* Authors: the Amanda Development Team. Its members are listed in a
* file named AUTHORS, in the root directory of this distribution.
*/
-/* $Id: amidxtaped.c,v 1.25.2.3.4.1.2.11.2.2 2005/10/02 13:48:42 martinea Exp $
+/* $Id: amidxtaped.c,v 1.73 2006/07/25 19:06:46 martinea Exp $
*
* This daemon extracts a dump image off a tape for amrecover and
* returns it over the network. It basically, reads a number of
*/
#include "amanda.h"
-#include "clock.h"
#include "version.h"
+#include "clock.h"
+#include "restore.h"
+#include "cmdline.h"
#include "changer.h"
-#include "tapeio.h"
#include "conffile.h"
#include "logfile.h"
+#include "amfeatures.h"
+#include "stream.h"
+#include "amandad.h"
+#include "server_util.h"
+
+#define amidxtaped_debug(i,x) do { \
+ if ((i) <= debug_amidxtaped) { \
+ dbprintf(x); \
+ } \
+} while (0)
+
+#define TIMEOUT 30
static char *pgm = "amidxtaped"; /* in case argv[0] is not set */
-char *conf_logdir = NULL;
-char *conf_logfile = NULL;
-int get_lock = 0;
-char *found_device = NULL;
+extern char *rst_conf_logfile;
+
+static int get_lock = 0;
+static int from_amandad;
-static char *get_client_line P((void));
+static am_feature_t *our_features = NULL;
+static am_feature_t *their_features = NULL;
+static g_option_t *g_options = NULL;
+static int ctlfdin, ctlfdout, datafdout;
+static char *amandad_auth = NULL;
+static FILE *cmdin, *cmdout;
+
+static char *get_client_line(FILE *in);
+static void check_security_buffer(char *);
+static char *get_client_line_fd(int);
+
+/* exit routine */
+static pid_t parent_pid = -1;
+static void cleanup(void);
+
+int main(int argc, char **argv);
/* get a line from client - line terminated by \r\n */
static char *
-get_client_line()
+get_client_line(FILE *in)
{
static char *line = NULL;
char *part = NULL;
- int len;
+ size_t len;
amfree(line);
while(1) {
- if((part = agets(stdin)) == NULL) {
+ if((part = agets(in)) == NULL) {
if(errno != 0) {
- dbprintf(("%s: read error: %s\n",
- debug_prefix_time(NULL), strerror(errno)));
+ dbprintf(_("read error: %s\n"), strerror(errno));
} else {
- dbprintf(("%s: EOF reached\n", debug_prefix_time(NULL)));
+ dbprintf(_("EOF reached\n"));
}
if(line) {
- dbprintf(("%s: unprocessed input:\n", debug_prefix_time(NULL)));
- dbprintf(("-----\n"));
- dbprintf(("%s\n", line));
- dbprintf(("-----\n"));
+ dbprintf(_("s: unprocessed input:\n"));
+ dbprintf("-----\n");
+ dbprintf("%s\n", line);
+ dbprintf("-----\n");
}
amfree(line);
amfree(part);
- if(get_lock) {
- unlink(conf_logfile);
- }
dbclose();
exit(1);
- /* NOTREACHED */
+ /*NOTREACHED*/
}
if(line) {
strappend(line, part);
*/
strappend(line, "\n");
}
- dbprintf(("%s: > %s\n", debug_prefix_time(NULL), line));
+ dbprintf("> %s\n", line);
return line;
}
-int found = 0;
-char *searchlabel = NULL;
-int nslots, backwards;
-
-int scan_init(rc, ns, bk)
-int rc, ns, bk;
+/* get a line from client - line terminated by \r\n */
+static char *
+get_client_line_fd(
+ int fd)
{
- if(rc) {
- if(get_lock) {
- unlink(conf_logfile);
- }
- error("could not get changer info: %s", changer_resultstr);
+ static char *line = NULL;
+ static size_t line_size = 0;
+ char *s = line;
+ size_t len = 0;
+ char c;
+ ssize_t nb;
+
+ if(line == NULL) { /* first time only, allocate initial buffer */
+ s = line = alloc(128);
+ line_size = 128;
}
- nslots = ns;
- backwards = bk;
+ while(1) {
+ nb = read(fd, &c, 1);
+ if (nb <= 0) {
+ /* EOF or error */
+ if ((nb <= 0) && ((errno == EINTR) || (errno == EAGAIN))) {
+ /* Keep looping if failure is temporary */
+ continue;
+ }
+ dbprintf(_("%s: Control pipe read error - %s\n"),
+ pgm, strerror(errno));
+ break;
+ }
- return 0;
+ if(len >= line_size-1) { /* increase buffer size */
+ line_size *= 2;
+ line = realloc(line, line_size);
+ if (line == NULL) {
+ error(_("Memory reallocation failure"));
+ /*NOTREACHED*/
+ }
+ s = &line[len];
+ }
+ *s = c;
+ if(c == '\n') {
+ if(len > 0 && *(s-1) == '\r') { /* remove '\r' */
+ s--;
+ len--;
+ }
+ *s = '\0';
+ return line;
+ }
+ s++;
+ len++;
+ }
+ line[len] = '\0';
+ return line;
}
-int taperscan_slot(rc, slotstr, device)
-int rc;
-char *slotstr;
-char *device;
+void
+check_security_buffer(
+ char * buffer)
{
- char *errstr;
- char *datestamp = NULL, *label = NULL;
+ socklen_t_equiv i;
+ struct sockaddr_in addr;
+ char *s, *fp, ch;
+ char *errstr = NULL;
- if(rc == 2) {
- dbprintf(("%s: fatal slot %s: %s\n", debug_prefix_time(NULL),
- slotstr, changer_resultstr));
- return 1;
- }
- else if(rc == 1) {
- dbprintf(("%s: slot %s: %s\n", debug_prefix_time(NULL),
- slotstr, changer_resultstr));
- return 0;
+ dbprintf(_("check_security_buffer(buffer='%s')\n"), buffer);
+
+ i = SIZEOF(addr);
+ if (getpeername(0, (struct sockaddr *)&addr, &i) == -1) {
+ error(_("getpeername: %s"), strerror(errno));
+ /*NOTREACHED*/
}
- else {
- if((errstr = tape_rdlabel(device, &datestamp, &label)) != NULL) {
- dbprintf(("%s: slot %s: %s\n", debug_prefix_time(NULL),
- slotstr, errstr));
- } else {
- /* got an amanda tape */
- dbprintf(("%s: slot %s: date %-8s label %s",
- debug_prefix_time(NULL), slotstr, datestamp, label));
- if(strcmp(label, FAKE_LABEL) == 0 ||
- strcmp(label, searchlabel) == 0) {
- /* it's the one we are looking for, stop here */
- found_device = newstralloc(found_device,device);
- dbprintf((" (exact label match)\n"));
- found = 1;
- return 1;
- }
- else {
- dbprintf((" (no match)\n"));
- }
- }
+ if ((addr.sin_family != (sa_family_t)AF_INET)
+ || (ntohs(addr.sin_port) == 20)) {
+ error(_("connection rejected from %s family %d port %d"),
+ inet_ntoa(addr.sin_addr), addr.sin_family, htons(addr.sin_port));
+ /*NOTREACHED*/
}
- return 0;
-}
+ /* do the security thing */
+ s = buffer;
+ ch = *s++;
-int lock_logfile()
-{
- conf_logdir = getconf_str(CNF_LOGDIR);
- if (*conf_logdir == '/') {
- conf_logdir = stralloc(conf_logdir);
- } else {
- conf_logdir = stralloc2(config_dir, conf_logdir);
+ skip_whitespace(s, ch);
+ if (ch == '\0') {
+ error(_("cannot parse SECURITY line"));
+ /*NOTREACHED*/
}
- conf_logfile = vstralloc(conf_logdir, "/log", NULL);
- if (access(conf_logfile, F_OK) == 0) {
- error("%s exists: amdump or amflush is already running, or you must run amcleanup", conf_logfile);
+ fp = s-1;
+ skip_non_whitespace(s, ch);
+ s[-1] = '\0';
+ if (strcmp(fp, "SECURITY") != 0) {
+ error(_("cannot parse SECURITY line"));
+ /*NOTREACHED*/
+ }
+ skip_whitespace(s, ch);
+ if (!check_security((sockaddr_union *)&addr, s-1, 0, &errstr)) {
+ error(_("security check failed: %s"), errstr);
+ /*NOTREACHED*/
}
- log_add(L_INFO, "amidxtaped");
- return 1;
}
-
-int main(argc, argv)
-int argc;
-char **argv;
+int
+main(
+ int argc,
+ char ** argv)
{
- int amrestore_nargs;
- char **amrestore_args;
char *buf = NULL;
- int i;
- socklen_t socklen;
- char *amrestore_path;
- pid_t pid;
- int isafile;
- struct stat stat_tape;
- char *tapename = NULL;
- char *s, *fp;
- int ch;
- char *errstr = NULL;
+ int data_sock = -1;
+ in_port_t data_port = (in_port_t)-1;
+ socklen_t_equiv socklen;
struct sockaddr_in addr;
- amwait_t status;
-
- int re_header = 0;
- int re_end = 0;
- char *re_label = NULL;
- char *re_fsf = NULL;
- char *re_device = NULL;
- char *re_host = NULL;
- char *re_disk = NULL;
- char *re_datestamp = NULL;
+ GSList *dumpspecs;
+ tapelist_t *tapes = NULL;
+ char *their_feature_string = NULL;
+ rst_flags_t *rst_flags;
+ int use_changer = 0;
+ int re_end;
char *re_config = NULL;
+ char *conf_tapetype;
+ tapetype_t *tape;
+ char *line;
+ char *tapedev;
+ dumpspec_t *ds;
- safe_fd(-1, 0);
+ /*
+ * Configure program for internationalization:
+ * 1) Only set the message locale for now.
+ * 2) Set textdomain for all amanda related programs to "amanda"
+ * We don't want to be forced to support dozens of message catalogs.
+ */
+ setlocale(LC_MESSAGES, "C");
+ textdomain("amanda");
+
+ safe_fd(DATA_FD_OFFSET, 4);
safe_cd();
+ /* Don't die when child closes pipe */
+ signal(SIGPIPE, SIG_IGN);
+
+ rst_flags = new_rst_flags();
+ rst_flags->mask_splits = 1; /* for older clients */
+ rst_flags->amidxtaped = 1;
+ our_features = am_init_feature_set();
+ their_features = am_set_default_feature_set();
+
/*
* When called via inetd, it is not uncommon to forget to put the
* argv[0] value on the config line. On some systems (e.g. Solaris)
set_pname(pgm);
-#ifdef FORCE_USERID
-
- /* we'd rather not run as root */
-
- if(geteuid() == 0) {
- if(client_uid == (uid_t) -1) {
- error("error [cannot find user %s in passwd file]\n", CLIENT_LOGIN);
- }
-
- initgroups(CLIENT_LOGIN, client_gid);
- setgid(client_gid);
- setuid(client_uid);
+ if(argv && argv[1] && strcmp(argv[1], "amandad") == 0) {
+ from_amandad = 1;
+ if(argv[2])
+ amandad_auth = argv[2];
+ }
+ else {
+ from_amandad = 0;
+ safe_fd(-1, 0);
}
-
-#endif /* FORCE_USERID */
/* initialize */
/* close stderr first so that debug file becomes it - amrestore
chats to stderr, which we don't want going to client */
/* if no debug file, ship to bit bucket */
(void)close(STDERR_FILENO);
- dbopen();
+ dbopen(DBG_SUBDIR_SERVER);
startclock();
- dbprintf(("%s: version %s\n", pgm, version()));
-#ifdef DEBUG_CODE
- if(dbfd() != -1 && dbfd() != STDERR_FILENO)
- {
- if(dup2(dbfd(),STDERR_FILENO) != STDERR_FILENO)
- {
- perror("amidxtaped can't redirect stderr to the debug file");
- dbprintf(("%s: can't redirect stderr to the debug file\n",
- debug_prefix_time(NULL)));
- return 1;
- }
- }
-#else
- if ((i = open("/dev/null", O_WRONLY)) == -1 ||
- (i != STDERR_FILENO &&
- (dup2(i, STDERR_FILENO) != STDERR_FILENO ||
- close(i) != 0))) {
- perror("amidxtaped can't redirect stderr");
- return 1;
- }
-#endif
+ dbprintf(_("%s: version %s\n"), pgm, version());
+ debug_dup_stderr_to_debug();
if (! (argc >= 1 && argv != NULL && argv[0] != NULL)) {
- dbprintf(("%s: WARNING: argv[0] not defined: check inetd.conf\n",
- debug_prefix_time(NULL)));
- }
-
- socklen = sizeof (addr);
- if (getpeername(0, (struct sockaddr *)&addr, &socklen) == -1)
- error("getpeername: %s", strerror(errno));
- if (addr.sin_family != AF_INET || ntohs(addr.sin_port) == 20) {
- error("connection rejected from %s family %d port %d",
- inet_ntoa(addr.sin_addr), addr.sin_family, htons(addr.sin_port));
+ dbprintf(_("WARNING: argv[0] not defined: check inetd.conf\n"));
}
- /* do the security thing */
- amfree(buf);
- buf = stralloc(get_client_line());
- s = buf;
- ch = *s++;
+ if(from_amandad == 0) {
+ socklen = SIZEOF(addr);
+ if (getpeername(0, (struct sockaddr *)&addr, &socklen) == -1) {
+ error(_("getpeername: %s"), strerror(errno));
+ /*NOTREACHED*/
+ }
+ if ((addr.sin_family != (sa_family_t)AF_INET)
+ || (ntohs(addr.sin_port) == 20)) {
+ error(_("connection rejected from %s family %d port %d"),
+ inet_ntoa(addr.sin_addr), addr.sin_family,
+ htons(addr.sin_port));
+ /*NOTREACHED*/
+ }
- skip_whitespace(s, ch);
- if (ch == '\0')
- {
- error("cannot parse SECURITY line");
- }
- fp = s-1;
- skip_non_whitespace(s, ch);
- s[-1] = '\0';
- if (strcmp(fp, "SECURITY") != 0)
- {
- error("cannot parse SECURITY line");
+ /* do the security thing */
+ amfree(buf);
+ fflush(stdout);
+ cmdout = stdout;
+ cmdin = stdin;
+ buf = stralloc(get_client_line(cmdin));
+ check_security_buffer(buf);
}
- skip_whitespace(s, ch);
- if (!security_ok(&addr, s-1, 0, &errstr)) {
- error("security check failed: %s", errstr);
+ else {
+ ctlfdout = DATA_FD_OFFSET + 0;
+ ctlfdin = DATA_FD_OFFSET + 1;
+ datafdout = DATA_FD_OFFSET + 2;
+ close(DATA_FD_OFFSET +3);
+
+ /* read the REQ packet */
+ for(; (line = agets(stdin)) != NULL; free(line)) {
+ if(strncmp_const(line, "OPTIONS ") == 0) {
+ if (g_options)
+ error(_("ERROR recover program sent multiple OPTIONS"));
+ g_options = parse_g_options(line+8, 1);
+ if(!g_options->hostname) {
+ g_options->hostname = alloc(MAX_HOSTNAME_LENGTH+1);
+ gethostname(g_options->hostname, MAX_HOSTNAME_LENGTH);
+ g_options->hostname[MAX_HOSTNAME_LENGTH] = '\0';
+ }
+ }
+ }
+ amfree(line);
+
+ if(amandad_auth && g_options->auth) {
+ if(strcasecmp(amandad_auth, g_options->auth) != 0) {
+ g_printf(_("ERROR recover program ask for auth=%s while amidxtaped is configured for '%s'\n"),
+ g_options->auth, amandad_auth);
+ error(_("ERROR recover program ask for auth=%s while amidxtaped is configured for '%s'"),
+ g_options->auth, amandad_auth);
+ /*NOTREACHED*/
+ }
+ }
+ /* send the REP packet */
+ g_printf("CONNECT CTL %d DATA %d\n", DATA_FD_OFFSET, DATA_FD_OFFSET+1);
+ g_printf("\n");
+ fflush(stdout);
+ fclose(stdin);
+ fclose(stdout);
+ cmdout = fdopen(ctlfdout, "a");
+ if (!cmdout) {
+ error(_("amidxtaped: Can't fdopen(ctlfdout): %s"), strerror(errno));
+ /*NOTREACHED*/
+ }
+ cmdin = fdopen(ctlfdin, "r");
+ if (!cmdin) {
+ error(_("amidxtaped: Can't fdopen(ctlfdin): %s"), strerror(errno));
+ /*NOTREACHED*/
+ }
}
- /* get the number of arguments */
- amrestore_nargs = 0;
- do {
+ ds = dumpspec_new(NULL, NULL, NULL, NULL);
+ for (re_end = 0; re_end == 0; ) {
+ char *s, ch;
amfree(buf);
- buf = stralloc(get_client_line());
- if(strncmp(buf, "LABEL=", 6) == 0) {
- re_label = stralloc(buf+6);
+ buf = stralloc(get_client_line(cmdin));
+ s = buf;
+ if(strncmp_const_skip(buf, "LABEL=", s, ch) == 0) {
+ tapes = unmarshal_tapelist_str(s);
}
- else if(strncmp(buf, "FSF=", 4) == 0) {
- int fsf = atoi(buf+4);
- if(fsf > 0) {
- re_fsf = stralloc(buf+4);
- }
+ else if(strncmp_const_skip(buf, "FSF=", s, ch) == 0) {
+ rst_flags->fsf = OFF_T_ATOI(s);
+ }
+ else if(strncmp_const_skip(buf, "HEADER", s, ch) == 0) {
+ rst_flags->headers = 1;
}
- else if(strncmp(buf, "HEADER", 6) == 0) {
- re_header = 1;
+ else if(strncmp_const_skip(buf, "FEATURES=", s, ch) == 0) {
+ char *our_feature_string = NULL;
+ their_feature_string = stralloc(s);
+ am_release_feature_set(their_features);
+ their_features = am_string_to_feature(their_feature_string);
+ amfree(their_feature_string);
+ our_feature_string = am_feature_to_string(our_features);
+ if(from_amandad == 1)
+ g_fprintf(cmdout,"FEATURES=%s\r\n", our_feature_string);
+ else
+ g_fprintf(cmdout,"%s", our_feature_string);
+ fflush(cmdout);
+ amfree(our_feature_string);
}
- else if(strncmp(buf, "DEVICE=", 7) == 0) {
- re_device = stralloc(buf+7);
+ else if(strncmp_const_skip(buf, "DEVICE=", s, ch) == 0) {
+ rst_flags->alt_tapedev= stralloc(s);
}
- else if(strncmp(buf, "HOST=", 5) == 0) {
- re_host = stralloc(buf+5);
+ else if(strncmp_const_skip(buf, "HOST=", s, ch) == 0) {
+ if (ds->host) {
+ dbprintf(_("WARNING: HOST appeared twice in client request.\n"));
+ amfree(ds->host);
+ }
+ ds->host = stralloc(s);
}
- else if(strncmp(buf, "DISK=", 5) == 0) {
- re_disk = stralloc(buf+5);
+ else if(strncmp_const_skip(buf, "DISK=", s, ch) == 0) {
+ if (ds->disk) {
+ dbprintf(_("WARNING: DISK appeared twice in client request.\n"));
+ amfree(ds->disk);
+ }
+ ds->disk = stralloc(s);
}
- else if(strncmp(buf, "DATESTAMP=", 10) == 0) {
- re_datestamp = stralloc(buf+10);
+ else if(strncmp_const_skip(buf, "DATESTAMP=", s, ch) == 0) {
+ if (ds->datestamp) {
+ dbprintf(_("WARNING: DATESTAMP appeared twice in client request.\n"));
+ amfree(ds->datestamp);
+ }
+ ds->datestamp = stralloc(s);
}
- else if(strncmp(buf, "END", 3) == 0) {
+ else if(strncmp_const(buf, "END") == 0) {
re_end = 1;
}
- else if(strncmp(buf, "CONFIG=", 7) == 0) {
- re_config = stralloc(buf+7);
+ else if(strncmp_const_skip(buf, "CONFIG=", s, ch) == 0) {
+ re_config = stralloc(s);
+ if(strlen(re_config) == 0)
+ amfree(re_config);
}
else if(buf[0] != '\0' && buf[0] >= '0' && buf[0] <= '9') {
- amrestore_nargs = atoi(buf);
re_end = 1;
}
- else {
- }
- } while (re_end == 0);
+ }
+ amfree(buf);
if(re_config) {
- char *conffile;
- config_dir = vstralloc(CONFIG_DIR, "/", re_config, "/", NULL);
- conffile = stralloc2(config_dir, CONFFILE_NAME);
- if (read_conffile(conffile)) {
- dbprintf(("%s: config '%s' not found\n",
- debug_prefix_time(NULL), re_config));
- amfree(re_fsf);
- amfree(re_label);
- amfree(re_config);
- }
- }
- else {
- amfree(re_fsf);
- amfree(re_label);
+ config_init(CONFIG_INIT_EXPLICIT_NAME, re_config);
+ dbrename(re_config, DBG_SUBDIR_SERVER);
+ } else {
+ config_init(0, NULL);
}
- if(re_device && re_config &&
- strcmp(re_device, getconf_str(CNF_TAPEDEV)) == 0) {
- get_lock = lock_logfile();
+ if (config_errors(NULL) >= CFGERR_ERRORS) {
+ g_critical(_("errors processing config file"));
}
- if(re_label && re_config &&
- strcmp(re_device, getconf_str(CNF_AMRECOVER_CHANGER)) == 0) {
+ check_running_as(RUNNING_AS_DUMPUSER_PREFERRED);
- if(changer_init() == 0) {
- dbprintf(("%s: No changer available\n",
- debug_prefix_time(NULL)));
- }
- else {
- searchlabel = stralloc(re_label);
- changer_find(scan_init, taperscan_slot, searchlabel);
- if(found == 0) {
- dbprintf(("%s: Can't find label \"%s\"\n",
- debug_prefix_time(NULL), searchlabel));
- if(get_lock) {
- unlink(conf_logfile);
- }
- dbclose();
- exit(1);
- }
- else {
- re_device=stralloc(found_device);
- dbprintf(("%s: label \"%s\" found\n",
- debug_prefix_time(NULL), searchlabel));
- }
- }
+ if(tapes &&
+ (!rst_flags->alt_tapedev ||
+ (re_config && ( strcmp(rst_flags->alt_tapedev,
+ getconf_str(CNF_AMRECOVER_CHANGER)) == 0 ||
+ strcmp(rst_flags->alt_tapedev,
+ getconf_str(CNF_TPCHANGER)) == 0 ) ) ) ) {
+ /* We need certain options, if restoring from more than one tape */
+ if(tapes->next && !am_has_feature(their_features, fe_recover_splits)) {
+ error(_("Client must support split dumps to restore requested data."));
+ /*NOTREACHED*/
+ }
+ dbprintf(_("Restoring from changer, checking labels\n"));
+ rst_flags->check_labels = 1;
+ use_changer = 1;
+ }
+
+ /* build the dumpspec list from our single dumpspec */
+ dumpspecs = g_slist_append(NULL, (gpointer)ds);
+ ds = NULL;
+
+ if(!tapes && rst_flags->alt_tapedev){
+ dbprintf(_("Looks like we're restoring from a holding file...\n"));
+ tapes = unmarshal_tapelist_str(rst_flags->alt_tapedev);
+ tapes->isafile = 1;
+ amfree(rst_flags->alt_tapedev);
+ rst_flags->alt_tapedev = NULL;
+ use_changer = FALSE;
+ }
+
+ tapedev = getconf_str(CNF_TAPEDEV);
+ /* If we'll be stepping on the tape server's devices, lock them. */
+ if(re_config &&
+ (use_changer || (rst_flags->alt_tapedev && tapedev &&
+ strcmp(rst_flags->alt_tapedev, tapedev) == 0) ) ) {
+ dbprintf(_("Locking devices\n"));
+ parent_pid = getpid();
+ atexit(cleanup);
+ get_lock = lock_logfile();
}
+ if (get_lock)
+ log_add(L_INFO, "%s pid %ld", get_pname(), (long)getpid());
- if(re_fsf && re_config && getconf_int(CNF_AMRECOVER_DO_FSF) == 0) {
- amfree(re_fsf);
- }
- if(re_label && re_config && getconf_int(CNF_AMRECOVER_CHECK_LABEL) == 0) {
- amfree(re_label);
+ /* Init the tape changer */
+ if(tapes && use_changer && changer_init() == 0) {
+ dbprintf(_("No changer available\n"));
}
- dbprintf(("%s: amrestore_nargs=%d\n",
- debug_prefix_time(NULL),
- amrestore_nargs));
-
- amrestore_args = (char **)alloc((amrestore_nargs+12)*sizeof(char *));
- i = 0;
- amrestore_args[i++] = "amrestore";
- if(re_header || re_device || re_host || re_disk || re_datestamp ||
- re_label || re_fsf) {
-
- amrestore_args[i++] = "-p";
- if(re_header) amrestore_args[i++] = "-h";
- if(re_label) {
- amrestore_args[i++] = "-l";
- amrestore_args[i++] = re_label;
- }
- if(re_fsf) {
- amrestore_args[i++] = "-f";
- amrestore_args[i++] = re_fsf;
- }
- if(re_device) amrestore_args[i++] = re_device;
- if(re_host) amrestore_args[i++] = re_host;
- if(re_disk) amrestore_args[i++] = re_disk;
- if(re_datestamp) amrestore_args[i++] = re_datestamp;
- }
- else { /* fe_amidxtaped_nargs */
- while (i <= amrestore_nargs) {
- amrestore_args[i++] = stralloc(get_client_line());
- }
+ /* Read the default block size from the tape type */
+ if(re_config && (conf_tapetype = getconf_str(CNF_TAPETYPE)) != NULL) {
+ tape = lookup_tapetype(conf_tapetype);
+ rst_flags->blocksize = tapetype_get_blocksize(tape) * 1024;
}
- amrestore_args[i] = NULL;
-
- amrestore_path = vstralloc(sbindir, "/", "amrestore", NULL);
- /* so got all the arguments, now ready to execv */
- dbprintf(("%s: Ready to execv amrestore with:\n", debug_prefix_time(NULL)));
- dbprintf(("path = %s\n", amrestore_path));
- for (i = 0; amrestore_args[i] != NULL; i++)
- {
- dbprintf(("argv[%d] = \"%s\"\n", i, amrestore_args[i]));
+ if(rst_flags->fsf && re_config &&
+ getconf_boolean(CNF_AMRECOVER_DO_FSF) == 0) {
+ rst_flags->fsf = (off_t)0;
}
- if ((pid = fork()) == 0)
- {
- /* child */
- (void)execv(amrestore_path, amrestore_args);
-
- /* only get here if exec failed */
- dbprintf(("%s: child could not exec %s: %s\n",
- debug_prefix_time(NULL),
- amrestore_path,
- strerror(errno)));
- return 1;
- /*NOT REACHED*/
+ if (!use_changer && re_config &&
+ getconf_boolean(CNF_AMRECOVER_CHECK_LABEL) == 0) {
+ rst_flags->check_labels = 0;
}
- /* this is the parent */
- if (pid == -1)
- {
- dbprintf(("%s: error forking amrestore child: %s\n",
- debug_prefix_time(NULL), strerror(errno)));
- if(get_lock) {
- unlink(conf_logfile);
+ /* establish a distinct data connection for dumpfile data */
+ if(am_has_feature(their_features, fe_recover_splits)) {
+ if(from_amandad == 1) {
+ rst_flags->pipe_to_fd = datafdout;
}
- dbclose();
- return 1;
- }
+ else {
+ int data_fd;
+ char *buf;
- /* wait for the child to do the restore */
- if (waitpid(pid, &status, 0) == -1)
- {
- dbprintf(("%s: error waiting for amrestore child: %s\n",
- debug_prefix_time(NULL), strerror(errno)));
- if(get_lock) {
- unlink(conf_logfile);
- }
- dbclose();
- return 1;
- }
- /* amrestore often sees the pipe reader (ie restore) quit in the middle
- of the file because it has extracted all of the files needed. This
- results in an exit status of 2. This unfortunately is the exit
- status returned by many errors. Only when the exit status is 1 is it
- guaranteed that an error existed. In all cases we should rewind the
- tape if we can so that a retry starts from the correct place */
- if (WIFEXITED(status) != 0)
- {
-
- dbprintf(("%s: amrestore terminated normally with status: %d\n",
- debug_prefix_time(NULL), WEXITSTATUS(status)));
- }
- else
- {
- dbprintf(("%s: amrestore terminated abnormally.\n",
- debug_prefix_time(NULL)));
- }
+ dbprintf(_("Client understands split dumpfiles\n"));
- /* rewind tape */
- if(re_device) {
- tapename = re_device;
- }
- else {
- /* the first non-option argument is the tape device */
- for (i = 1; i <= amrestore_nargs; i++)
- if (amrestore_args[i][0] != '-')
- break;
- if (i > amrestore_nargs) {
- dbprintf(("%s: Couldn't find tape in arguments\n",
- debug_prefix_time(NULL)));
- if(get_lock) {
- unlink(conf_logfile);
+ if((data_sock = stream_server(AF_INET, &data_port, STREAM_BUFSIZE,
+ STREAM_BUFSIZE, 0)) < 0){
+ error(_("could not create data socket: %s"), strerror(errno));
+ /*NOTREACHED*/
}
- dbclose();
- return 1;
- }
+ dbprintf(_("Local port %d set aside for data\n"), data_port);
- tapename = stralloc(amrestore_args[i]);
- }
- if (tape_stat(tapename, &stat_tape) != 0) {
- error("could not stat %s: %s", tapename, strerror(errno));
- }
- isafile = S_ISREG((stat_tape.st_mode));
- if (!isafile) {
- char *errstr = NULL;
+ /* tell client where to connect */
+ g_printf(_("CONNECT %hu\n"), (unsigned short)data_port);
+ fflush(stdout);
- dbprintf(("%s: rewinding tape ...\n", debug_prefix_time(NULL)));
- errstr = tape_rewind(tapename);
+ if((data_fd = stream_accept(data_sock, TIMEOUT, STREAM_BUFSIZE,
+ STREAM_BUFSIZE)) < 0){
+ error(_("stream_accept failed for client data connection: %s\n"),
+ strerror(errno));
+ /*NOTREACHED*/
+ }
- if (errstr != NULL) {
- dbprintf(("%s: %s\n", debug_prefix_time(NULL), errstr));
- amfree(errstr);
- } else {
- dbprintf(("%s: done\n", debug_prefix_time(NULL)));
- }
- }
+ buf = get_client_line_fd(data_fd);
- if(get_lock) {
- unlink(conf_logfile);
+ check_security_buffer(buf);
+ rst_flags->pipe_to_fd = data_fd;
+ }
}
-
- amfree(tapename);
+ else {
+ rst_flags->pipe_to_fd = fileno(stdout);
+ cmdout = stderr;
+ }
+ dbprintf(_("Sending output to file descriptor %d\n"), rst_flags->pipe_to_fd);
+
+
+ tapedev = getconf_str(CNF_TAPEDEV);
+ if(get_lock == 0 &&
+ re_config &&
+ (use_changer || (rst_flags->alt_tapedev && tapedev &&
+ strcmp(rst_flags->alt_tapedev, tapedev) == 0) ) ) {
+ char *process_name = get_master_process(rst_conf_logfile);
+ send_message(cmdout, rst_flags, their_features,
+ _("%s exists: %s is already running, "
+ "or you must run amcleanup"),
+ rst_conf_logfile, process_name);
+ error(_("%s exists: %s is already running, "
+ "or you must run amcleanup"),
+ rst_conf_logfile, process_name);
+ }
+
+ /* make sure our restore flags aren't crazy */
+ if (check_rst_flags(rst_flags) == -1) {
+ if (rst_flags->pipe_to_fd != -1)
+ aclose(rst_flags->pipe_to_fd);
+ send_message(cmdout, rst_flags, their_features,
+ _("restore flags are crazy"));
+ exit(1);
+ }
+
+ /* actual restoration */
+ search_tapes(cmdout, cmdin, use_changer, tapes, dumpspecs, rst_flags,
+ their_features);
+ dbprintf(_("Restoration finished\n"));
+
+ /* cleanup */
+ if(rst_flags->pipe_to_fd != -1) aclose(rst_flags->pipe_to_fd);
+ free_tapelist(tapes);
+
+ am_release_feature_set(their_features);
+
+ amfree(rst_flags->alt_tapedev);
+ amfree(rst_flags);
+ dumpspec_list_free(dumpspecs);
+ amfree(re_config);
dbclose();
return 0;
}
+
+static void
+cleanup(void)
+{
+ if (parent_pid == getpid()) {
+ if (get_lock) {
+ log_add(L_INFO, "pid-done %ld\n", (long)getpid());
+ unlink(rst_conf_logfile);
+ }
+ }
+}