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(JCR *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(JCR *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(JCR *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(JCR *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(JCR *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  *  for a specified PoolId
 252  *  The caller must free ids if non-NULL.
 253  *
 254  *  Returns 0: on failure
 255  *          1: on success
 256  */
 257 int db_get_media_ids(JCR *jcr, B_DB *mdb, uint32_t PoolId, int *num_ids, uint32_t *ids[])
 258 {
 259    int i = 0;
 260    uint32_t *id;
 261    MEDIA_DBR omr;
 262    int len;
 263
 264    db_lock(mdb);
 265    *ids = NULL;
 266    if (!bdb_open_media_file(mdb)) {
 267       db_unlock(mdb);
 268       return 0;
 269    }
 270    fseek(mdb->mediafd, 0L, SEEK_SET);   /* rewind file */
 271    /* Linear search through Pool records
 272     */
 273    len = sizeof(omr);
 274    if (mdb->control.MediaId == 0) {
 275       db_unlock(mdb);
 276       return 0;
 277    }
 278    *num_ids = mdb->control.MediaId;
 279    id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
 280    while (fread(&omr, len, 1, mdb->mediafd) > 0) {
 281       if (PoolId == omr.MediaId) {
 282          id[i++] = omr.MediaId;
 283       }
 284    }
 285    *ids = id;
 286    db_unlock(mdb);
 287    return 1;
 288 }
 289
 290 /*
 291  * Get Media Record
 292  * If the MediaId is non-zero, we get its record,
 293  *  otherwise, we search on the MediaName
 294  *
 295  * Returns: 0 on failure
 296  *          id on success
 297  */
 298 int db_get_media_record(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
 299 {
 300    faddr_t rec_addr;
 301    int found = 0;
 302    int stat = 0;
 303    int len;
 304    MEDIA_DBR omr;
 305
 306    db_lock(mdb);
 307    if (!bdb_open_media_file(mdb)) {
 308       db_unlock(mdb);
 309       return 0;
 310    }
 311    fseek(mdb->mediafd, 0L, SEEK_SET);   /* rewind file */
 312    rec_addr = 0;
 313    /* Linear search through Media records
 314     */
 315    len = sizeof(omr);
 316    while (fread(&omr, len, 1, mdb->mediafd) > 0) {
 317       if (omr.MediaId == 0) {
 318          continue;                    /* deleted record */
 319       }
 320       Dmsg1(200, "VolName=%s\n", omr.VolumeName);
 321       /* If id not zero, search by Id */
 322       if (mr->MediaId != 0) {
 323          Dmsg1(200, "MediaId=%d\n", mr->MediaId);
 324          if (mr->MediaId == omr.MediaId) {
 325            found = 1;
 326          }
 327       /* Search by Name */
 328       } else if (strcmp(mr->VolumeName, omr.VolumeName) == 0) {
 329          found = 1;
 330       }
 331       if (!found) {
 332          rec_addr = ftell(mdb->mediafd); /* save start next record */
 333          continue;
 334       }
 335       /* Found desired record, now return it */
 336       memcpy(mr, &omr, len);
 337       mr->rec_addr = rec_addr;
 338       stat = omr.MediaId;
 339       Dmsg3(200, "Found media record: MediaId=%d Name=%s MediaType=%s\n",
 340          omr.MediaId, omr.VolumeName, mr->MediaType);
 341       break;
 342    }
 343    if (stat == 0) {
 344       strcpy(mdb->errmsg, "Could not find requested Media record.\n");
 345    }
 346    db_unlock(mdb);
 347    return stat;
 348 }
 349
 350 /*
 351  * Find VolumeNames for a give JobId
 352  *  Returns: 0 on error or no Volumes found
 353  *           number of volumes on success
 354  *              Volumes are concatenated in VolumeNames
 355  *              separated by a vertical bar (|).
 356  */
 357 int db_get_job_volume_names(JCR *jcr, B_DB *mdb, uint32_t JobId, POOLMEM **VolumeNames)
 358 {
 359    int found = 0;
 360    JOBMEDIA_DBR jm;
 361    MEDIA_DBR mr;
 362    int jmlen, mrlen;
 363
 364    db_lock(mdb);
 365    if (!bdb_open_jobmedia_file(mdb)) {
 366       db_unlock(mdb);
 367       return 0;
 368    }
 369    if (!bdb_open_media_file(mdb)) {
 370       db_unlock(mdb);
 371       return 0;
 372    }
 373    jmlen = sizeof(jm);
 374    mrlen = sizeof(mr);
 375    *VolumeNames[0] = 0;
 376    fseek(mdb->jobmediafd, 0L, SEEK_SET); /* rewind the file */
 377    while (fread(&jm, jmlen, 1, mdb->jobmediafd) > 0) {
 378       if (jm.JobId == JobId) {
 379          /* Now look up VolumeName in Media file given MediaId */
 380          fseek(mdb->mediafd, 0L, SEEK_SET);
 381          while (fread(&mr, mrlen, 1, mdb->mediafd) > 0) {
 382             if (jm.MediaId == mr.MediaId) {
 383                if (*VolumeNames[0] != 0) {      /* if not first name, */
 384                   pm_strcat(VolumeNames, "|");  /* add separator */
 385                }
 386                pm_strcat(VolumeNames, mr.VolumeName); /* add Volume Name */
 387                found++;
 388             }
 389          }
 390       }
 391    }
 392    if (!found) {
 393       strcpy(mdb->errmsg, "No Volumes found.\n");
 394    }
 395    db_unlock(mdb);
 396    return found;
 397 }
 398
 399 /*
 400  * Get Client Record
 401  * If the ClientId is non-zero, we get its record,
 402  *  otherwise, we search on the Name
 403  *
 404  * Returns: 0 on failure
 405  *          id on success
 406  */
 407 int db_get_client_record(JCR *jcr, B_DB *mdb, CLIENT_DBR *cr)
 408 {
 409    CLIENT_DBR lcr;
 410    int len;
 411    int stat = 0;
 412
 413    db_lock(mdb);
 414    if (!bdb_open_client_file(mdb)) {
 415       db_unlock(mdb);
 416       return 0;
 417    }
 418    fseek(mdb->clientfd, 0L, SEEK_SET);   /* rewind file */
 419    /*
 420     * Linear search through Client records
 421     */
 422    len = sizeof(lcr);
 423    while (fread(&lcr, len, 1, mdb->clientfd)) {
 424       /* If id not zero, search by Id */
 425       if (cr->ClientId != 0) {
 426          if (cr->ClientId != lcr.ClientId) {
 427             continue;
 428          }
 429       /* Search by Name */
 430       } else if (strcmp(cr->Name, lcr.Name) != 0) {
 431          continue;                 /* not found */
 432       }
 433       memcpy(cr, &lcr, len);
 434       stat = lcr.ClientId;
 435       Dmsg2(200, "Found Client record: ClientId=%d Name=%s\n",
 436             lcr.ClientId, lcr.Name);
 437       break;
 438    }
 439    if (!stat) {
 440       strcpy(mdb->errmsg, "Client record not found.\n");
 441    }
 442    db_unlock(mdb);
 443    return stat;
 444 }
 445
 446 /*
 447  * Get FileSet Record   (We read the FILESET_DBR structure)
 448  * If the FileSetId is non-zero, we get its record,
 449  *  otherwise, we search on the FileSet (its name).
 450  *
 451  * Returns: 0 on failure
 452  *          id on success
 453  */
 454 int db_get_fileset_record(JCR *jcr, B_DB *mdb, FILESET_DBR *fsr)
 455 {
 456    FILESET_DBR lfsr;
 457    int stat = 0;
 458
 459    db_lock(mdb);
 460    if (!bdb_open_fileset_file(mdb)) {
 461       db_unlock(mdb);
 462       return 0;
 463    }
 464    fseek(mdb->filesetfd, 0L, SEEK_SET);   /* rewind file */
 465    /*
 466     * Linear search through FileSet records
 467     */
 468    while (fread(&lfsr, sizeof(lfsr), 1, mdb->filesetfd) > 0) {
 469       /* If id not zero, search by Id */
 470       if (fsr->FileSetId != 0) {
 471          if (fsr->FileSetId != lfsr.FileSetId) {
 472             continue;
 473          }
 474       /* Search by Name & MD5 */
 475       } else if (strcmp(fsr->FileSet, lfsr.FileSet) != 0 ||
 476                  strcmp(fsr->MD5, lfsr.MD5) != 0) {
 477          continue;                 /* not found */
 478       }
 479       /* Found desired record, now return it */
 480       memcpy(fsr, &lfsr, sizeof(lfsr));
 481       stat = fsr->FileSetId;
 482       Dmsg2(200, "Found FileSet record: FileSetId=%d FileSet=%s\n",
 483             lfsr.FileSetId, lfsr.FileSet);
 484       break;
 485    }
 486    if (!stat) {
 487       strcpy(mdb->errmsg, "FileSet record not found.\n");
 488    }
 489    db_unlock(mdb);
 490    return stat;
 491 }
 492
 493
 494
 495 int db_get_file_attributes_record(JCR *jcr, B_DB *mdb, char *fname, JOB_DBR *jr, FILE_DBR *fdbr)
 496 { return 0; }
 497
 498 int db_get_job_volume_parameters(JCR *jcr, B_DB *mdb, uint32_t JobId, VOL_PARAMS **VolParams)
 499 { return 0; }
 500
 501 int db_get_client_ids(JCR *jcr, B_DB *mdb, int *num_ids, uint32_t *ids[])
 502 { return 0; }
 503
 504 int db_get_counter_record(JCR *jcr, B_DB *mdb, COUNTER_DBR *cr)
 505 { return 0; }
 506
 507
 508 #endif /* HAVE_BACULA_DB */