admin: Option parser macros.
[tripe] / server / admin.c
1 /* -*-c-*-
2 *
3 * $Id$
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
29 /*----- Header files ------------------------------------------------------*/
30
31 #include "tripe.h"
32
33 /*----- Global variables --------------------------------------------------*/
34
35 #ifndef NTRACE
36
37 const trace_opt tr_opts[] = {
38 { 't', T_TUNNEL, "tunnel events" },
39 { 'r', T_PEER, "peer events" },
40 { 'a', T_ADMIN, "admin interface" },
41 { 's', T_KEYSET, "symmetric keyset management" },
42 { 'x', T_KEYEXCH, "key exchange" },
43 { 'm', T_KEYMGMT, "key management" },
44 { 'l', T_CHAL, "challenge management" },
45 { 'p', T_PACKET, "packet contents" },
46 { 'c', T_CRYPTO, "crypto details" },
47 { 'A', T_ALL, "all of the above" },
48 { 0, 0, 0 }
49 };
50
51 unsigned tr_flags = 0;
52 #endif
53
54 static 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
62 /*----- Static variables --------------------------------------------------*/
63
64 static admin *admins;
65 static admin *a_dead;
66 static sel_file sock;
67 static const char *sockname;
68 static unsigned flags = 0;
69 static admin *a_stdin = 0;
70 static 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)
76 #define T_PING SEC(5)
77
78 static void a_destroy(admin */*a*/);
79
80 #define BOOL(x) ((x) ? "t" : "nil")
81
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
95 static ssize_t trywrite(admin *a, const char *p, size_t sz)
96 {
97 ssize_t n, done = 0;
98
99 again:
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);
114 a_warn("ADMIN", "client-write-error", "?ERRNO", A_END);
115 return (-1);
116 }
117 }
118 return (done);
119 }
120
121 /* --- @doqueue@ -- *
122 *
123 * Arguments: @oqueue *q@ = pointer to output queue
124 * @const char *p@ = pointer to buffer to write
125 * @size_t sz@ = size of buffer
126 *
127 * Returns: Nonzero if the queue was previously empty.
128 *
129 * Use: Queues data to be written later.
130 */
131
132 static int doqueue(oqueue *q, const char *p, size_t sz)
133 {
134 obuf *o;
135 int rc = 0;
136 size_t n;
137
138 o = q->tl;
139 if (!o)
140 rc = 1;
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;
148 if (q->tl)
149 q->tl->next = o;
150 else
151 q->hd = o;
152 q->tl = o;
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);
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
178 static 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);
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
207 static void a_flush(int fd, unsigned mode, void *v)
208 {
209 admin *a = v;
210 obuf *o, *oo;
211 ssize_t n;
212
213 o = a->out.hd;
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 }
224 a->out.hd = o;
225 if (!o) {
226 a->out.tl = 0;
227 sel_rmfile(&a->w);
228 }
229 }
230
231 /*----- Utility functions -------------------------------------------------*/
232
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
244 static 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
272 static 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
322 /* --- @a_write@, @a_vwrite@ --- *
323 *
324 * Arguments: @admin *a@ = admin connection to write to
325 * @const char *status@ = status code to report
326 * @const char *tag@ = tag string, or null
327 * @const char *fmt@ = pointer to format string
328 * @va_list ap@ = arguments in list
329 * @...@ = other arguments
330 *
331 * Returns: ---
332 *
333 * Use: Sends a message to an admin connection.
334 */
335
336 static void a_vwrite(admin *a, const char *status, const char *tag,
337 const char *fmt, va_list ap)
338 {
339 dstr d = DSTR_INIT;
340 if (tag) dstr_puts(&d, "BG");
341 dstr_puts(&d, status);
342 if (tag) quotify(&d, tag);
343 a_vformat(&d, fmt, ap);
344 dstr_putc(&d, '\n');
345 dosend(a, d.buf, d.len);
346 dstr_destroy(&d);
347 }
348
349 static void a_write(admin *a, const char *status, const char *tag,
350 const char *fmt, ...)
351 {
352 va_list ap;
353 va_start(ap, fmt);
354 a_vwrite(a, status, tag, fmt, ap);
355 va_end(ap);
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
369 static void a_ok(admin *a) { a_write(a, "OK", 0, A_END); }
370
371 static void a_info(admin *a, const char *fmt, ...)
372 {
373 va_list ap;
374 va_start(ap, fmt);
375 a_vwrite(a, "INFO", 0, fmt, ap);
376 va_end(ap);
377 }
378
379 static void a_fail(admin *a, const char *fmt, ...)
380 {
381 va_list ap;
382 va_start(ap, fmt);
383 a_vwrite(a, "FAIL", 0, fmt, ap);
384 va_end(ap);
385 }
386
387 /* --- @a_alert@, @a_valert@, @a_rawalert@ --- *
388 *
389 * Arguments: @unsigned f_and, f_eq@ = filter for connections
390 * @const char *status@ = status string
391 * @const char *fmt@ = pointer to format string
392 * @const char *p@ = pointer to raw string
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
403 static void a_rawalert(unsigned f_and, unsigned f_eq, const char *status,
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;
411 dstr_puts(&d, status);
412 if (p) {
413 dstr_putc(&d, ' ');
414 dstr_putm(&d, p, sz);
415 }
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 }
424 dstr_destroy(&d);
425 }
426
427 static 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;
434 a_vformat(&d, fmt, ap);
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*/
440 static 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
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
460 void a_warn(const char *fmt, ...)
461 {
462 va_list ap;
463
464 va_start(ap, fmt);
465 if (flags & F_INIT)
466 a_valert(0, 0, "WARN", fmt, ap);
467 else {
468 dstr d = DSTR_INIT;
469 fprintf(stderr, "%s: ", QUIS);
470 a_vformat(&d, fmt, ap);
471 dstr_putc(&d, '\n');
472 dstr_write(&d, stderr);
473 dstr_destroy(&d);
474 }
475 va_end(ap);
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 *
486 * Use: Custom trace output handler. Sends trace messages to
487 * interested admin connections.
488 */
489
490 #ifndef NTRACE
491 static void a_trace(const char *p, size_t sz, void *v)
492 { a_rawalert(AF_TRACE, AF_TRACE, "TRACE", p, sz); }
493 #endif
494
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
505 void 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
514 /* --- @a_quit@ --- *
515 *
516 * Arguments: ---
517 *
518 * Returns: ---
519 *
520 * Use: Shuts things down nicely.
521 */
522
523 void a_quit(void)
524 {
525 peer *p;
526
527 close(sock.fd);
528 unlink(sockname);
529 while ((p = p_first()) != 0)
530 p_destroy(p);
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
544 static 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:
553 sprintf(buf, "%i", sig);
554 p = buf;
555 break;
556 }
557 a_warn("SERVER", "quit", "signal", "%s", p, A_END);
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
571 static void a_sighup(int sig, void *v)
572 {
573 a_warn("SERVER", "ignore", "signal", "SIGHUP", A_END);
574 }
575
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
583 static 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
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
609 static peer *a_findpeer(admin *a, const char *pn)
610 {
611 peer *p;
612
613 if ((p = p_find(pn)) == 0)
614 a_fail(a, "unknown-peer", "%s", pn, A_END);
615 return (p);
616 }
617
618 /*----- Backgrounded operations -------------------------------------------*/
619
620 #define BGTAG(bg) \
621 (((admin_bgop *)(bg))->tag ? ((admin_bgop *)(bg))->tag : "<foreground>")
622
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
633 static void a_bgrelease(admin_bgop *bg)
634 {
635 admin *a = bg->a;
636
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;
643 xfree(bg);
644 if (!a->bg && (a->f & AF_CLOSE)) a_destroy(a);
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
658 static void a_bgok(admin_bgop *bg)
659 { a_write(bg->a, "OK", bg->tag, A_END); }
660
661 static 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
669 static 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
689 static 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;
702 if (a->bg) a->bg->prev = bg;
703 a->bg = bg;
704 T( trace(T_ADMIN, "admin: add bgop %s", BGTAG(bg)); )
705 if (tag) a_write(a, "DETACH", tag, A_END);
706 }
707
708 /*----- Name resolution operations ----------------------------------------*/
709
710 /* --- @a_resolved@ --- *
711 *
712 * Arguments: @struct hostent *h@ = pointer to resolved hostname
713 * @void *v@ = pointer to resolver operation
714 *
715 * Returns: ---
716 *
717 * Use: Handles a completed name resolution.
718 */
719
720 static void a_resolved(struct hostent *h, void *v)
721 {
722 admin_resop *r = v;
723
724 T( trace(T_ADMIN, "admin: resop %s resolved", BGTAG(r)); )
725 TIMER;
726 if (!h) {
727 a_bgfail(&r->bg, "resolve-error", "%s", r->addr, A_END);
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 }
737
738 /* --- @a_restimer@ --- *
739 *
740 * Arguments: @struct timeval *tv@ = timer
741 * @void *v@ = pointer to resolver operation
742 *
743 * Returns: ---
744 *
745 * Use: Times out a resolver.
746 */
747
748 static void a_restimer(struct timeval *tv, void *v)
749 {
750 admin_resop *r = v;
751
752 T( trace(T_ADMIN, "admin: resop %s timeout", BGTAG(r)); )
753 a_bgfail(&r->bg, "resolver-timeout", "%s", r->addr, A_END);
754 r->func(r, ARES_FAIL);
755 bres_abort(&r->r);
756 xfree(r->addr);
757 a_bgrelease(&r->bg);
758 }
759
760 /* --- @a_rescancel@ --- *
761 *
762 * Arguments: @admin_bgop *bg@ = background operation
763 *
764 * Returns: ---
765 *
766 * Use: Cancels an add operation.
767 */
768
769 static void a_rescancel(admin_bgop *bg)
770 {
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);
778 }
779
780 /* --- @a_resolve@ --- *
781 *
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
788 *
789 * Returns: ---
790 *
791 * Use: Cranks up a resolver job.
792 */
793
794 static void a_resolve(admin *a, admin_resop *r, const char *tag,
795 void (*func)(struct admin_resop *, int),
796 unsigned ac, char *av[])
797 {
798 struct timeval tv;
799 unsigned long pt;
800 char *p;
801 int i = 0;
802
803 /* --- Fill in the easy bits of address --- */
804
805 r->bg.tag = "<starting>";
806 r->addr = 0;
807 r->func = func;
808 if (mystrieq(av[i], "inet")) i++;
809 if (ac - i != 2) {
810 a_fail(a, "bad-addr-syntax", "[inet] ADDRESS PORT", A_END);
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) {
820 a_fail(a, "unknown-service", "%s", av[i + 1], A_END);
821 goto fail;
822 }
823 pt = ntohs(s->s_port);
824 }
825 if (pt == 0 || pt >= 65536) {
826 a_fail(a, "invalid-port", "%lu", pt, A_END);
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
859 fail:
860 func(r, ARES_FAIL);
861 if (r->addr) xfree(r->addr);
862 xfree(r);
863 }
864
865 /*----- Option parsing ----------------------------------------------------*/
866
867 #define OPTIONS(argc, argv, guts) do { \
868 char **o_av = argv; \
869 for (;; o_av++) { \
870 if (!*o_av) \
871 break; \
872 if (mystrieq(*o_av, "--")) { \
873 o_av++; \
874 break; \
875 } \
876 guts \
877 if (**o_av == '-') \
878 goto bad_syntax; \
879 break; \
880 } \
881 argc -= o_av - argv; \
882 argv = o_av; \
883 } while (0)
884
885 #define OPT(name, guts) if (mystrieq(*o_av, name)) { guts continue; }
886
887 #define OPTARG(name, arg, guts) OPT(name, { \
888 const char *arg; \
889 arg = *++o_av; \
890 if (!arg) goto bad_syntax; \
891 guts \
892 })
893
894 #define OPTTIME(name, arg, guts) OPTARG(name, o_arg, { \
895 long arg; \
896 if ((arg = a_parsetime(o_arg)) < 0) { \
897 a_fail(a, "bad-time-spec", "%s", o_arg, A_END); \
898 goto fail; \
899 } \
900 guts \
901 })
902
903 /*----- Adding peers ------------------------------------------------------*/
904
905 /* --- @a_doadd@ --- *
906 *
907 * Arguments: @admin_resop *r@ = resolver operation
908 * @int rc@ = how it worked
909 *
910 * Returns: ---
911 *
912 * Use: Handles a completed resolution.
913 */
914
915 static void a_doadd(admin_resop *r, int rc)
916 {
917 admin_addop *add = (admin_addop *)r;
918
919 T( trace(T_ADMIN, "admin: done add op %s", BGTAG(add)); )
920
921 if (rc == ARES_OK) {
922 add->peer.sasz = add->r.sasz;
923 add->peer.sa = add->r.sa;
924 if (p_find(add->peer.name))
925 a_bgfail(&add->r.bg, "peer-exists", "%s", add->peer.name, A_END);
926 else if (!p_create(&add->peer))
927 a_bgfail(&add->r.bg, "peer-create-fail", "%s", add->peer.name, A_END);
928 else
929 a_bgok(&add->r.bg);
930 }
931
932 xfree(add->peer.name);
933 }
934
935 /* --- @acmd_add@ --- *
936 *
937 * Arguments: @admin *a@ = connection which requested the addition
938 * @unsigned ac@ = argument count
939 * @char *av[]@ = pointer to the argument list
940 *
941 * Returns: ---
942 *
943 * Use: Adds a new peer.
944 */
945
946 static void acmd_add(admin *a, unsigned ac, char *av[])
947 {
948 const char *tag = 0;
949 admin_addop *add;
950
951 /* --- Set stuff up --- */
952
953 add = xmalloc(sizeof(*add));
954 add->peer.name = 0;
955 add->peer.t_ka = 0;
956 add->peer.tops = tun_default;
957
958 /* --- Parse options --- */
959
960 OPTIONS(ac, av, {
961 OPTARG("-background", arg, { tag = arg; })
962 OPTARG("-tunnel", arg, {
963 unsigned i;
964 for (i = 0;; i++) {
965 if (!tunnels[i]) {
966 a_fail(a, "unknown-tunnel", "%s", arg, A_END);
967 goto fail;
968 }
969 if (mystrieq(arg, tunnels[i]->name)) {
970 add->peer.tops = tunnels[i];
971 break;
972 }
973 }
974 })
975 OPTTIME("-keepalive", t, { add->peer.t_ka = t; })
976 });
977
978 /* --- Make sure someone's not got there already --- */
979
980 if (!*av)
981 goto bad_syntax;
982 if (p_find(*av)) {
983 a_fail(a, "peer-exists", "%s", *av, A_END);
984 goto fail;
985 }
986 add->peer.name = xstrdup(*av++);
987 ac--;
988
989 /* --- Crank up the resolver --- */
990
991 a_resolve(a, &add->r, tag, a_doadd, ac, av);
992 return;
993
994 /* --- Clearing up --- */
995
996 bad_syntax:
997 a_fail(a, "bad-syntax", "add", "[OPTIONS] PEER ADDR ...", A_END);
998 fail:
999 if (add->peer.name) xfree(add->peer.name);
1000 xfree(add);
1001 return;
1002 }
1003
1004 /*----- Ping --------------------------------------------------------------*/
1005
1006 /* --- @a_pingcancel@ --- *
1007 *
1008 * Arguments: @admin_bgop *bg@ = background operation block
1009 *
1010 * Returns: ---
1011 *
1012 * Use: Cancels a running ping.
1013 */
1014
1015 static void a_pingcancel(admin_bgop *bg)
1016 {
1017 admin_pingop *pg = (admin_pingop *)bg;
1018 T( trace(T_ADMIN, "admin: cancel ping op %s", BGTAG(pg)); )
1019 p_pingdone(&pg->ping, PING_NONOTIFY);
1020 }
1021
1022 /* --- @a_pong@ --- *
1023 *
1024 * Arguments: @int rc@ = return code
1025 * @void *v@ = ping operation block
1026 *
1027 * Returns: ---
1028 *
1029 * Use: Collects what happened to a ping message.
1030 */
1031
1032 static void a_pong(int rc, void *v)
1033 {
1034 admin_pingop *pg = v;
1035 struct timeval tv;
1036 double millis;
1037
1038 switch (rc) {
1039 case PING_OK:
1040 gettimeofday(&tv, 0);
1041 tv_sub(&tv, &tv, &pg->pingtime);
1042 millis = (double)tv.tv_sec * 1000 + (double)tv.tv_usec/1000;
1043 a_bginfo(&pg->bg, "ping-ok", "%.1f", millis, A_END);
1044 a_bgok(&pg->bg);
1045 break;
1046 case PING_TIMEOUT:
1047 a_bginfo(&pg->bg, "ping-timeout", A_END);
1048 a_bgok(&pg->bg);
1049 break;
1050 case PING_PEERDIED:
1051 a_bginfo(&pg->bg, "ping-peer-died", A_END);
1052 a_bgok(&pg->bg);
1053 break;
1054 default:
1055 abort();
1056 }
1057 T( trace(T_ADMIN, "admin: ponged ping op %s", BGTAG(pg)); )
1058 a_bgrelease(&pg->bg);
1059 }
1060
1061 /* --- @acmd_ping@, @acmd_eping@ --- *
1062 *
1063 * Arguments: @admin *a@ = connection which requested the ping
1064 * @unsigned ac@ = argument count
1065 * @char *av[]@ = pointer to the argument list
1066 *
1067 * Returns: ---
1068 *
1069 * Use: Pings a peer.
1070 */
1071
1072 static void a_ping(admin *a, unsigned ac, char *av[],
1073 const char *cmd, unsigned msg)
1074 {
1075 long t = T_PING;
1076 peer *p;
1077 admin_pingop *pg = 0;
1078 const char *tag = 0;
1079
1080 OPTIONS(ac, av, {
1081 OPTARG("-background", arg, { tag = arg; })
1082 OPTTIME("-timeout", arg, { t = arg; })
1083 });
1084 if (!*av || av[1]) goto bad_syntax;
1085 if ((p = a_findpeer(a, *av)) == 0)
1086 return;
1087 pg = xmalloc(sizeof(*pg));
1088 gettimeofday(&pg->pingtime, 0);
1089 a_bgadd(a, &pg->bg, tag, a_pingcancel);
1090 T( trace(T_ADMIN, "admin: ping op %s: %s to %s",
1091 BGTAG(pg), cmd, p_name(p)); )
1092 if (p_pingsend(p, &pg->ping, msg, t, a_pong, pg)) {
1093 a_bgfail(&pg->bg, "ping-send-failed", A_END);
1094 a_bgrelease(&pg->bg);
1095 }
1096 return;
1097
1098 bad_syntax:
1099 a_fail(a, "bad-syntax", "%s", cmd, "[OPTIONS] PEER", cmd, A_END);
1100 fail:
1101 return;
1102 }
1103
1104 static void acmd_ping(admin *a, unsigned ac, char *av[])
1105 { a_ping(a, ac, av, "ping", MISC_PING); }
1106 static void acmd_eping(admin *a, unsigned ac, char *av[])
1107 { a_ping(a, ac, av, "eping", MISC_EPING); }
1108
1109 /*----- Administration commands -------------------------------------------*/
1110
1111 /* --- Miscellaneous commands --- */
1112
1113 /* --- @traceish@ --- *
1114 *
1115 * Arguments: @admin *a@ = connection to complain on
1116 * @unsigned ac@ = number of arguments
1117 * @char *av[]@ = vector of arguments
1118 * @const char *what@ = what we're messing with
1119 * @const trace_opt *tt@ = options table
1120 * @unsigned *ff@ = where the flags are
1121 *
1122 * Returns: Nonzero if anything changed.
1123 *
1124 * Use: Guts of trace-ish commands like `trace' and `watch'.
1125 */
1126
1127 static int traceish(admin *a, unsigned ac, char *av[],
1128 const char *what, const trace_opt *tt, unsigned *ff)
1129 {
1130 int ch = 0;
1131
1132 if (!ac || strcmp(av[0], "?") == 0) {
1133 const trace_opt *t;
1134 for (t = tt; t->ch; t++) {
1135 a_info(a, "*%c%c %s",
1136 t->ch, (*ff & t->f) == t->f ? '+' : ' ', t->help, A_END);
1137 }
1138 } else {
1139 unsigned sense = 1;
1140 unsigned f = *ff;
1141 const trace_opt *t;
1142 char *p = av[0];
1143
1144 while (*p) {
1145 switch (*p) {
1146 case '+': sense = 1; break;
1147 case '-': sense = 0; break;
1148 default:
1149 for (t = tt; t->ch; t++) {
1150 if (t->ch == *p) {
1151 if (sense) f |= t->f;
1152 else f &= ~t->f;
1153 goto tropt_ok;
1154 }
1155 }
1156 a_fail(a, "bad-%s-option", what, "%c", *p, A_END);
1157 return (0);
1158 tropt_ok:;
1159 break;
1160 }
1161 p++;
1162 }
1163 *ff = f;
1164 ch = 1;
1165 }
1166 a_ok(a);
1167 return (ch);
1168 }
1169
1170 #ifndef NTRACE
1171
1172 static void acmd_trace(admin *a, unsigned ac, char *av[])
1173 {
1174 if (traceish(a, ac, av, "trace", tr_opts, &tr_flags))
1175 trace_level(tr_flags);
1176 }
1177
1178 #endif
1179
1180 static void acmd_watch(admin *a, unsigned ac, char *av[])
1181 {
1182 traceish(a, ac, av, "watch", w_opts, &a->f);
1183 }
1184
1185 static void alertcmd(admin *a, unsigned f_and, unsigned f_eq,
1186 const char *tag, unsigned ac, char *av[])
1187 {
1188 dstr d = DSTR_INIT;
1189 unsigned i;
1190
1191 dstr_puts(&d, "USER");
1192 for (i = 0; i < ac; i++)
1193 quotify(&d, av[i]);
1194 dstr_putz(&d);
1195 a_rawalert(f_and, f_eq, tag, d.buf, d.len);
1196 dstr_destroy(&d);
1197 a_ok(a);
1198 }
1199
1200 static void acmd_notify(admin *a, unsigned ac, char *av[])
1201 { alertcmd(a, AF_NOTE, AF_NOTE, "NOTE", ac, av); }
1202 static void acmd_warn(admin *a, unsigned ac, char *av[])
1203 { alertcmd(a, AF_WARN, AF_WARN, "WARN", ac, av); }
1204
1205 static void acmd_port(admin *a, unsigned ac, char *av[])
1206 {
1207 a_info(a, "%u", p_port(), A_END);
1208 a_ok(a);
1209 }
1210
1211 static void acmd_daemon(admin *a, unsigned ac, char *av[])
1212 {
1213 if (flags & F_DAEMON)
1214 a_fail(a, "already-daemon", A_END);
1215 else {
1216 a_notify("DAEMON", A_END);
1217 if (a_stdin)
1218 a_destroy(a_stdin);
1219 if (u_daemon())
1220 a_fail(a, "daemon-error", "?ERRNO", A_END);
1221 else {
1222 flags |= F_DAEMON;
1223 a_ok(a);
1224 }
1225 }
1226 }
1227
1228 static void acmd_list(admin *a, unsigned ac, char *av[])
1229 {
1230 peer *p;
1231 for (p = p_first(); p; p = p_next(p))
1232 a_info(a, "%s", p_name(p), A_END);
1233 a_ok(a);
1234 }
1235
1236 static void acmd_ifname(admin *a, unsigned ac, char *av[])
1237 {
1238 peer *p;
1239
1240 if ((p = a_findpeer(a, av[0])) != 0) {
1241 a_info(a, "%s", p_ifname(p), A_END);
1242 a_ok(a);
1243 }
1244 }
1245
1246 static void acmd_setifname(admin *a, unsigned ac, char *av[])
1247 {
1248 peer *p;
1249
1250 if ((p = a_findpeer(a, av[0])) != 0) {
1251 a_notify("NEWIFNAME", "?PEER", p, "%s", p_ifname(p), "%s", av[1], A_END);
1252 p_setifname(p, av[1]);
1253 a_ok(a);
1254 }
1255 }
1256
1257 static void acmd_getchal(admin *a, unsigned ac, char *av[])
1258 {
1259 buf b;
1260
1261 buf_init(&b, buf_i, PKBUFSZ);
1262 c_new(&b);
1263 a_info(a, "?B64", BBASE(&b), (size_t)BLEN(&b), A_END);
1264 a_ok(a);
1265 }
1266
1267 static void acmd_checkchal(admin *a, unsigned ac, char *av[])
1268 {
1269 base64_ctx b64;
1270 buf b;
1271 dstr d = DSTR_INIT;
1272
1273 base64_init(&b64);
1274 base64_decode(&b64, av[0], strlen(av[0]), &d);
1275 base64_decode(&b64, 0, 0, &d);
1276 buf_init(&b, d.buf, d.len);
1277 if (c_check(&b) || BBAD(&b) || BLEFT(&b))
1278 a_fail(a, "invalid-challenge", A_END);
1279 else
1280 a_ok(a);
1281 dstr_destroy(&d);
1282 }
1283
1284 static void acmd_greet(admin *a, unsigned ac, char *av[])
1285 {
1286 peer *p;
1287 base64_ctx b64;
1288 dstr d = DSTR_INIT;
1289
1290 if ((p = a_findpeer(a, av[0])) != 0) {
1291 base64_init(&b64);
1292 base64_decode(&b64, av[1], strlen(av[1]), &d);
1293 base64_decode(&b64, 0, 0, &d);
1294 p_greet(p, d.buf, d.len);
1295 dstr_destroy(&d);
1296 a_ok(a);
1297 }
1298 }
1299
1300 static void acmd_addr(admin *a, unsigned ac, char *av[])
1301 {
1302 peer *p;
1303 const addr *ad;
1304
1305 if ((p = a_findpeer(a, av[0])) != 0) {
1306 ad = p_addr(p);
1307 assert(ad->sa.sa_family == AF_INET);
1308 a_info(a, "?ADDR", ad, A_END);
1309 a_ok(a);
1310 }
1311 }
1312
1313 static void acmd_peerinfo(admin *a, unsigned ac, char *av[])
1314 {
1315 peer *p;
1316 const peerspec *ps;
1317
1318 if ((p = a_findpeer(a, av[0])) != 0) {
1319 ps = p_spec(p);
1320 a_info(a, "tunnel=%s", ps->tops->name, A_END);
1321 a_info(a, "keepalive=%lu", ps->t_ka, A_END);
1322 a_ok(a);
1323 }
1324 }
1325
1326 static void acmd_servinfo(admin *a, unsigned ac, char *av[])
1327 {
1328 a_info(a, "implementation=edgeware-tripe", A_END);
1329 a_info(a, "version=%s", VERSION, A_END);
1330 a_info(a, "daemon=%s", BOOL(flags & F_DAEMON), A_END);
1331 a_ok(a);
1332 }
1333
1334 static void acmd_stats(admin *a, unsigned ac, char *av[])
1335 {
1336 peer *p;
1337 stats *st;
1338
1339 if ((p = a_findpeer(a, av[0])) == 0)
1340 return;
1341
1342 st = p_stats(p);
1343 a_info(a, "start-time=%s", timestr(st->t_start), A_END);
1344 a_info(a, "last-packet-time=%s", timestr(st->t_last), A_END);
1345 a_info(a, "last-keyexch-time=%s", timestr(st->t_kx), A_END);
1346 a_info(a, "packets-in=%lu bytes-in=%lu", st->n_in, st->sz_in, A_END);
1347 a_info(a, "packets-out=%lu bytes-out=%lu",
1348 st->n_out, st->sz_out, A_END);
1349 a_info(a, "keyexch-packets-in=%lu keyexch-bytes-in=%lu",
1350 st->n_kxin, st->sz_kxin, A_END);
1351 a_info(a, "keyexch-packets-out=%lu keyexch-bytes-out=%lu",
1352 st->n_kxout, st->sz_kxout, A_END);
1353 a_info(a, "ip-packets-in=%lu ip-bytes-in=%lu",
1354 st->n_ipin, st->sz_ipin, A_END);
1355 a_info(a, "ip-packets-out=%lu ip-bytes-out=%lu",
1356 st->n_ipout, st->sz_ipout, A_END);
1357 a_info(a, "rejected-packets=%lu", st->n_reject, A_END);
1358 a_ok(a);
1359 }
1360
1361 static void acmd_kill(admin *a, unsigned ac, char *av[])
1362 {
1363 peer *p;
1364 if ((p = a_findpeer(a, av[0])) != 0) {
1365 p_destroy(p);
1366 a_ok(a);
1367 }
1368 }
1369
1370 static void acmd_forcekx(admin *a, unsigned ac, char *av[])
1371 {
1372 peer *p;
1373 if ((p = a_findpeer(a, av[0])) != 0) {
1374 kx_start(&p->kx, 1);
1375 a_ok(a);
1376 }
1377 }
1378
1379 static void acmd_reload(admin *a, unsigned ac, char *av[])
1380 { p_keyreload(); a_ok(a); }
1381
1382 static void acmd_quit(admin *a, unsigned ac, char *av[])
1383 {
1384 a_warn("SERVER", "quit", "admin-request", A_END);
1385 a_ok(a);
1386 a_quit();
1387 }
1388
1389 static void acmd_version(admin *a, unsigned ac, char *av[])
1390 {
1391 a_info(a, "%s", PACKAGE, "%s", VERSION, A_END);
1392 a_ok(a);
1393 }
1394
1395 static void acmd_tunnels(admin *a, unsigned ac, char *av[])
1396 {
1397 int i;
1398 for (i = 0; tunnels[i]; i++)
1399 a_info(a, "%s", tunnels[i]->name, A_END);
1400 a_ok(a);
1401 }
1402
1403 /* --- The command table and help --- */
1404
1405 typedef struct acmd {
1406 const char *name;
1407 const char *help;
1408 unsigned argmin, argmax;
1409 void (*func)(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1410 } acmd;
1411
1412 static void acmd_help(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1413
1414 static const acmd acmdtab[] = {
1415 { "add", "[OPTIONS] PEER ADDR ...", 2, 0xffff, acmd_add },
1416 { "addr", "PEER", 1, 1, acmd_addr },
1417 { "checkchal", "CHAL", 1, 1, acmd_checkchal },
1418 { "daemon", 0, 0, 0, acmd_daemon },
1419 { "eping", "[OPTIONS] PEER", 1, 0xffff, acmd_eping },
1420 { "forcekx", "PEER", 1, 1, acmd_forcekx },
1421 { "getchal", 0, 0, 0, acmd_getchal },
1422 { "greet", "PEER CHAL", 2, 2, acmd_greet },
1423 { "help", 0, 0, 0, acmd_help },
1424 { "ifname", "PEER", 1, 1, acmd_ifname },
1425 { "kill", "PEER", 1, 1, acmd_kill },
1426 { "list", 0, 0, 0, acmd_list },
1427 { "notify", "MESSAGE ...", 1, 0xffff, acmd_notify },
1428 { "peerinfo", "PEER", 1, 1, acmd_peerinfo },
1429 { "ping", "[OPTIONS] PEER", 1, 0xffff, acmd_ping },
1430 { "port", 0, 0, 0, acmd_port },
1431 { "quit", 0, 0, 0, acmd_quit },
1432 { "reload", 0, 0, 0, acmd_reload },
1433 { "servinfo", 0, 0, 0, acmd_servinfo },
1434 { "setifname", "PEER NEW-NAME", 2, 2, acmd_setifname },
1435 { "stats", "PEER", 1, 1, acmd_stats },
1436 #ifndef NTRACE
1437 { "trace", "[OPTIONS]", 0, 1, acmd_trace },
1438 #endif
1439 { "tunnels", 0, 0, 0, acmd_tunnels },
1440 { "version", 0, 0, 0, acmd_version },
1441 { "warn", "MESSAGE ...", 1, 0xffff, acmd_warn },
1442 { "watch", "[OPTIONS]", 0, 1, acmd_watch },
1443 { 0, 0, 0, 0, 0 }
1444 };
1445
1446 static void acmd_help(admin *a, unsigned ac, char *av[])
1447 {
1448 const acmd *c;
1449 for (c = acmdtab; c->name; c++) {
1450 if (c->help)
1451 a_info(a, "%s", c->name, "*%s", c->help, A_END);
1452 else
1453 a_info(a, "%s", c->name, A_END);
1454 }
1455 a_ok(a);
1456 }
1457
1458 /*----- Connection handling -----------------------------------------------*/
1459
1460 /* --- @a_destroypending@ --- *
1461 *
1462 * Arguments: ---
1463 *
1464 * Returns: ---
1465 *
1466 * Use: Destroys pending admin connections at a safe time.
1467 */
1468
1469 static void a_destroypending(void)
1470 {
1471 admin *a, *aa;
1472 admin_bgop *bg, *bbg;
1473
1474 /* --- Destroy connections marked as pending --- */
1475
1476 for (a = a_dead; a; a = aa) {
1477 aa = a->next;
1478 assert(a->f & AF_DEAD);
1479
1480 /* --- Report what we're doing --- */
1481
1482 T( trace(T_ADMIN, "admin: completing destruction of connection %u",
1483 a->seq); )
1484
1485 /* --- Abort any background jobs in progress --- */
1486
1487 for (bg = a->bg; bg; bg = bbg) {
1488 bbg = bg->next;
1489 bg->cancel(bg);
1490 if (bg->tag) xfree(bg->tag);
1491 xfree(bg);
1492 }
1493
1494 /* --- Close file descriptors and selectory --- */
1495
1496 selbuf_destroy(&a->b);
1497 if (a->b.reader.fd != a->w.fd) close(a->b.reader.fd);
1498 close(a->w.fd);
1499 if (a_stdin == a) a_stdin = 0;
1500
1501 /* --- Done --- */
1502
1503 DESTROY(a);
1504 }
1505
1506 /* --- All pending destruction completed --- */
1507
1508 a_dead = 0;
1509 }
1510
1511 /* --- @a_destroy@ --- *
1512 *
1513 * Arguments: @admin *a@ = pointer to an admin block
1514 *
1515 * Returns: ---
1516 *
1517 * Use: Destroys an admin block. This requires a certain amount of
1518 * care.
1519 */
1520
1521 static void freequeue(oqueue *q)
1522 {
1523 obuf *o, *oo;
1524
1525 for (o = q->hd; o; o = oo) {
1526 oo = o->next;
1527 xfree(o);
1528 }
1529 q->hd = q->tl = 0;
1530 }
1531
1532 static void a_destroy(admin *a)
1533 {
1534 if (a->f & AF_DEAD)
1535 return;
1536
1537 if (a->next) a->next->prev = a->prev;
1538 if (a->prev) a->prev->next = a->next;
1539 else admins = a->next;
1540
1541 if (a->out.hd) sel_rmfile(&a->w);
1542 freequeue(&a->out);
1543
1544 a->f |= AF_DEAD;
1545 a->next = a_dead;
1546 a_dead = a;
1547
1548 T( trace(T_ADMIN, "admin: killing connection %u", a->seq); )
1549 }
1550
1551 /* --- @a_line@ --- *
1552 *
1553 * Arguments: @char *p@ = pointer to the line read
1554 * @size_t len@ = length of the line
1555 * @void *vp@ = pointer to my admin block
1556 *
1557 * Returns: ---
1558 *
1559 * Use: Handles a line of input.
1560 */
1561
1562 static void a_line(char *p, size_t len, void *vp)
1563 {
1564 admin *a = vp;
1565 const acmd *c;
1566 char *av[16];
1567 size_t ac;
1568
1569 TIMER;
1570 if (a->f & AF_DEAD)
1571 return;
1572 if (!p) {
1573 if (!a->bg)
1574 a_destroy(a);
1575 else {
1576 a->f |= AF_CLOSE;
1577 selbuf_disable(&a->b);
1578 }
1579 return;
1580 }
1581 ac = str_qsplit(p, av, 16, 0, STRF_QUOTE);
1582 if (!ac)
1583 return;
1584 for (c = acmdtab; c->name; c++) {
1585 if (mystrieq(av[0], c->name)) {
1586 ac--;
1587 if (c->argmin > ac || ac > c->argmax) {
1588 if (!c->help)
1589 a_fail(a, "bad-syntax", "%s", c->name, "", A_END);
1590 else
1591 a_fail(a, "bad-syntax", "%s", c->name, "%s", c->help, A_END);
1592 } else
1593 c->func(a, ac, av + 1);
1594 return;
1595 }
1596 }
1597 a_fail(a, "unknown-command", "%s", av[0], A_END);
1598 }
1599
1600 /* --- @a_create@ --- *
1601 *
1602 * Arguments: @int fd_in, fd_out@ = file descriptors to use
1603 * @unsigned f@ = initial flags to set
1604 *
1605 * Returns: ---
1606 *
1607 * Use: Creates a new admin connection.
1608 */
1609
1610 void a_create(int fd_in, int fd_out, unsigned f)
1611 {
1612 admin *a = CREATE(admin);
1613
1614 T( static unsigned seq = 0;
1615 a->seq = seq++; )
1616 T( trace(T_ADMIN, "admin: accepted connection %u", a->seq); )
1617 a->bg = 0;
1618 a->ref = 0;
1619 a->f = f;
1620 if (fd_in == STDIN_FILENO) a_stdin = a;
1621 fdflags(fd_in, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
1622 if (fd_out != fd_in)
1623 fdflags(fd_out, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
1624 selbuf_init(&a->b, &sel, fd_in, a_line, a);
1625 sel_initfile(&sel, &a->w, fd_out, SEL_WRITE, a_flush, a);
1626 a->out.hd = a->out.tl = 0;
1627 a->next = admins;
1628 a->prev = 0;
1629 if (admins) admins->prev = a;
1630 admins = a;
1631 }
1632
1633 /* --- @a_accept@ --- *
1634 *
1635 * Arguments: @int fd@ = file descriptor to accept
1636 * @unsigned mode@ = what to do
1637 * @void *v@ = uninteresting pointer
1638 *
1639 * Returns: ---
1640 *
1641 * Use: Accepts a new admin connection.
1642 */
1643
1644 static void a_accept(int fd, unsigned mode, void *v)
1645 {
1646 int nfd;
1647 struct sockaddr_un sun;
1648 size_t sz = sizeof(sun);
1649
1650 if ((nfd = accept(fd, (struct sockaddr *)&sun, &sz)) < 0) {
1651 if (errno != EINTR && errno != EAGAIN && errno != EWOULDBLOCK &&
1652 errno != ECONNABORTED && errno != EPROTO)
1653 a_warn("ADMIN", "accept-error", "?ERRNO", A_END);
1654 return;
1655 }
1656 a_create(nfd, nfd, 0);
1657 }
1658
1659 /* --- @a_preselect@ --- *
1660 *
1661 * Arguments: ---
1662 *
1663 * Returns: ---
1664 *
1665 * Use: Informs the admin module that we're about to select again,
1666 * and that it should do cleanup things it has delayed until a
1667 * `safe' time.
1668 */
1669
1670 void a_preselect(void) { if (a_dead) a_destroypending(); }
1671
1672 /* --- @a_daemon@ --- *
1673 *
1674 * Arguments: ---
1675 *
1676 * Returns: ---
1677 *
1678 * Use: Informs the admin module that it's a daemon.
1679 */
1680
1681 void a_daemon(void)
1682 {
1683 flags |= F_DAEMON;
1684 }
1685
1686 /* --- @a_init@ --- *
1687 *
1688 * Arguments: @const char *name@ = socket name to create
1689 *
1690 * Returns: ---
1691 *
1692 * Use: Creates the admin listening socket.
1693 */
1694
1695 void a_init(const char *name)
1696 {
1697 int fd;
1698 int n = 5;
1699 struct sockaddr_un sun;
1700 struct sigaction sa;
1701 size_t sz;
1702
1703 /* --- Set up the socket address --- */
1704
1705 sz = strlen(name) + 1;
1706 if (sz > sizeof(sun.sun_path))
1707 die(EXIT_FAILURE, "socket name `%s' too long", name);
1708 BURN(sun);
1709 sun.sun_family = AF_UNIX;
1710 memcpy(sun.sun_path, name, sz);
1711 sz += offsetof(struct sockaddr_un, sun_path);
1712
1713 /* --- Attempt to bind to the socket --- */
1714
1715 umask(0077);
1716 again:
1717 if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0)
1718 die(EXIT_FAILURE, "couldn't create socket: %s", strerror(errno));
1719 if (bind(fd, (struct sockaddr *)&sun, sz) < 0) {
1720 struct stat st;
1721 int e = errno;
1722 if (errno != EADDRINUSE) {
1723 die(EXIT_FAILURE, "couldn't bind to address `%s': %s",
1724 sun.sun_path, strerror(e));
1725 }
1726 if (!n)
1727 die(EXIT_FAILURE, "too many retries; giving up");
1728 n--;
1729 if (!connect(fd, (struct sockaddr *)&sun, sz)) {
1730 die(EXIT_FAILURE, "server already listening on admin socket `%s'",
1731 sun.sun_path);
1732 }
1733 if (errno != ECONNREFUSED)
1734 die(EXIT_FAILURE, "couldn't bind to address: %s", strerror(e));
1735 if (stat(sun.sun_path, &st)) {
1736 die(EXIT_FAILURE, "couldn't stat `%s': %s",
1737 sun.sun_path, strerror(errno));
1738 }
1739 if (!S_ISSOCK(st.st_mode))
1740 die(EXIT_FAILURE, "object `%s' isn't a socket", sun.sun_path);
1741 T( trace(T_ADMIN, "admin: stale socket found; removing it"); )
1742 unlink(sun.sun_path);
1743 close(fd);
1744 goto again;
1745 }
1746 chmod(sun.sun_path, 0600);
1747 fdflags(fd, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
1748 if (listen(fd, 5))
1749 die(EXIT_FAILURE, "couldn't listen on socket: %s", strerror(errno));
1750
1751 /* --- Listen to the socket --- */
1752
1753 sel_initfile(&sel, &sock, fd, SEL_READ, a_accept, 0);
1754 sel_addfile(&sock);
1755 sockname = name;
1756 bres_init(&sel);
1757 T( trace_custom(a_trace, 0);
1758 trace(T_ADMIN, "admin: enabled custom tracing"); )
1759 flags |= F_INIT;
1760
1761 /* --- Set up signal handlers --- */
1762
1763 sig_add(&s_term, SIGTERM, a_sigdie, 0);
1764 sig_add(&s_hup, SIGHUP, a_sighup, 0);
1765 signal(SIGPIPE, SIG_IGN);
1766 sigaction(SIGINT, 0, &sa);
1767 if (sa.sa_handler != SIG_IGN)
1768 sig_add(&s_int, SIGINT, a_sigdie, 0);
1769 }
1770
1771 /*----- That's all, folks -------------------------------------------------*/