简单实现windows系统普通或强制关机重启注销的功能

简单实现windows系统普通或强制关机重启注销的功能

#ifndef __H_SHUTDOWN_H__
#define __H_SHUTDOWN_H__

typedef enum _SHUTDOWN_ACTION
{
    ShutdownNoReboot,
    ShutdownReboot,
    ShutdownPowerOff
} SHUTDOWN_ACTION;

typedef DWORD (WINAPI* lpNtShutdownSystem)(SHUTDOWN_ACTION Action);

typedef DWORD (WINAPI* lpNtSetSystemPowerState)(IN POWER_ACTION SystemAction,
                                                IN SYSTEM_POWER_STATE MinSystemState,
                                                IN ULONG Flags);

class KsSystemDown
{
protected:
    HMODULE hNtdll;
    TOKEN_PRIVILEGES tkp;
    HANDLE hToken;

    lpNtShutdownSystem NtShutdownSystem;
    lpNtSetSystemPowerState NtSetSystemPowerState;

public:
    BOOL SystemShutdown(SHUTDOWN_ACTION action)
    {
        if (!NtShutdownSystem)
            return FALSE;

        DWORD dwRet = NtShutdownSystem(action);
        if (dwRet == 0)
            return TRUE;
        else
            return FALSE;
    }

    BOOL SystemPowerdown(IN POWER_ACTION SystemAction,IN SYSTEM_POWER_STATE MinSystemState,IN ULONG Flags)
    {
        if (!NtSetSystemPowerState)
            return FALSE;

        DWORD dwRet = NtSetSystemPowerState(SystemAction,MinSystemState,Flags);
        if (dwRet == 0)
            return TRUE;
        else
            return FALSE;
    }

    BOOL CommonLogoff()
    {
        return Shutdown(EWX_LOGOFF);
    }

    BOOL CommonShutdown()
    {
        return Shutdown((EWX_POWEROFF|EWX_SHUTDOWN));
    }

    BOOL CommonReBoot()
    {
        return Shutdown(EWX_REBOOT);
    }

    BOOL FastLogoff()
    {
        return Shutdown(EWX_LOGOFF|EWX_FORCE);
    }

    BOOL FastShutdown()
    {
        return Shutdown(EWX_POWEROFF|EWX_SHUTDOWN|EWX_FORCE);
    }

    BOOL FastReboot()
    {
        return Shutdown(EWX_REBOOT|EWX_FORCE);
    }

    BOOL Shutdown(DWORD dwValue)
    {
        return ExitWindowsEx(dwValue,0);
    }

    BOOL SuperShutdown()
    {
        if (NtShutdownSystem != NULL)
            return NtShutdownSystem(ShutdownNoReboot);
        else
            return FALSE;
    }

    BOOL SuperReboot()
    {
        if (NtShutdownSystem != NULL)
            return NtShutdownSystem(ShutdownReboot);
        else
            return FALSE;
    }

public:
    KsSystemDown():
      NtShutdownSystem(NULL),NtSetSystemPowerState(NULL)
      {
          OpenProcessToken(GetCurrentProcess(),TOKEN_QUERY|TOKEN_ADJUST_PRIVILEGES,&hToken);
          LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid);
          tkp.PrivilegeCount = 1;
          tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
          AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, NULL, 0);
          CloseHandle(hToken);

          hNtdll = GetModuleHandle("ntdll.dll");
          if (hNtdll)
          {
              NtShutdownSystem = (lpNtShutdownSystem)GetProcAddress(hNtdll,"NtShutdownSystem");
              NtSetSystemPowerState = (lpNtSetSystemPowerState)GetProcAddress(hNtdll,"NtSetSystemPowerState");
          }
      }

      ~KsSystemDown()
      {
          tkp.Privileges[0].Attributes = 0; 
          AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES) NULL, 0);
      }
};

#endif