libexec/rpc.rstatd/rstatd.c

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