system/wine
system/wine
1
0
Fork 0
mirror of git://source.winehq.org/git/wine.git synced 2024-07-20 22:04:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

iphlpapi/tests: Call GetExtendedTcp/UdpTable() in a loop.

Browse source 
Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=54588
This commit is contained in:
Hans Leidekker 2023-09-13 14:27:07 +02:00 committed by Alexandre Julliard
parent c2c0fefb99
commit 46e3f259c1
1 changed files with 47 additions and 56 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

103
dlls/iphlpapi/tests/iphlpapi.c
View file

@ -1866,9 +1866,27 @@ static void test_GetAdaptersAddresses(void)
free(ptr);
}
static DWORD get_extended_tcp_table( ULONG family, TCP_TABLE_CLASS class, void **table )
{
DWORD ret, size = 0;
*table = NULL;
ret = pGetExtendedTcpTable( NULL, &size, TRUE, family, class, 0 );
if (ret != ERROR_INSUFFICIENT_BUFFER) return ret;
*table = malloc( size );
ret = pGetExtendedTcpTable( *table, &size, TRUE, family, class, 0 );
while (ret == ERROR_INSUFFICIENT_BUFFER)
{
*table = realloc( *table, size );
ret = pGetExtendedTcpTable( *table, &size, TRUE, family, class, 0 );
}
return ret;
}
static void test_GetExtendedTcpTable(void)
{
DWORD ret, size;
DWORD ret;
MIB_TCPTABLE *table;
MIB_TCPTABLE_OWNER_PID *table_pid;
MIB_TCPTABLE_OWNER_MODULE *table_module;
@ -1881,57 +1899,27 @@ static void test_GetExtendedTcpTable(void)
ret = pGetExtendedTcpTable( NULL, NULL, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
ok( ret == ERROR_INVALID_PARAMETER, "got %lu\n", ret );
size = 0;
ret = pGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table = malloc( size );
ret = pGetExtendedTcpTable( table, &size, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
ret = get_extended_tcp_table( AF_INET, TCP_TABLE_BASIC_ALL, (void **)&table );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table );
size = 0;
ret = pGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_BASIC_LISTENER, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table = malloc( size );
ret = pGetExtendedTcpTable( table, &size, TRUE, AF_INET, TCP_TABLE_BASIC_LISTENER, 0 );
ret = get_extended_tcp_table( AF_INET, TCP_TABLE_BASIC_LISTENER, (void **)&table );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table );
size = 0;
ret = pGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table_pid = malloc( size );
ret = pGetExtendedTcpTable( table_pid, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0 );
ret = get_extended_tcp_table( AF_INET, TCP_TABLE_OWNER_PID_ALL, (void **)&table_pid );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table_pid );
size = 0;
ret = pGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_LISTENER, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table_pid = malloc( size );
ret = pGetExtendedTcpTable( table_pid, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_LISTENER, 0 );
ret = get_extended_tcp_table( AF_INET, TCP_TABLE_OWNER_PID_LISTENER, (void **)&table_pid );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table_pid );
size = 0;
ret = pGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_ALL, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table_module = malloc( size );
ret = pGetExtendedTcpTable( table_module, &size, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_ALL, 0 );
ret = get_extended_tcp_table( AF_INET, TCP_TABLE_OWNER_MODULE_ALL, (void **)&table_module );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table_module );
size = 0;
ret = pGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_LISTENER, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table_module = malloc( size );
ret = pGetExtendedTcpTable( table_module, &size, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_LISTENER, 0 );
ret = get_extended_tcp_table( AF_INET, TCP_TABLE_OWNER_MODULE_LISTENER, (void **)&table_module );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table_module );
}
@ -1985,9 +1973,27 @@ static void test_AllocateAndGetTcpExTableFromStack(void)
ok( ret == ERROR_NOT_SUPPORTED, "got %lu\n", ret );
}
static DWORD get_extended_udp_table( ULONG family, UDP_TABLE_CLASS class, void **table )
{
DWORD ret, size = 0;
*table = NULL;
ret = pGetExtendedUdpTable( NULL, &size, TRUE, family, class, 0 );
if (ret != ERROR_INSUFFICIENT_BUFFER) return ret;
*table = malloc( size );
ret = pGetExtendedUdpTable( *table, &size, TRUE, family, class, 0 );
while (ret == ERROR_INSUFFICIENT_BUFFER)
{
*table = realloc( *table, size );
ret = pGetExtendedUdpTable( *table, &size, TRUE, family, class, 0 );
}
return ret;
}
static void test_GetExtendedUdpTable(void)
{
DWORD ret, size;
DWORD ret;
MIB_UDPTABLE *table;
MIB_UDPTABLE_OWNER_PID *table_pid;
MIB_UDPTABLE_OWNER_MODULE *table_module;
@ -2000,30 +2006,15 @@ static void test_GetExtendedUdpTable(void)
ret = pGetExtendedUdpTable( NULL, NULL, TRUE, AF_INET, UDP_TABLE_BASIC, 0 );
ok( ret == ERROR_INVALID_PARAMETER, "got %lu\n", ret );
size = 0;
ret = pGetExtendedUdpTable( NULL, &size, TRUE, AF_INET, UDP_TABLE_BASIC, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table = malloc( size );
ret = pGetExtendedUdpTable( table, &size, TRUE, AF_INET, UDP_TABLE_BASIC, 0 );
ret = get_extended_udp_table( AF_INET, UDP_TABLE_BASIC, (void **)&table );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table );
size = 0;
ret = pGetExtendedUdpTable( NULL, &size, TRUE, AF_INET, UDP_TABLE_OWNER_PID, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table_pid = malloc( size );
ret = pGetExtendedUdpTable( table_pid, &size, TRUE, AF_INET, UDP_TABLE_OWNER_PID, 0 );
ret = get_extended_udp_table( AF_INET, UDP_TABLE_OWNER_PID, (void **)&table_pid );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table_pid );
size = 0;
ret = pGetExtendedUdpTable( NULL, &size, TRUE, AF_INET, UDP_TABLE_OWNER_MODULE, 0 );
ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %lu\n", ret );
table_module = malloc( size );
ret = pGetExtendedUdpTable( table_module, &size, TRUE, AF_INET, UDP_TABLE_OWNER_MODULE, 0 );
ret = get_extended_udp_table( AF_INET, UDP_TABLE_OWNER_MODULE, (void **)&table_module );
ok( ret == ERROR_SUCCESS, "got %lu\n", ret );
free( table_module );
}