Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-28 18:15:11 +00:00
Code Issues Projects Releases Packages Wiki Activity

[TCPIP] Don't assume the TEB is always defined

Browse source 
Fixes the crash in kmtest:TcpIpConnect test.
This commit is contained in:
Pierre Schweitzer 2018-12-31 13:59:58 +01:00
parent ccb8c906b0
commit 89a4cde14a
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
drivers/network/tcpip/tcpip/fileobjs.c
View file

@ -396,6 +396,7 @@ NTSTATUS FileOpenAddress(
PVOID Options) PVOID Options)
{ {
PADDRESS_FILE AddrFile; PADDRESS_FILE AddrFile;
PTEB Teb;
TI_DbgPrint(MID_TRACE, ("Called (Proto %d).\n", Protocol)); TI_DbgPrint(MID_TRACE, ("Called (Proto %d).\n", Protocol));
@ -429,7 +430,12 @@ NTSTATUS FileOpenAddress(
AddrFile->BCast = 1; AddrFile->BCast = 1;
AddrFile->HeaderIncl = 1; AddrFile->HeaderIncl = 1;
AddrFile->ProcessId = PsGetCurrentProcessId(); AddrFile->ProcessId = PsGetCurrentProcessId();
AddrFile->SubProcessTag = ((PTEB)PsGetCurrentThreadTeb())->SubProcessTag;
Teb = PsGetCurrentThreadTeb();
if (Teb != NULL) {
AddrFile->SubProcessTag = Teb->SubProcessTag;
}
KeQuerySystemTime(&AddrFile->CreationTime); KeQuerySystemTime(&AddrFile->CreationTime);
/* Make sure address is a local unicast address or 0 */ /* Make sure address is a local unicast address or 0 */