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