2010-11-06 23:04:47 +01:00
///////////////////////////////////////////////////////////////////////////////
// LameXP - Audio Encoder Front-End
2011-01-01 17:04:25 +01:00
// Copyright (C) 2004-2011 LoRd_MuldeR <MuldeR2@GMX.de>
2010-11-06 23:04:47 +01:00
//
// 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
///////////////////////////////////////////////////////////////////////////////
# include "Global.h"
//Qt includes
# include <QApplication>
# include <QMessageBox>
# include <QDir>
# include <QUuid>
# include <QMap>
# include <QDate>
# include <QIcon>
# include <QPlastiqueStyle>
# include <QImageReader>
# include <QSharedMemory>
# include <QSysInfo>
# include <QStringList>
2010-11-08 19:29:36 +01:00
# include <QSystemSemaphore>
2010-11-19 13:31:45 +01:00
# include <QMutex>
2010-11-22 21:45:00 +01:00
# include <QTextCodec>
2010-11-25 16:56:32 +01:00
# include <QLibrary>
2010-12-07 22:58:28 +01:00
# include <QRegExp>
2010-12-28 21:26:16 +01:00
# include <QResource>
# include <QTranslator>
2010-11-06 23:04:47 +01:00
//LameXP includes
2010-11-07 23:06:30 +01:00
# include "Resource.h"
2010-11-06 23:04:47 +01:00
# include "LockedFile.h"
2010-11-07 23:06:30 +01:00
//CRT includes
# include <io.h>
# include <fcntl.h>
2010-11-15 04:42:06 +01:00
# include <intrin.h>
2010-11-07 23:06:30 +01:00
2011-04-11 21:55:34 +02:00
//COM includes
# include <Objbase.h>
2010-11-07 16:32:54 +01:00
//Debug only includes
# ifdef _DEBUG
# include <Psapi.h>
# endif //_DEBUG
2010-11-06 23:04:47 +01:00
2010-11-11 19:37:16 +01:00
//Static build only macros
# ifdef QT_NODLL
# pragma warning(disable:4101)
# define LAMEXP_INIT_QT_STATIC_PLUGIN(X) Q_IMPORT_PLUGIN(X)
# else
# define LAMEXP_INIT_QT_STATIC_PLUGIN(X)
# endif
2010-11-08 19:29:36 +01:00
///////////////////////////////////////////////////////////////////////////////
// TYPES
///////////////////////////////////////////////////////////////////////////////
typedef struct
{
unsigned int command ;
unsigned int reserved_1 ;
unsigned int reserved_2 ;
char parameter [ 4096 ] ;
} lamexp_ipc_t ;
2010-11-06 23:04:47 +01:00
///////////////////////////////////////////////////////////////////////////////
// GLOBAL VARS
///////////////////////////////////////////////////////////////////////////////
//Build version
2011-01-20 22:54:10 +01:00
static const struct
{
unsigned int ver_major ;
unsigned int ver_minor ;
unsigned int ver_build ;
char * ver_release_name ;
}
g_lamexp_version =
{
VER_LAMEXP_MAJOR ,
VER_LAMEXP_MINOR ,
VER_LAMEXP_BUILD ,
VER_LAMEXP_RNAME
} ;
2010-11-06 23:04:47 +01:00
//Build date
static QDate g_lamexp_version_date ;
static const char * g_lamexp_months [ ] = { " Jan " , " Feb " , " Mar " , " Apr " , " May " , " Jun " , " Jul " , " Aug " , " Sep " , " Oct " , " Nov " , " Dec " } ;
static const char * g_lamexp_version_raw_date = __DATE__ ;
2011-03-04 23:40:09 +01:00
//Console attached flag
static bool g_lamexp_console_attached = false ;
2011-04-07 22:40:47 +02:00
//Compiler detection
//The following code was borrowed from MPC-HC project: http://mpc-hc.sf.net/
# if defined(__INTEL_COMPILER)
# if (__INTEL_COMPILER >= 1200)
static const char * g_lamexp_version_compiler = " ICL 12.x " ;
# elif (__INTEL_COMPILER >= 1100)
static const char * g_lamexp_version_compiler = = " ICL 11.x " ;
# elif (__INTEL_COMPILER >= 1000)
static const char * g_lamexp_version_compiler = = " ICL 10.x " ;
2010-11-20 22:14:29 +01:00
# else
2011-04-07 22:40:47 +02:00
# error Compiler is not supported!
# endif
# elif defined(_MSC_VER)
# if (_MSC_VER == 1600)
# if (_MSC_FULL_VER >= 160040219)
static const char * g_lamexp_version_compiler = " MSVC 2010-SP1 " ;
2010-11-20 22:14:29 +01:00
# else
2011-04-07 22:40:47 +02:00
static const char * g_lamexp_version_compiler = " MSVC 2010 " ;
2010-11-20 22:14:29 +01:00
# endif
2011-04-07 22:40:47 +02:00
# elif (_MSC_VER == 1500)
# if (_MSC_FULL_VER >= 150030729)
static const char * g_lamexp_version_compiler = " MSVC 2008-SP1 " ;
# else
static const char * g_lamexp_version_compiler = " MSVC 2008 " ;
# endif
# else
# error Compiler is not supported!
2010-11-20 22:14:29 +01:00
# endif
2011-04-07 22:40:47 +02:00
// 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)
LAMEXP_COMPILER_WARNING ( " SSE instruction set is enabled! " )
# elif (_M_IX86_FP == 2)
LAMEXP_COMPILER_WARNING ( " SSE2 instruction set is enabled! " )
# endif
# endif
# else
# error Compiler is not supported!
2010-11-20 22:14:29 +01:00
# endif
2011-02-28 01:28:27 +01:00
//Official web-site URL
static const char * g_lamexp_website_url = " http://mulder.dummwiedeutsch.de/ " ;
2011-03-20 14:28:27 +01:00
static const char * g_lamexp_support_url = " http://forum.doom9.org/showthread.php?t=157726 " ;
2011-02-28 01:28:27 +01:00
2010-11-12 15:58:53 +01:00
//Tool versions (expected)
static const unsigned int g_lamexp_toolver_neroaac = VER_LAMEXP_TOOL_NEROAAC ;
2010-11-06 23:04:47 +01:00
//Special folders
static QString g_lamexp_temp_folder ;
//Tools
static QMap < QString , LockedFile * > g_lamexp_tool_registry ;
2010-11-12 15:58:53 +01:00
static QMap < QString , unsigned int > g_lamexp_tool_versions ;
2010-11-06 23:04:47 +01:00
2010-12-28 21:26:16 +01:00
//Languages
2011-01-21 00:10:51 +01:00
static struct
{
QMap < QString , QString > files ;
QMap < QString , QString > names ;
QMap < QString , unsigned int > sysid ;
}
g_lamexp_translation ;
//Translator
2010-12-28 21:26:16 +01:00
static QTranslator * g_lamexp_currentTranslator = NULL ;
2010-11-06 23:04:47 +01:00
//Shared memory
2011-01-20 22:54:10 +01:00
static const struct
{
char * sharedmem ;
char * semaphore_read ;
char * semaphore_write ;
}
g_lamexp_ipc_uuid =
{
" {21A68A42-6923-43bb-9CF6-64BF151942EE} " ,
" {7A605549-F58C-4d78-B4E5-06EFC34F405B} " ,
" {60AA8D04-F6B8-497d-81EB-0F600F4A65B5} "
} ;
static struct
{
QSharedMemory * sharedmem ;
QSystemSemaphore * semaphore_read ;
QSystemSemaphore * semaphore_write ;
}
g_lamexp_ipc_ptr =
{
NULL , NULL , NULL
} ;
2010-11-06 23:04:47 +01:00
2010-11-11 19:37:16 +01:00
//Image formats
2011-03-21 02:16:18 +01:00
static const char * g_lamexp_imageformats [ ] = { " png " , " jpg " , " gif " , " ico " , " svg " , NULL } ;
2010-11-11 19:37:16 +01:00
2010-11-19 13:31:45 +01:00
//Global locks
static QMutex g_lamexp_message_mutex ;
2010-11-06 23:04:47 +01:00
///////////////////////////////////////////////////////////////////////////////
// GLOBAL FUNCTIONS
///////////////////////////////////////////////////////////////////////////////
/*
* Version getters
*/
2011-01-20 22:54:10 +01:00
unsigned int lamexp_version_major ( void ) { return g_lamexp_version . ver_major ; }
unsigned int lamexp_version_minor ( void ) { return g_lamexp_version . ver_minor ; }
unsigned int lamexp_version_build ( void ) { return g_lamexp_version . ver_build ; }
const char * lamexp_version_release ( void ) { return g_lamexp_version . ver_release_name ; }
2010-11-20 22:14:29 +01:00
const char * lamexp_version_compiler ( void ) { return g_lamexp_version_compiler ; }
2010-11-12 15:58:53 +01:00
unsigned int lamexp_toolver_neroaac ( void ) { return g_lamexp_toolver_neroaac ; }
2011-03-20 14:28:27 +01:00
/*
* URL getters
*/
2011-02-28 01:28:27 +01:00
const char * lamexp_website_url ( void ) { return g_lamexp_website_url ; }
2011-03-20 14:28:27 +01:00
const char * lamexp_support_url ( void ) { return g_lamexp_support_url ; }
2010-11-07 23:06:30 +01:00
2011-03-20 14:28:27 +01:00
/*
* Check for Demo ( pre - release ) version
*/
2010-11-07 23:06:30 +01:00
bool lamexp_version_demo ( void )
2011-02-12 22:17:42 +01:00
{
2011-04-16 21:21:08 +02:00
char buffer [ 128 ] ;
bool releaseVersion = false ;
if ( ! strcpy_s ( buffer , 128 , g_lamexp_version . ver_release_name ) )
{
char * context , * prefix = strtok_s ( buffer , " -,; " , & context ) ;
if ( prefix )
{
releaseVersion = ( ! _stricmp ( prefix , " Final " ) ) | | ( ! _stricmp ( prefix , " Hotfix " ) ) ;
}
}
return LAMEXP_DEBUG | | ( ! releaseVersion ) ;
2010-11-07 23:06:30 +01:00
}
2010-11-06 23:04:47 +01:00
2011-03-20 14:28:27 +01:00
/*
* Calculate expiration date
*/
2011-01-20 22:06:59 +01:00
QDate lamexp_version_expires ( void )
{
2011-03-12 21:58:46 +01:00
return lamexp_version_date ( ) . addDays ( LAMEXP_DEBUG ? 2 : 30 ) ;
2011-01-20 22:06:59 +01:00
}
2010-11-06 23:04:47 +01:00
/*
* Get build date date
*/
const QDate & lamexp_version_date ( void )
{
if ( ! g_lamexp_version_date . isValid ( ) )
{
char temp [ 32 ] ;
int date [ 3 ] ;
char * this_token = NULL ;
char * next_token = NULL ;
strcpy_s ( temp , 32 , g_lamexp_version_raw_date ) ;
this_token = strtok_s ( temp , " " , & next_token ) ;
for ( int i = 0 ; i < 3 ; i + + )
{
date [ i ] = - 1 ;
if ( this_token )
{
for ( int j = 0 ; j < 12 ; j + + )
{
if ( ! _strcmpi ( this_token , g_lamexp_months [ j ] ) )
{
date [ i ] = j + 1 ;
break ;
}
}
if ( date [ i ] < 0 )
{
date [ i ] = atoi ( this_token ) ;
}
this_token = strtok_s ( NULL , " " , & next_token ) ;
}
}
if ( date [ 0 ] > = 0 & & date [ 1 ] > = 0 & & date [ 2 ] > = 0 )
{
g_lamexp_version_date = QDate ( date [ 2 ] , date [ 0 ] , date [ 1 ] ) ;
}
}
return g_lamexp_version_date ;
}
2010-11-19 13:31:45 +01:00
/*
* Qt message handler
*/
void lamexp_message_handler ( QtMsgType type , const char * msg )
{
static HANDLE hConsole = NULL ;
2011-03-12 01:50:43 +01:00
static const char * GURU_MEDITATION = " \n \n GURU MEDITATION !!! \n \n " ;
2011-03-10 02:07:51 +01:00
const char * buffer = NULL , * text = msg ;
char temp [ 1024 ] ;
2011-03-12 01:50:43 +01:00
QMutexLocker lock ( & g_lamexp_message_mutex ) ;
2011-03-10 02:07:51 +01:00
if ( ! strncmp ( msg , " @BASE64@ " , 8 ) )
{
buffer = _strdup ( QByteArray : : fromBase64 ( msg + 8 ) . constData ( ) ) ;
if ( buffer ) text = buffer ;
}
2010-11-19 13:31:45 +01:00
if ( ! hConsole )
{
2011-03-04 23:40:09 +01:00
if ( g_lamexp_console_attached )
{
hConsole = CreateFile ( L " CONOUT$ " , GENERIC_WRITE , FILE_SHARE_WRITE | FILE_SHARE_READ , NULL , OPEN_EXISTING , NULL , NULL ) ;
if ( hConsole = = INVALID_HANDLE_VALUE ) hConsole = NULL ;
}
2010-11-19 13:31:45 +01:00
}
2011-03-04 23:40:09 +01:00
if ( hConsole )
2010-11-19 13:31:45 +01:00
{
2011-03-10 02:07:51 +01:00
int len = sprintf_s ( temp , 1024 , " %s \n " , text ) ;
2011-03-04 23:40:09 +01:00
CONSOLE_SCREEN_BUFFER_INFO bufferInfo ;
GetConsoleScreenBufferInfo ( hConsole , & bufferInfo ) ;
SetConsoleOutputCP ( CP_UTF8 ) ;
switch ( type )
{
case QtCriticalMsg :
case QtFatalMsg :
fflush ( stdout ) ;
fflush ( stderr ) ;
SetConsoleTextAttribute ( hConsole , FOREGROUND_RED | FOREGROUND_INTENSITY ) ;
2011-03-12 01:50:43 +01:00
WriteFile ( hConsole , GURU_MEDITATION , strlen ( GURU_MEDITATION ) , NULL , NULL ) ;
2011-03-10 02:07:51 +01:00
WriteFile ( hConsole , temp , len , NULL , NULL ) ;
FlushFileBuffers ( hConsole ) ;
2011-03-04 23:40:09 +01:00
break ;
case QtWarningMsg :
SetConsoleTextAttribute ( hConsole , FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY ) ;
2011-03-10 02:07:51 +01:00
WriteFile ( hConsole , temp , len , NULL , NULL ) ;
FlushFileBuffers ( hConsole ) ;
2011-03-04 23:40:09 +01:00
break ;
default :
SetConsoleTextAttribute ( hConsole , FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY ) ;
2011-03-10 02:07:51 +01:00
WriteFile ( hConsole , temp , len , NULL , NULL ) ;
FlushFileBuffers ( hConsole ) ;
2011-03-04 23:40:09 +01:00
break ;
}
SetConsoleTextAttribute ( hConsole , FOREGROUND_BLUE | FOREGROUND_GREEN | FOREGROUND_RED ) ;
2010-11-19 13:31:45 +01:00
}
2011-03-04 23:40:09 +01:00
else
{
switch ( type )
{
case QtCriticalMsg :
case QtFatalMsg :
2011-03-10 02:07:51 +01:00
sprintf_s ( temp , 1024 , " [LameXP][C] %s " , text ) ;
2011-03-04 23:40:09 +01:00
break ;
case QtWarningMsg :
2011-03-10 02:07:51 +01:00
sprintf_s ( temp , 1024 , " [LameXP][W] %s " , text ) ;
2011-03-04 23:40:09 +01:00
break ;
default :
2011-03-10 02:07:51 +01:00
sprintf_s ( temp , 1024 , " [LameXP][I] %s " , text ) ;
2011-03-04 23:40:09 +01:00
break ;
}
2011-03-10 02:07:51 +01:00
while ( char * ptr = strchr ( temp , ' \n ' ) ) * ptr = ' \t ' ;
strcat_s ( temp , 1024 , " \n " ) ;
OutputDebugStringA ( temp ) ;
2011-03-04 23:40:09 +01:00
}
2010-11-19 13:31:45 +01:00
if ( type = = QtCriticalMsg | | type = = QtFatalMsg )
{
2010-11-20 03:11:12 +01:00
lock . unlock ( ) ;
2011-03-10 02:07:51 +01:00
MessageBoxW ( NULL , QWCHAR ( QString : : fromUtf8 ( text ) ) , L " LameXP - GURU MEDITATION " , MB_ICONERROR | MB_TOPMOST | MB_TASKMODAL ) ;
2010-11-19 13:31:45 +01:00
FatalAppExit ( 0 , L " The application has encountered a critical error and will exit now! " ) ;
TerminateProcess ( GetCurrentProcess ( ) , - 1 ) ;
}
2011-03-10 02:07:51 +01:00
if ( buffer ) free ( ( void * ) buffer ) ;
}
2010-11-19 13:31:45 +01:00
2010-11-07 23:06:30 +01:00
/*
* Initialize the console
*/
void lamexp_init_console ( int argc , char * argv [ ] )
{
2011-02-14 19:03:36 +01:00
bool enableConsole = lamexp_version_demo ( ) ;
2011-03-12 21:58:46 +01:00
2011-03-25 16:17:19 +01:00
if ( ! LAMEXP_DEBUG )
2010-11-07 23:06:30 +01:00
{
2011-03-25 16:17:19 +01:00
for ( int i = 0 ; i < argc ; i + + )
2010-11-07 23:06:30 +01:00
{
2011-03-25 16:17:19 +01:00
if ( ! _stricmp ( argv [ i ] , " --console " ) )
{
enableConsole = true ;
}
else if ( ! _stricmp ( argv [ i ] , " --no-console " ) )
{
enableConsole = false ;
}
2011-02-14 19:03:36 +01:00
}
}
2010-11-07 23:06:30 +01:00
2011-02-14 19:03:36 +01:00
if ( enableConsole )
{
2011-03-04 23:40:09 +01:00
if ( ! g_lamexp_console_attached )
{
g_lamexp_console_attached = AllocConsole ( ) ;
}
if ( g_lamexp_console_attached )
2011-02-14 19:03:36 +01:00
{
//-------------------------------------------------------------------
//See: http://support.microsoft.com/default.aspx?scid=kb;en-us;105305
//-------------------------------------------------------------------
int hCrtStdOut = _open_osfhandle ( ( long ) GetStdHandle ( STD_OUTPUT_HANDLE ) , _O_TEXT ) ;
int hCrtStdErr = _open_osfhandle ( ( long ) GetStdHandle ( STD_ERROR_HANDLE ) , _O_TEXT ) ;
FILE * hfStdOut = _fdopen ( hCrtStdOut , " w " ) ;
FILE * hfStderr = _fdopen ( hCrtStdErr , " w " ) ;
* stdout = * hfStdOut ;
* stderr = * hfStderr ;
setvbuf ( stdout , NULL , _IONBF , 0 ) ;
setvbuf ( stderr , NULL , _IONBF , 0 ) ;
2010-11-07 23:06:30 +01:00
}
2011-02-14 19:03:36 +01:00
2011-03-04 23:40:09 +01:00
if ( HWND hwndConsole = GetConsoleWindow ( ) )
{
HMENU hMenu = GetSystemMenu ( hwndConsole , 0 ) ;
EnableMenuItem ( hMenu , SC_CLOSE , MF_BYCOMMAND | MF_GRAYED ) ;
RemoveMenu ( hMenu , SC_CLOSE , MF_BYCOMMAND ) ;
SetWindowLong ( hwndConsole , GWL_STYLE , GetWindowLong ( hwndConsole , GWL_STYLE ) & ( ~ WS_MAXIMIZEBOX ) ) ;
SetWindowLong ( hwndConsole , GWL_STYLE , GetWindowLong ( hwndConsole , GWL_STYLE ) & ( ~ WS_MINIMIZEBOX ) ) ;
SetConsoleCtrlHandler ( NULL , TRUE ) ;
SetConsoleTitle ( L " LameXP - Audio Encoder Front-End | Debug Console " ) ;
SetConsoleOutputCP ( CP_UTF8 ) ;
}
2010-11-07 23:06:30 +01:00
}
}
2010-11-15 04:42:06 +01:00
/*
* Detect CPU features
*/
lamexp_cpu_t lamexp_detect_cpu_features ( void )
{
2010-11-25 16:56:32 +01:00
typedef BOOL ( WINAPI * IsWow64ProcessFun ) ( __in HANDLE hProcess , __out PBOOL Wow64Process ) ;
typedef VOID ( WINAPI * GetNativeSystemInfoFun ) ( __out LPSYSTEM_INFO lpSystemInfo ) ;
static IsWow64ProcessFun IsWow64ProcessPtr = NULL ;
static GetNativeSystemInfoFun GetNativeSystemInfoPtr = NULL ;
2010-11-15 04:42:06 +01:00
2010-11-25 16:56:32 +01:00
lamexp_cpu_t features ;
SYSTEM_INFO systemInfo ;
2010-11-15 04:42:06 +01:00
int CPUInfo [ 4 ] = { - 1 } ;
2010-11-26 00:29:53 +01:00
char CPUIdentificationString [ 0x40 ] ;
2010-11-25 16:56:32 +01:00
char CPUBrandString [ 0x40 ] ;
2011-02-07 22:01:06 +01:00
2010-11-25 16:56:32 +01:00
memset ( & features , 0 , sizeof ( lamexp_cpu_t ) ) ;
memset ( & systemInfo , 0 , sizeof ( SYSTEM_INFO ) ) ;
2010-11-26 00:29:53 +01:00
memset ( CPUIdentificationString , 0 , sizeof ( CPUIdentificationString ) ) ;
2010-11-25 16:56:32 +01:00
memset ( CPUBrandString , 0 , sizeof ( CPUBrandString ) ) ;
2010-11-15 04:42:06 +01:00
__cpuid ( CPUInfo , 0 ) ;
2010-11-26 00:29:53 +01:00
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 ) ;
strcpy_s ( features . vendor , 0x40 , CPUIdentificationString ) ;
2010-11-15 04:42:06 +01:00
if ( CPUInfo [ 0 ] > = 1 )
{
__cpuid ( CPUInfo , 1 ) ;
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 ) ;
}
__cpuid ( CPUInfo , 0x80000000 ) ;
2011-02-07 22:01:06 +01:00
int nExIds = max ( min ( CPUInfo [ 0 ] , 0x80000004 ) , 0x80000000 ) ;
2010-11-15 04:42:06 +01:00
2011-02-07 22:01:06 +01:00
for ( int i = 0x80000002 ; i < = nExIds ; + + i )
2010-11-15 04:42:06 +01:00
{
__cpuid ( CPUInfo , i ) ;
2011-02-07 22:01:06 +01:00
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 ;
}
2010-11-15 04:42:06 +01:00
}
strcpy_s ( features . brand , 0x40 , CPUBrandString ) ;
2010-11-15 14:02:58 +01:00
2011-02-07 22:01:06 +01:00
if ( strlen ( features . brand ) < 1 ) strcpy_s ( features . brand , 0x40 , " Unknown " ) ;
if ( strlen ( features . vendor ) < 1 ) strcpy_s ( features . vendor , 0x40 , " Unknown " ) ;
2010-11-25 16:56:32 +01:00
# if !defined(_M_X64 ) && !defined(_M_IA64)
if ( ! IsWow64ProcessPtr | | ! GetNativeSystemInfoPtr )
2010-11-15 14:02:58 +01:00
{
2010-11-25 16:56:32 +01:00
QLibrary Kernel32Lib ( " kernel32.dll " ) ;
IsWow64ProcessPtr = ( IsWow64ProcessFun ) Kernel32Lib . resolve ( " IsWow64Process " ) ;
GetNativeSystemInfoPtr = ( GetNativeSystemInfoFun ) Kernel32Lib . resolve ( " GetNativeSystemInfo " ) ;
2010-11-15 14:02:58 +01:00
}
2010-11-25 16:56:32 +01:00
if ( IsWow64ProcessPtr )
{
BOOL x64 = FALSE ;
if ( IsWow64ProcessPtr ( GetCurrentProcess ( ) , & x64 ) )
{
features . x64 = x64 ;
}
}
if ( GetNativeSystemInfoPtr )
{
GetNativeSystemInfoPtr ( & systemInfo ) ;
}
else
{
GetSystemInfo ( & systemInfo ) ;
}
features . count = systemInfo . dwNumberOfProcessors ;
# else
2010-11-15 14:02:58 +01:00
GetNativeSystemInfo ( & systemInfo ) ;
features . count = systemInfo . dwNumberOfProcessors ;
2010-11-25 16:56:32 +01:00
features . x64 = true ;
# endif
2010-11-15 14:02:58 +01:00
2010-11-15 04:42:06 +01:00
return features ;
}
/*
* Check for debugger
*/
2011-02-18 00:53:36 +01:00
# if !defined(_DEBUG) || defined(NDEBUG)
void WINAPI lamexp_debug_thread_proc ( __in LPVOID lpParameter )
2010-11-15 04:42:06 +01:00
{
2011-02-18 00:53:36 +01:00
while ( ! IsDebuggerPresent ( ) )
2010-11-15 04:42:06 +01:00
{
Sleep ( 333 ) ;
}
TerminateProcess ( GetCurrentProcess ( ) , - 1 ) ;
}
2011-02-18 00:53:36 +01:00
HANDLE lamexp_debug_thread_init ( void )
{
if ( IsDebuggerPresent ( ) )
{
FatalAppExit ( 0 , L " Not a debug build. Please unload debugger and try again! " ) ;
TerminateProcess ( GetCurrentProcess ( ) , - 1 ) ;
}
return CreateThread ( NULL , NULL , reinterpret_cast < LPTHREAD_START_ROUTINE > ( & lamexp_debug_thread_proc ) , NULL , NULL , NULL ) ;
}
static const HANDLE g_debug_thread = lamexp_debug_thread_init ( ) ;
# endif
2010-11-15 04:42:06 +01:00
2010-12-23 03:13:11 +01:00
/*
* Check for compatibility mode
*/
2011-03-08 01:12:40 +01:00
static bool lamexp_check_compatibility_mode ( const char * exportName , const char * executableName )
2010-12-23 03:13:11 +01:00
{
QLibrary kernel32 ( " kernel32.dll " ) ;
2010-12-27 01:50:22 +01:00
if ( exportName ! = NULL )
{
if ( kernel32 . resolve ( exportName ) ! = NULL )
{
2011-03-08 01:12:40 +01:00
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 ( ) ) ;
2010-12-27 01:50:22 +01:00
return false ;
}
}
return true ;
}
/*
* Check for process elevation
*/
static bool lamexp_check_elevation ( void )
{
2011-01-03 22:24:58 +01:00
typedef enum { lamexp_token_elevationType_class = 18 , lamexp_token_elevation_class = 20 } LAMEXP_TOKEN_INFORMATION_CLASS ;
typedef enum { lamexp_elevationType_default = 1 , lamexp_elevationType_full , lamexp_elevationType_limited } LAMEXP_TOKEN_ELEVATION_TYPE ;
2010-12-27 01:50:22 +01:00
HANDLE hToken = NULL ;
bool bIsProcessElevated = false ;
2010-12-23 03:13:11 +01:00
2010-12-27 01:50:22 +01:00
if ( OpenProcessToken ( GetCurrentProcess ( ) , TOKEN_QUERY , & hToken ) )
2010-12-23 03:13:11 +01:00
{
2011-01-03 22:24:58 +01:00
LAMEXP_TOKEN_ELEVATION_TYPE tokenElevationType ;
2010-12-27 01:50:22 +01:00
DWORD returnLength ;
2011-01-03 22:24:58 +01:00
if ( GetTokenInformation ( hToken , ( TOKEN_INFORMATION_CLASS ) lamexp_token_elevationType_class , & tokenElevationType , sizeof ( LAMEXP_TOKEN_ELEVATION_TYPE ) , & returnLength ) )
2010-12-23 03:13:11 +01:00
{
2011-01-03 22:24:58 +01:00
if ( returnLength = = sizeof ( LAMEXP_TOKEN_ELEVATION_TYPE ) )
2010-12-23 03:13:11 +01:00
{
2011-01-03 22:24:58 +01:00
switch ( tokenElevationType )
{
case lamexp_elevationType_default :
qDebug ( " Process token elevation type: Default -> UAC is disabled. \n " ) ;
break ;
case lamexp_elevationType_full :
qWarning ( " Process token elevation type: Full -> potential security risk! \n " ) ;
bIsProcessElevated = true ;
break ;
case lamexp_elevationType_limited :
qDebug ( " Process token elevation type: Limited -> not elevated. \n " ) ;
break ;
}
2010-12-23 03:13:11 +01:00
}
}
2010-12-27 01:50:22 +01:00
CloseHandle ( hToken ) ;
}
else
{
qWarning ( " Failed to open process token! " ) ;
2010-12-23 03:13:11 +01:00
}
2010-12-27 01:50:22 +01:00
return ! bIsProcessElevated ;
2010-12-23 03:13:11 +01:00
}
2010-11-15 04:42:06 +01:00
2010-11-06 23:04:47 +01:00
/*
* Initialize Qt framework
*/
bool lamexp_init_qt ( int argc , char * argv [ ] )
{
static bool qt_initialized = false ;
//Don't initialized again, if done already
if ( qt_initialized )
{
return true ;
}
2011-03-08 01:12:40 +01:00
//Extract executable name from argv[] array
char * executableName = argv [ 0 ] ;
while ( char * temp = strpbrk ( executableName , " \\ /:? " ) )
{
executableName = temp + 1 ;
}
2010-11-06 23:04:47 +01:00
//Check Qt version
2011-03-08 01:12:40 +01:00
qDebug ( " Using Qt Framework v%s, compiled with Qt v%s [%s] " , qVersion ( ) , QT_VERSION_STR , QT_PACKAGEDATE_STR ) ;
if ( _stricmp ( qVersion ( ) , QT_VERSION_STR ) )
{
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 ( ) ) ;
return false ;
}
2010-11-06 23:04:47 +01:00
//Check the Windows version
2010-11-11 19:37:16 +01:00
switch ( QSysInfo : : windowsVersion ( ) & QSysInfo : : WV_NT_based )
2010-11-06 23:04:47 +01:00
{
case QSysInfo : : WV_XP :
2010-12-27 01:50:22 +01:00
qDebug ( " Running on Windows XP. \n " ) ;
2011-03-08 01:12:40 +01:00
lamexp_check_compatibility_mode ( " GetLargePageMinimum " , executableName ) ;
2010-11-06 23:04:47 +01:00
break ;
case QSysInfo : : WV_2003 :
2010-12-27 01:50:22 +01:00
qDebug ( " Running on Windows Server 2003 or Windows XP x64-Edition. \n " ) ;
2011-03-08 01:12:40 +01:00
lamexp_check_compatibility_mode ( " GetLocaleInfoEx " , executableName ) ;
2010-11-06 23:04:47 +01:00
break ;
case QSysInfo : : WV_VISTA :
2010-12-27 01:50:22 +01:00
qDebug ( " Running on Windows Vista or Windows Server 2008. \n " ) ;
2011-03-08 01:12:40 +01:00
lamexp_check_compatibility_mode ( " CreateRemoteThreadEx " , executableName ) ;
2010-11-06 23:04:47 +01:00
break ;
case QSysInfo : : WV_WINDOWS7 :
qDebug ( " Running on Windows 7 or Windows Server 2008 R2. \n " ) ;
2011-03-08 01:12:40 +01:00
lamexp_check_compatibility_mode ( NULL , executableName ) ;
2010-11-06 23:04:47 +01:00
break ;
default :
2011-04-08 15:36:41 +02:00
qFatal ( " %s " , QApplication : : tr ( " Executable '%1' requires Windows XP or later. " ) . arg ( QString : : fromLatin1 ( executableName ) ) . toLatin1 ( ) . constData ( ) ) ;
2010-11-06 23:04:47 +01:00
break ;
}
2010-12-23 03:13:11 +01:00
2011-04-17 00:17:43 +02:00
//Check for Wine
QLibrary ntdll ( " ntdll.dll " ) ;
bool isWine = ( ntdll . resolve ( " wine_get_version " ) ! = NULL ) | | ( ntdll . resolve ( " wine_nt_to_unix_file_name " ) ! = NULL ) ;
if ( isWine ) qWarning ( " It appears we are running under Wine, unexpected things might happen! \n " ) ;
2010-11-06 23:04:47 +01:00
//Create Qt application instance and setup version info
2011-04-04 14:55:46 +02:00
QDate date = QDate : : currentDate ( ) ;
2010-11-06 23:04:47 +01:00
QApplication * application = new QApplication ( argc , argv ) ;
application - > setApplicationName ( " LameXP - Audio Encoder Front-End " ) ;
application - > setApplicationVersion ( QString ( ) . sprintf ( " %d.%02d.%04d " , lamexp_version_major ( ) , lamexp_version_minor ( ) , lamexp_version_build ( ) ) ) ;
application - > setOrganizationName ( " LoRd_MuldeR " ) ;
application - > setOrganizationDomain ( " mulder.dummwiedeutsch.de " ) ;
2011-04-04 14:55:46 +02:00
application - > setWindowIcon ( ( date . month ( ) = = 12 & & date . day ( ) > = 24 & & date . day ( ) < = 26 ) ? QIcon ( " :/MainIcon2.png " ) : QIcon ( " :/MainIcon.png " ) ) ;
2010-11-06 23:04:47 +01:00
//Load plugins from application directory
QCoreApplication : : setLibraryPaths ( QStringList ( ) < < QApplication : : applicationDirPath ( ) ) ;
2010-11-07 23:06:30 +01:00
qDebug ( " Library Path: \n %s \n " , QApplication : : libraryPaths ( ) . first ( ) . toUtf8 ( ) . constData ( ) ) ;
2010-11-11 19:37:16 +01:00
//Init static Qt plugins
LAMEXP_INIT_QT_STATIC_PLUGIN ( qico ) ;
LAMEXP_INIT_QT_STATIC_PLUGIN ( qsvg ) ;
2010-11-11 00:25:23 +01:00
2010-11-06 23:04:47 +01:00
//Check for supported image formats
QList < QByteArray > supportedFormats = QImageReader : : supportedImageFormats ( ) ;
2010-11-11 19:37:16 +01:00
for ( int i = 0 ; g_lamexp_imageformats [ i ] ; i + + )
2010-11-06 23:04:47 +01:00
{
2010-11-11 19:37:16 +01:00
if ( ! supportedFormats . contains ( g_lamexp_imageformats [ i ] ) )
{
2011-03-08 01:12:40 +01:00
qFatal ( " Qt initialization error: QImageIOHandler for '%s' missing! " , g_lamexp_imageformats [ i ] ) ;
2010-11-11 19:37:16 +01:00
return false ;
}
2010-11-06 23:04:47 +01:00
}
2010-12-30 16:12:21 +01:00
2011-01-04 19:58:18 +01:00
//Add default translations
2011-01-21 00:10:51 +01:00
g_lamexp_translation . files . insert ( LAMEXP_DEFAULT_LANGID , " " ) ;
g_lamexp_translation . names . insert ( LAMEXP_DEFAULT_LANGID , " English " ) ;
2011-01-04 19:58:18 +01:00
2010-12-27 01:50:22 +01:00
//Check for process elevation
if ( ! lamexp_check_elevation ( ) )
{
if ( QMessageBox : : warning ( NULL , " LameXP " , " <nobr>LameXP was started with elevated rights. This is a potential security risk!</nobr> " , " Quit Program (Recommended) " , " Ignore " ) = = 0 )
{
return false ;
}
}
2011-03-18 01:12:17 +01:00
//Update console icon, if a console is attached
2011-04-17 00:17:43 +02:00
if ( g_lamexp_console_attached & & ! isWine )
2011-03-18 01:12:17 +01:00
{
typedef DWORD ( __stdcall * SetConsoleIconFun ) ( HICON ) ;
QLibrary kernel32 ( " kernel32.dll " ) ;
SetConsoleIconFun SetConsoleIconPtr = ( SetConsoleIconFun ) kernel32 . resolve ( " SetConsoleIcon " ) ;
if ( SetConsoleIconPtr )
{
SetConsoleIconPtr ( QIcon ( " :/icons/sound.png " ) . pixmap ( 16 , 16 ) . toWinHICON ( ) ) ;
}
}
2010-11-06 23:04:47 +01:00
//Done
qt_initialized = true ;
return true ;
}
/*
2010-11-08 21:47:35 +01:00
* Initialize IPC
2010-11-06 23:04:47 +01:00
*/
2010-11-08 21:47:35 +01:00
int lamexp_init_ipc ( void )
2010-11-06 23:04:47 +01:00
{
2011-01-20 22:54:10 +01:00
if ( g_lamexp_ipc_ptr . sharedmem & & g_lamexp_ipc_ptr . semaphore_read & & g_lamexp_ipc_ptr . semaphore_write )
2010-11-06 23:04:47 +01:00
{
2010-11-08 21:47:35 +01:00
return 0 ;
2010-11-06 23:04:47 +01:00
}
2010-11-11 22:58:02 +01:00
2011-01-20 22:54:10 +01:00
g_lamexp_ipc_ptr . semaphore_read = new QSystemSemaphore ( QString ( g_lamexp_ipc_uuid . semaphore_read ) , 0 ) ;
g_lamexp_ipc_ptr . semaphore_write = new QSystemSemaphore ( QString ( g_lamexp_ipc_uuid . semaphore_write ) , 0 ) ;
2010-11-08 19:29:36 +01:00
2011-01-20 22:54:10 +01:00
if ( g_lamexp_ipc_ptr . semaphore_read - > error ( ) ! = QSystemSemaphore : : NoError )
2010-11-08 19:29:36 +01:00
{
2011-01-20 22:54:10 +01:00
QString errorMessage = g_lamexp_ipc_ptr . semaphore_read - > errorString ( ) ;
LAMEXP_DELETE ( g_lamexp_ipc_ptr . semaphore_read ) ;
LAMEXP_DELETE ( g_lamexp_ipc_ptr . semaphore_write ) ;
2010-11-08 19:29:36 +01:00
qFatal ( " Failed to create system smaphore: %s " , errorMessage . toUtf8 ( ) . constData ( ) ) ;
2010-11-08 21:47:35 +01:00
return - 1 ;
2010-11-08 19:29:36 +01:00
}
2011-01-20 22:54:10 +01:00
if ( g_lamexp_ipc_ptr . semaphore_write - > error ( ) ! = QSystemSemaphore : : NoError )
2010-11-08 19:29:36 +01:00
{
2011-01-20 22:54:10 +01:00
QString errorMessage = g_lamexp_ipc_ptr . semaphore_write - > errorString ( ) ;
LAMEXP_DELETE ( g_lamexp_ipc_ptr . semaphore_read ) ;
LAMEXP_DELETE ( g_lamexp_ipc_ptr . semaphore_write ) ;
2010-11-08 19:29:36 +01:00
qFatal ( " Failed to create system smaphore: %s " , errorMessage . toUtf8 ( ) . constData ( ) ) ;
2010-11-08 21:47:35 +01:00
return - 1 ;
2010-11-08 19:29:36 +01:00
}
2011-01-20 22:54:10 +01:00
g_lamexp_ipc_ptr . sharedmem = new QSharedMemory ( QString ( g_lamexp_ipc_uuid . sharedmem ) , NULL ) ;
2010-11-06 23:04:47 +01:00
2011-01-20 22:54:10 +01:00
if ( ! g_lamexp_ipc_ptr . sharedmem - > create ( sizeof ( lamexp_ipc_t ) ) )
2010-11-06 23:04:47 +01:00
{
2011-01-20 22:54:10 +01:00
if ( g_lamexp_ipc_ptr . sharedmem - > error ( ) = = QSharedMemory : : AlreadyExists )
2010-11-06 23:04:47 +01:00
{
2011-01-20 22:54:10 +01:00
g_lamexp_ipc_ptr . sharedmem - > attach ( ) ;
if ( g_lamexp_ipc_ptr . sharedmem - > error ( ) = = QSharedMemory : : NoError )
2010-11-08 19:29:36 +01:00
{
2010-11-08 21:47:35 +01:00
return 1 ;
2010-11-08 19:29:36 +01:00
}
else
{
2011-01-20 22:54:10 +01:00
QString errorMessage = g_lamexp_ipc_ptr . sharedmem - > errorString ( ) ;
2010-11-08 21:47:35 +01:00
qFatal ( " Failed to attach to shared memory: %s " , errorMessage . toUtf8 ( ) . constData ( ) ) ;
return - 1 ;
2010-11-08 19:29:36 +01:00
}
2010-11-06 23:04:47 +01:00
}
else
{
2011-01-20 22:54:10 +01:00
QString errorMessage = g_lamexp_ipc_ptr . sharedmem - > errorString ( ) ;
2010-11-07 23:06:30 +01:00
qFatal ( " Failed to create shared memory: %s " , errorMessage . toUtf8 ( ) . constData ( ) ) ;
2010-11-08 21:47:35 +01:00
return - 1 ;
2010-11-06 23:04:47 +01:00
}
}
2011-01-20 22:54:10 +01:00
memset ( g_lamexp_ipc_ptr . sharedmem - > data ( ) , 0 , sizeof ( lamexp_ipc_t ) ) ;
g_lamexp_ipc_ptr . semaphore_write - > release ( ) ;
2010-11-08 19:29:36 +01:00
2010-11-08 21:47:35 +01:00
return 0 ;
2010-11-06 23:04:47 +01:00
}
2010-11-08 19:29:36 +01:00
/*
* IPC send message
*/
void lamexp_ipc_send ( unsigned int command , const char * message )
{
2011-01-20 22:54:10 +01:00
if ( ! g_lamexp_ipc_ptr . sharedmem | | ! g_lamexp_ipc_ptr . semaphore_read | | ! g_lamexp_ipc_ptr . semaphore_write )
2010-11-08 19:29:36 +01:00
{
throw " Shared memory for IPC not initialized yet. " ;
}
lamexp_ipc_t * lamexp_ipc = new lamexp_ipc_t ;
memset ( lamexp_ipc , 0 , sizeof ( lamexp_ipc_t ) ) ;
lamexp_ipc - > command = command ;
2010-11-09 22:06:11 +01:00
if ( message )
{
strcpy_s ( lamexp_ipc - > parameter , 4096 , message ) ;
}
2010-11-08 19:29:36 +01:00
2011-01-20 22:54:10 +01:00
if ( g_lamexp_ipc_ptr . semaphore_write - > acquire ( ) )
2010-11-08 21:47:35 +01:00
{
2011-01-20 22:54:10 +01:00
memcpy ( g_lamexp_ipc_ptr . sharedmem - > data ( ) , lamexp_ipc , sizeof ( lamexp_ipc_t ) ) ;
g_lamexp_ipc_ptr . semaphore_read - > release ( ) ;
2010-11-08 21:47:35 +01:00
}
2010-11-08 19:29:36 +01:00
LAMEXP_DELETE ( lamexp_ipc ) ;
}
/*
* IPC read message
*/
void lamexp_ipc_read ( unsigned int * command , char * message , size_t buffSize )
{
* command = 0 ;
message [ 0 ] = ' \0 ' ;
2011-01-20 22:54:10 +01:00
if ( ! g_lamexp_ipc_ptr . sharedmem | | ! g_lamexp_ipc_ptr . semaphore_read | | ! g_lamexp_ipc_ptr . semaphore_write )
2010-11-08 19:29:36 +01:00
{
throw " Shared memory for IPC not initialized yet. " ;
}
lamexp_ipc_t * lamexp_ipc = new lamexp_ipc_t ;
memset ( lamexp_ipc , 0 , sizeof ( lamexp_ipc_t ) ) ;
2011-01-20 22:54:10 +01:00
if ( g_lamexp_ipc_ptr . semaphore_read - > acquire ( ) )
2010-11-08 19:29:36 +01:00
{
2011-01-20 22:54:10 +01:00
memcpy ( lamexp_ipc , g_lamexp_ipc_ptr . sharedmem - > data ( ) , sizeof ( lamexp_ipc_t ) ) ;
g_lamexp_ipc_ptr . semaphore_write - > release ( ) ;
2010-11-08 21:47:35 +01:00
if ( ! ( lamexp_ipc - > reserved_1 | | lamexp_ipc - > reserved_2 ) )
{
* command = lamexp_ipc - > command ;
strcpy_s ( message , buffSize , lamexp_ipc - > parameter ) ;
}
else
{
qWarning ( " Malformed IPC message, will be ignored " ) ;
}
2010-11-08 19:29:36 +01:00
}
2010-11-08 21:47:35 +01:00
LAMEXP_DELETE ( lamexp_ipc ) ;
}
2011-01-24 00:04:07 +01:00
/*
* Check for LameXP " portable " mode
*/
bool lamexp_portable_mode ( void )
{
QString baseName = QFileInfo ( QApplication : : applicationFilePath ( ) ) . completeBaseName ( ) ;
return baseName . contains ( " lamexp " , Qt : : CaseInsensitive ) & & baseName . contains ( " portable " , Qt : : CaseInsensitive ) ;
}
2010-12-07 22:58:28 +01:00
/*
* Get a random string
*/
QString lamexp_rand_str ( void )
{
QRegExp regExp ( " \\ {( \\ w+) - ( \ \ w + ) - ( \ \ w + ) - ( \ \ w + ) - ( \ \ w + ) \ \ } " ) ;
QString uuid = QUuid : : createUuid ( ) . toString ( ) ;
if ( regExp . indexIn ( uuid ) > = 0 )
{
return QString ( ) . append ( regExp . cap ( 1 ) ) . append ( regExp . cap ( 2 ) ) . append ( regExp . cap ( 3 ) ) . append ( regExp . cap ( 4 ) ) . append ( regExp . cap ( 5 ) ) ;
}
throw " The RegExp didn't match on the UUID string. This shouldn't happen ;-) " ;
}
2010-11-06 23:04:47 +01:00
/*
* Get LameXP temp folder
*/
2011-02-25 22:03:39 +01:00
const QString & lamexp_temp_folder2 ( void )
2010-11-06 23:04:47 +01:00
{
2010-12-12 13:44:11 +01:00
static const char * TEMP_STR = " Temp " ;
2011-03-08 01:12:40 +01:00
const QString WRITE_TEST_DATA = lamexp_rand_str ( ) ;
const QString SUB_FOLDER = lamexp_rand_str ( ) ;
2010-12-07 22:58:28 +01:00
2011-03-08 01:12:40 +01:00
//Already initialized?
if ( ! g_lamexp_temp_folder . isEmpty ( ) )
{
if ( QDir ( g_lamexp_temp_folder ) . exists ( ) )
{
return g_lamexp_temp_folder ;
}
else
{
g_lamexp_temp_folder . clear ( ) ;
}
}
//Try the %TMP% or %TEMP% directory first
QDir temp = QDir : : temp ( ) ;
if ( temp . exists ( ) )
{
temp . mkdir ( SUB_FOLDER ) ;
if ( temp . cd ( SUB_FOLDER ) & & temp . exists ( ) )
{
QFile testFile ( QString ( " %1/~%2.tmp " ) . arg ( temp . canonicalPath ( ) , lamexp_rand_str ( ) ) ) ;
if ( testFile . open ( QIODevice : : ReadWrite ) )
{
if ( testFile . write ( WRITE_TEST_DATA . toLatin1 ( ) . constData ( ) ) > = strlen ( WRITE_TEST_DATA . toLatin1 ( ) . constData ( ) ) )
{
g_lamexp_temp_folder = temp . canonicalPath ( ) ;
}
testFile . remove ( ) ;
}
}
if ( ! g_lamexp_temp_folder . isEmpty ( ) )
{
return g_lamexp_temp_folder ;
}
}
//Create TEMP folder in %LOCALAPPDATA%
QDir localAppData = QDir ( lamexp_known_folder ( lamexp_folder_localappdata ) ) ;
if ( ! localAppData . path ( ) . isEmpty ( ) )
{
if ( ! localAppData . exists ( ) )
{
localAppData . mkpath ( " . " ) ;
}
if ( localAppData . exists ( ) )
{
if ( ! localAppData . entryList ( QDir : : AllDirs ) . contains ( TEMP_STR , Qt : : CaseInsensitive ) )
{
localAppData . mkdir ( TEMP_STR ) ;
}
if ( localAppData . cd ( TEMP_STR ) & & localAppData . exists ( ) )
{
localAppData . mkdir ( SUB_FOLDER ) ;
if ( localAppData . cd ( SUB_FOLDER ) & & localAppData . exists ( ) )
{
QFile testFile ( QString ( " %1/~%2.tmp " ) . arg ( localAppData . canonicalPath ( ) , lamexp_rand_str ( ) ) ) ;
if ( testFile . open ( QIODevice : : ReadWrite ) )
{
if ( testFile . write ( WRITE_TEST_DATA . toLatin1 ( ) . constData ( ) ) > = strlen ( WRITE_TEST_DATA . toLatin1 ( ) . constData ( ) ) )
{
g_lamexp_temp_folder = localAppData . canonicalPath ( ) ;
}
testFile . remove ( ) ;
}
}
}
}
if ( ! g_lamexp_temp_folder . isEmpty ( ) )
{
return g_lamexp_temp_folder ;
}
}
//Failed to create TEMP folder!
2011-03-09 00:20:36 +01:00
qFatal ( " Temporary directory could not be initialized! \n \n First attempt: \n %s \n \n Second attempt: \n %s " , temp . canonicalPath ( ) . toUtf8 ( ) . constData ( ) , localAppData . canonicalPath ( ) . toUtf8 ( ) . constData ( ) ) ;
2011-03-08 01:12:40 +01:00
return g_lamexp_temp_folder ;
}
2010-11-06 23:04:47 +01:00
/*
* Clean folder
*/
2011-03-21 17:36:43 +01:00
bool lamexp_clean_folder ( const QString & folderPath )
2010-11-06 23:04:47 +01:00
{
QDir tempFolder ( folderPath ) ;
2010-12-21 01:09:25 +01:00
QFileInfoList entryList = tempFolder . entryInfoList ( QDir : : AllEntries | QDir : : NoDotAndDotDot ) ;
2010-11-06 23:04:47 +01:00
for ( int i = 0 ; i < entryList . count ( ) ; i + + )
{
if ( entryList . at ( i ) . isDir ( ) )
{
2010-11-15 22:07:46 +01:00
lamexp_clean_folder ( entryList . at ( i ) . canonicalFilePath ( ) ) ;
2010-11-06 23:04:47 +01:00
}
else
{
2010-12-21 01:09:25 +01:00
for ( int j = 0 ; j < 3 ; j + + )
{
if ( lamexp_remove_file ( entryList . at ( i ) . canonicalFilePath ( ) ) )
{
break ;
}
}
2010-11-06 23:04:47 +01:00
}
}
tempFolder . rmdir ( " . " ) ;
return ! tempFolder . exists ( ) ;
}
/*
* Register tool
*/
2010-11-12 15:58:53 +01:00
void lamexp_register_tool ( const QString & toolName , LockedFile * file , unsigned int version )
2010-11-06 23:04:47 +01:00
{
2010-11-11 22:58:02 +01:00
if ( g_lamexp_tool_registry . contains ( toolName . toLower ( ) ) )
2010-11-06 23:04:47 +01:00
{
throw " lamexp_register_tool: Tool is already registered! " ;
}
2010-11-11 22:58:02 +01:00
g_lamexp_tool_registry . insert ( toolName . toLower ( ) , file ) ;
2010-11-12 15:58:53 +01:00
g_lamexp_tool_versions . insert ( toolName . toLower ( ) , version ) ;
2010-11-06 23:04:47 +01:00
}
/*
2010-11-11 22:58:02 +01:00
* Check for tool
*/
bool lamexp_check_tool ( const QString & toolName )
{
return g_lamexp_tool_registry . contains ( toolName . toLower ( ) ) ;
}
/*
2010-12-15 22:39:18 +01:00
* Lookup tool path
2010-11-06 23:04:47 +01:00
*/
const QString lamexp_lookup_tool ( const QString & toolName )
{
2010-11-11 22:58:02 +01:00
if ( g_lamexp_tool_registry . contains ( toolName . toLower ( ) ) )
2010-11-06 23:04:47 +01:00
{
2010-11-12 15:58:53 +01:00
return g_lamexp_tool_registry . value ( toolName . toLower ( ) ) - > filePath ( ) ;
2010-11-06 23:04:47 +01:00
}
else
{
return QString ( ) ;
}
2010-11-07 16:32:54 +01:00
}
2010-11-12 15:58:53 +01:00
/*
2010-12-15 22:39:18 +01:00
* Lookup tool version
2010-11-12 15:58:53 +01:00
*/
unsigned int lamexp_tool_version ( const QString & toolName )
{
if ( g_lamexp_tool_versions . contains ( toolName . toLower ( ) ) )
{
return g_lamexp_tool_versions . value ( toolName . toLower ( ) ) ;
}
else
{
return UINT_MAX ;
}
}
/*
* Version number to human - readable string
*/
2011-01-14 23:34:31 +01:00
const QString lamexp_version2string ( const QString & pattern , unsigned int version , const QString & defaultText )
2010-11-12 15:58:53 +01:00
{
2010-12-15 22:39:18 +01:00
if ( version = = UINT_MAX )
{
2011-01-14 23:34:31 +01:00
return defaultText ;
2010-12-15 22:39:18 +01:00
}
2010-11-12 15:58:53 +01:00
QString result = pattern ;
int digits = result . count ( " ? " , Qt : : CaseInsensitive ) ;
if ( digits < 1 )
{
return result ;
}
int pos = 0 ;
QString versionStr = QString ( ) . sprintf ( QString ( ) . sprintf ( " %%0%du " , digits ) . toLatin1 ( ) . constData ( ) , version ) ;
int index = result . indexOf ( " ? " , Qt : : CaseInsensitive ) ;
while ( index > = 0 & & pos < versionStr . length ( ) )
{
result [ index ] = versionStr [ pos + + ] ;
index = result . indexOf ( " ? " , Qt : : CaseInsensitive ) ;
}
return result ;
}
2011-01-04 19:58:18 +01:00
/*
* Register a new translation
*/
bool lamexp_translation_register ( const QString & langId , const QString & qmFile , const QString & langName , unsigned int & systemId )
{
if ( qmFile . isEmpty ( ) | | langName . isEmpty ( ) | | systemId < 1 )
{
return false ;
}
2011-01-21 00:10:51 +01:00
g_lamexp_translation . files . insert ( langId , qmFile ) ;
g_lamexp_translation . names . insert ( langId , langName ) ;
g_lamexp_translation . sysid . insert ( langId , systemId ) ;
2011-01-04 19:58:18 +01:00
return true ;
}
2010-12-28 21:26:16 +01:00
/*
2010-12-30 16:12:21 +01:00
* Get list of all translations
2010-12-28 21:26:16 +01:00
*/
QStringList lamexp_query_translations ( void )
{
2011-01-21 00:10:51 +01:00
return g_lamexp_translation . files . keys ( ) ;
2010-12-30 16:12:21 +01:00
}
/*
* Get translation name
*/
QString lamexp_translation_name ( const QString & langId )
{
2011-01-21 00:10:51 +01:00
return g_lamexp_translation . names . value ( langId . toLower ( ) , QString ( ) ) ;
2010-12-28 21:26:16 +01:00
}
2011-01-02 01:09:05 +01:00
/*
* Get translation system id
*/
2011-01-04 19:58:18 +01:00
unsigned int lamexp_translation_sysid ( const QString & langId )
2011-01-02 01:09:05 +01:00
{
2011-01-21 00:10:51 +01:00
return g_lamexp_translation . sysid . value ( langId . toLower ( ) , 0 ) ;
2011-01-02 01:09:05 +01:00
}
2010-12-28 21:26:16 +01:00
/*
* Install a new translator
*/
2010-12-30 16:12:21 +01:00
bool lamexp_install_translator ( const QString & langId )
2010-12-28 21:26:16 +01:00
{
bool success = false ;
2010-12-30 16:12:21 +01:00
if ( langId . isEmpty ( ) | | langId . toLower ( ) . compare ( LAMEXP_DEFAULT_LANGID ) = = 0 )
2010-12-28 21:26:16 +01:00
{
2011-01-06 00:53:52 +01:00
success = lamexp_install_translator_from_file ( QString ( ) ) ;
2010-12-28 21:26:16 +01:00
}
else
{
2011-01-21 00:10:51 +01:00
QString qmFile = g_lamexp_translation . files . value ( langId . toLower ( ) , QString ( ) ) ;
2010-12-28 21:26:16 +01:00
if ( ! qmFile . isEmpty ( ) )
{
2011-01-06 00:53:52 +01:00
success = lamexp_install_translator_from_file ( QString ( " :/localization/%1 " ) . arg ( qmFile ) ) ;
2010-12-28 21:26:16 +01:00
}
else
{
2010-12-30 16:12:21 +01:00
qWarning ( " Translation '%s' not available! " , langId . toLatin1 ( ) . constData ( ) ) ;
2010-12-28 21:26:16 +01:00
}
}
return success ;
}
2011-01-06 00:53:52 +01:00
/*
* Install a new translator from file
*/
bool lamexp_install_translator_from_file ( const QString & qmFile )
{
bool success = false ;
if ( ! g_lamexp_currentTranslator )
{
g_lamexp_currentTranslator = new QTranslator ( ) ;
}
if ( ! qmFile . isEmpty ( ) )
{
QString qmPath = QFileInfo ( qmFile ) . canonicalFilePath ( ) ;
QApplication : : removeTranslator ( g_lamexp_currentTranslator ) ;
success = g_lamexp_currentTranslator - > load ( qmPath ) ;
QApplication : : installTranslator ( g_lamexp_currentTranslator ) ;
if ( ! success )
{
qWarning ( " Failed to load translation: \n \" %s \" " , qmPath . toLatin1 ( ) . constData ( ) ) ;
}
}
else
{
QApplication : : removeTranslator ( g_lamexp_currentTranslator ) ;
success = true ;
}
return success ;
}
2010-12-15 18:43:21 +01:00
/*
* Locate known folder on local system
*/
QString lamexp_known_folder ( lamexp_known_folder_t folder_id )
{
typedef HRESULT ( WINAPI * SHGetKnownFolderPathFun ) ( __in const GUID & rfid , __in DWORD dwFlags , __in HANDLE hToken , __out PWSTR * ppszPath ) ;
typedef HRESULT ( WINAPI * SHGetFolderPathFun ) ( __in HWND hwndOwner , __in int nFolder , __in HANDLE hToken , __in DWORD dwFlags , __out LPWSTR pszPath ) ;
static const int CSIDL_LOCAL_APPDATA = 0x001c ;
static const int CSIDL_PROGRAM_FILES = 0x0026 ;
2010-12-19 21:23:43 +01:00
static const int CSIDL_SYSTEM_FOLDER = 0x0025 ;
2010-12-15 18:43:21 +01:00
static const GUID GUID_LOCAL_APPDATA = { 0xF1B32785 , 0x6FBA , 0x4FCF , { 0x9D , 0x55 , 0x7B , 0x8E , 0x7F , 0x15 , 0x70 , 0x91 } } ;
static const GUID GUID_LOCAL_APPDATA_LOW = { 0xA520A1A4 , 0x1780 , 0x4FF6 , { 0xBD , 0x18 , 0x16 , 0x73 , 0x43 , 0xC5 , 0xAF , 0x16 } } ;
static const GUID GUID_PROGRAM_FILES = { 0x905e63b6 , 0xc1bf , 0x494e , { 0xb2 , 0x9c , 0x65 , 0xb7 , 0x32 , 0xd3 , 0xd2 , 0x1a } } ;
2010-12-19 21:23:43 +01:00
static const GUID GUID_SYSTEM_FOLDER = { 0x1AC14E77 , 0x02E7 , 0x4E5D , { 0xB7 , 0x44 , 0x2E , 0xB1 , 0xAE , 0x51 , 0x98 , 0xB7 } } ;
2010-12-15 18:43:21 +01:00
2010-12-15 19:57:51 +01:00
static QLibrary * Kernel32Lib = NULL ;
2010-12-15 18:43:21 +01:00
static SHGetKnownFolderPathFun SHGetKnownFolderPathPtr = NULL ;
static SHGetFolderPathFun SHGetFolderPathPtr = NULL ;
if ( ( ! SHGetKnownFolderPathPtr ) & & ( ! SHGetFolderPathPtr ) )
{
2010-12-15 19:57:51 +01:00
if ( ! Kernel32Lib ) Kernel32Lib = new QLibrary ( " shell32.dll " ) ;
SHGetKnownFolderPathPtr = ( SHGetKnownFolderPathFun ) Kernel32Lib - > resolve ( " SHGetKnownFolderPath " ) ;
SHGetFolderPathPtr = ( SHGetFolderPathFun ) Kernel32Lib - > resolve ( " SHGetFolderPathW " ) ;
2010-12-15 18:43:21 +01:00
}
int folderCSIDL = - 1 ;
GUID folderGUID = { 0x0000 , 0x0000 , 0x0000 , { 0x00 , 0x00 , 0x00 , 0x00 , 0x00 , 0x00 , 0x00 , 0x00 } } ;
switch ( folder_id )
{
case lamexp_folder_localappdata :
folderCSIDL = CSIDL_LOCAL_APPDATA ;
folderGUID = GUID_LOCAL_APPDATA ;
break ;
case lamexp_folder_programfiles :
folderCSIDL = CSIDL_PROGRAM_FILES ;
folderGUID = GUID_PROGRAM_FILES ;
break ;
2010-12-19 21:23:43 +01:00
case lamexp_folder_systemfolder :
folderCSIDL = CSIDL_SYSTEM_FOLDER ;
folderGUID = GUID_SYSTEM_FOLDER ;
break ;
2010-12-15 18:43:21 +01:00
default :
return QString ( ) ;
break ;
}
QString folder ;
if ( SHGetKnownFolderPathPtr )
{
WCHAR * path = NULL ;
if ( SHGetKnownFolderPathPtr ( folderGUID , 0x00008000 , NULL , & path ) = = S_OK )
{
2010-12-15 19:57:51 +01:00
//MessageBoxW(0, path, L"SHGetKnownFolderPath", MB_TOPMOST);
2010-12-15 18:43:21 +01:00
QDir folderTemp = QDir ( QDir : : fromNativeSeparators ( QString : : fromUtf16 ( reinterpret_cast < const unsigned short * > ( path ) ) ) ) ;
if ( ! folderTemp . exists ( ) )
{
folderTemp . mkpath ( " . " ) ;
}
if ( folderTemp . exists ( ) )
{
folder = folderTemp . canonicalPath ( ) ;
}
CoTaskMemFree ( path ) ;
}
}
else if ( SHGetFolderPathPtr )
{
WCHAR * path = new WCHAR [ 4096 ] ;
if ( SHGetFolderPathPtr ( NULL , folderCSIDL , NULL , NULL , path ) = = S_OK )
{
2010-12-15 19:57:51 +01:00
//MessageBoxW(0, path, L"SHGetFolderPathW", MB_TOPMOST);
2010-12-15 18:43:21 +01:00
QDir folderTemp = QDir ( QDir : : fromNativeSeparators ( QString : : fromUtf16 ( reinterpret_cast < const unsigned short * > ( path ) ) ) ) ;
if ( ! folderTemp . exists ( ) )
{
folderTemp . mkpath ( " . " ) ;
}
if ( folderTemp . exists ( ) )
{
folder = folderTemp . canonicalPath ( ) ;
}
}
delete [ ] path ;
}
return folder ;
}
2010-11-12 15:58:53 +01:00
2010-12-21 01:09:25 +01:00
/*
* Safely remove a file
*/
bool lamexp_remove_file ( const QString & filename )
{
if ( ! QFileInfo ( filename ) . exists ( ) | | ! QFileInfo ( filename ) . isFile ( ) )
{
return true ;
}
else
{
if ( ! QFile : : remove ( filename ) )
{
2011-02-25 22:03:39 +01:00
DWORD attributes = GetFileAttributesW ( QWCHAR ( filename ) ) ;
SetFileAttributesW ( QWCHAR ( filename ) , ( attributes & ( ~ FILE_ATTRIBUTE_READONLY ) ) ) ;
2010-12-21 01:09:25 +01:00
if ( ! QFile : : remove ( filename ) )
{
qWarning ( " Could not delete \" %s \" " , filename . toLatin1 ( ) . constData ( ) ) ;
return false ;
}
else
{
return true ;
}
}
else
{
return true ;
}
}
}
2010-12-19 00:50:22 +01:00
2011-01-02 20:47:26 +01:00
/*
* Check if visual themes are enabled ( WinXP and later )
*/
bool lamexp_themes_enabled ( void )
{
typedef int ( WINAPI * IsAppThemedFun ) ( void ) ;
bool isAppThemed = false ;
QLibrary uxTheme ( QString ( " %1/UxTheme.dll " ) . arg ( lamexp_known_folder ( lamexp_folder_systemfolder ) ) ) ;
IsAppThemedFun IsAppThemedPtr = ( IsAppThemedFun ) uxTheme . resolve ( " IsAppThemed " ) ;
if ( IsAppThemedPtr )
{
isAppThemed = IsAppThemedPtr ( ) ;
if ( ! isAppThemed )
{
qWarning ( " Theme support is disabled for this process! " ) ;
}
}
return isAppThemed ;
}
2010-12-19 00:50:22 +01:00
/*
2010-12-28 21:26:16 +01:00
* Get number of free bytes on disk
2010-12-19 00:50:22 +01:00
*/
__int64 lamexp_free_diskspace ( const QString & path )
{
ULARGE_INTEGER freeBytesAvailable , totalNumberOfBytes , totalNumberOfFreeBytes ;
if ( GetDiskFreeSpaceExW ( reinterpret_cast < const wchar_t * > ( QDir : : toNativeSeparators ( path ) . utf16 ( ) ) , & freeBytesAvailable , & totalNumberOfBytes , & totalNumberOfFreeBytes ) )
{
return freeBytesAvailable . QuadPart ;
}
else
{
return 0 ;
}
}
2011-03-18 01:12:17 +01:00
bool lamexp_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 < wchar_t * > ( QWCHAR ( message ) ) , timeout , forceShutdown , FALSE , reason ) ;
}
}
}
return false ;
}
2010-12-28 21:26:16 +01:00
/*
* Finalization function ( final clean - up )
*/
void lamexp_finalization ( void )
{
//Free all tools
if ( ! g_lamexp_tool_registry . isEmpty ( ) )
{
QStringList keys = g_lamexp_tool_registry . keys ( ) ;
for ( int i = 0 ; i < keys . count ( ) ; i + + )
{
LAMEXP_DELETE ( g_lamexp_tool_registry [ keys . at ( i ) ] ) ;
}
g_lamexp_tool_registry . clear ( ) ;
g_lamexp_tool_versions . clear ( ) ;
}
//Delete temporary files
if ( ! g_lamexp_temp_folder . isEmpty ( ) )
{
for ( int i = 0 ; i < 100 ; i + + )
{
if ( lamexp_clean_folder ( g_lamexp_temp_folder ) )
{
break ;
}
Sleep ( 125 ) ;
}
g_lamexp_temp_folder . clear ( ) ;
}
//Clear languages
2010-12-30 16:12:21 +01:00
if ( g_lamexp_currentTranslator )
{
QApplication : : removeTranslator ( g_lamexp_currentTranslator ) ;
LAMEXP_DELETE ( g_lamexp_currentTranslator ) ;
}
2011-01-21 00:10:51 +01:00
g_lamexp_translation . files . clear ( ) ;
g_lamexp_translation . names . clear ( ) ;
2010-12-28 21:26:16 +01:00
//Destroy Qt application object
QApplication * application = dynamic_cast < QApplication * > ( QApplication : : instance ( ) ) ;
LAMEXP_DELETE ( application ) ;
//Detach from shared memory
2011-01-20 22:54:10 +01:00
if ( g_lamexp_ipc_ptr . sharedmem ) g_lamexp_ipc_ptr . sharedmem - > detach ( ) ;
LAMEXP_DELETE ( g_lamexp_ipc_ptr . sharedmem ) ;
LAMEXP_DELETE ( g_lamexp_ipc_ptr . semaphore_read ) ;
LAMEXP_DELETE ( g_lamexp_ipc_ptr . semaphore_write ) ;
2010-12-28 21:26:16 +01:00
}
2010-11-07 16:32:54 +01:00
/*
* Get number private bytes [ debug only ]
*/
SIZE_T lamexp_dbg_private_bytes ( void )
{
# ifdef _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 //_DEBUG
}