1 /*-
2 * SPDX-License-Identifier: BSD-4-Clause
3 *
4 * Copyright (c) 1995
5 * A.R. Gordon ([email protected]). All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. All advertising materials mentioning features or use of this software
16 * must display the following acknowledgement:
17 * This product includes software developed for the FreeBSD project
18 * 4. Neither the name of the author nor the names of any co-contributors
19 * may be used to endorse or promote products derived from this software
20 * without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY ANDREW GORDON AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 *
34 */
35
36 /* main() function for status monitor daemon. Some of the code in this */
37 /* file was generated by running rpcgen /usr/include/rpcsvc/sm_inter.x */
38 /* The actual program logic is in the file procs.c */
39
40 #include <sys/cdefs.h>
41 __FBSDID("$FreeBSD$");
42
43 #include <err.h>
44 #include <errno.h>
45 #include <stdio.h>
46 #include <stdlib.h>
47 #include <rpc/rpc.h>
48 #include <rpc/rpc_com.h>
49 #include <string.h>
50 #include <syslog.h>
51 #include <sys/types.h>
52 #include <sys/socket.h>
53 #include <sys/wait.h>
54 #include <netinet/in.h>
55 #include <arpa/inet.h>
56 #include <netdb.h>
57 #include <signal.h>
58 #include <unistd.h>
59 #include "statd.h"
60
61 #define GETPORT_MAXTRY 20 /* Max tries to get a port # */
62
63 int debug = 0; /* Controls syslog() calls for debug messages */
64
65 static char **hosts, *svcport_str = NULL;
66 static int nhosts = 0;
67 static int xcreated = 0;
68 static int mallocd_svcport = 0;
69 static int *sock_fd;
70 static int sock_fdcnt;
71 static int sock_fdpos;
72
73 static int create_service(struct netconfig *nconf);
74 static void complete_service(struct netconfig *nconf, char *port_str);
75 static void clearout_service(void);
76 static void handle_sigchld(int sig);
77 void out_of_mem(void) __dead2;
78
79 static void usage(void) __dead2;
80
81 int
main(int argc,char ** argv)82 main(int argc, char **argv)
83 {
84 struct sigaction sa;
85 struct netconfig *nconf;
86 void *nc_handle;
87 in_port_t svcport;
88 int ch, i, s;
89 char *endptr;
90 char **hosts_bak;
91 int have_v6 = 1;
92 int foreground = 0;
93 int maxrec = RPC_MAXDATASIZE;
94 int attempt_cnt, port_len, port_pos, ret;
95 char **port_list;
96
97 while ((ch = getopt(argc, argv, "dFh:p:")) != -1)
98 switch (ch) {
99 case 'd':
100 debug = 1;
101 break;
102 case 'F':
103 foreground = 1;
104 break;
105 case 'h':
106 ++nhosts;
107 hosts_bak = hosts;
108 hosts_bak = realloc(hosts, nhosts * sizeof(char *));
109 if (hosts_bak == NULL) {
110 if (hosts != NULL) {
111 for (i = 0; i < nhosts; i++)
112 free(hosts[i]);
113 free(hosts);
114 out_of_mem();
115 }
116 }
117 hosts = hosts_bak;
118 hosts[nhosts - 1] = strdup(optarg);
119 if (hosts[nhosts - 1] == NULL) {
120 for (i = 0; i < (nhosts - 1); i++)
121 free(hosts[i]);
122 free(hosts);
123 out_of_mem();
124 }
125 break;
126 case 'p':
127 endptr = NULL;
128 svcport = (in_port_t)strtoul(optarg, &endptr, 10);
129 if (endptr == NULL || *endptr != '\0' || svcport == 0 ||
130 svcport >= IPPORT_MAX)
131 usage();
132
133 svcport_str = strdup(optarg);
134 break;
135 default:
136 usage();
137 }
138 argc -= optind;
139 argv += optind;
140
141 (void)rpcb_unset(SM_PROG, SM_VERS, NULL);
142
143 /*
144 * Check if IPv6 support is present.
145 */
146 s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
147 if (s < 0)
148 have_v6 = 0;
149 else
150 close(s);
151
152 rpc_control(RPC_SVC_CONNMAXREC_SET, &maxrec);
153
154 /*
155 * If no hosts were specified, add a wildcard entry to bind to
156 * INADDR_ANY. Otherwise make sure 127.0.0.1 and ::1 are added to the
157 * list.
158 */
159 if (nhosts == 0) {
160 hosts = malloc(sizeof(char *));
161 if (hosts == NULL)
162 out_of_mem();
163
164 hosts[0] = strdup("*");
165 nhosts = 1;
166 } else {
167 hosts_bak = hosts;
168 if (have_v6) {
169 hosts_bak = realloc(hosts, (nhosts + 2) *
170 sizeof(char *));
171 if (hosts_bak == NULL) {
172 for (i = 0; i < nhosts; i++)
173 free(hosts[i]);
174 free(hosts);
175 out_of_mem();
176 } else
177 hosts = hosts_bak;
178
179 nhosts += 2;
180 hosts[nhosts - 2] = strdup("::1");
181 } else {
182 hosts_bak = realloc(hosts, (nhosts + 1) * sizeof(char *));
183 if (hosts_bak == NULL) {
184 for (i = 0; i < nhosts; i++)
185 free(hosts[i]);
186
187 free(hosts);
188 out_of_mem();
189 } else {
190 nhosts += 1;
191 hosts = hosts_bak;
192 }
193 }
194 hosts[nhosts - 1] = strdup("127.0.0.1");
195 }
196
197 attempt_cnt = 1;
198 sock_fdcnt = 0;
199 sock_fd = NULL;
200 port_list = NULL;
201 port_len = 0;
202 nc_handle = setnetconfig();
203 while ((nconf = getnetconfig(nc_handle))) {
204 /* We want to listen only on udp6, tcp6, udp, tcp transports */
205 if (nconf->nc_flag & NC_VISIBLE) {
206 /* Skip if there's no IPv6 support */
207 if (have_v6 == 0 && strcmp(nconf->nc_protofmly, "inet6") == 0) {
208 /* DO NOTHING */
209 } else {
210 ret = create_service(nconf);
211 if (ret == 1)
212 /* Ignore this call */
213 continue;
214 if (ret < 0) {
215 /*
216 * Failed to bind port, so close off
217 * all sockets created and try again
218 * if the port# was dynamically
219 * assigned via bind(2).
220 */
221 clearout_service();
222 if (mallocd_svcport != 0 &&
223 attempt_cnt < GETPORT_MAXTRY) {
224 free(svcport_str);
225 svcport_str = NULL;
226 mallocd_svcport = 0;
227 } else {
228 errno = EADDRINUSE;
229 syslog(LOG_ERR,
230 "bindresvport_sa: %m");
231 exit(1);
232 }
233
234 /* Start over at the first service. */
235 free(sock_fd);
236 sock_fdcnt = 0;
237 sock_fd = NULL;
238 nc_handle = setnetconfig();
239 attempt_cnt++;
240 } else if (mallocd_svcport != 0 &&
241 attempt_cnt == GETPORT_MAXTRY) {
242 /*
243 * For the last attempt, allow
244 * different port #s for each nconf
245 * by saving the svcport_str and
246 * setting it back to NULL.
247 */
248 port_list = realloc(port_list,
249 (port_len + 1) * sizeof(char *));
250 if (port_list == NULL)
251 out_of_mem();
252 port_list[port_len++] = svcport_str;
253 svcport_str = NULL;
254 mallocd_svcport = 0;
255 }
256 }
257 }
258 }
259
260 /*
261 * Successfully bound the ports, so call complete_service() to
262 * do the rest of the setup on the service(s).
263 */
264 sock_fdpos = 0;
265 port_pos = 0;
266 nc_handle = setnetconfig();
267 while ((nconf = getnetconfig(nc_handle))) {
268 /* We want to listen only on udp6, tcp6, udp, tcp transports */
269 if (nconf->nc_flag & NC_VISIBLE) {
270 /* Skip if there's no IPv6 support */
271 if (have_v6 == 0 && strcmp(nconf->nc_protofmly, "inet6") == 0) {
272 /* DO NOTHING */
273 } else if (port_list != NULL) {
274 if (port_pos >= port_len) {
275 syslog(LOG_ERR, "too many port#s");
276 exit(1);
277 }
278 complete_service(nconf, port_list[port_pos++]);
279 } else
280 complete_service(nconf, svcport_str);
281 }
282 }
283 endnetconfig(nc_handle);
284 free(sock_fd);
285 if (port_list != NULL) {
286 for (port_pos = 0; port_pos < port_len; port_pos++)
287 free(port_list[port_pos]);
288 free(port_list);
289 }
290
291 init_file("/var/db/statd.status");
292
293 /* Note that it is NOT sensible to run this program from inetd - the */
294 /* protocol assumes that it will run immediately at boot time. */
295 if ((foreground == 0) && daemon(0, 0) < 0) {
296 err(1, "cannot fork");
297 /* NOTREACHED */
298 }
299
300 openlog("rpc.statd", 0, LOG_DAEMON);
301 if (debug) syslog(LOG_INFO, "Starting - debug enabled");
302 else syslog(LOG_INFO, "Starting");
303
304 /* Install signal handler to collect exit status of child processes */
305 sa.sa_handler = handle_sigchld;
306 sigemptyset(&sa.sa_mask);
307 sigaddset(&sa.sa_mask, SIGCHLD);
308 sa.sa_flags = SA_RESTART;
309 sigaction(SIGCHLD, &sa, NULL);
310
311 /* Initialisation now complete - start operating */
312 notify_hosts(); /* Forks a process (if necessary) to do the */
313 /* SM_NOTIFY calls, which may be slow. */
314
315 svc_run(); /* Should never return */
316 exit(1);
317 }
318
319 /*
320 * This routine creates and binds sockets on the appropriate
321 * addresses. It gets called one time for each transport.
322 * It returns 0 upon success, 1 for ingore the call and -1 to indicate
323 * bind failed with EADDRINUSE.
324 * Any file descriptors that have been created are stored in sock_fd and
325 * the total count of them is maintained in sock_fdcnt.
326 */
327 static int
create_service(struct netconfig * nconf)328 create_service(struct netconfig *nconf)
329 {
330 struct addrinfo hints, *res = NULL;
331 struct sockaddr_in *sin;
332 struct sockaddr_in6 *sin6;
333 struct __rpc_sockinfo si;
334 int aicode;
335 int fd;
336 int nhostsbak;
337 int r;
338 u_int32_t host_addr[4]; /* IPv4 or IPv6 */
339 int mallocd_res;
340
341 if ((nconf->nc_semantics != NC_TPI_CLTS) &&
342 (nconf->nc_semantics != NC_TPI_COTS) &&
343 (nconf->nc_semantics != NC_TPI_COTS_ORD))
344 return (1); /* not my type */
345
346 /*
347 * XXX - using RPC library internal functions.
348 */
349 if (!__rpc_nconf2sockinfo(nconf, &si)) {
350 syslog(LOG_ERR, "cannot get information for %s",
351 nconf->nc_netid);
352 return (1);
353 }
354
355 /* Get rpc.statd's address on this transport */
356 memset(&hints, 0, sizeof hints);
357 hints.ai_family = si.si_af;
358 hints.ai_socktype = si.si_socktype;
359 hints.ai_protocol = si.si_proto;
360
361 /*
362 * Bind to specific IPs if asked to
363 */
364 nhostsbak = nhosts;
365 while (nhostsbak > 0) {
366 --nhostsbak;
367 sock_fd = realloc(sock_fd, (sock_fdcnt + 1) * sizeof(int));
368 if (sock_fd == NULL)
369 out_of_mem();
370 sock_fd[sock_fdcnt++] = -1; /* Set invalid for now. */
371 mallocd_res = 0;
372 hints.ai_flags = AI_PASSIVE;
373
374 /*
375 * XXX - using RPC library internal functions.
376 */
377 if ((fd = __rpc_nconf2fd(nconf)) < 0) {
378 syslog(LOG_ERR, "cannot create socket for %s",
379 nconf->nc_netid);
380 continue;
381 }
382 switch (hints.ai_family) {
383 case AF_INET:
384 if (inet_pton(AF_INET, hosts[nhostsbak],
385 host_addr) == 1) {
386 hints.ai_flags |= AI_NUMERICHOST;
387 } else {
388 /*
389 * Skip if we have an AF_INET6 address.
390 */
391 if (inet_pton(AF_INET6, hosts[nhostsbak],
392 host_addr) == 1) {
393 close(fd);
394 continue;
395 }
396 }
397 break;
398 case AF_INET6:
399 if (inet_pton(AF_INET6, hosts[nhostsbak],
400 host_addr) == 1) {
401 hints.ai_flags |= AI_NUMERICHOST;
402 } else {
403 /*
404 * Skip if we have an AF_INET address.
405 */
406 if (inet_pton(AF_INET, hosts[nhostsbak],
407 host_addr) == 1) {
408 close(fd);
409 continue;
410 }
411 }
412 break;
413 default:
414 break;
415 }
416
417 /*
418 * If no hosts were specified, just bind to INADDR_ANY
419 */
420 if (strcmp("*", hosts[nhostsbak]) == 0) {
421 if (svcport_str == NULL) {
422 res = malloc(sizeof(struct addrinfo));
423 if (res == NULL)
424 out_of_mem();
425 mallocd_res = 1;
426 res->ai_flags = hints.ai_flags;
427 res->ai_family = hints.ai_family;
428 res->ai_protocol = hints.ai_protocol;
429 switch (res->ai_family) {
430 case AF_INET:
431 sin = malloc(sizeof(struct sockaddr_in));
432 if (sin == NULL)
433 out_of_mem();
434 sin->sin_family = AF_INET;
435 sin->sin_port = htons(0);
436 sin->sin_addr.s_addr = htonl(INADDR_ANY);
437 res->ai_addr = (struct sockaddr*) sin;
438 res->ai_addrlen = (socklen_t)
439 sizeof(struct sockaddr_in);
440 break;
441 case AF_INET6:
442 sin6 = malloc(sizeof(struct sockaddr_in6));
443 if (sin6 == NULL)
444 out_of_mem();
445 sin6->sin6_family = AF_INET6;
446 sin6->sin6_port = htons(0);
447 sin6->sin6_addr = in6addr_any;
448 res->ai_addr = (struct sockaddr*) sin6;
449 res->ai_addrlen = (socklen_t)
450 sizeof(struct sockaddr_in6);
451 break;
452 default:
453 syslog(LOG_ERR, "bad addr fam %d",
454 res->ai_family);
455 exit(1);
456 }
457 } else {
458 if ((aicode = getaddrinfo(NULL, svcport_str,
459 &hints, &res)) != 0) {
460 syslog(LOG_ERR,
461 "cannot get local address for %s: %s",
462 nconf->nc_netid,
463 gai_strerror(aicode));
464 close(fd);
465 continue;
466 }
467 }
468 } else {
469 if ((aicode = getaddrinfo(hosts[nhostsbak], svcport_str,
470 &hints, &res)) != 0) {
471 syslog(LOG_ERR,
472 "cannot get local address for %s: %s",
473 nconf->nc_netid, gai_strerror(aicode));
474 close(fd);
475 continue;
476 }
477 }
478
479 /* Store the fd. */
480 sock_fd[sock_fdcnt - 1] = fd;
481
482 /* Now, attempt the bind. */
483 r = bindresvport_sa(fd, res->ai_addr);
484 if (r != 0) {
485 if (errno == EADDRINUSE && mallocd_svcport != 0) {
486 if (mallocd_res != 0) {
487 free(res->ai_addr);
488 free(res);
489 } else
490 freeaddrinfo(res);
491 return (-1);
492 }
493 syslog(LOG_ERR, "bindresvport_sa: %m");
494 exit(1);
495 }
496
497 if (svcport_str == NULL) {
498 svcport_str = malloc(NI_MAXSERV * sizeof(char));
499 if (svcport_str == NULL)
500 out_of_mem();
501 mallocd_svcport = 1;
502
503 if (getnameinfo(res->ai_addr,
504 res->ai_addr->sa_len, NULL, NI_MAXHOST,
505 svcport_str, NI_MAXSERV * sizeof(char),
506 NI_NUMERICHOST | NI_NUMERICSERV))
507 errx(1, "Cannot get port number");
508 }
509 if (mallocd_res != 0) {
510 free(res->ai_addr);
511 free(res);
512 } else
513 freeaddrinfo(res);
514 res = NULL;
515 }
516 return (0);
517 }
518
519 /*
520 * Called after all the create_service() calls have succeeded, to complete
521 * the setup and registration.
522 */
523 static void
complete_service(struct netconfig * nconf,char * port_str)524 complete_service(struct netconfig *nconf, char *port_str)
525 {
526 struct addrinfo hints, *res = NULL;
527 struct __rpc_sockinfo si;
528 struct netbuf servaddr;
529 SVCXPRT *transp = NULL;
530 int aicode, fd, nhostsbak;
531 int registered = 0;
532
533 if ((nconf->nc_semantics != NC_TPI_CLTS) &&
534 (nconf->nc_semantics != NC_TPI_COTS) &&
535 (nconf->nc_semantics != NC_TPI_COTS_ORD))
536 return; /* not my type */
537
538 /*
539 * XXX - using RPC library internal functions.
540 */
541 if (!__rpc_nconf2sockinfo(nconf, &si)) {
542 syslog(LOG_ERR, "cannot get information for %s",
543 nconf->nc_netid);
544 return;
545 }
546
547 nhostsbak = nhosts;
548 while (nhostsbak > 0) {
549 --nhostsbak;
550 if (sock_fdpos >= sock_fdcnt) {
551 /* Should never happen. */
552 syslog(LOG_ERR, "Ran out of socket fd's");
553 return;
554 }
555 fd = sock_fd[sock_fdpos++];
556 if (fd < 0)
557 continue;
558
559 if (nconf->nc_semantics != NC_TPI_CLTS)
560 listen(fd, SOMAXCONN);
561
562 transp = svc_tli_create(fd, nconf, NULL,
563 RPC_MAXDATASIZE, RPC_MAXDATASIZE);
564
565 if (transp != (SVCXPRT *) NULL) {
566 if (!svc_register(transp, SM_PROG, SM_VERS,
567 sm_prog_1, 0)) {
568 syslog(LOG_ERR, "can't register on %s",
569 nconf->nc_netid);
570 } else {
571 if (!svc_reg(transp, SM_PROG, SM_VERS,
572 sm_prog_1, NULL))
573 syslog(LOG_ERR,
574 "can't register %s SM_PROG service",
575 nconf->nc_netid);
576 }
577 } else
578 syslog(LOG_WARNING, "can't create %s services",
579 nconf->nc_netid);
580
581 if (registered == 0) {
582 registered = 1;
583 memset(&hints, 0, sizeof hints);
584 hints.ai_flags = AI_PASSIVE;
585 hints.ai_family = si.si_af;
586 hints.ai_socktype = si.si_socktype;
587 hints.ai_protocol = si.si_proto;
588
589
590 if ((aicode = getaddrinfo(NULL, port_str, &hints,
591 &res)) != 0) {
592 syslog(LOG_ERR, "cannot get local address: %s",
593 gai_strerror(aicode));
594 exit(1);
595 }
596
597 servaddr.buf = malloc(res->ai_addrlen);
598 memcpy(servaddr.buf, res->ai_addr, res->ai_addrlen);
599 servaddr.len = res->ai_addrlen;
600
601 rpcb_set(SM_PROG, SM_VERS, nconf, &servaddr);
602
603 xcreated++;
604 freeaddrinfo(res);
605 }
606 } /* end while */
607 }
608
609 /*
610 * Clear out sockets after a failure to bind one of them, so that the
611 * cycle of socket creation/binding can start anew.
612 */
613 static void
clearout_service(void)614 clearout_service(void)
615 {
616 int i;
617
618 for (i = 0; i < sock_fdcnt; i++) {
619 if (sock_fd[i] >= 0) {
620 shutdown(sock_fd[i], SHUT_RDWR);
621 close(sock_fd[i]);
622 }
623 }
624 }
625
626 static void
usage(void)627 usage(void)
628 {
629 fprintf(stderr, "usage: rpc.statd [-d] [-F] [-h <bindip>] [-p <port>]\n");
630 exit(1);
631 }
632
633 /* handle_sigchld ---------------------------------------------------------- */
634 /*
635 Purpose: Catch SIGCHLD and collect process status
636 Retruns: Nothing.
637 Notes: No special action required, other than to collect the
638 process status and hence allow the child to die:
639 we only use child processes for asynchronous transmission
640 of SM_NOTIFY to other systems, so it is normal for the
641 children to exit when they have done their work.
642 */
643
handle_sigchld(int sig __unused)644 static void handle_sigchld(int sig __unused)
645 {
646 int pid, status;
647 pid = wait4(-1, &status, WNOHANG, (struct rusage*)0);
648 if (!pid) syslog(LOG_ERR, "Phantom SIGCHLD??");
649 else if (status == 0)
650 {
651 if (debug) syslog(LOG_DEBUG, "Child %d exited OK", pid);
652 }
653 else syslog(LOG_ERR, "Child %d failed with status %d", pid,
654 WEXITSTATUS(status));
655 }
656
657 /*
658 * Out of memory, fatal
659 */
660 void
out_of_mem(void)661 out_of_mem(void)
662 {
663
664 syslog(LOG_ERR, "out of memory");
665 exit(2);
666 }
667