1 ///////////////////////////////////////////////////////////////////////////////
2 // MuldeR's Utilities for Qt
3 // Copyright (C) 2004-2014 LoRd_MuldeR <MuldeR2@GMX.de>
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.
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.
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
19 // http://www.gnu.org/licenses/lgpl-2.1.txt
20 //////////////////////////////////////////////////////////////////////////////////
25 #include <MUtils/Global.h>
26 #include <MUtils/OSSupport.h>
27 #include "CriticalSection_Win32.h"
30 #define WIN32_LEAN_AND_MEAN 1
36 #include <QReadWriteLock>
41 static const DWORD g_main_thread_id = GetCurrentThreadId();
43 ///////////////////////////////////////////////////////////////////////////////
45 ///////////////////////////////////////////////////////////////////////////////
47 typedef HRESULT (WINAPI *SHGetKnownFolderPath_t)(const GUID &rfid, DWORD dwFlags, HANDLE hToken, PWSTR *ppszPath);
48 typedef HRESULT (WINAPI *SHGetFolderPath_t)(HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPWSTR pszPath);
50 static QMap<size_t, QString>* g_known_folders_map;
51 static SHGetKnownFolderPath_t g_known_folders_fpGetKnownFolderPath;
52 static SHGetFolderPath_t g_known_folders_fpGetFolderPath;
53 static QReadWriteLock g_known_folders_lock;
55 const QString &MUtils::OS::known_folder(known_folder_t folder_id)
57 static const int CSIDL_FLAG_CREATE = 0x8000;
58 typedef enum { KF_FLAG_CREATE = 0x00008000 } kf_flags_t;
67 { 0x001c, {0xF1B32785,0x6FBA,0x4FCF,{0x9D,0x55,0x7B,0x8E,0x7F,0x15,0x70,0x91}} }, //CSIDL_LOCAL_APPDATA
68 { 0x0026, {0x905e63b6,0xc1bf,0x494e,{0xb2,0x9c,0x65,0xb7,0x32,0xd3,0xd2,0x1a}} }, //CSIDL_PROGRAM_FILES
69 { 0x0024, {0xF38BF404,0x1D43,0x42F2,{0x93,0x05,0x67,0xDE,0x0B,0x28,0xFC,0x23}} }, //CSIDL_WINDOWS_FOLDER
70 { 0x0025, {0x1AC14E77,0x02E7,0x4E5D,{0xB7,0x44,0x2E,0xB1,0xAE,0x51,0x98,0xB7}} }, //CSIDL_SYSTEM_FOLDER
73 size_t folderId = size_t(-1);
77 case FOLDER_LOCALAPPDATA: folderId = 0; break;
78 case FOLDER_PROGRAMFILES: folderId = 1; break;
79 case FOLDER_SYSTROOT_DIR: folderId = 2; break;
80 case FOLDER_SYSTEMFOLDER: folderId = 3; break;
83 if(folderId == size_t(-1))
85 qWarning("Invalid 'known' folder was requested!");
86 return *reinterpret_cast<QString*>(NULL);
89 QReadLocker readLock(&g_known_folders_lock);
92 if(g_known_folders_map)
94 if(g_known_folders_map->contains(folderId))
96 return g_known_folders_map->operator[](folderId);
100 //Obtain write lock to initialize
102 QWriteLocker writeLock(&g_known_folders_lock);
104 //Still not in cache?
105 if(g_known_folders_map)
107 if(g_known_folders_map->contains(folderId))
109 return g_known_folders_map->operator[](folderId);
113 //Initialize on first call
114 if(!g_known_folders_map)
116 QLibrary shell32("shell32.dll");
119 g_known_folders_fpGetFolderPath = (SHGetFolderPath_t) shell32.resolve("SHGetFolderPathW");
120 g_known_folders_fpGetKnownFolderPath = (SHGetKnownFolderPath_t) shell32.resolve("SHGetKnownFolderPath");
122 g_known_folders_map = new QMap<size_t, QString>();
127 //Now try to get the folder path!
128 if(g_known_folders_fpGetKnownFolderPath)
131 if(g_known_folders_fpGetKnownFolderPath(s_folders[folderId].guid, KF_FLAG_CREATE, NULL, &path) == S_OK)
133 //MessageBoxW(0, path, L"SHGetKnownFolderPath", MB_TOPMOST);
134 QDir folderTemp = QDir(QDir::fromNativeSeparators(MUTILS_WCHAR2QSTR(path)));
135 if(folderTemp.exists())
137 folderPath = folderTemp.canonicalPath();
142 else if(g_known_folders_fpGetFolderPath)
144 QScopedArrayPointer<WCHAR> path(new WCHAR[4096]);
145 if(g_known_folders_fpGetFolderPath(NULL, s_folders[folderId].csidl | CSIDL_FLAG_CREATE, NULL, NULL, path.data()) == S_OK)
147 //MessageBoxW(0, path, L"SHGetFolderPathW", MB_TOPMOST);
148 QDir folderTemp = QDir(QDir::fromNativeSeparators(MUTILS_WCHAR2QSTR(path.data())));
149 if(folderTemp.exists())
151 folderPath = folderTemp.canonicalPath();
157 g_known_folders_map->insert(folderId, folderPath);
158 return g_known_folders_map->operator[](folderId);
161 ///////////////////////////////////////////////////////////////////////////////
163 ///////////////////////////////////////////////////////////////////////////////
165 static CriticalSection g_fatal_exit_lock;
166 static volatile bool g_fatal_exit_flag = true;
168 static DWORD WINAPI fatal_exit_helper(LPVOID lpParameter)
170 MessageBoxA(NULL, ((LPCSTR) lpParameter), "LameXP - Guru Meditation", MB_OK | MB_ICONERROR | MB_TASKMODAL | MB_TOPMOST | MB_SETFOREGROUND);
174 void MUtils::OS::fatal_exit(const char* const errorMessage)
176 g_fatal_exit_lock.enter();
178 if(!g_fatal_exit_flag)
180 return; /*prevent recursive invocation*/
183 g_fatal_exit_flag = false;
185 if(g_main_thread_id != GetCurrentThreadId())
187 if(HANDLE hThreadMain = OpenThread(THREAD_SUSPEND_RESUME, FALSE, g_main_thread_id))
189 SuspendThread(hThreadMain); /*stop main thread*/
193 if(HANDLE hThread = CreateThread(NULL, 0, fatal_exit_helper, (LPVOID) errorMessage, 0, NULL))
195 WaitForSingleObject(hThread, INFINITE);
200 TerminateProcess(GetCurrentProcess(), 666);
204 ///////////////////////////////////////////////////////////////////////////////