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 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, void (*dispatch)(), int defport)
  51 {
  52         struct sockaddr_in saddr;
  53         SVCXPRT *transp;
  54         int     sock;
  55         socklen_t asize;
  56
  57         asize = sizeof(saddr);
  58         sock = 0;
  59         if (getsockname(0, (struct sockaddr *) &saddr, &asize) == 0
  60             && saddr.sin_family == AF_INET) {
  61                 socklen_t ssize = sizeof (int);
  62                 int fdtype = 0;
  63                 if (getsockopt(0, SOL_SOCKET, SO_TYPE,
  64                                 (char *)&fdtype, &ssize) == -1)
  65                         xlog(L_FATAL, "getsockopt failed: %s", strerror(errno));
  66                 /* inetd passes a UDP socket or a listening TCP socket.
  67                  * listen will fail on a connected TCP socket(passed by rsh).
  68                  */
  69                 if (!(fdtype == SOCK_STREAM && listen(0,5) == -1)) {
  70                         _rpcfdtype = fdtype;
  71                         _rpcpmstart = 1;
  72                 }
  73         }
  74         if (!_rpcpmstart) {
  75                 pmap_unset(prog, vers);
  76                 sock = RPC_ANYSOCK;
  77         }
  78
  79         if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_DGRAM)) {
  80                 static SVCXPRT *last_transp = NULL;
  81
  82                 if (_rpcpmstart == 0) {
  83                         if (last_transp
  84                             && (!defport || defport == last_transp->xp_port)) {
  85                                 transp = last_transp;
  86                                 goto udp_transport;
  87                         }
  88                         if (defport == 0)
  89                                 sock = RPC_ANYSOCK;
  90                         else if ((sock = makesock(defport, IPPROTO_UDP)) < 0) {
  91                                 xlog(L_FATAL, "%s: cannot make a UDP socket\n",
  92                                                 name);
  93                         }
  94                 }
  95                 if (sock == RPC_ANYSOCK)
  96                         sock = svcudp_socket (prog, 1);
  97                 transp = svcudp_create(sock);
  98                 if (transp == NULL) {
  99                         xlog(L_FATAL, "cannot create udp service.");
 100                 }
 101       udp_transport:
 102                 if (!svc_register(transp, prog, vers, dispatch, IPPROTO_UDP)) {
 103                         xlog(L_FATAL, "unable to register (%s, %d, udp).",
 104                                         name, vers);
 105                 }
 106                 last_transp = transp;
 107         }
 108
 109         if ((_rpcfdtype == 0) || (_rpcfdtype == SOCK_STREAM)) {
 110                 static SVCXPRT *last_transp = NULL;
 111
 112                 if (_rpcpmstart == 0) {
 113                         if (last_transp
 114                             && (!defport || defport == last_transp->xp_port)) {
 115                                 transp = last_transp;
 116                                 goto tcp_transport;
 117                         }
 118                         if (defport == 0)
 119                                 sock = RPC_ANYSOCK;
 120                         else if ((sock = makesock(defport, IPPROTO_TCP)) < 0) {
 121                                 xlog(L_FATAL, "%s: cannot make a TCP socket\n",
 122                                                 name);
 123                         }
 124                 }
 125                 if (sock == RPC_ANYSOCK)
 126                         sock = svctcp_socket (prog, 1);
 127                 transp = svctcp_create(sock, 0, 0);
 128                 if (transp == NULL) {
 129                         xlog(L_FATAL, "cannot create tcp service.");
 130                 }
 131       tcp_transport:
 132                 if (!svc_register(transp, prog, vers, dispatch, IPPROTO_TCP)) {
 133                         xlog(L_FATAL, "unable to register (%s, %d, tcp).",
 134                                         name, vers);
 135                 }
 136                 last_transp = transp;
 137         }
 138
 139         if (_rpcpmstart) {
 140                 signal (SIGALRM, closedown);
 141                 alarm (_RPCSVC_CLOSEDOWN);
 142         }
 143 }
 144
 145 static void closedown(sig)
 146 int sig;
 147 {
 148         (void) signal(sig, closedown);
 149         if (_rpcsvcdirty == 0) {
 150                 static int size;
 151                 int i, openfd;
 152
 153                 if (_rpcfdtype == SOCK_DGRAM)
 154                         exit(0);
 155                 if (size == 0) {
 156                         size = getdtablesize();
 157                 }
 158                 for (i = 0, openfd = 0; i < size && openfd < 2; i++)
 159                         if (FD_ISSET(i, &svc_fdset))
 160                                 openfd++;
 161                 if (openfd <= 1)
 162                         exit(0);
 163         }
 164         (void) alarm(_RPCSVC_CLOSEDOWN);
 165 }
 166
 167 int makesock(int port, int proto)
 168 {
 169         struct sockaddr_in sin;
 170         int     s;
 171         int     sock_type;
 172         int     val;
 173
 174         sock_type = (proto == IPPROTO_UDP) ? SOCK_DGRAM : SOCK_STREAM;
 175         s = socket(AF_INET, sock_type, proto);
 176         if (s < 0) {
 177                 xlog(L_FATAL, "Could not make a socket: %s\n",
 178                                         strerror(errno));
 179                 return (-1);
 180         }
 181         memset((char *) &sin, 0, sizeof(sin));
 182         sin.sin_family = AF_INET;
 183         sin.sin_addr.s_addr = INADDR_ANY;
 184         sin.sin_port = htons(port);
 185
 186         val = 1;
 187         if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) < 0)
 188                 xlog(L_ERROR, "setsockopt failed: %s\n", strerror(errno));
 189
 190 #if 0
 191         /* I was told it didn't work with gigabit ethernet.
 192            Don't bothet with it.  H.J. */
 193 #ifdef SO_SNDBUF
 194         {
 195                 int sblen, rblen;
 196
 197                 /* 1024 for rpc & transport overheads */
 198                 sblen = rblen = socksz + 1024;
 199                 if (setsockopt(s, SOL_SOCKET, SO_SNDBUF, &sblen, sizeof sblen) < 0 ||
 200                     setsockopt(s, SOL_SOCKET, SO_RCVBUF, &rblen, sizeof rblen) < 0)
 201                         xlog(L_ERROR, "setsockopt failed: %s\n", strerror(errno));
 202         }
 203 #endif                          /* SO_SNDBUF */
 204 #endif
 205
 206         if (bind(s, (struct sockaddr *) &sin, sizeof(sin)) == -1) {
 207                 xlog(L_FATAL, "Could not bind name to socket: %s\n",
 208                                         strerror(errno));
 209                 return (-1);
 210         }
 211         return (s);
 212 }
 213
 214
 215 /* Log an incoming call. */
 216 void
 217 rpc_logcall(struct svc_req *rqstp, char *xname, char *arg)
 218 {
 219         char            buff[1024];
 220         int             buflen=sizeof(buff);
 221         int             len;
 222         char            *sp;
 223         int             i;
 224
 225         if (!xlog_enabled(D_CALL))
 226                 return;
 227
 228         sp = buff;
 229         switch (rqstp->rq_cred.oa_flavor) {
 230         case AUTH_NULL:
 231                 sprintf(sp, "NULL");
 232                 break;
 233         case AUTH_UNIX: {
 234                 struct authunix_parms *unix_cred;
 235                 time_t time;
 236                 struct tm *tm;
 237
 238                 unix_cred = (struct authunix_parms *) rqstp->rq_clntcred;
 239                 time = unix_cred->aup_time;
 240                 tm = localtime(&time);
 241                 snprintf(sp, buflen, "UNIX %d/%d/%d %02d:%02d:%02d %s %d.%d",
 242                         tm->tm_year, tm->tm_mon + 1, tm->tm_mday,
 243                         tm->tm_hour, tm->tm_min, tm->tm_sec,
 244                         unix_cred->aup_machname,
 245                         unix_cred->aup_uid,
 246                         unix_cred->aup_gid);
 247                 sp[buflen-1] = 0;
 248                 len = strlen(sp);
 249                 sp += buflen;
 250                 buflen -= len;
 251                 if ((int) unix_cred->aup_len > 0) {
 252                         snprintf(sp, buflen, "+%d", unix_cred->aup_gids[0]);
 253                         sp[buflen-1] = 0;
 254                         len = strlen(sp);
 255                         sp += buflen;
 256                         buflen -= len;
 257                         for (i = 1; i < unix_cred->aup_len; i++) {
 258                                 snprintf(sp, buflen, ",%d",
 259                                         unix_cred->aup_gids[i]);
 260                                 sp[buflen-1] = 0;
 261                                 len = strlen(sp);
 262                                 sp += buflen;
 263                                 buflen -= len;
 264                         }
 265                 }
 266                 }
 267                 break;
 268         default:
 269                 sprintf(sp, "CRED %d", rqstp->rq_cred.oa_flavor);
 270         }
 271         xlog(D_CALL, "%s [%s]\n\t%s\n", xname, buff, arg);
 272 }