git.sur5r.net Git - bacula/bacula/blob - bacula/src/cats/bdb_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  * Bacula Catalog Database routines written specifically
   8  *  for Bacula.  Note, these routines are VERY dumb and
   9  *  do not provide all the functionality of an SQL database.
  10  *  The purpose of these routines is to ensure that Bacula
  11  *  can limp along if no real database is loaded on the
  12  *  system.
  13  *
  14  *    Kern Sibbald, January MMI
  15  *
  16  *    Version $Id$
  17  */
  18
  19 /*
  20    Copyright (C) 2001-2003 Kern Sibbald and John Walker
  21
  22    This program is free software; you can redistribute it and/or
  23    modify it under the terms of the GNU General Public License as
  24    published by the Free Software Foundation; either version 2 of
  25    the License, or (at your option) any later version.
  26
  27    This program is distributed in the hope that it will be useful,
  28    but WITHOUT ANY WARRANTY; without even the implied warranty of
  29    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  30    General Public License for more details.
  31
  32    You should have received a copy of the GNU General Public
  33    License along with this program; if not, write to the Free
  34    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  35    MA 02111-1307, USA.
  36
  37  */
  38
  39
  40 /* The following is necessary so that we do not include
  41  * the dummy external definition of DB.
  42  */
  43 #define __SQL_C                       /* indicate that this is sql.c */
  44
  45 #include "bacula.h"
  46 #include "cats.h"
  47 #include "bdb.h"
  48
  49 #ifdef HAVE_BACULA_DB
  50
  51 /* Forward referenced functions */
  52
  53
  54 /* -----------------------------------------------------------------------
  55  *
  56  *   Bacula specific defines and subroutines
  57  *
  58  * -----------------------------------------------------------------------
  59  */
  60
  61
  62 /*
  63  * Get Job record for given JobId
  64  * Returns: 0 on failure
  65  *          1 on success
  66  */
  67
  68 int db_get_job_record(void *jcr, B_DB *mdb, JOB_DBR *jr)
  69 {
  70    JOB_DBR ojr;
  71    faddr_t rec_addr;
  72    int found = 0;
  73    int stat = 0;
  74    int len;
  75
  76    db_lock(mdb);
  77    if (jr->JobId == 0 && jr->Name[0] == 0) { /* he wants # of Job records */
  78       jr->JobId = mdb->control.JobId;
  79       db_unlock(mdb);
  80       return 1;
  81    }
  82    Dmsg0(200, "Open Jobs\n");
  83    if (!bdb_open_jobs_file(mdb)) {
  84       db_unlock(mdb);
  85       return 0;
  86    }
  87    fseek(mdb->jobfd, 0L, SEEK_SET);   /* rewind file */
  88    rec_addr = 0;
  89    /* Linear search through Job records
  90     */
  91    len = sizeof(ojr);
  92    while (fread(&ojr, len, 1, mdb->jobfd) > 0) {
  93       /* If id not zero, search by Id */
  94       if (jr->JobId != 0) {
  95          if (jr->JobId == ojr.JobId) {
  96            found = 1;
  97          }
  98       /* Search by Job */
  99       } else if (strcmp(jr->Job, ojr.Job) == 0) {
 100          found = 1;
 101          Dmsg1(200, "Found Job: %s\n", ojr.Job);
 102       }
 103       if (!found) {
 104          rec_addr = ftell(mdb->jobfd); /* save start next record */
 105          continue;
 106       }
 107       /* Found desired record, now return it */
 108       memcpy(jr, &ojr, len);
 109       jr->rec_addr = rec_addr;
 110       stat = ojr.JobId;
 111       Dmsg2(200, "Found job record: JobId=%d Job=%s",
 112          ojr.JobId, ojr.Job);
 113       break;
 114    }
 115    if (!found) {
 116       strcpy(mdb->errmsg, "Job record not found.\n");
 117    }
 118    db_unlock(mdb);
 119    Dmsg1(200, "Return job stat=%d\n", stat);
 120    return stat;
 121 }
 122
 123
 124 /*
 125  * Get the number of pool records
 126  *
 127  * Returns: -1 on failure
 128  *          number on success
 129  */
 130 int db_get_num_pool_records(void *jcr, B_DB *mdb)
 131 {
 132    int stat = 0;
 133
 134    db_lock(mdb);
 135    stat = mdb->control.PoolId;
 136    db_unlock(mdb);
 137    return stat;
 138 }
 139
 140 /*
 141  * This function returns a list of all the Pool record ids.
 142  *  The caller must free ids if non-NULL.
 143  *
 144  *  Returns 0: on failure
 145  *          1: on success
 146  */
 147 int db_get_pool_ids(void *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 148 {
 149    int i = 0;
 150    uint32_t *id;
 151    POOL_DBR opr;
 152    int len;
 153
 154    db_lock(mdb);
 155    *ids = NULL;
 156    if (!bdb_open_pools_file(mdb)) {
 157       db_unlock(mdb);
 158       return 0;
 159    }
 160    fseek(mdb->poolfd, 0L, SEEK_SET);   /* rewind file */
 161    /* Linear search through Pool records
 162     */
 163    len = sizeof(opr);
 164    *num_ids = mdb->control.PoolId;
 165    id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 166    while (fread(&opr, len, 1, mdb->poolfd) > 0) {
 167       id[i++] = opr.PoolId;
 168    }
 169    *ids = id;
 170    db_unlock(mdb);
 171    return 1;
 172 }
 173
 174
 175 /*
 176  * Get Pool Record
 177  * If the PoolId is non-zero, we get its record,
 178  *  otherwise, we search on the PoolName
 179  *
 180  * Returns: 0 on failure
 181  *          id on success
 182  */
 183 int db_get_pool_record(void *jcr, B_DB *mdb, POOL_DBR *pr)
 184 {
 185    POOL_DBR opr;
 186    faddr_t rec_addr;
 187    int found = 0;
 188    int stat = 0;
 189    int len;
 190
 191    db_lock(mdb);
 192    Dmsg0(200, "Open pools\n");
 193    if (!bdb_open_pools_file(mdb)) {
 194       db_unlock(mdb);
 195       return 0;
 196    }
 197    fseek(mdb->poolfd, 0L, SEEK_SET);   /* rewind file */
 198    rec_addr = 0;
 199    /* Linear search through Pool records
 200     */
 201    len = sizeof(opr);
 202    while (fread(&opr, len, 1, mdb->poolfd) > 0) {
 203       /* If id not zero, search by Id */
 204       if (pr->PoolId != 0) {
 205          if (pr->PoolId == opr.PoolId) {
 206            found = 1;
 207          }
 208       /* Search by Name */
 209       } else if (strcmp(pr->Name, opr.Name) == 0) {
 210          found = 1;
 211          Dmsg1(200, "Found pool: %s\n", opr.Name);
 212       }
 213       if (!found) {
 214          rec_addr = ftell(mdb->poolfd); /* save start next record */
 215          continue;
 216       }
 217       /* Found desired record, now return it */
 218       memcpy(pr, &opr, len);
 219       pr->rec_addr = rec_addr;
 220       stat = opr.PoolId;
 221       Dmsg3(200, "Found pool record: PoolId=%d Name=%s PoolType=%s\n",
 222          opr.PoolId, opr.Name, opr.PoolType);
 223       break;
 224    }
 225    if (!found) {
 226       strcpy(mdb->errmsg, "Pool record not found.\n");
 227    }
 228    db_unlock(mdb);
 229    Dmsg1(200, "Return pool stat=%d\n", stat);
 230    return stat;
 231 }
 232
 233 /*
 234  * Get the number of Media records
 235  *
 236  * Returns: -1 on failure
 237  *          number on success
 238  */
 239 int db_get_num_media_records(void *jcr, B_DB *mdb)
 240 {
 241    int stat = 0;
 242
 243    db_lock(mdb);
 244    stat = mdb->control.MediaId;
 245    db_unlock(mdb);
 246    return stat;
 247 }
 248
 249 /*
 250  * This function returns a list of all the Media record ids.
 251  *  The caller must free ids if non-NULL.
 252  *
 253  *  Returns 0: on failure
 254  *          1: on success
 255  */
 256 int db_get_media_ids(void *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 257 {
 258    int i = 0;
 259    uint32_t *id;
 260    MEDIA_DBR omr;
 261    int len;
 262
 263    db_lock(mdb);
 264    *ids = NULL;
 265    if (!bdb_open_media_file(mdb)) {
 266       db_unlock(mdb);
 267       return 0;
 268    }
 269    fseek(mdb->mediafd, 0L, SEEK_SET);   /* rewind file */
 270    /* Linear search through Pool records
 271     */
 272    len = sizeof(omr);
 273    if (mdb->control.MediaId == 0) {
 274       db_unlock(mdb);
 275       return 0;
 276    }
 277    *num_ids = mdb->control.MediaId;
 278    id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 279    while (fread(&omr, len, 1, mdb->mediafd) > 0) {
 280       id[i++] = omr.MediaId;
 281    }
 282    *ids = id;
 283    db_unlock(mdb);
 284    return 1;
 285 }
 286
 287 /*
 288  * Get Media Record
 289  * If the MediaId is non-zero, we get its record,
 290  *  otherwise, we search on the MediaName
 291  *
 292  * Returns: 0 on failure
 293  *          id on success
 294  */
 295 int db_get_media_record(void *jcr, B_DB *mdb, MEDIA_DBR *mr)
 296 {
 297    faddr_t rec_addr;
 298    int found = 0;
 299    int stat = 0;
 300    int len;
 301    MEDIA_DBR omr;
 302
 303    db_lock(mdb);
 304    if (!bdb_open_media_file(mdb)) {
 305       db_unlock(mdb);
 306       return 0;
 307    }
 308    fseek(mdb->mediafd, 0L, SEEK_SET);   /* rewind file */
 309    rec_addr = 0;
 310    /* Linear search through Media records
 311     */
 312    len = sizeof(omr);
 313    while (fread(&omr, len, 1, mdb->mediafd) > 0) {
 314       if (omr.MediaId == 0) {
 315          continue;                    /* deleted record */
 316       }
 317       Dmsg1(200, "VolName=%s\n", omr.VolumeName);
 318       /* If id not zero, search by Id */
 319       if (mr->MediaId != 0) {
 320          Dmsg1(200, "MediaId=%d\n", mr->MediaId);
 321          if (mr->MediaId == omr.MediaId) {
 322            found = 1;
 323          }
 324       /* Search by Name */
 325       } else if (strcmp(mr->VolumeName, omr.VolumeName) == 0) {
 326          found = 1;
 327       }
 328       if (!found) {
 329          rec_addr = ftell(mdb->mediafd); /* save start next record */
 330          continue;
 331       }
 332       /* Found desired record, now return it */
 333       memcpy(mr, &omr, len);
 334       mr->rec_addr = rec_addr;
 335       stat = omr.MediaId;
 336       Dmsg3(200, "Found media record: MediaId=%d Name=%s MediaType=%s\n",
 337          omr.MediaId, omr.VolumeName, mr->MediaType);
 338       break;
 339    }
 340    if (stat == 0) {
 341       strcpy(mdb->errmsg, "Could not find requested Media record.\n");
 342    }
 343    db_unlock(mdb);
 344    return stat;
 345 }
 346
 347 /*
 348  * Find VolumeNames for a give JobId
 349  *  Returns: 0 on error or no Volumes found
 350  *           number of volumes on success
 351  *              Volumes are concatenated in VolumeNames
 352  *              separated by a vertical bar (|).
 353  */
 354 int db_get_job_volume_names(void *jcr, B_DB *mdb, uint32_t JobId, POOLMEM **VolumeNames)
 355 {
 356    int found = 0;
 357    JOBMEDIA_DBR jm;
 358    MEDIA_DBR mr;
 359    int jmlen, mrlen;
 360
 361    db_lock(mdb);
 362    if (!bdb_open_jobmedia_file(mdb)) {
 363       db_unlock(mdb);
 364       return 0;
 365    }
 366    if (!bdb_open_media_file(mdb)) {
 367       db_unlock(mdb);
 368       return 0;
 369    }
 370    jmlen = sizeof(jm);
 371    mrlen = sizeof(mr);
 372    *VolumeNames[0] = 0;
 373    fseek(mdb->jobmediafd, 0L, SEEK_SET); /* rewind the file */
 374    while (fread(&jm, jmlen, 1, mdb->jobmediafd) > 0) {
 375       if (jm.JobId == JobId) {
 376          /* Now look up VolumeName in Media file given MediaId */
 377          fseek(mdb->mediafd, 0L, SEEK_SET);
 378          while (fread(&mr, mrlen, 1, mdb->mediafd) > 0) {
 379             if (jm.MediaId == mr.MediaId) {
 380                if (*VolumeNames[0] != 0) {      /* if not first name, */
 381                   pm_strcat(VolumeNames, "|");  /* add separator */
 382                }
 383                pm_strcat(VolumeNames, mr.VolumeName); /* add Volume Name */
 384                found++;
 385             }
 386          }
 387       }
 388    }
 389    if (!found) {
 390       strcpy(mdb->errmsg, "No Volumes found.\n");
 391    }
 392    db_unlock(mdb);
 393    return found;
 394 }
 395
 396 /*
 397  * Get Client Record
 398  * If the ClientId is non-zero, we get its record,
 399  *  otherwise, we search on the Name
 400  *
 401  * Returns: 0 on failure
 402  *          id on success
 403  */
 404 int db_get_client_record(void *jcr, B_DB *mdb, CLIENT_DBR *cr)
 405 {
 406    CLIENT_DBR lcr;
 407    int len;
 408    int stat = 0;
 409
 410    db_lock(mdb);
 411    if (!bdb_open_client_file(mdb)) {
 412       db_unlock(mdb);
 413       return 0;
 414    }
 415    fseek(mdb->clientfd, 0L, SEEK_SET);   /* rewind file */
 416    /*
 417     * Linear search through Client records
 418     */
 419    len = sizeof(lcr);
 420    while (fread(&lcr, len, 1, mdb->clientfd)) {
 421       /* If id not zero, search by Id */
 422       if (cr->ClientId != 0) {
 423          if (cr->ClientId != lcr.ClientId) {
 424             continue;
 425          }
 426       /* Search by Name */
 427       } else if (strcmp(cr->Name, lcr.Name) != 0) {
 428          continue;                 /* not found */
 429       }
 430       memcpy(cr, &lcr, len);
 431       stat = lcr.ClientId;
 432       Dmsg2(200, "Found Client record: ClientId=%d Name=%s\n",
 433             lcr.ClientId, lcr.Name);
 434       break;
 435    }
 436    if (!stat) {
 437       strcpy(mdb->errmsg, "Client record not found.\n");
 438    }
 439    db_unlock(mdb);
 440    return stat;
 441 }
 442
 443 /*
 444  * Get FileSet Record   (We read the FILESET_DBR structure)
 445  * If the FileSetId is non-zero, we get its record,
 446  *  otherwise, we search on the FileSet (its name).
 447  *
 448  * Returns: 0 on failure
 449  *          id on success
 450  */
 451 int db_get_fileset_record(void *jcr, B_DB *mdb, FILESET_DBR *fsr)
 452 {
 453    FILESET_DBR lfsr;
 454    int stat = 0;
 455
 456    db_lock(mdb);
 457    if (!bdb_open_fileset_file(mdb)) {
 458       db_unlock(mdb);
 459       return 0;
 460    }
 461    fseek(mdb->filesetfd, 0L, SEEK_SET);   /* rewind file */
 462    /*
 463     * Linear search through FileSet records
 464     */
 465    while (fread(&lfsr, sizeof(lfsr), 1, mdb->filesetfd) > 0) {
 466       /* If id not zero, search by Id */
 467       if (fsr->FileSetId != 0) {
 468          if (fsr->FileSetId != lfsr.FileSetId) {
 469             continue;
 470          }
 471       /* Search by Name & MD5 */
 472       } else if (strcmp(fsr->FileSet, lfsr.FileSet) != 0 ||
 473                  strcmp(fsr->MD5, lfsr.MD5) != 0) {
 474          continue;                 /* not found */
 475       }
 476       /* Found desired record, now return it */
 477       memcpy(fsr, &lfsr, sizeof(lfsr));
 478       stat = fsr->FileSetId;
 479       Dmsg2(200, "Found FileSet record: FileSetId=%d FileSet=%s\n",
 480             lfsr.FileSetId, lfsr.FileSet);
 481       break;
 482    }
 483    if (!stat) {
 484       strcpy(mdb->errmsg, "FileSet record not found.\n");
 485    }
 486    db_unlock(mdb);
 487    return stat;
 488 }
 489
 490
 491
 492 int db_get_file_attributes_record(void *jcr, B_DB *mdb, char *fname, FILE_DBR *fdbr)
 493 { return 0; }
 494
 495 int db_get_job_volume_parameters(void *jcr, B_DB *mdb, uint32_t JobId, VOL_PARAMS **VolParams)
 496 { return 0; }
 497
 498 int db_get_client_ids(void *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 499 { return 0; }
 500
 501 #endif /* HAVE_BACULA_DB */