X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/a6d8c3f336deebfff152e1d3f6d646ba1dc59eea..e72b18a9bd9f89847e49e87c1159c74d7bf0fbf3:/clientname.c

diff --git a/clientname.c b/clientname.c
index caf0af6c..661d4ab6 100644
--- a/clientname.c
+++ b/clientname.c
@@ -34,6 +34,8 @@
 #include "rsync.h"
 
 static const char default_name[] = "UNKNOWN";
+extern int am_daemon;
+extern int am_server;
 
 
 /**
@@ -44,6 +46,7 @@ char *client_addr(int fd)
 	struct sockaddr_storage ss;
 	socklen_t length = sizeof ss;
 	char *ssh_client, *p;
+	int len;
 	static char addr_buf[100];
 	static int initialised;
 
@@ -51,19 +54,24 @@ char *client_addr(int fd)
 
 	initialised = 1;
 
-	if ((ssh_client = getenv("SSH_CLIENT")) != NULL) {
-		strlcpy(addr_buf, ssh_client, sizeof(addr_buf));
-		/* truncate SSH_CLIENT to just IP address */
-		p = strchr(addr_buf, ' ');
-		if (p)
-			*p = '\0';
-		else
-			strlcpy(addr_buf, "0.0.0.0", sizeof("0.0.0.0"));
-	} else
+	if (am_server) {
+		/* daemon over --rsh mode */
+		strcpy(addr_buf, "0.0.0.0");
+		if ((ssh_client = getenv("SSH_CLIENT")) != NULL) {
+			/* truncate SSH_CLIENT to just IP address */
+			p = strchr(ssh_client, ' ');
+			if (p) {
+				len = MIN((unsigned int) (p - ssh_client), 
+						sizeof(addr_buf) - 1);
+				strncpy(addr_buf, ssh_client, len);
+				*(addr_buf + len) = '\0';
+			}
+		}
+	} else {
 		client_sockaddr(fd, &ss, &length);
-
-	getnameinfo((struct sockaddr *)&ss, length,
-		addr_buf, sizeof(addr_buf), NULL, 0, NI_NUMERICHOST);
+		getnameinfo((struct sockaddr *)&ss, length,
+			    addr_buf, sizeof addr_buf, NULL, 0, NI_NUMERICHOST);
+	}
 
 	return addr_buf;
 }
@@ -93,47 +101,59 @@ char *client_name(int fd)
 	static char name_buf[100];
 	static char port_buf[100];
 	static int initialised;
+	struct sockaddr_storage ss, *ssp;
+	struct sockaddr_in sin;
+#ifdef INET6
+	struct sockaddr_in6 sin6;
+#endif
+	socklen_t ss_len;
 
 	if (initialised) return name_buf;
 
 	strcpy(name_buf, default_name);
 	initialised = 1;
 
-	if (getenv("SSH_CLIENT") != NULL) {
-		/* Look up name of IP address given in $SSH_CLIENT */
-#ifdef INET6
-		int af = AF_INET6;
-		struct sockaddr_in6 sin;
-#else
-		int af = AF_INET;
-		struct sockaddr_in sin;
-#endif
-		socklen_t sin_len = sizeof sin;
-
-		memset(&sin, 0, sin_len);
+	if (am_server) {
+		/* daemon over --rsh mode */
 
+		char *addr = client_addr(fd);
 #ifdef INET6
-		sin.sin6_family = af;
-		inet_pton(af, client_addr(fd), &sin.sin6_addr.s6_addr);
-#else
-		sin.sin_family = af;
-		inet_pton(af, client_addr(fd), &sin.sin_addr.s_addr);
+		int dots = 0;
+		char *p;
+
+		for (p = addr; *p && (dots <= 3); p++) {
+		    if (*p == '.')
+			dots++;
+		}
+		if (dots > 3) {
+			/* more than 4 parts to IP address, must be ipv6 */
+			ssp = (struct sockaddr_storage *) &sin6;
+			ss_len = sizeof sin6;
+			memset(ssp, 0, ss_len);
+			inet_pton(AF_INET6, addr, &sin6.sin6_addr);
+			sin6.sin6_family = AF_INET6;
+		} else
 #endif
+		{
+			ssp = (struct sockaddr_storage *) &sin;
+			ss_len = sizeof sin;
+			memset(ssp, 0, ss_len);
+			inet_pton(AF_INET, addr, &sin.sin_addr);
+			sin.sin_family = AF_INET;
+		}
 
-		if (!lookup_name(fd, (struct sockaddr_storage *)&sin, sin_len, 
-				name_buf, sizeof name_buf, port_buf, sizeof port_buf))
-			check_name(fd, (struct sockaddr_storage *)&sin, name_buf);
 	} else {
-		struct sockaddr_storage ss;
-		socklen_t ss_len = sizeof ss;
+		ss_len = sizeof ss;
+		ssp = &ss;
 
 		client_sockaddr(fd, &ss, &ss_len);
 
-		if (!lookup_name(fd, &ss, ss_len, name_buf, sizeof name_buf, 
-				port_buf, sizeof port_buf))
-			check_name(fd, &ss, name_buf);
 	}
 
+	if (!lookup_name(fd, ssp, ss_len, name_buf, sizeof name_buf, 
+			port_buf, sizeof port_buf))
+		check_name(fd, ssp, name_buf);
+
 	return name_buf;
 }
 
@@ -149,6 +169,8 @@ void client_sockaddr(int fd,
 		     struct sockaddr_storage *ss,
 		     socklen_t *ss_len)
 {
+	memset(ss, 0, sizeof(*ss));
+
 	if (getpeername(fd, (struct sockaddr *) ss, ss_len)) {
 		/* FIXME: Can we really not continue? */
 		rprintf(FERROR, RSYNC_NAME ": getpeername on fd%d failed: %s\n",
@@ -252,9 +274,23 @@ int compare_addrinfo_sockaddr(const struct addrinfo *ai,
 
 		sin1 = (const struct sockaddr_in6 *) ss;
 		sin2 = (const struct sockaddr_in6 *) ai->ai_addr;
-		
-		return memcmp(&sin1->sin6_addr, &sin2->sin6_addr,
-			      sizeof sin1->sin6_addr);
+
+		if (ai->ai_addrlen < sizeof(struct sockaddr_in6)) {
+			rprintf(FERROR,
+				"%s: too short sockaddr_in6; length=%d\n",
+				fn, ai->ai_addrlen);
+			return 1;
+		}
+
+		if (memcmp(&sin1->sin6_addr, &sin2->sin6_addr,
+			   sizeof sin1->sin6_addr))
+			return 1;
+
+#ifdef HAVE_SOCKADDR_IN6_SCOPE_ID
+		if (sin1->sin6_scope_id != sin2->sin6_scope_id)
+			return 1;
+#endif
+		return 0;
 	}
 #endif /* INET6 */
 	else {