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