win-capture: get-graphics-offsets support for UWP/AppContainer

losttech · Apr 24, 2023 · f428a41 · f428a41
1 parent a4bfe03
commit f428a41
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/plugins/win-capture/get-graphics-offsets/CMakeLists.txt b/plugins/win-capture/get-graphics-offsets/CMakeLists.txt
@@ -9,6 +9,8 @@ target_sources(
           dxgi-offsets.cpp
           d3d8-offsets.cpp
           d3d9-offsets.cpp
+          ../nt-stuff.c
+          ../nt-stuff.h
           ../graphics-hook-info.h
           ../hook-helpers.h)
 

diff --git a/plugins/win-capture/get-graphics-offsets/get-graphics-offsets.c b/plugins/win-capture/get-graphics-offsets/get-graphics-offsets.c
@@ -1,11 +1,20 @@
 #include <inttypes.h>
 #include <stdio.h>
+#include <stdlib.h>
 #include <windows.h>
 #include "get-graphics-offsets.h"
 
 struct hook_info *open_hook_info_shmem(const char *shmemName)
 {
-	HANDLE shmem = OpenFileMappingA(FILE_MAP_WRITE, false, shmemName);
+	HANDLE shmem;
+	if (shmemName[0] == '\\') {
+		wchar_t wideName[MAX_PATH + 1];
+		mbstowcs(wideName, shmemName, MAX_PATH);
+		shmem = nt_open_map(wideName);
+	} else {
+		shmem = OpenFileMappingA(FILE_MAP_WRITE, false, shmemName);
+	}
+
 	if (!shmem) {
 		return NULL;
 	}

diff --git a/plugins/win-capture/get-graphics-offsets/get-graphics-offsets.h b/plugins/win-capture/get-graphics-offsets/get-graphics-offsets.h
@@ -3,6 +3,7 @@
 #include <stdint.h>
 #include <stdbool.h>
 #include "../graphics-hook-info.h"
+#include "../nt-stuff.h"
 
 #define DUMMY_WNDCLASS "get_addrs_wndclass"