Bluetooth: hidp: fix buffer overflow
[pandora-kernel.git] / net / bluetooth / hidp / sock.c
1 /*
2    HIDP implementation for Linux Bluetooth stack (BlueZ).
3    Copyright (C) 2003-2004 Marcel Holtmann <marcel@holtmann.org>
4
5    This program is free software; you can redistribute it and/or modify
6    it under the terms of the GNU General Public License version 2 as
7    published by the Free Software Foundation;
8
9    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
10    OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
11    FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT OF THIRD PARTY RIGHTS.
12    IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) AND AUTHOR(S) BE LIABLE FOR ANY
13    CLAIM, OR ANY SPECIAL INDIRECT OR CONSEQUENTIAL DAMAGES, OR ANY DAMAGES
14    WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15    ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17
18    ALL LIABILITY, INCLUDING LIABILITY FOR INFRINGEMENT OF ANY PATENTS,
19    COPYRIGHTS, TRADEMARKS OR OTHER RIGHTS, RELATING TO USE OF THIS
20    SOFTWARE IS DISCLAIMED.
21 */
22
23 #include <linux/module.h>
24
25 #include <linux/types.h>
26 #include <linux/capability.h>
27 #include <linux/errno.h>
28 #include <linux/kernel.h>
29 #include <linux/poll.h>
30 #include <linux/fcntl.h>
31 #include <linux/skbuff.h>
32 #include <linux/socket.h>
33 #include <linux/ioctl.h>
34 #include <linux/file.h>
35 #include <linux/init.h>
36 #include <linux/compat.h>
37 #include <linux/gfp.h>
38 #include <net/sock.h>
39
40 #include "hidp.h"
41
42 static int hidp_sock_release(struct socket *sock)
43 {
44         struct sock *sk = sock->sk;
45
46         BT_DBG("sock %p sk %p", sock, sk);
47
48         if (!sk)
49                 return 0;
50
51         sock_orphan(sk);
52         sock_put(sk);
53
54         return 0;
55 }
56
57 static int hidp_sock_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
58 {
59         void __user *argp = (void __user *) arg;
60         struct hidp_connadd_req ca;
61         struct hidp_conndel_req cd;
62         struct hidp_connlist_req cl;
63         struct hidp_conninfo ci;
64         struct socket *csock;
65         struct socket *isock;
66         int err;
67
68         BT_DBG("cmd %x arg %lx", cmd, arg);
69
70         switch (cmd) {
71         case HIDPCONNADD:
72                 if (!capable(CAP_NET_ADMIN))
73                         return -EACCES;
74
75                 if (copy_from_user(&ca, argp, sizeof(ca)))
76                         return -EFAULT;
77
78                 csock = sockfd_lookup(ca.ctrl_sock, &err);
79                 if (!csock)
80                         return err;
81
82                 isock = sockfd_lookup(ca.intr_sock, &err);
83                 if (!isock) {
84                         sockfd_put(csock);
85                         return err;
86                 }
87                 ca.name[sizeof(ca.name)-1] = 0;
88
89                 if (csock->sk->sk_state != BT_CONNECTED ||
90                                 isock->sk->sk_state != BT_CONNECTED) {
91                         sockfd_put(csock);
92                         sockfd_put(isock);
93                         return -EBADFD;
94                 }
95
96                 err = hidp_add_connection(&ca, csock, isock);
97                 if (!err) {
98                         if (copy_to_user(argp, &ca, sizeof(ca)))
99                                 err = -EFAULT;
100                 } else {
101                         sockfd_put(csock);
102                         sockfd_put(isock);
103                 }
104
105                 return err;
106
107         case HIDPCONNDEL:
108                 if (!capable(CAP_NET_ADMIN))
109                         return -EACCES;
110
111                 if (copy_from_user(&cd, argp, sizeof(cd)))
112                         return -EFAULT;
113
114                 return hidp_del_connection(&cd);
115
116         case HIDPGETCONNLIST:
117                 if (copy_from_user(&cl, argp, sizeof(cl)))
118                         return -EFAULT;
119
120                 if (cl.cnum <= 0)
121                         return -EINVAL;
122
123                 err = hidp_get_connlist(&cl);
124                 if (!err && copy_to_user(argp, &cl, sizeof(cl)))
125                         return -EFAULT;
126
127                 return err;
128
129         case HIDPGETCONNINFO:
130                 if (copy_from_user(&ci, argp, sizeof(ci)))
131                         return -EFAULT;
132
133                 err = hidp_get_conninfo(&ci);
134                 if (!err && copy_to_user(argp, &ci, sizeof(ci)))
135                         return -EFAULT;
136
137                 return err;
138         }
139
140         return -EINVAL;
141 }
142
143 #ifdef CONFIG_COMPAT
144 struct compat_hidp_connadd_req {
145         int   ctrl_sock;        /* Connected control socket */
146         int   intr_sock;        /* Connected interrupt socket */
147         __u16 parser;
148         __u16 rd_size;
149         compat_uptr_t rd_data;
150         __u8  country;
151         __u8  subclass;
152         __u16 vendor;
153         __u16 product;
154         __u16 version;
155         __u32 flags;
156         __u32 idle_to;
157         char  name[128];
158 };
159
160 static int hidp_sock_compat_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
161 {
162         if (cmd == HIDPGETCONNLIST) {
163                 struct hidp_connlist_req cl;
164                 uint32_t uci;
165                 int err;
166
167                 if (get_user(cl.cnum, (uint32_t __user *) arg) ||
168                                 get_user(uci, (u32 __user *) (arg + 4)))
169                         return -EFAULT;
170
171                 cl.ci = compat_ptr(uci);
172
173                 if (cl.cnum <= 0)
174                         return -EINVAL;
175
176                 err = hidp_get_connlist(&cl);
177
178                 if (!err && put_user(cl.cnum, (uint32_t __user *) arg))
179                         err = -EFAULT;
180
181                 return err;
182         } else if (cmd == HIDPCONNADD) {
183                 struct compat_hidp_connadd_req ca;
184                 struct hidp_connadd_req __user *uca;
185
186                 uca = compat_alloc_user_space(sizeof(*uca));
187
188                 if (copy_from_user(&ca, (void __user *) arg, sizeof(ca)))
189                         return -EFAULT;
190
191                 if (put_user(ca.ctrl_sock, &uca->ctrl_sock) ||
192                                 put_user(ca.intr_sock, &uca->intr_sock) ||
193                                 put_user(ca.parser, &uca->parser) ||
194                                 put_user(ca.rd_size, &uca->rd_size) ||
195                                 put_user(compat_ptr(ca.rd_data), &uca->rd_data) ||
196                                 put_user(ca.country, &uca->country) ||
197                                 put_user(ca.subclass, &uca->subclass) ||
198                                 put_user(ca.vendor, &uca->vendor) ||
199                                 put_user(ca.product, &uca->product) ||
200                                 put_user(ca.version, &uca->version) ||
201                                 put_user(ca.flags, &uca->flags) ||
202                                 put_user(ca.idle_to, &uca->idle_to) ||
203                                 copy_to_user(&uca->name[0], &ca.name[0], 128))
204                         return -EFAULT;
205
206                 arg = (unsigned long) uca;
207
208                 /* Fall through. We don't actually write back any _changes_
209                    to the structure anyway, so there's no need to copy back
210                    into the original compat version */
211         }
212
213         return hidp_sock_ioctl(sock, cmd, arg);
214 }
215 #endif
216
217 static const struct proto_ops hidp_sock_ops = {
218         .family         = PF_BLUETOOTH,
219         .owner          = THIS_MODULE,
220         .release        = hidp_sock_release,
221         .ioctl          = hidp_sock_ioctl,
222 #ifdef CONFIG_COMPAT
223         .compat_ioctl   = hidp_sock_compat_ioctl,
224 #endif
225         .bind           = sock_no_bind,
226         .getname        = sock_no_getname,
227         .sendmsg        = sock_no_sendmsg,
228         .recvmsg        = sock_no_recvmsg,
229         .poll           = sock_no_poll,
230         .listen         = sock_no_listen,
231         .shutdown       = sock_no_shutdown,
232         .setsockopt     = sock_no_setsockopt,
233         .getsockopt     = sock_no_getsockopt,
234         .connect        = sock_no_connect,
235         .socketpair     = sock_no_socketpair,
236         .accept         = sock_no_accept,
237         .mmap           = sock_no_mmap
238 };
239
240 static struct proto hidp_proto = {
241         .name           = "HIDP",
242         .owner          = THIS_MODULE,
243         .obj_size       = sizeof(struct bt_sock)
244 };
245
246 static int hidp_sock_create(struct net *net, struct socket *sock, int protocol,
247                             int kern)
248 {
249         struct sock *sk;
250
251         BT_DBG("sock %p", sock);
252
253         if (sock->type != SOCK_RAW)
254                 return -ESOCKTNOSUPPORT;
255
256         sk = sk_alloc(net, PF_BLUETOOTH, GFP_ATOMIC, &hidp_proto);
257         if (!sk)
258                 return -ENOMEM;
259
260         sock_init_data(sock, sk);
261
262         sock->ops = &hidp_sock_ops;
263
264         sock->state = SS_UNCONNECTED;
265
266         sock_reset_flag(sk, SOCK_ZAPPED);
267
268         sk->sk_protocol = protocol;
269         sk->sk_state    = BT_OPEN;
270
271         return 0;
272 }
273
274 static const struct net_proto_family hidp_sock_family_ops = {
275         .family = PF_BLUETOOTH,
276         .owner  = THIS_MODULE,
277         .create = hidp_sock_create
278 };
279
280 int __init hidp_init_sockets(void)
281 {
282         int err;
283
284         err = proto_register(&hidp_proto, 0);
285         if (err < 0)
286                 return err;
287
288         err = bt_sock_register(BTPROTO_HIDP, &hidp_sock_family_ops);
289         if (err < 0)
290                 goto error;
291
292         return 0;
293
294 error:
295         BT_ERR("Can't register HIDP socket");
296         proto_unregister(&hidp_proto);
297         return err;
298 }
299
300 void __exit hidp_cleanup_sockets(void)
301 {
302         if (bt_sock_unregister(BTPROTO_HIDP) < 0)
303                 BT_ERR("Can't unregister HIDP socket");
304
305         proto_unregister(&hidp_proto);
306 }