关机小程序的初步实现

 今天寝室一堆人都打游戏，吵得睡不着，果断起来把这一阵以来一直在纠结的关机小工具初步实现以及一些注意事项发上来，作为自己的一个备忘记录，也作为其他人的一个参考。

 

1. #include <windows.h> 
2. #include <iostream> 
3. using namespace std; 
4.  
5. void AjustPrivilege_LOGOFF() 
6. { 
7.     HANDLE hToken; 
8.     TOKEN_PRIVILEGES tkp_logoff; 
9.      
10.     OpenProcessToken( 
11.         GetCurrentProcess(), 
12.         TOKEN_ADJUST_PRIVILEGES | 
13.         TOKEN_QUERY, 
14.         &hToken 
15.         ); 
16.  
17.     LookupPrivilegeValue( 
18.         NULL, 
19.         SE_DEBUG_NAME, 
20.         &tkp_logoff.Privileges[0].Luid 
21.         ); 
22.  
23.     tkp_logoff.PrivilegeCount=1; 
24.     tkp_logoff.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED; 
25.  
26.     AdjustTokenPrivileges( 
27.         hToken,    
28.         FALSE,    
29.         &tkp_logoff,    
30.         0, 
31.         (PTOKEN_PRIVILEGES)NULL,  
32.         0); 
33. } 
34.  
35. void AjustPrivilege_SHUTDOWN() 
36. { 
37.     HANDLE hToken; 
38.     TOKEN_PRIVILEGES tkp_shutdown; 
39.      
40.     OpenProcessToken( 
41.         GetCurrentProcess(), 
42.         TOKEN_ADJUST_PRIVILEGES | 
43.         TOKEN_QUERY, 
44.         &hToken 
45.         ); 
46.  
47.     LookupPrivilegeValue( 
48.         NULL, 
49.         SE_SHUTDOWN_NAME, 
50.         &tkp_shutdown.Privileges[0].Luid 
51.         ); 
52.  
53.     tkp_shutdown.PrivilegeCount=1; 
54.     tkp_shutdown.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED; 
55.  
56.     AdjustTokenPrivileges( 
57.         hToken,    
58.         FALSE,    
59.         &tkp_shutdown,    
60.         0, 
61.         (PTOKEN_PRIVILEGES)NULL,  
62.         0); 
63. } 
64.  
65. int main() 
66. { 
67.     int choose; 
68.  
69.     cout<<"choose:"<<endl<<"1-logoff"<<endl<<"2-shutdown"<<endl; 
70.     cin>>choose; 
71.  
72.     switch(choose) 
73.     { 
74.     case 1: 
75.         AjustPrivilege_LOGOFF(); 
76.         ExitWindowsEx(EWX_LOGOFF,0); 
77.         cout<<"now,logoff is start..."<<endl; 
78.         break; 
79.     case 2: 
80.         AjustPrivilege_SHUTDOWN(); 
81.         ExitWindowsEx(EWX_SHUTDOWN,0); 
82.         cout<<"now,shutdown is start..."<<endl; 
83.         break; 
84.     default: 
85.         cout<<"fuck!"<<endl; 
86.         break; 
87.     } 
88. } 

以上是目前为止的全部代码。

下面做简要分析。

 

 

1. void AjustPrivilege_LOGOFF() 
2. { 
3.     HANDLE hToken; 
4.     TOKEN_PRIVILEGES tkp_logoff; 
5.      
6.     OpenProcessToken( 
7.         GetCurrentProcess(), 
8.         TOKEN_ADJUST_PRIVILEGES | 
9.         TOKEN_QUERY, 
10.         &hToken 
11.         ); 
12.  
13.     LookupPrivilegeValue( 
14.         NULL, 
15.         SE_DEBUG_NAME, 
16.         &tkp_logoff.Privileges[0].Luid 
17.         ); 
18.  
19.     tkp_logoff.PrivilegeCount=1; 
20.     tkp_logoff.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED; 
21.  
22.     AdjustTokenPrivileges( 
23.         hToken,    
24.         FALSE,    
25.         &tkp_logoff,    
26.         0, 
27.         (PTOKEN_PRIVILEGES)NULL,  
28.         0); 
29. } 
30.  
31. void AjustPrivilege_SHUTDOWN() 
32. { 
33.     HANDLE hToken; 
34.     TOKEN_PRIVILEGES tkp_shutdown; 
35.      
36.     OpenProcessToken( 
37.         GetCurrentProcess(), 
38.         TOKEN_ADJUST_PRIVILEGES | 
39.         TOKEN_QUERY, 
40.         &hToken 
41.         ); 
42.  
43.     LookupPrivilegeValue( 
44.         NULL, 
45.         SE_SHUTDOWN_NAME, 
46.         &tkp_shutdown.Privileges[0].Luid 
47.         ); 
48.  
49.     tkp_shutdown.PrivilegeCount=1; 
50.     tkp_shutdown.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED; 
51.  
52.     AdjustTokenPrivileges( 
53.         hToken,    
54.         FALSE,    
55.         &tkp_shutdown,    
56.         0, 
57.         (PTOKEN_PRIVILEGES)NULL,  
58.         0); 
59. } 

以上两个函数是用于获得关机层次的操作所需的权限的。此项是在win2k之后就加入了windows产品中的一个安全措施。要注意的是，在编写代码期间我曾连续4天纠结于这个问题。下面来详细解析一下。

我们看到，两个函数的内容其实上没有太大区别，但是为什么我要写两个呢？我在最初的时候并不知道要这么写，所以只写了一个。但是悲剧的情况出现了。大家注意对比这两条语句：

 

1. LookupPrivilegeValue( 
2.         NULL, 
3.         SE_DEBUG_NAME, 
4.         &tkp_logoff.Privileges[0].Luid 
5.         ); 
6.  
7. LookupPrivilegeValue( 
8.         NULL, 
9.         SE_SHUTDOWN_NAME, 
10.         &tkp_shutdown.Privileges[0].Luid 
11.         ); 

有没有发现，第二个参数是不一样的。我当时就不知道这个区别，卡在这里很久。因为只写了上头一个，那么就只能实现注销，如果只写了下头一个，就只能实现关机，不论后面的ExitWindowsEx()函数的参数是EWX_LOGOFF还是EWX_SHUTDOWN。

也就是说，可以这样理解：最终的操作其实主要是跟LookupPrivilegeValue()这个函数相关，而ExitWindowsEx()只是在最后操作时起一点作用。<---这部分是我个人的理解，希望大牛指正。

然后，我们回到这两个函数来看。他们的作用已经说过，都是为了提升权限。那么，方便一些比较懒的读者不想去百度或者GOOGLE，每条语句我们都来分析一下好了，但不会做详细讲解，因为我自己都不太懂（笑）。

 

1. OpenProcessToken(
2. GetCurrentProcess(),
3. TOKEN_ADJUST_PRIVILEGES |
4. TOKEN_QUERY,
5. &hToken
6. );

此函数用于获取进程标记，最后一个参数便是我自己所创造的句柄。

 

1. LookupPrivilegeValue( 
2.         NULL, 
3.         SE_DEBUG_NAME, 
4.         &tkp_logoff.Privileges[0].Luid 
5.         ); 

获得本地机唯一的标识，注意，如果需要远程关机，参数会有相应变化，具体情况请查阅最新版MSDN。

 

1. tkp_logoff.PrivilegeCount=1; 
2. tkp_logoff.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED; 

第一条语句用于指定操作的个数，第二条语句用于提升权限复制。这两条语句我是没有太明白的，详细情况请查阅最新版MSDN。

 

1. AdjustTokenPrivileges( 
2.         hToken,    
3.         FALSE,    
4.         &tkp_logoff,    
5.         0, 
6.         (PTOKEN_PRIVILEGES)NULL,  
7.         0); 

用于启用或禁用特权一个有TOKEN_ADJUST_PRIVILEGES访问的访问令牌，这个函数我也不太明白，详细情况请查阅最新版MSDN。

 

目前就完成了这些，只实现了本程序操作计算机进行注销和关机，当然，如果想要加入重启或者休眠之类的功能也是可以的，不过要记得添加相应的函数。只要记住一点，函数与操作相对应就OK。下一步准备实现定时功能。以我对自己的了解，查阅实例，分析实例，再加上编写代码和测试，没有4天拿不下来，估计文章就得1周左右的时间了，如果有朋友想要看完这个系列估计有的等了。<---我估计没人回看我这菜鸟写的东西的(笑)...

 

不管怎么说，尽请期待后续吧。

转载于:https://blog.51cto.com/moowoo/532388