net/ieee802154/socket.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * IEEE802154.4 socket interface
   4  *
   5  * Copyright 2007, 2008 Siemens AG
   6  *
   7  * Written by:
   8  * Sergey Lapin <slapin@ossfans.org>
   9  * Maxim Gorbachyov <maxim.gorbachev@siemens.com>
  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>      /* For TIOCOUTQ/INQ */
  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 /* Utility function for families */
  31 static struct net_device*
  32 ieee802154_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
  81 static 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
  92 static 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
 100 static 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
 111 static 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
 125 static 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
 153 static 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 /* RAW Sockets (802.15.4 created in userspace) */
 171 static HLIST_HEAD(raw_head);
 172 static DEFINE_RWLOCK(raw_lock);
 173
 174 static 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
 184 static 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
 192 static void raw_close(struct sock *sk, long timeout)
 193 {
 194         sk_common_release(sk);
 195 }
 196
 197 static 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         err = ieee802154_sockaddr_check_size(uaddr, len);
 205         if (err < 0)
 206                 return err;
 207
 208         uaddr = (struct sockaddr_ieee802154 *)_uaddr;
 209         if (uaddr->family != AF_IEEE802154)
 210                 return -EINVAL;
 211
 212         lock_sock(sk);
 213
 214         ieee802154_addr_from_sa(&addr, &uaddr->addr);
 215         dev = ieee802154_get_dev(sock_net(sk), &addr);
 216         if (!dev) {
 217                 err = -ENODEV;
 218                 goto out;
 219         }
 220
 221         sk->sk_bound_dev_if = dev->ifindex;
 222         sk_dst_reset(sk);
 223
 224         dev_put(dev);
 225 out:
 226         release_sock(sk);
 227
 228         return err;
 229 }
 230
 231 static int raw_connect(struct sock *sk, struct sockaddr *uaddr,
 232                        int addr_len)
 233 {
 234         return -ENOTSUPP;
 235 }
 236
 237 static int raw_disconnect(struct sock *sk, int flags)
 238 {
 239         return 0;
 240 }
 241
 242 static int raw_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
 243 {
 244         struct net_device *dev;
 245         unsigned int mtu;
 246         struct sk_buff *skb;
 247         int hlen, tlen;
 248         int err;
 249
 250         if (msg->msg_flags & MSG_OOB) {
 251                 pr_debug("msg->msg_flags = 0x%x\n", msg->msg_flags);
 252                 return -EOPNOTSUPP;
 253         }
 254
 255         lock_sock(sk);
 256         if (!sk->sk_bound_dev_if)
 257                 dev = dev_getfirstbyhwtype(sock_net(sk), ARPHRD_IEEE802154);
 258         else
 259                 dev = dev_get_by_index(sock_net(sk), sk->sk_bound_dev_if);
 260         release_sock(sk);
 261
 262         if (!dev) {
 263                 pr_debug("no dev\n");
 264                 err = -ENXIO;
 265                 goto out;
 266         }
 267
 268         mtu = IEEE802154_MTU;
 269         pr_debug("name = %s, mtu = %u\n", dev->name, mtu);
 270
 271         if (size > mtu) {
 272                 pr_debug("size = %zu, mtu = %u\n", size, mtu);
 273                 err = -EMSGSIZE;
 274                 goto out_dev;
 275         }
 276         if (!size) {
 277                 err = 0;
 278                 goto out_dev;
 279         }
 280
 281         hlen = LL_RESERVED_SPACE(dev);
 282         tlen = dev->needed_tailroom;
 283         skb = sock_alloc_send_skb(sk, hlen + tlen + size,
 284                                   msg->msg_flags & MSG_DONTWAIT, &err);
 285         if (!skb)
 286                 goto out_dev;
 287
 288         skb_reserve(skb, hlen);
 289
 290         skb_reset_mac_header(skb);
 291         skb_reset_network_header(skb);
 292
 293         err = memcpy_from_msg(skb_put(skb, size), msg, size);
 294         if (err < 0)
 295                 goto out_skb;
 296
 297         skb->dev = dev;
 298         skb->protocol = htons(ETH_P_IEEE802154);
 299
 300         err = dev_queue_xmit(skb);
 301         if (err > 0)
 302                 err = net_xmit_errno(err);
 303
 304         dev_put(dev);
 305
 306         return err ?: size;
 307
 308 out_skb:
 309         kfree_skb(skb);
 310 out_dev:
 311         dev_put(dev);
 312 out:
 313         return err;
 314 }
 315
 316 static int raw_recvmsg(struct sock *sk, struct msghdr *msg, size_t len,
 317                        int noblock, int flags, int *addr_len)
 318 {
 319         size_t copied = 0;
 320         int err = -EOPNOTSUPP;
 321         struct sk_buff *skb;
 322
 323         skb = skb_recv_datagram(sk, flags, noblock, &err);
 324         if (!skb)
 325                 goto out;
 326
 327         copied = skb->len;
 328         if (len < copied) {
 329                 msg->msg_flags |= MSG_TRUNC;
 330                 copied = len;
 331         }
 332
 333         err = skb_copy_datagram_msg(skb, 0, msg, copied);
 334         if (err)
 335                 goto done;
 336
 337         sock_recv_ts_and_drops(msg, sk, skb);
 338
 339         if (flags & MSG_TRUNC)
 340                 copied = skb->len;
 341 done:
 342         skb_free_datagram(sk, skb);
 343 out:
 344         if (err)
 345                 return err;
 346         return copied;
 347 }
 348
 349 static int raw_rcv_skb(struct sock *sk, struct sk_buff *skb)
 350 {
 351         skb = skb_share_check(skb, GFP_ATOMIC);
 352         if (!skb)
 353                 return NET_RX_DROP;
 354
 355         if (sock_queue_rcv_skb(sk, skb) < 0) {
 356                 kfree_skb(skb);
 357                 return NET_RX_DROP;
 358         }
 359
 360         return NET_RX_SUCCESS;
 361 }
 362
 363 static void ieee802154_raw_deliver(struct net_device *dev, struct sk_buff *skb)
 364 {
 365         struct sock *sk;
 366
 367         read_lock(&raw_lock);
 368         sk_for_each(sk, &raw_head) {
 369                 bh_lock_sock(sk);
 370                 if (!sk->sk_bound_dev_if ||
 371                     sk->sk_bound_dev_if == dev->ifindex) {
 372                         struct sk_buff *clone;
 373
 374                         clone = skb_clone(skb, GFP_ATOMIC);
 375                         if (clone)
 376                                 raw_rcv_skb(sk, clone);
 377                 }
 378                 bh_unlock_sock(sk);
 379         }
 380         read_unlock(&raw_lock);
 381 }
 382
 383 static int raw_getsockopt(struct sock *sk, int level, int optname,
 384                           char __user *optval, int __user *optlen)
 385 {
 386         return -EOPNOTSUPP;
 387 }
 388
 389 static int raw_setsockopt(struct sock *sk, int level, int optname,
 390                           char __user *optval, unsigned int optlen)
 391 {
 392         return -EOPNOTSUPP;
 393 }
 394
 395 static struct proto ieee802154_raw_prot = {
 396         .name           = "IEEE-802.15.4-RAW",
 397         .owner          = THIS_MODULE,
 398         .obj_size       = sizeof(struct sock),
 399         .close          = raw_close,
 400         .bind           = raw_bind,
 401         .sendmsg        = raw_sendmsg,
 402         .recvmsg        = raw_recvmsg,
 403         .hash           = raw_hash,
 404         .unhash         = raw_unhash,
 405         .connect        = raw_connect,
 406         .disconnect     = raw_disconnect,
 407         .getsockopt     = raw_getsockopt,
 408         .setsockopt     = raw_setsockopt,
 409 };
 410
 411 static const struct proto_ops ieee802154_raw_ops = {
 412         .family            = PF_IEEE802154,
 413         .owner             = THIS_MODULE,
 414         .release           = ieee802154_sock_release,
 415         .bind              = ieee802154_sock_bind,
 416         .connect           = ieee802154_sock_connect,
 417         .socketpair        = sock_no_socketpair,
 418         .accept            = sock_no_accept,
 419         .getname           = sock_no_getname,
 420         .poll              = datagram_poll,
 421         .ioctl             = ieee802154_sock_ioctl,
 422         .gettstamp         = sock_gettstamp,
 423         .listen            = sock_no_listen,
 424         .shutdown          = sock_no_shutdown,
 425         .setsockopt        = sock_common_setsockopt,
 426         .getsockopt        = sock_common_getsockopt,
 427         .sendmsg           = ieee802154_sock_sendmsg,
 428         .recvmsg           = sock_common_recvmsg,
 429         .mmap              = sock_no_mmap,
 430         .sendpage          = sock_no_sendpage,
 431 #ifdef CONFIG_COMPAT
 432         .compat_setsockopt = compat_sock_common_setsockopt,
 433         .compat_getsockopt = compat_sock_common_getsockopt,
 434 #endif
 435 };
 436
 437 /* DGRAM Sockets (802.15.4 dataframes) */
 438 static HLIST_HEAD(dgram_head);
 439 static DEFINE_RWLOCK(dgram_lock);
 440
 441 struct dgram_sock {
 442         struct sock sk;
 443
 444         struct ieee802154_addr src_addr;
 445         struct ieee802154_addr dst_addr;
 446
 447         unsigned int bound:1;
 448         unsigned int connected:1;
 449         unsigned int want_ack:1;
 450         unsigned int want_lqi:1;
 451         unsigned int secen:1;
 452         unsigned int secen_override:1;
 453         unsigned int seclevel:3;
 454         unsigned int seclevel_override:1;
 455 };
 456
 457 static inline struct dgram_sock *dgram_sk(const struct sock *sk)
 458 {
 459         return container_of(sk, struct dgram_sock, sk);
 460 }
 461
 462 static int dgram_hash(struct sock *sk)
 463 {
 464         write_lock_bh(&dgram_lock);
 465         sk_add_node(sk, &dgram_head);
 466         sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
 467         write_unlock_bh(&dgram_lock);
 468
 469         return 0;
 470 }
 471
 472 static void dgram_unhash(struct sock *sk)
 473 {
 474         write_lock_bh(&dgram_lock);
 475         if (sk_del_node_init(sk))
 476                 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
 477         write_unlock_bh(&dgram_lock);
 478 }
 479
 480 static int dgram_init(struct sock *sk)
 481 {
 482         struct dgram_sock *ro = dgram_sk(sk);
 483
 484         ro->want_ack = 1;
 485         ro->want_lqi = 0;
 486         return 0;
 487 }
 488
 489 static void dgram_close(struct sock *sk, long timeout)
 490 {
 491         sk_common_release(sk);
 492 }
 493
 494 static int dgram_bind(struct sock *sk, struct sockaddr *uaddr, int len)
 495 {
 496         struct sockaddr_ieee802154 *addr = (struct sockaddr_ieee802154 *)uaddr;
 497         struct ieee802154_addr haddr;
 498         struct dgram_sock *ro = dgram_sk(sk);
 499         int err = -EINVAL;
 500         struct net_device *dev;
 501
 502         lock_sock(sk);
 503
 504         ro->bound = 0;
 505
 506         err = ieee802154_sockaddr_check_size(addr, len);
 507         if (err < 0)
 508                 goto out;
 509
 510         if (addr->family != AF_IEEE802154) {
 511                 err = -EINVAL;
 512                 goto out;
 513         }
 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;
 531 out_put:
 532         dev_put(dev);
 533 out:
 534         release_sock(sk);
 535
 536         return err;
 537 }
 538
 539 static 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                         /* We will only return the amount
 559                          * of this packet since that is all
 560                          * that will be read.
 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 /* FIXME: autobind */
 573 static 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         err = ieee802154_sockaddr_check_size(addr, len);
 581         if (err < 0)
 582                 return err;
 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         DECLARE_SOCKADDR(struct sockaddr_ieee802154*, daddr, msg->msg_name);
 622         int hlen, tlen;
 623         int err;
 624
 625         if (msg->msg_flags & MSG_OOB) {
 626                 pr_debug("msg->msg_flags = 0x%x\n", msg->msg_flags);
 627                 return -EOPNOTSUPP;
 628         }
 629
 630         if (msg->msg_name) {
 631                 if (ro->connected)
 632                         return -EISCONN;
 633                 if (msg->msg_namelen < IEEE802154_MIN_NAMELEN)
 634                         return -EINVAL;
 635                 err = ieee802154_sockaddr_check_size(daddr, msg->msg_namelen);
 636                 if (err < 0)
 637                         return err;
 638                 ieee802154_addr_from_sa(&dst_addr, &daddr->addr);
 639         } else {
 640                 if (!ro->connected)
 641                         return -EDESTADDRREQ;
 642                 dst_addr = ro->dst_addr;
 643         }
 644
 645         if (!ro->bound)
 646                 dev = dev_getfirstbyhwtype(sock_net(sk), ARPHRD_IEEE802154);
 647         else
 648                 dev = ieee802154_get_dev(sock_net(sk), &ro->src_addr);
 649
 650         if (!dev) {
 651                 pr_debug("no dev\n");
 652                 err = -ENXIO;
 653                 goto out;
 654         }
 655         mtu = IEEE802154_MTU;
 656         pr_debug("name = %s, mtu = %u\n", dev->name, mtu);
 657
 658         if (size > mtu) {
 659                 pr_debug("size = %zu, mtu = %u\n", size, mtu);
 660                 err = -EMSGSIZE;
 661                 goto out_dev;
 662         }
 663
 664         hlen = LL_RESERVED_SPACE(dev);
 665         tlen = dev->needed_tailroom;
 666         skb = sock_alloc_send_skb(sk, hlen + tlen + size,
 667                                   msg->msg_flags & MSG_DONTWAIT,
 668                                   &err);
 669         if (!skb)
 670                 goto out_dev;
 671
 672         skb_reserve(skb, hlen);
 673
 674         skb_reset_network_header(skb);
 675
 676         cb = mac_cb_init(skb);
 677         cb->type = IEEE802154_FC_TYPE_DATA;
 678         cb->ackreq = ro->want_ack;
 679         cb->secen = ro->secen;
 680         cb->secen_override = ro->secen_override;
 681         cb->seclevel = ro->seclevel;
 682         cb->seclevel_override = ro->seclevel_override;
 683
 684         err = wpan_dev_hard_header(skb, dev, &dst_addr,
 685                                    ro->bound ? &ro->src_addr : NULL, size);
 686         if (err < 0)
 687                 goto out_skb;
 688
 689         err = memcpy_from_msg(skb_put(skb, size), msg, size);
 690         if (err < 0)
 691                 goto out_skb;
 692
 693         skb->dev = dev;
 694         skb->protocol = htons(ETH_P_IEEE802154);
 695
 696         err = dev_queue_xmit(skb);
 697         if (err > 0)
 698                 err = net_xmit_errno(err);
 699
 700         dev_put(dev);
 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         struct dgram_sock *ro = dgram_sk(sk);
 719         DECLARE_SOCKADDR(struct sockaddr_ieee802154 *, saddr, msg->msg_name);
 720
 721         skb = skb_recv_datagram(sk, flags, noblock, &err);
 722         if (!skb)
 723                 goto out;
 724
 725         copied = skb->len;
 726         if (len < copied) {
 727                 msg->msg_flags |= MSG_TRUNC;
 728                 copied = len;
 729         }
 730
 731         /* FIXME: skip headers if necessary ?! */
 732         err = skb_copy_datagram_msg(skb, 0, msg, copied);
 733         if (err)
 734                 goto done;
 735
 736         sock_recv_ts_and_drops(msg, sk, skb);
 737
 738         if (saddr) {
 739                 /* Clear the implicit padding in struct sockaddr_ieee802154
 740                  * (16 bits between 'family' and 'addr') and in struct
 741                  * ieee802154_addr_sa (16 bits at the end of the structure).
 742                  */
 743                 memset(saddr, 0, sizeof(*saddr));
 744
 745                 saddr->family = AF_IEEE802154;
 746                 ieee802154_addr_to_sa(&saddr->addr, &mac_cb(skb)->source);
 747                 *addr_len = sizeof(*saddr);
 748         }
 749
 750         if (ro->want_lqi) {
 751                 err = put_cmsg(msg, SOL_IEEE802154, WPAN_WANTLQI,
 752                                sizeof(uint8_t), &(mac_cb(skb)->lqi));
 753                 if (err)
 754                         goto done;
 755         }
 756
 757         if (flags & MSG_TRUNC)
 758                 copied = skb->len;
 759 done:
 760         skb_free_datagram(sk, skb);
 761 out:
 762         if (err)
 763                 return err;
 764         return copied;
 765 }
 766
 767 static int dgram_rcv_skb(struct sock *sk, struct sk_buff *skb)
 768 {
 769         skb = skb_share_check(skb, GFP_ATOMIC);
 770         if (!skb)
 771                 return NET_RX_DROP;
 772
 773         if (sock_queue_rcv_skb(sk, skb) < 0) {
 774                 kfree_skb(skb);
 775                 return NET_RX_DROP;
 776         }
 777
 778         return NET_RX_SUCCESS;
 779 }
 780
 781 static inline bool
 782 ieee802154_match_sock(__le64 hw_addr, __le16 pan_id, __le16 short_addr,
 783                       struct dgram_sock *ro)
 784 {
 785         if (!ro->bound)
 786                 return true;
 787
 788         if (ro->src_addr.mode == IEEE802154_ADDR_LONG &&
 789             hw_addr == ro->src_addr.extended_addr)
 790                 return true;
 791
 792         if (ro->src_addr.mode == IEEE802154_ADDR_SHORT &&
 793             pan_id == ro->src_addr.pan_id &&
 794             short_addr == ro->src_addr.short_addr)
 795                 return true;
 796
 797         return false;
 798 }
 799
 800 static int ieee802154_dgram_deliver(struct net_device *dev, struct sk_buff *skb)
 801 {
 802         struct sock *sk, *prev = NULL;
 803         int ret = NET_RX_SUCCESS;
 804         __le16 pan_id, short_addr;
 805         __le64 hw_addr;
 806
 807         /* Data frame processing */
 808         BUG_ON(dev->type != ARPHRD_IEEE802154);
 809
 810         pan_id = dev->ieee802154_ptr->pan_id;
 811         short_addr = dev->ieee802154_ptr->short_addr;
 812         hw_addr = dev->ieee802154_ptr->extended_addr;
 813
 814         read_lock(&dgram_lock);
 815         sk_for_each(sk, &dgram_head) {
 816                 if (ieee802154_match_sock(hw_addr, pan_id, short_addr,
 817                                           dgram_sk(sk))) {
 818                         if (prev) {
 819                                 struct sk_buff *clone;
 820
 821                                 clone = skb_clone(skb, GFP_ATOMIC);
 822                                 if (clone)
 823                                         dgram_rcv_skb(prev, clone);
 824                         }
 825
 826                         prev = sk;
 827                 }
 828         }
 829
 830         if (prev) {
 831                 dgram_rcv_skb(prev, skb);
 832         } else {
 833                 kfree_skb(skb);
 834                 ret = NET_RX_DROP;
 835         }
 836         read_unlock(&dgram_lock);
 837
 838         return ret;
 839 }
 840
 841 static int dgram_getsockopt(struct sock *sk, int level, int optname,
 842                             char __user *optval, int __user *optlen)
 843 {
 844         struct dgram_sock *ro = dgram_sk(sk);
 845
 846         int val, len;
 847
 848         if (level != SOL_IEEE802154)
 849                 return -EOPNOTSUPP;
 850
 851         if (get_user(len, optlen))
 852                 return -EFAULT;
 853
 854         len = min_t(unsigned int, len, sizeof(int));
 855
 856         switch (optname) {
 857         case WPAN_WANTACK:
 858                 val = ro->want_ack;
 859                 break;
 860         case WPAN_WANTLQI:
 861                 val = ro->want_lqi;
 862                 break;
 863         case WPAN_SECURITY:
 864                 if (!ro->secen_override)
 865                         val = WPAN_SECURITY_DEFAULT;
 866                 else if (ro->secen)
 867                         val = WPAN_SECURITY_ON;
 868                 else
 869                         val = WPAN_SECURITY_OFF;
 870                 break;
 871         case WPAN_SECURITY_LEVEL:
 872                 if (!ro->seclevel_override)
 873                         val = WPAN_SECURITY_LEVEL_DEFAULT;
 874                 else
 875                         val = ro->seclevel;
 876                 break;
 877         default:
 878                 return -ENOPROTOOPT;
 879         }
 880
 881         if (put_user(len, optlen))
 882                 return -EFAULT;
 883         if (copy_to_user(optval, &val, len))
 884                 return -EFAULT;
 885         return 0;
 886 }
 887
 888 static int dgram_setsockopt(struct sock *sk, int level, int optname,
 889                             char __user *optval, unsigned int optlen)
 890 {
 891         struct dgram_sock *ro = dgram_sk(sk);
 892         struct net *net = sock_net(sk);
 893         int val;
 894         int err = 0;
 895
 896         if (optlen < sizeof(int))
 897                 return -EINVAL;
 898
 899         if (get_user(val, (int __user *)optval))
 900                 return -EFAULT;
 901
 902         lock_sock(sk);
 903
 904         switch (optname) {
 905         case WPAN_WANTACK:
 906                 ro->want_ack = !!val;
 907                 break;
 908         case WPAN_WANTLQI:
 909                 ro->want_lqi = !!val;
 910                 break;
 911         case WPAN_SECURITY:
 912                 if (!ns_capable(net->user_ns, CAP_NET_ADMIN) &&
 913                     !ns_capable(net->user_ns, CAP_NET_RAW)) {
 914                         err = -EPERM;
 915                         break;
 916                 }
 917
 918                 switch (val) {
 919                 case WPAN_SECURITY_DEFAULT:
 920                         ro->secen_override = 0;
 921                         break;
 922                 case WPAN_SECURITY_ON:
 923                         ro->secen_override = 1;
 924                         ro->secen = 1;
 925                         break;
 926                 case WPAN_SECURITY_OFF:
 927                         ro->secen_override = 1;
 928                         ro->secen = 0;
 929                         break;
 930                 default:
 931                         err = -EINVAL;
 932                         break;
 933                 }
 934                 break;
 935         case WPAN_SECURITY_LEVEL:
 936                 if (!ns_capable(net->user_ns, CAP_NET_ADMIN) &&
 937                     !ns_capable(net->user_ns, CAP_NET_RAW)) {
 938                         err = -EPERM;
 939                         break;
 940                 }
 941
 942                 if (val < WPAN_SECURITY_LEVEL_DEFAULT ||
 943                     val > IEEE802154_SCF_SECLEVEL_ENC_MIC128) {
 944                         err = -EINVAL;
 945                 } else if (val == WPAN_SECURITY_LEVEL_DEFAULT) {
 946                         ro->seclevel_override = 0;
 947                 } else {
 948                         ro->seclevel_override = 1;
 949                         ro->seclevel = val;
 950                 }
 951                 break;
 952         default:
 953                 err = -ENOPROTOOPT;
 954                 break;
 955         }
 956
 957         release_sock(sk);
 958         return err;
 959 }
 960
 961 static struct proto ieee802154_dgram_prot = {
 962         .name           = "IEEE-802.15.4-MAC",
 963         .owner          = THIS_MODULE,
 964         .obj_size       = sizeof(struct dgram_sock),
 965         .init           = dgram_init,
 966         .close          = dgram_close,
 967         .bind           = dgram_bind,
 968         .sendmsg        = dgram_sendmsg,
 969         .recvmsg        = dgram_recvmsg,
 970         .hash           = dgram_hash,
 971         .unhash         = dgram_unhash,
 972         .connect        = dgram_connect,
 973         .disconnect     = dgram_disconnect,
 974         .ioctl          = dgram_ioctl,
 975         .getsockopt     = dgram_getsockopt,
 976         .setsockopt     = dgram_setsockopt,
 977 };
 978
 979 static const struct proto_ops ieee802154_dgram_ops = {
 980         .family            = PF_IEEE802154,
 981         .owner             = THIS_MODULE,
 982         .release           = ieee802154_sock_release,
 983         .bind              = ieee802154_sock_bind,
 984         .connect           = ieee802154_sock_connect,
 985         .socketpair        = sock_no_socketpair,
 986         .accept            = sock_no_accept,
 987         .getname           = sock_no_getname,
 988         .poll              = datagram_poll,
 989         .ioctl             = ieee802154_sock_ioctl,
 990         .gettstamp         = sock_gettstamp,
 991         .listen            = sock_no_listen,
 992         .shutdown          = sock_no_shutdown,
 993         .setsockopt        = sock_common_setsockopt,
 994         .getsockopt        = sock_common_getsockopt,
 995         .sendmsg           = ieee802154_sock_sendmsg,
 996         .recvmsg           = sock_common_recvmsg,
 997         .mmap              = sock_no_mmap,
 998         .sendpage          = sock_no_sendpage,
 999 #ifdef CONFIG_COMPAT
1000         .compat_setsockopt = compat_sock_common_setsockopt,
1001         .compat_getsockopt = compat_sock_common_getsockopt,
1002 #endif
1003 };
1004
1005 static void ieee802154_sock_destruct(struct sock *sk)
1006 {
1007         skb_queue_purge(&sk->sk_receive_queue);
1008 }
1009
1010 /* Create a socket. Initialise the socket, blank the addresses
1011  * set the state.
1012  */
1013 static int ieee802154_create(struct net *net, struct socket *sock,
1014                              int protocol, int kern)
1015 {
1016         struct sock *sk;
1017         int rc;
1018         struct proto *proto;
1019         const struct proto_ops *ops;
1020
1021         if (!net_eq(net, &init_net))
1022                 return -EAFNOSUPPORT;
1023
1024         switch (sock->type) {
1025         case SOCK_RAW:
1026                 rc = -EPERM;
1027                 if (!capable(CAP_NET_RAW))
1028                         goto out;
1029                 proto = &ieee802154_raw_prot;
1030                 ops = &ieee802154_raw_ops;
1031                 break;
1032         case SOCK_DGRAM:
1033                 proto = &ieee802154_dgram_prot;
1034                 ops = &ieee802154_dgram_ops;
1035                 break;
1036         default:
1037                 rc = -ESOCKTNOSUPPORT;
1038                 goto out;
1039         }
1040
1041         rc = -ENOMEM;
1042         sk = sk_alloc(net, PF_IEEE802154, GFP_KERNEL, proto, kern);
1043         if (!sk)
1044                 goto out;
1045         rc = 0;
1046
1047         sock->ops = ops;
1048
1049         sock_init_data(sock, sk);
1050         sk->sk_destruct = ieee802154_sock_destruct;
1051         sk->sk_family = PF_IEEE802154;
1052
1053         /* Checksums on by default */
1054         sock_set_flag(sk, SOCK_ZAPPED);
1055
1056         if (sk->sk_prot->hash) {
1057                 rc = sk->sk_prot->hash(sk);
1058                 if (rc) {
1059                         sk_common_release(sk);
1060                         goto out;
1061                 }
1062         }
1063
1064         if (sk->sk_prot->init) {
1065                 rc = sk->sk_prot->init(sk);
1066                 if (rc)
1067                         sk_common_release(sk);
1068         }
1069 out:
1070         return rc;
1071 }
1072
1073 static const struct net_proto_family ieee802154_family_ops = {
1074         .family         = PF_IEEE802154,
1075         .create         = ieee802154_create,
1076         .owner          = THIS_MODULE,
1077 };
1078
1079 static int ieee802154_rcv(struct sk_buff *skb, struct net_device *dev,
1080                           struct packet_type *pt, struct net_device *orig_dev)
1081 {
1082         if (!netif_running(dev))
1083                 goto drop;
1084         pr_debug("got frame, type %d, dev %p\n", dev->type, dev);
1085 #ifdef DEBUG
1086         print_hex_dump_bytes("ieee802154_rcv ",
1087                              DUMP_PREFIX_NONE, skb->data, skb->len);
1088 #endif
1089
1090         if (!net_eq(dev_net(dev), &init_net))
1091                 goto drop;
1092
1093         ieee802154_raw_deliver(dev, skb);
1094
1095         if (dev->type != ARPHRD_IEEE802154)
1096                 goto drop;
1097
1098         if (skb->pkt_type != PACKET_OTHERHOST)
1099                 return ieee802154_dgram_deliver(dev, skb);
1100
1101 drop:
1102         kfree_skb(skb);
1103         return NET_RX_DROP;
1104 }
1105
1106 static struct packet_type ieee802154_packet_type = {
1107         .type = htons(ETH_P_IEEE802154),
1108         .func = ieee802154_rcv,
1109 };
1110
1111 static int __init af_ieee802154_init(void)
1112 {
1113         int rc;
1114
1115         rc = proto_register(&ieee802154_raw_prot, 1);
1116         if (rc)
1117                 goto out;
1118
1119         rc = proto_register(&ieee802154_dgram_prot, 1);
1120         if (rc)
1121                 goto err_dgram;
1122
1123         /* Tell SOCKET that we are alive */
1124         rc = sock_register(&ieee802154_family_ops);
1125         if (rc)
1126                 goto err_sock;
1127         dev_add_pack(&ieee802154_packet_type);
1128
1129         rc = 0;
1130         goto out;
1131
1132 err_sock:
1133         proto_unregister(&ieee802154_dgram_prot);
1134 err_dgram:
1135         proto_unregister(&ieee802154_raw_prot);
1136 out:
1137         return rc;
1138 }
1139
1140 static void __exit af_ieee802154_remove(void)
1141 {
1142         dev_remove_pack(&ieee802154_packet_type);
1143         sock_unregister(PF_IEEE802154);
1144         proto_unregister(&ieee802154_dgram_prot);
1145         proto_unregister(&ieee802154_raw_prot);
1146 }
1147
1148 module_init(af_ieee802154_init);
1149 module_exit(af_ieee802154_remove);
1150
1151 MODULE_LICENSE("GPL");
1152 MODULE_ALIAS_NETPROTO(PF_IEEE802154);