git.sur5r.net Git - bacula/bacula/blob - bacula/src/dird/newvol.c

   1 /*
   2    Bacula® - The Network Backup Solution
   3
   4    Copyright (C) 2000-2008 Free Software Foundation Europe e.V.
   5
   6    The main author of Bacula is Kern Sibbald, with contributions from
   7    many others, a complete list can be found in the file AUTHORS.
   8    This program is Free Software; you can redistribute it and/or
   9    modify it under the terms of version three of the GNU Affero General Public
  10    License as published by the Free Software Foundation and included
  11    in the file LICENSE.
  12
  13    This program is distributed in the hope that it will be useful, but
  14    WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16    General Public License for more details.
  17
  18    You should have received a copy of the GNU Affero General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  21    02110-1301, USA.
  22
  23    Bacula® is a registered trademark of Kern Sibbald.
  24    The licensor of Bacula is the Free Software Foundation Europe
  25    (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
  26    Switzerland, email:ftf@fsfeurope.org.
  27 */
  28 /*
  29  *
  30  *   Bacula Director -- newvol.c -- creates new Volumes in
  31  *    catalog Media table from the LabelFormat specification.
  32  *
  33  *     Kern Sibbald, May MMI
  34  *
  35  *    This routine runs as a thread and must be thread reentrant.
  36  *
  37  *  Basic tasks done here:
  38  *      If possible create a new Media entry
  39  *
  40  *   Version $Id$
  41  */
  42
  43 #include "bacula.h"
  44 #include "dird.h"
  45
  46 /* Forward referenced functions */
  47 static bool create_simple_name(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr);
  48 static bool perform_full_name_substitution(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr);
  49
  50
  51 /*
  52  * Automatic Volume name creation using the LabelFormat
  53  *
  54  *  The media record must have the PoolId filled in when
  55  *   calling this routine.
  56  */
  57 bool newVolume(JCR *jcr, MEDIA_DBR *mr)
  58 {
  59    POOL_DBR pr;
  60
  61    memset(&pr, 0, sizeof(pr));
  62
  63    /* See if we can create a new Volume */
  64    db_lock(jcr->db);
  65    pr.PoolId = mr->PoolId;
  66    if (!db_get_pool_record(jcr, jcr->db, &pr)) {
  67       goto bail_out;
  68    }
  69    if (pr.MaxVols == 0 || pr.NumVols < pr.MaxVols) {
  70       memset(mr, 0, sizeof(MEDIA_DBR));
  71       set_pool_dbr_defaults_in_media_dbr(mr, &pr);
  72       jcr->VolumeName[0] = 0;
  73       bstrncpy(mr->MediaType, jcr->wstore->media_type, sizeof(mr->MediaType));
  74       generate_job_event(jcr, "NewVolume"); /* return bool */
  75       generate_plugin_event(jcr, bEventNewVolume); /* return void... */
  76       if (jcr->VolumeName[0] && is_volume_name_legal(NULL, jcr->VolumeName)) {
  77          bstrncpy(mr->VolumeName, jcr->VolumeName, sizeof(mr->VolumeName));
  78       /* Check for special characters */
  79       } else if (pr.LabelFormat[0] && pr.LabelFormat[0] != '*') {
  80          if (is_volume_name_legal(NULL, pr.LabelFormat)) {
  81             /* No special characters, so apply simple algorithm */
  82             if (!create_simple_name(jcr, mr, &pr)) {
  83                goto bail_out;
  84             }
  85          } else {  /* try full substitution */
  86             /* Found special characters, so try substitution */
  87             if (!perform_full_name_substitution(jcr, mr, &pr)) {
  88                goto bail_out;
  89             }
  90             if (!is_volume_name_legal(NULL, mr->VolumeName)) {
  91                Jmsg(jcr, M_ERROR, 0, _("Illegal character in Volume name \"%s\"\n"),
  92                   mr->VolumeName);
  93                goto bail_out;
  94             }
  95          }
  96       } else {
  97          goto bail_out;
  98       }
  99       pr.NumVols++;
 100       mr->Enabled = 1;
 101       if (db_create_media_record(jcr, jcr->db, mr) &&
 102          db_update_pool_record(jcr, jcr->db, &pr)) {
 103          db_unlock(jcr->db);
 104          Jmsg(jcr, M_INFO, 0, _("Created new Volume \"%s\" in catalog.\n"), mr->VolumeName);
 105          Dmsg1(90, "Created new Volume=%s\n", mr->VolumeName);
 106          return true;
 107       } else {
 108          Jmsg(jcr, M_ERROR, 0, "%s", db_strerror(jcr->db));
 109       }
 110    }
 111 bail_out:
 112    db_unlock(jcr->db);
 113    return false;
 114 }
 115
 116 static bool create_simple_name(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr)
 117 {
 118    char name[MAXSTRING];
 119    char num[20];
 120    db_int64_ctx ctx;
 121    POOL_MEM query(PM_MESSAGE);
 122    char ed1[50];
 123
 124    /* See if volume already exists */
 125    mr->VolumeName[0] = 0;
 126    bstrncpy(name, pr->LabelFormat, sizeof(name));
 127    ctx.value = 0;
 128    Mmsg(query, "SELECT MAX(MediaId) FROM Media,Pool WHERE Pool.PoolId=%s",
 129         edit_int64(pr->PoolId, ed1));
 130    if (!db_sql_query(jcr->db, query.c_str(), db_int64_handler, (void *)&ctx)) {
 131       Jmsg(jcr, M_WARNING, 0, _("SQL failed, but ignored. ERR=%s\n"), db_strerror(jcr->db));
 132       ctx.value = pr->NumVols+1;
 133    }
 134    for (int i=(int)ctx.value+1; i<(int)ctx.value+100; i++) {
 135       MEDIA_DBR tmr;
 136       memset(&tmr, 0, sizeof(tmr));
 137       sprintf(num, "%04d", i);
 138       bstrncpy(tmr.VolumeName, name, sizeof(tmr.VolumeName));
 139       bstrncat(tmr.VolumeName, num, sizeof(tmr.VolumeName));
 140       if (db_get_media_record(jcr, jcr->db, &tmr)) {
 141          Jmsg(jcr, M_WARNING, 0,
 142              _("Wanted to create Volume \"%s\", but it already exists. Trying again.\n"),
 143              tmr.VolumeName);
 144          continue;
 145       }
 146       bstrncpy(mr->VolumeName, name, sizeof(mr->VolumeName));
 147       bstrncat(mr->VolumeName, num, sizeof(mr->VolumeName));
 148       break;                    /* Got good name */
 149    }
 150    if (mr->VolumeName[0] == 0) {
 151       Jmsg(jcr, M_ERROR, 0, _("Too many failures. Giving up creating Volume name.\n"));
 152       return false;
 153    }
 154    return true;
 155 }
 156
 157 /*
 158  * Perform full substitution on Label
 159  */
 160 static bool perform_full_name_substitution(JCR *jcr, MEDIA_DBR *mr, POOL_DBR *pr)
 161 {
 162    bool ok = false;
 163    POOLMEM *label = get_pool_memory(PM_FNAME);
 164    jcr->NumVols = pr->NumVols;
 165    if (variable_expansion(jcr, pr->LabelFormat, &label)) {
 166       bstrncpy(mr->VolumeName, label, sizeof(mr->VolumeName));
 167       ok = true;
 168    }
 169    free_pool_memory(label);
 170    return ok;
 171 }