X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/d144e43bb3ea77bdc9d1dd68a648166970822dd6..82ad07c4182f744c07b96a15df4572e559ed7dc8:/batch.c

diff --git a/batch.c b/batch.c
index 121e34ff..e0eaa39b 100644
--- a/batch.c
+++ b/batch.c
@@ -36,16 +36,19 @@ extern int always_checksum;
 extern int do_compression;
 extern int def_compress_level;
 extern int protocol_version;
+extern int flist_extra_ndx;
 extern char *batch_name;
 
 extern struct filter_list_struct filter_list;
 
+static int tweaked_preserve_uid;
+static int tweaked_preserve_gid;
 static int tweaked_compress_level;
 
 static int *flag_ptr[] = {
 	&recurse,		/* 0 */
-	&preserve_uid,		/* 1 */
-	&preserve_gid,		/* 2 */
+	&tweaked_preserve_uid,	/* 1 */
+	&tweaked_preserve_gid,	/* 2 */
 	&preserve_links,	/* 3 */
 	&preserve_devices,	/* 4 */
 	&preserve_hard_links,	/* 5 */
@@ -72,6 +75,8 @@ void write_stream_flags(int fd)
 {
 	int i, flags;
 
+	tweaked_preserve_uid = preserve_uid != 0;
+	tweaked_preserve_gid = preserve_gid != 0;
 #if Z_DEFAULT_COMPRESSION == -1
 	tweaked_compress_level = do_compression ? def_compress_level + 2 : 0;
 #else
@@ -113,6 +118,16 @@ void read_stream_flags(int fd)
 			xfer_dirs = 0;
 	}
 
+	if (tweaked_preserve_uid) {
+		if (!preserve_uid)
+			preserve_uid = flist_extra_ndx++;
+	} else
+		preserve_uid = 0;
+	if (tweaked_preserve_gid) {
+		if (!preserve_gid)
+			preserve_gid = flist_extra_ndx++;
+	} else
+		preserve_gid = 0;
 	if (tweaked_compress_level == 0 || tweaked_compress_level == 2)
 		do_compression = 0;
 	else {