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