Bump version.
[LameXP.git] / src / Global.cpp
blob323d6e1448739345c0c9d44f68e8f0226f20ecd1
1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2012 LoRd_MuldeR <MuldeR2@GMX.de>
4 //
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; either version 2 of the License, or
8 // (at your option) any later version.
9 //
10 // This program 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
13 // GNU General Public License for more details.
15 // You should have received a copy of the GNU General Public License along
16 // with this program; if not, write to the Free Software Foundation, Inc.,
17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 // http://www.gnu.org/licenses/gpl-2.0.txt
20 ///////////////////////////////////////////////////////////////////////////////
22 #include "Global.h"
24 //Qt includes
25 #include <QApplication>
26 #include <QMessageBox>
27 #include <QDir>
28 #include <QUuid>
29 #include <QMap>
30 #include <QDate>
31 #include <QIcon>
32 #include <QPlastiqueStyle>
33 #include <QImageReader>
34 #include <QSharedMemory>
35 #include <QSysInfo>
36 #include <QStringList>
37 #include <QSystemSemaphore>
38 #include <QMutex>
39 #include <QTextCodec>
40 #include <QLibrary>
41 #include <QRegExp>
42 #include <QResource>
43 #include <QTranslator>
44 #include <QEventLoop>
45 #include <QTimer>
46 #include <QLibraryInfo>
47 #include <QEvent>
49 //LameXP includes
50 #include "Resource.h"
51 #include "LockedFile.h"
53 //CRT includes
54 #include <iostream>
55 #include <fstream>
56 #include <io.h>
57 #include <fcntl.h>
58 #include <intrin.h>
59 #include <math.h>
60 #include <time.h>
61 #include <process.h>
63 //COM includes
64 #include <Objbase.h>
65 #include <PowrProf.h>
67 //Debug only includes
68 #if LAMEXP_DEBUG
69 #include <Psapi.h>
70 #endif
72 //Initialize static Qt plugins
73 #ifdef QT_NODLL
74 Q_IMPORT_PLUGIN(qico)
75 Q_IMPORT_PLUGIN(qsvg)
76 #endif
78 ///////////////////////////////////////////////////////////////////////////////
79 // TYPES
80 ///////////////////////////////////////////////////////////////////////////////
82 static const size_t g_lamexp_ipc_slots = 128;
84 typedef struct
86 unsigned int command;
87 unsigned int reserved_1;
88 unsigned int reserved_2;
89 char parameter[4096];
91 lamexp_ipc_data_t;
93 typedef struct
95 unsigned int pos_write;
96 unsigned int pos_read;
97 lamexp_ipc_data_t data[g_lamexp_ipc_slots];
99 lamexp_ipc_t;
101 ///////////////////////////////////////////////////////////////////////////////
102 // GLOBAL VARS
103 ///////////////////////////////////////////////////////////////////////////////
105 //Build version
106 static const struct
108 unsigned int ver_major;
109 unsigned int ver_minor;
110 unsigned int ver_build;
111 char *ver_release_name;
113 g_lamexp_version =
115 VER_LAMEXP_MAJOR,
116 VER_LAMEXP_MINOR,
117 VER_LAMEXP_BUILD,
118 VER_LAMEXP_RNAME
121 //Build date
122 static QDate g_lamexp_version_date;
123 static const char *g_lamexp_months[] = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};
124 static const char *g_lamexp_version_raw_date = __DATE__;
125 static const char *g_lamexp_version_raw_time = __TIME__;
127 //Console attached flag
128 static bool g_lamexp_console_attached = false;
130 //Compiler detection
131 //The following code was borrowed from MPC-HC project: http://mpc-hc.sf.net/
132 #if defined(__INTEL_COMPILER)
133 #if (__INTEL_COMPILER >= 1200)
134 static const char *g_lamexp_version_compiler = "ICL 12.x";
135 #elif (__INTEL_COMPILER >= 1100)
136 static const char *g_lamexp_version_compiler = = "ICL 11.x";
137 #elif (__INTEL_COMPILER >= 1000)
138 static const char *g_lamexp_version_compiler = = "ICL 10.x";
139 #else
140 #error Compiler is not supported!
141 #endif
142 #elif defined(_MSC_VER)
143 #if (_MSC_VER == 1600)
144 #if (_MSC_FULL_VER >= 160040219)
145 static const char *g_lamexp_version_compiler = "MSVC 2010-SP1";
146 #else
147 static const char *g_lamexp_version_compiler = "MSVC 2010";
148 #endif
149 #elif (_MSC_VER == 1500)
150 #if (_MSC_FULL_VER >= 150030729)
151 static const char *g_lamexp_version_compiler = "MSVC 2008-SP1";
152 #else
153 static const char *g_lamexp_version_compiler = "MSVC 2008";
154 #endif
155 #else
156 #error Compiler is not supported!
157 #endif
159 // Note: /arch:SSE and /arch:SSE2 are only available for the x86 platform
160 #if !defined(_M_X64) && defined(_M_IX86_FP)
161 #if (_M_IX86_FP == 1)
162 LAMEXP_COMPILER_WARNING("SSE instruction set is enabled!")
163 #elif (_M_IX86_FP == 2)
164 LAMEXP_COMPILER_WARNING("SSE2 instruction set is enabled!")
165 #endif
166 #endif
167 #else
168 #error Compiler is not supported!
169 #endif
171 //Architecture detection
172 #if defined(_M_X64)
173 static const char *g_lamexp_version_arch = "x64";
174 #elif defined(_M_IX86)
175 static const char *g_lamexp_version_arch = "x86";
176 #else
177 #error Architecture is not supported!
178 #endif
180 //Official web-site URL
181 static const char *g_lamexp_website_url = "http://lamexp.sourceforge.net/";
182 static const char *g_lamexp_support_url = "http://forum.doom9.org/showthread.php?t=157726";
184 //Tool versions (expected versions!)
185 static const unsigned int g_lamexp_toolver_neroaac = VER_LAMEXP_TOOL_NEROAAC;
186 static const unsigned int g_lamexp_toolver_fhgaacenc = VER_LAMEXP_TOOL_FHGAACENC;
187 static const unsigned int g_lamexp_toolver_qaacenc = VER_LAMEXP_TOOL_QAAC;
188 static const unsigned int g_lamexp_toolver_coreaudio = VER_LAMEXP_TOOL_COREAUDIO;
190 //Special folders
191 static QString g_lamexp_temp_folder;
193 //Tools
194 static QMap<QString, LockedFile*> g_lamexp_tool_registry;
195 static QMap<QString, unsigned int> g_lamexp_tool_versions;
197 //Languages
198 static struct
200 QMap<QString, QString> files;
201 QMap<QString, QString> names;
202 QMap<QString, unsigned int> sysid;
203 QMap<QString, unsigned int> cntry;
205 g_lamexp_translation;
207 //Translator
208 static QTranslator *g_lamexp_currentTranslator = NULL;
210 //Shared memory
211 static const struct
213 char *sharedmem;
214 char *semaphore_read;
215 char *semaphore_read_mutex;
216 char *semaphore_write;
217 char *semaphore_write_mutex;
219 g_lamexp_ipc_uuid =
221 "{21A68A42-6923-43bb-9CF6-64BF151942EE}",
222 "{7A605549-F58C-4d78-B4E5-06EFC34F405B}",
223 "{60AA8D04-F6B8-497d-81EB-0F600F4A65B5}",
224 "{726061D5-1615-4B82-871C-75FD93458E46}",
225 "{1A616023-AA6A-4519-8AF3-F7736E899977}"
227 static struct
229 QSharedMemory *sharedmem;
230 QSystemSemaphore *semaphore_read;
231 QSystemSemaphore *semaphore_read_mutex;
232 QSystemSemaphore *semaphore_write;
233 QSystemSemaphore *semaphore_write_mutex;
235 g_lamexp_ipc_ptr =
237 NULL, NULL, NULL
240 //Image formats
241 static const char *g_lamexp_imageformats[] = {"png", "jpg", "gif", "ico", "svg", NULL};
243 //Global locks
244 static QMutex g_lamexp_message_mutex;
246 //Main thread ID
247 static const DWORD g_main_thread_id = GetCurrentThreadId();
249 //Log file
250 static FILE *g_lamexp_log_file = NULL;
252 ///////////////////////////////////////////////////////////////////////////////
253 // GLOBAL FUNCTIONS
254 ///////////////////////////////////////////////////////////////////////////////
257 * Version getters
259 unsigned int lamexp_version_major(void) { return g_lamexp_version.ver_major; }
260 unsigned int lamexp_version_minor(void) { return g_lamexp_version.ver_minor; }
261 unsigned int lamexp_version_build(void) { return g_lamexp_version.ver_build; }
262 const char *lamexp_version_release(void) { return g_lamexp_version.ver_release_name; }
263 const char *lamexp_version_time(void) { return g_lamexp_version_raw_time; }
264 const char *lamexp_version_compiler(void) { return g_lamexp_version_compiler; }
265 const char *lamexp_version_arch(void) { return g_lamexp_version_arch; }
266 unsigned int lamexp_toolver_neroaac(void) { return g_lamexp_toolver_neroaac; }
267 unsigned int lamexp_toolver_fhgaacenc(void) { return g_lamexp_toolver_fhgaacenc; }
268 unsigned int lamexp_toolver_qaacenc(void) { return g_lamexp_toolver_qaacenc; }
269 unsigned int lamexp_toolver_coreaudio(void) { return g_lamexp_toolver_coreaudio; }
272 * URL getters
274 const char *lamexp_website_url(void) { return g_lamexp_website_url; }
275 const char *lamexp_support_url(void) { return g_lamexp_support_url; }
278 * Check for Demo (pre-release) version
280 bool lamexp_version_demo(void)
282 char buffer[128];
283 bool releaseVersion = false;
284 if(!strncpy_s(buffer, 128, g_lamexp_version.ver_release_name, _TRUNCATE))
286 char *context, *prefix = strtok_s(buffer, "-,; ", &context);
287 if(prefix)
289 releaseVersion = (!_stricmp(prefix, "Final")) || (!_stricmp(prefix, "Hotfix"));
292 return LAMEXP_DEBUG || (!releaseVersion);
296 * Calculate expiration date
298 QDate lamexp_version_expires(void)
300 return lamexp_version_date().addDays(LAMEXP_DEBUG ? 2 : 30);
304 * Get build date date
306 const QDate &lamexp_version_date(void)
308 if(!g_lamexp_version_date.isValid())
310 int date[3] = {0, 0, 0}; char temp[12] = {'\0'};
311 strncpy_s(temp, 12, g_lamexp_version_raw_date, _TRUNCATE);
313 if(strlen(temp) == 11)
315 temp[3] = temp[6] = '\0';
316 date[2] = atoi(&temp[4]);
317 date[0] = atoi(&temp[7]);
319 for(int j = 0; j < 12; j++)
321 if(!_strcmpi(&temp[0], g_lamexp_months[j]))
323 date[1] = j+1;
324 break;
328 g_lamexp_version_date = QDate(date[0], date[1], date[2]);
331 if(!g_lamexp_version_date.isValid())
333 qFatal("Internal error: Date format could not be recognized!");
337 return g_lamexp_version_date;
341 * Get the native operating system version
343 DWORD lamexp_get_os_version(void)
345 static DWORD osVersion = 0;
347 if(!osVersion)
349 OSVERSIONINFO osVerInfo;
350 memset(&osVerInfo, 0, sizeof(OSVERSIONINFO));
351 osVerInfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
353 if(GetVersionEx(&osVerInfo) == TRUE)
355 if(osVerInfo.dwPlatformId != VER_PLATFORM_WIN32_NT)
357 throw "Ouuups: Not running under Windows NT. This is not supposed to happen!";
359 osVersion = (DWORD)((osVerInfo.dwMajorVersion << 16) | (osVerInfo.dwMinorVersion & 0xffff));
363 return osVersion;
367 * Check if we are running under wine
369 bool lamexp_detect_wine(void)
371 static bool isWine = false;
372 static bool isWine_initialized = false;
374 if(!isWine_initialized)
376 QLibrary ntdll("ntdll.dll");
377 if(ntdll.load())
379 if(ntdll.resolve("wine_nt_to_unix_file_name") != NULL) isWine = true;
380 if(ntdll.resolve("wine_get_version") != NULL) isWine = true;
381 ntdll.unload();
383 isWine_initialized = true;
386 return isWine;
390 * Global exception handler
392 LONG WINAPI lamexp_exception_handler(__in struct _EXCEPTION_POINTERS *ExceptionInfo)
394 if(GetCurrentThreadId() != g_main_thread_id)
396 HANDLE mainThread = OpenThread(THREAD_TERMINATE, FALSE, g_main_thread_id);
397 if(mainThread) TerminateThread(mainThread, ULONG_MAX);
400 FatalAppExit(0, L"Unhandeled exception handler invoked, application will exit!");
401 TerminateProcess(GetCurrentProcess(), -1);
402 return LONG_MAX;
406 * Invalid parameters handler
408 void lamexp_invalid_param_handler(const wchar_t*, const wchar_t*, const wchar_t*, unsigned int, uintptr_t)
410 if(GetCurrentThreadId() != g_main_thread_id)
412 HANDLE mainThread = OpenThread(THREAD_TERMINATE, FALSE, g_main_thread_id);
413 if(mainThread) TerminateThread(mainThread, ULONG_MAX);
417 FatalAppExit(0, L"Invalid parameter handler invoked, application will exit!");
418 TerminateProcess(GetCurrentProcess(), -1);
422 * Change console text color
424 static void lamexp_console_color(FILE* file, WORD attributes)
426 const HANDLE hConsole = (HANDLE)(_get_osfhandle(_fileno(file)));
427 if((hConsole != NULL) && (hConsole != INVALID_HANDLE_VALUE))
429 SetConsoleTextAttribute(hConsole, attributes);
434 * Qt message handler
436 void lamexp_message_handler(QtMsgType type, const char *msg)
438 static const char *GURU_MEDITATION = "\n\nGURU MEDITATION !!!\n\n";
440 QMutexLocker lock(&g_lamexp_message_mutex);
442 if(g_lamexp_log_file)
444 static char prefix[] = "DWCF";
445 int index = qBound(0, static_cast<int>(type), 3);
446 unsigned int timestamp = static_cast<unsigned int>(_time64(NULL) % 3600I64);
447 QString str = QString::fromUtf8(msg).trimmed().replace('\n', '\t');
448 fprintf(g_lamexp_log_file, "[%c][%04u] %s\r\n", prefix[index], timestamp, str.toUtf8().constData());
449 fflush(g_lamexp_log_file);
452 if(g_lamexp_console_attached)
454 UINT oldOutputCP = GetConsoleOutputCP();
455 if(oldOutputCP != CP_UTF8) SetConsoleOutputCP(CP_UTF8);
457 switch(type)
459 case QtCriticalMsg:
460 case QtFatalMsg:
461 fflush(stdout);
462 fflush(stderr);
463 lamexp_console_color(stderr, FOREGROUND_RED | FOREGROUND_INTENSITY);
464 fprintf(stderr, GURU_MEDITATION);
465 fprintf(stderr, "%s\n", msg);
466 fflush(stderr);
467 break;
468 case QtWarningMsg:
469 lamexp_console_color(stderr, FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY);
470 fprintf(stderr, "%s\n", msg);
471 fflush(stderr);
472 break;
473 default:
474 lamexp_console_color(stderr, FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY);
475 fprintf(stderr, "%s\n", msg);
476 fflush(stderr);
477 break;
480 lamexp_console_color(stderr, FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED);
481 if(oldOutputCP != CP_UTF8) SetConsoleOutputCP(oldOutputCP);
483 else
485 QString temp("[LameXP][%1] %2");
487 switch(type)
489 case QtCriticalMsg:
490 case QtFatalMsg:
491 temp = temp.arg("C", QString::fromUtf8(msg));
492 break;
493 case QtWarningMsg:
494 temp = temp.arg("W", QString::fromUtf8(msg));
495 break;
496 default:
497 temp = temp.arg("I", QString::fromUtf8(msg));
498 break;
501 temp.replace("\n", "\t").append("\n");
502 OutputDebugStringA(temp.toLatin1().constData());
505 if(type == QtCriticalMsg || type == QtFatalMsg)
507 lock.unlock();
509 if(GetCurrentThreadId() != g_main_thread_id)
511 HANDLE mainThread = OpenThread(THREAD_TERMINATE, FALSE, g_main_thread_id);
512 if(mainThread) TerminateThread(mainThread, ULONG_MAX);
515 MessageBoxW(NULL, QWCHAR(QString::fromUtf8(msg)), L"LameXP - GURU MEDITATION", MB_ICONERROR | MB_TOPMOST | MB_TASKMODAL);
516 FatalAppExit(0, L"The application has encountered a critical error and will exit now!");
517 TerminateProcess(GetCurrentProcess(), -1);
522 * Initialize the console
524 void lamexp_init_console(int argc, char* argv[])
526 bool enableConsole = lamexp_version_demo();
528 if(_environ)
530 wchar_t *logfile = NULL;
531 size_t logfile_len = 0;
532 if(!_wdupenv_s(&logfile, &logfile_len, L"LAMEXP_LOGFILE"))
534 if(logfile && (logfile_len > 0))
536 FILE *temp = NULL;
537 if(!_wfopen_s(&temp, logfile, L"wb"))
539 fprintf(temp, "%c%c%c", 0xEF, 0xBB, 0xBF);
540 g_lamexp_log_file = temp;
542 free(logfile);
547 if(!LAMEXP_DEBUG)
549 for(int i = 0; i < argc; i++)
551 if(!_stricmp(argv[i], "--console"))
553 enableConsole = true;
555 else if(!_stricmp(argv[i], "--no-console"))
557 enableConsole = false;
562 if(enableConsole)
564 if(!g_lamexp_console_attached)
566 if(AllocConsole() != FALSE)
568 SetConsoleCtrlHandler(NULL, TRUE);
569 SetConsoleTitle(L"LameXP - Audio Encoder Front-End | Debug Console");
570 SetConsoleOutputCP(CP_UTF8);
571 g_lamexp_console_attached = true;
575 if(g_lamexp_console_attached)
577 //-------------------------------------------------------------------
578 //See: http://support.microsoft.com/default.aspx?scid=kb;en-us;105305
579 //-------------------------------------------------------------------
580 const int flags = _O_WRONLY | _O_U8TEXT;
581 int hCrtStdOut = _open_osfhandle((intptr_t) GetStdHandle(STD_OUTPUT_HANDLE), flags);
582 int hCrtStdErr = _open_osfhandle((intptr_t) GetStdHandle(STD_ERROR_HANDLE), flags);
583 FILE *hfStdOut = (hCrtStdOut >= 0) ? _fdopen(hCrtStdOut, "wb") : NULL;
584 FILE *hfStdErr = (hCrtStdErr >= 0) ? _fdopen(hCrtStdErr, "wb") : NULL;
585 if(hfStdOut) { *stdout = *hfStdOut; std::cout.rdbuf(new std::filebuf(hfStdOut)); }
586 if(hfStdErr) { *stderr = *hfStdErr; std::cerr.rdbuf(new std::filebuf(hfStdErr)); }
589 HWND hwndConsole = GetConsoleWindow();
591 if((hwndConsole != NULL) && (hwndConsole != INVALID_HANDLE_VALUE))
593 HMENU hMenu = GetSystemMenu(hwndConsole, 0);
594 EnableMenuItem(hMenu, SC_CLOSE, MF_BYCOMMAND | MF_GRAYED);
595 RemoveMenu(hMenu, SC_CLOSE, MF_BYCOMMAND);
597 SetWindowPos(hwndConsole, HWND_TOP, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE|SWP_NOZORDER|SWP_FRAMECHANGED);
598 SetWindowLong(hwndConsole, GWL_STYLE, GetWindowLong(hwndConsole, GWL_STYLE) & (~WS_MAXIMIZEBOX) & (~WS_MINIMIZEBOX));
599 SetWindowPos(hwndConsole, HWND_TOP, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE|SWP_NOZORDER|SWP_FRAMECHANGED);
605 * Detect CPU features
607 lamexp_cpu_t lamexp_detect_cpu_features(int argc, char **argv)
609 typedef BOOL (WINAPI *IsWow64ProcessFun)(__in HANDLE hProcess, __out PBOOL Wow64Process);
610 typedef VOID (WINAPI *GetNativeSystemInfoFun)(__out LPSYSTEM_INFO lpSystemInfo);
612 static IsWow64ProcessFun IsWow64ProcessPtr = NULL;
613 static GetNativeSystemInfoFun GetNativeSystemInfoPtr = NULL;
615 lamexp_cpu_t features;
616 SYSTEM_INFO systemInfo;
617 int CPUInfo[4] = {-1};
618 char CPUIdentificationString[0x40];
619 char CPUBrandString[0x40];
621 memset(&features, 0, sizeof(lamexp_cpu_t));
622 memset(&systemInfo, 0, sizeof(SYSTEM_INFO));
623 memset(CPUIdentificationString, 0, sizeof(CPUIdentificationString));
624 memset(CPUBrandString, 0, sizeof(CPUBrandString));
626 __cpuid(CPUInfo, 0);
627 memcpy(CPUIdentificationString, &CPUInfo[1], sizeof(int));
628 memcpy(CPUIdentificationString + 4, &CPUInfo[3], sizeof(int));
629 memcpy(CPUIdentificationString + 8, &CPUInfo[2], sizeof(int));
630 features.intel = (_stricmp(CPUIdentificationString, "GenuineIntel") == 0);
631 strncpy_s(features.vendor, 0x40, CPUIdentificationString, _TRUNCATE);
633 if(CPUInfo[0] >= 1)
635 __cpuid(CPUInfo, 1);
636 features.mmx = (CPUInfo[3] & 0x800000) || false;
637 features.sse = (CPUInfo[3] & 0x2000000) || false;
638 features.sse2 = (CPUInfo[3] & 0x4000000) || false;
639 features.ssse3 = (CPUInfo[2] & 0x200) || false;
640 features.sse3 = (CPUInfo[2] & 0x1) || false;
641 features.ssse3 = (CPUInfo[2] & 0x200) || false;
642 features.stepping = CPUInfo[0] & 0xf;
643 features.model = ((CPUInfo[0] >> 4) & 0xf) + (((CPUInfo[0] >> 16) & 0xf) << 4);
644 features.family = ((CPUInfo[0] >> 8) & 0xf) + ((CPUInfo[0] >> 20) & 0xff);
647 __cpuid(CPUInfo, 0x80000000);
648 int nExIds = qMax<int>(qMin<int>(CPUInfo[0], 0x80000004), 0x80000000);
650 for(int i = 0x80000002; i <= nExIds; ++i)
652 __cpuid(CPUInfo, i);
653 switch(i)
655 case 0x80000002:
656 memcpy(CPUBrandString, CPUInfo, sizeof(CPUInfo));
657 break;
658 case 0x80000003:
659 memcpy(CPUBrandString + 16, CPUInfo, sizeof(CPUInfo));
660 break;
661 case 0x80000004:
662 memcpy(CPUBrandString + 32, CPUInfo, sizeof(CPUInfo));
663 break;
667 strncpy_s(features.brand, 0x40, CPUBrandString, _TRUNCATE);
669 if(strlen(features.brand) < 1) strncpy_s(features.brand, 0x40, "Unknown", _TRUNCATE);
670 if(strlen(features.vendor) < 1) strncpy_s(features.vendor, 0x40, "Unknown", _TRUNCATE);
672 #if !defined(_M_X64 ) && !defined(_M_IA64)
673 if(!IsWow64ProcessPtr || !GetNativeSystemInfoPtr)
675 QLibrary Kernel32Lib("kernel32.dll");
676 IsWow64ProcessPtr = (IsWow64ProcessFun) Kernel32Lib.resolve("IsWow64Process");
677 GetNativeSystemInfoPtr = (GetNativeSystemInfoFun) Kernel32Lib.resolve("GetNativeSystemInfo");
679 if(IsWow64ProcessPtr)
681 BOOL x64 = FALSE;
682 if(IsWow64ProcessPtr(GetCurrentProcess(), &x64))
684 features.x64 = x64;
687 if(GetNativeSystemInfoPtr)
689 GetNativeSystemInfoPtr(&systemInfo);
691 else
693 GetSystemInfo(&systemInfo);
695 features.count = qBound(1UL, systemInfo.dwNumberOfProcessors, 64UL);
696 #else
697 GetNativeSystemInfo(&systemInfo);
698 features.count = systemInfo.dwNumberOfProcessors;
699 features.x64 = true;
700 #endif
702 if((argv != NULL) && (argc > 0))
704 bool flag = false;
705 for(int i = 0; i < argc; i++)
707 if(!_stricmp("--force-cpu-no-64bit", argv[i])) { flag = true; features.x64 = false; }
708 if(!_stricmp("--force-cpu-no-sse", argv[i])) { flag = true; features.sse = features.sse2 = features.sse3 = features.ssse3 = false; }
709 if(!_stricmp("--force-cpu-no-intel", argv[i])) { flag = true; features.intel = false; }
711 if(flag) qWarning("CPU flags overwritten by user-defined parameters. Take care!\n");
714 return features;
718 * Check for debugger (detect routine)
720 static __forceinline bool lamexp_check_for_debugger(void)
722 if(IsDebuggerPresent())
724 return true;
727 __try
729 CloseHandle((HANDLE) 0x7FFFFFFF);
731 __except(EXCEPTION_EXECUTE_HANDLER)
733 return true;
736 __try
738 DebugBreak();
740 __except(EXCEPTION_EXECUTE_HANDLER)
742 return false;
745 return true;
749 * Check for debugger (thread proc)
751 static unsigned int __stdcall lamexp_debug_thread_proc(LPVOID lpParameter)
753 while(!lamexp_check_for_debugger())
755 Sleep(32);
757 if(HANDLE thrd = OpenThread(THREAD_TERMINATE, FALSE, g_main_thread_id))
759 if(TerminateThread(thrd, -1))
761 FatalAppExit(0, L"Not a debug build. Please unload debugger and try again!");
763 CloseHandle(thrd);
765 TerminateProcess(GetCurrentProcess(), -1);
766 return 666;
770 * Check for debugger (startup routine)
772 static HANDLE lamexp_debug_thread_init(void)
774 if(lamexp_check_for_debugger())
776 FatalAppExit(0, L"Not a debug build. Please unload debugger and try again!");
777 TerminateProcess(GetCurrentProcess(), -1);
780 return (HANDLE) _beginthreadex(NULL, 0, lamexp_debug_thread_proc, NULL, 0, NULL);
784 * Check for compatibility mode
786 static bool lamexp_check_compatibility_mode(const char *exportName, const char *executableName)
788 QLibrary kernel32("kernel32.dll");
790 if(exportName != NULL)
792 if(kernel32.resolve(exportName) != NULL)
794 qWarning("Function '%s' exported from 'kernel32.dll' -> Windows compatibility mode!", exportName);
795 qFatal("%s", QApplication::tr("Executable '%1' doesn't support Windows compatibility mode.").arg(QString::fromLatin1(executableName)).toLatin1().constData());
796 return false;
800 return true;
804 * Computus according to H. Lichtenberg
806 static bool lamexp_computus(const QDate &date)
808 int X = date.year();
809 int A = X % 19;
810 int K = X / 100;
811 int M = 15 + (3*K + 3) / 4 - (8*K + 13) / 25;
812 int D = (19*A + M) % 30;
813 int S = 2 - (3*K + 3) / 4;
814 int R = D / 29 + (D / 28 - D / 29) * (A / 11);
815 int OG = 21 + D - R;
816 int SZ = 7 - (X + X / 4 + S) % 7;
817 int OE = 7 - (OG - SZ) % 7;
818 int OS = (OG + OE);
820 if(OS > 31)
822 return (date.month() == 4) && (date.day() == (OS - 31));
824 else
826 return (date.month() == 3) && (date.day() == OS);
831 * Check for Thanksgiving
833 static bool lamexp_thanksgiving(const QDate &date)
835 int day = 0;
837 switch(QDate(date.year(), 11, 1).dayOfWeek())
839 case 1: day = 25; break;
840 case 2: day = 24; break;
841 case 3: day = 23; break;
842 case 4: day = 22; break;
843 case 5: day = 28; break;
844 case 6: day = 27; break;
845 case 7: day = 26; break;
848 return (date.month() == 11) && (date.day() == day);
852 * Initialize app icon
854 QIcon lamexp_app_icon(const QDate *date, const QTime *time)
856 QDate currentDate = (date) ? QDate(*date) : QDate::currentDate();
857 QTime currentTime = (time) ? QTime(*time) : QTime::currentTime();
859 if(lamexp_thanksgiving(currentDate))
861 return QIcon(":/MainIcon6.png");
863 else if(((currentDate.month() == 12) && (currentDate.day() == 31) && (currentTime.hour() >= 20)) || ((currentDate.month() == 1) && (currentDate.day() == 1) && (currentTime.hour() <= 19)))
865 return QIcon(":/MainIcon5.png");
867 else if(((currentDate.month() == 10) && (currentDate.day() == 31) && (currentTime.hour() >= 12)) || ((currentDate.month() == 11) && (currentDate.day() == 1) && (currentTime.hour() <= 11)))
869 return QIcon(":/MainIcon4.png");
871 else if((currentDate.month() == 12) && (currentDate.day() >= 24) && (currentDate.day() <= 26))
873 return QIcon(":/MainIcon3.png");
875 else if(lamexp_computus(currentDate))
877 return QIcon(":/MainIcon2.png");
879 else
881 return QIcon(":/MainIcon1.png");
886 * Broadcast event to all windows
888 static bool lamexp_broadcast(int eventType, bool onlyToVisible)
890 if(QApplication *app = dynamic_cast<QApplication*>(QApplication::instance()))
892 qDebug("Broadcasting %d", eventType);
894 bool allOk = true;
895 QEvent poEvent(static_cast<QEvent::Type>(eventType));
896 QWidgetList list = app->topLevelWidgets();
898 while(!list.isEmpty())
900 QWidget *widget = list.takeFirst();
901 if(!onlyToVisible || widget->isVisible())
903 if(!app->sendEvent(widget, &poEvent))
905 allOk = false;
910 qDebug("Broadcast %d done (%s)", eventType, (allOk ? "OK" : "Stopped"));
911 return allOk;
913 else
915 qWarning("Broadcast failed, could not get QApplication instance!");
916 return false;
921 * Qt event filter
923 static bool lamexp_event_filter(void *message, long *result)
925 if((!(LAMEXP_DEBUG)) && lamexp_check_for_debugger())
927 FatalAppExit(0, L"Not a debug build. Please unload debugger and try again!");
928 TerminateProcess(GetCurrentProcess(), -1);
931 switch(reinterpret_cast<MSG*>(message)->message)
933 case WM_QUERYENDSESSION:
934 qWarning("WM_QUERYENDSESSION message received!");
935 *result = lamexp_broadcast(lamexp_event_queryendsession, false) ? TRUE : FALSE;
936 return true;
937 case WM_ENDSESSION:
938 qWarning("WM_ENDSESSION message received!");
939 if(reinterpret_cast<MSG*>(message)->wParam == TRUE)
941 lamexp_broadcast(lamexp_event_endsession, false);
942 if(QApplication *app = reinterpret_cast<QApplication*>(QApplication::instance()))
944 app->closeAllWindows();
945 app->quit();
947 lamexp_finalization();
948 exit(1);
950 *result = 0;
951 return true;
952 default:
953 /*ignore this message and let Qt handle it*/
954 return false;
959 * Check for process elevation
961 static bool lamexp_check_elevation(void)
963 typedef enum { lamexp_token_elevationType_class = 18, lamexp_token_elevation_class = 20 } LAMEXP_TOKEN_INFORMATION_CLASS;
964 typedef enum { lamexp_elevationType_default = 1, lamexp_elevationType_full, lamexp_elevationType_limited } LAMEXP_TOKEN_ELEVATION_TYPE;
966 HANDLE hToken = NULL;
967 bool bIsProcessElevated = false;
969 if(OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
971 LAMEXP_TOKEN_ELEVATION_TYPE tokenElevationType;
972 DWORD returnLength;
973 if(GetTokenInformation(hToken, (TOKEN_INFORMATION_CLASS) lamexp_token_elevationType_class, &tokenElevationType, sizeof(LAMEXP_TOKEN_ELEVATION_TYPE), &returnLength))
975 if(returnLength == sizeof(LAMEXP_TOKEN_ELEVATION_TYPE))
977 switch(tokenElevationType)
979 case lamexp_elevationType_default:
980 qDebug("Process token elevation type: Default -> UAC is disabled.\n");
981 break;
982 case lamexp_elevationType_full:
983 qWarning("Process token elevation type: Full -> potential security risk!\n");
984 bIsProcessElevated = true;
985 break;
986 case lamexp_elevationType_limited:
987 qDebug("Process token elevation type: Limited -> not elevated.\n");
988 break;
992 CloseHandle(hToken);
994 else
996 qWarning("Failed to open process token!");
999 return !bIsProcessElevated;
1003 * Initialize Qt framework
1005 bool lamexp_init_qt(int argc, char* argv[])
1007 static bool qt_initialized = false;
1008 typedef BOOL (WINAPI *SetDllDirectoryProc)(WCHAR *lpPathName);
1010 //Don't initialized again, if done already
1011 if(qt_initialized)
1013 return true;
1016 //Secure DLL loading
1017 QLibrary kernel32("kernel32.dll");
1018 if(kernel32.load())
1020 SetDllDirectoryProc pSetDllDirectory = (SetDllDirectoryProc) kernel32.resolve("SetDllDirectoryW");
1021 if(pSetDllDirectory != NULL) pSetDllDirectory(L"");
1022 kernel32.unload();
1025 //Extract executable name from argv[] array
1026 char *executableName = argv[0];
1027 while(char *temp = strpbrk(executableName, "\\/:?"))
1029 executableName = temp + 1;
1032 //Check Qt version
1033 qDebug("Using Qt v%s [%s], %s, %s", qVersion(), QLibraryInfo::buildDate().toString(Qt::ISODate).toLatin1().constData(), (qSharedBuild() ? "DLL" : "Static"), QLibraryInfo::buildKey().toLatin1().constData());
1034 qDebug("Compiled with Qt v%s [%s], %s\n", QT_VERSION_STR, QT_PACKAGEDATE_STR, QT_BUILD_KEY);
1035 if(_stricmp(qVersion(), QT_VERSION_STR))
1037 qFatal("%s", QApplication::tr("Executable '%1' requires Qt v%2, but found Qt v%3.").arg(QString::fromLatin1(executableName), QString::fromLatin1(QT_VERSION_STR), QString::fromLatin1(qVersion())).toLatin1().constData());
1038 return false;
1040 if(QLibraryInfo::buildKey().compare(QString::fromLatin1(QT_BUILD_KEY), Qt::CaseInsensitive))
1042 qFatal("%s", QApplication::tr("Executable '%1' was built for Qt '%2', but found Qt '%3'.").arg(QString::fromLatin1(executableName), QString::fromLatin1(QT_BUILD_KEY), QLibraryInfo::buildKey()).toLatin1().constData());
1043 return false;
1046 //Check the Windows version
1047 switch(QSysInfo::windowsVersion() & QSysInfo::WV_NT_based)
1049 case 0:
1050 case QSysInfo::WV_NT:
1051 qFatal("%s", QApplication::tr("Executable '%1' requires Windows 2000 or later.").arg(QString::fromLatin1(executableName)).toLatin1().constData());
1052 break;
1053 case QSysInfo::WV_2000:
1054 qDebug("Running on Windows 2000 (not officially supported!).\n");
1055 lamexp_check_compatibility_mode("GetNativeSystemInfo", executableName);
1056 break;
1057 case QSysInfo::WV_XP:
1058 qDebug("Running on Windows XP.\n");
1059 lamexp_check_compatibility_mode("GetLargePageMinimum", executableName);
1060 break;
1061 case QSysInfo::WV_2003:
1062 qDebug("Running on Windows Server 2003 or Windows XP x64-Edition.\n");
1063 lamexp_check_compatibility_mode("GetLocaleInfoEx", executableName);
1064 break;
1065 case QSysInfo::WV_VISTA:
1066 qDebug("Running on Windows Vista or Windows Server 2008.\n");
1067 lamexp_check_compatibility_mode("CreateRemoteThreadEx", executableName);
1068 break;
1069 case QSysInfo::WV_WINDOWS7:
1070 qDebug("Running on Windows 7 or Windows Server 2008 R2.\n");
1071 lamexp_check_compatibility_mode(NULL, executableName);
1072 break;
1073 default:
1075 DWORD osVersionNo = lamexp_get_os_version();
1076 qWarning("Running on an unknown/untested WinNT-based OS (v%u.%u).\n", HIWORD(osVersionNo), LOWORD(osVersionNo));
1078 break;
1081 //Check for Wine
1082 if(lamexp_detect_wine())
1084 qWarning("It appears we are running under Wine, unexpected things might happen!\n");
1088 //Create Qt application instance and setup version info
1089 QApplication *application = new QApplication(argc, argv);
1090 application->setApplicationName("LameXP - Audio Encoder Front-End");
1091 application->setApplicationVersion(QString().sprintf("%d.%02d.%04d", lamexp_version_major(), lamexp_version_minor(), lamexp_version_build()));
1092 application->setOrganizationName("LoRd_MuldeR");
1093 application->setOrganizationDomain("mulder.at.gg");
1094 application->setWindowIcon(lamexp_app_icon());
1095 application->setEventFilter(lamexp_event_filter);
1097 //Set text Codec for locale
1098 QTextCodec::setCodecForLocale(QTextCodec::codecForName("UTF-8"));
1100 //Load plugins from application directory
1101 QCoreApplication::setLibraryPaths(QStringList() << QApplication::applicationDirPath());
1102 qDebug("Library Path:\n%s\n", QApplication::libraryPaths().first().toUtf8().constData());
1104 //Check for supported image formats
1105 QList<QByteArray> supportedFormats = QImageReader::supportedImageFormats();
1106 for(int i = 0; g_lamexp_imageformats[i]; i++)
1108 if(!supportedFormats.contains(g_lamexp_imageformats[i]))
1110 qFatal("Qt initialization error: QImageIOHandler for '%s' missing!", g_lamexp_imageformats[i]);
1111 return false;
1115 //Add default translations
1116 g_lamexp_translation.files.insert(LAMEXP_DEFAULT_LANGID, "");
1117 g_lamexp_translation.names.insert(LAMEXP_DEFAULT_LANGID, "English");
1119 //Check for process elevation
1120 if((!lamexp_check_elevation()) && (!lamexp_detect_wine()))
1122 if(QMessageBox::warning(NULL, "LameXP", "<nobr>LameXP was started with elevated rights. This is a potential security risk!</nobr>", "Quit Program (Recommended)", "Ignore") == 0)
1124 return false;
1128 //Update console icon, if a console is attached
1129 if(g_lamexp_console_attached && (!lamexp_detect_wine()))
1131 typedef DWORD (__stdcall *SetConsoleIconFun)(HICON);
1132 QLibrary kernel32("kernel32.dll");
1133 if(kernel32.load())
1135 SetConsoleIconFun SetConsoleIconPtr = (SetConsoleIconFun) kernel32.resolve("SetConsoleIcon");
1136 if(SetConsoleIconPtr != NULL) SetConsoleIconPtr(QIcon(":/icons/sound.png").pixmap(16, 16).toWinHICON());
1137 kernel32.unload();
1141 //Done
1142 qt_initialized = true;
1143 return true;
1147 * Initialize IPC
1149 int lamexp_init_ipc(void)
1151 if(g_lamexp_ipc_ptr.sharedmem && g_lamexp_ipc_ptr.semaphore_read && g_lamexp_ipc_ptr.semaphore_write && g_lamexp_ipc_ptr.semaphore_read_mutex && g_lamexp_ipc_ptr.semaphore_write_mutex)
1153 return 0;
1156 g_lamexp_ipc_ptr.semaphore_read = new QSystemSemaphore(QString(g_lamexp_ipc_uuid.semaphore_read), 0);
1157 g_lamexp_ipc_ptr.semaphore_write = new QSystemSemaphore(QString(g_lamexp_ipc_uuid.semaphore_write), 0);
1158 g_lamexp_ipc_ptr.semaphore_read_mutex = new QSystemSemaphore(QString(g_lamexp_ipc_uuid.semaphore_read_mutex), 0);
1159 g_lamexp_ipc_ptr.semaphore_write_mutex = new QSystemSemaphore(QString(g_lamexp_ipc_uuid.semaphore_write_mutex), 0);
1161 if(g_lamexp_ipc_ptr.semaphore_read->error() != QSystemSemaphore::NoError)
1163 QString errorMessage = g_lamexp_ipc_ptr.semaphore_read->errorString();
1164 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read);
1165 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write);
1166 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read_mutex);
1167 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write_mutex);
1168 qFatal("Failed to create system smaphore: %s", errorMessage.toUtf8().constData());
1169 return -1;
1171 if(g_lamexp_ipc_ptr.semaphore_write->error() != QSystemSemaphore::NoError)
1173 QString errorMessage = g_lamexp_ipc_ptr.semaphore_write->errorString();
1174 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read);
1175 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write);
1176 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read_mutex);
1177 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write_mutex);
1178 qFatal("Failed to create system smaphore: %s", errorMessage.toUtf8().constData());
1179 return -1;
1181 if(g_lamexp_ipc_ptr.semaphore_read_mutex->error() != QSystemSemaphore::NoError)
1183 QString errorMessage = g_lamexp_ipc_ptr.semaphore_read_mutex->errorString();
1184 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read);
1185 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write);
1186 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read_mutex);
1187 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write_mutex);
1188 qFatal("Failed to create system smaphore: %s", errorMessage.toUtf8().constData());
1189 return -1;
1191 if(g_lamexp_ipc_ptr.semaphore_write_mutex->error() != QSystemSemaphore::NoError)
1193 QString errorMessage = g_lamexp_ipc_ptr.semaphore_write_mutex->errorString();
1194 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read);
1195 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write);
1196 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read_mutex);
1197 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write_mutex);
1198 qFatal("Failed to create system smaphore: %s", errorMessage.toUtf8().constData());
1199 return -1;
1202 g_lamexp_ipc_ptr.sharedmem = new QSharedMemory(QString(g_lamexp_ipc_uuid.sharedmem), NULL);
1204 if(!g_lamexp_ipc_ptr.sharedmem->create(sizeof(lamexp_ipc_t)))
1206 if(g_lamexp_ipc_ptr.sharedmem->error() == QSharedMemory::AlreadyExists)
1208 g_lamexp_ipc_ptr.sharedmem->attach();
1209 if(g_lamexp_ipc_ptr.sharedmem->error() == QSharedMemory::NoError)
1211 return 1;
1213 else
1215 QString errorMessage = g_lamexp_ipc_ptr.sharedmem->errorString();
1216 LAMEXP_DELETE(g_lamexp_ipc_ptr.sharedmem);
1217 qFatal("Failed to attach to shared memory: %s", errorMessage.toUtf8().constData());
1218 return -1;
1221 else
1223 QString errorMessage = g_lamexp_ipc_ptr.sharedmem->errorString();
1224 LAMEXP_DELETE(g_lamexp_ipc_ptr.sharedmem);
1225 qFatal("Failed to create shared memory: %s", errorMessage.toUtf8().constData());
1226 return -1;
1230 memset(g_lamexp_ipc_ptr.sharedmem->data(), 0, sizeof(lamexp_ipc_t));
1231 g_lamexp_ipc_ptr.semaphore_write->release(g_lamexp_ipc_slots);
1232 g_lamexp_ipc_ptr.semaphore_read_mutex->release();
1233 g_lamexp_ipc_ptr.semaphore_write_mutex->release();
1235 return 0;
1239 * IPC send message
1241 void lamexp_ipc_send(unsigned int command, const char* message)
1243 if(!g_lamexp_ipc_ptr.sharedmem || !g_lamexp_ipc_ptr.semaphore_read || !g_lamexp_ipc_ptr.semaphore_write || !g_lamexp_ipc_ptr.semaphore_read_mutex || !g_lamexp_ipc_ptr.semaphore_write_mutex)
1245 throw "Shared memory for IPC not initialized yet.";
1248 lamexp_ipc_data_t ipc_data;
1249 memset(&ipc_data, 0, sizeof(lamexp_ipc_data_t));
1250 ipc_data.command = command;
1252 if(message)
1254 strncpy_s(ipc_data.parameter, 4096, message, _TRUNCATE);
1257 if(g_lamexp_ipc_ptr.semaphore_write->acquire())
1259 if(g_lamexp_ipc_ptr.semaphore_write_mutex->acquire())
1261 lamexp_ipc_t *ptr = reinterpret_cast<lamexp_ipc_t*>(g_lamexp_ipc_ptr.sharedmem->data());
1262 memcpy(&ptr->data[ptr->pos_write], &ipc_data, sizeof(lamexp_ipc_data_t));
1263 ptr->pos_write = (ptr->pos_write + 1) % g_lamexp_ipc_slots;
1264 g_lamexp_ipc_ptr.semaphore_read->release();
1265 g_lamexp_ipc_ptr.semaphore_write_mutex->release();
1271 * IPC read message
1273 void lamexp_ipc_read(unsigned int *command, char* message, size_t buffSize)
1275 *command = 0;
1276 message[0] = '\0';
1278 if(!g_lamexp_ipc_ptr.sharedmem || !g_lamexp_ipc_ptr.semaphore_read || !g_lamexp_ipc_ptr.semaphore_write || !g_lamexp_ipc_ptr.semaphore_read_mutex || !g_lamexp_ipc_ptr.semaphore_write_mutex)
1280 throw "Shared memory for IPC not initialized yet.";
1283 lamexp_ipc_data_t ipc_data;
1284 memset(&ipc_data, 0, sizeof(lamexp_ipc_data_t));
1286 if(g_lamexp_ipc_ptr.semaphore_read->acquire())
1288 if(g_lamexp_ipc_ptr.semaphore_read_mutex->acquire())
1290 lamexp_ipc_t *ptr = reinterpret_cast<lamexp_ipc_t*>(g_lamexp_ipc_ptr.sharedmem->data());
1291 memcpy(&ipc_data, &ptr->data[ptr->pos_read], sizeof(lamexp_ipc_data_t));
1292 ptr->pos_read = (ptr->pos_read + 1) % g_lamexp_ipc_slots;
1293 g_lamexp_ipc_ptr.semaphore_write->release();
1294 g_lamexp_ipc_ptr.semaphore_read_mutex->release();
1296 if(!(ipc_data.reserved_1 || ipc_data.reserved_2))
1298 *command = ipc_data.command;
1299 strncpy_s(message, buffSize, ipc_data.parameter, _TRUNCATE);
1301 else
1303 qWarning("Malformed IPC message, will be ignored");
1310 * Check for LameXP "portable" mode
1312 bool lamexp_portable_mode(void)
1314 QString baseName = QFileInfo(QApplication::applicationFilePath()).completeBaseName();
1315 int idx1 = baseName.indexOf("lamexp", 0, Qt::CaseInsensitive);
1316 int idx2 = baseName.lastIndexOf("portable", -1, Qt::CaseInsensitive);
1317 return (idx1 >= 0) && (idx2 >= 0) && (idx1 < idx2);
1321 * Get a random string
1323 QString lamexp_rand_str(void)
1325 QRegExp regExp("\\{(\\w+)-(\\w+)-(\\w+)-(\\w+)-(\\w+)\\}");
1326 QString uuid = QUuid::createUuid().toString();
1328 if(regExp.indexIn(uuid) >= 0)
1330 return QString().append(regExp.cap(1)).append(regExp.cap(2)).append(regExp.cap(3)).append(regExp.cap(4)).append(regExp.cap(5));
1333 throw "The RegExp didn't match on the UUID string. This shouldn't happen ;-)";
1337 * Get LameXP temp folder
1339 const QString &lamexp_temp_folder2(void)
1341 static const char *TEMP_STR = "Temp";
1342 const QString WRITE_TEST_DATA = lamexp_rand_str();
1343 const QString SUB_FOLDER = lamexp_rand_str();
1345 //Already initialized?
1346 if(!g_lamexp_temp_folder.isEmpty())
1348 if(QDir(g_lamexp_temp_folder).exists())
1350 return g_lamexp_temp_folder;
1352 else
1354 g_lamexp_temp_folder.clear();
1358 //Try the %TMP% or %TEMP% directory first
1359 QDir temp = QDir::temp();
1360 if(temp.exists())
1362 temp.mkdir(SUB_FOLDER);
1363 if(temp.cd(SUB_FOLDER) && temp.exists())
1365 QFile testFile(QString("%1/~%2.tmp").arg(temp.canonicalPath(), lamexp_rand_str()));
1366 if(testFile.open(QIODevice::ReadWrite))
1368 if(testFile.write(WRITE_TEST_DATA.toLatin1().constData()) >= strlen(WRITE_TEST_DATA.toLatin1().constData()))
1370 g_lamexp_temp_folder = temp.canonicalPath();
1372 testFile.remove();
1375 if(!g_lamexp_temp_folder.isEmpty())
1377 return g_lamexp_temp_folder;
1381 //Create TEMP folder in %LOCALAPPDATA%
1382 QDir localAppData = QDir(lamexp_known_folder(lamexp_folder_localappdata));
1383 if(!localAppData.path().isEmpty())
1385 if(!localAppData.exists())
1387 localAppData.mkpath(".");
1389 if(localAppData.exists())
1391 if(!localAppData.entryList(QDir::AllDirs).contains(TEMP_STR, Qt::CaseInsensitive))
1393 localAppData.mkdir(TEMP_STR);
1395 if(localAppData.cd(TEMP_STR) && localAppData.exists())
1397 localAppData.mkdir(SUB_FOLDER);
1398 if(localAppData.cd(SUB_FOLDER) && localAppData.exists())
1400 QFile testFile(QString("%1/~%2.tmp").arg(localAppData.canonicalPath(), lamexp_rand_str()));
1401 if(testFile.open(QIODevice::ReadWrite))
1403 if(testFile.write(WRITE_TEST_DATA.toLatin1().constData()) >= strlen(WRITE_TEST_DATA.toLatin1().constData()))
1405 g_lamexp_temp_folder = localAppData.canonicalPath();
1407 testFile.remove();
1412 if(!g_lamexp_temp_folder.isEmpty())
1414 return g_lamexp_temp_folder;
1418 //Failed to create TEMP folder!
1419 qFatal("Temporary directory could not be initialized!\n\nFirst attempt:\n%s\n\nSecond attempt:\n%s", temp.canonicalPath().toUtf8().constData(), localAppData.canonicalPath().toUtf8().constData());
1420 return g_lamexp_temp_folder;
1424 * Clean folder
1426 bool lamexp_clean_folder(const QString &folderPath)
1428 QDir tempFolder(folderPath);
1429 QFileInfoList entryList = tempFolder.entryInfoList(QDir::AllEntries | QDir::NoDotAndDotDot);
1431 for(int i = 0; i < entryList.count(); i++)
1433 if(entryList.at(i).isDir())
1435 lamexp_clean_folder(entryList.at(i).canonicalFilePath());
1437 else
1439 for(int j = 0; j < 3; j++)
1441 if(lamexp_remove_file(entryList.at(i).canonicalFilePath()))
1443 break;
1449 tempFolder.rmdir(".");
1450 return !tempFolder.exists();
1454 * Register tool
1456 void lamexp_register_tool(const QString &toolName, LockedFile *file, unsigned int version)
1458 if(g_lamexp_tool_registry.contains(toolName.toLower()))
1460 throw "lamexp_register_tool: Tool is already registered!";
1463 g_lamexp_tool_registry.insert(toolName.toLower(), file);
1464 g_lamexp_tool_versions.insert(toolName.toLower(), version);
1468 * Check for tool
1470 bool lamexp_check_tool(const QString &toolName)
1472 return g_lamexp_tool_registry.contains(toolName.toLower());
1476 * Lookup tool path
1478 const QString lamexp_lookup_tool(const QString &toolName)
1480 if(g_lamexp_tool_registry.contains(toolName.toLower()))
1482 return g_lamexp_tool_registry.value(toolName.toLower())->filePath();
1484 else
1486 return QString();
1491 * Lookup tool version
1493 unsigned int lamexp_tool_version(const QString &toolName)
1495 if(g_lamexp_tool_versions.contains(toolName.toLower()))
1497 return g_lamexp_tool_versions.value(toolName.toLower());
1499 else
1501 return UINT_MAX;
1506 * Version number to human-readable string
1508 const QString lamexp_version2string(const QString &pattern, unsigned int version, const QString &defaultText)
1510 if(version == UINT_MAX)
1512 return defaultText;
1515 QString result = pattern;
1516 int digits = result.count("?", Qt::CaseInsensitive);
1518 if(digits < 1)
1520 return result;
1523 int pos = 0;
1524 QString versionStr = QString().sprintf(QString().sprintf("%%0%du", digits).toLatin1().constData(), version);
1525 int index = result.indexOf("?", Qt::CaseInsensitive);
1527 while(index >= 0 && pos < versionStr.length())
1529 result[index] = versionStr[pos++];
1530 index = result.indexOf("?", Qt::CaseInsensitive);
1533 return result;
1537 * Register a new translation
1539 bool lamexp_translation_register(const QString &langId, const QString &qmFile, const QString &langName, unsigned int &systemId, unsigned int &country)
1541 if(qmFile.isEmpty() || langName.isEmpty() || systemId < 1)
1543 return false;
1546 g_lamexp_translation.files.insert(langId, qmFile);
1547 g_lamexp_translation.names.insert(langId, langName);
1548 g_lamexp_translation.sysid.insert(langId, systemId);
1549 g_lamexp_translation.cntry.insert(langId, country);
1551 return true;
1555 * Get list of all translations
1557 QStringList lamexp_query_translations(void)
1559 return g_lamexp_translation.files.keys();
1563 * Get translation name
1565 QString lamexp_translation_name(const QString &langId)
1567 return g_lamexp_translation.names.value(langId.toLower(), QString());
1571 * Get translation system id
1573 unsigned int lamexp_translation_sysid(const QString &langId)
1575 return g_lamexp_translation.sysid.value(langId.toLower(), 0);
1579 * Get translation script id
1581 unsigned int lamexp_translation_country(const QString &langId)
1583 return g_lamexp_translation.cntry.value(langId.toLower(), 0);
1587 * Install a new translator
1589 bool lamexp_install_translator(const QString &langId)
1591 bool success = false;
1593 if(langId.isEmpty() || langId.toLower().compare(LAMEXP_DEFAULT_LANGID) == 0)
1595 success = lamexp_install_translator_from_file(QString());
1597 else
1599 QString qmFile = g_lamexp_translation.files.value(langId.toLower(), QString());
1600 if(!qmFile.isEmpty())
1602 success = lamexp_install_translator_from_file(QString(":/localization/%1").arg(qmFile));
1604 else
1606 qWarning("Translation '%s' not available!", langId.toLatin1().constData());
1610 return success;
1614 * Install a new translator from file
1616 bool lamexp_install_translator_from_file(const QString &qmFile)
1618 bool success = false;
1620 if(!g_lamexp_currentTranslator)
1622 g_lamexp_currentTranslator = new QTranslator();
1625 if(!qmFile.isEmpty())
1627 QString qmPath = QFileInfo(qmFile).canonicalFilePath();
1628 QApplication::removeTranslator(g_lamexp_currentTranslator);
1629 success = g_lamexp_currentTranslator->load(qmPath);
1630 QApplication::installTranslator(g_lamexp_currentTranslator);
1631 if(!success)
1633 qWarning("Failed to load translation:\n\"%s\"", qmPath.toLatin1().constData());
1636 else
1638 QApplication::removeTranslator(g_lamexp_currentTranslator);
1639 success = true;
1642 return success;
1646 * Locate known folder on local system
1648 QString lamexp_known_folder(lamexp_known_folder_t folder_id)
1650 typedef HRESULT (WINAPI *SHGetKnownFolderPathFun)(__in const GUID &rfid, __in DWORD dwFlags, __in HANDLE hToken, __out PWSTR *ppszPath);
1651 typedef HRESULT (WINAPI *SHGetFolderPathFun)(__in HWND hwndOwner, __in int nFolder, __in HANDLE hToken, __in DWORD dwFlags, __out LPWSTR pszPath);
1653 static const int CSIDL_LOCAL_APPDATA = 0x001c;
1654 static const int CSIDL_PROGRAM_FILES = 0x0026;
1655 static const int CSIDL_SYSTEM_FOLDER = 0x0025;
1656 static const GUID GUID_LOCAL_APPDATA = {0xF1B32785,0x6FBA,0x4FCF,{0x9D,0x55,0x7B,0x8E,0x7F,0x15,0x70,0x91}};
1657 static const GUID GUID_LOCAL_APPDATA_LOW = {0xA520A1A4,0x1780,0x4FF6,{0xBD,0x18,0x16,0x73,0x43,0xC5,0xAF,0x16}};
1658 static const GUID GUID_PROGRAM_FILES = {0x905e63b6,0xc1bf,0x494e,{0xb2,0x9c,0x65,0xb7,0x32,0xd3,0xd2,0x1a}};
1659 static const GUID GUID_SYSTEM_FOLDER = {0x1AC14E77,0x02E7,0x4E5D,{0xB7,0x44,0x2E,0xB1,0xAE,0x51,0x98,0xB7}};
1661 static QLibrary *Kernel32Lib = NULL;
1662 static SHGetKnownFolderPathFun SHGetKnownFolderPathPtr = NULL;
1663 static SHGetFolderPathFun SHGetFolderPathPtr = NULL;
1665 if((!SHGetKnownFolderPathPtr) && (!SHGetFolderPathPtr))
1667 if(!Kernel32Lib) Kernel32Lib = new QLibrary("shell32.dll");
1668 SHGetKnownFolderPathPtr = (SHGetKnownFolderPathFun) Kernel32Lib->resolve("SHGetKnownFolderPath");
1669 SHGetFolderPathPtr = (SHGetFolderPathFun) Kernel32Lib->resolve("SHGetFolderPathW");
1672 int folderCSIDL = -1;
1673 GUID folderGUID = {0x0000,0x0000,0x0000,{0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}};
1675 switch(folder_id)
1677 case lamexp_folder_localappdata:
1678 folderCSIDL = CSIDL_LOCAL_APPDATA;
1679 folderGUID = GUID_LOCAL_APPDATA;
1680 break;
1681 case lamexp_folder_programfiles:
1682 folderCSIDL = CSIDL_PROGRAM_FILES;
1683 folderGUID = GUID_PROGRAM_FILES;
1684 break;
1685 case lamexp_folder_systemfolder:
1686 folderCSIDL = CSIDL_SYSTEM_FOLDER;
1687 folderGUID = GUID_SYSTEM_FOLDER;
1688 break;
1689 default:
1690 return QString();
1691 break;
1694 QString folder;
1696 if(SHGetKnownFolderPathPtr)
1698 WCHAR *path = NULL;
1699 if(SHGetKnownFolderPathPtr(folderGUID, 0x00008000, NULL, &path) == S_OK)
1701 //MessageBoxW(0, path, L"SHGetKnownFolderPath", MB_TOPMOST);
1702 QDir folderTemp = QDir(QDir::fromNativeSeparators(QString::fromUtf16(reinterpret_cast<const unsigned short*>(path))));
1703 if(!folderTemp.exists())
1705 folderTemp.mkpath(".");
1707 if(folderTemp.exists())
1709 folder = folderTemp.canonicalPath();
1711 CoTaskMemFree(path);
1714 else if(SHGetFolderPathPtr)
1716 WCHAR *path = new WCHAR[4096];
1717 if(SHGetFolderPathPtr(NULL, folderCSIDL, NULL, NULL, path) == S_OK)
1719 //MessageBoxW(0, path, L"SHGetFolderPathW", MB_TOPMOST);
1720 QDir folderTemp = QDir(QDir::fromNativeSeparators(QString::fromUtf16(reinterpret_cast<const unsigned short*>(path))));
1721 if(!folderTemp.exists())
1723 folderTemp.mkpath(".");
1725 if(folderTemp.exists())
1727 folder = folderTemp.canonicalPath();
1730 delete [] path;
1733 return folder;
1737 * Safely remove a file
1739 bool lamexp_remove_file(const QString &filename)
1741 if(!QFileInfo(filename).exists() || !QFileInfo(filename).isFile())
1743 return true;
1745 else
1747 if(!QFile::remove(filename))
1749 DWORD attributes = GetFileAttributesW(QWCHAR(filename));
1750 SetFileAttributesW(QWCHAR(filename), (attributes & (~FILE_ATTRIBUTE_READONLY)));
1751 if(!QFile::remove(filename))
1753 qWarning("Could not delete \"%s\"", filename.toLatin1().constData());
1754 return false;
1756 else
1758 return true;
1761 else
1763 return true;
1769 * Check if visual themes are enabled (WinXP and later)
1771 bool lamexp_themes_enabled(void)
1773 typedef int (WINAPI *IsAppThemedFun)(void);
1775 bool isAppThemed = false;
1776 QLibrary uxTheme(QString("%1/UxTheme.dll").arg(lamexp_known_folder(lamexp_folder_systemfolder)));
1777 IsAppThemedFun IsAppThemedPtr = (IsAppThemedFun) uxTheme.resolve("IsAppThemed");
1779 if(IsAppThemedPtr)
1781 isAppThemed = IsAppThemedPtr();
1782 if(!isAppThemed)
1784 qWarning("Theme support is disabled for this process!");
1788 return isAppThemed;
1792 * Get number of free bytes on disk
1794 unsigned __int64 lamexp_free_diskspace(const QString &path, bool *ok)
1796 ULARGE_INTEGER freeBytesAvailable, totalNumberOfBytes, totalNumberOfFreeBytes;
1797 if(GetDiskFreeSpaceExW(reinterpret_cast<const wchar_t*>(QDir::toNativeSeparators(path).utf16()), &freeBytesAvailable, &totalNumberOfBytes, &totalNumberOfFreeBytes))
1799 if(ok) *ok = true;
1800 return freeBytesAvailable.QuadPart;
1802 else
1804 if(ok) *ok = false;
1805 return 0;
1810 * Check if computer does support hibernation
1812 bool lamexp_is_hibernation_supported(void)
1814 bool hibernationSupported = false;
1816 SYSTEM_POWER_CAPABILITIES pwrCaps;
1817 SecureZeroMemory(&pwrCaps, sizeof(SYSTEM_POWER_CAPABILITIES));
1819 if(GetPwrCapabilities(&pwrCaps))
1821 hibernationSupported = pwrCaps.SystemS4 && pwrCaps.HiberFilePresent;
1824 return hibernationSupported;
1828 * Shutdown the computer
1830 bool lamexp_shutdown_computer(const QString &message, const unsigned long timeout, const bool forceShutdown, const bool hibernate)
1832 HANDLE hToken = NULL;
1834 if(OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
1836 TOKEN_PRIVILEGES privileges;
1837 memset(&privileges, 0, sizeof(TOKEN_PRIVILEGES));
1838 privileges.PrivilegeCount = 1;
1839 privileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
1841 if(LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &privileges.Privileges[0].Luid))
1843 if(AdjustTokenPrivileges(hToken, FALSE, &privileges, NULL, NULL, NULL))
1845 if(hibernate)
1847 if(SetSuspendState(TRUE, TRUE, TRUE))
1849 return true;
1852 const DWORD reason = SHTDN_REASON_MAJOR_APPLICATION | SHTDN_REASON_FLAG_PLANNED;
1853 return InitiateSystemShutdownEx(NULL, const_cast<wchar_t*>(QWCHAR(message)), timeout, forceShutdown ? TRUE : FALSE, FALSE, reason);
1858 return false;
1862 * Make a window blink (to draw user's attention)
1864 void lamexp_blink_window(QWidget *poWindow, unsigned int count, unsigned int delay)
1866 static QMutex blinkMutex;
1868 const double maxOpac = 1.0;
1869 const double minOpac = 0.3;
1870 const double delOpac = 0.1;
1872 if(!blinkMutex.tryLock())
1874 qWarning("Blinking is already in progress, skipping!");
1875 return;
1880 const int steps = static_cast<int>(ceil(maxOpac - minOpac) / delOpac);
1881 const int sleep = static_cast<int>(floor(static_cast<double>(delay) / static_cast<double>(steps)));
1882 const double opacity = poWindow->windowOpacity();
1884 for(unsigned int i = 0; i < count; i++)
1886 for(double x = maxOpac; x >= minOpac; x -= delOpac)
1888 poWindow->setWindowOpacity(x);
1889 QApplication::processEvents();
1890 Sleep(sleep);
1893 for(double x = minOpac; x <= maxOpac; x += delOpac)
1895 poWindow->setWindowOpacity(x);
1896 QApplication::processEvents();
1897 Sleep(sleep);
1901 poWindow->setWindowOpacity(opacity);
1902 QApplication::processEvents();
1903 blinkMutex.unlock();
1905 catch (...)
1907 blinkMutex.unlock();
1908 qWarning("Exception error while blinking!");
1913 * Remove forbidden characters from a filename
1915 const QString lamexp_clean_filename(const QString &str)
1917 QString newStr(str);
1919 newStr.replace("\\", "-");
1920 newStr.replace(" / ", ", ");
1921 newStr.replace("/", ",");
1922 newStr.replace(":", "-");
1923 newStr.replace("*", "x");
1924 newStr.replace("?", "");
1925 newStr.replace("<", "[");
1926 newStr.replace(">", "]");
1927 newStr.replace("|", "!");
1929 return newStr.simplified();
1933 * Remove forbidden characters from a file path
1935 const QString lamexp_clean_filepath(const QString &str)
1937 QStringList parts = QString(str).replace("\\", "/").split("/");
1939 for(int i = 0; i < parts.count(); i++)
1941 parts[i] = lamexp_clean_filename(parts[i]);
1944 return parts.join("/");
1948 * Get a list of all available Qt Text Codecs
1950 QStringList lamexp_available_codepages(bool noAliases)
1952 QStringList codecList;
1954 QList<QByteArray> availableCodecs = QTextCodec::availableCodecs();
1955 while(!availableCodecs.isEmpty())
1957 QByteArray current = availableCodecs.takeFirst();
1958 if(!(current.startsWith("system") || current.startsWith("System")))
1960 codecList << QString::fromLatin1(current.constData(), current.size());
1961 if(noAliases)
1963 if(QTextCodec *currentCodec = QTextCodec::codecForName(current.constData()))
1966 QList<QByteArray> aliases = currentCodec->aliases();
1967 while(!aliases.isEmpty()) availableCodecs.removeAll(aliases.takeFirst());
1973 return codecList;
1977 * Finalization function (final clean-up)
1979 void lamexp_finalization(void)
1981 qDebug("lamexp_finalization()");
1983 //Free all tools
1984 if(!g_lamexp_tool_registry.isEmpty())
1986 QStringList keys = g_lamexp_tool_registry.keys();
1987 for(int i = 0; i < keys.count(); i++)
1989 LAMEXP_DELETE(g_lamexp_tool_registry[keys.at(i)]);
1991 g_lamexp_tool_registry.clear();
1992 g_lamexp_tool_versions.clear();
1995 //Delete temporary files
1996 if(!g_lamexp_temp_folder.isEmpty())
1998 for(int i = 0; i < 100; i++)
2000 if(lamexp_clean_folder(g_lamexp_temp_folder))
2002 break;
2004 Sleep(125);
2006 g_lamexp_temp_folder.clear();
2009 //Clear languages
2010 if(g_lamexp_currentTranslator)
2012 QApplication::removeTranslator(g_lamexp_currentTranslator);
2013 LAMEXP_DELETE(g_lamexp_currentTranslator);
2015 g_lamexp_translation.files.clear();
2016 g_lamexp_translation.names.clear();
2018 //Destroy Qt application object
2019 QApplication *application = dynamic_cast<QApplication*>(QApplication::instance());
2020 LAMEXP_DELETE(application);
2022 //Detach from shared memory
2023 if(g_lamexp_ipc_ptr.sharedmem) g_lamexp_ipc_ptr.sharedmem->detach();
2024 LAMEXP_DELETE(g_lamexp_ipc_ptr.sharedmem);
2025 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_read);
2026 LAMEXP_DELETE(g_lamexp_ipc_ptr.semaphore_write);
2028 //Close log file
2029 if(g_lamexp_log_file)
2031 fclose(g_lamexp_log_file);
2032 g_lamexp_log_file = NULL;
2037 * Initialize debug thread
2039 static const HANDLE g_debug_thread = LAMEXP_DEBUG ? NULL : lamexp_debug_thread_init();
2042 * Get number private bytes [debug only]
2044 SIZE_T lamexp_dbg_private_bytes(void)
2046 #if LAMEXP_DEBUG
2047 PROCESS_MEMORY_COUNTERS_EX memoryCounters;
2048 memoryCounters.cb = sizeof(PROCESS_MEMORY_COUNTERS_EX);
2049 GetProcessMemoryInfo(GetCurrentProcess(), (PPROCESS_MEMORY_COUNTERS) &memoryCounters, sizeof(PROCESS_MEMORY_COUNTERS_EX));
2050 return memoryCounters.PrivateUsage;
2051 #else
2052 throw "Cannot call this function in a non-debug build!";
2053 #endif //LAMEXP_DEBUG