net: check the length of the socket address passed to connect(2)
[linux-2.6/linux-acpi-2.6/ibm-acpi-2.6.git] / net / ieee802154 / af_ieee802154.c
blob01beb6c112057692bfc981a781c948cc5accb594
1 /*
2 * IEEE802154.4 socket interface
4 * Copyright 2007, 2008 Siemens AG
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.
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.
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.
19 * Written by:
20 * Sergey Lapin <slapin@ossfans.org>
21 * Maxim Gorbachyov <maxim.gorbachev@siemens.com>
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 <net/datalink.h>
32 #include <net/psnap.h>
33 #include <net/sock.h>
34 #include <net/tcp_states.h>
35 #include <net/route.h>
37 #include <net/af_ieee802154.h>
38 #include <net/ieee802154_netdev.h>
40 #include "af802154.h"
43 * Utility function for families
45 struct net_device *ieee802154_get_dev(struct net *net,
46 struct ieee802154_addr *addr)
48 struct net_device *dev = NULL;
49 struct net_device *tmp;
50 u16 pan_id, short_addr;
52 switch (addr->addr_type) {
53 case IEEE802154_ADDR_LONG:
54 rtnl_lock();
55 dev = dev_getbyhwaddr(net, ARPHRD_IEEE802154, addr->hwaddr);
56 if (dev)
57 dev_hold(dev);
58 rtnl_unlock();
59 break;
60 case IEEE802154_ADDR_SHORT:
61 if (addr->pan_id == 0xffff ||
62 addr->short_addr == IEEE802154_ADDR_UNDEF ||
63 addr->short_addr == 0xffff)
64 break;
66 rtnl_lock();
68 for_each_netdev(net, tmp) {
69 if (tmp->type != ARPHRD_IEEE802154)
70 continue;
72 pan_id = ieee802154_mlme_ops(tmp)->get_pan_id(tmp);
73 short_addr =
74 ieee802154_mlme_ops(tmp)->get_short_addr(tmp);
76 if (pan_id == addr->pan_id &&
77 short_addr == addr->short_addr) {
78 dev = tmp;
79 dev_hold(dev);
80 break;
84 rtnl_unlock();
85 break;
86 default:
87 pr_warning("Unsupported ieee802154 address type: %d\n",
88 addr->addr_type);
89 break;
92 return dev;
95 static int ieee802154_sock_release(struct socket *sock)
97 struct sock *sk = sock->sk;
99 if (sk) {
100 sock->sk = NULL;
101 sk->sk_prot->close(sk, 0);
103 return 0;
105 static int ieee802154_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
106 struct msghdr *msg, size_t len)
108 struct sock *sk = sock->sk;
110 return sk->sk_prot->sendmsg(iocb, sk, msg, len);
113 static int ieee802154_sock_bind(struct socket *sock, struct sockaddr *uaddr,
114 int addr_len)
116 struct sock *sk = sock->sk;
118 if (sk->sk_prot->bind)
119 return sk->sk_prot->bind(sk, uaddr, addr_len);
121 return sock_no_bind(sock, uaddr, addr_len);
124 static int ieee802154_sock_connect(struct socket *sock, struct sockaddr *uaddr,
125 int addr_len, int flags)
127 struct sock *sk = sock->sk;
129 if (addr_len < sizeof(uaddr->sa_family))
130 return -EINVAL;
132 if (uaddr->sa_family == AF_UNSPEC)
133 return sk->sk_prot->disconnect(sk, flags);
135 return sk->sk_prot->connect(sk, uaddr, addr_len);
138 static int ieee802154_dev_ioctl(struct sock *sk, struct ifreq __user *arg,
139 unsigned int cmd)
141 struct ifreq ifr;
142 int ret = -ENOIOCTLCMD;
143 struct net_device *dev;
145 if (copy_from_user(&ifr, arg, sizeof(struct ifreq)))
146 return -EFAULT;
148 ifr.ifr_name[IFNAMSIZ-1] = 0;
150 dev_load(sock_net(sk), ifr.ifr_name);
151 dev = dev_get_by_name(sock_net(sk), ifr.ifr_name);
153 if (dev->type == ARPHRD_IEEE802154 && dev->netdev_ops->ndo_do_ioctl)
154 ret = dev->netdev_ops->ndo_do_ioctl(dev, &ifr, cmd);
156 if (!ret && copy_to_user(arg, &ifr, sizeof(struct ifreq)))
157 ret = -EFAULT;
158 dev_put(dev);
160 return ret;
163 static int ieee802154_sock_ioctl(struct socket *sock, unsigned int cmd,
164 unsigned long arg)
166 struct sock *sk = sock->sk;
168 switch (cmd) {
169 case SIOCGSTAMP:
170 return sock_get_timestamp(sk, (struct timeval __user *)arg);
171 case SIOCGSTAMPNS:
172 return sock_get_timestampns(sk, (struct timespec __user *)arg);
173 case SIOCGIFADDR:
174 case SIOCSIFADDR:
175 return ieee802154_dev_ioctl(sk, (struct ifreq __user *)arg,
176 cmd);
177 default:
178 if (!sk->sk_prot->ioctl)
179 return -ENOIOCTLCMD;
180 return sk->sk_prot->ioctl(sk, cmd, arg);
184 static const struct proto_ops ieee802154_raw_ops = {
185 .family = PF_IEEE802154,
186 .owner = THIS_MODULE,
187 .release = ieee802154_sock_release,
188 .bind = ieee802154_sock_bind,
189 .connect = ieee802154_sock_connect,
190 .socketpair = sock_no_socketpair,
191 .accept = sock_no_accept,
192 .getname = sock_no_getname,
193 .poll = datagram_poll,
194 .ioctl = ieee802154_sock_ioctl,
195 .listen = sock_no_listen,
196 .shutdown = sock_no_shutdown,
197 .setsockopt = sock_common_setsockopt,
198 .getsockopt = sock_common_getsockopt,
199 .sendmsg = ieee802154_sock_sendmsg,
200 .recvmsg = sock_common_recvmsg,
201 .mmap = sock_no_mmap,
202 .sendpage = sock_no_sendpage,
203 #ifdef CONFIG_COMPAT
204 .compat_setsockopt = compat_sock_common_setsockopt,
205 .compat_getsockopt = compat_sock_common_getsockopt,
206 #endif
209 static const struct proto_ops ieee802154_dgram_ops = {
210 .family = PF_IEEE802154,
211 .owner = THIS_MODULE,
212 .release = ieee802154_sock_release,
213 .bind = ieee802154_sock_bind,
214 .connect = ieee802154_sock_connect,
215 .socketpair = sock_no_socketpair,
216 .accept = sock_no_accept,
217 .getname = sock_no_getname,
218 .poll = datagram_poll,
219 .ioctl = ieee802154_sock_ioctl,
220 .listen = sock_no_listen,
221 .shutdown = sock_no_shutdown,
222 .setsockopt = sock_common_setsockopt,
223 .getsockopt = sock_common_getsockopt,
224 .sendmsg = ieee802154_sock_sendmsg,
225 .recvmsg = sock_common_recvmsg,
226 .mmap = sock_no_mmap,
227 .sendpage = sock_no_sendpage,
228 #ifdef CONFIG_COMPAT
229 .compat_setsockopt = compat_sock_common_setsockopt,
230 .compat_getsockopt = compat_sock_common_getsockopt,
231 #endif
236 * Create a socket. Initialise the socket, blank the addresses
237 * set the state.
239 static int ieee802154_create(struct net *net, struct socket *sock,
240 int protocol, int kern)
242 struct sock *sk;
243 int rc;
244 struct proto *proto;
245 const struct proto_ops *ops;
247 if (!net_eq(net, &init_net))
248 return -EAFNOSUPPORT;
250 switch (sock->type) {
251 case SOCK_RAW:
252 proto = &ieee802154_raw_prot;
253 ops = &ieee802154_raw_ops;
254 break;
255 case SOCK_DGRAM:
256 proto = &ieee802154_dgram_prot;
257 ops = &ieee802154_dgram_ops;
258 break;
259 default:
260 rc = -ESOCKTNOSUPPORT;
261 goto out;
264 rc = -ENOMEM;
265 sk = sk_alloc(net, PF_IEEE802154, GFP_KERNEL, proto);
266 if (!sk)
267 goto out;
268 rc = 0;
270 sock->ops = ops;
272 sock_init_data(sock, sk);
273 /* FIXME: sk->sk_destruct */
274 sk->sk_family = PF_IEEE802154;
276 /* Checksums on by default */
277 sock_set_flag(sk, SOCK_ZAPPED);
279 if (sk->sk_prot->hash)
280 sk->sk_prot->hash(sk);
282 if (sk->sk_prot->init) {
283 rc = sk->sk_prot->init(sk);
284 if (rc)
285 sk_common_release(sk);
287 out:
288 return rc;
291 static const struct net_proto_family ieee802154_family_ops = {
292 .family = PF_IEEE802154,
293 .create = ieee802154_create,
294 .owner = THIS_MODULE,
297 static int ieee802154_rcv(struct sk_buff *skb, struct net_device *dev,
298 struct packet_type *pt, struct net_device *orig_dev)
300 if (!netif_running(dev))
301 return -ENODEV;
302 pr_debug("got frame, type %d, dev %p\n", dev->type, dev);
303 #ifdef DEBUG
304 print_hex_dump_bytes("ieee802154_rcv ", DUMP_PREFIX_NONE, skb->data, skb->len);
305 #endif
307 if (!net_eq(dev_net(dev), &init_net))
308 goto drop;
310 ieee802154_raw_deliver(dev, skb);
312 if (dev->type != ARPHRD_IEEE802154)
313 goto drop;
315 if (skb->pkt_type != PACKET_OTHERHOST)
316 return ieee802154_dgram_deliver(dev, skb);
318 drop:
319 kfree_skb(skb);
320 return NET_RX_DROP;
324 static struct packet_type ieee802154_packet_type = {
325 .type = __constant_htons(ETH_P_IEEE802154),
326 .func = ieee802154_rcv,
329 static int __init af_ieee802154_init(void)
331 int rc = -EINVAL;
333 rc = proto_register(&ieee802154_raw_prot, 1);
334 if (rc)
335 goto out;
337 rc = proto_register(&ieee802154_dgram_prot, 1);
338 if (rc)
339 goto err_dgram;
341 /* Tell SOCKET that we are alive */
342 rc = sock_register(&ieee802154_family_ops);
343 if (rc)
344 goto err_sock;
345 dev_add_pack(&ieee802154_packet_type);
347 rc = 0;
348 goto out;
350 err_sock:
351 proto_unregister(&ieee802154_dgram_prot);
352 err_dgram:
353 proto_unregister(&ieee802154_raw_prot);
354 out:
355 return rc;
357 static void __exit af_ieee802154_remove(void)
359 dev_remove_pack(&ieee802154_packet_type);
360 sock_unregister(PF_IEEE802154);
361 proto_unregister(&ieee802154_dgram_prot);
362 proto_unregister(&ieee802154_raw_prot);
365 module_init(af_ieee802154_init);
366 module_exit(af_ieee802154_remove);
368 MODULE_LICENSE("GPL");
369 MODULE_ALIAS_NETPROTO(PF_IEEE802154);