#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdio.h>
#include <sys/types.h>
#include "ntpd.h"
#include "ntp_lists.h"
#include "ntp_stdlib.h"
#include "ntp_control.h"
#include <ntp_random.h>
#ifdef OPENSSL
#include "openssl/rand.h"
#endif
#ifdef SYS_WINNT
extern int accept_wildcard_if_for_winnt;
#endif
#define AM_MODES 7
#define NO_PEER 0
int AM[AM_MODES][AM_MODES] = {
{ AM_ERR, AM_NEWPASS, AM_NOMATCH, AM_FXMIT, AM_MANYCAST, AM_NEWBCL},
{ AM_ERR, AM_PROCPKT, AM_PROCPKT, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH},
{ AM_ERR, AM_PROCPKT, AM_ERR, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH},
{ AM_ERR, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH, AM_PROCPKT, AM_NOMATCH},
{ AM_ERR, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH},
{ AM_ERR, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH},
{ AM_ERR, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH, AM_NOMATCH, AM_PROCPKT},
};
#define MATCH_ASSOC(x, y) AM[(x)][(y)]
struct peer *peer_hash[NTP_HASH_SIZE];
int peer_hash_count[NTP_HASH_SIZE];
struct peer *assoc_hash[NTP_HASH_SIZE];
int assoc_hash_count[NTP_HASH_SIZE];
static struct peer *peer_free;
int peer_free_count;
static associd_t current_association_ID;
#define INIT_PEER_ALLOC 15
#define INC_PEER_ALLOC 5
u_long peer_timereset;
u_long findpeer_calls;
u_long assocpeer_calls;
u_long peer_allocations;
u_long peer_demobilizations;
int total_peer_structs;
int peer_associations;
int peer_preempt;
static struct peer init_peer_alloc[INIT_PEER_ALLOC];
static void getmorepeermem (void);
static struct interface *select_peerinterface (struct peer *, sockaddr_u *, struct interface *, u_char);
static int score(struct peer *);
void
init_peer(void)
{
register int i;
memset(peer_hash, 0, sizeof(peer_hash));
memset(peer_hash_count, 0, sizeof(peer_hash_count));
memset(assoc_hash, 0, sizeof(assoc_hash));
memset(assoc_hash_count, 0, sizeof(assoc_hash_count));
findpeer_calls = peer_allocations = 0;
assocpeer_calls = peer_demobilizations = 0;
peer_free = NULL;
for (i = 0; i < INIT_PEER_ALLOC; i++)
LINK_SLIST(peer_free, &init_peer_alloc[i], next);
total_peer_structs = INIT_PEER_ALLOC;
peer_free_count = INIT_PEER_ALLOC;
while ((current_association_ID = ntp_random() & 0xffff) == 0);
}
static void
getmorepeermem(void)
{
register int i;
register struct peer *peer;
peer = (struct peer *)emalloc(INC_PEER_ALLOC *
sizeof(struct peer));
for (i = 0; i < INC_PEER_ALLOC; i++) {
LINK_SLIST(peer_free, peer, next);
peer++;
}
total_peer_structs += INC_PEER_ALLOC;
peer_free_count += INC_PEER_ALLOC;
}
struct peer *
findexistingpeer(
sockaddr_u *addr,
struct peer *start_peer,
int mode
)
{
register struct peer *peer;
if (NULL == start_peer)
peer = peer_hash[NTP_HASH_ADDR(addr)];
else
peer = start_peer->next;
while (peer != NULL) {
if (SOCK_EQ(addr, &peer->srcadr)
&& NSRCPORT(addr) == NSRCPORT(&peer->srcadr)
&& (-1 == mode || peer->hmode == mode))
break;
peer = peer->next;
}
return (peer);
}
struct peer *
findpeer(
sockaddr_u *srcadr,
struct interface *dstadr,
int pkt_mode,
int *action
)
{
register struct peer *peer;
u_int hash;
findpeer_calls++;
hash = NTP_HASH_ADDR(srcadr);
for (peer = peer_hash[hash]; peer != NULL; peer = peer->next) {
if (SOCK_EQ(srcadr, &peer->srcadr) &&
NSRCPORT(srcadr) == NSRCPORT(&peer->srcadr)) {
*action = MATCH_ASSOC(peer->hmode, pkt_mode);
if (*action == AM_ERR)
return ((struct peer *)0);
if (*action != AM_NOMATCH)
break;
}
}
if (peer == 0) {
*action = MATCH_ASSOC(NO_PEER, pkt_mode);
return ((struct peer *)0);
}
set_peerdstadr(peer, dstadr);
return (peer);
}
struct peer *
findpeerbyassoc(
u_int assoc
)
{
register struct peer *peer;
u_int hash;
assocpeer_calls++;
hash = assoc & NTP_HASH_MASK;
for (peer = assoc_hash[hash]; peer != 0; peer =
peer->ass_next) {
if (assoc == peer->associd)
return (peer);
}
return (NULL);
}
void
clear_all(void)
{
struct peer *peer, *next_peer;
int n;
for (n = 0; n < NTP_HASH_SIZE; n++) {
for (peer = peer_hash[n]; peer != 0; peer = next_peer) {
next_peer = peer->next;
if (!(peer->cast_flags & (MDF_ACAST |
MDF_MCAST | MDF_BCAST))) {
peer_clear(peer, "STEP");
}
}
}
#ifdef DEBUG
if (debug)
printf("clear_all: at %lu\n", current_time);
#endif
}
int
score_all(
struct peer *peer
)
{
struct peer *speer, *next_peer;
int n;
int temp, tamp;
tamp = score(peer);
temp = 100;
for (n = 0; n < NTP_HASH_SIZE; n++) {
for (speer = peer_hash[n]; speer != 0; speer =
next_peer) {
int x;
next_peer = speer->next;
if ((x = score(speer)) < temp && (peer->flags &
FLAG_PREEMPT))
temp = x;
}
}
#ifdef DEBUG
if (debug)
printf("score_all: at %lu score %d min %d\n",
current_time, tamp, temp);
#endif
if (tamp != temp)
temp = 0;
return (temp);
}
static int
score(
struct peer *peer
)
{
int temp;
temp = 0;
if (!(peer->flash & TEST10))
temp++;
if (!(peer->flash & TEST13))
temp++;
if (!(peer->flash & TEST12))
temp++;
if (!(peer->flash & TEST11))
temp++;
if (peer->status >= CTL_PST_SEL_SELCAND)
temp++;
if (peer->status != CTL_PST_SEL_EXCESS)
temp++;
return (temp);
}
void
unpeer(
struct peer *peer_to_remove
)
{
register struct peer *unlinked;
int hash;
char tbuf[80];
snprintf(tbuf, sizeof(tbuf), "assoc %d",
peer_to_remove->associd);
report_event(PEVNT_DEMOBIL, peer_to_remove, tbuf);
set_peerdstadr(peer_to_remove, NULL);
hash = NTP_HASH_ADDR(&peer_to_remove->srcadr);
peer_hash_count[hash]--;
peer_demobilizations++;
peer_associations--;
if (peer_to_remove->flags & FLAG_PREEMPT)
peer_preempt--;
#ifdef REFCLOCK
if (peer_to_remove->flags & FLAG_REFCLOCK)
refclock_unpeer(peer_to_remove);
#endif
peer_to_remove->action = 0;
UNLINK_SLIST(unlinked, peer_hash[hash], peer_to_remove, next,
struct peer);
if (NULL == unlinked) {
peer_hash_count[hash]++;
msyslog(LOG_ERR, "peer struct for %s not in table!",
stoa(&peer_to_remove->srcadr));
}
hash = peer_to_remove->associd & NTP_HASH_MASK;
assoc_hash_count[hash]--;
UNLINK_SLIST(unlinked, assoc_hash[hash], peer_to_remove,
ass_next, struct peer);
if (NULL == unlinked) {
assoc_hash_count[hash]++;
msyslog(LOG_ERR,
"peer struct for %s not in association table!",
stoa(&peer_to_remove->srcadr));
}
free(peer_to_remove->dns_name);
peer_to_remove->dns_name = NULL;
LINK_SLIST(peer_free, peer_to_remove, next);
peer_free_count++;
}
struct peer *
peer_config(
sockaddr_u *srcadr,
struct interface *dstadr,
int hmode,
int version,
int minpoll,
int maxpoll,
u_int flags,
int ttl,
keyid_t key,
u_char *keystr,
char *dns_name
)
{
u_char cast_flags;
u_long next_update;
struct peer *peer = NULL;
switch (hmode) {
case MODE_BROADCAST:
if (IS_MCAST(srcadr))
cast_flags = MDF_MCAST;
else
cast_flags = MDF_BCAST;
break;
case MODE_CLIENT:
if (IS_MCAST(srcadr))
cast_flags = MDF_ACAST;
else
cast_flags = MDF_UCAST;
break;
default:
cast_flags = MDF_UCAST;
}
if (mode_ntpdate)
flags |= FLAG_IBURST;
peer = newpeer(srcadr, dstadr, hmode, version, minpoll, maxpoll,
flags | FLAG_CONFIG, cast_flags, ttl, key);
if (peer) {
peer->dns_name = strdup(dns_name);
msyslog(LOG_NOTICE, "peer %s @ %s", dns_name, stoa(&peer->srcadr));
next_update = get_dns_flags(dns_name, peer);
if (dns_timer == 0 || (dns_timer > next_update))
dns_timer = next_update;
}
return peer;
}
void
set_peerdstadr(
struct peer *peer,
struct interface *interface
)
{
struct peer *unlinked;
if (peer->dstadr != interface) {
if (interface != NULL && (peer->cast_flags &
MDF_BCLNT) && (interface->flags & INT_MCASTIF) &&
peer->burst) {
return;
}
if (peer->dstadr != NULL) {
peer->dstadr->peercnt--;
UNLINK_SLIST(unlinked, peer->dstadr->peers,
peer, ilink, struct peer);
msyslog(LOG_INFO,
"%s interface %s -> %s",
stoa(&peer->srcadr),
stoa(&peer->dstadr->sin),
(interface != NULL)
? stoa(&interface->sin)
: "(null)");
}
peer->dstadr = interface;
if (peer->dstadr != NULL) {
LINK_SLIST(peer->dstadr->peers, peer, ilink);
peer->dstadr->peercnt++;
}
}
}
static void
peer_refresh_interface(
struct peer *peer
)
{
struct interface *niface, *piface;
niface = select_peerinterface(peer, &peer->srcadr, NULL,
peer->cast_flags);
#ifdef DEBUG
if (debug > 3)
{
printf(
"peer_refresh_interface: %s->%s mode %d vers %d poll %d %d flags 0x%x 0x%x ttl %d key %08x: new interface: ",
peer->dstadr == NULL ? "<null>" :
stoa(&peer->dstadr->sin), stoa(&peer->srcadr),
peer->hmode, peer->version, peer->minpoll,
peer->maxpoll, peer->flags, peer->cast_flags,
peer->ttl, peer->keyid);
if (niface != NULL) {
printf(
"fd=%d, bfd=%d, name=%.16s, flags=0x%x, scope=%d, ",
niface->fd, niface->bfd, niface->name,
niface->flags, niface->scopeid);
printf(", sin=%s", stoa((&niface->sin)));
if (niface->flags & INT_BROADCAST)
printf(", bcast=%s,",
stoa((&niface->bcast)));
printf(", mask=%s\n", stoa((&niface->mask)));
} else {
printf("<NONE>\n");
}
}
#endif
piface = peer->dstadr;
set_peerdstadr(peer, niface);
if (peer->dstadr) {
if (peer->dstadr != piface && !(peer->cast_flags &
MDF_ACAST) && peer->pmode != MODE_BROADCAST)
peer_clear(peer, "XFAC");
if (peer->cast_flags & MDF_BCAST) {
enable_broadcast(peer->dstadr, &peer->srcadr);
}
if (peer->cast_flags & MDF_MCAST) {
enable_multicast_if(peer->dstadr,
&peer->srcadr);
}
}
}
void
refresh_all_peerinterfaces(void)
{
struct peer *peer, *next_peer;
int n;
for (n = 0; n < NTP_HASH_SIZE; n++) {
for (peer = peer_hash[n]; peer != 0; peer = next_peer) {
next_peer = peer->next;
peer_refresh_interface(peer);
}
}
}
static struct interface *
select_peerinterface(
struct peer * peer,
sockaddr_u * srcadr,
struct interface * dstadr,
u_char cast_flags
)
{
struct interface *interface;
if (ISREFCLOCKADR(srcadr))
interface = loopback_interface;
else
if (cast_flags & (MDF_BCLNT | MDF_ACAST | MDF_MCAST | MDF_BCAST)) {
interface = findbcastinter(srcadr);
#ifdef DEBUG
if (debug > 3) {
if (interface != NULL)
printf("Found *-cast interface address %s, for address %s\n",
stoa(&(interface)->sin), stoa(srcadr));
else
printf("No *-cast local address found for address %s\n",
stoa(srcadr));
}
#endif
if (interface == ANY_INTERFACE_CHOOSE(srcadr))
interface = findinterface(srcadr);
}
else if (dstadr != NULL && dstadr !=
ANY_INTERFACE_CHOOSE(srcadr))
interface = dstadr;
else
interface = findinterface(srcadr);
if (interface != NULL && interface->flags & INT_WILDCARD)
#ifdef SYS_WINNT
if ( !accept_wildcard_if_for_winnt )
#endif
interface = NULL;
return interface;
}
struct peer *
newpeer(
sockaddr_u *srcadr,
struct interface *dstadr,
int hmode,
int version,
int minpoll,
int maxpoll,
u_int flags,
u_char cast_flags,
int ttl,
keyid_t key
)
{
struct peer *peer;
u_int hash;
char tbuf[80];
#ifdef OPENSSL
if (!crypto_flags) {
if (key > NTP_MAXKEY) {
return (NULL);
} else if (flags & FLAG_SKEY) {
msyslog(LOG_ERR, "Autokey not configured");
return (NULL);
}
}
#endif
peer = findexistingpeer(srcadr, NULL, hmode);
if (dstadr != NULL) {
while (peer != NULL) {
if (peer->dstadr == dstadr)
break;
if (dstadr == ANY_INTERFACE_CHOOSE(srcadr) &&
peer->dstadr == findinterface(srcadr))
break;
peer = findexistingpeer(srcadr, peer, hmode);
}
}
if (peer != NULL)
return (NULL);
if (peer_free_count == 0)
getmorepeermem();
UNLINK_HEAD_SLIST(peer, peer_free, next);
peer_free_count--;
peer_associations++;
if (flags & FLAG_PREEMPT)
peer_preempt++;
memset(peer, 0, sizeof(*peer));
peer->associd = current_association_ID;
if (++current_association_ID == 0)
++current_association_ID;
DPRINTF(3, ("newpeer: cast flags: 0x%x for address: %s\n",
cast_flags, stoa(srcadr)));
peer->srcadr = *srcadr;
set_peerdstadr(peer, select_peerinterface(peer, srcadr, dstadr,
cast_flags));
peer->hmode = (u_char)hmode;
peer->version = (u_char)version;
peer->flags = flags;
if (minpoll == 0)
peer->minpoll = NTP_MINDPOLL;
else {
peer->flags |= FLAG_UMINPOLL;
peer->minpoll = (u_char)min(minpoll, NTP_MAXPOLL);
}
if (maxpoll == 0)
peer->maxpoll = NTP_MAXDPOLL;
else {
peer->flags |= FLAG_UMAXPOLL;
peer->maxpoll = (u_char)max(maxpoll, NTP_MINPOLL);
}
if (peer->minpoll > peer->maxpoll)
peer->minpoll = peer->maxpoll;
if (peer->dstadr)
DPRINTF(3, ("newpeer: using fd %d and our addr %s\n",
peer->dstadr->fd, stoa(&peer->dstadr->sin)));
else
DPRINTF(3, ("newpeer: local interface currently not bound\n"));
if ((cast_flags & MDF_BCAST) && peer->dstadr)
enable_broadcast(peer->dstadr, srcadr);
if ((cast_flags & MDF_MCAST) && peer->dstadr)
enable_multicast_if(peer->dstadr, srcadr);
#ifdef OPENSSL
if (key > NTP_MAXKEY)
peer->flags |= FLAG_SKEY;
#endif
peer->cast_flags = cast_flags;
peer->ttl = (u_char)ttl;
peer->keyid = key;
peer->precision = sys_precision;
peer->hpoll = peer->minpoll;
if (cast_flags & MDF_ACAST)
peer_clear(peer, "ACST");
else if (cast_flags & MDF_MCAST)
peer_clear(peer, "MCST");
else if (cast_flags & MDF_BCAST)
peer_clear(peer, "BCST");
else
peer_clear(peer, "INIT");
if (mode_ntpdate)
peer_ntpdate++;
peer->timereset = current_time;
peer->timereachable = current_time;
peer->timereceived = current_time;
#ifdef REFCLOCK
if (ISREFCLOCKADR(&peer->srcadr)) {
if (maxpoll == 0)
peer->maxpoll = peer->minpoll;
if (!refclock_newpeer(peer)) {
set_peerdstadr(peer, NULL);
LINK_SLIST(peer_free, peer, next);
peer_free_count++;
return (NULL);
}
}
#endif
hash = NTP_HASH_ADDR(&peer->srcadr);
LINK_SLIST(peer_hash[hash], peer, next);
peer_hash_count[hash]++;
hash = peer->associd & NTP_HASH_MASK;
LINK_SLIST(assoc_hash[hash], peer, ass_next);
assoc_hash_count[hash]++;
snprintf(tbuf, sizeof(tbuf), "assoc %d", peer->associd);
report_event(PEVNT_MOBIL, peer, tbuf);
DPRINTF(1, ("newpeer: %s->%s mode %d vers %d poll %d %d flags 0x%x 0x%x ttl %d key %08x\n",
peer->dstadr == NULL ? "<null>" : stoa(&peer->dstadr->sin),
stoa(&peer->srcadr), peer->hmode, peer->version,
peer->minpoll, peer->maxpoll, peer->flags, peer->cast_flags,
peer->ttl, peer->keyid));
return (peer);
}
void
peer_clr_stats(void)
{
findpeer_calls = 0;
assocpeer_calls = 0;
peer_allocations = 0;
peer_demobilizations = 0;
peer_timereset = current_time;
}
void
peer_reset(
struct peer *peer
)
{
if (peer == NULL)
return;
peer->timereset = current_time;
peer->sent = 0;
peer->received = 0;
peer->processed = 0;
peer->badauth = 0;
peer->bogusorg = 0;
peer->oldpkt = 0;
peer->seldisptoolarge = 0;
peer->selbroken = 0;
}
void
peer_all_reset(void)
{
struct peer *peer;
int hash;
for (hash = 0; hash < NTP_HASH_SIZE; hash++)
for (peer = peer_hash[hash]; peer != 0; peer = peer->next)
peer_reset(peer);
}
struct peer *
findmanycastpeer(
struct recvbuf *rbufp
)
{
register struct peer *peer;
struct pkt *pkt;
l_fp p_org;
int i;
pkt = &rbufp->recv_pkt;
for (i = 0; i < NTP_HASH_SIZE; i++) {
if (peer_hash_count[i] == 0)
continue;
for (peer = peer_hash[i]; peer != 0; peer =
peer->next) {
if (peer->cast_flags & MDF_ACAST) {
NTOHL_FP(&pkt->org, &p_org);
if (L_ISEQU(&p_org, &peer->aorg))
return (peer);
}
}
}
return (NULL);
}
#include <dns_util.h>
u_long get_dns_flags(
char *dns_name,
struct peer* peer)
{
dns_handle_t handle;
dns_reply_t *reply;
dns_TXT_record_t *txt;
uint16_t klass, type;
handle = dns_open(NULL);
if (!handle) {
return 0;
}
dns_class_number("IN", &klass);
dns_type_number("TXT", &type);
reply = dns_lookup(handle, dns_name, klass, type);
if (reply) {
int a;
#if DEBUG
if (debug > 1) {
dns_print_reply(reply, stdout, 0xffff);
printf("status: %d\n", reply->status);
printf("answer count: %d\n", reply->header->ancount);
}
#endif
for (a = 0; a < reply->header->ancount; a++) {
if ((reply->answer[a]->dnstype != type) ||
(reply->answer[a]->dnsclass != klass))
continue;
txt = reply->answer[a]->data.TXT;
if (txt) {
int s;
u_int old_flags = peer->flags & (FLAG_BURST | FLAG_IBURST);
u_int new_flags = 0;
u_char minpoll = NTP_MINDPOLL;
u_char maxpoll = NTP_MAXDPOLL;
for (s = 0; s < txt->string_count; s++) {
#if DEBUG
if (debug > 1)
printf("%d: %s\n", s, txt->strings[s]);
#endif
if (0 == strncmp(txt->strings[s], "ntp ", 4)) {
char *next = txt->strings[s]+4;
char *arg, *p;
long arg_val;
if (peer->dns_ttl != reply->answer[a]->ttl) {
peer->dns_ttl = reply->answer[a]->ttl;
msyslog(LOG_INFO, "DNS %s ttl %d",
dns_name, peer->dns_ttl);
}
if (peer->dns_ttl != 0)
peer->dns_update = current_time + peer->dns_ttl;
while (NULL != (p = strsep(&next, " \t"))) {
switch (*p) {
case 'b':
if (strcmp(p, "burst") == 0) {
new_flags |= FLAG_BURST;
} else {
msyslog(LOG_WARNING, "DNS %s unknown configuration [%s]", dns_name, p);
}
break;
case 'i':
if (0 == strcmp(p, "iburst")) {
new_flags |= FLAG_IBURST;
} else {
msyslog(LOG_WARNING, "DNS %s unknown configuration [%s]", dns_name, p);
}
break;
case 'm':
if ((0 == (strcmp(p, "minpoll"))) || (0 == strcmp(p, "maxpoll"))) {
if (next) {
arg = strsep(&next, " \t");
arg_val = strtol(arg, NULL, 10);
if (p[1] == 'i') {
minpoll = (u_char)max(NTP_MINPOLL, arg_val);
} else {
maxpoll = (u_char)min(NTP_MAXPOLL, arg_val);
}
} else {
msyslog(LOG_WARNING, "DNS %s option %s missing numeric argument", dns_name, p);
}
} else {
msyslog(LOG_WARNING, "DNS %s unknown configuration [%s]", dns_name, p);
}
break;
default:
msyslog(LOG_WARNING, "DNS %s unknown configuration [%s]", dns_name, p);
break;
}
}
}
}
if (0 == (peer->flags & FLAG_UMINPOLL) && (minpoll != peer->minpoll)) {
peer->minpoll = minpoll;
msyslog(LOG_INFO, "DNS %s minpoll %d",
dns_name, peer->minpoll);
}
if (0 == (peer->flags & FLAG_UMAXPOLL) && (maxpoll != peer->maxpoll)) {
peer->maxpoll = maxpoll;
msyslog(LOG_INFO, "DNS %s maxpoll %d",
dns_name, peer->maxpoll);
}
if (new_flags != old_flags) {
u_int changes;
if (peer->flags & FLAG_UIBURST)
new_flags |= FLAG_IBURST;
if (peer->flags & FLAG_IBURST)
new_flags |= FLAG_IBURST;
changes = new_flags ^ old_flags;
if (changes & FLAG_IBURST) {
if (new_flags & FLAG_IBURST)
peer->flags |= FLAG_IBURST;
else
peer->flags &= ~FLAG_IBURST;
msyslog(LOG_INFO, "DNS %s %ciburst",
dns_name,
(new_flags & FLAG_IBURST) ? '+' : '-');
}
if (changes & FLAG_BURST) {
if (new_flags & FLAG_BURST)
peer->flags |= FLAG_BURST;
else
peer->flags &= ~FLAG_BURST;
msyslog(LOG_INFO, "DNS %s %cburst",
dns_name,
(new_flags & FLAG_BURST) ? '+' : '-');
}
}
}
}
dns_free_reply(reply);
}
dns_free(handle);
return peer->dns_update;
}