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 dhgrp *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->grp;
1733   algs = &kd->algs;
1734
1735   g->ops->grpinfo(g, a);
1736   a_info(a,
1737          "hash=%s", algs->h->name,
1738          "mgf=%s", algs->mgf->name,
1739          "hash-sz=%lu", (unsigned long)algs->h->hashsz,
1740          A_END);
1741   a_info(a,
1742          "bulk-transform=%s", algs->bulk->ops->name,
1743          "bulk-overhead=%lu",
1744          (unsigned long)algs->bulk->ops->overhead(algs->bulk),
1745          A_END);
1746   algs->bulk->ops->alginfo(algs->bulk, a);
1747   a_info(a,
1748          "cipher-data-limit=%lu",
1749          (unsigned long)algs->bulk->ops->expsz(algs->bulk),
1750          A_END);
1751   a_ok(a);
1752 }
1753
1754 static void acmd_list(admin *a, unsigned ac, char *av[])
1755 {
1756   FOREACH_PEER(p, {  a_info(a, "%s", p_name(p), A_END); });
1757   a_ok(a);
1758 }
1759
1760 static void acmd_ifname(admin *a, unsigned ac, char *av[])
1761 {
1762   peer *p;
1763
1764   if ((p = a_findpeer(a, av[0])) != 0) {
1765     a_info(a, "%s", p_ifname(p), A_END);
1766     a_ok(a);
1767   }
1768 }
1769
1770 static void acmd_setifname(admin *a, unsigned ac, char *av[])
1771 {
1772   peer *p;
1773
1774   if ((p = a_findpeer(a, av[0])) != 0) {
1775     a_notify("NEWIFNAME", "?PEER", p, "%s", p_ifname(p), "%s", av[1], A_END);
1776     p_setifname(p, av[1]);
1777     a_ok(a);
1778   }
1779 }
1780
1781 static void acmd_getchal(admin *a, unsigned ac, char *av[])
1782 {
1783   buf b;
1784
1785   buf_init(&b, buf_i, PKBUFSZ);
1786   c_new(&b);
1787   a_info(a, "?B64", BBASE(&b), (size_t)BLEN(&b), A_END);
1788   a_ok(a);
1789 }
1790
1791 static void acmd_checkchal(admin *a, unsigned ac, char *av[])
1792 {
1793   base64_ctx b64;
1794   buf b;
1795   dstr d = DSTR_INIT;
1796
1797   base64_init(&b64);
1798   base64_decode(&b64, av[0], strlen(av[0]), &d);
1799   base64_decode(&b64, 0, 0, &d);
1800   buf_init(&b, d.buf, d.len);
1801   if (c_check(&b) || BBAD(&b) || BLEFT(&b))
1802     a_fail(a, "invalid-challenge", A_END);
1803   else
1804     a_ok(a);
1805   dstr_destroy(&d);
1806 }
1807
1808 static void acmd_greet(admin *a, unsigned ac, char *av[])
1809 {
1810   peer *p;
1811   base64_ctx b64;
1812   dstr d = DSTR_INIT;
1813
1814   if ((p = a_findpeer(a, av[0])) != 0) {
1815     base64_init(&b64);
1816     base64_decode(&b64, av[1], strlen(av[1]), &d);
1817     base64_decode(&b64, 0, 0, &d);
1818     p_greet(p, d.buf, d.len);
1819     dstr_destroy(&d);
1820     a_ok(a);
1821   }
1822 }
1823
1824 static void acmd_addr(admin *a, unsigned ac, char *av[])
1825 {
1826   peer *p;
1827   const addr *ad;
1828
1829   if ((p = a_findpeer(a, av[0])) != 0) {
1830     ad = p_addr(p);
1831     assert(ad->sa.sa_family == AF_INET);
1832     a_info(a, "?ADDR", ad, A_END);
1833     a_ok(a);
1834   }
1835 }
1836
1837 static void acmd_peerinfo(admin *a, unsigned ac, char *av[])
1838 {
1839   peer *p;
1840   const peerspec *ps;
1841   const char *ptag;
1842
1843   if ((p = a_findpeer(a, av[0])) != 0) {
1844     ps = p_spec(p);
1845     a_info(a, "tunnel=%s", ps->tops->name, A_END);
1846     a_info(a, "key=%s", p_tag(p),
1847            "current-key=%s", p->kx.kpub->tag, A_END);
1848     if ((ptag = p_privtag(p)) == 0) ptag = "(default)";
1849     a_info(a, "private-key=%s", ptag,
1850            "current-private-key=%s", p->kx.kpriv->tag, A_END);
1851     a_info(a, "keepalive=%lu", ps->t_ka, A_END);
1852     a_ok(a);
1853   }
1854 }
1855
1856 static void acmd_servinfo(admin *a, unsigned ac, char *av[])
1857 {
1858   a_info(a, "implementation=edgeware-tripe", A_END);
1859   a_info(a, "version=%s", VERSION, A_END);
1860   a_info(a, "daemon=%s", BOOL(flags & F_DAEMON), A_END);
1861   a_ok(a);
1862 }
1863
1864 static void acmd_stats(admin *a, unsigned ac, char *av[])
1865 {
1866   peer *p;
1867   stats *st;
1868
1869   if ((p = a_findpeer(a, av[0])) == 0)
1870     return;
1871
1872   st = p_stats(p);
1873   a_info(a, "start-time=%s", timestr(st->t_start), A_END);
1874   a_info(a, "last-packet-time=%s", timestr(st->t_last), A_END);
1875   a_info(a, "last-keyexch-time=%s", timestr(st->t_kx), A_END);
1876   a_info(a, "packets-in=%lu", st->n_in, "bytes-in=%lu", st->sz_in, A_END);
1877   a_info(a,
1878          "packets-out=%lu", st->n_out,
1879          "bytes-out=%lu", st->sz_out,
1880          A_END);
1881   a_info(a,
1882          "keyexch-packets-in=%lu", st->n_kxin,
1883          "keyexch-bytes-in=%lu", st->sz_kxin,
1884          A_END);
1885   a_info(a,
1886          "keyexch-packets-out=%lu", st->n_kxout,
1887          "keyexch-bytes-out=%lu", st->sz_kxout,
1888          A_END);
1889   a_info(a,
1890          "ip-packets-in=%lu", st->n_ipin,
1891          "ip-bytes-in=%lu", st->sz_ipin,
1892          A_END);
1893   a_info(a,
1894          "ip-packets-out=%lu", st->n_ipout,
1895          "ip-bytes-out=%lu", st->sz_ipout,
1896          A_END);
1897   a_info(a, "rejected-packets=%lu", st->n_reject, A_END);
1898   a_ok(a);
1899 }
1900
1901 static void acmd_kill(admin *a, unsigned ac, char *av[])
1902 {
1903   peer *p;
1904
1905   if ((p = a_findpeer(a, av[0])) != 0) {
1906     p_destroy(p);
1907     a_ok(a);
1908   }
1909 }
1910
1911 static void acmd_forcekx(admin *a, unsigned ac, char *av[])
1912 {
1913   peer *p;
1914
1915   if ((p = a_findpeer(a, av[0])) != 0) {
1916     kx_start(&p->kx, 1);
1917     a_ok(a);
1918   }
1919 }
1920
1921 static void acmd_reload(admin *a, unsigned ac, char *av[])
1922   { p_keyreload(); a_ok(a); }
1923
1924 static void acmd_quit(admin *a, unsigned ac, char *av[])
1925 {
1926   a_warn("SERVER", "quit", "admin-request", A_END);
1927   a_ok(a);
1928   a_quit();
1929 }
1930
1931 static void acmd_version(admin *a, unsigned ac, char *av[])
1932 {
1933   a_info(a, "%s", PACKAGE, "%s", VERSION, A_END);
1934   a_ok(a);
1935 }
1936
1937 static void acmd_tunnels(admin *a, unsigned ac, char *av[])
1938 {
1939   int i;
1940
1941   for (i = 0; tunnels[i]; i++)
1942     a_info(a, "%s", tunnels[i]->name, A_END);
1943   a_ok(a);
1944 }
1945
1946 /* --- The command table and help --- */
1947
1948 typedef struct acmd {
1949   const char *name;
1950   const char *help;
1951   unsigned argmin, argmax;
1952   void (*func)(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1953 } acmd;
1954
1955 static void acmd_help(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1956
1957 static const acmd acmdtab[] = {
1958   { "add",      "[OPTIONS] PEER ADDR ...", 2,   0xffff, acmd_add },
1959   { "addr",     "PEER",                 1,      1,      acmd_addr },
1960   { "algs",     "[PEER]",               0,      1,      acmd_algs },
1961   { "bgcancel", "TAG",                  1,      1,      acmd_bgcancel },
1962   { "checkchal", "CHAL",                1,      1,      acmd_checkchal },
1963   { "daemon",   0,                      0,      0,      acmd_daemon },
1964   { "eping",    "[OPTIONS] PEER",       1,      0xffff, acmd_eping },
1965   { "forcekx",  "PEER",                 1,      1,      acmd_forcekx },
1966   { "getchal",  0,                      0,      0,      acmd_getchal },
1967   { "greet",    "PEER CHAL",            2,      2,      acmd_greet },
1968   { "help",     0,                      0,      0,      acmd_help },
1969   { "ifname",   "PEER",                 1,      1,      acmd_ifname },
1970   { "jobs",     0,                      0,      0,      acmd_jobs },
1971   { "kill",     "PEER",                 1,      1,      acmd_kill },
1972   { "list",     0,                      0,      0,      acmd_list },
1973   { "notify",   "MESSAGE ...",          1,      0xffff, acmd_notify },
1974   { "peerinfo", "PEER",                 1,      1,      acmd_peerinfo },
1975   { "ping",     "[OPTIONS] PEER",       1,      0xffff, acmd_ping },
1976   { "port",     0,                      0,      0,      acmd_port },
1977   { "quit",     0,                      0,      0,      acmd_quit },
1978   { "reload",   0,                      0,      0,      acmd_reload },
1979   { "servinfo", 0,                      0,      0,      acmd_servinfo },
1980   { "setifname", "PEER NEW-NAME",       2,      2,      acmd_setifname },
1981   { "svcclaim", "SERVICE VERSION",      2,      2,      acmd_svcclaim },
1982   { "svcensure", "SERVICE [VERSION]",   1,      2,      acmd_svcensure },
1983   { "svcfail",  "JOBID TOKENS...",      1,      0xffff, acmd_svcfail },
1984   { "svcinfo",  "JOBID TOKENS...",      1,      0xffff, acmd_svcinfo },
1985   { "svclist",  0,                      0,      0,      acmd_svclist },
1986   { "svcok",    "JOBID",                1,      1,      acmd_svcok },
1987   { "svcquery", "SERVICE",              1,      1,      acmd_svcquery },
1988   { "svcrelease", "SERVICE",            1,      1,      acmd_svcrelease },
1989   { "svcsubmit", "[OPTIONS] SERVICE TOKENS...",
1990                                         2,      0xffff, acmd_svcsubmit },
1991   { "stats",    "PEER",                 1,      1,      acmd_stats },
1992 #ifndef NTRACE
1993   { "trace",    "[OPTIONS]",            0,      1,      acmd_trace },
1994 #endif
1995   { "tunnels",  0,                      0,      0,      acmd_tunnels },
1996   { "version",  0,                      0,      0,      acmd_version },
1997   { "warn",     "MESSAGE ...",          1,      0xffff, acmd_warn },
1998   { "watch",    "[OPTIONS]",            0,      1,      acmd_watch },
1999   { 0,          0,                      0,      0,      0 }
2000 };
2001
2002 static void acmd_help(admin *a, unsigned ac, char *av[])
2003 {
2004   const acmd *c;
2005
2006   for (c = acmdtab; c->name; c++) {
2007     if (c->help)
2008       a_info(a, "%s", c->name, "*%s", c->help, A_END);
2009     else
2010       a_info(a, "%s", c->name, A_END);
2011   }
2012   a_ok(a);
2013 }
2014
2015 /*----- Connection handling -----------------------------------------------*/
2016
2017 /* --- @a_destroypending@ --- *
2018  *
2019  * Arguments:   ---
2020  *
2021  * Returns:     ---
2022  *
2023  * Use:         Destroys pending admin connections at a safe time.
2024  */
2025
2026 static void a_destroypending(void)
2027 {
2028   admin *a, *aa, *head;
2029   admin_bgop *bg, *bbg;
2030   admin_service *svc, *ssvc;
2031
2032   /* --- Destroy connections marked as pending --- *
2033    *
2034    * Slightly messy.  Killing clients may cause others to finally die.  Make
2035    * sure that they can be put on the list without clobbering anything or
2036    * getting lost.
2037    */
2038
2039   while (a_dead) {
2040     head = a_dead;
2041     a_dead = 0;
2042     for (a = head; a; a = aa) {
2043       aa = a->next;
2044       assert(a->f & AF_DEAD);
2045
2046       /* --- Report what we're doing --- */
2047
2048       T( trace(T_ADMIN, "admin: completing destruction of connection %u",
2049                a->seq); )
2050
2051       /* --- If this is the foreground client then shut down --- */
2052
2053       if (a->f & AF_FOREGROUND) {
2054         T( trace(T_ADMIN, "admin: foreground client quit: shutting down"); )
2055         a_warn("SERVER", "quit", "foreground-eof", A_END);
2056         a_quit();
2057       }
2058
2059       /* --- Abort any background jobs in progress --- */
2060
2061       for (bg = a->bg; bg; bg = bbg) {
2062         bbg = bg->next;
2063         bg->cancel(bg);
2064         if (bg->tag) xfree(bg->tag);
2065         xfree(bg);
2066       }
2067
2068       /* --- Release services I hold, and abort pending jobs --- */
2069
2070       for (svc = a->svcs; svc; svc = ssvc) {
2071         ssvc = svc->next;
2072         a_svcrelease(svc);
2073       }
2074       a_jobtablefinal(&a->j);
2075
2076       /* --- Close file descriptors and selectory --- */
2077
2078       selbuf_destroy(&a->b);
2079       if (a->b.reader.fd != a->w.fd) close(a->b.reader.fd);
2080       close(a->w.fd);
2081       if (a_stdin == a) a_stdin = 0;
2082
2083       /* --- Done --- */
2084
2085       DESTROY(a);
2086     }
2087   }
2088 }
2089
2090 /* --- @a_destroy@ --- *
2091  *
2092  * Arguments:   @admin *a@ = pointer to an admin block
2093  *
2094  * Returns:     ---
2095  *
2096  * Use:         Destroys an admin block.  This requires a certain amount of
2097  *              care.
2098  */
2099
2100 static void freequeue(oqueue *q)
2101 {
2102   obuf *o, *oo;
2103
2104   for (o = q->hd; o; o = oo) {
2105     oo = o->next;
2106     xfree(o);
2107   }
2108   q->hd = q->tl = 0;
2109 }
2110
2111 static void a_destroy(admin *a)
2112 {
2113   if (a->f & AF_DEAD)
2114     return;
2115
2116   if (a->next) a->next->prev = a->prev;
2117   if (a->prev) a->prev->next = a->next;
2118   else admins = a->next;
2119
2120   if (a->out.hd) sel_rmfile(&a->w);
2121   freequeue(&a->out);
2122
2123   a->f |= AF_DEAD;
2124   a->next = a_dead;
2125   a_dead = a;
2126
2127   T( trace(T_ADMIN, "admin: killing connection %u", a->seq); )
2128 }
2129
2130 /* --- @a_line@ --- *
2131  *
2132  * Arguments:   @char *p@ = pointer to the line read
2133  *              @size_t len@ = length of the line
2134  *              @void *vp@ = pointer to my admin block
2135  *
2136  * Returns:     ---
2137  *
2138  * Use:         Handles a line of input.
2139  */
2140
2141 static void a_line(char *p, size_t len, void *vp)
2142 {
2143   admin *a = vp;
2144   const acmd *c;
2145   char *av[16 + 1];
2146   size_t ac;
2147
2148   QUICKRAND;
2149   if (a->f & AF_DEAD)
2150     return;
2151   if (!p) {
2152     if (!a->bg)
2153       a_destroy(a);
2154     else {
2155       a->f |= AF_CLOSE;
2156       selbuf_disable(&a->b);
2157     }
2158     return;
2159   }
2160   ac = str_qsplit(p, av, N(av) - 1, 0, STRF_QUOTE);
2161   if (!ac)
2162     return;
2163   av[ac] = 0;
2164   for (c = acmdtab; c->name; c++) {
2165     if (mystrieq(av[0], c->name)) {
2166       ac--;
2167       if (c->argmin > ac || ac > c->argmax) {
2168         if (!c->help)
2169           a_fail(a, "bad-syntax", "%s", c->name, "", A_END);
2170         else
2171           a_fail(a, "bad-syntax", "%s", c->name, "%s", c->help, A_END);
2172       } else
2173         c->func(a, ac, av + 1);
2174       return;
2175     }
2176   }
2177   a_fail(a, "unknown-command", "%s", av[0], A_END);
2178 }
2179
2180 /* --- @a_create@ --- *
2181  *
2182  * Arguments:   @int fd_in, fd_out@ = file descriptors to use
2183  *              @unsigned f@ = initial flags to set
2184  *
2185  * Returns:     ---
2186  *
2187  * Use:         Creates a new admin connection.
2188  */
2189
2190 void a_create(int fd_in, int fd_out, unsigned f)
2191 {
2192   admin *a = CREATE(admin);
2193
2194   T( static unsigned seq = 0;
2195      a->seq = seq++; )
2196   T( trace(T_ADMIN, "admin: accepted connection %u", a->seq); )
2197   a->bg = 0;
2198   a->ref = 0;
2199   a->svcs = 0;
2200   a_jobtableinit(&a->j);
2201   a->f = f;
2202   if (fd_in == STDIN_FILENO) a_stdin = a;
2203   fdflags(fd_in, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2204   if (fd_out != fd_in)
2205     fdflags(fd_out, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2206   selbuf_init(&a->b, &sel, fd_in, a_line, a);
2207   sel_initfile(&sel, &a->w, fd_out, SEL_WRITE, a_flush, a);
2208   a->out.hd = a->out.tl = 0;
2209   a->next = admins;
2210   a->prev = 0;
2211   if (admins) admins->prev = a;
2212   admins = a;
2213 }
2214
2215 /* --- @a_accept@ --- *
2216  *
2217  * Arguments:   @int fd@ = file descriptor to accept
2218  *              @unsigned mode@ = what to do
2219  *              @void *v@ = uninteresting pointer
2220  *
2221  * Returns:     ---
2222  *
2223  * Use:         Accepts a new admin connection.
2224  */
2225
2226 static void a_accept(int fd, unsigned mode, void *v)
2227 {
2228   int nfd;
2229   struct sockaddr_un sun;
2230   socklen_t sz = sizeof(sun);
2231
2232   if ((nfd = accept(fd, (struct sockaddr *)&sun, &sz)) < 0) {
2233     if (errno != EINTR && errno != EAGAIN && errno != EWOULDBLOCK &&
2234         errno != ECONNABORTED && errno != EPROTO)
2235       a_warn("ADMIN", "accept-error", "?ERRNO", A_END);
2236     return;
2237   }
2238   a_create(nfd, nfd, 0);
2239 }
2240
2241 /* --- @a_preselect@ --- *
2242  *
2243  * Arguments:   ---
2244  *
2245  * Returns:     ---
2246  *
2247  * Use:         Informs the admin module that we're about to select again,
2248  *              and that it should do cleanup things it has delayed until a
2249  *              `safe' time.
2250  */
2251
2252 void a_preselect(void) { if (a_dead) a_destroypending(); }
2253
2254 /* --- @a_daemon@ --- *
2255  *
2256  * Arguments:   ---
2257  *
2258  * Returns:     ---
2259  *
2260  * Use:         Informs the admin module that it's a daemon.
2261  */
2262
2263 void a_daemon(void) { flags |= F_DAEMON; }
2264
2265 /* --- @a_init@ --- *
2266  *
2267  * Arguments:   @const char *name@ = socket name to create
2268  *              @uid_t u@ = user to own the socket
2269  *              @gid_t g@ = group to own the socket
2270  *              @mode_t m@ = permissions to set on the socket
2271  *
2272  * Returns:     ---
2273  *
2274  * Use:         Creates the admin listening socket.
2275  */
2276
2277 void a_init(const char *name, uid_t u, gid_t g, mode_t m)
2278 {
2279   int fd;
2280   int n = 5;
2281   struct sockaddr_un sun;
2282   struct sigaction sa;
2283   size_t sz;
2284   mode_t omask;
2285
2286   /* --- Create services table --- */
2287
2288   sym_create(&a_svcs);
2289
2290   /* --- Set up the socket address --- */
2291
2292   sz = strlen(name) + 1;
2293   if (sz > sizeof(sun.sun_path))
2294     die(EXIT_FAILURE, "socket name `%s' too long", name);
2295   BURN(sun);
2296   sun.sun_family = AF_UNIX;
2297   memcpy(sun.sun_path, name, sz);
2298   sz += offsetof(struct sockaddr_un, sun_path);
2299
2300   /* --- Attempt to bind to the socket --- */
2301
2302   omask = umask(0077);
2303 again:
2304   if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0)
2305     die(EXIT_FAILURE, "couldn't create socket: %s", strerror(errno));
2306   if (bind(fd, (struct sockaddr *)&sun, sz) < 0) {
2307     struct stat st;
2308     int e = errno;
2309     if (errno != EADDRINUSE) {
2310       die(EXIT_FAILURE, "couldn't bind to address `%s': %s",
2311           sun.sun_path, strerror(e));
2312     }
2313     if (!n)
2314       die(EXIT_FAILURE, "too many retries; giving up");
2315     n--;
2316     if (!connect(fd, (struct sockaddr *)&sun, sz)) {
2317       die(EXIT_FAILURE, "server already listening on admin socket `%s'",
2318           sun.sun_path);
2319     }
2320     if (errno != ECONNREFUSED)
2321       die(EXIT_FAILURE, "couldn't bind to address: %s", strerror(e));
2322     if (stat(sun.sun_path, &st)) {
2323       die(EXIT_FAILURE, "couldn't stat `%s': %s",
2324           sun.sun_path, strerror(errno));
2325     }
2326     if (!S_ISSOCK(st.st_mode))
2327       die(EXIT_FAILURE, "object `%s' isn't a socket", sun.sun_path);
2328     T( trace(T_ADMIN, "admin: stale socket found; removing it"); )
2329     unlink(sun.sun_path);
2330     close(fd);
2331     goto again;
2332   }
2333   if (chown(sun.sun_path, u, g)) {
2334     die(EXIT_FAILURE, "failed to set socket owner: %s",
2335         strerror(errno));
2336   }
2337   if (chmod(sun.sun_path, m)) {
2338     die(EXIT_FAILURE, "failed to set socket permissions: %s",
2339         strerror(errno));
2340   }
2341   umask(omask);
2342   fdflags(fd, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2343   if (listen(fd, 5))
2344     die(EXIT_FAILURE, "couldn't listen on socket: %s", strerror(errno));
2345
2346   /* --- Listen to the socket --- */
2347
2348   sel_initfile(&sel, &sock, fd, SEL_READ, a_accept, 0);
2349   sel_addfile(&sock);
2350   sockname = name;
2351   bres_init(&sel);
2352   T( trace_custom(a_trace, 0);
2353      trace(T_ADMIN, "admin: enabled custom tracing"); )
2354   flags |= F_INIT;
2355
2356   /* --- Set up signal handlers --- */
2357
2358   sig_add(&s_term, SIGTERM, a_sigdie, 0);
2359   sig_add(&s_hup, SIGHUP, a_sighup, 0);
2360   signal(SIGPIPE, SIG_IGN);
2361   sigaction(SIGINT, 0, &sa);
2362   if (sa.sa_handler != SIG_IGN)
2363     sig_add(&s_int, SIGINT, a_sigdie, 0);
2364 }
2365
2366 /*----- That's all, folks -------------------------------------------------*/