git.sur5r.net Git - bacula/bacula/blob - bacula/src/cats/sql_get.c

   1 /*
   2  * Bacula Catalog Database Get record interface routines
   3  *  Note, these routines generally get a record by id or
   4  *        by name.  If more logic is involved, the routine
   5  *        should be in find.c
   6  *
   7  *    Kern Sibbald, March 2000
   8  *
   9  *    Version $Id$
  10  */
  11
  12 /*
  13    Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
  14
  15    This program is free software; you can redistribute it and/or
  16    modify it under the terms of the GNU General Public License as
  17    published by the Free Software Foundation; either version 2 of
  18    the License, or (at your option) any later version.
  19
  20    This program is distributed in the hope that it will be useful,
  21    but WITHOUT ANY WARRANTY; without even the implied warranty of
  22    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  23    General Public License for more details.
  24
  25    You should have received a copy of the GNU General Public
  26    License along with this program; if not, write to the Free
  27    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  28    MA 02111-1307, USA.
  29
  30  */
  31
  32 /* *****FIXME**** fix fixed length of select_cmd[] and insert_cmd[] */
  33
  34 /* The following is necessary so that we do not include
  35  * the dummy external definition of DB.
  36  */
  37 #define __SQL_C                       /* indicate that this is sql.c */
  38
  39 #include "bacula.h"
  40 #include "cats.h"
  41
  42 #if    HAVE_MYSQL || HAVE_SQLITE
  43
  44 /* -----------------------------------------------------------------------
  45  *
  46  *   Generic Routines (or almost generic)
  47  *
  48  * -----------------------------------------------------------------------
  49  */
  50
  51 /* Forward referenced functions */
  52 static int db_get_file_record(B_DB *mdb, FILE_DBR *fdbr);
  53 static int db_get_filename_record(B_DB *mdb, char *fname);
  54 static int db_get_path_record(B_DB *mdb, char *path);
  55
  56
  57 /* Imported subroutines */
  58 extern void print_result(B_DB *mdb);
  59 extern int QueryDB(char *file, int line, B_DB *db, char *select_cmd);
  60
  61
  62 /*
  63  * Given a full filename (with path), look up the File record
  64  * (with attributes) in the database.
  65  *
  66  *  Returns: 0 on failure
  67  *           1 on success with the File record in FILE_DBR
  68  */
  69 int db_get_file_attributes_record(B_DB *mdb, char *fname, FILE_DBR *fdbr)
  70 {
  71    int fnl, pnl;
  72    char *l, *p;
  73    int stat;
  74    char file[MAXSTRING];
  75    char spath[MAXSTRING];
  76    char buf[MAXSTRING];
  77    Dmsg1(20, "Enter get_file_from_catalog fname=%s \n", fname);
  78
  79    /* Find path without the filename.
  80     * I.e. everything after the last / is a "filename".
  81     * OK, maybe it is a directory name, but we treat it like
  82     * a filename. If we don't find a / then the whole name
  83     * must be a path name (e.g. c:).
  84     */
  85    for (p=l=fname; *p; p++) {
  86       if (*p == '/') {
  87          l = p;
  88       }
  89    }
  90    if (*l == '/') {                   /* did we find a slash? */
  91       l++;                            /* yes, point to filename */
  92    } else {                           /* no, whole thing must be path name */
  93       l = p;
  94    }
  95
  96    /* If filename doesn't exist (i.e. root directory), we
  97     * simply create a blank name consisting of a single
  98     * space. This makes handling zero length filenames
  99     * easier.
 100     */
 101    fnl = p - l;
 102    if (fnl > 255) {
 103       Jmsg1(mdb->jcr, M_WARNING, 0, _("Filename truncated to 255 chars: %s\n"), l);
 104       fnl = 255;
 105    }
 106    if (fnl > 0) {
 107       strncpy(file, l, fnl);          /* copy filename */
 108       file[fnl] = 0;
 109    } else {
 110       file[0] = ' ';                  /* blank filename */
 111       file[1] = 0;
 112       fnl = 1;
 113    }
 114
 115    pnl = l - fname;
 116    if (pnl > 255) {
 117       Jmsg1(mdb->jcr, M_WARNING, 0, _("Path name truncated to 255 chars: %s\n"), fname);
 118       pnl = 255;
 119    }
 120    strncpy(spath, fname, pnl);
 121    spath[pnl] = 0;
 122
 123    if (pnl == 0) {
 124       Mmsg1(&mdb->errmsg, _("Path length is zero. File=%s\n"), fname);
 125       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 126       spath[0] = ' ';
 127       spath[1] = 0;
 128       pnl = 1;
 129    }
 130
 131    Dmsg1(400, "spath=%s\n", spath);
 132    Dmsg1(400, "file=%s\n", file);
 133
 134    db_escape_string(buf, file, fnl);
 135    fdbr->FilenameId = db_get_filename_record(mdb, buf);
 136    Dmsg2(400, "db_get_filename_record FilenameId=%u file=%s\n", fdbr->FilenameId, buf);
 137
 138    db_escape_string(buf, spath, pnl);
 139    fdbr->PathId = db_get_path_record(mdb, buf);
 140    Dmsg2(400, "db_get_path_record PathId=%u path=%s\n", fdbr->PathId, buf);
 141
 142    stat = db_get_file_record(mdb, fdbr);
 143
 144    return stat;
 145 }
 146
 147
 148 /*
 149  * Get a File record
 150  * Returns: 0 on failure
 151  *          1 on success
 152  *
 153  *  DO NOT use Jmsg in this routine.
 154  *
 155  *  Note in this routine, we do not use Jmsg because it may be
 156  *    called to get attributes of a non-existent file, which is
 157  *    "normal" if a new file is found during Verify.
 158  */
 159 static
 160 int db_get_file_record(B_DB *mdb, FILE_DBR *fdbr)
 161 {
 162    SQL_ROW row;
 163    int stat = 0;
 164
 165    db_lock(mdb);
 166    Mmsg(&mdb->cmd,
 167 "SELECT FileId, LStat, MD5 from File where File.JobId=%u and File.PathId=%u and \
 168 File.FilenameId=%u", fdbr->JobId, fdbr->PathId, fdbr->FilenameId);
 169
 170    Dmsg3(050, "Get_file_record JobId=%u FilenameId=%u PathId=%u\n",
 171       fdbr->JobId, fdbr->FilenameId, fdbr->PathId);
 172
 173    Dmsg1(100, "Query=%s\n", mdb->cmd);
 174
 175    if (QUERY_DB(mdb, mdb->cmd)) {
 176
 177       mdb->num_rows = sql_num_rows(mdb);
 178       Dmsg1(050, "get_file_record num_rows=%d\n", (int)mdb->num_rows);
 179
 180       if (mdb->num_rows > 1) {
 181          Mmsg1(&mdb->errmsg, _("get_file_record want 1 got rows=%d\n"),
 182             mdb->num_rows);
 183       }
 184       if (mdb->num_rows >= 1) {
 185          if ((row = sql_fetch_row(mdb)) == NULL) {
 186             Mmsg1(&mdb->errmsg, _("Error fetching row: %s\n"), sql_strerror(mdb));
 187          } else {
 188             fdbr->FileId = (FileId_t)strtod(row[0], NULL);
 189             strncpy(fdbr->LStat, row[1], sizeof(fdbr->LStat));
 190             fdbr->LStat[sizeof(fdbr->LStat)] = 0;
 191             strncpy(fdbr->MD5, row[2], sizeof(fdbr->MD5));
 192             fdbr->MD5[sizeof(fdbr->MD5)] = 0;
 193             stat = 1;
 194          }
 195       } else {
 196          Mmsg2(&mdb->errmsg, _("File record not found for PathId=%u FilenameId=%u\n"),
 197             fdbr->PathId, fdbr->FilenameId);
 198       }
 199       sql_free_result(mdb);
 200    }
 201    db_unlock(mdb);
 202    return stat;
 203
 204 }
 205
 206 /* Get Filename record
 207  * Returns: 0 on failure
 208  *          FilenameId on success
 209  *
 210  *   DO NOT use Jmsg in this routine (see notes for get_file_record)
 211  */
 212 static int db_get_filename_record(B_DB *mdb, char *fname)
 213 {
 214    SQL_ROW row;
 215    int FilenameId = 0;
 216
 217    if (*fname == 0) {
 218       Mmsg0(&mdb->errmsg, _("Null name given to db_get_filename_record\n"));
 219       Emsg0(M_ABORT, 0, mdb->errmsg);
 220    }
 221
 222    db_lock(mdb);
 223    Mmsg(&mdb->cmd, "SELECT FilenameId FROM Filename WHERE Name='%s'", fname);
 224    if (QUERY_DB(mdb, mdb->cmd)) {
 225
 226       mdb->num_rows = sql_num_rows(mdb);
 227
 228       if (mdb->num_rows > 1) {
 229          Mmsg1(&mdb->errmsg, _("More than one Filename!: %d\n"), (int)(mdb->num_rows));
 230       }
 231       if (mdb->num_rows >= 1) {
 232          if ((row = sql_fetch_row(mdb)) == NULL) {
 233             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 234          } else {
 235             FilenameId = atoi(row[0]);
 236             if (FilenameId <= 0) {
 237                Mmsg2(&mdb->errmsg, _("Get DB Filename record %s found bad record: %d\n"),
 238                   mdb->cmd, FilenameId);
 239                FilenameId = 0;
 240             }
 241          }
 242       } else {
 243          Mmsg1(&mdb->errmsg, _("Filename record: %s not found.\n"), fname);
 244       }
 245       sql_free_result(mdb);
 246    }
 247    db_unlock(mdb);
 248    return FilenameId;
 249 }
 250
 251 /* Get path record
 252  * Returns: 0 on failure
 253  *          PathId on success
 254  *
 255  *   DO NOT use Jmsg in this routine (see notes for get_file_record)
 256  */
 257 static int db_get_path_record(B_DB *mdb, char *path)
 258 {
 259    SQL_ROW row;
 260    uint32_t PathId = 0;
 261
 262    if (*path == 0) {
 263       Emsg0(M_ABORT, 0, _("Null path given to db_get_path_record\n"));
 264    }
 265
 266    db_lock(mdb);
 267
 268    if (mdb->cached_path_id != 0 && strcmp(mdb->cached_path, path) == 0) {
 269       db_unlock(mdb);
 270       return mdb->cached_path_id;
 271    }
 272
 273    Mmsg(&mdb->cmd, "SELECT PathId FROM Path WHERE Path='%s'", path);
 274
 275    if (QUERY_DB(mdb, mdb->cmd)) {
 276       char ed1[30];
 277       mdb->num_rows = sql_num_rows(mdb);
 278
 279       if (mdb->num_rows > 1) {
 280          Mmsg1(&mdb->errmsg, _("More than one Path!: %s\n"),
 281             edit_uint64(mdb->num_rows, ed1));
 282          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 283       } else if (mdb->num_rows == 1) {
 284          if ((row = sql_fetch_row(mdb)) == NULL) {
 285             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 286          } else {
 287             PathId = atoi(row[0]);
 288             if (PathId <= 0) {
 289                Mmsg2(&mdb->errmsg, _("Get DB path record %s found bad record: %u\n"),
 290                   mdb->cmd, PathId);
 291                PathId = 0;
 292             } else {
 293                /* Cache path */
 294                if (PathId != mdb->cached_path_id) {
 295                   mdb->cached_path_id = PathId;
 296                   mdb->cached_path = check_pool_memory_size(mdb->cached_path,
 297                      strlen(path)+1);
 298                   strcpy(mdb->cached_path, path);
 299                }
 300             }
 301          }
 302       } else {
 303          Mmsg1(&mdb->errmsg, _("Path record: %s not found.\n"), path);
 304       }
 305       sql_free_result(mdb);
 306    }
 307    db_unlock(mdb);
 308    return PathId;
 309 }
 310
 311
 312 /*
 313  * Get Job record for given JobId or Job name
 314  * Returns: 0 on failure
 315  *          1 on success
 316  */
 317 int db_get_job_record(B_DB *mdb, JOB_DBR *jr)
 318 {
 319    SQL_ROW row;
 320
 321    db_lock(mdb);
 322    if (jr->JobId == 0) {
 323       Mmsg(&mdb->cmd, "SELECT VolSessionId,VolSessionTime,\
 324 PoolId,StartTime,EndTime,JobFiles,JobBytes,JobTDate,Job,JobStatus,\
 325 Type,Level \
 326 FROM Job WHERE Job='%s'", jr->Job);
 327     } else {
 328       Mmsg(&mdb->cmd, "SELECT VolSessionId,VolSessionTime,\
 329 PoolId,StartTime,EndTime,JobFiles,JobBytes,JobTDate,Job,JobStatus,\
 330 Type,Level \
 331 FROM Job WHERE JobId=%u", jr->JobId);
 332     }
 333
 334    if (!QUERY_DB(mdb, mdb->cmd)) {
 335       db_unlock(mdb);
 336       return 0;                       /* failed */
 337    }
 338    if ((row = sql_fetch_row(mdb)) == NULL) {
 339       Mmsg1(&mdb->errmsg, _("No Job found for JobId %u\n"), jr->JobId);
 340       sql_free_result(mdb);
 341       db_unlock(mdb);
 342       return 0;                       /* failed */
 343    }
 344
 345    jr->VolSessionId = atol(row[0]);
 346    jr->VolSessionTime = atol(row[1]);
 347    jr->PoolId = atoi(row[2]);
 348    bstrncpy(jr->cStartTime, row[3]!=NULL?row[3]:"", sizeof(jr->cStartTime));
 349    bstrncpy(jr->cEndTime, row[4]!=NULL?row[4]:"", sizeof(jr->cEndTime));
 350    jr->JobFiles = atol(row[5]);
 351    jr->JobBytes = (uint64_t)strtod(row[6], NULL);
 352    jr->JobTDate = (utime_t)strtod(row[7], NULL);
 353    bstrncpy(jr->Job, row[8]!=NULL?row[8]:"", sizeof(jr->Job));
 354    jr->JobStatus = (int)*row[9];
 355    jr->Type = (int)*row[10];
 356    jr->Level = (int)*row[11];
 357    sql_free_result(mdb);
 358
 359    db_unlock(mdb);
 360    return 1;
 361 }
 362
 363 /*
 364  * Find VolumeNames for a give JobId
 365  *  Returns: 0 on error or no Volumes found
 366  *           number of volumes on success
 367  *              Volumes are concatenated in VolumeNames
 368  *              separated by a vertical bar (|).
 369  *
 370  *  Returns: number of volumes on success
 371  */
 372 int db_get_job_volume_names(B_DB *mdb, uint32_t JobId, POOLMEM **VolumeNames)
 373 {
 374    SQL_ROW row;
 375    int stat = 0;
 376    int i;
 377
 378    db_lock(mdb);
 379    Mmsg(&mdb->cmd,
 380 "SELECT VolumeName FROM JobMedia,Media WHERE JobMedia.JobId=%u \
 381 AND JobMedia.MediaId=Media.MediaId", JobId);
 382
 383    Dmsg1(130, "VolNam=%s\n", mdb->cmd);
 384    *VolumeNames[0] = 0;
 385    if (QUERY_DB(mdb, mdb->cmd)) {
 386       mdb->num_rows = sql_num_rows(mdb);
 387       Dmsg1(130, "Num rows=%d\n", mdb->num_rows);
 388       if (mdb->num_rows <= 0) {
 389          Mmsg1(&mdb->errmsg, _("No volumes found for JobId=%d\n"), JobId);
 390          stat = 0;
 391       } else {
 392          stat = mdb->num_rows;
 393          for (i=0; i < stat; i++) {
 394             if ((row = sql_fetch_row(mdb)) == NULL) {
 395                Mmsg2(&mdb->errmsg, _("Error fetching row %d: ERR=%s\n"), i, sql_strerror(mdb));
 396                Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 397                stat = 0;
 398                break;
 399             } else {
 400                if (*VolumeNames[0] != 0) {
 401                   pm_strcat(VolumeNames, "|");
 402                }
 403                pm_strcat(VolumeNames, row[0]);
 404             }
 405          }
 406       }
 407       sql_free_result(mdb);
 408    }
 409    db_unlock(mdb);
 410    return stat;
 411 }
 412
 413 /*
 414  * Find Volume parameters for a give JobId
 415  *  Returns: 0 on error or no Volumes found
 416  *           number of volumes on success
 417  *           List of Volumes and start/end file/blocks (malloced structure!)
 418  *
 419  *  Returns: number of volumes on success
 420  */
 421 int db_get_job_volume_parameters(B_DB *mdb, uint32_t JobId, VOL_PARAMS **VolParams)
 422 {
 423    SQL_ROW row;
 424    int stat = 0;
 425    int i;
 426    VOL_PARAMS *Vols = NULL;
 427
 428    db_lock(mdb);
 429    Mmsg(&mdb->cmd,
 430 "SELECT VolumeName,StartFile,EndFile,StartBlock,EndBlock"
 431 " FROM JobMedia,Media WHERE JobMedia.JobId=%u"
 432 " AND JobMedia.MediaId=Media.MediaId", JobId);
 433
 434    Dmsg1(130, "VolNam=%s\n", mdb->cmd);
 435    if (QUERY_DB(mdb, mdb->cmd)) {
 436       mdb->num_rows = sql_num_rows(mdb);
 437       Dmsg1(130, "Num rows=%d\n", mdb->num_rows);
 438       if (mdb->num_rows <= 0) {
 439          Mmsg1(&mdb->errmsg, _("No volumes found for JobId=%d\n"), JobId);
 440          stat = 0;
 441       } else {
 442          stat = mdb->num_rows;
 443          if (stat > 0) {
 444             *VolParams = Vols = (VOL_PARAMS *)malloc(stat * sizeof(VOL_PARAMS));
 445          }
 446          for (i=0; i < stat; i++) {
 447             if ((row = sql_fetch_row(mdb)) == NULL) {
 448                Mmsg2(&mdb->errmsg, _("Error fetching row %d: ERR=%s\n"), i, sql_strerror(mdb));
 449                Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 450                stat = 0;
 451                break;
 452             } else {
 453                bstrncpy(Vols[i].VolumeName, row[0], MAX_NAME_LENGTH);
 454                Vols[i].StartFile = atoi(row[1]);
 455                Vols[i].EndFile = atoi(row[2]);
 456                Vols[i].StartBlock = atoi(row[3]);
 457                Vols[i].EndBlock = atoi(row[4]);
 458             }
 459          }
 460       }
 461       sql_free_result(mdb);
 462    }
 463    db_unlock(mdb);
 464    return stat;
 465 }
 466
 467
 468
 469 /*
 470  * Get the number of pool records
 471  *
 472  * Returns: -1 on failure
 473  *          number on success
 474  */
 475 int db_get_num_pool_records(B_DB *mdb)
 476 {
 477    int stat = 0;
 478
 479    db_lock(mdb);
 480    Mmsg(&mdb->cmd, "SELECT count(*) from Pool");
 481    stat = get_sql_record_max(mdb);
 482    db_unlock(mdb);
 483    return stat;
 484 }
 485
 486 /*
 487  * This function returns a list of all the Pool record ids.
 488  *  The caller must free ids if non-NULL.
 489  *
 490  *  Returns 0: on failure
 491  *          1: on success
 492  */
 493 int db_get_pool_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
 494 {
 495    SQL_ROW row;
 496    int stat = 0;
 497    int i = 0;
 498    uint32_t *id;
 499
 500    db_lock(mdb);
 501    *ids = NULL;
 502    Mmsg(&mdb->cmd, "SELECT PoolId FROM Pool");
 503    if (QUERY_DB(mdb, mdb->cmd)) {
 504       *num_ids = sql_num_rows(mdb);
 505       if (*num_ids > 0) {
 506          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 507          while ((row = sql_fetch_row(mdb)) != NULL) {
 508             id[i++] = (uint32_t)atoi(row[0]);
 509          }
 510          *ids = id;
 511       }
 512       sql_free_result(mdb);
 513       stat = 1;
 514    } else {
 515       Mmsg(&mdb->errmsg, _("Pool id select failed: ERR=%s\n"), sql_strerror(mdb));
 516       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 517       stat = 0;
 518    }
 519    db_unlock(mdb);
 520    return stat;
 521 }
 522
 523
 524 /* Get Pool Record
 525  * If the PoolId is non-zero, we get its record,
 526  *  otherwise, we search on the PoolName
 527  *
 528  * Returns: 0 on failure
 529  *          id on success
 530  */
 531 int db_get_pool_record(B_DB *mdb, POOL_DBR *pdbr)
 532 {
 533    SQL_ROW row;
 534    int stat = 0;
 535
 536    db_lock(mdb);
 537    if (pdbr->PoolId != 0) {               /* find by id */
 538       Mmsg(&mdb->cmd,
 539 "SELECT PoolId,Name,NumVols,MaxVols,UseOnce,UseCatalog,AcceptAnyVolume,\
 540 AutoPrune,Recycle,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 541 MaxVolBytes,PoolType,LabelFormat FROM Pool WHERE Pool.PoolId=%d", pdbr->PoolId);
 542    } else {                           /* find by name */
 543       Mmsg(&mdb->cmd,
 544 "SELECT PoolId,Name,NumVols,MaxVols,UseOnce,UseCatalog,AcceptAnyVolume,\
 545 AutoPrune,Recycle,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 546 MaxVolBytes,PoolType,LabelFormat FROM Pool WHERE Pool.Name='%s'", pdbr->Name);
 547    }
 548
 549    if (QUERY_DB(mdb, mdb->cmd)) {
 550       mdb->num_rows = sql_num_rows(mdb);
 551       if (mdb->num_rows > 1) {
 552          char ed1[30];
 553          Mmsg1(&mdb->errmsg, _("More than one Pool!: %s\n"),
 554             edit_uint64(mdb->num_rows, ed1));
 555          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 556       } else if (mdb->num_rows == 1) {
 557          if ((row = sql_fetch_row(mdb)) == NULL) {
 558             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 559             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 560          } else {
 561             pdbr->PoolId = atoi(row[0]);
 562             bstrncpy(pdbr->Name, row[1]!=NULL?row[1]:"", sizeof(pdbr->Name));
 563             pdbr->NumVols = atoi(row[2]);
 564             pdbr->MaxVols = atoi(row[3]);
 565             pdbr->UseOnce = atoi(row[4]);
 566             pdbr->UseCatalog = atoi(row[5]);
 567             pdbr->AcceptAnyVolume = atoi(row[6]);
 568             pdbr->AutoPrune = atoi(row[7]);
 569             pdbr->Recycle = atoi(row[8]);
 570             pdbr->VolRetention = (utime_t)strtod(row[9], NULL);
 571             pdbr->VolUseDuration = (utime_t)strtod(row[10], NULL);
 572             pdbr->MaxVolJobs = atoi(row[11]);
 573             pdbr->MaxVolFiles = atoi(row[12]);
 574             pdbr->MaxVolBytes = (uint64_t)strtod(row[13], NULL);
 575             bstrncpy(pdbr->PoolType, row[13]!=NULL?row[14]:"", sizeof(pdbr->PoolType));
 576             bstrncpy(pdbr->LabelFormat, row[14]!=NULL?row[15]:"", sizeof(pdbr->LabelFormat));
 577             stat = pdbr->PoolId;
 578          }
 579       }
 580       sql_free_result(mdb);
 581    }
 582    db_unlock(mdb);
 583    return stat;
 584 }
 585
 586 /* Get FileSet Record
 587  * If the FileSetId is non-zero, we get its record,
 588  *  otherwise, we search on the name
 589  *
 590  * Returns: 0 on failure
 591  *          id on success
 592  */
 593 int db_get_fileset_record(B_DB *mdb, FILESET_DBR *fsr)
 594 {
 595    SQL_ROW row;
 596    int stat = 0;
 597
 598    db_lock(mdb);
 599    if (fsr->FileSetId != 0) {               /* find by id */
 600       Mmsg(&mdb->cmd,
 601            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 602            "WHERE FileSetId=%u", fsr->FileSetId);
 603    } else {                           /* find by name */
 604       Mmsg(&mdb->cmd,
 605            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 606            "WHERE FileSet='%s'", fsr->FileSet);
 607    }
 608
 609    if (QUERY_DB(mdb, mdb->cmd)) {
 610       mdb->num_rows = sql_num_rows(mdb);
 611       if (mdb->num_rows > 1) {
 612          char ed1[30];
 613          Mmsg1(&mdb->errmsg, _("Error got %s FileSets but expected only one!\n"),
 614             edit_uint64(mdb->num_rows, ed1));
 615          sql_data_seek(mdb, mdb->num_rows-1);
 616       }
 617       if ((row = sql_fetch_row(mdb)) == NULL) {
 618          Mmsg1(&mdb->errmsg, _("Error: FileSet record \"%s\" not found\n"), fsr->FileSet);
 619       } else {
 620          fsr->FileSetId = atoi(row[0]);
 621          bstrncpy(fsr->FileSet, row[1]!=NULL?row[1]:"", sizeof(fsr->FileSet));
 622          bstrncpy(fsr->MD5, row[2]!=NULL?row[2]:"", sizeof(fsr->MD5));
 623          stat = fsr->FileSetId;
 624       }
 625       sql_free_result(mdb);
 626    }
 627    db_unlock(mdb);
 628    return stat;
 629 }
 630
 631
 632 /*
 633  * Get the number of Media records
 634  *
 635  * Returns: -1 on failure
 636  *          number on success
 637  */
 638 int db_get_num_media_records(B_DB *mdb)
 639 {
 640    int stat = 0;
 641
 642    db_lock(mdb);
 643    Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 644    stat = get_sql_record_max(mdb);
 645    db_unlock(mdb);
 646    return stat;
 647 }
 648
 649
 650 /*
 651  * This function returns a list of all the Media record ids.
 652  *  The caller must free ids if non-NULL.
 653  *
 654  *  Returns 0: on failure
 655  *          1: on success
 656  */
 657 int db_get_media_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
 658 {
 659    SQL_ROW row;
 660    int stat = 0;
 661    int i = 0;
 662    uint32_t *id;
 663
 664    db_lock(mdb);
 665    *ids = NULL;
 666    Mmsg(&mdb->cmd, "SELECT MediaId FROM Media");
 667    if (QUERY_DB(mdb, mdb->cmd)) {
 668       *num_ids = sql_num_rows(mdb);
 669       if (*num_ids > 0) {
 670          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 671          while ((row = sql_fetch_row(mdb)) != NULL) {
 672             id[i++] = (uint32_t)atoi(row[0]);
 673          }
 674          *ids = id;
 675       }
 676       sql_free_result(mdb);
 677       stat = 1;
 678    } else {
 679       Mmsg(&mdb->errmsg, _("Media id select failed: ERR=%s\n"), sql_strerror(mdb));
 680       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 681       stat = 0;
 682    }
 683    db_unlock(mdb);
 684    return stat;
 685 }
 686
 687
 688 /* Get Media Record
 689  *
 690  * Returns: 0 on failure
 691  *          id on success
 692  */
 693 int db_get_media_record(B_DB *mdb, MEDIA_DBR *mr)
 694 {
 695    SQL_ROW row;
 696    int stat = 0;
 697
 698    db_lock(mdb);
 699    if (mr->MediaId == 0 && mr->VolumeName[0] == 0) {
 700       Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 701       mr->MediaId = get_sql_record_max(mdb);
 702       db_unlock(mdb);
 703       return 1;
 704    }
 705    if (mr->MediaId != 0) {               /* find by id */
 706       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 707 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 708 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 709 Recycle,Slot, FirstWritten \
 710 FROM Media WHERE MediaId=%d", mr->MediaId);
 711    } else {                           /* find by name */
 712       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 713 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 714 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 715 Recycle,Slot,FirstWritten \
 716 FROM Media WHERE VolumeName='%s'", mr->VolumeName);
 717    }
 718
 719    if (QUERY_DB(mdb, mdb->cmd)) {
 720       mdb->num_rows = sql_num_rows(mdb);
 721       if (mdb->num_rows > 1) {
 722          char ed1[30];
 723          Mmsg1(&mdb->errmsg, _("More than one Volume!: %s\n"),
 724             edit_uint64(mdb->num_rows, ed1));
 725          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 726       } else if (mdb->num_rows == 1) {
 727          if ((row = sql_fetch_row(mdb)) == NULL) {
 728             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 729             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 730          } else {
 731             /* return values */
 732             mr->MediaId = atoi(row[0]);
 733             bstrncpy(mr->VolumeName, row[1]!=NULL?row[1]:"", sizeof(mr->VolumeName));
 734             mr->VolJobs = atoi(row[2]);
 735             mr->VolFiles = atoi(row[3]);
 736             mr->VolBlocks = atoi(row[4]);
 737             mr->VolBytes = (uint64_t)strtod(row[5], NULL);
 738             mr->VolMounts = atoi(row[6]);
 739             mr->VolErrors = atoi(row[7]);
 740             mr->VolWrites = atoi(row[8]);
 741             mr->MaxVolBytes = (uint64_t)strtod(row[9], NULL);
 742             mr->VolCapacityBytes = (uint64_t)strtod(row[10], NULL);
 743             bstrncpy(mr->MediaType, row[11]!=NULL?row[11]:"", sizeof(mr->MediaType));
 744             bstrncpy(mr->VolStatus, row[12]!=NULL?row[12]:"", sizeof(mr->VolStatus));
 745             mr->PoolId = atoi(row[13]);
 746             mr->VolRetention = (utime_t)strtod(row[14], NULL);
 747             mr->VolUseDuration = (utime_t)strtod(row[15], NULL);
 748             mr->MaxVolJobs = atoi(row[16]);
 749             mr->MaxVolFiles = atoi(row[17]);
 750             mr->Recycle = atoi(row[18]);
 751             mr->Slot = atoi(row[19]);
 752             bstrncpy(mr->cFirstWritten, row[20]!=NULL?row[20]:"", sizeof(mr->cFirstWritten));
 753             stat = mr->MediaId;
 754          }
 755       } else {
 756          Mmsg0(&mdb->errmsg, _("Media record not found.\n"));
 757       }
 758       sql_free_result(mdb);
 759    }
 760    db_unlock(mdb);
 761    return stat;
 762 }
 763
 764
 765 #endif /* HAVE_MYSQL || HAVE_SQLITE */