diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/CMakeLists.txt | 1 | ||||
| -rw-r--r-- | tests/resolv.c | 164 |
2 files changed, 165 insertions, 0 deletions
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 4dc0bd60..5dcc1379 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -81,6 +81,7 @@ add_nng_test(reqrep 5) add_nng_test(pipeline 5) add_nng_test(pollfd 5) add_nng_test(pubsub 5) +add_nng_test(resolv 10) add_nng_test(sock 5) add_nng_test(survey 5) add_nng_test(tcp 5) diff --git a/tests/resolv.c b/tests/resolv.c new file mode 100644 index 00000000..9c07d35f --- /dev/null +++ b/tests/resolv.c @@ -0,0 +1,164 @@ +// +// Copyright 2016 Garrett D'Amore <garrett@damore.org> +// +// This software is supplied under the terms of the MIT License, a +// copy of which should be located in the distribution where this +// file was obtained (LICENSE.txt). A copy of the license may also be +// found online at https://opensource.org/licenses/MIT. +// + +#include "core/nng_impl.h" +#include "convey.h" +#include "stubs.h" + +#include <string.h> +#include <arpa/inet.h> + +static const char * +ip4str(void *addr) +{ + static char buf[256]; + + return (inet_ntop(AF_INET, addr, buf, sizeof (buf))); +} + + +static const char * +ip6str(void *addr) +{ + static char buf[256]; + + return (inet_ntop(AF_INET6, addr, buf, sizeof (buf))); +} + + +TestMain("TCP Resolver", { + nni_init(); + + Convey("Localhost IPv4 resolves", { + nni_aio aio; + const char *str; + memset(&aio, 0, sizeof (aio)); + nni_aio_init(&aio, NULL, NULL); + nni_plat_tcp_resolv("localhost", "80", NNG_AF_INET, 1, + &aio); + nni_aio_wait(&aio); + So(nni_aio_result(&aio) == 0); + So(aio.a_naddrs == 1); + So(aio.a_addrs[0].s_un.s_in.sa_family == NNG_AF_INET); + So(aio.a_addrs[0].s_un.s_in.sa_port == ntohs(80)); + So(aio.a_addrs[0].s_un.s_in.sa_addr == ntohl(0x7f000001)); + str = ip4str(&aio.a_addrs[0].s_un.s_in.sa_addr); + So(strcmp(str, "127.0.0.1") == 0); + nni_aio_fini(&aio); + } + ); + Convey("Localhost IPv6 resolves", { + nni_aio aio; + memset(&aio, 0, sizeof (aio)); + const char *str; + nni_aio_init(&aio, NULL, NULL); + nni_plat_tcp_resolv("localhost", "80", NNG_AF_INET6, 1, + &aio); + nni_aio_wait(&aio); + So(nni_aio_result(&aio) == 0); + So(aio.a_naddrs == 1); + So(aio.a_addrs[0].s_un.s_in6.sa_family == NNG_AF_INET6); + So(aio.a_addrs[0].s_un.s_in6.sa_port == ntohs(80)); + str = ip6str(&aio.a_addrs[0].s_un.s_in6.sa_addr); + So(strcmp(str, "::1") == 0); + nni_aio_fini(&aio); + } + ); + Convey("Localhost UNSPEC resolves", { + nni_aio aio; + memset(&aio, 0, sizeof (aio)); + const char *str; + int i; + nni_aio_init(&aio, NULL, NULL); + nni_plat_tcp_resolv("localhost", "80", NNG_AF_UNSPEC, 1, + &aio); + nni_aio_wait(&aio); + So(nni_aio_result(&aio) == 0); + So(aio.a_naddrs == 2); + for (i = 0; i < 2; i++) { + switch (aio.a_addrs[i].s_un.s_family) { + case NNG_AF_INET6: + So(aio.a_addrs[i].s_un.s_in6.sa_port == + ntohs(80)); + str = + ip6str(&aio.a_addrs[i].s_un.s_in6.sa_addr); + So(strcmp(str, "::1") == 0); + break; + + case NNG_AF_INET: + So(aio.a_addrs[i].s_un.s_in.sa_port == + ntohs(80)); + str = + ip4str(&aio.a_addrs[i].s_un.s_in.sa_addr); + So(strcmp(str, "127.0.0.1") == 0); + break; + default: + So(1 == 0); + } + } + So(aio.a_addrs[0].s_un.s_family != + aio.a_addrs[1].s_un.s_family); + nni_aio_fini(&aio); + } + ); + Convey("Google DNS IPv4 resolves", { + nni_aio aio; + const char *str; + memset(&aio, 0, sizeof (aio)); + nni_aio_init(&aio, NULL, NULL); + nni_plat_tcp_resolv("google-public-dns-a.google.com", + "80", NNG_AF_INET, 1, &aio); + nni_aio_wait(&aio); + So(nni_aio_result(&aio) == 0); + So(aio.a_naddrs == 1); + So(aio.a_addrs[0].s_un.s_in.sa_family == NNG_AF_INET); + So(aio.a_addrs[0].s_un.s_in.sa_port == ntohs(80)); + str = ip4str(&aio.a_addrs[0].s_un.s_in.sa_addr); + So(strcmp(str, "8.8.8.8") == 0); + nni_aio_fini(&aio); + } + ); + Convey("Numeric resolves", { + nni_aio aio; + const char *str; + memset(&aio, 0, sizeof (aio)); + nni_aio_init(&aio, NULL, NULL); + nni_plat_tcp_resolv("8.8.4.4", + "80", NNG_AF_INET, 1, &aio); + nni_aio_wait(&aio); + So(nni_aio_result(&aio) == 0); + So(aio.a_naddrs == 1); + So(aio.a_addrs[0].s_un.s_in.sa_family == NNG_AF_INET); + So(aio.a_addrs[0].s_un.s_in.sa_port == ntohs(80)); + str = ip4str(&aio.a_addrs[0].s_un.s_in.sa_addr); + So(strcmp(str, "8.8.4.4") == 0); + nni_aio_fini(&aio); + } + ); + Convey("Name service resolves", { + nni_aio aio; + const char *str; + memset(&aio, 0, sizeof (aio)); + nni_aio_init(&aio, NULL, NULL); + nni_plat_tcp_resolv("8.8.4.4", + "http", NNG_AF_INET, 1, &aio); + nni_aio_wait(&aio); + So(nni_aio_result(&aio) == 0); + So(aio.a_naddrs == 1); + So(aio.a_addrs[0].s_un.s_in.sa_family == NNG_AF_INET); + So(aio.a_addrs[0].s_un.s_in.sa_port == ntohs(80)); + str = ip4str(&aio.a_addrs[0].s_un.s_in.sa_addr); + So(strcmp(str, "8.8.4.4") == 0); + nni_aio_fini(&aio); + } + ); + + nni_fini(); + } + ) |