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-2003 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
  33 /* The following is necessary so that we do not include
  34  * the dummy external definition of DB.
  35  */
  36 #define __SQL_C                       /* indicate that this is sql.c */
  37
  38 #include "bacula.h"
  39 #include "cats.h"
  40
  41 #if    HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL
  42
  43 /* -----------------------------------------------------------------------
  44  *
  45  *   Generic Routines (or almost generic)
  46  *
  47  * -----------------------------------------------------------------------
  48  */
  49
  50 /* Forward referenced functions */
  51 static int db_get_file_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr, FILE_DBR *fdbr);
  52 static int db_get_filename_record(JCR *jcr, B_DB *mdb);
  53 static int db_get_path_record(JCR *jcr, B_DB *mdb);
  54
  55
  56 /* Imported subroutines */
  57 extern void print_result(B_DB *mdb);
  58 extern int QueryDB(const char *file, int line, JCR *jcr, B_DB *db, char *select_cmd);
  59 extern void split_path_and_file(JCR *jcr, B_DB *mdb, const char *fname);
  60
  61
  62
  63 /*
  64  * Given a full filename (with path), look up the File record
  65  * (with attributes) in the database.
  66  *
  67  *  Returns: 0 on failure
  68  *           1 on success with the File record in FILE_DBR
  69  */
  70 int db_get_file_attributes_record(JCR *jcr, B_DB *mdb, char *fname, JOB_DBR *jr, FILE_DBR *fdbr)
  71 {
  72    int stat;
  73    Dmsg1(100, "db_get_file_att_record fname=%s \n", fname);
  74
  75    db_lock(mdb);
  76    split_path_and_file(jcr, mdb, fname);
  77
  78    fdbr->FilenameId = db_get_filename_record(jcr, mdb);
  79
  80    fdbr->PathId = db_get_path_record(jcr, mdb);
  81
  82    stat = db_get_file_record(jcr, mdb, jr, fdbr);
  83
  84    db_unlock(mdb);
  85
  86    return stat;
  87 }
  88
  89
  90 /*
  91  * Get a File record
  92  * Returns: 0 on failure
  93  *          1 on success
  94  *
  95  *  DO NOT use Jmsg in this routine.
  96  *
  97  *  Note in this routine, we do not use Jmsg because it may be
  98  *    called to get attributes of a non-existent file, which is
  99  *    "normal" if a new file is found during Verify.
 100  */
 101 static
 102 int db_get_file_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr, FILE_DBR *fdbr)
 103 {
 104    SQL_ROW row;
 105    int stat = 0;
 106
 107    if (jcr->JobLevel == L_VERIFY_DISK_TO_CATALOG) {
 108    Mmsg(&mdb->cmd,
 109 "SELECT FileId, LStat, MD5 FROM File,Job WHERE "
 110 "File.JobId=Job.JobId AND File.PathId=%u AND "
 111 "File.FilenameId=%u AND Job.Type='B' AND Job.JobSTATUS='T' AND "
 112 "ClientId=%u ORDER BY StartTime DESC LIMIT 1",
 113       fdbr->PathId, fdbr->FilenameId, jr->ClientId);
 114    } else {
 115       Mmsg(&mdb->cmd,
 116 "SELECT FileId, LStat, MD5 FROM File WHERE File.JobId=%u AND File.PathId=%u AND "
 117 "File.FilenameId=%u", fdbr->JobId, fdbr->PathId, fdbr->FilenameId);
 118    }
 119    Dmsg3(050, "Get_file_record JobId=%u FilenameId=%u PathId=%u\n",
 120       fdbr->JobId, fdbr->FilenameId, fdbr->PathId);
 121
 122    Dmsg1(100, "Query=%s\n", mdb->cmd);
 123
 124    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 125
 126       mdb->num_rows = sql_num_rows(mdb);
 127       Dmsg1(050, "get_file_record num_rows=%d\n", (int)mdb->num_rows);
 128
 129       if (mdb->num_rows > 1) {
 130          Mmsg1(&mdb->errmsg, _("get_file_record want 1 got rows=%d\n"),
 131             mdb->num_rows);
 132       }
 133       if (mdb->num_rows >= 1) {
 134          if ((row = sql_fetch_row(mdb)) == NULL) {
 135             Mmsg1(&mdb->errmsg, _("Error fetching row: %s\n"), sql_strerror(mdb));
 136          } else {
 137             fdbr->FileId = (FileId_t)str_to_int64(row[0]);
 138             bstrncpy(fdbr->LStat, row[1], sizeof(fdbr->LStat));
 139             bstrncpy(fdbr->SIG, row[2], sizeof(fdbr->SIG));
 140             stat = 1;
 141          }
 142       } else {
 143          Mmsg2(&mdb->errmsg, _("File record for PathId=%u FilenameId=%u not found.\n"),
 144             fdbr->PathId, fdbr->FilenameId);
 145       }
 146       sql_free_result(mdb);
 147    } else {
 148       Mmsg(&mdb->errmsg, _("File record not found in Catalog.\n"));
 149    }
 150    return stat;
 151
 152 }
 153
 154 /* Get Filename record
 155  * Returns: 0 on failure
 156  *          FilenameId on success
 157  *
 158  *   DO NOT use Jmsg in this routine (see notes for get_file_record)
 159  */
 160 static int db_get_filename_record(JCR *jcr, B_DB *mdb)
 161 {
 162    SQL_ROW row;
 163    int FilenameId = 0;
 164
 165    mdb->esc_name = check_pool_memory_size(mdb->esc_name, 2*mdb->fnl+2);
 166    db_escape_string(mdb->esc_name, mdb->fname, mdb->fnl);
 167
 168    Mmsg(&mdb->cmd, "SELECT FilenameId FROM Filename WHERE Name='%s'", mdb->esc_name);
 169    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 170       char ed1[30];
 171
 172       mdb->num_rows = sql_num_rows(mdb);
 173
 174       if (mdb->num_rows > 1) {
 175          Mmsg2(&mdb->errmsg, _("More than one Filename!: %s for file: %s\n"),
 176             edit_uint64(mdb->num_rows, ed1), mdb->fname);
 177          Jmsg(jcr, M_WARNING, 0, "%s", mdb->errmsg);
 178       }
 179       if (mdb->num_rows >= 1) {
 180          if ((row = sql_fetch_row(mdb)) == NULL) {
 181             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 182          } else {
 183             FilenameId = atoi(row[0]);
 184             if (FilenameId <= 0) {
 185                Mmsg2(&mdb->errmsg, _("Get DB Filename record %s found bad record: %d\n"),
 186                   mdb->cmd, FilenameId);
 187                FilenameId = 0;
 188             }
 189          }
 190       } else {
 191          Mmsg1(&mdb->errmsg, _("Filename record: %s not found.\n"), mdb->fname);
 192       }
 193       sql_free_result(mdb);
 194    } else {
 195       Mmsg(&mdb->errmsg, _("Filename record: %s not found in Catalog.\n"), mdb->fname);
 196    }
 197    return FilenameId;
 198 }
 199
 200 /* Get path record
 201  * Returns: 0 on failure
 202  *          PathId on success
 203  *
 204  *   DO NOT use Jmsg in this routine (see notes for get_file_record)
 205  */
 206 static int db_get_path_record(JCR *jcr, B_DB *mdb)
 207 {
 208    SQL_ROW row;
 209    uint32_t PathId = 0;
 210
 211    mdb->esc_name = check_pool_memory_size(mdb->esc_name, 2*mdb->pnl+2);
 212    db_escape_string(mdb->esc_name, mdb->path, mdb->pnl);
 213
 214    if (mdb->cached_path_id != 0 && mdb->cached_path_len == mdb->pnl &&
 215        strcmp(mdb->cached_path, mdb->path) == 0) {
 216       return mdb->cached_path_id;
 217    }
 218
 219    Mmsg(&mdb->cmd, "SELECT PathId FROM Path WHERE Path='%s'", mdb->esc_name);
 220
 221    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 222       char ed1[30];
 223       mdb->num_rows = sql_num_rows(mdb);
 224
 225       if (mdb->num_rows > 1) {
 226          Mmsg2(&mdb->errmsg, _("More than one Path!: %s for path: %s\n"),
 227             edit_uint64(mdb->num_rows, ed1), mdb->path);
 228          Jmsg(jcr, M_WARNING, 0, "%s", mdb->errmsg);
 229       }
 230       /* Even if there are multiple paths, take the first one */
 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             PathId = atoi(row[0]);
 236             if (PathId <= 0) {
 237                Mmsg2(&mdb->errmsg, _("Get DB path record %s found bad record: %u\n"),
 238                   mdb->cmd, PathId);
 239                PathId = 0;
 240             } else {
 241                /* Cache path */
 242                if (PathId != mdb->cached_path_id) {
 243                   mdb->cached_path_id = PathId;
 244                   mdb->cached_path_len = mdb->pnl;
 245                   pm_strcpy(&mdb->cached_path, mdb->path);
 246                }
 247             }
 248          }
 249       } else {
 250          Mmsg1(&mdb->errmsg, _("Path record: %s not found.\n"), mdb->path);
 251       }
 252       sql_free_result(mdb);
 253    } else {
 254       Mmsg(&mdb->errmsg, _("Path record: %s not found in Catalog.\n"), mdb->path);
 255    }
 256    return PathId;
 257 }
 258
 259
 260 /*
 261  * Get Job record for given JobId or Job name
 262  * Returns: 0 on failure
 263  *          1 on success
 264  */
 265 int db_get_job_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr)
 266 {
 267    SQL_ROW row;
 268
 269    db_lock(mdb);
 270    if (jr->JobId == 0) {
 271       Mmsg(&mdb->cmd, "SELECT VolSessionId,VolSessionTime,"
 272 "PoolId,StartTime,EndTime,JobFiles,JobBytes,JobTDate,Job,JobStatus,"
 273 "Type,Level,ClientId "
 274 "FROM Job WHERE Job='%s'", jr->Job);
 275     } else {
 276       Mmsg(&mdb->cmd, "SELECT VolSessionId,VolSessionTime,"
 277 "PoolId,StartTime,EndTime,JobFiles,JobBytes,JobTDate,Job,JobStatus,"
 278 "Type,Level,ClientId "
 279 "FROM Job WHERE JobId=%u", jr->JobId);
 280     }
 281
 282    if (!QUERY_DB(jcr, mdb, mdb->cmd)) {
 283       db_unlock(mdb);
 284       return 0;                       /* failed */
 285    }
 286    if ((row = sql_fetch_row(mdb)) == NULL) {
 287       Mmsg1(&mdb->errmsg, _("No Job found for JobId %u\n"), jr->JobId);
 288       sql_free_result(mdb);
 289       db_unlock(mdb);
 290       return 0;                       /* failed */
 291    }
 292
 293    jr->VolSessionId = str_to_uint64(row[0]);
 294    jr->VolSessionTime = str_to_uint64(row[1]);
 295    jr->PoolId = str_to_uint64(row[2]);
 296    bstrncpy(jr->cStartTime, row[3]!=NULL?row[3]:"", sizeof(jr->cStartTime));
 297    bstrncpy(jr->cEndTime, row[4]!=NULL?row[4]:"", sizeof(jr->cEndTime));
 298    jr->JobFiles = atol(row[5]);
 299    jr->JobBytes = str_to_int64(row[6]);
 300    jr->JobTDate = str_to_int64(row[7]);
 301    bstrncpy(jr->Job, row[8]!=NULL?row[8]:"", sizeof(jr->Job));
 302    jr->JobStatus = (int)*row[9];
 303    jr->Type = (int)*row[10];
 304    jr->Level = (int)*row[11];
 305    jr->ClientId = str_to_uint64(row[12]!=NULL?row[12]:(char *)"");
 306    sql_free_result(mdb);
 307
 308    db_unlock(mdb);
 309    return 1;
 310 }
 311
 312 /*
 313  * Find VolumeNames for a given JobId
 314  *  Returns: 0 on error or no Volumes found
 315  *           number of volumes on success
 316  *              Volumes are concatenated in VolumeNames
 317  *              separated by a vertical bar (|) in the order
 318  *              that they were written.
 319  *
 320  *  Returns: number of volumes on success
 321  */
 322 int db_get_job_volume_names(JCR *jcr, B_DB *mdb, uint32_t JobId, POOLMEM **VolumeNames)
 323 {
 324    SQL_ROW row;
 325    int stat = 0;
 326    int i;
 327
 328    db_lock(mdb);
 329    /* Get one entry per VolumeName, but "sort" by VolIndex */
 330    Mmsg(&mdb->cmd,
 331         "SELECT VolumeName,MAX(VolIndex) FROM JobMedia,Media WHERE "
 332         "JobMedia.JobId=%u AND JobMedia.MediaId=Media.MediaId "
 333         "GROUP BY VolumeName "
 334         "ORDER BY 2 ASC", JobId);
 335
 336    Dmsg1(130, "VolNam=%s\n", mdb->cmd);
 337    *VolumeNames[0] = 0;
 338    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 339       mdb->num_rows = sql_num_rows(mdb);
 340       Dmsg1(130, "Num rows=%d\n", mdb->num_rows);
 341       if (mdb->num_rows <= 0) {
 342          Mmsg1(&mdb->errmsg, _("No volumes found for JobId=%d\n"), JobId);
 343          stat = 0;
 344       } else {
 345          stat = mdb->num_rows;
 346          for (i=0; i < stat; i++) {
 347             if ((row = sql_fetch_row(mdb)) == NULL) {
 348                Mmsg2(&mdb->errmsg, _("Error fetching row %d: ERR=%s\n"), i, sql_strerror(mdb));
 349                Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 350                stat = 0;
 351                break;
 352             } else {
 353                if (*VolumeNames[0] != 0) {
 354                   pm_strcat(VolumeNames, "|");
 355                }
 356                pm_strcat(VolumeNames, row[0]);
 357             }
 358          }
 359       }
 360       sql_free_result(mdb);
 361    } else {
 362       Mmsg(&mdb->errmsg, _("No Volume for JobId %d found in Catalog.\n"), JobId);
 363    }
 364    db_unlock(mdb);
 365    return stat;
 366 }
 367
 368 /*
 369  * Find Volume parameters for a give JobId
 370  *  Returns: 0 on error or no Volumes found
 371  *           number of volumes on success
 372  *           List of Volumes and start/end file/blocks (malloced structure!)
 373  *
 374  *  Returns: number of volumes on success
 375  */
 376 int db_get_job_volume_parameters(JCR *jcr, B_DB *mdb, uint32_t JobId, VOL_PARAMS **VolParams)
 377 {
 378    SQL_ROW row;
 379    int stat = 0;
 380    int i;
 381    VOL_PARAMS *Vols = NULL;
 382
 383    db_lock(mdb);
 384    Mmsg(&mdb->cmd,
 385 "SELECT VolumeName,FirstIndex,LastIndex,StartFile,EndFile,StartBlock,EndBlock"
 386 " FROM JobMedia,Media WHERE JobMedia.JobId=%u"
 387 " AND JobMedia.MediaId=Media.MediaId ORDER BY VolIndex,JobMediaId", JobId);
 388
 389    Dmsg1(130, "VolNam=%s\n", mdb->cmd);
 390    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 391       mdb->num_rows = sql_num_rows(mdb);
 392       Dmsg1(130, "Num rows=%d\n", mdb->num_rows);
 393       if (mdb->num_rows <= 0) {
 394          Mmsg1(&mdb->errmsg, _("No volumes found for JobId=%d\n"), JobId);
 395          stat = 0;
 396       } else {
 397          stat = mdb->num_rows;
 398          if (stat > 0) {
 399             *VolParams = Vols = (VOL_PARAMS *)malloc(stat * sizeof(VOL_PARAMS));
 400          }
 401          for (i=0; i < stat; i++) {
 402             if ((row = sql_fetch_row(mdb)) == NULL) {
 403                Mmsg2(&mdb->errmsg, _("Error fetching row %d: ERR=%s\n"), i, sql_strerror(mdb));
 404                Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 405                stat = 0;
 406                break;
 407             } else {
 408                bstrncpy(Vols[i].VolumeName, row[0], MAX_NAME_LENGTH);
 409                Vols[i].FirstIndex = atoi(row[1]);
 410                Vols[i].LastIndex = atoi(row[2]);
 411                Vols[i].StartFile = atoi(row[3]);
 412                Vols[i].EndFile = atoi(row[4]);
 413                Vols[i].StartBlock = atoi(row[5]);
 414                Vols[i].EndBlock = atoi(row[6]);
 415             }
 416          }
 417       }
 418       sql_free_result(mdb);
 419    }
 420    db_unlock(mdb);
 421    return stat;
 422 }
 423
 424
 425
 426 /*
 427  * Get the number of pool records
 428  *
 429  * Returns: -1 on failure
 430  *          number on success
 431  */
 432 int db_get_num_pool_records(JCR *jcr, B_DB *mdb)
 433 {
 434    int stat = 0;
 435
 436    db_lock(mdb);
 437    Mmsg(&mdb->cmd, "SELECT count(*) from Pool");
 438    stat = get_sql_record_max(jcr, mdb);
 439    db_unlock(mdb);
 440    return stat;
 441 }
 442
 443 /*
 444  * This function returns a list of all the Pool record ids.
 445  *  The caller must free ids if non-NULL.
 446  *
 447  *  Returns 0: on failure
 448  *          1: on success
 449  */
 450 int db_get_pool_ids(JCR *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 451 {
 452    SQL_ROW row;
 453    int stat = 0;
 454    int i = 0;
 455    uint32_t *id;
 456
 457    db_lock(mdb);
 458    *ids = NULL;
 459    Mmsg(&mdb->cmd, "SELECT PoolId FROM Pool");
 460    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 461       *num_ids = sql_num_rows(mdb);
 462       if (*num_ids > 0) {
 463          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 464          while ((row = sql_fetch_row(mdb)) != NULL) {
 465             id[i++] = str_to_uint64(row[0]);
 466          }
 467          *ids = id;
 468       }
 469       sql_free_result(mdb);
 470       stat = 1;
 471    } else {
 472       Mmsg(&mdb->errmsg, _("Pool id select failed: ERR=%s\n"), sql_strerror(mdb));
 473       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 474       stat = 0;
 475    }
 476    db_unlock(mdb);
 477    return stat;
 478 }
 479
 480 /*
 481  * This function returns a list of all the Client record ids.
 482  *  The caller must free ids if non-NULL.
 483  *
 484  *  Returns 0: on failure
 485  *          1: on success
 486  */
 487 int db_get_client_ids(JCR *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 488 {
 489    SQL_ROW row;
 490    int stat = 0;
 491    int i = 0;
 492    uint32_t *id;
 493
 494    db_lock(mdb);
 495    *ids = NULL;
 496    Mmsg(&mdb->cmd, "SELECT ClientId FROM Client");
 497    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 498       *num_ids = sql_num_rows(mdb);
 499       if (*num_ids > 0) {
 500          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 501          while ((row = sql_fetch_row(mdb)) != NULL) {
 502             id[i++] = (uint32_t)atoi(row[0]);
 503          }
 504          *ids = id;
 505       }
 506       sql_free_result(mdb);
 507       stat = 1;
 508    } else {
 509       Mmsg(&mdb->errmsg, _("Client id select failed: ERR=%s\n"), sql_strerror(mdb));
 510       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 511       stat = 0;
 512    }
 513    db_unlock(mdb);
 514    return stat;
 515 }
 516
 517
 518
 519 /* Get Pool Record
 520  * If the PoolId is non-zero, we get its record,
 521  *  otherwise, we search on the PoolName
 522  *
 523  * Returns: 0 on failure
 524  *          id on success
 525  */
 526 int db_get_pool_record(JCR *jcr, B_DB *mdb, POOL_DBR *pdbr)
 527 {
 528    SQL_ROW row;
 529    int stat = 0;
 530
 531    db_lock(mdb);
 532    if (pdbr->PoolId != 0) {               /* find by id */
 533       Mmsg(&mdb->cmd,
 534 "SELECT PoolId,Name,NumVols,MaxVols,UseOnce,UseCatalog,AcceptAnyVolume,\
 535 AutoPrune,Recycle,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 536 MaxVolBytes,PoolType,LabelFormat FROM Pool WHERE Pool.PoolId=%u", pdbr->PoolId);
 537    } else {                           /* find by name */
 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.Name='%s'", pdbr->Name);
 542    }
 543
 544    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 545       mdb->num_rows = sql_num_rows(mdb);
 546       if (mdb->num_rows > 1) {
 547          char ed1[30];
 548          Mmsg1(&mdb->errmsg, _("More than one Pool!: %s\n"),
 549             edit_uint64(mdb->num_rows, ed1));
 550          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 551       } else if (mdb->num_rows == 1) {
 552          if ((row = sql_fetch_row(mdb)) == NULL) {
 553             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 554             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 555          } else {
 556             pdbr->PoolId = str_to_int64(row[0]);
 557             bstrncpy(pdbr->Name, row[1]!=NULL?row[1]:"", sizeof(pdbr->Name));
 558             pdbr->NumVols = str_to_int64(row[2]);
 559             pdbr->MaxVols = str_to_int64(row[3]);
 560             pdbr->UseOnce = str_to_int64(row[4]);
 561             pdbr->UseCatalog = str_to_int64(row[5]);
 562             pdbr->AcceptAnyVolume = str_to_int64(row[6]);
 563             pdbr->AutoPrune = str_to_int64(row[7]);
 564             pdbr->Recycle = str_to_int64(row[8]);
 565             pdbr->VolRetention = str_to_int64(row[9]);
 566             pdbr->VolUseDuration = str_to_int64(row[10]);
 567             pdbr->MaxVolJobs = str_to_int64(row[11]);
 568             pdbr->MaxVolFiles = str_to_int64(row[12]);
 569             pdbr->MaxVolBytes = str_to_uint64(row[13]);
 570             bstrncpy(pdbr->PoolType, row[13]!=NULL?row[14]:"", sizeof(pdbr->PoolType));
 571             bstrncpy(pdbr->LabelFormat, row[14]!=NULL?row[15]:"", sizeof(pdbr->LabelFormat));
 572             stat = pdbr->PoolId;
 573          }
 574       } else {
 575          Mmsg(&mdb->errmsg, _("Pool record not found in Catalog.\n"));
 576       }
 577       sql_free_result(mdb);
 578    } else {
 579       Mmsg(&mdb->errmsg, _("Pool record not found in Catalog.\n"));
 580    }
 581    db_unlock(mdb);
 582    return stat;
 583 }
 584
 585 /* Get Client Record
 586  * If the ClientId is non-zero, we get its record,
 587  *  otherwise, we search on the Client Name
 588  *
 589  * Returns: 0 on failure
 590  *          1 on success
 591  */
 592 int db_get_client_record(JCR *jcr, B_DB *mdb, CLIENT_DBR *cdbr)
 593 {
 594    SQL_ROW row;
 595    int stat = 0;
 596
 597    db_lock(mdb);
 598    if (cdbr->ClientId != 0) {               /* find by id */
 599       Mmsg(&mdb->cmd,
 600 "SELECT ClientId,Name,Uname,AutoPrune,FileRetention,JobRetention "
 601 "FROM Client WHERE Client.ClientId=%u", cdbr->ClientId);
 602    } else {                           /* find by name */
 603       Mmsg(&mdb->cmd,
 604 "SELECT ClientId,Name,Uname,AutoPrune,FileRetention,JobRetention "
 605 "FROM Client WHERE Client.Name='%s'", cdbr->Name);
 606    }
 607
 608    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 609       mdb->num_rows = sql_num_rows(mdb);
 610       if (mdb->num_rows > 1) {
 611          char ed1[30];
 612          Mmsg1(&mdb->errmsg, _("More than one Client!: %s\n"),
 613             edit_uint64(mdb->num_rows, ed1));
 614          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 615       } else if (mdb->num_rows == 1) {
 616          if ((row = sql_fetch_row(mdb)) == NULL) {
 617             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 618             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 619          } else {
 620             cdbr->ClientId = str_to_int64(row[0]);
 621             bstrncpy(cdbr->Name, row[1]!=NULL?row[1]:"", sizeof(cdbr->Name));
 622             bstrncpy(cdbr->Uname, row[2]!=NULL?row[1]:"", sizeof(cdbr->Uname));
 623             cdbr->AutoPrune = str_to_int64(row[3]);
 624             cdbr->FileRetention = str_to_int64(row[4]);
 625             cdbr->JobRetention = str_to_int64(row[5]);
 626             stat = 1;
 627          }
 628       } else {
 629          Mmsg(&mdb->errmsg, _("Client record not found in Catalog.\n"));
 630       }
 631       sql_free_result(mdb);
 632    } else {
 633       Mmsg(&mdb->errmsg, _("Client record not found in Catalog.\n"));
 634    }
 635    db_unlock(mdb);
 636    return stat;
 637 }
 638
 639 /*
 640  * Get Counter Record
 641  *
 642  * Returns: 0 on failure
 643  *          1 on success
 644  */
 645 int db_get_counter_record(JCR *jcr, B_DB *mdb, COUNTER_DBR *cr)
 646 {
 647    SQL_ROW row;
 648
 649    db_lock(mdb);
 650    Mmsg(&mdb->cmd, "SELECT MinValue,MaxValue,CurrentValue,WrapCounter "
 651       "FROM Counters WHERE Counter='%s'", cr->Counter);
 652
 653    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 654
 655       mdb->num_rows = sql_num_rows(mdb);
 656
 657       /* If more than one, report error, but return first row */
 658       if (mdb->num_rows > 1) {
 659          Mmsg1(&mdb->errmsg, _("More than one Counter!: %d\n"), (int)(mdb->num_rows));
 660          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 661       }
 662       if (mdb->num_rows >= 1) {
 663          if ((row = sql_fetch_row(mdb)) == NULL) {
 664             Mmsg1(&mdb->errmsg, _("error fetching Counter row: %s\n"), sql_strerror(mdb));
 665             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 666             sql_free_result(mdb);
 667             db_unlock(mdb);
 668             return 0;
 669          }
 670          cr->MinValue = atoi(row[0]);
 671          cr->MaxValue = atoi(row[1]);
 672          cr->CurrentValue = atoi(row[2]);
 673          if (row[3]) {
 674             bstrncpy(cr->WrapCounter, row[3], sizeof(cr->WrapCounter));
 675          } else {
 676             cr->WrapCounter[0] = 0;
 677          }
 678          sql_free_result(mdb);
 679          db_unlock(mdb);
 680          return 1;
 681       }
 682       sql_free_result(mdb);
 683    } else {
 684       Mmsg(&mdb->errmsg, _("Counter record: %s not found in Catalog.\n"), cr->Counter);
 685    }
 686    db_unlock(mdb);
 687    return 0;
 688 }
 689
 690
 691 /* Get FileSet Record
 692  * If the FileSetId is non-zero, we get its record,
 693  *  otherwise, we search on the name
 694  *
 695  * Returns: 0 on failure
 696  *          id on success
 697  */
 698 int db_get_fileset_record(JCR *jcr, B_DB *mdb, FILESET_DBR *fsr)
 699 {
 700    SQL_ROW row;
 701    int stat = 0;
 702
 703    db_lock(mdb);
 704    if (fsr->FileSetId != 0) {               /* find by id */
 705       Mmsg(&mdb->cmd,
 706            "SELECT FileSetId,FileSet,MD5,CreateTime FROM FileSet "
 707            "WHERE FileSetId=%u", fsr->FileSetId);
 708    } else {                           /* find by name */
 709       Mmsg(&mdb->cmd,
 710            "SELECT FileSetId,FileSet,CreateTime,MD5 FROM FileSet "
 711            "WHERE FileSet='%s'", fsr->FileSet);
 712    }
 713
 714    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 715       mdb->num_rows = sql_num_rows(mdb);
 716       if (mdb->num_rows > 1) {
 717          char ed1[30];
 718          Mmsg1(&mdb->errmsg, _("Error got %s FileSets but expected only one!\n"),
 719             edit_uint64(mdb->num_rows, ed1));
 720          sql_data_seek(mdb, mdb->num_rows-1);
 721       }
 722       if ((row = sql_fetch_row(mdb)) == NULL) {
 723          Mmsg1(&mdb->errmsg, _("FileSet record \"%s\" not found.\n"), fsr->FileSet);
 724       } else {
 725          fsr->FileSetId = atoi(row[0]);
 726          bstrncpy(fsr->FileSet, row[1]!=NULL?row[1]:"", sizeof(fsr->FileSet));
 727          bstrncpy(fsr->MD5, row[2]!=NULL?row[2]:"", sizeof(fsr->MD5));
 728          bstrncpy(fsr->cCreateTime, row[3]!=NULL?row[3]:"", sizeof(fsr->cCreateTime));
 729          stat = fsr->FileSetId;
 730       }
 731       sql_free_result(mdb);
 732    } else {
 733       Mmsg(&mdb->errmsg, _("FileSet record not found in Catalog.\n"));
 734    }
 735    db_unlock(mdb);
 736    return stat;
 737 }
 738
 739
 740 /*
 741  * Get the number of Media records
 742  *
 743  * Returns: -1 on failure
 744  *          number on success
 745  */
 746 int db_get_num_media_records(JCR *jcr, B_DB *mdb)
 747 {
 748    int stat = 0;
 749
 750    db_lock(mdb);
 751    Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 752    stat = get_sql_record_max(jcr, mdb);
 753    db_unlock(mdb);
 754    return stat;
 755 }
 756
 757
 758 /*
 759  * This function returns a list of all the Media record ids for
 760  *     the current Pool.
 761  *  The caller must free ids if non-NULL.
 762  *
 763  *  Returns 0: on failure
 764  *          1: on success
 765  */
 766 int db_get_media_ids(JCR *jcr, B_DB *mdb, uint32_t PoolId, int *num_ids, uint32_t *ids[])
 767 {
 768    SQL_ROW row;
 769    int stat = 0;
 770    int i = 0;
 771    uint32_t *id;
 772
 773    db_lock(mdb);
 774    *ids = NULL;
 775    Mmsg(&mdb->cmd, "SELECT MediaId FROM Media WHERE PoolId=%u", PoolId);
 776    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 777       *num_ids = sql_num_rows(mdb);
 778       if (*num_ids > 0) {
 779          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 780          while ((row = sql_fetch_row(mdb)) != NULL) {
 781             id[i++] = (uint32_t)atoi(row[0]);
 782          }
 783          *ids = id;
 784       }
 785       sql_free_result(mdb);
 786       stat = 1;
 787    } else {
 788       Mmsg(&mdb->errmsg, _("Media id select failed: ERR=%s\n"), sql_strerror(mdb));
 789       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 790       stat = 0;
 791    }
 792    db_unlock(mdb);
 793    return stat;
 794 }
 795
 796
 797 /* Get Media Record
 798  *
 799  * Returns: 0 on failure
 800  *          id on success
 801  */
 802 int db_get_media_record(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
 803 {
 804    SQL_ROW row;
 805    int stat = 0;
 806
 807    db_lock(mdb);
 808    if (mr->MediaId == 0 && mr->VolumeName[0] == 0) {
 809       Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 810       mr->MediaId = get_sql_record_max(jcr, mdb);
 811       db_unlock(mdb);
 812       return 1;
 813    }
 814    if (mr->MediaId != 0) {               /* find by id */
 815       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,"
 816          "VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,"
 817          "MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,"
 818          "Recycle,Slot,FirstWritten,LastWritten,InChanger "
 819          "FROM Media WHERE MediaId=%d", mr->MediaId);
 820    } else {                           /* find by name */
 821       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,"
 822          "VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,"
 823          "MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,"
 824          "Recycle,Slot,FirstWritten,LastWritten,InChanger "
 825          "FROM Media WHERE VolumeName='%s'", mr->VolumeName);
 826    }
 827
 828    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 829       mdb->num_rows = sql_num_rows(mdb);
 830       if (mdb->num_rows > 1) {
 831          char ed1[30];
 832          Mmsg1(&mdb->errmsg, _("More than one Volume!: %s\n"),
 833             edit_uint64(mdb->num_rows, ed1));
 834          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 835       } else if (mdb->num_rows == 1) {
 836          if ((row = sql_fetch_row(mdb)) == NULL) {
 837             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 838             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 839          } else {
 840             /* return values */
 841             mr->MediaId = str_to_int64(row[0]);
 842             bstrncpy(mr->VolumeName, row[1]!=NULL?row[1]:"", sizeof(mr->VolumeName));
 843             mr->VolJobs = str_to_int64(row[2]);
 844             mr->VolFiles = str_to_int64(row[3]);
 845             mr->VolBlocks = str_to_int64(row[4]);
 846             mr->VolBytes = str_to_uint64(row[5]);
 847             mr->VolMounts = str_to_int64(row[6]);
 848             mr->VolErrors = str_to_int64(row[7]);
 849             mr->VolWrites = str_to_int64(row[8]);
 850             mr->MaxVolBytes = str_to_uint64(row[9]);
 851             mr->VolCapacityBytes = str_to_uint64(row[10]);
 852             bstrncpy(mr->MediaType, row[11]!=NULL?row[11]:"", sizeof(mr->MediaType));
 853             bstrncpy(mr->VolStatus, row[12]!=NULL?row[12]:"", sizeof(mr->VolStatus));
 854             mr->PoolId = str_to_int64(row[13]);
 855             mr->VolRetention = str_to_uint64(row[14]);
 856             mr->VolUseDuration = str_to_uint64(row[15]);
 857             mr->MaxVolJobs = str_to_int64(row[16]);
 858             mr->MaxVolFiles = str_to_int64(row[17]);
 859             mr->Recycle = str_to_int64(row[18]);
 860             mr->Slot = str_to_int64(row[19]);
 861             bstrncpy(mr->cFirstWritten, row[20]!=NULL?row[20]:"", sizeof(mr->cFirstWritten));
 862             mr->FirstWritten = (time_t)str_to_utime(mr->cFirstWritten);
 863             bstrncpy(mr->cLastWritten, row[21]!=NULL?row[21]:"", sizeof(mr->cLastWritten));
 864             mr->LastWritten = (time_t)str_to_utime(mr->cLastWritten);
 865             mr->InChanger = str_to_uint64(row[22]);
 866             stat = mr->MediaId;
 867          }
 868       } else {
 869          if (mr->MediaId != 0) {
 870             Mmsg1(&mdb->errmsg, _("Media record MediaId=%u not found.\n"), mr->MediaId);
 871          } else {
 872             Mmsg1(&mdb->errmsg, _("Media record for Volume \"%s\" not found.\n"),
 873                   mr->VolumeName);
 874          }
 875       }
 876       sql_free_result(mdb);
 877    } else {
 878       Mmsg(&mdb->errmsg, _("Media record not found in Catalog.\n"));
 879    }
 880    db_unlock(mdb);
 881    return stat;
 882 }
 883
 884
 885 #endif /* HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL */