From 6bdf3a6ba7c169e5d2e21f7d79bc798fb6e28c28 Mon Sep 17 00:00:00 2001
From: Ziqing Hui <zhui@codeweavers.com>
Date: Mon, 21 Mar 2022 17:05:54 +0800
Subject: [PATCH] ws2_32/tests: Test getaddrinfo() for IPv6 DNS querying.

The tests show that if we don't have a global IPv6 address, getaddrinfo()
will not return IPv6 results. Specifing AF_UNSPEC makes it return IPv4 results.

Signed-off-by: Ziqing Hui <zhui@codeweavers.com>
Signed-off-by: Huw Davies <huw@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
---
 dlls/ws2_32/tests/protocol.c | 65 ++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/dlls/ws2_32/tests/protocol.c b/dlls/ws2_32/tests/protocol.c
index 5f728a9c4a4..9a6c8610718 100644
--- a/dlls/ws2_32/tests/protocol.c
+++ b/dlls/ws2_32/tests/protocol.c
@@ -2122,6 +2122,17 @@ static void compare_addrinfo(ADDRINFO *a, ADDRINFO *b)
     ok(!a && !b, "Expected both addresses null (%p != %p)\n", a, b);
 }
 
+static BOOL ipv6_found(ADDRINFOA *addr)
+{
+    ADDRINFOA *p;
+    for (p = addr; p; p = p->ai_next)
+    {
+        if (p->ai_family == AF_INET6)
+            return TRUE;
+    }
+    return FALSE;
+}
+
 static void test_getaddrinfo(void)
 {
     int i, ret;
@@ -2129,6 +2140,7 @@ static void test_getaddrinfo(void)
     SOCKADDR_IN *sockaddr;
     CHAR name[256], *ip;
     DWORD size = sizeof(name);
+    BOOL has_ipv6_addr;
 
     memset(&hint, 0, sizeof(ADDRINFOA));
     GetComputerNameExA( ComputerNamePhysicalDnsHostname, name, &size );
@@ -2396,6 +2408,59 @@ static void test_getaddrinfo(void)
     ok(sockaddr->sin_family == AF_INET, "ai_addr->sin_family == %d\n", sockaddr->sin_family);
     ok(sockaddr->sin_port == 0, "ai_addr->sin_port == %d\n", sockaddr->sin_port);
     freeaddrinfo(result);
+
+    /* Check whether we have global IPv6 address */
+    result = NULL;
+    ret = getaddrinfo("", NULL, NULL, &result);
+    ok(!ret, "getaddrinfo failed with %d\n", WSAGetLastError());
+    has_ipv6_addr = FALSE;
+    for (p = result; p; p = p->ai_next)
+    {
+        if (p->ai_family == AF_INET6)
+        {
+            IN6_ADDR *a = &((SOCKADDR_IN6 *)p->ai_addr)->sin6_addr;
+            if (!IN6_IS_ADDR_LINKLOCAL(a) && !IN6_IS_ADDR_LOOPBACK(a) && !IN6_IS_ADDR_UNSPECIFIED(a))
+            {
+                has_ipv6_addr = TRUE;
+                break;
+            }
+        }
+    }
+    freeaddrinfo(result);
+
+    result = NULL;
+    ret = getaddrinfo("www.kernel.org", NULL, NULL, &result);
+    ok(!ret, "getaddrinfo failed with %d\n", WSAGetLastError());
+    if (!has_ipv6_addr)
+        todo_wine ok(!ipv6_found(result), "IPv6 address is returned.\n");
+    freeaddrinfo(result);
+
+    for (i = 0; i < ARRAY_SIZE(hinttests); i++)
+    {
+        if (hinttests[i].family != AF_UNSPEC || hinttests[i].error) continue;
+        winetest_push_context("Test %u", i);
+
+        hint.ai_flags = 0;
+        hint.ai_family = hinttests[i].family;
+        hint.ai_socktype = hinttests[i].socktype;
+        hint.ai_protocol = hinttests[i].protocol;
+
+        result = NULL;
+        ret = getaddrinfo("www.kernel.org", NULL, &hint, &result);
+        ok(!ret, "Got unexpected ret %d\n", ret);
+        if (!has_ipv6_addr)
+            todo_wine ok(!ipv6_found(result), "IPv6 address is returned.\n");
+        freeaddrinfo(result);
+
+        hint.ai_family = AF_INET6;
+        result = NULL;
+        ret = getaddrinfo("www.kernel.org", NULL, &hint, &result);
+        if (!has_ipv6_addr)
+            todo_wine ok(ret == WSANO_DATA, "Got unexpected ret %d\n", ret);
+        freeaddrinfo(result);
+
+        winetest_pop_context();
+    }
 }
 
 static void test_dns(void)