xref: /dragonfly/usr.sbin/rpc.lockd/lockd.c (revision 65cc0652) 
1 /*
2  * Copyright (c) 1995
3  *	A.R. Gordon (andrew.gordon@net-tel.co.uk).  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed for the FreeBSD project
16  * 4. Neither the name of the author nor the names of any co-contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY ANDREW GORDON AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  *
32  * $NetBSD: lockd.c,v 1.7 2000/08/12 18:08:44 thorpej Exp $
33  * $FreeBSD: src/usr.sbin/rpc.lockd/lockd.c,v 1.6 2001/03/19 12:50:09 alfred Exp $
34  */
35 
36 /*
37  * main() function for NFS lock daemon.  Most of the code in this
38  * file was generated by running rpcgen /usr/include/rpcsvc/nlm_prot.x.
39  *
40  * The actual program logic is in the file lock_proc.c
41  */
42 
43 #include <sys/types.h>
44 #include <sys/socket.h>
45 
46 #include <netinet/in.h>
47 #include <arpa/inet.h>
48 
49 #include <err.h>
50 #include <stdio.h>
51 #include <stdlib.h>
52 #include <errno.h>
53 #include <syslog.h>
54 #include <signal.h>
55 #include <string.h>
56 #include <unistd.h>
57 #include <libutil.h>
58 #include <netconfig.h>
59 #include <netdb.h>
60 
61 #include <rpc/rpc.h>
62 #include <rpc/rpc_com.h>
63 #include <rpcsvc/sm_inter.h>
64 
65 #include "lockd.h"
66 #include <rpcsvc/nlm_prot.h>
67 
68 int		debug_level = 0;	/* 0 = no debugging syslog() calls */
69 int		_rpcsvcdirty = 0;
70 
71 int grace_expired;
72 char **hosts, *svcport_str = NULL;
73 int nhosts = 0;
74 int xcreated = 0;
75 
76 #ifndef IPPORT_MAX
77 #define IPPORT_MAX	65535
78 #endif
79 
80 void	create_service(struct netconfig *);
81 
82 void	nlm_prog_0(struct svc_req *, SVCXPRT *);
83 void	nlm_prog_1(struct svc_req *, SVCXPRT *);
84 void	nlm_prog_3(struct svc_req *, SVCXPRT *);
85 void	nlm_prog_4(struct svc_req *, SVCXPRT *);
86 void	out_of_mem(void) __dead2;
87 
88 static void	usage(void) __dead2;
89 
90 void	sigalarm_handler(void);
91 
92 int
93 main(int argc, char **argv)
94 {
95 	int ch, i, s;
96 	void *nc_handle;
97 	char *endptr, **hosts_bak;
98 	struct sigaction sigalarm;
99 	int grace_period = 30;
100 	struct netconfig *nconf;
101 	int have_v6 = 1;
102 	int maxrec = RPC_MAXDATASIZE;
103 	in_port_t svcport = 0;
104 
105 	while ((ch = getopt(argc, argv, "d:g:h:p:")) != (-1)) {
106 		switch (ch) {
107 		case 'd':
108 			debug_level = atoi(optarg);
109 			if (!debug_level) {
110 				usage();
111 				/* NOTREACHED */
112 			}
113 			break;
114 		case 'g':
115 			grace_period = atoi(optarg);
116 			if (!grace_period) {
117 				usage();
118 				/* NOTREACHED */
119 			}
120 			break;
121 		case 'h':
122 			++nhosts;
123 			hosts_bak = hosts;
124 			hosts_bak = realloc(hosts, nhosts * sizeof(char *));
125 			if (hosts_bak == NULL) {
126 				if (hosts != NULL) {
127 					for (i = 0; i < nhosts; i++)
128 						free(hosts[i]);
129 					free(hosts);
130 					out_of_mem();
131 				}
132 			}
133 			hosts = hosts_bak;
134 			hosts[nhosts - 1] = strdup(optarg);
135 			if (hosts[nhosts - 1] == NULL) {
136 				for (i = 0; i < (nhosts - 1); i++)
137 					free(hosts[i]);
138 				free(hosts);
139 				out_of_mem();
140 			}
141 			break;
142 		case 'p':
143 			endptr = NULL;
144 			svcport = (in_port_t)strtoul(optarg, &endptr, 10);
145 			if (endptr == NULL || *endptr != '\0' ||
146 			    svcport == 0 || svcport >= IPPORT_MAX)
147 				usage();
148 			svcport_str = strdup(optarg);
149 			break;
150 		default:
151 		case '?':
152 			usage();
153 			/* NOTREACHED */
154 		}
155 	}
156 	if (geteuid()) { /* This command allowed only to root */
157 		fprintf(stderr, "Sorry. You are not superuser\n");
158 		exit(1);
159         }
160 
161 	rpcb_unset(NLM_PROG, NLM_SM, NULL);
162 	rpcb_unset(NLM_PROG, NLM_VERS, NULL);
163 	rpcb_unset(NLM_PROG, NLM_VERSX, NULL);
164 	rpcb_unset(NLM_PROG, NLM_VERS4, NULL);
165 
166 	/*
167 	 * Check if IPv6 support is present.
168 	 */
169 	s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
170 	if (s < 0)
171 		have_v6 = 0;
172 	else
173 		close(s);
174 
175 	rpc_control(RPC_SVC_CONNMAXREC_SET, &maxrec);
176 
177 	/*
178 	 * If no hosts were specified, add a wildcard entry to bind to
179 	 * INADDR_ANY. Otherwise make sure 127.0.0.1 and ::1 are added to the
180 	 * list.
181 	 */
182 	if (nhosts == 0) {
183 		hosts = malloc(sizeof(char**));
184 		if (hosts == NULL)
185 			out_of_mem();
186 
187 		hosts[0] = "*";
188 		nhosts = 1;
189 	} else {
190 		hosts_bak = hosts;
191 		if (have_v6) {
192 			hosts_bak = realloc(hosts, (nhosts + 2) *
193 			    sizeof(char *));
194 			if (hosts_bak == NULL) {
195 				for (i = 0; i < nhosts; i++)
196 					free(hosts[i]);
197 				free(hosts);
198 				out_of_mem();
199 			} else
200 				hosts = hosts_bak;
201 
202 			nhosts += 2;
203 			hosts[nhosts - 2] = "::1";
204 		} else {
205 			hosts_bak = realloc(hosts, (nhosts + 1) * sizeof(char *));
206 			if (hosts_bak == NULL) {
207 				for (i = 0; i < nhosts; i++)
208 					free(hosts[i]);
209 
210 				free(hosts);
211 				out_of_mem();
212 			} else {
213 				nhosts += 1;
214 				hosts = hosts_bak;
215 			}
216 		}
217 		hosts[nhosts - 1] = "127.0.0.1";
218 	}
219 
220 	nc_handle = setnetconfig();
221 	while ((nconf = getnetconfig(nc_handle))) {
222 		/* We want to listen only on udp6, tcp6, udp, tcp transports */
223 		if (nconf->nc_flag & NC_VISIBLE) {
224 			/* Skip if there's no IPv6 support */
225 			if (have_v6 == 0 && strcmp(nconf->nc_protofmly, "inet6") == 0) {
226 				/* DO NOTHING */
227 			} else {
228 				create_service(nconf);
229 			}
230 		}
231 	}
232 	endnetconfig(nc_handle);
233 
234 	/*
235 	 * Note that it is NOT sensible to run this program from inetd - the
236 	 * protocol assumes that it will run immediately at boot time.
237 	 */
238 	if (daemon(0, debug_level > 0)) {
239 		err(1, "cannot fork");
240 		/* NOTREACHED */
241 	}
242 
243 	openlog("rpc.lockd", 0, LOG_DAEMON);
244 	if (debug_level)
245 		syslog(LOG_INFO, "Starting, debug level %d", debug_level);
246 	else
247 		syslog(LOG_INFO, "Starting");
248 
249 	sigalarm.sa_handler = (sig_t) sigalarm_handler;
250 	sigemptyset(&sigalarm.sa_mask);
251 	sigalarm.sa_flags = SA_RESETHAND; /* should only happen once */
252 	sigalarm.sa_flags |= SA_RESTART;
253 	if (sigaction(SIGALRM, &sigalarm, NULL) != 0) {
254 		syslog(LOG_WARNING, "sigaction(SIGALRM) failed: %s",
255 		    strerror(errno));
256 		exit(1);
257 	}
258 	grace_expired = 0;
259 	alarm(grace_period);
260 
261 	svc_run();		/* Should never return */
262 	exit(1);
263 }
264 
265 /*
266  * This routine creates and binds sockets on the appropriate
267  * addresses. It gets called one time for each transport and
268  * registrates the service with rpcbind on that trasport.
269  */
270 void
271 create_service(struct netconfig *nconf)
272 {
273 	struct addrinfo hints, *res = NULL;
274 	struct sockaddr_in *sin;
275 	struct sockaddr_in6 *sin6;
276 	struct __rpc_sockinfo si;
277 	struct netbuf servaddr;
278 	SVCXPRT	*transp = NULL;
279 	int aicode;
280 	int fd;
281 	int nhostsbak;
282 	int r;
283 	int registered = 0;
284 	u_int32_t host_addr[4];  /* IPv4 or IPv6 */
285 
286 	if ((nconf->nc_semantics != NC_TPI_CLTS) &&
287 	    (nconf->nc_semantics != NC_TPI_COTS) &&
288 	    (nconf->nc_semantics != NC_TPI_COTS_ORD))
289 		return;	/* not my type */
290 
291 	/*
292 	 * XXX - using RPC library internal functions.
293 	 */
294 	if (!__rpc_nconf2sockinfo(nconf, &si)) {
295 		syslog(LOG_ERR, "cannot get information for %s",
296 		    nconf->nc_netid);
297 		return;
298 	}
299 
300 	/* Get rpc.statd's address on this transport */
301 	memset(&hints, 0, sizeof hints);
302 	hints.ai_flags = AI_PASSIVE;
303 	hints.ai_family = si.si_af;
304 	hints.ai_socktype = si.si_socktype;
305 	hints.ai_protocol = si.si_proto;
306 
307 	/*
308 	 * Bind to specific IPs if asked to
309 	 */
310 	nhostsbak = nhosts;
311 	while (nhostsbak > 0) {
312 		--nhostsbak;
313 
314 		/*
315 		 * XXX - using RPC library internal functions.
316 		 */
317 		if ((fd = __rpc_nconf2fd(nconf)) < 0) {
318 			syslog(LOG_ERR, "cannot create socket for %s",
319 			    nconf->nc_netid);
320 			continue;
321 		}
322 
323 		switch (hints.ai_family) {
324 			case AF_INET:
325 				if (inet_pton(AF_INET, hosts[nhostsbak],
326 				    host_addr) == 1) {
327 					hints.ai_flags &= AI_NUMERICHOST;
328 				} else {
329 					/*
330 					 * Skip if we have an AF_INET6 address.
331 					 */
332 					if (inet_pton(AF_INET6, hosts[nhostsbak],
333 					    host_addr) == 1) {
334 						close(fd);
335 						continue;
336 					}
337 				}
338 				break;
339 			case AF_INET6:
340 				if (inet_pton(AF_INET6, hosts[nhostsbak],
341 				    host_addr) == 1) {
342 					hints.ai_flags &= AI_NUMERICHOST;
343 				} else {
344 					/*
345 					 * Skip if we have an AF_INET address.
346 					 */
347 					if (inet_pton(AF_INET, hosts[nhostsbak],
348 					    host_addr) == 1) {
349 						close(fd);
350 						continue;
351 					}
352 				}
353 				break;
354 			default:
355 				break;
356 		}
357 
358 		/*
359 		 * If no hosts were specified, just bind to INADDR_ANY
360 		 */
361 		if (strcmp("*", hosts[nhostsbak]) == 0) {
362 			if (svcport_str == NULL) {
363 				res = malloc(sizeof(struct addrinfo));
364 				if (res == NULL)
365 					out_of_mem();
366 				res->ai_flags = hints.ai_flags;
367 				res->ai_family = hints.ai_family;
368 				res->ai_protocol = hints.ai_protocol;
369 				switch (res->ai_family) {
370 					case AF_INET:
371 						sin = malloc(sizeof(struct sockaddr_in));
372 						if (sin == NULL)
373 							out_of_mem();
374 						sin->sin_family = AF_INET;
375 						sin->sin_port = htons(0);
376 						sin->sin_addr.s_addr = htonl(INADDR_ANY);
377 						res->ai_addr = (struct sockaddr*) sin;
378 						res->ai_addrlen = (socklen_t)
379 						    sizeof(res->ai_addr);
380 						break;
381 					case AF_INET6:
382 						sin6 = malloc(sizeof(struct sockaddr_in6));
383 						if (sin6 == NULL)
384 							out_of_mem();
385 						sin6->sin6_family = AF_INET6;
386 						sin6->sin6_port = htons(0);
387 						sin6->sin6_addr = in6addr_any;
388 						res->ai_addr = (struct sockaddr*) sin6;
389 						res->ai_addrlen = (socklen_t) sizeof(res->ai_addr);
390 						break;
391 					default:
392 						break;
393 				}
394 			} else {
395 				if ((aicode = getaddrinfo(NULL, svcport_str,
396 				    &hints, &res)) != 0) {
397 					syslog(LOG_ERR,
398 					    "cannot get local address for %s: %s",
399 					    nconf->nc_netid,
400 					    gai_strerror(aicode));
401 					continue;
402 				}
403 			}
404 		} else {
405 			if ((aicode = getaddrinfo(hosts[nhostsbak], svcport_str,
406 			    &hints, &res)) != 0) {
407 				syslog(LOG_ERR,
408 				    "cannot get local address for %s: %s",
409 				    nconf->nc_netid, gai_strerror(aicode));
410 				continue;
411 			}
412 		}
413 
414 		r = bindresvport_sa(fd, res->ai_addr);
415 		if (r != 0) {
416 			syslog(LOG_ERR, "bindresvport_sa: %m");
417 			exit(1);
418 		}
419 
420 		transp = svc_tli_create(fd, nconf, NULL,
421 		    RPC_MAXDATASIZE, RPC_MAXDATASIZE);
422 
423 		if (transp != NULL) {
424 			if (!svc_reg(transp, NLM_PROG, NLM_SM, nlm_prog_0,
425 			    NULL))
426 				syslog(LOG_ERR,
427 				    "can't register %s NLM_PROG, NLM_SM service",
428 				    nconf->nc_netid);
429 
430 			if (!svc_reg(transp, NLM_PROG, NLM_VERS, nlm_prog_1,
431 			    NULL))
432 				syslog(LOG_ERR,
433 				    "can't register %s NLM_PROG, NLM_VERS service",
434 				    nconf->nc_netid);
435 
436 			if (!svc_reg(transp, NLM_PROG, NLM_VERSX, nlm_prog_3,
437 			    NULL))
438 				syslog(LOG_ERR,
439 				    "can't register %s NLM_PROG, NLM_VERSX service",
440 				    nconf->nc_netid);
441 
442 			if (!svc_reg(transp, NLM_PROG, NLM_VERS4, nlm_prog_4,
443 			    NULL))
444 				syslog(LOG_ERR,
445 				    "can't register %s NLM_PROG, NLM_VERS4 service",
446 				    nconf->nc_netid);
447 
448 		} else
449 			syslog(LOG_WARNING, "can't create %s services",
450 			    nconf->nc_netid);
451 
452 		if (registered == 0) {
453 			registered = 1;
454 			memset(&hints, 0, sizeof hints);
455 			hints.ai_flags = AI_PASSIVE;
456 			hints.ai_family = si.si_af;
457 			hints.ai_socktype = si.si_socktype;
458 			hints.ai_protocol = si.si_proto;
459 
460 			if (svcport_str == NULL) {
461 				svcport_str = malloc(NI_MAXSERV * sizeof(char));
462 				if (svcport_str == NULL)
463 					out_of_mem();
464 
465 				if (getnameinfo(res->ai_addr,
466 				    res->ai_addr->sa_len, NULL, NI_MAXHOST,
467 				    svcport_str, NI_MAXSERV * sizeof(char),
468 				    NI_NUMERICHOST | NI_NUMERICSERV))
469 					errx(1, "Cannot get port number");
470 			}
471 
472 			if((aicode = getaddrinfo(NULL, svcport_str, &hints,
473 			    &res)) != 0) {
474 				syslog(LOG_ERR, "cannot get local address: %s",
475 				    gai_strerror(aicode));
476 				exit(1);
477 			}
478 
479 			servaddr.buf = malloc(res->ai_addrlen);
480 			memcpy(servaddr.buf, res->ai_addr, res->ai_addrlen);
481 			servaddr.len = res->ai_addrlen;
482 
483 			rpcb_set(NLM_PROG, NLM_SM, nconf, &servaddr);
484 			rpcb_set(NLM_PROG, NLM_VERS, nconf, &servaddr);
485 			rpcb_set(NLM_PROG, NLM_VERSX, nconf, &servaddr);
486 			rpcb_set(NLM_PROG, NLM_VERS4, nconf, &servaddr);
487 
488 			xcreated++;
489 			freeaddrinfo(res);
490 		}
491 	} /* end while */
492 }
493 
494 void
495 sigalarm_handler(void)
496 {
497 
498 	grace_expired = 1;
499 }
500 
501 static void
502 usage(void)
503 {
504 	errx(1, "usage: rpc.lockd [-d <debuglevel>]"
505 	    " [-g <grace period>] [-h <bindip>] [-p <port>]");
506 }
507 
508 /*
509  * Out of memory, fatal
510  */
511 void
512 out_of_mem(void)
513 {
514 	syslog(LOG_ERR, "out of memory");
515 	exit(2);
516 }
517