]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/stored/acquire.c
Implement first cut DCR in SD
[bacula/bacula] / bacula / src / stored / acquire.c
index 85e1aef4aa611e6570ae6113ba0991696935f8a8..c44fd8e3bf802768d31263ccd318cb1365d4cef1 100644 (file)
 
 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
 
+DCR *new_dcr(JCR *jcr, DEVICE *dev)
+{
+   DCR *dcr = (DCR *)malloc(sizeof(DCR));
+   memset(dcr, 0, sizeof(DCR));
+   jcr->dcr = dcr;
+   dcr->jcr = jcr;
+   dcr->dev = dev;
+   dcr->block = new_block(dev);
+   dcr->record = new_record();
+   dcr->spool_fd = -1;
+   return dcr;
+}
+
+void free_dcr(DCR *dcr)
+{
+   if (dcr->block) {
+      free_block(dcr->block);
+   }
+   if (dcr->record) {
+      free_record(dcr->record);
+   }
+   dcr->jcr->dcr = NULL;
+   free(dcr);
+}
+
+
 /********************************************************************* 
  * Acquire device for reading. We permit (for the moment)
  *  only one reader.  We read the Volume label from the block and
@@ -38,29 +64,41 @@ static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
  *  Returns: 0 if failed for any reason
  *          1 if successful
  */
-int acquire_device_for_read(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
+DCR *acquire_device_for_read(JCR *jcr)
 {
-   int stat = 0;
-   int tape_previously_mounted;
+   bool vol_ok = false;
+   bool tape_previously_mounted;
+   bool tape_initially_mounted;
    VOL_LIST *vol;
    int autochanger = 0;
    int i;
-
+   DCR *dcr = jcr->dcr;
+   DEVICE *dev;
+   
+   /* Called for each volume */
+   if (!dcr) {
+      dcr = new_dcr(jcr, jcr->device->dev);
+   }
+   dev = dcr->dev;
    if (device_is_unmounted(dev)) {
-      Jmsg(jcr, M_WARNING, 0, _("device is BLOCKED due to user unmount.\n"));
+      Jmsg(jcr, M_WARNING, 0, _("device %s is BLOCKED due to user unmount.\n"),
+        dev_name(dev));
    }
    lock_device(dev);
    block_device(dev, BST_DOING_ACQUIRE);
    unlock_device(dev);
 
-   tape_previously_mounted = dev_state(dev, ST_READ) || dev_state(dev, ST_APPEND);
-
    if (dev_state(dev, ST_READ) || dev->num_writers > 0) {
       Jmsg2(jcr, M_FATAL, 0, _("Device %s is busy. Job %d canceled.\n"), 
            dev_name(dev), jcr->JobId);
       goto get_out;
    }
 
+   tape_previously_mounted = dev_state(dev, ST_READ) || 
+                            dev_state(dev, ST_APPEND) ||
+                            dev_state(dev, ST_LABEL);
+   tape_initially_mounted = tape_previously_mounted;
+
    /* Find next Volume, if any */
    vol = jcr->VolList;
    if (!vol) {
@@ -72,6 +110,7 @@ int acquire_device_for_read(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
       vol = vol->next;
    }
    pm_strcpy(&jcr->VolumeName, vol->VolumeName);
+   bstrncpy(dcr->VolumeName, vol->VolumeName, sizeof(dcr->VolumeName));
 
    for (i=0; i<5; i++) {
       if (job_canceled(jcr)) {
@@ -85,18 +124,22 @@ int acquire_device_for_read(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
        */
       for ( ; !(dev->state & ST_OPENED); ) {
          Dmsg1(120, "bstored: open vol=%s\n", jcr->VolumeName);
-        if (open_dev(dev, jcr->VolumeName, READ_ONLY) < 0) {
+        if (open_dev(dev, dcr->VolumeName, READ_ONLY) < 0) {
             Jmsg(jcr, M_FATAL, 0, _("Open device %s volume %s failed, ERR=%s\n"), 
-               dev_name(dev), jcr->VolumeName, strerror_dev(dev));
+               dev_name(dev), dcr->VolumeName, strerror_dev(dev));
            goto get_out;
         }
          Dmsg1(129, "open_dev %s OK\n", dev_name(dev));
       }
-      dev->state &= ~ST_LABEL;          /* force reread of label */
+      /****FIXME***** do not reread label if ioctl() says we are
+       *  correctly possitioned.  Possibly have way user can turn
+       *  this optimization (to be implemented) off.
+       */
+      dcr->dev->state &= ~ST_LABEL;          /* force reread of label */
       Dmsg0(200, "calling read-vol-label\n");
-      switch (read_dev_volume_label(jcr, dev, block)) {
+      switch (read_dev_volume_label(jcr, dev, dcr->block)) {
       case VOL_OK:
-        stat = 1;
+        vol_ok = true;
         break;                    /* got it */
       case VOL_IO_ERROR:
         /*
@@ -108,10 +151,16 @@ int acquire_device_for_read(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
             Jmsg(jcr, M_WARNING, 0, "%s", jcr->errmsg);                         
         }
         goto default_path;
+      case VOL_NAME_ERROR:
+        if (tape_initially_mounted) {
+           tape_initially_mounted = false;
+           goto default_path;
+        }
+        /* Fall through */
       default:
          Jmsg(jcr, M_WARNING, 0, "%s", jcr->errmsg);
 default_path:
-        tape_previously_mounted = 1;
+        tape_previously_mounted = true;
          Dmsg0(200, "dir_get_volume_info\n");
         if (!dir_get_volume_info(jcr, GET_VOL_INFO_FOR_READ)) { 
             Jmsg1(jcr, M_WARNING, 0, "%s", jcr->errmsg);
@@ -134,13 +183,14 @@ default_path:
       } /* end switch */
       break;
    } /* end for loop */
-   if (stat == 0) {
+   if (!vol_ok) {
       Jmsg1(jcr, M_FATAL, 0, _("Too many errors trying to mount device \"%s\".\n"),
            dev_name(dev));
       goto get_out;
    }
 
-   dev->state |= ST_READ;
+   dev->state &= ~ST_APPEND;         /* clear any previous append mode */
+   dev->state |= ST_READ;            /* set read mode */
    attach_jcr_to_device(dev, jcr);    /* attach jcr to device */
    Jmsg(jcr, M_INFO, 0, _("Ready to read from volume \"%s\" on device %s.\n"),
       jcr->VolumeName, dev_name(dev));
@@ -149,7 +199,11 @@ get_out:
    P(dev->mutex); 
    unblock_device(dev);
    V(dev->mutex);
-   return stat;
+   if (!vol_ok) {
+      free_dcr(dcr);
+      dcr = NULL;
+   }
+   return dcr;
 }
 
 /*
@@ -162,15 +216,18 @@ get_out:
  *   multiple devices (for files), thus we have our own mutex 
  *   on top of the device mutex.
  */
-DEVICE *acquire_device_for_append(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
+DCR *acquire_device_for_append(JCR *jcr)
 {
    int release = 0;
    bool recycle = false;
    bool do_mount = false;
-   DEVICE *rtn_dev = NULL;
+   DCR *dcr;
+   DEVICE *dev = jcr->device->dev;
 
+   dcr = new_dcr(jcr, dev);
    if (device_is_unmounted(dev)) {
-      Jmsg(jcr, M_WARNING, 0, _("device is BLOCKED due to user unmount.\n"));
+      Jmsg(jcr, M_WARNING, 0, _("device %s is BLOCKED due to user unmount.\n"),
+        dev_name(dev));
    }
    lock_device(dev);
    block_device(dev, BST_DOING_ACQUIRE);
@@ -190,6 +247,7 @@ DEVICE *acquire_device_for_append(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
        *    dir_find_next_appendable_volume
        */
       pm_strcpy(&jcr->VolumeName, dev->VolHdr.VolName);
+      bstrncpy(dcr->VolumeName, dev->VolHdr.VolName, sizeof(dcr->VolumeName));
       if (!dir_get_volume_info(jcr, GET_VOL_INFO_FOR_WRITE) &&
          !(dir_find_next_appendable_volume(jcr) &&
            strcmp(dev->VolHdr.VolName, jcr->VolumeName) == 0)) { /* wrong tape mounted */
@@ -230,7 +288,7 @@ DEVICE *acquire_device_for_append(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
           recycle = strcmp(jcr->VolCatInfo.VolCatStatus, "Recycle") == 0;
          if (recycle && dev->num_writers != 0) {
              Jmsg(jcr, M_FATAL, 0, _("Cannot recycle volume \"%s\""
-                  " because it is in use by another job."));
+                  " because it is in use by another job.\n"));
             goto get_out;
          }
        }
@@ -245,10 +303,10 @@ DEVICE *acquire_device_for_append(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
    }
 
    if (do_mount || recycle) {
-      if (!mount_next_write_volume(jcr, dev, block, release)) {
+      if (!mount_next_write_volume(jcr, dev, dcr->block, release)) {
         if (!job_canceled(jcr)) {
             /* Reduce "noise" -- don't print if job canceled */
-            Jmsg(jcr, M_FATAL, 0, _("Could not ready device %s for append.\n"),
+            Jmsg(jcr, M_FATAL, 0, _("Could not ready device \"%s\" for append.\n"),
               dev_name(dev));
         }
         goto get_out;
@@ -260,18 +318,21 @@ DEVICE *acquire_device_for_append(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
       jcr->NumVolumes = 1;
    }
    attach_jcr_to_device(dev, jcr);    /* attach jcr to device */
-   rtn_dev = dev;                    /* return device */
+   goto ok_out;
 
 /*
  * If we jump here, it is an error return because
  *  rtn_dev will still be NULL
  */
 get_out:
+   free_dcr(dcr);
+   dcr = NULL;
+ok_out:
    P(dev->mutex); 
    unblock_device(dev);
    V(dev->mutex);
    V(mutex);                         /* unlock other threads */
-   return rtn_dev;
+   return dcr;
 }
 
 /*
@@ -279,8 +340,9 @@ get_out:
  *  the device remains open.
  *
  */
-int release_device(JCR *jcr, DEVICE *dev)
+int release_device(JCR *jcr)
 {
+   DEVICE *dev = jcr->dcr->dev;   
    lock_device(dev);
    Dmsg1(100, "release_device device is %s\n", dev_is_tape(dev)?"tape":"disk");
    if (dev_state(dev, ST_READ)) {
@@ -292,46 +354,35 @@ int release_device(JCR *jcr, DEVICE *dev)
       /******FIXME**** send read volume usage statistics to director */
 
    } else if (dev->num_writers > 0) {
-      ASSERT(dev_state(dev, ST_APPEND));
       dev->num_writers--;
       Dmsg1(100, "There are %d writers in release_device\n", dev->num_writers);
-      if (dev->num_writers == 0) {
-        /* If we are the only writer, write EOF after job */
-        if (dev_state(dev, ST_LABEL)) {
-            Dmsg0(100, "dir_create_jobmedia_record. Release\n");
-           if (!dir_create_jobmedia_record(jcr)) {
-               Jmsg(jcr, M_ERROR, 0, _("Could not create JobMedia record for Volume=\"%s\" Job=%s\n"),
-              jcr->VolCatInfo.VolCatName, jcr->Job);
-           }
-           if (dev_can_write(dev)) {
-              weof_dev(dev, 1);
-           }
-           dev->VolCatInfo.VolCatJobs++;              /* increment number of jobs */
-           /* Note! do volume update before close, which zaps VolCatInfo */
-            Dmsg0(100, "dir_update_vol_info. Release0\n");
-           dir_update_volume_info(jcr, dev, 0); /* send Volume info to Director */
-        }
-
-        if (!dev_is_tape(dev) || !dev_cap(dev, CAP_ALWAYSOPEN)) {
-           offline_or_rewind_dev(dev);
-           close_dev(dev);
-        }
-      } else if (dev_state(dev, ST_LABEL)) {
+      if (dev_state(dev, ST_LABEL)) {
          Dmsg0(100, "dir_create_jobmedia_record. Release\n");
         if (!dir_create_jobmedia_record(jcr)) {
             Jmsg(jcr, M_ERROR, 0, _("Could not create JobMedia record for Volume=\"%s\" Job=%s\n"),
-              jcr->VolCatInfo.VolCatName, jcr->Job);
+           jcr->VolCatInfo.VolCatName, jcr->Job);
+        }
+        /* If no more writers, write an EOF */
+        if (!dev->num_writers && dev_can_write(dev)) {
+           weof_dev(dev, 1);
         }
-         Dmsg0(100, "dir_update_vol_info. Release1\n");
+        dev->VolCatInfo.VolCatFiles = dev->file;   /* set number of files */
         dev->VolCatInfo.VolCatJobs++;              /* increment number of jobs */
+        /* Note! do volume update before close, which zaps VolCatInfo */
+         Dmsg0(100, "dir_update_vol_info. Release0\n");
         dir_update_volume_info(jcr, dev, 0); /* send Volume info to Director */
       }
+
+      if (!dev->num_writers && (!dev_is_tape(dev) || !dev_cap(dev, CAP_ALWAYSOPEN))) {
+        offline_or_rewind_dev(dev);
+        close_dev(dev);
+      }
    } else {
       Jmsg2(jcr, M_ERROR, 0, _("BAD ERROR: release_device %s, Volume \"%s\" not in use.\n"), 
            dev_name(dev), NPRT(jcr->VolumeName));
    }
    detach_jcr_from_device(dev, jcr);
-   if (dev->prev && !dev_state(dev, ST_READ) && dev->num_writers == 0) {
+   if (dev->prev && !dev_state(dev, ST_READ) && !dev->num_writers) {
       P(mutex);
       unlock_device(dev);
       dev->prev->next = dev->next;    /* dechain */
@@ -340,5 +391,7 @@ int release_device(JCR *jcr, DEVICE *dev)
    } else {
       unlock_device(dev);
    }
+   free_dcr(jcr->dcr);
+   jcr->dcr = NULL;
    return 1;
 }