Add gtlc query.

[moira.git] / server / mr_main.c

diff --git a/server/mr_main.c b/server/mr_main.c
index 0fd93bc6384612b0cc681ccae85bcf44b5ad0d96..8b916152079a00dfe13c79666a8c60f0c6fe1a16 100644 (file)
--- a/server/mr_main.c
+++ b/server/mr_main.c
@@ -32,8 +32,6 @@
 
 RCSID("$Header$");
 
-extern char *krb_get_lrealm(char *, int);
-
 client *cur_client;
 
 char *whoami;
@@ -162,7 +160,8 @@ 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");
       exit(1);
@@ -232,11 +231,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 +261,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 +284,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 +291,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 +342,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);
@@ -344,8 +370,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));
+       }
     }
 }