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