BangC Factory

char GetOSVersionType()
{

    // 계산된 윈도우즈 버전을 저장할 변수
    char version = -1;

    // 윈도우즈 버전에 관련된 정보를 얻어올 변수
    OSVERSIONINFOEX osvi;

    // 구분할 변수 ( 0 이면, OSVERSIONINFO 구조체 정보를 얻었다는 뜻입니다. )
    BOOL version_ex_flag = 0;

    // 운영체제의 버전과 관련된 정보를 얻기 위해 구조체 변수를 초기화합니다.
    ZeroMemory(&osvi, sizeof(OSVERSIONINFOEX));

    // 현재 운영체제의 버전을 알수 없기 때문에 OSVERSIONINFOEX 속성으로 버전정보를 얻습니다.
    osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);

 

    //  윈도우가 아니라면 이 함수는 실패할 것입니다.
    if(!(version_ex_flag = GetVersionEx((OSVERSIONINFO *)&osvi))){

        osvi.dwOSVersionInfoSize = sizeof (OSVERSIONINFO);
        if(!GetVersionEx((OSVERSIONINFO *)&osvi)) return -1;
    }

 

    // 얻어온 OSVERSIONINFOEX 구조체 변수의 값을 이용하여 버전을 구별합니다.
    switch(osvi.dwPlatformId){
        // NT 기술 기반의 운영체제인 경우
        case VER_PLATFORM_WIN32_NT:
            if(osvi.dwMajorVersion <= 4) version = 4;
            else if(osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 0) version = 5;
            else if(version_ex_flag) version = 6;
            break;

        // 윈도우즈 9x 기반의 운영체제인 경우
        case VER_PLATFORM_WIN32_WINDOWS:
            if(osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 0) version = 1;
            else if(osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 10) version = 2;
            else if(osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 90) version = 3;
            break;
    }

 

    return version;
}

 

char GetNtPrivilege()
{
    HANDLE h_token;
    TOKEN_PRIVILEGES privilege_info;

 

    // 현재 프로세스의 권한과 관련된 정보를 변경하기 위해 토큰정보를 엽니다.

    if(!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES |
                                                    TOKEN_QUERY, &h_token)){
        // 권한과 관련된 정보 접근에 실패함..
        return 0;
    }

 

    // 현재 프로세스가 SE_SHUTDOWN_NAME 권한을 사용할수 있도록 설정한다.

    LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &privilege_info.Privileges[0].Luid);
    privilege_info.PrivilegeCount = 1;
    privilege_info.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

 

    // 지정한 값으로 권한을 조정한다.
    AdjustTokenPrivileges(h_token, FALSE, &privilege_info, 0, (PTOKEN_PRIVILEGES)NULL, 0);

    if(GetLastError() != ERROR_SUCCESS){
        // 권한 조정에 실패한 경우...

        return 0;
    }

 

    return 1;
}

 

void PowerOff()
{

    if(GetOSVersionType()) <= 3{

        // 현재 운영체제가 Windows 95, Windows 98, Windows ME 인 경우...
        ExitWindowsEx(EWX_LOGOFF | EWX_POWEROFF | EWX_SHUTDOWN, 0);
        ExitWindowsEx(EWX_FORCE, 0);
    } else if(GetNtPrivilege()){

        ExitWindowsEx(EWX_FORCE | EWX_POWEROFF, 0);

    }

}