#include "stdafx.h"#include "windows.h"#include <iostream>using namespace std;#pragma comment(lib, "user32.lib")#pragma comment(lib, "advapi32.lib")BOOL _tmain(int argc, _TCHAR* argv[]){ HANDLE hToken; TOKEN_PRIVILEGES tkp; UINT action; if (argc > 2) return FALSE; if (wcscmp(argv[1], L"/r") == 0) action = EWX_REBOOT; else if (wcscmp(argv[1], L"/s") == 0) action = EWX_SHUTDOWN; else return FALSE; // Get a token for this process. if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) { cout << "Open process token failed"<<endl; return FALSE; } // 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); if (GetLastError() != ERROR_SUCCESS) { cout << "Set privileges failed."<<endl; return FALSE; } // Shut down the system and force all applications to close. if (!ExitWindowsEx(action | EWX_FORCE, SHTDN_REASON_MAJOR_OPERATINGSYSTEM | SHTDN_REASON_MINOR_UPGRADE | SHTDN_REASON_FLAG_PLANNED)) { cout << "ExitWindowsEx failed." <<endl; return FALSE; } //shutdown was successful return TRUE;}
Attached Files
-
powermgm.rar 49.86KB
560 downloads
, better it works doesn't it 
)
