?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
staging:rtl8712 Aligned code with open parenthesis
[linux/fpc-iii.git] / net / ieee802154 / socket.c
blobe0bd013a1e5ed9fc8fbd16329a877f0231e13271
1 /*
2 * IEEE802154.4 socket interface
3 *
4 * Copyright 2007, 2008 Siemens AG
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2
8 * as published by the Free Software Foundation.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * Written by:
16 * Sergey Lapin <slapin@ossfans.org>
17 * Maxim Gorbachyov <maxim.gorbachev@siemens.com>
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> /* For TIOCOUTQ/INQ */
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 /* Utility function for families */
38 static struct net_device*
39 ieee802154_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
88 static 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
99 static 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
107 static 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
118 static 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
132 static 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
160 static 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 /* RAW Sockets (802.15.4 created in userspace) */
182 static HLIST_HEAD(raw_head);
183 static DEFINE_RWLOCK(raw_lock);
184
185 static 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
195 static 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
203 static void raw_close(struct sock *sk, long timeout)
204 {
205 sk_common_release(sk);
206 }
207
208 static 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);
235 out:
236 release_sock(sk);
237
238 return err;
239 }
240
241 static int raw_connect(struct sock *sk, struct sockaddr *uaddr,
242 int addr_len)
243 {
244 return -ENOTSUPP;
245 }
246
247 static int raw_disconnect(struct sock *sk, int flags)
248 {
249 return 0;
250 }
251
252 static 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->sk = sk;
305 skb->protocol = htons(ETH_P_IEEE802154);
306
307 dev_put(dev);
308
309 err = dev_queue_xmit(skb);
310 if (err > 0)
311 err = net_xmit_errno(err);
312
313 return err ?: size;
314
315 out_skb:
316 kfree_skb(skb);
317 out_dev:
318 dev_put(dev);
319 out:
320 return err;
321 }
322
323 static int raw_recvmsg(struct sock *sk, struct msghdr *msg, size_t len,
324 int noblock, int flags, int *addr_len)
325 {
326 size_t copied = 0;
327 int err = -EOPNOTSUPP;
328 struct sk_buff *skb;
329
330 skb = skb_recv_datagram(sk, flags, noblock, &err);
331 if (!skb)
332 goto out;
333
334 copied = skb->len;
335 if (len < copied) {
336 msg->msg_flags |= MSG_TRUNC;
337 copied = len;
338 }
339
340 err = skb_copy_datagram_msg(skb, 0, msg, copied);
341 if (err)
342 goto done;
343
344 sock_recv_ts_and_drops(msg, sk, skb);
345
346 if (flags & MSG_TRUNC)
347 copied = skb->len;
348 done:
349 skb_free_datagram(sk, skb);
350 out:
351 if (err)
352 return err;
353 return copied;
354 }
355
356 static int raw_rcv_skb(struct sock *sk, struct sk_buff *skb)
357 {
358 skb = skb_share_check(skb, GFP_ATOMIC);
359 if (!skb)
360 return NET_RX_DROP;
361
362 if (sock_queue_rcv_skb(sk, skb) < 0) {
363 kfree_skb(skb);
364 return NET_RX_DROP;
365 }
366
367 return NET_RX_SUCCESS;
368 }
369
370 static void ieee802154_raw_deliver(struct net_device *dev, struct sk_buff *skb)
371 {
372 struct sock *sk;
373
374 read_lock(&raw_lock);
375 sk_for_each(sk, &raw_head) {
376 bh_lock_sock(sk);
377 if (!sk->sk_bound_dev_if ||
378 sk->sk_bound_dev_if == dev->ifindex) {
379 struct sk_buff *clone;
380
381 clone = skb_clone(skb, GFP_ATOMIC);
382 if (clone)
383 raw_rcv_skb(sk, clone);
384 }
385 bh_unlock_sock(sk);
386 }
387 read_unlock(&raw_lock);
388 }
389
390 static int raw_getsockopt(struct sock *sk, int level, int optname,
391 char __user *optval, int __user *optlen)
392 {
393 return -EOPNOTSUPP;
394 }
395
396 static int raw_setsockopt(struct sock *sk, int level, int optname,
397 char __user *optval, unsigned int optlen)
398 {
399 return -EOPNOTSUPP;
400 }
401
402 static struct proto ieee802154_raw_prot = {
403 .name = "IEEE-802.15.4-RAW",
404 .owner = THIS_MODULE,
405 .obj_size = sizeof(struct sock),
406 .close = raw_close,
407 .bind = raw_bind,
408 .sendmsg = raw_sendmsg,
409 .recvmsg = raw_recvmsg,
410 .hash = raw_hash,
411 .unhash = raw_unhash,
412 .connect = raw_connect,
413 .disconnect = raw_disconnect,
414 .getsockopt = raw_getsockopt,
415 .setsockopt = raw_setsockopt,
416 };
417
418 static const struct proto_ops ieee802154_raw_ops = {
419 .family = PF_IEEE802154,
420 .owner = THIS_MODULE,
421 .release = ieee802154_sock_release,
422 .bind = ieee802154_sock_bind,
423 .connect = ieee802154_sock_connect,
424 .socketpair = sock_no_socketpair,
425 .accept = sock_no_accept,
426 .getname = sock_no_getname,
427 .poll = datagram_poll,
428 .ioctl = ieee802154_sock_ioctl,
429 .listen = sock_no_listen,
430 .shutdown = sock_no_shutdown,
431 .setsockopt = sock_common_setsockopt,
432 .getsockopt = sock_common_getsockopt,
433 .sendmsg = ieee802154_sock_sendmsg,
434 .recvmsg = sock_common_recvmsg,
435 .mmap = sock_no_mmap,
436 .sendpage = sock_no_sendpage,
437 #ifdef CONFIG_COMPAT
438 .compat_setsockopt = compat_sock_common_setsockopt,
439 .compat_getsockopt = compat_sock_common_getsockopt,
440 #endif
441 };
442
443 /* DGRAM Sockets (802.15.4 dataframes) */
444 static HLIST_HEAD(dgram_head);
445 static DEFINE_RWLOCK(dgram_lock);
446
447 struct dgram_sock {
448 struct sock sk;
449
450 struct ieee802154_addr src_addr;
451 struct ieee802154_addr dst_addr;
452
453 unsigned int bound:1;
454 unsigned int connected:1;
455 unsigned int want_ack:1;
456 unsigned int secen:1;
457 unsigned int secen_override:1;
458 unsigned int seclevel:3;
459 unsigned int seclevel_override:1;
460 };
461
462 static inline struct dgram_sock *dgram_sk(const struct sock *sk)
463 {
464 return container_of(sk, struct dgram_sock, sk);
465 }
466
467 static int dgram_hash(struct sock *sk)
468 {
469 write_lock_bh(&dgram_lock);
470 sk_add_node(sk, &dgram_head);
471 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
472 write_unlock_bh(&dgram_lock);
473
474 return 0;
475 }
476
477 static void dgram_unhash(struct sock *sk)
478 {
479 write_lock_bh(&dgram_lock);
480 if (sk_del_node_init(sk))
481 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
482 write_unlock_bh(&dgram_lock);
483 }
484
485 static int dgram_init(struct sock *sk)
486 {
487 struct dgram_sock *ro = dgram_sk(sk);
488
489 ro->want_ack = 1;
490 return 0;
491 }
492
493 static void dgram_close(struct sock *sk, long timeout)
494 {
495 sk_common_release(sk);
496 }
497
498 static int dgram_bind(struct sock *sk, struct sockaddr *uaddr, int len)
499 {
500 struct sockaddr_ieee802154 *addr = (struct sockaddr_ieee802154 *)uaddr;
501 struct ieee802154_addr haddr;
502 struct dgram_sock *ro = dgram_sk(sk);
503 int err = -EINVAL;
504 struct net_device *dev;
505
506 lock_sock(sk);
507
508 ro->bound = 0;
509
510 if (len < sizeof(*addr))
511 goto out;
512
513 if (addr->family != AF_IEEE802154)
514 goto out;
515
516 ieee802154_addr_from_sa(&haddr, &addr->addr);
517 dev = ieee802154_get_dev(sock_net(sk), &haddr);
518 if (!dev) {
519 err = -ENODEV;
520 goto out;
521 }
522
523 if (dev->type != ARPHRD_IEEE802154) {
524 err = -ENODEV;
525 goto out_put;
526 }
527
528 ro->src_addr = haddr;
529
530 ro->bound = 1;
531 err = 0;
532 out_put:
533 dev_put(dev);
534 out:
535 release_sock(sk);
536
537 return err;
538 }
539
540 static int dgram_ioctl(struct sock *sk, int cmd, unsigned long arg)
541 {
542 switch (cmd) {
543 case SIOCOUTQ:
544 {
545 int amount = sk_wmem_alloc_get(sk);
546
547 return put_user(amount, (int __user *)arg);
548 }
549
550 case SIOCINQ:
551 {
552 struct sk_buff *skb;
553 unsigned long amount;
554
555 amount = 0;
556 spin_lock_bh(&sk->sk_receive_queue.lock);
557 skb = skb_peek(&sk->sk_receive_queue);
558 if (skb) {
559 /* We will only return the amount
560 * of this packet since that is all
561 * that will be read.
562 */
563 amount = skb->len - ieee802154_hdr_length(skb);
564 }
565 spin_unlock_bh(&sk->sk_receive_queue.lock);
566 return put_user(amount, (int __user *)arg);
567 }
568 }
569
570 return -ENOIOCTLCMD;
571 }
572
573 /* FIXME: autobind */
574 static int dgram_connect(struct sock *sk, struct sockaddr *uaddr,
575 int len)
576 {
577 struct sockaddr_ieee802154 *addr = (struct sockaddr_ieee802154 *)uaddr;
578 struct dgram_sock *ro = dgram_sk(sk);
579 int err = 0;
580
581 if (len < sizeof(*addr))
582 return -EINVAL;
583
584 if (addr->family != AF_IEEE802154)
585 return -EINVAL;
586
587 lock_sock(sk);
588
589 if (!ro->bound) {
590 err = -ENETUNREACH;
591 goto out;
592 }
593
594 ieee802154_addr_from_sa(&ro->dst_addr, &addr->addr);
595 ro->connected = 1;
596
597 out:
598 release_sock(sk);
599 return err;
600 }
601
602 static int dgram_disconnect(struct sock *sk, int flags)
603 {
604 struct dgram_sock *ro = dgram_sk(sk);
605
606 lock_sock(sk);
607 ro->connected = 0;
608 release_sock(sk);
609
610 return 0;
611 }
612
613 static int dgram_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
614 {
615 struct net_device *dev;
616 unsigned int mtu;
617 struct sk_buff *skb;
618 struct ieee802154_mac_cb *cb;
619 struct dgram_sock *ro = dgram_sk(sk);
620 struct ieee802154_addr dst_addr;
621 int hlen, tlen;
622 int err;
623
624 if (msg->msg_flags & MSG_OOB) {
625 pr_debug("msg->msg_flags = 0x%x\n", msg->msg_flags);
626 return -EOPNOTSUPP;
627 }
628
629 if (!ro->connected && !msg->msg_name)
630 return -EDESTADDRREQ;
631 else if (ro->connected && msg->msg_name)
632 return -EISCONN;
633
634 if (!ro->bound)
635 dev = dev_getfirstbyhwtype(sock_net(sk), ARPHRD_IEEE802154);
636 else
637 dev = ieee802154_get_dev(sock_net(sk), &ro->src_addr);
638
639 if (!dev) {
640 pr_debug("no dev\n");
641 err = -ENXIO;
642 goto out;
643 }
644 mtu = IEEE802154_MTU;
645 pr_debug("name = %s, mtu = %u\n", dev->name, mtu);
646
647 if (size > mtu) {
648 pr_debug("size = %Zu, mtu = %u\n", size, mtu);
649 err = -EMSGSIZE;
650 goto out_dev;
651 }
652
653 hlen = LL_RESERVED_SPACE(dev);
654 tlen = dev->needed_tailroom;
655 skb = sock_alloc_send_skb(sk, hlen + tlen + size,
656 msg->msg_flags & MSG_DONTWAIT,
657 &err);
658 if (!skb)
659 goto out_dev;
660
661 skb_reserve(skb, hlen);
662
663 skb_reset_network_header(skb);
664
665 cb = mac_cb_init(skb);
666 cb->type = IEEE802154_FC_TYPE_DATA;
667 cb->ackreq = ro->want_ack;
668
669 if (msg->msg_name) {
670 DECLARE_SOCKADDR(struct sockaddr_ieee802154*,
671 daddr, msg->msg_name);
672
673 ieee802154_addr_from_sa(&dst_addr, &daddr->addr);
674 } else {
675 dst_addr = ro->dst_addr;
676 }
677
678 cb->secen = ro->secen;
679 cb->secen_override = ro->secen_override;
680 cb->seclevel = ro->seclevel;
681 cb->seclevel_override = ro->seclevel_override;
682
683 err = wpan_dev_hard_header(skb, dev, &dst_addr,
684 ro->bound ? &ro->src_addr : NULL, size);
685 if (err < 0)
686 goto out_skb;
687
688 err = memcpy_from_msg(skb_put(skb, size), msg, size);
689 if (err < 0)
690 goto out_skb;
691
692 skb->dev = dev;
693 skb->sk = sk;
694 skb->protocol = htons(ETH_P_IEEE802154);
695
696 dev_put(dev);
697
698 err = dev_queue_xmit(skb);
699 if (err > 0)
700 err = net_xmit_errno(err);
701
702 return err ?: size;
703
704 out_skb:
705 kfree_skb(skb);
706 out_dev:
707 dev_put(dev);
708 out:
709 return err;
710 }
711
712 static int dgram_recvmsg(struct sock *sk, struct msghdr *msg, size_t len,
713 int noblock, int flags, int *addr_len)
714 {
715 size_t copied = 0;
716 int err = -EOPNOTSUPP;
717 struct sk_buff *skb;
718 DECLARE_SOCKADDR(struct sockaddr_ieee802154 *, saddr, msg->msg_name);
719
720 skb = skb_recv_datagram(sk, flags, noblock, &err);
721 if (!skb)
722 goto out;
723
724 copied = skb->len;
725 if (len < copied) {
726 msg->msg_flags |= MSG_TRUNC;
727 copied = len;
728 }
729
730 /* FIXME: skip headers if necessary ?! */
731 err = skb_copy_datagram_msg(skb, 0, msg, copied);
732 if (err)
733 goto done;
734
735 sock_recv_ts_and_drops(msg, sk, skb);
736
737 if (saddr) {
738 /* Clear the implicit padding in struct sockaddr_ieee802154
739 * (16 bits between 'family' and 'addr') and in struct
740 * ieee802154_addr_sa (16 bits at the end of the structure).
741 */
742 memset(saddr, 0, sizeof(*saddr));
743
744 saddr->family = AF_IEEE802154;
745 ieee802154_addr_to_sa(&saddr->addr, &mac_cb(skb)->source);
746 *addr_len = sizeof(*saddr);
747 }
748
749 if (flags & MSG_TRUNC)
750 copied = skb->len;
751 done:
752 skb_free_datagram(sk, skb);
753 out:
754 if (err)
755 return err;
756 return copied;
757 }
758
759 static int dgram_rcv_skb(struct sock *sk, struct sk_buff *skb)
760 {
761 skb = skb_share_check(skb, GFP_ATOMIC);
762 if (!skb)
763 return NET_RX_DROP;
764
765 if (sock_queue_rcv_skb(sk, skb) < 0) {
766 kfree_skb(skb);
767 return NET_RX_DROP;
768 }
769
770 return NET_RX_SUCCESS;
771 }
772
773 static inline bool
774 ieee802154_match_sock(__le64 hw_addr, __le16 pan_id, __le16 short_addr,
775 struct dgram_sock *ro)
776 {
777 if (!ro->bound)
778 return true;
779
780 if (ro->src_addr.mode == IEEE802154_ADDR_LONG &&
781 hw_addr == ro->src_addr.extended_addr)
782 return true;
783
784 if (ro->src_addr.mode == IEEE802154_ADDR_SHORT &&
785 pan_id == ro->src_addr.pan_id &&
786 short_addr == ro->src_addr.short_addr)
787 return true;
788
789 return false;
790 }
791
792 static int ieee802154_dgram_deliver(struct net_device *dev, struct sk_buff *skb)
793 {
794 struct sock *sk, *prev = NULL;
795 int ret = NET_RX_SUCCESS;
796 __le16 pan_id, short_addr;
797 __le64 hw_addr;
798
799 /* Data frame processing */
800 BUG_ON(dev->type != ARPHRD_IEEE802154);
801
802 pan_id = dev->ieee802154_ptr->pan_id;
803 short_addr = dev->ieee802154_ptr->short_addr;
804 hw_addr = dev->ieee802154_ptr->extended_addr;
805
806 read_lock(&dgram_lock);
807 sk_for_each(sk, &dgram_head) {
808 if (ieee802154_match_sock(hw_addr, pan_id, short_addr,
809 dgram_sk(sk))) {
810 if (prev) {
811 struct sk_buff *clone;
812
813 clone = skb_clone(skb, GFP_ATOMIC);
814 if (clone)
815 dgram_rcv_skb(prev, clone);
816 }
817
818 prev = sk;
819 }
820 }
821
822 if (prev) {
823 dgram_rcv_skb(prev, skb);
824 } else {
825 kfree_skb(skb);
826 ret = NET_RX_DROP;
827 }
828 read_unlock(&dgram_lock);
829
830 return ret;
831 }
832
833 static int dgram_getsockopt(struct sock *sk, int level, int optname,
834 char __user *optval, int __user *optlen)
835 {
836 struct dgram_sock *ro = dgram_sk(sk);
837
838 int val, len;
839
840 if (level != SOL_IEEE802154)
841 return -EOPNOTSUPP;
842
843 if (get_user(len, optlen))
844 return -EFAULT;
845
846 len = min_t(unsigned int, len, sizeof(int));
847
848 switch (optname) {
849 case WPAN_WANTACK:
850 val = ro->want_ack;
851 break;
852 case WPAN_SECURITY:
853 if (!ro->secen_override)
854 val = WPAN_SECURITY_DEFAULT;
855 else if (ro->secen)
856 val = WPAN_SECURITY_ON;
857 else
858 val = WPAN_SECURITY_OFF;
859 break;
860 case WPAN_SECURITY_LEVEL:
861 if (!ro->seclevel_override)
862 val = WPAN_SECURITY_LEVEL_DEFAULT;
863 else
864 val = ro->seclevel;
865 break;
866 default:
867 return -ENOPROTOOPT;
868 }
869
870 if (put_user(len, optlen))
871 return -EFAULT;
872 if (copy_to_user(optval, &val, len))
873 return -EFAULT;
874 return 0;
875 }
876
877 static int dgram_setsockopt(struct sock *sk, int level, int optname,
878 char __user *optval, unsigned int optlen)
879 {
880 struct dgram_sock *ro = dgram_sk(sk);
881 struct net *net = sock_net(sk);
882 int val;
883 int err = 0;
884
885 if (optlen < sizeof(int))
886 return -EINVAL;
887
888 if (get_user(val, (int __user *)optval))
889 return -EFAULT;
890
891 lock_sock(sk);
892
893 switch (optname) {
894 case WPAN_WANTACK:
895 ro->want_ack = !!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
947 static 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
965 static 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 .listen = sock_no_listen,
977 .shutdown = sock_no_shutdown,
978 .setsockopt = sock_common_setsockopt,
979 .getsockopt = sock_common_getsockopt,
980 .sendmsg = ieee802154_sock_sendmsg,
981 .recvmsg = sock_common_recvmsg,
982 .mmap = sock_no_mmap,
983 .sendpage = sock_no_sendpage,
984 #ifdef CONFIG_COMPAT
985 .compat_setsockopt = compat_sock_common_setsockopt,
986 .compat_getsockopt = compat_sock_common_getsockopt,
987 #endif
988 };
989
990 /* Create a socket. Initialise the socket, blank the addresses
991 * set the state.
992 */
993 static int ieee802154_create(struct net *net, struct socket *sock,
994 int protocol, int kern)
995 {
996 struct sock *sk;
997 int rc;
998 struct proto *proto;
999 const struct proto_ops *ops;
1000
1001 if (!net_eq(net, &init_net))
1002 return -EAFNOSUPPORT;
1003
1004 switch (sock->type) {
1005 case SOCK_RAW:
1006 proto = &ieee802154_raw_prot;
1007 ops = &ieee802154_raw_ops;
1008 break;
1009 case SOCK_DGRAM:
1010 proto = &ieee802154_dgram_prot;
1011 ops = &ieee802154_dgram_ops;
1012 break;
1013 default:
1014 rc = -ESOCKTNOSUPPORT;
1015 goto out;
1016 }
1017
1018 rc = -ENOMEM;
1019 sk = sk_alloc(net, PF_IEEE802154, GFP_KERNEL, proto, kern);
1020 if (!sk)
1021 goto out;
1022 rc = 0;
1023
1024 sock->ops = ops;
1025
1026 sock_init_data(sock, sk);
1027 /* FIXME: sk->sk_destruct */
1028 sk->sk_family = PF_IEEE802154;
1029
1030 /* Checksums on by default */
1031 sock_set_flag(sk, SOCK_ZAPPED);
1032
1033 if (sk->sk_prot->hash) {
1034 rc = sk->sk_prot->hash(sk);
1035 if (rc) {
1036 sk_common_release(sk);
1037 goto out;
1038 }
1039 }
1040
1041 if (sk->sk_prot->init) {
1042 rc = sk->sk_prot->init(sk);
1043 if (rc)
1044 sk_common_release(sk);
1045 }
1046 out:
1047 return rc;
1048 }
1049
1050 static const struct net_proto_family ieee802154_family_ops = {
1051 .family = PF_IEEE802154,
1052 .create = ieee802154_create,
1053 .owner = THIS_MODULE,
1054 };
1055
1056 static int ieee802154_rcv(struct sk_buff *skb, struct net_device *dev,
1057 struct packet_type *pt, struct net_device *orig_dev)
1058 {
1059 if (!netif_running(dev))
1060 goto drop;
1061 pr_debug("got frame, type %d, dev %p\n", dev->type, dev);
1062 #ifdef DEBUG
1063 print_hex_dump_bytes("ieee802154_rcv ",
1064 DUMP_PREFIX_NONE, skb->data, skb->len);
1065 #endif
1066
1067 if (!net_eq(dev_net(dev), &init_net))
1068 goto drop;
1069
1070 ieee802154_raw_deliver(dev, skb);
1071
1072 if (dev->type != ARPHRD_IEEE802154)
1073 goto drop;
1074
1075 if (skb->pkt_type != PACKET_OTHERHOST)
1076 return ieee802154_dgram_deliver(dev, skb);
1077
1078 drop:
1079 kfree_skb(skb);
1080 return NET_RX_DROP;
1081 }
1082
1083 static struct packet_type ieee802154_packet_type = {
1084 .type = htons(ETH_P_IEEE802154),
1085 .func = ieee802154_rcv,
1086 };
1087
1088 static int __init af_ieee802154_init(void)
1089 {
1090 int rc = -EINVAL;
1091
1092 rc = proto_register(&ieee802154_raw_prot, 1);
1093 if (rc)
1094 goto out;
1095
1096 rc = proto_register(&ieee802154_dgram_prot, 1);
1097 if (rc)
1098 goto err_dgram;
1099
1100 /* Tell SOCKET that we are alive */
1101 rc = sock_register(&ieee802154_family_ops);
1102 if (rc)
1103 goto err_sock;
1104 dev_add_pack(&ieee802154_packet_type);
1105
1106 rc = 0;
1107 goto out;
1108
1109 err_sock:
1110 proto_unregister(&ieee802154_dgram_prot);
1111 err_dgram:
1112 proto_unregister(&ieee802154_raw_prot);
1113 out:
1114 return rc;
1115 }
1116
1117 static void __exit af_ieee802154_remove(void)
1118 {
1119 dev_remove_pack(&ieee802154_packet_type);
1120 sock_unregister(PF_IEEE802154);
1121 proto_unregister(&ieee802154_dgram_prot);
1122 proto_unregister(&ieee802154_raw_prot);
1123 }
1124
1125 module_init(af_ieee802154_init);
1126 module_exit(af_ieee802154_remove);
1127
1128 MODULE_LICENSE("GPL");
1129 MODULE_ALIAS_NETPROTO(PF_IEEE802154);
Linux with added FPC-III support