git.sur5r.net Git - bacula/bacula/blob - bacula/src/stored/bextract.c

   1 /*
   2  *
   3  *  Dumb program to extract files from a Bacula backup.
   4  *
   5  *   Kern E. Sibbald
   6  *
   7  *   Version $Id$
   8  *
   9  */
  10 /*
  11    Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
  12
  13    This program is free software; you can redistribute it and/or
  14    modify it under the terms of the GNU General Public License as
  15    published by the Free Software Foundation; either version 2 of
  16    the License, or (at your option) any later version.
  17
  18    This program is distributed in the hope that it will be useful,
  19    but WITHOUT ANY WARRANTY; without even the implied warranty of
  20    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  21    General Public License for more details.
  22
  23    You should have received a copy of the GNU General Public
  24    License along with this program; if not, write to the Free
  25    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  26    MA 02111-1307, USA.
  27
  28  */
  29
  30 #include "bacula.h"
  31 #include "stored.h"
  32 #include "findlib/find.h"
  33
  34 #ifdef HAVE_CYGWIN
  35 int win32_client = 1;
  36 #else
  37 int win32_client = 0;
  38 #endif
  39
  40
  41 static void do_extract(char *fname);
  42 static void record_cb(JCR *jcr, DEVICE *dev, DEV_BLOCK *block, DEV_RECORD *rec);
  43
  44 static DEVICE *dev = NULL;
  45 static BFILE bfd;
  46 static JCR *jcr;
  47 static FF_PKT my_ff;
  48 static FF_PKT *ff = &my_ff;
  49 static BSR *bsr = NULL;
  50 static int extract = FALSE;
  51 static int non_support_data = 0;
  52 static int non_support_attr = 0;
  53 static long total = 0;
  54 static ATTR *attr;
  55 static char *where;
  56 static uint32_t num_files = 0;
  57 static uint32_t compress_buf_size = 70000;
  58 static POOLMEM *compress_buf;
  59 static int prog_name_msg = 0;
  60 static int win32_data_msg = 0;
  61 static char *VolumeName = NULL;
  62
  63 static char *wbuf;                    /* write buffer address */
  64 static uint32_t wsize;                /* write size */
  65 static uint64_t fileAddr = 0;         /* file write address */
  66
  67 #define CONFIG_FILE "bacula-sd.conf"
  68 char *configfile;
  69
  70
  71 static void usage()
  72 {
  73    fprintf(stderr,
  74 "\nVersion: " VERSION " (" BDATE ")\n\n"
  75 "Usage: bextract [-d debug_level] <bacula-archive> <directory-to-store-files>\n"
  76 "       -b <file>       specify a bootstrap file\n"
  77 "       -c <file>       specify a configuration file\n"
  78 "       -dnn            set debug level to nn\n"
  79 "       -e <file>       exclude list\n"
  80 "       -i <file>       include list\n"
  81 "       -V              specify Volume names (separated by |)\n"
  82 "       -?              print this message\n\n");
  83    exit(1);
  84 }
  85
  86
  87 int main (int argc, char *argv[])
  88 {
  89    int ch;
  90    FILE *fd;
  91    char line[1000];
  92    int got_inc = FALSE;
  93
  94    working_directory = "/tmp";
  95    my_name_is(argc, argv, "bextract");
  96    init_msg(NULL, NULL);              /* setup message handler */
  97
  98    memset(ff, 0, sizeof(FF_PKT));
  99    init_include_exclude_files(ff);
 100    binit(&bfd);
 101
 102    while ((ch = getopt(argc, argv, "b:c:d:e:i:?")) != -1) {
 103       switch (ch) {
 104       case 'b':                    /* bootstrap file */
 105          bsr = parse_bsr(NULL, optarg);
 106 //       dump_bsr(bsr);
 107          break;
 108
 109       case 'c':                    /* specify config file */
 110          if (configfile != NULL) {
 111             free(configfile);
 112          }
 113          configfile = bstrdup(optarg);
 114          break;
 115
 116       case 'd':                    /* debug level */
 117          debug_level = atoi(optarg);
 118          if (debug_level <= 0)
 119             debug_level = 1;
 120          break;
 121
 122       case 'e':                    /* exclude list */
 123          if ((fd = fopen(optarg, "r")) == NULL) {
 124             Pmsg2(0, "Could not open exclude file: %s, ERR=%s\n",
 125                optarg, strerror(errno));
 126             exit(1);
 127          }
 128          while (fgets(line, sizeof(line), fd) != NULL) {
 129             strip_trailing_junk(line);
 130             Dmsg1(900, "add_exclude %s\n", line);
 131             add_fname_to_exclude_list(ff, line);
 132          }
 133          fclose(fd);
 134          break;
 135
 136       case 'i':                    /* include list */
 137          if ((fd = fopen(optarg, "r")) == NULL) {
 138             Pmsg2(0, "Could not open include file: %s, ERR=%s\n",
 139                optarg, strerror(errno));
 140             exit(1);
 141          }
 142          while (fgets(line, sizeof(line), fd) != NULL) {
 143             strip_trailing_junk(line);
 144             Dmsg1(900, "add_include %s\n", line);
 145             add_fname_to_include_list(ff, 0, line);
 146          }
 147          fclose(fd);
 148          got_inc = TRUE;
 149          break;
 150
 151       case 'V':                    /* Volume name */
 152          VolumeName = optarg;
 153          break;
 154
 155       case '?':
 156       default:
 157          usage();
 158
 159       } /* end switch */
 160    } /* end while */
 161    argc -= optind;
 162    argv += optind;
 163
 164    if (argc != 2) {
 165       Pmsg0(0, "Wrong number of arguments: \n");
 166       usage();
 167    }
 168
 169    if (configfile == NULL) {
 170       configfile = bstrdup(CONFIG_FILE);
 171    }
 172
 173    parse_config(configfile);
 174
 175    if (!got_inc) {                            /* If no include file, */
 176       add_fname_to_include_list(ff, 0, "/");  /*   include everything */
 177    }
 178
 179    where = argv[1];
 180    do_extract(argv[0]);
 181
 182    if (bsr) {
 183       free_bsr(bsr);
 184    }
 185    if (prog_name_msg) {
 186       Pmsg1(000, "%d Program Name and/or Program Data Stream records ignored.\n",
 187          prog_name_msg);
 188    }
 189    if (win32_data_msg) {
 190       Pmsg1(000, "%d Win32 data or Win32 gzip data stream records. Ignored.\n",
 191          win32_data_msg);
 192    }
 193    return 0;
 194 }
 195
 196 static void do_extract(char *devname)
 197 {
 198    struct stat statp;
 199    jcr = setup_jcr("bextract", devname, bsr, VolumeName);
 200    dev = setup_to_access_device(jcr, 1);    /* acquire for read */
 201    if (!dev) {
 202       exit(1);
 203    }
 204
 205    /* Make sure where directory exists and that it is a directory */
 206    if (stat(where, &statp) < 0) {
 207       Emsg2(M_ERROR_TERM, 0, "Cannot stat %s. It must exist. ERR=%s\n",
 208          where, strerror(errno));
 209    }
 210    if (!S_ISDIR(statp.st_mode)) {
 211       Emsg1(M_ERROR_TERM, 0, "%s must be a directory.\n", where);
 212    }
 213
 214    free(jcr->where);
 215    jcr->where = bstrdup(where);
 216    attr = new_attr();
 217
 218    compress_buf = get_memory(compress_buf_size);
 219
 220    read_records(jcr, dev, record_cb, mount_next_read_volume);
 221    /* If output file is still open, it was the last one in the
 222     * archive since we just hit an end of file, so close the file.
 223     */
 224    if (is_bopen(&bfd)) {
 225       set_attributes(jcr, attr, &bfd);
 226    }
 227    release_device(jcr, dev);
 228
 229    free_attr(attr);
 230    term_dev(dev);
 231    free_jcr(jcr);
 232    printf("%u files restored.\n", num_files);
 233    return;
 234 }
 235
 236 /*
 237  * Called here for each record from read_records()
 238  */
 239 static void record_cb(JCR *jcr, DEVICE *dev, DEV_BLOCK *block, DEV_RECORD *rec)
 240 {
 241    int stat;
 242
 243    if (rec->FileIndex < 0) {
 244       return;                         /* we don't want labels */
 245    }
 246
 247    /* File Attributes stream */
 248
 249    switch (rec->Stream) {
 250    case STREAM_UNIX_ATTRIBUTES:
 251    case STREAM_UNIX_ATTRIBUTES_EX:
 252
 253       /* If extracting, it was from previous stream, so
 254        * close the output file.
 255        */
 256       if (extract) {
 257          if (!is_bopen(&bfd)) {
 258             Emsg0(M_ERROR, 0, _("Logic error output file should be open but is not.\n"));
 259          }
 260          extract = FALSE;
 261          set_attributes(jcr, attr, &bfd);
 262       }
 263
 264       if (!unpack_attributes_record(jcr, rec->Stream, rec->data, attr)) {
 265          Emsg0(M_ERROR_TERM, 0, _("Cannot continue.\n"));
 266       }
 267
 268       if (attr->file_index != rec->FileIndex) {
 269          Emsg2(M_ERROR_TERM, 0, _("Record header file index %ld not equal record index %ld\n"),
 270             rec->FileIndex, attr->file_index);
 271       }
 272
 273       if (file_is_included(ff, attr->fname) && !file_is_excluded(ff, attr->fname)) {
 274          uint32_t LinkFI;
 275
 276          decode_stat(attr->attr, &attr->statp, &LinkFI);
 277
 278          build_attr_output_fnames(jcr, attr);
 279
 280          extract = FALSE;
 281          stat = create_file(jcr, attr, &bfd, REPLACE_ALWAYS);
 282          switch (stat) {
 283          case CF_ERROR:
 284          case CF_SKIP:
 285             break;
 286          case CF_EXTRACT:
 287             extract = TRUE;
 288             print_ls_output(jcr, attr);
 289             num_files++;
 290             fileAddr = 0;
 291             break;
 292          case CF_CREATED:
 293             set_attributes(jcr, attr, &bfd);
 294             print_ls_output(jcr, attr);
 295             num_files++;
 296             fileAddr = 0;
 297             break;
 298          }
 299       }
 300
 301    /* Windows Backup data stream */
 302    case STREAM_WIN32_DATA:
 303       if (!is_win32_backup()) {
 304          if (!non_support_data) {
 305             Jmsg(jcr, M_ERROR, 0, _("Win32 backup data not supported on this Client.\n"));
 306          }
 307          extract = FALSE;
 308          non_support_data++;
 309          return;
 310       }
 311       goto extract_data;
 312
 313
 314    /* Data stream and extracting */
 315    case STREAM_FILE_DATA:
 316    case STREAM_SPARSE_DATA:
 317
 318 extract_data:
 319       if (extract) {
 320          if (rec->Stream == STREAM_SPARSE_DATA) {
 321             ser_declare;
 322             uint64_t faddr;
 323             wbuf = rec->data + SPARSE_FADDR_SIZE;
 324             wsize = rec->data_len - SPARSE_FADDR_SIZE;
 325             ser_begin(rec->data, SPARSE_FADDR_SIZE);
 326             unser_uint64(faddr);
 327             if (fileAddr != faddr) {
 328                fileAddr = faddr;
 329                if (blseek(&bfd, (off_t)fileAddr, SEEK_SET) < 0) {
 330                   Emsg2(M_ERROR_TERM, 0, _("Seek error on %s: %s\n"),
 331                      attr->ofname, strerror(errno));
 332                }
 333             }
 334          } else {
 335             wbuf = rec->data;
 336             wsize = rec->data_len;
 337          }
 338          total += wsize;
 339          Dmsg2(8, "Write %u bytes, total=%u\n", wsize, total);
 340          if ((uint32_t)bwrite(&bfd, wbuf, wsize) != wsize) {
 341             Emsg2(M_ERROR_TERM, 0, _("Write error on %s: %s\n"),
 342                attr->ofname, strerror(errno));
 343          }
 344          fileAddr += wsize;
 345       }
 346
 347    /* Windows Backup GZIP data stream */
 348    case STREAM_WIN32_GZIP_DATA:
 349       if (!is_win32_backup()) {
 350          if (!non_support_attr) {
 351             Jmsg(jcr, M_ERROR, 0, _("Win32 GZIP backup data not supported on this Client.\n"));
 352          }
 353          extract = FALSE;
 354          non_support_attr++;
 355          return;
 356       }
 357       /* Fall through desired */
 358
 359    /* GZIP data stream */
 360    case STREAM_GZIP_DATA:
 361    case STREAM_SPARSE_GZIP_DATA:
 362 #ifdef HAVE_LIBZ
 363       if (extract) {
 364          uLongf compress_len;
 365          int stat;
 366
 367          if (rec->Stream == STREAM_SPARSE_GZIP_DATA) {
 368             ser_declare;
 369             uint64_t faddr;
 370             wbuf = rec->data + SPARSE_FADDR_SIZE;
 371             wsize = rec->data_len - SPARSE_FADDR_SIZE;
 372             ser_begin(rec->data, SPARSE_FADDR_SIZE);
 373             unser_uint64(faddr);
 374             if (fileAddr != faddr) {
 375                fileAddr = faddr;
 376                if (blseek(&bfd, (off_t)fileAddr, SEEK_SET) < 0) {
 377                   Emsg2(M_ERROR, 0, _("Seek error on %s: %s\n"),
 378                      attr->ofname, strerror(errno));
 379                }
 380             }
 381          } else {
 382             wbuf = rec->data;
 383             wsize = rec->data_len;
 384          }
 385          compress_len = compress_buf_size;
 386          if ((stat=uncompress((Bytef *)compress_buf, &compress_len,
 387                (const Bytef *)wbuf, (uLong)wsize) != Z_OK)) {
 388             Emsg1(M_ERROR_TERM, 0, _("Uncompression error. ERR=%d\n"), stat);
 389          }
 390
 391          Dmsg2(100, "Write uncompressed %d bytes, total before write=%d\n", compress_len, total);
 392          if ((uLongf)bwrite(&bfd, compress_buf, (size_t)compress_len) != compress_len) {
 393             Pmsg0(0, "===Write error===\n");
 394             Emsg2(M_ERROR_TERM, 0, _("Write error on %s: %s\n"),
 395                attr->ofname, strerror(errno));
 396          }
 397          total += compress_len;
 398          fileAddr += compress_len;
 399          Dmsg2(100, "Compress len=%d uncompressed=%d\n", rec->data_len,
 400             compress_len);
 401       }
 402 #else
 403       if (extract) {
 404          Emsg0(M_ERROR, 0, "GZIP data stream found, but GZIP not configured!\n");
 405          extract = FALSE;
 406          return;
 407       }
 408 #endif
 409
 410    case STREAM_MD5_SIGNATURE:
 411    case STREAM_SHA1_SIGNATURE:
 412       break;
 413
 414    case STREAM_PROGRAM_NAMES:
 415    case STREAM_PROGRAM_DATA:
 416       if (!prog_name_msg) {
 417          Pmsg0(000, "Got Program Name or Data Stream. Ignored.\n");
 418          prog_name_msg++;
 419       }
 420       break;
 421
 422    default:
 423       /* If extracting, wierd stream (not 1 or 2), close output file anyway */
 424       if (extract) {
 425          if (!is_bopen(&bfd)) {
 426             Emsg0(M_ERROR, 0, "Logic error output file should be open but is not.\n");
 427          }
 428          extract = FALSE;
 429          set_attributes(jcr, attr, &bfd);
 430       }
 431       Jmsg(jcr, M_ERROR, 0, _("Unknown stream=%d ignored. This shouldn't happen!\n"),
 432          rec->Stream);
 433       break;
 434
 435    } /* end switch */
 436 }
 437
 438
 439
 440
 441 /* Dummies to replace askdir.c */
 442 int     dir_get_volume_info(JCR *jcr, int writing) { return 1;}
 443 int     dir_find_next_appendable_volume(JCR *jcr) { return 1;}
 444 int     dir_update_volume_info(JCR *jcr, VOLUME_CAT_INFO *vol, int relabel) { return 1; }
 445 int     dir_create_jobmedia_record(JCR *jcr) { return 1; }
 446 int     dir_ask_sysop_to_mount_next_volume(JCR *jcr, DEVICE *dev) { return 1; }
 447 int     dir_update_file_attributes(JCR *jcr, DEV_RECORD *rec) { return 1;}
 448 int     dir_send_job_status(JCR *jcr) {return 1;}
 449
 450
 451 int dir_ask_sysop_to_mount_volume(JCR *jcr, DEVICE *dev)
 452 {
 453    fprintf(stderr, "Mount Volume %s on device %s and press return when ready: ",
 454       jcr->VolumeName, dev_name(dev));
 455    getchar();
 456    return 1;
 457 }