cfpkt_skbuff.c 8.71 KB
Newer Older
1 2
/*
 * Copyright (C) ST-Ericsson AB 2010
3
 * Author:	Sjur Brendeland
4 5 6
 * License terms: GNU General Public License (GPL) version 2
 */

Joe Perches's avatar
Joe Perches committed
7 8
#define pr_fmt(fmt) KBUILD_MODNAME ":%s(): " fmt, __func__

9 10 11
#include <linux/string.h>
#include <linux/skbuff.h>
#include <linux/hardirq.h>
12
#include <linux/export.h>
13 14
#include <net/caif/cfpkt.h>

15
#define PKT_PREFIX  48
16
#define PKT_POSTFIX 2
17
#define PKT_LEN_WHEN_EXTENDING 128
Joe Perches's avatar
Joe Perches committed
18 19 20 21 22 23
#define PKT_ERROR(pkt, errmsg)		   \
do {					   \
	cfpkt_priv(pkt)->erronous = true;  \
	skb_reset_tail_pointer(&pkt->skb); \
	pr_warn(errmsg);		   \
} while (0)
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45

struct cfpktq {
	struct sk_buff_head head;
	atomic_t count;
	/* Lock protects count updates */
	spinlock_t lock;
};

/*
 * net/caif/ is generic and does not
 * understand SKB, so we do this typecast
 */
struct cfpkt {
	struct sk_buff skb;
};

/* Private data inside SKB */
struct cfpkt_priv_data {
	struct dev_info dev_info;
	bool erronous;
};

Stephen Hemminger's avatar
Stephen Hemminger committed
46
static inline struct cfpkt_priv_data *cfpkt_priv(struct cfpkt *pkt)
47 48 49 50
{
	return (struct cfpkt_priv_data *) pkt->skb.cb;
}

Stephen Hemminger's avatar
Stephen Hemminger committed
51
static inline bool is_erronous(struct cfpkt *pkt)
52 53 54 55
{
	return cfpkt_priv(pkt)->erronous;
}

Stephen Hemminger's avatar
Stephen Hemminger committed
56
static inline struct sk_buff *pkt_to_skb(struct cfpkt *pkt)
57 58 59 60
{
	return &pkt->skb;
}

Stephen Hemminger's avatar
Stephen Hemminger committed
61
static inline struct cfpkt *skb_to_pkt(struct sk_buff *skb)
62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105
{
	return (struct cfpkt *) skb;
}

struct cfpkt *cfpkt_fromnative(enum caif_direction dir, void *nativepkt)
{
	struct cfpkt *pkt = skb_to_pkt(nativepkt);
	cfpkt_priv(pkt)->erronous = false;
	return pkt;
}
EXPORT_SYMBOL(cfpkt_fromnative);

void *cfpkt_tonative(struct cfpkt *pkt)
{
	return (void *) pkt;
}
EXPORT_SYMBOL(cfpkt_tonative);

static struct cfpkt *cfpkt_create_pfx(u16 len, u16 pfx)
{
	struct sk_buff *skb;

	if (likely(in_interrupt()))
		skb = alloc_skb(len + pfx, GFP_ATOMIC);
	else
		skb = alloc_skb(len + pfx, GFP_KERNEL);

	if (unlikely(skb == NULL))
		return NULL;

	skb_reserve(skb, pfx);
	return skb_to_pkt(skb);
}

inline struct cfpkt *cfpkt_create(u16 len)
{
	return cfpkt_create_pfx(len + PKT_POSTFIX, PKT_PREFIX);
}

void cfpkt_destroy(struct cfpkt *pkt)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	kfree_skb(skb);
}
106

107 108 109 110 111
inline bool cfpkt_more(struct cfpkt *pkt)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	return skb->len > 0;
}
112

113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
int cfpkt_peek_head(struct cfpkt *pkt, void *data, u16 len)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	if (skb_headlen(skb) >= len) {
		memcpy(data, skb->data, len);
		return 0;
	}
	return !cfpkt_extr_head(pkt, data, len) &&
	    !cfpkt_add_head(pkt, data, len);
}

int cfpkt_extr_head(struct cfpkt *pkt, void *data, u16 len)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	u8 *from;
	if (unlikely(is_erronous(pkt)))
		return -EPROTO;

	if (unlikely(len > skb->len)) {
Joe Perches's avatar
Joe Perches committed
132
		PKT_ERROR(pkt, "read beyond end of packet\n");
133 134 135 136 137
		return -EPROTO;
	}

	if (unlikely(len > skb_headlen(skb))) {
		if (unlikely(skb_linearize(skb) != 0)) {
Joe Perches's avatar
Joe Perches committed
138
			PKT_ERROR(pkt, "linearize failed\n");
139 140 141 142 143
			return -EPROTO;
		}
	}
	from = skb_pull(skb, len);
	from -= len;
144 145
	if (data)
		memcpy(data, from, len);
146 147
	return 0;
}
148
EXPORT_SYMBOL(cfpkt_extr_head);
149 150 151 152 153 154 155 156 157 158

int cfpkt_extr_trail(struct cfpkt *pkt, void *dta, u16 len)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	u8 *data = dta;
	u8 *from;
	if (unlikely(is_erronous(pkt)))
		return -EPROTO;

	if (unlikely(skb_linearize(skb) != 0)) {
Joe Perches's avatar
Joe Perches committed
159
		PKT_ERROR(pkt, "linearize failed\n");
160 161 162
		return -EPROTO;
	}
	if (unlikely(skb->data + len > skb_tail_pointer(skb))) {
Joe Perches's avatar
Joe Perches committed
163
		PKT_ERROR(pkt, "read beyond end of packet\n");
164 165 166 167 168 169 170
		return -EPROTO;
	}
	from = skb_tail_pointer(skb) - len;
	skb_trim(skb, skb->len - len);
	memcpy(data, from, len);
	return 0;
}
171

172 173 174 175
int cfpkt_pad_trail(struct cfpkt *pkt, u16 len)
{
	return cfpkt_add_body(pkt, NULL, len);
}
176

177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
int cfpkt_add_body(struct cfpkt *pkt, const void *data, u16 len)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	struct sk_buff *lastskb;
	u8 *to;
	u16 addlen = 0;


	if (unlikely(is_erronous(pkt)))
		return -EPROTO;

	lastskb = skb;

	/* Check whether we need to add space at the tail */
	if (unlikely(skb_tailroom(skb) < len)) {
		if (likely(len < PKT_LEN_WHEN_EXTENDING))
			addlen = PKT_LEN_WHEN_EXTENDING;
		else
			addlen = len;
	}

	/* Check whether we need to change the SKB before writing to the tail */
	if (unlikely((addlen > 0) || skb_cloned(skb) || skb_shared(skb))) {

		/* Make sure data is writable */
		if (unlikely(skb_cow_data(skb, addlen, &lastskb) < 0)) {
Joe Perches's avatar
Joe Perches committed
203
			PKT_ERROR(pkt, "cow failed\n");
204 205 206 207 208 209 210 211
			return -EPROTO;
		}
		/*
		 * Is the SKB non-linear after skb_cow_data()? If so, we are
		 * going to add data to the last SKB, so we need to adjust
		 * lengths of the top SKB.
		 */
		if (lastskb != skb) {
Joe Perches's avatar
Joe Perches committed
212
			pr_warn("Packet is non-linear\n");
213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235
			skb->len += len;
			skb->data_len += len;
		}
	}

	/* All set to put the last SKB and optionally write data there. */
	to = skb_put(lastskb, len);
	if (likely(data))
		memcpy(to, data, len);
	return 0;
}

inline int cfpkt_addbdy(struct cfpkt *pkt, u8 data)
{
	return cfpkt_add_body(pkt, &data, 1);
}

int cfpkt_add_head(struct cfpkt *pkt, const void *data2, u16 len)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	struct sk_buff *lastskb;
	u8 *to;
	const u8 *data = data2;
236
	int ret;
237 238 239
	if (unlikely(is_erronous(pkt)))
		return -EPROTO;
	if (unlikely(skb_headroom(skb) < len)) {
Joe Perches's avatar
Joe Perches committed
240
		PKT_ERROR(pkt, "no headroom\n");
241 242 243 244
		return -EPROTO;
	}

	/* Make sure data is writable */
245 246
	ret = skb_cow_data(skb, 0, &lastskb);
	if (unlikely(ret < 0)) {
Joe Perches's avatar
Joe Perches committed
247
		PKT_ERROR(pkt, "cow failed\n");
248
		return ret;
249 250 251 252 253 254
	}

	to = skb_push(skb, len);
	memcpy(to, data, len);
	return 0;
}
255
EXPORT_SYMBOL(cfpkt_add_head);
256 257 258 259 260

inline int cfpkt_add_trail(struct cfpkt *pkt, const void *data, u16 len)
{
	return cfpkt_add_body(pkt, data, len);
}
261

262 263 264 265 266
inline u16 cfpkt_getlen(struct cfpkt *pkt)
{
	struct sk_buff *skb = pkt_to_skb(pkt);
	return skb->len;
}
267

268
inline u16 cfpkt_iterate(struct cfpkt *pkt,
269 270
			 u16 (*iter_func)(u16, void *, u16),
			 u16 data)
271 272 273 274 275 276 277 278
{
	/*
	 * Don't care about the performance hit of linearizing,
	 * Checksum should not be used on high-speed interfaces anyway.
	 */
	if (unlikely(is_erronous(pkt)))
		return -EPROTO;
	if (unlikely(skb_linearize(&pkt->skb) != 0)) {
Joe Perches's avatar
Joe Perches committed
279
		PKT_ERROR(pkt, "linearize failed\n");
280 281 282 283
		return -EPROTO;
	}
	return iter_func(data, pkt->skb.data, cfpkt_getlen(pkt));
}
284

285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
int cfpkt_setlen(struct cfpkt *pkt, u16 len)
{
	struct sk_buff *skb = pkt_to_skb(pkt);


	if (unlikely(is_erronous(pkt)))
		return -EPROTO;

	if (likely(len <= skb->len)) {
		if (unlikely(skb->data_len))
			___pskb_trim(skb, len);
		else
			skb_trim(skb, len);

			return cfpkt_getlen(pkt);
	}

	/* Need to expand SKB */
	if (unlikely(!cfpkt_pad_trail(pkt, len - skb->len)))
Joe Perches's avatar
Joe Perches committed
304
		PKT_ERROR(pkt, "skb_pad_trail failed\n");
305 306 307 308 309

	return cfpkt_getlen(pkt);
}

struct cfpkt *cfpkt_append(struct cfpkt *dstpkt,
310 311
			   struct cfpkt *addpkt,
			   u16 expectlen)
312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329
{
	struct sk_buff *dst = pkt_to_skb(dstpkt);
	struct sk_buff *add = pkt_to_skb(addpkt);
	u16 addlen = skb_headlen(add);
	u16 neededtailspace;
	struct sk_buff *tmp;
	u16 dstlen;
	u16 createlen;
	if (unlikely(is_erronous(dstpkt) || is_erronous(addpkt))) {
		return dstpkt;
	}
	if (expectlen > addlen)
		neededtailspace = expectlen;
	else
		neededtailspace = addlen;

	if (dst->tail + neededtailspace > dst->end) {
		/* Create a dumplicate of 'dst' with more tail space */
330
		struct cfpkt *tmppkt;
331 332
		dstlen = skb_headlen(dst);
		createlen = dstlen + neededtailspace;
333 334
		tmppkt = cfpkt_create(createlen + PKT_PREFIX + PKT_POSTFIX);
		if (tmppkt == NULL)
335
			return NULL;
336
		tmp = pkt_to_skb(tmppkt);
337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
		skb_set_tail_pointer(tmp, dstlen);
		tmp->len = dstlen;
		memcpy(tmp->data, dst->data, dstlen);
		cfpkt_destroy(dstpkt);
		dst = tmp;
	}
	memcpy(skb_tail_pointer(dst), add->data, skb_headlen(add));
	cfpkt_destroy(addpkt);
	dst->tail += addlen;
	dst->len += addlen;
	return skb_to_pkt(dst);
}

struct cfpkt *cfpkt_split(struct cfpkt *pkt, u16 pos)
{
	struct sk_buff *skb2;
	struct sk_buff *skb = pkt_to_skb(pkt);
354
	struct cfpkt *tmppkt;
355 356 357 358 359 360 361
	u8 *split = skb->data + pos;
	u16 len2nd = skb_tail_pointer(skb) - split;

	if (unlikely(is_erronous(pkt)))
		return NULL;

	if (skb->data + pos > skb_tail_pointer(skb)) {
Joe Perches's avatar
Joe Perches committed
362
		PKT_ERROR(pkt, "trying to split beyond end of packet\n");
363 364 365 366
		return NULL;
	}

	/* Create a new packet for the second part of the data */
367 368 369 370 371 372
	tmppkt = cfpkt_create_pfx(len2nd + PKT_PREFIX + PKT_POSTFIX,
				  PKT_PREFIX);
	if (tmppkt == NULL)
		return NULL;
	skb2 = pkt_to_skb(tmppkt);

373 374 375 376 377 378 379 380 381 382 383

	if (skb2 == NULL)
		return NULL;

	/* Reduce the length of the original packet */
	skb_set_tail_pointer(skb, pos);
	skb->len = pos;

	memcpy(skb2->data, split, len2nd);
	skb2->tail += len2nd;
	skb2->len += len2nd;
384
	skb2->priority = skb->priority;
385 386 387
	return skb_to_pkt(skb2);
}

Stephen Hemminger's avatar
Stephen Hemminger committed
388
bool cfpkt_erroneous(struct cfpkt *pkt)
389 390 391 392 393 394 395 396
{
	return cfpkt_priv(pkt)->erronous;
}

struct caif_payload_info *cfpkt_info(struct cfpkt *pkt)
{
	return (struct caif_payload_info *)&pkt_to_skb(pkt)->cb;
}
397
EXPORT_SYMBOL(cfpkt_info);
398 399 400 401 402 403

void cfpkt_set_prio(struct cfpkt *pkt, int prio)
{
	pkt_to_skb(pkt)->priority = prio;
}
EXPORT_SYMBOL(cfpkt_set_prio);