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