hddtemp, ntpd plugin: Allow numeric `Port' arguments.
[collectd.git] / 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    = AI_ADDRCONFIG;
359         ai_hints.ai_family   = PF_UNSPEC;
360         ai_hints.ai_socktype = SOCK_DGRAM;
361         ai_hints.ai_protocol = IPPROTO_UDP;
362
363         if ((status = getaddrinfo (host, port, &ai_hints, &ai_list)) != 0)
364         {
365                 char errbuf[1024];
366                 ERROR ("ntpd plugin: getaddrinfo (%s, %s): %s",
367                                 host, port,
368                                 (status == EAI_SYSTEM)
369                                 ? sstrerror (errno, errbuf, sizeof (errbuf))
370                                 : gai_strerror (status));
371                 return (-1);
372         }
373
374         for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
375         {
376                 /* create our socket descriptor */
377                 if ((sock_descr = socket (ai_ptr->ai_family,
378                                                 ai_ptr->ai_socktype,
379                                                 ai_ptr->ai_protocol)) < 0)
380                         continue;
381
382                 /* connect to the ntpd */
383                 if (connect (sock_descr, ai_ptr->ai_addr, ai_ptr->ai_addrlen))
384                 {
385                         close (sock_descr);
386                         sock_descr = -1;
387                         continue;
388                 }
389
390                 break;
391         }
392
393         freeaddrinfo (ai_list);
394
395         if (sock_descr < 0)
396         {
397                 DEBUG ("Unable to connect to server.");
398                 ERROR ("ntpd plugin: Unable to connect to server.");
399         }
400
401         return (sock_descr);
402 }
403
404 /* For a description of the arguments see `ntpd_do_query' below. */
405 static int ntpd_receive_response (int req_code, int *res_items, int *res_size,
406                 char **res_data, int res_item_size)
407 {
408         int              sd;
409         struct pollfd    poll_s;
410         struct resp_pkt  res;
411         int              status;
412         int              done;
413         int              i;
414
415         char            *items;
416         size_t           items_num;
417
418         struct timeval   time_end;
419         struct timeval   time_now;
420         int              timeout;
421
422         int              pkt_item_num;        /* items in this packet */
423         int              pkt_item_len;        /* size of the items in this packet */
424         int              pkt_sequence;
425         char             pkt_recvd[MAXSEQ+1]; /* sequence numbers that have been received */
426         int              pkt_recvd_num;       /* number of packets that have been received */
427         int              pkt_lastseq;         /* the last sequence number */
428         ssize_t          pkt_padding;         /* Padding in this packet */
429
430         if ((sd = ntpd_connect ()) < 0)
431                 return (-1);
432
433         items = NULL;
434         items_num = 0;
435
436         memset (pkt_recvd, '\0', sizeof (pkt_recvd));
437         pkt_recvd_num = 0;
438         pkt_lastseq   = -1;
439
440         *res_items = 0;
441         *res_size  = 0;
442         *res_data  = NULL;
443
444         if (gettimeofday (&time_end, NULL) < 0)
445         {
446                 char errbuf[1024];
447                 ERROR ("ntpd plugin: gettimeofday failed: %s",
448                                 sstrerror (errno, errbuf, sizeof (errbuf)));
449                 return (-1);
450         }
451         time_end.tv_sec++; /* wait for a most one second */
452
453         done = 0;
454         while (done == 0)
455         {
456                 if (gettimeofday (&time_now, NULL) < 0)
457                 {
458                         char errbuf[1024];
459                         ERROR ("ntpd plugin: gettimeofday failed: %s",
460                                         sstrerror (errno, errbuf, sizeof (errbuf)));
461                         return (-1);
462                 }
463
464                 /* timeout reached */
465                 if ((timeout = timeval_sub (&time_end, &time_now)) == 0)
466                         break;
467
468                 poll_s.fd      = sd;
469                 poll_s.events  = POLLIN | POLLPRI;
470                 poll_s.revents = 0;
471                 
472                 DEBUG ("Polling for %ims", timeout);
473                 status = poll (&poll_s, 1, timeout);
474
475                 if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR)))
476                         continue;
477
478                 if (status < 0)
479                 {
480                         char errbuf[1024];
481                         ERROR ("ntpd plugin: poll failed: %s",
482                                         sstrerror (errno, errbuf, sizeof (errbuf)));
483                         return (-1);
484                 }
485
486                 if (status == 0) /* timeout */
487                 {
488                         DEBUG ("timeout reached.");
489                         break;
490                 }
491
492                 memset ((void *) &res, '\0', sizeof (res));
493                 status = recv (sd, (void *) &res, sizeof (res), 0 /* no flags */);
494
495                 if ((status < 0) && ((errno == EAGAIN) || (errno == EINTR)))
496                         continue;
497
498                 if (status < 0)
499                 {
500                         char errbuf[1024];
501                         DEBUG ("recv(2) failed: %s",
502                                         sstrerror (errno, errbuf, sizeof (errbuf)));
503                         DEBUG ("Closing socket #%i", sd);
504                         close (sd);
505                         sock_descr = sd = -1;
506                         return (-1);
507                 }
508
509                 DEBUG ("recv'd %i bytes", status);
510
511                 /* 
512                  * Do some sanity checks first
513                  */
514                 if (status < RESP_HEADER_SIZE)
515                 {
516                         WARNING ("ntpd plugin: Short (%i bytes) packet received",
517                                         (int) status);
518                         continue;
519                 }
520                 if (INFO_MODE (res.rm_vn_mode) != MODE_PRIVATE)
521                 {
522                         NOTICE ("ntpd plugin: Packet received with mode %i",
523                                         INFO_MODE (res.rm_vn_mode));
524                         continue;
525                 }
526                 if (INFO_IS_AUTH (res.auth_seq))
527                 {
528                         NOTICE ("ntpd plugin: Encrypted packet received");
529                         continue;
530                 }
531                 if (!ISRESPONSE (res.rm_vn_mode))
532                 {
533                         NOTICE ("ntpd plugin: Received request packet, "
534                                         "wanted response");
535                         continue;
536                 }
537                 if (INFO_MBZ (res.mbz_itemsize))
538                 {
539                         WARNING ("ntpd plugin: Received packet with nonzero "
540                                         "MBZ field!");
541                         continue;
542                 }
543                 if (res.implementation != IMPL_XNTPD)
544                 {
545                         WARNING ("ntpd plugin: Asked for request of type %i, "
546                                         "got %i", (int) IMPL_XNTPD, (int) res.implementation);
547                         continue;
548                 }
549
550                 /* Check for error code */
551                 if (INFO_ERR (res.err_nitems) != 0)
552                 {
553                         ERROR ("ntpd plugin: Received error code %i",
554                                         (int) INFO_ERR(res.err_nitems));
555                         return ((int) INFO_ERR (res.err_nitems));
556                 }
557
558                 /* extract number of items in this packet and the size of these items */
559                 pkt_item_num = INFO_NITEMS (res.err_nitems);
560                 pkt_item_len = INFO_ITEMSIZE (res.mbz_itemsize);
561                 DEBUG ("pkt_item_num = %i; pkt_item_len = %i;",
562                                 pkt_item_num, pkt_item_len);
563
564                 /* Check if the reported items fit in the packet */
565                 if ((pkt_item_num * pkt_item_len) > (status - RESP_HEADER_SIZE))
566                 {
567                         ERROR ("ntpd plugin: %i items * %i bytes > "
568                                         "%i bytes - %i bytes header",
569                                         (int) pkt_item_num, (int) pkt_item_len,
570                                         (int) status, (int) RESP_HEADER_SIZE);
571                         continue;
572                 }
573
574                 if (pkt_item_len > res_item_size)
575                 {
576                         ERROR ("ntpd plugin: (pkt_item_len = %i) "
577                                         ">= (res_item_size = %i)",
578                                         pkt_item_len, res_item_size);
579                         continue;
580                 }
581
582                 /* If this is the first packet (time wise, not sequence wise),
583                  * set `res_size'. If it's not the first packet check if the
584                  * items have the same size. Discard invalid packets. */
585                 if (items_num == 0) /* first packet */
586                 {
587                         DEBUG ("*res_size = %i", pkt_item_len);
588                         *res_size = pkt_item_len;
589                 }
590                 else if (*res_size != pkt_item_len)
591                 {
592                         DEBUG ("Error: *res_size = %i; pkt_item_len = %i;",
593                                         *res_size, pkt_item_len);
594                         ERROR ("Item sizes differ.");
595                         continue;
596                 }
597
598                 /*
599                  * Because the items in the packet may be smaller than the
600                  * items requested, the following holds true:
601                  */
602                 assert ((*res_size == pkt_item_len)
603                                 && (pkt_item_len <= res_item_size));
604
605                 /* Calculate the padding. No idea why there might be any padding.. */
606                 pkt_padding = 0;
607                 if (pkt_item_len < res_item_size)
608                         pkt_padding = res_item_size - pkt_item_len;
609                 DEBUG ("res_item_size = %i; pkt_padding = %i;",
610                                 res_item_size, pkt_padding);
611
612                 /* Extract the sequence number */
613                 pkt_sequence = INFO_SEQ (res.auth_seq);
614                 if ((pkt_sequence < 0) || (pkt_sequence > MAXSEQ))
615                 {
616                         ERROR ("ntpd plugin: Received packet with sequence %i",
617                                         pkt_sequence);
618                         continue;
619                 }
620
621                 /* Check if this sequence has been received before. If so, discard it. */
622                 if (pkt_recvd[pkt_sequence] != '\0')
623                 {
624                         NOTICE ("ntpd plugin: Sequence %i received twice",
625                                         pkt_sequence);
626                         continue;
627                 }
628
629                 /* If `pkt_lastseq != -1' another packet without `more bit' has
630                  * been received. */
631                 if (!ISMORE (res.rm_vn_mode))
632                 {
633                         if (pkt_lastseq != -1)
634                         {
635                                 ERROR ("ntpd plugin: Two packets which both "
636                                                 "claim to be the last one in the "
637                                                 "sequence have been received.");
638                                 continue;
639                         }
640                         pkt_lastseq = pkt_sequence;
641                         DEBUG ("Last sequence = %i;", pkt_lastseq);
642                 }
643
644                 /*
645                  * Enough with the checks. Copy the data now.
646                  * We start by allocating some more memory.
647                  */
648                 DEBUG ("realloc (%p, %i)", (void *) *res_data,
649                                 (items_num + pkt_item_num) * res_item_size);
650                 items = realloc ((void *) *res_data,
651                                 (items_num + pkt_item_num) * res_item_size);
652                 if (items == NULL)
653                 {
654                         items = *res_data;
655                         ERROR ("ntpd plugin: realloc failed.");
656                         continue;
657                 }
658                 items_num += pkt_item_num;
659                 *res_data = items;
660
661                 for (i = 0; i < pkt_item_num; i++)
662                 {
663                         /* dst: There are already `*res_items' items with
664                          *      res_item_size bytes each in in `*res_data'. Set
665                          *      dst to the first byte after that. */
666                         void *dst = (void *) (*res_data + ((*res_items) * res_item_size));
667                         /* src: We use `pkt_item_len' to calculate the offset
668                          *      from the beginning of the packet, because the
669                          *      items in the packet may be smaller than the
670                          *      items that were requested. We skip `i' such
671                          *      items. */
672                         void *src = (void *) (((char *) res.data) + (i * pkt_item_len));
673
674                         /* Set the padding to zeros */
675                         if (pkt_padding != 0)
676                                 memset (dst, '\0', res_item_size);
677                         memcpy (dst, src, (size_t) pkt_item_len);
678
679                         /* Increment `*res_items' by one, so `dst' will end up
680                          * one further in the next round. */
681                         (*res_items)++;
682                 } /* for (pkt_item_num) */
683
684                 pkt_recvd[pkt_sequence] = (char) 1;
685                 pkt_recvd_num++;
686
687                 if ((pkt_recvd_num - 1) == pkt_lastseq)
688                         done = 1;
689         } /* while (done == 0) */
690
691         return (0);
692 } /* int ntpd_receive_response */
693
694 /* For a description of the arguments see `ntpd_do_query' below. */
695 static int ntpd_send_request (int req_code, int req_items, int req_size, char *req_data)
696 {
697         int             sd;
698         struct req_pkt  req;
699         size_t          req_data_len;
700         int             status;
701
702         assert (req_items >= 0);
703         assert (req_size  >= 0);
704
705         if ((sd = ntpd_connect ()) < 0)
706                 return (-1);
707
708         memset ((void *) &req, '\0', sizeof (req));
709         req.rm_vn_mode = RM_VN_MODE(0, 0, 0);
710         req.auth_seq   = AUTH_SEQ (0, 0);
711         req.implementation = IMPL_XNTPD;
712         req.request = (unsigned char) req_code;
713
714         req_data_len = (size_t) (req_items * req_size);
715
716         assert (((req_data != NULL) && (req_data_len > 0))
717                         || ((req_data == NULL) && (req_items == 0) && (req_size == 0)));
718
719         req.err_nitems   = ERR_NITEMS (0, req_items);
720         req.mbz_itemsize = MBZ_ITEMSIZE (req_size);
721         
722         if (req_data != NULL)
723                 memcpy ((void *) req.data, (const void *) req_data, req_data_len);
724
725         DEBUG ("req_items = %i; req_size = %i; req_data = %p;",
726                         req_items, req_size, (void *) req_data);
727
728         status = swrite (sd, (const char *) &req, REQ_LEN_NOMAC);
729         if (status < 0)
730         {
731                 DEBUG ("`swrite' failed. Closing socket #%i", sd);
732                 close (sd);
733                 sock_descr = sd = -1;
734                 return (status);
735         }
736
737         return (0);
738 }
739
740 /*
741  * ntpd_do_query:
742  *
743  * req_code:      Type of request packet
744  * req_items:     Numver of items in the request
745  * req_size:      Size of one item in the request
746  * req_data:      Data of the request packet
747  * res_items:     Pointer to where the number returned items will be stored.
748  * res_size:      Pointer to where the size of one returned item will be stored.
749  * res_data:      This is where a pointer to the (allocated) data will be stored.
750  * res_item_size: Size of one returned item. (used to calculate padding)
751  *
752  * returns:       zero upon success, non-zero otherwise.
753  */
754 static int ntpd_do_query (int req_code, int req_items, int req_size, char *req_data,
755                 int *res_items, int *res_size, char **res_data, int res_item_size)
756 {
757         int status;
758
759         status = ntpd_send_request (req_code, req_items, req_size, req_data);
760         if (status != 0)
761                 return (status);
762
763         status = ntpd_receive_response (req_code, res_items, res_size, res_data,
764                         res_item_size);
765         return (status);
766 }
767
768 static double ntpd_read_fp (int32_t val_int)
769 {
770         double val_double;
771
772         val_int = ntohl (val_int);
773         val_double = ((double) val_int) / FP_FRAC;
774
775         return (val_double);
776 }
777
778 static int ntpd_read (void)
779 {
780         struct info_kernel *ik;
781         int                 ik_num;
782         int                 ik_size;
783
784         struct info_peer_summary *ps;
785         int                       ps_num;
786         int                       ps_size;
787
788         int status;
789         int i;
790
791         ik      = NULL;
792         ik_num  = 0;
793         ik_size = 0;
794
795         status = ntpd_do_query (REQ_GET_KERNEL,
796                         0, 0, NULL, /* request data */
797                         &ik_num, &ik_size, (char **) ((void *) &ik), /* response data */
798                         sizeof (struct info_kernel));
799
800         if (status != 0)
801         {
802                 DEBUG ("ntpd_do_query failed with status %i", status);
803                 return (-1);
804         }
805         if ((ik == NULL) || (ik_num == 0) || (ik_size == 0))
806         {
807                 DEBUG ("ntpd_do_query returned: ik = %p; ik_num = %i; ik_size = %i;",
808                                 (void *) ik, ik_num, ik_size);
809                 return (-1);
810         }
811
812         /* kerninfo -> estimated error */
813
814         DEBUG ("info_kernel:\n"
815                         "  pll offset    = %.8f\n"
816                         "  pll frequency = %.8f\n" /* drift compensation */
817                         "  est error     = %.8f\n",
818                         ntpd_read_fp (ik->offset),
819                         ntpd_read_fp (ik->freq),
820                         ntpd_read_fp (ik->esterror));
821
822         ntpd_submit ("frequency_offset", "loop",  ntpd_read_fp (ik->freq));
823         ntpd_submit ("time_offset",      "loop",  ntpd_read_fp (ik->offset));
824         ntpd_submit ("time_offset",      "error", ntpd_read_fp (ik->esterror));
825
826         free (ik);
827         ik = NULL;
828
829         status = ntpd_do_query (REQ_PEER_LIST_SUM,
830                         0, 0, NULL, /* request data */
831                         &ps_num, &ps_size, (char **) ((void *) &ps), /* response data */
832                         sizeof (struct info_peer_summary));
833         if (status != 0)
834         {
835                 DEBUG ("ntpd_do_query failed with status %i", status);
836                 return (-1);
837         }
838         if ((ps == NULL) || (ps_num == 0) || (ps_size == 0))
839         {
840                 DEBUG ("ntpd_do_query returned: ps = %p; ps_num = %i; ps_size = %i;",
841                                 (void *) ps, ps_num, ps_size);
842                 return (-1);
843         }
844
845         for (i = 0; i < ps_num; i++)
846         {
847                 struct info_peer_summary *ptr;
848                 double offset;
849
850                 char peername[NI_MAXHOST];
851                 int refclock_id;
852                 
853                 ptr = ps + i;
854                 refclock_id = 0;
855
856                 /*
857                 if (((ntohl (ptr->dstadr) & 0xFFFFFF00) == 0x7F000000) || (ptr->dstadr == 0))
858                         continue;
859                         */
860
861                 /* Convert the `long floating point' offset value to double */
862                 M_LFPTOD (ntohl (ptr->offset_int), ntohl (ptr->offset_frc), offset);
863
864                 if (ptr->v6_flag)
865                 {
866                         struct sockaddr_in6 sa;
867
868                         memset (&sa, 0, sizeof (sa));
869                         sa.sin6_family = AF_INET6;
870                         sa.sin6_port = htons (123);
871                         memcpy (&sa.sin6_addr, &ptr->srcadr6, sizeof (struct in6_addr));
872
873                         status = getnameinfo ((const struct sockaddr *) &sa,
874                                         sizeof (sa),
875                                         peername, sizeof (peername),
876                                         NULL, 0, 0 /* no flags */);
877                         if (status != 0)
878                         {
879                                 char errbuf[1024];
880                                 ERROR ("ntpd plugin: getnameinfo failed: %s",
881                                                 (status == EAI_SYSTEM)
882                                                 ? sstrerror (errno, errbuf, sizeof (errbuf))
883                                                 : gai_strerror (status));
884                                 continue;
885                         }
886                 }
887                 else if ((ntohl (ptr->srcadr) & REFCLOCK_MASK) == REFCLOCK_ADDR)
888                 {
889                         struct in_addr  addr_obj;
890                         char *addr_str;
891
892                         refclock_id = (ntohl (ptr->srcadr) >> 8) & 0x000000FF;
893
894                         if (refclock_id < refclock_names_num)
895                         {
896                                 strncpy (peername, refclock_names[refclock_id],
897                                                 sizeof (peername));
898                         }
899                         else
900                         {
901                                 memset ((void *) &addr_obj, '\0', sizeof (addr_obj));
902                                 addr_obj.s_addr = ptr->srcadr;
903                                 addr_str = inet_ntoa (addr_obj);
904
905                                 strncpy (peername, addr_str, sizeof (peername));
906                         }
907                 }
908                 else /* IPv4 */
909                 {
910                         struct in_addr  addr_obj;
911                         struct hostent *addr_he;
912                         char           *addr_str;
913
914                         memset ((void *) &addr_obj, '\0', sizeof (addr_obj));
915                         addr_obj.s_addr = ptr->srcadr;
916                         addr_str = inet_ntoa (addr_obj);
917
918                         addr_he = gethostbyaddr ((const void *) &addr_obj,
919                                         sizeof (addr_obj), AF_INET);
920                         if (addr_he != NULL)
921                         {
922                                 strncpy (peername, addr_he->h_name, sizeof (peername));
923                         }
924                         else
925                         {
926                                 strncpy (peername, addr_str, sizeof (peername));
927                         }
928                 }
929
930                 DEBUG ("peer %i:\n"
931                                 "  peername   = %s\n"
932                                 "  srcadr     = 0x%08x\n"
933                                 "  delay      = %f\n"
934                                 "  offset_int = %i\n"
935                                 "  offset_frc = %i\n"
936                                 "  offset     = %f\n"
937                                 "  dispersion = %f\n",
938                                 i,
939                                 peername,
940                                 ntohl (ptr->srcadr),
941                                 ntpd_read_fp (ptr->delay),
942                                 ntohl (ptr->offset_int),
943                                 ntohl (ptr->offset_frc),
944                                 offset,
945                                 ntpd_read_fp (ptr->dispersion));
946
947                 if (refclock_id != 1) /* not the system clock (offset will always be zero.. */
948                         ntpd_submit ("time_offset", peername, offset);
949                 ntpd_submit ("time_dispersion", peername, ntpd_read_fp (ptr->dispersion));
950                 if (refclock_id == 0) /* not a reference clock */
951                         ntpd_submit ("delay", peername, ntpd_read_fp (ptr->delay));
952         }
953
954         free (ps);
955         ps = NULL;
956
957         return (0);
958 } /* int ntpd_read */
959 #endif /* NTPD_HAVE_READ */
960
961 void module_register (void)
962 {
963 #if NTPD_HAVE_READ
964         plugin_register_config ("ntpd", ntpd_config,
965                         config_keys, config_keys_num);
966         plugin_register_read ("ntpd", ntpd_read);
967 #endif /* NTPD_HAVE_READ */
968 } /* void module_register */