net/ieee802154/socket.c

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