+// This is a supervisor program that handles start/stop/restart of
+// the various Citadel System components, when we are running in
+// a Docker container.
+//
+// Copyright (c) 2021 by the citadel.org team
+//
+// This program is open source software. Use, duplication, or disclosure
+// is subject to the terms of the GNU General Public License, version 3.
+// The program is distributed without any warranty, expressed or implied.
+
+#define CTDL_DIR "/citadel-data"
+
+#include <stdlib.h>
+#include <unistd.h>
+#include <stdio.h>
+#include <sys/wait.h>
+#include <errno.h>
+#include <signal.h>
+#include <string.h>
+#include <limits.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+
+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 all child process to exit...\n");
+ who_exited = waitpid(-1, &status, 0);
+ if (who_exited == citserver_pid) {
+ what_exited = "Citadel Server";
+ }
+ else if (who_exited == webcit_pid) {
+ what_exited = "WebCit HTTP";
+ }
+ else if (who_exited == webcits_pid) {
+ what_exited = "WebCit HTTPS";
+ }
+ else {
+ what_exited = "unknown";
+ }
+ if (who_exited >= 0) {
+ fprintf(stderr, "ctdlvisor: %d (%s) ended, status=%d\n", who_exited, what_exited, status);
+ }
+ } while (who_exited >= 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() {
+ pid_t pid = fork();
+ if (pid == 0) {
+ fprintf(stderr, "ctdlvisor: executing citserver\n");
+ //detach_from_tty();
+ execlp("/usr/local/citadel/citserver", "citserver", "-x9", "-h", CTDL_DIR, NULL);
+ exit(errno);
+ }
+ else {
+ fprintf(stderr, "ctdlvisor: citserver running on pid=%d\n", pid);
+ return(pid);
+ }
+}
+
+
+pid_t start_webcit() {
+ pid_t pid = fork();
+ if (pid == 0) {
+ fprintf(stderr, "ctdlvisor: executing webcit (http)\n");
+ //detach_from_tty();
+ execlp("/usr/local/webcit/webcit", "webcit", "-x9", "-p", "80", "uds", CTDL_DIR, NULL);
+ exit(errno);
+ }
+ else {
+ fprintf(stderr, "ctdlvisor: webcit (HTTP) running on pid=%d\n", pid);
+ return(pid);
+ }
+}
+
+
+pid_t start_webcits() {
+ pid_t pid = fork();
+ if (pid == 0) {
+ fprintf(stderr, "ctdlvisor: executing webcit (https)\n");
+ //detach_from_tty();
+ execlp("/usr/local/webcit/webcit", "webcit", "-x9", "-s", "-p", "443", "uds", CTDL_DIR, NULL);
+ exit(errno);
+ }
+ else {
+ fprintf(stderr, "ctdlvisor: webcit (HTTPS) running on pid=%d\n", pid);
+ return(pid);
+ }
+}
+
+
+void main_loop(void) {
+ int status;
+ pid_t who_exited;
+ int citserver_exit_code = 0;
+
+ do {
+ who_exited = waitpid(-1, &status, 0);
+ fprintf(stderr, "ctdlvisor: pid=%d exited, status=%d, exitcode=%d\n", who_exited, status, WEXITSTATUS(status));
+
+ // A *deliberate* exit of citserver will cause ctdlvisor to shut the whole AppImage down.
+ // If it crashes, however, we will start it back up.
+ if (who_exited == citserver_pid) {
+ citserver_exit_code = WEXITSTATUS(status);
+ if ((WIFEXITED(status)) && (citserver_exit_code == 0)) {
+ fprintf(stderr, "ctdlvisor: citserver exited normally - ending AppImage session\n");
+ shutting_down = 1;
+ kill(webcit_pid, SIGTERM);
+ kill(webcits_pid, SIGTERM);
+ }
+ else if ((WIFEXITED(status)) && (citserver_exit_code >= 101) && (citserver_exit_code <= 109)) {
+ fprintf(stderr, "ctdlvisor: citserver exited intentionally - ending AppImage session\n");
+ shutting_down = 1;
+ kill(webcit_pid, SIGTERM);
+ kill(webcits_pid, SIGTERM);
+ }
+ else {
+ if (WIFSIGNALED(status)) {
+ fprintf(stderr, "ctdlvisor: citserver crashed on signal %d\n", WTERMSIG(status));
+ }
+ citserver_pid = start_citadel();
+ }
+ }
+
+ // WebCit processes are restarted if they exit for any reason.
+ if ((who_exited == webcit_pid) && (!shutting_down)) webcit_pid = start_webcit();
+ if ((who_exited == webcits_pid) && (!shutting_down)) webcits_pid = start_webcits();
+
+ // If we somehow end up in an endless loop, at least slow it down.
+ sleep(1);
+
+ } while (who_exited >= 0);
+ ctdlvisor_exit(citserver_exit_code);
+}
+
+
+void install_client_link(void) { // FIXME this is all furkokt and needs to be rethought now that it's docker and not appimage
+
+ FILE *fp;
+ char path_to_link[PATH_MAX];
+ snprintf(path_to_link, sizeof path_to_link, CTDL_DIR "citadel");
+ fp = fopen(path_to_link, "w");
+ if (!fp) {
+ fprintf(stderr, "%s\n", strerror(errno));
+ return;
+ }
+
+ fprintf(fp, "#!/bin/bash\n"
+ "export APPDIR=%s\n"
+ "export LD_LIBRARY_PATH=${APPDIR}/usr/bin:$LD_LIBRARY_PATH\n"
+ "export PATH=${APPDIR}/usr/bin:$PATH\n"
+ "exec citadel\n"
+ ,
+ getenv("APPDIR")
+ );
+
+ fchmod(fileno(fp), 0755);
+ fclose(fp);
+}
+
+
+int main(int argc, char **argv) {
+ int a;
+ int migrate_mode = 0;
+
+ fprintf(stderr, "ctdlvisor: Welcome to the Citadel System running in a container.\n");
+ fprintf(stderr, "ctdlvisor: command line arguments: ");
+ for (a=0; a<argc; ++a) {
+ fprintf(stderr, "%s ", argv[a]);
+ }
+ fprintf(stderr, "\n");
+
+ char *dirs[] = {
+ CTDL_DIR,
+ CTDL_DIR "/data",
+ CTDL_DIR "/files",
+ CTDL_DIR "/keys"
+ };
+
+ for (a=0; a<4; ++a) {
+ mkdir(dirs[a], 0777);
+ if (access(dirs[a], R_OK|W_OK|X_OK)) {
+ fprintf(stderr, "ctdlvisor: %s: %s\n", dirs[a], strerror(errno));
+ ctdlvisor_exit(errno);
+ }
+ else {
+ fprintf(stderr, "ctdlvisor: %s is writable\n", dirs[a]);
+ }
+ }
+
+ symlink(CTDL_DIR "/keys", "/usr/local/webcit/keys");
+
+ /* parse command-line arguments */
+ while ((a=getopt(argc, argv, "cm")) != EOF) switch(a) {
+
+ // test this binary for compatibility and exit
+ case 'c':
+ fprintf(stderr, "%s: binary compatibility confirmed\n", argv[0]);
+ exit(0);
+ break;
+
+ // run ctdlmigrate only
+ case 'm':
+ migrate_mode = 1;
+ break;
+
+ // any other parameter makes it crash and burn
+ default:
+ fprintf(stderr, "usage\n");
+ exit(1);
+ }
+
+
+ signal(SIGHUP, signal_handler);
+
+ // "migrate mode" means we just start the server and then run ctdlmigrate interactively.
+ if (migrate_mode) {
+ citserver_pid = start_citadel();
+ fprintf(stderr, "ctdlvisor: waiting a moment for citserver to initialize...\n");
+ sleep(5);
+ char bin[1024];
+ sprintf(bin, "/usr/local/citadel/ctdlmigrate -h %s", CTDL_DIR);
+ system(bin);
+ kill(citserver_pid, SIGTERM);
+ }
+
+ // Otherwise, it's just a normal happy day in Citadel land.
+ else {
+ signal(SIGTERM, signal_handler);
+ signal(SIGINT, signal_handler);
+ signal(SIGQUIT, signal_handler);
+
+ citserver_pid = start_citadel(); // start Citadel Server
+ webcit_pid = start_webcit(); // start WebCit HTTP
+ webcits_pid = start_webcits(); // start WebCit HTTPS
+
+ install_client_link();
+
+ main_loop();
+ }
+
+ ctdlvisor_exit(0);
+}
projects / citadel-docker.git / commitdiff