[reactos] 21/26: [NTOS:FSRTL] Check return status of FsRtl locking functions

[Jérôme Gardou]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=46af941c4cc8a922559b10...

commit 46af941c4cc8a922559b1038813878793b26dc6e
Author:     Jérôme Gardou [e-mail]
AuthorDate: Fri Jun 18 10:01:59 2021 +0200
Commit:     Jérôme Gardou [e-mail]
CommitDate: Mon Jun 28 10:20:57 2021 +0200

[NTOS:FSRTL] Check return status of FsRtl locking functions

CORE-17637
---
 ntoskrnl/fsrtl/filelock.c | 42 +++++++++++++++++++++++++-----------------
 1 file changed, 25 insertions(+), 17 deletions(-)

diff --git a/ntoskrnl/fsrtl/filelock.c b/ntoskrnl/fsrtl/filelock.c
index f92f9397c96..624d2dd7047 100644
--- a/ntoskrnl/fsrtl/filelock.c
+++ b/ntoskrnl/fsrtl/filelock.c
@@ -994,6 +994,7 @@ FsRtlFastUnlockSingle(IN PFILE_LOCK FileLock,
     InternalInfo->Generation++;
     while ((NextMatchingLockIrp = IoCsqRemoveNextIrp(&InternalInfo->Csq, &Find)))
     {
+        NTSTATUS Status;
         if (NextMatchingLockIrp->IoStatus.Information == InternalInfo->Generation)
         {
             // We've already looked at this one, meaning that we looped.
@@ -1011,7 +1012,9 @@ FsRtlFastUnlockSingle(IN PFILE_LOCK FileLock,
         // because somebody else snatched part of the range in a new thread.
         DPRINT("Locking another IRP %p for %p %wZ\n",
                NextMatchingLockIrp, FileLock, &FileObject->FileName);
-        FsRtlProcessFileLock(InternalInfo->BelongsTo, NextMatchingLockIrp, NULL);
+        Status = FsRtlProcessFileLock(InternalInfo->BelongsTo, NextMatchingLockIrp, NULL);
+        if (!NT_SUCCESS(Status))
+            return Status;
     }

DPRINT("Success %wZ\n", &FileObject->FileName);
@@ -1173,23 +1176,25 @@ FsRtlProcessFileLock(IN PFILE_LOCK FileLock,
     {
         /* A lock */
     case IRP_MN_LOCK:
-
+    {
         /* Call the private lock routine */
-        FsRtlPrivateLock(FileLock,
-                         IoStackLocation->FileObject,
-                         &IoStackLocation->
-                         Parameters.LockControl.ByteOffset,
-                         IoStackLocation->Parameters.LockControl.Length,
-                         IoGetRequestorProcess(Irp),
-                         IoStackLocation->Parameters.LockControl.Key,
-                         IoStackLocation->Flags & SL_FAIL_IMMEDIATELY,
-                         IoStackLocation->Flags & SL_EXCLUSIVE_LOCK,
-                         &IoStatusBlock,
-                         Irp,
-                         Context,
-                         FALSE);
+        BOOLEAN Result = FsRtlPrivateLock(FileLock,
+                                          IoStackLocation->FileObject,
+                                          &IoStackLocation->Parameters.LockControl.ByteOffset,
+                                          IoStackLocation->Parameters.LockControl.Length,
+                                          IoGetRequestorProcess(Irp),
+                                          IoStackLocation->Parameters.LockControl.Key,
+                                          IoStackLocation->Flags & SL_FAIL_IMMEDIATELY,
+                                          IoStackLocation->Flags & SL_EXCLUSIVE_LOCK,
+                                          &IoStatusBlock,
+                                          Irp,
+                                          Context,
+                                          FALSE);
+        /* FsRtlPrivateLock has _Must_inspect_result_. Just check this is consistent on debug builds */
+        NT_ASSERT(Result == NT_SUCCESS(IoStatusBlock.Status));
+        (void)Result;
         return IoStatusBlock.Status;
-
+    }
         /* A single unlock */
     case IRP_MN_UNLOCK_SINGLE:

@@ -1299,7 +1304,10 @@ FsRtlUninitializeFileLock(IN PFILE_LOCK FileLock)
         }
         while ((Irp = IoCsqRemoveNextIrp(&InternalInfo->Csq, NULL)) != NULL)
         {
-            FsRtlProcessFileLock(FileLock, Irp, NULL);
+            NTSTATUS Status = FsRtlProcessFileLock(FileLock, Irp, NULL);
+            /* FsRtlProcessFileLock has _Must_inspect_result_ */
+            NT_ASSERT(NT_SUCCESS(Status));
+            (void)Status;
         }
         ExFreePoolWithTag(InternalInfo, TAG_FLOCK);
         FileLock->LockInformation = NULL;