net/ieee802154/af_ieee802154.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 <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 #include "af802154.h"
  38
  39 /* Utility function for families */
  40 struct net_device*
  41 ieee802154_get_dev(struct net *net, const struct ieee802154_addr *addr)
  42 {
  43         struct net_device *dev = NULL;
  44         struct net_device *tmp;
  45         __le16 pan_id, short_addr;
  46         u8 hwaddr[IEEE802154_ADDR_LEN];
  47
  48         switch (addr->mode) {
  49         case IEEE802154_ADDR_LONG:
  50                 ieee802154_devaddr_to_raw(hwaddr, addr->extended_addr);
  51                 rcu_read_lock();
  52                 dev = dev_getbyhwaddr_rcu(net, ARPHRD_IEEE802154, hwaddr);
  53                 if (dev)
  54                         dev_hold(dev);
  55                 rcu_read_unlock();
  56                 break;
  57         case IEEE802154_ADDR_SHORT:
  58                 if (addr->pan_id == cpu_to_le16(IEEE802154_PANID_BROADCAST) ||
  59                     addr->short_addr == cpu_to_le16(IEEE802154_ADDR_UNDEF) ||
  60                     addr->short_addr == cpu_to_le16(IEEE802154_ADDR_BROADCAST))
  61                         break;
  62
  63                 rtnl_lock();
  64
  65                 for_each_netdev(net, tmp) {
  66                         if (tmp->type != ARPHRD_IEEE802154)
  67                                 continue;
  68
  69                         pan_id = ieee802154_mlme_ops(tmp)->get_pan_id(tmp);
  70                         short_addr =
  71                                 ieee802154_mlme_ops(tmp)->get_short_addr(tmp);
  72
  73                         if (pan_id == addr->pan_id &&
  74                             short_addr == addr->short_addr) {
  75                                 dev = tmp;
  76                                 dev_hold(dev);
  77                                 break;
  78                         }
  79                 }
  80
  81                 rtnl_unlock();
  82                 break;
  83         default:
  84                 pr_warn("Unsupported ieee802154 address type: %d\n",
  85                         addr->mode);
  86                 break;
  87         }
  88
  89         return dev;
  90 }
  91
  92 static int ieee802154_sock_release(struct socket *sock)
  93 {
  94         struct sock *sk = sock->sk;
  95
  96         if (sk) {
  97                 sock->sk = NULL;
  98                 sk->sk_prot->close(sk, 0);
  99         }
 100         return 0;
 101 }
 102
 103 static int ieee802154_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
 104                                    struct msghdr *msg, size_t len)
 105 {
 106         struct sock *sk = sock->sk;
 107
 108         return sk->sk_prot->sendmsg(iocb, sk, msg, len);
 109 }
 110
 111 static int ieee802154_sock_bind(struct socket *sock, struct sockaddr *uaddr,
 112                                 int addr_len)
 113 {
 114         struct sock *sk = sock->sk;
 115
 116         if (sk->sk_prot->bind)
 117                 return sk->sk_prot->bind(sk, uaddr, addr_len);
 118
 119         return sock_no_bind(sock, uaddr, addr_len);
 120 }
 121
 122 static int ieee802154_sock_connect(struct socket *sock, struct sockaddr *uaddr,
 123                                    int addr_len, int flags)
 124 {
 125         struct sock *sk = sock->sk;
 126
 127         if (addr_len < sizeof(uaddr->sa_family))
 128                 return -EINVAL;
 129
 130         if (uaddr->sa_family == AF_UNSPEC)
 131                 return sk->sk_prot->disconnect(sk, flags);
 132
 133         return sk->sk_prot->connect(sk, uaddr, addr_len);
 134 }
 135
 136 static int ieee802154_dev_ioctl(struct sock *sk, struct ifreq __user *arg,
 137                                 unsigned int cmd)
 138 {
 139         struct ifreq ifr;
 140         int ret = -ENOIOCTLCMD;
 141         struct net_device *dev;
 142
 143         if (copy_from_user(&ifr, arg, sizeof(struct ifreq)))
 144                 return -EFAULT;
 145
 146         ifr.ifr_name[IFNAMSIZ-1] = 0;
 147
 148         dev_load(sock_net(sk), ifr.ifr_name);
 149         dev = dev_get_by_name(sock_net(sk), ifr.ifr_name);
 150
 151         if (!dev)
 152                 return -ENODEV;
 153
 154         if (dev->type == ARPHRD_IEEE802154 && dev->netdev_ops->ndo_do_ioctl)
 155                 ret = dev->netdev_ops->ndo_do_ioctl(dev, &ifr, cmd);
 156
 157         if (!ret && copy_to_user(arg, &ifr, sizeof(struct ifreq)))
 158                 ret = -EFAULT;
 159         dev_put(dev);
 160
 161         return ret;
 162 }
 163
 164 static int ieee802154_sock_ioctl(struct socket *sock, unsigned int cmd,
 165                                  unsigned long arg)
 166 {
 167         struct sock *sk = sock->sk;
 168
 169         switch (cmd) {
 170         case SIOCGSTAMP:
 171                 return sock_get_timestamp(sk, (struct timeval __user *)arg);
 172         case SIOCGSTAMPNS:
 173                 return sock_get_timestampns(sk, (struct timespec __user *)arg);
 174         case SIOCGIFADDR:
 175         case SIOCSIFADDR:
 176                 return ieee802154_dev_ioctl(sk, (struct ifreq __user *)arg,
 177                                 cmd);
 178         default:
 179                 if (!sk->sk_prot->ioctl)
 180                         return -ENOIOCTLCMD;
 181                 return sk->sk_prot->ioctl(sk, cmd, arg);
 182         }
 183 }
 184
 185 static const struct proto_ops ieee802154_raw_ops = {
 186         .family            = PF_IEEE802154,
 187         .owner             = THIS_MODULE,
 188         .release           = ieee802154_sock_release,
 189         .bind              = ieee802154_sock_bind,
 190         .connect           = ieee802154_sock_connect,
 191         .socketpair        = sock_no_socketpair,
 192         .accept            = sock_no_accept,
 193         .getname           = sock_no_getname,
 194         .poll              = datagram_poll,
 195         .ioctl             = ieee802154_sock_ioctl,
 196         .listen            = sock_no_listen,
 197         .shutdown          = sock_no_shutdown,
 198         .setsockopt        = sock_common_setsockopt,
 199         .getsockopt        = sock_common_getsockopt,
 200         .sendmsg           = ieee802154_sock_sendmsg,
 201         .recvmsg           = sock_common_recvmsg,
 202         .mmap              = sock_no_mmap,
 203         .sendpage          = sock_no_sendpage,
 204 #ifdef CONFIG_COMPAT
 205         .compat_setsockopt = compat_sock_common_setsockopt,
 206         .compat_getsockopt = compat_sock_common_getsockopt,
 207 #endif
 208 };
 209
 210 static const struct proto_ops ieee802154_dgram_ops = {
 211         .family            = PF_IEEE802154,
 212         .owner             = THIS_MODULE,
 213         .release           = ieee802154_sock_release,
 214         .bind              = ieee802154_sock_bind,
 215         .connect           = ieee802154_sock_connect,
 216         .socketpair        = sock_no_socketpair,
 217         .accept            = sock_no_accept,
 218         .getname           = sock_no_getname,
 219         .poll              = datagram_poll,
 220         .ioctl             = ieee802154_sock_ioctl,
 221         .listen            = sock_no_listen,
 222         .shutdown          = sock_no_shutdown,
 223         .setsockopt        = sock_common_setsockopt,
 224         .getsockopt        = sock_common_getsockopt,
 225         .sendmsg           = ieee802154_sock_sendmsg,
 226         .recvmsg           = sock_common_recvmsg,
 227         .mmap              = sock_no_mmap,
 228         .sendpage          = sock_no_sendpage,
 229 #ifdef CONFIG_COMPAT
 230         .compat_setsockopt = compat_sock_common_setsockopt,
 231         .compat_getsockopt = compat_sock_common_getsockopt,
 232 #endif
 233 };
 234
 235 /* Create a socket. Initialise the socket, blank the addresses
 236  * set the state.
 237  */
 238 static int ieee802154_create(struct net *net, struct socket *sock,
 239                              int protocol, int kern)
 240 {
 241         struct sock *sk;
 242         int rc;
 243         struct proto *proto;
 244         const struct proto_ops *ops;
 245
 246         if (!net_eq(net, &init_net))
 247                 return -EAFNOSUPPORT;
 248
 249         switch (sock->type) {
 250         case SOCK_RAW:
 251                 proto = &ieee802154_raw_prot;
 252                 ops = &ieee802154_raw_ops;
 253                 break;
 254         case SOCK_DGRAM:
 255                 proto = &ieee802154_dgram_prot;
 256                 ops = &ieee802154_dgram_ops;
 257                 break;
 258         default:
 259                 rc = -ESOCKTNOSUPPORT;
 260                 goto out;
 261         }
 262
 263         rc = -ENOMEM;
 264         sk = sk_alloc(net, PF_IEEE802154, GFP_KERNEL, proto);
 265         if (!sk)
 266                 goto out;
 267         rc = 0;
 268
 269         sock->ops = ops;
 270
 271         sock_init_data(sock, sk);
 272         /* FIXME: sk->sk_destruct */
 273         sk->sk_family = PF_IEEE802154;
 274
 275         /* Checksums on by default */
 276         sock_set_flag(sk, SOCK_ZAPPED);
 277
 278         if (sk->sk_prot->hash)
 279                 sk->sk_prot->hash(sk);
 280
 281         if (sk->sk_prot->init) {
 282                 rc = sk->sk_prot->init(sk);
 283                 if (rc)
 284                         sk_common_release(sk);
 285         }
 286 out:
 287         return rc;
 288 }
 289
 290 static const struct net_proto_family ieee802154_family_ops = {
 291         .family         = PF_IEEE802154,
 292         .create         = ieee802154_create,
 293         .owner          = THIS_MODULE,
 294 };
 295
 296 static int ieee802154_rcv(struct sk_buff *skb, struct net_device *dev,
 297                           struct packet_type *pt, struct net_device *orig_dev)
 298 {
 299         if (!netif_running(dev))
 300                 goto drop;
 301         pr_debug("got frame, type %d, dev %p\n", dev->type, dev);
 302 #ifdef DEBUG
 303         print_hex_dump_bytes("ieee802154_rcv ",
 304                              DUMP_PREFIX_NONE, skb->data, skb->len);
 305 #endif
 306
 307         if (!net_eq(dev_net(dev), &init_net))
 308                 goto drop;
 309
 310         ieee802154_raw_deliver(dev, skb);
 311
 312         if (dev->type != ARPHRD_IEEE802154)
 313                 goto drop;
 314
 315         if (skb->pkt_type != PACKET_OTHERHOST)
 316                 return ieee802154_dgram_deliver(dev, skb);
 317
 318 drop:
 319         kfree_skb(skb);
 320         return NET_RX_DROP;
 321 }
 322
 323 static struct packet_type ieee802154_packet_type = {
 324         .type = htons(ETH_P_IEEE802154),
 325         .func = ieee802154_rcv,
 326 };
 327
 328 static int __init af_ieee802154_init(void)
 329 {
 330         int rc = -EINVAL;
 331
 332         rc = proto_register(&ieee802154_raw_prot, 1);
 333         if (rc)
 334                 goto out;
 335
 336         rc = proto_register(&ieee802154_dgram_prot, 1);
 337         if (rc)
 338                 goto err_dgram;
 339
 340         /* Tell SOCKET that we are alive */
 341         rc = sock_register(&ieee802154_family_ops);
 342         if (rc)
 343                 goto err_sock;
 344         dev_add_pack(&ieee802154_packet_type);
 345
 346         rc = 0;
 347         goto out;
 348
 349 err_sock:
 350         proto_unregister(&ieee802154_dgram_prot);
 351 err_dgram:
 352         proto_unregister(&ieee802154_raw_prot);
 353 out:
 354         return rc;
 355 }
 356
 357 static void __exit af_ieee802154_remove(void)
 358 {
 359         dev_remove_pack(&ieee802154_packet_type);
 360         sock_unregister(PF_IEEE802154);
 361         proto_unregister(&ieee802154_dgram_prot);
 362         proto_unregister(&ieee802154_raw_prot);
 363 }
 364
 365 module_init(af_ieee802154_init);
 366 module_exit(af_ieee802154_remove);
 367
 368 MODULE_LICENSE("GPL");
 369 MODULE_ALIAS_NETPROTO(PF_IEEE802154);