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)str_to_int64(row[0]);
 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 = str_to_uint64(row[0]);
 346    jr->VolSessionTime = str_to_uint64(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 = str_to_int64(row[6]);
 352    jr->JobTDate = str_to_int64(row[7]);
 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  * This function returns a list of all the Client record ids.
 525  *  The caller must free ids if non-NULL.
 526  *
 527  *  Returns 0: on failure
 528  *          1: on success
 529  */
 530 int db_get_client_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
 531 {
 532    SQL_ROW row;
 533    int stat = 0;
 534    int i = 0;
 535    uint32_t *id;
 536
 537    db_lock(mdb);
 538    *ids = NULL;
 539    Mmsg(&mdb->cmd, "SELECT ClientId FROM Client");
 540    if (QUERY_DB(mdb, mdb->cmd)) {
 541       *num_ids = sql_num_rows(mdb);
 542       if (*num_ids > 0) {
 543          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 544          while ((row = sql_fetch_row(mdb)) != NULL) {
 545             id[i++] = (uint32_t)atoi(row[0]);
 546          }
 547          *ids = id;
 548       }
 549       sql_free_result(mdb);
 550       stat = 1;
 551    } else {
 552       Mmsg(&mdb->errmsg, _("Client id select failed: ERR=%s\n"), sql_strerror(mdb));
 553       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 554       stat = 0;
 555    }
 556    db_unlock(mdb);
 557    return stat;
 558 }
 559
 560
 561
 562 /* Get Pool Record
 563  * If the PoolId is non-zero, we get its record,
 564  *  otherwise, we search on the PoolName
 565  *
 566  * Returns: 0 on failure
 567  *          id on success
 568  */
 569 int db_get_pool_record(B_DB *mdb, POOL_DBR *pdbr)
 570 {
 571    SQL_ROW row;
 572    int stat = 0;
 573
 574    db_lock(mdb);
 575    if (pdbr->PoolId != 0) {               /* find by id */
 576       Mmsg(&mdb->cmd,
 577 "SELECT PoolId,Name,NumVols,MaxVols,UseOnce,UseCatalog,AcceptAnyVolume,\
 578 AutoPrune,Recycle,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 579 MaxVolBytes,PoolType,LabelFormat FROM Pool WHERE Pool.PoolId=%u", pdbr->PoolId);
 580    } else {                           /* find by name */
 581       Mmsg(&mdb->cmd,
 582 "SELECT PoolId,Name,NumVols,MaxVols,UseOnce,UseCatalog,AcceptAnyVolume,\
 583 AutoPrune,Recycle,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 584 MaxVolBytes,PoolType,LabelFormat FROM Pool WHERE Pool.Name='%s'", pdbr->Name);
 585    }
 586
 587    if (QUERY_DB(mdb, mdb->cmd)) {
 588       mdb->num_rows = sql_num_rows(mdb);
 589       if (mdb->num_rows > 1) {
 590          char ed1[30];
 591          Mmsg1(&mdb->errmsg, _("More than one Pool!: %s\n"),
 592             edit_uint64(mdb->num_rows, ed1));
 593          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 594       } else if (mdb->num_rows == 1) {
 595          if ((row = sql_fetch_row(mdb)) == NULL) {
 596             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 597             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 598          } else {
 599             pdbr->PoolId = str_to_int64(row[0]);
 600             bstrncpy(pdbr->Name, row[1]!=NULL?row[1]:"", sizeof(pdbr->Name));
 601             pdbr->NumVols = str_to_int64(row[2]);
 602             pdbr->MaxVols = str_to_int64(row[3]);
 603             pdbr->UseOnce = str_to_int64(row[4]);
 604             pdbr->UseCatalog = str_to_int64(row[5]);
 605             pdbr->AcceptAnyVolume = str_to_int64(row[6]);
 606             pdbr->AutoPrune = str_to_int64(row[7]);
 607             pdbr->Recycle = str_to_int64(row[8]);
 608             pdbr->VolRetention = str_to_int64(row[9]);
 609             pdbr->VolUseDuration = str_to_int64(row[10]);
 610             pdbr->MaxVolJobs = str_to_int64(row[11]);
 611             pdbr->MaxVolFiles = str_to_int64(row[12]);
 612             pdbr->MaxVolBytes = str_to_uint64(row[13]);
 613             bstrncpy(pdbr->PoolType, row[13]!=NULL?row[14]:"", sizeof(pdbr->PoolType));
 614             bstrncpy(pdbr->LabelFormat, row[14]!=NULL?row[15]:"", sizeof(pdbr->LabelFormat));
 615             stat = pdbr->PoolId;
 616          }
 617       }
 618       sql_free_result(mdb);
 619    }
 620    db_unlock(mdb);
 621    return stat;
 622 }
 623
 624 /* Get Client Record
 625  * If the ClientId is non-zero, we get its record,
 626  *  otherwise, we search on the Client Name
 627  *
 628  * Returns: 0 on failure
 629  *          1 on success
 630  */
 631 int db_get_client_record(B_DB *mdb, CLIENT_DBR *cdbr)
 632 {
 633    SQL_ROW row;
 634    int stat = 0;
 635
 636    db_lock(mdb);
 637    if (cdbr->ClientId != 0) {               /* find by id */
 638       Mmsg(&mdb->cmd,
 639 "SELECT ClientId,Name,Uname,AutoPrune,FileRetention,JobRetention "
 640 "FROM Client WHERE Client.ClientId=%u", cdbr->ClientId);
 641    } else {                           /* find by name */
 642       Mmsg(&mdb->cmd,
 643 "SELECT ClientId,Name,Uname,AutoPrune,FileRetention,JobRetention "
 644 "FROM Client WHERE Client.Name='%s'", cdbr->Name);
 645    }
 646
 647    if (QUERY_DB(mdb, mdb->cmd)) {
 648       mdb->num_rows = sql_num_rows(mdb);
 649       if (mdb->num_rows > 1) {
 650          char ed1[30];
 651          Mmsg1(&mdb->errmsg, _("More than one Client!: %s\n"),
 652             edit_uint64(mdb->num_rows, ed1));
 653          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 654       } else if (mdb->num_rows == 1) {
 655          if ((row = sql_fetch_row(mdb)) == NULL) {
 656             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 657             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 658          } else {
 659             cdbr->ClientId = str_to_int64(row[0]);
 660             bstrncpy(cdbr->Name, row[1]!=NULL?row[1]:"", sizeof(cdbr->Name));
 661             bstrncpy(cdbr->Uname, row[2]!=NULL?row[1]:"", sizeof(cdbr->Uname));
 662             cdbr->AutoPrune = str_to_int64(row[3]);
 663             cdbr->FileRetention = str_to_int64(row[4]);
 664             cdbr->JobRetention = str_to_int64(row[5]);
 665             stat = 1;
 666          }
 667       }
 668       sql_free_result(mdb);
 669    }
 670    db_unlock(mdb);
 671    return stat;
 672 }
 673
 674
 675 /* Get FileSet Record
 676  * If the FileSetId is non-zero, we get its record,
 677  *  otherwise, we search on the name
 678  *
 679  * Returns: 0 on failure
 680  *          id on success
 681  */
 682 int db_get_fileset_record(B_DB *mdb, FILESET_DBR *fsr)
 683 {
 684    SQL_ROW row;
 685    int stat = 0;
 686
 687    db_lock(mdb);
 688    if (fsr->FileSetId != 0) {               /* find by id */
 689       Mmsg(&mdb->cmd,
 690            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 691            "WHERE FileSetId=%u", fsr->FileSetId);
 692    } else {                           /* find by name */
 693       Mmsg(&mdb->cmd,
 694            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 695            "WHERE FileSet='%s'", fsr->FileSet);
 696    }
 697
 698    if (QUERY_DB(mdb, mdb->cmd)) {
 699       mdb->num_rows = sql_num_rows(mdb);
 700       if (mdb->num_rows > 1) {
 701          char ed1[30];
 702          Mmsg1(&mdb->errmsg, _("Error got %s FileSets but expected only one!\n"),
 703             edit_uint64(mdb->num_rows, ed1));
 704          sql_data_seek(mdb, mdb->num_rows-1);
 705       }
 706       if ((row = sql_fetch_row(mdb)) == NULL) {
 707          Mmsg1(&mdb->errmsg, _("Error: FileSet record \"%s\" not found\n"), fsr->FileSet);
 708       } else {
 709          fsr->FileSetId = atoi(row[0]);
 710          bstrncpy(fsr->FileSet, row[1]!=NULL?row[1]:"", sizeof(fsr->FileSet));
 711          bstrncpy(fsr->MD5, row[2]!=NULL?row[2]:"", sizeof(fsr->MD5));
 712          stat = fsr->FileSetId;
 713       }
 714       sql_free_result(mdb);
 715    }
 716    db_unlock(mdb);
 717    return stat;
 718 }
 719
 720
 721 /*
 722  * Get the number of Media records
 723  *
 724  * Returns: -1 on failure
 725  *          number on success
 726  */
 727 int db_get_num_media_records(B_DB *mdb)
 728 {
 729    int stat = 0;
 730
 731    db_lock(mdb);
 732    Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 733    stat = get_sql_record_max(mdb);
 734    db_unlock(mdb);
 735    return stat;
 736 }
 737
 738
 739 /*
 740  * This function returns a list of all the Media record ids.
 741  *  The caller must free ids if non-NULL.
 742  *
 743  *  Returns 0: on failure
 744  *          1: on success
 745  */
 746 int db_get_media_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
 747 {
 748    SQL_ROW row;
 749    int stat = 0;
 750    int i = 0;
 751    uint32_t *id;
 752
 753    db_lock(mdb);
 754    *ids = NULL;
 755    Mmsg(&mdb->cmd, "SELECT MediaId FROM Media");
 756    if (QUERY_DB(mdb, mdb->cmd)) {
 757       *num_ids = sql_num_rows(mdb);
 758       if (*num_ids > 0) {
 759          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 760          while ((row = sql_fetch_row(mdb)) != NULL) {
 761             id[i++] = (uint32_t)atoi(row[0]);
 762          }
 763          *ids = id;
 764       }
 765       sql_free_result(mdb);
 766       stat = 1;
 767    } else {
 768       Mmsg(&mdb->errmsg, _("Media id select failed: ERR=%s\n"), sql_strerror(mdb));
 769       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 770       stat = 0;
 771    }
 772    db_unlock(mdb);
 773    return stat;
 774 }
 775
 776
 777 /* Get Media Record
 778  *
 779  * Returns: 0 on failure
 780  *          id on success
 781  */
 782 int db_get_media_record(B_DB *mdb, MEDIA_DBR *mr)
 783 {
 784    SQL_ROW row;
 785    int stat = 0;
 786
 787    db_lock(mdb);
 788    if (mr->MediaId == 0 && mr->VolumeName[0] == 0) {
 789       Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 790       mr->MediaId = get_sql_record_max(mdb);
 791       db_unlock(mdb);
 792       return 1;
 793    }
 794    if (mr->MediaId != 0) {               /* find by id */
 795       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 796 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 797 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 798 Recycle,Slot, FirstWritten \
 799 FROM Media WHERE MediaId=%d", mr->MediaId);
 800    } else {                           /* find by name */
 801       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 802 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 803 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 804 Recycle,Slot,FirstWritten \
 805 FROM Media WHERE VolumeName='%s'", mr->VolumeName);
 806    }
 807
 808    if (QUERY_DB(mdb, mdb->cmd)) {
 809       mdb->num_rows = sql_num_rows(mdb);
 810       if (mdb->num_rows > 1) {
 811          char ed1[30];
 812          Mmsg1(&mdb->errmsg, _("More than one Volume!: %s\n"),
 813             edit_uint64(mdb->num_rows, ed1));
 814          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 815       } else if (mdb->num_rows == 1) {
 816          if ((row = sql_fetch_row(mdb)) == NULL) {
 817             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 818             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 819          } else {
 820             /* return values */
 821             mr->MediaId = str_to_int64(row[0]);
 822             bstrncpy(mr->VolumeName, row[1]!=NULL?row[1]:"", sizeof(mr->VolumeName));
 823             mr->VolJobs = str_to_int64(row[2]);
 824             mr->VolFiles = str_to_int64(row[3]);
 825             mr->VolBlocks = str_to_int64(row[4]);
 826             mr->VolBytes = str_to_uint64(row[5]);
 827             mr->VolMounts = str_to_int64(row[6]);
 828             mr->VolErrors = str_to_int64(row[7]);
 829             mr->VolWrites = str_to_int64(row[8]);
 830             mr->MaxVolBytes = str_to_uint64(row[9]);
 831             mr->VolCapacityBytes = str_to_uint64(row[10]);
 832             bstrncpy(mr->MediaType, row[11]!=NULL?row[11]:"", sizeof(mr->MediaType));
 833             bstrncpy(mr->VolStatus, row[12]!=NULL?row[12]:"", sizeof(mr->VolStatus));
 834             mr->PoolId = str_to_int64(row[13]);
 835             mr->VolRetention = str_to_uint64(row[14]);
 836             mr->VolUseDuration = str_to_uint64(row[15]);
 837             mr->MaxVolJobs = str_to_int64(row[16]);
 838             mr->MaxVolFiles = str_to_int64(row[17]);
 839             mr->Recycle = str_to_int64(row[18]);
 840             mr->Slot = str_to_int64(row[19]);
 841             bstrncpy(mr->cFirstWritten, row[20]!=NULL?row[20]:"", sizeof(mr->cFirstWritten));
 842             stat = mr->MediaId;
 843          }
 844       } else {
 845          Mmsg0(&mdb->errmsg, _("Media record not found.\n"));
 846       }
 847       sql_free_result(mdb);
 848    }
 849    db_unlock(mdb);
 850    return stat;
 851 }
 852
 853
 854 #endif /* HAVE_MYSQL || HAVE_SQLITE */