return TDI_INVALID_PARAMETER;
case IP_MIB_ADDRTABLE_ENTRY_ID:
- if (ID->toi_entity.tei_entity != CL_NL_ENTITY &&
- ID->toi_entity.tei_entity != CO_NL_ENTITY)
- return TDI_INVALID_PARAMETER;
-
if (ID->toi_type != INFO_TYPE_PROVIDER)
return TDI_INVALID_PARAMETER;
- return InfoTdiQueryGetAddrTable(ID->toi_entity, Buffer, BufferSize);
+ if (ID->toi_entity.tei_entity == CL_NL_ENTITY ||
+ ID->toi_entity.tei_entity == CO_NL_ENTITY)
+ return InfoTdiQueryGetAddrTable(ID->toi_entity, Buffer, BufferSize);
+ else if (ID->toi_entity.tei_entity == CO_TL_ENTITY)
+ if ((EntityListContext = GetContext(ID->toi_entity)))
+ return InfoTdiQueryGetConnectionTcpTable(EntityListContext, Buffer, BufferSize, TRUE);
+ else
+ return TDI_INVALID_PARAMETER;
+ else
+ return TDI_INVALID_PARAMETER;
case IP_MIB_ARPTABLE_ENTRY_ID:
if (ID->toi_type != INFO_TYPE_PROVIDER)
return TDI_INVALID_PARAMETER;
else if (ID->toi_entity.tei_entity == CO_TL_ENTITY)
if ((EntityListContext = GetContext(ID->toi_entity)))
- return InfoTdiQueryGetConnectionTcpTable(EntityListContext, Buffer, BufferSize);
+ return InfoTdiQueryGetConnectionTcpTable(EntityListContext, Buffer, BufferSize, FALSE);
else
return TDI_INVALID_PARAMETER;
else if (ID->toi_entity.tei_entity == CL_TL_ENTITY)
TDI_STATUS InfoTdiQueryGetConnectionTcpTable(PADDRESS_FILE AddrFile,
PNDIS_BUFFER Buffer,
- PUINT BufferSize)
+ PUINT BufferSize,
+ BOOLEAN Extended)
{
- MIB_TCPROW TcpRow;
+ SIZE_T Size;
+ MIB_TCPROW_OWNER_PID TcpRow;
TDI_STATUS Status = TDI_SUCCESS;
TI_DbgPrint(DEBUG_INFO, ("Called.\n"));
TcpRow.dwLocalAddr = AddrFile->Address.Address.IPv4Address;
TcpRow.dwLocalPort = AddrFile->Port;
+ TcpRow.dwOwningPid = (DWORD)AddrFile->ProcessId;
+ if (Extended)
+ {
+ Size = sizeof(MIB_TCPROW_OWNER_PID);
+ }
+ else
+ {
+ Size = sizeof(MIB_TCPROW);
+ }
if (AddrFile->Listener != NULL)
{
EndPoint = AddrFile->Listener->AddressFile;
- TcpRow.State = MIB_TCP_STATE_LISTEN;
+ TcpRow.dwState = MIB_TCP_STATE_LISTEN;
TcpRow.dwRemoteAddr = EndPoint->Address.Address.IPv4Address;
TcpRow.dwRemotePort = EndPoint->Port;
}
TcpRow.dwRemotePort = ntohs(EndPoint.Address[0].Address[0].sin_port);
}
- Status = TCPGetSocketStatus(AddrFile->Connection, (PULONG)&TcpRow.State);
+ Status = TCPGetSocketStatus(AddrFile->Connection, &TcpRow.dwState);
ASSERT(NT_SUCCESS(Status));
}
else
{
- TcpRow.State = 0;
+ TcpRow.dwState = 0;
TcpRow.dwRemoteAddr = 0;
TcpRow.dwRemotePort = 0;
}
if (NT_SUCCESS(Status))
{
- Status = InfoCopyOut( (PCHAR)&TcpRow, sizeof(TcpRow),
+ Status = InfoCopyOut( (PCHAR)&TcpRow, Size,
Buffer, BufferSize );
}