1 /* 6LoWPAN fragment reassembly
5 * Alexander Aring <aar@pengutronix.de>
7 * Based on: net/ipv6/reassembly.c
9 * This program is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU General Public License
11 * as published by the Free Software Foundation; either version
12 * 2 of the License, or (at your option) any later version.
15 #define pr_fmt(fmt) "6LoWPAN: " fmt
17 #include <linux/net.h>
18 #include <linux/list.h>
19 #include <linux/netdevice.h>
20 #include <linux/random.h>
21 #include <linux/jhash.h>
22 #include <linux/skbuff.h>
23 #include <linux/slab.h>
24 #include <linux/export.h>
26 #include <net/ieee802154_netdev.h>
27 #include <net/6lowpan.h>
29 #include <net/inet_frag.h>
31 #include "6lowpan_i.h"
33 static const char lowpan_frags_cache_name
[] = "lowpan-frags";
35 static struct inet_frags lowpan_frags
;
37 static int lowpan_frag_reasm(struct lowpan_frag_queue
*fq
,
38 struct sk_buff
*prev
, struct net_device
*ldev
);
40 static unsigned int lowpan_hash_frag(u16 tag
, u16 d_size
,
41 const struct ieee802154_addr
*saddr
,
42 const struct ieee802154_addr
*daddr
)
44 net_get_random_once(&lowpan_frags
.rnd
, sizeof(lowpan_frags
.rnd
));
45 return jhash_3words(ieee802154_addr_hash(saddr
),
46 ieee802154_addr_hash(daddr
),
47 (__force u32
)(tag
+ (d_size
<< 16)),
51 static unsigned int lowpan_hashfn(const struct inet_frag_queue
*q
)
53 const struct lowpan_frag_queue
*fq
;
55 fq
= container_of(q
, struct lowpan_frag_queue
, q
);
56 return lowpan_hash_frag(fq
->tag
, fq
->d_size
, &fq
->saddr
, &fq
->daddr
);
59 static bool lowpan_frag_match(const struct inet_frag_queue
*q
, const void *a
)
61 const struct lowpan_frag_queue
*fq
;
62 const struct lowpan_create_arg
*arg
= a
;
64 fq
= container_of(q
, struct lowpan_frag_queue
, q
);
65 return fq
->tag
== arg
->tag
&& fq
->d_size
== arg
->d_size
&&
66 ieee802154_addr_equal(&fq
->saddr
, arg
->src
) &&
67 ieee802154_addr_equal(&fq
->daddr
, arg
->dst
);
70 static void lowpan_frag_init(struct inet_frag_queue
*q
, const void *a
)
72 const struct lowpan_create_arg
*arg
= a
;
73 struct lowpan_frag_queue
*fq
;
75 fq
= container_of(q
, struct lowpan_frag_queue
, q
);
78 fq
->d_size
= arg
->d_size
;
79 fq
->saddr
= *arg
->src
;
80 fq
->daddr
= *arg
->dst
;
83 static void lowpan_frag_expire(unsigned long data
)
85 struct frag_queue
*fq
;
88 fq
= container_of((struct inet_frag_queue
*)data
, struct frag_queue
, q
);
89 net
= container_of(fq
->q
.net
, struct net
, ieee802154_lowpan
.frags
);
91 spin_lock(&fq
->q
.lock
);
93 if (fq
->q
.flags
& INET_FRAG_COMPLETE
)
96 inet_frag_kill(&fq
->q
, &lowpan_frags
);
98 spin_unlock(&fq
->q
.lock
);
99 inet_frag_put(&fq
->q
, &lowpan_frags
);
102 static inline struct lowpan_frag_queue
*
103 fq_find(struct net
*net
, const struct lowpan_802154_cb
*cb
,
104 const struct ieee802154_addr
*src
,
105 const struct ieee802154_addr
*dst
)
107 struct inet_frag_queue
*q
;
108 struct lowpan_create_arg arg
;
110 struct netns_ieee802154_lowpan
*ieee802154_lowpan
=
111 net_ieee802154_lowpan(net
);
114 arg
.d_size
= cb
->d_size
;
118 hash
= lowpan_hash_frag(cb
->d_tag
, cb
->d_size
, src
, dst
);
120 q
= inet_frag_find(&ieee802154_lowpan
->frags
,
121 &lowpan_frags
, &arg
, hash
);
122 if (IS_ERR_OR_NULL(q
)) {
123 inet_frag_maybe_warn_overflow(q
, pr_fmt());
126 return container_of(q
, struct lowpan_frag_queue
, q
);
129 static int lowpan_frag_queue(struct lowpan_frag_queue
*fq
,
130 struct sk_buff
*skb
, u8 frag_type
)
132 struct sk_buff
*prev
, *next
;
133 struct net_device
*ldev
;
136 if (fq
->q
.flags
& INET_FRAG_COMPLETE
)
139 offset
= lowpan_802154_cb(skb
)->d_offset
<< 3;
140 end
= lowpan_802154_cb(skb
)->d_size
;
142 /* Is this the final fragment? */
143 if (offset
+ skb
->len
== end
) {
144 /* If we already have some bits beyond end
145 * or have different end, the segment is corrupted.
147 if (end
< fq
->q
.len
||
148 ((fq
->q
.flags
& INET_FRAG_LAST_IN
) && end
!= fq
->q
.len
))
150 fq
->q
.flags
|= INET_FRAG_LAST_IN
;
153 if (end
> fq
->q
.len
) {
154 /* Some bits beyond end -> corruption. */
155 if (fq
->q
.flags
& INET_FRAG_LAST_IN
)
161 /* Find out which fragments are in front and at the back of us
162 * in the chain of fragments so far. We must know where to put
163 * this fragment, right?
165 prev
= fq
->q
.fragments_tail
;
167 lowpan_802154_cb(prev
)->d_offset
<
168 lowpan_802154_cb(skb
)->d_offset
) {
173 for (next
= fq
->q
.fragments
; next
!= NULL
; next
= next
->next
) {
174 if (lowpan_802154_cb(next
)->d_offset
>=
175 lowpan_802154_cb(skb
)->d_offset
)
181 /* Insert this fragment in the chain of fragments. */
184 fq
->q
.fragments_tail
= skb
;
188 fq
->q
.fragments
= skb
;
194 fq
->q
.stamp
= skb
->tstamp
;
195 if (frag_type
== LOWPAN_DISPATCH_FRAG1
)
196 fq
->q
.flags
|= INET_FRAG_FIRST_IN
;
198 fq
->q
.meat
+= skb
->len
;
199 add_frag_mem_limit(fq
->q
.net
, skb
->truesize
);
201 if (fq
->q
.flags
== (INET_FRAG_FIRST_IN
| INET_FRAG_LAST_IN
) &&
202 fq
->q
.meat
== fq
->q
.len
) {
204 unsigned long orefdst
= skb
->_skb_refdst
;
206 skb
->_skb_refdst
= 0UL;
207 res
= lowpan_frag_reasm(fq
, prev
, ldev
);
208 skb
->_skb_refdst
= orefdst
;
218 /* Check if this packet is complete.
219 * Returns NULL on failure by any reason, and pointer
220 * to current nexthdr field in reassembled frame.
222 * It is called with locked fq, and caller must check that
223 * queue is eligible for reassembly i.e. it is not COMPLETE,
224 * the last and the first frames arrived and all the bits are here.
226 static int lowpan_frag_reasm(struct lowpan_frag_queue
*fq
, struct sk_buff
*prev
,
227 struct net_device
*ldev
)
229 struct sk_buff
*fp
, *head
= fq
->q
.fragments
;
232 inet_frag_kill(&fq
->q
, &lowpan_frags
);
234 /* Make the one we just received the head. */
237 fp
= skb_clone(head
, GFP_ATOMIC
);
242 fp
->next
= head
->next
;
244 fq
->q
.fragments_tail
= fp
;
247 skb_morph(head
, fq
->q
.fragments
);
248 head
->next
= fq
->q
.fragments
->next
;
250 consume_skb(fq
->q
.fragments
);
251 fq
->q
.fragments
= head
;
254 /* Head of list must not be cloned. */
255 if (skb_unclone(head
, GFP_ATOMIC
))
258 /* If the first fragment is fragmented itself, we split
259 * it to two chunks: the first with data and paged part
260 * and the second, holding only fragments.
262 if (skb_has_frag_list(head
)) {
263 struct sk_buff
*clone
;
266 clone
= alloc_skb(0, GFP_ATOMIC
);
269 clone
->next
= head
->next
;
271 skb_shinfo(clone
)->frag_list
= skb_shinfo(head
)->frag_list
;
272 skb_frag_list_init(head
);
273 for (i
= 0; i
< skb_shinfo(head
)->nr_frags
; i
++)
274 plen
+= skb_frag_size(&skb_shinfo(head
)->frags
[i
]);
275 clone
->len
= head
->data_len
- plen
;
276 clone
->data_len
= clone
->len
;
277 head
->data_len
-= clone
->len
;
278 head
->len
-= clone
->len
;
279 add_frag_mem_limit(fq
->q
.net
, clone
->truesize
);
282 WARN_ON(head
== NULL
);
284 sum_truesize
= head
->truesize
;
285 for (fp
= head
->next
; fp
;) {
288 struct sk_buff
*next
= fp
->next
;
290 sum_truesize
+= fp
->truesize
;
291 if (skb_try_coalesce(head
, fp
, &headstolen
, &delta
)) {
292 kfree_skb_partial(fp
, headstolen
);
294 if (!skb_shinfo(head
)->frag_list
)
295 skb_shinfo(head
)->frag_list
= fp
;
296 head
->data_len
+= fp
->len
;
297 head
->len
+= fp
->len
;
298 head
->truesize
+= fp
->truesize
;
302 sub_frag_mem_limit(fq
->q
.net
, sum_truesize
);
306 head
->tstamp
= fq
->q
.stamp
;
308 fq
->q
.fragments
= NULL
;
309 fq
->q
.fragments_tail
= NULL
;
313 net_dbg_ratelimited("lowpan_frag_reasm: no memory for reassembly\n");
317 static int lowpan_frag_rx_handlers_result(struct sk_buff
*skb
,
318 lowpan_rx_result res
)
322 return NET_RX_SUCCESS
;
324 /* nobody cared about this packet */
325 net_warn_ratelimited("%s: received unknown dispatch\n",
330 /* all others failure */
335 static lowpan_rx_result
lowpan_frag_rx_h_iphc(struct sk_buff
*skb
)
339 if (!lowpan_is_iphc(*skb_network_header(skb
)))
342 ret
= lowpan_iphc_decompress(skb
);
349 static int lowpan_invoke_frag_rx_handlers(struct sk_buff
*skb
)
351 lowpan_rx_result res
;
353 #define CALL_RXH(rxh) \
356 if (res != RX_CONTINUE) \
360 /* likely at first */
361 CALL_RXH(lowpan_frag_rx_h_iphc
);
362 CALL_RXH(lowpan_rx_h_ipv6
);
365 return lowpan_frag_rx_handlers_result(skb
, res
);
369 #define LOWPAN_FRAG_DGRAM_SIZE_HIGH_MASK 0x07
370 #define LOWPAN_FRAG_DGRAM_SIZE_HIGH_SHIFT 8
372 static int lowpan_get_cb(struct sk_buff
*skb
, u8 frag_type
,
373 struct lowpan_802154_cb
*cb
)
376 u8 high
= 0, low
= 0;
379 fail
= lowpan_fetch_skb(skb
, &high
, 1);
380 fail
|= lowpan_fetch_skb(skb
, &low
, 1);
381 /* remove the dispatch value and use first three bits as high value
382 * for the datagram size
384 cb
->d_size
= (high
& LOWPAN_FRAG_DGRAM_SIZE_HIGH_MASK
) <<
385 LOWPAN_FRAG_DGRAM_SIZE_HIGH_SHIFT
| low
;
386 fail
|= lowpan_fetch_skb(skb
, &d_tag
, 2);
387 cb
->d_tag
= ntohs(d_tag
);
389 if (frag_type
== LOWPAN_DISPATCH_FRAGN
) {
390 fail
|= lowpan_fetch_skb(skb
, &cb
->d_offset
, 1);
392 skb_reset_network_header(skb
);
394 /* check if datagram_size has ipv6hdr on FRAG1 */
395 fail
|= cb
->d_size
< sizeof(struct ipv6hdr
);
396 /* check if we can dereference the dispatch value */
406 int lowpan_frag_rcv(struct sk_buff
*skb
, u8 frag_type
)
408 struct lowpan_frag_queue
*fq
;
409 struct net
*net
= dev_net(skb
->dev
);
410 struct lowpan_802154_cb
*cb
= lowpan_802154_cb(skb
);
411 struct ieee802154_hdr hdr
;
414 if (ieee802154_hdr_peek_addrs(skb
, &hdr
) < 0)
417 err
= lowpan_get_cb(skb
, frag_type
, cb
);
421 if (frag_type
== LOWPAN_DISPATCH_FRAG1
) {
422 err
= lowpan_invoke_frag_rx_handlers(skb
);
423 if (err
== NET_RX_DROP
)
427 if (cb
->d_size
> IPV6_MIN_MTU
) {
428 net_warn_ratelimited("lowpan_frag_rcv: datagram size exceeds MTU\n");
432 fq
= fq_find(net
, cb
, &hdr
.source
, &hdr
.dest
);
436 spin_lock(&fq
->q
.lock
);
437 ret
= lowpan_frag_queue(fq
, skb
, frag_type
);
438 spin_unlock(&fq
->q
.lock
);
440 inet_frag_put(&fq
->q
, &lowpan_frags
);
452 static struct ctl_table lowpan_frags_ns_ctl_table
[] = {
454 .procname
= "6lowpanfrag_high_thresh",
455 .data
= &init_net
.ieee802154_lowpan
.frags
.high_thresh
,
456 .maxlen
= sizeof(int),
458 .proc_handler
= proc_dointvec_minmax
,
459 .extra1
= &init_net
.ieee802154_lowpan
.frags
.low_thresh
462 .procname
= "6lowpanfrag_low_thresh",
463 .data
= &init_net
.ieee802154_lowpan
.frags
.low_thresh
,
464 .maxlen
= sizeof(int),
466 .proc_handler
= proc_dointvec_minmax
,
468 .extra2
= &init_net
.ieee802154_lowpan
.frags
.high_thresh
471 .procname
= "6lowpanfrag_time",
472 .data
= &init_net
.ieee802154_lowpan
.frags
.timeout
,
473 .maxlen
= sizeof(int),
475 .proc_handler
= proc_dointvec_jiffies
,
480 /* secret interval has been deprecated */
481 static int lowpan_frags_secret_interval_unused
;
482 static struct ctl_table lowpan_frags_ctl_table
[] = {
484 .procname
= "6lowpanfrag_secret_interval",
485 .data
= &lowpan_frags_secret_interval_unused
,
486 .maxlen
= sizeof(int),
488 .proc_handler
= proc_dointvec_jiffies
,
493 static int __net_init
lowpan_frags_ns_sysctl_register(struct net
*net
)
495 struct ctl_table
*table
;
496 struct ctl_table_header
*hdr
;
497 struct netns_ieee802154_lowpan
*ieee802154_lowpan
=
498 net_ieee802154_lowpan(net
);
500 table
= lowpan_frags_ns_ctl_table
;
501 if (!net_eq(net
, &init_net
)) {
502 table
= kmemdup(table
, sizeof(lowpan_frags_ns_ctl_table
),
507 table
[0].data
= &ieee802154_lowpan
->frags
.high_thresh
;
508 table
[0].extra1
= &ieee802154_lowpan
->frags
.low_thresh
;
509 table
[0].extra2
= &init_net
.ieee802154_lowpan
.frags
.high_thresh
;
510 table
[1].data
= &ieee802154_lowpan
->frags
.low_thresh
;
511 table
[1].extra2
= &ieee802154_lowpan
->frags
.high_thresh
;
512 table
[2].data
= &ieee802154_lowpan
->frags
.timeout
;
514 /* Don't export sysctls to unprivileged users */
515 if (net
->user_ns
!= &init_user_ns
)
516 table
[0].procname
= NULL
;
519 hdr
= register_net_sysctl(net
, "net/ieee802154/6lowpan", table
);
523 ieee802154_lowpan
->sysctl
.frags_hdr
= hdr
;
527 if (!net_eq(net
, &init_net
))
533 static void __net_exit
lowpan_frags_ns_sysctl_unregister(struct net
*net
)
535 struct ctl_table
*table
;
536 struct netns_ieee802154_lowpan
*ieee802154_lowpan
=
537 net_ieee802154_lowpan(net
);
539 table
= ieee802154_lowpan
->sysctl
.frags_hdr
->ctl_table_arg
;
540 unregister_net_sysctl_table(ieee802154_lowpan
->sysctl
.frags_hdr
);
541 if (!net_eq(net
, &init_net
))
545 static struct ctl_table_header
*lowpan_ctl_header
;
547 static int __init
lowpan_frags_sysctl_register(void)
549 lowpan_ctl_header
= register_net_sysctl(&init_net
,
550 "net/ieee802154/6lowpan",
551 lowpan_frags_ctl_table
);
552 return lowpan_ctl_header
== NULL
? -ENOMEM
: 0;
555 static void lowpan_frags_sysctl_unregister(void)
557 unregister_net_sysctl_table(lowpan_ctl_header
);
560 static inline int lowpan_frags_ns_sysctl_register(struct net
*net
)
565 static inline void lowpan_frags_ns_sysctl_unregister(struct net
*net
)
569 static inline int __init
lowpan_frags_sysctl_register(void)
574 static inline void lowpan_frags_sysctl_unregister(void)
579 static int __net_init
lowpan_frags_init_net(struct net
*net
)
581 struct netns_ieee802154_lowpan
*ieee802154_lowpan
=
582 net_ieee802154_lowpan(net
);
585 ieee802154_lowpan
->frags
.high_thresh
= IPV6_FRAG_HIGH_THRESH
;
586 ieee802154_lowpan
->frags
.low_thresh
= IPV6_FRAG_LOW_THRESH
;
587 ieee802154_lowpan
->frags
.timeout
= IPV6_FRAG_TIMEOUT
;
589 res
= inet_frags_init_net(&ieee802154_lowpan
->frags
);
592 res
= lowpan_frags_ns_sysctl_register(net
);
594 inet_frags_uninit_net(&ieee802154_lowpan
->frags
);
598 static void __net_exit
lowpan_frags_exit_net(struct net
*net
)
600 struct netns_ieee802154_lowpan
*ieee802154_lowpan
=
601 net_ieee802154_lowpan(net
);
603 lowpan_frags_ns_sysctl_unregister(net
);
604 inet_frags_exit_net(&ieee802154_lowpan
->frags
, &lowpan_frags
);
607 static struct pernet_operations lowpan_frags_ops
= {
608 .init
= lowpan_frags_init_net
,
609 .exit
= lowpan_frags_exit_net
,
612 int __init
lowpan_net_frag_init(void)
616 ret
= lowpan_frags_sysctl_register();
620 ret
= register_pernet_subsys(&lowpan_frags_ops
);
624 lowpan_frags
.hashfn
= lowpan_hashfn
;
625 lowpan_frags
.constructor
= lowpan_frag_init
;
626 lowpan_frags
.destructor
= NULL
;
627 lowpan_frags
.skb_free
= NULL
;
628 lowpan_frags
.qsize
= sizeof(struct frag_queue
);
629 lowpan_frags
.match
= lowpan_frag_match
;
630 lowpan_frags
.frag_expire
= lowpan_frag_expire
;
631 lowpan_frags
.frags_cache_name
= lowpan_frags_cache_name
;
632 ret
= inet_frags_init(&lowpan_frags
);
638 lowpan_frags_sysctl_unregister();
642 void lowpan_net_frag_exit(void)
644 inet_frags_fini(&lowpan_frags
);
645 lowpan_frags_sysctl_unregister();
646 unregister_pernet_subsys(&lowpan_frags_ops
);