Davide Beatrici : mmdevapi: Query MemoryWineUnixFuncs virtual memory and store the resulting handle.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 0888d2242df7a5fb310cd9473465fbfe3374a615
URL:    https://gitlab.winehq.org/wine/wine/-/commit/0888d2242df7a5fb310cd9473465fbf...

Author: Davide Beatrici <git(a)davidebeatrici.dev>
Date:   Sat Mar  4 02:31:53 2023 +0100

mmdevapi: Query MemoryWineUnixFuncs virtual memory and store the resulting handle.

In preparation for calling the audio driver's UNIX functions directly.

---

 dlls/mmdevapi/main.c     | 12 +++++++++++-
 dlls/mmdevapi/mmdevapi.h |  4 ++++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/dlls/mmdevapi/main.c b/dlls/mmdevapi/main.c
index c792661af33..b019d2e427c 100644
--- a/dlls/mmdevapi/main.c
+++ b/dlls/mmdevapi/main.c
@@ -65,6 +65,7 @@ static const char *get_priority_string(int prio)
 
 static BOOL load_driver(const WCHAR *name, DriverFuncs *driver)
 {
+    NTSTATUS status;
     WCHAR driver_module[264];
 
     lstrcpyW(driver_module, L"wine");
@@ -80,8 +81,14 @@ static BOOL load_driver(const WCHAR *name, DriverFuncs *driver)
         return FALSE;
     }
 
+    if ((status = NtQueryVirtualMemory(GetCurrentProcess(), driver->module, MemoryWineUnixFuncs,
+        &driver->module_unixlib, sizeof(driver->module_unixlib), NULL))) {
+        ERR("Unable to load UNIX functions: %lx\n", status);
+        goto fail;
+    }
+
 #define LDFC(n) do { driver->p##n = (void*)GetProcAddress(driver->module, #n);\
-        if(!driver->p##n) { FreeLibrary(driver->module); return FALSE; } } while(0)
+        if(!driver->p##n) { goto fail; } } while(0)
     LDFC(GetPriority);
     LDFC(GetEndpointIDs);
     LDFC(GetAudioEndpoint);
@@ -98,6 +105,9 @@ static BOOL load_driver(const WCHAR *name, DriverFuncs *driver)
             wine_dbgstr_w(driver_module), get_priority_string(driver->priority));
 
     return TRUE;
+fail:
+    FreeLibrary(driver->module);
+    return FALSE;
 }
 
 static BOOL WINAPI init_driver(INIT_ONCE *once, void *param, void **context)
diff --git a/dlls/mmdevapi/mmdevapi.h b/dlls/mmdevapi/mmdevapi.h
index d694a3930aa..16eb3606fea 100644
--- a/dlls/mmdevapi/mmdevapi.h
+++ b/dlls/mmdevapi/mmdevapi.h
@@ -18,13 +18,17 @@
 
 #include "unixlib.h"
 
+#include <winternl.h>
+
 #include <wine/list.h>
+#include <wine/unixlib.h>
 
 extern HRESULT MMDevEnum_Create(REFIID riid, void **ppv) DECLSPEC_HIDDEN;
 extern void MMDevEnum_Free(void) DECLSPEC_HIDDEN;
 
 typedef struct _DriverFuncs {
     HMODULE module;
+    unixlib_handle_t module_unixlib;
     WCHAR module_name[64];
     int priority;