|
|
/* $OpenBSD: ntpd.c,v 1.41 2006/02/21 23:47:00 stevesk Exp $ */
|
|
|
|
|
|
/*
|
|
|
* Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>
|
|
|
*
|
|
|
* 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 THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
|
* WHATSOEVER RESULTING FROM LOSS OF MIND, 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/types.h>
|
|
|
#include <sys/socket.h>
|
|
|
#include <sys/wait.h>
|
|
|
#include <netinet/in.h>
|
|
|
#include <errno.h>
|
|
|
#include <poll.h>
|
|
|
#include <pwd.h>
|
|
|
#include <resolv.h>
|
|
|
#include <signal.h>
|
|
|
#include <stdio.h>
|
|
|
#include <stdlib.h>
|
|
|
#include <string.h>
|
|
|
#include <unistd.h>
|
|
|
|
|
|
#include "ntpd.h"
|
|
|
|
|
|
void sighdlr(int);
|
|
|
__dead void usage(void);
|
|
|
int main(int, char *[]);
|
|
|
int check_child(pid_t, const char *);
|
|
|
int dispatch_imsg(struct ntpd_conf *);
|
|
|
int ntpd_adjtime(double);
|
|
|
void ntpd_settime(double);
|
|
|
|
|
|
volatile sig_atomic_t quit = 0;
|
|
|
volatile sig_atomic_t reconfig = 0;
|
|
|
volatile sig_atomic_t sigchld = 0;
|
|
|
struct imsgbuf *ibuf;
|
|
|
|
|
|
void
|
|
|
sighdlr(int sig)
|
|
|
{
|
|
|
switch (sig) {
|
|
|
case SIGTERM:
|
|
|
case SIGINT:
|
|
|
quit = 1;
|
|
|
break;
|
|
|
case SIGCHLD:
|
|
|
sigchld = 1;
|
|
|
break;
|
|
|
case SIGHUP:
|
|
|
reconfig = 1;
|
|
|
break;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
__dead void
|
|
|
usage(void)
|
|
|
{
|
|
|
extern char *__progname;
|
|
|
|
|
|
fprintf(stderr, "usage: %s [-dSs] [-f file]\n", __progname);
|
|
|
exit(1);
|
|
|
}
|
|
|
|
|
|
#define POLL_MAX 8
|
|
|
#define PFD_PIPE 0
|
|
|
|
|
|
int
|
|
|
main(int argc, char *argv[])
|
|
|
{
|
|
|
struct ntpd_conf conf;
|
|
|
struct pollfd pfd[POLL_MAX];
|
|
|
pid_t chld_pid = 0, pid;
|
|
|
const char *conffile;
|
|
|
int ch, nfds, timeout = INFTIM;
|
|
|
int pipe_chld[2];
|
|
|
|
|
|
conffile = CONFFILE;
|
|
|
|
|
|
bzero(&conf, sizeof(conf));
|
|
|
|
|
|
log_init(1); /* log to stderr until daemonized */
|
|
|
res_init(); /* XXX */
|
|
|
|
|
|
while ((ch = getopt(argc, argv, "df:sS")) != -1) {
|
|
|
switch (ch) {
|
|
|
case 'd':
|
|
|
conf.debug = 1;
|
|
|
break;
|
|
|
case 'f':
|
|
|
conffile = optarg;
|
|
|
break;
|
|
|
case 's':
|
|
|
conf.settime = 1;
|
|
|
break;
|
|
|
case 'S':
|
|
|
conf.settime = 0;
|
|
|
break;
|
|
|
default:
|
|
|
usage();
|
|
|
/* NOTREACHED */
|
|
|
}
|
|
|
}
|
|
|
|
|
|
if (parse_config(conffile, &conf))
|
|
|
exit(1);
|
|
|
|
|
|
if (geteuid()) {
|
|
|
fprintf(stderr, "ntpd: need root privileges\n");
|
|
|
exit(1);
|
|
|
}
|
|
|
|
|
|
if (getpwnam(NTPD_USER) == NULL) {
|
|
|
fprintf(stderr, "ntpd: unknown user %s\n", NTPD_USER);
|
|
|
exit(1);
|
|
|
}
|
|
|
endpwent();
|
|
|
|
|
|
if (!conf.settime) {
|
|
|
log_init(conf.debug);
|
|
|
if (!conf.debug)
|
|
|
if (daemon(1, 0))
|
|
|
fatal("daemon");
|
|
|
} else
|
|
|
timeout = SETTIME_TIMEOUT * 1000;
|
|
|
|
|
|
if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe_chld) == -1)
|
|
|
fatal("socketpair");
|
|
|
|
|
|
/* fork child process */
|
|
|
chld_pid = ntp_main(pipe_chld, &conf);
|
|
|
|
|
|
setproctitle("[priv]");
|
|
|
|
|
|
signal(SIGTERM, sighdlr);
|
|
|
signal(SIGINT, sighdlr);
|
|
|
signal(SIGCHLD, sighdlr);
|
|
|
signal(SIGHUP, sighdlr);
|
|
|
|
|
|
close(pipe_chld[1]);
|
|
|
|
|
|
if ((ibuf = malloc(sizeof(struct imsgbuf))) == NULL)
|
|
|
fatal(NULL);
|
|
|
imsg_init(ibuf, pipe_chld[0]);
|
|
|
|
|
|
while (quit == 0) {
|
|
|
pfd[PFD_PIPE].fd = ibuf->fd;
|
|
|
pfd[PFD_PIPE].events = POLLIN;
|
|
|
if (ibuf->w.queued)
|
|
|
pfd[PFD_PIPE].events |= POLLOUT;
|
|
|
|
|
|
if ((nfds = poll(pfd, 1, timeout)) == -1)
|
|
|
if (errno != EINTR) {
|
|
|
log_warn("poll error");
|
|
|
quit = 1;
|
|
|
}
|
|
|
|
|
|
if (nfds == 0 && conf.settime) {
|
|
|
conf.settime = 0;
|
|
|
timeout = INFTIM;
|
|
|
log_init(conf.debug);
|
|
|
log_debug("no reply received in time, skipping initial "
|
|
|
"time setting");
|
|
|
if (!conf.debug)
|
|
|
if (daemon(1, 0))
|
|
|
fatal("daemon");
|
|
|
}
|
|
|
|
|
|
if (nfds > 0 && (pfd[PFD_PIPE].revents & POLLOUT))
|
|
|
if (msgbuf_write(&ibuf->w) < 0) {
|
|
|
log_warn("pipe write error (to child)");
|
|
|
quit = 1;
|
|
|
}
|
|
|
|
|
|
if (nfds > 0 && pfd[PFD_PIPE].revents & POLLIN) {
|
|
|
nfds--;
|
|
|
if (dispatch_imsg(&conf) == -1)
|
|
|
quit = 1;
|
|
|
}
|
|
|
|
|
|
if (sigchld) {
|
|
|
if (check_child(chld_pid, "child")) {
|
|
|
quit = 1;
|
|
|
chld_pid = 0;
|
|
|
}
|
|
|
sigchld = 0;
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
signal(SIGCHLD, SIG_DFL);
|
|
|
|
|
|
if (chld_pid)
|
|
|
kill(chld_pid, SIGTERM);
|
|
|
|
|
|
do {
|
|
|
if ((pid = wait(NULL)) == -1 &&
|
|
|
errno != EINTR && errno != ECHILD)
|
|
|
fatal("wait");
|
|
|
} while (pid != -1 || (pid == -1 && errno == EINTR));
|
|
|
|
|
|
msgbuf_clear(&ibuf->w);
|
|
|
free(ibuf);
|
|
|
log_info("Terminating");
|
|
|
return (0);
|
|
|
}
|
|
|
|
|
|
int
|
|
|
check_child(pid_t pid, const char *pname)
|
|
|
{
|
|
|
int status, sig;
|
|
|
char *signame;
|
|
|
|
|
|
if (waitpid(pid, &status, WNOHANG) > 0) {
|
|
|
if (WIFEXITED(status)) {
|
|
|
log_warnx("Lost child: %s exited", pname);
|
|
|
return (1);
|
|
|
}
|
|
|
if (WIFSIGNALED(status)) {
|
|
|
sig = WTERMSIG(status);
|
|
|
signame = strsignal(sig) ? strsignal(sig) : "unknown";
|
|
|
log_warnx("Lost child: %s terminated; signal %d (%s)",
|
|
|
pname, sig, signame);
|
|
|
return (1);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
return (0);
|
|
|
}
|
|
|
|
|
|
int
|
|
|
dispatch_imsg(struct ntpd_conf *conf)
|
|
|
{
|
|
|
struct imsg imsg;
|
|
|
int n, cnt;
|
|
|
double d;
|
|
|
char *name;
|
|
|
struct ntp_addr *h, *hn;
|
|
|
struct buf *buf;
|
|
|
|
|
|
if ((n = imsg_read(ibuf)) == -1)
|
|
|
return (-1);
|
|
|
|
|
|
if (n == 0) { /* connection closed */
|
|
|
log_warnx("dispatch_imsg in main: pipe closed");
|
|
|
return (-1);
|
|
|
}
|
|
|
|
|
|
for (;;) {
|
|
|
if ((n = imsg_get(ibuf, &imsg)) == -1)
|
|
|
return (-1);
|
|
|
|
|
|
if (n == 0)
|
|
|
break;
|
|
|
|
|
|
switch (imsg.hdr.type) {
|
|
|
case IMSG_ADJTIME:
|
|
|
if (imsg.hdr.len != IMSG_HEADER_SIZE + sizeof(d))
|
|
|
fatalx("invalid IMSG_ADJTIME received");
|
|
|
memcpy(&d, imsg.data, sizeof(d));
|
|
|
n = ntpd_adjtime(d);
|
|
|
imsg_compose(ibuf, IMSG_ADJTIME, 0, 0, &n, sizeof(n));
|
|
|
break;
|
|
|
case IMSG_SETTIME:
|
|
|
if (imsg.hdr.len != IMSG_HEADER_SIZE + sizeof(d))
|
|
|
fatalx("invalid IMSG_SETTIME received");
|
|
|
if (!conf->settime)
|
|
|
break;
|
|
|
log_init(conf->debug);
|
|
|
memcpy(&d, imsg.data, sizeof(d));
|
|
|
ntpd_settime(d);
|
|
|
/* daemonize now */
|
|
|
if (!conf->debug)
|
|
|
if (daemon(1, 0))
|
|
|
fatal("daemon");
|
|
|
conf->settime = 0;
|
|
|
break;
|
|
|
case IMSG_HOST_DNS:
|
|
|
name = imsg.data;
|
|
|
if (imsg.hdr.len < 1 + IMSG_HEADER_SIZE)
|
|
|
fatalx("invalid IMSG_HOST_DNS received");
|
|
|
imsg.hdr.len -= 1 + IMSG_HEADER_SIZE;
|
|
|
if (name[imsg.hdr.len] != '\0' ||
|
|
|
strlen(name) != imsg.hdr.len)
|
|
|
fatalx("invalid IMSG_HOST_DNS received");
|
|
|
if ((cnt = host_dns(name, &hn)) == -1)
|
|
|
break;
|
|
|
buf = imsg_create(ibuf, IMSG_HOST_DNS,
|
|
|
imsg.hdr.peerid, 0,
|
|
|
cnt * sizeof(struct sockaddr_storage));
|
|
|
if (buf == NULL)
|
|
|
break;
|
|
|
if (cnt > 0)
|
|
|
for (h = hn; h != NULL; h = h->next)
|
|
|
imsg_add(buf, &h->ss, sizeof(h->ss));
|
|
|
|
|
|
imsg_close(ibuf, buf);
|
|
|
break;
|
|
|
default:
|
|
|
break;
|
|
|
}
|
|
|
imsg_free(&imsg);
|
|
|
}
|
|
|
return (0);
|
|
|
}
|
|
|
|
|
|
int
|
|
|
ntpd_adjtime(double d)
|
|
|
{
|
|
|
struct timeval tv, olddelta;
|
|
|
int synced = 0;
|
|
|
static int firstadj = 1;
|
|
|
|
|
|
if (d >= (double)LOG_NEGLIGEE / 1000 ||
|
|
|
d <= -1 * (double)LOG_NEGLIGEE / 1000)
|
|
|
log_info("adjusting local clock by %fs", d);
|
|
|
else
|
|
|
log_debug("adjusting local clock by %fs", d);
|
|
|
d_to_tv(d, &tv);
|
|
|
if (adjtime(&tv, &olddelta) == -1)
|
|
|
log_warn("adjtime failed");
|
|
|
else if (!firstadj && olddelta.tv_sec == 0 && olddelta.tv_usec == 0)
|
|
|
synced = 1;
|
|
|
firstadj = 0;
|
|
|
return (synced);
|
|
|
}
|
|
|
|
|
|
void
|
|
|
ntpd_settime(double d)
|
|
|
{
|
|
|
struct timeval tv, curtime;
|
|
|
char buf[80];
|
|
|
time_t tval;
|
|
|
|
|
|
/* if the offset is small, don't call settimeofday */
|
|
|
if (d < SETTIME_MIN_OFFSET && d > -SETTIME_MIN_OFFSET)
|
|
|
return;
|
|
|
|
|
|
if (gettimeofday(&curtime, NULL) == -1) {
|
|
|
log_warn("gettimeofday");
|
|
|
return;
|
|
|
}
|
|
|
d_to_tv(d, &tv);
|
|
|
curtime.tv_usec += tv.tv_usec + 1000000;
|
|
|
curtime.tv_sec += tv.tv_sec - 1 + (curtime.tv_usec / 1000000);
|
|
|
curtime.tv_usec %= 1000000;
|
|
|
|
|
|
if (settimeofday(&curtime, NULL) == -1) {
|
|
|
log_warn("settimeofday");
|
|
|
return;
|
|
|
}
|
|
|
tval = curtime.tv_sec;
|
|
|
strftime(buf, sizeof(buf), "%a %b %e %H:%M:%S %Z %Y",
|
|
|
localtime(&tval));
|
|
|
log_info("set local clock to %s (offset %fs)", buf, d);
|
|
|
}
|