Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-15 09:01:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

[IPHLPAPI] Implement getOwnerUdoTable(), to get UDP connections with owner PID

Browse source
This commit is contained in:
Pierre Schweitzer 2018-11-24 21:12:44 +01:00
parent 13de7f08af
commit 6fb6c7b50b
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
1 changed files with 61 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

61
dll/win32/iphlpapi/ipstats_reactos.c
View file

@ -700,6 +700,67 @@ 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;
}
DWORD getNumTcpEntries(void)
{
DWORD numEntities;