2 * Copyright (c) 1998-2008 Apple Inc. All rights reserved.
4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
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.
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
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
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
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.
26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
28 /* Copyright (c) 1995 NeXT Computer, Inc. All Rights Reserved */
30 * Copyright (c) 1982, 1986, 1988, 1990, 1993
31 * The Regents of the University of California. All rights reserved.
33 * Redistribution and use in source and binary forms, with or without
34 * modification, are permitted provided that the following conditions
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.
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
61 * @(#)uipc_socket2.c 8.1 (Berkeley) 6/10/93
62 * $FreeBSD: src/sys/kern/uipc_socket2.c,v 1.55.2.9 2001/07/26 18:53:02 peter Exp $
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,
71 #include <sys/param.h>
72 #include <sys/systm.h>
73 #include <sys/domain.h>
74 #include <sys/kernel.h>
75 #include <sys/proc_internal.h>
76 #include <sys/kauth.h>
77 #include <sys/malloc.h>
79 #include <sys/protosw.h>
81 #include <sys/socket.h>
82 #include <sys/socketvar.h>
83 #include <sys/signalvar.h>
84 #include <sys/sysctl.h>
86 #include <kern/locks.h>
87 #include <net/route.h>
88 #include <netinet/in.h>
89 #include <netinet/in_pcb.h>
90 #include <sys/kdebug.h>
91 #include <libkern/OSAtomic.h>
94 #include <security/mac_framework.h>
97 /* TODO: this should be in a header file somewhere */
98 extern void postevent(struct socket
*, struct sockbuf
*, int);
100 #define DBG_FNC_SBDROP NETDBG_CODE(DBG_NETSOCK, 4)
101 #define DBG_FNC_SBAPPEND NETDBG_CODE(DBG_NETSOCK, 5)
103 static inline void sbcompress(struct sockbuf
*, struct mbuf
*, struct mbuf
*);
104 static struct socket
*sonewconn_internal(struct socket
*, int);
105 static int sbappendaddr_internal(struct sockbuf
*, struct sockaddr
*,
106 struct mbuf
*, struct mbuf
*);
107 static int sbappendcontrol_internal(struct sockbuf
*, struct mbuf
*,
111 * Primitive routines for operating on sockets and socket buffers
113 static int soqlimitcompat
= 1;
114 static int soqlencomp
= 0;
116 /* Based on the number of mbuf clusters configured, high_sb_max and sb_max can get
117 * scaled up or down to suit that memory configuration. high_sb_max is a higher
118 * limit on sb_max that is checked when sb_max gets set through sysctl.
121 u_int32_t sb_max
= SB_MAX
; /* XXX should be static */
122 u_int32_t high_sb_max
= SB_MAX
;
124 static u_int32_t sb_efficiency
= 8; /* parameter for sbreserve() */
125 __private_extern__
unsigned int total_mb_cnt
= 0;
126 __private_extern__
unsigned int total_cl_cnt
= 0;
127 __private_extern__
int sbspace_factor
= 8;
130 * Procedures to manipulate state flags of socket
131 * and do appropriate wakeups. Normal sequence from the
132 * active (originating) side is that soisconnecting() is
133 * called during processing of connect() call,
134 * resulting in an eventual call to soisconnected() if/when the
135 * connection is established. When the connection is torn down
136 * soisdisconnecting() is called during processing of disconnect() call,
137 * and soisdisconnected() is called when the connection to the peer
138 * is totally severed. The semantics of these routines are such that
139 * connectionless protocols can call soisconnected() and soisdisconnected()
140 * only, bypassing the in-progress calls when setting up a ``connection''
143 * From the passive side, a socket is created with
144 * two queues of sockets: so_incomp for connections in progress
145 * and so_comp for connections already made and awaiting user acceptance.
146 * As a protocol is preparing incoming connections, it creates a socket
147 * structure queued on so_incomp by calling sonewconn(). When the connection
148 * is established, soisconnected() is called, and transfers the
149 * socket structure to so_comp, making it available to accept().
151 * If a socket is closed with sockets on either
152 * so_incomp or so_comp, these sockets are dropped.
154 * If higher level protocols are implemented in
155 * the kernel, the wakeups done here will sometimes
156 * cause software-interrupt process scheduling.
159 soisconnecting(struct socket
*so
)
162 so
->so_state
&= ~(SS_ISCONNECTED
|SS_ISDISCONNECTING
);
163 so
->so_state
|= SS_ISCONNECTING
;
165 sflt_notify(so
, sock_evt_connecting
, NULL
);
169 soisconnected(struct socket
*so
)
171 struct socket
*head
= so
->so_head
;
173 so
->so_state
&= ~(SS_ISCONNECTING
|SS_ISDISCONNECTING
|SS_ISCONFIRMING
);
174 so
->so_state
|= SS_ISCONNECTED
;
176 sflt_notify(so
, sock_evt_connected
, NULL
);
178 if (head
&& (so
->so_state
& SS_INCOMP
)) {
179 so
->so_state
&= ~SS_INCOMP
;
180 so
->so_state
|= SS_COMP
;
181 if (head
->so_proto
->pr_getlock
!= NULL
) {
182 socket_unlock(so
, 0);
183 socket_lock(head
, 1);
185 postevent(head
, 0, EV_RCONN
);
186 TAILQ_REMOVE(&head
->so_incomp
, so
, so_list
);
188 TAILQ_INSERT_TAIL(&head
->so_comp
, so
, so_list
);
190 wakeup_one((caddr_t
)&head
->so_timeo
);
191 if (head
->so_proto
->pr_getlock
!= NULL
) {
192 socket_unlock(head
, 1);
196 postevent(so
, 0, EV_WCONN
);
197 wakeup((caddr_t
)&so
->so_timeo
);
204 soisdisconnecting(struct socket
*so
)
206 so
->so_state
&= ~SS_ISCONNECTING
;
207 so
->so_state
|= (SS_ISDISCONNECTING
|SS_CANTRCVMORE
|SS_CANTSENDMORE
);
208 sflt_notify(so
, sock_evt_disconnecting
, NULL
);
209 wakeup((caddr_t
)&so
->so_timeo
);
215 soisdisconnected(struct socket
*so
)
217 so
->so_state
&= ~(SS_ISCONNECTING
|SS_ISCONNECTED
|SS_ISDISCONNECTING
);
218 so
->so_state
|= (SS_CANTRCVMORE
|SS_CANTSENDMORE
|SS_ISDISCONNECTED
);
219 sflt_notify(so
, sock_evt_disconnected
, NULL
);
220 wakeup((caddr_t
)&so
->so_timeo
);
226 * When an attempt at a new connection is noted on a socket
227 * which accepts connections, sonewconn is called. If the
228 * connection is possible (subject to space constraints, etc.)
229 * then we allocate a new structure, propoerly linked into the
230 * data structure of the original socket, and return this.
231 * Connstatus may be 0, or SO_ISCONFIRMING, or SO_ISCONNECTED.
233 static struct socket
*
234 sonewconn_internal(struct socket
*head
, int connstatus
)
236 int so_qlen
, error
= 0;
238 lck_mtx_t
*mutex_held
;
240 if (head
->so_proto
->pr_getlock
!= NULL
)
241 mutex_held
= (*head
->so_proto
->pr_getlock
)(head
, 0);
243 mutex_held
= head
->so_proto
->pr_domain
->dom_mtx
;
244 lck_mtx_assert(mutex_held
, LCK_MTX_ASSERT_OWNED
);
248 * This is the default case; so_qlen represents the
249 * sum of both incomplete and completed queues.
251 so_qlen
= head
->so_qlen
;
254 * When kern.ipc.soqlencomp is set to 1, so_qlen
255 * represents only the completed queue. Since we
256 * cannot let the incomplete queue goes unbounded
257 * (in case of SYN flood), we cap the incomplete
258 * queue length to at most somaxconn, and use that
259 * as so_qlen so that we fail immediately below.
261 so_qlen
= head
->so_qlen
- head
->so_incqlen
;
262 if (head
->so_incqlen
> somaxconn
)
267 (soqlimitcompat
? head
->so_qlimit
: (3 * head
->so_qlimit
/ 2)))
268 return ((struct socket
*)0);
269 so
= soalloc(1, head
->so_proto
->pr_domain
->dom_family
,
272 return ((struct socket
*)0);
273 /* check if head was closed during the soalloc */
274 if (head
->so_proto
== NULL
) {
276 return ((struct socket
*)0);
280 so
->so_type
= head
->so_type
;
281 so
->so_options
= head
->so_options
&~ SO_ACCEPTCONN
;
282 so
->so_linger
= head
->so_linger
;
283 so
->so_state
= head
->so_state
| SS_NOFDREF
;
284 so
->so_proto
= head
->so_proto
;
285 so
->so_timeo
= head
->so_timeo
;
286 so
->so_pgid
= head
->so_pgid
;
287 so
->so_uid
= head
->so_uid
;
288 /* inherit socket options stored in so_flags */
289 so
->so_flags
= head
->so_flags
& (SOF_NOSIGPIPE
|
296 so
->next_lock_lr
= 0;
297 so
->next_unlock_lr
= 0;
300 so
->so_rcv
.sb_flags
|= SB_RECV
; /* XXX */
301 so
->so_rcv
.sb_so
= so
->so_snd
.sb_so
= so
;
302 TAILQ_INIT(&so
->so_evlist
);
305 #if CONFIG_MACF_SOCKET
306 mac_socket_label_associate_accept(head
, so
);
309 /* inherit traffic management properties of listener */
310 so
->so_traffic_mgt_flags
= head
->so_traffic_mgt_flags
&
311 (TRAFFIC_MGT_SO_BACKGROUND
| TRAFFIC_MGT_SO_BG_REGULATE
);
312 so
->so_background_thread
= head
->so_background_thread
;
314 if (soreserve(so
, head
->so_snd
.sb_hiwat
, head
->so_rcv
.sb_hiwat
)) {
317 return ((struct socket
*)0);
321 * Must be done with head unlocked to avoid deadlock
322 * for protocol with per socket mutexes.
324 if (head
->so_proto
->pr_unlock
)
325 socket_unlock(head
, 0);
326 if (((*so
->so_proto
->pr_usrreqs
->pru_attach
)(so
, 0, NULL
) != 0) ||
330 if (head
->so_proto
->pr_unlock
)
331 socket_lock(head
, 0);
332 return ((struct socket
*)0);
334 if (head
->so_proto
->pr_unlock
)
335 socket_lock(head
, 0);
337 so
->so_proto
->pr_domain
->dom_refs
++;
341 TAILQ_INSERT_TAIL(&head
->so_comp
, so
, so_list
);
342 so
->so_state
|= SS_COMP
;
344 TAILQ_INSERT_TAIL(&head
->so_incomp
, so
, so_list
);
345 so
->so_state
|= SS_INCOMP
;
351 /* Attach socket filters for this protocol */
356 so
->so_state
|= connstatus
;
358 wakeup((caddr_t
)&head
->so_timeo
);
365 sonewconn(struct socket
*head
, int connstatus
, const struct sockaddr
*from
)
368 struct socket_filter_entry
*filter
;
371 for (filter
= head
->so_filt
; filter
&& (error
== 0);
372 filter
= filter
->sfe_next_onsocket
) {
373 if (filter
->sfe_filter
->sf_filter
.sf_connect_in
) {
377 socket_unlock(head
, 0);
379 error
= filter
->sfe_filter
->sf_filter
.
380 sf_connect_in(filter
->sfe_cookie
, head
, from
);
384 socket_lock(head
, 0);
392 return (sonewconn_internal(head
, connstatus
));
396 * Socantsendmore indicates that no more data will be sent on the
397 * socket; it would normally be applied to a socket when the user
398 * informs the system that no more data is to be sent, by the protocol
399 * code (in case PRU_SHUTDOWN). Socantrcvmore indicates that no more data
400 * will be received, and will normally be applied to the socket by a
401 * protocol when it detects that the peer will send no more data.
402 * Data queued for reading in the socket may yet be read.
406 socantsendmore(struct socket
*so
)
408 so
->so_state
|= SS_CANTSENDMORE
;
409 sflt_notify(so
, sock_evt_cantsendmore
, NULL
);
414 socantrcvmore(struct socket
*so
)
416 so
->so_state
|= SS_CANTRCVMORE
;
417 sflt_notify(so
, sock_evt_cantrecvmore
, NULL
);
422 * Wait for data to arrive at/drain from a socket buffer.
429 sbwait(struct sockbuf
*sb
)
433 struct socket
*so
= sb
->sb_so
;
434 lck_mtx_t
*mutex_held
;
437 lr_saved
= (uintptr_t) __builtin_return_address(0);
439 if (so
->so_proto
->pr_getlock
!= NULL
)
440 mutex_held
= (*so
->so_proto
->pr_getlock
)(so
, 0);
442 mutex_held
= so
->so_proto
->pr_domain
->dom_mtx
;
444 sb
->sb_flags
|= SB_WAIT
;
446 if (so
->so_usecount
< 1)
447 panic("sbwait: so=%p refcount=%d\n", so
, so
->so_usecount
);
448 ts
.tv_sec
= sb
->sb_timeo
.tv_sec
;
449 ts
.tv_nsec
= sb
->sb_timeo
.tv_usec
* 1000;
450 error
= msleep((caddr_t
)&sb
->sb_cc
, mutex_held
,
451 (sb
->sb_flags
& SB_NOINTR
) ? PSOCK
: PSOCK
| PCATCH
, "sbwait", &ts
);
453 lck_mtx_assert(mutex_held
, LCK_MTX_ASSERT_OWNED
);
455 if (so
->so_usecount
< 1)
456 panic("sbwait: so=%p refcount=%d\n", so
, so
->so_usecount
);
458 if ((so
->so_state
& SS_DRAINING
)) {
466 * Lock a sockbuf already known to be locked;
467 * return any error returned from sleep (EINTR).
473 sb_lock(struct sockbuf
*sb
)
475 struct socket
*so
= sb
->sb_so
;
476 lck_mtx_t
*mutex_held
;
480 panic("sb_lock: null so back pointer sb=%p\n", sb
);
482 while (sb
->sb_flags
& SB_LOCK
) {
483 sb
->sb_flags
|= SB_WANT
;
484 if (so
->so_proto
->pr_getlock
!= NULL
)
485 mutex_held
= (*so
->so_proto
->pr_getlock
)(so
, 0);
487 mutex_held
= so
->so_proto
->pr_domain
->dom_mtx
;
488 if (so
->so_usecount
< 1)
489 panic("sb_lock: so=%p refcount=%d\n", so
,
492 error
= msleep((caddr_t
)&sb
->sb_flags
, mutex_held
,
493 (sb
->sb_flags
& SB_NOINTR
) ? PSOCK
: PSOCK
| PCATCH
,
495 if (so
->so_usecount
< 1)
496 panic("sb_lock: 2 so=%p refcount=%d\n", so
,
501 sb
->sb_flags
|= SB_LOCK
;
506 * Wakeup processes waiting on a socket buffer.
507 * Do asynchronous notification via SIGIO
508 * if the socket has the SS_ASYNC flag set.
511 sowakeup(struct socket
*so
, struct sockbuf
*sb
)
513 sb
->sb_flags
&= ~SB_SEL
;
514 selwakeup(&sb
->sb_sel
);
515 if (sb
->sb_flags
& SB_WAIT
) {
516 sb
->sb_flags
&= ~SB_WAIT
;
517 wakeup((caddr_t
)&sb
->sb_cc
);
519 if (so
->so_state
& SS_ASYNC
) {
521 gsignal(-so
->so_pgid
, SIGIO
);
522 else if (so
->so_pgid
> 0)
523 proc_signal(so
->so_pgid
, SIGIO
);
525 if (sb
->sb_flags
& SB_KNOTE
) {
526 KNOTE(&sb
->sb_sel
.si_note
, SO_FILT_HINT_LOCKED
);
528 if (sb
->sb_flags
& SB_UPCALL
) {
529 void (*so_upcall
)(struct socket
*, caddr_t
, int);
530 caddr_t so_upcallarg
;
532 so_upcall
= so
->so_upcall
;
533 so_upcallarg
= so
->so_upcallarg
;
534 /* Let close know that we're about to do an upcall */
535 so
->so_flags
|= SOF_UPCALLINUSE
;
537 socket_unlock(so
, 0);
538 (*so_upcall
)(so
, so_upcallarg
, M_DONTWAIT
);
541 so
->so_flags
&= ~SOF_UPCALLINUSE
;
542 /* Tell close that it's safe to proceed */
543 if (so
->so_flags
& SOF_CLOSEWAIT
)
544 wakeup((caddr_t
)&so
->so_upcall
);
549 * Socket buffer (struct sockbuf) utility routines.
551 * Each socket contains two socket buffers: one for sending data and
552 * one for receiving data. Each buffer contains a queue of mbufs,
553 * information about the number of mbufs and amount of data in the
554 * queue, and other fields allowing select() statements and notification
555 * on data availability to be implemented.
557 * Data stored in a socket buffer is maintained as a list of records.
558 * Each record is a list of mbufs chained together with the m_next
559 * field. Records are chained together with the m_nextpkt field. The upper
560 * level routine soreceive() expects the following conventions to be
561 * observed when placing information in the receive buffer:
563 * 1. If the protocol requires each message be preceded by the sender's
564 * name, then a record containing that name must be present before
565 * any associated data (mbuf's must be of type MT_SONAME).
566 * 2. If the protocol supports the exchange of ``access rights'' (really
567 * just additional data associated with the message), and there are
568 * ``rights'' to be received, then a record containing this data
569 * should be present (mbuf's must be of type MT_RIGHTS).
570 * 3. If a name or rights record exists, then it must be followed by
571 * a data record, perhaps of zero length.
573 * Before using a new socket structure it is first necessary to reserve
574 * buffer space to the socket, by calling sbreserve(). This should commit
575 * some of the available buffer space in the system buffer pool for the
576 * socket (currently, it does nothing but enforce limits). The space
577 * should be released by calling sbrelease() when the socket is destroyed.
585 soreserve(struct socket
*so
, u_int32_t sndcc
, u_int32_t rcvcc
)
588 if (sbreserve(&so
->so_snd
, sndcc
) == 0)
590 if (sbreserve(&so
->so_rcv
, rcvcc
) == 0)
592 if (so
->so_rcv
.sb_lowat
== 0)
593 so
->so_rcv
.sb_lowat
= 1;
594 if (so
->so_snd
.sb_lowat
== 0)
595 so
->so_snd
.sb_lowat
= MCLBYTES
;
596 if (so
->so_snd
.sb_lowat
> so
->so_snd
.sb_hiwat
)
597 so
->so_snd
.sb_lowat
= so
->so_snd
.sb_hiwat
;
601 selthreadclear(&so
->so_snd
.sb_sel
);
603 sbrelease(&so
->so_snd
);
609 * Allot mbufs to a sockbuf.
610 * Attempt to scale mbmax so that mbcnt doesn't become limiting
611 * if buffering efficiency is near the normal case.
614 sbreserve(struct sockbuf
*sb
, u_int32_t cc
)
616 if ((u_quad_t
)cc
> (u_quad_t
)sb_max
* MCLBYTES
/ (MSIZE
+ MCLBYTES
))
619 sb
->sb_mbmax
= min(cc
* sb_efficiency
, sb_max
);
620 if (sb
->sb_lowat
> sb
->sb_hiwat
)
621 sb
->sb_lowat
= sb
->sb_hiwat
;
626 * Free mbufs held by a socket, and reserved mbuf space.
628 /* WARNING needs to do selthreadclear() before calling this */
630 sbrelease(struct sockbuf
*sb
)
638 * Routines to add and remove
639 * data from an mbuf queue.
641 * The routines sbappend() or sbappendrecord() are normally called to
642 * append new mbufs to a socket buffer, after checking that adequate
643 * space is available, comparing the function sbspace() with the amount
644 * of data to be added. sbappendrecord() differs from sbappend() in
645 * that data supplied is treated as the beginning of a new record.
646 * To place a sender's address, optional access rights, and data in a
647 * socket receive buffer, sbappendaddr() should be used. To place
648 * access rights and data in a socket receive buffer, sbappendrights()
649 * should be used. In either case, the new data begins a new record.
650 * Note that unlike sbappend() and sbappendrecord(), these routines check
651 * for the caller that there will be enough space to store the data.
652 * Each fails if there is not enough space, or if it cannot find mbufs
653 * to store additional information in.
655 * Reliable protocols may use the socket send buffer to hold data
656 * awaiting acknowledgement. Data is normally copied from a socket
657 * send buffer in a protocol with m_copy for output to a peer,
658 * and then removing the data from the socket buffer with sbdrop()
659 * or sbdroprecord() when the data is acknowledged by the peer.
663 * Append mbuf chain m to the last record in the
664 * socket buffer sb. The additional space associated
665 * the mbuf chain is recorded in sb. Empty mbufs are
666 * discarded and mbufs are compacted where possible.
669 sbappend(struct sockbuf
*sb
, struct mbuf
*m
)
671 struct socket
*so
= sb
->sb_so
;
673 if (m
== NULL
|| (sb
->sb_flags
& SB_DROP
)) {
679 SBLASTRECORDCHK(sb
, "sbappend 1");
681 if (sb
->sb_lastrecord
!= NULL
&& (sb
->sb_mbtail
->m_flags
& M_EOR
))
682 return (sbappendrecord(sb
, m
));
684 if (sb
->sb_flags
& SB_RECV
) {
685 int error
= sflt_data_in(so
, NULL
, &m
, NULL
, 0, NULL
);
686 SBLASTRECORDCHK(sb
, "sbappend 2");
688 if (error
!= EJUSTRETURN
)
694 /* If this is the first record, it's also the last record */
695 if (sb
->sb_lastrecord
== NULL
)
696 sb
->sb_lastrecord
= m
;
698 sbcompress(sb
, m
, sb
->sb_mbtail
);
699 SBLASTRECORDCHK(sb
, "sbappend 3");
704 * Similar to sbappend, except that this is optimized for stream sockets.
707 sbappendstream(struct sockbuf
*sb
, struct mbuf
*m
)
709 struct socket
*so
= sb
->sb_so
;
711 if (m
->m_nextpkt
!= NULL
|| (sb
->sb_mb
!= sb
->sb_lastrecord
))
712 panic("sbappendstream: nexpkt %p || mb %p != lastrecord %p\n",
713 m
->m_nextpkt
, sb
->sb_mb
, sb
->sb_lastrecord
);
715 SBLASTMBUFCHK(sb
, __func__
);
717 if (m
== NULL
|| (sb
->sb_flags
& SB_DROP
)) {
723 if (sb
->sb_flags
& SB_RECV
) {
724 int error
= sflt_data_in(so
, NULL
, &m
, NULL
, 0, NULL
);
725 SBLASTRECORDCHK(sb
, "sbappendstream 1");
727 if (error
!= EJUSTRETURN
)
733 sbcompress(sb
, m
, sb
->sb_mbtail
);
734 sb
->sb_lastrecord
= sb
->sb_mb
;
735 SBLASTRECORDCHK(sb
, "sbappendstream 2");
741 sbcheck(struct sockbuf
*sb
)
745 u_int32_t len
= 0, mbcnt
= 0;
746 lck_mtx_t
*mutex_held
;
748 if (sb
->sb_so
->so_proto
->pr_getlock
!= NULL
)
749 mutex_held
= (*sb
->sb_so
->so_proto
->pr_getlock
)(sb
->sb_so
, 0);
751 mutex_held
= sb
->sb_so
->so_proto
->pr_domain
->dom_mtx
;
753 lck_mtx_assert(mutex_held
, LCK_MTX_ASSERT_OWNED
);
758 for (m
= sb
->sb_mb
; m
; m
= n
) {
760 for (; m
; m
= m
->m_next
) {
763 /* XXX pretty sure this is bogus */
764 if (m
->m_flags
& M_EXT
)
765 mbcnt
+= m
->m_ext
.ext_size
;
768 if (len
!= sb
->sb_cc
|| mbcnt
!= sb
->sb_mbcnt
) {
769 panic("cc %ld != %ld || mbcnt %ld != %ld\n", len
, sb
->sb_cc
,
770 mbcnt
, sb
->sb_mbcnt
);
776 sblastrecordchk(struct sockbuf
*sb
, const char *where
)
778 struct mbuf
*m
= sb
->sb_mb
;
780 while (m
&& m
->m_nextpkt
)
783 if (m
!= sb
->sb_lastrecord
) {
784 printf("sblastrecordchk: mb %p lastrecord %p last %p\n",
785 sb
->sb_mb
, sb
->sb_lastrecord
, m
);
786 printf("packet chain:\n");
787 for (m
= sb
->sb_mb
; m
!= NULL
; m
= m
->m_nextpkt
)
789 panic("sblastrecordchk from %s", where
);
794 sblastmbufchk(struct sockbuf
*sb
, const char *where
)
796 struct mbuf
*m
= sb
->sb_mb
;
799 while (m
&& m
->m_nextpkt
)
802 while (m
&& m
->m_next
)
805 if (m
!= sb
->sb_mbtail
) {
806 printf("sblastmbufchk: mb %p mbtail %p last %p\n",
807 sb
->sb_mb
, sb
->sb_mbtail
, m
);
808 printf("packet tree:\n");
809 for (m
= sb
->sb_mb
; m
!= NULL
; m
= m
->m_nextpkt
) {
811 for (n
= m
; n
!= NULL
; n
= n
->m_next
)
815 panic("sblastmbufchk from %s", where
);
820 * Similar to sbappend, except the mbuf chain begins a new record.
823 sbappendrecord(struct sockbuf
*sb
, struct mbuf
*m0
)
828 if (m0
== NULL
|| (sb
->sb_flags
& SB_DROP
)) {
834 for (m
= m0
; m
!= NULL
; m
= m
->m_next
)
837 if (space
> sbspace(sb
) && !(sb
->sb_flags
& SB_UNIX
)) {
842 if (sb
->sb_flags
& SB_RECV
) {
843 int error
= sflt_data_in(sb
->sb_so
, NULL
, &m0
, NULL
,
844 sock_data_filt_flag_record
, NULL
);
846 SBLASTRECORDCHK(sb
, "sbappendrecord 1");
847 if (error
!= EJUSTRETURN
)
854 * Note this permits zero length records.
857 SBLASTRECORDCHK(sb
, "sbappendrecord 2");
858 if (sb
->sb_lastrecord
!= NULL
) {
859 sb
->sb_lastrecord
->m_nextpkt
= m0
;
863 sb
->sb_lastrecord
= m0
;
868 if (m
&& (m0
->m_flags
& M_EOR
)) {
869 m0
->m_flags
&= ~M_EOR
;
872 sbcompress(sb
, m
, m0
);
873 SBLASTRECORDCHK(sb
, "sbappendrecord 3");
878 * As above except that OOB data
879 * is inserted at the beginning of the sockbuf,
880 * but after any other OOB data.
883 sbinsertoob(struct sockbuf
*sb
, struct mbuf
*m0
)
891 SBLASTRECORDCHK(sb
, "sbinsertoob 1");
893 if ((sb
->sb_flags
& SB_RECV
) != 0) {
894 int error
= sflt_data_in(sb
->sb_so
, NULL
, &m0
, NULL
,
895 sock_data_filt_flag_oob
, NULL
);
897 SBLASTRECORDCHK(sb
, "sbinsertoob 2");
899 if (error
!= EJUSTRETURN
) {
906 for (mp
= &sb
->sb_mb
; *mp
; mp
= &((*mp
)->m_nextpkt
)) {
912 continue; /* WANT next train */
917 goto again
; /* inspect THIS train further */
922 * Put the first mbuf on the queue.
923 * Note this permits zero length records.
928 /* m0 is actually the new tail */
929 sb
->sb_lastrecord
= m0
;
934 if (m
&& (m0
->m_flags
& M_EOR
)) {
935 m0
->m_flags
&= ~M_EOR
;
938 sbcompress(sb
, m
, m0
);
939 SBLASTRECORDCHK(sb
, "sbinsertoob 3");
944 * Append address and data, and optionally, control (ancillary) data
945 * to the receive queue of a socket. If present,
946 * m0 must include a packet header with total length.
947 * Returns 0 if no space in sockbuf or insufficient mbufs.
949 * Returns: 0 No space/out of mbufs
953 sbappendaddr_internal(struct sockbuf
*sb
, struct sockaddr
*asa
,
954 struct mbuf
*m0
, struct mbuf
*control
)
956 struct mbuf
*m
, *n
, *nlast
;
957 int space
= asa
->sa_len
;
959 if (m0
&& (m0
->m_flags
& M_PKTHDR
) == 0)
960 panic("sbappendaddr");
963 space
+= m0
->m_pkthdr
.len
;
964 for (n
= control
; n
; n
= n
->m_next
) {
966 if (n
->m_next
== 0) /* keep pointer to last control buf */
969 if (space
> sbspace(sb
))
971 if (asa
->sa_len
> MLEN
)
973 MGET(m
, M_DONTWAIT
, MT_SONAME
);
976 m
->m_len
= asa
->sa_len
;
977 bcopy((caddr_t
)asa
, mtod(m
, caddr_t
), asa
->sa_len
);
979 n
->m_next
= m0
; /* concatenate data to control */
984 SBLASTRECORDCHK(sb
, "sbappendadddr 1");
986 for (n
= m
; n
->m_next
!= NULL
; n
= n
->m_next
)
991 if (sb
->sb_lastrecord
!= NULL
) {
992 sb
->sb_lastrecord
->m_nextpkt
= m
;
996 sb
->sb_lastrecord
= m
;
997 sb
->sb_mbtail
= nlast
;
999 SBLASTMBUFCHK(sb
, __func__
);
1000 SBLASTRECORDCHK(sb
, "sbappendadddr 2");
1002 postevent(0, sb
, EV_RWBYTES
);
1007 * Returns: 0 Error: No space/out of mbufs/etc.
1010 * Imputed: (*error_out) errno for error
1012 * sflt_data_in:??? [whatever a filter author chooses]
1015 sbappendaddr(struct sockbuf
*sb
, struct sockaddr
*asa
, struct mbuf
*m0
,
1016 struct mbuf
*control
, int *error_out
)
1019 boolean_t sb_unix
= (sb
->sb_flags
& SB_UNIX
);
1024 if (m0
&& (m0
->m_flags
& M_PKTHDR
) == 0)
1025 panic("sbappendaddrorfree");
1027 if (sb
->sb_flags
& SB_DROP
) {
1030 if (control
!= NULL
&& !sb_unix
)
1032 if (error_out
!= NULL
)
1033 *error_out
= EINVAL
;
1037 /* Call socket data in filters */
1038 if ((sb
->sb_flags
& SB_RECV
) != 0) {
1040 error
= sflt_data_in(sb
->sb_so
, asa
, &m0
, &control
, 0, NULL
);
1041 SBLASTRECORDCHK(sb
, __func__
);
1043 if (error
!= EJUSTRETURN
) {
1046 if (control
!= NULL
&& !sb_unix
)
1055 result
= sbappendaddr_internal(sb
, asa
, m0
, control
);
1059 if (control
!= NULL
&& !sb_unix
)
1062 *error_out
= ENOBUFS
;
1069 sbappendcontrol_internal(struct sockbuf
*sb
, struct mbuf
*m0
,
1070 struct mbuf
*control
)
1072 struct mbuf
*m
, *mlast
, *n
;
1076 panic("sbappendcontrol");
1078 for (m
= control
; ; m
= m
->m_next
) {
1083 n
= m
; /* save pointer to last control buffer */
1084 for (m
= m0
; m
; m
= m
->m_next
)
1086 if (space
> sbspace(sb
) && !(sb
->sb_flags
& SB_UNIX
))
1088 n
->m_next
= m0
; /* concatenate data to control */
1090 SBLASTRECORDCHK(sb
, "sbappendcontrol 1");
1092 for (m
= control
; m
->m_next
!= NULL
; m
= m
->m_next
)
1097 if (sb
->sb_lastrecord
!= NULL
) {
1098 sb
->sb_lastrecord
->m_nextpkt
= control
;
1100 sb
->sb_mb
= control
;
1102 sb
->sb_lastrecord
= control
;
1103 sb
->sb_mbtail
= mlast
;
1105 SBLASTMBUFCHK(sb
, __func__
);
1106 SBLASTRECORDCHK(sb
, "sbappendcontrol 2");
1108 postevent(0, sb
, EV_RWBYTES
);
1113 sbappendcontrol(struct sockbuf
*sb
, struct mbuf
*m0
, struct mbuf
*control
,
1117 boolean_t sb_unix
= (sb
->sb_flags
& SB_UNIX
);
1122 if (sb
->sb_flags
& SB_DROP
) {
1125 if (control
!= NULL
&& !sb_unix
)
1127 if (error_out
!= NULL
)
1128 *error_out
= EINVAL
;
1132 if (sb
->sb_flags
& SB_RECV
) {
1135 error
= sflt_data_in(sb
->sb_so
, NULL
, &m0
, &control
, 0, NULL
);
1136 SBLASTRECORDCHK(sb
, __func__
);
1138 if (error
!= EJUSTRETURN
) {
1141 if (control
!= NULL
&& !sb_unix
)
1150 result
= sbappendcontrol_internal(sb
, m0
, control
);
1154 if (control
!= NULL
&& !sb_unix
)
1157 *error_out
= ENOBUFS
;
1164 * Compress mbuf chain m into the socket
1165 * buffer sb following mbuf n. If n
1166 * is null, the buffer is presumed empty.
1169 sbcompress(struct sockbuf
*sb
, struct mbuf
*m
, struct mbuf
*n
)
1175 /* There is nothing to compress; just update the tail */
1176 for (; n
->m_next
!= NULL
; n
= n
->m_next
)
1183 eor
|= m
->m_flags
& M_EOR
;
1184 if (m
->m_len
== 0 && (eor
== 0 ||
1185 (((o
= m
->m_next
) || (o
= n
)) && o
->m_type
== m
->m_type
))) {
1186 if (sb
->sb_lastrecord
== m
)
1187 sb
->sb_lastrecord
= m
->m_next
;
1191 if (n
&& (n
->m_flags
& M_EOR
) == 0 &&
1195 m
->m_len
<= MCLBYTES
/ 4 && /* XXX: Don't copy too much */
1196 m
->m_len
<= M_TRAILINGSPACE(n
) &&
1197 n
->m_type
== m
->m_type
) {
1198 bcopy(mtod(m
, caddr_t
), mtod(n
, caddr_t
) + n
->m_len
,
1199 (unsigned)m
->m_len
);
1200 n
->m_len
+= m
->m_len
;
1201 sb
->sb_cc
+= m
->m_len
;
1202 if (m
->m_type
!= MT_DATA
&& m
->m_type
!= MT_HEADER
&&
1203 m
->m_type
!= MT_OOBDATA
)
1204 /* XXX: Probably don't need.*/
1205 sb
->sb_ctl
+= m
->m_len
;
1216 m
->m_flags
&= ~M_EOR
;
1224 printf("semi-panic: sbcompress\n");
1227 SBLASTMBUFCHK(sb
, __func__
);
1228 postevent(0, sb
, EV_RWBYTES
);
1232 sb_empty_assert(struct sockbuf
*sb
, const char *where
)
1234 if (!(sb
->sb_cc
== 0 && sb
->sb_mb
== NULL
&& sb
->sb_mbcnt
== 0 &&
1235 sb
->sb_mbtail
== NULL
&& sb
->sb_lastrecord
== NULL
)) {
1236 panic("%s: sb %p so %p cc %d mbcnt %d mb %p mbtail %p "
1237 "lastrecord %p\n", where
, sb
, sb
->sb_so
, sb
->sb_cc
,
1238 sb
->sb_mbcnt
, sb
->sb_mb
, sb
->sb_mbtail
, sb
->sb_lastrecord
);
1244 * Free all mbufs in a sockbuf.
1245 * Check that all resources are reclaimed.
1248 sbflush(struct sockbuf
*sb
)
1250 if (sb
->sb_so
== NULL
)
1251 panic("sbflush sb->sb_so already null sb=%p\n", sb
);
1252 (void) sblock(sb
, M_WAIT
);
1253 while (sb
->sb_mbcnt
) {
1255 * Don't call sbdrop(sb, 0) if the leading mbuf is non-empty:
1256 * we would loop forever. Panic instead.
1258 if (!sb
->sb_cc
&& (sb
->sb_mb
== NULL
|| sb
->sb_mb
->m_len
))
1260 sbdrop(sb
, (int)sb
->sb_cc
);
1262 sb_empty_assert(sb
, __func__
);
1263 postevent(0, sb
, EV_RWBYTES
);
1264 sbunlock(sb
, 1); /* keep socket locked */
1269 * Drop data from (the front of) a sockbuf.
1270 * use m_freem_list to free the mbuf structures
1271 * under a single lock... this is done by pruning
1272 * the top of the tree from the body by keeping track
1273 * of where we get to in the tree and then zeroing the
1274 * two pertinent pointers m_nextpkt and m_next
1275 * the socket buffer is then updated to point at the new
1276 * top of the tree and the pruned area is released via
1280 sbdrop(struct sockbuf
*sb
, int len
)
1282 struct mbuf
*m
, *free_list
, *ml
;
1283 struct mbuf
*next
, *last
;
1285 KERNEL_DEBUG((DBG_FNC_SBDROP
| DBG_FUNC_START
), sb
, len
, 0, 0, 0);
1287 next
= (m
= sb
->sb_mb
) ? m
->m_nextpkt
: 0;
1288 free_list
= last
= m
;
1289 ml
= (struct mbuf
*)0;
1295 * temporarily replacing this panic with printf
1296 * because it occurs occasionally when closing
1297 * a socket when there is no harm in ignoring
1298 * it. This problem will be investigated
1301 /* panic("sbdrop"); */
1302 printf("sbdrop - count not zero\n");
1305 * zero the counts. if we have no mbufs,
1306 * we have no data (PR-2986815)
1313 next
= m
->m_nextpkt
;
1316 if (m
->m_len
> len
) {
1320 if (m
->m_type
!= MT_DATA
&& m
->m_type
!= MT_HEADER
&&
1321 m
->m_type
!= MT_OOBDATA
)
1331 while (m
&& m
->m_len
== 0) {
1338 ml
->m_next
= (struct mbuf
*)0;
1339 last
->m_nextpkt
= (struct mbuf
*)0;
1340 m_freem_list(free_list
);
1344 m
->m_nextpkt
= next
;
1350 * First part is an inline SB_EMPTY_FIXUP(). Second part
1351 * makes sure sb_lastrecord is up-to-date if we dropped
1352 * part of the last record.
1356 sb
->sb_mbtail
= NULL
;
1357 sb
->sb_lastrecord
= NULL
;
1358 } else if (m
->m_nextpkt
== NULL
) {
1359 sb
->sb_lastrecord
= m
;
1362 postevent(0, sb
, EV_RWBYTES
);
1364 KERNEL_DEBUG((DBG_FNC_SBDROP
| DBG_FUNC_END
), sb
, 0, 0, 0, 0);
1368 * Drop a record off the front of a sockbuf
1369 * and move the next record to the front.
1372 sbdroprecord(struct sockbuf
*sb
)
1374 struct mbuf
*m
, *mn
;
1378 sb
->sb_mb
= m
->m_nextpkt
;
1386 postevent(0, sb
, EV_RWBYTES
);
1390 * Create a "control" mbuf containing the specified data
1391 * with the specified type for presentation on a socket buffer.
1394 sbcreatecontrol(caddr_t p
, int size
, int type
, int level
)
1399 if (CMSG_SPACE((u_int
)size
) > MLEN
)
1400 return ((struct mbuf
*)NULL
);
1401 if ((m
= m_get(M_DONTWAIT
, MT_CONTROL
)) == NULL
)
1402 return ((struct mbuf
*)NULL
);
1403 cp
= mtod(m
, struct cmsghdr
*);
1404 /* XXX check size? */
1405 (void) memcpy(CMSG_DATA(cp
), p
, size
);
1406 m
->m_len
= CMSG_SPACE(size
);
1407 cp
->cmsg_len
= CMSG_LEN(size
);
1408 cp
->cmsg_level
= level
;
1409 cp
->cmsg_type
= type
;
1414 * Some routines that return EOPNOTSUPP for entry points that are not
1415 * supported by a protocol. Fill in as needed.
1418 pru_abort_notsupp(__unused
struct socket
*so
)
1420 return (EOPNOTSUPP
);
1424 pru_accept_notsupp(__unused
struct socket
*so
, __unused
struct sockaddr
**nam
)
1426 return (EOPNOTSUPP
);
1430 pru_attach_notsupp(__unused
struct socket
*so
, __unused
int proto
,
1431 __unused
struct proc
*p
)
1433 return (EOPNOTSUPP
);
1437 pru_bind_notsupp(__unused
struct socket
*so
, __unused
struct sockaddr
*nam
,
1438 __unused
struct proc
*p
)
1440 return (EOPNOTSUPP
);
1444 pru_connect_notsupp(__unused
struct socket
*so
, __unused
struct sockaddr
*nam
,
1445 __unused
struct proc
*p
)
1447 return (EOPNOTSUPP
);
1451 pru_connect2_notsupp(__unused
struct socket
*so1
, __unused
struct socket
*so2
)
1453 return (EOPNOTSUPP
);
1457 pru_control_notsupp(__unused
struct socket
*so
, __unused u_long cmd
,
1458 __unused caddr_t data
, __unused
struct ifnet
*ifp
, __unused
struct proc
*p
)
1460 return (EOPNOTSUPP
);
1464 pru_detach_notsupp(__unused
struct socket
*so
)
1466 return (EOPNOTSUPP
);
1470 pru_disconnect_notsupp(__unused
struct socket
*so
)
1472 return (EOPNOTSUPP
);
1476 pru_listen_notsupp(__unused
struct socket
*so
, __unused
struct proc
*p
)
1478 return (EOPNOTSUPP
);
1482 pru_peeraddr_notsupp(__unused
struct socket
*so
, __unused
struct sockaddr
**nam
)
1484 return (EOPNOTSUPP
);
1488 pru_rcvd_notsupp(__unused
struct socket
*so
, __unused
int flags
)
1490 return (EOPNOTSUPP
);
1494 pru_rcvoob_notsupp(__unused
struct socket
*so
, __unused
struct mbuf
*m
,
1497 return (EOPNOTSUPP
);
1501 pru_send_notsupp(__unused
struct socket
*so
, __unused
int flags
,
1502 __unused
struct mbuf
*m
, __unused
struct sockaddr
*addr
,
1503 __unused
struct mbuf
*control
, __unused
struct proc
*p
)
1506 return (EOPNOTSUPP
);
1511 * This isn't really a ``null'' operation, but it's the default one
1512 * and doesn't do anything destructive.
1515 pru_sense_null(struct socket
*so
, void *ub
, int isstat64
)
1517 if (isstat64
!= 0) {
1518 struct stat64
*sb64
;
1520 sb64
= (struct stat64
*)ub
;
1521 sb64
->st_blksize
= so
->so_snd
.sb_hiwat
;
1525 sb
= (struct stat
*)ub
;
1526 sb
->st_blksize
= so
->so_snd
.sb_hiwat
;
1534 pru_sosend_notsupp(__unused
struct socket
*so
, __unused
struct sockaddr
*addr
,
1535 __unused
struct uio
*uio
, __unused
struct mbuf
*top
,
1536 __unused
struct mbuf
*control
, __unused
int flags
)
1539 return (EOPNOTSUPP
);
1543 pru_soreceive_notsupp(__unused
struct socket
*so
,
1544 __unused
struct sockaddr
**paddr
,
1545 __unused
struct uio
*uio
, __unused
struct mbuf
**mp0
,
1546 __unused
struct mbuf
**controlp
, __unused
int *flagsp
)
1548 return (EOPNOTSUPP
);
1552 pru_shutdown_notsupp(__unused
struct socket
*so
)
1554 return (EOPNOTSUPP
);
1558 pru_sockaddr_notsupp(__unused
struct socket
*so
, __unused
struct sockaddr
**nam
)
1560 return (EOPNOTSUPP
);
1564 pru_sopoll_notsupp(__unused
struct socket
*so
, __unused
int events
,
1565 __unused kauth_cred_t cred
, __unused
void *wql
)
1567 return (EOPNOTSUPP
);
1573 * The following are macros on BSD and functions on Darwin
1577 * Do we need to notify the other side when I/O is possible?
1581 sb_notify(struct sockbuf
*sb
)
1583 return ((sb
->sb_flags
&
1584 (SB_WAIT
|SB_SEL
|SB_ASYNC
|SB_UPCALL
|SB_KNOTE
)) != 0);
1588 * How much space is there in a socket buffer (so->so_snd or so->so_rcv)?
1589 * This is problematical if the fields are unsigned, as the space might
1590 * still be negative (cc > hiwat or mbcnt > mbmax). Should detect
1591 * overflow and return 0.
1594 sbspace(struct sockbuf
*sb
)
1597 imin((int)(sb
->sb_hiwat
- sb
->sb_cc
),
1598 (int)(sb
->sb_mbmax
- sb
->sb_mbcnt
));
1605 /* do we have to send all at once on a socket? */
1607 sosendallatonce(struct socket
*so
)
1609 return (so
->so_proto
->pr_flags
& PR_ATOMIC
);
1612 /* can we read something from so? */
1614 soreadable(struct socket
*so
)
1616 return (so
->so_rcv
.sb_cc
>= so
->so_rcv
.sb_lowat
||
1617 (so
->so_state
& SS_CANTRCVMORE
) ||
1618 so
->so_comp
.tqh_first
|| so
->so_error
);
1621 /* can we write something to so? */
1624 sowriteable(struct socket
*so
)
1626 return ((sbspace(&(so
)->so_snd
) >= (so
)->so_snd
.sb_lowat
&&
1627 ((so
->so_state
&SS_ISCONNECTED
) ||
1628 (so
->so_proto
->pr_flags
&PR_CONNREQUIRED
) == 0)) ||
1629 (so
->so_state
& SS_CANTSENDMORE
) ||
1633 /* adjust counters in sb reflecting allocation of m */
1636 sballoc(struct sockbuf
*sb
, struct mbuf
*m
)
1639 sb
->sb_cc
+= m
->m_len
;
1640 if (m
->m_type
!= MT_DATA
&& m
->m_type
!= MT_HEADER
&&
1641 m
->m_type
!= MT_OOBDATA
)
1642 sb
->sb_ctl
+= m
->m_len
;
1643 sb
->sb_mbcnt
+= MSIZE
;
1645 if (m
->m_flags
& M_EXT
) {
1646 sb
->sb_mbcnt
+= m
->m_ext
.ext_size
;
1647 cnt
+= m
->m_ext
.ext_size
/ MSIZE
;
1649 OSAddAtomic(cnt
, &total_mb_cnt
);
1652 /* adjust counters in sb reflecting freeing of m */
1654 sbfree(struct sockbuf
*sb
, struct mbuf
*m
)
1657 sb
->sb_cc
-= m
->m_len
;
1658 if (m
->m_type
!= MT_DATA
&& m
->m_type
!= MT_HEADER
&&
1659 m
->m_type
!= MT_OOBDATA
)
1660 sb
->sb_ctl
-= m
->m_len
;
1661 sb
->sb_mbcnt
-= MSIZE
;
1662 if (m
->m_flags
& M_EXT
) {
1663 sb
->sb_mbcnt
-= m
->m_ext
.ext_size
;
1664 cnt
-= m
->m_ext
.ext_size
/ MSIZE
;
1666 OSAddAtomic(cnt
, &total_mb_cnt
);
1670 * Set lock on sockbuf sb; sleep if lock is already held.
1671 * Unless SB_NOINTR is set on sockbuf, sleep is interruptible.
1672 * Returns error without lock if sleep is interrupted.
1674 * Returns: 0 Success
1679 sblock(struct sockbuf
*sb
, int wf
)
1683 if (sb
->sb_flags
& SB_LOCK
)
1684 error
= (wf
== M_WAIT
) ? sb_lock(sb
) : EWOULDBLOCK
;
1686 sb
->sb_flags
|= SB_LOCK
;
1691 /* release lock on sockbuf sb */
1693 sbunlock(struct sockbuf
*sb
, int keeplocked
)
1695 struct socket
*so
= sb
->sb_so
;
1697 lck_mtx_t
*mutex_held
;
1699 lr_saved
= __builtin_return_address(0);
1701 sb
->sb_flags
&= ~SB_LOCK
;
1703 if (sb
->sb_flags
& SB_WANT
) {
1704 sb
->sb_flags
&= ~SB_WANT
;
1705 if (so
->so_usecount
< 0) {
1706 panic("sbunlock: b4 wakeup so=%p ref=%d lr=%p "
1707 "sb_flags=%x lrh= %s\n", sb
->sb_so
, so
->so_usecount
,
1708 lr_saved
, sb
->sb_flags
, solockhistory_nr(so
));
1711 wakeup((caddr_t
)&(sb
)->sb_flags
);
1713 if (keeplocked
== 0) { /* unlock on exit */
1714 if (so
->so_proto
->pr_getlock
!= NULL
)
1715 mutex_held
= (*so
->so_proto
->pr_getlock
)(so
, 0);
1717 mutex_held
= so
->so_proto
->pr_domain
->dom_mtx
;
1719 lck_mtx_assert(mutex_held
, LCK_MTX_ASSERT_OWNED
);
1722 if (so
->so_usecount
< 0)
1723 panic("sbunlock: unlock on exit so=%p ref=%d lr=%p "
1724 "sb_flags=%x lrh= %s\n", so
, so
->so_usecount
, lr_saved
,
1725 sb
->sb_flags
, solockhistory_nr(so
));
1726 so
->unlock_lr
[so
->next_unlock_lr
] = lr_saved
;
1727 so
->next_unlock_lr
= (so
->next_unlock_lr
+1) % SO_LCKDBG_MAX
;
1728 lck_mtx_unlock(mutex_held
);
1733 sorwakeup(struct socket
*so
)
1735 if (sb_notify(&so
->so_rcv
))
1736 sowakeup(so
, &so
->so_rcv
);
1740 sowwakeup(struct socket
*so
)
1742 if (sb_notify(&so
->so_snd
))
1743 sowakeup(so
, &so
->so_snd
);
1745 #endif /* __APPLE__ */
1748 * Make a copy of a sockaddr in a malloced buffer of type M_SONAME.
1751 dup_sockaddr(struct sockaddr
*sa
, int canwait
)
1753 struct sockaddr
*sa2
;
1755 MALLOC(sa2
, struct sockaddr
*, sa
->sa_len
, M_SONAME
,
1756 canwait
? M_WAITOK
: M_NOWAIT
);
1758 bcopy(sa
, sa2
, sa
->sa_len
);
1763 * Create an external-format (``xsocket'') structure using the information
1764 * in the kernel-format socket structure pointed to by so. This is done
1765 * to reduce the spew of irrelevant information over this interface,
1766 * to isolate user code from changes in the kernel structure, and
1767 * potentially to provide information-hiding if we decide that
1768 * some of this information should be hidden from users.
1771 sotoxsocket(struct socket
*so
, struct xsocket
*xso
)
1773 xso
->xso_len
= sizeof (*xso
);
1774 xso
->xso_so
= (_XSOCKET_PTR(struct socket
*))(uintptr_t)so
;
1775 xso
->so_type
= so
->so_type
;
1776 xso
->so_options
= so
->so_options
;
1777 xso
->so_linger
= so
->so_linger
;
1778 xso
->so_state
= so
->so_state
;
1779 xso
->so_pcb
= (_XSOCKET_PTR(caddr_t
))(uintptr_t)so
->so_pcb
;
1781 xso
->xso_protocol
= so
->so_proto
->pr_protocol
;
1782 xso
->xso_family
= so
->so_proto
->pr_domain
->dom_family
;
1784 xso
->xso_protocol
= xso
->xso_family
= 0;
1786 xso
->so_qlen
= so
->so_qlen
;
1787 xso
->so_incqlen
= so
->so_incqlen
;
1788 xso
->so_qlimit
= so
->so_qlimit
;
1789 xso
->so_timeo
= so
->so_timeo
;
1790 xso
->so_error
= so
->so_error
;
1791 xso
->so_pgid
= so
->so_pgid
;
1792 xso
->so_oobmark
= so
->so_oobmark
;
1793 sbtoxsockbuf(&so
->so_snd
, &xso
->so_snd
);
1794 sbtoxsockbuf(&so
->so_rcv
, &xso
->so_rcv
);
1795 xso
->so_uid
= so
->so_uid
;
1799 #if !CONFIG_EMBEDDED
1802 sotoxsocket64(struct socket
*so
, struct xsocket64
*xso
)
1804 xso
->xso_len
= sizeof (*xso
);
1805 xso
->xso_so
= (u_int64_t
)(uintptr_t)so
;
1806 xso
->so_type
= so
->so_type
;
1807 xso
->so_options
= so
->so_options
;
1808 xso
->so_linger
= so
->so_linger
;
1809 xso
->so_state
= so
->so_state
;
1810 xso
->so_pcb
= (u_int64_t
)(uintptr_t)so
->so_pcb
;
1812 xso
->xso_protocol
= so
->so_proto
->pr_protocol
;
1813 xso
->xso_family
= so
->so_proto
->pr_domain
->dom_family
;
1815 xso
->xso_protocol
= xso
->xso_family
= 0;
1817 xso
->so_qlen
= so
->so_qlen
;
1818 xso
->so_incqlen
= so
->so_incqlen
;
1819 xso
->so_qlimit
= so
->so_qlimit
;
1820 xso
->so_timeo
= so
->so_timeo
;
1821 xso
->so_error
= so
->so_error
;
1822 xso
->so_pgid
= so
->so_pgid
;
1823 xso
->so_oobmark
= so
->so_oobmark
;
1824 sbtoxsockbuf(&so
->so_snd
, &xso
->so_snd
);
1825 sbtoxsockbuf(&so
->so_rcv
, &xso
->so_rcv
);
1826 xso
->so_uid
= so
->so_uid
;
1829 #endif /* !CONFIG_EMBEDDED */
1832 * This does the same for sockbufs. Note that the xsockbuf structure,
1833 * since it is always embedded in a socket, does not include a self
1834 * pointer nor a length. We make this entry point public in case
1835 * some other mechanism needs it.
1838 sbtoxsockbuf(struct sockbuf
*sb
, struct xsockbuf
*xsb
)
1840 xsb
->sb_cc
= sb
->sb_cc
;
1841 xsb
->sb_hiwat
= sb
->sb_hiwat
;
1842 xsb
->sb_mbcnt
= sb
->sb_mbcnt
;
1843 xsb
->sb_mbmax
= sb
->sb_mbmax
;
1844 xsb
->sb_lowat
= sb
->sb_lowat
;
1845 xsb
->sb_flags
= sb
->sb_flags
;
1846 xsb
->sb_timeo
= (short)
1847 (sb
->sb_timeo
.tv_sec
* hz
) + sb
->sb_timeo
.tv_usec
/ tick
;
1848 if (xsb
->sb_timeo
== 0 && sb
->sb_timeo
.tv_usec
!= 0)
1853 soisbackground(struct socket
*so
)
1855 return (so
->so_traffic_mgt_flags
& TRAFFIC_MGT_SO_BACKGROUND
);
1859 * Here is the definition of some of the basic objects in the kern.ipc
1860 * branch of the MIB.
1862 SYSCTL_NODE(_kern
, KERN_IPC
, ipc
, CTLFLAG_RW
|CTLFLAG_LOCKED
, 0, "IPC");
1864 /* Check that the maximum socket buffer size is within a range */
1867 sysctl_sb_max(__unused
struct sysctl_oid
*oidp
, __unused
void *arg1
,
1868 __unused
int arg2
, struct sysctl_req
*req
)
1870 u_int32_t new_value
;
1872 int error
= sysctl_io_number(req
, sb_max
, sizeof(u_int32_t
), &new_value
,
1874 if (!error
&& changed
) {
1875 if (new_value
> LOW_SB_MAX
&&
1876 new_value
<= high_sb_max
) {
1885 SYSCTL_PROC(_kern_ipc
, KIPC_MAXSOCKBUF
, maxsockbuf
, CTLTYPE_INT
| CTLFLAG_RW
,
1886 &sb_max
, 0, &sysctl_sb_max
, "IU", "Maximum socket buffer size");
1888 SYSCTL_INT(_kern_ipc
, OID_AUTO
, maxsockets
, CTLFLAG_RD
,
1889 &maxsockets
, 0, "Maximum number of sockets avaliable");
1890 SYSCTL_INT(_kern_ipc
, KIPC_SOCKBUF_WASTE
, sockbuf_waste_factor
, CTLFLAG_RW
,
1891 &sb_efficiency
, 0, "");
1892 SYSCTL_INT(_kern_ipc
, OID_AUTO
, sbspace_factor
, CTLFLAG_RW
,
1893 &sbspace_factor
, 0, "Ratio of mbuf/cluster use for socket layers");
1894 SYSCTL_INT(_kern_ipc
, KIPC_NMBCLUSTERS
, nmbclusters
, CTLFLAG_RD
,
1895 &nmbclusters
, 0, "");
1896 SYSCTL_INT(_kern_ipc
, OID_AUTO
, njcl
, CTLFLAG_RD
, &njcl
, 0, "");
1897 SYSCTL_INT(_kern_ipc
, OID_AUTO
, njclbytes
, CTLFLAG_RD
, &njclbytes
, 0, "");
1898 SYSCTL_INT(_kern_ipc
, KIPC_SOQLIMITCOMPAT
, soqlimitcompat
, CTLFLAG_RW
,
1899 &soqlimitcompat
, 1, "Enable socket queue limit compatibility");
1900 SYSCTL_INT(_kern_ipc
, OID_AUTO
, soqlencomp
, CTLFLAG_RW
,
1901 &soqlencomp
, 0, "Listen backlog represents only complete queue");