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