Removed the global variable 'home_specified'

[citadel.git] / citadel / sendcommand.c

diff --git a/citadel/sendcommand.c b/citadel/sendcommand.c
index 9b331cc2456d44df78ab9ad3932d367a80608c19..45ff15c5daa99c089879bbe54f4aa31dae4da871 100644 (file)
--- a/citadel/sendcommand.c
+++ b/citadel/sendcommand.c
@@ -29,8 +29,8 @@
 #include <signal.h>
 #include <errno.h>
 #include <limits.h>
+#include <libcitadel.h>
 #include "citadel.h"
-#include "tools.h"
 #include "citadel_ipc.h"
 #include "server.h"
 #include "config.h"
@@ -73,6 +73,8 @@ void remove_lockfile(void)
  */
 void nq_cleanup(int e)
 {
+       if (e == SIGALRM)
+               fprintf(stderr, "\nWatch dog time out.\n");
        remove_lockfile();
        exit(e);
 }
@@ -161,6 +163,7 @@ void sendcommand_die(void) {
 }
 
 
+
 /*
  * main
  */
@@ -169,6 +172,7 @@ int main(int argc, char **argv)
        int a;
        char cmd[SIZ];
        char buf[SIZ];
+       int watchdog = 5;
 
        int relh=0;
        int home=0;
@@ -177,9 +181,8 @@ int main(int argc, char **argv)
        fd_set read_fd;
        struct timeval tv;
        int ret, err;
-
-       CtdlInitBase64Table();
-
+       int server_shutting_down = 0;
+       
        strcpy(ctdl_home_directory, DEFAULT_PORT);
 
        strcpy(cmd, "");
@@ -194,8 +197,11 @@ int main(int argc, char **argv)
                        else
                                safestrncpy(relhome, &argv[a][2],
                                                        sizeof relhome);
-                       home_specified = 1;
                        home=1;
+               } else if (!strncmp(argv[a], "-w", 2)) {
+                       watchdog = atoi(&argv[a][2]);
+                       if (watchdog<1)
+                               watchdog=1;
                } else {
                        if (!IsEmptyStr(cmd))
                                strcat(cmd, " ");
@@ -203,7 +209,7 @@ int main(int argc, char **argv)
                }
        }
 
-       calc_dirs_n_files(relh, home, relhome, ctdldir);
+       calc_dirs_n_files(relh, home, relhome, ctdldir, 0);
        get_config();
 
        signal(SIGINT, cleanup);
@@ -217,7 +223,7 @@ int main(int argc, char **argv)
                        ctdl_home_directory);
        fflush(stderr);
 
-       alarm(5);
+       alarm(watchdog);
        signal(SIGALRM, nq_cleanup); /* Set up a watchdog type timer in case we hang */
        
        np_attach_to_server(UDS, ctdl_home_directory);
@@ -232,10 +238,14 @@ int main(int argc, char **argv)
        tv.tv_sec = 0;
        tv.tv_usec = 1000;
 
+       if (!strncasecmp(&buf[1], "31", 2)) {
+               server_shutting_down = 1;
+       }
+
        if (buf[0] == '1') {
                while (CtdlIPC_chat_recv(ipc, buf), strcmp(buf, "000")) {
                        printf("%s\n", buf);
-                       alarm(5); /* Kick the watchdog timer */
+                       alarm(watchdog); /* Kick the watchdog timer */
                }
        } else if (buf[0] == '4') {
                do {
@@ -275,16 +285,23 @@ int main(int argc, char **argv)
                                        fflush (stdout);
                                }
                        }
-                       alarm(5); /* Kick the watchdog timer */
+                       alarm(watchdog); /* Kick the watchdog timer */
                } while (strcmp(buf, "000"));
                CtdlIPC_chat_send(ipc, "\n");
                CtdlIPC_chat_send(ipc, "000");
        }
        alarm(0);       /* Shutdown the watchdog timer */
        fprintf(stderr, "sendcommand: processing ended.\n");
-       if (strcasecmp(cmd, "DOWN"))
-               cleanup(0);
-       else    /* If we downed the server we can't to do CtdlIPCQuit in cleanup()*/
+
+       /* Clean up and log off ... unless the server indicated that the command
+        * we sent is shutting it down, in which case we want to just cut the
+        * connection and exit.
+        */
+       if (server_shutting_down) {
                nq_cleanup(0);
+       }
+       else {
+               cleanup(0);
+       }
        return 0;
 }