[reactos] 102/179: [WINESYNC] dbghelp: Make dll builtin PE path search helper more generic.

[winesync]

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

commit e01d28abd821f1bdac44479581046e280736057c
Author:     winesync <ros-dev(a)reactos.org>
AuthorDate: Fri Sep 11 17:10:41 2020 +0200
Commit:     Jérôme Gardou <jerome.gardou(a)reactos.org>
CommitDate: Wed Sep 16 10:35:47 2020 +0200

    [WINESYNC] dbghelp: Make dll builtin PE path search helper more generic.
    
    Signed-off-by: Jacek Caban <jacek(a)codeweavers.com>
    Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>
    
    wine commit id 21af2e194792aaa263c144c8fb42fe678ad2ecd7 by Jacek Caban
<jacek(a)codeweavers.com>
---
 dll/win32/dbghelp/dbghelp_private.h |  1 +
 dll/win32/dbghelp/path.c            | 88 +++++++++++++++++++++++++++++++++++
 dll/win32/dbghelp/pe_module.c       | 91 ++++++++-----------------------------
 sdk/tools/winesync/dbghelp.cfg      |  2 +-
 4 files changed, 110 insertions(+), 72 deletions(-)

diff --git a/dll/win32/dbghelp/dbghelp_private.h b/dll/win32/dbghelp/dbghelp_private.h
index 661a939eec8..ac528c5dbe5 100644
--- a/dll/win32/dbghelp/dbghelp_private.h
+++ b/dll/win32/dbghelp/dbghelp_private.h
@@ -703,6 +703,7 @@ extern BOOL         path_find_symbol_file(const struct process* pcs,
const struc
                                           PCSTR full_path, const GUID* guid, DWORD dw1,
DWORD dw2,
                                           WCHAR *buffer, BOOL* is_unmatched)
DECLSPEC_HIDDEN;
 extern WCHAR *get_dos_file_name(const WCHAR *filename) DECLSPEC_HIDDEN;
+extern BOOL search_dll_path(const WCHAR *name, BOOL (*match)(void*, HANDLE, const
WCHAR*), void *param) DECLSPEC_HIDDEN;
 extern const WCHAR* file_name(const WCHAR* str) DECLSPEC_HIDDEN;
 extern const char* file_nameA(const char* str) DECLSPEC_HIDDEN;
 
diff --git a/dll/win32/dbghelp/path.c b/dll/win32/dbghelp/path.c
index 1b04d97e136..e10318d35c6 100644
--- a/dll/win32/dbghelp/path.c
+++ b/dll/win32/dbghelp/path.c
@@ -720,3 +720,91 @@ WCHAR *get_dos_file_name(const WCHAR *filename)
     }
     return dos_path;
 }
+
+#ifndef __REACTOS__
+BOOL search_dll_path(const WCHAR *name, BOOL (*match)(void*, HANDLE, const WCHAR*), void
*param)
+{
+    size_t len, i;
+    HANDLE file;
+    WCHAR *buf;
+    BOOL ret;
+
+    static const WCHAR winebuilddirW[] =
{'W','I','N','E','B','U','I','L','D','D','I','R',0};
+    static const WCHAR winedlldirW[] =
{'W','I','N','E','D','L','L','D','I','R','%','u',0};
+
+    name = file_name(name);
+
+    if ((len = GetEnvironmentVariableW(winebuilddirW, NULL, 0)))
+    {
+        WCHAR *p, *end;
+        const WCHAR dllsW[] = {
'\\','d','l','l','s','\\' };
+        const WCHAR programsW[] = {
'\\','p','r','o','g','r','a','m','s','\\'
};
+        const WCHAR dot_dllW[] = {'.','d','l','l',0};
+        const WCHAR dot_exeW[] = {'.','e','x','e',0};
+        const WCHAR dot_soW[] = {'.','s','o',0};
+
+        if (!(buf = heap_alloc((len + 8 + 3 * lstrlenW(name)) * sizeof(WCHAR)))) return
FALSE;
+        end = buf + GetEnvironmentVariableW(winebuilddirW, buf, len);
+
+        memcpy(end, dllsW, sizeof(dllsW));
+        strcpyW(end + ARRAY_SIZE(dllsW), name);
+        if ((p = strrchrW(end, '.')) && !lstrcmpW(p, dot_soW)) *p = 0;
+        if ((p = strrchrW(end, '.')) && !lstrcmpW(p, dot_dllW)) *p = 0;
+        p = end + strlenW(end);
+        *p++ = '\\';
+        strcpyW(p, name);
+        file = CreateFileW(buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, NULL);
+        if (file != INVALID_HANDLE_VALUE)
+        {
+            ret = match(param, file, buf);
+            CloseHandle(file);
+            if (ret) goto found;
+        }
+
+        memcpy(end, programsW, sizeof(programsW));
+        end += ARRAY_SIZE(programsW);
+        strcpyW(end, name);
+        if ((p = strrchrW(end, '.')) && !lstrcmpW(p, dot_soW)) *p = 0;
+        if ((p = strrchrW(end, '.')) && !lstrcmpW(p, dot_exeW)) *p = 0;
+        p = end + strlenW(end);
+        *p++ = '\\';
+        strcpyW(p, name);
+        file = CreateFileW(buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, NULL);
+        if (file != INVALID_HANDLE_VALUE)
+        {
+            ret = match(param, file, buf);
+            CloseHandle(file);
+            if (ret) goto found;
+        }
+
+        heap_free(buf);
+    }
+
+    for (i = 0;; i++)
+    {
+        WCHAR env_name[64];
+        sprintfW(env_name, winedlldirW, i);
+        if (!(len = GetEnvironmentVariableW(env_name, NULL, 0))) break;
+        if (!(buf = heap_alloc((len + lstrlenW(name) + 2) * sizeof(WCHAR)))) return
FALSE;
+
+        len = GetEnvironmentVariableW(env_name, buf, len);
+        buf[len++] = '\\';
+        strcpyW(buf + len, name);
+        file = CreateFileW(buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, NULL);
+        if (file != INVALID_HANDLE_VALUE)
+        {
+            ret = match(param, file, buf);
+            CloseHandle(file);
+            if (ret) goto found;
+        }
+        heap_free(buf);
+    }
+
+    return FALSE;
+
+found:
+    TRACE("found %s\n", debugstr_w(buf));
+    heap_free(buf);
+    return TRUE;
+}
+#endif
diff --git a/dll/win32/dbghelp/pe_module.c b/dll/win32/dbghelp/pe_module.c
index 0eb97e9e0c7..62328bb8d49 100644
--- a/dll/win32/dbghelp/pe_module.c
+++ b/dll/win32/dbghelp/pe_module.c
@@ -775,68 +775,23 @@ BOOL pe_load_debug_info(const struct process* pcs, struct module*
module)
 }
 
 #ifndef __REACTOS__
-static WCHAR *find_builtin_pe(const WCHAR *path, HANDLE *file)
+struct builtin_search
 {
-    const WCHAR *base_name;
-    size_t len, i;
-    WCHAR *buf;
+    WCHAR *path;
+    struct image_file_map fmap;
+};
 
-    static const WCHAR winebuilddirW[] =
{'W','I','N','E','B','U','I','L','D','D','I','R',0};
-    static const WCHAR winedlldirW[] =
{'W','I','N','E','D','L','L','D','I','R','%','u',0};
+static BOOL search_builtin_pe(void *param, HANDLE handle, const WCHAR *path)
+{
+    struct builtin_search *search = param;
+    size_t size;
 
-    if ((base_name = strrchrW(path, '\\'))) base_name++;
-    else base_name = path;
+    if (!pe_map_file(handle, &search->fmap, DMT_PE)) return FALSE;
 
-    if ((len = GetEnvironmentVariableW(winebuilddirW, NULL, 0)))
-    {
-        WCHAR *p, *end;
-        const WCHAR dllsW[] = {
'\\','d','l','l','s','\\' };
-        const WCHAR programsW[] = {
'\\','p','r','o','g','r','a','m','s','\\'
};
-        const WCHAR dot_dllW[] = {'.','d','l','l',0};
-        const WCHAR dot_exeW[] = {'.','e','x','e',0};
-
-        if (!(buf = heap_alloc((len + 8 + 2 * lstrlenW(base_name)) * sizeof(WCHAR))))
return NULL;
-        end = buf + GetEnvironmentVariableW(winebuilddirW, buf, len);
-
-        memcpy(end, dllsW, sizeof(dllsW));
-        strcpyW(end + ARRAY_SIZE(dllsW), base_name);
-        if ((p = strchrW(end, '.')) && !lstrcmpW(p, dot_dllW)) *p = 0;
-        p = end + strlenW(end);
-        *p++ = '\\';
-        strcpyW(p, base_name);
-        *file = CreateFileW(buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, NULL);
-        if (*file != INVALID_HANDLE_VALUE) return buf;
-
-        memcpy(end, programsW, sizeof(programsW));
-        end += ARRAY_SIZE(programsW);
-        strcpyW(end, base_name);
-        if ((p = strchrW(end, '.')) && !lstrcmpW(p, dot_exeW)) *p = 0;
-        p = end + strlenW(end);
-        *p++ = '\\';
-        strcpyW(p, base_name);
-        *file = CreateFileW(buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, NULL);
-        if (*file != INVALID_HANDLE_VALUE) return buf;
-
-        heap_free(buf);
-    }
-
-    for (i = 0;; i++)
-    {
-        WCHAR name[64];
-        sprintfW(name, winedlldirW, i);
-        if (!(len = GetEnvironmentVariableW(name, NULL, 0))) break;
-        if (!(buf = heap_alloc((len + lstrlenW(base_name) + 2) * sizeof(WCHAR)))) return
NULL;
-
-        GetEnvironmentVariableW(name, buf, len);
-        buf[len++] = '\\';
-        strcpyW(buf + len, base_name);
-        *file = CreateFileW(buf, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL, NULL);
-        if (*file != INVALID_HANDLE_VALUE) return buf;
-
-        heap_free(buf);
-    }
-
-    return NULL;
+    size = (lstrlenW(path) + 1) * sizeof(WCHAR);
+    if ((search->path = heap_alloc(size)))
+        memcpy(search->path, path, size);
+    return TRUE;
 }
 #endif
 
@@ -870,18 +825,12 @@ struct module* pe_load_native_module(struct process* pcs, const
WCHAR* name,
     if (pe_map_file(hFile, &modfmt->u.pe_info->fmap, DMT_PE))
     {
 #ifndef __REACTOS__
-        WCHAR *builtin_path = NULL;
-        HANDLE builtin_module;
-        if (modfmt->u.pe_info->fmap.u.pe.builtin && (builtin_path =
find_builtin_pe(loaded_name, &builtin_module)))
+        struct builtin_search builtin = { NULL };
+        if (modfmt->u.pe_info->fmap.u.pe.builtin &&
search_dll_path(loaded_name, search_builtin_pe, &builtin))
         {
-            struct image_file_map builtin_fmap;
-            if (pe_map_file(builtin_module, &builtin_fmap, DMT_PE))
-            {
-                TRACE("reloaded %s from %s\n", debugstr_w(loaded_name),
debugstr_w(builtin_path));
-                image_unmap_file(&modfmt->u.pe_info->fmap);
-                modfmt->u.pe_info->fmap = builtin_fmap;
-            }
-            CloseHandle(builtin_module);
+            TRACE("reloaded %s from %s\n", debugstr_w(loaded_name),
debugstr_w(builtin.path));
+            image_unmap_file(&modfmt->u.pe_info->fmap);
+            modfmt->u.pe_info->fmap = builtin.fmap;
         }
 #endif
         if (!base) base =
modfmt->u.pe_info->fmap.u.pe.ntheader.OptionalHeader.ImageBase;
@@ -895,7 +844,7 @@ struct module* pe_load_native_module(struct process* pcs, const WCHAR*
name,
 #ifdef __REACTOS__
             module->real_path = NULL;
 #else
-            module->real_path = builtin_path;
+            module->real_path = builtin.path;
 #endif
             modfmt->module = module;
             modfmt->remove = pe_module_remove;
@@ -912,7 +861,7 @@ struct module* pe_load_native_module(struct process* pcs, const WCHAR*
name,
         {
             ERR("could not load the module '%s'\n",
debugstr_w(loaded_name));
 #ifndef __REACTOS__
-            heap_free(module->real_path);
+            heap_free(builtin.path);
 #endif
             image_unmap_file(&modfmt->u.pe_info->fmap);
         }
diff --git a/sdk/tools/winesync/dbghelp.cfg b/sdk/tools/winesync/dbghelp.cfg
index 3ef3ca41a9b..d0e56271d9c 100644
--- a/sdk/tools/winesync/dbghelp.cfg
+++ b/sdk/tools/winesync/dbghelp.cfg
@@ -4,4 +4,4 @@ files:
   include/dbghelp.h: sdk/include/psdk/dbghelp.h
   include/wine/mscvpdb.h: sdk/include/reactos/wine/mscvpdb.h
 tags:
-  wine: 77e880e6d22ef9be031f783fbb4b6a5e8c8826e8
+  wine: 21af2e194792aaa263c144c8fb42fe678ad2ecd7