Detect if glibc provides socklen_t and use that instead
[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 #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 }