Fix --remove-source-files sanity check w/--copy-links the right way.
[rsync.git] / cleanup.c
index 971154ceca4058572ab899aa5178e418ff127323..c3ae799925b2531373bf03cc3ca76c853c40a563 100644 (file)
--- a/cleanup.c
+++ b/cleanup.c
@@ -4,7 +4,7 @@
  * Copyright (C) 1996-2000 Andrew Tridgell
  * Copyright (C) 1996 Paul Mackerras
  * Copyright (C) 2002 Martin Pool
- * Copyright (C) 2003-2013 Wayne Davison
+ * Copyright (C) 2003-2018 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
@@ -22,6 +22,7 @@
 
 #include "rsync.h"
 
+extern int dry_run;
 extern int am_server;
 extern int am_daemon;
 extern int am_receiver;
@@ -87,7 +88,7 @@ int cleanup_got_literal = 0;
 static const char *cleanup_fname;
 static const char *cleanup_new_fname;
 static struct file_struct *cleanup_file;
-static int cleanup_fd_r, cleanup_fd_w;
+static int cleanup_fd_r = -1, cleanup_fd_w = -1;
 static pid_t cleanup_pid = 0;
 
 pid_t cleanup_child_pid = -1;
@@ -102,15 +103,15 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
        static int switch_step = 0;
        static int exit_code = 0, exit_line = 0;
        static const char *exit_file = NULL;
-       static int unmodified_code = 0;
+       static int first_code = 0;
 
        SIGACTION(SIGUSR1, SIG_IGN);
        SIGACTION(SIGUSR2, SIG_IGN);
 
-       if (exit_code) { /* Preserve first exit info when recursing. */
-               code = exit_code;
-               file = exit_file;
-               line = exit_line;
+       if (!exit_code) { /* Preserve first error exit info when recursing. */
+               exit_code = code;
+               exit_file = file;
+               exit_line = line < 0 ? -line : line;
        }
 
        /* If this is the exit at the end of the run, the server side
@@ -124,9 +125,7 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
 #include "case_N.h" /* case 0: */
                switch_step++;
 
-               exit_code = unmodified_code = code;
-               exit_file = file;
-               exit_line = line;
+               first_code = code;
 
                if (output_needs_newline) {
                        fputc('\n', stdout);
@@ -139,7 +138,6 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
                                who_am_i(), code, file, line);
                }
 
-               /* FALLTHROUGH */
 #include "case_N.h"
                switch_step++;
 
@@ -148,38 +146,41 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
                        int pid = wait_process(cleanup_child_pid, &status, WNOHANG);
                        if (pid == cleanup_child_pid) {
                                status = WEXITSTATUS(status);
-                               if (status > code)
-                                       code = exit_code = status;
+                               if (status > exit_code)
+                                       exit_code = status;
                        }
                }
 
-               /* FALLTHROUGH */
 #include "case_N.h"
                switch_step++;
 
-               if (cleanup_got_literal && cleanup_fname && cleanup_new_fname
-                && keep_partial && handle_partial_dir(cleanup_new_fname, PDIR_CREATE)) {
-                       int tweak_modtime = 0;
-                       const char *fname = cleanup_fname;
-                       cleanup_fname = NULL;
-                       if (cleanup_fd_r != -1)
+               if (cleanup_got_literal && (cleanup_fname || cleanup_fd_w != -1)) {
+                       if (cleanup_fd_r != -1) {
                                close(cleanup_fd_r);
+                               cleanup_fd_r = -1;
+                       }
                        if (cleanup_fd_w != -1) {
                                flush_write_file(cleanup_fd_w);
                                close(cleanup_fd_w);
+                               cleanup_fd_w = -1;
                        }
-                       if (!partial_dir) {
-                           /* We don't want to leave a partial file with a modern time or it
-                            * could be skipped via --update.  Setting the time to something
-                            * really old also helps it to stand out as unfinished in an ls. */
-                           tweak_modtime = 1;
-                           cleanup_file->modtime = 0;
+                       if (cleanup_fname && cleanup_new_fname && keep_partial
+                        && handle_partial_dir(cleanup_new_fname, PDIR_CREATE)) {
+                               int tweak_modtime = 0;
+                               const char *fname = cleanup_fname;
+                               cleanup_fname = NULL;
+                               if (!partial_dir) {
+                                   /* We don't want to leave a partial file with a modern time or it
+                                    * could be skipped via --update.  Setting the time to something
+                                    * really old also helps it to stand out as unfinished in an ls. */
+                                   tweak_modtime = 1;
+                                   cleanup_file->modtime = 0;
+                               }
+                               finish_transfer(cleanup_new_fname, fname, NULL, NULL,
+                                               cleanup_file, tweak_modtime, !partial_dir);
                        }
-                       finish_transfer(cleanup_new_fname, fname, NULL, NULL,
-                                       cleanup_file, tweak_modtime, !partial_dir);
                }
 
-               /* FALLTHROUGH */
 #include "case_N.h"
                switch_step++;
 
@@ -188,16 +189,15 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
                        if (code == RERR_SIGNAL)
                                io_flush(FULL_FLUSH);
                }
-               if (!code)
+               if (!exit_code && !code)
                        io_flush(FULL_FLUSH);
 
-               /* FALLTHROUGH */
 #include "case_N.h"
                switch_step++;
 
                if (cleanup_fname)
                        do_unlink(cleanup_fname);
-               if (code)
+               if (exit_code)
                        kill_all(SIGUSR1);
                if (cleanup_pid && cleanup_pid == getpid()) {
                        char *pidf = lp_pid_file();
@@ -205,33 +205,34 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
                                unlink(lp_pid_file());
                }
 
-               if (code == 0) {
+               if (exit_code == 0) {
+                       if (code)
+                               exit_code = code;
                        if (io_error & IOERR_DEL_LIMIT)
-                               code = exit_code = RERR_DEL_LIMIT;
+                               exit_code = RERR_DEL_LIMIT;
                        if (io_error & IOERR_VANISHED)
-                               code = exit_code = RERR_VANISHED;
+                               exit_code = RERR_VANISHED;
                        if (io_error & IOERR_GENERAL || got_xfer_error)
-                               code = exit_code = RERR_PARTIAL;
+                               exit_code = RERR_PARTIAL;
                }
 
                /* If line < 0, this exit is after a MSG_ERROR_EXIT event, so
                 * we don't want to output a duplicate error. */
-               if ((code && line > 0)
+               if ((exit_code && line > 0)
                 || am_daemon || (logfile_name && (am_server || !INFO_GTE(STATS, 1))))
-                       log_exit(code, file, line);
+                       log_exit(exit_code, exit_file, exit_line);
 
-               /* FALLTHROUGH */
 #include "case_N.h"
                switch_step++;
 
                if (DEBUG_GTE(EXIT, 1)) {
                        rprintf(FINFO,
                                "[%s] _exit_cleanup(code=%d, file=%s, line=%d): "
-                               "about to call exit(%d)\n",
-                               who_am_i(), unmodified_code, file, line, code);
+                               "about to call exit(%d)%s\n",
+                               who_am_i(), first_code, exit_file, exit_line, exit_code,
+                               dry_run ? " (DRY RUN)" : "");
                }
 
-               /* FALLTHROUGH */
 #include "case_N.h"
                switch_step++;
 
@@ -247,11 +248,10 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
                        noop_io_until_death();
                }
 
-               /* FALLTHROUGH */
 #include "case_N.h"
                switch_step++;
 
-               if (am_server && code)
+               if (am_server && exit_code)
                        msleep(100);
                close_all();
 
@@ -260,12 +260,13 @@ NORETURN void _exit_cleanup(int code, const char *file, int line)
                break;
        }
 
-       exit(code);
+       exit(exit_code);
 }
 
 void cleanup_disable(void)
 {
        cleanup_fname = cleanup_new_fname = NULL;
+       cleanup_fd_r = cleanup_fd_w = -1;
        cleanup_got_literal = 0;
 }