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