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,EndBlock"
 365 " FROM JobMedia,Media WHERE JobMedia.JobId=%u"
 366 " AND JobMedia.MediaId=Media.MediaId ORDER BY VolIndex,JobMediaId", 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(JCR *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(JCR *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++] = str_to_uint64(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(JCR *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(JCR *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(JCR *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 Counter Record
 612  *
 613  * Returns: 0 on failure
 614  *          1 on success
 615  */
 616 int db_get_counter_record(JCR *jcr, B_DB *mdb, COUNTER_DBR *cr)
 617 {
 618    SQL_ROW row;
 619
 620    db_lock(mdb);
 621    Mmsg(&mdb->cmd, "SELECT MinValue,MaxValue,CurrentValue,WrapCounter "
 622       "FROM Counters WHERE Counter='%s'", cr->Counter);
 623
 624    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 625
 626       mdb->num_rows = sql_num_rows(mdb);
 627
 628       /* If more than one, report error, but return first row */
 629       if (mdb->num_rows > 1) {
 630          Mmsg1(&mdb->errmsg, _("More than one Counter!: %d\n"), (int)(mdb->num_rows));
 631          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 632       }
 633       if (mdb->num_rows >= 1) {
 634          if ((row = sql_fetch_row(mdb)) == NULL) {
 635             Mmsg1(&mdb->errmsg, _("error fetching Counter row: %s\n"), sql_strerror(mdb));
 636             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 637             sql_free_result(mdb);
 638             db_unlock(mdb);
 639             return 0;
 640          }
 641          cr->MinValue = atoi(row[0]);
 642          cr->MaxValue = atoi(row[1]);
 643          cr->CurrentValue = atoi(row[2]);
 644          if (row[3]) {
 645             bstrncpy(cr->WrapCounter, row[3], sizeof(cr->WrapCounter));
 646          } else {
 647             cr->WrapCounter[0] = 0;
 648          }
 649          sql_free_result(mdb);
 650          db_unlock(mdb);
 651          return 1;
 652       }
 653       sql_free_result(mdb);
 654    }
 655    db_unlock(mdb);
 656    return 0;
 657 }
 658
 659
 660 /* Get FileSet Record
 661  * If the FileSetId is non-zero, we get its record,
 662  *  otherwise, we search on the name
 663  *
 664  * Returns: 0 on failure
 665  *          id on success
 666  */
 667 int db_get_fileset_record(JCR *jcr, B_DB *mdb, FILESET_DBR *fsr)
 668 {
 669    SQL_ROW row;
 670    int stat = 0;
 671
 672    db_lock(mdb);
 673    if (fsr->FileSetId != 0) {               /* find by id */
 674       Mmsg(&mdb->cmd,
 675            "SELECT FileSetId,FileSet,MD5,CreateTime FROM FileSet "
 676            "WHERE FileSetId=%u", fsr->FileSetId);
 677    } else {                           /* find by name */
 678       Mmsg(&mdb->cmd,
 679            "SELECT FileSetId,FileSet,CreateTime,MD5 FROM FileSet "
 680            "WHERE FileSet='%s'", fsr->FileSet);
 681    }
 682
 683    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 684       mdb->num_rows = sql_num_rows(mdb);
 685       if (mdb->num_rows > 1) {
 686          char ed1[30];
 687          Mmsg1(&mdb->errmsg, _("Error got %s FileSets but expected only one!\n"),
 688             edit_uint64(mdb->num_rows, ed1));
 689          sql_data_seek(mdb, mdb->num_rows-1);
 690       }
 691       if ((row = sql_fetch_row(mdb)) == NULL) {
 692          Mmsg1(&mdb->errmsg, _("FileSet record \"%s\" not found.\n"), fsr->FileSet);
 693       } else {
 694          fsr->FileSetId = atoi(row[0]);
 695          bstrncpy(fsr->FileSet, row[1]!=NULL?row[1]:"", sizeof(fsr->FileSet));
 696          bstrncpy(fsr->MD5, row[2]!=NULL?row[2]:"", sizeof(fsr->MD5));
 697          bstrncpy(fsr->cCreateTime, row[3]!=NULL?row[3]:"", sizeof(fsr->cCreateTime));
 698          stat = fsr->FileSetId;
 699       }
 700       sql_free_result(mdb);
 701    }
 702    db_unlock(mdb);
 703    return stat;
 704 }
 705
 706
 707 /*
 708  * Get the number of Media records
 709  *
 710  * Returns: -1 on failure
 711  *          number on success
 712  */
 713 int db_get_num_media_records(JCR *jcr, B_DB *mdb)
 714 {
 715    int stat = 0;
 716
 717    db_lock(mdb);
 718    Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 719    stat = get_sql_record_max(jcr, mdb);
 720    db_unlock(mdb);
 721    return stat;
 722 }
 723
 724
 725 /*
 726  * This function returns a list of all the Media record ids for
 727  *     the current Pool.
 728  *  The caller must free ids if non-NULL.
 729  *
 730  *  Returns 0: on failure
 731  *          1: on success
 732  */
 733 int db_get_media_ids(JCR *jcr, B_DB *mdb, uint32_t PoolId, int *num_ids, uint32_t *ids[])
 734 {
 735    SQL_ROW row;
 736    int stat = 0;
 737    int i = 0;
 738    uint32_t *id;
 739
 740    db_lock(mdb);
 741    *ids = NULL;
 742    Mmsg(&mdb->cmd, "SELECT MediaId FROM Media WHERE PoolId=%u", PoolId);
 743    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 744       *num_ids = sql_num_rows(mdb);
 745       if (*num_ids > 0) {
 746          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 747          while ((row = sql_fetch_row(mdb)) != NULL) {
 748             id[i++] = (uint32_t)atoi(row[0]);
 749          }
 750          *ids = id;
 751       }
 752       sql_free_result(mdb);
 753       stat = 1;
 754    } else {
 755       Mmsg(&mdb->errmsg, _("Media id select failed: ERR=%s\n"), sql_strerror(mdb));
 756       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 757       stat = 0;
 758    }
 759    db_unlock(mdb);
 760    return stat;
 761 }
 762
 763
 764 /* Get Media Record
 765  *
 766  * Returns: 0 on failure
 767  *          id on success
 768  */
 769 int db_get_media_record(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
 770 {
 771    SQL_ROW row;
 772    int stat = 0;
 773
 774    db_lock(mdb);
 775    if (mr->MediaId == 0 && mr->VolumeName[0] == 0) {
 776       Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 777       mr->MediaId = get_sql_record_max(jcr, mdb);
 778       db_unlock(mdb);
 779       return 1;
 780    }
 781    if (mr->MediaId != 0) {               /* find by id */
 782       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 783 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 784 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 785 Recycle,Slot,FirstWritten,LastWritten \
 786 FROM Media WHERE MediaId=%d", mr->MediaId);
 787    } else {                           /* find by name */
 788       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 789 VolBytes,VolMounts,VolErrors,VolWrites,MaxVolBytes,VolCapacityBytes,\
 790 MediaType,VolStatus,PoolId,VolRetention,VolUseDuration,MaxVolJobs,MaxVolFiles,\
 791 Recycle,Slot,FirstWritten,LastWritten \
 792 FROM Media WHERE VolumeName='%s'", mr->VolumeName);
 793    }
 794
 795    if (QUERY_DB(jcr, mdb, mdb->cmd)) {
 796       mdb->num_rows = sql_num_rows(mdb);
 797       if (mdb->num_rows > 1) {
 798          char ed1[30];
 799          Mmsg1(&mdb->errmsg, _("More than one Volume!: %s\n"),
 800             edit_uint64(mdb->num_rows, ed1));
 801          Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 802       } else if (mdb->num_rows == 1) {
 803          if ((row = sql_fetch_row(mdb)) == NULL) {
 804             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 805             Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
 806          } else {
 807             /* return values */
 808             mr->MediaId = str_to_int64(row[0]);
 809             bstrncpy(mr->VolumeName, row[1]!=NULL?row[1]:"", sizeof(mr->VolumeName));
 810             mr->VolJobs = str_to_int64(row[2]);
 811             mr->VolFiles = str_to_int64(row[3]);
 812             mr->VolBlocks = str_to_int64(row[4]);
 813             mr->VolBytes = str_to_uint64(row[5]);
 814             mr->VolMounts = str_to_int64(row[6]);
 815             mr->VolErrors = str_to_int64(row[7]);
 816             mr->VolWrites = str_to_int64(row[8]);
 817             mr->MaxVolBytes = str_to_uint64(row[9]);
 818             mr->VolCapacityBytes = str_to_uint64(row[10]);
 819             bstrncpy(mr->MediaType, row[11]!=NULL?row[11]:"", sizeof(mr->MediaType));
 820             bstrncpy(mr->VolStatus, row[12]!=NULL?row[12]:"", sizeof(mr->VolStatus));
 821             mr->PoolId = str_to_int64(row[13]);
 822             mr->VolRetention = str_to_uint64(row[14]);
 823             mr->VolUseDuration = str_to_uint64(row[15]);
 824             mr->MaxVolJobs = str_to_int64(row[16]);
 825             mr->MaxVolFiles = str_to_int64(row[17]);
 826             mr->Recycle = str_to_int64(row[18]);
 827             mr->Slot = str_to_int64(row[19]);
 828             bstrncpy(mr->cFirstWritten, row[20]!=NULL?row[20]:"", sizeof(mr->cFirstWritten));
 829             mr->FirstWritten = (time_t)str_to_utime(mr->cFirstWritten);
 830             bstrncpy(mr->cLastWritten, row[21]!=NULL?row[21]:"", sizeof(mr->cLastWritten));
 831             mr->LastWritten = (time_t)str_to_utime(mr->cLastWritten);
 832             stat = mr->MediaId;
 833          }
 834       } else {
 835          if (mr->MediaId != 0) {
 836             Mmsg1(&mdb->errmsg, _("Media record MediaId=%u not found.\n"), mr->MediaId);
 837          } else {
 838             Mmsg1(&mdb->errmsg, _("Media record for Volume \"%s\" not found.\n"),
 839                   mr->VolumeName);
 840          }
 841       }
 842       sql_free_result(mdb);
 843    }
 844    db_unlock(mdb);
 845    return stat;
 846 }
 847
 848
 849 #endif /* HAVE_MYSQL || HAVE_SQLITE */