diff options
Diffstat (limited to 'contrib/netbsd-tests/fs/nfs/nfsservice/nfsd.c')
-rw-r--r-- | contrib/netbsd-tests/fs/nfs/nfsservice/nfsd.c | 571 |
1 files changed, 571 insertions, 0 deletions
diff --git a/contrib/netbsd-tests/fs/nfs/nfsservice/nfsd.c b/contrib/netbsd-tests/fs/nfs/nfsservice/nfsd.c new file mode 100644 index 0000000..b062501 --- /dev/null +++ b/contrib/netbsd-tests/fs/nfs/nfsservice/nfsd.c @@ -0,0 +1,571 @@ +/* $NetBSD: nfsd.c,v 1.4 2013/10/19 17:45:00 christos Exp $ */ + +/* + * Copyright (c) 1989, 1993, 1994 + * The Regents of the University of California. All rights reserved. + * + * This code is derived from software contributed to Berkeley by + * Rick Macklem at The University of Guelph. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. Neither the name of the University nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#include <sys/cdefs.h> +#ifndef lint +__COPYRIGHT("@(#) Copyright (c) 1989, 1993, 1994\ + The Regents of the University of California. All rights reserved."); +#endif /* not lint */ + +#ifndef lint +#if 0 +static char sccsid[] = "@(#)nfsd.c 8.9 (Berkeley) 3/29/95"; +#else +__RCSID("$NetBSD: nfsd.c,v 1.4 2013/10/19 17:45:00 christos Exp $"); +#endif +#endif /* not lint */ + +#include <sys/param.h> +#include <sys/ioctl.h> +#include <sys/stat.h> +#include <sys/wait.h> +#include <sys/uio.h> +#include <sys/ucred.h> +#include <sys/mount.h> +#include <sys/socket.h> +#include <sys/socketvar.h> +#include <poll.h> + +#include <rpc/rpc.h> +#include <rpc/pmap_clnt.h> +#include <rpc/pmap_prot.h> + +#include <nfs/rpcv2.h> +#include <nfs/nfsproto.h> +#include <nfs/nfs.h> + +#include <err.h> +#include <errno.h> +#include <fcntl.h> +#include <grp.h> +#include <pwd.h> +#include <pthread.h> +#include <signal.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <syslog.h> +#include <unistd.h> +#include <netdb.h> + +#include <rump/rump.h> +#include <rump/rump_syscalls.h> + +/* Global defs */ +#ifdef DEBUG +#define syslog(e, s, args...) \ +do { \ + fprintf(stderr,(s), ## args); \ + fprintf(stderr, "\n"); \ +} while (/*CONSTCOND*/0) +int debug = 1; +#else +int debug = 0; +#endif + +int main __P((int, char **)); +void nonfs __P((int)); +void usage __P((void)); + +static void * +child(void *arg) +{ + struct nfsd_srvargs nsd; + int nfssvc_flag; + + nfssvc_flag = NFSSVC_NFSD; + memset(&nsd, 0, sizeof(nsd)); + while (rump_sys_nfssvc(nfssvc_flag, &nsd) < 0) { + if (errno != ENEEDAUTH) { + syslog(LOG_ERR, "nfssvc: %m %d", errno); + exit(1); + } + nfssvc_flag = NFSSVC_NFSD | NFSSVC_AUTHINFAIL; + } + + return NULL; +} + +/* + * Nfs server daemon mostly just a user context for nfssvc() + * + * 1 - do file descriptor and signal cleanup + * 2 - create the nfsd thread(s) + * 3 - create server socket(s) + * 4 - register socket with portmap + * + * For connectionless protocols, just pass the socket into the kernel via + * nfssvc(). + * For connection based sockets, loop doing accepts. When you get a new + * socket from accept, pass the msgsock into the kernel via nfssvc(). + * The arguments are: + * -c - support iso cltp clients + * -r - reregister with portmapper + * -t - support tcp nfs clients + * -u - support udp nfs clients + * followed by "n" which is the number of nfsd threads to create + */ +int nfsd_main(int, char**); +int +nfsd_main(argc, argv) + int argc; + char *argv[]; +{ + struct nfsd_args nfsdargs; + struct addrinfo *ai_udp, *ai_tcp, *ai_udp6, *ai_tcp6, hints; + struct netconfig *nconf_udp, *nconf_tcp, *nconf_udp6, *nconf_tcp6; + struct netbuf nb_udp, nb_tcp, nb_udp6, nb_tcp6; + struct sockaddr_in inetpeer; + struct pollfd set[4]; + socklen_t len; + int ch, connect_type_cnt, i, msgsock; + int nfsdcnt, on = 1, reregister, sock, tcpflag, tcpsock; + int tcp6sock, ip6flag; + int tp4cnt, tp4flag, tpipcnt, udpflag, ecode, s; + int error = 0; + +#define DEFNFSDCNT 4 + nfsdcnt = DEFNFSDCNT; + reregister = tcpflag = tp4cnt = tp4flag = tpipcnt = 0; + udpflag = ip6flag = 0; + nconf_udp = nconf_tcp = nconf_udp6 = nconf_tcp6 = NULL; + tcpsock = tcp6sock = -1; +#define GETOPT "6n:rtu" +#define USAGE "[-rtu] [-n num_servers]" + while ((ch = getopt(argc, argv, GETOPT)) != -1) { + switch (ch) { + case '6': + ip6flag = 1; + s = socket(PF_INET6, SOCK_DGRAM, IPPROTO_UDP); + if (s < 0 && (errno == EPROTONOSUPPORT || + errno == EPFNOSUPPORT || errno == EAFNOSUPPORT)) + ip6flag = 0; + else + close(s); + break; + case 'n': + nfsdcnt = atoi(optarg); + if (nfsdcnt < 1) { + warnx("nfsd count %d; reset to %d", nfsdcnt, DEFNFSDCNT); + nfsdcnt = DEFNFSDCNT; + } + break; + case 'r': + reregister = 1; + break; + case 't': + tcpflag = 1; + break; + case 'u': + udpflag = 1; + break; + default: + case '?': + usage(); + }; + } + argv += optind; + argc -= optind; + + /* + * XXX + * Backward compatibility, trailing number is the count of daemons. + */ + if (argc > 1) + usage(); + if (argc == 1) { + nfsdcnt = atoi(argv[0]); + if (nfsdcnt < 1) { + warnx("nfsd count %d; reset to %d", nfsdcnt, DEFNFSDCNT); + nfsdcnt = DEFNFSDCNT; + } + } + + /* + * If none of TCP or UDP are specified, default to UDP only. + */ + if (tcpflag == 0 && udpflag == 0) + udpflag = 1; + + if (debug == 0) { + fprintf(stderr, "non-debug not supported here\n"); + exit(1); + +#ifdef not_the_debug_man + daemon(0, 0); + (void)signal(SIGHUP, SIG_IGN); + (void)signal(SIGINT, SIG_IGN); + (void)signal(SIGQUIT, SIG_IGN); + (void)signal(SIGSYS, nonfs); +#endif + } + + if (udpflag) { + memset(&hints, 0, sizeof hints); + hints.ai_flags = AI_PASSIVE; + hints.ai_family = PF_INET; + hints.ai_socktype = SOCK_DGRAM; + hints.ai_protocol = IPPROTO_UDP; + + ecode = getaddrinfo(NULL, "nfs", &hints, &ai_udp); + if (ecode != 0) { + syslog(LOG_ERR, "getaddrinfo udp: %s", + gai_strerror(ecode)); + exit(1); + } + + nconf_udp = getnetconfigent("udp"); + + if (nconf_udp == NULL) + err(1, "getnetconfigent udp failed"); + + nb_udp.buf = ai_udp->ai_addr; + nb_udp.len = nb_udp.maxlen = ai_udp->ai_addrlen; + if (reregister) + if (!rpcb_set(RPCPROG_NFS, 2, nconf_udp, &nb_udp)) + err(1, "rpcb_set udp failed"); + } + + if (tcpflag) { + memset(&hints, 0, sizeof hints); + hints.ai_flags = AI_PASSIVE; + hints.ai_family = PF_INET; + hints.ai_socktype = SOCK_STREAM; + hints.ai_protocol = IPPROTO_TCP; + + ecode = getaddrinfo(NULL, "nfs", &hints, &ai_tcp); + if (ecode != 0) { + syslog(LOG_ERR, "getaddrinfo tcp: %s", + gai_strerror(ecode)); + exit(1); + } + + nconf_tcp = getnetconfigent("tcp"); + + if (nconf_tcp == NULL) + err(1, "getnetconfigent tcp failed"); + + nb_tcp.buf = ai_tcp->ai_addr; + nb_tcp.len = nb_tcp.maxlen = ai_tcp->ai_addrlen; + if (reregister) + if (!rpcb_set(RPCPROG_NFS, 2, nconf_tcp, &nb_tcp)) + err(1, "rpcb_set tcp failed"); + } + + if (udpflag && ip6flag) { + memset(&hints, 0, sizeof hints); + hints.ai_flags = AI_PASSIVE; + hints.ai_family = PF_INET6; + hints.ai_socktype = SOCK_DGRAM; + hints.ai_protocol = IPPROTO_UDP; + + ecode = getaddrinfo(NULL, "nfs", &hints, &ai_udp6); + if (ecode != 0) { + syslog(LOG_ERR, "getaddrinfo udp: %s", + gai_strerror(ecode)); + exit(1); + } + + nconf_udp6 = getnetconfigent("udp6"); + + if (nconf_udp6 == NULL) + err(1, "getnetconfigent udp6 failed"); + + nb_udp6.buf = ai_udp6->ai_addr; + nb_udp6.len = nb_udp6.maxlen = ai_udp6->ai_addrlen; + if (reregister) + if (!rpcb_set(RPCPROG_NFS, 2, nconf_udp6, &nb_udp6)) + err(1, "rpcb_set udp6 failed"); + } + + if (tcpflag && ip6flag) { + memset(&hints, 0, sizeof hints); + hints.ai_flags = AI_PASSIVE; + hints.ai_family = PF_INET6; + hints.ai_socktype = SOCK_STREAM; + hints.ai_protocol = IPPROTO_TCP; + + ecode = getaddrinfo(NULL, "nfs", &hints, &ai_tcp6); + if (ecode != 0) { + syslog(LOG_ERR, "getaddrinfo tcp: %s", + gai_strerror(ecode)); + exit(1); + } + + nconf_tcp6 = getnetconfigent("tcp6"); + + if (nconf_tcp6 == NULL) + err(1, "getnetconfigent tcp6 failed"); + + nb_tcp6.buf = ai_tcp6->ai_addr; + nb_tcp6.len = nb_tcp6.maxlen = ai_tcp6->ai_addrlen; + if (reregister) + if (!rpcb_set(RPCPROG_NFS, 2, nconf_tcp6, &nb_tcp6)) + err(1, "rpcb_set tcp6 failed"); + } + + openlog("nfsd", LOG_PID, LOG_DAEMON); + + for (i = 0; i < nfsdcnt; i++) { + pthread_t t; + pthread_create(&t, NULL, child, NULL); + } + + /* If we are serving udp, set up the socket. */ + if (udpflag) { + if ((sock = rump_sys_socket(ai_udp->ai_family, ai_udp->ai_socktype, + ai_udp->ai_protocol)) < 0) { + syslog(LOG_ERR, "can't create udp socket"); + exit(1); + } + if (bind(sock, ai_udp->ai_addr, ai_udp->ai_addrlen) < 0) { + syslog(LOG_ERR, "can't bind udp addr"); + exit(1); + } + if (!rpcb_set(RPCPROG_NFS, 2, nconf_udp, &nb_udp) || + !rpcb_set(RPCPROG_NFS, 3, nconf_udp, &nb_udp)) { + syslog(LOG_ERR, "can't register with udp portmap"); + exit(1); + } + nfsdargs.sock = sock; + nfsdargs.name = NULL; + nfsdargs.namelen = 0; + if (rump_sys_nfssvc(NFSSVC_ADDSOCK, &nfsdargs) < 0) { + syslog(LOG_ERR, "can't add UDP socket"); + exit(1); + } + (void)rump_sys_close(sock); + } + + if (udpflag &&ip6flag) { + if ((sock = rump_sys_socket(ai_udp6->ai_family, ai_udp6->ai_socktype, + ai_udp6->ai_protocol)) < 0) { + syslog(LOG_ERR, "can't create udp socket"); + exit(1); + } + if (rump_sys_setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, + &on, sizeof on) < 0) { + syslog(LOG_ERR, "can't set v6-only binding for udp6 " + "socket: %m"); + exit(1); + } + if (rump_sys_bind(sock, ai_udp6->ai_addr, ai_udp6->ai_addrlen) < 0) { + syslog(LOG_ERR, "can't bind udp addr"); + exit(1); + } + if (!rpcb_set(RPCPROG_NFS, 2, nconf_udp6, &nb_udp6) || + !rpcb_set(RPCPROG_NFS, 3, nconf_udp6, &nb_udp6)) { + syslog(LOG_ERR, "can't register with udp portmap"); + exit(1); + } + nfsdargs.sock = sock; + nfsdargs.name = NULL; + nfsdargs.namelen = 0; + if (rump_sys_nfssvc(NFSSVC_ADDSOCK, &nfsdargs) < 0) { + syslog(LOG_ERR, "can't add UDP6 socket"); + exit(1); + } + (void)rump_sys_close(sock); + } + + /* Now set up the master server socket waiting for tcp connections. */ + on = 1; + connect_type_cnt = 0; + if (tcpflag) { + if ((tcpsock = rump_sys_socket(ai_tcp->ai_family, ai_tcp->ai_socktype, + ai_tcp->ai_protocol)) < 0) { + syslog(LOG_ERR, "can't create tcp socket"); + exit(1); + } + if (setsockopt(tcpsock, + SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on)) < 0) + syslog(LOG_ERR, "setsockopt SO_REUSEADDR: %m"); + if (bind(tcpsock, ai_tcp->ai_addr, ai_tcp->ai_addrlen) < 0) { + syslog(LOG_ERR, "can't bind tcp addr"); + exit(1); + } + if (rump_sys_listen(tcpsock, 5) < 0) { + syslog(LOG_ERR, "listen failed"); + exit(1); + } + if (!rpcb_set(RPCPROG_NFS, 2, nconf_tcp, &nb_tcp) || + !rpcb_set(RPCPROG_NFS, 3, nconf_tcp, &nb_tcp)) { + syslog(LOG_ERR, "can't register tcp with rpcbind"); + exit(1); + } + set[0].fd = tcpsock; + set[0].events = POLLIN; + connect_type_cnt++; + } else + set[0].fd = -1; + + if (tcpflag && ip6flag) { + if ((tcp6sock = socket(ai_tcp6->ai_family, ai_tcp6->ai_socktype, + ai_tcp6->ai_protocol)) < 0) { + syslog(LOG_ERR, "can't create tcp socket"); + exit(1); + } + if (setsockopt(tcp6sock, + SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on)) < 0) + syslog(LOG_ERR, "setsockopt SO_REUSEADDR: %m"); + if (setsockopt(tcp6sock, IPPROTO_IPV6, IPV6_V6ONLY, + &on, sizeof on) < 0) { + syslog(LOG_ERR, "can't set v6-only binding for tcp6 " + "socket: %m"); + exit(1); + } + if (bind(tcp6sock, ai_tcp6->ai_addr, ai_tcp6->ai_addrlen) < 0) { + syslog(LOG_ERR, "can't bind tcp6 addr"); + exit(1); + } + if (listen(tcp6sock, 5) < 0) { + syslog(LOG_ERR, "listen failed"); + exit(1); + } + if (!rpcb_set(RPCPROG_NFS, 2, nconf_tcp6, &nb_tcp6) || + !rpcb_set(RPCPROG_NFS, 3, nconf_tcp6, &nb_tcp6)) { + syslog(LOG_ERR, "can't register tcp6 with rpcbind"); + exit(1); + } + set[1].fd = tcp6sock; + set[1].events = POLLIN; + connect_type_cnt++; + } else + set[1].fd = -1; + + set[2].fd = -1; + set[3].fd = -1; + + if (connect_type_cnt == 0) { + pause(); + exit(0); + } + + /* + * Loop forever accepting connections and passing the sockets + * into the kernel for the mounts. + */ + for (;;) { + if (rump_sys_poll(set, 4, INFTIM) < 1) { + syslog(LOG_ERR, "poll failed: %m"); + exit(1); + } + + len = sizeof(inetpeer); + if ((msgsock = accept(tcpsock, + (struct sockaddr *)&inetpeer, &len)) < 0) { + syslog(LOG_ERR, "accept failed: %d", error); + exit(1); + } + memset(inetpeer.sin_zero, 0, sizeof(inetpeer.sin_zero)); + if (setsockopt(msgsock, SOL_SOCKET, + SO_KEEPALIVE, (char *)&on, sizeof(on)) < 0) + syslog(LOG_ERR, + "setsockopt SO_KEEPALIVE: %m"); + nfsdargs.sock = msgsock; + nfsdargs.name = (caddr_t)&inetpeer; + nfsdargs.namelen = sizeof(inetpeer); + rump_sys_nfssvc(NFSSVC_ADDSOCK, &nfsdargs); + (void)rump_sys_close(msgsock); + +#ifdef notyet + if (set[1].revents & POLLIN) { + len = sizeof(inet6peer); + if ((msgsock = rump_sys_accept(tcp6sock, + (struct sockaddr *)&inet6peer, &len, &error)) < 0) { + syslog(LOG_ERR, "accept failed: %m"); + exit(1); + } + if (rump_sys_setsockopt(msgsock, SOL_SOCKET, + SO_KEEPALIVE, (char *)&on, sizeof(on), &error) < 0) + syslog(LOG_ERR, + "setsockopt SO_KEEPALIVE: %m"); + nfsdargs.sock = msgsock; + nfsdargs.name = (caddr_t)&inet6peer; + nfsdargs.namelen = sizeof(inet6peer); + rump_sys_nfssvc(NFSSVC_ADDSOCK, &nfsdargs, &error); + (void)rump_sys_close(msgsock, &error); + } + + if (set[2].revents & POLLIN) { + len = sizeof(isopeer); + if ((msgsock = rump_sys_accept(tp4sock, + (struct sockaddr *)&isopeer, &len, &error)) < 0) { + syslog(LOG_ERR, "accept failed: %m"); + exit(1); + } + if (rump_sys_setsockopt(msgsock, SOL_SOCKET, + SO_KEEPALIVE, (char *)&on, sizeof(on), &error) < 0) + syslog(LOG_ERR, + "setsockopt SO_KEEPALIVE: %m"); + nfsdargs.sock = msgsock; + nfsdargs.name = (caddr_t)&isopeer; + nfsdargs.namelen = len; + rump_sys_nfssvc(NFSSVC_ADDSOCK, &nfsdargs, &error); + (void)rump_sys_close(msgsock, &error); + } + + if (set[3].revents & POLLIN) { + len = sizeof(inetpeer); + if ((msgsock = rump_sys_accept(tpipsock, + (struct sockaddr *)&inetpeer, &len)) < 0) { + syslog(LOG_ERR, "accept failed: %m"); + exit(1); + } + if (setsockopt(msgsock, SOL_SOCKET, + SO_KEEPALIVE, (char *)&on, sizeof(on)) < 0) + syslog(LOG_ERR, "setsockopt SO_KEEPALIVE: %m"); + nfsdargs.sock = msgsock; + nfsdargs.name = (caddr_t)&inetpeer; + nfsdargs.namelen = len; + rump_sys_nfssvc(NFSSVC_ADDSOCK, &nfsdargs); + (void)rump_sys_close(msgsock); + } +#endif /* notyet */ + } +} + +void +usage() +{ + (void)fprintf(stderr, "usage: nfsd %s\n", USAGE); + exit(1); +} + +void +nonfs(signo) + int signo; +{ + syslog(LOG_ERR, "missing system call: NFS not available."); +} |