X-Git-Url: http://git.osdn.net/view?a=blobdiff_plain;f=src%2FOSSupport_Win32.cpp;h=8aab56a0c6d5a9379e88becbf91eae69ebbca3be;hb=d42da03bf1601c24d5005811cb5327e32fb8d62c;hp=1b9698d6f95ab1efdcec4df4ac5eb5d903da7c84;hpb=2086c3dde6a28d34d6439d7080aad2353c0c7eaa;p=mutilities%2FMUtilities.git diff --git a/src/OSSupport_Win32.cpp b/src/OSSupport_Win32.cpp index 1b9698d..8aab56a 100644 --- a/src/OSSupport_Win32.cpp +++ b/src/OSSupport_Win32.cpp @@ -32,6 +32,7 @@ #include #include #include +#include //Qt #include @@ -46,19 +47,235 @@ static const DWORD g_main_thread_id = GetCurrentThreadId(); // SYSTEM MESSAGE /////////////////////////////////////////////////////////////////////////////// +static const UINT g_msgBoxFlags = MB_TOPMOST | MB_TASKMODAL | MB_SETFOREGROUND; + void MUtils::OS::system_message_nfo(const wchar_t *const title, const wchar_t *const text) { - MessageBoxW(NULL, text, title, MB_TOPMOST | MB_TASKMODAL | MB_ICONINFORMATION); + MessageBoxW(NULL, text, title, g_msgBoxFlags | MB_ICONINFORMATION); } void MUtils::OS::system_message_wrn(const wchar_t *const title, const wchar_t *const text) { - MessageBoxW(NULL, text, title, MB_TOPMOST | MB_TASKMODAL | MB_ICONWARNING); + MessageBoxW(NULL, text, title, g_msgBoxFlags | MB_ICONWARNING); } void MUtils::OS::system_message_err(const wchar_t *const title, const wchar_t *const text) { - MessageBoxW(NULL, text, title, MB_TOPMOST | MB_TASKMODAL | MB_ICONERROR); + MessageBoxW(NULL, text, title, g_msgBoxFlags | MB_ICONERROR); +} + +/////////////////////////////////////////////////////////////////////////////// +// FETCH CLI ARGUMENTS +/////////////////////////////////////////////////////////////////////////////// + +static QReadWriteLock g_arguments_lock; +static QScopedPointer g_arguments_list; + +const QStringList &MUtils::OS::arguments(void) +{ + QReadLocker readLock(&g_arguments_lock); + + //Already initialized? + if(!g_arguments_list.isNull()) + { + return (*(g_arguments_list.data())); + } + + readLock.unlock(); + QWriteLocker writeLock(&g_arguments_lock); + + //Still not initialized? + if(!g_arguments_list.isNull()) + { + return (*(g_arguments_list.data())); + } + + g_arguments_list.reset(new QStringList); + int nArgs = 0; + LPWSTR *szArglist = CommandLineToArgvW(GetCommandLineW(), &nArgs); + + if(NULL != szArglist) + { + for(int i = 0; i < nArgs; i++) + { + *(g_arguments_list.data()) << MUTILS_QSTR(szArglist[i]); + } + LocalFree(szArglist); + } + else + { + qWarning("CommandLineToArgvW() has failed !!!"); + } + + return (*(g_arguments_list.data())); +} + +/////////////////////////////////////////////////////////////////////////////// +// OS VERSION DETECTION +/////////////////////////////////////////////////////////////////////////////// + +static bool g_os_version_initialized = false; +static MUtils::OS::Version::os_version_t g_os_version_info = MUtils::OS::Version::UNKNOWN_OPSYS; +static QReadWriteLock g_os_version_lock; + +//Maps marketing names to the actual Windows NT versions +static const struct +{ + MUtils::OS::Version::os_version_t version; + const char friendlyName[64]; +} +g_os_version_lut[] = +{ + { MUtils::OS::Version::WINDOWS_WIN2K, "Windows 2000" }, //2000 + { MUtils::OS::Version::WINDOWS_WINXP, "Windows XP or Windows XP Media Center Edition" }, //XP + { MUtils::OS::Version::WINDOWS_XPX64, "Windows Server 2003 or Windows XP x64" }, //XP_x64 + { MUtils::OS::Version::WINDOWS_VISTA, "Windows Vista or Windows Server 2008" }, //Vista + { MUtils::OS::Version::WINDOWS_WIN70, "Windows 7 or Windows Server 2008 R2" }, //7 + { MUtils::OS::Version::WINDOWS_WIN80, "Windows 8 or Windows Server 2012" }, //8 + { MUtils::OS::Version::WINDOWS_WIN81, "Windows 8.1 or Windows Server 2012 R2" }, //8.1 + { MUtils::OS::Version::WINDOWS_WN100, "Windows 10 or Windows Server 2014 (Preview)" }, //10 + { MUtils::OS::Version::UNKNOWN_OPSYS, "N/A" } +}; + +static bool verify_os_version(const DWORD major, const DWORD minor) +{ + OSVERSIONINFOEXW osvi; + DWORDLONG dwlConditionMask = 0; + + //Initialize the OSVERSIONINFOEX structure + memset(&osvi, 0, sizeof(OSVERSIONINFOEXW)); + osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEXW); + osvi.dwMajorVersion = major; + osvi.dwMinorVersion = minor; + osvi.dwPlatformId = VER_PLATFORM_WIN32_NT; + + //Initialize the condition mask + VER_SET_CONDITION(dwlConditionMask, VER_MAJORVERSION, VER_GREATER_EQUAL); + VER_SET_CONDITION(dwlConditionMask, VER_MINORVERSION, VER_GREATER_EQUAL); + VER_SET_CONDITION(dwlConditionMask, VER_PLATFORMID, VER_EQUAL); + + // Perform the test + const BOOL ret = VerifyVersionInfoW(&osvi, VER_MAJORVERSION | VER_MINORVERSION | VER_PLATFORMID, dwlConditionMask); + + //Error checking + if(!ret) + { + if(GetLastError() != ERROR_OLD_WIN_VERSION) + { + qWarning("VerifyVersionInfo() system call has failed!"); + } + } + + return (ret != FALSE); +} + +static bool get_real_os_version(unsigned int *major, unsigned int *minor, bool *pbOverride) +{ + *major = *minor = 0; + *pbOverride = false; + + //Initialize local variables + OSVERSIONINFOEXW osvi; + memset(&osvi, 0, sizeof(OSVERSIONINFOEXW)); + osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEXW); + + //Try GetVersionEx() first + if(GetVersionExW((LPOSVERSIONINFOW)&osvi) == FALSE) + { + qWarning("GetVersionEx() has failed, cannot detect Windows version!"); + return false; + } + + //Make sure we are running on NT + if(osvi.dwPlatformId == VER_PLATFORM_WIN32_NT) + { + *major = osvi.dwMajorVersion; + *minor = osvi.dwMinorVersion; + } + else + { + qWarning("Not running on Windows NT, unsupported operating system!"); + return false; + } + + //Determine the real *major* version first + forever + { + const DWORD nextMajor = (*major) + 1; + if(verify_os_version(nextMajor, 0)) + { + *pbOverride = true; + *major = nextMajor; + *minor = 0; + continue; + } + break; + } + + //Now also determine the real *minor* version + forever + { + const DWORD nextMinor = (*minor) + 1; + if(verify_os_version((*major), nextMinor)) + { + *pbOverride = true; + *minor = nextMinor; + continue; + } + break; + } + + return true; +} + +const MUtils::OS::Version::os_version_t &MUtils::OS::os_version(void) +{ + QReadLocker readLock(&g_os_version_lock); + + //Already initialized? + if(g_os_version_initialized) + { + return g_os_version_info; + } + + readLock.unlock(); + QWriteLocker writeLock(&g_os_version_lock); + + //Initialized now? + if(g_os_version_initialized) + { + return g_os_version_info; + } + + //Detect OS version + unsigned int major, minor; bool overrideFlg; + if(get_real_os_version(&major, &minor, &overrideFlg)) + { + g_os_version_info.type = Version::OS_WINDOWS; + g_os_version_info.versionMajor = major; + g_os_version_info.versionMinor = minor; + g_os_version_info.overrideFlag = overrideFlg; + } + else + { + qWarning("Failed to determin the operating system version!"); + } + + g_os_version_initialized = true; + return g_os_version_info; +} + +const char *MUtils::OS::os_friendly_name(const MUtils::OS::Version::os_version_t &os_version) +{ + for(size_t i = 0; g_os_version_lut[i].version != MUtils::OS::Version::UNKNOWN_OPSYS; i++) + { + if(os_version == g_os_version_lut[i].version) + { + return g_os_version_lut[i].friendlyName; + } + } + + return NULL; } /////////////////////////////////////////////////////////////////////////////// @@ -264,11 +481,11 @@ static volatile bool g_fatal_exit_flag = true; static DWORD WINAPI fatal_exit_helper(LPVOID lpParameter) { - MessageBoxA(NULL, ((LPCSTR) lpParameter), "GURU MEDITATION", MB_OK | MB_ICONERROR | MB_TASKMODAL | MB_TOPMOST | MB_SETFOREGROUND); + MUtils::OS::system_message_err((LPWSTR) lpParameter, L"GURU MEDITATION"); return 0; } -void MUtils::OS::fatal_exit(const char* const errorMessage) +void MUtils::OS::fatal_exit(const wchar_t* const errorMessage) { g_fatal_exit_lock.enter();