net/atm/common.c

   1 /* net/atm/common.c - ATM sockets (common part for PVC and SVC) */
   2
   3 /* Written 1995-2000 by Werner Almesberger, EPFL LRC/ICA */
   4
   5 #define pr_fmt(fmt) KBUILD_MODNAME ":%s: " fmt, __func__
   6
   7 #include <linux/module.h>
   8 #include <linux/kmod.h>
   9 #include <linux/net.h>          /* struct socket, struct proto_ops */
  10 #include <linux/atm.h>          /* ATM stuff */
  11 #include <linux/atmdev.h>
  12 #include <linux/socket.h>       /* SOL_SOCKET */
  13 #include <linux/errno.h>        /* error codes */
  14 #include <linux/capability.h>
  15 #include <linux/mm.h>
  16 #include <linux/sched.h>
  17 #include <linux/time.h>         /* struct timeval */
  18 #include <linux/skbuff.h>
  19 #include <linux/bitops.h>
  20 #include <linux/init.h>
  21 #include <linux/slab.h>
  22 #include <net/sock.h>           /* struct sock */
  23 #include <linux/uaccess.h>
  24 #include <linux/poll.h>
  25
  26 #include <linux/atomic.h>
  27
  28 #include "resources.h"          /* atm_find_dev */
  29 #include "common.h"             /* prototypes */
  30 #include "protocols.h"          /* atm_init_<transport> */
  31 #include "addr.h"               /* address registry */
  32 #include "signaling.h"          /* for WAITING and sigd_attach */
  33
  34 struct hlist_head vcc_hash[VCC_HTABLE_SIZE];
  35 EXPORT_SYMBOL(vcc_hash);
  36
  37 DEFINE_RWLOCK(vcc_sklist_lock);
  38 EXPORT_SYMBOL(vcc_sklist_lock);
  39
  40 static ATOMIC_NOTIFIER_HEAD(atm_dev_notify_chain);
  41
  42 static void __vcc_insert_socket(struct sock *sk)
  43 {
  44         struct atm_vcc *vcc = atm_sk(sk);
  45         struct hlist_head *head = &vcc_hash[vcc->vci & (VCC_HTABLE_SIZE - 1)];
  46         sk->sk_hash = vcc->vci & (VCC_HTABLE_SIZE - 1);
  47         sk_add_node(sk, head);
  48 }
  49
  50 void vcc_insert_socket(struct sock *sk)
  51 {
  52         write_lock_irq(&vcc_sklist_lock);
  53         __vcc_insert_socket(sk);
  54         write_unlock_irq(&vcc_sklist_lock);
  55 }
  56 EXPORT_SYMBOL(vcc_insert_socket);
  57
  58 static void vcc_remove_socket(struct sock *sk)
  59 {
  60         write_lock_irq(&vcc_sklist_lock);
  61         sk_del_node_init(sk);
  62         write_unlock_irq(&vcc_sklist_lock);
  63 }
  64
  65 static struct sk_buff *alloc_tx(struct atm_vcc *vcc, unsigned int size)
  66 {
  67         struct sk_buff *skb;
  68         struct sock *sk = sk_atm(vcc);
  69
  70         if (sk_wmem_alloc_get(sk) && !atm_may_send(vcc, size)) {
  71                 pr_debug("Sorry: wmem_alloc = %d, size = %d, sndbuf = %d\n",
  72                          sk_wmem_alloc_get(sk), size, sk->sk_sndbuf);
  73                 return NULL;
  74         }
  75         while (!(skb = alloc_skb(size, GFP_KERNEL)))
  76                 schedule();
  77         pr_debug("%d += %d\n", sk_wmem_alloc_get(sk), skb->truesize);
  78         atomic_add(skb->truesize, &sk->sk_wmem_alloc);
  79         return skb;
  80 }
  81
  82 static void vcc_sock_destruct(struct sock *sk)
  83 {
  84         if (atomic_read(&sk->sk_rmem_alloc))
  85                 printk(KERN_DEBUG "%s: rmem leakage (%d bytes) detected.\n",
  86                        __func__, atomic_read(&sk->sk_rmem_alloc));
  87
  88         if (atomic_read(&sk->sk_wmem_alloc))
  89                 printk(KERN_DEBUG "%s: wmem leakage (%d bytes) detected.\n",
  90                        __func__, atomic_read(&sk->sk_wmem_alloc));
  91 }
  92
  93 static void vcc_def_wakeup(struct sock *sk)
  94 {
  95         struct socket_wq *wq;
  96
  97         rcu_read_lock();
  98         wq = rcu_dereference(sk->sk_wq);
  99         if (wq_has_sleeper(wq))
 100                 wake_up(&wq->wait);
 101         rcu_read_unlock();
 102 }
 103
 104 static inline int vcc_writable(struct sock *sk)
 105 {
 106         struct atm_vcc *vcc = atm_sk(sk);
 107
 108         return (vcc->qos.txtp.max_sdu +
 109                 atomic_read(&sk->sk_wmem_alloc)) <= sk->sk_sndbuf;
 110 }
 111
 112 static void vcc_write_space(struct sock *sk)
 113 {
 114         struct socket_wq *wq;
 115
 116         rcu_read_lock();
 117
 118         if (vcc_writable(sk)) {
 119                 wq = rcu_dereference(sk->sk_wq);
 120                 if (wq_has_sleeper(wq))
 121                         wake_up_interruptible(&wq->wait);
 122
 123                 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
 124         }
 125
 126         rcu_read_unlock();
 127 }
 128
 129 static void vcc_release_cb(struct sock *sk)
 130 {
 131         struct atm_vcc *vcc = atm_sk(sk);
 132
 133         if (vcc->release_cb)
 134                 vcc->release_cb(vcc);
 135 }
 136
 137 static struct proto vcc_proto = {
 138         .name     = "VCC",
 139         .owner    = THIS_MODULE,
 140         .obj_size = sizeof(struct atm_vcc),
 141         .release_cb = vcc_release_cb,
 142 };
 143
 144 int vcc_create(struct net *net, struct socket *sock, int protocol, int family)
 145 {
 146         struct sock *sk;
 147         struct atm_vcc *vcc;
 148
 149         sock->sk = NULL;
 150         if (sock->type == SOCK_STREAM)
 151                 return -EINVAL;
 152         sk = sk_alloc(net, family, GFP_KERNEL, &vcc_proto);
 153         if (!sk)
 154                 return -ENOMEM;
 155         sock_init_data(sock, sk);
 156         sk->sk_state_change = vcc_def_wakeup;
 157         sk->sk_write_space = vcc_write_space;
 158
 159         vcc = atm_sk(sk);
 160         vcc->dev = NULL;
 161         memset(&vcc->local, 0, sizeof(struct sockaddr_atmsvc));
 162         memset(&vcc->remote, 0, sizeof(struct sockaddr_atmsvc));
 163         vcc->qos.txtp.max_sdu = 1 << 16; /* for meta VCs */
 164         atomic_set(&sk->sk_wmem_alloc, 1);
 165         atomic_set(&sk->sk_rmem_alloc, 0);
 166         vcc->push = NULL;
 167         vcc->pop = NULL;
 168         vcc->owner = NULL;
 169         vcc->push_oam = NULL;
 170         vcc->release_cb = NULL;
 171         vcc->vpi = vcc->vci = 0; /* no VCI/VPI yet */
 172         vcc->atm_options = vcc->aal_options = 0;
 173         sk->sk_destruct = vcc_sock_destruct;
 174         return 0;
 175 }
 176
 177 static void vcc_destroy_socket(struct sock *sk)
 178 {
 179         struct atm_vcc *vcc = atm_sk(sk);
 180         struct sk_buff *skb;
 181
 182         set_bit(ATM_VF_CLOSE, &vcc->flags);
 183         clear_bit(ATM_VF_READY, &vcc->flags);
 184         if (vcc->dev) {
 185                 if (vcc->dev->ops->close)
 186                         vcc->dev->ops->close(vcc);
 187                 if (vcc->push)
 188                         vcc->push(vcc, NULL); /* atmarpd has no push */
 189                 module_put(vcc->owner);
 190
 191                 while ((skb = skb_dequeue(&sk->sk_receive_queue)) != NULL) {
 192                         atm_return(vcc, skb->truesize);
 193                         kfree_skb(skb);
 194                 }
 195
 196                 module_put(vcc->dev->ops->owner);
 197                 atm_dev_put(vcc->dev);
 198         }
 199
 200         vcc_remove_socket(sk);
 201 }
 202
 203 int vcc_release(struct socket *sock)
 204 {
 205         struct sock *sk = sock->sk;
 206
 207         if (sk) {
 208                 lock_sock(sk);
 209                 vcc_destroy_socket(sock->sk);
 210                 release_sock(sk);
 211                 sock_put(sk);
 212         }
 213
 214         return 0;
 215 }
 216
 217 void vcc_release_async(struct atm_vcc *vcc, int reply)
 218 {
 219         struct sock *sk = sk_atm(vcc);
 220
 221         set_bit(ATM_VF_CLOSE, &vcc->flags);
 222         sk->sk_shutdown |= RCV_SHUTDOWN;
 223         sk->sk_err = -reply;
 224         clear_bit(ATM_VF_WAITING, &vcc->flags);
 225         sk->sk_state_change(sk);
 226 }
 227 EXPORT_SYMBOL(vcc_release_async);
 228
 229 void vcc_process_recv_queue(struct atm_vcc *vcc)
 230 {
 231         struct sk_buff_head queue, *rq;
 232         struct sk_buff *skb, *tmp;
 233         unsigned long flags;
 234
 235         __skb_queue_head_init(&queue);
 236         rq = &sk_atm(vcc)->sk_receive_queue;
 237
 238         spin_lock_irqsave(&rq->lock, flags);
 239         skb_queue_splice_init(rq, &queue);
 240         spin_unlock_irqrestore(&rq->lock, flags);
 241
 242         skb_queue_walk_safe(&queue, skb, tmp) {
 243                 __skb_unlink(skb, &queue);
 244                 vcc->push(vcc, skb);
 245         }
 246 }
 247 EXPORT_SYMBOL(vcc_process_recv_queue);
 248
 249 void atm_dev_signal_change(struct atm_dev *dev, char signal)
 250 {
 251         pr_debug("%s signal=%d dev=%p number=%d dev->signal=%d\n",
 252                 __func__, signal, dev, dev->number, dev->signal);
 253
 254         /* atm driver sending invalid signal */
 255         WARN_ON(signal < ATM_PHY_SIG_LOST || signal > ATM_PHY_SIG_FOUND);
 256
 257         if (dev->signal == signal)
 258                 return; /* no change */
 259
 260         dev->signal = signal;
 261
 262         atomic_notifier_call_chain(&atm_dev_notify_chain, signal, dev);
 263 }
 264 EXPORT_SYMBOL(atm_dev_signal_change);
 265
 266 void atm_dev_release_vccs(struct atm_dev *dev)
 267 {
 268         int i;
 269
 270         write_lock_irq(&vcc_sklist_lock);
 271         for (i = 0; i < VCC_HTABLE_SIZE; i++) {
 272                 struct hlist_head *head = &vcc_hash[i];
 273                 struct hlist_node *tmp;
 274                 struct sock *s;
 275                 struct atm_vcc *vcc;
 276
 277                 sk_for_each_safe(s, tmp, head) {
 278                         vcc = atm_sk(s);
 279                         if (vcc->dev == dev) {
 280                                 vcc_release_async(vcc, -EPIPE);
 281                                 sk_del_node_init(s);
 282                         }
 283                 }
 284         }
 285         write_unlock_irq(&vcc_sklist_lock);
 286 }
 287 EXPORT_SYMBOL(atm_dev_release_vccs);
 288
 289 static int adjust_tp(struct atm_trafprm *tp, unsigned char aal)
 290 {
 291         int max_sdu;
 292
 293         if (!tp->traffic_class)
 294                 return 0;
 295         switch (aal) {
 296         case ATM_AAL0:
 297                 max_sdu = ATM_CELL_SIZE-1;
 298                 break;
 299         case ATM_AAL34:
 300                 max_sdu = ATM_MAX_AAL34_PDU;
 301                 break;
 302         default:
 303                 pr_warning("AAL problems ... (%d)\n", aal);
 304                 /* fall through */
 305         case ATM_AAL5:
 306                 max_sdu = ATM_MAX_AAL5_PDU;
 307         }
 308         if (!tp->max_sdu)
 309                 tp->max_sdu = max_sdu;
 310         else if (tp->max_sdu > max_sdu)
 311                 return -EINVAL;
 312         if (!tp->max_cdv)
 313                 tp->max_cdv = ATM_MAX_CDV;
 314         return 0;
 315 }
 316
 317 static int check_ci(const struct atm_vcc *vcc, short vpi, int vci)
 318 {
 319         struct hlist_head *head = &vcc_hash[vci & (VCC_HTABLE_SIZE - 1)];
 320         struct sock *s;
 321         struct atm_vcc *walk;
 322
 323         sk_for_each(s, head) {
 324                 walk = atm_sk(s);
 325                 if (walk->dev != vcc->dev)
 326                         continue;
 327                 if (test_bit(ATM_VF_ADDR, &walk->flags) && walk->vpi == vpi &&
 328                     walk->vci == vci && ((walk->qos.txtp.traffic_class !=
 329                     ATM_NONE && vcc->qos.txtp.traffic_class != ATM_NONE) ||
 330                     (walk->qos.rxtp.traffic_class != ATM_NONE &&
 331                     vcc->qos.rxtp.traffic_class != ATM_NONE)))
 332                         return -EADDRINUSE;
 333         }
 334
 335         /* allow VCCs with same VPI/VCI iff they don't collide on
 336            TX/RX (but we may refuse such sharing for other reasons,
 337            e.g. if protocol requires to have both channels) */
 338
 339         return 0;
 340 }
 341
 342 static int find_ci(const struct atm_vcc *vcc, short *vpi, int *vci)
 343 {
 344         static short p;        /* poor man's per-device cache */
 345         static int c;
 346         short old_p;
 347         int old_c;
 348         int err;
 349
 350         if (*vpi != ATM_VPI_ANY && *vci != ATM_VCI_ANY) {
 351                 err = check_ci(vcc, *vpi, *vci);
 352                 return err;
 353         }
 354         /* last scan may have left values out of bounds for current device */
 355         if (*vpi != ATM_VPI_ANY)
 356                 p = *vpi;
 357         else if (p >= 1 << vcc->dev->ci_range.vpi_bits)
 358                 p = 0;
 359         if (*vci != ATM_VCI_ANY)
 360                 c = *vci;
 361         else if (c < ATM_NOT_RSV_VCI || c >= 1 << vcc->dev->ci_range.vci_bits)
 362                         c = ATM_NOT_RSV_VCI;
 363         old_p = p;
 364         old_c = c;
 365         do {
 366                 if (!check_ci(vcc, p, c)) {
 367                         *vpi = p;
 368                         *vci = c;
 369                         return 0;
 370                 }
 371                 if (*vci == ATM_VCI_ANY) {
 372                         c++;
 373                         if (c >= 1 << vcc->dev->ci_range.vci_bits)
 374                                 c = ATM_NOT_RSV_VCI;
 375                 }
 376                 if ((c == ATM_NOT_RSV_VCI || *vci != ATM_VCI_ANY) &&
 377                     *vpi == ATM_VPI_ANY) {
 378                         p++;
 379                         if (p >= 1 << vcc->dev->ci_range.vpi_bits)
 380                                 p = 0;
 381                 }
 382         } while (old_p != p || old_c != c);
 383         return -EADDRINUSE;
 384 }
 385
 386 static int __vcc_connect(struct atm_vcc *vcc, struct atm_dev *dev, short vpi,
 387                          int vci)
 388 {
 389         struct sock *sk = sk_atm(vcc);
 390         int error;
 391
 392         if ((vpi != ATM_VPI_UNSPEC && vpi != ATM_VPI_ANY &&
 393             vpi >> dev->ci_range.vpi_bits) || (vci != ATM_VCI_UNSPEC &&
 394             vci != ATM_VCI_ANY && vci >> dev->ci_range.vci_bits))
 395                 return -EINVAL;
 396         if (vci > 0 && vci < ATM_NOT_RSV_VCI && !capable(CAP_NET_BIND_SERVICE))
 397                 return -EPERM;
 398         error = -ENODEV;
 399         if (!try_module_get(dev->ops->owner))
 400                 return error;
 401         vcc->dev = dev;
 402         write_lock_irq(&vcc_sklist_lock);
 403         if (test_bit(ATM_DF_REMOVED, &dev->flags) ||
 404             (error = find_ci(vcc, &vpi, &vci))) {
 405                 write_unlock_irq(&vcc_sklist_lock);
 406                 goto fail_module_put;
 407         }
 408         vcc->vpi = vpi;
 409         vcc->vci = vci;
 410         __vcc_insert_socket(sk);
 411         write_unlock_irq(&vcc_sklist_lock);
 412         switch (vcc->qos.aal) {
 413         case ATM_AAL0:
 414                 error = atm_init_aal0(vcc);
 415                 vcc->stats = &dev->stats.aal0;
 416                 break;
 417         case ATM_AAL34:
 418                 error = atm_init_aal34(vcc);
 419                 vcc->stats = &dev->stats.aal34;
 420                 break;
 421         case ATM_NO_AAL:
 422                 /* ATM_AAL5 is also used in the "0 for default" case */
 423                 vcc->qos.aal = ATM_AAL5;
 424                 /* fall through */
 425         case ATM_AAL5:
 426                 error = atm_init_aal5(vcc);
 427                 vcc->stats = &dev->stats.aal5;
 428                 break;
 429         default:
 430                 error = -EPROTOTYPE;
 431         }
 432         if (!error)
 433                 error = adjust_tp(&vcc->qos.txtp, vcc->qos.aal);
 434         if (!error)
 435                 error = adjust_tp(&vcc->qos.rxtp, vcc->qos.aal);
 436         if (error)
 437                 goto fail;
 438         pr_debug("VCC %d.%d, AAL %d\n", vpi, vci, vcc->qos.aal);
 439         pr_debug("  TX: %d, PCR %d..%d, SDU %d\n",
 440                  vcc->qos.txtp.traffic_class,
 441                  vcc->qos.txtp.min_pcr,
 442                  vcc->qos.txtp.max_pcr,
 443                  vcc->qos.txtp.max_sdu);
 444         pr_debug("  RX: %d, PCR %d..%d, SDU %d\n",
 445                  vcc->qos.rxtp.traffic_class,
 446                  vcc->qos.rxtp.min_pcr,
 447                  vcc->qos.rxtp.max_pcr,
 448                  vcc->qos.rxtp.max_sdu);
 449
 450         if (dev->ops->open) {
 451                 error = dev->ops->open(vcc);
 452                 if (error)
 453                         goto fail;
 454         }
 455         return 0;
 456
 457 fail:
 458         vcc_remove_socket(sk);
 459 fail_module_put:
 460         module_put(dev->ops->owner);
 461         /* ensure we get dev module ref count correct */
 462         vcc->dev = NULL;
 463         return error;
 464 }
 465
 466 int vcc_connect(struct socket *sock, int itf, short vpi, int vci)
 467 {
 468         struct atm_dev *dev;
 469         struct atm_vcc *vcc = ATM_SD(sock);
 470         int error;
 471
 472         pr_debug("(vpi %d, vci %d)\n", vpi, vci);
 473         if (sock->state == SS_CONNECTED)
 474                 return -EISCONN;
 475         if (sock->state != SS_UNCONNECTED)
 476                 return -EINVAL;
 477         if (!(vpi || vci))
 478                 return -EINVAL;
 479
 480         if (vpi != ATM_VPI_UNSPEC && vci != ATM_VCI_UNSPEC)
 481                 clear_bit(ATM_VF_PARTIAL, &vcc->flags);
 482         else
 483                 if (test_bit(ATM_VF_PARTIAL, &vcc->flags))
 484                         return -EINVAL;
 485         pr_debug("(TX: cl %d,bw %d-%d,sdu %d; "
 486                  "RX: cl %d,bw %d-%d,sdu %d,AAL %s%d)\n",
 487                  vcc->qos.txtp.traffic_class, vcc->qos.txtp.min_pcr,
 488                  vcc->qos.txtp.max_pcr, vcc->qos.txtp.max_sdu,
 489                  vcc->qos.rxtp.traffic_class, vcc->qos.rxtp.min_pcr,
 490                  vcc->qos.rxtp.max_pcr, vcc->qos.rxtp.max_sdu,
 491                  vcc->qos.aal == ATM_AAL5 ? "" :
 492                  vcc->qos.aal == ATM_AAL0 ? "" : " ??? code ",
 493                  vcc->qos.aal == ATM_AAL0 ? 0 : vcc->qos.aal);
 494         if (!test_bit(ATM_VF_HASQOS, &vcc->flags))
 495                 return -EBADFD;
 496         if (vcc->qos.txtp.traffic_class == ATM_ANYCLASS ||
 497             vcc->qos.rxtp.traffic_class == ATM_ANYCLASS)
 498                 return -EINVAL;
 499         if (likely(itf != ATM_ITF_ANY)) {
 500                 dev = try_then_request_module(atm_dev_lookup(itf),
 501                                               "atm-device-%d", itf);
 502         } else {
 503                 dev = NULL;
 504                 mutex_lock(&atm_dev_mutex);
 505                 if (!list_empty(&atm_devs)) {
 506                         dev = list_entry(atm_devs.next,
 507                                          struct atm_dev, dev_list);
 508                         atm_dev_hold(dev);
 509                 }
 510                 mutex_unlock(&atm_dev_mutex);
 511         }
 512         if (!dev)
 513                 return -ENODEV;
 514         error = __vcc_connect(vcc, dev, vpi, vci);
 515         if (error) {
 516                 atm_dev_put(dev);
 517                 return error;
 518         }
 519         if (vpi == ATM_VPI_UNSPEC || vci == ATM_VCI_UNSPEC)
 520                 set_bit(ATM_VF_PARTIAL, &vcc->flags);
 521         if (test_bit(ATM_VF_READY, &ATM_SD(sock)->flags))
 522                 sock->state = SS_CONNECTED;
 523         return 0;
 524 }
 525
 526 int vcc_recvmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg,
 527                 size_t size, int flags)
 528 {
 529         struct sock *sk = sock->sk;
 530         struct atm_vcc *vcc;
 531         struct sk_buff *skb;
 532         int copied, error = -EINVAL;
 533
 534         if (sock->state != SS_CONNECTED)
 535                 return -ENOTCONN;
 536
 537         /* only handle MSG_DONTWAIT and MSG_PEEK */
 538         if (flags & ~(MSG_DONTWAIT | MSG_PEEK))
 539                 return -EOPNOTSUPP;
 540
 541         vcc = ATM_SD(sock);
 542         if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
 543             test_bit(ATM_VF_CLOSE, &vcc->flags) ||
 544             !test_bit(ATM_VF_READY, &vcc->flags))
 545                 return 0;
 546
 547         skb = skb_recv_datagram(sk, flags, flags & MSG_DONTWAIT, &error);
 548         if (!skb)
 549                 return error;
 550
 551         copied = skb->len;
 552         if (copied > size) {
 553                 copied = size;
 554                 msg->msg_flags |= MSG_TRUNC;
 555         }
 556
 557         error = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
 558         if (error)
 559                 return error;
 560         sock_recv_ts_and_drops(msg, sk, skb);
 561
 562         if (!(flags & MSG_PEEK)) {
 563                 pr_debug("%d -= %d\n", atomic_read(&sk->sk_rmem_alloc),
 564                          skb->truesize);
 565                 atm_return(vcc, skb->truesize);
 566         }
 567
 568         skb_free_datagram(sk, skb);
 569         return copied;
 570 }
 571
 572 int vcc_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *m,
 573                 size_t total_len)
 574 {
 575         struct sock *sk = sock->sk;
 576         DEFINE_WAIT(wait);
 577         struct atm_vcc *vcc;
 578         struct sk_buff *skb;
 579         int eff, error;
 580         const void __user *buff;
 581         int size;
 582
 583         lock_sock(sk);
 584         if (sock->state != SS_CONNECTED) {
 585                 error = -ENOTCONN;
 586                 goto out;
 587         }
 588         if (m->msg_name) {
 589                 error = -EISCONN;
 590                 goto out;
 591         }
 592         if (m->msg_iovlen != 1) {
 593                 error = -ENOSYS; /* fix this later @@@ */
 594                 goto out;
 595         }
 596         buff = m->msg_iov->iov_base;
 597         size = m->msg_iov->iov_len;
 598         vcc = ATM_SD(sock);
 599         if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
 600             test_bit(ATM_VF_CLOSE, &vcc->flags) ||
 601             !test_bit(ATM_VF_READY, &vcc->flags)) {
 602                 error = -EPIPE;
 603                 send_sig(SIGPIPE, current, 0);
 604                 goto out;
 605         }
 606         if (!size) {
 607                 error = 0;
 608                 goto out;
 609         }
 610         if (size < 0 || size > vcc->qos.txtp.max_sdu) {
 611                 error = -EMSGSIZE;
 612                 goto out;
 613         }
 614
 615         eff = (size+3) & ~3; /* align to word boundary */
 616         prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
 617         error = 0;
 618         while (!(skb = alloc_tx(vcc, eff))) {
 619                 if (m->msg_flags & MSG_DONTWAIT) {
 620                         error = -EAGAIN;
 621                         break;
 622                 }
 623                 schedule();
 624                 if (signal_pending(current)) {
 625                         error = -ERESTARTSYS;
 626                         break;
 627                 }
 628                 if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
 629                     test_bit(ATM_VF_CLOSE, &vcc->flags) ||
 630                     !test_bit(ATM_VF_READY, &vcc->flags)) {
 631                         error = -EPIPE;
 632                         send_sig(SIGPIPE, current, 0);
 633                         break;
 634                 }
 635                 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
 636         }
 637         finish_wait(sk_sleep(sk), &wait);
 638         if (error)
 639                 goto out;
 640         skb->dev = NULL; /* for paths shared with net_device interfaces */
 641         ATM_SKB(skb)->atm_options = vcc->atm_options;
 642         if (copy_from_user(skb_put(skb, size), buff, size)) {
 643                 kfree_skb(skb);
 644                 error = -EFAULT;
 645                 goto out;
 646         }
 647         if (eff != size)
 648                 memset(skb->data + size, 0, eff-size);
 649         error = vcc->dev->ops->send(vcc, skb);
 650         error = error ? error : size;
 651 out:
 652         release_sock(sk);
 653         return error;
 654 }
 655
 656 unsigned int vcc_poll(struct file *file, struct socket *sock, poll_table *wait)
 657 {
 658         struct sock *sk = sock->sk;
 659         struct atm_vcc *vcc;
 660         unsigned int mask;
 661
 662         sock_poll_wait(file, sk_sleep(sk), wait);
 663         mask = 0;
 664
 665         vcc = ATM_SD(sock);
 666
 667         /* exceptional events */
 668         if (sk->sk_err)
 669                 mask = POLLERR;
 670
 671         if (test_bit(ATM_VF_RELEASED, &vcc->flags) ||
 672             test_bit(ATM_VF_CLOSE, &vcc->flags))
 673                 mask |= POLLHUP;
 674
 675         /* readable? */
 676         if (!skb_queue_empty(&sk->sk_receive_queue))
 677                 mask |= POLLIN | POLLRDNORM;
 678
 679         /* writable? */
 680         if (sock->state == SS_CONNECTING &&
 681             test_bit(ATM_VF_WAITING, &vcc->flags))
 682                 return mask;
 683
 684         if (vcc->qos.txtp.traffic_class != ATM_NONE &&
 685             vcc_writable(sk))
 686                 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
 687
 688         return mask;
 689 }
 690
 691 static int atm_change_qos(struct atm_vcc *vcc, struct atm_qos *qos)
 692 {
 693         int error;
 694
 695         /*
 696          * Don't let the QoS change the already connected AAL type nor the
 697          * traffic class.
 698          */
 699         if (qos->aal != vcc->qos.aal ||
 700             qos->rxtp.traffic_class != vcc->qos.rxtp.traffic_class ||
 701             qos->txtp.traffic_class != vcc->qos.txtp.traffic_class)
 702                 return -EINVAL;
 703         error = adjust_tp(&qos->txtp, qos->aal);
 704         if (!error)
 705                 error = adjust_tp(&qos->rxtp, qos->aal);
 706         if (error)
 707                 return error;
 708         if (!vcc->dev->ops->change_qos)
 709                 return -EOPNOTSUPP;
 710         if (sk_atm(vcc)->sk_family == AF_ATMPVC)
 711                 return vcc->dev->ops->change_qos(vcc, qos, ATM_MF_SET);
 712         return svc_change_qos(vcc, qos);
 713 }
 714
 715 static int check_tp(const struct atm_trafprm *tp)
 716 {
 717         /* @@@ Should be merged with adjust_tp */
 718         if (!tp->traffic_class || tp->traffic_class == ATM_ANYCLASS)
 719                 return 0;
 720         if (tp->traffic_class != ATM_UBR && !tp->min_pcr && !tp->pcr &&
 721             !tp->max_pcr)
 722                 return -EINVAL;
 723         if (tp->min_pcr == ATM_MAX_PCR)
 724                 return -EINVAL;
 725         if (tp->min_pcr && tp->max_pcr && tp->max_pcr != ATM_MAX_PCR &&
 726             tp->min_pcr > tp->max_pcr)
 727                 return -EINVAL;
 728         /*
 729          * We allow pcr to be outside [min_pcr,max_pcr], because later
 730          * adjustment may still push it in the valid range.
 731          */
 732         return 0;
 733 }
 734
 735 static int check_qos(const struct atm_qos *qos)
 736 {
 737         int error;
 738
 739         if (!qos->txtp.traffic_class && !qos->rxtp.traffic_class)
 740                 return -EINVAL;
 741         if (qos->txtp.traffic_class != qos->rxtp.traffic_class &&
 742             qos->txtp.traffic_class && qos->rxtp.traffic_class &&
 743             qos->txtp.traffic_class != ATM_ANYCLASS &&
 744             qos->rxtp.traffic_class != ATM_ANYCLASS)
 745                 return -EINVAL;
 746         error = check_tp(&qos->txtp);
 747         if (error)
 748                 return error;
 749         return check_tp(&qos->rxtp);
 750 }
 751
 752 int vcc_setsockopt(struct socket *sock, int level, int optname,
 753                    char __user *optval, unsigned int optlen)
 754 {
 755         struct atm_vcc *vcc;
 756         unsigned long value;
 757         int error;
 758
 759         if (__SO_LEVEL_MATCH(optname, level) && optlen != __SO_SIZE(optname))
 760                 return -EINVAL;
 761
 762         vcc = ATM_SD(sock);
 763         switch (optname) {
 764         case SO_ATMQOS:
 765         {
 766                 struct atm_qos qos;
 767
 768                 if (copy_from_user(&qos, optval, sizeof(qos)))
 769                         return -EFAULT;
 770                 error = check_qos(&qos);
 771                 if (error)
 772                         return error;
 773                 if (sock->state == SS_CONNECTED)
 774                         return atm_change_qos(vcc, &qos);
 775                 if (sock->state != SS_UNCONNECTED)
 776                         return -EBADFD;
 777                 vcc->qos = qos;
 778                 set_bit(ATM_VF_HASQOS, &vcc->flags);
 779                 return 0;
 780         }
 781         case SO_SETCLP:
 782                 if (get_user(value, (unsigned long __user *)optval))
 783                         return -EFAULT;
 784                 if (value)
 785                         vcc->atm_options |= ATM_ATMOPT_CLP;
 786                 else
 787                         vcc->atm_options &= ~ATM_ATMOPT_CLP;
 788                 return 0;
 789         default:
 790                 if (level == SOL_SOCKET)
 791                         return -EINVAL;
 792                 break;
 793         }
 794         if (!vcc->dev || !vcc->dev->ops->setsockopt)
 795                 return -EINVAL;
 796         return vcc->dev->ops->setsockopt(vcc, level, optname, optval, optlen);
 797 }
 798
 799 int vcc_getsockopt(struct socket *sock, int level, int optname,
 800                    char __user *optval, int __user *optlen)
 801 {
 802         struct atm_vcc *vcc;
 803         int len;
 804
 805         if (get_user(len, optlen))
 806                 return -EFAULT;
 807         if (__SO_LEVEL_MATCH(optname, level) && len != __SO_SIZE(optname))
 808                 return -EINVAL;
 809
 810         vcc = ATM_SD(sock);
 811         switch (optname) {
 812         case SO_ATMQOS:
 813                 if (!test_bit(ATM_VF_HASQOS, &vcc->flags))
 814                         return -EINVAL;
 815                 return copy_to_user(optval, &vcc->qos, sizeof(vcc->qos))
 816                         ? -EFAULT : 0;
 817         case SO_SETCLP:
 818                 return put_user(vcc->atm_options & ATM_ATMOPT_CLP ? 1 : 0,
 819                                 (unsigned long __user *)optval) ? -EFAULT : 0;
 820         case SO_ATMPVC:
 821         {
 822                 struct sockaddr_atmpvc pvc;
 823
 824                 if (!vcc->dev || !test_bit(ATM_VF_ADDR, &vcc->flags))
 825                         return -ENOTCONN;
 826                 memset(&pvc, 0, sizeof(pvc));
 827                 pvc.sap_family = AF_ATMPVC;
 828                 pvc.sap_addr.itf = vcc->dev->number;
 829                 pvc.sap_addr.vpi = vcc->vpi;
 830                 pvc.sap_addr.vci = vcc->vci;
 831                 return copy_to_user(optval, &pvc, sizeof(pvc)) ? -EFAULT : 0;
 832         }
 833         default:
 834                 if (level == SOL_SOCKET)
 835                         return -EINVAL;
 836                 break;
 837         }
 838         if (!vcc->dev || !vcc->dev->ops->getsockopt)
 839                 return -EINVAL;
 840         return vcc->dev->ops->getsockopt(vcc, level, optname, optval, len);
 841 }
 842
 843 int register_atmdevice_notifier(struct notifier_block *nb)
 844 {
 845         return atomic_notifier_chain_register(&atm_dev_notify_chain, nb);
 846 }
 847 EXPORT_SYMBOL_GPL(register_atmdevice_notifier);
 848
 849 void unregister_atmdevice_notifier(struct notifier_block *nb)
 850 {
 851         atomic_notifier_chain_unregister(&atm_dev_notify_chain, nb);
 852 }
 853 EXPORT_SYMBOL_GPL(unregister_atmdevice_notifier);
 854
 855 static int __init atm_init(void)
 856 {
 857         int error;
 858
 859         error = proto_register(&vcc_proto, 0);
 860         if (error < 0)
 861                 goto out;
 862         error = atmpvc_init();
 863         if (error < 0) {
 864                 pr_err("atmpvc_init() failed with %d\n", error);
 865                 goto out_unregister_vcc_proto;
 866         }
 867         error = atmsvc_init();
 868         if (error < 0) {
 869                 pr_err("atmsvc_init() failed with %d\n", error);
 870                 goto out_atmpvc_exit;
 871         }
 872         error = atm_proc_init();
 873         if (error < 0) {
 874                 pr_err("atm_proc_init() failed with %d\n", error);
 875                 goto out_atmsvc_exit;
 876         }
 877         error = atm_sysfs_init();
 878         if (error < 0) {
 879                 pr_err("atm_sysfs_init() failed with %d\n", error);
 880                 goto out_atmproc_exit;
 881         }
 882 out:
 883         return error;
 884 out_atmproc_exit:
 885         atm_proc_exit();
 886 out_atmsvc_exit:
 887         atmsvc_exit();
 888 out_atmpvc_exit:
 889         atmsvc_exit();
 890 out_unregister_vcc_proto:
 891         proto_unregister(&vcc_proto);
 892         goto out;
 893 }
 894
 895 static void __exit atm_exit(void)
 896 {
 897         atm_proc_exit();
 898         atm_sysfs_exit();
 899         atmsvc_exit();
 900         atmpvc_exit();
 901         proto_unregister(&vcc_proto);
 902 }
 903
 904 subsys_initcall(atm_init);
 905
 906 module_exit(atm_exit);
 907
 908 MODULE_LICENSE("GPL");
 909 MODULE_ALIAS_NETPROTO(PF_ATMPVC);
 910 MODULE_ALIAS_NETPROTO(PF_ATMSVC);