• 欢迎访问开心洋葱网站,在线教程,推荐使用最新版火狐浏览器和Chrome浏览器访问本网站,欢迎加入开心洋葱 QQ群
  • 为方便开心洋葱网用户,开心洋葱官网已经开启复制功能!
  • 欢迎访问开心洋葱网站,手机也能访问哦~欢迎加入开心洋葱多维思维学习平台 QQ群
  • 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏开心洋葱吧~~~~~~~~~~~~~!
  • 由于近期流量激增,小站的ECS没能经的起亲们的访问,本站依然没有盈利,如果各位看如果觉着文字不错,还请看官给小站打个赏~~~~~~~~~~~~~!

VC实现windows系统强制关机重启和注销功能的代码

C# 水墨上仙 2614次浏览

VC实现windows系统强制关机重启和注销功能的代码
来源:http://blog.csdn.net/cackeme/article/details/7825751

#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


喜欢 (0)
加载中……