Fedora kernel-2.6.17-1.2142_FC4 patched with stable patch-2.6.17.4-vs2.0.2-rc26.diff
[linux-2.6.git] / net / atm / svc.c
index 035068b..3a180cf 100644 (file)
@@ -53,27 +53,27 @@ static void svc_disconnect(struct atm_vcc *vcc)
 {
        DEFINE_WAIT(wait);
        struct sk_buff *skb;
+       struct sock *sk = sk_atm(vcc);
 
        DPRINTK("svc_disconnect %p\n",vcc);
        if (test_bit(ATM_VF_REGIS,&vcc->flags)) {
-               prepare_to_wait(vcc->sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
+               prepare_to_wait(sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
                sigd_enq(vcc,as_close,NULL,NULL,NULL);
                while (!test_bit(ATM_VF_RELEASED,&vcc->flags) && sigd) {
                        schedule();
-                       prepare_to_wait(vcc->sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
+                       prepare_to_wait(sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
                }
-               finish_wait(vcc->sk->sk_sleep, &wait);
+               finish_wait(sk->sk_sleep, &wait);
        }
        /* beware - socket is still in use by atmsigd until the last
           as_indicate has been answered */
-       while ((skb = skb_dequeue(&vcc->sk->sk_receive_queue)) != NULL) {
+       while ((skb = skb_dequeue(&sk->sk_receive_queue)) != NULL) {
+               atm_return(vcc, skb->truesize);
                DPRINTK("LISTEN REL\n");
                sigd_enq2(NULL,as_reject,vcc,NULL,NULL,&vcc->qos,0);
                dev_kfree_skb(skb);
        }
-       clear_bit(ATM_VF_REGIS,&vcc->flags);
-       clear_bit(ATM_VF_RELEASED,&vcc->flags);
-       clear_bit(ATM_VF_CLOSE,&vcc->flags);
+       clear_bit(ATM_VF_REGIS, &vcc->flags);
        /* ... may retry later */
 }
 
@@ -90,10 +90,8 @@ static int svc_release(struct socket *sock)
                /* VCC pointer is used as a reference, so we must not free it
                   (thereby subjecting it to re-use) before all pending connections
                   are closed */
-               sock_hold(sk);
-               vcc_release(sock);
                svc_disconnect(vcc);
-               sock_put(sk);
+               vcc_release(sock);
        }
        return 0;
 }
@@ -120,10 +118,6 @@ static int svc_bind(struct socket *sock,struct sockaddr *sockaddr,
                goto out;
        }
        vcc = ATM_SD(sock);
-       if (test_bit(ATM_VF_SESSION, &vcc->flags)) {
-               error = -EINVAL;
-               goto out;
-       }
        addr = (struct sockaddr_atmsvc *) sockaddr;
        if (addr->sas_family != AF_ATMSVC) {
                error = -EAFNOSUPPORT;
@@ -194,10 +188,6 @@ static int svc_connect(struct socket *sock,struct sockaddr *sockaddr,
                }
                break;
        case SS_UNCONNECTED:
-               if (test_bit(ATM_VF_SESSION, &vcc->flags)) {
-                       error = -EINVAL;
-                       goto out;
-               }
                addr = (struct sockaddr_atmsvc *) sockaddr;
                if (addr->sas_family != AF_ATMSVC) {
                        error = -EAFNOSUPPORT;
@@ -290,7 +280,8 @@ static int svc_connect(struct socket *sock,struct sockaddr *sockaddr,
  */
        if (!(error = vcc_connect(sock, vcc->itf, vcc->vpi, vcc->vci)))
                sock->state = SS_CONNECTED;
-       else (void) svc_disconnect(vcc);
+       else
+               (void) svc_disconnect(vcc);
 out:
        release_sock(sk);
        return error;
@@ -311,6 +302,7 @@ static int svc_listen(struct socket *sock,int backlog)
                error = -EINVAL;
                goto out;
        }
+       vcc_insert_socket(sk);
        set_bit(ATM_VF_WAITING, &vcc->flags);
        prepare_to_wait(sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
        sigd_enq(vcc,as_listen,NULL,NULL,&vcc->local);
@@ -324,8 +316,7 @@ static int svc_listen(struct socket *sock,int backlog)
                goto out;
        }
        set_bit(ATM_VF_LISTEN,&vcc->flags);
-       vcc->sk->sk_max_ack_backlog = backlog > 0 ? backlog :
-                                                   ATM_BACKLOG_DEFAULT;
+       sk->sk_max_ack_backlog = backlog > 0 ? backlog : ATM_BACKLOG_DEFAULT;
        error = -sk->sk_err;
 out:
        release_sock(sk);
@@ -354,8 +345,8 @@ static int svc_accept(struct socket *sock,struct socket *newsock,int flags)
        while (1) {
                DEFINE_WAIT(wait);
 
-               prepare_to_wait(old_vcc->sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
-               while (!(skb = skb_dequeue(&old_vcc->sk->sk_receive_queue)) &&
+               prepare_to_wait(sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
+               while (!(skb = skb_dequeue(&sk->sk_receive_queue)) &&
                       sigd) {
                        if (test_bit(ATM_VF_RELEASED,&old_vcc->flags)) break;
                        if (test_bit(ATM_VF_CLOSE,&old_vcc->flags)) {
@@ -373,9 +364,9 @@ static int svc_accept(struct socket *sock,struct socket *newsock,int flags)
                                error = -ERESTARTSYS;
                                break;
                        }
-                       prepare_to_wait(old_vcc->sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
+                       prepare_to_wait(sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
                }
-               finish_wait(old_vcc->sk->sk_sleep, &wait);
+               finish_wait(sk->sk_sleep, &wait);
                if (error)
                        goto out;
                if (!skb) {
@@ -391,7 +382,7 @@ static int svc_accept(struct socket *sock,struct socket *newsock,int flags)
                error = vcc_connect(newsock, msg->pvc.sap_addr.itf,
                                    msg->pvc.sap_addr.vpi, msg->pvc.sap_addr.vci);
                dev_kfree_skb(skb);
-               old_vcc->sk->sk_ack_backlog--;
+               sk->sk_ack_backlog--;
                if (error) {
                        sigd_enq2(NULL,as_reject,old_vcc,NULL,NULL,
                            &old_vcc->qos,error);
@@ -400,23 +391,23 @@ static int svc_accept(struct socket *sock,struct socket *newsock,int flags)
                }
                /* wait should be short, so we ignore the non-blocking flag */
                set_bit(ATM_VF_WAITING, &new_vcc->flags);
-               prepare_to_wait(new_vcc->sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
+               prepare_to_wait(sk_atm(new_vcc)->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
                sigd_enq(new_vcc,as_accept,old_vcc,NULL,NULL);
                while (test_bit(ATM_VF_WAITING, &new_vcc->flags) && sigd) {
                        release_sock(sk);
                        schedule();
                        lock_sock(sk);
-                       prepare_to_wait(new_vcc->sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
+                       prepare_to_wait(sk_atm(new_vcc)->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
                }
-               finish_wait(new_vcc->sk->sk_sleep, &wait);
+               finish_wait(sk_atm(new_vcc)->sk_sleep, &wait);
                if (!sigd) {
                        error = -EUNATCH;
                        goto out;
                }
-               if (!new_vcc->sk->sk_err)
+               if (!sk_atm(new_vcc)->sk_err)
                        break;
-               if (new_vcc->sk->sk_err != ERESTARTSYS) {
-                       error = -new_vcc->sk->sk_err;
+               if (sk_atm(new_vcc)->sk_err != ERESTARTSYS) {
+                       error = -sk_atm(new_vcc)->sk_err;
                        goto out;
                }
        }
@@ -442,40 +433,65 @@ static int svc_getname(struct socket *sock,struct sockaddr *sockaddr,
 
 int svc_change_qos(struct atm_vcc *vcc,struct atm_qos *qos)
 {
+       struct sock *sk = sk_atm(vcc);
        DEFINE_WAIT(wait);
 
        set_bit(ATM_VF_WAITING, &vcc->flags);
-       prepare_to_wait(vcc->sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
+       prepare_to_wait(sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
        sigd_enq2(vcc,as_modify,NULL,NULL,&vcc->local,qos,0);
        while (test_bit(ATM_VF_WAITING, &vcc->flags) &&
               !test_bit(ATM_VF_RELEASED, &vcc->flags) && sigd) {
                schedule();
-               prepare_to_wait(vcc->sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
+               prepare_to_wait(sk->sk_sleep, &wait, TASK_UNINTERRUPTIBLE);
        }
-       finish_wait(vcc->sk->sk_sleep, &wait);
+       finish_wait(sk->sk_sleep, &wait);
        if (!sigd) return -EUNATCH;
-       return -vcc->sk->sk_err;
+       return -sk->sk_err;
 }
 
 
-static int svc_setsockopt(struct socket *sock,int level,int optname,
-    char __user *optval,int optlen)
+static int svc_setsockopt(struct socket *sock, int level, int optname,
+                         char __user *optval, int optlen)
 {
        struct sock *sk = sock->sk;
-       struct atm_vcc *vcc;
-       int error = 0;
+       struct atm_vcc *vcc = ATM_SD(sock);
+       int value, error = 0;
 
-       if (!__SO_LEVEL_MATCH(optname, level) || optname != SO_ATMSAP ||
-           optlen != sizeof(struct atm_sap)) {
-               error = vcc_setsockopt(sock, level, optname, optval, optlen);
-               goto out;
-       }
-       vcc = ATM_SD(sock);
-       if (copy_from_user(&vcc->sap, optval, optlen)) {
-               error = -EFAULT;
-               goto out;
+       lock_sock(sk);
+       switch (optname) {
+               case SO_ATMSAP:
+                       if (level != SOL_ATM || optlen != sizeof(struct atm_sap)) {
+                               error = -EINVAL;
+                               goto out;
+                       }
+                       if (copy_from_user(&vcc->sap, optval, optlen)) {
+                               error = -EFAULT;
+                               goto out;
+                       }
+                       set_bit(ATM_VF_HASSAP, &vcc->flags);
+                       break;
+               case SO_MULTIPOINT:
+                       if (level != SOL_ATM || optlen != sizeof(int)) {
+                               error = -EINVAL;
+                               goto out;
+                       }
+                       if (get_user(value, (int __user *) optval)) {
+                               error = -EFAULT;
+                               goto out;
+                       }
+                       if (value == 1) {
+                               set_bit(ATM_VF_SESSION, &vcc->flags);
+                       } else if (value == 0) {
+                               clear_bit(ATM_VF_SESSION, &vcc->flags);
+                       } else {
+                               error = -EINVAL;
+                       }
+                       break;
+               default:
+                       error = vcc_setsockopt(sock, level, optname,
+                                              optval, optlen);
        }
-       set_bit(ATM_VF_HASSAP, &vcc->flags);
+
 out:
        release_sock(sk);
        return error;
@@ -511,7 +527,93 @@ out:
 }
 
 
-static struct proto_ops svc_proto_ops = {
+static int svc_addparty(struct socket *sock, struct sockaddr *sockaddr,
+                       int sockaddr_len, int flags)
+{
+       DEFINE_WAIT(wait);
+       struct sock *sk = sock->sk;
+       struct atm_vcc *vcc = ATM_SD(sock);
+       int error;
+
+       lock_sock(sk);
+       set_bit(ATM_VF_WAITING, &vcc->flags);
+       prepare_to_wait(sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
+       sigd_enq(vcc, as_addparty, NULL, NULL,
+                (struct sockaddr_atmsvc *) sockaddr);
+       if (flags & O_NONBLOCK) {
+               finish_wait(sk->sk_sleep, &wait);
+               error = -EINPROGRESS;
+               goto out;
+       }
+       DPRINTK("svc_addparty added wait queue\n");
+       while (test_bit(ATM_VF_WAITING, &vcc->flags) && sigd) {
+               schedule();
+               prepare_to_wait(sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
+       }
+       finish_wait(sk->sk_sleep, &wait);
+       error = xchg(&sk->sk_err_soft, 0);
+out:
+       release_sock(sk);
+       return error;
+}
+
+
+static int svc_dropparty(struct socket *sock, int ep_ref)
+{
+       DEFINE_WAIT(wait);
+       struct sock *sk = sock->sk;
+       struct atm_vcc *vcc = ATM_SD(sock);
+       int error;
+
+       lock_sock(sk);
+       set_bit(ATM_VF_WAITING, &vcc->flags);
+       prepare_to_wait(sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
+       sigd_enq2(vcc, as_dropparty, NULL, NULL, NULL, NULL, ep_ref);
+       while (test_bit(ATM_VF_WAITING, &vcc->flags) && sigd) {
+               schedule();
+               prepare_to_wait(sk->sk_sleep, &wait, TASK_INTERRUPTIBLE);
+       }
+       finish_wait(sk->sk_sleep, &wait);
+       if (!sigd) {
+               error = -EUNATCH;
+               goto out;
+       }
+       error = xchg(&sk->sk_err_soft, 0);
+out:
+       release_sock(sk);
+       return error;
+}
+
+
+static int svc_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
+{
+       int error, ep_ref;
+       struct sockaddr_atmsvc sa;
+       struct atm_vcc *vcc = ATM_SD(sock);
+  
+       switch (cmd) {
+               case ATM_ADDPARTY:
+                       if (!test_bit(ATM_VF_SESSION, &vcc->flags))
+                               return -EINVAL;
+                       if (copy_from_user(&sa, (void __user *) arg, sizeof(sa)))
+                               return -EFAULT;
+                       error = svc_addparty(sock, (struct sockaddr *) &sa, sizeof(sa), 0);
+                       break;
+               case ATM_DROPPARTY:
+                       if (!test_bit(ATM_VF_SESSION, &vcc->flags))
+                               return -EINVAL;
+                       if (copy_from_user(&ep_ref, (void __user *) arg, sizeof(int)))
+                               return -EFAULT;
+                       error = svc_dropparty(sock, ep_ref);
+                       break;
+               default:
+                       error = vcc_ioctl(sock, cmd, arg);
+       }
+
+       return error;
+}
+
+static const struct proto_ops svc_proto_ops = {
        .family =       PF_ATMSVC,
        .owner =        THIS_MODULE,
 
@@ -522,7 +624,7 @@ static struct proto_ops svc_proto_ops = {
        .accept =       svc_accept,
        .getname =      svc_getname,
        .poll =         vcc_poll,
-       .ioctl =        vcc_ioctl,
+       .ioctl =        svc_ioctl,
        .listen =       svc_listen,
        .shutdown =     svc_shutdown,
        .setsockopt =   svc_setsockopt,