X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/56c473b795bad9d29430bf794f9204bd096c9c3d..91eee5946ad6d033303abfe184085c87a283da29:/clientserver.c

diff --git a/clientserver.c b/clientserver.c
index 32deee76..39e5b90d 100644
--- a/clientserver.c
+++ b/clientserver.c
@@ -32,7 +32,7 @@ int start_socket_client(char *host, char *path, int argc, char *argv[])
 	int sargc=0;
 	char line[1024];
 	char *p;
-	int version;
+	extern int remote_version;
 
 	fd = open_socket_out(host, rsync_port);
 	if (fd == -1) {
@@ -48,16 +48,16 @@ int start_socket_client(char *host, char *path, int argc, char *argv[])
 
 	sargs[sargc] = NULL;
 
+	io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION);
+
 	if (!read_line(fd, line, sizeof(line)-1)) {
 		return -1;
 	}
 
-	if (sscanf(line,"@RSYNCD: %d", &version) != 1) {
+	if (sscanf(line,"@RSYNCD: %d", &remote_version) != 1) {
 		return -1;
 	}
 
-	io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION);
-
 	p = strchr(path,'/');
 	if (p) *p = 0;
 	io_printf(fd,"%s\n",path);
@@ -108,6 +108,11 @@ static int rsync_module(int fd, int i)
 		return -1;
 	}
 
+	if (!claim_connection(lp_lock_file(), lp_max_connections())) {
+		rprintf(FERROR,"ERROR: max connections reached\n");
+		return -1;
+	}
+
 	rprintf(FINFO,"rsync on module %s from %s (%s)\n",
 		lp_name(i), host, addr);
 
@@ -209,9 +214,9 @@ static int start_daemon(int fd)
 {
 	char line[200];
 	char *motd;
-	int version;
 	int i = -1;
 	extern char *config_file;
+	extern int remote_version;
 
 	if (!lp_load(config_file)) {
 		exit_cleanup(1);
@@ -221,14 +226,6 @@ static int start_daemon(int fd)
 
 	io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION);
 
-	if (!read_line(fd, line, sizeof(line)-1)) {
-		return -1;
-	}
-
-	if (sscanf(line,"@RSYNCD: %d", &version) != 1) {
-		return -1;
-	}	
-
 	motd = lp_motd_file();
 	if (*motd) {
 		FILE *f = fopen(motd,"r");
@@ -243,6 +240,14 @@ static int start_daemon(int fd)
 		io_printf(fd,"\n");
 	}
 
+	if (!read_line(fd, line, sizeof(line)-1)) {
+		return -1;
+	}
+
+	if (sscanf(line,"@RSYNCD: %d", &remote_version) != 1) {
+		return -1;
+	}	
+
 	while (i == -1) {
 
 		line[0] = 0;