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