reactos/modules/rostests/apitests/iphlpapi/GetExtendedUdpTable.c

176 lines
4.6 KiB
C
Raw Normal View History

/*
* 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();
}