]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/dird/job.c
This commit was manufactured by cvs2svn to create tag
[bacula/bacula] / bacula / src / dird / job.c
index ebf9bac1b4b65c4f390d4c993e8d2170e26ac731..e84a6422ccb1ee96e5cdbe9c52ed962f7ac3f6a4 100644 (file)
@@ -7,7 +7,7 @@
  *    Version $Id$
  */
 /*
-   Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
+   Copyright (C) 2000-2003 Kern Sibbald and John Walker
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
 #include "dird.h"
 
 /* Forward referenced subroutines */
-static void job_thread(void *arg);
-static char *edit_run_codes(JCR *jcr, char *omsg, char *imsg);
+static void *job_thread(void *arg);
 
 /* Exported subroutines */
-void run_job(JCR *jcr);
-void init_job_server(int max_workers);
 
 
 /* Imported subroutines */
 extern void term_scheduler();
 extern void term_ua_server();
 extern int do_backup(JCR *jcr);
+extern int do_admin(JCR *jcr);
 extern int do_restore(JCR *jcr);
 extern int do_verify(JCR *jcr);
-extern void backup_cleanup(void);
-
-/* Queue of jobs to be run */
-static workq_t job_wq;               /* our job work queue */
 
+jobq_t job_queue;
 
 void init_job_server(int max_workers)
 {
    int stat;
-
-   if ((stat = workq_init(&job_wq, max_workers, job_thread)) != 0) {
-      Emsg1(M_ABORT, 0, _("Could not init job work queue: ERR=%s\n"), strerror(stat));
+   if ((stat = jobq_init(&job_queue, max_workers, job_thread)) != 0) {
+      Emsg1(M_ABORT, 0, _("Could not init job queue: ERR=%s\n"), strerror(stat));
    }
    return;
 }
@@ -69,20 +63,23 @@ void run_job(JCR *jcr)
 {
    int stat, errstat;
 
+   sm_check(__FILE__, __LINE__, True);
    init_msg(jcr, jcr->messages);
    create_unique_job_name(jcr, jcr->job->hdr.name);
+   set_jcr_job_status(jcr, JS_Created);
    jcr->jr.SchedTime = jcr->sched_time;
    jcr->jr.StartTime = jcr->start_time;
+   jcr->jr.EndTime = 0;              /* perhaps rescheduled, clear it */
    jcr->jr.Type = jcr->JobType;
    jcr->jr.Level = jcr->JobLevel;
    jcr->jr.JobStatus = jcr->JobStatus;
-   strcpy(jcr->jr.Name, jcr->job->hdr.name);
-   strcpy(jcr->jr.Job, jcr->Job);
+   bstrncpy(jcr->jr.Name, jcr->job->hdr.name, sizeof(jcr->jr.Name));
+   bstrncpy(jcr->jr.Job, jcr->Job, sizeof(jcr->jr.Job));
 
    /* Initialize termination condition variable */
    if ((errstat = pthread_cond_init(&jcr->term_wait, NULL)) != 0) {
       Jmsg1(jcr, M_FATAL, 0, _("Unable to init job cond variable: ERR=%s\n"), strerror(errstat));
-      jcr->JobStatus = JS_ErrorTerminated;
+      set_jcr_job_status(jcr, JS_ErrorTerminated);
       free_jcr(jcr);
       return;
    }
@@ -92,11 +89,15 @@ void run_job(JCR *jcr)
     */
    Dmsg0(50, "Open database\n");
    jcr->db=db_init_database(jcr, jcr->catalog->db_name, jcr->catalog->db_user,
-                           jcr->catalog->db_password);
-   if (!db_open_database(jcr->db)) {
-      Jmsg(jcr, M_FATAL, 0, "%s", db_strerror(jcr->db));
-      db_close_database(jcr->db);
-      jcr->JobStatus = JS_ErrorTerminated;
+                           jcr->catalog->db_password, jcr->catalog->db_address,
+                           jcr->catalog->db_port, jcr->catalog->db_socket);
+   if (!jcr->db || !db_open_database(jcr, jcr->db)) {
+      Jmsg(jcr, M_FATAL, 0, _("Could not open database \"%s\".\n"),
+                jcr->catalog->db_name);
+      if (jcr->db) {
+         Jmsg(jcr, M_FATAL, 0, "%s", db_strerror(jcr->db));
+      }
+      set_jcr_job_status(jcr, JS_ErrorTerminated);
       free_jcr(jcr);
       return;
    }
@@ -106,61 +107,77 @@ void run_job(JCR *jcr)
     * Create Job record  
     */
    jcr->jr.JobStatus = jcr->JobStatus;
-   if (!db_create_job_record(jcr->db, &jcr->jr)) {
+   if (!db_create_job_record(jcr, jcr->db, &jcr->jr)) {
       Jmsg(jcr, M_FATAL, 0, "%s", db_strerror(jcr->db));
-      db_close_database(jcr->db);
-      jcr->JobStatus = JS_ErrorTerminated;
+      set_jcr_job_status(jcr, JS_ErrorTerminated);
       free_jcr(jcr);
       return;
    }
    jcr->JobId = jcr->jr.JobId;
    ASSERT(jcr->jr.JobId > 0);
 
-   Dmsg4(30, "Created job record JobId=%d Name=%s Type=%c Level=%c\n", 
+   Dmsg4(50, "Created job record JobId=%d Name=%s Type=%c Level=%c\n", 
        jcr->JobId, jcr->Job, jcr->jr.Type, jcr->jr.Level);
    Dmsg0(200, "Add jrc to work queue\n");
 
-
    /* Queue the job to be run */
-   if ((stat = workq_add(&job_wq, (void *)jcr)) != 0) {
-      Emsg1(M_ABORT, 0, _("Could not add job to work queue: ERR=%s\n"), strerror(stat));
+   if ((stat = jobq_add(&job_queue, jcr)) != 0) {
+      Emsg1(M_ABORT, 0, _("Could not add job queue: ERR=%s\n"), strerror(stat));
    }
-   Dmsg0(200, "Done run_job()\n");
+   Dmsg0(100, "Done run_job()\n");
 }
 
 /* 
- * This is the engine called by workq_add() when we were pulled               
+ * This is the engine called by job_add() when we were pulled               
  *  from the work queue.
- *  At this point, we are running in our own thread 
+ *  At this point, we are running in our own thread and all
+ *    necessary resources are allocated -- see jobq.c
  */
-static void job_thread(void *arg)
+static void *job_thread(void *arg)
 {
-   time_t now;
    JCR *jcr = (JCR *)arg;
 
-   time(&now);
-
-   Dmsg0(100, "=====Start Job=========\n");
-   jcr->start_time = now;            /* set the real start time */
-   if (jcr->job->MaxStartDelay != 0 && jcr->job->MaxStartDelay <
-       (btime_t)(jcr->start_time - jcr->sched_time)) {
-      Jmsg(jcr, M_FATAL, 0, _("Job cancelled because max delay time exceeded.\n"));
-      jcr->JobStatus = JS_ErrorTerminated;
-      update_job_end_record(jcr);
-   } else {
-
-      /* Run Job */
-      jcr->JobStatus = JS_Running;
-
-      if (jcr->job->RunBeforeJob) {
-        POOLMEM *before = get_pool_memory(PM_FNAME);
-        int status;
-        
-        before = edit_run_codes(jcr, before, jcr->job->RunBeforeJob);
-        status = run_program(before, 0, NULL);
-        free_pool_memory(before);
-      }
-      switch (jcr->JobType) {
+   pthread_detach(pthread_self());
+   sm_check(__FILE__, __LINE__, True);
+
+   for ( ;; ) {
+
+      Dmsg0(200, "=====Start Job=========\n");
+      jcr->start_time = time(NULL);     /* set the real start time */
+      set_jcr_job_status(jcr, JS_Running);
+
+      if (job_canceled(jcr)) {
+        update_job_end_record(jcr);
+      } else if (jcr->job->MaxStartDelay != 0 && jcr->job->MaxStartDelay <
+         (utime_t)(jcr->start_time - jcr->sched_time)) {
+         Jmsg(jcr, M_FATAL, 0, _("Job canceled because max start delay time exceeded.\n"));
+        set_jcr_job_status(jcr, JS_Canceled);
+        update_job_end_record(jcr);
+      } else {
+
+        /* Run Job */
+        if (jcr->job->RunBeforeJob) {
+           POOLMEM *before = get_pool_memory(PM_FNAME);
+           int status;
+           BPIPE *bpipe;
+           char line[MAXSTRING];
+           
+            before = edit_job_codes(jcr, before, jcr->job->RunBeforeJob, "");
+            bpipe = open_bpipe(before, 0, "r");
+           free_pool_memory(before);
+           while (fgets(line, sizeof(line), bpipe->rfd)) {
+               Jmsg(jcr, M_INFO, 0, _("RunBefore: %s"), line);
+           }
+           status = close_bpipe(bpipe);
+           if (status != 0) {
+               Jmsg(jcr, M_FATAL, 0, _("RunBeforeJob returned non-zero status=%d\n"),
+                 status);
+              set_jcr_job_status(jcr, JS_FatalError);
+              update_job_end_record(jcr);
+              goto bail_out;
+           }
+        }
+        switch (jcr->JobType) {
         case JT_BACKUP:
            do_backup(jcr);
            if (jcr->JobStatus == JS_Terminated) {
@@ -180,28 +197,56 @@ static void job_thread(void *arg)
            }
            break;
         case JT_ADMIN:
-           /* No actual job */
-           do_autoprune(jcr);
-           jcr->JobStatus = JS_Terminated;
+           do_admin(jcr);
+           if (jcr->JobStatus == JS_Terminated) {
+              do_autoprune(jcr);
+           }
            break;
         default:
             Pmsg1(0, "Unimplemented job type: %d\n", jcr->JobType);
            break;
         }
+        if ((jcr->job->RunAfterJob && jcr->JobStatus == JS_Terminated) ||
+            (jcr->job->RunAfterFailedJob && jcr->JobStatus != JS_Terminated)) {
+           POOLMEM *after = get_pool_memory(PM_FNAME);
+           int status;
+           BPIPE *bpipe;
+           char line[MAXSTRING];
+           
+           if (jcr->JobStatus == JS_Terminated) {
+               after = edit_job_codes(jcr, after, jcr->job->RunAfterJob, "");
+           } else {
+               after = edit_job_codes(jcr, after, jcr->job->RunAfterFailedJob, "");
+           }
+            bpipe = open_bpipe(after, 0, "r");
+           free_pool_memory(after);
+           while (fgets(line, sizeof(line), bpipe->rfd)) {
+               Jmsg(jcr, M_INFO, 0, _("RunAfter: %s"), line);
+           }
+           status = close_bpipe(bpipe);
+           if (status != 0) {
+              if (jcr->JobStatus == JS_Terminated) {
+                  Jmsg(jcr, M_FATAL, 0, _("RunAfterJob returned non-zero status=%d\n"),
+                      status);
+              } else {
+                  Jmsg(jcr, M_FATAL, 0, _("RunAfterFailedJob returned non-zero status=%d\n"),
+                      status);
+              }
+              set_jcr_job_status(jcr, JS_FatalError);
+              update_job_end_record(jcr);
+           }
+        }
+      }
+bail_out:
+      break;
    }
-   if (jcr->job->RunAfterJob) {
-      POOLMEM *after = get_pool_memory(PM_FNAME);
-      int status;
-      
-      after = edit_run_codes(jcr, after, jcr->job->RunAfterJob);
-      status = run_program(after, 0, NULL);
-      free_pool_memory(after);
-   }
-   Dmsg0(50, "Before free jcr\n");
-   free_jcr(jcr);
+
    Dmsg0(50, "======== End Job ==========\n");
+   sm_check(__FILE__, __LINE__, True);
+   return NULL;
 }
 
+
 /*
  * Get or create a Client record for this Job
  */
@@ -210,21 +255,26 @@ int get_or_create_client_record(JCR *jcr)
    CLIENT_DBR cr;
 
    memset(&cr, 0, sizeof(cr));
-   strcpy(cr.Name, jcr->client->hdr.name);
+   bstrncpy(cr.Name, jcr->client->hdr.name, sizeof(cr.Name));
    cr.AutoPrune = jcr->client->AutoPrune;
    cr.FileRetention = jcr->client->FileRetention;
    cr.JobRetention = jcr->client->JobRetention;
-   if (jcr->client_name) {
-      free_pool_memory(jcr->client_name);
+   if (!jcr->client_name) {
+      jcr->client_name = get_pool_memory(PM_NAME);
    }
-   jcr->client_name = get_memory(strlen(jcr->client->hdr.name) + 1);
-   strcpy(jcr->client_name, jcr->client->hdr.name);
-   if (!db_create_client_record(jcr->db, &cr)) {
-      Jmsg(jcr, M_FATAL, 0, _("Could not create Client record. %s"), 
+   pm_strcpy(&jcr->client_name, jcr->client->hdr.name);
+   if (!db_create_client_record(jcr, jcr->db, &cr)) {
+      Jmsg(jcr, M_FATAL, 0, _("Could not create Client record. ERR=%s\n"), 
         db_strerror(jcr->db));
       return 0;
    }
    jcr->jr.ClientId = cr.ClientId;
+   if (cr.Uname[0]) {
+      if (!jcr->client_uname) {
+        jcr->client_uname = get_pool_memory(PM_NAME);
+      }
+      pm_strcpy(&jcr->client_uname, cr.Uname);
+   }
    Dmsg2(100, "Created Client %s record %d\n", jcr->client->hdr.name, 
       jcr->jr.ClientId);
    return 1;
@@ -246,7 +296,7 @@ void update_job_end_record(JCR *jcr)
    jcr->jr.JobBytes = jcr->JobBytes;
    jcr->jr.VolSessionId = jcr->VolSessionId;
    jcr->jr.VolSessionTime = jcr->VolSessionTime;
-   if (!db_update_job_end_record(jcr->db, &jcr->jr)) {
+   if (!db_update_job_end_record(jcr, jcr->db, &jcr->jr)) {
       Jmsg(jcr, M_WARNING, 0, _("Error updating job record. %s"), 
         db_strerror(jcr->db));
    }
@@ -274,20 +324,21 @@ void create_unique_job_name(JCR *jcr, char *base_name)
     * thus unique Job Name 
     */
    P(mutex);                         /* lock creation of jobs */
-   time(&now);
+   now = time(NULL);
    while (now == last_start_time) {
-      sleep(1);
-      time(&now);
+      bmicrosleep(0, 500000);
+      now = time(NULL);
    }
    last_start_time = now;
    V(mutex);                         /* allow creation of jobs */
    jcr->start_time = now;
    /* Form Unique JobName */
    localtime_r(&now, &tm);
-   strftime(dt, sizeof(dt), "%Y-%m-%d.%H:%M:%S", &tm); 
-   strncpy(name, base_name, sizeof(name));
+   /* Use only characters that are permitted in Windows filenames */
+   strftime(dt, sizeof(dt), "%Y-%m-%d_%H.%M.%S", &tm); 
+   bstrncpy(name, base_name, sizeof(name));
    name[sizeof(name)-22] = 0;         /* truncate if too long */
-   sprintf(jcr->Job, "%s.%s", name, dt); /* add date & time */
+   bsnprintf(jcr->Job, sizeof(jcr->Job), "%s.%s", name, dt); /* add date & time */
    /* Convert spaces into underscores */
    for (p=jcr->Job; *p; p++) {
       if (*p == ' ') {
@@ -305,32 +356,43 @@ void dird_free_jcr(JCR *jcr)
 {
    Dmsg0(200, "Start dird free_jcr\n");
 
+   if (jcr->sd_auth_key) {
+      free(jcr->sd_auth_key);
+      jcr->sd_auth_key = NULL;
+   }
+   if (jcr->where) {
+      free(jcr->where);
+      jcr->where = NULL;
+   }
    if (jcr->file_bsock) {
       Dmsg0(200, "Close File bsock\n");
       bnet_close(jcr->file_bsock);
+      jcr->file_bsock = NULL;
    }
    if (jcr->store_bsock) {
       Dmsg0(200, "Close Store bsock\n");
       bnet_close(jcr->store_bsock);
+      jcr->store_bsock = NULL;
    }
    if (jcr->fname) {  
       Dmsg0(200, "Free JCR fname\n");
       free_pool_memory(jcr->fname);
+      jcr->fname = NULL;
    }
    if (jcr->stime) {
       Dmsg0(200, "Free JCR stime\n");
       free_pool_memory(jcr->stime);
-   }
-   if (jcr->db) {
-      Dmsg0(200, "Close DB\n");
-      db_close_database(jcr->db);
-   }
-   if (jcr->RestoreWhere) {
-      free(jcr->RestoreWhere);
+      jcr->stime = NULL;
    }
    if (jcr->RestoreBootstrap) {
       free(jcr->RestoreBootstrap);
+      jcr->RestoreBootstrap = NULL;
+   }
+   if (jcr->client_uname) {
+      free_pool_memory(jcr->client_uname);
+      jcr->client_uname = NULL;
    }
+   pthread_cond_destroy(&jcr->term_wait);
    Dmsg0(200, "End dird free_jcr\n");
 }
 
@@ -345,20 +407,29 @@ void set_jcr_defaults(JCR *jcr, JOB *job)
 {
    jcr->job = job;
    jcr->JobType = job->JobType;
-   jcr->JobLevel = job->level;
+   switch (jcr->JobType) {
+   case JT_ADMIN:
+   case JT_RESTORE:
+      jcr->JobLevel = L_NONE;
+      break;
+   default:
+      jcr->JobLevel = job->level;
+      break;
+   }
+   jcr->JobPriority = job->Priority;
    jcr->store = job->storage;
    jcr->client = job->client;
-   if (jcr->client_name) {
-      free_pool_memory(jcr->client_name);
+   if (!jcr->client_name) {
+      jcr->client_name = get_pool_memory(PM_NAME);
    }
-   jcr->client_name = get_memory(strlen(jcr->client->hdr.name) + 1);
-   strcpy(jcr->client_name, jcr->client->hdr.name);
+   pm_strcpy(&jcr->client_name, jcr->client->hdr.name);
    jcr->pool = job->pool;
    jcr->catalog = job->client->catalog;
    jcr->fileset = job->fileset;
    jcr->messages = job->messages; 
    if (jcr->RestoreBootstrap) {
       free(jcr->RestoreBootstrap);
+      jcr->RestoreBootstrap = NULL;
    }
    /* This can be overridden by Console program */
    if (job->RestoreBootstrap) {
@@ -373,85 +444,12 @@ void set_jcr_defaults(JCR *jcr, JOB *job)
       case JT_BACKUP:
         jcr->JobLevel = L_INCREMENTAL;
         break;
+      case JT_RESTORE:
+      case JT_ADMIN:
+        jcr->JobLevel = L_NONE;
+        break;
       default:
         break;
       }
    }
 }
-
-/*
- * Edit codes into Run command
- *  %% = %
- *  %c = Client's name
- *  %d = Director's name
- *  %i = JobId
- *  %e = Job Exit
- *  %j = Job
- *  %l = Job Level
- *  %n = Job name
- *  %t = Job type
- *
- *  omsg = edited output message
- *  imsg = input string containing edit codes (%x)
- *
- */
-static char *edit_run_codes(JCR *jcr, char *omsg, char *imsg) 
-{
-   char *p;
-   const char *str;
-   char add[20];
-
-   *omsg = 0;
-   Dmsg1(200, "edit_run_codes: %s\n", imsg);
-   for (p=imsg; *p; p++) {
-      if (*p == '%') {
-        switch (*++p) {
-         case '%':
-            str = "%";
-           break;
-         case 'c':
-           str = jcr->client_name;
-           if (!str) {
-               str = "";
-           }
-           break;
-         case 'd':
-           str = my_name;
-           break;
-         case 'e':
-           str = job_status_to_str(jcr->JobStatus);
-           break;
-         case 'i':
-            sprintf(add, "%d", jcr->JobId);
-           str = add;
-           break;
-         case 'j':                    /* Job */
-           str = jcr->Job;
-           break;
-         case 'l':
-           str = job_level_to_str(jcr->JobLevel);
-           break;
-         case 'n':
-           str = jcr->job->hdr.name;
-           break;
-         case 't':
-           str = job_type_to_str(jcr->JobType);
-           break;
-        default:
-            add[0] = '%';
-           add[1] = *p;
-           add[2] = 0;
-           str = add;
-           break;
-        }
-      } else {
-        add[0] = *p;
-        add[1] = 0;
-        str = add;
-      }
-      Dmsg1(200, "add_str %s\n", str);
-      pm_strcat(&omsg, (char *)str);
-      Dmsg1(200, "omsg=%s\n", omsg);
-   }
-   return omsg;
-}