/////////////////////////////////////////////////////////////////////////////// // Simple x264 Launcher // Copyright (C) 2004-2014 LoRd_MuldeR // // This program is free software; you can redistribute it and/or modify // it under the terms of the GNU General Public License as published by // the Free Software Foundation; either version 2 of the License, or // (at your option) any later version. // // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU General Public License for more details. // // You should have received a copy of the GNU General Public License along // with this program; if not, write to the Free Software Foundation, Inc., // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. // // http://www.gnu.org/licenses/gpl-2.0.txt /////////////////////////////////////////////////////////////////////////////// //x264 includes #include "global.h" #include "targetver.h" //Windows includes #define NOMINMAX #define WIN32_LEAN_AND_MEAN #include #include #include #include #include #include //C++ includes #include #include #include #include //VLD #include //Version #define ENABLE_X264_VERSION_INCLUDE #include "version.h" #undef ENABLE_X264_VERSION_INCLUDE //Qt includes #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include //CRT includes #include #include #include #include #include //Debug only includes #if X264_DEBUG #include #endif //Global types typedef HRESULT (WINAPI *SHGetKnownFolderPath_t)(const GUID &rfid, DWORD dwFlags, HANDLE hToken, PWSTR *ppszPath); typedef HRESULT (WINAPI *SHGetFolderPath_t)(HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPWSTR pszPath); //Global vars static bool g_x264_console_attached = false; static QMutex g_x264_message_mutex; static const DWORD g_main_thread_id = GetCurrentThreadId(); static FILE *g_x264_log_file = NULL; static QDate g_x264_version_date; //Const static const char *g_x264_months[] = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"}; static const char *g_x264_imageformats[] = {"png", "jpg", "gif", "ico", "svg", NULL}; //Build version static const struct { unsigned int ver_major; unsigned int ver_minor; unsigned int ver_patch; unsigned int ver_build; const char* ver_date; const char* ver_time; } g_x264_version = { (VER_X264_MAJOR), (VER_X264_MINOR), (VER_X264_PATCH), (VER_X264_BUILD), __DATE__, __TIME__, }; //CLI Arguments static struct { QStringList *list; QReadWriteLock lock; } g_x264_argv; //OS Version static struct { bool bInitialized; x264_os_version_t version; QReadWriteLock lock; } g_x264_os_version; //Special folders static struct { QMap *knownFolders; SHGetKnownFolderPath_t getKnownFolderPath; SHGetFolderPath_t getFolderPath; QReadWriteLock lock; } g_x264_known_folder; //%TEMP% folder static struct { QString *path; QReadWriteLock lock; } g_x264_temp_folder; //Wine detection static struct { bool bInitialized; bool bIsWine; QReadWriteLock lock; } g_x264_wine; //Portable Mode static struct { bool bInitialized; bool bPortableModeEnabled; QReadWriteLock lock; } g_x264_portable; //Known Windows versions - maps marketing names to the actual Windows NT versions const x264_os_version_t x264_winver_win2k = {5,0}; const x264_os_version_t x264_winver_winxp = {5,1}; const x264_os_version_t x264_winver_xpx64 = {5,2}; const x264_os_version_t x264_winver_vista = {6,0}; const x264_os_version_t x264_winver_win70 = {6,1}; const x264_os_version_t x264_winver_win80 = {6,2}; const x264_os_version_t x264_winver_win81 = {6,3}; //GURU MEDITATION static const char *GURU_MEDITATION = "\n\nGURU MEDITATION !!!\n\n"; /////////////////////////////////////////////////////////////////////////////// // MACROS /////////////////////////////////////////////////////////////////////////////// //String helper #define CLEAN_OUTPUT_STRING(STR) do \ { \ const char CTRL_CHARS[3] = { '\r', '\n', '\t' }; \ for(size_t i = 0; i < 3; i++) \ { \ while(char *pos = strchr((STR), CTRL_CHARS[i])) *pos = char(0x20); \ } \ } \ while(0) //String helper #define TRIM_LEFT(STR) do \ { \ const char WHITE_SPACE[4] = { char(0x20), '\r', '\n', '\t' }; \ for(size_t i = 0; i < 4; i++) \ { \ while(*(STR) == WHITE_SPACE[i]) (STR)++; \ } \ } \ while(0) //Check for CLI flag static inline bool _CHECK_FLAG(const int argc, char **argv, const char *flag) { for(int i = 1; i < argc; i++) { if(_stricmp(argv[i], flag) == 0) return true; } return false; } #define CHECK_FLAG(FLAG) _CHECK_FLAG(argc, argv, "--" FLAG) #define X264_ZERO_MEMORY(X) SecureZeroMemory(&X, sizeof(X)) /////////////////////////////////////////////////////////////////////////////// // COMPILER INFO /////////////////////////////////////////////////////////////////////////////// /* * Disclaimer: Parts of the following code were borrowed from MPC-HC project: http://mpc-hc.sf.net/ */ //Compiler detection #if defined(__INTEL_COMPILER) #if (__INTEL_COMPILER >= 1300) static const char *g_x264_version_compiler = "ICL 13." X264_MAKE_STR(__INTEL_COMPILER_BUILD_DATE); #elif (__INTEL_COMPILER >= 1200) static const char *g_x264_version_compiler = "ICL 12." X264_MAKE_STR(__INTEL_COMPILER_BUILD_DATE); #elif (__INTEL_COMPILER >= 1100) static const char *g_x264_version_compiler = "ICL 11.x"; #elif (__INTEL_COMPILER >= 1000) static const char *g_x264_version_compiler = "ICL 10.x"; #else #error Compiler is not supported! #endif #elif defined(_MSC_VER) #if (_MSC_VER == 1800) #if (_MSC_FULL_VER < 180021005) static const char *g_x264_version_compiler = "MSVC 2013-Beta"; #elif (_MSC_FULL_VER == 180021005) static const char *g_x264_version_compiler = "MSVC 2013"; #else #error Compiler version is not supported yet! #endif #elif (_MSC_VER == 1700) #if (_MSC_FULL_VER < 170050727) static const char *g_x264_version_compiler = "MSVC 2012-Beta"; #elif (_MSC_FULL_VER < 170051020) static const char *g_x264_version_compiler = "MSVC 2012"; #elif (_MSC_FULL_VER < 170051106) static const char *g_x264_version_compiler = "MSVC 2012.1-CTP"; #elif (_MSC_FULL_VER < 170060315) static const char *g_x264_version_compiler = "MSVC 2012.1"; #elif (_MSC_FULL_VER < 170060610) static const char *g_x264_version_compiler = "MSVC 2012.2"; #elif (_MSC_FULL_VER == 170060610) static const char *g_x264_version_compiler = "MSVC 2012.3"; #else #error Compiler version is not supported yet! #endif #elif (_MSC_VER == 1600) #if (_MSC_FULL_VER < 160040219) static const char *g_x264_version_compiler = "MSVC 2010"; #elif (_MSC_FULL_VER == 160040219) static const char *g_x264_version_compiler = "MSVC 2010-SP1"; #else #error Compiler version is not supported yet! #endif #elif (_MSC_VER == 1500) #if (_MSC_FULL_VER >= 150030729) static const char *g_x264_version_compiler = "MSVC 2008-SP1"; #else static const char *g_x264_version_compiler = "MSVC 2008"; #endif #else #error Compiler is not supported! #endif // Note: /arch:SSE and /arch:SSE2 are only available for the x86 platform #if !defined(_M_X64) && defined(_M_IX86_FP) #if (_M_IX86_FP == 1) X264_COMPILER_WARNING("SSE instruction set is enabled!") #elif (_M_IX86_FP == 2) X264_COMPILER_WARNING("SSE2 (or higher) instruction set is enabled!") #endif #endif #else #error Compiler is not supported! #endif //Architecture detection #if defined(_M_X64) static const char *g_x264_version_arch = "x64"; #elif defined(_M_IX86) static const char *g_x264_version_arch = "x86"; #else #error Architecture is not supported! #endif /////////////////////////////////////////////////////////////////////////////// // GLOBAL FUNCTIONS /////////////////////////////////////////////////////////////////////////////// static __forceinline bool x264_check_for_debugger(void); /* * Suspend calling thread for N milliseconds */ inline void x264_sleep(const unsigned int delay) { Sleep(delay); } /* * Global exception handler */ LONG WINAPI x264_exception_handler(__in struct _EXCEPTION_POINTERS *ExceptionInfo) { if(GetCurrentThreadId() != g_main_thread_id) { HANDLE mainThread = OpenThread(THREAD_TERMINATE, FALSE, g_main_thread_id); if(mainThread) TerminateThread(mainThread, ULONG_MAX); } x264_fatal_exit(L"Unhandeled exception handler invoked, application will exit!"); return LONG_MAX; } /* * Invalid parameters handler */ void x264_invalid_param_handler(const wchar_t*, const wchar_t*, const wchar_t*, unsigned int, uintptr_t) { if(GetCurrentThreadId() != g_main_thread_id) { HANDLE mainThread = OpenThread(THREAD_TERMINATE, FALSE, g_main_thread_id); if(mainThread) TerminateThread(mainThread, ULONG_MAX); } x264_fatal_exit(L"Invalid parameter handler invoked, application will exit!"); } /* * Get a random string */ QString x264_rand_str(const bool bLong) { const QUuid uuid = QUuid::createUuid().toString(); const unsigned int u1 = uuid.data1; const unsigned int u2 = (((unsigned int)(uuid.data2)) << 16) | ((unsigned int)(uuid.data3)); const unsigned int u3 = (((unsigned int)(uuid.data4[0])) << 24) | (((unsigned int)(uuid.data4[1])) << 16) | (((unsigned int)(uuid.data4[2])) << 8) | ((unsigned int)(uuid.data4[3])); const unsigned int u4 = (((unsigned int)(uuid.data4[4])) << 24) | (((unsigned int)(uuid.data4[5])) << 16) | (((unsigned int)(uuid.data4[6])) << 8) | ((unsigned int)(uuid.data4[7])); return bLong ? QString().sprintf("%08x%08x%08x%08x", u1, u2, u3, u4) : QString().sprintf("%08x%08x", (u1 ^ u2), (u3 ^ u4)); } /* * Robert Jenkins' 96 bit Mix Function * Source: http://www.concentric.net/~Ttwang/tech/inthash.htm */ static unsigned int x264_mix(const unsigned int x, const unsigned int y, const unsigned int z) { unsigned int a = x; unsigned int b = y; unsigned int c = z; a=a-b; a=a-c; a=a^(c >> 13); b=b-c; b=b-a; b=b^(a << 8); c=c-a; c=c-b; c=c^(b >> 13); a=a-b; a=a-c; a=a^(c >> 12); b=b-c; b=b-a; b=b^(a << 16); c=c-a; c=c-b; c=c^(b >> 5); a=a-b; a=a-c; a=a^(c >> 3); b=b-c; b=b-a; b=b^(a << 10); c=c-a; c=c-b; c=c^(b >> 15); return c; } /* * Seeds the random number generator * Note: Altough rand_s() doesn't need a seed, this must be called pripr to x264_rand(), just to to be sure! */ void x264_seed_rand(void) { qsrand(x264_mix(clock(), time(NULL), _getpid())); } /* * Returns a randum number * Note: This function uses rand_s() if available, but falls back to qrand() otherwise */ unsigned int x264_rand(void) { quint32 rnd = 0; if(rand_s(&rnd) == 0) { return rnd; } for(size_t i = 0; i < sizeof(unsigned int); i++) { rnd = (rnd << 8) ^ qrand(); } return rnd; } /* * Change console text color */ static void x264_console_color(FILE* file, WORD attributes) { const HANDLE hConsole = (HANDLE)(_get_osfhandle(_fileno(file))); if((hConsole != NULL) && (hConsole != INVALID_HANDLE_VALUE)) { SetConsoleTextAttribute(hConsole, attributes); } } /* * Determines the current date, resistant against certain manipulations */ QDate x264_current_date_safe(void) { const DWORD MAX_PROC = 1024; DWORD *processes = new DWORD[MAX_PROC]; DWORD bytesReturned = 0; if(!EnumProcesses(processes, sizeof(DWORD) * MAX_PROC, &bytesReturned)) { X264_DELETE_ARRAY(processes); return QDate::currentDate(); } const DWORD procCount = bytesReturned / sizeof(DWORD); ULARGE_INTEGER lastStartTime; memset(&lastStartTime, 0, sizeof(ULARGE_INTEGER)); for(DWORD i = 0; i < procCount; i++) { HANDLE hProc = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, processes[i]); if(hProc) { FILETIME processTime[4]; if(GetProcessTimes(hProc, &processTime[0], &processTime[1], &processTime[2], &processTime[3])) { ULARGE_INTEGER timeCreation; timeCreation.LowPart = processTime[0].dwLowDateTime; timeCreation.HighPart = processTime[0].dwHighDateTime; if(timeCreation.QuadPart > lastStartTime.QuadPart) { lastStartTime.QuadPart = timeCreation.QuadPart; } } CloseHandle(hProc); } } X264_DELETE_ARRAY(processes); FILETIME lastStartTime_fileTime; lastStartTime_fileTime.dwHighDateTime = lastStartTime.HighPart; lastStartTime_fileTime.dwLowDateTime = lastStartTime.LowPart; FILETIME lastStartTime_localTime; if(!FileTimeToLocalFileTime(&lastStartTime_fileTime, &lastStartTime_localTime)) { memcpy(&lastStartTime_localTime, &lastStartTime_fileTime, sizeof(FILETIME)); } SYSTEMTIME lastStartTime_system; if(!FileTimeToSystemTime(&lastStartTime_localTime, &lastStartTime_system)) { memset(&lastStartTime_system, 0, sizeof(SYSTEMTIME)); lastStartTime_system.wYear = 1970; lastStartTime_system.wMonth = lastStartTime_system.wDay = 1; } const QDate currentDate = QDate::currentDate(); const QDate processDate = QDate(lastStartTime_system.wYear, lastStartTime_system.wMonth, lastStartTime_system.wDay); return (currentDate >= processDate) ? currentDate : processDate; } /* * Output logging message to console */ static void x264_write_console(const int type, const char *msg) { __try { if(_isatty(_fileno(stderr))) { UINT oldOutputCP = GetConsoleOutputCP(); if(oldOutputCP != CP_UTF8) SetConsoleOutputCP(CP_UTF8); switch(type) { case QtCriticalMsg: case QtFatalMsg: x264_console_color(stderr, FOREGROUND_RED | FOREGROUND_INTENSITY); fprintf(stderr, GURU_MEDITATION); fprintf(stderr, "%s\n", msg); fflush(stderr); break; case QtWarningMsg: x264_console_color(stderr, FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY); fprintf(stderr, "%s\n", msg); fflush(stderr); break; default: x264_console_color(stderr, FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY); fprintf(stderr, "%s\n", msg); fflush(stderr); break; } x264_console_color(stderr, FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED); if(oldOutputCP != CP_UTF8) SetConsoleOutputCP(oldOutputCP); } } __except(1) { /*ignore any exception that might occur here!*/ } } /* * Output logging message to debugger */ static void x264_write_dbg_out(const int type, const char *msg) { const char *FORMAT = "[sx264l][%c] %s\n"; __try { char buffer[512]; const char* input = msg; TRIM_LEFT(input); switch(type) { case QtCriticalMsg: case QtFatalMsg: _snprintf_s(buffer, 512, _TRUNCATE, FORMAT, 'C', input); break; case QtWarningMsg: _snprintf_s(buffer, 512, _TRUNCATE, FORMAT, 'W', input); break; default: _snprintf_s(buffer, 512, _TRUNCATE, FORMAT, 'I', input); break; } char *temp = &buffer[0]; CLEAN_OUTPUT_STRING(temp); OutputDebugStringA(temp); } __except(1) { /*ignore any exception that might occur here!*/ } } /* * Output logging message to logfile */ static void x264_write_logfile(const int type, const char *msg) { const char *FORMAT = "[%c][%04u] %s\r\n"; __try { if(g_x264_log_file) { char buffer[512]; strncpy_s(buffer, 512, msg, _TRUNCATE); char *temp = &buffer[0]; TRIM_LEFT(temp); CLEAN_OUTPUT_STRING(temp); const unsigned int timestamp = static_cast(_time64(NULL) % 3600I64); switch(type) { case QtCriticalMsg: case QtFatalMsg: fprintf(g_x264_log_file, FORMAT, 'C', timestamp, temp); break; case QtWarningMsg: fprintf(g_x264_log_file, FORMAT, 'W', timestamp, temp); break; default: fprintf(g_x264_log_file, FORMAT, 'I', timestamp, temp); break; } fflush(g_x264_log_file); } } __except(1) { /*ignore any exception that might occur here!*/ } } /* * Qt message handler */ void x264_message_handler(QtMsgType type, const char *msg) { if((!msg) || (!(msg[0]))) { return; } QMutexLocker lock(&g_x264_message_mutex); if(g_x264_log_file) { x264_write_logfile(type, msg); } if(g_x264_console_attached) { x264_write_console(type, msg); } else { x264_write_dbg_out(type, msg); } if((type == QtCriticalMsg) || (type == QtFatalMsg)) { lock.unlock(); x264_fatal_exit(L"The application has encountered a critical error and will exit now!", QString::fromUtf8(msg).toLatin1().constData()); } } /* * Initialize the console */ void x264_init_console(const int argc, char **argv) { bool enableConsole = x264_is_prerelease() || (X264_DEBUG); if(_environ) { wchar_t *logfile = NULL; size_t logfile_len = 0; if(!_wdupenv_s(&logfile, &logfile_len, L"X264_LAUNCHER_LOGFILE")) { if(logfile && (logfile_len > 0)) { FILE *temp = NULL; if(!_wfopen_s(&temp, logfile, L"wb")) { fprintf(temp, "%c%c%c", 0xEF, 0xBB, 0xBF); g_x264_log_file = temp; } free(logfile); } } } if(!X264_DEBUG) { if(CHECK_FLAG("console")) enableConsole = true; if(CHECK_FLAG("no-console")) enableConsole = false; } if(enableConsole) { if(!g_x264_console_attached) { if(AllocConsole()) { SetConsoleCtrlHandler(NULL, TRUE); SetConsoleTitle(L"Simple x264 Launcher | Debug Console"); SetConsoleOutputCP(CP_UTF8); g_x264_console_attached = true; } } if(g_x264_console_attached) { //------------------------------------------------------------------- //See: http://support.microsoft.com/default.aspx?scid=kb;en-us;105305 //------------------------------------------------------------------- const int flags = _O_WRONLY | _O_U8TEXT; int hCrtStdOut = _open_osfhandle((intptr_t) GetStdHandle(STD_OUTPUT_HANDLE), flags); int hCrtStdErr = _open_osfhandle((intptr_t) GetStdHandle(STD_ERROR_HANDLE), flags); FILE *hfStdOut = (hCrtStdOut >= 0) ? _fdopen(hCrtStdOut, "wb") : NULL; FILE *hfStdErr = (hCrtStdErr >= 0) ? _fdopen(hCrtStdErr, "wb") : NULL; if(hfStdOut) { *stdout = *hfStdOut; std::cout.rdbuf(new std::filebuf(hfStdOut)); } if(hfStdErr) { *stderr = *hfStdErr; std::cerr.rdbuf(new std::filebuf(hfStdErr)); } } HWND hwndConsole = GetConsoleWindow(); if((hwndConsole != NULL) && (hwndConsole != INVALID_HANDLE_VALUE)) { HMENU hMenu = GetSystemMenu(hwndConsole, 0); EnableMenuItem(hMenu, SC_CLOSE, MF_BYCOMMAND | MF_GRAYED); RemoveMenu(hMenu, SC_CLOSE, MF_BYCOMMAND); SetWindowPos(hwndConsole, HWND_TOP, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE|SWP_NOZORDER|SWP_FRAMECHANGED); SetWindowLong(hwndConsole, GWL_STYLE, GetWindowLong(hwndConsole, GWL_STYLE) & (~WS_MAXIMIZEBOX) & (~WS_MINIMIZEBOX)); SetWindowPos(hwndConsole, HWND_TOP, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE|SWP_NOZORDER|SWP_FRAMECHANGED); } } } /* * Version info */ unsigned int x264_version_major(void) { return g_x264_version.ver_major; } unsigned int x264_version_minor(void) { return (g_x264_version.ver_minor * 10) + (g_x264_version.ver_patch % 10); } unsigned int x264_version_build(void) { return g_x264_version.ver_build; } const char *x264_version_compiler(void) { return g_x264_version_compiler; } const char *x264_version_arch(void) { return g_x264_version_arch; } /* * Get CLI arguments */ const QStringList &x264_arguments(void) { QReadLocker readLock(&g_x264_argv.lock); if(!g_x264_argv.list) { readLock.unlock(); QWriteLocker writeLock(&g_x264_argv.lock); g_x264_argv.list = new QStringList; int nArgs = 0; LPWSTR *szArglist = CommandLineToArgvW(GetCommandLineW(), &nArgs); if(NULL != szArglist) { for(int i = 0; i < nArgs; i++) { (*g_x264_argv.list) << WCHAR2QSTR(szArglist[i]); } LocalFree(szArglist); } else { qWarning("CommandLineToArgvW() has failed !!!"); } } return (*g_x264_argv.list); } /* * Check for portable mode */ bool x264_portable(void) { QReadLocker readLock(&g_x264_portable.lock); if(g_x264_portable.bInitialized) { return g_x264_portable.bPortableModeEnabled; } readLock.unlock(); QWriteLocker writeLock(&g_x264_portable.lock); if(!g_x264_portable.bInitialized) { if(VER_X264_PORTABLE_EDITION) { qWarning("Simple x264 Launcher portable edition!\n"); g_x264_portable.bPortableModeEnabled = true; } else { QString baseName = QFileInfo(QApplication::applicationFilePath()).completeBaseName(); int idx1 = baseName.indexOf("x264", 0, Qt::CaseInsensitive); int idx2 = baseName.lastIndexOf("portable", -1, Qt::CaseInsensitive); g_x264_portable.bPortableModeEnabled = (idx1 >= 0) && (idx2 >= 0) && (idx1 < idx2); } g_x264_portable.bInitialized = true; } return g_x264_portable.bPortableModeEnabled; } /* * Get data path (i.e. path to store config files) */ const QString &x264_data_path(void) { static QString pathCache; if(pathCache.isNull()) { if(!x264_portable()) { pathCache = QDesktopServices::storageLocation(QDesktopServices::DataLocation); } if(pathCache.isEmpty() || x264_portable()) { pathCache = QApplication::applicationDirPath(); } if(!QDir(pathCache).mkpath(".")) { qWarning("Data directory could not be created:\n%s\n", pathCache.toUtf8().constData()); pathCache = QDir::currentPath(); } } return pathCache; } /* * Get build date date */ const QDate &x264_version_date(void) { if(!g_x264_version_date.isValid()) { int date[3] = {0, 0, 0}; char temp[12] = {'\0'}; strncpy_s(temp, 12, g_x264_version.ver_date, _TRUNCATE); if(strlen(temp) == 11) { temp[3] = temp[6] = '\0'; date[2] = atoi(&temp[4]); date[0] = atoi(&temp[7]); for(int j = 0; j < 12; j++) { if(!_strcmpi(&temp[0], g_x264_months[j])) { date[1] = j+1; break; } } g_x264_version_date = QDate(date[0], date[1], date[2]); } if(!g_x264_version_date.isValid()) { qFatal("Internal error: Date format could not be recognized!"); } } return g_x264_version_date; } const char *x264_version_time(void) { return g_x264_version.ver_time; } bool x264_is_prerelease(void) { return (VER_X264_PRE_RELEASE); } /* * CPUID prototype (actual function is in ASM code) */ extern "C" { void x264_cpu_cpuid(unsigned int op, unsigned int *eax, unsigned int *ebx, unsigned int *ecx, unsigned int *edx); } /* * Detect CPU features */ x264_cpu_t x264_detect_cpu_features(const int argc, char **argv) { typedef BOOL (WINAPI *IsWow64ProcessFun)(__in HANDLE hProcess, __out PBOOL Wow64Process); x264_cpu_t features; SYSTEM_INFO systemInfo; unsigned int CPUInfo[4]; char CPUIdentificationString[0x40]; char CPUBrandString[0x40]; memset(&features, 0, sizeof(x264_cpu_t)); memset(&systemInfo, 0, sizeof(SYSTEM_INFO)); memset(CPUIdentificationString, 0, sizeof(CPUIdentificationString)); memset(CPUBrandString, 0, sizeof(CPUBrandString)); x264_cpu_cpuid(0, &CPUInfo[0], &CPUInfo[1], &CPUInfo[2], &CPUInfo[3]); memcpy(CPUIdentificationString, &CPUInfo[1], sizeof(int)); memcpy(CPUIdentificationString + 4, &CPUInfo[3], sizeof(int)); memcpy(CPUIdentificationString + 8, &CPUInfo[2], sizeof(int)); features.intel = (_stricmp(CPUIdentificationString, "GenuineIntel") == 0); strncpy_s(features.vendor, 0x40, CPUIdentificationString, _TRUNCATE); if(CPUInfo[0] >= 1) { x264_cpu_cpuid(1, &CPUInfo[0], &CPUInfo[1], &CPUInfo[2], &CPUInfo[3]); features.mmx = (CPUInfo[3] & 0x800000) || false; features.sse = (CPUInfo[3] & 0x2000000) || false; features.sse2 = (CPUInfo[3] & 0x4000000) || false; features.ssse3 = (CPUInfo[2] & 0x200) || false; features.sse3 = (CPUInfo[2] & 0x1) || false; features.ssse3 = (CPUInfo[2] & 0x200) || false; features.stepping = CPUInfo[0] & 0xf; features.model = ((CPUInfo[0] >> 4) & 0xf) + (((CPUInfo[0] >> 16) & 0xf) << 4); features.family = ((CPUInfo[0] >> 8) & 0xf) + ((CPUInfo[0] >> 20) & 0xff); if(features.sse) features.mmx2 = true; //MMXEXT is a subset of SSE! } x264_cpu_cpuid(0x80000000, &CPUInfo[0], &CPUInfo[1], &CPUInfo[2], &CPUInfo[3]); int nExIds = qMax(qMin(CPUInfo[0], 0x80000004), 0x80000000); if((_stricmp(CPUIdentificationString, "AuthenticAMD") == 0) && (nExIds >= 0x80000001U)) { x264_cpu_cpuid(0x80000001, &CPUInfo[0], &CPUInfo[1], &CPUInfo[2], &CPUInfo[3]); features.mmx2 = features.mmx2 || (CPUInfo[3] & 0x00400000U); } for(int i = 0x80000002; i <= nExIds; ++i) { x264_cpu_cpuid(i, &CPUInfo[0], &CPUInfo[1], &CPUInfo[2], &CPUInfo[3]); switch(i) { case 0x80000002: memcpy(CPUBrandString, CPUInfo, sizeof(CPUInfo)); break; case 0x80000003: memcpy(CPUBrandString + 16, CPUInfo, sizeof(CPUInfo)); break; case 0x80000004: memcpy(CPUBrandString + 32, CPUInfo, sizeof(CPUInfo)); break; } } strncpy_s(features.brand, 0x40, CPUBrandString, _TRUNCATE); if(strlen(features.brand) < 1) strncpy_s(features.brand, 0x40, "Unknown", _TRUNCATE); if(strlen(features.vendor) < 1) strncpy_s(features.vendor, 0x40, "Unknown", _TRUNCATE); #if (!(defined(_M_X64) || defined(_M_IA64))) QLibrary Kernel32Lib("kernel32.dll"); if(IsWow64ProcessFun IsWow64ProcessPtr = (IsWow64ProcessFun) Kernel32Lib.resolve("IsWow64Process")) { BOOL x64flag = FALSE; if(IsWow64ProcessPtr(GetCurrentProcess(), &x64flag)) { features.x64 = (x64flag == TRUE); } } #else features.x64 = true; #endif DWORD_PTR procAffinity, sysAffinity; if(GetProcessAffinityMask(GetCurrentProcess(), &procAffinity, &sysAffinity)) { for(DWORD_PTR mask = 1; mask; mask <<= 1) { features.count += ((sysAffinity & mask) ? (1) : (0)); } } if(features.count < 1) { GetNativeSystemInfo(&systemInfo); features.count = qBound(1UL, systemInfo.dwNumberOfProcessors, 64UL); } bool flag = false; if(CHECK_FLAG("force-cpu-no-64bit")) { flag = true; features.x64 = false; } if(CHECK_FLAG("force-cpu-no-sse")) { flag = true; features.sse = features.sse2 = features.sse3 = features.ssse3 = false; } if(CHECK_FLAG("force-cpu-no-intel")) { flag = true; features.intel = false; } if(flag) qWarning("CPU flags overwritten by user-defined parameters. Take care!\n"); return features; } /* * Verify a specific Windows version */ static bool x264_verify_os_version(const DWORD major, const DWORD minor) { OSVERSIONINFOEXW osvi; DWORDLONG dwlConditionMask = 0; //Initialize the OSVERSIONINFOEX structure memset(&osvi, 0, sizeof(OSVERSIONINFOEXW)); osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEXW); osvi.dwMajorVersion = major; osvi.dwMinorVersion = minor; osvi.dwPlatformId = VER_PLATFORM_WIN32_NT; //Initialize the condition mask VER_SET_CONDITION(dwlConditionMask, VER_MAJORVERSION, VER_GREATER_EQUAL); VER_SET_CONDITION(dwlConditionMask, VER_MINORVERSION, VER_GREATER_EQUAL); VER_SET_CONDITION(dwlConditionMask, VER_PLATFORMID, VER_EQUAL); // Perform the test const BOOL ret = VerifyVersionInfoW(&osvi, VER_MAJORVERSION | VER_MINORVERSION | VER_PLATFORMID, dwlConditionMask); //Error checking if(!ret) { if(GetLastError() != ERROR_OLD_WIN_VERSION) { qWarning("VerifyVersionInfo() system call has failed!"); } } return (ret != FALSE); } /* * Determine the *real* Windows version */ static bool x264_get_real_os_version(unsigned int *major, unsigned int *minor, bool *pbOverride) { *major = *minor = 0; *pbOverride = false; //Initialize local variables OSVERSIONINFOEXW osvi; memset(&osvi, 0, sizeof(OSVERSIONINFOEXW)); osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEXW); //Try GetVersionEx() first if(GetVersionExW((LPOSVERSIONINFOW)&osvi) == FALSE) { qWarning("GetVersionEx() has failed, cannot detect Windows version!"); return false; } //Make sure we are running on NT if(osvi.dwPlatformId == VER_PLATFORM_WIN32_NT) { *major = osvi.dwMajorVersion; *minor = osvi.dwMinorVersion; } else { qWarning("Not running on Windows NT, unsupported operating system!"); return false; } //Determine the real *major* version first forever { const DWORD nextMajor = (*major) + 1; if(x264_verify_os_version(nextMajor, 0)) { *pbOverride = true; *major = nextMajor; *minor = 0; continue; } break; } //Now also determine the real *minor* version forever { const DWORD nextMinor = (*minor) + 1; if(x264_verify_os_version((*major), nextMinor)) { *pbOverride = true; *minor = nextMinor; continue; } break; } return true; } /* * Get the native operating system version */ const x264_os_version_t &x264_get_os_version(void) { QReadLocker readLock(&g_x264_os_version.lock); //Already initialized? if(g_x264_os_version.bInitialized) { return g_x264_os_version.version; } readLock.unlock(); QWriteLocker writeLock(&g_x264_os_version.lock); //Detect OS version if(!g_x264_os_version.bInitialized) { unsigned int major, minor; bool oflag; if(x264_get_real_os_version(&major, &minor, &oflag)) { g_x264_os_version.version.versionMajor = major; g_x264_os_version.version.versionMinor = minor; g_x264_os_version.version.overrideFlag = oflag; g_x264_os_version.bInitialized = true; } else { qWarning("Failed to determin the operating system version!"); } } return g_x264_os_version.version; } /* * Check for compatibility mode */ static bool x264_check_compatibility_mode(const char *exportName, const char *executableName) { QLibrary kernel32("kernel32.dll"); if(exportName != NULL) { if(kernel32.resolve(exportName) != NULL) { qWarning("Function '%s' exported from 'kernel32.dll' -> Windows compatibility mode!", exportName); qFatal("%s", QApplication::tr("Executable '%1' doesn't support Windows compatibility mode.").arg(QString::fromLatin1(executableName)).toLatin1().constData()); return false; } } return true; } /* * Check if we are running under wine */ bool x264_detect_wine(void) { QReadLocker readLock(&g_x264_wine.lock); //Already initialized? if(g_x264_wine.bInitialized) { return g_x264_wine.bIsWine; } readLock.unlock(); QWriteLocker writeLock(&g_x264_wine.lock); if(!g_x264_wine.bInitialized) { g_x264_wine.bIsWine = false; QLibrary ntdll("ntdll.dll"); if(ntdll.load()) { if(ntdll.resolve("wine_nt_to_unix_file_name") != NULL) g_x264_wine.bIsWine = true; if(ntdll.resolve("wine_get_version") != NULL) g_x264_wine.bIsWine = true; ntdll.unload(); } g_x264_wine.bInitialized = true; } return g_x264_wine.bIsWine; } /* * Qt event filter */ static bool x264_event_filter(void *message, long *result) { if((!(X264_DEBUG)) && x264_check_for_debugger()) { x264_fatal_exit(L"Not a debug build. Please unload debugger and try again!"); } //switch(reinterpret_cast(message)->message) //{ //case WM_QUERYENDSESSION: // qWarning("WM_QUERYENDSESSION message received!"); // *result = x264_broadcast(x264_event_queryendsession, false) ? TRUE : FALSE; // return true; //case WM_ENDSESSION: // qWarning("WM_ENDSESSION message received!"); // if(reinterpret_cast(message)->wParam == TRUE) // { // x264_broadcast(x264_event_endsession, false); // if(QApplication *app = reinterpret_cast(QApplication::instance())) // { // app->closeAllWindows(); // app->quit(); // } // x264_finalization(); // exit(1); // } // *result = 0; // return true; //default: // /*ignore this message and let Qt handle it*/ // return false; //} return false; } /* * Check for process elevation */ static bool x264_process_is_elevated(bool *bIsUacEnabled = NULL) { bool bIsProcessElevated = false; if(bIsUacEnabled) *bIsUacEnabled = false; HANDLE hToken = NULL; if(OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken)) { TOKEN_ELEVATION_TYPE tokenElevationType; DWORD returnLength; if(GetTokenInformation(hToken, TokenElevationType, &tokenElevationType, sizeof(TOKEN_ELEVATION_TYPE), &returnLength)) { if(returnLength == sizeof(TOKEN_ELEVATION_TYPE)) { switch(tokenElevationType) { case TokenElevationTypeDefault: qDebug("Process token elevation type: Default -> UAC is disabled.\n"); break; case TokenElevationTypeFull: qWarning("Process token elevation type: Full -> potential security risk!\n"); bIsProcessElevated = true; if(bIsUacEnabled) *bIsUacEnabled = true; break; case TokenElevationTypeLimited: qDebug("Process token elevation type: Limited -> not elevated.\n"); if(bIsUacEnabled) *bIsUacEnabled = true; break; default: qWarning("Unknown tokenElevationType value: %d", tokenElevationType); break; } } else { qWarning("GetTokenInformation() return an unexpected size!"); } } CloseHandle(hToken); } else { qWarning("Failed to open process token!"); } return bIsProcessElevated; } /* * Check if the current user is an administartor (helper function) */ static bool x264_user_is_admin_helper(void) { HANDLE hToken = NULL; if(!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken)) { return false; } DWORD dwSize = 0; if(!GetTokenInformation(hToken, TokenGroups, NULL, 0, &dwSize)) { if(GetLastError() != ERROR_INSUFFICIENT_BUFFER) { CloseHandle(hToken); return false; } } PTOKEN_GROUPS lpGroups = (PTOKEN_GROUPS) malloc(dwSize); if(!lpGroups) { CloseHandle(hToken); return false; } if(!GetTokenInformation(hToken, TokenGroups, lpGroups, dwSize, &dwSize)) { free(lpGroups); CloseHandle(hToken); return false; } PSID lpSid = NULL; SID_IDENTIFIER_AUTHORITY Authority = {SECURITY_NT_AUTHORITY}; if(!AllocateAndInitializeSid(&Authority, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &lpSid)) { free(lpGroups); CloseHandle(hToken); return false; } bool bResult = false; for(DWORD i = 0; i < lpGroups->GroupCount; i++) { if(EqualSid(lpSid, lpGroups->Groups[i].Sid)) { bResult = true; break; } } FreeSid(lpSid); free(lpGroups); CloseHandle(hToken); return bResult; } /* * Check if the current user is an administartor */ bool x264_user_is_admin(void) { bool isAdmin = false; //Check for process elevation and UAC support first! if(x264_process_is_elevated(&isAdmin)) { qWarning("Process is elevated -> user is admin!"); return true; } //If not elevated and UAC is not available -> user must be in admin group! if(!isAdmin) { qDebug("UAC is disabled/unavailable -> checking for Administrators group"); isAdmin = x264_user_is_admin_helper(); } return isAdmin; } /* * Initialize Qt framework */ bool x264_init_qt(int &argc, char **argv) { static bool qt_initialized = false; typedef BOOL (WINAPI *SetDllDirectoryProc)(WCHAR *lpPathName); const QStringList &arguments = x264_arguments(); //Don't initialized again, if done already if(qt_initialized) { return true; } //Secure DLL loading QLibrary kernel32("kernel32.dll"); if(kernel32.load()) { SetDllDirectoryProc pSetDllDirectory = (SetDllDirectoryProc) kernel32.resolve("SetDllDirectoryW"); if(pSetDllDirectory != NULL) pSetDllDirectory(L""); } //Extract executable name from argv[] array QString executableName = QLatin1String("x264_launcher.exe"); if(arguments.count() > 0) { static const char *delimiters = "\\/:?"; executableName = arguments[0].trimmed(); for(int i = 0; delimiters[i]; i++) { int temp = executableName.lastIndexOf(QChar(delimiters[i])); if(temp >= 0) executableName = executableName.mid(temp + 1); } executableName = executableName.trimmed(); if(executableName.isEmpty()) { executableName = QLatin1String("x264_launcher.exe"); } } //Check Qt version #ifdef QT_BUILD_KEY qDebug("Using Qt v%s [%s], %s, %s", qVersion(), QLibraryInfo::buildDate().toString(Qt::ISODate).toLatin1().constData(), (qSharedBuild() ? "DLL" : "Static"), QLibraryInfo::buildKey().toLatin1().constData()); qDebug("Compiled with Qt v%s [%s], %s\n", QT_VERSION_STR, QT_PACKAGEDATE_STR, QT_BUILD_KEY); if(_stricmp(qVersion(), QT_VERSION_STR)) { qFatal("%s", QApplication::tr("Executable '%1' requires Qt v%2, but found Qt v%3.").arg(executableName, QString::fromLatin1(QT_VERSION_STR), QString::fromLatin1(qVersion())).toLatin1().constData()); return false; } if(QLibraryInfo::buildKey().compare(QString::fromLatin1(QT_BUILD_KEY), Qt::CaseInsensitive)) { qFatal("%s", QApplication::tr("Executable '%1' was built for Qt '%2', but found Qt '%3'.").arg(executableName, QString::fromLatin1(QT_BUILD_KEY), QLibraryInfo::buildKey()).toLatin1().constData()); return false; } #else qDebug("Using Qt v%s [%s], %s", qVersion(), QLibraryInfo::buildDate().toString(Qt::ISODate).toLatin1().constData(), (qSharedBuild() ? "DLL" : "Static")); qDebug("Compiled with Qt v%s [%s]\n", QT_VERSION_STR, QT_PACKAGEDATE_STR); #endif //Check the Windows version const x264_os_version_t &osVersionNo = x264_get_os_version(); if(osVersionNo < x264_winver_winxp) { qFatal("%s", QApplication::tr("Executable '%1' requires Windows XP or later.").arg(executableName).toLatin1().constData()); } //Supported Windows version? if(osVersionNo == x264_winver_winxp) { qDebug("Running on Windows XP or Windows XP Media Center Edition.\n"); //x264_check_compatibility_mode("GetLargePageMinimum", executableName); } else if(osVersionNo == x264_winver_xpx64) { qDebug("Running on Windows Server 2003, Windows Server 2003 R2 or Windows XP x64.\n"); //x264_check_compatibility_mode("GetLocaleInfoEx", executableName); } else if(osVersionNo == x264_winver_vista) { qDebug("Running on Windows Vista or Windows Server 2008.\n"); //x264_check_compatibility_mode("CreateRemoteThreadEx", executableName*/); } else if(osVersionNo == x264_winver_win70) { qDebug("Running on Windows 7 or Windows Server 2008 R2.\n"); //x264_check_compatibility_mode("CreateFile2", executableName); } else if(osVersionNo == x264_winver_win80) { qDebug("Running on Windows 8 or Windows Server 2012.\n"); //x264_check_compatibility_mode("FindPackagesByPackageFamily", executableName); } else if(osVersionNo == x264_winver_win81) { qDebug("Running on Windows 8.1 or Windows Server 2012 R2.\n"); //x264_check_compatibility_mode(NULL, executableName); } else { const QString message = QString().sprintf("Running on an unknown WindowsNT-based system (v%u.%u).", osVersionNo.versionMajor, osVersionNo.versionMinor); qWarning("%s\n", QUTF8(message)); MessageBoxW(NULL, QWCHAR(message), L"Simple x264 Launcher", MB_OK | MB_TOPMOST | MB_ICONWARNING); } //Check for compat mode if(osVersionNo.overrideFlag && (osVersionNo <= x264_winver_win81)) { qWarning("Windows compatibility mode detected!"); if(!arguments.contains("--ignore-compat-mode", Qt::CaseInsensitive)) { qFatal("%s", QApplication::tr("Executable '%1' doesn't support Windows compatibility mode.").arg(executableName).toLatin1().constData()); return false; } } //Check for Wine if(x264_detect_wine()) { qWarning("It appears we are running under Wine, unexpected things might happen!\n"); } //Set text Codec for locale QTextCodec::setCodecForLocale(QTextCodec::codecForName("UTF-8")); //Create Qt application instance QApplication *application = new QApplication(argc, argv); //Load plugins from application directory QCoreApplication::setLibraryPaths(QStringList() << QApplication::applicationDirPath()); qDebug("Library Path:\n%s\n", QUTF8(QApplication::libraryPaths().first())); //Create Qt application instance and setup version info application->setApplicationName("Simple x264 Launcher"); application->setApplicationVersion(QString().sprintf("%d.%02d", x264_version_major(), x264_version_minor())); application->setOrganizationName("LoRd_MuldeR"); application->setOrganizationDomain("mulder.at.gg"); application->setWindowIcon(QIcon(":/icons/movie.ico")); application->setEventFilter(x264_event_filter); //Check for supported image formats QList supportedFormats = QImageReader::supportedImageFormats(); for(int i = 0; g_x264_imageformats[i]; i++) { if(!supportedFormats.contains(g_x264_imageformats[i])) { qFatal("Qt initialization error: QImageIOHandler for '%s' missing!", g_x264_imageformats[i]); return false; } } //Add default translations /* QWriteLocker writeLockTranslations(&g_x264_translation.lock); if(!g_x264_translation.files) g_x264_translation.files = new QMap(); if(!g_x264_translation.names) g_x264_translation.names = new QMap(); g_x264_translation.files->insert(X264_DEFAULT_LANGID, ""); g_x264_translation.names->insert(X264_DEFAULT_LANGID, "English"); writeLockTranslations.unlock(); */ //Check for process elevation if(x264_process_is_elevated() && (!x264_detect_wine())) { QMessageBox messageBox(QMessageBox::Warning, "Simple x264 Launcher", "Simple x264 Launcher was started with 'elevated' rights, altough it does not need these rights.
Running an applications with unnecessary rights is a potential security risk!
", QMessageBox::NoButton, NULL, Qt::Dialog | Qt::MSWindowsFixedSizeDialogHint | Qt::WindowStaysOnTopHint); messageBox.addButton("Quit Program (Recommended)", QMessageBox::NoRole); messageBox.addButton("Ignore", QMessageBox::NoRole); if(messageBox.exec() == 0) { return false; } } //Update console icon, if a console is attached #if QT_VERSION < QT_VERSION_CHECK(5,0,0) if(g_x264_console_attached && (!x264_detect_wine())) { typedef DWORD (__stdcall *SetConsoleIconFun)(HICON); QLibrary kernel32("kernel32.dll"); if(kernel32.load()) { SetConsoleIconFun SetConsoleIconPtr = (SetConsoleIconFun) kernel32.resolve("SetConsoleIcon"); QPixmap pixmap = QIcon(":/icons/movie.ico").pixmap(16, 16); if((SetConsoleIconPtr != NULL) && (!pixmap.isNull())) SetConsoleIconPtr(pixmap.toWinHICON()); kernel32.unload(); } } #endif //Done qt_initialized = true; return true; } /* * Suspend or resume process */ bool x264_suspendProcess(const QProcess *proc, const bool suspend) { if(Q_PID pid = proc->pid()) { if(suspend) { return (SuspendThread(pid->hThread) != ((DWORD) -1)); } else { return (ResumeThread(pid->hThread) != ((DWORD) -1)); } } else { return false; } } /* * Convert path to short/ANSI path */ QString x264_path2ansi(const QString &longPath, bool makeLowercase) { QString shortPath = longPath; const QString longPathNative = QDir::toNativeSeparators(longPath); DWORD buffSize = GetShortPathNameW(QWCHAR(longPathNative), NULL, NULL); if(buffSize > 0) { wchar_t *buffer = (wchar_t*) _malloca(sizeof(wchar_t) * buffSize); DWORD result = GetShortPathNameW(QWCHAR(longPathNative), buffer, buffSize); if((result > 0) && (result < buffSize)) { shortPath = QDir::fromNativeSeparators(QString::fromUtf16(reinterpret_cast(buffer), result)); if(makeLowercase) { QFileInfo info(shortPath); shortPath = QString("%1/%2").arg(info.absolutePath(), info.fileName().toLower()); } } _freea(buffer); buffer = NULL; } return shortPath; } /* * Set the process priority class for current process */ bool x264_change_process_priority(const int priority) { return x264_change_process_priority(GetCurrentProcess(), priority); } /* * Set the process priority class for specified process */ bool x264_change_process_priority(const QProcess *proc, const int priority) { if(Q_PID qPid = proc->pid()) { return x264_change_process_priority(qPid->hProcess, priority); } else { return false; } } /* * Set the process priority class for specified process */ bool x264_change_process_priority(void *hProcess, const int priority) { bool ok = false; switch(qBound(-2, priority, 2)) { case 2: ok = (SetPriorityClass(hProcess, HIGH_PRIORITY_CLASS) == TRUE); break; case 1: if(!(ok = (SetPriorityClass(hProcess, ABOVE_NORMAL_PRIORITY_CLASS) == TRUE))) { ok = (SetPriorityClass(hProcess, HIGH_PRIORITY_CLASS) == TRUE); } break; case 0: ok = (SetPriorityClass(hProcess, NORMAL_PRIORITY_CLASS) == TRUE); break; case -1: if(!(ok = (SetPriorityClass(hProcess, BELOW_NORMAL_PRIORITY_CLASS) == TRUE))) { ok = (SetPriorityClass(hProcess, IDLE_PRIORITY_CLASS) == TRUE); } break; case -2: ok = (SetPriorityClass(hProcess, IDLE_PRIORITY_CLASS) == TRUE); break; } return ok; } /* * Play a sound (from resources) */ bool x264_play_sound(const unsigned short uiSoundIdx, const bool bAsync, const wchar_t *alias) { if(alias) { return PlaySound(alias, GetModuleHandle(NULL), (SND_ALIAS | (bAsync ? SND_ASYNC : SND_SYNC))) == TRUE; } else { return PlaySound(MAKEINTRESOURCE(uiSoundIdx), GetModuleHandle(NULL), (SND_RESOURCE | (bAsync ? SND_ASYNC : SND_SYNC))) == TRUE; } } /* * Current process ID */ unsigned int x264_process_id(void) { return GetCurrentProcessId(); } /* * Current process ID */ unsigned int x264_process_id(QProcess &process) { if(Q_PID pid = process.pid()) { return pid->dwProcessId; } return NULL; } /* * Make a window blink (to draw user's attention) */ void x264_blink_window(QWidget *poWindow, unsigned int count, unsigned int delay) { static QMutex blinkMutex; const double maxOpac = 1.0; const double minOpac = 0.3; const double delOpac = 0.1; if(!blinkMutex.tryLock()) { qWarning("Blinking is already in progress, skipping!"); return; } try { const int steps = static_cast(ceil(maxOpac - minOpac) / delOpac); const int sleep = static_cast(floor(static_cast(delay) / static_cast(steps))); const double opacity = poWindow->windowOpacity(); for(unsigned int i = 0; i < count; i++) { for(double x = maxOpac; x >= minOpac; x -= delOpac) { poWindow->setWindowOpacity(x); QApplication::processEvents(); Sleep(sleep); } for(double x = minOpac; x <= maxOpac; x += delOpac) { poWindow->setWindowOpacity(x); QApplication::processEvents(); Sleep(sleep); } } poWindow->setWindowOpacity(opacity); QApplication::processEvents(); blinkMutex.unlock(); } catch(...) { blinkMutex.unlock(); qWarning("Exception error while blinking!"); } } /* * Bring the specifed window to the front */ static bool x264_bring_to_front(const HWND hWin) { if(hWin) { const bool ret = (SetForegroundWindow(hWin) != FALSE); SwitchToThisWindow(hWin, TRUE); return ret; } return false; } /* * Bring the specifed window to the front */ bool x264_bring_to_front(const QWidget *win) { if(win) { return x264_bring_to_front(win->winId()); } return false; } /* * Bring window of the specifed process to the front (callback) */ static BOOL CALLBACK x264_bring_process_to_front_helper(HWND hwnd, LPARAM lParam) { DWORD processId = *reinterpret_cast(lParam); DWORD windowProcessId = NULL; GetWindowThreadProcessId(hwnd, &windowProcessId); if(windowProcessId == processId) { x264_bring_to_front(hwnd); return FALSE; } return TRUE; } /* * Bring window of the specifed process to the front */ bool x264_bring_process_to_front(const unsigned long pid) { return EnumWindows(x264_bring_process_to_front_helper, reinterpret_cast(&pid)) == TRUE; } /* * Check if file is a valid Win32/Win64 executable */ bool x264_is_executable(const QString &path) { bool bIsExecutable = false; DWORD binaryType; if(GetBinaryType(QWCHAR(QDir::toNativeSeparators(path)), &binaryType)) { bIsExecutable = (binaryType == SCS_32BIT_BINARY || binaryType == SCS_64BIT_BINARY); } return bIsExecutable; } /* * Read value from registry */ QString x264_query_reg_string(const bool bUser, const QString &path, const QString &name) { QString result; HKEY hKey = NULL; if(RegOpenKey((bUser ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE), QWCHAR(path), &hKey) == ERROR_SUCCESS) { const size_t DATA_LEN = 2048; wchar_t data[DATA_LEN]; DWORD type = REG_NONE, size = sizeof(wchar_t) * DATA_LEN; if(RegQueryValueEx(hKey, QWCHAR(name), NULL, &type, ((BYTE*)&data[0]), &size) == ERROR_SUCCESS) { if((type == REG_SZ) || (type == REG_EXPAND_SZ)) { result = WCHAR2QSTR(&data[0]); } } RegCloseKey(hKey); } return result; } /* * Locate known folder on local system */ const QString &x264_known_folder(x264_known_folder_t folder_id) { static const int CSIDL_FLAG_CREATE = 0x8000; typedef enum { KF_FLAG_CREATE = 0x00008000 } kf_flags_t; struct { const int csidl; const GUID guid; } static s_folders[] = { { 0x001c, {0xF1B32785,0x6FBA,0x4FCF,{0x9D,0x55,0x7B,0x8E,0x7F,0x15,0x70,0x91}} }, //CSIDL_LOCAL_APPDATA { 0x0026, {0x905e63b6,0xc1bf,0x494e,{0xb2,0x9c,0x65,0xb7,0x32,0xd3,0xd2,0x1a}} }, //CSIDL_PROGRAM_FILES { 0x0024, {0xF38BF404,0x1D43,0x42F2,{0x93,0x05,0x67,0xDE,0x0B,0x28,0xFC,0x23}} }, //CSIDL_WINDOWS_FOLDER { 0x0025, {0x1AC14E77,0x02E7,0x4E5D,{0xB7,0x44,0x2E,0xB1,0xAE,0x51,0x98,0xB7}} }, //CSIDL_SYSTEM_FOLDER }; size_t folderId = size_t(-1); switch(folder_id) { case x264_folder_localappdata: folderId = 0; break; case x264_folder_programfiles: folderId = 1; break; case x264_folder_systroot_dir: folderId = 2; break; case x264_folder_systemfolder: folderId = 3; break; } if(folderId == size_t(-1)) { qWarning("Invalid 'known' folder was requested!"); return *reinterpret_cast(NULL); } QReadLocker readLock(&g_x264_known_folder.lock); //Already in cache? if(g_x264_known_folder.knownFolders) { if(g_x264_known_folder.knownFolders->contains(folderId)) { return (*g_x264_known_folder.knownFolders)[folderId]; } } //Obtain write lock to initialize readLock.unlock(); QWriteLocker writeLock(&g_x264_known_folder.lock); //Still not in cache? if(g_x264_known_folder.knownFolders) { if(g_x264_known_folder.knownFolders->contains(folderId)) { return (*g_x264_known_folder.knownFolders)[folderId]; } } //Initialize on first call if(!g_x264_known_folder.knownFolders) { QLibrary shell32("shell32.dll"); if(shell32.load()) { g_x264_known_folder.getFolderPath = (SHGetFolderPath_t) shell32.resolve("SHGetFolderPathW"); g_x264_known_folder.getKnownFolderPath = (SHGetKnownFolderPath_t) shell32.resolve("SHGetKnownFolderPath"); } g_x264_known_folder.knownFolders = new QMap(); } QString folderPath; //Now try to get the folder path! if(g_x264_known_folder.getKnownFolderPath) { WCHAR *path = NULL; if(g_x264_known_folder.getKnownFolderPath(s_folders[folderId].guid, KF_FLAG_CREATE, NULL, &path) == S_OK) { //MessageBoxW(0, path, L"SHGetKnownFolderPath", MB_TOPMOST); QDir folderTemp = QDir(QDir::fromNativeSeparators(QString::fromUtf16(reinterpret_cast(path)))); if(folderTemp.exists()) { folderPath = folderTemp.canonicalPath(); } CoTaskMemFree(path); } } else if(g_x264_known_folder.getFolderPath) { WCHAR *path = new WCHAR[4096]; if(g_x264_known_folder.getFolderPath(NULL, s_folders[folderId].csidl | CSIDL_FLAG_CREATE, NULL, NULL, path) == S_OK) { //MessageBoxW(0, path, L"SHGetFolderPathW", MB_TOPMOST); QDir folderTemp = QDir(QDir::fromNativeSeparators(QString::fromUtf16(reinterpret_cast(path)))); if(folderTemp.exists()) { folderPath = folderTemp.canonicalPath(); } } X264_DELETE_ARRAY(path); } //Update cache g_x264_known_folder.knownFolders->insert(folderId, folderPath); return (*g_x264_known_folder.knownFolders)[folderId]; } /* * Try to initialize the folder (with *write* access) */ static QString x264_try_init_folder(const QString &folderPath) { static const char *DATA = "Lorem ipsum dolor sit amet, consectetur, adipisci velit!"; bool success = false; const QFileInfo folderInfo(folderPath); const QDir folderDir(folderInfo.absoluteFilePath()); //Create folder, if it does *not* exist yet for(int i = 0; i < 16; i++) { if(folderDir.exists()) break; folderDir.mkpath("."); } //Make sure folder exists now *and* is writable if(folderDir.exists()) { const QByteArray testData = QByteArray(DATA); for(int i = 0; i < 32; i++) { QFile testFile(folderDir.absoluteFilePath(QString("~%1.tmp").arg(x264_rand_str()))); if(testFile.open(QIODevice::ReadWrite | QIODevice::Truncate)) { if(testFile.write(testData) >= testData.size()) { success = true; } testFile.remove(); testFile.close(); } if(success) break; } } return (success ? folderDir.canonicalPath() : QString()); } /* * Detect the TEMP directory */ const QString &x264_temp_directory(void) { QReadLocker readLock(&g_x264_temp_folder.lock); if(g_x264_temp_folder.path) { return *g_x264_temp_folder.path; } readLock.unlock(); QWriteLocker writeLock(&g_x264_temp_folder.lock); if(!g_x264_temp_folder.path) { //Try %TEMP% first g_x264_temp_folder.path = new QString(x264_try_init_folder(QDir::temp().absolutePath())); //Fall back to %LOCALAPPDATA%, if %TEMP% didn't work if(g_x264_temp_folder.path->isEmpty()) { qWarning("%%TEMP%% directory not found -> falling back to %%LOCALAPPDATA%%"); static const x264_known_folder_t folderId[2] = { x264_folder_localappdata, x264_folder_systroot_dir }; for(size_t id = 0; (g_x264_temp_folder.path->isEmpty() && (id < 2)); id++) { const QString &localAppData = x264_known_folder(x264_folder_localappdata); if(!localAppData.isEmpty()) { *g_x264_temp_folder.path = x264_try_init_folder(QString("%1/Temp").arg(localAppData)); } else { qWarning("%%LOCALAPPDATA%% directory could not be found!"); } } } //Failed to init TEMP folder? if(g_x264_temp_folder.path->isEmpty()) { qWarning("Temporary directory could not be initialized !!!"); } } return *g_x264_temp_folder.path; } /* * Display the window's close button */ bool x264_enable_close_button(const QWidget *win, const bool bEnable) { bool ok = false; if(HMENU hMenu = GetSystemMenu(win->winId(), FALSE)) { ok = (EnableMenuItem(hMenu, SC_CLOSE, MF_BYCOMMAND | (bEnable ? MF_ENABLED : MF_GRAYED)) == TRUE); } return ok; } /* * Play beep sound */ bool x264_beep(int beepType) { switch(beepType) { case x264_beep_info: return MessageBeep(MB_ICONASTERISK) == TRUE; break; case x264_beep_warning: return MessageBeep(MB_ICONEXCLAMATION) == TRUE; break; case x264_beep_error: return MessageBeep(MB_ICONHAND) == TRUE; break; default: return false; } } /* * Shutdown the computer */ bool x264_shutdown_computer(const QString &message, const unsigned long timeout, const bool forceShutdown) { HANDLE hToken = NULL; if(OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) { TOKEN_PRIVILEGES privileges; memset(&privileges, 0, sizeof(TOKEN_PRIVILEGES)); privileges.PrivilegeCount = 1; privileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED; if(LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &privileges.Privileges[0].Luid)) { if(AdjustTokenPrivileges(hToken, FALSE, &privileges, NULL, NULL, NULL)) { const DWORD reason = SHTDN_REASON_MAJOR_APPLICATION | SHTDN_REASON_FLAG_PLANNED; return InitiateSystemShutdownEx(NULL, const_cast(QWCHAR(message)), timeout, forceShutdown ? TRUE : FALSE, FALSE, reason); } } } return false; } /* * Check the network connection status */ int x264_network_status(void) { DWORD dwFlags; const BOOL ret = IsNetworkAlive(&dwFlags); if(GetLastError() == 0) { return (ret != FALSE) ? x264_network_yes : x264_network_non; } return x264_network_err; } /* * Setup QPorcess object */ void x264_init_process(QProcess &process, const QString &wokringDir, const bool bReplaceTempDir) { //Environment variable names static const char *const s_envvar_names_temp[] = { "TEMP", "TMP", "TMPDIR", "HOME", "USERPROFILE", "HOMEPATH", NULL }; static const char *const s_envvar_names_remove[] = { "WGETRC", "SYSTEM_WGETRC", "HTTP_PROXY", "FTP_PROXY", "NO_PROXY", "GNUPGHOME", "LC_ALL", "LC_COLLATE", "LC_CTYPE", "LC_MESSAGES", "LC_MONETARY", "LC_NUMERIC", "LC_TIME", "LANG", NULL }; //Initialize environment QProcessEnvironment env = process.processEnvironment(); if(env.isEmpty()) env = QProcessEnvironment::systemEnvironment(); //Clean a number of enviroment variables that might affect our tools for(size_t i = 0; s_envvar_names_remove[i]; i++) { env.remove(QString::fromLatin1(s_envvar_names_remove[i])); env.remove(QString::fromLatin1(s_envvar_names_remove[i]).toLower()); } const QString tempDir = QDir::toNativeSeparators(x264_temp_directory()); //Replace TEMP directory in environment if(bReplaceTempDir) { for(size_t i = 0; s_envvar_names_temp[i]; i++) { env.insert(s_envvar_names_temp[i], tempDir); } } //Setup PATH variable const QString path = env.value("PATH", QString()).trimmed(); env.insert("PATH", path.isEmpty() ? tempDir : QString("%1;%2").arg(tempDir, path)); //Setup QPorcess object process.setWorkingDirectory(wokringDir); process.setProcessChannelMode(QProcess::MergedChannels); process.setReadChannel(QProcess::StandardOutput); process.setProcessEnvironment(env); } /* * Inform the system that it is in use, thereby preventing the system from entering sleep */ bool x264_set_thread_execution_state(const bool systemRequired) { EXECUTION_STATE state = NULL; if(systemRequired) { state = SetThreadExecutionState(ES_CONTINUOUS | ES_SYSTEM_REQUIRED | ES_AWAYMODE_REQUIRED); } else { state = SetThreadExecutionState(ES_CONTINUOUS); } return (state != NULL); } /* * Exception class */ X264Exception::X264Exception(const char *message, ...) : runtime_error(message) { va_list args; va_start(args, message); vsnprintf_s(m_message, MAX_MSGLEN, _TRUNCATE, message, args); va_end(args); } /* * Check for debugger (detect routine) */ static __forceinline bool x264_check_for_debugger(void) { __try { CloseHandle((HANDLE)((DWORD_PTR)-3)); } __except(1) { return true; } __try { __debugbreak(); } __except(1) { return IsDebuggerPresent(); } return true; } /* * Check for debugger (thread proc) */ static unsigned int __stdcall x264_debug_thread_proc(LPVOID lpParameter) { SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_LOWEST); forever { if(x264_check_for_debugger()) { x264_fatal_exit(L"Not a debug build. Please unload debugger and try again!"); return 666; } x264_sleep(100); } } /* * Check for debugger (startup routine) */ static HANDLE x264_debug_thread_init() { if(x264_check_for_debugger()) { x264_fatal_exit(L"Not a debug build. Please unload debugger and try again!"); } const uintptr_t h = _beginthreadex(NULL, 0, x264_debug_thread_proc, NULL, 0, NULL); return (HANDLE)(h^0xdeadbeef); } /* * Fatal application exit */ #pragma intrinsic(_InterlockedExchange) void x264_fatal_exit(const wchar_t* exitMessage, const char* errorBoxMessage) { static volatile long bFatalFlag = 0L; if(_InterlockedExchange(&bFatalFlag, 1L) == 0L) { if(GetCurrentThreadId() != g_main_thread_id) { HANDLE mainThread = OpenThread(THREAD_TERMINATE, FALSE, g_main_thread_id); if(mainThread) TerminateThread(mainThread, ULONG_MAX); } if(errorBoxMessage) { MessageBoxA(NULL, errorBoxMessage, "Simple x264 Launcher - GURU MEDITATION", MB_ICONERROR | MB_TOPMOST | MB_TASKMODAL); } FatalAppExit(0, exitMessage); for(;;) { TerminateProcess(GetCurrentProcess(), -1); } } } /* * Entry point checks */ static DWORD x264_entry_check(void); static DWORD g_x264_entry_check_result = x264_entry_check(); static DWORD g_x264_entry_check_flag = 0x789E09B2; static DWORD x264_entry_check(void) { volatile DWORD retVal = 0xA199B5AF; if(g_x264_entry_check_flag != 0x8761F64D) { x264_fatal_exit(L"Application initialization has failed, take care!"); } return retVal; } /* * Application entry point (runs before static initializers) */ extern "C" { int WinMainCRTStartup(void); int x264_entry_point(void) { if((!X264_DEBUG) && x264_check_for_debugger()) { x264_fatal_exit(L"Not a debug build. Please unload debugger and try again!"); } if(g_x264_entry_check_flag != 0x789E09B2) { x264_fatal_exit(L"Application initialization has failed, take care!"); } //Zero *before* constructors are called X264_ZERO_MEMORY(g_x264_argv); X264_ZERO_MEMORY(g_x264_os_version); X264_ZERO_MEMORY(g_x264_portable); X264_ZERO_MEMORY(g_x264_known_folder); X264_ZERO_MEMORY(g_x264_temp_folder); //Make sure we will pass the check g_x264_entry_check_flag = ~g_x264_entry_check_flag; //Now initialize the C Runtime library! return WinMainCRTStartup(); } } /* * Initialize debug thread */ static const HANDLE g_debug_thread = X264_DEBUG ? NULL : x264_debug_thread_init(); /* * Get number private bytes [debug only] */ size_t x264_dbg_private_bytes(void) { #if X264_DEBUG PROCESS_MEMORY_COUNTERS_EX memoryCounters; memoryCounters.cb = sizeof(PROCESS_MEMORY_COUNTERS_EX); GetProcessMemoryInfo(GetCurrentProcess(), (PPROCESS_MEMORY_COUNTERS) &memoryCounters, sizeof(PROCESS_MEMORY_COUNTERS_EX)); return memoryCounters.PrivateUsage; #else THROW("Cannot call this function in a non-debug build!"); #endif //X264_DEBUG } /* * Finalization function */ void x264_finalization(void) { //Destroy Qt application object QApplication *application = dynamic_cast(QApplication::instance()); X264_DELETE(application); //Free STDOUT and STDERR buffers if(g_x264_console_attached) { if(std::filebuf *tmp = dynamic_cast(std::cout.rdbuf())) { std::cout.rdbuf(NULL); X264_DELETE(tmp); } if(std::filebuf *tmp = dynamic_cast(std::cerr.rdbuf())) { std::cerr.rdbuf(NULL); X264_DELETE(tmp); } } //Clear CLI args X264_DELETE(g_x264_argv.list); //Clear folders cache X264_DELETE(g_x264_known_folder.knownFolders); X264_DELETE(g_x264_temp_folder.path); }