3 * Admin interface for configuration
5 * (c) 2001 Straylight/Edgeware
8 /*----- Licensing notice --------------------------------------------------*
10 * This file is part of Trivial IP Encryption (TrIPE).
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.
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.
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.
27 /*----- Header files ------------------------------------------------------*/
31 /*----- Global variables --------------------------------------------------*/
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 { 'p', T_PACKET, "packet contents" },
44 { 'c', T_CRYPTO, "crypto details" },
45 { 'A', T_ALL, "all of the above" },
49 unsigned tr_flags = 0;
52 static const trace_opt w_opts[] = {
53 { 't', AF_TRACE, "trace messages" },
54 { 'n', AF_NOTE, "asynchronous notifications" },
55 { 'w', AF_WARN, "warnings" },
56 { 'A', AF_ALLMSGS, "all of the above" },
60 /*----- Static variables --------------------------------------------------*/
65 static const char *sockname;
66 static sym_table a_svcs;
67 static unsigned flags = 0;
68 static admin *a_stdin = 0;
69 static sig s_term, s_int, s_hup;
74 #define T_RESOLVE SEC(30)
77 static void a_destroy(admin */*a*/);
79 #define BOOL(x) ((x) ? "t" : "nil")
81 /*----- Output functions --------------------------------------------------*/
83 /* --- @trywrite@ --- *
85 * Arguments: @admin *a@ = pointer to an admin block
86 * @const char *p@ = pointer to buffer to write
87 * @size_t sz@ = size of data to write
89 * Returns: The number of bytes written, or less than zero on error.
91 * Use: Attempts to write data to a client.
94 static ssize_t trywrite(admin *a, const char *p, size_t sz)
101 n = write(a->w.fd, p, sz);
111 if (errno != EAGAIN && errno != EWOULDBLOCK) {
113 a_warn("ADMIN", "client-write-error", "?ERRNO", A_END);
120 /* --- @doqueue@ -- *
122 * Arguments: @oqueue *q@ = pointer to output queue
123 * @const char *p@ = pointer to buffer to write
124 * @size_t sz@ = size of buffer
126 * Returns: Nonzero if the queue was previously empty.
128 * Use: Queues data to be written later.
131 static int doqueue(oqueue *q, const char *p, size_t sz)
140 else if (o->p_in < o->buf + OBUFSZ)
144 o = xmalloc(sizeof(obuf));
146 o->p_in = o->p_out = o->buf;
154 n = o->buf + OBUFSZ - o->p_in;
157 memcpy(o->p_in, p, n);
166 /* --- @dosend@ --- *
168 * Arguemnts: @admin *a@ = pointer to an admin block
169 * @const char *p@ = pointer to buffer to write
170 * @size_t sz@ = size of data to write
174 * Use: Sends data to an admin client.
177 static void dosend(admin *a, const char *p, size_t sz)
184 if ((n = trywrite(a, p, sz)) < 0)
191 if (doqueue(&a->out, p, sz))
195 /* --- @a_flush@ --- *
197 * Arguments: @int fd@ = file descriptor
198 * @unsigned mode@ = what's happening
199 * @void *v@ = pointer to my admin block
203 * Use: Flushes buffers when a client is ready to read again.
206 static void a_flush(int fd, unsigned mode, void *v)
214 if ((n = trywrite(a, o->p_out, o->p_in - o->p_out)) < 0)
217 if (o->p_in < o->p_out)
230 /*----- Utility functions -------------------------------------------------*/
232 /* --- @a_vformat@ --- *
234 * Arguments: @dstr *d@ = where to leave the formatted message
235 * @const char *fmt@ = pointer to format string
236 * @va_list ap@ = arguments in list
240 * Use: Main message token formatting driver.
243 static void a_vformat(dstr *d, const char *fmt, va_list ap)
250 dstr_vputf(d, fmt + 1, &ap);
251 } else if (*fmt == '?') {
252 if (strcmp(fmt, "?ADDR") == 0) {
253 const addr *a = va_arg(ap, const addr *);
254 switch (a->sa.sa_family) {
256 u_quotify(d, "INET");
257 u_quotify(d, inet_ntoa(a->sin.sin_addr));
258 dstr_putf(d, " %u", (unsigned)ntohs(a->sin.sin_port));
263 } else if (strcmp(fmt, "?B64") == 0) {
264 const octet *p = va_arg(ap, const octet *);
265 size_t n = va_arg(ap, size_t);
271 base64_encode(&b64, p, n, d);
272 base64_encode(&b64, 0, 0, d);
273 while (d->len && d->buf[d->len - 1] == '=') d->len--;
274 } else if (strcmp(fmt, "?TOKENS") == 0) {
275 const char *const *av = va_arg(ap, const char *const *);
276 while (*av) u_quotify(d, *av++);
277 } else if (strcmp(fmt, "?PEER") == 0)
278 u_quotify(d, p_name(va_arg(ap, peer *)));
279 else if (strcmp(fmt, "?ERRNO") == 0) {
280 dstr_putf(d, " E%d", errno);
281 u_quotify(d, strerror(errno));
285 if (*fmt == '!') fmt++;
287 dstr_vputf(&dd, fmt, &ap);
288 u_quotify(d, dd.buf);
290 fmt = va_arg(ap, const char *);
296 /* --- @a_write@, @a_vwrite@ --- *
298 * Arguments: @admin *a@ = admin connection to write to
299 * @const char *status@ = status code to report
300 * @const char *tag@ = tag string, or null
301 * @const char *fmt@ = pointer to format string
302 * @va_list ap@ = arguments in list
303 * @...@ = other arguments
307 * Use: Sends a message to an admin connection.
310 static void a_vwrite(admin *a, const char *status, const char *tag,
311 const char *fmt, va_list ap)
315 if (tag) dstr_puts(&d, "BG");
316 dstr_puts(&d, status);
317 if (tag) u_quotify(&d, tag);
318 a_vformat(&d, fmt, ap);
320 dosend(a, d.buf, d.len);
324 static void a_write(admin *a, const char *status, const char *tag,
325 const char *fmt, ...)
330 a_vwrite(a, status, tag, fmt, ap);
334 /* --- @a_ok@, @a_info@, @a_fail@ --- *
336 * Arguments: @admin *a@ = connection
337 * @const char *fmt@ = format string
338 * @...@ = other arguments
342 * Use: Convenience functions for @a_write@.
345 static void a_ok(admin *a) { a_write(a, "OK", 0, A_END); }
347 static void a_info(admin *a, const char *fmt, ...)
352 a_vwrite(a, "INFO", 0, fmt, ap);
356 static void a_fail(admin *a, const char *fmt, ...)
361 a_vwrite(a, "FAIL", 0, fmt, ap);
365 /* --- @a_alert@, @a_valert@, @a_rawalert@ --- *
367 * Arguments: @unsigned f_and, f_eq@ = filter for connections
368 * @const char *status@ = status string
369 * @const char *fmt@ = pointer to format string
370 * @const char *p@ = pointer to raw string
371 * @size_t sz@ = size of raw string
372 * @va_list ap@ = arguments in list
373 * @...@ = other arguments
377 * Use: Write a message to all admin connections matched by the given
381 static void a_rawalert(unsigned f_and, unsigned f_eq, const char *status,
382 const char *p, size_t sz)
387 if (!(flags & F_INIT))
389 dstr_puts(&d, status);
392 dstr_putm(&d, p, sz);
397 for (a = admins; a; a = aa) {
399 if ((a->f & f_and) == f_eq)
400 dosend(a, d.buf, d.len);
405 static void a_valert(unsigned f_and, unsigned f_eq, const char *status,
406 const char *fmt, va_list ap)
410 if (!(flags & F_INIT))
412 a_vformat(&d, fmt, ap);
413 a_rawalert(f_and, f_eq, status, fmt ? d.buf : 0, fmt ? d.len : 0);
417 static void a_alert(unsigned f_and, unsigned f_eq, const char *status,
418 const char *fmt, ...)
423 a_valert(f_and, f_eq, status, fmt, ap);
427 /* --- @a_warn@ --- *
429 * Arguments: @const char *fmt@ = pointer to format string
430 * @...@ = other arguments
434 * Use: Informs all admin connections of a warning.
437 void a_warn(const char *fmt, ...)
443 a_valert(0, 0, "WARN", fmt, ap);
446 fprintf(stderr, "%s: ", QUIS);
447 a_vformat(&d, fmt, ap);
449 dstr_write(&d, stderr);
455 /* --- @a_trace@ --- *
457 * Arguments: @const char *p@ = pointer to a buffer
458 * @size_t sz@ = size of the buffer
459 * @void *v@ = uninteresting pointer
463 * Use: Custom trace output handler. Sends trace messages to
464 * interested admin connections.
468 static void a_trace(const char *p, size_t sz, void *v)
469 { a_rawalert(AF_TRACE, AF_TRACE, "TRACE", p, sz); }
472 /* --- @a_notify@ --- *
474 * Arguments: @const char *fmt@ = pointer to format string
475 * @...@ = other arguments
479 * Use: Sends a notification to interested admin connections.
482 void a_notify(const char *fmt, ...)
487 a_valert(AF_NOTE, AF_NOTE, "NOTE", fmt, ap);
491 /* --- @a_quit@ --- *
497 * Use: Shuts things down nicely.
506 while ((p = p_first()) != 0)
511 /* --- @a_sigdie@ --- *
513 * Arguments: @int sig@ = signal number
514 * @void *v@ = an uninteresting argument
518 * Use Shuts down on receipt of a fatal signal.
521 static void a_sigdie(int sig, void *v)
527 case SIGTERM: p = "SIGTERM"; break;
528 case SIGINT: p = "SIGINT"; break;
530 sprintf(buf, "%i", sig);
534 a_warn("SERVER", "quit", "signal", "%s", p, A_END);
538 /* --- @a_sighup@ --- *
540 * Arguments: @int sig@ = signal number
541 * @void *v@ = an uninteresting argument
545 * Use Logs a message about SIGHUP not being useful.
548 static void a_sighup(int sig, void *v)
549 { a_warn("SERVER", "ignore", "signal", "SIGHUP", A_END); }
551 /* --- @a_parsetime@ --- *
553 * Arguments; @const char *p@ = time string to parse
555 * Returns: Time in seconds, or @< 0@ on error.
558 static long a_parsetime(const char *p)
561 long t = strtol(p, &q, 0);
567 case 's': if (q[1] != 0)
574 /* --- @a_findpeer@ --- *
576 * Arguments: @admin *a@ = admin connection
577 * @const char *pn@ = peer name
579 * Returns: The peer, or null if not there.
581 * Use: Finds a peer, reporting an error if it failed.
584 static peer *a_findpeer(admin *a, const char *pn)
588 if ((p = p_find(pn)) == 0)
589 a_fail(a, "unknown-peer", "%s", pn, A_END);
593 /*----- Backgrounded operations -------------------------------------------*/
596 (((admin_bgop *)(bg))->tag ? ((admin_bgop *)(bg))->tag : "<foreground>")
598 /* --- @a_bgfind@ --- *
600 * Arguments: @admin *a@ = a client block
601 * @const char *tag@ = the requested tag
603 * Returns: The requested background job, or null.
606 static admin_bgop *a_bgfind(admin *a, const char *tag)
610 for (bg = a->bg; bg; bg = bg->next) {
611 if (bg->tag && strcmp(tag, bg->tag) == 0)
617 /* --- @a_bgrelease@ --- *
619 * Arguments: @admin_bgop *bg@ = backgrounded operation
623 * Use: Removes a backgrounded operation from the queue, since
624 * (presumably) it's done.
627 static void a_bgrelease(admin_bgop *bg)
631 T( trace(T_ADMIN, "admin: release bgop %s", BGTAG(bg)); )
632 if (bg->tag) xfree(bg->tag);
633 else selbuf_enable(&a->b);
634 if (bg->next) bg->next->prev = bg->prev;
635 if (bg->prev) bg->prev->next = bg->next;
636 else a->bg = bg->next;
638 if (!a->bg && (a->f & AF_CLOSE)) a_destroy(a);
641 /* --- @a_bgok@, @a_bginfo@, @a_bgfail@ --- *
643 * Arguments: @admin_bgop *bg@ = backgrounded operation
644 * @const char *fmt@ = format string
645 * @...@ = other arguments
649 * Use: Convenience functions for @a_write@.
652 static void a_bgok(admin_bgop *bg)
653 { a_write(bg->a, "OK", bg->tag, A_END); }
655 static void a_bginfo(admin_bgop *bg, const char *fmt, ...)
659 a_vwrite(bg->a, "INFO", bg->tag, fmt, ap);
663 static void a_bgfail(admin_bgop *bg, const char *fmt, ...)
667 a_vwrite(bg->a, "FAIL", bg->tag, fmt, ap);
671 /* --- @a_bgadd@ --- *
673 * Arguments: @admin *a@ = administration connection
674 * @admin_bgop *bg@ = pointer to background operation
675 * @const char *tag@ = background tag, or null for foreground
676 * @void (*cancel)(admin_bgop *)@ = cancel function
678 * Returns: Zero for success, nonzero on failure.
680 * Use: Links a background job into the list.
683 static int a_bgadd(admin *a, admin_bgop *bg, const char *tag,
684 void (*cancel)(admin_bgop *))
687 if (a_bgfind(a, tag)) {
688 a_fail(a, "tag-exists", "%s", tag, A_END);
691 bg->tag = xstrdup(tag);
695 selbuf_disable(&a->b);
701 if (a->bg) a->bg->prev = bg;
703 T( trace(T_ADMIN, "admin: add bgop %s", BGTAG(bg)); )
704 if (tag) a_write(a, "DETACH", tag, A_END);
708 /*----- Job table manipulation --------------------------------------------*/
711 #define JOB_INDEXMASK ((1ul << JOB_SHIFT) - 1)
712 #define JOB_SEQMASK ((1ul << (32 - JOB_SHIFT)) - 1)
714 #define JOB_END 0xfffffffful
716 static unsigned long a_joboffset;
718 /* --- @a_jobidencode@ --- *
720 * Arguments: @admin_svcop *svc@ = pointer to a service operation
722 * Returns: A jobid for this job, in an internal static buffer.
724 * Use: Constructs a jobid. In order to dissuade people from
725 * predicting jobids, we obfuscate them.
727 * A `raw' jobid consists of two 16-bit fields. The low 16 bits
728 * are an index into a big array. The high 16 bits are a
729 * sequence number recording how many times that slot has been
732 * This `raw' jobid is then obfuscated by adding a randomly-
733 * generated offset, and multiplying (mod %$2^{32}$%) by a fixed
737 static const char *a_jobidencode(admin_svcop *svc)
739 admin_jobtable *j = &svc->prov->j;
744 assert(svc->index <= JOB_INDEXMASK);
745 seq = j->v[svc->index].seq;
746 assert(seq <= JOB_SEQMASK);
747 pre = (unsigned long)svc->index | ((unsigned long)seq << JOB_SHIFT);
748 sprintf(buf, "J%08lx", ((pre + a_joboffset) * 0x0f87a7a3ul) & 0xffffffff);
752 /* --- @a_jobiddecode@ --- *
754 * Arguments: @admin_jobtable *j@ = pointer to a job table
755 * @const char *jid@ = pointer to a jobid string
757 * Returns: A pointer to the job's @svcop@ structure.
760 static admin_svcop *a_jobiddecode(admin_jobtable *j, const char *jid)
767 for (i = 1; i < 9; i++) {
768 if (!isxdigit((unsigned char)jid[i]))
773 pre = strtoul(jid + 1, 0, 16);
774 pre = ((pre * 0xbd11c40bul) - a_joboffset) & 0xffffffff;
775 i = pre & JOB_INDEXMASK;
776 if (i >= j->n || j->v[i].seq != (pre >> JOB_SHIFT))
778 return (j->v[i].u.op);
781 /* --- @a_jobcreate@ --- *
783 * Arguments: @admin *a@ = pointer to administration client
785 * Returns: A pointer to a freshly-allocated @svcop@, or null.
787 * Use: Allocates a fresh @svcop@ and links it into a job table.
790 static admin_svcop *a_jobcreate(admin *a)
795 admin_jobtable *j = &a->j;
797 if (j->free != JOB_END) {
799 j->free = j->v[i].u.next;
802 if (j->sz > JOB_INDEXMASK)
807 j->v = xmalloc(j->sz * sizeof(*j->v));
810 j->v = xrealloc(j->v, j->sz * sizeof(*j->v), sz * sizeof(*j->v));
816 svc = xmalloc(sizeof(*svc));
819 svc->next = j->active;
821 if (j->active) j->active->prev = svc;
824 IF_TRACING(T_ADMIN, {
825 trace(T_ADMIN, "admin: created job %s (%u)", a_jobidencode(svc), i);
830 /* --- @a_jobdestroy@ --- *
832 * Arguments: @admin_svcop *svc@ = pointer to job block
836 * Use: Frees up a completed (or cancelled) job.
839 static void a_jobdestroy(admin_svcop *svc)
841 admin *a = svc->prov;
842 admin_jobtable *j = &a->j;
843 unsigned i = svc->index;
845 IF_TRACING(T_ADMIN, {
846 trace(T_ADMIN, "admin: destroying job %s (%u)", a_jobidencode(svc), i);
848 assert(j->v[i].u.op = svc);
849 j->v[i].u.next = j->free;
852 if (svc->next) svc->next->prev = svc->prev;
853 if (svc->prev) svc->prev->next = svc->next;
854 else j->active = svc->next;
857 /* --- @a_jobtableinit@ --- *
859 * Arguments: @admin_jobtable *j@ = pointer to job table
863 * Use: Initializes a job table.
866 static void a_jobtableinit(admin_jobtable *j)
869 a_joboffset = GR_RANGE(&rand_global, 0xffffffff) + 1;
876 /* --- @a_jobtablefinal@ --- *
878 * Arguments: @admin_jobtable *j@ = pointer to job table
882 * Use: Closes down a job table.
885 static void a_jobtablefinal(admin_jobtable *j)
887 admin_svcop *svc, *ssvc;
889 for (svc = j->active; svc; svc = ssvc) {
891 a_bgfail(&svc->bg, "provider-failed", A_END);
892 a_bgrelease(&svc->bg);
894 if (j->v) xfree(j->v);
897 /*----- Services infrastructure -------------------------------------------*/
899 /* --- @a_svcfind@ --- *
901 * Arguments: @admin *a@ = the requesting client
902 * @const char *name@ = service name wanted
904 * Returns: The service requested, or null.
906 * Use: Finds a service; reports an error if the service couldn't be
910 static admin_service *a_svcfind(admin *a, const char *name)
914 if ((svc = sym_find(&a_svcs, name, -1, 0, 0)) == 0) {
915 a_fail(a, "unknown-service", "%s", name, A_END);
921 /* --- @a_svcunlink@ --- *
923 * Arguments: @admin_service *svc@ = pointer to service structure
927 * Use: Unlinks the service from its provider, without issuing a
928 * message or freeing the structure. The version string is
932 static void a_svcunlink(admin_service *svc)
935 svc->next->prev = svc->prev;
937 svc->prev->next = svc->next;
939 svc->prov->svcs = svc->next;
943 /* --- @a_svcrelease@ --- *
945 * Arguments: @admin_service *svc@ = pointer to service structure
949 * Use: Releases a service and frees its structure.
952 static void a_svcrelease(admin_service *svc)
954 a_notify("SVCRELEASE", "%s", SYM_NAME(svc), A_END);
956 sym_remove(&a_svcs, svc);
959 /*----- Name resolution operations ----------------------------------------*/
961 /* --- @a_resolved@ --- *
963 * Arguments: @struct hostent *h@ = pointer to resolved hostname
964 * @void *v@ = pointer to resolver operation
968 * Use: Handles a completed name resolution.
971 static void a_resolved(struct hostent *h, void *v)
975 T( trace(T_ADMIN, "admin: resop %s resolved", BGTAG(r)); )
978 a_bgfail(&r->bg, "resolve-error", "%s", r->addr, A_END);
979 r->func(r, ARES_FAIL);
981 memcpy(&r->sa.sin.sin_addr, h->h_addr, sizeof(struct in_addr));
989 /* --- @a_restimer@ --- *
991 * Arguments: @struct timeval *tv@ = timer
992 * @void *v@ = pointer to resolver operation
996 * Use: Times out a resolver.
999 static void a_restimer(struct timeval *tv, void *v)
1003 T( trace(T_ADMIN, "admin: resop %s timeout", BGTAG(r)); )
1004 a_bgfail(&r->bg, "resolver-timeout", "%s", r->addr, A_END);
1005 r->func(r, ARES_FAIL);
1008 a_bgrelease(&r->bg);
1011 /* --- @a_rescancel@ --- *
1013 * Arguments: @admin_bgop *bg@ = background operation
1017 * Use: Cancels an add operation.
1020 static void a_rescancel(admin_bgop *bg)
1022 admin_resop *r = (admin_resop *)bg;
1024 T( trace(T_ADMIN, "admin: cancel resop %s", BGTAG(r)); )
1025 r->func(r, ARES_FAIL);
1031 /* --- @a_resolve@ --- *
1033 * Arguments: @admin *a@ = administration connection
1034 * @admin_resop *r@ = resolver operation to run
1035 * @const char *tag@ = background operation tag
1036 * @void (*func)(struct admin_resop *, int@ = handler function
1037 * @unsigned ac@ = number of remaining arguments
1038 * @char *av[]@ = pointer to remaining arguments
1042 * Use: Cranks up a resolver job.
1045 static void a_resolve(admin *a, admin_resop *r, const char *tag,
1046 void (*func)(struct admin_resop *, int),
1047 unsigned ac, char *av[])
1054 /* --- Fill in the easy bits of address --- */
1056 r->bg.tag = "<starting>";
1059 if (mystrieq(av[i], "inet")) i++;
1060 if (ac - i != 1 && ac - i != 2) {
1061 a_fail(a, "bad-addr-syntax", "[inet] ADDRESS [PORT]", A_END);
1064 r->sa.sin.sin_family = AF_INET;
1065 r->sasz = sizeof(r->sa.sin);
1066 r->addr = xstrdup(av[i]);
1070 pt = strtoul(av[i + 1], &p, 0);
1072 struct servent *s = getservbyname(av[i + 1], "udp");
1074 a_fail(a, "unknown-service", "%s", av[i + 1], A_END);
1077 pt = ntohs(s->s_port);
1080 if (pt == 0 || pt >= 65536) {
1081 a_fail(a, "invalid-port", "%lu", pt, A_END);
1084 r->sa.sin.sin_port = htons(pt);
1086 /* --- Report backgrounding --- *
1088 * Do this for consistency of interface, even if we're going to get the
1089 * answer straight away.
1092 if (a_bgadd(a, &r->bg, tag, a_rescancel))
1094 T( trace(T_ADMIN, "admin: %u, resop %s, hostname `%s'",
1095 a->seq, BGTAG(r), r->addr); )
1097 /* --- If the name is numeric, do it the easy way --- */
1099 if (inet_aton(av[i], &r->sa.sin.sin_addr)) {
1100 T( trace(T_ADMIN, "admin: resop %s done the easy way", BGTAG(r)); )
1103 a_bgrelease(&r->bg);
1107 /* --- Store everything for later and crank up the resolver --- */
1109 gettimeofday(&tv, 0);
1110 tv.tv_sec += T_RESOLVE;
1111 sel_addtimer(&sel, &r->t, &tv, a_restimer, r);
1112 bres_byname(&r->r, r->addr, a_resolved, r);
1117 if (r->addr) xfree(r->addr);
1121 /*----- Option parsing ----------------------------------------------------*/
1123 #define OPTIONS(argc, argv, guts) do { \
1124 char **o_av = argv; \
1128 if (mystrieq(*o_av, "--")) { \
1133 if (**o_av == '-') \
1137 argc -= o_av - argv; \
1141 #define OPT(name, guts) if (mystrieq(*o_av, name)) { guts continue; }
1143 #define OPTARG(name, arg, guts) OPT(name, { \
1146 if (!arg) goto bad_syntax; \
1150 #define OPTTIME(name, arg, guts) OPTARG(name, o_arg, { \
1152 if ((arg = a_parsetime(o_arg)) < 0) { \
1153 a_fail(a, "bad-time-spec", "%s", o_arg, A_END); \
1159 /*----- Adding peers ------------------------------------------------------*/
1161 /* --- @a_doadd@ --- *
1163 * Arguments: @admin_resop *r@ = resolver operation
1164 * @int rc@ = how it worked
1168 * Use: Handles a completed resolution.
1171 static void a_doadd(admin_resop *r, int rc)
1173 admin_addop *add = (admin_addop *)r;
1175 T( trace(T_ADMIN, "admin: done add op %s", BGTAG(add)); )
1177 if (rc == ARES_OK) {
1178 add->peer.sasz = add->r.sasz;
1179 add->peer.sa = add->r.sa;
1180 if (p_find(add->peer.name))
1181 a_bgfail(&add->r.bg, "peer-exists", "%s", add->peer.name, A_END);
1182 else if (!p_create(&add->peer))
1183 a_bgfail(&add->r.bg, "peer-create-fail", "%s", add->peer.name, A_END);
1188 xfree(add->peer.name);
1191 /* --- @acmd_add@ --- *
1193 * Arguments: @admin *a@ = connection which requested the addition
1194 * @unsigned ac@ = argument count
1195 * @char *av[]@ = pointer to the argument list
1199 * Use: Adds a new peer.
1202 static void acmd_add(admin *a, unsigned ac, char *av[])
1204 const char *tag = 0;
1207 /* --- Set stuff up --- */
1209 add = xmalloc(sizeof(*add));
1212 add->peer.tops = tun_default;
1214 /* --- Parse options --- */
1217 OPTARG("-background", arg, { tag = arg; })
1218 OPTARG("-tunnel", arg, {
1222 a_fail(a, "unknown-tunnel", "%s", arg, A_END);
1225 if (mystrieq(arg, tunnels[i]->name)) {
1226 add->peer.tops = tunnels[i];
1231 OPTTIME("-keepalive", t, { add->peer.t_ka = t; })
1234 /* --- Make sure someone's not got there already --- */
1236 if (!av[0] || !av[1])
1239 a_fail(a, "peer-exists", "%s", *av, A_END);
1242 add->peer.name = xstrdup(*av++);
1245 /* --- Crank up the resolver --- */
1247 a_resolve(a, &add->r, tag, a_doadd, ac, av);
1250 /* --- Clearing up --- */
1253 a_fail(a, "bad-syntax", "add", "[OPTIONS] PEER ADDR ...", A_END);
1255 if (add->peer.name) xfree(add->peer.name);
1260 /*----- Ping --------------------------------------------------------------*/
1262 /* --- @a_pingcancel@ --- *
1264 * Arguments: @admin_bgop *bg@ = background operation block
1268 * Use: Cancels a running ping.
1271 static void a_pingcancel(admin_bgop *bg)
1273 admin_pingop *pg = (admin_pingop *)bg;
1274 T( trace(T_ADMIN, "admin: cancel ping op %s", BGTAG(pg)); )
1275 p_pingdone(&pg->ping, PING_NONOTIFY);
1278 /* --- @a_pong@ --- *
1280 * Arguments: @int rc@ = return code
1281 * @void *v@ = ping operation block
1285 * Use: Collects what happened to a ping message.
1288 static void a_pong(int rc, void *v)
1290 admin_pingop *pg = v;
1296 gettimeofday(&tv, 0);
1297 tv_sub(&tv, &tv, &pg->pingtime);
1298 millis = (double)tv.tv_sec * 1000 + (double)tv.tv_usec/1000;
1299 a_bginfo(&pg->bg, "ping-ok", "%.1f", millis, A_END);
1303 a_bginfo(&pg->bg, "ping-timeout", A_END);
1307 a_bginfo(&pg->bg, "ping-peer-died", A_END);
1313 T( trace(T_ADMIN, "admin: ponged ping op %s", BGTAG(pg)); )
1314 a_bgrelease(&pg->bg);
1317 /* --- @acmd_ping@, @acmd_eping@ --- *
1319 * Arguments: @admin *a@ = connection which requested the ping
1320 * @unsigned ac@ = argument count
1321 * @char *av[]@ = pointer to the argument list
1325 * Use: Pings a peer.
1328 static void a_ping(admin *a, unsigned ac, char *av[],
1329 const char *cmd, unsigned msg)
1333 admin_pingop *pg = 0;
1334 const char *tag = 0;
1337 OPTARG("-background", arg, { tag = arg; })
1338 OPTTIME("-timeout", arg, { t = arg; })
1340 if (!*av || av[1]) goto bad_syntax;
1341 if ((p = a_findpeer(a, *av)) == 0)
1343 pg = xmalloc(sizeof(*pg));
1344 gettimeofday(&pg->pingtime, 0);
1345 if (a_bgadd(a, &pg->bg, tag, a_pingcancel))
1347 T( trace(T_ADMIN, "admin: ping op %s: %s to %s",
1348 BGTAG(pg), cmd, p_name(p)); )
1349 if (p_pingsend(p, &pg->ping, msg, t, a_pong, pg)) {
1350 a_bgfail(&pg->bg, "ping-send-failed", A_END);
1351 a_bgrelease(&pg->bg);
1356 a_fail(a, "bad-syntax", "%s", cmd, "[OPTIONS] PEER", cmd, A_END);
1362 static void acmd_ping(admin *a, unsigned ac, char *av[])
1363 { a_ping(a, ac, av, "ping", MISC_PING); }
1364 static void acmd_eping(admin *a, unsigned ac, char *av[])
1365 { a_ping(a, ac, av, "eping", MISC_EPING); }
1367 /*----- Service commands --------------------------------------------------*/
1369 static void acmd_svcclaim(admin *a, unsigned ac, char *av[])
1374 svc = sym_find(&a_svcs, av[0], -1, sizeof(*svc), &f);
1376 if (versioncmp(av[1], svc->version) <= 0) {
1379 "%s", SYM_NAME(svc),
1384 a_write(svc->prov, "SVCCLAIM", 0, "%s", av[0], "%s", av[1], A_END);
1388 svc->version = xstrdup(av[1]);
1389 svc->next = a->svcs;
1391 if (a->svcs) a->svcs->prev = svc;
1393 a_notify("SVCCLAIM", "%s", SYM_NAME(svc), "%s", svc->version, A_END);
1397 static void acmd_svcrelease(admin *a, unsigned ac, char *av[])
1401 if ((svc = a_svcfind(a, av[0])) == 0)
1403 if (svc->prov != a) {
1404 a_fail(a, "not-service-provider", "%s", SYM_NAME(svc), A_END);
1411 static void acmd_svcensure(admin *a, unsigned ac, char *av[])
1415 if ((svc = a_svcfind(a, av[0])) == 0)
1417 if (av[1] && versioncmp(svc->version, av[1]) < 0) {
1418 a_fail(a, "service-too-old",
1419 "%s", SYM_NAME(svc),
1427 static void acmd_svcquery(admin *a, unsigned ac, char *av[])
1431 if ((svc = a_svcfind(a, av[0])) != 0) {
1432 a_info(a, "name=%s", SYM_NAME(svc), "version=%s", svc->version, A_END);
1437 static void acmd_svclist(admin *a, unsigned ac, char *av[])
1442 for (sym_mkiter(&i, &a_svcs); (svc = sym_next(&i)) != 0; )
1443 a_info(a, "%s", SYM_NAME(svc), "%s", svc->version, A_END);
1447 static void a_canceljob(admin_bgop *bg)
1449 admin_svcop *svc = (admin_svcop *)bg;
1451 a_write(svc->prov, "SVCCANCEL", 0, "%s", a_jobidencode(svc), A_END);
1455 static void acmd_svcsubmit(admin *a, unsigned ac, char *av[])
1457 const char *tag = 0;
1460 const char *ver = 0;
1464 OPTARG("-background", arg, { tag = arg; })
1465 OPTARG("-version", arg, { ver = arg; })
1467 if (!*av) goto bad_syntax;
1468 if ((svc = a_svcfind(a, *av)) == 0) goto fail;
1469 if (ver && versioncmp(svc->version, ver) < 0) {
1470 a_fail(a, "service-too-old",
1471 "%s", SYM_NAME(svc),
1476 if ((svcop = a_jobcreate(svc->prov)) == 0) {
1477 a_fail(a, "provider-overloaded", A_END);
1480 if (a_bgadd(a, &svcop->bg, tag, a_canceljob)) {
1481 a_jobdestroy(svcop);
1484 a_write(svc->prov, "SVCJOB", 0,
1485 "%s", a_jobidencode(svcop),
1491 a_fail(a, "bad-syntax", "svcsubmit", "[OPTIONS] SERVICE ARGS...", A_END);
1498 static void a_replycmd(admin *a, const char *status, int donep, char *av[])
1502 if ((svc = a_jobiddecode(&a->j, av[0])) == 0) {
1503 a_fail(a, "unknown-jobid", av[0], A_END);
1506 a_write(svc->bg.a, status, svc->bg.tag, "?TOKENS", av + 1, A_END);
1509 a_bgrelease(&svc->bg);
1514 static void acmd_svcok(admin *a, unsigned ac, char *av[])
1515 { a_replycmd(a, "OK", 1, av); }
1516 static void acmd_svcinfo(admin *a, unsigned ac, char *av[])
1517 { a_replycmd(a, "INFO", 0, av); }
1518 static void acmd_svcfail(admin *a, unsigned ac, char *av[])
1519 { a_replycmd(a, "FAIL", 1, av); }
1521 /*----- Administration commands -------------------------------------------*/
1523 /* --- Miscellaneous commands --- */
1525 /* --- @traceish@ --- *
1527 * Arguments: @admin *a@ = connection to complain on
1528 * @unsigned ac@ = number of arguments
1529 * @char *av[]@ = vector of arguments
1530 * @const char *what@ = what we're messing with
1531 * @const trace_opt *tt@ = options table
1532 * @unsigned *ff@ = where the flags are
1534 * Returns: Nonzero if anything changed.
1536 * Use: Guts of trace-ish commands like `trace' and `watch'.
1539 static int traceish(admin *a, unsigned ac, char *av[],
1540 const char *what, const trace_opt *tt, unsigned *ff)
1544 if (!ac || strcmp(av[0], "?") == 0) {
1546 for (t = tt; t->ch; t++) {
1547 a_info(a, "*%c%c %s",
1548 t->ch, (*ff & t->f) == t->f ? '+' : ' ', t->help, A_END);
1558 case '+': sense = 1; break;
1559 case '-': sense = 0; break;
1561 for (t = tt; t->ch; t++) {
1563 if (sense) f |= t->f;
1568 a_fail(a, "bad-%s-option", what, "%c", *p, A_END);
1584 static void acmd_trace(admin *a, unsigned ac, char *av[])
1586 if (traceish(a, ac, av, "trace", tr_opts, &tr_flags))
1587 trace_level(tr_flags);
1592 static void acmd_watch(admin *a, unsigned ac, char *av[])
1593 { traceish(a, ac, av, "watch", w_opts, &a->f); }
1595 static void alertcmd(admin *a, unsigned f_and, unsigned f_eq,
1596 const char *status, char *av[])
1597 { a_alert(f_and, f_eq, status, "USER", "?TOKENS", av, A_END); a_ok(a); }
1598 static void acmd_notify(admin *a, unsigned ac, char *av[])
1599 { alertcmd(a, AF_NOTE, AF_NOTE, "NOTE", av); }
1600 static void acmd_warn(admin *a, unsigned ac, char *av[])
1601 { alertcmd(a, AF_WARN, AF_WARN, "WARN", av); }
1603 static void acmd_port(admin *a, unsigned ac, char *av[])
1604 { a_info(a, "%u", p_port(), A_END); a_ok(a); }
1606 static void acmd_daemon(admin *a, unsigned ac, char *av[])
1608 if (flags & F_DAEMON)
1609 a_fail(a, "already-daemon", A_END);
1611 a_notify("DAEMON", A_END);
1615 a_fail(a, "daemon-error", "?ERRNO", A_END);
1623 static void acmd_jobs(admin *a, unsigned ac, char *av[])
1627 for (bg = a->bg; bg; bg = bg->next) {
1629 a_info(a, "%s", bg->tag, A_END);
1634 static void acmd_bgcancel(admin *a, unsigned ac, char *av[])
1638 if ((bg = a_bgfind(a, av[0])) == 0)
1639 a_fail(a, "unknown-tag", "%s", av[0], A_END);
1647 static void acmd_list(admin *a, unsigned ac, char *av[])
1651 for (p = p_first(); p; p = p_next(p))
1652 a_info(a, "%s", p_name(p), A_END);
1656 static void acmd_ifname(admin *a, unsigned ac, char *av[])
1660 if ((p = a_findpeer(a, av[0])) != 0) {
1661 a_info(a, "%s", p_ifname(p), A_END);
1666 static void acmd_setifname(admin *a, unsigned ac, char *av[])
1670 if ((p = a_findpeer(a, av[0])) != 0) {
1671 a_notify("NEWIFNAME", "?PEER", p, "%s", p_ifname(p), "%s", av[1], A_END);
1672 p_setifname(p, av[1]);
1677 static void acmd_getchal(admin *a, unsigned ac, char *av[])
1681 buf_init(&b, buf_i, PKBUFSZ);
1683 a_info(a, "?B64", BBASE(&b), (size_t)BLEN(&b), A_END);
1687 static void acmd_checkchal(admin *a, unsigned ac, char *av[])
1694 base64_decode(&b64, av[0], strlen(av[0]), &d);
1695 base64_decode(&b64, 0, 0, &d);
1696 buf_init(&b, d.buf, d.len);
1697 if (c_check(&b) || BBAD(&b) || BLEFT(&b))
1698 a_fail(a, "invalid-challenge", A_END);
1704 static void acmd_greet(admin *a, unsigned ac, char *av[])
1710 if ((p = a_findpeer(a, av[0])) != 0) {
1712 base64_decode(&b64, av[1], strlen(av[1]), &d);
1713 base64_decode(&b64, 0, 0, &d);
1714 p_greet(p, d.buf, d.len);
1720 static void acmd_addr(admin *a, unsigned ac, char *av[])
1725 if ((p = a_findpeer(a, av[0])) != 0) {
1727 assert(ad->sa.sa_family == AF_INET);
1728 a_info(a, "?ADDR", ad, A_END);
1733 static void acmd_peerinfo(admin *a, unsigned ac, char *av[])
1738 if ((p = a_findpeer(a, av[0])) != 0) {
1740 a_info(a, "tunnel=%s", ps->tops->name, A_END);
1741 a_info(a, "keepalive=%lu", ps->t_ka, A_END);
1746 static void acmd_servinfo(admin *a, unsigned ac, char *av[])
1748 a_info(a, "implementation=edgeware-tripe", A_END);
1749 a_info(a, "version=%s", VERSION, A_END);
1750 a_info(a, "daemon=%s", BOOL(flags & F_DAEMON), A_END);
1754 static void acmd_stats(admin *a, unsigned ac, char *av[])
1759 if ((p = a_findpeer(a, av[0])) == 0)
1763 a_info(a, "start-time=%s", timestr(st->t_start), A_END);
1764 a_info(a, "last-packet-time=%s", timestr(st->t_last), A_END);
1765 a_info(a, "last-keyexch-time=%s", timestr(st->t_kx), A_END);
1766 a_info(a, "packets-in=%lu", st->n_in, "bytes-in=%lu", st->sz_in, A_END);
1768 "packets-out=%lu", st->n_out,
1769 "bytes-out=%lu", st->sz_out,
1772 "keyexch-packets-in=%lu", st->n_kxin,
1773 "keyexch-bytes-in=%lu", st->sz_kxin,
1776 "keyexch-packets-out=%lu", st->n_kxout,
1777 "keyexch-bytes-out=%lu", st->sz_kxout,
1780 "ip-packets-in=%lu", st->n_ipin,
1781 "ip-bytes-in=%lu", st->sz_ipin,
1784 "ip-packets-out=%lu", st->n_ipout,
1785 "ip-bytes-out=%lu", st->sz_ipout,
1787 a_info(a, "rejected-packets=%lu", st->n_reject, A_END);
1791 static void acmd_kill(admin *a, unsigned ac, char *av[])
1795 if ((p = a_findpeer(a, av[0])) != 0) {
1801 static void acmd_forcekx(admin *a, unsigned ac, char *av[])
1805 if ((p = a_findpeer(a, av[0])) != 0) {
1806 kx_start(&p->kx, 1);
1811 static void acmd_reload(admin *a, unsigned ac, char *av[])
1812 { p_keyreload(); a_ok(a); }
1814 static void acmd_quit(admin *a, unsigned ac, char *av[])
1816 a_warn("SERVER", "quit", "admin-request", A_END);
1821 static void acmd_version(admin *a, unsigned ac, char *av[])
1823 a_info(a, "%s", PACKAGE, "%s", VERSION, A_END);
1827 static void acmd_tunnels(admin *a, unsigned ac, char *av[])
1831 for (i = 0; tunnels[i]; i++)
1832 a_info(a, "%s", tunnels[i]->name, A_END);
1836 /* --- The command table and help --- */
1838 typedef struct acmd {
1841 unsigned argmin, argmax;
1842 void (*func)(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1845 static void acmd_help(admin */*a*/, unsigned /*ac*/, char */*av*/[]);
1847 static const acmd acmdtab[] = {
1848 { "add", "[OPTIONS] PEER ADDR ...", 2, 0xffff, acmd_add },
1849 { "addr", "PEER", 1, 1, acmd_addr },
1850 { "bgcancel", "TAG", 1, 1, acmd_bgcancel },
1851 { "checkchal", "CHAL", 1, 1, acmd_checkchal },
1852 { "daemon", 0, 0, 0, acmd_daemon },
1853 { "eping", "[OPTIONS] PEER", 1, 0xffff, acmd_eping },
1854 { "forcekx", "PEER", 1, 1, acmd_forcekx },
1855 { "getchal", 0, 0, 0, acmd_getchal },
1856 { "greet", "PEER CHAL", 2, 2, acmd_greet },
1857 { "help", 0, 0, 0, acmd_help },
1858 { "ifname", "PEER", 1, 1, acmd_ifname },
1859 { "jobs", 0, 0, 0, acmd_jobs },
1860 { "kill", "PEER", 1, 1, acmd_kill },
1861 { "list", 0, 0, 0, acmd_list },
1862 { "notify", "MESSAGE ...", 1, 0xffff, acmd_notify },
1863 { "peerinfo", "PEER", 1, 1, acmd_peerinfo },
1864 { "ping", "[OPTIONS] PEER", 1, 0xffff, acmd_ping },
1865 { "port", 0, 0, 0, acmd_port },
1866 { "quit", 0, 0, 0, acmd_quit },
1867 { "reload", 0, 0, 0, acmd_reload },
1868 { "servinfo", 0, 0, 0, acmd_servinfo },
1869 { "setifname", "PEER NEW-NAME", 2, 2, acmd_setifname },
1870 { "svcclaim", "SERVICE VERSION", 2, 2, acmd_svcclaim },
1871 { "svcensure", "SERVICE [VERSION]", 1, 2, acmd_svcensure },
1872 { "svcfail", "JOBID TOKENS...", 1, 0xffff, acmd_svcfail },
1873 { "svcinfo", "JOBID TOKENS...", 1, 0xffff, acmd_svcinfo },
1874 { "svclist", 0, 0, 0, acmd_svclist },
1875 { "svcok", "JOBID", 1, 1, acmd_svcok },
1876 { "svcquery", "SERVICE", 1, 1, acmd_svcquery },
1877 { "svcrelease", "SERVICE", 1, 1, acmd_svcrelease },
1878 { "svcsubmit", "[OPTIONS] SERVICE TOKENS...",
1879 2, 0xffff, acmd_svcsubmit },
1880 { "stats", "PEER", 1, 1, acmd_stats },
1882 { "trace", "[OPTIONS]", 0, 1, acmd_trace },
1884 { "tunnels", 0, 0, 0, acmd_tunnels },
1885 { "version", 0, 0, 0, acmd_version },
1886 { "warn", "MESSAGE ...", 1, 0xffff, acmd_warn },
1887 { "watch", "[OPTIONS]", 0, 1, acmd_watch },
1891 static void acmd_help(admin *a, unsigned ac, char *av[])
1895 for (c = acmdtab; c->name; c++) {
1897 a_info(a, "%s", c->name, "*%s", c->help, A_END);
1899 a_info(a, "%s", c->name, A_END);
1904 /*----- Connection handling -----------------------------------------------*/
1906 /* --- @a_destroypending@ --- *
1912 * Use: Destroys pending admin connections at a safe time.
1915 static void a_destroypending(void)
1918 admin_bgop *bg, *bbg;
1919 admin_service *svc, *ssvc;
1921 /* --- Destroy connections marked as pending --- */
1923 for (a = a_dead; a; a = aa) {
1925 assert(a->f & AF_DEAD);
1927 /* --- Report what we're doing --- */
1929 T( trace(T_ADMIN, "admin: completing destruction of connection %u",
1932 /* --- Abort any background jobs in progress --- */
1934 for (bg = a->bg; bg; bg = bbg) {
1937 if (bg->tag) xfree(bg->tag);
1941 /* --- Release services I hold, and abort pending jobs --- */
1943 for (svc = a->svcs; svc; svc = ssvc) {
1947 a_jobtablefinal(&a->j);
1949 /* --- Close file descriptors and selectory --- */
1951 selbuf_destroy(&a->b);
1952 if (a->b.reader.fd != a->w.fd) close(a->b.reader.fd);
1954 if (a_stdin == a) a_stdin = 0;
1961 /* --- All pending destruction completed --- */
1966 /* --- @a_destroy@ --- *
1968 * Arguments: @admin *a@ = pointer to an admin block
1972 * Use: Destroys an admin block. This requires a certain amount of
1976 static void freequeue(oqueue *q)
1980 for (o = q->hd; o; o = oo) {
1987 static void a_destroy(admin *a)
1992 if (a->next) a->next->prev = a->prev;
1993 if (a->prev) a->prev->next = a->next;
1994 else admins = a->next;
1996 if (a->out.hd) sel_rmfile(&a->w);
2003 T( trace(T_ADMIN, "admin: killing connection %u", a->seq); )
2006 /* --- @a_line@ --- *
2008 * Arguments: @char *p@ = pointer to the line read
2009 * @size_t len@ = length of the line
2010 * @void *vp@ = pointer to my admin block
2014 * Use: Handles a line of input.
2017 static void a_line(char *p, size_t len, void *vp)
2032 selbuf_disable(&a->b);
2036 ac = str_qsplit(p, av, 16, 0, STRF_QUOTE);
2039 for (c = acmdtab; c->name; c++) {
2040 if (mystrieq(av[0], c->name)) {
2042 if (c->argmin > ac || ac > c->argmax) {
2044 a_fail(a, "bad-syntax", "%s", c->name, "", A_END);
2046 a_fail(a, "bad-syntax", "%s", c->name, "%s", c->help, A_END);
2048 c->func(a, ac, av + 1);
2052 a_fail(a, "unknown-command", "%s", av[0], A_END);
2055 /* --- @a_create@ --- *
2057 * Arguments: @int fd_in, fd_out@ = file descriptors to use
2058 * @unsigned f@ = initial flags to set
2062 * Use: Creates a new admin connection.
2065 void a_create(int fd_in, int fd_out, unsigned f)
2067 admin *a = CREATE(admin);
2069 T( static unsigned seq = 0;
2071 T( trace(T_ADMIN, "admin: accepted connection %u", a->seq); )
2075 a_jobtableinit(&a->j);
2077 if (fd_in == STDIN_FILENO) a_stdin = a;
2078 fdflags(fd_in, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2079 if (fd_out != fd_in)
2080 fdflags(fd_out, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2081 selbuf_init(&a->b, &sel, fd_in, a_line, a);
2082 sel_initfile(&sel, &a->w, fd_out, SEL_WRITE, a_flush, a);
2083 a->out.hd = a->out.tl = 0;
2086 if (admins) admins->prev = a;
2090 /* --- @a_accept@ --- *
2092 * Arguments: @int fd@ = file descriptor to accept
2093 * @unsigned mode@ = what to do
2094 * @void *v@ = uninteresting pointer
2098 * Use: Accepts a new admin connection.
2101 static void a_accept(int fd, unsigned mode, void *v)
2104 struct sockaddr_un sun;
2105 size_t sz = sizeof(sun);
2107 if ((nfd = accept(fd, (struct sockaddr *)&sun, &sz)) < 0) {
2108 if (errno != EINTR && errno != EAGAIN && errno != EWOULDBLOCK &&
2109 errno != ECONNABORTED && errno != EPROTO)
2110 a_warn("ADMIN", "accept-error", "?ERRNO", A_END);
2113 a_create(nfd, nfd, 0);
2116 /* --- @a_preselect@ --- *
2122 * Use: Informs the admin module that we're about to select again,
2123 * and that it should do cleanup things it has delayed until a
2127 void a_preselect(void) { if (a_dead) a_destroypending(); }
2129 /* --- @a_daemon@ --- *
2135 * Use: Informs the admin module that it's a daemon.
2138 void a_daemon(void) { flags |= F_DAEMON; }
2140 /* --- @a_init@ --- *
2142 * Arguments: @const char *name@ = socket name to create
2146 * Use: Creates the admin listening socket.
2149 void a_init(const char *name)
2153 struct sockaddr_un sun;
2154 struct sigaction sa;
2157 /* --- Create services table --- */
2159 sym_create(&a_svcs);
2161 /* --- Set up the socket address --- */
2163 sz = strlen(name) + 1;
2164 if (sz > sizeof(sun.sun_path))
2165 die(EXIT_FAILURE, "socket name `%s' too long", name);
2167 sun.sun_family = AF_UNIX;
2168 memcpy(sun.sun_path, name, sz);
2169 sz += offsetof(struct sockaddr_un, sun_path);
2171 /* --- Attempt to bind to the socket --- */
2175 if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0)
2176 die(EXIT_FAILURE, "couldn't create socket: %s", strerror(errno));
2177 if (bind(fd, (struct sockaddr *)&sun, sz) < 0) {
2180 if (errno != EADDRINUSE) {
2181 die(EXIT_FAILURE, "couldn't bind to address `%s': %s",
2182 sun.sun_path, strerror(e));
2185 die(EXIT_FAILURE, "too many retries; giving up");
2187 if (!connect(fd, (struct sockaddr *)&sun, sz)) {
2188 die(EXIT_FAILURE, "server already listening on admin socket `%s'",
2191 if (errno != ECONNREFUSED)
2192 die(EXIT_FAILURE, "couldn't bind to address: %s", strerror(e));
2193 if (stat(sun.sun_path, &st)) {
2194 die(EXIT_FAILURE, "couldn't stat `%s': %s",
2195 sun.sun_path, strerror(errno));
2197 if (!S_ISSOCK(st.st_mode))
2198 die(EXIT_FAILURE, "object `%s' isn't a socket", sun.sun_path);
2199 T( trace(T_ADMIN, "admin: stale socket found; removing it"); )
2200 unlink(sun.sun_path);
2204 chmod(sun.sun_path, 0600);
2205 fdflags(fd, O_NONBLOCK, O_NONBLOCK, FD_CLOEXEC, FD_CLOEXEC);
2207 die(EXIT_FAILURE, "couldn't listen on socket: %s", strerror(errno));
2209 /* --- Listen to the socket --- */
2211 sel_initfile(&sel, &sock, fd, SEL_READ, a_accept, 0);
2215 T( trace_custom(a_trace, 0);
2216 trace(T_ADMIN, "admin: enabled custom tracing"); )
2219 /* --- Set up signal handlers --- */
2221 sig_add(&s_term, SIGTERM, a_sigdie, 0);
2222 sig_add(&s_hup, SIGHUP, a_sighup, 0);
2223 signal(SIGPIPE, SIG_IGN);
2224 sigaction(SIGINT, 0, &sa);
2225 if (sa.sa_handler != SIG_IGN)
2226 sig_add(&s_int, SIGINT, a_sigdie, 0);
2229 /*----- That's all, folks -------------------------------------------------*/