64,685
社区成员
发帖
与我相关
我的任务
分享
#include <windows.h>
#pragma comment( lib, "advapi32.lib" )
BOOL MySystemShutdown( LPTSTR lpMsg )
{
HANDLE hToken; // handle to process token
TOKEN_PRIVILEGES tkp; // pointer to token structure
BOOL fResult; // system shutdown flag
// Get the current process token handle so we can get shutdown
// privilege.
if (!OpenProcessToken(GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
return FALSE;
// Get the LUID for 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 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)
return FALSE;
// Display the shutdown dialog box and start the countdown.
fResult = InitiateSystemShutdown(
NULL, // shut down local computer
lpMsg, // message for user
30, // time-out period, in seconds
FALSE, // ask user to close apps
TRUE); // reboot after shutdown
if (!fResult)
return FALSE;
// Disable shutdown privilege.
tkp.Privileges[0].Attributes = 0;
AdjustTokenPrivileges(hToken, FALSE, &tkp, 0,
(PTOKEN_PRIVILEGES) NULL, 0);
return TRUE;
}
HINSTANCE h=LoadLibrary("ntdll.dll");
if(!h)
{
MessageBox("加载'ntdll.dll'失败!");
return;
}
typedef int (__stdcall *AdjustPrivilege)(int,int,int,int *);
AdjustPrivilege SetPrivilege=NULL;
SetPrivilege=(AdjustPrivilege)GetProcAddress(h,"RtlAdjustPrivilege");
if(!SetPrivilege)
{
MessageBox("加载'RtlAdjustPrivilege'失败!");
return;
}
typedef int (__stdcall *ShutdownSystem)(int);
ShutdownSystem Shutdown=NULL;
Shutdown=(ShutdownSystem)GetProcAddress(h,"NtShutdownSystem");
if(!Shutdown)
{
MessageBox("加载'NtShutdownSystem'失败!");
return;
}
const int SE_SHUTDOWN_PRIVILEGE = 19;
const int shutdown = 0;
const int RESTART = 1;
const int POWEROFF = 2;
int *a;
(*SetPrivilege)(SE_SHUTDOWN_PRIVILEGE,true,false,a);
(*Shutdown)(POWEROFF);