mirror of
https://github.com/reactos/reactos.git
synced 2025-07-31 14:51:44 +00:00
[IPHLPAPI_APITEST] Add tests for AllocateAndGetTcpExTable2FromStack()
This commit is contained in:
Pierre Schweitzer
parent
7ce2f09a70
commit
ec605e9138
3 changed files with 359 additions and 5 deletions
|
|
@ -0,0 +1,351 @@
|
|||
/*
|
||||
* PROJECT: ReactOS API Tests
|
||||
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
|
||||
* PURPOSE: Tests for TCP connections enumeration functions
|
||||
* COPYRIGHT: Copyright 2018 Pierre Schweitzer
|
||||
*/
|
||||
|
||||
#include <apitest.h>
|
||||
|
||||
#define WIN32_NO_STATUS
|
||||
#include <iphlpapi.h>
|
||||
#include <winsock2.h>
|
||||
|
||||
static DWORD (WINAPI * pAllocateAndGetTcpExTable2FromStack)(PVOID*,BOOL,HANDLE,DWORD,DWORD,TCP_TABLE_CLASS);
|
||||
|
||||
START_TEST(AllocateAndGetTcpExTable2FromStack)
|
||||
{
|
||||
WSADATA wsaData;
|
||||
SOCKET sock;
|
||||
SOCKADDR_IN server;
|
||||
PMIB_TCPTABLE TcpTable;
|
||||
PMIB_TCPTABLE_OWNER_PID TcpTableOwner;
|
||||
PMIB_TCPTABLE_OWNER_MODULE TcpTableOwnerMod;
|
||||
DWORD i;
|
||||
BOOLEAN Found;
|
||||
HINSTANCE hIpHlpApi;
|
||||
SYSTEMTIME Creation;
|
||||
DWORD Pid = GetCurrentProcessId();
|
||||
|
||||
hIpHlpApi = GetModuleHandleW(L"iphlpapi.dll");
|
||||
if (!hIpHlpApi)
|
||||
{
|
||||
skip("Failed to load iphlpapi.dll\n");
|
||||
return;
|
||||
}
|
||||
|
||||
pAllocateAndGetTcpExTable2FromStack = (void *)GetProcAddress(hIpHlpApi, "AllocateAndGetTcpExTable2FromStack");
|
||||
if (pAllocateAndGetTcpExTable2FromStack == NULL)
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack not found\n");
|
||||
return;
|
||||
}
|
||||
|
||||
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
|
||||
{
|
||||
skip("Failed to init WS2\n");
|
||||
return;
|
||||
}
|
||||
|
||||
GetSystemTime(&Creation);
|
||||
|
||||
sock = socket(AF_INET, SOCK_STREAM, 0);
|
||||
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 (listen(sock, SOMAXCONN) == SOCKET_ERROR)
|
||||
{
|
||||
skip("Cannot listen on socket\n");
|
||||
goto quit2;
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTable, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_BASIC_ALL) == ERROR_SUCCESS)
|
||||
{
|
||||
ok(TcpTable->dwNumEntries > 0, "No TCP connections?!\n");
|
||||
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTable->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTable->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTable->table[i].dwLocalAddr == 0 &&
|
||||
TcpTable->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTable->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
ok(Found, "Our socket wasn't found!\n");
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTable);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTable, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_BASIC_CONNECTIONS) == ERROR_SUCCESS)
|
||||
{
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTable->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTable->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTable->table[i].dwLocalAddr == 0 &&
|
||||
TcpTable->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTable->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
ok(Found == FALSE, "Our socket was found!\n");
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTable);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTable, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_BASIC_LISTENER) == ERROR_SUCCESS)
|
||||
{
|
||||
ok(TcpTable->dwNumEntries > 0, "No TCP connections?!\n");
|
||||
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTable->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTable->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTable->table[i].dwLocalAddr == 0 &&
|
||||
TcpTable->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTable->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
ok(Found, "Our socket wasn't found!\n");
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTable);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwner, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_PID_ALL) == ERROR_SUCCESS)
|
||||
{
|
||||
ok(TcpTableOwner->dwNumEntries > 0, "No TCP connections?!\n");
|
||||
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTableOwner->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTableOwner->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTableOwner->table[i].dwLocalAddr == 0 &&
|
||||
TcpTableOwner->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTableOwner->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!Found)
|
||||
{
|
||||
skip("Our socket wasn't found!\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(TcpTableOwner->table[i].dwOwningPid == Pid, "Invalid owner\n");
|
||||
}
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTableOwner);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwner, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_PID_CONNECTIONS) == ERROR_SUCCESS)
|
||||
{
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTableOwner->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTableOwner->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTableOwner->table[i].dwLocalAddr == 0 &&
|
||||
TcpTableOwner->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTableOwner->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
ok(Found == FALSE, "Our socket was found!\n");
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTableOwner);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwner, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_PID_LISTENER) == ERROR_SUCCESS)
|
||||
{
|
||||
ok(TcpTableOwner->dwNumEntries > 0, "No TCP connections?!\n");
|
||||
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTableOwner->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTableOwner->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTableOwner->table[i].dwLocalAddr == 0 &&
|
||||
TcpTableOwner->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTableOwner->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!Found)
|
||||
{
|
||||
skip("Our socket wasn't found!\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(TcpTableOwner->table[i].dwOwningPid == Pid, "Invalid owner\n");
|
||||
}
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTableOwner);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwnerMod, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_MODULE_ALL) == ERROR_SUCCESS)
|
||||
{
|
||||
ok(TcpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n");
|
||||
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTableOwnerMod->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTableOwnerMod->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTableOwnerMod->table[i].dwLocalAddr == 0 &&
|
||||
TcpTableOwnerMod->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTableOwnerMod->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!Found)
|
||||
{
|
||||
skip("Our socket wasn't found!\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
SYSTEMTIME SockTime;
|
||||
|
||||
ok(TcpTableOwnerMod->table[i].dwOwningPid == Pid, "Invalid owner\n");
|
||||
|
||||
CopyMemory(&SockTime, &TcpTableOwnerMod->table[i].liCreateTimestamp, sizeof(SYSTEMTIME));
|
||||
ok(Creation.wYear == SockTime.wYear, "Invalid year\n");
|
||||
ok(Creation.wMonth == SockTime.wMonth, "Invalid month\n");
|
||||
ok(Creation.wDayOfWeek == SockTime.wDayOfWeek, "Invalid day of week\n");
|
||||
ok(Creation.wDay == SockTime.wDay, "Invalid day\n");
|
||||
ok(Creation.wHour == SockTime.wHour, "Invalid hour\n");
|
||||
ok(Creation.wMinute == SockTime.wMinute, "Invalid minute\n");
|
||||
}
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTableOwnerMod);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwnerMod, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_MODULE_CONNECTIONS) == ERROR_SUCCESS)
|
||||
{
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTableOwnerMod->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTableOwnerMod->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTableOwnerMod->table[i].dwLocalAddr == 0 &&
|
||||
TcpTableOwnerMod->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTableOwnerMod->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
ok(Found == FALSE, "Our socket was found!\n");
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTableOwnerMod);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwnerMod, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_MODULE_LISTENER) == ERROR_SUCCESS)
|
||||
{
|
||||
ok(TcpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n");
|
||||
|
||||
Found = FALSE;
|
||||
for (i = 0; i < TcpTableOwnerMod->dwNumEntries; ++i)
|
||||
{
|
||||
if (TcpTableOwnerMod->table[i].dwState == MIB_TCP_STATE_LISTEN &&
|
||||
TcpTableOwnerMod->table[i].dwLocalAddr == 0 &&
|
||||
TcpTableOwnerMod->table[i].dwLocalPort == htons(9876) &&
|
||||
TcpTableOwnerMod->table[i].dwRemoteAddr == 0)
|
||||
{
|
||||
Found = TRUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!Found)
|
||||
{
|
||||
skip("Our socket wasn't found!\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
SYSTEMTIME SockTime;
|
||||
|
||||
ok(TcpTableOwnerMod->table[i].dwOwningPid == Pid, "Invalid owner\n");
|
||||
|
||||
CopyMemory(&SockTime, &TcpTableOwnerMod->table[i].liCreateTimestamp, sizeof(SYSTEMTIME));
|
||||
ok(Creation.wYear == SockTime.wYear, "Invalid year\n");
|
||||
ok(Creation.wMonth == SockTime.wMonth, "Invalid month\n");
|
||||
ok(Creation.wDayOfWeek == SockTime.wDayOfWeek, "Invalid day of week\n");
|
||||
ok(Creation.wDay == SockTime.wDay, "Invalid day\n");
|
||||
ok(Creation.wHour == SockTime.wHour, "Invalid hour\n");
|
||||
ok(Creation.wMinute == SockTime.wMinute, "Invalid minute\n");
|
||||
}
|
||||
|
||||
HeapFree(GetProcessHeap(), 0, TcpTableOwnerMod);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("AllocateAndGetTcpExTable2FromStack failure\n");
|
||||
}
|
||||
|
||||
quit2:
|
||||
closesocket(sock);
|
||||
quit:
|
||||
WSACleanup();
|
||||
}
|
||||
|
|
@ -1,5 +1,6 @@
|
|||
|
||||
list(APPEND SOURCE
|
||||
AllocateAndGetTcpExTable2FromStack.c
|
||||
GetInterfaceName.c
|
||||
GetNetworkParams.c
|
||||
icmp.c
|
||||
|
|
@ -9,5 +10,5 @@ list(APPEND SOURCE
|
|||
add_executable(iphlpapi_apitest ${SOURCE})
|
||||
target_link_libraries(iphlpapi_apitest wine ${PSEH_LIB})
|
||||
set_module_type(iphlpapi_apitest win32cui)
|
||||
add_importlibs(iphlpapi_apitest iphlpapi advapi32 msvcrt kernel32 ntdll)
|
||||
add_importlibs(iphlpapi_apitest iphlpapi ws2_32 advapi32 msvcrt kernel32 ntdll)
|
||||
add_rostests_file(TARGET iphlpapi_apitest)
|
||||
|
|
|
|||
|
|
@ -3,6 +3,7 @@
|
|||
#define STANDALONE
|
||||
#include <apitest.h>
|
||||
|
||||
extern void func_AllocateAndGetTcpExTable2FromStack(void);
|
||||
extern void func_GetInterfaceName(void);
|
||||
extern void func_GetNetworkParams(void);
|
||||
extern void func_icmp(void);
|
||||
|
|
@ -10,10 +11,11 @@ extern void func_SendARP(void);
|
|||
|
||||
const struct test winetest_testlist[] =
|
||||
{
|
||||
{ "GetInterfaceName", func_GetInterfaceName },
|
||||
{ "GetNetworkParams", func_GetNetworkParams },
|
||||
{ "icmp", func_icmp },
|
||||
{ "SendARP", func_SendARP },
|
||||
{ "AllocateAndGetTcpExTable2FromStack", func_AllocateAndGetTcpExTable2FromStack },
|
||||
{ "GetInterfaceName", func_GetInterfaceName },
|
||||
{ "GetNetworkParams", func_GetNetworkParams },
|
||||
{ "icmp", func_icmp },
|
||||
{ "SendARP", func_SendARP },
|
||||
|
||||
{ 0, 0 }
|
||||
};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue