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