sockets: strengthen test suite IP protocol availability checks

Instead of just checking whether it is possible to bind() on a socket, also
check that we can successfully connect() to the socket we bound to. This
more closely replicates the level of functionality that tests will actually
use.

Reviewed-by: Eric Blake <eblake@redhat.com>
Reviewed-by: Marc-André Lureau <marcandre.lureau@redhat.com>
Signed-off-by: Daniel P. Berrange <berrange@redhat.com>
This commit is contained in:
Daniel P. Berrange 2017-12-22 11:48:49 +00:00 committed by Daniel P. Berrangé
parent 9b589ffb12
commit abd983c0e0
2 changed files with 60 additions and 11 deletions

View File

@ -30,10 +30,15 @@
# define EAI_ADDRFAMILY 0 # define EAI_ADDRFAMILY 0
#endif #endif
int socket_can_bind(const char *hostname) int socket_can_bind_connect(const char *hostname)
{ {
int fd = -1; int lfd = -1, cfd = -1, afd = -1;
struct addrinfo ai, *res = NULL; struct addrinfo ai, *res = NULL;
struct sockaddr_storage ss;
socklen_t sslen = sizeof(ss);
int soerr;
socklen_t soerrlen = sizeof(soerr);
bool check_soerr = false;
int rc; int rc;
int ret = -1; int ret = -1;
@ -54,20 +59,64 @@ int socket_can_bind(const char *hostname)
goto cleanup; goto cleanup;
} }
fd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol); lfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol);
if (fd < 0) { if (lfd < 0) {
goto cleanup; goto cleanup;
} }
if (bind(fd, res->ai_addr, res->ai_addrlen) < 0) { cfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol);
if (cfd < 0) {
goto cleanup; goto cleanup;
} }
if (bind(lfd, res->ai_addr, res->ai_addrlen) < 0) {
goto cleanup;
}
if (listen(lfd, 1) < 0) {
goto cleanup;
}
if (getsockname(lfd, (struct sockaddr *)&ss, &sslen) < 0) {
goto cleanup;
}
qemu_set_nonblock(cfd);
if (connect(cfd, (struct sockaddr *)&ss, sslen) < 0) {
if (errno == EINPROGRESS) {
check_soerr = true;
} else {
goto cleanup;
}
}
sslen = sizeof(ss);
afd = accept(lfd, (struct sockaddr *)&ss, &sslen);
if (afd < 0) {
goto cleanup;
}
if (check_soerr) {
if (qemu_getsockopt(cfd, SOL_SOCKET, SO_ERROR, &soerr, &soerrlen) < 0) {
goto cleanup;
}
if (soerr) {
errno = soerr;
goto cleanup;
}
}
ret = 0; ret = 0;
cleanup: cleanup:
if (fd != -1) { if (afd != -1) {
close(fd); close(afd);
}
if (cfd != -1) {
close(cfd);
}
if (lfd != -1) {
close(lfd);
} }
if (res) { if (res) {
freeaddrinfo(res); freeaddrinfo(res);
@ -80,7 +129,7 @@ int socket_check_protocol_support(bool *has_ipv4, bool *has_ipv6)
{ {
*has_ipv4 = *has_ipv6 = false; *has_ipv4 = *has_ipv6 = false;
if (socket_can_bind("127.0.0.1") < 0) { if (socket_can_bind_connect("127.0.0.1") < 0) {
if (errno != EADDRNOTAVAIL) { if (errno != EADDRNOTAVAIL) {
return -1; return -1;
} }
@ -88,7 +137,7 @@ int socket_check_protocol_support(bool *has_ipv4, bool *has_ipv6)
*has_ipv4 = true; *has_ipv4 = true;
} }
if (socket_can_bind("::1") < 0) { if (socket_can_bind_connect("::1") < 0) {
if (errno != EADDRNOTAVAIL) { if (errno != EADDRNOTAVAIL) {
return -1; return -1;
} }

View File

@ -21,13 +21,13 @@
/* /*
* @hostname: a DNS name or numeric IP address * @hostname: a DNS name or numeric IP address
* *
* Check whether it is possible to bind to ports * Check whether it is possible to bind & connect to ports
* on the DNS name or IP address @hostname. If an IP address * on the DNS name or IP address @hostname. If an IP address
* is used, it must not be a wildcard address. * is used, it must not be a wildcard address.
* *
* Returns 0 on success, -1 on error with errno set * Returns 0 on success, -1 on error with errno set
*/ */
int socket_can_bind(const char *hostname); int socket_can_bind_connect(const char *hostname);
/* /*
* @has_ipv4: set to true on return if IPv4 is available * @has_ipv4: set to true on return if IPv4 is available