[KMTEST] Add a NULL check

This commit is contained in:
Timo Kreuzer 2022-11-28 09:56:03 +02:00
parent 9e12af569f
commit 52fe43e87b

View file

@ -459,6 +459,11 @@ TestIrpHandler(
ok(Irp->AssociatedIrp.SystemBuffer == NULL, "A SystemBuffer was allocated!\n"); ok(Irp->AssociatedIrp.SystemBuffer == NULL, "A SystemBuffer was allocated!\n");
OrigBuffer = Buffer = MapAndLockUserBuffer(Irp, Length); OrigBuffer = Buffer = MapAndLockUserBuffer(Irp, Length);
ok(Buffer != NULL, "Null pointer!\n"); ok(Buffer != NULL, "Null pointer!\n");
if (Buffer == NULL)
{
Status = STATUS_UNSUCCESSFUL;
goto Exit;
}
if (Offset.QuadPart < Fcb->Header.FileSize.QuadPart) if (Offset.QuadPart < Fcb->Header.FileSize.QuadPart)
{ {
@ -545,6 +550,7 @@ TestIrpHandler(
IoCompleteRequest(Irp, IO_NO_INCREMENT); IoCompleteRequest(Irp, IO_NO_INCREMENT);
} }
Exit:
FsRtlExitFileSystem(); FsRtlExitFileSystem();
return Status; return Status;