2 * Bacula File Daemon Job processing
4 * Kern Sibbald, October MM
10 Copyright (C) 2000-2003 Kern Sibbald and John Walker
12 This program is free software; you can redistribute it and/or
13 modify it under the terms of the GNU General Public License as
14 published by the Free Software Foundation; either version 2 of
15 the License, or (at your option) any later version.
17 This program is distributed in the hope that it will be useful,
18 but WITHOUT ANY WARRANTY; without even the implied warranty of
19 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20 General Public License for more details.
22 You should have received a copy of the GNU General Public
23 License along with this program; if not, write to the Free
24 Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
32 extern char my_name[];
33 extern CLIENT *me; /* our client resource */
35 /* Imported functions */
36 extern int status_cmd(JCR *jcr);
38 /* Forward referenced functions */
39 static int backup_cmd(JCR *jcr);
40 static int bootstrap_cmd(JCR *jcr);
41 static int cancel_cmd(JCR *jcr);
42 static int setdebug_cmd(JCR *jcr);
43 static int estimate_cmd(JCR *jcr);
44 static int exclude_cmd(JCR *jcr);
45 static int hello_cmd(JCR *jcr);
46 static int job_cmd(JCR *jcr);
47 static int include_cmd(JCR *jcr);
48 static int level_cmd(JCR *jcr);
49 static int verify_cmd(JCR *jcr);
50 static int restore_cmd(JCR *jcr);
51 static int storage_cmd(JCR *jcr);
52 static int session_cmd(JCR *jcr);
53 static int response(JCR *jcr, BSOCK *sd, char *resp, char *cmd);
54 static void filed_free_jcr(JCR *jcr);
55 static int open_sd_read_session(JCR *jcr);
56 static int send_bootstrap_file(JCR *jcr);
57 static int runbefore_cmd(JCR *jcr);
58 static int runafter_cmd(JCR *jcr);
59 static int run_cmd(JCR *jcr, char *cmd, char *name);
62 /* Exported functions */
70 * The following are the recognized commands from the Director.
72 static struct s_cmds cmds[] = {
73 {"backup", backup_cmd},
74 {"cancel", cancel_cmd},
75 {"setdebug=", setdebug_cmd},
76 {"estimate", estimate_cmd},
77 {"exclude", exclude_cmd},
79 {"include", include_cmd},
81 {"level = ", level_cmd},
82 {"restore", restore_cmd},
83 {"session", session_cmd},
84 {"status", status_cmd},
85 {"storage ", storage_cmd},
86 {"verify", verify_cmd},
87 {"bootstrap", bootstrap_cmd},
88 {"RunBeforeJob", runbefore_cmd},
89 {"RunAfterJob", runafter_cmd},
90 {NULL, NULL} /* list terminator */
93 /* Commands received from director that need scanning */
94 static char jobcmd[] = "JobId=%d Job=%127s SDid=%d SDtime=%d Authorization=%100s";
95 static char storaddr[] = "storage address=%s port=%d ssl=%d\n";
96 static char sessioncmd[] = "session %127s %ld %ld %ld %ld %ld %ld\n";
97 static char restorecmd[] = "restore replace=%c prelinks=%d where=%s\n";
98 static char restorecmd1[] = "restore replace=%c prelinks=%d where=\n";
99 static char verifycmd[] = "verify level=%30s\n";
100 static char estimatecmd[] = "estimate listing=%d\n";
101 static char runbefore[] = "RunBeforeJob %s\n";
102 static char runafter[] = "RunAfterJob %s\n";
104 /* Responses sent to Director */
105 static char errmsg[] = "2999 Invalid command\n";
106 static char no_auth[] = "2998 No Authorization\n";
107 static char OKinc[] = "2000 OK include\n";
108 static char OKest[] = "2000 OK estimate files=%u bytes=%s\n";
109 static char OKexc[] = "2000 OK exclude\n";
110 static char OKlevel[] = "2000 OK level\n";
111 static char OKbackup[] = "2000 OK backup\n";
112 static char OKbootstrap[] = "2000 OK bootstrap\n";
113 static char OKverify[] = "2000 OK verify\n";
114 static char OKrestore[] = "2000 OK restore\n";
115 static char OKsession[] = "2000 OK session\n";
116 static char OKstore[] = "2000 OK storage\n";
117 static char OKjob[] = "2000 OK Job " HOST_OS "," DISTNAME "," DISTVER;
118 static char OKsetdebug[] = "2000 OK setdebug=%d\n";
119 static char BADjob[] = "2901 Bad Job\n";
120 static char EndJob[] = "2800 End Job TermCode=%d JobFiles=%u ReadBytes=%s JobBytes=%s Errors=%u\n";
121 static char OKRunBefore[] = "2000 OK RunBefore\n";
122 static char OKRunAfter[] = "2000 OK RunAfter\n";
124 /* Responses received from Storage Daemon */
125 static char OK_end[] = "3000 OK end\n";
126 static char OK_close[] = "3000 OK close Status = %d\n";
127 static char OK_open[] = "3000 OK open ticket = %d\n";
128 static char OK_data[] = "3000 OK data\n";
129 static char OK_append[] = "3000 OK append data\n";
130 static char OKSDbootstrap[] = "3000 OK bootstrap\n";
133 /* Commands sent to Storage Daemon */
134 static char append_open[] = "append open session\n";
135 static char append_data[] = "append data %d\n";
136 static char append_end[] = "append end session %d\n";
137 static char append_close[] = "append close session %d\n";
138 static char read_open[] = "read open session = %s %ld %ld %ld %ld %ld %ld\n";
139 static char read_data[] = "read data %d\n";
140 static char read_close[] = "read close session %d\n";
143 * Accept requests from a Director
145 * NOTE! We are running as a separate thread
147 * Send output one line
148 * at a time followed by a zero length transmission.
150 * Return when the connection is terminated or there
153 * Basic task here is:
154 * Authenticate Director (during Hello command).
155 * Accept commands one at a time from the Director
159 void *handle_client_request(void *dirp)
163 BSOCK *dir = (BSOCK *)dirp;
165 jcr = new_jcr(sizeof(JCR), filed_free_jcr); /* create JCR */
166 jcr->dir_bsock = dir;
167 jcr->ff = init_find_files();
168 jcr->start_time = time(NULL);
169 jcr->last_fname = get_pool_memory(PM_FNAME);
170 jcr->last_fname[0] = 0;
171 jcr->client_name = get_memory(strlen(my_name) + 1);
172 pm_strcpy(&jcr->client_name, my_name);
174 enable_backup_privileges(NULL, 1 /* ignore_errors */);
176 /**********FIXME******* add command handler error code */
178 for (quit=0; !quit;) {
181 if (bnet_recv(dir) < 0) {
182 break; /* connection terminated */
184 dir->msg[dir->msglen] = 0;
185 Dmsg1(100, "<dird: %s", dir->msg);
187 for (i=0; cmds[i].cmd; i++) {
188 if (strncmp(cmds[i].cmd, dir->msg, strlen(cmds[i].cmd)) == 0) {
189 if (!jcr->authenticated && cmds[i].func != hello_cmd) {
190 bnet_fsend(dir, no_auth);
193 found = TRUE; /* indicate command found */
194 if (!cmds[i].func(jcr)) { /* do command */
195 quit = TRUE; /* error or fully terminated, get out */
196 Dmsg0(20, "Command error or Job done.\n");
201 if (!found) { /* command not found */
202 bnet_fsend(dir, errmsg);
208 /* Inform Storage daemon that we are done */
209 if (jcr->store_bsock) {
210 bnet_sig(jcr->store_bsock, BNET_TERMINATE);
213 if (jcr->RunAfterJob && !job_canceled(jcr)) {
214 run_cmd(jcr, jcr->RunAfterJob, "ClientRunAfterJob");
217 /* Inform Director that we are done */
218 bnet_sig(dir, BNET_TERMINATE);
220 Dmsg0(100, "Calling term_find_files\n");
221 term_find_files((FF_PKT *)jcr->ff);
222 Dmsg0(100, "Done with term_find_files\n");
223 free_jcr(jcr); /* destroy JCR record */
224 Dmsg0(100, "Done with free_jcr\n");
229 * Hello from Director he must identify himself and provide his
232 static int hello_cmd(JCR *jcr)
234 Dmsg0(120, "Calling Authenticate\n");
235 if (!authenticate_director(jcr)) {
238 Dmsg0(120, "OK Authenticate\n");
239 jcr->authenticated = TRUE;
246 static int cancel_cmd(JCR *jcr)
248 BSOCK *dir = jcr->dir_bsock;
249 char Job[MAX_NAME_LENGTH];
252 if (sscanf(dir->msg, "cancel Job=%127s", Job) == 1) {
253 if (!(cjcr=get_jcr_by_full_name(Job))) {
254 bnet_fsend(dir, "2901 Job %s not found.\n", Job);
256 if (cjcr->store_bsock) {
258 cjcr->store_bsock->timed_out = 1;
259 cjcr->store_bsock->terminated = 1;
261 pthread_kill(cjcr->my_thread_id, TIMEOUT_SIGNAL);
265 set_jcr_job_status(cjcr, JS_Canceled);
267 bnet_fsend(dir, "2001 Job %s marked to be canceled.\n", Job);
270 bnet_fsend(dir, "2902 Error scanning cancel command.\n");
272 bnet_sig(dir, BNET_EOD);
278 * Set debug level as requested by the Director
281 static int setdebug_cmd(JCR *jcr)
283 BSOCK *dir = jcr->dir_bsock;
286 Dmsg1(110, "setdebug_cmd: %s", dir->msg);
287 if (sscanf(dir->msg, "setdebug=%d", &level) != 1 || level < 0) {
288 pm_strcpy(&jcr->errmsg, dir->msg);
289 bnet_fsend(dir, "2991 Bad setdebug command: %s\n", jcr->errmsg);
293 return bnet_fsend(dir, OKsetdebug, level);
297 static int estimate_cmd(JCR *jcr)
299 BSOCK *dir = jcr->dir_bsock;
302 if (sscanf(dir->msg, estimatecmd, &jcr->listing) != 1) {
303 pm_strcpy(&jcr->errmsg, dir->msg);
304 Jmsg(jcr, M_FATAL, 0, _("Bad estimate command: %s"), jcr->errmsg);
305 bnet_fsend(dir, "2992 Bad estimate command.\n");
309 bnet_fsend(dir, OKest, jcr->num_files_examined,
310 edit_uint64_with_commas(jcr->JobBytes, ed2));
311 bnet_sig(dir, BNET_EOD);
316 * Get JobId and Storage Daemon Authorization key from Director
318 static int job_cmd(JCR *jcr)
320 BSOCK *dir = jcr->dir_bsock;
321 POOLMEM *sd_auth_key;
323 sd_auth_key = get_memory(dir->msglen);
324 if (sscanf(dir->msg, jobcmd, &jcr->JobId, jcr->Job,
325 &jcr->VolSessionId, &jcr->VolSessionTime,
327 pm_strcpy(&jcr->errmsg, dir->msg);
328 Jmsg(jcr, M_FATAL, 0, _("Bad Job Command: %s"), jcr->errmsg);
329 bnet_fsend(dir, BADjob);
330 free_pool_memory(sd_auth_key);
333 jcr->sd_auth_key = bstrdup(sd_auth_key);
334 free_pool_memory(sd_auth_key);
335 Dmsg2(120, "JobId=%d Auth=%s\n", jcr->JobId, jcr->sd_auth_key);
336 return bnet_fsend(dir, OKjob);
339 static int runbefore_cmd(JCR *jcr)
342 BSOCK *dir = jcr->dir_bsock;
343 POOLMEM *cmd = get_memory(dir->msglen+1);
345 Dmsg1(100, "runbefore_cmd: %s", dir->msg);
346 if (sscanf(dir->msg, runbefore, cmd) != 1) {
347 pm_strcpy(&jcr->errmsg, dir->msg);
348 Jmsg1(jcr, M_FATAL, 0, _("Bad RunBeforeJob command: %s\n"), jcr->errmsg);
349 bnet_fsend(dir, "2905 Bad RunBeforeJob command.\n");
355 /* Run the command now */
356 stat = run_cmd(jcr, cmd, "ClientRunBeforeJob");
359 bnet_fsend(dir, OKRunBefore);
362 bnet_fsend(dir, "2905 Bad RunBeforeJob command.\n");
367 static int runafter_cmd(JCR *jcr)
369 BSOCK *dir = jcr->dir_bsock;
370 POOLMEM *msg = get_memory(dir->msglen+1);
372 Dmsg1(100, "runafter_cmd: %s", dir->msg);
373 if (sscanf(dir->msg, runafter, msg) != 1) {
374 pm_strcpy(&jcr->errmsg, dir->msg);
375 Jmsg1(jcr, M_FATAL, 0, _("Bad RunAfter command: %s\n"), jcr->errmsg);
376 bnet_fsend(dir, "2905 Bad RunAfterJob command.\n");
381 if (jcr->RunAfterJob) {
382 free_pool_memory(jcr->RunAfterJob);
384 jcr->RunAfterJob = get_pool_memory(PM_FNAME);
385 pm_strcpy(&jcr->RunAfterJob, msg);
386 free_pool_memory(msg);
387 return bnet_fsend(dir, OKRunAfter);
390 static int run_cmd(JCR *jcr, char *cmd, char *name)
392 POOLMEM *ecmd = get_pool_memory(PM_FNAME);
395 char line[MAXSTRING];
397 ecmd = edit_job_codes(jcr, ecmd, cmd, "");
398 bpipe = open_bpipe(ecmd, 0, "r");
399 free_pool_memory(ecmd);
400 while (fgets(line, sizeof(line), bpipe->rfd)) {
401 Jmsg(jcr, M_INFO, 0, _("%s: %s"), name, line);
403 status = close_bpipe(bpipe);
405 Jmsg(jcr, M_FATAL, 0, _("%s returned non-zero status=%d\n"), name,
407 set_jcr_job_status(jcr, JS_FatalError);
417 static void add_fname_to_list(JCR *jcr, char *fname, int list)
427 /* Skip leading options -- currently ignored */
428 for (p=fname; *p && *p != ' '; p++)
430 /* Skip spaces, and q points to first space */
431 for (q=NULL; *p && *p == ' '; p++) {
439 fn = get_pool_memory(PM_FNAME);
440 fn = edit_job_codes(jcr, fn, p, "");
441 bpipe = open_bpipe(fn, 0, "r");
442 free_pool_memory(fn);
444 Jmsg(jcr, M_FATAL, 0, _("Cannot run program: %s. ERR=%s\n"),
448 /* Copy File options */
449 if (list == INC_LIST) {
450 *q = 0; /* terminate options */
453 optlen = strlen(buf);
457 while (fgets(buf+optlen, sizeof(buf)-optlen, bpipe->rfd)) {
458 strip_trailing_junk(buf);
459 if (list == INC_LIST) {
460 add_fname_to_include_list((FF_PKT *)jcr->ff, 1, buf);
462 add_fname_to_exclude_list((FF_PKT *)jcr->ff, buf);
465 if ((stat=close_bpipe(bpipe)) != 0) {
466 Jmsg(jcr, M_FATAL, 0, _("Error running program: %s. RtnStat=%d ERR=%s\n"),
467 p, stat, strerror(errno));
472 p++; /* skip over < */
473 if ((ffd = fopen(p, "r")) == NULL) {
474 Jmsg(jcr, M_FATAL, 0, _("Cannot open %s file: %s. ERR=%s\n"),
475 list==INC_LIST?"included":"excluded", p, strerror(errno));
478 /* Copy File options */
479 if (list == INC_LIST) {
480 *q = 0; /* terminate options */
483 optlen = strlen(buf);
487 while (fgets(buf+optlen, sizeof(buf)-optlen, ffd)) {
488 strip_trailing_junk(buf);
489 if (list == INC_LIST) {
490 add_fname_to_include_list((FF_PKT *)jcr->ff, 1, buf);
492 add_fname_to_exclude_list((FF_PKT *)jcr->ff, buf);
498 if (list == INC_LIST) {
499 add_fname_to_include_list((FF_PKT *)jcr->ff, 1, fname);
501 add_fname_to_exclude_list((FF_PKT *)jcr->ff, p);
509 * Get list of files/directories to include from Director
512 static int include_cmd(JCR *jcr)
514 BSOCK *dir = jcr->dir_bsock;
516 while (bnet_recv(dir) >= 0) {
517 dir->msg[dir->msglen] = 0;
518 strip_trailing_junk(dir->msg);
519 Dmsg1(010, "include file: %s\n", dir->msg);
520 add_fname_to_list(jcr, dir->msg, INC_LIST);
523 return bnet_fsend(dir, OKinc);
527 * Get list of files to exclude from Director
530 static int exclude_cmd(JCR *jcr)
532 BSOCK *dir = jcr->dir_bsock;
534 while (bnet_recv(dir) >= 0) {
535 dir->msg[dir->msglen] = 0;
536 strip_trailing_junk(dir->msg);
537 add_fname_to_list(jcr, dir->msg, EXC_LIST);
538 Dmsg1(110, "<dird: exclude file %s\n", dir->msg);
541 return bnet_fsend(dir, OKexc);
545 static int bootstrap_cmd(JCR *jcr)
547 BSOCK *dir = jcr->dir_bsock;
548 POOLMEM *fname = get_pool_memory(PM_FNAME);
551 if (jcr->RestoreBootstrap) {
552 unlink(jcr->RestoreBootstrap);
553 free_pool_memory(jcr->RestoreBootstrap);
555 Mmsg(&fname, "%s/%s.%s.bootstrap", me->working_directory, me->hdr.name,
557 Dmsg1(400, "bootstrap=%s\n", fname);
558 jcr->RestoreBootstrap = fname;
559 bs = fopen(fname, "a+"); /* create file */
562 * Suck up what he is sending to us so that he will then
563 * read our error message.
565 while (bnet_recv(dir) >= 0)
568 Jmsg(jcr, M_FATAL, 0, _("Could not create bootstrap file %s: ERR=%s\n"),
569 jcr->RestoreBootstrap, strerror(errno));
570 free_pool_memory(jcr->RestoreBootstrap);
571 jcr->RestoreBootstrap = NULL;
572 set_jcr_job_status(jcr, JS_ErrorTerminated);
576 while (bnet_recv(dir) >= 0) {
577 Dmsg1(200, "filed<dird: bootstrap file %s\n", dir->msg);
582 return bnet_fsend(dir, OKbootstrap);
587 * Get backup level from Director
590 static int level_cmd(JCR *jcr)
592 BSOCK *dir = jcr->dir_bsock;
598 level = get_memory(dir->msglen+1);
599 Dmsg1(110, "level_cmd: %s", dir->msg);
600 if (sscanf(dir->msg, "level = %s ", level) != 1) {
601 pm_strcpy(&jcr->errmsg, dir->msg);
602 Jmsg1(jcr, M_FATAL, 0, _("Bad level command: %s\n"), jcr->errmsg);
606 /* Base backup requested? */
607 if (strcmp(level, "base") == 0) {
608 jcr->JobLevel = L_BASE;
609 /* Full backup requested? */
610 } else if (strcmp(level, "full") == 0) {
611 jcr->JobLevel = L_FULL;
613 * Backup requested since <date> <time>
614 * This form is also used for incremental and differential
616 } else if (strcmp(level, "since") == 0) {
617 jcr->JobLevel = L_SINCE;
618 if (sscanf(dir->msg, "level = since %d-%d-%d %d:%d:%d mtime_only=%d",
619 &tm.tm_year, &tm.tm_mon, &tm.tm_mday,
620 &tm.tm_hour, &tm.tm_min, &tm.tm_sec, &mtime_only) != 7) {
621 pm_strcpy(&jcr->errmsg, dir->msg);
622 Jmsg1(jcr, M_FATAL, 0, _("Bad scan of date/time: %s\n"), jcr->errmsg);
628 tm.tm_wday = tm.tm_yday = 0;
631 Dmsg2(100, "Got since time: %s mtime_only=%d\n", ctime(&mtime), mtime_only);
632 jcr->incremental = 1; /* set incremental or decremental backup */
633 jcr->mtime = mtime; /* set since time */
634 jcr->mtime_only = mtime_only; /* and what to compare */
636 Jmsg1(jcr, M_FATAL, 0, "Unknown backup level: %s\n", level);
641 return bnet_fsend(dir, OKlevel);
645 * Get session parameters from Director -- this is for a Restore command
647 static int session_cmd(JCR *jcr)
649 BSOCK *dir = jcr->dir_bsock;
651 Dmsg1(100, "SessionCmd: %s", dir->msg);
652 if (sscanf(dir->msg, sessioncmd, jcr->VolumeName,
653 &jcr->VolSessionId, &jcr->VolSessionTime,
654 &jcr->StartFile, &jcr->EndFile,
655 &jcr->StartBlock, &jcr->EndBlock) != 7) {
656 pm_strcpy(&jcr->errmsg, dir->msg);
657 Jmsg(jcr, M_FATAL, 0, "Bad session command: %s", jcr->errmsg);
661 return bnet_fsend(dir, OKsession);
665 * Get address of storage daemon from Director
668 static int storage_cmd(JCR *jcr)
670 int stored_port; /* storage daemon port */
671 int enable_ssl; /* enable ssl to sd */
672 BSOCK *dir = jcr->dir_bsock;
673 BSOCK *sd; /* storage daemon bsock */
675 Dmsg1(100, "StorageCmd: %s", dir->msg);
676 if (sscanf(dir->msg, storaddr, &jcr->stored_addr, &stored_port, &enable_ssl) != 3) {
677 pm_strcpy(&jcr->errmsg, dir->msg);
678 Jmsg(jcr, M_FATAL, 0, _("Bad storage command: %s"), jcr->errmsg);
681 Dmsg3(110, "Open storage: %s:%d ssl=%d\n", jcr->stored_addr, stored_port, enable_ssl);
682 /* Open command communications with Storage daemon */
683 /* Try to connect for 1 hour at 10 second intervals */
684 sd = bnet_connect(jcr, 10, me->SDConnectTimeout, _("Storage daemon"),
685 jcr->stored_addr, NULL, stored_port, 1);
687 Jmsg(jcr, M_FATAL, 0, _("Failed to connect to Storage daemon: %s:%d\n"),
688 jcr->stored_addr, stored_port);
692 jcr->store_bsock = sd;
694 bnet_fsend(sd, "Hello Start Job %s\n", jcr->Job);
695 if (!authenticate_storagedaemon(jcr)) {
696 Jmsg(jcr, M_FATAL, 0, _("Failed to authenticate Storage daemon.\n"));
699 Dmsg0(110, "Authenticated with SD.\n");
701 /* Send OK to Director */
702 return bnet_fsend(dir, OKstore);
707 * Do a backup. For now, we handle only Full and Incremental.
709 static int backup_cmd(JCR *jcr)
711 BSOCK *dir = jcr->dir_bsock;
712 BSOCK *sd = jcr->store_bsock;
715 char ed1[50], ed2[50];
717 set_jcr_job_status(jcr, JS_Blocked);
718 jcr->JobType = JT_BACKUP;
719 Dmsg1(100, "begin backup ff=%p\n", (FF_PKT *)jcr->ff);
722 Jmsg(jcr, M_FATAL, 0, _("Cannot contact Storage daemon\n"));
726 bnet_fsend(dir, OKbackup);
727 Dmsg1(110, "bfiled>dird: %s", dir->msg);
730 * Send Append Open Session to Storage daemon
732 bnet_fsend(sd, append_open);
733 Dmsg1(110, ">stored: %s", sd->msg);
735 * Expect to receive back the Ticket number
737 if (bget_msg(sd) >= 0) {
738 Dmsg1(110, "<stored: %s", sd->msg);
739 if (sscanf(sd->msg, OK_open, &jcr->Ticket) != 1) {
740 Jmsg(jcr, M_FATAL, 0, _("Bad response to append open: %s\n"), sd->msg);
743 Dmsg1(110, "Got Ticket=%d\n", jcr->Ticket);
745 Jmsg(jcr, M_FATAL, 0, _("Bad response from stored to open command\n"));
750 * Send Append data command to Storage daemon
752 bnet_fsend(sd, append_data, jcr->Ticket);
753 Dmsg1(110, ">stored: %s", sd->msg);
756 * Expect to get OK data
758 Dmsg1(110, "<stored: %s", sd->msg);
759 if (!response(jcr, sd, OK_data, "Append Data")) {
764 * Send Files to Storage daemon
766 Dmsg1(110, "begin blast ff=%p\n", (FF_PKT *)jcr->ff);
767 if (!blast_data_to_storage_daemon(jcr, NULL)) {
768 set_jcr_job_status(jcr, JS_ErrorTerminated);
769 bnet_suppress_error_messages(sd, 1);
771 set_jcr_job_status(jcr, JS_Terminated);
772 if (jcr->JobStatus != JS_Terminated) {
773 bnet_suppress_error_messages(sd, 1);
774 goto cleanup; /* bail out now */
777 * Expect to get response to append_data from Storage daemon
779 if (!response(jcr, sd, OK_append, "Append Data")) {
780 set_jcr_job_status(jcr, JS_ErrorTerminated);
785 * Send Append End Data to Storage daemon
787 bnet_fsend(sd, append_end, jcr->Ticket);
789 if (!response(jcr, sd, OK_end, "Append End")) {
790 set_jcr_job_status(jcr, JS_ErrorTerminated);
795 * Send Append Close to Storage daemon
797 bnet_fsend(sd, append_close, jcr->Ticket);
798 while (bget_msg(sd) >= 0) { /* stop on signal or error */
799 if (sscanf(sd->msg, OK_close, &SDJobStatus) == 1) {
801 Dmsg2(200, "SDJobStatus = %d %c\n", SDJobStatus, (char)SDJobStatus);
805 Jmsg(jcr, M_FATAL, 0, _("Append Close with SD failed.\n"));
808 if (SDJobStatus != JS_Terminated) {
809 Jmsg(jcr, M_FATAL, 0, _("Bad status %d returned from Storage Daemon.\n"),
816 bnet_fsend(dir, EndJob, jcr->JobStatus, jcr->JobFiles,
817 edit_uint64(jcr->ReadBytes, ed1),
818 edit_uint64(jcr->JobBytes, ed2), jcr->Errors);
820 return 0; /* return and stop command loop */
824 * Do a Verify for Director
827 static int verify_cmd(JCR *jcr)
829 BSOCK *dir = jcr->dir_bsock;
830 BSOCK *sd = jcr->store_bsock;
831 char level[100], ed1[50], ed2[50];
833 jcr->JobType = JT_VERIFY;
834 if (sscanf(dir->msg, verifycmd, level) != 1) {
835 bnet_fsend(dir, "2994 Bad verify command: %s\n", dir->msg);
838 if (strcasecmp(level, "init") == 0) {
839 jcr->JobLevel = L_VERIFY_INIT;
840 } else if (strcasecmp(level, "catalog") == 0){
841 jcr->JobLevel = L_VERIFY_CATALOG;
842 } else if (strcasecmp(level, "volume") == 0){
843 jcr->JobLevel = L_VERIFY_VOLUME_TO_CATALOG;
844 } else if (strcasecmp(level, "data") == 0){
845 jcr->JobLevel = L_VERIFY_DATA;
846 } else if (strcasecmp(level, "disk_to_catalog") == 0) {
847 jcr->JobLevel = L_VERIFY_DISK_TO_CATALOG;
849 bnet_fsend(dir, "2994 Bad verify level: %s\n", dir->msg);
853 bnet_fsend(dir, OKverify);
854 Dmsg1(110, "bfiled>dird: %s", dir->msg);
856 switch (jcr->JobLevel) {
858 case L_VERIFY_CATALOG:
861 case L_VERIFY_VOLUME_TO_CATALOG:
862 if (!open_sd_read_session(jcr)) {
865 start_dir_heartbeat(jcr);
866 do_verify_volume(jcr);
867 stop_dir_heartbeat(jcr);
869 * Send Close session command to Storage daemon
871 bnet_fsend(sd, read_close, jcr->Ticket);
872 Dmsg1(130, "bfiled>stored: %s", sd->msg);
874 /* ****FIXME**** check response */
875 bget_msg(sd); /* get OK */
877 /* Inform Storage daemon that we are done */
878 bnet_sig(sd, BNET_TERMINATE);
881 case L_VERIFY_DISK_TO_CATALOG:
885 bnet_fsend(dir, "2994 Bad verify level: %s\n", dir->msg);
889 bnet_sig(dir, BNET_EOD);
891 /* Send termination status back to Dir */
892 bnet_fsend(dir, EndJob, jcr->JobStatus, jcr->JobFiles,
893 edit_uint64(jcr->ReadBytes, ed1),
894 edit_uint64(jcr->JobBytes, ed2), jcr->Errors);
896 /* Inform Director that we are done */
897 bnet_sig(dir, BNET_TERMINATE);
898 return 0; /* return and terminate command loop */
902 * Do a Restore for Director
905 static int restore_cmd(JCR *jcr)
907 BSOCK *dir = jcr->dir_bsock;
908 BSOCK *sd = jcr->store_bsock;
912 char ed1[50], ed2[50];
915 * Scan WHERE (base directory for restore) from command
917 Dmsg0(150, "restore command\n");
918 /* Pickup where string */
919 where = get_memory(dir->msglen+1);
922 if (sscanf(dir->msg, restorecmd, &replace, &prefix_links, where) != 3) {
923 if (sscanf(dir->msg, restorecmd1, &replace, &prefix_links) != 2) {
924 pm_strcpy(&jcr->errmsg, dir->msg);
925 Jmsg(jcr, M_FATAL, 0, _("Bad replace command. CMD=%s\n"), jcr->errmsg);
930 /* Turn / into nothing */
931 if (where[0] == '/' && where[1] == 0) {
935 Dmsg2(150, "Got replace %c, where=%s\n", replace, where);
936 unbash_spaces(where);
937 jcr->where = bstrdup(where);
938 free_pool_memory(where);
939 jcr->replace = replace;
940 jcr->prefix_links = prefix_links;
942 bnet_fsend(dir, OKrestore);
943 Dmsg1(110, "bfiled>dird: %s", dir->msg);
945 jcr->JobType = JT_RESTORE;
947 set_jcr_job_status(jcr, JS_Blocked);
948 if (!open_sd_read_session(jcr)) {
949 set_jcr_job_status(jcr, JS_ErrorTerminated);
953 set_jcr_job_status(jcr, JS_Running);
956 * Do restore of files and data
958 start_dir_heartbeat(jcr);
960 stop_dir_heartbeat(jcr);
962 set_jcr_job_status(jcr, JS_Terminated);
963 if (jcr->JobStatus != JS_Terminated) {
964 bnet_suppress_error_messages(sd, 1);
968 * Send Close session command to Storage daemon
970 bnet_fsend(sd, read_close, jcr->Ticket);
971 Dmsg1(130, "bfiled>stored: %s", sd->msg);
973 bget_msg(sd); /* get OK */
975 /* Inform Storage daemon that we are done */
976 bnet_sig(sd, BNET_TERMINATE);
980 /* Send termination status back to Dir */
981 bnet_fsend(dir, EndJob, jcr->JobStatus, jcr->JobFiles,
982 edit_uint64(jcr->ReadBytes, ed1),
983 edit_uint64(jcr->JobBytes, ed2), jcr->Errors);
985 /* Inform Director that we are done */
986 bnet_sig(dir, BNET_TERMINATE);
988 Dmsg0(130, "Done in job.c\n");
989 return 0; /* return and terminate command loop */
992 static int open_sd_read_session(JCR *jcr)
994 BSOCK *sd = jcr->store_bsock;
997 Jmsg(jcr, M_FATAL, 0, _("Improper calling sequence.\n"));
1000 Dmsg4(120, "VolSessId=%ld VolsessT=%ld SF=%ld EF=%ld\n",
1001 jcr->VolSessionId, jcr->VolSessionTime, jcr->StartFile, jcr->EndFile);
1002 Dmsg2(120, "JobId=%d vol=%s\n", jcr->JobId, "DummyVolume");
1004 * Open Read Session with Storage daemon
1006 bnet_fsend(sd, read_open, jcr->VolumeName,
1007 jcr->VolSessionId, jcr->VolSessionTime, jcr->StartFile, jcr->EndFile,
1008 jcr->StartBlock, jcr->EndBlock);
1009 Dmsg1(110, ">stored: %s", sd->msg);
1014 if (bget_msg(sd) >= 0) {
1015 Dmsg1(110, "bfiled<stored: %s", sd->msg);
1016 if (sscanf(sd->msg, OK_open, &jcr->Ticket) != 1) {
1017 Jmsg(jcr, M_FATAL, 0, _("Bad response to SD read open: %s\n"), sd->msg);
1020 Dmsg1(110, "bfiled: got Ticket=%d\n", jcr->Ticket);
1022 Jmsg(jcr, M_FATAL, 0, _("Bad response from stored to read open command\n"));
1026 if (!send_bootstrap_file(jcr)) {
1031 * Start read of data with Storage daemon
1033 bnet_fsend(sd, read_data, jcr->Ticket);
1034 Dmsg1(110, ">stored: %s", sd->msg);
1039 if (!response(jcr, sd, OK_data, "Read Data")) {
1046 * Destroy the Job Control Record and associated
1047 * resources (sockets).
1049 static void filed_free_jcr(JCR *jcr)
1051 if (jcr->store_bsock) {
1052 bnet_close(jcr->store_bsock);
1054 if (jcr->RestoreBootstrap) {
1055 unlink(jcr->RestoreBootstrap);
1056 free_pool_memory(jcr->RestoreBootstrap);
1057 jcr->RestoreBootstrap = NULL;
1059 if (jcr->last_fname) {
1060 free_pool_memory(jcr->last_fname);
1062 if (jcr->RunAfterJob) {
1063 free_pool_memory(jcr->RunAfterJob);
1069 * Get response from Storage daemon to a command we
1070 * sent. Check that the response is OK.
1072 * Returns: 0 on failure
1075 int response(JCR *jcr, BSOCK *sd, char *resp, char *cmd)
1080 if (bget_msg(sd) > 0) {
1081 Dmsg0(110, sd->msg);
1082 if (strcmp(sd->msg, resp) == 0) {
1086 if (job_canceled(jcr)) {
1087 return 0; /* if canceled avoid useless error messages */
1089 if (is_bnet_error(sd)) {
1090 Jmsg2(jcr, M_FATAL, 0, _("Comm error with SD. bad response to %s. ERR=%s\n"),
1091 cmd, bnet_strerror(sd));
1093 Jmsg3(jcr, M_FATAL, 0, _("Bad response to %s command. Wanted %s, got %s\n"),
1094 cmd, resp, sd->msg);
1099 static int send_bootstrap_file(JCR *jcr)
1103 BSOCK *sd = jcr->store_bsock;
1104 char *bootstrap = "bootstrap\n";
1107 Dmsg1(400, "send_bootstrap_file: %s\n", jcr->RestoreBootstrap);
1108 if (!jcr->RestoreBootstrap) {
1111 bs = fopen(jcr->RestoreBootstrap, "r");
1113 Jmsg(jcr, M_FATAL, 0, _("Could not open bootstrap file %s: ERR=%s\n"),
1114 jcr->RestoreBootstrap, strerror(errno));
1115 set_jcr_job_status(jcr, JS_ErrorTerminated);
1118 pm_strcpy(&sd->msg, bootstrap);
1119 sd->msglen = strlen(sd->msg);
1121 while (fgets(buf, sizeof(buf), bs)) {
1122 sd->msglen = Mmsg(&sd->msg, "%s", buf);
1125 bnet_sig(sd, BNET_EOD);
1127 if (!response(jcr, sd, OKSDbootstrap, "Bootstrap")) {
1128 set_jcr_job_status(jcr, JS_ErrorTerminated);
1134 if (jcr->RestoreBootstrap) {
1135 unlink(jcr->RestoreBootstrap);
1136 free_pool_memory(jcr->RestoreBootstrap);
1137 jcr->RestoreBootstrap = NULL;