Black box test for get_interface_address6_via_udp_socket_hack().

2 files changed, 64 insertions, 0 deletions

diff --git a/src/common/address.h b/src/common/address.h
index 8c6ee5abbb..6d9d78893f 100644
--- a/src/common/address.h
+++ b/src/common/address.h
@@ -274,6 +274,9 @@ tor_addr_port_t *tor_addr_port_new(const tor_addr_t *addr, uint16_t port);
 
 #ifdef ADDRESS_PRIVATE
 STATIC smartlist_t *get_interface_addresses_raw(int severity);
+STATIC int get_interface_address6_via_udp_socket_hack(int severity,
+                                                      sa_family_t family,
+                                                      tor_addr_t *addr);
 
 #ifdef HAVE_IFADDRS_TO_SMARTLIST
 STATIC smartlist_t *ifaddrs_to_smartlist(const struct ifaddrs *ifa);
diff --git a/src/test/test_address.c b/src/test/test_address.c
index 7f7347fa88..57bd70cf18 100644
--- a/src/test/test_address.c
+++ b/src/test/test_address.c
@@ -452,10 +452,71 @@ test_address_get_if_addrs_ioctl(void *arg)
 
 #endif
 
+/** Return 1 iff an address exactly equal to <b>tor_addr</b> is in
+ * <b>smartlist</b>. Otherwise, return 0.
+ */
+static int
+smartlist_contain_tor_addr(smartlist_t *smartlist, tor_addr_t *tor_addr)
+{
+  int success = 0;
+
+  SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, a) {
+    if (tor_addr_compare(tor_addr,a,CMP_EXACT) == 0) {
+      success = 1;
+      break;
+    }
+  } SMARTLIST_FOREACH_END(a);
+
+  return success;
+}
+
+static void
+test_address_udp_socket_trick_blackbox(void *arg)
+{
+  smartlist_t *all_addrs;
+  tor_addr_t *addr4;
+  tor_addr_t *addr6;
+  int retval;
+
+  (void)arg;
+
+  addr4 = tor_malloc(sizeof(tor_addr_t));
+  addr6 = tor_malloc(sizeof(tor_addr_t));
+
+  all_addrs = get_interface_addresses_raw(LOG_DEBUG);
+  retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
+                                                      AF_INET,addr4);
+
+  tt_assert(all_addrs);
+  tt_assert(retval == 0);
+
+  tt_assert(smartlist_contain_tor_addr(all_addrs,addr4));
+
+#if 0
+  retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
+                                                      AF_INET6,addr6);
+
+  tt_assert(smartlist_contain_tor_addr(all_addrs,addr6));
+#endif
+
+  retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
+                                                      AF_CCITT,addr4);
+
+  tt_assert(retval == -1);
+
+  done:
+  tor_free(addr4);
+  tor_free(addr6);
+  SMARTLIST_FOREACH(all_addrs, tor_addr_t *, t, tor_free(t));
+  smartlist_free(all_addrs);
+  return;
+}
+
 #define ADDRESS_TEST(name, flags) \
   { #name, test_address_ ## name, flags, NULL, NULL }
 
 struct testcase_t address_tests[] = {
+  ADDRESS_TEST(udp_socket_trick_blackbox, TT_FORK),
 #ifdef HAVE_IFADDRS_TO_SMARTLIST
   ADDRESS_TEST(get_if_addrs_ifaddrs, TT_FORK),
   ADDRESS_TEST(ifaddrs_to_smartlist, 0),