move signal setup to a proc and call it at appropriate times;

[moira.git] / server / mr_main.c

diff --git a/server/mr_main.c b/server/mr_main.c
index aaf1093c22be5ed0e53a8c9e01a3a1f662b2563c..2d76d215bd919071416d1a3b68211fb48135a2b7 100644 (file)
--- a/server/mr_main.c
+++ b/server/mr_main.c
@@ -20,10 +20,13 @@ 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;
 
@@ -53,7 +56,7 @@ extern void do_client();
 extern int sigshut();
 void clist_append();
 void oplist_append();
-void reapchild();
+void reapchild(), godormant(), gowakeup();
 
 extern time_t now;
 
@@ -72,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);
        
@@ -106,16 +110,21 @@ main(argc, argv)
        get_krbrlm(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.
@@ -123,17 +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 (SIGCHLD, reapchild)) < 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) {
@@ -154,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.
@@ -167,13 +169,27 @@ 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");
+                       com_err(whoami, errno, " error from op_select");
                        continue;
                } else if (status != -2) {
                        com_err(whoami, 0, " wrong return from op_select_any");
@@ -204,6 +220,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.
@@ -288,7 +323,6 @@ 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++;
@@ -442,11 +476,62 @@ void reapchild()
     union wait status;
     int pid;
 
+    if (takedown || dormant == ASLEEP)
+      return;
     while ((pid = wait3(&status, WNOHANG, (struct rusage *)0)) > 0) {
        if  (status.w_termsig == 0 && status.w_retcode == 0)
-         com_err(whoami, 0, "dcm started successfully");
+         com_err(whoami, 0, "child exited successfully");
        else
-         com_err(whoami, 0, "%d: startdcm exits with signal %d status %d",
+         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);
+    }
+}