Split up the ifuncs.h file into 3 .h files.
[rsync/rsync.git] / flist.c
diff --git a/flist.c b/flist.c
index 2cf72b1..766f1b3 100644 (file)
--- a/flist.c
+++ b/flist.c
@@ -23,6 +23,7 @@
 #include "rsync.h"
 #include "ifuncs.h"
 #include "rounding.h"
+#include "inums.h"
 #include "io.h"
 
 extern int am_root;
@@ -65,7 +66,9 @@ extern int protocol_version;
 extern int sanitize_paths;
 extern int munge_symlinks;
 extern int need_unsorted_flist;
+extern int sender_symlink_iconv;
 extern int output_needs_newline;
+extern int sender_keeps_checksum;
 extern int unsort_ndx;
 extern struct stats stats;
 extern char *filesfrom_host;
@@ -319,7 +322,7 @@ static void flist_expand(struct file_list *flist, int extra)
        if (DEBUG_GTE(FLIST, 1) && flist->malloced != FLIST_START) {
                rprintf(FCLIENT, "[%s] expand file_list pointer array to %s bytes, did%s move\n",
                    who_am_i(),
-                   big_num(sizeof flist->files[0] * flist->malloced, 0),
+                   big_num(sizeof flist->files[0] * flist->malloced),
                    (new_ptr == flist->files) ? " not" : "");
        }
 
@@ -375,7 +378,7 @@ int change_pathname(struct file_struct *file, const char *dir, int dirlen)
        if (!change_dir(dir, CD_NORMAL)) {
          chdir_error:
                io_error |= IOERR_GENERAL;
-               rsyserr(FERROR, errno, "change_dir %s failed", full_fname(dir));
+               rsyserr(FERROR_XFER, errno, "change_dir %s failed", full_fname(dir));
                if (dir != orig_dir)
                        change_dir(orig_dir, CD_NORMAL);
                pathname = NULL;
@@ -386,7 +389,11 @@ int change_pathname(struct file_struct *file, const char *dir, int dirlen)
        return 1;
 }
 
-static void send_file_entry(int f, const char *fname, struct file_struct *file, int ndx, int first_ndx)
+static void send_file_entry(int f, const char *fname, struct file_struct *file,
+#ifdef SUPPORT_LINKS
+                           const char *symlink_name, int symlink_len,
+#endif
+                           int ndx, int first_ndx)
 {
        static time_t modtime;
        static mode_t mode;
@@ -399,7 +406,9 @@ static void send_file_entry(int f, const char *fname, struct file_struct *file,
        static gid_t gid;
        static const char *user_name, *group_name;
        static char lastname[MAXPATHLEN];
+#ifdef SUPPORT_HARD_LINKS
        int first_hlink_ndx = -1;
+#endif
        int l1, l2;
        int xflags;
 
@@ -472,6 +481,17 @@ static void send_file_entry(int f, const char *fname, struct file_struct *file,
                                np->data = (void*)(long)(first_ndx + ndx + 1);
                                xflags |= XMIT_HLINK_FIRST;
                        }
+                       if (DEBUG_GTE(HLINK, 1)) {
+                               if (first_hlink_ndx >= 0) {
+                                       rprintf(FINFO, "[%s] #%d hard-links #%d (%sabbrev)\n",
+                                               who_am_i(), first_ndx + ndx, first_hlink_ndx,
+                                               first_hlink_ndx >= first_ndx ? "" : "un");
+                               } else if (DEBUG_GTE(HLINK, 3)) {
+                                       rprintf(FINFO, "[%s] dev:inode for #%d is %s:%s\n",
+                                               who_am_i(), first_ndx + ndx,
+                                               big_num(tmp_dev), big_num(tmp_ino));
+                               }
+                       }
                } else {
                        if (tmp_dev == dev) {
                                if (protocol_version >= 28)
@@ -518,11 +538,13 @@ static void send_file_entry(int f, const char *fname, struct file_struct *file,
                write_byte(f, l2);
        write_buf(f, fname + l1, l2);
 
+#ifdef SUPPORT_HARD_LINKS
        if (first_hlink_ndx >= 0) {
                write_varint(f, first_hlink_ndx);
                if (first_hlink_ndx >= first_ndx)
                        goto the_end;
        }
+#endif
 
        write_varlong30(f, F_LENGTH(file), 3);
        if (!(xflags & XMIT_SAME_TIME)) {
@@ -575,11 +597,9 @@ static void send_file_entry(int f, const char *fname, struct file_struct *file,
        }
 
 #ifdef SUPPORT_LINKS
-       if (preserve_links && S_ISLNK(mode)) {
-               const char *sl = F_SYMLINK(file);
-               int len = strlen(sl);
-               write_varint30(f, len);
-               write_buf(f, sl, len);
+       if (symlink_len) {
+               write_varint30(f, symlink_len);
+               write_buf(f, symlink_name, symlink_len);
        }
 #endif
 
@@ -609,7 +629,9 @@ static void send_file_entry(int f, const char *fname, struct file_struct *file,
                write_buf(f, sum, checksum_len);
        }
 
+#ifdef SUPPORT_HARD_LINKS
   the_end:
+#endif
        strlcpy(lastname, fname, MAXPATHLEN);
 
        if (S_ISREG(mode) || S_ISLNK(mode))
@@ -671,16 +693,16 @@ static struct file_struct *recv_file_entry(struct file_list *flist,
                xbuf outbuf, inbuf;
 
                INIT_CONST_XBUF(outbuf, thisname);
-               INIT_XBUF(inbuf, lastname, basename_len, -1);
+               INIT_XBUF(inbuf, lastname, basename_len, (size_t)-1);
 
                if (iconvbufs(ic_recv, &inbuf, &outbuf, 0) < 0) {
                        io_error |= IOERR_GENERAL;
-                       rprintf(FINFO,
+                       rprintf(FERROR_UTF8,
                            "[%s] cannot convert filename: %s (%s)\n",
                            who_am_i(), lastname, strerror(errno));
                        outbuf.len = 0;
                }
-               outbuf.buf[outbuf.len] = '\0';
+               thisname[outbuf.len] = '\0';
        }
 #endif
 
@@ -713,6 +735,11 @@ static struct file_struct *recv_file_entry(struct file_list *flist,
                                first_hlink_ndx, flist->ndx_start + flist->used);
                        exit_cleanup(RERR_PROTOCOL);
                }
+               if (DEBUG_GTE(HLINK, 1)) {
+                       rprintf(FINFO, "[%s] #%d hard-links #%d (%sabbrev)\n",
+                               who_am_i(), flist->used+flist->ndx_start, first_hlink_ndx,
+                               first_hlink_ndx >= flist->ndx_start ? "" : "un");
+               }
                if (first_hlink_ndx >= flist->ndx_start) {
                        struct file_struct *first = flist->files[first_hlink_ndx - flist->ndx_start];
                        file_length = F_LENGTH(first);
@@ -811,6 +838,13 @@ static struct file_struct *recv_file_entry(struct file_list *flist,
                                linkname_len - 1);
                        overflow_exit("recv_file_entry");
                }
+#ifdef ICONV_OPTION
+               /* We don't know how much extra room we need to convert
+                * the as-yet-unread symlink data, so let's hope that a
+                * double-size buffer is plenty. */
+               if (sender_symlink_iconv)
+                       linkname_len *= 2;
+#endif
                if (munge_symlinks)
                        linkname_len += SYMLINK_PREFIX_LEN;
        }
@@ -945,14 +979,40 @@ static struct file_struct *recv_file_entry(struct file_list *flist,
                if (first_hlink_ndx >= flist->ndx_start) {
                        struct file_struct *first = flist->files[first_hlink_ndx - flist->ndx_start];
                        memcpy(bp, F_SYMLINK(first), linkname_len);
-               } else if (munge_symlinks) {
-                       strlcpy(bp, SYMLINK_PREFIX, linkname_len);
-                       bp += SYMLINK_PREFIX_LEN;
-                       linkname_len -= SYMLINK_PREFIX_LEN;
-                       read_sbuf(f, bp, linkname_len - 1);
                } else {
-                       read_sbuf(f, bp, linkname_len - 1);
-                       if (sanitize_paths)
+                       if (munge_symlinks) {
+                               strlcpy(bp, SYMLINK_PREFIX, linkname_len);
+                               bp += SYMLINK_PREFIX_LEN;
+                               linkname_len -= SYMLINK_PREFIX_LEN;
+                       }
+#ifdef ICONV_OPTION
+                       if (sender_symlink_iconv) {
+                               xbuf outbuf, inbuf;
+
+                               alloc_len = linkname_len;
+                               linkname_len /= 2;
+
+                               /* Read the symlink data into the end of our double-sized
+                                * buffer and then convert it into the right spot. */
+                               INIT_XBUF(inbuf, bp + alloc_len - linkname_len,
+                                         linkname_len - 1, (size_t)-1);
+                               read_sbuf(f, inbuf.buf, inbuf.len);
+                               INIT_XBUF(outbuf, bp, 0, alloc_len);
+
+                               if (iconvbufs(ic_recv, &inbuf, &outbuf, 0) < 0) {
+                                       io_error |= IOERR_GENERAL;
+                                       rprintf(FERROR_XFER,
+                                           "[%s] cannot convert symlink data for: %s (%s)\n",
+                                           who_am_i(), full_fname(thisname), strerror(errno));
+                                       bp = (char*)file->basename;
+                                       *bp++ = '\0';
+                                       outbuf.len = 0;
+                               }
+                               bp[outbuf.len] = '\0';
+                       } else
+#endif
+                               read_sbuf(f, bp, linkname_len - 1);
+                       if (sanitize_paths && !munge_symlinks && *bp)
                                sanitize_path(bp, bp, "", lastdir_depth, SP_DEFAULT);
                }
        }
@@ -1045,7 +1105,7 @@ struct file_struct *make_file(const char *fname, struct file_list *flist,
 
        if (strlcpy(thisname, fname, sizeof thisname) >= sizeof thisname) {
                io_error |= IOERR_GENERAL;
-               rprintf(FINFO, "skipping overly long name: %s\n", fname);
+               rprintf(FERROR_XFER, "skipping overly long name: %s\n", fname);
                return NULL;
        }
        clean_fname(thisname, 0);
@@ -1090,7 +1150,7 @@ struct file_struct *make_file(const char *fname, struct file_list *flist,
                        }
                } else {
                        io_error |= IOERR_GENERAL;
-                       rsyserr(FERROR_XFER, save_errno, "readlink %s failed",
+                       rsyserr(FERROR_XFER, save_errno, "readlink_stat(%s) failed",
                                full_fname(thisname));
                }
                return NULL;
@@ -1187,6 +1247,12 @@ struct file_struct *make_file(const char *fname, struct file_list *flist,
                extra_len += EXTRA_LEN;
 #endif
 
+       if (always_checksum && am_sender && S_ISREG(st.st_mode)) {
+               file_checksum(thisname, tmp_sum, st.st_size);
+               if (sender_keeps_checksum)
+                       extra_len += SUM_EXTRA_CNT * EXTRA_LEN;
+       }
+
 #if EXTRA_ROUNDING > 0
        if (extra_len & (EXTRA_ROUNDING * EXTRA_LEN))
                extra_len = (extra_len | (EXTRA_ROUNDING * EXTRA_LEN)) + EXTRA_LEN;
@@ -1213,8 +1279,8 @@ struct file_struct *make_file(const char *fname, struct file_list *flist,
                if (protocol_version >= 28
                 ? (!S_ISDIR(st.st_mode) && st.st_nlink > 1)
                 : S_ISREG(st.st_mode)) {
-                       tmp_dev = st.st_dev;
-                       tmp_ino = st.st_ino;
+                       tmp_dev = (int64)st.st_dev + 1;
+                       tmp_ino = (int64)st.st_ino;
                } else
                        tmp_dev = 0;
        }
@@ -1250,9 +1316,6 @@ struct file_struct *make_file(const char *fname, struct file_list *flist,
                memcpy(bp + basename_len, linkname, linkname_len);
 #endif
 
-       if (always_checksum && am_sender && S_ISREG(st.st_mode))
-               file_checksum(thisname, tmp_sum, st.st_size);
-
        if (am_sender)
                F_PATHNAME(file) = pathname;
        else if (!pool)
@@ -1283,6 +1346,9 @@ struct file_struct *make_file(const char *fname, struct file_list *flist,
                return NULL;
        }
 
+       if (sender_keeps_checksum && S_ISREG(st.st_mode))
+               memcpy(F_SUM(file), tmp_sum, checksum_len);
+
        if (unsort_ndx)
                F_NDX(file) = dir_count;
 
@@ -1310,10 +1376,27 @@ static struct file_struct *send_file_name(int f, struct file_list *flist,
 
        if (f >= 0) {
                char fbuf[MAXPATHLEN];
+#ifdef SUPPORT_LINKS
+               const char *symlink_name;
+               int symlink_len;
+#ifdef ICONV_OPTION
+               char symlink_buf[MAXPATHLEN];
+#endif
+#endif
 #if defined SUPPORT_ACLS || defined SUPPORT_XATTRS
                stat_x sx;
 #endif
 
+#ifdef SUPPORT_LINKS
+               if (preserve_links && S_ISLNK(file->mode)) {
+                       symlink_name = F_SYMLINK(file);
+                       symlink_len = strlen(symlink_name);
+               } else {
+                       symlink_name = NULL;
+                       symlink_len = 0;
+               }
+#endif
+
 #ifdef ICONV_OPTION
                if (ic_send != (iconv_t)-1) {
                        xbuf outbuf, inbuf;
@@ -1326,19 +1409,38 @@ static struct file_struct *send_file_name(int f, struct file_list *flist,
                                if (iconvbufs(ic_send, &inbuf, &outbuf, 0) < 0)
                                        goto convert_error;
                                outbuf.size += 2;
-                               outbuf.buf[outbuf.len++] = '/';
+                               fbuf[outbuf.len++] = '/';
                        }
 
                        INIT_XBUF_STRLEN(inbuf, (char*)file->basename);
                        if (iconvbufs(ic_send, &inbuf, &outbuf, 0) < 0) {
                          convert_error:
                                io_error |= IOERR_GENERAL;
-                               rprintf(FINFO,
+                               rprintf(FERROR_XFER,
                                    "[%s] cannot convert filename: %s (%s)\n",
                                    who_am_i(), f_name(file, fbuf), strerror(errno));
                                return NULL;
                        }
-                       outbuf.buf[outbuf.len] = '\0';
+                       fbuf[outbuf.len] = '\0';
+
+#ifdef SUPPORT_LINKS
+                       if (symlink_len && sender_symlink_iconv) {
+                               INIT_XBUF(inbuf, (char*)symlink_name, symlink_len, (size_t)-1);
+                               INIT_CONST_XBUF(outbuf, symlink_buf);
+                               if (iconvbufs(ic_send, &inbuf, &outbuf, 0) < 0) {
+                                       io_error |= IOERR_GENERAL;
+                                       f_name(file, fbuf);
+                                       rprintf(FERROR_XFER,
+                                           "[%s] cannot convert symlink data for: %s (%s)\n",
+                                           who_am_i(), full_fname(fbuf), strerror(errno));
+                                       return NULL;
+                               }
+                               symlink_buf[outbuf.len] = '\0';
+
+                               symlink_name = symlink_buf;
+                               symlink_len = outbuf.len;
+                       }
+#endif
                } else
 #endif
                        f_name(file, fbuf);
@@ -1363,7 +1465,11 @@ static struct file_struct *send_file_name(int f, struct file_list *flist,
                }
 #endif
 
-               send_file_entry(f, fbuf, file, flist->used, flist->ndx_start);
+               send_file_entry(f, fbuf, file,
+#ifdef SUPPORT_LINKS
+                               symlink_name, symlink_len,
+#endif
+                               flist->used, flist->ndx_start);
 
 #ifdef SUPPORT_ACLS
                if (preserve_acls && !S_ISLNK(file->mode)) {
@@ -1558,14 +1664,14 @@ static void send_directory(int f, struct file_list *flist, char *fbuf, int len,
                        continue;
                if (strlcpy(p, dname, remainder) >= remainder) {
                        io_error |= IOERR_GENERAL;
-                       rprintf(FINFO,
+                       rprintf(FERROR_XFER,
                                "cannot send long-named file %s\n",
                                full_fname(fbuf));
                        continue;
                }
                if (dname[0] == '\0') {
                        io_error |= IOERR_GENERAL;
-                       rprintf(FINFO,
+                       rprintf(FERROR_XFER,
                                "cannot send file with empty name in %s\n",
                                full_fname(fbuf));
                        continue;
@@ -1811,7 +1917,12 @@ void send_extra_file_list(int f, int at_least)
                        dp = F_DIR_NODE_P(file);
                }
 
-               write_byte(f, 0);
+               if (protocol_version < 31 || io_error == save_io_error || ignore_errors)
+                       write_byte(f, 0);
+               else {
+                       write_shortint(f, XMIT_EXTENDED_FLAGS|XMIT_IO_ERROR_ENDLIST);
+                       write_int(f, io_error);
+               }
 
                if (need_unsorted_flist) {
                        if (!(flist->sorted = new_array(struct file_struct *, flist->used)))
@@ -1852,7 +1963,7 @@ void send_extra_file_list(int f, int at_least)
        }
 
   finish:
-       if (io_error != save_io_error && !ignore_errors)
+       if (io_error != save_io_error && protocol_version == 30 && !ignore_errors)
                send_msg_int(MSG_IO_ERROR, io_error);
 }
 
@@ -2108,7 +2219,13 @@ struct file_list *send_file_list(int f, int argc, char *argv[])
                stats.flist_buildtime = 1;
        start_tv = end_tv;
 
-       write_byte(f, 0); /* Indicate end of file list */
+       /* Indicate end of file list */
+       if (protocol_version < 31 || io_error == 0 || ignore_errors)
+               write_byte(f, 0);
+       else {
+               write_shortint(f, XMIT_EXTENDED_FLAGS|XMIT_IO_ERROR_ENDLIST);
+               write_int(f, io_error);
+       }
 
 #ifdef SUPPORT_HARD_LINKS
        if (preserve_hard_links && protocol_version >= 30 && !inc_recurse)
@@ -2146,7 +2263,7 @@ struct file_list *send_file_list(int f, int argc, char *argv[])
        /* send the io_error flag */
        if (protocol_version < 30)
                write_int(f, ignore_errors ? 0 : io_error);
-       else if (io_error && !ignore_errors)
+       else if (io_error && protocol_version == 30 && !ignore_errors)
                send_msg_int(MSG_IO_ERROR, io_error);
 
        if (disable_buffering)
@@ -2164,7 +2281,7 @@ struct file_list *send_file_list(int f, int argc, char *argv[])
        if (inc_recurse) {
                send_dir_depth = 1;
                add_dirs_to_tree(-1, flist, dir_count);
-               if (!file_total || strcmp(flist->sorted[0]->basename, ".") != 0)
+               if (!file_total || strcmp(flist->sorted[flist->low]->basename, ".") != 0)
                        flist->parent_ndx = -1;
                flist_done_allocating(flist);
                if (send_dir_ndx < 0) {
@@ -2188,12 +2305,13 @@ struct file_list *recv_file_list(int f)
        int dstart, flags;
        int64 start_read;
 
-       if (!first_flist)
+       if (!first_flist) {
+               if (show_filelist_p())
+                       start_filelist_progress("receiving file list");
+               else if (inc_recurse && INFO_GTE(FLIST, 1) && !am_server)
+                       rprintf(FCLIENT, "receiving incremental file list\n");
                rprintf(FLOG, "receiving file list\n");
-       if (show_filelist_p())
-               start_filelist_progress("receiving file list");
-       else if (inc_recurse && INFO_GTE(FLIST, 1) && !am_server && !first_flist)
-               rprintf(FCLIENT, "receiving incremental file list\n");
+       }
 
        start_read = stats.total_read;
 
@@ -2216,10 +2334,22 @@ struct file_list *recv_file_list(int f)
        while ((flags = read_byte(f)) != 0) {
                struct file_struct *file;
 
-               flist_expand(flist, 1);
-
                if (protocol_version >= 28 && (flags & XMIT_EXTENDED_FLAGS))
                        flags |= read_byte(f) << 8;
+
+               if (flags == (XMIT_EXTENDED_FLAGS|XMIT_IO_ERROR_ENDLIST)) {
+                       int err;
+                       if (protocol_version < 31) {
+                               rprintf(FERROR, "Invalid flist flag: %x\n", flags);
+                               exit_cleanup(RERR_PROTOCOL);
+                       }
+                       err = read_int(f);
+                       if (!ignore_errors)
+                               io_error |= err;
+                       break;
+               }
+
+               flist_expand(flist, 1);
                file = recv_file_entry(flist, flags, f);
 
                if (inc_recurse && S_ISDIR(file->mode)) {
@@ -2232,8 +2362,8 @@ struct file_list *recv_file_list(int f)
                maybe_emit_filelist_progress(flist->used);
 
                if (DEBUG_GTE(FLIST, 2)) {
-                       rprintf(FINFO, "recv_file_name(%s)\n",
-                               f_name(file, NULL));
+                       char *name = f_name(file, NULL);
+                       rprintf(FINFO, "recv_file_name(%s)\n", NS(name));
                }
        }
        file_total += flist->used;
@@ -2283,12 +2413,11 @@ struct file_list *recv_file_list(int f)
 
        if (protocol_version < 30) {
                /* Recv the io_error flag */
-               if (ignore_errors)
-                       read_int(f);
-               else
-                       io_error |= read_int(f);
+               int err = read_int(f);
+               if (!ignore_errors)
+                       io_error |= err;
        } else if (inc_recurse && flist->ndx_start == 1) {
-               if (!file_total || strcmp(flist->sorted[0]->basename, ".") != 0)
+               if (!file_total || strcmp(flist->sorted[flist->low]->basename, ".") != 0)
                        flist->parent_ndx = -1;
        }
 
@@ -2408,14 +2537,16 @@ struct file_list *flist_new(int flags, char *msg)
 
        if (flags & FLIST_TEMP) {
                if (!(flist->file_pool = pool_create(SMALL_EXTENT, 0,
-                                               out_of_memory, POOL_INTERN)))
+                                                    out_of_memory,
+                                                    POOL_INTERN)))
                        out_of_memory(msg);
        } else {
                /* This is a doubly linked list with prev looping back to
                 * the end of the list, but the last next pointer is NULL. */
                if (!first_flist) {
                        flist->file_pool = pool_create(NORMAL_EXTENT, 0,
-                                               out_of_memory, POOL_INTERN);
+                                                      out_of_memory,
+                                                      POOL_INTERN);
                        if (!flist->file_pool)
                                out_of_memory(msg);
 
@@ -2689,7 +2820,7 @@ static void output_flist(struct file_list *flist)
                        "[%s] i=%d %s %s%s%s%s mode=0%o len=%s%s%s flags=%x\n",
                        who, i + flist->ndx_start,
                        root, dir, slash, name, trail,
-                       (int)file->mode, big_num(F_LENGTH(file), 0),
+                       (int)file->mode, comma_num(F_LENGTH(file)),
                        uidbuf, gidbuf, file->flags);
        }
 }