Use krb5 auth, not krb4.

[moira.git] / server / mr_main.c

diff --git a/server/mr_main.c b/server/mr_main.c
index 1c662259ab3f88a5ed81bad66bef2093d8d4ce85..a8e91fc907e176744495de2dc87f3e35ea906c1e 100644 (file)
--- a/server/mr_main.c
+++ b/server/mr_main.c
@@ -29,11 +29,10 @@
 #include <unistd.h>
 
 #include <krb.h>
+#include <krb5.h>
 
 RCSID("$Header$");
 
-extern char *krb_get_lrealm(char *, int);
-
 client *cur_client;
 
 char *whoami;
@@ -44,6 +43,7 @@ time_t now;
 
 char *host;
 char krb_realm[REALM_SZ];
+krb5_context context = NULL;
 
 /* Client array and associated data. This needs to be global for _list_users */
 client **clients;
@@ -106,6 +106,14 @@ 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);
+    }
+  krb5_init_ets(context);
+
   /*
    * Database initialization.  Only init if database should be open.
    */
@@ -162,9 +170,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);
@@ -185,7 +194,7 @@ int main(int argc, char **argv)
   while (!takedown)
     {
       int i;
-      struct timeval timeout;
+      struct timeval timeout = {60, 0}; /* 1 minute */
 
       /* If we're supposed to go down and we can, do it */
       if (((dormant == AWAKE) && (nclients == 0) &&
@@ -202,8 +211,7 @@ int main(int argc, char **argv)
       /* Block until something happens. */
       memcpy(&readfds, &xreadfds, sizeof(readfds));
       memcpy(&writefds, &xwritefds, sizeof(writefds));
-      /* XXX set timeout */
-      if (select(nfds, &readfds, &writefds, NULL, NULL) == -1)
+      if (select(nfds, &readfds, &writefds, NULL, &timeout) == -1)
        {
          if (errno != EINTR)
            com_err(whoami, errno, "in select");
@@ -233,11 +241,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)
@@ -263,6 +271,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,
@@ -284,7 +294,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;
@@ -292,19 +301,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;
 
@@ -318,6 +352,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 +380,10 @@ void reapchild(int x)
       if (pid == inc_pid)
        inc_running = 0;
       if (!takedown && (WTERMSIG(status) != 0 || WEXITSTATUS(status) != 0))
-       com_err(whoami, 0, "%d: child exits with signal %d status %d",
-               pid, WTERMSIG(status), WEXITSTATUS(status));
+       {
+         critical_alert("moirad", "%d: child exits with signal %d status %d",
+                        pid, WTERMSIG(status), WEXITSTATUS(status));
+       }
     }
 }