ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

iphlpapi: Add tests for GetExtendedTcpTable.

This commit is contained in:
Hans Leidekker 2012-04-16 12:35:44 +02:00 committed by Alexandre Julliard
parent e78ac4711c
commit d2dbd9bdac
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
dlls/iphlpapi/tests/iphlpapi.c
View File

@ -66,6 +66,7 @@ typedef DWORD (WINAPI *GetPerAdapterInfoFunc)(ULONG,PIP_PER_ADAPTER_INFO,PULONG)
typedef DWORD (WINAPI *GetAdaptersAddressesFunc)(ULONG,ULONG,PVOID,PIP_ADAPTER_ADDRESSES,PULONG); typedef DWORD (WINAPI *GetAdaptersAddressesFunc)(ULONG,ULONG,PVOID,PIP_ADAPTER_ADDRESSES,PULONG);
typedef DWORD (WINAPI *NotifyAddrChangeFunc)(PHANDLE,LPOVERLAPPED); typedef DWORD (WINAPI *NotifyAddrChangeFunc)(PHANDLE,LPOVERLAPPED);
typedef BOOL (WINAPI *CancelIPChangeNotifyFunc)(LPOVERLAPPED); typedef BOOL (WINAPI *CancelIPChangeNotifyFunc)(LPOVERLAPPED);
typedef DWORD (WINAPI *GetExtendedTcpTableFunc)(PVOID,PDWORD,BOOL,ULONG,TCP_TABLE_CLASS,ULONG);
static GetNumberOfInterfacesFunc gGetNumberOfInterfaces = NULL; static GetNumberOfInterfacesFunc gGetNumberOfInterfaces = NULL;
static GetIpAddrTableFunc gGetIpAddrTable = NULL; static GetIpAddrTableFunc gGetIpAddrTable = NULL;
@ -87,6 +88,7 @@ static GetPerAdapterInfoFunc gGetPerAdapterInfo = NULL;
static GetAdaptersAddressesFunc gGetAdaptersAddresses = NULL; static GetAdaptersAddressesFunc gGetAdaptersAddresses = NULL;
static NotifyAddrChangeFunc gNotifyAddrChange = NULL; static NotifyAddrChangeFunc gNotifyAddrChange = NULL;
static CancelIPChangeNotifyFunc gCancelIPChangeNotify = NULL; static CancelIPChangeNotifyFunc gCancelIPChangeNotify = NULL;
static GetExtendedTcpTableFunc gGetExtendedTcpTable = NULL;
static void loadIPHlpApi(void) static void loadIPHlpApi(void)
{ {
@ -130,6 +132,8 @@ static void loadIPHlpApi(void)
hLibrary, "NotifyAddrChange"); hLibrary, "NotifyAddrChange");
gCancelIPChangeNotify = (CancelIPChangeNotifyFunc)GetProcAddress( gCancelIPChangeNotify = (CancelIPChangeNotifyFunc)GetProcAddress(
hLibrary, "CancelIPChangeNotify"); hLibrary, "CancelIPChangeNotify");
gGetExtendedTcpTable = (GetExtendedTcpTableFunc)GetProcAddress(
hLibrary, "GetExtendedTcpTable");
} }
} }
@ -154,6 +158,7 @@ static void freeIPHlpApi(void)
gGetUdpTable = NULL; gGetUdpTable = NULL;
gNotifyAddrChange = NULL; gNotifyAddrChange = NULL;
gCancelIPChangeNotify = NULL; gCancelIPChangeNotify = NULL;
gGetExtendedTcpTable = NULL;
FreeLibrary(hLibrary); FreeLibrary(hLibrary);
hLibrary = NULL; hLibrary = NULL;
} }
@ -970,6 +975,39 @@ static void test_GetAdaptersAddresses(void)
HeapFree(GetProcessHeap(), 0, ptr); HeapFree(GetProcessHeap(), 0, ptr);
} }
static void test_GetExtendedTcpTable(void)
{
DWORD ret, size;
MIB_TCPTABLE *table;
MIB_TCPTABLE_OWNER_PID *table_pid;
if (!gGetExtendedTcpTable)
{
win_skip("GetExtendedTcpTable not available\n");
return;
}
ret = gGetExtendedTcpTable( NULL, NULL, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
ok( ret == ERROR_INVALID_PARAMETER, "got %u\n", ret );
size = 0;
ret = gGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %u\n", ret );
table = HeapAlloc( GetProcessHeap(), 0, size );
ret = gGetExtendedTcpTable( table, &size, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
ok( ret == ERROR_SUCCESS, "got %u\n", ret );
HeapFree( GetProcessHeap(), 0, table );
size = 0;
ret = gGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %u\n", ret );
table_pid = HeapAlloc( GetProcessHeap(), 0, size );
ret = gGetExtendedTcpTable( table_pid, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0 );
ok( ret == ERROR_SUCCESS, "got %u\n", ret );
HeapFree( GetProcessHeap(), 0, table_pid );
}
START_TEST(iphlpapi) START_TEST(iphlpapi)
{ {
@ -987,6 +1025,7 @@ START_TEST(iphlpapi)
testWin2KFunctions(); testWin2KFunctions();
test_GetAdaptersAddresses(); test_GetAdaptersAddresses();
test_GetExtendedTcpTable();
freeIPHlpApi(); freeIPHlpApi();
} }
} }