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