| /* See LICENSE file for copyright and license details. */ |
| #include <sys/types.h> |
| #include <sys/wait.h> |
| |
| #include <errno.h> |
| #include <limits.h> |
| #include <signal.h> |
| #include <stdarg.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| #include <ctype.h> |
| #include <string.h> |
| #include <syslog.h> |
| #include <time.h> |
| #include <unistd.h> |
| |
| #include "queue.h" |
| #include "util.h" |
| |
| struct field { |
| enum { |
| ERROR, |
| WILDCARD, |
| NUMBER, |
| RANGE, |
| REPEAT, |
| LIST |
| } type; |
| long *val; |
| int len; |
| }; |
| |
| struct ctabentry { |
| struct field min; |
| struct field hour; |
| struct field mday; |
| struct field mon; |
| struct field wday; |
| char *cmd; |
| TAILQ_ENTRY(ctabentry) entry; |
| }; |
| |
| struct jobentry { |
| char *cmd; |
| pid_t pid; |
| TAILQ_ENTRY(jobentry) entry; |
| }; |
| |
| static sig_atomic_t chldreap; |
| static sig_atomic_t reload; |
| static sig_atomic_t quit; |
| static TAILQ_HEAD(, ctabentry) ctabhead = TAILQ_HEAD_INITIALIZER(ctabhead); |
| static TAILQ_HEAD(, jobentry) jobhead = TAILQ_HEAD_INITIALIZER(jobhead); |
| static char *config = "/etc/crontab"; |
| static char *pidfile = "/var/run/crond.pid"; |
| static int nflag; |
| |
| static void |
| loginfo(const char *fmt, ...) |
| { |
| va_list ap; |
| va_start(ap, fmt); |
| if (nflag == 0) |
| vsyslog(LOG_INFO, fmt, ap); |
| else |
| vfprintf(stdout, fmt, ap); |
| fflush(stdout); |
| va_end(ap); |
| } |
| |
| static void |
| logwarn(const char *fmt, ...) |
| { |
| va_list ap; |
| va_start(ap, fmt); |
| if (nflag == 0) |
| vsyslog(LOG_WARNING, fmt, ap); |
| else |
| vfprintf(stderr, fmt, ap); |
| va_end(ap); |
| } |
| |
| static void |
| logerr(const char *fmt, ...) |
| { |
| va_list ap; |
| va_start(ap, fmt); |
| if (nflag == 0) |
| vsyslog(LOG_ERR, fmt, ap); |
| else |
| vfprintf(stderr, fmt, ap); |
| va_end(ap); |
| } |
| |
| static void |
| runjob(char *cmd) |
| { |
| struct jobentry *je; |
| time_t t; |
| pid_t pid; |
| |
| t = time(NULL); |
| |
| /* If command is already running, skip it */ |
| TAILQ_FOREACH(je, &jobhead, entry) { |
| if (strcmp(je->cmd, cmd) == 0) { |
| loginfo("already running %s pid: %d at %s", |
| je->cmd, je->pid, ctime(&t)); |
| return; |
| } |
| } |
| |
| switch ((pid = fork())) { |
| case -1: |
| logerr("error: failed to fork job: %s time: %s", |
| cmd, ctime(&t)); |
| return; |
| case 0: |
| setsid(); |
| loginfo("run: %s pid: %d at %s", |
| cmd, getpid(), ctime(&t)); |
| execl("/bin/sh", "/bin/sh", "-c", cmd, (char *)NULL); |
| logerr("error: failed to execute job: %s time: %s", |
| cmd, ctime(&t)); |
| _exit(1); |
| default: |
| je = emalloc(sizeof(*je)); |
| je->cmd = estrdup(cmd); |
| je->pid = pid; |
| TAILQ_INSERT_TAIL(&jobhead, je, entry); |
| } |
| } |
| |
| static void |
| waitjob(void) |
| { |
| struct jobentry *je, *tmp; |
| int status; |
| time_t t; |
| pid_t pid; |
| |
| t = time(NULL); |
| |
| while ((pid = waitpid(-1, &status, WNOHANG | WUNTRACED)) > 0) { |
| je = NULL; |
| TAILQ_FOREACH(tmp, &jobhead, entry) { |
| if (tmp->pid == pid) { |
| je = tmp; |
| break; |
| } |
| } |
| if (je) { |
| TAILQ_REMOVE(&jobhead, je, entry); |
| free(je->cmd); |
| free(je); |
| } |
| if (WIFEXITED(status) == 1) |
| loginfo("complete: pid: %d returned: %d time: %s", |
| pid, WEXITSTATUS(status), ctime(&t)); |
| else if (WIFSIGNALED(status) == 1) |
| loginfo("complete: pid: %d terminated by signal: %s time: %s", |
| pid, strsignal(WTERMSIG(status)), ctime(&t)); |
| else if (WIFSTOPPED(status) == 1) |
| loginfo("complete: pid: %d stopped by signal: %s time: %s", |
| pid, strsignal(WSTOPSIG(status)), ctime(&t)); |
| } |
| } |
| |
| static int |
| isleap(int year) |
| { |
| if (year % 400 == 0) |
| return 1; |
| if (year % 100 == 0) |
| return 0; |
| return (year % 4 == 0); |
| } |
| |
| static int |
| daysinmon(int mon, int year) |
| { |
| int days[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 }; |
| if (year < 1900) |
| year += 1900; |
| if (isleap(year)) |
| days[1] = 29; |
| return days[mon]; |
| } |
| |
| static int |
| matchentry(struct ctabentry *cte, struct tm *tm) |
| { |
| struct { |
| struct field *f; |
| int tm; |
| int len; |
| } matchtbl[] = { |
| { .f = &cte->min, .tm = tm->tm_min, .len = 60 }, |
| { .f = &cte->hour, .tm = tm->tm_hour, .len = 24 }, |
| { .f = &cte->mday, .tm = tm->tm_mday, .len = daysinmon(tm->tm_mon, tm->tm_year) }, |
| { .f = &cte->mon, .tm = tm->tm_mon, .len = 12 }, |
| { .f = &cte->wday, .tm = tm->tm_wday, .len = 7 }, |
| }; |
| size_t i; |
| int j; |
| |
| for (i = 0; i < LEN(matchtbl); i++) { |
| switch (matchtbl[i].f->type) { |
| case WILDCARD: |
| continue; |
| case NUMBER: |
| if (matchtbl[i].f->val[0] == matchtbl[i].tm) |
| continue; |
| break; |
| case RANGE: |
| if (matchtbl[i].f->val[0] <= matchtbl[i].tm) |
| if (matchtbl[i].f->val[1] >= matchtbl[i].tm) |
| continue; |
| break; |
| case REPEAT: |
| if (matchtbl[i].tm > 0) { |
| if (matchtbl[i].tm % matchtbl[i].f->val[0] == 0) |
| continue; |
| } else { |
| if (matchtbl[i].len % matchtbl[i].f->val[0] == 0) |
| continue; |
| } |
| break; |
| case LIST: |
| for (j = 0; j < matchtbl[i].f->len; j++) |
| if (matchtbl[i].f->val[j] == matchtbl[i].tm) |
| break; |
| if (j < matchtbl[i].f->len) |
| continue; |
| break; |
| default: |
| break; |
| } |
| break; |
| } |
| if (i != LEN(matchtbl)) |
| return 0; |
| return 1; |
| } |
| |
| static int |
| parsefield(const char *field, long low, long high, struct field *f) |
| { |
| int i; |
| char *e1, *e2; |
| const char *p; |
| |
| p = field; |
| while (isdigit(*p)) |
| p++; |
| |
| f->type = ERROR; |
| |
| switch (*p) { |
| case '*': |
| if (strcmp(field, "*") == 0) { |
| f->val = NULL; |
| f->len = 0; |
| f->type = WILDCARD; |
| } else if (strncmp(field, "*/", 2) == 0) { |
| f->val = emalloc(sizeof(*f->val)); |
| f->len = 1; |
| |
| errno = 0; |
| f->val[0] = strtol(field + 2, &e1, 10); |
| if (e1[0] != '\0' || errno != 0 || f->val[0] == 0) |
| break; |
| |
| f->type = REPEAT; |
| } |
| break; |
| case '\0': |
| f->val = emalloc(sizeof(*f->val)); |
| f->len = 1; |
| |
| errno = 0; |
| f->val[0] = strtol(field, &e1, 10); |
| if (e1[0] != '\0' || errno != 0) |
| break; |
| |
| f->type = NUMBER; |
| break; |
| case '-': |
| f->val = emalloc(2 * sizeof(*f->val)); |
| f->len = 2; |
| |
| errno = 0; |
| f->val[0] = strtol(field, &e1, 10); |
| if (e1[0] != '-' || errno != 0) |
| break; |
| |
| errno = 0; |
| f->val[1] = strtol(e1 + 1, &e2, 10); |
| if (e2[0] != '\0' || errno != 0) |
| break; |
| |
| f->type = RANGE; |
| break; |
| case ',': |
| for (i = 1; isdigit(*p) || *p == ','; p++) |
| if (*p == ',') |
| i++; |
| f->val = emalloc(i * sizeof(*f->val)); |
| f->len = i; |
| |
| errno = 0; |
| f->val[0] = strtol(field, &e1, 10); |
| if (f->val[0] < low || f->val[0] > high) |
| break; |
| |
| for (i = 1; *e1 == ',' && errno == 0; i++) { |
| errno = 0; |
| f->val[i] = strtol(e1 + 1, &e2, 10); |
| e1 = e2; |
| } |
| if (e1[0] != '\0' || errno != 0) |
| break; |
| |
| f->type = LIST; |
| break; |
| default: |
| return -1; |
| } |
| |
| for (i = 0; i < f->len; i++) |
| if (f->val[i] < low || f->val[i] > high) |
| f->type = ERROR; |
| |
| if (f->type == ERROR) { |
| free(f->val); |
| return -1; |
| } |
| |
| return 0; |
| } |
| |
| static void |
| freecte(struct ctabentry *cte, int nfields) |
| { |
| switch (nfields) { |
| case 6: |
| free(cte->cmd); |
| case 5: |
| free(cte->wday.val); |
| case 4: |
| free(cte->mon.val); |
| case 3: |
| free(cte->mday.val); |
| case 2: |
| free(cte->hour.val); |
| case 1: |
| free(cte->min.val); |
| } |
| free(cte); |
| } |
| |
| static void |
| unloadentries(void) |
| { |
| struct ctabentry *cte, *tmp; |
| |
| for (cte = TAILQ_FIRST(&ctabhead); cte; cte = tmp) { |
| tmp = TAILQ_NEXT(cte, entry); |
| TAILQ_REMOVE(&ctabhead, cte, entry); |
| freecte(cte, 6); |
| } |
| } |
| |
| static int |
| loadentries(void) |
| { |
| struct ctabentry *cte; |
| FILE *fp; |
| char *line = NULL, *p, *col; |
| int r = 0, y; |
| size_t size = 0; |
| ssize_t len; |
| struct fieldlimits { |
| char *name; |
| long min; |
| long max; |
| struct field *f; |
| } flim[] = { |
| { "min", 0, 59, NULL }, |
| { "hour", 0, 23, NULL }, |
| { "mday", 1, 31, NULL }, |
| { "mon", 1, 12, NULL }, |
| { "wday", 0, 6, NULL } |
| }; |
| size_t x; |
| |
| if ((fp = fopen(config, "r")) == NULL) { |
| logerr("error: can't open %s: %s\n", config, strerror(errno)); |
| return -1; |
| } |
| |
| for (y = 0; (len = getline(&line, &size, fp)) != -1; y++) { |
| p = line; |
| if (line[0] == '#' || line[0] == '\n' || line[0] == '\0') |
| continue; |
| |
| cte = emalloc(sizeof(*cte)); |
| flim[0].f = &cte->min; |
| flim[1].f = &cte->hour; |
| flim[2].f = &cte->mday; |
| flim[3].f = &cte->mon; |
| flim[4].f = &cte->wday; |
| |
| for (x = 0; x < LEN(flim); x++) { |
| do |
| col = strsep(&p, "\t\n "); |
| while (col && col[0] == '\0'); |
| |
| if (!col || parsefield(col, flim[x].min, flim[x].max, flim[x].f) < 0) { |
| logerr("error: failed to parse `%s' field on line %d\n", |
| flim[x].name, y + 1); |
| freecte(cte, x); |
| r = -1; |
| break; |
| } |
| } |
| |
| if (r == -1) |
| break; |
| |
| col = strsep(&p, "\n"); |
| if (col) |
| while (col[0] == '\t' || col[0] == ' ') |
| col++; |
| if (!col || col[0] == '\0') { |
| logerr("error: missing `cmd' field on line %d\n", |
| y + 1); |
| freecte(cte, 5); |
| r = -1; |
| break; |
| } |
| cte->cmd = estrdup(col); |
| |
| TAILQ_INSERT_TAIL(&ctabhead, cte, entry); |
| } |
| |
| if (r < 0) |
| unloadentries(); |
| |
| free(line); |
| fclose(fp); |
| |
| return r; |
| } |
| |
| static void |
| reloadentries(void) |
| { |
| unloadentries(); |
| if (loadentries() < 0) |
| logwarn("warning: discarding old crontab entries\n"); |
| } |
| |
| static void |
| sighandler(int sig) |
| { |
| switch (sig) { |
| case SIGCHLD: |
| chldreap = 1; |
| break; |
| case SIGHUP: |
| reload = 1; |
| break; |
| case SIGTERM: |
| quit = 1; |
| break; |
| } |
| } |
| |
| static void |
| usage(void) |
| { |
| eprintf("usage: %s [-f file] [-n]\n", argv0); |
| } |
| |
| int |
| main(int argc, char *argv[]) |
| { |
| FILE *fp; |
| struct ctabentry *cte; |
| time_t t; |
| struct tm *tm; |
| struct sigaction sa; |
| |
| ARGBEGIN { |
| case 'n': |
| nflag = 1; |
| break; |
| case 'f': |
| config = EARGF(usage()); |
| break; |
| default: |
| usage(); |
| } ARGEND |
| |
| if (argc > 0) |
| usage(); |
| |
| if (nflag == 0) { |
| openlog(argv[0], LOG_CONS | LOG_PID, LOG_CRON); |
| if (daemon(1, 0) < 0) { |
| logerr("error: failed to daemonize %s\n", strerror(errno)); |
| return 1; |
| } |
| if ((fp = fopen(pidfile, "w"))) { |
| fprintf(fp, "%d\n", getpid()); |
| fclose(fp); |
| } |
| } |
| |
| sa.sa_handler = sighandler; |
| sigfillset(&sa.sa_mask); |
| sa.sa_flags = SA_RESTART; |
| sigaction(SIGCHLD, &sa, NULL); |
| sigaction(SIGHUP, &sa, NULL); |
| sigaction(SIGTERM, &sa, NULL); |
| |
| loadentries(); |
| |
| while (1) { |
| t = time(NULL); |
| sleep(60 - t % 60); |
| |
| if (quit == 1) { |
| if (nflag == 0) |
| unlink(pidfile); |
| unloadentries(); |
| /* Don't wait or kill forked processes, just exit */ |
| break; |
| } |
| |
| if (reload == 1 || chldreap == 1) { |
| if (reload == 1) { |
| reloadentries(); |
| reload = 0; |
| } |
| if (chldreap == 1) { |
| waitjob(); |
| chldreap = 0; |
| } |
| continue; |
| } |
| |
| TAILQ_FOREACH(cte, &ctabhead, entry) { |
| t = time(NULL); |
| tm = localtime(&t); |
| if (matchentry(cte, tm) == 1) |
| runjob(cte->cmd); |
| } |
| } |
| |
| if (nflag == 0) |
| closelog(); |
| |
| return 0; |
| } |