TnbWindows.h

[詳解]

#pragma once
#include "TnbThread.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CCtrlAltDel : public CThreadlet::IRunner
{
    CCtrlAltDel(void) {}
 
    virtual void Run(LPARAM lParam)
    {
        HDESK old_desktop = ::GetThreadDesktop(::GetCurrentThreadId());
        DWORD dwDesiredAccess =      DESKTOP_CREATEMENU | DESKTOP_CREATEWINDOW
                                    | DESKTOP_ENUMERATE | DESKTOP_HOOKCONTROL
                                    | DESKTOP_WRITEOBJECTS | DESKTOP_READOBJECTS
                                    | DESKTOP_SWITCHDESKTOP | GENERIC_WRITE;
        HDESK h = ::OpenDesktop(_T("Winlogon"), 0, FALSE, dwDesiredAccess);
        if ( ::SetThreadDesktop(h) ) 
        {
            ::PostMessage(HWND_BROADCAST, WM_HOTKEY, 0, MAKELONG(MOD_ALT | MOD_CONTROL, VK_DELETE));
        }
        ::SetThreadDesktop(old_desktop);
    }
public:
    static void Exec(void)
    {
        CThreadlet::Start(0, new CCtrlAltDel);
    }
};
 
 
 
class CPowerManager
{
    CPowerManager(void)
    {
    }
 
    static bool ms_AddAuthority(LPCTSTR lpName = SE_SHUTDOWN_NAME)
    {
        // アクセストークンを開く
        HANDLE hHandle;
        if ( ! ::OpenProcessToken(::GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hHandle) )
        {
            return false;
        }
        TOKEN_PRIVILEGES token;
        token.PrivilegeCount = 1;   // 数は1つ
        if ( ! ::LookupPrivilegeValue(NULL, lpName, &(token.Privileges[0].Luid)) )
        {
            ::CloseHandle(hHandle);
            return false;
        }
        token.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; // 有効
        if ( ! ::AdjustTokenPrivileges(hHandle, FALSE, &token, 0, NULL, NULL) )
        {
            ::CloseHandle(hHandle);
            return false;
        }
        // アクセストークンを閉じる
        return !! ::CloseHandle(hHandle);
    }
 
    static bool ms_AddShutdownAuthority(LPCTSTR lpMachineName)
    {
        LPCTSTR P = SE_REMOTE_SHUTDOWN_NAME;
        if ( lpMachineName == NULL || *lpMachineName == 0 )
        {
            P = SE_SHUTDOWN_NAME;
        }
        return ms_AddAuthority(P);
    }
 
public:
 
    static bool Shutdown95(UINT uFlags = EWX_POWEROFF)
    {
        if ( ! ms_AddAuthority() )
        {
            return false;
        }
        return !! ::ExitWindowsEx(uFlags, 0);
    }
 
    static bool ShutdownNt(LPCTSTR lpMachineName, LPCTSTR lpMessage, DWORD dwTimeout,
                        bool bForceAppsClosed = false, bool bRebootAfterShutdown = true)
    {
        if ( ! ms_AddShutdownAuthority(lpMachineName) )
        {
            return false;
        }
        BOOL r = ::InitiateSystemShutdown(
                    CSimpleStr(lpMachineName), CSimpleStr(lpMessage), dwTimeout, bForceAppsClosed, bRebootAfterShutdown);
        return !! r;
    }
 
    static bool Shutdown(bool bForceAppsClosed = false, bool bRebootAfterShutdown = true)
    {
        bool r;
        OSVERSIONINFO tOsVerInfo = { sizeof(OSVERSIONINFO) };
        ::GetVersionEx(&tOsVerInfo);
        if ( tOsVerInfo.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS )
        {
            //Windows95/98系
            UINT uFlag = bRebootAfterShutdown ? EWX_REBOOT : EWX_POWEROFF;
            if ( bForceAppsClosed )
            {
                uFlag |= EWX_FORCE;
            }
            r = Shutdown95(uFlag);
        }
        else
        {
            //WindowsNT系
            r = ShutdownNt(NULL, NULL, 0, bForceAppsClosed, bRebootAfterShutdown);
        }
        return r;
    }
 
    static bool AbortShutdown(LPCTSTR lpMachineName)
    {
        if ( ! ms_AddShutdownAuthority(lpMachineName) )
        {
            return false;
        }
        BOOL r = ::AbortSystemShutdown(CSimpleStr(lpMachineName));
        return !! r;
    }
 
    static bool Suspend(bool fSuspend = true, bool fForce = false)
    {
        if ( ! ms_AddAuthority() )
        {
            return false;
        }
        return !! ::SetSystemPowerState(fSuspend, fForce);
    }
 
    static bool GetStatus(LPSYSTEM_POWER_STATUS _lpSystemPowerStatus)
    {
        return !! ::GetSystemPowerStatus(_lpSystemPowerStatus);
    }
 
    static bool SetShutdownNotifyLevel(WORD level)
    {
        DWORD l;
        DWORD f;
        if ( ::GetProcessShutdownParameters(&l, &f) )
        {
            return !! ::SetProcessShutdownParameters(level, f);
        }
        return false;
    }
};
 
 
 
};//TNB