2 * Amanda, The Advanced Maryland Automatic Network Disk Archiver
3 * Copyright (c) 1991-1999 University of Maryland at College Park
6 * Permission to use, copy, modify, distribute, and sell this software and its
7 * documentation for any purpose is hereby granted without fee, provided that
8 * the above copyright notice appear in all copies and that both that
9 * copyright notice and this permission notice appear in supporting
10 * documentation, and that the name of U.M. not be used in advertising or
11 * publicity pertaining to distribution of the software without specific,
12 * written prior permission. U.M. makes no representations about the
13 * suitability of this software for any purpose. It is provided "as is"
14 * without express or implied warranty.
16 * U.M. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL U.M.
18 * BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
19 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
20 * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
21 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
23 * Authors: the Amanda Development Team. Its members are listed in a
24 * file named AUTHORS, in the root directory of this distribution.
28 * $Id: amandad.c,v 1.18 2006/08/21 20:17:09 martinea Exp $
30 * handle client-host side of Amanda network communications, including
31 * security checks, execution of the proper service, and acking the
39 #include "amfeatures.h"
47 #define REP_TIMEOUT (6*60*60) /* secs for service to reply */
48 #define ACK_TIMEOUT 10 /* XXX should be configurable */
49 #define STDERR_PIPE (DATA_FD_COUNT + 1)
51 #define amandad_debug(i, ...) do { \
52 if ((i) <= debug_amandad) { \
53 dbprintf(__VA_ARGS__); \
58 * These are the actions for entering the state machine
60 typedef enum { A_START, A_RECVPKT, A_RECVREP, A_PENDING, A_FINISH, A_CONTINUE,
61 A_SENDNAK, A_TIMEOUT } action_t;
64 * This is a state in the state machine. It is a function pointer to
65 * the function that actually implements the state.
67 struct active_service;
68 typedef action_t (*state_t)(struct active_service *, action_t, pkt_t *);
70 /* string that we scan for in sendbackup's MESG stream */
71 static const char info_end_str[] = "sendbackup: info end\n";
72 #define INFO_END_LEN (sizeof(info_end_str)-1)
75 * Here are the services that we allow.
76 * Must be in the same order as services[].
87 static struct services {
92 { "noop", 1, SERVICE_NOOP },
93 { "sendsize", 1, SERVICE_SENDSIZE },
94 { "sendbackup", 1, SERVICE_SENDBACKUP },
95 { "selfcheck", 1, SERVICE_SELFCHECK },
96 { "amindexd", 0, SERVICE_AMINDEXD },
97 { "amidxtaped", 0, SERVICE_AMIDXTAPED }
99 #define NSERVICES (int)(sizeof(services) / sizeof(services[0]))
102 * This structure describes an active running service.
104 * An active service is something running that we have received
105 * a request for. This structure holds info on that service, including
106 * file descriptors for data, etc, as well as the security handle
107 * for communications with the amanda server.
109 struct active_service {
110 service_t service; /* service name */
111 char *cmd; /* name of command we ran */
112 char *arguments; /* arguments we sent it */
113 security_handle_t *security_handle; /* remote server */
114 state_t state; /* how far this has progressed */
115 pid_t pid; /* pid of subprocess */
116 int send_partial_reply; /* send PREP packet */
117 int reqfd; /* pipe to write requests */
118 int repfd; /* pipe to read replies */
119 int errfd; /* pipe to read stderr */
120 event_handle_t *ev_repfd; /* read event handle for repfd */
121 event_handle_t *ev_reptimeout; /* timeout for rep data */
122 event_handle_t *ev_errfd; /* read event handle for errfd */
123 pkt_t rep_pkt; /* rep packet we're sending out */
124 char *errbuf; /* buffer to read the err into */
125 char *repbuf; /* buffer to read the rep into */
126 size_t bufsize; /* length of repbuf */
127 size_t repbufsize; /* length of repbuf */
128 int repretry; /* times we'll retry sending the rep */
129 int seen_info_end; /* have we seen "sendbackup info end\n"? */
130 char info_end_buf[INFO_END_LEN]; /* last few bytes read, used for scanning for info end */
133 * General user streams to the process, and their equivalent
136 struct datafd_handle {
137 int fd_read; /* pipe to child process */
138 int fd_write; /* pipe to child process */
139 event_handle_t *ev_read; /* it's read event handle */
140 event_handle_t *ev_write; /* it's write event handle */
141 security_stream_t *netfd; /* stream to amanda server */
142 struct active_service *as; /* pointer back to our enclosure */
143 } data[DATA_FD_COUNT];
144 char databuf[NETWORK_BLOCK_BYTES]; /* buffer to relay netfd data in */
148 * Queue of outstanding requests that we are running.
150 GSList *serviceq = NULL;
152 static int wait_30s = 1;
153 static int exit_on_qlength = 1;
154 static char *auth = NULL;
155 static kencrypt_type amandad_kencrypt = KENCRYPT_NONE;
157 int main(int argc, char **argv);
159 static int allocstream(struct active_service *, int);
160 static void exit_check(void *);
161 static void protocol_accept(security_handle_t *, pkt_t *);
162 static void state_machine(struct active_service *, action_t, pkt_t *);
164 static action_t s_sendack(struct active_service *, action_t, pkt_t *);
165 static action_t s_repwait(struct active_service *, action_t, pkt_t *);
166 static action_t s_processrep(struct active_service *, action_t, pkt_t *);
167 static action_t s_sendrep(struct active_service *, action_t, pkt_t *);
168 static action_t s_ackwait(struct active_service *, action_t, pkt_t *);
170 static void repfd_recv(void *);
171 static void process_errfd(void *cookie);
172 static void errfd_recv(void *);
173 static void timeout_repfd(void *);
174 static void protocol_recv(void *, pkt_t *, security_status_t);
175 static void process_readnetfd(void *);
176 static void process_writenetfd(void *, void *, ssize_t);
177 static struct active_service *service_new(security_handle_t *,
178 const char *, service_t, const char *);
179 static void service_delete(struct active_service *);
180 static int writebuf(struct active_service *, const void *, size_t);
181 static ssize_t do_sendpkt(security_handle_t *handle, pkt_t *pkt);
182 static char *amandad_get_security_conf (char *, void *);
184 static const char *state2str(state_t);
185 static const char *action2str(action_t);
195 const security_driver_t *secdrv;
197 char *pgm = "amandad"; /* in case argv[0] is not set */
198 #if defined(USE_REUSEADDR)
204 * Configure program for internationalization:
205 * 1) Only set the message locale for now.
206 * 2) Set textdomain for all amanda related programs to "amanda"
207 * We don't want to be forced to support dozens of message catalogs.
209 setlocale(LC_MESSAGES, "C");
210 textdomain("amanda");
216 * When called via inetd, it is not uncommon to forget to put the
217 * argv[0] value on the config line. On some systems (e.g. Solaris)
218 * this causes argv and/or argv[0] to be NULL, so we have to be
219 * careful getting our name.
221 if ((argv == NULL) || (argv[0] == NULL)) {
222 pgm = "amandad"; /* in case argv[0] is not set */
224 pgm = basename(argv[0]); /* Strip of leading path get debug name */
227 dbopen(DBG_SUBDIR_AMANDAD);
230 error(_("argv == NULL\n"));
234 /* Don't die when child closes pipe */
235 signal(SIGPIPE, SIG_IGN);
237 /* Parse the configuration; we'll handle errors later */
238 config_init(CONFIG_INIT_CLIENT, NULL);
240 if (geteuid() == 0) {
241 check_running_as(RUNNING_AS_ROOT);
242 initgroups(CLIENT_LOGIN, get_client_gid());
243 setgid(get_client_gid());
244 setegid(get_client_gid());
245 seteuid(get_client_uid());
247 check_running_as(RUNNING_AS_CLIENT_LOGIN);
250 add_amanda_log_handler(amanda_log_stderr);
251 add_amanda_log_handler(amanda_log_syslog);
254 * ad-hoc argument parsing
256 * We accept -auth=[authentication type]
260 * We also add a list of services that amandad can launch
263 in = 0; out = 1; /* default to stdin/stdout */
265 for (i = 1; i < argc; i++) {
267 * Get a driver for a security type specified after -auth=
269 if (strncmp(argv[i], "-auth=", strlen("-auth=")) == 0) {
270 argv[i] += strlen("-auth=");
271 secdrv = security_getdriver(argv[i]);
273 if (secdrv == NULL) {
274 error(_("no driver for security type '%s'\n"), argv[i]);
281 * If -no-exit is specified, always run even after requests have
284 else if (strcmp(argv[i], "-no-exit") == 0) {
290 * Allow us to directly bind to a udp port for debugging.
291 * This may only apply to some security types.
293 else if (strncmp(argv[i], "-udp=", strlen("-udp=")) == 0) {
295 struct sockaddr_in6 sin;
297 struct sockaddr_in sin;
300 argv[i] += strlen("-udp=");
302 in = out = socket(AF_INET6, SOCK_DGRAM, 0);
304 in = out = socket(AF_INET, SOCK_DGRAM, 0);
307 error(_("can't create dgram socket: %s\n"), strerror(errno));
311 r = setsockopt(in, SOL_SOCKET, SO_REUSEADDR,
312 (void *)&on, (socklen_t_equiv)sizeof(on));
314 dbprintf(_("amandad: setsockopt(SO_REUSEADDR) failed: %s\n"),
320 sin.sin6_family = (sa_family_t)AF_INET6;
321 sin.sin6_addr = in6addr_any;
322 sin.sin6_port = (in_port_t)htons((in_port_t)atoi(argv[i]));
324 sin.sin_family = (sa_family_t)AF_INET;
325 sin.sin_addr.s_addr = INADDR_ANY;
326 sin.sin_port = (in_port_t)htons((in_port_t)atoi(argv[i]));
328 if (bind(in, (struct sockaddr *)&sin, (socklen_t_equiv)sizeof(sin)) < 0) {
329 error(_("can't bind to port %d: %s\n"), atoi(argv[i]),
335 * Ditto for tcp ports.
337 else if (strncmp(argv[i], "-tcp=", strlen("-tcp=")) == 0) {
339 struct sockaddr_in6 sin;
341 struct sockaddr_in sin;
346 argv[i] += strlen("-tcp=");
348 sock = socket(AF_INET6, SOCK_STREAM, 0);
350 sock = socket(AF_INET, SOCK_STREAM, 0);
353 error(_("can't create tcp socket: %s\n"), strerror(errno));
357 r = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
358 (void *)&on, (socklen_t_equiv)sizeof(on));
360 dbprintf(_("amandad: setsockopt(SO_REUSEADDR) failed: %s\n"),
365 sin.sin6_family = (sa_family_t)AF_INET6;
366 sin.sin6_addr = in6addr_any;
367 sin.sin6_port = (in_port_t)htons((in_port_t)atoi(argv[i]));
369 sin.sin_family = (sa_family_t)AF_INET;
370 sin.sin_addr.s_addr = INADDR_ANY;
371 sin.sin_port = (in_port_t)htons((in_port_t)atoi(argv[i]));
373 if (bind(sock, (struct sockaddr *)&sin, (socklen_t_equiv)sizeof(sin)) < 0) {
374 error(_("can't bind to port %d: %s\n"), atoi(argv[i]),
379 n = (socklen_t_equiv)sizeof(sin);
380 in = out = accept(sock, (struct sockaddr *)&sin, &n);
383 * It must be a service name
386 /* clear all services */
388 for (j = 0; j < (int)NSERVICES; j++)
389 services[j].active = 0;
393 if(strcmp(argv[i],"amdump") == 0) {
394 services[0].active = 1;
395 services[1].active = 1;
396 services[2].active = 1;
397 services[3].active = 1;
400 for (j = 0; j < (int)NSERVICES; j++)
401 if (strcmp(services[j].name, argv[i]) == 0)
403 if (j == (int)NSERVICES) {
404 dbprintf(_("%s: invalid service\n"), argv[i]);
407 services[j].active = 1;
413 * If no security type specified, use BSD
415 if (secdrv == NULL) {
416 secdrv = security_getdriver("BSD");
418 if (secdrv == NULL) {
419 error(_("no driver for default security type 'BSD'\n"));
424 if(strcasecmp(auth, "rsh") == 0 ||
425 strcasecmp(auth, "ssh") == 0 ||
426 strcasecmp(auth, "local") == 0 ||
427 strcasecmp(auth, "bsdtcp") == 0) {
432 #ifndef SINGLE_USERID
433 if (geteuid() == 0) {
434 if (strcasecmp(auth, "krb5") != 0) {
436 /* lookup our local user name */
437 if ((pwd = getpwnam(CLIENT_LOGIN)) == NULL) {
438 error(_("getpwnam(%s) failed."), CLIENT_LOGIN);
441 if (pwd->pw_uid != 0) {
442 error(_("'amandad' must be run as user '%s' when using '%s' authentication"),
447 if (strcasecmp(auth, "krb5") == 0) {
448 error(_("'amandad' must be run as user 'root' when using 'krb5' authentication"));
457 dbprintf(_("version %s\n"), VERSION);
458 for (i = 0; version_info[i] != NULL; i++) {
459 dbprintf(" %s", version_info[i]);
462 if (! (argc >= 1 && argv != NULL && argv[0] != NULL)) {
463 dbprintf(_("WARNING: argv[0] not defined: check inetd.conf\n"));
466 /* krb5 require the euid to be 0 */
467 if (strcasecmp(auth, "krb5") == 0) {
472 * Schedule to call protocol_accept() when new security handles
473 * are created on stdin.
475 security_accept(secdrv, amandad_get_security_conf, in, out, protocol_accept, NULL);
478 * Schedule an event that will try to exit every 30 seconds if there
479 * are no requests outstanding.
482 (void)event_register((event_id_t)30, EV_TIME, exit_check, &no_exit);
485 * Call event_loop() with an arg of 0, telling it to block until all
486 * events are completed.
497 * This runs periodically and checks to see if we have any active services
498 * still running. If we don't, then we quit.
506 assert(cookie != NULL);
507 no_exit = *(int *)cookie;
510 * If things are still running, then don't exit.
512 if (g_slist_length(serviceq) > 0)
516 * If the caller asked us to never exit, then we're done
526 * Handles new incoming protocol handles. This is a callback for
527 * security_accept(), which gets called when new handles are detected.
531 security_handle_t * handle,
536 struct active_service *as;
537 char *pktbody, *tok, *service, *arguments;
538 char *service_path = NULL;
539 GSList *errlist = NULL;
545 * If handle is NULL, then the connection is closed.
552 * If we have errors (not warnings) from the config file, let the server
553 * know immediately. Unfortunately, we only get one ERROR line, so if there
554 * are multiple errors, we just show the first.
556 if (config_errors(&errlist) >= CFGERR_ERRORS) {
557 GSList *iter = errlist;
559 gboolean multiple_errors = FALSE;
562 errmsg = (char *)iter->data;
564 multiple_errors = TRUE;
566 errmsg = "(no error message)";
569 pkt_init(&pkt_out, P_NAK, "ERROR %s%s", errmsg,
570 multiple_errors? _(" (additional errors not displayed)"):"");
571 do_sendpkt(handle, &pkt_out);
572 amfree(pkt_out.body);
573 security_close(handle);
578 * If pkt is NULL, then there was a problem with the new connection.
581 dbprintf(_("accept error: %s\n"), security_geterror(handle));
582 pkt_init(&pkt_out, P_NAK, "ERROR %s\n", security_geterror(handle));
583 do_sendpkt(handle, &pkt_out);
584 amfree(pkt_out.body);
585 security_close(handle);
589 dbprintf(_("accept recv %s pkt:\n<<<<<\n%s>>>>>\n"),
590 pkt_type2str(pkt->type), pkt->body);
593 * If this is not a REQ packet, just forget about it.
595 if (pkt->type != P_REQ) {
596 dbprintf(_("received unexpected %s packet:\n<<<<<\n%s>>>>>\n\n"),
597 pkt_type2str(pkt->type), pkt->body);
598 security_close(handle);
602 pktbody = service = arguments = NULL;
606 * Parse out the service and arguments
609 pktbody = stralloc(pkt->body);
611 tok = strtok(pktbody, " ");
614 if (strcmp(tok, "SERVICE") != 0)
617 tok = strtok(NULL, " \n");
620 service = stralloc(tok);
622 /* we call everything else 'arguments' */
623 tok = strtok(NULL, "");
626 arguments = stralloc(tok);
628 /* see if it's one we allow */
629 for (i = 0; i < (int)NSERVICES; i++)
630 if (services[i].active == 1 && strcmp(services[i].name, service) == 0)
632 if (i == (int)NSERVICES) {
633 dbprintf(_("%s: invalid service\n"), service);
634 pkt_init(&pkt_out, P_NAK, _("ERROR %s: invalid service, add '%s' as argument to amandad\n"), service, service);
638 service_path = vstralloc(amlibexecdir, "/", service, NULL);
639 if (access(service_path, X_OK) < 0) {
640 dbprintf(_("can't execute %s: %s\n"), service_path, strerror(errno));
641 pkt_init(&pkt_out, P_NAK,
642 _("ERROR execute access to \"%s\" denied\n"),
647 /* see if its already running */
648 for (iter = serviceq; iter != NULL; iter = g_slist_next(iter)) {
649 as = (struct active_service *)iter->data;
650 if (strcmp(as->cmd, service_path) == 0 &&
651 strcmp(as->arguments, arguments) == 0) {
652 dbprintf(_("%s %s: already running, acking req\n"),
654 pkt_init_empty(&pkt_out, P_ACK);
655 goto send_pkt_out_no_delete;
660 * create a new service instance, and send the arguments down
663 dbprintf(_("creating new service: %s\n%s\n"), service, arguments);
664 as = service_new(handle, service_path, services[i].service, arguments);
665 if (writebuf(as, arguments, strlen(arguments)) < 0) {
666 const char *errmsg = strerror(errno);
667 dbprintf(_("error sending arguments to %s: %s\n"), service, errmsg);
668 pkt_init(&pkt_out, P_NAK, _("ERROR error writing arguments to %s: %s\n"),
676 amfree(service_path);
680 * Move to the sendack state, and start up the state
683 as->state = s_sendack;
684 state_machine(as, A_START, NULL);
688 pkt_init(&pkt_out, P_NAK, _("ERROR invalid REQ\n"));
689 dbprintf(_("received invalid %s packet:\n<<<<<\n%s>>>>>\n\n"),
690 pkt_type2str(pkt->type), pkt->body);
695 send_pkt_out_no_delete:
697 amfree(service_path);
700 do_sendpkt(handle, &pkt_out);
701 security_close(handle);
702 amfree(pkt_out.body);
706 * Handles incoming protocol packets. Routes responses to the proper
711 struct active_service * as,
719 amandad_debug(1, _("state_machine: %p entering\n"), as);
721 curstate = as->state;
722 amandad_debug(1, _("state_machine: %p curstate=%s action=%s\n"), as,
723 state2str(curstate), action2str(action));
724 retaction = (*curstate)(as, action, pkt);
725 amandad_debug(1, _("state_machine: %p curstate=%s returned %s (nextstate=%s)\n"),
726 as, state2str(curstate), action2str(retaction),
727 state2str(as->state));
731 * State has queued up and is now blocking on input.
734 amandad_debug(1, _("state_machine: %p leaving (A_PENDING)\n"), as);
738 * service has switched states. Loop.
744 * state has determined that the packet it received was bogus.
745 * Send a nak, and return.
748 dbprintf(_("received unexpected %s packet\n"),
749 pkt_type2str(pkt->type));
750 dbprintf(_("<<<<<\n%s----\n\n"), pkt->body);
751 pkt_init(&nak, P_NAK, _("ERROR unexpected packet type %s\n"),
752 pkt_type2str(pkt->type));
753 do_sendpkt(as->security_handle, &nak);
755 security_recvpkt(as->security_handle, protocol_recv, as, -1);
756 amandad_debug(1, _("state_machine: %p leaving (A_SENDNAK)\n"), as);
760 * Service is done. Remove it and finish.
763 amandad_debug(1, _("state_machine: %p leaving (A_FINISH)\n"), as);
776 * This state just sends an ack. After that, we move to the repwait
777 * state to wait for REP data to arrive from the subprocess.
781 struct active_service * as,
787 (void)action; /* Quiet unused parameter warning */
788 (void)pkt; /* Quiet unused parameter warning */
790 pkt_init_empty(&ack, P_ACK);
791 if (do_sendpkt(as->security_handle, &ack) < 0) {
792 dbprintf(_("error sending ACK: %s\n"),
793 security_geterror(as->security_handle));
800 * move to the repwait state
801 * Setup a listener for data on the reply fd, but also
802 * listen for packets over the wire, as the server may
803 * poll us if we take a long time.
804 * Setup a timeout that will fire if it takes too long to
807 as->state = s_repwait;
808 as->ev_repfd = event_register((event_id_t)as->repfd, EV_READFD, repfd_recv, as);
809 as->ev_reptimeout = event_register(REP_TIMEOUT, EV_TIME,
812 as->ev_errfd = event_register((event_id_t)as->errfd, EV_READFD, errfd_recv, as);
813 security_recvpkt(as->security_handle, protocol_recv, as, -1);
818 * This is the repwait state. We have responded to the initial REQ with
819 * an ACK, and we are now waiting for the process we spawned to pass us
820 * data to send in a REP.
824 struct active_service * as,
837 * We normally shouldn't receive any packets while waiting
838 * for our REP data, but in some cases we do.
840 if (action == A_RECVPKT) {
843 * Another req for something that's running. Just send an ACK
844 * and go back and wait for more data.
846 if (pkt->type == P_REQ) {
847 dbprintf(_("received dup P_REQ packet, ACKing it\n"));
848 amfree(as->rep_pkt.body);
849 pkt_init_empty(&as->rep_pkt, P_ACK);
850 do_sendpkt(as->security_handle, &as->rep_pkt);
851 security_recvpkt(as->security_handle, protocol_recv, as, -1);
854 /* something unexpected. Nak it */
858 if (action == A_TIMEOUT) {
859 amfree(as->rep_pkt.body);
860 pkt_init(&as->rep_pkt, P_NAK, _("ERROR timeout on reply pipe\n"));
861 dbprintf(_("%s timed out waiting for REP data\n"), as->cmd);
862 do_sendpkt(as->security_handle, &as->rep_pkt);
866 assert(action == A_RECVREP);
867 if(as->bufsize == 0) {
868 as->bufsize = NETWORK_BLOCK_BYTES;
869 as->repbuf = alloc(as->bufsize);
873 n = read(as->repfd, as->repbuf + as->repbufsize,
874 as->bufsize - as->repbufsize - 1);
875 } while ((n < 0) && ((errno == EINTR) || (errno == EAGAIN)));
877 const char *errstr = strerror(errno);
878 dbprintf(_("read error on reply pipe: %s\n"), errstr);
879 amfree(as->rep_pkt.body);
880 pkt_init(&as->rep_pkt, P_NAK, _("ERROR read error on reply pipe: %s\n"),
882 do_sendpkt(as->security_handle, &as->rep_pkt);
886 /* If end of service, wait for process status */
888 pid = waitpid(as->pid, &retstat, WNOHANG);
889 if (as->service == SERVICE_NOOP ||
890 as->service == SERVICE_SENDSIZE ||
891 as->service == SERVICE_SELFCHECK) {
893 while (t<5 && pid == 0) {
896 pid = waitpid(as->pid, &retstat, WNOHANG);
903 pid = waitpid(as->pid, &retstat, WNOHANG);
907 if (! WIFEXITED(retstat)) {
909 code = WTERMSIG(retstat);
910 } else if (WEXITSTATUS(retstat) != 0) {
912 code = WEXITSTATUS(retstat);
915 dbprintf(_("service %s failed: pid %u exited with %s %d\n"),
916 (as->cmd)?as->cmd:_("??UNKONWN??"),
920 _("ERROR service %s failed: pid %u exited with %s %d\n"),
921 (as->cmd)?as->cmd:_("??UNKONWN??"), (unsigned)as->pid,
923 if (as->repbufsize + strlen(msg) >= (as->bufsize - 1)) {
925 repbuf_temp = alloc(as->bufsize);
926 memcpy(repbuf_temp, as->repbuf, as->repbufsize + 1);
928 as->repbuf = repbuf_temp;
930 strcpy(as->repbuf + as->repbufsize, msg);
931 as->repbufsize += strlen(msg);
938 * If we got some data, go back and wait for more, or EOF. Nul terminate
941 as->repbuf[n + as->repbufsize] = '\0';
944 if(as->repbufsize >= (as->bufsize - 1)) {
946 repbuf_temp = alloc(as->bufsize);
947 memcpy(repbuf_temp, as->repbuf, as->repbufsize + 1);
949 as->repbuf = repbuf_temp;
951 else if(as->send_partial_reply) {
952 amfree(as->rep_pkt.body);
953 pkt_init(&as->rep_pkt, P_PREP, "%s", as->repbuf);
954 do_sendpkt(as->security_handle, &as->rep_pkt);
955 amfree(as->rep_pkt.body);
956 pkt_init_empty(&as->rep_pkt, P_REP);
963 * If we got 0, then we hit EOF. Process the data and release
968 assert(as->ev_repfd != NULL);
969 event_release(as->ev_repfd);
972 assert(as->ev_reptimeout != NULL);
973 event_release(as->ev_reptimeout);
974 as->ev_reptimeout = NULL;
976 as->state = s_processrep;
982 * After we have read in all of the rep data, we process it and send
983 * it out as a REP packet.
987 struct active_service * as,
993 (void)action; /* Quiet unused parameter warning */
994 (void)pkt; /* Quiet unused parameter warning */
997 * Copy the rep lines into the outgoing packet.
999 * If this line is a CONNECT, translate it
1000 * Format is "CONNECT <tag> <handle> <tag> <handle> etc...
1003 * CONNECT DATA 4 MESG 5 INDEX 6
1005 * The tags are arbitrary. The handles are in the DATA_FD pool.
1006 * We need to map these to security streams and pass them back
1007 * to the amanda server. If the handle is -1, then we don't map.
1009 if (strncmp_const(as->repbuf,"KENCRYPT\n") == 0) {
1010 amandad_kencrypt = KENCRYPT_WILL_DO;
1011 repbuf = stralloc(as->repbuf + 9);
1013 repbuf = stralloc(as->repbuf);
1015 amfree(as->rep_pkt.body);
1016 pkt_init_empty(&as->rep_pkt, P_REP);
1017 tok = strtok(repbuf, " ");
1020 if (strcmp(tok, "CONNECT") == 0) {
1021 char *line, *nextbuf;
1023 /* Save the entire line */
1024 line = strtok(NULL, "\n");
1025 /* Save the buf following the line */
1026 nextbuf = strtok(NULL, "");
1028 if (line == NULL || nextbuf == NULL)
1031 pkt_cat(&as->rep_pkt, "CONNECT");
1033 /* loop over the id/handle pairs */
1036 tok = strtok(line, " ");
1037 line = NULL; /* keep working from line */
1040 pkt_cat(&as->rep_pkt, " %s", tok);
1043 tok = strtok(NULL, " \n");
1046 /* convert the handle into something the server can process */
1047 pkt_cat(&as->rep_pkt, " %d", allocstream(as, atoi(tok)));
1049 pkt_cat(&as->rep_pkt, "\n%s", nextbuf);
1052 pkt_cat(&as->rep_pkt, "%s", as->repbuf);
1056 * We've setup our REP packet in as->rep_pkt. Now move to the transmission
1059 as->state = s_sendrep;
1060 as->repretry = getconf_int(CNF_REP_TRIES);
1062 return (A_CONTINUE);
1066 * This is the state where we send the REP we just collected from our child.
1070 struct active_service * as,
1074 (void)action; /* Quiet unused parameter warning */
1075 (void)pkt; /* Quiet unused parameter warning */
1078 * Transmit it and move to the ack state.
1080 do_sendpkt(as->security_handle, &as->rep_pkt);
1081 security_recvpkt(as->security_handle, protocol_recv, as, ACK_TIMEOUT);
1082 as->state = s_ackwait;
1087 * This is the state in which we wait for the server to ACK the REP
1092 struct active_service * as,
1096 struct datafd_handle *dh;
1100 * If we got a timeout, try again, but eventually give up.
1102 if (action == A_TIMEOUT) {
1103 if (--as->repretry > 0) {
1104 as->state = s_sendrep;
1105 return (A_CONTINUE);
1107 dbprintf(_("timeout waiting for ACK for our REP\n"));
1110 amandad_debug(1, _("received ACK, now opening streams\n"));
1112 assert(action == A_RECVPKT);
1114 if (pkt->type == P_REQ) {
1115 dbprintf(_("received dup P_REQ packet, resending REP\n"));
1116 as->state = s_sendrep;
1117 return (A_CONTINUE);
1120 if (pkt->type != P_ACK)
1123 if (amandad_kencrypt == KENCRYPT_WILL_DO) {
1124 amandad_kencrypt = KENCRYPT_YES;
1128 * Got the ack, now open the pipes
1130 for (dh = &as->data[0]; dh < &as->data[DATA_FD_COUNT]; dh++) {
1131 if (dh->netfd == NULL)
1133 dbprintf("opening security stream for fd %d\n", (int)(dh - as->data) + DATA_FD_OFFSET);
1134 if (security_stream_accept(dh->netfd) < 0) {
1135 dbprintf(_("stream %td accept failed: %s\n"),
1136 dh - &as->data[0], security_geterror(as->security_handle));
1137 security_stream_close(dh->netfd);
1142 /* setup an event for reads from it. As a special case, don't start
1143 * listening on as->data[0] until we read some data on another fd, if
1144 * the service is sendbackup. This ensures that we send a MESG or
1145 * INDEX token before any DATA tokens, as dumper assumes. This is a
1146 * hack, if that wasn't already obvious! */
1147 if (dh != &as->data[0] || as->service != SERVICE_SENDBACKUP) {
1148 dh->ev_read = event_register((event_id_t)dh->fd_read, EV_READFD,
1149 process_readnetfd, dh);
1151 amandad_debug(1, "Skipping registration of sendbackup's data FD\n");
1154 security_stream_read(dh->netfd, process_writenetfd, dh);
1159 * Pipes are open, so auth them. Count them at the same time.
1161 for (npipes = 0, dh = &as->data[0]; dh < &as->data[DATA_FD_COUNT]; dh++) {
1162 if (dh->netfd == NULL)
1164 if (security_stream_auth(dh->netfd) < 0) {
1165 security_stream_close(dh->netfd);
1167 event_release(dh->ev_read);
1168 event_release(dh->ev_write);
1170 dh->ev_write = NULL;
1177 * If no pipes are open, then we're done. Otherwise, just start running.
1178 * The event handlers on all of the pipes will take it from here.
1180 amandad_debug(1, _("at end of s_ackwait, npipes is %d\n"), npipes);
1184 security_close(as->security_handle);
1185 as->security_handle = NULL;
1191 * Called when a repfd has received data
1197 struct active_service *as = cookie;
1200 assert(as->ev_repfd != NULL);
1202 state_machine(as, A_RECVREP, NULL);
1209 struct active_service *as = cookie;
1211 /* Process errfd before sending the REP packet */
1213 SELECT_ARG_TYPE readset;
1217 memset(&tv, 0, SIZEOF(tv));
1219 FD_SET(as->errfd, &readset);
1220 nfound = select(as->errfd+1, &readset, NULL, NULL, &tv);
1221 if (nfound && FD_ISSET(as->errfd, &readset)) {
1228 * Called when a errfd has received data
1234 struct active_service *as = cookie;
1240 assert(as->ev_errfd != NULL);
1242 n = read(as->errfd, &buf, 32768);
1245 /* Terminate it with '\0' */
1249 as->errbuf = vstrextend(&as->errbuf, buf, NULL);
1251 as->errbuf = stralloc(buf);
1253 } else if (n == 0) {
1254 event_release(as->ev_errfd);
1255 as->ev_errfd = NULL;
1256 } else { /* n < 0 */
1257 event_release(as->ev_errfd);
1258 as->ev_errfd = NULL;
1259 g_snprintf(buf, 32768,
1260 "error reading stderr or service: %s\n", strerror(errno));
1263 /* for each line terminate by '\n' */
1264 while (as->errbuf != NULL && (r = strchr(as->errbuf, '\n')) != NULL) {
1268 s = vstrallocf("ERROR service %s: %s\n",
1269 services[as->service].name, as->errbuf);
1271 /* Add to repbuf, error message will be in the REP packet if it
1272 * is not already sent
1275 if (as->bufsize == 0) {
1276 as->bufsize = NETWORK_BLOCK_BYTES;
1277 as->repbuf = alloc(as->bufsize);
1279 while (as->bufsize < as->repbufsize + n) {
1282 repbuf_temp = alloc(as->bufsize);
1283 memcpy(repbuf_temp, as->repbuf, as->repbufsize + 1);
1285 as->repbuf = repbuf_temp;
1287 memcpy(as->repbuf + as->repbufsize, s, n);
1288 as->repbufsize += n;
1292 /* remove first line from buffer */
1301 * Called when a repfd has timed out
1307 struct active_service *as = cookie;
1310 assert(as->ev_reptimeout != NULL);
1312 state_machine(as, A_TIMEOUT, NULL);
1316 * Called when a handle has received data
1322 security_status_t status)
1324 struct active_service *as = cookie;
1330 dbprintf(_("received %s pkt:\n<<<<<\n%s>>>>>\n"),
1331 pkt_type2str(pkt->type), pkt->body);
1332 state_machine(as, A_RECVPKT, pkt);
1335 dbprintf(_("timeout\n"));
1336 state_machine(as, A_TIMEOUT, NULL);
1339 dbprintf(_("receive error: %s\n"),
1340 security_geterror(as->security_handle));
1346 * This is a generic relay function that just reads data from one of
1347 * the process's pipes and passes it up the equivalent security_stream_t
1354 struct datafd_handle *dh = cookie;
1355 struct active_service *as = dh->as;
1361 n = read(dh->fd_read, as->databuf, SIZEOF(as->databuf));
1362 } while ((n < 0) && ((errno == EINTR) || (errno == EAGAIN)));
1368 pkt_init(&nak, P_NAK, _("A ERROR data descriptor %d broken: %s\n"),
1369 dh->fd_read, strerror(errno));
1373 * Process has closed the pipe. Just remove this event handler.
1374 * If all pipes are closed, shut down this service.
1377 event_release(dh->ev_read);
1379 if(dh->ev_write == NULL) {
1380 security_stream_close(dh->netfd);
1383 for (dh = &as->data[0]; dh < &as->data[DATA_FD_COUNT]; dh++) {
1384 if (dh->netfd != NULL)
1391 /* Handle the special case of recognizing "sendbackup info end"
1392 * from sendbackup's MESG fd */
1393 if (as->service == SERVICE_SENDBACKUP && !as->seen_info_end && dh == &as->data[1]) {
1394 /* make a buffer containing the combined data from info_end_buf
1395 * and what we've read this time, and search it for info_end_strj
1396 * This includes a NULL byte for strstr's sanity. */
1397 char *combined_buf = malloc(INFO_END_LEN + n + 1);
1398 memcpy(combined_buf, as->info_end_buf, INFO_END_LEN);
1399 memcpy(combined_buf+INFO_END_LEN, as->databuf, n);
1400 combined_buf[INFO_END_LEN+n] = '\0';
1402 as->seen_info_end = (strstr(combined_buf, info_end_str) != NULL);
1404 /* fill info_end_buf from the tail end of combined_buf */
1405 memcpy(as->info_end_buf, combined_buf + n, INFO_END_LEN);
1407 /* if we did see info_end_str, start reading the data fd (fd 0) */
1408 if (as->seen_info_end) {
1409 struct datafd_handle *dh = &as->data[0];
1410 amandad_debug(1, "Opening datafd to sendbackup (delayed until sendbackup sent header info)\n");
1411 dh->ev_read = event_register((event_id_t)dh->fd_read, EV_READFD,
1412 process_readnetfd, dh);
1414 amandad_debug(1, "sendbackup header info still not complete\n");
1418 if (security_stream_write(dh->netfd, as->databuf, (size_t)n) < 0) {
1419 /* stream has croaked */
1420 pkt_init(&nak, P_NAK, _("ERROR write error on stream %d: %s\n"),
1421 security_stream_id(dh->netfd),
1422 security_stream_geterror(dh->netfd));
1428 do_sendpkt(as->security_handle, &nak);
1434 * This is a generic relay function that just read data from one of
1435 * the security_stream_t and passes it up the equivalent process's pipes
1443 struct datafd_handle *dh;
1445 assert(cookie != NULL);
1448 if (dh->fd_write <= 0) {
1449 dbprintf(_("process_writenetfd: dh->fd_write <= 0\n"));
1450 } else if (size > 0) {
1451 full_write(dh->fd_write, buf, (size_t)size);
1452 security_stream_read(dh->netfd, process_writenetfd, dh);
1455 aclose(dh->fd_write);
1461 * Convert a local stream handle (DATA_FD...) into something that
1462 * can be sent to the amanda server.
1464 * Returns a number that should be sent to the server in the REP packet.
1468 struct active_service * as,
1471 struct datafd_handle *dh;
1473 /* note that handle is in the range DATA_FD_OFFSET to DATA_FD_COUNT, but
1474 * it is NOT a file descriptor! */
1476 /* if the handle is -1, then we don't bother */
1480 /* make sure the handle's kosher */
1481 if (handle < DATA_FD_OFFSET || handle >= DATA_FD_OFFSET + DATA_FD_COUNT)
1484 /* get a pointer into our handle array */
1485 dh = &as->data[handle - DATA_FD_OFFSET];
1487 /* make sure we're not already using the net handle */
1488 if (dh->netfd != NULL)
1491 /* allocate a stream from the security layer and return */
1492 dh->netfd = security_stream_server(as->security_handle);
1493 if (dh->netfd == NULL) {
1494 dbprintf(_("couldn't open stream to server: %s\n"),
1495 security_geterror(as->security_handle));
1500 * convert the stream into a numeric id that can be sent to the
1503 return (security_stream_id(dh->netfd));
1507 * Create a new service instance
1509 static struct active_service *
1511 security_handle_t * security_handle,
1514 const char * arguments)
1517 int data_read[DATA_FD_COUNT + 2][2];
1518 int data_write[DATA_FD_COUNT + 2][2];
1519 struct active_service *as;
1523 assert(security_handle != NULL);
1524 assert(cmd != NULL);
1525 assert(arguments != NULL);
1527 /* a plethora of pipes */
1528 /* data_read[0] : stdin
1529 * data_write[0] : stdout
1530 * data_read[1], data_write[1] : first stream
1531 * data_read[2], data_write[2] : second stream
1532 * data_read[3], data_write[3] : third stream
1533 * data_write[4] : stderr
1535 for (i = 0; i < DATA_FD_COUNT + 1; i++) {
1536 if (pipe(data_read[i]) < 0) {
1537 error(_("pipe: %s\n"), strerror(errno));
1540 if (pipe(data_write[i]) < 0) {
1541 error(_("pipe: %s\n"), strerror(errno));
1545 if (pipe(data_write[STDERR_PIPE]) < 0) {
1546 error(_("pipe: %s\n"), strerror(errno));
1550 switch(pid = fork()) {
1552 error(_("could not fork service %s: %s\n"), cmd, strerror(errno));
1556 * The parent. Close the far ends of our pipes and return.
1558 as = g_new0(struct active_service, 1);
1559 as->cmd = stralloc(cmd);
1560 as->arguments = stralloc(arguments);
1561 as->security_handle = security_handle;
1563 as->service = service;
1565 as->send_partial_reply = 0;
1566 as->seen_info_end = FALSE;
1567 /* fill in info_end_buf with non-null characters */
1568 memset(as->info_end_buf, '-', sizeof(as->info_end_buf));
1569 if(service == SERVICE_SENDSIZE) {
1570 g_option_t *g_options;
1571 char *option_str, *p;
1573 option_str = stralloc(as->arguments+8);
1574 p = strchr(option_str,'\n');
1577 g_options = parse_g_options(option_str, 1);
1578 if(am_has_feature(g_options->features, fe_partial_estimate)) {
1579 as->send_partial_reply = 1;
1581 free_g_options(g_options);
1585 /* write to the request pipe */
1586 aclose(data_read[0][0]);
1587 as->reqfd = data_read[0][1];
1590 * read from the reply pipe
1592 as->repfd = data_write[0][0];
1593 aclose(data_write[0][1]);
1594 as->ev_repfd = NULL;
1599 as->rep_pkt.body = NULL;
1602 * read from the stderr pipe
1604 as->errfd = data_write[STDERR_PIPE][0];
1605 aclose(data_write[STDERR_PIPE][1]);
1606 as->ev_errfd = NULL;
1610 * read from the rest of the general-use pipes
1611 * (netfds are opened as the client requests them)
1613 for (i = 0; i < DATA_FD_COUNT; i++) {
1614 aclose(data_read[i + 1][1]);
1615 aclose(data_write[i + 1][0]);
1616 as->data[i].fd_read = data_read[i + 1][0];
1617 as->data[i].fd_write = data_write[i + 1][1];
1618 as->data[i].ev_read = NULL;
1619 as->data[i].ev_write = NULL;
1620 as->data[i].netfd = NULL;
1621 as->data[i].as = as;
1624 /* add it to the service queue */
1625 /* increment the active service count */
1626 serviceq = g_slist_append(serviceq, (gpointer)as);
1631 * The child. Put our pipes in their advertised locations
1636 * The data stream is stdin in the new process
1638 if (dup2(data_read[0][0], 0) < 0) {
1639 error(_("dup %d to %d failed: %s\n"), data_read[0][0], 0,
1643 aclose(data_read[0][0]);
1644 aclose(data_read[0][1]);
1647 * The reply stream is stdout
1649 if (dup2(data_write[0][1], 1) < 0) {
1650 error(_("dup %d to %d failed: %s\n"), data_write[0][1], 1,
1653 aclose(data_write[0][0]);
1654 aclose(data_write[0][1]);
1656 for (i = 0; i < DATA_FD_COUNT; i++) {
1657 aclose(data_read[i + 1][0]);
1658 aclose(data_write[i + 1][1]);
1662 * Make sure they are not open in the range DATA_FD_OFFSET to
1663 * DATA_FD_OFFSET + DATA_FD_COUNT*2 - 1
1665 for (i = 0; i < DATA_FD_COUNT; i++) {
1666 while(data_read[i + 1][1] >= DATA_FD_OFFSET &&
1667 data_read[i + 1][1] <= DATA_FD_OFFSET + DATA_FD_COUNT*2 - 1) {
1668 newfd = dup(data_read[i + 1][1]);
1670 error(_("Can't dup out off DATA_FD range"));
1671 data_read[i + 1][1] = newfd;
1673 while(data_write[i + 1][0] >= DATA_FD_OFFSET &&
1674 data_write[i + 1][0] <= DATA_FD_OFFSET + DATA_FD_COUNT*2 - 1) {
1675 newfd = dup(data_write[i + 1][0]);
1677 error(_("Can't dup out off DATA_FD range"));
1678 data_write[i + 1][0] = newfd;
1681 while(data_write[4][0] >= DATA_FD_OFFSET &&
1682 data_write[4][0] <= DATA_FD_OFFSET + DATA_FD_COUNT*2 - 1) {
1683 newfd = dup(data_write[4][0]);
1685 error(_("Can't dup out off DATA_FD range"));
1686 data_write[4][0] = newfd;
1688 while(data_write[4][1] >= DATA_FD_OFFSET &&
1689 data_write[4][1] <= DATA_FD_OFFSET + DATA_FD_COUNT*2 - 1) {
1690 newfd = dup(data_write[4][1]);
1692 error(_("Can't dup out off DATA_FD range"));
1693 data_write[4][1] = newfd;
1696 for (i = 0; i < DATA_FD_COUNT*2; i++)
1697 close(DATA_FD_OFFSET + i);
1700 * The rest start at the offset defined in amandad.h, and continue
1701 * through the internal defined.
1703 for (i = 0; i < DATA_FD_COUNT; i++) {
1704 if (dup2(data_read[i + 1][1], i*2 + DATA_FD_OFFSET) < 0) {
1705 error(_("dup %d to %d failed: %s\n"), data_read[i + 1][1],
1706 i + DATA_FD_OFFSET, strerror(errno));
1708 aclose(data_read[i + 1][1]);
1710 if (dup2(data_write[i + 1][0], i*2 + 1 + DATA_FD_OFFSET) < 0) {
1711 error(_("dup %d to %d failed: %s\n"), data_write[i + 1][0],
1712 i + DATA_FD_OFFSET, strerror(errno));
1714 aclose(data_write[i + 1][0]);
1717 /* close all unneeded fd */
1718 close(STDERR_FILENO);
1719 dup2(data_write[STDERR_PIPE][1], 2);
1720 aclose(data_write[STDERR_PIPE][0]);
1721 aclose(data_write[STDERR_PIPE][1]);
1722 safe_fd(DATA_FD_OFFSET, DATA_FD_COUNT*2);
1724 execle(cmd, cmd, "amandad", auth, (char *)NULL, safe_env());
1725 error(_("could not exec service %s: %s\n"), cmd, strerror(errno));
1732 * Unallocate a service instance
1736 struct active_service * as)
1741 struct datafd_handle *dh;
1743 amandad_debug(1, _("closing service: %s\n"),
1744 (as->cmd)?as->cmd:_("??UNKONWN??"));
1748 assert(as->cmd != NULL);
1751 assert(as->arguments != NULL);
1752 amfree(as->arguments);
1754 if (as->reqfd != -1)
1756 if (as->repfd != -1)
1758 if (as->errfd != -1) {
1763 if (as->ev_repfd != NULL)
1764 event_release(as->ev_repfd);
1765 if (as->ev_reptimeout != NULL)
1766 event_release(as->ev_reptimeout);
1767 if (as->ev_errfd != NULL)
1768 event_release(as->ev_errfd);
1770 for (i = 0; i < DATA_FD_COUNT; i++) {
1773 aclose(dh->fd_read);
1774 aclose(dh->fd_write);
1776 if (dh->netfd != NULL)
1777 security_stream_close(dh->netfd);
1779 if (dh->ev_read != NULL)
1780 event_release(dh->ev_read);
1781 if (dh->ev_write != NULL)
1782 event_release(dh->ev_write);
1785 if (as->security_handle != NULL)
1786 security_close(as->security_handle);
1788 assert(as->pid > 0);
1789 kill(as->pid, SIGTERM);
1790 pid = waitpid(as->pid, NULL, WNOHANG);
1792 while (pid != as->pid && count > 0) {
1795 pid = waitpid(as->pid, NULL, WNOHANG);
1797 if (pid != as->pid) {
1798 g_debug("Process %d failed to exit", (int)as->pid);
1801 serviceq = g_slist_remove(serviceq, (gpointer)as);
1804 amfree(as->arguments);
1806 amfree(as->rep_pkt.body);
1809 if(exit_on_qlength == 0 && g_slist_length(serviceq) == 0) {
1816 * Like 'fullwrite', but does the work in a child process so pipelines
1821 struct active_service * as,
1828 switch (pid=fork()) {
1833 waitpid(pid, NULL, WNOHANG);
1834 return 0; /* this is the parent */
1836 case 0: /* this is the child */
1838 writesize = full_write(as->reqfd, bufp, size);
1839 exit(writesize != size);
1847 security_handle_t * handle,
1850 dbprintf(_("sending %s pkt:\n<<<<<\n%s>>>>>\n"),
1851 pkt_type2str(pkt->type), pkt->body);
1853 return security_sendpkt(handle, pkt);
1859 * Convert a state into a string
1865 static const struct {
1869 #define X(state) { state, stringize(state) }
1879 for (i = 0; i < (int)(sizeof(states) / sizeof(states[0])); i++)
1880 if (state == states[i].state)
1881 return (states[i].str);
1882 return (_("INVALID STATE"));
1886 * Convert an action into a string
1892 static const struct {
1896 #define X(action) { action, stringize(action) }
1909 for (i = 0; i < (int)(sizeof(actions) / sizeof(actions[0])); i++)
1910 if (action == actions[i].action)
1911 return (actions[i].str);
1912 return (_("UNKNOWN ACTION"));
1916 amandad_get_security_conf(
1920 (void)arg; /* Quiet unused parameter warning */
1922 if (!string || !*string)
1925 if (strcmp(string, "kencrypt")==0) {
1926 if (amandad_kencrypt == KENCRYPT_YES)