2 * support/nfs/rpcmisc.c
4 * Miscellaneous functions for RPC startup and shutdown.
5 * This code is partially snarfed from rpcgen -s tcp -s udp,
6 * partly written by Mark Shand, Donald Becker, and Rick
7 * Sladkey. It was tweaked slightly by Olaf Kirch to be
8 * usable by both unfsd and mountd.
10 * This software may be used for any purpose provided
11 * the above copyright notice is retained. It is supplied
12 * as is, with no warranty expressed or implied.
19 #include <sys/types.h>
20 #include <sys/ioctl.h>
22 #include <sys/socket.h>
24 #include <rpc/pmap_clnt.h>
25 #include <netinet/in.h>
37 #if SIZEOF_SOCKLEN_T - 0 == 0
41 static void closedown(int sig);
42 static int makesock(int port, int proto);
44 #define _RPCSVC_CLOSEDOWN 120
50 rpc_init(char *name, int prog, int vers,
51 void (*dispatch)(struct svc_req *, register SVCXPRT *),
54 struct sockaddr_in saddr;
59 asize = sizeof(saddr);
61 if (getsockname(0, (struct sockaddr *) &saddr, &asize) == 0
62 && saddr.sin_family == AF_INET) {
63 socklen_t ssize = sizeof (int);
65 if (getsockopt(0, SOL_SOCKET, SO_TYPE,
66 (char *)&fdtype, &ssize) == -1)
67 xlog(L_FATAL, "getsockopt failed: %s", strerror(errno));
68 /* inetd passes a UDP socket or a listening TCP socket.
69 * listen will fail on a connected TCP socket(passed by rsh).
71 if (!(fdtype == SOCK_STREAM && listen(0,5) == -1)) {
77 pmap_unset(prog, vers);
81 if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_DGRAM)) {
82 static SVCXPRT *last_transp = NULL;
84 if (_rpcpmstart == 0) {
86 && (!defport || defport == last_transp->xp_port)) {
93 sock = makesock(defport, IPPROTO_UDP);
95 if (sock == RPC_ANYSOCK)
96 sock = svcudp_socket (prog, 1);
97 transp = svcudp_create(sock);
99 xlog(L_FATAL, "cannot create udp service.");
102 if (!svc_register(transp, prog, vers, dispatch, IPPROTO_UDP)) {
103 xlog(L_FATAL, "unable to register (%s, %d, udp).",
106 last_transp = transp;
109 if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_STREAM)) {
110 static SVCXPRT *last_transp = NULL;
112 if (_rpcpmstart == 0) {
114 && (!defport || defport == last_transp->xp_port)) {
115 transp = last_transp;
121 sock = makesock(defport, IPPROTO_TCP);
123 if (sock == RPC_ANYSOCK)
124 sock = svctcp_socket (prog, 1);
125 transp = svctcp_create(sock, 0, 0);
126 if (transp == NULL) {
127 xlog(L_FATAL, "cannot create tcp service.");
130 if (!svc_register(transp, prog, vers, dispatch, IPPROTO_TCP)) {
131 xlog(L_FATAL, "unable to register (%s, %d, tcp).",
134 last_transp = transp;
138 signal (SIGALRM, closedown);
139 alarm (_RPCSVC_CLOSEDOWN);
143 static void closedown(sig)
146 (void) signal(sig, closedown);
147 if (_rpcsvcdirty == 0) {
151 if (_rpcfdtype == SOCK_DGRAM)
154 size = getdtablesize();
156 for (i = 0, openfd = 0; i < size && openfd < 2; i++)
157 if (FD_ISSET(i, &svc_fdset))
162 (void) alarm(_RPCSVC_CLOSEDOWN);
166 * Create listener socket for a given port
168 * Return an open network socket on success; otherwise return -1
169 * if some error occurs.
172 makesock(int port, int proto)
174 struct sockaddr_in sin;
175 int sock, sock_type, val;
177 sock_type = (proto == IPPROTO_UDP) ? SOCK_DGRAM : SOCK_STREAM;
178 sock = socket(AF_INET, sock_type, proto);
180 xlog(L_FATAL, "Could not make a socket: %s",
184 memset((char *) &sin, 0, sizeof(sin));
185 sin.sin_family = AF_INET;
186 sin.sin_addr.s_addr = htonl(INADDR_ANY);
187 sin.sin_port = htons(port);
190 if (proto == IPPROTO_TCP)
191 if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
192 &val, sizeof(val)) < 0)
193 xlog(L_ERROR, "setsockopt failed: %s",
196 if (bind(sock, (struct sockaddr *) &sin, sizeof(sin)) == -1) {
197 xlog(L_FATAL, "Could not bind name to socket: %s",