OSDN Git Service

Fixed is_library_file() for older Windows versions.
[mutilities/MUtilities.git] / src / OSSupport_Win32.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // MuldeR's Utilities for Qt
3 // Copyright (C) 2004-2016 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 void get_file_path_translate(QString &path)
732 {
733         static const DWORD BUFSIZE = 2048;
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
753 static QString get_file_path_fallback(const HANDLE &hFile)
754 {
755         QString filePath;
756
757         const HANDLE hFileMap = CreateFileMappingW(hFile, NULL, PAGE_READONLY, 0, 1, NULL);
758         if (hFileMap)
759         {
760                 void *const pMem = MapViewOfFile(hFileMap, FILE_MAP_READ, 0, 0, 1);
761                 if (pMem)
762                 {
763                         static const size_t BUFFSIZE = 2048;
764                         wchar_t buffer[BUFFSIZE];
765                         if (GetMappedFileNameW(GetCurrentProcess(), pMem, buffer, BUFFSIZE) > 0)
766                         {
767                                 filePath = MUTILS_QSTR(buffer);
768                         }
769                         UnmapViewOfFile(pMem);
770                 }
771                 CloseHandle(hFileMap);
772         }
773
774         if (!filePath.isEmpty())
775         {
776                 get_file_path_translate(filePath);
777         }
778
779         return filePath;
780 }
781
782 QString MUtils::OS::get_file_path(const int &fd)
783 {
784         if (fd >= 0)
785         {
786                 const GetPathNameByHandleFun getPathNameByHandleFun = MUtils::Win32Utils::resolve<GetPathNameByHandleFun>(QLatin1String("kernel32"), QLatin1String("GetFinalPathNameByHandleW"));
787                 if (!getPathNameByHandleFun)
788                 {
789                         return get_file_path_fallback((HANDLE)_get_osfhandle(fd));
790                 }
791
792                 const HANDLE handle = (HANDLE) _get_osfhandle(fd);
793                 const DWORD len = getPathNameByHandleFun(handle, NULL, 0, FILE_NAME_OPENED);
794                 if (len > 0)
795                 {
796                         if (wchar_t *const buffer = (wchar_t*)_malloca(sizeof(wchar_t) * len))
797                         {
798                                 const DWORD ret = getPathNameByHandleFun(handle, buffer, len, FILE_NAME_OPENED);
799                                 if ((ret > 0) && (ret < len))
800                                 {
801                                         const QString path(MUTILS_QSTR(buffer));
802                                         return path.startsWith(QLatin1String("\\\\?\\")) ? path.mid(4) : path;
803                                 }
804                                 _freea(buffer);
805                         }
806                 }
807         }
808
809         return QString();
810 }
811
812 ///////////////////////////////////////////////////////////////////////////////
813 // PROCESS ELEVATION
814 ///////////////////////////////////////////////////////////////////////////////
815
816 static bool user_is_admin_helper(void)
817 {
818         HANDLE hToken = NULL;
819         if(!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
820         {
821                 return false;
822         }
823
824         DWORD dwSize = 0;
825         if(!GetTokenInformation(hToken, TokenGroups, NULL, 0, &dwSize))
826         {
827                 if(GetLastError() != ERROR_INSUFFICIENT_BUFFER)
828                 {
829                         CloseHandle(hToken);
830                         return false;
831                 }
832         }
833
834         PTOKEN_GROUPS lpGroups = (PTOKEN_GROUPS) malloc(dwSize);
835         if(!lpGroups)
836         {
837                 CloseHandle(hToken);
838                 return false;
839         }
840
841         if(!GetTokenInformation(hToken, TokenGroups, lpGroups, dwSize, &dwSize))
842         {
843                 free(lpGroups);
844                 CloseHandle(hToken);
845                 return false;
846         }
847
848         PSID lpSid = NULL; SID_IDENTIFIER_AUTHORITY Authority = {SECURITY_NT_AUTHORITY};
849         if(!AllocateAndInitializeSid(&Authority, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &lpSid))
850         {
851                 free(lpGroups);
852                 CloseHandle(hToken);
853                 return false;
854         }
855
856         bool bResult = false;
857         for(DWORD i = 0; i < lpGroups->GroupCount; i++)
858         {
859                 if(EqualSid(lpSid, lpGroups->Groups[i].Sid))
860                 {
861                         bResult = true;
862                         break;
863                 }
864         }
865
866         FreeSid(lpSid);
867         free(lpGroups);
868         CloseHandle(hToken);
869         return bResult;
870 }
871
872 bool MUtils::OS::is_elevated(bool *bIsUacEnabled)
873 {
874         if(bIsUacEnabled)
875         {
876                 *bIsUacEnabled = false;
877         }
878
879         bool bIsProcessElevated = false;
880         HANDLE hToken = NULL;
881         
882         if(OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
883         {
884                 TOKEN_ELEVATION_TYPE tokenElevationType;
885                 DWORD returnLength;
886                 if(GetTokenInformation(hToken, TokenElevationType, &tokenElevationType, sizeof(TOKEN_ELEVATION_TYPE), &returnLength))
887                 {
888                         if(returnLength == sizeof(TOKEN_ELEVATION_TYPE))
889                         {
890                                 switch(tokenElevationType)
891                                 {
892                                 case TokenElevationTypeDefault:
893                                         qDebug("Process token elevation type: Default -> UAC is disabled.\n");
894                                         break;
895                                 case TokenElevationTypeFull:
896                                         qWarning("Process token elevation type: Full -> potential security risk!\n");
897                                         bIsProcessElevated = true;
898                                         if(bIsUacEnabled) *bIsUacEnabled = true;
899                                         break;
900                                 case TokenElevationTypeLimited:
901                                         qDebug("Process token elevation type: Limited -> not elevated.\n");
902                                         if(bIsUacEnabled) *bIsUacEnabled = true;
903                                         break;
904                                 default:
905                                         qWarning("Unknown tokenElevationType value: %d", tokenElevationType);
906                                         break;
907                                 }
908                         }
909                         else
910                         {
911                                 qWarning("GetTokenInformation() return an unexpected size!");
912                         }
913                 }
914                 CloseHandle(hToken);
915         }
916         else
917         {
918                 qWarning("Failed to open process token!");
919         }
920
921         return bIsProcessElevated;
922 }
923
924 bool MUtils::OS::user_is_admin(void)
925 {
926         bool isAdmin = false;
927
928         //Check for process elevation and UAC support first!
929         if(MUtils::OS::is_elevated(&isAdmin))
930         {
931                 qWarning("Process is elevated -> user is admin!");
932                 return true;
933         }
934         
935         //If not elevated and UAC is not available -> user must be in admin group!
936         if(!isAdmin)
937         {
938                 qDebug("UAC is disabled/unavailable -> checking for Administrators group");
939                 isAdmin = user_is_admin_helper();
940         }
941
942         return isAdmin;
943 }
944
945 ///////////////////////////////////////////////////////////////////////////////
946 // NETWORK STATE
947 ///////////////////////////////////////////////////////////////////////////////
948
949 int MUtils::OS::network_status(void)
950 {
951         DWORD dwFlags;
952         const BOOL ret = IsNetworkAlive(&dwFlags);
953         if(GetLastError() == 0)
954         {
955                 return (ret != FALSE) ? NETWORK_TYPE_YES : NETWORK_TYPE_NON;
956         }
957         return NETWORK_TYPE_ERR;
958 }
959
960 ///////////////////////////////////////////////////////////////////////////////
961 // MESSAGE HANDLER
962 ///////////////////////////////////////////////////////////////////////////////
963
964 bool MUtils::OS::handle_os_message(const void *const message, long *result)
965 {
966         const MSG *const msg = reinterpret_cast<const MSG*>(message);
967
968         switch(msg->message)
969         {
970         case WM_QUERYENDSESSION:
971                 qWarning("WM_QUERYENDSESSION message received!");
972                 *result = MUtils::GUI::broadcast(MUtils::GUI::USER_EVENT_QUERYENDSESSION, false) ? TRUE : FALSE;
973                 return true;
974         case WM_ENDSESSION:
975                 qWarning("WM_ENDSESSION message received!");
976                 if(msg->wParam == TRUE)
977                 {
978                         MUtils::GUI::broadcast(MUtils::GUI::USER_EVENT_ENDSESSION, false);
979                         MUtils::GUI::force_quit();
980                         exit(1);
981                 }
982                 *result = 0;
983                 return true;
984         default:
985                 /*ignore this message and let Qt handle it*/
986                 return false;
987         }
988 }
989
990 ///////////////////////////////////////////////////////////////////////////////
991 // SLEEP
992 ///////////////////////////////////////////////////////////////////////////////
993
994 void MUtils::OS::sleep_ms(const size_t &duration)
995 {
996         Sleep((DWORD) duration);
997 }
998
999 ///////////////////////////////////////////////////////////////////////////////
1000 // EXECUTABLE CHECK
1001 ///////////////////////////////////////////////////////////////////////////////
1002
1003 static bool libraryAsImageResourceSupported()
1004 {
1005         OSVERSIONINFOEXW osvi;
1006         memset(&osvi, 0, sizeof(OSVERSIONINFOEXW));
1007         osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEXW);
1008         if(GetVersionExW((LPOSVERSIONINFOW)&osvi))
1009         {
1010                 if(osvi.dwPlatformId == VER_PLATFORM_WIN32_NT)
1011                 {
1012                         return (osvi.dwMajorVersion >= 6U);
1013                 }
1014         }
1015         return false;
1016 }
1017
1018 bool MUtils::OS::is_executable_file(const QString &path)
1019 {
1020         DWORD binaryType;
1021         if(GetBinaryType(MUTILS_WCHR(QDir::toNativeSeparators(path)), &binaryType))
1022         {
1023                 return ((binaryType == SCS_32BIT_BINARY) || (binaryType == SCS_64BIT_BINARY));
1024         }
1025
1026         const DWORD errorCode = GetLastError();
1027         qWarning("GetBinaryType() failed with error: 0x%08X", errorCode);
1028         return false;
1029 }
1030
1031 bool MUtils::OS::is_library_file(const QString &path)
1032 {
1033         const DWORD flags = libraryAsImageResourceSupported() ? (LOAD_LIBRARY_AS_DATAFILE | LOAD_LIBRARY_AS_IMAGE_RESOURCE) : (LOAD_LIBRARY_AS_DATAFILE | DONT_RESOLVE_DLL_REFERENCES);
1034         if (const HMODULE hMod = LoadLibraryEx(MUTILS_WCHR(QDir::toNativeSeparators(path)), NULL, flags))
1035         {
1036                 FreeLibrary(hMod);
1037                 return true;
1038         }
1039
1040         const DWORD errorCode = GetLastError();
1041         qWarning("LoadLibraryEx() failed with error: 0x%08X", errorCode);
1042         return false;
1043 }
1044
1045 ///////////////////////////////////////////////////////////////////////////////
1046 // HIBERNATION / SHUTDOWN
1047 ///////////////////////////////////////////////////////////////////////////////
1048
1049 bool MUtils::OS::is_hibernation_supported(void)
1050 {
1051         bool hibernationSupported = false;
1052
1053         SYSTEM_POWER_CAPABILITIES pwrCaps;
1054         SecureZeroMemory(&pwrCaps, sizeof(SYSTEM_POWER_CAPABILITIES));
1055         
1056         if(GetPwrCapabilities(&pwrCaps))
1057         {
1058                 hibernationSupported = pwrCaps.SystemS4 && pwrCaps.HiberFilePresent;
1059         }
1060
1061         return hibernationSupported;
1062 }
1063
1064 bool MUtils::OS::shutdown_computer(const QString &message, const unsigned long timeout, const bool forceShutdown, const bool hibernate)
1065 {
1066         HANDLE hToken = NULL;
1067
1068         if(OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
1069         {
1070                 TOKEN_PRIVILEGES privileges;
1071                 memset(&privileges, 0, sizeof(TOKEN_PRIVILEGES));
1072                 privileges.PrivilegeCount = 1;
1073                 privileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
1074                 
1075                 if(LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &privileges.Privileges[0].Luid))
1076                 {
1077                         if(AdjustTokenPrivileges(hToken, FALSE, &privileges, NULL, NULL, NULL))
1078                         {
1079                                 if(hibernate)
1080                                 {
1081                                         if(SetSuspendState(TRUE, TRUE, TRUE))
1082                                         {
1083                                                 return true;
1084                                         }
1085                                 }
1086                                 const DWORD reason = SHTDN_REASON_MAJOR_APPLICATION | SHTDN_REASON_FLAG_PLANNED;
1087                                 return InitiateSystemShutdownEx(NULL, const_cast<wchar_t*>(MUTILS_WCHR(message)), timeout, forceShutdown ? TRUE : FALSE, FALSE, reason);
1088                         }
1089                 }
1090         }
1091         
1092         return false;
1093 }
1094
1095 ///////////////////////////////////////////////////////////////////////////////
1096 // FREE DISKSPACE
1097 ///////////////////////////////////////////////////////////////////////////////
1098
1099 bool MUtils::OS::free_diskspace(const QString &path, quint64 &freeSpace)
1100 {
1101         ULARGE_INTEGER freeBytesAvailable, totalNumberOfBytes, totalNumberOfFreeBytes;
1102         if(GetDiskFreeSpaceExW(reinterpret_cast<const wchar_t*>(QDir::toNativeSeparators(path).utf16()), &freeBytesAvailable, &totalNumberOfBytes, &totalNumberOfFreeBytes))
1103         {
1104                 freeSpace = freeBytesAvailable.QuadPart;
1105                 return true;;
1106         }
1107
1108         freeSpace = -1;
1109         return false;
1110 }
1111
1112 ///////////////////////////////////////////////////////////////////////////////
1113 // SHELL OPEN
1114 ///////////////////////////////////////////////////////////////////////////////
1115
1116 bool MUtils::OS::shell_open(const QWidget *parent, const QString &url, const QString &parameters, const QString &directory, const bool explore)
1117 {
1118         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;
1119 }
1120
1121 bool MUtils::OS::shell_open(const QWidget *parent, const QString &url, const bool explore)
1122 {
1123         return shell_open(parent, url, QString(), QString(), explore);
1124 }
1125
1126 ///////////////////////////////////////////////////////////////////////////////
1127 // OPEN MEDIA FILE
1128 ///////////////////////////////////////////////////////////////////////////////
1129
1130 bool MUtils::OS::open_media_file(const QString &mediaFilePath)
1131 {
1132         const static wchar_t *registryPrefix[2] = { L"SOFTWARE\\", L"SOFTWARE\\Wow6432Node\\" };
1133         const static wchar_t *registryKeys[3] = 
1134         {
1135                 L"Microsoft\\Windows\\CurrentVersion\\Uninstall\\{97D341C8-B0D1-4E4A-A49A-C30B52F168E9}",
1136                 L"Microsoft\\Windows\\CurrentVersion\\Uninstall\\{DB9E4EAB-2717-499F-8D56-4CC8A644AB60}",
1137                 L"foobar2000"
1138         };
1139         const static wchar_t *appNames[4] = { L"smplayer_portable.exe", L"smplayer.exe", L"MPUI.exe", L"foobar2000.exe" };
1140         const static wchar_t *valueNames[2] = { L"InstallLocation", L"InstallDir" };
1141
1142         for(size_t i = 0; i < 3; i++)
1143         {
1144                 for(size_t j = 0; j < 2; j++)
1145                 {
1146                         QString mplayerPath;
1147                         HKEY registryKeyHandle = NULL;
1148
1149                         const QString currentKey = MUTILS_QSTR(registryPrefix[j]).append(MUTILS_QSTR(registryKeys[i]));
1150                         if(RegOpenKeyExW(HKEY_LOCAL_MACHINE, MUTILS_WCHR(currentKey), 0, KEY_READ, &registryKeyHandle) == ERROR_SUCCESS)
1151                         {
1152                                 for(size_t k = 0; k < 2; k++)
1153                                 {
1154                                         wchar_t Buffer[4096];
1155                                         DWORD BuffSize = sizeof(wchar_t*) * 4096;
1156                                         DWORD DataType = REG_NONE;
1157                                         if(RegQueryValueExW(registryKeyHandle, valueNames[k], 0, &DataType, reinterpret_cast<BYTE*>(Buffer), &BuffSize) == ERROR_SUCCESS)
1158                                         {
1159                                                 if((DataType == REG_SZ) || (DataType == REG_EXPAND_SZ) || (DataType == REG_LINK))
1160                                                 {
1161                                                         mplayerPath = MUTILS_QSTR(Buffer);
1162                                                         break;
1163                                                 }
1164                                         }
1165                                 }
1166                                 RegCloseKey(registryKeyHandle);
1167                         }
1168
1169                         if(!mplayerPath.isEmpty())
1170                         {
1171                                 QDir mplayerDir(mplayerPath);
1172                                 if(mplayerDir.exists())
1173                                 {
1174                                         for(size_t k = 0; k < 4; k++)
1175                                         {
1176                                                 if(mplayerDir.exists(MUTILS_QSTR(appNames[k])))
1177                                                 {
1178                                                         qDebug("Player found at:\n%s\n", MUTILS_UTF8(mplayerDir.absoluteFilePath(MUTILS_QSTR(appNames[k]))));
1179                                                         QProcess::startDetached(mplayerDir.absoluteFilePath(MUTILS_QSTR(appNames[k])), QStringList() << QDir::toNativeSeparators(mediaFilePath));
1180                                                         return true;
1181                                                 }
1182                                         }
1183                                 }
1184                         }
1185                 }
1186         }
1187         return false;
1188 }
1189
1190 ///////////////////////////////////////////////////////////////////////////////
1191 // DEBUGGER CHECK
1192 ///////////////////////////////////////////////////////////////////////////////
1193
1194 static bool change_process_priority_helper(const HANDLE hProcess, const int priority)
1195 {
1196         bool ok = false;
1197
1198         switch(qBound(-2, priority, 2))
1199         {
1200         case 2:
1201                 ok = (SetPriorityClass(hProcess, HIGH_PRIORITY_CLASS) == TRUE);
1202                 break;
1203         case 1:
1204                 if(!(ok = (SetPriorityClass(hProcess, ABOVE_NORMAL_PRIORITY_CLASS) == TRUE)))
1205                 {
1206                         ok = (SetPriorityClass(hProcess, HIGH_PRIORITY_CLASS) == TRUE);
1207                 }
1208                 break;
1209         case 0:
1210                 ok = (SetPriorityClass(hProcess, NORMAL_PRIORITY_CLASS) == TRUE);
1211                 break;
1212         case -1:
1213                 if(!(ok = (SetPriorityClass(hProcess, BELOW_NORMAL_PRIORITY_CLASS) == TRUE)))
1214                 {
1215                         ok = (SetPriorityClass(hProcess, IDLE_PRIORITY_CLASS) == TRUE);
1216                 }
1217                 break;
1218         case -2:
1219                 ok = (SetPriorityClass(hProcess, IDLE_PRIORITY_CLASS) == TRUE);
1220                 break;
1221         }
1222
1223         return ok;
1224 }
1225
1226 bool MUtils::OS::change_process_priority(const int priority)
1227 {
1228         return change_process_priority_helper(GetCurrentProcess(), priority);
1229 }
1230
1231 bool MUtils::OS::change_process_priority(const QProcess *proc, const int priority)
1232 {
1233         if(Q_PID qPid = proc->pid())
1234         {
1235                 return change_process_priority_helper(qPid->hProcess, priority);
1236         }
1237         else
1238         {
1239                 return false;
1240         }
1241 }
1242
1243 ///////////////////////////////////////////////////////////////////////////////
1244 // PROCESS ID
1245 ///////////////////////////////////////////////////////////////////////////////
1246
1247 quint32 MUtils::OS::process_id(const QProcess *const proc)
1248 {
1249         PROCESS_INFORMATION *procInf = proc->pid();
1250         return (procInf) ? procInf->dwProcessId : NULL;
1251 }
1252
1253 ///////////////////////////////////////////////////////////////////////////////
1254 // PROCESS SUSPEND/RESUME
1255 ///////////////////////////////////////////////////////////////////////////////
1256
1257 bool MUtils::OS::suspend_process(const QProcess *proc, const bool suspend)
1258 {
1259         if(Q_PID pid = proc->pid())
1260         {
1261                 if(suspend)
1262                 {
1263                         return (SuspendThread(pid->hThread) != ((DWORD) -1));
1264                 }
1265                 else
1266                 {
1267                         return (ResumeThread(pid->hThread)  != ((DWORD) -1));
1268                 }
1269         }
1270         else
1271         {
1272                 return false;
1273         }
1274 }
1275
1276 ///////////////////////////////////////////////////////////////////////////////
1277 // SYSTEM TIMER
1278 ///////////////////////////////////////////////////////////////////////////////
1279
1280 bool MUtils::OS::setup_timer_resolution(const quint32 &interval)
1281 {
1282         return timeBeginPeriod(interval) == TIMERR_NOERROR;
1283 }
1284
1285 bool MUtils::OS::reset_timer_resolution(const quint32 &interval)
1286 {
1287         return timeEndPeriod(interval) == TIMERR_NOERROR;
1288 }
1289
1290 ///////////////////////////////////////////////////////////////////////////////
1291 // SET FILE TIME
1292 ///////////////////////////////////////////////////////////////////////////////
1293
1294 static QScopedPointer<QDateTime> s_epoch;
1295 static QReadWriteLock            s_epochLock;
1296
1297 static const QDateTime *get_epoch(void)
1298 {
1299         QReadLocker rdLock(&s_epochLock);
1300
1301         if (s_epoch.isNull())
1302         {
1303                 rdLock.unlock();
1304                 QWriteLocker wrLock(&s_epochLock);
1305                 if (s_epoch.isNull())
1306                 {
1307                         s_epoch.reset(new QDateTime(QDate(1601, 1, 1), QTime(0, 0, 0, 0), Qt::UTC));
1308                 }
1309                 wrLock.unlock();
1310                 rdLock.relock();
1311         }
1312
1313         return s_epoch.data();
1314 }
1315
1316 static FILETIME *qt_time_to_file_time(FILETIME *const fileTime, const QDateTime &dateTime)
1317 {
1318         memset(fileTime, 0, sizeof(FILETIME));
1319
1320         if (const QDateTime *const epoch = get_epoch())
1321         {
1322                 const qint64 msecs = epoch->msecsTo(dateTime);
1323                 if (msecs > 0)
1324                 {
1325                         const quint64 ticks = 10000U * quint64(msecs);
1326                         fileTime->dwHighDateTime = ((ticks >> 32) & 0xFFFFFFFF);
1327                         fileTime->dwLowDateTime = (ticks & 0xFFFFFFFF);
1328                         return fileTime;
1329                 }
1330         }
1331
1332         return NULL;
1333 }
1334
1335 static bool set_file_time(const HANDLE hFile, const QDateTime &created, const QDateTime &lastMod, const QDateTime &lastAcc)
1336 {
1337         FILETIME ftCreated, ftLastMod, ftLastAcc;
1338
1339         FILETIME *const pCreated = created.isValid() ? qt_time_to_file_time(&ftCreated, created) : NULL;
1340         FILETIME *const pLastMod = lastMod.isValid() ? qt_time_to_file_time(&ftLastMod, lastMod) : NULL;
1341         FILETIME *const pLastAcc = lastAcc.isValid() ? qt_time_to_file_time(&ftLastAcc, lastAcc) : NULL;
1342
1343         if (pCreated || pLastMod || pLastAcc)
1344         {
1345                 return (SetFileTime(hFile, pCreated, pLastAcc, pLastMod) != FALSE);
1346         }
1347
1348         return false;
1349 }
1350
1351 bool MUtils::OS::set_file_time(const QFile &file, const QDateTime &created, const QDateTime &lastMod, const QDateTime &lastAcc)
1352 {
1353         const int fd = file.handle();
1354         if (fd >= 0)
1355         {
1356                 return set_file_time((HANDLE)_get_osfhandle(fd), created, lastMod, lastAcc);
1357         }
1358         return false;
1359 }
1360
1361 bool MUtils::OS::set_file_time(const QString &path, const QDateTime &created, const QDateTime &lastMod, const QDateTime &lastAcc)
1362 {
1363         const HANDLE hFile = CreateFileW(MUTILS_WCHR(path), FILE_WRITE_ATTRIBUTES, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, 0);
1364         bool okay = false;
1365         if ((hFile != NULL) && (hFile != INVALID_HANDLE_VALUE))
1366         {
1367                 okay = set_file_time(hFile, created, lastMod, lastAcc);
1368                 CloseHandle(hFile);
1369         }
1370         return okay;
1371 }
1372
1373 ///////////////////////////////////////////////////////////////////////////////
1374 // CHECK KEY STATE
1375 ///////////////////////////////////////////////////////////////////////////////
1376
1377 bool MUtils::OS::check_key_state_esc(void)
1378 {
1379         return (GetAsyncKeyState(VK_ESCAPE) & 0x0001) != 0;
1380 }
1381
1382 ///////////////////////////////////////////////////////////////////////////////
1383 // SHELL CHANGE NOTIFICATION
1384 ///////////////////////////////////////////////////////////////////////////////
1385
1386 void MUtils::OS::shell_change_notification(void)
1387 {
1388         SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
1389 }
1390
1391 ///////////////////////////////////////////////////////////////////////////////
1392 // WOW64 REDIRECTION
1393 ///////////////////////////////////////////////////////////////////////////////
1394
1395 typedef BOOL (_stdcall *Wow64DisableWow64FsRedirectionFun)(void *OldValue);
1396 typedef BOOL (_stdcall *Wow64RevertWow64FsRedirectionFun )(void *OldValue);
1397
1398 bool MUtils::OS::wow64fsredir_disable(void *oldValue)
1399 {
1400         const Wow64DisableWow64FsRedirectionFun wow64redir_disable = MUtils::Win32Utils::resolve<Wow64DisableWow64FsRedirectionFun>(QLatin1String("kernel32"), QLatin1String("Wow64DisableWow64FsRedirection"));
1401         if(wow64redir_disable)
1402         {
1403                 if (wow64redir_disable(oldValue))
1404                 {
1405                         return true;
1406                 }
1407         }
1408         return false;
1409 }
1410
1411 bool MUtils::OS::wow64fsredir_revert(void *oldValue)
1412 {
1413         const Wow64RevertWow64FsRedirectionFun wow64redir_disable = MUtils::Win32Utils::resolve<Wow64RevertWow64FsRedirectionFun>(QLatin1String("kernel32"), QLatin1String("Wow64RevertWow64FsRedirection"));
1414         if (wow64redir_disable)
1415         {
1416                 if (wow64redir_disable(oldValue))
1417                 {
1418                         return true;
1419                 }
1420         }
1421         return false;
1422 }
1423
1424 ///////////////////////////////////////////////////////////////////////////////
1425 // DEBUGGER CHECK
1426 ///////////////////////////////////////////////////////////////////////////////
1427
1428 #if (!(MUTILS_DEBUG))
1429 static __forceinline bool is_debugger_present(void)
1430 {
1431         __try
1432         {
1433                 CloseHandle((HANDLE)((DWORD_PTR)-3));
1434         }
1435         __except(1)
1436         {
1437                 return true;
1438         }
1439
1440         BOOL bHaveDebugger = FALSE;
1441         if(CheckRemoteDebuggerPresent(GetCurrentProcess(), &bHaveDebugger))
1442         {
1443                 if(bHaveDebugger) return true;
1444         }
1445
1446         return IsDebuggerPresent();
1447 }
1448 static __forceinline bool check_debugger_helper(void)
1449 {
1450         if(is_debugger_present())
1451         {
1452                 MUtils::OS::fatal_exit(L"Not a debug build. Please unload debugger and try again!");
1453                 return true;
1454         }
1455         return false;
1456 }
1457 #else
1458 #define check_debugger_helper() (false)
1459 #endif
1460
1461 void MUtils::OS::check_debugger(void)
1462 {
1463         check_debugger_helper();
1464 }
1465
1466 static volatile bool g_debug_check = check_debugger_helper();
1467
1468 ///////////////////////////////////////////////////////////////////////////////
1469 // FATAL EXIT
1470 ///////////////////////////////////////////////////////////////////////////////
1471
1472 static MUtils::Internal::CriticalSection g_fatal_exit_lock;
1473 static volatile bool g_fatal_exit_flag = true;
1474
1475 static DWORD WINAPI fatal_exit_helper(LPVOID lpParameter)
1476 {
1477         MUtils::OS::system_message_err(L"GURU MEDITATION", (LPWSTR) lpParameter);
1478         return 0;
1479 }
1480
1481 void MUtils::OS::fatal_exit(const wchar_t* const errorMessage)
1482 {
1483         g_fatal_exit_lock.enter();
1484         
1485         if(!g_fatal_exit_flag)
1486         {
1487                 return; /*prevent recursive invocation*/
1488         }
1489
1490         g_fatal_exit_flag = false;
1491
1492         if(g_main_thread_id != GetCurrentThreadId())
1493         {
1494                 if(HANDLE hThreadMain = OpenThread(THREAD_SUSPEND_RESUME, FALSE, g_main_thread_id))
1495                 {
1496                         SuspendThread(hThreadMain); /*stop main thread*/
1497                 }
1498         }
1499
1500         if(HANDLE hThread = CreateThread(NULL, 0, fatal_exit_helper, (LPVOID) errorMessage, 0, NULL))
1501         {
1502                 WaitForSingleObject(hThread, INFINITE);
1503         }
1504
1505         for(;;)
1506         {
1507                 TerminateProcess(GetCurrentProcess(), 666);
1508         }
1509 }
1510
1511 ///////////////////////////////////////////////////////////////////////////////