[TCPIP] In InfoTdiQueryGetConnectionTcpTable(), only return if there's a connection

This commit is contained in:
Pierre Schweitzer 2018-11-25 09:32:30 +01:00
parent f87d9caf77
commit ce7f9211a0
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B

View file

@ -186,7 +186,7 @@ TDI_STATUS InfoTdiQueryGetConnectionTcpTable(PADDRESS_FILE AddrFile,
{ {
SIZE_T Size; SIZE_T Size;
MIB_TCPROW_OWNER_PID TcpRow; MIB_TCPROW_OWNER_PID TcpRow;
TDI_STATUS Status = TDI_SUCCESS; TDI_STATUS Status = TDI_INVALID_REQUEST;
TI_DbgPrint(DEBUG_INFO, ("Called.\n")); TI_DbgPrint(DEBUG_INFO, ("Called.\n"));
@ -211,6 +211,8 @@ TDI_STATUS InfoTdiQueryGetConnectionTcpTable(PADDRESS_FILE AddrFile,
TcpRow.dwLocalPort = AddrFile->Port; TcpRow.dwLocalPort = AddrFile->Port;
TcpRow.dwRemoteAddr = EndPoint->Address.Address.IPv4Address; TcpRow.dwRemoteAddr = EndPoint->Address.Address.IPv4Address;
TcpRow.dwRemotePort = EndPoint->Port; TcpRow.dwRemotePort = EndPoint->Port;
Status = TDI_SUCCESS;
} }
else if (AddrFile->Connection != NULL && else if (AddrFile->Connection != NULL &&
AddrFile->Connection->SocketContext != NULL) AddrFile->Connection->SocketContext != NULL)
@ -238,12 +240,6 @@ TDI_STATUS InfoTdiQueryGetConnectionTcpTable(PADDRESS_FILE AddrFile,
} }
} }
} }
else
{
TcpRow.dwState = 0;
TcpRow.dwRemoteAddr = 0;
TcpRow.dwRemotePort = 0;
}
if (NT_SUCCESS(Status)) if (NT_SUCCESS(Status))
{ {