]> git.saurik.com Git - apple/xnu.git/blame - bsd/kern/uipc_socket.c
xnu-1504.15.3.tar.gz
[apple/xnu.git] / bsd / kern / uipc_socket.c
CommitLineData
1c79356b 1/*
b0d623f7 2 * Copyright (c) 1998-2008 Apple Inc. All rights reserved.
5d5c5d0d 3 *
2d21ac55 4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
1c79356b 5 *
2d21ac55
A
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
8f6c56a5 14 *
2d21ac55
A
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
8f6c56a5
A
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
2d21ac55
A
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
8f6c56a5 25 *
2d21ac55 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
1c79356b 27 */
1c79356b
A
28/* Copyright (c) 1995 NeXT Computer, Inc. All Rights Reserved */
29/*
30 * Copyright (c) 1982, 1986, 1988, 1990, 1993
31 * The Regents of the University of California. All rights reserved.
32 *
33 * Redistribution and use in source and binary forms, with or without
34 * modification, are permitted provided that the following conditions
35 * are met:
36 * 1. Redistributions of source code must retain the above copyright
37 * notice, this list of conditions and the following disclaimer.
38 * 2. Redistributions in binary form must reproduce the above copyright
39 * notice, this list of conditions and the following disclaimer in the
40 * documentation and/or other materials provided with the distribution.
41 * 3. All advertising materials mentioning features or use of this software
42 * must display the following acknowledgement:
43 * This product includes software developed by the University of
44 * California, Berkeley and its contributors.
45 * 4. Neither the name of the University nor the names of its contributors
46 * may be used to endorse or promote products derived from this software
47 * without specific prior written permission.
48 *
49 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
50 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
51 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
52 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
53 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
54 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
55 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
56 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
57 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
58 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
59 * SUCH DAMAGE.
60 *
9bccf70c
A
61 * @(#)uipc_socket.c 8.3 (Berkeley) 4/15/94
62 * $FreeBSD: src/sys/kern/uipc_socket.c,v 1.68.2.16 2001/06/14 20:46:06 ume Exp $
1c79356b 63 */
2d21ac55
A
64/*
65 * NOTICE: This file was modified by SPARTA, Inc. in 2005 to introduce
66 * support for mandatory and extensible security protections. This notice
67 * is included in support of clause 2.2 (b) of the Apple Public License,
68 * Version 2.0.
69 */
1c79356b
A
70
71#include <sys/param.h>
72#include <sys/systm.h>
55e303ae 73#include <sys/filedesc.h>
2d21ac55 74#include <sys/proc.h>
91447636
A
75#include <sys/proc_internal.h>
76#include <sys/kauth.h>
77#include <sys/file_internal.h>
1c79356b
A
78#include <sys/fcntl.h>
79#include <sys/malloc.h>
80#include <sys/mbuf.h>
81#include <sys/domain.h>
82#include <sys/kernel.h>
55e303ae 83#include <sys/event.h>
1c79356b
A
84#include <sys/poll.h>
85#include <sys/protosw.h>
86#include <sys/socket.h>
87#include <sys/socketvar.h>
88#include <sys/resourcevar.h>
89#include <sys/signalvar.h>
90#include <sys/sysctl.h>
91#include <sys/uio.h>
92#include <sys/ev.h>
93#include <sys/kdebug.h>
2d21ac55 94#include <sys/un.h>
d1ecb069 95#include <sys/user.h>
1c79356b
A
96#include <net/route.h>
97#include <netinet/in.h>
98#include <netinet/in_pcb.h>
99#include <kern/zalloc.h>
91447636 100#include <kern/locks.h>
1c79356b 101#include <machine/limits.h>
2d21ac55
A
102#include <libkern/OSAtomic.h>
103#include <pexpert/pexpert.h>
b0d623f7 104#include <kern/assert.h>
2d21ac55
A
105
106#if CONFIG_MACF
107#include <security/mac.h>
108#include <security/mac_framework.h>
109#endif /* MAC */
110
1c79356b
A
111int so_cache_hw = 0;
112int so_cache_timeouts = 0;
113int so_cache_max_freed = 0;
114int cached_sock_count = 0;
b0d623f7 115__private_extern__ int max_cached_sock_count = MAX_CACHED_SOCKETS;
1c79356b
A
116struct socket *socket_cache_head = 0;
117struct socket *socket_cache_tail = 0;
b0d623f7 118u_int32_t so_cache_time = 0;
1c79356b
A
119int so_cache_init_done = 0;
120struct zone *so_cache_zone;
1c79356b 121
91447636
A
122static lck_grp_t *so_cache_mtx_grp;
123static lck_attr_t *so_cache_mtx_attr;
124static lck_grp_attr_t *so_cache_mtx_grp_attr;
125lck_mtx_t *so_cache_mtx;
126
1c79356b
A
127#include <machine/limits.h>
128
2d21ac55
A
129static void filt_sordetach(struct knote *kn);
130static int filt_soread(struct knote *kn, long hint);
131static void filt_sowdetach(struct knote *kn);
132static int filt_sowrite(struct knote *kn, long hint);
2d21ac55
A
133
134static int
135sooptcopyin_timeval(struct sockopt *sopt, struct timeval * tv_p);
136
137static int
138sooptcopyout_timeval(struct sockopt *sopt, const struct timeval * tv_p);
55e303ae 139
b0d623f7
A
140static struct filterops soread_filtops = {
141 .f_isfd = 1,
142 .f_detach = filt_sordetach,
143 .f_event = filt_soread,
144};
145static struct filterops sowrite_filtops = {
146 .f_isfd = 1,
147 .f_detach = filt_sowdetach,
148 .f_event = filt_sowrite,
149};
55e303ae 150
2d21ac55 151#define EVEN_MORE_LOCKING_DEBUG 0
1c79356b
A
152int socket_debug = 0;
153int socket_zone = M_SOCKET;
154so_gen_t so_gencnt; /* generation count for sockets */
155
156MALLOC_DEFINE(M_SONAME, "soname", "socket name");
157MALLOC_DEFINE(M_PCB, "pcb", "protocol control block");
158
2d21ac55
A
159#define DBG_LAYER_IN_BEG NETDBG_CODE(DBG_NETSOCK, 0)
160#define DBG_LAYER_IN_END NETDBG_CODE(DBG_NETSOCK, 2)
161#define DBG_LAYER_OUT_BEG NETDBG_CODE(DBG_NETSOCK, 1)
162#define DBG_LAYER_OUT_END NETDBG_CODE(DBG_NETSOCK, 3)
163#define DBG_FNC_SOSEND NETDBG_CODE(DBG_NETSOCK, (4 << 8) | 1)
164#define DBG_FNC_SORECEIVE NETDBG_CODE(DBG_NETSOCK, (8 << 8))
165#define DBG_FNC_SOSHUTDOWN NETDBG_CODE(DBG_NETSOCK, (9 << 8))
1c79356b 166
2d21ac55 167#define MAX_SOOPTGETM_SIZE (128 * MCLBYTES)
1c79356b 168
91447636 169
1c79356b
A
170SYSCTL_DECL(_kern_ipc);
171
2d21ac55
A
172int somaxconn = SOMAXCONN;
173SYSCTL_INT(_kern_ipc, KIPC_SOMAXCONN, somaxconn, CTLFLAG_RW, &somaxconn, 0, "");
1c79356b
A
174
175/* Should we get a maximum also ??? */
fa4905b1 176static int sosendmaxchain = 65536;
1c79356b 177static int sosendminchain = 16384;
55e303ae 178static int sorecvmincopy = 16384;
1c79356b 179SYSCTL_INT(_kern_ipc, OID_AUTO, sosendminchain, CTLFLAG_RW, &sosendminchain,
2d21ac55 180 0, "");
55e303ae 181SYSCTL_INT(_kern_ipc, OID_AUTO, sorecvmincopy, CTLFLAG_RW, &sorecvmincopy,
2d21ac55
A
182 0, "");
183
184/*
185 * Set to enable jumbo clusters (if available) for large writes when
186 * the socket is marked with SOF_MULTIPAGES; see below.
187 */
188int sosendjcl = 1;
189SYSCTL_INT(_kern_ipc, OID_AUTO, sosendjcl, CTLFLAG_RW, &sosendjcl, 0, "");
1c79356b 190
2d21ac55
A
191/*
192 * Set this to ignore SOF_MULTIPAGES and use jumbo clusters for large
193 * writes on the socket for all protocols on any network interfaces,
194 * depending upon sosendjcl above. Be extra careful when setting this
195 * to 1, because sending down packets that cross physical pages down to
196 * broken drivers (those that falsely assume that the physical pages
197 * are contiguous) might lead to system panics or silent data corruption.
198 * When set to 0, the system will respect SOF_MULTIPAGES, which is set
199 * only for TCP sockets whose outgoing interface is IFNET_MULTIPAGES
200 * capable. Set this to 1 only for testing/debugging purposes.
201 */
202int sosendjcl_ignore_capab = 0;
203SYSCTL_INT(_kern_ipc, OID_AUTO, sosendjcl_ignore_capab, CTLFLAG_RW,
204 &sosendjcl_ignore_capab, 0, "");
1c79356b
A
205
206/*
207 * Socket operation routines.
208 * These routines are called by the routines in
209 * sys_socket.c or from a system process, and
210 * implement the semantics of socket operations by
211 * switching out to the protocol specific routines.
212 */
213
2d21ac55
A
214/* sys_generic.c */
215extern void postevent(struct socket *, struct sockbuf *, int);
216extern void evsofree(struct socket *);
217
218/* TODO: these should be in header file */
219extern int get_inpcb_str_size(void);
220extern int get_tcp_str_size(void);
221extern struct domain *pffinddomain(int);
222extern struct protosw *pffindprotonotype(int, int);
223extern int soclose_locked(struct socket *);
224extern int soo_kqfilter(struct fileproc *, struct knote *, struct proc *);
225
d1ecb069
A
226extern int uthread_get_background_state(uthread_t);
227
9bccf70c 228#ifdef __APPLE__
91447636
A
229
230vm_size_t so_cache_zone_element_size;
231
2d21ac55
A
232static int sodelayed_copy(struct socket *, struct uio *, struct mbuf **, int *);
233static void cached_sock_alloc(struct socket **, int);
234static void cached_sock_free(struct socket *);
235static void so_cache_timer(void *);
236
237void soclose_wait_locked(struct socket *so);
b0d623f7 238int so_isdstlocal(struct socket *so);
91447636
A
239
240
2d21ac55
A
241void
242socketinit(void)
1c79356b 243{
2d21ac55 244 vm_size_t str_size;
1c79356b 245
91447636
A
246 if (so_cache_init_done) {
247 printf("socketinit: already called...\n");
248 return;
249 }
250
593a1d5f 251 PE_parse_boot_argn("socket_debug", &socket_debug, sizeof (socket_debug));
2d21ac55 252
91447636
A
253 /*
254 * allocate lock group attribute and group for socket cache mutex
255 */
256 so_cache_mtx_grp_attr = lck_grp_attr_alloc_init();
91447636 257
2d21ac55
A
258 so_cache_mtx_grp = lck_grp_alloc_init("so_cache",
259 so_cache_mtx_grp_attr);
260
91447636
A
261 /*
262 * allocate the lock attribute for socket cache mutex
263 */
264 so_cache_mtx_attr = lck_attr_alloc_init();
91447636 265
2d21ac55
A
266 so_cache_init_done = 1;
267
268 /* cached sockets mutex */
269 so_cache_mtx = lck_mtx_alloc_init(so_cache_mtx_grp, so_cache_mtx_attr);
1c79356b 270
2d21ac55 271 if (so_cache_mtx == NULL)
91447636
A
272 return; /* we're hosed... */
273
2d21ac55
A
274 str_size = (vm_size_t)(sizeof (struct socket) + 4 +
275 get_inpcb_str_size() + 4 + get_tcp_str_size());
276
277 so_cache_zone = zinit(str_size, 120000*str_size, 8192, "socache zone");
0b4c1975 278 zone_change(so_cache_zone, Z_NOENCRYPT, TRUE);
1c79356b 279#if TEMPDEBUG
2d21ac55 280 printf("cached_sock_alloc -- so_cache_zone size is %x\n", str_size);
1c79356b 281#endif
2d21ac55 282 timeout(so_cache_timer, NULL, (SO_CACHE_FLUSH_INTERVAL * hz));
91447636 283
2d21ac55 284 so_cache_zone_element_size = str_size;
1c79356b 285
2d21ac55 286 sflt_init();
1c79356b
A
287}
288
2d21ac55
A
289static void
290cached_sock_alloc(struct socket **so, int waitok)
1c79356b 291{
2d21ac55 292 caddr_t temp;
b0d623f7 293 register uintptr_t offset;
1c79356b 294
91447636
A
295 lck_mtx_lock(so_cache_mtx);
296
2d21ac55
A
297 if (cached_sock_count) {
298 cached_sock_count--;
299 *so = socket_cache_head;
300 if (*so == 0)
301 panic("cached_sock_alloc: cached sock is null");
1c79356b 302
2d21ac55
A
303 socket_cache_head = socket_cache_head->cache_next;
304 if (socket_cache_head)
305 socket_cache_head->cache_prev = 0;
306 else
307 socket_cache_tail = 0;
91447636
A
308
309 lck_mtx_unlock(so_cache_mtx);
1c79356b 310
2d21ac55
A
311 temp = (*so)->so_saved_pcb;
312 bzero((caddr_t)*so, sizeof (struct socket));
1c79356b 313#if TEMPDEBUG
2d21ac55
A
314 kprintf("cached_sock_alloc - retreiving cached sock %p - "
315 "count == %d\n", *so, cached_sock_count);
1c79356b 316#endif
2d21ac55
A
317 (*so)->so_saved_pcb = temp;
318 (*so)->cached_in_sock_layer = 1;
319 } else {
1c79356b 320#if TEMPDEBUG
2d21ac55 321 kprintf("Allocating cached sock %p from memory\n", *so);
1c79356b
A
322#endif
323
2d21ac55 324 lck_mtx_unlock(so_cache_mtx);
1c79356b 325
2d21ac55
A
326 if (waitok)
327 *so = (struct socket *)zalloc(so_cache_zone);
328 else
329 *so = (struct socket *)zalloc_noblock(so_cache_zone);
1c79356b 330
2d21ac55
A
331 if (*so == 0)
332 return;
1c79356b 333
2d21ac55 334 bzero((caddr_t)*so, sizeof (struct socket));
1c79356b 335
2d21ac55
A
336 /*
337 * Define offsets for extra structures into our single block of
338 * memory. Align extra structures on longword boundaries.
339 */
b0d623f7
A
340
341 offset = (uintptr_t) *so;
2d21ac55 342 offset += sizeof (struct socket);
b0d623f7
A
343
344 offset = ALIGN(offset);
345
2d21ac55
A
346 (*so)->so_saved_pcb = (caddr_t)offset;
347 offset += get_inpcb_str_size();
b0d623f7
A
348
349 offset = ALIGN(offset);
1c79356b 350
2d21ac55
A
351 ((struct inpcb *)(*so)->so_saved_pcb)->inp_saved_ppcb =
352 (caddr_t)offset;
1c79356b 353#if TEMPDEBUG
2d21ac55
A
354 kprintf("Allocating cached socket - %p, pcb=%p tcpcb=%p\n",
355 *so, (*so)->so_saved_pcb,
1c79356b
A
356 ((struct inpcb *)(*so)->so_saved_pcb)->inp_saved_ppcb);
357#endif
2d21ac55 358 }
1c79356b 359
2d21ac55 360 (*so)->cached_in_sock_layer = 1;
1c79356b
A
361}
362
2d21ac55
A
363static void
364cached_sock_free(struct socket *so)
1c79356b 365{
1c79356b 366
91447636 367 lck_mtx_lock(so_cache_mtx);
1c79356b 368
b0d623f7 369 if (++cached_sock_count > max_cached_sock_count) {
1c79356b 370 --cached_sock_count;
91447636 371 lck_mtx_unlock(so_cache_mtx);
1c79356b 372#if TEMPDEBUG
2d21ac55 373 kprintf("Freeing overflowed cached socket %p\n", so);
1c79356b 374#endif
91447636 375 zfree(so_cache_zone, so);
2d21ac55 376 } else {
1c79356b 377#if TEMPDEBUG
2d21ac55 378 kprintf("Freeing socket %p into cache\n", so);
1c79356b
A
379#endif
380 if (so_cache_hw < cached_sock_count)
381 so_cache_hw = cached_sock_count;
382
383 so->cache_next = socket_cache_head;
384 so->cache_prev = 0;
385 if (socket_cache_head)
386 socket_cache_head->cache_prev = so;
387 else
388 socket_cache_tail = so;
389
390 so->cache_timestamp = so_cache_time;
391 socket_cache_head = so;
91447636 392 lck_mtx_unlock(so_cache_mtx);
1c79356b
A
393 }
394
395#if TEMPDEBUG
2d21ac55
A
396 kprintf("Freed cached sock %p into cache - count is %d\n",
397 so, cached_sock_count);
1c79356b 398#endif
1c79356b
A
399}
400
2d21ac55
A
401static void
402so_cache_timer(__unused void *dummy)
1c79356b
A
403{
404 register struct socket *p;
1c79356b 405 register int n_freed = 0;
1c79356b 406
91447636 407 lck_mtx_lock(so_cache_mtx);
1c79356b 408
91447636 409 ++so_cache_time;
1c79356b 410
2d21ac55 411 while ((p = socket_cache_tail)) {
1c79356b 412 if ((so_cache_time - p->cache_timestamp) < SO_CACHE_TIME_LIMIT)
2d21ac55 413 break;
1c79356b
A
414
415 so_cache_timeouts++;
1c79356b 416
2d21ac55
A
417 if ((socket_cache_tail = p->cache_prev))
418 p->cache_prev->cache_next = 0;
419 if (--cached_sock_count == 0)
420 socket_cache_head = 0;
1c79356b 421
91447636 422 zfree(so_cache_zone, p);
2d21ac55
A
423
424 if (++n_freed >= SO_CACHE_MAX_FREE_BATCH) {
425 so_cache_max_freed++;
1c79356b
A
426 break;
427 }
428 }
91447636 429 lck_mtx_unlock(so_cache_mtx);
1c79356b
A
430
431 timeout(so_cache_timer, NULL, (SO_CACHE_FLUSH_INTERVAL * hz));
1c79356b 432}
9bccf70c 433#endif /* __APPLE__ */
1c79356b
A
434
435/*
436 * Get a socket structure from our zone, and initialize it.
437 * We don't implement `waitok' yet (see comments in uipc_domain.c).
438 * Note that it would probably be better to allocate socket
439 * and PCB at the same time, but I'm not convinced that all
440 * the protocols can be easily modified to do this.
441 */
442struct socket *
2d21ac55 443soalloc(int waitok, int dom, int type)
1c79356b
A
444{
445 struct socket *so;
446
2d21ac55
A
447 if ((dom == PF_INET) && (type == SOCK_STREAM)) {
448 cached_sock_alloc(&so, waitok);
449 } else {
450 MALLOC_ZONE(so, struct socket *, sizeof (*so), socket_zone,
451 M_WAITOK);
452 if (so != NULL)
453 bzero(so, sizeof (*so));
1c79356b
A
454 }
455 /* XXX race condition for reentrant kernel */
91447636 456//###LD Atomic add for so_gencnt
2d21ac55
A
457 if (so != NULL) {
458 so->so_gencnt = ++so_gencnt;
459 so->so_zone = socket_zone;
460#if CONFIG_MACF_SOCKET
461 /* Convert waitok to M_WAITOK/M_NOWAIT for MAC Framework. */
462 if (mac_socket_label_init(so, !waitok) != 0) {
463 sodealloc(so);
464 return (NULL);
465 }
466#endif /* MAC_SOCKET */
1c79356b
A
467 }
468
2d21ac55 469 return (so);
1c79356b
A
470}
471
2d21ac55
A
472/*
473 * Returns: 0 Success
474 * EAFNOSUPPORT
475 * EPROTOTYPE
476 * EPROTONOSUPPORT
477 * ENOBUFS
478 * <pru_attach>:ENOBUFS[AF_UNIX]
479 * <pru_attach>:ENOBUFS[TCP]
480 * <pru_attach>:ENOMEM[TCP]
481 * <pru_attach>:EISCONN[TCP]
482 * <pru_attach>:??? [other protocol families, IPSEC]
483 */
1c79356b 484int
2d21ac55 485socreate(int dom, struct socket **aso, int type, int proto)
1c79356b
A
486{
487 struct proc *p = current_proc();
488 register struct protosw *prp;
9bccf70c 489 register struct socket *so;
1c79356b 490 register int error = 0;
d1ecb069
A
491 thread_t thread;
492 struct uthread *ut;
493
55e303ae
A
494#if TCPDEBUG
495 extern int tcpconsdebug;
496#endif
1c79356b
A
497 if (proto)
498 prp = pffindproto(dom, proto, type);
499 else
500 prp = pffindtype(dom, type);
9bccf70c 501
2d21ac55
A
502 if (prp == 0 || prp->pr_usrreqs->pru_attach == 0) {
503 if (pffinddomain(dom) == NULL) {
504 return (EAFNOSUPPORT);
505 }
506 if (proto != 0) {
507 if (pffindprotonotype(dom, proto) != NULL) {
508 return (EPROTOTYPE);
509 }
510 }
9bccf70c
A
511 return (EPROTONOSUPPORT);
512 }
1c79356b
A
513 if (prp->pr_type != type)
514 return (EPROTOTYPE);
b0d623f7 515 so = soalloc(1, dom, type);
1c79356b
A
516 if (so == 0)
517 return (ENOBUFS);
518
519 TAILQ_INIT(&so->so_incomp);
520 TAILQ_INIT(&so->so_comp);
521 so->so_type = type;
522
b0d623f7
A
523 so->so_uid = kauth_cred_getuid(kauth_cred_get());
524 if (!suser(kauth_cred_get(), NULL))
525 so->so_state = SS_PRIV;
526
1c79356b 527 so->so_proto = prp;
9bccf70c 528#ifdef __APPLE__
1c79356b 529 so->so_rcv.sb_flags |= SB_RECV; /* XXX */
91447636 530 so->so_rcv.sb_so = so->so_snd.sb_so = so;
9bccf70c 531#endif
0c530ab8
A
532 so->next_lock_lr = 0;
533 so->next_unlock_lr = 0;
2d21ac55
A
534
535#if CONFIG_MACF_SOCKET
536 mac_socket_label_associate(kauth_cred_get(), so);
537#endif /* MAC_SOCKET */
538
91447636 539//### Attachement will create the per pcb lock if necessary and increase refcount
2d21ac55
A
540 /*
541 * for creation, make sure it's done before
542 * socket is inserted in lists
543 */
544 so->so_usecount++;
91447636
A
545
546 error = (*prp->pr_usrreqs->pru_attach)(so, proto, p);
1c79356b 547 if (error) {
2d21ac55
A
548 /*
549 * Warning:
550 * If so_pcb is not zero, the socket will be leaked,
551 * so protocol attachment handler must be coded carefuly
55e303ae 552 */
1c79356b 553 so->so_state |= SS_NOFDREF;
37839358
A
554 so->so_usecount--;
555 sofreelastref(so, 1); /* will deallocate the socket */
1c79356b
A
556 return (error);
557 }
9bccf70c 558#ifdef __APPLE__
1c79356b 559 prp->pr_domain->dom_refs++;
1c79356b 560 TAILQ_INIT(&so->so_evlist);
91447636
A
561
562 /* Attach socket filters for this protocol */
563 sflt_initsock(so);
55e303ae
A
564#if TCPDEBUG
565 if (tcpconsdebug == 2)
566 so->so_options |= SO_DEBUG;
567#endif
9bccf70c 568#endif
d1ecb069
A
569 /*
570 * If this is a background thread/task, mark the socket as such.
571 */
572 thread = current_thread();
573 ut = get_bsdthread_info(thread);
574 if (uthread_get_background_state(ut)) {
575 socket_set_traffic_mgt_flags(so, TRAFFIC_MGT_SO_BACKGROUND);
576 so->so_background_thread = thread;
577 /*
578 * In case setpriority(PRIO_DARWIN_THREAD) was called
579 * on this thread, regulate network (TCP) traffics.
580 */
581 if (ut->uu_flag & UT_BACKGROUND_TRAFFIC_MGT) {
582 socket_set_traffic_mgt_flags(so,
583 TRAFFIC_MGT_SO_BG_REGULATE);
584 }
585 }
586
1c79356b
A
587 *aso = so;
588 return (0);
589}
590
2d21ac55
A
591/*
592 * Returns: 0 Success
593 * <pru_bind>:EINVAL Invalid argument [COMMON_START]
594 * <pru_bind>:EAFNOSUPPORT Address family not supported
595 * <pru_bind>:EADDRNOTAVAIL Address not available.
596 * <pru_bind>:EINVAL Invalid argument
597 * <pru_bind>:EAFNOSUPPORT Address family not supported [notdef]
598 * <pru_bind>:EACCES Permission denied
599 * <pru_bind>:EADDRINUSE Address in use
600 * <pru_bind>:EAGAIN Resource unavailable, try again
601 * <pru_bind>:EPERM Operation not permitted
602 * <pru_bind>:???
603 * <sf_bind>:???
604 *
605 * Notes: It's not possible to fully enumerate the return codes above,
606 * since socket filter authors and protocol family authors may
607 * not choose to limit their error returns to those listed, even
608 * though this may result in some software operating incorrectly.
609 *
610 * The error codes which are enumerated above are those known to
611 * be returned by the tcp_usr_bind function supplied.
612 */
1c79356b 613int
2d21ac55 614sobind(struct socket *so, struct sockaddr *nam)
1c79356b
A
615{
616 struct proc *p = current_proc();
91447636 617 int error = 0;
2d21ac55
A
618 struct socket_filter_entry *filter;
619 int filtered = 0;
1c79356b 620
91447636
A
621 socket_lock(so, 1);
622
2d21ac55
A
623 /*
624 * If this is a bind request on a previously-accepted socket
625 * that has been marked as inactive, reject it now before
626 * we go any further.
627 */
628 if (so->so_flags & SOF_DEFUNCT) {
629 error = EINVAL;
630 goto out;
631 }
632
91447636
A
633 /* Socket filter */
634 error = 0;
635 for (filter = so->so_filt; filter && (error == 0);
2d21ac55 636 filter = filter->sfe_next_onsocket) {
91447636
A
637 if (filter->sfe_filter->sf_filter.sf_bind) {
638 if (filtered == 0) {
639 filtered = 1;
640 sflt_use(so);
641 socket_unlock(so, 0);
1c79356b 642 }
2d21ac55
A
643 error = filter->sfe_filter->sf_filter.
644 sf_bind(filter->sfe_cookie, so, nam);
1c79356b
A
645 }
646 }
91447636
A
647 if (filtered != 0) {
648 socket_lock(so, 0);
649 sflt_unuse(so);
650 }
651 /* End socket filter */
2d21ac55 652
91447636
A
653 if (error == 0)
654 error = (*so->so_proto->pr_usrreqs->pru_bind)(so, nam, p);
2d21ac55 655out:
91447636 656 socket_unlock(so, 1);
2d21ac55 657
91447636
A
658 if (error == EJUSTRETURN)
659 error = 0;
2d21ac55 660
1c79356b
A
661 return (error);
662}
663
664void
2d21ac55 665sodealloc(struct socket *so)
1c79356b
A
666{
667 so->so_gencnt = ++so_gencnt;
668
2d21ac55
A
669#if CONFIG_MACF_SOCKET
670 mac_socket_label_destroy(so);
671#endif /* MAC_SOCKET */
672 if (so->cached_in_sock_layer == 1) {
673 cached_sock_free(so);
674 } else {
675 if (so->cached_in_sock_layer == -1)
676 panic("sodealloc: double dealloc: so=%p\n", so);
677 so->cached_in_sock_layer = -1;
678 FREE_ZONE(so, sizeof (*so), so->so_zone);
91447636 679 }
1c79356b
A
680}
681
2d21ac55
A
682/*
683 * Returns: 0 Success
684 * EINVAL
685 * EOPNOTSUPP
686 * <pru_listen>:EINVAL[AF_UNIX]
687 * <pru_listen>:EINVAL[TCP]
688 * <pru_listen>:EADDRNOTAVAIL[TCP] Address not available.
689 * <pru_listen>:EINVAL[TCP] Invalid argument
690 * <pru_listen>:EAFNOSUPPORT[TCP] Address family not supported [notdef]
691 * <pru_listen>:EACCES[TCP] Permission denied
692 * <pru_listen>:EADDRINUSE[TCP] Address in use
693 * <pru_listen>:EAGAIN[TCP] Resource unavailable, try again
694 * <pru_listen>:EPERM[TCP] Operation not permitted
695 * <sf_listen>:???
696 *
697 * Notes: Other <pru_listen> returns depend on the protocol family; all
698 * <sf_listen> returns depend on what the filter author causes
699 * their filter to return.
700 */
1c79356b 701int
2d21ac55 702solisten(struct socket *so, int backlog)
1c79356b 703{
1c79356b 704 struct proc *p = current_proc();
2d21ac55
A
705 int error = 0;
706 struct socket_filter_entry *filter;
707 int filtered = 0;
1c79356b 708
91447636 709 socket_lock(so, 1);
2d21ac55
A
710 if (so->so_proto == NULL) {
711 error = EINVAL;
712 goto out;
713 }
714 if ((so->so_proto->pr_flags & PR_CONNREQUIRED) == 0) {
715 error = EOPNOTSUPP;
716 goto out;
717 }
718
719 /*
720 * If the listen request is made on a socket that is not fully
721 * disconnected, or on a previously-accepted socket that has
722 * been marked as inactive, reject the request now.
723 */
724 if ((so->so_state &
725 (SS_ISCONNECTED|SS_ISCONNECTING|SS_ISDISCONNECTING)) ||
726 (so->so_flags & SOF_DEFUNCT)) {
727 error = EINVAL;
728 goto out;
729 }
730
731 if ((so->so_restrictions & SO_RESTRICT_DENYIN) != 0) {
732 error = EPERM;
733 goto out;
734 }
735
736 error = 0;
737 for (filter = so->so_filt; filter && (error == 0);
738 filter = filter->sfe_next_onsocket) {
739 if (filter->sfe_filter->sf_filter.sf_listen) {
740 if (filtered == 0) {
741 filtered = 1;
742 sflt_use(so);
743 socket_unlock(so, 0);
91447636 744 }
2d21ac55
A
745 error = filter->sfe_filter->sf_filter.
746 sf_listen(filter->sfe_cookie, so);
91447636 747 }
2d21ac55
A
748 }
749 if (filtered != 0) {
750 socket_lock(so, 0);
751 sflt_unuse(so);
91447636
A
752 }
753
754 if (error == 0) {
755 error = (*so->so_proto->pr_usrreqs->pru_listen)(so, p);
756 }
2d21ac55 757
1c79356b 758 if (error) {
91447636
A
759 if (error == EJUSTRETURN)
760 error = 0;
2d21ac55 761 goto out;
1c79356b 762 }
2d21ac55 763
91447636 764 if (TAILQ_EMPTY(&so->so_comp))
1c79356b 765 so->so_options |= SO_ACCEPTCONN;
2d21ac55
A
766 /*
767 * POSIX: The implementation may have an upper limit on the length of
768 * the listen queue-either global or per accepting socket. If backlog
769 * exceeds this limit, the length of the listen queue is set to the
770 * limit.
771 *
772 * If listen() is called with a backlog argument value that is less
773 * than 0, the function behaves as if it had been called with a backlog
774 * argument value of 0.
775 *
776 * A backlog argument of 0 may allow the socket to accept connections,
777 * in which case the length of the listen queue may be set to an
778 * implementation-defined minimum value.
779 */
780 if (backlog <= 0 || backlog > somaxconn)
1c79356b 781 backlog = somaxconn;
1c79356b 782
2d21ac55
A
783 so->so_qlimit = backlog;
784out:
91447636 785 socket_unlock(so, 1);
2d21ac55 786 return (error);
1c79356b
A
787}
788
1c79356b 789void
2d21ac55 790sofreelastref(struct socket *so, int dealloc)
9bccf70c 791{
1c79356b
A
792 struct socket *head = so->so_head;
793
2d21ac55 794 /* Assume socket is locked */
1c79356b 795
3a60a9f5
A
796 /* Remove any filters - may be called more than once */
797 sflt_termsock(so);
2d21ac55
A
798
799 if ((!(so->so_flags & SOF_PCBCLEARING)) ||
800 ((so->so_state & SS_NOFDREF) == 0)) {
9bccf70c 801#ifdef __APPLE__
0b4e3aa0
A
802 selthreadclear(&so->so_snd.sb_sel);
803 selthreadclear(&so->so_rcv.sb_sel);
cc9f6e38
A
804 so->so_rcv.sb_flags &= ~SB_UPCALL;
805 so->so_snd.sb_flags &= ~SB_UPCALL;
9bccf70c 806#endif
1c79356b 807 return;
0b4e3aa0 808 }
9bccf70c 809 if (head != NULL) {
91447636 810 socket_lock(head, 1);
9bccf70c
A
811 if (so->so_state & SS_INCOMP) {
812 TAILQ_REMOVE(&head->so_incomp, so, so_list);
813 head->so_incqlen--;
814 } else if (so->so_state & SS_COMP) {
815 /*
816 * We must not decommission a socket that's
817 * on the accept(2) queue. If we do, then
818 * accept(2) may hang after select(2) indicated
819 * that the listening socket was ready.
820 */
821#ifdef __APPLE__
822 selthreadclear(&so->so_snd.sb_sel);
823 selthreadclear(&so->so_rcv.sb_sel);
cc9f6e38
A
824 so->so_rcv.sb_flags &= ~SB_UPCALL;
825 so->so_snd.sb_flags &= ~SB_UPCALL;
9bccf70c 826#endif
91447636 827 socket_unlock(head, 1);
9bccf70c
A
828 return;
829 } else {
830 panic("sofree: not queued");
831 }
1c79356b 832 head->so_qlen--;
9bccf70c 833 so->so_state &= ~SS_INCOMP;
1c79356b 834 so->so_head = NULL;
91447636 835 socket_unlock(head, 1);
1c79356b 836 }
9bccf70c 837#ifdef __APPLE__
0b4e3aa0 838 selthreadclear(&so->so_snd.sb_sel);
1c79356b 839 sbrelease(&so->so_snd);
9bccf70c 840#endif
1c79356b 841 sorflush(so);
2d21ac55 842
91447636
A
843 /* 3932268: disable upcall */
844 so->so_rcv.sb_flags &= ~SB_UPCALL;
845 so->so_snd.sb_flags &= ~SB_UPCALL;
2d21ac55 846
91447636
A
847 if (dealloc)
848 sodealloc(so);
1c79356b
A
849}
850
2d21ac55
A
851void
852soclose_wait_locked(struct socket *so)
853{
854 lck_mtx_t *mutex_held;
855
856 if (so->so_proto->pr_getlock != NULL)
857 mutex_held = (*so->so_proto->pr_getlock)(so, 0);
858 else
859 mutex_held = so->so_proto->pr_domain->dom_mtx;
860 lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
861
4a3eedf9
A
862 /*
863 * Double check here and return if there's no outstanding upcall;
864 * otherwise proceed further only if SOF_UPCALLCLOSEWAIT is set.
865 */
866 if (!(so->so_flags & SOF_UPCALLINUSE) ||
867 !(so->so_flags & SOF_UPCALLCLOSEWAIT))
2d21ac55
A
868 return;
869
870 so->so_flags |= SOF_CLOSEWAIT;
871 (void) msleep((caddr_t)&so->so_upcall, mutex_held, (PZERO - 1),
872 "soclose_wait_locked", NULL);
873 lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
874 so->so_flags &= ~SOF_CLOSEWAIT;
875}
876
1c79356b
A
877/*
878 * Close a socket on last file table reference removal.
879 * Initiate disconnect if connected.
880 * Free socket when disconnect complete.
881 */
882int
2d21ac55 883soclose_locked(struct socket *so)
1c79356b 884{
1c79356b 885 int error = 0;
2d21ac55 886 lck_mtx_t *mutex_held;
91447636 887 struct timespec ts;
1c79356b 888
91447636 889 if (so->so_usecount == 0) {
2d21ac55 890 panic("soclose: so=%p refcount=0\n", so);
1c79356b
A
891 }
892
91447636 893 sflt_notify(so, sock_evt_closing, NULL);
2d21ac55 894
91447636 895 if ((so->so_options & SO_ACCEPTCONN)) {
2d21ac55
A
896 struct socket *sp, *sonext;
897 int socklock = 0;
898
899 /*
900 * We do not want new connection to be added
901 * to the connection queues
902 */
91447636 903 so->so_options &= ~SO_ACCEPTCONN;
2d21ac55
A
904
905 for (sp = TAILQ_FIRST(&so->so_incomp); sp != NULL; sp = sonext) {
906 sonext = TAILQ_NEXT(sp, so_list);
907
908 /* Radar 5350314
909 * skip sockets thrown away by tcpdropdropblreq
910 * they will get cleanup by the garbage collection.
911 * otherwise, remove the incomp socket from the queue
912 * and let soabort trigger the appropriate cleanup.
91447636 913 */
2d21ac55
A
914 if (sp->so_flags & SOF_OVERFLOW)
915 continue;
916
ff6e181a 917 if (so->so_proto->pr_getlock != NULL) {
2d21ac55
A
918 /* lock ordering for consistency with the rest of the stack,
919 * we lock the socket first and then grabb the head.
920 */
91447636 921 socket_unlock(so, 0);
ff6e181a 922 socket_lock(sp, 1);
ff6e181a 923 socket_lock(so, 0);
2d21ac55
A
924 socklock = 1;
925 }
926
927 TAILQ_REMOVE(&so->so_incomp, sp, so_list);
928 so->so_incqlen--;
929
930 if (sp->so_state & SS_INCOMP) {
931 sp->so_state &= ~SS_INCOMP;
932 sp->so_head = NULL;
933
934 (void) soabort(sp);
ff6e181a 935 }
2d21ac55
A
936
937 if (socklock)
938 socket_unlock(sp, 1);
91447636
A
939 }
940
941 while ((sp = TAILQ_FIRST(&so->so_comp)) != NULL) {
91447636 942 /* Dequeue from so_comp since sofree() won't do it */
2d21ac55 943 TAILQ_REMOVE(&so->so_comp, sp, so_list);
91447636 944 so->so_qlen--;
ff6e181a
A
945
946 if (so->so_proto->pr_getlock != NULL) {
947 socket_unlock(so, 0);
948 socket_lock(sp, 1);
949 }
950
2d21ac55
A
951 if (sp->so_state & SS_COMP) {
952 sp->so_state &= ~SS_COMP;
953 sp->so_head = NULL;
954
955 (void) soabort(sp);
956 }
91447636 957
ff6e181a 958 if (so->so_proto->pr_getlock != NULL) {
91447636 959 socket_unlock(sp, 1);
ff6e181a
A
960 socket_lock(so, 0);
961 }
91447636
A
962 }
963 }
964 if (so->so_pcb == 0) {
965 /* 3915887: mark the socket as ready for dealloc */
966 so->so_flags |= SOF_PCBCLEARING;
1c79356b 967 goto discard;
91447636 968 }
1c79356b
A
969 if (so->so_state & SS_ISCONNECTED) {
970 if ((so->so_state & SS_ISDISCONNECTING) == 0) {
91447636 971 error = sodisconnectlocked(so);
1c79356b
A
972 if (error)
973 goto drop;
974 }
975 if (so->so_options & SO_LINGER) {
976 if ((so->so_state & SS_ISDISCONNECTING) &&
977 (so->so_state & SS_NBIO))
978 goto drop;
2d21ac55 979 if (so->so_proto->pr_getlock != NULL)
91447636 980 mutex_held = (*so->so_proto->pr_getlock)(so, 0);
2d21ac55 981 else
91447636 982 mutex_held = so->so_proto->pr_domain->dom_mtx;
1c79356b 983 while (so->so_state & SS_ISCONNECTED) {
91447636 984 ts.tv_sec = (so->so_linger/100);
2d21ac55
A
985 ts.tv_nsec = (so->so_linger % 100) *
986 NSEC_PER_USEC * 1000 * 10;
987 error = msleep((caddr_t)&so->so_timeo,
988 mutex_held, PSOCK | PCATCH, "soclose", &ts);
91447636 989 if (error) {
2d21ac55
A
990 /*
991 * It's OK when the time fires,
992 * don't report an error
993 */
91447636
A
994 if (error == EWOULDBLOCK)
995 error = 0;
1c79356b 996 break;
91447636 997 }
1c79356b
A
998 }
999 }
1000 }
1001drop:
91447636 1002 if (so->so_usecount == 0)
2d21ac55 1003 panic("soclose: usecount is zero so=%p\n", so);
91447636 1004 if (so->so_pcb && !(so->so_flags & SOF_PCBCLEARING)) {
1c79356b
A
1005 int error2 = (*so->so_proto->pr_usrreqs->pru_detach)(so);
1006 if (error == 0)
1007 error = error2;
1008 }
91447636 1009 if (so->so_usecount <= 0)
2d21ac55 1010 panic("soclose: usecount is zero so=%p\n", so);
1c79356b 1011discard:
e3027f41 1012 if (so->so_pcb && so->so_state & SS_NOFDREF)
1c79356b
A
1013 panic("soclose: NOFDREF");
1014 so->so_state |= SS_NOFDREF;
9bccf70c 1015#ifdef __APPLE__
1c79356b
A
1016 so->so_proto->pr_domain->dom_refs--;
1017 evsofree(so);
9bccf70c 1018#endif
91447636 1019 so->so_usecount--;
1c79356b 1020 sofree(so);
1c79356b
A
1021 return (error);
1022}
1023
91447636 1024int
2d21ac55 1025soclose(struct socket *so)
91447636
A
1026{
1027 int error = 0;
1028 socket_lock(so, 1);
2d21ac55
A
1029
1030 if (so->so_flags & SOF_UPCALLINUSE)
1031 soclose_wait_locked(so);
1032
1033 if (so->so_retaincnt == 0) {
91447636 1034 error = soclose_locked(so);
2d21ac55
A
1035 } else {
1036 /*
1037 * if the FD is going away, but socket is
1038 * retained in kernel remove its reference
1039 */
91447636
A
1040 so->so_usecount--;
1041 if (so->so_usecount < 2)
2d21ac55
A
1042 panic("soclose: retaincnt non null and so=%p "
1043 "usecount=%d\n", so, so->so_usecount);
91447636
A
1044 }
1045 socket_unlock(so, 1);
1046 return (error);
1047}
1048
1c79356b
A
1049/*
1050 * Must be called at splnet...
1051 */
2d21ac55 1052/* Should already be locked */
1c79356b 1053int
2d21ac55 1054soabort(struct socket *so)
1c79356b 1055{
9bccf70c 1056 int error;
1c79356b 1057
91447636 1058#ifdef MORE_LOCKING_DEBUG
2d21ac55 1059 lck_mtx_t *mutex_held;
91447636 1060
2d21ac55 1061 if (so->so_proto->pr_getlock != NULL)
91447636 1062 mutex_held = (*so->so_proto->pr_getlock)(so, 0);
2d21ac55 1063 else
91447636
A
1064 mutex_held = so->so_proto->pr_domain->dom_mtx;
1065 lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
1066#endif
1067
2d21ac55
A
1068 if ((so->so_flags & SOF_ABORTED) == 0) {
1069 so->so_flags |= SOF_ABORTED;
1070 error = (*so->so_proto->pr_usrreqs->pru_abort)(so);
1071 if (error) {
1072 sofree(so);
1073 return (error);
1074 }
9bccf70c
A
1075 }
1076 return (0);
1c79356b
A
1077}
1078
1079int
2d21ac55 1080soacceptlock(struct socket *so, struct sockaddr **nam, int dolock)
9bccf70c 1081{
1c79356b 1082 int error;
91447636 1083
2d21ac55
A
1084 if (dolock)
1085 socket_lock(so, 1);
1c79356b
A
1086
1087 if ((so->so_state & SS_NOFDREF) == 0)
1088 panic("soaccept: !NOFDREF");
1089 so->so_state &= ~SS_NOFDREF;
1090 error = (*so->so_proto->pr_usrreqs->pru_accept)(so, nam);
2d21ac55
A
1091
1092 if (dolock)
1093 socket_unlock(so, 1);
1c79356b
A
1094 return (error);
1095}
2d21ac55 1096
91447636 1097int
2d21ac55 1098soaccept(struct socket *so, struct sockaddr **nam)
91447636
A
1099{
1100 return (soacceptlock(so, nam, 1));
1101}
1c79356b
A
1102
1103int
2d21ac55
A
1104soacceptfilter(struct socket *so)
1105{
1106 struct sockaddr *local = NULL, *remote = NULL;
1107 struct socket_filter_entry *filter;
1108 int error = 0, filtered = 0;
1109 struct socket *head = so->so_head;
1110
1111 /*
b0d623f7 1112 * Hold the lock even if this socket
2d21ac55 1113 * has not been made visible to the filter(s).
b0d623f7
A
1114 * For sockets with global locks, this protect against the
1115 * head or peer going away
2d21ac55 1116 */
b0d623f7
A
1117 socket_lock(so, 1);
1118 if (sogetaddr_locked(so, &remote, 1) != 0 ||
1119 sogetaddr_locked(so, &local, 0) != 0) {
2d21ac55
A
1120 so->so_state &= ~(SS_NOFDREF | SS_COMP);
1121 so->so_head = NULL;
b0d623f7 1122 socket_unlock(so, 1);
2d21ac55
A
1123 soclose(so);
1124 /* Out of resources; try it again next time */
1125 error = ECONNABORTED;
1126 goto done;
1127 }
1128
1129 /*
1130 * At this point, we have a reference on the listening socket
1131 * so we know it won't be going away. Do the same for the newly
1132 * accepted socket while we invoke the accept callback routine.
1133 */
2d21ac55
A
1134 for (filter = so->so_filt; filter != NULL && error == 0;
1135 filter = filter->sfe_next_onsocket) {
1136 if (filter->sfe_filter->sf_filter.sf_accept != NULL) {
1137 if (!filtered) {
1138 filtered = 1;
1139 sflt_use(so);
1140 socket_unlock(so, 0);
1141 }
1142 error = filter->sfe_filter->sf_filter.
1143 sf_accept(filter->sfe_cookie,
1144 head, so, local, remote);
1145 }
1146 }
1147
1148 if (filtered) {
1149 socket_lock(so, 0);
1150 sflt_unuse(so);
1151 }
1152
1153 /*
1154 * If we get EJUSTRETURN from one of the filters, mark this socket
1155 * as inactive and return it anyway. This newly accepted socket
1156 * will be disconnected later before we hand it off to the caller.
1157 */
1158 if (error == EJUSTRETURN) {
1159 error = 0;
1160 so->so_flags |= SOF_DEFUNCT;
1161 /* Prevent data from being appended to the socket buffers */
1162 so->so_snd.sb_flags |= SB_DROP;
1163 so->so_rcv.sb_flags |= SB_DROP;
1164 }
1165
1166 if (error != 0) {
1167 /*
1168 * This may seem like a duplication to the above error
1169 * handling part when we return ECONNABORTED, except
1170 * the following is done while holding the lock since
1171 * the socket has been exposed to the filter(s) earlier.
1172 */
1173 so->so_state &= ~(SS_NOFDREF | SS_COMP);
1174 so->so_head = NULL;
1175 socket_unlock(so, 1);
1176 soclose(so);
1177 /* Propagate socket filter's error code to the caller */
1178 } else {
1179 socket_unlock(so, 1);
1180 }
1181done:
1182 /* Callee checks for NULL pointer */
1183 sock_freeaddr(remote);
1184 sock_freeaddr(local);
1185 return (error);
1186}
1c79356b 1187
2d21ac55
A
1188/*
1189 * Returns: 0 Success
1190 * EOPNOTSUPP Operation not supported on socket
1191 * EISCONN Socket is connected
1192 * <pru_connect>:EADDRNOTAVAIL Address not available.
1193 * <pru_connect>:EINVAL Invalid argument
1194 * <pru_connect>:EAFNOSUPPORT Address family not supported [notdef]
1195 * <pru_connect>:EACCES Permission denied
1196 * <pru_connect>:EADDRINUSE Address in use
1197 * <pru_connect>:EAGAIN Resource unavailable, try again
1198 * <pru_connect>:EPERM Operation not permitted
1199 * <sf_connect_out>:??? [anything a filter writer might set]
1200 */
1201int
1202soconnectlock(struct socket *so, struct sockaddr *nam, int dolock)
1c79356b 1203{
1c79356b
A
1204 int error;
1205 struct proc *p = current_proc();
1c79356b 1206
2d21ac55
A
1207 if (dolock)
1208 socket_lock(so, 1);
91447636 1209
2d21ac55
A
1210 /*
1211 * If this is a listening socket or if this is a previously-accepted
1212 * socket that has been marked as inactive, reject the connect request.
1213 */
1214 if ((so->so_options & SO_ACCEPTCONN) || (so->so_flags & SOF_DEFUNCT)) {
1215 if (dolock)
1216 socket_unlock(so, 1);
1c79356b 1217 return (EOPNOTSUPP);
91447636 1218 }
2d21ac55
A
1219
1220 if ((so->so_restrictions & SO_RESTRICT_DENYOUT) != 0) {
1221 if (dolock)
1222 socket_unlock(so, 1);
1223 return (EPERM);
1224 }
1225
1c79356b
A
1226 /*
1227 * If protocol is connection-based, can only connect once.
1228 * Otherwise, if connected, try to disconnect first.
1229 * This allows user to disconnect by connecting to, e.g.,
1230 * a null address.
1231 */
1232 if (so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING) &&
1233 ((so->so_proto->pr_flags & PR_CONNREQUIRED) ||
2d21ac55 1234 (error = sodisconnectlocked(so)))) {
1c79356b 1235 error = EISCONN;
2d21ac55 1236 } else {
91447636
A
1237 /*
1238 * Run connect filter before calling protocol:
1239 * - non-blocking connect returns before completion;
1240 */
2d21ac55
A
1241 struct socket_filter_entry *filter;
1242 int filtered = 0;
1243
1244 error = 0;
1245 for (filter = so->so_filt; filter && (error == 0);
1246 filter = filter->sfe_next_onsocket) {
1247 if (filter->sfe_filter->sf_filter.sf_connect_out) {
1248 if (filtered == 0) {
1249 filtered = 1;
1250 sflt_use(so);
1251 socket_unlock(so, 0);
91447636 1252 }
2d21ac55
A
1253 error = filter->sfe_filter->sf_filter.
1254 sf_connect_out(filter->sfe_cookie, so, nam);
91447636 1255 }
91447636 1256 }
2d21ac55
A
1257 if (filtered != 0) {
1258 socket_lock(so, 0);
1259 sflt_unuse(so);
1260 }
1261
91447636
A
1262 if (error) {
1263 if (error == EJUSTRETURN)
1264 error = 0;
2d21ac55
A
1265 if (dolock)
1266 socket_unlock(so, 1);
1267 return (error);
91447636 1268 }
2d21ac55 1269
1c79356b 1270 error = (*so->so_proto->pr_usrreqs->pru_connect)(so, nam, p);
1c79356b 1271 }
2d21ac55
A
1272 if (dolock)
1273 socket_unlock(so, 1);
1c79356b
A
1274 return (error);
1275}
1276
91447636 1277int
2d21ac55 1278soconnect(struct socket *so, struct sockaddr *nam)
91447636
A
1279{
1280 return (soconnectlock(so, nam, 1));
1281}
1282
2d21ac55
A
1283/*
1284 * Returns: 0 Success
1285 * <pru_connect2>:EINVAL[AF_UNIX]
1286 * <pru_connect2>:EPROTOTYPE[AF_UNIX]
1287 * <pru_connect2>:??? [other protocol families]
1288 *
1289 * Notes: <pru_connect2> is not supported by [TCP].
1290 */
1c79356b 1291int
2d21ac55 1292soconnect2(struct socket *so1, struct socket *so2)
1c79356b 1293{
1c79356b 1294 int error;
91447636 1295
0c530ab8 1296 socket_lock(so1, 1);
2d21ac55 1297 if (so2->so_proto->pr_lock)
0c530ab8 1298 socket_lock(so2, 1);
1c79356b
A
1299
1300 error = (*so1->so_proto->pr_usrreqs->pru_connect2)(so1, so2);
2d21ac55 1301
0c530ab8 1302 socket_unlock(so1, 1);
2d21ac55 1303 if (so2->so_proto->pr_lock)
0c530ab8 1304 socket_unlock(so2, 1);
1c79356b
A
1305 return (error);
1306}
1307
1308int
2d21ac55 1309sodisconnectlocked(struct socket *so)
1c79356b 1310{
1c79356b 1311 int error;
1c79356b
A
1312
1313 if ((so->so_state & SS_ISCONNECTED) == 0) {
1314 error = ENOTCONN;
1315 goto bad;
1316 }
1317 if (so->so_state & SS_ISDISCONNECTING) {
1318 error = EALREADY;
1319 goto bad;
1320 }
2d21ac55 1321
1c79356b 1322 error = (*so->so_proto->pr_usrreqs->pru_disconnect)(so);
2d21ac55 1323
9bccf70c 1324 if (error == 0) {
91447636 1325 sflt_notify(so, sock_evt_disconnected, NULL);
1c79356b 1326 }
1c79356b 1327bad:
1c79356b
A
1328 return (error);
1329}
2d21ac55
A
1330
1331/* Locking version */
91447636 1332int
2d21ac55 1333sodisconnect(struct socket *so)
91447636 1334{
2d21ac55 1335 int error;
91447636
A
1336
1337 socket_lock(so, 1);
1338 error = sodisconnectlocked(so);
1339 socket_unlock(so, 1);
2d21ac55 1340 return (error);
91447636 1341}
1c79356b
A
1342
1343#define SBLOCKWAIT(f) (((f) & MSG_DONTWAIT) ? M_DONTWAIT : M_WAIT)
91447636
A
1344
1345/*
1346 * sosendcheck will lock the socket buffer if it isn't locked and
1347 * verify that there is space for the data being inserted.
2d21ac55
A
1348 *
1349 * Returns: 0 Success
1350 * EPIPE
1351 * sblock:EWOULDBLOCK
1352 * sblock:EINTR
1353 * sbwait:EBADF
1354 * sbwait:EINTR
1355 * [so_error]:???
91447636 1356 */
91447636 1357static int
b0d623f7
A
1358sosendcheck(struct socket *so, struct sockaddr *addr, int32_t resid, int32_t clen,
1359 int32_t atomic, int flags, int *sblocked)
91447636 1360{
b0d623f7
A
1361 int error = 0;
1362 int32_t space;
3a60a9f5 1363 int assumelock = 0;
91447636
A
1364
1365restart:
1366 if (*sblocked == 0) {
3a60a9f5 1367 if ((so->so_snd.sb_flags & SB_LOCK) != 0 &&
2d21ac55
A
1368 so->so_send_filt_thread != 0 &&
1369 so->so_send_filt_thread == current_thread()) {
3a60a9f5
A
1370 /*
1371 * We're being called recursively from a filter,
1372 * allow this to continue. Radar 4150520.
1373 * Don't set sblocked because we don't want
1374 * to perform an unlock later.
1375 */
1376 assumelock = 1;
2d21ac55 1377 } else {
3a60a9f5
A
1378 error = sblock(&so->so_snd, SBLOCKWAIT(flags));
1379 if (error) {
2d21ac55 1380 return (error);
3a60a9f5
A
1381 }
1382 *sblocked = 1;
1383 }
91447636 1384 }
2d21ac55
A
1385
1386 /*
1387 * If a send attempt is made on a previously-accepted socket
1388 * that has been marked as inactive (disconnected), reject
1389 * the request.
1390 */
1391 if (so->so_flags & SOF_DEFUNCT)
1392 return (ENOTCONN);
1393
1394 if (so->so_state & SS_CANTSENDMORE)
1395 return (EPIPE);
1396
91447636
A
1397 if (so->so_error) {
1398 error = so->so_error;
1399 so->so_error = 0;
2d21ac55 1400 return (error);
91447636 1401 }
2d21ac55 1402
91447636 1403 if ((so->so_state & SS_ISCONNECTED) == 0) {
2d21ac55 1404 if ((so->so_proto->pr_flags & PR_CONNREQUIRED) != 0) {
91447636 1405 if ((so->so_state & SS_ISCONFIRMING) == 0 &&
2d21ac55
A
1406 !(resid == 0 && clen != 0))
1407 return (ENOTCONN);
1408 } else if (addr == 0 && !(flags&MSG_HOLD)) {
1409 return ((so->so_proto->pr_flags & PR_CONNREQUIRED) ?
1410 ENOTCONN : EDESTADDRREQ);
1411 }
91447636
A
1412 }
1413 space = sbspace(&so->so_snd);
1414 if (flags & MSG_OOB)
1415 space += 1024;
1416 if ((atomic && resid > so->so_snd.sb_hiwat) ||
2d21ac55
A
1417 clen > so->so_snd.sb_hiwat)
1418 return (EMSGSIZE);
1419 if (space < resid + clen &&
b0d623f7 1420 (atomic || space < (int32_t)so->so_snd.sb_lowat || space < clen)) {
2d21ac55
A
1421 if ((so->so_state & SS_NBIO) || (flags & MSG_NBIO) ||
1422 assumelock) {
1423 return (EWOULDBLOCK);
3a60a9f5 1424 }
91447636
A
1425 sbunlock(&so->so_snd, 1);
1426 error = sbwait(&so->so_snd);
1427 if (error) {
2d21ac55 1428 return (error);
91447636
A
1429 }
1430 goto restart;
1431 }
2d21ac55
A
1432
1433 return (0);
91447636
A
1434}
1435
1c79356b
A
1436/*
1437 * Send on a socket.
1438 * If send must go all at once and message is larger than
1439 * send buffering, then hard error.
1440 * Lock against other senders.
1441 * If must go all at once and not enough room now, then
1442 * inform user that this would block and do nothing.
1443 * Otherwise, if nonblocking, send as much as possible.
1444 * The data to be sent is described by "uio" if nonzero,
1445 * otherwise by the mbuf chain "top" (which must be null
1446 * if uio is not). Data provided in mbuf chain must be small
1447 * enough to send all at once.
1448 *
1449 * Returns nonzero on error, timeout or signal; callers
1450 * must check for short counts if EINTR/ERESTART are returned.
1451 * Data and control buffers are freed on return.
1452 * Experiment:
1453 * MSG_HOLD: go thru most of sosend(), but just enqueue the mbuf
1454 * MSG_SEND: go thru as for MSG_HOLD on current fragment, then
1455 * point at the mbuf chain being constructed and go from there.
2d21ac55
A
1456 *
1457 * Returns: 0 Success
1458 * EOPNOTSUPP
1459 * EINVAL
1460 * ENOBUFS
1461 * uiomove:EFAULT
1462 * sosendcheck:EPIPE
1463 * sosendcheck:EWOULDBLOCK
1464 * sosendcheck:EINTR
1465 * sosendcheck:EBADF
1466 * sosendcheck:EINTR
1467 * sosendcheck:??? [value from so_error]
1468 * <pru_send>:ECONNRESET[TCP]
1469 * <pru_send>:EINVAL[TCP]
1470 * <pru_send>:ENOBUFS[TCP]
1471 * <pru_send>:EADDRINUSE[TCP]
1472 * <pru_send>:EADDRNOTAVAIL[TCP]
1473 * <pru_send>:EAFNOSUPPORT[TCP]
1474 * <pru_send>:EACCES[TCP]
1475 * <pru_send>:EAGAIN[TCP]
1476 * <pru_send>:EPERM[TCP]
1477 * <pru_send>:EMSGSIZE[TCP]
1478 * <pru_send>:EHOSTUNREACH[TCP]
1479 * <pru_send>:ENETUNREACH[TCP]
1480 * <pru_send>:ENETDOWN[TCP]
1481 * <pru_send>:ENOMEM[TCP]
1482 * <pru_send>:ENOBUFS[TCP]
1483 * <pru_send>:???[TCP] [ignorable: mostly IPSEC/firewall/DLIL]
1484 * <pru_send>:EINVAL[AF_UNIX]
1485 * <pru_send>:EOPNOTSUPP[AF_UNIX]
1486 * <pru_send>:EPIPE[AF_UNIX]
1487 * <pru_send>:ENOTCONN[AF_UNIX]
1488 * <pru_send>:EISCONN[AF_UNIX]
1489 * <pru_send>:???[AF_UNIX] [whatever a filter author chooses]
1490 * <sf_data_out>:??? [whatever a filter author chooses]
1491 *
1492 * Notes: Other <pru_send> returns depend on the protocol family; all
1493 * <sf_data_out> returns depend on what the filter author causes
1494 * their filter to return.
1c79356b
A
1495 */
1496int
2d21ac55
A
1497sosend(struct socket *so, struct sockaddr *addr, struct uio *uio,
1498 struct mbuf *top, struct mbuf *control, int flags)
1c79356b
A
1499{
1500 struct mbuf **mp;
fa4905b1 1501 register struct mbuf *m, *freelist = NULL;
b0d623f7 1502 register int32_t space, len, resid;
91447636 1503 int clen = 0, error, dontroute, mlen, sendflags;
1c79356b 1504 int atomic = sosendallatonce(so) || top;
91447636 1505 int sblocked = 0;
1c79356b 1506 struct proc *p = current_proc();
1c79356b 1507
2d21ac55 1508 if (uio) {
91447636
A
1509 // LP64todo - fix this!
1510 resid = uio_resid(uio);
2d21ac55 1511 } else {
1c79356b 1512 resid = top->m_pkthdr.len;
2d21ac55
A
1513 }
1514 KERNEL_DEBUG((DBG_FNC_SOSEND | DBG_FUNC_START), so, resid,
1515 so->so_snd.sb_cc, so->so_snd.sb_lowat, so->so_snd.sb_hiwat);
1c79356b 1516
91447636 1517 socket_lock(so, 1);
2d21ac55
A
1518 if (so->so_type != SOCK_STREAM && (flags & MSG_OOB) != 0) {
1519 error = EOPNOTSUPP;
1520 socket_unlock(so, 1);
1521 goto out;
1522 }
91447636 1523
1c79356b
A
1524 /*
1525 * In theory resid should be unsigned.
1526 * However, space must be signed, as it might be less than 0
1527 * if we over-committed, and we must use a signed comparison
1528 * of space and resid. On the other hand, a negative resid
1529 * causes us to loop sending 0-length segments to the protocol.
1530 *
1531 * Also check to make sure that MSG_EOR isn't used on SOCK_STREAM
1532 * type sockets since that's an error.
1533 */
91447636 1534 if (resid < 0 || (so->so_type == SOCK_STREAM && (flags & MSG_EOR))) {
1c79356b 1535 error = EINVAL;
91447636 1536 socket_unlock(so, 1);
1c79356b
A
1537 goto out;
1538 }
1539
1540 dontroute =
1541 (flags & MSG_DONTROUTE) && (so->so_options & SO_DONTROUTE) == 0 &&
1542 (so->so_proto->pr_flags & PR_ATOMIC);
b0d623f7 1543 OSIncrementAtomicLong(&p->p_stats->p_ru.ru_msgsnd);
1c79356b
A
1544 if (control)
1545 clen = control->m_len;
1c79356b 1546
1c79356b 1547 do {
2d21ac55
A
1548 error = sosendcheck(so, addr, resid, clen, atomic, flags,
1549 &sblocked);
91447636 1550 if (error) {
3a60a9f5 1551 goto release;
1c79356b 1552 }
1c79356b 1553 mp = &top;
2d21ac55
A
1554 space = sbspace(&so->so_snd) - clen + ((flags & MSG_OOB) ?
1555 1024 : 0);
fa4905b1 1556
1c79356b 1557 do {
2d21ac55
A
1558 struct socket_filter_entry *filter;
1559 int filtered;
1560 boolean_t recursive;
1561
1562 if (uio == NULL) {
91447636
A
1563 /*
1564 * Data is prepackaged in "top".
1565 */
1566 resid = 0;
1c79356b
A
1567 if (flags & MSG_EOR)
1568 top->m_flags |= M_EOR;
91447636 1569 } else {
2d21ac55
A
1570 int chainlength;
1571 int bytes_to_copy;
1572 boolean_t jumbocl;
1573
b0d623f7 1574 bytes_to_copy = imin(resid, space);
2d21ac55 1575
91447636
A
1576 if (sosendminchain > 0) {
1577 chainlength = 0;
2d21ac55 1578 } else {
91447636 1579 chainlength = sosendmaxchain;
2d21ac55
A
1580 }
1581
1582 /*
1583 * Attempt to use larger than system page-size
1584 * clusters for large writes only if there is
1585 * a jumbo cluster pool and if the socket is
1586 * marked accordingly.
1587 */
1588 jumbocl = sosendjcl && njcl > 0 &&
1589 ((so->so_flags & SOF_MULTIPAGES) ||
1590 sosendjcl_ignore_capab);
1591
91447636 1592 socket_unlock(so, 0);
2d21ac55 1593
91447636
A
1594 do {
1595 int num_needed;
1596 int hdrs_needed = (top == 0) ? 1 : 0;
2d21ac55 1597
91447636 1598 /*
2d21ac55
A
1599 * try to maintain a local cache of mbuf
1600 * clusters needed to complete this
1601 * write the list is further limited to
1602 * the number that are currently needed
1603 * to fill the socket this mechanism
1604 * allows a large number of mbufs/
1605 * clusters to be grabbed under a single
1606 * mbuf lock... if we can't get any
1607 * clusters, than fall back to trying
1608 * for mbufs if we fail early (or
1609 * miscalcluate the number needed) make
1610 * sure to release any clusters we
1611 * haven't yet consumed.
91447636 1612 */
2d21ac55
A
1613 if (freelist == NULL &&
1614 bytes_to_copy > NBPG && jumbocl) {
1615 num_needed =
1616 bytes_to_copy / M16KCLBYTES;
1617
1618 if ((bytes_to_copy -
1619 (num_needed * M16KCLBYTES))
1620 >= MINCLSIZE)
1621 num_needed++;
91447636 1622
2d21ac55
A
1623 freelist =
1624 m_getpackets_internal(
1625 (unsigned int *)&num_needed,
1626 hdrs_needed, M_WAIT, 0,
1627 M16KCLBYTES);
1628 /*
1629 * Fall back to 4K cluster size
1630 * if allocation failed
1631 */
1632 }
1633
1634 if (freelist == NULL &&
1635 bytes_to_copy > MCLBYTES) {
1636 num_needed =
1637 bytes_to_copy / NBPG;
1638
1639 if ((bytes_to_copy -
1640 (num_needed * NBPG)) >=
1641 MINCLSIZE)
91447636 1642 num_needed++;
2d21ac55
A
1643
1644 freelist =
1645 m_getpackets_internal(
1646 (unsigned int *)&num_needed,
1647 hdrs_needed, M_WAIT, 0,
1648 NBPG);
1649 /*
1650 * Fall back to cluster size
1651 * if allocation failed
1652 */
91447636 1653 }
2d21ac55
A
1654
1655 if (freelist == NULL &&
1656 bytes_to_copy > MINCLSIZE) {
1657 num_needed =
1658 bytes_to_copy / MCLBYTES;
1659
1660 if ((bytes_to_copy -
1661 (num_needed * MCLBYTES)) >=
1662 MINCLSIZE)
91447636 1663 num_needed++;
2d21ac55
A
1664
1665 freelist =
1666 m_getpackets_internal(
1667 (unsigned int *)&num_needed,
1668 hdrs_needed, M_WAIT, 0,
1669 MCLBYTES);
1670 /*
1671 * Fall back to a single mbuf
1672 * if allocation failed
1673 */
91447636 1674 }
2d21ac55 1675
91447636
A
1676 if (freelist == NULL) {
1677 if (top == 0)
2d21ac55
A
1678 MGETHDR(freelist,
1679 M_WAIT, MT_DATA);
91447636 1680 else
2d21ac55
A
1681 MGET(freelist,
1682 M_WAIT, MT_DATA);
91447636
A
1683
1684 if (freelist == NULL) {
1685 error = ENOBUFS;
1686 socket_lock(so, 0);
3a60a9f5 1687 goto release;
91447636
A
1688 }
1689 /*
2d21ac55
A
1690 * For datagram protocols,
1691 * leave room for protocol
1692 * headers in first mbuf.
91447636 1693 */
2d21ac55
A
1694 if (atomic && top == 0 &&
1695 bytes_to_copy < MHLEN) {
1696 MH_ALIGN(freelist,
1697 bytes_to_copy);
1698 }
91447636
A
1699 }
1700 m = freelist;
1701 freelist = m->m_next;
1702 m->m_next = NULL;
2d21ac55 1703
91447636
A
1704 if ((m->m_flags & M_EXT))
1705 mlen = m->m_ext.ext_size;
1706 else if ((m->m_flags & M_PKTHDR))
2d21ac55
A
1707 mlen =
1708 MHLEN - m_leadingspace(m);
91447636
A
1709 else
1710 mlen = MLEN;
b0d623f7 1711 len = imin(mlen, bytes_to_copy);
91447636
A
1712
1713 chainlength += len;
2d21ac55 1714
91447636 1715 space -= len;
fa4905b1 1716
2d21ac55 1717 error = uiomove(mtod(m, caddr_t),
b0d623f7 1718 len, uio);
2d21ac55 1719
91447636 1720 resid = uio_resid(uio);
2d21ac55 1721
91447636
A
1722 m->m_len = len;
1723 *mp = m;
1724 top->m_pkthdr.len += len;
2d21ac55 1725 if (error)
91447636
A
1726 break;
1727 mp = &m->m_next;
1728 if (resid <= 0) {
1729 if (flags & MSG_EOR)
1730 top->m_flags |= M_EOR;
1731 break;
1732 }
1733 bytes_to_copy = min(resid, space);
2d21ac55
A
1734
1735 } while (space > 0 &&
1736 (chainlength < sosendmaxchain || atomic ||
1737 resid < MINCLSIZE));
1738
91447636 1739 socket_lock(so, 0);
2d21ac55 1740
91447636
A
1741 if (error)
1742 goto release;
1743 }
2d21ac55
A
1744
1745 if (flags & (MSG_HOLD|MSG_SEND)) {
3a60a9f5
A
1746 /* Enqueue for later, go away if HOLD */
1747 register struct mbuf *mb1;
2d21ac55 1748 if (so->so_temp && (flags & MSG_FLUSH)) {
3a60a9f5
A
1749 m_freem(so->so_temp);
1750 so->so_temp = NULL;
1751 }
1752 if (so->so_temp)
1753 so->so_tail->m_next = top;
1754 else
1755 so->so_temp = top;
1756 mb1 = top;
1757 while (mb1->m_next)
2d21ac55 1758 mb1 = mb1->m_next;
3a60a9f5 1759 so->so_tail = mb1;
2d21ac55 1760 if (flags & MSG_HOLD) {
3a60a9f5
A
1761 top = NULL;
1762 goto release;
1763 }
1764 top = so->so_temp;
2d21ac55
A
1765 }
1766 if (dontroute)
1767 so->so_options |= SO_DONTROUTE;
1768
1769 /* Compute flags here, for pru_send and NKEs */
1770 sendflags = (flags & MSG_OOB) ? PRUS_OOB :
1771 /*
1772 * If the user set MSG_EOF, the protocol
1773 * understands this flag and nothing left to
1774 * send then use PRU_SEND_EOF instead of PRU_SEND.
1775 */
1776 ((flags & MSG_EOF) &&
1777 (so->so_proto->pr_flags & PR_IMPLOPCL) &&
1778 (resid <= 0)) ?
1c79356b 1779 PRUS_EOF :
2d21ac55
A
1780 /* If there is more to send set PRUS_MORETOCOME */
1781 (resid > 0 && space > 0) ? PRUS_MORETOCOME : 0;
1782
91447636
A
1783 /*
1784 * Socket filter processing
1785 */
2d21ac55
A
1786 recursive = (so->so_send_filt_thread != NULL);
1787 filtered = 0;
1788 error = 0;
1789 for (filter = so->so_filt; filter && (error == 0);
1790 filter = filter->sfe_next_onsocket) {
1791 if (filter->sfe_filter->sf_filter.sf_data_out) {
1792 int so_flags = 0;
1793 if (filtered == 0) {
1794 filtered = 1;
1795 so->so_send_filt_thread =
1796 current_thread();
1797 sflt_use(so);
1798 socket_unlock(so, 0);
1799 so_flags =
1800 (sendflags & MSG_OOB) ?
1801 sock_data_filt_flag_oob : 0;
91447636 1802 }
2d21ac55
A
1803 error = filter->sfe_filter->sf_filter.
1804 sf_data_out(filter->sfe_cookie, so,
1805 addr, &top, &control, so_flags);
91447636 1806 }
2d21ac55
A
1807 }
1808
1809 if (filtered) {
1810 /*
1811 * At this point, we've run at least one
1812 * filter. The socket is unlocked as is
1813 * the socket buffer. Clear the recorded
1814 * filter thread only when we are outside
1815 * of a filter's context. This allows for
1816 * a filter to issue multiple inject calls
1817 * from its sf_data_out callback routine.
1818 */
1819 socket_lock(so, 0);
1820 sflt_unuse(so);
1821 if (!recursive)
3a60a9f5 1822 so->so_send_filt_thread = 0;
2d21ac55
A
1823 if (error) {
1824 if (error == EJUSTRETURN) {
1825 error = 0;
1826 clen = 0;
1827 control = 0;
1828 top = 0;
1c79356b 1829 }
2d21ac55
A
1830
1831 goto release;
1c79356b
A
1832 }
1833 }
91447636
A
1834 /*
1835 * End Socket filter processing
1836 */
2d21ac55 1837
91447636
A
1838 if (error == EJUSTRETURN) {
1839 /* A socket filter handled this data */
1840 error = 0;
2d21ac55
A
1841 } else {
1842 error = (*so->so_proto->pr_usrreqs->pru_send)
1843 (so, sendflags, top, addr, control, p);
91447636 1844 }
9bccf70c 1845#ifdef __APPLE__
2d21ac55
A
1846 if (flags & MSG_SEND)
1847 so->so_temp = NULL;
9bccf70c 1848#endif
2d21ac55
A
1849 if (dontroute)
1850 so->so_options &= ~SO_DONTROUTE;
1851
1852 clen = 0;
1853 control = 0;
1854 top = 0;
1855 mp = &top;
1856 if (error)
1857 goto release;
1c79356b
A
1858 } while (resid && space > 0);
1859 } while (resid);
1860
1861release:
3a60a9f5
A
1862 if (sblocked)
1863 sbunlock(&so->so_snd, 0); /* will unlock socket */
1864 else
1865 socket_unlock(so, 1);
1c79356b
A
1866out:
1867 if (top)
1868 m_freem(top);
1869 if (control)
1870 m_freem(control);
fa4905b1 1871 if (freelist)
2d21ac55 1872 m_freem_list(freelist);
1c79356b 1873
2d21ac55
A
1874 KERNEL_DEBUG(DBG_FNC_SOSEND | DBG_FUNC_END, so, resid, so->so_snd.sb_cc,
1875 space, error);
1c79356b
A
1876
1877 return (error);
1878}
1879
1880/*
1881 * Implement receive operations on a socket.
1882 * We depend on the way that records are added to the sockbuf
1883 * by sbappend*. In particular, each record (mbufs linked through m_next)
1884 * must begin with an address if the protocol so specifies,
1885 * followed by an optional mbuf or mbufs containing ancillary data,
1886 * and then zero or more mbufs of data.
1887 * In order to avoid blocking network interrupts for the entire time here,
1888 * we splx() while doing the actual copy to user space.
1889 * Although the sockbuf is locked, new data may still be appended,
1890 * and thus we must maintain consistency of the sockbuf during that time.
1891 *
1892 * The caller may receive the data as a single mbuf chain by supplying
1893 * an mbuf **mp0 for use in returning the chain. The uio is then used
1894 * only for the count in uio_resid.
2d21ac55
A
1895 *
1896 * Returns: 0 Success
1897 * ENOBUFS
1898 * ENOTCONN
1899 * EWOULDBLOCK
1900 * uiomove:EFAULT
1901 * sblock:EWOULDBLOCK
1902 * sblock:EINTR
1903 * sbwait:EBADF
1904 * sbwait:EINTR
1905 * sodelayed_copy:EFAULT
1906 * <pru_rcvoob>:EINVAL[TCP]
1907 * <pru_rcvoob>:EWOULDBLOCK[TCP]
1908 * <pru_rcvoob>:???
1909 * <pr_domain->dom_externalize>:EMSGSIZE[AF_UNIX]
1910 * <pr_domain->dom_externalize>:ENOBUFS[AF_UNIX]
1911 * <pr_domain->dom_externalize>:???
1912 *
1913 * Notes: Additional return values from calls through <pru_rcvoob> and
1914 * <pr_domain->dom_externalize> depend on protocols other than
1915 * TCP or AF_UNIX, which are documented above.
1c79356b
A
1916 */
1917int
2d21ac55
A
1918soreceive(struct socket *so, struct sockaddr **psa, struct uio *uio,
1919 struct mbuf **mp0, struct mbuf **controlp, int *flagsp)
1c79356b 1920{
91447636
A
1921 register struct mbuf *m, **mp, *ml = NULL;
1922 register int flags, len, error, offset;
1c79356b
A
1923 struct protosw *pr = so->so_proto;
1924 struct mbuf *nextrecord;
1925 int moff, type = 0;
91447636 1926 int orig_resid = uio_resid(uio);
2d21ac55
A
1927 struct mbuf *free_list;
1928 int delayed_copy_len;
55e303ae
A
1929 int can_delay;
1930 int need_event;
1931 struct proc *p = current_proc();
1932
2d21ac55
A
1933 // LP64todo - fix this!
1934 KERNEL_DEBUG(DBG_FNC_SORECEIVE | DBG_FUNC_START, so, uio_resid(uio),
1935 so->so_rcv.sb_cc, so->so_rcv.sb_lowat, so->so_rcv.sb_hiwat);
1c79356b 1936
91447636 1937 socket_lock(so, 1);
1c79356b 1938
91447636
A
1939#ifdef MORE_LOCKING_DEBUG
1940 if (so->so_usecount == 1)
1941 panic("soreceive: so=%x no other reference on socket\n", so);
1942#endif
1c79356b
A
1943 mp = mp0;
1944 if (psa)
1945 *psa = 0;
1946 if (controlp)
1947 *controlp = 0;
1948 if (flagsp)
1949 flags = *flagsp &~ MSG_EOR;
1950 else
1951 flags = 0;
2d21ac55
A
1952
1953 /*
1954 * If a recv attempt is made on a previously-accepted socket
1955 * that has been marked as inactive (disconnected), reject
1956 * the request.
1957 */
1958 if (so->so_flags & SOF_DEFUNCT) {
1959 struct sockbuf *sb = &so->so_rcv;
1960
1961 /*
1962 * This socket should have been disconnected and flushed
1963 * prior to being returned from accept; there should be
1964 * no data on its receive list, so panic otherwise.
1965 */
1966 sb_empty_assert(sb, __func__);
1967 socket_unlock(so, 1);
1968 return (ENOTCONN);
1969 }
1970
1971 /*
1972 * When SO_WANTOOBFLAG is set we try to get out-of-band data
1973 * regardless of the flags argument. Here is the case were
1974 * out-of-band data is not inline.
1975 */
1976 if ((flags & MSG_OOB) ||
1977 ((so->so_options & SO_WANTOOBFLAG) != 0 &&
1978 (so->so_options & SO_OOBINLINE) == 0 &&
1979 (so->so_oobmark || (so->so_state & SS_RCVATMARK)))) {
1c79356b 1980 m = m_get(M_WAIT, MT_DATA);
55e303ae 1981 if (m == NULL) {
91447636 1982 socket_unlock(so, 1);
2d21ac55
A
1983 KERNEL_DEBUG(DBG_FNC_SORECEIVE | DBG_FUNC_END,
1984 ENOBUFS, 0, 0, 0, 0);
9bccf70c 1985 return (ENOBUFS);
55e303ae 1986 }
1c79356b
A
1987 error = (*pr->pr_usrreqs->pru_rcvoob)(so, m, flags & MSG_PEEK);
1988 if (error)
1989 goto bad;
91447636 1990 socket_unlock(so, 0);
1c79356b
A
1991 do {
1992 error = uiomove(mtod(m, caddr_t),
b0d623f7 1993 imin(uio_resid(uio), m->m_len), uio);
1c79356b 1994 m = m_free(m);
91447636
A
1995 } while (uio_resid(uio) && error == 0 && m);
1996 socket_lock(so, 0);
1c79356b
A
1997bad:
1998 if (m)
1999 m_freem(m);
9bccf70c
A
2000#ifdef __APPLE__
2001 if ((so->so_options & SO_WANTOOBFLAG) != 0) {
2002 if (error == EWOULDBLOCK || error == EINVAL) {
2d21ac55 2003 /*
9bccf70c 2004 * Let's try to get normal data:
2d21ac55
A
2005 * EWOULDBLOCK: out-of-band data not
2006 * receive yet. EINVAL: out-of-band data
2007 * already read.
9bccf70c
A
2008 */
2009 error = 0;
2010 goto nooob;
2d21ac55 2011 } else if (error == 0 && flagsp) {
9bccf70c 2012 *flagsp |= MSG_OOB;
2d21ac55
A
2013 }
2014 }
91447636 2015 socket_unlock(so, 1);
2d21ac55
A
2016 KERNEL_DEBUG(DBG_FNC_SORECEIVE | DBG_FUNC_END, error,
2017 0, 0, 0, 0);
9bccf70c 2018#endif
1c79356b
A
2019 return (error);
2020 }
2021nooob:
2022 if (mp)
2023 *mp = (struct mbuf *)0;
91447636 2024 if (so->so_state & SS_ISCONFIRMING && uio_resid(uio))
1c79356b
A
2025 (*pr->pr_usrreqs->pru_rcvd)(so, 0);
2026
55e303ae
A
2027
2028 free_list = (struct mbuf *)0;
2029 delayed_copy_len = 0;
1c79356b 2030restart:
91447636
A
2031#ifdef MORE_LOCKING_DEBUG
2032 if (so->so_usecount <= 1)
2d21ac55
A
2033 printf("soreceive: sblock so=%p ref=%d on socket\n",
2034 so, so->so_usecount);
91447636 2035#endif
6601e61a
A
2036 /*
2037 * See if the socket has been closed (SS_NOFDREF|SS_CANTRCVMORE)
2038 * and if so just return to the caller. This could happen when
2039 * soreceive() is called by a socket upcall function during the
2040 * time the socket is freed. The socket buffer would have been
2041 * locked across the upcall, therefore we cannot put this thread
2042 * to sleep (else we will deadlock) or return EWOULDBLOCK (else
2043 * we may livelock), because the lock on the socket buffer will
2044 * only be released when the upcall routine returns to its caller.
2045 * Because the socket has been officially closed, there can be
2046 * no further read on it.
2047 */
2048 if ((so->so_state & (SS_NOFDREF | SS_CANTRCVMORE)) ==
2049 (SS_NOFDREF | SS_CANTRCVMORE)) {
2050 socket_unlock(so, 1);
2051 return (0);
2052 }
2053
9bccf70c
A
2054 error = sblock(&so->so_rcv, SBLOCKWAIT(flags));
2055 if (error) {
91447636 2056 socket_unlock(so, 1);
2d21ac55
A
2057 KERNEL_DEBUG(DBG_FNC_SORECEIVE | DBG_FUNC_END, error,
2058 0, 0, 0, 0);
1c79356b
A
2059 return (error);
2060 }
1c79356b
A
2061
2062 m = so->so_rcv.sb_mb;
2063 /*
2064 * If we have less data than requested, block awaiting more
2065 * (subject to any timeout) if:
2066 * 1. the current count is less than the low water mark, or
2067 * 2. MSG_WAITALL is set, and it is possible to do the entire
2068 * receive operation at once if we block (resid <= hiwat).
2069 * 3. MSG_DONTWAIT is not set
2070 * If MSG_WAITALL is set but resid is larger than the receive buffer,
2071 * we have to do the receive in sections, and thus risk returning
2072 * a short count if a timeout or signal occurs after we start.
2073 */
2074 if (m == 0 || (((flags & MSG_DONTWAIT) == 0 &&
91447636 2075 so->so_rcv.sb_cc < uio_resid(uio)) &&
2d21ac55 2076 (so->so_rcv.sb_cc < so->so_rcv.sb_lowat ||
91447636 2077 ((flags & MSG_WAITALL) && uio_resid(uio) <= so->so_rcv.sb_hiwat)) &&
1c79356b 2078 m->m_nextpkt == 0 && (pr->pr_flags & PR_ATOMIC) == 0)) {
2d21ac55
A
2079 /*
2080 * Panic if we notice inconsistencies in the socket's
2081 * receive list; both sb_mb and sb_cc should correctly
2082 * reflect the contents of the list, otherwise we may
2083 * end up with false positives during select() or poll()
2084 * which could put the application in a bad state.
2085 */
2086 if (m == NULL && so->so_rcv.sb_cc != 0)
b0d623f7 2087 panic("soreceive corrupted so_rcv: m %p cc %u",
2d21ac55 2088 m, so->so_rcv.sb_cc);
55e303ae 2089
1c79356b
A
2090 if (so->so_error) {
2091 if (m)
2092 goto dontblock;
2093 error = so->so_error;
2094 if ((flags & MSG_PEEK) == 0)
2095 so->so_error = 0;
2096 goto release;
2097 }
2098 if (so->so_state & SS_CANTRCVMORE) {
2099 if (m)
2100 goto dontblock;
2101 else
2102 goto release;
2103 }
2104 for (; m; m = m->m_next)
2d21ac55 2105 if (m->m_type == MT_OOBDATA || (m->m_flags & M_EOR)) {
1c79356b
A
2106 m = so->so_rcv.sb_mb;
2107 goto dontblock;
2108 }
2109 if ((so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) == 0 &&
2110 (so->so_proto->pr_flags & PR_CONNREQUIRED)) {
2111 error = ENOTCONN;
2112 goto release;
2113 }
91447636 2114 if (uio_resid(uio) == 0)
1c79356b 2115 goto release;
2d21ac55
A
2116 if ((so->so_state & SS_NBIO) ||
2117 (flags & (MSG_DONTWAIT|MSG_NBIO))) {
1c79356b
A
2118 error = EWOULDBLOCK;
2119 goto release;
2120 }
2d21ac55
A
2121 SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 1");
2122 SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 1");
91447636 2123 sbunlock(&so->so_rcv, 1);
2d21ac55 2124#if EVEN_MORE_LOCKING_DEBUG
1c79356b 2125 if (socket_debug)
2d21ac55 2126 printf("Waiting for socket data\n");
91447636 2127#endif
55e303ae 2128
1c79356b 2129 error = sbwait(&so->so_rcv);
2d21ac55 2130#if EVEN_MORE_LOCKING_DEBUG
1c79356b 2131 if (socket_debug)
2d21ac55 2132 printf("SORECEIVE - sbwait returned %d\n", error);
91447636
A
2133#endif
2134 if (so->so_usecount < 1)
2d21ac55
A
2135 panic("soreceive: after 2nd sblock so=%p ref=%d on "
2136 "socket\n", so, so->so_usecount);
9bccf70c 2137 if (error) {
91447636 2138 socket_unlock(so, 1);
2d21ac55
A
2139 KERNEL_DEBUG(DBG_FNC_SORECEIVE | DBG_FUNC_END, error,
2140 0, 0, 0, 0);
2141 return (error);
1c79356b
A
2142 }
2143 goto restart;
2144 }
2145dontblock:
b0d623f7 2146 OSIncrementAtomicLong(&p->p_stats->p_ru.ru_msgrcv);
2d21ac55
A
2147 SBLASTRECORDCHK(&so->so_rcv, "soreceive 1");
2148 SBLASTMBUFCHK(&so->so_rcv, "soreceive 1");
1c79356b
A
2149 nextrecord = m->m_nextpkt;
2150 if ((pr->pr_flags & PR_ADDR) && m->m_type == MT_SONAME) {
2151 KASSERT(m->m_type == MT_SONAME, ("receive 1a"));
2d21ac55
A
2152#if CONFIG_MACF_SOCKET_SUBSET
2153 /*
2154 * Call the MAC framework for policy checking if we're in
2155 * the user process context and the socket isn't connected.
2156 */
2157 if (p != kernproc && !(so->so_state & SS_ISCONNECTED)) {
2158 struct mbuf *m0 = m;
2159 /*
2160 * Dequeue this record (temporarily) from the receive
2161 * list since we're about to drop the socket's lock
2162 * where a new record may arrive and be appended to
2163 * the list. Upon MAC policy failure, the record
2164 * will be freed. Otherwise, we'll add it back to
2165 * the head of the list. We cannot rely on SB_LOCK
2166 * because append operation uses the socket's lock.
2167 */
2168 do {
2169 m->m_nextpkt = NULL;
2170 sbfree(&so->so_rcv, m);
2171 m = m->m_next;
2172 } while (m != NULL);
2173 m = m0;
2174 so->so_rcv.sb_mb = nextrecord;
2175 SB_EMPTY_FIXUP(&so->so_rcv);
2176 SBLASTRECORDCHK(&so->so_rcv, "soreceive 1a");
2177 SBLASTMBUFCHK(&so->so_rcv, "soreceive 1a");
2178 socket_unlock(so, 0);
2179 if (mac_socket_check_received(proc_ucred(p), so,
2180 mtod(m, struct sockaddr *)) != 0) {
2181 /*
2182 * MAC policy failure; free this record and
2183 * process the next record (or block until
2184 * one is available). We have adjusted sb_cc
2185 * and sb_mbcnt above so there is no need to
2186 * call sbfree() again.
2187 */
2188 do {
2189 m = m_free(m);
2190 } while (m != NULL);
2191 /*
2192 * Clear SB_LOCK but don't unlock the socket.
2193 * Process the next record or wait for one.
2194 */
2195 socket_lock(so, 0);
2196 sbunlock(&so->so_rcv, 1);
2197 goto restart;
2198 }
2199 socket_lock(so, 0);
2200 /*
2201 * Re-adjust the socket receive list and re-enqueue
2202 * the record in front of any packets which may have
2203 * been appended while we dropped the lock.
2204 */
2205 for (m = m0; m->m_next != NULL; m = m->m_next)
2206 sballoc(&so->so_rcv, m);
2207 sballoc(&so->so_rcv, m);
2208 if (so->so_rcv.sb_mb == NULL) {
2209 so->so_rcv.sb_lastrecord = m0;
2210 so->so_rcv.sb_mbtail = m;
2211 }
2212 m = m0;
2213 nextrecord = m->m_nextpkt = so->so_rcv.sb_mb;
2214 so->so_rcv.sb_mb = m;
2215 SBLASTRECORDCHK(&so->so_rcv, "soreceive 1b");
2216 SBLASTMBUFCHK(&so->so_rcv, "soreceive 1b");
2217 }
2218#endif /* CONFIG_MACF_SOCKET_SUBSET */
1c79356b 2219 orig_resid = 0;
4a249263 2220 if (psa) {
1c79356b 2221 *psa = dup_sockaddr(mtod(m, struct sockaddr *),
2d21ac55 2222 mp0 == 0);
4a249263
A
2223 if ((*psa == 0) && (flags & MSG_NEEDSA)) {
2224 error = EWOULDBLOCK;
2225 goto release;
2226 }
2227 }
1c79356b
A
2228 if (flags & MSG_PEEK) {
2229 m = m->m_next;
2230 } else {
2231 sbfree(&so->so_rcv, m);
91447636 2232 if (m->m_next == 0 && so->so_rcv.sb_cc != 0)
2d21ac55
A
2233 panic("soreceive: about to create invalid "
2234 "socketbuf");
1c79356b
A
2235 MFREE(m, so->so_rcv.sb_mb);
2236 m = so->so_rcv.sb_mb;
2d21ac55
A
2237 if (m != NULL) {
2238 m->m_nextpkt = nextrecord;
2239 } else {
2240 so->so_rcv.sb_mb = nextrecord;
2241 SB_EMPTY_FIXUP(&so->so_rcv);
2242 }
1c79356b
A
2243 }
2244 }
2d21ac55
A
2245
2246 /*
2247 * Process one or more MT_CONTROL mbufs present before any data mbufs
2248 * in the first mbuf chain on the socket buffer. If MSG_PEEK, we
2249 * just copy the data; if !MSG_PEEK, we call into the protocol to
2250 * perform externalization.
2251 */
2252 if (m != NULL && m->m_type == MT_CONTROL) {
2253 struct mbuf *cm = NULL, *cmn;
2254 struct mbuf **cme = &cm;
2255 struct sockbuf *sb_rcv = &so->so_rcv;
2256
2257 /*
2258 * Externalizing the control messages would require us to
2259 * drop the socket's lock below. Once we re-acquire the
2260 * lock, the mbuf chain might change. In order to preserve
2261 * consistency, we unlink all control messages from the
2262 * first mbuf chain in one shot and link them separately
2263 * onto a different chain.
2264 */
2265 do {
2266 if (flags & MSG_PEEK) {
2267 if (controlp != NULL) {
2268 *controlp = m_copy(m, 0, m->m_len);
2269 controlp = &(*controlp)->m_next;
91447636 2270 }
2d21ac55 2271 m = m->m_next;
1c79356b 2272 } else {
2d21ac55
A
2273 m->m_nextpkt = NULL;
2274 sbfree(sb_rcv, m);
2275 sb_rcv->sb_mb = m->m_next;
2276 m->m_next = NULL;
2277 *cme = m;
2278 cme = &(*cme)->m_next;
2279 m = sb_rcv->sb_mb;
2280 }
2281 } while (m != NULL && m->m_type == MT_CONTROL);
2282
2283 if (!(flags & MSG_PEEK)) {
2284 if (sb_rcv->sb_mb != NULL) {
2285 sb_rcv->sb_mb->m_nextpkt = nextrecord;
2286 } else {
2287 sb_rcv->sb_mb = nextrecord;
2288 SB_EMPTY_FIXUP(sb_rcv);
1c79356b 2289 }
2d21ac55
A
2290 if (nextrecord == NULL)
2291 sb_rcv->sb_lastrecord = m;
1c79356b 2292 }
2d21ac55
A
2293
2294 SBLASTRECORDCHK(&so->so_rcv, "soreceive ctl");
2295 SBLASTMBUFCHK(&so->so_rcv, "soreceive ctl");
2296
2297 while (cm != NULL) {
2298 int cmsg_type;
2299
2300 cmn = cm->m_next;
2301 cm->m_next = NULL;
2302 cmsg_type = mtod(cm, struct cmsghdr *)->cmsg_type;
2303
2304 /*
2305 * Call the protocol to externalize SCM_RIGHTS message
2306 * and return the modified message to the caller upon
2307 * success. Otherwise, all other control messages are
2308 * returned unmodified to the caller. Note that we
2309 * only get into this loop if MSG_PEEK is not set.
2310 */
2311 if (pr->pr_domain->dom_externalize != NULL &&
2312 cmsg_type == SCM_RIGHTS) {
2313 /*
2314 * Release socket lock: see 3903171. This
2315 * would also allow more records to be appended
2316 * to the socket buffer. We still have SB_LOCK
2317 * set on it, so we can be sure that the head
2318 * of the mbuf chain won't change.
2319 */
2320 socket_unlock(so, 0);
2321 error = (*pr->pr_domain->dom_externalize)(cm);
2322 socket_lock(so, 0);
2323 } else {
2324 error = 0;
2325 }
2326
2327 if (controlp != NULL && error == 0) {
2328 *controlp = cm;
2329 controlp = &(*controlp)->m_next;
2330 orig_resid = 0;
2331 } else {
2332 (void) m_free(cm);
2333 }
2334 cm = cmn;
1c79356b 2335 }
2d21ac55
A
2336 orig_resid = 0;
2337 if (sb_rcv->sb_mb != NULL)
2338 nextrecord = sb_rcv->sb_mb->m_nextpkt;
2339 else
2340 nextrecord = NULL;
1c79356b 2341 }
2d21ac55
A
2342
2343 if (m != NULL) {
2344 if (!(flags & MSG_PEEK)) {
2345 /*
2346 * We get here because m points to an mbuf following
2347 * any MT_SONAME or MT_CONTROL mbufs which have been
2348 * processed above. In any case, m should be pointing
2349 * to the head of the mbuf chain, and the nextrecord
2350 * should be either NULL or equal to m->m_nextpkt.
2351 * See comments above about SB_LOCK.
2352 */
2353 if (m != so->so_rcv.sb_mb || m->m_nextpkt != nextrecord)
2354 panic("soreceive: post-control !sync so=%p "
2355 "m=%p nextrecord=%p\n", so, m, nextrecord);
2356
2357 if (nextrecord == NULL)
2358 so->so_rcv.sb_lastrecord = m;
2359 }
1c79356b
A
2360 type = m->m_type;
2361 if (type == MT_OOBDATA)
2362 flags |= MSG_OOB;
2d21ac55
A
2363 } else {
2364 if (!(flags & MSG_PEEK)) {
2365 so->so_rcv.sb_mb = nextrecord;
2366 SB_EMPTY_FIXUP(&so->so_rcv);
2367 }
1c79356b 2368 }
2d21ac55
A
2369 SBLASTRECORDCHK(&so->so_rcv, "soreceive 2");
2370 SBLASTMBUFCHK(&so->so_rcv, "soreceive 2");
2371
1c79356b
A
2372 moff = 0;
2373 offset = 0;
fa4905b1 2374
91447636 2375 if (!(flags & MSG_PEEK) && uio_resid(uio) > sorecvmincopy)
2d21ac55 2376 can_delay = 1;
55e303ae 2377 else
2d21ac55 2378 can_delay = 0;
55e303ae
A
2379
2380 need_event = 0;
fa4905b1 2381
91447636 2382 while (m && (uio_resid(uio) - delayed_copy_len) > 0 && error == 0) {
1c79356b
A
2383 if (m->m_type == MT_OOBDATA) {
2384 if (type != MT_OOBDATA)
2385 break;
2d21ac55 2386 } else if (type == MT_OOBDATA) {
1c79356b 2387 break;
2d21ac55 2388 }
9bccf70c 2389 /*
2d21ac55 2390 * Make sure to allways set MSG_OOB event when getting
9bccf70c
A
2391 * out of band data inline.
2392 */
1c79356b 2393 if ((so->so_options & SO_WANTOOBFLAG) != 0 &&
2d21ac55
A
2394 (so->so_options & SO_OOBINLINE) != 0 &&
2395 (so->so_state & SS_RCVATMARK) != 0) {
9bccf70c
A
2396 flags |= MSG_OOB;
2397 }
1c79356b 2398 so->so_state &= ~SS_RCVATMARK;
91447636 2399 len = uio_resid(uio) - delayed_copy_len;
1c79356b
A
2400 if (so->so_oobmark && len > so->so_oobmark - offset)
2401 len = so->so_oobmark - offset;
2402 if (len > m->m_len - moff)
2403 len = m->m_len - moff;
2404 /*
2405 * If mp is set, just pass back the mbufs.
2406 * Otherwise copy them out via the uio, then free.
2407 * Sockbuf must be consistent here (points to current mbuf,
2408 * it points to next record) when we drop priority;
2409 * we must note any additions to the sockbuf when we
2410 * block interrupts again.
2411 */
2412 if (mp == 0) {
2d21ac55
A
2413 SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove");
2414 SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove");
55e303ae 2415 if (can_delay && len == m->m_len) {
2d21ac55 2416 /*
55e303ae
A
2417 * only delay the copy if we're consuming the
2418 * mbuf and we're NOT in MSG_PEEK mode
2419 * and we have enough data to make it worthwile
2d21ac55
A
2420 * to drop and retake the lock... can_delay
2421 * reflects the state of the 2 latter
2422 * constraints moff should always be zero
2423 * in these cases
55e303ae 2424 */
2d21ac55 2425 delayed_copy_len += len;
55e303ae 2426 } else {
2d21ac55
A
2427 if (delayed_copy_len) {
2428 error = sodelayed_copy(so, uio,
2429 &free_list, &delayed_copy_len);
55e303ae
A
2430
2431 if (error) {
55e303ae
A
2432 goto release;
2433 }
2d21ac55
A
2434 /*
2435 * can only get here if MSG_PEEK is not
2436 * set therefore, m should point at the
2437 * head of the rcv queue; if it doesn't,
2438 * it means something drastically
2439 * changed while we were out from behind
2440 * the lock in sodelayed_copy. perhaps
2441 * a RST on the stream. in any event,
2442 * the stream has been interrupted. it's
2443 * probably best just to return whatever
2444 * data we've moved and let the caller
2445 * sort it out...
2446 */
55e303ae 2447 if (m != so->so_rcv.sb_mb) {
2d21ac55 2448 break;
55e303ae
A
2449 }
2450 }
91447636 2451 socket_unlock(so, 0);
2d21ac55
A
2452 error = uiomove(mtod(m, caddr_t) + moff,
2453 (int)len, uio);
91447636 2454 socket_lock(so, 0);
55e303ae 2455
55e303ae 2456 if (error)
2d21ac55 2457 goto release;
55e303ae 2458 }
2d21ac55 2459 } else {
91447636 2460 uio_setresid(uio, (uio_resid(uio) - len));
2d21ac55 2461 }
1c79356b
A
2462 if (len == m->m_len - moff) {
2463 if (m->m_flags & M_EOR)
2464 flags |= MSG_EOR;
2465 if (flags & MSG_PEEK) {
2466 m = m->m_next;
2467 moff = 0;
2468 } else {
2469 nextrecord = m->m_nextpkt;
2470 sbfree(&so->so_rcv, m);
91447636 2471 m->m_nextpkt = NULL;
55e303ae 2472
1c79356b
A
2473 if (mp) {
2474 *mp = m;
2475 mp = &m->m_next;
2476 so->so_rcv.sb_mb = m = m->m_next;
2477 *mp = (struct mbuf *)0;
2478 } else {
55e303ae 2479 if (free_list == NULL)
2d21ac55
A
2480 free_list = m;
2481 else
2482 ml->m_next = m;
2483 ml = m;
14353aa8 2484 so->so_rcv.sb_mb = m = m->m_next;
2d21ac55 2485 ml->m_next = 0;
1c79356b 2486 }
2d21ac55 2487 if (m != NULL) {
1c79356b 2488 m->m_nextpkt = nextrecord;
2d21ac55
A
2489 if (nextrecord == NULL)
2490 so->so_rcv.sb_lastrecord = m;
2491 } else {
2492 so->so_rcv.sb_mb = nextrecord;
2493 SB_EMPTY_FIXUP(&so->so_rcv);
2494 }
2495 SBLASTRECORDCHK(&so->so_rcv, "soreceive 3");
2496 SBLASTMBUFCHK(&so->so_rcv, "soreceive 3");
1c79356b
A
2497 }
2498 } else {
2d21ac55 2499 if (flags & MSG_PEEK) {
1c79356b 2500 moff += len;
2d21ac55 2501 } else {
1c79356b
A
2502 if (mp)
2503 *mp = m_copym(m, 0, len, M_WAIT);
2504 m->m_data += len;
2505 m->m_len -= len;
2506 so->so_rcv.sb_cc -= len;
2507 }
2508 }
2509 if (so->so_oobmark) {
2510 if ((flags & MSG_PEEK) == 0) {
2511 so->so_oobmark -= len;
2512 if (so->so_oobmark == 0) {
2d21ac55
A
2513 so->so_state |= SS_RCVATMARK;
2514 /*
2515 * delay posting the actual event until
2516 * after any delayed copy processing
2517 * has finished
2518 */
2519 need_event = 1;
2520 break;
1c79356b
A
2521 }
2522 } else {
2523 offset += len;
2524 if (offset == so->so_oobmark)
2525 break;
2526 }
2527 }
2d21ac55 2528 if (flags & MSG_EOR)
1c79356b
A
2529 break;
2530 /*
2d21ac55
A
2531 * If the MSG_WAITALL or MSG_WAITSTREAM flag is set
2532 * (for non-atomic socket), we must not quit until
2533 * "uio->uio_resid == 0" or an error termination.
2534 * If a signal/timeout occurs, return with a short
2535 * count but without error. Keep sockbuf locked
2536 * against other readers.
1c79356b 2537 */
2d21ac55
A
2538 while (flags & (MSG_WAITALL|MSG_WAITSTREAM) && m == 0 &&
2539 (uio_resid(uio) - delayed_copy_len) > 0 &&
1c79356b
A
2540 !sosendallatonce(so) && !nextrecord) {
2541 if (so->so_error || so->so_state & SS_CANTRCVMORE)
2d21ac55 2542 goto release;
fa4905b1 2543
2d21ac55
A
2544 /*
2545 * Depending on the protocol (e.g. TCP), the following
2546 * might cause the socket lock to be dropped and later
2547 * be reacquired, and more data could have arrived and
2548 * have been appended to the receive socket buffer by
2549 * the time it returns. Therefore, we only sleep in
2550 * sbwait() below if and only if the socket buffer is
2551 * empty, in order to avoid a false sleep.
2552 */
2553 if (pr->pr_flags & PR_WANTRCVD && so->so_pcb &&
2554 (((struct inpcb *)so->so_pcb)->inp_state !=
2555 INPCB_STATE_DEAD))
2556 (*pr->pr_usrreqs->pru_rcvd)(so, flags);
2557
2558 SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2");
2559 SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2");
2560
2561 if (so->so_rcv.sb_mb == NULL && sbwait(&so->so_rcv)) {
2562 error = 0;
55e303ae 2563 goto release;
fa4905b1 2564 }
55e303ae 2565 /*
2d21ac55
A
2566 * have to wait until after we get back from the sbwait
2567 * to do the copy because we will drop the lock if we
2568 * have enough data that has been delayed... by dropping
2569 * the lock we open up a window allowing the netisr
2570 * thread to process the incoming packets and to change
2571 * the state of this socket... we're issuing the sbwait
2572 * because the socket is empty and we're expecting the
2573 * netisr thread to wake us up when more packets arrive;
2574 * if we allow that processing to happen and then sbwait
2575 * we could stall forever with packets sitting in the
2576 * socket if no further packets arrive from the remote
2577 * side.
55e303ae 2578 *
2d21ac55
A
2579 * we want to copy before we've collected all the data
2580 * to satisfy this request to allow the copy to overlap
2581 * the incoming packet processing on an MP system
55e303ae 2582 */
2d21ac55
A
2583 if (delayed_copy_len > sorecvmincopy &&
2584 (delayed_copy_len > (so->so_rcv.sb_hiwat / 2))) {
2585 error = sodelayed_copy(so, uio,
2586 &free_list, &delayed_copy_len);
55e303ae
A
2587
2588 if (error)
2d21ac55 2589 goto release;
1c79356b
A
2590 }
2591 m = so->so_rcv.sb_mb;
fa4905b1 2592 if (m) {
1c79356b 2593 nextrecord = m->m_nextpkt;
fa4905b1 2594 }
1c79356b
A
2595 }
2596 }
91447636
A
2597#ifdef MORE_LOCKING_DEBUG
2598 if (so->so_usecount <= 1)
2d21ac55
A
2599 panic("soreceive: after big while so=%p ref=%d on socket\n",
2600 so, so->so_usecount);
91447636 2601#endif
1c79356b
A
2602
2603 if (m && pr->pr_flags & PR_ATOMIC) {
9bccf70c 2604#ifdef __APPLE__
2d21ac55 2605 if (so->so_options & SO_DONTTRUNC) {
1c79356b 2606 flags |= MSG_RCVMORE;
2d21ac55 2607 } else {
9bccf70c
A
2608#endif
2609 flags |= MSG_TRUNC;
1c79356b
A
2610 if ((flags & MSG_PEEK) == 0)
2611 (void) sbdroprecord(&so->so_rcv);
9bccf70c 2612#ifdef __APPLE__
1c79356b 2613 }
9bccf70c 2614#endif
1c79356b 2615 }
2d21ac55
A
2616
2617 /*
2618 * pru_rcvd below (for TCP) may cause more data to be received
2619 * if the socket lock is dropped prior to sending the ACK; some
2620 * legacy OpenTransport applications don't handle this well
2621 * (if it receives less data than requested while MSG_HAVEMORE
2622 * is set), and so we set the flag now based on what we know
2623 * prior to calling pru_rcvd.
2624 */
2625 if ((so->so_options & SO_WANTMORE) && so->so_rcv.sb_cc > 0)
2626 flags |= MSG_HAVEMORE;
2627
1c79356b 2628 if ((flags & MSG_PEEK) == 0) {
2d21ac55 2629 if (m == 0) {
1c79356b 2630 so->so_rcv.sb_mb = nextrecord;
2d21ac55
A
2631 /*
2632 * First part is an inline SB_EMPTY_FIXUP(). Second
2633 * part makes sure sb_lastrecord is up-to-date if
2634 * there is still data in the socket buffer.
2635 */
2636 if (so->so_rcv.sb_mb == NULL) {
2637 so->so_rcv.sb_mbtail = NULL;
2638 so->so_rcv.sb_lastrecord = NULL;
2639 } else if (nextrecord->m_nextpkt == NULL) {
2640 so->so_rcv.sb_lastrecord = nextrecord;
2641 }
2642 }
2643 SBLASTRECORDCHK(&so->so_rcv, "soreceive 4");
2644 SBLASTMBUFCHK(&so->so_rcv, "soreceive 4");
1c79356b
A
2645 if (pr->pr_flags & PR_WANTRCVD && so->so_pcb)
2646 (*pr->pr_usrreqs->pru_rcvd)(so, flags);
2647 }
9bccf70c 2648#ifdef __APPLE__
55e303ae 2649 if (delayed_copy_len) {
91447636 2650 error = sodelayed_copy(so, uio, &free_list, &delayed_copy_len);
55e303ae
A
2651
2652 if (error)
2d21ac55 2653 goto release;
55e303ae
A
2654 }
2655 if (free_list) {
2d21ac55 2656 m_freem_list((struct mbuf *)free_list);
55e303ae
A
2657 free_list = (struct mbuf *)0;
2658 }
2659 if (need_event)
2d21ac55 2660 postevent(so, 0, EV_OOB);
9bccf70c 2661#endif
91447636 2662 if (orig_resid == uio_resid(uio) && orig_resid &&
1c79356b 2663 (flags & MSG_EOR) == 0 && (so->so_state & SS_CANTRCVMORE) == 0) {
91447636 2664 sbunlock(&so->so_rcv, 1);
1c79356b
A
2665 goto restart;
2666 }
2667
2668 if (flagsp)
2669 *flagsp |= flags;
2670release:
91447636
A
2671#ifdef MORE_LOCKING_DEBUG
2672 if (so->so_usecount <= 1)
2d21ac55
A
2673 panic("soreceive: release so=%p ref=%d on socket\n",
2674 so, so->so_usecount);
91447636 2675#endif
55e303ae 2676 if (delayed_copy_len) {
2d21ac55 2677 error = sodelayed_copy(so, uio, &free_list, &delayed_copy_len);
55e303ae
A
2678 }
2679 if (free_list) {
2d21ac55 2680 m_freem_list((struct mbuf *)free_list);
55e303ae 2681 }
91447636 2682 sbunlock(&so->so_rcv, 0); /* will unlock socket */
1c79356b 2683
2d21ac55
A
2684 // LP64todo - fix this!
2685 KERNEL_DEBUG(DBG_FNC_SORECEIVE | DBG_FUNC_END, so, uio_resid(uio),
2686 so->so_rcv.sb_cc, 0, error);
1c79356b
A
2687
2688 return (error);
2689}
2690
2d21ac55
A
2691/*
2692 * Returns: 0 Success
2693 * uiomove:EFAULT
2694 */
2695static int
2696sodelayed_copy(struct socket *so, struct uio *uio, struct mbuf **free_list,
2697 int *resid)
55e303ae 2698{
2d21ac55 2699 int error = 0;
55e303ae
A
2700 struct mbuf *m;
2701
2702 m = *free_list;
2703
91447636 2704 socket_unlock(so, 0);
55e303ae 2705
2d21ac55
A
2706 while (m && error == 0) {
2707
2708 error = uiomove(mtod(m, caddr_t), (int)m->m_len, uio);
2709
2710 m = m->m_next;
2711 }
2712 m_freem_list(*free_list);
2713
2714 *free_list = (struct mbuf *)NULL;
2715 *resid = 0;
2716
2717 socket_lock(so, 0);
55e303ae 2718
2d21ac55
A
2719 return (error);
2720}
2721
2722
2723/*
2724 * Returns: 0 Success
2725 * EINVAL
2726 * ENOTCONN
2727 * <pru_shutdown>:EINVAL
2728 * <pru_shutdown>:EADDRNOTAVAIL[TCP]
2729 * <pru_shutdown>:ENOBUFS[TCP]
2730 * <pru_shutdown>:EMSGSIZE[TCP]
2731 * <pru_shutdown>:EHOSTUNREACH[TCP]
2732 * <pru_shutdown>:ENETUNREACH[TCP]
2733 * <pru_shutdown>:ENETDOWN[TCP]
2734 * <pru_shutdown>:ENOMEM[TCP]
2735 * <pru_shutdown>:EACCES[TCP]
2736 * <pru_shutdown>:EMSGSIZE[TCP]
2737 * <pru_shutdown>:ENOBUFS[TCP]
2738 * <pru_shutdown>:???[TCP] [ignorable: mostly IPSEC/firewall/DLIL]
2739 * <pru_shutdown>:??? [other protocol families]
2740 */
2741int
2742soshutdown(struct socket *so, int how)
2743{
2744 int error;
55e303ae 2745
2d21ac55
A
2746 switch (how) {
2747 case SHUT_RD:
2748 case SHUT_WR:
2749 case SHUT_RDWR:
2750 socket_lock(so, 1);
2751 if ((so->so_state &
2752 (SS_ISCONNECTED|SS_ISCONNECTING|SS_ISDISCONNECTING)) == 0) {
2753 error = ENOTCONN;
2754 } else {
2755 error = soshutdownlock(so, how);
2756 }
2757 socket_unlock(so, 1);
2758 break;
2759 default:
2760 error = EINVAL;
2761 break;
55e303ae 2762 }
55e303ae
A
2763
2764 return (error);
2765}
2766
1c79356b 2767int
2d21ac55 2768soshutdownlock(struct socket *so, int how)
1c79356b 2769{
2d21ac55
A
2770 struct protosw *pr = so->so_proto;
2771 int error = 0;
1c79356b 2772
91447636 2773 sflt_notify(so, sock_evt_shutdown, &how);
1c79356b 2774
9bccf70c 2775 if (how != SHUT_WR) {
2d21ac55
A
2776 if ((so->so_state & SS_CANTRCVMORE) != 0) {
2777 /* read already shut down */
2778 error = ENOTCONN;
2779 goto done;
2780 }
1c79356b
A
2781 sorflush(so);
2782 postevent(so, 0, EV_RCLOSED);
2783 }
9bccf70c 2784 if (how != SHUT_RD) {
2d21ac55
A
2785 if ((so->so_state & SS_CANTSENDMORE) != 0) {
2786 /* write already shut down */
2787 error = ENOTCONN;
2788 goto done;
2789 }
2790 error = (*pr->pr_usrreqs->pru_shutdown)(so);
2791 postevent(so, 0, EV_WCLOSED);
1c79356b 2792 }
2d21ac55
A
2793done:
2794 KERNEL_DEBUG(DBG_FNC_SOSHUTDOWN | DBG_FUNC_END, 0, 0, 0, 0, 0);
2795 return (error);
1c79356b
A
2796}
2797
2798void
2d21ac55 2799sorflush(struct socket *so)
1c79356b
A
2800{
2801 register struct sockbuf *sb = &so->so_rcv;
2802 register struct protosw *pr = so->so_proto;
1c79356b 2803 struct sockbuf asb;
1c79356b 2804
91447636 2805#ifdef MORE_LOCKING_DEBUG
2d21ac55 2806 lck_mtx_t *mutex_held;
91447636 2807
2d21ac55 2808 if (so->so_proto->pr_getlock != NULL)
91447636 2809 mutex_held = (*so->so_proto->pr_getlock)(so, 0);
2d21ac55 2810 else
91447636
A
2811 mutex_held = so->so_proto->pr_domain->dom_mtx;
2812 lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
2813#endif
2814
2815 sflt_notify(so, sock_evt_flush_read, NULL);
1c79356b
A
2816
2817 sb->sb_flags |= SB_NOINTR;
2818 (void) sblock(sb, M_WAIT);
1c79356b 2819 socantrcvmore(so);
91447636 2820 sbunlock(sb, 1);
9bccf70c 2821#ifdef __APPLE__
0b4e3aa0 2822 selthreadclear(&sb->sb_sel);
9bccf70c 2823#endif
1c79356b
A
2824 asb = *sb;
2825 bzero((caddr_t)sb, sizeof (*sb));
91447636 2826 sb->sb_so = so; /* reestablish link to socket */
9bccf70c
A
2827 if (asb.sb_flags & SB_KNOTE) {
2828 sb->sb_sel.si_note = asb.sb_sel.si_note;
2829 sb->sb_flags = SB_KNOTE;
2830 }
2d21ac55
A
2831 if (asb.sb_flags & SB_DROP)
2832 sb->sb_flags |= SB_DROP;
2833 if (asb.sb_flags & SB_UNIX)
2834 sb->sb_flags |= SB_UNIX;
2835 if ((pr->pr_flags & PR_RIGHTS) && pr->pr_domain->dom_dispose) {
1c79356b 2836 (*pr->pr_domain->dom_dispose)(asb.sb_mb);
2d21ac55 2837 }
1c79356b
A
2838 sbrelease(&asb);
2839}
2840
2841/*
2842 * Perhaps this routine, and sooptcopyout(), below, ought to come in
2843 * an additional variant to handle the case where the option value needs
2844 * to be some kind of integer, but not a specific size.
2845 * In addition to their use here, these functions are also called by the
2846 * protocol-level pr_ctloutput() routines.
2d21ac55
A
2847 *
2848 * Returns: 0 Success
2849 * EINVAL
2850 * copyin:EFAULT
1c79356b
A
2851 */
2852int
2d21ac55 2853sooptcopyin(struct sockopt *sopt, void *buf, size_t len, size_t minlen)
1c79356b
A
2854{
2855 size_t valsize;
2856
2857 /*
2858 * If the user gives us more than we wanted, we ignore it,
2859 * but if we don't get the minimum length the caller
2860 * wants, we return EINVAL. On success, sopt->sopt_valsize
2861 * is set to however much we actually retrieved.
2862 */
2863 if ((valsize = sopt->sopt_valsize) < minlen)
2d21ac55 2864 return (EINVAL);
1c79356b
A
2865 if (valsize > len)
2866 sopt->sopt_valsize = valsize = len;
2867
b0d623f7 2868 if (sopt->sopt_p != kernproc)
1c79356b
A
2869 return (copyin(sopt->sopt_val, buf, valsize));
2870
91447636 2871 bcopy(CAST_DOWN(caddr_t, sopt->sopt_val), buf, valsize);
2d21ac55
A
2872 return (0);
2873}
2874
2875/*
2876 * sooptcopyin_timeval
2877 * Copy in a timeval value into tv_p, and take into account whether the
2878 * the calling process is 64-bit or 32-bit. Moved the sanity checking
2879 * code here so that we can verify the 64-bit tv_sec value before we lose
2880 * the top 32-bits assigning tv64.tv_sec to tv_p->tv_sec.
2881 */
2882static int
2883sooptcopyin_timeval(struct sockopt *sopt, struct timeval * tv_p)
2884{
2885 int error;
b0d623f7 2886
2d21ac55 2887 if (proc_is64bit(sopt->sopt_p)) {
b0d623f7 2888 struct user64_timeval tv64;
2d21ac55
A
2889
2890 if (sopt->sopt_valsize < sizeof(tv64)) {
2891 return (EINVAL);
2892 }
2893 sopt->sopt_valsize = sizeof(tv64);
b0d623f7
A
2894 if (sopt->sopt_p != kernproc) {
2895 error = copyin(sopt->sopt_val, &tv64, sizeof(tv64));
2896 if (error != 0)
2897 return (error);
2898 } else {
2899 bcopy(CAST_DOWN(caddr_t, sopt->sopt_val), &tv64,
2900 sizeof(tv64));
2d21ac55
A
2901 }
2902 if (tv64.tv_sec < 0 || tv64.tv_sec > LONG_MAX
2903 || tv64.tv_usec < 0 || tv64.tv_usec >= 1000000) {
2904 return (EDOM);
2905 }
2906 tv_p->tv_sec = tv64.tv_sec;
2907 tv_p->tv_usec = tv64.tv_usec;
2908 } else {
b0d623f7
A
2909 struct user32_timeval tv32;
2910
2911 if (sopt->sopt_valsize < sizeof(tv32)) {
2d21ac55
A
2912 return (EINVAL);
2913 }
b0d623f7
A
2914 sopt->sopt_valsize = sizeof(tv32);
2915 if (sopt->sopt_p != kernproc) {
2916 error = copyin(sopt->sopt_val, &tv32, sizeof(tv32));
2d21ac55
A
2917 if (error != 0) {
2918 return (error);
2919 }
2920 } else {
b0d623f7
A
2921 bcopy(CAST_DOWN(caddr_t, sopt->sopt_val), &tv32,
2922 sizeof(tv32));
2d21ac55 2923 }
b0d623f7
A
2924#ifndef __LP64__ // K64todo "comparison is always false due to limited range of data type"
2925 if (tv32.tv_sec < 0 || tv32.tv_sec > LONG_MAX
2926 || tv32.tv_usec < 0 || tv32.tv_usec >= 1000000) {
2d21ac55
A
2927 return (EDOM);
2928 }
b0d623f7
A
2929#endif
2930 tv_p->tv_sec = tv32.tv_sec;
2931 tv_p->tv_usec = tv32.tv_usec;
2d21ac55
A
2932 }
2933 return (0);
1c79356b
A
2934}
2935
2d21ac55
A
2936/*
2937 * Returns: 0 Success
2938 * EINVAL
2939 * ENOPROTOOPT
2940 * ENOBUFS
2941 * EDOM
2942 * sooptcopyin:EINVAL
2943 * sooptcopyin:EFAULT
2944 * sooptcopyin_timeval:EINVAL
2945 * sooptcopyin_timeval:EFAULT
2946 * sooptcopyin_timeval:EDOM
2947 * <pr_ctloutput>:EOPNOTSUPP[AF_UNIX]
2948 * <pr_ctloutput>:???w
2949 * sflt_attach_private:??? [whatever a filter author chooses]
2950 * <sf_setoption>:??? [whatever a filter author chooses]
2951 *
2952 * Notes: Other <pru_listen> returns depend on the protocol family; all
2953 * <sf_listen> returns depend on what the filter author causes
2954 * their filter to return.
2955 */
1c79356b 2956int
2d21ac55 2957sosetopt(struct socket *so, struct sockopt *sopt)
1c79356b
A
2958{
2959 int error, optval;
2960 struct linger l;
2961 struct timeval tv;
2d21ac55
A
2962 struct socket_filter_entry *filter;
2963 int filtered = 0;
2964#if CONFIG_MACF_SOCKET
2965 struct mac extmac;
2966#endif /* MAC_SOCKET */
91447636
A
2967
2968 socket_lock(so, 1);
2d21ac55 2969 if ((so->so_state & (SS_CANTRCVMORE | SS_CANTSENDMORE))
b0d623f7
A
2970 == (SS_CANTRCVMORE | SS_CANTSENDMORE) &&
2971 (so->so_flags & SOF_NPX_SETOPTSHUT) == 0) {
2d21ac55
A
2972 /* the socket has been shutdown, no more sockopt's */
2973 error = EINVAL;
2974 goto bad;
2975 }
1c79356b 2976
9bccf70c
A
2977 if (sopt->sopt_dir != SOPT_SET) {
2978 sopt->sopt_dir = SOPT_SET;
2979 }
2980
2d21ac55
A
2981 error = 0;
2982 for (filter = so->so_filt; filter && (error == 0);
2983 filter = filter->sfe_next_onsocket) {
2984 if (filter->sfe_filter->sf_filter.sf_setoption) {
2985 if (filtered == 0) {
2986 filtered = 1;
2987 sflt_use(so);
2988 socket_unlock(so, 0);
91447636 2989 }
2d21ac55
A
2990 error = filter->sfe_filter->sf_filter.
2991 sf_setoption(filter->sfe_cookie, so, sopt);
91447636 2992 }
2d21ac55
A
2993 }
2994
2995 if (filtered != 0) {
2996 socket_lock(so, 0);
2997 sflt_unuse(so);
2998
2999 if (error) {
3000 if (error == EJUSTRETURN)
3001 error = 0;
3002 goto bad;
1c79356b 3003 }
1c79356b
A
3004 }
3005
3006 error = 0;
3007 if (sopt->sopt_level != SOL_SOCKET) {
91447636 3008 if (so->so_proto && so->so_proto->pr_ctloutput) {
2d21ac55 3009 error = (*so->so_proto->pr_ctloutput)(so, sopt);
91447636
A
3010 socket_unlock(so, 1);
3011 return (error);
3012 }
1c79356b
A
3013 error = ENOPROTOOPT;
3014 } else {
3015 switch (sopt->sopt_name) {
3016 case SO_LINGER:
91447636 3017 case SO_LINGER_SEC:
2d21ac55 3018 error = sooptcopyin(sopt, &l, sizeof (l), sizeof (l));
1c79356b
A
3019 if (error)
3020 goto bad;
3021
2d21ac55
A
3022 so->so_linger = (sopt->sopt_name == SO_LINGER) ?
3023 l.l_linger : l.l_linger * hz;
1c79356b
A
3024 if (l.l_onoff)
3025 so->so_options |= SO_LINGER;
3026 else
3027 so->so_options &= ~SO_LINGER;
3028 break;
3029
3030 case SO_DEBUG:
3031 case SO_KEEPALIVE:
3032 case SO_DONTROUTE:
3033 case SO_USELOOPBACK:
3034 case SO_BROADCAST:
3035 case SO_REUSEADDR:
3036 case SO_REUSEPORT:
3037 case SO_OOBINLINE:
3038 case SO_TIMESTAMP:
9bccf70c 3039#ifdef __APPLE__
1c79356b
A
3040 case SO_DONTTRUNC:
3041 case SO_WANTMORE:
9bccf70c
A
3042 case SO_WANTOOBFLAG:
3043#endif
2d21ac55
A
3044 error = sooptcopyin(sopt, &optval, sizeof (optval),
3045 sizeof (optval));
1c79356b
A
3046 if (error)
3047 goto bad;
3048 if (optval)
3049 so->so_options |= sopt->sopt_name;
3050 else
3051 so->so_options &= ~sopt->sopt_name;
3052 break;
3053
3054 case SO_SNDBUF:
3055 case SO_RCVBUF:
3056 case SO_SNDLOWAT:
3057 case SO_RCVLOWAT:
2d21ac55
A
3058 error = sooptcopyin(sopt, &optval, sizeof (optval),
3059 sizeof (optval));
1c79356b
A
3060 if (error)
3061 goto bad;
3062
3063 /*
3064 * Values < 1 make no sense for any of these
3065 * options, so disallow them.
3066 */
3067 if (optval < 1) {
3068 error = EINVAL;
3069 goto bad;
3070 }
3071
3072 switch (sopt->sopt_name) {
3073 case SO_SNDBUF:
3074 case SO_RCVBUF:
3075 if (sbreserve(sopt->sopt_name == SO_SNDBUF ?
2d21ac55 3076 &so->so_snd : &so->so_rcv,
b0d623f7 3077 (u_int32_t) optval) == 0) {
1c79356b
A
3078 error = ENOBUFS;
3079 goto bad;
3080 }
2d21ac55
A
3081 if (sopt->sopt_name == SO_SNDBUF)
3082 so->so_snd.sb_flags |= SB_USRSIZE;
3083 else
3084 so->so_rcv.sb_flags |= SB_USRSIZE;
1c79356b
A
3085 break;
3086
3087 /*
3088 * Make sure the low-water is never greater than
3089 * the high-water.
3090 */
3091 case SO_SNDLOWAT:
3092 so->so_snd.sb_lowat =
3093 (optval > so->so_snd.sb_hiwat) ?
3094 so->so_snd.sb_hiwat : optval;
3095 break;
3096 case SO_RCVLOWAT:
3097 so->so_rcv.sb_lowat =
3098 (optval > so->so_rcv.sb_hiwat) ?
3099 so->so_rcv.sb_hiwat : optval;
3100 break;
3101 }
3102 break;
3103
3104 case SO_SNDTIMEO:
3105 case SO_RCVTIMEO:
2d21ac55 3106 error = sooptcopyin_timeval(sopt, &tv);
1c79356b
A
3107 if (error)
3108 goto bad;
3109
1c79356b
A
3110 switch (sopt->sopt_name) {
3111 case SO_SNDTIMEO:
91447636 3112 so->so_snd.sb_timeo = tv;
1c79356b
A
3113 break;
3114 case SO_RCVTIMEO:
91447636 3115 so->so_rcv.sb_timeo = tv;
1c79356b
A
3116 break;
3117 }
3118 break;
3119
3120 case SO_NKE:
9bccf70c
A
3121 {
3122 struct so_nke nke;
1c79356b 3123
2d21ac55
A
3124 error = sooptcopyin(sopt, &nke, sizeof (nke),
3125 sizeof (nke));
1c79356b 3126 if (error)
2d21ac55 3127 goto bad;
1c79356b 3128
2d21ac55
A
3129 error = sflt_attach_private(so, NULL,
3130 nke.nke_handle, 1);
1c79356b
A
3131 break;
3132 }
3133
9bccf70c 3134 case SO_NOSIGPIPE:
2d21ac55
A
3135 error = sooptcopyin(sopt, &optval, sizeof (optval),
3136 sizeof (optval));
3137 if (error)
3138 goto bad;
3139 if (optval)
3140 so->so_flags |= SOF_NOSIGPIPE;
3141 else
3142 so->so_flags &= ~SOF_NOSIGPIPE;
3143
9bccf70c
A
3144 break;
3145
55e303ae 3146 case SO_NOADDRERR:
2d21ac55
A
3147 error = sooptcopyin(sopt, &optval, sizeof (optval),
3148 sizeof (optval));
3149 if (error)
3150 goto bad;
3151 if (optval)
3152 so->so_flags |= SOF_NOADDRAVAIL;
3153 else
3154 so->so_flags &= ~SOF_NOADDRAVAIL;
3155
3156 break;
3157
3158 case SO_REUSESHAREUID:
3159 error = sooptcopyin(sopt, &optval, sizeof (optval),
3160 sizeof (optval));
3161 if (error)
3162 goto bad;
3163 if (optval)
3164 so->so_flags |= SOF_REUSESHAREUID;
3165 else
3166 so->so_flags &= ~SOF_REUSESHAREUID;
3167 break;
3168#ifdef __APPLE_API_PRIVATE
3169 case SO_NOTIFYCONFLICT:
3170 if (kauth_cred_issuser(kauth_cred_get()) == 0) {
3171 error = EPERM;
3172 goto bad;
3173 }
3174 error = sooptcopyin(sopt, &optval, sizeof (optval),
3175 sizeof (optval));
3176 if (error)
3177 goto bad;
3178 if (optval)
3179 so->so_flags |= SOF_NOTIFYCONFLICT;
3180 else
3181 so->so_flags &= ~SOF_NOTIFYCONFLICT;
3182 break;
3183#endif
3184 case SO_RESTRICTIONS:
3185 if (kauth_cred_issuser(kauth_cred_get()) == 0) {
3186 error = EPERM;
3187 goto bad;
3188 }
3189 error = sooptcopyin(sopt, &optval, sizeof (optval),
3190 sizeof (optval));
3191 if (error)
3192 goto bad;
3193 so->so_restrictions = (optval & (SO_RESTRICT_DENYIN |
3194 SO_RESTRICT_DENYOUT | SO_RESTRICT_DENYSET));
3195 break;
3196
3197 case SO_LABEL:
3198#if CONFIG_MACF_SOCKET
3199 if ((error = sooptcopyin(sopt, &extmac, sizeof (extmac),
3200 sizeof (extmac))) != 0)
3201 goto bad;
3202
3203 error = mac_setsockopt_label(proc_ucred(sopt->sopt_p),
3204 so, &extmac);
3205#else
3206 error = EOPNOTSUPP;
3207#endif /* MAC_SOCKET */
55e303ae
A
3208 break;
3209
4a3eedf9
A
3210#ifdef __APPLE_API_PRIVATE
3211 case SO_UPCALLCLOSEWAIT:
3212 error = sooptcopyin(sopt, &optval, sizeof (optval),
3213 sizeof (optval));
3214 if (error)
3215 goto bad;
3216 if (optval)
3217 so->so_flags |= SOF_UPCALLCLOSEWAIT;
3218 else
3219 so->so_flags &= ~SOF_UPCALLCLOSEWAIT;
3220 break;
3221#endif
3222
b0d623f7
A
3223 case SO_RANDOMPORT:
3224 error = sooptcopyin(sopt, &optval, sizeof (optval),
3225 sizeof (optval));
3226 if (error)
3227 goto bad;
3228 if (optval)
3229 so->so_flags |= SOF_BINDRANDOMPORT;
3230 else
3231 so->so_flags &= ~SOF_BINDRANDOMPORT;
3232 break;
3233
3234 case SO_NP_EXTENSIONS: {
3235 struct so_np_extensions sonpx;
3236
3237 error = sooptcopyin(sopt, &sonpx, sizeof(sonpx), sizeof(sonpx));
3238 if (error)
3239 goto bad;
3240 if (sonpx.npx_mask & ~SONPX_MASK_VALID) {
3241 error = EINVAL;
3242 goto bad;
3243 }
3244 /*
3245 * Only one bit defined for now
3246 */
3247 if ((sonpx.npx_mask & SONPX_SETOPTSHUT)) {
3248 if ((sonpx.npx_flags & SONPX_SETOPTSHUT))
3249 so->so_flags |= SOF_NPX_SETOPTSHUT;
3250 else
3251 so->so_flags &= ~SOF_NPX_SETOPTSHUT;
3252 }
3253 break;
3254 }
3255
d41d1dae
A
3256#if PKT_PRIORITY
3257 case SO_TRAFFIC_CLASS: {
3258 error = sooptcopyin(sopt, &optval, sizeof (optval),
3259 sizeof (optval));
3260 if (error)
3261 goto bad;
3262 if (optval < SO_TC_BE || optval > SO_TC_VO) {
3263 error = EINVAL;
3264 goto bad;
3265 }
3266 so->so_traffic_class = optval;
3267 }
3268#endif /* PKT_PRIORITY */
3269
1c79356b
A
3270 default:
3271 error = ENOPROTOOPT;
3272 break;
3273 }
3274 if (error == 0 && so->so_proto && so->so_proto->pr_ctloutput) {
2d21ac55 3275 (void) ((*so->so_proto->pr_ctloutput)(so, sopt));
1c79356b
A
3276 }
3277 }
3278bad:
91447636 3279 socket_unlock(so, 1);
1c79356b
A
3280 return (error);
3281}
3282
2d21ac55 3283/* Helper routines for getsockopt */
1c79356b 3284int
2d21ac55 3285sooptcopyout(struct sockopt *sopt, void *buf, size_t len)
1c79356b
A
3286{
3287 int error;
3288 size_t valsize;
3289
3290 error = 0;
3291
3292 /*
3293 * Documented get behavior is that we always return a value,
3294 * possibly truncated to fit in the user's buffer.
3295 * Traditional behavior is that we always tell the user
3296 * precisely how much we copied, rather than something useful
3297 * like the total amount we had available for her.
3298 * Note that this interface is not idempotent; the entire answer must
3299 * generated ahead of time.
3300 */
3301 valsize = min(len, sopt->sopt_valsize);
3302 sopt->sopt_valsize = valsize;
91447636 3303 if (sopt->sopt_val != USER_ADDR_NULL) {
b0d623f7 3304 if (sopt->sopt_p != kernproc)
1c79356b
A
3305 error = copyout(buf, sopt->sopt_val, valsize);
3306 else
91447636 3307 bcopy(buf, CAST_DOWN(caddr_t, sopt->sopt_val), valsize);
1c79356b 3308 }
2d21ac55
A
3309 return (error);
3310}
3311
3312static int
3313sooptcopyout_timeval(struct sockopt *sopt, const struct timeval * tv_p)
3314{
3315 int error;
3316 size_t len;
b0d623f7
A
3317 struct user64_timeval tv64;
3318 struct user32_timeval tv32;
2d21ac55
A
3319 const void * val;
3320 size_t valsize;
b0d623f7 3321
2d21ac55
A
3322 error = 0;
3323 if (proc_is64bit(sopt->sopt_p)) {
b0d623f7 3324 len = sizeof(tv64);
2d21ac55
A
3325 tv64.tv_sec = tv_p->tv_sec;
3326 tv64.tv_usec = tv_p->tv_usec;
3327 val = &tv64;
3328 } else {
b0d623f7
A
3329 len = sizeof(tv32);
3330 tv32.tv_sec = tv_p->tv_sec;
3331 tv32.tv_usec = tv_p->tv_usec;
3332 val = &tv32;
2d21ac55
A
3333 }
3334 valsize = min(len, sopt->sopt_valsize);
3335 sopt->sopt_valsize = valsize;
3336 if (sopt->sopt_val != USER_ADDR_NULL) {
b0d623f7 3337 if (sopt->sopt_p != kernproc)
2d21ac55
A
3338 error = copyout(val, sopt->sopt_val, valsize);
3339 else
3340 bcopy(val, CAST_DOWN(caddr_t, sopt->sopt_val), valsize);
3341 }
3342 return (error);
1c79356b
A
3343}
3344
2d21ac55
A
3345/*
3346 * Return: 0 Success
3347 * ENOPROTOOPT
3348 * <pr_ctloutput>:EOPNOTSUPP[AF_UNIX]
3349 * <pr_ctloutput>:???
3350 * <sf_getoption>:???
3351 */
1c79356b 3352int
2d21ac55 3353sogetopt(struct socket *so, struct sockopt *sopt)
1c79356b
A
3354{
3355 int error, optval;
3356 struct linger l;
3357 struct timeval tv;
2d21ac55
A
3358 struct socket_filter_entry *filter;
3359 int filtered = 0;
3360#if CONFIG_MACF_SOCKET
3361 struct mac extmac;
3362#endif /* MAC_SOCKET */
1c79356b 3363
2d21ac55
A
3364 if (sopt->sopt_dir != SOPT_GET) {
3365 sopt->sopt_dir = SOPT_GET;
3366 }
9bccf70c 3367
91447636 3368 socket_lock(so, 1);
2d21ac55
A
3369
3370 error = 0;
3371 for (filter = so->so_filt; filter && (error == 0);
3372 filter = filter->sfe_next_onsocket) {
3373 if (filter->sfe_filter->sf_filter.sf_getoption) {
3374 if (filtered == 0) {
3375 filtered = 1;
3376 sflt_use(so);
3377 socket_unlock(so, 0);
91447636 3378 }
2d21ac55
A
3379 error = filter->sfe_filter->sf_filter.
3380 sf_getoption(filter->sfe_cookie, so, sopt);
91447636 3381 }
2d21ac55
A
3382 }
3383 if (filtered != 0) {
3384 socket_lock(so, 0);
3385 sflt_unuse(so);
3386
3387 if (error) {
3388 if (error == EJUSTRETURN)
3389 error = 0;
3390 socket_unlock(so, 1);
3391 return (error);
1c79356b 3392 }
1c79356b
A
3393 }
3394
3395 error = 0;
3396 if (sopt->sopt_level != SOL_SOCKET) {
3397 if (so->so_proto && so->so_proto->pr_ctloutput) {
2d21ac55 3398 error = (*so->so_proto->pr_ctloutput)(so, sopt);
91447636
A
3399 socket_unlock(so, 1);
3400 return (error);
3401 } else {
3402 socket_unlock(so, 1);
1c79356b 3403 return (ENOPROTOOPT);
91447636 3404 }
1c79356b
A
3405 } else {
3406 switch (sopt->sopt_name) {
3407 case SO_LINGER:
91447636 3408 case SO_LINGER_SEC:
1c79356b 3409 l.l_onoff = so->so_options & SO_LINGER;
2d21ac55
A
3410 l.l_linger = (sopt->sopt_name == SO_LINGER) ?
3411 so->so_linger : so->so_linger / hz;
3412 error = sooptcopyout(sopt, &l, sizeof (l));
1c79356b
A
3413 break;
3414
3415 case SO_USELOOPBACK:
3416 case SO_DONTROUTE:
3417 case SO_DEBUG:
3418 case SO_KEEPALIVE:
3419 case SO_REUSEADDR:
3420 case SO_REUSEPORT:
3421 case SO_BROADCAST:
3422 case SO_OOBINLINE:
3423 case SO_TIMESTAMP:
9bccf70c 3424#ifdef __APPLE__
1c79356b
A
3425 case SO_DONTTRUNC:
3426 case SO_WANTMORE:
9bccf70c
A
3427 case SO_WANTOOBFLAG:
3428#endif
1c79356b
A
3429 optval = so->so_options & sopt->sopt_name;
3430integer:
2d21ac55 3431 error = sooptcopyout(sopt, &optval, sizeof (optval));
1c79356b
A
3432 break;
3433
3434 case SO_TYPE:
3435 optval = so->so_type;
3436 goto integer;
3437
9bccf70c 3438#ifdef __APPLE__
1c79356b 3439 case SO_NREAD:
2d21ac55
A
3440 if (so->so_proto->pr_flags & PR_ATOMIC) {
3441 int pkt_total;
3442 struct mbuf *m1;
1c79356b 3443
2d21ac55
A
3444 pkt_total = 0;
3445 m1 = so->so_rcv.sb_mb;
9bccf70c 3446 while (m1) {
2d21ac55
A
3447 if (m1->m_type == MT_DATA || m1->m_type == MT_HEADER ||
3448 m1->m_type == MT_OOBDATA)
1c79356b 3449 pkt_total += m1->m_len;
1c79356b
A
3450 m1 = m1->m_next;
3451 }
3452 optval = pkt_total;
2d21ac55
A
3453 } else {
3454 optval = so->so_rcv.sb_cc - so->so_rcv.sb_ctl;
3455 }
1c79356b 3456 goto integer;
2d21ac55 3457
91447636
A
3458 case SO_NWRITE:
3459 optval = so->so_snd.sb_cc;
2d21ac55 3460 goto integer;
9bccf70c 3461#endif
1c79356b
A
3462 case SO_ERROR:
3463 optval = so->so_error;
3464 so->so_error = 0;
3465 goto integer;
3466
3467 case SO_SNDBUF:
3468 optval = so->so_snd.sb_hiwat;
3469 goto integer;
3470
3471 case SO_RCVBUF:
3472 optval = so->so_rcv.sb_hiwat;
3473 goto integer;
3474
3475 case SO_SNDLOWAT:
3476 optval = so->so_snd.sb_lowat;
3477 goto integer;
3478
3479 case SO_RCVLOWAT:
3480 optval = so->so_rcv.sb_lowat;
3481 goto integer;
3482
3483 case SO_SNDTIMEO:
3484 case SO_RCVTIMEO:
91447636 3485 tv = (sopt->sopt_name == SO_SNDTIMEO ?
2d21ac55 3486 so->so_snd.sb_timeo : so->so_rcv.sb_timeo);
1c79356b 3487
2d21ac55
A
3488 error = sooptcopyout_timeval(sopt, &tv);
3489 break;
1c79356b 3490
91447636
A
3491 case SO_NOSIGPIPE:
3492 optval = (so->so_flags & SOF_NOSIGPIPE);
3493 goto integer;
9bccf70c 3494
55e303ae 3495 case SO_NOADDRERR:
91447636
A
3496 optval = (so->so_flags & SOF_NOADDRAVAIL);
3497 goto integer;
55e303ae 3498
2d21ac55
A
3499 case SO_REUSESHAREUID:
3500 optval = (so->so_flags & SOF_REUSESHAREUID);
3501 goto integer;
3502
3503#ifdef __APPLE_API_PRIVATE
3504 case SO_NOTIFYCONFLICT:
3505 optval = (so->so_flags & SOF_NOTIFYCONFLICT);
3506 goto integer;
3507#endif
3508 case SO_RESTRICTIONS:
3509 optval = so->so_restrictions & (SO_RESTRICT_DENYIN |
3510 SO_RESTRICT_DENYOUT | SO_RESTRICT_DENYSET);
3511 goto integer;
3512
3513 case SO_LABEL:
3514#if CONFIG_MACF_SOCKET
3515 if ((error = sooptcopyin(sopt, &extmac, sizeof (extmac),
3516 sizeof (extmac))) != 0 ||
3517 (error = mac_socket_label_get(proc_ucred(
3518 sopt->sopt_p), so, &extmac)) != 0)
3519 break;
3520
3521 error = sooptcopyout(sopt, &extmac, sizeof (extmac));
3522#else
3523 error = EOPNOTSUPP;
3524#endif /* MAC_SOCKET */
3525 break;
3526
3527 case SO_PEERLABEL:
3528#if CONFIG_MACF_SOCKET
3529 if ((error = sooptcopyin(sopt, &extmac, sizeof (extmac),
3530 sizeof (extmac))) != 0 ||
3531 (error = mac_socketpeer_label_get(proc_ucred(
3532 sopt->sopt_p), so, &extmac)) != 0)
3533 break;
3534
3535 error = sooptcopyout(sopt, &extmac, sizeof (extmac));
3536#else
3537 error = EOPNOTSUPP;
3538#endif /* MAC_SOCKET */
3539 break;
3540
4a3eedf9
A
3541#ifdef __APPLE_API_PRIVATE
3542 case SO_UPCALLCLOSEWAIT:
3543 optval = (so->so_flags & SOF_UPCALLCLOSEWAIT);
3544 goto integer;
3545#endif
b0d623f7
A
3546 case SO_RANDOMPORT:
3547 optval = (so->so_flags & SOF_BINDRANDOMPORT);
3548 goto integer;
3549
3550 case SO_NP_EXTENSIONS: {
3551 struct so_np_extensions sonpx;
3552
3553 sonpx.npx_flags = (so->so_flags & SOF_NPX_SETOPTSHUT) ? SONPX_SETOPTSHUT : 0;
3554 sonpx.npx_mask = SONPX_MASK_VALID;
4a3eedf9 3555
b0d623f7
A
3556 error = sooptcopyout(sopt, &sonpx, sizeof(struct so_np_extensions));
3557 break;
3558 }
d41d1dae
A
3559#if PKT_PRIORITY
3560 case SO_TRAFFIC_CLASS:
3561 optval = so->so_traffic_class;
3562 goto integer;
3563#endif /* PKT_PRIORITY */
3564
1c79356b
A
3565 default:
3566 error = ENOPROTOOPT;
3567 break;
3568 }
91447636 3569 socket_unlock(so, 1);
1c79356b
A
3570 return (error);
3571 }
3572}
3573
9bccf70c 3574/* XXX; prepare mbuf for (__FreeBSD__ < 3) routines. */
1c79356b 3575int
9bccf70c 3576soopt_getm(struct sockopt *sopt, struct mbuf **mp)
1c79356b
A
3577{
3578 struct mbuf *m, *m_prev;
3579 int sopt_size = sopt->sopt_valsize;
b0d623f7 3580 int how;
1c79356b 3581
a3d08fcd 3582 if (sopt_size > MAX_SOOPTGETM_SIZE)
2d21ac55 3583 return (EMSGSIZE);
a3d08fcd 3584
b0d623f7
A
3585 how = sopt->sopt_p != kernproc ? M_WAIT : M_DONTWAIT;
3586 MGET(m, how, MT_DATA);
1c79356b 3587 if (m == 0)
2d21ac55 3588 return (ENOBUFS);
1c79356b 3589 if (sopt_size > MLEN) {
b0d623f7 3590 MCLGET(m, how);
1c79356b
A
3591 if ((m->m_flags & M_EXT) == 0) {
3592 m_free(m);
2d21ac55 3593 return (ENOBUFS);
1c79356b
A
3594 }
3595 m->m_len = min(MCLBYTES, sopt_size);
3596 } else {
3597 m->m_len = min(MLEN, sopt_size);
3598 }
3599 sopt_size -= m->m_len;
3600 *mp = m;
3601 m_prev = m;
3602
3603 while (sopt_size) {
b0d623f7 3604 MGET(m, how, MT_DATA);
1c79356b
A
3605 if (m == 0) {
3606 m_freem(*mp);
2d21ac55 3607 return (ENOBUFS);
1c79356b
A
3608 }
3609 if (sopt_size > MLEN) {
b0d623f7 3610 MCLGET(m, how);
1c79356b
A
3611 if ((m->m_flags & M_EXT) == 0) {
3612 m_freem(*mp);
2d21ac55 3613 return (ENOBUFS);
1c79356b
A
3614 }
3615 m->m_len = min(MCLBYTES, sopt_size);
3616 } else {
3617 m->m_len = min(MLEN, sopt_size);
3618 }
3619 sopt_size -= m->m_len;
3620 m_prev->m_next = m;
3621 m_prev = m;
3622 }
2d21ac55 3623 return (0);
1c79356b
A
3624}
3625
3626/* XXX; copyin sopt data into mbuf chain for (__FreeBSD__ < 3) routines. */
3627int
9bccf70c 3628soopt_mcopyin(struct sockopt *sopt, struct mbuf *m)
1c79356b
A
3629{
3630 struct mbuf *m0 = m;
3631
91447636 3632 if (sopt->sopt_val == USER_ADDR_NULL)
2d21ac55 3633 return (0);
1c79356b 3634 while (m != NULL && sopt->sopt_valsize >= m->m_len) {
b0d623f7 3635 if (sopt->sopt_p != kernproc) {
1c79356b
A
3636 int error;
3637
2d21ac55
A
3638 error = copyin(sopt->sopt_val, mtod(m, char *),
3639 m->m_len);
1c79356b
A
3640 if (error != 0) {
3641 m_freem(m0);
2d21ac55 3642 return (error);
1c79356b 3643 }
2d21ac55
A
3644 } else {
3645 bcopy(CAST_DOWN(caddr_t, sopt->sopt_val),
3646 mtod(m, char *), m->m_len);
3647 }
1c79356b 3648 sopt->sopt_valsize -= m->m_len;
2d21ac55 3649 sopt->sopt_val += m->m_len;
1c79356b
A
3650 m = m->m_next;
3651 }
3652 if (m != NULL) /* should be allocated enoughly at ip6_sooptmcopyin() */
9bccf70c 3653 panic("soopt_mcopyin");
2d21ac55 3654 return (0);
1c79356b
A
3655}
3656
3657/* XXX; copyout mbuf chain data into soopt for (__FreeBSD__ < 3) routines. */
3658int
9bccf70c 3659soopt_mcopyout(struct sockopt *sopt, struct mbuf *m)
1c79356b
A
3660{
3661 struct mbuf *m0 = m;
3662 size_t valsize = 0;
3663
91447636 3664 if (sopt->sopt_val == USER_ADDR_NULL)
2d21ac55 3665 return (0);
1c79356b 3666 while (m != NULL && sopt->sopt_valsize >= m->m_len) {
b0d623f7 3667 if (sopt->sopt_p != kernproc) {
1c79356b
A
3668 int error;
3669
2d21ac55
A
3670 error = copyout(mtod(m, char *), sopt->sopt_val,
3671 m->m_len);
1c79356b
A
3672 if (error != 0) {
3673 m_freem(m0);
2d21ac55 3674 return (error);
1c79356b 3675 }
2d21ac55
A
3676 } else {
3677 bcopy(mtod(m, char *),
3678 CAST_DOWN(caddr_t, sopt->sopt_val), m->m_len);
3679 }
3680 sopt->sopt_valsize -= m->m_len;
3681 sopt->sopt_val += m->m_len;
3682 valsize += m->m_len;
3683 m = m->m_next;
1c79356b
A
3684 }
3685 if (m != NULL) {
3686 /* enough soopt buffer should be given from user-land */
3687 m_freem(m0);
2d21ac55 3688 return (EINVAL);
1c79356b
A
3689 }
3690 sopt->sopt_valsize = valsize;
2d21ac55 3691 return (0);
1c79356b
A
3692}
3693
9bccf70c 3694void
2d21ac55 3695sohasoutofband(struct socket *so)
9bccf70c 3696{
9bccf70c 3697
9bccf70c
A
3698 if (so->so_pgid < 0)
3699 gsignal(-so->so_pgid, SIGURG);
2d21ac55
A
3700 else if (so->so_pgid > 0)
3701 proc_signal(so->so_pgid, SIGURG);
9bccf70c
A
3702 selwakeup(&so->so_rcv.sb_sel);
3703}
3704
3705int
91447636 3706sopoll(struct socket *so, int events, __unused kauth_cred_t cred, void * wql)
9bccf70c
A
3707{
3708 struct proc *p = current_proc();
3709 int revents = 0;
91447636
A
3710
3711 socket_lock(so, 1);
9bccf70c
A
3712
3713 if (events & (POLLIN | POLLRDNORM))
3714 if (soreadable(so))
3715 revents |= events & (POLLIN | POLLRDNORM);
3716
3717 if (events & (POLLOUT | POLLWRNORM))
3718 if (sowriteable(so))
3719 revents |= events & (POLLOUT | POLLWRNORM);
3720
3721 if (events & (POLLPRI | POLLRDBAND))
3722 if (so->so_oobmark || (so->so_state & SS_RCVATMARK))
3723 revents |= events & (POLLPRI | POLLRDBAND);
3724
3725 if (revents == 0) {
3726 if (events & (POLLIN | POLLPRI | POLLRDNORM | POLLRDBAND)) {
2d21ac55
A
3727 /*
3728 * Darwin sets the flag first,
3729 * BSD calls selrecord first
3730 */
9bccf70c
A
3731 so->so_rcv.sb_flags |= SB_SEL;
3732 selrecord(p, &so->so_rcv.sb_sel, wql);
3733 }
3734
3735 if (events & (POLLOUT | POLLWRNORM)) {
2d21ac55
A
3736 /*
3737 * Darwin sets the flag first,
3738 * BSD calls selrecord first
3739 */
9bccf70c
A
3740 so->so_snd.sb_flags |= SB_SEL;
3741 selrecord(p, &so->so_snd.sb_sel, wql);
3742 }
3743 }
3744
91447636 3745 socket_unlock(so, 1);
9bccf70c
A
3746 return (revents);
3747}
55e303ae 3748
55e303ae 3749int
2d21ac55
A
3750soo_kqfilter(__unused struct fileproc *fp, struct knote *kn,
3751 __unused struct proc *p)
55e303ae 3752{
91447636 3753 struct socket *so = (struct socket *)kn->kn_fp->f_fglob->fg_data;
55e303ae 3754 struct sockbuf *sb;
2d21ac55 3755
91447636 3756 socket_lock(so, 1);
55e303ae 3757
2d21ac55
A
3758#if CONFIG_MACF_SOCKET
3759 if (mac_socket_check_kqfilter(proc_ucred(p), kn, so) != 0) {
3760 socket_unlock(so, 1);
3761 return (1);
3762 }
3763#endif /* MAC_SOCKET */
3764
55e303ae
A
3765 switch (kn->kn_filter) {
3766 case EVFILT_READ:
b0d623f7 3767 kn->kn_fop = &soread_filtops;
55e303ae
A
3768 sb = &so->so_rcv;
3769 break;
3770 case EVFILT_WRITE:
3771 kn->kn_fop = &sowrite_filtops;
3772 sb = &so->so_snd;
3773 break;
3774 default:
91447636 3775 socket_unlock(so, 1);
55e303ae
A
3776 return (1);
3777 }
3778
55e303ae
A
3779 if (KNOTE_ATTACH(&sb->sb_sel.si_note, kn))
3780 sb->sb_flags |= SB_KNOTE;
91447636 3781 socket_unlock(so, 1);
55e303ae
A
3782 return (0);
3783}
3784
3785static void
3786filt_sordetach(struct knote *kn)
3787{
91447636 3788 struct socket *so = (struct socket *)kn->kn_fp->f_fglob->fg_data;
55e303ae 3789
91447636
A
3790 socket_lock(so, 1);
3791 if (so->so_rcv.sb_flags & SB_KNOTE)
55e303ae
A
3792 if (KNOTE_DETACH(&so->so_rcv.sb_sel.si_note, kn))
3793 so->so_rcv.sb_flags &= ~SB_KNOTE;
91447636 3794 socket_unlock(so, 1);
55e303ae
A
3795}
3796
3797/*ARGSUSED*/
3798static int
3799filt_soread(struct knote *kn, long hint)
3800{
91447636 3801 struct socket *so = (struct socket *)kn->kn_fp->f_fglob->fg_data;
55e303ae 3802
91447636
A
3803 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3804 socket_lock(so, 1);
3805
b0d623f7
A
3806 if (so->so_options & SO_ACCEPTCONN) {
3807 int isempty;
3808
3809 /* Radar 6615193 handle the listen case dynamically
3810 * for kqueue read filter. This allows to call listen() after registering
3811 * the kqueue EVFILT_READ.
3812 */
3813
3814 kn->kn_data = so->so_qlen;
3815 isempty = ! TAILQ_EMPTY(&so->so_comp);
3816
3817 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3818 socket_unlock(so, 1);
3819
3820 return (isempty);
3821 }
3822
3823 /* socket isn't a listener */
3824
2d21ac55
A
3825 kn->kn_data = so->so_rcv.sb_cc - so->so_rcv.sb_ctl;
3826
91447636
A
3827 if (so->so_oobmark) {
3828 if (kn->kn_flags & EV_OOBAND) {
2d21ac55 3829 kn->kn_data -= so->so_oobmark;
91447636
A
3830 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3831 socket_unlock(so, 1);
3832 return (1);
3833 }
3834 kn->kn_data = so->so_oobmark;
3835 kn->kn_flags |= EV_OOBAND;
3836 } else {
91447636
A
3837 if (so->so_state & SS_CANTRCVMORE) {
3838 kn->kn_flags |= EV_EOF;
3839 kn->kn_fflags = so->so_error;
3840 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3841 socket_unlock(so, 1);
3842 return (1);
3843 }
55e303ae 3844 }
91447636
A
3845
3846 if (so->so_state & SS_RCVATMARK) {
3847 if (kn->kn_flags & EV_OOBAND) {
3848 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3849 socket_unlock(so, 1);
3850 return (1);
3851 }
3852 kn->kn_flags |= EV_OOBAND;
3853 } else if (kn->kn_flags & EV_OOBAND) {
3854 kn->kn_data = 0;
3855 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3856 socket_unlock(so, 1);
3857 return (0);
3858 }
3859
3860 if (so->so_error) { /* temporary udp error */
3861 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3862 socket_unlock(so, 1);
55e303ae 3863 return (1);
91447636
A
3864 }
3865
3866 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3867 socket_unlock(so, 1);
3868
2d21ac55
A
3869 return ((kn->kn_flags & EV_OOBAND) ||
3870 kn->kn_data >= ((kn->kn_sfflags & NOTE_LOWAT) ?
3871 kn->kn_sdata : so->so_rcv.sb_lowat));
55e303ae
A
3872}
3873
3874static void
3875filt_sowdetach(struct knote *kn)
3876{
91447636
A
3877 struct socket *so = (struct socket *)kn->kn_fp->f_fglob->fg_data;
3878 socket_lock(so, 1);
55e303ae 3879
2d21ac55 3880 if (so->so_snd.sb_flags & SB_KNOTE)
55e303ae
A
3881 if (KNOTE_DETACH(&so->so_snd.sb_sel.si_note, kn))
3882 so->so_snd.sb_flags &= ~SB_KNOTE;
91447636 3883 socket_unlock(so, 1);
55e303ae
A
3884}
3885
3886/*ARGSUSED*/
3887static int
3888filt_sowrite(struct knote *kn, long hint)
3889{
91447636
A
3890 struct socket *so = (struct socket *)kn->kn_fp->f_fglob->fg_data;
3891
3892 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3893 socket_lock(so, 1);
55e303ae
A
3894
3895 kn->kn_data = sbspace(&so->so_snd);
3896 if (so->so_state & SS_CANTSENDMORE) {
2d21ac55 3897 kn->kn_flags |= EV_EOF;
55e303ae 3898 kn->kn_fflags = so->so_error;
91447636
A
3899 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3900 socket_unlock(so, 1);
55e303ae
A
3901 return (1);
3902 }
91447636
A
3903 if (so->so_error) { /* temporary udp error */
3904 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3905 socket_unlock(so, 1);
55e303ae 3906 return (1);
91447636 3907 }
55e303ae 3908 if (((so->so_state & SS_ISCONNECTED) == 0) &&
91447636
A
3909 (so->so_proto->pr_flags & PR_CONNREQUIRED)) {
3910 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3911 socket_unlock(so, 1);
55e303ae 3912 return (0);
91447636
A
3913 }
3914 if ((hint & SO_FILT_HINT_LOCKED) == 0)
3915 socket_unlock(so, 1);
55e303ae
A
3916 if (kn->kn_sfflags & NOTE_LOWAT)
3917 return (kn->kn_data >= kn->kn_sdata);
3918 return (kn->kn_data >= so->so_snd.sb_lowat);
3919}
3920
b0d623f7
A
3921#define SO_LOCK_HISTORY_STR_LEN (2 * SO_LCKDBG_MAX * (2 + sizeof(void *) + 1) + 1)
3922
3923__private_extern__ const char * solockhistory_nr(struct socket *so)
55e303ae 3924{
b0d623f7
A
3925 size_t n = 0;
3926 int i;
3927 static char lock_history_str[SO_LOCK_HISTORY_STR_LEN];
55e303ae 3928
b0d623f7
A
3929 for (i = SO_LCKDBG_MAX - 1; i >= 0; i--) {
3930 n += snprintf(lock_history_str + n, SO_LOCK_HISTORY_STR_LEN - n, "%lx:%lx ",
3931 (uintptr_t) so->lock_lr[(so->next_lock_lr + i) % SO_LCKDBG_MAX],
3932 (uintptr_t) so->unlock_lr[(so->next_unlock_lr + i) % SO_LCKDBG_MAX]);
3933 }
3934 return lock_history_str;
55e303ae
A
3935}
3936
91447636 3937int
2d21ac55 3938socket_lock(struct socket *so, int refcount)
91447636 3939{
b0d623f7
A
3940 int error = 0;
3941 void *lr_saved;
0c530ab8 3942
b0d623f7 3943 lr_saved = __builtin_return_address(0);
91447636
A
3944
3945 if (so->so_proto->pr_lock) {
3946 error = (*so->so_proto->pr_lock)(so, refcount, lr_saved);
2d21ac55 3947 } else {
91447636 3948#ifdef MORE_LOCKING_DEBUG
2d21ac55
A
3949 lck_mtx_assert(so->so_proto->pr_domain->dom_mtx,
3950 LCK_MTX_ASSERT_NOTOWNED);
91447636
A
3951#endif
3952 lck_mtx_lock(so->so_proto->pr_domain->dom_mtx);
3953 if (refcount)
3954 so->so_usecount++;
b0d623f7 3955 so->lock_lr[so->next_lock_lr] = lr_saved;
0c530ab8 3956 so->next_lock_lr = (so->next_lock_lr+1) % SO_LCKDBG_MAX;
91447636
A
3957 }
3958
2d21ac55 3959 return (error);
91447636
A
3960}
3961
3962int
2d21ac55 3963socket_unlock(struct socket *so, int refcount)
91447636 3964{
b0d623f7
A
3965 int error = 0;
3966 void *lr_saved;
2d21ac55 3967 lck_mtx_t *mutex_held;
91447636 3968
b0d623f7 3969 lr_saved = __builtin_return_address(0);
91447636
A
3970
3971 if (so->so_proto == NULL)
2d21ac55 3972 panic("socket_unlock null so_proto so=%p\n", so);
91447636 3973
2d21ac55 3974 if (so && so->so_proto->pr_unlock) {
91447636 3975 error = (*so->so_proto->pr_unlock)(so, refcount, lr_saved);
2d21ac55 3976 } else {
91447636
A
3977 mutex_held = so->so_proto->pr_domain->dom_mtx;
3978#ifdef MORE_LOCKING_DEBUG
3979 lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
3980#endif
b0d623f7 3981 so->unlock_lr[so->next_unlock_lr] = lr_saved;
0c530ab8
A
3982 so->next_unlock_lr = (so->next_unlock_lr+1) % SO_LCKDBG_MAX;
3983
91447636
A
3984 if (refcount) {
3985 if (so->so_usecount <= 0)
b0d623f7
A
3986 panic("socket_unlock: bad refcount=%d so=%p (%d, %d, %d) lrh=%s",
3987 so->so_usecount, so, so->so_proto->pr_domain->dom_family,
3988 so->so_type, so->so_proto->pr_protocol,
3989 solockhistory_nr(so));
3990
91447636
A
3991 so->so_usecount--;
3992 if (so->so_usecount == 0) {
3993 sofreelastref(so, 1);
3994 }
91447636
A
3995 }
3996 lck_mtx_unlock(mutex_held);
3997 }
3998
2d21ac55 3999 return (error);
91447636 4000}
2d21ac55
A
4001
4002/* Called with socket locked, will unlock socket */
91447636 4003void
2d21ac55 4004sofree(struct socket *so)
91447636
A
4005{
4006
2d21ac55
A
4007 lck_mtx_t *mutex_held;
4008 if (so->so_proto->pr_getlock != NULL)
91447636 4009 mutex_held = (*so->so_proto->pr_getlock)(so, 0);
2d21ac55 4010 else
91447636
A
4011 mutex_held = so->so_proto->pr_domain->dom_mtx;
4012 lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
2d21ac55 4013
91447636
A
4014 sofreelastref(so, 0);
4015}
4016
4017void
2d21ac55 4018soreference(struct socket *so)
91447636
A
4019{
4020 socket_lock(so, 1); /* locks & take one reference on socket */
4021 socket_unlock(so, 0); /* unlock only */
4022}
4023
4024void
2d21ac55 4025sodereference(struct socket *so)
91447636
A
4026{
4027 socket_lock(so, 0);
4028 socket_unlock(so, 1);
4029}
2d21ac55
A
4030
4031/*
4032 * Set or clear SOF_MULTIPAGES on the socket to enable or disable the
4033 * possibility of using jumbo clusters. Caller must ensure to hold
4034 * the socket lock.
4035 */
4036void
4037somultipages(struct socket *so, boolean_t set)
4038{
4039 if (set)
4040 so->so_flags |= SOF_MULTIPAGES;
4041 else
4042 so->so_flags &= ~SOF_MULTIPAGES;
4043}
b0d623f7
A
4044
4045int
4046so_isdstlocal(struct socket *so) {
4047
4048 struct inpcb *inp = (struct inpcb *)so->so_pcb;
4049
4050 if (so->so_proto->pr_domain->dom_family == AF_INET) {
4051 return inaddr_local(inp->inp_faddr);
4052 } else if (so->so_proto->pr_domain->dom_family == AF_INET6) {
4053 return in6addr_local(&inp->in6p_faddr);
4054 }
4055 return 0;
4056}