Merge branch 'for-linus' of git://git.kernel.org/pub/scm/linux/kernel/git/geert/linux...
[safe/jmp/linux-2.6] / net / phonet / socket.c
index dfd4061..69c8b82 100644 (file)
 
 #include <linux/kernel.h>
 #include <linux/net.h>
+#include <linux/poll.h>
 #include <net/sock.h>
 #include <net/tcp_states.h>
 
 #include <linux/phonet.h>
 #include <net/phonet/phonet.h>
+#include <net/phonet/pep.h>
 #include <net/phonet/pn_dev.h>
 
 static int pn_socket_release(struct socket *sock)
@@ -43,32 +45,50 @@ static int pn_socket_release(struct socket *sock)
        return 0;
 }
 
+#define PN_HASHSIZE    16
+#define PN_HASHMASK    (PN_HASHSIZE-1)
+
+
 static struct  {
-       struct hlist_head hlist;
+       struct hlist_head hlist[PN_HASHSIZE];
        spinlock_t lock;
-} pnsocks = {
-       .hlist = HLIST_HEAD_INIT,
-       .lock = __SPIN_LOCK_UNLOCKED(pnsocks.lock),
-};
+} pnsocks;
+
+void __init pn_sock_init(void)
+{
+       unsigned i;
+
+       for (i = 0; i < PN_HASHSIZE; i++)
+               INIT_HLIST_HEAD(pnsocks.hlist + i);
+       spin_lock_init(&pnsocks.lock);
+}
+
+static struct hlist_head *pn_hash_list(u16 obj)
+{
+       return pnsocks.hlist + (obj & PN_HASHMASK);
+}
 
 /*
  * Find address based on socket address, match only certain fields.
  * Also grab sock if it was found. Remember to sock_put it later.
  */
-struct sock *pn_find_sock_by_sa(const struct sockaddr_pn *spn)
+struct sock *pn_find_sock_by_sa(struct net *net, const struct sockaddr_pn *spn)
 {
        struct hlist_node *node;
        struct sock *sknode;
        struct sock *rval = NULL;
        u16 obj = pn_sockaddr_get_object(spn);
        u8 res = spn->spn_resource;
+       struct hlist_head *hlist = pn_hash_list(obj);
 
        spin_lock_bh(&pnsocks.lock);
 
-       sk_for_each(sknode, node, &pnsocks.hlist) {
+       sk_for_each(sknode, node, hlist) {
                struct pn_sock *pn = pn_sk(sknode);
                BUG_ON(!pn->sobject); /* unbound socket */
 
+               if (!net_eq(sock_net(sknode), net))
+                       continue;
                if (pn_port(obj)) {
                        /* Look up socket by port */
                        if (pn_port(pn->sobject) != pn_port(obj))
@@ -78,8 +98,8 @@ struct sock *pn_find_sock_by_sa(const struct sockaddr_pn *spn)
                        if (pn->resource != res)
                                continue;
                }
-               if (pn_addr(pn->sobject)
-                && pn_addr(pn->sobject) != pn_addr(obj))
+               if (pn_addr(pn->sobject) &&
+                   pn_addr(pn->sobject) != pn_addr(obj))
                        continue;
 
                rval = sknode;
@@ -90,13 +110,44 @@ struct sock *pn_find_sock_by_sa(const struct sockaddr_pn *spn)
        spin_unlock_bh(&pnsocks.lock);
 
        return rval;
+}
+
+/* Deliver a broadcast packet (only in bottom-half) */
+void pn_deliver_sock_broadcast(struct net *net, struct sk_buff *skb)
+{
+       struct hlist_head *hlist = pnsocks.hlist;
+       unsigned h;
+
+       spin_lock(&pnsocks.lock);
+       for (h = 0; h < PN_HASHSIZE; h++) {
+               struct hlist_node *node;
+               struct sock *sknode;
+
+               sk_for_each(sknode, node, hlist) {
+                       struct sk_buff *clone;
 
+                       if (!net_eq(sock_net(sknode), net))
+                               continue;
+                       if (!sock_flag(sknode, SOCK_BROADCAST))
+                               continue;
+
+                       clone = skb_clone(skb, GFP_ATOMIC);
+                       if (clone) {
+                               sock_hold(sknode);
+                               sk_receive_skb(sknode, clone, 0);
+                       }
+               }
+               hlist++;
+       }
+       spin_unlock(&pnsocks.lock);
 }
 
 void pn_sock_hash(struct sock *sk)
 {
+       struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);
+
        spin_lock_bh(&pnsocks.lock);
-       sk_add_node(sk, &pnsocks.hlist);
+       sk_add_node(sk, hlist);
        spin_unlock_bh(&pnsocks.lock);
 }
 EXPORT_SYMBOL(pn_sock_hash);
@@ -109,6 +160,8 @@ void pn_sock_unhash(struct sock *sk)
 }
 EXPORT_SYMBOL(pn_sock_unhash);
 
+static DEFINE_MUTEX(port_mutex);
+
 static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
 {
        struct sock *sk = sock->sk;
@@ -128,7 +181,7 @@ static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
 
        handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
        saddr = pn_addr(handle);
-       if (saddr && phonet_address_lookup(saddr))
+       if (saddr && phonet_address_lookup(sock_net(sk), saddr))
                return -EADDRNOTAVAIL;
 
        lock_sock(sk);
@@ -136,9 +189,11 @@ static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
                err = -EINVAL; /* attempt to rebind */
                goto out;
        }
+       WARN_ON(sk_hashed(sk));
+       mutex_lock(&port_mutex);
        err = sk->sk_prot->get_port(sk, pn_port(handle));
        if (err)
-               goto out;
+               goto out_port;
 
        /* get_port() sets the port, bind() sets the address if applicable */
        pn->sobject = pn_object(saddr, pn_port(pn->sobject));
@@ -146,6 +201,8 @@ static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
 
        /* Enable RX on the socket */
        sk->sk_prot->hash(sk);
+out_port:
+       mutex_unlock(&port_mutex);
 out:
        release_sock(sk);
        return err;
@@ -166,6 +223,24 @@ static int pn_socket_autobind(struct socket *sock)
        return 0; /* socket was already bound */
 }
 
+static int pn_socket_accept(struct socket *sock, struct socket *newsock,
+                               int flags)
+{
+       struct sock *sk = sock->sk;
+       struct sock *newsk;
+       int err;
+
+       newsk = sk->sk_prot->accept(sk, flags, &err);
+       if (!newsk)
+               return err;
+
+       lock_sock(newsk);
+       sock_graft(newsk, newsock);
+       newsock->state = SS_CONNECTED;
+       release_sock(newsk);
+       return 0;
+}
+
 static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
                                int *sockaddr_len, int peer)
 {
@@ -182,6 +257,35 @@ static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
        return 0;
 }
 
+static unsigned int pn_socket_poll(struct file *file, struct socket *sock,
+                                       poll_table *wait)
+{
+       struct sock *sk = sock->sk;
+       struct pep_sock *pn = pep_sk(sk);
+       unsigned int mask = 0;
+
+       poll_wait(file, &sock->wait, wait);
+
+       switch (sk->sk_state) {
+       case TCP_LISTEN:
+               return hlist_empty(&pn->ackq) ? 0 : POLLIN;
+       case TCP_CLOSE:
+               return POLLERR;
+       }
+
+       if (!skb_queue_empty(&sk->sk_receive_queue))
+               mask |= POLLIN | POLLRDNORM;
+       if (!skb_queue_empty(&pn->ctrlreq_queue))
+               mask |= POLLPRI;
+       if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
+               return POLLHUP;
+
+       if (sk->sk_state == TCP_ESTABLISHED && atomic_read(&pn->tx_credits))
+               mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
+
+       return mask;
+}
+
 static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
                                unsigned long arg)
 {
@@ -220,6 +324,30 @@ static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
        return sk->sk_prot->ioctl(sk, cmd, arg);
 }
 
+static int pn_socket_listen(struct socket *sock, int backlog)
+{
+       struct sock *sk = sock->sk;
+       int err = 0;
+
+       if (sock->state != SS_UNCONNECTED)
+               return -EINVAL;
+       if (pn_socket_autobind(sock))
+               return -ENOBUFS;
+
+       lock_sock(sk);
+       if (sk->sk_state != TCP_CLOSE) {
+               err = -EINVAL;
+               goto out;
+       }
+
+       sk->sk_state = TCP_LISTEN;
+       sk->sk_ack_backlog = 0;
+       sk->sk_max_ack_backlog = backlog;
+out:
+       release_sock(sk);
+       return err;
+}
+
 static int pn_socket_sendmsg(struct kiocb *iocb, struct socket *sock,
                                struct msghdr *m, size_t total_len)
 {
@@ -256,21 +384,44 @@ const struct proto_ops phonet_dgram_ops = {
        .sendpage       = sock_no_sendpage,
 };
 
-static DEFINE_MUTEX(port_mutex);
+const struct proto_ops phonet_stream_ops = {
+       .family         = AF_PHONET,
+       .owner          = THIS_MODULE,
+       .release        = pn_socket_release,
+       .bind           = pn_socket_bind,
+       .connect        = sock_no_connect,
+       .socketpair     = sock_no_socketpair,
+       .accept         = pn_socket_accept,
+       .getname        = pn_socket_getname,
+       .poll           = pn_socket_poll,
+       .ioctl          = pn_socket_ioctl,
+       .listen         = pn_socket_listen,
+       .shutdown       = sock_no_shutdown,
+       .setsockopt     = sock_common_setsockopt,
+       .getsockopt     = sock_common_getsockopt,
+#ifdef CONFIG_COMPAT
+       .compat_setsockopt = compat_sock_common_setsockopt,
+       .compat_getsockopt = compat_sock_common_getsockopt,
+#endif
+       .sendmsg        = pn_socket_sendmsg,
+       .recvmsg        = sock_common_recvmsg,
+       .mmap           = sock_no_mmap,
+       .sendpage       = sock_no_sendpage,
+};
+EXPORT_SYMBOL(phonet_stream_ops);
 
 /* allocate port for a socket */
 int pn_sock_get_port(struct sock *sk, unsigned short sport)
 {
        static int port_cur;
+       struct net *net = sock_net(sk);
        struct pn_sock *pn = pn_sk(sk);
        struct sockaddr_pn try_sa;
        struct sock *tmpsk;
 
        memset(&try_sa, 0, sizeof(struct sockaddr_pn));
        try_sa.spn_family = AF_PHONET;
-
-       mutex_lock(&port_mutex);
-
+       WARN_ON(!mutex_is_locked(&port_mutex));
        if (!sport) {
                /* search free port */
                int port, pmin, pmax;
@@ -282,7 +433,7 @@ int pn_sock_get_port(struct sock *sk, unsigned short sport)
                                port_cur = pmin;
 
                        pn_sockaddr_set_port(&try_sa, port_cur);
-                       tmpsk = pn_find_sock_by_sa(&try_sa);
+                       tmpsk = pn_find_sock_by_sa(net, &try_sa);
                        if (tmpsk == NULL) {
                                sport = port_cur;
                                goto found;
@@ -292,21 +443,122 @@ int pn_sock_get_port(struct sock *sk, unsigned short sport)
        } else {
                /* try to find specific port */
                pn_sockaddr_set_port(&try_sa, sport);
-               tmpsk = pn_find_sock_by_sa(&try_sa);
+               tmpsk = pn_find_sock_by_sa(net, &try_sa);
                if (tmpsk == NULL)
                        /* No sock there! We can use that port... */
                        goto found;
                else
                        sock_put(tmpsk);
        }
-       mutex_unlock(&port_mutex);
-
        /* the port must be in use already */
        return -EADDRINUSE;
 
 found:
-       mutex_unlock(&port_mutex);
        pn->sobject = pn_object(pn_addr(pn->sobject), sport);
        return 0;
 }
 EXPORT_SYMBOL(pn_sock_get_port);
+
+#ifdef CONFIG_PROC_FS
+static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
+{
+       struct net *net = seq_file_net(seq);
+       struct hlist_head *hlist = pnsocks.hlist;
+       struct hlist_node *node;
+       struct sock *sknode;
+       unsigned h;
+
+       for (h = 0; h < PN_HASHSIZE; h++) {
+               sk_for_each(sknode, node, hlist) {
+                       if (!net_eq(net, sock_net(sknode)))
+                               continue;
+                       if (!pos)
+                               return sknode;
+                       pos--;
+               }
+               hlist++;
+       }
+       return NULL;
+}
+
+static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
+{
+       struct net *net = seq_file_net(seq);
+
+       do
+               sk = sk_next(sk);
+       while (sk && !net_eq(net, sock_net(sk)));
+
+       return sk;
+}
+
+static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
+       __acquires(pnsocks.lock)
+{
+       spin_lock_bh(&pnsocks.lock);
+       return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
+}
+
+static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
+{
+       struct sock *sk;
+
+       if (v == SEQ_START_TOKEN)
+               sk = pn_sock_get_idx(seq, 0);
+       else
+               sk = pn_sock_get_next(seq, v);
+       (*pos)++;
+       return sk;
+}
+
+static void pn_sock_seq_stop(struct seq_file *seq, void *v)
+       __releases(pnsocks.lock)
+{
+       spin_unlock_bh(&pnsocks.lock);
+}
+
+static int pn_sock_seq_show(struct seq_file *seq, void *v)
+{
+       int len;
+
+       if (v == SEQ_START_TOKEN)
+               seq_printf(seq, "%s%n", "pt  loc  rem rs st tx_queue rx_queue "
+                       "  uid inode ref pointer drops", &len);
+       else {
+               struct sock *sk = v;
+               struct pn_sock *pn = pn_sk(sk);
+
+               seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
+                       "%d %p %d%n",
+                       sk->sk_protocol, pn->sobject, 0, pn->resource,
+                       sk->sk_state,
+                       sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
+                       sock_i_uid(sk), sock_i_ino(sk),
+                       atomic_read(&sk->sk_refcnt), sk,
+                       atomic_read(&sk->sk_drops), &len);
+       }
+       seq_printf(seq, "%*s\n", 127 - len, "");
+       return 0;
+}
+
+static const struct seq_operations pn_sock_seq_ops = {
+       .start = pn_sock_seq_start,
+       .next = pn_sock_seq_next,
+       .stop = pn_sock_seq_stop,
+       .show = pn_sock_seq_show,
+};
+
+static int pn_sock_open(struct inode *inode, struct file *file)
+{
+       return seq_open_net(inode, file, &pn_sock_seq_ops,
+                               sizeof(struct seq_net_private));
+}
+
+const struct file_operations pn_sock_seq_fops = {
+       .owner = THIS_MODULE,
+       .open = pn_sock_open,
+       .read = seq_read,
+       .llseek = seq_lseek,
+       .release = seq_release_net,
+};
+#endif