]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/stored/job.c
- Convert more atoi to str_to_int64() for DB.
[bacula/bacula] / bacula / src / stored / job.c
index fc8c882b2b2b73d6dce798dddfcf6b1f834f111b..4dc568b35e170ea1750cde4b6c66b92a2da8b74d 100644 (file)
@@ -1,11 +1,13 @@
 /*
  *   Job control and execution for Storage Daemon
  *
+ *   Kern Sibbald, MM
+ *
  *   Version $Id$
  *
  */
 /*
-   Copyright (C) 2000-2003 Kern Sibbald and John Walker
+   Copyright (C) 2000-2005 Kern Sibbald
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
@@ -34,22 +36,28 @@ extern uint32_t VolSessionTime;
 extern uint32_t newVolSessionId();
 
 /* Forward referenced functions */
-static int use_device_cmd(JCR *jcr);
+static bool use_storage_cmd(JCR *jcr);
 
 /* Requests from the Director daemon */
-static char jobcmd[]     = "JobId=%d job=%127s job_name=%127s client_name=%127s "
-         "type=%d level=%d FileSet=%127s NoAttr=%d SpoolAttr=%d FileSetMD5=%127s "
-         "SpoolData=%d";
-static char use_device[] = "use device=%s media_type=%s pool_name=%s pool_type=%s\n";
+static char jobcmd[] = "JobId=%d job=%127s job_name=%127s client_name=%127s "
+      "type=%d level=%d FileSet=%127s NoAttr=%d SpoolAttr=%d FileSetMD5=%127s "
+      "SpoolData=%d WritePartAfterJob=%d NewVol=%d\n";
+static char use_storage[]  = "use storage=%127s media_type=%127s "
+   "pool_name=%127s pool_type=%127s append=%d\n";
+static char use_device[]  = "use device=%127s\n";
+//static char query_device[] = "query device=%127s";
+
 
 /* Responses sent to Director daemon */
 static char OKjob[]     = "3000 OK Job SDid=%u SDtime=%u Authorization=%s\n";
-static char OK_device[] = "3000 OK use device\n";
-static char NO_device[] = "3914 Device \"%s\" not in SD Device resources.\n";
+static char OK_device[] = "3000 OK use device device=%s\n";
+static char NO_device[] = "3924 Device \"%s\" not in SD Device resources.\n";
+static char NOT_open[]  = "3925 Device \"%s\" could not be opened or does not exist.\n";
 static char BAD_use[]   = "3913 Bad use command: %s\n";
 static char BAD_job[]   = "3915 Bad Job command: %s\n";
-
-
+//static char OK_query[]  = "3001 OK query\n";
+//static char NO_query[]  = "3918 Query failed\n";
+//static char BAD_query[] = "3917 Bad query command: %s\n";
 
 /*
  * Director requests us to start a job
@@ -61,50 +69,41 @@ static char BAD_job[]   = "3915 Bad Job command: %s\n";
  *  - Return when the connection is terminated or
  *    there is an error.
  */
-int job_cmd(JCR *jcr)
+bool job_cmd(JCR *jcr)
 {
-   int JobId, errstat;
+   int JobId;
    char auth_key[100];
    BSOCK *dir = jcr->dir_bsock;
-   POOLMEM *job_name, *client_name, *job, *fileset_name, *fileset_md5;
+   POOL_MEM job_name, client_name, job, fileset_name, fileset_md5;
    int JobType, level, spool_attributes, no_attributes, spool_data;
-   struct timeval tv;
-   struct timezone tz;
-   struct timespec timeout;
+   int write_part_after_job, NewVol;
+
    JCR *ojcr;
 
    /*
     * Get JobId and permissions from Director
     */
-
-   Dmsg1(100, "Job_cmd: %s\n", dir->msg);
-   job = get_memory(dir->msglen);
-   job_name = get_memory(dir->msglen);
-   client_name = get_memory(dir->msglen);
-   fileset_name = get_memory(dir->msglen);
-   fileset_md5 = get_memory(dir->msglen);
-   if (sscanf(dir->msg, jobcmd, &JobId, job, job_name, client_name,
-             &JobType, &level, fileset_name, &no_attributes,
-             &spool_attributes, fileset_md5, &spool_data) != 11) {
-      pm_strcpy(&jcr->errmsg, dir->msg);
+   Dmsg1(100, "<dird: %s\n", dir->msg);
+   if (sscanf(dir->msg, jobcmd, &JobId, job.c_str(), job_name.c_str(),
+             client_name.c_str(),
+             &JobType, &level, fileset_name.c_str(), &no_attributes,
+             &spool_attributes, fileset_md5.c_str(), &spool_data, 
+             &write_part_after_job, &NewVol) != 13) {
+      pm_strcpy(jcr->errmsg, dir->msg);
       bnet_fsend(dir, BAD_job, jcr->errmsg);
+      Dmsg1(100, ">dird: %s\n", dir->msg);
       Emsg1(M_FATAL, 0, _("Bad Job Command from Director: %s\n"), jcr->errmsg);
-      free_memory(job);
-      free_memory(job_name);
-      free_memory(client_name);
-      free_memory(fileset_name);
-      free_memory(fileset_md5);
       set_jcr_job_status(jcr, JS_ErrorTerminated);
-      return 0;
+      return false;
    }
-   /*        
+   /*
     * Since this job could be rescheduled, we
     *  check to see if we have it already. If so
     *  free the old jcr and use the new one.
     */
-   ojcr = get_jcr_by_full_name(job);
+   ojcr = get_jcr_by_full_name(job.c_str());
    if (ojcr && !ojcr->authenticated) {
-      Dmsg2(100, "Found ojcr=0x%x Job %s\n", (unsigned)(long)ojcr, job);
+      Dmsg2(100, "Found ojcr=0x%x Job %s\n", (unsigned)(long)ojcr, job.c_str());
       free_jcr(ojcr);
    }
    jcr->JobId = JobId;
@@ -113,25 +112,22 @@ int job_cmd(JCR *jcr)
    bstrncpy(jcr->Job, job, sizeof(jcr->Job));
    unbash_spaces(job_name);
    jcr->job_name = get_pool_memory(PM_NAME);
-   pm_strcpy(&jcr->job_name, job_name);
+   pm_strcpy(jcr->job_name, job_name);
    unbash_spaces(client_name);
    jcr->client_name = get_pool_memory(PM_NAME);
-   pm_strcpy(&jcr->client_name, client_name);
+   pm_strcpy(jcr->client_name, client_name);
    unbash_spaces(fileset_name);
    jcr->fileset_name = get_pool_memory(PM_NAME);
-   pm_strcpy(&jcr->fileset_name, fileset_name);
+   pm_strcpy(jcr->fileset_name, fileset_name);
    jcr->JobType = JobType;
    jcr->JobLevel = level;
    jcr->no_attributes = no_attributes;
    jcr->spool_attributes = spool_attributes;
    jcr->spool_data = spool_data;
+   jcr->write_part_after_job = write_part_after_job;
    jcr->fileset_md5 = get_pool_memory(PM_NAME);
-   pm_strcpy(&jcr->fileset_md5, fileset_md5);
-   free_memory(job);
-   free_memory(job_name);
-   free_memory(client_name);
-   free_memory(fileset_name);
-   free_memory(fileset_md5);
+   pm_strcpy(jcr->fileset_md5, fileset_md5);
+   jcr->NewVolEachJob = NewVol;
 
    jcr->authenticated = false;
 
@@ -140,24 +136,48 @@ int job_cmd(JCR *jcr)
     */
    make_session_key(auth_key, NULL, 1);
    bnet_fsend(dir, OKjob, jcr->VolSessionId, jcr->VolSessionTime, auth_key);
-   Dmsg1(110, ">dird: %s", dir->msg);
+   Dmsg1(100, ">dird: %s", dir->msg);
    jcr->sd_auth_key = bstrdup(auth_key);
-   memset(auth_key, 0, sizeof(auth_key));    
+   memset(auth_key, 0, sizeof(auth_key));
+   return true;
+}
 
+bool use_cmd(JCR *jcr) 
+{
    /*
     * Wait for the device, media, and pool information
     */
-   if (!use_device_cmd(jcr)) {
+   if (!use_storage_cmd(jcr)) {
       set_jcr_job_status(jcr, JS_ErrorTerminated);
       memset(jcr->sd_auth_key, 0, strlen(jcr->sd_auth_key));
-      return 0;
+      return false;
+   }
+   return true;
+}
+
+bool run_cmd(JCR *jcr)
+{
+   struct timeval tv;
+   struct timezone tz;
+   struct timespec timeout;
+   int errstat;
+
+   Dmsg1(100, "Run_cmd: %s\n", jcr->dir_bsock->msg);
+   /* The following jobs don't need the FD */
+   switch (jcr->JobType) {
+   case JT_MIGRATION:
+   case JT_COPY:
+   case JT_ARCHIVE:
+      jcr->authenticated = true;
+      run_job(jcr);
+      return false;
    }
 
    set_jcr_job_status(jcr, JS_WaitFD);         /* wait for FD to connect */
    dir_send_job_status(jcr);
 
    gettimeofday(&tv, &tz);
-   timeout.tv_nsec = tv.tv_usec * 1000; 
+   timeout.tv_nsec = tv.tv_usec * 1000;
    timeout.tv_sec = tv.tv_sec + 30 * 60;       /* wait 30 minutes */
 
    Dmsg1(100, "%s waiting on FD to contact SD\n", jcr->Job);
@@ -181,13 +201,13 @@ int job_cmd(JCR *jcr)
       Dmsg1(100, "Running job %s\n", jcr->Job);
       run_job(jcr);                  /* Run the job */
    }
-   return 0;
+   return false;
 }
 
 /*
  * After receiving a connection (in job.c) if it is
  *   from the File daemon, this routine is called.
- */  
+ */
 void handle_filed_connection(BSOCK *fd, char *job_name)
 {
    JCR *jcr;
@@ -205,10 +225,12 @@ void handle_filed_connection(BSOCK *fd, char *job_name)
    Dmsg1(110, "Found Job %s\n", job_name);
 
    if (jcr->authenticated) {
-      Pmsg2(000, "Hey!!!! JobId %u Job %s already authenticated.\n", 
+      Jmsg2(jcr, M_FATAL, 0, "Hey!!!! JobId %u Job %s already authenticated.\n",
         jcr->JobId, jcr->Job);
+      free_jcr(jcr);
+      return;
    }
-  
+
    /*
     * Authenticate the File daemon
     */
@@ -231,108 +253,277 @@ void handle_filed_connection(BSOCK *fd, char *job_name)
 }
 
 
-/*  
+/*
  *   Use Device command from Director
- *   He tells is what Device Name to use, the Media Type, 
+ *   He tells is what Device Name to use, the Media Type,
  *     the Pool Name, and the Pool Type.
  *
  *    Ensure that the device exists and is opened, then store
  *     the media and pool info in the JCR.
  */
-static int use_device_cmd(JCR *jcr)
+class DIRSTORE {
+public:
+   alist *device;
+   char name[MAX_NAME_LENGTH];
+   char media_type[MAX_NAME_LENGTH];
+   char pool_name[MAX_NAME_LENGTH];
+   char pool_type[MAX_NAME_LENGTH];
+};
+
+class DIRDEVICE {
+   alist *device;
+};
+   
+static bool use_storage_cmd(JCR *jcr)
 {
-   POOLMEM *dev_name, *media_type, *pool_name, *pool_type;
+   POOL_MEM store_name, dev_name, media_type, pool_name, pool_type;
    BSOCK *dir = jcr->dir_bsock;
    DEVRES *device;
+   AUTOCHANGER *changer;
+   int append;
+   bool ok;
 
-   if (bnet_recv(dir) <= 0) {
-      Jmsg0(jcr, M_FATAL, 0, _("No Device from Director\n"));
-      return 0;
-   }
-   
-   Dmsg1(120, "Use device: %s", dir->msg);
-   dev_name = get_memory(dir->msglen);
-   media_type = get_memory(dir->msglen);
-   pool_name = get_memory(dir->msglen);
-   pool_type = get_memory(dir->msglen);
-   if (sscanf(dir->msg, use_device, dev_name, media_type, pool_name, pool_type) == 4) {
-      unbash_spaces(dev_name);
+   /*
+    * If there are multiple devices, the director sends us
+    *  use_device for each device that it wants to use.
+    */
+   Dmsg1(100, "<dird: %s", dir->msg);
+   ok = sscanf(dir->msg, use_storage, store_name.c_str(), 
+               media_type.c_str(), pool_name.c_str(), 
+              pool_type.c_str(), &append) == 5;
+   if (ok) {
+      unbash_spaces(store_name);
       unbash_spaces(media_type);
       unbash_spaces(pool_name);
       unbash_spaces(pool_type);
-      device = NULL;
+      if (bnet_recv(dir) <= 0) {
+        return false;   
+      }
+      ok = sscanf(dir->msg, use_device, dev_name.c_str()) == 1;
+      if (!ok) {
+        return false;
+      }
+      /* Eat to BNET_EOD */
+      while (bnet_recv(dir) > 0) {
+      }
       LockRes();
-      while ((device=(DEVRES *)GetNextRes(R_DEVICE, (RES *)device))) {
+      foreach_res(device, R_DEVICE) {
         /* Find resource, and make sure we were able to open it */
-        if (strcmp(device->hdr.name, dev_name) == 0 && device->dev) {
-            Dmsg1(120, "Found device %s\n", device->hdr.name);
-           jcr->pool_name = get_memory(strlen(pool_name) + 1);
-           strcpy(jcr->pool_name, pool_name);
-           jcr->pool_type = get_memory(strlen(pool_type) + 1);
-           strcpy(jcr->pool_type, pool_type);
-           jcr->media_type = get_memory(strlen(media_type) + 1);
-           strcpy(jcr->media_type, media_type);
-           jcr->dev_name = get_memory(strlen(dev_name) + 1);
-           strcpy(jcr->dev_name, dev_name);
-           jcr->device = device;
-           Dmsg4(120, use_device, dev_name, media_type, pool_name, pool_type);
-           free_memory(dev_name);
-           free_memory(media_type);
-           free_memory(pool_name);
-           free_memory(pool_type);
+        if (fnmatch(dev_name.c_str(), device->hdr.name, 0) == 0 &&
+            strcmp(device->media_type, media_type.c_str()) == 0) {
+           const int name_len = MAX_NAME_LENGTH;
+           DCR *dcr;
            UnlockRes();
-           return bnet_fsend(dir, OK_device);
+           if (!device->dev) {
+              device->dev = init_dev(jcr, NULL, device);
+           }
+           if (!device->dev) {
+               Jmsg(jcr, M_WARNING, 0, _("\n"
+                  "     Device \"%s\" requested by DIR could not be opened or does not exist.\n"),
+                   dev_name.c_str());
+              bnet_fsend(dir, NOT_open, dev_name.c_str());
+               Dmsg1(100, ">dird: %s\n", dir->msg);
+              return false;
+           }  
+           dcr = new_dcr(jcr, device->dev);
+           if (!dcr) {
+               bnet_fsend(dir, _("3926 Could not get dcr for device: %s\n"), dev_name.c_str());
+               Dmsg1(100, ">dird: %s\n", dir->msg);
+              return false;
+           }
+            Dmsg1(100, "Found device %s\n", device->hdr.name);
+           bstrncpy(dcr->pool_name, pool_name, name_len);
+           bstrncpy(dcr->pool_type, pool_type, name_len);
+           bstrncpy(dcr->media_type, media_type, name_len);
+           bstrncpy(dcr->dev_name, dev_name, name_len);
+           jcr->dcr = dcr;
+           if (append == SD_APPEND) {
+              ok = reserve_device_for_append(dcr);
+           } else {
+              ok = reserve_device_for_read(dcr);
+           }
+           if (!ok) {
+               bnet_fsend(dir, _("3927 Could not reserve device: %s\n"), dev_name.c_str());
+               Dmsg1(100, ">dird: %s\n", dir->msg);
+              free_dcr(jcr->dcr);
+              return false;
+           }
+            Dmsg1(220, "Got: %s", dir->msg);
+           bash_spaces(dev_name);
+           ok = bnet_fsend(dir, OK_device, dev_name.c_str());
+            Dmsg1(100, ">dird: %s\n", dir->msg);
+           return ok;
         }
       }
+
+      foreach_res(changer, R_AUTOCHANGER) {
+        /* Find resource, and make sure we were able to open it */
+        if (fnmatch(dev_name.c_str(), changer->hdr.name, 0) == 0) {
+           const int name_len = MAX_NAME_LENGTH;
+           DCR *dcr;
+           /* Try each device in this AutoChanger */
+           foreach_alist(device, changer->device) {
+               Dmsg1(100, "Try changer device %s\n", device->hdr.name);
+              if (!device->dev) {
+                 device->dev = init_dev(jcr, NULL, device);
+              }
+              if (!device->dev) {
+                  Dmsg1(100, "Device %s could not be opened. Skipped\n", dev_name.c_str());
+                  Jmsg(jcr, M_WARNING, 0, _("\n"
+                     "     Device \"%s\" in changer \"%s\" requested by DIR could not be opened or does not exist.\n"),
+                      device->hdr.name, dev_name.c_str());
+                 continue;
+              }
+              if (!device->dev->autoselect) {
+                 continue;           /* device is not available */
+              }
+              dcr = new_dcr(jcr, device->dev);
+              if (!dcr) {
+                  bnet_fsend(dir, _("3926 Could not get dcr for device: %s\n"), dev_name.c_str());
+                  Dmsg1(100, ">dird: %s\n", dir->msg);
+                 UnlockRes();
+                 return false;
+              }
+               Dmsg1(100, "Found changer device %s\n", device->hdr.name);
+              bstrncpy(dcr->pool_name, pool_name, name_len);
+              bstrncpy(dcr->pool_type, pool_type, name_len);
+              bstrncpy(dcr->media_type, media_type, name_len);
+              bstrncpy(dcr->dev_name, dev_name, name_len);
+              jcr->dcr = dcr;
+              if (append == SD_APPEND) {
+                 ok = reserve_device_for_append(dcr);
+              } else {
+                 ok = reserve_device_for_read(dcr);
+              }
+              if (!ok) {
+                  Jmsg(jcr, M_WARNING, 0, _("Could not reserve device: %s\n"), dev_name.c_str());
+                 free_dcr(jcr->dcr);
+                 continue;
+              }
+               Dmsg1(100, "Device %s opened.\n", dev_name.c_str());
+              UnlockRes();
+              pm_strcpy(dev_name, device->hdr.name);
+              bash_spaces(dev_name);
+              ok = bnet_fsend(dir, OK_device, dev_name.c_str());
+               Dmsg1(100, ">dird: %s\n", dir->msg);
+              return ok;
+           }
+           break;                    /* we found it but could not open a device */
+        }
+      }
+
       UnlockRes();
       if (verbose) {
         unbash_spaces(dir->msg);
-        pm_strcpy(&jcr->errmsg, dir->msg);
+        pm_strcpy(jcr->errmsg, dir->msg);
          Jmsg(jcr, M_INFO, 0, _("Failed command: %s\n"), jcr->errmsg);
       }
       Jmsg(jcr, M_FATAL, 0, _("\n"
-         "     Device \"%s\" requested by Dir not found in SD Device resources.\n"),
-          dev_name);
-      bnet_fsend(dir, NO_device, dev_name);
+         "     Device \"%s\" with MediaType \"%s\" requested by DIR not found in SD Device resources.\n"),
+          dev_name.c_str(), media_type.c_str());
+      bnet_fsend(dir, NO_device, dev_name.c_str());
+      Dmsg1(100, ">dird: %s\n", dir->msg);
    } else {
       unbash_spaces(dir->msg);
-      pm_strcpy(&jcr->errmsg, dir->msg);
+      pm_strcpy(jcr->errmsg, dir->msg);
       if (verbose) {
          Jmsg(jcr, M_INFO, 0, _("Failed command: %s\n"), jcr->errmsg);
       }
       Jmsg(jcr, M_FATAL, 0, _("Bad Use Device command: %s\n"), jcr->errmsg);
       bnet_fsend(dir, BAD_use, jcr->errmsg);
+      Dmsg1(100, ">dird: %s\n", dir->msg);
    }
 
-   free_memory(dev_name);
-   free_memory(media_type);
-   free_memory(pool_name);
-   free_memory(pool_type);
-   return 0;                         /* ERROR return */
+   return false;                     /* ERROR return */
 }
 
-/* 
+#ifdef needed
+/*
+ *   Query Device command from Director
+ *   Sends Storage Daemon's information on the device to the
+ *    caller (presumably the Director).
+ *   This command always returns "true" so that the line is
+ *    not closed on an error.
+ *
+ */
+bool query_cmd(JCR *jcr)
+{
+   POOL_MEM dev_name, VolumeName, MediaType, ChangerName;
+   BSOCK *dir = jcr->dir_bsock;
+   DEVRES *device;
+   AUTOCHANGER *changer;
+   bool ok;
+
+   Dmsg1(100, "Query_cmd: %s", dir->msg);
+   ok = sscanf(dir->msg, query_device, dev_name.c_str()) == 1;
+   Dmsg1(100, "<dird: %s\n", dir->msg);
+   if (ok) {
+      unbash_spaces(dev_name);
+      LockRes();
+      foreach_res(device, R_DEVICE) {
+        /* Find resource, and make sure we were able to open it */
+        if (fnmatch(dev_name.c_str(), device->hdr.name, 0) == 0) {
+           if (!device->dev) {
+              device->dev = init_dev(jcr, NULL, device);
+           }
+           if (!device->dev) {
+              break;
+           }  
+           UnlockRes();
+           ok = dir_update_device(jcr, device->dev);
+           if (ok) {
+              ok = bnet_fsend(dir, OK_query);
+           } else {
+              bnet_fsend(dir, NO_query);
+           }
+           return ok;
+        }
+      }
+      foreach_res(changer, R_AUTOCHANGER) {
+        /* Find resource, and make sure we were able to open it */
+        if (fnmatch(dev_name.c_str(), changer->hdr.name, 0) == 0) {
+           UnlockRes();
+           if (!changer->device || changer->device->size() == 0) {
+              continue;              /* no devices */
+           }
+           ok = dir_update_changer(jcr, changer);
+           if (ok) {
+              ok = bnet_fsend(dir, OK_query);
+           } else {
+              bnet_fsend(dir, NO_query);
+           }
+           return ok;
+        }
+      }
+      /* If we get here, the device/autochanger was not found */
+      UnlockRes();
+      unbash_spaces(dir->msg);
+      pm_strcpy(jcr->errmsg, dir->msg);
+      bnet_fsend(dir, NO_device, dev_name.c_str());
+      Dmsg1(100, ">dird: %s\n", dir->msg);
+   } else {
+      unbash_spaces(dir->msg);
+      pm_strcpy(jcr->errmsg, dir->msg);
+      bnet_fsend(dir, BAD_query, jcr->errmsg);
+      Dmsg1(100, ">dird: %s\n", dir->msg);
+   }
+
+   return true;
+}
+
+#endif
+
+
+/*
  * Destroy the Job Control Record and associated
  * resources (sockets).
  */
-void stored_free_jcr(JCR *jcr) 
+void stored_free_jcr(JCR *jcr)
 {
    if (jcr->file_bsock) {
       bnet_close(jcr->file_bsock);
       jcr->file_bsock = NULL;
    }
-   if (jcr->pool_name) {
-      free_memory(jcr->pool_name);
-   }
-   if (jcr->pool_type) {
-      free_memory(jcr->pool_type);
-   }
-   if (jcr->media_type) {
-      free_memory(jcr->media_type);
-   }
-   if (jcr->dev_name) {
-      free_memory(jcr->dev_name);
-   }
    if (jcr->job_name) {
       free_pool_memory(jcr->job_name);
    }
@@ -355,10 +546,14 @@ void stored_free_jcr(JCR *jcr)
       free_pool_memory(jcr->RestoreBootstrap);
       jcr->RestoreBootstrap = NULL;
    }
-   if (jcr->next_dev && jcr->prev_dev) {
+   if (jcr->next_dev || jcr->prev_dev) {
       Emsg0(M_FATAL, 0, _("In free_jcr(), but still attached to device!!!!\n"));
    }
    pthread_cond_destroy(&jcr->job_start_wait);
+   if (jcr->dcrs) {
+      delete jcr->dcrs;
+   }
+   jcr->dcrs = NULL;
    if (jcr->dcr) {
       free_dcr(jcr->dcr);
       jcr->dcr = NULL;