xref: /freebsd/contrib/ntp/ntpq/ntpq.c (revision 4c1a82cea504df7a79f5bd8f7d0a41cacccff16e)
1 /*
2  * ntpq - query an NTP server using mode 6 commands
3  */
4 #include <config.h>
5 #include <ctype.h>
6 #include <signal.h>
7 #include <setjmp.h>
8 #include <stddef.h>
9 #include <stdio.h>
10 #include <sys/types.h>
11 #include <sys/time.h>
12 #ifdef HAVE_UNISTD_H
13 # include <unistd.h>
14 #endif
15 #ifdef HAVE_FCNTL_H
16 # include <fcntl.h>
17 #endif
18 #ifdef SYS_WINNT
19 # include <mswsock.h>
20 #endif
21 #include <isc/net.h>
22 #include <isc/result.h>
23 
24 #include "ntpq.h"
25 #include "ntp_assert.h"
26 #include "ntp_stdlib.h"
27 #include "ntp_unixtime.h"
28 #include "ntp_calendar.h"
29 #include "ntp_select.h"
30 #include "ntp_assert.h"
31 #include "lib_strbuf.h"
32 #include "ntp_lineedit.h"
33 #include "ntp_debug.h"
34 #ifdef OPENSSL
35 # include "openssl/evp.h"
36 # include "openssl/objects.h"
37 # include "openssl/err.h"
38 # ifdef SYS_WINNT
39 #  include "openssl/opensslv.h"
40 #  if !defined(HAVE_EVP_MD_DO_ALL_SORTED) && OPENSSL_VERSION_NUMBER > 0x10000000L
41 #     define HAVE_EVP_MD_DO_ALL_SORTED	1
42 #  endif
43 # endif
44 # include "libssl_compat.h"
45 # ifdef HAVE_OPENSSL_CMAC_H
46 #  include <openssl/cmac.h>
47 #  define CMAC "AES128CMAC"
48 # endif
49 #endif
50 #include <ssl_applink.c>
51 
52 #include "ntp_libopts.h"
53 #include "safecast.h"
54 
55 #ifdef SYS_VXWORKS		/* vxWorks needs mode flag -casey*/
56 # define open(name, flags)   open(name, flags, 0777)
57 # define SERVER_PORT_NUM     123
58 #endif
59 
60 /* we use COMMAND as an autogen keyword */
61 #ifdef COMMAND
62 # undef COMMAND
63 #endif
64 
65 /*
66  * Because we potentially understand a lot of commands we will run
67  * interactive if connected to a terminal.
68  */
69 int interactive = 0;		/* set to 1 when we should prompt */
70 const char *prompt = "ntpq> ";	/* prompt to ask him about */
71 
72 /*
73  * use old readvars behavior?  --old-rv processing in ntpq resets
74  * this value based on the presence or absence of --old-rv.  It is
75  * initialized to 1 here to maintain backward compatibility with
76  * libntpq clients such as ntpsnmpd, which are free to reset it as
77  * desired.
78  */
79 int	old_rv = 1;
80 
81 /*
82  * How should we display the refid?
83  * REFID_HASH, REFID_IPV4
84  */
85 te_Refid drefid = -1;
86 
87 /*
88  * for get_systime()
89  */
90 s_char	sys_precision;		/* local clock precision (log2 s) */
91 
92 /*
93  * Keyid used for authenticated requests.  Obtained on the fly.
94  */
95 u_long info_auth_keyid = 0;
96 
97 static	int	info_auth_keytype = NID_md5;	/* MD5 */
98 static	size_t	info_auth_hashlen = 16;		/* MD5 */
99 u_long	current_time;		/* needed by authkeys; not used */
100 
101 /*
102  * Flag which indicates we should always send authenticated requests
103  */
104 int always_auth = 0;
105 
106 /*
107  * Flag which indicates raw mode output.
108  */
109 int rawmode = 0;
110 
111 /*
112  * Packet version number we use
113  */
114 u_char pktversion = NTP_OLDVERSION + 1;
115 
116 
117 /*
118  * Format values
119  */
120 #define	PADDING	0
121 #define	HA	1	/* host address */
122 #define	NA	2	/* network address */
123 #define	LP	3	/* leap (print in binary) */
124 #define	RF	4	/* refid (sometimes string, sometimes not) */
125 #define	AR	5	/* array of times */
126 #define FX	6	/* test flags */
127 #define TS	7	/* l_fp timestamp in hex */
128 #define	OC	8	/* integer, print in octal */
129 #define	EOV	255	/* end of table */
130 
131 /*
132  * For the most part ntpq simply displays what ntpd provides in the
133  * mostly plain-text mode 6 responses.  A few variable names are by
134  * default "cooked" to provide more human-friendly output.
135  */
136 const var_format cookedvars[] = {
137 	{ "leap",		LP },
138 	{ "reach",		OC },
139 	{ "refid",		RF },
140 	{ "reftime",		TS },
141 	{ "clock",		TS },
142 	{ "org",		TS },
143 	{ "rec",		TS },
144 	{ "xmt",		TS },
145 	{ "flash",		FX },
146 	{ "srcadr",		HA },
147 	{ "peeradr",		HA },	/* compat with others */
148 	{ "dstadr",		NA },
149 	{ "filtdelay",		AR },
150 	{ "filtoffset",		AR },
151 	{ "filtdisp",		AR },
152 	{ "filterror",		AR },	/* compat with others */
153 };
154 
155 
156 
157 /*
158  * flasher bits
159  */
160 static const char *tstflagnames[] = {
161 	"pkt_dup",		/* TEST1 */
162 	"pkt_bogus",		/* TEST2 */
163 	"pkt_unsync",		/* TEST3 */
164 	"pkt_denied",		/* TEST4 */
165 	"pkt_auth",		/* TEST5 */
166 	"pkt_stratum",		/* TEST6 */
167 	"pkt_header",		/* TEST7 */
168 	"pkt_autokey",		/* TEST8 */
169 	"pkt_crypto",		/* TEST9 */
170 	"peer_stratum",		/* TEST10 */
171 	"peer_dist",		/* TEST11 */
172 	"peer_loop",		/* TEST12 */
173 	"peer_unreach"		/* TEST13 */
174 };
175 
176 
177 int		ntpqmain	(int,	char **);
178 /*
179  * Built in command handler declarations
180  */
181 static	int	openhost	(const char *, int);
182 static	void	dump_hex_printable(const void *, size_t);
183 static	int	sendpkt		(void *, size_t);
184 static	int	getresponse	(int, int, u_short *, size_t *, const char **, int);
185 static	int	sendrequest	(int, associd_t, int, size_t, const char *);
186 static	char *	tstflags	(u_long);
187 #ifndef BUILD_AS_LIB
188 static	void	getcmds		(void);
189 #ifndef SYS_WINNT
190 static	int	abortcmd	(void);
191 #endif	/* SYS_WINNT */
192 static	void	docmd		(const char *);
193 static	void	tokenize	(const char *, char **, int *);
194 static	int	getarg		(const char *, int, arg_v *);
195 #endif	/* BUILD_AS_LIB */
196 static	int	findcmd		(const char *, struct xcmd *,
197 				 struct xcmd *, struct xcmd **);
198 static	int	rtdatetolfp	(char *, l_fp *);
199 static	int	decodearr	(char *, int *, l_fp *, int);
200 static	void	help		(struct parse *, FILE *);
201 static	int	helpsort	(const void *, const void *);
202 static	void	printusage	(struct xcmd *, FILE *);
203 static	void	timeout		(struct parse *, FILE *);
204 static	void	auth_delay	(struct parse *, FILE *);
205 static	void	host		(struct parse *, FILE *);
206 static	void	ntp_poll	(struct parse *, FILE *);
207 static	void	keyid		(struct parse *, FILE *);
208 static	void	keytype		(struct parse *, FILE *);
209 static	void	passwd		(struct parse *, FILE *);
210 static	void	hostnames	(struct parse *, FILE *);
211 static	void	setdebug	(struct parse *, FILE *);
212 static	void	quit		(struct parse *, FILE *);
213 static	void	showdrefid	(struct parse *, FILE *);
214 static	void	version		(struct parse *, FILE *);
215 static	void	raw		(struct parse *, FILE *);
216 static	void	cooked		(struct parse *, FILE *);
217 static	void	authenticate	(struct parse *, FILE *);
218 static	void	ntpversion	(struct parse *, FILE *);
219 static	void	warning		(const char *, ...) NTP_PRINTF(1, 2);
220 static	void	error		(const char *, ...) NTP_PRINTF(1, 2);
221 static	u_long	getkeyid	(const char *);
222 static	void	atoascii	(const char *, size_t, char *, size_t);
223 static	void	cookedprint	(int, size_t, const char *, int, int, FILE *);
224 static	void	rawprint	(int, size_t, const char *, int, int, FILE *);
225 static	void	startoutput	(void);
226 static	void	output		(FILE *, const char *, const char *);
227 static	void	endoutput	(FILE *);
228 static	void	outputarr	(FILE *, char *, int, l_fp *);
229 static	int	assoccmp	(const void *, const void *);
230 	u_short	varfmt		(const char *);
231 	void	ntpq_custom_opt_handler(tOptions *, tOptDesc *);
232 
233 #ifndef BUILD_AS_LIB
234 static	char   *list_digest_names(void);
235 static	char   *insert_cmac	(char *list);
236 static	void	on_ctrlc	(void);
237 static	int	my_easprintf	(char**, const char *, ...) NTP_PRINTF(2, 3);
238 # if defined(OPENSSL) && defined(HAVE_EVP_MD_DO_ALL_SORTED)
239 static	void	list_md_fn	(const EVP_MD *m, const char *from,
240 				 const char *to, void *arg);
241 # endif /* defined(OPENSSL) && defined(HAVE_EVP_MD_DO_ALL_SORTED) */
242 #endif /* !defined(BUILD_AS_LIB) */
243 
244 
245 /* read a character from memory and expand to integer */
246 static inline int
247 pgetc(
248 	const char *cp
249 	)
250 {
251 	return (int)*(const unsigned char*)cp;
252 }
253 
254 
255 
256 /*
257  * Built-in commands we understand
258  */
259 struct xcmd builtins[] = {
260 	{ "?",		help,		{  OPT|NTP_STR, NO, NO, NO },
261 	  { "command", "", "", "" },
262 	  "tell the use and syntax of commands" },
263 	{ "help",	help,		{  OPT|NTP_STR, NO, NO, NO },
264 	  { "command", "", "", "" },
265 	  "tell the use and syntax of commands" },
266 	{ "timeout",	timeout,	{ OPT|NTP_UINT, NO, NO, NO },
267 	  { "msec", "", "", "" },
268 	  "set the primary receive time out" },
269 	{ "delay",	auth_delay,	{ OPT|NTP_INT, NO, NO, NO },
270 	  { "msec", "", "", "" },
271 	  "set the delay added to encryption time stamps" },
272 	{ "host",	host,		{ OPT|NTP_STR, OPT|NTP_STR, NO, NO },
273 	  { "-4|-6", "hostname", "", "" },
274 	  "specify the host whose NTP server we talk to" },
275 	{ "poll",	ntp_poll,	{ OPT|NTP_UINT, OPT|NTP_STR, NO, NO },
276 	  { "n", "verbose", "", "" },
277 	  "poll an NTP server in client mode `n' times" },
278 	{ "passwd",	passwd,		{ OPT|NTP_STR, NO, NO, NO },
279 	  { "", "", "", "" },
280 	  "specify a password to use for authenticated requests"},
281 	{ "hostnames",	hostnames,	{ OPT|NTP_STR, NO, NO, NO },
282 	  { "yes|no", "", "", "" },
283 	  "specify whether hostnames or net numbers are printed"},
284 	{ "debug",	setdebug,	{ OPT|NTP_STR, NO, NO, NO },
285 	  { "no|more|less", "", "", "" },
286 	  "set/change debugging level" },
287 	{ "quit",	quit,		{ NO, NO, NO, NO },
288 	  { "", "", "", "" },
289 	  "exit ntpq" },
290 	{ "exit",	quit,		{ NO, NO, NO, NO },
291 	  { "", "", "", "" },
292 	  "exit ntpq" },
293 	{ "keyid",	keyid,		{ OPT|NTP_UINT, NO, NO, NO },
294 	  { "key#", "", "", "" },
295 	  "set keyid to use for authenticated requests" },
296 	{ "drefid",	showdrefid,	{ OPT|NTP_STR, NO, NO, NO },
297 	  { "hash|ipv4", "", "", "" },
298 	  "display refid's as IPv4 or hash" },
299 	{ "version",	version,	{ NO, NO, NO, NO },
300 	  { "", "", "", "" },
301 	  "print version number" },
302 	{ "raw",	raw,		{ NO, NO, NO, NO },
303 	  { "", "", "", "" },
304 	  "do raw mode variable output" },
305 	{ "cooked",	cooked,		{ NO, NO, NO, NO },
306 	  { "", "", "", "" },
307 	  "do cooked mode variable output" },
308 	{ "authenticate", authenticate,	{ OPT|NTP_STR, NO, NO, NO },
309 	  { "yes|no", "", "", "" },
310 	  "always authenticate requests to this server" },
311 	{ "ntpversion",	ntpversion,	{ OPT|NTP_UINT, NO, NO, NO },
312 	  { "version number", "", "", "" },
313 	  "set the NTP version number to use for requests" },
314 	{ "keytype",	keytype,	{ OPT|NTP_STR, NO, NO, NO },
315 	  { "key type %s", "", "", "" },
316 	  NULL },
317 	{ 0,		0,		{ NO, NO, NO, NO },
318 	  { "", "", "", "" }, "" }
319 };
320 
321 
322 /*
323  * Default values we use.
324  */
325 #define	DEFHOST		"localhost"	/* default host name */
326 #define	DEFTIMEOUT	5		/* wait 5 seconds for 1st pkt */
327 #define	DEFSTIMEOUT	3		/* and 3 more for each additional */
328 /*
329  * Requests are automatically retried once, so total timeout with no
330  * response is a bit over 2 * DEFTIMEOUT, or 10 seconds.  At the other
331  * extreme, a request eliciting 32 packets of responses each for some
332  * reason nearly DEFSTIMEOUT seconds after the prior in that series,
333  * with a single packet dropped, would take around 32 * DEFSTIMEOUT, or
334  * 93 seconds to fail each of two times, or 186 seconds.
335  * Some commands involve a series of requests, such as "peers" and
336  * "mrulist", so the cumulative timeouts are even longer for those.
337  */
338 #define	DEFDELAY	0x51EB852	/* 20 milliseconds, l_fp fraction */
339 #define	LENHOSTNAME	256		/* host name is 256 characters long */
340 #define	MAXCMDS		100		/* maximum commands on cmd line */
341 #define	MAXHOSTS	200		/* maximum hosts on cmd line */
342 #define	MAXLINE		512		/* maximum line length */
343 #define	MAXTOKENS	(1+MAXARGS+2)	/* maximum number of usable tokens */
344 #define	MAXVARLEN	256		/* maximum length of a variable name */
345 #define	MAXVALLEN	2048		/* maximum length of a variable value */
346 #define	MAXOUTLINE	72		/* maximum length of an output line */
347 #define SCREENWIDTH	76		/* nominal screen width in columns */
348 
349 /*
350  * Some variables used and manipulated locally
351  */
352 struct sock_timeval tvout = { DEFTIMEOUT, 0 };	/* time out for reads */
353 struct sock_timeval tvsout = { DEFSTIMEOUT, 0 };/* secondary time out */
354 l_fp delay_time;				/* delay time */
355 char currenthost[LENHOSTNAME];			/* current host name */
356 int currenthostisnum;				/* is prior text from IP? */
357 struct sockaddr_in hostaddr;			/* host address */
358 int showhostnames = 1;				/* show host names by default */
359 int wideremote = 0;				/* show wide remote names? */
360 
361 int ai_fam_templ;				/* address family */
362 int ai_fam_default;				/* default address family */
363 SOCKET sockfd;					/* fd socket is opened on */
364 int havehost = 0;				/* set to 1 when host open */
365 int s_port = 0;
366 struct servent *server_entry = NULL;		/* server entry for ntp */
367 
368 
369 /*
370  * Sequence number used for requests.  It is incremented before
371  * it is used.
372  */
373 u_short sequence;
374 
375 /*
376  * Holds data returned from queries.  Declare buffer long to be sure of
377  * alignment.
378  */
379 #define	DATASIZE	(MAXFRAGS*480)	/* maximum amount of data */
380 long pktdata[DATASIZE/sizeof(long)];
381 
382 /*
383  * assoc_cache[] is a dynamic array which allows references to
384  * associations using &1 ... &N for n associations, avoiding manual
385  * lookup of the current association IDs for a given ntpd.  It also
386  * caches the status word for each association, retrieved incidentally.
387  */
388 struct association *	assoc_cache;
389 u_int assoc_cache_slots;/* count of allocated array entries */
390 u_int numassoc;		/* number of cached associations */
391 
392 /*
393  * For commands typed on the command line (with the -c option)
394  */
395 size_t numcmds = 0;
396 const char *ccmds[MAXCMDS];
397 #define	ADDCMD(cp)	if (numcmds < MAXCMDS) ccmds[numcmds++] = (cp)
398 
399 /*
400  * When multiple hosts are specified.
401  */
402 
403 u_int numhosts;
404 
405 chost chosts[MAXHOSTS];
406 #define	ADDHOST(cp)						\
407 	do {							\
408 		if (numhosts < MAXHOSTS) {			\
409 			chosts[numhosts].name = (cp);		\
410 			chosts[numhosts].fam = ai_fam_templ;	\
411 			numhosts++;				\
412 		}						\
413 	} while (0)
414 
415 /*
416  * Macro definitions we use
417  */
418 #define	ISSPACE(c)	((c) == ' ' || (c) == '\t')
419 #define	ISEOL(c)	((c) == '\n' || (c) == '\r' || (c) == '\0')
420 #define	STREQ(a, b)	(*(a) == *(b) && strcmp((a), (b)) == 0)
421 
422 /*
423  * Jump buffer for longjumping back to the command level.
424  *
425  * Since we do this from a signal handler, we use 'sig{set,long}jmp()'
426  * if available. The signal is blocked by default during the excution of
427  * a signal handler, and it is unspecified if '{set,long}jmp()' save and
428  * restore the signal mask. They do on BSD, it depends on the GLIBC
429  * version on Linux, and the gods know what happens on other OSes...
430  *
431  * So we use the 'sig{set,long}jmp()' functions where available, because
432  * for them the semantics are well-defined. If we have to fall back to
433  * '{set,long}jmp()', the CTRL-C handling might be a bit erratic.
434  */
435 #if HAVE_DECL_SIGSETJMP && HAVE_DECL_SIGLONGJMP
436 # define JMP_BUF	sigjmp_buf
437 # define SETJMP(x)	sigsetjmp((x), 1)
438 # define LONGJMP(x, v)	siglongjmp((x),(v))
439 #else
440 # define JMP_BUF	jmp_buf
441 # define SETJMP(x)	setjmp((x))
442 # define LONGJMP(x, v)	longjmp((x),(v))
443 #endif
444 static	JMP_BUF		interrupt_buf;
445 static	volatile int	jump = 0;
446 
447 /*
448  * Points at file being currently printed into
449  */
450 FILE *current_output = NULL;
451 
452 /*
453  * Command table imported from ntpdc_ops.c
454  */
455 extern struct xcmd opcmds[];
456 
457 char const *progname;
458 
459 #ifdef NO_MAIN_ALLOWED
460 #ifndef BUILD_AS_LIB
461 CALL(ntpq,"ntpq",ntpqmain);
462 
463 void clear_globals(void)
464 {
465 	extern int ntp_optind;
466 	showhostnames = 0;	/* don'tshow host names by default */
467 	ntp_optind = 0;
468 	server_entry = NULL;	/* server entry for ntp */
469 	havehost = 0;		/* set to 1 when host open */
470 	numassoc = 0;		/* number of cached associations */
471 	numcmds = 0;
472 	numhosts = 0;
473 }
474 #endif /* !BUILD_AS_LIB */
475 #endif /* NO_MAIN_ALLOWED */
476 
477 /*
478  * main - parse arguments and handle options
479  */
480 #ifndef NO_MAIN_ALLOWED
481 int
482 main(
483 	int argc,
484 	char *argv[]
485 	)
486 {
487 	return ntpqmain(argc, argv);
488 }
489 #endif
490 
491 
492 #ifndef BUILD_AS_LIB
493 int
494 ntpqmain(
495 	int argc,
496 	char *argv[]
497 	)
498 {
499 	u_int ihost;
500 	size_t icmd;
501 
502 
503 #ifdef SYS_VXWORKS
504 	clear_globals();
505 	taskPrioritySet(taskIdSelf(), 100 );
506 #endif
507 
508 	delay_time.l_ui = 0;
509 	delay_time.l_uf = DEFDELAY;
510 
511 	init_lib();	/* sets up ipv4_works, ipv6_works */
512 	ssl_applink();
513 	init_auth();
514 
515 	/* Check to see if we have IPv6. Otherwise default to IPv4 */
516 	if (!ipv6_works)
517 		ai_fam_default = AF_INET;
518 
519 	/* Fixup keytype's help based on available digest names */
520 
521 	{
522 	    char *list;
523 	    char *msg;
524 
525 	    list = list_digest_names();
526 
527 	    for (icmd = 0; icmd < sizeof(builtins)/sizeof(*builtins); icmd++) {
528 		if (strcmp("keytype", builtins[icmd].keyword) == 0) {
529 		    break;
530 		}
531 	    }
532 
533 	    /* CID: 1295478 */
534 	    /* This should only "trip" if "keytype" is removed from builtins */
535 	    INSIST(icmd < sizeof(builtins)/sizeof(*builtins));
536 
537 #ifdef OPENSSL
538 	    builtins[icmd].desc[0] = "digest-name";
539 	    my_easprintf(&msg,
540 			 "set key type to use for authenticated requests, one of:%s",
541 			 list);
542 #else
543 	    builtins[icmd].desc[0] = "md5";
544 	    my_easprintf(&msg,
545 			 "set key type to use for authenticated requests (%s)",
546 			 list);
547 #endif
548 	    builtins[icmd].comment = msg;
549 	    free(list);
550 	}
551 
552 	progname = argv[0];
553 
554 	{
555 		int optct = ntpOptionProcess(&ntpqOptions, argc, argv);
556 		argc -= optct;
557 		argv += optct;
558 	}
559 
560 	/*
561 	 * Process options other than -c and -p, which are specially
562 	 * handled by ntpq_custom_opt_handler().
563 	 */
564 
565 	debug = OPT_VALUE_SET_DEBUG_LEVEL;
566 
567 	if (HAVE_OPT(IPV4))
568 		ai_fam_templ = AF_INET;
569 	else if (HAVE_OPT(IPV6))
570 		ai_fam_templ = AF_INET6;
571 	else
572 		ai_fam_templ = ai_fam_default;
573 
574 	if (HAVE_OPT(INTERACTIVE))
575 		interactive = 1;
576 
577 	if (HAVE_OPT(NUMERIC))
578 		showhostnames = 0;
579 
580 	if (HAVE_OPT(WIDE))
581 		wideremote = 1;
582 
583 	old_rv = HAVE_OPT(OLD_RV);
584 
585 	drefid = OPT_VALUE_REFID;
586 
587 	if (0 == argc) {
588 		ADDHOST(DEFHOST);
589 	} else {
590 		for (ihost = 0; ihost < (u_int)argc; ihost++) {
591 			if ('-' == *argv[ihost]) {
592 				//
593 				// If I really cared I'd also check:
594 				// 0 == argv[ihost][2]
595 				//
596 				// and there are other cases as well...
597 				//
598 				if ('4' == argv[ihost][1]) {
599 					ai_fam_templ = AF_INET;
600 					continue;
601 				} else if ('6' == argv[ihost][1]) {
602 					ai_fam_templ = AF_INET6;
603 					continue;
604 				} else {
605 					// XXX Throw a usage error
606 				}
607 			}
608 			ADDHOST(argv[ihost]);
609 		}
610 	}
611 
612 	if (numcmds == 0 && interactive == 0
613 	    && isatty(fileno(stdin)) && isatty(fileno(stderr))) {
614 		interactive = 1;
615 	}
616 
617 	set_ctrl_c_hook(on_ctrlc);
618 #ifndef SYS_WINNT /* Under NT cannot handle SIGINT, WIN32 spawns a handler */
619 	if (interactive)
620 		push_ctrl_c_handler(abortcmd);
621 #endif /* SYS_WINNT */
622 
623 	if (numcmds == 0) {
624 		(void) openhost(chosts[0].name, chosts[0].fam);
625 		getcmds();
626 	} else {
627 		for (ihost = 0; ihost < numhosts; ihost++) {
628 			if (openhost(chosts[ihost].name, chosts[ihost].fam)) {
629 				if (ihost && current_output)
630 					fputc('\n', current_output);
631 				for (icmd = 0; icmd < numcmds; icmd++) {
632 					if (icmd && current_output)
633 						fputc('\n', current_output);
634 					docmd(ccmds[icmd]);
635 				}
636 			}
637 		}
638 	}
639 #ifdef SYS_WINNT
640 	WSACleanup();
641 #endif /* SYS_WINNT */
642 	return 0;
643 }
644 #endif /* !BUILD_AS_LIB */
645 
646 /*
647  * openhost - open a socket to a host
648  */
649 static	int
650 openhost(
651 	const char *hname,
652 	int	    fam
653 	)
654 {
655 	const char svc[] = "ntp";
656 	char temphost[LENHOSTNAME];
657 	int a_info;
658 	struct addrinfo hints, *ai;
659 	sockaddr_u addr;
660 	size_t octets;
661 	const char *cp;
662 	char name[LENHOSTNAME];
663 
664 	/*
665 	 * We need to get by the [] if they were entered
666 	 */
667 	if (*hname == '[') {
668 		cp = strchr(hname + 1, ']');
669 		if (!cp || (octets = (size_t)(cp - hname) - 1) >= sizeof(name)) {
670 			errno = EINVAL;
671 			warning("%s", "bad hostname/address");
672 			return 0;
673 		}
674 		memcpy(name, hname + 1, octets);
675 		name[octets] = '\0';
676 		hname = name;
677 	}
678 
679 	/*
680 	 * First try to resolve it as an ip address and if that fails,
681 	 * do a fullblown (dns) lookup. That way we only use the dns
682 	 * when it is needed and work around some implementations that
683 	 * will return an "IPv4-mapped IPv6 address" address if you
684 	 * give it an IPv4 address to lookup.
685 	 */
686 	ZERO(hints);
687 	hints.ai_family = fam;
688 	hints.ai_protocol = IPPROTO_UDP;
689 	hints.ai_socktype = SOCK_DGRAM;
690 	hints.ai_flags = Z_AI_NUMERICHOST;
691 	ai = NULL;
692 
693 	a_info = getaddrinfo(hname, svc, &hints, &ai);
694 	if (a_info == EAI_NONAME
695 #ifdef EAI_NODATA
696 	    || a_info == EAI_NODATA
697 #endif
698 	   ) {
699 		hints.ai_flags = AI_CANONNAME;
700 #ifdef AI_ADDRCONFIG
701 		hints.ai_flags |= AI_ADDRCONFIG;
702 #endif
703 		a_info = getaddrinfo(hname, svc, &hints, &ai);
704 	}
705 #ifdef AI_ADDRCONFIG
706 	/* Some older implementations don't like AI_ADDRCONFIG. */
707 	if (a_info == EAI_BADFLAGS) {
708 		hints.ai_flags &= ~AI_ADDRCONFIG;
709 		a_info = getaddrinfo(hname, svc, &hints, &ai);
710 	}
711 #endif
712 	if (a_info != 0) {
713 		fprintf(stderr, "%s\n", gai_strerror(a_info));
714 		return 0;
715 	}
716 
717 	INSIST(ai != NULL);
718 	ZERO(addr);
719 	octets = min(sizeof(addr), ai->ai_addrlen);
720 	memcpy(&addr, ai->ai_addr, octets);
721 
722 	if (ai->ai_canonname == NULL) {
723 		strlcpy(temphost, stoa(&addr), sizeof(temphost));
724 		currenthostisnum = TRUE;
725 	} else {
726 		strlcpy(temphost, ai->ai_canonname, sizeof(temphost));
727 		currenthostisnum = FALSE;
728 	}
729 
730 	if (debug > 2)
731 		printf("Opening host %s (%s)\n",
732 			temphost,
733 			(ai->ai_family == AF_INET)
734 			? "AF_INET"
735 			: (ai->ai_family == AF_INET6)
736 			  ? "AF_INET6"
737 			  : "AF-???"
738 			);
739 
740 	if (havehost == 1) {
741 		if (debug > 2)
742 			printf("Closing old host %s\n", currenthost);
743 		closesocket(sockfd);
744 		havehost = 0;
745 	}
746 	strlcpy(currenthost, temphost, sizeof(currenthost));
747 
748 	/* port maps to the same location in both families */
749 	s_port = NSRCPORT(&addr);
750 #ifdef SYS_VXWORKS
751 	((struct sockaddr_in6 *)&hostaddr)->sin6_port = htons(SERVER_PORT_NUM);
752 	if (ai->ai_family == AF_INET)
753 		*(struct sockaddr_in *)&hostaddr=
754 			*((struct sockaddr_in *)ai->ai_addr);
755 	else
756 		*(struct sockaddr_in6 *)&hostaddr=
757 			*((struct sockaddr_in6 *)ai->ai_addr);
758 #endif /* SYS_VXWORKS */
759 
760 #ifdef SYS_WINNT
761 	{
762 		int optionValue = SO_SYNCHRONOUS_NONALERT;
763 		int err;
764 
765 		err = setsockopt(INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE,
766 				 (void *)&optionValue, sizeof(optionValue));
767 		if (err) {
768 			mfprintf(stderr,
769 				 "setsockopt(SO_SYNCHRONOUS_NONALERT)"
770 				 " error: %m\n");
771 			freeaddrinfo(ai);
772 			exit(1);
773 		}
774 	}
775 #endif /* SYS_WINNT */
776 
777 	sockfd = socket(ai->ai_family, ai->ai_socktype,
778 			ai->ai_protocol);
779 	if (sockfd == INVALID_SOCKET) {
780 		error("socket");
781 		freeaddrinfo(ai);
782 		return 0;
783 	}
784 
785 
786 #ifdef NEED_RCVBUF_SLOP
787 # ifdef SO_RCVBUF
788 	{ int rbufsize = DATASIZE + 2048;	/* 2K for slop */
789 	if (setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF,
790 		       (void *)&rbufsize, sizeof(int)) == -1)
791 		error("setsockopt");
792 	}
793 # endif
794 #endif
795 
796 	if
797 #ifdef SYS_VXWORKS
798 	   (connect(sockfd, (struct sockaddr *)&hostaddr,
799 		    sizeof(hostaddr)) == -1)
800 #else
801 	   (connect(sockfd, (struct sockaddr *)ai->ai_addr,
802 		ai->ai_addrlen) == -1)
803 #endif /* SYS_VXWORKS */
804 	{
805 		error("connect");
806 		freeaddrinfo(ai);
807 		return 0;
808 	}
809 	freeaddrinfo(ai);
810 	havehost = 1;
811 	numassoc = 0;
812 
813 	return 1;
814 }
815 
816 
817 static void
818 dump_hex_printable(
819 	const void *	data,
820 	size_t		len
821 	)
822 {
823 	/* every line shows at most 16 bytes, so we need a buffer of
824 	 *   4 * 16 (2 xdigits, 1 char, one sep for xdigits)
825 	 * + 2 * 1  (block separators)
826 	 * + <LF> + <NUL>
827 	 *---------------
828 	 *  68 bytes
829 	 */
830 	static const char s_xdig[16] = "0123456789ABCDEF";
831 
832 	char lbuf[68];
833 	int  ch, rowlen;
834 	const u_char * cdata = data;
835 	char *xptr, *pptr;
836 
837 	while (len) {
838 		memset(lbuf, ' ', sizeof(lbuf));
839 		xptr = lbuf;
840 		pptr = lbuf + 3*16 + 2;
841 
842 		rowlen = (len > 16) ? 16 : (int)len;
843 		len -= rowlen;
844 
845 		do {
846 			ch = *cdata++;
847 
848 			*xptr++ = s_xdig[ch >> 4  ];
849 			*xptr++ = s_xdig[ch & 0x0F];
850 			if (++xptr == lbuf + 3*8)
851 				++xptr;
852 
853 			*pptr++ = isprint(ch) ? (char)ch : '.';
854 		} while (--rowlen);
855 
856 		*pptr++ = '\n';
857 		*pptr   = '\0';
858 		fputs(lbuf, stdout);
859 	}
860 }
861 
862 
863 /* XXX ELIMINATE sendpkt similar in ntpq.c, ntpdc.c, ntp_io.c, ntptrace.c */
864 /*
865  * sendpkt - send a packet to the remote host
866  */
867 static int
868 sendpkt(
869 	void *	xdata,
870 	size_t	xdatalen
871 	)
872 {
873 	if (debug >= 3)
874 		printf("Sending %zu octets\n", xdatalen);
875 
876 	if (send(sockfd, xdata, xdatalen, 0) == -1) {
877 		warning("write to %s failed", currenthost);
878 		return -1;
879 	}
880 
881 	if (debug >= 4) {
882 		printf("Request packet:\n");
883 		dump_hex_printable(xdata, xdatalen);
884 	}
885 	return 0;
886 }
887 
888 /*
889  * getresponse - get a (series of) response packet(s) and return the data
890  */
891 static int
892 getresponse(
893 	int opcode,
894 	int associd,
895 	u_short *rstatus,
896 	size_t *rsize,
897 	const char **rdata,
898 	int timeo
899 	)
900 {
901 	struct ntp_control rpkt;
902 	struct sock_timeval tvo;
903 	u_short offsets[MAXFRAGS+1];
904 	u_short counts[MAXFRAGS+1];
905 	u_short offset;
906 	u_short count;
907 	size_t numfrags;
908 	size_t f;
909 	size_t ff;
910 	int seenlastfrag;
911 	int shouldbesize;
912 	fd_set fds;
913 	int n;
914 	int errcode;
915 	/* absolute timeout checks. Not 'time_t' by intention! */
916 	uint32_t tobase;	/* base value for timeout */
917 	uint32_t tospan;	/* timeout span (max delay) */
918 	uint32_t todiff;	/* current delay */
919 
920 	memset(offsets, 0, sizeof(offsets));
921 	memset(counts , 0, sizeof(counts ));
922 
923 	/*
924 	 * This is pretty tricky.  We may get between 1 and MAXFRAG packets
925 	 * back in response to the request.  We peel the data out of
926 	 * each packet and collect it in one long block.  When the last
927 	 * packet in the sequence is received we'll know how much data we
928 	 * should have had.  Note we use one long time out, should reconsider.
929 	 */
930 	*rsize = 0;
931 	if (rstatus)
932 		*rstatus = 0;
933 	*rdata = (char *)pktdata;
934 
935 	numfrags = 0;
936 	seenlastfrag = 0;
937 
938 	tobase = (uint32_t)time(NULL);
939 
940 	FD_ZERO(&fds);
941 
942 	/*
943 	 * Loop until we have an error or a complete response.  Nearly all
944 	 * code paths to loop again use continue.
945 	 */
946 	for (;;) {
947 
948 		if (numfrags == 0)
949 			tvo = tvout;
950 		else
951 			tvo = tvsout;
952 		tospan = (uint32_t)tvo.tv_sec + (tvo.tv_usec != 0);
953 
954 		FD_SET(sockfd, &fds);
955 		n = select(sockfd+1, &fds, NULL, NULL, &tvo);
956 		if (n == -1) {
957 #if !defined(SYS_WINNT) && defined(EINTR)
958 			/* Windows does not know about EINTR (until very
959 			 * recently) and the handling of console events
960 			 * is *very* different from POSIX/UNIX signal
961 			 * handling anyway.
962 			 *
963 			 * Under non-windows targets we map EINTR as
964 			 * 'last packet was received' and try to exit
965 			 * the receive sequence.
966 			 */
967 			if (errno == EINTR) {
968 				seenlastfrag = 1;
969 				goto maybe_final;
970 			}
971 #endif
972 			warning("select fails");
973 			return -1;
974 		}
975 
976 		/*
977 		 * Check if this is already too late. Trash the data and
978 		 * fake a timeout if this is so.
979 		 */
980 		todiff = (((uint32_t)time(NULL)) - tobase) & 0x7FFFFFFFu;
981 		if ((n > 0) && (todiff > tospan)) {
982 			n = recv(sockfd, (char *)&rpkt, sizeof(rpkt), 0);
983 			n -= n; /* faked timeout return from 'select()',
984 				 * execute RMW cycle on 'n'
985 				 */
986 		}
987 
988 		if (n <= 0) {
989 			/*
990 			 * Timed out.  Return what we have
991 			 */
992 			if (numfrags == 0) {
993 				if (timeo)
994 					fprintf(stderr,
995 						"%s: timed out, nothing received\n",
996 						currenthost);
997 				return ERR_TIMEOUT;
998 			}
999 			if (timeo)
1000 				fprintf(stderr,
1001 					"%s: timed out with incomplete data\n",
1002 					currenthost);
1003 			if (debug) {
1004 				fprintf(stderr,
1005 					"ERR_INCOMPLETE: Received fragments:\n");
1006 				for (f = 0; f < numfrags; f++)
1007 					fprintf(stderr,
1008 						"%2u: %5d %5d\t%3d octets\n",
1009 						(u_int)f, offsets[f],
1010 						offsets[f] +
1011 						counts[f],
1012 						counts[f]);
1013 				fprintf(stderr,
1014 					"last fragment %sreceived\n",
1015 					(seenlastfrag)
1016 					    ? ""
1017 					    : "not ");
1018 			}
1019 			return ERR_INCOMPLETE;
1020 		}
1021 
1022 		n = recv(sockfd, (char *)&rpkt, sizeof(rpkt), 0);
1023 		if (n < 0) {
1024 			warning("read");
1025 			return -1;
1026 		}
1027 
1028 		if (debug >= 4) {
1029 			printf("Response packet:\n");
1030 			dump_hex_printable(&rpkt, n);
1031 		}
1032 
1033 		/*
1034 		 * Check for format errors.  Bug proofing.
1035 		 */
1036 		if (n < (int)CTL_HEADER_LEN) {
1037 			if (debug)
1038 				printf("Short (%d byte) packet received\n", n);
1039 			continue;
1040 		}
1041 		if (PKT_VERSION(rpkt.li_vn_mode) > NTP_VERSION
1042 		    || PKT_VERSION(rpkt.li_vn_mode) < NTP_OLDVERSION) {
1043 			if (debug)
1044 				printf("Packet received with version %d\n",
1045 				       PKT_VERSION(rpkt.li_vn_mode));
1046 			continue;
1047 		}
1048 		if (PKT_MODE(rpkt.li_vn_mode) != MODE_CONTROL) {
1049 			if (debug)
1050 				printf("Packet received with mode %d\n",
1051 				       PKT_MODE(rpkt.li_vn_mode));
1052 			continue;
1053 		}
1054 		if (!CTL_ISRESPONSE(rpkt.r_m_e_op)) {
1055 			if (debug)
1056 				printf("Received request packet, wanted response\n");
1057 			continue;
1058 		}
1059 
1060 		/*
1061 		 * Check opcode and sequence number for a match.
1062 		 * Could be old data getting to us.
1063 		 */
1064 		if (ntohs(rpkt.sequence) != sequence) {
1065 			if (debug)
1066 				printf("Received sequnce number %d, wanted %d\n",
1067 				       ntohs(rpkt.sequence), sequence);
1068 			continue;
1069 		}
1070 		if (CTL_OP(rpkt.r_m_e_op) != opcode) {
1071 			if (debug)
1072 			    printf(
1073 				    "Received opcode %d, wanted %d (sequence number okay)\n",
1074 				    CTL_OP(rpkt.r_m_e_op), opcode);
1075 			continue;
1076 		}
1077 
1078 		/*
1079 		 * Check the error code.  If non-zero, return it.
1080 		 */
1081 		if (CTL_ISERROR(rpkt.r_m_e_op)) {
1082 			errcode = (ntohs(rpkt.status) >> 8) & 0xff;
1083 			if (CTL_ISMORE(rpkt.r_m_e_op))
1084 				TRACE(1, ("Error code %d received on not-final packet\n",
1085 					  errcode));
1086 			if (errcode == CERR_UNSPEC)
1087 				return ERR_UNSPEC;
1088 			return errcode;
1089 		}
1090 
1091 		/*
1092 		 * Check the association ID to make sure it matches what
1093 		 * we sent.
1094 		 */
1095 		if (ntohs(rpkt.associd) != associd) {
1096 			TRACE(1, ("Association ID %d doesn't match expected %d\n",
1097 				  ntohs(rpkt.associd), associd));
1098 			/*
1099 			 * Hack for silly fuzzballs which, at the time of writing,
1100 			 * return an assID of sys.peer when queried for system variables.
1101 			 */
1102 #ifdef notdef
1103 			continue;
1104 #endif
1105 		}
1106 
1107 		/*
1108 		 * Collect offset and count.  Make sure they make sense.
1109 		 */
1110 		offset = ntohs(rpkt.offset);
1111 		count = ntohs(rpkt.count);
1112 
1113 		/*
1114 		 * validate received payload size is padded to next 32-bit
1115 		 * boundary and no smaller than claimed by rpkt.count
1116 		 */
1117 		if (n & 0x3) {
1118 			TRACE(1, ("Response packet not padded, size = %d\n",
1119 				  n));
1120 			continue;
1121 		}
1122 
1123 		shouldbesize = (CTL_HEADER_LEN + count + 3) & ~3;
1124 
1125 		if (n < shouldbesize) {
1126 			printf("Response packet claims %u octets payload, above %ld received\n",
1127 			       count, (long)(n - CTL_HEADER_LEN));
1128 			return ERR_INCOMPLETE;
1129 		}
1130 
1131 		if (debug >= 3 && shouldbesize > n) {
1132 			u_int32 key;
1133 			u_int32 *lpkt;
1134 			int maclen;
1135 
1136 			/*
1137 			 * Usually we ignore authentication, but for debugging purposes
1138 			 * we watch it here.
1139 			 */
1140 			/* round to 8 octet boundary */
1141 			shouldbesize = (shouldbesize + 7) & ~7;
1142 
1143 			maclen = n - shouldbesize;
1144 			if (maclen >= (int)MIN_MAC_LEN) {
1145 				printf(
1146 					"Packet shows signs of authentication (total %d, data %d, mac %d)\n",
1147 					n, shouldbesize, maclen);
1148 				lpkt = (u_int32 *)&rpkt;
1149 				printf("%08lx %08lx %08lx %08lx %08lx %08lx\n",
1150 				       (u_long)ntohl(lpkt[(n - maclen)/sizeof(u_int32) - 3]),
1151 				       (u_long)ntohl(lpkt[(n - maclen)/sizeof(u_int32) - 2]),
1152 				       (u_long)ntohl(lpkt[(n - maclen)/sizeof(u_int32) - 1]),
1153 				       (u_long)ntohl(lpkt[(n - maclen)/sizeof(u_int32)]),
1154 				       (u_long)ntohl(lpkt[(n - maclen)/sizeof(u_int32) + 1]),
1155 				       (u_long)ntohl(lpkt[(n - maclen)/sizeof(u_int32) + 2]));
1156 				key = ntohl(lpkt[(n - maclen) / sizeof(u_int32)]);
1157 				printf("Authenticated with keyid %lu\n", (u_long)key);
1158 				if (key != 0 && key != info_auth_keyid) {
1159 					printf("We don't know that key\n");
1160 				} else {
1161 					if (authdecrypt(key, (u_int32 *)&rpkt,
1162 					    n - maclen, maclen)) {
1163 						printf("Auth okay!\n");
1164 					} else {
1165 						printf("Auth failed!\n");
1166 					}
1167 				}
1168 			}
1169 		}
1170 
1171 		TRACE(2, ("Got packet, size = %d\n", n));
1172 		if (count > (n - CTL_HEADER_LEN)) {
1173 			TRACE(1, ("Received count of %u octets, data in packet is %ld\n",
1174 				  count, (long)n - CTL_HEADER_LEN));
1175 			continue;
1176 		}
1177 		if (count == 0 && CTL_ISMORE(rpkt.r_m_e_op)) {
1178 			TRACE(1, ("Received count of 0 in non-final fragment\n"));
1179 			continue;
1180 		}
1181 		if (offset + count > sizeof(pktdata)) {
1182 			TRACE(1, ("Offset %u, count %u, too big for buffer\n",
1183 				  offset, count));
1184 			return ERR_TOOMUCH;
1185 		}
1186 		if (seenlastfrag && !CTL_ISMORE(rpkt.r_m_e_op)) {
1187 			TRACE(1, ("Received second last fragment packet\n"));
1188 			continue;
1189 		}
1190 
1191 		/*
1192 		 * So far, so good.  Record this fragment, making sure it doesn't
1193 		 * overlap anything.
1194 		 */
1195 		TRACE(2, ("Packet okay\n"));
1196 
1197 		if (numfrags > (MAXFRAGS - 1)) {
1198 			TRACE(2, ("Number of fragments exceeds maximum %d\n",
1199 				  MAXFRAGS - 1));
1200 			return ERR_TOOMUCH;
1201 		}
1202 
1203 		/*
1204 		 * Find the position for the fragment relative to any
1205 		 * previously received.
1206 		 */
1207 		for (f = 0;
1208 		     f < numfrags && offsets[f] < offset;
1209 		     f++) {
1210 			/* empty body */ ;
1211 		}
1212 
1213 		if (f < numfrags && offset == offsets[f]) {
1214 			TRACE(1, ("duplicate %u octets at %u ignored, prior %u at %u\n",
1215 				  count, offset, counts[f], offsets[f]));
1216 			continue;
1217 		}
1218 
1219 		if (f > 0 && (offsets[f-1] + counts[f-1]) > offset) {
1220 			TRACE(1, ("received frag at %u overlaps with %u octet frag at %u\n",
1221 				  offset, counts[f-1], offsets[f-1]));
1222 			continue;
1223 		}
1224 
1225 		if (f < numfrags && (offset + count) > offsets[f]) {
1226 			TRACE(1, ("received %u octet frag at %u overlaps with frag at %u\n",
1227 				  count, offset, offsets[f]));
1228 			continue;
1229 		}
1230 
1231 		for (ff = numfrags; ff > f; ff--) {
1232 			offsets[ff] = offsets[ff-1];
1233 			counts[ff] = counts[ff-1];
1234 		}
1235 		offsets[f] = offset;
1236 		counts[f] = count;
1237 		numfrags++;
1238 
1239 		/*
1240 		 * Got that stuffed in right.  Figure out if this was the last.
1241 		 * Record status info out of the last packet.
1242 		 */
1243 		if (!CTL_ISMORE(rpkt.r_m_e_op)) {
1244 			seenlastfrag = 1;
1245 			if (rstatus != 0)
1246 				*rstatus = ntohs(rpkt.status);
1247 		}
1248 
1249 		/*
1250 		 * Copy the data into the data buffer, and bump the
1251 		 * timout base in case we need more.
1252 		 */
1253 		memcpy((char *)pktdata + offset, &rpkt.u, count);
1254 		tobase = (uint32_t)time(NULL);
1255 
1256 		/*
1257 		 * If we've seen the last fragment, look for holes in the sequence.
1258 		 * If there aren't any, we're done.
1259 		 */
1260 #if !defined(SYS_WINNT) && defined(EINTR)
1261 		maybe_final:
1262 #endif
1263 
1264 		if (seenlastfrag && offsets[0] == 0) {
1265 			for (f = 1; f < numfrags; f++)
1266 				if (offsets[f-1] + counts[f-1] !=
1267 				    offsets[f])
1268 					break;
1269 			if (f == numfrags) {
1270 				*rsize = offsets[f-1] + counts[f-1];
1271 				TRACE(1, ("%lu packets reassembled into response\n",
1272 					  (u_long)numfrags));
1273 				return 0;
1274 			}
1275 		}
1276 	}  /* giant for (;;) collecting response packets */
1277 }  /* getresponse() */
1278 
1279 
1280 /*
1281  * sendrequest - format and send a request packet
1282  */
1283 static int
1284 sendrequest(
1285 	int opcode,
1286 	associd_t associd,
1287 	int auth,
1288 	size_t qsize,
1289 	const char *qdata
1290 	)
1291 {
1292 	struct ntp_control qpkt;
1293 	size_t	pktsize;
1294 	u_long	key_id;
1295 	char *	pass;
1296 	size_t	maclen;
1297 
1298 	/*
1299 	 * Check to make sure the data will fit in one packet
1300 	 */
1301 	if (qsize > CTL_MAX_DATA_LEN) {
1302 		fprintf(stderr,
1303 			"***Internal error!  qsize (%zu) too large\n",
1304 			qsize);
1305 		return 1;
1306 	}
1307 
1308 	/*
1309 	 * Fill in the packet
1310 	 */
1311 	qpkt.li_vn_mode = PKT_LI_VN_MODE(0, pktversion, MODE_CONTROL);
1312 	qpkt.r_m_e_op = (u_char)(opcode & CTL_OP_MASK);
1313 	qpkt.sequence = htons(sequence);
1314 	qpkt.status = 0;
1315 	qpkt.associd = htons((u_short)associd);
1316 	qpkt.offset = 0;
1317 	qpkt.count = htons((u_short)qsize);
1318 
1319 	pktsize = CTL_HEADER_LEN;
1320 
1321 	/*
1322 	 * If we have data, copy and pad it out to a 32-bit boundary.
1323 	 */
1324 	if (qsize > 0) {
1325 		memcpy(&qpkt.u, qdata, (size_t)qsize);
1326 		pktsize += qsize;
1327 		while (pktsize & (sizeof(u_int32) - 1)) {
1328 			qpkt.u.data[qsize++] = 0;
1329 			pktsize++;
1330 		}
1331 	}
1332 
1333 	/*
1334 	 * If it isn't authenticated we can just send it.  Otherwise
1335 	 * we're going to have to think about it a little.
1336 	 */
1337 	if (!auth && !always_auth) {
1338 		return sendpkt(&qpkt, pktsize);
1339 	}
1340 
1341 	/*
1342 	 * Pad out packet to a multiple of 8 octets to be sure
1343 	 * receiver can handle it.
1344 	 */
1345 	while (pktsize & 7) {
1346 		qpkt.u.data[qsize++] = 0;
1347 		pktsize++;
1348 	}
1349 
1350 	/*
1351 	 * Get the keyid and the password if we don't have one.
1352 	 */
1353 	if (info_auth_keyid == 0) {
1354 		key_id = getkeyid("Keyid: ");
1355 		if (key_id == 0 || key_id > NTP_MAXKEY) {
1356 			fprintf(stderr,
1357 				"Invalid key identifier\n");
1358 			return 1;
1359 		}
1360 		info_auth_keyid = key_id;
1361 	}
1362 	if (!authistrusted(info_auth_keyid)) {
1363 		pass = getpass_keytype(info_auth_keytype);
1364 		if ('\0' == pass[0]) {
1365 			fprintf(stderr, "Invalid password\n");
1366 			return 1;
1367 		}
1368 		authusekey(info_auth_keyid, info_auth_keytype,
1369 			   (u_char *)pass);
1370 		authtrust(info_auth_keyid, 1);
1371 	}
1372 
1373 	/*
1374 	 * Do the encryption.
1375 	 */
1376 	maclen = authencrypt(info_auth_keyid, (void *)&qpkt, pktsize);
1377 	if (!maclen) {
1378 		fprintf(stderr, "Key not found\n");
1379 		return 1;
1380 	} else if ((size_t)maclen != (info_auth_hashlen + sizeof(keyid_t))) {
1381 		fprintf(stderr,
1382 			"%zu octet MAC, %zu expected with %zu octet digest\n",
1383 			maclen, (info_auth_hashlen + sizeof(keyid_t)),
1384 			info_auth_hashlen);
1385 		return 1;
1386 	}
1387 
1388 	return sendpkt((char *)&qpkt, pktsize + maclen);
1389 }
1390 
1391 
1392 /*
1393  * show_error_msg - display the error text for a mode 6 error response.
1394  */
1395 void
1396 show_error_msg(
1397 	int		m6resp,
1398 	associd_t	associd
1399 	)
1400 {
1401 	if (numhosts > 1)
1402 		fprintf(stderr, "server=%s ", currenthost);
1403 
1404 	switch (m6resp) {
1405 
1406 	case CERR_BADFMT:
1407 		fprintf(stderr,
1408 		    "***Server reports a bad format request packet\n");
1409 		break;
1410 
1411 	case CERR_PERMISSION:
1412 		fprintf(stderr,
1413 		    "***Server disallowed request (authentication?)\n");
1414 		break;
1415 
1416 	case CERR_BADOP:
1417 		fprintf(stderr,
1418 		    "***Server reports a bad opcode in request\n");
1419 		break;
1420 
1421 	case CERR_BADASSOC:
1422 		fprintf(stderr,
1423 		    "***Association ID %d unknown to server\n",
1424 		    associd);
1425 		break;
1426 
1427 	case CERR_UNKNOWNVAR:
1428 		fprintf(stderr,
1429 		    "***A request variable unknown to the server\n");
1430 		break;
1431 
1432 	case CERR_BADVALUE:
1433 		fprintf(stderr,
1434 		    "***Server indicates a request variable was bad\n");
1435 		break;
1436 
1437 	case ERR_UNSPEC:
1438 		fprintf(stderr,
1439 		    "***Server returned an unspecified error\n");
1440 		break;
1441 
1442 	case ERR_TIMEOUT:
1443 		fprintf(stderr, "***Request timed out\n");
1444 		break;
1445 
1446 	case ERR_INCOMPLETE:
1447 		fprintf(stderr,
1448 		    "***Response from server was incomplete\n");
1449 		break;
1450 
1451 	case ERR_TOOMUCH:
1452 		fprintf(stderr,
1453 		    "***Buffer size exceeded for returned data\n");
1454 		break;
1455 
1456 	default:
1457 		fprintf(stderr,
1458 		    "***Server returns unknown error code %d\n",
1459 		    m6resp);
1460 	}
1461 }
1462 
1463 /*
1464  * doquery - send a request and process the response, displaying
1465  *	     error messages for any error responses.
1466  */
1467 int
1468 doquery(
1469 	int opcode,
1470 	associd_t associd,
1471 	int auth,
1472 	size_t qsize,
1473 	const char *qdata,
1474 	u_short *rstatus,
1475 	size_t *rsize,
1476 	const char **rdata
1477 	)
1478 {
1479 	return doqueryex(opcode, associd, auth, qsize, qdata, rstatus,
1480 			 rsize, rdata, FALSE);
1481 }
1482 
1483 
1484 /*
1485  * doqueryex - send a request and process the response, optionally
1486  *	       displaying error messages for any error responses.
1487  */
1488 int
1489 doqueryex(
1490 	int opcode,
1491 	associd_t associd,
1492 	int auth,
1493 	size_t qsize,
1494 	const char *qdata,
1495 	u_short *rstatus,
1496 	size_t *rsize,
1497 	const char **rdata,
1498 	int quiet
1499 	)
1500 {
1501 	int res;
1502 	int done;
1503 
1504 	/*
1505 	 * Check to make sure host is open
1506 	 */
1507 	if (!havehost) {
1508 		fprintf(stderr, "***No host open, use `host' command\n");
1509 		return -1;
1510 	}
1511 
1512 	done = 0;
1513 	sequence++;
1514 
1515     again:
1516 	/*
1517 	 * send a request
1518 	 */
1519 	res = sendrequest(opcode, associd, auth, qsize, qdata);
1520 	if (res != 0)
1521 		return res;
1522 
1523 	/*
1524 	 * Get the response.  If we got a standard error, print a message
1525 	 */
1526 	res = getresponse(opcode, associd, rstatus, rsize, rdata, done);
1527 
1528 	if (res > 0) {
1529 		if (!done && (res == ERR_TIMEOUT || res == ERR_INCOMPLETE)) {
1530 			if (res == ERR_INCOMPLETE) {
1531 				/*
1532 				 * better bump the sequence so we don't
1533 				 * get confused about differing fragments.
1534 				 */
1535 				sequence++;
1536 			}
1537 			done = 1;
1538 			goto again;
1539 		}
1540 		if (!quiet)
1541 			show_error_msg(res, associd);
1542 
1543 	}
1544 	return res;
1545 }
1546 
1547 
1548 #ifndef BUILD_AS_LIB
1549 /*
1550  * getcmds - read commands from the standard input and execute them
1551  */
1552 static void
1553 getcmds(void)
1554 {
1555 	char *	line;
1556 	int	count;
1557 
1558 	ntp_readline_init(interactive ? prompt : NULL);
1559 
1560 	for (;;) {
1561 		line = ntp_readline(&count);
1562 		if (NULL == line)
1563 			break;
1564 		docmd(line);
1565 		free(line);
1566 	}
1567 
1568 	ntp_readline_uninit();
1569 }
1570 #endif /* !BUILD_AS_LIB */
1571 
1572 
1573 #if !defined(SYS_WINNT) && !defined(BUILD_AS_LIB)
1574 /*
1575  * abortcmd - catch interrupts and abort the current command
1576  */
1577 static int
1578 abortcmd(void)
1579 {
1580 	if (current_output == stdout)
1581 		(void) fflush(stdout);
1582 	putc('\n', stderr);
1583 	(void) fflush(stderr);
1584 	if (jump) {
1585 		jump = 0;
1586 		LONGJMP(interrupt_buf, 1);
1587 	}
1588 	return TRUE;
1589 }
1590 #endif	/* !SYS_WINNT && !BUILD_AS_LIB */
1591 
1592 
1593 #ifndef	BUILD_AS_LIB
1594 /*
1595  * docmd - decode the command line and execute a command
1596  */
1597 static void
1598 docmd(
1599 	const char *cmdline
1600 	)
1601 {
1602 	char *tokens[1+MAXARGS+2];
1603 	struct parse pcmd;
1604 	int ntok;
1605 	static int i;
1606 	struct xcmd *xcmd;
1607 
1608 	/*
1609 	 * Tokenize the command line.  If nothing on it, return.
1610 	 */
1611 	tokenize(cmdline, tokens, &ntok);
1612 	if (ntok == 0)
1613 	    return;
1614 
1615 	/*
1616 	 * Find the appropriate command description.
1617 	 */
1618 	i = findcmd(tokens[0], builtins, opcmds, &xcmd);
1619 	if (i == 0) {
1620 		(void) fprintf(stderr, "***Command `%s' unknown\n",
1621 			       tokens[0]);
1622 		return;
1623 	} else if (i >= 2) {
1624 		(void) fprintf(stderr, "***Command `%s' ambiguous\n",
1625 			       tokens[0]);
1626 		return;
1627 	}
1628 
1629 	/* Warn about ignored extra args */
1630 	for (i = MAXARGS + 1; i < ntok ; ++i) {
1631 		fprintf(stderr, "***Extra arg `%s' ignored\n", tokens[i]);
1632 	}
1633 
1634 	/*
1635 	 * Save the keyword, then walk through the arguments, interpreting
1636 	 * as we go.
1637 	 */
1638 	pcmd.keyword = tokens[0];
1639 	pcmd.nargs = 0;
1640 	for (i = 0; i < MAXARGS && xcmd->arg[i] != NO; i++) {
1641 		if ((i+1) >= ntok) {
1642 			if (!(xcmd->arg[i] & OPT)) {
1643 				printusage(xcmd, stderr);
1644 				return;
1645 			}
1646 			break;
1647 		}
1648 		if ((xcmd->arg[i] & OPT) && (*tokens[i+1] == '>'))
1649 			break;
1650 		if (!getarg(tokens[i+1], (int)xcmd->arg[i], &pcmd.argval[i]))
1651 			return;
1652 		pcmd.nargs++;
1653 	}
1654 
1655 	i++;
1656 	if (i < ntok && *tokens[i] == '>') {
1657 		char *fname;
1658 
1659 		if (*(tokens[i]+1) != '\0')
1660 			fname = tokens[i]+1;
1661 		else if ((i+1) < ntok)
1662 			fname = tokens[i+1];
1663 		else {
1664 			(void) fprintf(stderr, "***No file for redirect\n");
1665 			return;
1666 		}
1667 
1668 		current_output = fopen(fname, "w");
1669 		if (current_output == NULL) {
1670 			(void) fprintf(stderr, "***Error opening %s: ", fname);
1671 			perror("");
1672 			return;
1673 		}
1674 	} else {
1675 		current_output = stdout;
1676 	}
1677 
1678 	if (interactive) {
1679 		if ( ! SETJMP(interrupt_buf)) {
1680 			jump = 1;
1681 			(xcmd->handler)(&pcmd, current_output);
1682 			jump = 0;
1683 		} else {
1684 			fflush(current_output);
1685 			fputs("\n >>> command aborted <<<\n", stderr);
1686 			fflush(stderr);
1687 		}
1688 
1689 	} else {
1690 		jump = 0;
1691 		(xcmd->handler)(&pcmd, current_output);
1692 	}
1693 	if ((NULL != current_output) && (stdout != current_output)) {
1694 		(void)fclose(current_output);
1695 		current_output = NULL;
1696 	}
1697 }
1698 
1699 
1700 /*
1701  * tokenize - turn a command line into tokens
1702  *
1703  * SK: Modified to allow a quoted string
1704  *
1705  * HMS: If the first character of the first token is a ':' then (after
1706  * eating inter-token whitespace) the 2nd token is the rest of the line.
1707  */
1708 
1709 static void
1710 tokenize(
1711 	const char *line,
1712 	char **tokens,
1713 	int *ntok
1714 	)
1715 {
1716 	register const char *cp;
1717 	register char *sp;
1718 	static char tspace[MAXLINE];
1719 
1720 	sp = tspace;
1721 	cp = line;
1722 	for (*ntok = 0; *ntok < MAXTOKENS; (*ntok)++) {
1723 		tokens[*ntok] = sp;
1724 
1725 		/* Skip inter-token whitespace */
1726 		while (ISSPACE(*cp))
1727 		    cp++;
1728 
1729 		/* If we're at EOL we're done */
1730 		if (ISEOL(*cp))
1731 		    break;
1732 
1733 		/* If this is the 2nd token and the first token begins
1734 		 * with a ':', then just grab to EOL.
1735 		 */
1736 
1737 		if (*ntok == 1 && tokens[0][0] == ':') {
1738 			do {
1739 				if (sp - tspace >= MAXLINE)
1740 					goto toobig;
1741 				*sp++ = *cp++;
1742 			} while (!ISEOL(*cp));
1743 		}
1744 
1745 		/* Check if this token begins with a double quote.
1746 		 * If yes, continue reading till the next double quote
1747 		 */
1748 		else if (*cp == '\"') {
1749 			++cp;
1750 			do {
1751 				if (sp - tspace >= MAXLINE)
1752 					goto toobig;
1753 				*sp++ = *cp++;
1754 			} while ((*cp != '\"') && !ISEOL(*cp));
1755 			/* HMS: a missing closing " should be an error */
1756 		}
1757 		else {
1758 			do {
1759 				if (sp - tspace >= MAXLINE)
1760 					goto toobig;
1761 				*sp++ = *cp++;
1762 			} while ((*cp != '\"') && !ISSPACE(*cp) && !ISEOL(*cp));
1763 			/* HMS: Why check for a " in the previous line? */
1764 		}
1765 
1766 		if (sp - tspace >= MAXLINE)
1767 			goto toobig;
1768 		*sp++ = '\0';
1769 	}
1770 	return;
1771 
1772   toobig:
1773 	*ntok = 0;
1774 	fprintf(stderr,
1775 		"***Line `%s' is too big\n",
1776 		line);
1777 	return;
1778 }
1779 
1780 
1781 /*
1782  * getarg - interpret an argument token
1783  */
1784 static int
1785 getarg(
1786 	const char *str,
1787 	int code,
1788 	arg_v *argp
1789 	)
1790 {
1791 	u_long ul;
1792 
1793 	switch (code & ~OPT) {
1794 	case NTP_STR:
1795 		argp->string = str;
1796 		break;
1797 
1798 	case NTP_ADD:
1799 		if (!getnetnum(str, &argp->netnum, NULL, 0))
1800 			return 0;
1801 		break;
1802 
1803 	case NTP_UINT:
1804 		if ('&' == str[0]) {
1805 			if (!atouint(&str[1], &ul)) {
1806 				fprintf(stderr,
1807 					"***Association index `%s' invalid/undecodable\n",
1808 					str);
1809 				return 0;
1810 			}
1811 			if (0 == numassoc) {
1812 				dogetassoc(stdout);
1813 				if (0 == numassoc) {
1814 					fprintf(stderr,
1815 						"***No associations found, `%s' unknown\n",
1816 						str);
1817 					return 0;
1818 				}
1819 			}
1820 			ul = min(ul, numassoc);
1821 			argp->uval = assoc_cache[ul - 1].assid;
1822 			break;
1823 		}
1824 		if (!atouint(str, &argp->uval)) {
1825 			fprintf(stderr, "***Illegal unsigned value %s\n",
1826 				str);
1827 			return 0;
1828 		}
1829 		break;
1830 
1831 	case NTP_INT:
1832 		if (!atoint(str, &argp->ival)) {
1833 			fprintf(stderr, "***Illegal integer value %s\n",
1834 				str);
1835 			return 0;
1836 		}
1837 		break;
1838 
1839 	case IP_VERSION:
1840 		if (!strcmp("-6", str)) {
1841 			argp->ival = 6;
1842 		} else if (!strcmp("-4", str)) {
1843 			argp->ival = 4;
1844 		} else {
1845 			fprintf(stderr, "***Version must be either 4 or 6\n");
1846 			return 0;
1847 		}
1848 		break;
1849 	}
1850 
1851 	return 1;
1852 }
1853 #endif	/* !BUILD_AS_LIB */
1854 
1855 
1856 /*
1857  * findcmd - find a command in a command description table
1858  */
1859 static int
1860 findcmd(
1861 	const char *	str,
1862 	struct xcmd *	clist1,
1863 	struct xcmd *	clist2,
1864 	struct xcmd **	cmd
1865 	)
1866 {
1867 	struct xcmd *cl;
1868 	size_t clen;
1869 	int nmatch;
1870 	struct xcmd *nearmatch = NULL;
1871 	struct xcmd *clist;
1872 
1873 	clen = strlen(str);
1874 	nmatch = 0;
1875 	if (clist1 != 0)
1876 	    clist = clist1;
1877 	else if (clist2 != 0)
1878 	    clist = clist2;
1879 	else
1880 	    return 0;
1881 
1882     again:
1883 	for (cl = clist; cl->keyword != 0; cl++) {
1884 		/* do a first character check, for efficiency */
1885 		if (*str != *(cl->keyword))
1886 		    continue;
1887 		if (strncmp(str, cl->keyword, (unsigned)clen) == 0) {
1888 			/*
1889 			 * Could be extact match, could be approximate.
1890 			 * Is exact if the length of the keyword is the
1891 			 * same as the str.
1892 			 */
1893 			if (*((cl->keyword) + clen) == '\0') {
1894 				*cmd = cl;
1895 				return 1;
1896 			}
1897 			nmatch++;
1898 			nearmatch = cl;
1899 		}
1900 	}
1901 
1902 	/*
1903 	 * See if there is more to do.  If so, go again.  Sorry about the
1904 	 * goto, too much looking at BSD sources...
1905 	 */
1906 	if (clist == clist1 && clist2 != 0) {
1907 		clist = clist2;
1908 		goto again;
1909 	}
1910 
1911 	/*
1912 	 * If we got extactly 1 near match, use it, else return number
1913 	 * of matches.
1914 	 */
1915 	if (nmatch == 1) {
1916 		*cmd = nearmatch;
1917 		return 1;
1918 	}
1919 	return nmatch;
1920 }
1921 
1922 
1923 /*
1924  * getnetnum - given a host name, return its net number
1925  *	       and (optional) full name
1926  */
1927 int
1928 getnetnum(
1929 	const char *hname,
1930 	sockaddr_u *num,
1931 	char *fullhost,
1932 	int af
1933 	)
1934 {
1935 	struct addrinfo hints, *ai = NULL;
1936 
1937 	ZERO(hints);
1938 	hints.ai_flags = AI_CANONNAME;
1939 #ifdef AI_ADDRCONFIG
1940 	hints.ai_flags |= AI_ADDRCONFIG;
1941 #endif
1942 
1943 	/*
1944 	 * decodenetnum only works with addresses, but handles syntax
1945 	 * that getaddrinfo doesn't:  [2001::1]:1234
1946 	 */
1947 	if (decodenetnum(hname, num)) {
1948 		if (fullhost != NULL)
1949 			getnameinfo(&num->sa, SOCKLEN(num), fullhost,
1950 				    LENHOSTNAME, NULL, 0, 0);
1951 		return 1;
1952 	} else if (getaddrinfo(hname, "ntp", &hints, &ai) == 0) {
1953 		INSIST(sizeof(*num) >= ai->ai_addrlen);
1954 		memcpy(num, ai->ai_addr, ai->ai_addrlen);
1955 		if (fullhost != NULL) {
1956 			if (ai->ai_canonname != NULL)
1957 				strlcpy(fullhost, ai->ai_canonname,
1958 					LENHOSTNAME);
1959 			else
1960 				getnameinfo(&num->sa, SOCKLEN(num),
1961 					    fullhost, LENHOSTNAME, NULL,
1962 					    0, 0);
1963 		}
1964 		freeaddrinfo(ai);
1965 		return 1;
1966 	}
1967 	fprintf(stderr, "***Can't find host %s\n", hname);
1968 
1969 	return 0;
1970 }
1971 
1972 
1973 /*
1974  * nntohost - convert network number to host name.  This routine enforces
1975  *	       the showhostnames setting.
1976  */
1977 const char *
1978 nntohost(
1979 	sockaddr_u *netnum
1980 	)
1981 {
1982 	return nntohost_col(netnum, LIB_BUFLENGTH - 1, FALSE);
1983 }
1984 
1985 
1986 /*
1987  * nntohost_col - convert network number to host name in fixed width.
1988  *		  This routine enforces the showhostnames setting.
1989  *		  When displaying hostnames longer than the width,
1990  *		  the first part of the hostname is displayed.  When
1991  *		  displaying numeric addresses longer than the width,
1992  *		  Such as IPv6 addresses, the caller decides whether
1993  *		  the first or last of the numeric address is used.
1994  */
1995 const char *
1996 nntohost_col(
1997 	sockaddr_u *	addr,
1998 	size_t		width,
1999 	int		preserve_lowaddrbits
2000 	)
2001 {
2002 	const char *	out;
2003 
2004 	if (!showhostnames || SOCK_UNSPEC(addr)) {
2005 		if (preserve_lowaddrbits)
2006 			out = trunc_left(stoa(addr), width);
2007 		else
2008 			out = trunc_right(stoa(addr), width);
2009 	} else if (ISREFCLOCKADR(addr)) {
2010 		out = refnumtoa(addr);
2011 	} else {
2012 		out = trunc_right(socktohost(addr), width);
2013 	}
2014 	return out;
2015 }
2016 
2017 
2018 /*
2019  * nntohostp() is the same as nntohost() plus a :port suffix
2020  */
2021 const char *
2022 nntohostp(
2023 	sockaddr_u *netnum
2024 	)
2025 {
2026 	const char *	hostn;
2027 	char *		buf;
2028 
2029 	if (!showhostnames || SOCK_UNSPEC(netnum))
2030 		return sptoa(netnum);
2031 	else if (ISREFCLOCKADR(netnum))
2032 		return refnumtoa(netnum);
2033 
2034 	hostn = socktohost(netnum);
2035 	LIB_GETBUF(buf);
2036 	snprintf(buf, LIB_BUFLENGTH, "%s:%u", hostn, SRCPORT(netnum));
2037 
2038 	return buf;
2039 }
2040 
2041 /*
2042  * rtdatetolfp - decode an RT-11 date into an l_fp
2043  */
2044 static int
2045 rtdatetolfp(
2046 	char *str,
2047 	l_fp *lfp
2048 	)
2049 {
2050 	register char *cp;
2051 	register int i;
2052 	struct calendar cal;
2053 	char buf[4];
2054 
2055 	cal.yearday = 0;
2056 
2057 	/*
2058 	 * An RT-11 date looks like:
2059 	 *
2060 	 * d[d]-Mth-y[y] hh:mm:ss
2061 	 *
2062 	 * (No docs, but assume 4-digit years are also legal...)
2063 	 *
2064 	 * d[d]-Mth-y[y[y[y]]] hh:mm:ss
2065 	 */
2066 	cp = str;
2067 	if (!isdigit(pgetc(cp))) {
2068 		if (*cp == '-') {
2069 			/*
2070 			 * Catch special case
2071 			 */
2072 			L_CLR(lfp);
2073 			return 1;
2074 		}
2075 		return 0;
2076 	}
2077 
2078 	cal.monthday = (u_char) (*cp++ - '0');	/* ascii dependent */
2079 	if (isdigit(pgetc(cp))) {
2080 		cal.monthday = (u_char)((cal.monthday << 3) + (cal.monthday << 1));
2081 		cal.monthday = (u_char)(cal.monthday + *cp++ - '0');
2082 	}
2083 
2084 	if (*cp++ != '-')
2085 	    return 0;
2086 
2087 	for (i = 0; i < 3; i++)
2088 	    buf[i] = *cp++;
2089 	buf[3] = '\0';
2090 
2091 	for (i = 0; i < 12; i++)
2092 	    if (STREQ(buf, months[i]))
2093 		break;
2094 	if (i == 12)
2095 	    return 0;
2096 	cal.month = (u_char)(i + 1);
2097 
2098 	if (*cp++ != '-')
2099 	    return 0;
2100 
2101 	if (!isdigit(pgetc(cp)))
2102 	    return 0;
2103 	cal.year = (u_short)(*cp++ - '0');
2104 	if (isdigit(pgetc(cp))) {
2105 		cal.year = (u_short)((cal.year << 3) + (cal.year << 1));
2106 		cal.year = (u_short)(*cp++ - '0');
2107 	}
2108 	if (isdigit(pgetc(cp))) {
2109 		cal.year = (u_short)((cal.year << 3) + (cal.year << 1));
2110 		cal.year = (u_short)(cal.year + *cp++ - '0');
2111 	}
2112 	if (isdigit(pgetc(cp))) {
2113 		cal.year = (u_short)((cal.year << 3) + (cal.year << 1));
2114 		cal.year = (u_short)(cal.year + *cp++ - '0');
2115 	}
2116 
2117 	/*
2118 	 * Catch special case.  If cal.year == 0 this is a zero timestamp.
2119 	 */
2120 	if (cal.year == 0) {
2121 		L_CLR(lfp);
2122 		return 1;
2123 	}
2124 
2125 	if (*cp++ != ' ' || !isdigit(pgetc(cp)))
2126 	    return 0;
2127 	cal.hour = (u_char)(*cp++ - '0');
2128 	if (isdigit(pgetc(cp))) {
2129 		cal.hour = (u_char)((cal.hour << 3) + (cal.hour << 1));
2130 		cal.hour = (u_char)(cal.hour + *cp++ - '0');
2131 	}
2132 
2133 	if (*cp++ != ':' || !isdigit(pgetc(cp)))
2134 	    return 0;
2135 	cal.minute = (u_char)(*cp++ - '0');
2136 	if (isdigit(pgetc(cp))) {
2137 		cal.minute = (u_char)((cal.minute << 3) + (cal.minute << 1));
2138 		cal.minute = (u_char)(cal.minute + *cp++ - '0');
2139 	}
2140 
2141 	if (*cp++ != ':' || !isdigit(pgetc(cp)))
2142 	    return 0;
2143 	cal.second = (u_char)(*cp++ - '0');
2144 	if (isdigit(pgetc(cp))) {
2145 		cal.second = (u_char)((cal.second << 3) + (cal.second << 1));
2146 		cal.second = (u_char)(cal.second + *cp++ - '0');
2147 	}
2148 
2149 	/*
2150 	 * For RT-11, 1972 seems to be the pivot year
2151 	 */
2152 	if (cal.year < 72)
2153 		cal.year += 2000;
2154 	if (cal.year < 100)
2155 		cal.year += 1900;
2156 
2157 	lfp->l_ui = caltontp(&cal);
2158 	lfp->l_uf = 0;
2159 	return 1;
2160 }
2161 
2162 
2163 /*
2164  * decodets - decode a timestamp into an l_fp format number, with
2165  *	      consideration of fuzzball formats.
2166  */
2167 int
2168 decodets(
2169 	char *str,
2170 	l_fp *lfp
2171 	)
2172 {
2173 	char *cp;
2174 	char buf[30];
2175 	size_t b;
2176 
2177 	/*
2178 	 * If it starts with a 0x, decode as hex.
2179 	 */
2180 	if (*str == '0' && (*(str+1) == 'x' || *(str+1) == 'X'))
2181 		return hextolfp(str+2, lfp);
2182 
2183 	/*
2184 	 * If it starts with a '"', try it as an RT-11 date.
2185 	 */
2186 	if (*str == '"') {
2187 		cp = str + 1;
2188 		b = 0;
2189 		while ('"' != *cp && '\0' != *cp &&
2190 		       b < COUNTOF(buf) - 1)
2191 			buf[b++] = *cp++;
2192 		buf[b] = '\0';
2193 		return rtdatetolfp(buf, lfp);
2194 	}
2195 
2196 	/*
2197 	 * Might still be hex.  Check out the first character.  Talk
2198 	 * about heuristics!
2199 	 */
2200 	if ((*str >= 'A' && *str <= 'F') || (*str >= 'a' && *str <= 'f'))
2201 		return hextolfp(str, lfp);
2202 
2203 	/*
2204 	 * Try it as a decimal.  If this fails, try as an unquoted
2205 	 * RT-11 date.  This code should go away eventually.
2206 	 */
2207 	if (atolfp(str, lfp))
2208 		return 1;
2209 
2210 	return rtdatetolfp(str, lfp);
2211 }
2212 
2213 
2214 /*
2215  * decodetime - decode a time value.  It should be in milliseconds
2216  */
2217 int
2218 decodetime(
2219 	char *str,
2220 	l_fp *lfp
2221 	)
2222 {
2223 	return mstolfp(str, lfp);
2224 }
2225 
2226 
2227 /*
2228  * decodeint - decode an integer
2229  */
2230 int
2231 decodeint(
2232 	char *str,
2233 	long *val
2234 	)
2235 {
2236 	if (*str == '0') {
2237 		if (*(str+1) == 'x' || *(str+1) == 'X')
2238 		    return hextoint(str+2, (u_long *)val);
2239 		return octtoint(str, (u_long *)val);
2240 	}
2241 	return atoint(str, val);
2242 }
2243 
2244 
2245 /*
2246  * decodeuint - decode an unsigned integer
2247  */
2248 int
2249 decodeuint(
2250 	char *str,
2251 	u_long *val
2252 	)
2253 {
2254 	if (*str == '0') {
2255 		if (*(str + 1) == 'x' || *(str + 1) == 'X')
2256 			return (hextoint(str + 2, val));
2257 		return (octtoint(str, val));
2258 	}
2259 	return (atouint(str, val));
2260 }
2261 
2262 
2263 /*
2264  * decodearr - decode an array of time values
2265  */
2266 static int
2267 decodearr(
2268 	char *cp,
2269 	int  *narr,
2270 	l_fp *lfpa,
2271 	int   amax
2272 	)
2273 {
2274 	char *bp;
2275 	char buf[60];
2276 
2277 	*narr = 0;
2278 
2279 	while (*narr < amax && *cp) {
2280 		if (isspace(pgetc(cp))) {
2281 			do
2282 				++cp;
2283 			while (*cp && isspace(pgetc(cp)));
2284 		} else {
2285 			bp = buf;
2286 			do {
2287 				if (bp != (buf + sizeof(buf) - 1))
2288 					*bp++ = *cp;
2289 				++cp;
2290 			} while (*cp && !isspace(pgetc(cp)));
2291 			*bp = '\0';
2292 
2293 			if (!decodetime(buf, lfpa))
2294 				return 0;
2295 			++(*narr);
2296 			++lfpa;
2297 		}
2298 	}
2299 	return 1;
2300 }
2301 
2302 
2303 /*
2304  * Finally, the built in command handlers
2305  */
2306 
2307 /*
2308  * help - tell about commands, or details of a particular command
2309  */
2310 static void
2311 help(
2312 	struct parse *pcmd,
2313 	FILE *fp
2314 	)
2315 {
2316 	struct xcmd *xcp = NULL;	/* quiet warning */
2317 	const char *cmd;
2318 	const char *list[100];
2319 	size_t word, words;
2320 	size_t row, rows;
2321 	size_t col, cols;
2322 	size_t length;
2323 
2324 	if (pcmd->nargs == 0) {
2325 		words = 0;
2326 		for (xcp = builtins; xcp->keyword != NULL; xcp++) {
2327 			if (*(xcp->keyword) != '?' &&
2328 			    words < COUNTOF(list))
2329 				list[words++] = xcp->keyword;
2330 		}
2331 		for (xcp = opcmds; xcp->keyword != NULL; xcp++)
2332 			if (words < COUNTOF(list))
2333 				list[words++] = xcp->keyword;
2334 
2335 		qsort((void *)list, words, sizeof(list[0]), helpsort);
2336 		col = 0;
2337 		for (word = 0; word < words; word++) {
2338 			length = strlen(list[word]);
2339 			col = max(col, length);
2340 		}
2341 
2342 		cols = SCREENWIDTH / ++col;
2343 		rows = (words + cols - 1) / cols;
2344 
2345 		fprintf(fp, "ntpq commands:\n");
2346 
2347 		for (row = 0; row < rows; row++) {
2348 			for (word = row; word < words; word += rows)
2349 				fprintf(fp, "%-*.*s", (int)col,
2350 					(int)col - 1, list[word]);
2351 			fprintf(fp, "\n");
2352 		}
2353 	} else {
2354 		cmd = pcmd->argval[0].string;
2355 		words = findcmd(cmd, builtins, opcmds, &xcp);
2356 		if (words == 0) {
2357 			fprintf(stderr,
2358 				"Command `%s' is unknown\n", cmd);
2359 			return;
2360 		} else if (words >= 2) {
2361 			fprintf(stderr,
2362 				"Command `%s' is ambiguous\n", cmd);
2363 			return;
2364 		}
2365 		fprintf(fp, "function: %s\n", xcp->comment);
2366 		printusage(xcp, fp);
2367 	}
2368 }
2369 
2370 
2371 /*
2372  * helpsort - do hostname qsort comparisons
2373  */
2374 static int
2375 helpsort(
2376 	const void *t1,
2377 	const void *t2
2378 	)
2379 {
2380 	const char * const *	name1 = t1;
2381 	const char * const *	name2 = t2;
2382 
2383 	return strcmp(*name1, *name2);
2384 }
2385 
2386 
2387 /*
2388  * printusage - print usage information for a command
2389  */
2390 static void
2391 printusage(
2392 	struct xcmd *xcp,
2393 	FILE *fp
2394 	)
2395 {
2396 	register int i;
2397 
2398 	/* XXX: Do we need to warn about extra args here too? */
2399 
2400 	(void) fprintf(fp, "usage: %s", xcp->keyword);
2401 	for (i = 0; i < MAXARGS && xcp->arg[i] != NO; i++) {
2402 		if (xcp->arg[i] & OPT)
2403 		    (void) fprintf(fp, " [ %s ]", xcp->desc[i]);
2404 		else
2405 		    (void) fprintf(fp, " %s", xcp->desc[i]);
2406 	}
2407 	(void) fprintf(fp, "\n");
2408 }
2409 
2410 
2411 /*
2412  * timeout - set time out time
2413  */
2414 static void
2415 timeout(
2416 	struct parse *pcmd,
2417 	FILE *fp
2418 	)
2419 {
2420 	int val;
2421 
2422 	if (pcmd->nargs == 0) {
2423 		val = (int)tvout.tv_sec * 1000 + tvout.tv_usec / 1000;
2424 		(void) fprintf(fp, "primary timeout %d ms\n", val);
2425 	} else {
2426 		tvout.tv_sec = pcmd->argval[0].uval / 1000;
2427 		tvout.tv_usec = (pcmd->argval[0].uval - ((long)tvout.tv_sec * 1000))
2428 			* 1000;
2429 	}
2430 }
2431 
2432 
2433 /*
2434  * auth_delay - set delay for auth requests
2435  */
2436 static void
2437 auth_delay(
2438 	struct parse *pcmd,
2439 	FILE *fp
2440 	)
2441 {
2442 	int isneg;
2443 	u_long val;
2444 
2445 	if (pcmd->nargs == 0) {
2446 		val = delay_time.l_ui * 1000 + delay_time.l_uf / 4294967;
2447 		(void) fprintf(fp, "delay %lu ms\n", val);
2448 	} else {
2449 		if (pcmd->argval[0].ival < 0) {
2450 			isneg = 1;
2451 			val = (u_long)(-pcmd->argval[0].ival);
2452 		} else {
2453 			isneg = 0;
2454 			val = (u_long)pcmd->argval[0].ival;
2455 		}
2456 
2457 		delay_time.l_ui = val / 1000;
2458 		val %= 1000;
2459 		delay_time.l_uf = val * 4294967;	/* 2**32/1000 */
2460 
2461 		if (isneg)
2462 		    L_NEG(&delay_time);
2463 	}
2464 }
2465 
2466 
2467 /*
2468  * host - set the host we are dealing with.
2469  */
2470 static void
2471 host(
2472 	struct parse *pcmd,
2473 	FILE *fp
2474 	)
2475 {
2476 	int i;
2477 
2478 	if (pcmd->nargs == 0) {
2479 		if (havehost)
2480 			(void) fprintf(fp, "current host is %s\n",
2481 					   currenthost);
2482 		else
2483 			(void) fprintf(fp, "no current host\n");
2484 		return;
2485 	}
2486 
2487 	i = 0;
2488 	ai_fam_templ = ai_fam_default;
2489 	if (pcmd->nargs == 2) {
2490 		if (!strcmp("-4", pcmd->argval[i].string))
2491 			ai_fam_templ = AF_INET;
2492 		else if (!strcmp("-6", pcmd->argval[i].string))
2493 			ai_fam_templ = AF_INET6;
2494 		else
2495 			goto no_change;
2496 		i = 1;
2497 	}
2498 	if (openhost(pcmd->argval[i].string, ai_fam_templ)) {
2499 		fprintf(fp, "current host set to %s\n", currenthost);
2500 	} else {
2501     no_change:
2502 		if (havehost)
2503 			fprintf(fp, "current host remains %s\n",
2504 				currenthost);
2505 		else
2506 			fprintf(fp, "still no current host\n");
2507 	}
2508 }
2509 
2510 
2511 /*
2512  * poll - do one (or more) polls of the host via NTP
2513  */
2514 /*ARGSUSED*/
2515 static void
2516 ntp_poll(
2517 	struct parse *pcmd,
2518 	FILE *fp
2519 	)
2520 {
2521 	(void) fprintf(fp, "poll not implemented yet\n");
2522 }
2523 
2524 
2525 /*
2526  * showdrefid2str - return a string explanation of the value of drefid
2527  */
2528 static const char *
2529 showdrefid2str(void)
2530 {
2531 	switch (drefid) {
2532 	    case REFID_HASH:
2533 	    	return "hash";
2534 	    case REFID_IPV4:
2535 	    	return "ipv4";
2536 	    default:
2537 	    	return "Unknown";
2538 	}
2539 }
2540 
2541 
2542 /*
2543  * drefid - display/change "display hash"
2544  */
2545 static void
2546 showdrefid(
2547 	struct parse *pcmd,
2548 	FILE *fp
2549 	)
2550 {
2551 	if (pcmd->nargs == 0) {
2552 		(void) fprintf(fp, "drefid value is %s\n", showdrefid2str());
2553 		return;
2554 	} else if (STREQ(pcmd->argval[0].string, "hash")) {
2555 		drefid = REFID_HASH;
2556 	} else if (STREQ(pcmd->argval[0].string, "ipv4")) {
2557 		drefid = REFID_IPV4;
2558 	} else {
2559 		(void) fprintf(fp, "What?\n");
2560 		return;
2561 	}
2562 	(void) fprintf(fp, "drefid value set to %s\n", showdrefid2str());
2563 }
2564 
2565 
2566 /*
2567  * keyid - get a keyid to use for authenticating requests
2568  */
2569 static void
2570 keyid(
2571 	struct parse *pcmd,
2572 	FILE *fp
2573 	)
2574 {
2575 	if (pcmd->nargs == 0) {
2576 		if (info_auth_keyid == 0)
2577 		    (void) fprintf(fp, "no keyid defined\n");
2578 		else
2579 		    (void) fprintf(fp, "keyid is %lu\n", (u_long)info_auth_keyid);
2580 	} else {
2581 		/* allow zero so that keyid can be cleared. */
2582 		if(pcmd->argval[0].uval > NTP_MAXKEY)
2583 		    (void) fprintf(fp, "Invalid key identifier\n");
2584 		info_auth_keyid = pcmd->argval[0].uval;
2585 	}
2586 }
2587 
2588 /*
2589  * keytype - get type of key to use for authenticating requests
2590  */
2591 static void
2592 keytype(
2593 	struct parse *pcmd,
2594 	FILE *fp
2595 	)
2596 {
2597 	const char *	digest_name;
2598 	size_t		digest_len;
2599 	int		key_type;
2600 
2601 	if (!pcmd->nargs) {
2602 		fprintf(fp, "keytype is %s with %lu octet digests\n",
2603 			keytype_name(info_auth_keytype),
2604 			(u_long)info_auth_hashlen);
2605 		return;
2606 	}
2607 
2608 	digest_name = pcmd->argval[0].string;
2609 	digest_len = 0;
2610 	key_type = keytype_from_text(digest_name, &digest_len);
2611 
2612 	if (!key_type) {
2613 		fprintf(fp, "keytype is not valid. "
2614 #ifdef OPENSSL
2615 			"Type \"help keytype\" for the available digest types.\n");
2616 #else
2617 			"Only \"md5\" is available.\n");
2618 #endif
2619 		return;
2620 	}
2621 
2622 	info_auth_keytype = key_type;
2623 	info_auth_hashlen = digest_len;
2624 }
2625 
2626 
2627 /*
2628  * passwd - get an authentication key
2629  */
2630 /*ARGSUSED*/
2631 static void
2632 passwd(
2633 	struct parse *pcmd,
2634 	FILE *fp
2635 	)
2636 {
2637 	const char *pass;
2638 
2639 	if (info_auth_keyid == 0) {
2640 		info_auth_keyid = getkeyid("Keyid: ");
2641 		if (info_auth_keyid == 0) {
2642 			(void)fprintf(fp, "Keyid must be defined\n");
2643 			return;
2644 		}
2645 	}
2646 	if (pcmd->nargs >= 1)
2647 		pass = pcmd->argval[0].string;
2648 	else {
2649 		pass = getpass_keytype(info_auth_keytype);
2650 		if ('\0' == pass[0]) {
2651 			fprintf(fp, "Password unchanged\n");
2652 			return;
2653 		}
2654 	}
2655 	authusekey(info_auth_keyid, info_auth_keytype,
2656 		   (const u_char *)pass);
2657 	authtrust(info_auth_keyid, 1);
2658 }
2659 
2660 
2661 /*
2662  * hostnames - set the showhostnames flag
2663  */
2664 static void
2665 hostnames(
2666 	struct parse *pcmd,
2667 	FILE *fp
2668 	)
2669 {
2670 	if (pcmd->nargs == 0) {
2671 		if (showhostnames)
2672 		    (void) fprintf(fp, "hostnames being shown\n");
2673 		else
2674 		    (void) fprintf(fp, "hostnames not being shown\n");
2675 	} else {
2676 		if (STREQ(pcmd->argval[0].string, "yes"))
2677 		    showhostnames = 1;
2678 		else if (STREQ(pcmd->argval[0].string, "no"))
2679 		    showhostnames = 0;
2680 		else
2681 		    (void)fprintf(stderr, "What?\n");
2682 	}
2683 }
2684 
2685 
2686 
2687 /*
2688  * setdebug - set/change debugging level
2689  */
2690 static void
2691 setdebug(
2692 	struct parse *pcmd,
2693 	FILE *fp
2694 	)
2695 {
2696 	if (pcmd->nargs == 0) {
2697 		(void) fprintf(fp, "debug level is %d\n", debug);
2698 		return;
2699 	} else if (STREQ(pcmd->argval[0].string, "no")) {
2700 		debug = 0;
2701 	} else if (STREQ(pcmd->argval[0].string, "more")) {
2702 		debug++;
2703 	} else if (STREQ(pcmd->argval[0].string, "less")) {
2704 		debug--;
2705 	} else {
2706 		(void) fprintf(fp, "What?\n");
2707 		return;
2708 	}
2709 	(void) fprintf(fp, "debug level set to %d\n", debug);
2710 }
2711 
2712 
2713 /*
2714  * quit - stop this nonsense
2715  */
2716 /*ARGSUSED*/
2717 static void
2718 quit(
2719 	struct parse *pcmd,
2720 	FILE *fp
2721 	)
2722 {
2723 	if (havehost)
2724 	    closesocket(sockfd);	/* cleanliness next to godliness */
2725 	exit(0);
2726 }
2727 
2728 
2729 /*
2730  * version - print the current version number
2731  */
2732 /*ARGSUSED*/
2733 static void
2734 version(
2735 	struct parse *pcmd,
2736 	FILE *fp
2737 	)
2738 {
2739 
2740 	(void) fprintf(fp, "%s\n", Version);
2741 	return;
2742 }
2743 
2744 
2745 /*
2746  * raw - set raw mode output
2747  */
2748 /*ARGSUSED*/
2749 static void
2750 raw(
2751 	struct parse *pcmd,
2752 	FILE *fp
2753 	)
2754 {
2755 	rawmode = 1;
2756 	(void) fprintf(fp, "Output set to raw\n");
2757 }
2758 
2759 
2760 /*
2761  * cooked - set cooked mode output
2762  */
2763 /*ARGSUSED*/
2764 static void
2765 cooked(
2766 	struct parse *pcmd,
2767 	FILE *fp
2768 	)
2769 {
2770 	rawmode = 0;
2771 	(void) fprintf(fp, "Output set to cooked\n");
2772 	return;
2773 }
2774 
2775 
2776 /*
2777  * authenticate - always authenticate requests to this host
2778  */
2779 static void
2780 authenticate(
2781 	struct parse *pcmd,
2782 	FILE *fp
2783 	)
2784 {
2785 	if (pcmd->nargs == 0) {
2786 		if (always_auth) {
2787 			(void) fprintf(fp,
2788 				       "authenticated requests being sent\n");
2789 		} else
2790 		    (void) fprintf(fp,
2791 				   "unauthenticated requests being sent\n");
2792 	} else {
2793 		if (STREQ(pcmd->argval[0].string, "yes")) {
2794 			always_auth = 1;
2795 		} else if (STREQ(pcmd->argval[0].string, "no")) {
2796 			always_auth = 0;
2797 		} else
2798 		    (void)fprintf(stderr, "What?\n");
2799 	}
2800 }
2801 
2802 
2803 /*
2804  * ntpversion - choose the NTP version to use
2805  */
2806 static void
2807 ntpversion(
2808 	struct parse *pcmd,
2809 	FILE *fp
2810 	)
2811 {
2812 	if (pcmd->nargs == 0) {
2813 		(void) fprintf(fp,
2814 			       "NTP version being claimed is %d\n", pktversion);
2815 	} else {
2816 		if (pcmd->argval[0].uval < NTP_OLDVERSION
2817 		    || pcmd->argval[0].uval > NTP_VERSION) {
2818 			(void) fprintf(stderr, "versions %d to %d, please\n",
2819 				       NTP_OLDVERSION, NTP_VERSION);
2820 		} else {
2821 			pktversion = (u_char) pcmd->argval[0].uval;
2822 		}
2823 	}
2824 }
2825 
2826 
2827 static void __attribute__((__format__(__printf__, 1, 0)))
2828 vwarning(const char *fmt, va_list ap)
2829 {
2830 	int serrno = errno;
2831 	(void) fprintf(stderr, "%s: ", progname);
2832 	vfprintf(stderr, fmt, ap);
2833 	(void) fprintf(stderr, ": %s\n", strerror(serrno));
2834 }
2835 
2836 /*
2837  * warning - print a warning message
2838  */
2839 static void __attribute__((__format__(__printf__, 1, 2)))
2840 warning(
2841 	const char *fmt,
2842 	...
2843 	)
2844 {
2845 	va_list ap;
2846 	va_start(ap, fmt);
2847 	vwarning(fmt, ap);
2848 	va_end(ap);
2849 }
2850 
2851 
2852 /*
2853  * error - print a message and exit
2854  */
2855 static void __attribute__((__format__(__printf__, 1, 2)))
2856 error(
2857 	const char *fmt,
2858 	...
2859 	)
2860 {
2861 	va_list ap;
2862 	va_start(ap, fmt);
2863 	vwarning(fmt, ap);
2864 	va_end(ap);
2865 	exit(1);
2866 }
2867 /*
2868  * getkeyid - prompt the user for a keyid to use
2869  */
2870 static u_long
2871 getkeyid(
2872 	const char *keyprompt
2873 	)
2874 {
2875 	int c;
2876 	FILE *fi;
2877 	char pbuf[20];
2878 	size_t i;
2879 	size_t ilim;
2880 
2881 #ifndef SYS_WINNT
2882 	if ((fi = fdopen(open("/dev/tty", 2), "r")) == NULL)
2883 #else
2884 	if ((fi = _fdopen(open("CONIN$", _O_TEXT), "r")) == NULL)
2885 #endif /* SYS_WINNT */
2886 		fi = stdin;
2887 	else
2888 		setbuf(fi, (char *)NULL);
2889 	fprintf(stderr, "%s", keyprompt); fflush(stderr);
2890 	for (i = 0, ilim = COUNTOF(pbuf) - 1;
2891 	     i < ilim && (c = getc(fi)) != '\n' && c != EOF;
2892 	     )
2893 		pbuf[i++] = (char)c;
2894 	pbuf[i] = '\0';
2895 	if (fi != stdin)
2896 		fclose(fi);
2897 
2898 	return (u_long) atoi(pbuf);
2899 }
2900 
2901 
2902 /*
2903  * atoascii - printable-ize possibly ascii data using the character
2904  *	      transformations cat -v uses.
2905  */
2906 static void
2907 atoascii(
2908 	const char *in,
2909 	size_t in_octets,
2910 	char *out,
2911 	size_t out_octets
2912 	)
2913 {
2914 	const u_char *	pchIn;
2915 	const u_char *	pchInLimit;
2916 	u_char *	pchOut;
2917 	u_char		c;
2918 
2919 	pchIn = (const u_char *)in;
2920 	pchInLimit = pchIn + in_octets;
2921 	pchOut = (u_char *)out;
2922 
2923 	if (NULL == pchIn) {
2924 		if (0 < out_octets)
2925 			*pchOut = '\0';
2926 		return;
2927 	}
2928 
2929 #define	ONEOUT(c)					\
2930 do {							\
2931 	if (0 == --out_octets) {			\
2932 		*pchOut = '\0';				\
2933 		return;					\
2934 	}						\
2935 	*pchOut++ = (c);				\
2936 } while (0)
2937 
2938 	for (	; pchIn < pchInLimit; pchIn++) {
2939 		c = *pchIn;
2940 		if ('\0' == c)
2941 			break;
2942 		if (c & 0x80) {
2943 			ONEOUT('M');
2944 			ONEOUT('-');
2945 			c &= 0x7f;
2946 		}
2947 		if (c < ' ') {
2948 			ONEOUT('^');
2949 			ONEOUT((u_char)(c + '@'));
2950 		} else if (0x7f == c) {
2951 			ONEOUT('^');
2952 			ONEOUT('?');
2953 		} else
2954 			ONEOUT(c);
2955 	}
2956 	ONEOUT('\0');
2957 
2958 #undef ONEOUT
2959 }
2960 
2961 
2962 /*
2963  * makeascii - print possibly ascii data using the character
2964  *	       transformations that cat -v uses.
2965  */
2966 void
2967 makeascii(
2968 	size_t length,
2969 	const char *data,
2970 	FILE *fp
2971 	)
2972 {
2973 	const u_char *data_u_char;
2974 	const u_char *cp;
2975 	int c;
2976 
2977 	data_u_char = (const u_char *)data;
2978 
2979 	for (cp = data_u_char; cp < data_u_char + length; cp++) {
2980 		c = (int)*cp;
2981 		if (c & 0x80) {
2982 			putc('M', fp);
2983 			putc('-', fp);
2984 			c &= 0x7f;
2985 		}
2986 
2987 		if (c < ' ') {
2988 			putc('^', fp);
2989 			putc(c + '@', fp);
2990 		} else if (0x7f == c) {
2991 			putc('^', fp);
2992 			putc('?', fp);
2993 		} else
2994 			putc(c, fp);
2995 	}
2996 }
2997 
2998 
2999 /*
3000  * asciize - same thing as makeascii except add a newline
3001  */
3002 void
3003 asciize(
3004 	int length,
3005 	char *data,
3006 	FILE *fp
3007 	)
3008 {
3009 	makeascii(length, data, fp);
3010 	putc('\n', fp);
3011 }
3012 
3013 
3014 /*
3015  * truncate string to fit clipping excess at end.
3016  *	"too long"	->	"too l"
3017  * Used for hostnames.
3018  */
3019 const char *
3020 trunc_right(
3021 	const char *	src,
3022 	size_t		width
3023 	)
3024 {
3025 	size_t	sl;
3026 	char *	out;
3027 
3028 
3029 	sl = strlen(src);
3030 	if (sl > width && LIB_BUFLENGTH - 1 > width && width > 0) {
3031 		LIB_GETBUF(out);
3032 		memcpy(out, src, width);
3033 		out[width] = '\0';
3034 
3035 		return out;
3036 	}
3037 
3038 	return src;
3039 }
3040 
3041 
3042 /*
3043  * truncate string to fit by preserving right side and using '_' to hint
3044  *	"too long"	->	"_long"
3045  * Used for local IPv6 addresses, where low bits differentiate.
3046  */
3047 const char *
3048 trunc_left(
3049 	const char *	src,
3050 	size_t		width
3051 	)
3052 {
3053 	size_t	sl;
3054 	char *	out;
3055 
3056 
3057 	sl = strlen(src);
3058 	if (sl > width && LIB_BUFLENGTH - 1 > width && width > 1) {
3059 		LIB_GETBUF(out);
3060 		out[0] = '_';
3061 		memcpy(&out[1], &src[sl + 1 - width], width);
3062 
3063 		return out;
3064 	}
3065 
3066 	return src;
3067 }
3068 
3069 
3070 /*
3071  * Some circular buffer space
3072  */
3073 #define	CBLEN	80
3074 #define	NUMCB	6
3075 
3076 char circ_buf[NUMCB][CBLEN];
3077 int nextcb = 0;
3078 
3079 /* --------------------------------------------------------------------
3080  * Parsing a response value list
3081  *
3082  * This sounds simple (and it actually is not really hard) but it has
3083  * some pitfalls.
3084  *
3085  * Rule1: CR/LF is never embedded in an item
3086  * Rule2: An item is a name, optionally followed by a value
3087  * Rule3: The value is separated from the name by a '='
3088  * Rule4: Items are separated by a ','
3089  * Rule5: values can be quoted by '"', in which case they can contain
3090  *        arbitrary characters but *not* '"', CR and LF
3091  *
3092  * There are a few implementations out there that require a somewhat
3093  * relaxed attitude when parsing a value list, especially since we want
3094  * to copy names and values into local buffers. If these would overflow,
3095  * the item should be skipped without terminating the parsing sequence.
3096  *
3097  * Also, for empty values, there might be a '=' after the name or not;
3098  * we treat that equivalent.
3099  *
3100  * Parsing an item definitely breaks on a CR/LF. If an item is not
3101  * followed by a comma (','), parsing stops. In the middle of a quoted
3102  * character sequence CR/LF terminates the parsing finally without
3103  * returning a value.
3104  *
3105  * White space and other noise is ignored when parsing the data buffer;
3106  * only CR, LF, ',', '=' and '"' are characters with a special meaning.
3107  * White space is stripped from the names and values *after* working
3108  * through the buffer, before making the local copies. If whitespace
3109  * stripping results in an empty name, parsing resumes.
3110  */
3111 
3112 /*
3113  * nextvar parsing helpers
3114  */
3115 
3116 /* predicate: allowed chars inside a quoted string */
3117 static int/*BOOL*/ cp_qschar(int ch)
3118 {
3119 	return ch && (ch != '"' && ch != '\r' && ch != '\n');
3120 }
3121 
3122 /* predicate: allowed chars inside an unquoted string */
3123 static int/*BOOL*/ cp_uqchar(int ch)
3124 {
3125 	return ch && (ch != ',' && ch != '"' && ch != '\r' && ch != '\n');
3126 }
3127 
3128 /* predicate: allowed chars inside a value name */
3129 static int/*BOOL*/ cp_namechar(int ch)
3130 {
3131 	return ch && (ch != ',' && ch != '=' && ch != '\r' && ch != '\n');
3132 }
3133 
3134 /* predicate: characters *between* list items. We're relaxed here. */
3135 static int/*BOOL*/ cp_ivspace(int ch)
3136 {
3137 	return (ch == ',' || (ch > 0 && ch <= ' '));
3138 }
3139 
3140 /* get current character (or NUL when on end) */
3141 static inline int
3142 pf_getch(
3143 	const char **	datap,
3144 	const char *	endp
3145 	)
3146 {
3147 	return (*datap != endp)
3148 	    ? *(const unsigned char*)*datap
3149 	    : '\0';
3150 }
3151 
3152 /* get next character (or NUL when on end) */
3153 static inline int
3154 pf_nextch(
3155 	const char **	datap,
3156 	const char *	endp
3157 	)
3158 {
3159 	return (*datap != endp && ++(*datap) != endp)
3160 	    ? *(const unsigned char*)*datap
3161 	    : '\0';
3162 }
3163 
3164 static size_t
3165 str_strip(
3166 	const char ** 	datap,
3167 	size_t		len
3168 	)
3169 {
3170 	static const char empty[] = "";
3171 
3172 	if (*datap && len) {
3173 		const char * cpl = *datap;
3174 		const char * cpr = cpl + len;
3175 
3176 		while (cpl != cpr && *(const unsigned char*)cpl <= ' ')
3177 			++cpl;
3178 		while (cpl != cpr && *(const unsigned char*)(cpr - 1) <= ' ')
3179 			--cpr;
3180 		*datap = cpl;
3181 		len = (size_t)(cpr - cpl);
3182 	} else {
3183 		*datap = empty;
3184 		len = 0;
3185 	}
3186 	return len;
3187 }
3188 
3189 static void
3190 pf_error(
3191 	const char *	what,
3192 	const char *	where,
3193 	const char *	whend
3194 	)
3195 {
3196 #   ifndef BUILD_AS_LIB
3197 
3198 	FILE *	ofp = (debug > 0) ? stdout : stderr;
3199 	size_t	len = (size_t)(whend - where);
3200 
3201 	if (len > 50) /* *must* fit into an 'int'! */
3202 		len = 50;
3203 	fprintf(ofp, "nextvar: %s: '%.*s'\n",
3204 		what, (int)len, where);
3205 
3206 #   else  /*defined(BUILD_AS_LIB)*/
3207 
3208 	UNUSED_ARG(what);
3209 	UNUSED_ARG(where);
3210 	UNUSED_ARG(whend);
3211 
3212 #   endif /*defined(BUILD_AS_LIB)*/
3213 }
3214 
3215 /*
3216  * nextvar - find the next variable in the buffer
3217  */
3218 int/*BOOL*/
3219 nextvar(
3220 	size_t *datalen,
3221 	const char **datap,
3222 	char **vname,
3223 	char **vvalue
3224 	)
3225 {
3226 	enum PState 	{ sDone, sInit, sName, sValU, sValQ };
3227 
3228 	static char	name[MAXVARLEN], value[MAXVALLEN];
3229 
3230 	const char	*cp, *cpend;
3231 	const char	*np, *vp;
3232 	size_t		nlen, vlen;
3233 	int		ch;
3234 	enum PState	st;
3235 
3236 	cpend = *datap + *datalen;
3237 
3238   again:
3239 	np   = vp   = NULL;
3240 	nlen = vlen = 0;
3241 
3242 	st = sInit;
3243 	ch = pf_getch(datap, cpend);
3244 
3245 	while (st != sDone) {
3246 		switch (st)
3247 		{
3248 		case sInit:	/* handle inter-item chars */
3249 			while (cp_ivspace(ch))
3250 				ch = pf_nextch(datap, cpend);
3251 			if (cp_namechar(ch)) {
3252 				np = *datap;
3253 				cp = np;
3254 				st = sName;
3255 				ch = pf_nextch(datap, cpend);
3256 			} else {
3257 				goto final_done;
3258 			}
3259 			break;
3260 
3261 		case sName:	/* collect name */
3262 			while (cp_namechar(ch))
3263 				ch = pf_nextch(datap, cpend);
3264 			nlen = (size_t)(*datap - np);
3265 			if (ch == '=') {
3266 				ch = pf_nextch(datap, cpend);
3267 				vp = *datap;
3268 				st = sValU;
3269 			} else {
3270 				if (ch != ',')
3271 					*datap = cpend;
3272 				st = sDone;
3273 			}
3274 			break;
3275 
3276 		case sValU:	/* collect unquoted part(s) of value */
3277 			while (cp_uqchar(ch))
3278 				ch = pf_nextch(datap, cpend);
3279 			if (ch == '"') {
3280 				ch = pf_nextch(datap, cpend);
3281 				st = sValQ;
3282 			} else {
3283 				vlen = (size_t)(*datap - vp);
3284 				if (ch != ',')
3285 					*datap = cpend;
3286 				st = sDone;
3287 			}
3288 			break;
3289 
3290 		case sValQ:	/* collect quoted part(s) of value */
3291 			while (cp_qschar(ch))
3292 				ch = pf_nextch(datap, cpend);
3293 			if (ch == '"') {
3294 				ch = pf_nextch(datap, cpend);
3295 				st = sValU;
3296 			} else {
3297 				pf_error("no closing quote, stop", cp, cpend);
3298 				goto final_done;
3299 			}
3300 			break;
3301 
3302 		default:
3303 			pf_error("state machine error, stop", *datap, cpend);
3304 			goto final_done;
3305 		}
3306 	}
3307 
3308 	/* If name or value do not fit their buffer, croak and start
3309 	 * over. If there's no name at all after whitespace stripping,
3310 	 * redo silently.
3311 	 */
3312 	nlen = str_strip(&np, nlen);
3313 	vlen = str_strip(&vp, vlen);
3314 
3315 	if (nlen == 0) {
3316 		goto again;
3317 	}
3318 	if (nlen >= sizeof(name)) {
3319 		pf_error("runaway name", np, cpend);
3320 		goto again;
3321 	}
3322 	if (vlen >= sizeof(value)) {
3323 		pf_error("runaway value", vp, cpend);
3324 		goto again;
3325 	}
3326 
3327 	/* copy name and value into NUL-terminated buffers */
3328 	memcpy(name, np, nlen);
3329 	name[nlen] = '\0';
3330 	*vname = name;
3331 
3332 	memcpy(value, vp, vlen);
3333 	value[vlen] = '\0';
3334 	*vvalue = value;
3335 
3336 	/* check if there's more to do or if we are finshed */
3337 	*datalen = (size_t)(cpend - *datap);
3338 	return TRUE;
3339 
3340   final_done:
3341 	*datap = cpend;
3342 	*datalen = 0;
3343 	return FALSE;
3344 }
3345 
3346 
3347 u_short
3348 varfmt(const char * varname)
3349 {
3350 	u_int n;
3351 
3352 	for (n = 0; n < COUNTOF(cookedvars); n++)
3353 		if (!strcmp(varname, cookedvars[n].varname))
3354 			return cookedvars[n].fmt;
3355 
3356 	return PADDING;
3357 }
3358 
3359 
3360 /*
3361  * printvars - print variables returned in response packet
3362  */
3363 void
3364 printvars(
3365 	size_t length,
3366 	const char *data,
3367 	int status,
3368 	int sttype,
3369 	int quiet,
3370 	FILE *fp
3371 	)
3372 {
3373 	if (rawmode)
3374 	    rawprint(sttype, length, data, status, quiet, fp);
3375 	else
3376 	    cookedprint(sttype, length, data, status, quiet, fp);
3377 }
3378 
3379 
3380 /*
3381  * rawprint - do a printout of the data in raw mode
3382  */
3383 static void
3384 rawprint(
3385 	int datatype,
3386 	size_t length,
3387 	const char *data,
3388 	int status,
3389 	int quiet,
3390 	FILE *fp
3391 	)
3392 {
3393 	const char *cp;
3394 	const char *cpend;
3395 
3396 	/*
3397 	 * Essentially print the data as is.  We reformat unprintables, though.
3398 	 */
3399 	cp = data;
3400 	cpend = data + length;
3401 
3402 	if (!quiet)
3403 		(void) fprintf(fp, "status=0x%04x,\n", status);
3404 
3405 	while (cp < cpend) {
3406 		if (*cp == '\r') {
3407 			/*
3408 			 * If this is a \r and the next character is a
3409 			 * \n, supress this, else pretty print it.  Otherwise
3410 			 * just output the character.
3411 			 */
3412 			if (cp == (cpend - 1) || *(cp + 1) != '\n')
3413 			    makeascii(1, cp, fp);
3414 		} else if (isspace(pgetc(cp)) || isprint(pgetc(cp)))
3415 			putc(*cp, fp);
3416 		else
3417 			makeascii(1, cp, fp);
3418 		cp++;
3419 	}
3420 }
3421 
3422 
3423 /*
3424  * Global data used by the cooked output routines
3425  */
3426 int out_chars;		/* number of characters output */
3427 int out_linecount;	/* number of characters output on this line */
3428 
3429 
3430 /*
3431  * startoutput - get ready to do cooked output
3432  */
3433 static void
3434 startoutput(void)
3435 {
3436 	out_chars = 0;
3437 	out_linecount = 0;
3438 }
3439 
3440 
3441 /*
3442  * output - output a variable=value combination
3443  */
3444 static void
3445 output(
3446 	FILE *fp,
3447 	const char *name,
3448 	const char *value
3449 	)
3450 {
3451 	int len;
3452 
3453 	/* strlen of "name=value" */
3454 	len = size2int_sat(strlen(name) + 1 + strlen(value));
3455 
3456 	if (out_chars != 0) {
3457 		out_chars += 2;
3458 		if ((out_linecount + len + 2) > MAXOUTLINE) {
3459 			fputs(",\n", fp);
3460 			out_linecount = 0;
3461 		} else {
3462 			fputs(", ", fp);
3463 			out_linecount += 2;
3464 		}
3465 	}
3466 
3467 	fputs(name, fp);
3468 	putc('=', fp);
3469 	fputs(value, fp);
3470 	out_chars += len;
3471 	out_linecount += len;
3472 }
3473 
3474 
3475 /*
3476  * endoutput - terminate a block of cooked output
3477  */
3478 static void
3479 endoutput(
3480 	FILE *fp
3481 	)
3482 {
3483 	if (out_chars != 0)
3484 		putc('\n', fp);
3485 }
3486 
3487 
3488 /*
3489  * outputarr - output an array of values
3490  */
3491 static void
3492 outputarr(
3493 	FILE *fp,
3494 	char *name,
3495 	int narr,
3496 	l_fp *lfp
3497 	)
3498 {
3499 	char *bp;
3500 	char *cp;
3501 	size_t i;
3502 	size_t len;
3503 	char buf[256];
3504 
3505 	bp = buf;
3506 	/*
3507 	 * Hack to align delay and offset values
3508 	 */
3509 	for (i = (int)strlen(name); i < 11; i++)
3510 		*bp++ = ' ';
3511 
3512 	for (i = narr; i > 0; i--) {
3513 		if (i != (size_t)narr)
3514 			*bp++ = ' ';
3515 		cp = lfptoms(lfp, 2);
3516 		len = strlen(cp);
3517 		if (len > 7) {
3518 			cp[7] = '\0';
3519 			len = 7;
3520 		}
3521 		while (len < 7) {
3522 			*bp++ = ' ';
3523 			len++;
3524 		}
3525 		while (*cp != '\0')
3526 		    *bp++ = *cp++;
3527 		lfp++;
3528 	}
3529 	*bp = '\0';
3530 	output(fp, name, buf);
3531 }
3532 
3533 static char *
3534 tstflags(
3535 	u_long val
3536 	)
3537 {
3538 #	if CBLEN < 10
3539 #	 error BLEN is too small -- increase!
3540 #	endif
3541 
3542 	char *cp, *s;
3543 	size_t cb, i;
3544 	int l;
3545 
3546 	s = cp = circ_buf[nextcb];
3547 	if (++nextcb >= NUMCB)
3548 		nextcb = 0;
3549 	cb = sizeof(circ_buf[0]);
3550 
3551 	l = snprintf(cp, cb, "%02lx", val);
3552 	if (l < 0 || (size_t)l >= cb)
3553 		goto fail;
3554 	cp += l;
3555 	cb -= l;
3556 	if (!val) {
3557 		l = strlcat(cp, " ok", cb);
3558 		if ((size_t)l >= cb)
3559 			goto fail;
3560 		cp += l;
3561 		cb -= l;
3562 	} else {
3563 		const char *sep;
3564 
3565 		sep = " ";
3566 		for (i = 0; i < COUNTOF(tstflagnames); i++) {
3567 			if (val & 0x1) {
3568 				l = snprintf(cp, cb, "%s%s", sep,
3569 					     tstflagnames[i]);
3570 				if (l < 0)
3571 					goto fail;
3572 				if ((size_t)l >= cb) {
3573 					cp += cb - 4;
3574 					cb = 4;
3575 					l = strlcpy (cp, "...", cb);
3576 					cp += l;
3577 					cb -= l;
3578 					break;
3579 				}
3580 				sep = ", ";
3581 				cp += l;
3582 				cb -= l;
3583 			}
3584 			val >>= 1;
3585 		}
3586 	}
3587 
3588 	return s;
3589 
3590   fail:
3591 	*cp = '\0';
3592 	return s;
3593 }
3594 
3595 /*
3596  * cookedprint - output variables in cooked mode
3597  */
3598 static void
3599 cookedprint(
3600 	int datatype,
3601 	size_t length,
3602 	const char *data,
3603 	int status,
3604 	int quiet,
3605 	FILE *fp
3606 	)
3607 {
3608 	char *name;
3609 	char *value;
3610 	char output_raw;
3611 	int fmt;
3612 	l_fp lfp;
3613 	sockaddr_u hval;
3614 	u_long uval;
3615 	int narr;
3616 	size_t len;
3617 	l_fp lfparr[8];
3618 	char b[12];
3619 	char bn[2 * MAXVARLEN];
3620 	char bv[2 * MAXVALLEN];
3621 
3622 	UNUSED_ARG(datatype);
3623 
3624 	if (!quiet)
3625 		fprintf(fp, "status=%04x %s,\n", status,
3626 			statustoa(datatype, status));
3627 
3628 	startoutput();
3629 	while (nextvar(&length, &data, &name, &value)) {
3630 		fmt = varfmt(name);
3631 		output_raw = 0;
3632 		switch (fmt) {
3633 
3634 		case PADDING:
3635 			output_raw = '*';
3636 			break;
3637 
3638 		case TS:
3639 			if (!value || !decodets(value, &lfp))
3640 				output_raw = '?';
3641 			else
3642 				output(fp, name, prettydate(&lfp));
3643 			break;
3644 
3645 		case HA:	/* fallthru */
3646 		case NA:
3647 			if (!value || !decodenetnum(value, &hval)) {
3648 				output_raw = '?';
3649 			} else if (fmt == HA){
3650 				output(fp, name, nntohost(&hval));
3651 			} else {
3652 				output(fp, name, stoa(&hval));
3653 			}
3654 			break;
3655 
3656 		case RF:
3657 			if (!value) {
3658 				output_raw = '?';
3659 			} else if (decodenetnum(value, &hval)) {
3660 				if (ISREFCLOCKADR(&hval))
3661 					output(fp, name,
3662 					       refnumtoa(&hval));
3663 				else
3664 					output(fp, name, stoa(&hval));
3665 			} else if (strlen(value) <= 4) {
3666 				output(fp, name, value);
3667 			} else {
3668 				output_raw = '?';
3669 			}
3670 			break;
3671 
3672 		case LP:
3673 			if (!value || !decodeuint(value, &uval) || uval > 3) {
3674 				output_raw = '?';
3675 			} else {
3676 				b[0] = (0x2 & uval)
3677 					   ? '1'
3678 					   : '0';
3679 				b[1] = (0x1 & uval)
3680 					   ? '1'
3681 					   : '0';
3682 				b[2] = '\0';
3683 				output(fp, name, b);
3684 			}
3685 			break;
3686 
3687 		case OC:
3688 			if (!value || !decodeuint(value, &uval)) {
3689 				output_raw = '?';
3690 			} else {
3691 				snprintf(b, sizeof(b), "%03lo", uval);
3692 				output(fp, name, b);
3693 			}
3694 			break;
3695 
3696 		case AR:
3697 			if (!value || !decodearr(value, &narr, lfparr, 8))
3698 				output_raw = '?';
3699 			else
3700 				outputarr(fp, name, narr, lfparr);
3701 			break;
3702 
3703 		case FX:
3704 			if (!value || !decodeuint(value, &uval))
3705 				output_raw = '?';
3706 			else
3707 				output(fp, name, tstflags(uval));
3708 			break;
3709 
3710 		default:
3711 			fprintf(stderr, "Internal error in cookedprint, %s=%s, fmt %d\n",
3712 				name, value, fmt);
3713 			output_raw = '?';
3714 			break;
3715 		}
3716 
3717 		if (output_raw != 0) {
3718 			/* TALOS-CAN-0063: avoid buffer overrun */
3719 			atoascii(name, MAXVARLEN, bn, sizeof(bn));
3720 			if (output_raw != '*') {
3721 				atoascii(value, MAXVALLEN,
3722 					 bv, sizeof(bv) - 1);
3723 				len = strlen(bv);
3724 				bv[len] = output_raw;
3725 				bv[len+1] = '\0';
3726 			} else {
3727 				atoascii(value, MAXVALLEN,
3728 					 bv, sizeof(bv));
3729 			}
3730 			output(fp, bn, bv);
3731 		}
3732 	}
3733 	endoutput(fp);
3734 }
3735 
3736 
3737 /*
3738  * sortassoc - sort associations in the cache into ascending order
3739  */
3740 void
3741 sortassoc(void)
3742 {
3743 	if (numassoc > 1)
3744 		qsort(assoc_cache, (size_t)numassoc,
3745 		      sizeof(assoc_cache[0]), &assoccmp);
3746 }
3747 
3748 
3749 /*
3750  * assoccmp - compare two associations
3751  */
3752 static int
3753 assoccmp(
3754 	const void *t1,
3755 	const void *t2
3756 	)
3757 {
3758 	const struct association *ass1 = t1;
3759 	const struct association *ass2 = t2;
3760 
3761 	if (ass1->assid < ass2->assid)
3762 		return -1;
3763 	if (ass1->assid > ass2->assid)
3764 		return 1;
3765 	return 0;
3766 }
3767 
3768 
3769 /*
3770  * grow_assoc_cache() - enlarge dynamic assoc_cache array
3771  *
3772  * The strategy is to add an assumed 4k page size at a time, leaving
3773  * room for malloc() bookkeeping overhead equivalent to 4 pointers.
3774  */
3775 void
3776 grow_assoc_cache(void)
3777 {
3778 	static size_t	prior_sz;
3779 	size_t		new_sz;
3780 
3781 	new_sz = prior_sz + 4 * 1024;
3782 	if (0 == prior_sz) {
3783 		new_sz -= 4 * sizeof(void *);
3784 	}
3785 	assoc_cache = erealloc_zero(assoc_cache, new_sz, prior_sz);
3786 	prior_sz = new_sz;
3787 	assoc_cache_slots = (u_int)(new_sz / sizeof(assoc_cache[0]));
3788 }
3789 
3790 
3791 /*
3792  * ntpq_custom_opt_handler - autoopts handler for -c and -p
3793  *
3794  * By default, autoopts loses the relative order of -c and -p options
3795  * on the command line.  This routine replaces the default handler for
3796  * those routines and builds a list of commands to execute preserving
3797  * the order.
3798  */
3799 void
3800 ntpq_custom_opt_handler(
3801 	tOptions *pOptions,
3802 	tOptDesc *pOptDesc
3803 	)
3804 {
3805 	switch (pOptDesc->optValue) {
3806 
3807 	default:
3808 		fprintf(stderr,
3809 			"ntpq_custom_opt_handler unexpected option '%c' (%d)\n",
3810 			pOptDesc->optValue, pOptDesc->optValue);
3811 		exit(1);
3812 
3813 	case 'c':
3814 		ADDCMD(pOptDesc->pzLastArg);
3815 		break;
3816 
3817 	case 'p':
3818 		ADDCMD("peers");
3819 		break;
3820 	}
3821 }
3822 /*
3823  * Obtain list of digest names
3824  */
3825 
3826 #if defined(OPENSSL) && !defined(HAVE_EVP_MD_DO_ALL_SORTED)
3827 # if defined(_MSC_VER) && OPENSSL_VERSION_NUMBER >= 0x10100000L
3828 #  define HAVE_EVP_MD_DO_ALL_SORTED
3829 # endif
3830 #endif
3831 
3832 #ifdef OPENSSL
3833 # ifdef HAVE_EVP_MD_DO_ALL_SORTED
3834 #  define K_PER_LINE	8
3835 #  define K_NL_PFX_STR	"\n    "
3836 #  define K_DELIM_STR	", "
3837 
3838 struct hstate {
3839 	char *list;
3840 	const char **seen;
3841 	int idx;
3842 };
3843 
3844 
3845 #  ifndef BUILD_AS_LIB
3846 static void
3847 list_md_fn(const EVP_MD *m, const char *from, const char *to, void *arg)
3848 {
3849 	size_t 	       len, n;
3850 	const char    *name, **seen;
3851 	struct hstate *hstate = arg;
3852 	const char    *cp;
3853 
3854 	/* m is MD obj, from is name or alias, to is base name for alias */
3855 	if (!m || !from || to)
3856 		return; /* Ignore aliases */
3857 
3858 	/* Discard MACs that NTP won't accept. */
3859 	/* Keep this consistent with keytype_from_text() in ssl_init.c. */
3860 	if (EVP_MD_size(m) > (MAX_MAC_LEN - sizeof(keyid_t)))
3861 		return;
3862 
3863 	name = EVP_MD_name(m);
3864 
3865 	/* Lowercase names aren't accepted by keytype_from_text in ssl_init.c */
3866 
3867 	for (cp = name; *cp; cp++)
3868 		if (islower((unsigned char)*cp))
3869 			return;
3870 
3871 	len = (cp - name) + 1;
3872 
3873 	/* There are duplicates.  Discard if name has been seen. */
3874 
3875 	for (seen = hstate->seen; *seen; seen++)
3876 		if (!strcmp(*seen, name))
3877 			return;
3878 
3879 	n = (seen - hstate->seen) + 2;
3880 	hstate->seen = erealloc(hstate->seen, n * sizeof(*seen));
3881 	hstate->seen[n-2] = name;
3882 	hstate->seen[n-1] = NULL;
3883 
3884 	if (hstate->list != NULL)
3885 		len += strlen(hstate->list);
3886 
3887 	len += (hstate->idx >= K_PER_LINE)
3888 	    ? strlen(K_NL_PFX_STR)
3889 	    : strlen(K_DELIM_STR);
3890 
3891 	if (hstate->list == NULL) {
3892 		hstate->list = (char *)emalloc(len);
3893 		hstate->list[0] = '\0';
3894 	} else {
3895 		hstate->list = (char *)erealloc(hstate->list, len);
3896 	}
3897 
3898 	sprintf(hstate->list + strlen(hstate->list), "%s%s",
3899 		((hstate->idx >= K_PER_LINE) ? K_NL_PFX_STR : K_DELIM_STR),
3900 		name);
3901 
3902 	if (hstate->idx >= K_PER_LINE)
3903 		hstate->idx = 1;
3904 	else
3905 		hstate->idx++;
3906 }
3907 #  endif /* !defined(BUILD_AS_LIB) */
3908 
3909 #  ifndef BUILD_AS_LIB
3910 /* Insert CMAC into SSL digests list */
3911 static char *
3912 insert_cmac(char *list)
3913 {
3914 #ifdef ENABLE_CMAC
3915 	int insert;
3916 	size_t len;
3917 
3918 
3919 	/* If list empty, we need to insert CMAC on new line */
3920 	insert = (!list || !*list);
3921 
3922 	if (insert) {
3923 		len = strlen(K_NL_PFX_STR) + strlen(CMAC);
3924 		list = (char *)erealloc(list, len + 1);
3925 		sprintf(list, "%s%s", K_NL_PFX_STR, CMAC);
3926 	} else {	/* List not empty */
3927 		/* Check if CMAC already in list - future proofing */
3928 		const char *cmac_sn;
3929 		char *cmac_p;
3930 
3931 		cmac_sn = OBJ_nid2sn(NID_cmac);
3932 		cmac_p = list;
3933 		insert = cmac_sn != NULL && *cmac_sn != '\0';
3934 
3935 		/* CMAC in list if found, followed by nul char or ',' */
3936 		while (insert && NULL != (cmac_p = strstr(cmac_p, cmac_sn))) {
3937 			cmac_p += strlen(cmac_sn);
3938 			/* Still need to insert if not nul and not ',' */
3939 			insert = *cmac_p && ',' != *cmac_p;
3940 		}
3941 
3942 		/* Find proper insertion point */
3943 		if (insert) {
3944 			char *last_nl;
3945 			char *point;
3946 			char *delim;
3947 			int found;
3948 
3949 			/* Default to start if list empty */
3950 			found = 0;
3951 			delim = list;
3952 			len = strlen(list);
3953 
3954 			/* While new lines */
3955 			while (delim < list + len && *delim &&
3956 			       !strncmp(K_NL_PFX_STR, delim, strlen(K_NL_PFX_STR))) {
3957 				point = delim + strlen(K_NL_PFX_STR);
3958 
3959 				/* While digest names on line */
3960 				while (point < list + len && *point) {
3961 					/* Another digest after on same or next line? */
3962 					delim = strstr( point, K_DELIM_STR);
3963 					last_nl = strstr( point, K_NL_PFX_STR);
3964 
3965 					/* No - end of list */
3966 					if (!delim && !last_nl) {
3967 						delim = list + len;
3968 					} else
3969 						/* New line and no delim or before delim? */
3970 						if (last_nl && (!delim || last_nl < delim)) {
3971 							delim = last_nl;
3972 						}
3973 
3974 					/* Found insertion point where CMAC before entry? */
3975 					if (strncmp(CMAC, point, delim - point) < 0) {
3976 						found = 1;
3977 						break;
3978 					}
3979 
3980 					if (delim < list + len && *delim &&
3981 					    !strncmp(K_DELIM_STR, delim, strlen(K_DELIM_STR))) {
3982 						point += strlen(K_DELIM_STR);
3983 					} else {
3984 						break;
3985 					}
3986 				} /* While digest names on line */
3987 			} /* While new lines */
3988 
3989 			/* If found in list */
3990 			if (found) {
3991 				/* insert cmac and delim */
3992 				/* Space for list could move - save offset */
3993 				ptrdiff_t p_offset = point - list;
3994 				len += strlen(CMAC) + strlen(K_DELIM_STR);
3995 				list = (char *)erealloc(list, len + 1);
3996 				point = list + p_offset;
3997 				/* move to handle src/dest overlap */
3998 				memmove(point + strlen(CMAC) + strlen(K_DELIM_STR),
3999 					point, strlen(point) + 1);
4000 				strncpy(point, CMAC, strlen(CMAC));
4001 				strncpy(point + strlen(CMAC), K_DELIM_STR, strlen(K_DELIM_STR));
4002 			} else {	/* End of list */
4003 				/* append delim and cmac */
4004 				len += strlen(K_DELIM_STR) + strlen(CMAC);
4005 				list = (char *)erealloc(list, len + 1);
4006 				strcpy(list + strlen(list), K_DELIM_STR);
4007 				strcpy(list + strlen(list), CMAC);
4008 			}
4009 		} /* insert */
4010 	} /* List not empty */
4011 #endif /*ENABLE_CMAC*/
4012 	return list;
4013 }
4014 #  endif /* !defined(BUILD_AS_LIB) */
4015 # endif
4016 #endif
4017 
4018 
4019 #ifndef BUILD_AS_LIB
4020 static char *
4021 list_digest_names(void)
4022 {
4023 	char *list = NULL;
4024 
4025 #ifdef OPENSSL
4026 # ifdef HAVE_EVP_MD_DO_ALL_SORTED
4027 	struct hstate hstate = { NULL, NULL, K_PER_LINE+1 };
4028 
4029 	/* replace calloc(1, sizeof(const char *)) */
4030 	hstate.seen = (const char **)emalloc_zero(sizeof(const char *));
4031 
4032 	INIT_SSL();
4033 	EVP_MD_do_all_sorted(list_md_fn, &hstate);
4034 	list = hstate.list;
4035 	free(hstate.seen);
4036 
4037 	list = insert_cmac(list);	/* Insert CMAC into SSL digests list */
4038 
4039 # else
4040 	list = (char *)emalloc(sizeof("md5, others (upgrade to OpenSSL-1.0 for full list)"));
4041 	strcpy(list, "md5, others (upgrade to OpenSSL-1.0 for full list)");
4042 # endif
4043 #else
4044 	list = (char *)emalloc(sizeof("md5"));
4045 	strcpy(list, "md5");
4046 #endif
4047 
4048 	return list;
4049 }
4050 #endif /* !defined(BUILD_AS_LIB) */
4051 
4052 #define CTRLC_STACK_MAX 4
4053 static volatile size_t		ctrlc_stack_len = 0;
4054 static volatile Ctrl_C_Handler	ctrlc_stack[CTRLC_STACK_MAX];
4055 
4056 
4057 
4058 int/*BOOL*/
4059 push_ctrl_c_handler(
4060 	Ctrl_C_Handler func
4061 	)
4062 {
4063 	size_t size = ctrlc_stack_len;
4064 	if (func && (size < CTRLC_STACK_MAX)) {
4065 		ctrlc_stack[size] = func;
4066 		ctrlc_stack_len = size + 1;
4067 		return TRUE;
4068 	}
4069 	return FALSE;
4070 }
4071 
4072 int/*BOOL*/
4073 pop_ctrl_c_handler(
4074 	Ctrl_C_Handler func
4075 	)
4076 {
4077 	size_t size = ctrlc_stack_len;
4078 	if (size) {
4079 		--size;
4080 		if (func == NULL || func == ctrlc_stack[size]) {
4081 			ctrlc_stack_len = size;
4082 			return TRUE;
4083 		}
4084 	}
4085 	return FALSE;
4086 }
4087 
4088 #ifndef BUILD_AS_LIB
4089 static void
4090 on_ctrlc(void)
4091 {
4092 	size_t size = ctrlc_stack_len;
4093 	while (size)
4094 		if ((*ctrlc_stack[--size])())
4095 			break;
4096 }
4097 #endif /* !defined(BUILD_AS_LIB) */
4098 
4099 #ifndef BUILD_AS_LIB
4100 static int
4101 my_easprintf(
4102 	char ** 	ppinto,
4103 	const char *	fmt   ,
4104 	...
4105 	)
4106 {
4107 	va_list	va;
4108 	int	prc;
4109 	size_t	len = 128;
4110 	char *	buf = emalloc(len);
4111 
4112   again:
4113 	/* Note: we expect the memory allocation to fail long before the
4114 	 * increment in buffer size actually overflows.
4115 	 */
4116 	buf = (buf) ? erealloc(buf, len) : emalloc(len);
4117 
4118 	va_start(va, fmt);
4119 	prc = vsnprintf(buf, len, fmt, va);
4120 	va_end(va);
4121 
4122 	if (prc < 0) {
4123 		/* might be very old vsnprintf. Or actually MSVC... */
4124 		len += len >> 1;
4125 		goto again;
4126 	}
4127 	if ((size_t)prc >= len) {
4128 		/* at least we have the proper size now... */
4129 		len = (size_t)prc + 1;
4130 		goto again;
4131 	}
4132 	if ((size_t)prc < (len - 32))
4133 		buf = erealloc(buf, (size_t)prc + 1);
4134 	*ppinto = buf;
4135 	return prc;
4136 }
4137 #endif /* !defined(BUILD_AS_LIB) */
4138