1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20#include <linux/net.h>
21#include <linux/capability.h>
22#include <linux/module.h>
23#include <linux/if_arp.h>
24#include <linux/if.h>
25#include <linux/termios.h>
26#include <linux/list.h>
27#include <linux/slab.h>
28#include <net/datalink.h>
29#include <net/psnap.h>
30#include <net/sock.h>
31#include <net/tcp_states.h>
32#include <net/route.h>
33
34#include <net/af_ieee802154.h>
35#include <net/ieee802154_netdev.h>
36
37
38static struct net_device*
39ieee802154_get_dev(struct net *net, const struct ieee802154_addr *addr)
40{
41 struct net_device *dev = NULL;
42 struct net_device *tmp;
43 __le16 pan_id, short_addr;
44 u8 hwaddr[IEEE802154_ADDR_LEN];
45
46 switch (addr->mode) {
47 case IEEE802154_ADDR_LONG:
48 ieee802154_devaddr_to_raw(hwaddr, addr->extended_addr);
49 rcu_read_lock();
50 dev = dev_getbyhwaddr_rcu(net, ARPHRD_IEEE802154, hwaddr);
51 if (dev)
52 dev_hold(dev);
53 rcu_read_unlock();
54 break;
55 case IEEE802154_ADDR_SHORT:
56 if (addr->pan_id == cpu_to_le16(IEEE802154_PANID_BROADCAST) ||
57 addr->short_addr == cpu_to_le16(IEEE802154_ADDR_UNDEF) ||
58 addr->short_addr == cpu_to_le16(IEEE802154_ADDR_BROADCAST))
59 break;
60
61 rtnl_lock();
62
63 for_each_netdev(net, tmp) {
64 if (tmp->type != ARPHRD_IEEE802154)
65 continue;
66
67 pan_id = tmp->ieee802154_ptr->pan_id;
68 short_addr = tmp->ieee802154_ptr->short_addr;
69 if (pan_id == addr->pan_id &&
70 short_addr == addr->short_addr) {
71 dev = tmp;
72 dev_hold(dev);
73 break;
74 }
75 }
76
77 rtnl_unlock();
78 break;
79 default:
80 pr_warn("Unsupported ieee802154 address type: %d\n",
81 addr->mode);
82 break;
83 }
84
85 return dev;
86}
87
88static int ieee802154_sock_release(struct socket *sock)
89{
90 struct sock *sk = sock->sk;
91
92 if (sk) {
93 sock->sk = NULL;
94 sk->sk_prot->close(sk, 0);
95 }
96 return 0;
97}
98
99static int ieee802154_sock_sendmsg(struct socket *sock, struct msghdr *msg,
100 size_t len)
101{
102 struct sock *sk = sock->sk;
103
104 return sk->sk_prot->sendmsg(sk, msg, len);
105}
106
107static int ieee802154_sock_bind(struct socket *sock, struct sockaddr *uaddr,
108 int addr_len)
109{
110 struct sock *sk = sock->sk;
111
112 if (sk->sk_prot->bind)
113 return sk->sk_prot->bind(sk, uaddr, addr_len);
114
115 return sock_no_bind(sock, uaddr, addr_len);
116}
117
118static int ieee802154_sock_connect(struct socket *sock, struct sockaddr *uaddr,
119 int addr_len, int flags)
120{
121 struct sock *sk = sock->sk;
122
123 if (addr_len < sizeof(uaddr->sa_family))
124 return -EINVAL;
125
126 if (uaddr->sa_family == AF_UNSPEC)
127 return sk->sk_prot->disconnect(sk, flags);
128
129 return sk->sk_prot->connect(sk, uaddr, addr_len);
130}
131
132static int ieee802154_dev_ioctl(struct sock *sk, struct ifreq __user *arg,
133 unsigned int cmd)
134{
135 struct ifreq ifr;
136 int ret = -ENOIOCTLCMD;
137 struct net_device *dev;
138
139 if (copy_from_user(&ifr, arg, sizeof(struct ifreq)))
140 return -EFAULT;
141
142 ifr.ifr_name[IFNAMSIZ-1] = 0;
143
144 dev_load(sock_net(sk), ifr.ifr_name);
145 dev = dev_get_by_name(sock_net(sk), ifr.ifr_name);
146
147 if (!dev)
148 return -ENODEV;
149
150 if (dev->type == ARPHRD_IEEE802154 && dev->netdev_ops->ndo_do_ioctl)
151 ret = dev->netdev_ops->ndo_do_ioctl(dev, &ifr, cmd);
152
153 if (!ret && copy_to_user(arg, &ifr, sizeof(struct ifreq)))
154 ret = -EFAULT;
155 dev_put(dev);
156
157 return ret;
158}
159
160static int ieee802154_sock_ioctl(struct socket *sock, unsigned int cmd,
161 unsigned long arg)
162{
163 struct sock *sk = sock->sk;
164
165 switch (cmd) {
166 case SIOCGSTAMP:
167 return sock_get_timestamp(sk, (struct timeval __user *)arg);
168 case SIOCGSTAMPNS:
169 return sock_get_timestampns(sk, (struct timespec __user *)arg);
170 case SIOCGIFADDR:
171 case SIOCSIFADDR:
172 return ieee802154_dev_ioctl(sk, (struct ifreq __user *)arg,
173 cmd);
174 default:
175 if (!sk->sk_prot->ioctl)
176 return -ENOIOCTLCMD;
177 return sk->sk_prot->ioctl(sk, cmd, arg);
178 }
179}
180
181
182static HLIST_HEAD(raw_head);
183static DEFINE_RWLOCK(raw_lock);
184
185static int raw_hash(struct sock *sk)
186{
187 write_lock_bh(&raw_lock);
188 sk_add_node(sk, &raw_head);
189 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
190 write_unlock_bh(&raw_lock);
191
192 return 0;
193}
194
195static void raw_unhash(struct sock *sk)
196{
197 write_lock_bh(&raw_lock);
198 if (sk_del_node_init(sk))
199 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
200 write_unlock_bh(&raw_lock);
201}
202
203static void raw_close(struct sock *sk, long timeout)
204{
205 sk_common_release(sk);
206}
207
208static int raw_bind(struct sock *sk, struct sockaddr *_uaddr, int len)
209{
210 struct ieee802154_addr addr;
211 struct sockaddr_ieee802154 *uaddr = (struct sockaddr_ieee802154 *)_uaddr;
212 int err = 0;
213 struct net_device *dev = NULL;
214
215 if (len < sizeof(*uaddr))
216 return -EINVAL;
217
218 uaddr = (struct sockaddr_ieee802154 *)_uaddr;
219 if (uaddr->family != AF_IEEE802154)
220 return -EINVAL;
221
222 lock_sock(sk);
223
224 ieee802154_addr_from_sa(&addr, &uaddr->addr);
225 dev = ieee802154_get_dev(sock_net(sk), &addr);
226 if (!dev) {
227 err = -ENODEV;
228 goto out;
229 }
230
231 sk->sk_bound_dev_if = dev->ifindex;
232 sk_dst_reset(sk);
233
234 dev_put(dev);
235out:
236 release_sock(sk);
237
238 return err;
239}
240
241static int raw_connect(struct sock *sk, struct sockaddr *uaddr,
242 int addr_len)
243{
244 return -ENOTSUPP;
245}
246
247static int raw_disconnect(struct sock *sk, int flags)
248{
249 return 0;
250}
251
252static int raw_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
253{
254 struct net_device *dev;
255 unsigned int mtu;
256 struct sk_buff *skb;
257 int hlen, tlen;
258 int err;
259
260 if (msg->msg_flags & MSG_OOB) {
261 pr_debug("msg->msg_flags = 0x%x\n", msg->msg_flags);
262 return -EOPNOTSUPP;
263 }
264
265 lock_sock(sk);
266 if (!sk->sk_bound_dev_if)
267 dev = dev_getfirstbyhwtype(sock_net(sk), ARPHRD_IEEE802154);
268 else
269 dev = dev_get_by_index(sock_net(sk), sk->sk_bound_dev_if);
270 release_sock(sk);
271
272 if (!dev) {
273 pr_debug("no dev\n");
274 err = -ENXIO;
275 goto out;
276 }
277
278 mtu = IEEE802154_MTU;
279 pr_debug("name = %s, mtu = %u\n", dev->name, mtu);
280
281 if (size > mtu) {
282 pr_debug("size = %zu, mtu = %u\n", size, mtu);
283 err = -EMSGSIZE;
284 goto out_dev;
285 }
286
287 hlen = LL_RESERVED_SPACE(dev);
288 tlen = dev->needed_tailroom;
289 skb = sock_alloc_send_skb(sk, hlen + tlen + size,
290 msg->msg_flags & MSG_DONTWAIT, &err);
291 if (!skb)
292 goto out_dev;
293
294 skb_reserve(skb, hlen);
295
296 skb_reset_mac_header(skb);
297 skb_reset_network_header(skb);
298
299 err = memcpy_from_msg(skb_put(skb, size), msg, size);
300 if (err < 0)
301 goto out_skb;
302
303 skb->dev = dev;
304 skb->protocol = htons(ETH_P_IEEE802154);
305
306 err = dev_queue_xmit(skb);
307 if (err > 0)
308 err = net_xmit_errno(err);
309
310 dev_put(dev);
311
312 return err ?: size;
313
314out_skb:
315 kfree_skb(skb);
316out_dev:
317 dev_put(dev);
318out:
319 return err;
320}
321
322static int raw_recvmsg(struct sock *sk, struct msghdr *msg, size_t len,
323 int noblock, int flags, int *addr_len)
324{
325 size_t copied = 0;
326 int err = -EOPNOTSUPP;
327 struct sk_buff *skb;
328
329 skb = skb_recv_datagram(sk, flags, noblock, &err);
330 if (!skb)
331 goto out;
332
333 copied = skb->len;
334 if (len < copied) {
335 msg->msg_flags |= MSG_TRUNC;
336 copied = len;
337 }
338
339 err = skb_copy_datagram_msg(skb, 0, msg, copied);
340 if (err)
341 goto done;
342
343 sock_recv_ts_and_drops(msg, sk, skb);
344
345 if (flags & MSG_TRUNC)
346 copied = skb->len;
347done:
348 skb_free_datagram(sk, skb);
349out:
350 if (err)
351 return err;
352 return copied;
353}
354
355static int raw_rcv_skb(struct sock *sk, struct sk_buff *skb)
356{
357 skb = skb_share_check(skb, GFP_ATOMIC);
358 if (!skb)
359 return NET_RX_DROP;
360
361 if (sock_queue_rcv_skb(sk, skb) < 0) {
362 kfree_skb(skb);
363 return NET_RX_DROP;
364 }
365
366 return NET_RX_SUCCESS;
367}
368
369static void ieee802154_raw_deliver(struct net_device *dev, struct sk_buff *skb)
370{
371 struct sock *sk;
372
373 read_lock(&raw_lock);
374 sk_for_each(sk, &raw_head) {
375 bh_lock_sock(sk);
376 if (!sk->sk_bound_dev_if ||
377 sk->sk_bound_dev_if == dev->ifindex) {
378 struct sk_buff *clone;
379
380 clone = skb_clone(skb, GFP_ATOMIC);
381 if (clone)
382 raw_rcv_skb(sk, clone);
383 }
384 bh_unlock_sock(sk);
385 }
386 read_unlock(&raw_lock);
387}
388
389static int raw_getsockopt(struct sock *sk, int level, int optname,
390 char __user *optval, int __user *optlen)
391{
392 return -EOPNOTSUPP;
393}
394
395static int raw_setsockopt(struct sock *sk, int level, int optname,
396 char __user *optval, unsigned int optlen)
397{
398 return -EOPNOTSUPP;
399}
400
401static struct proto ieee802154_raw_prot = {
402 .name = "IEEE-802.15.4-RAW",
403 .owner = THIS_MODULE,
404 .obj_size = sizeof(struct sock),
405 .close = raw_close,
406 .bind = raw_bind,
407 .sendmsg = raw_sendmsg,
408 .recvmsg = raw_recvmsg,
409 .hash = raw_hash,
410 .unhash = raw_unhash,
411 .connect = raw_connect,
412 .disconnect = raw_disconnect,
413 .getsockopt = raw_getsockopt,
414 .setsockopt = raw_setsockopt,
415};
416
417static const struct proto_ops ieee802154_raw_ops = {
418 .family = PF_IEEE802154,
419 .owner = THIS_MODULE,
420 .release = ieee802154_sock_release,
421 .bind = ieee802154_sock_bind,
422 .connect = ieee802154_sock_connect,
423 .socketpair = sock_no_socketpair,
424 .accept = sock_no_accept,
425 .getname = sock_no_getname,
426 .poll = datagram_poll,
427 .ioctl = ieee802154_sock_ioctl,
428 .listen = sock_no_listen,
429 .shutdown = sock_no_shutdown,
430 .setsockopt = sock_common_setsockopt,
431 .getsockopt = sock_common_getsockopt,
432 .sendmsg = ieee802154_sock_sendmsg,
433 .recvmsg = sock_common_recvmsg,
434 .mmap = sock_no_mmap,
435 .sendpage = sock_no_sendpage,
436#ifdef CONFIG_COMPAT
437 .compat_setsockopt = compat_sock_common_setsockopt,
438 .compat_getsockopt = compat_sock_common_getsockopt,
439#endif
440};
441
442
443static HLIST_HEAD(dgram_head);
444static DEFINE_RWLOCK(dgram_lock);
445
446struct dgram_sock {
447 struct sock sk;
448
449 struct ieee802154_addr src_addr;
450 struct ieee802154_addr dst_addr;
451
452 unsigned int bound:1;
453 unsigned int connected:1;
454 unsigned int want_ack:1;
455 unsigned int secen:1;
456 unsigned int secen_override:1;
457 unsigned int seclevel:3;
458 unsigned int seclevel_override:1;
459};
460
461static inline struct dgram_sock *dgram_sk(const struct sock *sk)
462{
463 return container_of(sk, struct dgram_sock, sk);
464}
465
466static int dgram_hash(struct sock *sk)
467{
468 write_lock_bh(&dgram_lock);
469 sk_add_node(sk, &dgram_head);
470 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
471 write_unlock_bh(&dgram_lock);
472
473 return 0;
474}
475
476static void dgram_unhash(struct sock *sk)
477{
478 write_lock_bh(&dgram_lock);
479 if (sk_del_node_init(sk))
480 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
481 write_unlock_bh(&dgram_lock);
482}
483
484static int dgram_init(struct sock *sk)
485{
486 struct dgram_sock *ro = dgram_sk(sk);
487
488 ro->want_ack = 1;
489 return 0;
490}
491
492static void dgram_close(struct sock *sk, long timeout)
493{
494 sk_common_release(sk);
495}
496
497static int dgram_bind(struct sock *sk, struct sockaddr *uaddr, int len)
498{
499 struct sockaddr_ieee802154 *addr = (struct sockaddr_ieee802154 *)uaddr;
500 struct ieee802154_addr haddr;
501 struct dgram_sock *ro = dgram_sk(sk);
502 int err = -EINVAL;
503 struct net_device *dev;
504
505 lock_sock(sk);
506
507 ro->bound = 0;
508
509 if (len < sizeof(*addr))
510 goto out;
511
512 if (addr->family != AF_IEEE802154)
513 goto out;
514
515 ieee802154_addr_from_sa(&haddr, &addr->addr);
516 dev = ieee802154_get_dev(sock_net(sk), &haddr);
517 if (!dev) {
518 err = -ENODEV;
519 goto out;
520 }
521
522 if (dev->type != ARPHRD_IEEE802154) {
523 err = -ENODEV;
524 goto out_put;
525 }
526
527 ro->src_addr = haddr;
528
529 ro->bound = 1;
530 err = 0;
531out_put:
532 dev_put(dev);
533out:
534 release_sock(sk);
535
536 return err;
537}
538
539static int dgram_ioctl(struct sock *sk, int cmd, unsigned long arg)
540{
541 switch (cmd) {
542 case SIOCOUTQ:
543 {
544 int amount = sk_wmem_alloc_get(sk);
545
546 return put_user(amount, (int __user *)arg);
547 }
548
549 case SIOCINQ:
550 {
551 struct sk_buff *skb;
552 unsigned long amount;
553
554 amount = 0;
555 spin_lock_bh(&sk->sk_receive_queue.lock);
556 skb = skb_peek(&sk->sk_receive_queue);
557 if (skb) {
558
559
560
561
562 amount = skb->len - ieee802154_hdr_length(skb);
563 }
564 spin_unlock_bh(&sk->sk_receive_queue.lock);
565 return put_user(amount, (int __user *)arg);
566 }
567 }
568
569 return -ENOIOCTLCMD;
570}
571
572
573static int dgram_connect(struct sock *sk, struct sockaddr *uaddr,
574 int len)
575{
576 struct sockaddr_ieee802154 *addr = (struct sockaddr_ieee802154 *)uaddr;
577 struct dgram_sock *ro = dgram_sk(sk);
578 int err = 0;
579
580 if (len < sizeof(*addr))
581 return -EINVAL;
582
583 if (addr->family != AF_IEEE802154)
584 return -EINVAL;
585
586 lock_sock(sk);
587
588 if (!ro->bound) {
589 err = -ENETUNREACH;
590 goto out;
591 }
592
593 ieee802154_addr_from_sa(&ro->dst_addr, &addr->addr);
594 ro->connected = 1;
595
596out:
597 release_sock(sk);
598 return err;
599}
600
601static int dgram_disconnect(struct sock *sk, int flags)
602{
603 struct dgram_sock *ro = dgram_sk(sk);
604
605 lock_sock(sk);
606 ro->connected = 0;
607 release_sock(sk);
608
609 return 0;
610}
611
612static int dgram_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
613{
614 struct net_device *dev;
615 unsigned int mtu;
616 struct sk_buff *skb;
617 struct ieee802154_mac_cb *cb;
618 struct dgram_sock *ro = dgram_sk(sk);
619 struct ieee802154_addr dst_addr;
620 int hlen, tlen;
621 int err;
622
623 if (msg->msg_flags & MSG_OOB) {
624 pr_debug("msg->msg_flags = 0x%x\n", msg->msg_flags);
625 return -EOPNOTSUPP;
626 }
627
628 if (!ro->connected && !msg->msg_name)
629 return -EDESTADDRREQ;
630 else if (ro->connected && msg->msg_name)
631 return -EISCONN;
632
633 if (!ro->bound)
634 dev = dev_getfirstbyhwtype(sock_net(sk), ARPHRD_IEEE802154);
635 else
636 dev = ieee802154_get_dev(sock_net(sk), &ro->src_addr);
637
638 if (!dev) {
639 pr_debug("no dev\n");
640 err = -ENXIO;
641 goto out;
642 }
643 mtu = IEEE802154_MTU;
644 pr_debug("name = %s, mtu = %u\n", dev->name, mtu);
645
646 if (size > mtu) {
647 pr_debug("size = %zu, mtu = %u\n", size, mtu);
648 err = -EMSGSIZE;
649 goto out_dev;
650 }
651
652 hlen = LL_RESERVED_SPACE(dev);
653 tlen = dev->needed_tailroom;
654 skb = sock_alloc_send_skb(sk, hlen + tlen + size,
655 msg->msg_flags & MSG_DONTWAIT,
656 &err);
657 if (!skb)
658 goto out_dev;
659
660 skb_reserve(skb, hlen);
661
662 skb_reset_network_header(skb);
663
664 cb = mac_cb_init(skb);
665 cb->type = IEEE802154_FC_TYPE_DATA;
666 cb->ackreq = ro->want_ack;
667
668 if (msg->msg_name) {
669 DECLARE_SOCKADDR(struct sockaddr_ieee802154*,
670 daddr, msg->msg_name);
671
672 ieee802154_addr_from_sa(&dst_addr, &daddr->addr);
673 } else {
674 dst_addr = ro->dst_addr;
675 }
676
677 cb->secen = ro->secen;
678 cb->secen_override = ro->secen_override;
679 cb->seclevel = ro->seclevel;
680 cb->seclevel_override = ro->seclevel_override;
681
682 err = wpan_dev_hard_header(skb, dev, &dst_addr,
683 ro->bound ? &ro->src_addr : NULL, size);
684 if (err < 0)
685 goto out_skb;
686
687 err = memcpy_from_msg(skb_put(skb, size), msg, size);
688 if (err < 0)
689 goto out_skb;
690
691 skb->dev = dev;
692 skb->protocol = htons(ETH_P_IEEE802154);
693
694 err = dev_queue_xmit(skb);
695 if (err > 0)
696 err = net_xmit_errno(err);
697
698 dev_put(dev);
699
700 return err ?: size;
701
702out_skb:
703 kfree_skb(skb);
704out_dev:
705 dev_put(dev);
706out:
707 return err;
708}
709
710static int dgram_recvmsg(struct sock *sk, struct msghdr *msg, size_t len,
711 int noblock, int flags, int *addr_len)
712{
713 size_t copied = 0;
714 int err = -EOPNOTSUPP;
715 struct sk_buff *skb;
716 DECLARE_SOCKADDR(struct sockaddr_ieee802154 *, saddr, msg->msg_name);
717
718 skb = skb_recv_datagram(sk, flags, noblock, &err);
719 if (!skb)
720 goto out;
721
722 copied = skb->len;
723 if (len < copied) {
724 msg->msg_flags |= MSG_TRUNC;
725 copied = len;
726 }
727
728
729 err = skb_copy_datagram_msg(skb, 0, msg, copied);
730 if (err)
731 goto done;
732
733 sock_recv_ts_and_drops(msg, sk, skb);
734
735 if (saddr) {
736
737
738
739
740 memset(saddr, 0, sizeof(*saddr));
741
742 saddr->family = AF_IEEE802154;
743 ieee802154_addr_to_sa(&saddr->addr, &mac_cb(skb)->source);
744 *addr_len = sizeof(*saddr);
745 }
746
747 if (flags & MSG_TRUNC)
748 copied = skb->len;
749done:
750 skb_free_datagram(sk, skb);
751out:
752 if (err)
753 return err;
754 return copied;
755}
756
757static int dgram_rcv_skb(struct sock *sk, struct sk_buff *skb)
758{
759 skb = skb_share_check(skb, GFP_ATOMIC);
760 if (!skb)
761 return NET_RX_DROP;
762
763 if (sock_queue_rcv_skb(sk, skb) < 0) {
764 kfree_skb(skb);
765 return NET_RX_DROP;
766 }
767
768 return NET_RX_SUCCESS;
769}
770
771static inline bool
772ieee802154_match_sock(__le64 hw_addr, __le16 pan_id, __le16 short_addr,
773 struct dgram_sock *ro)
774{
775 if (!ro->bound)
776 return true;
777
778 if (ro->src_addr.mode == IEEE802154_ADDR_LONG &&
779 hw_addr == ro->src_addr.extended_addr)
780 return true;
781
782 if (ro->src_addr.mode == IEEE802154_ADDR_SHORT &&
783 pan_id == ro->src_addr.pan_id &&
784 short_addr == ro->src_addr.short_addr)
785 return true;
786
787 return false;
788}
789
790static int ieee802154_dgram_deliver(struct net_device *dev, struct sk_buff *skb)
791{
792 struct sock *sk, *prev = NULL;
793 int ret = NET_RX_SUCCESS;
794 __le16 pan_id, short_addr;
795 __le64 hw_addr;
796
797
798 BUG_ON(dev->type != ARPHRD_IEEE802154);
799
800 pan_id = dev->ieee802154_ptr->pan_id;
801 short_addr = dev->ieee802154_ptr->short_addr;
802 hw_addr = dev->ieee802154_ptr->extended_addr;
803
804 read_lock(&dgram_lock);
805 sk_for_each(sk, &dgram_head) {
806 if (ieee802154_match_sock(hw_addr, pan_id, short_addr,
807 dgram_sk(sk))) {
808 if (prev) {
809 struct sk_buff *clone;
810
811 clone = skb_clone(skb, GFP_ATOMIC);
812 if (clone)
813 dgram_rcv_skb(prev, clone);
814 }
815
816 prev = sk;
817 }
818 }
819
820 if (prev) {
821 dgram_rcv_skb(prev, skb);
822 } else {
823 kfree_skb(skb);
824 ret = NET_RX_DROP;
825 }
826 read_unlock(&dgram_lock);
827
828 return ret;
829}
830
831static int dgram_getsockopt(struct sock *sk, int level, int optname,
832 char __user *optval, int __user *optlen)
833{
834 struct dgram_sock *ro = dgram_sk(sk);
835
836 int val, len;
837
838 if (level != SOL_IEEE802154)
839 return -EOPNOTSUPP;
840
841 if (get_user(len, optlen))
842 return -EFAULT;
843
844 len = min_t(unsigned int, len, sizeof(int));
845
846 switch (optname) {
847 case WPAN_WANTACK:
848 val = ro->want_ack;
849 break;
850 case WPAN_SECURITY:
851 if (!ro->secen_override)
852 val = WPAN_SECURITY_DEFAULT;
853 else if (ro->secen)
854 val = WPAN_SECURITY_ON;
855 else
856 val = WPAN_SECURITY_OFF;
857 break;
858 case WPAN_SECURITY_LEVEL:
859 if (!ro->seclevel_override)
860 val = WPAN_SECURITY_LEVEL_DEFAULT;
861 else
862 val = ro->seclevel;
863 break;
864 default:
865 return -ENOPROTOOPT;
866 }
867
868 if (put_user(len, optlen))
869 return -EFAULT;
870 if (copy_to_user(optval, &val, len))
871 return -EFAULT;
872 return 0;
873}
874
875static int dgram_setsockopt(struct sock *sk, int level, int optname,
876 char __user *optval, unsigned int optlen)
877{
878 struct dgram_sock *ro = dgram_sk(sk);
879 struct net *net = sock_net(sk);
880 int val;
881 int err = 0;
882
883 if (optlen < sizeof(int))
884 return -EINVAL;
885
886 if (get_user(val, (int __user *)optval))
887 return -EFAULT;
888
889 lock_sock(sk);
890
891 switch (optname) {
892 case WPAN_WANTACK:
893 ro->want_ack = !!val;
894 break;
895 case WPAN_SECURITY:
896 if (!ns_capable(net->user_ns, CAP_NET_ADMIN) &&
897 !ns_capable(net->user_ns, CAP_NET_RAW)) {
898 err = -EPERM;
899 break;
900 }
901
902 switch (val) {
903 case WPAN_SECURITY_DEFAULT:
904 ro->secen_override = 0;
905 break;
906 case WPAN_SECURITY_ON:
907 ro->secen_override = 1;
908 ro->secen = 1;
909 break;
910 case WPAN_SECURITY_OFF:
911 ro->secen_override = 1;
912 ro->secen = 0;
913 break;
914 default:
915 err = -EINVAL;
916 break;
917 }
918 break;
919 case WPAN_SECURITY_LEVEL:
920 if (!ns_capable(net->user_ns, CAP_NET_ADMIN) &&
921 !ns_capable(net->user_ns, CAP_NET_RAW)) {
922 err = -EPERM;
923 break;
924 }
925
926 if (val < WPAN_SECURITY_LEVEL_DEFAULT ||
927 val > IEEE802154_SCF_SECLEVEL_ENC_MIC128) {
928 err = -EINVAL;
929 } else if (val == WPAN_SECURITY_LEVEL_DEFAULT) {
930 ro->seclevel_override = 0;
931 } else {
932 ro->seclevel_override = 1;
933 ro->seclevel = val;
934 }
935 break;
936 default:
937 err = -ENOPROTOOPT;
938 break;
939 }
940
941 release_sock(sk);
942 return err;
943}
944
945static struct proto ieee802154_dgram_prot = {
946 .name = "IEEE-802.15.4-MAC",
947 .owner = THIS_MODULE,
948 .obj_size = sizeof(struct dgram_sock),
949 .init = dgram_init,
950 .close = dgram_close,
951 .bind = dgram_bind,
952 .sendmsg = dgram_sendmsg,
953 .recvmsg = dgram_recvmsg,
954 .hash = dgram_hash,
955 .unhash = dgram_unhash,
956 .connect = dgram_connect,
957 .disconnect = dgram_disconnect,
958 .ioctl = dgram_ioctl,
959 .getsockopt = dgram_getsockopt,
960 .setsockopt = dgram_setsockopt,
961};
962
963static const struct proto_ops ieee802154_dgram_ops = {
964 .family = PF_IEEE802154,
965 .owner = THIS_MODULE,
966 .release = ieee802154_sock_release,
967 .bind = ieee802154_sock_bind,
968 .connect = ieee802154_sock_connect,
969 .socketpair = sock_no_socketpair,
970 .accept = sock_no_accept,
971 .getname = sock_no_getname,
972 .poll = datagram_poll,
973 .ioctl = ieee802154_sock_ioctl,
974 .listen = sock_no_listen,
975 .shutdown = sock_no_shutdown,
976 .setsockopt = sock_common_setsockopt,
977 .getsockopt = sock_common_getsockopt,
978 .sendmsg = ieee802154_sock_sendmsg,
979 .recvmsg = sock_common_recvmsg,
980 .mmap = sock_no_mmap,
981 .sendpage = sock_no_sendpage,
982#ifdef CONFIG_COMPAT
983 .compat_setsockopt = compat_sock_common_setsockopt,
984 .compat_getsockopt = compat_sock_common_getsockopt,
985#endif
986};
987
988
989
990
991static int ieee802154_create(struct net *net, struct socket *sock,
992 int protocol, int kern)
993{
994 struct sock *sk;
995 int rc;
996 struct proto *proto;
997 const struct proto_ops *ops;
998
999 if (!net_eq(net, &init_net))
1000 return -EAFNOSUPPORT;
1001
1002 switch (sock->type) {
1003 case SOCK_RAW:
1004 proto = &ieee802154_raw_prot;
1005 ops = &ieee802154_raw_ops;
1006 break;
1007 case SOCK_DGRAM:
1008 proto = &ieee802154_dgram_prot;
1009 ops = &ieee802154_dgram_ops;
1010 break;
1011 default:
1012 rc = -ESOCKTNOSUPPORT;
1013 goto out;
1014 }
1015
1016 rc = -ENOMEM;
1017 sk = sk_alloc(net, PF_IEEE802154, GFP_KERNEL, proto, kern);
1018 if (!sk)
1019 goto out;
1020 rc = 0;
1021
1022 sock->ops = ops;
1023
1024 sock_init_data(sock, sk);
1025
1026 sk->sk_family = PF_IEEE802154;
1027
1028
1029 sock_set_flag(sk, SOCK_ZAPPED);
1030
1031 if (sk->sk_prot->hash) {
1032 rc = sk->sk_prot->hash(sk);
1033 if (rc) {
1034 sk_common_release(sk);
1035 goto out;
1036 }
1037 }
1038
1039 if (sk->sk_prot->init) {
1040 rc = sk->sk_prot->init(sk);
1041 if (rc)
1042 sk_common_release(sk);
1043 }
1044out:
1045 return rc;
1046}
1047
1048static const struct net_proto_family ieee802154_family_ops = {
1049 .family = PF_IEEE802154,
1050 .create = ieee802154_create,
1051 .owner = THIS_MODULE,
1052};
1053
1054static int ieee802154_rcv(struct sk_buff *skb, struct net_device *dev,
1055 struct packet_type *pt, struct net_device *orig_dev)
1056{
1057 if (!netif_running(dev))
1058 goto drop;
1059 pr_debug("got frame, type %d, dev %p\n", dev->type, dev);
1060#ifdef DEBUG
1061 print_hex_dump_bytes("ieee802154_rcv ",
1062 DUMP_PREFIX_NONE, skb->data, skb->len);
1063#endif
1064
1065 if (!net_eq(dev_net(dev), &init_net))
1066 goto drop;
1067
1068 ieee802154_raw_deliver(dev, skb);
1069
1070 if (dev->type != ARPHRD_IEEE802154)
1071 goto drop;
1072
1073 if (skb->pkt_type != PACKET_OTHERHOST)
1074 return ieee802154_dgram_deliver(dev, skb);
1075
1076drop:
1077 kfree_skb(skb);
1078 return NET_RX_DROP;
1079}
1080
1081static struct packet_type ieee802154_packet_type = {
1082 .type = htons(ETH_P_IEEE802154),
1083 .func = ieee802154_rcv,
1084};
1085
1086static int __init af_ieee802154_init(void)
1087{
1088 int rc = -EINVAL;
1089
1090 rc = proto_register(&ieee802154_raw_prot, 1);
1091 if (rc)
1092 goto out;
1093
1094 rc = proto_register(&ieee802154_dgram_prot, 1);
1095 if (rc)
1096 goto err_dgram;
1097
1098
1099 rc = sock_register(&ieee802154_family_ops);
1100 if (rc)
1101 goto err_sock;
1102 dev_add_pack(&ieee802154_packet_type);
1103
1104 rc = 0;
1105 goto out;
1106
1107err_sock:
1108 proto_unregister(&ieee802154_dgram_prot);
1109err_dgram:
1110 proto_unregister(&ieee802154_raw_prot);
1111out:
1112 return rc;
1113}
1114
1115static void __exit af_ieee802154_remove(void)
1116{
1117 dev_remove_pack(&ieee802154_packet_type);
1118 sock_unregister(PF_IEEE802154);
1119 proto_unregister(&ieee802154_dgram_prot);
1120 proto_unregister(&ieee802154_raw_prot);
1121}
1122
1123module_init(af_ieee802154_init);
1124module_exit(af_ieee802154_remove);
1125
1126MODULE_LICENSE("GPL");
1127MODULE_ALIAS_NETPROTO(PF_IEEE802154);
1128
