libexec/rpc.rstatd/rstatd.c

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