Changeset 398

Timestamp:

9/26/2008 4:57:59 AM (5 years ago)

Author:

lowjoel

Message:

Alert the user is the current process is running at a higher privilege level than the previous process. Should fix many elevation issues. Fixes #48

File:

• trunk/Eraser.cpp (modified) (2 diffs)

trunk/Eraser.cpp

@@ -76 +76 @@
 // registered so that we can unregister it in ExitInstance
 static BOOL bClassRegistered = FALSE;
+__declspec(dllimport) bool IsProcessElevated(HANDLE process);
 
 BOOL CEraserApp::FirstInstance()
 {
-    CWnd *pWndPrev = CWnd::FindWindow(szEraserClassName, NULL);
+    CWnd *pWndPrev = CWnd::FindWindow(szEraserClassName, NULL);
     CWnd *pWndChild;
 
@@ -85 +86 @@
     if (pWndPrev)
     {
+        // Determine the elevation status of both processes.
+        unsigned elevationStatus = 0;
+        {
+            DWORD pid = 0;
+            GetWindowThreadProcessId(pWndPrev->m_hWnd, &pid);
+            HANDLE process = OpenProcess(PROCESS_QUERY_INFORMATION, false, pid);
+            if (IsProcessElevated(process))
+                elevationStatus |= 2;
+            if (IsProcessElevated(GetCurrentProcess()))
+                elevationStatus |= 1;
+        }
+
+        if (!(elevationStatus & 2) && (elevationStatus & 1))
+        {
+            switch (AfxMessageBox("Another instance of Eraser is still running, but this "
+                "new instance is running at a higher privilege level than the other.\n\n"
+                "Do you want to close the previous instance? "
+                "All active erasing tasks will be terminated", MB_YESNO))
+            {
+            case IDYES:
+                pWndPrev->SendMessage(WM_CLOSE);
+                break;
+            }
+
+            return TRUE;
+        }
+
         // if so, does it have any popups?
         pWndChild = pWndPrev->GetLastActivePopup();