#include "stdafx.h"
#include "SystemInformation.h"

OPERATING_SYSTEM SystemInformation::currenOS = UNKNOWN_OS;

bool SystemInformation::getSystemInformation()
{
    RTL_OSVERSIONINFOW osvi = {0};
    osvi.dwOSVersionInfoSize = sizeof(RTL_OSVERSIONINFOW);
    SYSTEM_INFO si = {0};
    def_GetNativeSystemInfo _GetNativeSystemInfo = 0;
    typedef LONG(WINAPI * tRtlGetVersion)(PRTL_OSVERSIONINFOW);
    tRtlGetVersion pRtlGetVersion = (tRtlGetVersion)GetProcAddress(GetModuleHandleW(L"ntdll.dll"), "RtlGetVersion");
    if(!pRtlGetVersion)
        return false;

    if(pRtlGetVersion(&osvi) != 0)
        return false;

    if((osvi.dwMajorVersion < 5) || ((osvi.dwMajorVersion == 5) && (osvi.dwMinorVersion == 0)))
    {
        return false;
    }

    _GetNativeSystemInfo = (def_GetNativeSystemInfo)GetProcAddress(GetModuleHandleW(L"kernel32.dll"), "GetNativeSystemInfo");
    if(_GetNativeSystemInfo)
    {
        _GetNativeSystemInfo(&si);
    }
    else
    {
        GetSystemInfo(&si);
    }

    bool isX64 = si.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64;
    bool isX86 = si.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_INTEL;


    DWORD major = osvi.dwMajorVersion;
    DWORD minor = osvi.dwMinorVersion;

    if(isX64 && major == 5 && minor == 2)
    {
        currenOS = WIN_XP_64;
    }
    else if(isX86 && major == 5 && minor == 1)
    {
        currenOS = WIN_XP_32;
    }
    else if(isX64 && major == 6 && minor == 0)
    {
        currenOS = WIN_VISTA_64;
    }
    else if(isX86 && major == 6 && minor == 0)
    {
        currenOS = WIN_VISTA_32;
    }
    else if(isX64 && major == 6 && minor == 1)
    {
        currenOS = WIN_7_64;
    }
    else if(isX86 && major == 6 && minor == 1)
    {
        currenOS = WIN_7_32;
    }
    else if(isX64 && major == 6 && minor == 2)
    {
        currenOS = WIN_8_64;
    }
    else if(isX86 && major == 6 && minor == 2)
    {
        currenOS = WIN_8_32;
    }
    else
    {
        currenOS = UNKNOWN_OS;
    }

    return (currenOS != UNKNOWN_OS);
}