net/unix/af_unix.c

   1 /*
   2  * NET4:        Implementation of BSD Unix domain sockets.
   3  *
   4  * Authors:     Alan Cox, <alan@lxorguk.ukuu.org.uk>
   5  *
   6  *              This program is free software; you can redistribute it and/or
   7  *              modify it under the terms of the GNU General Public License
   8  *              as published by the Free Software Foundation; either version
   9  *              2 of the License, or (at your option) any later version.
  10  *
  11  * Fixes:
  12  *              Linus Torvalds  :       Assorted bug cures.
  13  *              Niibe Yutaka    :       async I/O support.
  14  *              Carsten Paeth   :       PF_UNIX check, address fixes.
  15  *              Alan Cox        :       Limit size of allocated blocks.
  16  *              Alan Cox        :       Fixed the stupid socketpair bug.
  17  *              Alan Cox        :       BSD compatibility fine tuning.
  18  *              Alan Cox        :       Fixed a bug in connect when interrupted.
  19  *              Alan Cox        :       Sorted out a proper draft version of
  20  *                                      file descriptor passing hacked up from
  21  *                                      Mike Shaver's work.
  22  *              Marty Leisner   :       Fixes to fd passing
  23  *              Nick Nevin      :       recvmsg bugfix.
  24  *              Alan Cox        :       Started proper garbage collector
  25  *              Heiko EiBfeldt  :       Missing verify_area check
  26  *              Alan Cox        :       Started POSIXisms
  27  *              Andreas Schwab  :       Replace inode by dentry for proper
  28  *                                      reference counting
  29  *              Kirk Petersen   :       Made this a module
  30  *          Christoph Rohland   :       Elegant non-blocking accept/connect algorithm.
  31  *                                      Lots of bug fixes.
  32  *           Alexey Kuznetosv   :       Repaired (I hope) bugs introduces
  33  *                                      by above two patches.
  34  *           Andrea Arcangeli   :       If possible we block in connect(2)
  35  *                                      if the max backlog of the listen socket
  36  *                                      is been reached. This won't break
  37  *                                      old apps and it will avoid huge amount
  38  *                                      of socks hashed (this for unix_gc()
  39  *                                      performances reasons).
  40  *                                      Security fix that limits the max
  41  *                                      number of socks to 2*max_files and
  42  *                                      the number of skb queueable in the
  43  *                                      dgram receiver.
  44  *              Artur Skawina   :       Hash function optimizations
  45  *           Alexey Kuznetsov   :       Full scale SMP. Lot of bugs are introduced 8)
  46  *            Malcolm Beattie   :       Set peercred for socketpair
  47  *           Michal Ostrowski   :       Module initialization cleanup.
  48  *           Arnaldo C. Melo    :       Remove MOD_{INC,DEC}_USE_COUNT,
  49  *                                      the core infrastructure is doing that
  50  *                                      for all net proto families now (2.5.69+)
  51  *
  52  *
  53  * Known differences from reference BSD that was tested:
  54  *
  55  *      [TO FIX]
  56  *      ECONNREFUSED is not returned from one end of a connected() socket to the
  57  *              other the moment one end closes.
  58  *      fstat() doesn't return st_dev=0, and give the blksize as high water mark
  59  *              and a fake inode identifier (nor the BSD first socket fstat twice bug).
  60  *      [NOT TO FIX]
  61  *      accept() returns a path name even if the connecting socket has closed
  62  *              in the meantime (BSD loses the path and gives up).
  63  *      accept() returns 0 length path for an unbound connector. BSD returns 16
  64  *              and a null first byte in the path (but not for gethost/peername - BSD bug ??)
  65  *      socketpair(...SOCK_RAW..) doesn't panic the kernel.
  66  *      BSD af_unix apparently has connect forgetting to block properly.
  67  *              (need to check this with the POSIX spec in detail)
  68  *
  69  * Differences from 2.0.0-11-... (ANK)
  70  *      Bug fixes and improvements.
  71  *              - client shutdown killed server socket.
  72  *              - removed all useless cli/sti pairs.
  73  *
  74  *      Semantic changes/extensions.
  75  *              - generic control message passing.
  76  *              - SCM_CREDENTIALS control message.
  77  *              - "Abstract" (not FS based) socket bindings.
  78  *                Abstract names are sequences of bytes (not zero terminated)
  79  *                started by 0, so that this name space does not intersect
  80  *                with BSD names.
  81  */
  82
  83 #include <linux/module.h>
  84 #include <linux/kernel.h>
  85 #include <linux/signal.h>
  86 #include <linux/sched.h>
  87 #include <linux/errno.h>
  88 #include <linux/string.h>
  89 #include <linux/stat.h>
  90 #include <linux/dcache.h>
  91 #include <linux/namei.h>
  92 #include <linux/socket.h>
  93 #include <linux/un.h>
  94 #include <linux/fcntl.h>
  95 #include <linux/termios.h>
  96 #include <linux/sockios.h>
  97 #include <linux/net.h>
  98 #include <linux/in.h>
  99 #include <linux/fs.h>
 100 #include <linux/slab.h>
 101 #include <asm/uaccess.h>
 102 #include <linux/skbuff.h>
 103 #include <linux/netdevice.h>
 104 #include <net/net_namespace.h>
 105 #include <net/sock.h>
 106 #include <net/tcp_states.h>
 107 #include <net/af_unix.h>
 108 #include <linux/proc_fs.h>
 109 #include <linux/seq_file.h>
 110 #include <net/scm.h>
 111 #include <linux/init.h>
 112 #include <linux/poll.h>
 113 #include <linux/rtnetlink.h>
 114 #include <linux/mount.h>
 115 #include <net/checksum.h>
 116 #include <linux/security.h>
 117
 118 static struct hlist_head unix_socket_table[UNIX_HASH_SIZE + 1];
 119 static DEFINE_SPINLOCK(unix_table_lock);
 120 static atomic_long_t unix_nr_socks;
 121
 122 #define unix_sockets_unbound    (&unix_socket_table[UNIX_HASH_SIZE])
 123
 124 #define UNIX_ABSTRACT(sk)       (unix_sk(sk)->addr->hash != UNIX_HASH_SIZE)
 125
 126 #ifdef CONFIG_SECURITY_NETWORK
 127 static void unix_get_secdata(struct scm_cookie *scm, struct sk_buff *skb)
 128 {
 129         memcpy(UNIXSID(skb), &scm->secid, sizeof(u32));
 130 }
 131
 132 static inline void unix_set_secdata(struct scm_cookie *scm, struct sk_buff *skb)
 133 {
 134         scm->secid = *UNIXSID(skb);
 135 }
 136 #else
 137 static inline void unix_get_secdata(struct scm_cookie *scm, struct sk_buff *skb)
 138 { }
 139
 140 static inline void unix_set_secdata(struct scm_cookie *scm, struct sk_buff *skb)
 141 { }
 142 #endif /* CONFIG_SECURITY_NETWORK */
 143
 144 /*
 145  *  SMP locking strategy:
 146  *    hash table is protected with spinlock unix_table_lock
 147  *    each socket state is protected by separate spin lock.
 148  */
 149
 150 static inline unsigned unix_hash_fold(__wsum n)
 151 {
 152         unsigned int hash = (__force unsigned int)csum_fold(n);
 153
 154         hash ^= hash>>8;
 155         return hash&(UNIX_HASH_SIZE-1);
 156 }
 157
 158 #define unix_peer(sk) (unix_sk(sk)->peer)
 159
 160 static inline int unix_our_peer(struct sock *sk, struct sock *osk)
 161 {
 162         return unix_peer(osk) == sk;
 163 }
 164
 165 static inline int unix_may_send(struct sock *sk, struct sock *osk)
 166 {
 167         return unix_peer(osk) == NULL || unix_our_peer(sk, osk);
 168 }
 169
 170 static inline int unix_recvq_full(struct sock const *sk)
 171 {
 172         return skb_queue_len(&sk->sk_receive_queue) > sk->sk_max_ack_backlog;
 173 }
 174
 175 static struct sock *unix_peer_get(struct sock *s)
 176 {
 177         struct sock *peer;
 178
 179         unix_state_lock(s);
 180         peer = unix_peer(s);
 181         if (peer)
 182                 sock_hold(peer);
 183         unix_state_unlock(s);
 184         return peer;
 185 }
 186
 187 static inline void unix_release_addr(struct unix_address *addr)
 188 {
 189         if (atomic_dec_and_test(&addr->refcnt))
 190                 kfree(addr);
 191 }
 192
 193 /*
 194  *      Check unix socket name:
 195  *              - should be not zero length.
 196  *              - if started by not zero, should be NULL terminated (FS object)
 197  *              - if started by zero, it is abstract name.
 198  */
 199
 200 static int unix_mkname(struct sockaddr_un *sunaddr, int len, unsigned *hashp)
 201 {
 202         if (len <= sizeof(short) || len > sizeof(*sunaddr))
 203                 return -EINVAL;
 204         if (!sunaddr || sunaddr->sun_family != AF_UNIX)
 205                 return -EINVAL;
 206         if (sunaddr->sun_path[0]) {
 207                 /*
 208                  * This may look like an off by one error but it is a bit more
 209                  * subtle. 108 is the longest valid AF_UNIX path for a binding.
 210                  * sun_path[108] doesn't as such exist.  However in kernel space
 211                  * we are guaranteed that it is a valid memory location in our
 212                  * kernel address buffer.
 213                  */
 214                 ((char *)sunaddr)[len] = 0;
 215                 len = strlen(sunaddr->sun_path)+1+sizeof(short);
 216                 return len;
 217         }
 218
 219         *hashp = unix_hash_fold(csum_partial(sunaddr, len, 0));
 220         return len;
 221 }
 222
 223 static void __unix_remove_socket(struct sock *sk)
 224 {
 225         sk_del_node_init(sk);
 226 }
 227
 228 static void __unix_insert_socket(struct hlist_head *list, struct sock *sk)
 229 {
 230         WARN_ON(!sk_unhashed(sk));
 231         sk_add_node(sk, list);
 232 }
 233
 234 static inline void unix_remove_socket(struct sock *sk)
 235 {
 236         spin_lock(&unix_table_lock);
 237         __unix_remove_socket(sk);
 238         spin_unlock(&unix_table_lock);
 239 }
 240
 241 static inline void unix_insert_socket(struct hlist_head *list, struct sock *sk)
 242 {
 243         spin_lock(&unix_table_lock);
 244         __unix_insert_socket(list, sk);
 245         spin_unlock(&unix_table_lock);
 246 }
 247
 248 static struct sock *__unix_find_socket_byname(struct net *net,
 249                                               struct sockaddr_un *sunname,
 250                                               int len, int type, unsigned hash)
 251 {
 252         struct sock *s;
 253         struct hlist_node *node;
 254
 255         sk_for_each(s, node, &unix_socket_table[hash ^ type]) {
 256                 struct unix_sock *u = unix_sk(s);
 257
 258                 if (!net_eq(sock_net(s), net))
 259                         continue;
 260
 261                 if (u->addr->len == len &&
 262                     !memcmp(u->addr->name, sunname, len))
 263                         goto found;
 264         }
 265         s = NULL;
 266 found:
 267         return s;
 268 }
 269
 270 static inline struct sock *unix_find_socket_byname(struct net *net,
 271                                                    struct sockaddr_un *sunname,
 272                                                    int len, int type,
 273                                                    unsigned hash)
 274 {
 275         struct sock *s;
 276
 277         spin_lock(&unix_table_lock);
 278         s = __unix_find_socket_byname(net, sunname, len, type, hash);
 279         if (s)
 280                 sock_hold(s);
 281         spin_unlock(&unix_table_lock);
 282         return s;
 283 }
 284
 285 static struct sock *unix_find_socket_byinode(struct inode *i)
 286 {
 287         struct sock *s;
 288         struct hlist_node *node;
 289
 290         spin_lock(&unix_table_lock);
 291         sk_for_each(s, node,
 292                     &unix_socket_table[i->i_ino & (UNIX_HASH_SIZE - 1)]) {
 293                 struct dentry *dentry = unix_sk(s)->dentry;
 294
 295                 if (dentry && dentry->d_inode == i) {
 296                         sock_hold(s);
 297                         goto found;
 298                 }
 299         }
 300         s = NULL;
 301 found:
 302         spin_unlock(&unix_table_lock);
 303         return s;
 304 }
 305
 306 /* Support code for asymmetrically connected dgram sockets
 307  *
 308  * If a datagram socket is connected to a socket not itself connected
 309  * to the first socket (eg, /dev/log), clients may only enqueue more
 310  * messages if the present receive queue of the server socket is not
 311  * "too large". This means there's a second writeability condition
 312  * poll and sendmsg need to test. The dgram recv code will do a wake
 313  * up on the peer_wait wait queue of a socket upon reception of a
 314  * datagram which needs to be propagated to sleeping would-be writers
 315  * since these might not have sent anything so far. This can't be
 316  * accomplished via poll_wait because the lifetime of the server
 317  * socket might be less than that of its clients if these break their
 318  * association with it or if the server socket is closed while clients
 319  * are still connected to it and there's no way to inform "a polling
 320  * implementation" that it should let go of a certain wait queue
 321  *
 322  * In order to propagate a wake up, a wait_queue_t of the client
 323  * socket is enqueued on the peer_wait queue of the server socket
 324  * whose wake function does a wake_up on the ordinary client socket
 325  * wait queue. This connection is established whenever a write (or
 326  * poll for write) hit the flow control condition and broken when the
 327  * association to the server socket is dissolved or after a wake up
 328  * was relayed.
 329  */
 330
 331 static int unix_dgram_peer_wake_relay(wait_queue_t *q, unsigned mode, int flags,
 332                                       void *key)
 333 {
 334         struct unix_sock *u;
 335         wait_queue_head_t *u_sleep;
 336
 337         u = container_of(q, struct unix_sock, peer_wake);
 338
 339         __remove_wait_queue(&unix_sk(u->peer_wake.private)->peer_wait,
 340                             q);
 341         u->peer_wake.private = NULL;
 342
 343         /* relaying can only happen while the wq still exists */
 344         u_sleep = sk_sleep(&u->sk);
 345         if (u_sleep)
 346                 wake_up_interruptible_poll(u_sleep, key);
 347
 348         return 0;
 349 }
 350
 351 static int unix_dgram_peer_wake_connect(struct sock *sk, struct sock *other)
 352 {
 353         struct unix_sock *u, *u_other;
 354         int rc;
 355
 356         u = unix_sk(sk);
 357         u_other = unix_sk(other);
 358         rc = 0;
 359         spin_lock(&u_other->peer_wait.lock);
 360
 361         if (!u->peer_wake.private) {
 362                 u->peer_wake.private = other;
 363                 __add_wait_queue(&u_other->peer_wait, &u->peer_wake);
 364
 365                 rc = 1;
 366         }
 367
 368         spin_unlock(&u_other->peer_wait.lock);
 369         return rc;
 370 }
 371
 372 static void unix_dgram_peer_wake_disconnect(struct sock *sk,
 373                                             struct sock *other)
 374 {
 375         struct unix_sock *u, *u_other;
 376
 377         u = unix_sk(sk);
 378         u_other = unix_sk(other);
 379         spin_lock(&u_other->peer_wait.lock);
 380
 381         if (u->peer_wake.private == other) {
 382                 __remove_wait_queue(&u_other->peer_wait, &u->peer_wake);
 383                 u->peer_wake.private = NULL;
 384         }
 385
 386         spin_unlock(&u_other->peer_wait.lock);
 387 }
 388
 389 static void unix_dgram_peer_wake_disconnect_wakeup(struct sock *sk,
 390                                                    struct sock *other)
 391 {
 392         unix_dgram_peer_wake_disconnect(sk, other);
 393         wake_up_interruptible_poll(sk_sleep(sk),
 394                                    POLLOUT |
 395                                    POLLWRNORM |
 396                                    POLLWRBAND);
 397 }
 398
 399 /* preconditions:
 400  *      - unix_peer(sk) == other
 401  *      - association is stable
 402  */
 403 static int unix_dgram_peer_wake_me(struct sock *sk, struct sock *other)
 404 {
 405         int connected;
 406
 407         connected = unix_dgram_peer_wake_connect(sk, other);
 408
 409         if (unix_recvq_full(other))
 410                 return 1;
 411
 412         if (connected)
 413                 unix_dgram_peer_wake_disconnect(sk, other);
 414
 415         return 0;
 416 }
 417
 418 static inline int unix_writable(struct sock *sk)
 419 {
 420         return (atomic_read(&sk->sk_wmem_alloc) << 2) <= sk->sk_sndbuf;
 421 }
 422
 423 static void unix_write_space(struct sock *sk)
 424 {
 425         struct socket_wq *wq;
 426
 427         rcu_read_lock();
 428         if (unix_writable(sk)) {
 429                 wq = rcu_dereference(sk->sk_wq);
 430                 if (wq_has_sleeper(wq))
 431                         wake_up_interruptible_sync_poll(&wq->wait,
 432                                 POLLOUT | POLLWRNORM | POLLWRBAND);
 433                 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
 434         }
 435         rcu_read_unlock();
 436 }
 437
 438 /* When dgram socket disconnects (or changes its peer), we clear its receive
 439  * queue of packets arrived from previous peer. First, it allows to do
 440  * flow control based only on wmem_alloc; second, sk connected to peer
 441  * may receive messages only from that peer. */
 442 static void unix_dgram_disconnected(struct sock *sk, struct sock *other)
 443 {
 444         if (!skb_queue_empty(&sk->sk_receive_queue)) {
 445                 skb_queue_purge(&sk->sk_receive_queue);
 446                 wake_up_interruptible_all(&unix_sk(sk)->peer_wait);
 447
 448                 /* If one link of bidirectional dgram pipe is disconnected,
 449                  * we signal error. Messages are lost. Do not make this,
 450                  * when peer was not connected to us.
 451                  */
 452                 if (!sock_flag(other, SOCK_DEAD) && unix_peer(other) == sk) {
 453                         other->sk_err = ECONNRESET;
 454                         other->sk_error_report(other);
 455                 }
 456         }
 457 }
 458
 459 static void unix_sock_destructor(struct sock *sk)
 460 {
 461         struct unix_sock *u = unix_sk(sk);
 462
 463         skb_queue_purge(&sk->sk_receive_queue);
 464
 465         WARN_ON(atomic_read(&sk->sk_wmem_alloc));
 466         WARN_ON(!sk_unhashed(sk));
 467         WARN_ON(sk->sk_socket);
 468         if (!sock_flag(sk, SOCK_DEAD)) {
 469                 printk(KERN_INFO "Attempt to release alive unix socket: %p\n", sk);
 470                 return;
 471         }
 472
 473         if (u->addr)
 474                 unix_release_addr(u->addr);
 475
 476         atomic_long_dec(&unix_nr_socks);
 477         local_bh_disable();
 478         sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
 479         local_bh_enable();
 480 #ifdef UNIX_REFCNT_DEBUG
 481         printk(KERN_DEBUG "UNIX %p is destroyed, %ld are still alive.\n", sk,
 482                 atomic_long_read(&unix_nr_socks));
 483 #endif
 484 }
 485
 486 static void unix_release_sock(struct sock *sk, int embrion)
 487 {
 488         struct unix_sock *u = unix_sk(sk);
 489         struct dentry *dentry;
 490         struct vfsmount *mnt;
 491         struct sock *skpair;
 492         struct sk_buff *skb;
 493         int state;
 494
 495         unix_remove_socket(sk);
 496
 497         /* Clear state */
 498         unix_state_lock(sk);
 499         sock_orphan(sk);
 500         sk->sk_shutdown = SHUTDOWN_MASK;
 501         dentry       = u->dentry;
 502         u->dentry    = NULL;
 503         mnt          = u->mnt;
 504         u->mnt       = NULL;
 505         state = sk->sk_state;
 506         sk->sk_state = TCP_CLOSE;
 507         unix_state_unlock(sk);
 508
 509         wake_up_interruptible_all(&u->peer_wait);
 510
 511         skpair = unix_peer(sk);
 512
 513         if (skpair != NULL) {
 514                 if (sk->sk_type == SOCK_STREAM || sk->sk_type == SOCK_SEQPACKET) {
 515                         unix_state_lock(skpair);
 516                         /* No more writes */
 517                         skpair->sk_shutdown = SHUTDOWN_MASK;
 518                         if (!skb_queue_empty(&sk->sk_receive_queue) || embrion)
 519                                 skpair->sk_err = ECONNRESET;
 520                         unix_state_unlock(skpair);
 521                         skpair->sk_state_change(skpair);
 522                         sk_wake_async(skpair, SOCK_WAKE_WAITD, POLL_HUP);
 523                 }
 524
 525                 unix_dgram_peer_wake_disconnect(sk, skpair);
 526                 sock_put(skpair); /* It may now die */
 527                 unix_peer(sk) = NULL;
 528         }
 529
 530         /* Try to flush out this socket. Throw out buffers at least */
 531
 532         while ((skb = skb_dequeue(&sk->sk_receive_queue)) != NULL) {
 533                 if (state == TCP_LISTEN)
 534                         unix_release_sock(skb->sk, 1);
 535                 /* passed fds are erased in the kfree_skb hook        */
 536                 kfree_skb(skb);
 537         }
 538
 539         if (dentry) {
 540                 dput(dentry);
 541                 mntput(mnt);
 542         }
 543
 544         sock_put(sk);
 545
 546         /* ---- Socket is dead now and most probably destroyed ---- */
 547
 548         /*
 549          * Fixme: BSD difference: In BSD all sockets connected to use get
 550          *        ECONNRESET and we die on the spot. In Linux we behave
 551          *        like files and pipes do and wait for the last
 552          *        dereference.
 553          *
 554          * Can't we simply set sock->err?
 555          *
 556          *        What the above comment does talk about? --ANK(980817)
 557          */
 558
 559         if (unix_tot_inflight)
 560                 unix_gc();              /* Garbage collect fds */
 561 }
 562
 563 static void init_peercred(struct sock *sk)
 564 {
 565         put_pid(sk->sk_peer_pid);
 566         if (sk->sk_peer_cred)
 567                 put_cred(sk->sk_peer_cred);
 568         sk->sk_peer_pid  = get_pid(task_tgid(current));
 569         sk->sk_peer_cred = get_current_cred();
 570 }
 571
 572 static void copy_peercred(struct sock *sk, struct sock *peersk)
 573 {
 574         put_pid(sk->sk_peer_pid);
 575         if (sk->sk_peer_cred)
 576                 put_cred(sk->sk_peer_cred);
 577         sk->sk_peer_pid  = get_pid(peersk->sk_peer_pid);
 578         sk->sk_peer_cred = get_cred(peersk->sk_peer_cred);
 579 }
 580
 581 static int unix_listen(struct socket *sock, int backlog)
 582 {
 583         int err;
 584         struct sock *sk = sock->sk;
 585         struct unix_sock *u = unix_sk(sk);
 586         struct pid *old_pid = NULL;
 587         const struct cred *old_cred = NULL;
 588
 589         err = -EOPNOTSUPP;
 590         if (sock->type != SOCK_STREAM && sock->type != SOCK_SEQPACKET)
 591                 goto out;       /* Only stream/seqpacket sockets accept */
 592         err = -EINVAL;
 593         if (!u->addr)
 594                 goto out;       /* No listens on an unbound socket */
 595         unix_state_lock(sk);
 596         if (sk->sk_state != TCP_CLOSE && sk->sk_state != TCP_LISTEN)
 597                 goto out_unlock;
 598         if (backlog > sk->sk_max_ack_backlog)
 599                 wake_up_interruptible_all(&u->peer_wait);
 600         sk->sk_max_ack_backlog  = backlog;
 601         sk->sk_state            = TCP_LISTEN;
 602         /* set credentials so connect can copy them */
 603         init_peercred(sk);
 604         err = 0;
 605
 606 out_unlock:
 607         unix_state_unlock(sk);
 608         put_pid(old_pid);
 609         if (old_cred)
 610                 put_cred(old_cred);
 611 out:
 612         return err;
 613 }
 614
 615 static int unix_release(struct socket *);
 616 static int unix_bind(struct socket *, struct sockaddr *, int);
 617 static int unix_stream_connect(struct socket *, struct sockaddr *,
 618                                int addr_len, int flags);
 619 static int unix_socketpair(struct socket *, struct socket *);
 620 static int unix_accept(struct socket *, struct socket *, int);
 621 static int unix_getname(struct socket *, struct sockaddr *, int *, int);
 622 static unsigned int unix_poll(struct file *, struct socket *, poll_table *);
 623 static unsigned int unix_dgram_poll(struct file *, struct socket *,
 624                                     poll_table *);
 625 static int unix_ioctl(struct socket *, unsigned int, unsigned long);
 626 static int unix_shutdown(struct socket *, int);
 627 static int unix_stream_sendmsg(struct kiocb *, struct socket *,
 628                                struct msghdr *, size_t);
 629 static int unix_stream_recvmsg(struct kiocb *, struct socket *,
 630                                struct msghdr *, size_t, int);
 631 static int unix_dgram_sendmsg(struct kiocb *, struct socket *,
 632                               struct msghdr *, size_t);
 633 static int unix_dgram_recvmsg(struct kiocb *, struct socket *,
 634                               struct msghdr *, size_t, int);
 635 static int unix_dgram_connect(struct socket *, struct sockaddr *,
 636                               int, int);
 637 static int unix_seqpacket_sendmsg(struct kiocb *, struct socket *,
 638                                   struct msghdr *, size_t);
 639 static int unix_seqpacket_recvmsg(struct kiocb *, struct socket *,
 640                                   struct msghdr *, size_t, int);
 641
 642 static const struct proto_ops unix_stream_ops = {
 643         .family =       PF_UNIX,
 644         .owner =        THIS_MODULE,
 645         .release =      unix_release,
 646         .bind =         unix_bind,
 647         .connect =      unix_stream_connect,
 648         .socketpair =   unix_socketpair,
 649         .accept =       unix_accept,
 650         .getname =      unix_getname,
 651         .poll =         unix_poll,
 652         .ioctl =        unix_ioctl,
 653         .listen =       unix_listen,
 654         .shutdown =     unix_shutdown,
 655         .setsockopt =   sock_no_setsockopt,
 656         .getsockopt =   sock_no_getsockopt,
 657         .sendmsg =      unix_stream_sendmsg,
 658         .recvmsg =      unix_stream_recvmsg,
 659         .mmap =         sock_no_mmap,
 660         .sendpage =     sock_no_sendpage,
 661 };
 662
 663 static const struct proto_ops unix_dgram_ops = {
 664         .family =       PF_UNIX,
 665         .owner =        THIS_MODULE,
 666         .release =      unix_release,
 667         .bind =         unix_bind,
 668         .connect =      unix_dgram_connect,
 669         .socketpair =   unix_socketpair,
 670         .accept =       sock_no_accept,
 671         .getname =      unix_getname,
 672         .poll =         unix_dgram_poll,
 673         .ioctl =        unix_ioctl,
 674         .listen =       sock_no_listen,
 675         .shutdown =     unix_shutdown,
 676         .setsockopt =   sock_no_setsockopt,
 677         .getsockopt =   sock_no_getsockopt,
 678         .sendmsg =      unix_dgram_sendmsg,
 679         .recvmsg =      unix_dgram_recvmsg,
 680         .mmap =         sock_no_mmap,
 681         .sendpage =     sock_no_sendpage,
 682 };
 683
 684 static const struct proto_ops unix_seqpacket_ops = {
 685         .family =       PF_UNIX,
 686         .owner =        THIS_MODULE,
 687         .release =      unix_release,
 688         .bind =         unix_bind,
 689         .connect =      unix_stream_connect,
 690         .socketpair =   unix_socketpair,
 691         .accept =       unix_accept,
 692         .getname =      unix_getname,
 693         .poll =         unix_dgram_poll,
 694         .ioctl =        unix_ioctl,
 695         .listen =       unix_listen,
 696         .shutdown =     unix_shutdown,
 697         .setsockopt =   sock_no_setsockopt,
 698         .getsockopt =   sock_no_getsockopt,
 699         .sendmsg =      unix_seqpacket_sendmsg,
 700         .recvmsg =      unix_seqpacket_recvmsg,
 701         .mmap =         sock_no_mmap,
 702         .sendpage =     sock_no_sendpage,
 703 };
 704
 705 static struct proto unix_proto = {
 706         .name                   = "UNIX",
 707         .owner                  = THIS_MODULE,
 708         .obj_size               = sizeof(struct unix_sock),
 709 };
 710
 711 /*
 712  * AF_UNIX sockets do not interact with hardware, hence they
 713  * dont trigger interrupts - so it's safe for them to have
 714  * bh-unsafe locking for their sk_receive_queue.lock. Split off
 715  * this special lock-class by reinitializing the spinlock key:
 716  */
 717 static struct lock_class_key af_unix_sk_receive_queue_lock_key;
 718
 719 static struct sock *unix_create1(struct net *net, struct socket *sock)
 720 {
 721         struct sock *sk = NULL;
 722         struct unix_sock *u;
 723
 724         atomic_long_inc(&unix_nr_socks);
 725         if (atomic_long_read(&unix_nr_socks) > 2 * get_max_files())
 726                 goto out;
 727
 728         sk = sk_alloc(net, PF_UNIX, GFP_KERNEL, &unix_proto);
 729         if (!sk)
 730                 goto out;
 731
 732         sock_init_data(sock, sk);
 733         lockdep_set_class(&sk->sk_receive_queue.lock,
 734                                 &af_unix_sk_receive_queue_lock_key);
 735
 736         sk->sk_write_space      = unix_write_space;
 737         sk->sk_max_ack_backlog  = net->unx.sysctl_max_dgram_qlen;
 738         sk->sk_destruct         = unix_sock_destructor;
 739         u         = unix_sk(sk);
 740         u->dentry = NULL;
 741         u->mnt    = NULL;
 742         spin_lock_init(&u->lock);
 743         atomic_long_set(&u->inflight, 0);
 744         INIT_LIST_HEAD(&u->link);
 745         mutex_init(&u->readlock); /* single task reading lock */
 746         init_waitqueue_head(&u->peer_wait);
 747         init_waitqueue_func_entry(&u->peer_wake, unix_dgram_peer_wake_relay);
 748         unix_insert_socket(unix_sockets_unbound, sk);
 749 out:
 750         if (sk == NULL)
 751                 atomic_long_dec(&unix_nr_socks);
 752         else {
 753                 local_bh_disable();
 754                 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
 755                 local_bh_enable();
 756         }
 757         return sk;
 758 }
 759
 760 static int unix_create(struct net *net, struct socket *sock, int protocol,
 761                        int kern)
 762 {
 763         if (protocol && protocol != PF_UNIX)
 764                 return -EPROTONOSUPPORT;
 765
 766         sock->state = SS_UNCONNECTED;
 767
 768         switch (sock->type) {
 769         case SOCK_STREAM:
 770                 sock->ops = &unix_stream_ops;
 771                 break;
 772                 /*
 773                  *      Believe it or not BSD has AF_UNIX, SOCK_RAW though
 774                  *      nothing uses it.
 775                  */
 776         case SOCK_RAW:
 777                 sock->type = SOCK_DGRAM;
 778         case SOCK_DGRAM:
 779                 sock->ops = &unix_dgram_ops;
 780                 break;
 781         case SOCK_SEQPACKET:
 782                 sock->ops = &unix_seqpacket_ops;
 783                 break;
 784         default:
 785                 return -ESOCKTNOSUPPORT;
 786         }
 787
 788         return unix_create1(net, sock) ? 0 : -ENOMEM;
 789 }
 790
 791 static int unix_release(struct socket *sock)
 792 {
 793         struct sock *sk = sock->sk;
 794
 795         if (!sk)
 796                 return 0;
 797
 798         unix_release_sock(sk, 0);
 799         sock->sk = NULL;
 800
 801         return 0;
 802 }
 803
 804 static int unix_autobind(struct socket *sock)
 805 {
 806         struct sock *sk = sock->sk;
 807         struct net *net = sock_net(sk);
 808         struct unix_sock *u = unix_sk(sk);
 809         static u32 ordernum = 1;
 810         struct unix_address *addr;
 811         int err;
 812         unsigned int retries = 0;
 813
 814         err = mutex_lock_interruptible(&u->readlock);
 815         if (err)
 816                 return err;
 817
 818         err = 0;
 819         if (u->addr)
 820                 goto out;
 821
 822         err = -ENOMEM;
 823         addr = kzalloc(sizeof(*addr) + sizeof(short) + 16, GFP_KERNEL);
 824         if (!addr)
 825                 goto out;
 826
 827         addr->name->sun_family = AF_UNIX;
 828         atomic_set(&addr->refcnt, 1);
 829
 830 retry:
 831         addr->len = sprintf(addr->name->sun_path+1, "%05x", ordernum) + 1 + sizeof(short);
 832         addr->hash = unix_hash_fold(csum_partial(addr->name, addr->len, 0));
 833
 834         spin_lock(&unix_table_lock);
 835         ordernum = (ordernum+1)&0xFFFFF;
 836
 837         if (__unix_find_socket_byname(net, addr->name, addr->len, sock->type,
 838                                       addr->hash)) {
 839                 spin_unlock(&unix_table_lock);
 840                 /*
 841                  * __unix_find_socket_byname() may take long time if many names
 842                  * are already in use.
 843                  */
 844                 cond_resched();
 845                 /* Give up if all names seems to be in use. */
 846                 if (retries++ == 0xFFFFF) {
 847                         err = -ENOSPC;
 848                         kfree(addr);
 849                         goto out;
 850                 }
 851                 goto retry;
 852         }
 853         addr->hash ^= sk->sk_type;
 854
 855         __unix_remove_socket(sk);
 856         u->addr = addr;
 857         __unix_insert_socket(&unix_socket_table[addr->hash], sk);
 858         spin_unlock(&unix_table_lock);
 859         err = 0;
 860
 861 out:    mutex_unlock(&u->readlock);
 862         return err;
 863 }
 864
 865 static struct sock *unix_find_other(struct net *net,
 866                                     struct sockaddr_un *sunname, int len,
 867                                     int type, unsigned hash, int *error)
 868 {
 869         struct sock *u;
 870         struct path path;
 871         int err = 0;
 872
 873         if (sunname->sun_path[0]) {
 874                 struct inode *inode;
 875                 err = kern_path(sunname->sun_path, LOOKUP_FOLLOW, &path);
 876                 if (err)
 877                         goto fail;
 878                 inode = path.dentry->d_inode;
 879                 err = inode_permission(inode, MAY_WRITE);
 880                 if (err)
 881                         goto put_fail;
 882
 883                 err = -ECONNREFUSED;
 884                 if (!S_ISSOCK(inode->i_mode))
 885                         goto put_fail;
 886                 u = unix_find_socket_byinode(inode);
 887                 if (!u)
 888                         goto put_fail;
 889
 890                 if (u->sk_type == type)
 891                         touch_atime(path.mnt, path.dentry);
 892
 893                 path_put(&path);
 894
 895                 err = -EPROTOTYPE;
 896                 if (u->sk_type != type) {
 897                         sock_put(u);
 898                         goto fail;
 899                 }
 900         } else {
 901                 err = -ECONNREFUSED;
 902                 u = unix_find_socket_byname(net, sunname, len, type, hash);
 903                 if (u) {
 904                         struct dentry *dentry;
 905                         dentry = unix_sk(u)->dentry;
 906                         if (dentry)
 907                                 touch_atime(unix_sk(u)->mnt, dentry);
 908                 } else
 909                         goto fail;
 910         }
 911         return u;
 912
 913 put_fail:
 914         path_put(&path);
 915 fail:
 916         *error = err;
 917         return NULL;
 918 }
 919
 920
 921 static int unix_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len)
 922 {
 923         struct sock *sk = sock->sk;
 924         struct net *net = sock_net(sk);
 925         struct unix_sock *u = unix_sk(sk);
 926         struct sockaddr_un *sunaddr = (struct sockaddr_un *)uaddr;
 927         char *sun_path = sunaddr->sun_path;
 928         struct dentry *dentry = NULL;
 929         struct path path;
 930         int err;
 931         unsigned hash;
 932         struct unix_address *addr;
 933         struct hlist_head *list;
 934
 935         err = -EINVAL;
 936         if (sunaddr->sun_family != AF_UNIX)
 937                 goto out;
 938
 939         if (addr_len == sizeof(short)) {
 940                 err = unix_autobind(sock);
 941                 goto out;
 942         }
 943
 944         err = unix_mkname(sunaddr, addr_len, &hash);
 945         if (err < 0)
 946                 goto out;
 947         addr_len = err;
 948
 949         err = mutex_lock_interruptible(&u->readlock);
 950         if (err)
 951                 goto out;
 952
 953         err = -EINVAL;
 954         if (u->addr)
 955                 goto out_up;
 956
 957         err = -ENOMEM;
 958         addr = kmalloc(sizeof(*addr)+addr_len, GFP_KERNEL);
 959         if (!addr)
 960                 goto out_up;
 961
 962         memcpy(addr->name, sunaddr, addr_len);
 963         addr->len = addr_len;
 964         addr->hash = hash ^ sk->sk_type;
 965         atomic_set(&addr->refcnt, 1);
 966
 967         if (sun_path[0]) {
 968                 unsigned int mode;
 969                 err = 0;
 970                 /*
 971                  * Get the parent directory, calculate the hash for last
 972                  * component.
 973                  */
 974                 dentry = kern_path_create(AT_FDCWD, sun_path, &path, 0);
 975                 err = PTR_ERR(dentry);
 976                 if (IS_ERR(dentry))
 977                         goto out_mknod_parent;
 978
 979                 /*
 980                  * All right, let's create it.
 981                  */
 982                 mode = S_IFSOCK |
 983                        (SOCK_INODE(sock)->i_mode & ~current_umask());
 984                 err = mnt_want_write(path.mnt);
 985                 if (err)
 986                         goto out_mknod_dput;
 987                 err = security_path_mknod(&path, dentry, mode, 0);
 988                 if (err)
 989                         goto out_mknod_drop_write;
 990                 err = vfs_mknod(path.dentry->d_inode, dentry, mode, 0);
 991 out_mknod_drop_write:
 992                 mnt_drop_write(path.mnt);
 993                 if (err)
 994                         goto out_mknod_dput;
 995                 mutex_unlock(&path.dentry->d_inode->i_mutex);
 996                 dput(path.dentry);
 997                 path.dentry = dentry;
 998
 999                 addr->hash = UNIX_HASH_SIZE;
1000         }
1001
1002         spin_lock(&unix_table_lock);
1003
1004         if (!sun_path[0]) {
1005                 err = -EADDRINUSE;
1006                 if (__unix_find_socket_byname(net, sunaddr, addr_len,
1007                                               sk->sk_type, hash)) {
1008                         unix_release_addr(addr);
1009                         goto out_unlock;
1010                 }
1011
1012                 list = &unix_socket_table[addr->hash];
1013         } else {
1014                 list = &unix_socket_table[dentry->d_inode->i_ino & (UNIX_HASH_SIZE-1)];
1015                 u->dentry = path.dentry;
1016                 u->mnt    = path.mnt;
1017         }
1018
1019         err = 0;
1020         __unix_remove_socket(sk);
1021         u->addr = addr;
1022         __unix_insert_socket(list, sk);
1023
1024 out_unlock:
1025         spin_unlock(&unix_table_lock);
1026 out_up:
1027         mutex_unlock(&u->readlock);
1028 out:
1029         return err;
1030
1031 out_mknod_dput:
1032         dput(dentry);
1033         mutex_unlock(&path.dentry->d_inode->i_mutex);
1034         path_put(&path);
1035 out_mknod_parent:
1036         if (err == -EEXIST)
1037                 err = -EADDRINUSE;
1038         unix_release_addr(addr);
1039         goto out_up;
1040 }
1041
1042 static void unix_state_double_lock(struct sock *sk1, struct sock *sk2)
1043 {
1044         if (unlikely(sk1 == sk2) || !sk2) {
1045                 unix_state_lock(sk1);
1046                 return;
1047         }
1048         if (sk1 < sk2) {
1049                 unix_state_lock(sk1);
1050                 unix_state_lock_nested(sk2);
1051         } else {
1052                 unix_state_lock(sk2);
1053                 unix_state_lock_nested(sk1);
1054         }
1055 }
1056
1057 static void unix_state_double_unlock(struct sock *sk1, struct sock *sk2)
1058 {
1059         if (unlikely(sk1 == sk2) || !sk2) {
1060                 unix_state_unlock(sk1);
1061                 return;
1062         }
1063         unix_state_unlock(sk1);
1064         unix_state_unlock(sk2);
1065 }
1066
1067 static int unix_dgram_connect(struct socket *sock, struct sockaddr *addr,
1068                               int alen, int flags)
1069 {
1070         struct sock *sk = sock->sk;
1071         struct net *net = sock_net(sk);
1072         struct sockaddr_un *sunaddr = (struct sockaddr_un *)addr;
1073         struct sock *other;
1074         unsigned hash;
1075         int err;
1076
1077         if (addr->sa_family != AF_UNSPEC) {
1078                 err = unix_mkname(sunaddr, alen, &hash);
1079                 if (err < 0)
1080                         goto out;
1081                 alen = err;
1082
1083                 if (test_bit(SOCK_PASSCRED, &sock->flags) &&
1084                     !unix_sk(sk)->addr && (err = unix_autobind(sock)) != 0)
1085                         goto out;
1086
1087 restart:
1088                 other = unix_find_other(net, sunaddr, alen, sock->type, hash, &err);
1089                 if (!other)
1090                         goto out;
1091
1092                 unix_state_double_lock(sk, other);
1093
1094                 /* Apparently VFS overslept socket death. Retry. */
1095                 if (sock_flag(other, SOCK_DEAD)) {
1096                         unix_state_double_unlock(sk, other);
1097                         sock_put(other);
1098                         goto restart;
1099                 }
1100
1101                 err = -EPERM;
1102                 if (!unix_may_send(sk, other))
1103                         goto out_unlock;
1104
1105                 err = security_unix_may_send(sk->sk_socket, other->sk_socket);
1106                 if (err)
1107                         goto out_unlock;
1108
1109         } else {
1110                 /*
1111                  *      1003.1g breaking connected state with AF_UNSPEC
1112                  */
1113                 other = NULL;
1114                 unix_state_double_lock(sk, other);
1115         }
1116
1117         /*
1118          * If it was connected, reconnect.
1119          */
1120         if (unix_peer(sk)) {
1121                 struct sock *old_peer = unix_peer(sk);
1122                 unix_peer(sk) = other;
1123                 unix_dgram_peer_wake_disconnect_wakeup(sk, old_peer);
1124
1125                 unix_state_double_unlock(sk, other);
1126
1127                 if (other != old_peer)
1128                         unix_dgram_disconnected(sk, old_peer);
1129                 sock_put(old_peer);
1130         } else {
1131                 unix_peer(sk) = other;
1132                 unix_state_double_unlock(sk, other);
1133         }
1134         return 0;
1135
1136 out_unlock:
1137         unix_state_double_unlock(sk, other);
1138         sock_put(other);
1139 out:
1140         return err;
1141 }
1142
1143 static long unix_wait_for_peer(struct sock *other, long timeo)
1144 {
1145         struct unix_sock *u = unix_sk(other);
1146         int sched;
1147         DEFINE_WAIT(wait);
1148
1149         prepare_to_wait_exclusive(&u->peer_wait, &wait, TASK_INTERRUPTIBLE);
1150
1151         sched = !sock_flag(other, SOCK_DEAD) &&
1152                 !(other->sk_shutdown & RCV_SHUTDOWN) &&
1153                 unix_recvq_full(other);
1154
1155         unix_state_unlock(other);
1156
1157         if (sched)
1158                 timeo = schedule_timeout(timeo);
1159
1160         finish_wait(&u->peer_wait, &wait);
1161         return timeo;
1162 }
1163
1164 static int unix_stream_connect(struct socket *sock, struct sockaddr *uaddr,
1165                                int addr_len, int flags)
1166 {
1167         struct sockaddr_un *sunaddr = (struct sockaddr_un *)uaddr;
1168         struct sock *sk = sock->sk;
1169         struct net *net = sock_net(sk);
1170         struct unix_sock *u = unix_sk(sk), *newu, *otheru;
1171         struct sock *newsk = NULL;
1172         struct sock *other = NULL;
1173         struct sk_buff *skb = NULL;
1174         unsigned hash;
1175         int st;
1176         int err;
1177         long timeo;
1178
1179         err = unix_mkname(sunaddr, addr_len, &hash);
1180         if (err < 0)
1181                 goto out;
1182         addr_len = err;
1183
1184         if (test_bit(SOCK_PASSCRED, &sock->flags) && !u->addr &&
1185             (err = unix_autobind(sock)) != 0)
1186                 goto out;
1187
1188         timeo = sock_sndtimeo(sk, flags & O_NONBLOCK);
1189
1190         /* First of all allocate resources.
1191            If we will make it after state is locked,
1192            we will have to recheck all again in any case.
1193          */
1194
1195         err = -ENOMEM;
1196
1197         /* create new sock for complete connection */
1198         newsk = unix_create1(sock_net(sk), NULL);
1199         if (newsk == NULL)
1200                 goto out;
1201
1202         /* Allocate skb for sending to listening sock */
1203         skb = sock_wmalloc(newsk, 1, 0, GFP_KERNEL);
1204         if (skb == NULL)
1205                 goto out;
1206
1207 restart:
1208         /*  Find listening sock. */
1209         other = unix_find_other(net, sunaddr, addr_len, sk->sk_type, hash, &err);
1210         if (!other)
1211                 goto out;
1212
1213         /* Latch state of peer */
1214         unix_state_lock(other);
1215
1216         /* Apparently VFS overslept socket death. Retry. */
1217         if (sock_flag(other, SOCK_DEAD)) {
1218                 unix_state_unlock(other);
1219                 sock_put(other);
1220                 goto restart;
1221         }
1222
1223         err = -ECONNREFUSED;
1224         if (other->sk_state != TCP_LISTEN)
1225                 goto out_unlock;
1226         if (other->sk_shutdown & RCV_SHUTDOWN)
1227                 goto out_unlock;
1228
1229         if (unix_recvq_full(other)) {
1230                 err = -EAGAIN;
1231                 if (!timeo)
1232                         goto out_unlock;
1233
1234                 timeo = unix_wait_for_peer(other, timeo);
1235
1236                 err = sock_intr_errno(timeo);
1237                 if (signal_pending(current))
1238                         goto out;
1239                 sock_put(other);
1240                 goto restart;
1241         }
1242
1243         /* Latch our state.
1244
1245            It is tricky place. We need to grab our state lock and cannot
1246            drop lock on peer. It is dangerous because deadlock is
1247            possible. Connect to self case and simultaneous
1248            attempt to connect are eliminated by checking socket
1249            state. other is TCP_LISTEN, if sk is TCP_LISTEN we
1250            check this before attempt to grab lock.
1251
1252            Well, and we have to recheck the state after socket locked.
1253          */
1254         st = sk->sk_state;
1255
1256         switch (st) {
1257         case TCP_CLOSE:
1258                 /* This is ok... continue with connect */
1259                 break;
1260         case TCP_ESTABLISHED:
1261                 /* Socket is already connected */
1262                 err = -EISCONN;
1263                 goto out_unlock;
1264         default:
1265                 err = -EINVAL;
1266                 goto out_unlock;
1267         }
1268
1269         unix_state_lock_nested(sk);
1270
1271         if (sk->sk_state != st) {
1272                 unix_state_unlock(sk);
1273                 unix_state_unlock(other);
1274                 sock_put(other);
1275                 goto restart;
1276         }
1277
1278         err = security_unix_stream_connect(sk, other, newsk);
1279         if (err) {
1280                 unix_state_unlock(sk);
1281                 goto out_unlock;
1282         }
1283
1284         /* The way is open! Fastly set all the necessary fields... */
1285
1286         sock_hold(sk);
1287         unix_peer(newsk)        = sk;
1288         newsk->sk_state         = TCP_ESTABLISHED;
1289         newsk->sk_type          = sk->sk_type;
1290         init_peercred(newsk);
1291         newu = unix_sk(newsk);
1292         RCU_INIT_POINTER(newsk->sk_wq, &newu->peer_wq);
1293         otheru = unix_sk(other);
1294
1295         /* copy address information from listening to new sock*/
1296         if (otheru->addr) {
1297                 atomic_inc(&otheru->addr->refcnt);
1298                 newu->addr = otheru->addr;
1299         }
1300         if (otheru->dentry) {
1301                 newu->dentry    = dget(otheru->dentry);
1302                 newu->mnt       = mntget(otheru->mnt);
1303         }
1304
1305         /* Set credentials */
1306         copy_peercred(sk, other);
1307
1308         sock->state     = SS_CONNECTED;
1309         sk->sk_state    = TCP_ESTABLISHED;
1310         sock_hold(newsk);
1311
1312         smp_mb__after_atomic_inc();     /* sock_hold() does an atomic_inc() */
1313         unix_peer(sk)   = newsk;
1314
1315         unix_state_unlock(sk);
1316
1317         /* take ten and and send info to listening sock */
1318         spin_lock(&other->sk_receive_queue.lock);
1319         __skb_queue_tail(&other->sk_receive_queue, skb);
1320         spin_unlock(&other->sk_receive_queue.lock);
1321         unix_state_unlock(other);
1322         other->sk_data_ready(other, 0);
1323         sock_put(other);
1324         return 0;
1325
1326 out_unlock:
1327         if (other)
1328                 unix_state_unlock(other);
1329
1330 out:
1331         kfree_skb(skb);
1332         if (newsk)
1333                 unix_release_sock(newsk, 0);
1334         if (other)
1335                 sock_put(other);
1336         return err;
1337 }
1338
1339 static int unix_socketpair(struct socket *socka, struct socket *sockb)
1340 {
1341         struct sock *ska = socka->sk, *skb = sockb->sk;
1342
1343         /* Join our sockets back to back */
1344         sock_hold(ska);
1345         sock_hold(skb);
1346         unix_peer(ska) = skb;
1347         unix_peer(skb) = ska;
1348         init_peercred(ska);
1349         init_peercred(skb);
1350
1351         if (ska->sk_type != SOCK_DGRAM) {
1352                 ska->sk_state = TCP_ESTABLISHED;
1353                 skb->sk_state = TCP_ESTABLISHED;
1354                 socka->state  = SS_CONNECTED;
1355                 sockb->state  = SS_CONNECTED;
1356         }
1357         return 0;
1358 }
1359
1360 static void unix_sock_inherit_flags(const struct socket *old,
1361                                     struct socket *new)
1362 {
1363         if (test_bit(SOCK_PASSCRED, &old->flags))
1364                 set_bit(SOCK_PASSCRED, &new->flags);
1365         if (test_bit(SOCK_PASSSEC, &old->flags))
1366                 set_bit(SOCK_PASSSEC, &new->flags);
1367 }
1368
1369 static int unix_accept(struct socket *sock, struct socket *newsock, int flags)
1370 {
1371         struct sock *sk = sock->sk;
1372         struct sock *tsk;
1373         struct sk_buff *skb;
1374         int err;
1375
1376         err = -EOPNOTSUPP;
1377         if (sock->type != SOCK_STREAM && sock->type != SOCK_SEQPACKET)
1378                 goto out;
1379
1380         err = -EINVAL;
1381         if (sk->sk_state != TCP_LISTEN)
1382                 goto out;
1383
1384         /* If socket state is TCP_LISTEN it cannot change (for now...),
1385          * so that no locks are necessary.
1386          */
1387
1388         skb = skb_recv_datagram(sk, 0, flags&O_NONBLOCK, &err);
1389         if (!skb) {
1390                 /* This means receive shutdown. */
1391                 if (err == 0)
1392                         err = -EINVAL;
1393                 goto out;
1394         }
1395
1396         tsk = skb->sk;
1397         skb_free_datagram(sk, skb);
1398         wake_up_interruptible(&unix_sk(sk)->peer_wait);
1399
1400         /* attach accepted sock to socket */
1401         unix_state_lock(tsk);
1402         newsock->state = SS_CONNECTED;
1403         unix_sock_inherit_flags(sock, newsock);
1404         sock_graft(tsk, newsock);
1405         unix_state_unlock(tsk);
1406         return 0;
1407
1408 out:
1409         return err;
1410 }
1411
1412
1413 static int unix_getname(struct socket *sock, struct sockaddr *uaddr, int *uaddr_len, int peer)
1414 {
1415         struct sock *sk = sock->sk;
1416         struct unix_sock *u;
1417         DECLARE_SOCKADDR(struct sockaddr_un *, sunaddr, uaddr);
1418         int err = 0;
1419
1420         if (peer) {
1421                 sk = unix_peer_get(sk);
1422
1423                 err = -ENOTCONN;
1424                 if (!sk)
1425                         goto out;
1426                 err = 0;
1427         } else {
1428                 sock_hold(sk);
1429         }
1430
1431         u = unix_sk(sk);
1432         unix_state_lock(sk);
1433         if (!u->addr) {
1434                 sunaddr->sun_family = AF_UNIX;
1435                 sunaddr->sun_path[0] = 0;
1436                 *uaddr_len = sizeof(short);
1437         } else {
1438                 struct unix_address *addr = u->addr;
1439
1440                 *uaddr_len = addr->len;
1441                 memcpy(sunaddr, addr->name, *uaddr_len);
1442         }
1443         unix_state_unlock(sk);
1444         sock_put(sk);
1445 out:
1446         return err;
1447 }
1448
1449 static void unix_detach_fds(struct scm_cookie *scm, struct sk_buff *skb)
1450 {
1451         int i;
1452
1453         scm->fp = UNIXCB(skb).fp;
1454         UNIXCB(skb).fp = NULL;
1455
1456         for (i = scm->fp->count-1; i >= 0; i--)
1457                 unix_notinflight(scm->fp->fp[i]);
1458 }
1459
1460 static void unix_destruct_scm(struct sk_buff *skb)
1461 {
1462         struct scm_cookie scm;
1463         memset(&scm, 0, sizeof(scm));
1464         scm.pid  = UNIXCB(skb).pid;
1465         scm.cred = UNIXCB(skb).cred;
1466         if (UNIXCB(skb).fp)
1467                 unix_detach_fds(&scm, skb);
1468
1469         /* Alas, it calls VFS */
1470         /* So fscking what? fput() had been SMP-safe since the last Summer */
1471         scm_destroy(&scm);
1472         sock_wfree(skb);
1473 }
1474
1475 #define MAX_RECURSION_LEVEL 4
1476
1477 static int unix_attach_fds(struct scm_cookie *scm, struct sk_buff *skb)
1478 {
1479         int i;
1480         unsigned char max_level = 0;
1481         int unix_sock_count = 0;
1482
1483         for (i = scm->fp->count - 1; i >= 0; i--) {
1484                 struct sock *sk = unix_get_socket(scm->fp->fp[i]);
1485
1486                 if (sk) {
1487                         unix_sock_count++;
1488                         max_level = max(max_level,
1489                                         unix_sk(sk)->recursion_level);
1490                 }
1491         }
1492         if (unlikely(max_level > MAX_RECURSION_LEVEL))
1493                 return -ETOOMANYREFS;
1494
1495         /*
1496          * Need to duplicate file references for the sake of garbage
1497          * collection.  Otherwise a socket in the fps might become a
1498          * candidate for GC while the skb is not yet queued.
1499          */
1500         UNIXCB(skb).fp = scm_fp_dup(scm->fp);
1501         if (!UNIXCB(skb).fp)
1502                 return -ENOMEM;
1503
1504         if (unix_sock_count) {
1505                 for (i = scm->fp->count - 1; i >= 0; i--)
1506                         unix_inflight(scm->fp->fp[i]);
1507         }
1508         return max_level;
1509 }
1510
1511 static int unix_scm_to_skb(struct scm_cookie *scm, struct sk_buff *skb, bool send_fds)
1512 {
1513         int err = 0;
1514
1515         UNIXCB(skb).pid  = get_pid(scm->pid);
1516         if (scm->cred)
1517                 UNIXCB(skb).cred = get_cred(scm->cred);
1518         UNIXCB(skb).fp = NULL;
1519         if (scm->fp && send_fds)
1520                 err = unix_attach_fds(scm, skb);
1521
1522         skb->destructor = unix_destruct_scm;
1523         return err;
1524 }
1525
1526 /*
1527  * Some apps rely on write() giving SCM_CREDENTIALS
1528  * We include credentials if source or destination socket
1529  * asserted SOCK_PASSCRED.
1530  */
1531 static void maybe_add_creds(struct sk_buff *skb, const struct socket *sock,
1532                             const struct sock *other)
1533 {
1534         if (UNIXCB(skb).cred)
1535                 return;
1536         if (test_bit(SOCK_PASSCRED, &sock->flags) ||
1537             !other->sk_socket ||
1538             test_bit(SOCK_PASSCRED, &other->sk_socket->flags)) {
1539                 UNIXCB(skb).pid  = get_pid(task_tgid(current));
1540                 UNIXCB(skb).cred = get_current_cred();
1541         }
1542 }
1543
1544 /*
1545  *      Send AF_UNIX data.
1546  */
1547
1548 static int unix_dgram_sendmsg(struct kiocb *kiocb, struct socket *sock,
1549                               struct msghdr *msg, size_t len)
1550 {
1551         struct sock_iocb *siocb = kiocb_to_siocb(kiocb);
1552         struct sock *sk = sock->sk;
1553         struct net *net = sock_net(sk);
1554         struct unix_sock *u = unix_sk(sk);
1555         struct sockaddr_un *sunaddr = msg->msg_name;
1556         struct sock *other = NULL;
1557         int namelen = 0; /* fake GCC */
1558         int err;
1559         unsigned hash;
1560         struct sk_buff *skb;
1561         long timeo;
1562         struct scm_cookie tmp_scm;
1563         int max_level;
1564         int sk_locked;
1565
1566         if (NULL == siocb->scm)
1567                 siocb->scm = &tmp_scm;
1568         wait_for_unix_gc();
1569         err = scm_send(sock, msg, siocb->scm, false);
1570         if (err < 0)
1571                 return err;
1572
1573         err = -EOPNOTSUPP;
1574         if (msg->msg_flags&MSG_OOB)
1575                 goto out;
1576
1577         if (msg->msg_namelen) {
1578                 err = unix_mkname(sunaddr, msg->msg_namelen, &hash);
1579                 if (err < 0)
1580                         goto out;
1581                 namelen = err;
1582         } else {
1583                 sunaddr = NULL;
1584                 err = -ENOTCONN;
1585                 other = unix_peer_get(sk);
1586                 if (!other)
1587                         goto out;
1588         }
1589
1590         if (test_bit(SOCK_PASSCRED, &sock->flags) && !u->addr
1591             && (err = unix_autobind(sock)) != 0)
1592                 goto out;
1593
1594         err = -EMSGSIZE;
1595         if (len > sk->sk_sndbuf - 32)
1596                 goto out;
1597
1598         skb = sock_alloc_send_skb(sk, len, msg->msg_flags&MSG_DONTWAIT, &err);
1599         if (skb == NULL)
1600                 goto out;
1601
1602         err = unix_scm_to_skb(siocb->scm, skb, true);
1603         if (err < 0)
1604                 goto out_free;
1605         max_level = err + 1;
1606         unix_get_secdata(siocb->scm, skb);
1607
1608         skb_reset_transport_header(skb);
1609         err = memcpy_fromiovec(skb_put(skb, len), msg->msg_iov, len);
1610         if (err)
1611                 goto out_free;
1612
1613         timeo = sock_sndtimeo(sk, msg->msg_flags & MSG_DONTWAIT);
1614
1615 restart:
1616         if (!other) {
1617                 err = -ECONNRESET;
1618                 if (sunaddr == NULL)
1619                         goto out_free;
1620
1621                 other = unix_find_other(net, sunaddr, namelen, sk->sk_type,
1622                                         hash, &err);
1623                 if (other == NULL)
1624                         goto out_free;
1625         }
1626
1627         if (sk_filter(other, skb) < 0) {
1628                 /* Toss the packet but do not return any error to the sender */
1629                 err = len;
1630                 goto out_free;
1631         }
1632
1633         sk_locked = 0;
1634         unix_state_lock(other);
1635 restart_locked:
1636         err = -EPERM;
1637         if (!unix_may_send(sk, other))
1638                 goto out_unlock;
1639
1640         if (unlikely(sock_flag(other, SOCK_DEAD))) {
1641                 /*
1642                  *      Check with 1003.1g - what should
1643                  *      datagram error
1644                  */
1645                 unix_state_unlock(other);
1646                 sock_put(other);
1647
1648                 if (!sk_locked)
1649                         unix_state_lock(sk);
1650
1651                 err = 0;
1652                 if (unix_peer(sk) == other) {
1653                         unix_peer(sk) = NULL;
1654                         unix_dgram_peer_wake_disconnect_wakeup(sk, other);
1655
1656                         unix_state_unlock(sk);
1657
1658                         unix_dgram_disconnected(sk, other);
1659                         sock_put(other);
1660                         err = -ECONNREFUSED;
1661                 } else {
1662                         unix_state_unlock(sk);
1663                 }
1664
1665                 other = NULL;
1666                 if (err)
1667                         goto out_free;
1668                 goto restart;
1669         }
1670
1671         err = -EPIPE;
1672         if (other->sk_shutdown & RCV_SHUTDOWN)
1673                 goto out_unlock;
1674
1675         if (sk->sk_type != SOCK_SEQPACKET) {
1676                 err = security_unix_may_send(sk->sk_socket, other->sk_socket);
1677                 if (err)
1678                         goto out_unlock;
1679         }
1680
1681         if (unlikely(unix_peer(other) != sk && unix_recvq_full(other))) {
1682                 if (timeo) {
1683                         timeo = unix_wait_for_peer(other, timeo);
1684
1685                         err = sock_intr_errno(timeo);
1686                         if (signal_pending(current))
1687                                 goto out_free;
1688
1689                         goto restart;
1690                 }
1691
1692                 if (!sk_locked) {
1693                         unix_state_unlock(other);
1694                         unix_state_double_lock(sk, other);
1695                 }
1696
1697                 if (unix_peer(sk) != other ||
1698                     unix_dgram_peer_wake_me(sk, other)) {
1699                         err = -EAGAIN;
1700                         sk_locked = 1;
1701                         goto out_unlock;
1702                 }
1703
1704                 if (!sk_locked) {
1705                         sk_locked = 1;
1706                         goto restart_locked;
1707                 }
1708         }
1709
1710         if (unlikely(sk_locked))
1711                 unix_state_unlock(sk);
1712
1713         if (sock_flag(other, SOCK_RCVTSTAMP))
1714                 __net_timestamp(skb);
1715         maybe_add_creds(skb, sock, other);
1716         skb_queue_tail(&other->sk_receive_queue, skb);
1717         if (max_level > unix_sk(other)->recursion_level)
1718                 unix_sk(other)->recursion_level = max_level;
1719         unix_state_unlock(other);
1720         other->sk_data_ready(other, len);
1721         sock_put(other);
1722         scm_destroy(siocb->scm);
1723         return len;
1724
1725 out_unlock:
1726         if (sk_locked)
1727                 unix_state_unlock(sk);
1728         unix_state_unlock(other);
1729 out_free:
1730         kfree_skb(skb);
1731 out:
1732         if (other)
1733                 sock_put(other);
1734         scm_destroy(siocb->scm);
1735         return err;
1736 }
1737
1738
1739 static int unix_stream_sendmsg(struct kiocb *kiocb, struct socket *sock,
1740                                struct msghdr *msg, size_t len)
1741 {
1742         struct sock_iocb *siocb = kiocb_to_siocb(kiocb);
1743         struct sock *sk = sock->sk;
1744         struct sock *other = NULL;
1745         int err, size;
1746         struct sk_buff *skb;
1747         int sent = 0;
1748         struct scm_cookie tmp_scm;
1749         bool fds_sent = false;
1750         int max_level;
1751
1752         if (NULL == siocb->scm)
1753                 siocb->scm = &tmp_scm;
1754         wait_for_unix_gc();
1755         err = scm_send(sock, msg, siocb->scm, false);
1756         if (err < 0)
1757                 return err;
1758
1759         err = -EOPNOTSUPP;
1760         if (msg->msg_flags&MSG_OOB)
1761                 goto out_err;
1762
1763         if (msg->msg_namelen) {
1764                 err = sk->sk_state == TCP_ESTABLISHED ? -EISCONN : -EOPNOTSUPP;
1765                 goto out_err;
1766         } else {
1767                 err = -ENOTCONN;
1768                 other = unix_peer(sk);
1769                 if (!other)
1770                         goto out_err;
1771         }
1772
1773         if (sk->sk_shutdown & SEND_SHUTDOWN)
1774                 goto pipe_err;
1775
1776         while (sent < len) {
1777                 /*
1778                  *      Optimisation for the fact that under 0.01% of X
1779                  *      messages typically need breaking up.
1780                  */
1781
1782                 size = len-sent;
1783
1784                 /* Keep two messages in the pipe so it schedules better */
1785                 if (size > ((sk->sk_sndbuf >> 1) - 64))
1786                         size = (sk->sk_sndbuf >> 1) - 64;
1787
1788                 if (size > SKB_MAX_ALLOC)
1789                         size = SKB_MAX_ALLOC;
1790
1791                 /*
1792                  *      Grab a buffer
1793                  */
1794
1795                 skb = sock_alloc_send_skb(sk, size, msg->msg_flags&MSG_DONTWAIT,
1796                                           &err);
1797
1798                 if (skb == NULL)
1799                         goto out_err;
1800
1801                 /*
1802                  *      If you pass two values to the sock_alloc_send_skb
1803                  *      it tries to grab the large buffer with GFP_NOFS
1804                  *      (which can fail easily), and if it fails grab the
1805                  *      fallback size buffer which is under a page and will
1806                  *      succeed. [Alan]
1807                  */
1808                 size = min_t(int, size, skb_tailroom(skb));
1809
1810
1811                 /* Only send the fds in the first buffer */
1812                 err = unix_scm_to_skb(siocb->scm, skb, !fds_sent);
1813                 if (err < 0) {
1814                         kfree_skb(skb);
1815                         goto out_err;
1816                 }
1817                 max_level = err + 1;
1818                 fds_sent = true;
1819
1820                 err = memcpy_fromiovec(skb_put(skb, size), msg->msg_iov, size);
1821                 if (err) {
1822                         kfree_skb(skb);
1823                         goto out_err;
1824                 }
1825
1826                 unix_state_lock(other);
1827
1828                 if (sock_flag(other, SOCK_DEAD) ||
1829                     (other->sk_shutdown & RCV_SHUTDOWN))
1830                         goto pipe_err_free;
1831
1832                 maybe_add_creds(skb, sock, other);
1833                 skb_queue_tail(&other->sk_receive_queue, skb);
1834                 if (max_level > unix_sk(other)->recursion_level)
1835                         unix_sk(other)->recursion_level = max_level;
1836                 unix_state_unlock(other);
1837                 other->sk_data_ready(other, size);
1838                 sent += size;
1839         }
1840
1841         scm_destroy(siocb->scm);
1842         siocb->scm = NULL;
1843
1844         return sent;
1845
1846 pipe_err_free:
1847         unix_state_unlock(other);
1848         kfree_skb(skb);
1849 pipe_err:
1850         if (sent == 0 && !(msg->msg_flags&MSG_NOSIGNAL))
1851                 send_sig(SIGPIPE, current, 0);
1852         err = -EPIPE;
1853 out_err:
1854         scm_destroy(siocb->scm);
1855         siocb->scm = NULL;
1856         return sent ? : err;
1857 }
1858
1859 static int unix_seqpacket_sendmsg(struct kiocb *kiocb, struct socket *sock,
1860                                   struct msghdr *msg, size_t len)
1861 {
1862         int err;
1863         struct sock *sk = sock->sk;
1864
1865         err = sock_error(sk);
1866         if (err)
1867                 return err;
1868
1869         if (sk->sk_state != TCP_ESTABLISHED)
1870                 return -ENOTCONN;
1871
1872         if (msg->msg_namelen)
1873                 msg->msg_namelen = 0;
1874
1875         return unix_dgram_sendmsg(kiocb, sock, msg, len);
1876 }
1877
1878 static int unix_seqpacket_recvmsg(struct kiocb *iocb, struct socket *sock,
1879                               struct msghdr *msg, size_t size,
1880                               int flags)
1881 {
1882         struct sock *sk = sock->sk;
1883
1884         if (sk->sk_state != TCP_ESTABLISHED)
1885                 return -ENOTCONN;
1886
1887         return unix_dgram_recvmsg(iocb, sock, msg, size, flags);
1888 }
1889
1890 static void unix_copy_addr(struct msghdr *msg, struct sock *sk)
1891 {
1892         struct unix_sock *u = unix_sk(sk);
1893
1894         if (u->addr) {
1895                 msg->msg_namelen = u->addr->len;
1896                 memcpy(msg->msg_name, u->addr->name, u->addr->len);
1897         }
1898 }
1899
1900 static int unix_dgram_recvmsg(struct kiocb *iocb, struct socket *sock,
1901                               struct msghdr *msg, size_t size,
1902                               int flags)
1903 {
1904         struct sock_iocb *siocb = kiocb_to_siocb(iocb);
1905         struct scm_cookie tmp_scm;
1906         struct sock *sk = sock->sk;
1907         struct unix_sock *u = unix_sk(sk);
1908         int noblock = flags & MSG_DONTWAIT;
1909         struct sk_buff *skb;
1910         int err;
1911
1912         err = -EOPNOTSUPP;
1913         if (flags&MSG_OOB)
1914                 goto out;
1915
1916         err = mutex_lock_interruptible(&u->readlock);
1917         if (unlikely(err)) {
1918                 /* recvmsg() in non blocking mode is supposed to return -EAGAIN
1919                  * sk_rcvtimeo is not honored by mutex_lock_interruptible()
1920                  */
1921                 err = noblock ? -EAGAIN : -ERESTARTSYS;
1922                 goto out;
1923         }
1924
1925         skb = skb_recv_datagram(sk, flags, noblock, &err);
1926         if (!skb) {
1927                 unix_state_lock(sk);
1928                 /* Signal EOF on disconnected non-blocking SEQPACKET socket. */
1929                 if (sk->sk_type == SOCK_SEQPACKET && err == -EAGAIN &&
1930                     (sk->sk_shutdown & RCV_SHUTDOWN))
1931                         err = 0;
1932                 unix_state_unlock(sk);
1933                 goto out_unlock;
1934         }
1935
1936         wake_up_interruptible_sync_poll(&u->peer_wait,
1937                                         POLLOUT | POLLWRNORM | POLLWRBAND);
1938
1939         if (msg->msg_name)
1940                 unix_copy_addr(msg, skb->sk);
1941
1942         if (size > skb->len)
1943                 size = skb->len;
1944         else if (size < skb->len)
1945                 msg->msg_flags |= MSG_TRUNC;
1946
1947         err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, size);
1948         if (err)
1949                 goto out_free;
1950
1951         if (sock_flag(sk, SOCK_RCVTSTAMP))
1952                 __sock_recv_timestamp(msg, sk, skb);
1953
1954         if (!siocb->scm) {
1955                 siocb->scm = &tmp_scm;
1956                 memset(&tmp_scm, 0, sizeof(tmp_scm));
1957         }
1958         scm_set_cred(siocb->scm, UNIXCB(skb).pid, UNIXCB(skb).cred);
1959         unix_set_secdata(siocb->scm, skb);
1960
1961         if (!(flags & MSG_PEEK)) {
1962                 if (UNIXCB(skb).fp)
1963                         unix_detach_fds(siocb->scm, skb);
1964         } else {
1965                 /* It is questionable: on PEEK we could:
1966                    - do not return fds - good, but too simple 8)
1967                    - return fds, and do not return them on read (old strategy,
1968                      apparently wrong)
1969                    - clone fds (I chose it for now, it is the most universal
1970                      solution)
1971
1972                    POSIX 1003.1g does not actually define this clearly
1973                    at all. POSIX 1003.1g doesn't define a lot of things
1974                    clearly however!
1975
1976                 */
1977                 if (UNIXCB(skb).fp)
1978                         siocb->scm->fp = scm_fp_dup(UNIXCB(skb).fp);
1979         }
1980         err = size;
1981
1982         scm_recv(sock, msg, siocb->scm, flags);
1983
1984 out_free:
1985         skb_free_datagram(sk, skb);
1986 out_unlock:
1987         mutex_unlock(&u->readlock);
1988 out:
1989         return err;
1990 }
1991
1992 /*
1993  *      Sleep until data has arrive. But check for races..
1994  */
1995
1996 static long unix_stream_data_wait(struct sock *sk, long timeo)
1997 {
1998         DEFINE_WAIT(wait);
1999
2000         unix_state_lock(sk);
2001
2002         for (;;) {
2003                 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
2004
2005                 if (!skb_queue_empty(&sk->sk_receive_queue) ||
2006                     sk->sk_err ||
2007                     (sk->sk_shutdown & RCV_SHUTDOWN) ||
2008                     signal_pending(current) ||
2009                     !timeo)
2010                         break;
2011
2012                 set_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
2013                 unix_state_unlock(sk);
2014                 timeo = schedule_timeout(timeo);
2015                 unix_state_lock(sk);
2016
2017                 if (sock_flag(sk, SOCK_DEAD))
2018                         break;
2019
2020                 clear_bit(SOCK_ASYNC_WAITDATA, &sk->sk_socket->flags);
2021         }
2022
2023         finish_wait(sk_sleep(sk), &wait);
2024         unix_state_unlock(sk);
2025         return timeo;
2026 }
2027
2028
2029
2030 static int unix_stream_recvmsg(struct kiocb *iocb, struct socket *sock,
2031                                struct msghdr *msg, size_t size,
2032                                int flags)
2033 {
2034         struct sock_iocb *siocb = kiocb_to_siocb(iocb);
2035         struct scm_cookie tmp_scm;
2036         struct sock *sk = sock->sk;
2037         struct unix_sock *u = unix_sk(sk);
2038         struct sockaddr_un *sunaddr = msg->msg_name;
2039         int copied = 0;
2040         int noblock = flags & MSG_DONTWAIT;
2041         int check_creds = 0;
2042         int target;
2043         int err = 0;
2044         long timeo;
2045
2046         err = -EINVAL;
2047         if (sk->sk_state != TCP_ESTABLISHED)
2048                 goto out;
2049
2050         err = -EOPNOTSUPP;
2051         if (flags&MSG_OOB)
2052                 goto out;
2053
2054         target = sock_rcvlowat(sk, flags&MSG_WAITALL, size);
2055         timeo = sock_rcvtimeo(sk, noblock);
2056
2057         /* Lock the socket to prevent queue disordering
2058          * while sleeps in memcpy_tomsg
2059          */
2060
2061         if (!siocb->scm) {
2062                 siocb->scm = &tmp_scm;
2063                 memset(&tmp_scm, 0, sizeof(tmp_scm));
2064         }
2065
2066         mutex_lock(&u->readlock);
2067
2068         do {
2069                 int chunk;
2070                 struct sk_buff *skb;
2071
2072                 unix_state_lock(sk);
2073                 if (sock_flag(sk, SOCK_DEAD)) {
2074                         err = -ECONNRESET;
2075                         goto unlock;
2076                 }
2077                 skb = skb_peek(&sk->sk_receive_queue);
2078                 if (skb == NULL) {
2079                         unix_sk(sk)->recursion_level = 0;
2080                         if (copied >= target)
2081                                 goto unlock;
2082
2083                         /*
2084                          *      POSIX 1003.1g mandates this order.
2085                          */
2086
2087                         err = sock_error(sk);
2088                         if (err)
2089                                 goto unlock;
2090                         if (sk->sk_shutdown & RCV_SHUTDOWN)
2091                                 goto unlock;
2092
2093                         unix_state_unlock(sk);
2094                         err = -EAGAIN;
2095                         if (!timeo)
2096                                 break;
2097                         mutex_unlock(&u->readlock);
2098
2099                         timeo = unix_stream_data_wait(sk, timeo);
2100
2101                         if (signal_pending(current)) {
2102                                 err = sock_intr_errno(timeo);
2103                                 scm_destroy(siocb->scm);
2104                                 goto out;
2105                         }
2106
2107                         mutex_lock(&u->readlock);
2108                         continue;
2109  unlock:
2110                         unix_state_unlock(sk);
2111                         break;
2112                 }
2113                 unix_state_unlock(sk);
2114
2115                 if (check_creds) {
2116                         /* Never glue messages from different writers */
2117                         if ((UNIXCB(skb).pid  != siocb->scm->pid) ||
2118                             (UNIXCB(skb).cred != siocb->scm->cred))
2119                                 break;
2120                 } else if (test_bit(SOCK_PASSCRED, &sock->flags)) {
2121                         /* Copy credentials */
2122                         scm_set_cred(siocb->scm, UNIXCB(skb).pid, UNIXCB(skb).cred);
2123                         check_creds = 1;
2124                 }
2125
2126                 /* Copy address just once */
2127                 if (sunaddr) {
2128                         unix_copy_addr(msg, skb->sk);
2129                         sunaddr = NULL;
2130                 }
2131
2132                 chunk = min_t(unsigned int, skb->len, size);
2133                 if (memcpy_toiovec(msg->msg_iov, skb->data, chunk)) {
2134                         if (copied == 0)
2135                                 copied = -EFAULT;
2136                         break;
2137                 }
2138                 copied += chunk;
2139                 size -= chunk;
2140
2141                 /* Mark read part of skb as used */
2142                 if (!(flags & MSG_PEEK)) {
2143                         skb_pull(skb, chunk);
2144
2145                         if (UNIXCB(skb).fp)
2146                                 unix_detach_fds(siocb->scm, skb);
2147
2148                         if (skb->len)
2149                                 break;
2150
2151                         skb_unlink(skb, &sk->sk_receive_queue);
2152                         consume_skb(skb);
2153
2154                         if (siocb->scm->fp)
2155                                 break;
2156                 } else {
2157                         /* It is questionable, see note in unix_dgram_recvmsg.
2158                          */
2159                         if (UNIXCB(skb).fp)
2160                                 siocb->scm->fp = scm_fp_dup(UNIXCB(skb).fp);
2161
2162                         break;
2163                 }
2164         } while (size);
2165
2166         mutex_unlock(&u->readlock);
2167         scm_recv(sock, msg, siocb->scm, flags);
2168 out:
2169         return copied ? : err;
2170 }
2171
2172 static int unix_shutdown(struct socket *sock, int mode)
2173 {
2174         struct sock *sk = sock->sk;
2175         struct sock *other;
2176
2177         mode = (mode+1)&(RCV_SHUTDOWN|SEND_SHUTDOWN);
2178
2179         if (!mode)
2180                 return 0;
2181
2182         unix_state_lock(sk);
2183         sk->sk_shutdown |= mode;
2184         other = unix_peer(sk);
2185         if (other)
2186                 sock_hold(other);
2187         unix_state_unlock(sk);
2188         sk->sk_state_change(sk);
2189
2190         if (other &&
2191                 (sk->sk_type == SOCK_STREAM || sk->sk_type == SOCK_SEQPACKET)) {
2192
2193                 int peer_mode = 0;
2194
2195                 if (mode&RCV_SHUTDOWN)
2196                         peer_mode |= SEND_SHUTDOWN;
2197                 if (mode&SEND_SHUTDOWN)
2198                         peer_mode |= RCV_SHUTDOWN;
2199                 unix_state_lock(other);
2200                 other->sk_shutdown |= peer_mode;
2201                 unix_state_unlock(other);
2202                 other->sk_state_change(other);
2203                 if (peer_mode == SHUTDOWN_MASK)
2204                         sk_wake_async(other, SOCK_WAKE_WAITD, POLL_HUP);
2205                 else if (peer_mode & RCV_SHUTDOWN)
2206                         sk_wake_async(other, SOCK_WAKE_WAITD, POLL_IN);
2207         }
2208         if (other)
2209                 sock_put(other);
2210
2211         return 0;
2212 }
2213
2214 static int unix_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
2215 {
2216         struct sock *sk = sock->sk;
2217         long amount = 0;
2218         int err;
2219
2220         switch (cmd) {
2221         case SIOCOUTQ:
2222                 amount = sk_wmem_alloc_get(sk);
2223                 err = put_user(amount, (int __user *)arg);
2224                 break;
2225         case SIOCINQ:
2226                 {
2227                         struct sk_buff *skb;
2228
2229                         if (sk->sk_state == TCP_LISTEN) {
2230                                 err = -EINVAL;
2231                                 break;
2232                         }
2233
2234                         spin_lock(&sk->sk_receive_queue.lock);
2235                         if (sk->sk_type == SOCK_STREAM ||
2236                             sk->sk_type == SOCK_SEQPACKET) {
2237                                 skb_queue_walk(&sk->sk_receive_queue, skb)
2238                                         amount += skb->len;
2239                         } else {
2240                                 skb = skb_peek(&sk->sk_receive_queue);
2241                                 if (skb)
2242                                         amount = skb->len;
2243                         }
2244                         spin_unlock(&sk->sk_receive_queue.lock);
2245                         err = put_user(amount, (int __user *)arg);
2246                         break;
2247                 }
2248
2249         default:
2250                 err = -ENOIOCTLCMD;
2251                 break;
2252         }
2253         return err;
2254 }
2255
2256 static unsigned int unix_poll(struct file *file, struct socket *sock, poll_table *wait)
2257 {
2258         struct sock *sk = sock->sk;
2259         unsigned int mask;
2260
2261         sock_poll_wait(file, sk_sleep(sk), wait);
2262         mask = 0;
2263
2264         /* exceptional events? */
2265         if (sk->sk_err)
2266                 mask |= POLLERR;
2267         if (sk->sk_shutdown == SHUTDOWN_MASK)
2268                 mask |= POLLHUP;
2269         if (sk->sk_shutdown & RCV_SHUTDOWN)
2270                 mask |= POLLRDHUP | POLLIN | POLLRDNORM;
2271
2272         /* readable? */
2273         if (!skb_queue_empty(&sk->sk_receive_queue))
2274                 mask |= POLLIN | POLLRDNORM;
2275
2276         /* Connection-based need to check for termination and startup */
2277         if ((sk->sk_type == SOCK_STREAM || sk->sk_type == SOCK_SEQPACKET) &&
2278             sk->sk_state == TCP_CLOSE)
2279                 mask |= POLLHUP;
2280
2281         /*
2282          * we set writable also when the other side has shut down the
2283          * connection. This prevents stuck sockets.
2284          */
2285         if (unix_writable(sk))
2286                 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
2287
2288         return mask;
2289 }
2290
2291 static unsigned int unix_dgram_poll(struct file *file, struct socket *sock,
2292                                     poll_table *wait)
2293 {
2294         struct sock *sk = sock->sk, *other;
2295         unsigned int mask, writable;
2296
2297         sock_poll_wait(file, sk_sleep(sk), wait);
2298         mask = 0;
2299
2300         /* exceptional events? */
2301         if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
2302                 mask |= POLLERR;
2303         if (sk->sk_shutdown & RCV_SHUTDOWN)
2304                 mask |= POLLRDHUP | POLLIN | POLLRDNORM;
2305         if (sk->sk_shutdown == SHUTDOWN_MASK)
2306                 mask |= POLLHUP;
2307
2308         /* readable? */
2309         if (!skb_queue_empty(&sk->sk_receive_queue))
2310                 mask |= POLLIN | POLLRDNORM;
2311
2312         /* Connection-based need to check for termination and startup */
2313         if (sk->sk_type == SOCK_SEQPACKET) {
2314                 if (sk->sk_state == TCP_CLOSE)
2315                         mask |= POLLHUP;
2316                 /* connection hasn't started yet? */
2317                 if (sk->sk_state == TCP_SYN_SENT)
2318                         return mask;
2319         }
2320
2321         /* No write status requested, avoid expensive OUT tests. */
2322         if (wait && !(wait->key & (POLLWRBAND | POLLWRNORM | POLLOUT)))
2323                 return mask;
2324
2325         writable = unix_writable(sk);
2326         if (writable) {
2327                 unix_state_lock(sk);
2328
2329                 other = unix_peer(sk);
2330                 if (other && unix_peer(other) != sk &&
2331                     unix_recvq_full(other) &&
2332                     unix_dgram_peer_wake_me(sk, other))
2333                         writable = 0;
2334
2335                 unix_state_unlock(sk);
2336         }
2337
2338         if (writable)
2339                 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
2340         else
2341                 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
2342
2343         return mask;
2344 }
2345
2346 #ifdef CONFIG_PROC_FS
2347 static struct sock *first_unix_socket(int *i)
2348 {
2349         for (*i = 0; *i <= UNIX_HASH_SIZE; (*i)++) {
2350                 if (!hlist_empty(&unix_socket_table[*i]))
2351                         return __sk_head(&unix_socket_table[*i]);
2352         }
2353         return NULL;
2354 }
2355
2356 static struct sock *next_unix_socket(int *i, struct sock *s)
2357 {
2358         struct sock *next = sk_next(s);
2359         /* More in this chain? */
2360         if (next)
2361                 return next;
2362         /* Look for next non-empty chain. */
2363         for ((*i)++; *i <= UNIX_HASH_SIZE; (*i)++) {
2364                 if (!hlist_empty(&unix_socket_table[*i]))
2365                         return __sk_head(&unix_socket_table[*i]);
2366         }
2367         return NULL;
2368 }
2369
2370 struct unix_iter_state {
2371         struct seq_net_private p;
2372         int i;
2373 };
2374
2375 static struct sock *unix_seq_idx(struct seq_file *seq, loff_t pos)
2376 {
2377         struct unix_iter_state *iter = seq->private;
2378         loff_t off = 0;
2379         struct sock *s;
2380
2381         for (s = first_unix_socket(&iter->i); s; s = next_unix_socket(&iter->i, s)) {
2382                 if (sock_net(s) != seq_file_net(seq))
2383                         continue;
2384                 if (off == pos)
2385                         return s;
2386                 ++off;
2387         }
2388         return NULL;
2389 }
2390
2391 static void *unix_seq_start(struct seq_file *seq, loff_t *pos)
2392         __acquires(unix_table_lock)
2393 {
2394         spin_lock(&unix_table_lock);
2395         return *pos ? unix_seq_idx(seq, *pos - 1) : SEQ_START_TOKEN;
2396 }
2397
2398 static void *unix_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2399 {
2400         struct unix_iter_state *iter = seq->private;
2401         struct sock *sk = v;
2402         ++*pos;
2403
2404         if (v == SEQ_START_TOKEN)
2405                 sk = first_unix_socket(&iter->i);
2406         else
2407                 sk = next_unix_socket(&iter->i, sk);
2408         while (sk && (sock_net(sk) != seq_file_net(seq)))
2409                 sk = next_unix_socket(&iter->i, sk);
2410         return sk;
2411 }
2412
2413 static void unix_seq_stop(struct seq_file *seq, void *v)
2414         __releases(unix_table_lock)
2415 {
2416         spin_unlock(&unix_table_lock);
2417 }
2418
2419 static int unix_seq_show(struct seq_file *seq, void *v)
2420 {
2421
2422         if (v == SEQ_START_TOKEN)
2423                 seq_puts(seq, "Num       RefCount Protocol Flags    Type St "
2424                          "Inode Path\n");
2425         else {
2426                 struct sock *s = v;
2427                 struct unix_sock *u = unix_sk(s);
2428                 unix_state_lock(s);
2429
2430                 seq_printf(seq, "%pK: %08X %08X %08X %04X %02X %5lu",
2431                         s,
2432                         atomic_read(&s->sk_refcnt),
2433                         0,
2434                         s->sk_state == TCP_LISTEN ? __SO_ACCEPTCON : 0,
2435                         s->sk_type,
2436                         s->sk_socket ?
2437                         (s->sk_state == TCP_ESTABLISHED ? SS_CONNECTED : SS_UNCONNECTED) :
2438                         (s->sk_state == TCP_ESTABLISHED ? SS_CONNECTING : SS_DISCONNECTING),
2439                         sock_i_ino(s));
2440
2441                 if (u->addr) {
2442                         int i, len;
2443                         seq_putc(seq, ' ');
2444
2445                         i = 0;
2446                         len = u->addr->len - sizeof(short);
2447                         if (!UNIX_ABSTRACT(s))
2448                                 len--;
2449                         else {
2450                                 seq_putc(seq, '@');
2451                                 i++;
2452                         }
2453                         for ( ; i < len; i++)
2454                                 seq_putc(seq, u->addr->name->sun_path[i]);
2455                 }
2456                 unix_state_unlock(s);
2457                 seq_putc(seq, '\n');
2458         }
2459
2460         return 0;
2461 }
2462
2463 static const struct seq_operations unix_seq_ops = {
2464         .start  = unix_seq_start,
2465         .next   = unix_seq_next,
2466         .stop   = unix_seq_stop,
2467         .show   = unix_seq_show,
2468 };
2469
2470 static int unix_seq_open(struct inode *inode, struct file *file)
2471 {
2472         return seq_open_net(inode, file, &unix_seq_ops,
2473                             sizeof(struct unix_iter_state));
2474 }
2475
2476 static const struct file_operations unix_seq_fops = {
2477         .owner          = THIS_MODULE,
2478         .open           = unix_seq_open,
2479         .read           = seq_read,
2480         .llseek         = seq_lseek,
2481         .release        = seq_release_net,
2482 };
2483
2484 #endif
2485
2486 static const struct net_proto_family unix_family_ops = {
2487         .family = PF_UNIX,
2488         .create = unix_create,
2489         .owner  = THIS_MODULE,
2490 };
2491
2492
2493 static int __net_init unix_net_init(struct net *net)
2494 {
2495         int error = -ENOMEM;
2496
2497         net->unx.sysctl_max_dgram_qlen = 10;
2498         if (unix_sysctl_register(net))
2499                 goto out;
2500
2501 #ifdef CONFIG_PROC_FS
2502         if (!proc_net_fops_create(net, "unix", 0, &unix_seq_fops)) {
2503                 unix_sysctl_unregister(net);
2504                 goto out;
2505         }
2506 #endif
2507         error = 0;
2508 out:
2509         return error;
2510 }
2511
2512 static void __net_exit unix_net_exit(struct net *net)
2513 {
2514         unix_sysctl_unregister(net);
2515         proc_net_remove(net, "unix");
2516 }
2517
2518 static struct pernet_operations unix_net_ops = {
2519         .init = unix_net_init,
2520         .exit = unix_net_exit,
2521 };
2522
2523 static int __init af_unix_init(void)
2524 {
2525         int rc = -1;
2526         struct sk_buff *dummy_skb;
2527
2528         BUILD_BUG_ON(sizeof(struct unix_skb_parms) > sizeof(dummy_skb->cb));
2529
2530         rc = proto_register(&unix_proto, 1);
2531         if (rc != 0) {
2532                 printk(KERN_CRIT "%s: Cannot create unix_sock SLAB cache!\n",
2533                        __func__);
2534                 goto out;
2535         }
2536
2537         sock_register(&unix_family_ops);
2538         register_pernet_subsys(&unix_net_ops);
2539 out:
2540         return rc;
2541 }
2542
2543 static void __exit af_unix_exit(void)
2544 {
2545         sock_unregister(PF_UNIX);
2546         proto_unregister(&unix_proto);
2547         unregister_pernet_subsys(&unix_net_ops);
2548 }
2549
2550 /* Earlier than device_initcall() so that other drivers invoking
2551    request_module() don't end up in a loop when modprobe tries
2552    to use a UNIX socket. But later than subsys_initcall() because
2553    we depend on stuff initialised there */
2554 fs_initcall(af_unix_init);
2555 module_exit(af_unix_exit);
2556
2557 MODULE_LICENSE("GPL");
2558 MODULE_ALIAS_NETPROTO(PF_UNIX);