Fincer
/
openntpd-openbsd
mirror of https://github.com/Fincer/openntpd-openbsd
1
0
Fork 0
Code Issues 0 Projects 0 Releases 61 Wiki Activity
Source code pulled from OpenBSD for OpenNTPD. The place to contribute to this code is via the OpenBSD CVS tree.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
3808 Commits
49 Branches
393 MiB
 
 
 
 
 
 
Tree: 265f65c52e
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '265f65c52e'
${ noResults }
openntpd-openbsd/src/usr.sbin/ntpd/ntpd.c

277 lines
5.4 KiB
Raw Blame History

/* $OpenBSD: ntpd.c,v 1.14 2004/08/12 16:33:59 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 <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 dispatch_imsg(void);
void ntpd_adjtime(double);
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));
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 (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();
log_init(debug);
if (!debug)
daemon(1, 0);
if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, pipe_chld) == -1)
fatal("socketpair");
/* 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 (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
dispatch_imsg(void)
{
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))
fatal("invalid IMSG_ADJTIME received");
memcpy(&d, imsg.data, sizeof(d));
ntpd_adjtime(d);
break;
case IMSG_HOST_DNS:
name = imsg.data;
if (imsg.hdr.len != strlen(name) + 1 + IMSG_HEADER_SIZE)
fatal("invalid IMSG_HOST_DNS received");
if ((cnt = host_dns(name, &hn)) > 0) {
buf = imsg_create(&ibuf, IMSG_HOST_DNS,
imsg.hdr.peerid,
cnt * sizeof(struct sockaddr_storage));
if (buf == NULL)
break;
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);
}
void
ntpd_adjtime(double d)
{
struct timeval tv;
d_to_tv(d, &tv);
log_info("adjusting local clock by %fs", d);
if (adjtime(&tv, NULL) == -1)
log_warn("adjtime failed");
}