[ros-diffs] [fireball] 51208: [NTDLL] - Implement LdrEnumerateLoadedModules().

Author: fireball
Date: Wed Mar 30 21:21:42 2011
New Revision: 51208

URL: http://svn.reactos.org/svn/reactos?rev=51208&view=rev
Log:
[NTDLL]
- Implement LdrEnumerateLoadedModules().

Modified:
    trunk/reactos/dll/ntdll/def/ntdll.pspec
    trunk/reactos/dll/ntdll/ldr/ldrapi.c

Modified: trunk/reactos/dll/ntdll/def/ntdll.pspec
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/dll/ntdll/def/ntdll.pspec?…
==============================================================================
--- trunk/reactos/dll/ntdll/def/ntdll.pspec [iso-8859-1] (original)
+++ trunk/reactos/dll/ntdll/def/ntdll.pspec [iso-8859-1] Wed Mar 30 21:21:42 2011
@@ -77,7 +77,7 @@
 //@ stdcall LdrDestroyOutOfProcessImage
 @ stdcall LdrDisableThreadCalloutsForDll(long)
 @ stdcall LdrEnumResources(ptr ptr long ptr ptr)
-//@ stdcall LdrEnumerateLoadedModules
+@ stdcall LdrEnumerateLoadedModules(long ptr long)
 //@ stdcall LdrFindCreateProcessManifest // 5.1 and 5.2 only
 @ stdcall LdrFindEntryForAddress(ptr ptr)
 @ stdcall LdrFindResourceDirectory_U(long ptr long ptr)

Modified: trunk/reactos/dll/ntdll/ldr/ldrapi.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/dll/ntdll/ldr/ldrapi.c?rev…
==============================================================================
--- trunk/reactos/dll/ntdll/ldr/ldrapi.c [iso-8859-1] (original)
+++ trunk/reactos/dll/ntdll/ldr/ldrapi.c [iso-8859-1] Wed Mar 30 21:21:42 2011
@@ -530,4 +530,66 @@
     return LdrQueryProcessModuleInformationEx(0, 0, ModuleInformation, Size,
ReturnedSize);
 }
 
+NTSTATUS
+NTAPI
+LdrEnumerateLoadedModules(BOOLEAN ReservedFlag, PLDR_ENUM_CALLBACK EnumProc, ULONG
Context)
+{
+    PLIST_ENTRY ListHead, ListEntry;
+    PLDR_DATA_TABLE_ENTRY LdrEntry;
+    NTSTATUS Status;
+    ULONG Cookie;
+    BOOLEAN Stop = FALSE;
+
+    /* Check parameters */
+    if (ReservedFlag || !EnumProc) return STATUS_INVALID_PARAMETER;
+
+    /* Acquire the loader lock */
+    Status = LdrLockLoaderLock(0, NULL, &Cookie);
+    if (!NT_SUCCESS(Status)) return Status;
+
+    /* Loop all the modules and call enum proc */
+    ListHead = &NtCurrentPeb()->Ldr->InLoadOrderModuleList;
+    ListEntry = ListHead->Flink;
+    while (ListHead != ListEntry)
+    {
+        /* Get the entry */
+        LdrEntry = CONTAINING_RECORD(ListEntry, LDR_DATA_TABLE_ENTRY, InLoadOrderLinks);
+
+        /* Call the enumeration proc inside SEH */
+        _SEH2_TRY
+        {
+            EnumProc(LdrEntry, Context, &Stop);
+        }
+        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+        {
+            /* Ignoring the exception */
+        } _SEH2_END;
+
+        /* Break if we were asked to stop enumeration */
+        if (Stop)
+        {
+            /* Release loader lock */
+            Status = LdrUnlockLoaderLock(0, Cookie);
+
+            /* Reset any successful status to STATUS_SUCCESS, but leave
+               failure to the caller */
+            if (NT_SUCCESS(Status))
+                Status = STATUS_SUCCESS;
+
+            /* Return any possible failure status */
+            return Status;
+        }
+
+        /* Advance to the next module */
+        ListEntry = ListEntry->Flink;
+    }
+
+    /* Release loader lock, it must succeed this time */
+    Status = LdrUnlockLoaderLock(0, Cookie);
+    ASSERT(NT_SUCCESS(Status));
+
+    /* Return success */
+    return STATUS_SUCCESS;
+}
+
 /* EOF */