Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-05 15:23:03 +00:00
Code Issues Projects Releases Packages Wiki Activity

[TCPIP] Stubplement returning module information for TCP and UDP connections

Browse source
This commit is contained in:
Pierre Schweitzer 2018-11-27 22:27:45 +01:00
parent e486d8aaa6
commit f718b1d719
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
2 changed files with 43 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

17
drivers/network/tcpip/tcpip/info.c
View file

@ -308,6 +308,23 @@ TDI_STATUS InfoTdiQueryInformationEx(
else else
return TDI_INVALID_PARAMETER; return TDI_INVALID_PARAMETER;
case IP_SPECIFIC_MODULE_ENTRY_ID:
if (ID->toi_type != INFO_TYPE_PROVIDER)
return TDI_INVALID_PARAMETER;
if (ID->toi_entity.tei_entity == CO_TL_ENTITY)
if ((EntityListContext = GetContext(ID->toi_entity)))
return InfoTdiQueryGetConnectionTcpTable(EntityListContext, Buffer, BufferSize, TcpUdpClassOwner);
else
return TDI_INVALID_PARAMETER;
else if (ID->toi_entity.tei_entity == CL_TL_ENTITY)
if ((EntityListContext = GetContext(ID->toi_entity)))
return InfoTdiQueryGetConnectionUdpTable(EntityListContext, Buffer, BufferSize, TcpUdpClassOwner);
else
return TDI_INVALID_PARAMETER;
else
return TDI_INVALID_PARAMETER;
#if 0 #if 0
case IP_INTFC_INFO_ID: case IP_INTFC_INFO_ID:
if (ID->toi_type != INFO_TYPE_PROVIDER) if (ID->toi_type != INFO_TYPE_PROVIDER)

31
drivers/network/tcpip/tcpip/ninfo.c
View file

@ -185,22 +185,27 @@ TDI_STATUS InfoTdiQueryGetConnectionTcpTable(PADDRESS_FILE AddrFile,
TDI_TCPUDP_CLASS_INFO Class) TDI_TCPUDP_CLASS_INFO Class)
{ {
SIZE_T Size; SIZE_T Size;
MIB_TCPROW_OWNER_PID TcpRow; MIB_TCPROW_OWNER_MODULE TcpRow;
TDI_STATUS Status = TDI_INVALID_REQUEST; TDI_STATUS Status = TDI_INVALID_REQUEST;
TI_DbgPrint(DEBUG_INFO, ("Called.\n")); TI_DbgPrint(DEBUG_INFO, ("Called.\n"));
TcpRow.dwOwningPid = (DWORD)AddrFile->ProcessId;
if (Class == TcpUdpClassOwnerPid) if (Class == TcpUdpClassOwnerPid)
{ {
Size = sizeof(MIB_TCPROW_OWNER_PID); Size = sizeof(MIB_TCPROW_OWNER_PID);
} }
else if (Class == TcpUdpClassOwner)
{
Size = sizeof(MIB_TCPROW_OWNER_MODULE);
}
else else
{ {
ASSERT(Class != TcpUdpClassOwner);
Size = sizeof(MIB_TCPROW); Size = sizeof(MIB_TCPROW);
} }
TcpRow.dwOwningPid = (DWORD)AddrFile->ProcessId;
TcpRow.liCreateTimestamp = AddrFile->CreationTime; /* FIXME: to check */
if (AddrFile->Listener != NULL) if (AddrFile->Listener != NULL)
{ {
PADDRESS_FILE EndPoint; PADDRESS_FILE EndPoint;
@ -244,6 +249,12 @@ TDI_STATUS InfoTdiQueryGetConnectionTcpTable(PADDRESS_FILE AddrFile,
if (NT_SUCCESS(Status)) if (NT_SUCCESS(Status))
{ {
if (Class == TcpUdpClassOwner)
{
/* FIXME */
RtlZeroMemory(&TcpRow.OwningModuleInfo[0], sizeof(TcpRow.OwningModuleInfo));
}
Status = InfoCopyOut( (PCHAR)&TcpRow, Size, Status = InfoCopyOut( (PCHAR)&TcpRow, Size,
Buffer, BufferSize ); Buffer, BufferSize );
} }
@ -259,7 +270,7 @@ TDI_STATUS InfoTdiQueryGetConnectionUdpTable(PADDRESS_FILE AddrFile,
TDI_TCPUDP_CLASS_INFO Class) TDI_TCPUDP_CLASS_INFO Class)
{ {
SIZE_T Size; SIZE_T Size;
MIB_UDPROW_OWNER_PID UdpRow; MIB_UDPROW_OWNER_MODULE UdpRow;
TDI_STATUS Status = TDI_INVALID_REQUEST; TDI_STATUS Status = TDI_INVALID_REQUEST;
TI_DbgPrint(DEBUG_INFO, ("Called.\n")); TI_DbgPrint(DEBUG_INFO, ("Called.\n"));
@ -268,15 +279,25 @@ TDI_STATUS InfoTdiQueryGetConnectionUdpTable(PADDRESS_FILE AddrFile,
{ {
Size = sizeof(MIB_UDPROW_OWNER_PID); Size = sizeof(MIB_UDPROW_OWNER_PID);
} }
else if (Class == TcpUdpClassOwner)
{
Size = sizeof(MIB_UDPROW_OWNER_MODULE);
}
else else
{ {
ASSERT(Class != TcpUdpClassOwner);
Size = sizeof(MIB_UDPROW); Size = sizeof(MIB_UDPROW);
} }
UdpRow.dwLocalAddr = AddrFile->Address.Address.IPv4Address; UdpRow.dwLocalAddr = AddrFile->Address.Address.IPv4Address;
UdpRow.dwLocalPort = AddrFile->Port; UdpRow.dwLocalPort = AddrFile->Port;
UdpRow.dwOwningPid = (DWORD)AddrFile->ProcessId; UdpRow.dwOwningPid = (DWORD)AddrFile->ProcessId;
UdpRow.liCreateTimestamp = AddrFile->CreationTime; /* FIXME: to check */
UdpRow.dwFlags = 0; /* FIXME */
if (Class == TcpUdpClassOwner)
{
/* FIXME */
RtlZeroMemory(&UdpRow.OwningModuleInfo[0], sizeof(UdpRow.OwningModuleInfo));
}
Status = InfoCopyOut( (PCHAR)&UdpRow, Status = InfoCopyOut( (PCHAR)&UdpRow,
Size, Buffer, BufferSize ); Size, Buffer, BufferSize );