LCOV - code coverage report
Current view: top level - src/client - restore.c (source / functions) Hit Total Coverage
Test: burp-coverage-clean.info Lines: 161 285 56.5 %
Date: 2018-05-29 13:53:20 Functions: 9 13 69.2 %

          Line data    Source code
       1             : #include "../burp.h"
       2             : #include "../alloc.h"
       3             : #include "../asfd.h"
       4             : #include "../async.h"
       5             : #include "../attribs.h"
       6             : #include "../berrno.h"
       7             : #include "../cmd.h"
       8             : #include "../cntr.h"
       9             : #include "../fsops.h"
      10             : #include "../handy.h"
      11             : #include "../pathcmp.h"
      12             : #include "../log.h"
      13             : #include "../prepend.h"
      14             : #include "../protocol2/blk.h"
      15             : #include "cvss.h"
      16             : #include "protocol1/restore.h"
      17             : #include "protocol2/restore.h"
      18             : #include "restore.h"
      19             : 
      20           7 : int restore_interrupt(struct asfd *asfd,
      21             :         struct sbuf *sb, const char *msg, struct cntr *cntr,
      22             :         enum protocol protocol)
      23             : {
      24           7 :         int ret=0;
      25           7 :         char *path=NULL;
      26           7 :         struct iobuf *rbuf=asfd->rbuf;
      27             : 
      28           7 :         if(cntr)
      29             :         {
      30           0 :                 cntr_add(cntr, CMD_WARNING, 1);
      31           0 :                 logp("WARNING: %s\n", msg);
      32           0 :                 if(asfd->write_str(asfd, CMD_WARNING, msg)) goto end;
      33             :         }
      34             : 
      35             :         // If it is file data, get the server
      36             :         // to interrupt the flow and move on.
      37           7 :         if(!iobuf_is_filedata(&sb->path)
      38           0 :           && !iobuf_is_vssdata(&sb->path))
      39             :                 return 0;
      40             : 
      41           7 :         if(protocol==PROTO_1)
      42           0 :                 path=sb->protocol1->datapth.buf;
      43           7 :         else if(protocol==PROTO_2)
      44           7 :                 path=sb->path.buf;
      45             : 
      46           7 :         if(!path) return 0;
      47             : 
      48           7 :         if(asfd->write_str(asfd, CMD_INTERRUPT, path))
      49             :                 goto end;
      50             : 
      51             :         // Read to the end file marker.
      52             :         while(1)
      53             :         {
      54         714 :                 iobuf_free_content(rbuf);
      55         714 :                 if(asfd->read(asfd))
      56             :                         goto end;
      57         714 :                 if(!rbuf->len) continue;
      58             : 
      59         714 :                 switch(rbuf->cmd)
      60             :                 {
      61             :                         case CMD_APPEND:
      62             :                         case CMD_DATA:
      63         707 :                                 continue;
      64             :                         case CMD_END_FILE:
      65             :                                 ret=0;
      66             :                                 goto end;
      67             :                         default:
      68           0 :                                 iobuf_log_unexpected(rbuf, __func__);
      69           0 :                                 goto end;
      70             :                 }
      71             :         }
      72             : end:
      73           7 :         iobuf_free_content(rbuf);
      74           7 :         return ret;
      75             : }
      76             : 
      77           6 : static int make_link(
      78             : #ifdef HAVE_WIN32
      79             :         struct asfd *asfd,
      80             :         struct cntr *cntr,
      81             : #endif
      82             :         const char *fname, const char *lnk,
      83             :         enum cmd cmd, const char *restore_prefix)
      84             : {
      85           6 :         int ret=-1;
      86             : 
      87             : #ifdef HAVE_WIN32
      88             :         logw(asfd, cntr, "windows seems not to support hardlinks or symlinks\n");
      89             : #else
      90           6 :         unlink(fname);
      91           6 :         if(cmd==CMD_HARD_LINK)
      92             :         {
      93           1 :                 char *flnk=NULL;
      94           1 :                 if(!(flnk=prepend_s(restore_prefix, lnk)))
      95             :                 {
      96           0 :                         log_out_of_memory(__func__);
      97           0 :                         return -1;
      98             :                 }
      99             :                 //printf("%s -> %s\n", fname, flnk);
     100           1 :                 ret=link(flnk, fname);
     101           1 :                 free_w(&flnk);
     102             :         }
     103           5 :         else if(cmd==CMD_SOFT_LINK)
     104             :         {
     105             :                 //printf("%s -> %s\n", fname, lnk);
     106           5 :                 ret=symlink(lnk, fname);
     107             :         }
     108             :         else
     109             :         {
     110           0 :                 logp("unexpected link command: %c\n", cmd);
     111           0 :                 ret=-1;
     112             :         }
     113             : #endif
     114             : 
     115           6 :         if(ret) logp("could not %slink %s -> %s: %s\n",
     116             :                 cmd==CMD_HARD_LINK?"hard":"sym",
     117           0 :                 fname, lnk, strerror(errno));
     118             : 
     119             :         return ret;
     120             : }
     121             : 
     122             : // FIX THIS: Maybe should be in bfile.c.
     123          18 : enum ofr_e open_for_restore(struct asfd *asfd, struct BFILE *bfd, const char *path,
     124             :         struct sbuf *sb, int vss_restore, struct cntr *cntr,
     125             :         enum protocol protocol)
     126             : {
     127             :         static int flags;
     128          18 :         if(bfd->mode!=BF_CLOSED)
     129             :         {
     130             : #ifdef HAVE_WIN32
     131             :                 if(bfd->path && !strcmp(bfd->path, path))
     132             :                 {
     133             :                         // Already open after restoring the VSS data.
     134             :                         // Time now for the actual file data.
     135             :                         return OFR_OK;
     136             :                 }
     137             :                 else
     138             :                 {
     139             : #endif
     140          12 :                         if(bfd->close(bfd, asfd))
     141             :                         {
     142           0 :                                 logp("error closing %s in %s()\n",
     143             :                                         path, __func__);
     144           0 :                                 return OFR_ERROR;
     145             :                         }
     146             : #ifdef HAVE_WIN32
     147             :                 }
     148             : #endif
     149             :         }
     150             : 
     151             : #ifdef HAVE_WIN32
     152             :         // Some massive hacks to work around times that winattr was not
     153             :         // getting set correctly inside server side backups.
     154             :         // The EFS one will stop burp segfaulting when restoring affected
     155             :         // EFS files.
     156             :         if(sb->path.cmd==CMD_EFS_FILE)
     157             :                 sb->winattr |= FILE_ATTRIBUTE_ENCRYPTED;
     158             :         if(S_ISDIR(sb->statp.st_mode))
     159             :                 sb->winattr |= FILE_ATTRIBUTE_DIRECTORY;
     160             : #endif
     161             : 
     162          18 :         bfile_init(bfd, sb->winattr, cntr);
     163          18 :         bfd->set_attribs_on_close=1;
     164             : #ifdef HAVE_WIN32
     165             :         bfd->set_win32_api(bfd, vss_restore);
     166             : #else
     167             :         // Abuse the vss_restore option to mean vss_strip on non-Windows.
     168          18 :         bfd->set_vss_strip(bfd, !vss_restore);
     169             : #endif
     170          18 :         if(S_ISDIR(sb->statp.st_mode))
     171             :         {
     172             :                 // Windows directories are treated as having file data.
     173           0 :                 flags=O_WRONLY|O_BINARY;
     174           0 :                 mkdir(path, 0777);
     175             :         }
     176             :         else
     177          18 :                 flags=O_WRONLY|O_BINARY|O_CREAT|O_TRUNC;
     178             : 
     179          18 :         if(bfd->open(bfd, asfd, path, flags, S_IRUSR | S_IWUSR))
     180             :         {
     181             :                 struct berrno be;
     182           0 :                 berrno_init(&be);
     183           0 :                 char msg[256]="";
     184           0 :                 snprintf(msg, sizeof(msg), "Could not open for writing %s: %s",
     185           0 :                         path, berrno_bstrerror(&be, errno));
     186           0 :                 if(restore_interrupt(asfd, sb, msg, cntr, protocol))
     187             :                         return OFR_ERROR;
     188           0 :                 return OFR_CONTINUE;
     189             :         }
     190             :         // Add attributes to bfd so that they can be set when it is closed.
     191          18 :         bfd->winattr=sb->winattr;
     192          18 :         memcpy(&bfd->statp, &sb->statp, sizeof(struct stat));
     193          18 :         return OFR_OK;
     194             : }
     195             : 
     196             : static char *build_msg(const char *text, const char *param)
     197             : {
     198             :         static char msg[256]="";
     199           0 :         snprintf(msg, sizeof(msg), text, param);
     200             :         return msg;
     201             : }
     202             : 
     203             : #ifndef HAVE_WIN32
     204           0 : static void do_logw(struct asfd *asfd, struct cntr *cntr,
     205             :         const char *text, const char *param)
     206             : {
     207           0 :         logw(asfd, cntr, "%s", build_msg(text, param));
     208           0 : }
     209             : #endif
     210             : 
     211             : static int warn_and_interrupt(struct asfd *asfd, struct sbuf *sb,
     212             :         struct cntr *cntr, enum protocol protocol,
     213             :         const char *text, const char *param)
     214             : {
     215           0 :         return restore_interrupt(asfd, sb, build_msg(text, param), cntr,
     216             :                 protocol);
     217             : }
     218             : 
     219           0 : static int restore_special(struct asfd *asfd, struct sbuf *sb,
     220             :         const char *fname, enum action act, struct cntr *cntr,
     221             :         enum protocol protocol)
     222             : {
     223           0 :         int ret=0;
     224           0 :         char *rpath=NULL;
     225             : #ifdef HAVE_WIN32
     226             :         logw(asfd, cntr, "Cannot restore special files to Windows: %s\n", fname);
     227             :         goto end;
     228             : #else
     229           0 :         struct stat statp=sb->statp;
     230             : 
     231           0 :         if(act==ACTION_VERIFY)
     232             :         {
     233           0 :                 cntr_add(cntr, CMD_SPECIAL, 1);
     234           0 :                 return 0;
     235             :         }
     236             : 
     237           0 :         if(build_path(fname, "", &rpath, NULL))
     238             :         {
     239             :                 // failed - do a warning
     240           0 :                 if(restore_interrupt(asfd, sb,
     241           0 :                         build_msg("build path failed: %s", fname),
     242             :                         cntr, protocol))
     243           0 :                                 ret=-1;
     244             :                 goto end;
     245             :         }
     246           0 :         if(S_ISFIFO(statp.st_mode))
     247             :         {
     248           0 :                 if(mkfifo(rpath, statp.st_mode) && errno!=EEXIST)
     249           0 :                         do_logw(asfd, cntr,
     250           0 :                                 "Cannot make fifo: %s\n", strerror(errno));
     251             :                 else
     252             :                 {
     253           0 :                         attribs_set(asfd, rpath, &statp, sb->winattr, cntr);
     254           0 :                         cntr_add(cntr, CMD_SPECIAL, 1);
     255             :                 }
     256             :         }
     257           0 :         else if(S_ISSOCK(statp.st_mode))
     258             :         {
     259           0 :                 if(mksock(rpath))
     260           0 :                         do_logw(asfd, cntr,
     261           0 :                                 "Cannot make socket: %s\n", strerror(errno));
     262             :                 else
     263             :                 {
     264           0 :                         attribs_set(asfd, rpath, &statp, sb->winattr, cntr);
     265           0 :                         cntr_add(cntr, CMD_SPECIAL, 1);
     266             :                 }
     267             :         }
     268             : #ifdef S_IFDOOR     // Solaris high speed RPC mechanism
     269             :         else if (S_ISDOOR(statp.st_mode))
     270             :                 do_logw(asfd, cntr,
     271             :                         "Skipping restore of door file: %s\n", fname);
     272             : #endif
     273             : #ifdef S_IFPORT     // Solaris event port for handling AIO
     274             :         else if (S_ISPORT(statp.st_mode))
     275             :                 do_logw(asfd, cntr,
     276             :                         "Skipping restore of event port file: %s\n", fname);
     277             : #endif
     278           0 :         else if(mknod(fname, statp.st_mode, statp.st_rdev) && errno!=EEXIST)
     279           0 :                 do_logw(asfd, cntr, "Cannot make node: %s\n", strerror(errno));
     280             :         else
     281             :         {
     282           0 :                 attribs_set(asfd, rpath, &statp, sb->winattr, cntr);
     283           0 :                 cntr_add(cntr, CMD_SPECIAL, 1);
     284             :         }
     285             : #endif
     286             : end:
     287           0 :         free_w(&rpath);
     288           0 :         return ret;
     289             : }
     290             : 
     291           0 : int restore_dir(struct asfd *asfd, struct sbuf *sb,
     292             :         const char *dname, enum action act, struct cntr *cntr,
     293             :         enum protocol protocol)
     294             : {
     295           0 :         int ret=0;
     296           0 :         char *rpath=NULL;
     297           0 :         if(act==ACTION_RESTORE)
     298             :         {
     299           0 :                 if(build_path(dname, "", &rpath, NULL))
     300             :                 {
     301           0 :                         ret=warn_and_interrupt(asfd, sb, cntr, protocol,
     302             :                                 "build path failed: %s", dname);
     303           0 :                         goto end;
     304             :                 }
     305           0 :                 else if(is_dir_lstat(rpath)<=0)
     306             :                 {
     307           0 :                         if(mkdir(rpath, 0777))
     308             :                         {
     309           0 :                                 ret=warn_and_interrupt(asfd, sb, cntr, protocol,
     310           0 :                                         "mkdir error: %s", strerror(errno));
     311           0 :                                 goto end;
     312             :                         }
     313             :                 }
     314           0 :                 attribs_set(asfd, rpath, &(sb->statp), sb->winattr, cntr);
     315           0 :                 if(!ret) cntr_add(cntr, sb->path.cmd, 1);
     316             :         }
     317           0 :         else cntr_add(cntr, sb->path.cmd, 1);
     318             : end:
     319           0 :         free_w(&rpath);
     320           0 :         return ret;
     321             : }
     322             : 
     323           6 : static int restore_link(struct asfd *asfd, struct sbuf *sb,
     324             :         const char *fname, enum action act, struct cntr *cntr,
     325             :         enum protocol protocol, const char *restore_prefix)
     326             : {
     327           6 :         int ret=0;
     328             : 
     329           6 :         if(act==ACTION_RESTORE)
     330             :         {
     331           6 :                 char *rpath=NULL;
     332           6 :                 if(build_path(fname, "", &rpath, NULL))
     333             :                 {
     334           0 :                         ret=warn_and_interrupt(asfd, sb, cntr, protocol,
     335             :                                 "build path failed: %s", fname);
     336           0 :                         goto end;
     337             :                 }
     338          12 :                 else if(make_link(
     339             : #ifdef HAVE_WIN32
     340             :                         asfd,
     341             :                         cntr,
     342             : #endif
     343           6 :                         fname, sb->link.buf,
     344             :                         sb->link.cmd, restore_prefix))
     345             :                 {
     346           0 :                         ret=warn_and_interrupt(asfd, sb, cntr, protocol,
     347             :                                 "could not create link", "");
     348           0 :                         goto end;
     349             :                 }
     350             :                 else if(!ret)
     351             :                 {
     352           6 :                         attribs_set(asfd, fname,
     353             :                                 &(sb->statp), sb->winattr, cntr);
     354           6 :                         cntr_add(cntr, sb->path.cmd, 1);
     355             :                 }
     356           6 :                 free_w(&rpath);
     357             :         }
     358           0 :         else cntr_add(cntr, sb->path.cmd, 1);
     359             : end:
     360           6 :         return ret;
     361             : }
     362             : 
     363             : static void strip_invalid_characters(char **path)
     364             : {
     365             : #ifdef HAVE_WIN32
     366             :       char *ch = *path;
     367             :       if (ch[0] != 0 && ch[1] != 0) {
     368             :          ch += 2;
     369             :          while (*ch) {
     370             :             switch (*ch) {
     371             :             case ':':
     372             :             case '<':
     373             :             case '>':
     374             :             case '*':
     375             :             case '?':
     376             :             case '|':
     377             :                *ch = '_';
     378             :                 break;
     379             :             }
     380             :             ch++;
     381             :          }
     382             :       }
     383             : #endif
     384             : }
     385             : 
     386             : static const char *act_str(enum action act)
     387             : {
     388             :         static const char *ret=NULL;
     389          20 :         if(act==ACTION_RESTORE) ret="restore";
     390           0 :         else ret="verify";
     391          20 :         return ret;
     392             : }
     393             : 
     394             : #ifndef UTEST
     395             : static
     396             : #endif
     397           7 : void strip_from_path(char *path, const char *strip)
     398             : {
     399             :         char *p;
     400             :         char *src;
     401             :         size_t len;
     402          14 :         if(!path
     403           7 :           || !strip
     404           6 :           || strlen(path)<=strlen(strip)
     405           5 :           || !(p=strstr(path, strip)))
     406             :                 return;
     407             : 
     408           5 :         len=strlen(p)-strlen(strip)+1;
     409           5 :         src=p+strlen(strip);
     410           5 :         memmove(p, src, len);
     411             : }
     412             : 
     413             : 
     414             : // Return 1 for ok, -1 for error, 0 for too many components stripped.
     415           0 : static int strip_path_components(struct asfd *asfd,
     416             :         struct sbuf *sb, int strip, struct cntr *cntr, enum protocol protocol)
     417             : {
     418           0 :         int s=0;
     419           0 :         char *tmp=NULL;
     420           0 :         char *cp=sb->path.buf;
     421           0 :         char *dp=NULL;
     422           0 :         for(s=0; cp && *cp && s<strip; s++)
     423             :         {
     424           0 :                 if(!(dp=strchr(cp, '/')))
     425             :                 {
     426           0 :                         char msg[256]="";
     427           0 :                         snprintf(msg, sizeof(msg),
     428             :                                 "Stripped too many components: %s",
     429             :                                 iobuf_to_printable(&sb->path));
     430           0 :                         if(restore_interrupt(asfd, sb, msg, cntr, protocol))
     431             :                                 return -1;
     432           0 :                         return 0;
     433             :                 }
     434           0 :                 cp=dp+1;
     435             :         }
     436           0 :         if(!cp)
     437             :         {
     438           0 :                 char msg[256]="";
     439           0 :                 snprintf(msg, sizeof(msg),
     440             :                         "Stripped too many components: %s",
     441             :                         iobuf_to_printable(&sb->path));
     442           0 :                 if(restore_interrupt(asfd, sb, msg, cntr, protocol))
     443             :                         return -1;
     444           0 :                 return 0;
     445             :         }
     446           0 :         if(!(tmp=strdup_w(cp, __func__)))
     447             :                 return -1;
     448           0 :         free_w(&sb->path.buf);
     449           0 :         sb->path.buf=tmp;
     450           0 :         return 1;
     451             : }
     452             : 
     453          32 : static int overwrite_ok(struct sbuf *sb,
     454             :         int overwrite,
     455             : #ifdef HAVE_WIN32
     456             :         struct BFILE *bfd,
     457             : #endif
     458             :         const char *fullpath)
     459             : {
     460             :         struct stat checkstat;
     461             : 
     462             :         // User specified overwrite is OK.
     463             : #ifdef HAVE_WIN32
     464             :         if(overwrite) return 1;
     465             : #else
     466             :         // User specified overwrite is OK,
     467             :         // UNLESS we are trying to overwrite the file with trailing VSS data.
     468          32 :         if(overwrite)
     469           0 :                 return (sb->path.cmd!=CMD_VSS_T
     470           0 :                         && sb->path.cmd!=CMD_ENC_VSS_T);
     471             : #endif
     472             : 
     473          32 :         if(!S_ISDIR(sb->statp.st_mode)
     474          32 :           && sb->path.cmd!=CMD_METADATA
     475          32 :           && sb->path.cmd!=CMD_ENC_METADATA
     476          32 :           && sb->path.cmd!=CMD_VSS
     477          32 :           && sb->path.cmd!=CMD_ENC_VSS)
     478             :         {
     479             : #ifdef HAVE_WIN32
     480             :                 // If Windows previously got some VSS data, it needs to append
     481             :                 // the file data to the already open bfd.
     482             :                 // And trailing VSS data.
     483             :                 if(bfd->mode!=BF_CLOSED
     484             :                   && (sb->path.cmd==CMD_FILE || sb->path.cmd==CMD_ENC_FILE
     485             :                       || sb->path.cmd==CMD_VSS_T || sb->path.cmd==CMD_ENC_VSS_T)
     486             :                   && bfd->path && !strcmp(bfd->path, fullpath))
     487             :                         return 1;
     488             : #endif
     489             :                 // If we have file data and the destination is
     490             :                 // a fifo, it is OK to write to the fifo.
     491          32 :                 if((sb->path.cmd==CMD_FILE || sb->path.cmd==CMD_ENC_FILE)
     492          26 :                   && S_ISFIFO(sb->statp.st_mode))
     493             :                         return 1;
     494             : 
     495             :                 // File path exists. Do not overwrite.
     496          32 :                 if(!lstat(fullpath, &checkstat)) return 0;
     497             :         }
     498             : 
     499             :         return 1;
     500             : }
     501             : 
     502             : #define RESTORE_STREAM  "restore_stream"
     503             : // Used to have "restore_spool". Removed for simplicity.
     504             : 
     505             : static char *restore_style=NULL;
     506             : 
     507           2 : static enum asl_ret restore_style_func(struct asfd *asfd,
     508             :         __attribute__ ((unused)) struct conf **confs,
     509             :         __attribute__ ((unused)) void *param)
     510             : {
     511           2 :         char msg[32]="";
     512           2 :         restore_style=NULL;
     513           2 :         if(strcmp(asfd->rbuf->buf, RESTORE_STREAM))
     514             :         {
     515           0 :                 iobuf_log_unexpected(asfd->rbuf, __func__);
     516           0 :                 return ASL_END_ERROR;
     517             :         }
     518           2 :         snprintf(msg, sizeof(msg), "%s_ok", asfd->rbuf->buf);
     519           2 :         if(asfd->write_str(asfd, CMD_GEN, msg))
     520             :                 return ASL_END_ERROR;
     521           2 :         restore_style=asfd->rbuf->buf;
     522           2 :         iobuf_init(asfd->rbuf);
     523           2 :         return ASL_END_OK;
     524             : }
     525             : 
     526           8 : static char *get_restore_style(struct asfd *asfd, struct conf **confs)
     527             : {
     528           8 :         if(get_protocol(confs)==PROTO_1)
     529           5 :                 return strdup_w(RESTORE_STREAM, __func__);
     530           3 :         if(asfd->simple_loop(asfd, confs, NULL, __func__,
     531             :                 restore_style_func)) return NULL;
     532           2 :         return restore_style;
     533             : }
     534             : 
     535           8 : int do_restore_client(struct asfd *asfd,
     536             :         struct conf **confs, enum action act, int vss_restore)
     537             : {
     538           8 :         int ret=-1;
     539           8 :         char msg[512]="";
     540           8 :         struct sbuf *sb=NULL;
     541           8 :         struct blk *blk=NULL;
     542           8 :         struct BFILE *bfd=NULL;
     543           8 :         char *fullpath=NULL;
     544           8 :         char *style=NULL;
     545           8 :         struct cntr *cntr=get_cntr(confs);
     546           8 :         enum protocol protocol=get_protocol(confs);
     547           8 :         int strip=get_int(confs[OPT_STRIP]);
     548           8 :         int overwrite=get_int(confs[OPT_OVERWRITE]);
     549           8 :         const char *strip_path=get_string(confs[OPT_STRIP_FROM_PATH]);
     550           8 :         const char *backup=get_string(confs[OPT_BACKUP]);
     551           8 :         const char *regex=get_string(confs[OPT_REGEX]);
     552           8 :         const char *restore_prefix=get_string(confs[OPT_RESTOREPREFIX]);
     553           8 :         const char *encryption_password=get_string(confs[OPT_ENCRYPTION_PASSWORD]);
     554             : 
     555           8 :         if(!(bfd=bfile_alloc())) goto end;
     556             : 
     557           8 :         bfile_init(bfd, 0, cntr);
     558           8 :         bfd->set_attribs_on_close=1;
     559             : 
     560          16 :         snprintf(msg, sizeof(msg), "%s %s:%s", act_str(act),
     561             :                 backup?backup:"", regex?regex:"");
     562           8 :         logp("doing %s\n", msg);
     563           8 :         if(asfd->write_str(asfd, CMD_GEN, msg)
     564           8 :           || asfd_read_expect(asfd, CMD_GEN, "ok"))
     565             :                 goto error;
     566           8 :         logp("doing %s confirmed\n", act_str(act));
     567             : 
     568             : #if defined(HAVE_WIN32)
     569             :         if(act==ACTION_RESTORE) win32_enable_backup_privileges();
     570             : #endif
     571             : 
     572           8 :         logfmt("\n");
     573             : 
     574           8 :         if(cntr_recv(asfd, confs))
     575             :                 goto error;
     576             : 
     577           8 :         if(!(style=get_restore_style(asfd, confs)))
     578             :                 goto error;
     579             : 
     580           7 :         if(!(sb=sbuf_alloc(protocol))
     581           7 :           || (protocol==PROTO_2 && !(blk=blk_alloc())))
     582             :         {
     583           0 :                 log_and_send_oom(asfd);
     584           0 :                 goto error;
     585             :         }
     586             : 
     587             :         while(1)
     588             :         {
     589          52 :                 sbuf_free_content(sb);
     590          52 :                 if(protocol==PROTO_1)
     591          13 :                         sb->flags |= SBUF_CLIENT_RESTORE_HACK;
     592             : 
     593          52 :                 switch(sbuf_fill_from_net(sb, asfd, blk, cntr))
     594             :                 {
     595             :                         case 0: break;
     596           4 :                         case 1: if(asfd->write_str(asfd, CMD_GEN,
     597             :                                 "restoreend ok")) goto error;
     598             :                                 goto end; // It was OK.
     599             :                         default:
     600             :                         case -1: goto error;
     601             :                 }
     602             : 
     603          46 :                 if(protocol==PROTO_2)
     604             :                 {
     605          37 :                         if(blk->data)
     606             :                         {
     607          14 :                                 int wret=0;
     608          14 :                                 if(act==ACTION_VERIFY)
     609           0 :                                         cntr_add(cntr, CMD_DATA, 1);
     610             :                                 else
     611          14 :                                         wret=write_protocol2_data(asfd,
     612             :                                                 bfd, blk, vss_restore);
     613          14 :                                 blk_free_content(blk);
     614          14 :                                 blk->data=NULL;
     615          14 :                                 if(wret) goto error;
     616          14 :                                 continue;
     617             :                         }
     618          23 :                         else if(sb->endfile.buf)
     619             :                         {
     620           0 :                                 continue;
     621             :                         }
     622             :                 }
     623             : 
     624          32 :                 switch(sb->path.cmd)
     625             :                 {
     626             :                         case CMD_DIRECTORY:
     627             :                         case CMD_FILE:
     628             :                         case CMD_ENC_FILE:
     629             :                         case CMD_SOFT_LINK:
     630             :                         case CMD_HARD_LINK:
     631             :                         case CMD_SPECIAL:
     632             :                         case CMD_METADATA:
     633             :                         case CMD_ENC_METADATA:
     634             :                         case CMD_VSS:
     635             :                         case CMD_ENC_VSS:
     636             :                         case CMD_VSS_T:
     637             :                         case CMD_ENC_VSS_T:
     638             :                         case CMD_EFS_FILE:
     639          32 :                                 if(strip)
     640             :                                 {
     641             :                                         int s;
     642           0 :                                         s=strip_path_components(asfd,
     643             :                                                 sb, strip, cntr, protocol);
     644           0 :                                         if(s<0) goto error;
     645           0 :                                         if(s==0)
     646             :                                         {
     647             :                                                 // Too many components stripped
     648             :                                                 // - carry on.
     649           0 :                                                 continue;
     650             :                                         }
     651             :                                         // It is OK, sb.path is now stripped.
     652             :                                 }
     653          32 :                                 if(strip_path)
     654             :                                 {
     655           0 :                                         strip_from_path(sb->path.buf,
     656             :                                                 strip_path);
     657             :                                         // Strip links if their path is absolute
     658           0 :                                         if(sb->link.buf && !is_absolute(sb->link.buf))
     659           0 :                                                 strip_from_path(sb->link.buf,
     660             :                                                         strip_path);
     661             :                                 }
     662          32 :                                 free_w(&fullpath);
     663          32 :                                 if(!(fullpath=prepend_s(restore_prefix,
     664          32 :                                         sb->path.buf)))
     665             :                                 {
     666           0 :                                         log_and_send_oom(asfd);
     667           0 :                                         goto error;
     668             :                                 }
     669          32 :                                 if(act==ACTION_RESTORE)
     670             :                                 {
     671          32 :                                   strip_invalid_characters(&fullpath);
     672          32 :                                   if(!overwrite_ok(sb, overwrite,
     673             : #ifdef HAVE_WIN32
     674             :                                         bfd,
     675             : #endif
     676             :                                         fullpath))
     677             :                                   {
     678           7 :                                         char msg[512]="";
     679             :                                         // Something exists at that path.
     680           7 :                                         snprintf(msg, sizeof(msg),
     681             :                                                 "Path exists: %s\n", fullpath);
     682           7 :                                         if(restore_interrupt(asfd,
     683             :                                                 sb, msg, cntr, protocol))
     684             :                                                         goto error;
     685           7 :                                         continue;
     686             :                                   }
     687             :                                 }
     688             :                                 break;
     689             :                         case CMD_MESSAGE:
     690             :                         case CMD_WARNING:
     691           0 :                                 log_recvd(&sb->path, cntr, 1);
     692           0 :                                 logfmt("\n");
     693           0 :                                 continue;
     694             :                         default:
     695             :                                 break;
     696             :                 }
     697             : 
     698          25 :                 switch(sb->path.cmd)
     699             :                 {
     700             :                         // These are the same in both protocol1 and protocol2.
     701             :                         case CMD_DIRECTORY:
     702           0 :                                 if(restore_dir(asfd, sb, fullpath, act, cntr,
     703             :                                         protocol))
     704             :                                                 goto error;
     705           0 :                                 continue;
     706             :                         case CMD_SOFT_LINK:
     707             :                         case CMD_HARD_LINK:
     708           6 :                                 if(restore_link(asfd, sb, fullpath, act, cntr,
     709             :                                         protocol, restore_prefix))
     710             :                                                 goto error;
     711           6 :                                 continue;
     712             :                         case CMD_SPECIAL:
     713           0 :                                 if(restore_special(asfd, sb,
     714             :                                         fullpath, act, cntr, protocol))
     715             :                                                 goto error;
     716           0 :                                 continue;
     717             :                         default:
     718             :                                 break;
     719             :                 }
     720             : 
     721          19 :                 if(protocol==PROTO_2)
     722             :                 {
     723          14 :                         if(restore_switch_protocol2(asfd, sb, fullpath, act,
     724             :                                 bfd, vss_restore, cntr))
     725             :                                         goto error;
     726             :                 }
     727             :                 else
     728             :                 {
     729           5 :                         if(restore_switch_protocol1(asfd, sb, fullpath, act,
     730             :                                 bfd, vss_restore, cntr, encryption_password))
     731             :                                         goto error;
     732             :                 }
     733             :         }
     734             : 
     735             : end:
     736             :         ret=0;
     737             : error:
     738             :         // It is possible for a fd to still be open.
     739           8 :         bfd->close(bfd, asfd);
     740           8 :         bfile_free(&bfd);
     741             : 
     742           8 :         cntr_print_end(cntr);
     743           8 :         cntr_print(cntr, act, asfd);
     744             : 
     745          12 :         if(!ret) logp("%s finished\n", act_str(act));
     746           4 :         else logp("ret: %d\n", ret);
     747             : 
     748           8 :         sbuf_free(&sb);
     749           8 :         free_w(&style);
     750           8 :         free_w(&fullpath);
     751           8 :         blk_free(&blk);
     752             : 
     753           8 :         return ret;
     754             : }

Generated by: LCOV version 1.13