Remove dependency from socketpair.c on address.h

Also refactor some annoying code in tor_ersatz_socketpair.
This commit is contained in:
Nick Mathewson 2018-08-01 09:24:06 -04:00
parent 2884639ad6
commit 9b24609af0
2 changed files with 92 additions and 60 deletions

View File

@ -2,6 +2,7 @@ orconfig.h
siphash.h siphash.h
ht.h ht.h
lib/arch/*.h
lib/cc/*.h lib/cc/*.h
lib/container/*.h lib/container/*.h
lib/ctime/*.h lib/ctime/*.h

View File

@ -4,22 +4,81 @@
#include "lib/net/socketpair.h" #include "lib/net/socketpair.h"
#include "lib/net/socket.h" #include "lib/net/socket.h"
#include "lib/net/address.h" #include "lib/arch/bytes.h"
#include <errno.h> #include <errno.h>
#include <string.h> #include <string.h>
#ifdef HAVE_NETINET_IN_H
#include <netinet/in.h>
#endif
#ifdef _WIN32
#include <winsock2.h>
#include <windows.h>
#endif
#ifdef NEED_ERSATZ_SOCKETPAIR #ifdef NEED_ERSATZ_SOCKETPAIR
static inline socklen_t // Avoid warning about call to memcmp.
SIZEOF_SOCKADDR(int domain) #define raw_memcmp memcmp
static tor_socket_t
get_local_listener(int family, int type)
{ {
switch (domain) { struct sockaddr_in sin;
case AF_INET: struct sockaddr_in6 sin6;
return sizeof(struct sockaddr_in); struct sockaddr *sa;
case AF_INET6: int len;
return sizeof(struct sockaddr_in6);
default: memset(&sin, 0, sizeof(sin));
memset(&sin6, 0, sizeof(sin6));
tor_socket_t sock = TOR_INVALID_SOCKET;
sock = tor_open_socket(family, type, 0);
if (!SOCKET_OK(sock)) {
return TOR_INVALID_SOCKET;
}
if (family == AF_INET) {
sa = (struct sockaddr *) &sin;
sin.sin_family = AF_INET;
sin.sin_addr.s_addr = tor_htonl(0x7f000001);
len = sizeof(sin);
} else {
sa = (struct sockaddr *) &sin6;
sin6.sin6_family = AF_INET;
sin6.sin6_addr.s6_addr[15] = 1;
len = sizeof(sin6);
}
if (bind(sock, sa, len) == -1)
goto err;
if (listen(sock, 1) == -1)
goto err;
return sock;
err:
tor_close_socket(sock);
return TOR_INVALID_SOCKET;
}
static int
sockaddr_eq(struct sockaddr *sa1, struct sockaddr *sa2)
{
if (sa1->sa_family != sa2->sa_family)
return 0;
if (sa1->sa_family == AF_INET6) {
struct sockaddr_in6 *sin6_1 = (struct sockaddr_in6 *) sa1;
struct sockaddr_in6 *sin6_2 = (struct sockaddr_in6 *) sa2;
return sin6_1->sin6_port == sin6_2->sin6_port &&
0==raw_memcmp(sin6_1->sin6_addr.s6_addr, sin6_2->sin6_addr.s6_addr, 16);
} else if (sa1->sa_family == AF_INET) {
struct sockaddr_in *sin_1 = (struct sockaddr_in *) sa1;
struct sockaddr_in *sin_2 = (struct sockaddr_in *) sa2;
return sin_1->sin_port == sin_2->sin_port &&
sin_1->sin_addr.s_addr == sin_2->sin_addr.s_addr;
} else {
return 0; return 0;
} }
} }
@ -39,21 +98,17 @@ tor_ersatz_socketpair(int family, int type, int protocol, tor_socket_t fd[2])
tor_socket_t listener = TOR_INVALID_SOCKET; tor_socket_t listener = TOR_INVALID_SOCKET;
tor_socket_t connector = TOR_INVALID_SOCKET; tor_socket_t connector = TOR_INVALID_SOCKET;
tor_socket_t acceptor = TOR_INVALID_SOCKET; tor_socket_t acceptor = TOR_INVALID_SOCKET;
tor_addr_t listen_tor_addr; struct sockaddr_storage accepted_addr_ss;
struct sockaddr_storage connect_addr_ss, listen_addr_ss; struct sockaddr_storage connect_addr_ss;
struct sockaddr *listen_addr = (struct sockaddr *) &listen_addr_ss;
uint16_t listen_port = 0;
tor_addr_t connect_tor_addr;
uint16_t connect_port = 0;
struct sockaddr *connect_addr = (struct sockaddr *) &connect_addr_ss; struct sockaddr *connect_addr = (struct sockaddr *) &connect_addr_ss;
struct sockaddr *accepted_addr = (struct sockaddr *) &accepted_addr_ss;
socklen_t size; socklen_t size;
int saved_errno = -1; int saved_errno = -1;
int ersatz_domain = AF_INET; int ersatz_domain = AF_INET;
socklen_t addrlen = sizeof(struct sockaddr_in);
memset(&connect_tor_addr, 0, sizeof(connect_tor_addr)); memset(&accepted_addr_ss, 0, sizeof(accepted_addr_ss));
memset(&connect_addr_ss, 0, sizeof(connect_addr_ss)); memset(&connect_addr_ss, 0, sizeof(connect_addr_ss));
memset(&listen_tor_addr, 0, sizeof(listen_tor_addr));
memset(&listen_addr_ss, 0, sizeof(listen_addr_ss));
if (protocol if (protocol
#ifdef AF_UNIX #ifdef AF_UNIX
@ -70,14 +125,15 @@ tor_ersatz_socketpair(int family, int type, int protocol, tor_socket_t fd[2])
return -EINVAL; return -EINVAL;
} }
listener = tor_open_socket(ersatz_domain, type, 0); listener = get_local_listener(ersatz_domain, type);
if (!SOCKET_OK(listener)) { if (!SOCKET_OK(listener)) {
int first_errno = tor_socket_errno(-1); int first_errno = tor_socket_errno(-1);
if (first_errno == SOCK_ERRNO(EPROTONOSUPPORT) if (first_errno == SOCK_ERRNO(EPROTONOSUPPORT)) {
&& ersatz_domain == AF_INET) {
/* Assume we're on an IPv6-only system */ /* Assume we're on an IPv6-only system */
ersatz_domain = AF_INET6; ersatz_domain = AF_INET6;
listener = tor_open_socket(ersatz_domain, type, 0); addrlen = sizeof(struct sockaddr_in6);
listener = get_local_listener(ersatz_domain, type);
}
if (!SOCKET_OK(listener)) { if (!SOCKET_OK(listener)) {
/* Keep the previous behaviour, which was to return the IPv4 error. /* Keep the previous behaviour, which was to return the IPv4 error.
* (This may be less informative on IPv6-only systems.) * (This may be less informative on IPv6-only systems.)
@ -87,25 +143,6 @@ tor_ersatz_socketpair(int family, int type, int protocol, tor_socket_t fd[2])
return -first_errno; return -first_errno;
} }
} }
}
/* If there is no 127.0.0.1 or ::1, this will and must fail. Otherwise, we
* risk exposing a socketpair on a routable IP address. (Some BSD jails
* use a routable address for localhost. Fortunately, they have the real
* AF_UNIX socketpair.) */
if (ersatz_domain == AF_INET) {
tor_addr_from_ipv4h(&listen_tor_addr, INADDR_LOOPBACK);
} else {
tor_addr_parse(&listen_tor_addr, "[::1]");
}
tor_assert(tor_addr_is_loopback(&listen_tor_addr));
size = tor_addr_to_sockaddr(&listen_tor_addr,
0 /* kernel chooses port. */,
listen_addr,
sizeof(listen_addr_ss));
if (bind(listener, listen_addr, size) == -1)
goto tidy_up_and_fail;
if (listen(listener, 1) == -1)
goto tidy_up_and_fail;
connector = tor_open_socket(ersatz_domain, type, 0); connector = tor_open_socket(ersatz_domain, type, 0);
if (!SOCKET_OK(connector)) if (!SOCKET_OK(connector))
@ -114,33 +151,27 @@ tor_ersatz_socketpair(int family, int type, int protocol, tor_socket_t fd[2])
size = sizeof(connect_addr_ss); size = sizeof(connect_addr_ss);
if (getsockname(listener, connect_addr, &size) == -1) if (getsockname(listener, connect_addr, &size) == -1)
goto tidy_up_and_fail; goto tidy_up_and_fail;
if (size != SIZEOF_SOCKADDR (connect_addr->sa_family)) if (size != addrlen)
goto abort_tidy_up_and_fail; goto abort_tidy_up_and_fail;
if (connect(connector, connect_addr, size) == -1) if (connect(connector, connect_addr, size) == -1)
goto tidy_up_and_fail; goto tidy_up_and_fail;
size = sizeof(listen_addr_ss); size = sizeof(accepted_addr_ss);
acceptor = tor_accept_socket(listener, listen_addr, &size); acceptor = tor_accept_socket(listener, accepted_addr, &size);
if (!SOCKET_OK(acceptor)) if (!SOCKET_OK(acceptor))
goto tidy_up_and_fail; goto tidy_up_and_fail;
if (size != SIZEOF_SOCKADDR(listen_addr->sa_family)) if (size != addrlen)
goto abort_tidy_up_and_fail; goto abort_tidy_up_and_fail;
/* Now check we are talking to ourself by matching port and host on the /* Now check we are talking to ourself by matching port and host on the
two sockets. */ two sockets. */
if (getsockname(connector, connect_addr, &size) == -1) if (getsockname(connector, connect_addr, &size) == -1)
goto tidy_up_and_fail; goto tidy_up_and_fail;
/* Set *_tor_addr and *_port to the address and port that was used */ /* Set *_tor_addr and *_port to the address and port that was used */
tor_addr_from_sockaddr(&listen_tor_addr, listen_addr, &listen_port); if (!sockaddr_eq(accepted_addr, connect_addr))
tor_addr_from_sockaddr(&connect_tor_addr, connect_addr, &connect_port);
if (size != SIZEOF_SOCKADDR (connect_addr->sa_family)
|| tor_addr_compare(&listen_tor_addr, &connect_tor_addr, CMP_SEMANTIC)
|| listen_port != connect_port) {
goto abort_tidy_up_and_fail; goto abort_tidy_up_and_fail;
}
tor_close_socket(listener); tor_close_socket(listener);
fd[0] = connector; fd[0] = connector;
fd[1] = acceptor; fd[1] = acceptor;
return 0; return 0;
abort_tidy_up_and_fail: abort_tidy_up_and_fail: