[PATCH 2/5] iphlpapi/tests: Confirm that GetExtendedUdpTable associates a socket with the correct PID.

[Tim Clem]

From: Tim Clem

---
 dlls/iphlpapi/tests/iphlpapi.c | 92 ++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)

diff --git a/dlls/iphlpapi/tests/iphlpapi.c b/dlls/iphlpapi/tests/iphlpapi.c
index 1248dca59e7..da3a06d54ca 100644
--- a/dlls/iphlpapi/tests/iphlpapi.c
+++ b/dlls/iphlpapi/tests/iphlpapi.c
@@ -2113,6 +2113,96 @@ static void test_GetExtendedUdpTable(void)
     free( table_module );
 }

+/* Test that the UDP_TABLE_OWNER_PID table contains an entry for a socket we
+   make, and associates it with our process. */
+static void test_GetExtendedUdpTable_owner( int family )
+{
+    SOCKET sock;
+    int port;
+    DWORD i, ret;
+    void *raw_table = NULL;
+
+    winetest_push_context( "%s", family == AF_INET ? "AF_INET" : "AF_INET6" );
+
+    sock = socket( family, SOCK_DGRAM, IPPROTO_UDP );
+    ok( sock != INVALID_SOCKET, "socket error %d\n", WSAGetLastError() );
+
+    if (family == AF_INET)
+    {
+        struct sockaddr_in addr = { 0 };
+        int addr_len = sizeof(addr);
+
+        addr.sin_family = AF_INET;
+        addr.sin_addr.s_addr = htonl( INADDR_LOOPBACK );
+        addr.sin_port = 0;
+
+        ok( bind( sock, (struct sockaddr *)&addr, addr_len ) == 0, "bind error %d\n", WSAGetLastError() );
+        ok( getsockname( sock, (struct sockaddr *)&addr, &addr_len ) == 0, "getsockname error %d\n", WSAGetLastError() );
+
+        port = addr.sin_port;
+    }
+    else
+    {
+        struct sockaddr_in6 addr = { 0 };
+        int addr_len = sizeof(addr);
+
+        addr.sin6_family = AF_INET6;
+        addr.sin6_addr = in6addr_loopback;
+        addr.sin6_port = 0;
+
+        ok( bind( sock, (struct sockaddr *)&addr, addr_len ) == 0, "bind error %d\n", WSAGetLastError() );
+        ok( getsockname( sock, (struct sockaddr *)&addr, &addr_len ) == 0, "getsockname error %d\n", WSAGetLastError() );
+
+        port = addr.sin6_port;
+    }
+
+    ret = get_extended_udp_table( family, UDP_TABLE_OWNER_PID, &raw_table );
+    if (ret != ERROR_SUCCESS)
+    {
+        skip( "error %lu getting UDP table\n", ret );
+        goto done;
+    }
+
+    if (family == AF_INET)
+    {
+        MIB_UDPTABLE_OWNER_PID *table = raw_table;
+        BOOL found_it = FALSE;
+        for (i = 0; i < table->dwNumEntries; i++)
+        {
+            MIB_UDPROW_OWNER_PID *row = &table->table[i];
+            if (row->dwLocalPort == port && row->dwLocalAddr == htonl( INADDR_LOOPBACK ))
+            {
+                ok( row->dwOwningPid == GetCurrentProcessId(), "unexpected socket owner %04lx\n", row->dwOwningPid );
+                found_it = TRUE;
+                break;
+            }
+        }
+        ok( found_it, "no table entry for socket\n" );
+    }
+    else
+    {
+        MIB_UDP6TABLE_OWNER_PID *table = raw_table;
+        BOOL found_it = FALSE;
+        for (i = 0; i < table->dwNumEntries; i++)
+        {
+            MIB_UDP6ROW_OWNER_PID *row = &table->table[i];
+            if (row->dwLocalPort == port && IN6_IS_ADDR_LOOPBACK( (IN6_ADDR*)&row->ucLocalAddr ))
+            {
+                ok( row->dwOwningPid == GetCurrentProcessId(), "unexpected socket owner %04lx\n", row->dwOwningPid );
+                found_it = TRUE;
+                break;
+            }
+        }
+        ok( found_it, "no table entry for socket\n" );
+    }
+
+done:
+    closesocket( sock );
+    free( raw_table );
+
+    winetest_pop_context();
+}
+
 static void test_CreateSortedAddressPairs(void)
 {
     SOCKADDR_IN6 dst[2];
@@ -3031,6 +3121,8 @@ START_TEST(iphlpapi)
     test_GetExtendedTcpTable_owner(AF_INET);
     test_GetExtendedTcpTable_owner(AF_INET6);
     test_GetExtendedUdpTable();
+    test_GetExtendedUdpTable_owner(AF_INET);
+    test_GetExtendedUdpTable_owner(AF_INET6);
     test_AllocateAndGetTcpExTableFromStack();
     test_CreateSortedAddressPairs();
     test_interface_identifier_conversion();