SourceCodeSky

#include "stdafx.h"
#include "Shutdown.h"
#include "reason.h"
#include "resource.h"
bool InitPrivileges()
{
    HANDLE token;
    TOKEN_PRIVILEGES tkp;
    if(!::OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,&token))
        return false;
    ::LookupPrivilegeValue(NULL,SE_SHUTDOWN_NAME,&tkp.Privileges[0].Luid);
    tkp.PrivilegeCount=1;
    tkp.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED;
    if(!AdjustTokenPrivileges(token,false,&tkp,0,(PTOKEN_PRIVILEGES)NULL,0))
        return false;
    if(GetLastError()!=ERROR_SUCCESS)
        return false;
    return true;
}
bool kgdShutdown(bool bShutdown)
{
    InitPrivileges();
    if(bShutdown)
    {
        if(!ExitWindowsEx(EWX_FORCE|EWX_SHUTDOWN,SHTDN_REASON_MAJOR_OPERATINGSYSTEM|SHTDN_REASON_MINOR_UPGRADE|
        SHTDN_REASON_FLAG_PLANNED))
        return false;
    }
    else
    {
        if(!ExitWindowsEx(EWX_FORCE|EWX_REBOOT,SHTDN_REASON_MAJOR_OPERATINGSYSTEM|SHTDN_REASON_MINOR_UPGRADE|
        SHTDN_REASON_FLAG_PLANNED))
        return false;
    }
    
    return true;
}
void kgdHibernate(bool bHibernate)
{
    InitPrivileges();
    if(bHibernate)
    {
        SetSystemPowerState(false,true);
    }
    else
    {
        SetSystemPowerState(true,true);
    }
}
void ShutdownComputer(ShutdownOption so)
{
    switch(so)
    {
    case Shutdown:
        kgdShutdown(true);
        break;
    case Reboot:
        kgdShutdown(false);
        break;
    case Logoff:
        ::ExitWindowsEx(EWX_FORCE|EWX_LOGOFF,0);
        break;
    case LockScreen:
        ::LockWorkStation();
        break;
    case Hibernate:
        kgdHibernate(true);
        break;
    case Suspend:
        kgdHibernate(false);
        break;
    }
}