X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=libbb%2Fxconnect.c;h=bc0691531dbba55056e912ab768b3ecd0c20c8b9;hb=666da5e2c6edec979966d16771818b32dcfafe04;hp=27459791ec1ea9b9f65cda99647a8272e706e1fc;hpb=e6dc439b3a3fa2a64f9e938ac4e5810025c04242;p=oweals%2Fbusybox.git

diff --git a/libbb/xconnect.c b/libbb/xconnect.c
index 27459791e..bc0691531 100644
--- a/libbb/xconnect.c
+++ b/libbb/xconnect.c
@@ -2,46 +2,44 @@
 /*
  * Utility routines.
  *
- * Connect to host at port using address resolusion from getaddrinfo
+ * Connect to host at port using address resolution from getaddrinfo
  *
  */
 
-#include <unistd.h>
-#include <string.h>
-#include <stdlib.h>
-#include <sys/types.h>
-#include <sys/socket.h>
-#include <errno.h>
-#include <netdb.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <arpa/inet.h>
 #include "libbb.h"
 
-int bb_getport(char *port)
+/* Return network byte ordered port number for a service.
+ * If "port" is a number use it as the port.
+ * If "port" is a name it is looked up in /etc/services, if it isnt found return
+ * default_port
+ */
+unsigned short bb_lookup_port(const char *port, const char *protocol, unsigned short default_port)
 {
-	int port_nr;
-	char *endptr;
-	struct servent *tserv;
+	unsigned short port_nr = htons(default_port);
+	if (port) {
+		char *endptr;
+		int old_errno;
+		long port_long;
 
-	if (!port) {
-		return -1;
-	}
-	port_nr=strtol(port, &endptr, 10);
-	if (errno != 0 || *endptr!='\0' || endptr==port || port_nr < 1 || port_nr > 65536) 
-	{
-		if (port_nr==0 && (tserv = getservbyname(port, "tcp")) != NULL) {
-			port_nr = tserv->s_port;
+		/* Since this is a lib function, we're not allowed to reset errno to 0.
+		 * Doing so could break an app that is deferring checking of errno. */
+		old_errno = errno;
+		errno = 0;
+		port_long = strtol(port, &endptr, 10);
+		if (errno != 0 || *endptr!='\0' || endptr==port || port_long < 0 || port_long > 65535) {
+			struct servent *tserv = getservbyname(port, protocol);
+			if (tserv) {
+				port_nr = tserv->s_port;
+			}
 		} else {
-			return -1;
+			port_nr = htons(port_long);
 		}
-	} else {
-		port_nr = htons(port_nr);
+		errno = old_errno;
 	}
 	return port_nr;
 }
 
-void bb_lookup_host(struct sockaddr_in *s_in, char *host, char *port)
+void bb_lookup_host(struct sockaddr_in *s_in, const char *host)
 {
 	struct hostent *he;
 
@@ -49,19 +47,108 @@ void bb_lookup_host(struct sockaddr_in *s_in, char *host, char *port)
 	s_in->sin_family = AF_INET;
 	he = xgethostbyname(host);
 	memcpy(&(s_in->sin_addr), he->h_addr_list[0], he->h_length);
+}
 
-	if (port) {
-		s_in->sin_port=bb_getport(port);
+void xconnect(int s, const struct sockaddr *s_addr, socklen_t addrlen)
+{
+	if (connect(s, s_addr, addrlen) < 0) {
+		if (ENABLE_FEATURE_CLEAN_UP) close(s);
+		if (s_addr->sa_family == AF_INET)
+			bb_perror_msg_and_die("%s (%s)",
+				"cannot connect to remote host",
+				inet_ntoa(((struct sockaddr_in *)s_addr)->sin_addr));
+		bb_perror_msg_and_die("cannot connect to remote host");
+	}
+}
+
+int xconnect_tcp_v4(struct sockaddr_in *s_addr)
+{
+	int s = xsocket(AF_INET, SOCK_STREAM, 0);
+	xconnect(s, (struct sockaddr*) s_addr, sizeof(*s_addr));
+	return s;
+}
+
+static const int one = 1;
+int setsockopt_reuseaddr(int fd)
+{
+	return setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
+}
+int setsockopt_broadcast(int fd)
+{
+	return setsockopt(fd, SOL_SOCKET, SO_BROADCAST, &one, sizeof(one));
+}
+
+int dotted2sockaddr(const char *dotted, struct sockaddr* sp, int socklen)
+{
+	union {
+		struct in_addr a4;
+#if ENABLE_FEATURE_IPV6
+		struct in6_addr a6;
+#endif
+	} a;
+
+	/* TODO maybe: port spec? like n.n.n.n:nn */
+
+#if ENABLE_FEATURE_IPV6
+	if (socklen >= sizeof(struct sockaddr_in6)
+	 && inet_pton(AF_INET6, dotted, &a.a6) > 0
+	) {
+		((struct sockaddr_in6*)sp)->sin6_family = AF_INET6;
+		((struct sockaddr_in6*)sp)->sin6_addr = a.a6;
+		/* ((struct sockaddr_in6*)sp)->sin6_port = */
+		return 0; /* success */
+	}
+#endif
+	if (socklen >= sizeof(struct sockaddr_in)
+	 && inet_pton(AF_INET, dotted, &a.a4) > 0
+	) {
+		((struct sockaddr_in*)sp)->sin_family = AF_INET;
+		((struct sockaddr_in*)sp)->sin_addr = a.a4;
+		/* ((struct sockaddr_in*)sp)->sin_port = */
+		return 0; /* success */
 	}
+	return 1;
 }
 
-int xconnect(struct sockaddr_in *s_addr)
+int xsocket_stream_ip4or6(sa_family_t *fp)
 {
-	int s = socket(AF_INET, SOCK_STREAM, 0);
-	if (connect(s, (struct sockaddr_in *)s_addr, sizeof(struct sockaddr_in)) < 0)
+	int fd;
+#if ENABLE_FEATURE_IPV6
+	fd = socket(AF_INET6, SOCK_STREAM, 0);
+	if (fp) *fp = AF_INET6;
+	if (fd < 0)
+#endif
 	{
-		bb_perror_msg_and_die("Unable to connect to remote host (%s)", 
-				inet_ntoa(s_addr->sin_addr));
+		fd = xsocket(AF_INET, SOCK_STREAM, 0);
+		if (fp) *fp = AF_INET;
 	}
-	return s;
+	return fd;
+}
+
+int create_and_bind_socket_ip4or6(const char *hostaddr, int port)
+{
+	int fd;
+	sockaddr_inet sa;
+
+	memset(&sa, 0, sizeof(sa));
+	if (hostaddr) {
+		if (dotted2sockaddr(hostaddr, &sa.sa, sizeof(sa)))
+			bb_error_msg_and_die("bad address '%s'", hostaddr);
+		/* user specified bind addr dictates family */
+		fd = xsocket(sa.sa.sa_family, SOCK_STREAM, 0);
+	} else 
+		fd = xsocket_stream_ip4or6(&sa.sa.sa_family);
+	setsockopt_reuseaddr(fd);
+
+	/* if (port >= 0) { */
+#if ENABLE_FEATURE_IPV6
+		if (sa.sa.sa_family == AF_INET6 /* && !sa.sin6.sin6_port */)
+			sa.sin6.sin6_port = htons(port);
+#endif
+		if (sa.sa.sa_family == AF_INET /* && !sa.sin.sin_port */)
+			sa.sin.sin_port = htons(port);
+	/* } */
+
+	xbind(fd, &sa.sa, sizeof(sa));
+	return fd;
 }