安全关机程序[通俗易懂]

大家好，又见面了，我是你们的朋友全栈君。

安全关机程序

最近在实验室用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); }

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/162969.html原文链接：https://javaforall.cn