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