contrib/ntp/ntpdc/ntpdc.c

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