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(void *jcr, B_DB *mdb, FILE_DBR *fdbr);
  52 static int db_get_filename_record(void *jcr, B_DB *mdb);
  53 static int db_get_path_record(void *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, void *jcr, B_DB *db, char *select_cmd);
  59 extern void split_path_and_filename(void *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(void *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(void *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->MD5, row[2], sizeof(fdbr->MD5));
 132             stat = 1;
 133          }
 134       } else {
 135          Mmsg2(&mdb->errmsg, _("File record not found for PathId=%u FilenameId=%u\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(void *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(void *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(void *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 = atoi(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(void *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(void *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,EndBlock"
 365 " FROM JobMedia,Media WHERE JobMedia.JobId=%u"
 366 " AND JobMedia.MediaId=Media.MediaId", JobId);
 367
 368    Dmsg1(130, "VolNam=%s\n", mdb->cmd);
 369    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 370       mdb->num_rows = sql_num_rows(mdb);
 371       Dmsg1(130, "Num rows=%d\n", mdb->num_rows);
 372       if (mdb->num_rows <= 0) {
 373          Mmsg1(&mdb->errmsg, _("No volumes found for JobId=%d\n"), JobId);
 374          stat = 0;
 375       } else {
 376          stat = mdb->num_rows;
 377          if (stat > 0) {
 378             *VolParams = Vols = (VOL_PARAMS *)malloc(stat * sizeof(VOL_PARAMS));
 379          }
 380          for (i=0; i < stat; i++) {
 381             if ((row = sql_fetch_row(mdb)) == NULL) {
 382                Mmsg2(&mdb->errmsg, _("Error fetching row %d: ERR=%s\n"), i, sql_strerror(mdb));
 383                Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 384                stat = 0;
 385                break;
 386             } else {
 387                bstrncpy(Vols[i].VolumeName, row[0], MAX_NAME_LENGTH);
 388                Vols[i].FirstIndex = atoi(row[1]);
 389                Vols[i].LastIndex = atoi(row[2]);
 390                Vols[i].StartFile = atoi(row[3]);
 391                Vols[i].EndFile = atoi(row[4]);
 392                Vols[i].StartBlock = atoi(row[5]);
 393                Vols[i].EndBlock = atoi(row[6]);
 394             }
 395          }
 396       }
 397       sql_free_result(mdb);
 398    }
 399    db_unlock(mdb);
 400    return stat;
 401 }
 402
 403
 404
 405 /*
 406  * Get the number of pool records
 407  *
 408  * Returns: -1 on failure
 409  *          number on success
 410  */
 411 int db_get_num_pool_records(void *jcr, B_DB *mdb)
 412 {
 413    int stat = 0;
 414
 415    db_lock(mdb);
 416    Mmsg(&mdb->cmd, "SELECT count(*) from Pool");
 417    stat = get_sql_record_max(jcr, mdb);
 418    db_unlock(mdb);
 419    return stat;
 420 }
 421
 422 /*
 423  * This function returns a list of all the Pool record ids.
 424  *  The caller must free ids if non-NULL.
 425  *
 426  *  Returns 0: on failure
 427  *          1: on success
 428  */
 429 int db_get_pool_ids(void *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 430 {
 431    SQL_ROW row;
 432    int stat = 0;
 433    int i = 0;
 434    uint32_t *id;
 435
 436    db_lock(mdb);
 437    *ids = NULL;
 438    Mmsg(&mdb->cmd, "SELECT PoolId FROM Pool");
 439    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 440       *num_ids = sql_num_rows(mdb);
 441       if (*num_ids > 0) {
 442          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 443          while ((row = sql_fetch_row(mdb)) != NULL) {
 444             id[i++] = (uint32_t)atoi(row[0]);
 445          }
 446          *ids = id;
 447       }
 448       sql_free_result(mdb);
 449       stat = 1;
 450    } else {
 451       Mmsg(&mdb->errmsg, _("Pool id select failed: ERR=%s\n"), sql_strerror(mdb));
 452       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 453       stat = 0;
 454    }
 455    db_unlock(mdb);
 456    return stat;
 457 }
 458
 459 /*
 460  * This function returns a list of all the Client record ids.
 461  *  The caller must free ids if non-NULL.
 462  *
 463  *  Returns 0: on failure
 464  *          1: on success
 465  */
 466 int db_get_client_ids(void *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 467 {
 468    SQL_ROW row;
 469    int stat = 0;
 470    int i = 0;
 471    uint32_t *id;
 472
 473    db_lock(mdb);
 474    *ids = NULL;
 475    Mmsg(&mdb->cmd, "SELECT ClientId FROM Client");
 476    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 477       *num_ids = sql_num_rows(mdb);
 478       if (*num_ids > 0) {
 479          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 480          while ((row = sql_fetch_row(mdb)) != NULL) {
 481             id[i++] = (uint32_t)atoi(row[0]);
 482          }
 483          *ids = id;
 484       }
 485       sql_free_result(mdb);
 486       stat = 1;
 487    } else {
 488       Mmsg(&mdb->errmsg, _("Client id select failed: ERR=%s\n"), sql_strerror(mdb));
 489       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 490       stat = 0;
 491    }
 492    db_unlock(mdb);
 493    return stat;
 494 }
 495
 496
 497
 498 /* Get Pool Record
 499  * If the PoolId is non-zero, we get its record,
 500  *  otherwise, we search on the PoolName
 501  *
 502  * Returns: 0 on failure
 503  *          id on success
 504  */
 505 int db_get_pool_record(void *jcr, B_DB *mdb, POOL_DBR *pdbr)
 506 {
 507    SQL_ROW row;
 508    int stat = 0;
 509
 510    db_lock(mdb);
 511    if (pdbr->PoolId != 0) {               /* find by id */
 512       Mmsg(&mdb->cmd,
 513 "SELECT PoolId,Name,NumVols,MaxVols,UseOnce,UseCatalog,AcceptAnyVolume,\
 514 AutoPrune,Recycle,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 515 MaxVolBytes,PoolType,LabelFormat FROM Pool WHERE Pool.PoolId=%u", pdbr->PoolId);
 516    } else {                           /* find by name */
 517       Mmsg(&mdb->cmd,
 518 "SELECT PoolId,Name,NumVols,MaxVols,UseOnce,UseCatalog,AcceptAnyVolume,\
 519 AutoPrune,Recycle,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 520 MaxVolBytes,PoolType,LabelFormat FROM Pool WHERE Pool.Name='%s'", pdbr->Name);
 521    }
 522
 523    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 524       mdb->num_rows = sql_num_rows(mdb);
 525       if (mdb->num_rows > 1) {
 526          char ed1[30];
 527          Mmsg1(&mdb->errmsg, _("More than one Pool!: %s\n"),
 528             edit_uint64(mdb->num_rows, ed1));
 529          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 530       } else if (mdb->num_rows == 1) {
 531          if ((row = sql_fetch_row(mdb)) == NULL) {
 532             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 533             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 534          } else {
 535             pdbr->PoolId = str_to_int64(row[0]);
 536             bstrncpy(pdbr->Name, row[1]!=NULL?row[1]:"", sizeof(pdbr->Name));
 537             pdbr->NumVols = str_to_int64(row[2]);
 538             pdbr->MaxVols = str_to_int64(row[3]);
 539             pdbr->UseOnce = str_to_int64(row[4]);
 540             pdbr->UseCatalog = str_to_int64(row[5]);
 541             pdbr->AcceptAnyVolume = str_to_int64(row[6]);
 542             pdbr->AutoPrune = str_to_int64(row[7]);
 543             pdbr->Recycle = str_to_int64(row[8]);
 544             pdbr->VolRetention = str_to_int64(row[9]);
 545             pdbr->VolUseDuration = str_to_int64(row[10]);
 546             pdbr->MaxVolJobs = str_to_int64(row[11]);
 547             pdbr->MaxVolFiles = str_to_int64(row[12]);
 548             pdbr->MaxVolBytes = str_to_uint64(row[13]);
 549             bstrncpy(pdbr->PoolType, row[13]!=NULL?row[14]:"", sizeof(pdbr->PoolType));
 550             bstrncpy(pdbr->LabelFormat, row[14]!=NULL?row[15]:"", sizeof(pdbr->LabelFormat));
 551             stat = pdbr->PoolId;
 552          }
 553       }
 554       sql_free_result(mdb);
 555    }
 556    db_unlock(mdb);
 557    return stat;
 558 }
 559
 560 /* Get Client Record
 561  * If the ClientId is non-zero, we get its record,
 562  *  otherwise, we search on the Client Name
 563  *
 564  * Returns: 0 on failure
 565  *          1 on success
 566  */
 567 int db_get_client_record(void *jcr, B_DB *mdb, CLIENT_DBR *cdbr)
 568 {
 569    SQL_ROW row;
 570    int stat = 0;
 571
 572    db_lock(mdb);
 573    if (cdbr->ClientId != 0) {               /* find by id */
 574       Mmsg(&mdb->cmd,
 575 "SELECT ClientId,Name,Uname,AutoPrune,FileRetention,JobRetention "
 576 "FROM Client WHERE Client.ClientId=%u", cdbr->ClientId);
 577    } else {                           /* find by name */
 578       Mmsg(&mdb->cmd,
 579 "SELECT ClientId,Name,Uname,AutoPrune,FileRetention,JobRetention "
 580 "FROM Client WHERE Client.Name='%s'", cdbr->Name);
 581    }
 582
 583    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 584       mdb->num_rows = sql_num_rows(mdb);
 585       if (mdb->num_rows > 1) {
 586          char ed1[30];
 587          Mmsg1(&mdb->errmsg, _("More than one Client!: %s\n"),
 588             edit_uint64(mdb->num_rows, ed1));
 589          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 590       } else if (mdb->num_rows == 1) {
 591          if ((row = sql_fetch_row(mdb)) == NULL) {
 592             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 593             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 594          } else {
 595             cdbr->ClientId = str_to_int64(row[0]);
 596             bstrncpy(cdbr->Name, row[1]!=NULL?row[1]:"", sizeof(cdbr->Name));
 597             bstrncpy(cdbr->Uname, row[2]!=NULL?row[1]:"", sizeof(cdbr->Uname));
 598             cdbr->AutoPrune = str_to_int64(row[3]);
 599             cdbr->FileRetention = str_to_int64(row[4]);
 600             cdbr->JobRetention = str_to_int64(row[5]);
 601             stat = 1;
 602          }
 603       }
 604       sql_free_result(mdb);
 605    }
 606    db_unlock(mdb);
 607    return stat;
 608 }
 609
 610
 611 /* Get FileSet Record
 612  * If the FileSetId is non-zero, we get its record,
 613  *  otherwise, we search on the name
 614  *
 615  * Returns: 0 on failure
 616  *          id on success
 617  */
 618 int db_get_fileset_record(void *jcr, B_DB *mdb, FILESET_DBR *fsr)
 619 {
 620    SQL_ROW row;
 621    int stat = 0;
 622
 623    db_lock(mdb);
 624    if (fsr->FileSetId != 0) {               /* find by id */
 625       Mmsg(&mdb->cmd,
 626            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 627            "WHERE FileSetId=%u", fsr->FileSetId);
 628    } else {                           /* find by name */
 629       Mmsg(&mdb->cmd,
 630            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 631            "WHERE FileSet='%s'", fsr->FileSet);
 632    }
 633
 634    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 635       mdb->num_rows = sql_num_rows(mdb);
 636       if (mdb->num_rows > 1) {
 637          char ed1[30];
 638          Mmsg1(&mdb->errmsg, _("Error got %s FileSets but expected only one!\n"),
 639             edit_uint64(mdb->num_rows, ed1));
 640          sql_data_seek(mdb, mdb->num_rows-1);
 641       }
 642       if ((row = sql_fetch_row(mdb)) == NULL) {
 643          Mmsg1(&mdb->errmsg, _("Error: FileSet record \"%s\" not found\n"), fsr->FileSet);
 644       } else {
 645          fsr->FileSetId = atoi(row[0]);
 646          bstrncpy(fsr->FileSet, row[1]!=NULL?row[1]:"", sizeof(fsr->FileSet));
 647          bstrncpy(fsr->MD5, row[2]!=NULL?row[2]:"", sizeof(fsr->MD5));
 648          stat = fsr->FileSetId;
 649       }
 650       sql_free_result(mdb);
 651    }
 652    db_unlock(mdb);
 653    return stat;
 654 }
 655
 656
 657 /*
 658  * Get the number of Media records
 659  *
 660  * Returns: -1 on failure
 661  *          number on success
 662  */
 663 int db_get_num_media_records(void *jcr, B_DB *mdb)
 664 {
 665    int stat = 0;
 666
 667    db_lock(mdb);
 668    Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 669    stat = get_sql_record_max(jcr, mdb);
 670    db_unlock(mdb);
 671    return stat;
 672 }
 673
 674
 675 /*
 676  * This function returns a list of all the Media record ids.
 677  *  The caller must free ids if non-NULL.
 678  *
 679  *  Returns 0: on failure
 680  *          1: on success
 681  */
 682 int db_get_media_ids(void *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 683 {
 684    SQL_ROW row;
 685    int stat = 0;
 686    int i = 0;
 687    uint32_t *id;
 688
 689    db_lock(mdb);
 690    *ids = NULL;
 691    Mmsg(&mdb->cmd, "SELECT MediaId FROM Media");
 692    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 693       *num_ids = sql_num_rows(mdb);
 694       if (*num_ids > 0) {
 695          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 696          while ((row = sql_fetch_row(mdb)) != NULL) {
 697             id[i++] = (uint32_t)atoi(row[0]);
 698          }
 699          *ids = id;
 700       }
 701       sql_free_result(mdb);
 702       stat = 1;
 703    } else {
 704       Mmsg(&mdb->errmsg, _("Media id select failed: ERR=%s\n"), sql_strerror(mdb));
 705       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 706       stat = 0;
 707    }
 708    db_unlock(mdb);
 709    return stat;
 710 }
 711
 712
 713 /* Get Media Record
 714  *
 715  * Returns: 0 on failure
 716  *          id on success
 717  */
 718 int db_get_media_record(void *jcr, B_DB *mdb, MEDIA_DBR *mr)
 719 {
 720    SQL_ROW row;
 721    int stat = 0;
 722
 723    db_lock(mdb);
 724    if (mr->MediaId == 0 && mr->VolumeName[0] == 0) {
 725       Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 726       mr->MediaId = get_sql_record_max(jcr, mdb);
 727       db_unlock(mdb);
 728       return 1;
 729    }
 730    if (mr->MediaId != 0) {               /* find by id */
 731       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 732 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 733 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 734 Recycle,Slot,FirstWritten,LastWritten \
 735 FROM Media WHERE MediaId=%d", mr->MediaId);
 736    } else {                           /* find by name */
 737       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 738 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 739 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 740 Recycle,Slot,FirstWritten,LastWritten \
 741 FROM Media WHERE VolumeName='%s'", mr->VolumeName);
 742    }
 743
 744    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 745       mdb->num_rows = sql_num_rows(mdb);
 746       if (mdb->num_rows > 1) {
 747          char ed1[30];
 748          Mmsg1(&mdb->errmsg, _("More than one Volume!: %s\n"),
 749             edit_uint64(mdb->num_rows, ed1));
 750          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 751       } else if (mdb->num_rows == 1) {
 752          if ((row = sql_fetch_row(mdb)) == NULL) {
 753             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 754             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 755          } else {
 756             /* return values */
 757             mr->MediaId = str_to_int64(row[0]);
 758             bstrncpy(mr->VolumeName, row[1]!=NULL?row[1]:"", sizeof(mr->VolumeName));
 759             mr->VolJobs = str_to_int64(row[2]);
 760             mr->VolFiles = str_to_int64(row[3]);
 761             mr->VolBlocks = str_to_int64(row[4]);
 762             mr->VolBytes = str_to_uint64(row[5]);
 763             mr->VolMounts = str_to_int64(row[6]);
 764             mr->VolErrors = str_to_int64(row[7]);
 765             mr->VolWrites = str_to_int64(row[8]);
 766             mr->MaxVolBytes = str_to_uint64(row[9]);
 767             mr->VolCapacityBytes = str_to_uint64(row[10]);
 768             bstrncpy(mr->MediaType, row[11]!=NULL?row[11]:"", sizeof(mr->MediaType));
 769             bstrncpy(mr->VolStatus, row[12]!=NULL?row[12]:"", sizeof(mr->VolStatus));
 770             mr->PoolId = str_to_int64(row[13]);
 771             mr->VolRetention = str_to_uint64(row[14]);
 772             mr->VolUseDuration = str_to_uint64(row[15]);
 773             mr->MaxVolJobs = str_to_int64(row[16]);
 774             mr->MaxVolFiles = str_to_int64(row[17]);
 775             mr->Recycle = str_to_int64(row[18]);
 776             mr->Slot = str_to_int64(row[19]);
 777             bstrncpy(mr->cFirstWritten, row[20]!=NULL?row[20]:"", sizeof(mr->cFirstWritten));
 778             mr->FirstWritten = (time_t)str_to_utime(mr->cFirstWritten);
 779             bstrncpy(mr->cLastWritten, row[21]!=NULL?row[21]:"", sizeof(mr->cLastWritten));
 780             mr->LastWritten = (time_t)str_to_utime(mr->cLastWritten);
 781             stat = mr->MediaId;
 782          }
 783       } else {
 784          Mmsg0(&mdb->errmsg, _("Media record not found.\n"));
 785       }
 786       sql_free_result(mdb);
 787    }
 788    db_unlock(mdb);
 789    return stat;
 790 }
 791
 792
 793 #endif /* HAVE_MYSQL || HAVE_SQLITE */