Home | History | Annotate | Line # | Download | only in nfs
      1 /*	$NetBSD: nfs_boot.c,v 1.90 2024/07/05 04:31:54 rin Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1995, 1997 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Adam Glass and Gordon W. Ross.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 /*
     33  * Support for NFS diskless booting, specifically getting information
     34  * about where to mount root from, what pathnames, etc.
     35  */
     36 
     37 #include <sys/cdefs.h>
     38 __KERNEL_RCSID(0, "$NetBSD: nfs_boot.c,v 1.90 2024/07/05 04:31:54 rin Exp $");
     39 
     40 #ifdef _KERNEL_OPT
     41 #include "opt_nfs.h"
     42 #include "opt_tftproot.h"
     43 #include "opt_nfs_boot.h"
     44 #endif
     45 
     46 #ifdef NFS_BOOT_TCP
     47 #undef NFS_BOOT_UDP
     48 #endif
     49 
     50 #include <sys/param.h>
     51 #include <sys/systm.h>
     52 #include <sys/kernel.h>
     53 #include <sys/device.h>
     54 #include <sys/ioctl.h>
     55 #include <sys/proc.h>
     56 #include <sys/mount.h>
     57 #include <sys/mbuf.h>
     58 #include <sys/reboot.h>
     59 #include <sys/socket.h>
     60 #include <sys/socketvar.h>
     61 
     62 #include <net/if.h>
     63 #include <net/route.h>
     64 #include <net/if_ether.h>
     65 #include <net/if_types.h>
     66 
     67 #include <netinet/in.h>
     68 #include <netinet/if_inarp.h>
     69 
     70 #include <nfs/rpcv2.h>
     71 #include <nfs/krpc.h>
     72 #include <nfs/xdr_subs.h>
     73 
     74 #include <nfs/nfsproto.h>
     75 #include <nfs/nfs.h>
     76 #include <nfs/nfsmount.h>
     77 #include <nfs/nfsdiskless.h>
     78 
     79 /*
     80  * There are three implementations of NFS diskless boot.
     81  * One implementation uses BOOTP (RFC951, RFC1048),
     82  * Sun RPC/bootparams or static configuration.  See the
     83  * files:
     84  *    nfs_bootdhcp.c:   BOOTP (RFC951, RFC1048)
     85  *    nfs_bootparam.c:  Sun RPC/bootparams
     86  *    nfs_bootstatic.c: honour config(1) description
     87  */
     88 #if defined(NFS_BOOT_BOOTP) || defined(NFS_BOOT_DHCP)
     89 int nfs_boot_rfc951 = 1; /* BOOTP enabled (default) */
     90 #endif
     91 #ifdef NFS_BOOT_BOOTPARAM
     92 int nfs_boot_bootparam = 1; /* BOOTPARAM enabled (default) */
     93 #endif
     94 #ifdef NFS_BOOT_BOOTSTATIC
     95 int nfs_boot_bootstatic = 1; /* BOOTSTATIC enabled (default) */
     96 #endif
     97 
     98 #define IP_MIN_MTU 576
     99 
    100 /* mountd RPC */
    101 static int md_mount(struct sockaddr_in *mdsin, char *path,
    102 	struct nfs_args *argp, struct lwp *l);
    103 
    104 static int nfs_boot_delroute_matcher(struct rtentry *, void *);
    105 static void nfs_boot_defrt(struct in_addr *);
    106 static  int nfs_boot_getfh(struct nfs_dlmount *ndm, struct lwp *);
    107 
    108 
    109 /*
    110  * Called with an empty nfs_diskless struct to be filled in.
    111  * Find an interface, determine its ip address (etc.) and
    112  * save all the boot parameters in the nfs_diskless struct.
    113  */
    114 int
    115 nfs_boot_init(struct nfs_diskless *nd, struct lwp *lwp)
    116 {
    117 	struct ifnet *ifp;
    118 	int error = 0;
    119 	int flags __unused;
    120 
    121 	/* Explicitly necessary or build fails
    122 	 * due to unused variable, otherwise.
    123 	 */
    124 	flags = 0;
    125 
    126 	/*
    127 	 * Find the network interface.
    128 	 */
    129 	ifp = ifunit(device_xname(root_device));
    130 	if (ifp == NULL) {
    131 		printf("nfs_boot: '%s' not found\n",
    132 		       device_xname(root_device));
    133 		return (ENXIO);
    134 	}
    135 	nd->nd_ifp = ifp;
    136 
    137 	error = EADDRNOTAVAIL; /* ??? */
    138 #if defined(NFS_BOOT_BOOTSTATIC)
    139 	if (error && nfs_boot_bootstatic) {
    140 		printf("nfs_boot: trying static\n");
    141 		error = nfs_bootstatic(nd, lwp, &flags);
    142 	}
    143 #endif
    144 #if defined(NFS_BOOT_BOOTP) || defined(NFS_BOOT_DHCP)
    145 	if (error && nfs_boot_rfc951) {
    146 #if defined(NFS_BOOT_DHCP)
    147 		printf("nfs_boot: trying DHCP/BOOTP\n");
    148 #else
    149 		printf("nfs_boot: trying BOOTP\n");
    150 #endif
    151 		error = nfs_bootdhcp(nd, lwp, &flags);
    152 	}
    153 #endif
    154 #ifdef NFS_BOOT_BOOTPARAM
    155 	if (error && nfs_boot_bootparam) {
    156 		printf("nfs_boot: trying RARP (and RPC/bootparam)\n");
    157 		error = nfs_bootparam(nd, lwp, &flags);
    158 	}
    159 #endif
    160 	if (error)
    161 		return (error);
    162 
    163 	/*
    164 	 * Set MTU if passed
    165 	 */
    166 	if (nd->nd_mtu >= IP_MIN_MTU )
    167 		nfs_boot_setmtu(nd->nd_ifp, nd->nd_mtu, lwp);
    168 
    169 	/*
    170 	 * If the gateway address is set, add a default route.
    171 	 * (The mountd RPCs may go across a gateway.)
    172 	 */
    173 	if (nd->nd_gwip.s_addr)
    174 		nfs_boot_defrt(&nd->nd_gwip);
    175 
    176 #ifdef TFTPROOT
    177 	if (nd->nd_nomount)
    178 		goto out;
    179 #endif
    180 	/*
    181 	 * Now fetch the NFS file handles as appropriate.
    182 	 */
    183 	error = nfs_boot_getfh(&nd->nd_root, lwp);
    184 
    185 	if (error)
    186 		nfs_boot_cleanup(nd, lwp);
    187 
    188 #ifdef TFTPROOT
    189 out:
    190 #endif
    191 	return (error);
    192 }
    193 
    194 void
    195 nfs_boot_cleanup(struct nfs_diskless *nd, struct lwp *lwp)
    196 {
    197 
    198 	nfs_boot_deladdress(nd->nd_ifp, lwp, nd->nd_myip.s_addr);
    199 	nfs_boot_ifupdown(nd->nd_ifp, lwp, 0);
    200 	nfs_boot_flushrt(nd->nd_ifp);
    201 }
    202 
    203 int
    204 nfs_boot_ifupdown(struct ifnet *ifp, struct lwp *lwp, int up)
    205 {
    206 	struct socket *so;
    207 	struct ifreq ireq;
    208 	int error;
    209 
    210 	memset(&ireq, 0, sizeof(ireq));
    211 	memcpy(ireq.ifr_name, ifp->if_xname, IFNAMSIZ);
    212 
    213 	/*
    214 	 * Get a socket to use for various things in here.
    215 	 * After this, use "goto out" to cleanup and return.
    216 	 */
    217 	error = socreate(AF_INET, &so, SOCK_DGRAM, 0, lwp, NULL);
    218 	if (error) {
    219 		printf("ifupdown: socreate, error=%d\n", error);
    220 		return (error);
    221 	}
    222 
    223 	/*
    224 	 * Bring up the interface. (just set the "up" flag)
    225 	 * Get the old interface flags and or IFF_UP into them so
    226 	 * things like media selection flags are not clobbered.
    227 	 */
    228 	error = ifioctl(so, SIOCGIFFLAGS, (void *)&ireq, lwp);
    229 	if (error) {
    230 		printf("ifupdown: GIFFLAGS, error=%d\n", error);
    231 		goto out;
    232 	}
    233 	if (up)
    234 		ireq.ifr_flags |= IFF_UP;
    235 	else
    236 		ireq.ifr_flags &= ~IFF_UP;
    237 	error = ifioctl(so, SIOCSIFFLAGS, &ireq, lwp);
    238 	if (error) {
    239 		printf("ifupdown: SIFFLAGS, error=%d\n", error);
    240 		goto out;
    241 	}
    242 
    243 	if (up)
    244 		/* give the link some time to get up */
    245 		tsleep(nfs_boot_ifupdown, PZERO, "nfsbif", 3 * hz);
    246 out:
    247 	soclose(so);
    248 	return (error);
    249 }
    250 
    251 void
    252 nfs_boot_setmtu(struct ifnet *ifp, int mtu, struct lwp *lwp)
    253 {
    254 	struct socket *so;
    255 	struct ifreq ireq;
    256 	int error;
    257 
    258 	memset(&ireq, 0, sizeof(ireq));
    259 	memcpy(ireq.ifr_name, ifp->if_xname, IFNAMSIZ);
    260 
    261 	/*
    262 	 * Get a socket to use for various things in here.
    263 	 * After this, use "goto out" to cleanup and return.
    264 	 */
    265 	error = socreate(AF_INET, &so, SOCK_DGRAM, 0, lwp, NULL);
    266 	if (error) {
    267 		printf("setmtu: socreate, error=%d\n", error);
    268 		return;
    269 	}
    270 
    271 	/*
    272 	 * Get structure, set the new MTU, push structure.
    273 	 */
    274 	error = ifioctl(so, SIOCGIFMTU, (void *)&ireq, lwp);
    275 	if (error) {
    276 		printf("setmtu: GIFMTU, error=%d\n", error);
    277 		goto out;
    278 	}
    279 
    280 	ireq.ifr_mtu = mtu;
    281 
    282 	error = ifioctl(so, SIOCSIFMTU, &ireq, lwp);
    283 	if (error) {
    284 		printf("setmtu: SIFMTU, error=%d\n", error);
    285 		goto out;
    286 	}
    287 
    288 out:
    289 	soclose(so);
    290 	return;
    291 }
    292 
    293 int
    294 nfs_boot_setaddress(struct ifnet *ifp, struct lwp *lwp,
    295 		uint32_t addr, uint32_t netmask, uint32_t braddr)
    296 {
    297 	struct socket *so;
    298 	struct ifaliasreq iareq;
    299 	struct sockaddr_in *sin;
    300 	int error;
    301 
    302 	/*
    303 	 * Get a socket to use for various things in here.
    304 	 * After this, use "goto out" to cleanup and return.
    305 	 */
    306 	error = socreate(AF_INET, &so, SOCK_DGRAM, 0, lwp, NULL);
    307 	if (error) {
    308 		printf("setaddress: socreate, error=%d\n", error);
    309 		return (error);
    310 	}
    311 
    312 	memset(&iareq, 0, sizeof(iareq));
    313 	memcpy(iareq.ifra_name, ifp->if_xname, IFNAMSIZ);
    314 
    315 	/* Set the I/F address */
    316 	sin = (struct sockaddr_in *)&iareq.ifra_addr;
    317 	sin->sin_len = sizeof(*sin);
    318 	sin->sin_family = AF_INET;
    319 	sin->sin_addr.s_addr = addr;
    320 
    321 	/* Set the netmask */
    322 	if (netmask != INADDR_ANY) {
    323 		sin = (struct sockaddr_in *)&iareq.ifra_mask;
    324 		sin->sin_len = sizeof(*sin);
    325 		sin->sin_family = AF_INET;
    326 		sin->sin_addr.s_addr = netmask;
    327 	} /* else leave subnetmask unspecified (len=0) */
    328 
    329 	/* Set the broadcast addr. */
    330 	if (braddr != INADDR_ANY) {
    331 		sin = (struct sockaddr_in *)&iareq.ifra_broadaddr;
    332 		sin->sin_len = sizeof(*sin);
    333 		sin->sin_family = AF_INET;
    334 		sin->sin_addr.s_addr = braddr;
    335 	} /* else leave broadcast addr unspecified (len=0) */
    336 
    337 	error = ifioctl(so, SIOCAIFADDR, (void *)&iareq, lwp);
    338 	if (error) {
    339 		printf("setaddress, error=%d\n", error);
    340 		goto out;
    341 	}
    342 
    343 	/* give the link some time to get up */
    344 	tsleep(nfs_boot_setaddress, PZERO, "nfsbtd", 3 * hz);
    345 out:
    346 	soclose(so);
    347 	return (error);
    348 }
    349 
    350 int
    351 nfs_boot_deladdress(struct ifnet *ifp, struct lwp *lwp, uint32_t addr)
    352 {
    353 	struct socket *so;
    354 	struct ifreq ifr;
    355 	struct sockaddr_in sin;
    356 	struct in_addr ia = {.s_addr = addr};
    357 	int error;
    358 
    359 	/*
    360 	 * Get a socket to use for various things in here.
    361 	 * After this, use "goto out" to cleanup and return.
    362 	 */
    363 	error = socreate(AF_INET, &so, SOCK_DGRAM, 0, lwp, NULL);
    364 	if (error) {
    365 		printf("deladdress: socreate, error=%d\n", error);
    366 		return (error);
    367 	}
    368 
    369 	memset(&ifr, 0, sizeof(ifr));
    370 	memcpy(ifr.ifr_name, ifp->if_xname, IFNAMSIZ);
    371 
    372 	sockaddr_in_init(&sin, &ia, 0);
    373 	ifreq_setaddr(SIOCDIFADDR, &ifr, sintocsa(&sin));
    374 
    375 	error = ifioctl(so, SIOCDIFADDR, &ifr, lwp);
    376 	if (error) {
    377 		printf("deladdress, error=%d\n", error);
    378 		goto out;
    379 	}
    380 
    381 out:
    382 	soclose(so);
    383 	return (error);
    384 }
    385 
    386 int
    387 nfs_boot_setrecvtimo(struct socket *so)
    388 {
    389 	struct timeval tv;
    390 
    391 	tv.tv_sec = 1;
    392 	tv.tv_usec = 0;
    393 
    394 	return (so_setsockopt(NULL, so, SOL_SOCKET, SO_RCVTIMEO, &tv,
    395 	    sizeof(tv)));
    396 }
    397 
    398 int
    399 nfs_boot_enbroadcast(struct socket *so)
    400 {
    401 	int32_t on;
    402 
    403 	on = 1;
    404 	return (so_setsockopt(NULL, so, SOL_SOCKET, SO_BROADCAST, &on,
    405 	    sizeof(on)));
    406 }
    407 
    408 int
    409 nfs_boot_sobind_ipport(struct socket *so, uint16_t port, struct lwp *l)
    410 {
    411 	struct sockaddr_in sin;
    412 	int error;
    413 
    414 	sin.sin_len = sizeof(sin);
    415 	sin.sin_family = AF_INET;
    416 	sin.sin_addr.s_addr = INADDR_ANY;
    417 	sin.sin_port = htons(port);
    418 	error = sobind(so, (struct sockaddr *)&sin, l);
    419 	return (error);
    420 }
    421 
    422 /*
    423  * What is the longest we will wait before re-sending a request?
    424  * Note this is also the frequency of "timeout" messages.
    425  * The re-send loop counts up linearly to this maximum, so the
    426  * first complaint will happen after (1+2+3+4+5)=15 seconds.
    427  */
    428 #define	MAX_RESEND_DELAY 5	/* seconds */
    429 #define TOTAL_TIMEOUT   30	/* seconds */
    430 
    431 int
    432 nfs_boot_sendrecv(struct socket *so, struct sockaddr_in *nam,
    433 		int (*sndproc)(struct mbuf *, void *, int),
    434 		struct mbuf *snd,
    435 		int (*rcvproc)(struct mbuf **, void *),
    436 		struct mbuf **rcv, struct mbuf **from_p,
    437 		void *context, struct lwp *lwp)
    438 {
    439 	int error, rcvflg, timo, secs, waited;
    440 	struct mbuf *m, *from;
    441 	struct uio uio;
    442 
    443 	/* Free at end if not null. */
    444 	from = NULL;
    445 
    446 	/*
    447 	 * Send it, repeatedly, until a reply is received,
    448 	 * but delay each re-send by an increasing amount.
    449 	 * If the delay hits the maximum, start complaining.
    450 	 */
    451 	waited = timo = 0;
    452 send_again:
    453 	waited += timo;
    454 	if (waited >= TOTAL_TIMEOUT)
    455 		return (ETIMEDOUT);
    456 
    457 	/* Determine new timeout. */
    458 	if (timo < MAX_RESEND_DELAY)
    459 		timo++;
    460 	else
    461 		printf("nfs_boot: timeout...\n");
    462 
    463 	if (sndproc) {
    464 		error = (*sndproc)(snd, context, waited);
    465 		if (error)
    466 			goto out;
    467 	}
    468 
    469 	/* Send request (or re-send). */
    470 	m = m_copypacket(snd, M_WAIT);
    471 	if (m == NULL) {
    472 		error = ENOBUFS;
    473 		goto out;
    474 	}
    475 	error = (*so->so_send)(so, (struct sockaddr *)nam, NULL,
    476 	    m, NULL, 0, lwp);
    477 	if (error) {
    478 		printf("nfs_boot: sosend: %d\n", error);
    479 		goto out;
    480 	}
    481 	m = NULL;
    482 
    483 	/*
    484 	 * Wait for up to timo seconds for a reply.
    485 	 * The socket receive timeout was set to 1 second.
    486 	 */
    487 
    488 	secs = timo;
    489 	for (;;) {
    490 		m_freem(from);
    491 		from = NULL;
    492 		m_freem(m);
    493 		m = NULL;
    494 		uio.uio_resid = 1 << 16; /* ??? */
    495 		rcvflg = 0;
    496 		error = (*so->so_receive)(so, &from, &uio, &m, NULL, &rcvflg);
    497 		if (error == EWOULDBLOCK) {
    498 			if (--secs <= 0)
    499 				goto send_again;
    500 			continue;
    501 		}
    502 		if (error)
    503 			goto out;
    504 #ifdef DIAGNOSTIC
    505 		if (!m || !(m->m_flags & M_PKTHDR)
    506 		    || (1 << 16) - uio.uio_resid != m->m_pkthdr.len)
    507 			panic("nfs_boot_sendrecv: return size");
    508 #endif
    509 
    510 		if ((*rcvproc)(&m, context))
    511 			continue;
    512 
    513 		if (rcv)
    514 			*rcv = m;
    515 		else
    516 			m_freem(m);
    517 		if (from_p) {
    518 			*from_p = from;
    519 			from = NULL;
    520 		}
    521 		break;
    522 	}
    523 out:
    524 	m_freem(from);
    525 	return (error);
    526 }
    527 
    528 /*
    529  * Install a default route to the passed IP address.
    530  */
    531 static void
    532 nfs_boot_defrt(struct in_addr *gw_ip)
    533 {
    534 	struct sockaddr dst, gw, mask;
    535 	struct sockaddr_in *sin;
    536 	int error;
    537 
    538 	/* Destination: (default) */
    539 	memset((void *)&dst, 0, sizeof(dst));
    540 	dst.sa_len = sizeof(dst);
    541 	dst.sa_family = AF_INET;
    542 	/* Gateway: */
    543 	memset((void *)&gw, 0, sizeof(gw));
    544 	sin = (struct sockaddr_in *)&gw;
    545 	sin->sin_len = sizeof(*sin);
    546 	sin->sin_family = AF_INET;
    547 	sin->sin_addr.s_addr = gw_ip->s_addr;
    548 	/* Mask: (zero length) */
    549 	/* XXX - Just pass a null pointer? */
    550 	memset(&mask, 0, sizeof(mask));
    551 
    552 	/* add, dest, gw, mask, flags, 0 */
    553 	error = rtrequest(RTM_ADD, &dst, &gw, &mask,
    554 			  (RTF_UP | RTF_GATEWAY | RTF_STATIC), NULL);
    555 	if (error) {
    556 		printf("nfs_boot: add route, error=%d\n", error);
    557 		error = 0;
    558 	}
    559 }
    560 
    561 static int
    562 nfs_boot_delroute_matcher(struct rtentry *rt, void *w)
    563 {
    564 
    565 	if ((void *)rt->rt_ifp != w)
    566 		return 0;
    567 
    568 	return 1;
    569 }
    570 
    571 void
    572 nfs_boot_flushrt(struct ifnet *ifp)
    573 {
    574 
    575 	rt_delete_matched_entries(AF_INET, nfs_boot_delroute_matcher, ifp, false);
    576 }
    577 
    578 /*
    579  * Get an initial NFS file handle using Sun RPC/mountd.
    580  * Separate function because we used to call it twice.
    581  * (once for root and once for swap)
    582  *
    583  * ndm  output
    584  */
    585 static int
    586 nfs_boot_getfh(struct nfs_dlmount *ndm, struct lwp *l)
    587 {
    588 	struct nfs_args *args;
    589 	struct sockaddr_in *sin;
    590 	char *pathname;
    591 	int error;
    592 	u_int16_t port;
    593 
    594 	args = &ndm->ndm_args;
    595 
    596 	/* Initialize mount args. */
    597 	memset((void *) args, 0, sizeof(*args));
    598 	args->addr     = &ndm->ndm_saddr;
    599 	args->addrlen  = args->addr->sa_len;
    600 #ifdef NFS_BOOT_UDP
    601 	args->sotype   = SOCK_DGRAM;
    602 #else
    603 	args->sotype   = SOCK_STREAM;
    604 #endif
    605 	args->fh       = ndm->ndm_fh;
    606 	args->hostname = ndm->ndm_host;
    607 	args->flags    = NFSMNT_NOCONN | NFSMNT_RESVPORT;
    608 
    609 #ifndef NFS_V2_ONLY
    610 	args->flags    |= NFSMNT_NFSV3;
    611 #endif
    612 #ifdef	NFS_BOOT_OPTIONS
    613 	args->flags    |= NFS_BOOT_OPTIONS;
    614 #endif
    615 #ifdef	NFS_BOOT_RWSIZE
    616 	/*
    617 	 * Reduce rsize,wsize for interfaces that consistently
    618 	 * drop fragments of long UDP messages.  (i.e. wd8003).
    619 	 * You can always change these later via remount.
    620 	 */
    621 	args->flags   |= NFSMNT_WSIZE | NFSMNT_RSIZE;
    622 	args->wsize    = NFS_BOOT_RWSIZE;
    623 	args->rsize    = NFS_BOOT_RWSIZE;
    624 #endif
    625 
    626 	/*
    627 	 * Find the pathname part of the "server:pathname"
    628 	 * string left in ndm->ndm_host by nfs_boot_init.
    629 	 */
    630 	pathname = strchr(ndm->ndm_host, ':');
    631 	if (pathname == 0) {
    632 		printf("nfs_boot: getfh - no pathname\n");
    633 		return (EIO);
    634 	}
    635 	pathname++;
    636 
    637 	/*
    638 	 * Get file handle using RPC to mountd/mount
    639 	 */
    640 	sin = (struct sockaddr_in *)&ndm->ndm_saddr;
    641 	error = md_mount(sin, pathname, args, l);
    642 	if (error) {
    643 		printf("nfs_boot: mountd `%s', error=%d\n",
    644 		       ndm->ndm_host, error);
    645 		return (error);
    646 	}
    647 
    648 	/* Set port number for NFS use. */
    649 	/* XXX: NFS port is always 2049, right? */
    650 retry:
    651 	error = krpc_portmap(sin, NFS_PROG,
    652 		    (args->flags & NFSMNT_NFSV3) ? NFS_VER3 : NFS_VER2,
    653 		    (args->sotype == SOCK_STREAM) ? IPPROTO_TCP : IPPROTO_UDP,
    654 		    &port, l);
    655 	if (port == htons(0))
    656 		error = EIO;
    657 	if (error) {
    658 		if (args->sotype == SOCK_STREAM) {
    659 			args->sotype = SOCK_DGRAM;
    660 			goto retry;
    661 		}
    662 		printf("nfs_boot: portmap NFS, error=%d\n", error);
    663 		return (error);
    664 	}
    665 	sin->sin_port = port;
    666 	return (0);
    667 }
    668 
    669 
    670 /*
    671  * RPC: mountd/mount
    672  * Given a server pathname, get an NFS file handle.
    673  * Also, sets sin->sin_port to the NFS service port.
    674  *
    675  * mdsin   mountd server address
    676  */
    677 static int
    678 md_mount(struct sockaddr_in *mdsin, char *path,
    679 	 struct nfs_args *argp, struct lwp *lwp)
    680 {
    681 	/* The RPC structures */
    682 	struct rdata {
    683 		u_int32_t errno;
    684 		union {
    685 			u_int8_t  v2fh[NFSX_V2FH];
    686 			struct {
    687 				u_int32_t fhlen;
    688 				u_int8_t  fh[1];
    689 			} v3fh;
    690 		} fh;
    691 	} *rdata;
    692 	struct mbuf *m;
    693 	u_int8_t *fh;
    694 	int minlen, error;
    695 	int mntver;
    696 
    697 	mntver = (argp->flags & NFSMNT_NFSV3) ? 3 : 2;
    698 	do {
    699 		/*
    700 		 * Get port number for MOUNTD.
    701 		 */
    702 		error = krpc_portmap(mdsin, RPCPROG_MNT, mntver,
    703 		                    IPPROTO_UDP, &mdsin->sin_port, lwp);
    704 		if (error)
    705 			continue;
    706 
    707 		/* This mbuf is consumed by krpc_call. */
    708 		m = xdr_string_encode(path, strlen(path));
    709 		if (m == NULL)
    710 			return ENOMEM;
    711 
    712 		/* Do RPC to mountd. */
    713 		error = krpc_call(mdsin, RPCPROG_MNT, mntver,
    714 		                  RPCMNT_MOUNT, &m, NULL, lwp);
    715 		if (error != EPROGMISMATCH)
    716 			break;
    717 		/* Try lower version of mountd. */
    718 	} while (--mntver >= 1);
    719 	if (error) {
    720 		printf("nfs_boot: mountd error=%d\n", error);
    721 		return error;
    722 	}
    723 	if (mntver != 3)
    724 		argp->flags &= ~NFSMNT_NFSV3;
    725 
    726 	/* The reply might have only the errno. */
    727 	if (m->m_len < 4)
    728 		goto bad;
    729 	/* Have at least errno, so check that. */
    730 	rdata = mtod(m, struct rdata *);
    731 	error = fxdr_unsigned(u_int32_t, rdata->errno);
    732 	if (error)
    733 		goto out;
    734 
    735 	/* Have errno==0, so the fh must be there. */
    736 	if (mntver == 3) {
    737 		argp->fhsize   = fxdr_unsigned(u_int32_t, rdata->fh.v3fh.fhlen);
    738 		if (argp->fhsize > NFSX_V3FHMAX)
    739 			goto bad;
    740 		minlen = 2 * sizeof(u_int32_t) + argp->fhsize;
    741 	} else {
    742 		argp->fhsize   = NFSX_V2FH;
    743 		minlen = sizeof(u_int32_t) + argp->fhsize;
    744 	}
    745 
    746 	if (m->m_len < minlen) {
    747 		m = m_pullup(m, minlen);
    748 		if (m == NULL)
    749 			return(EBADRPC);
    750 		rdata = mtod(m, struct rdata *);
    751 	}
    752 
    753 	fh = (mntver == 3) ?
    754 		rdata->fh.v3fh.fh : rdata->fh.v2fh;
    755 	memcpy(argp->fh, fh, argp->fhsize);
    756 
    757 	goto out;
    758 
    759 bad:
    760 	error = EBADRPC;
    761 
    762 out:
    763 	m_freem(m);
    764 	return error;
    765 }
    766