server/: Use modern functions for address/text conversions.
[tripe] / server / admin.c
1 /* -*-c-*-
2 *
3 * Admin interface for configuration
4 *
5 * (c) 2001 Straylight/Edgeware
6 */
7
8 /*----- Licensing notice --------------------------------------------------*
9 *
10 * This file is part of Trivial IP Encryption (TrIPE).
11 *
12 * TrIPE is free software: you can redistribute it and/or modify it under
13 * the terms of the GNU General Public License as published by the Free
14 * Software Foundation; either version 3 of the License, or (at your
15 * option) any later version.
16 *
17 * TrIPE is distributed in the hope that it will be useful, but WITHOUT
18 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
20 * for more details.
21 *
22 * You should have received a copy of the GNU General Public License
23 * along with TrIPE. If not, see <https://www.gnu.org/licenses/>.
24 */
25
26 /*----- Header files ------------------------------------------------------*/
27
28 #include "tripe.h"
29
30 /*----- Global variables --------------------------------------------------*/
31
32 #ifndef NTRACE
33
34 const trace_opt tr_opts[] = {
35 { 't', T_TUNNEL, "tunnel events" },
36 { 'r', T_PEER, "peer events" },
37 { 'a', T_ADMIN, "admin interface" },
38 { 's', T_KEYSET, "symmetric keyset management" },
39 { 'x', T_KEYEXCH, "key exchange" },
40 { 'm', T_KEYMGMT, "key management" },
41 { 'l', T_CHAL, "challenge management" },
42 { 'v', T_PRIVSEP, "privilege separation" },
43 { 'p', T_PACKET, "packet contents" },
44 { 'c', T_CRYPTO, "crypto details" },
45 { 'A', T_ALL, "all of the above" },
46 { 0, 0, 0 }
47 };
48
49 unsigned tr_flags = 0;
50 #endif
51
52 static const trace_opt w_opts[] = {
53 #ifndef NTRACE
54 { 't', AF_TRACE, "trace messages" },
55 #endif
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 sym_table a_svcs;
69 static unsigned flags = 0;
70 static admin *a_stdin = 0;
71 static sig s_term, s_int, s_hup;
72
73 #define F_DAEMON 1u
74 #define F_INIT 2u
75 #define F_FOREGROUND 4u
76
77 #define T_RESOLVE SEC(30)
78 #define T_PING SEC(5)
79
80 static void a_destroy(admin */*a*/);
81
82 #define BOOL(x) ((x) ? "t" : "nil")
83
84 /*----- Output functions --------------------------------------------------*/
85
86 /* --- @trywrite@ --- *
87 *
88 * Arguments: @admin *a@ = pointer to an admin block
89 * @const char *p@ = pointer to buffer to write
90 * @size_t sz@ = size of data to write
91 *
92 * Returns: The number of bytes written, or less than zero on error.
93 *
94 * Use: Attempts to write data to a client.
95 */
96
97 static ssize_t trywrite(admin *a, const char *p, size_t sz)
98 {
99 ssize_t n, done = 0;
100
101 again:
102 if (!sz)
103 return (done);
104 n = write(a->w.fd, p, sz);
105 if (n > 0) {
106 done += n;
107 p += n;
108 sz -= n;
109 goto again;
110 }
111 if (n < 0) {
112 if (errno == EINTR)
113 goto again;
114 if (errno != EAGAIN && errno != EWOULDBLOCK) {
115 a_destroy(a);
116 a_warn("ADMIN", "client-write-error", "?ERRNO", A_END);
117 return (-1);
118 }
119 }
120 return (done);
121 }
122
123 /* --- @doqueue@ -- *
124 *
125 * Arguments: @oqueue *q@ = pointer to output queue
126 * @const char *p@ = pointer to buffer to write
127 * @size_t sz@ = size of buffer
128 *
129 * Returns: Nonzero if the queue was previously empty.
130 *
131 * Use: Queues data to be written later.
132 */
133
134 static int doqueue(oqueue *q, const char *p, size_t sz)
135 {
136 obuf *o;
137 int rc = 0;
138 size_t n;
139
140 o = q->tl;
141 if (!o)
142 rc = 1;
143 else if (o->p_in < o->buf + OBUFSZ)
144 goto noalloc;
145
146 do {
147 o = xmalloc(sizeof(obuf));
148 o->next = 0;
149 o->p_in = o->p_out = o->buf;
150 if (q->tl)
151 q->tl->next = o;
152 else
153 q->hd = o;
154 q->tl = o;
155
156 noalloc:
157 n = o->buf + OBUFSZ - o->p_in;
158 if (n > sz)
159 n = sz;
160 memcpy(o->p_in, p, n);
161 o->p_in += n;
162 p += n;
163 sz -= n;
164 } while (sz);
165
166 return (rc);
167 }
168
169 /* --- @dosend@ --- *
170 *
171 * Arguemnts: @admin *a@ = pointer to an admin block
172 * @const char *p@ = pointer to buffer to write
173 * @size_t sz@ = size of data to write
174 *
175 * Returns: ---
176 *
177 * Use: Sends data to an admin client.
178 */
179
180 static void dosend(admin *a, const char *p, size_t sz)
181 {
182 ssize_t n;
183
184 if (a->f & AF_DEAD)
185 return;
186 if (!a->out.hd) {
187 if ((n = trywrite(a, p, sz)) < 0)
188 return;
189 p += n;
190 sz -= n;
191 if (!sz)
192 return;
193 }
194 if (doqueue(&a->out, p, sz))
195 sel_addfile(&a->w);
196 }
197
198 /* --- @a_flush@ --- *
199 *
200 * Arguments: @int fd@ = file descriptor
201 * @unsigned mode@ = what's happening
202 * @void *v@ = pointer to my admin block
203 *
204 * Returns: ---
205 *
206 * Use: Flushes buffers when a client is ready to read again.
207 */
208
209 static void a_flush(int fd, unsigned mode, void *v)
210 {
211 admin *a = v;
212 obuf *o, *oo;
213 ssize_t n;
214
215 o = a->out.hd;
216 while (o) {
217 if ((n = trywrite(a, o->p_out, o->p_in - o->p_out)) < 0)
218 return;
219 o->p_out += n;
220 if (o->p_in < o->p_out)
221 break;
222 oo = o;
223 o = o->next;
224 xfree(oo);
225 }
226 a->out.hd = o;
227 if (!o) {
228 a->out.tl = 0;
229 sel_rmfile(&a->w);
230 }
231 }
232
233 /*----- Utility functions -------------------------------------------------*/
234
235 /* --- @a_vformat@ --- *
236 *
237 * Arguments: @dstr *d@ = where to leave the formatted message
238 * @const char *fmt@ = pointer to format string
239 * @va_list *ap@ = arguments in list
240 *
241 * Returns: ---
242 *
243 * Use: Main message token formatting driver. The arguments are
244 * interleaved formatting tokens and their parameters, finally
245 * terminated by an entry @A_END@.
246 *
247 * Tokens recognized:
248 *
249 * * "*..." ... -- pretokenized @dstr_putf@-like string
250 *
251 * * "?ADDR" SOCKADDR -- a socket address, to be converted
252 *
253 * * "?B64" BUFFER SIZE -- binary data to be base64-encoded
254 *
255 * * "?TOKENS" VECTOR -- null-terminated vector of tokens
256 *
257 * * "?PEER" PEER -- peer's name
258 *
259 * * "?ERRNO" ERRNO -- system error code
260 *
261 * * "[!]..." ... -- @dstr_putf@-like string as single token
262 */
263
264 void a_vformat(dstr *d, const char *fmt, va_list *ap)
265 {
266 dstr dd = DSTR_INIT;
267
268 while (fmt) {
269 if (*fmt == '*') {
270 if (d->len) dstr_putc(d, ' ');
271 dstr_vputf(d, fmt + 1, ap);
272 } else if (*fmt == '?') {
273 if (strcmp(fmt, "?ADDR") == 0) {
274 const addr *a = va_arg(*ap, const addr *);
275 char name[NI_MAXHOST], serv[NI_MAXSERV];
276 int ix, err;
277 if ((err = getnameinfo(&a->sa, addrsz(a),
278 name, sizeof(name), serv, sizeof(serv),
279 (NI_NUMERICHOST | NI_NUMERICSERV |
280 NI_DGRAM)))) {
281 dstr_putf(d, " E%d", err);
282 u_quotify(d, gai_strerror(err));
283 } else {
284 ix = afix(a->sa.sa_family); assert(ix >= 0);
285 u_quotify(d, aftab[ix].name);
286 u_quotify(d, name);
287 u_quotify(d, serv);
288 }
289 } else if (strcmp(fmt, "?B64") == 0) {
290 const octet *p = va_arg(*ap, const octet *);
291 size_t n = va_arg(*ap, size_t);
292 codec *b64 = base64_class.encoder(CDCF_NOEQPAD, "", 0);
293 dstr_putc(d, ' ');
294 b64->ops->code(b64, p, n, d);
295 b64->ops->code(b64, 0, 0, d);
296 b64->ops->destroy(b64);
297 } else if (strcmp(fmt, "?TOKENS") == 0) {
298 const char *const *av = va_arg(*ap, const char *const *);
299 while (*av) u_quotify(d, *av++);
300 } else if (strcmp(fmt, "?PEER") == 0)
301 u_quotify(d, p_name(va_arg(*ap, peer *)));
302 else if (strcmp(fmt, "?ERRNO") == 0) {
303 dstr_putf(d, " E%d", errno);
304 u_quotify(d, strerror(errno));
305 } else
306 abort();
307 } else {
308 if (*fmt == '!') fmt++;
309 DRESET(&dd);
310 dstr_vputf(&dd, fmt, ap);
311 u_quotify(d, dd.buf);
312 }
313 fmt = va_arg(*ap, const char *);
314 }
315 dstr_putz(d);
316
317 dstr_destroy(&dd);
318 }
319
320 /* --- @a_format@ --- *
321 *
322 * Arguments: @dstr *d@ = where to leave the formatted message
323 * @const char *fmt@ = pointer to format string
324 *
325 * Returns: ---
326 *
327 * Use: Writes a tokenized message into a string, for later
328 * presentation.
329 */
330
331 void a_format(dstr *d, const char *fmt, ...)
332 {
333 va_list ap;
334
335 va_start(ap, fmt);
336 a_vformat(d, fmt, &ap);
337 va_end(ap);
338 }
339
340 /* --- @a_write@, @a_vwrite@ --- *
341 *
342 * Arguments: @admin *a@ = admin connection to write to
343 * @const char *status@ = status code to report
344 * @const char *tag@ = tag string, or null
345 * @const char *fmt@ = pointer to format string
346 * @va_list *ap@ = arguments in list
347 * @...@ = other arguments
348 *
349 * Returns: ---
350 *
351 * Use: Sends a message to an admin connection.
352 */
353
354 static void a_vwrite(admin *a, const char *status, const char *tag,
355 const char *fmt, va_list *ap)
356 {
357 dstr d = DSTR_INIT;
358
359 if (tag) dstr_puts(&d, "BG");
360 dstr_puts(&d, status);
361 if (tag) u_quotify(&d, tag);
362 a_vformat(&d, fmt, ap);
363 dstr_putc(&d, '\n');
364 dosend(a, d.buf, d.len);
365 dstr_destroy(&d);
366 }
367
368 static void a_write(admin *a, const char *status, const char *tag,
369 const char *fmt, ...)
370 {
371 va_list ap;
372
373 va_start(ap, fmt);
374 a_vwrite(a, status, tag, fmt, &ap);
375 va_end(ap);
376 }
377
378 /* --- @a_ok@, @a_fail@ --- *
379 *
380 * Arguments: @admin *a@ = connection
381 * @const char *fmt@ = format string
382 * @...@ = other arguments
383 *
384 * Returns: ---
385 *
386 * Use: Convenience functions for @a_write@.
387 */
388
389 static void a_ok(admin *a) { a_write(a, "OK", 0, A_END); }
390
391 static void a_fail(admin *a, const char *fmt, ...)
392 {
393 va_list ap;
394
395 va_start(ap, fmt);
396 a_vwrite(a, "FAIL", 0, fmt, &ap);
397 va_end(ap);
398 }
399
400 /* --- @a_info@ --- *
401 *
402 * Arguments: @admin *a@ = connection
403 * @const char *fmt@ = format string
404 * @...@ = other arguments
405 *
406 * Returns: ---
407 *
408 * Use: Report information to an admin client.
409 */
410
411 void a_info(admin *a, const char *fmt, ...)
412 {
413 va_list ap;
414
415 va_start(ap, fmt);
416 a_vwrite(a, "INFO", 0, fmt, &ap);
417 va_end(ap);
418 }
419
420 /* --- @a_alert@, @a_valert@, @a_rawalert@ --- *
421 *
422 * Arguments: @unsigned f_and, f_eq@ = filter for connections
423 * @const char *status@ = status string
424 * @const char *fmt@ = pointer to format string
425 * @const char *p@ = pointer to raw string
426 * @size_t sz@ = size of raw string
427 * @va_list *ap@ = arguments in list
428 * @...@ = other arguments
429 *
430 * Returns: ---
431 *
432 * Use: Write a message to all admin connections matched by the given
433 * filter.
434 */
435
436 static void a_rawalert(unsigned f_and, unsigned f_eq, const char *status,
437 const char *p, size_t sz)
438 {
439 admin *a, *aa;
440 dstr d = DSTR_INIT;
441
442 if (!(flags & F_INIT))
443 return;
444 dstr_puts(&d, status);
445 if (p) {
446 dstr_putc(&d, ' ');
447 dstr_putm(&d, p, sz);
448 }
449 dstr_putc(&d, '\n');
450 p = d.buf;
451 sz = d.len;
452 for (a = admins; a; a = aa) {
453 aa = a->next;
454 if ((a->f & f_and) == f_eq)
455 dosend(a, d.buf, d.len);
456 }
457 dstr_destroy(&d);
458 }
459
460 static void a_valert(unsigned f_and, unsigned f_eq, const char *status,
461 const char *fmt, va_list *ap)
462 {
463 dstr d = DSTR_INIT;
464
465 if (!(flags & F_INIT))
466 return;
467 a_vformat(&d, fmt, ap);
468 a_rawalert(f_and, f_eq, status, fmt ? d.buf : 0, fmt ? d.len : 0);
469 dstr_destroy(&d);
470 }
471
472 static void a_alert(unsigned f_and, unsigned f_eq, const char *status,
473 const char *fmt, ...)
474 {
475 va_list ap;
476
477 va_start(ap, fmt);
478 a_valert(f_and, f_eq, status, fmt, &ap);
479 va_end(ap);
480 }
481
482 /* --- @a_warn@ --- *
483 *
484 * Arguments: @const char *fmt@ = pointer to format string
485 * @...@ = other arguments
486 *
487 * Returns: ---
488 *
489 * Use: Informs all admin connections of a warning.
490 */
491
492 void a_warn(const char *fmt, ...)
493 {
494 va_list ap;
495
496 va_start(ap, fmt);
497 if (flags & F_INIT)
498 a_valert(0, 0, "WARN", fmt, &ap);
499 else {
500 dstr d = DSTR_INIT;
501 fprintf(stderr, "%s: ", QUIS);
502 a_vformat(&d, fmt, &ap);
503 dstr_putc(&d, '\n');
504 dstr_write(&d, stderr);
505 dstr_destroy(&d);
506 }
507 va_end(ap);
508 }
509
510 /* --- @a_trace@ --- *
511 *
512 * Arguments: @const char *p@ = pointer to a buffer
513 * @size_t sz@ = size of the buffer
514 * @void *v@ = uninteresting pointer
515 *
516 * Returns: ---
517 *
518 * Use: Custom trace output handler. Sends trace messages to
519 * interested admin connections.
520 */
521
522 #ifndef NTRACE
523 static void a_trace(const char *p, size_t sz, void *v)
524 { a_rawalert(AF_TRACE, AF_TRACE, "TRACE", p, sz); }
525 #endif
526
527 /* --- @a_notify@ --- *
528 *
529 * Arguments: @const char *fmt@ = pointer to format string
530 * @...@ = other arguments
531 *
532 * Returns: ---
533 *
534 * Use: Sends a notification to interested admin connections.
535 */
536
537 void a_notify(const char *fmt, ...)
538 {
539 va_list ap;
540
541 va_start(ap, fmt);
542 a_valert(AF_NOTE, AF_NOTE, "NOTE", fmt, &ap);
543 va_end(ap);
544 }
545
546 /* --- @a_quit@ --- *
547 *
548 * Arguments: ---
549 *
550 * Returns: ---
551 *
552 * Use: Shuts things down nicely.
553 */
554
555 void a_quit(void)
556 {
557 close(sock.fd);
558 unlink(sockname);
559 FOREACH_PEER(p, { p_destroy(p); });
560 ps_quit();
561 exit(0);
562 }
563
564 /* --- @a_sigdie@ --- *
565 *
566 * Arguments: @int sig@ = signal number
567 * @void *v@ = an uninteresting argument
568 *
569 * Returns: ---
570 *
571 * Use Shuts down on receipt of a fatal signal.
572 */
573
574 static void a_sigdie(int sig, void *v)
575 {
576 char *p;
577 char buf[20];
578
579 switch (sig) {
580 case SIGTERM: p = "SIGTERM"; break;
581 case SIGINT: p = "SIGINT"; break;
582 default:
583 sprintf(buf, "%i", sig);
584 p = buf;
585 break;
586 }
587 a_warn("SERVER", "quit", "signal", "%s", p, A_END);
588 a_quit();
589 }
590
591 /* --- @a_sighup@ --- *
592 *
593 * Arguments: @int sig@ = signal number
594 * @void *v@ = an uninteresting argument
595 *
596 * Returns: ---
597 *
598 * Use Logs a message about SIGHUP not being useful.
599 */
600
601 static void a_sighup(int sig, void *v)
602 { a_warn("SERVER", "ignore", "signal", "SIGHUP", A_END); }
603
604 /* --- @a_parsetime@ --- *
605 *
606 * Arguments; @const char *p@ = time string to parse
607 *
608 * Returns: Time in seconds, or @< 0@ on error.
609 */
610
611 static long a_parsetime(const char *p)
612 {
613 char *q;
614 long t = strtol(p, &q, 0);
615
616 switch (*q) {
617 case 'd': t *= 24;
618 case 'h': t *= 60;
619 case 'm': t *= 60;
620 case 's': if (q[1] != 0)
621 default: t = -1;
622 case 0: break;
623 }
624 return (t);
625 }
626
627 /* --- @a_findpeer@ --- *
628 *
629 * Arguments: @admin *a@ = admin connection
630 * @const char *pn@ = peer name
631 *
632 * Returns: The peer, or null if not there.
633 *
634 * Use: Finds a peer, reporting an error if it failed.
635 */
636
637 static peer *a_findpeer(admin *a, const char *pn)
638 {
639 peer *p;
640
641 if ((p = p_find(pn)) == 0)
642 a_fail(a, "unknown-peer", "%s", pn, A_END);
643 return (p);
644 }
645
646 /*----- Backgrounded operations -------------------------------------------*/
647
648 #define BGTAG(bg) \
649 (((admin_bgop *)(bg))->tag ? ((admin_bgop *)(bg))->tag : "<foreground>")
650
651 /* --- @a_bgfind@ --- *
652 *
653 * Arguments: @admin *a@ = a client block
654 * @const char *tag@ = the requested tag
655 *
656 * Returns: The requested background job, or null.
657 */
658
659 static admin_bgop *a_bgfind(admin *a, const char *tag)
660 {
661 admin_bgop *bg;
662
663 for (bg = a->bg; bg; bg = bg->next) {
664 if (bg->tag && strcmp(tag, bg->tag) == 0)
665 return (bg);
666 }
667 return (0);
668 }
669
670 /* --- @a_bgrelease@ --- *
671 *
672 * Arguments: @admin_bgop *bg@ = backgrounded operation
673 *
674 * Returns: ---
675 *
676 * Use: Removes a backgrounded operation from the queue, since
677 * (presumably) it's done.
678 */
679
680 static void a_bgrelease(admin_bgop *bg)
681 {
682 admin *a = bg->a;
683
684 T( trace(T_ADMIN, "admin: release bgop %s", BGTAG(bg)); )
685 if (bg->tag) xfree(bg->tag);
686 else selbuf_enable(&a->b);
687 if (bg->next) bg->next->prev = bg->prev;
688 if (bg->prev) bg->prev->next = bg->next;
689 else a->bg = bg->next;
690 xfree(bg);
691 if (!a->bg && (a->f & AF_CLOSE)) a_destroy(a);
692 }
693
694 /* --- @a_bgok@, @a_bginfo@, @a_bgfail@ --- *
695 *
696 * Arguments: @admin_bgop *bg@ = backgrounded operation
697 * @const char *fmt@ = format string
698 * @...@ = other arguments
699 *
700 * Returns: ---
701 *
702 * Use: Convenience functions for @a_write@.
703 */
704
705 static void a_bgok(admin_bgop *bg)
706 { a_write(bg->a, "OK", bg->tag, A_END); }
707
708 static void a_bginfo(admin_bgop *bg, const char *fmt, ...)
709 {
710 va_list ap;
711 va_start(ap, fmt);
712 a_vwrite(bg->a, "INFO", bg->tag, fmt, &ap);
713 va_end(ap);
714 }
715
716 static void a_bgfail(admin_bgop *bg, const char *fmt, ...)
717 {
718 va_list ap;
719 va_start(ap, fmt);
720 a_vwrite(bg->a, "FAIL", bg->tag, fmt, &ap);
721 va_end(ap);
722 }
723
724 /* --- @a_bgadd@ --- *
725 *
726 * Arguments: @admin *a@ = administration connection
727 * @admin_bgop *bg@ = pointer to background operation
728 * @const char *tag@ = background tag, or null for foreground
729 * @void (*cancel)(admin_bgop *)@ = cancel function
730 *
731 * Returns: Zero for success, nonzero on failure.
732 *
733 * Use: Links a background job into the list.
734 */
735
736 static int a_bgadd(admin *a, admin_bgop *bg, const char *tag,
737 void (*cancel)(admin_bgop *))
738 {
739 if (tag) {
740 if (a_bgfind(a, tag)) {
741 a_fail(a, "tag-exists", "%s", tag, A_END);
742 return (-1);
743 }
744 bg->tag = xstrdup(tag);
745 }
746 else {
747 bg->tag = 0;
748 selbuf_disable(&a->b);
749 }
750 bg->a = a;
751 bg->cancel = cancel;
752 bg->next = a->bg;
753 bg->prev = 0;
754 if (a->bg) a->bg->prev = bg;
755 a->bg = bg;
756 T( trace(T_ADMIN, "admin: add bgop %s", BGTAG(bg)); )
757 if (tag) a_write(a, "DETACH", tag, A_END);
758 return (0);
759 }
760
761 /*----- Job table manipulation --------------------------------------------*/
762
763 #define JOB_SHIFT 16
764 #define JOB_INDEXMASK ((1ul << JOB_SHIFT) - 1)
765 #define JOB_SEQMASK ((1ul << (32 - JOB_SHIFT)) - 1)
766
767 #define JOB_END 0xfffffffful
768
769 static unsigned long a_joboffset;
770
771 /* --- @a_jobidencode@ --- *
772 *
773 * Arguments: @admin_svcop *svc@ = pointer to a service operation
774 *
775 * Returns: A jobid for this job, in an internal static buffer.
776 *
777 * Use: Constructs a jobid. In order to dissuade people from
778 * predicting jobids, we obfuscate them.
779 *
780 * A `raw' jobid consists of two 16-bit fields. The low 16 bits
781 * are an index into a big array. The high 16 bits are a
782 * sequence number recording how many times that slot has been
783 * reused.
784 *
785 * This `raw' jobid is then obfuscated by adding a randomly-
786 * generated offset, and multiplying (mod %$2^{32}$%) by a fixed
787 * odd constant.
788 */
789
790 static const char *a_jobidencode(admin_svcop *svc)
791 {
792 admin_jobtable *j = &svc->prov->j;
793 static char buf[10];
794 unsigned long pre;
795 unsigned seq;
796
797 assert(svc->index <= JOB_INDEXMASK);
798 seq = j->v[svc->index].seq;
799 assert(seq <= JOB_SEQMASK);
800 pre = (unsigned long)svc->index | ((unsigned long)seq << JOB_SHIFT);
801 sprintf(buf, "J%08lx", ((pre + a_joboffset) * 0x0f87a7a3ul) & 0xffffffff);
802 return (buf);
803 }
804
805 /* --- @a_jobiddecode@ --- *
806 *
807 * Arguments: @admin_jobtable *j@ = pointer to a job table
808 * @const char *jid@ = pointer to a jobid string
809 *
810 * Returns: A pointer to the job's @svcop@ structure.
811 */
812
813 static admin_svcop *a_jobiddecode(admin_jobtable *j, const char *jid)
814 {
815 unsigned i;
816 unsigned long pre;
817
818 if (jid[0] != 'J')
819 return (0);
820 for (i = 1; i < 9; i++) {
821 if (!isxdigit((unsigned char)jid[i]))
822 return (0);
823 }
824 if (jid[9] != 0)
825 return (0);
826 pre = strtoul(jid + 1, 0, 16);
827 pre = ((pre * 0xbd11c40bul) - a_joboffset) & 0xffffffff;
828 i = pre & JOB_INDEXMASK;
829 if (i >= j->n || j->v[i].seq != (pre >> JOB_SHIFT))
830 return (0);
831 return (j->v[i].u.op);
832 }
833
834 /* --- @a_jobcreate@ --- *
835 *
836 * Arguments: @admin *a@ = pointer to administration client
837 *
838 * Returns: A pointer to a freshly-allocated @svcop@, or null.
839 *
840 * Use: Allocates a fresh @svcop@ and links it into a job table.
841 */
842
843 static admin_svcop *a_jobcreate(admin *a)
844 {
845 admin_svcop *svc;
846 unsigned i;
847 unsigned sz;
848 admin_jobtable *j = &a->j;
849
850 if (j->free != JOB_END) {
851 i = j->free;
852 j->free = j->v[i].u.next;
853 } else {
854 if (j->n == j->sz) {
855 if (j->sz > JOB_INDEXMASK)
856 return (0);
857 sz = j->sz;
858 if (!sz) {
859 j->sz = 16;
860 j->v = xmalloc(j->sz * sizeof(*j->v));
861 } else {
862 j->sz = sz << 1;
863 j->v = xrealloc(j->v, j->sz * sizeof(*j->v), sz * sizeof(*j->v));
864 }
865 }
866 i = j->n++;
867 j->v[i].seq = 0;
868 }
869 svc = xmalloc(sizeof(*svc));
870 svc->index = i;
871 svc->prov = a;
872 svc->next = j->active;
873 svc->prev = 0;
874 if (j->active) j->active->prev = svc;
875 j->active = svc;
876 j->v[i].u.op = svc;
877 IF_TRACING(T_ADMIN, {
878 trace(T_ADMIN, "admin: created job %s (%u)", a_jobidencode(svc), i);
879 })
880 return (svc);
881 }
882
883 /* --- @a_jobdestroy@ --- *
884 *
885 * Arguments: @admin_svcop *svc@ = pointer to job block
886 *
887 * Returns: ---
888 *
889 * Use: Frees up a completed (or cancelled) job.
890 */
891
892 static void a_jobdestroy(admin_svcop *svc)
893 {
894 admin *a = svc->prov;
895 admin_jobtable *j = &a->j;
896 unsigned i = svc->index;
897
898 IF_TRACING(T_ADMIN, {
899 trace(T_ADMIN, "admin: destroying job %s (%u)", a_jobidencode(svc), i);
900 })
901 assert(j->v[i].u.op = svc);
902 j->v[i].u.next = j->free;
903 j->v[i].seq++;
904 j->free = i;
905 if (svc->next) svc->next->prev = svc->prev;
906 if (svc->prev) svc->prev->next = svc->next;
907 else j->active = svc->next;
908 }
909
910 /* --- @a_jobtableinit@ --- *
911 *
912 * Arguments: @admin_jobtable *j@ = pointer to job table
913 *
914 * Returns: ---
915 *
916 * Use: Initializes a job table.
917 */
918
919 static void a_jobtableinit(admin_jobtable *j)
920 {
921 if (!a_joboffset)
922 a_joboffset = GR_RANGE(&rand_global, 0xffffffff) + 1;
923 j->n = j->sz = 0;
924 j->active = 0;
925 j->free = JOB_END;
926 j->v = 0;
927 }
928
929 /* --- @a_jobtablefinal@ --- *
930 *
931 * Arguments: @admin_jobtable *j@ = pointer to job table
932 *
933 * Returns: ---
934 *
935 * Use: Closes down a job table.
936 */
937
938 static void a_jobtablefinal(admin_jobtable *j)
939 {
940 admin_svcop *svc, *ssvc;
941
942 for (svc = j->active; svc; svc = ssvc) {
943 ssvc = svc->next;
944 a_bgfail(&svc->bg, "provider-failed", A_END);
945 a_bgrelease(&svc->bg);
946 }
947 if (j->v) xfree(j->v);
948 }
949
950 /*----- Services infrastructure -------------------------------------------*/
951
952 /* --- @a_svcfind@ --- *
953 *
954 * Arguments: @admin *a@ = the requesting client
955 * @const char *name@ = service name wanted
956 *
957 * Returns: The service requested, or null.
958 *
959 * Use: Finds a service; reports an error if the service couldn't be
960 * found.
961 */
962
963 static admin_service *a_svcfind(admin *a, const char *name)
964 {
965 admin_service *svc;
966
967 if ((svc = sym_find(&a_svcs, name, -1, 0, 0)) == 0) {
968 a_fail(a, "unknown-service", "%s", name, A_END);
969 return (0);
970 }
971 return (svc);
972 }
973
974 /* --- @a_svcunlink@ --- *
975 *
976 * Arguments: @admin_service *svc@ = pointer to service structure
977 *
978 * Returns: ---
979 *
980 * Use: Unlinks the service from its provider, without issuing a
981 * message or freeing the structure. The version string is
982 * freed, however.
983 */
984
985 static void a_svcunlink(admin_service *svc)
986 {
987 if (svc->next)
988 svc->next->prev = svc->prev;
989 if (svc->prev)
990 svc->prev->next = svc->next;
991 else
992 svc->prov->svcs = svc->next;
993 xfree(svc->version);
994 }
995
996 /* --- @a_svcrelease@ --- *
997 *
998 * Arguments: @admin_service *svc@ = pointer to service structure
999 *
1000 * Returns: ---
1001 *
1002 * Use: Releases a service and frees its structure.
1003 */
1004
1005 static void a_svcrelease(admin_service *svc)
1006 {
1007 a_notify("SVCRELEASE", "%s", SYM_NAME(svc), A_END);
1008 a_svcunlink(svc);
1009 sym_remove(&a_svcs, svc);
1010 }
1011
1012 /*----- Name resolution operations ----------------------------------------*/
1013
1014 /* --- @a_resolved@ --- *
1015 *
1016 * Arguments: @struct hostent *h@ = pointer to resolved hostname
1017 * @void *v@ = pointer to resolver operation
1018 *
1019 * Returns: ---
1020 *
1021 * Use: Handles a completed name resolution.
1022 */
1023
1024 static void a_resolved(struct hostent *h, void *v)
1025 {
1026 admin_resop *r = v;
1027
1028 QUICKRAND;
1029 if (!h) {
1030 T( trace(T_ADMIN, "admin: resop %s failed: %s",
1031 BGTAG(r), hstrerror(h_errno)); )
1032 a_bgfail(&r->bg, "resolve-error", "%s", r->addr, A_END);
1033 r->func(r, ARES_FAIL);
1034 } else {
1035 T( trace(T_ADMIN, "admin: resop %s ok", BGTAG(r)); )
1036 r->sa.sin.sin_family = AF_INET;
1037 memcpy(&r->sa.sin.sin_addr, h->h_addr, sizeof(struct in_addr));
1038 setport(&r->sa, r->port);
1039 r->func(r, ARES_OK);
1040 }
1041 sel_rmtimer(&r->t);
1042 xfree(r->addr);
1043 a_bgrelease(&r->bg);
1044 }
1045
1046 /* --- @a_restimer@ --- *
1047 *
1048 * Arguments: @struct timeval *tv@ = timer
1049 * @void *v@ = pointer to resolver operation
1050 *
1051 * Returns: ---
1052 *
1053 * Use: Times out a resolver.
1054 */
1055
1056 static void a_restimer(struct timeval *tv, void *v)
1057 {
1058 admin_resop *r = v;
1059
1060 T( trace(T_ADMIN, "admin: resop %s timeout", BGTAG(r)); )
1061 a_bgfail(&r->bg, "resolver-timeout", "%s", r->addr, A_END);
1062 r->func(r, ARES_FAIL);
1063 bres_abort(&r->r);
1064 xfree(r->addr);
1065 a_bgrelease(&r->bg);
1066 }
1067
1068 /* --- @a_rescancel@ --- *
1069 *
1070 * Arguments: @admin_bgop *bg@ = background operation
1071 *
1072 * Returns: ---
1073 *
1074 * Use: Cancels an add operation.
1075 */
1076
1077 static void a_rescancel(admin_bgop *bg)
1078 {
1079 admin_resop *r = (admin_resop *)bg;
1080
1081 T( trace(T_ADMIN, "admin: cancel resop %s", BGTAG(r)); )
1082 r->func(r, ARES_FAIL);
1083 sel_rmtimer(&r->t);
1084 xfree(r->addr);
1085 bres_abort(&r->r);
1086 }
1087
1088 /* --- @a_resolve@ --- *
1089 *
1090 * Arguments: @admin *a@ = administration connection
1091 * @admin_resop *r@ = resolver operation to run
1092 * @const char *tag@ = background operation tag
1093 * @void (*func)(struct admin_resop *, int@ = handler function
1094 * @unsigned ac@ = number of remaining arguments
1095 * @char *av[]@ = pointer to remaining arguments
1096 *
1097 * Returns: ---
1098 *
1099 * Use: Cranks up a resolver job.
1100 */
1101
1102 static void a_resolve(admin *a, admin_resop *r, const char *tag,
1103 void (*func)(struct admin_resop *, int),
1104 unsigned ac, char *av[])
1105 {
1106 struct timeval tv;
1107 unsigned long pt;
1108 int af = AF_UNSPEC;
1109 const char *fam = "ANY";
1110 char *p;
1111 int i = 0, j;
1112 struct addrinfo *ai, *ailist, aihint = { 0 };
1113
1114 /* --- Fill in the easy bits of address --- */
1115
1116 r->bg.tag = "<starting>";
1117 r->addr = 0;
1118 r->func = func;
1119 if (mystrieq(av[i], "any"))
1120 { fam = "ANY"; af = AF_UNSPEC; i++; }
1121 else for (j = 0; j < NADDRFAM; j++) {
1122 if (mystrieq(av[i], aftab[j].name)) {
1123 assert(udpsock[j].fd >= 0);
1124 fam = aftab[j].name;
1125 af = aftab[j].af;
1126 i++;
1127 break;
1128 }
1129 }
1130 if (ac - i != 1 && ac - i != 2) {
1131 a_fail(a, "bad-addr-syntax", "[FAMILY] ADDRESS [PORT]", A_END);
1132 goto fail;
1133 }
1134 r->addr = xstrdup(av[i]);
1135 if (!av[i + 1])
1136 pt = TRIPE_PORT;
1137 else {
1138 pt = strtoul(av[i + 1], &p, 0);
1139 if (*p) {
1140 struct servent *s = getservbyname(av[i + 1], "udp");
1141 if (!s) {
1142 a_fail(a, "unknown-service", "%s", av[i + 1], A_END);
1143 goto fail;
1144 }
1145 pt = ntohs(s->s_port);
1146 }
1147 }
1148 if (pt == 0 || pt >= 65536) {
1149 a_fail(a, "invalid-port", "%lu", pt, A_END);
1150 goto fail;
1151 }
1152 r->port = pt;
1153
1154 /* --- Report backgrounding --- *
1155 *
1156 * Do this for consistency of interface, even if we're going to get the
1157 * answer straight away.
1158 */
1159
1160 if (a_bgadd(a, &r->bg, tag, a_rescancel))
1161 goto fail;
1162 T( trace(T_ADMIN, "admin: %u, resop %s, hostname `%s', family `%s'",
1163 a->seq, BGTAG(r), r->addr, fam); )
1164
1165 /* --- Make sure the address family is something we can implement --- */
1166
1167 if (af != AF_UNSPEC && af != AF_INET) {
1168 T( trace(T_ADMIN, "admin: resop %s failed: unsupported address family",
1169 BGTAG(r)); )
1170 a_bgfail(&r->bg, "resolve-error", "%s", r->addr, A_END);
1171 goto fail_release;
1172 }
1173 assert(udpsock[AFIX_INET].fd >= 0);
1174
1175 /* --- If the name is numeric, do it the easy way --- */
1176
1177 aihint.ai_family = AF_INET;
1178 aihint.ai_socktype = SOCK_DGRAM;
1179 aihint.ai_protocol = IPPROTO_UDP;
1180 aihint.ai_flags = AI_NUMERICHOST;
1181 if (!getaddrinfo(av[i], 0, &aihint, &ailist)) {
1182 for (ai = ailist; ai; ai = ai->ai_next) {
1183 if ((j = afix(ai->ai_family)) >= 0 && udpsock[j].fd >= 0)
1184 break;
1185 }
1186 if (!ai) {
1187 T( trace(T_ADMIN, "admin: resop %s failed: "
1188 "no suitable addresses returned", BGTAG(r)); )
1189 a_bgfail(&r->bg, "resolve-error", "%s" , r->addr, A_END);
1190 func(r, ARES_FAIL);
1191 } else {
1192 T( trace(T_ADMIN, "admin: resop %s done the easy way", BGTAG(r)); )
1193 assert(ai->ai_addrlen <= sizeof(r->sa));
1194 memcpy(&r->sa, ai->ai_addr, ai->ai_addrlen);
1195 setport(&r->sa, r->port);
1196 func(r, ARES_OK);
1197 }
1198 freeaddrinfo(ailist);
1199 xfree(r->addr);
1200 a_bgrelease(&r->bg);
1201 return;
1202 }
1203
1204 /* --- Store everything for later and crank up the resolver --- */
1205
1206 gettimeofday(&tv, 0);
1207 tv.tv_sec += T_RESOLVE;
1208 sel_addtimer(&sel, &r->t, &tv, a_restimer, r);
1209 bres_byname(&r->r, r->addr, a_resolved, r);
1210 return;
1211
1212 fail:
1213 func(r, ARES_FAIL);
1214 if (r->addr) xfree(r->addr);
1215 xfree(r);
1216 return;
1217
1218 fail_release:
1219 func(r, ARES_FAIL);
1220 xfree(r->addr);
1221 a_bgrelease(&r->bg);
1222 }
1223
1224 /*----- Option parsing ----------------------------------------------------*/
1225
1226 #define OPTIONS(argc, argv, guts) do { \
1227 char **o_av = argv; \
1228 for (;; o_av++) { \
1229 if (!*o_av) \
1230 break; \
1231 if (mystrieq(*o_av, "--")) { \
1232 o_av++; \
1233 break; \
1234 } \
1235 guts \
1236 if (**o_av == '-') \
1237 goto bad_syntax; \
1238 break; \
1239 } \
1240 argc -= o_av - argv; \
1241 argv = o_av; \
1242 } while (0)
1243
1244 #define OPT(name, guts) if (mystrieq(*o_av, name)) { guts continue; }
1245
1246 #define OPTARG(name, arg, guts) OPT(name, { \
1247 const char *arg; \
1248 arg = *++o_av; \
1249 if (!arg) goto bad_syntax; \
1250 guts \
1251 })
1252
1253 #define OPTTIME(name, arg, guts) OPTARG(name, o_arg, { \
1254 long arg; \
1255 if ((arg = a_parsetime(o_arg)) < 0) { \
1256 a_fail(a, "bad-time-spec", "%s", o_arg, A_END); \
1257 goto fail; \
1258 } \
1259 guts \
1260 })
1261
1262 /*----- Adding peers ------------------------------------------------------*/
1263
1264 /* --- @a_doadd@ --- *
1265 *
1266 * Arguments: @admin_resop *r@ = resolver operation
1267 * @int rc@ = how it worked
1268 *
1269 * Returns: ---
1270 *
1271 * Use: Handles a completed resolution.
1272 */
1273
1274 static void a_doadd(admin_resop *r, int rc)
1275 {
1276 admin_addop *add = (admin_addop *)r;
1277
1278 T( trace(T_ADMIN, "admin: done add op %s", BGTAG(add)); )
1279
1280 if (rc == ARES_OK) {
1281 add->peer.sa = add->r.sa;
1282 if (p_findbyaddr(&add->r.sa))
1283 a_bgfail(&add->r.bg, "peer-addr-exists", "?ADDR", &add->r.sa, A_END);
1284 else if (p_find(add->peer.name))
1285 a_bgfail(&add->r.bg, "peer-exists", "%s", add->peer.name, A_END);
1286 else if (!p_create(&add->peer))
1287 a_bgfail(&add->r.bg, "peer-create-fail", "%s", add->peer.name, A_END);
1288 else
1289 a_bgok(&add->r.bg);
1290 }
1291
1292 if (add->peer.tag) xfree(add->peer.tag);
1293 if (add->peer.privtag) xfree(add->peer.privtag);
1294 xfree(add->peer.name);
1295 }
1296
1297 /* --- @acmd_add@ --- *
1298 *
1299 * Arguments: @admin *a@ = connection which requested the addition
1300 * @unsigned ac@ = argument count
1301 * @char *av[]@ = pointer to the argument list
1302 *
1303 * Returns: ---
1304 *
1305 * Use: Adds a new peer.
1306 */
1307
1308 static void acmd_add(admin *a, unsigned ac, char *av[])
1309 {
1310 const char *tag = 0;
1311 admin_addop *add;
1312
1313 /* --- Set stuff up --- */
1314
1315 add = xmalloc(sizeof(*add));
1316 add->peer.name = 0;
1317 add->peer.tag = 0;
1318 add->peer.privtag = 0;
1319 add->peer.t_ka = 0;
1320 add->peer.tops = tun_default;
1321 add->peer.f = 0;
1322
1323 /* --- Parse options --- */
1324
1325 OPTIONS(ac, av, {
1326 OPTARG("-background", arg, { tag = arg; })
1327 OPTARG("-tunnel", arg, {
1328 unsigned i;
1329 for (i = 0;; i++) {
1330 if (!tunnels[i]) {
1331 a_fail(a, "unknown-tunnel", "%s", arg, A_END);
1332 goto fail;
1333 }
1334 if (mystrieq(arg, tunnels[i]->name)) {
1335 add->peer.tops = tunnels[i];
1336 break;
1337 }
1338 }
1339 })
1340 OPTTIME("-keepalive", t, { add->peer.t_ka = t; })
1341 OPT("-cork", { add->peer.f |= KXF_CORK; })
1342 OPTARG("-key", arg, {
1343 if (add->peer.tag) xfree(add->peer.tag);
1344 add->peer.tag = xstrdup(arg);
1345 })
1346 OPT("-mobile", { add->peer.f |= PSF_MOBILE; })
1347 OPTARG("-priv", arg, {
1348 if (add->peer.privtag) xfree(add->peer.privtag);
1349 add->peer.privtag = xstrdup(arg);
1350 })
1351 });
1352
1353 /* --- Make sure someone's not got there already --- */
1354
1355 if (!av[0] || !av[1])
1356 goto bad_syntax;
1357 if (p_find(*av)) {
1358 a_fail(a, "peer-exists", "%s", *av, A_END);
1359 goto fail;
1360 }
1361 add->peer.name = xstrdup(*av++);
1362 ac--;
1363
1364 /* --- Crank up the resolver --- */
1365
1366 a_resolve(a, &add->r, tag, a_doadd, ac, av);
1367 return;
1368
1369 /* --- Clearing up --- */
1370
1371 bad_syntax:
1372 a_fail(a, "bad-syntax", "add", "[OPTIONS] PEER ADDR ...", A_END);
1373 fail:
1374 if (add->peer.name) xfree(add->peer.name);
1375 if (add->peer.tag) xfree(add->peer.tag);
1376 if (add->peer.privtag) xfree(add->peer.privtag);
1377 xfree(add);
1378 return;
1379 }
1380
1381 /*----- Ping --------------------------------------------------------------*/
1382
1383 /* --- @a_pingcancel@ --- *
1384 *
1385 * Arguments: @admin_bgop *bg@ = background operation block
1386 *
1387 * Returns: ---
1388 *
1389 * Use: Cancels a running ping.
1390 */
1391
1392 static void a_pingcancel(admin_bgop *bg)
1393 {
1394 admin_pingop *pg = (admin_pingop *)bg;
1395 T( trace(T_ADMIN, "admin: cancel ping op %s", BGTAG(pg)); )
1396 p_pingdone(&pg->ping, PING_NONOTIFY);
1397 }
1398
1399 /* --- @a_pong@ --- *
1400 *
1401 * Arguments: @int rc@ = return code
1402 * @void *v@ = ping operation block
1403 *
1404 * Returns: ---
1405 *
1406 * Use: Collects what happened to a ping message.
1407 */
1408
1409 static void a_pong(int rc, void *v)
1410 {
1411 admin_pingop *pg = v;
1412 struct timeval tv;
1413 double millis;
1414
1415 switch (rc) {
1416 case PING_OK:
1417 gettimeofday(&tv, 0);
1418 tv_sub(&tv, &tv, &pg->pingtime);
1419 millis = (double)tv.tv_sec * 1000 + (double)tv.tv_usec/1000;
1420 a_bginfo(&pg->bg, "ping-ok", "%.1f", millis, A_END);
1421 a_bgok(&pg->bg);
1422 break;
1423 case PING_TIMEOUT:
1424 a_bginfo(&pg->bg, "ping-timeout", A_END);
1425 a_bgok(&pg->bg);
1426 break;
1427 case PING_PEERDIED:
1428 a_bginfo(&pg->bg, "ping-peer-died", A_END);
1429 a_bgok(&pg->bg);
1430 break;
1431 default:
1432 abort();
1433 }
1434 T( trace(T_ADMIN, "admin: ponged ping op %s", BGTAG(pg)); )
1435 a_bgrelease(&pg->bg);
1436 }
1437
1438 /* --- @acmd_ping@, @acmd_eping@ --- *
1439 *
1440 * Arguments: @admin *a@ = connection which requested the ping
1441 * @unsigned ac@ = argument count
1442 * @char *av[]@ = pointer to the argument list
1443 *
1444 * Returns: ---
1445 *
1446 * Use: Pings a peer.
1447 */
1448
1449 static void a_ping(admin *a, unsigned ac, char *av[],
1450 const char *cmd, unsigned msg)
1451 {
1452 long t = T_PING;
1453 peer *p;
1454 admin_pingop *pg = 0;
1455 const char *tag = 0;
1456
1457 OPTIONS(ac, av, {
1458 OPTARG("-background", arg, { tag = arg; })
1459 OPTTIME("-timeout", arg, { t = arg; })
1460 });
1461 if (!*av || av[1]) goto bad_syntax;
1462 if ((p = a_findpeer(a, *av)) == 0)
1463 return;
1464 pg = xmalloc(sizeof(*pg));
1465 gettimeofday(&pg->pingtime, 0);
1466 if (a_bgadd(a, &pg->bg, tag, a_pingcancel))
1467 goto fail;
1468 T( trace(T_ADMIN, "admin: ping op %s: %s to %s",
1469 BGTAG(pg), cmd, p_name(p)); )
1470 if (p_pingsend(p, &pg->ping, msg, t, a_pong, pg)) {
1471 a_bgfail(&pg->bg, "ping-send-failed", A_END);
1472 a_bgrelease(&pg->bg);
1473 }
1474 return;
1475
1476 bad_syntax:
1477 a_fail(a, "bad-syntax", "%s", cmd, "[OPTIONS] PEER", cmd, A_END);
1478 fail:
1479 if (pg) xfree(pg);
1480 return;
1481 }
1482
1483 static void acmd_ping(admin *a, unsigned ac, char *av[])
1484 { a_ping(a, ac, av, "ping", MISC_PING); }
1485 static void acmd_eping(admin *a, unsigned ac, char *av[])
1486 { a_ping(a, ac, av, "eping", MISC_EPING); }
1487
1488 /*----- Service commands --------------------------------------------------*/
1489
1490 static void acmd_svcclaim(admin *a, unsigned ac, char *av[])
1491 {
1492 admin_service *svc;
1493 unsigned f;
1494
1495 svc = sym_find(&a_svcs, av[0], -1, sizeof(*svc), &f);
1496 if (f) {
1497 if (versioncmp(av[1], svc->version) <= 0) {
1498 a_fail(a,
1499 "service-exists",
1500 "%s", SYM_NAME(svc),
1501 "%s", svc->version,
1502 A_END);
1503 return;
1504 }
1505 a_write(svc->prov, "SVCCLAIM", 0, "%s", av[0], "%s", av[1], A_END);
1506 a_svcunlink(svc);
1507 }
1508 svc->prov = a;
1509 svc->version = xstrdup(av[1]);
1510 svc->next = a->svcs;
1511 svc->prev = 0;
1512 if (a->svcs) a->svcs->prev = svc;
1513 a->svcs = svc;
1514 a_notify("SVCCLAIM", "%s", SYM_NAME(svc), "%s", svc->version, A_END);
1515 a_ok(a);
1516 }
1517
1518 static void acmd_svcrelease(admin *a, unsigned ac, char *av[])
1519 {
1520 admin_service *svc;
1521
1522 if ((svc = a_svcfind(a, av[0])) == 0)
1523 return;
1524 if (svc->prov != a) {
1525 a_fail(a, "not-service-provider", "%s", SYM_NAME(svc), A_END);
1526 return;
1527 }
1528 a_svcrelease(svc);
1529 a_ok(a);
1530 }
1531
1532 static void acmd_svcensure(admin *a, unsigned ac, char *av[])
1533 {
1534 admin_service *svc;
1535
1536 if ((svc = a_svcfind(a, av[0])) == 0)
1537 return;
1538 if (av[1] && versioncmp(svc->version, av[1]) < 0) {
1539 a_fail(a, "service-too-old",
1540 "%s", SYM_NAME(svc),
1541 "%s", svc->version,
1542 A_END);
1543 return;
1544 }
1545 a_ok(a);
1546 }
1547
1548 static void acmd_svcquery(admin *a, unsigned ac, char *av[])
1549 {
1550 admin_service *svc;
1551
1552 if ((svc = a_svcfind(a, av[0])) != 0) {
1553 a_info(a, "name=%s", SYM_NAME(svc), "version=%s", svc->version, A_END);
1554 a_ok(a);
1555 }
1556 }
1557
1558 static void acmd_svclist(admin *a, unsigned ac, char *av[])
1559 {
1560 admin_service *svc;
1561 sym_iter i;
1562
1563 for (sym_mkiter(&i, &a_svcs); (svc = sym_next(&i)) != 0; )
1564 a_info(a, "%s", SYM_NAME(svc), "%s", svc->version, A_END);
1565 a_ok(a);
1566 }
1567
1568 static void a_canceljob(admin_bgop *bg)
1569 {
1570 admin_svcop *svc = (admin_svcop *)bg;
1571
1572 a_write(svc->prov, "SVCCANCEL", 0, "%s", a_jobidencode(svc), A_END);
1573 a_jobdestroy(svc);
1574 }
1575
1576 static void acmd_svcsubmit(admin *a, unsigned ac, char *av[])
1577 {
1578 const char *tag = 0;
1579 admin_service *svc;
1580 admin_svcop *svcop;
1581 const char *ver = 0;
1582 dstr d = DSTR_INIT;
1583
1584 OPTIONS(ac, av, {
1585 OPTARG("-background", arg, { tag = arg; })
1586 OPTARG("-version", arg, { ver = arg; })
1587 });
1588 if (!*av) goto bad_syntax;
1589 if ((svc = a_svcfind(a, *av)) == 0) goto fail;
1590 if (ver && versioncmp(svc->version, ver) < 0) {
1591 a_fail(a, "service-too-old",
1592 "%s", SYM_NAME(svc),
1593 "%s", svc->version,
1594 A_END);
1595 goto fail;
1596 }
1597 if ((svcop = a_jobcreate(svc->prov)) == 0) {
1598 a_fail(a, "provider-overloaded", A_END);
1599 goto fail;
1600 }
1601 if (a_bgadd(a, &svcop->bg, tag, a_canceljob)) {
1602 a_jobdestroy(svcop);
1603 goto fail;
1604 }
1605 a_write(svc->prov, "SVCJOB", 0,
1606 "%s", a_jobidencode(svcop),
1607 "?TOKENS", av,
1608 A_END);
1609 goto done;
1610
1611 bad_syntax:
1612 a_fail(a, "bad-syntax", "svcsubmit", "[OPTIONS] SERVICE ARGS...", A_END);
1613 fail:
1614 done:
1615 dstr_destroy(&d);
1616 return;
1617 }
1618
1619 static void a_replycmd(admin *a, const char *status, int donep, char *av[])
1620 {
1621 admin_svcop *svc;
1622
1623 if ((svc = a_jobiddecode(&a->j, av[0])) == 0) {
1624 a_fail(a, "unknown-jobid", av[0], A_END);
1625 return;
1626 }
1627 a_write(svc->bg.a, status, svc->bg.tag, "?TOKENS", av + 1, A_END);
1628 if (donep) {
1629 a_jobdestroy(svc);
1630 a_bgrelease(&svc->bg);
1631 }
1632 a_ok(a);
1633 }
1634
1635 static void acmd_svcok(admin *a, unsigned ac, char *av[])
1636 { a_replycmd(a, "OK", 1, av); }
1637 static void acmd_svcinfo(admin *a, unsigned ac, char *av[])
1638 { a_replycmd(a, "INFO", 0, av); }
1639 static void acmd_svcfail(admin *a, unsigned ac, char *av[])
1640 { a_replycmd(a, "FAIL", 1, av); }
1641
1642 /*----- Administration commands -------------------------------------------*/
1643
1644 /* --- Miscellaneous commands --- */
1645
1646 /* --- @traceish@ --- *
1647 *
1648 * Arguments: @admin *a@ = connection to complain on
1649 * @unsigned ac@ = number of arguments
1650 * @char *av[]@ = vector of arguments
1651 * @const char *what@ = what we're messing with
1652 * @const trace_opt *tt@ = options table
1653 * @unsigned *ff@ = where the flags are
1654 *
1655 * Returns: Nonzero if anything changed.
1656 *
1657 * Use: Guts of trace-ish commands like `trace' and `watch'.
1658 */
1659
1660 static int traceish(admin *a, unsigned ac, char *av[],
1661 const char *what, const trace_opt *tt, unsigned *ff)
1662 {
1663 int ch = 0;
1664
1665 if (!ac || strcmp(av[0], "?") == 0) {
1666 const trace_opt *t;
1667 for (t = tt; t->ch; t++) {
1668 a_info(a, "*%c%c %s",
1669 t->ch, (*ff & t->f) == t->f ? '+' : ' ', t->help, A_END);
1670 }
1671 } else {
1672 unsigned sense = 1;
1673 unsigned f = *ff;
1674 const trace_opt *t;
1675 char *p = av[0];
1676
1677 while (*p) {
1678 switch (*p) {
1679 case '+': sense = 1; break;
1680 case '-': sense = 0; break;
1681 default:
1682 for (t = tt; t->ch; t++) {
1683 if (t->ch == *p) {
1684 if (sense) f |= t->f;
1685 else f &= ~t->f;
1686 goto tropt_ok;
1687 }
1688 }
1689 a_fail(a, "bad-%s-option", what, "%c", *p, A_END);
1690 return (0);
1691 tropt_ok:;
1692 break;
1693 }
1694 p++;
1695 }
1696 *ff = f;
1697 ch = 1;
1698 }
1699 a_ok(a);
1700 return (ch);
1701 }
1702
1703 #ifndef NTRACE
1704
1705 static void acmd_trace(admin *a, unsigned ac, char *av[])
1706 {
1707 if (traceish(a, ac, av, "trace", tr_opts, &tr_flags))
1708 trace_level(tr_flags);
1709 }
1710
1711 #endif
1712
1713 static void acmd_watch(admin *a, unsigned ac, char *av[])
1714 { traceish(a, ac, av, "watch", w_opts, &a->f); }
1715
1716 static void alertcmd(admin *a, unsigned f_and, unsigned f_eq,
1717 const char *status, char *av[])
1718 { a_alert(f_and, f_eq, status, "USER", "?TOKENS", av, A_END); a_ok(a); }
1719 static void acmd_notify(admin *a, unsigned ac, char *av[])
1720 { alertcmd(a, AF_NOTE, AF_NOTE, "NOTE", av); }
1721 static void acmd_warn(admin *a, unsigned ac, char *av[])
1722 { alertcmd(a, AF_WARN, AF_WARN, "WARN", av); }
1723
1724 static void acmd_port(admin *a, unsigned ac, char *av[])
1725 {
1726 int i;
1727
1728 if (ac) {
1729 for (i = 0; i < NADDRFAM; i++)
1730 if (mystrieq(av[0], aftab[i].name)) goto found;
1731 a_fail(a, "unknown-address-family", "%s", av[0], A_END);
1732 return;
1733 found:
1734 assert(udpsock[i].fd >= 0);
1735 } else {
1736 for (i = 0; i < NADDRFAM; i++)
1737 if (udpsock[i].fd >= 0) goto found;
1738 abort();
1739 }
1740 a_info(a, "%u", p_port(i), A_END);
1741 a_ok(a);
1742 }
1743
1744 static void acmd_daemon(admin *a, unsigned ac, char *av[])
1745 {
1746 if (flags & F_DAEMON)
1747 a_fail(a, "already-daemon", A_END);
1748 else {
1749 a_notify("DAEMON", A_END);
1750 if (a_stdin)
1751 a_destroy(a_stdin);
1752 if (daemonize())
1753 a_fail(a, "daemon-error", "?ERRNO", A_END);
1754 else {
1755 flags |= F_DAEMON;
1756 a_ok(a);
1757 }
1758 }
1759 }
1760
1761 static void acmd_jobs(admin *a, unsigned ac, char *av[])
1762 {
1763 admin_bgop *bg;
1764
1765 for (bg = a->bg; bg; bg = bg->next) {
1766 assert(bg->tag);
1767 a_info(a, "%s", bg->tag, A_END);
1768 }
1769 a_ok(a);
1770 }
1771
1772 static void acmd_bgcancel(admin *a, unsigned ac, char *av[])
1773 {
1774 admin_bgop *bg;
1775
1776 if ((bg = a_bgfind(a, av[0])) == 0)
1777 a_fail(a, "unknown-tag", "%s", av[0], A_END);
1778 else {
1779 bg->cancel(bg);
1780 a_bgrelease(bg);
1781 a_ok(a);
1782 }
1783 }
1784
1785 static void acmd_algs(admin *a, unsigned ac, char *av[])
1786 {
1787 peer *p;
1788 const kdata *kd;
1789 const dhgrp *g;
1790 const algswitch *algs;
1791
1792 if (!ac)
1793 kd = master;
1794 else {
1795 if ((p = a_findpeer(a, av[0])) == 0) return;
1796 kd = p->kx.kpriv;
1797 }
1798 g = kd->grp;
1799 algs = &kd->algs;
1800
1801 g->ops->grpinfo(g, a);
1802 a_info(a,
1803 "hash=%s", algs->h->name,
1804 "mgf=%s", algs->mgf->name,
1805 "hash-sz=%lu", (unsigned long)algs->h->hashsz,
1806 A_END);
1807 a_info(a,
1808 "bulk-transform=%s", algs->bulk->ops->name,
1809 "bulk-overhead=%lu",
1810 (unsigned long)algs->bulk->ops->overhead(algs->bulk),
1811 A_END);
1812 algs->bulk->ops->alginfo(algs->bulk, a);
1813 a_info(a,
1814 "cipher-data-limit=%lu",
1815 (unsigned long)algs->bulk->ops->expsz(algs->bulk),
1816 A_END);
1817 a_ok(a);
1818 }
1819
1820 static void acmd_list(admin *a, unsigned ac, char *av[])
1821 {
1822 FOREACH_PEER(p, { a_info(a, "%s", p_name(p), A_END); });
1823 a_ok(a);
1824 }
1825
1826 static void acmd_ifname(admin *a, unsigned ac, char *av[])
1827 {
1828 peer *p;
1829
1830 if ((p = a_findpeer(a, av[0])) != 0) {
1831 a_info(a, "%s", p_ifname(p), A_END);
1832 a_ok(a);
1833 }
1834 }
1835
1836 static void acmd_setifname(admin *a, unsigned ac, char *av[])
1837 {
1838 peer *p;
1839
1840 if ((p = a_findpeer(a, av[0])) != 0) {
1841 a_notify("NEWIFNAME", "?PEER", p, "%s", p_ifname(p), "%s", av[1], A_END);
1842 p_setifname(p, av[1]);
1843 a_ok(a);
1844 }
1845 }
1846
1847 static void acmd_getchal(admin *a, unsigned ac, char *av[])
1848 {
1849 buf b;
1850
1851 buf_init(&b, buf_i, PKBUFSZ);
1852 c_new(&b);
1853 a_info(a, "?B64", BBASE(&b), (size_t)BLEN(&b), A_END);
1854 a_ok(a);
1855 }
1856
1857 static void acmd_checkchal(admin *a, unsigned ac, char *av[])
1858 {
1859 codec *b64 = base64_class.decoder(CDCF_NOEQPAD);
1860 int err;
1861 buf b;
1862 dstr d = DSTR_INIT;
1863
1864 if ((err = b64->ops->code(b64, av[0], strlen(av[0]), &d)) != 0 ||
1865 (err = b64->ops->code(b64, 0, 0, &d)) != 0)
1866 a_fail(a, "bad-base64", "%s", codec_strerror(err), A_END);
1867 else {
1868 buf_init(&b, d.buf, d.len);
1869 if (c_check(&b) || BBAD(&b) || BLEFT(&b))
1870 a_fail(a, "invalid-challenge", A_END);
1871 else
1872 a_ok(a);
1873 }
1874 b64->ops->destroy(b64);
1875 dstr_destroy(&d);
1876 }
1877
1878 static void acmd_greet(admin *a, unsigned ac, char *av[])
1879 {
1880 peer *p;
1881 int err;
1882 codec *b64;
1883 dstr d = DSTR_INIT;
1884
1885 if ((p = a_findpeer(a, av[0])) == 0) return;
1886 b64 = base64_class.decoder(CDCF_NOEQPAD);
1887 if ((err = b64->ops->code(b64, av[1], strlen(av[1]), &d)) != 0 ||
1888 (err = b64->ops->code(b64, 0, 0, &d)) != 0)
1889 a_fail(a, "bad-base64", "%s", codec_strerror(err), A_END);
1890 else {
1891 p_greet(p, d.buf, d.len);
1892 a_ok(a);
1893 }
1894 b64->ops->destroy(b64);
1895 dstr_destroy(&d);
1896 }
1897
1898 static void acmd_addr(admin *a, unsigned ac, char *av[])
1899 {
1900 peer *p;
1901 const addr *ad;
1902
1903 if ((p = a_findpeer(a, av[0])) != 0) {
1904 ad = p_addr(p);
1905 a_info(a, "?ADDR", ad, A_END);
1906 a_ok(a);
1907 }
1908 }
1909
1910 static void acmd_peerinfo(admin *a, unsigned ac, char *av[])
1911 {
1912 peer *p;
1913 const peerspec *ps;
1914 const char *ptag;
1915
1916 if ((p = a_findpeer(a, av[0])) != 0) {
1917 ps = p_spec(p);
1918 a_info(a, "tunnel=%s", ps->tops->name, A_END);
1919 a_info(a, "key=%s", p_tag(p),
1920 "current-key=%s", p->kx.kpub->tag, A_END);
1921 if ((ptag = p_privtag(p)) == 0) ptag = "(default)";
1922 a_info(a, "private-key=%s", ptag,
1923 "current-private-key=%s", p->kx.kpriv->tag, A_END);
1924 a_info(a, "keepalive=%lu", ps->t_ka, A_END);
1925 a_info(a, "corked=%s", BOOL(p->kx.f&KXF_CORK),
1926 "mobile=%s", BOOL(ps->f&PSF_MOBILE),
1927 A_END);
1928 a_ok(a);
1929 }
1930 }
1931
1932 static void acmd_servinfo(admin *a, unsigned ac, char *av[])
1933 {
1934 a_info(a, "implementation=edgeware-tripe", A_END);
1935 a_info(a, "version=%s", VERSION, A_END);
1936 a_info(a, "daemon=%s", BOOL(flags & F_DAEMON), A_END);
1937 a_ok(a);
1938 }
1939
1940 static void acmd_stats(admin *a, unsigned ac, char *av[])
1941 {
1942 peer *p;
1943 stats *st;
1944
1945 if ((p = a_findpeer(a, av[0])) == 0)
1946 return;
1947
1948 st = p_stats(p);
1949 a_info(a, "start-time=%s", timestr(st->t_start), A_END);
1950 a_info(a, "last-packet-time=%s", timestr(st->t_last), A_END);
1951 a_info(a, "last-keyexch-time=%s", timestr(st->t_kx), A_END);
1952 a_info(a, "packets-in=%lu", st->n_in, "bytes-in=%lu", st->sz_in, A_END);
1953 a_info(a,
1954 "packets-out=%lu", st->n_out,
1955 "bytes-out=%lu", st->sz_out,
1956 A_END);
1957 a_info(a,
1958 "keyexch-packets-in=%lu", st->n_kxin,
1959 "keyexch-bytes-in=%lu", st->sz_kxin,
1960 A_END);
1961 a_info(a,
1962 "keyexch-packets-out=%lu", st->n_kxout,
1963 "keyexch-bytes-out=%lu", st->sz_kxout,
1964 A_END);
1965 a_info(a,
1966 "ip-packets-in=%lu", st->n_ipin,
1967 "ip-bytes-in=%lu", st->sz_ipin,
1968 A_END);
1969 a_info(a,
1970 "ip-packets-out=%lu", st->n_ipout,
1971 "ip-bytes-out=%lu", st->sz_ipout,
1972 A_END);
1973 a_info(a, "rejected-packets=%lu", st->n_reject, A_END);
1974 a_ok(a);
1975 }
1976
1977 static void acmd_kill(admin *a, unsigned ac, char *av[])
1978 {
1979 peer *p;
1980
1981 if ((p = a_findpeer(a, av[0])) != 0) {
1982 p_destroy(p);
1983 a_ok(a);
1984 }
1985 }
1986
1987 static void acmd_forcekx(admin *a, unsigned ac, char *av[])
1988 {
1989 peer *p;
1990
1991 if ((p = a_findpeer(a, av[0])) != 0) {
1992 kx_start(&p->kx, 1);
1993 a_ok(a);
1994 }
1995 }
1996
1997 static void acmd_reload(admin *a, unsigned ac, char *av[])
1998 { p_keyreload(); a_ok(a); }
1999
2000 static void acmd_quit(admin *a, unsigned ac, char *av[])
2001 {
2002 a_warn("SERVER", "quit", "admin-request", A_END);
2003 a_ok(a);
2004 a_quit();
2005 }
2006
2007 static void acmd_version(admin *a, unsigned ac, char *av[])
2008 {
2009 a_info(a, "%s", PACKAGE, "%s", VERSION, A_END);
2010 a_ok(a);
2011 }
2012
2013 static void acmd_tunnels(admin *a, unsigned ac, char *av[])
2014 {
2015 int i;
2016
2017 for (i = 0; tunnels[i]; i++)
2018 a_info(a, "%s", tunnels[i]->name, A_END);
2019 a_ok(a);
2020 }
2021
2022 /* --- The command table and help --- */
2023
2024 typedef struct acmd {
2025 const char *name;
2026 const char *help;
2027 unsigned argmin, argmax;
2028 void (*func)(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
2029 } acmd;
2030
2031 static void acmd_help(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
2032
2033 static const acmd acmdtab[] = {
2034 { "add", "[OPTIONS] PEER ADDR ...", 2, 0xffff, acmd_add },
2035 { "addr", "PEER", 1, 1, acmd_addr },
2036 { "algs", "[PEER]", 0, 1, acmd_algs },
2037 { "bgcancel", "TAG", 1, 1, acmd_bgcancel },
2038 { "checkchal", "CHAL", 1, 1, acmd_checkchal },
2039 { "daemon", 0, 0, 0, acmd_daemon },
2040 { "eping", "[OPTIONS] PEER", 1, 0xffff, acmd_eping },
2041 { "forcekx", "PEER", 1, 1, acmd_forcekx },
2042 { "getchal", 0, 0, 0, acmd_getchal },
2043 { "greet", "PEER CHAL", 2, 2, acmd_greet },
2044 { "help", 0, 0, 0, acmd_help },
2045 { "ifname", "PEER", 1, 1, acmd_ifname },
2046 { "jobs", 0, 0, 0, acmd_jobs },
2047 { "kill", "PEER", 1, 1, acmd_kill },
2048 { "list", 0, 0, 0, acmd_list },
2049 { "notify", "MESSAGE ...", 1, 0xffff, acmd_notify },
2050 { "peerinfo", "PEER", 1, 1, acmd_peerinfo },
2051 { "ping", "[OPTIONS] PEER", 1, 0xffff, acmd_ping },
2052 { "port", "[FAMILY]", 0, 1, acmd_port },
2053 { "quit", 0, 0, 0, acmd_quit },
2054 { "reload", 0, 0, 0, acmd_reload },
2055 { "servinfo", 0, 0, 0, acmd_servinfo },
2056 { "setifname", "PEER NEW-NAME", 2, 2, acmd_setifname },
2057 { "stats", "PEER", 1, 1, acmd_stats },
2058 { "svcclaim", "SERVICE VERSION", 2, 2, acmd_svcclaim },
2059 { "svcensure", "SERVICE [VERSION]", 1, 2, acmd_svcensure },
2060 { "svcfail", "JOBID TOKENS...", 1, 0xffff, acmd_svcfail },
2061 { "svcinfo", "JOBID TOKENS...", 1, 0xffff, acmd_svcinfo },
2062 { "svclist", 0, 0, 0, acmd_svclist },
2063 { "svcok", "JOBID", 1, 1, acmd_svcok },
2064 { "svcquery", "SERVICE", 1, 1, acmd_svcquery },
2065 { "svcrelease", "SERVICE", 1, 1, acmd_svcrelease },
2066 { "svcsubmit", "[OPTIONS] SERVICE TOKENS...",
2067 2, 0xffff, acmd_svcsubmit },
2068 #ifndef NTRACE
2069 { "trace", "[OPTIONS]", 0, 1, acmd_trace },
2070 #endif
2071 { "tunnels", 0, 0, 0, acmd_tunnels },
2072 { "version", 0, 0, 0, acmd_version },
2073 { "warn", "MESSAGE ...", 1, 0xffff, acmd_warn },
2074 { "watch", "[OPTIONS]", 0, 1, acmd_watch },
2075 { 0, 0, 0, 0, 0 }
2076 };
2077
2078 static void acmd_help(admin *a, unsigned ac, char *av[])
2079 {
2080 const acmd *c;
2081
2082 for (c = acmdtab; c->name; c++) {
2083 if (c->help)
2084 a_info(a, "%s", c->name, "*%s", c->help, A_END);
2085 else
2086 a_info(a, "%s", c->name, A_END);
2087 }
2088 a_ok(a);
2089 }
2090
2091 /*----- Connection handling -----------------------------------------------*/
2092
2093 /* --- @a_destroypending@ --- *
2094 *
2095 * Arguments: ---
2096 *
2097 * Returns: ---
2098 *
2099 * Use: Destroys pending admin connections at a safe time.
2100 */
2101
2102 static void a_destroypending(void)
2103 {
2104 admin *a, *aa, *head;
2105 admin_bgop *bg, *bbg;
2106 admin_service *svc, *ssvc;
2107
2108 /* --- Destroy connections marked as pending --- *
2109 *
2110 * Slightly messy. Killing clients may cause others to finally die. Make
2111 * sure that they can be put on the list without clobbering anything or
2112 * getting lost.
2113 */
2114
2115 while (a_dead) {
2116 head = a_dead;
2117 a_dead = 0;
2118 for (a = head; a; a = aa) {
2119 aa = a->next;
2120 assert(a->f & AF_DEAD);
2121
2122 /* --- Report what we're doing --- */
2123
2124 T( trace(T_ADMIN, "admin: completing destruction of connection %u",
2125 a->seq); )
2126
2127 /* --- If this is the foreground client then shut down --- */
2128
2129 if (a->f & AF_FOREGROUND) {
2130 T( trace(T_ADMIN, "admin: foreground client quit: shutting down"); )
2131 a_warn("SERVER", "quit", "foreground-eof", A_END);
2132 a_quit();
2133 }
2134
2135 /* --- Abort any background jobs in progress --- */
2136
2137 for (bg = a->bg; bg; bg = bbg) {
2138 bbg = bg->next;
2139 bg->cancel(bg);
2140 if (bg->tag) xfree(bg->tag);
2141 xfree(bg);
2142 }
2143
2144 /* --- Release services I hold, and abort pending jobs --- */
2145
2146 for (svc = a->svcs; svc; svc = ssvc) {
2147 ssvc = svc->next;
2148 a_svcrelease(svc);
2149 }
2150 a_jobtablefinal(&a->j);
2151
2152 /* --- Close file descriptors and selectory --- */
2153
2154 selbuf_destroy(&a->b);
2155 if (a->b.reader.fd != a->w.fd) close(a->b.reader.fd);
2156 close(a->w.fd);
2157 if (a_stdin == a) a_stdin = 0;
2158
2159 /* --- Done --- */
2160
2161 DESTROY(a);
2162 }
2163 }
2164 }
2165
2166 /* --- @a_destroy@ --- *
2167 *
2168 * Arguments: @admin *a@ = pointer to an admin block
2169 *
2170 * Returns: ---
2171 *
2172 * Use: Destroys an admin block. This requires a certain amount of
2173 * care.
2174 */
2175
2176 static void freequeue(oqueue *q)
2177 {
2178 obuf *o, *oo;
2179
2180 for (o = q->hd; o; o = oo) {
2181 oo = o->next;
2182 xfree(o);
2183 }
2184 q->hd = q->tl = 0;
2185 }
2186
2187 static void a_destroy(admin *a)
2188 {
2189 if (a->f & AF_DEAD)
2190 return;
2191
2192 if (a->next) a->next->prev = a->prev;
2193 if (a->prev) a->prev->next = a->next;
2194 else admins = a->next;
2195
2196 if (a->out.hd) sel_rmfile(&a->w);
2197 freequeue(&a->out);
2198
2199 a->f |= AF_DEAD;
2200 a->next = a_dead;
2201 a_dead = a;
2202
2203 T( trace(T_ADMIN, "admin: killing connection %u", a->seq); )
2204 }
2205
2206 /* --- @a_line@ --- *
2207 *
2208 * Arguments: @char *p@ = pointer to the line read
2209 * @size_t len@ = length of the line
2210 * @void *vp@ = pointer to my admin block
2211 *
2212 * Returns: ---
2213 *
2214 * Use: Handles a line of input.
2215 */
2216
2217 static void a_line(char *p, size_t len, void *vp)
2218 {
2219 admin *a = vp;
2220 const acmd *c;
2221 char *av[16 + 1];
2222 size_t ac;
2223
2224 QUICKRAND;
2225 if (a->f & AF_DEAD)
2226 return;
2227 if (!p) {
2228 if (!a->bg)
2229 a_destroy(a);
2230 else {
2231 a->f |= AF_CLOSE;
2232 selbuf_disable(&a->b);
2233 }
2234 return;
2235 }
2236 ac = str_qsplit(p, av, N(av) - 1, 0, STRF_QUOTE);
2237 if (!ac)
2238 return;
2239 av[ac] = 0;
2240 for (c = acmdtab; c->name; c++) {
2241 if (mystrieq(av[0], c->name)) {
2242 ac--;
2243 if (c->argmin > ac || ac > c->argmax) {
2244 if (!c->help)
2245 a_fail(a, "bad-syntax", "%s", c->name, "", A_END);
2246 else
2247 a_fail(a, "bad-syntax", "%s", c->name, "%s", c->help, A_END);
2248 } else
2249 c->func(a, ac, av + 1);
2250 return;
2251 }
2252 }
2253 a_fail(a, "unknown-command", "%s", av[0], A_END);
2254 }
2255
2256 /* --- @a_create@ --- *
2257 *
2258 * Arguments: @int fd_in, fd_out@ = file descriptors to use
2259 * @unsigned f@ = initial flags to set
2260 *
2261 * Returns: ---
2262 *
2263 * Use: Creates a new admin connection.
2264 */
2265
2266 void a_create(int fd_in, int fd_out, unsigned f)
2267 {
2268 admin *a = CREATE(admin);
2269
2270 T( static unsigned seq = 0;
2271 a->seq = seq++; )
2272 T( trace(T_ADMIN, "admin: accepted connection %u", a->seq); )
2273 a->bg = 0;
2274 a->ref = 0;
2275 a->svcs = 0;
2276 a_jobtableinit(&a->j);
2277 a->f = f;
2278 if (fd_in == STDIN_FILENO) a_stdin = a;
2279 fdflags(fd_in, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2280 if (fd_out != fd_in)
2281 fdflags(fd_out, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2282 selbuf_init(&a->b, &sel, fd_in, a_line, a);
2283 sel_initfile(&sel, &a->w, fd_out, SEL_WRITE, a_flush, a);
2284 a->out.hd = a->out.tl = 0;
2285 a->next = admins;
2286 a->prev = 0;
2287 if (admins) admins->prev = a;
2288 admins = a;
2289 }
2290
2291 /* --- @a_accept@ --- *
2292 *
2293 * Arguments: @int fd@ = file descriptor to accept
2294 * @unsigned mode@ = what to do
2295 * @void *v@ = uninteresting pointer
2296 *
2297 * Returns: ---
2298 *
2299 * Use: Accepts a new admin connection.
2300 */
2301
2302 static void a_accept(int fd, unsigned mode, void *v)
2303 {
2304 int nfd;
2305 struct sockaddr_un sun;
2306 socklen_t sz = sizeof(sun);
2307
2308 if ((nfd = accept(fd, (struct sockaddr *)&sun, &sz)) < 0) {
2309 if (errno != EINTR && errno != EAGAIN && errno != EWOULDBLOCK &&
2310 errno != ECONNABORTED && errno != EPROTO)
2311 a_warn("ADMIN", "accept-error", "?ERRNO", A_END);
2312 return;
2313 }
2314 a_create(nfd, nfd, 0);
2315 }
2316
2317 /* --- @a_preselect@ --- *
2318 *
2319 * Arguments: ---
2320 *
2321 * Returns: ---
2322 *
2323 * Use: Informs the admin module that we're about to select again,
2324 * and that it should do cleanup things it has delayed until a
2325 * `safe' time.
2326 */
2327
2328 void a_preselect(void) { if (a_dead) a_destroypending(); }
2329
2330 /* --- @a_daemon@ --- *
2331 *
2332 * Arguments: ---
2333 *
2334 * Returns: ---
2335 *
2336 * Use: Informs the admin module that it's a daemon.
2337 */
2338
2339 void a_daemon(void) { flags |= F_DAEMON; }
2340
2341 /* --- @a_init@ --- *
2342 *
2343 * Arguments: @const char *name@ = socket name to create
2344 * @uid_t u@ = user to own the socket
2345 * @gid_t g@ = group to own the socket
2346 * @mode_t m@ = permissions to set on the socket
2347 *
2348 * Returns: ---
2349 *
2350 * Use: Creates the admin listening socket.
2351 */
2352
2353 void a_init(const char *name, uid_t u, gid_t g, mode_t m)
2354 {
2355 int fd;
2356 int n = 5;
2357 struct sockaddr_un sun;
2358 struct sigaction sa;
2359 size_t sz;
2360 mode_t omask;
2361
2362 /* --- Create services table --- */
2363
2364 sym_create(&a_svcs);
2365
2366 /* --- Set up the socket address --- */
2367
2368 sz = strlen(name) + 1;
2369 if (sz > sizeof(sun.sun_path))
2370 die(EXIT_FAILURE, "socket name `%s' too long", name);
2371 BURN(sun);
2372 sun.sun_family = AF_UNIX;
2373 memcpy(sun.sun_path, name, sz);
2374 sz += offsetof(struct sockaddr_un, sun_path);
2375
2376 /* --- Attempt to bind to the socket --- */
2377
2378 omask = umask(0077);
2379 again:
2380 if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0)
2381 die(EXIT_FAILURE, "couldn't create socket: %s", strerror(errno));
2382 if (bind(fd, (struct sockaddr *)&sun, sz) < 0) {
2383 struct stat st;
2384 int e = errno;
2385 if (errno != EADDRINUSE) {
2386 die(EXIT_FAILURE, "couldn't bind to address `%s': %s",
2387 sun.sun_path, strerror(e));
2388 }
2389 if (!n)
2390 die(EXIT_FAILURE, "too many retries; giving up");
2391 n--;
2392 if (!connect(fd, (struct sockaddr *)&sun, sz)) {
2393 die(EXIT_FAILURE, "server already listening on admin socket `%s'",
2394 sun.sun_path);
2395 }
2396 if (errno != ECONNREFUSED)
2397 die(EXIT_FAILURE, "couldn't bind to address: %s", strerror(e));
2398 if (stat(sun.sun_path, &st)) {
2399 die(EXIT_FAILURE, "couldn't stat `%s': %s",
2400 sun.sun_path, strerror(errno));
2401 }
2402 if (!S_ISSOCK(st.st_mode))
2403 die(EXIT_FAILURE, "object `%s' isn't a socket", sun.sun_path);
2404 T( trace(T_ADMIN, "admin: stale socket found; removing it"); )
2405 unlink(sun.sun_path);
2406 close(fd);
2407 goto again;
2408 }
2409 if (chown(sun.sun_path, u, g)) {
2410 die(EXIT_FAILURE, "failed to set socket owner: %s",
2411 strerror(errno));
2412 }
2413 if (chmod(sun.sun_path, m)) {
2414 die(EXIT_FAILURE, "failed to set socket permissions: %s",
2415 strerror(errno));
2416 }
2417 umask(omask);
2418 fdflags(fd, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2419 if (listen(fd, 5))
2420 die(EXIT_FAILURE, "couldn't listen on socket: %s", strerror(errno));
2421
2422 /* --- Listen to the socket --- */
2423
2424 sel_initfile(&sel, &sock, fd, SEL_READ, a_accept, 0);
2425 sel_addfile(&sock);
2426 sockname = name;
2427 bres_init(&sel);
2428 T( trace_custom(a_trace, 0);
2429 trace(T_ADMIN, "admin: enabled custom tracing"); )
2430 flags |= F_INIT;
2431
2432 /* --- Set up signal handlers --- */
2433
2434 sig_add(&s_term, SIGTERM, a_sigdie, 0);
2435 sig_add(&s_hup, SIGHUP, a_sighup, 0);
2436 sigaction(SIGINT, 0, &sa);
2437 if (sa.sa_handler != SIG_IGN)
2438 sig_add(&s_int, SIGINT, a_sigdie, 0);
2439 }
2440
2441 /*----- That's all, folks -------------------------------------------------*/