Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-04 00:45:49 +00:00
Code Issues Projects Releases Packages Wiki Activity

[IPHLPAPI_APITEST] Test GetExtendedTcpTable rather than AllocateAndGetTcpExTable2FromStack

Browse source
This commit is contained in:
Pierre Schweitzer 2018-11-29 08:42:13 +01:00
parent 60203092d7
commit 1e9fb35fbc
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
3 changed files with 52 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/rostests/apitests/iphlpapi/CMakeLists.txt
View file

@ -1,6 +1,6 @@
list(APPEND SOURCE list(APPEND SOURCE
AllocateAndGetTcpExTable2FromStack.c GetExtendedTcpTable.c
GetInterfaceName.c GetInterfaceName.c
GetNetworkParams.c GetNetworkParams.c
icmp.c icmp.c

80
modules/rostests/apitests/iphlpapi/AllocateAndGetTcpExTable2FromStack.c → modules/rostests/apitests/iphlpapi/GetExtendedTcpTable.c
View file

@ -11,9 +11,34 @@
#include <iphlpapi.h> #include <iphlpapi.h>
#include <winsock2.h> #include <winsock2.h>
static DWORD (WINAPI * pAllocateAndGetTcpExTable2FromStack)(PVOID*,BOOL,HANDLE,DWORD,DWORD,TCP_TABLE_CLASS); DWORD GetExtendedTcpTableWithAlloc(PVOID *TcpTable, BOOL Order, DWORD Family, TCP_TABLE_CLASS Class)
{
DWORD ret;
DWORD Size = 0;
START_TEST(AllocateAndGetTcpExTable2FromStack) *TcpTable = NULL;
ret = GetExtendedTcpTable(*TcpTable, &Size, Order, Family, Class, 0);
if (ret == ERROR_INSUFFICIENT_BUFFER)
{
*TcpTable = HeapAlloc(GetProcessHeap(), 0, Size);
if (*TcpTable == NULL)
{
return ERROR_OUTOFMEMORY;
}
ret = GetExtendedTcpTable(*TcpTable, &Size, Order, Family, Class, 0);
if (ret != NO_ERROR)
{
HeapFree(GetProcessHeap(), 0, *TcpTable);
*TcpTable = NULL;
}
}
return ret;
}
START_TEST(GetExtendedTcpTable)
{ {
WSADATA wsaData; WSADATA wsaData;
SOCKET sock; SOCKET sock;
@ -23,25 +48,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
PMIB_TCPTABLE_OWNER_MODULE TcpTableOwnerMod; PMIB_TCPTABLE_OWNER_MODULE TcpTableOwnerMod;
DWORD i; DWORD i;
BOOLEAN Found; BOOLEAN Found;
HINSTANCE hIpHlpApi;
FILETIME Creation; FILETIME Creation;
LARGE_INTEGER CreationTime; LARGE_INTEGER CreationTime;
DWORD Pid = GetCurrentProcessId(); 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) if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
{ {
skip("Failed to init WS2\n"); skip("Failed to init WS2\n");
@ -76,7 +86,7 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
goto quit2; goto quit2;
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTable, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_BASIC_ALL) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTable, TRUE, AF_INET, TCP_TABLE_BASIC_ALL) == ERROR_SUCCESS)
{ {
ok(TcpTable->dwNumEntries > 0, "No TCP connections?!\n"); ok(TcpTable->dwNumEntries > 0, "No TCP connections?!\n");
@ -98,10 +108,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTable, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_BASIC_CONNECTIONS) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTable, TRUE, AF_INET, TCP_TABLE_BASIC_CONNECTIONS) == ERROR_SUCCESS)
{ {
Found = FALSE; Found = FALSE;
for (i = 0; i < TcpTable->dwNumEntries; ++i) for (i = 0; i < TcpTable->dwNumEntries; ++i)
@ -121,10 +131,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTable, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_BASIC_LISTENER) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTable, TRUE, AF_INET, TCP_TABLE_BASIC_LISTENER) == ERROR_SUCCESS)
{ {
ok(TcpTable->dwNumEntries > 0, "No TCP connections?!\n"); ok(TcpTable->dwNumEntries > 0, "No TCP connections?!\n");
@ -146,10 +156,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwner, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_PID_ALL) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTableOwner, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL) == ERROR_SUCCESS)
{ {
ok(TcpTableOwner->dwNumEntries > 0, "No TCP connections?!\n"); ok(TcpTableOwner->dwNumEntries > 0, "No TCP connections?!\n");
@ -179,10 +189,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwner, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_PID_CONNECTIONS) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTableOwner, TRUE, AF_INET, TCP_TABLE_OWNER_PID_CONNECTIONS) == ERROR_SUCCESS)
{ {
Found = FALSE; Found = FALSE;
for (i = 0; i < TcpTableOwner->dwNumEntries; ++i) for (i = 0; i < TcpTableOwner->dwNumEntries; ++i)
@ -202,10 +212,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwner, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_PID_LISTENER) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTableOwner, TRUE, AF_INET, TCP_TABLE_OWNER_PID_LISTENER) == ERROR_SUCCESS)
{ {
ok(TcpTableOwner->dwNumEntries > 0, "No TCP connections?!\n"); ok(TcpTableOwner->dwNumEntries > 0, "No TCP connections?!\n");
@ -235,10 +245,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwnerMod, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_MODULE_ALL) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTableOwnerMod, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_ALL) == ERROR_SUCCESS)
{ {
ok(TcpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n"); ok(TcpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n");
@ -271,10 +281,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwnerMod, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_MODULE_CONNECTIONS) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTableOwnerMod, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_CONNECTIONS) == ERROR_SUCCESS)
{ {
Found = FALSE; Found = FALSE;
for (i = 0; i < TcpTableOwnerMod->dwNumEntries; ++i) for (i = 0; i < TcpTableOwnerMod->dwNumEntries; ++i)
@ -294,10 +304,10 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
if (pAllocateAndGetTcpExTable2FromStack((PVOID *)&TcpTableOwnerMod, TRUE, GetProcessHeap(), 0, AF_INET, TCP_TABLE_OWNER_MODULE_LISTENER) == ERROR_SUCCESS) if (GetExtendedTcpTableWithAlloc((PVOID *)&TcpTableOwnerMod, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_LISTENER) == ERROR_SUCCESS)
{ {
ok(TcpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n"); ok(TcpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n");
@ -330,7 +340,7 @@ START_TEST(AllocateAndGetTcpExTable2FromStack)
} }
else else
{ {
skip("AllocateAndGetTcpExTable2FromStack failure\n"); skip("GetExtendedTcpTableWithAlloc failure\n");
} }
quit2: quit2:

12
modules/rostests/apitests/iphlpapi/testlist.c
View file

@ -3,7 +3,7 @@
#define STANDALONE #define STANDALONE
#include <apitest.h> #include <apitest.h>
extern void func_AllocateAndGetTcpExTable2FromStack(void); extern void func_GetExtendedTcpTable(void);
extern void func_GetInterfaceName(void); extern void func_GetInterfaceName(void);
extern void func_GetNetworkParams(void); extern void func_GetNetworkParams(void);
extern void func_icmp(void); extern void func_icmp(void);
@ -11,11 +11,11 @@ extern void func_SendARP(void);
const struct test winetest_testlist[] = const struct test winetest_testlist[] =
{ {
{ "AllocateAndGetTcpExTable2FromStack", func_AllocateAndGetTcpExTable2FromStack }, { "GetExtendedTcpTable", func_GetExtendedTcpTable },
{ "GetInterfaceName", func_GetInterfaceName }, { "GetInterfaceName", func_GetInterfaceName },
{ "GetNetworkParams", func_GetNetworkParams }, { "GetNetworkParams", func_GetNetworkParams },
{ "icmp", func_icmp }, { "icmp", func_icmp },
{ "SendARP", func_SendARP }, { "SendARP", func_SendARP },
{ 0, 0 } { 0, 0 }
}; };