Windows 7下阻止系统关机

  1 #include <windows.h>
  2 #include <tchar.h>
  3 #include "resource.h"
  4 
  5 //标记是否已经阻止关机，默认为未阻止
  6 BOOL blockedFlag = FALSE;
  7 
  8 //这俩函数纯粹是撒娇，为了突出它们才拿出来改个名的
  9 BOOL BlockShutdown(HWND hwnd)
 10 {
 11     if (ShutdownBlockReasonCreate(hwnd, _T("不准关机！除非你点强制关机 :-(")))
 12     {
 13         return TRUE;
 14     }
 15     return FALSE;
 16 }
 17 
 18 BOOL UnblockShutdown(HWND hwnd)
 19 {
 20     if (ShutdownBlockReasonDestroy(hwnd))
 21     {
 22         return TRUE;
 23     }
 24     return FALSE;
 25 }
 26 
 27 INT_PTR CALLBACK MainDialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
 28 {
 29     switch(uMsg)
 30     {
 31     case WM_INITDIALOG:
 32         SetDlgItemText(hwndDlg, IDC_STATIC_STATUS, _T("未阻止关机"));
 33         return TRUE;
 34     case WM_CLOSE:
 35         if (blockedFlag)
 36         {
 37             if (UnblockShutdown(hwndDlg))
 38             {
 39                 EndDialog(hwndDlg, 0);
 40                 return TRUE;
 41             }
 42             if (IDYES == MessageBox(hwndDlg, _T("解除阻止失败……还要继续关闭程序么（关闭程序后不影响关机）？"), _T("提示"), MB_YESNO))
 43             {
 44                 EndDialog(hwndDlg, 0);
 45                 return TRUE;
 46             }
 47             return FALSE;
 48         }
 49         EndDialog(hwndDlg, 0);
 50         return TRUE;
 51     case WM_QUERYENDSESSION:
 52         if (blockedFlag)        //不要以为只调用BlockShutdownReasonCreate就行了，WM_QUERYENDSESSION也得拦住
 53         {
 54             return TRUE;
 55         }
 56         return FALSE;
 57     case WM_COMMAND:
 58         switch(LOWORD(wParam))
 59         {
 60         case IDC_BUTTON_BLOCK:
 61             if (blockedFlag)
 62             {
 63                 MessageBox(hwndDlg, _T("已经阻止关机，请点击关机测试。"), _T("提示"), MB_OK);
 64             }
 65             else
 66             {
 67                 if (BlockShutdown(hwndDlg))
 68                 {
 69                     SetDlgItemText(hwndDlg, IDC_STATIC_STATUS, _T("已经阻止关机"));
 70                     blockedFlag = TRUE;
 71                 }
 72                 else
 73                 {
 74                     MessageBox(hwndDlg, _T("阻止关机失败了……"), _T("提示"), MB_OK);
 75                 }
 76             }
 77             return TRUE;
 78         case IDC_BUTTON_UNBLOCK:
 79             if (!blockedFlag)
 80             {
 81                 MessageBox(hwndDlg, _T("未阻止关机，无需解除。"), _T("提示"), MB_OK);
 82             }
 83             else
 84             {
 85                 if (UnblockShutdown(hwndDlg))
 86                 {
 87                     SetDlgItemText(hwndDlg, IDC_STATIC_STATUS, _T("未阻止关机"));
 88                     blockedFlag = FALSE;
 89                 }
 90                 else
 91                 {
 92                     MessageBox(hwndDlg, _T("解除阻止失败了……"), _T("提示"), MB_OK);
 93                 }
 94             }
 95             return TRUE;
 96         default:
 97             return FALSE;
 98         }
 99     default:
100         return FALSE;
101     }
102     return FALSE;
103 }
104 int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd)
105 {
106     DialogBox(hInstance, MAKEINTRESOURCE(IDD_MAINDIALOG), NULL, MainDialogProc);
107     return 0;
108 }