Tweaked the output a little.

[rsync/rsync.git] / batch.c

diff --git a/batch.c b/batch.c
index f8529db..b1859fc 100644 (file)
--- a/batch.c
+++ b/batch.c
@@ -1,9 +1,24 @@
-/* -*- c-file-style: "linux" -*-
-
-   Weiss 1/1999
-   Batch utilities for rsync.
-
-*/
+/*
+ * Support for the batch-file options.
+ *
+ * Copyright (C) 1999 Weiss
+ * Copyright (C) 2004 Chris Shoemaker
+ * Copyright (C) 2004, 2005, 2006 Wayne Davison
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA.
+ */
 
 #include "rsync.h"
 #include "zlib/zlib.h"
@@ -21,16 +36,19 @@ extern int always_checksum;
 extern int do_compression;
 extern int def_compress_level;
 extern int protocol_version;
+extern int flist_extra_cnt;
 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 */
@@ -57,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
@@ -98,6 +118,16 @@ void read_stream_flags(int fd)
                        xfer_dirs = 0;
        }
 
+       if (tweaked_preserve_uid) {
+               if (!preserve_uid)
+                       preserve_uid = ++flist_extra_cnt;
+       } else
+               preserve_uid = 0;
+       if (tweaked_preserve_gid) {
+               if (!preserve_gid)
+                       preserve_gid = ++flist_extra_cnt;
+       } else
+               preserve_gid = 0;
        if (tweaked_compress_level == 0 || tweaked_compress_level == 2)
                do_compression = 0;
        else {