]> CyberLeo.Net >> Repos - FreeBSD/releng/9.3.git/blob - contrib/ntp/ntpdc/ntpdc.c
o Fix invalid TCP checksums with pf(4). [EN-16:02.pf]
[FreeBSD/releng/9.3.git] / contrib / ntp / ntpdc / ntpdc.c
1 /*
2  * ntpdc - control and monitor your ntpd daemon
3  */
4 #include <config.h>
5 #include <stdio.h>
6 #include <stddef.h>
7 #include <ctype.h>
8 #include <signal.h>
9 #include <setjmp.h>
10 #ifdef HAVE_UNISTD_H
11 # include <unistd.h>
12 #endif
13 #ifdef HAVE_FCNTL_H
14 # include <fcntl.h>
15 #endif
16 #ifdef SYS_WINNT
17 # include <mswsock.h>
18 #endif
19 #include <isc/net.h>
20 #include <isc/result.h>
21
22 #include "ntpdc.h"
23 #include "ntp_select.h"
24 #include "ntp_stdlib.h"
25 #include "ntp_assert.h"
26 #include "ntp_lineedit.h"
27 #ifdef OPENSSL
28 #include "openssl/evp.h"
29 #include "openssl/objects.h"
30 #endif
31 #include <ssl_applink.c>
32
33 #include "ntp_libopts.h"
34 #include "ntpdc-opts.h"
35 #include "safecast.h"
36
37 #ifdef SYS_VXWORKS
38                                 /* vxWorks needs mode flag -casey*/
39 # define open(name, flags)   open(name, flags, 0777)
40 # define SERVER_PORT_NUM     123
41 #endif
42
43 /* We use COMMAND as an autogen keyword */
44 #ifdef COMMAND
45 # undef COMMAND
46 #endif
47
48 /*
49  * Because we now potentially understand a lot of commands (and
50  * it requires a lot of commands to talk to ntpd) we will run
51  * interactive if connected to a terminal.
52  */
53 static  int     interactive = 0;        /* set to 1 when we should prompt */
54 static  const char *    prompt = "ntpdc> ";     /* prompt to ask him about */
55
56 /*
57  * Keyid used for authenticated requests.  Obtained on the fly.
58  */
59 static  u_long  info_auth_keyid;
60 static int keyid_entered = 0;
61
62 static  int     info_auth_keytype = NID_md5;    /* MD5 */
63 static  size_t  info_auth_hashlen = 16;         /* MD5 */
64 u_long  current_time;           /* needed by authkeys; not used */
65
66 /*
67  * for get_systime()
68  */
69 s_char  sys_precision;          /* local clock precision (log2 s) */
70
71 int             ntpdcmain       (int,   char **);
72 /*
73  * Built in command handler declarations
74  */
75 static  int     openhost        (const char *);
76 static  int     sendpkt         (void *, size_t);
77 static  void    growpktdata     (void);
78 static  int     getresponse     (int, int, size_t *, size_t *, const char **, size_t);
79 static  int     sendrequest     (int, int, int, size_t, size_t, const char *);
80 static  void    getcmds         (void);
81 static  RETSIGTYPE abortcmd     (int);
82 static  void    docmd           (const char *);
83 static  void    tokenize        (const char *, char **, int *);
84 static  int     findcmd         (char *, struct xcmd *, struct xcmd *, struct xcmd **);
85 static  int     getarg          (char *, int, arg_v *);
86 static  int     getnetnum       (const char *, sockaddr_u *, char *, int);
87 static  void    help            (struct parse *, FILE *);
88 static  int     helpsort        (const void *, const void *);
89 static  void    printusage      (struct xcmd *, FILE *);
90 static  void    timeout         (struct parse *, FILE *);
91 static  void    my_delay        (struct parse *, FILE *);
92 static  void    host            (struct parse *, FILE *);
93 static  void    keyid           (struct parse *, FILE *);
94 static  void    keytype         (struct parse *, FILE *);
95 static  void    passwd          (struct parse *, FILE *);
96 static  void    hostnames       (struct parse *, FILE *);
97 static  void    setdebug        (struct parse *, FILE *);
98 static  void    quit            (struct parse *, FILE *);
99 static  void    version         (struct parse *, FILE *);
100 static  void    warning         (const char *, ...)
101     __attribute__((__format__(__printf__, 1, 2)));
102 static  void    error           (const char *, ...)
103     __attribute__((__format__(__printf__, 1, 2)));
104 static  u_long  getkeyid        (const char *);
105
106
107
108 /*
109  * Built-in commands we understand
110  */
111 static  struct xcmd builtins[] = {
112         { "?",          help,           {  OPT|NTP_STR, NO, NO, NO },
113           { "command", "", "", "" },
114           "tell the use and syntax of commands" },
115         { "help",       help,           {  OPT|NTP_STR, NO, NO, NO },
116           { "command", "", "", "" },
117           "tell the use and syntax of commands" },
118         { "timeout",    timeout,        { OPT|NTP_UINT, NO, NO, NO },
119           { "msec", "", "", "" },
120           "set the primary receive time out" },
121         { "delay",      my_delay,       { OPT|NTP_INT, NO, NO, NO },
122           { "msec", "", "", "" },
123           "set the delay added to encryption time stamps" },
124         { "host",       host,           { OPT|NTP_STR, OPT|NTP_STR, NO, NO },
125           { "-4|-6", "hostname", "", "" },
126           "specify the host whose NTP server we talk to" },
127         { "passwd",     passwd,         { OPT|NTP_STR, NO, NO, NO },
128           { "", "", "", "" },
129           "specify a password to use for authenticated requests"},
130         { "hostnames",  hostnames,      { OPT|NTP_STR, NO, NO, NO },
131           { "yes|no", "", "", "" },
132           "specify whether hostnames or net numbers are printed"},
133         { "debug",      setdebug,       { OPT|NTP_STR, NO, NO, NO },
134           { "no|more|less", "", "", "" },
135           "set/change debugging level" },
136         { "quit",       quit,           { NO, NO, NO, NO },
137           { "", "", "", "" },
138           "exit ntpdc" },
139         { "exit",       quit,           { NO, NO, NO, NO },
140           { "", "", "", "" },
141           "exit ntpdc" },
142         { "keyid",      keyid,          { OPT|NTP_UINT, NO, NO, NO },
143           { "key#", "", "", "" },
144           "set/show keyid to use for authenticated requests" },
145         { "keytype",    keytype,        { OPT|NTP_STR, NO, NO, NO },
146           { "(md5|des)", "", "", "" },
147           "set/show key authentication type for authenticated requests (des|md5)" },
148         { "version",    version,        { NO, NO, NO, NO },
149           { "", "", "", "" },
150           "print version number" },
151         { 0,            0,              { NO, NO, NO, NO },
152           { "", "", "", "" }, "" }
153 };
154
155
156 /*
157  * Default values we use.
158  */
159 #define DEFHOST         "localhost"     /* default host name */
160 #define DEFTIMEOUT      (5)             /* 5 second time out */
161 #define DEFSTIMEOUT     (2)             /* 2 second time out after first */
162 #define DEFDELAY        0x51EB852       /* 20 milliseconds, l_fp fraction */
163 #define LENHOSTNAME     256             /* host name is 256 characters long */
164 #define MAXCMDS         100             /* maximum commands on cmd line */
165 #define MAXHOSTS        200             /* maximum hosts on cmd line */
166 #define MAXLINE         512             /* maximum line length */
167 #define MAXTOKENS       (1+1+MAXARGS+MOREARGS+2)        /* maximum number of usable tokens */
168 #define SCREENWIDTH     78              /* nominal screen width in columns */
169
170 /*
171  * Some variables used and manipulated locally
172  */
173 static  struct sock_timeval tvout = { DEFTIMEOUT, 0 };  /* time out for reads */
174 static  struct sock_timeval tvsout = { DEFSTIMEOUT, 0 };/* secondary time out */
175 static  l_fp delay_time;                                /* delay time */
176 static  char currenthost[LENHOSTNAME];                  /* current host name */
177 int showhostnames = 1;                                  /* show host names by default */
178
179 static  int ai_fam_templ;                               /* address family */
180 static  int ai_fam_default;                             /* default address family */
181 static  SOCKET sockfd;                                  /* fd socket is opened on */
182 static  int havehost = 0;                               /* set to 1 when host open */
183 int s_port = 0;
184
185 /*
186  * Holds data returned from queries.  We allocate INITDATASIZE
187  * octets to begin with, increasing this as we need to.
188  */
189 #define INITDATASIZE    (sizeof(struct resp_pkt) * 16)
190 #define INCDATASIZE     (sizeof(struct resp_pkt) * 8)
191
192 static  char *pktdata;
193 static  int pktdatasize;
194
195 /*
196  * These are used to help the magic with old and new versions of ntpd.
197  */
198 int impl_ver = IMPL_XNTPD;
199 static int req_pkt_size = REQ_LEN_NOMAC;
200
201 /*
202  * For commands typed on the command line (with the -c option)
203  */
204 static  int numcmds = 0;
205 static  const char *ccmds[MAXCMDS];
206 #define ADDCMD(cp)      if (numcmds < MAXCMDS) ccmds[numcmds++] = (cp)
207
208 /*
209  * When multiple hosts are specified.
210  */
211 static  int numhosts = 0;
212 static  const char *chosts[MAXHOSTS];
213 #define ADDHOST(cp)     if (numhosts < MAXHOSTS) chosts[numhosts++] = (cp)
214
215 /*
216  * Error codes for internal use
217  */
218 #define ERR_INCOMPLETE          16
219 #define ERR_TIMEOUT             17
220
221 /*
222  * Macro definitions we use
223  */
224 #define ISSPACE(c)      ((c) == ' ' || (c) == '\t')
225 #define ISEOL(c)        ((c) == '\n' || (c) == '\r' || (c) == '\0')
226 #define STREQ(a, b)     (*(a) == *(b) && strcmp((a), (b)) == 0)
227
228 /*
229  * Jump buffer for longjumping back to the command level
230  */
231 static  jmp_buf interrupt_buf;
232 static  volatile int jump = 0;
233
234 /*
235  * Pointer to current output unit
236  */
237 static  FILE *current_output;
238
239 /*
240  * Command table imported from ntpdc_ops.c
241  */
242 extern struct xcmd opcmds[];
243
244 char const *progname;
245
246 #ifdef NO_MAIN_ALLOWED
247 CALL(ntpdc,"ntpdc",ntpdcmain);
248 #else
249 int
250 main(
251         int argc,
252         char *argv[]
253         )
254 {
255         return ntpdcmain(argc, argv);
256 }
257 #endif
258
259 #ifdef SYS_VXWORKS
260 void clear_globals(void)
261 {
262     showhostnames = 0;              /* show host names by default */
263     havehost = 0;                   /* set to 1 when host open */
264     numcmds = 0;
265     numhosts = 0;
266 }
267 #endif
268
269 /*
270  * main - parse arguments and handle options
271  */
272 int
273 ntpdcmain(
274         int argc,
275         char *argv[]
276         )
277 {
278
279         delay_time.l_ui = 0;
280         delay_time.l_uf = DEFDELAY;
281
282 #ifdef SYS_VXWORKS
283         clear_globals();
284         taskPrioritySet(taskIdSelf(), 100 );
285 #endif
286
287         init_lib();     /* sets up ipv4_works, ipv6_works */
288         ssl_applink();
289         init_auth();
290
291         /* Check to see if we have IPv6. Otherwise default to IPv4 */
292         if (!ipv6_works)
293                 ai_fam_default = AF_INET;
294
295         progname = argv[0];
296
297         {
298                 int optct = ntpOptionProcess(&ntpdcOptions, argc, argv);
299                 argc -= optct;
300                 argv += optct;
301         }
302
303         if (HAVE_OPT(IPV4))
304                 ai_fam_templ = AF_INET;
305         else if (HAVE_OPT(IPV6))
306                 ai_fam_templ = AF_INET6;
307         else
308                 ai_fam_templ = ai_fam_default;
309
310         if (HAVE_OPT(COMMAND)) {
311                 int             cmdct = STACKCT_OPT( COMMAND );
312                 const char**    cmds  = STACKLST_OPT( COMMAND );
313
314                 while (cmdct-- > 0) {
315                         ADDCMD(*cmds++);
316                 }
317         }
318
319         debug = OPT_VALUE_SET_DEBUG_LEVEL;
320
321         if (HAVE_OPT(INTERACTIVE)) {
322                 interactive = 1;
323         }
324
325         if (HAVE_OPT(NUMERIC)) {
326                 showhostnames = 0;
327         }
328
329         if (HAVE_OPT(LISTPEERS)) {
330                 ADDCMD("listpeers");
331         }
332
333         if (HAVE_OPT(PEERS)) {
334                 ADDCMD("peers");
335         }
336
337         if (HAVE_OPT(SHOWPEERS)) {
338                 ADDCMD("dmpeers");
339         }
340
341         if (ntp_optind == argc) {
342                 ADDHOST(DEFHOST);
343         } else {
344                 for (; ntp_optind < argc; ntp_optind++)
345                     ADDHOST(argv[ntp_optind]);
346         }
347
348         if (numcmds == 0 && interactive == 0
349             && isatty(fileno(stdin)) && isatty(fileno(stderr))) {
350                 interactive = 1;
351         }
352
353 #ifndef SYS_WINNT /* Under NT cannot handle SIGINT, WIN32 spawns a handler */
354         if (interactive)
355             (void) signal_no_reset(SIGINT, abortcmd);
356 #endif /* SYS_WINNT */
357
358         /*
359          * Initialize the packet data buffer
360          */
361         pktdatasize = INITDATASIZE;
362         pktdata = emalloc(INITDATASIZE);
363
364         if (numcmds == 0) {
365                 (void) openhost(chosts[0]);
366                 getcmds();
367         } else {
368                 int ihost;
369                 int icmd;
370
371                 for (ihost = 0; ihost < numhosts; ihost++) {
372                         if (openhost(chosts[ihost]))
373                             for (icmd = 0; icmd < numcmds; icmd++) {
374                                     if (numhosts > 1) 
375                                         printf ("--- %s ---\n",chosts[ihost]);
376                                     docmd(ccmds[icmd]);
377                             }
378                 }
379         }
380 #ifdef SYS_WINNT
381         WSACleanup();
382 #endif
383         return(0);
384 } /* main end */
385
386
387 /*
388  * openhost - open a socket to a host
389  */
390 static int
391 openhost(
392         const char *hname
393         )
394 {
395         char temphost[LENHOSTNAME];
396         int a_info, i;
397         struct addrinfo hints, *ai = NULL;
398         sockaddr_u addr;
399         size_t octets;
400         register const char *cp;
401         char name[LENHOSTNAME];
402         char service[5];
403
404         /*
405          * We need to get by the [] if they were entered 
406          */
407         
408         cp = hname;
409         
410         if (*cp == '[') {
411                 cp++;   
412                 for (i = 0; *cp && *cp != ']'; cp++, i++)
413                         name[i] = *cp;
414                 if (*cp == ']') {
415                         name[i] = '\0';
416                         hname = name;
417                 } else {
418                         return 0;
419                 }
420         }       
421
422         /*
423          * First try to resolve it as an ip address and if that fails,
424          * do a fullblown (dns) lookup. That way we only use the dns
425          * when it is needed and work around some implementations that
426          * will return an "IPv4-mapped IPv6 address" address if you
427          * give it an IPv4 address to lookup.
428          */
429         strlcpy(service, "ntp", sizeof(service));
430         ZERO(hints);
431         hints.ai_family = ai_fam_templ;
432         hints.ai_protocol = IPPROTO_UDP;
433         hints.ai_socktype = SOCK_DGRAM;
434         hints.ai_flags = Z_AI_NUMERICHOST;
435
436         a_info = getaddrinfo(hname, service, &hints, &ai);
437         if (a_info == EAI_NONAME
438 #ifdef EAI_NODATA
439             || a_info == EAI_NODATA
440 #endif
441            ) {
442                 hints.ai_flags = AI_CANONNAME;
443 #ifdef AI_ADDRCONFIG
444                 hints.ai_flags |= AI_ADDRCONFIG;
445 #endif
446                 a_info = getaddrinfo(hname, service, &hints, &ai);      
447         }
448         /* Some older implementations don't like AI_ADDRCONFIG. */
449         if (a_info == EAI_BADFLAGS) {
450                 hints.ai_flags = AI_CANONNAME;
451                 a_info = getaddrinfo(hname, service, &hints, &ai);      
452         }
453         if (a_info != 0) {
454                 fprintf(stderr, "%s\n", gai_strerror(a_info));
455                 if (ai != NULL)
456                         freeaddrinfo(ai);
457                 return 0;
458         }
459
460         /* 
461          * getaddrinfo() has returned without error so ai should not 
462          * be NULL.
463          */
464         INSIST(ai != NULL);
465         ZERO(addr);
466         octets = min(sizeof(addr), ai->ai_addrlen);
467         memcpy(&addr, ai->ai_addr, octets);
468
469         if (ai->ai_canonname == NULL)
470                 strlcpy(temphost, stoa(&addr), sizeof(temphost));
471         else
472                 strlcpy(temphost, ai->ai_canonname, sizeof(temphost));
473
474         if (debug > 2)
475                 printf("Opening host %s\n", temphost);
476
477         if (havehost == 1) {
478                 if (debug > 2)
479                         printf("Closing old host %s\n", currenthost);
480                 closesocket(sockfd);
481                 havehost = 0;
482         }
483         strlcpy(currenthost, temphost, sizeof(currenthost));
484         
485         /* port maps to the same in both families */
486         s_port = NSRCPORT(&addr);; 
487 #ifdef SYS_VXWORKS
488         ((struct sockaddr_in6 *)&hostaddr)->sin6_port = htons(SERVER_PORT_NUM);
489         if (ai->ai_family == AF_INET)
490                 *(struct sockaddr_in *)&hostaddr= 
491                         *((struct sockaddr_in *)ai->ai_addr);
492         else 
493                 *(struct sockaddr_in6 *)&hostaddr= 
494                         *((struct sockaddr_in6 *)ai->ai_addr);
495 #endif /* SYS_VXWORKS */
496
497 #ifdef SYS_WINNT
498         {
499                 int optionValue = SO_SYNCHRONOUS_NONALERT;
500                 int err;
501
502                 err = setsockopt(INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE, (char *)&optionValue, sizeof(optionValue));
503                 if (err != NO_ERROR) {
504                         (void) fprintf(stderr, "cannot open nonoverlapped sockets\n");
505                         exit(1);
506                 }
507         }
508 #endif /* SYS_WINNT */
509
510         sockfd = socket(ai->ai_family, SOCK_DGRAM, 0);
511         if (sockfd == INVALID_SOCKET) {
512                 error("socket");
513                 exit(-1);
514         }
515         
516 #ifdef NEED_RCVBUF_SLOP
517 # ifdef SO_RCVBUF
518         {
519                 int rbufsize = INITDATASIZE + 2048; /* 2K for slop */
520
521                 if (setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF,
522                                &rbufsize, sizeof(int)) == -1)
523                     error("setsockopt");
524         }
525 # endif
526 #endif
527
528 #ifdef SYS_VXWORKS
529         if (connect(sockfd, (struct sockaddr *)&hostaddr, 
530                     sizeof(hostaddr)) == -1)
531 #else
532         if (connect(sockfd, ai->ai_addr, ai->ai_addrlen) == -1)
533 #endif /* SYS_VXWORKS */
534         {
535                 error("connect");
536                 exit(-1);
537         }
538
539         freeaddrinfo(ai);
540         havehost = 1;
541         req_pkt_size = REQ_LEN_NOMAC;
542         impl_ver = IMPL_XNTPD;
543         return 1;
544 }
545
546
547 /* XXX ELIMINATE sendpkt similar in ntpq.c, ntpdc.c, ntp_io.c, ntptrace.c */
548 /*
549  * sendpkt - send a packet to the remote host
550  */
551 static int
552 sendpkt(
553         void *  xdata,
554         size_t  xdatalen
555         )
556 {
557         if (send(sockfd, xdata, xdatalen, 0) == -1) {
558                 warning("write to %s failed", currenthost);
559                 return -1;
560         }
561
562         return 0;
563 }
564
565
566 /*
567  * growpktdata - grow the packet data area
568  */
569 static void
570 growpktdata(void)
571 {
572         size_t priorsz;
573
574         priorsz = (size_t)pktdatasize;
575         pktdatasize += INCDATASIZE;
576         pktdata = erealloc_zero(pktdata, (size_t)pktdatasize, priorsz);
577 }
578
579
580 /*
581  * getresponse - get a (series of) response packet(s) and return the data
582  */
583 static int
584 getresponse(
585         int implcode,
586         int reqcode,
587         size_t *ritems,
588         size_t *rsize,
589         const char **rdata,
590         size_t esize
591         )
592 {
593         struct resp_pkt rpkt;
594         struct sock_timeval tvo;
595         size_t items;
596         size_t i;
597         size_t size;
598         size_t datasize;
599         char *datap;
600         char *tmp_data;
601         char haveseq[MAXSEQ+1];
602         int firstpkt;
603         int lastseq;
604         int numrecv;
605         int seq;
606         fd_set fds;
607         ssize_t n;
608         size_t pad;
609
610         /*
611          * This is pretty tricky.  We may get between 1 and many packets
612          * back in response to the request.  We peel the data out of
613          * each packet and collect it in one long block.  When the last
614          * packet in the sequence is received we'll know how many we
615          * should have had.  Note we use one long time out, should reconsider.
616          */
617         *ritems = 0;
618         *rsize = 0;
619         firstpkt = 1;
620         numrecv = 0;
621         *rdata = datap = pktdata;
622         lastseq = 999;  /* too big to be a sequence number */
623         ZERO(haveseq);
624         FD_ZERO(&fds);
625
626     again:
627         if (firstpkt)
628                 tvo = tvout;
629         else
630                 tvo = tvsout;
631         
632         FD_SET(sockfd, &fds);
633         n = select(sockfd+1, &fds, NULL, NULL, &tvo);
634         if (n == -1) {
635                 warning("select fails");
636                 return -1;
637         }
638         if (n == 0) {
639                 /*
640                  * Timed out.  Return what we have
641                  */
642                 if (firstpkt) {
643                         (void) fprintf(stderr,
644                                        "%s: timed out, nothing received\n",
645                                        currenthost);
646                         return ERR_TIMEOUT;
647                 } else {
648                         (void) fprintf(stderr,
649                                        "%s: timed out with incomplete data\n",
650                                        currenthost);
651                         if (debug) {
652                                 printf("Received sequence numbers");
653                                 for (n = 0; n <= MAXSEQ; n++)
654                                     if (haveseq[n])
655                                         printf(" %zd,", n);
656                                 if (lastseq != 999)
657                                     printf(" last frame received\n");
658                                 else
659                                     printf(" last frame not received\n");
660                         }
661                         return ERR_INCOMPLETE;
662                 }
663         }
664
665         n = recv(sockfd, (char *)&rpkt, sizeof(rpkt), 0);
666         if (n == -1) {
667                 warning("read");
668                 return -1;
669         }
670
671
672         /*
673          * Check for format errors.  Bug proofing.
674          */
675         if (n < (ssize_t)RESP_HEADER_SIZE) {
676                 if (debug)
677                         printf("Short (%zd byte) packet received\n", n);
678                 goto again;
679         }
680         if (INFO_VERSION(rpkt.rm_vn_mode) > NTP_VERSION ||
681             INFO_VERSION(rpkt.rm_vn_mode) < NTP_OLDVERSION) {
682                 if (debug)
683                         printf("Packet received with version %d\n",
684                                INFO_VERSION(rpkt.rm_vn_mode));
685                 goto again;
686         }
687         if (INFO_MODE(rpkt.rm_vn_mode) != MODE_PRIVATE) {
688                 if (debug)
689                         printf("Packet received with mode %d\n",
690                                INFO_MODE(rpkt.rm_vn_mode));
691                 goto again;
692         }
693         if (INFO_IS_AUTH(rpkt.auth_seq)) {
694                 if (debug)
695                         printf("Encrypted packet received\n");
696                 goto again;
697         }
698         if (!ISRESPONSE(rpkt.rm_vn_mode)) {
699                 if (debug)
700                         printf("Received request packet, wanted response\n");
701                 goto again;
702         }
703         if (INFO_MBZ(rpkt.mbz_itemsize) != 0) {
704                 if (debug)
705                         printf("Received packet with nonzero MBZ field!\n");
706                 goto again;
707         }
708
709         /*
710          * Check implementation/request.  Could be old data getting to us.
711          */
712         if (rpkt.implementation != implcode || rpkt.request != reqcode) {
713                 if (debug)
714                         printf(
715                             "Received implementation/request of %d/%d, wanted %d/%d",
716                             rpkt.implementation, rpkt.request,
717                             implcode, reqcode);
718                 goto again;
719         }
720
721         /*
722          * Check the error code.  If non-zero, return it.
723          */
724         if (INFO_ERR(rpkt.err_nitems) != INFO_OKAY) {
725                 if (debug && ISMORE(rpkt.rm_vn_mode)) {
726                         printf("Error code %d received on not-final packet\n",
727                                INFO_ERR(rpkt.err_nitems));
728                 }
729                 return (int)INFO_ERR(rpkt.err_nitems);
730         }
731
732         /*
733          * Collect items and size.  Make sure they make sense.
734          */
735         items = INFO_NITEMS(rpkt.err_nitems);
736         size = INFO_ITEMSIZE(rpkt.mbz_itemsize);
737         if (esize > size)
738                 pad = esize - size;
739         else 
740                 pad = 0;
741         datasize = items * size;
742         if ((size_t)datasize > (n-RESP_HEADER_SIZE)) {
743                 if (debug)
744                     printf(
745                             "Received items %zu, size %zu (total %zu), data in packet is %zu\n",
746                             items, size, datasize, n-RESP_HEADER_SIZE);
747                 goto again;
748         }
749
750         /*
751          * If this isn't our first packet, make sure the size matches
752          * the other ones.
753          */
754         if (!firstpkt && size != *rsize) {
755                 if (debug)
756                     printf("Received itemsize %zu, previous %zu\n",
757                            size, *rsize);
758                 goto again;
759         }
760         /*
761          * If we've received this before, +toss it
762          */
763         seq = INFO_SEQ(rpkt.auth_seq);
764         if (haveseq[seq]) {
765                 if (debug)
766                     printf("Received duplicate sequence number %d\n", seq);
767                 goto again;
768         }
769         haveseq[seq] = 1;
770
771         /*
772          * If this is the last in the sequence, record that.
773          */
774         if (!ISMORE(rpkt.rm_vn_mode)) {
775                 if (lastseq != 999) {
776                         printf("Received second end sequence packet\n");
777                         goto again;
778                 }
779                 lastseq = seq;
780         }
781
782         /*
783          * So far, so good.  Copy this data into the output array.
784          */
785         if ((datap + datasize + (pad * items)) > (pktdata + pktdatasize)) {
786                 size_t offset = datap - pktdata;
787                 growpktdata();
788                 *rdata = pktdata; /* might have been realloced ! */
789                 datap = pktdata + offset;
790         }
791         /* 
792          * We now move the pointer along according to size and number of
793          * items.  This is so we can play nice with older implementations
794          */
795
796         tmp_data = rpkt.u.data;
797         for (i = 0; i < items; i++) {
798                 memcpy(datap, tmp_data, (unsigned)size);
799                 tmp_data += size;
800                 zero_mem(datap + size, pad);
801                 datap += size + pad;
802         }
803
804         if (firstpkt) {
805                 firstpkt = 0;
806                 *rsize = size + pad;
807         }
808         *ritems += items;
809
810         /*
811          * Finally, check the count of received packets.  If we've got them
812          * all, return
813          */
814         ++numrecv;
815         if (numrecv <= lastseq)
816                 goto again;
817         return INFO_OKAY;
818 }
819
820
821 /*
822  * sendrequest - format and send a request packet
823  *
824  * Historically, ntpdc has used a fixed-size request packet regardless
825  * of the actual payload size.  When authenticating, the timestamp, key
826  * ID, and digest have been placed just before the end of the packet.
827  * With the introduction in late 2009 of support for authenticated
828  * ntpdc requests using larger 20-octet digests (vs. 16 for MD5), we
829  * come up four bytes short.
830  *
831  * To maintain interop while allowing for larger digests, the behavior
832  * is unchanged when using 16-octet digests.  For larger digests, the
833  * timestamp, key ID, and digest are placed immediately following the
834  * request payload, with the overall packet size variable.  ntpd can
835  * distinguish 16-octet digests by the overall request size being
836  * REQ_LEN_NOMAC + 4 + 16 with the auth bit enabled.  When using a
837  * longer digest, that request size should be avoided.
838  *
839  * With the form used with 20-octet and larger digests, the timestamp,
840  * key ID, and digest are located by ntpd relative to the start of the
841  * packet, and the size of the digest is then implied by the packet
842  * size.
843  */
844 static int
845 sendrequest(
846         int implcode,
847         int reqcode,
848         int auth,
849         size_t qitems,
850         size_t qsize,
851         const char *qdata
852         )
853 {
854         struct req_pkt qpkt;
855         size_t  datasize;
856         size_t  reqsize;
857         u_long  key_id;
858         l_fp    ts;
859         l_fp *  ptstamp;
860         size_t  maclen;
861         char *  pass;
862
863         ZERO(qpkt);
864         qpkt.rm_vn_mode = RM_VN_MODE(0, 0, 0);
865         qpkt.implementation = (u_char)implcode;
866         qpkt.request = (u_char)reqcode;
867
868         datasize = qitems * qsize;
869         if (datasize && qdata != NULL) {
870                 memcpy(qpkt.u.data, qdata, datasize);
871                 qpkt.err_nitems = ERR_NITEMS(0, qitems);
872                 qpkt.mbz_itemsize = MBZ_ITEMSIZE(qsize);
873         } else {
874                 qpkt.err_nitems = ERR_NITEMS(0, 0);
875                 qpkt.mbz_itemsize = MBZ_ITEMSIZE(qsize);  /* allow for optional first item */
876         }
877
878         if (!auth || (keyid_entered && info_auth_keyid == 0)) {
879                 qpkt.auth_seq = AUTH_SEQ(0, 0);
880                 return sendpkt(&qpkt, req_pkt_size);
881         }
882
883         if (info_auth_keyid == 0) {
884                 key_id = getkeyid("Keyid: ");
885                 if (!key_id) {
886                         fprintf(stderr, "Invalid key identifier\n");
887                         return 1;
888                 }
889                 info_auth_keyid = key_id;
890         }
891         if (!authistrusted(info_auth_keyid)) {
892                 pass = getpass_keytype(info_auth_keytype);
893                 if ('\0' == pass[0]) {
894                         fprintf(stderr, "Invalid password\n");
895                         return 1;
896                 }
897                 authusekey(info_auth_keyid, info_auth_keytype,
898                            (u_char *)pass);
899                 authtrust(info_auth_keyid, 1);
900         }
901         qpkt.auth_seq = AUTH_SEQ(1, 0);
902         if (info_auth_hashlen > 16) {
903                 /*
904                  * Only ntpd which expects REQ_LEN_NOMAC plus maclen
905                  * octets in an authenticated request using a 16 octet
906                  * digest (that is, a newer ntpd) will handle digests
907                  * larger than 16 octets, so for longer digests, do
908                  * not attempt to shorten the requests for downlevel
909                  * ntpd compatibility.
910                  */
911                 if (REQ_LEN_NOMAC != req_pkt_size)
912                         return 1;
913                 reqsize = REQ_LEN_HDR + datasize + sizeof(*ptstamp);
914                 /* align to 32 bits */
915                 reqsize = (reqsize + 3) & ~3;
916         } else
917                 reqsize = req_pkt_size;
918         ptstamp = (void *)((char *)&qpkt + reqsize);
919         ptstamp--;
920         get_systime(&ts);
921         L_ADD(&ts, &delay_time);
922         HTONL_FP(&ts, ptstamp);
923         maclen = authencrypt(
924                 info_auth_keyid, (void *)&qpkt, size2int_chk(reqsize));
925         if (!maclen) {  
926                 fprintf(stderr, "Key not found\n");
927                 return 1;
928         } else if (maclen != (int)(info_auth_hashlen + sizeof(keyid_t))) {
929                 fprintf(stderr,
930                         "%zu octet MAC, %zu expected with %zu octet digest\n",
931                         maclen, (info_auth_hashlen + sizeof(keyid_t)),
932                         info_auth_hashlen);
933                 return 1;
934         }
935         return sendpkt(&qpkt, reqsize + maclen);
936 }
937
938
939 /*
940  * doquery - send a request and process the response
941  */
942 int
943 doquery(
944         int implcode,
945         int reqcode,
946         int auth,
947         size_t qitems,
948         size_t qsize,
949         const char *qdata,
950         size_t *ritems,
951         size_t *rsize,
952         const char **rdata,
953         int quiet_mask,
954         int esize
955         )
956 {
957         int res;
958         char junk[512];
959         fd_set fds;
960         struct sock_timeval tvzero;
961
962         /*
963          * Check to make sure host is open
964          */
965         if (!havehost) {
966                 (void) fprintf(stderr, "***No host open, use `host' command\n");
967                 return -1;
968         }
969
970         /*
971          * Poll the socket and clear out any pending data
972          */
973 again:
974         do {
975                 tvzero.tv_sec = tvzero.tv_usec = 0;
976                 FD_ZERO(&fds);
977                 FD_SET(sockfd, &fds);
978                 res = select(sockfd+1, &fds, NULL, NULL, &tvzero);
979                 if (res == -1) {
980                         warning("polling select");
981                         return -1;
982                 } else if (res > 0)
983
984                     (void) recv(sockfd, junk, sizeof junk, 0);
985         } while (res > 0);
986
987
988         /*
989          * send a request
990          */
991         res = sendrequest(implcode, reqcode, auth, qitems, qsize, qdata);
992         if (res != 0)
993                 return res;
994         
995         /*
996          * Get the response.  If we got a standard error, print a message
997          */
998         res = getresponse(implcode, reqcode, ritems, rsize, rdata, esize);
999
1000         /*
1001          * Try to be compatible with older implementations of ntpd.
1002          */
1003         if (res == INFO_ERR_FMT && req_pkt_size != 48) {
1004                 int oldsize;
1005
1006                 oldsize = req_pkt_size;
1007
1008                 switch(req_pkt_size) {
1009                 case REQ_LEN_NOMAC:
1010                         req_pkt_size = 160;
1011                         break;
1012                 case 160:
1013                         req_pkt_size = 48;
1014                         break;
1015                 }
1016                 if (impl_ver == IMPL_XNTPD) {
1017                         fprintf(stderr,
1018                             "***Warning changing to older implementation\n");
1019                         return INFO_ERR_IMPL;
1020                 }
1021
1022                 fprintf(stderr,
1023                     "***Warning changing the request packet size from %d to %d\n",
1024                     oldsize, req_pkt_size);
1025                 goto again;
1026         }
1027
1028         /* log error message if not told to be quiet */
1029         if ((res > 0) && (((1 << res) & quiet_mask) == 0)) {
1030                 switch(res) {
1031                 case INFO_ERR_IMPL:
1032                         /* Give us a chance to try the older implementation. */
1033                         if (implcode == IMPL_XNTPD)
1034                                 break;
1035                         (void) fprintf(stderr,
1036                                        "***Server implementation incompatible with our own\n");
1037                         break;
1038                 case INFO_ERR_REQ:
1039                         (void) fprintf(stderr,
1040                                        "***Server doesn't implement this request\n");
1041                         break;
1042                 case INFO_ERR_FMT:
1043                         (void) fprintf(stderr,
1044                                        "***Server reports a format error in the received packet (shouldn't happen)\n");
1045                         break;
1046                 case INFO_ERR_NODATA:
1047                         (void) fprintf(stderr,
1048                                        "***Server reports data not found\n");
1049                         break;
1050                 case INFO_ERR_AUTH:
1051                         (void) fprintf(stderr, "***Permission denied\n");
1052                         break;
1053                 case ERR_TIMEOUT:
1054                         (void) fprintf(stderr, "***Request timed out\n");
1055                         break;
1056                 case ERR_INCOMPLETE:
1057                         (void) fprintf(stderr,
1058                                        "***Response from server was incomplete\n");
1059                         break;
1060                 default:
1061                         (void) fprintf(stderr,
1062                                        "***Server returns unknown error code %d\n", res);
1063                         break;
1064                 }
1065         }
1066         return res;
1067 }
1068
1069
1070 /*
1071  * getcmds - read commands from the standard input and execute them
1072  */
1073 static void
1074 getcmds(void)
1075 {
1076         char *  line;
1077         int     count;
1078
1079         ntp_readline_init(interactive ? prompt : NULL);
1080
1081         for (;;) {
1082                 line = ntp_readline(&count);
1083                 if (NULL == line)
1084                         break;
1085                 docmd(line);
1086                 free(line);
1087         }
1088
1089         ntp_readline_uninit();
1090 }
1091
1092
1093 #ifndef SYS_WINNT /* Under NT cannot handle SIGINT, WIN32 spawns a handler */
1094 /*
1095  * abortcmd - catch interrupts and abort the current command
1096  */
1097 static RETSIGTYPE
1098 abortcmd(
1099         int sig
1100         )
1101 {
1102
1103         if (current_output == stdout)
1104             (void) fflush(stdout);
1105         putc('\n', stderr);
1106         (void) fflush(stderr);
1107         if (jump) longjmp(interrupt_buf, 1);
1108 }
1109 #endif /* SYS_WINNT */
1110
1111 /*
1112  * docmd - decode the command line and execute a command
1113  */
1114 static void
1115 docmd(
1116         const char *cmdline
1117         )
1118 {
1119         char *tokens[1+MAXARGS+MOREARGS+2];
1120         struct parse pcmd;
1121         int ntok;
1122         int i, ti;
1123         int rval;
1124         struct xcmd *xcmd;
1125
1126         ai_fam_templ = ai_fam_default;
1127         /*
1128          * Tokenize the command line.  If nothing on it, return.
1129          */
1130         if (strlen(cmdline) >= MAXLINE) {
1131                 fprintf(stderr, "***Command ignored, more than %d characters:\n%s\n",
1132                         MAXLINE - 1, cmdline);
1133                 return;
1134         }
1135         tokenize(cmdline, tokens, &ntok);
1136         if (ntok == 0)
1137             return;
1138         
1139         /*
1140          * Find the appropriate command description.
1141          */
1142         i = findcmd(tokens[0], builtins, opcmds, &xcmd);
1143         if (i == 0) {
1144                 (void) fprintf(stderr, "***Command `%s' unknown\n",
1145                                tokens[0]);
1146                 return;
1147         } else if (i >= 2) {
1148                 (void) fprintf(stderr, "***Command `%s' ambiguous\n",
1149                                tokens[0]);
1150                 return;
1151         }
1152         
1153         /*
1154          * Save the keyword, then walk through the arguments, interpreting
1155          * as we go.
1156          */
1157         pcmd.keyword = tokens[0];
1158         pcmd.nargs = 0;
1159         ti = 1;
1160         for (i = 0; i < MAXARGS && xcmd->arg[i] != NO;) {
1161                 if ((i+ti) >= ntok) {
1162                         if (!(xcmd->arg[i] & OPT)) {
1163                                 printusage(xcmd, stderr);
1164                                 return;
1165                         }
1166                         break;
1167                 }
1168                 if ((xcmd->arg[i] & OPT) && (*tokens[i+ti] == '>'))
1169                         break;
1170                 rval = getarg(tokens[i+ti], (int)xcmd->arg[i], &pcmd.argval[i]);
1171                 if (rval == -1) {
1172                         ti++;
1173                         continue;
1174                 }
1175                 if (rval == 0)
1176                         return;
1177                 pcmd.nargs++;
1178                 i++;
1179         }
1180
1181         /* Any extra args are assumed to be "OPT|NTP_STR". */
1182         for ( ; i < MAXARGS + MOREARGS;) {
1183              if ((i+ti) >= ntok)
1184                   break;
1185                 rval = getarg(tokens[i+ti], (int)(OPT|NTP_STR), &pcmd.argval[i]);
1186                 if (rval == -1) {
1187                         ti++;
1188                         continue;
1189                 }
1190                 if (rval == 0)
1191                         return;
1192                 pcmd.nargs++;
1193                 i++;
1194         }
1195
1196         i += ti;
1197         if (i < ntok && *tokens[i] == '>') {
1198                 char *fname;
1199
1200                 if (*(tokens[i]+1) != '\0')
1201                     fname = tokens[i]+1;
1202                 else if ((i+1) < ntok)
1203                     fname = tokens[i+1];
1204                 else {
1205                         (void) fprintf(stderr, "***No file for redirect\n");
1206                         return;
1207                 }
1208
1209                 current_output = fopen(fname, "w");
1210                 if (current_output == NULL) {
1211                         (void) fprintf(stderr, "***Error opening %s: ", fname);
1212                         perror("");
1213                         return;
1214                 }
1215         } else {
1216                 current_output = stdout;
1217         }
1218
1219         if (interactive && setjmp(interrupt_buf)) {
1220                 return;
1221         } else {
1222                 jump = 1;
1223                 (xcmd->handler)(&pcmd, current_output);
1224                 jump = 0;
1225                 if (current_output != stdout)
1226                         (void) fclose(current_output);
1227                 current_output = NULL;
1228         }
1229 }
1230
1231
1232 /*
1233  * tokenize - turn a command line into tokens
1234  */
1235 static void
1236 tokenize(
1237         const char *line,
1238         char **tokens,
1239         int *ntok
1240         )
1241 {
1242         register const char *cp;
1243         register char *sp;
1244         static char tspace[MAXLINE];
1245
1246         sp = tspace;
1247         cp = line;
1248         for (*ntok = 0; *ntok < MAXTOKENS; (*ntok)++) {
1249                 tokens[*ntok] = sp;
1250                 while (ISSPACE(*cp))
1251                     cp++;
1252                 if (ISEOL(*cp))
1253                     break;
1254                 do {
1255                         *sp++ = *cp++;
1256                 } while (!ISSPACE(*cp) && !ISEOL(*cp));
1257
1258                 *sp++ = '\0';
1259         }
1260 }
1261
1262
1263
1264 /*
1265  * findcmd - find a command in a command description table
1266  */
1267 static int
1268 findcmd(
1269         register char *str,
1270         struct xcmd *clist1,
1271         struct xcmd *clist2,
1272         struct xcmd **cmd
1273         )
1274 {
1275         register struct xcmd *cl;
1276         size_t clen;
1277         int nmatch;
1278         struct xcmd *nearmatch = NULL;
1279         struct xcmd *clist;
1280
1281         clen = strlen(str);
1282         nmatch = 0;
1283         if (clist1 != 0)
1284             clist = clist1;
1285         else if (clist2 != 0)
1286             clist = clist2;
1287         else
1288             return 0;
1289
1290     again:
1291         for (cl = clist; cl->keyword != 0; cl++) {
1292                 /* do a first character check, for efficiency */
1293                 if (*str != *(cl->keyword))
1294                     continue;
1295                 if (strncmp(str, cl->keyword, (unsigned)clen) == 0) {
1296                         /*
1297                          * Could be extact match, could be approximate.
1298                          * Is exact if the length of the keyword is the
1299                          * same as the str.
1300                          */
1301                         if (*((cl->keyword) + clen) == '\0') {
1302                                 *cmd = cl;
1303                                 return 1;
1304                         }
1305                         nmatch++;
1306                         nearmatch = cl;
1307                 }
1308         }
1309
1310                                 /*
1311                                  * See if there is more to do.  If so, go again.  Sorry about the
1312                                  * goto, too much looking at BSD sources...
1313                                  */
1314         if (clist == clist1 && clist2 != 0) {
1315                 clist = clist2;
1316                 goto again;
1317         }
1318
1319                                 /*
1320                                  * If we got extactly 1 near match, use it, else return number
1321                                  * of matches.
1322                                  */
1323         if (nmatch == 1) {
1324                 *cmd = nearmatch;
1325                 return 1;
1326         }
1327         return nmatch;
1328 }
1329
1330
1331 /*
1332  * getarg - interpret an argument token
1333  *
1334  * string is always set.
1335  * type is set to the decoded type.
1336  *
1337  * return:       0 - failure
1338  *               1 - success
1339  *              -1 - skip to next token
1340  */
1341 static int
1342 getarg(
1343         char *str,
1344         int code,
1345         arg_v *argp
1346         )
1347 {
1348         int isneg;
1349         char *cp, *np;
1350         static const char *digits = "0123456789";
1351
1352         ZERO(*argp);
1353         argp->string = str;
1354         argp->type   = code & ~OPT;
1355
1356         switch (argp->type) {
1357             case NTP_STR:
1358                 break;
1359             case NTP_ADD:
1360                 if (!strcmp("-6", str)) {
1361                         ai_fam_templ = AF_INET6;
1362                         return -1;
1363                 } else if (!strcmp("-4", str)) {
1364                         ai_fam_templ = AF_INET;
1365                         return -1;
1366                 }
1367                 if (!getnetnum(str, &(argp->netnum), (char *)0, 0)) {
1368                         return 0;
1369                 }
1370                 break;
1371             case NTP_INT:
1372             case NTP_UINT:
1373                 isneg = 0;
1374                 np = str;
1375                 if (*np == '-') {
1376                         np++;
1377                         isneg = 1;
1378                 }
1379
1380                 argp->uval = 0;
1381                 do {
1382                         cp = strchr(digits, *np);
1383                         if (cp == NULL) {
1384                                 (void) fprintf(stderr,
1385                                                "***Illegal integer value %s\n", str);
1386                                 return 0;
1387                         }
1388                         argp->uval *= 10;
1389                         argp->uval += (u_long)(cp - digits);
1390                 } while (*(++np) != '\0');
1391
1392                 if (isneg) {
1393                         if ((code & ~OPT) == NTP_UINT) {
1394                                 (void) fprintf(stderr,
1395                                                "***Value %s should be unsigned\n", str);
1396                                 return 0;
1397                         }
1398                         argp->ival = -argp->ival;
1399                 }
1400                 break;
1401             case IP_VERSION:
1402                 if (!strcmp("-6", str))
1403                         argp->ival = 6 ;
1404                 else if (!strcmp("-4", str))
1405                         argp->ival = 4 ;
1406                 else {
1407                         (void) fprintf(stderr,
1408                             "***Version must be either 4 or 6\n");
1409                         return 0;
1410                 }
1411                 break;
1412         }
1413
1414         return 1;
1415 }
1416
1417
1418 /*
1419  * getnetnum - given a host name, return its net number
1420  *             and (optional) full name
1421  */
1422 static int
1423 getnetnum(
1424         const char *hname,
1425         sockaddr_u *num,
1426         char *fullhost,
1427         int af
1428         )
1429 {
1430         struct addrinfo hints, *ai = NULL;
1431
1432         ZERO(hints);
1433         hints.ai_flags = AI_CANONNAME;
1434 #ifdef AI_ADDRCONFIG
1435         hints.ai_flags |= AI_ADDRCONFIG;
1436 #endif
1437         
1438         /*
1439          * decodenetnum only works with addresses, but handles syntax
1440          * that getaddrinfo doesn't:  [2001::1]:1234
1441          */
1442         if (decodenetnum(hname, num)) {
1443                 if (fullhost != NULL)
1444                         getnameinfo(&num->sa, SOCKLEN(num), fullhost,
1445                                     LENHOSTNAME, NULL, 0, 0);
1446                 return 1;
1447         } else if (getaddrinfo(hname, "ntp", &hints, &ai) == 0) {
1448                 INSIST(sizeof(*num) >= ai->ai_addrlen);
1449                 memcpy(num, ai->ai_addr, ai->ai_addrlen);
1450                 if (fullhost != NULL) {
1451                         if (ai->ai_canonname != NULL)
1452                                 strlcpy(fullhost, ai->ai_canonname,
1453                                         LENHOSTNAME);
1454                         else
1455                                 getnameinfo(&num->sa, SOCKLEN(num),
1456                                             fullhost, LENHOSTNAME, NULL,
1457                                             0, 0);
1458                 }
1459                 return 1;
1460         }
1461         fprintf(stderr, "***Can't find host %s\n", hname);
1462
1463         return 0;
1464 }
1465
1466
1467 /*
1468  * nntohost - convert network number to host name.  This routine enforces
1469  *             the showhostnames setting.
1470  */
1471 const char *
1472 nntohost(
1473         sockaddr_u *netnum
1474         )
1475 {
1476         if (!showhostnames || SOCK_UNSPEC(netnum))
1477                 return stoa(netnum);
1478         else if (ISREFCLOCKADR(netnum))
1479                 return refnumtoa(netnum);
1480         else
1481                 return socktohost(netnum);
1482 }
1483
1484
1485 /*
1486  * Finally, the built in command handlers
1487  */
1488
1489 /*
1490  * help - tell about commands, or details of a particular command
1491  */
1492 static void
1493 help(
1494         struct parse *pcmd,
1495         FILE *fp
1496         )
1497 {
1498         struct xcmd *xcp;
1499         char *cmd;
1500         const char *list[100];
1501         size_t word, words;
1502         size_t row, rows;
1503         size_t col, cols;
1504         size_t length;
1505
1506         if (pcmd->nargs == 0) {
1507                 words = 0;
1508                 for (xcp = builtins; xcp->keyword != 0; xcp++) {
1509                         if (*(xcp->keyword) != '?')
1510                                 list[words++] = xcp->keyword;
1511                 }
1512                 for (xcp = opcmds; xcp->keyword != 0; xcp++)
1513                         list[words++] = xcp->keyword;
1514
1515                 qsort((void *)list, words, sizeof(list[0]), helpsort);
1516                 col = 0;
1517                 for (word = 0; word < words; word++) {
1518                         length = strlen(list[word]);
1519                         col = max(col, length);
1520                 }
1521
1522                 cols = SCREENWIDTH / ++col;
1523                 rows = (words + cols - 1) / cols;
1524
1525                 fprintf(fp, "ntpdc commands:\n");
1526
1527                 for (row = 0; row < rows; row++) {
1528                         for (word = row; word < words; word += rows)
1529                                 fprintf(fp, "%-*.*s", (int)col,
1530                                         (int)col - 1, list[word]);
1531                         fprintf(fp, "\n");
1532                 }
1533         } else {
1534                 cmd = pcmd->argval[0].string;
1535                 words = findcmd(cmd, builtins, opcmds, &xcp);
1536                 if (words == 0) {
1537                         fprintf(stderr,
1538                                 "Command `%s' is unknown\n", cmd);
1539                         return;
1540                 } else if (words >= 2) {
1541                         fprintf(stderr,
1542                                 "Command `%s' is ambiguous\n", cmd);
1543                         return;
1544                 }
1545                 fprintf(fp, "function: %s\n", xcp->comment);
1546                 printusage(xcp, fp);
1547         }
1548 }
1549
1550
1551 /*
1552  * helpsort - do hostname qsort comparisons
1553  */
1554 static int
1555 helpsort(
1556         const void *t1,
1557         const void *t2
1558         )
1559 {
1560         const char * const *    name1 = t1;
1561         const char * const *    name2 = t2;
1562
1563         return strcmp(*name1, *name2);
1564 }
1565
1566
1567 /*
1568  * printusage - print usage information for a command
1569  */
1570 static void
1571 printusage(
1572         struct xcmd *xcp,
1573         FILE *fp
1574         )
1575 {
1576         int i, opt46;
1577
1578         opt46 = 0;
1579         (void) fprintf(fp, "usage: %s", xcp->keyword);
1580         for (i = 0; i < MAXARGS && xcp->arg[i] != NO; i++) {
1581                 if (opt46 == 0 && (xcp->arg[i] & ~OPT) == NTP_ADD) {
1582                         (void) fprintf(fp, " [ -4|-6 ]");
1583                         opt46 = 1;
1584                 }
1585                 if (xcp->arg[i] & OPT)
1586                     (void) fprintf(fp, " [ %s ]", xcp->desc[i]);
1587                 else
1588                     (void) fprintf(fp, " %s", xcp->desc[i]);
1589         }
1590         (void) fprintf(fp, "\n");
1591 }
1592
1593
1594 /*
1595  * timeout - set time out time
1596  */
1597 static void
1598 timeout(
1599         struct parse *pcmd,
1600         FILE *fp
1601         )
1602 {
1603         int val;
1604
1605         if (pcmd->nargs == 0) {
1606                 val = tvout.tv_sec * 1000 + tvout.tv_usec / 1000;
1607                 (void) fprintf(fp, "primary timeout %d ms\n", val);
1608         } else {
1609                 tvout.tv_sec = pcmd->argval[0].uval / 1000;
1610                 tvout.tv_usec = (pcmd->argval[0].uval - (tvout.tv_sec * 1000))
1611                         * 1000;
1612         }
1613 }
1614
1615
1616 /*
1617  * my_delay - set delay for auth requests
1618  */
1619 static void
1620 my_delay(
1621         struct parse *pcmd,
1622         FILE *fp
1623         )
1624 {
1625         int isneg;
1626         u_long val;
1627
1628         if (pcmd->nargs == 0) {
1629                 val = delay_time.l_ui * 1000 + delay_time.l_uf / 4294967;
1630                 (void) fprintf(fp, "delay %lu ms\n", val);
1631         } else {
1632                 if (pcmd->argval[0].ival < 0) {
1633                         isneg = 1;
1634                         val = (u_long)(-pcmd->argval[0].ival);
1635                 } else {
1636                         isneg = 0;
1637                         val = (u_long)pcmd->argval[0].ival;
1638                 }
1639
1640                 delay_time.l_ui = val / 1000;
1641                 val %= 1000;
1642                 delay_time.l_uf = val * 4294967;        /* 2**32/1000 */
1643
1644                 if (isneg)
1645                     L_NEG(&delay_time);
1646         }
1647 }
1648
1649
1650 /*
1651  * host - set the host we are dealing with.
1652  */
1653 static void
1654 host(
1655         struct parse *pcmd,
1656         FILE *fp
1657         )
1658 {
1659         int i;
1660
1661         if (pcmd->nargs == 0) {
1662                 if (havehost)
1663                     (void) fprintf(fp, "current host is %s\n", currenthost);
1664                 else
1665                     (void) fprintf(fp, "no current host\n");
1666                 return;
1667         }
1668
1669         i = 0;
1670         if (pcmd->nargs == 2) {
1671                 if (!strcmp("-4", pcmd->argval[i].string))
1672                         ai_fam_templ = AF_INET;
1673                 else if (!strcmp("-6", pcmd->argval[i].string))
1674                         ai_fam_templ = AF_INET6;
1675                 else {
1676                         if (havehost)
1677                                 (void) fprintf(fp,
1678                                     "current host remains %s\n", currenthost);
1679                         else
1680                                 (void) fprintf(fp, "still no current host\n");
1681                         return;
1682                 }
1683                 i = 1;
1684         }
1685         if (openhost(pcmd->argval[i].string)) {
1686                 (void) fprintf(fp, "current host set to %s\n", currenthost);
1687         } else {
1688                 if (havehost)
1689                     (void) fprintf(fp,
1690                                    "current host remains %s\n", currenthost);
1691                 else
1692                     (void) fprintf(fp, "still no current host\n");
1693         }
1694 }
1695
1696
1697 /*
1698  * keyid - get a keyid to use for authenticating requests
1699  */
1700 static void
1701 keyid(
1702         struct parse *pcmd,
1703         FILE *fp
1704         )
1705 {
1706         if (pcmd->nargs == 0) {
1707                 if (info_auth_keyid == 0 && !keyid_entered)
1708                     (void) fprintf(fp, "no keyid defined\n");
1709                 else if (info_auth_keyid == 0 && keyid_entered)
1710                     (void) fprintf(fp, "no keyid will be sent\n");
1711                 else
1712                     (void) fprintf(fp, "keyid is %lu\n", (u_long)info_auth_keyid);
1713         } else {
1714                 info_auth_keyid = pcmd->argval[0].uval;
1715                 keyid_entered = 1;
1716         }
1717 }
1718
1719
1720 /*
1721  * keytype - get type of key to use for authenticating requests
1722  */
1723 static void
1724 keytype(
1725         struct parse *pcmd,
1726         FILE *fp
1727         )
1728 {
1729         const char *    digest_name;
1730         size_t          digest_len;
1731         int             key_type;
1732
1733         if (!pcmd->nargs) {
1734                 fprintf(fp, "keytype is %s with %lu octet digests\n",
1735                         keytype_name(info_auth_keytype),
1736                         (u_long)info_auth_hashlen);
1737                 return;
1738         }
1739
1740         digest_name = pcmd->argval[0].string;
1741         digest_len = 0;
1742         key_type = keytype_from_text(digest_name, &digest_len);
1743
1744         if (!key_type) {
1745                 fprintf(fp, "keytype must be 'md5'%s\n",
1746 #ifdef OPENSSL
1747                         " or a digest type provided by OpenSSL");
1748 #else
1749                         "");
1750 #endif
1751                 return;
1752         }
1753
1754         info_auth_keytype = key_type;
1755         info_auth_hashlen = digest_len;
1756 }
1757
1758
1759 /*
1760  * passwd - get an authentication key
1761  */
1762 /*ARGSUSED*/
1763 static void
1764 passwd(
1765         struct parse *pcmd,
1766         FILE *fp
1767         )
1768 {
1769         char *pass;
1770
1771         if (info_auth_keyid == 0) {
1772                 info_auth_keyid = getkeyid("Keyid: ");
1773                 if (info_auth_keyid == 0) {
1774                         (void)fprintf(fp, "Keyid must be defined\n");
1775                         return;
1776                 }
1777         }
1778         if (pcmd->nargs >= 1)
1779                 pass = pcmd->argval[0].string;
1780         else {
1781                 pass = getpass_keytype(info_auth_keytype);
1782                 if ('\0' == *pass) {
1783                         fprintf(fp, "Password unchanged\n");
1784                         return;
1785                 }
1786         }
1787         authusekey(info_auth_keyid, info_auth_keytype, (u_char *)pass);
1788         authtrust(info_auth_keyid, 1);
1789 }
1790
1791
1792 /*
1793  * hostnames - set the showhostnames flag
1794  */
1795 static void
1796 hostnames(
1797         struct parse *pcmd,
1798         FILE *fp
1799         )
1800 {
1801         if (pcmd->nargs == 0) {
1802                 if (showhostnames)
1803                     (void) fprintf(fp, "hostnames being shown\n");
1804                 else
1805                     (void) fprintf(fp, "hostnames not being shown\n");
1806         } else {
1807                 if (STREQ(pcmd->argval[0].string, "yes"))
1808                     showhostnames = 1;
1809                 else if (STREQ(pcmd->argval[0].string, "no"))
1810                     showhostnames = 0;
1811                 else
1812                     (void)fprintf(stderr, "What?\n");
1813         }
1814 }
1815
1816
1817 /*
1818  * setdebug - set/change debugging level
1819  */
1820 static void
1821 setdebug(
1822         struct parse *pcmd,
1823         FILE *fp
1824         )
1825 {
1826         if (pcmd->nargs == 0) {
1827                 (void) fprintf(fp, "debug level is %d\n", debug);
1828                 return;
1829         } else if (STREQ(pcmd->argval[0].string, "no")) {
1830                 debug = 0;
1831         } else if (STREQ(pcmd->argval[0].string, "more")) {
1832                 debug++;
1833         } else if (STREQ(pcmd->argval[0].string, "less")) {
1834                 debug--;
1835         } else {
1836                 (void) fprintf(fp, "What?\n");
1837                 return;
1838         }
1839         (void) fprintf(fp, "debug level set to %d\n", debug);
1840 }
1841
1842
1843 /*
1844  * quit - stop this nonsense
1845  */
1846 /*ARGSUSED*/
1847 static void
1848 quit(
1849         struct parse *pcmd,
1850         FILE *fp
1851         )
1852 {
1853         if (havehost)
1854             closesocket(sockfd);
1855         exit(0);
1856 }
1857
1858
1859 /*
1860  * version - print the current version number
1861  */
1862 /*ARGSUSED*/
1863 static void
1864 version(
1865         struct parse *pcmd,
1866         FILE *fp
1867         )
1868 {
1869
1870         (void) fprintf(fp, "%s\n", Version);
1871         return;
1872 }
1873
1874
1875 static void __attribute__((__format__(__printf__, 1, 0)))
1876 vwarning(const char *fmt, va_list ap)
1877 {
1878         int serrno = errno;
1879         (void) fprintf(stderr, "%s: ", progname);
1880         vfprintf(stderr, fmt, ap);
1881         (void) fprintf(stderr, ": %s\n", strerror(serrno));
1882 }
1883
1884 /*
1885  * warning - print a warning message
1886  */
1887 static void __attribute__((__format__(__printf__, 1, 2)))
1888 warning(
1889         const char *fmt,
1890         ...
1891         )
1892 {
1893         va_list ap;
1894         va_start(ap, fmt);
1895         vwarning(fmt, ap);
1896         va_end(ap);
1897 }
1898
1899
1900 /*
1901  * error - print a message and exit
1902  */
1903 static void __attribute__((__format__(__printf__, 1, 2)))
1904 error(
1905         const char *fmt,
1906         ...
1907         )
1908 {
1909         va_list ap;
1910         va_start(ap, fmt);
1911         vwarning(fmt, ap);
1912         va_end(ap);
1913         exit(1);
1914 }
1915
1916 /*
1917  * getkeyid - prompt the user for a keyid to use
1918  */
1919 static u_long
1920 getkeyid(
1921         const char *keyprompt
1922         )
1923 {
1924         int c;
1925         FILE *fi;
1926         char pbuf[20];
1927         size_t i;
1928         size_t ilim;
1929
1930 #ifndef SYS_WINNT
1931         if ((fi = fdopen(open("/dev/tty", 2), "r")) == NULL)
1932 #else
1933         if ((fi = _fdopen(open("CONIN$", _O_TEXT), "r")) == NULL)
1934 #endif /* SYS_WINNT */
1935                 fi = stdin;
1936         else
1937                 setbuf(fi, (char *)NULL);
1938         fprintf(stderr, "%s", keyprompt); fflush(stderr);
1939         for (i = 0, ilim = COUNTOF(pbuf) - 1;
1940              i < ilim && (c = getc(fi)) != '\n' && c != EOF;
1941              )
1942                 pbuf[i++] = (char)c;
1943         pbuf[i] = '\0';
1944         if (fi != stdin)
1945                 fclose(fi);
1946
1947         return (u_long) atoi(pbuf);
1948 }