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, 2001, 2002 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 /* *****FIXME**** fix fixed length of select_cmd[] and insert_cmd[] */
  33
  34 /* The following is necessary so that we do not include
  35  * the dummy external definition of DB.
  36  */
  37 #define __SQL_C                       /* indicate that this is sql.c */
  38
  39 #include "bacula.h"
  40 #include "cats.h"
  41
  42 #if    HAVE_MYSQL || HAVE_SQLITE
  43
  44 /* -----------------------------------------------------------------------
  45  *
  46  *   Generic Routines (or almost generic)
  47  *
  48  * -----------------------------------------------------------------------
  49  */
  50
  51 /* Forward referenced functions */
  52 static int db_get_file_record(B_DB *mdb, FILE_DBR *fdbr);
  53 static int db_get_filename_record(B_DB *mdb, char *fname);
  54 static int db_get_path_record(B_DB *mdb, char *path);
  55
  56
  57 /* Imported subroutines */
  58 extern void print_result(B_DB *mdb);
  59 extern int QueryDB(char *file, int line, B_DB *db, char *select_cmd);
  60
  61
  62 /*
  63  * Given a full filename (with path), look up the File record
  64  * (with attributes) in the database.
  65  *
  66  *  Returns: 0 on failure
  67  *           1 on success with the File record in FILE_DBR
  68  */
  69 int db_get_file_attributes_record(B_DB *mdb, char *fname, FILE_DBR *fdbr)
  70 {
  71    int fnl, pnl;
  72    char *l, *p;
  73    uint64_t id;
  74    char file[MAXSTRING];
  75    char spath[MAXSTRING];
  76    char buf[MAXSTRING];
  77    Dmsg1(20, "Enter get_file_from_catalog fname=%s \n", fname);
  78
  79    /* Find path without the filename.
  80     * I.e. everything after the last / is a "filename".
  81     * OK, maybe it is a directory name, but we treat it like
  82     * a filename. If we don't find a / then the whole name
  83     * must be a path name (e.g. c:).
  84     */
  85    for (p=l=fname; *p; p++) {
  86       if (*p == '/') {
  87          l = p;
  88       }
  89    }
  90    if (*l == '/') {                   /* did we find a slash? */
  91       l++;                            /* yes, point to filename */
  92    } else {                           /* no, whole thing must be path name */
  93       l = p;
  94    }
  95
  96    /* If filename doesn't exist (i.e. root directory), we
  97     * simply create a blank name consisting of a single
  98     * space. This makes handling zero length filenames
  99     * easier.
 100     */
 101    fnl = p - l;
 102    if (fnl > 255) {
 103       Jmsg1(mdb->jcr, M_WARNING, 0, _("Filename truncated to 255 chars: %s\n"), l);
 104       fnl = 255;
 105    }
 106    if (fnl > 0) {
 107       strncpy(file, l, fnl);          /* copy filename */
 108       file[fnl] = 0;
 109    } else {
 110       file[0] = ' ';                  /* blank filename */
 111       file[1] = 0;
 112       fnl = 1;
 113    }
 114
 115    pnl = l - fname;
 116    if (pnl > 255) {
 117       Jmsg1(mdb->jcr, M_WARNING, 0, _("Path name truncated to 255 chars: %s\n"), fname);
 118       pnl = 255;
 119    }
 120    strncpy(spath, fname, pnl);
 121    spath[pnl] = 0;
 122
 123    if (pnl == 0) {
 124       Mmsg1(&mdb->errmsg, _("Path length is zero. File=%s\n"), fname);
 125       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 126       spath[0] = ' ';
 127       spath[1] = 0;
 128       pnl = 1;
 129    }
 130
 131    Dmsg1(100, "spath=%s\n", spath);
 132    Dmsg1(100, "file=%s\n", file);
 133
 134    db_escape_string(buf, file, fnl);
 135    fdbr->FilenameId = db_get_filename_record(mdb, buf);
 136    Dmsg2(100, "db_get_filename_record FilenameId=%d file=%s\n", fdbr->FilenameId, buf);
 137
 138    db_escape_string(buf, spath, pnl);
 139    fdbr->PathId = db_get_path_record(mdb, buf);
 140    Dmsg2(100, "db_get_path_record PathId=%d path=%s\n", fdbr->PathId, buf);
 141
 142    id = db_get_file_record(mdb, fdbr);
 143
 144    return id;
 145 }
 146
 147
 148
 149 /* Get a File record
 150  * Returns: 0 on failure
 151  *          1 on success
 152  */
 153 static
 154 int db_get_file_record(B_DB *mdb, FILE_DBR *fdbr)
 155 {
 156    SQL_ROW row;
 157    int stat = 0;
 158
 159    db_lock(mdb);
 160    Mmsg(&mdb->cmd,
 161 "SELECT FileId, LStat, MD5 from File where File.JobId=%d and File.PathId=%d and \
 162 File.FilenameId=%d", fdbr->JobId, fdbr->PathId, fdbr->FilenameId);
 163
 164    Dmsg3(050, "Get_file_record JobId=%d FilenameId=%d PathId=%d\n",
 165       fdbr->JobId, fdbr->FilenameId, fdbr->PathId);
 166
 167    Dmsg1(100, "Query=%s\n", mdb->cmd);
 168
 169    if (QUERY_DB(mdb, mdb->cmd)) {
 170
 171       mdb->num_rows = sql_num_rows(mdb);
 172       Dmsg1(050, "get_file_record num_rows=%d\n", (int)mdb->num_rows);
 173
 174       /*
 175        * Note, we can find more than one File record with the same
 176        *  filename if the file is linked.   ????????
 177        */
 178       if (mdb->num_rows > 1) {
 179          Jmsg1(mdb->jcr, M_WARNING, 0, _("get_file_record want 1 got rows=%d\n"), mdb->num_rows);
 180          Jmsg1(mdb->jcr, M_ERROR, 0, "%s", mdb->cmd);
 181       }
 182       if (mdb->num_rows >= 1) {
 183          if ((row = sql_fetch_row(mdb)) == NULL) {
 184             Mmsg1(&mdb->errmsg, "Error fetching row: %s\n", sql_strerror(mdb));
 185             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 186          } else {
 187             fdbr->FileId = (FileId_t)strtod(row[0], NULL);
 188             strncpy(fdbr->LStat, row[1], sizeof(fdbr->LStat));
 189             fdbr->LStat[sizeof(fdbr->LStat)] = 0;
 190             strncpy(fdbr->MD5, row[2], sizeof(fdbr->MD5));
 191             fdbr->MD5[sizeof(fdbr->MD5)] = 0;
 192             stat = 1;
 193          }
 194       }
 195       sql_free_result(mdb);
 196    }
 197    db_unlock(mdb);
 198    return stat;
 199
 200 }
 201
 202 /* Get Filename record
 203  * Returns: 0 on failure
 204  *          FilenameId on success
 205  */
 206 static int db_get_filename_record(B_DB *mdb, char *fname)
 207 {
 208    SQL_ROW row;
 209    int FilenameId = 0;
 210
 211    if (*fname == 0) {
 212       Mmsg0(&mdb->errmsg, _("Null name given to db_get_filename_record\n"));
 213       Emsg0(M_ABORT, 0, mdb->errmsg);
 214    }
 215
 216    db_lock(mdb);
 217    Mmsg(&mdb->cmd, "SELECT FilenameId FROM Filename WHERE Name='%s'", fname);
 218    if (QUERY_DB(mdb, mdb->cmd)) {
 219
 220       mdb->num_rows = sql_num_rows(mdb);
 221
 222       if (mdb->num_rows > 1) {
 223          Mmsg1(&mdb->errmsg, _("More than one Filename!: %d\n"), (int)(mdb->num_rows));
 224          Jmsg(mdb->jcr, M_WARNING, 0, "%s", mdb->errmsg);
 225       }
 226       if (mdb->num_rows >= 1) {
 227          if ((row = sql_fetch_row(mdb)) == NULL) {
 228             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 229             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 230          } else {
 231             FilenameId = atoi(row[0]);
 232             if (FilenameId <= 0) {
 233                Mmsg2(&mdb->errmsg, _("Get DB Filename record %s found bad record: %d\n"),
 234                   mdb->cmd, FilenameId);
 235                Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 236                FilenameId = 0;
 237             }
 238          }
 239       }
 240       sql_free_result(mdb);
 241    }
 242    db_unlock(mdb);
 243    return FilenameId;
 244 }
 245
 246 /* Get path record
 247  * Returns: 0 on failure
 248  *          PathId on success
 249  */
 250 static int db_get_path_record(B_DB *mdb, char *path)
 251 {
 252    SQL_ROW row;
 253    uint32_t PathId = 0;
 254
 255    if (*path == 0) {
 256       Emsg0(M_ABORT, 0, _("Null path given to db_get_path_record\n"));
 257    }
 258
 259    db_lock(mdb);
 260
 261    if (mdb->cached_path_id != 0 && strcmp(mdb->cached_path, path) == 0) {
 262       db_unlock(mdb);
 263       return mdb->cached_path_id;
 264    }
 265
 266    Mmsg(&mdb->cmd, "SELECT PathId FROM Path WHERE Path='%s'", path);
 267
 268    if (QUERY_DB(mdb, mdb->cmd)) {
 269       char ed1[30];
 270       mdb->num_rows = sql_num_rows(mdb);
 271
 272       if (mdb->num_rows > 1) {
 273          Mmsg1(&mdb->errmsg, _("More than one Path!: %s\n"),
 274             edit_uint64(mdb->num_rows, ed1));
 275          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 276       } else if (mdb->num_rows == 1) {
 277          if ((row = sql_fetch_row(mdb)) == NULL) {
 278             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 279             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 280          } else {
 281             PathId = atoi(row[0]);
 282             if (PathId <= 0) {
 283                Mmsg2(&mdb->errmsg, _("Get DB path record %s found bad record: %d\n"),
 284                   mdb->cmd, PathId);
 285                Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 286                PathId = 0;
 287             } else {
 288                /* Cache path */
 289                if (PathId != mdb->cached_path_id) {
 290                   mdb->cached_path_id = PathId;
 291                   mdb->cached_path = check_pool_memory_size(mdb->cached_path,
 292                      strlen(path)+1);
 293                   strcpy(mdb->cached_path, path);
 294                }
 295             }
 296          }
 297       }
 298       sql_free_result(mdb);
 299    }
 300    db_unlock(mdb);
 301    return PathId;
 302 }
 303
 304
 305 /*
 306  * Get Job record for given JobId or Job name
 307  * Returns: 0 on failure
 308  *          1 on success
 309  */
 310 int db_get_job_record(B_DB *mdb, JOB_DBR *jr)
 311 {
 312    SQL_ROW row;
 313
 314    db_lock(mdb);
 315    if (jr->JobId == 0) {
 316       Mmsg(&mdb->cmd, "SELECT VolSessionId, VolSessionTime, \
 317 PoolId, StartTime, EndTime, JobFiles, JobBytes, JobTDate, Job, JobStatus \
 318 FROM Job WHERE Job='%s'", jr->Job);
 319     } else {
 320       Mmsg(&mdb->cmd, "SELECT VolSessionId, VolSessionTime, \
 321 PoolId, StartTime, EndTime, JobFiles, JobBytes, JobTDate, Job, JobStatus \
 322 FROM Job WHERE JobId=%d", jr->JobId);
 323     }
 324
 325    if (!QUERY_DB(mdb, mdb->cmd)) {
 326       db_unlock(mdb);
 327       return 0;                       /* failed */
 328    }
 329    if ((row = sql_fetch_row(mdb)) == NULL) {
 330       Mmsg1(&mdb->errmsg, _("No Job found for JobId %d\n"), jr->JobId);
 331       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 332       sql_free_result(mdb);
 333       db_unlock(mdb);
 334       return 0;                       /* failed */
 335    }
 336
 337    jr->VolSessionId = atol(row[0]);
 338    jr->VolSessionTime = atol(row[1]);
 339    jr->PoolId = atoi(row[2]);
 340    strcpy(jr->cStartTime, row[3]);
 341    strcpy(jr->cEndTime, row[4]);
 342    jr->JobFiles = atol(row[5]);
 343    jr->JobBytes = (uint64_t)strtod(row[6], NULL);
 344    jr->JobTDate = (btime_t)strtod(row[7], NULL);
 345    strcpy(jr->Job, row[8]);
 346    jr->JobStatus = (int)*row[9];
 347    sql_free_result(mdb);
 348
 349    db_unlock(mdb);
 350    return 1;
 351 }
 352
 353 /*
 354  * Find VolumeNames for a give JobId
 355  *  Returns: 0 on error or no Volumes found
 356  *           number of volumes on success
 357  *              Volumes are concatenated in VolumeNames
 358  *              separated by a vertical bar (|).
 359  */
 360 int db_get_job_volume_names(B_DB *mdb, uint32_t JobId, POOLMEM **VolumeNames)
 361 {
 362    SQL_ROW row;
 363    int stat = 0;
 364    int i;
 365
 366    db_lock(mdb);
 367    Mmsg(&mdb->cmd,
 368 "SELECT VolumeName FROM JobMedia,Media WHERE JobMedia.JobId=%d \
 369 AND JobMedia.MediaId=Media.MediaId", JobId);
 370
 371    Dmsg1(130, "VolNam=%s\n", mdb->cmd);
 372    *VolumeNames[0] = 0;
 373    if (QUERY_DB(mdb, mdb->cmd)) {
 374       mdb->num_rows = sql_num_rows(mdb);
 375       Dmsg1(130, "Num rows=%d\n", mdb->num_rows);
 376       if (mdb->num_rows <= 0) {
 377          Mmsg1(&mdb->errmsg, _("No volumes found for JobId=%d"), JobId);
 378          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 379          stat = 0;
 380       } else {
 381          stat = mdb->num_rows;
 382          for (i=0; i < stat; i++) {
 383             if ((row = sql_fetch_row(mdb)) == NULL) {
 384                Mmsg2(&mdb->errmsg, _("Error fetching row %d: ERR=%s\n"), i, sql_strerror(mdb));
 385                Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 386                stat = 0;
 387                break;
 388             } else {
 389                if (*VolumeNames[0] != 0) {
 390                   pm_strcat(VolumeNames, "|");
 391                }
 392                pm_strcat(VolumeNames, row[0]);
 393             }
 394          }
 395       }
 396       sql_free_result(mdb);
 397    }
 398    db_unlock(mdb);
 399    return stat;
 400 }
 401
 402
 403 /*
 404  * Get the number of pool records
 405  *
 406  * Returns: -1 on failure
 407  *          number on success
 408  */
 409 int db_get_num_pool_records(B_DB *mdb)
 410 {
 411    int stat = 0;
 412
 413    db_lock(mdb);
 414    Mmsg(&mdb->cmd, "SELECT count(*) from Pool");
 415    stat = get_sql_record_max(mdb);
 416    db_unlock(mdb);
 417    return stat;
 418 }
 419
 420 /*
 421  * This function returns a list of all the Pool record ids.
 422  *  The caller must free ids if non-NULL.
 423  *
 424  *  Returns 0: on failure
 425  *          1: on success
 426  */
 427 int db_get_pool_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
 428 {
 429    SQL_ROW row;
 430    int stat = 0;
 431    int i = 0;
 432    uint32_t *id;
 433
 434    db_lock(mdb);
 435    *ids = NULL;
 436    Mmsg(&mdb->cmd, "SELECT PoolId FROM Pool");
 437    if (QUERY_DB(mdb, mdb->cmd)) {
 438       *num_ids = sql_num_rows(mdb);
 439       if (*num_ids > 0) {
 440          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 441          while ((row = sql_fetch_row(mdb)) != NULL) {
 442             id[i++] = (uint32_t)atoi(row[0]);
 443          }
 444          *ids = id;
 445       }
 446       sql_free_result(mdb);
 447       stat = 1;
 448    } else {
 449       Mmsg(&mdb->errmsg, _("Pool id select failed: ERR=%s\n"), sql_strerror(mdb));
 450       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 451       stat = 0;
 452    }
 453    db_unlock(mdb);
 454    return stat;
 455 }
 456
 457
 458 /* Get Pool Record
 459  * If the PoolId is non-zero, we get its record,
 460  *  otherwise, we search on the PoolName
 461  *
 462  * Returns: 0 on failure
 463  *          id on success
 464  */
 465 int db_get_pool_record(B_DB *mdb, POOL_DBR *pdbr)
 466 {
 467    SQL_ROW row;
 468    int stat = 0;
 469
 470    db_lock(mdb);
 471    if (pdbr->PoolId != 0) {               /* find by id */
 472       Mmsg(&mdb->cmd,
 473 "SELECT PoolId, Name, NumVols, MaxVols, UseOnce, UseCatalog, AcceptAnyVolume, \
 474 AutoPrune, Recycle, VolRetention, \
 475 PoolType, LabelFormat FROM Pool WHERE Pool.PoolId=%d", pdbr->PoolId);
 476    } else {                           /* find by name */
 477       Mmsg(&mdb->cmd,
 478 "SELECT PoolId, Name, NumVols, MaxVols, UseOnce, UseCatalog, AcceptAnyVolume, \
 479 AutoPrune, Recycle, VolRetention, \
 480 PoolType, LabelFormat FROM Pool WHERE Pool.Name='%s'", pdbr->Name);
 481    }
 482
 483    if (QUERY_DB(mdb, mdb->cmd)) {
 484       mdb->num_rows = sql_num_rows(mdb);
 485       if (mdb->num_rows > 1) {
 486          char ed1[30];
 487          Mmsg1(&mdb->errmsg, _("More than one Pool!: %s\n"),
 488             edit_uint64(mdb->num_rows, ed1));
 489          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 490       } else if (mdb->num_rows == 1) {
 491          if ((row = sql_fetch_row(mdb)) == NULL) {
 492             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 493             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 494          } else {
 495             pdbr->PoolId = atoi(row[0]);
 496             strcpy(pdbr->Name, row[1]);
 497             pdbr->NumVols = atoi(row[2]);
 498             pdbr->MaxVols = atoi(row[3]);
 499             pdbr->UseOnce = atoi(row[4]);
 500             pdbr->UseCatalog = atoi(row[5]);
 501             pdbr->AcceptAnyVolume = atoi(row[6]);
 502             pdbr->AutoPrune = atoi(row[7]);
 503             pdbr->Recycle = atoi(row[8]);
 504             pdbr->VolRetention = (btime_t)strtod(row[9], NULL);
 505             strcpy(pdbr->PoolType, row[10]);
 506             if (row[11]) {
 507                strcpy(pdbr->LabelFormat, row[11]);
 508             } else {
 509                pdbr->LabelFormat[0] = 0;
 510             }
 511             stat = pdbr->PoolId;
 512          }
 513       }
 514       sql_free_result(mdb);
 515    }
 516    db_unlock(mdb);
 517    return stat;
 518 }
 519
 520 /* Get FileSet Record
 521  * If the FileSetId is non-zero, we get its record,
 522  *  otherwise, we search on the name
 523  *
 524  * Returns: 0 on failure
 525  *          id on success
 526  */
 527 int db_get_fileset_record(B_DB *mdb, FILESET_DBR *fsr)
 528 {
 529    SQL_ROW row;
 530    int stat = 0;
 531
 532    db_lock(mdb);
 533    if (fsr->FileSetId != 0) {               /* find by id */
 534       Mmsg(&mdb->cmd,
 535            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 536            "WHERE FileSetId=%u", fsr->FileSetId);
 537    } else {                           /* find by name */
 538       Mmsg(&mdb->cmd,
 539            "SELECT FileSetId, FileSet, MD5 FROM FileSet "
 540            "WHERE FileSet='%s'", fsr->FileSet);
 541    }
 542
 543    if (QUERY_DB(mdb, mdb->cmd)) {
 544       mdb->num_rows = sql_num_rows(mdb);
 545       if (mdb->num_rows > 1) {
 546          char ed1[30];
 547          Mmsg1(&mdb->errmsg, _("Got %s FileSets expected only one!\n"),
 548             edit_uint64(mdb->num_rows, ed1));
 549          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 550          sql_data_seek(mdb, mdb->num_rows-1);
 551       }
 552       if ((row = sql_fetch_row(mdb)) == NULL) {
 553          Mmsg1(&mdb->errmsg, _("Error fetching row get_fileset: %s\n"), sql_strerror(mdb));
 554          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 555       } else {
 556          fsr->FileSetId = atoi(row[0]);
 557          strcpy(fsr->FileSet, row[1]);
 558          strcpy(fsr->MD5, row[2]);
 559          stat = fsr->FileSetId;
 560       }
 561       sql_free_result(mdb);
 562    }
 563    db_unlock(mdb);
 564    return stat;
 565 }
 566
 567
 568 /*
 569  * Get the number of Media records
 570  *
 571  * Returns: -1 on failure
 572  *          number on success
 573  */
 574 int db_get_num_media_records(B_DB *mdb)
 575 {
 576    int stat = 0;
 577
 578    db_lock(mdb);
 579    Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 580    stat = get_sql_record_max(mdb);
 581    db_unlock(mdb);
 582    return stat;
 583 }
 584
 585
 586 /*
 587  * This function returns a list of all the Media record ids.
 588  *  The caller must free ids if non-NULL.
 589  *
 590  *  Returns 0: on failure
 591  *          1: on success
 592  */
 593 int db_get_media_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
 594 {
 595    SQL_ROW row;
 596    int stat = 0;
 597    int i = 0;
 598    uint32_t *id;
 599
 600    db_lock(mdb);
 601    *ids = NULL;
 602    Mmsg(&mdb->cmd, "SELECT MediaId FROM Media");
 603    if (QUERY_DB(mdb, mdb->cmd)) {
 604       *num_ids = sql_num_rows(mdb);
 605       if (*num_ids > 0) {
 606          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 607          while ((row = sql_fetch_row(mdb)) != NULL) {
 608             id[i++] = (uint32_t)atoi(row[0]);
 609          }
 610          *ids = id;
 611       }
 612       sql_free_result(mdb);
 613       stat = 1;
 614    } else {
 615       Mmsg(&mdb->errmsg, _("Media id select failed: ERR=%s\n"), sql_strerror(mdb));
 616       Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 617       stat = 0;
 618    }
 619    db_unlock(mdb);
 620    return stat;
 621 }
 622
 623
 624 /* Get Media Record
 625  *
 626  * Returns: 0 on failure
 627  *          id on success
 628  */
 629 int db_get_media_record(B_DB *mdb, MEDIA_DBR *mr)
 630 {
 631    SQL_ROW row;
 632    int stat = 0;
 633
 634    db_lock(mdb);
 635    if (mr->MediaId == 0 && mr->VolumeName[0] == 0) {
 636       Mmsg(&mdb->cmd, "SELECT count(*) from Media");
 637       mr->MediaId = get_sql_record_max(mdb);
 638       db_unlock(mdb);
 639       return 1;
 640    }
 641    if (mr->MediaId != 0) {               /* find by id */
 642       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 643 VolBytes,VolMounts,VolErrors,VolWrites,VolMaxBytes,VolCapacityBytes,\
 644 MediaType,VolStatus,PoolId,VolRetention,Recycle,Slot \
 645 FROM Media WHERE MediaId=%d", mr->MediaId);
 646    } else {                           /* find by name */
 647       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
 648 VolBytes,VolMounts,VolErrors,VolWrites,VolMaxBytes,VolCapacityBytes,\
 649 MediaType,VolStatus,PoolId,VolRetention,Recycle,Slot \
 650 FROM Media WHERE VolumeName='%s'", mr->VolumeName);
 651    }
 652
 653    if (QUERY_DB(mdb, mdb->cmd)) {
 654       mdb->num_rows = sql_num_rows(mdb);
 655       if (mdb->num_rows > 1) {
 656          char ed1[30];
 657          Mmsg1(&mdb->errmsg, _("More than one Volume!: %s\n"),
 658             edit_uint64(mdb->num_rows, ed1));
 659          Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 660       } else if (mdb->num_rows == 1) {
 661          if ((row = sql_fetch_row(mdb)) == NULL) {
 662             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
 663             Jmsg(mdb->jcr, M_ERROR, 0, "%s", mdb->errmsg);
 664          } else {
 665             /* return values */
 666             mr->MediaId = atoi(row[0]);
 667             strcpy(mr->VolumeName, row[1]);
 668             mr->VolJobs = atoi(row[2]);
 669             mr->VolFiles = atoi(row[3]);
 670             mr->VolBlocks = atoi(row[4]);
 671             mr->VolBytes = (uint64_t)strtod(row[5], NULL);
 672             mr->VolMounts = atoi(row[6]);
 673             mr->VolErrors = atoi(row[7]);
 674             mr->VolWrites = atoi(row[8]);
 675             mr->VolMaxBytes = (uint64_t)strtod(row[9], NULL);
 676             mr->VolCapacityBytes = (uint64_t)strtod(row[10], NULL);
 677             strcpy(mr->MediaType, row[11]);
 678             strcpy(mr->VolStatus, row[12]);
 679             mr->PoolId = atoi(row[13]);
 680             mr->VolRetention = (btime_t)strtod(row[14], NULL);
 681             mr->Recycle = atoi(row[15]);
 682             mr->Slot = atoi(row[16]);
 683             stat = mr->MediaId;
 684          }
 685       } else {
 686          Mmsg0(&mdb->errmsg, _("Media record not found.\n"));
 687       }
 688       sql_free_result(mdb);
 689    }
 690    db_unlock(mdb);
 691    return stat;
 692 }
 693
 694
 695 #endif /* HAVE_MYSQL || HAVE_SQLITE */