180 lines
6.2 KiB
C#
180 lines
6.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace CC_Functions.W32
|
|
{
|
|
public static class Power
|
|
{
|
|
[DllImport("ntdll.dll", SetLastError = true)]
|
|
public static extern IntPtr RtlAdjustPrivilege(int Privilege, bool bEnablePrivilege, bool IsThreadPrivilege, out bool PreviousValue);
|
|
[DllImport("ntdll.dll")]
|
|
private static extern uint NtRaiseHardError(
|
|
uint ErrorStatus,
|
|
uint NumberOfParameters,
|
|
uint UnicodeStringParameterMask,
|
|
IntPtr Parameters,
|
|
uint ValidResponseOption,
|
|
out uint Response
|
|
);
|
|
[DllImport("user32.dll", SetLastError = true)]
|
|
[return: MarshalAs(UnmanagedType.Bool)]
|
|
static extern bool ExitWindowsEx(ExitWindows uFlags, ShutdownReason dwReason);
|
|
[Flags]
|
|
public enum ShutdownReason : uint
|
|
{
|
|
MajorApplication = 0x00040000,
|
|
MajorHardware = 0x00010000,
|
|
MajorLegacyApi = 0x00070000,
|
|
MajorOperatingSystem = 0x00020000,
|
|
MajorOther = 0x00000000,
|
|
MajorPower = 0x00060000,
|
|
MajorSoftware = 0x00030000,
|
|
MajorSystem = 0x00050000,
|
|
|
|
MinorBlueScreen = 0x0000000F,
|
|
MinorCordUnplugged = 0x0000000b,
|
|
MinorDisk = 0x00000007,
|
|
MinorEnvironment = 0x0000000c,
|
|
MinorHardwareDriver = 0x0000000d,
|
|
MinorHotfix = 0x00000011,
|
|
MinorHung = 0x00000005,
|
|
MinorInstallation = 0x00000002,
|
|
MinorMaintenance = 0x00000001,
|
|
MinorMMC = 0x00000019,
|
|
MinorNetworkConnectivity = 0x00000014,
|
|
MinorNetworkCard = 0x00000009,
|
|
MinorOther = 0x00000000,
|
|
MinorOtherDriver = 0x0000000e,
|
|
MinorPowerSupply = 0x0000000a,
|
|
MinorProcessor = 0x00000008,
|
|
MinorReconfig = 0x00000004,
|
|
MinorSecurity = 0x00000013,
|
|
MinorSecurityFix = 0x00000012,
|
|
MinorSecurityFixUninstall = 0x00000018,
|
|
MinorServicePack = 0x00000010,
|
|
MinorServicePackUninstall = 0x00000016,
|
|
MinorTermSrv = 0x00000020,
|
|
MinorUnstable = 0x00000006,
|
|
MinorUpgrade = 0x00000003,
|
|
MinorWMI = 0x00000015,
|
|
|
|
FlagUserDefined = 0x40000000,
|
|
FlagPlanned = 0x80000000
|
|
}
|
|
[Flags]
|
|
public enum ExitWindows : uint
|
|
{
|
|
// ONE of the following five:
|
|
LogOff = 0x00,
|
|
ShutDown = 0x01,
|
|
Reboot = 0x02,
|
|
PowerOff = 0x08,
|
|
RestartApps = 0x40,
|
|
// plus AT MOST ONE of the following two:
|
|
Force = 0x04,
|
|
ForceIfHung = 0x10,
|
|
}
|
|
[Flags]
|
|
public enum ShutdownMode : uint
|
|
{
|
|
LogOff = 0x00,
|
|
ShutDown = 0x01,
|
|
Reboot = 0x02,
|
|
PowerOff = 0x08,
|
|
RestartApps = 0x40,
|
|
BSoD = 0x29a
|
|
}
|
|
[Flags]
|
|
public enum ShutdownMod : uint
|
|
{
|
|
None = 0x00,
|
|
Force = 0x04,
|
|
ForceIfHung = 0x10
|
|
}
|
|
public static unsafe void RaiseEvent(ShutdownMode mode, ShutdownReason reason = ShutdownReason.MinorOther, ShutdownMod mod = ShutdownMod.None)
|
|
{
|
|
if (mode == ShutdownMode.BSoD)
|
|
{
|
|
bool t1;
|
|
uint t2;
|
|
RtlAdjustPrivilege(19, true, false, out t1);
|
|
NtRaiseHardError(0xc0000022, 0, 0, IntPtr.Zero, 6, out t2);
|
|
}
|
|
else
|
|
{
|
|
Privileges.EnablePrivilege(SecurityEntity.SE_SHUTDOWN_NAME);
|
|
ExitWindowsEx((ExitWindows)((uint)mode | (uint)mod), reason);
|
|
}
|
|
}
|
|
}
|
|
class ugly_copy_paste_code
|
|
{
|
|
/*
|
|
[StructLayout(LayoutKind.Sequential, Pack = 1)]
|
|
internal struct TokPriv1Luid
|
|
{
|
|
public int Count;
|
|
public long Luid;
|
|
public int Attr;
|
|
}
|
|
|
|
[DllImport("kernel32.dll", ExactSpelling = true)]
|
|
internal static extern IntPtr GetCurrentProcess();
|
|
|
|
[DllImport("advapi32.dll", ExactSpelling = true, SetLastError = true)]
|
|
internal static extern bool OpenProcessToken(IntPtr h, int acc, ref IntPtr
|
|
phtok);
|
|
|
|
[DllImport("advapi32.dll", SetLastError = true)]
|
|
internal static extern bool LookupPrivilegeValue(string host, string name,
|
|
ref long pluid);
|
|
|
|
[DllImport("advapi32.dll", ExactSpelling = true, SetLastError = true)]
|
|
internal static extern bool AdjustTokenPrivileges(IntPtr htok, bool disall,
|
|
ref TokPriv1Luid newst, int len, IntPtr prev, IntPtr relen);
|
|
|
|
[DllImport("user32.dll", ExactSpelling = true, SetLastError = true)]
|
|
internal static extern bool ExitWindowsEx(int flg, int rea);
|
|
|
|
[Flags()]
|
|
public enum ExitWindowsValues
|
|
{
|
|
Logoff = 0x00000000,
|
|
Shutdown = 0x00000001,
|
|
Reboot = 0x00000002,
|
|
Force = 0x00000004,
|
|
Poweroff = 0x00000008,
|
|
For
|
|
}
|
|
internal const int SE_PRIVILEGE_ENABLED = 0x00000002;
|
|
internal const int TOKEN_QUERY = 0x00000008;
|
|
internal const int TOKEN_ADJUST_PRIVILEGES = 0x00000020;
|
|
internal const string SE_SHUTDOWN_NAME = "SeShutdownPrivilege";
|
|
internal const int EWX_LOGOFF = 0x00000000;
|
|
internal const int EWX_SHUTDOWN = 0x00000001;
|
|
internal const int EWX_REBOOT = 0x00000002;
|
|
internal const int EWX_FORCE = 0x00000004;
|
|
internal const int EWX_POWEROFF = 0x00000008;
|
|
internal const int EWX_FORCEIFHUNG = 0x00000010;
|
|
|
|
private void DoExitWin(int flg)
|
|
{
|
|
TokPriv1Luid tp;
|
|
IntPtr hproc = GetCurrentProcess();
|
|
IntPtr htok = IntPtr.Zero;
|
|
OpenProcessToken(hproc, TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, ref htok);
|
|
tp.Count = 1;
|
|
tp.Luid = 0;
|
|
tp.Attr = SE_PRIVILEGE_ENABLED;
|
|
LookupPrivilegeValue(null, SE_SHUTDOWN_NAME, ref tp.Luid);
|
|
AdjustTokenPrivileges(htok, false, ref tp, 0, IntPtr.Zero, IntPtr.Zero);
|
|
ExitWindowsEx(flg, 0);
|
|
}
|
|
*/
|
|
}
|
|
}
|