#ifdef HAVE_CONFIG_H
# include <config.h>
#endif
#include "ntp_machine.h"
#include "ntpd.h"
#include "ntp_stdlib.h"
#include <stdio.h>
#include <signal.h>
#ifdef HAVE_SYS_SIGNAL_H
# include <sys/signal.h>
#endif
#ifdef HAVE_UNISTD_H
# include <unistd.h>
#endif
#if defined(HAVE_IO_COMPLETION_PORT)
# include "ntp_iocompletionport.h"
# include "ntp_timer.h"
#endif
#ifdef PUBKEY
#include "ntp_crypto.h"
#endif
volatile int alarm_flag;
static u_long adjust_timer;
static u_long keys_timer;
static u_long hourly_timer;
static u_long huffpuff_timer;
#ifdef AUTOKEY
static u_long revoke_timer;
u_long sys_revoke = 1 << KEY_REVOKE;
#endif
volatile u_long alarm_overflow;
#define MINUTE 60
#define HOUR (60*60)
u_long current_time;
u_long timer_timereset;
u_long timer_overflows;
u_long timer_xmtcalls;
#if defined(VMS)
static int vmstimer[2];
static int vmsinc[2];
#endif
#if defined SYS_WINNT
static HANDLE WaitableTimerHandle = NULL;
#else
static RETSIGTYPE alarming P((int));
#endif
void
init_timer(void)
{
#if !defined(VMS)
# if !defined SYS_WINNT || defined(SYS_CYGWIN32)
# ifndef HAVE_TIMER_SETTIME
struct itimerval itimer;
# else
static timer_t ntpd_timerid;
struct itimerspec itimer;
# endif
# else
HANDLE hToken;
TOKEN_PRIVILEGES tkp;
# endif
#endif
alarm_flag = 0;
alarm_overflow = 0;
adjust_timer = 1;
hourly_timer = HOUR;
huffpuff_timer = 0;
current_time = 0;
timer_overflows = 0;
timer_xmtcalls = 0;
timer_timereset = 0;
#if !defined(SYS_WINNT)
# if !defined(VMS)
# if defined(HAVE_TIMER_CREATE) && defined(HAVE_TIMER_SETTIME)
if (timer_create (CLOCK_REALTIME, NULL, &ntpd_timerid) ==
# ifdef SYS_VXWORKS
ERROR
# else
-1
# endif
)
{
fprintf (stderr, "timer create FAILED\n");
exit (0);
}
(void) signal_no_reset(SIGALRM, alarming);
itimer.it_interval.tv_sec = itimer.it_value.tv_sec = (1<<EVENT_TIMEOUT);
itimer.it_interval.tv_nsec = itimer.it_value.tv_nsec = 0;
timer_settime(ntpd_timerid, 0 , &itimer, NULL);
# else
(void) signal_no_reset(SIGALRM, alarming);
itimer.it_interval.tv_sec = itimer.it_value.tv_sec = (1<<EVENT_TIMEOUT);
itimer.it_interval.tv_usec = itimer.it_value.tv_usec = 0;
setitimer(ITIMER_REAL, &itimer, (struct itimerval *)0);
# endif
# else
vmsinc[0] = 10000000;
vmsinc[1] = 0;
lib$emul(&(1<<EVENT_TIMEOUT), &vmsinc, &0, &vmsinc);
sys$gettim(&vmstimer);
lib$addx(&vmsinc, &vmstimer, &vmstimer);
sys$setimr(0, &vmstimer, alarming, alarming, 0);
# endif
#else
_tzset();
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) {
msyslog(LOG_ERR, "OpenProcessToken failed: %m");
exit(1);
}
LookupPrivilegeValue(NULL, SE_SYSTEMTIME_NAME, &tkp.Privileges[0].Luid);
tkp.PrivilegeCount = 1;
tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES) NULL, 0);
if (GetLastError() != ERROR_SUCCESS) {
msyslog(LOG_ERR, "AdjustTokenPrivileges failed: %m");
}
WaitableTimerHandle = CreateWaitableTimer(NULL, FALSE, NULL);
if (WaitableTimerHandle == NULL) {
msyslog(LOG_ERR, "CreateWaitableTimer failed: %m");
exit(1);
}
else {
DWORD Period = (1<<EVENT_TIMEOUT) * 1000;
LARGE_INTEGER DueTime;
DueTime.QuadPart = Period * 10000i64;
if (!SetWaitableTimer(WaitableTimerHandle, &DueTime, Period, NULL, NULL, FALSE) != NO_ERROR) {
msyslog(LOG_ERR, "SetWaitableTimer failed: %m");
exit(1);
}
}
#endif
}
#if defined(SYS_WINNT)
extern HANDLE
get_timer_handle(void)
{
return WaitableTimerHandle;
}
#endif
void
timer(void)
{
register struct peer *peer, *next_peer;
u_int n;
current_time += (1<<EVENT_TIMEOUT);
if (adjust_timer <= current_time) {
adjust_timer += 1;
adj_host_clock();
}
for (n = 0; n < HASH_SIZE; n++) {
for (peer = peer_hash[n]; peer != 0; peer = next_peer) {
next_peer = peer->next;
if (peer->action && peer->nextaction <= current_time)
peer->action(peer);
if (peer->nextdate <= current_time) {
#ifdef REFCLOCK
if (peer->flags & FLAG_REFCLOCK)
refclock_transmit(peer);
else
transmit(peer);
#else
transmit(peer);
#endif
}
}
}
if (keys_timer <= current_time) {
keys_timer += MINUTE;
auth_agekeys();
}
if (huffpuff_timer <= current_time) {
huffpuff_timer += HUFFPUFF;
huffpuff();
}
#ifdef AUTOKEY
if (revoke_timer <= current_time) {
revoke_timer += sys_revoke;
expire_all();
#ifdef DEBUG
if (debug)
printf("key expire: at %lu next %lu\n",
current_time, revoke_timer);
#endif
}
#endif
if (hourly_timer <= current_time) {
hourly_timer += HOUR;
hourly_stats();
}
}
#ifndef SYS_WINNT
static RETSIGTYPE
alarming(
int sig
)
{
#if !defined(VMS)
if (initializing)
return;
if (alarm_flag)
alarm_overflow++;
else
alarm_flag++;
#else
if (!initializing) {
if (alarm_flag) alarm_overflow++;
else alarm_flag = 1;
}
lib$addx(&vmsinc,&vmstimer,&vmstimer);
sys$setimr(0,&vmstimer,alarming,alarming,0);
#endif
}
#endif
void
timer_clr_stats(void)
{
timer_overflows = 0;
timer_xmtcalls = 0;
timer_timereset = current_time;
}