C/C++控制Windows关机/注销/重启的正确姿势

简介

说到代码控制Windows关机/注销/重启的方式，有很多种，最简单的不过就是控制命令行，使用system("pause")函数执行一个shutdown -s -t 0，关机就完成了。但这种方式还要借助于命令行的方式解决问题。而Windows早就提供给我们直接控制关机/注销/重启的API了，在WindwosNT系统之前，只需调用ExitWindowsEx()就OK了。但自从出现了WindowsNT系统后，权限意识大大提高，为了提高系统的安全性，微软要求Windows执行关机/重启这类命令时一定要先提升进程权限，再执行ExitWindowsEx()函数。所以在NT代表的系统中，提升权限就得用到OpenProcessToken()函数打开进程的权限令牌，然后使用LookupPrivilegeValue()函数提取出关机权限所对应的Luid，再调用AdjustTokenPrivileges()函数修改对应权限即可。虽然操作繁琐了一些，但是并不复杂，相对于执行命令行的关机指令而言更加具有健壮性，不容易被拦截。

C++代码样例

/*
本程序只演示关机，注销，重启三种功能，主要目的是理解进程权限的控制，其他功能可详见MSDN文档说明
*/
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <conio.h>
#include <ctype.h>
#include <iostream>
#include <windows.h>

using namespace std;

/*
提升进程权限
*/
bool improvePv()
{
	HANDLE hToken;
	TOKEN_PRIVILEGES tkp;
	if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &hToken)) return false;
	if (!LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid)) return false;
	tkp.PrivilegeCount = 1;
	tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
	if (!AdjustTokenPrivileges(hToken, FALSE, &tkp, NULL, NULL, NULL)) return false;
	return true;
}

/*
关机
*/
bool powerOffProc()
{
	if (!improvePv() || !ExitWindowsEx(EWX_POWEROFF | EWX_FORCE, SHTDN_REASON_MAJOR_APPLICATION)) return false;
	return true;
}

/*
注销
*/
bool logOffProc()
{
	if (!improvePv() || !ExitWindowsEx(EWX_LOGOFF | EWX_FORCE, SHTDN_REASON_MAJOR_APPLICATION)) return false;
	return true;
}

/*
重启
*/
bool reBootProc()
{
	if (!improvePv() || !ExitWindowsEx(EWX_REBOOT | EWX_FORCE, SHTDN_REASON_MAJOR_APPLICATION)) return false;
	return true;
}

int main(void)
{
	CHAR ch;
	printf(">>>>>>>>>>>>>>>>>>>>>> Demo >>>>>>>>>>>>>>>>>\n*\n");
	printf("* 1. Power_Off\n*\n");
	printf("* 2. Log_Off\n*\n");
	printf("* 3. ReBoot\n*\n");
	printf(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>\n");
	ch = getch();
	while (1)
	{
		switch (ch)
		{
		case '1':
			if (!powerOffProc())
			{
				printf("Process Error!\n");
				continue;
			}
			return 0;
		case '2':
			if (!logOffProc())
			{
				printf("Process Error!\n");
				continue;
			}
			return 0;
		case '3':
			if (!reBootProc())
			{
				printf("Process Error!\n");
				continue;
			}
			return 0;
		default:
			printf("Error!\n");
		}
	}
	system("pause");
	return 0;
}