xref: /xnu-10002.61.3/bsd/nfs/nfs_socket.c (revision 0f4c859e951fba394238ab619495c4e1d54d0f34)
1*0f4c859eSApple OSS Distributions /*
2*0f4c859eSApple OSS Distributions  * Copyright (c) 2000-2020 Apple Inc. All rights reserved.
3*0f4c859eSApple OSS Distributions  *
4*0f4c859eSApple OSS Distributions  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5*0f4c859eSApple OSS Distributions  *
6*0f4c859eSApple OSS Distributions  * This file contains Original Code and/or Modifications of Original Code
7*0f4c859eSApple OSS Distributions  * as defined in and that are subject to the Apple Public Source License
8*0f4c859eSApple OSS Distributions  * Version 2.0 (the 'License'). You may not use this file except in
9*0f4c859eSApple OSS Distributions  * compliance with the License. The rights granted to you under the License
10*0f4c859eSApple OSS Distributions  * may not be used to create, or enable the creation or redistribution of,
11*0f4c859eSApple OSS Distributions  * unlawful or unlicensed copies of an Apple operating system, or to
12*0f4c859eSApple OSS Distributions  * circumvent, violate, or enable the circumvention or violation of, any
13*0f4c859eSApple OSS Distributions  * terms of an Apple operating system software license agreement.
14*0f4c859eSApple OSS Distributions  *
15*0f4c859eSApple OSS Distributions  * Please obtain a copy of the License at
16*0f4c859eSApple OSS Distributions  * http://www.opensource.apple.com/apsl/ and read it before using this file.
17*0f4c859eSApple OSS Distributions  *
18*0f4c859eSApple OSS Distributions  * The Original Code and all software distributed under the License are
19*0f4c859eSApple OSS Distributions  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20*0f4c859eSApple OSS Distributions  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21*0f4c859eSApple OSS Distributions  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22*0f4c859eSApple OSS Distributions  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23*0f4c859eSApple OSS Distributions  * Please see the License for the specific language governing rights and
24*0f4c859eSApple OSS Distributions  * limitations under the License.
25*0f4c859eSApple OSS Distributions  *
26*0f4c859eSApple OSS Distributions  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27*0f4c859eSApple OSS Distributions  */
28*0f4c859eSApple OSS Distributions /* Copyright (c) 1995 NeXT Computer, Inc. All Rights Reserved */
29*0f4c859eSApple OSS Distributions /*
30*0f4c859eSApple OSS Distributions  * Copyright (c) 1989, 1991, 1993, 1995
31*0f4c859eSApple OSS Distributions  *	The Regents of the University of California.  All rights reserved.
32*0f4c859eSApple OSS Distributions  *
33*0f4c859eSApple OSS Distributions  * This code is derived from software contributed to Berkeley by
34*0f4c859eSApple OSS Distributions  * Rick Macklem at The University of Guelph.
35*0f4c859eSApple OSS Distributions  *
36*0f4c859eSApple OSS Distributions  * Redistribution and use in source and binary forms, with or without
37*0f4c859eSApple OSS Distributions  * modification, are permitted provided that the following conditions
38*0f4c859eSApple OSS Distributions  * are met:
39*0f4c859eSApple OSS Distributions  * 1. Redistributions of source code must retain the above copyright
40*0f4c859eSApple OSS Distributions  *    notice, this list of conditions and the following disclaimer.
41*0f4c859eSApple OSS Distributions  * 2. Redistributions in binary form must reproduce the above copyright
42*0f4c859eSApple OSS Distributions  *    notice, this list of conditions and the following disclaimer in the
43*0f4c859eSApple OSS Distributions  *    documentation and/or other materials provided with the distribution.
44*0f4c859eSApple OSS Distributions  * 3. All advertising materials mentioning features or use of this software
45*0f4c859eSApple OSS Distributions  *    must display the following acknowledgement:
46*0f4c859eSApple OSS Distributions  *	This product includes software developed by the University of
47*0f4c859eSApple OSS Distributions  *	California, Berkeley and its contributors.
48*0f4c859eSApple OSS Distributions  * 4. Neither the name of the University nor the names of its contributors
49*0f4c859eSApple OSS Distributions  *    may be used to endorse or promote products derived from this software
50*0f4c859eSApple OSS Distributions  *    without specific prior written permission.
51*0f4c859eSApple OSS Distributions  *
52*0f4c859eSApple OSS Distributions  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
53*0f4c859eSApple OSS Distributions  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
54*0f4c859eSApple OSS Distributions  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
55*0f4c859eSApple OSS Distributions  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
56*0f4c859eSApple OSS Distributions  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
57*0f4c859eSApple OSS Distributions  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
58*0f4c859eSApple OSS Distributions  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
59*0f4c859eSApple OSS Distributions  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
60*0f4c859eSApple OSS Distributions  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
61*0f4c859eSApple OSS Distributions  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
62*0f4c859eSApple OSS Distributions  * SUCH DAMAGE.
63*0f4c859eSApple OSS Distributions  *
64*0f4c859eSApple OSS Distributions  *	@(#)nfs_socket.c	8.5 (Berkeley) 3/30/95
65*0f4c859eSApple OSS Distributions  * FreeBSD-Id: nfs_socket.c,v 1.30 1997/10/28 15:59:07 bde Exp $
66*0f4c859eSApple OSS Distributions  */
67*0f4c859eSApple OSS Distributions 
68*0f4c859eSApple OSS Distributions #include <nfs/nfs_conf.h>
69*0f4c859eSApple OSS Distributions #if CONFIG_NFS_SERVER
70*0f4c859eSApple OSS Distributions 
71*0f4c859eSApple OSS Distributions /*
72*0f4c859eSApple OSS Distributions  * Socket operations for use by nfs
73*0f4c859eSApple OSS Distributions  */
74*0f4c859eSApple OSS Distributions 
75*0f4c859eSApple OSS Distributions #include <sys/systm.h>
76*0f4c859eSApple OSS Distributions #include <sys/kauth.h>
77*0f4c859eSApple OSS Distributions #include <sys/mount_internal.h>
78*0f4c859eSApple OSS Distributions #include <sys/kpi_mbuf.h>
79*0f4c859eSApple OSS Distributions #include <IOKit/IOLib.h>
80*0f4c859eSApple OSS Distributions 
81*0f4c859eSApple OSS Distributions #include <netinet/in.h>
82*0f4c859eSApple OSS Distributions 
83*0f4c859eSApple OSS Distributions #include <nfs/rpcv2.h>
84*0f4c859eSApple OSS Distributions #include <nfs/nfsproto.h>
85*0f4c859eSApple OSS Distributions #include <nfs/nfs.h>
86*0f4c859eSApple OSS Distributions #include <nfs/xdr_subs.h>
87*0f4c859eSApple OSS Distributions #include <nfs/nfsm_subs.h>
88*0f4c859eSApple OSS Distributions #include <nfs/nfs_gss.h>
89*0f4c859eSApple OSS Distributions 
90*0f4c859eSApple OSS Distributions ZONE_DEFINE(nfsrv_descript_zone, "NFSV3 srvdesc",
91*0f4c859eSApple OSS Distributions     sizeof(struct nfsrv_descript), ZC_NONE);
92*0f4c859eSApple OSS Distributions 
93*0f4c859eSApple OSS Distributions int nfsrv_sock_max_rec_queue_length = 128; /* max # RPC records queued on (UDP) socket */
94*0f4c859eSApple OSS Distributions 
95*0f4c859eSApple OSS Distributions uint32_t nfsrv_unprocessed_rpc_current = 0; /* Current bytes of unprocessed RPC records */
96*0f4c859eSApple OSS Distributions uint32_t nfsrv_unprocessed_rpc_max = (64 * 1024 * 1024); /* Max bytes of unprocessed RPC records - 64MB by default */
97*0f4c859eSApple OSS Distributions 
98*0f4c859eSApple OSS Distributions int nfsrv_getstream(struct nfsrv_sock *, int);
99*0f4c859eSApple OSS Distributions int nfsrv_getreq(struct nfsrv_descript *);
100*0f4c859eSApple OSS Distributions extern int nfsv3_procid[NFS_NPROCS];
101*0f4c859eSApple OSS Distributions 
102*0f4c859eSApple OSS Distributions #define NFS_TRYLOCK_MSEC_SLEEP 1
103*0f4c859eSApple OSS Distributions 
104*0f4c859eSApple OSS Distributions /*
105*0f4c859eSApple OSS Distributions  * compare two sockaddr structures
106*0f4c859eSApple OSS Distributions  */
107*0f4c859eSApple OSS Distributions int
nfs_sockaddr_cmp(struct sockaddr * sa1,struct sockaddr * sa2)108*0f4c859eSApple OSS Distributions nfs_sockaddr_cmp(struct sockaddr *sa1, struct sockaddr *sa2)
109*0f4c859eSApple OSS Distributions {
110*0f4c859eSApple OSS Distributions 	if (!sa1) {
111*0f4c859eSApple OSS Distributions 		return -1;
112*0f4c859eSApple OSS Distributions 	}
113*0f4c859eSApple OSS Distributions 	if (!sa2) {
114*0f4c859eSApple OSS Distributions 		return 1;
115*0f4c859eSApple OSS Distributions 	}
116*0f4c859eSApple OSS Distributions 	if (sa1->sa_family != sa2->sa_family) {
117*0f4c859eSApple OSS Distributions 		return (sa1->sa_family < sa2->sa_family) ? -1 : 1;
118*0f4c859eSApple OSS Distributions 	}
119*0f4c859eSApple OSS Distributions 	if (sa1->sa_len != sa2->sa_len) {
120*0f4c859eSApple OSS Distributions 		return (sa1->sa_len < sa2->sa_len) ? -1 : 1;
121*0f4c859eSApple OSS Distributions 	}
122*0f4c859eSApple OSS Distributions 	if (sa1->sa_family == AF_INET) {
123*0f4c859eSApple OSS Distributions 		return bcmp(&((struct sockaddr_in*)sa1)->sin_addr,
124*0f4c859eSApple OSS Distributions 		           &((struct sockaddr_in*)sa2)->sin_addr, sizeof(((struct sockaddr_in*)sa1)->sin_addr));
125*0f4c859eSApple OSS Distributions 	}
126*0f4c859eSApple OSS Distributions 	if (sa1->sa_family == AF_INET6) {
127*0f4c859eSApple OSS Distributions 		return bcmp(&((struct sockaddr_in6*)sa1)->sin6_addr,
128*0f4c859eSApple OSS Distributions 		           &((struct sockaddr_in6*)sa2)->sin6_addr, sizeof(((struct sockaddr_in6*)sa1)->sin6_addr));
129*0f4c859eSApple OSS Distributions 	}
130*0f4c859eSApple OSS Distributions 	return -1;
131*0f4c859eSApple OSS Distributions }
132*0f4c859eSApple OSS Distributions 
133*0f4c859eSApple OSS Distributions /*
134*0f4c859eSApple OSS Distributions  * Generate the rpc reply header
135*0f4c859eSApple OSS Distributions  * siz arg. is used to decide if adding a cluster is worthwhile
136*0f4c859eSApple OSS Distributions  */
137*0f4c859eSApple OSS Distributions int
nfsrv_rephead(struct nfsrv_descript * nd,__unused struct nfsrv_sock * slp,struct nfsm_chain * nmrepp,size_t siz)138*0f4c859eSApple OSS Distributions nfsrv_rephead(
139*0f4c859eSApple OSS Distributions 	struct nfsrv_descript *nd,
140*0f4c859eSApple OSS Distributions 	__unused struct nfsrv_sock *slp,
141*0f4c859eSApple OSS Distributions 	struct nfsm_chain *nmrepp,
142*0f4c859eSApple OSS Distributions 	size_t siz)
143*0f4c859eSApple OSS Distributions {
144*0f4c859eSApple OSS Distributions 	mbuf_t mrep;
145*0f4c859eSApple OSS Distributions 	u_int32_t *tl;
146*0f4c859eSApple OSS Distributions 	struct nfsm_chain nmrep;
147*0f4c859eSApple OSS Distributions 	int err, error;
148*0f4c859eSApple OSS Distributions 
149*0f4c859eSApple OSS Distributions 	err = nd->nd_repstat;
150*0f4c859eSApple OSS Distributions 	if (err && (nd->nd_vers == NFS_VER2)) {
151*0f4c859eSApple OSS Distributions 		siz = 0;
152*0f4c859eSApple OSS Distributions 	}
153*0f4c859eSApple OSS Distributions 
154*0f4c859eSApple OSS Distributions 	/*
155*0f4c859eSApple OSS Distributions 	 * If this is a big reply, use a cluster else
156*0f4c859eSApple OSS Distributions 	 * try and leave leading space for the lower level headers.
157*0f4c859eSApple OSS Distributions 	 */
158*0f4c859eSApple OSS Distributions 	siz += RPC_REPLYSIZ;
159*0f4c859eSApple OSS Distributions 	if (siz >= nfs_mbuf_minclsize) {
160*0f4c859eSApple OSS Distributions 		error = mbuf_getpacket(MBUF_WAITOK, &mrep);
161*0f4c859eSApple OSS Distributions 	} else {
162*0f4c859eSApple OSS Distributions 		error = mbuf_gethdr(MBUF_WAITOK, MBUF_TYPE_DATA, &mrep);
163*0f4c859eSApple OSS Distributions 	}
164*0f4c859eSApple OSS Distributions 	if (error) {
165*0f4c859eSApple OSS Distributions 		/* unable to allocate packet */
166*0f4c859eSApple OSS Distributions 		/* XXX should we keep statistics for these errors? */
167*0f4c859eSApple OSS Distributions 		return error;
168*0f4c859eSApple OSS Distributions 	}
169*0f4c859eSApple OSS Distributions 	if (siz < nfs_mbuf_minclsize) {
170*0f4c859eSApple OSS Distributions 		/* leave space for lower level headers */
171*0f4c859eSApple OSS Distributions 		tl = mbuf_data(mrep);
172*0f4c859eSApple OSS Distributions 		tl += 80 / sizeof(*tl);  /* XXX max_hdr? XXX */
173*0f4c859eSApple OSS Distributions 		mbuf_setdata(mrep, tl, 6 * NFSX_UNSIGNED);
174*0f4c859eSApple OSS Distributions 	}
175*0f4c859eSApple OSS Distributions 	nfsm_chain_init(&nmrep, mrep);
176*0f4c859eSApple OSS Distributions 	nfsm_chain_add_32(error, &nmrep, nd->nd_retxid);
177*0f4c859eSApple OSS Distributions 	nfsm_chain_add_32(error, &nmrep, RPC_REPLY);
178*0f4c859eSApple OSS Distributions 	if (err == ERPCMISMATCH || (err & NFSERR_AUTHERR)) {
179*0f4c859eSApple OSS Distributions 		nfsm_chain_add_32(error, &nmrep, RPC_MSGDENIED);
180*0f4c859eSApple OSS Distributions 		if (err & NFSERR_AUTHERR) {
181*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_AUTHERR);
182*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, (err & ~NFSERR_AUTHERR));
183*0f4c859eSApple OSS Distributions 		} else {
184*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_MISMATCH);
185*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_VER2);
186*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_VER2);
187*0f4c859eSApple OSS Distributions 		}
188*0f4c859eSApple OSS Distributions 	} else {
189*0f4c859eSApple OSS Distributions 		/* reply status */
190*0f4c859eSApple OSS Distributions 		nfsm_chain_add_32(error, &nmrep, RPC_MSGACCEPTED);
191*0f4c859eSApple OSS Distributions 		if (nd->nd_gss_context != NULL) {
192*0f4c859eSApple OSS Distributions 			/* RPCSEC_GSS verifier */
193*0f4c859eSApple OSS Distributions 			error = nfs_gss_svc_verf_put(nd, &nmrep);
194*0f4c859eSApple OSS Distributions 			if (error) {
195*0f4c859eSApple OSS Distributions 				nfsm_chain_add_32(error, &nmrep, RPC_SYSTEM_ERR);
196*0f4c859eSApple OSS Distributions 				goto done;
197*0f4c859eSApple OSS Distributions 			}
198*0f4c859eSApple OSS Distributions 		} else {
199*0f4c859eSApple OSS Distributions 			/* RPCAUTH_NULL verifier */
200*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPCAUTH_NULL);
201*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, 0);
202*0f4c859eSApple OSS Distributions 		}
203*0f4c859eSApple OSS Distributions 		/* accepted status */
204*0f4c859eSApple OSS Distributions 		switch (err) {
205*0f4c859eSApple OSS Distributions 		case EPROGUNAVAIL:
206*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_PROGUNAVAIL);
207*0f4c859eSApple OSS Distributions 			break;
208*0f4c859eSApple OSS Distributions 		case EPROGMISMATCH:
209*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_PROGMISMATCH);
210*0f4c859eSApple OSS Distributions 			/* XXX hard coded versions? */
211*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, NFS_VER2);
212*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, NFS_VER3);
213*0f4c859eSApple OSS Distributions 			break;
214*0f4c859eSApple OSS Distributions 		case EPROCUNAVAIL:
215*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_PROCUNAVAIL);
216*0f4c859eSApple OSS Distributions 			break;
217*0f4c859eSApple OSS Distributions 		case EBADRPC:
218*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_GARBAGE);
219*0f4c859eSApple OSS Distributions 			break;
220*0f4c859eSApple OSS Distributions 		default:
221*0f4c859eSApple OSS Distributions 			nfsm_chain_add_32(error, &nmrep, RPC_SUCCESS);
222*0f4c859eSApple OSS Distributions 			if (nd->nd_gss_context != NULL) {
223*0f4c859eSApple OSS Distributions 				error = nfs_gss_svc_prepare_reply(nd, &nmrep);
224*0f4c859eSApple OSS Distributions 			}
225*0f4c859eSApple OSS Distributions 			if (err != NFSERR_RETVOID) {
226*0f4c859eSApple OSS Distributions 				nfsm_chain_add_32(error, &nmrep,
227*0f4c859eSApple OSS Distributions 				    (err ? nfsrv_errmap(nd, err) : 0));
228*0f4c859eSApple OSS Distributions 			}
229*0f4c859eSApple OSS Distributions 			break;
230*0f4c859eSApple OSS Distributions 		}
231*0f4c859eSApple OSS Distributions 	}
232*0f4c859eSApple OSS Distributions 
233*0f4c859eSApple OSS Distributions done:
234*0f4c859eSApple OSS Distributions 	nfsm_chain_build_done(error, &nmrep);
235*0f4c859eSApple OSS Distributions 	if (error) {
236*0f4c859eSApple OSS Distributions 		/* error composing reply header */
237*0f4c859eSApple OSS Distributions 		/* XXX should we keep statistics for these errors? */
238*0f4c859eSApple OSS Distributions 		mbuf_freem(mrep);
239*0f4c859eSApple OSS Distributions 		return error;
240*0f4c859eSApple OSS Distributions 	}
241*0f4c859eSApple OSS Distributions 
242*0f4c859eSApple OSS Distributions 	*nmrepp = nmrep;
243*0f4c859eSApple OSS Distributions 	if ((err != 0) && (err != NFSERR_RETVOID)) {
244*0f4c859eSApple OSS Distributions 		OSAddAtomic64(1, &nfsrvstats.srvrpc_errs);
245*0f4c859eSApple OSS Distributions 	}
246*0f4c859eSApple OSS Distributions 	return 0;
247*0f4c859eSApple OSS Distributions }
248*0f4c859eSApple OSS Distributions 
249*0f4c859eSApple OSS Distributions /*
250*0f4c859eSApple OSS Distributions  * The nfs server send routine.
251*0f4c859eSApple OSS Distributions  *
252*0f4c859eSApple OSS Distributions  * - return EINTR or ERESTART if interrupted by a signal
253*0f4c859eSApple OSS Distributions  * - return EPIPE if a connection is lost for connection based sockets (TCP...)
254*0f4c859eSApple OSS Distributions  * - do any cleanup required by recoverable socket errors (???)
255*0f4c859eSApple OSS Distributions  */
256*0f4c859eSApple OSS Distributions int
nfsrv_send(struct nfsrv_sock * slp,mbuf_t nam,mbuf_t top)257*0f4c859eSApple OSS Distributions nfsrv_send(struct nfsrv_sock *slp, mbuf_t nam, mbuf_t top)
258*0f4c859eSApple OSS Distributions {
259*0f4c859eSApple OSS Distributions 	int error;
260*0f4c859eSApple OSS Distributions 	socket_t so = slp->ns_so;
261*0f4c859eSApple OSS Distributions 	struct sockaddr *sendnam;
262*0f4c859eSApple OSS Distributions 	struct msghdr msg;
263*0f4c859eSApple OSS Distributions 
264*0f4c859eSApple OSS Distributions 	bzero(&msg, sizeof(msg));
265*0f4c859eSApple OSS Distributions 	if (nam && !sock_isconnected(so) && (slp->ns_sotype != SOCK_STREAM)) {
266*0f4c859eSApple OSS Distributions 		if ((sendnam = mbuf_data(nam))) {
267*0f4c859eSApple OSS Distributions 			msg.msg_name = (caddr_t)sendnam;
268*0f4c859eSApple OSS Distributions 			msg.msg_namelen = sendnam->sa_len;
269*0f4c859eSApple OSS Distributions 		}
270*0f4c859eSApple OSS Distributions 	}
271*0f4c859eSApple OSS Distributions 	if (NFSRV_IS_DBG(NFSRV_FAC_SRV, 15)) {
272*0f4c859eSApple OSS Distributions 		nfs_dump_mbuf(__func__, __LINE__, "nfsrv_send\n", top);
273*0f4c859eSApple OSS Distributions 	}
274*0f4c859eSApple OSS Distributions 	error = sock_sendmbuf(so, &msg, top, 0, NULL);
275*0f4c859eSApple OSS Distributions 	if (!error) {
276*0f4c859eSApple OSS Distributions 		return 0;
277*0f4c859eSApple OSS Distributions 	}
278*0f4c859eSApple OSS Distributions 	log(LOG_INFO, "nfsd send error %d\n", error);
279*0f4c859eSApple OSS Distributions 
280*0f4c859eSApple OSS Distributions 	if ((error == EWOULDBLOCK) && (slp->ns_sotype == SOCK_STREAM)) {
281*0f4c859eSApple OSS Distributions 		error = EPIPE;  /* zap TCP sockets if they time out on send */
282*0f4c859eSApple OSS Distributions 	}
283*0f4c859eSApple OSS Distributions 	/* Handle any recoverable (soft) socket errors here. (???) */
284*0f4c859eSApple OSS Distributions 	if (error != EINTR && error != ERESTART && error != EIO &&
285*0f4c859eSApple OSS Distributions 	    error != EWOULDBLOCK && error != EPIPE) {
286*0f4c859eSApple OSS Distributions 		error = 0;
287*0f4c859eSApple OSS Distributions 	}
288*0f4c859eSApple OSS Distributions 
289*0f4c859eSApple OSS Distributions 	return error;
290*0f4c859eSApple OSS Distributions }
291*0f4c859eSApple OSS Distributions 
292*0f4c859eSApple OSS Distributions /*
293*0f4c859eSApple OSS Distributions  * Socket upcall routine for the nfsd sockets.
294*0f4c859eSApple OSS Distributions  * The caddr_t arg is a pointer to the "struct nfsrv_sock".
295*0f4c859eSApple OSS Distributions  * Essentially do as much as possible non-blocking, else punt and it will
296*0f4c859eSApple OSS Distributions  * be called with MBUF_WAITOK from an nfsd.
297*0f4c859eSApple OSS Distributions  */
298*0f4c859eSApple OSS Distributions void
nfsrv_rcv(socket_t so,void * arg,int waitflag)299*0f4c859eSApple OSS Distributions nfsrv_rcv(socket_t so, void *arg, int waitflag)
300*0f4c859eSApple OSS Distributions {
301*0f4c859eSApple OSS Distributions 	struct nfsrv_sock *slp = arg;
302*0f4c859eSApple OSS Distributions 
303*0f4c859eSApple OSS Distributions 	while (1) {
304*0f4c859eSApple OSS Distributions 		if (!nfsd_thread_count || !(slp->ns_flag & SLP_VALID)) {
305*0f4c859eSApple OSS Distributions 			return;
306*0f4c859eSApple OSS Distributions 		}
307*0f4c859eSApple OSS Distributions 		if (lck_rw_try_lock_exclusive(&slp->ns_rwlock)) {
308*0f4c859eSApple OSS Distributions 			/* Exclusive lock acquired */
309*0f4c859eSApple OSS Distributions 			break;
310*0f4c859eSApple OSS Distributions 		}
311*0f4c859eSApple OSS Distributions 		IOSleep(NFS_TRYLOCK_MSEC_SLEEP);
312*0f4c859eSApple OSS Distributions 	}
313*0f4c859eSApple OSS Distributions 
314*0f4c859eSApple OSS Distributions 	nfsrv_rcv_locked(so, slp, waitflag);
315*0f4c859eSApple OSS Distributions 	/* Note: ns_rwlock gets dropped when called with MBUF_DONTWAIT */
316*0f4c859eSApple OSS Distributions }
317*0f4c859eSApple OSS Distributions void
nfsrv_rcv_locked(socket_t so,struct nfsrv_sock * slp,int waitflag)318*0f4c859eSApple OSS Distributions nfsrv_rcv_locked(socket_t so, struct nfsrv_sock *slp, int waitflag)
319*0f4c859eSApple OSS Distributions {
320*0f4c859eSApple OSS Distributions 	mbuf_t m, mp, mhck, m2;
321*0f4c859eSApple OSS Distributions 	int ns_flag = 0, error;
322*0f4c859eSApple OSS Distributions 	struct msghdr   msg;
323*0f4c859eSApple OSS Distributions 	size_t bytes_read;
324*0f4c859eSApple OSS Distributions 
325*0f4c859eSApple OSS Distributions 	if ((slp->ns_flag & SLP_VALID) == 0) {
326*0f4c859eSApple OSS Distributions 		if (waitflag == MBUF_DONTWAIT) {
327*0f4c859eSApple OSS Distributions 			lck_rw_done(&slp->ns_rwlock);
328*0f4c859eSApple OSS Distributions 		}
329*0f4c859eSApple OSS Distributions 		return;
330*0f4c859eSApple OSS Distributions 	}
331*0f4c859eSApple OSS Distributions 
332*0f4c859eSApple OSS Distributions #ifdef notdef
333*0f4c859eSApple OSS Distributions 	/*
334*0f4c859eSApple OSS Distributions 	 * Define this to test for nfsds handling this under heavy load.
335*0f4c859eSApple OSS Distributions 	 */
336*0f4c859eSApple OSS Distributions 	if (waitflag == MBUF_DONTWAIT) {
337*0f4c859eSApple OSS Distributions 		ns_flag = SLP_NEEDQ;
338*0f4c859eSApple OSS Distributions 		goto dorecs;
339*0f4c859eSApple OSS Distributions 	}
340*0f4c859eSApple OSS Distributions #endif
341*0f4c859eSApple OSS Distributions 	if (slp->ns_sotype == SOCK_STREAM) {
342*0f4c859eSApple OSS Distributions 		/*
343*0f4c859eSApple OSS Distributions 		 * If there are already records on the queue, defer soreceive()
344*0f4c859eSApple OSS Distributions 		 * to an(other) nfsd so that there is feedback to the TCP layer that
345*0f4c859eSApple OSS Distributions 		 * the nfs servers are heavily loaded.
346*0f4c859eSApple OSS Distributions 		 */
347*0f4c859eSApple OSS Distributions 		if (slp->ns_rec) {
348*0f4c859eSApple OSS Distributions 			ns_flag = SLP_NEEDQ;
349*0f4c859eSApple OSS Distributions 			goto dorecs;
350*0f4c859eSApple OSS Distributions 		}
351*0f4c859eSApple OSS Distributions 
352*0f4c859eSApple OSS Distributions 		/*
353*0f4c859eSApple OSS Distributions 		 * Do soreceive().
354*0f4c859eSApple OSS Distributions 		 */
355*0f4c859eSApple OSS Distributions 		bytes_read = 1000000000;
356*0f4c859eSApple OSS Distributions 		error = sock_receivembuf(so, NULL, &mp, MSG_DONTWAIT, &bytes_read);
357*0f4c859eSApple OSS Distributions 		if (error || mp == NULL) {
358*0f4c859eSApple OSS Distributions 			if (error == EWOULDBLOCK) {
359*0f4c859eSApple OSS Distributions 				ns_flag = (waitflag == MBUF_DONTWAIT) ? SLP_NEEDQ : 0;
360*0f4c859eSApple OSS Distributions 			} else {
361*0f4c859eSApple OSS Distributions 				ns_flag = SLP_DISCONN;
362*0f4c859eSApple OSS Distributions 			}
363*0f4c859eSApple OSS Distributions 			goto dorecs;
364*0f4c859eSApple OSS Distributions 		}
365*0f4c859eSApple OSS Distributions 		m = mp;
366*0f4c859eSApple OSS Distributions 		if (slp->ns_rawend) {
367*0f4c859eSApple OSS Distributions 			if ((error = mbuf_setnext(slp->ns_rawend, m))) {
368*0f4c859eSApple OSS Distributions 				panic("nfsrv_rcv: mbuf_setnext failed %d", error);
369*0f4c859eSApple OSS Distributions 			}
370*0f4c859eSApple OSS Distributions 			slp->ns_cc += bytes_read;
371*0f4c859eSApple OSS Distributions 		} else {
372*0f4c859eSApple OSS Distributions 			slp->ns_raw = m;
373*0f4c859eSApple OSS Distributions 			slp->ns_cc = bytes_read;
374*0f4c859eSApple OSS Distributions 		}
375*0f4c859eSApple OSS Distributions 		while ((m2 = mbuf_next(m))) {
376*0f4c859eSApple OSS Distributions 			m = m2;
377*0f4c859eSApple OSS Distributions 		}
378*0f4c859eSApple OSS Distributions 		slp->ns_rawend = m;
379*0f4c859eSApple OSS Distributions 
380*0f4c859eSApple OSS Distributions 		/*
381*0f4c859eSApple OSS Distributions 		 * Now try and parse record(s) out of the raw stream data.
382*0f4c859eSApple OSS Distributions 		 */
383*0f4c859eSApple OSS Distributions 		error = nfsrv_getstream(slp, waitflag);
384*0f4c859eSApple OSS Distributions 		if (error) {
385*0f4c859eSApple OSS Distributions 			if (error == EWOULDBLOCK) {
386*0f4c859eSApple OSS Distributions 				ns_flag = SLP_NEEDQ;
387*0f4c859eSApple OSS Distributions 			} else {
388*0f4c859eSApple OSS Distributions 				ns_flag = SLP_DISCONN;
389*0f4c859eSApple OSS Distributions 			}
390*0f4c859eSApple OSS Distributions 		}
391*0f4c859eSApple OSS Distributions 	} else {
392*0f4c859eSApple OSS Distributions 		struct sockaddr_storage nam;
393*0f4c859eSApple OSS Distributions 
394*0f4c859eSApple OSS Distributions 		if (slp->ns_reccnt >= nfsrv_sock_max_rec_queue_length) {
395*0f4c859eSApple OSS Distributions 			/* already have max # RPC records queued on this socket */
396*0f4c859eSApple OSS Distributions 			ns_flag = SLP_NEEDQ;
397*0f4c859eSApple OSS Distributions 			goto dorecs;
398*0f4c859eSApple OSS Distributions 		}
399*0f4c859eSApple OSS Distributions 
400*0f4c859eSApple OSS Distributions 		bzero(&msg, sizeof(msg));
401*0f4c859eSApple OSS Distributions 		msg.msg_name = (caddr_t)&nam;
402*0f4c859eSApple OSS Distributions 		msg.msg_namelen = sizeof(nam);
403*0f4c859eSApple OSS Distributions 
404*0f4c859eSApple OSS Distributions 		do {
405*0f4c859eSApple OSS Distributions 			bytes_read = 1000000000;
406*0f4c859eSApple OSS Distributions 			error = sock_receivembuf(so, &msg, &mp, MSG_DONTWAIT | MSG_NEEDSA, &bytes_read);
407*0f4c859eSApple OSS Distributions 			if (mp) {
408*0f4c859eSApple OSS Distributions 				if (msg.msg_name && (mbuf_get(MBUF_WAITOK, MBUF_TYPE_SONAME, &mhck) == 0)) {
409*0f4c859eSApple OSS Distributions 					mbuf_setlen(mhck, nam.ss_len);
410*0f4c859eSApple OSS Distributions 					bcopy(&nam, mbuf_data(mhck), nam.ss_len);
411*0f4c859eSApple OSS Distributions 					m = mhck;
412*0f4c859eSApple OSS Distributions 					if (mbuf_setnext(m, mp)) {
413*0f4c859eSApple OSS Distributions 						/* trouble... just drop it */
414*0f4c859eSApple OSS Distributions 						printf("nfsrv_rcv: mbuf_setnext failed\n");
415*0f4c859eSApple OSS Distributions 						mbuf_free(mhck);
416*0f4c859eSApple OSS Distributions 						m = mp;
417*0f4c859eSApple OSS Distributions 					}
418*0f4c859eSApple OSS Distributions 				} else {
419*0f4c859eSApple OSS Distributions 					m = mp;
420*0f4c859eSApple OSS Distributions 				}
421*0f4c859eSApple OSS Distributions 				if (slp->ns_recend) {
422*0f4c859eSApple OSS Distributions 					mbuf_setnextpkt(slp->ns_recend, m);
423*0f4c859eSApple OSS Distributions 				} else {
424*0f4c859eSApple OSS Distributions 					slp->ns_rec = m;
425*0f4c859eSApple OSS Distributions 					slp->ns_flag |= SLP_DOREC;
426*0f4c859eSApple OSS Distributions 				}
427*0f4c859eSApple OSS Distributions 				slp->ns_recend = m;
428*0f4c859eSApple OSS Distributions 				mbuf_setnextpkt(m, NULL);
429*0f4c859eSApple OSS Distributions 				slp->ns_reccnt++;
430*0f4c859eSApple OSS Distributions 			}
431*0f4c859eSApple OSS Distributions 		} while (mp);
432*0f4c859eSApple OSS Distributions 	}
433*0f4c859eSApple OSS Distributions 
434*0f4c859eSApple OSS Distributions 	/*
435*0f4c859eSApple OSS Distributions 	 * Now try and process the request records, non-blocking.
436*0f4c859eSApple OSS Distributions 	 */
437*0f4c859eSApple OSS Distributions dorecs:
438*0f4c859eSApple OSS Distributions 	if (ns_flag) {
439*0f4c859eSApple OSS Distributions 		slp->ns_flag |= ns_flag;
440*0f4c859eSApple OSS Distributions 	}
441*0f4c859eSApple OSS Distributions 	if (waitflag == MBUF_DONTWAIT) {
442*0f4c859eSApple OSS Distributions 		int wake = (slp->ns_flag & SLP_WORKTODO);
443*0f4c859eSApple OSS Distributions 		lck_rw_done(&slp->ns_rwlock);
444*0f4c859eSApple OSS Distributions 		if (wake && nfsd_thread_count) {
445*0f4c859eSApple OSS Distributions 			while (1) {
446*0f4c859eSApple OSS Distributions 				if ((slp->ns_flag & SLP_VALID) == 0) {
447*0f4c859eSApple OSS Distributions 					break;
448*0f4c859eSApple OSS Distributions 				}
449*0f4c859eSApple OSS Distributions 				if (lck_mtx_try_lock(&nfsd_mutex)) {
450*0f4c859eSApple OSS Distributions 					/* Mutex acquired */
451*0f4c859eSApple OSS Distributions 					nfsrv_wakenfsd(slp);
452*0f4c859eSApple OSS Distributions 					lck_mtx_unlock(&nfsd_mutex);
453*0f4c859eSApple OSS Distributions 					break;
454*0f4c859eSApple OSS Distributions 				}
455*0f4c859eSApple OSS Distributions 				IOSleep(NFS_TRYLOCK_MSEC_SLEEP);
456*0f4c859eSApple OSS Distributions 			}
457*0f4c859eSApple OSS Distributions 		}
458*0f4c859eSApple OSS Distributions 	}
459*0f4c859eSApple OSS Distributions }
460*0f4c859eSApple OSS Distributions 
461*0f4c859eSApple OSS Distributions /*
462*0f4c859eSApple OSS Distributions  * Try and extract an RPC request from the mbuf data list received on a
463*0f4c859eSApple OSS Distributions  * stream socket. The "waitflag" argument indicates whether or not it
464*0f4c859eSApple OSS Distributions  * can sleep.
465*0f4c859eSApple OSS Distributions  */
466*0f4c859eSApple OSS Distributions int
nfsrv_getstream(struct nfsrv_sock * slp,int waitflag)467*0f4c859eSApple OSS Distributions nfsrv_getstream(struct nfsrv_sock *slp, int waitflag)
468*0f4c859eSApple OSS Distributions {
469*0f4c859eSApple OSS Distributions 	mbuf_t m;
470*0f4c859eSApple OSS Distributions 	char *cp1, *cp2, *mdata;
471*0f4c859eSApple OSS Distributions 	int error;
472*0f4c859eSApple OSS Distributions 	size_t len, mlen;
473*0f4c859eSApple OSS Distributions 	mbuf_t om, m2, recm;
474*0f4c859eSApple OSS Distributions 	u_int32_t recmark;
475*0f4c859eSApple OSS Distributions 
476*0f4c859eSApple OSS Distributions 	if (slp->ns_flag & SLP_GETSTREAM) {
477*0f4c859eSApple OSS Distributions 		panic("nfs getstream");
478*0f4c859eSApple OSS Distributions 	}
479*0f4c859eSApple OSS Distributions 	slp->ns_flag |= SLP_GETSTREAM;
480*0f4c859eSApple OSS Distributions 	for (;;) {
481*0f4c859eSApple OSS Distributions 		if (slp->ns_reclen == 0) {
482*0f4c859eSApple OSS Distributions 			if (slp->ns_cc < NFSX_UNSIGNED) {
483*0f4c859eSApple OSS Distributions 				slp->ns_flag &= ~SLP_GETSTREAM;
484*0f4c859eSApple OSS Distributions 				return 0;
485*0f4c859eSApple OSS Distributions 			}
486*0f4c859eSApple OSS Distributions 			m = slp->ns_raw;
487*0f4c859eSApple OSS Distributions 			mdata = mbuf_data(m);
488*0f4c859eSApple OSS Distributions 			mlen = mbuf_len(m);
489*0f4c859eSApple OSS Distributions 			if (mlen >= NFSX_UNSIGNED) {
490*0f4c859eSApple OSS Distributions 				bcopy(mdata, (caddr_t)&recmark, NFSX_UNSIGNED);
491*0f4c859eSApple OSS Distributions 				mdata += NFSX_UNSIGNED;
492*0f4c859eSApple OSS Distributions 				mlen -= NFSX_UNSIGNED;
493*0f4c859eSApple OSS Distributions 				mbuf_setdata(m, mdata, mlen);
494*0f4c859eSApple OSS Distributions 			} else {
495*0f4c859eSApple OSS Distributions 				cp1 = (caddr_t)&recmark;
496*0f4c859eSApple OSS Distributions 				cp2 = mdata;
497*0f4c859eSApple OSS Distributions 				while (cp1 < ((caddr_t)&recmark) + NFSX_UNSIGNED) {
498*0f4c859eSApple OSS Distributions 					while (mlen == 0) {
499*0f4c859eSApple OSS Distributions 						m = mbuf_next(m);
500*0f4c859eSApple OSS Distributions 						cp2 = mbuf_data(m);
501*0f4c859eSApple OSS Distributions 						mlen = mbuf_len(m);
502*0f4c859eSApple OSS Distributions 					}
503*0f4c859eSApple OSS Distributions 					*cp1++ = *cp2++;
504*0f4c859eSApple OSS Distributions 					mlen--;
505*0f4c859eSApple OSS Distributions 					mbuf_setdata(m, cp2, mlen);
506*0f4c859eSApple OSS Distributions 				}
507*0f4c859eSApple OSS Distributions 			}
508*0f4c859eSApple OSS Distributions 			slp->ns_cc -= NFSX_UNSIGNED;
509*0f4c859eSApple OSS Distributions 			recmark = ntohl(recmark);
510*0f4c859eSApple OSS Distributions 			slp->ns_reclen = recmark & ~0x80000000;
511*0f4c859eSApple OSS Distributions 			if (recmark & 0x80000000) {
512*0f4c859eSApple OSS Distributions 				slp->ns_flag |= SLP_LASTFRAG;
513*0f4c859eSApple OSS Distributions 			} else {
514*0f4c859eSApple OSS Distributions 				slp->ns_flag &= ~SLP_LASTFRAG;
515*0f4c859eSApple OSS Distributions 			}
516*0f4c859eSApple OSS Distributions 			if (slp->ns_reclen <= 0 || slp->ns_reclen > NFS_MAXPACKET) {
517*0f4c859eSApple OSS Distributions 				slp->ns_flag &= ~SLP_GETSTREAM;
518*0f4c859eSApple OSS Distributions 				return EINVAL;
519*0f4c859eSApple OSS Distributions 			}
520*0f4c859eSApple OSS Distributions 			/* check if we have reached the max allowed memory consumption */
521*0f4c859eSApple OSS Distributions 			if (nfsrv_unprocessed_rpc_max && (nfsrv_unprocessed_rpc_current + slp->ns_reclen > nfsrv_unprocessed_rpc_max)) {
522*0f4c859eSApple OSS Distributions 				slp->ns_flag &= ~SLP_GETSTREAM;
523*0f4c859eSApple OSS Distributions 				printf("nfsrv_getstream: nfsrv_unprocessed_rpc_current (%u) has reached the max allowed consumption (%u)\n", nfsrv_unprocessed_rpc_current, nfsrv_unprocessed_rpc_max);
524*0f4c859eSApple OSS Distributions 				return ENOBUFS;
525*0f4c859eSApple OSS Distributions 			}
526*0f4c859eSApple OSS Distributions 			OSAddAtomic(slp->ns_reclen, &nfsrv_unprocessed_rpc_current);
527*0f4c859eSApple OSS Distributions 			slp->ns_recslen += slp->ns_reclen;
528*0f4c859eSApple OSS Distributions 		}
529*0f4c859eSApple OSS Distributions 
530*0f4c859eSApple OSS Distributions 		/*
531*0f4c859eSApple OSS Distributions 		 * Now get the record part.
532*0f4c859eSApple OSS Distributions 		 *
533*0f4c859eSApple OSS Distributions 		 * Note that slp->ns_reclen may be 0.  Linux sometimes
534*0f4c859eSApple OSS Distributions 		 * generates 0-length RPCs
535*0f4c859eSApple OSS Distributions 		 */
536*0f4c859eSApple OSS Distributions 		recm = NULL;
537*0f4c859eSApple OSS Distributions 		if (slp->ns_cc == slp->ns_reclen) {
538*0f4c859eSApple OSS Distributions 			recm = slp->ns_raw;
539*0f4c859eSApple OSS Distributions 			slp->ns_raw = slp->ns_rawend = NULL;
540*0f4c859eSApple OSS Distributions 			slp->ns_cc = slp->ns_reclen = 0;
541*0f4c859eSApple OSS Distributions 		} else if (slp->ns_cc > slp->ns_reclen) {
542*0f4c859eSApple OSS Distributions 			len = 0;
543*0f4c859eSApple OSS Distributions 			m = slp->ns_raw;
544*0f4c859eSApple OSS Distributions 			mlen = mbuf_len(m);
545*0f4c859eSApple OSS Distributions 			mdata = mbuf_data(m);
546*0f4c859eSApple OSS Distributions 			om = NULL;
547*0f4c859eSApple OSS Distributions 			while (len < slp->ns_reclen) {
548*0f4c859eSApple OSS Distributions 				if ((len + mlen) > slp->ns_reclen) {
549*0f4c859eSApple OSS Distributions 					if (mbuf_copym(m, 0, slp->ns_reclen - len, waitflag, &m2)) {
550*0f4c859eSApple OSS Distributions 						slp->ns_flag &= ~SLP_GETSTREAM;
551*0f4c859eSApple OSS Distributions 						return EWOULDBLOCK;
552*0f4c859eSApple OSS Distributions 					}
553*0f4c859eSApple OSS Distributions 					if (om) {
554*0f4c859eSApple OSS Distributions 						if (mbuf_setnext(om, m2)) {
555*0f4c859eSApple OSS Distributions 							/* trouble... just drop it */
556*0f4c859eSApple OSS Distributions 							printf("nfsrv_getstream: mbuf_setnext failed\n");
557*0f4c859eSApple OSS Distributions 							mbuf_freem(m2);
558*0f4c859eSApple OSS Distributions 							slp->ns_flag &= ~SLP_GETSTREAM;
559*0f4c859eSApple OSS Distributions 							return EWOULDBLOCK;
560*0f4c859eSApple OSS Distributions 						}
561*0f4c859eSApple OSS Distributions 						recm = slp->ns_raw;
562*0f4c859eSApple OSS Distributions 					} else {
563*0f4c859eSApple OSS Distributions 						recm = m2;
564*0f4c859eSApple OSS Distributions 					}
565*0f4c859eSApple OSS Distributions 					mdata += slp->ns_reclen - len;
566*0f4c859eSApple OSS Distributions 					mlen -= slp->ns_reclen - len;
567*0f4c859eSApple OSS Distributions 					mbuf_setdata(m, mdata, mlen);
568*0f4c859eSApple OSS Distributions 					len = slp->ns_reclen;
569*0f4c859eSApple OSS Distributions 				} else if ((len + mlen) == slp->ns_reclen) {
570*0f4c859eSApple OSS Distributions 					om = m;
571*0f4c859eSApple OSS Distributions 					len += mlen;
572*0f4c859eSApple OSS Distributions 					m = mbuf_next(m);
573*0f4c859eSApple OSS Distributions 					recm = slp->ns_raw;
574*0f4c859eSApple OSS Distributions 					if (mbuf_setnext(om, NULL)) {
575*0f4c859eSApple OSS Distributions 						printf("nfsrv_getstream: mbuf_setnext failed 2\n");
576*0f4c859eSApple OSS Distributions 						slp->ns_flag &= ~SLP_GETSTREAM;
577*0f4c859eSApple OSS Distributions 						return EWOULDBLOCK;
578*0f4c859eSApple OSS Distributions 					}
579*0f4c859eSApple OSS Distributions 					mlen = mbuf_len(m);
580*0f4c859eSApple OSS Distributions 					mdata = mbuf_data(m);
581*0f4c859eSApple OSS Distributions 				} else {
582*0f4c859eSApple OSS Distributions 					om = m;
583*0f4c859eSApple OSS Distributions 					len += mlen;
584*0f4c859eSApple OSS Distributions 					m = mbuf_next(m);
585*0f4c859eSApple OSS Distributions 					mlen = mbuf_len(m);
586*0f4c859eSApple OSS Distributions 					mdata = mbuf_data(m);
587*0f4c859eSApple OSS Distributions 				}
588*0f4c859eSApple OSS Distributions 			}
589*0f4c859eSApple OSS Distributions 			slp->ns_raw = m;
590*0f4c859eSApple OSS Distributions 			slp->ns_cc -= len;
591*0f4c859eSApple OSS Distributions 			slp->ns_reclen = 0;
592*0f4c859eSApple OSS Distributions 		} else {
593*0f4c859eSApple OSS Distributions 			slp->ns_flag &= ~SLP_GETSTREAM;
594*0f4c859eSApple OSS Distributions 			return 0;
595*0f4c859eSApple OSS Distributions 		}
596*0f4c859eSApple OSS Distributions 
597*0f4c859eSApple OSS Distributions 		/*
598*0f4c859eSApple OSS Distributions 		 * Accumulate the fragments into a record.
599*0f4c859eSApple OSS Distributions 		 */
600*0f4c859eSApple OSS Distributions 		if (slp->ns_frag == NULL) {
601*0f4c859eSApple OSS Distributions 			slp->ns_frag = recm;
602*0f4c859eSApple OSS Distributions 		} else {
603*0f4c859eSApple OSS Distributions 			m = slp->ns_frag;
604*0f4c859eSApple OSS Distributions 			while ((m2 = mbuf_next(m))) {
605*0f4c859eSApple OSS Distributions 				m = m2;
606*0f4c859eSApple OSS Distributions 			}
607*0f4c859eSApple OSS Distributions 			if ((error = mbuf_setnext(m, recm))) {
608*0f4c859eSApple OSS Distributions 				panic("nfsrv_getstream: mbuf_setnext failed 3, %d", error);
609*0f4c859eSApple OSS Distributions 			}
610*0f4c859eSApple OSS Distributions 		}
611*0f4c859eSApple OSS Distributions 		if (slp->ns_flag & SLP_LASTFRAG) {
612*0f4c859eSApple OSS Distributions 			if (slp->ns_recend) {
613*0f4c859eSApple OSS Distributions 				mbuf_setnextpkt(slp->ns_recend, slp->ns_frag);
614*0f4c859eSApple OSS Distributions 			} else {
615*0f4c859eSApple OSS Distributions 				slp->ns_rec = slp->ns_frag;
616*0f4c859eSApple OSS Distributions 				slp->ns_flag |= SLP_DOREC;
617*0f4c859eSApple OSS Distributions 				OSAddAtomic(-slp->ns_recslen, &nfsrv_unprocessed_rpc_current);
618*0f4c859eSApple OSS Distributions 				slp->ns_recslen = 0;
619*0f4c859eSApple OSS Distributions 			}
620*0f4c859eSApple OSS Distributions 			slp->ns_recend = slp->ns_frag;
621*0f4c859eSApple OSS Distributions 			slp->ns_frag = NULL;
622*0f4c859eSApple OSS Distributions 		}
623*0f4c859eSApple OSS Distributions 	}
624*0f4c859eSApple OSS Distributions }
625*0f4c859eSApple OSS Distributions 
626*0f4c859eSApple OSS Distributions /*
627*0f4c859eSApple OSS Distributions  * Parse an RPC header.
628*0f4c859eSApple OSS Distributions  */
629*0f4c859eSApple OSS Distributions int
nfsrv_dorec(struct nfsrv_sock * slp,struct nfsd * nfsd,struct nfsrv_descript ** ndp)630*0f4c859eSApple OSS Distributions nfsrv_dorec(
631*0f4c859eSApple OSS Distributions 	struct nfsrv_sock *slp,
632*0f4c859eSApple OSS Distributions 	struct nfsd *nfsd,
633*0f4c859eSApple OSS Distributions 	struct nfsrv_descript **ndp)
634*0f4c859eSApple OSS Distributions {
635*0f4c859eSApple OSS Distributions 	mbuf_t m;
636*0f4c859eSApple OSS Distributions 	mbuf_t nam;
637*0f4c859eSApple OSS Distributions 	struct nfsrv_descript *nd;
638*0f4c859eSApple OSS Distributions 	int error = 0;
639*0f4c859eSApple OSS Distributions 
640*0f4c859eSApple OSS Distributions 	*ndp = NULL;
641*0f4c859eSApple OSS Distributions 	if (!(slp->ns_flag & (SLP_VALID | SLP_DOREC)) || (slp->ns_rec == NULL)) {
642*0f4c859eSApple OSS Distributions 		return ENOBUFS;
643*0f4c859eSApple OSS Distributions 	}
644*0f4c859eSApple OSS Distributions 	nd = zalloc(nfsrv_descript_zone);
645*0f4c859eSApple OSS Distributions 	m = slp->ns_rec;
646*0f4c859eSApple OSS Distributions 	slp->ns_rec = mbuf_nextpkt(m);
647*0f4c859eSApple OSS Distributions 	if (slp->ns_rec) {
648*0f4c859eSApple OSS Distributions 		mbuf_setnextpkt(m, NULL);
649*0f4c859eSApple OSS Distributions 	} else {
650*0f4c859eSApple OSS Distributions 		slp->ns_flag &= ~SLP_DOREC;
651*0f4c859eSApple OSS Distributions 		slp->ns_recend = NULL;
652*0f4c859eSApple OSS Distributions 	}
653*0f4c859eSApple OSS Distributions 	slp->ns_reccnt--;
654*0f4c859eSApple OSS Distributions 	if (mbuf_type(m) == MBUF_TYPE_SONAME) {
655*0f4c859eSApple OSS Distributions 		nam = m;
656*0f4c859eSApple OSS Distributions 		m = mbuf_next(m);
657*0f4c859eSApple OSS Distributions 		if ((error = mbuf_setnext(nam, NULL))) {
658*0f4c859eSApple OSS Distributions 			panic("nfsrv_dorec: mbuf_setnext failed %d", error);
659*0f4c859eSApple OSS Distributions 		}
660*0f4c859eSApple OSS Distributions 	} else {
661*0f4c859eSApple OSS Distributions 		nam = NULL;
662*0f4c859eSApple OSS Distributions 	}
663*0f4c859eSApple OSS Distributions 	nd->nd_nam2 = nam;
664*0f4c859eSApple OSS Distributions 	nfsm_chain_dissect_init(error, &nd->nd_nmreq, m);
665*0f4c859eSApple OSS Distributions 	if (!error) {
666*0f4c859eSApple OSS Distributions 		error = nfsrv_getreq(nd);
667*0f4c859eSApple OSS Distributions 	}
668*0f4c859eSApple OSS Distributions 	if (error) {
669*0f4c859eSApple OSS Distributions 		if (nam) {
670*0f4c859eSApple OSS Distributions 			mbuf_freem(nam);
671*0f4c859eSApple OSS Distributions 		}
672*0f4c859eSApple OSS Distributions 		if (nd->nd_gss_context) {
673*0f4c859eSApple OSS Distributions 			nfs_gss_svc_ctx_deref(nd->nd_gss_context);
674*0f4c859eSApple OSS Distributions 		}
675*0f4c859eSApple OSS Distributions 		NFS_ZFREE(nfsrv_descript_zone, nd);
676*0f4c859eSApple OSS Distributions 		return error;
677*0f4c859eSApple OSS Distributions 	}
678*0f4c859eSApple OSS Distributions 	nd->nd_mrep = NULL;
679*0f4c859eSApple OSS Distributions 	*ndp = nd;
680*0f4c859eSApple OSS Distributions 	nfsd->nfsd_nd = nd;
681*0f4c859eSApple OSS Distributions 	return 0;
682*0f4c859eSApple OSS Distributions }
683*0f4c859eSApple OSS Distributions 
684*0f4c859eSApple OSS Distributions /*
685*0f4c859eSApple OSS Distributions  * Parse an RPC request
686*0f4c859eSApple OSS Distributions  * - verify it
687*0f4c859eSApple OSS Distributions  * - fill in the cred struct.
688*0f4c859eSApple OSS Distributions  */
689*0f4c859eSApple OSS Distributions int
nfsrv_getreq(struct nfsrv_descript * nd)690*0f4c859eSApple OSS Distributions nfsrv_getreq(struct nfsrv_descript *nd)
691*0f4c859eSApple OSS Distributions {
692*0f4c859eSApple OSS Distributions 	struct nfsm_chain *nmreq;
693*0f4c859eSApple OSS Distributions 	int len, i;
694*0f4c859eSApple OSS Distributions 	u_int32_t nfsvers, auth_type;
695*0f4c859eSApple OSS Distributions 	int error = 0;
696*0f4c859eSApple OSS Distributions 	uid_t user_id;
697*0f4c859eSApple OSS Distributions 	gid_t group_id;
698*0f4c859eSApple OSS Distributions 	short ngroups;
699*0f4c859eSApple OSS Distributions 	uint32_t val;
700*0f4c859eSApple OSS Distributions 
701*0f4c859eSApple OSS Distributions 	nd->nd_cr = NULL;
702*0f4c859eSApple OSS Distributions 	nd->nd_gss_context = NULL;
703*0f4c859eSApple OSS Distributions 	nd->nd_gss_seqnum = 0;
704*0f4c859eSApple OSS Distributions 	nd->nd_gss_mb = NULL;
705*0f4c859eSApple OSS Distributions 
706*0f4c859eSApple OSS Distributions 	user_id = group_id = -2;
707*0f4c859eSApple OSS Distributions 	val = auth_type = len = 0;
708*0f4c859eSApple OSS Distributions 
709*0f4c859eSApple OSS Distributions 	nmreq = &nd->nd_nmreq;
710*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, nd->nd_retxid); // XID
711*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, val);           // RPC Call
712*0f4c859eSApple OSS Distributions 	if (!error && (val != RPC_CALL)) {
713*0f4c859eSApple OSS Distributions 		error = EBADRPC;
714*0f4c859eSApple OSS Distributions 	}
715*0f4c859eSApple OSS Distributions 	nfsmout_if(error);
716*0f4c859eSApple OSS Distributions 	nd->nd_repstat = 0;
717*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, val);   // RPC Version
718*0f4c859eSApple OSS Distributions 	nfsmout_if(error);
719*0f4c859eSApple OSS Distributions 	if (val != RPC_VER2) {
720*0f4c859eSApple OSS Distributions 		nd->nd_repstat = ERPCMISMATCH;
721*0f4c859eSApple OSS Distributions 		nd->nd_procnum = NFSPROC_NOOP;
722*0f4c859eSApple OSS Distributions 		return 0;
723*0f4c859eSApple OSS Distributions 	}
724*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, val);   // RPC Program Number
725*0f4c859eSApple OSS Distributions 	nfsmout_if(error);
726*0f4c859eSApple OSS Distributions 	if (val != NFS_PROG) {
727*0f4c859eSApple OSS Distributions 		nd->nd_repstat = EPROGUNAVAIL;
728*0f4c859eSApple OSS Distributions 		nd->nd_procnum = NFSPROC_NOOP;
729*0f4c859eSApple OSS Distributions 		return 0;
730*0f4c859eSApple OSS Distributions 	}
731*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, nfsvers);// NFS Version Number
732*0f4c859eSApple OSS Distributions 	nfsmout_if(error);
733*0f4c859eSApple OSS Distributions 	if ((nfsvers < NFS_VER2) || (nfsvers > NFS_VER3)) {
734*0f4c859eSApple OSS Distributions 		nd->nd_repstat = EPROGMISMATCH;
735*0f4c859eSApple OSS Distributions 		nd->nd_procnum = NFSPROC_NOOP;
736*0f4c859eSApple OSS Distributions 		return 0;
737*0f4c859eSApple OSS Distributions 	}
738*0f4c859eSApple OSS Distributions 	nd->nd_vers = nfsvers;
739*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, nd->nd_procnum);// NFS Procedure Number
740*0f4c859eSApple OSS Distributions 	nfsmout_if(error);
741*0f4c859eSApple OSS Distributions 	if ((nd->nd_procnum >= NFS_NPROCS) ||
742*0f4c859eSApple OSS Distributions 	    ((nd->nd_vers == NFS_VER2) && (nd->nd_procnum > NFSV2PROC_STATFS))) {
743*0f4c859eSApple OSS Distributions 		nd->nd_repstat = EPROCUNAVAIL;
744*0f4c859eSApple OSS Distributions 		nd->nd_procnum = NFSPROC_NOOP;
745*0f4c859eSApple OSS Distributions 		return 0;
746*0f4c859eSApple OSS Distributions 	}
747*0f4c859eSApple OSS Distributions 	if (nfsvers != NFS_VER3) {
748*0f4c859eSApple OSS Distributions 		nd->nd_procnum = nfsv3_procid[nd->nd_procnum];
749*0f4c859eSApple OSS Distributions 	}
750*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, auth_type);     // Auth Flavor
751*0f4c859eSApple OSS Distributions 	nfsm_chain_get_32(error, nmreq, len);           // Auth Length
752*0f4c859eSApple OSS Distributions 	if (!error && (len < 0 || len > RPCAUTH_MAXSIZ)) {
753*0f4c859eSApple OSS Distributions 		error = EBADRPC;
754*0f4c859eSApple OSS Distributions 	}
755*0f4c859eSApple OSS Distributions 	nfsmout_if(error);
756*0f4c859eSApple OSS Distributions 
757*0f4c859eSApple OSS Distributions 	/* Handle authentication */
758*0f4c859eSApple OSS Distributions 	if (auth_type == RPCAUTH_SYS) {
759*0f4c859eSApple OSS Distributions 		struct posix_cred temp_pcred;
760*0f4c859eSApple OSS Distributions 		if (nd->nd_procnum == NFSPROC_NULL) {
761*0f4c859eSApple OSS Distributions 			return 0;
762*0f4c859eSApple OSS Distributions 		}
763*0f4c859eSApple OSS Distributions 		nd->nd_sec = RPCAUTH_SYS;
764*0f4c859eSApple OSS Distributions 		nfsm_chain_adv(error, nmreq, NFSX_UNSIGNED);    // skip stamp
765*0f4c859eSApple OSS Distributions 		nfsm_chain_get_32(error, nmreq, len);           // hostname length
766*0f4c859eSApple OSS Distributions 		if (len < 0 || len > NFS_MAXNAMLEN) {
767*0f4c859eSApple OSS Distributions 			error = EBADRPC;
768*0f4c859eSApple OSS Distributions 		}
769*0f4c859eSApple OSS Distributions 		nfsm_chain_adv(error, nmreq, nfsm_rndup(len));  // skip hostname
770*0f4c859eSApple OSS Distributions 		nfsmout_if(error);
771*0f4c859eSApple OSS Distributions 
772*0f4c859eSApple OSS Distributions 		/* create a temporary credential using the bits from the wire */
773*0f4c859eSApple OSS Distributions 		bzero(&temp_pcred, sizeof(temp_pcred));
774*0f4c859eSApple OSS Distributions 		nfsm_chain_get_32(error, nmreq, user_id);
775*0f4c859eSApple OSS Distributions 		nfsm_chain_get_32(error, nmreq, group_id);
776*0f4c859eSApple OSS Distributions 		temp_pcred.cr_groups[0] = group_id;
777*0f4c859eSApple OSS Distributions 		nfsm_chain_get_32(error, nmreq, len);           // extra GID count
778*0f4c859eSApple OSS Distributions 		if ((len < 0) || (len > RPCAUTH_UNIXGIDS)) {
779*0f4c859eSApple OSS Distributions 			error = EBADRPC;
780*0f4c859eSApple OSS Distributions 		}
781*0f4c859eSApple OSS Distributions 		nfsmout_if(error);
782*0f4c859eSApple OSS Distributions 		for (i = 1; i <= len; i++) {
783*0f4c859eSApple OSS Distributions 			if (i < NGROUPS) {
784*0f4c859eSApple OSS Distributions 				nfsm_chain_get_32(error, nmreq, temp_pcred.cr_groups[i]);
785*0f4c859eSApple OSS Distributions 			} else {
786*0f4c859eSApple OSS Distributions 				nfsm_chain_adv(error, nmreq, NFSX_UNSIGNED);
787*0f4c859eSApple OSS Distributions 			}
788*0f4c859eSApple OSS Distributions 		}
789*0f4c859eSApple OSS Distributions 		nfsmout_if(error);
790*0f4c859eSApple OSS Distributions 		ngroups = (len >= NGROUPS) ? NGROUPS : (short)(len + 1);
791*0f4c859eSApple OSS Distributions 		if (ngroups > 1) {
792*0f4c859eSApple OSS Distributions 			nfsrv_group_sort(&temp_pcred.cr_groups[0], ngroups);
793*0f4c859eSApple OSS Distributions 		}
794*0f4c859eSApple OSS Distributions 		nfsm_chain_adv(error, nmreq, NFSX_UNSIGNED);    // verifier flavor (should be AUTH_NONE)
795*0f4c859eSApple OSS Distributions 		nfsm_chain_get_32(error, nmreq, len);           // verifier length
796*0f4c859eSApple OSS Distributions 		if (len < 0 || len > RPCAUTH_MAXSIZ) {
797*0f4c859eSApple OSS Distributions 			error = EBADRPC;
798*0f4c859eSApple OSS Distributions 		}
799*0f4c859eSApple OSS Distributions 		if (len > 0) {
800*0f4c859eSApple OSS Distributions 			nfsm_chain_adv(error, nmreq, nfsm_rndup(len));
801*0f4c859eSApple OSS Distributions 		}
802*0f4c859eSApple OSS Distributions 
803*0f4c859eSApple OSS Distributions 		/* request creation of a real credential */
804*0f4c859eSApple OSS Distributions 		temp_pcred.cr_uid = user_id;
805*0f4c859eSApple OSS Distributions 		temp_pcred.cr_ngroups = ngroups;
806*0f4c859eSApple OSS Distributions 		nd->nd_cr = posix_cred_create(&temp_pcred);
807*0f4c859eSApple OSS Distributions 		if (nd->nd_cr == NULL) {
808*0f4c859eSApple OSS Distributions 			nd->nd_repstat = ENOMEM;
809*0f4c859eSApple OSS Distributions 			nd->nd_procnum = NFSPROC_NOOP;
810*0f4c859eSApple OSS Distributions 			return 0;
811*0f4c859eSApple OSS Distributions 		}
812*0f4c859eSApple OSS Distributions 	} else if (auth_type == RPCSEC_GSS) {
813*0f4c859eSApple OSS Distributions 		error = nfs_gss_svc_cred_get(nd, nmreq);
814*0f4c859eSApple OSS Distributions 		if (error) {
815*0f4c859eSApple OSS Distributions 			if (error == EINVAL) {
816*0f4c859eSApple OSS Distributions 				goto nfsmout;   // drop the request
817*0f4c859eSApple OSS Distributions 			}
818*0f4c859eSApple OSS Distributions 			nd->nd_repstat = error;
819*0f4c859eSApple OSS Distributions 			nd->nd_procnum = NFSPROC_NOOP;
820*0f4c859eSApple OSS Distributions 			return 0;
821*0f4c859eSApple OSS Distributions 		}
822*0f4c859eSApple OSS Distributions 	} else {
823*0f4c859eSApple OSS Distributions 		if (nd->nd_procnum == NFSPROC_NULL) {   // assume it's AUTH_NONE
824*0f4c859eSApple OSS Distributions 			return 0;
825*0f4c859eSApple OSS Distributions 		}
826*0f4c859eSApple OSS Distributions 		nd->nd_repstat = (NFSERR_AUTHERR | AUTH_REJECTCRED);
827*0f4c859eSApple OSS Distributions 		nd->nd_procnum = NFSPROC_NOOP;
828*0f4c859eSApple OSS Distributions 		return 0;
829*0f4c859eSApple OSS Distributions 	}
830*0f4c859eSApple OSS Distributions 	return 0;
831*0f4c859eSApple OSS Distributions nfsmout:
832*0f4c859eSApple OSS Distributions 	if (IS_VALID_CRED(nd->nd_cr)) {
833*0f4c859eSApple OSS Distributions 		kauth_cred_unref(&nd->nd_cr);
834*0f4c859eSApple OSS Distributions 	}
835*0f4c859eSApple OSS Distributions 	nfsm_chain_cleanup(nmreq);
836*0f4c859eSApple OSS Distributions 	return error;
837*0f4c859eSApple OSS Distributions }
838*0f4c859eSApple OSS Distributions 
839*0f4c859eSApple OSS Distributions /*
840*0f4c859eSApple OSS Distributions  * Search for a sleeping nfsd and wake it up.
841*0f4c859eSApple OSS Distributions  * SIDE EFFECT: If none found, make sure the socket is queued up so that one
842*0f4c859eSApple OSS Distributions  * of the running nfsds will go look for the work in the nfsrv_sockwait list.
843*0f4c859eSApple OSS Distributions  * Note: Must be called with nfsd_mutex held.
844*0f4c859eSApple OSS Distributions  */
845*0f4c859eSApple OSS Distributions void
nfsrv_wakenfsd(struct nfsrv_sock * slp)846*0f4c859eSApple OSS Distributions nfsrv_wakenfsd(struct nfsrv_sock *slp)
847*0f4c859eSApple OSS Distributions {
848*0f4c859eSApple OSS Distributions 	struct nfsd *nd;
849*0f4c859eSApple OSS Distributions 
850*0f4c859eSApple OSS Distributions 	while (1) {
851*0f4c859eSApple OSS Distributions 		if ((slp->ns_flag & SLP_VALID) == 0) {
852*0f4c859eSApple OSS Distributions 			return;
853*0f4c859eSApple OSS Distributions 		}
854*0f4c859eSApple OSS Distributions 		if (lck_rw_try_lock_exclusive(&slp->ns_rwlock)) {
855*0f4c859eSApple OSS Distributions 			/* Exclusive lock acquired */
856*0f4c859eSApple OSS Distributions 			break;
857*0f4c859eSApple OSS Distributions 		}
858*0f4c859eSApple OSS Distributions 		IOSleep(NFS_TRYLOCK_MSEC_SLEEP);
859*0f4c859eSApple OSS Distributions 	}
860*0f4c859eSApple OSS Distributions 
861*0f4c859eSApple OSS Distributions 	/* if there's work to do on this socket, make sure it's queued up */
862*0f4c859eSApple OSS Distributions 	if ((slp->ns_flag & SLP_WORKTODO) && !(slp->ns_flag & SLP_QUEUED)) {
863*0f4c859eSApple OSS Distributions 		TAILQ_INSERT_TAIL(&nfsrv_sockwait, slp, ns_svcq);
864*0f4c859eSApple OSS Distributions 		slp->ns_flag |= SLP_WAITQ;
865*0f4c859eSApple OSS Distributions 	}
866*0f4c859eSApple OSS Distributions 	lck_rw_done(&slp->ns_rwlock);
867*0f4c859eSApple OSS Distributions 
868*0f4c859eSApple OSS Distributions 	/* wake up a waiting nfsd, if possible */
869*0f4c859eSApple OSS Distributions 	nd = TAILQ_FIRST(&nfsd_queue);
870*0f4c859eSApple OSS Distributions 	if (!nd) {
871*0f4c859eSApple OSS Distributions 		return;
872*0f4c859eSApple OSS Distributions 	}
873*0f4c859eSApple OSS Distributions 
874*0f4c859eSApple OSS Distributions 	TAILQ_REMOVE(&nfsd_queue, nd, nfsd_queue);
875*0f4c859eSApple OSS Distributions 	nd->nfsd_flag &= ~NFSD_WAITING;
876*0f4c859eSApple OSS Distributions 	wakeup(nd);
877*0f4c859eSApple OSS Distributions }
878*0f4c859eSApple OSS Distributions 
879*0f4c859eSApple OSS Distributions #endif /* CONFIG_NFS_SERVER */
880