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