support/nfs/rpcmisc.c

   1 /*
   2  * support/nfs/rpcmisc.c
   3  *
   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.
   9  *
  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.
  13  */
  14
  15 #ifdef HAVE_CONFIG_H
  16 #include <config.h>
  17 #endif
  18
  19 #include <sys/types.h>
  20 #include <sys/ioctl.h>
  21 #include <sys/stat.h>
  22 #include <sys/socket.h>
  23 #include <rpc/rpc.h>
  24 #include <rpc/pmap_clnt.h>
  25 #include <netinet/in.h>
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29 #include <signal.h>
  30 #include <fcntl.h>
  31 #include <memory.h>
  32 #include <errno.h>
  33 #include <unistd.h>
  34 #include <time.h>
  35 #include "nfslib.h"
  36
  37 #if SIZEOF_SOCKLEN_T - 0 == 0
  38 #define socklen_t int
  39 #endif
  40
  41 static void     closedown(int sig);
  42 static int      makesock(int port, int proto);
  43
  44 #define _RPCSVC_CLOSEDOWN       120
  45 int     _rpcpmstart = 0;
  46 int     _rpcfdtype = 0;
  47 int     _rpcsvcdirty = 0;
  48
  49 void
  50 rpc_init(char *name, int prog, int vers,
  51          void (*dispatch)(struct svc_req *, register SVCXPRT *),
  52          int defport)
  53 {
  54         struct sockaddr_in saddr;
  55         SVCXPRT *transp;
  56         int     sock;
  57         socklen_t asize;
  58
  59         asize = sizeof(saddr);
  60         sock = 0;
  61         if (getsockname(0, (struct sockaddr *) &saddr, &asize) == 0
  62             && saddr.sin_family == AF_INET) {
  63                 socklen_t ssize = sizeof (int);
  64                 int fdtype = 0;
  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).
  70                  */
  71                 if (!(fdtype == SOCK_STREAM && listen(0,5) == -1)) {
  72                         _rpcfdtype = fdtype;
  73                         _rpcpmstart = 1;
  74                 }
  75         }
  76         if (!_rpcpmstart) {
  77                 pmap_unset(prog, vers);
  78                 sock = RPC_ANYSOCK;
  79         }
  80
  81         if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_DGRAM)) {
  82                 static SVCXPRT *last_transp = NULL;
  83
  84                 if (_rpcpmstart == 0) {
  85                         if (last_transp
  86                             && (!defport || defport == last_transp->xp_port)) {
  87                                 transp = last_transp;
  88                                 goto udp_transport;
  89                         }
  90                         if (defport == 0)
  91                                 sock = RPC_ANYSOCK;
  92                         else if ((sock = makesock(defport, IPPROTO_UDP)) < 0) {
  93                                 xlog(L_FATAL, "%s: cannot make a UDP socket\n",
  94                                                 name);
  95                         }
  96                 }
  97                 if (sock == RPC_ANYSOCK)
  98                         sock = svcudp_socket (prog, 1);
  99                 transp = svcudp_create(sock);
 100                 if (transp == NULL) {
 101                         xlog(L_FATAL, "cannot create udp service.");
 102                 }
 103       udp_transport:
 104                 if (!svc_register(transp, prog, vers, dispatch, IPPROTO_UDP)) {
 105                         xlog(L_FATAL, "unable to register (%s, %d, udp).",
 106                                         name, vers);
 107                 }
 108                 last_transp = transp;
 109         }
 110
 111         if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_STREAM)) {
 112                 static SVCXPRT *last_transp = NULL;
 113
 114                 if (_rpcpmstart == 0) {
 115                         if (last_transp
 116                             && (!defport || defport == last_transp->xp_port)) {
 117                                 transp = last_transp;
 118                                 goto tcp_transport;
 119                         }
 120                         if (defport == 0)
 121                                 sock = RPC_ANYSOCK;
 122                         else if ((sock = makesock(defport, IPPROTO_TCP)) < 0) {
 123                                 xlog(L_FATAL, "%s: cannot make a TCP socket\n",
 124                                                 name);
 125                         }
 126                 }
 127                 if (sock == RPC_ANYSOCK)
 128                         sock = svctcp_socket (prog, 1);
 129                 transp = svctcp_create(sock, 0, 0);
 130                 if (transp == NULL) {
 131                         xlog(L_FATAL, "cannot create tcp service.");
 132                 }
 133       tcp_transport:
 134                 if (!svc_register(transp, prog, vers, dispatch, IPPROTO_TCP)) {
 135                         xlog(L_FATAL, "unable to register (%s, %d, tcp).",
 136                                         name, vers);
 137                 }
 138                 last_transp = transp;
 139         }
 140
 141         if (_rpcpmstart) {
 142                 signal (SIGALRM, closedown);
 143                 alarm (_RPCSVC_CLOSEDOWN);
 144         }
 145 }
 146
 147 static void closedown(sig)
 148 int sig;
 149 {
 150         (void) signal(sig, closedown);
 151         if (_rpcsvcdirty == 0) {
 152                 static int size;
 153                 int i, openfd;
 154
 155                 if (_rpcfdtype == SOCK_DGRAM)
 156                         exit(0);
 157                 if (size == 0) {
 158                         size = getdtablesize();
 159                 }
 160                 for (i = 0, openfd = 0; i < size && openfd < 2; i++)
 161                         if (FD_ISSET(i, &svc_fdset))
 162                                 openfd++;
 163                 if (openfd <= 1)
 164                         exit(0);
 165         }
 166         (void) alarm(_RPCSVC_CLOSEDOWN);
 167 }
 168
 169 /*
 170  * Create listener socket for a given port
 171  *
 172  * Return an open network socket on success; otherwise return -1
 173  * if some error occurs.
 174  */
 175 static int
 176 makesock(int port, int proto)
 177 {
 178         struct sockaddr_in sin;
 179         int     sock, sock_type, val;
 180
 181         sock_type = (proto == IPPROTO_UDP) ? SOCK_DGRAM : SOCK_STREAM;
 182         sock = socket(AF_INET, sock_type, proto);
 183         if (sock < 0) {
 184                 xlog(L_FATAL, "Could not make a socket: %s",
 185                                         strerror(errno));
 186                 return -1;
 187         }
 188         memset((char *) &sin, 0, sizeof(sin));
 189         sin.sin_family = AF_INET;
 190         sin.sin_addr.s_addr = htonl(INADDR_ANY);
 191         sin.sin_port = htons(port);
 192
 193         val = 1;
 194         if (proto == IPPROTO_TCP)
 195                 if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
 196                                &val, sizeof(val)) < 0)
 197                         xlog(L_ERROR, "setsockopt failed: %s",
 198                              strerror(errno));
 199
 200         if (bind(sock, (struct sockaddr *) &sin, sizeof(sin)) == -1) {
 201                 xlog(L_FATAL, "Could not bind name to socket: %s",
 202                                         strerror(errno));
 203                 return -1;
 204         }
 205
 206         return sock;
 207 }