安全关机程序

安全关机程序

最近在实验室用ftp下点东西，但是由于实验室晚上12点就会断电。于是
需要在此之前关掉机器，图省事就用WindowsXP自带的计划任务每次设置
成11：50就调用“shutdown -s”命令自动关机。但是好几次都发现没法
正常关机，第二天早上起来就会检测磁盘。于是就做了个实验，发现确实
当使用flashfxp下载东西时，关机会不能正常关机，等待确定终止flashfxp
程序。

发现原因后，很简单，先把用户进程全部terminate掉，然后再自动关机。
于是就有了本文。

具体内容很简单，用CreateToolhelp32Snapshot函数得到当前进程快照，
然后Process32First和Process32Next函数循环得到进程的ID号。然后再
调用OpenProcess得到进程句柄和其优先级类。从而判断是否是用户进程，
将之Terminate掉（使用TerminateProcess函数）。最后调用ExitWindowsEx
函数关闭机器。

为了防止意外，在程序中还再次调用windows程序：shutdown -s命令来关
闭机器。

具体代码如下：

/*
   AutoShutDown.cpp
*/

#include <windows.h>
#include <tlhelp32.h>
#include <stdio.h>

BOOL fn_KillAllUserProcess();

BOOL fn_ShutdownComputer();

void error(char * szErrorMessage);

void main( )
{
  fn_KillAllUserProcess();
  fn_ShutdownComputer();
}

BOOL fn_KillAllUserProcess()
{
 HANDLE hProcessSnap;
 HANDLE hProcess;
 PROCESSENTRY32 pe32;
 DWORD  dwPriorityClass;

 
 // Take a snapshot of all processes in the system.
 hProcessSnap = CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0

);
 if( hProcessSnap == INVALID_HANDLE_VALUE )
 {
  return FALSE;
 }
 
 // Set the size of the structure before using it.
 pe32.dwSize = sizeof( PROCESSENTRY32 );
 
 // Retrieve information about the first process,
 // and exit if unsuccessful
 if( !Process32First( hProcessSnap, &pe32 ) )
 {
  CloseHandle( hProcessSnap );     // Must clean up the

snapshot object!
  return  FALSE;
 }
 
 // Now walk the snapshot of processes, and
 // display information about each process in turn
 do
 {
  // Retrieve the priority class.
  dwPriorityClass = 0;
  hProcess = OpenProcess( PROCESS_ALL_ACCESS, FALSE,

pe32.th32ProcessID );
  if( hProcess != NULL )
  {
   dwPriorityClass = GetPriorityClass( hProcess );
  }
  
  //judge if the user's process and if the process of

this program
  if (dwPriorityClass == 32 && stricmp(pe32.szExeFile,

"AutoShutDown.exe") != 0)
  {
   //terminate the prcess
   TerminateProcess(hProcess, 0);
  }
  
  if (hProcess != NULL)
   CloseHandle(hProcess);
  
 } while( Process32Next( hProcessSnap, &pe32 ) );
 
 CloseHandle( hProcessSnap );

 return TRUE ;

}

BOOL fn_ShutdownComputer()
{
 HANDLE hToken;
 TOKEN_PRIVILEGES tkp;
 
 // Get a token for this process.
 
 if (!OpenProcessToken(GetCurrentProcess(),
        TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
  error("OpenProcessToken");
 
 // Get the LUID for the shutdown privilege.
 
 LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME,
        &tkp.Privileges[0].Luid);
 
 tkp.PrivilegeCount = 1;  // one privilege to set   
 tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 
 // Get the shutdown privilege for this process.
 
 AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
        (PTOKEN_PRIVILEGES)NULL, 0);
 
 // Cannot test the return value of AdjustTokenPrivileges.
 
 if (GetLastError() != ERROR_SUCCESS)
  error("AdjustTokenPrivileges");
 
 // Shut down the system and force all applications to close.
 
 if (!ExitWindowsEx(EWX_SHUTDOWN | EWX_FORCE, 0))
  error("ExitWindowsEx");

 //sleep 30 seconds
 Sleep(30000);

 //use the onther way to shundown the computer
 system("shutdown -s");

 return true;
}

void error(char * szErrorMessage)
{
 printf("%s/n", szErrorMessage);
}