uipc_usrreq.c revision 1.154 1 1.154 rtr /* $NetBSD: uipc_usrreq.c,v 1.154 2014/06/22 08:10:18 rtr Exp $ */
2 1.30 thorpej
3 1.30 thorpej /*-
4 1.121 mrg * Copyright (c) 1998, 2000, 2004, 2008, 2009 The NetBSD Foundation, Inc.
5 1.30 thorpej * All rights reserved.
6 1.30 thorpej *
7 1.30 thorpej * This code is derived from software contributed to The NetBSD Foundation
8 1.30 thorpej * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
9 1.121 mrg * NASA Ames Research Center, and by Andrew Doran.
10 1.30 thorpej *
11 1.30 thorpej * Redistribution and use in source and binary forms, with or without
12 1.30 thorpej * modification, are permitted provided that the following conditions
13 1.30 thorpej * are met:
14 1.30 thorpej * 1. Redistributions of source code must retain the above copyright
15 1.30 thorpej * notice, this list of conditions and the following disclaimer.
16 1.30 thorpej * 2. Redistributions in binary form must reproduce the above copyright
17 1.30 thorpej * notice, this list of conditions and the following disclaimer in the
18 1.30 thorpej * documentation and/or other materials provided with the distribution.
19 1.30 thorpej *
20 1.30 thorpej * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
21 1.30 thorpej * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22 1.30 thorpej * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23 1.30 thorpej * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
24 1.30 thorpej * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25 1.30 thorpej * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26 1.30 thorpej * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27 1.30 thorpej * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28 1.30 thorpej * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29 1.30 thorpej * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30 1.30 thorpej * POSSIBILITY OF SUCH DAMAGE.
31 1.30 thorpej */
32 1.10 cgd
33 1.1 cgd /*
34 1.8 mycroft * Copyright (c) 1982, 1986, 1989, 1991, 1993
35 1.8 mycroft * The Regents of the University of California. All rights reserved.
36 1.1 cgd *
37 1.1 cgd * Redistribution and use in source and binary forms, with or without
38 1.1 cgd * modification, are permitted provided that the following conditions
39 1.1 cgd * are met:
40 1.1 cgd * 1. Redistributions of source code must retain the above copyright
41 1.1 cgd * notice, this list of conditions and the following disclaimer.
42 1.1 cgd * 2. Redistributions in binary form must reproduce the above copyright
43 1.1 cgd * notice, this list of conditions and the following disclaimer in the
44 1.1 cgd * documentation and/or other materials provided with the distribution.
45 1.67 agc * 3. Neither the name of the University nor the names of its contributors
46 1.67 agc * may be used to endorse or promote products derived from this software
47 1.67 agc * without specific prior written permission.
48 1.67 agc *
49 1.67 agc * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
50 1.67 agc * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
51 1.67 agc * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
52 1.67 agc * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
53 1.67 agc * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
54 1.67 agc * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
55 1.67 agc * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
56 1.67 agc * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
57 1.67 agc * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
58 1.67 agc * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
59 1.67 agc * SUCH DAMAGE.
60 1.67 agc *
61 1.67 agc * @(#)uipc_usrreq.c 8.9 (Berkeley) 5/14/95
62 1.67 agc */
63 1.67 agc
64 1.67 agc /*
65 1.67 agc * Copyright (c) 1997 Christopher G. Demetriou. All rights reserved.
66 1.67 agc *
67 1.67 agc * Redistribution and use in source and binary forms, with or without
68 1.67 agc * modification, are permitted provided that the following conditions
69 1.67 agc * are met:
70 1.67 agc * 1. Redistributions of source code must retain the above copyright
71 1.67 agc * notice, this list of conditions and the following disclaimer.
72 1.67 agc * 2. Redistributions in binary form must reproduce the above copyright
73 1.67 agc * notice, this list of conditions and the following disclaimer in the
74 1.67 agc * documentation and/or other materials provided with the distribution.
75 1.1 cgd * 3. All advertising materials mentioning features or use of this software
76 1.1 cgd * must display the following acknowledgement:
77 1.1 cgd * This product includes software developed by the University of
78 1.1 cgd * California, Berkeley and its contributors.
79 1.1 cgd * 4. Neither the name of the University nor the names of its contributors
80 1.1 cgd * may be used to endorse or promote products derived from this software
81 1.1 cgd * without specific prior written permission.
82 1.1 cgd *
83 1.1 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
84 1.1 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
85 1.1 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
86 1.1 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
87 1.1 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
88 1.1 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
89 1.1 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
90 1.1 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
91 1.1 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
92 1.1 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
93 1.1 cgd * SUCH DAMAGE.
94 1.1 cgd *
95 1.31 fvdl * @(#)uipc_usrreq.c 8.9 (Berkeley) 5/14/95
96 1.1 cgd */
97 1.53 lukem
98 1.53 lukem #include <sys/cdefs.h>
99 1.154 rtr __KERNEL_RCSID(0, "$NetBSD: uipc_usrreq.c,v 1.154 2014/06/22 08:10:18 rtr Exp $");
100 1.1 cgd
101 1.7 mycroft #include <sys/param.h>
102 1.8 mycroft #include <sys/systm.h>
103 1.7 mycroft #include <sys/proc.h>
104 1.7 mycroft #include <sys/filedesc.h>
105 1.7 mycroft #include <sys/domain.h>
106 1.7 mycroft #include <sys/protosw.h>
107 1.7 mycroft #include <sys/socket.h>
108 1.7 mycroft #include <sys/socketvar.h>
109 1.7 mycroft #include <sys/unpcb.h>
110 1.7 mycroft #include <sys/un.h>
111 1.7 mycroft #include <sys/namei.h>
112 1.7 mycroft #include <sys/vnode.h>
113 1.7 mycroft #include <sys/file.h>
114 1.7 mycroft #include <sys/stat.h>
115 1.7 mycroft #include <sys/mbuf.h>
116 1.91 elad #include <sys/kauth.h>
117 1.101 ad #include <sys/kmem.h>
118 1.106 ad #include <sys/atomic.h>
119 1.119 pooka #include <sys/uidinfo.h>
120 1.121 mrg #include <sys/kernel.h>
121 1.121 mrg #include <sys/kthread.h>
122 1.1 cgd
123 1.1 cgd /*
124 1.1 cgd * Unix communications domain.
125 1.1 cgd *
126 1.1 cgd * TODO:
127 1.134 manu * RDM
128 1.1 cgd * rethink name space problems
129 1.1 cgd * need a proper out-of-band
130 1.112 ad *
131 1.112 ad * Notes on locking:
132 1.112 ad *
133 1.112 ad * The generic rules noted in uipc_socket2.c apply. In addition:
134 1.112 ad *
135 1.112 ad * o We have a global lock, uipc_lock.
136 1.112 ad *
137 1.112 ad * o All datagram sockets are locked by uipc_lock.
138 1.112 ad *
139 1.112 ad * o For stream socketpairs, the two endpoints are created sharing the same
140 1.112 ad * independent lock. Sockets presented to PRU_CONNECT2 must already have
141 1.112 ad * matching locks.
142 1.112 ad *
143 1.112 ad * o Stream sockets created via socket() start life with their own
144 1.112 ad * independent lock.
145 1.112 ad *
146 1.112 ad * o Stream connections to a named endpoint are slightly more complicated.
147 1.112 ad * Sockets that have called listen() have their lock pointer mutated to
148 1.112 ad * the global uipc_lock. When establishing a connection, the connecting
149 1.112 ad * socket also has its lock mutated to uipc_lock, which matches the head
150 1.112 ad * (listening socket). We create a new socket for accept() to return, and
151 1.112 ad * that also shares the head's lock. Until the connection is completely
152 1.112 ad * done on both ends, all three sockets are locked by uipc_lock. Once the
153 1.112 ad * connection is complete, the association with the head's lock is broken.
154 1.112 ad * The connecting socket and the socket returned from accept() have their
155 1.112 ad * lock pointers mutated away from uipc_lock, and back to the connecting
156 1.112 ad * socket's original, independent lock. The head continues to be locked
157 1.112 ad * by uipc_lock.
158 1.112 ad *
159 1.112 ad * o If uipc_lock is determined to be a significant source of contention,
160 1.112 ad * it could easily be hashed out. It is difficult to simply make it an
161 1.112 ad * independent lock because of visibility / garbage collection issues:
162 1.112 ad * if a socket has been associated with a lock at any point, that lock
163 1.112 ad * must remain valid until the socket is no longer visible in the system.
164 1.112 ad * The lock must not be freed or otherwise destroyed until any sockets
165 1.112 ad * that had referenced it have also been destroyed.
166 1.1 cgd */
167 1.93 christos const struct sockaddr_un sun_noname = {
168 1.145 christos .sun_len = offsetof(struct sockaddr_un, sun_path),
169 1.93 christos .sun_family = AF_LOCAL,
170 1.93 christos };
171 1.1 cgd ino_t unp_ino; /* prototype for fake inode numbers */
172 1.1 cgd
173 1.152 rmind static void unp_detach(struct socket *);
174 1.92 ad struct mbuf *unp_addsockcred(struct lwp *, struct mbuf *);
175 1.121 mrg static void unp_mark(file_t *);
176 1.121 mrg static void unp_scan(struct mbuf *, void (*)(file_t *), int);
177 1.121 mrg static void unp_discard_now(file_t *);
178 1.121 mrg static void unp_discard_later(file_t *);
179 1.121 mrg static void unp_thread(void *);
180 1.121 mrg static void unp_thread_kick(void);
181 1.112 ad static kmutex_t *uipc_lock;
182 1.112 ad
183 1.121 mrg static kcondvar_t unp_thread_cv;
184 1.121 mrg static lwp_t *unp_thread_lwp;
185 1.121 mrg static SLIST_HEAD(,file) unp_thread_discard;
186 1.121 mrg static int unp_defer;
187 1.121 mrg
188 1.112 ad /*
189 1.112 ad * Initialize Unix protocols.
190 1.112 ad */
191 1.112 ad void
192 1.112 ad uipc_init(void)
193 1.112 ad {
194 1.121 mrg int error;
195 1.112 ad
196 1.112 ad uipc_lock = mutex_obj_alloc(MUTEX_DEFAULT, IPL_NONE);
197 1.121 mrg cv_init(&unp_thread_cv, "unpgc");
198 1.121 mrg
199 1.121 mrg error = kthread_create(PRI_NONE, KTHREAD_MPSAFE, NULL, unp_thread,
200 1.121 mrg NULL, &unp_thread_lwp, "unpgc");
201 1.121 mrg if (error != 0)
202 1.121 mrg panic("uipc_init %d", error);
203 1.112 ad }
204 1.112 ad
205 1.112 ad /*
206 1.112 ad * A connection succeeded: disassociate both endpoints from the head's
207 1.112 ad * lock, and make them share their own lock. There is a race here: for
208 1.112 ad * a very brief time one endpoint will be locked by a different lock
209 1.112 ad * than the other end. However, since the current thread holds the old
210 1.112 ad * lock (the listening socket's lock, the head) access can still only be
211 1.112 ad * made to one side of the connection.
212 1.112 ad */
213 1.112 ad static void
214 1.112 ad unp_setpeerlocks(struct socket *so, struct socket *so2)
215 1.112 ad {
216 1.112 ad struct unpcb *unp;
217 1.112 ad kmutex_t *lock;
218 1.112 ad
219 1.112 ad KASSERT(solocked2(so, so2));
220 1.112 ad
221 1.112 ad /*
222 1.112 ad * Bail out if either end of the socket is not yet fully
223 1.112 ad * connected or accepted. We only break the lock association
224 1.112 ad * with the head when the pair of sockets stand completely
225 1.112 ad * on their own.
226 1.112 ad */
227 1.125 yamt KASSERT(so->so_head == NULL);
228 1.125 yamt if (so2->so_head != NULL)
229 1.112 ad return;
230 1.112 ad
231 1.112 ad /*
232 1.112 ad * Drop references to old lock. A third reference (from the
233 1.112 ad * queue head) must be held as we still hold its lock. Bonus:
234 1.112 ad * we don't need to worry about garbage collecting the lock.
235 1.112 ad */
236 1.112 ad lock = so->so_lock;
237 1.112 ad KASSERT(lock == uipc_lock);
238 1.112 ad mutex_obj_free(lock);
239 1.112 ad mutex_obj_free(lock);
240 1.112 ad
241 1.112 ad /*
242 1.112 ad * Grab stream lock from the initiator and share between the two
243 1.112 ad * endpoints. Issue memory barrier to ensure all modifications
244 1.112 ad * become globally visible before the lock change. so2 is
245 1.112 ad * assumed not to have a stream lock, because it was created
246 1.112 ad * purely for the server side to accept this connection and
247 1.112 ad * started out life using the domain-wide lock.
248 1.112 ad */
249 1.112 ad unp = sotounpcb(so);
250 1.112 ad KASSERT(unp->unp_streamlock != NULL);
251 1.112 ad KASSERT(sotounpcb(so2)->unp_streamlock == NULL);
252 1.112 ad lock = unp->unp_streamlock;
253 1.112 ad unp->unp_streamlock = NULL;
254 1.112 ad mutex_obj_hold(lock);
255 1.112 ad membar_exit();
256 1.127 bouyer /*
257 1.127 bouyer * possible race if lock is not held - see comment in
258 1.127 bouyer * uipc_usrreq(PRU_ACCEPT).
259 1.127 bouyer */
260 1.127 bouyer KASSERT(mutex_owned(lock));
261 1.115 ad solockreset(so, lock);
262 1.115 ad solockreset(so2, lock);
263 1.112 ad }
264 1.112 ad
265 1.112 ad /*
266 1.112 ad * Reset a socket's lock back to the domain-wide lock.
267 1.112 ad */
268 1.112 ad static void
269 1.112 ad unp_resetlock(struct socket *so)
270 1.112 ad {
271 1.112 ad kmutex_t *olock, *nlock;
272 1.112 ad struct unpcb *unp;
273 1.112 ad
274 1.112 ad KASSERT(solocked(so));
275 1.112 ad
276 1.112 ad olock = so->so_lock;
277 1.112 ad nlock = uipc_lock;
278 1.112 ad if (olock == nlock)
279 1.112 ad return;
280 1.112 ad unp = sotounpcb(so);
281 1.112 ad KASSERT(unp->unp_streamlock == NULL);
282 1.112 ad unp->unp_streamlock = olock;
283 1.112 ad mutex_obj_hold(nlock);
284 1.112 ad mutex_enter(nlock);
285 1.115 ad solockreset(so, nlock);
286 1.112 ad mutex_exit(olock);
287 1.112 ad }
288 1.112 ad
289 1.112 ad static void
290 1.112 ad unp_free(struct unpcb *unp)
291 1.112 ad {
292 1.112 ad if (unp->unp_addr)
293 1.112 ad free(unp->unp_addr, M_SONAME);
294 1.112 ad if (unp->unp_streamlock != NULL)
295 1.112 ad mutex_obj_free(unp->unp_streamlock);
296 1.152 rmind kmem_free(unp, sizeof(*unp));
297 1.112 ad }
298 1.30 thorpej
299 1.20 mycroft int
300 1.76 matt unp_output(struct mbuf *m, struct mbuf *control, struct unpcb *unp,
301 1.92 ad struct lwp *l)
302 1.20 mycroft {
303 1.20 mycroft struct socket *so2;
304 1.77 matt const struct sockaddr_un *sun;
305 1.20 mycroft
306 1.153 christos /* XXX: server side closed the socket */
307 1.153 christos if (unp->unp_conn == NULL)
308 1.153 christos return ECONNREFUSED;
309 1.20 mycroft so2 = unp->unp_conn->unp_socket;
310 1.112 ad
311 1.112 ad KASSERT(solocked(so2));
312 1.112 ad
313 1.20 mycroft if (unp->unp_addr)
314 1.20 mycroft sun = unp->unp_addr;
315 1.20 mycroft else
316 1.20 mycroft sun = &sun_noname;
317 1.30 thorpej if (unp->unp_conn->unp_flags & UNP_WANTCRED)
318 1.92 ad control = unp_addsockcred(l, control);
319 1.82 christos if (sbappendaddr(&so2->so_rcv, (const struct sockaddr *)sun, m,
320 1.20 mycroft control) == 0) {
321 1.112 ad so2->so_rcv.sb_overflowed++;
322 1.98 martin unp_dispose(control);
323 1.20 mycroft m_freem(control);
324 1.20 mycroft m_freem(m);
325 1.60 christos return (ENOBUFS);
326 1.20 mycroft } else {
327 1.20 mycroft sorwakeup(so2);
328 1.20 mycroft return (0);
329 1.20 mycroft }
330 1.20 mycroft }
331 1.20 mycroft
332 1.20 mycroft void
333 1.112 ad unp_setaddr(struct socket *so, struct mbuf *nam, bool peeraddr)
334 1.20 mycroft {
335 1.77 matt const struct sockaddr_un *sun;
336 1.112 ad struct unpcb *unp;
337 1.112 ad bool ext;
338 1.20 mycroft
339 1.127 bouyer KASSERT(solocked(so));
340 1.112 ad unp = sotounpcb(so);
341 1.112 ad ext = false;
342 1.20 mycroft
343 1.112 ad for (;;) {
344 1.112 ad sun = NULL;
345 1.112 ad if (peeraddr) {
346 1.112 ad if (unp->unp_conn && unp->unp_conn->unp_addr)
347 1.112 ad sun = unp->unp_conn->unp_addr;
348 1.112 ad } else {
349 1.112 ad if (unp->unp_addr)
350 1.112 ad sun = unp->unp_addr;
351 1.112 ad }
352 1.112 ad if (sun == NULL)
353 1.112 ad sun = &sun_noname;
354 1.112 ad nam->m_len = sun->sun_len;
355 1.112 ad if (nam->m_len > MLEN && !ext) {
356 1.112 ad sounlock(so);
357 1.112 ad MEXTMALLOC(nam, MAXPATHLEN * 2, M_WAITOK);
358 1.112 ad solock(so);
359 1.112 ad ext = true;
360 1.112 ad } else {
361 1.112 ad KASSERT(nam->m_len <= MAXPATHLEN * 2);
362 1.112 ad memcpy(mtod(nam, void *), sun, (size_t)nam->m_len);
363 1.112 ad break;
364 1.112 ad }
365 1.112 ad }
366 1.20 mycroft }
367 1.20 mycroft
368 1.151 rmind static int
369 1.151 rmind unp_usrreq(struct socket *so, int req, struct mbuf *m, struct mbuf *nam,
370 1.151 rmind struct mbuf *control, struct lwp *l)
371 1.1 cgd {
372 1.152 rmind struct unpcb *unp;
373 1.46 augustss struct socket *so2;
374 1.75 christos u_int newhiwat;
375 1.46 augustss int error = 0;
376 1.1 cgd
377 1.152 rmind KASSERT(req != PRU_ATTACH);
378 1.152 rmind KASSERT(req != PRU_DETACH);
379 1.154 rtr KASSERT(req != PRU_CONTROL);
380 1.152 rmind
381 1.152 rmind KASSERT(solocked(so));
382 1.152 rmind unp = sotounpcb(so);
383 1.20 mycroft
384 1.152 rmind KASSERT(!control || (req == PRU_SEND || req == PRU_SENDOOB));
385 1.152 rmind if (unp == NULL) {
386 1.152 rmind error = EINVAL;
387 1.152 rmind goto release;
388 1.1 cgd }
389 1.20 mycroft
390 1.1 cgd switch (req) {
391 1.1 cgd case PRU_BIND:
392 1.90 christos KASSERT(l != NULL);
393 1.112 ad error = unp_bind(so, nam, l);
394 1.1 cgd break;
395 1.1 cgd
396 1.1 cgd case PRU_LISTEN:
397 1.112 ad /*
398 1.112 ad * If the socket can accept a connection, it must be
399 1.112 ad * locked by uipc_lock.
400 1.112 ad */
401 1.112 ad unp_resetlock(so);
402 1.122 yamt if (unp->unp_vnode == NULL)
403 1.1 cgd error = EINVAL;
404 1.1 cgd break;
405 1.1 cgd
406 1.1 cgd case PRU_CONNECT:
407 1.90 christos KASSERT(l != NULL);
408 1.86 christos error = unp_connect(so, nam, l);
409 1.1 cgd break;
410 1.1 cgd
411 1.1 cgd case PRU_CONNECT2:
412 1.72 matt error = unp_connect2(so, (struct socket *)nam, PRU_CONNECT2);
413 1.1 cgd break;
414 1.1 cgd
415 1.1 cgd case PRU_DISCONNECT:
416 1.1 cgd unp_disconnect(unp);
417 1.1 cgd break;
418 1.1 cgd
419 1.1 cgd case PRU_ACCEPT:
420 1.112 ad KASSERT(so->so_lock == uipc_lock);
421 1.72 matt /*
422 1.72 matt * Mark the initiating STREAM socket as connected *ONLY*
423 1.72 matt * after it's been accepted. This prevents a client from
424 1.72 matt * overrunning a server and receiving ECONNREFUSED.
425 1.72 matt */
426 1.145 christos if (unp->unp_conn == NULL) {
427 1.145 christos /*
428 1.145 christos * This will use the empty socket and will not
429 1.145 christos * allocate.
430 1.145 christos */
431 1.145 christos unp_setaddr(so, nam, true);
432 1.112 ad break;
433 1.145 christos }
434 1.112 ad so2 = unp->unp_conn->unp_socket;
435 1.112 ad if (so2->so_state & SS_ISCONNECTING) {
436 1.112 ad KASSERT(solocked2(so, so->so_head));
437 1.112 ad KASSERT(solocked2(so2, so->so_head));
438 1.112 ad soisconnected(so2);
439 1.112 ad }
440 1.112 ad /*
441 1.112 ad * If the connection is fully established, break the
442 1.112 ad * association with uipc_lock and give the connected
443 1.152 rmind * pair a separate lock to share.
444 1.127 bouyer * There is a race here: sotounpcb(so2)->unp_streamlock
445 1.127 bouyer * is not locked, so when changing so2->so_lock
446 1.127 bouyer * another thread can grab it while so->so_lock is still
447 1.127 bouyer * pointing to the (locked) uipc_lock.
448 1.129 wiz * this should be harmless, except that this makes
449 1.127 bouyer * solocked2() and solocked() unreliable.
450 1.127 bouyer * Another problem is that unp_setaddr() expects the
451 1.127 bouyer * the socket locked. Grabing sotounpcb(so2)->unp_streamlock
452 1.127 bouyer * fixes both issues.
453 1.112 ad */
454 1.127 bouyer mutex_enter(sotounpcb(so2)->unp_streamlock);
455 1.112 ad unp_setpeerlocks(so2, so);
456 1.112 ad /*
457 1.112 ad * Only now return peer's address, as we may need to
458 1.112 ad * block in order to allocate memory.
459 1.112 ad *
460 1.112 ad * XXX Minor race: connection can be broken while
461 1.112 ad * lock is dropped in unp_setaddr(). We will return
462 1.112 ad * error == 0 and sun_noname as the peer address.
463 1.112 ad */
464 1.112 ad unp_setaddr(so, nam, true);
465 1.127 bouyer /* so_lock now points to unp_streamlock */
466 1.127 bouyer mutex_exit(so2->so_lock);
467 1.1 cgd break;
468 1.1 cgd
469 1.1 cgd case PRU_SHUTDOWN:
470 1.1 cgd socantsendmore(so);
471 1.1 cgd unp_shutdown(unp);
472 1.1 cgd break;
473 1.1 cgd
474 1.1 cgd case PRU_RCVD:
475 1.1 cgd switch (so->so_type) {
476 1.1 cgd
477 1.1 cgd case SOCK_DGRAM:
478 1.1 cgd panic("uipc 1");
479 1.1 cgd /*NOTREACHED*/
480 1.1 cgd
481 1.134 manu case SOCK_SEQPACKET: /* FALLTHROUGH */
482 1.1 cgd case SOCK_STREAM:
483 1.1 cgd #define rcv (&so->so_rcv)
484 1.1 cgd #define snd (&so2->so_snd)
485 1.1 cgd if (unp->unp_conn == 0)
486 1.1 cgd break;
487 1.1 cgd so2 = unp->unp_conn->unp_socket;
488 1.112 ad KASSERT(solocked2(so, so2));
489 1.1 cgd /*
490 1.1 cgd * Adjust backpressure on sender
491 1.1 cgd * and wakeup any waiting to write.
492 1.1 cgd */
493 1.1 cgd snd->sb_mbmax += unp->unp_mbcnt - rcv->sb_mbcnt;
494 1.1 cgd unp->unp_mbcnt = rcv->sb_mbcnt;
495 1.75 christos newhiwat = snd->sb_hiwat + unp->unp_cc - rcv->sb_cc;
496 1.81 christos (void)chgsbsize(so2->so_uidinfo,
497 1.75 christos &snd->sb_hiwat, newhiwat, RLIM_INFINITY);
498 1.1 cgd unp->unp_cc = rcv->sb_cc;
499 1.1 cgd sowwakeup(so2);
500 1.1 cgd #undef snd
501 1.1 cgd #undef rcv
502 1.1 cgd break;
503 1.1 cgd
504 1.1 cgd default:
505 1.1 cgd panic("uipc 2");
506 1.1 cgd }
507 1.1 cgd break;
508 1.1 cgd
509 1.1 cgd case PRU_SEND:
510 1.30 thorpej /*
511 1.30 thorpej * Note: unp_internalize() rejects any control message
512 1.30 thorpej * other than SCM_RIGHTS, and only allows one. This
513 1.30 thorpej * has the side-effect of preventing a caller from
514 1.30 thorpej * forging SCM_CREDS.
515 1.30 thorpej */
516 1.90 christos if (control) {
517 1.112 ad sounlock(so);
518 1.112 ad error = unp_internalize(&control);
519 1.112 ad solock(so);
520 1.112 ad if (error != 0) {
521 1.111 mlelstv m_freem(control);
522 1.111 mlelstv m_freem(m);
523 1.111 mlelstv break;
524 1.111 mlelstv }
525 1.83 yamt }
526 1.1 cgd switch (so->so_type) {
527 1.1 cgd
528 1.1 cgd case SOCK_DGRAM: {
529 1.112 ad KASSERT(so->so_lock == uipc_lock);
530 1.1 cgd if (nam) {
531 1.111 mlelstv if ((so->so_state & SS_ISCONNECTED) != 0)
532 1.1 cgd error = EISCONN;
533 1.111 mlelstv else {
534 1.112 ad /*
535 1.112 ad * Note: once connected, the
536 1.112 ad * socket's lock must not be
537 1.112 ad * dropped until we have sent
538 1.112 ad * the message and disconnected.
539 1.112 ad * This is necessary to prevent
540 1.112 ad * intervening control ops, like
541 1.112 ad * another connection.
542 1.112 ad */
543 1.111 mlelstv error = unp_connect(so, nam, l);
544 1.20 mycroft }
545 1.1 cgd } else {
546 1.111 mlelstv if ((so->so_state & SS_ISCONNECTED) == 0)
547 1.1 cgd error = ENOTCONN;
548 1.111 mlelstv }
549 1.111 mlelstv if (error) {
550 1.111 mlelstv unp_dispose(control);
551 1.111 mlelstv m_freem(control);
552 1.111 mlelstv m_freem(m);
553 1.111 mlelstv break;
554 1.1 cgd }
555 1.147 martin KASSERT(l != NULL);
556 1.92 ad error = unp_output(m, control, unp, l);
557 1.1 cgd if (nam)
558 1.1 cgd unp_disconnect(unp);
559 1.1 cgd break;
560 1.1 cgd }
561 1.1 cgd
562 1.134 manu case SOCK_SEQPACKET: /* FALLTHROUGH */
563 1.1 cgd case SOCK_STREAM:
564 1.1 cgd #define rcv (&so2->so_rcv)
565 1.1 cgd #define snd (&so->so_snd)
566 1.87 christos if (unp->unp_conn == NULL) {
567 1.87 christos error = ENOTCONN;
568 1.87 christos break;
569 1.87 christos }
570 1.1 cgd so2 = unp->unp_conn->unp_socket;
571 1.112 ad KASSERT(solocked2(so, so2));
572 1.30 thorpej if (unp->unp_conn->unp_flags & UNP_WANTCRED) {
573 1.30 thorpej /*
574 1.30 thorpej * Credentials are passed only once on
575 1.134 manu * SOCK_STREAM and SOCK_SEQPACKET.
576 1.30 thorpej */
577 1.30 thorpej unp->unp_conn->unp_flags &= ~UNP_WANTCRED;
578 1.92 ad control = unp_addsockcred(l, control);
579 1.30 thorpej }
580 1.1 cgd /*
581 1.1 cgd * Send to paired receive port, and then reduce
582 1.1 cgd * send buffer hiwater marks to maintain backpressure.
583 1.1 cgd * Wake up readers.
584 1.1 cgd */
585 1.1 cgd if (control) {
586 1.112 ad if (sbappendcontrol(rcv, m, control) != 0)
587 1.112 ad control = NULL;
588 1.134 manu } else {
589 1.134 manu switch(so->so_type) {
590 1.134 manu case SOCK_SEQPACKET:
591 1.134 manu sbappendrecord(rcv, m);
592 1.134 manu break;
593 1.134 manu case SOCK_STREAM:
594 1.134 manu sbappend(rcv, m);
595 1.134 manu break;
596 1.134 manu default:
597 1.134 manu panic("uipc_usrreq");
598 1.134 manu break;
599 1.134 manu }
600 1.134 manu }
601 1.1 cgd snd->sb_mbmax -=
602 1.1 cgd rcv->sb_mbcnt - unp->unp_conn->unp_mbcnt;
603 1.1 cgd unp->unp_conn->unp_mbcnt = rcv->sb_mbcnt;
604 1.75 christos newhiwat = snd->sb_hiwat -
605 1.75 christos (rcv->sb_cc - unp->unp_conn->unp_cc);
606 1.81 christos (void)chgsbsize(so->so_uidinfo,
607 1.75 christos &snd->sb_hiwat, newhiwat, RLIM_INFINITY);
608 1.1 cgd unp->unp_conn->unp_cc = rcv->sb_cc;
609 1.1 cgd sorwakeup(so2);
610 1.1 cgd #undef snd
611 1.1 cgd #undef rcv
612 1.112 ad if (control != NULL) {
613 1.112 ad unp_dispose(control);
614 1.112 ad m_freem(control);
615 1.112 ad }
616 1.1 cgd break;
617 1.1 cgd
618 1.1 cgd default:
619 1.1 cgd panic("uipc 4");
620 1.1 cgd }
621 1.1 cgd break;
622 1.1 cgd
623 1.1 cgd case PRU_ABORT:
624 1.112 ad (void)unp_drop(unp, ECONNABORTED);
625 1.88 matt KASSERT(so->so_head == NULL);
626 1.152 rmind KASSERT(so->so_pcb != NULL);
627 1.152 rmind unp_detach(so);
628 1.1 cgd break;
629 1.1 cgd
630 1.1 cgd case PRU_SENSE:
631 1.1 cgd ((struct stat *) m)->st_blksize = so->so_snd.sb_hiwat;
632 1.134 manu switch (so->so_type) {
633 1.134 manu case SOCK_SEQPACKET: /* FALLTHROUGH */
634 1.134 manu case SOCK_STREAM:
635 1.134 manu if (unp->unp_conn == 0)
636 1.134 manu break;
637 1.134 manu
638 1.1 cgd so2 = unp->unp_conn->unp_socket;
639 1.112 ad KASSERT(solocked2(so, so2));
640 1.1 cgd ((struct stat *) m)->st_blksize += so2->so_rcv.sb_cc;
641 1.134 manu break;
642 1.134 manu default:
643 1.134 manu break;
644 1.1 cgd }
645 1.1 cgd ((struct stat *) m)->st_dev = NODEV;
646 1.1 cgd if (unp->unp_ino == 0)
647 1.1 cgd unp->unp_ino = unp_ino++;
648 1.25 kleink ((struct stat *) m)->st_atimespec =
649 1.25 kleink ((struct stat *) m)->st_mtimespec =
650 1.25 kleink ((struct stat *) m)->st_ctimespec = unp->unp_ctime;
651 1.1 cgd ((struct stat *) m)->st_ino = unp->unp_ino;
652 1.1 cgd return (0);
653 1.1 cgd
654 1.1 cgd case PRU_RCVOOB:
655 1.20 mycroft error = EOPNOTSUPP;
656 1.20 mycroft break;
657 1.1 cgd
658 1.1 cgd case PRU_SENDOOB:
659 1.22 mycroft m_freem(control);
660 1.20 mycroft m_freem(m);
661 1.1 cgd error = EOPNOTSUPP;
662 1.1 cgd break;
663 1.1 cgd
664 1.1 cgd case PRU_SOCKADDR:
665 1.112 ad unp_setaddr(so, nam, false);
666 1.1 cgd break;
667 1.1 cgd
668 1.1 cgd case PRU_PEERADDR:
669 1.112 ad unp_setaddr(so, nam, true);
670 1.1 cgd break;
671 1.1 cgd
672 1.1 cgd default:
673 1.1 cgd panic("piusrreq");
674 1.1 cgd }
675 1.20 mycroft
676 1.1 cgd release:
677 1.1 cgd return (error);
678 1.1 cgd }
679 1.1 cgd
680 1.1 cgd /*
681 1.30 thorpej * Unix domain socket option processing.
682 1.30 thorpej */
683 1.30 thorpej int
684 1.118 plunky uipc_ctloutput(int op, struct socket *so, struct sockopt *sopt)
685 1.30 thorpej {
686 1.30 thorpej struct unpcb *unp = sotounpcb(so);
687 1.30 thorpej int optval = 0, error = 0;
688 1.30 thorpej
689 1.112 ad KASSERT(solocked(so));
690 1.112 ad
691 1.118 plunky if (sopt->sopt_level != 0) {
692 1.100 dyoung error = ENOPROTOOPT;
693 1.30 thorpej } else switch (op) {
694 1.30 thorpej
695 1.30 thorpej case PRCO_SETOPT:
696 1.118 plunky switch (sopt->sopt_name) {
697 1.30 thorpej case LOCAL_CREDS:
698 1.72 matt case LOCAL_CONNWAIT:
699 1.118 plunky error = sockopt_getint(sopt, &optval);
700 1.118 plunky if (error)
701 1.118 plunky break;
702 1.118 plunky switch (sopt->sopt_name) {
703 1.30 thorpej #define OPTSET(bit) \
704 1.30 thorpej if (optval) \
705 1.30 thorpej unp->unp_flags |= (bit); \
706 1.30 thorpej else \
707 1.30 thorpej unp->unp_flags &= ~(bit);
708 1.30 thorpej
709 1.118 plunky case LOCAL_CREDS:
710 1.118 plunky OPTSET(UNP_WANTCRED);
711 1.118 plunky break;
712 1.118 plunky case LOCAL_CONNWAIT:
713 1.118 plunky OPTSET(UNP_CONNWAIT);
714 1.118 plunky break;
715 1.30 thorpej }
716 1.30 thorpej break;
717 1.30 thorpej #undef OPTSET
718 1.30 thorpej
719 1.30 thorpej default:
720 1.30 thorpej error = ENOPROTOOPT;
721 1.30 thorpej break;
722 1.30 thorpej }
723 1.30 thorpej break;
724 1.30 thorpej
725 1.30 thorpej case PRCO_GETOPT:
726 1.112 ad sounlock(so);
727 1.118 plunky switch (sopt->sopt_name) {
728 1.99 he case LOCAL_PEEREID:
729 1.99 he if (unp->unp_flags & UNP_EIDSVALID) {
730 1.118 plunky error = sockopt_set(sopt,
731 1.118 plunky &unp->unp_connid, sizeof(unp->unp_connid));
732 1.99 he } else {
733 1.99 he error = EINVAL;
734 1.99 he }
735 1.99 he break;
736 1.30 thorpej case LOCAL_CREDS:
737 1.30 thorpej #define OPTBIT(bit) (unp->unp_flags & (bit) ? 1 : 0)
738 1.30 thorpej
739 1.99 he optval = OPTBIT(UNP_WANTCRED);
740 1.118 plunky error = sockopt_setint(sopt, optval);
741 1.30 thorpej break;
742 1.30 thorpej #undef OPTBIT
743 1.30 thorpej
744 1.30 thorpej default:
745 1.30 thorpej error = ENOPROTOOPT;
746 1.30 thorpej break;
747 1.30 thorpej }
748 1.112 ad solock(so);
749 1.30 thorpej break;
750 1.30 thorpej }
751 1.30 thorpej return (error);
752 1.30 thorpej }
753 1.30 thorpej
754 1.30 thorpej /*
755 1.1 cgd * Both send and receive buffers are allocated PIPSIZ bytes of buffering
756 1.1 cgd * for stream sockets, although the total for sender and receiver is
757 1.1 cgd * actually only PIPSIZ.
758 1.1 cgd * Datagram sockets really use the sendspace as the maximum datagram size,
759 1.1 cgd * and don't really want to reserve the sendspace. Their recvspace should
760 1.1 cgd * be large enough for at least one max-size datagram plus address.
761 1.1 cgd */
762 1.1 cgd #define PIPSIZ 4096
763 1.1 cgd u_long unpst_sendspace = PIPSIZ;
764 1.1 cgd u_long unpst_recvspace = PIPSIZ;
765 1.1 cgd u_long unpdg_sendspace = 2*1024; /* really max datagram size */
766 1.1 cgd u_long unpdg_recvspace = 4*1024;
767 1.1 cgd
768 1.121 mrg u_int unp_rights; /* files in flight */
769 1.121 mrg u_int unp_rights_ratio = 2; /* limit, fraction of maxfiles */
770 1.1 cgd
771 1.152 rmind static int
772 1.152 rmind unp_attach(struct socket *so, int proto)
773 1.1 cgd {
774 1.152 rmind struct unpcb *unp = sotounpcb(so);
775 1.152 rmind u_long sndspc, rcvspc;
776 1.1 cgd int error;
777 1.80 perry
778 1.152 rmind KASSERT(unp == NULL);
779 1.152 rmind
780 1.112 ad switch (so->so_type) {
781 1.152 rmind case SOCK_SEQPACKET:
782 1.152 rmind /* FALLTHROUGH */
783 1.112 ad case SOCK_STREAM:
784 1.112 ad if (so->so_lock == NULL) {
785 1.112 ad so->so_lock = mutex_obj_alloc(MUTEX_DEFAULT, IPL_NONE);
786 1.112 ad solock(so);
787 1.112 ad }
788 1.152 rmind sndspc = unpst_sendspace;
789 1.152 rmind rcvspc = unpst_recvspace;
790 1.112 ad break;
791 1.1 cgd
792 1.112 ad case SOCK_DGRAM:
793 1.112 ad if (so->so_lock == NULL) {
794 1.112 ad mutex_obj_hold(uipc_lock);
795 1.112 ad so->so_lock = uipc_lock;
796 1.112 ad solock(so);
797 1.112 ad }
798 1.152 rmind sndspc = unpdg_sendspace;
799 1.152 rmind rcvspc = unpdg_recvspace;
800 1.112 ad break;
801 1.8 mycroft
802 1.112 ad default:
803 1.112 ad panic("unp_attach");
804 1.1 cgd }
805 1.152 rmind
806 1.152 rmind if (so->so_snd.sb_hiwat == 0 || so->so_rcv.sb_hiwat == 0) {
807 1.152 rmind error = soreserve(so, sndspc, rcvspc);
808 1.152 rmind if (error) {
809 1.152 rmind return error;
810 1.152 rmind }
811 1.152 rmind }
812 1.152 rmind
813 1.152 rmind unp = kmem_zalloc(sizeof(*unp), KM_SLEEP);
814 1.152 rmind nanotime(&unp->unp_ctime);
815 1.14 mycroft unp->unp_socket = so;
816 1.15 mycroft so->so_pcb = unp;
817 1.152 rmind
818 1.152 rmind KASSERT(solocked(so));
819 1.152 rmind return 0;
820 1.1 cgd }
821 1.1 cgd
822 1.152 rmind static void
823 1.152 rmind unp_detach(struct socket *so)
824 1.1 cgd {
825 1.152 rmind struct unpcb *unp;
826 1.112 ad vnode_t *vp;
827 1.112 ad
828 1.152 rmind unp = sotounpcb(so);
829 1.152 rmind KASSERT(unp != NULL);
830 1.152 rmind KASSERT(solocked(so));
831 1.112 ad retry:
832 1.112 ad if ((vp = unp->unp_vnode) != NULL) {
833 1.112 ad sounlock(so);
834 1.112 ad /* Acquire v_interlock to protect against unp_connect(). */
835 1.113 ad /* XXXAD racy */
836 1.135 rmind mutex_enter(vp->v_interlock);
837 1.112 ad vp->v_socket = NULL;
838 1.148 hannken mutex_exit(vp->v_interlock);
839 1.148 hannken vrele(vp);
840 1.112 ad solock(so);
841 1.112 ad unp->unp_vnode = NULL;
842 1.1 cgd }
843 1.1 cgd if (unp->unp_conn)
844 1.1 cgd unp_disconnect(unp);
845 1.112 ad while (unp->unp_refs) {
846 1.112 ad KASSERT(solocked2(so, unp->unp_refs->unp_socket));
847 1.112 ad if (unp_drop(unp->unp_refs, ECONNRESET)) {
848 1.112 ad solock(so);
849 1.112 ad goto retry;
850 1.112 ad }
851 1.112 ad }
852 1.112 ad soisdisconnected(so);
853 1.112 ad so->so_pcb = NULL;
854 1.8 mycroft if (unp_rights) {
855 1.8 mycroft /*
856 1.121 mrg * Normally the receive buffer is flushed later, in sofree,
857 1.121 mrg * but if our receive buffer holds references to files that
858 1.121 mrg * are now garbage, we will enqueue those file references to
859 1.121 mrg * the garbage collector and kick it into action.
860 1.8 mycroft */
861 1.112 ad sorflush(so);
862 1.112 ad unp_free(unp);
863 1.121 mrg unp_thread_kick();
864 1.14 mycroft } else
865 1.112 ad unp_free(unp);
866 1.1 cgd }
867 1.1 cgd
868 1.154 rtr static int
869 1.154 rtr unp_ioctl(struct socket *so, struct mbuf *m, struct mbuf *nam,
870 1.154 rtr struct mbuf *control, struct lwp *l)
871 1.154 rtr {
872 1.154 rtr return EOPNOTSUPP;
873 1.154 rtr }
874 1.154 rtr
875 1.146 christos /*
876 1.146 christos * Allocate the new sockaddr. We have to allocate one
877 1.146 christos * extra byte so that we can ensure that the pathname
878 1.146 christos * is nul-terminated. Note that unlike linux, we don't
879 1.146 christos * include in the address length the NUL in the path
880 1.146 christos * component, because doing so, would exceed sizeof(sockaddr_un)
881 1.146 christos * for fully occupied pathnames. Linux is also inconsistent,
882 1.146 christos * because it does not include the NUL in the length of
883 1.146 christos * what it calls "abstract" unix sockets.
884 1.146 christos */
885 1.146 christos static struct sockaddr_un *
886 1.146 christos makeun(struct mbuf *nam, size_t *addrlen) {
887 1.146 christos struct sockaddr_un *sun;
888 1.146 christos
889 1.146 christos *addrlen = nam->m_len + 1;
890 1.146 christos sun = malloc(*addrlen, M_SONAME, M_WAITOK);
891 1.146 christos m_copydata(nam, 0, nam->m_len, (void *)sun);
892 1.146 christos *(((char *)sun) + nam->m_len) = '\0';
893 1.146 christos sun->sun_len = strlen(sun->sun_path) +
894 1.146 christos offsetof(struct sockaddr_un, sun_path);
895 1.146 christos return sun;
896 1.146 christos }
897 1.146 christos
898 1.5 andrew int
899 1.112 ad unp_bind(struct socket *so, struct mbuf *nam, struct lwp *l)
900 1.1 cgd {
901 1.27 thorpej struct sockaddr_un *sun;
902 1.112 ad struct unpcb *unp;
903 1.106 ad vnode_t *vp;
904 1.1 cgd struct vattr vattr;
905 1.27 thorpej size_t addrlen;
906 1.1 cgd int error;
907 1.133 dholland struct pathbuf *pb;
908 1.1 cgd struct nameidata nd;
909 1.112 ad proc_t *p;
910 1.1 cgd
911 1.112 ad unp = sotounpcb(so);
912 1.112 ad if (unp->unp_vnode != NULL)
913 1.20 mycroft return (EINVAL);
914 1.109 ad if ((unp->unp_flags & UNP_BUSY) != 0) {
915 1.109 ad /*
916 1.109 ad * EALREADY may not be strictly accurate, but since this
917 1.109 ad * is a major application error it's hardly a big deal.
918 1.109 ad */
919 1.109 ad return (EALREADY);
920 1.109 ad }
921 1.109 ad unp->unp_flags |= UNP_BUSY;
922 1.112 ad sounlock(so);
923 1.109 ad
924 1.112 ad p = l->l_proc;
925 1.146 christos sun = makeun(nam, &addrlen);
926 1.27 thorpej
927 1.133 dholland pb = pathbuf_create(sun->sun_path);
928 1.133 dholland if (pb == NULL) {
929 1.133 dholland error = ENOMEM;
930 1.133 dholland goto bad;
931 1.133 dholland }
932 1.133 dholland NDINIT(&nd, CREATE, FOLLOW | LOCKPARENT | TRYEMULROOT, pb);
933 1.27 thorpej
934 1.1 cgd /* SHOULD BE ABLE TO ADOPT EXISTING AND wakeup() ALA FIFO's */
935 1.133 dholland if ((error = namei(&nd)) != 0) {
936 1.133 dholland pathbuf_destroy(pb);
937 1.27 thorpej goto bad;
938 1.133 dholland }
939 1.9 mycroft vp = nd.ni_vp;
940 1.96 hannken if (vp != NULL) {
941 1.9 mycroft VOP_ABORTOP(nd.ni_dvp, &nd.ni_cnd);
942 1.9 mycroft if (nd.ni_dvp == vp)
943 1.9 mycroft vrele(nd.ni_dvp);
944 1.1 cgd else
945 1.9 mycroft vput(nd.ni_dvp);
946 1.1 cgd vrele(vp);
947 1.133 dholland pathbuf_destroy(pb);
948 1.96 hannken error = EADDRINUSE;
949 1.96 hannken goto bad;
950 1.1 cgd }
951 1.128 pooka vattr_null(&vattr);
952 1.1 cgd vattr.va_type = VSOCK;
953 1.84 jmmv vattr.va_mode = ACCESSPERMS & ~(p->p_cwdi->cwdi_cmask);
954 1.16 christos error = VOP_CREATE(nd.ni_dvp, &nd.ni_vp, &nd.ni_cnd, &vattr);
955 1.133 dholland if (error) {
956 1.149 hannken vput(nd.ni_dvp);
957 1.133 dholland pathbuf_destroy(pb);
958 1.27 thorpej goto bad;
959 1.133 dholland }
960 1.9 mycroft vp = nd.ni_vp;
961 1.150 hannken vn_lock(vp, LK_EXCLUSIVE | LK_RETRY);
962 1.112 ad solock(so);
963 1.1 cgd vp->v_socket = unp->unp_socket;
964 1.1 cgd unp->unp_vnode = vp;
965 1.27 thorpej unp->unp_addrlen = addrlen;
966 1.27 thorpej unp->unp_addr = sun;
967 1.99 he unp->unp_connid.unp_pid = p->p_pid;
968 1.112 ad unp->unp_connid.unp_euid = kauth_cred_geteuid(l->l_cred);
969 1.112 ad unp->unp_connid.unp_egid = kauth_cred_getegid(l->l_cred);
970 1.99 he unp->unp_flags |= UNP_EIDSBIND;
971 1.130 hannken VOP_UNLOCK(vp);
972 1.149 hannken vput(nd.ni_dvp);
973 1.109 ad unp->unp_flags &= ~UNP_BUSY;
974 1.133 dholland pathbuf_destroy(pb);
975 1.1 cgd return (0);
976 1.27 thorpej
977 1.27 thorpej bad:
978 1.27 thorpej free(sun, M_SONAME);
979 1.112 ad solock(so);
980 1.109 ad unp->unp_flags &= ~UNP_BUSY;
981 1.27 thorpej return (error);
982 1.1 cgd }
983 1.1 cgd
984 1.5 andrew int
985 1.86 christos unp_connect(struct socket *so, struct mbuf *nam, struct lwp *l)
986 1.1 cgd {
987 1.46 augustss struct sockaddr_un *sun;
988 1.106 ad vnode_t *vp;
989 1.46 augustss struct socket *so2, *so3;
990 1.99 he struct unpcb *unp, *unp2, *unp3;
991 1.27 thorpej size_t addrlen;
992 1.1 cgd int error;
993 1.133 dholland struct pathbuf *pb;
994 1.1 cgd struct nameidata nd;
995 1.1 cgd
996 1.109 ad unp = sotounpcb(so);
997 1.109 ad if ((unp->unp_flags & UNP_BUSY) != 0) {
998 1.109 ad /*
999 1.109 ad * EALREADY may not be strictly accurate, but since this
1000 1.109 ad * is a major application error it's hardly a big deal.
1001 1.109 ad */
1002 1.109 ad return (EALREADY);
1003 1.109 ad }
1004 1.109 ad unp->unp_flags |= UNP_BUSY;
1005 1.112 ad sounlock(so);
1006 1.109 ad
1007 1.146 christos sun = makeun(nam, &addrlen);
1008 1.133 dholland pb = pathbuf_create(sun->sun_path);
1009 1.133 dholland if (pb == NULL) {
1010 1.133 dholland error = ENOMEM;
1011 1.133 dholland goto bad2;
1012 1.133 dholland }
1013 1.27 thorpej
1014 1.133 dholland NDINIT(&nd, LOOKUP, FOLLOW | LOCKLEAF | TRYEMULROOT, pb);
1015 1.133 dholland
1016 1.133 dholland if ((error = namei(&nd)) != 0) {
1017 1.133 dholland pathbuf_destroy(pb);
1018 1.27 thorpej goto bad2;
1019 1.133 dholland }
1020 1.9 mycroft vp = nd.ni_vp;
1021 1.1 cgd if (vp->v_type != VSOCK) {
1022 1.1 cgd error = ENOTSOCK;
1023 1.1 cgd goto bad;
1024 1.1 cgd }
1025 1.133 dholland pathbuf_destroy(pb);
1026 1.102 pooka if ((error = VOP_ACCESS(vp, VWRITE, l->l_cred)) != 0)
1027 1.1 cgd goto bad;
1028 1.112 ad /* Acquire v_interlock to protect against unp_detach(). */
1029 1.135 rmind mutex_enter(vp->v_interlock);
1030 1.1 cgd so2 = vp->v_socket;
1031 1.112 ad if (so2 == NULL) {
1032 1.135 rmind mutex_exit(vp->v_interlock);
1033 1.1 cgd error = ECONNREFUSED;
1034 1.1 cgd goto bad;
1035 1.1 cgd }
1036 1.1 cgd if (so->so_type != so2->so_type) {
1037 1.135 rmind mutex_exit(vp->v_interlock);
1038 1.1 cgd error = EPROTOTYPE;
1039 1.1 cgd goto bad;
1040 1.1 cgd }
1041 1.112 ad solock(so);
1042 1.112 ad unp_resetlock(so);
1043 1.135 rmind mutex_exit(vp->v_interlock);
1044 1.112 ad if ((so->so_proto->pr_flags & PR_CONNREQUIRED) != 0) {
1045 1.112 ad /*
1046 1.112 ad * This may seem somewhat fragile but is OK: if we can
1047 1.112 ad * see SO_ACCEPTCONN set on the endpoint, then it must
1048 1.112 ad * be locked by the domain-wide uipc_lock.
1049 1.112 ad */
1050 1.132 yamt KASSERT((so2->so_options & SO_ACCEPTCONN) == 0 ||
1051 1.112 ad so2->so_lock == uipc_lock);
1052 1.1 cgd if ((so2->so_options & SO_ACCEPTCONN) == 0 ||
1053 1.144 rmind (so3 = sonewconn(so2, false)) == NULL) {
1054 1.1 cgd error = ECONNREFUSED;
1055 1.112 ad sounlock(so);
1056 1.1 cgd goto bad;
1057 1.1 cgd }
1058 1.1 cgd unp2 = sotounpcb(so2);
1059 1.1 cgd unp3 = sotounpcb(so3);
1060 1.26 thorpej if (unp2->unp_addr) {
1061 1.26 thorpej unp3->unp_addr = malloc(unp2->unp_addrlen,
1062 1.26 thorpej M_SONAME, M_WAITOK);
1063 1.36 perry memcpy(unp3->unp_addr, unp2->unp_addr,
1064 1.26 thorpej unp2->unp_addrlen);
1065 1.26 thorpej unp3->unp_addrlen = unp2->unp_addrlen;
1066 1.26 thorpej }
1067 1.30 thorpej unp3->unp_flags = unp2->unp_flags;
1068 1.112 ad unp3->unp_connid.unp_pid = l->l_proc->p_pid;
1069 1.112 ad unp3->unp_connid.unp_euid = kauth_cred_geteuid(l->l_cred);
1070 1.112 ad unp3->unp_connid.unp_egid = kauth_cred_getegid(l->l_cred);
1071 1.99 he unp3->unp_flags |= UNP_EIDSVALID;
1072 1.99 he if (unp2->unp_flags & UNP_EIDSBIND) {
1073 1.99 he unp->unp_connid = unp2->unp_connid;
1074 1.99 he unp->unp_flags |= UNP_EIDSVALID;
1075 1.99 he }
1076 1.112 ad so2 = so3;
1077 1.33 thorpej }
1078 1.72 matt error = unp_connect2(so, so2, PRU_CONNECT);
1079 1.112 ad sounlock(so);
1080 1.27 thorpej bad:
1081 1.1 cgd vput(vp);
1082 1.27 thorpej bad2:
1083 1.27 thorpej free(sun, M_SONAME);
1084 1.112 ad solock(so);
1085 1.109 ad unp->unp_flags &= ~UNP_BUSY;
1086 1.1 cgd return (error);
1087 1.1 cgd }
1088 1.1 cgd
1089 1.5 andrew int
1090 1.76 matt unp_connect2(struct socket *so, struct socket *so2, int req)
1091 1.1 cgd {
1092 1.46 augustss struct unpcb *unp = sotounpcb(so);
1093 1.46 augustss struct unpcb *unp2;
1094 1.1 cgd
1095 1.1 cgd if (so2->so_type != so->so_type)
1096 1.1 cgd return (EPROTOTYPE);
1097 1.112 ad
1098 1.112 ad /*
1099 1.112 ad * All three sockets involved must be locked by same lock:
1100 1.112 ad *
1101 1.112 ad * local endpoint (so)
1102 1.112 ad * remote endpoint (so2)
1103 1.131 yamt * queue head (so2->so_head, only if PR_CONNREQUIRED)
1104 1.112 ad */
1105 1.112 ad KASSERT(solocked2(so, so2));
1106 1.125 yamt KASSERT(so->so_head == NULL);
1107 1.125 yamt if (so2->so_head != NULL) {
1108 1.125 yamt KASSERT(so2->so_lock == uipc_lock);
1109 1.125 yamt KASSERT(solocked2(so2, so2->so_head));
1110 1.112 ad }
1111 1.112 ad
1112 1.1 cgd unp2 = sotounpcb(so2);
1113 1.1 cgd unp->unp_conn = unp2;
1114 1.1 cgd switch (so->so_type) {
1115 1.1 cgd
1116 1.1 cgd case SOCK_DGRAM:
1117 1.1 cgd unp->unp_nextref = unp2->unp_refs;
1118 1.1 cgd unp2->unp_refs = unp;
1119 1.1 cgd soisconnected(so);
1120 1.1 cgd break;
1121 1.1 cgd
1122 1.134 manu case SOCK_SEQPACKET: /* FALLTHROUGH */
1123 1.1 cgd case SOCK_STREAM:
1124 1.1 cgd unp2->unp_conn = unp;
1125 1.72 matt if (req == PRU_CONNECT &&
1126 1.72 matt ((unp->unp_flags | unp2->unp_flags) & UNP_CONNWAIT))
1127 1.72 matt soisconnecting(so);
1128 1.72 matt else
1129 1.72 matt soisconnected(so);
1130 1.1 cgd soisconnected(so2);
1131 1.112 ad /*
1132 1.112 ad * If the connection is fully established, break the
1133 1.112 ad * association with uipc_lock and give the connected
1134 1.112 ad * pair a seperate lock to share. For CONNECT2, we
1135 1.112 ad * require that the locks already match (the sockets
1136 1.112 ad * are created that way).
1137 1.112 ad */
1138 1.125 yamt if (req == PRU_CONNECT) {
1139 1.125 yamt KASSERT(so2->so_head != NULL);
1140 1.112 ad unp_setpeerlocks(so, so2);
1141 1.125 yamt }
1142 1.1 cgd break;
1143 1.1 cgd
1144 1.1 cgd default:
1145 1.1 cgd panic("unp_connect2");
1146 1.1 cgd }
1147 1.1 cgd return (0);
1148 1.1 cgd }
1149 1.1 cgd
1150 1.5 andrew void
1151 1.76 matt unp_disconnect(struct unpcb *unp)
1152 1.1 cgd {
1153 1.46 augustss struct unpcb *unp2 = unp->unp_conn;
1154 1.112 ad struct socket *so;
1155 1.1 cgd
1156 1.1 cgd if (unp2 == 0)
1157 1.1 cgd return;
1158 1.1 cgd unp->unp_conn = 0;
1159 1.112 ad so = unp->unp_socket;
1160 1.112 ad switch (so->so_type) {
1161 1.1 cgd case SOCK_DGRAM:
1162 1.1 cgd if (unp2->unp_refs == unp)
1163 1.1 cgd unp2->unp_refs = unp->unp_nextref;
1164 1.1 cgd else {
1165 1.1 cgd unp2 = unp2->unp_refs;
1166 1.1 cgd for (;;) {
1167 1.112 ad KASSERT(solocked2(so, unp2->unp_socket));
1168 1.1 cgd if (unp2 == 0)
1169 1.1 cgd panic("unp_disconnect");
1170 1.1 cgd if (unp2->unp_nextref == unp)
1171 1.1 cgd break;
1172 1.1 cgd unp2 = unp2->unp_nextref;
1173 1.1 cgd }
1174 1.1 cgd unp2->unp_nextref = unp->unp_nextref;
1175 1.1 cgd }
1176 1.1 cgd unp->unp_nextref = 0;
1177 1.112 ad so->so_state &= ~SS_ISCONNECTED;
1178 1.1 cgd break;
1179 1.1 cgd
1180 1.134 manu case SOCK_SEQPACKET: /* FALLTHROUGH */
1181 1.1 cgd case SOCK_STREAM:
1182 1.112 ad KASSERT(solocked2(so, unp2->unp_socket));
1183 1.112 ad soisdisconnected(so);
1184 1.1 cgd unp2->unp_conn = 0;
1185 1.1 cgd soisdisconnected(unp2->unp_socket);
1186 1.1 cgd break;
1187 1.1 cgd }
1188 1.1 cgd }
1189 1.1 cgd
1190 1.5 andrew void
1191 1.76 matt unp_shutdown(struct unpcb *unp)
1192 1.1 cgd {
1193 1.1 cgd struct socket *so;
1194 1.1 cgd
1195 1.134 manu switch(unp->unp_socket->so_type) {
1196 1.134 manu case SOCK_SEQPACKET: /* FALLTHROUGH */
1197 1.134 manu case SOCK_STREAM:
1198 1.134 manu if (unp->unp_conn && (so = unp->unp_conn->unp_socket))
1199 1.134 manu socantrcvmore(so);
1200 1.134 manu break;
1201 1.134 manu default:
1202 1.134 manu break;
1203 1.134 manu }
1204 1.1 cgd }
1205 1.1 cgd
1206 1.112 ad bool
1207 1.76 matt unp_drop(struct unpcb *unp, int errno)
1208 1.1 cgd {
1209 1.1 cgd struct socket *so = unp->unp_socket;
1210 1.1 cgd
1211 1.112 ad KASSERT(solocked(so));
1212 1.112 ad
1213 1.1 cgd so->so_error = errno;
1214 1.1 cgd unp_disconnect(unp);
1215 1.1 cgd if (so->so_head) {
1216 1.112 ad so->so_pcb = NULL;
1217 1.112 ad /* sofree() drops the socket lock */
1218 1.14 mycroft sofree(so);
1219 1.112 ad unp_free(unp);
1220 1.112 ad return true;
1221 1.1 cgd }
1222 1.112 ad return false;
1223 1.1 cgd }
1224 1.1 cgd
1225 1.1 cgd #ifdef notdef
1226 1.76 matt unp_drain(void)
1227 1.1 cgd {
1228 1.1 cgd
1229 1.1 cgd }
1230 1.1 cgd #endif
1231 1.1 cgd
1232 1.5 andrew int
1233 1.136 christos unp_externalize(struct mbuf *rights, struct lwp *l, int flags)
1234 1.1 cgd {
1235 1.138 christos struct cmsghdr * const cm = mtod(rights, struct cmsghdr *);
1236 1.138 christos struct proc * const p = l->l_proc;
1237 1.106 ad file_t **rp;
1238 1.138 christos int error = 0;
1239 1.47 thorpej
1240 1.138 christos const size_t nfds = (cm->cmsg_len - CMSG_ALIGN(sizeof(*cm))) /
1241 1.106 ad sizeof(file_t *);
1242 1.143 drochner if (nfds == 0)
1243 1.143 drochner goto noop;
1244 1.1 cgd
1245 1.138 christos int * const fdp = kmem_alloc(nfds * sizeof(int), KM_SLEEP);
1246 1.101 ad rw_enter(&p->p_cwdi->cwdi_lock, RW_READER);
1247 1.50 thorpej
1248 1.121 mrg /* Make sure the recipient should be able to see the files.. */
1249 1.140 christos rp = (file_t **)CMSG_DATA(cm);
1250 1.140 christos for (size_t i = 0; i < nfds; i++) {
1251 1.140 christos file_t * const fp = *rp++;
1252 1.140 christos if (fp == NULL) {
1253 1.140 christos error = EINVAL;
1254 1.140 christos goto out;
1255 1.140 christos }
1256 1.140 christos /*
1257 1.140 christos * If we are in a chroot'ed directory, and
1258 1.140 christos * someone wants to pass us a directory, make
1259 1.140 christos * sure it's inside the subtree we're allowed
1260 1.140 christos * to access.
1261 1.140 christos */
1262 1.140 christos if (p->p_cwdi->cwdi_rdir != NULL && fp->f_type == DTYPE_VNODE) {
1263 1.140 christos vnode_t *vp = (vnode_t *)fp->f_data;
1264 1.140 christos if ((vp->v_type == VDIR) &&
1265 1.140 christos !vn_isunder(vp, p->p_cwdi->cwdi_rdir, l)) {
1266 1.140 christos error = EPERM;
1267 1.140 christos goto out;
1268 1.39 sommerfe }
1269 1.39 sommerfe }
1270 1.39 sommerfe }
1271 1.50 thorpej
1272 1.50 thorpej restart:
1273 1.24 cgd /*
1274 1.50 thorpej * First loop -- allocate file descriptor table slots for the
1275 1.121 mrg * new files.
1276 1.24 cgd */
1277 1.138 christos for (size_t i = 0; i < nfds; i++) {
1278 1.106 ad if ((error = fd_alloc(p, 0, &fdp[i])) != 0) {
1279 1.49 thorpej /*
1280 1.50 thorpej * Back out what we've done so far.
1281 1.49 thorpej */
1282 1.138 christos while (i-- > 0) {
1283 1.106 ad fd_abort(p, NULL, fdp[i]);
1284 1.106 ad }
1285 1.50 thorpej if (error == ENOSPC) {
1286 1.106 ad fd_tryexpand(p);
1287 1.50 thorpej error = 0;
1288 1.138 christos goto restart;
1289 1.50 thorpej }
1290 1.138 christos /*
1291 1.138 christos * This is the error that has historically
1292 1.138 christos * been returned, and some callers may
1293 1.138 christos * expect it.
1294 1.138 christos */
1295 1.138 christos error = EMSGSIZE;
1296 1.138 christos goto out;
1297 1.49 thorpej }
1298 1.1 cgd }
1299 1.24 cgd
1300 1.24 cgd /*
1301 1.50 thorpej * Now that adding them has succeeded, update all of the
1302 1.121 mrg * file passing state and affix the descriptors.
1303 1.112 ad */
1304 1.106 ad rp = (file_t **)CMSG_DATA(cm);
1305 1.138 christos int *ofdp = (int *)CMSG_DATA(cm);
1306 1.138 christos for (size_t i = 0; i < nfds; i++) {
1307 1.138 christos file_t * const fp = *rp++;
1308 1.138 christos const int fd = fdp[i];
1309 1.106 ad atomic_dec_uint(&unp_rights);
1310 1.136 christos fd_set_exclose(l, fd, (flags & O_CLOEXEC) != 0);
1311 1.136 christos fd_affix(p, fp, fd);
1312 1.138 christos /*
1313 1.138 christos * Done with this file pointer, replace it with a fd;
1314 1.138 christos */
1315 1.138 christos *ofdp++ = fd;
1316 1.106 ad mutex_enter(&fp->f_lock);
1317 1.50 thorpej fp->f_msgcount--;
1318 1.106 ad mutex_exit(&fp->f_lock);
1319 1.106 ad /*
1320 1.106 ad * Note that fd_affix() adds a reference to the file.
1321 1.106 ad * The file may already have been closed by another
1322 1.106 ad * LWP in the process, so we must drop the reference
1323 1.106 ad * added by unp_internalize() with closef().
1324 1.106 ad */
1325 1.106 ad closef(fp);
1326 1.50 thorpej }
1327 1.50 thorpej
1328 1.50 thorpej /*
1329 1.138 christos * Adjust length, in case of transition from large file_t
1330 1.138 christos * pointers to ints.
1331 1.50 thorpej */
1332 1.138 christos if (sizeof(file_t *) != sizeof(int)) {
1333 1.138 christos cm->cmsg_len = CMSG_LEN(nfds * sizeof(int));
1334 1.138 christos rights->m_len = CMSG_SPACE(nfds * sizeof(int));
1335 1.138 christos }
1336 1.50 thorpej out:
1337 1.138 christos if (__predict_false(error != 0)) {
1338 1.141 riastrad file_t **const fpp = (file_t **)CMSG_DATA(cm);
1339 1.141 riastrad for (size_t i = 0; i < nfds; i++)
1340 1.141 riastrad unp_discard_now(fpp[i]);
1341 1.141 riastrad /*
1342 1.141 riastrad * Truncate the array so that nobody will try to interpret
1343 1.141 riastrad * what is now garbage in it.
1344 1.141 riastrad */
1345 1.141 riastrad cm->cmsg_len = CMSG_LEN(0);
1346 1.141 riastrad rights->m_len = CMSG_SPACE(0);
1347 1.138 christos }
1348 1.143 drochner rw_exit(&p->p_cwdi->cwdi_lock);
1349 1.143 drochner kmem_free(fdp, nfds * sizeof(int));
1350 1.138 christos
1351 1.143 drochner noop:
1352 1.141 riastrad /*
1353 1.141 riastrad * Don't disclose kernel memory in the alignment space.
1354 1.141 riastrad */
1355 1.141 riastrad KASSERT(cm->cmsg_len <= rights->m_len);
1356 1.141 riastrad memset(&mtod(rights, char *)[cm->cmsg_len], 0, rights->m_len -
1357 1.141 riastrad cm->cmsg_len);
1358 1.139 christos return error;
1359 1.1 cgd }
1360 1.1 cgd
1361 1.5 andrew int
1362 1.112 ad unp_internalize(struct mbuf **controlp)
1363 1.1 cgd {
1364 1.121 mrg filedesc_t *fdescp = curlwp->l_fd;
1365 1.108 yamt struct mbuf *control = *controlp;
1366 1.73 martin struct cmsghdr *newcm, *cm = mtod(control, struct cmsghdr *);
1367 1.106 ad file_t **rp, **files;
1368 1.106 ad file_t *fp;
1369 1.46 augustss int i, fd, *fdp;
1370 1.106 ad int nfds, error;
1371 1.121 mrg u_int maxmsg;
1372 1.106 ad
1373 1.106 ad error = 0;
1374 1.106 ad newcm = NULL;
1375 1.38 thorpej
1376 1.106 ad /* Sanity check the control message header. */
1377 1.66 jdolecek if (cm->cmsg_type != SCM_RIGHTS || cm->cmsg_level != SOL_SOCKET ||
1378 1.117 christos cm->cmsg_len > control->m_len ||
1379 1.117 christos cm->cmsg_len < CMSG_ALIGN(sizeof(*cm)))
1380 1.1 cgd return (EINVAL);
1381 1.24 cgd
1382 1.106 ad /*
1383 1.106 ad * Verify that the file descriptors are valid, and acquire
1384 1.106 ad * a reference to each.
1385 1.106 ad */
1386 1.47 thorpej nfds = (cm->cmsg_len - CMSG_ALIGN(sizeof(*cm))) / sizeof(int);
1387 1.47 thorpej fdp = (int *)CMSG_DATA(cm);
1388 1.121 mrg maxmsg = maxfiles / unp_rights_ratio;
1389 1.24 cgd for (i = 0; i < nfds; i++) {
1390 1.24 cgd fd = *fdp++;
1391 1.121 mrg if (atomic_inc_uint_nv(&unp_rights) > maxmsg) {
1392 1.121 mrg atomic_dec_uint(&unp_rights);
1393 1.121 mrg nfds = i;
1394 1.121 mrg error = EAGAIN;
1395 1.121 mrg goto out;
1396 1.121 mrg }
1397 1.137 martin if ((fp = fd_getfile(fd)) == NULL
1398 1.137 martin || fp->f_type == DTYPE_KQUEUE) {
1399 1.137 martin if (fp)
1400 1.137 martin fd_putfile(fd);
1401 1.121 mrg atomic_dec_uint(&unp_rights);
1402 1.120 pooka nfds = i;
1403 1.106 ad error = EBADF;
1404 1.106 ad goto out;
1405 1.101 ad }
1406 1.24 cgd }
1407 1.24 cgd
1408 1.106 ad /* Allocate new space and copy header into it. */
1409 1.106 ad newcm = malloc(CMSG_SPACE(nfds * sizeof(file_t *)), M_MBUF, M_WAITOK);
1410 1.106 ad if (newcm == NULL) {
1411 1.106 ad error = E2BIG;
1412 1.106 ad goto out;
1413 1.106 ad }
1414 1.106 ad memcpy(newcm, cm, sizeof(struct cmsghdr));
1415 1.106 ad files = (file_t **)CMSG_DATA(newcm);
1416 1.106 ad
1417 1.24 cgd /*
1418 1.106 ad * Transform the file descriptors into file_t pointers, in
1419 1.24 cgd * reverse order so that if pointers are bigger than ints, the
1420 1.106 ad * int won't get until we're done. No need to lock, as we have
1421 1.106 ad * already validated the descriptors with fd_getfile().
1422 1.24 cgd */
1423 1.94 cbiere fdp = (int *)CMSG_DATA(cm) + nfds;
1424 1.94 cbiere rp = files + nfds;
1425 1.24 cgd for (i = 0; i < nfds; i++) {
1426 1.126 ad fp = fdescp->fd_dt->dt_ff[*--fdp]->ff_file;
1427 1.106 ad KASSERT(fp != NULL);
1428 1.106 ad mutex_enter(&fp->f_lock);
1429 1.94 cbiere *--rp = fp;
1430 1.1 cgd fp->f_count++;
1431 1.1 cgd fp->f_msgcount++;
1432 1.106 ad mutex_exit(&fp->f_lock);
1433 1.106 ad }
1434 1.106 ad
1435 1.106 ad out:
1436 1.106 ad /* Release descriptor references. */
1437 1.106 ad fdp = (int *)CMSG_DATA(cm);
1438 1.106 ad for (i = 0; i < nfds; i++) {
1439 1.106 ad fd_putfile(*fdp++);
1440 1.121 mrg if (error != 0) {
1441 1.121 mrg atomic_dec_uint(&unp_rights);
1442 1.121 mrg }
1443 1.1 cgd }
1444 1.73 martin
1445 1.106 ad if (error == 0) {
1446 1.108 yamt if (control->m_flags & M_EXT) {
1447 1.108 yamt m_freem(control);
1448 1.108 yamt *controlp = control = m_get(M_WAIT, MT_CONTROL);
1449 1.108 yamt }
1450 1.106 ad MEXTADD(control, newcm, CMSG_SPACE(nfds * sizeof(file_t *)),
1451 1.73 martin M_MBUF, NULL, NULL);
1452 1.73 martin cm = newcm;
1453 1.106 ad /*
1454 1.106 ad * Adjust message & mbuf to note amount of space
1455 1.106 ad * actually used.
1456 1.106 ad */
1457 1.106 ad cm->cmsg_len = CMSG_LEN(nfds * sizeof(file_t *));
1458 1.106 ad control->m_len = CMSG_SPACE(nfds * sizeof(file_t *));
1459 1.73 martin }
1460 1.73 martin
1461 1.106 ad return error;
1462 1.30 thorpej }
1463 1.30 thorpej
1464 1.30 thorpej struct mbuf *
1465 1.92 ad unp_addsockcred(struct lwp *l, struct mbuf *control)
1466 1.30 thorpej {
1467 1.30 thorpej struct sockcred *sc;
1468 1.142 christos struct mbuf *m;
1469 1.142 christos void *p;
1470 1.30 thorpej
1471 1.142 christos m = sbcreatecontrol1(&p, SOCKCREDSIZE(kauth_cred_ngroups(l->l_cred)),
1472 1.142 christos SCM_CREDS, SOL_SOCKET, M_WAITOK);
1473 1.142 christos if (m == NULL)
1474 1.142 christos return control;
1475 1.142 christos
1476 1.142 christos sc = p;
1477 1.92 ad sc->sc_uid = kauth_cred_getuid(l->l_cred);
1478 1.92 ad sc->sc_euid = kauth_cred_geteuid(l->l_cred);
1479 1.92 ad sc->sc_gid = kauth_cred_getgid(l->l_cred);
1480 1.92 ad sc->sc_egid = kauth_cred_getegid(l->l_cred);
1481 1.92 ad sc->sc_ngroups = kauth_cred_ngroups(l->l_cred);
1482 1.142 christos
1483 1.142 christos for (int i = 0; i < sc->sc_ngroups; i++)
1484 1.92 ad sc->sc_groups[i] = kauth_cred_group(l->l_cred, i);
1485 1.30 thorpej
1486 1.142 christos return m_add(control, m);
1487 1.1 cgd }
1488 1.1 cgd
1489 1.39 sommerfe /*
1490 1.121 mrg * Do a mark-sweep GC of files in the system, to free up any which are
1491 1.121 mrg * caught in flight to an about-to-be-closed socket. Additionally,
1492 1.121 mrg * process deferred file closures.
1493 1.39 sommerfe */
1494 1.121 mrg static void
1495 1.121 mrg unp_gc(file_t *dp)
1496 1.1 cgd {
1497 1.121 mrg extern struct domain unixdomain;
1498 1.121 mrg file_t *fp, *np;
1499 1.46 augustss struct socket *so, *so1;
1500 1.121 mrg u_int i, old, new;
1501 1.121 mrg bool didwork;
1502 1.1 cgd
1503 1.121 mrg KASSERT(curlwp == unp_thread_lwp);
1504 1.121 mrg KASSERT(mutex_owned(&filelist_lock));
1505 1.106 ad
1506 1.121 mrg /*
1507 1.121 mrg * First, process deferred file closures.
1508 1.121 mrg */
1509 1.121 mrg while (!SLIST_EMPTY(&unp_thread_discard)) {
1510 1.121 mrg fp = SLIST_FIRST(&unp_thread_discard);
1511 1.121 mrg KASSERT(fp->f_unpcount > 0);
1512 1.121 mrg KASSERT(fp->f_count > 0);
1513 1.121 mrg KASSERT(fp->f_msgcount > 0);
1514 1.121 mrg KASSERT(fp->f_count >= fp->f_unpcount);
1515 1.121 mrg KASSERT(fp->f_count >= fp->f_msgcount);
1516 1.121 mrg KASSERT(fp->f_msgcount >= fp->f_unpcount);
1517 1.121 mrg SLIST_REMOVE_HEAD(&unp_thread_discard, f_unplist);
1518 1.121 mrg i = fp->f_unpcount;
1519 1.121 mrg fp->f_unpcount = 0;
1520 1.121 mrg mutex_exit(&filelist_lock);
1521 1.121 mrg for (; i != 0; i--) {
1522 1.121 mrg unp_discard_now(fp);
1523 1.121 mrg }
1524 1.121 mrg mutex_enter(&filelist_lock);
1525 1.121 mrg }
1526 1.39 sommerfe
1527 1.121 mrg /*
1528 1.121 mrg * Clear mark bits. Ensure that we don't consider new files
1529 1.121 mrg * entering the file table during this loop (they will not have
1530 1.121 mrg * FSCAN set).
1531 1.121 mrg */
1532 1.106 ad unp_defer = 0;
1533 1.106 ad LIST_FOREACH(fp, &filehead, f_list) {
1534 1.121 mrg for (old = fp->f_flag;; old = new) {
1535 1.121 mrg new = atomic_cas_uint(&fp->f_flag, old,
1536 1.121 mrg (old | FSCAN) & ~(FMARK|FDEFER));
1537 1.121 mrg if (__predict_true(old == new)) {
1538 1.121 mrg break;
1539 1.121 mrg }
1540 1.121 mrg }
1541 1.106 ad }
1542 1.39 sommerfe
1543 1.39 sommerfe /*
1544 1.121 mrg * Iterate over the set of sockets, marking ones believed (based on
1545 1.121 mrg * refcount) to be referenced from a process, and marking for rescan
1546 1.121 mrg * sockets which are queued on a socket. Recan continues descending
1547 1.121 mrg * and searching for sockets referenced by sockets (FDEFER), until
1548 1.121 mrg * there are no more socket->socket references to be discovered.
1549 1.39 sommerfe */
1550 1.1 cgd do {
1551 1.121 mrg didwork = false;
1552 1.121 mrg for (fp = LIST_FIRST(&filehead); fp != NULL; fp = np) {
1553 1.121 mrg KASSERT(mutex_owned(&filelist_lock));
1554 1.121 mrg np = LIST_NEXT(fp, f_list);
1555 1.106 ad mutex_enter(&fp->f_lock);
1556 1.121 mrg if ((fp->f_flag & FDEFER) != 0) {
1557 1.106 ad atomic_and_uint(&fp->f_flag, ~FDEFER);
1558 1.1 cgd unp_defer--;
1559 1.106 ad KASSERT(fp->f_count != 0);
1560 1.1 cgd } else {
1561 1.101 ad if (fp->f_count == 0 ||
1562 1.121 mrg (fp->f_flag & FMARK) != 0 ||
1563 1.121 mrg fp->f_count == fp->f_msgcount ||
1564 1.121 mrg fp->f_unpcount != 0) {
1565 1.106 ad mutex_exit(&fp->f_lock);
1566 1.1 cgd continue;
1567 1.101 ad }
1568 1.1 cgd }
1569 1.106 ad atomic_or_uint(&fp->f_flag, FMARK);
1570 1.39 sommerfe
1571 1.1 cgd if (fp->f_type != DTYPE_SOCKET ||
1572 1.112 ad (so = fp->f_data) == NULL ||
1573 1.101 ad so->so_proto->pr_domain != &unixdomain ||
1574 1.121 mrg (so->so_proto->pr_flags & PR_RIGHTS) == 0) {
1575 1.106 ad mutex_exit(&fp->f_lock);
1576 1.1 cgd continue;
1577 1.101 ad }
1578 1.121 mrg
1579 1.121 mrg /* Gain file ref, mark our position, and unlock. */
1580 1.121 mrg didwork = true;
1581 1.121 mrg LIST_INSERT_AFTER(fp, dp, f_list);
1582 1.121 mrg fp->f_count++;
1583 1.106 ad mutex_exit(&fp->f_lock);
1584 1.121 mrg mutex_exit(&filelist_lock);
1585 1.101 ad
1586 1.112 ad /*
1587 1.121 mrg * Mark files referenced from sockets queued on the
1588 1.121 mrg * accept queue as well.
1589 1.112 ad */
1590 1.112 ad solock(so);
1591 1.39 sommerfe unp_scan(so->so_rcv.sb_mb, unp_mark, 0);
1592 1.121 mrg if ((so->so_options & SO_ACCEPTCONN) != 0) {
1593 1.54 matt TAILQ_FOREACH(so1, &so->so_q0, so_qe) {
1594 1.39 sommerfe unp_scan(so1->so_rcv.sb_mb, unp_mark, 0);
1595 1.39 sommerfe }
1596 1.54 matt TAILQ_FOREACH(so1, &so->so_q, so_qe) {
1597 1.39 sommerfe unp_scan(so1->so_rcv.sb_mb, unp_mark, 0);
1598 1.39 sommerfe }
1599 1.39 sommerfe }
1600 1.112 ad sounlock(so);
1601 1.121 mrg
1602 1.121 mrg /* Re-lock and restart from where we left off. */
1603 1.121 mrg closef(fp);
1604 1.121 mrg mutex_enter(&filelist_lock);
1605 1.121 mrg np = LIST_NEXT(dp, f_list);
1606 1.121 mrg LIST_REMOVE(dp, f_list);
1607 1.1 cgd }
1608 1.121 mrg /*
1609 1.121 mrg * Bail early if we did nothing in the loop above. Could
1610 1.121 mrg * happen because of concurrent activity causing unp_defer
1611 1.121 mrg * to get out of sync.
1612 1.121 mrg */
1613 1.121 mrg } while (unp_defer != 0 && didwork);
1614 1.101 ad
1615 1.8 mycroft /*
1616 1.121 mrg * Sweep pass.
1617 1.8 mycroft *
1618 1.121 mrg * We grab an extra reference to each of the files that are
1619 1.121 mrg * not otherwise accessible and then free the rights that are
1620 1.121 mrg * stored in messages on them.
1621 1.8 mycroft */
1622 1.121 mrg for (fp = LIST_FIRST(&filehead); fp != NULL; fp = np) {
1623 1.121 mrg KASSERT(mutex_owned(&filelist_lock));
1624 1.121 mrg np = LIST_NEXT(fp, f_list);
1625 1.106 ad mutex_enter(&fp->f_lock);
1626 1.121 mrg
1627 1.121 mrg /*
1628 1.121 mrg * Ignore non-sockets.
1629 1.121 mrg * Ignore dead sockets, or sockets with pending close.
1630 1.121 mrg * Ignore sockets obviously referenced elsewhere.
1631 1.121 mrg * Ignore sockets marked as referenced by our scan.
1632 1.121 mrg * Ignore new sockets that did not exist during the scan.
1633 1.121 mrg */
1634 1.121 mrg if (fp->f_type != DTYPE_SOCKET ||
1635 1.121 mrg fp->f_count == 0 || fp->f_unpcount != 0 ||
1636 1.121 mrg fp->f_count != fp->f_msgcount ||
1637 1.121 mrg (fp->f_flag & (FMARK | FSCAN)) != FSCAN) {
1638 1.121 mrg mutex_exit(&fp->f_lock);
1639 1.121 mrg continue;
1640 1.8 mycroft }
1641 1.121 mrg
1642 1.121 mrg /* Gain file ref, mark our position, and unlock. */
1643 1.121 mrg LIST_INSERT_AFTER(fp, dp, f_list);
1644 1.121 mrg fp->f_count++;
1645 1.106 ad mutex_exit(&fp->f_lock);
1646 1.121 mrg mutex_exit(&filelist_lock);
1647 1.121 mrg
1648 1.121 mrg /*
1649 1.121 mrg * Flush all data from the socket's receive buffer.
1650 1.121 mrg * This will cause files referenced only by the
1651 1.121 mrg * socket to be queued for close.
1652 1.121 mrg */
1653 1.121 mrg so = fp->f_data;
1654 1.121 mrg solock(so);
1655 1.121 mrg sorflush(so);
1656 1.121 mrg sounlock(so);
1657 1.121 mrg
1658 1.121 mrg /* Re-lock and restart from where we left off. */
1659 1.121 mrg closef(fp);
1660 1.121 mrg mutex_enter(&filelist_lock);
1661 1.121 mrg np = LIST_NEXT(dp, f_list);
1662 1.121 mrg LIST_REMOVE(dp, f_list);
1663 1.121 mrg }
1664 1.121 mrg }
1665 1.121 mrg
1666 1.121 mrg /*
1667 1.121 mrg * Garbage collector thread. While SCM_RIGHTS messages are in transit,
1668 1.121 mrg * wake once per second to garbage collect. Run continually while we
1669 1.121 mrg * have deferred closes to process.
1670 1.121 mrg */
1671 1.121 mrg static void
1672 1.121 mrg unp_thread(void *cookie)
1673 1.121 mrg {
1674 1.121 mrg file_t *dp;
1675 1.121 mrg
1676 1.121 mrg /* Allocate a dummy file for our scans. */
1677 1.121 mrg if ((dp = fgetdummy()) == NULL) {
1678 1.121 mrg panic("unp_thread");
1679 1.1 cgd }
1680 1.101 ad
1681 1.121 mrg mutex_enter(&filelist_lock);
1682 1.121 mrg for (;;) {
1683 1.121 mrg KASSERT(mutex_owned(&filelist_lock));
1684 1.121 mrg if (SLIST_EMPTY(&unp_thread_discard)) {
1685 1.121 mrg if (unp_rights != 0) {
1686 1.121 mrg (void)cv_timedwait(&unp_thread_cv,
1687 1.121 mrg &filelist_lock, hz);
1688 1.121 mrg } else {
1689 1.121 mrg cv_wait(&unp_thread_cv, &filelist_lock);
1690 1.121 mrg }
1691 1.112 ad }
1692 1.121 mrg unp_gc(dp);
1693 1.39 sommerfe }
1694 1.121 mrg /* NOTREACHED */
1695 1.121 mrg }
1696 1.121 mrg
1697 1.121 mrg /*
1698 1.121 mrg * Kick the garbage collector into action if there is something for
1699 1.121 mrg * it to process.
1700 1.121 mrg */
1701 1.121 mrg static void
1702 1.121 mrg unp_thread_kick(void)
1703 1.121 mrg {
1704 1.121 mrg
1705 1.121 mrg if (!SLIST_EMPTY(&unp_thread_discard) || unp_rights != 0) {
1706 1.121 mrg mutex_enter(&filelist_lock);
1707 1.121 mrg cv_signal(&unp_thread_cv);
1708 1.121 mrg mutex_exit(&filelist_lock);
1709 1.44 thorpej }
1710 1.1 cgd }
1711 1.1 cgd
1712 1.5 andrew void
1713 1.76 matt unp_dispose(struct mbuf *m)
1714 1.1 cgd {
1715 1.8 mycroft
1716 1.1 cgd if (m)
1717 1.121 mrg unp_scan(m, unp_discard_later, 1);
1718 1.1 cgd }
1719 1.1 cgd
1720 1.5 andrew void
1721 1.106 ad unp_scan(struct mbuf *m0, void (*op)(file_t *), int discard)
1722 1.1 cgd {
1723 1.46 augustss struct mbuf *m;
1724 1.121 mrg file_t **rp, *fp;
1725 1.46 augustss struct cmsghdr *cm;
1726 1.121 mrg int i, qfds;
1727 1.1 cgd
1728 1.1 cgd while (m0) {
1729 1.48 thorpej for (m = m0; m; m = m->m_next) {
1730 1.121 mrg if (m->m_type != MT_CONTROL ||
1731 1.121 mrg m->m_len < sizeof(*cm)) {
1732 1.121 mrg continue;
1733 1.121 mrg }
1734 1.121 mrg cm = mtod(m, struct cmsghdr *);
1735 1.121 mrg if (cm->cmsg_level != SOL_SOCKET ||
1736 1.121 mrg cm->cmsg_type != SCM_RIGHTS)
1737 1.121 mrg continue;
1738 1.121 mrg qfds = (cm->cmsg_len - CMSG_ALIGN(sizeof(*cm)))
1739 1.121 mrg / sizeof(file_t *);
1740 1.121 mrg rp = (file_t **)CMSG_DATA(cm);
1741 1.121 mrg for (i = 0; i < qfds; i++) {
1742 1.121 mrg fp = *rp;
1743 1.121 mrg if (discard) {
1744 1.121 mrg *rp = 0;
1745 1.39 sommerfe }
1746 1.121 mrg (*op)(fp);
1747 1.121 mrg rp++;
1748 1.1 cgd }
1749 1.48 thorpej }
1750 1.52 thorpej m0 = m0->m_nextpkt;
1751 1.1 cgd }
1752 1.1 cgd }
1753 1.1 cgd
1754 1.5 andrew void
1755 1.106 ad unp_mark(file_t *fp)
1756 1.1 cgd {
1757 1.101 ad
1758 1.39 sommerfe if (fp == NULL)
1759 1.39 sommerfe return;
1760 1.80 perry
1761 1.39 sommerfe /* If we're already deferred, don't screw up the defer count */
1762 1.106 ad mutex_enter(&fp->f_lock);
1763 1.101 ad if (fp->f_flag & (FMARK | FDEFER)) {
1764 1.106 ad mutex_exit(&fp->f_lock);
1765 1.1 cgd return;
1766 1.101 ad }
1767 1.39 sommerfe
1768 1.39 sommerfe /*
1769 1.121 mrg * Minimize the number of deferrals... Sockets are the only type of
1770 1.121 mrg * file which can hold references to another file, so just mark
1771 1.121 mrg * other files, and defer unmarked sockets for the next pass.
1772 1.39 sommerfe */
1773 1.39 sommerfe if (fp->f_type == DTYPE_SOCKET) {
1774 1.39 sommerfe unp_defer++;
1775 1.106 ad KASSERT(fp->f_count != 0);
1776 1.106 ad atomic_or_uint(&fp->f_flag, FDEFER);
1777 1.39 sommerfe } else {
1778 1.106 ad atomic_or_uint(&fp->f_flag, FMARK);
1779 1.39 sommerfe }
1780 1.106 ad mutex_exit(&fp->f_lock);
1781 1.1 cgd }
1782 1.1 cgd
1783 1.121 mrg static void
1784 1.121 mrg unp_discard_now(file_t *fp)
1785 1.1 cgd {
1786 1.106 ad
1787 1.39 sommerfe if (fp == NULL)
1788 1.39 sommerfe return;
1789 1.106 ad
1790 1.121 mrg KASSERT(fp->f_count > 0);
1791 1.121 mrg KASSERT(fp->f_msgcount > 0);
1792 1.121 mrg
1793 1.106 ad mutex_enter(&fp->f_lock);
1794 1.1 cgd fp->f_msgcount--;
1795 1.106 ad mutex_exit(&fp->f_lock);
1796 1.106 ad atomic_dec_uint(&unp_rights);
1797 1.106 ad (void)closef(fp);
1798 1.1 cgd }
1799 1.121 mrg
1800 1.121 mrg static void
1801 1.121 mrg unp_discard_later(file_t *fp)
1802 1.121 mrg {
1803 1.121 mrg
1804 1.121 mrg if (fp == NULL)
1805 1.121 mrg return;
1806 1.121 mrg
1807 1.121 mrg KASSERT(fp->f_count > 0);
1808 1.121 mrg KASSERT(fp->f_msgcount > 0);
1809 1.121 mrg
1810 1.121 mrg mutex_enter(&filelist_lock);
1811 1.121 mrg if (fp->f_unpcount++ == 0) {
1812 1.121 mrg SLIST_INSERT_HEAD(&unp_thread_discard, fp, f_unplist);
1813 1.121 mrg }
1814 1.121 mrg mutex_exit(&filelist_lock);
1815 1.121 mrg }
1816 1.151 rmind
1817 1.151 rmind const struct pr_usrreqs unp_usrreqs = {
1818 1.152 rmind .pr_attach = unp_attach,
1819 1.152 rmind .pr_detach = unp_detach,
1820 1.154 rtr .pr_ioctl = unp_ioctl,
1821 1.151 rmind .pr_generic = unp_usrreq,
1822 1.151 rmind };
1823