[reactos] 01/02: [IPHLPAPI] Reduce code duplication and use a single function for TCP and UDP enumerations
30 Nov
2018
30 Nov
'18
7:06 p.m.
https://git.reactos.org/?p=reactos.git;a=commitdiff;h=bf052e120b3ad451b1b45…
commit bf052e120b3ad451b1b450cc7c1ebf12b1f57ffa
Author: Pierre Schweitzer <pierre(a)reactos.org>
AuthorDate: Fri Nov 30 19:42:38 2018 +0100
Commit: Pierre Schweitzer <pierre(a)reactos.org>
CommitDate: Fri Nov 30 20:06:29 2018 +0100
[IPHLPAPI] Reduce code duplication and use a single function for TCP and UDP
enumerations
Furthermore, memory was improperly allocated before (alignment issues) leading to
heap
corruption under certain circumstances
---
dll/win32/iphlpapi/iphlpapi_main.c | 24 +--
dll/win32/iphlpapi/ipstats.h | 30 +---
dll/win32/iphlpapi/ipstats_reactos.c | 291 ++++-------------------------------
3 files changed, 53 insertions(+), 292 deletions(-)
diff --git a/dll/win32/iphlpapi/iphlpapi_main.c b/dll/win32/iphlpapi/iphlpapi_main.c
index 023f2fd082..11bddb40c5 100644
--- a/dll/win32/iphlpapi/iphlpapi_main.c
+++ b/dll/win32/iphlpapi/iphlpapi_main.c
@@ -1040,7 +1040,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
{
case TCP_TABLE_BASIC_ALL:
{
- PMIB_TCPTABLE pOurTcpTable = getTcpTable();
+ PMIB_TCPTABLE pOurTcpTable = getTcpTable(ClassBasic);
PMIB_TCPTABLE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1066,7 +1066,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_BASIC_CONNECTIONS:
{
- PMIB_TCPTABLE pOurTcpTable = getTcpTable();
+ PMIB_TCPTABLE pOurTcpTable = getTcpTable(ClassBasic);
PMIB_TCPTABLE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1110,7 +1110,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_BASIC_LISTENER:
{
- PMIB_TCPTABLE pOurTcpTable = getTcpTable();
+ PMIB_TCPTABLE pOurTcpTable = getTcpTable(ClassBasic);
PMIB_TCPTABLE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1154,7 +1154,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_OWNER_PID_ALL:
{
- PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getOwnerTcpTable();
+ PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getTcpTable(ClassModulePid);
PMIB_TCPTABLE_OWNER_PID pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1181,7 +1181,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_OWNER_PID_CONNECTIONS:
{
- PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getOwnerTcpTable();
+ PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getTcpTable(ClassModulePid);
PMIB_TCPTABLE_OWNER_PID pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1226,7 +1226,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_OWNER_PID_LISTENER:
{
- PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getOwnerTcpTable();
+ PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getTcpTable(ClassModulePid);
PMIB_TCPTABLE_OWNER_PID pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1271,7 +1271,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_OWNER_MODULE_ALL:
{
- PMIB_TCPTABLE_OWNER_MODULE pOurTcpTable = getOwnerModTcpTable();
+ PMIB_TCPTABLE_OWNER_MODULE pOurTcpTable = getTcpTable(ClassModule);
PMIB_TCPTABLE_OWNER_MODULE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1298,7 +1298,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_OWNER_MODULE_CONNECTIONS:
{
- PMIB_TCPTABLE_OWNER_MODULE pOurTcpTable = getOwnerModTcpTable();
+ PMIB_TCPTABLE_OWNER_MODULE pOurTcpTable = getTcpTable(ClassModule);
PMIB_TCPTABLE_OWNER_MODULE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1343,7 +1343,7 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize,
BOOL bOrder, U
case TCP_TABLE_OWNER_MODULE_LISTENER:
{
- PMIB_TCPTABLE_OWNER_MODULE pOurTcpTable = getOwnerModTcpTable();
+ PMIB_TCPTABLE_OWNER_MODULE pOurTcpTable = getTcpTable(ClassModule);
PMIB_TCPTABLE_OWNER_MODULE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
@@ -1450,7 +1450,7 @@ DWORD WINAPI GetExtendedUdpTable(PVOID pUdpTable, PDWORD pdwSize,
BOOL bOrder, U
{
case UDP_TABLE_BASIC:
{
- PMIB_UDPTABLE pOurUdpTable = getUdpTable();
+ PMIB_UDPTABLE pOurUdpTable = getUdpTable(ClassBasic);
PMIB_UDPTABLE pTheirUdpTable = pUdpTable;
if (pOurUdpTable)
@@ -1476,7 +1476,7 @@ DWORD WINAPI GetExtendedUdpTable(PVOID pUdpTable, PDWORD pdwSize,
BOOL bOrder, U
case UDP_TABLE_OWNER_PID:
{
- PMIB_UDPTABLE_OWNER_PID pOurUdpTable = getOwnerUdpTable();
+ PMIB_UDPTABLE_OWNER_PID pOurUdpTable = getUdpTable(ClassModulePid);
PMIB_UDPTABLE_OWNER_PID pTheirUdpTable = pUdpTable;
if (pOurUdpTable)
@@ -1502,7 +1502,7 @@ DWORD WINAPI GetExtendedUdpTable(PVOID pUdpTable, PDWORD pdwSize,
BOOL bOrder, U
case UDP_TABLE_OWNER_MODULE:
{
- PMIB_UDPTABLE_OWNER_MODULE pOurUdpTable = getOwnerModUdpTable();
+ PMIB_UDPTABLE_OWNER_MODULE pOurUdpTable = getUdpTable(ClassModule);
PMIB_UDPTABLE_OWNER_MODULE pTheirUdpTable = pUdpTable;
if (pOurUdpTable)
diff --git a/dll/win32/iphlpapi/ipstats.h b/dll/win32/iphlpapi/ipstats.h
index 6becd0014c..66d78c946f 100644
--- a/dll/win32/iphlpapi/ipstats.h
+++ b/dll/win32/iphlpapi/ipstats.h
@@ -76,6 +76,12 @@ typedef struct _RouteTable {
RouteEntry routes[1];
} RouteTable;
+typedef enum _CLASS_TABLE {
+ ClassBasic,
+ ClassModulePid,
+ ClassModule
+} CLASS_TABLE;
+
/* Allocates and returns to you the route table, or NULL if it can't allocate
* enough memory. free() the returned table.
*/
@@ -95,17 +101,7 @@ DWORD getNumUdpEntries(void);
/* Allocates and returns to you the UDP state table, or NULL if it can't
* allocate enough memory. free() the returned table.
*/
-PMIB_UDPTABLE getUdpTable(void);
-
-/* Allocates and returns to you the UDP state table with owner PID,
- * or NULL if it can't allocate enough memory. free() the returned table.
- */
-PMIB_UDPTABLE_OWNER_PID getOwnerUdpTable(void);
-
-/* Allocates and returns to you the UDP state table with module info,
- * or NULL if it can't allocate enough memory. free() the returned table.
- */
-PMIB_UDPTABLE_OWNER_MODULE getOwnerModUdpTable(void);
+PVOID getUdpTable(CLASS_TABLE Class);
/* Returns the number of entries in the TCP state table. */
DWORD getNumTcpEntries(void);
@@ -113,16 +109,6 @@ DWORD getNumTcpEntries(void);
/* Allocates and returns to you the TCP state table, or NULL if it can't
* allocate enough memory. free() the returned table.
*/
-PMIB_TCPTABLE getTcpTable(void);
-
-/* Allocates and returns to you the TCP state table with owner PID,
- * or NULL if it can't allocate enough memory. free() the returned table.
- */
-PMIB_TCPTABLE_OWNER_PID getOwnerTcpTable(void);
-
-/* Allocates and returns to you the TCP state table with module info,
- * or NULL if it can't allocate enough memory. free() the returned table.
- */
-PMIB_TCPTABLE_OWNER_MODULE getOwnerModTcpTable(void);
+PVOID getTcpTable(CLASS_TABLE Class);
#endif /* ndef WINE_IPSTATS_H_ */
diff --git a/dll/win32/iphlpapi/ipstats_reactos.c b/dll/win32/iphlpapi/ipstats_reactos.c
index 84351ea9a0..5906388fc4 100644
--- a/dll/win32/iphlpapi/ipstats_reactos.c
+++ b/dll/win32/iphlpapi/ipstats_reactos.c
@@ -590,6 +590,21 @@ PMIB_IPNETTABLE getArpTable(void)
return IpArpTable;
}
+struct _TABLE_CALL
+{
+ DWORD TOIID;
+ SIZE_T UdpSize;
+ SIZE_T TcpSize;
+ SIZE_T UdpOffset;
+ SIZE_T TcpOffset;
+} UdpTcpTableCall[] = {
+ {IP_MIB_ARPTABLE_ENTRY_ID, sizeof(MIB_UDPROW), sizeof(MIB_TCPROW),
FIELD_OFFSET(MIB_UDPTABLE, table), FIELD_OFFSET(MIB_TCPTABLE, table)},
+ {IP_MIB_ADDRTABLE_ENTRY_ID, sizeof(MIB_UDPROW_OWNER_PID),
sizeof(MIB_TCPROW_OWNER_PID), FIELD_OFFSET(MIB_UDPTABLE_OWNER_PID, table),
FIELD_OFFSET(MIB_TCPTABLE_OWNER_PID, table)},
+ {IP_SPECIFIC_MODULE_ENTRY_ID, sizeof(MIB_UDPROW_OWNER_MODULE),
sizeof(MIB_TCPROW_OWNER_MODULE), FIELD_OFFSET(MIB_UDPTABLE_OWNER_MODULE, table),
FIELD_OFFSET(MIB_TCPTABLE_OWNER_MODULE, table)},
+};
+
+#define Add2Ptr(PTR, INC) (PVOID)((ULONG_PTR)(PTR) + (INC))
+
DWORD getNumUdpEntries(void)
{
DWORD numEntities;
@@ -639,7 +654,7 @@ DWORD getNumUdpEntries(void)
return totalNumber;
}
-PMIB_UDPTABLE getUdpTable(void)
+PVOID getUdpTable(CLASS_TABLE Class)
{
DWORD numEntities, returnSize;
TDIEntityID *entitySet;
@@ -647,7 +662,7 @@ PMIB_UDPTABLE getUdpTable(void)
int i, totalNumber, TmpIdx, CurrIdx = 0;
NTSTATUS status;
PMIB_UDPTABLE IpUdpTable = NULL;
- PMIB_UDPROW AdapterUdpTable = NULL;
+ PVOID AdapterUdpTable = NULL;
TRACE("called.\n");
@@ -661,7 +676,7 @@ PMIB_UDPTABLE getUdpTable(void)
IpUdpTable = HeapAlloc
( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_UDPROW) * totalNumber) );
+ UdpTcpTableCall[Class].UdpOffset + (UdpTcpTableCall[Class].UdpSize * totalNumber) );
if (!IpUdpTable) {
closeTcpFile(tcpFile);
return NULL;
@@ -676,17 +691,19 @@ PMIB_UDPTABLE getUdpTable(void)
status = tdiGetSetOfThings( tcpFile,
INFO_CLASS_PROTOCOL,
INFO_TYPE_PROVIDER,
- IP_MIB_ARPTABLE_ENTRY_ID,
+ UdpTcpTableCall[Class].TOIID,
CL_TL_ENTITY,
entitySet[i].tei_instance,
0,
- sizeof(MIB_UDPROW),
- (PVOID *)&AdapterUdpTable,
+ UdpTcpTableCall[Class].UdpSize,
+ &AdapterUdpTable,
&returnSize );
if( status == STATUS_SUCCESS ) {
for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
- IpUdpTable->table[CurrIdx] = AdapterUdpTable[TmpIdx];
+ CopyMemory(Add2Ptr(IpUdpTable, UdpTcpTableCall[Class].UdpOffset +
UdpTcpTableCall[Class].UdpSize * CurrIdx),
+ Add2Ptr(AdapterUdpTable, UdpTcpTableCall[Class].UdpSize *
TmpIdx),
+ UdpTcpTableCall[Class].UdpSize);
tdiFreeThingSet( AdapterUdpTable );
}
}
@@ -700,128 +717,6 @@ PMIB_UDPTABLE getUdpTable(void)
return IpUdpTable;
}
-PMIB_UDPTABLE_OWNER_PID getOwnerUdpTable(void)
-{
- DWORD numEntities, returnSize;
- TDIEntityID *entitySet;
- HANDLE tcpFile;
- int i, totalNumber, TmpIdx, CurrIdx = 0;
- NTSTATUS status;
- PMIB_UDPTABLE_OWNER_PID IpOwnerUdpTable = NULL;
- PMIB_UDPROW_OWNER_PID AdapterOwnerUdpTable = NULL;
-
- TRACE("called.\n");
-
- totalNumber = getNumUdpEntries();
-
- status = openTcpFile( &tcpFile, FILE_READ_DATA );
- if( !NT_SUCCESS(status) ) {
- ERR("openTcpFile returned 0x%08lx\n", status);
- return 0;
- }
-
- IpOwnerUdpTable = HeapAlloc
- ( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_UDPROW_OWNER_PID) * totalNumber) );
- if (!IpOwnerUdpTable) {
- closeTcpFile(tcpFile);
- return NULL;
- }
-
- status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
-
- for( i = 0; i < numEntities; i++ ) {
- if( entitySet[i].tei_entity == CL_TL_ENTITY &&
- hasArp( tcpFile, &entitySet[i] ) ) {
-
- status = tdiGetSetOfThings( tcpFile,
- INFO_CLASS_PROTOCOL,
- INFO_TYPE_PROVIDER,
- IP_MIB_ADDRTABLE_ENTRY_ID,
- CL_TL_ENTITY,
- entitySet[i].tei_instance,
- 0,
- sizeof(MIB_UDPROW_OWNER_PID),
- (PVOID *)&AdapterOwnerUdpTable,
- &returnSize );
-
- if( status == STATUS_SUCCESS ) {
- for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
- IpOwnerUdpTable->table[CurrIdx] = AdapterOwnerUdpTable[TmpIdx];
- tdiFreeThingSet( AdapterOwnerUdpTable );
- }
- }
- }
-
- closeTcpFile( tcpFile );
-
- tdiFreeThingSet( entitySet );
- IpOwnerUdpTable->dwNumEntries = CurrIdx;
-
- return IpOwnerUdpTable;
-}
-
-PMIB_UDPTABLE_OWNER_MODULE getOwnerModUdpTable(void)
-{
- DWORD numEntities, returnSize;
- TDIEntityID *entitySet;
- HANDLE tcpFile;
- int i, totalNumber, TmpIdx, CurrIdx = 0;
- NTSTATUS status;
- PMIB_UDPTABLE_OWNER_MODULE IpOwnerModUdpTable = NULL;
- PMIB_UDPROW_OWNER_MODULE AdapterOwnerModUdpTable = NULL;
-
- TRACE("called.\n");
-
- totalNumber = getNumUdpEntries();
-
- status = openTcpFile( &tcpFile, FILE_READ_DATA );
- if( !NT_SUCCESS(status) ) {
- ERR("openTcpFile returned 0x%08lx\n", status);
- return 0;
- }
-
- IpOwnerModUdpTable = HeapAlloc
- ( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_UDPROW_OWNER_MODULE) * totalNumber) );
- if (!IpOwnerModUdpTable) {
- closeTcpFile(tcpFile);
- return NULL;
- }
-
- status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
-
- for( i = 0; i < numEntities; i++ ) {
- if( entitySet[i].tei_entity == CL_TL_ENTITY &&
- hasArp( tcpFile, &entitySet[i] ) ) {
-
- status = tdiGetSetOfThings( tcpFile,
- INFO_CLASS_PROTOCOL,
- INFO_TYPE_PROVIDER,
- IP_SPECIFIC_MODULE_ENTRY_ID,
- CL_TL_ENTITY,
- entitySet[i].tei_instance,
- 0,
- sizeof(MIB_UDPROW_OWNER_MODULE),
- (PVOID *)&AdapterOwnerModUdpTable,
- &returnSize );
-
- if( status == STATUS_SUCCESS ) {
- for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
- IpOwnerModUdpTable->table[CurrIdx] =
AdapterOwnerModUdpTable[TmpIdx];
- tdiFreeThingSet( AdapterOwnerModUdpTable );
- }
- }
- }
-
- closeTcpFile( tcpFile );
-
- tdiFreeThingSet( entitySet );
- IpOwnerModUdpTable->dwNumEntries = CurrIdx;
-
- return IpOwnerModUdpTable;
-}
-
DWORD getNumTcpEntries(void)
{
DWORD numEntities;
@@ -871,7 +766,7 @@ DWORD getNumTcpEntries(void)
return totalNumber;
}
-PMIB_TCPTABLE getTcpTable(void)
+PVOID getTcpTable(CLASS_TABLE Class)
{
DWORD numEntities, returnSize;
TDIEntityID *entitySet;
@@ -879,7 +774,7 @@ PMIB_TCPTABLE getTcpTable(void)
int i, totalNumber, TmpIdx, CurrIdx = 0;
NTSTATUS status;
PMIB_TCPTABLE IpTcpTable = NULL;
- PMIB_TCPROW AdapterTcpTable = NULL;
+ PVOID AdapterTcpTable = NULL;
TRACE("called.\n");
@@ -893,7 +788,7 @@ PMIB_TCPTABLE getTcpTable(void)
IpTcpTable = HeapAlloc
( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_TCPROW) * totalNumber) );
+ UdpTcpTableCall[Class].TcpOffset + (UdpTcpTableCall[Class].TcpSize * totalNumber) );
if (!IpTcpTable) {
closeTcpFile(tcpFile);
return NULL;
@@ -908,17 +803,19 @@ PMIB_TCPTABLE getTcpTable(void)
status = tdiGetSetOfThings( tcpFile,
INFO_CLASS_PROTOCOL,
INFO_TYPE_PROVIDER,
- IP_MIB_ARPTABLE_ENTRY_ID,
+ UdpTcpTableCall[Class].TOIID,
CO_TL_ENTITY,
entitySet[i].tei_instance,
0,
- sizeof(MIB_TCPROW),
- (PVOID *)&AdapterTcpTable,
+ UdpTcpTableCall[Class].TcpSize,
+ &AdapterTcpTable,
&returnSize );
if( status == STATUS_SUCCESS ) {
for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
- IpTcpTable->table[CurrIdx] = AdapterTcpTable[TmpIdx];
+ CopyMemory(Add2Ptr(IpTcpTable, UdpTcpTableCall[Class].TcpOffset +
UdpTcpTableCall[Class].TcpSize * CurrIdx),
+ Add2Ptr(AdapterTcpTable, UdpTcpTableCall[Class].TcpSize *
TmpIdx),
+ UdpTcpTableCall[Class].TcpSize);
tdiFreeThingSet( AdapterTcpTable );
}
}
@@ -931,125 +828,3 @@ PMIB_TCPTABLE getTcpTable(void)
return IpTcpTable;
}
-
-PMIB_TCPTABLE_OWNER_PID getOwnerTcpTable(void)
-{
- DWORD numEntities, returnSize;
- TDIEntityID *entitySet;
- HANDLE tcpFile;
- int i, totalNumber, TmpIdx, CurrIdx = 0;
- NTSTATUS status;
- PMIB_TCPTABLE_OWNER_PID IpOwnerTcpTable = NULL;
- PMIB_TCPROW_OWNER_PID AdapterOwnerTcpTable = NULL;
-
- TRACE("called.\n");
-
- totalNumber = getNumTcpEntries();
-
- status = openTcpFile( &tcpFile, FILE_READ_DATA );
- if( !NT_SUCCESS(status) ) {
- ERR("openTcpFile returned 0x%08lx\n", status);
- return 0;
- }
-
- IpOwnerTcpTable = HeapAlloc
- ( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_TCPROW_OWNER_PID) * totalNumber) );
- if (!IpOwnerTcpTable) {
- closeTcpFile(tcpFile);
- return NULL;
- }
-
- status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
-
- for( i = 0; i < numEntities; i++ ) {
- if( entitySet[i].tei_entity == CO_TL_ENTITY &&
- hasArp( tcpFile, &entitySet[i] ) ) {
-
- status = tdiGetSetOfThings( tcpFile,
- INFO_CLASS_PROTOCOL,
- INFO_TYPE_PROVIDER,
- IP_MIB_ADDRTABLE_ENTRY_ID,
- CO_TL_ENTITY,
- entitySet[i].tei_instance,
- 0,
- sizeof(MIB_TCPROW_OWNER_PID),
- (PVOID *)&AdapterOwnerTcpTable,
- &returnSize );
-
- if( status == STATUS_SUCCESS ) {
- for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
- IpOwnerTcpTable->table[CurrIdx] = AdapterOwnerTcpTable[TmpIdx];
- tdiFreeThingSet( AdapterOwnerTcpTable );
- }
- }
- }
-
- closeTcpFile( tcpFile );
-
- tdiFreeThingSet( entitySet );
- IpOwnerTcpTable->dwNumEntries = CurrIdx;
-
- return IpOwnerTcpTable;
-}
-
-PMIB_TCPTABLE_OWNER_MODULE getOwnerModTcpTable(void)
-{
- DWORD numEntities, returnSize;
- TDIEntityID *entitySet;
- HANDLE tcpFile;
- int i, totalNumber, TmpIdx, CurrIdx = 0;
- NTSTATUS status;
- PMIB_TCPTABLE_OWNER_MODULE IpOwnerModTcpTable = NULL;
- PMIB_TCPROW_OWNER_MODULE AdapterOwnerModTcpTable = NULL;
-
- TRACE("called.\n");
-
- totalNumber = getNumTcpEntries();
-
- status = openTcpFile( &tcpFile, FILE_READ_DATA );
- if( !NT_SUCCESS(status) ) {
- ERR("openTcpFile returned 0x%08lx\n", status);
- return 0;
- }
-
- IpOwnerModTcpTable = HeapAlloc
- ( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_TCPROW_OWNER_MODULE) * totalNumber) );
- if (!IpOwnerModTcpTable) {
- closeTcpFile(tcpFile);
- return NULL;
- }
-
- status = tdiGetEntityIDSet( tcpFile, &entitySet, &numEntities );
-
- for( i = 0; i < numEntities; i++ ) {
- if( entitySet[i].tei_entity == CO_TL_ENTITY &&
- hasArp( tcpFile, &entitySet[i] ) ) {
-
- status = tdiGetSetOfThings( tcpFile,
- INFO_CLASS_PROTOCOL,
- INFO_TYPE_PROVIDER,
- IP_SPECIFIC_MODULE_ENTRY_ID,
- CO_TL_ENTITY,
- entitySet[i].tei_instance,
- 0,
- sizeof(MIB_TCPROW_OWNER_MODULE),
- (PVOID *)&AdapterOwnerModTcpTable,
- &returnSize );
-
- if( status == STATUS_SUCCESS ) {
- for( TmpIdx = 0; TmpIdx < returnSize; TmpIdx++, CurrIdx++ )
- IpOwnerModTcpTable->table[CurrIdx] =
AdapterOwnerModTcpTable[TmpIdx];
- tdiFreeThingSet( AdapterOwnerModTcpTable );
- }
- }
- }
-
- closeTcpFile( tcpFile );
-
- tdiFreeThingSet( entitySet );
- IpOwnerModTcpTable->dwNumEntries = CurrIdx;
-
- return IpOwnerModTcpTable;
-}
2177
days inactive
2177
days old
0 comments
1 participants
participants (1)
-
Pierre Schweitzer