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