X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=bacula%2Fsrc%2Fdird%2Fbsr.c;h=4e03daa5a257079118b3b929e9f2ddbc160b7ee4;hb=6e637fce30a1fdbc2da43552f513f529db4d4e87;hp=f656817ee19ff63624600f010cd3f5f3e57a3f4e;hpb=5eafeb9e405cc7f96048f30554c27206b54634f0;p=bacula%2Fbacula diff --git a/bacula/src/dird/bsr.c b/bacula/src/dird/bsr.c index f656817ee1..4e03daa5a2 100644 --- a/bacula/src/dird/bsr.c +++ b/bacula/src/dird/bsr.c @@ -9,32 +9,39 @@ * * Version $Id$ */ - /* - Copyright (C) 2002-2005 Kern Sibbald + Bacula® - The Network Backup Solution + + Copyright (C) 2002-2006 Free Software Foundation Europe e.V. - This program is free software; you can redistribute it and/or - modify it under the terms of the GNU General Public License as - published by the Free Software Foundation; either version 2 of - the License, or (at your option) any later version. + The main author of Bacula is Kern Sibbald, with contributions from + many others, a complete list can be found in the file AUTHORS. + This program is Free Software; you can redistribute it and/or + modify it under the terms of version two of the GNU General Public + License as published by the Free Software Foundation and included + in the file LICENSE. - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of + This program is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. - You should have received a copy of the GNU General Public - License along with this program; if not, write to the Free - Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, - MA 02111-1307, USA. + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + 02110-1301, USA. - */ + Bacula® is a registered trademark of John Walker. + The licensor of Bacula is the Free Software Foundation Europe + (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich, + Switzerland, email:ftf@fsfeurope.org. +*/ #include "bacula.h" #include "dird.h" /* Forward referenced functions */ -static uint32_t write_bsr(UAContext *ua, RBSR *bsr, FILE *fd); +static uint32_t write_bsr(UAContext *ua, RESTORE_CTX &rx, FILE *fd); void print_bsr(UAContext *ua, RBSR *bsr); @@ -51,12 +58,34 @@ RBSR_FINDEX *new_findex() /* Free all BSR FileIndex entries */ static void free_findex(RBSR_FINDEX *fi) { - if (fi) { - free_findex(fi->next); + RBSR_FINDEX *next; + for ( ; fi; fi=next) { + next = fi->next; free(fi); } } +/* + * Get storage device name from Storage resource + */ +static bool get_storage_device(char *device, char *storage) +{ + STORE *store; + if (storage[0] == 0) { + return false; + } + store = (STORE *)GetResWithName(R_STORAGE, storage); + if (!store) { + return false; + } + DEVICE *dev = (DEVICE *)(store->device->first()); + if (!dev) { + return false; + } + bstrncpy(device, dev->hdr.name, MAX_NAME_LENGTH); + return true; +} + /* * Our data structures were not designed completely * correctly, so the file indexes cover the full @@ -67,7 +96,7 @@ static void free_findex(RBSR_FINDEX *fi) * We are called here once for each JobMedia record * for each Volume. */ -static uint32_t write_findex(UAContext *ua, RBSR_FINDEX *fi, +static uint32_t write_findex(RBSR_FINDEX *fi, int32_t FirstIndex, int32_t LastIndex, FILE *fd) { uint32_t count = 0; @@ -111,14 +140,14 @@ static bool is_volume_selected(RBSR_FINDEX *fi, static void print_findex(UAContext *ua, RBSR_FINDEX *fi) { - bsendmsg(ua, "fi=0x%lx\n", fi); + ua->send_msg("fi=0x%lx\n", fi); for ( ; fi; fi=fi->next) { if (fi->findex == fi->findex2) { - bsendmsg(ua, "FileIndex=%d\n", fi->findex); -// Dmsg1(100, "FileIndex=%d\n", fi->findex); + ua->send_msg("FileIndex=%d\n", fi->findex); + Dmsg1(1000, "FileIndex=%d\n", fi->findex); } else { - bsendmsg(ua, "FileIndex=%d-%d\n", fi->findex, fi->findex2); -// Dmsg2(100, "FileIndex=%d-%d\n", fi->findex, fi->findex2); + ua->send_msg("FileIndex=%d-%d\n", fi->findex, fi->findex2); + Dmsg2(1000, "FileIndex=%d-%d\n", fi->findex, fi->findex2); } } } @@ -134,12 +163,13 @@ RBSR *new_bsr() /* Free the entire BSR */ void free_bsr(RBSR *bsr) { - if (bsr) { + RBSR *next; + for ( ; bsr; bsr=next) { free_findex(bsr->fi); if (bsr->VolParams) { free(bsr->VolParams); } - free_bsr(bsr->next); + next = bsr->next; free(bsr); } } @@ -148,159 +178,305 @@ void free_bsr(RBSR *bsr) * Complete the BSR by filling in the VolumeName and * VolSessionId and VolSessionTime using the JobId */ -int complete_bsr(UAContext *ua, RBSR *bsr) +bool complete_bsr(UAContext *ua, RBSR *bsr) { - if (bsr) { + for ( ; bsr; bsr=bsr->next) { JOB_DBR jr; memset(&jr, 0, sizeof(jr)); jr.JobId = bsr->JobId; if (!db_get_job_record(ua->jcr, ua->db, &jr)) { - bsendmsg(ua, _("Unable to get Job record. ERR=%s\n"), db_strerror(ua->db)); - return 0; + ua->error_msg(_("Unable to get Job record. ERR=%s\n"), db_strerror(ua->db)); + return false; } bsr->VolSessionId = jr.VolSessionId; bsr->VolSessionTime = jr.VolSessionTime; if ((bsr->VolCount=db_get_job_volume_parameters(ua->jcr, ua->db, bsr->JobId, &(bsr->VolParams))) == 0) { - bsendmsg(ua, _("Unable to get Job Volume Parameters. ERR=%s\n"), db_strerror(ua->db)); + ua->error_msg(_("Unable to get Job Volume Parameters. ERR=%s\n"), db_strerror(ua->db)); if (bsr->VolParams) { free(bsr->VolParams); bsr->VolParams = NULL; } - return 0; + return false; } - return complete_bsr(ua, bsr->next); } - return 1; + return true; +} + +static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; +static uint32_t uniq = 0; + +static void make_unique_restore_filename(UAContext *ua, POOL_MEM &fname) +{ + JCR *jcr = ua->jcr; + int i = find_arg_with_value(ua, "bootstrap"); + if (i >= 0) { + Mmsg(fname, "%s", ua->argv[i]); + jcr->unlink_bsr = false; + } else { + P(mutex); + uniq++; + V(mutex); + Mmsg(fname, "%s/%s.restore.%u.bsr", working_directory, my_name, uniq); + jcr->unlink_bsr = true; + } + if (jcr->RestoreBootstrap) { + free(jcr->RestoreBootstrap); + } + jcr->RestoreBootstrap = bstrdup(fname.c_str()); } /* * Write the bootstrap records to file */ -uint32_t write_bsr_file(UAContext *ua, RBSR *bsr) +uint32_t write_bsr_file(UAContext *ua, RESTORE_CTX &rx) { FILE *fd; - POOLMEM *fname = get_pool_memory(PM_MESSAGE); + POOL_MEM fname(PM_MESSAGE); + POOL_MEM volmsg(PM_MESSAGE); uint32_t count = 0;; bool err; + char *p; + JobId_t JobId; + char Device[MAX_NAME_LENGTH]; - Mmsg(fname, "%s/restore.bsr", working_directory); - fd = fopen(fname, "w+"); + make_unique_restore_filename(ua, fname); + fd = fopen(fname.c_str(), "w+b"); if (!fd) { berrno be; - bsendmsg(ua, _("Unable to create bootstrap file %s. ERR=%s\n"), - fname, be.strerror()); + ua->error_msg(_("Unable to create bootstrap file %s. ERR=%s\n"), + fname.c_str(), be.bstrerror()); goto bail_out; } /* Write them to file */ - count = write_bsr(ua, bsr, fd); + count = write_bsr(ua, rx, fd); err = ferror(fd); fclose(fd); + if (count == 0) { + ua->info_msg(_("No files found to restore/migrate. No bootstrap file written.\n")); + goto bail_out; + } if (err) { - bsendmsg(ua, _("Error writing bsr file.\n")); + ua->error_msg(_("Error writing bsr file.\n")); count = 0; goto bail_out; } - bsendmsg(ua, _("Bootstrap records written to %s\n"), fname); + ua->send_msg(_("Bootstrap records written to %s\n"), fname.c_str()); /* Tell the user what he will need to mount */ - bsendmsg(ua, "\n"); - bsendmsg(ua, _("The job will require the following Volumes:\n")); + ua->send_msg("\n"); + ua->send_msg(_("The job will require the following\n" + " Volume(s) Storage(s) SD Device(s)\n" + "===========================================================================\n")); /* Create Unique list of Volumes using prompt list */ start_prompt(ua, ""); - for (RBSR *nbsr=bsr; nbsr; nbsr=nbsr->next) { - for (int i=0; i < nbsr->VolCount; i++) { - if (nbsr->VolParams[i].VolumeName[0]) { - add_prompt(ua, nbsr->VolParams[i].VolumeName); + if (*rx.JobIds) { + /* Ensure that the volumes are printed in JobId order */ + for (p=rx.JobIds; get_next_jobid_from_list(&p, &JobId) > 0; ) { + for (RBSR *nbsr=rx.bsr; nbsr; nbsr=nbsr->next) { + if (JobId != nbsr->JobId) { + continue; + } + for (int i=0; i < nbsr->VolCount; i++) { + if (nbsr->VolParams[i].VolumeName[0]) { + if (!get_storage_device(Device, nbsr->VolParams[i].Storage)) { + Device[0] = 0; + } + Mmsg(volmsg, "%-25.25s %-25.25s %-25.25s", + nbsr->VolParams[i].VolumeName, + nbsr->VolParams[i].Storage, Device); + add_prompt(ua, volmsg.c_str()); + } + } + } + } + } else { + /* Print Volumes in any order */ + for (RBSR *nbsr=rx.bsr; nbsr; nbsr=nbsr->next) { + for (int i=0; i < nbsr->VolCount; i++) { + if (nbsr->VolParams[i].VolumeName[0]) { + if (!get_storage_device(Device, nbsr->VolParams[i].Storage)) { + Device[0] = 0; + } + Mmsg(volmsg, "%-25.25s %-25.25s %-25.25s", + nbsr->VolParams[i].VolumeName, + nbsr->VolParams[i].Storage, Device); + add_prompt(ua, volmsg.c_str()); + } } } } for (int i=0; i < ua->num_prompts; i++) { - bsendmsg(ua, " %s\n", ua->prompt[i]); + ua->send_msg(" %s\n", ua->prompt[i]); free(ua->prompt[i]); } if (ua->num_prompts == 0) { - bsendmsg(ua, _("No Volumes found to restore.\n")); + ua->send_msg(_("No Volumes found to restore.\n")); count = 0; } ua->num_prompts = 0; - bsendmsg(ua, "\n"); + ua->send_msg("\n"); bail_out: - free_pool_memory(fname); return count; } -static uint32_t write_bsr(UAContext *ua, RBSR *bsr, FILE *fd) +/* + * Here we actually write out the details of the bsr file. + * Note, there is one bsr for each JobId, but the bsr may + * have multiple volumes, which have been entered in the + * order they were written. + * The bsrs must be written out in the order the JobIds + * are found in the jobid list. + */ +static uint32_t write_bsr(UAContext *ua, RESTORE_CTX &rx, FILE *fd) { uint32_t count = 0; uint32_t total_count = 0; uint32_t LastIndex = 0; bool first = true; - if (bsr) { - /* - * For a given volume, loop over all the JobMedia records. - * VolCount is the number of JobMedia records. - */ - for (int i=0; i < bsr->VolCount; i++) { - if (!is_volume_selected(bsr->fi, bsr->VolParams[i].FirstIndex, - bsr->VolParams[i].LastIndex)) { - bsr->VolParams[i].VolumeName[0] = 0; /* zap VolumeName */ - continue; - } - fprintf(fd, "Volume=\"%s\"\n", bsr->VolParams[i].VolumeName); - fprintf(fd, "MediaType=\"%s\"\n", bsr->VolParams[i].MediaType); - fprintf(fd, "VolSessionId=%u\n", bsr->VolSessionId); - fprintf(fd, "VolSessionTime=%u\n", bsr->VolSessionTime); - if (bsr->VolParams[i].StartFile == bsr->VolParams[i].EndFile) { - fprintf(fd, "VolFile=%u\n", bsr->VolParams[i].StartFile); - } else { - fprintf(fd, "VolFile=%u-%u\n", bsr->VolParams[i].StartFile, - bsr->VolParams[i].EndFile); - } - if (bsr->VolParams[i].StartBlock == bsr->VolParams[i].EndBlock) { - fprintf(fd, "VolFile=%u\n", bsr->VolParams[i].StartBlock); - } else { - fprintf(fd, "VolBlock=%u-%u\n", bsr->VolParams[i].StartBlock, - bsr->VolParams[i].EndBlock); + char *p; + JobId_t JobId; + char device[MAX_NAME_LENGTH]; + RBSR *bsr; + if (*rx.JobIds == 0) { + for (bsr=rx.bsr; bsr; bsr=bsr->next) { + /* + * For a given volume, loop over all the JobMedia records. + * VolCount is the number of JobMedia records. + */ + for (int i=0; i < bsr->VolCount; i++) { + if (!is_volume_selected(bsr->fi, bsr->VolParams[i].FirstIndex, + bsr->VolParams[i].LastIndex)) { + bsr->VolParams[i].VolumeName[0] = 0; /* zap VolumeName */ + continue; + } + if (!rx.store) { + find_storage_resource(ua, rx, bsr->VolParams[i].Storage, + bsr->VolParams[i].MediaType); + } + fprintf(fd, "Volume=\"%s\"\n", bsr->VolParams[i].VolumeName); + fprintf(fd, "MediaType=\"%s\"\n", bsr->VolParams[i].MediaType); + if (get_storage_device(device, bsr->VolParams[i].Storage)) { + fprintf(fd, "Device=\"%s\"\n", device); + } + if (bsr->VolParams[i].Slot > 0) { + fprintf(fd, "Slot=%d\n", bsr->VolParams[i].Slot); + } + fprintf(fd, "VolSessionId=%u\n", bsr->VolSessionId); + fprintf(fd, "VolSessionTime=%u\n", bsr->VolSessionTime); + if (bsr->VolParams[i].StartFile == bsr->VolParams[i].EndFile) { + fprintf(fd, "VolFile=%u\n", bsr->VolParams[i].StartFile); + } else { + fprintf(fd, "VolFile=%u-%u\n", bsr->VolParams[i].StartFile, + bsr->VolParams[i].EndFile); + } + if (bsr->VolParams[i].StartBlock == bsr->VolParams[i].EndBlock) { + fprintf(fd, "VolBlock=%u\n", bsr->VolParams[i].StartBlock); + } else { + fprintf(fd, "VolBlock=%u-%u\n", bsr->VolParams[i].StartBlock, + bsr->VolParams[i].EndBlock); + } + // Dmsg2(100, "bsr VolParam FI=%u LI=%u\n", + // bsr->VolParams[i].FirstIndex, bsr->VolParams[i].LastIndex); + + count = write_findex(bsr->fi, bsr->VolParams[i].FirstIndex, + bsr->VolParams[i].LastIndex, fd); + if (count) { + fprintf(fd, "Count=%u\n", count); + } + total_count += count; + /* If the same file is present on two tapes or in two files + * on a tape, it is a continuation, and should not be treated + * twice in the totals. + */ + if (!first && LastIndex == bsr->VolParams[i].FirstIndex) { + total_count--; + } + first = false; + LastIndex = bsr->VolParams[i].LastIndex; } -// Dmsg2(100, "bsr VolParam FI=%u LI=%u\n", -// bsr->VolParams[i].FirstIndex, bsr->VolParams[i].LastIndex); - - count = write_findex(ua, bsr->fi, bsr->VolParams[i].FirstIndex, - bsr->VolParams[i].LastIndex, fd); - if (count) { - fprintf(fd, "Count=%u\n", count); + } + return total_count; + } + for (p=rx.JobIds; get_next_jobid_from_list(&p, &JobId) > 0; ) { + for (bsr=rx.bsr; bsr; bsr=bsr->next) { + if (JobId != bsr->JobId) { + continue; } - total_count += count; - /* If the same file is present on two tapes or in two files - * on a tape, it is a continuation, and should not be treated - * twice in the totals. + /* + * For a given volume, loop over all the JobMedia records. + * VolCount is the number of JobMedia records. */ - if (!first && LastIndex == bsr->VolParams[i].FirstIndex) { - total_count--; + for (int i=0; i < bsr->VolCount; i++) { + if (!is_volume_selected(bsr->fi, bsr->VolParams[i].FirstIndex, + bsr->VolParams[i].LastIndex)) { + bsr->VolParams[i].VolumeName[0] = 0; /* zap VolumeName */ + continue; + } + if (!rx.store) { + find_storage_resource(ua, rx, bsr->VolParams[i].Storage, + bsr->VolParams[i].MediaType); + } + fprintf(fd, "Volume=\"%s\"\n", bsr->VolParams[i].VolumeName); + fprintf(fd, "MediaType=\"%s\"\n", bsr->VolParams[i].MediaType); + if (get_storage_device(device, bsr->VolParams[i].Storage)) { + fprintf(fd, "Device=\"%s\"\n", device); + } + if (bsr->VolParams[i].Slot > 0) { + fprintf(fd, "Slot=%d\n", bsr->VolParams[i].Slot); + } + fprintf(fd, "VolSessionId=%u\n", bsr->VolSessionId); + fprintf(fd, "VolSessionTime=%u\n", bsr->VolSessionTime); + if (bsr->VolParams[i].StartFile == bsr->VolParams[i].EndFile) { + fprintf(fd, "VolFile=%u\n", bsr->VolParams[i].StartFile); + } else { + fprintf(fd, "VolFile=%u-%u\n", bsr->VolParams[i].StartFile, + bsr->VolParams[i].EndFile); + } + if (bsr->VolParams[i].StartBlock == bsr->VolParams[i].EndBlock) { + fprintf(fd, "VolBlock=%u\n", bsr->VolParams[i].StartBlock); + } else { + fprintf(fd, "VolBlock=%u-%u\n", bsr->VolParams[i].StartBlock, + bsr->VolParams[i].EndBlock); + } + // Dmsg2(100, "bsr VolParam FI=%u LI=%u\n", + // bsr->VolParams[i].FirstIndex, bsr->VolParams[i].LastIndex); + + count = write_findex(bsr->fi, bsr->VolParams[i].FirstIndex, + bsr->VolParams[i].LastIndex, fd); + if (count) { + fprintf(fd, "Count=%u\n", count); + } + total_count += count; + /* If the same file is present on two tapes or in two files + * on a tape, it is a continuation, and should not be treated + * twice in the totals. + */ + if (!first && LastIndex == bsr->VolParams[i].FirstIndex) { + total_count--; + } + first = false; + LastIndex = bsr->VolParams[i].LastIndex; } - first = false; - LastIndex = bsr->VolParams[i].LastIndex; } - write_bsr(ua, bsr->next, fd); } return total_count; } void print_bsr(UAContext *ua, RBSR *bsr) { - if (bsr) { + for ( ; bsr; bsr=bsr->next) { for (int i=0; i < bsr->VolCount; i++) { - bsendmsg(ua, "Volume=\"%s\"\n", bsr->VolParams[i].VolumeName); - bsendmsg(ua, "MediaType\"%s\"\n", bsr->VolParams[i].MediaType); - bsendmsg(ua, "VolSessionId=%u\n", bsr->VolSessionId); - bsendmsg(ua, "VolSessionTime=%u\n", bsr->VolSessionTime); - bsendmsg(ua, "VolFile=%u-%u\n", bsr->VolParams[i].StartFile, + ua->send_msg("Volume=\"%s\"\n", bsr->VolParams[i].VolumeName); + ua->send_msg("MediaType\"%s\"\n", bsr->VolParams[i].MediaType); + ua->send_msg("VolSessionId=%u\n", bsr->VolSessionId); + ua->send_msg("VolSessionTime=%u\n", bsr->VolSessionTime); + ua->send_msg("VolFile=%u-%u\n", bsr->VolParams[i].StartFile, bsr->VolParams[i].EndFile); - bsendmsg(ua, "VolBlock=%u-%u\n", bsr->VolParams[i].StartBlock, + ua->send_msg("VolBlock=%u-%u\n", bsr->VolParams[i].StartBlock, bsr->VolParams[i].EndBlock); print_findex(ua, bsr->fi); }