git.sur5r.net Git - bacula/bacula/blob - bacula/src/filed/restore.c

   1 /*
   2  *  Bacula File Daemon  restore.c Restorefiles.
   3  *
   4  *    Kern Sibbald, November MM
   5  *
   6  *   Version $Id$
   7  *
   8  */
   9 /*
  10    Copyright (C) 2000-2003 Kern Sibbald and John Walker
  11
  12    This program is free software; you can redistribute it and/or
  13    modify it under the terms of the GNU General Public License as
  14    published by the Free Software Foundation; either version 2 of
  15    the License, or (at your option) any later version.
  16
  17    This program is distributed in the hope that it will be useful,
  18    but WITHOUT ANY WARRANTY; without even the implied warranty of
  19    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  20    General Public License for more details.
  21
  22    You should have received a copy of the GNU General Public
  23    License along with this program; if not, write to the Free
  24    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  25    MA 02111-1307, USA.
  26
  27  */
  28
  29 #include "bacula.h"
  30 #include "filed.h"
  31
  32 /* Data received from Storage Daemon */
  33 static char rec_header[] = "rechdr %ld %ld %ld %ld %ld";
  34
  35 /* Forward referenced functions */
  36 static void print_ls_output(JCR *jcr, char *fname, char *lname, int type, struct stat *statp);
  37
  38 #define RETRY 10                      /* retry wait time */
  39
  40 /*
  41  * Restore the requested files.
  42  *
  43  */
  44 void do_restore(JCR *jcr)
  45 {
  46    int wherelen;
  47    BSOCK *sd;
  48    POOLMEM *fname;                    /* original file name */
  49    POOLMEM *ofile;                    /* output name with possible prefix */
  50    POOLMEM *lname;                    /* link name with possible prefix */
  51    POOLMEM *attribsEx;                /* Extended attributes (Win32) */
  52    int32_t stream;
  53    uint32_t size;
  54    uint32_t VolSessionId, VolSessionTime, file_index;
  55    uint32_t record_file_index;
  56    struct stat statp;
  57    int extract = FALSE;
  58    int ofd = -1;
  59    int type, stat;
  60    uint32_t total = 0;                /* Job total but only 32 bits for debug */
  61    char *wbuf;                        /* write buffer */
  62    uint32_t wsize;                    /* write size */
  63    uint64_t fileAddr = 0;             /* file write address */
  64
  65    wherelen = strlen(jcr->where);
  66
  67    sd = jcr->store_bsock;
  68    set_jcr_job_status(jcr, JS_Running);
  69
  70    if (!bnet_set_buffer_size(sd, MAX_NETWORK_BUFFER_SIZE, BNET_SETBUF_READ)) {
  71       set_jcr_job_status(jcr, JS_ErrorTerminated);
  72       return;
  73    }
  74    jcr->buf_size = sd->msglen;
  75
  76    fname = get_pool_memory(PM_FNAME);
  77    ofile = get_pool_memory(PM_FNAME);
  78    lname = get_pool_memory(PM_FNAME);
  79    attribsEx = get_pool_memory(PM_FNAME);
  80
  81 #ifdef HAVE_LIBZ
  82    uint32_t compress_buf_size = jcr->buf_size + 12 + ((jcr->buf_size+999) / 1000) + 100;
  83    jcr->compress_buf = (char *)bmalloc(compress_buf_size);
  84 #endif
  85
  86    /*
  87     * Get a record from the Storage daemon. We are guaranteed to
  88     *   receive records in the following order:
  89     *   1. Stream record header
  90     *   2. Stream data
  91     *        a. Attributes (Unix or Win32)
  92     *    or  b. File data for the file
  93     *    or  c. Possibly MD5 record
  94     *   3. Repeat step 1
  95     */
  96    while (bnet_recv(sd) >= 0 && !job_cancelled(jcr)) {
  97       /*
  98        * First we expect a Stream Record Header
  99        */
 100       if (sscanf(sd->msg, rec_header, &VolSessionId, &VolSessionTime, &file_index,
 101           &stream, &size) != 5) {
 102          Jmsg1(jcr, M_FATAL, 0, _("Record header scan error: %s\n"), sd->msg);
 103          goto bail_out;
 104       }
 105       Dmsg2(30, "Got hdr: FilInx=%d Stream=%d.\n", file_index, stream);
 106
 107       /*
 108        * Now we expect the Stream Data
 109        */
 110       if (bnet_recv(sd) < 0) {
 111          Jmsg1(jcr, M_FATAL, 0, _("Data record error. ERR=%s\n"), bnet_strerror(sd));
 112          goto bail_out;
 113       }
 114       if (size != (uint32_t)sd->msglen) {
 115          Jmsg2(jcr, M_FATAL, 0, _("Actual data size %d not same as header %d\n"), sd->msglen, size);
 116          goto bail_out;
 117       }
 118       Dmsg1(30, "Got stream data, len=%d\n", sd->msglen);
 119
 120       /* File Attributes stream */
 121       if (stream == STREAM_UNIX_ATTRIBUTES || stream == STREAM_WIN32_ATTRIBUTES) {
 122          char *ap, *lp, *fp, *apex;
 123
 124          Dmsg1(30, "Stream=Unix Attributes. extract=%d\n", extract);
 125          /* If extracting, it was from previous stream, so
 126           * close the output file.
 127           */
 128          if (extract) {
 129             if (ofd < 0) {
 130                Jmsg0(jcr, M_ERROR, 0, _("Logic error output file should be open\n"));
 131             }
 132             set_attributes(jcr, fname, ofile, lname, type, stream,
 133                            &statp, attribsEx, &ofd);
 134             extract = FALSE;
 135             Dmsg0(30, "Stop extracting.\n");
 136          }
 137
 138          if ((int)sizeof_pool_memory(fname) <  sd->msglen) {
 139             fname = realloc_pool_memory(fname, sd->msglen + 1);
 140          }
 141          if ((int)sizeof_pool_memory(ofile) < sd->msglen + wherelen + 1) {
 142             ofile = realloc_pool_memory(ofile, sd->msglen + wherelen + 1);
 143          }
 144          if ((int)sizeof_pool_memory(lname) < sd->msglen + wherelen + 1) {
 145             lname = realloc_pool_memory(lname, sd->msglen + wherelen + 1);
 146          }
 147          *fname = 0;
 148          *lname = 0;
 149
 150          /*
 151           * An Attributes record consists of:
 152           *    File_index
 153           *    Type   (FT_types)
 154           *    Filename
 155           *    Attributes
 156           *    Link name (if file linked i.e. FT_LNK)
 157           *    Extended attributes (Win32)
 158           *
 159           */
 160          Dmsg1(100, "Attr: %s\n", sd->msg);
 161          if (sscanf(sd->msg, "%d %d", &record_file_index, &type) != 2) {
 162             Jmsg(jcr, M_FATAL, 0, _("Error scanning attributes: %s\n"), sd->msg);
 163             Dmsg1(100, "\nError scanning attributes. %s\n", sd->msg);
 164             goto bail_out;
 165          }
 166          Dmsg2(100, "Got Attr: FilInx=%d type=%d\n", record_file_index, type);
 167          if (record_file_index != file_index) {
 168             Jmsg(jcr, M_FATAL, 0, _("Record header file index %ld not equal record index %ld\n"),
 169                file_index, record_file_index);
 170             Dmsg0(100, "File index error\n");
 171             goto bail_out;
 172          }
 173          ap = sd->msg;
 174          while (*ap++ != ' ')         /* skip record file index */
 175             ;
 176          while (*ap++ != ' ')         /* skip type */
 177             ;
 178          /* Save filename and position to attributes */
 179          fp = fname;
 180          while (*ap != 0) {
 181             *fp++  = *ap++;           /* copy filename to fname */
 182          }
 183          *fp = *ap++;                 /* terminate filename & point to attribs */
 184
 185          /* Skip to Link name */
 186          if (type == FT_LNK || type == FT_LNKSAVED) {
 187             lp = ap;
 188             while (*lp++ != 0) {
 189                ;
 190             }
 191          } else {
 192             lp = "";
 193          }
 194
 195          if (stream == STREAM_WIN32_ATTRIBUTES) {
 196             apex = ap;                   /* start at attributes */
 197             while (*apex++ != 0) {       /* skip attributes */
 198                ;
 199             }
 200             while (*apex++ != 0) {       /* skip link name */
 201                ;
 202             }
 203             pm_strcpy(&attribsEx, apex); /* make a copy */
 204          } else {
 205             *attribsEx = 0;              /* no extended attributes */
 206          }
 207
 208          Dmsg3(200, "File %s\nattrib=%s\nattribsEx=%s\n", fname, ap, attribsEx);
 209
 210          decode_stat(ap, &statp);
 211          /*
 212           * Prepend the where directory so that the
 213           * files are put where the user wants.
 214           *
 215           * We do a little jig here to handle Win32 files with
 216           *   a drive letter -- we simply strip the drive: from
 217           *   every filename if a prefix is supplied.
 218           *
 219           */
 220          if (jcr->where[0] == 0) {
 221             strcpy(ofile, fname);
 222             strcpy(lname, lp);
 223          } else {
 224             char *fn;
 225             strcpy(ofile, jcr->where);  /* copy prefix */
 226             if (win32_client && fname[1] == ':') {
 227                fn = fname+2;          /* skip over drive: */
 228             } else {
 229                fn = fname;            /* take whole name */
 230             }
 231             /* Ensure where is terminated with a slash */
 232             if (jcr->where[wherelen-1] != '/' && fn[0] != '/') {
 233                strcat(ofile, "/");
 234             }
 235             strcat(ofile, fn);        /* copy rest of name */
 236             /* Fixup link name */
 237             if (type == FT_LNK || type == FT_LNKSAVED) {
 238                if (lp[0] == '/') {      /* if absolute path */
 239                   strcpy(lname, jcr->where);
 240                }
 241                if (win32_client && lp[1] == ':') {
 242                   strcat(lname, lp+2); /* copy rest of name */
 243                } else {
 244                   strcat(lname, lp);   /* On Unix systems we take everything */
 245                }
 246             }
 247          }
 248
 249          Dmsg1(30, "Outfile=%s\n", ofile);
 250
 251          extract = FALSE;
 252          stat = create_file(jcr, fname, ofile, lname, type,
 253                             stream, &statp, attribsEx, &ofd, jcr->replace);
 254          switch (stat) {
 255          case CF_ERROR:
 256          case CF_SKIP:
 257             break;
 258          case CF_EXTRACT:
 259             extract = TRUE;
 260             P(jcr->mutex);
 261             pm_strcpy(&jcr->last_fname, ofile);
 262             V(jcr->mutex);
 263             /* Fall-through wanted */
 264          case CF_CREATED:
 265             jcr->JobFiles++;
 266             fileAddr = 0;
 267             print_ls_output(jcr, ofile, lname, type, &statp);
 268             break;
 269          }
 270
 271          jcr->num_files_examined++;
 272
 273       /* Data stream */
 274       } else if (stream == STREAM_FILE_DATA || stream == STREAM_SPARSE_DATA) {
 275          if (extract) {
 276             if (stream == STREAM_SPARSE_DATA) {
 277                ser_declare;
 278                uint64_t faddr;
 279                char ec1[50];
 280
 281                wbuf = sd->msg + SPARSE_FADDR_SIZE;
 282                wsize = sd->msglen - SPARSE_FADDR_SIZE;
 283                ser_begin(sd->msg, SPARSE_FADDR_SIZE);
 284                unser_uint64(faddr);
 285                if (fileAddr != faddr) {
 286                   fileAddr = faddr;
 287                   if (lseek(ofd, (off_t)fileAddr, SEEK_SET) < 0) {
 288                      Jmsg3(jcr, M_ERROR, 0, _("Seek to %s error on %s: ERR=%s\n"),
 289                          edit_uint64(fileAddr, ec1), ofile, strerror(errno));
 290                      goto bail_out;
 291                   }
 292                }
 293             } else {
 294                wbuf = sd->msg;
 295                wsize = sd->msglen;
 296             }
 297             Dmsg2(30, "Write %u bytes, total before write=%u\n", wsize, total);
 298             if ((uint32_t)write(ofd, wbuf, wsize) != wsize) {
 299                Dmsg0(0, "===Write error===\n");
 300                Jmsg2(jcr, M_ERROR, 0, _("Write error on %s: %s\n"), ofile, strerror(errno));
 301                goto bail_out;
 302             }
 303             total += wsize;
 304             jcr->JobBytes += wsize;
 305             fileAddr += wsize;
 306          }
 307
 308       /* GZIP data stream */
 309       } else if (stream == STREAM_GZIP_DATA || stream == STREAM_SPARSE_GZIP_DATA) {
 310 #ifdef HAVE_LIBZ
 311          if (extract) {
 312             ser_declare;
 313             uLong compress_len;
 314             uint64_t faddr;
 315             char ec1[50];
 316             int stat;
 317
 318             if (stream == STREAM_SPARSE_GZIP_DATA) {
 319                wbuf = sd->msg + SPARSE_FADDR_SIZE;
 320                wsize = sd->msglen - SPARSE_FADDR_SIZE;
 321                ser_begin(sd->msg, SPARSE_FADDR_SIZE);
 322                unser_uint64(faddr);
 323                if (fileAddr != faddr) {
 324                   fileAddr = faddr;
 325                   if (lseek(ofd, (off_t)fileAddr, SEEK_SET) < 0) {
 326                      Jmsg3(jcr, M_ERROR, 0, _("Seek to %s error on %s: ERR=%s\n"),
 327                          edit_uint64(fileAddr, ec1), ofile, strerror(errno));
 328                      goto bail_out;
 329                   }
 330                }
 331             } else {
 332                wbuf = sd->msg;
 333                wsize = sd->msglen;
 334             }
 335             compress_len = compress_buf_size;
 336             Dmsg2(100, "Comp_len=%d msglen=%d\n", compress_len, wsize);
 337             if ((stat=uncompress((Byte *)jcr->compress_buf, &compress_len,
 338                   (const Byte *)wbuf, (uLong)wsize)) != Z_OK) {
 339                Jmsg(jcr, M_ERROR, 0, _("Uncompression error. ERR=%d\n"), stat);
 340                goto bail_out;
 341             }
 342
 343             Dmsg2(100, "Write uncompressed %d bytes, total before write=%d\n", compress_len, total);
 344             if ((uLong)write(ofd, jcr->compress_buf, compress_len) != compress_len) {
 345                Dmsg0(0, "===Write error===\n");
 346                Jmsg2(jcr, M_ERROR, 0, "Write error on %s: %s\n", ofile, strerror(errno));
 347                goto bail_out;
 348             }
 349             total += compress_len;
 350             jcr->JobBytes += compress_len;
 351             fileAddr += compress_len;
 352          }
 353 #else
 354          if (extract) {
 355             Jmsg(jcr, M_ERROR, 0, "GZIP data stream found, but GZIP not configured!\n");
 356             goto bail_out;
 357          }
 358 #endif
 359       /* If extracting, wierd stream (not 1 or 2), close output file anyway */
 360       } else if (extract) {
 361          Dmsg1(30, "Found wierd stream %d\n", stream);
 362          if (ofd < 0) {
 363             Jmsg0(jcr, M_ERROR, 0, _("Logic error output file should be open\n"));
 364          }
 365          set_attributes(jcr, fname, ofile, lname, type, stream,
 366                         &statp, attribsEx, &ofd);
 367          extract = FALSE;
 368       } else if (stream != STREAM_MD5_SIGNATURE) {
 369          Dmsg2(0, "None of above!!! stream=%d data=%s\n", stream,sd->msg);
 370       }
 371    }
 372
 373    /* If output file is still open, it was the last one in the
 374     * archive since we just hit an end of file, so close the file.
 375     */
 376    if (ofd >= 0) {
 377       set_attributes(jcr, fname, ofile, lname, type, stream,
 378                      &statp, attribsEx, &ofd);
 379    }
 380    set_jcr_job_status(jcr, JS_Terminated);
 381    goto ok_out;
 382
 383 bail_out:
 384    set_jcr_job_status(jcr, JS_ErrorTerminated);
 385 ok_out:
 386    if (jcr->compress_buf) {
 387       free(jcr->compress_buf);
 388       jcr->compress_buf = NULL;
 389    }
 390    free_pool_memory(fname);
 391    free_pool_memory(ofile);
 392    free_pool_memory(lname);
 393    free_pool_memory(attribsEx);
 394    Dmsg2(10, "End Do Restore. Files=%d Bytes=%" lld "\n", jcr->JobFiles,
 395       jcr->JobBytes);
 396 }
 397
 398 extern char *getuser(uid_t uid);
 399 extern char *getgroup(gid_t gid);
 400
 401 /*
 402  * Print an ls style message, also send INFO
 403  */
 404 static void print_ls_output(JCR *jcr, char *fname, char *lname, int type, struct stat *statp)
 405 {
 406    char buf[2000];
 407    char ec1[30];
 408    char *p, *f;
 409    int n;
 410
 411    p = encode_mode(statp->st_mode, buf);
 412    n = sprintf(p, "  %2d ", (uint32_t)statp->st_nlink);
 413    p += n;
 414    n = sprintf(p, "%-8.8s %-8.8s", getuser(statp->st_uid), getgroup(statp->st_gid));
 415    p += n;
 416    n = sprintf(p, "%8.8s ", edit_uint64(statp->st_size, ec1));
 417    p += n;
 418    p = encode_time(statp->st_ctime, p);
 419    *p++ = ' ';
 420    *p++ = ' ';
 421    for (f=fname; *f && (p-buf) < (int)sizeof(buf)-10; ) {
 422       *p++ = *f++;
 423    }
 424    if (type == FT_LNK) {
 425       *p++ = ' ';
 426       *p++ = '-';
 427       *p++ = '>';
 428       *p++ = ' ';
 429       /* Copy link name */
 430       for (f=lname; *f && (p-buf) < (int)sizeof(buf)-10; ) {
 431          *p++ = *f++;
 432       }
 433    }
 434    *p++ = '\n';
 435    *p = 0;
 436    Dmsg0(20, buf);
 437    Jmsg(jcr, M_INFO, 0, "%s", buf);
 438 }