]> andersk Git - moira.git/blobdiff - server/mr_main.c
new kerberos libraries; always get children's status to avoid zombies
[moira.git] / server / mr_main.c
index cfd30febd311f87b60aefde22139c3b58a47cf9f..6c1b018dd942191decd3f56586163b3cc956adbe 100644 (file)
@@ -4,7 +4,8 @@
  *     $Header$
  *
  *     Copyright (C) 1987 by the Massachusetts Institute of Technology
- *
+ *     For copying and distribution information, please see the file
+ *     <mit-copyright.h>.
  *
  *     SMS server process.
  *
 
 static char *rcsid_sms_main_c = "$Header$";
 
+#include <mit-copyright.h>
 #include <strings.h>
+#include <sys/types.h>
 #include <sys/errno.h>
 #include <sys/signal.h>
+#include <sys/wait.h>
+#include <sys/stat.h>
 #include "sms_server.h"
+#include <krb_et.h>
 
 extern CONNECTION newconn, listencon;
 
@@ -50,6 +56,7 @@ extern void do_client();
 extern int sigshut();
 void clist_append();
 void oplist_append();
+void reapchild(), godormant(), gowakeup();
 
 extern time_t now;
 
@@ -68,13 +75,14 @@ main(argc, argv)
 {
        int status;
        time_t tardy;
+       struct stat stbuf;
        
        whoami = argv[0];
        /*
         * Error handler init.
         */
-       init_sms_err_tbl();
-       init_krb_err_tbl();
+       initialize_sms_error_table();
+       initialize_krb_error_table();
        set_com_err_hook(sms_com_err);
        setlinebuf(stderr);
        
@@ -99,19 +107,24 @@ main(argc, argv)
        gdb_debug(0); /* this can be patched, if necessary, to enable */
                      /* GDB level debugging .. */
        krb_realm = malloc(REALM_SZ);
-       get_krbrlm(krb_realm, 1);
+       krb_get_lrealm(krb_realm, 1);
        
        /*
-        * Database initialization.
+        * Database initialization.  Only init if database should be open.
         */
 
-       if ((status = sms_open_database()) != 0) {
+       if (stat(SMS_MOTD_FILE, &stbuf) != 0) {
+           if ((status = sms_open_database()) != 0) {
                com_err(whoami, status, " when trying to open database.");
                exit(1);
+           }
+           sanity_check_database();
+       } else {
+           dormant = ASLEEP;
+           com_err(whoami, 0, "sleeping, not opening database");
        }
        
        sanity_check_queries();
-       sanity_check_database();
 
        /*
         * Set up client array handler.
@@ -119,16 +132,7 @@ main(argc, argv)
        nclients = 0;
        clients = (client **) malloc(0);
        
-       /*
-        * Signal handlers
-        *      There should probably be a few more of these.
-        */
-       
-       if ((((int)signal (SIGTERM, sigshut)) < 0) ||
-           (((int)signal (SIGHUP, sigshut)) < 0)) {
-               com_err(whoami, errno, " Unable to establish signal handler.");
-               exit(1);
-       }
+       sms_setup_signals();
        
        journal = fopen(JOURNAL, "a");
        if (journal == NULL) {
@@ -149,7 +153,10 @@ main(argc, argv)
        
        com_err(whoami, 0, "started (pid %d)", getpid());
        com_err(whoami, 0, rcsid_sms_main_c);
-       send_zgram("SMS", "server started");
+       if (dormant != ASLEEP)
+         send_zgram("SMS", "server started");
+       else
+         send_zgram("SMS", "server started, but database closed");
 
        /*
         * Run until shut down.
@@ -162,13 +169,28 @@ main(argc, argv)
 #ifdef notdef
                com_err(whoami, 0, "tick");
 #endif notdef
+               if (dormant == SLEEPY) {
+                   sms_close_database();
+                   com_err(whoami, 0, "database closed");
+                   sms_setup_signals();
+                   send_zgram("SMS", "database closed");
+                   dormant = ASLEEP;
+               } else if (dormant == GROGGY) {
+                   sms_open_database();
+                   com_err(whoami, 0, "database open");
+                   sms_setup_signals();
+                   send_zgram("SMS", "database open again");
+                   dormant = AWAKE;
+               }
+
                errno = 0;
                status = op_select_any(op_list, 0,
                                       (fd_set *)NULL, (fd_set *)NULL,
                                       (fd_set *)NULL, (struct timeval *)NULL);
 
                if (status == -1) {
-                       com_err(whoami, errno, " error from op_select");
+                       if (errno != EINTR)
+                         com_err(whoami, errno, " error from op_select");
                        continue;
                } else if (status != -2) {
                        com_err(whoami, 0, " wrong return from op_select_any");
@@ -199,6 +221,25 @@ main(argc, argv)
                                 * Sleep here to prevent hosing?
                                 */
                        }
+                       /* if the new connection is our only connection,
+                        * and the server is supposed to be down, then go
+                        * down now.
+                        */
+                       if ((dormant == AWAKE) && (nclients == 1) &&
+                           (stat(SMS_MOTD_FILE, &stbuf) == 0)) {
+                           com_err(whoami, 0, "motd file exists, slumbertime");
+                           dormant = SLEEPY;
+                       }
+                       /* on new connection, if we are no longer supposed
+                        * to be down, then wake up.
+                        */
+                       if ((dormant == ASLEEP) &&
+                           (stat(SMS_MOTD_FILE, &stbuf) == -1) &&
+                           (errno == ENOENT)) {
+                           com_err(whoami, 0, "motd file no longer exists, waking up");
+                           dormant = GROGGY;
+                       }
+                         
                }
                /*
                 * Handle any existing connections.
@@ -283,12 +324,11 @@ new_connection()
         */
        cp = (client *)malloc(sizeof *cp);
        bzero(cp, sizeof(*cp));
-       cp->state = CL_STARTING;
        cp->action = CL_ACCEPT;
        cp->con = newconn;
        cp->id = counter++;
        cp->args = NULL;
-       cp->clname = NULL;
+       cp->clname[0] = NULL;
        cp->reply.sms_argv = NULL;
        cp->first = NULL;
        cp->last = NULL;
@@ -430,3 +470,65 @@ oplist_delete(oplp, op)
        }
        abort();
 }
+
+
+void reapchild()
+{
+    union wait status;
+    int pid;
+
+    while ((pid = wait3(&status, WNOHANG, (struct rusage *)0)) > 0) {
+       if  (!takedown && (status.w_termsig != 0 || status.w_retcode != 0))
+         com_err(whoami, 0, "%d: child exits with signal %d status %d",
+                 pid, status.w_termsig, status.w_retcode);
+    }
+}
+
+
+void godormant()
+{
+    switch (dormant) {
+    case AWAKE:
+    case GROGGY:
+       com_err(whoami, 0, "requested to go dormant");
+       break;
+    case ASLEEP:
+       com_err(whoami, 0, "already asleep");
+       break;
+    case SLEEPY:
+       break;
+    }
+    dormant = SLEEPY;
+}
+
+
+void gowakeup()
+{
+    switch (dormant) {
+    case ASLEEP:
+    case SLEEPY:
+       com_err(whoami, 0, "Good morning");
+       break;
+    case AWAKE:
+       com_err(whoami, 0, "already awake");
+       break;
+    case GROGGY:
+       break;
+    }
+    dormant = GROGGY;
+}
+
+       
+sms_setup_signals()
+{
+    /* There should probably be a few more of these. */
+       
+    if ((((int)signal (SIGTERM, sigshut)) < 0) ||
+       (((int)signal (SIGCHLD, reapchild)) < 0) ||
+       (((int)signal (SIGUSR1, godormant)) < 0) ||
+       (((int)signal (SIGUSR2, gowakeup)) < 0) ||
+       (((int)signal (SIGHUP, sigshut)) < 0)) {
+       com_err(whoami, errno, " Unable to establish signal handlers.");
+       exit(1);
+    }
+}
This page took 0.047823 seconds and 4 git commands to generate.