[KMTESTS:CC] Add missing FsRtlEnter/ExitFileSystem() calls

Spotted by Thomas
This commit is contained in:
Pierre Schweitzer 2018-09-01 12:18:51 +02:00
parent 8f284f8a0c
commit b0d1522886
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B

View file

@ -331,6 +331,8 @@ TestMessageHandler(
{ {
NTSTATUS Status = STATUS_SUCCESS; NTSTATUS Status = STATUS_SUCCESS;
FsRtlEnterFileSystem();
switch (ControlCode) switch (ControlCode)
{ {
case IOCTL_START_TEST: case IOCTL_START_TEST:
@ -348,6 +350,8 @@ TestMessageHandler(
break; break;
} }
FsRtlExitFileSystem();
return Status; return Status;
} }
@ -365,6 +369,8 @@ TestIrpHandler(
DPRINT("IRP %x/%x\n", IoStack->MajorFunction, IoStack->MinorFunction); DPRINT("IRP %x/%x\n", IoStack->MajorFunction, IoStack->MinorFunction);
ASSERT(IoStack->MajorFunction == IRP_MJ_READ); ASSERT(IoStack->MajorFunction == IRP_MJ_READ);
FsRtlEnterFileSystem();
Status = STATUS_NOT_SUPPORTED; Status = STATUS_NOT_SUPPORTED;
Irp->IoStatus.Information = 0; Irp->IoStatus.Information = 0;
@ -420,5 +426,7 @@ TestIrpHandler(
IoCompleteRequest(Irp, IO_NO_INCREMENT); IoCompleteRequest(Irp, IO_NO_INCREMENT);
} }
FsRtlExitFileSystem();
return Status; return Status;
} }