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