[PATCH v3 1/2] ntdll: Add arm64ec_get_module_metadata helper.

[Billy Laws]

From: Billy Laws

---
 dlls/ntdll/ntdll_misc.h     |  1 +
 dlls/ntdll/signal_arm64ec.c | 18 ++++++++++++++++++
 dlls/ntdll/unwind.c         | 12 ++----------
 3 files changed, 21 insertions(+), 10 deletions(-)

diff --git a/dlls/ntdll/ntdll_misc.h b/dlls/ntdll/ntdll_misc.h
index 22a3a6ced60..fb5fb311646 100644
--- a/dlls/ntdll/ntdll_misc.h
+++ b/dlls/ntdll/ntdll_misc.h
@@ -169,6 +169,7 @@ extern void heap_thread_detach(void);

extern NTSTATUS arm64ec_process_init( HMODULE module );
 extern NTSTATUS arm64ec_thread_init(void);
+extern IMAGE_ARM64EC_METADATA *arm64ec_get_module_metadata( HMODULE module );
 extern void arm64ec_update_hybrid_metadata( void *module, IMAGE_NT_HEADERS *nt,
                                             const IMAGE_ARM64EC_METADATA *metadata );
 extern void invoke_arm64ec_syscall(void);
diff --git a/dlls/ntdll/signal_arm64ec.c b/dlls/ntdll/signal_arm64ec.c
index fbd49fd26c7..3106d9f57a3 100644
--- a/dlls/ntdll/signal_arm64ec.c
+++ b/dlls/ntdll/signal_arm64ec.c
@@ -220,6 +220,24 @@ NTSTATUS arm64ec_thread_init(void)
 }

+/**********************************************************************
+ *           arm64ec_get_module_metadata
+ */
+IMAGE_ARM64EC_METADATA *arm64ec_get_module_metadata( HMODULE module )
+{
+    IMAGE_LOAD_CONFIG_DIRECTORY *cfg;
+    ULONG size;
+
+    if (!(cfg = RtlImageDirectoryEntryToData( module, TRUE,
+                                              IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG, &size )))
+        return NULL;
+
+    size = min( size, cfg->Size );
+    if (size <= offsetof( IMAGE_LOAD_CONFIG_DIRECTORY, CHPEMetadataPointer )) return NULL;
+    return (IMAGE_ARM64EC_METADATA *)cfg->CHPEMetadataPointer;
+}
+
+
 static void update_hybrid_pointer( void *module, const IMAGE_SECTION_HEADER *sec, UINT rva, void *ptr )
 {
     if (!rva) return;
diff --git a/dlls/ntdll/unwind.c b/dlls/ntdll/unwind.c
index 91b07a7cc8c..c11bd9ddb9c 100644
--- a/dlls/ntdll/unwind.c
+++ b/dlls/ntdll/unwind.c
@@ -267,16 +267,8 @@ PRUNTIME_FUNCTION WINAPI RtlLookupFunctionTable( ULONG_PTR pc, ULONG_PTR *base,
 #ifdef __arm64ec__
     if (RtlIsEcCode( pc ))
     {
-        IMAGE_LOAD_CONFIG_DIRECTORY *cfg;
-        IMAGE_ARM64EC_METADATA *metadata;
-        ULONG size;
-
-        if (!(cfg = RtlImageDirectoryEntryToData( module->DllBase, TRUE,
-                                                  IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG, &size )))
-            return NULL;
-        size = min( size, cfg->Size );
-        if (size <= offsetof( IMAGE_LOAD_CONFIG_DIRECTORY, CHPEMetadataPointer )) return NULL;
-        metadata = (IMAGE_ARM64EC_METADATA *)cfg->CHPEMetadataPointer;
+        IMAGE_ARM64EC_METADATA *metadata = arm64ec_get_module_metadata( module->DllBase );
+        if (!metadata) return NULL;
         *len = metadata->ExtraRFETableSize;
         return (RUNTIME_FUNCTION *)(*base + metadata->ExtraRFETable);
     }