|
|
/* $OpenBSD: ntpd.c,v 1.3 2004/06/01 16:27:09 henning 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 <arpa/inet.h>
|
|
|
#include <err.h>
|
|
|
#include <errno.h>
|
|
|
#include <fcntl.h>
|
|
|
#include <poll.h>
|
|
|
#include <pwd.h>
|
|
|
#include <signal.h>
|
|
|
#include <stdio.h>
|
|
|
#include <stdlib.h>
|
|
|
#include <string.h>
|
|
|
#include <unistd.h>
|
|
|
|
|
|
#include "ntpd.h"
|
|
|
|
|
|
void sighdlr(int);
|
|
|
void usage(void);
|
|
|
int main(int, char *[]);
|
|
|
int check_child(pid_t, const char *);
|
|
|
int reconfigure(char *);
|
|
|
int dispatch_imsg(void);
|
|
|
|
|
|
int rfd = -1;
|
|
|
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;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
void
|
|
|
usage(void)
|
|
|
{
|
|
|
extern char *__progname;
|
|
|
|
|
|
fprintf(stderr, "usage: %s [-d] [-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;
|
|
|
char *conffile;
|
|
|
int debug = 0;
|
|
|
int ch, nfds;
|
|
|
int pipe_chld[2];
|
|
|
|
|
|
conffile = CONFFILE;
|
|
|
|
|
|
bzero(&conf, sizeof(conf));
|
|
|
TAILQ_INIT(&conf.listen_addrs);
|
|
|
|
|
|
log_init(1); /* log to stderr until daemonized */
|
|
|
|
|
|
while ((ch = getopt(argc, argv, "df:")) != -1) {
|
|
|
switch (ch) {
|
|
|
case 'd':
|
|
|
debug = 1;
|
|
|
break;
|
|
|
case 'f':
|
|
|
conffile = optarg;
|
|
|
break;
|
|
|
default:
|
|
|
usage();
|
|
|
/* NOTREACHED */
|
|
|
}
|
|
|
}
|
|
|
|
|
|
if (geteuid())
|
|
|
errx(1, "need root privileges");
|
|
|
|
|
|
if (getpwnam(NTPD_USER) == NULL)
|
|
|
errx(1, "unknown user %s", NTPD_USER);
|
|
|
endpwent();
|
|
|
|
|
|
log_init(debug);
|
|
|
|
|
|
if (!debug)
|
|
|
daemon(1, 0);
|
|
|
|
|
|
log_info("startup");
|
|
|
|
|
|
if (pipe(pipe_chld) == -1)
|
|
|
fatal("pipe");
|
|
|
|
|
|
/* fork children */
|
|
|
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]);
|
|
|
|
|
|
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, INFTIM)) == -1)
|
|
|
if (errno != EINTR) {
|
|
|
log_warn("poll error");
|
|
|
quit = 1;
|
|
|
}
|
|
|
|
|
|
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() == -1)
|
|
|
quit = 1;
|
|
|
}
|
|
|
|
|
|
if (reconfig) {
|
|
|
log_info("rereading config");
|
|
|
reconfigure(conffile);
|
|
|
reconfig = 0;
|
|
|
}
|
|
|
|
|
|
if (sigchld) {
|
|
|
if (check_child(chld_pid, "child"))
|
|
|
quit = 1;
|
|
|
sigchld = 0;
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
signal(SIGCHLD, SIG_IGN);
|
|
|
|
|
|
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));
|
|
|
|
|
|
log_info("Terminating");
|
|
|
return (0);
|
|
|
}
|
|
|
|
|
|
int
|
|
|
check_child(pid_t pid, const char *pname)
|
|
|
{
|
|
|
int status;
|
|
|
|
|
|
if (waitpid(pid, &status, WNOHANG) > 0) {
|
|
|
if (WIFEXITED(status)) {
|
|
|
log_warnx("Lost child: %s exited", pname);
|
|
|
return (1);
|
|
|
}
|
|
|
if (WIFSIGNALED(status)) {
|
|
|
log_warnx("Lost child: %s terminated; signal %d",
|
|
|
pname, WTERMSIG(status));
|
|
|
return (1);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
return (0);
|
|
|
}
|
|
|
|
|
|
int
|
|
|
reconfigure(char *conffile)
|
|
|
{
|
|
|
return (-1);
|
|
|
}
|
|
|
|
|
|
int
|
|
|
dispatch_imsg(void)
|
|
|
{
|
|
|
struct imsg imsg;
|
|
|
int n;
|
|
|
|
|
|
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) {
|
|
|
default:
|
|
|
break;
|
|
|
}
|
|
|
imsg_free(&imsg);
|
|
|
}
|
|
|
return (0);
|
|
|
}
|