include cleanup: Update gfp.h and slab.h includes to prepare for breaking implicit...
[safe/jmp/linux-2.6] / 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  * You should have received a copy of the GNU General Public License along
16  * with this program; if not, write to the Free Software Foundation, Inc.,
17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  * Written by:
20  * Sergey Lapin <slapin@ossfans.org>
21  * Maxim Gorbachyov <maxim.gorbachev@siemens.com>
22  */
23
24 #include <linux/net.h>
25 #include <linux/capability.h>
26 #include <linux/module.h>
27 #include <linux/if_arp.h>
28 #include <linux/if.h>
29 #include <linux/termios.h>      /* For TIOCOUTQ/INQ */
30 #include <linux/list.h>
31 #include <linux/slab.h>
32 #include <net/datalink.h>
33 #include <net/psnap.h>
34 #include <net/sock.h>
35 #include <net/tcp_states.h>
36 #include <net/route.h>
37
38 #include <net/af_ieee802154.h>
39 #include <net/ieee802154_netdev.h>
40
41 #include "af802154.h"
42
43 /*
44  * Utility function for families
45  */
46 struct net_device *ieee802154_get_dev(struct net *net,
47                 struct ieee802154_addr *addr)
48 {
49         struct net_device *dev = NULL;
50         struct net_device *tmp;
51         u16 pan_id, short_addr;
52
53         switch (addr->addr_type) {
54         case IEEE802154_ADDR_LONG:
55                 rtnl_lock();
56                 dev = dev_getbyhwaddr(net, ARPHRD_IEEE802154, addr->hwaddr);
57                 if (dev)
58                         dev_hold(dev);
59                 rtnl_unlock();
60                 break;
61         case IEEE802154_ADDR_SHORT:
62                 if (addr->pan_id == 0xffff ||
63                     addr->short_addr == IEEE802154_ADDR_UNDEF ||
64                     addr->short_addr == 0xffff)
65                         break;
66
67                 rtnl_lock();
68
69                 for_each_netdev(net, tmp) {
70                         if (tmp->type != ARPHRD_IEEE802154)
71                                 continue;
72
73                         pan_id = ieee802154_mlme_ops(tmp)->get_pan_id(tmp);
74                         short_addr =
75                                 ieee802154_mlme_ops(tmp)->get_short_addr(tmp);
76
77                         if (pan_id == addr->pan_id &&
78                             short_addr == addr->short_addr) {
79                                 dev = tmp;
80                                 dev_hold(dev);
81                                 break;
82                         }
83                 }
84
85                 rtnl_unlock();
86                 break;
87         default:
88                 pr_warning("Unsupported ieee802154 address type: %d\n",
89                                 addr->addr_type);
90                 break;
91         }
92
93         return dev;
94 }
95
96 static int ieee802154_sock_release(struct socket *sock)
97 {
98         struct sock *sk = sock->sk;
99
100         if (sk) {
101                 sock->sk = NULL;
102                 sk->sk_prot->close(sk, 0);
103         }
104         return 0;
105 }
106 static int ieee802154_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
107                 struct msghdr *msg, size_t len)
108 {
109         struct sock *sk = sock->sk;
110
111         return sk->sk_prot->sendmsg(iocb, sk, msg, len);
112 }
113
114 static int ieee802154_sock_bind(struct socket *sock, struct sockaddr *uaddr,
115                 int addr_len)
116 {
117         struct sock *sk = sock->sk;
118
119         if (sk->sk_prot->bind)
120                 return sk->sk_prot->bind(sk, uaddr, addr_len);
121
122         return sock_no_bind(sock, uaddr, addr_len);
123 }
124
125 static int ieee802154_sock_connect(struct socket *sock, struct sockaddr *uaddr,
126                         int addr_len, int flags)
127 {
128         struct sock *sk = sock->sk;
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->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 static const struct proto_ops ieee802154_raw_ops = {
183         .family            = PF_IEEE802154,
184         .owner             = THIS_MODULE,
185         .release           = ieee802154_sock_release,
186         .bind              = ieee802154_sock_bind,
187         .connect           = ieee802154_sock_connect,
188         .socketpair        = sock_no_socketpair,
189         .accept            = sock_no_accept,
190         .getname           = sock_no_getname,
191         .poll              = datagram_poll,
192         .ioctl             = ieee802154_sock_ioctl,
193         .listen            = sock_no_listen,
194         .shutdown          = sock_no_shutdown,
195         .setsockopt        = sock_common_setsockopt,
196         .getsockopt        = sock_common_getsockopt,
197         .sendmsg           = ieee802154_sock_sendmsg,
198         .recvmsg           = sock_common_recvmsg,
199         .mmap              = sock_no_mmap,
200         .sendpage          = sock_no_sendpage,
201 #ifdef CONFIG_COMPAT
202         .compat_setsockopt = compat_sock_common_setsockopt,
203         .compat_getsockopt = compat_sock_common_getsockopt,
204 #endif
205 };
206
207 static const struct proto_ops ieee802154_dgram_ops = {
208         .family            = PF_IEEE802154,
209         .owner             = THIS_MODULE,
210         .release           = ieee802154_sock_release,
211         .bind              = ieee802154_sock_bind,
212         .connect           = ieee802154_sock_connect,
213         .socketpair        = sock_no_socketpair,
214         .accept            = sock_no_accept,
215         .getname           = sock_no_getname,
216         .poll              = datagram_poll,
217         .ioctl             = ieee802154_sock_ioctl,
218         .listen            = sock_no_listen,
219         .shutdown          = sock_no_shutdown,
220         .setsockopt        = sock_common_setsockopt,
221         .getsockopt        = sock_common_getsockopt,
222         .sendmsg           = ieee802154_sock_sendmsg,
223         .recvmsg           = sock_common_recvmsg,
224         .mmap              = sock_no_mmap,
225         .sendpage          = sock_no_sendpage,
226 #ifdef CONFIG_COMPAT
227         .compat_setsockopt = compat_sock_common_setsockopt,
228         .compat_getsockopt = compat_sock_common_getsockopt,
229 #endif
230 };
231
232
233 /*
234  * Create a socket. Initialise the socket, blank the addresses
235  * set the state.
236  */
237 static int ieee802154_create(struct net *net, struct socket *sock,
238                              int protocol, int kern)
239 {
240         struct sock *sk;
241         int rc;
242         struct proto *proto;
243         const struct proto_ops *ops;
244
245         if (!net_eq(net, &init_net))
246                 return -EAFNOSUPPORT;
247
248         switch (sock->type) {
249         case SOCK_RAW:
250                 proto = &ieee802154_raw_prot;
251                 ops = &ieee802154_raw_ops;
252                 break;
253         case SOCK_DGRAM:
254                 proto = &ieee802154_dgram_prot;
255                 ops = &ieee802154_dgram_ops;
256                 break;
257         default:
258                 rc = -ESOCKTNOSUPPORT;
259                 goto out;
260         }
261
262         rc = -ENOMEM;
263         sk = sk_alloc(net, PF_IEEE802154, GFP_KERNEL, proto);
264         if (!sk)
265                 goto out;
266         rc = 0;
267
268         sock->ops = ops;
269
270         sock_init_data(sock, sk);
271         /* FIXME: sk->sk_destruct */
272         sk->sk_family = PF_IEEE802154;
273
274         /* Checksums on by default */
275         sock_set_flag(sk, SOCK_ZAPPED);
276
277         if (sk->sk_prot->hash)
278                 sk->sk_prot->hash(sk);
279
280         if (sk->sk_prot->init) {
281                 rc = sk->sk_prot->init(sk);
282                 if (rc)
283                         sk_common_release(sk);
284         }
285 out:
286         return rc;
287 }
288
289 static const struct net_proto_family ieee802154_family_ops = {
290         .family         = PF_IEEE802154,
291         .create         = ieee802154_create,
292         .owner          = THIS_MODULE,
293 };
294
295 static int ieee802154_rcv(struct sk_buff *skb, struct net_device *dev,
296         struct packet_type *pt, struct net_device *orig_dev)
297 {
298         if (!netif_running(dev))
299                 return -ENODEV;
300         pr_debug("got frame, type %d, dev %p\n", dev->type, dev);
301 #ifdef DEBUG
302         print_hex_dump_bytes("ieee802154_rcv ", DUMP_PREFIX_NONE, skb->data, skb->len);
303 #endif
304
305         if (!net_eq(dev_net(dev), &init_net))
306                 goto drop;
307
308         ieee802154_raw_deliver(dev, skb);
309
310         if (dev->type != ARPHRD_IEEE802154)
311                 goto drop;
312
313         if (skb->pkt_type != PACKET_OTHERHOST)
314                 return ieee802154_dgram_deliver(dev, skb);
315
316 drop:
317         kfree_skb(skb);
318         return NET_RX_DROP;
319 }
320
321
322 static struct packet_type ieee802154_packet_type = {
323         .type = __constant_htons(ETH_P_IEEE802154),
324         .func = ieee802154_rcv,
325 };
326
327 static int __init af_ieee802154_init(void)
328 {
329         int rc = -EINVAL;
330
331         rc = proto_register(&ieee802154_raw_prot, 1);
332         if (rc)
333                 goto out;
334
335         rc = proto_register(&ieee802154_dgram_prot, 1);
336         if (rc)
337                 goto err_dgram;
338
339         /* Tell SOCKET that we are alive */
340         rc = sock_register(&ieee802154_family_ops);
341         if (rc)
342                 goto err_sock;
343         dev_add_pack(&ieee802154_packet_type);
344
345         rc = 0;
346         goto out;
347
348 err_sock:
349         proto_unregister(&ieee802154_dgram_prot);
350 err_dgram:
351         proto_unregister(&ieee802154_raw_prot);
352 out:
353         return rc;
354 }
355 static void __exit af_ieee802154_remove(void)
356 {
357         dev_remove_pack(&ieee802154_packet_type);
358         sock_unregister(PF_IEEE802154);
359         proto_unregister(&ieee802154_dgram_prot);
360         proto_unregister(&ieee802154_raw_prot);
361 }
362
363 module_init(af_ieee802154_init);
364 module_exit(af_ieee802154_remove);
365
366 MODULE_LICENSE("GPL");
367 MODULE_ALIAS_NETPROTO(PF_IEEE802154);