5 * Admin interface for configuration
7 * (c) 2001 Straylight/Edgeware
10 /*----- Licensing notice --------------------------------------------------*
12 * This file is part of Trivial IP Encryption (TrIPE).
14 * TrIPE is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU General Public License as published by
16 * the Free Software Foundation; either version 2 of the License, or
17 * (at your option) any later version.
19 * TrIPE is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 * GNU General Public License for more details.
24 * You should have received a copy of the GNU General Public License
25 * along with TrIPE; if not, write to the Free Software Foundation,
26 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
29 /*----- Header files ------------------------------------------------------*/
33 /*----- Global variables --------------------------------------------------*/
37 const trace_opt tr_opts
[] = {
38 { 't', T_TUNNEL
, "tunnel events" },
39 { 'r', T_PEER
, "peer events" },
40 { 'a', T_ADMIN
, "admin interface" },
41 { 's', T_KEYSET
, "symmetric keyset management" },
42 { 'x', T_KEYEXCH
, "key exchange" },
43 { 'm', T_KEYMGMT
, "key management" },
44 { 'l', T_CHAL
, "challenge management" },
45 { 'p', T_PACKET
, "packet contents" },
46 { 'c', T_CRYPTO
, "crypto details" },
47 { 'A', T_ALL
, "all of the above" },
51 unsigned tr_flags
= 0;
54 static const trace_opt w_opts
[] = {
55 { 't', AF_TRACE
, "trace messages" },
56 { 'n', AF_NOTE
, "asynchronous notifications" },
57 { 'w', AF_WARN
, "warnings" },
58 { 'A', AF_ALLMSGS
, "all of the above" },
62 /*----- Static variables --------------------------------------------------*/
67 static const char *sockname
;
68 static sym_table a_svcs
;
69 static unsigned flags
= 0;
70 static admin
*a_stdin
= 0;
71 static sig s_term
, s_int
, s_hup
;
76 #define T_RESOLVE SEC(30)
79 static void a_destroy(admin */
*a*/
);
81 #define BOOL(x) ((x) ? "t" : "nil")
83 /*----- Output functions --------------------------------------------------*/
85 /* --- @trywrite@ --- *
87 * Arguments: @admin *a@ = pointer to an admin block
88 * @const char *p@ = pointer to buffer to write
89 * @size_t sz@ = size of data to write
91 * Returns: The number of bytes written, or less than zero on error.
93 * Use: Attempts to write data to a client.
96 static ssize_t
trywrite(admin
*a
, const char *p
, size_t sz
)
103 n
= write(a
->w
.fd
, p
, sz
);
113 if (errno
!= EAGAIN
&& errno
!= EWOULDBLOCK
) {
115 a_warn("ADMIN", "client-write-error", "?ERRNO", A_END
);
122 /* --- @doqueue@ -- *
124 * Arguments: @oqueue *q@ = pointer to output queue
125 * @const char *p@ = pointer to buffer to write
126 * @size_t sz@ = size of buffer
128 * Returns: Nonzero if the queue was previously empty.
130 * Use: Queues data to be written later.
133 static int doqueue(oqueue
*q
, const char *p
, size_t sz
)
142 else if (o
->p_in
< o
->buf
+ OBUFSZ
)
146 o
= xmalloc(sizeof(obuf
));
148 o
->p_in
= o
->p_out
= o
->buf
;
156 n
= o
->buf
+ OBUFSZ
- o
->p_in
;
159 memcpy(o
->p_in
, p
, n
);
168 /* --- @dosend@ --- *
170 * Arguemnts: @admin *a@ = pointer to an admin block
171 * @const char *p@ = pointer to buffer to write
172 * @size_t sz@ = size of data to write
176 * Use: Sends data to an admin client.
179 static void dosend(admin
*a
, const char *p
, size_t sz
)
186 if ((n
= trywrite(a
, p
, sz
)) < 0)
193 if (doqueue(&a
->out
, p
, sz
))
197 /* --- @a_flush@ --- *
199 * Arguments: @int fd@ = file descriptor
200 * @unsigned mode@ = what's happening
201 * @void *v@ = pointer to my admin block
205 * Use: Flushes buffers when a client is ready to read again.
208 static void a_flush(int fd
, unsigned mode
, void *v
)
216 if ((n
= trywrite(a
, o
->p_out
, o
->p_in
- o
->p_out
)) < 0)
219 if (o
->p_in
< o
->p_out
)
232 /*----- Utility functions -------------------------------------------------*/
234 /* --- @a_vformat@ --- *
236 * Arguments: @dstr *d@ = where to leave the formatted message
237 * @const char *fmt@ = pointer to format string
238 * @va_list ap@ = arguments in list
242 * Use: Main message token formatting driver.
245 static void a_vformat(dstr
*d
, const char *fmt
, va_list ap
)
252 dstr_vputf(d
, fmt
+ 1, &ap
);
253 } else if (*fmt
== '?') {
254 if (strcmp(fmt
, "?ADDR") == 0) {
255 const addr
*a
= va_arg(ap
, const addr
*);
256 switch (a
->sa
.sa_family
) {
258 u_quotify(d
, "INET");
259 u_quotify(d
, inet_ntoa(a
->sin
.sin_addr
));
260 dstr_putf(d
, " %u", (unsigned)ntohs(a
->sin
.sin_port
));
265 } else if (strcmp(fmt
, "?B64") == 0) {
266 const octet
*p
= va_arg(ap
, const octet
*);
267 size_t n
= va_arg(ap
, size_t);
273 base64_encode(&b64
, p
, n
, d
);
274 base64_encode(&b64
, 0, 0, d
);
275 while (d
->len
&& d
->buf
[d
->len
- 1] == '=') d
->len
--;
276 } else if (strcmp(fmt
, "?TOKENS") == 0) {
277 const char *const *av
= va_arg(ap
, const char *const *);
278 while (*av
) u_quotify(d
, *av
++);
279 } else if (strcmp(fmt
, "?PEER") == 0)
280 u_quotify(d
, p_name(va_arg(ap
, peer
*)));
281 else if (strcmp(fmt
, "?ERRNO") == 0) {
282 dstr_putf(d
, " E%d", errno
);
283 u_quotify(d
, strerror(errno
));
287 if (*fmt
== '!') fmt
++;
289 dstr_vputf(&dd
, fmt
, &ap
);
290 u_quotify(d
, dd
.buf
);
292 fmt
= va_arg(ap
, const char *);
298 /* --- @a_write@, @a_vwrite@ --- *
300 * Arguments: @admin *a@ = admin connection to write to
301 * @const char *status@ = status code to report
302 * @const char *tag@ = tag string, or null
303 * @const char *fmt@ = pointer to format string
304 * @va_list ap@ = arguments in list
305 * @...@ = other arguments
309 * Use: Sends a message to an admin connection.
312 static void a_vwrite(admin
*a
, const char *status
, const char *tag
,
313 const char *fmt
, va_list ap
)
317 if (tag
) dstr_puts(&d
, "BG");
318 dstr_puts(&d
, status
);
319 if (tag
) u_quotify(&d
, tag
);
320 a_vformat(&d
, fmt
, ap
);
322 dosend(a
, d
.buf
, d
.len
);
326 static void a_write(admin
*a
, const char *status
, const char *tag
,
327 const char *fmt
, ...)
332 a_vwrite(a
, status
, tag
, fmt
, ap
);
336 /* --- @a_ok@, @a_info@, @a_fail@ --- *
338 * Arguments: @admin *a@ = connection
339 * @const char *fmt@ = format string
340 * @...@ = other arguments
344 * Use: Convenience functions for @a_write@.
347 static void a_ok(admin
*a
) { a_write(a
, "OK", 0, A_END
); }
349 static void a_info(admin
*a
, const char *fmt
, ...)
354 a_vwrite(a
, "INFO", 0, fmt
, ap
);
358 static void a_fail(admin
*a
, const char *fmt
, ...)
363 a_vwrite(a
, "FAIL", 0, fmt
, ap
);
367 /* --- @a_alert@, @a_valert@, @a_rawalert@ --- *
369 * Arguments: @unsigned f_and, f_eq@ = filter for connections
370 * @const char *status@ = status string
371 * @const char *fmt@ = pointer to format string
372 * @const char *p@ = pointer to raw string
373 * @size_t sz@ = size of raw string
374 * @va_list ap@ = arguments in list
375 * @...@ = other arguments
379 * Use: Write a message to all admin connections matched by the given
383 static void a_rawalert(unsigned f_and
, unsigned f_eq
, const char *status
,
384 const char *p
, size_t sz
)
389 if (!(flags
& F_INIT
))
391 dstr_puts(&d
, status
);
394 dstr_putm(&d
, p
, sz
);
399 for (a
= admins
; a
; a
= aa
) {
401 if ((a
->f
& f_and
) == f_eq
)
402 dosend(a
, d
.buf
, d
.len
);
407 static void a_valert(unsigned f_and
, unsigned f_eq
, const char *status
,
408 const char *fmt
, va_list ap
)
412 if (!(flags
& F_INIT
))
414 a_vformat(&d
, fmt
, ap
);
415 a_rawalert(f_and
, f_eq
, status
, fmt ? d
.buf
: 0, fmt ? d
.len
: 0);
419 static void a_alert(unsigned f_and
, unsigned f_eq
, const char *status
,
420 const char *fmt
, ...)
425 a_valert(f_and
, f_eq
, status
, fmt
, ap
);
429 /* --- @a_warn@ --- *
431 * Arguments: @const char *fmt@ = pointer to format string
432 * @...@ = other arguments
436 * Use: Informs all admin connections of a warning.
439 void a_warn(const char *fmt
, ...)
445 a_valert(0, 0, "WARN", fmt
, ap
);
448 fprintf(stderr
, "%s: ", QUIS
);
449 a_vformat(&d
, fmt
, ap
);
451 dstr_write(&d
, stderr
);
457 /* --- @a_trace@ --- *
459 * Arguments: @const char *p@ = pointer to a buffer
460 * @size_t sz@ = size of the buffer
461 * @void *v@ = uninteresting pointer
465 * Use: Custom trace output handler. Sends trace messages to
466 * interested admin connections.
470 static void a_trace(const char *p
, size_t sz
, void *v
)
471 { a_rawalert(AF_TRACE
, AF_TRACE
, "TRACE", p
, sz
); }
474 /* --- @a_notify@ --- *
476 * Arguments: @const char *fmt@ = pointer to format string
477 * @...@ = other arguments
481 * Use: Sends a notification to interested admin connections.
484 void a_notify(const char *fmt
, ...)
489 a_valert(AF_NOTE
, AF_NOTE
, "NOTE", fmt
, ap
);
493 /* --- @a_quit@ --- *
499 * Use: Shuts things down nicely.
508 while ((p
= p_first()) != 0)
513 /* --- @a_sigdie@ --- *
515 * Arguments: @int sig@ = signal number
516 * @void *v@ = an uninteresting argument
520 * Use Shuts down on receipt of a fatal signal.
523 static void a_sigdie(int sig
, void *v
)
529 case SIGTERM
: p
= "SIGTERM"; break;
530 case SIGINT
: p
= "SIGINT"; break;
532 sprintf(buf
, "%i", sig
);
536 a_warn("SERVER", "quit", "signal", "%s", p
, A_END
);
540 /* --- @a_sighup@ --- *
542 * Arguments: @int sig@ = signal number
543 * @void *v@ = an uninteresting argument
547 * Use Logs a message about SIGHUP not being useful.
550 static void a_sighup(int sig
, void *v
)
551 { a_warn("SERVER", "ignore", "signal", "SIGHUP", A_END
); }
553 /* --- @a_parsetime@ --- *
555 * Arguments; @const char *p@ = time string to parse
557 * Returns: Time in seconds, or @< 0@ on error.
560 static long a_parsetime(const char *p
)
563 long t
= strtol(p
, &q
, 0);
569 case 's': if (q
[1] != 0)
576 /* --- @a_findpeer@ --- *
578 * Arguments: @admin *a@ = admin connection
579 * @const char *pn@ = peer name
581 * Returns: The peer, or null if not there.
583 * Use: Finds a peer, reporting an error if it failed.
586 static peer
*a_findpeer(admin
*a
, const char *pn
)
590 if ((p
= p_find(pn
)) == 0)
591 a_fail(a
, "unknown-peer", "%s", pn
, A_END
);
595 /*----- Backgrounded operations -------------------------------------------*/
598 (((admin_bgop *)(bg))->tag ? ((admin_bgop *)(bg))->tag : "<foreground>")
600 /* --- @a_bgfind@ --- *
602 * Arguments: @admin *a@ = a client block
603 * @const char *tag@ = the requested tag
605 * Returns: The requested background job, or null.
608 static admin_bgop
*a_bgfind(admin
*a
, const char *tag
)
612 for (bg
= a
->bg
; bg
; bg
= bg
->next
) {
613 if (bg
->tag
&& strcmp(tag
, bg
->tag
) == 0)
619 /* --- @a_bgrelease@ --- *
621 * Arguments: @admin_bgop *bg@ = backgrounded operation
625 * Use: Removes a backgrounded operation from the queue, since
626 * (presumably) it's done.
629 static void a_bgrelease(admin_bgop
*bg
)
633 T( trace(T_ADMIN
, "admin: release bgop %s", BGTAG(bg
)); )
634 if (bg
->tag
) xfree(bg
->tag
);
635 else selbuf_enable(&a
->b
);
636 if (bg
->next
) bg
->next
->prev
= bg
->prev
;
637 if (bg
->prev
) bg
->prev
->next
= bg
->next
;
638 else a
->bg
= bg
->next
;
640 if (!a
->bg
&& (a
->f
& AF_CLOSE
)) a_destroy(a
);
643 /* --- @a_bgok@, @a_bginfo@, @a_bgfail@ --- *
645 * Arguments: @admin_bgop *bg@ = backgrounded operation
646 * @const char *fmt@ = format string
647 * @...@ = other arguments
651 * Use: Convenience functions for @a_write@.
654 static void a_bgok(admin_bgop
*bg
)
655 { a_write(bg
->a
, "OK", bg
->tag
, A_END
); }
657 static void a_bginfo(admin_bgop
*bg
, const char *fmt
, ...)
661 a_vwrite(bg
->a
, "INFO", bg
->tag
, fmt
, ap
);
665 static void a_bgfail(admin_bgop
*bg
, const char *fmt
, ...)
669 a_vwrite(bg
->a
, "FAIL", bg
->tag
, fmt
, ap
);
673 /* --- @a_bgadd@ --- *
675 * Arguments: @admin *a@ = administration connection
676 * @admin_bgop *bg@ = pointer to background operation
677 * @const char *tag@ = background tag, or null for foreground
678 * @void (*cancel)(admin_bgop *)@ = cancel function
680 * Returns: Zero for success, nonzero on failure.
682 * Use: Links a background job into the list.
685 static int a_bgadd(admin
*a
, admin_bgop
*bg
, const char *tag
,
686 void (*cancel
)(admin_bgop
*))
689 if (a_bgfind(a
, tag
)) {
690 a_fail(a
, "tag-exists", "%s", tag
, A_END
);
693 bg
->tag
= xstrdup(tag
);
697 selbuf_disable(&a
->b
);
703 if (a
->bg
) a
->bg
->prev
= bg
;
705 T( trace(T_ADMIN
, "admin: add bgop %s", BGTAG(bg
)); )
706 if (tag
) a_write(a
, "DETACH", tag
, A_END
);
710 /*----- Job table manipulation --------------------------------------------*/
713 #define JOB_INDEXMASK ((1ul << JOB_SHIFT) - 1)
714 #define JOB_SEQMASK ((1ul << (32 - JOB_SHIFT)) - 1)
716 #define JOB_END 0xfffffffful
718 static unsigned long a_joboffset
;
720 /* --- @a_jobidencode@ --- *
722 * Arguments: @admin_svcop *svc@ = pointer to a service operation
724 * Returns: A jobid for this job, in an internal static buffer.
726 * Use: Constructs a jobid. In order to dissuade people from
727 * predicting jobids, we obfuscate them.
729 * A `raw' jobid consists of two 16-bit fields. The low 16 bits
730 * are an index into a big array. The high 16 bits are a
731 * sequence number recording how many times that slot has been
734 * This `raw' jobid is then obfuscated by adding a randomly-
735 * generated offset, and multiplying (mod %$2^{32}$%) by a fixed
739 static const char *a_jobidencode(admin_svcop
*svc
)
741 admin_jobtable
*j
= &svc
->prov
->j
;
746 assert(svc
->index
<= JOB_INDEXMASK
);
747 seq
= j
->v
[svc
->index
].seq
;
748 assert(seq
<= JOB_SEQMASK
);
749 pre
= (unsigned long)svc
->index
| ((unsigned long)seq
<< JOB_SHIFT
);
750 sprintf(buf
, "J%08lx", ((pre
+ a_joboffset
) * 0x0f87a7a3ul
) & 0xffffffff);
754 /* --- @a_jobiddecode@ --- *
756 * Arguments: @admin_jobtable *j@ = pointer to a job table
757 * @const char *jid@ = pointer to a jobid string
759 * Returns: A pointer to the job's @svcop@ structure.
762 static admin_svcop
*a_jobiddecode(admin_jobtable
*j
, const char *jid
)
769 for (i
= 1; i
< 9; i
++) {
770 if (!isxdigit((unsigned char)jid
[i
]))
775 pre
= strtoul(jid
+ 1, 0, 16);
776 pre
= ((pre
* 0xbd11c40bul
) - a_joboffset
) & 0xffffffff;
777 i
= pre
& JOB_INDEXMASK
;
778 if (i
>= j
->n
|| j
->v
[i
].seq
!= (pre
>> JOB_SHIFT
))
780 return (j
->v
[i
].u
.op
);
783 /* --- @a_jobcreate@ --- *
785 * Arguments: @admin *a@ = pointer to administration client
787 * Returns: A pointer to a freshly-allocated @svcop@, or null.
789 * Use: Allocates a fresh @svcop@ and links it into a job table.
792 static admin_svcop
*a_jobcreate(admin
*a
)
797 admin_jobtable
*j
= &a
->j
;
799 if (j
->free
!= JOB_END
) {
801 j
->free
= j
->v
[i
].u
.next
;
804 if (j
->sz
> JOB_INDEXMASK
)
809 j
->v
= xmalloc(j
->sz
* sizeof(*j
->v
));
812 j
->v
= xrealloc(j
->v
, j
->sz
* sizeof(*j
->v
), sz
* sizeof(*j
->v
));
818 svc
= xmalloc(sizeof(*svc
));
821 svc
->next
= j
->active
;
823 if (j
->active
) j
->active
->prev
= svc
;
826 IF_TRACING(T_ADMIN
, {
827 trace(T_ADMIN
, "admin: created job %s (%u)", a_jobidencode(svc
), i
);
832 /* --- @a_jobdestroy@ --- *
834 * Arguments: @admin_svcop *svc@ = pointer to job block
838 * Use: Frees up a completed (or cancelled) job.
841 static void a_jobdestroy(admin_svcop
*svc
)
843 admin
*a
= svc
->prov
;
844 admin_jobtable
*j
= &a
->j
;
845 unsigned i
= svc
->index
;
847 IF_TRACING(T_ADMIN
, {
848 trace(T_ADMIN
, "admin: destroying job %s (%u)", a_jobidencode(svc
), i
);
850 assert(j
->v
[i
].u
.op
= svc
);
851 j
->v
[i
].u
.next
= j
->free
;
854 if (svc
->next
) svc
->next
->prev
= svc
->prev
;
855 if (svc
->prev
) svc
->prev
->next
= svc
->next
;
856 else j
->active
= svc
->next
;
859 /* --- @a_jobtableinit@ --- *
861 * Arguments: @admin_jobtable *j@ = pointer to job table
865 * Use: Initializes a job table.
868 static void a_jobtableinit(admin_jobtable
*j
)
871 a_joboffset
= GR_RANGE(&rand_global
, 0xffffffff) + 1;
878 /* --- @a_jobtablefinal@ --- *
880 * Arguments: @admin_jobtable *j@ = pointer to job table
884 * Use: Closes down a job table.
887 static void a_jobtablefinal(admin_jobtable
*j
)
889 admin_svcop
*svc
, *ssvc
;
891 for (svc
= j
->active
; svc
; svc
= ssvc
) {
893 a_bgfail(&svc
->bg
, "provider-failed", A_END
);
894 a_bgrelease(&svc
->bg
);
896 if (j
->v
) xfree(j
->v
);
899 /*----- Services infrastructure -------------------------------------------*/
901 /* --- @a_svcfind@ --- *
903 * Arguments: @admin *a@ = the requesting client
904 * @const char *name@ = service name wanted
906 * Returns: The service requested, or null.
908 * Use: Finds a service; reports an error if the service couldn't be
912 static admin_service
*a_svcfind(admin
*a
, const char *name
)
916 if ((svc
= sym_find(&a_svcs
, name
, -1, 0, 0)) == 0) {
917 a_fail(a
, "unknown-service", "%s", name
, A_END
);
923 /* --- @a_svcunlink@ --- *
925 * Arguments: @admin_service *svc@ = pointer to service structure
929 * Use: Unlinks the service from its provider, without issuing a
930 * message or freeing the structure. The version string is
934 static void a_svcunlink(admin_service
*svc
)
937 svc
->next
->prev
= svc
->prev
;
939 svc
->prev
->next
= svc
->next
;
941 svc
->prov
->svcs
= svc
->next
;
945 /* --- @a_svcrelease@ --- *
947 * Arguments: @admin_service *svc@ = pointer to service structure
951 * Use: Releases a service and frees its structure.
954 static void a_svcrelease(admin_service
*svc
)
956 a_notify("SVCRELEASE", "%s", SYM_NAME(svc
), A_END
);
958 sym_remove(&a_svcs
, svc
);
961 /*----- Name resolution operations ----------------------------------------*/
963 /* --- @a_resolved@ --- *
965 * Arguments: @struct hostent *h@ = pointer to resolved hostname
966 * @void *v@ = pointer to resolver operation
970 * Use: Handles a completed name resolution.
973 static void a_resolved(struct hostent
*h
, void *v
)
977 T( trace(T_ADMIN
, "admin: resop %s resolved", BGTAG(r
)); )
980 a_bgfail(&r
->bg
, "resolve-error", "%s", r
->addr
, A_END
);
981 r
->func(r
, ARES_FAIL
);
983 memcpy(&r
->sa
.sin
.sin_addr
, h
->h_addr
, sizeof(struct in_addr
));
991 /* --- @a_restimer@ --- *
993 * Arguments: @struct timeval *tv@ = timer
994 * @void *v@ = pointer to resolver operation
998 * Use: Times out a resolver.
1001 static void a_restimer(struct timeval
*tv
, void *v
)
1005 T( trace(T_ADMIN
, "admin: resop %s timeout", BGTAG(r
)); )
1006 a_bgfail(&r
->bg
, "resolver-timeout", "%s", r
->addr
, A_END
);
1007 r
->func(r
, ARES_FAIL
);
1010 a_bgrelease(&r
->bg
);
1013 /* --- @a_rescancel@ --- *
1015 * Arguments: @admin_bgop *bg@ = background operation
1019 * Use: Cancels an add operation.
1022 static void a_rescancel(admin_bgop
*bg
)
1024 admin_resop
*r
= (admin_resop
*)bg
;
1026 T( trace(T_ADMIN
, "admin: cancel resop %s", BGTAG(r
)); )
1027 r
->func(r
, ARES_FAIL
);
1033 /* --- @a_resolve@ --- *
1035 * Arguments: @admin *a@ = administration connection
1036 * @admin_resop *r@ = resolver operation to run
1037 * @const char *tag@ = background operation tag
1038 * @void (*func)(struct admin_resop *, int@ = handler function
1039 * @unsigned ac@ = number of remaining arguments
1040 * @char *av[]@ = pointer to remaining arguments
1044 * Use: Cranks up a resolver job.
1047 static void a_resolve(admin
*a
, admin_resop
*r
, const char *tag
,
1048 void (*func
)(struct admin_resop
*, int),
1049 unsigned ac
, char *av
[])
1056 /* --- Fill in the easy bits of address --- */
1058 r
->bg
.tag
= "<starting>";
1061 if (mystrieq(av
[i
], "inet")) i
++;
1063 a_fail(a
, "bad-addr-syntax", "[inet] ADDRESS PORT", A_END
);
1066 r
->sa
.sin
.sin_family
= AF_INET
;
1067 r
->sasz
= sizeof(r
->sa
.sin
);
1068 r
->addr
= xstrdup(av
[i
]);
1069 pt
= strtoul(av
[i
+ 1], &p
, 0);
1071 struct servent
*s
= getservbyname(av
[i
+ 1], "udp");
1073 a_fail(a
, "unknown-service", "%s", av
[i
+ 1], A_END
);
1076 pt
= ntohs(s
->s_port
);
1078 if (pt
== 0 || pt
>= 65536) {
1079 a_fail(a
, "invalid-port", "%lu", pt
, A_END
);
1082 r
->sa
.sin
.sin_port
= htons(pt
);
1084 /* --- Report backgrounding --- *
1086 * Do this for consistency of interface, even if we're going to get the
1087 * answer straight away.
1090 if (a_bgadd(a
, &r
->bg
, tag
, a_rescancel
))
1092 T( trace(T_ADMIN
, "admin: %u, resop %s, hostname `%s'",
1093 a
->seq
, BGTAG(r
), r
->addr
); )
1095 /* --- If the name is numeric, do it the easy way --- */
1097 if (inet_aton(av
[i
], &r
->sa
.sin
.sin_addr
)) {
1098 T( trace(T_ADMIN
, "admin: resop %s done the easy way", BGTAG(r
)); )
1101 a_bgrelease(&r
->bg
);
1105 /* --- Store everything for later and crank up the resolver --- */
1107 gettimeofday(&tv
, 0);
1108 tv
.tv_sec
+= T_RESOLVE
;
1109 sel_addtimer(&sel
, &r
->t
, &tv
, a_restimer
, r
);
1110 bres_byname(&r
->r
, r
->addr
, a_resolved
, r
);
1115 if (r
->addr
) xfree(r
->addr
);
1119 /*----- Option parsing ----------------------------------------------------*/
1121 #define OPTIONS(argc, argv, guts) do { \
1122 char **o_av = argv; \
1126 if (mystrieq(*o_av, "--")) { \
1131 if (**o_av == '-') \
1135 argc -= o_av - argv; \
1139 #define OPT(name, guts) if (mystrieq(*o_av, name)) { guts continue; }
1141 #define OPTARG(name, arg, guts) OPT(name, { \
1144 if (!arg) goto bad_syntax; \
1148 #define OPTTIME(name, arg, guts) OPTARG(name, o_arg, { \
1150 if ((arg = a_parsetime(o_arg)) < 0) { \
1151 a_fail(a, "bad-time-spec", "%s", o_arg, A_END); \
1157 /*----- Adding peers ------------------------------------------------------*/
1159 /* --- @a_doadd@ --- *
1161 * Arguments: @admin_resop *r@ = resolver operation
1162 * @int rc@ = how it worked
1166 * Use: Handles a completed resolution.
1169 static void a_doadd(admin_resop
*r
, int rc
)
1171 admin_addop
*add
= (admin_addop
*)r
;
1173 T( trace(T_ADMIN
, "admin: done add op %s", BGTAG(add
)); )
1175 if (rc
== ARES_OK
) {
1176 add
->peer
.sasz
= add
->r
.sasz
;
1177 add
->peer
.sa
= add
->r
.sa
;
1178 if (p_find(add
->peer
.name
))
1179 a_bgfail(&add
->r
.bg
, "peer-exists", "%s", add
->peer
.name
, A_END
);
1180 else if (!p_create(&add
->peer
))
1181 a_bgfail(&add
->r
.bg
, "peer-create-fail", "%s", add
->peer
.name
, A_END
);
1186 xfree(add
->peer
.name
);
1189 /* --- @acmd_add@ --- *
1191 * Arguments: @admin *a@ = connection which requested the addition
1192 * @unsigned ac@ = argument count
1193 * @char *av[]@ = pointer to the argument list
1197 * Use: Adds a new peer.
1200 static void acmd_add(admin
*a
, unsigned ac
, char *av
[])
1202 const char *tag
= 0;
1205 /* --- Set stuff up --- */
1207 add
= xmalloc(sizeof(*add
));
1210 add
->peer
.tops
= tun_default
;
1212 /* --- Parse options --- */
1215 OPTARG("-background", arg
, { tag
= arg
; })
1216 OPTARG("-tunnel", arg
, {
1220 a_fail(a
, "unknown-tunnel", "%s", arg
, A_END
);
1223 if (mystrieq(arg
, tunnels
[i
]->name
)) {
1224 add
->peer
.tops
= tunnels
[i
];
1229 OPTTIME("-keepalive", t
, { add
->peer
.t_ka
= t
; })
1232 /* --- Make sure someone's not got there already --- */
1234 if (!av
[0] || !av
[1])
1237 a_fail(a
, "peer-exists", "%s", *av
, A_END
);
1240 add
->peer
.name
= xstrdup(*av
++);
1243 /* --- Crank up the resolver --- */
1245 a_resolve(a
, &add
->r
, tag
, a_doadd
, ac
, av
);
1248 /* --- Clearing up --- */
1251 a_fail(a
, "bad-syntax", "add", "[OPTIONS] PEER ADDR ...", A_END
);
1253 if (add
->peer
.name
) xfree(add
->peer
.name
);
1258 /*----- Ping --------------------------------------------------------------*/
1260 /* --- @a_pingcancel@ --- *
1262 * Arguments: @admin_bgop *bg@ = background operation block
1266 * Use: Cancels a running ping.
1269 static void a_pingcancel(admin_bgop
*bg
)
1271 admin_pingop
*pg
= (admin_pingop
*)bg
;
1272 T( trace(T_ADMIN
, "admin: cancel ping op %s", BGTAG(pg
)); )
1273 p_pingdone(&pg
->ping
, PING_NONOTIFY
);
1276 /* --- @a_pong@ --- *
1278 * Arguments: @int rc@ = return code
1279 * @void *v@ = ping operation block
1283 * Use: Collects what happened to a ping message.
1286 static void a_pong(int rc
, void *v
)
1288 admin_pingop
*pg
= v
;
1294 gettimeofday(&tv
, 0);
1295 tv_sub(&tv
, &tv
, &pg
->pingtime
);
1296 millis
= (double)tv
.tv_sec
* 1000 + (double)tv
.tv_usec
/1000;
1297 a_bginfo(&pg
->bg
, "ping-ok", "%.1f", millis
, A_END
);
1301 a_bginfo(&pg
->bg
, "ping-timeout", A_END
);
1305 a_bginfo(&pg
->bg
, "ping-peer-died", A_END
);
1311 T( trace(T_ADMIN
, "admin: ponged ping op %s", BGTAG(pg
)); )
1312 a_bgrelease(&pg
->bg
);
1315 /* --- @acmd_ping@, @acmd_eping@ --- *
1317 * Arguments: @admin *a@ = connection which requested the ping
1318 * @unsigned ac@ = argument count
1319 * @char *av[]@ = pointer to the argument list
1323 * Use: Pings a peer.
1326 static void a_ping(admin
*a
, unsigned ac
, char *av
[],
1327 const char *cmd
, unsigned msg
)
1331 admin_pingop
*pg
= 0;
1332 const char *tag
= 0;
1335 OPTARG("-background", arg
, { tag
= arg
; })
1336 OPTTIME("-timeout", arg
, { t
= arg
; })
1338 if (!*av
|| av
[1]) goto bad_syntax
;
1339 if ((p
= a_findpeer(a
, *av
)) == 0)
1341 pg
= xmalloc(sizeof(*pg
));
1342 gettimeofday(&pg
->pingtime
, 0);
1343 if (a_bgadd(a
, &pg
->bg
, tag
, a_pingcancel
))
1345 T( trace(T_ADMIN
, "admin: ping op %s: %s to %s",
1346 BGTAG(pg
), cmd
, p_name(p
)); )
1347 if (p_pingsend(p
, &pg
->ping
, msg
, t
, a_pong
, pg
)) {
1348 a_bgfail(&pg
->bg
, "ping-send-failed", A_END
);
1349 a_bgrelease(&pg
->bg
);
1354 a_fail(a
, "bad-syntax", "%s", cmd
, "[OPTIONS] PEER", cmd
, A_END
);
1360 static void acmd_ping(admin
*a
, unsigned ac
, char *av
[])
1361 { a_ping(a
, ac
, av
, "ping", MISC_PING
); }
1362 static void acmd_eping(admin
*a
, unsigned ac
, char *av
[])
1363 { a_ping(a
, ac
, av
, "eping", MISC_EPING
); }
1365 /*----- Service commands --------------------------------------------------*/
1367 static void acmd_svcclaim(admin
*a
, unsigned ac
, char *av
[])
1372 svc
= sym_find(&a_svcs
, av
[0], -1, sizeof(*svc
), &f
);
1374 if (versioncmp(av
[1], svc
->version
) <= 0) {
1377 "%s", SYM_NAME(svc
),
1382 a_write(svc
->prov
, "SVCCLAIM", 0, "%s", av
[0], "%s", av
[1], A_END
);
1386 svc
->version
= xstrdup(av
[1]);
1387 svc
->next
= a
->svcs
;
1389 if (a
->svcs
) a
->svcs
->prev
= svc
;
1391 a_notify("SVCCLAIM", "%s", SYM_NAME(svc
), "%s", svc
->version
, A_END
);
1395 static void acmd_svcrelease(admin
*a
, unsigned ac
, char *av
[])
1399 if ((svc
= a_svcfind(a
, av
[0])) == 0)
1401 if (svc
->prov
!= a
) {
1402 a_fail(a
, "not-service-provider", "%s", SYM_NAME(svc
), A_END
);
1409 static void acmd_svcensure(admin
*a
, unsigned ac
, char *av
[])
1413 if ((svc
= a_svcfind(a
, av
[0])) == 0)
1415 if (av
[1] && versioncmp(svc
->version
, av
[1]) < 0) {
1416 a_fail(a
, "service-too-old",
1417 "%s", SYM_NAME(svc
),
1425 static void acmd_svcquery(admin
*a
, unsigned ac
, char *av
[])
1429 if ((svc
= a_svcfind(a
, av
[0])) != 0) {
1430 a_info(a
, "name=%s", SYM_NAME(svc
), "version=%s", svc
->version
, A_END
);
1435 static void acmd_svclist(admin
*a
, unsigned ac
, char *av
[])
1440 for (sym_mkiter(&i
, &a_svcs
); (svc
= sym_next(&i
)) != 0; )
1441 a_info(a
, "%s", SYM_NAME(svc
), "%s", svc
->version
, A_END
);
1445 static void a_canceljob(admin_bgop
*bg
)
1447 admin_svcop
*svc
= (admin_svcop
*)bg
;
1449 a_write(svc
->prov
, "SVCCANCEL", 0, "%s", a_jobidencode(svc
), A_END
);
1453 static void acmd_svcsubmit(admin
*a
, unsigned ac
, char *av
[])
1455 const char *tag
= 0;
1458 const char *ver
= 0;
1462 OPTARG("-background", arg
, { tag
= arg
; })
1463 OPTARG("-version", arg
, { ver
= arg
; })
1465 if (!*av
) goto bad_syntax
;
1466 if ((svc
= a_svcfind(a
, *av
)) == 0) goto fail
;
1467 if (ver
&& versioncmp(svc
->version
, ver
) < 0) {
1468 a_fail(a
, "service-too-old",
1469 "%s", SYM_NAME(svc
),
1474 if ((svcop
= a_jobcreate(svc
->prov
)) == 0) {
1475 a_fail(a
, "provider-overloaded", A_END
);
1478 if (a_bgadd(a
, &svcop
->bg
, tag
, a_canceljob
)) {
1479 a_jobdestroy(svcop
);
1482 a_write(svc
->prov
, "SVCJOB", 0,
1483 "%s", a_jobidencode(svcop
),
1489 a_fail(a
, "bad-syntax", "svcsubmit", "[OPTIONS] SERVICE ARGS...", A_END
);
1496 static void a_replycmd(admin
*a
, const char *status
, int donep
, char *av
[])
1500 if ((svc
= a_jobiddecode(&a
->j
, av
[0])) == 0) {
1501 a_fail(a
, "unknown-jobid", av
[0], A_END
);
1504 a_write(svc
->bg
.a
, status
, svc
->bg
.tag
, "?TOKENS", av
+ 1, A_END
);
1507 a_bgrelease(&svc
->bg
);
1512 static void acmd_svcok(admin
*a
, unsigned ac
, char *av
[])
1513 { a_replycmd(a
, "OK", 1, av
); }
1514 static void acmd_svcinfo(admin
*a
, unsigned ac
, char *av
[])
1515 { a_replycmd(a
, "INFO", 0, av
); }
1516 static void acmd_svcfail(admin
*a
, unsigned ac
, char *av
[])
1517 { a_replycmd(a
, "FAIL", 1, av
); }
1519 /*----- Administration commands -------------------------------------------*/
1521 /* --- Miscellaneous commands --- */
1523 /* --- @traceish@ --- *
1525 * Arguments: @admin *a@ = connection to complain on
1526 * @unsigned ac@ = number of arguments
1527 * @char *av[]@ = vector of arguments
1528 * @const char *what@ = what we're messing with
1529 * @const trace_opt *tt@ = options table
1530 * @unsigned *ff@ = where the flags are
1532 * Returns: Nonzero if anything changed.
1534 * Use: Guts of trace-ish commands like `trace' and `watch'.
1537 static int traceish(admin
*a
, unsigned ac
, char *av
[],
1538 const char *what
, const trace_opt
*tt
, unsigned *ff
)
1542 if (!ac
|| strcmp(av
[0], "?") == 0) {
1544 for (t
= tt
; t
->ch
; t
++) {
1545 a_info(a
, "*%c%c %s",
1546 t
->ch
, (*ff
& t
->f
) == t
->f ?
'+' : ' ', t
->help
, A_END
);
1556 case '+': sense
= 1; break;
1557 case '-': sense
= 0; break;
1559 for (t
= tt
; t
->ch
; t
++) {
1561 if (sense
) f
|= t
->f
;
1566 a_fail(a
, "bad-%s-option", what
, "%c", *p
, A_END
);
1582 static void acmd_trace(admin
*a
, unsigned ac
, char *av
[])
1584 if (traceish(a
, ac
, av
, "trace", tr_opts
, &tr_flags
))
1585 trace_level(tr_flags
);
1590 static void acmd_watch(admin
*a
, unsigned ac
, char *av
[])
1591 { traceish(a
, ac
, av
, "watch", w_opts
, &a
->f
); }
1593 static void alertcmd(admin
*a
, unsigned f_and
, unsigned f_eq
,
1594 const char *status
, char *av
[])
1595 { a_alert(f_and
, f_eq
, status
, "USER", "?TOKENS", av
, A_END
); a_ok(a
); }
1596 static void acmd_notify(admin
*a
, unsigned ac
, char *av
[])
1597 { alertcmd(a
, AF_NOTE
, AF_NOTE
, "NOTE", av
); }
1598 static void acmd_warn(admin
*a
, unsigned ac
, char *av
[])
1599 { alertcmd(a
, AF_WARN
, AF_WARN
, "WARN", av
); }
1601 static void acmd_port(admin
*a
, unsigned ac
, char *av
[])
1602 { a_info(a
, "%u", p_port(), A_END
); a_ok(a
); }
1604 static void acmd_daemon(admin
*a
, unsigned ac
, char *av
[])
1606 if (flags
& F_DAEMON
)
1607 a_fail(a
, "already-daemon", A_END
);
1609 a_notify("DAEMON", A_END
);
1613 a_fail(a
, "daemon-error", "?ERRNO", A_END
);
1621 static void acmd_jobs(admin
*a
, unsigned ac
, char *av
[])
1625 for (bg
= a
->bg
; bg
; bg
= bg
->next
) {
1627 a_info(a
, "%s", bg
->tag
, A_END
);
1632 static void acmd_bgcancel(admin
*a
, unsigned ac
, char *av
[])
1636 if ((bg
= a_bgfind(a
, av
[0])) == 0)
1637 a_fail(a
, "unknown-tag", "%s", av
[0], A_END
);
1645 static void acmd_list(admin
*a
, unsigned ac
, char *av
[])
1649 for (p
= p_first(); p
; p
= p_next(p
))
1650 a_info(a
, "%s", p_name(p
), A_END
);
1654 static void acmd_ifname(admin
*a
, unsigned ac
, char *av
[])
1658 if ((p
= a_findpeer(a
, av
[0])) != 0) {
1659 a_info(a
, "%s", p_ifname(p
), A_END
);
1664 static void acmd_setifname(admin
*a
, unsigned ac
, char *av
[])
1668 if ((p
= a_findpeer(a
, av
[0])) != 0) {
1669 a_notify("NEWIFNAME", "?PEER", p
, "%s", p_ifname(p
), "%s", av
[1], A_END
);
1670 p_setifname(p
, av
[1]);
1675 static void acmd_getchal(admin
*a
, unsigned ac
, char *av
[])
1679 buf_init(&b
, buf_i
, PKBUFSZ
);
1681 a_info(a
, "?B64", BBASE(&b
), (size_t)BLEN(&b
), A_END
);
1685 static void acmd_checkchal(admin
*a
, unsigned ac
, char *av
[])
1692 base64_decode(&b64
, av
[0], strlen(av
[0]), &d
);
1693 base64_decode(&b64
, 0, 0, &d
);
1694 buf_init(&b
, d
.buf
, d
.len
);
1695 if (c_check(&b
) || BBAD(&b
) || BLEFT(&b
))
1696 a_fail(a
, "invalid-challenge", A_END
);
1702 static void acmd_greet(admin
*a
, unsigned ac
, char *av
[])
1708 if ((p
= a_findpeer(a
, av
[0])) != 0) {
1710 base64_decode(&b64
, av
[1], strlen(av
[1]), &d
);
1711 base64_decode(&b64
, 0, 0, &d
);
1712 p_greet(p
, d
.buf
, d
.len
);
1718 static void acmd_addr(admin
*a
, unsigned ac
, char *av
[])
1723 if ((p
= a_findpeer(a
, av
[0])) != 0) {
1725 assert(ad
->sa
.sa_family
== AF_INET
);
1726 a_info(a
, "?ADDR", ad
, A_END
);
1731 static void acmd_peerinfo(admin
*a
, unsigned ac
, char *av
[])
1736 if ((p
= a_findpeer(a
, av
[0])) != 0) {
1738 a_info(a
, "tunnel=%s", ps
->tops
->name
, A_END
);
1739 a_info(a
, "keepalive=%lu", ps
->t_ka
, A_END
);
1744 static void acmd_servinfo(admin
*a
, unsigned ac
, char *av
[])
1746 a_info(a
, "implementation=edgeware-tripe", A_END
);
1747 a_info(a
, "version=%s", VERSION
, A_END
);
1748 a_info(a
, "daemon=%s", BOOL(flags
& F_DAEMON
), A_END
);
1752 static void acmd_stats(admin
*a
, unsigned ac
, char *av
[])
1757 if ((p
= a_findpeer(a
, av
[0])) == 0)
1761 a_info(a
, "start-time=%s", timestr(st
->t_start
), A_END
);
1762 a_info(a
, "last-packet-time=%s", timestr(st
->t_last
), A_END
);
1763 a_info(a
, "last-keyexch-time=%s", timestr(st
->t_kx
), A_END
);
1764 a_info(a
, "packets-in=%lu", st
->n_in
, "bytes-in=%lu", st
->sz_in
, A_END
);
1766 "packets-out=%lu", st
->n_out
,
1767 "bytes-out=%lu", st
->sz_out
,
1770 "keyexch-packets-in=%lu", st
->n_kxin
,
1771 "keyexch-bytes-in=%lu", st
->sz_kxin
,
1774 "keyexch-packets-out=%lu", st
->n_kxout
,
1775 "keyexch-bytes-out=%lu", st
->sz_kxout
,
1778 "ip-packets-in=%lu", st
->n_ipin
,
1779 "ip-bytes-in=%lu", st
->sz_ipin
,
1782 "ip-packets-out=%lu", st
->n_ipout
,
1783 "ip-bytes-out=%lu", st
->sz_ipout
,
1785 a_info(a
, "rejected-packets=%lu", st
->n_reject
, A_END
);
1789 static void acmd_kill(admin
*a
, unsigned ac
, char *av
[])
1793 if ((p
= a_findpeer(a
, av
[0])) != 0) {
1799 static void acmd_forcekx(admin
*a
, unsigned ac
, char *av
[])
1803 if ((p
= a_findpeer(a
, av
[0])) != 0) {
1804 kx_start(&p
->kx
, 1);
1809 static void acmd_reload(admin
*a
, unsigned ac
, char *av
[])
1810 { p_keyreload(); a_ok(a
); }
1812 static void acmd_quit(admin
*a
, unsigned ac
, char *av
[])
1814 a_warn("SERVER", "quit", "admin-request", A_END
);
1819 static void acmd_version(admin
*a
, unsigned ac
, char *av
[])
1821 a_info(a
, "%s", PACKAGE
, "%s", VERSION
, A_END
);
1825 static void acmd_tunnels(admin
*a
, unsigned ac
, char *av
[])
1829 for (i
= 0; tunnels
[i
]; i
++)
1830 a_info(a
, "%s", tunnels
[i
]->name
, A_END
);
1834 /* --- The command table and help --- */
1836 typedef struct acmd
{
1839 unsigned argmin
, argmax
;
1840 void (*func
)(admin */
*a*/
, unsigned /*ac*/, char */
*av*/
[]);
1843 static void acmd_help(admin */
*a*/
, unsigned /*ac*/, char */
*av*/
[]);
1845 static const acmd acmdtab
[] = {
1846 { "add", "[OPTIONS] PEER ADDR ...", 2, 0xffff, acmd_add
},
1847 { "addr", "PEER", 1, 1, acmd_addr
},
1848 { "bgcancel", "TAG", 1, 1, acmd_bgcancel
},
1849 { "checkchal", "CHAL", 1, 1, acmd_checkchal
},
1850 { "daemon", 0, 0, 0, acmd_daemon
},
1851 { "eping", "[OPTIONS] PEER", 1, 0xffff, acmd_eping
},
1852 { "forcekx", "PEER", 1, 1, acmd_forcekx
},
1853 { "getchal", 0, 0, 0, acmd_getchal
},
1854 { "greet", "PEER CHAL", 2, 2, acmd_greet
},
1855 { "help", 0, 0, 0, acmd_help
},
1856 { "ifname", "PEER", 1, 1, acmd_ifname
},
1857 { "jobs", 0, 0, 0, acmd_jobs
},
1858 { "kill", "PEER", 1, 1, acmd_kill
},
1859 { "list", 0, 0, 0, acmd_list
},
1860 { "notify", "MESSAGE ...", 1, 0xffff, acmd_notify
},
1861 { "peerinfo", "PEER", 1, 1, acmd_peerinfo
},
1862 { "ping", "[OPTIONS] PEER", 1, 0xffff, acmd_ping
},
1863 { "port", 0, 0, 0, acmd_port
},
1864 { "quit", 0, 0, 0, acmd_quit
},
1865 { "reload", 0, 0, 0, acmd_reload
},
1866 { "servinfo", 0, 0, 0, acmd_servinfo
},
1867 { "setifname", "PEER NEW-NAME", 2, 2, acmd_setifname
},
1868 { "svcclaim", "SERVICE VERSION", 2, 2, acmd_svcclaim
},
1869 { "svcensure", "SERVICE [VERSION]", 1, 2, acmd_svcensure
},
1870 { "svcfail", "JOBID TOKENS...", 1, 0xffff, acmd_svcfail
},
1871 { "svcinfo", "JOBID TOKENS...", 1, 0xffff, acmd_svcinfo
},
1872 { "svclist", 0, 0, 0, acmd_svclist
},
1873 { "svcok", "JOBID", 1, 1, acmd_svcok
},
1874 { "svcquery", "SERVICE", 1, 1, acmd_svcquery
},
1875 { "svcrelease", "SERVICE", 1, 1, acmd_svcrelease
},
1876 { "svcsubmit", "[OPTIONS] SERVICE TOKENS...",
1877 2, 0xffff, acmd_svcsubmit
},
1878 { "stats", "PEER", 1, 1, acmd_stats
},
1880 { "trace", "[OPTIONS]", 0, 1, acmd_trace
},
1882 { "tunnels", 0, 0, 0, acmd_tunnels
},
1883 { "version", 0, 0, 0, acmd_version
},
1884 { "warn", "MESSAGE ...", 1, 0xffff, acmd_warn
},
1885 { "watch", "[OPTIONS]", 0, 1, acmd_watch
},
1889 static void acmd_help(admin
*a
, unsigned ac
, char *av
[])
1893 for (c
= acmdtab
; c
->name
; c
++) {
1895 a_info(a
, "%s", c
->name
, "*%s", c
->help
, A_END
);
1897 a_info(a
, "%s", c
->name
, A_END
);
1902 /*----- Connection handling -----------------------------------------------*/
1904 /* --- @a_destroypending@ --- *
1910 * Use: Destroys pending admin connections at a safe time.
1913 static void a_destroypending(void)
1916 admin_bgop
*bg
, *bbg
;
1917 admin_service
*svc
, *ssvc
;
1919 /* --- Destroy connections marked as pending --- */
1921 for (a
= a_dead
; a
; a
= aa
) {
1923 assert(a
->f
& AF_DEAD
);
1925 /* --- Report what we're doing --- */
1927 T( trace(T_ADMIN
, "admin: completing destruction of connection %u",
1930 /* --- Abort any background jobs in progress --- */
1932 for (bg
= a
->bg
; bg
; bg
= bbg
) {
1935 if (bg
->tag
) xfree(bg
->tag
);
1939 /* --- Release services I hold, and abort pending jobs --- */
1941 for (svc
= a
->svcs
; svc
; svc
= ssvc
) {
1945 a_jobtablefinal(&a
->j
);
1947 /* --- Close file descriptors and selectory --- */
1949 selbuf_destroy(&a
->b
);
1950 if (a
->b
.reader
.fd
!= a
->w
.fd
) close(a
->b
.reader
.fd
);
1952 if (a_stdin
== a
) a_stdin
= 0;
1959 /* --- All pending destruction completed --- */
1964 /* --- @a_destroy@ --- *
1966 * Arguments: @admin *a@ = pointer to an admin block
1970 * Use: Destroys an admin block. This requires a certain amount of
1974 static void freequeue(oqueue
*q
)
1978 for (o
= q
->hd
; o
; o
= oo
) {
1985 static void a_destroy(admin
*a
)
1990 if (a
->next
) a
->next
->prev
= a
->prev
;
1991 if (a
->prev
) a
->prev
->next
= a
->next
;
1992 else admins
= a
->next
;
1994 if (a
->out
.hd
) sel_rmfile(&a
->w
);
2001 T( trace(T_ADMIN
, "admin: killing connection %u", a
->seq
); )
2004 /* --- @a_line@ --- *
2006 * Arguments: @char *p@ = pointer to the line read
2007 * @size_t len@ = length of the line
2008 * @void *vp@ = pointer to my admin block
2012 * Use: Handles a line of input.
2015 static void a_line(char *p
, size_t len
, void *vp
)
2030 selbuf_disable(&a
->b
);
2034 ac
= str_qsplit(p
, av
, 16, 0, STRF_QUOTE
);
2037 for (c
= acmdtab
; c
->name
; c
++) {
2038 if (mystrieq(av
[0], c
->name
)) {
2040 if (c
->argmin
> ac
|| ac
> c
->argmax
) {
2042 a_fail(a
, "bad-syntax", "%s", c
->name
, "", A_END
);
2044 a_fail(a
, "bad-syntax", "%s", c
->name
, "%s", c
->help
, A_END
);
2046 c
->func(a
, ac
, av
+ 1);
2050 a_fail(a
, "unknown-command", "%s", av
[0], A_END
);
2053 /* --- @a_create@ --- *
2055 * Arguments: @int fd_in, fd_out@ = file descriptors to use
2056 * @unsigned f@ = initial flags to set
2060 * Use: Creates a new admin connection.
2063 void a_create(int fd_in
, int fd_out
, unsigned f
)
2065 admin
*a
= CREATE(admin
);
2067 T( static unsigned seq
= 0;
2069 T( trace(T_ADMIN
, "admin: accepted connection %u", a
->seq
); )
2073 a_jobtableinit(&a
->j
);
2075 if (fd_in
== STDIN_FILENO
) a_stdin
= a
;
2076 fdflags(fd_in
, O_NONBLOCK
, O_NONBLOCK
, FD_CLOEXEC
, FD_CLOEXEC
);
2077 if (fd_out
!= fd_in
)
2078 fdflags(fd_out
, O_NONBLOCK
, O_NONBLOCK
, FD_CLOEXEC
, FD_CLOEXEC
);
2079 selbuf_init(&a
->b
, &sel
, fd_in
, a_line
, a
);
2080 sel_initfile(&sel
, &a
->w
, fd_out
, SEL_WRITE
, a_flush
, a
);
2081 a
->out
.hd
= a
->out
.tl
= 0;
2084 if (admins
) admins
->prev
= a
;
2088 /* --- @a_accept@ --- *
2090 * Arguments: @int fd@ = file descriptor to accept
2091 * @unsigned mode@ = what to do
2092 * @void *v@ = uninteresting pointer
2096 * Use: Accepts a new admin connection.
2099 static void a_accept(int fd
, unsigned mode
, void *v
)
2102 struct sockaddr_un sun
;
2103 size_t sz
= sizeof(sun
);
2105 if ((nfd
= accept(fd
, (struct sockaddr
*)&sun
, &sz
)) < 0) {
2106 if (errno
!= EINTR
&& errno
!= EAGAIN
&& errno
!= EWOULDBLOCK
&&
2107 errno
!= ECONNABORTED
&& errno
!= EPROTO
)
2108 a_warn("ADMIN", "accept-error", "?ERRNO", A_END
);
2111 a_create(nfd
, nfd
, 0);
2114 /* --- @a_preselect@ --- *
2120 * Use: Informs the admin module that we're about to select again,
2121 * and that it should do cleanup things it has delayed until a
2125 void a_preselect(void) { if (a_dead
) a_destroypending(); }
2127 /* --- @a_daemon@ --- *
2133 * Use: Informs the admin module that it's a daemon.
2136 void a_daemon(void) { flags
|= F_DAEMON
; }
2138 /* --- @a_init@ --- *
2140 * Arguments: @const char *name@ = socket name to create
2144 * Use: Creates the admin listening socket.
2147 void a_init(const char *name
)
2151 struct sockaddr_un sun
;
2152 struct sigaction sa
;
2155 /* --- Create services table --- */
2157 sym_create(&a_svcs
);
2159 /* --- Set up the socket address --- */
2161 sz
= strlen(name
) + 1;
2162 if (sz
> sizeof(sun
.sun_path
))
2163 die(EXIT_FAILURE
, "socket name `%s' too long", name
);
2165 sun
.sun_family
= AF_UNIX
;
2166 memcpy(sun
.sun_path
, name
, sz
);
2167 sz
+= offsetof(struct sockaddr_un
, sun_path
);
2169 /* --- Attempt to bind to the socket --- */
2173 if ((fd
= socket(PF_UNIX
, SOCK_STREAM
, 0)) < 0)
2174 die(EXIT_FAILURE
, "couldn't create socket: %s", strerror(errno
));
2175 if (bind(fd
, (struct sockaddr
*)&sun
, sz
) < 0) {
2178 if (errno
!= EADDRINUSE
) {
2179 die(EXIT_FAILURE
, "couldn't bind to address `%s': %s",
2180 sun
.sun_path
, strerror(e
));
2183 die(EXIT_FAILURE
, "too many retries; giving up");
2185 if (!connect(fd
, (struct sockaddr
*)&sun
, sz
)) {
2186 die(EXIT_FAILURE
, "server already listening on admin socket `%s'",
2189 if (errno
!= ECONNREFUSED
)
2190 die(EXIT_FAILURE
, "couldn't bind to address: %s", strerror(e
));
2191 if (stat(sun
.sun_path
, &st
)) {
2192 die(EXIT_FAILURE
, "couldn't stat `%s': %s",
2193 sun
.sun_path
, strerror(errno
));
2195 if (!S_ISSOCK(st
.st_mode
))
2196 die(EXIT_FAILURE
, "object `%s' isn't a socket", sun
.sun_path
);
2197 T( trace(T_ADMIN
, "admin: stale socket found; removing it"); )
2198 unlink(sun
.sun_path
);
2202 chmod(sun
.sun_path
, 0600);
2203 fdflags(fd
, O_NONBLOCK
, O_NONBLOCK
, FD_CLOEXEC
, FD_CLOEXEC
);
2205 die(EXIT_FAILURE
, "couldn't listen on socket: %s", strerror(errno
));
2207 /* --- Listen to the socket --- */
2209 sel_initfile(&sel
, &sock
, fd
, SEL_READ
, a_accept
, 0);
2213 T( trace_custom(a_trace
, 0);
2214 trace(T_ADMIN
, "admin: enabled custom tracing"); )
2217 /* --- Set up signal handlers --- */
2219 sig_add(&s_term
, SIGTERM
, a_sigdie
, 0);
2220 sig_add(&s_hup
, SIGHUP
, a_sighup
, 0);
2221 signal(SIGPIPE
, SIG_IGN
);
2222 sigaction(SIGINT
, 0, &sa
);
2223 if (sa
.sa_handler
!= SIG_IGN
)
2224 sig_add(&s_int
, SIGINT
, a_sigdie
, 0);
2227 /*----- That's all, folks -------------------------------------------------*/