{
case TCP_TABLE_BASIC_ALL:
{
- PMIB_TCPTABLE pOurTcpTable = getTcpTable();
+ PMIB_TCPTABLE pOurTcpTable = getTcpTable(ClassBasic);
PMIB_TCPTABLE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
case TCP_TABLE_BASIC_CONNECTIONS:
{
- PMIB_TCPTABLE pOurTcpTable = getTcpTable();
+ PMIB_TCPTABLE pOurTcpTable = getTcpTable(ClassBasic);
PMIB_TCPTABLE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
case TCP_TABLE_BASIC_LISTENER:
{
- PMIB_TCPTABLE pOurTcpTable = getTcpTable();
+ PMIB_TCPTABLE pOurTcpTable = getTcpTable(ClassBasic);
PMIB_TCPTABLE pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
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)
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)
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)
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)
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)
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)
{
case UDP_TABLE_BASIC:
{
- PMIB_UDPTABLE pOurUdpTable = getUdpTable();
+ PMIB_UDPTABLE pOurUdpTable = getUdpTable(ClassBasic);
PMIB_UDPTABLE pTheirUdpTable = pUdpTable;
if (pOurUdpTable)
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)
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)
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;
return totalNumber;
}
-PMIB_UDPTABLE getUdpTable(void)
+PVOID getUdpTable(CLASS_TABLE Class)
{
DWORD numEntities, returnSize;
TDIEntityID *entitySet;
int i, totalNumber, TmpIdx, CurrIdx = 0;
NTSTATUS status;
PMIB_UDPTABLE IpUdpTable = NULL;
- PMIB_UDPROW AdapterUdpTable = NULL;
+ PVOID AdapterUdpTable = NULL;
TRACE("called.\n");
IpUdpTable = HeapAlloc
( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_UDPROW) * totalNumber) );
+ UdpTcpTableCall[Class].UdpOffset + (UdpTcpTableCall[Class].UdpSize * totalNumber) );
if (!IpUdpTable) {
closeTcpFile(tcpFile);
return NULL;
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 );
}
}
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;
return totalNumber;
}
-PMIB_TCPTABLE getTcpTable(void)
+PVOID getTcpTable(CLASS_TABLE Class)
{
DWORD numEntities, returnSize;
TDIEntityID *entitySet;
int i, totalNumber, TmpIdx, CurrIdx = 0;
NTSTATUS status;
PMIB_TCPTABLE IpTcpTable = NULL;
- PMIB_TCPROW AdapterTcpTable = NULL;
+ PVOID AdapterTcpTable = NULL;
TRACE("called.\n");
IpTcpTable = HeapAlloc
( GetProcessHeap(), 0,
- sizeof(DWORD) + (sizeof(MIB_TCPROW) * totalNumber) );
+ UdpTcpTableCall[Class].TcpOffset + (UdpTcpTableCall[Class].TcpSize * totalNumber) );
if (!IpTcpTable) {
closeTcpFile(tcpFile);
return NULL;
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 );
}
}
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;
-}