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