Merge pull request #3 from love-code-yeyixiao/FixAndCheckAutoDelete

Fix and add two feature
sandboxie-plus · Mar 2, 2024 · b264e37 · b264e37
2 parents e45246e + 44d9792
commit b264e37
Show file tree

Hide file tree

Showing 6 changed files with 332 additions and 279 deletions.
diff --git a/Sandboxie/core/dll/gui.c b/Sandboxie/core/dll/gui.c
@@ -1629,6 +1629,8 @@ _FX LRESULT Gui_WindowProcA(
 
     if (uMsg == WM_SETTEXT && Gui_ShouldCreateTitle(hWnd))
         new_lParam = (LPARAM)Gui_CreateTitleA((UCHAR *)lParam);
+	if (uMsg == WM_CREATE)
+		ProtectScreen(hWnd);
     else
         new_lParam = lParam;
 
@@ -2702,3 +2704,26 @@ _FX BOOLEAN ComDlg32_Init(HMODULE module)
 
     return TRUE;
 }
+
+_FX VOID ProtectScreen(HWND hWnd) {
+	WCHAR buffer[BOXNAME_COUNT] = { L"\0" };
+	SbieApi_QueryProcess(GetCurrentProcess(), buffer, 0, 0, 0);
+	if (SbieApi_QueryConfBool(buffer, L"IsProtectScreen", FALSE)) {
+		OSVERSIONINFOEX os;
+		os.dwOSVersionInfoSize = sizeof(os);
+		if (GetVersionEx(&os))
+		{
+			if (os.dwMajorVersion == 6) {
+				HMODULE hDll = GetModuleHandleW(L"user32.dll");
+				if (hDll != NULL && hDll != INVALID_HANDLE_VALUE) {
+					typedef BOOL(*LPSETWINDOWDISPLAYAFFINITY)(HWND, DWORD);
+					LPSETWINDOWDISPLAYAFFINITY swda = GetProcAddress(hDll, "SetWindowDisplayAffinity");
+					if (swda) {
+						swda(hWnd, 0x00000001);
+					}
+
+				}
+			}
+		}
+	}
+}