chiark - git - mdw - tripe/blob - server/admin.c

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