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

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