Initial commit

24 files changed, 2851 insertions, 0 deletions

diff --git a/src/.gitignore b/src/.gitignore
new file mode 100644
index 0000000..3d36647
--- /dev/null
+++ b/src/.gitignore
@@ -0,0 +1 @@
+genrc
diff --git a/src/Makefile.am b/src/Makefile.am
new file mode 100644
index 0000000..fe24f3e
--- /dev/null
+++ b/src/Makefile.am
@@ -0,0 +1,32 @@
+bin_PROGRAMS = genrc
+genrc_SOURCES = \
+ genrc.c\
+ genrc.h\
+ err.c\
+ pidfrom.c\
+ pid_file.c\
+ pid_config.c\
+ pid_grep.c\
+ pid_proc.c\
+ pid_ps.c\
+ pidlist.c\
+ procscan.c\
+ com_status.c\
+ com_start.c\
+ com_stop.c\
+ com_restart.c\
+ com_reload.c\
+ transform.c\
+ match_exact.c\
+ match_glob.c\
+ match_regex.c\
+ sentinel.c
+
+AM_CPPFLAGS = @GRECS_INCLUDES@
+LDADD = @GRECS_LDADD@
+
+if COND_PCRE
+ genrc_SOURCES += match_pcre.c
+ AM_CPPFLAGS += -DHAVE_PCRE=1
+endif
+
diff --git a/src/com_reload.c b/src/com_reload.c
new file mode 100644
index 0000000..7ff1206
--- /dev/null
+++ b/src/com_reload.c
@@ -0,0 +1,17 @@
+#include "genrc.h"
+
+int
+com_reload(void)
+{
+        PIDLIST pids;
+        
+        if (genrc_no_reload)
+                return com_restart();
+        pidlist_init(&pids);
+        if (get_pid_list(genrc_pid_closure, &pids))
+                return 1;
+        pidlist_kill(&pids, genrc_signal_reload);
+        return 0;
+}
+
+        
diff --git a/src/com_restart.c b/src/com_restart.c
new file mode 100644
index 0000000..b3588a1
--- /dev/null
+++ b/src/com_restart.c
@@ -0,0 +1,12 @@
+#include "genrc.h"
+
+int
+com_restart(void)
+{
+        int rc;
+        rc = com_stop();
+        if (rc == 0)
+                rc = com_start();
+        return rc;
+}
+        
diff --git a/src/com_start.c b/src/com_start.c
new file mode 100644
index 0000000..73bb2ec
--- /dev/null
+++ b/src/com_start.c
@@ -0,0 +1,121 @@
+#include "genrc.h"
+
+void
+report_exec_error(int rc, char const *program)
+{
+        if (WIFEXITED(rc)) {
+                if (WEXITSTATUS(rc)) {
+                        genrc_error("%s exited with status %d",
+                                    program, WEXITSTATUS(rc));
+                }
+        } else if (WIFSIGNALED(rc)) {
+                char const *coremsg = "";
+#ifdef WCOREDUMP
+                if (WCOREDUMP(rc))
+                        coremsg = " (core dumped)";
+#endif
+                genrc_error("%s terminated on signal %d%s",
+                            program, WTERMSIG(rc), coremsg);
+        } else if (WIFSTOPPED(rc)) {
+                genrc_error("%s stopped on signal %d",
+                            program, WSTOPSIG(rc));
+        } else {
+                genrc_error("%s terminated with unrecognized status: %d",
+                            program, rc);
+        }
+}
+
+typedef void (*SIGHANDLER)(int);
+
+void
+sigchld(int sig)
+{
+}
+
+int
+timedwaitpid(pid_t pid, int *status)
+{
+        struct timeval now, stoptime, ttw;
+        int rc = -1;
+        SIGHANDLER oldsig;
+        
+        oldsig = signal(SIGCHLD, sigchld);
+        gettimeofday(&stoptime, NULL);
+        stoptime.tv_sec += genrc_timeout;
+        while (1) {
+                pid_t p;
+                
+                p = waitpid(pid, status, WNOHANG);
+                if (p == pid) {
+                        rc = 0;
+                        break;
+                }
+                if (p < 0 && errno != EINTR) {
+                        system_error(errno, "waitpid");
+                        break;
+                }
+
+                gettimeofday(&now, NULL);
+                if (timercmp(&now, &stoptime, >=))
+                        break;
+                timersub(&stoptime, &now, &ttw);
+                if (select(0, NULL, NULL, NULL, &ttw) < 0) {
+                        if (errno != EINTR) {
+                                system_error(errno, "select");
+                                break;
+                        }
+                }
+
+        }
+        signal(SIGCHLD, oldsig);
+        if (rc) {
+                kill(pid, SIGKILL);
+        }
+        return rc;
+}
+
+int
+com_start(void)
+{
+        pid_t pid;
+        int status;
+        PIDLIST pids;
+        char *p;
+        
+        pidlist_init(&pids);
+        if (get_pid_list(genrc_pid_closure, &pids) == 0) {
+                int running = pids.pidc > 0;
+                pidlist_free(&pids);
+                if (running) {
+                        genrc_error("%s is already running", genrc_program);
+                        return 1;
+                }
+        }
+
+        if ((p = getenv("GENRC_SENTINEL")) && *p == '1')
+                return sentinel();
+        
+        pid = fork();
+        if (pid == -1) {
+                system_error(errno, "fork");
+                return 1;
+        }
+        if (pid == 0) {
+                char *argv[] = { SHELL, "-c", NULL, NULL };
+                argv[2] = genrc_command;
+                execvp(SHELL, argv);
+                system_error(errno, "failed to exec %s", genrc_program);
+                exit(127);
+        }
+
+        if (timedwaitpid(pid, &status)) {
+                genrc_error("timed out waiting for %s to return",
+                            genrc_program);
+                return 1;
+        }
+        if (!(WIFEXITED(status) && WEXITSTATUS(status) == 0)) {
+                report_exec_error(status, genrc_program);
+                return 1;
+        }
+        return 0;
+}
diff --git a/src/com_status.c b/src/com_status.c
new file mode 100644
index 0000000..f68c6ef
--- /dev/null
+++ b/src/com_status.c
@@ -0,0 +1,38 @@
+#include "genrc.h"
+
+int
+com_status(void)
+{
+        PIDLIST pids;
+        int rc;
+
+        pidlist_init(&pids);
+        rc = get_pid_list(genrc_pid_closure, &pids);
+        if (rc == 0) {
+                if (pids.pidc == 0)
+                        printf("%s is not running\n", genrc_program);
+                else {
+                        printf("%s is running ", genrc_program);
+                        if (pids.pidc == 1) {
+                                printf("(pid %lu)",
+                                       (unsigned long) pids.pidv[0]);
+                        } else {