1 /*- 2 * Copyright (c) 1993, John Brezak 3 * 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 by the University of 16 * California, Berkeley and its contributors. 17 * 4. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 #ifndef lint 35 static const char rcsid[] = 36 "$FreeBSD$"; 37 #endif /* not lint */ 38 39 #include <stdlib.h> 40 #include <rpc/rpc.h> 41 #include <signal.h> 42 #include <syslog.h> 43 #include <rpc/pmap_clnt.h> 44 #include <rpcsvc/rstat.h> 45 46 extern void rstat_service(struct svc_req *, SVCXPRT *); 47 48 int from_inetd = 1; /* started from inetd ? */ 49 int closedown = 20; /* how long to wait before going dormant */ 50 51 void 52 cleanup(int sig __unused) 53 { 54 (void) pmap_unset(RSTATPROG, RSTATVERS_TIME); 55 (void) pmap_unset(RSTATPROG, RSTATVERS_SWTCH); 56 (void) pmap_unset(RSTATPROG, RSTATVERS_ORIG); 57 exit(0); 58 } 59 60 int 61 main(int argc, char *argv[]) 62 { 63 SVCXPRT *transp; 64 int sock = 0; 65 int proto = 0; 66 struct sockaddr_in from; 67 int fromlen; 68 69 if (argc == 2) 70 closedown = atoi(argv[1]); 71 if (closedown <= 0) 72 closedown = 20; 73 74 /* 75 * See if inetd started us 76 */ 77 fromlen = sizeof(from); 78 if (getsockname(0, (struct sockaddr *)&from, &fromlen) < 0) { 79 from_inetd = 0; 80 sock = RPC_ANYSOCK; 81 proto = IPPROTO_UDP; 82 } 83 84 if (!from_inetd) { 85 daemon(0, 0); 86 87 (void)pmap_unset(RSTATPROG, RSTATVERS_TIME); 88 (void)pmap_unset(RSTATPROG, RSTATVERS_SWTCH); 89 (void)pmap_unset(RSTATPROG, RSTATVERS_ORIG); 90 91 (void) signal(SIGINT, cleanup); 92 (void) signal(SIGTERM, cleanup); 93 (void) signal(SIGHUP, cleanup); 94 } 95 96 openlog("rpc.rstatd", LOG_CONS|LOG_PID, LOG_DAEMON); 97 98 transp = svcudp_create(sock); 99 if (transp == NULL) { 100 syslog(LOG_ERR, "cannot create udp service"); 101 exit(1); 102 } 103 if (!svc_register(transp, RSTATPROG, RSTATVERS_TIME, rstat_service, proto)) { 104 syslog(LOG_ERR, "unable to register (RSTATPROG, RSTATVERS_TIME, %s)", proto?"udp":"(inetd)"); 105 exit(1); 106 } 107 if (!svc_register(transp, RSTATPROG, RSTATVERS_SWTCH, rstat_service, proto)) { 108 syslog(LOG_ERR, "unable to register (RSTATPROG, RSTATVERS_SWTCH, %s)", proto?"udp":"(inetd)"); 109 exit(1); 110 } 111 if (!svc_register(transp, RSTATPROG, RSTATVERS_ORIG, rstat_service, proto)) { 112 syslog(LOG_ERR, "unable to register (RSTATPROG, RSTATVERS_ORIG, %s)", proto?"udp":"(inetd)"); 113 exit(1); 114 } 115 116 svc_run(); 117 syslog(LOG_ERR, "svc_run returned"); 118 exit(1); 119 } 120