3 * Bacula Director Job processing routines
5 * 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 /* Forward referenced subroutines */
33 static void *job_thread(void *arg);
34 static char *edit_run_codes(JCR *jcr, char *omsg, char *imsg);
35 static void release_resource_locks(JCR *jcr);
36 static int acquire_resource_locks(JCR *jcr);
38 static void backoff_resource_locks(JCR *jcr, int count);
41 /* Exported subroutines */
42 void run_job(JCR *jcr);
45 /* Imported subroutines */
46 extern void term_scheduler();
47 extern void term_ua_server();
48 extern int do_backup(JCR *jcr);
49 extern int do_admin(JCR *jcr);
50 extern int do_restore(JCR *jcr);
51 extern int do_verify(JCR *jcr);
54 static semlock_t job_lock;
55 static pthread_mutex_t mutex;
56 static pthread_cond_t resource_wait;
57 static int waiting = 0; /* count of waiting threads */
62 /* Queue of jobs to be run */
63 workq_t job_wq; /* our job work queue */
67 void init_job_server(int max_workers)
71 if ((stat = sem_init(&job_lock, max_workers)) != 0) {
72 Emsg1(M_ABORT, 0, _("Could not init job lock: ERR=%s\n"), strerror(stat));
74 if ((stat = pthread_mutex_init(&mutex, NULL)) != 0) {
75 Emsg1(M_ABORT, 0, _("Could not init resource mutex: ERR=%s\n"), strerror(stat));
77 if ((stat = pthread_cond_init(&resource_wait, NULL)) != 0) {
78 Emsg1(M_ABORT, 0, _("Could not init resource wait: ERR=%s\n"), strerror(stat));
83 if ((stat = jobq_init(&job_queue, max_workers, job_thread)) != 0) {
84 Emsg1(M_ABORT, 0, _("Could not init job queue: ERR=%s\n"), strerror(stat));
87 /* This is the OLD work queue code to go away */
88 if ((stat = workq_init(&job_wq, max_workers, job_thread)) != 0) {
89 Emsg1(M_ABORT, 0, _("Could not init job work queue: ERR=%s\n"), strerror(stat));
97 * Run a job -- typically called by the scheduler, but may also
98 * be called by the UA (Console program).
101 void run_job(JCR *jcr)
108 workq_ele_t *work_item;
112 sm_check(__FILE__, __LINE__, True);
113 init_msg(jcr, jcr->messages);
114 create_unique_job_name(jcr, jcr->job->hdr.name);
115 set_jcr_job_status(jcr, JS_Created);
116 jcr->jr.SchedTime = jcr->sched_time;
117 jcr->jr.StartTime = jcr->start_time;
118 jcr->jr.Type = jcr->JobType;
119 jcr->jr.Level = jcr->JobLevel;
120 jcr->jr.JobStatus = jcr->JobStatus;
121 bstrncpy(jcr->jr.Name, jcr->job->hdr.name, sizeof(jcr->jr.Name));
122 bstrncpy(jcr->jr.Job, jcr->Job, sizeof(jcr->jr.Job));
124 /* Initialize termination condition variable */
125 if ((errstat = pthread_cond_init(&jcr->term_wait, NULL)) != 0) {
126 Jmsg1(jcr, M_FATAL, 0, _("Unable to init job cond variable: ERR=%s\n"), strerror(errstat));
127 set_jcr_job_status(jcr, JS_ErrorTerminated);
135 Dmsg0(50, "Open database\n");
136 jcr->db=db_init_database(jcr, jcr->catalog->db_name, jcr->catalog->db_user,
137 jcr->catalog->db_password, jcr->catalog->db_address,
138 jcr->catalog->db_port, jcr->catalog->db_socket);
139 if (!db_open_database(jcr, jcr->db)) {
140 Jmsg(jcr, M_FATAL, 0, "%s", db_strerror(jcr->db));
141 set_jcr_job_status(jcr, JS_ErrorTerminated);
145 Dmsg0(50, "DB opened\n");
150 jcr->jr.JobStatus = jcr->JobStatus;
151 if (!db_create_job_record(jcr, jcr->db, &jcr->jr)) {
152 Jmsg(jcr, M_FATAL, 0, "%s", db_strerror(jcr->db));
153 set_jcr_job_status(jcr, JS_ErrorTerminated);
157 jcr->JobId = jcr->jr.JobId;
158 ASSERT(jcr->jr.JobId > 0);
160 Dmsg4(30, "Created job record JobId=%d Name=%s Type=%c Level=%c\n",
161 jcr->JobId, jcr->Job, jcr->jr.Type, jcr->jr.Level);
162 Dmsg0(200, "Add jrc to work queue\n");
165 if ((stat = pthread_create(&tid, NULL, job_thread, (void *)jcr)) != 0) {
166 Emsg1(M_ABORT, 0, _("Unable to create job thread: ERR=%s\n"), strerror(stat));
170 /* Queue the job to be run */
171 if ((stat = jobq_add(&job_queue, jcr)) != 0) {
172 Emsg1(M_ABORT, 0, _("Could not add job queue: ERR=%s\n"), strerror(stat));
175 /* Queue the job to be run */
176 if ((stat = workq_add(&job_wq, (void *)jcr, &work_item, 0)) != 0) {
177 Emsg1(M_ABORT, 0, _("Could not add job to work queue: ERR=%s\n"), strerror(stat));
179 jcr->work_item = work_item;
182 Dmsg0(200, "Done run_job()\n");
186 * This is the engine called by workq_add() when we were pulled
187 * from the work queue.
188 * At this point, we are running in our own thread
190 static void *job_thread(void *arg)
192 JCR *jcr = (JCR *)arg;
194 pthread_detach(pthread_self());
195 sm_check(__FILE__, __LINE__, True);
198 if (!acquire_resource_locks(jcr)) {
199 set_jcr_job_status(jcr, JS_Canceled);
202 Dmsg0(200, "=====Start Job=========\n");
203 jcr->start_time = time(NULL); /* set the real start time */
204 set_jcr_job_status(jcr, JS_Running);
206 if (job_canceled(jcr)) {
207 update_job_end_record(jcr);
208 } else if (jcr->job->MaxStartDelay != 0 && jcr->job->MaxStartDelay <
209 (utime_t)(jcr->start_time - jcr->sched_time)) {
210 Jmsg(jcr, M_FATAL, 0, _("Job canceled because max start delay time exceeded.\n"));
211 set_jcr_job_status(jcr, JS_Canceled);
212 update_job_end_record(jcr);
216 if (jcr->job->RunBeforeJob) {
217 POOLMEM *before = get_pool_memory(PM_FNAME);
220 char line[MAXSTRING];
222 before = edit_run_codes(jcr, before, jcr->job->RunBeforeJob);
223 bpipe = open_bpipe(before, 0, "r");
224 free_pool_memory(before);
225 while (fgets(line, sizeof(line), bpipe->rfd)) {
226 Jmsg(jcr, M_INFO, 0, _("RunBefore: %s"), line);
228 status = close_bpipe(bpipe);
230 Jmsg(jcr, M_FATAL, 0, _("RunBeforeJob returned non-zero status=%d\n"),
232 set_jcr_job_status(jcr, JS_FatalError);
233 update_job_end_record(jcr);
237 switch (jcr->JobType) {
240 if (jcr->JobStatus == JS_Terminated) {
246 if (jcr->JobStatus == JS_Terminated) {
252 if (jcr->JobStatus == JS_Terminated) {
258 if (jcr->JobStatus == JS_Terminated) {
263 Pmsg1(0, "Unimplemented job type: %d\n", jcr->JobType);
266 if (jcr->job->RunAfterJob) {
267 POOLMEM *after = get_pool_memory(PM_FNAME);
270 char line[MAXSTRING];
272 after = edit_run_codes(jcr, after, jcr->job->RunAfterJob);
273 bpipe = open_bpipe(after, 0, "r");
274 free_pool_memory(after);
275 while (fgets(line, sizeof(line), bpipe->rfd)) {
276 Jmsg(jcr, M_INFO, 0, _("RunAfter: %s"), line);
278 status = close_bpipe(bpipe);
280 Jmsg(jcr, M_FATAL, 0, _("RunAfterJob returned non-zero status=%d\n"),
282 set_jcr_job_status(jcr, JS_FatalError);
283 update_job_end_record(jcr);
288 release_resource_locks(jcr);
289 if (jcr->job->RescheduleOnError &&
290 jcr->JobStatus != JS_Terminated &&
291 jcr->JobStatus != JS_Canceled &&
292 jcr->job->RescheduleTimes > 0 &&
293 jcr->reschedule_count < jcr->job->RescheduleTimes) {
296 * Reschedule this job by cleaning it up, but
297 * reuse the same JobId if possible.
299 jcr->reschedule_count++;
300 jcr->sched_time = time(NULL) + jcr->job->RescheduleInterval;
301 Dmsg2(100, "Rescheduled Job %s to re-run in %d seconds.\n", jcr->Job,
302 (int)jcr->job->RescheduleInterval);
303 jcr->JobStatus = JS_Created; /* force new status */
304 dird_free_jcr(jcr); /* partial cleanup old stuff */
305 if (jcr->JobBytes == 0) {
306 continue; /* reschedule the job */
309 * Something was actually backed up, so we cannot reuse
310 * the old JobId or there will be database record
311 * conflicts. We now create a new job, copying the
312 * appropriate fields.
314 JCR *njcr = new_jcr(sizeof(JCR), dird_free_jcr);
315 set_jcr_defaults(njcr, jcr->job);
316 njcr->reschedule_count = jcr->reschedule_count;
317 njcr->JobLevel = jcr->JobLevel;
318 njcr->JobStatus = jcr->JobStatus;
319 njcr->pool = jcr->pool;
320 njcr->store = jcr->store;
321 njcr->messages = jcr->messages;
328 Dmsg0(200, "Close DB\n");
329 db_close_database(jcr, jcr->db);
335 Dmsg0(50, "======== End Job ==========\n");
336 sm_check(__FILE__, __LINE__, True);
341 * Acquire the resources needed. These locks limit the
342 * number of jobs by each resource. We have limits on
343 * Jobs, Clients, Storage, and total jobs.
345 static int acquire_resource_locks(JCR *jcr)
348 time_t now = time(NULL);
349 time_t wtime = jcr->sched_time - now;
351 /* Wait until scheduled time arrives */
352 if (wtime > 0 && verbose) {
353 Jmsg(jcr, M_INFO, 0, _("Job %s waiting %d seconds for scheduled start time.\n"),
355 set_jcr_job_status(jcr, JS_WaitStartTime);
357 /* Check every 30 seconds if canceled */
359 Dmsg2(100, "Waiting on sched time, jobid=%d secs=%d\n", jcr->JobId, wtime);
363 bmicrosleep(wtime, 0);
364 if (job_canceled(jcr)) {
367 wtime = jcr->sched_time - time(NULL);
375 /* Initialize semaphores */
376 if (jcr->store->sem.valid != SEMLOCK_VALID) {
377 if ((stat = sem_init(&jcr->store->sem, jcr->store->MaxConcurrentJobs)) != 0) {
378 Emsg1(M_ABORT, 0, _("Could not init Storage semaphore: ERR=%s\n"), strerror(stat));
381 if (jcr->client->sem.valid != SEMLOCK_VALID) {
382 if ((stat = sem_init(&jcr->client->sem, jcr->client->MaxConcurrentJobs)) != 0) {
383 Emsg1(M_ABORT, 0, _("Could not init Client semaphore: ERR=%s\n"), strerror(stat));
386 if (jcr->job->sem.valid != SEMLOCK_VALID) {
387 if ((stat = sem_init(&jcr->job->sem, jcr->job->MaxConcurrentJobs)) != 0) {
388 Emsg1(M_ABORT, 0, _("Could not init Job semaphore: ERR=%s\n"), strerror(stat));
393 /* Acquire semaphore */
394 set_jcr_job_status(jcr, JS_WaitJobRes);
395 if ((stat = sem_lock(&jcr->job->sem)) != 0) {
396 Emsg1(M_ABORT, 0, _("Could not acquire Job max jobs lock: ERR=%s\n"), strerror(stat));
398 set_jcr_job_status(jcr, JS_WaitClientRes);
399 if ((stat = sem_trylock(&jcr->client->sem)) != 0) {
401 backoff_resource_locks(jcr, 1);
404 Emsg1(M_ABORT, 0, _("Could not acquire Client max jobs lock: ERR=%s\n"), strerror(stat));
407 set_jcr_job_status(jcr, JS_WaitStoreRes);
408 if ((stat = sem_trylock(&jcr->store->sem)) != 0) {
410 backoff_resource_locks(jcr, 2);
413 Emsg1(M_ABORT, 0, _("Could not acquire Storage max jobs lock: ERR=%s\n"), strerror(stat));
416 set_jcr_job_status(jcr, JS_WaitMaxJobs);
417 if ((stat = sem_trylock(&job_lock)) != 0) {
419 backoff_resource_locks(jcr, 3);
422 Emsg1(M_ABORT, 0, _("Could not acquire max jobs lock: ERR=%s\n"), strerror(stat));
428 if (job_canceled(jcr)) {
433 * Wait for a resource to be released either by backoff or
434 * by a job terminating.
437 pthread_cond_wait(&resource_wait, &mutex);
442 jcr->acquired_resource_locks = true;
449 * We could not get all the resource locks because
450 * too many jobs are running, so release any locks
451 * we did acquire, giving others a chance to use them
454 static void backoff_resource_locks(JCR *jcr, int count)
459 sem_unlock(&jcr->store->sem);
460 /* Fall through wanted */
462 sem_unlock(&jcr->client->sem);
463 /* Fall through wanted */
465 sem_unlock(&jcr->job->sem);
469 * Since we released a lock, if there are any threads
470 * waiting, wake them up so that they can try again.
473 pthread_cond_broadcast(&resource_wait);
480 * This is called at the end of the job to release
481 * any resource limits on the number of jobs. If
482 * there are any other jobs waiting, we wake them
483 * up so that they can try again.
485 static void release_resource_locks(JCR *jcr)
487 if (!jcr->acquired_resource_locks) {
488 return; /* Job canceled, no locks acquired */
492 sem_unlock(&jcr->store->sem);
493 sem_unlock(&jcr->client->sem);
494 sem_unlock(&jcr->job->sem);
495 sem_unlock(&job_lock);
497 pthread_cond_broadcast(&resource_wait);
499 jcr->acquired_resource_locks = false;
505 * Get or create a Client record for this Job
507 int get_or_create_client_record(JCR *jcr)
511 memset(&cr, 0, sizeof(cr));
512 bstrncpy(cr.Name, jcr->client->hdr.name, sizeof(cr.Name));
513 cr.AutoPrune = jcr->client->AutoPrune;
514 cr.FileRetention = jcr->client->FileRetention;
515 cr.JobRetention = jcr->client->JobRetention;
516 if (!jcr->client_name) {
517 jcr->client_name = get_pool_memory(PM_NAME);
519 pm_strcpy(&jcr->client_name, jcr->client->hdr.name);
520 if (!db_create_client_record(jcr, jcr->db, &cr)) {
521 Jmsg(jcr, M_FATAL, 0, _("Could not create Client record. ERR=%s\n"),
522 db_strerror(jcr->db));
525 jcr->jr.ClientId = cr.ClientId;
527 if (!jcr->client_uname) {
528 jcr->client_uname = get_pool_memory(PM_NAME);
530 pm_strcpy(&jcr->client_uname, cr.Uname);
532 Dmsg2(100, "Created Client %s record %d\n", jcr->client->hdr.name,
539 * Write status and such in DB
541 void update_job_end_record(JCR *jcr)
543 if (jcr->jr.EndTime == 0) {
544 jcr->jr.EndTime = time(NULL);
546 jcr->end_time = jcr->jr.EndTime;
547 jcr->jr.JobId = jcr->JobId;
548 jcr->jr.JobStatus = jcr->JobStatus;
549 jcr->jr.JobFiles = jcr->JobFiles;
550 jcr->jr.JobBytes = jcr->JobBytes;
551 jcr->jr.VolSessionId = jcr->VolSessionId;
552 jcr->jr.VolSessionTime = jcr->VolSessionTime;
553 if (!db_update_job_end_record(jcr, jcr->db, &jcr->jr)) {
554 Jmsg(jcr, M_WARNING, 0, _("Error updating job record. %s"),
555 db_strerror(jcr->db));
560 * Takes base_name and appends (unique) current
561 * date and time to form unique job name.
563 * Returns: unique job name in jcr->Job
564 * date/time in jcr->start_time
566 void create_unique_job_name(JCR *jcr, char *base_name)
568 /* Job start mutex */
569 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
570 static time_t last_start_time = 0;
573 char dt[MAX_TIME_LENGTH];
574 char name[MAX_NAME_LENGTH];
577 /* Guarantee unique start time -- maximum one per second, and
578 * thus unique Job Name
580 P(mutex); /* lock creation of jobs */
582 while (now == last_start_time) {
583 bmicrosleep(0, 500000);
586 last_start_time = now;
587 V(mutex); /* allow creation of jobs */
588 jcr->start_time = now;
589 /* Form Unique JobName */
590 localtime_r(&now, &tm);
591 /* Use only characters that are permitted in Windows filenames */
592 strftime(dt, sizeof(dt), "%Y-%m-%d_%H.%M.%S", &tm);
593 bstrncpy(name, base_name, sizeof(name));
594 name[sizeof(name)-22] = 0; /* truncate if too long */
595 bsnprintf(jcr->Job, sizeof(jcr->Job), "%s.%s", name, dt); /* add date & time */
596 /* Convert spaces into underscores */
597 for (p=jcr->Job; *p; p++) {
605 * Free the Job Control Record if no one is still using it.
606 * Called from main free_jcr() routine in src/lib/jcr.c so
607 * that we can do our Director specific cleanup of the jcr.
609 void dird_free_jcr(JCR *jcr)
611 Dmsg0(200, "Start dird free_jcr\n");
613 if (jcr->sd_auth_key) {
614 free(jcr->sd_auth_key);
615 jcr->sd_auth_key = NULL;
621 if (jcr->file_bsock) {
622 Dmsg0(200, "Close File bsock\n");
623 bnet_close(jcr->file_bsock);
624 jcr->file_bsock = NULL;
626 if (jcr->store_bsock) {
627 Dmsg0(200, "Close Store bsock\n");
628 bnet_close(jcr->store_bsock);
629 jcr->store_bsock = NULL;
632 Dmsg0(200, "Free JCR fname\n");
633 free_pool_memory(jcr->fname);
637 Dmsg0(200, "Free JCR stime\n");
638 free_pool_memory(jcr->stime);
641 if (jcr->RestoreBootstrap) {
642 free(jcr->RestoreBootstrap);
643 jcr->RestoreBootstrap = NULL;
645 if (jcr->client_uname) {
646 free_pool_memory(jcr->client_uname);
647 jcr->client_uname = NULL;
649 Dmsg0(200, "End dird free_jcr\n");
653 * Set some defaults in the JCR necessary to
654 * run. These items are pulled from the job
655 * definition as defaults, but can be overridden
656 * later either by the Run record in the Schedule resource,
657 * or by the Console program.
659 void set_jcr_defaults(JCR *jcr, JOB *job)
662 jcr->JobType = job->JobType;
663 jcr->JobLevel = job->level;
664 jcr->JobPriority = job->Priority;
665 jcr->store = job->storage;
666 jcr->client = job->client;
667 if (!jcr->client_name) {
668 jcr->client_name = get_pool_memory(PM_NAME);
670 pm_strcpy(&jcr->client_name, jcr->client->hdr.name);
671 jcr->pool = job->pool;
672 jcr->catalog = job->client->catalog;
673 jcr->fileset = job->fileset;
674 jcr->messages = job->messages;
675 if (jcr->RestoreBootstrap) {
676 free(jcr->RestoreBootstrap);
678 /* This can be overridden by Console program */
679 if (job->RestoreBootstrap) {
680 jcr->RestoreBootstrap = bstrdup(job->RestoreBootstrap);
682 /* If no default level given, set one */
683 if (jcr->JobLevel == 0) {
684 switch (jcr->JobType) {
686 jcr->JobLevel = L_VERIFY_CATALOG;
689 jcr->JobLevel = L_INCREMENTAL;
693 jcr->JobLevel = L_FULL;
702 * Edit codes into Run command
705 * %d = Director's name
713 * omsg = edited output message
714 * imsg = input string containing edit codes (%x)
717 static char *edit_run_codes(JCR *jcr, char *omsg, char *imsg)
724 Dmsg1(200, "edit_run_codes: %s\n", imsg);
725 for (p=imsg; *p; p++) {
732 str = jcr->client_name;
741 str = job_status_to_str(jcr->JobStatus);
744 sprintf(add, "%d", jcr->JobId);
751 str = job_level_to_str(jcr->JobLevel);
754 str = jcr->job->hdr.name;
757 str = job_type_to_str(jcr->JobType);
771 Dmsg1(200, "add_str %s\n", str);
772 pm_strcat(&omsg, (char *)str);
773 Dmsg1(200, "omsg=%s\n", omsg);