- always flush the IO write buffer when reading

[rsync/rsync.git] / clientserver.c

diff --git a/clientserver.c b/clientserver.c
index 765cba7..63ff423 100644 (file)
--- a/clientserver.c
+++ b/clientserver.c
@@ -33,6 +33,8 @@ int start_socket_client(char *host, char *path, int argc, char *argv[])
        char line[MAXPATHLEN];
        char *p, *user=NULL;
        extern int remote_version;
+       extern int am_client;
+       extern int am_sender;
 
        p = strchr(host, '@');
        if (p) {
@@ -44,6 +46,8 @@ int start_socket_client(char *host, char *path, int argc, char *argv[])
        if (!user) user = getenv("USER");
        if (!user) user = getenv("LOGNAME");
 
+       am_client = 1;
+
        fd = open_socket_out(host, rsync_port);
        if (fd == -1) {
                exit_cleanup(1);
@@ -92,6 +96,9 @@ int start_socket_client(char *host, char *path, int argc, char *argv[])
        }
        io_printf(fd,"\n");
 
+       if (remote_version > 17 && !am_sender)
+               io_start_multiplex_in(fd);
+
        return client_run(fd, fd, -1, argc, argv);
 }
 
@@ -109,9 +116,12 @@ static int rsync_module(int fd, int i)
        char *addr = client_addr(fd);
        char *host = client_name(fd);
        char *name = lp_name(i);
+       char *user;
        int start_glob=0;
        char *request=NULL;
        extern int am_sender;
+       extern int remote_version;
+       extern int am_root;
 
        if (!allow_access(addr, host, lp_hosts_allow(i), lp_hosts_deny(i))) {
                rprintf(FERROR,"rsync denied on module %s from %s (%s)\n",
@@ -121,13 +131,6 @@ static int rsync_module(int fd, int i)
                return -1;
        }
 
-       if (!auth_server(fd, i, addr, "@RSYNCD: AUTHREQD ")) {
-               rprintf(FERROR,"auth failed on module %s from %s (%s)\n",
-                       name, client_name(fd), client_addr(fd));
-               io_printf(fd,"@ERROR: auth failed on module %s\n",name);
-               return -1;              
-       }
-
        if (!claim_connection(lp_lock_file(), lp_max_connections())) {
                rprintf(FERROR,"max connections (%d) reached\n",
                        lp_max_connections());
@@ -136,6 +139,15 @@ static int rsync_module(int fd, int i)
        }
 
        
+       user = auth_server(fd, i, addr, "@RSYNCD: AUTHREQD ");
+
+       if (!user) {
+               rprintf(FERROR,"auth failed on module %s from %s (%s)\n",
+                       name, client_name(fd), client_addr(fd));
+               io_printf(fd,"@ERROR: auth failed on module %s\n",name);
+               return -1;              
+       }
+
        module_id = i;
 
        if (lp_read_only(i))
@@ -162,7 +174,7 @@ static int rsync_module(int fd, int i)
        }
 
        p = lp_exclude_from(i);
-       add_exclude_file(p, 1);
+       add_exclude_file(p, 1, 0);
 
        p = lp_exclude(i);
        add_exclude_line(p);
@@ -193,6 +205,8 @@ static int rsync_module(int fd, int i)
                return -1;
        }
 
+       am_root = (getuid() == 0);
+
        io_printf(fd,"@RSYNCD: OK\n");
 
        argv[argc++] = "rsyncd";
@@ -233,9 +247,15 @@ static int rsync_module(int fd, int i)
        parse_arguments(argc, argv);
 
        if (request) {
-               rprintf(FINFO,"rsync %s %s from %s (%s)\n",
-                       am_sender?"on":"to",
-                       request, host, addr);
+               if (*user) {
+                       rprintf(FINFO,"rsync %s %s from %s@%s (%s)\n",
+                               am_sender?"on":"to",
+                               request, user, host, addr);
+               } else {
+                       rprintf(FINFO,"rsync %s %s from %s (%s)\n",
+                               am_sender?"on":"to",
+                               request, host, addr);
+               }
                free(request);
        }
 
@@ -246,6 +266,9 @@ static int rsync_module(int fd, int i)
        argp = argv + optind;
        optind = 0;
 
+       if (remote_version > 17 && am_sender)
+               io_start_multiplex_out(fd);
+
        start_server(fd, fd, argc, argp);
 
        return 0;