--- /dev/null
+/*
+ * PROJECT: ReactOS API Tests
+ * LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
+ * PURPOSE: Tests for UDP connections enumeration functions
+ * COPYRIGHT: Copyright 2018 Pierre Schweitzer
+ */
+
+#include <apitest.h>
+
+#define WIN32_NO_STATUS
+#include <iphlpapi.h>
+#include <winsock2.h>
+
+static DWORD GetExtendedUdpTableWithAlloc(PVOID *UdpTable, BOOL Order, DWORD Family, UDP_TABLE_CLASS Class)
+{
+ DWORD ret;
+ DWORD Size = 0;
+
+ *UdpTable = NULL;
+
+ ret = GetExtendedUdpTable(*UdpTable, &Size, Order, Family, Class, 0);
+ if (ret == ERROR_INSUFFICIENT_BUFFER)
+ {
+ *UdpTable = HeapAlloc(GetProcessHeap(), 0, Size);
+ if (*UdpTable == NULL)
+ {
+ return ERROR_OUTOFMEMORY;
+ }
+
+ ret = GetExtendedUdpTable(*UdpTable, &Size, Order, Family, Class, 0);
+ if (ret != NO_ERROR)
+ {
+ HeapFree(GetProcessHeap(), 0, *UdpTable);
+ *UdpTable = NULL;
+ }
+ }
+
+ return ret;
+}
+
+START_TEST(GetExtendedUdpTable)
+{
+ WSADATA wsaData;
+ SOCKET sock;
+ SOCKADDR_IN server;
+ PMIB_UDPTABLE UdpTable;
+ PMIB_UDPTABLE_OWNER_PID UdpTableOwner;
+ PMIB_UDPTABLE_OWNER_MODULE UdpTableOwnerMod;
+ DWORD i;
+ BOOLEAN Found;
+ FILETIME Creation;
+ LARGE_INTEGER CreationTime;
+ DWORD Pid = GetCurrentProcessId();
+
+ if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
+ {
+ skip("Failed to init WS2\n");
+ return;
+ }
+
+ GetSystemTimeAsFileTime(&Creation);
+ CreationTime.LowPart = Creation.dwLowDateTime;
+ CreationTime.HighPart = Creation.dwHighDateTime;
+
+ sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+ if (sock == INVALID_SOCKET)
+ {
+ skip("Cannot create socket\n");
+ goto quit;
+ }
+
+ ZeroMemory(&server, sizeof(SOCKADDR_IN));
+ server.sin_family = AF_INET;
+ server.sin_addr.s_addr = htonl(INADDR_ANY);
+ server.sin_port = htons(9876);
+
+ if (bind(sock, (SOCKADDR*)&server, sizeof(SOCKADDR_IN)) == SOCKET_ERROR)
+ {
+ skip("Cannot bind socket\n");
+ goto quit2;
+ }
+
+ if (GetExtendedUdpTableWithAlloc((PVOID *)&UdpTable, TRUE, AF_INET, UDP_TABLE_BASIC) == ERROR_SUCCESS)
+ {
+ ok(UdpTable->dwNumEntries > 0, "No UDP connections?!\n");
+
+ Found = FALSE;
+ for (i = 0; i < UdpTable->dwNumEntries; ++i)
+ {
+ if (UdpTable->table[i].dwLocalAddr == 0 &&
+ UdpTable->table[i].dwLocalPort == htons(9876))
+ {
+ Found = TRUE;
+ break;
+ }
+ }
+ ok(Found, "Our socket wasn't found!\n");
+
+ HeapFree(GetProcessHeap(), 0, UdpTable);
+ }
+ else
+ {
+ skip("GetExtendedUdpTableWithAlloc failure\n");
+ }
+
+ if (GetExtendedUdpTableWithAlloc((PVOID *)&UdpTableOwner, TRUE, AF_INET, UDP_TABLE_OWNER_PID) == ERROR_SUCCESS)
+ {
+ ok(UdpTableOwner->dwNumEntries > 0, "No UDP connections?!\n");
+
+ Found = FALSE;
+ for (i = 0; i < UdpTableOwner->dwNumEntries; ++i)
+ {
+ if (UdpTableOwner->table[i].dwLocalAddr == 0 &&
+ UdpTableOwner->table[i].dwLocalPort == htons(9876))
+ {
+ Found = TRUE;
+ break;
+ }
+ }
+
+ if (!Found)
+ {
+ skip("Our socket wasn't found!\n");
+ }
+ else
+ {
+ ok(UdpTableOwner->table[i].dwOwningPid == Pid, "Invalid owner\n");
+ }
+
+ HeapFree(GetProcessHeap(), 0, UdpTableOwner);
+ }
+ else
+ {
+ skip("GetExtendedUdpTableWithAlloc failure\n");
+ }
+
+ if (GetExtendedUdpTableWithAlloc((PVOID *)&UdpTableOwnerMod, TRUE, AF_INET, UDP_TABLE_OWNER_MODULE) == ERROR_SUCCESS)
+ {
+ ok(UdpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n");
+
+ Found = FALSE;
+ for (i = 0; i < UdpTableOwnerMod->dwNumEntries; ++i)
+ {
+ if (UdpTableOwnerMod->table[i].dwLocalAddr == 0 &&
+ UdpTableOwnerMod->table[i].dwLocalPort == htons(9876))
+ {
+ Found = TRUE;
+ break;
+ }
+ }
+
+ if (!Found)
+ {
+ skip("Our socket wasn't found!\n");
+ }
+ else
+ {
+ ok(UdpTableOwnerMod->table[i].dwOwningPid == Pid, "Invalid owner\n");
+
+ ok(UdpTableOwnerMod->table[i].liCreateTimestamp.QuadPart >= CreationTime.QuadPart, "Invalid time\n");
+ ok(UdpTableOwnerMod->table[i].liCreateTimestamp.QuadPart <= CreationTime.QuadPart + 60000000000LL, "Invalid time\n");
+ }
+
+ HeapFree(GetProcessHeap(), 0, UdpTableOwnerMod);
+ }
+ else
+ {
+ skip("GetExtendedUdpTableWithAlloc failure\n");
+ }
+
+quit2:
+ closesocket(sock);
+quit:
+ WSACleanup();
+}