summaryrefslogtreecommitdiffabout
path: root/src
authorSergey Poznyakoff <gray@gnu.org>2018-05-15 05:50:08 (GMT)
committer Sergey Poznyakoff <gray@gnu.org>2018-05-15 06:05:31 (GMT)
commitb5f4388a2da1b94ce8c8e45a990fd51b2f52dae4 (patch) (side-by-side diff)
tree28bcf8cf313309636357c470faed4456837db7c8 /src
downloadgenrc-b5f4388a2da1b94ce8c8e45a990fd51b2f52dae4.tar.gz
genrc-b5f4388a2da1b94ce8c8e45a990fd51b2f52dae4.tar.bz2
Initial commit
Diffstat (limited to 'src') (more/less context) (ignore whitespace changes)
-rw-r--r--src/.gitignore1
-rw-r--r--src/Makefile.am32
-rw-r--r--src/com_reload.c17
-rw-r--r--src/com_restart.c12
-rw-r--r--src/com_start.c121
-rw-r--r--src/com_status.c38
-rw-r--r--src/com_stop.c66
-rw-r--r--src/err.c47
-rw-r--r--src/genrc.c469
-rw-r--r--src/genrc.h137
-rw-r--r--src/match_exact.c20
-rw-r--r--src/match_glob.c22
-rw-r--r--src/match_pcre.c60
-rw-r--r--src/match_regex.c61
-rw-r--r--src/pid_config.c91
-rw-r--r--src/pid_file.c70
-rw-r--r--src/pid_grep.c80
-rw-r--r--src/pid_proc.c335
-rw-r--r--src/pid_ps.c137
-rw-r--r--src/pidfrom.c52
-rw-r--r--src/pidlist.c121
-rw-r--r--src/procscan.c95
-rw-r--r--src/sentinel.c207
-rw-r--r--src/transform.c560
24 files changed, 2851 insertions, 0 deletions
diff --git a/src/.gitignore b/src/.gitignore
new file mode 100644
index 0000000..3d36647
--- a/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
--- a/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
--- a/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
--- a/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
--- a/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
--- a/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 {
+ int i;
+ int delim = ' ';
+ printf("(pids");
+ for (i = 0; i < pids.pidc; i++) {
+ printf("%c%lu", delim,
+ (unsigned long) pids.pidv[i]);
+ delim = ',';
+ }
+ putchar(')');
+ }
+ putchar('\n');
+ }
+ } else {
+ printf("status unknown\n");
+ }
+ return 0;
+}
+
+
diff --git a/src/com_stop.c b/src/com_stop.c
new file mode 100644
index 0000000..5227358
--- a/dev/null
+++ b/src/com_stop.c
@@ -0,0 +1,66 @@
+#include "genrc.h"
+
+static void
+timermul(struct timeval *a, int b)
+{
+ a->tv_sec *= b;
+ a->tv_usec *= b;
+ a->tv_sec = a->tv_sec + a->tv_usec / 1000000;
+ a->tv_usec %= 1000000;
+}
+
+#define MIN_POLL_TTW 20000
+
+int
+com_stop(void)
+{
+ PIDLIST pids;
+ struct timeval stoptime, before, after, ttw, maxttw;
+ int ratio;
+
+ pidlist_init(&pids);
+ if (get_pid_list(genrc_pid_closure, &pids)) {
+ genrc_error("program status unknown");
+ return 1;
+ }
+
+ if (pids.pidc == 0) {
+ genrc_error("%s not running", genrc_program);
+ return 1;
+ }
+
+ gettimeofday(&stoptime, NULL);
+ stoptime.tv_sec += genrc_timeout;
+ ratio = 1;
+ while (pids.pidc) {
+ gettimeofday(&before, NULL);
+ if (timercmp(&before, &stoptime, >))
+ break;
+ pidlist_kill(&pids, genrc_signal_stop);
+ if (get_pid_list(genrc_pid_closure, &pids))
+ break;
+ if (pids.pidc == 0)
+ return 0;
+ gettimeofday(&after, NULL);
+ if (timercmp(&after, &stoptime, >=))
+ break;
+ if (ratio < 10)
+ ++ratio;
+ timersub(&stoptime, &after, &maxttw);
+ timersub(&after, &before, &ttw);
+ timermul(&ttw, ratio);
+ if (ttw.tv_sec < 0 || ttw.tv_usec < 0)
+ ttw.tv_sec = ttw.tv_usec = 0;
+ if (timercmp(&ttw, &maxttw, >))
+ ttw = maxttw;
+ if (ttw.tv_sec == 0 && ttw.tv_usec < MIN_POLL_TTW)
+ ttw.tv_usec = MIN_POLL_TTW;
+ if (select(0, NULL, NULL, NULL, &ttw) < 0 && errno != EINTR) {
+ system_error(errno, "select");
+ break;
+ }
+ }
+ return 1;
+}
+
+
diff --git a/src/err.c b/src/err.c
new file mode 100644
index 0000000..539a7b8
--- a/dev/null
+++ b/src/err.c
@@ -0,0 +1,47 @@
+#include "genrc.h"
+
+char *progname = "genrc";
+
+void
+setprogname(char const *s)
+{
+ progname = xstrdup(s);
+}
+
+void
+genrc_error(char const *fmt, ...)
+{
+ va_list ap;
+ fprintf(stderr, "%s: ", progname);
+ va_start(ap, fmt);
+ vfprintf(stderr, fmt, ap);
+ va_end(ap);
+ fputc('\n', stderr);
+}
+
+void
+usage_error(char const *fmt, ...)
+{
+ va_list ap;
+ fprintf(stderr, "%s: ", progname);
+ va_start(ap, fmt);
+ vfprintf(stderr, fmt, ap);
+ va_end(ap);
+ fputc('\n', stderr);
+ exit(1);
+}
+
+void
+system_error(int ec, char const *fmt, ...)
+{
+ va_list ap;
+ fprintf(stderr, "%s: ", progname);
+ va_start(ap, fmt);
+ vfprintf(stderr, fmt, ap);
+ va_end(ap);
+ fprintf(stderr, ": %s\n", strerror(ec));
+}
+
+
+
+
diff --git a/src/genrc.c b/src/genrc.c
new file mode 100644
index 0000000..64dd81a
--- a/dev/null
+++ b/src/genrc.c
@@ -0,0 +1,469 @@
+#include "genrc.h"
+#include <sys/ioctl.h>
+
+char *genrc_command;
+char *genrc_program;
+char *genrc_pid_from;
+unsigned genrc_timeout = 5;
+int genrc_no_reload;
+int genrc_signal_stop = SIGTERM;
+int genrc_signal_reload = SIGHUP;
+GENRC_PID_CLOSURE *genrc_pid_closure;
+char *genrc_pidfile;
+
+
+enum {
+ OPT_USAGE = 256,
+ OPT_VERSION,
+ OPT_SIGNAL_RELOAD,
+ OPT_NO_RELOAD,
+ OPT_SIGNAL_STOP
+};
+
+struct option longopts[] = {
+ { "help", no_argument, 0, 'h' },
+ { "usage", no_argument, 0, OPT_USAGE },
+ { "command", required_argument, 0, 'c' },
+ { "program", required_argument, 0, 'p' },
+ { "pid-from", required_argument, 0, 'P' },
+ { "pidfile", required_argument, 0, 'F' },
+ { "timeout", required_argument, 0, 't' },
+ { "signal-reload", required_argument, 0, OPT_SIGNAL_RELOAD },
+ { "no-reload", no_argument, 0, OPT_NO_RELOAD },
+ { "signal-stop", required_argument, 0, OPT_SIGNAL_STOP },
+ { "sentinel", no_argument, 0, 'S' },
+ { "version", no_argument, 0, OPT_VERSION },
+ { NULL }
+};
+char shortopts[] = "c:hF:P:p:St:";
+
+struct sigdefn {
+ char const *sig_name;
+ int sig_no;
+};
+
+#define S(s) { #s, s }
+static struct sigdefn sigdefn[] = {
+ S (SIGHUP),
+ S (SIGINT),
+ S (SIGQUIT),
+ S (SIGILL),
+ S (SIGTRAP),
+ S (SIGABRT),
+ S (SIGIOT),
+ S (SIGBUS),
+ S (SIGFPE),
+ S (SIGKILL),
+ S (SIGUSR1),
+ S (SIGSEGV),
+ S (SIGUSR2),
+ S (SIGPIPE),
+ S (SIGALRM),
+ S (SIGTERM),
+#ifdef SIGSTKFLT
+ S (SIGSTKFLT),
+#endif
+ S (SIGCHLD),
+ S (SIGCONT),
+ S (SIGSTOP),
+ S (SIGTSTP),
+ S (SIGTTIN),
+ S (SIGTTOU),
+#ifdef SIGURG
+ S (SIGURG),
+#endif
+#ifdef SIGXCPU
+ S (SIGXCPU),
+#endif
+#ifdef SIGXFSZ
+ S (SIGXFSZ),
+#endif
+#ifdef SIGVTALRM
+ S (SIGVTALRM),
+#endif
+#ifdef SIGPROF
+ S (SIGPROF),
+#endif
+#ifdef SIGWINCH
+ S (SIGWINCH),
+#endif
+#ifdef SIGPOLL
+ S (SIGPOLL),
+#endif
+#ifdef SIGIO
+ S (SIGIO),
+#endif
+#ifdef SIGPWR
+ S (SIGPWR),
+#endif
+#ifdef SIGSYS
+ S (SIGSYS),
+#endif
+ {NULL}
+};
+#undef S
+
+static int
+is_numeric_str(char const *s)
+{
+ while (*s) {
+ if (!isdigit(*s))
+ return 0;
+ s++;
+ }
+ return 1;
+}
+
+int
+sig_name_to_str(char const *s)
+{
+ if (is_numeric_str(s)) {
+ char *end;
+ unsigned long n;
+ errno = 0;
+ n = strtoul(s, &end, 10);
+ if (errno || *end || n > UINT_MAX)
+ return -1;
+ return n;
+ } else {
+ struct sigdefn *sd;
+
+ for (sd = sigdefn; sd->sig_name; sd++) {
+ if (s[0] == 's' || s[0] == 'S') {
+ if (strcasecmp(sd->sig_name, s) == 0)
+ return sd->sig_no;
+ } else if (strcasecmp(sd->sig_name + 3, s) == 0)
+ return sd->sig_no;
+ }
+ }
+ return -1;
+}
+
+char const *help_msg[] = {
+ "Usage: genrc [OPTIONS] COMMAND",
+ "generic rc program\n",
+ "COMMANDs are:\n",
+ " start start the program",
+ " stop stop the program",
+ " restart restart the running program",
+ " reload send a reload signal to the program",
+ " status display the program status",
+ "",
+ "OPTIONs are:\n",
+ " -h, --help display this help list",
+ " --usage display short usage information",
+ " -c, --command=COMMAND command line to run",
+ " -p, --program=PROGRAM name of the program to run",
+ " -P, --pid-from=SOURCE where to look for PIDs of the running programs",
+ " -F, --pidfile=NAME name of the PID file",
+ " (same as --pid-from=FILE:NAME)",
+ " -t, --timeout=SECONDS time to wait for the program to start up or",
+ " terminate",
+ " --signal-reload=SIG signal to send on reload (default: SIGHUP)",
+ " setting to 0 is equivalent to --no-reload",
+ " --no-reload makes reload equivalent to restart",
+ " --signal-stop=SIG signal to send in order to terminate the program",
+ " (default: SIGTERM)",
+ " --sentinel PROGRAM runs in foreground; disconnect from the",
+ " controlling terminal, run it and act as a sentinel",
+ " --version display program version and exist",
+ "",
+ "Influential environment variables and corresponding options:",
+ "",
+ " GENRC_COMMAND=COMMAND --command=COMMAND",
+ " GENRC_PROGRAM=NAME --program=NAME",
+ " GENRC_PID_FROM=SOURCE --pid-from=SOURCE",
+ " GENRC_TIMEOUT=SECONDS --timeout=SECONDS",
+ " GENRC_SENTINEL=1 --sentinel",
+ "",
+ " At least one of COMMAND or PROGRAM must be given.",
+ " If PROGRAM is supplied, but COMMAND is not, COMMAND is set to PROGRAM",
+ " Otherwise, if COMMAND is set, but PROGRAM is not, PROGRAM is set to the",
+ " first word (in the shell sense) in COMMAND.",
+ "",
+ "PID sources:",
+ "",
+ " FILE:<NAME>",
+ " Read PID from the file <NAME>",
+ "",
+ " CONFIG:<LANG>:<FILENAME>:<FQRN>",
+ " Name of the PID file is stored in relation <FQRN> of the configuration",
+ " file <FILENAME>, written in language <LANG>",
+ "",
+ " GREP:<FILE>:s/<RX>/<REPL>/[<FLAGS>][;...]",
+ " Grep for the first line in <FILE> that matches <RX>. If found, process",
+ " replace the matched portion according to <REPL> and <FLAGS>. Use",
+ " the resulting string as PID. More sed expressions can be supplied",
+ " separated with semicolons.",
+ "",
+ " PROC[:[<EXE>][:<FLAGS>]]",
+ " Look for matching program in /proc/<PID>/*. If <EXE> is not supplied",
+ " or empty, program name from --program will be used. <FLAGS> are:",
+ " e exact match",
+ " g glob pattern match",
+ " x extended POSIX regexp match (default)",
+ " p PCRE match",
+ " i case-insensitive match",
+ "",
+ " c match entire command line",
+ " r match real executable name (instead of argv0)",
+ " a signal all matching PIDs",
+ "",
+ " PS:[:[<EXE>][:<FLAGS>]]",
+ " Look for matching program in the output of 'ps -ef'. <EXE> and <FLAGS>",
+ " as described above",
+ "",
+ NULL
+};
+
+void
+help(void)
+{
+ int i;
+
+ for (i = 0; help_msg[i]; i++)
+ puts(help_msg[i]);
+}
+
+char const *usage_msg[] = {
+ "genrc",
+ "[-h]",
+ "[-F PIDFILE]",
+ "[-P SOURCE]",
+ "[-c COMMAND]",
+ "[-p PROGRAM]",
+ "[-t SECONDS]",
+ "[--command=COMMAND]",
+ "[--help]",
+ "[--no-reload]",
+ "[--pid-from=SOURCE]",
+ "[--pidfile=PIDFILE]",
+ "[--program=PROGRAM]",
+ "[--sentinel]",
+ "[--signal-reload=SIG]",
+ "[--signal-stop=SIG]",
+ "[--timeout=SECONDS]",
+ "[--usage]",
+ "{",
+ "start",
+ "|",
+ "stop",
+ "|",
+ "restart",
+ "|",
+ "reload",
+ "|",
+ "status",
+ "}",
+ NULL
+};
+
+static int
+screen_width(void)
+{
+ struct winsize ws;
+ ws.ws_col = ws.ws_row = 0;
+ if ((ioctl(1, TIOCGWINSZ, (char *) &ws) < 0) || ws.ws_col == 0) {
+ char *p = getenv("COLUMNS");
+ if (p)
+ ws.ws_col = atoi(p);
+ if (ws.ws_col == 0)
+ ws.ws_col = 80;
+ }
+ return ws.ws_col;
+}
+
+void
+usage(void)
+{
+ int i, j, pos = 0;
+ int width = screen_width();
+ if (width > 4)
+ width -= 4;
+ for (i = j = 0; usage_msg[i]; i++) {
+ int len = strlen(usage_msg[i]);
+ if (j > 0)
+ len++;
+ if (pos + len > width) {
+ putchar('\n');
+ pos = strlen(usage_msg[0]) + 1;
+ printf("%*s", pos, "");
+ j = 0;
+ } else if (j > 0)
+ putchar(' ');
+ j++;
+ printf("%s", usage_msg[i]);
+ pos += len;
+ }
+ putchar('\n');
+}
+
+static const char gplv3[] =
+ "License GPLv3+: GNU GPL version 3 or later "
+ "<http://gnu.org/licenses/gpl.html>\n"
+ "This is free software: you are free to change and redistribute it.\n"
+ "There is NO WARRANTY, to the extent permitted by law.\n\n";
+
+void
+version(void)
+{
+ printf("%s\n", PACKAGE_STRING);
+ printf("Copyryght (C) 2018 Sergey Poznyakoff\n");
+ printf("%s", gplv3);
+}
+
+
+typedef int (*GENRC_COMMAND)(void);
+
+struct comtab {
+ char const *com_name;
+ GENRC_COMMAND com_fun;
+};
+
+static struct comtab comtab[] = {
+ { "start", com_start },
+ { "stop", com_stop },
+ { "restart", com_restart },
+ { "reload", com_reload },
+ { "status", com_status },
+ { NULL, NULL }
+};
+
+GENRC_COMMAND
+find_command(char const *s)
+{
+ struct comtab *c;
+ for (c = comtab; c->com_name; c++)
+ if (strcmp(c->com_name, s) == 0)
+ break;
+ return c->com_fun;
+}
+
+int
+main(int argc, char **argv)
+{
+ int c;
+ char *p;
+ int no_reload = 0;
+ GENRC_COMMAND command;
+
+ setprogname(argv[0]);
+
+ while ((c = getopt_long(argc, argv, shortopts, longopts, NULL))
+ != EOF) {
+ switch (c) {
+ case 'h':
+ help();
+ exit(0);
+ case OPT_USAGE:
+ usage();
+ exit(0);
+ case OPT_VERSION:
+ version();
+ exit(0);
+ case 'c':
+ setenv("GENRC_COMMAND", optarg, 1);
+ break;
+ case 'p':
+ setenv("GENRC_PROGRAM", optarg, 1);
+ break;
+ case 'P':
+ setenv("GENRC_PID_FROM", optarg, 1);
+ break;
+ case 'F':
+ genrc_pidfile = optarg;
+ p = xmalloc(6 + strlen(optarg));
+ strcat(strcpy(p, "FILE:"), optarg);
+ setenv("GENRC_PID_FROM", p, 1);
+ free(p);
+ break;
+ case 't':
+ setenv("GENRC_TIMEOUT", optarg, 1);
+ break;
+ case 'S':
+ setenv("GENRC_SENTINEL", "1", 1);
+ break;
+ case OPT_NO_RELOAD:
+ no_reload = 1;
+ break;
+ case OPT_SIGNAL_RELOAD:
+ setenv("GENRC_SIGNAL_RELOAD", optarg, 1);
+ break;
+ case OPT_SIGNAL_STOP:
+ setenv("GENRC_SIGNAL_STOP", optarg, 1);
+ break;
+ default:
+ exit(1);
+ }
+ }
+
+ if ((p = getenv("GENRC_COMMAND")) != NULL)
+ genrc_command = p;
+ if ((p = getenv("GENRC_PROGRAM")) != NULL)
+ genrc_program = p;
+
+ if (no_reload)
+ genrc_no_reload = 1;
+ else if ((p = getenv("GENRC_SIGNAL_RELOAD")) != NULL) {
+ genrc_signal_reload = sig_name_to_str(p);
+ if (genrc_signal_reload == -1)
+ usage_error("%s: invalid signal number", p);
+ else if (genrc_signal_reload == 0)
+ genrc_no_reload = 1;
+ }
+
+ if ((p = getenv("GENRC_SIGNAL_STOP")) != NULL) {
+ genrc_signal_stop = sig_name_to_str(p);
+ if (genrc_signal_stop <= 0)
+ usage_error("%s: invalid signal number", p);
+ }
+
+ if ((p = getenv("GENRC_TIMEOUT")) != NULL) {
+ char *end;
+ unsigned long n;
+ errno = 0;
+ n = strtoul(p, &end, 10);
+ if (errno || *p || n > UINT_MAX)
+ usage_error("%s: invalid timeout", p);
+ if (n == 0)
+ genrc_no_reload = 1;
+ else
+ genrc_timeout = n;
+ }
+
+ if (!genrc_command) {
+ if (genrc_program) {
+ genrc_command = xstrdup(genrc_program);
+ } else {
+ usage_error("either --command (GENRC_COMMAND) or --program (GENRC_PROGRAM) must be given");
+ }
+ } else if (!genrc_program) {
+ struct wordsplit ws;
+ ws.ws_error = genrc_error;
+ if (wordsplit(genrc_command, &ws,
+ WRDSF_NOCMD|WRDSF_NOVAR|
+ WRDSF_ENOMEMABRT|WRDSF_SHOWERR|WRDSF_ERROR))
+ exit(1);
+ genrc_program = xstrdup(ws.ws_wordv[0]);
+ wordsplit_free(&ws);
+ }
+
+ if ((p = getenv("GENRC_PID_FROM")) != NULL) {
+ genrc_pid_closure = get_pid_closure(p);
+ } else {
+ genrc_pid_closure = get_pid_closure("PROC");
+ }
+
+ argc -= optind;
+ if (argc == 0)
+ usage_error("missing command verb");
+ if (argc > 1)
+ usage_error("too many arguments");
+ argv += optind;
+
+ command = find_command(argv[0]);
+ if (!command)
+ usage_error("unrecognized command: %s", argv[0]);
+ exit(command());
+}
+
diff --git a/src/genrc.h b/src/genrc.h
new file mode 100644
index 0000000..f782fd5
--- a/dev/null
+++ b/src/genrc.h
@@ -0,0 +1,137 @@
+#include <config.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <getopt.h>
+#include <signal.h>
+#include <limits.h>
+#include <errno.h>
+#include <string.h>
+#include <regex.h>
+#include <ctype.h>
+#include <sys/time.h>
+#include <sys/select.h>
+#include <sys/wait.h>
+#include "grecs.h"
+#include "wordsplit.h"
+
+void setprogname(char const *s);
+void genrc_error(char const *fmt, ...);
+void usage_error(char const *fmt, ...);
+void system_error(int ec, char const *fmt, ...);
+
+#define xmalloc grecs_malloc
+#define xzalloc grecs_zalloc
+#define xcalloc grecs_calloc
+#define xrealloc grecs_realloc
+#define xstrdup grecs_strdup
+
+#define SHELL "/bin/sh"
+
+pid_t file_read_pid(char const *filename);
+
+typedef struct transform *TRANSFORM;
+char *transform_string(TRANSFORM tf, const char *input);
+char *transform_string_if_match(TRANSFORM tf, const char *input);
+TRANSFORM compile_transform_expr(const char *expr, int cflags);
+
+struct pidlist {
+ pid_t *pidv;
+ size_t pidc;
+ size_t pidn;
+};
+typedef struct pidlist PIDLIST;
+
+void pidlist_init(PIDLIST *);
+void pidlist_free(PIDLIST *);
+void pidlist_clear(PIDLIST *plist);
+void pidlist_add(PIDLIST *, pid_t);
+ssize_t pidlist_index(PIDLIST *plist, pid_t p);
+int pidlist_member(PIDLIST *plist, pid_t p);
+int pidlist_remove(PIDLIST *plist, size_t i);
+void pidlist_kill(PIDLIST *plist, int sig);
+
+pid_t strtopid(char const *str);
+
+int pid_is_running(pid_t pid);
+
+
+enum {
+ MATCH_REGEX, /* extended POSIX regexp match (default) */
+ MATCH_PCRE, /* PCRE match (not implemented) */
+ MATCH_GLOB, /* glob pattern match */
+ MATCH_EXACT, /* exact match */
+};
+#define MATCH_DEFAULT MATCH_REGEX
+
+enum {
+ PROCF_ICASE = 0x01, /* ignore case */
+ PROCF_ALL = 0x02, /* use all pids */
+ PROCF_CMDLINE = 0x04, /* match against entire command line */
+ PROCF_EXE = 0x08 /* match against executable name */
+};
+#define PROCF_DEFAULT 0
+
+struct procscanbuf {
+ int match; /* match type */
+ int flags; /* match flags */
+ void *pattern;
+};
+
+typedef struct procscanbuf *PROCSCANBUF;
+
+PROCSCANBUF procscan_init(char const *pattern, char const *flagstr);
+void procscan_free(PROCSCANBUF buf);
+int procscan_match(PROCSCANBUF buf, char const *arg);
+
+void match_exact_init(PROCSCANBUF buf, char const *pattern);
+void match_exact_free(PROCSCANBUF buf);
+int match_exact(PROCSCANBUF buf, char const *arg);
+
+void match_glob_init(PROCSCANBUF buf, char const *pattern);
+void match_glob_free(PROCSCANBUF buf);
+int match_glob(PROCSCANBUF buf, char const *arg);
+
+void match_regex_init(PROCSCANBUF buf, char const *pattern);
+void match_regex_free(PROCSCANBUF buf);
+int match_regex(PROCSCANBUF buf, char const *arg);
+
+void match_pcre_init(PROCSCANBUF buf, char const *pattern);
+void match_pcre_free(PROCSCANBUF buf);
+int match_pcre(PROCSCANBUF buf, char const *arg);
+
+
+
+struct genrc_pid_closure {
+ char const *name;
+ int (*pid)(struct genrc_pid_closure *, PIDLIST *);
+};
+
+typedef struct genrc_pid_closure GENRC_PID_CLOSURE;
+GENRC_PID_CLOSURE *get_pid_closure(char const *str);
+int get_pid_list(GENRC_PID_CLOSURE *, PIDLIST *);
+
+GENRC_PID_CLOSURE *genrc_pid_file_init(int argc, char **argv);
+GENRC_PID_CLOSURE *genrc_pid_config_init(int argc, char **argv);
+GENRC_PID_CLOSURE *genrc_pid_grep_init(int argc, char **argv);
+GENRC_PID_CLOSURE *genrc_pid_proc_init(int argc, char **argv);
+GENRC_PID_CLOSURE *genrc_pid_ps_init(int argc, char **argv);
+
+extern char *genrc_command;
+extern char *genrc_program;
+extern char *genrc_pid_from;
+extern unsigned genrc_timeout;
+extern int genrc_no_reload;
+extern int genrc_signal_reload;
+extern int genrc_signal_stop;
+extern GENRC_PID_CLOSURE *genrc_pid_closure;
+extern char *genrc_pidfile;
+
+int sentinel(void);
+
+int com_start(void);
+int com_status(void);
+int com_stop(void);
+int com_restart(void);
+int com_reload(void);
+
diff --git a/src/match_exact.c b/src/match_exact.c
new file mode 100644
index 0000000..49be17f
--- a/dev/null
+++ b/src/match_exact.c
@@ -0,0 +1,20 @@
+#include "genrc.h"
+
+void
+match_exact_init(PROCSCANBUF buf, char const *pattern)
+{
+ buf->pattern = xstrdup(pattern ? pattern : genrc_program);
+}
+
+void
+match_exact_free(PROCSCANBUF buf)
+{
+ free(buf->pattern);
+}
+
+int
+match_exact(PROCSCANBUF buf, char const *arg)
+{
+ return ((buf->flags & PROCF_ICASE) ? strcasecmp : strcmp)
+ ((char*)buf->pattern, arg);
+}
diff --git a/src/match_glob.c b/src/match_glob.c
new file mode 100644
index 0000000..d2dfd71
--- a/dev/null
+++ b/src/match_glob.c
@@ -0,0 +1,22 @@
+#include "genrc.h"
+#define _GNU_SOURCE 1
+#include <fnmatch.h>
+
+void
+match_glob_init(PROCSCANBUF buf, char const *pattern)
+{
+ buf->pattern = xstrdup(pattern ? pattern : genrc_program);
+}
+
+void
+match_glob_free(PROCSCANBUF buf)
+{
+ free(buf->pattern);
+}
+
+int
+match_glob(PROCSCANBUF buf, char const *arg)
+{
+ return fnmatch((char*)buf->pattern, arg,
+ (buf->flags & PROCF_ICASE) ? FNM_CASEFOLD : 0);
+}
diff --git a/src/match_pcre.c b/src/match_pcre.c
new file mode 100644
index 0000000..55c4714
--- a/dev/null
+++ b/src/match_pcre.c
@@ -0,0 +1,60 @@
+#include "genrc.h"
+#include <pcre.h>
+
+static char *
+progname_pattern(char const *prog)
+{
+ grecs_txtacc_t acc;
+ char *ret;
+ static char const specials[] = "\\[]?.*+(){}|";
+
+ acc = grecs_txtacc_create();
+ grecs_txtacc_grow_char(acc, '^');
+ if (prog[0] != '/')
+ grecs_txtacc_grow_string(acc, "(?:/(?:.*/)?)?");
+ for (; *prog; prog++) {
+ if (strchr(specials, *prog))
+ grecs_txtacc_grow_char(acc, '\\');
+ grecs_txtacc_grow_char(acc, *prog);
+ }
+ grecs_txtacc_grow_char(acc, '$');
+ grecs_txtacc_grow_char(acc, 0);
+ ret = grecs_txtacc_finish(acc, 1);
+ grecs_txtacc_free(acc);
+ return ret;
+}
+
+void
+match_pcre_init(PROCSCANBUF buf, char const *pattern)
+{
+ pcre *pcre;
+ char const *error;
+ int error_offset;
+ char *tmp = NULL;
+ int cflags = 0;
+ if (buf->flags & PROCF_ICASE)
+ cflags |= PCRE_CASELESS;
+ if (!pattern) {
+ pattern = tmp = progname_pattern(genrc_program);
+ }
+ pcre = pcre_compile(pattern, cflags, &error, &error_offset, 0);
+ free(tmp);
+ if (!pcre)
+ usage_error("%s at offset %d", error, error_offset);
+ buf->pattern = pcre;
+}
+
+void
+match_pcre_free(PROCSCANBUF buf)
+{
+ pcre_free ((pcre*)buf->pattern);
+}
+
+int
+match_pcre(PROCSCANBUF buf, char const *arg)
+{
+ int rc = pcre_exec((pcre*)buf->pattern, 0, arg, strlen(arg),
+ 0, 0, NULL, 0);
+ return !(rc >= 0);
+ /* FIXME: error message unless rc == PCRE_ERROR_NOMATCH */
+}
diff --git a/src/match_regex.c b/src/match_regex.c
new file mode 100644
index 0000000..fc666c9
--- a/dev/null
+++ b/src/match_regex.c
@@ -0,0 +1,61 @@
+#include "genrc.h"
+
+static char *
+progname_pattern(char const *prog)
+{
+ grecs_txtacc_t acc;
+ char *ret;
+ static char const specials[] = "\\[]?.*+(){}|";
+
+ acc = grecs_txtacc_create();
+ grecs_txtacc_grow_char(acc, '^');
+ if (prog[0] != '/')
+ grecs_txtacc_grow_string(acc, "(/(.*/)?)?");
+ for (; *prog; prog++) {
+ if (strchr(specials, *prog))
+ grecs_txtacc_grow_char(acc, '\\');
+ grecs_txtacc_grow_char(acc, *prog);
+ }
+ grecs_txtacc_grow_char(acc, '$');
+ grecs_txtacc_grow_char(acc, 0);
+ ret = grecs_txtacc_finish(acc, 1);
+ grecs_txtacc_free(acc);
+ return ret;
+}
+
+void
+match_regex_init(PROCSCANBUF buf, char const *pattern)
+{
+ regex_t *rx = xmalloc(sizeof(*rx));
+ int rc;
+ char *tmp = NULL;
+ int cflags = REG_EXTENDED|REG_NOSUB;
+ if (buf->flags & PROCF_ICASE)
+ cflags |= REG_ICASE;
+ if (!pattern) {
+ pattern = tmp = progname_pattern(genrc_program);
+ }
+
+ rc = regcomp(rx, pattern, cflags);
+ free(tmp);
+ if (rc) {
+ char errbuf[512];
+ regerror(rc, rx, errbuf, sizeof(errbuf));
+ usage_error("invalid regular expression: %s", errbuf);
+ }
+ buf->pattern = rx;
+}
+
+void
+match_regex_free(PROCSCANBUF buf)
+{
+ regfree((regex_t*)buf->pattern);
+ free(buf->pattern);
+}
+
+int
+match_regex(PROCSCANBUF buf, char const *arg)
+{
+ return regexec((regex_t*)buf->pattern, arg, 0, NULL, 0);
+}
+
diff --git a/src/pid_config.c b/src/pid_config.c
new file mode 100644
index 0000000..b646cc0
--- a/dev/null
+++ b/src/pid_config.c
@@ -0,0 +1,91 @@
+#include "genrc.h"
+
+typedef struct grecs_node *(*GRECS_PARSER_FUNC)(const char *, int);
+
+struct config_pid_closure {
+ struct genrc_pid_closure generic;
+ char *filename;
+ char *fqrn;
+ GRECS_PARSER_FUNC parser;
+};
+
+static int
+pid_config_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist)
+{
+ struct config_pid_closure *config_clos =
+ (struct config_pid_closure *)clos;
+ struct grecs_node *tree, *node;
+ pid_t pid;
+
+ grecs_parser_fun = config_clos->parser;
+ tree = grecs_parse(config_clos->filename);
+ if (!tree)
+ exit(1);
+ node = grecs_find_node(tree, config_clos->fqrn);
+ if (!node)
+ usage_error("%s: relation %s not found",
+ config_clos->filename,
+ config_clos->fqrn);
+ if (node->type != grecs_node_stmt)
+ usage_error("%s: relation %s is not a statement",
+ config_clos->filename,
+ config_clos->fqrn);
+ if (node->v.value->type != GRECS_TYPE_STRING)
+ usage_error("%s: relation %s is not a string",
+ config_clos->filename,
+ config_clos->fqrn);
+
+ pid = file_read_pid(node->v.value->v.string);
+ grecs_tree_free(tree);
+
+ if (pid == -1)
+ return -1;
+ if (pid > 0)
+ pidlist_add(plist, pid);
+
+ return 0;
+}
+
+struct langtab {
+ char const *lang_name;
+ GRECS_PARSER_FUNC lang_parser;
+};
+
+static struct langtab langtab[] = {
+ { "GRECS", grecs_grecs_parser },
+ { "META1", grecs_meta1_parser },
+ { "BIND", grecs_bind_parser },
+ { "DHCPD", grecs_dhcpd_parser },
+ { "GIT", grecs_git_parser },
+ { "PATH", grecs_path_parser },
+ { NULL, NULL }
+};
+
+static GRECS_PARSER_FUNC
+find_parser(const char *lang)
+{
+ struct langtab *p;
+ for (p = langtab; p->lang_name; p++)
+ if (strcmp(p->lang_name, lang) == 0)
+ break;
+ return p->lang_parser;
+}
+
+GENRC_PID_CLOSURE *
+genrc_pid_config_init(int argc, char **argv)
+{
+ struct config_pid_closure *clos;
+
+ if (argc != 4)
+ usage_error("expected format: CONFIG:<LANG>:<FILE>:<FQRN>");
+
+ clos = xmalloc(sizeof(*clos));
+ clos->generic.pid = pid_config_get;
+ clos->parser = find_parser(argv[1]);
+ if (!clos->parser)
+ usage_error("%s: unsupported parser type", argv[1]);
+ clos->filename = xstrdup(argv[2]);
+ clos->fqrn = xstrdup(argv[3]);
+ return (GENRC_PID_CLOSURE *) clos;
+}
+
diff --git a/src/pid_file.c b/src/pid_file.c
new file mode 100644
index 0000000..344de12
--- a/dev/null
+++ b/src/pid_file.c
@@ -0,0 +1,70 @@
+#include "genrc.h"
+
+struct file_pid_closure {
+ struct genrc_pid_closure generic;
+ char *filename;
+};
+
+pid_t
+file_read_pid(char const *filename)
+{
+ FILE *fp;
+ char buf[256];
+ int n, ec, ok;
+ unsigned long pid;
+ char *end;
+
+ fp = fopen(filename, "r");
+ if (!fp) {
+ if (errno == ENOENT)
+ return 0;
+ system_error(errno, "can't open file %s", filename);
+ return -1;
+ }
+ ok = fgets(buf, sizeof(buf), fp) != NULL;
+ ec = errno;
+ fclose(fp);
+ if (!ok) {
+ system_error(ec, "error reading from %s", filename);
+ return -1;
+ }
+ n = strlen(buf);
+ if (n == 0) {
+ genrc_error("invalid line read from %s", filename);
+ return -1;
+ }
+ if (buf[n-1] == '\n')
+ buf[n-1] = 0;
+ errno = 0;
+ pid = strtoul(buf, &end, 10);
+ if (errno || *end) {
+ genrc_error("invalid line read from %s", filename);
+ return -1;
+ }
+ return pid_is_running(pid) ? pid : 0;
+}
+
+static int
+pid_file_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist)
+{
+ struct file_pid_closure *file_clos = (struct file_pid_closure *)clos;
+ pid_t pid = file_read_pid(file_clos->filename);
+ if (pid == -1)
+ return -1;
+ if (pid > 0)
+ pidlist_add(plist, pid);
+ return 0;
+}
+
+GENRC_PID_CLOSURE *
+genrc_pid_file_init(int argc, char **argv)
+{
+ struct file_pid_closure *clos;
+
+ if (argc != 2)
+ usage_error("%s: expected format: FILE:<FILENAME>", argv[0]);
+ clos = xmalloc(sizeof(*clos));
+ clos->generic.pid = pid_file_get;
+ clos->filename = xstrdup(argv[1]);
+ return (GENRC_PID_CLOSURE *)clos;
+}
diff --git a/src/pid_grep.c b/src/pid_grep.c
new file mode 100644
index 0000000..8869a2d
--- a/dev/null
+++ b/src/pid_grep.c
@@ -0,0 +1,80 @@
+#include "genrc.h"
+
+/*
+GREP:<FILE>:s/<RX>/<REPL>/[<FLAGS>]
+
+Grep for the first line in <FILE> that matches <RX>. If found, process the
+string using the SED expression and return the result.
+*/
+
+struct grep_pid_closure {
+ struct genrc_pid_closure generic;
+ char *filename;
+ TRANSFORM tf;
+};
+
+static int
+pid_grep_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist)
+{
+ struct grep_pid_closure *grep_clos = (struct grep_pid_closure *)clos;
+ FILE *fp;
+ char *buf = NULL;
+ size_t size = 0;
+ ssize_t n;
+ char *res = NULL;
+ pid_t pid;
+
+ fp = fopen(grep_clos->filename, "r");
+ if (!fp) {
+ if (errno == ENOENT)
+ return 0;
+ system_error(errno, "can't open file %s", grep_clos->filename);
+ return -1;
+ }
+
+ while ((n = grecs_getline(&buf, &size, fp)) > 0) {
+ buf[--n] = 0;
+ if (n == 0)
+ continue;
+ res = transform_string_if_match(grep_clos->tf, buf);
+ if (res)
+ break;
+ }
+ free(buf);
+ fclose(fp);
+
+ if (!res) {
+ usage_error("%s: no match for %s",
+ grep_clos->filename,
+ grep_clos->generic.name);
+ return -1;
+ }
+
+ pid = file_read_pid(res);
+ free(res);
+
+ if (pid == -1)
+ return -1;
+ if (pid > 0)
+ pidlist_add(plist, pid);
+
+ return 0;
+}
+
+GENRC_PID_CLOSURE *
+genrc_pid_grep_init(int argc, char **argv)
+{
+ struct grep_pid_closure *clos;
+
+ if (argc != 3)
+ usage_error("expected format: GREP:<FILE>:s/<RX>/<REPL>/[<FLAGS>];...");
+
+ clos = xmalloc(sizeof(*clos));
+ clos->generic.pid = pid_grep_get;
+ clos->filename = xstrdup(argv[1]);
+ clos->tf = compile_transform_expr(argv[2], REG_EXTENDED);
+
+ return (GENRC_PID_CLOSURE *) clos;
+}
+
+
diff --git a/src/pid_proc.c b/src/pid_proc.c
new file mode 100644
index 0000000..e460dff
--- a/dev/null
+++ b/src/pid_proc.c
@@ -0,0 +1,335 @@
+#include "genrc.h"
+#include <glob.h>
+#include <sys/stat.h>
+
+
+char *
+catfile(char const *dir, char const *file)
+{
+ char *ret;
+ size_t dlen, flen;
+
+ dlen = strlen(dir);
+ if (dlen > 0 && dir[dlen-1] == '/')
+ dlen--;
+ flen = strlen(file);
+ ret = xmalloc(dlen + flen + 2);
+ memcpy(ret, dir, dlen);
+ ret[dlen] = '/';
+ strcpy(ret + dlen + 1, file);
+ return ret;
+}
+
+#define INITIAL_READLINK_SIZE 128
+
+static int
+areadlink(const char *name, char **pbuf)
+{
+ size_t status = 0;
+ char *buf;
+ size_t size;
+ ssize_t linklen;
+
+ size = INITIAL_READLINK_SIZE;
+ buf = malloc(size);
+ if (!buf)
+ return -1;
+ while (1) {
+ char *p;
+ size_t newsize = size << 1;
+ if (newsize < size) {
+ status = ENAMETOOLONG;
+ break;
+ }
+ size = newsize;
+ p = realloc(buf, size);
+ if (!p)
+ free(buf);
+ buf = p;
+ if (!buf) {
+ status = ENOMEM;
+ break;
+ }
+
+ linklen = readlink(name, buf, size);
+ if (linklen < 0 && errno != ERANGE) {
+ status = errno;
+ break;
+ }
+
+ if ((size_t) linklen < size) {
+ buf[linklen++] = '\0';
+ status = 0;
+ break;
+ }
+ }
+
+ if (status) {
+ free(buf);
+ errno = status;
+ return -1;
+ }
+ *pbuf = buf;
+ return 0;
+}
+
+pid_t
+strtopid(char const *str)
+{
+ unsigned long n;
+ char *end;
+ errno = 0;
+ n = strtoul(str, &end, 10);
+ if (errno || *end)
+ return -1;
+ return (pid_t) n;
+}
+
+struct proc_pid_closure {
+ struct genrc_pid_closure generic;
+ PROCSCANBUF buf;
+};
+
+pid_t
+parent_pid(pid_t p)
+{
+ FILE *fp;
+ char *filename = NULL;
+ size_t s = 0;
+ char *line = NULL;
+ ssize_t n;
+ struct wordsplit ws;
+ pid_t pid;
+
+ grecs_asprintf(&filename, &s, "/proc/%lu/stat", (unsigned long)p);
+ fp = fopen(filename, "r");
+ if (!fp) {
+ system_error(errno, "can't open file %s", filename);
+ free(filename);
+ return -1;
+ }
+ s = 0;
+ n = grecs_getline(&line, &s, fp);
+ fclose(fp);
+ if (n == -1) {
+ system_error(errno, "error reading from %s", filename);
+ free(filename);
+ return -1;
+ }
+ free(filename);
+ if (n == 0)
+ return 0;
+
+ ws.ws_delim = " ";
+ ws.ws_error = genrc_error;
+ if (wordsplit(line, &ws,
+ WRDSF_NOCMD|WRDSF_NOVAR|WRDSF_QUOTE|
+ WRDSF_DELIM|WRDSF_ENOMEMABRT|WRDSF_SHOWERR|WRDSF_ERROR))
+ exit(1);
+ pid = strtopid(ws.ws_wordv[3]);
+ wordsplit_free(&ws);
+ return pid;
+}
+
+static void
+parent_pids_filter(PIDLIST *plist)
+{
+ int more;
+ size_t i;
+
+ do {
+ more = 0;
+ for (i = 0; i < plist->pidc; ) {
+ pid_t ppid = parent_pid(plist->pidv[i]);
+ if (ppid > 1 && pidlist_member(plist, ppid)) {
+ pidlist_remove(plist, i);
+ more = 1;
+ } else
+ i++;
+ }
+ } while (more);
+}
+
+static int
+match_exe(struct proc_pid_closure *clos, char const *dir)
+{
+ char *fname = catfile(dir, "exe");
+ char *exename;
+ struct stat st;
+ int rc = 0;
+
+ if (lstat(fname, &st)) {
+ if (errno != ENOENT)
+ system_error(errno, "stat %s", fname);
+ rc = -1;
+ } else {
+ if (!S_ISLNK(st.st_mode)) {
+ rc = -1;
+ } else if (areadlink(fname, &exename)) {
+ if (errno != ENOENT)
+ system_error(errno, "readlink %s", fname);
+ rc = -1;
+ }
+ }
+ free(fname);
+ if (rc == 0) {
+ rc = procscan_match(clos->buf, exename);
+ free(exename);
+ }
+ return rc;
+}
+
+static int
+match_cmdline(struct proc_pid_closure *clos, char const *dir)
+{
+ char *fname = catfile(dir, "cmdline");
+ FILE *fp;
+ int c;
+ int nul = 0;
+ grecs_txtacc_t acc;
+ int rc;
+ char *cmdline;
+
+ fp = fopen(fname, "r");
+ if (!fp) {
+ system_error(errno, "fopen %s", fname);
+ free(fname);
+ return -1;
+ }
+ acc = grecs_txtacc_create();
+
+ while ((c = fgetc(fp)) != EOF) {
+ if (nul) {
+ grecs_txtacc_grow_char(acc, ' ');
+ nul = 0;
+ }
+ if (c == 0)
+ nul = 1;
+ else
+ grecs_txtacc_grow_char(acc, c);
+ }
+ fclose(fp);
+ free(fname);
+
+ grecs_txtacc_grow_char(acc, 0);
+ cmdline = grecs_txtacc_finish(acc, 0);
+
+ rc = procscan_match(clos->buf, cmdline);
+ grecs_txtacc_free(acc);
+ return rc;
+}
+
+static int
+match_argv0(struct proc_pid_closure *clos, char const *dir)
+{
+ char *fname = catfile(dir, "cmdline");
+ FILE *fp;
+ int c;
+ grecs_txtacc_t acc;
+ int rc;
+ char *argv0;
+
+ fp = fopen(fname, "r");
+ if (!fp) {
+ system_error(errno, "fopen %s", fname);
+ free(fname);
+ return -1;
+ }
+ acc = grecs_txtacc_create();
+
+ while ((c = fgetc(fp)) != EOF) {
+ grecs_txtacc_grow_char(acc, c);
+ if (c == 0)
+ break;
+ }
+ fclose(fp);
+ free(fname);
+
+ argv0 = grecs_txtacc_finish(acc, 0);
+
+ rc = procscan_match(clos->buf, argv0);
+ grecs_txtacc_free(acc);
+ return rc;
+}
+
+static void
+scandir(struct proc_pid_closure *clos, char const *dir, PIDLIST *plist)
+{
+ int rc;
+ struct stat st;
+
+ if (stat(dir, &st)) {
+ if (errno != ENOENT)
+ system_error(errno, "stat %s", dir);
+ return;
+ }
+ if (!S_ISDIR(st.st_mode))
+ return;
+
+ if (clos->buf->flags & PROCF_CMDLINE) {
+ rc = match_cmdline(clos, dir);
+ } else if (clos->buf->flags & PROCF_EXE) {
+ rc = match_exe(clos, dir);
+ } else {
+ rc = match_argv0(clos, dir);
+ }
+
+ if (rc == 0) {
+ pidlist_add(plist, strtopid(dir + sizeof("/proc")));
+ }
+}
+
+
+static int
+globerr(const char *epath, int ec)
+{
+ system_error(ec, "%s", epath);
+ return 0;
+}
+
+static int
+pid_proc_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist)
+{
+ struct proc_pid_closure *proc_clos = (struct proc_pid_closure *)clos;
+ glob_t gl;
+ char const *dirpat = "/proc/[0-9]*";
+ int i;
+
+ switch (glob(dirpat, GLOB_NOSORT|GLOB_ONLYDIR, globerr, &gl)) {
+ case GLOB_NOSPACE:
+ grecs_alloc_die();
+ case GLOB_ABORTED:
+ genrc_error("read error during glob scan");
+ exit(1);
+ case GLOB_NOMATCH:
+ return 0;
+ }
+
+ for (i = 0; i < gl.gl_pathc; i++) {
+ scandir(proc_clos, gl.gl_pathv[i], plist);
+ }
+ globfree(&gl);
+
+ if (plist->pidc > 1 && !(proc_clos->buf->flags & PROCF_ALL))
+ parent_pids_filter(plist);
+
+ return 0;
+}
+
+GENRC_PID_CLOSURE *
+genrc_pid_proc_init(int argc, char **argv)
+{
+ struct proc_pid_closure *clos;
+
+ if (argc > 3)
+ usage_error("expected format: PROC[:[<EXE>][:<FLAGS>]]");
+
+ clos = xzalloc(sizeof *clos);
+
+ clos->buf = procscan_init((argc >= 2 && argv[1][0] != 0)
+ ? argv[1] : NULL,
+ argc > 2 ? argv[2] : NULL);
+ clos->generic.pid = pid_proc_get;
+
+ return (GENRC_PID_CLOSURE *)clos;
+}
diff --git a/src/pid_ps.c b/src/pid_ps.c
new file mode 100644
index 0000000..573304d
--- a/dev/null
+++ b/src/pid_ps.c
@@ -0,0 +1,137 @@
+#include "genrc.h"
+
+struct ps_pid_closure {
+ struct genrc_pid_closure generic;
+ PROCSCANBUF buf;
+};
+
+enum {
+ UID_COL = 0,
+ PID_COL,
+ PPID_COL,
+ C_COL,
+ STIME_COL,
+ TTY_COL,
+ TIME_COL,
+ CMD_COL
+};
+
+static int
+match_cmdline(struct ps_pid_closure *clos, struct wordsplit const *ws)
+{
+ grecs_txtacc_t acc;
+ size_t i;
+ int rc;
+ char *cmdline;
+
+ acc = grecs_txtacc_create();
+ grecs_txtacc_grow_string(acc, ws->ws_wordv[CMD_COL]);
+ for (i = CMD_COL + 1; i < ws->ws_wordc; i++) {
+ grecs_txtacc_grow_char(acc, ' ');
+ grecs_txtacc_grow_string(acc, ws->ws_wordv[i]);
+ }
+ grecs_txtacc_grow_char(acc, 0);
+ cmdline = grecs_txtacc_finish(acc, 0);
+ rc = procscan_match(clos->buf, cmdline);
+ grecs_txtacc_free(acc);
+ return rc;
+}
+
+static int
+match_argv0(struct ps_pid_closure *clos, struct wordsplit const *ws)
+{
+ return procscan_match(clos->buf, ws->ws_wordv[CMD_COL]);
+}
+
+static int
+match(struct ps_pid_closure *clos, struct wordsplit const *ws)
+{
+ return ((clos->buf->flags & PROCF_CMDLINE)
+ ? match_cmdline : match_argv0)(clos, ws);
+}
+
+static int
+pid_ps_get(GENRC_PID_CLOSURE *clos, PIDLIST *plist)
+{
+ struct ps_pid_closure *ps_clos = (struct ps_pid_closure *)clos;
+ FILE *fp;
+ struct wordsplit ws;
+ int wsflags;
+ ssize_t n;
+ char *buf = NULL;
+ size_t size = 0;
+ size_t line = 0;
+ PIDLIST pplist;
+
+ fp = popen("ps -ef", "r");
+ if (!fp) {
+ system_error(errno, "failed to run ps");
+ return -1;
+ }
+
+ ws.ws_delim = " ";
+ ws.ws_error = genrc_error;
+ wsflags = WRDSF_NOCMD
+ | WRDSF_NOVAR
+ | WRDSF_QUOTE
+ | WRDSF_DELIM
+ | WRDSF_SQUEEZE_DELIMS
+ | WRDSF_ENOMEMABRT
+ | WRDSF_SHOWERR
+ | WRDSF_ERROR;
+ pidlist_init(&pplist);
+ while ((n = grecs_getline(&buf, &size, fp)) > 0) {
+ line++;
+ if (line == 1)
+ continue;
+ buf[n-1] = 0;
+ if (wordsplit(buf, &ws, wsflags))
+ exit(1);
+ wsflags |= WRDSF_REUSE;
+ if (match(ps_clos, &ws) == 0) {
+ pid_t ppid = strtopid(ws.ws_wordv[PPID_COL]);
+ if (!pidlist_member(&pplist, ppid))
+ pidlist_add(&pplist, ppid);
+ pidlist_add(plist, strtopid(ws.ws_wordv[PID_COL]));
+ }
+ }
+ fclose(fp);
+ free(buf);
+ if (wsflags & WRDSF_REUSE)
+ wordsplit_free(&ws);
+ if (plist->pidc > 1 && !(ps_clos->buf->flags & PROCF_ALL)) {
+ size_t i;
+
+ for (i = 0; i < pplist.pidc; ) {
+ if (pidlist_member(plist, pplist.pidv[i]))
+ i++;
+ else {
+ pidlist_remove(&pplist, i);
+ }
+ }
+ pidlist_free(plist);
+ *plist = pplist;
+ }
+
+ return 0;
+}
+
+GENRC_PID_CLOSURE *
+genrc_pid_ps_init(int argc, char **argv)
+{
+ struct ps_pid_closure *clos;
+
+ if (argc > 3)
+ usage_error("expected format: PS[:[<EXE>][:<FLAGS>]]");
+
+ clos = xzalloc(sizeof *clos);
+
+ clos->buf = procscan_init((argc >= 2 && argv[1][0] != 0)
+ ? argv[1] : genrc_program,
+ argc > 2 ? argv[2] : NULL);
+ if (clos->buf->flags & PROCF_EXE)
+ usage_error("flag r isn't compatible with PS");
+ clos->generic.pid = pid_ps_get;
+
+ return (GENRC_PID_CLOSURE *)clos;
+}
diff --git a/src/pidfrom.c b/src/pidfrom.c
new file mode 100644
index 0000000..ab1f623
--- a/dev/null
+++ b/src/pidfrom.c
@@ -0,0 +1,52 @@
+#include "genrc.h"
+
+struct genrc_pid_source {
+ char *name;
+ GENRC_PID_CLOSURE *(*init)(int argc, char **argv);
+};
+
+struct genrc_pid_source sourcetab[] = {
+ { "FILE", genrc_pid_file_init },
+ { "CONFIG", genrc_pid_config_init },
+ { "GREP", genrc_pid_grep_init },
+ { "PROC", genrc_pid_proc_init },
+ { "PS", genrc_pid_ps_init },
+ { NULL }
+};
+
+GENRC_PID_CLOSURE *
+get_pid_closure(char const *str)
+{
+ struct genrc_pid_source *s;
+ struct wordsplit ws;
+ GENRC_PID_CLOSURE *pc;
+
+ ws.ws_error = genrc_error;
+ ws.ws_delim = ":";
+ if (wordsplit(str, &ws,
+ WRDSF_NOCMD|WRDSF_NOVAR|WRDSF_QUOTE|
+ WRDSF_DELIM|WRDSF_ENOMEMABRT|WRDSF_SHOWERR|WRDSF_ERROR))
+ exit(1);
+
+ if (ws.ws_wordc == 0)
+ usage_error("GENRC_PID_FROM argument is empty");
+ for (s = sourcetab; ; s++) {
+ if (!s->name)
+ usage_error("%s: unsupported PID source", str);
+ if (strcmp(s->name, ws.ws_wordv[0]) == 0)
+ break;
+ }
+ pc = s->init(ws.ws_wordc, ws.ws_wordv);
+ if (!pc)
+ exit(1);
+ pc->name = xstrdup(str);
+ return pc;
+}
+
+int
+get_pid_list(GENRC_PID_CLOSURE *clos, PIDLIST *plist)
+{
+ pidlist_clear(plist);
+ return clos->pid(clos, plist);
+}
+
diff --git a/src/pidlist.c b/src/pidlist.c
new file mode 100644
index 0000000..cf47c2e
--- a/dev/null
+++ b/src/pidlist.c
@@ -0,0 +1,121 @@
+#include "genrc.h"
+
+void *
+x2nrealloc(void *p, size_t *pn, size_t s)
+{
+ size_t n = *pn;
+
+ if (!p) {
+ if (!n) {
+ /* The approximate size to use for initial small
+ allocation requests, when the invoking code
+ specifies an old size of zero. 64 bytes is
+ the largest "small" request for the
+ GNU C library malloc. */
+ enum { DEFAULT_MXFAST = 64 };
+
+ n = DEFAULT_MXFAST / s;
+ n += !n;
+ }
+ } else {
+ /* Set N = ceil (1.5 * N) so that progress is made if N == 1.
+ Check for overflow, so that N * S stays in size_t range.
+ The check is slightly conservative, but an exact check isn't
+ worth the trouble. */
+ if ((size_t) -1 / 3 * 2 / s <= n)
+ grecs_alloc_die();
+ n += (n + 1) / 2;
+ }
+
+ *pn = n;
+ return grecs_realloc(p, n * s);
+}
+
+void
+pidlist_init(PIDLIST *plist)
+{
+ plist->pidv = NULL;
+ plist->pidc = 0;
+ plist->pidn = 0;
+}
+
+void
+pidlist_free(PIDLIST *plist)
+{
+ free(plist->pidv);
+}
+
+void
+pidlist_clear(PIDLIST *plist)
+{
+ plist->pidc = 0;
+}
+
+void
+pidlist_add(PIDLIST *plist, pid_t p)
+{
+ if (p <= 0)
+ return;
+ if (plist->pidc == plist->pidn) {
+ plist->pidv = x2nrealloc(plist->pidv, &plist->pidn,
+ sizeof(plist->pidv[0]));
+ }
+ plist->pidv[plist->pidc++] = p;
+}
+
+ssize_t
+pidlist_index(PIDLIST *plist, pid_t p)
+{
+ size_t i;
+ for (i = 0; i < plist->pidc; i++) {
+ if (plist->pidv[i] == p)
+ return i;
+ }
+ return -1;
+}
+
+int
+pidlist_member(PIDLIST *plist, pid_t p)
+{
+ return pidlist_index(plist, p) != -1;
+}
+
+int
+pidlist_remove(PIDLIST *plist, size_t i)
+{
+ if (i == -1)
+ return 0;
+
+ if (i < plist->pidc - 1) {
+ memmove(plist->pidv + i, plist->pidv + i + 1,
+ (plist->pidc - i - 1) * sizeof(plist->pidv[0]));
+ }
+ plist->pidc--;
+ return 1;
+}
+
+void
+pidlist_kill(PIDLIST *plist, int sig)
+{
+ size_t i;
+
+ for (i = 0; i < plist->pidc; i++) {
+ if (kill(plist->pidv[i], sig)) {
+ if (errno != ESRCH)
+ system_error(errno, "kill %lu",
+ (unsigned long) plist->pidv[i]);
+ }
+ }
+}
+
+int
+pid_is_running(pid_t pid)
+{
+ if (kill(pid, 0)) {
+ if (errno == ESRCH)
+ return 0;
+ system_error(errno, "kill %lu", (unsigned long)pid);
+ exit(1);
+ }
+ return 1;
+}
diff --git a/src/procscan.c b/src/procscan.c
new file mode 100644
index 0000000..1425f6a
--- a/dev/null
+++ b/src/procscan.c
@@ -0,0 +1,95 @@
+#include "genrc.h"
+
+struct flagdfn {
+ int letter; /* flag letter */
+ int is_match; /* is this flag a match mode */
+ int value; /* flag value */
+};
+
+static struct flagdfn flagtab[] = {
+ { 'e', 1, MATCH_EXACT },
+ { 'g', 1, MATCH_GLOB },
+ { 'x', 1, MATCH_REGEX },
+ { 'p', 1, MATCH_PCRE },
+ { 'i', 0, PROCF_ICASE },
+ { 'a', 0, PROCF_ALL },
+ { 'c', 0, PROCF_CMDLINE },
+ { 'r', 0, PROCF_EXE },
+ { 0 }
+};
+
+static struct flagdfn *
+find_flag(int let)
+{
+ struct flagdfn *p;
+
+ let = tolower(let);
+ for (p = flagtab; p->letter; p++)
+ if (p->letter == let)
+ return p;
+ return NULL;
+}
+
+struct match_type {
+ void (*match_init)(PROCSCANBUF, char const *);
+ void (*match_free)(PROCSCANBUF);
+ int (*match_func)(PROCSCANBUF, char const *);
+};
+
+struct match_type match_types[] = {
+ [MATCH_REGEX] = { match_regex_init, match_regex_free, match_regex },
+#ifdef HAVE_PCRE
+ [MATCH_PCRE] = { match_pcre_init, match_pcre_free, match_pcre },
+#endif
+ [MATCH_GLOB] = { match_glob_init, match_glob_free, match_glob },
+ [MATCH_EXACT] = { match_exact_init, match_exact_free, match_exact }
+};
+
+PROCSCANBUF
+procscan_init(char const *pattern, char const *flagstr)
+{
+ PROCSCANBUF buf;
+
+ buf = xmalloc(sizeof(buf));
+ buf->match = MATCH_DEFAULT;
+ buf->flags = PROCF_DEFAULT;
+ if (flagstr) {
+ char const *p;
+ for (p = flagstr; *p; p++) {
+ struct flagdfn *flag = find_flag(*p);
+ if (!flag)
+ usage_error("%s: unknown flag %c",
+ flagstr, *p);
+ if (flag->is_match) {
+ if (match_types[flag->value].match_init)
+ buf->match = flag->value;
+ else
+ usage_error("match type '%c' is not available", *p);
+ }
+ else if (isupper(*p))
+ buf->flags &= ~flag->value;
+ else
+ buf->flags |= flag->value;
+ }
+ }
+ match_types[buf->match].match_init(buf, pattern);
+ return buf;
+}
+
+void
+procscan_free(PROCSCANBUF buf)
+{
+ match_types[buf->match].match_free(buf);
+ free(buf);
+}
+
+int
+procscan_match(PROCSCANBUF buf, char const *arg)
+{
+ if (!arg)
+ return 1;
+ return match_types[buf->match].match_func(buf, arg);
+}
+
+
+
diff --git a/src/sentinel.c b/src/sentinel.c
new file mode 100644
index 0000000..0fe0714
--- a/dev/null
+++ b/src/sentinel.c
@@ -0,0 +1,207 @@
+#include "genrc.h"
+#include <unistd.h>
+#include <fcntl.h>
+#include <syslog.h>
+
+static void
+xpipe(int p[2])
+{
+ if (pipe(p)) {
+ system_error(errno, "pipe");
+ exit(1);
+ }
+}
+
+#define max(a,b) ((a) > (b) ? (a) : (b))
+
+static void
+write_pid_file(pid_t pid)
+{
+ if (genrc_pidfile) {
+ FILE *fp;
+ fp = fopen(genrc_pidfile, "w");
+ fprintf(fp, "%lu\n", (unsigned long)pid);
+ fclose(fp);
+ }
+}
+
+#define LOGBUFSIZE 1024
+
+struct log_buffer {
+ char buf[LOGBUFSIZE];
+ size_t pos;
+ int prio;
+};
+
+static void
+log_buffer_init(struct log_buffer *lb, int prio)
+{
+ lb->buf[sizeof(lb->buf) - 1] = 0;
+ lb->pos = 0;
+ lb->prio = prio;
+}
+
+static void
+log_buffer_read(int fd, struct log_buffer *lb)
+{
+ int rc;
+
+ rc = read(fd, lb->buf + lb->pos, 1);
+ if (rc == -1) {
+ syslog(LOG_ERR, "read: %m");
+ } else if (rc == 1) {
+ if (lb->pos == sizeof(lb->buf)-1) {
+ syslog(lb->prio, "%s", lb->buf);
+ lb->pos = 0;
+ } else if (lb->buf[lb->pos] == '\n') {
+ lb->buf[lb->pos] = 0;
+ syslog(lb->prio, "%s", lb->buf);
+ lb->pos = 0;
+ } else
+ lb->pos++;
+ }
+}
+
+void
+wait_loop(pid_t child, int out, int err)
+{
+ fd_set rdset;
+ int nfd = (out > err ? out : err) + 1;
+ struct log_buffer obuf, ebuf;
+
+ openlog(genrc_program, LOG_PID, LOG_DAEMON);
+ log_buffer_init(&obuf, LOG_INFO);
+ log_buffer_init(&ebuf, LOG_ERR);
+
+ while (1) {
+ int rc, status;
+
+ if (waitpid(child, &status, WNOHANG) == child) {
+ if (genrc_pidfile)
+ unlink(genrc_pidfile);
+ if (WIFEXITED(status)) {
+ syslog(LOG_INFO, "%s exited with status %d",
+ genrc_program, WEXITSTATUS(status));
+ _exit(WEXITSTATUS(status));
+ } else if (WIFSIGNALED(status)) {
+ char const *coremsg = "";
+#ifdef WCOREDUMP
+ if (WCOREDUMP(status))
+ coremsg = " (core dumped)";
+#endif
+ syslog(LOG_INFO, "%s terminated on signal %d%s",
+ genrc_program, WTERMSIG(status), coremsg);
+ } else if (WIFSTOPPED(status)) {
+ syslog(LOG_INFO, "%s stopped on signal %d",
+ genrc_program, WSTOPSIG(status));
+ } else {
+ syslog(LOG_INFO, "%s terminated; status %d",
+ genrc_program, rc);
+ }
+ break;
+ }
+
+ FD_ZERO(&rdset);
+ FD_SET(out, &rdset);
+ FD_SET(err, &rdset);
+ rc = select(nfd, &rdset, NULL, NULL, NULL);
+ if (rc == -1) {
+ if (errno == EINTR)
+ continue;
+ syslog(LOG_CRIT, "select: %m");
+ syslog(LOG_CRIT, "killing %lu and terminating",
+ (unsigned long) child);
+ kill(child, SIGKILL);
+ break;
+ }
+
+ if (FD_ISSET(out, &rdset)) {
+ log_buffer_read(out, &obuf);
+ }
+ if (FD_ISSET(err, &rdset)) {
+ log_buffer_read(err, &ebuf);
+ }
+ }
+ _exit(1);
+}
+
+pid_t
+start_command(int p[])
+{
+ int errpipe[2], outpipe[2];
+ pid_t pid;
+
+ xpipe(errpipe);
+ xpipe(outpipe);
+ pid = fork();
+ if (pid == -1) {
+ system_error(errno, "pipe");
+ return -1;
+ }
+
+ if (pid == 0) {
+ char *argv[] = { SHELL, "-c", NULL, NULL };
+ int i;
+
+ close(0);
+ close(1);
+ close(2);
+ open("/dev/null", O_RDWR);
+ dup(outpipe[1]);
+ dup(errpipe[1]);
+
+ i = max(max(outpipe[0],outpipe[1]),
+ max(errpipe[0],errpipe[1]));
+ while (--i > 2)
+ close(i);
+
+ argv[2] = genrc_command;
+ execvp(SHELL, argv);
+ system_error(errno, "failed to exec %s", genrc_program);
+ exit(127);
+ }
+
+ write_pid_file(pid);
+
+ close(outpipe[1]);
+ close(errpipe[1]);
+
+ p[0] = outpipe[0];
+ p[1] = errpipe[0];
+ return pid;
+}
+
+int
+sentinel(void)
+{
+ pid_t pid;
+ int p[2];
+
+ /* Detach from the controlling terminal */
+ pid = fork();
+ if (pid == -1) {
+ system_error(errno, "fork");
+ return -1;
+ }
+ if (pid) /* master */
+ return 0;
+ /* Run as a session leader */
+ setsid();
+
+ pid = fork();
+ if (pid == -1) {
+ system_error(errno, "fork");
+ exit(1);
+ }
+ if (pid)
+ _exit(0);
+
+ /* Grand-child */
+ pid = start_command(p);
+ if (pid == -1)
+ _exit(127);
+ wait_loop(pid, p[0], p[1]);
+ _exit(1);
+}
+
+
diff --git a/src/transform.c b/src/transform.c
new file mode 100644
index 0000000..d3cccb1
--- a/dev/null
+++ b/src/transform.c
@@ -0,0 +1,560 @@
+/* This file is part of genrc.
+ Copyright (C) 2006-2018 Sergey Poznyakoff.
+ (using my implementation for the GNU tar).
+
+ Genrc is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 3, or (at your option)
+ any later version.
+
+ Genrc is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with genrc. If not, see <http://www.gnu.org/licenses/>. */
+
+#include "genrc.h"
+
+enum transform_type {
+ transform_first,
+ transform_global
+};
+
+enum replace_segm_type {
+ segm_literal, /* Literal segment */
+ segm_backref, /* Back-reference segment */
+ segm_case_ctl /* Case control segment (GNU extension) */
+};
+
+enum case_ctl_type {
+ ctl_stop, /* Stop case conversion */
+ ctl_upcase_next, /* Turn the next character to uppercase */
+ ctl_locase_next, /* Turn the next character to lowercase */
+ ctl_upcase, /* Turn the replacement to uppercase until ctl_stop */
+ ctl_locase /* Turn the replacement to lowercase until ctl_stop */
+};
+
+struct replace_segm {
+ struct replace_segm *next;
+ enum replace_segm_type type;
+ union {
+ struct {
+ char *ptr;
+ size_t size;
+ } literal; /* type == segm_literal */
+ size_t ref; /* type == segm_backref */
+ enum case_ctl_type ctl; /* type == segm_case_ctl */
+ } v;
+};
+
+struct transform {
+ struct transform *next;
+ enum transform_type transform_type;
+ unsigned match_number;
+ regex_t regex;
+ /* Compiled replacement expression */
+ struct replace_segm *repl_head, *repl_tail;
+ size_t segm_count; /* Number of elements in the above list */
+};
+
+struct transform_list {
+ struct transform *head, *tail;
+};
+
+static struct transform *
+new_transform(struct transform_list *tlist)
+{
+ struct transform *p = xzalloc(sizeof *p);
+ if (tlist->tail)
+ tlist->tail->next = p;
+ else
+ tlist->head = p;
+ tlist->tail = p;
+ return p;
+}
+
+static struct replace_segm *
+add_segment(struct transform *tf)
+{
+ struct replace_segm *segm = xmalloc(sizeof *segm);
+ segm->next = NULL;
+ if (tf->repl_tail)
+ tf->repl_tail->next = segm;
+ else
+ tf->repl_head = segm;
+ tf->repl_tail = segm;
+ tf->segm_count++;
+ return segm;
+}
+
+static void
+add_literal_segment(struct transform *tf, char *str, char *end)
+{
+ size_t len = end - str;
+ if (len) {
+ struct replace_segm *segm = add_segment(tf);
+ segm->type = segm_literal;
+ segm->v.literal.ptr = xmalloc(len + 1);
+ memcpy(segm->v.literal.ptr, str, len);
+ segm->v.literal.ptr[len] = 0;
+ segm->v.literal.size = len;
+ }
+}
+
+static void
+add_char_segment(struct transform *tf, int chr)
+{
+ struct replace_segm *segm = add_segment(tf);
+ segm->type = segm_literal;
+ segm->v.literal.ptr = xmalloc(2);
+ segm->v.literal.ptr[0] = chr;
+ segm->v.literal.ptr[1] = 0;
+ segm->v.literal.size = 1;
+}
+
+static void
+add_backref_segment(struct transform *tf, size_t ref)
+{
+ struct replace_segm *segm = add_segment(tf);
+ segm->type = segm_backref;
+ segm->v.ref = ref;
+}
+
+static void
+add_case_ctl_segment(struct transform *tf, enum case_ctl_type ctl)
+{
+ struct replace_segm *segm = add_segment(tf);
+ segm->type = segm_case_ctl;
+ segm->v.ctl = ctl;
+}
+
+static const char *
+parse_transform_expr(struct transform_list *tlist,
+ const char *expr, int cflags)
+{
+ int delim;
+ int i, j, rc;
+ char *str, *beg, *cur;
+ const char *p;
+ struct transform *tf = new_transform(tlist);
+
+ if (expr[0] != 's')
+ usage_error("invalid transform expression: %s", expr);
+
+ delim = expr[1];
+
+ /* Scan regular expression */
+ for (i = 2; expr[i] && expr[i] != delim; i++)
+ if (expr[i] == '\\' && expr[i + 1])
+ i++;
+
+ if (expr[i] != delim)
+ usage_error("missing 2nd delimiter in position %d of expression %s",
+ i, expr);
+
+ /* Scan replacement expression */
+ for (j = i + 1; expr[j] && expr[j] != delim; j++)
+ if (expr[j] == '\\' && expr[j + 1])
+ j++;
+
+ if (expr[j] != delim)
+ usage_error("missing trailing delimiter in position %d of expression %s",
+ j, expr);
+
+ /* Check flags */
+ tf->transform_type = transform_first;
+ for (p = expr + j + 1; *p && *p != ';'; p++)
+ switch (*p) {
+ case 'g':
+ tf->transform_type = transform_global;
+ break;
+
+ case 'i':
+ cflags |= REG_ICASE;
+ break;
+
+ case 'x':
+ cflags |= REG_EXTENDED;
+ break;
+
+ case 'b':
+ cflags &= ~REG_EXTENDED;
+ break;
+
+ case '0':
+ case '1':
+ case '2':
+ case '3':
+ case '4':
+ case '5':
+ case '6':
+ case '7':
+ case '8':
+ case '9':
+ tf->match_number = strtoul(p, (char **)&p, 0);
+ p--;
+ break;
+
+ default:
+ usage_error("unknown flag in transform expression: %c",
+ *p);
+ }
+
+ if (*p == ';')
+ p++;
+
+ /* Extract and compile regex */
+ str = xmalloc(i - 1);
+ memcpy(str, expr + 2, i - 2);
+ str[i - 2] = 0;
+
+ rc = regcomp(&tf->regex, str, cflags);
+
+ if (rc) {
+ char errbuf[512];
+ regerror(rc, &tf->regex, errbuf, sizeof(errbuf));
+ usage_error("invalid transform expression: %s", errbuf);
+ }
+
+ if (str[0] == '^' || str[strlen(str) - 1] == '$')
+ tf->transform_type = transform_first;
+
+ free(str);
+
+ /* Extract and compile replacement expr */
+ i++;
+ str = xmalloc(j - i + 1);
+ memcpy(str, expr + i, j - i);
+ str[j - i] = 0;
+
+ for (cur = beg = str; *cur;) {
+ if (*cur == '\\') {
+ size_t n;
+
+ add_literal_segment(tf, beg, cur);
+ switch (*++cur) {
+ case '0':
+ case '1':
+ case '2':
+ case '3':
+ case '4':
+ case '5':
+ case '6':
+ case '7':
+ case '8':
+ case '9':
+ n = strtoul(cur, &cur, 10);
+ if (n > tf->regex.re_nsub)
+ usage_error("invalid transform replacement: "
+ "back reference out of range");
+ add_backref_segment(tf, n);
+ break;
+
+ case '\\':
+ add_char_segment(tf, '\\');
+ cur++;
+ break;
+
+ case 'a':
+ add_char_segment(tf, '\a');
+ cur++;
+ break;
+
+ case 'b':
+ add_char_segment(tf, '\b');
+ cur++;
+ break;
+
+ case 'f':
+ add_char_segment(tf, '\f');
+ cur++;
+ break;
+
+ case 'n':
+ add_char_segment(tf, '\n');
+ cur++;
+ break;
+
+ case 'r':
+ add_char_segment(tf, '\r');
+ cur++;
+ break;
+
+ case 't':
+ add_char_segment(tf, '\t');
+ cur++;
+ break;
+
+ case 'v':
+ add_char_segment(tf, '\v');
+ cur++;
+ break;
+
+ case '&':
+ add_char_segment(tf, '&');
+ cur++;
+ break;
+
+ case 'L':
+ /* Turn the replacement to lowercase until
+ a `\U' or `\E' is found, */
+ add_case_ctl_segment(tf, ctl_locase);
+ cur++;
+ break;
+
+ case 'l':
+ /* Turn the next character to lowercase, */
+ add_case_ctl_segment(tf, ctl_locase_next);
+ cur++;
+ break;
+
+ case 'U':
+ /* Turn the replacement to uppercase until
+ a `\L' or `\E' is found, */
+ add_case_ctl_segment(tf, ctl_upcase);
+ cur++;
+ break;
+
+ case 'u':
+ /* Turn the next character to uppercase, */
+ add_case_ctl_segment(tf, ctl_upcase_next);
+ cur++;
+ break;
+
+ case 'E':
+ /* Stop case conversion started by
+ `\L' or `\U'. */
+ add_case_ctl_segment(tf, ctl_stop);
+ cur++;
+ break;
+
+ default:
+ if (*cur == delim)
+ add_char_segment(tf, delim);
+ else {
+ char buf[2];
+ buf[0] = '\\';
+ buf[1] = *cur;
+ add_literal_segment(tf, buf, buf + 2);
+ }
+ cur++;
+ break;
+ }
+ beg = cur;
+ } else if (*cur == '&') {
+ add_literal_segment(tf, beg, cur);
+ add_backref_segment(tf, 0);
+ beg = ++cur;
+ } else
+ cur++;
+ }
+ add_literal_segment(tf, beg, cur);
+
+ return p;
+}
+
+TRANSFORM
+compile_transform_expr(const char *expr, int cflags)
+{
+ struct transform_list tlist = { NULL, NULL };
+ while (*expr)
+ expr = parse_transform_expr(&tlist, expr, cflags);
+ return tlist.head;
+}
+
+/* Run case conversion specified by CASE_CTL on array PTR of SIZE
+ characters. Returns pointer to statically allocated storage. */
+static const char *
+run_case_conv(enum case_ctl_type case_ctl, const char *ptr, size_t size)
+{
+ static char *case_ctl_buffer;
+ static size_t case_ctl_bufsize;
+ char *p;
+
+ if (case_ctl_bufsize < size) {
+ case_ctl_bufsize = size;
+ case_ctl_buffer = xrealloc(case_ctl_buffer, case_ctl_bufsize);
+ }
+ memcpy(case_ctl_buffer, ptr, size);
+ switch (case_ctl) {
+ case ctl_upcase_next:
+ case_ctl_buffer[0] = toupper(case_ctl_buffer[0]);
+ break;
+
+ case ctl_locase_next:
+ case_ctl_buffer[0] = tolower(case_ctl_buffer[0]);
+ break;
+
+ case ctl_upcase:
+ for (p = case_ctl_buffer; p < case_ctl_buffer + size; p++)
+ *p = toupper(*p);
+ break;
+
+ case ctl_locase:
+ for (p = case_ctl_buffer; p < case_ctl_buffer + size; p++)
+ *p = tolower(*p);
+ break;
+
+ case ctl_stop:
+ break;
+ }
+ return case_ctl_buffer;
+}
+
+
+struct case_ctl_state {
+ enum case_ctl_type case_ctl; /* Current case conversion op */
+ enum case_ctl_type save_ctl; /* Saved case_ctl for \u and \l */
+};
+
+/* Reset case conversion after a single-char operation */
+static void
+case_ctl_reset(struct case_ctl_state *state)
+{
+ if (state->case_ctl == ctl_upcase_next
+ || state->case_ctl == ctl_locase_next) {
+ state->case_ctl = state->save_ctl;
+ state->save_ctl = ctl_stop;
+ }
+}
+
+static void
+repl_loop(struct transform *tf, grecs_txtacc_t acc,
+ const char *input,
+ regmatch_t *rmp,
+ struct case_ctl_state *state)
+{
+ const char *ptr;
+ struct replace_segm *segm;
+
+ for (segm = tf->repl_head; segm; segm = segm->next) {
+ switch (segm->type) {
+ case segm_literal:
+ /* Literal segment */
+ if (state->case_ctl == ctl_stop)
+ ptr = segm->v.literal.ptr;
+ else {
+ ptr = run_case_conv(state->case_ctl,
+ segm->v.literal.ptr,
+ segm->v.literal.size);
+ case_ctl_reset(state);
+ }
+ grecs_txtacc_grow(acc, ptr, segm->v.literal.size);
+ break;
+
+ case segm_backref:
+ /* Back-reference segment */
+ if (rmp[segm->v.ref].rm_so != -1
+ && rmp[segm->v.ref].rm_eo != -1) {
+ size_t size = rmp[segm->v.ref].rm_eo -
+ rmp[segm->v.ref].rm_so;
+ ptr = input + rmp[segm->v.ref].rm_so;
+ if (state->case_ctl != ctl_stop) {
+ ptr = run_case_conv(state->case_ctl,
+ ptr, size);
+ case_ctl_reset(state);
+ }
+ grecs_txtacc_grow(acc, ptr, size);
+ }
+ break;
+
+ case segm_case_ctl:
+ switch (segm->v.ctl) {
+ case ctl_upcase_next:
+ case ctl_locase_next:
+ switch (state->save_ctl) {
+ case ctl_stop:
+ case ctl_upcase:
+ case ctl_locase:
+ state->save_ctl = state->case_ctl;
+ default:
+ break;
+ }
+ /*FALL THROUGH */
+
+ case ctl_upcase:
+ case ctl_locase:
+ case ctl_stop:
+ state->case_ctl = segm->v.ctl;
+ }
+ }
+ }
+}
+
+void
+_single_transform_name_to_acc(struct transform *tf, grecs_txtacc_t acc,
+ const char *input)
+{
+ regmatch_t *rmp;
+ int rc;
+ size_t nmatches = 0;
+ struct case_ctl_state ctl_state = { ctl_stop, ctl_stop };
+
+ rmp = xmalloc((tf->regex.re_nsub + 1) * sizeof(*rmp));
+
+ while (*input) {
+ size_t disp;
+
+ rc = regexec(&tf->regex, input, tf->regex.re_nsub + 1, rmp, 0);
+
+ if (rc == 0) {
+ disp = rmp[0].rm_eo;
+
+ if (rmp[0].rm_so)
+ grecs_txtacc_grow(acc, input, rmp[0].rm_so);
+
+ nmatches++;
+ if (tf->match_number && nmatches < tf->match_number) {
+ grecs_txtacc_grow(acc, input, disp);
+ input += disp;
+ continue;
+ }
+ repl_loop(tf, acc, input, rmp, &ctl_state);
+ } else {
+ disp = strlen(input);
+ grecs_txtacc_grow(acc, input, disp);
+ }
+
+ input += disp;
+
+ if (tf->transform_type == transform_first) {
+ grecs_txtacc_grow_string(acc, input);
+ break;
+ }
+ }
+
+ free(rmp);
+ grecs_txtacc_grow_char(acc, 0);
+}
+
+char *
+transform_string(TRANSFORM tf, const char *input)
+{
+ if (tf) {
+ grecs_txtacc_t acc = grecs_txtacc_create();
+ char *buf = NULL;
+
+ for (; tf; tf = tf->next) {
+ _single_transform_name_to_acc(tf, acc, input);
+ if (buf)
+ grecs_txtacc_free_string(acc, buf);
+ buf = grecs_txtacc_finish(acc, 0);
+ input = buf;
+ }
+ buf = xstrdup(buf);
+ grecs_txtacc_free(acc);
+ return buf;
+ }
+ return xstrdup(input);
+}
+
+char *
+transform_string_if_match(TRANSFORM tf, const char *input)
+{
+ int rc = regexec(&tf->regex, input, 0, NULL, 0);
+ if (rc == 0)
+ return transform_string(tf, input);
+ return NULL;
+}

Return to:

Send suggestions and report system problems to the System administrator.