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