refactor getting process handle of window

eriforce · Aug 27, 2024 · ed5e418 · ed5e418
1 parent 452ff29
commit ed5e418
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 13 deletions.
diff --git a/src/Magpie.App/ScalingService.cpp b/src/Magpie.App/ScalingService.cpp
@@ -330,16 +330,8 @@ void ScalingService::_ScaleForegroundWindow() {
 }
 
 static bool GetWindowIntegrityLevel(HWND hWnd, DWORD& integrityLevel) noexcept {
-	DWORD processId;
-	if (!GetWindowThreadProcessId(hWnd, &processId)) {
-		Logger::Get().Win32Error("GetWindowThreadProcessId 失败");
-		return false;
-	}
-
-	wil::unique_process_handle hProc(
-		OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, processId));
+	wil::unique_process_handle hProc = Win32Utils::GetWndProcessHandle(hWnd);
 	if (!hProc) {
-		Logger::Get().Win32Error("OpenProcess 失败");
 		return false;
 	}
 

diff --git a/src/Shared/Win32Utils.cpp b/src/Shared/Win32Utils.cpp
@@ -34,7 +34,7 @@ std::wstring Win32Utils::GetWndTitle(HWND hWnd) noexcept {
 	return title;
 }
 
-std::wstring Win32Utils::GetPathOfWnd(HWND hWnd) noexcept {
+wil::unique_process_handle Win32Utils::GetWndProcessHandle(HWND hWnd) noexcept {
 	wil::unique_process_handle hProc;
 
 	if (DWORD dwProcId = 0; GetWindowThreadProcessId(hWnd, &dwProcId)) {
@@ -58,10 +58,15 @@ std::wstring Win32Utils::GetPathOfWnd(HWND hWnd) noexcept {
 				Logger::Get().Win32Error("GetProcessHandleFromHwnd 失败");
 			}
 		}
+	}
 
-		if (!hProc) {
-			return {};
-		}
+	return hProc;
+}
+
+std::wstring Win32Utils::GetPathOfWnd(HWND hWnd) noexcept {
+	wil::unique_process_handle hProc = ::GetWndProcessHandle(hWnd);
+	if (!hProc) {
+		return {};
 	}
 
 	std::wstring fileName;

diff --git a/src/Shared/Win32Utils.h b/src/Shared/Win32Utils.h
@@ -14,6 +14,8 @@ struct Win32Utils {
 
 	static std::wstring GetWndTitle(HWND hWnd) noexcept;
 
+	static wil::unique_process_handle Win32Utils::GetWndProcessHandle(HWND hWnd) noexcept;
+
 	static std::wstring GetPathOfWnd(HWND hWnd) noexcept;
 
 	static UINT GetWindowShowCmd(HWND hWnd) noexcept;