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

[IPHLPAPI] Implement the IPv4 TCP_TABLE_OWNER_PID_* cases in GetExtendedTcpTable()

Browse source
This commit is contained in:
Pierre Schweitzer 2018-11-24 20:32:55 +01:00
parent dca7e5689f
commit c949a12506
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
2 changed files with 116 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

111
dll/win32/iphlpapi/iphlpapi_main.c
View file

@ -922,6 +922,117 @@ DWORD WINAPI GetExtendedTcpTable(PVOID pTcpTable, PDWORD pdwSize, BOOL bOrder, U
}
break;
case TCP_TABLE_OWNER_PID_ALL:
{
PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getOwnerTcpTable();
PMIB_TCPTABLE_OWNER_PID pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
{
if (sizeof(DWORD) + pOurTcpTable->dwNumEntries * sizeof(MIB_TCPROW_OWNER_PID) > *pdwSize || !pTheirTcpTable)
{
*pdwSize = sizeof(DWORD) + pOurTcpTable->dwNumEntries * sizeof(MIB_TCPROW_OWNER_PID);
ret = ERROR_INSUFFICIENT_BUFFER;
}
else
{
memcpy(pTheirTcpTable, pOurTcpTable, sizeof(DWORD) + pOurTcpTable->dwNumEntries * sizeof(MIB_TCPROW_OWNER_PID));
/* Don't sort on PID, so use basic helper */
if (bOrder)
qsort(pTheirTcpTable->table, pTheirTcpTable->dwNumEntries,
sizeof(MIB_TCPROW_OWNER_PID), TcpTableSorter);
}
}
}
break;
case TCP_TABLE_OWNER_PID_CONNECTIONS:
{
PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getOwnerTcpTable();
PMIB_TCPTABLE_OWNER_PID pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
{
for (i = 0, count = 0; i < pOurTcpTable->dwNumEntries; ++i)
{
if (pOurTcpTable->table[i].dwState != MIB_TCP_STATE_LISTEN)
{
++count;
}
}
if (sizeof(DWORD) + count * sizeof(MIB_TCPROW_OWNER_PID) > *pdwSize || !pTheirTcpTable)
{
*pdwSize = sizeof(DWORD) + count * sizeof(MIB_TCPROW_OWNER_PID);
ret = ERROR_INSUFFICIENT_BUFFER;
}
else
{
pTheirTcpTable->dwNumEntries = count;
for (i = 0, count = 0; i < pOurTcpTable->dwNumEntries; ++i)
{
if (pOurTcpTable->table[i].dwState != MIB_TCP_STATE_LISTEN)
{
memcpy(&pTheirTcpTable->table[count], &pOurTcpTable->table[i], sizeof(MIB_TCPROW_OWNER_PID));
++count;
}
}
ASSERT(count == pTheirTcpTable->dwNumEntries);
/* Don't sort on PID, so use basic helper */
if (bOrder)
qsort(pTheirTcpTable->table, pTheirTcpTable->dwNumEntries,
sizeof(MIB_TCPROW_OWNER_PID), TcpTableSorter);
}
}
}
break;
case TCP_TABLE_OWNER_PID_LISTENER:
{
PMIB_TCPTABLE_OWNER_PID pOurTcpTable = getOwnerTcpTable();
PMIB_TCPTABLE_OWNER_PID pTheirTcpTable = pTcpTable;
if (pOurTcpTable)
{
for (i = 0, count = 0; i < pOurTcpTable->dwNumEntries; ++i)
{
if (pOurTcpTable->table[i].dwState == MIB_TCP_STATE_LISTEN)
{
++count;
}
}
if (sizeof(DWORD) + count * sizeof(MIB_TCPROW_OWNER_PID) > *pdwSize || !pTheirTcpTable)
{
*pdwSize = sizeof(DWORD) + count * sizeof(MIB_TCPROW_OWNER_PID);
ret = ERROR_INSUFFICIENT_BUFFER;
}
else
{
pTheirTcpTable->dwNumEntries = count;
for (i = 0, count = 0; i < pOurTcpTable->dwNumEntries; ++i)
{
if (pOurTcpTable->table[i].dwState == MIB_TCP_STATE_LISTEN)
{
memcpy(&pTheirTcpTable->table[count], &pOurTcpTable->table[i], sizeof(MIB_TCPROW_OWNER_PID));
++count;
}
}
ASSERT(count == pTheirTcpTable->dwNumEntries);
/* Don't sort on PID, so use basic helper */
if (bOrder)
qsort(pTheirTcpTable->table, pTheirTcpTable->dwNumEntries,
sizeof(MIB_TCPROW_OWNER_PID), TcpTableSorter);
}
}
}
break;
default:
UNIMPLEMENTED;
ret = ERROR_INVALID_PARAMETER;

5
dll/win32/iphlpapi/ipstats.h
View file

@ -105,4 +105,9 @@ DWORD getNumTcpEntries(void);
*/
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);
#endif /* ndef WINE_IPSTATS_H_ */