]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/dird/newvol.c
regress: add new many-reload-test to standard scripts
[bacula/bacula] / bacula / src / dird / newvol.c
index 92c2b1f691bbaa1cf313ab8633dce77632e1e28d..8b74dbcbd2baee428b19f691bf06a8c33f87e64f 100644 (file)
@@ -1,3 +1,21 @@
+/*
+   Bacula(R) - The Network Backup Solution
+
+   Copyright (C) 2000-2015 Kern Sibbald
+
+   The original author of Bacula is Kern Sibbald, with contributions
+   from many others, a complete list can be found in the file AUTHORS.
+
+   You may use this file and others of this release according to the
+   license defined in the LICENSE file, which includes the Affero General
+   Public License, v3.0 ("AGPLv3") and some additional permissions and
+   terms pursuant to its AGPLv3 Section 7.
+
+   This notice must be preserved when any source code is 
+   conveyed and/or propagated.
+
+   Bacula(R) is a registered trademark of Kern Sibbald.
+*/
 /*
  *
  *   Bacula Director -- newvol.c -- creates new Volumes in
  *    This routine runs as a thread and must be thread reentrant.
  *
  *  Basic tasks done here:
- *     If possible create a new Media entry
+ *      If possible create a new Media entry
  *
- *   Version $Id$
- */
-/*
-   Copyright (C) 2000, 2001, 2002 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
-   published by the Free Software Foundation; either version 2 of
-   the License, or (at your option) any later version.
-
-   This program is distributed in the hope that it will be useful,
-   but WITHOUT ANY WARRANTY; without even the implied warranty of
-   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
-   General Public License for more details.
-
-   You should have received a copy of the GNU General Public
-   License along with this program; if not, write to the Free
-   Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
-   MA 02111-1307, USA.
-
  */
 
 #include "bacula.h"
 #include "dird.h"
 
+/* Forward referenced functions */
+static bool create_simple_name(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr);
+static bool perform_full_name_substitution(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr);
+
+
 /*
- * Really crude automatic Volume name creation using
- *  LabelFormat
+ * Automatic Volume name creation using the LabelFormat
+ *
+ *  The media record must have the PoolId filled in when
+ *   calling this routine.
  */
-int newVolume(JCR *jcr)
+bool newVolume(JCR *jcr, MEDIA_DBR *mr, STORE *store)
 {
-   MEDIA_DBR mr; 
    POOL_DBR pr;
-   char name[MAXSTRING];
 
-   memset(&mr, 0, sizeof(mr));
    memset(&pr, 0, sizeof(pr));
 
    /* See if we can create a new Volume */
-   pr.PoolId = jcr->PoolId;
-   if (db_get_pool_record(jcr->db, &pr) && pr.LabelFormat[0] &&
-       pr.LabelFormat[0] != '*') {
-      if (pr.MaxVols == 0 || pr.NumVols < pr.MaxVols) {
-        memset(&mr, 0, sizeof(mr));
-        mr.PoolId = jcr->PoolId;
-        strcpy(mr.MediaType, jcr->store->media_type);
-        strcpy(name, pr.LabelFormat);   
-         strcat(name, "%04d");
-        sprintf(mr.VolumeName, name, ++pr.NumVols);
-         strcpy(mr.VolStatus, "Append");
-        mr.Recycle = pr.Recycle;
-        mr.VolRetention = pr.VolRetention;
-        if (db_create_media_record(jcr->db, &mr) &&
-           db_update_pool_record(jcr->db, &pr) == 1) {
-            Dmsg1(90, "Created new Volume=%s\n", mr.VolumeName);
-           return 1;
-        } else {
-            Jmsg(jcr, M_ERROR, 0, "%s", db_strerror(jcr->db));
-        }
+   db_lock(jcr->db);
+   pr.PoolId = mr->PoolId;
+   if (!db_get_pool_numvols(jcr, jcr->db, &pr)) {
+      goto bail_out;
+   }
+   if (pr.MaxVols == 0 || pr.NumVols < pr.MaxVols) {
+      mr->clear();
+      set_pool_dbr_defaults_in_media_dbr(mr, &pr);
+      jcr->VolumeName[0] = 0;
+      bstrncpy(mr->MediaType, jcr->wstore->media_type, sizeof(mr->MediaType));
+      generate_plugin_event(jcr, bDirEventNewVolume); /* return void... */
+      if (jcr->VolumeName[0] && is_volume_name_legal(NULL, jcr->VolumeName)) {
+         bstrncpy(mr->VolumeName, jcr->VolumeName, sizeof(mr->VolumeName));
+      /* Check for special characters */
+      } else if (pr.LabelFormat[0] && pr.LabelFormat[0] != '*') {
+         if (is_volume_name_legal(NULL, pr.LabelFormat)) {
+            /* No special characters, so apply simple algorithm */
+            if (!create_simple_name(jcr, mr, &pr)) {
+               goto bail_out;
+            }
+         } else {  /* try full substitution */
+            /* Found special characters, so try substitution */
+            if (!perform_full_name_substitution(jcr, mr, &pr)) {
+               goto bail_out;
+            }
+            if (!is_volume_name_legal(NULL, mr->VolumeName)) {
+               Jmsg(jcr, M_ERROR, 0, _("Illegal character in Volume name \"%s\"\n"),
+                  mr->VolumeName);
+               goto bail_out;
+            }
+         }
+      } else {
+         goto bail_out;
       }
+      pr.NumVols++;
+      mr->Enabled = 1;
+      set_storageid_in_mr(store, mr);
+      if (db_create_media_record(jcr, jcr->db, mr) &&
+         db_update_pool_record(jcr, jcr->db, &pr)) {
+         Jmsg(jcr, M_INFO, 0, _("Created new Volume=\"%s\", Pool=\"%s\", MediaType=\"%s\" in catalog.\n"),
+            mr->VolumeName, pr.Name, mr->MediaType);
+         Dmsg1(90, "Created new Volume=%s\n", mr->VolumeName);
+         db_unlock(jcr->db);
+         return true;
+      } else {
+         Jmsg(jcr, M_ERROR, 0, "%s", db_strerror(jcr->db));
+      }
+   }
+bail_out:
+   db_unlock(jcr->db);
+   return false;
+}
+
+static bool create_simple_name(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr)
+{
+   char name[MAXSTRING];
+   char num[20];
+   db_int64_ctx ctx;
+   POOL_MEM query(PM_MESSAGE);
+   char ed1[50];
+
+   /* See if volume already exists */
+   mr->VolumeName[0] = 0;
+   bstrncpy(name, pr->LabelFormat, sizeof(name));
+   ctx.value = 0;
+   /* TODO: Remove Pool as it is not used in the query */
+   Mmsg(query, "SELECT MAX(MediaId) FROM Media,Pool WHERE Pool.PoolId=%s",
+        edit_int64(pr->PoolId, ed1));
+   if (!db_sql_query(jcr->db, query.c_str(), db_int64_handler, (void *)&ctx)) {
+      Jmsg(jcr, M_WARNING, 0, _("SQL failed, but ignored. ERR=%s\n"), db_strerror(jcr->db));
+      ctx.value = pr->NumVols+1;
+   }
+   for (int i=(int)ctx.value+1; i<(int)ctx.value+100; i++) {
+      MEDIA_DBR tmr;
+      sprintf(num, "%04d", i);
+      bstrncpy(tmr.VolumeName, name, sizeof(tmr.VolumeName));
+      bstrncat(tmr.VolumeName, num, sizeof(tmr.VolumeName));
+      if (db_get_media_record(jcr, jcr->db, &tmr)) {
+         Jmsg(jcr, M_WARNING, 0,
+             _("Wanted to create Volume \"%s\", but it already exists. Trying again.\n"),
+             tmr.VolumeName);
+         continue;
+      }
+      bstrncpy(mr->VolumeName, name, sizeof(mr->VolumeName));
+      bstrncat(mr->VolumeName, num, sizeof(mr->VolumeName));
+      break;                    /* Got good name */
+   }
+   if (mr->VolumeName[0] == 0) {
+      Jmsg(jcr, M_ERROR, 0, _("Too many failures. Giving up creating Volume name.\n"));
+      return false;
+   }
+   return true;
+}
+
+/*
+ * Perform full substitution on Label
+ */
+static bool perform_full_name_substitution(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr)
+{
+   bool ok = false;
+   POOLMEM *label = get_pool_memory(PM_FNAME);
+   jcr->NumVols = pr->NumVols;
+   if (variable_expansion(jcr, pr->LabelFormat, &label)) {
+      bstrncpy(mr->VolumeName, label, sizeof(mr->VolumeName));
+      ok = true;
    }
-   return 0;   
+   free_pool_memory(label);
+   return ok;
 }