[tor-commits] [torsocks/master] gethostbyaddr_r: Populate h_addrtype field

[dgoulet]

commit a44660bd6d32de6c58a51eaad7c019f8e9131cf3
Author: David Goulet <dgou...@torproject.org>
Date:   Fri Apr 20 11:58:12 2018 -0400

    gethostbyaddr_r: Populate h_addrtype field
    
    Fixes #25586
    
    Signed-off-by: David Goulet <dgou...@torproject.org>
---
 src/lib/gethostbyname.c |  1 +
 tests/test_dns.c        | 51 +++++++++++++++++++++++++++++++++++++++----------
 2 files changed, 42 insertions(+), 10 deletions(-)

diff --git a/src/lib/gethostbyname.c b/src/lib/gethostbyname.c
index e43ed4a..296099e 100644
--- a/src/lib/gethostbyname.c
+++ b/src/lib/gethostbyname.c
@@ -301,6 +301,7 @@ LIBC_GETHOSTBYADDR_R_RET_TYPE 
tsocks_gethostbyaddr_r(LIBC_GETHOSTBYADDR_R_SIG)
 
        he->h_aliases = NULL;
        he->h_length = strlen(he->h_name);
+       he->h_addrtype = type;
        /* Assign the address list within the data of the given buffer. */
        data->addr_list[0] = (char *) addr;
        data->addr_list[1] = NULL;
diff --git a/tests/test_dns.c b/tests/test_dns.c
index 2d87744..06fad12 100644
--- a/tests/test_dns.c
+++ b/tests/test_dns.c
@@ -26,7 +26,7 @@
 #include <tap/tap.h>
 #include "helpers.h"
 
-#define NUM_TESTS 4
+#define NUM_TESTS 5
 
 struct test_host {
        const char *name;
@@ -76,26 +76,56 @@ static void test_gethostbyname(const struct test_host *host)
        return;
 }
 
+static void test_gethostbyaddr_r(const struct test_host *host)
+{
+  int result;
+  in_addr_t addr;
+  struct hostent ret;
+  char buf[1024];
+  int buflen = sizeof buf;
+  struct hostent *result_entp;
+  int h_errno;
+
+  assert(host);
+  diag("gethostbyaddr_r test");
+
+  addr = inet_addr(host->ip);
+       result = gethostbyaddr_r((const void *)&addr,
+                               INET_ADDRSTRLEN, AF_INET, &ret, buf, buflen, 
&result_entp, &h_errno);
+
+  if (result) {
+    fail("Resolving address %s: %d", host->ip, result);
+  }
+
+  if (strcmp(host->name, result_entp->h_name) != 0) {
+    fail("Wrong resolved name: %s", result_entp->h_name);
+  }
+
+  if (result_entp->h_addrtype != AF_INET) {
+    fail("Wrong resolved address family: %d", result_entp->h_addrtype);
+  }
+
+  ok(1, "Resolved address");
+}
+
 static void test_gethostbyaddr(const struct test_host *host)
 {
        struct hostent *he;
-    in_addr_t addr;
+  in_addr_t addr;
 
        assert(host);
 
        diag("gethostbyaddr test");
 
-    addr = inet_addr(host->ip);
-
-    he = gethostbyaddr((const void *)&addr, INET_ADDRSTRLEN, AF_INET);
-    if (he) {
-               ok(strcmp(host->name, he->h_name) == 0,
-                               "Resolving address %s", host->ip);
-    } else {
+       addr = inet_addr(host->ip);
+       he = gethostbyaddr((const void *)&addr, INET_ADDRSTRLEN, AF_INET);
+       if (he) {
+               ok(strcmp(host->name, he->h_name) == 0, "Resolving address %s", 
host->ip);
+       } else {
                fail("Resolving address %s", host->ip);
        }
 
-    return;
+       return;
 }
 
 static void test_getaddrinfo(const struct test_host *host)
@@ -148,6 +178,7 @@ int main(int argc, char **argv)
        test_getaddrinfo(&tor_check);
        test_gethostbyname(&tor_dir_auth1);
        test_gethostbyaddr(&tor_dir_auth2);
+       test_gethostbyaddr_r(&tor_dir_auth2);
        test_getaddrinfo(&tor_localhost);
 
 end:



_______________________________________________
tor-commits mailing list
tor-commits@lists.torproject.org
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits