src/ntpd.c

   1 /**
   2  * collectd - src/ntpd.c
   3  * Copyright (C) 2006-2007  Florian octo Forster
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Florian octo Forster <octo at verplant.org>
  20  **/
  21
  22 #include "collectd.h"
  23 #include "common.h"
  24 #include "plugin.h"
  25 #include "configfile.h"
  26
  27 #if HAVE_STDINT_H
  28 # include <stdint.h>
  29 #endif
  30 #if HAVE_NETDB_H
  31 # include <netdb.h>
  32 #endif
  33 #if HAVE_SYS_SOCKET_H
  34 # include <sys/socket.h>
  35 #endif
  36 #if HAVE_NETINET_IN_H
  37 # include <netinet/in.h>
  38 #endif
  39 #if HAVE_ARPA_INET_H
  40 # include <arpa/inet.h> /* inet_ntoa */
  41 #endif
  42 #if HAVE_NETINET_TCP_H
  43 # include <netinet/tcp.h>
  44 #endif
  45 #if HAVE_POLL_H
  46 # include <poll.h>
  47 #endif
  48
  49 static const char *config_keys[] =
  50 {
  51         "Host",
  52         "Port",
  53         "ReverseLookups"
  54 };
  55 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
  56
  57 static int do_reverse_lookups = 1;
  58
  59 # define NTPD_DEFAULT_HOST "localhost"
  60 # define NTPD_DEFAULT_PORT "123"
  61 static int   sock_descr = -1;
  62 static char *ntpd_host = NULL;
  63 static char  ntpd_port[16];
  64
  65 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
  66  * The following definitions were copied from the NTPd distribution  *
  67  * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
  68 #define MAXFILENAME 128
  69 #define MAXSEQ  127
  70 #define MODE_PRIVATE 7
  71 #define NTP_OLDVERSION ((uint8_t) 1) /* oldest credible version */
  72 #define IMPL_XNTPD 3
  73 #define FP_FRAC 65536.0
  74
  75 #define REFCLOCK_ADDR 0x7f7f0000 /* 127.127.0.0 */
  76 #define REFCLOCK_MASK 0xffff0000 /* 255.255.0.0 */
  77
  78 /* This structure is missing the message authentication code, since collectd
  79  * doesn't use it. */
  80 struct req_pkt
  81 {
  82         uint8_t  rm_vn_mode;
  83         uint8_t  auth_seq;
  84         uint8_t  implementation;                /* implementation number */
  85         uint8_t  request;                       /* request number */
  86         uint16_t err_nitems;            /* error code/number of data items */
  87         uint16_t mbz_itemsize;          /* item size */
  88         char     data[MAXFILENAME + 48];        /* data area [32 prev](176 byte max) */
  89                                         /* struct conf_peer must fit */
  90 };
  91 #define REQ_LEN_NOMAC (sizeof(struct req_pkt))
  92
  93 /*
  94  * A response packet.  The length here is variable, this is a
  95  * maximally sized one.  Note that this implementation doesn't
  96  * authenticate responses.
  97  */
  98 #define RESP_HEADER_SIZE        (8)
  99 #define RESP_DATA_SIZE          (500)
 100
 101 struct resp_pkt
 102 {
 103         uint8_t  rm_vn_mode;           /* response, more, version, mode */
 104         uint8_t  auth_seq;             /* key, sequence number */
 105         uint8_t  implementation;       /* implementation number */
 106         uint8_t  request;              /* request number */
 107         uint16_t err_nitems;           /* error code/number of data items */
 108         uint16_t mbz_itemsize;         /* item size */
 109         char     data[RESP_DATA_SIZE]; /* data area */
 110 };
 111
 112 /*
 113  * Bit setting macros for multifield items.
 114  */
 115 #define RESP_BIT        0x80
 116 #define MORE_BIT        0x40
 117
 118 #define ISRESPONSE(rm_vn_mode)  (((rm_vn_mode)&RESP_BIT)!=0)
 119 #define ISMORE(rm_vn_mode)      (((rm_vn_mode)&MORE_BIT)!=0)
 120 #define INFO_VERSION(rm_vn_mode) ((uint8_t)(((rm_vn_mode)>>3)&0x7))
 121 #define INFO_MODE(rm_vn_mode)   ((rm_vn_mode)&0x7)
 122
 123 #define RM_VN_MODE(resp, more, version)         \
 124                                 ((uint8_t)(((resp)?RESP_BIT:0)\
 125                                 |((more)?MORE_BIT:0)\
 126                                 |((version?version:(NTP_OLDVERSION+1))<<3)\
 127                                 |(MODE_PRIVATE)))
 128
 129 #define INFO_IS_AUTH(auth_seq)  (((auth_seq) & 0x80) != 0)
 130 #define INFO_SEQ(auth_seq)      ((auth_seq)&0x7f)
 131 #define AUTH_SEQ(auth, seq)     ((uint8_t)((((auth)!=0)?0x80:0)|((seq)&0x7f)))
 132
 133 #define INFO_ERR(err_nitems)    ((uint16_t)((ntohs(err_nitems)>>12)&0xf))
 134 #define INFO_NITEMS(err_nitems) ((uint16_t)(ntohs(err_nitems)&0xfff))
 135 #define ERR_NITEMS(err, nitems) (htons((uint16_t)((((uint16_t)(err)<<12)&0xf000)\
 136                                 |((uint16_t)(nitems)&0xfff))))
 137
 138 #define INFO_MBZ(mbz_itemsize)  ((ntohs(mbz_itemsize)>>12)&0xf)
 139 #define INFO_ITEMSIZE(mbz_itemsize)     ((uint16_t)(ntohs(mbz_itemsize)&0xfff))
 140 #define MBZ_ITEMSIZE(itemsize)  (htons((uint16_t)(itemsize)))
 141
 142 /* negate a long float type */
 143 #define M_NEG(v_i, v_f) \
 144         do { \
 145                 if ((v_f) == 0) \
 146                 (v_i) = -((uint32_t)(v_i)); \
 147                 else { \
 148                         (v_f) = -((uint32_t)(v_f)); \
 149                         (v_i) = ~(v_i); \
 150                 } \
 151         } while(0)
 152 /* l_fp to double */
 153 #define M_LFPTOD(r_i, r_uf, d) \
 154         do { \
 155                 register int32_t  i; \
 156                 register uint32_t f; \
 157                 \
 158                 i = (r_i); \
 159                 f = (r_uf); \
 160                 if (i < 0) { \
 161                         M_NEG(i, f); \
 162                         (d) = -((double) i + ((double) f) / 4294967296.0); \
 163                 } else { \
 164                         (d) = (double) i + ((double) f) / 4294967296.0; \
 165                 } \
 166         } while (0)
 167
 168 #define REQ_PEER_LIST_SUM 1
 169 struct info_peer_summary
 170 {
 171         uint32_t dstadr;         /* local address (zero for undetermined) */
 172         uint32_t srcadr;         /* source address */
 173         uint16_t srcport;        /* source port */
 174         uint8_t stratum;         /* stratum of peer */
 175         int8_t hpoll;            /* host polling interval */
 176         int8_t ppoll;            /* peer polling interval */
 177         uint8_t reach;           /* reachability register */
 178         uint8_t flags;           /* flags, from above */
 179         uint8_t hmode;           /* peer mode */
 180         int32_t  delay;          /* peer.estdelay; s_fp */
 181         int32_t  offset_int;     /* peer.estoffset; integral part */
 182         int32_t  offset_frc;     /* peer.estoffset; fractional part */
 183         uint32_t dispersion;     /* peer.estdisp; u_fp */
 184         uint32_t v6_flag;        /* is this v6 or not */
 185         uint32_t unused1;        /* (unused) padding for dstadr6 */
 186         struct in6_addr dstadr6; /* local address (v6) */
 187         struct in6_addr srcadr6; /* source address (v6) */
 188 };
 189
 190 #define REQ_SYS_INFO 4
 191 struct info_sys
 192 {
 193         uint32_t peer;           /* system peer address (v4) */
 194         uint8_t  peer_mode;      /* mode we are syncing to peer in */
 195         uint8_t  leap;           /* system leap bits */
 196         uint8_t  stratum;        /* our stratum */
 197         int8_t   precision;      /* local clock precision */
 198         int32_t  rootdelay;      /* distance from sync source */
 199         uint32_t rootdispersion; /* dispersion from sync source */
 200         uint32_t refid;          /* reference ID of sync source */
 201         uint64_t reftime;        /* system reference time */
 202         uint32_t poll;           /* system poll interval */
 203         uint8_t  flags;          /* system flags */
 204         uint8_t  unused1;        /* unused */
 205         uint8_t  unused2;        /* unused */
 206         uint8_t  unused3;        /* unused */
 207         int32_t  bdelay;         /* default broadcast offset */
 208         int32_t  frequency;      /* frequency residual (scaled ppm)  */
 209         uint64_t authdelay;      /* default authentication delay */
 210         uint32_t stability;      /* clock stability (scaled ppm) */
 211         int32_t  v6_flag;        /* is this v6 or not */
 212         int32_t  unused4;        /* unused, padding for peer6 */
 213         struct in6_addr peer6;   /* system peer address (v6) */
 214 };
 215
 216 #define REQ_GET_KERNEL 38
 217 struct info_kernel
 218 {
 219         int32_t  offset;
 220         int32_t  freq;
 221         int32_t  maxerror;
 222         int32_t  esterror;
 223         uint16_t status;
 224         uint16_t shift;
 225         int32_t  constant;
 226         int32_t  precision;
 227         int32_t  tolerance;
 228         /* pps stuff */
 229         int32_t  ppsfreq;
 230         int32_t  jitter;
 231         int32_t  stabil;
 232         int32_t  jitcnt;
 233         int32_t  calcnt;
 234         int32_t  errcnt;
 235         int32_t  stbcnt;
 236 };
 237
 238 /* List of reference clock names */
 239 static char *refclock_names[] =
 240 {
 241         "UNKNOWN",    "LOCAL",        "GPS_TRAK",   "WWV_PST",     /*  0- 3 */
 242         "SPECTRACOM", "TRUETIME",     "IRIG_AUDIO", "CHU_AUDIO",   /*  4- 7 */
 243         "GENERIC",    "GPS_MX4200",   "GPS_AS2201", "GPS_ARBITER", /*  8-11 */
 244         "IRIG_TPRO",  "ATOM_LEITCH",  "MSF_EES",    "GPSTM_TRUE",  /* 12-15 */
 245         "GPS_BANC",   "GPS_DATUM",    "ACTS_NIST",  "WWV_HEATH",   /* 16-19 */
 246         "GPS_NMEA",   "GPS_VME",      "PPS",        "ACTS_PTB",    /* 20-23 */
 247         "ACTS_USNO",  "TRUETIME",     "GPS_HP",     "MSF_ARCRON",  /* 24-27 */
 248         "SHM",        "GPS_PALISADE", "GPS_ONCORE", "GPS_JUPITER", /* 28-31 */
 249         "CHRONOLOG",  "DUMBCLOCK",    "ULINK_M320", "PCF",         /* 32-35 */
 250         "WWV_AUDIO",  "GPS_FG",       "HOPF_S",     "HOPF_P",      /* 36-39 */
 251         "JJY",        "TT_IRIG",      "GPS_ZYFER",  "GPS_RIPENCC", /* 40-43 */
 252         "NEOCLK4X"                                                 /* 44    */
 253 };
 254 static int refclock_names_num = STATIC_ARRAY_SIZE (refclock_names);
 255 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 256  * End of the copied stuff..                                         *
 257  * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 258
 259 static int ntpd_config (const char *key, const char *value)
 260 {
 261         if (strcasecmp (key, "Host") == 0)
 262         {
 263                 if (ntpd_host != NULL)
 264                         free (ntpd_host);
 265                 if ((ntpd_host = strdup (value)) == NULL)
 266                         return (1);
 267         }
 268         else if (strcasecmp (key, "Port") == 0)
 269         {
 270                 int port = (int) (atof (value));
 271                 if ((port > 0) && (port <= 65535))
 272                         ssnprintf (ntpd_port, sizeof (ntpd_port),
 273                                         "%i", port);
 274                 else
 275                         sstrncpy (ntpd_port, value, sizeof (ntpd_port));
 276         }
 277         else if (strcasecmp (key, "ReverseLookups") == 0)
 278         {
 279                 if ((strcasecmp (value, "True") == 0)
 280                                 || (strcasecmp (value, "Yes") == 0)
 281                                 || (strcasecmp (value, "On") == 0))
 282                         do_reverse_lookups = 1;
 283                 else
 284                         do_reverse_lookups = 0;
 285         }
 286         else
 287         {
 288                 return (-1);
 289         }
 290
 291         return (0);
 292 }
 293
 294 static void ntpd_submit (char *type, char *type_inst, double value)
 295 {
 296         value_t values[1];
 297         value_list_t vl = VALUE_LIST_INIT;
 298
 299         values[0].gauge = value;
 300
 301         vl.values = values;
 302         vl.values_len = 1;
 303         vl.time = time (NULL);
 304         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
 305         sstrncpy (vl.plugin, "ntpd", sizeof (vl.plugin));
 306         sstrncpy (vl.plugin_instance, "", sizeof (vl.plugin_instance));
 307         sstrncpy (vl.type, type, sizeof (vl.type));
 308         sstrncpy (vl.type_instance, type_inst, sizeof (vl.type_instance));
 309
 310         plugin_dispatch_values (&vl);
 311 }
 312
 313 static int ntpd_connect (void)
 314 {
 315         char *host;
 316         char *port;
 317
 318         struct addrinfo  ai_hints;
 319         struct addrinfo *ai_list;
 320         struct addrinfo *ai_ptr;
 321         int              status;
 322
 323         if (sock_descr >= 0)
 324                 return (sock_descr);
 325
 326         DEBUG ("Opening a new socket");
 327
 328         host = ntpd_host;
 329         if (host == NULL)
 330                 host = NTPD_DEFAULT_HOST;
 331
 332         port = ntpd_port;
 333         if (strlen (port) == 0)
 334                 port = NTPD_DEFAULT_PORT;
 335
 336         memset (&ai_hints, '\0', sizeof (ai_hints));
 337         ai_hints.ai_flags    = 0;
 338 #ifdef AI_ADDRCONFIG
 339         ai_hints.ai_flags   |= AI_ADDRCONFIG;
 340 #endif
 341         ai_hints.ai_family   = PF_UNSPEC;
 342         ai_hints.ai_socktype = SOCK_DGRAM;
 343         ai_hints.ai_protocol = IPPROTO_UDP;
 344
 345         if ((status = getaddrinfo (host, port, &ai_hints, &ai_list)) != 0)
 346         {
 347                 char errbuf[1024];
 348                 ERROR ("ntpd plugin: getaddrinfo (%s, %s): %s",
 349                                 host, port,
 350                                 (status == EAI_SYSTEM)
 351                                 ? sstrerror (errno, errbuf, sizeof (errbuf))
 352                                 : gai_strerror (status));
 353                 return (-1);
 354         }
 355
 356         for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
 357         {
 358                 /* create our socket descriptor */
 359                 if ((sock_descr = socket (ai_ptr->ai_family,
 360                                                 ai_ptr->ai_socktype,
 361                                                 ai_ptr->ai_protocol)) < 0)
 362                         continue;
 363
 364                 /* connect to the ntpd */
 365                 if (connect (sock_descr, ai_ptr->ai_addr, ai_ptr->ai_addrlen))
 366                 {
 367                         close (sock_descr);
 368                         sock_descr = -1;
 369                         continue;
 370                 }
 371
 372                 break;
 373         }
 374
 375         freeaddrinfo (ai_list);
 376
 377         if (sock_descr < 0)
 378         {
 379                 ERROR ("ntpd plugin: Unable to connect to server.");
 380         }
 381
 382         return (sock_descr);
 383 }
 384
 385 /* For a description of the arguments see `ntpd_do_query' below. */
 386 static int ntpd_receive_response (int *res_items, int *res_size,
 387                 char **res_data, int res_item_size)
 388 {
 389         int              sd;
 390         struct pollfd    poll_s;
 391         struct resp_pkt  res;
 392         int              status;
 393         int              done;
 394         int              i;
 395
 396         char            *items;
 397         size_t           items_num;
 398
 399         struct timeval   time_end;
 400         struct timeval   time_now;
 401         int              timeout;
 402
 403         int              pkt_item_num;        /* items in this packet */
 404         int              pkt_item_len;        /* size of the items in this packet */
 405         int              pkt_sequence;
 406         char             pkt_recvd[MAXSEQ+1]; /* sequence numbers that have been received */
 407         int              pkt_recvd_num;       /* number of packets that have been received */
 408         int              pkt_lastseq;         /* the last sequence number */
 409         ssize_t          pkt_padding;         /* Padding in this packet */
 410
 411         if ((sd = ntpd_connect ()) < 0)
 412                 return (-1);
 413
 414         items = NULL;
 415         items_num = 0;
 416
 417         memset (pkt_recvd, '\0', sizeof (pkt_recvd));
 418         pkt_recvd_num = 0;
 419         pkt_lastseq   = -1;
 420
 421         *res_items = 0;
 422         *res_size  = 0;
 423         *res_data  = NULL;
 424
 425         if (gettimeofday (&time_end, NULL) < 0)
 426         {
 427                 char errbuf[1024];
 428                 ERROR ("ntpd plugin: gettimeofday failed: %s",
 429                                 sstrerror (errno, errbuf, sizeof (errbuf)));
 430                 return (-1);
 431         }
 432         time_end.tv_sec++; /* wait for a most one second */
 433
 434         done = 0;
 435         while (done == 0)
 436         {
 437                 struct timeval time_left;
 438
 439                 if (gettimeofday (&time_now, NULL) < 0)
 440                 {
 441                         char errbuf[1024];
 442                         ERROR ("ntpd plugin: gettimeofday failed: %s",
 443                                         sstrerror (errno, errbuf, sizeof (errbuf)));
 444                         return (-1);
 445                 }
 446
 447                 if (timeval_cmp (time_end, time_now, &time_left) <= 0)
 448                         timeout = 0;
 449                 else
 450                         timeout = 1000 * time_left.tv_sec
 451                                 + ((time_left.tv_usec + 500) / 1000);
 452
 453                 /* timeout reached */
 454                 if (timeout <= 0)
 455                         break;
 456
 457                 poll_s.fd      = sd;
 458                 poll_s.events  = POLLIN | POLLPRI;
 459                 poll_s.revents = 0;
 460
 461                 DEBUG ("Polling for %ims", timeout);
 462                 status = poll (&poll_s, 1, timeout);
 463
 464                 if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR)))
 465                         continue;
 466
 467                 if (status < 0)
 468                 {
 469                         char errbuf[1024];
 470                         ERROR ("ntpd plugin: poll failed: %s",
 471                                         sstrerror (errno, errbuf, sizeof (errbuf)));
 472                         return (-1);
 473                 }
 474
 475                 if (status == 0) /* timeout */
 476                 {
 477                         DEBUG ("timeout reached.");
 478                         break;
 479                 }
 480
 481                 memset ((void *) &res, '\0', sizeof (res));
 482                 status = recv (sd, (void *) &res, sizeof (res), 0 /* no flags */);
 483
 484                 if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR)))
 485                         continue;
 486
 487                 if (status < 0)
 488                 {
 489                         char errbuf[1024];
 490                         INFO ("recv(2) failed: %s",
 491                                         sstrerror (errno, errbuf, sizeof (errbuf)));
 492                         DEBUG ("Closing socket #%i", sd);
 493                         close (sd);
 494                         sock_descr = sd = -1;
 495                         return (-1);
 496                 }
 497
 498                 DEBUG ("recv'd %i bytes", status);
 499
 500                 /*
 501                  * Do some sanity checks first
 502                  */
 503                 if (status < RESP_HEADER_SIZE)
 504                 {
 505                         WARNING ("ntpd plugin: Short (%i bytes) packet received",
 506                                         (int) status);
 507                         continue;
 508                 }
 509                 if (INFO_MODE (res.rm_vn_mode) != MODE_PRIVATE)
 510                 {
 511                         NOTICE ("ntpd plugin: Packet received with mode %i",
 512                                         INFO_MODE (res.rm_vn_mode));
 513                         continue;
 514                 }
 515                 if (INFO_IS_AUTH (res.auth_seq))
 516                 {
 517                         NOTICE ("ntpd plugin: Encrypted packet received");
 518                         continue;
 519                 }
 520                 if (!ISRESPONSE (res.rm_vn_mode))
 521                 {
 522                         NOTICE ("ntpd plugin: Received request packet, "
 523                                         "wanted response");
 524                         continue;
 525                 }
 526                 if (INFO_MBZ (res.mbz_itemsize))
 527                 {
 528                         WARNING ("ntpd plugin: Received packet with nonzero "
 529                                         "MBZ field!");
 530                         continue;
 531                 }
 532                 if (res.implementation != IMPL_XNTPD)
 533                 {
 534                         WARNING ("ntpd plugin: Asked for request of type %i, "
 535                                         "got %i", (int) IMPL_XNTPD, (int) res.implementation);
 536                         continue;
 537                 }
 538
 539                 /* Check for error code */
 540                 if (INFO_ERR (res.err_nitems) != 0)
 541                 {
 542                         ERROR ("ntpd plugin: Received error code %i",
 543                                         (int) INFO_ERR(res.err_nitems));
 544                         return ((int) INFO_ERR (res.err_nitems));
 545                 }
 546
 547                 /* extract number of items in this packet and the size of these items */
 548                 pkt_item_num = INFO_NITEMS (res.err_nitems);
 549                 pkt_item_len = INFO_ITEMSIZE (res.mbz_itemsize);
 550                 DEBUG ("pkt_item_num = %i; pkt_item_len = %i;",
 551                                 pkt_item_num, pkt_item_len);
 552
 553                 /* Check if the reported items fit in the packet */
 554                 if ((pkt_item_num * pkt_item_len) > (status - RESP_HEADER_SIZE))
 555                 {
 556                         ERROR ("ntpd plugin: %i items * %i bytes > "
 557                                         "%i bytes - %i bytes header",
 558                                         (int) pkt_item_num, (int) pkt_item_len,
 559                                         (int) status, (int) RESP_HEADER_SIZE);
 560                         continue;
 561                 }
 562
 563                 if (pkt_item_len > res_item_size)
 564                 {
 565                         ERROR ("ntpd plugin: (pkt_item_len = %i) "
 566                                         ">= (res_item_size = %i)",
 567                                         pkt_item_len, res_item_size);
 568                         continue;
 569                 }
 570
 571                 /* If this is the first packet (time wise, not sequence wise),
 572                  * set `res_size'. If it's not the first packet check if the
 573                  * items have the same size. Discard invalid packets. */
 574                 if (items_num == 0) /* first packet */
 575                 {
 576                         DEBUG ("*res_size = %i", pkt_item_len);
 577                         *res_size = pkt_item_len;
 578                 }
 579                 else if (*res_size != pkt_item_len)
 580                 {
 581                         DEBUG ("Error: *res_size = %i; pkt_item_len = %i;",
 582                                         *res_size, pkt_item_len);
 583                         ERROR ("Item sizes differ.");
 584                         continue;
 585                 }
 586
 587                 /*
 588                  * Because the items in the packet may be smaller than the
 589                  * items requested, the following holds true:
 590                  */
 591                 assert ((*res_size == pkt_item_len)
 592                                 && (pkt_item_len <= res_item_size));
 593
 594                 /* Calculate the padding. No idea why there might be any padding.. */
 595                 pkt_padding = 0;
 596                 if (pkt_item_len < res_item_size)
 597                         pkt_padding = res_item_size - pkt_item_len;
 598                 DEBUG ("res_item_size = %i; pkt_padding = %zi;",
 599                                 res_item_size, pkt_padding);
 600
 601                 /* Extract the sequence number */
 602                 pkt_sequence = INFO_SEQ (res.auth_seq);
 603                 if ((pkt_sequence < 0) || (pkt_sequence > MAXSEQ))
 604                 {
 605                         ERROR ("ntpd plugin: Received packet with sequence %i",
 606                                         pkt_sequence);
 607                         continue;
 608                 }
 609
 610                 /* Check if this sequence has been received before. If so, discard it. */
 611                 if (pkt_recvd[pkt_sequence] != '\0')
 612                 {
 613                         NOTICE ("ntpd plugin: Sequence %i received twice",
 614                                         pkt_sequence);
 615                         continue;
 616                 }
 617
 618                 /* If `pkt_lastseq != -1' another packet without `more bit' has
 619                  * been received. */
 620                 if (!ISMORE (res.rm_vn_mode))
 621                 {
 622                         if (pkt_lastseq != -1)
 623                         {
 624                                 ERROR ("ntpd plugin: Two packets which both "
 625                                                 "claim to be the last one in the "
 626                                                 "sequence have been received.");
 627                                 continue;
 628                         }
 629                         pkt_lastseq = pkt_sequence;
 630                         DEBUG ("Last sequence = %i;", pkt_lastseq);
 631                 }
 632
 633                 /*
 634                  * Enough with the checks. Copy the data now.
 635                  * We start by allocating some more memory.
 636                  */
 637                 DEBUG ("realloc (%p, %zu)", (void *) *res_data,
 638                                 (items_num + pkt_item_num) * res_item_size);
 639                 items = realloc ((void *) *res_data,
 640                                 (items_num + pkt_item_num) * res_item_size);
 641                 if (items == NULL)
 642                 {
 643                         items = *res_data;
 644                         ERROR ("ntpd plugin: realloc failed.");
 645                         continue;
 646                 }
 647                 items_num += pkt_item_num;
 648                 *res_data = items;
 649
 650                 for (i = 0; i < pkt_item_num; i++)
 651                 {
 652                         /* dst: There are already `*res_items' items with
 653                          *      res_item_size bytes each in in `*res_data'. Set
 654                          *      dst to the first byte after that. */
 655                         void *dst = (void *) (*res_data + ((*res_items) * res_item_size));
 656                         /* src: We use `pkt_item_len' to calculate the offset
 657                          *      from the beginning of the packet, because the
 658                          *      items in the packet may be smaller than the
 659                          *      items that were requested. We skip `i' such
 660                          *      items. */
 661                         void *src = (void *) (((char *) res.data) + (i * pkt_item_len));
 662
 663                         /* Set the padding to zeros */
 664                         if (pkt_padding != 0)
 665                                 memset (dst, '\0', res_item_size);
 666                         memcpy (dst, src, (size_t) pkt_item_len);
 667
 668                         /* Increment `*res_items' by one, so `dst' will end up
 669                          * one further in the next round. */
 670                         (*res_items)++;
 671                 } /* for (pkt_item_num) */
 672
 673                 pkt_recvd[pkt_sequence] = (char) 1;
 674                 pkt_recvd_num++;
 675
 676                 if ((pkt_recvd_num - 1) == pkt_lastseq)
 677                         done = 1;
 678         } /* while (done == 0) */
 679
 680         return (0);
 681 } /* int ntpd_receive_response */
 682
 683 /* For a description of the arguments see `ntpd_do_query' below. */
 684 static int ntpd_send_request (int req_code, int req_items, int req_size, char *req_data)
 685 {
 686         int             sd;
 687         struct req_pkt  req;
 688         size_t          req_data_len;
 689         int             status;
 690
 691         assert (req_items >= 0);
 692         assert (req_size  >= 0);
 693
 694         if ((sd = ntpd_connect ()) < 0)
 695                 return (-1);
 696
 697         memset ((void *) &req, '\0', sizeof (req));
 698         req.rm_vn_mode = RM_VN_MODE(0, 0, 0);
 699         req.auth_seq   = AUTH_SEQ (0, 0);
 700         req.implementation = IMPL_XNTPD;
 701         req.request = (unsigned char) req_code;
 702
 703         req_data_len = (size_t) (req_items * req_size);
 704
 705         assert (((req_data != NULL) && (req_data_len > 0))
 706                         || ((req_data == NULL) && (req_items == 0) && (req_size == 0)));
 707
 708         req.err_nitems   = ERR_NITEMS (0, req_items);
 709         req.mbz_itemsize = MBZ_ITEMSIZE (req_size);
 710
 711         if (req_data != NULL)
 712                 memcpy ((void *) req.data, (const void *) req_data, req_data_len);
 713
 714         DEBUG ("req_items = %i; req_size = %i; req_data = %p;",
 715                         req_items, req_size, (void *) req_data);
 716
 717         status = swrite (sd, (const char *) &req, REQ_LEN_NOMAC);
 718         if (status < 0)
 719         {
 720                 DEBUG ("`swrite' failed. Closing socket #%i", sd);
 721                 close (sd);
 722                 sock_descr = sd = -1;
 723                 return (status);
 724         }
 725
 726         return (0);
 727 }
 728
 729 /*
 730  * ntpd_do_query:
 731  *
 732  * req_code:      Type of request packet
 733  * req_items:     Numver of items in the request
 734  * req_size:      Size of one item in the request
 735  * req_data:      Data of the request packet
 736  * res_items:     Pointer to where the number returned items will be stored.
 737  * res_size:      Pointer to where the size of one returned item will be stored.
 738  * res_data:      This is where a pointer to the (allocated) data will be stored.
 739  * res_item_size: Size of one returned item. (used to calculate padding)
 740  *
 741  * returns:       zero upon success, non-zero otherwise.
 742  */
 743 static int ntpd_do_query (int req_code, int req_items, int req_size, char *req_data,
 744                 int *res_items, int *res_size, char **res_data, int res_item_size)
 745 {
 746         int status;
 747
 748         status = ntpd_send_request (req_code, req_items, req_size, req_data);
 749         if (status != 0)
 750                 return (status);
 751
 752         status = ntpd_receive_response (res_items, res_size, res_data,
 753                         res_item_size);
 754         return (status);
 755 }
 756
 757 static double ntpd_read_fp (int32_t val_int)
 758 {
 759         double val_double;
 760
 761         val_int = ntohl (val_int);
 762         val_double = ((double) val_int) / FP_FRAC;
 763
 764         return (val_double);
 765 }
 766
 767 static int ntpd_read (void)
 768 {
 769         struct info_kernel *ik;
 770         int                 ik_num;
 771         int                 ik_size;
 772
 773         struct info_peer_summary *ps;
 774         int                       ps_num;
 775         int                       ps_size;
 776
 777         int status;
 778         int i;
 779
 780         ik      = NULL;
 781         ik_num  = 0;
 782         ik_size = 0;
 783
 784         status = ntpd_do_query (REQ_GET_KERNEL,
 785                         0, 0, NULL, /* request data */
 786                         &ik_num, &ik_size, (char **) ((void *) &ik), /* response data */
 787                         sizeof (struct info_kernel));
 788
 789         if (status != 0)
 790         {
 791                 DEBUG ("ntpd_do_query failed with status %i", status);
 792                 return (-1);
 793         }
 794         if ((ik == NULL) || (ik_num == 0) || (ik_size == 0))
 795         {
 796                 DEBUG ("ntpd_do_query returned: ik = %p; ik_num = %i; ik_size = %i;",
 797                                 (void *) ik, ik_num, ik_size);
 798                 return (-1);
 799         }
 800
 801         /* kerninfo -> estimated error */
 802
 803         DEBUG ("info_kernel:\n"
 804                         "  pll offset    = %.8f\n"
 805                         "  pll frequency = %.8f\n" /* drift compensation */
 806                         "  est error     = %.8f\n",
 807                         ntpd_read_fp (ik->offset),
 808                         ntpd_read_fp (ik->freq),
 809                         ntpd_read_fp (ik->esterror));
 810
 811         ntpd_submit ("frequency_offset", "loop",  ntpd_read_fp (ik->freq));
 812         ntpd_submit ("time_offset",      "loop",  ntpd_read_fp (ik->offset));
 813         ntpd_submit ("time_offset",      "error", ntpd_read_fp (ik->esterror));
 814
 815         free (ik);
 816         ik = NULL;
 817
 818         status = ntpd_do_query (REQ_PEER_LIST_SUM,
 819                         0, 0, NULL, /* request data */
 820                         &ps_num, &ps_size, (char **) ((void *) &ps), /* response data */
 821                         sizeof (struct info_peer_summary));
 822         if (status != 0)
 823         {
 824                 DEBUG ("ntpd_do_query failed with status %i", status);
 825                 return (-1);
 826         }
 827         if ((ps == NULL) || (ps_num == 0) || (ps_size == 0))
 828         {
 829                 DEBUG ("ntpd_do_query returned: ps = %p; ps_num = %i; ps_size = %i;",
 830                                 (void *) ps, ps_num, ps_size);
 831                 return (-1);
 832         }
 833
 834         for (i = 0; i < ps_num; i++)
 835         {
 836                 struct info_peer_summary *ptr;
 837                 double offset;
 838
 839                 char peername[NI_MAXHOST];
 840                 int refclock_id;
 841
 842                 ptr = ps + i;
 843                 refclock_id = 0;
 844
 845                 /* Convert the `long floating point' offset value to double */
 846                 M_LFPTOD (ntohl (ptr->offset_int), ntohl (ptr->offset_frc), offset);
 847
 848                 /* Special IP addresses for hardware clocks and stuff.. */
 849                 if (!ptr->v6_flag
 850                                 && ((ntohl (ptr->srcadr) & REFCLOCK_MASK)
 851                                         == REFCLOCK_ADDR))
 852                 {
 853                         struct in_addr  addr_obj;
 854                         char *addr_str;
 855
 856                         refclock_id = (ntohl (ptr->srcadr) >> 8) & 0x000000FF;
 857
 858                         if (refclock_id < refclock_names_num)
 859                         {
 860                                 sstrncpy (peername, refclock_names[refclock_id],
 861                                                 sizeof (peername));
 862                         }
 863                         else
 864                         {
 865                                 memset ((void *) &addr_obj, '\0', sizeof (addr_obj));
 866                                 addr_obj.s_addr = ptr->srcadr;
 867                                 addr_str = inet_ntoa (addr_obj);
 868
 869                                 sstrncpy (peername, addr_str, sizeof (peername));
 870                         }
 871                 }
 872                 else /* Normal network host. */
 873                 {
 874                         struct sockaddr_storage sa;
 875                         socklen_t sa_len;
 876                         int flags = 0;
 877
 878                         memset (&sa, '\0', sizeof (sa));
 879
 880                         if (ptr->v6_flag)
 881                         {
 882                                 struct sockaddr_in6 *sa_ptr;
 883                                 sa_ptr = (struct sockaddr_in6 *) &sa;
 884
 885                                 sa_ptr->sin6_family = AF_INET6;
 886                                 sa_ptr->sin6_port = htons (123);
 887                                 memcpy (&sa_ptr->sin6_addr, &ptr->srcadr6,
 888                                                 sizeof (struct in6_addr));
 889                                 sa_len = sizeof (struct sockaddr_in6);
 890                         }
 891                         else
 892                         {
 893                                 struct sockaddr_in *sa_ptr;
 894                                 sa_ptr = (struct sockaddr_in *) &sa;
 895
 896                                 sa_ptr->sin_family = AF_INET;
 897                                 sa_ptr->sin_port = htons (123);
 898                                 memcpy (&sa_ptr->sin_addr, &ptr->srcadr,
 899                                                 sizeof (struct in_addr));
 900                                 sa_len = sizeof (struct sockaddr_in);
 901                         }
 902
 903                         if (do_reverse_lookups == 0)
 904                                 flags |= NI_NUMERICHOST;
 905
 906                         status = getnameinfo ((const struct sockaddr *) &sa,
 907                                         sa_len,
 908                                         peername, sizeof (peername),
 909                                         NULL, 0, /* No port name */
 910                                         flags);
 911                         if (status != 0)
 912                         {
 913                                 char errbuf[1024];
 914                                 ERROR ("ntpd plugin: getnameinfo failed: %s",
 915                                                 (status == EAI_SYSTEM)
 916                                                 ? sstrerror (errno, errbuf, sizeof (errbuf))
 917                                                 : gai_strerror (status));
 918                                 continue;
 919                         }
 920                 }
 921
 922                 DEBUG ("peer %i:\n"
 923                                 "  peername   = %s\n"
 924                                 "  srcadr     = 0x%08x\n"
 925                                 "  delay      = %f\n"
 926                                 "  offset_int = %i\n"
 927                                 "  offset_frc = %i\n"
 928                                 "  offset     = %f\n"
 929                                 "  dispersion = %f\n",
 930                                 i,
 931                                 peername,
 932                                 ntohl (ptr->srcadr),
 933                                 ntpd_read_fp (ptr->delay),
 934                                 ntohl (ptr->offset_int),
 935                                 ntohl (ptr->offset_frc),
 936                                 offset,
 937                                 ntpd_read_fp (ptr->dispersion));
 938
 939                 if (refclock_id != 1) /* not the system clock (offset will always be zero.. */
 940                         ntpd_submit ("time_offset", peername, offset);
 941                 ntpd_submit ("time_dispersion", peername, ntpd_read_fp (ptr->dispersion));
 942                 if (refclock_id == 0) /* not a reference clock */
 943                         ntpd_submit ("delay", peername, ntpd_read_fp (ptr->delay));
 944         }
 945
 946         free (ps);
 947         ps = NULL;
 948
 949         return (0);
 950 } /* int ntpd_read */
 951
 952 void module_register (void)
 953 {
 954         plugin_register_config ("ntpd", ntpd_config,
 955                         config_keys, config_keys_num);
 956         plugin_register_read ("ntpd", ntpd_read);
 957 } /* void module_register */