mirror of
https://github.com/reactos/reactos.git
synced 2024-08-26 21:28:48 +00:00
197 lines
5.1 KiB
C
197 lines
5.1 KiB
C
/*
|
|
* ReactOS Task Manager
|
|
*
|
|
* shutdown.c
|
|
*
|
|
* Copyright (C) 1999 - 2001 Brian Palmer <brianp@reactos.org>
|
|
* 2011 Mário Kacmár /Mario Kacmar/ aka Kario (kario@szm.sk)
|
|
* 2014 Robert Naumann <gonzomdx@gmail.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
#include <ndk/exfuncs.h>
|
|
#include <ndk/pofuncs.h>
|
|
#include <ndk/rtlfuncs.h>
|
|
|
|
// Uncomment when NtInitiatePowerAction() is implemented
|
|
// #define NT_INITIATE_POWERACTION_IMPLEMENTED
|
|
|
|
static BOOL
|
|
EnablePrivilege(LPCWSTR lpszPrivilegeName, BOOL bEnablePrivilege)
|
|
{
|
|
BOOL Success;
|
|
HANDLE hToken;
|
|
TOKEN_PRIVILEGES tp;
|
|
|
|
Success = OpenProcessToken(GetCurrentProcess(),
|
|
TOKEN_ADJUST_PRIVILEGES,
|
|
&hToken);
|
|
if (!Success) return Success;
|
|
|
|
Success = LookupPrivilegeValueW(NULL,
|
|
lpszPrivilegeName,
|
|
&tp.Privileges[0].Luid);
|
|
if (!Success) goto Quit;
|
|
|
|
tp.PrivilegeCount = 1;
|
|
tp.Privileges[0].Attributes = (bEnablePrivilege ? SE_PRIVILEGE_ENABLED : 0);
|
|
|
|
Success = AdjustTokenPrivileges(hToken, FALSE, &tp, 0, NULL, NULL);
|
|
|
|
Quit:
|
|
CloseHandle(hToken);
|
|
return Success;
|
|
}
|
|
|
|
VOID
|
|
ShutDown_StandBy(VOID)
|
|
{
|
|
NTSTATUS Status;
|
|
|
|
if (!EnablePrivilege(SE_SHUTDOWN_NAME, TRUE))
|
|
{
|
|
ShowWin32Error(GetLastError());
|
|
return;
|
|
}
|
|
|
|
#ifdef NT_INITIATE_POWERACTION_IMPLEMENTED
|
|
Status = NtInitiatePowerAction(PowerActionSleep,
|
|
PowerSystemSleeping1,
|
|
0, FALSE);
|
|
#else
|
|
Status = NtSetSystemPowerState(PowerActionSleep,
|
|
PowerSystemSleeping1,
|
|
0);
|
|
#endif
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
ShowWin32Error(RtlNtStatusToDosError(Status));
|
|
|
|
EnablePrivilege(SE_SHUTDOWN_NAME, FALSE);
|
|
}
|
|
|
|
VOID
|
|
ShutDown_Hibernate(VOID)
|
|
{
|
|
NTSTATUS Status;
|
|
|
|
if (!EnablePrivilege(SE_SHUTDOWN_NAME, TRUE))
|
|
{
|
|
ShowWin32Error(GetLastError());
|
|
return;
|
|
}
|
|
|
|
#ifdef NT_INITIATE_POWERACTION_IMPLEMENTED
|
|
Status = NtInitiatePowerAction(PowerActionHibernate,
|
|
PowerSystemHibernate,
|
|
0, FALSE);
|
|
#else
|
|
Status = NtSetSystemPowerState(PowerActionHibernate,
|
|
PowerSystemHibernate,
|
|
0);
|
|
#endif
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
ShowWin32Error(RtlNtStatusToDosError(Status));
|
|
|
|
EnablePrivilege(SE_SHUTDOWN_NAME, FALSE);
|
|
}
|
|
|
|
VOID
|
|
ShutDown_PowerOff(VOID)
|
|
{
|
|
/* Trick: on Windows, pressing the CTRL key forces shutdown via NT API */
|
|
BOOL ForceShutdown = !!(GetKeyState(VK_CONTROL) & 0x8000);
|
|
|
|
if (!EnablePrivilege(SE_SHUTDOWN_NAME, TRUE))
|
|
{
|
|
ShowWin32Error(GetLastError());
|
|
return;
|
|
}
|
|
|
|
if (ForceShutdown)
|
|
{
|
|
NTSTATUS Status = NtShutdownSystem(ShutdownPowerOff);
|
|
if (!NT_SUCCESS(Status))
|
|
ShowWin32Error(RtlNtStatusToDosError(Status));
|
|
}
|
|
else
|
|
{
|
|
// The choice of EWX_SHUTDOWN or EWX_POWEROFF may be done with NtPowerInformation
|
|
if (!ExitWindowsEx(EWX_POWEROFF /* EWX_SHUTDOWN */, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER))
|
|
ShowWin32Error(GetLastError());
|
|
}
|
|
|
|
EnablePrivilege(SE_SHUTDOWN_NAME, FALSE);
|
|
}
|
|
|
|
VOID
|
|
ShutDown_Reboot(VOID)
|
|
{
|
|
/* Trick: on Windows, pressing the CTRL key forces reboot via NT API */
|
|
BOOL ForceReboot = !!(GetKeyState(VK_CONTROL) & 0x8000);
|
|
|
|
if (!EnablePrivilege(SE_SHUTDOWN_NAME, TRUE))
|
|
{
|
|
ShowWin32Error(GetLastError());
|
|
return;
|
|
}
|
|
|
|
if (ForceReboot)
|
|
{
|
|
NTSTATUS Status = NtShutdownSystem(ShutdownReboot);
|
|
if (!NT_SUCCESS(Status))
|
|
ShowWin32Error(RtlNtStatusToDosError(Status));
|
|
}
|
|
else
|
|
{
|
|
if (!ExitWindowsEx(EWX_REBOOT, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER))
|
|
ShowWin32Error(GetLastError());
|
|
}
|
|
|
|
EnablePrivilege(SE_SHUTDOWN_NAME, FALSE);
|
|
}
|
|
|
|
VOID
|
|
ShutDown_LogOffUser(VOID)
|
|
{
|
|
if (!ExitWindowsEx(EWX_LOGOFF, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER))
|
|
ShowWin32Error(GetLastError());
|
|
}
|
|
|
|
VOID
|
|
ShutDown_SwitchUser(VOID)
|
|
{
|
|
}
|
|
|
|
VOID
|
|
ShutDown_LockComputer(VOID)
|
|
{
|
|
if (!LockWorkStation())
|
|
ShowWin32Error(GetLastError());
|
|
}
|
|
|
|
VOID
|
|
ShutDown_Disconnect(VOID)
|
|
{
|
|
}
|
|
|
|
VOID
|
|
ShutDown_EjectComputer(VOID)
|
|
{
|
|
}
|