shutdown.c File Reference

#include "precomp.h"
#include <ndk/exfuncs.h>
#include <ndk/pofuncs.h>
#include <ndk/rtlfuncs.h>

Include dependency graph for shutdown.c:

Go to the source code of this file.

Functions
static BOOL	EnablePrivilege (LPCWSTR lpszPrivilegeName, BOOL bEnablePrivilege)
VOID	ShutDown_StandBy (VOID)
VOID	ShutDown_Hibernate (VOID)
VOID	ShutDown_PowerOff (VOID)
VOID	ShutDown_Reboot (VOID)
VOID	ShutDown_LogOffUser (VOID)
VOID	ShutDown_SwitchUser (VOID)
VOID	ShutDown_LockComputer (VOID)
VOID	ShutDown_Disconnect (VOID)
VOID	ShutDown_EjectComputer (VOID)

Function Documentation

EnablePrivilege()

static BOOL EnablePrivilege ( LPCWSTR  lpszPrivilegeName, BOOL  bEnablePrivilege  )

static

Definition at line 34 of file shutdown.c.

Referenced by ShutDown_Hibernate(), ShutDown_PowerOff(), ShutDown_Reboot(), and ShutDown_StandBy().

{
    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;
}

ShutDown_Disconnect()

VOID ShutDown_Disconnect

(

VOID

)

Definition at line 189 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
}

ShutDown_EjectComputer()

VOID ShutDown_EjectComputer

(

VOID

)

Definition at line 194 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
}

ShutDown_Hibernate()

VOID ShutDown_Hibernate

(

VOID

)

Definition at line 88 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
    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);
}

ShutDown_LockComputer()

VOID ShutDown_LockComputer

(

VOID

)

Definition at line 182 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
    if (!LockWorkStation())
        ShowWin32Error(GetLastError());
}

ShutDown_LogOffUser()

VOID ShutDown_LogOffUser

(

VOID

)

Definition at line 170 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
    if (!ExitWindowsEx(EWX_LOGOFF, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER))
        ShowWin32Error(GetLastError());
}

ShutDown_PowerOff()

VOID ShutDown_PowerOff

(

VOID

)

Definition at line 115 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
    /* 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);
}

ShutDown_Reboot()

VOID ShutDown_Reboot

(

VOID

)

Definition at line 143 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
    /* 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);
}

ShutDown_StandBy()

VOID ShutDown_StandBy

(

VOID

)

Definition at line 61 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
    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);
}

ShutDown_SwitchUser()

VOID ShutDown_SwitchUser

(

VOID

)

Definition at line 177 of file shutdown.c.

Referenced by TaskManagerWndProc().

{
}