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