diff options
author | Peter Maydell <peter.maydell@linaro.org> | 2016-03-15 15:51:06 +0000 |
---|---|---|
committer | Peter Maydell <peter.maydell@linaro.org> | 2016-03-15 15:51:06 +0000 |
commit | f84d587111cc165f0aa6aa49e434fb3afc897c2d (patch) | |
tree | 5f76f4ceb397565cc2333595d091cef0ec0611d0 /tests | |
parent | d41e0bed7ba7bad87b4e0cedf5580f29ee1a311f (diff) | |
parent | cfd47a71df51047833d182e9e97244e7816b57da (diff) |
Merge remote-tracking branch 'remotes/berrange/tags/pull-io-next-2016-03-15-1' into staging
Merge I/O fixes
# gpg: Signature made Tue 15 Mar 2016 14:42:43 GMT using RSA key ID 15104FDF
# gpg: Good signature from "Daniel P. Berrange <dan@berrange.com>"
# gpg: aka "Daniel P. Berrange <berrange@redhat.com>"
* remotes/berrange/tags/pull-io-next-2016-03-15-1:
io: stronger check for support for IPv4/6
Signed-off-by: Peter Maydell <peter.maydell@linaro.org>
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test-io-channel-socket.c | 70 |
1 files changed, 49 insertions, 21 deletions
diff --git a/tests/test-io-channel-socket.c b/tests/test-io-channel-socket.c index ae665f5ea0..87018acb8a 100644 --- a/tests/test-io-channel-socket.c +++ b/tests/test-io-channel-socket.c @@ -23,44 +23,72 @@ #include "io/channel-util.h" #include "io-channel-helpers.h" -static int check_bind(struct sockaddr *sa, socklen_t salen, bool *has_proto) +#ifndef AI_ADDRCONFIG +# define AI_ADDRCONFIG 0 +#endif +#ifndef AI_V4MAPPED +# define AI_V4MAPPED 0 +#endif +#ifndef EAI_ADDRFAMILY +# define EAI_ADDRFAMILY 0 +#endif + +static int check_bind(const char *hostname, bool *has_proto) { - int fd; + int fd = -1; + struct addrinfo ai, *res = NULL; + int rc; + int ret = -1; + + memset(&ai, 0, sizeof(ai)); + ai.ai_flags = AI_CANONNAME | AI_V4MAPPED | AI_ADDRCONFIG; + ai.ai_family = AF_UNSPEC; + ai.ai_socktype = SOCK_STREAM; + + /* lookup */ + rc = getaddrinfo(hostname, NULL, &ai, &res); + if (rc != 0) { + if (rc == EAI_ADDRFAMILY || + rc == EAI_FAMILY) { + *has_proto = false; + goto done; + } + goto cleanup; + } - fd = socket(sa->sa_family, SOCK_STREAM, 0); + fd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol); if (fd < 0) { - return -1; + goto cleanup; } - if (bind(fd, sa, salen) < 0) { - close(fd); + if (bind(fd, res->ai_addr, res->ai_addrlen) < 0) { if (errno == EADDRNOTAVAIL) { *has_proto = false; - return 0; + goto done; } - return -1; + goto cleanup; } - close(fd); *has_proto = true; - return 0; + done: + ret = 0; + + cleanup: + if (fd != -1) { + close(fd); + } + if (res) { + freeaddrinfo(res); + } + return ret; } static int check_protocol_support(bool *has_ipv4, bool *has_ipv6) { - struct sockaddr_in sin = { - .sin_family = AF_INET, - .sin_addr = { .s_addr = htonl(INADDR_LOOPBACK) }, - }; - struct sockaddr_in6 sin6 = { - .sin6_family = AF_INET6, - .sin6_addr = IN6ADDR_LOOPBACK_INIT, - }; - - if (check_bind((struct sockaddr *)&sin, sizeof(sin), has_ipv4) < 0) { + if (check_bind("127.0.0.1", has_ipv4) < 0) { return -1; } - if (check_bind((struct sockaddr *)&sin6, sizeof(sin6), has_ipv6) < 0) { + if (check_bind("::1", has_ipv6) < 0) { return -1; } |