2010-07-20 17:42:13 +00:00
|
|
|
/*-
|
2004-05-03 21:41:02 +00:00
|
|
|
* Copyright (c) 2003-2004 Sean M. Kelly <smkelly@FreeBSD.org>
|
2013-02-27 19:03:31 +00:00
|
|
|
* Copyright (c) 2013 iXsystems.com,
|
|
|
|
* author: Alfred Perlstein <alfred@freebsd.org>
|
|
|
|
*
|
2003-06-26 09:50:52 +00:00
|
|
|
* All rights reserved.
|
|
|
|
*
|
|
|
|
* Redistribution and use in source and binary forms, with or without
|
|
|
|
* modification, are permitted provided that the following conditions
|
|
|
|
* are met:
|
|
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
|
|
* notice, this list of conditions and the following disclaimer.
|
|
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
|
|
* documentation and/or other materials provided with the distribution.
|
|
|
|
*
|
|
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
|
|
* SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Software watchdog daemon.
|
|
|
|
*/
|
|
|
|
|
2003-07-03 03:37:04 +00:00
|
|
|
#include <sys/types.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
__FBSDID("$FreeBSD$");
|
|
|
|
|
2010-09-26 01:45:33 +00:00
|
|
|
#include <sys/mman.h>
|
2005-08-24 19:28:33 +00:00
|
|
|
#include <sys/param.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
#include <sys/rtprio.h>
|
|
|
|
#include <sys/stat.h>
|
2003-07-03 03:37:04 +00:00
|
|
|
#include <sys/time.h>
|
2013-07-27 20:47:01 +00:00
|
|
|
#include <sys/sysctl.h>
|
2004-02-28 20:56:35 +00:00
|
|
|
#include <sys/watchdog.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
|
|
|
|
#include <err.h>
|
2004-02-28 20:56:35 +00:00
|
|
|
#include <errno.h>
|
|
|
|
#include <fcntl.h>
|
2005-08-24 19:28:33 +00:00
|
|
|
#include <libutil.h>
|
2004-02-28 20:56:35 +00:00
|
|
|
#include <math.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
#include <paths.h>
|
2003-07-03 03:37:04 +00:00
|
|
|
#include <signal.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
#include <stdio.h>
|
2013-07-28 16:56:31 +00:00
|
|
|
#include <stdint.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
#include <stdlib.h>
|
2004-02-28 20:56:35 +00:00
|
|
|
#include <string.h>
|
2012-11-03 18:38:28 +00:00
|
|
|
#include <strings.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
#include <sysexits.h>
|
2013-02-27 19:03:31 +00:00
|
|
|
#include <syslog.h>
|
2003-06-26 09:50:52 +00:00
|
|
|
#include <unistd.h>
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
#include <getopt.h>
|
|
|
|
|
2013-08-10 01:48:15 +00:00
|
|
|
static long fetchtimeout(int opt,
|
|
|
|
const char *longopt, const char *myoptarg, int zero_ok);
|
2003-06-26 09:50:52 +00:00
|
|
|
static void parseargs(int, char *[]);
|
2013-07-27 20:47:01 +00:00
|
|
|
static int seconds_to_pow2ns(int);
|
2003-06-26 09:50:52 +00:00
|
|
|
static void sighandler(int);
|
|
|
|
static void watchdog_loop(void);
|
|
|
|
static int watchdog_init(void);
|
|
|
|
static int watchdog_onoff(int onoff);
|
2006-12-15 22:47:36 +00:00
|
|
|
static int watchdog_patpat(u_int timeout);
|
2003-06-26 09:50:52 +00:00
|
|
|
static void usage(void);
|
2013-07-27 20:47:01 +00:00
|
|
|
static int tstotv(struct timeval *tv, struct timespec *ts);
|
|
|
|
static int tvtohz(struct timeval *tv);
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2010-07-20 17:42:13 +00:00
|
|
|
static int debugging = 0;
|
|
|
|
static int end_program = 0;
|
|
|
|
static const char *pidfile = _PATH_VARRUN "watchdogd.pid";
|
2013-02-27 19:03:31 +00:00
|
|
|
static u_int timeout = WD_TO_128SEC;
|
|
|
|
static u_int pretimeout = 0;
|
2013-07-27 20:47:01 +00:00
|
|
|
static u_int timeout_sec;
|
2010-07-20 17:42:13 +00:00
|
|
|
static u_int passive = 0;
|
|
|
|
static int is_daemon = 0;
|
2013-02-27 19:03:31 +00:00
|
|
|
static int is_dry_run = 0; /* do not arm the watchdog, only
|
|
|
|
report on timing of the watch
|
|
|
|
program */
|
|
|
|
static int do_timedog = 0;
|
2013-03-26 19:43:18 +00:00
|
|
|
static int do_syslog = 1;
|
2010-07-20 17:42:13 +00:00
|
|
|
static int fd = -1;
|
|
|
|
static int nap = 1;
|
2013-02-27 19:03:31 +00:00
|
|
|
static int carp_thresh_seconds = -1;
|
2010-07-20 17:42:13 +00:00
|
|
|
static char *test_cmd = NULL;
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
static const char *getopt_shortopts;
|
|
|
|
|
|
|
|
static int pretimeout_set;
|
|
|
|
static int pretimeout_act;
|
|
|
|
static int pretimeout_act_set;
|
|
|
|
|
|
|
|
static int softtimeout_set;
|
|
|
|
static int softtimeout_act;
|
|
|
|
static int softtimeout_act_set;
|
|
|
|
|
|
|
|
static struct option longopts[] = {
|
|
|
|
{ "debug", no_argument, &debugging, 1 },
|
|
|
|
{ "pretimeout", required_argument, &pretimeout_set, 1 },
|
|
|
|
{ "pretimeout-action", required_argument, &pretimeout_act_set, 1 },
|
|
|
|
{ "softtimeout", no_argument, &softtimeout_set, 1 },
|
|
|
|
{ "softtimeout-action", required_argument, &softtimeout_act_set, 1 },
|
|
|
|
{ NULL, 0, NULL, 0}
|
|
|
|
};
|
|
|
|
|
2013-01-26 21:29:45 +00:00
|
|
|
/*
|
|
|
|
* Ask malloc() to map minimum-sized chunks of virtual address space at a time,
|
|
|
|
* so that mlockall() won't needlessly wire megabytes of unused memory into the
|
|
|
|
* process. This must be done using the malloc_conf string so that it gets set
|
|
|
|
* up before the first allocation, which happens before entry to main().
|
|
|
|
*/
|
|
|
|
const char * malloc_conf = "lg_chunk:0";
|
|
|
|
|
2003-06-26 09:50:52 +00:00
|
|
|
/*
|
2004-04-28 07:35:03 +00:00
|
|
|
* Periodically pat the watchdog, preventing it from firing.
|
2003-06-26 09:50:52 +00:00
|
|
|
*/
|
|
|
|
int
|
|
|
|
main(int argc, char *argv[])
|
|
|
|
{
|
|
|
|
struct rtprio rtp;
|
2005-08-24 19:28:33 +00:00
|
|
|
struct pidfh *pfh;
|
|
|
|
pid_t otherpid;
|
2003-06-26 09:50:52 +00:00
|
|
|
|
|
|
|
if (getuid() != 0)
|
|
|
|
errx(EX_SOFTWARE, "not super user");
|
|
|
|
|
|
|
|
parseargs(argc, argv);
|
|
|
|
|
2013-03-26 19:43:18 +00:00
|
|
|
if (do_syslog)
|
2013-02-27 19:03:31 +00:00
|
|
|
openlog("watchdogd", LOG_CONS|LOG_NDELAY|LOG_PERROR,
|
|
|
|
LOG_DAEMON);
|
|
|
|
|
2003-06-26 09:50:52 +00:00
|
|
|
rtp.type = RTP_PRIO_REALTIME;
|
|
|
|
rtp.prio = 0;
|
|
|
|
if (rtprio(RTP_SET, 0, &rtp) == -1)
|
|
|
|
err(EX_OSERR, "rtprio");
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
if (!is_dry_run && watchdog_init() == -1)
|
2003-07-03 03:37:04 +00:00
|
|
|
errx(EX_SOFTWARE, "unable to initialize watchdog");
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2004-02-28 20:56:35 +00:00
|
|
|
if (is_daemon) {
|
|
|
|
if (watchdog_onoff(1) == -1)
|
2009-12-21 15:50:37 +00:00
|
|
|
err(EX_OSERR, "patting the dog");
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2005-09-16 11:24:28 +00:00
|
|
|
pfh = pidfile_open(pidfile, 0600, &otherpid);
|
2005-08-24 19:28:33 +00:00
|
|
|
if (pfh == NULL) {
|
|
|
|
if (errno == EEXIST) {
|
2013-02-27 19:03:31 +00:00
|
|
|
watchdog_onoff(0);
|
2005-08-24 19:28:33 +00:00
|
|
|
errx(EX_SOFTWARE, "%s already running, pid: %d",
|
|
|
|
getprogname(), otherpid);
|
|
|
|
}
|
|
|
|
warn("Cannot open or create pidfile");
|
|
|
|
}
|
|
|
|
|
2004-02-28 20:56:35 +00:00
|
|
|
if (debugging == 0 && daemon(0, 0) == -1) {
|
|
|
|
watchdog_onoff(0);
|
2005-08-24 19:28:33 +00:00
|
|
|
pidfile_remove(pfh);
|
2004-02-28 20:56:35 +00:00
|
|
|
err(EX_OSERR, "daemon");
|
|
|
|
}
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2004-02-28 20:56:35 +00:00
|
|
|
signal(SIGHUP, SIG_IGN);
|
|
|
|
signal(SIGINT, sighandler);
|
|
|
|
signal(SIGTERM, sighandler);
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2005-08-24 19:28:33 +00:00
|
|
|
pidfile_write(pfh);
|
2010-09-26 01:45:33 +00:00
|
|
|
if (madvise(0, 0, MADV_PROTECT) != 0)
|
|
|
|
warn("madvise failed");
|
2012-08-30 08:07:37 +00:00
|
|
|
if (mlockall(MCL_CURRENT | MCL_FUTURE) != 0)
|
2012-08-28 08:38:53 +00:00
|
|
|
warn("mlockall failed");
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2004-02-28 20:56:35 +00:00
|
|
|
watchdog_loop();
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2004-02-28 20:56:35 +00:00
|
|
|
/* exiting */
|
2005-08-24 19:28:33 +00:00
|
|
|
pidfile_remove(pfh);
|
2004-02-28 20:56:35 +00:00
|
|
|
return (EX_OK);
|
|
|
|
} else {
|
|
|
|
if (passive)
|
|
|
|
timeout |= WD_PASSIVE;
|
|
|
|
else
|
|
|
|
timeout |= WD_ACTIVE;
|
2006-12-15 22:47:36 +00:00
|
|
|
if (watchdog_patpat(timeout) < 0)
|
2004-02-28 20:56:35 +00:00
|
|
|
err(EX_OSERR, "patting the dog");
|
|
|
|
return (EX_OK);
|
|
|
|
}
|
2003-06-26 09:50:52 +00:00
|
|
|
}
|
|
|
|
|
2013-07-27 20:47:01 +00:00
|
|
|
static void
|
|
|
|
pow2ns_to_ts(int pow2ns, struct timespec *ts)
|
|
|
|
{
|
|
|
|
uint64_t ns;
|
|
|
|
|
|
|
|
ns = 1ULL << pow2ns;
|
|
|
|
ts->tv_sec = ns / 1000000000ULL;
|
|
|
|
ts->tv_nsec = ns % 1000000000ULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Convert a timeout in seconds to N where 2^N nanoseconds is close to
|
|
|
|
* "seconds".
|
|
|
|
*
|
|
|
|
* The kernel expects the timeouts for watchdogs in "2^N nanosecond format".
|
|
|
|
*/
|
|
|
|
static u_int
|
|
|
|
parse_timeout_to_pow2ns(char opt, const char *longopt, const char *myoptarg)
|
|
|
|
{
|
|
|
|
double a;
|
|
|
|
u_int rv;
|
|
|
|
struct timespec ts;
|
|
|
|
struct timeval tv;
|
|
|
|
int ticks;
|
|
|
|
char shortopt[] = "- ";
|
|
|
|
|
|
|
|
if (!longopt)
|
|
|
|
shortopt[1] = opt;
|
|
|
|
|
2013-08-10 01:48:15 +00:00
|
|
|
a = fetchtimeout(opt, longopt, myoptarg, 1);
|
2013-07-27 20:47:01 +00:00
|
|
|
|
|
|
|
if (a == 0)
|
|
|
|
rv = WD_TO_NEVER;
|
|
|
|
else
|
|
|
|
rv = seconds_to_pow2ns(a);
|
|
|
|
pow2ns_to_ts(rv, &ts);
|
|
|
|
tstotv(&tv, &ts);
|
|
|
|
ticks = tvtohz(&tv);
|
|
|
|
if (debugging) {
|
|
|
|
printf("Timeout for %s%s "
|
|
|
|
"is 2^%d nanoseconds "
|
2013-07-28 16:56:31 +00:00
|
|
|
"(in: %s sec -> out: %jd sec %ld ns -> %d ticks)\n",
|
2013-07-27 20:47:01 +00:00
|
|
|
longopt ? "-" : "", longopt ? longopt : shortopt,
|
|
|
|
rv,
|
2013-07-28 16:56:31 +00:00
|
|
|
myoptarg, (intmax_t)ts.tv_sec, ts.tv_nsec, ticks);
|
2013-07-27 20:47:01 +00:00
|
|
|
}
|
|
|
|
if (ticks <= 0) {
|
|
|
|
errx(1, "Timeout for %s%s is too small, please choose a higher timeout.", longopt ? "-" : "", longopt ? longopt : shortopt);
|
|
|
|
}
|
|
|
|
|
|
|
|
return (rv);
|
|
|
|
}
|
|
|
|
|
2003-06-26 09:50:52 +00:00
|
|
|
/*
|
|
|
|
* Catch signals and begin shutdown process.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
sighandler(int signum)
|
|
|
|
{
|
|
|
|
|
|
|
|
if (signum == SIGINT || signum == SIGTERM)
|
|
|
|
end_program = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
2004-04-28 07:35:03 +00:00
|
|
|
* Open the watchdog device.
|
2003-06-26 09:50:52 +00:00
|
|
|
*/
|
|
|
|
static int
|
2009-12-29 22:53:27 +00:00
|
|
|
watchdog_init(void)
|
2003-06-26 09:50:52 +00:00
|
|
|
{
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
if (is_dry_run)
|
|
|
|
return 0;
|
|
|
|
|
2004-02-28 20:56:35 +00:00
|
|
|
fd = open("/dev/" _PATH_WATCHDOG, O_RDWR);
|
|
|
|
if (fd >= 0)
|
|
|
|
return (0);
|
|
|
|
warn("Could not open watchdog device");
|
|
|
|
return (-1);
|
2003-06-26 09:50:52 +00:00
|
|
|
}
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
/*
|
|
|
|
* If we are doing timing, then get the time.
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
watchdog_getuptime(struct timespec *tp)
|
|
|
|
{
|
|
|
|
int error;
|
|
|
|
|
|
|
|
if (!do_timedog)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
error = clock_gettime(CLOCK_UPTIME_FAST, tp);
|
|
|
|
if (error)
|
|
|
|
warn("clock_gettime");
|
|
|
|
return (error);
|
|
|
|
}
|
|
|
|
|
|
|
|
static long
|
|
|
|
watchdog_check_dogfunction_time(struct timespec *tp_start,
|
|
|
|
struct timespec *tp_end)
|
|
|
|
{
|
2013-03-26 19:43:18 +00:00
|
|
|
struct timeval tv_start, tv_end, tv_now, tv;
|
2013-02-27 19:03:31 +00:00
|
|
|
const char *cmd_prefix, *cmd;
|
2013-03-26 19:43:18 +00:00
|
|
|
struct timespec tp_now;
|
2013-02-27 19:03:31 +00:00
|
|
|
int sec;
|
|
|
|
|
|
|
|
if (!do_timedog)
|
|
|
|
return (0);
|
|
|
|
|
|
|
|
TIMESPEC_TO_TIMEVAL(&tv_start, tp_start);
|
|
|
|
TIMESPEC_TO_TIMEVAL(&tv_end, tp_end);
|
|
|
|
timersub(&tv_end, &tv_start, &tv);
|
|
|
|
sec = tv.tv_sec;
|
|
|
|
if (sec < carp_thresh_seconds)
|
|
|
|
return (sec);
|
|
|
|
|
|
|
|
if (test_cmd) {
|
|
|
|
cmd_prefix = "Watchdog program";
|
|
|
|
cmd = test_cmd;
|
|
|
|
} else {
|
|
|
|
cmd_prefix = "Watchdog operation";
|
|
|
|
cmd = "stat(\"/etc\", &sb)";
|
|
|
|
}
|
|
|
|
if (do_syslog)
|
|
|
|
syslog(LOG_CRIT, "%s: '%s' took too long: "
|
2013-03-26 19:43:18 +00:00
|
|
|
"%d.%06ld seconds >= %d seconds threshold",
|
2013-02-27 19:03:31 +00:00
|
|
|
cmd_prefix, cmd, sec, (long)tv.tv_usec,
|
|
|
|
carp_thresh_seconds);
|
2013-03-26 19:43:18 +00:00
|
|
|
else
|
|
|
|
warnx("%s: '%s' took too long: "
|
|
|
|
"%d.%06ld seconds >= %d seconds threshold",
|
|
|
|
cmd_prefix, cmd, sec, (long)tv.tv_usec,
|
|
|
|
carp_thresh_seconds);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Adjust the sleep interval again in case syslog(3) took a non-trivial
|
|
|
|
* amount of time to run.
|
|
|
|
*/
|
|
|
|
if (watchdog_getuptime(&tp_now))
|
|
|
|
return (sec);
|
|
|
|
TIMESPEC_TO_TIMEVAL(&tv_now, &tp_now);
|
|
|
|
timersub(&tv_now, &tv_start, &tv);
|
|
|
|
sec = tv.tv_sec;
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
return (sec);
|
|
|
|
}
|
|
|
|
|
2003-06-26 09:50:52 +00:00
|
|
|
/*
|
|
|
|
* Main program loop which is iterated every second.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
watchdog_loop(void)
|
|
|
|
{
|
2013-02-27 19:03:31 +00:00
|
|
|
struct timespec ts_start, ts_end;
|
2003-06-26 09:50:52 +00:00
|
|
|
struct stat sb;
|
2013-02-27 19:03:31 +00:00
|
|
|
long waited;
|
|
|
|
int error, failed;
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2006-12-15 22:47:36 +00:00
|
|
|
while (end_program != 2) {
|
2003-06-26 09:50:52 +00:00
|
|
|
failed = 0;
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
error = watchdog_getuptime(&ts_start);
|
|
|
|
if (error) {
|
|
|
|
end_program = 1;
|
|
|
|
goto try_end;
|
|
|
|
}
|
|
|
|
|
2004-02-28 20:56:35 +00:00
|
|
|
if (test_cmd != NULL)
|
|
|
|
failed = system(test_cmd);
|
|
|
|
else
|
|
|
|
failed = stat("/etc", &sb);
|
2003-06-26 09:50:52 +00:00
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
error = watchdog_getuptime(&ts_end);
|
|
|
|
if (error) {
|
|
|
|
end_program = 1;
|
|
|
|
goto try_end;
|
|
|
|
}
|
|
|
|
|
2003-06-26 09:50:52 +00:00
|
|
|
if (failed == 0)
|
2006-12-15 22:47:36 +00:00
|
|
|
watchdog_patpat(timeout|WD_ACTIVE);
|
2013-03-26 19:43:18 +00:00
|
|
|
|
|
|
|
waited = watchdog_check_dogfunction_time(&ts_start, &ts_end);
|
2013-02-27 19:03:31 +00:00
|
|
|
if (nap - waited > 0)
|
|
|
|
sleep(nap - waited);
|
2006-12-15 22:47:36 +00:00
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
try_end:
|
2006-12-15 22:47:36 +00:00
|
|
|
if (end_program != 0) {
|
|
|
|
if (watchdog_onoff(0) == 0) {
|
|
|
|
end_program = 2;
|
|
|
|
} else {
|
2013-01-26 22:02:40 +00:00
|
|
|
warnx("Could not stop the watchdog, not exiting");
|
2006-12-15 22:47:36 +00:00
|
|
|
end_program = 0;
|
|
|
|
}
|
|
|
|
}
|
2003-06-26 09:50:52 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Reset the watchdog timer. This function must be called periodically
|
|
|
|
* to keep the watchdog from firing.
|
|
|
|
*/
|
2010-07-20 17:42:13 +00:00
|
|
|
static int
|
2006-12-15 22:47:36 +00:00
|
|
|
watchdog_patpat(u_int t)
|
2003-06-26 09:50:52 +00:00
|
|
|
{
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
if (is_dry_run)
|
|
|
|
return 0;
|
|
|
|
|
2006-12-15 22:47:36 +00:00
|
|
|
return ioctl(fd, WDIOCPATPAT, &t);
|
2003-06-26 09:50:52 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Toggle the kernel's watchdog. This routine is used to enable and
|
|
|
|
* disable the watchdog.
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
watchdog_onoff(int onoff)
|
|
|
|
{
|
2013-02-27 19:03:31 +00:00
|
|
|
int error;
|
|
|
|
|
|
|
|
/* fake successful watchdog op if a dry run */
|
|
|
|
if (is_dry_run)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
if (onoff) {
|
|
|
|
/*
|
|
|
|
* Call the WDIOC_SETSOFT regardless of softtimeout_set
|
|
|
|
* because we'll need to turn it off if someone had turned
|
|
|
|
* it on.
|
|
|
|
*/
|
|
|
|
error = ioctl(fd, WDIOC_SETSOFT, &softtimeout_set);
|
|
|
|
if (error) {
|
|
|
|
warn("setting WDIOC_SETSOFT %d", softtimeout_set);
|
|
|
|
return (error);
|
|
|
|
}
|
|
|
|
error = watchdog_patpat((timeout|WD_ACTIVE));
|
|
|
|
if (error) {
|
|
|
|
warn("watchdog_patpat failed");
|
|
|
|
goto failsafe;
|
|
|
|
}
|
|
|
|
if (softtimeout_act_set) {
|
|
|
|
error = ioctl(fd, WDIOC_SETSOFTTIMEOUTACT,
|
|
|
|
&softtimeout_act);
|
|
|
|
if (error) {
|
|
|
|
warn("setting WDIOC_SETSOFTTIMEOUTACT %d",
|
|
|
|
softtimeout_act);
|
|
|
|
goto failsafe;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (pretimeout_set) {
|
|
|
|
error = ioctl(fd, WDIOC_SETPRETIMEOUT, &pretimeout);
|
|
|
|
if (error) {
|
|
|
|
warn("setting WDIOC_SETPRETIMEOUT %d",
|
|
|
|
pretimeout);
|
|
|
|
goto failsafe;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (pretimeout_act_set) {
|
|
|
|
error = ioctl(fd, WDIOC_SETPRETIMEOUTACT,
|
|
|
|
&pretimeout_act);
|
|
|
|
if (error) {
|
|
|
|
warn("setting WDIOC_SETPRETIMEOUTACT %d",
|
|
|
|
pretimeout_act);
|
|
|
|
goto failsafe;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/* pat one more time for good measure */
|
2006-12-15 22:47:36 +00:00
|
|
|
return watchdog_patpat((timeout|WD_ACTIVE));
|
2013-02-27 19:03:31 +00:00
|
|
|
} else {
|
2006-12-15 22:47:36 +00:00
|
|
|
return watchdog_patpat(0);
|
2013-02-27 19:03:31 +00:00
|
|
|
}
|
|
|
|
failsafe:
|
|
|
|
watchdog_patpat(0);
|
|
|
|
return (error);
|
2003-06-26 09:50:52 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Tell user how to use the program.
|
|
|
|
*/
|
|
|
|
static void
|
2009-12-29 22:53:27 +00:00
|
|
|
usage(void)
|
2003-06-26 09:50:52 +00:00
|
|
|
{
|
2004-02-28 20:56:35 +00:00
|
|
|
if (is_daemon)
|
2013-02-27 19:03:31 +00:00
|
|
|
fprintf(stderr, "usage:\n"
|
2013-03-26 19:43:18 +00:00
|
|
|
" watchdogd [-dnSw] [-e cmd] [-I file] [-s sleep] [-t timeout]\n"
|
2013-02-27 19:03:31 +00:00
|
|
|
" [-T script_timeout]\n"
|
|
|
|
" [--debug]\n"
|
|
|
|
" [--pretimeout seconds] [-pretimeout-action action]\n"
|
|
|
|
" [--softtimeout] [-softtimeout-action action]\n"
|
|
|
|
);
|
2004-02-28 20:56:35 +00:00
|
|
|
else
|
2006-03-06 07:42:52 +00:00
|
|
|
fprintf(stderr, "usage: watchdog [-d] [-t timeout]\n");
|
2003-06-26 09:50:52 +00:00
|
|
|
exit(EX_USAGE);
|
|
|
|
}
|
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
static long
|
2013-08-10 01:48:15 +00:00
|
|
|
fetchtimeout(int opt, const char *longopt, const char *myoptarg, int zero_ok)
|
2013-02-27 19:03:31 +00:00
|
|
|
{
|
|
|
|
const char *errstr;
|
|
|
|
char *p;
|
|
|
|
long rv;
|
|
|
|
|
|
|
|
errstr = NULL;
|
|
|
|
p = NULL;
|
|
|
|
errno = 0;
|
|
|
|
rv = strtol(myoptarg, &p, 0);
|
|
|
|
if ((p != NULL && *p != '\0') || errno != 0)
|
|
|
|
errstr = "is not a number";
|
2013-08-10 01:48:15 +00:00
|
|
|
if (rv < 0 || (!zero_ok && rv == 0))
|
2013-02-27 19:03:31 +00:00
|
|
|
errstr = "must be greater than zero";
|
|
|
|
if (errstr) {
|
|
|
|
if (longopt)
|
|
|
|
errx(EX_USAGE, "--%s argument %s", longopt, errstr);
|
|
|
|
else
|
|
|
|
errx(EX_USAGE, "-%c argument %s", opt, errstr);
|
|
|
|
}
|
|
|
|
return (rv);
|
|
|
|
}
|
|
|
|
|
|
|
|
struct act_tbl {
|
|
|
|
const char *at_act;
|
|
|
|
int at_value;
|
|
|
|
};
|
|
|
|
|
2013-04-08 08:05:15 +00:00
|
|
|
static const struct act_tbl act_tbl[] = {
|
2013-02-27 19:03:31 +00:00
|
|
|
{ "panic", WD_SOFT_PANIC },
|
|
|
|
{ "ddb", WD_SOFT_DDB },
|
|
|
|
{ "log", WD_SOFT_LOG },
|
|
|
|
{ "printf", WD_SOFT_PRINTF },
|
|
|
|
{ NULL, 0 }
|
|
|
|
};
|
|
|
|
|
|
|
|
static void
|
|
|
|
timeout_act_error(const char *lopt, const char *badact)
|
|
|
|
{
|
|
|
|
char *opts, *oldopts;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
opts = NULL;
|
|
|
|
for (i = 0; act_tbl[i].at_act != NULL; i++) {
|
|
|
|
oldopts = opts;
|
|
|
|
if (asprintf(&opts, "%s%s%s",
|
|
|
|
oldopts == NULL ? "" : oldopts,
|
|
|
|
oldopts == NULL ? "" : ", ",
|
|
|
|
act_tbl[i].at_act) == -1)
|
|
|
|
err(EX_OSERR, "malloc");
|
|
|
|
free(oldopts);
|
|
|
|
}
|
|
|
|
warnx("bad --%s argument '%s' must be one of (%s).",
|
|
|
|
lopt, badact, opts);
|
|
|
|
usage();
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Take a comma separated list of actions and or the flags
|
|
|
|
* together for the ioctl.
|
|
|
|
*/
|
|
|
|
static int
|
|
|
|
timeout_act_str2int(const char *lopt, const char *acts)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
char *dupacts, *tofree;
|
|
|
|
char *o;
|
|
|
|
int rv = 0;
|
|
|
|
|
|
|
|
tofree = dupacts = strdup(acts);
|
|
|
|
if (!tofree)
|
|
|
|
err(EX_OSERR, "malloc");
|
|
|
|
while ((o = strsep(&dupacts, ",")) != NULL) {
|
|
|
|
for (i = 0; act_tbl[i].at_act != NULL; i++) {
|
|
|
|
if (!strcmp(o, act_tbl[i].at_act)) {
|
|
|
|
rv |= act_tbl[i].at_value;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (act_tbl[i].at_act == NULL)
|
|
|
|
timeout_act_error(lopt, o);
|
|
|
|
}
|
|
|
|
free(tofree);
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
2013-07-27 20:47:01 +00:00
|
|
|
int
|
|
|
|
tstotv(struct timeval *tv, struct timespec *ts)
|
|
|
|
{
|
|
|
|
|
|
|
|
tv->tv_sec = ts->tv_sec;
|
|
|
|
tv->tv_usec = ts->tv_nsec / 1000;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Convert a timeval to a number of ticks.
|
|
|
|
* Mostly copied from the kernel.
|
|
|
|
*/
|
|
|
|
int
|
|
|
|
tvtohz(struct timeval *tv)
|
|
|
|
{
|
|
|
|
register unsigned long ticks;
|
|
|
|
register long sec, usec;
|
|
|
|
int hz;
|
|
|
|
size_t hzsize;
|
|
|
|
int error;
|
|
|
|
int tick;
|
|
|
|
|
|
|
|
hzsize = sizeof(hz);
|
|
|
|
|
|
|
|
error = sysctlbyname("kern.hz", &hz, &hzsize, NULL, 0);
|
|
|
|
if (error)
|
|
|
|
err(1, "sysctlbyname kern.hz");
|
|
|
|
|
|
|
|
tick = 1000000 / hz;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* If the number of usecs in the whole seconds part of the time
|
|
|
|
* difference fits in a long, then the total number of usecs will
|
|
|
|
* fit in an unsigned long. Compute the total and convert it to
|
|
|
|
* ticks, rounding up and adding 1 to allow for the current tick
|
|
|
|
* to expire. Rounding also depends on unsigned long arithmetic
|
|
|
|
* to avoid overflow.
|
|
|
|
*
|
|
|
|
* Otherwise, if the number of ticks in the whole seconds part of
|
|
|
|
* the time difference fits in a long, then convert the parts to
|
|
|
|
* ticks separately and add, using similar rounding methods and
|
|
|
|
* overflow avoidance. This method would work in the previous
|
|
|
|
* case but it is slightly slower and assumes that hz is integral.
|
|
|
|
*
|
|
|
|
* Otherwise, round the time difference down to the maximum
|
|
|
|
* representable value.
|
|
|
|
*
|
|
|
|
* If ints have 32 bits, then the maximum value for any timeout in
|
|
|
|
* 10ms ticks is 248 days.
|
|
|
|
*/
|
|
|
|
sec = tv->tv_sec;
|
|
|
|
usec = tv->tv_usec;
|
|
|
|
if (usec < 0) {
|
|
|
|
sec--;
|
|
|
|
usec += 1000000;
|
|
|
|
}
|
|
|
|
if (sec < 0) {
|
|
|
|
#ifdef DIAGNOSTIC
|
|
|
|
if (usec > 0) {
|
|
|
|
sec++;
|
|
|
|
usec -= 1000000;
|
|
|
|
}
|
|
|
|
printf("tvotohz: negative time difference %ld sec %ld usec\n",
|
|
|
|
sec, usec);
|
|
|
|
#endif
|
|
|
|
ticks = 1;
|
|
|
|
} else if (sec <= LONG_MAX / 1000000)
|
|
|
|
ticks = (sec * 1000000 + (unsigned long)usec + (tick - 1))
|
|
|
|
/ tick + 1;
|
|
|
|
else if (sec <= LONG_MAX / hz)
|
|
|
|
ticks = sec * hz
|
|
|
|
+ ((unsigned long)usec + (tick - 1)) / tick + 1;
|
|
|
|
else
|
|
|
|
ticks = LONG_MAX;
|
|
|
|
if (ticks > INT_MAX)
|
|
|
|
ticks = INT_MAX;
|
|
|
|
return ((int)ticks);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
seconds_to_pow2ns(int seconds)
|
|
|
|
{
|
|
|
|
uint64_t power;
|
|
|
|
uint64_t ns;
|
|
|
|
uint64_t shifted;
|
|
|
|
|
|
|
|
if (seconds <= 0)
|
|
|
|
errx(1, "seconds %d < 0", seconds);
|
|
|
|
ns = ((uint64_t)seconds) * 1000000000ULL;
|
|
|
|
power = flsll(ns);
|
|
|
|
shifted = 1ULL << power;
|
|
|
|
if (shifted <= ns) {
|
|
|
|
power++;
|
|
|
|
}
|
|
|
|
if (debugging) {
|
|
|
|
printf("shifted %lld\n", (long long)shifted);
|
|
|
|
printf("seconds_to_pow2ns: seconds: %d, ns %lld, power %d\n",
|
|
|
|
seconds, (long long)ns, (int)power);
|
|
|
|
}
|
|
|
|
return (power);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2003-06-26 09:50:52 +00:00
|
|
|
/*
|
|
|
|
* Handle the few command line arguments supported.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
parseargs(int argc, char *argv[])
|
|
|
|
{
|
2013-02-27 19:03:31 +00:00
|
|
|
int longindex;
|
2003-06-26 09:50:52 +00:00
|
|
|
int c;
|
2013-02-27 19:03:31 +00:00
|
|
|
const char *lopt;
|
2004-02-28 20:56:35 +00:00
|
|
|
|
2013-02-27 19:03:31 +00:00
|
|
|
/*
|
|
|
|
* if we end with a 'd' aka 'watchdogd' then we are the daemon program,
|
|
|
|
* otherwise run as a command line utility.
|
|
|
|
*/
|
2004-02-28 20:56:35 +00:00
|
|
|
c = strlen(argv[0]);
|
|
|
|
if (argv[0][c - 1] == 'd')
|
|
|
|
is_daemon = 1;
|
2013-02-27 19:03:31 +00:00
|
|
|
|
|
|
|
if (is_daemon)
|
|
|
|
getopt_shortopts = "I:de:ns:t:ST:w?";
|
|
|
|
else
|
|
|
|
getopt_shortopts = "dt:?";
|
|
|
|
|
|
|
|
while ((c = getopt_long(argc, argv, getopt_shortopts, longopts,
|
|
|
|
&longindex)) != -1) {
|
2003-06-26 09:50:52 +00:00
|
|
|
switch (c) {
|
|
|
|
case 'I':
|
|
|
|
pidfile = optarg;
|
|
|
|
break;
|
|
|
|
case 'd':
|
|
|
|
debugging = 1;
|
|
|
|
break;
|
2004-02-28 20:56:35 +00:00
|
|
|
case 'e':
|
|
|
|
test_cmd = strdup(optarg);
|
|
|
|
break;
|
2013-02-27 19:03:31 +00:00
|
|
|
case 'n':
|
|
|
|
is_dry_run = 1;
|
|
|
|
break;
|
2004-02-28 20:56:35 +00:00
|
|
|
#ifdef notyet
|
|
|
|
case 'p':
|
|
|
|
passive = 1;
|
|
|
|
break;
|
|
|
|
#endif
|
|
|
|
case 's':
|
2013-08-10 01:48:15 +00:00
|
|
|
nap = fetchtimeout(c, NULL, optarg, 0);
|
2013-02-27 19:03:31 +00:00
|
|
|
break;
|
|
|
|
case 'S':
|
2013-03-26 19:43:18 +00:00
|
|
|
do_syslog = 0;
|
2004-02-28 20:56:35 +00:00
|
|
|
break;
|
|
|
|
case 't':
|
2013-07-27 20:47:01 +00:00
|
|
|
timeout_sec = atoi(optarg);
|
|
|
|
timeout = parse_timeout_to_pow2ns(c, NULL, optarg);
|
|
|
|
if (debugging)
|
|
|
|
printf("Timeout is 2^%d nanoseconds\n",
|
|
|
|
timeout);
|
2004-02-28 20:56:35 +00:00
|
|
|
break;
|
2013-02-27 19:03:31 +00:00
|
|
|
case 'T':
|
2013-08-10 01:48:15 +00:00
|
|
|
carp_thresh_seconds =
|
|
|
|
fetchtimeout(c, "NULL", optarg, 0);
|
2013-02-27 19:03:31 +00:00
|
|
|
break;
|
|
|
|
case 'w':
|
|
|
|
do_timedog = 1;
|
|
|
|
break;
|
|
|
|
case 0:
|
|
|
|
lopt = longopts[longindex].name;
|
|
|
|
if (!strcmp(lopt, "pretimeout")) {
|
2013-08-10 01:48:15 +00:00
|
|
|
pretimeout = fetchtimeout(0, lopt, optarg, 0);
|
2013-02-27 19:03:31 +00:00
|
|
|
} else if (!strcmp(lopt, "pretimeout-action")) {
|
|
|
|
pretimeout_act = timeout_act_str2int(lopt,
|
|
|
|
optarg);
|
|
|
|
} else if (!strcmp(lopt, "softtimeout-action")) {
|
|
|
|
softtimeout_act = timeout_act_str2int(lopt,
|
|
|
|
optarg);
|
|
|
|
} else {
|
|
|
|
/* warnx("bad option at index %d: %s", optind,
|
|
|
|
argv[optind]);
|
|
|
|
usage();
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
break;
|
2003-06-26 09:50:52 +00:00
|
|
|
case '?':
|
|
|
|
default:
|
|
|
|
usage();
|
|
|
|
/* NOTREACHED */
|
|
|
|
}
|
|
|
|
}
|
2013-02-27 19:03:31 +00:00
|
|
|
|
|
|
|
if (carp_thresh_seconds == -1)
|
|
|
|
carp_thresh_seconds = nap;
|
|
|
|
|
2005-09-30 08:30:20 +00:00
|
|
|
if (argc != optind)
|
|
|
|
errx(EX_USAGE, "extra arguments.");
|
2004-02-28 20:56:35 +00:00
|
|
|
if (is_daemon && timeout < WD_TO_1SEC)
|
|
|
|
errx(EX_USAGE, "-t argument is less than one second.");
|
2013-07-27 20:47:01 +00:00
|
|
|
if (pretimeout_set) {
|
|
|
|
struct timespec ts;
|
|
|
|
|
|
|
|
pow2ns_to_ts(timeout, &ts);
|
2013-07-30 16:20:54 +00:00
|
|
|
if (pretimeout >= (uintmax_t)ts.tv_sec) {
|
2013-07-27 20:47:01 +00:00
|
|
|
errx(EX_USAGE,
|
|
|
|
"pretimeout (%d) >= timeout (%d -> %ld)\n"
|
|
|
|
"see manual section TIMEOUT RESOLUTION",
|
|
|
|
pretimeout, timeout_sec, (long)ts.tv_sec);
|
|
|
|
}
|
|
|
|
}
|
2003-06-26 09:50:52 +00:00
|
|
|
}
|