more appimage stuff related to installation
[citadel.git] / appimage / ctdlvisor.c
index cfeaf290a526f4cf767eedcbb9e64f18a8668c01..cedc2a4c49d2c285d5560e82f4014b38b5613d04 100644 (file)
@@ -24,9 +24,8 @@
 #include <signal.h>
 #include <string.h>
 #include <limits.h>
-
-void main_loop(void);
-void run_in_foreground(void);
+#include <sys/types.h>
+#include <sys/stat.h>
 
 char *data_directory = "/usr/local/citadel";
 char *http_port = "80";
@@ -34,17 +33,34 @@ char *https_port = "443";
 pid_t citserver_pid;
 pid_t webcit_pid;
 pid_t webcits_pid;
+int shutting_down = 0;
+
+// Call this instead of exit() just for common diagnostics etc.
+void ctdlvisor_exit(int code) {
+       printf("ctdlvisor: exit code %d\n", code);
+       exit(code);
+}
 
 
+// Interrupting this program with a signal will begin an orderly shutdown.
 void signal_handler(int signal) {
        fprintf(stderr, "ctdlvisor: caught signal %d\n", signal);
 
+       while(shutting_down) {
+               fprintf(stderr, "ctdlvisor: already shutting down\n");
+               sleep(1);
+       }
+
        int status;
        pid_t who_exited;
        char *what_exited = NULL;
 
+       shutting_down = 1;
+       kill(citserver_pid, SIGTERM);
+       kill(webcit_pid, SIGTERM);
+       kill(webcits_pid, SIGTERM);
        do {
-               fprintf(stderr, "ctdlvisor: waiting for any child process to exit...\n");
+               fprintf(stderr, "ctdlvisor: waiting for all child process to exit...\n");
                who_exited = waitpid(-1, &status, 0);
                if (who_exited == citserver_pid) {
                        what_exited = "Citadel Server";
@@ -61,11 +77,25 @@ void signal_handler(int signal) {
                fprintf(stderr, "ctdlvisor: pid=%d (%s) exited, status=%d, exitcode=%d\n", who_exited, what_exited, status, WEXITSTATUS(status));
        } while (who_exited >= 0);
 
-       printf("ctdlvisor: exiting from signal catcher.\n");
-       exit(0);
+       ctdlvisor_exit(0);
 }
 
 
+void detach_from_tty(void) {
+       signal(SIGHUP, SIG_IGN);
+       signal(SIGINT, SIG_IGN);
+       signal(SIGQUIT, SIG_IGN);
+
+       setsid();       // become our own process group leader
+       umask(0);
+       if (    (freopen("/dev/null", "r", stdin) != stdin) ||
+               (freopen("/dev/null", "w", stdout) != stdout) ||
+               (freopen("/dev/null", "w", stderr) != stderr)
+       ) {
+               fprintf(stderr, "sysdep: unable to reopen stdio: %s\n", strerror(errno));
+       }
+}
+
 
 pid_t start_citadel() {
        char bin[1024];
@@ -73,7 +103,7 @@ pid_t start_citadel() {
        pid_t pid = fork();
        if (pid == 0) {
                fprintf(stderr, "ctdlvisor: executing %s\n", bin);
-               close(0); close(1); close(2);
+               detach_from_tty();
                execlp(bin, "citserver", "-x9", "-h", data_directory, NULL);
                exit(errno);
        }
@@ -91,7 +121,7 @@ pid_t start_webcit() {
        pid_t pid = fork();
        if (pid == 0) {
                fprintf(stderr, "ctdlvisor: executing %s\n", bin);
-               close(0); close(1); close(2);
+               detach_from_tty();
                execlp(bin, "webcit", "-x9", wchome, "-p", http_port, "uds", data_directory, NULL);
                exit(errno);
        }
@@ -109,7 +139,7 @@ pid_t start_webcits() {
        pid_t pid = fork();
        if (pid == 0) {
                fprintf(stderr, "ctdlvisor: executing %s\n", bin);
-               close(0); close(1); close(2);
+               detach_from_tty();
                execlp(bin, "webcit", "-x9", wchome, "-s", "-p", https_port, "uds", data_directory, NULL);
                exit(errno);
        }
@@ -119,78 +149,9 @@ pid_t start_webcits() {
 }
 
 
-static char *usage =
-       "ctdlvisor: usage: ctdlvisor [-h data_directory] [-p http_port] [-s https_port] command\n"
-       "           'command' must be one of: run, install, start, stop\n"
-;
-
-int main(int argc, char **argv) {
-       int c;
-
-       while ((c = getopt (argc, argv, "h:p:s:")) != -1)  switch(c) {
-               case 'h':
-                       data_directory = optarg;
-                       break;
-               case 'p':
-                       http_port = optarg;
-                       break;
-               case 's':
-                       https_port = optarg;
-                       break;
-               default:
-                       fprintf(stderr, "%s", usage);
-                       exit(1);
-       }
-
-
-       if (argc != optind+1) {
-               fprintf(stderr, "%s", usage);
-               exit(1);
-       }
-
-       if (!strcasecmp(argv[optind], "run")) {
-               run_in_foreground();
-       }
-       else {
-               fprintf(stderr, "%s", usage);
-               exit(1);
-       }
-
-       exit(0);
-}
-
-
-void run_in_foreground(void) {
-       fprintf(stderr, "ctdlvisor: Welcome to the Citadel System, brought to you using AppImage.\n");
-       fprintf(stderr, "ctdlvisor: LD_LIBRARY_PATH = %s\n", getenv("LD_LIBRARY_PATH"));
-       fprintf(stderr, "ctdlvisor:            PATH = %s\n", getenv("PATH"));
-       fprintf(stderr, "ctdlvisor:          APPDIR = %s\n", getenv("APPDIR"));
-       fprintf(stderr, "ctdlvisor:  data directory = %s\n", data_directory);
-       fprintf(stderr, "ctdlvisor:       HTTP port = %s\n", http_port);
-       fprintf(stderr, "ctdlvisor:      HTTPS port = %s\n", https_port);
-
-       if (access(data_directory, R_OK|W_OK|X_OK)) {
-               fprintf(stderr, "ctdlvisor: %s: %s\n", data_directory, strerror(errno));
-               exit(errno);
-       }
-
-       signal(SIGTERM, signal_handler);
-       signal(SIGHUP, signal_handler);
-       signal(SIGINT, signal_handler);
-       signal(SIGQUIT, signal_handler);
-
-       citserver_pid = start_citadel();
-       webcit_pid = start_webcit();
-       webcits_pid = start_webcits();
-
-       main_loop();
-}
-
-
 void main_loop(void) {
        int status;
        pid_t who_exited;
-       int shutting_down = 0;
        int citserver_exit_code = 0;
 
        do {
@@ -227,7 +188,141 @@ void main_loop(void) {
                sleep(1);
 
        } while (who_exited >= 0);
+       ctdlvisor_exit(citserver_exit_code);
+}
+
+
+void run_in_foreground(void) {
+       fprintf(stderr, "ctdlvisor: Welcome to the Citadel System, brought to you using AppImage.\n");
+       fprintf(stderr, "ctdlvisor: LD_LIBRARY_PATH = %s\n", getenv("LD_LIBRARY_PATH"));
+       fprintf(stderr, "ctdlvisor:            PATH = %s\n", getenv("PATH"));
+       fprintf(stderr, "ctdlvisor:          APPDIR = %s\n", getenv("APPDIR"));
+       fprintf(stderr, "ctdlvisor:  data directory = %s\n", data_directory);
+       fprintf(stderr, "ctdlvisor:       HTTP port = %s\n", http_port);
+       fprintf(stderr, "ctdlvisor:      HTTPS port = %s\n", https_port);
+
+       if (access(data_directory, R_OK|W_OK|X_OK)) {
+               fprintf(stderr, "ctdlvisor: %s: %s\n", data_directory, strerror(errno));
+               ctdlvisor_exit(errno);
+       }
+
+       signal(SIGTERM, signal_handler);
+       signal(SIGHUP, signal_handler);
+       signal(SIGINT, signal_handler);
+       signal(SIGQUIT, signal_handler);
+
+       citserver_pid = start_citadel();
+       webcit_pid = start_webcit();
+       webcits_pid = start_webcits();
+
+       main_loop();
+       ctdlvisor_exit(0);
+}
+
+
+void install_as_service(void) {
+
+       // FIXME fail if some other citadel distribution is already there
+       // FIXME fail if any server processes are running
+       // FIXME interact with the user
+       // FIXME get port numbers and data directory
+       // FIXME create the data directory
+       // FIXME move the appimage into its permanent location
+
+       fprintf(stderr, "Installing as service\n");
+
+       FILE *fp = fopen("/etc/systemd/system/ctdl.service", "w");
+       fprintf(fp,     "# This unit file starts all Citadel services via the AppImage distribution.\n"
+                       "[Unit]\n"
+                       "Description=Citadel\n"
+                       "After=network.target\n"
+                       "[Service]\n"
+                       "ExecStart=/root/citadel/appimage/Citadel-x86_64.AppImage run -h %s -s %s -s %s\n"
+                       "ExecStop=/bin/kill $MAINPID\n"
+                       "KillMode=process\n"
+                       "Restart=on-failure\n"
+                       "LimitCORE=infinity\n"
+                       "[Install]\n"
+                       "WantedBy=multi-user.target\n"
+               ,
+                       data_directory, http_port, https_port
+       );
+       fclose(fp);
+
+       fprintf(stderr, "systemd unit file is installed.  Type 'systemctl enable ctdl' to have it start at boot.\n");
+}
+
+
+static char *usage =
+       "\n"
+       "ctdlvisor: usage: ctdlvisor [-h data_directory] [-p http_port] [-s https_port] command\n"
+       "           'command' must be one of: run, install, remove, upgrade, test, help\n"
+       "\n"
+;
+
+int main(int argc, char **argv) {
+       int c;
+
+       if (getenv("APPDIR") == NULL) {
+               fprintf(stderr, "ctdlvisor: APPDIR is not set.  This program must be run from within an AppImage.\n");
+               ctdlvisor_exit(1);
+       }
+
+       while ((c = getopt (argc, argv, "h:p:s:")) != -1)  switch(c) {
+               case 'h':
+                       data_directory = optarg;
+                       break;
+               case 'p':
+                       http_port = optarg;
+                       break;
+               case 's':
+                       https_port = optarg;
+                       break;
+               default:
+                       fprintf(stderr, "%s", usage);
+                       ctdlvisor_exit(1);
+       }
+
+
+       if (argc != optind+1) {
+               fprintf(stderr, "%s", usage);
+               ctdlvisor_exit(1);
+       }
+
+       if (!strcasecmp(argv[optind], "run")) {
+               run_in_foreground();
+       }
+       else if (!strcasecmp(argv[optind], "install")) {
+               install_as_service();
+       }
+       else if (!strcasecmp(argv[optind], "remove")) {
+               fprintf(stderr, "oops, this is not implemented yet\n");
+       }
+       else if (!strcasecmp(argv[optind], "upgrade")) {
+               fprintf(stderr, "oops, this is not implemented yet\n");
+       }
+       else if (!strcasecmp(argv[optind], "test")) {
+               fprintf(stderr, "oops, this is not implemented yet\n");
+       }
+       else if (!strcasecmp(argv[optind], "help")) {
+               fprintf(stderr, "%s", usage);
+               fprintf(stderr, "[-h dir]  Use 'dir' as the Citadel data directory (this directory must exist)\n"
+                               "[-p port] Listen for HTTP connections on 'port'\n"
+                               "[-s port] Listen for HTTPS connections on 'port'\n"
+                               "'command' must be one of:\n"
+                               "       run     - launch Citadel services (does not detach from terminal)\n"
+                               "       install - create systemd unit files for automatic startup at boot\n"
+                               "       remove  - delete systemd unit files to end automatic startup\n"
+                               "       upgrade - download and install a new version of this appimage\n"
+                               "       test    - test the appimage for binary compatibility with this host\n"
+                               "       help    - display this message\n"
+                               "\n"
+               );
+       }
+       else {
+               fprintf(stderr, "%s", usage);
+               ctdlvisor_exit(1);
+       }
 
-       printf("ctdlvisor: exit code %d\n", citserver_exit_code);
-       exit(citserver_exit_code);
+       ctdlvisor_exit(0);
 }