net/bluetooth/af_bluetooth.c

   1 /*
   2    BlueZ - Bluetooth protocol stack for Linux
   3    Copyright (C) 2000-2001 Qualcomm Incorporated
   4
   5    Written 2000,2001 by Maxim Krasnyansky <maxk@qualcomm.com>
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License version 2 as
   9    published by the Free Software Foundation;
  10
  11    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  12    OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  13    FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT OF THIRD PARTY RIGHTS.
  14    IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) AND AUTHOR(S) BE LIABLE FOR ANY
  15    CLAIM, OR ANY SPECIAL INDIRECT OR CONSEQUENTIAL DAMAGES, OR ANY DAMAGES
  16    WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  17    ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  18    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  19
  20    ALL LIABILITY, INCLUDING LIABILITY FOR INFRINGEMENT OF ANY PATENTS,
  21    COPYRIGHTS, TRADEMARKS OR OTHER RIGHTS, RELATING TO USE OF THIS
  22    SOFTWARE IS DISCLAIMED.
  23 */
  24
  25 /* Bluetooth address family and sockets. */
  26
  27 #include <linux/module.h>
  28
  29 #include <linux/types.h>
  30 #include <linux/list.h>
  31 #include <linux/errno.h>
  32 #include <linux/kernel.h>
  33 #include <linux/sched.h>
  34 #include <linux/skbuff.h>
  35 #include <linux/init.h>
  36 #include <linux/poll.h>
  37 #include <net/sock.h>
  38 #include <asm/ioctls.h>
  39 #include <linux/kmod.h>
  40
  41 #include <net/bluetooth/bluetooth.h>
  42
  43 #define VERSION "2.16"
  44
  45 /* Bluetooth sockets */
  46 #define BT_MAX_PROTO    8
  47 static const struct net_proto_family *bt_proto[BT_MAX_PROTO];
  48 static DEFINE_RWLOCK(bt_proto_lock);
  49
  50 static struct lock_class_key bt_lock_key[BT_MAX_PROTO];
  51 static const char *const bt_key_strings[BT_MAX_PROTO] = {
  52         "sk_lock-AF_BLUETOOTH-BTPROTO_L2CAP",
  53         "sk_lock-AF_BLUETOOTH-BTPROTO_HCI",
  54         "sk_lock-AF_BLUETOOTH-BTPROTO_SCO",
  55         "sk_lock-AF_BLUETOOTH-BTPROTO_RFCOMM",
  56         "sk_lock-AF_BLUETOOTH-BTPROTO_BNEP",
  57         "sk_lock-AF_BLUETOOTH-BTPROTO_CMTP",
  58         "sk_lock-AF_BLUETOOTH-BTPROTO_HIDP",
  59         "sk_lock-AF_BLUETOOTH-BTPROTO_AVDTP",
  60 };
  61
  62 static struct lock_class_key bt_slock_key[BT_MAX_PROTO];
  63 static const char *const bt_slock_key_strings[BT_MAX_PROTO] = {
  64         "slock-AF_BLUETOOTH-BTPROTO_L2CAP",
  65         "slock-AF_BLUETOOTH-BTPROTO_HCI",
  66         "slock-AF_BLUETOOTH-BTPROTO_SCO",
  67         "slock-AF_BLUETOOTH-BTPROTO_RFCOMM",
  68         "slock-AF_BLUETOOTH-BTPROTO_BNEP",
  69         "slock-AF_BLUETOOTH-BTPROTO_CMTP",
  70         "slock-AF_BLUETOOTH-BTPROTO_HIDP",
  71         "slock-AF_BLUETOOTH-BTPROTO_AVDTP",
  72 };
  73
  74 static inline void bt_sock_reclassify_lock(struct socket *sock, int proto)
  75 {
  76         struct sock *sk = sock->sk;
  77
  78         if (!sk)
  79                 return;
  80
  81         BUG_ON(sock_owned_by_user(sk));
  82
  83         sock_lock_init_class_and_name(sk,
  84                         bt_slock_key_strings[proto], &bt_slock_key[proto],
  85                                 bt_key_strings[proto], &bt_lock_key[proto]);
  86 }
  87
  88 int bt_sock_register(int proto, const struct net_proto_family *ops)
  89 {
  90         int err = 0;
  91
  92         if (proto < 0 || proto >= BT_MAX_PROTO)
  93                 return -EINVAL;
  94
  95         write_lock(&bt_proto_lock);
  96
  97         if (bt_proto[proto])
  98                 err = -EEXIST;
  99         else
 100                 bt_proto[proto] = ops;
 101
 102         write_unlock(&bt_proto_lock);
 103
 104         return err;
 105 }
 106 EXPORT_SYMBOL(bt_sock_register);
 107
 108 int bt_sock_unregister(int proto)
 109 {
 110         int err = 0;
 111
 112         if (proto < 0 || proto >= BT_MAX_PROTO)
 113                 return -EINVAL;
 114
 115         write_lock(&bt_proto_lock);
 116
 117         if (!bt_proto[proto])
 118                 err = -ENOENT;
 119         else
 120                 bt_proto[proto] = NULL;
 121
 122         write_unlock(&bt_proto_lock);
 123
 124         return err;
 125 }
 126 EXPORT_SYMBOL(bt_sock_unregister);
 127
 128 static int bt_sock_create(struct net *net, struct socket *sock, int proto,
 129                           int kern)
 130 {
 131         int err;
 132
 133         if (net != &init_net)
 134                 return -EAFNOSUPPORT;
 135
 136         if (proto < 0 || proto >= BT_MAX_PROTO)
 137                 return -EINVAL;
 138
 139         if (!bt_proto[proto])
 140                 request_module("bt-proto-%d", proto);
 141
 142         err = -EPROTONOSUPPORT;
 143
 144         read_lock(&bt_proto_lock);
 145
 146         if (bt_proto[proto] && try_module_get(bt_proto[proto]->owner)) {
 147                 err = bt_proto[proto]->create(net, sock, proto, kern);
 148                 bt_sock_reclassify_lock(sock, proto);
 149                 module_put(bt_proto[proto]->owner);
 150         }
 151
 152         read_unlock(&bt_proto_lock);
 153
 154         return err;
 155 }
 156
 157 void bt_sock_link(struct bt_sock_list *l, struct sock *sk)
 158 {
 159         write_lock_bh(&l->lock);
 160         sk_add_node(sk, &l->head);
 161         write_unlock_bh(&l->lock);
 162 }
 163 EXPORT_SYMBOL(bt_sock_link);
 164
 165 void bt_sock_unlink(struct bt_sock_list *l, struct sock *sk)
 166 {
 167         write_lock_bh(&l->lock);
 168         sk_del_node_init(sk);
 169         write_unlock_bh(&l->lock);
 170 }
 171 EXPORT_SYMBOL(bt_sock_unlink);
 172
 173 void bt_accept_enqueue(struct sock *parent, struct sock *sk)
 174 {
 175         BT_DBG("parent %p, sk %p", parent, sk);
 176
 177         sock_hold(sk);
 178         list_add_tail(&bt_sk(sk)->accept_q, &bt_sk(parent)->accept_q);
 179         bt_sk(sk)->parent = parent;
 180         parent->sk_ack_backlog++;
 181 }
 182 EXPORT_SYMBOL(bt_accept_enqueue);
 183
 184 void bt_accept_unlink(struct sock *sk)
 185 {
 186         BT_DBG("sk %p state %d", sk, sk->sk_state);
 187
 188         list_del_init(&bt_sk(sk)->accept_q);
 189         bt_sk(sk)->parent->sk_ack_backlog--;
 190         bt_sk(sk)->parent = NULL;
 191         sock_put(sk);
 192 }
 193 EXPORT_SYMBOL(bt_accept_unlink);
 194
 195 struct sock *bt_accept_dequeue(struct sock *parent, struct socket *newsock)
 196 {
 197         struct list_head *p, *n;
 198         struct sock *sk;
 199
 200         BT_DBG("parent %p", parent);
 201
 202         local_bh_disable();
 203         list_for_each_safe(p, n, &bt_sk(parent)->accept_q) {
 204                 sk = (struct sock *) list_entry(p, struct bt_sock, accept_q);
 205
 206                 bh_lock_sock(sk);
 207
 208                 /* FIXME: Is this check still needed */
 209                 if (sk->sk_state == BT_CLOSED) {
 210                         bh_unlock_sock(sk);
 211                         bt_accept_unlink(sk);
 212                         continue;
 213                 }
 214
 215                 if (sk->sk_state == BT_CONNECTED || !newsock ||
 216                                                 bt_sk(parent)->defer_setup) {
 217                         bt_accept_unlink(sk);
 218                         if (newsock)
 219                                 sock_graft(sk, newsock);
 220
 221                         bh_unlock_sock(sk);
 222                         local_bh_enable();
 223                         return sk;
 224                 }
 225
 226                 bh_unlock_sock(sk);
 227         }
 228         local_bh_enable();
 229
 230         return NULL;
 231 }
 232 EXPORT_SYMBOL(bt_accept_dequeue);
 233
 234 int bt_sock_recvmsg(struct kiocb *iocb, struct socket *sock,
 235                                 struct msghdr *msg, size_t len, int flags)
 236 {
 237         int noblock = flags & MSG_DONTWAIT;
 238         struct sock *sk = sock->sk;
 239         struct sk_buff *skb;
 240         size_t copied;
 241         int err;
 242
 243         BT_DBG("sock %p sk %p len %zu", sock, sk, len);
 244
 245         if (flags & (MSG_OOB))
 246                 return -EOPNOTSUPP;
 247
 248         skb = skb_recv_datagram(sk, flags, noblock, &err);
 249         if (!skb) {
 250                 if (sk->sk_shutdown & RCV_SHUTDOWN)
 251                         return 0;
 252                 return err;
 253         }
 254
 255         msg->msg_namelen = 0;
 256
 257         copied = skb->len;
 258         if (len < copied) {
 259                 msg->msg_flags |= MSG_TRUNC;
 260                 copied = len;
 261         }
 262
 263         skb_reset_transport_header(skb);
 264         err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
 265         if (err == 0)
 266                 sock_recv_ts_and_drops(msg, sk, skb);
 267
 268         skb_free_datagram(sk, skb);
 269
 270         return err ? : copied;
 271 }
 272 EXPORT_SYMBOL(bt_sock_recvmsg);
 273
 274 static long bt_sock_data_wait(struct sock *sk, long timeo)
 275 {
 276         DECLARE_WAITQUEUE(wait, current);
 277
 278         add_wait_queue(sk_sleep(sk), &wait);
 279         for (;;) {
 280                 set_current_state(TASK_INTERRUPTIBLE);
 281
 282                 if (!skb_queue_empty(&sk->sk_receive_queue))
 283                         break;
 284
 285                 if (sk->sk_err || (sk->sk_shutdown & RCV_SHUTDOWN))
 286                         break;
 287
 288                 if (signal_pending(current) || !timeo)
 289                         break;
 290
 291                 set_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
 292                 release_sock(sk);
 293                 timeo = schedule_timeout(timeo);
 294                 lock_sock(sk);
 295                 clear_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
 296         }
 297
 298         __set_current_state(TASK_RUNNING);
 299         remove_wait_queue(sk_sleep(sk), &wait);
 300         return timeo;
 301 }
 302
 303 int bt_sock_stream_recvmsg(struct kiocb *iocb, struct socket *sock,
 304                                struct msghdr *msg, size_t size, int flags)
 305 {
 306         struct sock *sk = sock->sk;
 307         int err = 0;
 308         size_t target, copied = 0;
 309         long timeo;
 310
 311         if (flags & MSG_OOB)
 312                 return -EOPNOTSUPP;
 313
 314         msg->msg_namelen = 0;
 315
 316         BT_DBG("sk %p size %zu", sk, size);
 317
 318         lock_sock(sk);
 319
 320         target = sock_rcvlowat(sk, flags & MSG_WAITALL, size);
 321         timeo  = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
 322
 323         do {
 324                 struct sk_buff *skb;
 325                 int chunk;
 326
 327                 skb = skb_dequeue(&sk->sk_receive_queue);
 328                 if (!skb) {
 329                         if (copied >= target)
 330                                 break;
 331
 332                         err = sock_error(sk);
 333                         if (err)
 334                                 break;
 335                         if (sk->sk_shutdown & RCV_SHUTDOWN)
 336                                 break;
 337
 338                         err = -EAGAIN;
 339                         if (!timeo)
 340                                 break;
 341
 342                         timeo = bt_sock_data_wait(sk, timeo);
 343
 344                         if (signal_pending(current)) {
 345                                 err = sock_intr_errno(timeo);
 346                                 goto out;
 347                         }
 348                         continue;
 349                 }
 350
 351                 chunk = min_t(unsigned int, skb->len, size);
 352                 if (memcpy_toiovec(msg->msg_iov, skb->data, chunk)) {
 353                         skb_queue_head(&sk->sk_receive_queue, skb);
 354                         if (!copied)
 355                                 copied = -EFAULT;
 356                         break;
 357                 }
 358                 copied += chunk;
 359                 size   -= chunk;
 360
 361                 sock_recv_ts_and_drops(msg, sk, skb);
 362
 363                 if (!(flags & MSG_PEEK)) {
 364                         skb_pull(skb, chunk);
 365                         if (skb->len) {
 366                                 skb_queue_head(&sk->sk_receive_queue, skb);
 367                                 break;
 368                         }
 369                         kfree_skb(skb);
 370
 371                 } else {
 372                         /* put message back and return */
 373                         skb_queue_head(&sk->sk_receive_queue, skb);
 374                         break;
 375                 }
 376         } while (size);
 377
 378 out:
 379         release_sock(sk);
 380         return copied ? : err;
 381 }
 382 EXPORT_SYMBOL(bt_sock_stream_recvmsg);
 383
 384 static inline unsigned int bt_accept_poll(struct sock *parent)
 385 {
 386         struct list_head *p, *n;
 387         struct sock *sk;
 388
 389         list_for_each_safe(p, n, &bt_sk(parent)->accept_q) {
 390                 sk = (struct sock *) list_entry(p, struct bt_sock, accept_q);
 391                 if (sk->sk_state == BT_CONNECTED ||
 392                                         (bt_sk(parent)->defer_setup &&
 393                                                 sk->sk_state == BT_CONNECT2))
 394                         return POLLIN | POLLRDNORM;
 395         }
 396
 397         return 0;
 398 }
 399
 400 unsigned int bt_sock_poll(struct file *file, struct socket *sock, poll_table *wait)
 401 {
 402         struct sock *sk = sock->sk;
 403         unsigned int mask = 0;
 404
 405         BT_DBG("sock %p, sk %p", sock, sk);
 406
 407         poll_wait(file, sk_sleep(sk), wait);
 408
 409         if (sk->sk_state == BT_LISTEN)
 410                 return bt_accept_poll(sk);
 411
 412         if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
 413                 mask |= POLLERR;
 414
 415         if (sk->sk_shutdown & RCV_SHUTDOWN)
 416                 mask |= POLLRDHUP | POLLIN | POLLRDNORM;
 417
 418         if (sk->sk_shutdown == SHUTDOWN_MASK)
 419                 mask |= POLLHUP;
 420
 421         if (!skb_queue_empty(&sk->sk_receive_queue))
 422                 mask |= POLLIN | POLLRDNORM;
 423
 424         if (sk->sk_state == BT_CLOSED)
 425                 mask |= POLLHUP;
 426
 427         if (sk->sk_state == BT_CONNECT ||
 428                         sk->sk_state == BT_CONNECT2 ||
 429                         sk->sk_state == BT_CONFIG)
 430                 return mask;
 431
 432         if (sock_writeable(sk))
 433                 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
 434         else
 435                 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
 436
 437         return mask;
 438 }
 439 EXPORT_SYMBOL(bt_sock_poll);
 440
 441 int bt_sock_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
 442 {
 443         struct sock *sk = sock->sk;
 444         struct sk_buff *skb;
 445         long amount;
 446         int err;
 447
 448         BT_DBG("sk %p cmd %x arg %lx", sk, cmd, arg);
 449
 450         switch (cmd) {
 451         case TIOCOUTQ:
 452                 if (sk->sk_state == BT_LISTEN)
 453                         return -EINVAL;
 454
 455                 amount = sk->sk_sndbuf - sk_wmem_alloc_get(sk);
 456                 if (amount < 0)
 457                         amount = 0;
 458                 err = put_user(amount, (int __user *) arg);
 459                 break;
 460
 461         case TIOCINQ:
 462                 if (sk->sk_state == BT_LISTEN)
 463                         return -EINVAL;
 464
 465                 lock_sock(sk);
 466                 skb = skb_peek(&sk->sk_receive_queue);
 467                 amount = skb ? skb->len : 0;
 468                 release_sock(sk);
 469                 err = put_user(amount, (int __user *) arg);
 470                 break;
 471
 472         case SIOCGSTAMP:
 473                 err = sock_get_timestamp(sk, (struct timeval __user *) arg);
 474                 break;
 475
 476         case SIOCGSTAMPNS:
 477                 err = sock_get_timestampns(sk, (struct timespec __user *) arg);
 478                 break;
 479
 480         default:
 481                 err = -ENOIOCTLCMD;
 482                 break;
 483         }
 484
 485         return err;
 486 }
 487 EXPORT_SYMBOL(bt_sock_ioctl);
 488
 489 int bt_sock_wait_state(struct sock *sk, int state, unsigned long timeo)
 490 {
 491         DECLARE_WAITQUEUE(wait, current);
 492         int err = 0;
 493
 494         BT_DBG("sk %p", sk);
 495
 496         add_wait_queue(sk_sleep(sk), &wait);
 497         set_current_state(TASK_INTERRUPTIBLE);
 498         while (sk->sk_state != state) {
 499                 if (!timeo) {
 500                         err = -EINPROGRESS;
 501                         break;
 502                 }
 503
 504                 if (signal_pending(current)) {
 505                         err = sock_intr_errno(timeo);
 506                         break;
 507                 }
 508
 509                 release_sock(sk);
 510                 timeo = schedule_timeout(timeo);
 511                 lock_sock(sk);
 512                 set_current_state(TASK_INTERRUPTIBLE);
 513
 514                 err = sock_error(sk);
 515                 if (err)
 516                         break;
 517         }
 518         __set_current_state(TASK_RUNNING);
 519         remove_wait_queue(sk_sleep(sk), &wait);
 520         return err;
 521 }
 522 EXPORT_SYMBOL(bt_sock_wait_state);
 523
 524 static struct net_proto_family bt_sock_family_ops = {
 525         .owner  = THIS_MODULE,
 526         .family = PF_BLUETOOTH,
 527         .create = bt_sock_create,
 528 };
 529
 530 static int __init bt_init(void)
 531 {
 532         int err;
 533
 534         BT_INFO("Core ver %s", VERSION);
 535
 536         err = bt_sysfs_init();
 537         if (err < 0)
 538                 return err;
 539
 540         err = sock_register(&bt_sock_family_ops);
 541         if (err < 0) {
 542                 bt_sysfs_cleanup();
 543                 return err;
 544         }
 545
 546         BT_INFO("HCI device and connection manager initialized");
 547
 548         err = hci_sock_init();
 549         if (err < 0)
 550                 goto error;
 551
 552         err = l2cap_init();
 553         if (err < 0)
 554                 goto sock_err;
 555
 556         err = sco_init();
 557         if (err < 0) {
 558                 l2cap_exit();
 559                 goto sock_err;
 560         }
 561
 562         return 0;
 563
 564 sock_err:
 565         hci_sock_cleanup();
 566
 567 error:
 568         sock_unregister(PF_BLUETOOTH);
 569         bt_sysfs_cleanup();
 570
 571         return err;
 572 }
 573
 574 static void __exit bt_exit(void)
 575 {
 576
 577         sco_exit();
 578
 579         l2cap_exit();
 580
 581         hci_sock_cleanup();
 582
 583         sock_unregister(PF_BLUETOOTH);
 584
 585         bt_sysfs_cleanup();
 586 }
 587
 588 subsys_initcall(bt_init);
 589 module_exit(bt_exit);
 590
 591 MODULE_AUTHOR("Marcel Holtmann <marcel@holtmann.org>");
 592 MODULE_DESCRIPTION("Bluetooth Core ver " VERSION);
 593 MODULE_VERSION(VERSION);
 594 MODULE_LICENSE("GPL");
 595 MODULE_ALIAS_NETPROTO(PF_BLUETOOTH);