Windows 7下阻止系统关机

　　好长时间不更新博客了，今天终于把毕业设计答辩完了，虽然没完全做到自己预先设定的目标，但是起码算是效果出来了。虽然很长时间不更博客，但是订阅的RSS倒也没扔下，坚持读一读，今天这篇文章就是根据msdn博客的一篇博文写成的，只是添加了一些代码，有愿意看英文的请移步到这里。

　　从Vista开始，想阻止系统关机就开始变麻烦了，不能只拦截WM_QUERYENDSESSION了，操作系统只给一个应用程序两秒钟的时间去保存自己的东西，两秒钟之后，不管做完了没有，Game Over！可是如果你正在刻录一张光盘呢？两秒钟是都不够的，所以Vista之后也给提供了一种方式阻止系统关机，就是ShutdownBlockReasonCreate函数，只要调用这个函数，第一个参数传递创建主窗口的句柄，第二个参数传递一个字符串，这个字符串说明阻止关机的原因，将会显示给用户。而且msdn提示说这个API函数只能由创建主窗口的线程调用，否则将返回ERROR_ACCESS_DENIED。而且完成后还得调用ShutdownBlockReasonDestroy函数使系统关机能够继续进行，参数只传递上面那个窗口的句柄就可以了。

　　下面这段代码就是完整实现，不能使用VS2005编译，因为自带的库还没有这俩函数，我是使用VS2010编译的，工程文件和可执行文件的下载地址在文章最下方。

  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 }

　　下图就是运行效果。

ShutdownBlocker

　　上面这个图应该很多人看到过，只是通常刚一显示就消失了，这样就不会消失了，直到程序调用了ShutdownBlockReasonDestroy。

　　还有一点需要注意的是，如果你的电脑安装了360安全卫士orQQ电脑管家等优化软件，通常都会一键优化给优化掉……你就看不到这个对话框了，程序直接被枪毙掉了，因为这样可以加快关机速度 :-(

　　完整的C语言工程和可执行程序文件都在压缩包里，为了方便没装VS2010的童鞋测试程序效果，msvcr100.dll我也放到release目录下了，下载地址是https://files.cnblogs.com/pianoid/ShutdownBlocker.rar。

posted @ 2012-06-20 20:04  pianoid  阅读(9447)  评论(0)  编辑  收藏  举报