Build without krb4 if it's unavailable.

[moira.git] / server / mr_main.c

diff --git a/server/mr_main.c b/server/mr_main.c
index ae3239687f1064b8d95ff18632bb5acee4e25c36..2935aa7f9f7610dc85d3550f19ad4a197c825af3 100644 (file)
--- a/server/mr_main.c
+++ b/server/mr_main.c
@@ -28,12 +28,13 @@
 #include <string.h>
 #include <unistd.h>
 
+#ifdef HAVE_KRB4
 #include <krb.h>
+#endif
+#include <krb5.h>
 
 RCSID("$Header$");
 
-extern char *krb_get_lrealm(char *, int);
-
 client *cur_client;
 
 char *whoami;
@@ -43,13 +44,16 @@ FILE *journal;
 time_t now;
 
 char *host;
-char krb_realm[REALM_SZ];
+krb5_context context = NULL;
+char *krb_realm = NULL;
 
 /* Client array and associated data. This needs to be global for _list_users */
 client **clients;
 int nclients, clientssize;
 
 int dormant;
+int child_exited_abnormally = 0;
+int child_pid, child_signal, child_status;
 
 void reapchild(int x);
 void godormant(int x);
@@ -104,7 +108,19 @@ int main(int argc, char **argv)
        }
     }
 
-  krb_get_lrealm(krb_realm, 1);
+  status = krb5_init_context(&context);
+  if (status)
+    {
+      com_err(whoami, status, "Initializing krb5 context.");
+      exit(1);
+    }
+
+  status = krb5_get_default_realm(context, &krb_realm);
+  if (status)
+    {
+      com_err(whoami, status, "Getting default Kerberos realm.");
+      exit(1);
+    }
 
   /*
    * Database initialization.  Only init if database should be open.
@@ -162,9 +178,10 @@ int main(int argc, char **argv)
   /*
    * Establish template connection.
    */
-  if (!(listener = mr_listen(port)))
+  listener = mr_listen(port);
+  if (listener == -1)
     {
-      com_err(whoami, status, "trying to create listening connection");
+      com_err(whoami, MR_ABORTED, "trying to create listening connection");
       exit(1);
     }
   FD_ZERO(&xreadfds);
@@ -213,6 +230,14 @@ int main(int argc, char **argv)
 
       if (takedown)
        break;
+
+      if (child_exited_abnormally)
+       {
+         critical_alert("moirad", "%d: child exits with signal %d status %d",
+                        child_pid, child_signal, child_status);
+         child_exited_abnormally = 0;
+       }
+
       time(&now);
       if (!inc_running || now - inc_started > INC_TIMEOUT)
        next_incremental();
@@ -232,11 +257,11 @@ int main(int argc, char **argv)
       /* Handle any new connections */
       if (FD_ISSET(listener, &readfds))
        {
-         int newconn;
+         int newconn, addrlen = sizeof(struct sockaddr_in);
          struct sockaddr_in addr;
          client *cp;
 
-         newconn = mr_accept(listener, &addr);
+         newconn = accept(listener, (struct sockaddr *)&addr, &addrlen);
          if (newconn == -1)
            com_err(whoami, errno, "accepting new connection");
          else if (newconn > 0)
@@ -262,6 +287,8 @@ int main(int argc, char **argv)
              cp->tuplessize = 1;
              cp->tuples = xmalloc(sizeof(mr_params));
              memset(cp->tuples, 0, sizeof(mr_params));
+             cp->state = CL_ACCEPTING;
+             cp->version = 2;
 
              cur_client = cp;
              com_err(whoami, 0,
@@ -283,7 +310,6 @@ int main(int argc, char **argv)
              if (!clients[i]->ntuples)
                {
                  FD_CLR(clients[i]->con, &xwritefds);
-                 /* Now that we're done writing we can read again */
                  FD_SET(clients[i]->con, &xreadfds);
                }
              clients[i]->last_time_used = now;
@@ -291,19 +317,44 @@ int main(int argc, char **argv)
 
          if (FD_ISSET(clients[i]->con, &readfds))
            {
-             client_read(clients[i]);
-             if (clients[i]->ntuples)
-               FD_SET(clients[i]->con, &xwritefds);
-             clients[i]->last_time_used = now;
+             if (clients[i]->state == CL_ACCEPTING)
+               {
+                 switch(mr_cont_accept(clients[i]->con,
+                                       &clients[i]->hsbuf,
+                                       &clients[i]->hslen))
+                   {
+                   case -1:
+                     break;
+
+                   case 0:
+                     clients[i]->state = CL_CLOSING;
+                     break;
+
+                   default:
+                     clients[i]->state = CL_ACTIVE;
+                     clients[i]->hsbuf = NULL;
+                     break;
+                   }
+               }
+             else
+               {
+                 client_read(clients[i]);
+                 if (clients[i]->ntuples)
+                   {
+                     FD_CLR(clients[i]->con, &xreadfds);
+                     FD_SET(clients[i]->con, &xwritefds);
+                   }
+                 clients[i]->last_time_used = now;
+               }
            }
 
          if (clients[i]->last_time_used < tardy)
            {
              com_err(whoami, 0, "Shutting down connection due to inactivity");
-             clients[i]->done = 1;
+             clients[i]->state = CL_CLOSING;
            }
 
-         if (clients[i]->done)
+         if (clients[i]->state == CL_CLOSING)
            {
              client *old;
 
@@ -317,6 +368,7 @@ int main(int argc, char **argv)
              FD_CLR(clients[i]->con, &xwritefds);
              free_rtn_tuples(clients[i]);
              free(clients[i]->tuples);
+             free(clients[i]->hsbuf);
              old = clients[i];
              clients[i] = clients[--nclients];
              free(old);
@@ -345,8 +397,10 @@ void reapchild(int x)
        inc_running = 0;
       if (!takedown && (WTERMSIG(status) != 0 || WEXITSTATUS(status) != 0))
        {
-         critical_error("moirad", "%d: child exits with signal %d status %d",
-                        pid, WTERMSIG(status), WEXITSTATUS(status));
+         child_exited_abnormally = 1;
+         child_pid = pid;
+         child_signal = WTERMSIG(status);
+         child_status = WEXITSTATUS(status);
        }
     }
 }