[ros-diffs] [reactos] 02/02: [NTOS:OB] Minor refactoring.

[Hermès Bélusca-Maïto]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=74513a75ab15608f2018f…

commit 74513a75ab15608f2018f52661e84e8435b72b98
Author:     Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org>
AuthorDate: Fri Sep 24 22:39:35 2021 +0200
Commit:     Hermès Bélusca-Maïto <hermes.belusca-maito(a)reactos.org>
CommitDate: Sat Sep 25 01:09:01 2021 +0200

    [NTOS:OB] Minor refactoring.
    
    - NtQuerySymbolicLinkObject(): Use an intermediate variable for the object header.
    - Simplify code in ObpLookupEntryDirectory() by calling
ObpReleaseLookupContextObject() instead.
    - Use TAG_OBJECT_TYPE instead of hardcoded tag values.
---
 ntoskrnl/ob/obdir.c  | 16 ++++------------
 ntoskrnl/ob/oblife.c |  4 ++--
 ntoskrnl/ob/oblink.c | 15 ++++++++++-----
 3 files changed, 16 insertions(+), 19 deletions(-)

diff --git a/ntoskrnl/ob/obdir.c b/ntoskrnl/ob/obdir.c
index cb7699ac840..3f3bc492c11 100644
--- a/ntoskrnl/ob/obdir.c
+++ b/ntoskrnl/ob/obdir.c
@@ -174,6 +174,7 @@ ObpLookupEntryDirectory(IN POBJECT_DIRECTORY Directory,
     PVOID FoundObject = NULL;
     PWSTR Buffer;
     POBJECT_DIRECTORY ShadowDirectory;
+
     PAGED_CODE();
 
     /* Check if we should search the shadow directory */
@@ -317,20 +318,11 @@ Quickie:
         }
     }
 
-    /* Check if we found an object already */
-    if (Context->Object)
-    {
-        /* We already did a lookup, so remove this object's query reference */
-        ObjectHeader = OBJECT_TO_OBJECT_HEADER(Context->Object);
-        HeaderNameInfo = OBJECT_HEADER_TO_NAME_INFO(ObjectHeader);
-        ObpDereferenceNameInfo(HeaderNameInfo);
-
-        /* Also dereference the object itself */
-        ObDereferenceObject(Context->Object);
-    }
+    /* Release any object previously looked up and replace it with the new one */
+    ObpReleaseLookupContextObject(Context);
+    Context->Object = FoundObject;
 
     /* Return the object we found */
-    Context->Object = FoundObject;
     return FoundObject;
 }
 
diff --git a/ntoskrnl/ob/oblife.c b/ntoskrnl/ob/oblife.c
index a01f5373bca..06c0d96e904 100644
--- a/ntoskrnl/ob/oblife.c
+++ b/ntoskrnl/ob/oblife.c
@@ -635,7 +635,7 @@ ObpAllocateObject(IN POBJECT_CREATE_INFORMATION ObjectCreateInfo,
     {
         /* Use default tag and non-paged pool */
         PoolType = NonPagedPool;
-        Tag = 'TjbO';
+        Tag = TAG_OBJECT_TYPE;
     }
     else
     {
@@ -1159,7 +1159,7 @@ ObCreateObjectType(IN PUNICODE_STRING TypeName,
 
         /* Set the hard-coded key and object count */
         LocalObjectType->TotalNumberOfObjects = 1;
-        LocalObjectType->Key = 'TjbO';
+        LocalObjectType->Key = TAG_OBJECT_TYPE;
     }
     else
     {
diff --git a/ntoskrnl/ob/oblink.c b/ntoskrnl/ob/oblink.c
index 7d03ba1bf96..47b813b802a 100644
--- a/ntoskrnl/ob/oblink.c
+++ b/ntoskrnl/ob/oblink.c
@@ -904,11 +904,13 @@ NtQuerySymbolicLinkObject(IN HANDLE LinkHandle,
                           OUT PUNICODE_STRING LinkTarget,
                           OUT PULONG ResultLength OPTIONAL)
 {
+    NTSTATUS Status;
     UNICODE_STRING SafeLinkTarget = { 0, 0, NULL };
     POBJECT_SYMBOLIC_LINK SymlinkObject;
-    KPROCESSOR_MODE PreviousMode = ExGetPreviousMode();
-    NTSTATUS Status;
+    POBJECT_HEADER ObjectHeader;
     ULONG LengthUsed;
+    KPROCESSOR_MODE PreviousMode = ExGetPreviousMode();
+
     PAGED_CODE();
 
     if (PreviousMode != KernelMode)
@@ -945,12 +947,15 @@ NtQuerySymbolicLinkObject(IN HANDLE LinkHandle,
                                        SYMBOLIC_LINK_QUERY,
                                        ObpSymbolicLinkObjectType,
                                        PreviousMode,
-                                       (PVOID *)&SymlinkObject,
+                                       (PVOID*)&SymlinkObject,
                                        NULL);
     if (NT_SUCCESS(Status))
     {
+        /* Get the object header */
+        ObjectHeader = OBJECT_TO_OBJECT_HEADER(SymlinkObject);
+
         /* Lock the object */
-        ObpAcquireObjectLock(OBJECT_TO_OBJECT_HEADER(SymlinkObject));
+        ObpAcquireObjectLock(ObjectHeader);
 
         /*
          * So here's the thing: If you specify a return length, then the
@@ -995,7 +1000,7 @@ NtQuerySymbolicLinkObject(IN HANDLE LinkHandle,
         _SEH2_END;
 
         /* Unlock and dereference the object */
-        ObpReleaseObjectLock(OBJECT_TO_OBJECT_HEADER(SymlinkObject));
+        ObpReleaseObjectLock(ObjectHeader);
         ObDereferenceObject(SymlinkObject);
     }