Unified the externs.
[rsync/rsync.git] / main.c
diff --git a/main.c b/main.c
index 1b07eee..619578f 100644 (file)
--- a/main.c
+++ b/main.c
 time_t starttime = 0;
 
 extern struct stats stats;
-extern char *files_from;
-extern int filesfrom_fd;
-extern char *remote_filesfrom_file;
+extern int am_root;
 extern int am_server;
 extern int am_sender;
 extern int am_daemon;
 extern int verbose;
+extern int blocking_io;
+extern int cvs_exclude;
+extern int delete_mode;
+extern int delete_excluded;
+extern int delete_after;
+extern int daemon_over_rsh;
+extern int do_stats;
+extern int dry_run;
+extern int list_only;
+extern int local_server;
+extern int log_got_error;
+extern int module_id;
+extern int orig_umask;
+extern int preserve_hard_links;
 extern int protocol_version;
+extern int recurse;
+extern int relative_paths;
+extern int rsync_port;
+extern int read_batch;
+extern int write_batch;
+extern int filesfrom_fd;
+extern pid_t cleanup_child_pid;
+extern char *files_from;
+extern char *remote_filesfrom_file;
+extern char *rsync_path;
+extern char *shell_cmd;
+extern struct file_list *batch_flist;
 
 /* there's probably never more than at most 2 outstanding child processes,
  * but set it higher just in case.
@@ -55,7 +79,7 @@ void wait_process(pid_t pid, int *status)
 
        while ((waited_pid = waitpid(pid, status, WNOHANG)) == 0) {
                msleep(20);
-               io_flush();
+               io_flush(FULL_FLUSH);
        }
 
        if ((waited_pid == -1) && (errno == ECHILD)) {
@@ -82,7 +106,6 @@ void wait_process(pid_t pid, int *status)
 static void report(int f)
 {
        time_t t = time(NULL);
-       extern int do_stats;
        int send_stats;
 
        if (do_stats && verbose > 1) {
@@ -198,11 +221,6 @@ static pid_t do_cmd(char *cmd,char *machine,char *user,char *path,int *f_in,int
        pid_t ret;
        char *tok,*dir=NULL;
        int dash_l_set = 0;
-       extern int local_server;
-       extern char *rsync_path;
-       extern int blocking_io;
-       extern int daemon_over_rsh;
-       extern int read_batch;
 
        if (!read_batch && !local_server) {
                char *rsh_env = getenv(RSYNC_RSH_ENV);
@@ -293,7 +311,6 @@ static char *get_local_name(struct file_list *flist,char *name)
 {
        STRUCT_STAT st;
        int e;
-       extern int orig_umask;
 
        if (verbose > 2)
                rprintf(FINFO,"get_local_name count=%d %s\n",
@@ -340,15 +357,11 @@ static char *get_local_name(struct file_list *flist,char *name)
 }
 
 
-
-
 static void do_server_sender(int f_in, int f_out, int argc,char *argv[])
 {
        int i;
        struct file_list *flist;
        char *dir = argv[0];
-       extern int relative_paths;
-       extern int recurse;
 
        if (verbose > 2)
                rprintf(FINFO,"server_sender starting pid=%d\n",(int)getpid());
@@ -383,13 +396,13 @@ static void do_server_sender(int f_in, int f_out, int argc,char *argv[])
        io_start_buffering_in(f_in);
        io_start_buffering_out(f_out);
        send_files(flist,f_out,f_in);
-       io_flush();
+       io_flush(FULL_FLUSH);
        report(f_out);
        if (protocol_version >= 24) {
                /* final goodbye message */
                read_int(f_in);
        }
-       io_flush();
+       io_flush(FULL_FLUSH);
        exit_cleanup(0);
 }
 
@@ -398,12 +411,7 @@ static int do_recv(int f_in,int f_out,struct file_list *flist,char *local_name)
 {
        int pid;
        int status=0;
-       int recv_pipe[2];
        int error_pipe[2];
-       extern int preserve_hard_links;
-       extern int delete_after;
-       extern int recurse;
-       extern int delete_mode;
 
        if (preserve_hard_links)
                init_hard_links(flist);
@@ -415,20 +423,14 @@ static int do_recv(int f_in,int f_out,struct file_list *flist,char *local_name)
                }
        }
 
-       if (fd_pair(recv_pipe) < 0) {
-               rprintf(FERROR,"pipe failed in do_recv\n");
-               exit_cleanup(RERR_SOCKETIO);
-       }
-
        if (fd_pair(error_pipe) < 0) {
                rprintf(FERROR,"error pipe failed in do_recv\n");
                exit_cleanup(RERR_SOCKETIO);
        }
 
-       io_flush();
+       io_flush(NORMAL_FLUSH);
 
        if ((pid=do_fork()) == 0) {
-               close(recv_pipe[0]);
                close(error_pipe[0]);
                if (f_in != f_out) close(f_out);
 
@@ -436,41 +438,39 @@ static int do_recv(int f_in,int f_out,struct file_list *flist,char *local_name)
                io_multiplexing_close();
 
                /* set place to send errors */
-               set_error_fd(error_pipe[1]);
+               set_msg_fd_out(error_pipe[1]);
 
-               recv_files(f_in,flist,local_name,recv_pipe[1]);
-               io_flush();
+               recv_files(f_in,flist,local_name);
+               io_flush(FULL_FLUSH);
                report(f_in);
 
-               write_int(recv_pipe[1],1);
-               close(recv_pipe[1]);
-               io_flush();
+               send_msg(MSG_DONE, "", 0);
+               io_flush(FULL_FLUSH);
                /* finally we go to sleep until our parent kills us
                 * with a USR2 signal. We sleep for a short time as on
                 * some OSes a signal won't interrupt a sleep! */
-               while (msleep(20))
-                       ;
+               while (1)
+                       msleep(20);
        }
 
-       close(recv_pipe[1]);
        close(error_pipe[1]);
        if (f_in != f_out) close(f_in);
 
        io_start_buffering_out(f_out);
 
-       io_set_error_fd(error_pipe[0]);
+       set_msg_fd_in(error_pipe[0]);
 
-       generate_files(f_out,flist,local_name,recv_pipe[0]);
+       generate_files(f_out, flist, local_name);
 
-       read_int(recv_pipe[0]);
-       close(recv_pipe[0]);
+       get_redo_num(); /* Read final -1, and any prior messages. */
+       io_flush(FULL_FLUSH);
        if (protocol_version >= 24) {
                /* send a final goodbye message */
                write_int(f_out, -1);
        }
-       io_flush();
+       io_flush(FULL_FLUSH);
 
-       io_set_error_fd(-1);
+       set_msg_fd_in(-1);
        kill(pid, SIGUSR2);
        wait_process(pid, &status);
        return status;
@@ -483,11 +483,6 @@ static void do_server_recv(int f_in, int f_out, int argc,char *argv[])
        struct file_list *flist;
        char *local_name=NULL;
        char *dir = NULL;
-       extern int delete_mode;
-       extern int delete_excluded;
-       extern int module_id;
-       extern int read_batch;
-       extern struct file_list *batch_flist;
 
        if (verbose > 2)
                rprintf(FINFO,"server_recv(%d) starting pid=%d\n",argc,(int)getpid());
@@ -554,9 +549,6 @@ int child_main(int argc, char *argv[])
 
 void start_server(int f_in, int f_out, int argc, char *argv[])
 {
-       extern int cvs_exclude;
-       extern int read_batch;
-
        setup_protocol(f_out, f_in);
 
        set_nonblocking(f_in);
@@ -588,10 +580,6 @@ int client_run(int f_in, int f_out, pid_t pid, int argc, char *argv[])
        struct file_list *flist = NULL;
        int status = 0, status2 = 0;
        char *local_name = NULL;
-       extern pid_t cleanup_child_pid;
-       extern int write_batch;
-       extern int read_batch;
-       extern struct file_list *batch_flist;
 
        cleanup_child_pid = pid;
        if (read_batch)
@@ -606,9 +594,6 @@ int client_run(int f_in, int f_out, pid_t pid, int argc, char *argv[])
                io_start_multiplex_in(f_in);
 
        if (am_sender) {
-               extern int cvs_exclude;
-               extern int delete_mode;
-               extern int delete_excluded;
                io_start_buffering_out(f_out);
                if (cvs_exclude)
                        add_cvs_excludes();
@@ -621,10 +606,10 @@ int client_run(int f_in, int f_out, pid_t pid, int argc, char *argv[])
                if (verbose > 3)
                        rprintf(FINFO,"file list sent\n");
 
-               io_flush();
+               io_flush(NORMAL_FLUSH);
                io_start_buffering_out(f_out);
                send_files(flist,f_out,f_in);
-               io_flush();
+               io_flush(FULL_FLUSH);
                if (protocol_version >= 24) {
                        /* final goodbye message */
                        read_int(f_in);
@@ -632,16 +617,15 @@ int client_run(int f_in, int f_out, pid_t pid, int argc, char *argv[])
                if (pid != -1) {
                        if (verbose > 3)
                                rprintf(FINFO,"client_run waiting on %d\n", (int) pid);
-                       io_flush();
+                       io_flush(FULL_FLUSH);
                        wait_process(pid, &status);
                }
                report(-1);
-               io_flush();
+               io_flush(FULL_FLUSH);
                exit_cleanup(status);
        }
 
        if (argc == 0) {
-               extern int list_only;
                list_only = 1;
        }
 
@@ -668,7 +652,7 @@ int client_run(int f_in, int f_out, pid_t pid, int argc, char *argv[])
        if (pid != -1) {
                if (verbose > 3)
                        rprintf(FINFO,"client_run2 waiting on %d\n", (int) pid);
-               io_flush();
+               io_flush(FULL_FLUSH);
                wait_process(pid, &status);
        }
 
@@ -708,11 +692,6 @@ static int start_client(int argc, char *argv[])
        int ret;
        pid_t pid;
        int f_in,f_out;
-       extern int local_server;
-       extern char *shell_cmd;
-       extern int rsync_port;
-       extern int daemon_over_rsh;
-       extern int read_batch;
        int rc;
 
        /* Don't clobber argv[] so that ps(1) can still show the right
@@ -863,7 +842,6 @@ static int start_client(int argc, char *argv[])
        }
 
        if (argc == 0 && !am_sender) {
-               extern int list_only;
                list_only = 1;
        }
 
@@ -896,7 +874,6 @@ static RETSIGTYPE sigusr1_handler(UNUSED(int val))
 
 static RETSIGTYPE sigusr2_handler(UNUSED(int val))
 {
-       extern int log_got_error;
        if (log_got_error) _exit(RERR_PARTIAL);
        _exit(0);
 }
@@ -979,11 +956,7 @@ static RETSIGTYPE rsync_panic_handler(UNUSED(int whatsig))
 
 int main(int argc,char *argv[])
 {
-       extern int am_root;
-       extern int orig_umask;
-       extern int dry_run;
        int ret;
-       extern int write_batch;
        int orig_argc;
        char **orig_argv;