OSDN Git Service

Now using Keccak/SHA-3 to verify the built-in files.
[lamexp/LameXP.git] / src / Thread_Initialization.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2012 LoRd_MuldeR <MuldeR2@GMX.de>
4 //
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; either version 2 of the License, or
8 // (at your option) any later version.
9 //
10 // This program is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 // GNU General Public License for more details.
14 //
15 // You should have received a copy of the GNU General Public License along
16 // with this program; if not, write to the Free Software Foundation, Inc.,
17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 //
19 // http://www.gnu.org/licenses/gpl-2.0.txt
20 ///////////////////////////////////////////////////////////////////////////////
21
22 #include "Thread_Initialization.h"
23
24 #include "LockedFile.h"
25 #include "Tools.h"
26
27 #include <QFileInfo>
28 #include <QCoreApplication>
29 #include <QProcess>
30 #include <QMap>
31 #include <QDir>
32 #include <QLibrary>
33 #include <QResource>
34 #include <QTime>
35 #include <QTextStream>
36 #include <QRunnable>
37 #include <QThreadPool>
38 #include <QMutex>
39
40 /* helper macros */
41 #define PRINT_CPU_TYPE(X) case X: qDebug("Selected CPU is: " #X)
42 static const double g_allowedExtractDelay = 12.0;
43
44 ////////////////////////////////////////////////////////////
45 // ExtractorTask class
46 ////////////////////////////////////////////////////////////
47
48 class ExtractorTask : public QRunnable
49 {
50 public:
51         ExtractorTask(const QDir &appDir, const QString &toolName, const QString &toolShortName, const QByteArray &toolHash, const unsigned int toolVersion)
52         :
53                 QRunnable(), m_appDir(appDir), m_toolName(toolName), m_toolShortName(toolShortName), m_toolHash(toolHash), m_toolVersion(toolVersion)
54         {
55                 /* Nothing to do */
56         }
57
58         static void clearFlags(void)
59         {
60                 s_bAbort = s_bCustom = false;
61                 s_errMsg[0] = '\0';
62         }
63
64         static bool getAbort(void) { return s_bAbort; }
65         static bool getCustom(void) { return s_bCustom; }
66         static char *const getError(void) { return s_errMsg; }
67
68 protected:
69         void run(void)
70         {
71                 try
72                 {
73                         LockedFile *lockedFile = NULL;
74                         unsigned int version = m_toolVersion;
75                         
76                         if(!s_bAbort)
77                         {
78                                 QFileInfo customTool(QString("%1/tools/%2/%3").arg(m_appDir.canonicalPath(), QString::number(lamexp_version_build()), m_toolShortName));
79                                 if(customTool.exists() && customTool.isFile())
80                                 {
81                                         qDebug("Setting up file: %s <- %s", m_toolShortName.toLatin1().constData(), m_appDir.relativeFilePath(customTool.canonicalFilePath()).toLatin1().constData());
82                                         lockedFile = new LockedFile(customTool.canonicalFilePath()); version = UINT_MAX; s_bCustom = true;
83                                 }
84                                 else
85                                 {
86                                         qDebug("Extracting file: %s -> %s", m_toolName.toLatin1().constData(), m_toolShortName.toLatin1().constData());
87                                         lockedFile = new LockedFile(QString(":/tools/%1").arg(m_toolName), QString("%1/lxp_%2").arg(lamexp_temp_folder2(), m_toolShortName), m_toolHash);
88                                 }
89
90                                 if(lockedFile)
91                                 {
92                                         QMutexLocker lock(&s_mutex);
93                                         lamexp_register_tool(m_toolShortName, lockedFile, version);
94                                 }
95                         }
96                 }
97                 catch(char *errorMsg)
98                 {
99                         qWarning("At least one of the required tools could not be initialized:\n%s", errorMsg);
100                         if(s_mutex.tryLock())
101                         {
102                                 if(!s_bAbort) { s_bAbort = true; strncpy_s(s_errMsg, 1024, errorMsg, _TRUNCATE); }
103                                 s_mutex.unlock();
104                         }
105                 }
106         }
107
108 private:
109         const QDir m_appDir;
110         const QString m_toolName;
111         const QString m_toolShortName;
112         const QByteArray m_toolHash;
113         const unsigned int m_toolVersion;
114
115         static volatile bool s_bAbort;
116         static volatile bool s_bCustom;
117         static QMutex s_mutex;
118         static char s_errMsg[1024];
119 };
120
121 volatile bool ExtractorTask::s_bAbort = false;
122 volatile bool ExtractorTask::s_bCustom = false;
123 char ExtractorTask::s_errMsg[1024] = {'\0'};
124 QMutex ExtractorTask::s_mutex;
125
126 ////////////////////////////////////////////////////////////
127 // Constructor
128 ////////////////////////////////////////////////////////////
129
130 InitializationThread::InitializationThread(const lamexp_cpu_t *cpuFeatures)
131 {
132         m_bSuccess = false;
133         memset(&m_cpuFeatures, 0, sizeof(lamexp_cpu_t));
134         m_slowIndicator = false;
135         
136         if(cpuFeatures)
137         {
138                 memcpy(&m_cpuFeatures, cpuFeatures, sizeof(lamexp_cpu_t));
139         }
140 }
141
142 ////////////////////////////////////////////////////////////
143 // Thread Main
144 ////////////////////////////////////////////////////////////
145
146 void InitializationThread::run()
147 {
148         m_bSuccess = false;
149         delay();
150
151         //CPU type selection
152         unsigned int cpuSupport = 0;
153         if(m_cpuFeatures.sse && m_cpuFeatures.sse2 && m_cpuFeatures.intel)
154         {
155                 cpuSupport = m_cpuFeatures.x64 ? CPU_TYPE_X64_SSE : CPU_TYPE_X86_SSE;
156         }
157         else
158         {
159                 cpuSupport = m_cpuFeatures.x64 ? CPU_TYPE_X64_GEN : CPU_TYPE_X86_GEN;
160         }
161
162         //Hack to disable x64 on Wine, as x64 binaries won't run under Wine (tested with Wine 1.4 under Ubuntu 12.04 x64)
163         if(cpuSupport & CPU_TYPE_X64_ALL)
164         {
165                 //DWORD osVerNo = lamexp_get_os_version();
166                 //if((HIWORD(osVerNo) == 6) && (LOWORD(osVerNo) == 2))
167                 if(lamexp_detect_wine())
168                 {
169                         qWarning("Running under Wine on a 64-Bit system. Going to disable all x64 support!\n");
170                         cpuSupport = (cpuSupport == CPU_TYPE_X64_SSE) ? CPU_TYPE_X86_SSE : CPU_TYPE_X86_GEN;
171                 }
172         }
173
174         //Print selected CPU type
175         switch(cpuSupport)
176         {
177                 PRINT_CPU_TYPE(CPU_TYPE_X86_GEN); break;
178                 PRINT_CPU_TYPE(CPU_TYPE_X86_SSE); break;
179                 PRINT_CPU_TYPE(CPU_TYPE_X64_GEN); break;
180                 PRINT_CPU_TYPE(CPU_TYPE_X64_SSE); break;
181                 default: throw "CPU support undefined!";
182         }
183
184         //Allocate maps
185         QMap<QString, QString> mapChecksum;
186         QMap<QString, unsigned int> mapVersion;
187         QMap<QString, unsigned int> mapCpuType;
188
189         //Init properties
190         for(int i = 0; i < INT_MAX; i++)
191         {
192                 if(!g_lamexp_tools[i].pcName && !g_lamexp_tools[i].pcHash && !g_lamexp_tools[i].uiVersion)
193                 {
194                         break;
195                 }
196                 else if(g_lamexp_tools[i].pcName && g_lamexp_tools[i].pcHash && g_lamexp_tools[i].uiVersion)
197                 {
198                         const QString currentTool = QString::fromLatin1(g_lamexp_tools[i].pcName);
199                         mapChecksum.insert(currentTool, QString::fromLatin1(g_lamexp_tools[i].pcHash));
200                         mapCpuType.insert(currentTool, g_lamexp_tools[i].uiCpuType);
201                         mapVersion.insert(currentTool, g_lamexp_tools[i].uiVersion);
202                 }
203                 else
204                 {
205                         qFatal("Inconsistent checksum data detected. Take care!");
206                 }
207         }
208
209         QDir toolsDir(":/tools/");
210         QList<QFileInfo> toolsList = toolsDir.entryInfoList(QStringList("*.*"), QDir::Files, QDir::Name);
211         QDir appDir = QDir(QCoreApplication::applicationDirPath()).canonicalPath();
212
213         QThreadPool *pool = new QThreadPool();
214         int idealThreadCount = QThread::idealThreadCount();
215         if(idealThreadCount > 0)
216         {
217                 pool->setMaxThreadCount(idealThreadCount * 2);
218         }
219         
220         ExtractorTask::clearFlags();
221
222         QTime timer;
223         timer.start();
224         
225         //Extract all files
226         while(!toolsList.isEmpty())
227         {
228                 try
229                 {
230                         QFileInfo currentTool = toolsList.takeFirst();
231                         QString toolName = currentTool.fileName().toLower();
232                         QString toolShortName = QString("%1.%2").arg(currentTool.baseName().toLower(), currentTool.suffix().toLower());
233                         
234                         QByteArray toolHash = mapChecksum.take(toolName).toLatin1();
235                         unsigned int toolCpuType = mapCpuType.take(toolName);
236                         unsigned int toolVersion = mapVersion.take(toolName);
237                         
238                         if(toolHash.size() != 96)
239                         {
240                                 throw "The required checksum is missing, take care!";
241                         }
242                         
243                         if(toolCpuType & cpuSupport)
244                         {
245                                 pool->start(new ExtractorTask(appDir, toolName, toolShortName, toolHash, toolVersion));
246                                 QThread::yieldCurrentThread();
247                         }
248                 }
249                 catch(char *errorMsg)
250                 {
251                         qFatal("At least one of the required tools could not be initialized:\n%s", errorMsg);
252                         return;
253                 }
254         }
255
256         //Wait for extrator threads to finish
257         pool->waitForDone();
258         LAMEXP_DELETE(pool);
259
260         //Make sure all files were extracted correctly
261         if(ExtractorTask::getAbort())
262         {
263                 qFatal("At least one of the required tools could not be initialized:\n%s", ExtractorTask::getError());
264                 return;
265         }
266
267         //Make sure all files were extracted
268         if(!mapChecksum.isEmpty())
269         {
270                 qFatal("At least one required tool could not be found:\n%s", toolsDir.filePath(mapChecksum.keys().first()).toLatin1().constData());
271                 return;
272         }
273
274         qDebug("All extracted.\n");
275
276         //Clean-up
277         mapChecksum.clear();
278         mapVersion.clear();
279         mapCpuType.clear();
280         
281         //Using any custom tools?
282         if(ExtractorTask::getCustom())
283         {
284                 qWarning("Warning: Using custom tools, you might encounter unexpected problems!\n");
285         }
286
287         //Check delay
288         double delayExtract = static_cast<double>(timer.elapsed()) / 1000.0;
289         if(delayExtract > g_allowedExtractDelay)
290         {
291                 m_slowIndicator = true;
292                 qWarning("Extracting tools took %.3f seconds -> probably slow realtime virus scanner.", delayExtract);
293                 qWarning("Please report performance problems to your anti-virus developer !!!\n");
294         }
295
296         //Register all translations
297         initTranslations();
298
299         //Look for AAC encoders
300         initNeroAac();
301         initFhgAac();
302         initQAac();
303
304         delay();
305         m_bSuccess = true;
306 }
307
308 ////////////////////////////////////////////////////////////
309 // PUBLIC FUNCTIONS
310 ////////////////////////////////////////////////////////////
311
312 void InitializationThread::delay(void)
313 {
314         const char *temp = "|/-\\";
315         printf("Thread is doing something important... ?\b", temp[4]);
316
317         for(int i = 0; i < 20; i++)
318         {
319                 printf("%c\b", temp[i%4]);
320         }
321
322         printf("Done\n\n");
323 }
324
325 void InitializationThread::initTranslations(void)
326 {
327         //Search for language files
328         QStringList qmFiles = QDir(":/localization").entryList(QStringList() << "LameXP_??.qm", QDir::Files, QDir::Name);
329
330         //Make sure we found at least one translation
331         if(qmFiles.count() < 1)
332         {
333                 qFatal("Could not find any translation files!");
334                 return;
335         }
336
337         //Add all available translations
338         while(!qmFiles.isEmpty())
339         {
340                 QString langId, langName;
341                 unsigned int systemId = 0, country = 0;
342                 QString qmFile = qmFiles.takeFirst();
343                 
344                 QRegExp langIdExp("LameXP_(\\w\\w)\\.qm", Qt::CaseInsensitive);
345                 if(langIdExp.indexIn(qmFile) >= 0)
346                 {
347                         langId = langIdExp.cap(1).toLower();
348                         QResource langRes = QResource(QString(":/localization/%1.txt").arg(qmFile));
349                         if(langRes.isValid() && langRes.size() > 0)
350                         {
351                                 QByteArray data = QByteArray::fromRawData(reinterpret_cast<const char*>(langRes.data()), langRes.size());
352                                 QTextStream stream(&data, QIODevice::ReadOnly);
353                                 stream.setAutoDetectUnicode(false); stream.setCodec("UTF-8");
354                                 while(!stream.atEnd())
355                                 {
356                                         QStringList langInfo = stream.readLine().simplified().split(",", QString::SkipEmptyParts);
357                                         if(langInfo.count() == 3)
358                                         {
359                                                 systemId = langInfo.at(0).trimmed().toUInt();
360                                                 country  = langInfo.at(1).trimmed().toUInt();
361                                                 langName = langInfo.at(2).trimmed();
362                                                 break;
363                                         }
364                                 }
365                         }
366                 }
367
368                 if(!(langId.isEmpty() || langName.isEmpty() || systemId == 0))
369                 {
370                         if(lamexp_translation_register(langId, qmFile, langName, systemId, country))
371                         {
372                                 qDebug("Registering translation: %s = %s (%u) [%u]", qmFile.toUtf8().constData(), langName.toUtf8().constData(), systemId, country);
373                         }
374                         else
375                         {
376                                 qWarning("Failed to register: %s", qmFile.toLatin1().constData());
377                         }
378                 }
379         }
380
381         qDebug("All registered.\n");
382 }
383
384 void InitializationThread::initNeroAac(void)
385 {
386         const QString appPath = QDir(QCoreApplication::applicationDirPath()).canonicalPath();
387         
388         QFileInfo neroFileInfo[3];
389         neroFileInfo[0] = QFileInfo(QString("%1/neroAacEnc.exe").arg(appPath));
390         neroFileInfo[1] = QFileInfo(QString("%1/neroAacDec.exe").arg(appPath));
391         neroFileInfo[2] = QFileInfo(QString("%1/neroAacTag.exe").arg(appPath));
392         
393         bool neroFilesFound = true;
394         for(int i = 0; i < 3; i++)      { if(!neroFileInfo[i].exists()) neroFilesFound = false; }
395
396         //Lock the Nero binaries
397         if(!neroFilesFound)
398         {
399                 qDebug("Nero encoder binaries not found -> AAC encoding support will be disabled!\n");
400                 return;
401         }
402
403         qDebug("Found Nero AAC encoder binary:\n%s\n", neroFileInfo[0].canonicalFilePath().toUtf8().constData());
404
405         LockedFile *neroBin[3];
406         for(int i = 0; i < 3; i++) neroBin[i] = NULL;
407
408         try
409         {
410                 for(int i = 0; i < 3; i++)
411                 {
412                         neroBin[i] = new LockedFile(neroFileInfo[i].canonicalFilePath());
413                 }
414         }
415         catch(...)
416         {
417                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
418                 qWarning("Failed to get excluive lock to Nero encoder binary -> AAC encoding support will be disabled!");
419                 return;
420         }
421
422         QProcess process;
423         process.setProcessChannelMode(QProcess::MergedChannels);
424         process.setReadChannel(QProcess::StandardOutput);
425         process.start(neroFileInfo[0].canonicalFilePath(), QStringList() << "-help");
426
427         if(!process.waitForStarted())
428         {
429                 qWarning("Nero process failed to create!");
430                 qWarning("Error message: \"%s\"\n", process.errorString().toLatin1().constData());
431                 process.kill();
432                 process.waitForFinished(-1);
433                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
434                 return;
435         }
436
437         unsigned int neroVersion = 0;
438
439         while(process.state() != QProcess::NotRunning)
440         {
441                 if(!process.waitForReadyRead())
442                 {
443                         if(process.state() == QProcess::Running)
444                         {
445                                 qWarning("Nero process time out -> killing!");
446                                 process.kill();
447                                 process.waitForFinished(-1);
448                                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
449                                 return;
450                         }
451                 }
452
453                 while(process.canReadLine())
454                 {
455                         QString line = QString::fromUtf8(process.readLine().constData()).simplified();
456                         QStringList tokens = line.split(" ", QString::SkipEmptyParts, Qt::CaseInsensitive);
457                         int index1 = tokens.indexOf("Package");
458                         int index2 = tokens.indexOf("version:");
459                         if(index1 >= 0 && index2 >= 0 && index1 + 1 == index2 && index2 < tokens.count() - 1)
460                         {
461                                 QStringList versionTokens = tokens.at(index2 + 1).split(".", QString::SkipEmptyParts, Qt::CaseInsensitive);
462                                 if(versionTokens.count() == 4)
463                                 {
464                                         neroVersion = 0;
465                                         neroVersion += qMin(9, qMax(0, versionTokens.at(3).toInt()));
466                                         neroVersion += qMin(9, qMax(0, versionTokens.at(2).toInt())) * 10;
467                                         neroVersion += qMin(9, qMax(0, versionTokens.at(1).toInt())) * 100;
468                                         neroVersion += qMin(9, qMax(0, versionTokens.at(0).toInt())) * 1000;
469                                 }
470                         }
471                 }
472         }
473
474         if(!(neroVersion > 0))
475         {
476                 qWarning("Nero AAC version could not be determined -> AAC encoding support will be disabled!");
477                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
478                 return;
479         }
480         
481         for(int i = 0; i < 3; i++)
482         {
483                 lamexp_register_tool(neroFileInfo[i].fileName(), neroBin[i], neroVersion);
484         }
485 }
486
487 void InitializationThread::initFhgAac(void)
488 {
489         const QString appPath = QDir(QCoreApplication::applicationDirPath()).canonicalPath();
490         
491         QFileInfo fhgFileInfo[5];
492         fhgFileInfo[0] = QFileInfo(QString("%1/fhgaacenc.exe").arg(appPath));
493         fhgFileInfo[1] = QFileInfo(QString("%1/enc_fhgaac.dll").arg(appPath));
494         fhgFileInfo[2] = QFileInfo(QString("%1/nsutil.dll").arg(appPath));
495         fhgFileInfo[3] = QFileInfo(QString("%1/libmp4v2.dll").arg(appPath));
496         fhgFileInfo[4] = QFileInfo(QString("%1/libsndfile-1.dll").arg(appPath));
497         
498         bool fhgFilesFound = true;
499         for(int i = 0; i < 5; i++)      { if(!fhgFileInfo[i].exists()) fhgFilesFound = false; }
500
501         //Lock the FhgAacEnc binaries
502         if(!fhgFilesFound)
503         {
504                 qDebug("FhgAacEnc binaries not found -> FhgAacEnc support will be disabled!\n");
505                 return;
506         }
507
508         qDebug("Found FhgAacEnc cli_exe:\n%s\n", fhgFileInfo[0].canonicalFilePath().toUtf8().constData());
509         qDebug("Found FhgAacEnc enc_dll:\n%s\n", fhgFileInfo[1].canonicalFilePath().toUtf8().constData());
510
511         LockedFile *fhgBin[5];
512         for(int i = 0; i < 5; i++) fhgBin[i] = NULL;
513
514         try
515         {
516                 for(int i = 0; i < 5; i++)
517                 {
518                         fhgBin[i] = new LockedFile(fhgFileInfo[i].canonicalFilePath());
519                 }
520         }
521         catch(...)
522         {
523                 for(int i = 0; i < 5; i++) LAMEXP_DELETE(fhgBin[i]);
524                 qWarning("Failed to get excluive lock to FhgAacEnc binary -> FhgAacEnc support will be disabled!");
525                 return;
526         }
527
528         QProcess process;
529         process.setProcessChannelMode(QProcess::MergedChannels);
530         process.setReadChannel(QProcess::StandardOutput);
531         process.start(fhgFileInfo[0].canonicalFilePath(), QStringList() << "--version");
532
533         if(!process.waitForStarted())
534         {
535                 qWarning("FhgAacEnc process failed to create!");
536                 qWarning("Error message: \"%s\"\n", process.errorString().toLatin1().constData());
537                 process.kill();
538                 process.waitForFinished(-1);
539                 for(int i = 0; i < 5; i++) LAMEXP_DELETE(fhgBin[i]);
540                 return;
541         }
542
543         QRegExp fhgAacEncSig("fhgaacenc version (\\d+) by tmkk", Qt::CaseInsensitive);
544         unsigned int fhgVersion = 0;
545
546         while(process.state() != QProcess::NotRunning)
547         {
548                 process.waitForReadyRead();
549                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
550                 {
551                         qWarning("FhgAacEnc process time out -> killing!");
552                         process.kill();
553                         process.waitForFinished(-1);
554                         for(int i = 0; i < 5; i++) LAMEXP_DELETE(fhgBin[i]);
555                         return;
556                 }
557                 while(process.bytesAvailable() > 0)
558                 {
559                         QString line = QString::fromUtf8(process.readLine().constData()).simplified();
560                         if(fhgAacEncSig.lastIndexIn(line) >= 0)
561                         {
562                                 bool ok = false;
563                                 unsigned int temp = fhgAacEncSig.cap(1).toUInt(&ok);
564                                 if(ok) fhgVersion = temp;
565                         }
566                 }
567         }
568
569         if(!(fhgVersion > 0))
570         {
571                 qWarning("FhgAacEnc version couldn't be determined -> FhgAacEnc support will be disabled!");
572                 for(int i = 0; i < 5; i++) LAMEXP_DELETE(fhgBin[i]);
573                 return;
574         }
575         else if(fhgVersion < lamexp_toolver_fhgaacenc())
576         {
577                 qWarning("FhgAacEnc version is too much outdated (%s) -> FhgAacEnc support will be disabled!", lamexp_version2string("????-??-??", fhgVersion, "N/A").toLatin1().constData());
578                 qWarning("Minimum required FhgAacEnc version currently is: %s\n", lamexp_version2string("????-??-??", lamexp_toolver_fhgaacenc(), "N/A").toLatin1().constData());
579                 for(int i = 0; i < 5; i++) LAMEXP_DELETE(fhgBin[i]);
580                 return;
581         }
582         
583         for(int i = 0; i < 5; i++)
584         {
585                 lamexp_register_tool(fhgFileInfo[i].fileName(), fhgBin[i], fhgVersion);
586         }
587 }
588
589 void InitializationThread::initQAac(void)
590 {
591         const QString appPath = QDir(QCoreApplication::applicationDirPath()).canonicalPath();
592
593         QFileInfo qaacFileInfo[2];
594         qaacFileInfo[0] = QFileInfo(QString("%1/qaac.exe").arg(appPath));
595         qaacFileInfo[1] = QFileInfo(QString("%1/libsoxrate.dll").arg(appPath));
596         
597         bool qaacFilesFound = true;
598         for(int i = 0; i < 2; i++)      { if(!qaacFileInfo[i].exists()) qaacFilesFound = false; }
599
600         //Lock the QAAC binaries
601         if(!qaacFilesFound)
602         {
603                 qDebug("QAAC binaries not found -> QAAC support will be disabled!\n");
604                 return;
605         }
606
607         qDebug("Found QAAC encoder:\n%s\n", qaacFileInfo[0].canonicalFilePath().toUtf8().constData());
608
609         LockedFile *qaacBin[2];
610         for(int i = 0; i < 2; i++) qaacBin[i] = NULL;
611
612         try
613         {
614                 for(int i = 0; i < 2; i++)
615                 {
616                         qaacBin[i] = new LockedFile(qaacFileInfo[i].canonicalFilePath());
617                 }
618         }
619         catch(...)
620         {
621                 for(int i = 0; i < 2; i++) LAMEXP_DELETE(qaacBin[i]);
622                 qWarning("Failed to get excluive lock to QAAC binary -> QAAC support will be disabled!");
623                 return;
624         }
625
626         QProcess process;
627         process.setProcessChannelMode(QProcess::MergedChannels);
628         process.setReadChannel(QProcess::StandardOutput);
629         process.start(qaacFileInfo[0].canonicalFilePath(), QStringList() << "--check");
630
631         if(!process.waitForStarted())
632         {
633                 qWarning("QAAC process failed to create!");
634                 qWarning("Error message: \"%s\"\n", process.errorString().toLatin1().constData());
635                 process.kill();
636                 process.waitForFinished(-1);
637                 for(int i = 0; i < 2; i++) LAMEXP_DELETE(qaacBin[i]);
638                 return;
639         }
640
641         QRegExp qaacEncSig("qaac (\\d)\\.(\\d)(\\d)", Qt::CaseInsensitive);
642         QRegExp coreEncSig("CoreAudioToolbox (\\d)\\.(\\d)\\.(\\d)\\.(\\d)", Qt::CaseInsensitive);
643         unsigned int qaacVersion = 0;
644         unsigned int coreVersion = 0;
645
646         while(process.state() != QProcess::NotRunning)
647         {
648                 process.waitForReadyRead();
649                 if(!process.bytesAvailable() && process.state() == QProcess::Running)
650                 {
651                         qWarning("QAAC process time out -> killing!");
652                         process.kill();
653                         process.waitForFinished(-1);
654                 for(int i = 0; i < 2; i++) LAMEXP_DELETE(qaacBin[i]);
655                         return;
656                 }
657                 while(process.bytesAvailable() > 0)
658                 {
659                         QString line = QString::fromUtf8(process.readLine().constData()).simplified();
660                         if(qaacEncSig.lastIndexIn(line) >= 0)
661                         {
662                                 unsigned int tmp[3] = {0, 0, 0};
663                                 bool ok[3] = {false, false, false};
664                                 tmp[0] = qaacEncSig.cap(1).toUInt(&ok[0]);
665                                 tmp[1] = qaacEncSig.cap(2).toUInt(&ok[1]);
666                                 tmp[2] = qaacEncSig.cap(3).toUInt(&ok[2]);
667                                 if(ok[0] && ok[1] && ok[2])
668                                 {
669                                         qaacVersion = (qBound(0U, tmp[0], 9U) * 100) + (qBound(0U, tmp[1], 9U) * 10) + qBound(0U, tmp[2], 9U);
670                                 }
671                         }
672                         if(coreEncSig.lastIndexIn(line) >= 0)
673                         {
674                                 unsigned int tmp[4] = {0, 0, 0, 0};
675                                 bool ok[4] = {false, false, false, false};
676                                 tmp[0] = coreEncSig.cap(1).toUInt(&ok[0]);
677                                 tmp[1] = coreEncSig.cap(2).toUInt(&ok[1]);
678                                 tmp[2] = coreEncSig.cap(3).toUInt(&ok[2]);
679                                 tmp[3] = coreEncSig.cap(4).toUInt(&ok[3]);
680                                 if(ok[0] && ok[1] && ok[2] && ok[3])
681                                 {
682                                         coreVersion = (qBound(0U, tmp[0], 9U) * 1000) + (qBound(0U, tmp[1], 9U) * 100) + (qBound(0U, tmp[2], 9U) * 10) + qBound(0U, tmp[3], 9U);
683                                 }
684                         }
685                 }
686         }
687
688         //qDebug("qaac %d, CoreAudioToolbox %d", qaacVersion, coreVersion);
689
690         if(!(qaacVersion > 0))
691         {
692                 qWarning("QAAC version couldn't be determined -> QAAC support will be disabled!");
693                 for(int i = 0; i < 2; i++) LAMEXP_DELETE(qaacBin[i]);
694                 return;
695         }
696         else if(qaacVersion < lamexp_toolver_qaacenc())
697         {
698                 qWarning("QAAC version is too much outdated (%s) -> QAAC support will be disabled!", lamexp_version2string("v?.??", qaacVersion, "N/A").toLatin1().constData());
699                 qWarning("Minimum required QAAC version currently is: %s.\n", lamexp_version2string("v?.??", lamexp_toolver_qaacenc(), "N/A").toLatin1().constData());
700                 for(int i = 0; i < 2; i++) LAMEXP_DELETE(qaacBin[i]);
701                 return;
702         }
703
704         if(!(coreVersion > 0))
705         {
706                 qWarning("CoreAudioToolbox version couldn't be determined -> QAAC support will be disabled!");
707                 for(int i = 0; i < 2; i++) LAMEXP_DELETE(qaacBin[i]);
708                 return;
709         }
710         else if(coreVersion < lamexp_toolver_coreaudio())
711         {
712                 qWarning("CoreAudioToolbox version is too much outdated (%s) -> QAAC support will be disabled!", lamexp_version2string("v?.?.?.?", coreVersion, "N/A").toLatin1().constData());
713                 qWarning("Minimum required CoreAudioToolbox version currently is: %s.\n", lamexp_version2string("v?.??", lamexp_toolver_coreaudio(), "N/A").toLatin1().constData());
714                 for(int i = 0; i < 2; i++) LAMEXP_DELETE(qaacBin[i]);
715                 return;
716         }
717
718         lamexp_register_tool(qaacFileInfo[0].fileName(), qaacBin[0], qaacVersion);
719         lamexp_register_tool(qaacFileInfo[1].fileName(), qaacBin[1], qaacVersion);
720 }
721
722 void InitializationThread::selfTest(void)
723 {
724         const unsigned int cpu[4] = {CPU_TYPE_X86_GEN, CPU_TYPE_X86_SSE, CPU_TYPE_X64_GEN, CPU_TYPE_X64_SSE};
725
726         for(size_t k = 0; k < 4; k++)
727         {
728                 qDebug("[TEST]");
729                 switch(cpu[k])
730                 {
731                         PRINT_CPU_TYPE(CPU_TYPE_X86_GEN); break;
732                         PRINT_CPU_TYPE(CPU_TYPE_X86_SSE); break;
733                         PRINT_CPU_TYPE(CPU_TYPE_X64_GEN); break;
734                         PRINT_CPU_TYPE(CPU_TYPE_X64_SSE); break;
735                         default: throw "CPU support undefined!";
736                 }
737                 int n = 0;
738                 for(int i = 0; i < INT_MAX; i++)
739                 {
740                         if(!g_lamexp_tools[i].pcName && !g_lamexp_tools[i].pcHash && !g_lamexp_tools[i].uiVersion)
741                         {
742                                 break;
743                         }
744                         if(g_lamexp_tools[i].uiCpuType & cpu[k])
745                         {
746                                 qDebug("%02i -> %s", ++n, g_lamexp_tools[i].pcName);
747                         }
748                 }
749                 if(n != 28)
750                 {
751                         qFatal("Tool count mismatch !!!");
752                 }
753                 qDebug("Done.\n");
754         }
755 }
756
757 ////////////////////////////////////////////////////////////
758 // EVENTS
759 ////////////////////////////////////////////////////////////
760
761 /*NONE*/