contrib/ntp/ntpdc/ntpdc.c

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