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