NetBSD/external/bsd/cron/dist/cron.c
2010-10-02 12:22:20 +00:00

560 lines
14 KiB
C

/* $NetBSD: cron.c,v 1.7 2010/10/02 12:22:20 tron Exp $ */
/* Copyright 1988,1990,1993,1994 by Paul Vixie
* All rights reserved
*/
/*
* Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
* Copyright (c) 1997,2000 by Internet Software Consortium, Inc.
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
* OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <sys/cdefs.h>
#if !defined(lint) && !defined(LINT)
#if 0
static char rcsid[] = "Id: cron.c,v 1.12 2004/01/23 18:56:42 vixie Exp";
#else
__RCSID("$NetBSD: cron.c,v 1.7 2010/10/02 12:22:20 tron Exp $");
#endif
#endif
#define MAIN_PROGRAM
#include "cron.h"
enum timejump { negative, small, medium, large };
static void usage(void),
run_reboot_jobs(cron_db *),
find_jobs(time_t, cron_db *, int, int),
set_time(int),
cron_sleep(int),
sigchld_handler(int),
sighup_handler(int),
sigchld_reaper(void),
quit(int),
parse_args(int c, char *v[]);
static volatile sig_atomic_t got_sighup, got_sigchld;
static time_t timeRunning, virtualTime, clockTime;
static long GMToff;
static void
usage(void) {
const char **dflags;
(void)fprintf(stderr, "usage: %s [-n] [-x [", getprogname());
for (dflags = DebugFlagNames; *dflags; dflags++)
(void)fprintf(stderr, "%s%s", *dflags, dflags[1] ? "," : "]");
(void)fprintf(stderr, "]\n");
exit(ERROR_EXIT);
}
int
main(int argc, char *argv[]) {
struct sigaction sact;
cron_db database;
setprogname(argv[0]);
(void)setlocale(LC_ALL, "");
(void)setvbuf(stdout, NULL, _IOLBF, 0);
(void)setvbuf(stderr, NULL, _IOLBF, 0);
NoFork = 0;
parse_args(argc, argv);
(void)memset(&sact, 0, sizeof sact);
(void)sigemptyset(&sact.sa_mask);
sact.sa_flags = 0;
#ifdef SA_RESTART
sact.sa_flags |= SA_RESTART;
#endif
sact.sa_handler = sigchld_handler;
(void) sigaction(SIGCHLD, &sact, NULL);
sact.sa_handler = sighup_handler;
(void) sigaction(SIGHUP, &sact, NULL);
sact.sa_handler = quit;
(void) sigaction(SIGINT, &sact, NULL);
(void) sigaction(SIGTERM, &sact, NULL);
acquire_daemonlock(0);
set_cron_uid();
set_cron_cwd();
if (setenv("PATH", _PATH_DEFPATH, 1) < 0) {
log_it("CRON", getpid(), "DEATH", "can't malloc");
exit(1);
}
/* if there are no debug flags turned on, fork as a daemon should.
*/
if (DebugFlags) {
#if DEBUGGING
(void)fprintf(stderr, "[%ld] cron started\n", (long)getpid());
#endif
} else if (NoFork == 0) {
if (daemon(1, 0)) {
log_it("CRON",getpid(),"DEATH","can't fork");
exit(1);
}
}
acquire_daemonlock(0);
database.head = NULL;
database.tail = NULL;
database.mtime = (time_t) 0;
load_database(&database);
set_time(TRUE);
run_reboot_jobs(&database);
timeRunning = virtualTime = clockTime;
/*
* Too many clocks, not enough time (Al. Einstein)
* These clocks are in minutes since the epoch, adjusted for timezone.
* virtualTime: is the time it *would* be if we woke up
* promptly and nobody ever changed the clock. It is
* monotonically increasing... unless a timejump happens.
* At the top of the loop, all jobs for 'virtualTime' have run.
* timeRunning: is the time we last awakened.
* clockTime: is the time when set_time was last called.
*/
for (;;) {
int timeDiff;
enum timejump wakeupKind;
/* ... wait for the time (in minutes) to change ... */
do {
cron_sleep(timeRunning + 1);
set_time(FALSE);
} while (clockTime == timeRunning);
timeRunning = clockTime;
/*
* Calculate how the current time differs from our virtual
* clock. Classify the change into one of 4 cases.
*/
timeDiff = timeRunning - virtualTime;
/* shortcut for the most common case */
if (timeDiff == 1) {
virtualTime = timeRunning;
find_jobs(virtualTime, &database, TRUE, TRUE);
} else {
if (timeDiff > (3*MINUTE_COUNT) ||
timeDiff < -(3*MINUTE_COUNT))
wakeupKind = large;
else if (timeDiff > 5)
wakeupKind = medium;
else if (timeDiff > 0)
wakeupKind = small;
else
wakeupKind = negative;
switch (wakeupKind) {
case small:
/*
* case 1: timeDiff is a small positive number
* (wokeup late) run jobs for each virtual
* minute until caught up.
*/
Debug(DSCH, ("[%ld], normal case %d minutes to go\n",
(long)getpid(), timeDiff));
do {
if (job_runqueue())
(void)sleep(10);
virtualTime++;
find_jobs(virtualTime, &database,
TRUE, TRUE);
} while (virtualTime < timeRunning);
break;
case medium:
/*
* case 2: timeDiff is a medium-sized positive
* number, for example because we went to DST
* run wildcard jobs once, then run any
* fixed-time jobs that would otherwise be
* skipped if we use up our minute (possible,
* if there are a lot of jobs to run) go
* around the loop again so that wildcard jobs
* have a chance to run, and we do our
* housekeeping.
*/
Debug(DSCH, ("[%ld], DST begins %d minutes to go\n",
(long)getpid(), timeDiff));
/* run wildcard jobs for current minute */
find_jobs(timeRunning, &database, TRUE, FALSE);
/* run fixed-time jobs for each minute missed */
do {
if (job_runqueue())
(void)sleep(10);
virtualTime++;
find_jobs(virtualTime, &database,
FALSE, TRUE);
set_time(FALSE);
} while (virtualTime < timeRunning &&
clockTime == timeRunning);
break;
case negative:
/*
* case 3: timeDiff is a small or medium-sized
* negative num, eg. because of DST ending.
* Just run the wildcard jobs. The fixed-time
* jobs probably have already run, and should
* not be repeated. Virtual time does not
* change until we are caught up.
*/
Debug(DSCH, ("[%ld], DST ends %d minutes to go\n",
(long)getpid(), timeDiff));
find_jobs(timeRunning, &database, TRUE, FALSE);
break;
default:
/*
* other: time has changed a *lot*,
* jump virtual time, and run everything
*/
Debug(DSCH, ("[%ld], clock jumped\n",
(long)getpid()));
virtualTime = timeRunning;
find_jobs(timeRunning, &database, TRUE, TRUE);
}
}
/* Jobs to be run (if any) are loaded; clear the queue. */
(void)job_runqueue();
/* Check to see if we received a signal while running jobs. */
if (got_sighup) {
got_sighup = 0;
log_close();
}
if (got_sigchld) {
got_sigchld = 0;
sigchld_reaper();
}
load_database(&database);
}
}
static void
run_reboot_jobs(cron_db *db) {
user *u;
entry *e;
for (u = db->head; u != NULL; u = u->next) {
for (e = u->crontab; e != NULL; e = e->next) {
if (e->flags & WHEN_REBOOT)
job_add(e, u, StartTime);
}
}
(void) job_runqueue();
}
static const char *
bitprint(const bitstr_t *b, size_t l)
{
size_t i, set, clear;
static char result[1024];
char tmp[1024];
result[0] = '\0';
for (i = 0;;) {
for (; i < l; i++)
if (bit_test(b, i))
break;
if (i == l)
return result;
set = i;
for (; i < l; i++)
if (!bit_test(b, i))
break;
clear = i;
if (set == 0 && clear == l) {
snprintf(result, sizeof(result), "*");
return result;
}
if (clear == l || set == clear - 1)
snprintf(tmp, sizeof(tmp), ",%zu", set);
else
snprintf(tmp, sizeof(tmp), ",%zu-%zu", set, clear - 1);
if (result[0] == '\0')
strlcpy(result, tmp + 1, sizeof(result));
else
strlcat(result, tmp, sizeof(result));
}
}
static const char *
flagsprint(int flags)
{
static char result[1024];
result[0] = '\0';
if (flags & MIN_STAR)
strlcat(result, ",min", sizeof(result));
if (flags & HR_STAR)
strlcat(result, ",hr", sizeof(result));
if (flags & DOM_STAR)
strlcat(result, ",dom", sizeof(result));
if (flags & DOW_STAR)
strlcat(result, ",dow", sizeof(result));
if (flags & WHEN_REBOOT)
strlcat(result, ",reboot", sizeof(result));
if (flags & DONT_LOG)
strlcat(result, ",nolog", sizeof(result));
return result + (result[0] == ',');
}
static void
printone(char *res, size_t len, const char *b)
{
int comma = strchr(b, ',') != NULL;
if (comma)
strlcat(res, "[", len);
strlcat(res, b, len);
strlcat(res, "]," + (comma ? 0 : 1), len);
}
static const char *
tick(const entry *e)
{
static char result[1024];
result[0] = '\0';
printone(result, sizeof(result), bitprint(e->minute, MINUTE_COUNT));
printone(result, sizeof(result), bitprint(e->hour, HOUR_COUNT));
printone(result, sizeof(result), bitprint(e->dom, DOM_COUNT));
printone(result, sizeof(result), bitprint(e->month, MONTH_COUNT));
printone(result, sizeof(result), bitprint(e->dow, DOW_COUNT));
strlcat(result, "flags=", sizeof(result));
strlcat(result, flagsprint(e->flags), sizeof(result));
return result;
}
static void
find_jobs(time_t vtime, cron_db *db, int doWild, int doNonWild) {
time_t virtualSecond = vtime * SECONDS_PER_MINUTE;
struct tm *tm;
int minute, hour, dom, month, dow;
user *u;
entry *e;
#ifndef CRON_LOCALTIME
char *orig_tz, *job_tz;
#define maketime(tz1, tz2) do { \
char *t = tz1; \
if (t != NULL && *t != '\0') \
setenv("TZ", t, 1); \
else if ((tz2) != NULL) \
setenv("TZ", (tz2), 1); \
else \
unsetenv("TZ"); \
tzset(); \
tm = localtime(&virtualSecond); \
minute = tm->tm_min -FIRST_MINUTE; \
hour = tm->tm_hour -FIRST_HOUR; \
dom = tm->tm_mday -FIRST_DOM; \
month = tm->tm_mon + 1 /* 0..11 -> 1..12 */ -FIRST_MONTH; \
dow = tm->tm_wday -FIRST_DOW; \
} while (/*CONSTCOND*/0)
orig_tz = getenv("TZ");
maketime(NULL, orig_tz);
#else
tm = gmtime(&virtualSecond);
#endif
Debug(DSCH, ("[%ld] tick(%d,%d,%d,%d,%d) %s %s\n",
(long)getpid(), minute, hour, dom, month, dow,
doWild?" ":"No wildcard",doNonWild?" ":"Wildcard only"));
/* the dom/dow situation is odd. '* * 1,15 * Sun' will run on the
* first and fifteenth AND every Sunday; '* * * * Sun' will run *only*
* on Sundays; '* * 1,15 * *' will run *only* the 1st and 15th. this
* is why we keep 'e->dow_star' and 'e->dom_star'. yes, it's bizarre.
* like many bizarre things, it's the standard.
*/
for (u = db->head; u != NULL; u = u->next) {
for (e = u->crontab; e != NULL; e = e->next) {
#ifndef CRON_LOCALTIME
job_tz = env_get("CRON_TZ", e->envp);
maketime(job_tz, orig_tz);
#else
#define job_tz "N/A"
#define orig_tz "N/A"
#endif
Debug(DSCH|DEXT, ("user [%s:%ld:%ld:...] "
"[jobtz=%s, origtz=%s] "
"tick(%s), cmd=\"%s\"\n",
e->pwd->pw_name, (long)e->pwd->pw_uid,
(long)e->pwd->pw_gid, job_tz, orig_tz,
tick(e), e->cmd));
if (bit_test(e->minute, minute) &&
bit_test(e->hour, hour) &&
bit_test(e->month, month) &&
( ((e->flags & DOM_STAR) || (e->flags & DOW_STAR))
? (bit_test(e->dow,dow) && bit_test(e->dom,dom))
: (bit_test(e->dow,dow) || bit_test(e->dom,dom))
)
) {
if ((doNonWild &&
!(e->flags & (MIN_STAR|HR_STAR))) ||
(doWild && (e->flags & (MIN_STAR|HR_STAR))))
job_add(e, u, StartTime);
}
}
}
#ifndef CRON_LOCALTIME
if (orig_tz != NULL)
setenv("TZ", orig_tz, 1);
else
unsetenv("TZ");
#endif
}
/*
* Set StartTime and clockTime to the current time.
* These are used for computing what time it really is right now.
* Note that clockTime is a unix wallclock time converted to minutes.
*/
static void
set_time(int initialize) {
#ifdef CRON_LOCALTIME
struct tm tm;
static int isdst;
StartTime = time(NULL);
/* We adjust the time to GMT so we can catch DST changes. */
tm = *localtime(&StartTime);
if (initialize || tm.tm_isdst != isdst) {
isdst = tm.tm_isdst;
GMToff = get_gmtoff(&StartTime, &tm);
Debug(DSCH, ("[%ld] GMToff=%ld\n",
(long)getpid(), (long)GMToff));
}
#else
StartTime = time(NULL);
#endif
clockTime = (StartTime + GMToff) / (time_t)SECONDS_PER_MINUTE;
}
/*
* Try to just hit the next minute.
*/
static void
cron_sleep(int target) {
time_t t1, t2;
int seconds_to_wait;
t1 = time(NULL) + GMToff;
seconds_to_wait = (int)(target * SECONDS_PER_MINUTE - t1);
Debug(DSCH, ("[%ld] Target time=%lld, sec-to-wait=%d\n",
(long)getpid(), (long long)target*SECONDS_PER_MINUTE,
seconds_to_wait));
while (seconds_to_wait > 0 && seconds_to_wait < 65) {
(void)sleep((unsigned int) seconds_to_wait);
/*
* Check to see if we were interrupted by a signal.
* If so, service the signal(s) then continue sleeping
* where we left off.
*/
if (got_sighup) {
got_sighup = 0;
log_close();
}
if (got_sigchld) {
got_sigchld = 0;
sigchld_reaper();
}
t2 = time(NULL) + GMToff;
seconds_to_wait -= (int)(t2 - t1);
t1 = t2;
}
}
/*ARGSUSED*/
static void
sighup_handler(int x __unused) {
got_sighup = 1;
}
/*ARGSUSED*/
static void
sigchld_handler(int x __unused) {
got_sigchld = 1;
}
/*ARGSUSED*/
static void
quit(int x __unused) {
(void) unlink(_PATH_CRON_PID);
_exit(0);
}
static void
sigchld_reaper(void) {
WAIT_T waiter;
PID_T pid;
do {
pid = waitpid(-1, &waiter, WNOHANG);
switch (pid) {
case -1:
if (errno == EINTR)
continue;
Debug(DPROC,
("[%ld] sigchld...no children\n",
(long)getpid()));
break;
case 0:
Debug(DPROC,
("[%ld] sigchld...no dead kids\n",
(long)getpid()));
break;
default:
Debug(DPROC,
("[%ld] sigchld...pid #%ld died, stat=%d\n",
(long)getpid(), (long)pid, WEXITSTATUS(waiter)));
break;
}
} while (pid > 0);
}
static void
parse_args(int argc, char *argv[]) {
int argch;
while (-1 != (argch = getopt(argc, argv, "nx:"))) {
switch (argch) {
default:
usage();
break;
case 'x':
if (!set_debug_flags(optarg))
usage();
break;
case 'n':
NoFork = 1;
break;
}
}
}