2 * Bacula File Daemon Job processing
4 * Kern Sibbald, October MM
10 Copyright (C) 2000-2004 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);
37 extern int qstatus_cmd(JCR *jcr);
39 /* Forward referenced functions */
40 static int backup_cmd(JCR *jcr);
41 static int bootstrap_cmd(JCR *jcr);
42 static int cancel_cmd(JCR *jcr);
43 static int setdebug_cmd(JCR *jcr);
44 static int estimate_cmd(JCR *jcr);
45 static int exclude_cmd(JCR *jcr);
46 static int hello_cmd(JCR *jcr);
47 static int job_cmd(JCR *jcr);
48 static int include_cmd(JCR *jcr);
49 static int fileset_cmd(JCR *jcr);
50 static int level_cmd(JCR *jcr);
51 static int verify_cmd(JCR *jcr);
52 static int restore_cmd(JCR *jcr);
53 static int storage_cmd(JCR *jcr);
54 static int session_cmd(JCR *jcr);
55 static int response(JCR *jcr, BSOCK *sd, char *resp, const char *cmd);
56 static void filed_free_jcr(JCR *jcr);
57 static int open_sd_read_session(JCR *jcr);
58 static int send_bootstrap_file(JCR *jcr);
59 static int runbefore_cmd(JCR *jcr);
60 static int runafter_cmd(JCR *jcr);
61 static int run_cmd(JCR *jcr, char *cmd, const char *name);
62 static void set_options(findFOPTS *fo, const char *opts);
65 /* Exported functions */
73 * The following are the recognized commands from the Director.
75 static struct s_cmds cmds[] = {
76 {"backup", backup_cmd},
77 {"cancel", cancel_cmd},
78 {"setdebug=", setdebug_cmd},
79 {"estimate", estimate_cmd},
80 {"exclude", exclude_cmd},
82 {"include", include_cmd},
83 {"fileset", fileset_cmd},
85 {"level = ", level_cmd},
86 {"restore", restore_cmd},
87 {"session", session_cmd},
88 {"status", status_cmd},
89 {".status", qstatus_cmd},
90 {"storage ", storage_cmd},
91 {"verify", verify_cmd},
92 {"bootstrap", bootstrap_cmd},
93 {"RunBeforeJob", runbefore_cmd},
94 {"RunAfterJob", runafter_cmd},
95 {NULL, NULL} /* list terminator */
98 /* Commands received from director that need scanning */
99 static char jobcmd[] = "JobId=%d Job=%127s SDid=%d SDtime=%d Authorization=%100s";
100 static char storaddr[] = "storage address=%s port=%d ssl=%d\n";
101 static char sessioncmd[] = "session %127s %ld %ld %ld %ld %ld %ld\n";
102 static char restorecmd[] = "restore replace=%c prelinks=%d where=%s\n";
103 static char restorecmd1[] = "restore replace=%c prelinks=%d where=\n";
104 static char verifycmd[] = "verify level=%30s\n";
105 static char estimatecmd[] = "estimate listing=%d\n";
106 static char runbefore[] = "RunBeforeJob %s\n";
107 static char runafter[] = "RunAfterJob %s\n";
109 /* Responses sent to Director */
110 static char errmsg[] = "2999 Invalid command\n";
111 static char no_auth[] = "2998 No Authorization\n";
112 static char OKinc[] = "2000 OK include\n";
113 static char OKest[] = "2000 OK estimate files=%u bytes=%s\n";
114 static char OKexc[] = "2000 OK exclude\n";
115 static char OKlevel[] = "2000 OK level\n";
116 static char OKbackup[] = "2000 OK backup\n";
117 static char OKbootstrap[] = "2000 OK bootstrap\n";
118 static char OKverify[] = "2000 OK verify\n";
119 static char OKrestore[] = "2000 OK restore\n";
120 static char OKsession[] = "2000 OK session\n";
121 static char OKstore[] = "2000 OK storage\n";
122 static char OKjob[] = "2000 OK Job %s,%s,%s";
123 static char OKsetdebug[] = "2000 OK setdebug=%d\n";
124 static char BADjob[] = "2901 Bad Job\n";
125 static char EndJob[] = "2800 End Job TermCode=%d JobFiles=%u ReadBytes=%s JobBytes=%s Errors=%u\n";
126 static char OKRunBefore[] = "2000 OK RunBefore\n";
127 static char OKRunAfter[] = "2000 OK RunAfter\n";
129 /* Responses received from Storage Daemon */
130 static char OK_end[] = "3000 OK end\n";
131 static char OK_close[] = "3000 OK close Status = %d\n";
132 static char OK_open[] = "3000 OK open ticket = %d\n";
133 static char OK_data[] = "3000 OK data\n";
134 static char OK_append[] = "3000 OK append data\n";
135 static char OKSDbootstrap[] = "3000 OK bootstrap\n";
138 /* Commands sent to Storage Daemon */
139 static char append_open[] = "append open session\n";
140 static char append_data[] = "append data %d\n";
141 static char append_end[] = "append end session %d\n";
142 static char append_close[] = "append close session %d\n";
143 static char read_open[] = "read open session = %s %ld %ld %ld %ld %ld %ld\n";
144 static char read_data[] = "read data %d\n";
145 static char read_close[] = "read close session %d\n";
148 * Accept requests from a Director
150 * NOTE! We are running as a separate thread
152 * Send output one line
153 * at a time followed by a zero length transmission.
155 * Return when the connection is terminated or there
158 * Basic task here is:
159 * Authenticate Director (during Hello command).
160 * Accept commands one at a time from the Director
164 void *handle_client_request(void *dirp)
169 BSOCK *dir = (BSOCK *)dirp;
171 jcr = new_jcr(sizeof(JCR), filed_free_jcr); /* create JCR */
172 jcr->dir_bsock = dir;
173 jcr->ff = init_find_files();
174 jcr->start_time = time(NULL);
175 jcr->last_fname = get_pool_memory(PM_FNAME);
176 jcr->last_fname[0] = 0;
177 jcr->client_name = get_memory(strlen(my_name) + 1);
178 pm_strcpy(&jcr->client_name, my_name);
180 enable_backup_privileges(NULL, 1 /* ignore_errors */);
182 /**********FIXME******* add command handler error code */
184 for (quit=false; !quit;) {
187 if (bnet_recv(dir) < 0) {
188 break; /* connection terminated */
190 dir->msg[dir->msglen] = 0;
191 Dmsg1(100, "<dird: %s", dir->msg);
193 for (i=0; cmds[i].cmd; i++) {
194 if (strncmp(cmds[i].cmd, dir->msg, strlen(cmds[i].cmd)) == 0) {
195 if (!jcr->authenticated && cmds[i].func != hello_cmd) {
196 bnet_fsend(dir, no_auth);
199 found = true; /* indicate command found */
200 Dmsg1(100, "Executing %s command.\n", cmds[i].cmd);
201 if (!cmds[i].func(jcr)) { /* do command */
202 quit = true; /* error or fully terminated, get out */
203 Dmsg0(20, "Quit command loop due to command error or Job done.\n");
208 if (!found) { /* command not found */
209 bnet_fsend(dir, errmsg);
215 /* Inform Storage daemon that we are done */
216 if (jcr->store_bsock) {
217 bnet_sig(jcr->store_bsock, BNET_TERMINATE);
220 if (jcr->RunAfterJob && !job_canceled(jcr)) {
221 run_cmd(jcr, jcr->RunAfterJob, "ClientRunAfterJob");
224 /* Inform Director that we are done */
225 bnet_sig(dir, BNET_TERMINATE);
227 /* Clean up fileset */
228 FF_PKT *ff = (FF_PKT *)jcr->ff;
229 findFILESET *fileset = ff->fileset;
232 /* Delete FileSet Include lists */
233 for (i=0; i<fileset->include_list.size(); i++) {
234 findINCEXE *incexe = (findINCEXE *)fileset->include_list.get(i);
235 for (j=0; j<incexe->opts_list.size(); j++) {
236 findFOPTS *fo = (findFOPTS *)incexe->opts_list.get(j);
237 for (k=0; k<fo->regex.size(); k++) {
238 regfree((regex_t *)fo->regex.get(k));
250 incexe->opts_list.destroy();
251 incexe->name_list.destroy();
253 fileset->include_list.destroy();
255 /* Delete FileSet Exclude lists */
256 for (i=0; i<fileset->exclude_list.size(); i++) {
257 findINCEXE *incexe = (findINCEXE *)fileset->exclude_list.get(i);
258 for (j=0; j<incexe->opts_list.size(); j++) {
259 findFOPTS *fo = (findFOPTS *)incexe->opts_list.get(j);
264 incexe->opts_list.destroy();
265 incexe->name_list.destroy();
267 fileset->exclude_list.destroy();
270 Dmsg0(100, "Calling term_find_files\n");
271 term_find_files((FF_PKT *)jcr->ff);
272 Dmsg0(100, "Done with term_find_files\n");
273 free_jcr(jcr); /* destroy JCR record */
274 Dmsg0(100, "Done with free_jcr\n");
279 * Hello from Director he must identify himself and provide his
282 static int hello_cmd(JCR *jcr)
284 Dmsg0(120, "Calling Authenticate\n");
285 if (!authenticate_director(jcr)) {
288 Dmsg0(120, "OK Authenticate\n");
289 jcr->authenticated = true;
296 static int cancel_cmd(JCR *jcr)
298 BSOCK *dir = jcr->dir_bsock;
299 char Job[MAX_NAME_LENGTH];
302 if (sscanf(dir->msg, "cancel Job=%127s", Job) == 1) {
303 if (!(cjcr=get_jcr_by_full_name(Job))) {
304 bnet_fsend(dir, "2901 Job %s not found.\n", Job);
306 if (cjcr->store_bsock) {
308 cjcr->store_bsock->timed_out = 1;
309 cjcr->store_bsock->terminated = 1;
311 * #if !defined(HAVE_CYGWIN) && !defined(HAVE_WIN32)
313 #if !defined(HAVE_CYGWIN)
314 pthread_kill(cjcr->my_thread_id, TIMEOUT_SIGNAL);
318 set_jcr_job_status(cjcr, JS_Canceled);
320 bnet_fsend(dir, _("2001 Job %s marked to be canceled.\n"), Job);
323 bnet_fsend(dir, _("2902 Error scanning cancel command.\n"));
325 bnet_sig(dir, BNET_EOD);
331 * Set debug level as requested by the Director
334 static int setdebug_cmd(JCR *jcr)
336 BSOCK *dir = jcr->dir_bsock;
337 int level, trace_flag;
339 Dmsg1(110, "setdebug_cmd: %s", dir->msg);
340 if (sscanf(dir->msg, "setdebug=%d trace=%d", &level, &trace_flag) != 2 || level < 0) {
341 pm_strcpy(&jcr->errmsg, dir->msg);
342 bnet_fsend(dir, "2991 Bad setdebug command: %s\n", jcr->errmsg);
346 set_trace(trace_flag);
347 return bnet_fsend(dir, OKsetdebug, level);
351 static int estimate_cmd(JCR *jcr)
353 BSOCK *dir = jcr->dir_bsock;
356 if (sscanf(dir->msg, estimatecmd, &jcr->listing) != 1) {
357 pm_strcpy(&jcr->errmsg, dir->msg);
358 Jmsg(jcr, M_FATAL, 0, _("Bad estimate command: %s"), jcr->errmsg);
359 bnet_fsend(dir, "2992 Bad estimate command.\n");
363 bnet_fsend(dir, OKest, jcr->num_files_examined,
364 edit_uint64_with_commas(jcr->JobBytes, ed2));
365 bnet_sig(dir, BNET_EOD);
370 * Get JobId and Storage Daemon Authorization key from Director
372 static int job_cmd(JCR *jcr)
374 BSOCK *dir = jcr->dir_bsock;
375 POOLMEM *sd_auth_key;
377 sd_auth_key = get_memory(dir->msglen);
378 if (sscanf(dir->msg, jobcmd, &jcr->JobId, jcr->Job,
379 &jcr->VolSessionId, &jcr->VolSessionTime,
381 pm_strcpy(&jcr->errmsg, dir->msg);
382 Jmsg(jcr, M_FATAL, 0, _("Bad Job Command: %s"), jcr->errmsg);
383 bnet_fsend(dir, BADjob);
384 free_pool_memory(sd_auth_key);
387 jcr->sd_auth_key = bstrdup(sd_auth_key);
388 free_pool_memory(sd_auth_key);
389 Dmsg2(120, "JobId=%d Auth=%s\n", jcr->JobId, jcr->sd_auth_key);
390 return bnet_fsend(dir, OKjob, HOST_OS, DISTNAME, DISTVER);
393 static int runbefore_cmd(JCR *jcr)
396 BSOCK *dir = jcr->dir_bsock;
397 POOLMEM *cmd = get_memory(dir->msglen+1);
399 Dmsg1(100, "runbefore_cmd: %s", dir->msg);
400 if (sscanf(dir->msg, runbefore, cmd) != 1) {
401 pm_strcpy(&jcr->errmsg, dir->msg);
402 Jmsg1(jcr, M_FATAL, 0, _("Bad RunBeforeJob command: %s\n"), jcr->errmsg);
403 bnet_fsend(dir, "2905 Bad RunBeforeJob command.\n");
409 /* Run the command now */
410 stat = run_cmd(jcr, cmd, "ClientRunBeforeJob");
413 bnet_fsend(dir, OKRunBefore);
416 bnet_fsend(dir, "2905 Bad RunBeforeJob command.\n");
421 static int runafter_cmd(JCR *jcr)
423 BSOCK *dir = jcr->dir_bsock;
424 POOLMEM *msg = get_memory(dir->msglen+1);
426 Dmsg1(100, "runafter_cmd: %s", dir->msg);
427 if (sscanf(dir->msg, runafter, msg) != 1) {
428 pm_strcpy(&jcr->errmsg, dir->msg);
429 Jmsg1(jcr, M_FATAL, 0, _("Bad RunAfter command: %s\n"), jcr->errmsg);
430 bnet_fsend(dir, "2905 Bad RunAfterJob command.\n");
435 if (jcr->RunAfterJob) {
436 free_pool_memory(jcr->RunAfterJob);
438 jcr->RunAfterJob = get_pool_memory(PM_FNAME);
439 pm_strcpy(&jcr->RunAfterJob, msg);
440 free_pool_memory(msg);
441 return bnet_fsend(dir, OKRunAfter);
444 static int run_cmd(JCR *jcr, char *cmd, const char *name)
446 POOLMEM *ecmd = get_pool_memory(PM_FNAME);
449 char line[MAXSTRING];
451 ecmd = edit_job_codes(jcr, ecmd, cmd, "");
452 bpipe = open_bpipe(ecmd, 0, "r");
453 free_pool_memory(ecmd);
455 Jmsg(jcr, M_FATAL, 0, _("%s could not execute\n"), name);
456 set_jcr_job_status(jcr, JS_FatalError);
459 while (fgets(line, sizeof(line), bpipe->rfd)) {
460 Jmsg(jcr, M_INFO, 0, _("%s: %s"), name, line);
462 status = close_bpipe(bpipe);
464 Jmsg(jcr, M_FATAL, 0, _("%s returned non-zero status=%d\n"), name,
466 set_jcr_job_status(jcr, JS_FatalError);
476 static void add_fname_to_list(JCR *jcr, char *fname, int list)
486 /* Skip leading options -- currently ignored */
487 for (p=fname; *p && *p != ' '; p++)
489 /* Skip spaces, and q points to first space */
490 for (q=NULL; *p && *p == ' '; p++) {
498 p++; /* skip over | */
499 fn = get_pool_memory(PM_FNAME);
500 fn = edit_job_codes(jcr, fn, p, "");
501 bpipe = open_bpipe(fn, 0, "r");
502 free_pool_memory(fn);
504 Jmsg(jcr, M_FATAL, 0, _("Cannot run program: %s. ERR=%s\n"),
508 /* Copy File options */
509 if (list == INC_LIST) {
510 *q = 0; /* terminate options */
513 optlen = strlen(buf);
517 while (fgets(buf+optlen, sizeof(buf)-optlen, bpipe->rfd)) {
518 strip_trailing_junk(buf);
519 if (list == INC_LIST) {
520 add_fname_to_include_list((FF_PKT *)jcr->ff, 1, buf);
522 add_fname_to_exclude_list((FF_PKT *)jcr->ff, buf);
525 if ((stat=close_bpipe(bpipe)) != 0) {
526 Jmsg(jcr, M_FATAL, 0, _("Error running program: %s. RtnStat=%d ERR=%s\n"),
527 p, stat, strerror(errno));
532 p++; /* skip over < */
533 if ((ffd = fopen(p, "r")) == NULL) {
535 Jmsg(jcr, M_FATAL, 0, _("Cannot open %s file: %s. ERR=%s\n"),
536 list==INC_LIST?"included":"excluded", p, be.strerror());
539 /* Copy File options */
540 if (list == INC_LIST) {
541 *q = 0; /* terminate options */
544 optlen = strlen(buf);
548 while (fgets(buf+optlen, sizeof(buf)-optlen, ffd)) {
549 strip_trailing_junk(buf);
550 if (list == INC_LIST) {
551 add_fname_to_include_list((FF_PKT *)jcr->ff, 1, buf);
553 add_fname_to_exclude_list((FF_PKT *)jcr->ff, buf);
559 if (list == INC_LIST) {
560 add_fname_to_include_list((FF_PKT *)jcr->ff, 1, fname);
562 add_fname_to_exclude_list((FF_PKT *)jcr->ff, p);
570 * Get list of files/directories to include from Director
573 static int include_cmd(JCR *jcr)
575 BSOCK *dir = jcr->dir_bsock;
577 while (bnet_recv(dir) >= 0) {
578 dir->msg[dir->msglen] = 0;
579 strip_trailing_junk(dir->msg);
580 Dmsg1(010, "include file: %s\n", dir->msg);
581 add_fname_to_list(jcr, dir->msg, INC_LIST);
584 return bnet_fsend(dir, OKinc);
587 static bool init_fileset(JCR *jcr)
590 findFILESET *fileset;
595 ff = (FF_PKT *)jcr->ff;
599 fileset = (findFILESET *)malloc(sizeof(findFILESET));
600 memset(fileset, 0, sizeof(findFILESET));
601 ff->fileset = fileset;
602 fileset->state = state_none;
603 fileset->include_list.init(1, true);
604 fileset->exclude_list.init(1, true);
608 static findFOPTS *start_options(FF_PKT *ff)
610 int state = ff->fileset->state;
611 findINCEXE *incexe = ff->fileset->incexe;
613 if (state != state_options) {
614 ff->fileset->state = state_options;
615 findFOPTS *fo = (findFOPTS *)malloc(sizeof(findFOPTS));
616 memset(fo, 0, sizeof(findFOPTS));
617 fo->regex.init(1, true);
618 fo->wild.init(1, true);
619 fo->base.init(1, true);
620 incexe->current_opts = fo;
621 incexe->opts_list.append(fo);
623 return incexe->current_opts;
628 * Add fname to include/exclude fileset list. First check for
629 * | and < and if necessary perform command.
631 static void add_file_to_fileset(JCR *jcr, const char *fname, findFILESET *fileset)
643 p++; /* skip over | */
644 fn = get_pool_memory(PM_FNAME);
645 fn = edit_job_codes(jcr, fn, p, "");
646 bpipe = open_bpipe(fn, 0, "r");
647 free_pool_memory(fn);
649 Jmsg(jcr, M_FATAL, 0, _("Cannot run program: %s. ERR=%s\n"),
653 while (fgets(buf, sizeof(buf), bpipe->rfd)) {
654 strip_trailing_junk(buf);
655 fileset->incexe->name_list.append(bstrdup(buf));
657 if ((stat=close_bpipe(bpipe)) != 0) {
658 Jmsg(jcr, M_FATAL, 0, _("Error running program: %s. RtnStat=%d ERR=%s\n"),
659 p, stat, strerror(errno));
664 p++; /* skip over < */
665 if ((ffd = fopen(p, "r")) == NULL) {
667 Jmsg(jcr, M_FATAL, 0, _("Cannot open FileSet input file: %s. ERR=%s\n"),
671 while (fgets(buf, sizeof(buf), ffd)) {
672 strip_trailing_junk(buf);
673 fileset->incexe->name_list.append(bstrdup(buf));
678 fileset->incexe->name_list.append(bstrdup(fname));
684 static void add_fileset(JCR *jcr, const char *item)
686 FF_PKT *ff = (FF_PKT *)jcr->ff;
687 findFILESET *fileset = ff->fileset;
688 int state = fileset->state;
689 findFOPTS *current_opts;
691 Dmsg1(100, "%s\n", item);
693 if (item[1] == ' ') { /* If string follows */
694 item += 2; /* point to string */
697 if (state == state_error) {
703 fileset->incexe = (findINCEXE *)malloc(sizeof(findINCEXE));
704 memset(fileset->incexe, 0, sizeof(findINCEXE));
705 fileset->incexe->opts_list.init(1, true);
706 fileset->incexe->name_list.init(1, true);
707 fileset->include_list.append(fileset->incexe);
711 fileset->incexe = (findINCEXE *)malloc(sizeof(findINCEXE));
712 memset(fileset->incexe, 0, sizeof(findINCEXE));
713 fileset->incexe->opts_list.init(1, true);
714 fileset->incexe->name_list.init(1, true);
715 fileset->exclude_list.append(fileset->incexe);
721 /* File item to either include/include list */
722 state = state_include;
723 add_file_to_fileset(jcr, item, fileset);
726 current_opts = start_options(ff);
730 preg = (regex_t *)malloc(sizeof(regex_t));
731 rc = regcomp(preg, item, REG_EXTENDED);
733 regerror(rc, preg, prbuf, sizeof(prbuf));
736 Jmsg(jcr, M_FATAL, 0, "REGEX %s compile error. ERR=%s\n", item, prbuf);
740 current_opts->regex.append(preg);
741 state = state_options;
744 current_opts = start_options(ff);
745 current_opts->base.append(bstrdup(item));
746 state = state_options;
749 current_opts = start_options(ff);
750 current_opts->wild.append(bstrdup(item));
751 state = state_options;
754 current_opts = start_options(ff);
755 set_options(current_opts, item);
756 state = state_options;
759 current_opts = start_options(ff);
760 current_opts->reader = bstrdup(item);
761 state = state_options;
764 current_opts = start_options(ff);
765 current_opts->writer = bstrdup(item);
766 state = state_options;
769 Jmsg(jcr, M_FATAL, 0, "Invalid FileSet command: %s\n", item);
773 ff->fileset->state = state;
776 static bool term_fileset(JCR *jcr)
778 FF_PKT *ff = (FF_PKT *)jcr->ff;
779 findFILESET *fileset = ff->fileset;
782 for (i=0; i<fileset->include_list.size(); i++) {
783 findINCEXE *incexe = (findINCEXE *)fileset->include_list.get(i);
785 for (j=0; j<incexe->opts_list.size(); j++) {
786 findFOPTS *fo = (findFOPTS *)incexe->opts_list.get(j);
787 for (k=0; k<fo->regex.size(); k++) {
788 Dmsg1(400, "R %s\n", (char *)fo->regex.get(k));
790 for (k=0; k<fo->wild.size(); k++) {
791 Dmsg1(400, "W %s\n", (char *)fo->wild.get(k));
793 for (k=0; k<fo->base.size(); k++) {
794 Dmsg1(400, "B %s\n", (char *)fo->base.get(k));
797 Dmsg1(400, "D %s\n", fo->reader);
800 Dmsg1(400, "T %s\n", fo->writer);
803 for (j=0; j<incexe->name_list.size(); j++) {
804 Dmsg1(400, "F %s\n", (char *)incexe->name_list.get(j));
807 for (i=0; i<fileset->exclude_list.size(); i++) {
808 findINCEXE *incexe = (findINCEXE *)fileset->exclude_list.get(i);
810 for (j=0; j<incexe->opts_list.size(); j++) {
811 findFOPTS *fo = (findFOPTS *)incexe->opts_list.get(j);
812 for (k=0; k<fo->regex.size(); k++) {
813 Dmsg1(400, "R %s\n", (char *)fo->regex.get(k));
815 for (k=0; k<fo->wild.size(); k++) {
816 Dmsg1(400, "W %s\n", (char *)fo->wild.get(k));
818 for (k=0; k<fo->base.size(); k++) {
819 Dmsg1(400, "B %s\n", (char *)fo->base.get(k));
822 for (j=0; j<incexe->name_list.size(); j++) {
823 Dmsg1(400, "F %s\n", (char *)incexe->name_list.get(j));
826 return ff->fileset->state != state_error;
831 * As an optimization, we should do this during
832 * "compile" time in filed/job.c, and keep only a bit mask
833 * and the Verify options.
835 static void set_options(findFOPTS *fo, const char *opts)
840 for (p=opts; *p; p++) {
842 case 'a': /* alway replace */
843 case '0': /* no option */
846 fo->flags |= FO_EXCLUDE;
849 fo->flags |= FO_MULTIFS;
851 case 'h': /* no recursion */
852 fo->flags |= FO_NO_RECURSION;
858 fo->flags |= FO_NOREPLACE;
860 case 'p': /* use portable data format */
861 fo->flags |= FO_PORTABLE;
863 case 'r': /* read fifo */
864 fo->flags |= FO_READFIFO;
867 fo->flags |= FO_SHA1;
870 fo->flags |= FO_SPARSE;
873 fo->flags |= FO_MTIMEONLY;
876 fo->flags |= FO_KEEPATIME;
878 case 'V': /* verify options */
879 /* Copy Verify Options */
880 for (j=0; *p && *p != ':'; p++) {
881 fo->VerifyOpts[j] = *p;
882 if (j < (int)sizeof(fo->VerifyOpts) - 1) {
886 fo->VerifyOpts[j] = 0;
889 fo->flags |= FO_IF_NEWER;
891 case 'Z': /* gzip compression */
892 fo->flags |= FO_GZIP;
893 fo->GZIP_level = *++p - '0';
894 Dmsg1(200, "Compression level=%d\n", fo->GZIP_level);
897 Emsg1(M_ERROR, 0, "Unknown include/exclude option: %c\n", *p);
905 * Director is passing his Fileset
907 static int fileset_cmd(JCR *jcr)
909 BSOCK *dir = jcr->dir_bsock;
911 if (!init_fileset(jcr)) {
914 while (bnet_recv(dir) >= 0) {
915 strip_trailing_junk(dir->msg);
916 Dmsg1(400, "Fileset: %s\n", dir->msg);
917 add_fileset(jcr, dir->msg);
919 if (!term_fileset(jcr)) {
922 return bnet_fsend(dir, OKinc);
927 * Get list of files to exclude from Director
930 static int exclude_cmd(JCR *jcr)
932 BSOCK *dir = jcr->dir_bsock;
934 while (bnet_recv(dir) >= 0) {
935 dir->msg[dir->msglen] = 0;
936 strip_trailing_junk(dir->msg);
937 add_fname_to_list(jcr, dir->msg, EXC_LIST);
938 Dmsg1(110, "<dird: exclude file %s\n", dir->msg);
941 return bnet_fsend(dir, OKexc);
945 static int bootstrap_cmd(JCR *jcr)
947 BSOCK *dir = jcr->dir_bsock;
948 POOLMEM *fname = get_pool_memory(PM_FNAME);
951 if (jcr->RestoreBootstrap) {
952 unlink(jcr->RestoreBootstrap);
953 free_pool_memory(jcr->RestoreBootstrap);
955 Mmsg(fname, "%s/%s.%s.bootstrap", me->working_directory, me->hdr.name,
957 Dmsg1(400, "bootstrap=%s\n", fname);
958 jcr->RestoreBootstrap = fname;
959 bs = fopen(fname, "a+"); /* create file */
963 * Suck up what he is sending to us so that he will then
964 * read our error message.
966 while (bnet_recv(dir) >= 0)
969 Jmsg(jcr, M_FATAL, 0, _("Could not create bootstrap file %s: ERR=%s\n"),
970 jcr->RestoreBootstrap, be.strerror());
971 free_pool_memory(jcr->RestoreBootstrap);
972 jcr->RestoreBootstrap = NULL;
973 set_jcr_job_status(jcr, JS_ErrorTerminated);
977 while (bnet_recv(dir) >= 0) {
978 Dmsg1(200, "filed<dird: bootstrap file %s\n", dir->msg);
983 return bnet_fsend(dir, OKbootstrap);
988 * Get backup level from Director
991 static int level_cmd(JCR *jcr)
993 BSOCK *dir = jcr->dir_bsock;
994 POOLMEM *level, *buf = NULL;
999 level = get_memory(dir->msglen+1);
1000 Dmsg1(110, "level_cmd: %s", dir->msg);
1001 if (sscanf(dir->msg, "level = %s ", level) != 1) {
1004 /* Base backup requested? */
1005 if (strcmp(level, "base") == 0) {
1006 jcr->JobLevel = L_BASE;
1007 /* Full backup requested? */
1008 } else if (strcmp(level, "full") == 0) {
1009 jcr->JobLevel = L_FULL;
1011 * Backup requested since <date> <time>
1012 * This form is also used for incremental and differential
1013 * This code is deprecated. See since_utime for new code.
1015 } else if (strcmp(level, "since") == 0) {
1016 jcr->JobLevel = L_SINCE;
1017 if (sscanf(dir->msg, "level = since %d-%d-%d %d:%d:%d mtime_only=%d",
1018 &tm.tm_year, &tm.tm_mon, &tm.tm_mday,
1019 &tm.tm_hour, &tm.tm_min, &tm.tm_sec, &mtime_only) != 7) {
1024 tm.tm_wday = tm.tm_yday = 0;
1026 mtime = mktime(&tm);
1027 Dmsg2(100, "Got since time: %s mtime_only=%d\n", ctime(&mtime), mtime_only);
1028 jcr->incremental = 1; /* set incremental or decremental backup */
1029 jcr->mtime = mtime; /* set since time */
1031 * We get his UTC since time, then sync the clocks and correct it
1032 * to agree with our clock.
1034 } else if (strcmp(level, "since_utime") == 0) {
1035 buf = get_memory(dir->msglen+1);
1036 utime_t since_time, adj;
1037 btime_t his_time, bt_start, rt=0, bt_adj=0;
1038 jcr->JobLevel = L_SINCE;
1039 if (sscanf(dir->msg, "level = since_utime %s mtime_only=%d",
1040 buf, &mtime_only) != 2) {
1043 since_time = str_to_uint64(buf); /* this is the since time */
1044 char ed1[50], ed2[50];
1046 * Sync clocks by polling him for the time. We take
1047 * 10 samples of his time throwing out the first two.
1049 for (int i=0; i<10; i++) {
1050 bt_start = get_current_btime();
1051 bnet_sig(dir, BNET_BTIME); /* poll for time */
1052 if (bnet_recv(dir) <= 0) { /* get response */
1055 if (sscanf(dir->msg, "btime %s", buf) != 1) {
1058 if (i < 2) { /* toss first two results */
1061 his_time = str_to_uint64(buf);
1062 rt = get_current_btime() - bt_start; /* compute round trip time */
1063 bt_adj -= his_time - bt_start - rt/2;
1064 Dmsg2(200, "rt=%s adj=%s\n", edit_uint64(rt, ed1), edit_uint64(bt_adj, ed2));
1067 bt_adj = bt_adj / 8; /* compute average time */
1068 Dmsg2(100, "rt=%s adj=%s\n", edit_uint64(rt, ed1), edit_uint64(bt_adj, ed2));
1069 adj = btime_to_utime(bt_adj);
1070 since_time += adj; /* adjust for clock difference */
1072 Jmsg(jcr, M_INFO, 0, _("Since time adjusted by %d seconds.\n"), adj);
1074 bnet_sig(dir, BNET_EOD);
1076 Dmsg2(100, "adj = %d since_time=%d\n", (int)adj, (int)since_time);
1077 jcr->incremental = 1; /* set incremental or decremental backup */
1078 jcr->mtime = (time_t)since_time; /* set since time */
1080 Jmsg1(jcr, M_FATAL, 0, "Unknown backup level: %s\n", level);
1088 return bnet_fsend(dir, OKlevel);
1091 pm_strcpy(&jcr->errmsg, dir->msg);
1092 Jmsg1(jcr, M_FATAL, 0, _("Bad level command: %s\n"), jcr->errmsg);
1101 * Get session parameters from Director -- this is for a Restore command
1103 static int session_cmd(JCR *jcr)
1105 BSOCK *dir = jcr->dir_bsock;
1107 Dmsg1(100, "SessionCmd: %s", dir->msg);
1108 if (sscanf(dir->msg, sessioncmd, jcr->VolumeName,
1109 &jcr->VolSessionId, &jcr->VolSessionTime,
1110 &jcr->StartFile, &jcr->EndFile,
1111 &jcr->StartBlock, &jcr->EndBlock) != 7) {
1112 pm_strcpy(&jcr->errmsg, dir->msg);
1113 Jmsg(jcr, M_FATAL, 0, "Bad session command: %s", jcr->errmsg);
1117 return bnet_fsend(dir, OKsession);
1121 * Get address of storage daemon from Director
1124 static int storage_cmd(JCR *jcr)
1126 int stored_port; /* storage daemon port */
1127 int enable_ssl; /* enable ssl to sd */
1128 BSOCK *dir = jcr->dir_bsock;
1129 BSOCK *sd; /* storage daemon bsock */
1131 Dmsg1(100, "StorageCmd: %s", dir->msg);
1132 if (sscanf(dir->msg, storaddr, &jcr->stored_addr, &stored_port, &enable_ssl) != 3) {
1133 pm_strcpy(&jcr->errmsg, dir->msg);
1134 Jmsg(jcr, M_FATAL, 0, _("Bad storage command: %s"), jcr->errmsg);
1137 Dmsg3(110, "Open storage: %s:%d ssl=%d\n", jcr->stored_addr, stored_port, enable_ssl);
1138 /* Open command communications with Storage daemon */
1139 /* Try to connect for 1 hour at 10 second intervals */
1140 sd = bnet_connect(jcr, 10, (int)me->SDConnectTimeout, _("Storage daemon"),
1141 jcr->stored_addr, NULL, stored_port, 1);
1143 Jmsg(jcr, M_FATAL, 0, _("Failed to connect to Storage daemon: %s:%d\n"),
1144 jcr->stored_addr, stored_port);
1145 Dmsg2(100, "Failed to connect to Storage daemon: %s:%d\n",
1146 jcr->stored_addr, stored_port);
1149 Dmsg0(110, "Connection OK to SD.\n");
1151 jcr->store_bsock = sd;
1153 bnet_fsend(sd, "Hello Start Job %s\n", jcr->Job);
1154 if (!authenticate_storagedaemon(jcr)) {
1155 Jmsg(jcr, M_FATAL, 0, _("Failed to authenticate Storage daemon.\n"));
1158 Dmsg0(110, "Authenticated with SD.\n");
1160 /* Send OK to Director */
1161 return bnet_fsend(dir, OKstore);
1166 * Do a backup. For now, we handle only Full and Incremental.
1168 static int backup_cmd(JCR *jcr)
1170 BSOCK *dir = jcr->dir_bsock;
1171 BSOCK *sd = jcr->store_bsock;
1174 char ed1[50], ed2[50];
1176 set_jcr_job_status(jcr, JS_Blocked);
1177 jcr->JobType = JT_BACKUP;
1178 Dmsg1(100, "begin backup ff=%p\n", (FF_PKT *)jcr->ff);
1181 Jmsg(jcr, M_FATAL, 0, _("Cannot contact Storage daemon\n"));
1185 bnet_fsend(dir, OKbackup);
1186 Dmsg1(110, "bfiled>dird: %s", dir->msg);
1189 * Send Append Open Session to Storage daemon
1191 bnet_fsend(sd, append_open);
1192 Dmsg1(110, ">stored: %s", sd->msg);
1194 * Expect to receive back the Ticket number
1196 if (bget_msg(sd) >= 0) {
1197 Dmsg1(110, "<stored: %s", sd->msg);
1198 if (sscanf(sd->msg, OK_open, &jcr->Ticket) != 1) {
1199 Jmsg(jcr, M_FATAL, 0, _("Bad response to append open: %s\n"), sd->msg);
1202 Dmsg1(110, "Got Ticket=%d\n", jcr->Ticket);
1204 Jmsg(jcr, M_FATAL, 0, _("Bad response from stored to open command\n"));
1209 * Send Append data command to Storage daemon
1211 bnet_fsend(sd, append_data, jcr->Ticket);
1212 Dmsg1(110, ">stored: %s", sd->msg);
1215 * Expect to get OK data
1217 Dmsg1(110, "<stored: %s", sd->msg);
1218 if (!response(jcr, sd, OK_data, "Append Data")) {
1223 * Send Files to Storage daemon
1225 Dmsg1(110, "begin blast ff=%p\n", (FF_PKT *)jcr->ff);
1226 if (!blast_data_to_storage_daemon(jcr, NULL)) {
1227 set_jcr_job_status(jcr, JS_ErrorTerminated);
1228 bnet_suppress_error_messages(sd, 1);
1229 Dmsg0(110, "Error in blast_data.\n");
1231 set_jcr_job_status(jcr, JS_Terminated);
1232 if (jcr->JobStatus != JS_Terminated) {
1233 bnet_suppress_error_messages(sd, 1);
1234 goto cleanup; /* bail out now */
1237 * Expect to get response to append_data from Storage daemon
1239 if (!response(jcr, sd, OK_append, "Append Data")) {
1240 set_jcr_job_status(jcr, JS_ErrorTerminated);
1245 * Send Append End Data to Storage daemon
1247 bnet_fsend(sd, append_end, jcr->Ticket);
1249 if (!response(jcr, sd, OK_end, "Append End")) {
1250 set_jcr_job_status(jcr, JS_ErrorTerminated);
1255 * Send Append Close to Storage daemon
1257 bnet_fsend(sd, append_close, jcr->Ticket);
1258 while (bget_msg(sd) >= 0) { /* stop on signal or error */
1259 if (sscanf(sd->msg, OK_close, &SDJobStatus) == 1) {
1261 Dmsg2(200, "SDJobStatus = %d %c\n", SDJobStatus, (char)SDJobStatus);
1265 Jmsg(jcr, M_FATAL, 0, _("Append Close with SD failed.\n"));
1268 if (SDJobStatus != JS_Terminated) {
1269 Jmsg(jcr, M_FATAL, 0, _("Bad status %d returned from Storage Daemon.\n"),
1275 bnet_fsend(dir, EndJob, jcr->JobStatus, jcr->JobFiles,
1276 edit_uint64(jcr->ReadBytes, ed1),
1277 edit_uint64(jcr->JobBytes, ed2), jcr->Errors);
1278 Dmsg1(110, "End FD msg: %s\n", dir->msg);
1280 return 0; /* return and stop command loop */
1284 * Do a Verify for Director
1287 static int verify_cmd(JCR *jcr)
1289 BSOCK *dir = jcr->dir_bsock;
1290 BSOCK *sd = jcr->store_bsock;
1291 char level[100], ed1[50], ed2[50];
1293 jcr->JobType = JT_VERIFY;
1294 if (sscanf(dir->msg, verifycmd, level) != 1) {
1295 bnet_fsend(dir, "2994 Bad verify command: %s\n", dir->msg);
1298 if (strcasecmp(level, "init") == 0) {
1299 jcr->JobLevel = L_VERIFY_INIT;
1300 } else if (strcasecmp(level, "catalog") == 0){
1301 jcr->JobLevel = L_VERIFY_CATALOG;
1302 } else if (strcasecmp(level, "volume") == 0){
1303 jcr->JobLevel = L_VERIFY_VOLUME_TO_CATALOG;
1304 } else if (strcasecmp(level, "data") == 0){
1305 jcr->JobLevel = L_VERIFY_DATA;
1306 } else if (strcasecmp(level, "disk_to_catalog") == 0) {
1307 jcr->JobLevel = L_VERIFY_DISK_TO_CATALOG;
1309 bnet_fsend(dir, "2994 Bad verify level: %s\n", dir->msg);
1313 bnet_fsend(dir, OKverify);
1314 Dmsg1(110, "bfiled>dird: %s", dir->msg);
1316 switch (jcr->JobLevel) {
1318 case L_VERIFY_CATALOG:
1321 case L_VERIFY_VOLUME_TO_CATALOG:
1322 if (!open_sd_read_session(jcr)) {
1325 start_dir_heartbeat(jcr);
1326 do_verify_volume(jcr);
1327 stop_dir_heartbeat(jcr);
1329 * Send Close session command to Storage daemon
1331 bnet_fsend(sd, read_close, jcr->Ticket);
1332 Dmsg1(130, "bfiled>stored: %s", sd->msg);
1334 /* ****FIXME**** check response */
1335 bget_msg(sd); /* get OK */
1337 /* Inform Storage daemon that we are done */
1338 bnet_sig(sd, BNET_TERMINATE);
1341 case L_VERIFY_DISK_TO_CATALOG:
1345 bnet_fsend(dir, "2994 Bad verify level: %s\n", dir->msg);
1349 bnet_sig(dir, BNET_EOD);
1351 /* Send termination status back to Dir */
1352 bnet_fsend(dir, EndJob, jcr->JobStatus, jcr->JobFiles,
1353 edit_uint64(jcr->ReadBytes, ed1),
1354 edit_uint64(jcr->JobBytes, ed2), jcr->Errors);
1356 /* Inform Director that we are done */
1357 bnet_sig(dir, BNET_TERMINATE);
1358 return 0; /* return and terminate command loop */
1362 * Do a Restore for Director
1365 static int restore_cmd(JCR *jcr)
1367 BSOCK *dir = jcr->dir_bsock;
1368 BSOCK *sd = jcr->store_bsock;
1372 char ed1[50], ed2[50];
1375 * Scan WHERE (base directory for restore) from command
1377 Dmsg0(150, "restore command\n");
1378 /* Pickup where string */
1379 where = get_memory(dir->msglen+1);
1382 if (sscanf(dir->msg, restorecmd, &replace, &prefix_links, where) != 3) {
1383 if (sscanf(dir->msg, restorecmd1, &replace, &prefix_links) != 2) {
1384 pm_strcpy(&jcr->errmsg, dir->msg);
1385 Jmsg(jcr, M_FATAL, 0, _("Bad replace command. CMD=%s\n"), jcr->errmsg);
1390 /* Turn / into nothing */
1391 if (where[0] == '/' && where[1] == 0) {
1395 Dmsg2(150, "Got replace %c, where=%s\n", replace, where);
1396 unbash_spaces(where);
1397 jcr->where = bstrdup(where);
1398 free_pool_memory(where);
1399 jcr->replace = replace;
1400 jcr->prefix_links = prefix_links;
1402 bnet_fsend(dir, OKrestore);
1403 Dmsg1(110, "bfiled>dird: %s", dir->msg);
1405 jcr->JobType = JT_RESTORE;
1407 set_jcr_job_status(jcr, JS_Blocked);
1408 if (!open_sd_read_session(jcr)) {
1409 set_jcr_job_status(jcr, JS_ErrorTerminated);
1413 set_jcr_job_status(jcr, JS_Running);
1416 * Do restore of files and data
1418 start_dir_heartbeat(jcr);
1420 stop_dir_heartbeat(jcr);
1422 set_jcr_job_status(jcr, JS_Terminated);
1423 if (jcr->JobStatus != JS_Terminated) {
1424 bnet_suppress_error_messages(sd, 1);
1428 * Send Close session command to Storage daemon
1430 bnet_fsend(sd, read_close, jcr->Ticket);
1431 Dmsg1(130, "bfiled>stored: %s", sd->msg);
1433 bget_msg(sd); /* get OK */
1435 /* Inform Storage daemon that we are done */
1436 bnet_sig(sd, BNET_TERMINATE);
1441 set_jcr_job_status(jcr, JS_ErrorTerminated);
1443 /* Send termination status back to Dir */
1444 bnet_fsend(dir, EndJob, jcr->JobStatus, jcr->JobFiles,
1445 edit_uint64(jcr->ReadBytes, ed1),
1446 edit_uint64(jcr->JobBytes, ed2), jcr->Errors);
1448 /* Inform Director that we are done */
1449 bnet_sig(dir, BNET_TERMINATE);
1451 Dmsg0(130, "Done in job.c\n");
1452 return 0; /* return and terminate command loop */
1455 static int open_sd_read_session(JCR *jcr)
1457 BSOCK *sd = jcr->store_bsock;
1460 Jmsg(jcr, M_FATAL, 0, _("Improper calling sequence.\n"));
1463 Dmsg4(120, "VolSessId=%ld VolsessT=%ld SF=%ld EF=%ld\n",
1464 jcr->VolSessionId, jcr->VolSessionTime, jcr->StartFile, jcr->EndFile);
1465 Dmsg2(120, "JobId=%d vol=%s\n", jcr->JobId, "DummyVolume");
1467 * Open Read Session with Storage daemon
1469 bnet_fsend(sd, read_open, jcr->VolumeName,
1470 jcr->VolSessionId, jcr->VolSessionTime, jcr->StartFile, jcr->EndFile,
1471 jcr->StartBlock, jcr->EndBlock);
1472 Dmsg1(110, ">stored: %s", sd->msg);
1477 if (bget_msg(sd) >= 0) {
1478 Dmsg1(110, "bfiled<stored: %s", sd->msg);
1479 if (sscanf(sd->msg, OK_open, &jcr->Ticket) != 1) {
1480 Jmsg(jcr, M_FATAL, 0, _("Bad response to SD read open: %s\n"), sd->msg);
1483 Dmsg1(110, "bfiled: got Ticket=%d\n", jcr->Ticket);
1485 Jmsg(jcr, M_FATAL, 0, _("Bad response from stored to read open command\n"));
1489 if (!send_bootstrap_file(jcr)) {
1494 * Start read of data with Storage daemon
1496 bnet_fsend(sd, read_data, jcr->Ticket);
1497 Dmsg1(110, ">stored: %s", sd->msg);
1502 if (!response(jcr, sd, OK_data, "Read Data")) {
1509 * Destroy the Job Control Record and associated
1510 * resources (sockets).
1512 static void filed_free_jcr(JCR *jcr)
1514 if (jcr->store_bsock) {
1515 bnet_close(jcr->store_bsock);
1517 if (jcr->RestoreBootstrap) {
1518 unlink(jcr->RestoreBootstrap);
1519 free_pool_memory(jcr->RestoreBootstrap);
1520 jcr->RestoreBootstrap = NULL;
1522 if (jcr->last_fname) {
1523 free_pool_memory(jcr->last_fname);
1525 if (jcr->RunAfterJob) {
1526 free_pool_memory(jcr->RunAfterJob);
1534 * Get response from Storage daemon to a command we
1535 * sent. Check that the response is OK.
1537 * Returns: 0 on failure
1540 int response(JCR *jcr, BSOCK *sd, char *resp, const char *cmd)
1545 if (bget_msg(sd) > 0) {
1546 Dmsg0(110, sd->msg);
1547 if (strcmp(sd->msg, resp) == 0) {
1551 if (job_canceled(jcr)) {
1552 return 0; /* if canceled avoid useless error messages */
1554 if (is_bnet_error(sd)) {
1555 Jmsg2(jcr, M_FATAL, 0, _("Comm error with SD. bad response to %s. ERR=%s\n"),
1556 cmd, bnet_strerror(sd));
1558 Jmsg3(jcr, M_FATAL, 0, _("Bad response to %s command. Wanted %s, got %s\n"),
1559 cmd, resp, sd->msg);
1564 static int send_bootstrap_file(JCR *jcr)
1568 BSOCK *sd = jcr->store_bsock;
1569 const char *bootstrap = "bootstrap\n";
1572 Dmsg1(400, "send_bootstrap_file: %s\n", jcr->RestoreBootstrap);
1573 if (!jcr->RestoreBootstrap) {
1576 bs = fopen(jcr->RestoreBootstrap, "r");
1579 Jmsg(jcr, M_FATAL, 0, _("Could not open bootstrap file %s: ERR=%s\n"),
1580 jcr->RestoreBootstrap, be.strerror());
1581 set_jcr_job_status(jcr, JS_ErrorTerminated);
1584 pm_strcpy(&sd->msg, bootstrap);
1585 sd->msglen = strlen(sd->msg);
1587 while (fgets(buf, sizeof(buf), bs)) {
1588 sd->msglen = Mmsg(sd->msg, "%s", buf);
1591 bnet_sig(sd, BNET_EOD);
1593 if (!response(jcr, sd, OKSDbootstrap, "Bootstrap")) {
1594 set_jcr_job_status(jcr, JS_ErrorTerminated);
1600 if (jcr->RestoreBootstrap) {
1601 unlink(jcr->RestoreBootstrap);
1602 free_pool_memory(jcr->RestoreBootstrap);
1603 jcr->RestoreBootstrap = NULL;