chiark / gitweb /
admin: Put all command options at the start of the command-line.
[tripe] / server / admin.c
CommitLineData
410c8acf 1/* -*-c-*-
2 *
2d752320 3 * $Id$
410c8acf 4 *
5 * Admin interface for configuration
6 *
7 * (c) 2001 Straylight/Edgeware
8 */
9
10/*----- Licensing notice --------------------------------------------------*
11 *
12 * This file is part of Trivial IP Encryption (TrIPE).
13 *
14 * TrIPE is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU General Public License as published by
16 * the Free Software Foundation; either version 2 of the License, or
17 * (at your option) any later version.
18 *
19 * TrIPE is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 * GNU General Public License for more details.
23 *
24 * You should have received a copy of the GNU General Public License
25 * along with TrIPE; if not, write to the Free Software Foundation,
26 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
27 */
28
410c8acf 29/*----- Header files ------------------------------------------------------*/
30
31#include "tripe.h"
32
33/*----- Global variables --------------------------------------------------*/
34
35#ifndef NTRACE
36
37const trace_opt tr_opts[] = {
38 { 't', T_TUNNEL, "tunnel events" },
39 { 'r', T_PEER, "peer events" },
40 { 'a', T_ADMIN, "admin interface" },
410c8acf 41 { 's', T_KEYSET, "symmetric keyset management" },
42 { 'x', T_KEYEXCH, "key exchange" },
43 { 'm', T_KEYMGMT, "key management" },
37941236 44 { 'l', T_CHAL, "challenge management" },
45 { 'p', T_PACKET, "packet contents" },
46 { 'c', T_CRYPTO, "crypto details" },
410c8acf 47 { 'A', T_ALL, "all of the above" },
48 { 0, 0, 0 }
49};
50
51unsigned tr_flags = 0;
52#endif
53
3cdc3f3a 54static const trace_opt w_opts[] = {
55 { 't', AF_TRACE, "trace messages" },
56 { 'n', AF_NOTE, "asynchronous notifications" },
57 { 'w', AF_WARN, "warnings" },
58 { 'A', AF_ALLMSGS, "all of the above" },
59 { 0, 0, 0 }
60};
61
410c8acf 62/*----- Static variables --------------------------------------------------*/
63
64static admin *admins;
c511e1f9 65static admin *a_dead;
410c8acf 66static sel_file sock;
67static const char *sockname;
68static unsigned flags = 0;
69static admin *a_stdin = 0;
70static sig s_term, s_int, s_hup;
71
72#define F_DAEMON 1u
73#define F_INIT 2u
74
75#define T_RESOLVE SEC(30)
0ba8de86 76#define T_PING SEC(5)
410c8acf 77
fd3cf232 78static void a_destroy(admin */*a*/);
fd3cf232 79
060ca767 80#define BOOL(x) ((x) ? "t" : "nil")
81
fd3cf232 82/*----- Output functions --------------------------------------------------*/
83
84/* --- @trywrite@ --- *
85 *
86 * Arguments: @admin *a@ = pointer to an admin block
87 * @const char *p@ = pointer to buffer to write
88 * @size_t sz@ = size of data to write
89 *
90 * Returns: The number of bytes written, or less than zero on error.
91 *
92 * Use: Attempts to write data to a client.
93 */
94
95static ssize_t trywrite(admin *a, const char *p, size_t sz)
96{
97 ssize_t n, done = 0;
98
99again:
100 if (!sz)
101 return (done);
102 n = write(a->w.fd, p, sz);
103 if (n > 0) {
104 done += n;
105 p += n;
106 sz -= n;
107 goto again;
108 }
109 if (n < 0) {
110 if (errno == EINTR)
111 goto again;
112 if (errno != EAGAIN && errno != EWOULDBLOCK) {
113 a_destroy(a);
f43df819 114 a_warn("ADMIN", "client-write-error", "?ERRNO", A_END);
fd3cf232 115 return (-1);
116 }
117 }
118 return (done);
119}
120
de014da6 121/* --- @doqueue@ -- *
fd3cf232 122 *
de014da6 123 * Arguments: @oqueue *q@ = pointer to output queue
fd3cf232 124 * @const char *p@ = pointer to buffer to write
de014da6 125 * @size_t sz@ = size of buffer
fd3cf232 126 *
de014da6 127 * Returns: Nonzero if the queue was previously empty.
fd3cf232 128 *
de014da6 129 * Use: Queues data to be written later.
fd3cf232 130 */
131
de014da6 132static int doqueue(oqueue *q, const char *p, size_t sz)
fd3cf232 133{
fd3cf232 134 obuf *o;
de014da6 135 int rc = 0;
136 size_t n;
fd3cf232 137
de014da6 138 o = q->tl;
fd3cf232 139 if (!o)
de014da6 140 rc = 1;
fd3cf232 141 else if (o->p_in < o->buf + OBUFSZ)
142 goto noalloc;
143
144 do {
145 o = xmalloc(sizeof(obuf));
146 o->next = 0;
147 o->p_in = o->p_out = o->buf;
de014da6 148 if (q->tl)
149 q->tl->next = o;
fd3cf232 150 else
de014da6 151 q->hd = o;
152 q->tl = o;
fd3cf232 153
154 noalloc:
155 n = o->buf + OBUFSZ - o->p_in;
156 if (n > sz)
157 n = sz;
158 memcpy(o->p_in, p, n);
159 o->p_in += n;
160 p += n;
161 sz -= n;
162 } while (sz);
de014da6 163
164 return (rc);
165}
166
167/* --- @dosend@ --- *
168 *
169 * Arguemnts: @admin *a@ = pointer to an admin block
170 * @const char *p@ = pointer to buffer to write
171 * @size_t sz@ = size of data to write
172 *
173 * Returns: ---
174 *
175 * Use: Sends data to an admin client.
176 */
177
178static void dosend(admin *a, const char *p, size_t sz)
179{
180 ssize_t n;
181
182 if (a->f & AF_DEAD)
183 return;
184 if (!a->out.hd) {
185 if ((n = trywrite(a, p, sz)) < 0)
186 return;
187 p += n;
188 sz -= n;
189 if (!sz)
190 return;
191 }
192 if (doqueue(&a->out, p, sz))
193 sel_addfile(&a->w);
fd3cf232 194}
195
196/* --- @a_flush@ --- *
197 *
198 * Arguments: @int fd@ = file descriptor
199 * @unsigned mode@ = what's happening
200 * @void *v@ = pointer to my admin block
201 *
202 * Returns: ---
203 *
204 * Use: Flushes buffers when a client is ready to read again.
205 */
206
207static void a_flush(int fd, unsigned mode, void *v)
208{
209 admin *a = v;
210 obuf *o, *oo;
211 ssize_t n;
212
de014da6 213 o = a->out.hd;
fd3cf232 214 while (o) {
215 if ((n = trywrite(a, o->p_out, o->p_in - o->p_out)) < 0)
216 return;
217 o->p_out += n;
218 if (o->p_in < o->p_out)
219 break;
220 oo = o;
221 o = o->next;
222 xfree(oo);
223 }
de014da6 224 a->out.hd = o;
fd3cf232 225 if (!o) {
de014da6 226 a->out.tl = 0;
fd3cf232 227 sel_rmfile(&a->w);
228 }
229}
230
410c8acf 231/*----- Utility functions -------------------------------------------------*/
232
f43df819
MW
233/* --- @quotify@ --- *
234 *
235 * Arguments: @dstr *d@ = where to write the answer
236 * @const char *p@ = string to quotify
237 *
238 * Returns: ---
239 *
240 * Use: Quotes the given string if necessary, according to our
241 * quoting rules.
242 */
243
244static void quotify(dstr *d, const char *p)
245{
246 if (d->len)
247 dstr_putc(d, ' ');
248 if (*p && !p[strcspn(p, "\"' \t\n\v")])
249 dstr_puts(d, p);
250 else {
251 dstr_putc(d, '\"');
252 while (*p) {
253 if (*p == '\\' || *p == '\"')
254 dstr_putc(d, '\\');
255 dstr_putc(d, *p++);
256 }
257 dstr_putc(d, '\"');
258 }
259}
260
261/* --- @a_vformat@ --- *
262 *
263 * Arguments: @dstr *d@ = where to leave the formatted message
264 * @const char *fmt@ = pointer to format string
265 * @va_list ap@ = arguments in list
266 *
267 * Returns: ---
268 *
269 * Use: Main message token formatting driver.
270 */
271
272static void a_vformat(dstr *d, const char *fmt, va_list ap)
273{
274 dstr dd = DSTR_INIT;
275
276 while (fmt) {
277 if (*fmt == '*') {
278 dstr_putc(d, ' ');
279 dstr_vputf(d, fmt + 1, &ap);
280 } else if (*fmt == '?') {
281 if (strcmp(fmt, "?ADDR") == 0) {
282 const addr *a = va_arg(ap, const addr *);
283 switch (a->sa.sa_family) {
284 case AF_INET:
285 quotify(d, "INET");
286 quotify(d, inet_ntoa(a->sin.sin_addr));
287 dstr_putf(d, " %u", (unsigned)ntohs(a->sin.sin_port));
288 break;
289 default:
290 abort();
291 }
292 } else if (strcmp(fmt, "?B64") == 0) {
293 const octet *p = va_arg(ap, const octet *);
294 size_t n = va_arg(ap, size_t);
295 base64_ctx b64;
296 dstr_putc(d, ' ');
297 base64_init(&b64);
298 b64.indent = "";
299 b64.maxline = 0;
300 base64_encode(&b64, p, n, d);
301 base64_encode(&b64, 0, 0, d);
302 while (d->len && d->buf[d->len - 1] == '=') d->len--;
303 } else if (strcmp(fmt, "?PEER") == 0)
304 quotify(d, p_name(va_arg(ap, peer *)));
305 else if (strcmp(fmt, "?ERRNO") == 0) {
306 dstr_putf(d, " E%d", errno);
307 quotify(d, strerror(errno));
308 } else
309 abort();
310 } else {
311 if (*fmt == '!') fmt++;
312 DRESET(&dd);
313 dstr_vputf(&dd, fmt, &ap);
314 quotify(d, dd.buf);
315 }
316 fmt = va_arg(ap, const char *);
317 }
318
319 dstr_destroy(&dd);
320}
321
3cdc3f3a 322/* --- @a_write@, @a_vwrite@ --- *
410c8acf 323 *
324 * Arguments: @admin *a@ = admin connection to write to
de014da6 325 * @const char *status@ = status code to report
326 * @const char *tag@ = tag string, or null
410c8acf 327 * @const char *fmt@ = pointer to format string
3cdc3f3a 328 * @va_list ap@ = arguments in list
410c8acf 329 * @...@ = other arguments
330 *
331 * Returns: ---
332 *
333 * Use: Sends a message to an admin connection.
334 */
335
de014da6 336static void a_vwrite(admin *a, const char *status, const char *tag,
337 const char *fmt, va_list ap)
410c8acf 338{
410c8acf 339 dstr d = DSTR_INIT;
de014da6 340 if (tag) dstr_puts(&d, "BG");
341 dstr_puts(&d, status);
f43df819
MW
342 if (tag) quotify(&d, tag);
343 a_vformat(&d, fmt, ap);
3cdc3f3a 344 dstr_putc(&d, '\n');
345 dosend(a, d.buf, d.len);
346 dstr_destroy(&d);
347}
348
de014da6 349static void a_write(admin *a, const char *status, const char *tag,
350 const char *fmt, ...)
3cdc3f3a 351{
352 va_list ap;
410c8acf 353 va_start(ap, fmt);
de014da6 354 a_vwrite(a, status, tag, fmt, ap);
410c8acf 355 va_end(ap);
3cdc3f3a 356}
357
358/* --- @a_ok@, @a_info@, @a_fail@ --- *
359 *
360 * Arguments: @admin *a@ = connection
361 * @const char *fmt@ = format string
362 * @...@ = other arguments
363 *
364 * Returns: ---
365 *
366 * Use: Convenience functions for @a_write@.
367 */
368
f43df819 369static void a_ok(admin *a) { a_write(a, "OK", 0, A_END); }
3cdc3f3a 370
371static void a_info(admin *a, const char *fmt, ...)
372{
373 va_list ap;
374 va_start(ap, fmt);
de014da6 375 a_vwrite(a, "INFO", 0, fmt, ap);
3cdc3f3a 376 va_end(ap);
377}
378
379static void a_fail(admin *a, const char *fmt, ...)
380{
381 va_list ap;
382 va_start(ap, fmt);
de014da6 383 a_vwrite(a, "FAIL", 0, fmt, ap);
3cdc3f3a 384 va_end(ap);
385}
386
387/* --- @a_alert@, @a_valert@, @a_rawalert@ --- *
388 *
389 * Arguments: @unsigned f_and, f_eq@ = filter for connections
de014da6 390 * @const char *status@ = status string
3cdc3f3a 391 * @const char *fmt@ = pointer to format string
de014da6 392 * @const char *p@ = pointer to raw string
3cdc3f3a 393 * @size_t sz@ = size of raw string
394 * @va_list ap@ = arguments in list
395 * @...@ = other arguments
396 *
397 * Returns: ---
398 *
399 * Use: Write a message to all admin connections matched by the given
400 * filter.
401 */
402
de014da6 403static void a_rawalert(unsigned f_and, unsigned f_eq, const char *status,
3cdc3f3a 404 const char *p, size_t sz)
405{
406 admin *a, *aa;
407 dstr d = DSTR_INIT;
408
409 if (!(flags & F_INIT))
410 return;
de014da6 411 dstr_puts(&d, status);
412 if (p) {
413 dstr_putc(&d, ' ');
3cdc3f3a 414 dstr_putm(&d, p, sz);
de014da6 415 }
3cdc3f3a 416 dstr_putc(&d, '\n');
417 p = d.buf;
418 sz = d.len;
419 for (a = admins; a; a = aa) {
420 aa = a->next;
421 if ((a->f & f_and) == f_eq)
422 dosend(a, d.buf, d.len);
423 }
410c8acf 424 dstr_destroy(&d);
425}
426
3cdc3f3a 427static void a_valert(unsigned f_and, unsigned f_eq, const char *tag,
428 const char *fmt, va_list ap)
429{
430 dstr d = DSTR_INIT;
431
432 if (!(flags & F_INIT))
433 return;
f43df819 434 a_vformat(&d, fmt, ap);
3cdc3f3a 435 a_rawalert(f_and, f_eq, tag, fmt ? d.buf : 0, fmt ? d.len : 0);
436 dstr_destroy(&d);
437}
438
439#if 0 /*unused*/
440static void a_alert(unsigned f_and, unsigned f_eq, const char *tag,
441 const char *fmt, ...)
442{
443 va_list ap;
444 va_start(ap, fmt);
445 a_valert(f_and, f_eq, tag, fmt, ap);
446 va_end(ap);
447}
448#endif
449
410c8acf 450/* --- @a_warn@ --- *
451 *
452 * Arguments: @const char *fmt@ = pointer to format string
453 * @...@ = other arguments
454 *
455 * Returns: ---
456 *
457 * Use: Informs all admin connections of a warning.
458 */
459
460void a_warn(const char *fmt, ...)
461{
462 va_list ap;
410c8acf 463
410c8acf 464 va_start(ap, fmt);
3cdc3f3a 465 if (flags & F_INIT)
466 a_valert(0, 0, "WARN", fmt, ap);
410c8acf 467 else {
f43df819 468 dstr d = DSTR_INIT;
3cdc3f3a 469 fprintf(stderr, "%s: ", QUIS);
f43df819
MW
470 a_vformat(&d, fmt, ap);
471 dstr_putc(&d, '\n');
472 dstr_write(&d, stderr);
473 dstr_destroy(&d);
410c8acf 474 }
3cdc3f3a 475 va_end(ap);
410c8acf 476}
477
478/* --- @a_trace@ --- *
479 *
480 * Arguments: @const char *p@ = pointer to a buffer
481 * @size_t sz@ = size of the buffer
482 * @void *v@ = uninteresting pointer
483 *
484 * Returns: ---
485 *
3cdc3f3a 486 * Use: Custom trace output handler. Sends trace messages to
487 * interested admin connections.
410c8acf 488 */
489
490#ifndef NTRACE
491static void a_trace(const char *p, size_t sz, void *v)
de014da6 492 { a_rawalert(AF_TRACE, AF_TRACE, "TRACE", p, sz); }
410c8acf 493#endif
494
3cdc3f3a 495/* --- @a_notify@ --- *
496 *
497 * Arguments: @const char *fmt@ = pointer to format string
498 * @...@ = other arguments
499 *
500 * Returns: ---
501 *
502 * Use: Sends a notification to interested admin connections.
503 */
504
505void a_notify(const char *fmt, ...)
506{
507 va_list ap;
508
509 va_start(ap, fmt);
510 a_valert(AF_NOTE, AF_NOTE, "NOTE", fmt, ap);
511 va_end(ap);
512}
513
410c8acf 514/* --- @a_quit@ --- *
515 *
516 * Arguments: ---
517 *
518 * Returns: ---
519 *
520 * Use: Shuts things down nicely.
521 */
522
523void a_quit(void)
524{
42da2a58 525 peer *p;
526
410c8acf 527 close(sock.fd);
528 unlink(sockname);
de014da6 529 while ((p = p_first()) != 0)
530 p_destroy(p);
410c8acf 531 exit(0);
532}
533
534/* --- @a_sigdie@ --- *
535 *
536 * Arguments: @int sig@ = signal number
537 * @void *v@ = an uninteresting argument
538 *
539 * Returns: ---
540 *
541 * Use Shuts down on receipt of a fatal signal.
542 */
543
544static void a_sigdie(int sig, void *v)
545{
546 char *p;
547 char buf[20];
548
549 switch (sig) {
550 case SIGTERM: p = "SIGTERM"; break;
551 case SIGINT: p = "SIGINT"; break;
552 default:
3cdc3f3a 553 sprintf(buf, "%i", sig);
410c8acf 554 p = buf;
555 break;
556 }
f43df819 557 a_warn("SERVER", "quit", "signal", "%s", p, A_END);
410c8acf 558 a_quit();
559}
560
561/* --- @a_sighup@ --- *
562 *
563 * Arguments: @int sig@ = signal number
564 * @void *v@ = an uninteresting argument
565 *
566 * Returns: ---
567 *
568 * Use Logs a message about SIGHUP not being useful.
569 */
570
571static void a_sighup(int sig, void *v)
572{
f43df819 573 a_warn("SERVER", "ignore", "signal", "SIGHUP", A_END);
410c8acf 574}
575
0ba8de86 576/* --- @a_parsetime@ --- *
577 *
578 * Arguments; @const char *p@ = time string to parse
579 *
580 * Returns: Time in seconds, or @< 0@ on error.
581 */
582
583static long a_parsetime(const char *p)
584{
585 char *q;
586 long t = strtol(p, &q, 0);
587
588 switch (*q) {
589 case 'd': t *= 24;
590 case 'h': t *= 60;
591 case 'm': t *= 60;
592 case 's': if (q[1] != 0)
593 default: t = -1;
594 case 0: break;
595 }
596 return (t);
597}
598
37941236 599/* --- @a_findpeer@ --- *
600 *
601 * Arguments: @admin *a@ = admin connection
602 * @const char *pn@ = peer name
603 *
604 * Returns: The peer, or null if not there.
605 *
606 * Use: Finds a peer, reporting an error if it failed.
607 */
608
609static peer *a_findpeer(admin *a, const char *pn)
610{
611 peer *p;
612
613 if ((p = p_find(pn)) == 0)
f43df819 614 a_fail(a, "unknown-peer", "%s", pn, A_END);
37941236 615 return (p);
616}
617
de014da6 618/*----- Backgrounded operations -------------------------------------------*/
619
060ca767 620#define BGTAG(bg) \
621 (((admin_bgop *)(bg))->tag ? ((admin_bgop *)(bg))->tag : "<foreground>")
622
de014da6 623/* --- @a_bgrelease@ --- *
624 *
625 * Arguments: @admin_bgop *bg@ = backgrounded operation
626 *
627 * Returns: ---
628 *
629 * Use: Removes a backgrounded operation from the queue, since
630 * (presumably) it's done.
631 */
632
633static void a_bgrelease(admin_bgop *bg)
634{
635 admin *a = bg->a;
636
060ca767 637 T( trace(T_ADMIN, "admin: release bgop %s", BGTAG(bg)); )
638 if (bg->tag) xfree(bg->tag);
639 else selbuf_enable(&a->b);
640 if (bg->next) bg->next->prev = bg->prev;
641 if (bg->prev) bg->prev->next = bg->next;
642 else a->bg = bg->next;
de014da6 643 xfree(bg);
547a9c13 644 if (!a->bg && (a->f & AF_CLOSE)) a_destroy(a);
de014da6 645}
646
647/* --- @a_bgok@, @a_bginfo@, @a_bgfail@ --- *
648 *
649 * Arguments: @admin_bgop *bg@ = backgrounded operation
650 * @const char *fmt@ = format string
651 * @...@ = other arguments
652 *
653 * Returns: ---
654 *
655 * Use: Convenience functions for @a_write@.
656 */
657
658static void a_bgok(admin_bgop *bg)
f43df819 659 { a_write(bg->a, "OK", bg->tag, A_END); }
de014da6 660
661static void a_bginfo(admin_bgop *bg, const char *fmt, ...)
662{
663 va_list ap;
664 va_start(ap, fmt);
665 a_vwrite(bg->a, "INFO", bg->tag, fmt, ap);
666 va_end(ap);
667}
668
669static void a_bgfail(admin_bgop *bg, const char *fmt, ...)
670{
671 va_list ap;
672 va_start(ap, fmt);
673 a_vwrite(bg->a, "FAIL", bg->tag, fmt, ap);
674 va_end(ap);
675}
676
677/* --- @a_bgadd@ --- *
678 *
679 * Arguments: @admin *a@ = administration connection
680 * @admin_bgop *bg@ = pointer to background operation
681 * @const char *tag@ = background tag, or null for foreground
682 * @void (*cancel)(admin_bgop *)@ = cancel function
683 *
684 * Returns: ---
685 *
686 * Use: Links a background job into the list.
687 */
688
689static void a_bgadd(admin *a, admin_bgop *bg, const char *tag,
690 void (*cancel)(admin_bgop *))
691{
692 if (tag)
693 bg->tag = xstrdup(tag);
694 else {
695 bg->tag = 0;
696 selbuf_disable(&a->b);
697 }
698 bg->a = a;
699 bg->cancel = cancel;
700 bg->next = a->bg;
701 bg->prev = 0;
060ca767 702 if (a->bg) a->bg->prev = bg;
de014da6 703 a->bg = bg;
060ca767 704 T( trace(T_ADMIN, "admin: add bgop %s", BGTAG(bg)); )
f43df819 705 if (tag) a_write(a, "DETACH", tag, A_END);
de014da6 706}
707
37941236 708/*----- Name resolution operations ----------------------------------------*/
de014da6 709
37941236 710/* --- @a_resolved@ --- *
de014da6 711 *
37941236 712 * Arguments: @struct hostent *h@ = pointer to resolved hostname
713 * @void *v@ = pointer to resolver operation
de014da6 714 *
715 * Returns: ---
716 *
37941236 717 * Use: Handles a completed name resolution.
de014da6 718 */
719
37941236 720static void a_resolved(struct hostent *h, void *v)
de014da6 721{
37941236 722 admin_resop *r = v;
723
724 T( trace(T_ADMIN, "admin: resop %s resolved", BGTAG(r)); )
725 TIMER;
726 if (!h) {
f43df819 727 a_bgfail(&r->bg, "resolve-error", "%s", r->addr, A_END);
37941236 728 r->func(r, ARES_FAIL);
729 } else {
730 memcpy(&r->sa.sin.sin_addr, h->h_addr, sizeof(struct in_addr));
731 r->func(r, ARES_OK);
732 }
733 sel_rmtimer(&r->t);
734 xfree(r->addr);
735 a_bgrelease(&r->bg);
736}
de014da6 737
37941236 738/* --- @a_restimer@ --- *
de014da6 739 *
37941236 740 * Arguments: @struct timeval *tv@ = timer
741 * @void *v@ = pointer to resolver operation
de014da6 742 *
743 * Returns: ---
744 *
37941236 745 * Use: Times out a resolver.
de014da6 746 */
747
37941236 748static void a_restimer(struct timeval *tv, void *v)
de014da6 749{
37941236 750 admin_resop *r = v;
751
752 T( trace(T_ADMIN, "admin: resop %s timeout", BGTAG(r)); )
f43df819 753 a_bgfail(&r->bg, "resolver-timeout", "%s", r->addr, A_END);
37941236 754 r->func(r, ARES_FAIL);
755 bres_abort(&r->r);
756 xfree(r->addr);
757 a_bgrelease(&r->bg);
de014da6 758}
759
37941236 760/* --- @a_rescancel@ --- *
de014da6 761 *
37941236 762 * Arguments: @admin_bgop *bg@ = background operation
de014da6 763 *
764 * Returns: ---
765 *
37941236 766 * Use: Cancels an add operation.
de014da6 767 */
768
37941236 769static void a_rescancel(admin_bgop *bg)
de014da6 770{
37941236 771 admin_resop *r = (admin_resop *)bg;
772
773 T( trace(T_ADMIN, "admin: cancel resop %s", BGTAG(r)); )
774 r->func(r, ARES_FAIL);
775 sel_rmtimer(&r->t);
776 xfree(r->addr);
777 bres_abort(&r->r);
de014da6 778}
37941236 779
780/* --- @a_resolve@ --- *
410c8acf 781 *
37941236 782 * Arguments: @admin *a@ = administration connection
783 * @admin_resop *r@ = resolver operation to run
784 * @const char *tag@ = background operation tag
785 * @void (*func)(struct admin_resop *, int@ = handler function
786 * @unsigned ac@ = number of remaining arguments
787 * @char *av[]@ = pointer to remaining arguments
410c8acf 788 *
789 * Returns: ---
790 *
37941236 791 * Use: Cranks up a resolver job.
410c8acf 792 */
793
37941236 794static void a_resolve(admin *a, admin_resop *r, const char *tag,
795 void (*func)(struct admin_resop *, int),
796 unsigned ac, char *av[])
410c8acf 797{
37941236 798 struct timeval tv;
799 unsigned long pt;
800 char *p;
801 int i = 0;
fd3cf232 802
37941236 803 /* --- Fill in the easy bits of address --- */
804
452bf3f6 805 r->bg.tag = "<starting>";
37941236 806 r->addr = 0;
807 r->func = func;
808 if (mystrieq(av[i], "inet")) i++;
809 if (ac - i != 2) {
f43df819 810 a_fail(a, "bad-addr-syntax", "[inet] ADDRESS PORT", A_END);
37941236 811 goto fail;
812 }
813 r->sa.sin.sin_family = AF_INET;
814 r->sasz = sizeof(r->sa.sin);
815 r->addr = xstrdup(av[i]);
816 pt = strtoul(av[i + 1], &p, 0);
817 if (*p) {
818 struct servent *s = getservbyname(av[i + 1], "udp");
819 if (!s) {
f43df819 820 a_fail(a, "unknown-service", "%s", av[i + 1], A_END);
37941236 821 goto fail;
822 }
823 pt = ntohs(s->s_port);
410c8acf 824 }
37941236 825 if (pt == 0 || pt >= 65536) {
f43df819 826 a_fail(a, "invalid-port", "%lu", pt, A_END);
37941236 827 goto fail;
828 }
829 r->sa.sin.sin_port = htons(pt);
830
831 /* --- Report backgrounding --- *
832 *
833 * Do this for consistency of interface, even if we're going to get the
834 * answer straight away.
835 */
836
837 a_bgadd(a, &r->bg, tag, a_rescancel);
838 T( trace(T_ADMIN, "admin: %u, resop %s, hostname `%s'",
839 a->seq, BGTAG(r), r->addr); )
840
841 /* --- If the name is numeric, do it the easy way --- */
842
843 if (inet_aton(av[i], &r->sa.sin.sin_addr)) {
844 T( trace(T_ADMIN, "admin: resop %s done the easy way", BGTAG(r)); )
845 func(r, ARES_OK);
846 xfree(r->addr);
847 a_bgrelease(&r->bg);
848 return;
849 }
850
851 /* --- Store everything for later and crank up the resolver --- */
852
853 gettimeofday(&tv, 0);
854 tv.tv_sec += T_RESOLVE;
855 sel_addtimer(&sel, &r->t, &tv, a_restimer, r);
856 bres_byname(&r->r, r->addr, a_resolved, r);
857 return;
858
859fail:
860 func(r, ARES_FAIL);
861 if (r->addr) xfree(r->addr);
862 xfree(r);
410c8acf 863}
864
37941236 865/*----- Adding peers ------------------------------------------------------*/
866
867/* --- @a_doadd@ --- *
410c8acf 868 *
37941236 869 * Arguments: @admin_resop *r@ = resolver operation
870 * @int rc@ = how it worked
410c8acf 871 *
872 * Returns: ---
873 *
37941236 874 * Use: Handles a completed resolution.
410c8acf 875 */
876
37941236 877static void a_doadd(admin_resop *r, int rc)
410c8acf 878{
37941236 879 admin_addop *add = (admin_addop *)r;
880
881 T( trace(T_ADMIN, "admin: done add op %s", BGTAG(add)); )
882
883 if (rc == ARES_OK) {
884 add->peer.sasz = add->r.sasz;
885 add->peer.sa = add->r.sa;
886 if (p_find(add->peer.name))
f43df819 887 a_bgfail(&add->r.bg, "peer-exists", "%s", add->peer.name, A_END);
37941236 888 else if (!p_create(&add->peer))
f43df819 889 a_bgfail(&add->r.bg, "peer-create-fail", "%s", add->peer.name, A_END);
37941236 890 else
891 a_bgok(&add->r.bg);
892 }
de014da6 893
37941236 894 xfree(add->peer.name);
410c8acf 895}
896
897/* --- @acmd_add@ --- *
898 *
899 * Arguments: @admin *a@ = connection which requested the addition
900 * @unsigned ac@ = argument count
901 * @char *av[]@ = pointer to the argument list
902 *
903 * Returns: ---
904 *
905 * Use: Adds a new peer.
906 */
907
908static void acmd_add(admin *a, unsigned ac, char *av[])
909{
42da2a58 910 unsigned i, j;
de014da6 911 const char *tag = 0;
9986f0b5 912 admin_addop *add;
410c8acf 913
0ba8de86 914 /* --- Set stuff up --- */
915
de014da6 916 add = xmalloc(sizeof(*add));
9986f0b5 917 add->peer.name = 0;
de014da6 918 add->peer.t_ka = 0;
919 add->peer.tops = tun_default;
37941236 920
42da2a58 921 /* --- Parse options --- */
922
9986f0b5 923 i = 0;
42da2a58 924 for (;;) {
925 if (!av[i])
926 goto bad_syntax;
de014da6 927 if (mystrieq(av[i], "-background")) {
928 if (!av[++i]) goto bad_syntax;
929 tag = av[i];
930 } else if (mystrieq(av[i], "-tunnel")) {
0ba8de86 931 if (!av[++i]) goto bad_syntax;
42da2a58 932 for (j = 0;; j++) {
933 if (!tunnels[j]) {
f43df819 934 a_fail(a, "unknown-tunnel", "%s", av[i], A_END);
37941236 935 goto fail;
42da2a58 936 }
937 if (mystrieq(av[i], tunnels[j]->name)) {
de014da6 938 add->peer.tops = tunnels[j];
42da2a58 939 break;
940 }
941 }
0ba8de86 942 } else if (mystrieq(av[i], "-keepalive")) {
943 long t;
944 if (!av[++i]) goto bad_syntax;
945 if ((t = a_parsetime(av[i])) < 0) {
f43df819 946 a_fail(a, "bad-time-spec", "%s", av[i], A_END);
37941236 947 goto fail;
0ba8de86 948 }
de014da6 949 add->peer.t_ka = t;
42da2a58 950 } else if (mystrieq(av[i], "--")) {
951 i++;
952 break;
953 } else
954 break;
0ba8de86 955 i++;
42da2a58 956 }
957
9986f0b5
MW
958 /* --- Make sure someone's not got there already --- */
959
960 if (!av[i])
961 goto bad_syntax;
962 if (p_find(av[i])) {
963 a_fail(a, "peer-exists", "%s", av[i], A_END);
964 goto fail;
965 }
966 add->peer.name = xstrdup(av[i++]);
967
37941236 968 /* --- Crank up the resolver --- */
de014da6 969
37941236 970 a_resolve(a, &add->r, tag, a_doadd, ac - i, av + i);
42da2a58 971 return;
972
37941236 973 /* --- Clearing up --- */
974
42da2a58 975bad_syntax:
9986f0b5 976 a_fail(a, "bad-syntax", "add", "[OPTIONS] PEER ADDR ...", A_END);
de014da6 977fail:
9986f0b5 978 if (add->peer.name) xfree(add->peer.name);
37941236 979 xfree(add);
0ba8de86 980 return;
981}
982
983/*----- Ping --------------------------------------------------------------*/
984
de014da6 985/* --- @a_pingcancel@ --- *
986 *
987 * Arguments: @admin_bgop *bg@ = background operation block
988 *
989 * Returns: ---
990 *
991 * Use: Cancels a running ping.
992 */
993
994static void a_pingcancel(admin_bgop *bg)
995{
996 admin_pingop *pg = (admin_pingop *)bg;
060ca767 997 T( trace(T_ADMIN, "admin: cancel ping op %s", BGTAG(pg)); )
de014da6 998 p_pingdone(&pg->ping, PING_NONOTIFY);
999}
1000
0ba8de86 1001/* --- @a_pong@ --- *
1002 *
1003 * Arguments: @int rc@ = return code
de014da6 1004 * @void *v@ = ping operation block
0ba8de86 1005 *
1006 * Returns: ---
1007 *
1008 * Use: Collects what happened to a ping message.
1009 */
1010
de014da6 1011static void a_pong(int rc, void *v)
0ba8de86 1012{
de014da6 1013 admin_pingop *pg = v;
0ba8de86 1014 struct timeval tv;
1015 double millis;
1016
0ba8de86 1017 switch (rc) {
1018 case PING_OK:
1019 gettimeofday(&tv, 0);
de014da6 1020 tv_sub(&tv, &tv, &pg->pingtime);
0ba8de86 1021 millis = (double)tv.tv_sec * 1000 + (double)tv.tv_usec/1000;
f43df819 1022 a_bginfo(&pg->bg, "ping-ok", "%.1f", millis, A_END);
de014da6 1023 a_bgok(&pg->bg);
0ba8de86 1024 break;
1025 case PING_TIMEOUT:
f43df819 1026 a_bginfo(&pg->bg, "ping-timeout", A_END);
de014da6 1027 a_bgok(&pg->bg);
0ba8de86 1028 break;
1029 case PING_PEERDIED:
f43df819 1030 a_bginfo(&pg->bg, "ping-peer-died", A_END);
de014da6 1031 a_bgok(&pg->bg);
0ba8de86 1032 break;
1033 default:
1034 abort();
1035 }
060ca767 1036 T( trace(T_ADMIN, "admin: ponged ping op %s", BGTAG(pg)); )
de014da6 1037 a_bgrelease(&pg->bg);
0ba8de86 1038}
1039
1040/* --- @acmd_ping@, @acmd_eping@ --- *
1041 *
1042 * Arguments: @admin *a@ = connection which requested the ping
1043 * @unsigned ac@ = argument count
1044 * @char *av[]@ = pointer to the argument list
1045 *
1046 * Returns: ---
1047 *
1048 * Use: Pings a peer.
1049 */
1050
1051static void a_ping(admin *a, unsigned ac, char *av[],
1052 const char *cmd, unsigned msg)
1053{
1054 long t = T_PING;
1055 int i;
1056 peer *p;
de014da6 1057 admin_pingop *pg = 0;
1058 const char *tag = 0;
0ba8de86 1059
1060 i = 0;
1061 for (;;) {
1062 if (!av[i])
1063 goto bad_syntax;
de014da6 1064 if (mystrieq(av[i], "-background")) {
1065 if (!av[++i]) goto bad_syntax;
1066 tag = av[i];
1067 } else if (mystrieq(av[i], "-timeout")) {
0ba8de86 1068 if (!av[++i]) goto bad_syntax;
1069 if ((t = a_parsetime(av[i])) < 0) {
f43df819 1070 a_fail(a, "bad-time-spec", "%s", av[i], A_END);
0ba8de86 1071 return;
1072 }
1073 } else if (mystrieq(av[i], "--")) {
1074 i++;
1075 break;
1076 } else
1077 break;
1078 i++;
1079 }
1080
1081 if (!av[i]) goto bad_syntax;
37941236 1082 if ((p = a_findpeer(a, av[i])) == 0)
0ba8de86 1083 return;
de014da6 1084 pg = xmalloc(sizeof(*pg));
1085 gettimeofday(&pg->pingtime, 0);
1086 a_bgadd(a, &pg->bg, tag, a_pingcancel);
060ca767 1087 T( trace(T_ADMIN, "admin: ping op %s: %s to %s",
1088 BGTAG(pg), cmd, p_name(p)); )
de014da6 1089 if (p_pingsend(p, &pg->ping, msg, t, a_pong, pg)) {
f43df819 1090 a_bgfail(&pg->bg, "ping-send-failed", A_END);
de014da6 1091 a_bgrelease(&pg->bg);
1092 }
0ba8de86 1093 return;
1094
1095bad_syntax:
f43df819 1096 a_fail(a, "bad-syntax", "%s", cmd, "[OPTIONS] PEER", cmd, A_END);
42da2a58 1097 return;
410c8acf 1098}
1099
0ba8de86 1100static void acmd_ping(admin *a, unsigned ac, char *av[])
1101 { a_ping(a, ac, av, "ping", MISC_PING); }
1102static void acmd_eping(admin *a, unsigned ac, char *av[])
1103 { a_ping(a, ac, av, "eping", MISC_EPING); }
0ba8de86 1104
410c8acf 1105/*----- Administration commands -------------------------------------------*/
1106
1107/* --- Miscellaneous commands --- */
1108
3cdc3f3a 1109/* --- @traceish@ --- *
1110 *
1111 * Arguments: @admin *a@ = connection to complain on
1112 * @unsigned ac@ = number of arguments
1113 * @char *av[]@ = vector of arguments
1114 * @const char *what@ = what we're messing with
1115 * @const trace_opt *tt@ = options table
1116 * @unsigned *ff@ = where the flags are
1117 *
1118 * Returns: Nonzero if anything changed.
1119 *
1120 * Use: Guts of trace-ish commands like `trace' and `watch'.
1121 */
410c8acf 1122
3cdc3f3a 1123static int traceish(admin *a, unsigned ac, char *av[],
1124 const char *what, const trace_opt *tt, unsigned *ff)
410c8acf 1125{
3cdc3f3a 1126 int ch = 0;
1127
410c8acf 1128 if (!ac || strcmp(av[0], "?") == 0) {
1129 const trace_opt *t;
3cdc3f3a 1130 for (t = tt; t->ch; t++) {
f43df819
MW
1131 a_info(a, "*%c%c %s",
1132 t->ch, (*ff & t->f) == t->f ? '+' : ' ', t->help, A_END);
410c8acf 1133 }
1134 } else {
1135 unsigned sense = 1;
3cdc3f3a 1136 unsigned f = *ff;
1137 const trace_opt *t;
410c8acf 1138 char *p = av[0];
1139
1140 while (*p) {
1141 switch (*p) {
1142 case '+': sense = 1; break;
1143 case '-': sense = 0; break;
1144 default:
3cdc3f3a 1145 for (t = tt; t->ch; t++) {
1146 if (t->ch == *p) {
1147 if (sense) f |= t->f;
1148 else f &= ~t->f;
410c8acf 1149 goto tropt_ok;
1150 }
1151 }
f43df819 1152 a_fail(a, "bad-%s-option", what, "%c", *p, A_END);
3cdc3f3a 1153 return (0);
410c8acf 1154 tropt_ok:;
1155 break;
1156 }
1157 p++;
1158 }
3cdc3f3a 1159 *ff = f;
1160 ch = 1;
410c8acf 1161 }
3cdc3f3a 1162 a_ok(a);
1163 return (ch);
1164}
1165
1166#ifndef NTRACE
1167
1168static void acmd_trace(admin *a, unsigned ac, char *av[])
1169{
1170 if (traceish(a, ac, av, "trace", tr_opts, &tr_flags))
1171 trace_level(tr_flags);
410c8acf 1172}
1173
1174#endif
1175
3cdc3f3a 1176static void acmd_watch(admin *a, unsigned ac, char *av[])
1177{
1178 traceish(a, ac, av, "watch", w_opts, &a->f);
1179}
1180
3cdc3f3a 1181static void alertcmd(admin *a, unsigned f_and, unsigned f_eq,
1182 const char *tag, unsigned ac, char *av[])
1183{
1184 dstr d = DSTR_INIT;
1185 unsigned i;
1186
1187 dstr_puts(&d, "USER");
1188 for (i = 0; i < ac; i++)
1189 quotify(&d, av[i]);
1190 dstr_putz(&d);
1191 a_rawalert(f_and, f_eq, tag, d.buf, d.len);
1192 dstr_destroy(&d);
1193 a_ok(a);
1194}
1195
1196static void acmd_notify(admin *a, unsigned ac, char *av[])
1197 { alertcmd(a, AF_NOTE, AF_NOTE, "NOTE", ac, av); }
1198static void acmd_warn(admin *a, unsigned ac, char *av[])
1199 { alertcmd(a, AF_WARN, AF_WARN, "WARN", ac, av); }
1200
410c8acf 1201static void acmd_port(admin *a, unsigned ac, char *av[])
1202{
f43df819 1203 a_info(a, "%u", p_port(), A_END);
3cdc3f3a 1204 a_ok(a);
410c8acf 1205}
1206
410c8acf 1207static void acmd_daemon(admin *a, unsigned ac, char *av[])
1208{
1209 if (flags & F_DAEMON)
f43df819 1210 a_fail(a, "already-daemon", A_END);
410c8acf 1211 else {
f43df819 1212 a_notify("DAEMON", A_END);
93bee2c4 1213 if (a_stdin)
410c8acf 1214 a_destroy(a_stdin);
410c8acf 1215 if (u_daemon())
f43df819 1216 a_fail(a, "daemon-error", "?ERRNO", A_END);
93bee2c4 1217 else {
410c8acf 1218 flags |= F_DAEMON;
3cdc3f3a 1219 a_ok(a);
93bee2c4 1220 }
410c8acf 1221 }
1222}
1223
1224static void acmd_list(admin *a, unsigned ac, char *av[])
1225{
1226 peer *p;
1227 for (p = p_first(); p; p = p_next(p))
f43df819 1228 a_info(a, "%s", p_name(p), A_END);
3cdc3f3a 1229 a_ok(a);
410c8acf 1230}
1231
1232static void acmd_ifname(admin *a, unsigned ac, char *av[])
1233{
1234 peer *p;
1235
37941236 1236 if ((p = a_findpeer(a, av[0])) != 0) {
f43df819 1237 a_info(a, "%s", p_ifname(p), A_END);
3cdc3f3a 1238 a_ok(a);
1239 }
410c8acf 1240}
1241
64cf2223
MW
1242static void acmd_setifname(admin *a, unsigned ac, char *av[])
1243{
1244 peer *p;
1245
1246 if ((p = a_findpeer(a, av[0])) != 0) {
1247 a_notify("NEWIFNAME", "?PEER", p, "%s", p_ifname(p), "%s", av[1], A_END);
1248 p_setifname(p, av[1]);
1249 a_ok(a);
1250 }
1251}
1252
37941236 1253static void acmd_getchal(admin *a, unsigned ac, char *av[])
1254{
1255 buf b;
1256
1257 buf_init(&b, buf_i, PKBUFSZ);
1258 c_new(&b);
f43df819 1259 a_info(a, "?B64", BBASE(&b), (size_t)BLEN(&b), A_END);
37941236 1260 a_ok(a);
1261}
1262
1263static void acmd_checkchal(admin *a, unsigned ac, char *av[])
1264{
1265 base64_ctx b64;
1266 buf b;
1267 dstr d = DSTR_INIT;
1268
1269 base64_init(&b64);
1270 base64_decode(&b64, av[0], strlen(av[0]), &d);
1271 base64_decode(&b64, 0, 0, &d);
1272 buf_init(&b, d.buf, d.len);
1273 if (c_check(&b) || BBAD(&b) || BLEFT(&b))
f43df819 1274 a_fail(a, "invalid-challenge", A_END);
37941236 1275 else
1276 a_ok(a);
1277 dstr_destroy(&d);
1278}
1279
1280static void acmd_greet(admin *a, unsigned ac, char *av[])
1281{
1282 peer *p;
1283 base64_ctx b64;
1284 dstr d = DSTR_INIT;
1285
1286 if ((p = a_findpeer(a, av[0])) != 0) {
1287 base64_init(&b64);
1288 base64_decode(&b64, av[1], strlen(av[1]), &d);
1289 base64_decode(&b64, 0, 0, &d);
1290 p_greet(p, d.buf, d.len);
1291 dstr_destroy(&d);
1292 a_ok(a);
1293 }
1294}
1295
410c8acf 1296static void acmd_addr(admin *a, unsigned ac, char *av[])
1297{
1298 peer *p;
1299 const addr *ad;
1300
37941236 1301 if ((p = a_findpeer(a, av[0])) != 0) {
410c8acf 1302 ad = p_addr(p);
1303 assert(ad->sa.sa_family == AF_INET);
f43df819 1304 a_info(a, "?ADDR", ad, A_END);
3cdc3f3a 1305 a_ok(a);
410c8acf 1306 }
1307}
1308
060ca767 1309static void acmd_peerinfo(admin *a, unsigned ac, char *av[])
1310{
1311 peer *p;
1312 const peerspec *ps;
1313
37941236 1314 if ((p = a_findpeer(a, av[0])) != 0) {
1315 ps = p_spec(p);
f43df819
MW
1316 a_info(a, "tunnel=%s", ps->tops->name, A_END);
1317 a_info(a, "keepalive=%lu", ps->t_ka, A_END);
37941236 1318 a_ok(a);
060ca767 1319 }
060ca767 1320}
1321
1322static void acmd_servinfo(admin *a, unsigned ac, char *av[])
1323{
f43df819
MW
1324 a_info(a, "implementation=edgeware-tripe", A_END);
1325 a_info(a, "version=%s", VERSION, A_END);
1326 a_info(a, "daemon=%s", BOOL(flags & F_DAEMON), A_END);
060ca767 1327 a_ok(a);
1328}
1329
c7053da9 1330static void acmd_stats(admin *a, unsigned ac, char *av[])
1331{
1332 peer *p;
1333 stats *st;
1334
37941236 1335 if ((p = a_findpeer(a, av[0])) == 0)
060ca767 1336 return;
060ca767 1337
1338 st = p_stats(p);
f43df819
MW
1339 a_info(a, "start-time=%s", timestr(st->t_start), A_END);
1340 a_info(a, "last-packet-time=%s", timestr(st->t_last), A_END);
1341 a_info(a, "last-keyexch-time=%s", timestr(st->t_kx), A_END);
1342 a_info(a, "packets-in=%lu bytes-in=%lu", st->n_in, st->sz_in, A_END);
060ca767 1343 a_info(a, "packets-out=%lu bytes-out=%lu",
f43df819 1344 st->n_out, st->sz_out, A_END);
060ca767 1345 a_info(a, "keyexch-packets-in=%lu keyexch-bytes-in=%lu",
f43df819 1346 st->n_kxin, st->sz_kxin, A_END);
060ca767 1347 a_info(a, "keyexch-packets-out=%lu keyexch-bytes-out=%lu",
f43df819 1348 st->n_kxout, st->sz_kxout, A_END);
060ca767 1349 a_info(a, "ip-packets-in=%lu ip-bytes-in=%lu",
f43df819 1350 st->n_ipin, st->sz_ipin, A_END);
060ca767 1351 a_info(a, "ip-packets-out=%lu ip-bytes-out=%lu",
f43df819
MW
1352 st->n_ipout, st->sz_ipout, A_END);
1353 a_info(a, "rejected-packets=%lu", st->n_reject, A_END);
060ca767 1354 a_ok(a);
c7053da9 1355}
1356
410c8acf 1357static void acmd_kill(admin *a, unsigned ac, char *av[])
1358{
1359 peer *p;
37941236 1360 if ((p = a_findpeer(a, av[0])) != 0) {
410c8acf 1361 p_destroy(p);
3cdc3f3a 1362 a_ok(a);
410c8acf 1363 }
1364}
1365
0ba8de86 1366static void acmd_forcekx(admin *a, unsigned ac, char *av[])
1367{
1368 peer *p;
37941236 1369 if ((p = a_findpeer(a, av[0])) != 0) {
de014da6 1370 kx_start(&p->kx, 1);
0ba8de86 1371 a_ok(a);
1372 }
1373}
1374
de014da6 1375static void acmd_reload(admin *a, unsigned ac, char *av[])
1376 { p_keyreload(); a_ok(a); }
1377
410c8acf 1378static void acmd_quit(admin *a, unsigned ac, char *av[])
1379{
f43df819 1380 a_warn("SERVER", "quit", "admin-request", A_END);
3cdc3f3a 1381 a_ok(a);
410c8acf 1382 a_quit();
1383}
1384
3cdc3f3a 1385static void acmd_version(admin *a, unsigned ac, char *av[])
1386{
f43df819 1387 a_info(a, "%s", PACKAGE, "%s", VERSION, A_END);
3cdc3f3a 1388 a_ok(a);
1389}
1390
42da2a58 1391static void acmd_tunnels(admin *a, unsigned ac, char *av[])
1392{
1393 int i;
1394 for (i = 0; tunnels[i]; i++)
f43df819 1395 a_info(a, "%s", tunnels[i]->name, A_END);
42da2a58 1396 a_ok(a);
1397}
1398
410c8acf 1399/* --- The command table and help --- */
1400
1401typedef struct acmd {
1402 const char *name;
1403 const char *help;
1404 unsigned argmin, argmax;
1405 void (*func)(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1406} acmd;
1407
1408static void acmd_help(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1409
1410static const acmd acmdtab[] = {
9986f0b5 1411 { "add", "[OPTIONS] PEER ADDR ...", 2, 0xffff, acmd_add },
f43df819
MW
1412 { "addr", "PEER", 1, 1, acmd_addr },
1413 { "checkchal", "CHAL", 1, 1, acmd_checkchal },
1414 { "daemon", 0, 0, 0, acmd_daemon },
1415 { "eping", "[OPTIONS] PEER", 1, 0xffff, acmd_eping },
1416 { "forcekx", "PEER", 1, 1, acmd_forcekx },
1417 { "getchal", 0, 0, 0, acmd_getchal },
1418 { "greet", "PEER CHAL", 2, 2, acmd_greet },
1419 { "help", 0, 0, 0, acmd_help },
1420 { "ifname", "PEER", 1, 1, acmd_ifname },
1421 { "kill", "PEER", 1, 1, acmd_kill },
1422 { "list", 0, 0, 0, acmd_list },
1423 { "notify", "MESSAGE ...", 1, 0xffff, acmd_notify },
1424 { "peerinfo", "PEER", 1, 1, acmd_peerinfo },
1425 { "ping", "[OPTIONS] PEER", 1, 0xffff, acmd_ping },
1426 { "port", 0, 0, 0, acmd_port },
1427 { "quit", 0, 0, 0, acmd_quit },
1428 { "reload", 0, 0, 0, acmd_reload },
1429 { "servinfo", 0, 0, 0, acmd_servinfo },
64cf2223 1430 { "setifname", "PEER NEW-NAME", 2, 2, acmd_setifname },
f43df819 1431 { "stats", "PEER", 1, 1, acmd_stats },
de014da6 1432#ifndef NTRACE
f43df819 1433 { "trace", "[OPTIONS]", 0, 1, acmd_trace },
de014da6 1434#endif
f43df819
MW
1435 { "tunnels", 0, 0, 0, acmd_tunnels },
1436 { "version", 0, 0, 0, acmd_version },
1437 { "warn", "MESSAGE ...", 1, 0xffff, acmd_warn },
1438 { "watch", "[OPTIONS]", 0, 1, acmd_watch },
410c8acf 1439 { 0, 0, 0, 0, 0 }
1440};
1441
1442static void acmd_help(admin *a, unsigned ac, char *av[])
1443{
1444 const acmd *c;
f43df819
MW
1445 for (c = acmdtab; c->name; c++) {
1446 if (c->help)
1447 a_info(a, "%s", c->name, "*%s", c->help, A_END);
1448 else
1449 a_info(a, "%s", c->name, A_END);
1450 }
3cdc3f3a 1451 a_ok(a);
410c8acf 1452}
1453
1454/*----- Connection handling -----------------------------------------------*/
1455
c511e1f9 1456/* --- @a_destroypending@ --- *
410c8acf 1457 *
c511e1f9 1458 * Arguments: ---
fd3cf232 1459 *
1460 * Returns: ---
1461 *
c511e1f9 1462 * Use: Destroys pending admin connections at a safe time.
fd3cf232 1463 */
1464
c511e1f9 1465static void a_destroypending(void)
410c8acf 1466{
c511e1f9 1467 admin *a, *aa;
de014da6 1468 admin_bgop *bg, *bbg;
de014da6 1469
c511e1f9 1470 /* --- Destroy connections marked as pending --- */
fd3cf232 1471
c511e1f9
MW
1472 for (a = a_dead; a; a = aa) {
1473 aa = a->next;
1474 assert(a->f & AF_DEAD);
1475
1476 /* --- Report what we're doing --- */
1477
1478 T( trace(T_ADMIN, "admin: completing destruction of connection %u",
1479 a->seq); )
1480
1481 /* --- Abort any background jobs in progress --- */
1482
1483 for (bg = a->bg; bg; bg = bbg) {
1484 bbg = bg->next;
1485 bg->cancel(bg);
1486 if (bg->tag) xfree(bg->tag);
1487 xfree(bg);
1488 }
1489
1490 /* --- Close file descriptors and selectory --- */
1491
1492 selbuf_destroy(&a->b);
1493 if (a->b.reader.fd != a->w.fd) close(a->b.reader.fd);
1494 close(a->w.fd);
1495 if (a_stdin == a) a_stdin = 0;
1496
1497 /* --- Done --- */
1498
1499 DESTROY(a);
410c8acf 1500 }
410c8acf 1501
c511e1f9 1502 /* --- All pending destruction completed --- */
060ca767 1503
c511e1f9 1504 a_dead = 0;
060ca767 1505}
1506
fd3cf232 1507/* --- @a_destroy@ --- *
1508 *
1509 * Arguments: @admin *a@ = pointer to an admin block
1510 *
1511 * Returns: ---
1512 *
1513 * Use: Destroys an admin block. This requires a certain amount of
1514 * care.
1515 */
1516
de014da6 1517static void freequeue(oqueue *q)
1518{
1519 obuf *o, *oo;
1520
1521 for (o = q->hd; o; o = oo) {
1522 oo = o->next;
1523 xfree(o);
1524 }
1525 q->hd = q->tl = 0;
1526}
1527
fd3cf232 1528static void a_destroy(admin *a)
1529{
fd3cf232 1530 if (a->f & AF_DEAD)
1531 return;
1532
c511e1f9
MW
1533 if (a->next) a->next->prev = a->prev;
1534 if (a->prev) a->prev->next = a->next;
1535 else admins = a->next;
fd3cf232 1536
c511e1f9 1537 if (a->out.hd) sel_rmfile(&a->w);
de014da6 1538 freequeue(&a->out);
fd3cf232 1539
c511e1f9
MW
1540 a->f |= AF_DEAD;
1541 a->next = a_dead;
1542 a_dead = a;
fd3cf232 1543
c511e1f9 1544 T( trace(T_ADMIN, "admin: killing connection %u", a->seq); )
fd3cf232 1545}
1546
410c8acf 1547/* --- @a_line@ --- *
1548 *
1549 * Arguments: @char *p@ = pointer to the line read
3008eabf 1550 * @size_t len@ = length of the line
410c8acf 1551 * @void *vp@ = pointer to my admin block
1552 *
1553 * Returns: ---
1554 *
1555 * Use: Handles a line of input.
1556 */
1557
3008eabf 1558static void a_line(char *p, size_t len, void *vp)
410c8acf 1559{
1560 admin *a = vp;
1561 const acmd *c;
3cdc3f3a 1562 char *av[16];
410c8acf 1563 size_t ac;
1564
8d0c7a83 1565 TIMER;
fd3cf232 1566 if (a->f & AF_DEAD)
1567 return;
410c8acf 1568 if (!p) {
de014da6 1569 if (!a->bg)
1570 a_destroy(a);
1571 else {
1572 a->f |= AF_CLOSE;
1573 selbuf_disable(&a->b);
1574 }
410c8acf 1575 return;
1576 }
3cdc3f3a 1577 ac = str_qsplit(p, av, 16, 0, STRF_QUOTE);
410c8acf 1578 if (!ac)
1579 return;
410c8acf 1580 for (c = acmdtab; c->name; c++) {
3cdc3f3a 1581 if (mystrieq(av[0], c->name)) {
410c8acf 1582 ac--;
f43df819
MW
1583 if (c->argmin > ac || ac > c->argmax) {
1584 if (!c->help)
1585 a_fail(a, "bad-syntax", "%s", c->name, "", A_END);
1586 else
1587 a_fail(a, "bad-syntax", "%s", c->name, "%s", c->help, A_END);
c511e1f9 1588 } else
410c8acf 1589 c->func(a, ac, av + 1);
1590 return;
1591 }
1592 }
f43df819 1593 a_fail(a, "unknown-command", "%s", av[0], A_END);
410c8acf 1594}
1595
1596/* --- @a_create@ --- *
1597 *
1598 * Arguments: @int fd_in, fd_out@ = file descriptors to use
3cdc3f3a 1599 * @unsigned f@ = initial flags to set
410c8acf 1600 *
1601 * Returns: ---
1602 *
1603 * Use: Creates a new admin connection.
1604 */
1605
3cdc3f3a 1606void a_create(int fd_in, int fd_out, unsigned f)
410c8acf 1607{
1608 admin *a = CREATE(admin);
3cdc3f3a 1609
93bee2c4 1610 T( static unsigned seq = 0;
1611 a->seq = seq++; )
410c8acf 1612 T( trace(T_ADMIN, "admin: accepted connection %u", a->seq); )
de014da6 1613 a->bg = 0;
042d5c20 1614 a->ref = 0;
3cdc3f3a 1615 a->f = f;
de014da6 1616 if (fd_in == STDIN_FILENO) a_stdin = a;
fd3cf232 1617 fdflags(fd_in, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
410c8acf 1618 if (fd_out != fd_in)
fd3cf232 1619 fdflags(fd_out, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
410c8acf 1620 selbuf_init(&a->b, &sel, fd_in, a_line, a);
fd3cf232 1621 sel_initfile(&sel, &a->w, fd_out, SEL_WRITE, a_flush, a);
de014da6 1622 a->out.hd = a->out.tl = 0;
410c8acf 1623 a->next = admins;
1624 a->prev = 0;
de014da6 1625 if (admins) admins->prev = a;
410c8acf 1626 admins = a;
1627}
1628
1629/* --- @a_accept@ --- *
1630 *
1631 * Arguments: @int fd@ = file descriptor to accept
1632 * @unsigned mode@ = what to do
1633 * @void *v@ = uninteresting pointer
1634 *
1635 * Returns: ---
1636 *
1637 * Use: Accepts a new admin connection.
1638 */
1639
1640static void a_accept(int fd, unsigned mode, void *v)
1641{
1642 int nfd;
1643 struct sockaddr_un sun;
1644 size_t sz = sizeof(sun);
1645
1646 if ((nfd = accept(fd, (struct sockaddr *)&sun, &sz)) < 0) {
1ecee057 1647 if (errno != EINTR && errno != EAGAIN && errno != EWOULDBLOCK &&
1648 errno != ECONNABORTED && errno != EPROTO)
f43df819 1649 a_warn("ADMIN", "accept-error", "?ERRNO", A_END);
410c8acf 1650 return;
1651 }
3cdc3f3a 1652 a_create(nfd, nfd, 0);
410c8acf 1653}
1654
c511e1f9
MW
1655/* --- @a_preselect@ --- *
1656 *
1657 * Arguments: ---
1658 *
1659 * Returns: ---
1660 *
1661 * Use: Informs the admin module that we're about to select again,
1662 * and that it should do cleanup things it has delayed until a
1663 * `safe' time.
1664 */
1665
1666void a_preselect(void) { if (a_dead) a_destroypending(); }
1667
410c8acf 1668/* --- @a_daemon@ --- *
1669 *
1670 * Arguments: ---
1671 *
1672 * Returns: ---
1673 *
1674 * Use: Informs the admin module that it's a daemon.
1675 */
1676
1677void a_daemon(void)
1678{
1679 flags |= F_DAEMON;
1680}
1681
1682/* --- @a_init@ --- *
1683 *
1684 * Arguments: @const char *name@ = socket name to create
1685 *
1686 * Returns: ---
1687 *
1688 * Use: Creates the admin listening socket.
1689 */
1690
1691void a_init(const char *name)
1692{
1693 int fd;
1694 int n = 5;
1695 struct sockaddr_un sun;
1696 struct sigaction sa;
1697 size_t sz;
1698
1699 /* --- Set up the socket address --- */
1700
1701 sz = strlen(name) + 1;
1702 if (sz > sizeof(sun.sun_path))
1703 die(EXIT_FAILURE, "socket name `%s' too long", name);
1704 BURN(sun);
1705 sun.sun_family = AF_UNIX;
1706 memcpy(sun.sun_path, name, sz);
1707 sz += offsetof(struct sockaddr_un, sun_path);
1708
1709 /* --- Attempt to bind to the socket --- */
1710
1711 umask(0077);
1712again:
1713 if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0)
1714 die(EXIT_FAILURE, "couldn't create socket: %s", strerror(errno));
1715 if (bind(fd, (struct sockaddr *)&sun, sz) < 0) {
1716 struct stat st;
1717 int e = errno;
1718 if (errno != EADDRINUSE) {
1719 die(EXIT_FAILURE, "couldn't bind to address `%s': %s",
1720 sun.sun_path, strerror(e));
1721 }
1722 if (!n)
1723 die(EXIT_FAILURE, "too many retries; giving up");
1724 n--;
1725 if (!connect(fd, (struct sockaddr *)&sun, sz)) {
1726 die(EXIT_FAILURE, "server already listening on admin socket `%s'",
1727 sun.sun_path);
1728 }
1729 if (errno != ECONNREFUSED)
1730 die(EXIT_FAILURE, "couldn't bind to address: %s", strerror(e));
1731 if (stat(sun.sun_path, &st)) {
1732 die(EXIT_FAILURE, "couldn't stat `%s': %s",
1733 sun.sun_path, strerror(errno));
1734 }
1735 if (!S_ISSOCK(st.st_mode))
1736 die(EXIT_FAILURE, "object `%s' isn't a socket", sun.sun_path);
c7053da9 1737 T( trace(T_ADMIN, "admin: stale socket found; removing it"); )
410c8acf 1738 unlink(sun.sun_path);
1739 close(fd);
1740 goto again;
1741 }
1742 chmod(sun.sun_path, 0600);
1743 fdflags(fd, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
1744 if (listen(fd, 5))
1745 die(EXIT_FAILURE, "couldn't listen on socket: %s", strerror(errno));
1746
1747 /* --- Listen to the socket --- */
1748
1749 sel_initfile(&sel, &sock, fd, SEL_READ, a_accept, 0);
1750 sel_addfile(&sock);
1751 sockname = name;
1752 bres_init(&sel);
1753 T( trace_custom(a_trace, 0);
1754 trace(T_ADMIN, "admin: enabled custom tracing"); )
1755 flags |= F_INIT;
1756
1757 /* --- Set up signal handlers --- */
1758
1759 sig_add(&s_term, SIGTERM, a_sigdie, 0);
1760 sig_add(&s_hup, SIGHUP, a_sighup, 0);
3cdc3f3a 1761 signal(SIGPIPE, SIG_IGN);
410c8acf 1762 sigaction(SIGINT, 0, &sa);
1763 if (sa.sa_handler != SIG_IGN)
1764 sig_add(&s_int, SIGINT, a_sigdie, 0);
1765}
1766
1767/*----- That's all, folks -------------------------------------------------*/