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