usr.sbin/rpc.statd/statd.c

   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  */
  33
  34 #ifndef lint
  35 static const char rcsid[] =
  36   "$FreeBSD: src/usr.sbin/rpc.statd/statd.c,v 1.6 1999/10/05 14:40:38 marcel Exp $";
  37 #endif /* not lint */
  38
  39 /* main() function for status monitor daemon.  Some of the code in this */
  40 /* file was generated by running rpcgen /usr/include/rpcsvc/sm_inter.x  */
  41 /* The actual program logic is in the file procs.c                      */
  42
  43 #include <err.h>
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <rpc/rpc.h>
  47 #include <rpc/pmap_clnt.h>
  48 #include <syslog.h>
  49 #include <sys/types.h>
  50 #include <sys/wait.h>
  51 #include <signal.h>
  52 #include "statd.h"
  53
  54 int debug = 0;          /* Controls syslog() calls for debug messages   */
  55
  56 extern void sm_prog_1(struct svc_req *rqstp, SVCXPRT *transp);
  57 static void handle_sigchld();
  58 static void usage __P((void));
  59
  60 int
  61 main(int argc, char **argv)
  62 {
  63   SVCXPRT *transp;
  64   struct sigaction sa;
  65
  66   if (argc > 1)
  67   {
  68     if (strcmp(argv[1], "-d"))
  69         usage();
  70     debug = 1;
  71   }
  72
  73   (void)pmap_unset(SM_PROG, SM_VERS);
  74
  75   transp = svcudp_create(RPC_ANYSOCK);
  76   if (transp == NULL)
  77     errx(1, "cannot create udp service");
  78   if (!svc_register(transp, SM_PROG, SM_VERS, sm_prog_1, IPPROTO_UDP))
  79     errx(1, "unable to register (SM_PROG, SM_VERS, udp)");
  80
  81   transp = svctcp_create(RPC_ANYSOCK, 0, 0);
  82   if (transp == NULL)
  83     errx(1, "cannot create tcp service");
  84   if (!svc_register(transp, SM_PROG, SM_VERS, sm_prog_1, IPPROTO_TCP))
  85     errx(1, "unable to register (SM_PROG, SM_VERS, tcp)");
  86   init_file("/var/db/statd.status");
  87
  88   /* Note that it is NOT sensible to run this program from inetd - the  */
  89   /* protocol assumes that it will run immediately at boot time.        */
  90   daemon(0, 0);
  91   openlog("rpc.statd", 0, LOG_DAEMON);
  92   if (debug) syslog(LOG_INFO, "Starting - debug enabled");
  93   else syslog(LOG_INFO, "Starting");
  94
  95   /* Install signal handler to collect exit status of child processes   */
  96   sa.sa_handler = handle_sigchld;
  97   sigemptyset(&sa.sa_mask);
  98   sigaddset(&sa.sa_mask, SIGCHLD);
  99   sa.sa_flags = SA_RESTART;
 100   sigaction(SIGCHLD, &sa, NULL);
 101
 102   /* Initialisation now complete - start operating                      */
 103   notify_hosts();       /* Forks a process (if necessary) to do the     */
 104                         /* SM_NOTIFY calls, which may be slow.          */
 105
 106   svc_run();    /* Should never return                                  */
 107   exit(1);
 108 }
 109
 110 static void
 111 usage()
 112 {
 113       fprintf(stderr, "usage: rpc.statd [-d]\n");
 114       exit(1);
 115 }
 116
 117 /* handle_sigchld ---------------------------------------------------------- */
 118 /*
 119    Purpose:     Catch SIGCHLD and collect process status
 120    Retruns:     Nothing.
 121    Notes:       No special action required, other than to collect the
 122                 process status and hence allow the child to die:
 123                 we only use child processes for asynchronous transmission
 124                 of SM_NOTIFY to other systems, so it is normal for the
 125                 children to exit when they have done their work.
 126 */
 127
 128 static void handle_sigchld(int sig, int code, struct sigcontext *scp)
 129 {
 130   int pid, status;
 131   pid = wait4(-1, &status, WNOHANG, (struct rusage*)0);
 132   if (!pid) syslog(LOG_ERR, "Phantom SIGCHLD??");
 133   else if (status == 0)
 134   {
 135     if (debug) syslog(LOG_DEBUG, "Child %d exited OK", pid);
 136   }
 137   else syslog(LOG_ERR, "Child %d failed with status %d", pid,
 138     WEXITSTATUS(status));
 139 }
 140