OSDN Git Service

dde3443aa6ae04f8638675f5f5f1571ba5f67e3b
[mutilities/MUtilities.git] / src / OSSupport_Win32.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // MuldeR's Utilities for Qt
3 // Copyright (C) 2004-2015 LoRd_MuldeR <MuldeR2@GMX.de>
4 //
5 // This library is free software; you can redistribute it and/or
6 // modify it under the terms of the GNU Lesser General Public
7 // License as published by the Free Software Foundation; either
8 // version 2.1 of the License, or (at your option) any later version.
9 //
10 // This library is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 // Lesser General Public License for more details.
14 //
15 // You should have received a copy of the GNU Lesser General Public
16 // License along with this library; if not, write to the Free Software
17 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
18 //
19 // http://www.gnu.org/licenses/lgpl-2.1.txt
20 //////////////////////////////////////////////////////////////////////////////////
21
22 //Win32 API
23 #define WIN32_LEAN_AND_MEAN 1
24 #include <Windows.h>
25 #include <Psapi.h>
26 #include <Sensapi.h>
27 #include <Shellapi.h>
28 #include <PowrProf.h>
29 #include <Mmsystem.h>
30 #pragma warning(push)
31 #pragma warning(disable:4091) //for MSVC2015
32 #include <ShlObj.h>
33 #pragma warning(pop)
34
35 //CRT
36 #include <io.h>
37
38 //Internal
39 #include <MUtils/Global.h>
40 #include <MUtils/OSSupport.h>
41 #include <MUtils/GUI.h>
42 #include "CriticalSection_Win32.h"
43 #include "Utils_Win32.h"
44
45 //Qt
46 #include <QMap>
47 #include <QReadWriteLock>
48 #include <QDir>
49 #include <QWidget>
50 #include <QProcess>
51
52 //Main thread ID
53 static const DWORD g_main_thread_id = GetCurrentThreadId();
54
55 ///////////////////////////////////////////////////////////////////////////////
56 // SYSTEM MESSAGE
57 ///////////////////////////////////////////////////////////////////////////////
58
59 static const UINT g_msgBoxFlags = MB_TOPMOST | MB_TASKMODAL | MB_SETFOREGROUND;
60
61 void MUtils::OS::system_message_nfo(const wchar_t *const title, const wchar_t *const text)
62 {
63         MessageBoxW(NULL, text, title, g_msgBoxFlags | MB_ICONINFORMATION);
64 }
65
66 void MUtils::OS::system_message_wrn(const wchar_t *const title, const wchar_t *const text)
67 {
68         MessageBoxW(NULL, text, title, g_msgBoxFlags | MB_ICONWARNING);
69 }
70
71 void MUtils::OS::system_message_err(const wchar_t *const title, const wchar_t *const text)
72 {
73         MessageBoxW(NULL, text, title, g_msgBoxFlags | MB_ICONERROR);
74 }
75
76 ///////////////////////////////////////////////////////////////////////////////
77 // FETCH CLI ARGUMENTS
78 ///////////////////////////////////////////////////////////////////////////////
79
80 static QReadWriteLock                          g_arguments_lock;
81 static QScopedPointer<MUtils::OS::ArgumentMap> g_arguments_list;
82
83 const QStringList MUtils::OS::crack_command_line(const QString &command_line)
84 {
85         int nArgs = 0;
86         LPWSTR *szArglist = CommandLineToArgvW(command_line.isNull() ? GetCommandLineW() : MUTILS_WCHR(command_line), &nArgs);
87
88         QStringList command_line_tokens;
89         if(NULL != szArglist)
90         {
91                 for(int i = 0; i < nArgs; i++)
92                 {
93                         const QString argStr = MUTILS_QSTR(szArglist[i]).trimmed();
94                         if(!argStr.isEmpty())
95                         {
96                                 command_line_tokens << argStr;
97                         }
98                 }
99                 LocalFree(szArglist);
100         }
101
102         return command_line_tokens;
103 }
104
105 const MUtils::OS::ArgumentMap &MUtils::OS::arguments(void)
106 {
107         QReadLocker readLock(&g_arguments_lock);
108
109         //Already initialized?
110         if(!g_arguments_list.isNull())
111         {
112                 return (*(g_arguments_list.data()));
113         }
114
115         readLock.unlock();
116         QWriteLocker writeLock(&g_arguments_lock);
117
118         //Still not initialized?
119         if(!g_arguments_list.isNull())
120         {
121                 return (*(g_arguments_list.data()));
122         }
123
124         g_arguments_list.reset(new ArgumentMap());
125         const QStringList argList = crack_command_line();
126
127         if(!argList.isEmpty())
128         {
129                 const QString argPrefix = QLatin1String("--");
130                 const QChar   separator = QLatin1Char('=');
131
132                 bool firstToken = true;
133                 for(QStringList::ConstIterator iter = argList.constBegin(); iter != argList.constEnd(); iter++)
134                 {
135                         if(firstToken)
136                         {
137                                 firstToken = false;
138                                 continue; /*skip executable file name*/
139                         }
140                         if(iter->startsWith(argPrefix))
141                         {
142                                 const QString argData = iter->mid(2).trimmed();
143                                 if(argData.length() > 0)
144                                 {
145                                         const int separatorIndex = argData.indexOf(separator);
146                                         if(separatorIndex > 0)
147                                         {
148                                                 const QString argKey = argData.left(separatorIndex).trimmed();
149                                                 const QString argVal = argData.mid(separatorIndex + 1).trimmed();
150                                                 g_arguments_list->insertMulti(argKey.toLower(), argVal);
151                                         }
152                                         else
153                                         {
154                                                 g_arguments_list->insertMulti(argData.toLower(), QString());
155                                         }
156                                 }
157                         }
158                 }
159         }
160         else if(argList.empty())
161         {
162                 qWarning("CommandLineToArgvW() has failed !!!");
163         }
164
165         return (*(g_arguments_list.data()));
166 }
167
168 ///////////////////////////////////////////////////////////////////////////////
169 // COPY FILE
170 ///////////////////////////////////////////////////////////////////////////////
171
172 typedef struct _progress_callback_data_t
173 {
174         MUtils::OS::progress_callback_t callback_function;
175         void *user_data;
176 }
177 progress_callback_data_t;
178
179 static DWORD __stdcall copy_file_progress(LARGE_INTEGER TotalFileSize, LARGE_INTEGER TotalBytesTransferred, LARGE_INTEGER StreamSize, LARGE_INTEGER StreamBytesTransferred, DWORD dwStreamNumber, DWORD dwCallbackReason, HANDLE hSourceFile, HANDLE hDestinationFile, LPVOID lpData)
180 {
181         if(const progress_callback_data_t *data = (progress_callback_data_t*) lpData)
182         {
183                 const double progress = qBound(0.0, double(TotalBytesTransferred.QuadPart) / double(TotalFileSize.QuadPart), 1.0);
184                 return data->callback_function(progress, data->user_data) ? PROGRESS_CONTINUE : PROGRESS_CANCEL;
185         }
186         return PROGRESS_CONTINUE;
187 }
188
189 MUTILS_API bool MUtils::OS::copy_file(const QString &sourcePath, const QString &outputPath, const bool &overwrite, const progress_callback_t callback, void *const userData)
190 {
191         progress_callback_data_t callback_data = { callback, userData };
192         BOOL cancel = FALSE;
193         const BOOL result = CopyFileExW(MUTILS_WCHR(QDir::toNativeSeparators(sourcePath)), MUTILS_WCHR(QDir::toNativeSeparators(outputPath)), ((callback_data.callback_function) ? copy_file_progress : NULL), ((callback_data.callback_function) ? &callback_data : NULL), &cancel, (overwrite ? 0 : COPY_FILE_FAIL_IF_EXISTS));
194
195         if(result == FALSE)
196         {
197                 const DWORD errorCode = GetLastError();
198                 if(errorCode != ERROR_REQUEST_ABORTED)
199                 {
200                         qWarning("CopyFile() failed with error code 0x%08X!", errorCode);
201                 }
202                 else
203                 {
204                         qWarning("CopyFile() operation was abroted by user!");
205                 }
206         }
207
208         return (result != FALSE);
209 }
210
211 ///////////////////////////////////////////////////////////////////////////////
212 // GET FILE VERSION
213 ///////////////////////////////////////////////////////////////////////////////
214
215 static bool get_file_version_helper(const QString fileName, PVOID buffer, const size_t &size, quint16 *const major, quint16 *const minor, quint16 *const patch, quint16 *const build)
216 {
217         if(!GetFileVersionInfo(MUTILS_WCHR(fileName), 0, size, buffer))
218         {
219                 qWarning("GetFileVersionInfo() has failed, file version cannot be determined!");
220                 return false;
221         }
222
223         VS_FIXEDFILEINFO *verInfo;
224         UINT verInfoLen;
225         if(!VerQueryValue(buffer, L"\\", (LPVOID*)(&verInfo), &verInfoLen))
226         {
227                 qWarning("VerQueryValue() has failed, file version cannot be determined!");
228                 return false;
229         }
230
231         if(major) *major = quint16((verInfo->dwFileVersionMS >> 16) & 0x0000FFFF);
232         if(minor) *minor = quint16((verInfo->dwFileVersionMS)       & 0x0000FFFF);
233         if(patch) *patch = quint16((verInfo->dwFileVersionLS >> 16) & 0x0000FFFF);
234         if(build) *build = quint16((verInfo->dwFileVersionLS)       & 0x0000FFFF);
235
236         return true;
237 }
238
239 bool MUtils::OS::get_file_version(const QString fileName, quint16 *const major, quint16 *const minor, quint16 *const patch, quint16 *const build)
240 {
241         if(major) *major = 0U; if(minor) *minor = 0U;
242         if(patch) *patch = 0U; if(build) *build = 0U;
243
244         const DWORD size = GetFileVersionInfoSize(MUTILS_WCHR(fileName), NULL);
245         if(size < 1)
246         {
247                 qWarning("GetFileVersionInfoSize() has failed, file version cannot be determined!");
248                 return false;
249         }
250         
251         PVOID buffer = _malloca(size);
252         if(!buffer)
253         {
254                 qWarning("Memory allocation has failed!");
255                 return false;
256         }
257
258         const bool success = get_file_version_helper(fileName, buffer, size, major, minor, patch, build);
259
260         _freea(buffer);
261         return success;
262 }
263
264 ///////////////////////////////////////////////////////////////////////////////
265 // OS VERSION DETECTION
266 ///////////////////////////////////////////////////////////////////////////////
267
268 static bool g_os_version_initialized = false;
269 static MUtils::OS::Version::os_version_t g_os_version_info = MUtils::OS::Version::UNKNOWN_OPSYS;
270 static QReadWriteLock g_os_version_lock;
271
272 //Maps marketing names to the actual Windows NT versions
273 static const struct
274 {
275         MUtils::OS::Version::os_version_t version;
276         const char friendlyName[64];
277 }
278 g_os_version_lut[] =
279 {
280         { MUtils::OS::Version::WINDOWS_WIN2K, "Windows 2000"                                  },        //2000
281         { MUtils::OS::Version::WINDOWS_WINXP, "Windows XP or Windows XP Media Center Edition" },        //XP
282         { MUtils::OS::Version::WINDOWS_XPX64, "Windows Server 2003 or Windows XP x64"         },        //XP_x64
283         { MUtils::OS::Version::WINDOWS_VISTA, "Windows Vista or Windows Server 2008"          },        //Vista
284         { MUtils::OS::Version::WINDOWS_WIN70, "Windows 7 or Windows Server 2008 R2"           },        //7
285         { MUtils::OS::Version::WINDOWS_WIN80, "Windows 8 or Windows Server 2012"              },        //8
286         { MUtils::OS::Version::WINDOWS_WIN81, "Windows 8.1 or Windows Server 2012 R2"         },        //8.1
287         { MUtils::OS::Version::WINDOWS_WN100, "Windows 10 or Windows Server 2014 (Preview)"   },        //10
288         { MUtils::OS::Version::UNKNOWN_OPSYS, "N/A" }
289 };
290
291 //OS version data dtructures
292 namespace MUtils
293 {
294         namespace OS
295         {
296                 namespace Version
297                 {
298                         //Comparision operators for os_version_t
299                         bool os_version_t::operator== (const os_version_t &rhs) const { return (type == rhs.type) && (versionMajor == rhs.versionMajor) && ((versionMinor == rhs.versionMinor)); }
300                         bool os_version_t::operator!= (const os_version_t &rhs) const { return (type != rhs.type) || (versionMajor != rhs.versionMajor) || ((versionMinor != rhs.versionMinor)); }
301                         bool os_version_t::operator>  (const os_version_t &rhs) const { return (type == rhs.type) && ((versionMajor > rhs.versionMajor) || ((versionMajor == rhs.versionMajor) && (versionMinor >  rhs.versionMinor))); }
302                         bool os_version_t::operator>= (const os_version_t &rhs) const { return (type == rhs.type) && ((versionMajor > rhs.versionMajor) || ((versionMajor == rhs.versionMajor) && (versionMinor >= rhs.versionMinor))); }
303                         bool os_version_t::operator<  (const os_version_t &rhs) const { return (type == rhs.type) && ((versionMajor < rhs.versionMajor) || ((versionMajor == rhs.versionMajor) && (versionMinor <  rhs.versionMinor))); }
304                         bool os_version_t::operator<= (const os_version_t &rhs) const { return (type == rhs.type) && ((versionMajor < rhs.versionMajor) || ((versionMajor == rhs.versionMajor) && (versionMinor <= rhs.versionMinor))); }
305
306                         //Known Windows NT versions
307                         const os_version_t WINDOWS_WIN2K = { OS_WINDOWS,  5, 0 };       // 2000
308                         const os_version_t WINDOWS_WINXP = { OS_WINDOWS,  5, 1 };       // XP
309                         const os_version_t WINDOWS_XPX64 = { OS_WINDOWS,  5, 2 };       // XP_x64
310                         const os_version_t WINDOWS_VISTA = { OS_WINDOWS,  6, 0 };       // Vista
311                         const os_version_t WINDOWS_WIN70 = { OS_WINDOWS,  6, 1 };       // 7
312                         const os_version_t WINDOWS_WIN80 = { OS_WINDOWS,  6, 2 };       // 8
313                         const os_version_t WINDOWS_WIN81 = { OS_WINDOWS,  6, 3 };       // 8.1
314                         const os_version_t WINDOWS_WN100 = { OS_WINDOWS, 10, 0 };       // 10
315
316                         //Unknown OS
317                         const os_version_t UNKNOWN_OPSYS = { OS_UNKNOWN, 0, 0 };        // N/A
318                 }
319         }
320 }
321
322 static bool verify_os_version(const DWORD major, const DWORD minor)
323 {
324         OSVERSIONINFOEXW osvi;
325         DWORDLONG dwlConditionMask = 0;
326
327         //Initialize the OSVERSIONINFOEX structure
328         memset(&osvi, 0, sizeof(OSVERSIONINFOEXW));
329         osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEXW);
330         osvi.dwMajorVersion = major;
331         osvi.dwMinorVersion = minor;
332         osvi.dwPlatformId = VER_PLATFORM_WIN32_NT;
333
334         //Initialize the condition mask
335         VER_SET_CONDITION(dwlConditionMask, VER_MAJORVERSION, VER_GREATER_EQUAL);
336         VER_SET_CONDITION(dwlConditionMask, VER_MINORVERSION, VER_GREATER_EQUAL);
337         VER_SET_CONDITION(dwlConditionMask, VER_PLATFORMID,   VER_EQUAL);
338
339         // Perform the test
340         const BOOL ret = VerifyVersionInfoW(&osvi, VER_MAJORVERSION | VER_MINORVERSION | VER_PLATFORMID, dwlConditionMask);
341
342         //Error checking
343         if(!ret)
344         {
345                 if(GetLastError() != ERROR_OLD_WIN_VERSION)
346                 {
347                         qWarning("VerifyVersionInfo() system call has failed!");
348                 }
349         }
350
351         return (ret != FALSE);
352 }
353
354 static bool get_real_os_version(unsigned int *major, unsigned int *minor, bool *pbOverride)
355 {
356         *major = *minor = 0;
357         *pbOverride = false;
358         
359         //Initialize local variables
360         OSVERSIONINFOEXW osvi;
361         memset(&osvi, 0, sizeof(OSVERSIONINFOEXW));
362         osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEXW);
363
364         //Try GetVersionEx() first
365         if(GetVersionExW((LPOSVERSIONINFOW)&osvi) == FALSE)
366         {
367                 qWarning("GetVersionEx() has failed, cannot detect Windows version!");
368                 return false;
369         }
370
371         //Make sure we are running on NT
372         if(osvi.dwPlatformId == VER_PLATFORM_WIN32_NT)
373         {
374                 *major = osvi.dwMajorVersion;
375                 *minor = osvi.dwMinorVersion;
376         }
377         else
378         {
379                 qWarning("Not running on Windows NT, unsupported operating system!");
380                 return false;
381         }
382
383         //Determine the real *major* version first
384         forever
385         {
386                 const DWORD nextMajor = (*major) + 1;
387                 if(verify_os_version(nextMajor, 0))
388                 {
389                         *pbOverride = true;
390                         *major = nextMajor;
391                         *minor = 0;
392                         continue;
393                 }
394                 break;
395         }
396
397         //Now also determine the real *minor* version
398         forever
399         {
400                 const DWORD nextMinor = (*minor) + 1;
401                 if(verify_os_version((*major), nextMinor))
402                 {
403                         *pbOverride = true;
404                         *minor = nextMinor;
405                         continue;
406                 }
407                 break;
408         }
409
410         //Workaround for the mess that is sometimes referred to as "Windows 10"
411         if(((*major) > 6) || (((*major) == 6) && ((*minor) >= 2)))
412         {
413                 quint16 kernel32_major, kernel32_minor;
414                 if(MUtils::OS::get_file_version(QLatin1String("kernel32"), &kernel32_major, &kernel32_minor))
415                 {
416                         if((kernel32_major > (*major)) || ((kernel32_major == (*major)) && (kernel32_minor > (*minor))))
417                         {
418                                 *major = kernel32_major;
419                                 *minor = kernel32_minor;
420                                 *pbOverride = true;
421                         }
422                 }
423         }
424
425         return true;
426 }
427
428 const MUtils::OS::Version::os_version_t &MUtils::OS::os_version(void)
429 {
430         QReadLocker readLock(&g_os_version_lock);
431
432         //Already initialized?
433         if(g_os_version_initialized)
434         {
435                 return g_os_version_info;
436         }
437         
438         readLock.unlock();
439         QWriteLocker writeLock(&g_os_version_lock);
440
441         //Initialized now?
442         if(g_os_version_initialized)
443         {
444                 return g_os_version_info;
445         }
446
447         //Detect OS version
448         unsigned int major, minor; bool overrideFlg;
449         if(get_real_os_version(&major, &minor, &overrideFlg))
450         {
451                 g_os_version_info.type = Version::OS_WINDOWS;
452                 g_os_version_info.versionMajor = major;
453                 g_os_version_info.versionMinor = minor;
454                 g_os_version_info.overrideFlag = overrideFlg;
455         }
456         else
457         {
458                 qWarning("Failed to determin the operating system version!");
459         }
460
461         //Completed
462         g_os_version_initialized = true;
463         return g_os_version_info;
464 }
465
466 const char *MUtils::OS::os_friendly_name(const MUtils::OS::Version::os_version_t &os_version)
467 {
468         for(size_t i = 0; g_os_version_lut[i].version != MUtils::OS::Version::UNKNOWN_OPSYS; i++)
469         {
470                 if(os_version == g_os_version_lut[i].version)
471                 {
472                         return g_os_version_lut[i].friendlyName;
473                 }
474         }
475
476         return NULL;
477 }
478
479 ///////////////////////////////////////////////////////////////////////////////
480 // WINE DETECTION
481 ///////////////////////////////////////////////////////////////////////////////
482
483 static bool g_wine_deteced = false;
484 static bool g_wine_initialized = false;
485 static QReadWriteLock g_wine_lock;
486
487 static const bool detect_wine(void)
488 {
489         void *const ptr = MUtils::Win32Utils::resolve<void*>(QLatin1String("ntdll"), QLatin1String("wine_get_version"));
490         return (ptr != NULL);
491 }
492
493 const bool &MUtils::OS::running_on_wine(void)
494 {
495         QReadLocker readLock(&g_wine_lock);
496
497         //Already initialized?
498         if(g_wine_initialized)
499         {
500                 return g_wine_deteced;
501         }
502
503         readLock.unlock();
504         QWriteLocker writeLock(&g_wine_lock);
505
506         //Initialized now?
507         if(g_wine_initialized)
508         {
509                 return g_wine_deteced;
510         }
511
512         //Try to detect Wine
513         g_wine_deteced = detect_wine();
514         g_wine_initialized = true;
515
516         return g_wine_deteced;
517 }
518
519 ///////////////////////////////////////////////////////////////////////////////
520 // KNWON FOLDERS
521 ///////////////////////////////////////////////////////////////////////////////
522
523 typedef QMap<size_t, QString> KFMap;
524 typedef HRESULT (WINAPI *SHGetKnownFolderPath_t)(const GUID &rfid, DWORD dwFlags, HANDLE hToken, PWSTR *ppszPath);
525 typedef HRESULT (WINAPI *SHGetFolderPath_t)     (HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPWSTR pszPath);
526
527 static QScopedPointer<KFMap>  g_known_folders_map;
528 static QReadWriteLock         g_known_folders_lock;
529
530 const QString &MUtils::OS::known_folder(known_folder_t folder_id)
531 {
532         typedef enum { KF_FLAG_CREATE = 0x00008000 } kf_flags_t;
533         
534         struct
535         {
536                 const int csidl;
537                 const GUID guid;
538         }
539         static s_folders[] =
540         {
541                 { 0x001c, {0xF1B32785,0x6FBA,0x4FCF,{0x9D,0x55,0x7B,0x8E,0x7F,0x15,0x70,0x91}} },  //CSIDL_LOCAL_APPDATA
542                 { 0x0026, {0x905e63b6,0xc1bf,0x494e,{0xb2,0x9c,0x65,0xb7,0x32,0xd3,0xd2,0x1a}} },  //CSIDL_PROGRAM_FILES
543                 { 0x0024, {0xF38BF404,0x1D43,0x42F2,{0x93,0x05,0x67,0xDE,0x0B,0x28,0xFC,0x23}} },  //CSIDL_WINDOWS_FOLDER
544                 { 0x0025, {0x1AC14E77,0x02E7,0x4E5D,{0xB7,0x44,0x2E,0xB1,0xAE,0x51,0x98,0xB7}} },  //CSIDL_SYSTEM_FOLDER
545         };
546
547         size_t folderId = size_t(-1);
548
549         switch(folder_id)
550         {
551                 case FOLDER_LOCALAPPDATA: folderId = 0; break;
552                 case FOLDER_PROGRAMFILES: folderId = 1; break;
553                 case FOLDER_SYSTROOT_DIR: folderId = 2; break;
554                 case FOLDER_SYSTEMFOLDER: folderId = 3; break;
555         }
556
557         if(folderId == size_t(-1))
558         {
559                 qWarning("Invalid 'known' folder was requested!");
560                 return *reinterpret_cast<QString*>(NULL);
561         }
562
563         QReadLocker readLock(&g_known_folders_lock);
564
565         //Already in cache?
566         if(!g_known_folders_map.isNull())
567         {
568                 if(g_known_folders_map->contains(folderId))
569                 {
570                         return g_known_folders_map->operator[](folderId);
571                 }
572         }
573
574         //Obtain write lock to initialize
575         readLock.unlock();
576         QWriteLocker writeLock(&g_known_folders_lock);
577
578         //Still not in cache?
579         if(!g_known_folders_map.isNull())
580         {
581                 if(g_known_folders_map->contains(folderId))
582                 {
583                         return g_known_folders_map->operator[](folderId);
584                 }
585         }
586
587         //Initialize on first call
588         if(g_known_folders_map.isNull())
589         {
590                 g_known_folders_map.reset(new QMap<size_t, QString>());
591         }
592
593         QString folderPath;
594
595         //Now try to get the folder path!
596         if(const SHGetKnownFolderPath_t known_folders_fpGetKnownFolderPath = MUtils::Win32Utils::resolve<SHGetKnownFolderPath_t>(QLatin1String("shell32"), QLatin1String("SHGetKnownFolderPath")))
597         {
598                 WCHAR *path = NULL;
599                 if(known_folders_fpGetKnownFolderPath(s_folders[folderId].guid, KF_FLAG_CREATE, NULL, &path) == S_OK)
600                 {
601                         //MessageBoxW(0, path, L"SHGetKnownFolderPath", MB_TOPMOST);
602                         QDir folderTemp = QDir(QDir::fromNativeSeparators(MUTILS_QSTR(path)));
603                         if(folderTemp.exists())
604                         {
605                                 folderPath = folderTemp.canonicalPath();
606                         }
607                         CoTaskMemFree(path);
608                 }
609         }
610         else if(const SHGetFolderPath_t known_folders_fpGetFolderPath = MUtils::Win32Utils::resolve<SHGetFolderPath_t>(QLatin1String("shell32"), QLatin1String("SHGetFolderPathW")))
611         {
612                 QScopedArrayPointer<WCHAR> path(new WCHAR[4096]);
613                 if(known_folders_fpGetFolderPath(NULL, s_folders[folderId].csidl | CSIDL_FLAG_CREATE, NULL, NULL, path.data()) == S_OK)
614                 {
615                         //MessageBoxW(0, path, L"SHGetFolderPathW", MB_TOPMOST);
616                         QDir folderTemp = QDir(QDir::fromNativeSeparators(MUTILS_QSTR(path.data())));
617                         if(folderTemp.exists())
618                         {
619                                 folderPath = folderTemp.canonicalPath();
620                         }
621                 }
622         }
623
624         //Update cache
625         g_known_folders_map->insert(folderId, folderPath);
626         return g_known_folders_map->operator[](folderId);
627 }
628
629 ///////////////////////////////////////////////////////////////////////////////
630 // CURRENT DATA & TIME
631 ///////////////////////////////////////////////////////////////////////////////
632
633 QDate MUtils::OS::current_date(void)
634 {
635         const DWORD MAX_PROC = 1024;
636         QScopedArrayPointer<DWORD> processes(new DWORD[MAX_PROC]);
637         DWORD bytesReturned = 0;
638         
639         if(!EnumProcesses(processes.data(), sizeof(DWORD) * MAX_PROC, &bytesReturned))
640         {
641                 return QDate::currentDate();
642         }
643
644         const DWORD procCount = bytesReturned / sizeof(DWORD);
645         ULARGE_INTEGER lastStartTime;
646         memset(&lastStartTime, 0, sizeof(ULARGE_INTEGER));
647
648         for(DWORD i = 0; i < procCount; i++)
649         {
650                 if(HANDLE hProc = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, processes[i]))
651                 {
652                         FILETIME processTime[4];
653                         if(GetProcessTimes(hProc, &processTime[0], &processTime[1], &processTime[2], &processTime[3]))
654                         {
655                                 ULARGE_INTEGER timeCreation;
656                                 timeCreation.LowPart  = processTime[0].dwLowDateTime;
657                                 timeCreation.HighPart = processTime[0].dwHighDateTime;
658                                 if(timeCreation.QuadPart > lastStartTime.QuadPart)
659                                 {
660                                         lastStartTime.QuadPart = timeCreation.QuadPart;
661                                 }
662                         }
663                         CloseHandle(hProc);
664                 }
665         }
666         
667         FILETIME lastStartTime_fileTime;
668         lastStartTime_fileTime.dwHighDateTime = lastStartTime.HighPart;
669         lastStartTime_fileTime.dwLowDateTime  = lastStartTime.LowPart;
670
671         FILETIME lastStartTime_localTime;
672         if(!FileTimeToLocalFileTime(&lastStartTime_fileTime, &lastStartTime_localTime))
673         {
674                 memcpy(&lastStartTime_localTime, &lastStartTime_fileTime, sizeof(FILETIME));
675         }
676         
677         SYSTEMTIME lastStartTime_system;
678         if(!FileTimeToSystemTime(&lastStartTime_localTime, &lastStartTime_system))
679         {
680                 memset(&lastStartTime_system, 0, sizeof(SYSTEMTIME));
681                 lastStartTime_system.wYear = 1970; lastStartTime_system.wMonth = lastStartTime_system.wDay = 1;
682         }
683
684         const QDate currentDate = QDate::currentDate();
685         const QDate processDate = QDate(lastStartTime_system.wYear, lastStartTime_system.wMonth, lastStartTime_system.wDay);
686         return (currentDate >= processDate) ? currentDate : processDate;
687 }
688
689 quint64 MUtils::OS::current_file_time(void)
690 {
691         FILETIME fileTime;
692         GetSystemTimeAsFileTime(&fileTime);
693
694         ULARGE_INTEGER temp;
695         temp.HighPart = fileTime.dwHighDateTime;
696         temp.LowPart = fileTime.dwLowDateTime;
697
698         return temp.QuadPart;
699 }
700
701 ///////////////////////////////////////////////////////////////////////////////
702 // FILE PATH FROM FD
703 ///////////////////////////////////////////////////////////////////////////////
704
705 typedef DWORD(_stdcall *GetPathNameByHandleFun)(HANDLE hFile, LPWSTR lpszFilePath, DWORD cchFilePath, DWORD dwFlags);
706
707 static QString get_file_path_drive_list(void)
708 {
709         QString list;
710         const DWORD len = GetLogicalDriveStringsW(0, NULL);
711         if (len > 0)
712         {
713                 if (wchar_t *const buffer = (wchar_t*) _malloca(sizeof(wchar_t) * len))
714                 {
715                         const DWORD ret = GetLogicalDriveStringsW(len, buffer);
716                         if ((ret > 0) && (ret < len))
717                         {
718                                 const wchar_t *ptr = buffer;
719                                 while (const size_t current_len = wcslen(ptr))
720                                 {
721                                         list.append(QChar(*reinterpret_cast<const ushort*>(ptr)));
722                                         ptr += (current_len + 1);
723                                 }
724                         }
725                         _freea(buffer);
726                 }
727         }
728         return list;
729 }
730
731 static QString &get_file_path_translate(QString &path)
732 {
733         static const DWORD BUFSIZE = 4096;
734         wchar_t buffer[BUFSIZE], drive[3];
735
736         const QString driveList = get_file_path_drive_list();
737         wcscpy_s(drive, 3, L"?:");
738         for (const wchar_t *current = MUTILS_WCHR(driveList); *current; current++)
739         {
740                 drive[0] = (*current);
741                 if (QueryDosDeviceW(drive, buffer, MAX_PATH))
742                 {
743                         const QString prefix = MUTILS_QSTR(buffer);
744                         if (path.startsWith(prefix, Qt::CaseInsensitive))
745                         {
746                                 path.remove(0, prefix.length()).prepend(QLatin1Char(':')).prepend(QChar(*reinterpret_cast<const ushort*>(current)));
747                                 break;
748                         }
749                 }
750         }
751
752         return path;
753 }
754
755 static QString get_file_path_fallback(const HANDLE &hFile)
756 {
757         QString filePath;
758
759         const HANDLE hFileMap = CreateFileMappingW(hFile, NULL, PAGE_READONLY, 0, 1, NULL);
760         if (hFileMap)
761         {
762                 void* pMem = MapViewOfFile(hFileMap, FILE_MAP_READ, 0, 0, 1);
763                 if (pMem)
764                 {
765                         static const DWORD BUFSIZE = 4096;
766                         wchar_t buffer[BUFSIZE];
767                         if (GetMappedFileNameW(GetCurrentProcess(), pMem, buffer, BUFSIZE))
768                         {
769                                 filePath = get_file_path_translate(MUTILS_QSTR(buffer));
770                         }
771                         UnmapViewOfFile(pMem);
772                 }
773                 CloseHandle(hFileMap);
774         }
775
776         return filePath;
777 }
778
779 QString MUtils::OS::get_file_path(const int &fd)
780 {
781         if (fd >= 0)
782         {
783                 const GetPathNameByHandleFun getPathNameByHandleFun = MUtils::Win32Utils::resolve<GetPathNameByHandleFun>(QLatin1String("kernel32"), QLatin1String("GetFinalPathNameByHandleW"));
784                 if (!getPathNameByHandleFun)
785                 {
786                         qWarning("MUtils::OS::get_file_path() --> fallback!");
787                         return get_file_path_fallback((HANDLE)_get_osfhandle(fd));
788                 }
789
790                 const HANDLE handle = (HANDLE) _get_osfhandle(fd);
791                 const DWORD len = getPathNameByHandleFun(handle, NULL, 0, FILE_NAME_OPENED);
792                 if (len > 0)
793                 {
794                         if (wchar_t *const buffer = (wchar_t*)_malloca(sizeof(wchar_t) * len))
795                         {
796                                 const DWORD ret = getPathNameByHandleFun(handle, buffer, len, FILE_NAME_OPENED);
797                                 if ((ret > 0) && (ret < len))
798                                 {
799                                         const QString path(MUTILS_QSTR(buffer));
800                                         return path.startsWith(QLatin1String("\\\\?\\")) ? path.mid(4) : path;
801                                 }
802                                 _freea(buffer);
803                         }
804                 }
805         }
806
807         return QString();
808 }
809
810 ///////////////////////////////////////////////////////////////////////////////
811 // PROCESS ELEVATION
812 ///////////////////////////////////////////////////////////////////////////////
813
814 static bool user_is_admin_helper(void)
815 {
816         HANDLE hToken = NULL;
817         if(!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
818         {
819                 return false;
820         }
821
822         DWORD dwSize = 0;
823         if(!GetTokenInformation(hToken, TokenGroups, NULL, 0, &dwSize))
824         {
825                 if(GetLastError() != ERROR_INSUFFICIENT_BUFFER)
826                 {
827                         CloseHandle(hToken);
828                         return false;
829                 }
830         }
831
832         PTOKEN_GROUPS lpGroups = (PTOKEN_GROUPS) malloc(dwSize);
833         if(!lpGroups)
834         {
835                 CloseHandle(hToken);
836                 return false;
837         }
838
839         if(!GetTokenInformation(hToken, TokenGroups, lpGroups, dwSize, &dwSize))
840         {
841                 free(lpGroups);
842                 CloseHandle(hToken);
843                 return false;
844         }
845
846         PSID lpSid = NULL; SID_IDENTIFIER_AUTHORITY Authority = {SECURITY_NT_AUTHORITY};
847         if(!AllocateAndInitializeSid(&Authority, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &lpSid))
848         {
849                 free(lpGroups);
850                 CloseHandle(hToken);
851                 return false;
852         }
853
854         bool bResult = false;
855         for(DWORD i = 0; i < lpGroups->GroupCount; i++)
856         {
857                 if(EqualSid(lpSid, lpGroups->Groups[i].Sid))
858                 {
859                         bResult = true;
860                         break;
861                 }
862         }
863
864         FreeSid(lpSid);
865         free(lpGroups);
866         CloseHandle(hToken);
867         return bResult;
868 }
869
870 bool MUtils::OS::is_elevated(bool *bIsUacEnabled)
871 {
872         if(bIsUacEnabled)
873         {
874                 *bIsUacEnabled = false;
875         }
876
877         bool bIsProcessElevated = false;
878         HANDLE hToken = NULL;
879         
880         if(OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
881         {
882                 TOKEN_ELEVATION_TYPE tokenElevationType;
883                 DWORD returnLength;
884                 if(GetTokenInformation(hToken, TokenElevationType, &tokenElevationType, sizeof(TOKEN_ELEVATION_TYPE), &returnLength))
885                 {
886                         if(returnLength == sizeof(TOKEN_ELEVATION_TYPE))
887                         {
888                                 switch(tokenElevationType)
889                                 {
890                                 case TokenElevationTypeDefault:
891                                         qDebug("Process token elevation type: Default -> UAC is disabled.\n");
892                                         break;
893                                 case TokenElevationTypeFull:
894                                         qWarning("Process token elevation type: Full -> potential security risk!\n");
895                                         bIsProcessElevated = true;
896                                         if(bIsUacEnabled) *bIsUacEnabled = true;
897                                         break;
898                                 case TokenElevationTypeLimited:
899                                         qDebug("Process token elevation type: Limited -> not elevated.\n");
900                                         if(bIsUacEnabled) *bIsUacEnabled = true;
901                                         break;
902                                 default:
903                                         qWarning("Unknown tokenElevationType value: %d", tokenElevationType);
904                                         break;
905                                 }
906                         }
907                         else
908                         {
909                                 qWarning("GetTokenInformation() return an unexpected size!");
910                         }
911                 }
912                 CloseHandle(hToken);
913         }
914         else
915         {
916                 qWarning("Failed to open process token!");
917         }
918
919         return bIsProcessElevated;
920 }
921
922 bool MUtils::OS::user_is_admin(void)
923 {
924         bool isAdmin = false;
925
926         //Check for process elevation and UAC support first!
927         if(MUtils::OS::is_elevated(&isAdmin))
928         {
929                 qWarning("Process is elevated -> user is admin!");
930                 return true;
931         }
932         
933         //If not elevated and UAC is not available -> user must be in admin group!
934         if(!isAdmin)
935         {
936                 qDebug("UAC is disabled/unavailable -> checking for Administrators group");
937                 isAdmin = user_is_admin_helper();
938         }
939
940         return isAdmin;
941 }
942
943 ///////////////////////////////////////////////////////////////////////////////
944 // NETWORK STATE
945 ///////////////////////////////////////////////////////////////////////////////
946
947 int MUtils::OS::network_status(void)
948 {
949         DWORD dwFlags;
950         const BOOL ret = IsNetworkAlive(&dwFlags);
951         if(GetLastError() == 0)
952         {
953                 return (ret != FALSE) ? NETWORK_TYPE_YES : NETWORK_TYPE_NON;
954         }
955         return NETWORK_TYPE_ERR;
956 }
957
958 ///////////////////////////////////////////////////////////////////////////////
959 // MESSAGE HANDLER
960 ///////////////////////////////////////////////////////////////////////////////
961
962 bool MUtils::OS::handle_os_message(const void *const message, long *result)
963 {
964         const MSG *const msg = reinterpret_cast<const MSG*>(message);
965
966         switch(msg->message)
967         {
968         case WM_QUERYENDSESSION:
969                 qWarning("WM_QUERYENDSESSION message received!");
970                 *result = MUtils::GUI::broadcast(MUtils::GUI::USER_EVENT_QUERYENDSESSION, false) ? TRUE : FALSE;
971                 return true;
972         case WM_ENDSESSION:
973                 qWarning("WM_ENDSESSION message received!");
974                 if(msg->wParam == TRUE)
975                 {
976                         MUtils::GUI::broadcast(MUtils::GUI::USER_EVENT_ENDSESSION, false);
977                         MUtils::GUI::force_quit();
978                         exit(1);
979                 }
980                 *result = 0;
981                 return true;
982         default:
983                 /*ignore this message and let Qt handle it*/
984                 return false;
985         }
986 }
987
988 ///////////////////////////////////////////////////////////////////////////////
989 // SLEEP
990 ///////////////////////////////////////////////////////////////////////////////
991
992 void MUtils::OS::sleep_ms(const size_t &duration)
993 {
994         Sleep((DWORD) duration);
995 }
996
997 ///////////////////////////////////////////////////////////////////////////////
998 // EXECUTABLE CHECK
999 ///////////////////////////////////////////////////////////////////////////////
1000
1001 bool MUtils::OS::is_executable_file(const QString &path)
1002 {
1003         bool bIsExecutable = false;
1004         DWORD binaryType;
1005         if(GetBinaryType(MUTILS_WCHR(QDir::toNativeSeparators(path)), &binaryType))
1006         {
1007                 bIsExecutable = (binaryType == SCS_32BIT_BINARY || binaryType == SCS_64BIT_BINARY);
1008         }
1009         return bIsExecutable;
1010 }
1011
1012 ///////////////////////////////////////////////////////////////////////////////
1013 // HIBERNATION / SHUTDOWN
1014 ///////////////////////////////////////////////////////////////////////////////
1015
1016 bool MUtils::OS::is_hibernation_supported(void)
1017 {
1018         bool hibernationSupported = false;
1019
1020         SYSTEM_POWER_CAPABILITIES pwrCaps;
1021         SecureZeroMemory(&pwrCaps, sizeof(SYSTEM_POWER_CAPABILITIES));
1022         
1023         if(GetPwrCapabilities(&pwrCaps))
1024         {
1025                 hibernationSupported = pwrCaps.SystemS4 && pwrCaps.HiberFilePresent;
1026         }
1027
1028         return hibernationSupported;
1029 }
1030
1031 bool MUtils::OS::shutdown_computer(const QString &message, const unsigned long timeout, const bool forceShutdown, const bool hibernate)
1032 {
1033         HANDLE hToken = NULL;
1034
1035         if(OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
1036         {
1037                 TOKEN_PRIVILEGES privileges;
1038                 memset(&privileges, 0, sizeof(TOKEN_PRIVILEGES));
1039                 privileges.PrivilegeCount = 1;
1040                 privileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
1041                 
1042                 if(LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &privileges.Privileges[0].Luid))
1043                 {
1044                         if(AdjustTokenPrivileges(hToken, FALSE, &privileges, NULL, NULL, NULL))
1045                         {
1046                                 if(hibernate)
1047                                 {
1048                                         if(SetSuspendState(TRUE, TRUE, TRUE))
1049                                         {
1050                                                 return true;
1051                                         }
1052                                 }
1053                                 const DWORD reason = SHTDN_REASON_MAJOR_APPLICATION | SHTDN_REASON_FLAG_PLANNED;
1054                                 return InitiateSystemShutdownEx(NULL, const_cast<wchar_t*>(MUTILS_WCHR(message)), timeout, forceShutdown ? TRUE : FALSE, FALSE, reason);
1055                         }
1056                 }
1057         }
1058         
1059         return false;
1060 }
1061
1062 ///////////////////////////////////////////////////////////////////////////////
1063 // FREE DISKSPACE
1064 ///////////////////////////////////////////////////////////////////////////////
1065
1066 bool MUtils::OS::free_diskspace(const QString &path, quint64 &freeSpace)
1067 {
1068         ULARGE_INTEGER freeBytesAvailable, totalNumberOfBytes, totalNumberOfFreeBytes;
1069         if(GetDiskFreeSpaceExW(reinterpret_cast<const wchar_t*>(QDir::toNativeSeparators(path).utf16()), &freeBytesAvailable, &totalNumberOfBytes, &totalNumberOfFreeBytes))
1070         {
1071                 freeSpace = freeBytesAvailable.QuadPart;
1072                 return true;;
1073         }
1074
1075         freeSpace = -1;
1076         return false;
1077 }
1078
1079 ///////////////////////////////////////////////////////////////////////////////
1080 // SHELL OPEN
1081 ///////////////////////////////////////////////////////////////////////////////
1082
1083 bool MUtils::OS::shell_open(const QWidget *parent, const QString &url, const QString &parameters, const QString &directory, const bool explore)
1084 {
1085         return ((int) ShellExecuteW((parent ? parent->winId() : NULL), (explore ? L"explore" : L"open"), MUTILS_WCHR(url), ((!parameters.isEmpty()) ? MUTILS_WCHR(parameters) : NULL), ((!directory.isEmpty()) ? MUTILS_WCHR(directory) : NULL), SW_SHOW)) > 32;
1086 }
1087
1088 bool MUtils::OS::shell_open(const QWidget *parent, const QString &url, const bool explore)
1089 {
1090         return shell_open(parent, url, QString(), QString(), explore);
1091 }
1092
1093 ///////////////////////////////////////////////////////////////////////////////
1094 // OPEN MEDIA FILE
1095 ///////////////////////////////////////////////////////////////////////////////
1096
1097 bool MUtils::OS::open_media_file(const QString &mediaFilePath)
1098 {
1099         const static wchar_t *registryPrefix[2] = { L"SOFTWARE\\", L"SOFTWARE\\Wow6432Node\\" };
1100         const static wchar_t *registryKeys[3] = 
1101         {
1102                 L"Microsoft\\Windows\\CurrentVersion\\Uninstall\\{97D341C8-B0D1-4E4A-A49A-C30B52F168E9}",
1103                 L"Microsoft\\Windows\\CurrentVersion\\Uninstall\\{DB9E4EAB-2717-499F-8D56-4CC8A644AB60}",
1104                 L"foobar2000"
1105         };
1106         const static wchar_t *appNames[4] = { L"smplayer_portable.exe", L"smplayer.exe", L"MPUI.exe", L"foobar2000.exe" };
1107         const static wchar_t *valueNames[2] = { L"InstallLocation", L"InstallDir" };
1108
1109         for(size_t i = 0; i < 3; i++)
1110         {
1111                 for(size_t j = 0; j < 2; j++)
1112                 {
1113                         QString mplayerPath;
1114                         HKEY registryKeyHandle = NULL;
1115
1116                         const QString currentKey = MUTILS_QSTR(registryPrefix[j]).append(MUTILS_QSTR(registryKeys[i]));
1117                         if(RegOpenKeyExW(HKEY_LOCAL_MACHINE, MUTILS_WCHR(currentKey), 0, KEY_READ, &registryKeyHandle) == ERROR_SUCCESS)
1118                         {
1119                                 for(size_t k = 0; k < 2; k++)
1120                                 {
1121                                         wchar_t Buffer[4096];
1122                                         DWORD BuffSize = sizeof(wchar_t*) * 4096;
1123                                         DWORD DataType = REG_NONE;
1124                                         if(RegQueryValueExW(registryKeyHandle, valueNames[k], 0, &DataType, reinterpret_cast<BYTE*>(Buffer), &BuffSize) == ERROR_SUCCESS)
1125                                         {
1126                                                 if((DataType == REG_SZ) || (DataType == REG_EXPAND_SZ) || (DataType == REG_LINK))
1127                                                 {
1128                                                         mplayerPath = MUTILS_QSTR(Buffer);
1129                                                         break;
1130                                                 }
1131                                         }
1132                                 }
1133                                 RegCloseKey(registryKeyHandle);
1134                         }
1135
1136                         if(!mplayerPath.isEmpty())
1137                         {
1138                                 QDir mplayerDir(mplayerPath);
1139                                 if(mplayerDir.exists())
1140                                 {
1141                                         for(size_t k = 0; k < 4; k++)
1142                                         {
1143                                                 if(mplayerDir.exists(MUTILS_QSTR(appNames[k])))
1144                                                 {
1145                                                         qDebug("Player found at:\n%s\n", MUTILS_UTF8(mplayerDir.absoluteFilePath(MUTILS_QSTR(appNames[k]))));
1146                                                         QProcess::startDetached(mplayerDir.absoluteFilePath(MUTILS_QSTR(appNames[k])), QStringList() << QDir::toNativeSeparators(mediaFilePath));
1147                                                         return true;
1148                                                 }
1149                                         }
1150                                 }
1151                         }
1152                 }
1153         }
1154         return false;
1155 }
1156
1157 ///////////////////////////////////////////////////////////////////////////////
1158 // DEBUGGER CHECK
1159 ///////////////////////////////////////////////////////////////////////////////
1160
1161 static bool change_process_priority_helper(const HANDLE hProcess, const int priority)
1162 {
1163         bool ok = false;
1164
1165         switch(qBound(-2, priority, 2))
1166         {
1167         case 2:
1168                 ok = (SetPriorityClass(hProcess, HIGH_PRIORITY_CLASS) == TRUE);
1169                 break;
1170         case 1:
1171                 if(!(ok = (SetPriorityClass(hProcess, ABOVE_NORMAL_PRIORITY_CLASS) == TRUE)))
1172                 {
1173                         ok = (SetPriorityClass(hProcess, HIGH_PRIORITY_CLASS) == TRUE);
1174                 }
1175                 break;
1176         case 0:
1177                 ok = (SetPriorityClass(hProcess, NORMAL_PRIORITY_CLASS) == TRUE);
1178                 break;
1179         case -1:
1180                 if(!(ok = (SetPriorityClass(hProcess, BELOW_NORMAL_PRIORITY_CLASS) == TRUE)))
1181                 {
1182                         ok = (SetPriorityClass(hProcess, IDLE_PRIORITY_CLASS) == TRUE);
1183                 }
1184                 break;
1185         case -2:
1186                 ok = (SetPriorityClass(hProcess, IDLE_PRIORITY_CLASS) == TRUE);
1187                 break;
1188         }
1189
1190         return ok;
1191 }
1192
1193 bool MUtils::OS::change_process_priority(const int priority)
1194 {
1195         return change_process_priority_helper(GetCurrentProcess(), priority);
1196 }
1197
1198 bool MUtils::OS::change_process_priority(const QProcess *proc, const int priority)
1199 {
1200         if(Q_PID qPid = proc->pid())
1201         {
1202                 return change_process_priority_helper(qPid->hProcess, priority);
1203         }
1204         else
1205         {
1206                 return false;
1207         }
1208 }
1209
1210 ///////////////////////////////////////////////////////////////////////////////
1211 // PROCESS ID
1212 ///////////////////////////////////////////////////////////////////////////////
1213
1214 quint32 MUtils::OS::process_id(const QProcess *const proc)
1215 {
1216         PROCESS_INFORMATION *procInf = proc->pid();
1217         return (procInf) ? procInf->dwProcessId : NULL;
1218 }
1219
1220 ///////////////////////////////////////////////////////////////////////////////
1221 // PROCESS SUSPEND/RESUME
1222 ///////////////////////////////////////////////////////////////////////////////
1223
1224 bool MUtils::OS::suspend_process(const QProcess *proc, const bool suspend)
1225 {
1226         if(Q_PID pid = proc->pid())
1227         {
1228                 if(suspend)
1229                 {
1230                         return (SuspendThread(pid->hThread) != ((DWORD) -1));
1231                 }
1232                 else
1233                 {
1234                         return (ResumeThread(pid->hThread)  != ((DWORD) -1));
1235                 }
1236         }
1237         else
1238         {
1239                 return false;
1240         }
1241 }
1242
1243 ///////////////////////////////////////////////////////////////////////////////
1244 // SYSTEM TIMER
1245 ///////////////////////////////////////////////////////////////////////////////
1246
1247 bool MUtils::OS::setup_timer_resolution(const quint32 &interval)
1248 {
1249         return timeBeginPeriod(interval) == TIMERR_NOERROR;
1250 }
1251
1252 bool MUtils::OS::reset_timer_resolution(const quint32 &interval)
1253 {
1254         return timeEndPeriod(interval) == TIMERR_NOERROR;
1255 }
1256
1257 ///////////////////////////////////////////////////////////////////////////////
1258 // SET FILE TIME
1259 ///////////////////////////////////////////////////////////////////////////////
1260
1261 static QScopedPointer<QDateTime> s_epoch;
1262 static QReadWriteLock            s_epochLock;
1263
1264 static const QDateTime *get_epoch(void)
1265 {
1266         QReadLocker rdLock(&s_epochLock);
1267
1268         if (s_epoch.isNull())
1269         {
1270                 rdLock.unlock();
1271                 QWriteLocker wrLock(&s_epochLock);
1272                 if (s_epoch.isNull())
1273                 {
1274                         s_epoch.reset(new QDateTime(QDate(1601, 1, 1), QTime(0, 0, 0, 0), Qt::UTC));
1275                 }
1276                 wrLock.unlock();
1277                 rdLock.relock();
1278         }
1279
1280         return s_epoch.data();
1281 }
1282
1283 static FILETIME *qt_time_to_file_time(FILETIME *const fileTime, const QDateTime &dateTime)
1284 {
1285         memset(fileTime, 0, sizeof(FILETIME));
1286
1287         if (const QDateTime *const epoch = get_epoch())
1288         {
1289                 const qint64 msecs = epoch->msecsTo(dateTime);
1290                 if (msecs > 0)
1291                 {
1292                         const quint64 ticks = 10000U * quint64(msecs);
1293                         fileTime->dwHighDateTime = ((ticks >> 32) & 0xFFFFFFFF);
1294                         fileTime->dwLowDateTime = (ticks & 0xFFFFFFFF);
1295                         return fileTime;
1296                 }
1297         }
1298
1299         return NULL;
1300 }
1301
1302 static bool set_file_time(const HANDLE hFile, const QDateTime &created, const QDateTime &lastMod, const QDateTime &lastAcc)
1303 {
1304         FILETIME ftCreated, ftLastMod, ftLastAcc;
1305
1306         FILETIME *const pCreated = created.isValid() ? qt_time_to_file_time(&ftCreated, created) : NULL;
1307         FILETIME *const pLastMod = lastMod.isValid() ? qt_time_to_file_time(&ftLastMod, lastMod) : NULL;
1308         FILETIME *const pLastAcc = lastAcc.isValid() ? qt_time_to_file_time(&ftLastAcc, lastAcc) : NULL;
1309
1310         if (pCreated || pLastMod || pLastAcc)
1311         {
1312                 return (SetFileTime(hFile, pCreated, pLastAcc, pLastMod) != FALSE);
1313         }
1314
1315         return false;
1316 }
1317
1318 bool MUtils::OS::set_file_time(const QFile &file, const QDateTime &created, const QDateTime &lastMod, const QDateTime &lastAcc)
1319 {
1320         const int fd = file.handle();
1321         if (fd >= 0)
1322         {
1323                 return set_file_time((HANDLE)_get_osfhandle(fd), created, lastMod, lastAcc);
1324         }
1325         return false;
1326 }
1327
1328 bool MUtils::OS::set_file_time(const QString &path, const QDateTime &created, const QDateTime &lastMod, const QDateTime &lastAcc)
1329 {
1330         const HANDLE hFile = CreateFileW(MUTILS_WCHR(path), FILE_WRITE_ATTRIBUTES, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, 0);
1331         bool okay = false;
1332         if ((hFile != NULL) && (hFile != INVALID_HANDLE_VALUE))
1333         {
1334                 okay = set_file_time(hFile, created, lastMod, lastAcc);
1335                 CloseHandle(hFile);
1336         }
1337         return okay;
1338 }
1339
1340 ///////////////////////////////////////////////////////////////////////////////
1341 // CHECK KEY STATE
1342 ///////////////////////////////////////////////////////////////////////////////
1343
1344 bool MUtils::OS::check_key_state_esc(void)
1345 {
1346         return (GetAsyncKeyState(VK_ESCAPE) & 0x0001) != 0;
1347 }
1348
1349 ///////////////////////////////////////////////////////////////////////////////
1350 // SHELL CHANGE NOTIFICATION
1351 ///////////////////////////////////////////////////////////////////////////////
1352
1353 void MUtils::OS::shell_change_notification(void)
1354 {
1355         SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
1356 }
1357
1358 ///////////////////////////////////////////////////////////////////////////////
1359 // WOW64 REDIRECTION
1360 ///////////////////////////////////////////////////////////////////////////////
1361
1362 typedef BOOL (_stdcall *Wow64DisableWow64FsRedirectionFun)(void *OldValue);
1363 typedef BOOL (_stdcall *Wow64RevertWow64FsRedirectionFun )(void *OldValue);
1364
1365 bool MUtils::OS::wow64fsredir_disable(void *oldValue)
1366 {
1367         const Wow64DisableWow64FsRedirectionFun wow64redir_disable = MUtils::Win32Utils::resolve<Wow64DisableWow64FsRedirectionFun>(QLatin1String("kernel32"), QLatin1String("Wow64DisableWow64FsRedirection"));
1368         if(wow64redir_disable)
1369         {
1370                 if (wow64redir_disable(oldValue))
1371                 {
1372                         return true;
1373                 }
1374         }
1375         return false;
1376 }
1377
1378 bool MUtils::OS::wow64fsredir_revert(void *oldValue)
1379 {
1380         const Wow64RevertWow64FsRedirectionFun wow64redir_disable = MUtils::Win32Utils::resolve<Wow64RevertWow64FsRedirectionFun>(QLatin1String("kernel32"), QLatin1String("Wow64RevertWow64FsRedirection"));
1381         if (wow64redir_disable)
1382         {
1383                 if (wow64redir_disable(oldValue))
1384                 {
1385                         return true;
1386                 }
1387         }
1388         return false;
1389 }
1390
1391 ///////////////////////////////////////////////////////////////////////////////
1392 // DEBUGGER CHECK
1393 ///////////////////////////////////////////////////////////////////////////////
1394
1395 #if (!(MUTILS_DEBUG))
1396 static __forceinline bool is_debugger_present(void)
1397 {
1398         __try
1399         {
1400                 CloseHandle((HANDLE)((DWORD_PTR)-3));
1401         }
1402         __except(1)
1403         {
1404                 return true;
1405         }
1406
1407         BOOL bHaveDebugger = FALSE;
1408         if(CheckRemoteDebuggerPresent(GetCurrentProcess(), &bHaveDebugger))
1409         {
1410                 if(bHaveDebugger) return true;
1411         }
1412
1413         return IsDebuggerPresent();
1414 }
1415 static __forceinline bool check_debugger_helper(void)
1416 {
1417         if(is_debugger_present())
1418         {
1419                 MUtils::OS::fatal_exit(L"Not a debug build. Please unload debugger and try again!");
1420                 return true;
1421         }
1422         return false;
1423 }
1424 #else
1425 #define check_debugger_helper() (false)
1426 #endif
1427
1428 void MUtils::OS::check_debugger(void)
1429 {
1430         check_debugger_helper();
1431 }
1432
1433 static volatile bool g_debug_check = check_debugger_helper();
1434
1435 ///////////////////////////////////////////////////////////////////////////////
1436 // FATAL EXIT
1437 ///////////////////////////////////////////////////////////////////////////////
1438
1439 static MUtils::Internal::CriticalSection g_fatal_exit_lock;
1440 static volatile bool g_fatal_exit_flag = true;
1441
1442 static DWORD WINAPI fatal_exit_helper(LPVOID lpParameter)
1443 {
1444         MUtils::OS::system_message_err(L"GURU MEDITATION", (LPWSTR) lpParameter);
1445         return 0;
1446 }
1447
1448 void MUtils::OS::fatal_exit(const wchar_t* const errorMessage)
1449 {
1450         g_fatal_exit_lock.enter();
1451         
1452         if(!g_fatal_exit_flag)
1453         {
1454                 return; /*prevent recursive invocation*/
1455         }
1456
1457         g_fatal_exit_flag = false;
1458
1459         if(g_main_thread_id != GetCurrentThreadId())
1460         {
1461                 if(HANDLE hThreadMain = OpenThread(THREAD_SUSPEND_RESUME, FALSE, g_main_thread_id))
1462                 {
1463                         SuspendThread(hThreadMain); /*stop main thread*/
1464                 }
1465         }
1466
1467         if(HANDLE hThread = CreateThread(NULL, 0, fatal_exit_helper, (LPVOID) errorMessage, 0, NULL))
1468         {
1469                 WaitForSingleObject(hThread, INFINITE);
1470         }
1471
1472         for(;;)
1473         {
1474                 TerminateProcess(GetCurrentProcess(), 666);
1475         }
1476 }
1477
1478 ///////////////////////////////////////////////////////////////////////////////