src/Thread_Initialization.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2016 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, but always including the *additional*
   9 // restrictions defined in the "License.txt" file.
  10 //
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU General Public License along
  17 // with this program; if not, write to the Free Software Foundation, Inc.,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20 // http://www.gnu.org/licenses/gpl-2.0.txt
  21 ///////////////////////////////////////////////////////////////////////////////
  22
  23 #include "Thread_Initialization.h"
  24
  25 //Internal
  26 #define LAMEXP_INC_TOOLS 1
  27 #include "Tools.h"
  28 #include "LockedFile.h"
  29 #include "FileHash.h"
  30 #include "Tool_Abstract.h"
  31
  32 //MUtils
  33 #include <MUtils/Global.h>
  34 #include <MUtils/OSSupport.h>
  35 #include <MUtils/Translation.h>
  36 #include <MUtils/Exception.h>
  37
  38 //Qt
  39 #include <QFileInfo>
  40 #include <QCoreApplication>
  41 #include <QProcess>
  42 #include <QMap>
  43 #include <QDir>
  44 #include <QResource>
  45 #include <QTextStream>
  46 #include <QRunnable>
  47 #include <QThreadPool>
  48 #include <QMutex>
  49 #include <QQueue>
  50 #include <QElapsedTimer>
  51 #include <QVector>
  52
  53 /* enable custom tools? */
  54 static const bool ENABLE_CUSTOM_TOOLS = true;
  55
  56 /* helper macros */
  57 #define PRINT_CPU_TYPE(X) case X: qDebug("Selected CPU is: " #X)
  58 #define MAKE_REGEXP(STR) (((STR) && ((STR)[0])) ? QRegExp((STR)) : QRegExp())
  59
  60 /* constants */
  61 static const double g_allowedExtractDelay = 12.0;
  62 static const size_t BUFF_SIZE = 512;
  63 static const size_t EXPECTED_TOOL_COUNT = 28;
  64
  65 /* number of CPU cores -> number of threads */
  66 static unsigned int cores2threads(const unsigned int cores)
  67 {
  68         static const size_t LUT_LEN = 4;
  69
  70         static const struct
  71         {
  72                 const unsigned int upperBound;
  73                 const double coeffs[4];
  74         }
  75         LUT[LUT_LEN] =
  76         {
  77                 {  4, { -0.052695810565,  0.158087431694, 4.982841530055, -1.088233151184 } },
  78                 {  8, {  0.042431693989, -0.983442622951, 9.548961748634, -7.176393442623 } },
  79                 { 12, { -0.006277322404,  0.185573770492, 0.196830601093, 17.762622950820 } },
  80                 { 32, {  0.000673497268, -0.064655737705, 3.199584699454,  5.751606557377 } }
  81         };
  82
  83         size_t index = 0;
  84         while((cores > LUT[index].upperBound) && (index < (LUT_LEN-1))) index++;
  85
  86         const double x = qBound(1.0, double(cores), double(LUT[LUT_LEN-1].upperBound));
  87         const double y = (LUT[index].coeffs[0] * pow(x, 3.0)) + (LUT[index].coeffs[1] * pow(x, 2.0)) + (LUT[index].coeffs[2] * x) + LUT[index].coeffs[3];
  88
  89         return qRound(abs(y));
  90 }
  91
  92 ////////////////////////////////////////////////////////////
  93 // BaseTask class
  94 ////////////////////////////////////////////////////////////
  95
  96 class BaseTask : public QRunnable
  97 {
  98 public:
  99         BaseTask(void)
 100         {
 101         }
 102
 103         ~BaseTask(void)
 104         {
 105         }
 106
 107         static void clearFlags(QMutexLocker &lock = QMutexLocker(&s_mutex))
 108         {
 109                 s_bExcept = false;
 110                 s_errMsg[0] = char(0);
 111         }
 112
 113         static bool getExcept(void)
 114         {
 115                 bool ret;
 116                 QMutexLocker lock(&s_mutex);
 117                 ret = s_bExcept;
 118                 return ret;
 119         }
 120
 121         static bool getErrMsg(char *buffer, const size_t buffSize)
 122         {
 123                 QMutexLocker lock(&s_mutex);
 124                 if(s_errMsg[0])
 125                 {
 126                         strncpy_s(buffer, BUFF_SIZE, s_errMsg, _TRUNCATE);
 127                         return true;
 128                 }
 129                 return false;
 130         }
 131
 132 protected:
 133         virtual void taskMain(void) = 0;
 134
 135         void run(void)
 136         {
 137                 try
 138                 {
 139                         if(!getExcept()) taskMain();
 140                 }
 141                 catch(const std::exception &e)
 142                 {
 143                         QMutexLocker lock(&s_mutex);
 144                         if(!s_bExcept)
 145                         {
 146                                 s_bExcept = true;
 147                                 strncpy_s(s_errMsg, BUFF_SIZE, e.what(), _TRUNCATE);
 148                         }
 149                         lock.unlock();
 150                         qWarning("OptionalInitTask exception error:\n%s\n\n", e.what());
 151                 }
 152                 catch(...)
 153                 {
 154                         QMutexLocker lock(&s_mutex);
 155                         if(!s_bExcept)
 156                         {
 157                                 s_bExcept = true;
 158                                 strncpy_s(s_errMsg, BUFF_SIZE, "Unknown exception error!", _TRUNCATE);
 159                         }
 160                         lock.unlock();
 161                         qWarning("OptionalInitTask encountered an unknown exception!");
 162                 }
 163         }
 164
 165         static volatile bool s_bExcept;
 166         static QMutex s_mutex;
 167         static char s_errMsg[BUFF_SIZE];
 168 };
 169
 170 QMutex BaseTask::s_mutex;
 171 char BaseTask::s_errMsg[BUFF_SIZE] = {'\0'};
 172 volatile bool BaseTask::s_bExcept = false;
 173
 174 ////////////////////////////////////////////////////////////
 175 // ExtractorTask class
 176 ////////////////////////////////////////////////////////////
 177
 178 class ExtractorTask : public BaseTask
 179 {
 180 public:
 181         ExtractorTask(QResource *const toolResource, const QDir &appDir, const QString &toolName, const QByteArray &toolHash, const unsigned int toolVersion, const QString &toolTag)
 182         :
 183                 m_appDir(appDir),
 184                 m_tempPath(MUtils::temp_folder()),
 185                 m_toolName(toolName),
 186                 m_toolHash(toolHash),
 187                 m_toolVersion(toolVersion),
 188                 m_toolTag(toolTag),
 189                 m_toolResource(toolResource)
 190         {
 191                 /* Nothing to do */
 192         }
 193
 194         ~ExtractorTask(void)
 195         {
 196         }
 197
 198         static bool getCustom(void)
 199         {
 200                 bool ret;
 201                 QMutexLocker lock(&s_mutex);
 202                 ret = s_bCustom;
 203                 return ret;
 204         }
 205
 206         static void clearFlags(void)
 207         {
 208                 QMutexLocker lock(&s_mutex);
 209                 s_bCustom = false;
 210                 BaseTask::clearFlags(lock);
 211         }
 212
 213 protected:
 214         void taskMain(void)
 215         {
 216                 QScopedPointer<LockedFile> lockedFile;
 217                 unsigned int version = m_toolVersion;
 218
 219                 const QFileInfo toolFileInfo(m_toolName);
 220                 const QString   toolShrtName = QString("%1.%2").arg(toolFileInfo.baseName().toLower(), toolFileInfo.suffix().toLower());
 221
 222                 //Try to load a "custom" tool first
 223                 if(ENABLE_CUSTOM_TOOLS)
 224                 {
 225                         const QFileInfo customTool(QString("%1/tools/%2/%3").arg(m_appDir.canonicalPath(), QString::number(lamexp_version_build()), toolShrtName));
 226                         if(customTool.exists() && customTool.isFile())
 227                         {
 228                                 qDebug("Setting up file: %s <- %s", toolShrtName.toLatin1().constData(), m_appDir.relativeFilePath(customTool.canonicalFilePath()).toLatin1().constData());
 229                                 try
 230                                 {
 231                                         lockedFile.reset(new LockedFile(customTool.canonicalFilePath()));
 232                                         version = UINT_MAX; s_bCustom = true;
 233                                 }
 234                                 catch(std::runtime_error&)
 235                                 {
 236                                         lockedFile.reset();
 237                                 }
 238                         }
 239                 }
 240
 241                 //Try to load the tool from the "cache" next
 242                 if(lockedFile.isNull())
 243                 {
 244                         const QFileInfo chachedTool(QString("%1/cache/%2").arg(m_appDir.canonicalPath(), toolFileInfo.fileName()));
 245                         if(chachedTool.exists() && chachedTool.isFile())
 246                         {
 247                                 qDebug("Validating file: %s <- %s", toolShrtName.toLatin1().constData(), m_appDir.relativeFilePath(chachedTool.canonicalFilePath()).toLatin1().constData());
 248                                 try
 249                                 {
 250                                         lockedFile.reset(new LockedFile(chachedTool.canonicalFilePath(), m_toolHash));
 251                                 }
 252                                 catch(std::runtime_error&)
 253                                 {
 254                                         lockedFile.reset();
 255                                 }
 256                         }
 257                 }
 258
 259                 //If still not initialized, extract tool now!
 260                 if(lockedFile.isNull())
 261                 {
 262                         qDebug("Extracting file: %s -> %s", m_toolName.toLatin1().constData(), toolShrtName.toLatin1().constData());
 263                         lockedFile.reset(new LockedFile(m_toolResource.data(), QString("%1/lxp_%2").arg(m_tempPath, toolShrtName), m_toolHash));
 264                 }
 265
 266                 //Register tool
 267                 lamexp_tools_register(toolShrtName, lockedFile.take(), version, m_toolTag);
 268         }
 269
 270 private:
 271         static volatile bool      s_bCustom;
 272         QScopedPointer<QResource> m_toolResource;
 273         const QDir                m_appDir;
 274         const QString             m_tempPath;
 275         const QString             m_toolName;
 276         const QByteArray          m_toolHash;
 277         const unsigned int        m_toolVersion;
 278         const QString             m_toolTag;
 279 };
 280
 281 volatile bool ExtractorTask::s_bCustom = false;
 282
 283 ////////////////////////////////////////////////////////////
 284 // InitAacEncTask class
 285 ////////////////////////////////////////////////////////////
 286
 287 class InitAacEncTask : public BaseTask
 288 {
 289 public:
 290         InitAacEncTask(const aac_encoder_t *const encoder_info)
 291         :
 292                 m_encoder_info(encoder_info)
 293         {
 294         }
 295
 296         ~InitAacEncTask(void)
 297         {
 298         }
 299
 300 protected:
 301         void taskMain(void)
 302         {
 303                 initAacEncImpl(m_encoder_info->toolName, m_encoder_info->fileNames, m_encoder_info->toolMinVersion, m_encoder_info->verDigits, m_encoder_info->verShift, m_encoder_info->verStr, MAKE_REGEXP(m_encoder_info->regExpVer), MAKE_REGEXP(m_encoder_info->regExpSig));
 304         }
 305
 306         static void initAacEncImpl(const char *const toolName, const char *const fileNames[], const quint32 &toolMinVersion, const quint32 &verDigits, const quint32 &verShift, const char *const verStr, QRegExp &regExpVer, QRegExp &regExpSig = QRegExp());
 307
 308 private:
 309         const aac_encoder_t *const m_encoder_info;
 310 };
 311
 312 ////////////////////////////////////////////////////////////
 313 // Constructor
 314 ////////////////////////////////////////////////////////////
 315
 316 InitializationThread::InitializationThread(const MUtils::CPUFetaures::cpu_info_t &cpuFeatures)
 317 :
 318         m_bSuccess(false),
 319         m_slowIndicator(false)
 320 {
 321
 322         memcpy(&m_cpuFeatures, &cpuFeatures, sizeof(MUtils::CPUFetaures::cpu_info_t));
 323 }
 324
 325 ////////////////////////////////////////////////////////////
 326 // Thread Main
 327 ////////////////////////////////////////////////////////////
 328
 329 void InitializationThread::run(void)
 330 {
 331         try
 332         {
 333                 doInit();
 334         }
 335         catch(const std::exception &error)
 336         {
 337                 MUTILS_PRINT_ERROR("\nGURU MEDITATION !!!\n\nException error:\n%s\n", error.what());
 338                 MUtils::OS::fatal_exit(L"Unhandeled C++ exception error, application will exit!");
 339         }
 340         catch(...)
 341         {
 342                 MUTILS_PRINT_ERROR("\nGURU MEDITATION !!!\n\nUnknown exception error!\n");
 343                 MUtils::OS::fatal_exit(L"Unhandeled C++ exception error, application will exit!");
 344         }
 345 }
 346
 347 double InitializationThread::doInit(const size_t threadCount)
 348 {
 349         m_bSuccess = false;
 350         delay();
 351
 352         //CPU type selection
 353         unsigned int cpuSupport = 0;
 354         if((m_cpuFeatures.features & MUtils::CPUFetaures::FLAG_SSE) && (m_cpuFeatures.features & MUtils::CPUFetaures::FLAG_SSE2) && m_cpuFeatures.intel)
 355         {
 356                 cpuSupport = m_cpuFeatures.x64 ? CPU_TYPE_X64_SSE : CPU_TYPE_X86_SSE;
 357         }
 358         else
 359         {
 360                 cpuSupport = m_cpuFeatures.x64 ? CPU_TYPE_X64_GEN : CPU_TYPE_X86_GEN;
 361         }
 362
 363         //Hack to disable x64 on Wine, as x64 binaries won't run under Wine (tested with Wine 1.4 under Ubuntu 12.04 x64)
 364         if(cpuSupport & CPU_TYPE_X64_ALL)
 365         {
 366                 if(MUtils::OS::running_on_wine())
 367                 {
 368                         qWarning("Running under Wine on a 64-Bit system. Going to disable all x64 support!\n");
 369                         cpuSupport = (cpuSupport == CPU_TYPE_X64_SSE) ? CPU_TYPE_X86_SSE : CPU_TYPE_X86_GEN;
 370                 }
 371         }
 372
 373         //Print selected CPU type
 374         switch(cpuSupport)
 375         {
 376                 PRINT_CPU_TYPE(CPU_TYPE_X86_GEN); break;
 377                 PRINT_CPU_TYPE(CPU_TYPE_X86_SSE); break;
 378                 PRINT_CPU_TYPE(CPU_TYPE_X64_GEN); break;
 379                 PRINT_CPU_TYPE(CPU_TYPE_X64_SSE); break;
 380                 default: MUTILS_THROW("CPU support undefined!");
 381         }
 382
 383         //Allocate queues
 384         QQueue<QString> queueToolName;
 385         QQueue<QString> queueChecksum;
 386         QQueue<QString> queueVersInfo;
 387         QQueue<unsigned int> queueVersions;
 388         QQueue<unsigned int> queueCpuTypes;
 389
 390         //Init properties
 391         for(int i = 0; true; i++)
 392         {
 393                 if(!(g_lamexp_tools[i].pcName || g_lamexp_tools[i].pcHash  || g_lamexp_tools[i].uiVersion))
 394                 {
 395                         break;
 396                 }
 397                 else if(g_lamexp_tools[i].pcName && g_lamexp_tools[i].pcHash && g_lamexp_tools[i].uiVersion)
 398                 {
 399                         queueToolName.enqueue(QString::fromLatin1(g_lamexp_tools[i].pcName));
 400                         queueChecksum.enqueue(QString::fromLatin1(g_lamexp_tools[i].pcHash));
 401                         queueVersInfo.enqueue(QString::fromLatin1(g_lamexp_tools[i].pcVersTag));
 402                         queueCpuTypes.enqueue(g_lamexp_tools[i].uiCpuType);
 403                         queueVersions.enqueue(g_lamexp_tools[i].uiVersion);
 404                 }
 405                 else
 406                 {
 407                         qFatal("Inconsistent checksum data detected. Take care!");
 408                 }
 409         }
 410
 411         QDir appDir = QDir(QCoreApplication::applicationDirPath()).canonicalPath();
 412
 413         QScopedPointer<QThreadPool> pool(new QThreadPool());
 414         pool->setMaxThreadCount((threadCount > 0) ? threadCount : qBound(2U, cores2threads(m_cpuFeatures.count), EXPECTED_TOOL_COUNT));
 415         /* qWarning("Using %u threads for extraction.", pool->maxThreadCount()); */
 416
 417         FileHash::selfTest();
 418         ExtractorTask::clearFlags();
 419
 420         //Start the timer
 421         QElapsedTimer timeExtractStart;
 422         timeExtractStart.start();
 423
 424         //Extract all files
 425         while(!(queueToolName.isEmpty() || queueChecksum.isEmpty() || queueVersInfo.isEmpty() || queueCpuTypes.isEmpty() || queueVersions.isEmpty()))
 426         {
 427                 const QString toolName = queueToolName.dequeue();
 428                 const QString checksum = queueChecksum.dequeue();
 429                 const QString versInfo = queueVersInfo.dequeue();
 430                 const unsigned int cpuType = queueCpuTypes.dequeue();
 431                 const unsigned int version = queueVersions.dequeue();
 432
 433                 const QByteArray toolHash(checksum.toLatin1());
 434                 if(toolHash.size() != 96)
 435                 {
 436                         qFatal("The checksum for \"%s\" has an invalid size!", MUTILS_UTF8(toolName));
 437                         return -1.0;
 438                 }
 439
 440                 QScopedPointer<QResource> resource(new QResource(QString(":/tools/%1").arg(toolName)));
 441                 if(!(resource->isValid() && resource->data()))
 442                 {
 443                         qFatal("The resource for \"%s\" could not be found!", MUTILS_UTF8(toolName));
 444                         return -1.0;
 445                 }
 446
 447                 if(cpuType & cpuSupport)
 448                 {
 449                         pool->start(new ExtractorTask(resource.take(), appDir, toolName, toolHash, version, versInfo));
 450                         continue;
 451                 }
 452         }
 453
 454         //Sanity Check
 455         if(!(queueToolName.isEmpty() && queueChecksum.isEmpty() && queueVersInfo.isEmpty() && queueCpuTypes.isEmpty() && queueVersions.isEmpty()))
 456         {
 457                 qFatal("Checksum queues *not* empty fater verification completed. Take care!");
 458         }
 459
 460         //Wait for extrator threads to finish
 461         pool->waitForDone();
 462
 463         //Performance measure
 464         const double delayExtract = double(timeExtractStart.elapsed()) / 1000.0;
 465         timeExtractStart.invalidate();
 466
 467         //Make sure all files were extracted correctly
 468         if(ExtractorTask::getExcept())
 469         {
 470                 char errorMsg[BUFF_SIZE];
 471                 if(ExtractorTask::getErrMsg(errorMsg, BUFF_SIZE))
 472                 {
 473                         qFatal("At least one of the required tools could not be initialized:\n%s", errorMsg);
 474                         return -1.0;
 475                 }
 476                 qFatal("At least one of the required tools could not be initialized!");
 477                 return -1.0;
 478         }
 479
 480         qDebug("All extracted.\n");
 481
 482         //Using any custom tools?
 483         if(ExtractorTask::getCustom())
 484         {
 485                 qWarning("Warning: Using custom tools, you might encounter unexpected problems!\n");
 486         }
 487
 488         //Check delay
 489         if(delayExtract > g_allowedExtractDelay)
 490         {
 491                 m_slowIndicator = true;
 492                 qWarning("Extracting tools took %.3f seconds -> probably slow realtime virus scanner.", delayExtract);
 493                 qWarning("Please report performance problems to your anti-virus developer !!!\n");
 494         }
 495         else
 496         {
 497                 qDebug("Extracting the tools took %.3f seconds (OK).\n", delayExtract);
 498         }
 499
 500         //Register all translations
 501         initTranslations();
 502
 503         //Look for AAC encoders
 504         InitAacEncTask::clearFlags();
 505         for(size_t i = 0; g_lamexp_aacenc[i].toolName; i++)
 506         {
 507                 pool->start(new InitAacEncTask(&(g_lamexp_aacenc[i])));
 508         }
 509         pool->waitForDone();
 510
 511         //Make sure initialization finished correctly
 512         if(InitAacEncTask::getExcept())
 513         {
 514                 char errorMsg[BUFF_SIZE];
 515                 if(InitAacEncTask::getErrMsg(errorMsg, BUFF_SIZE))
 516                 {
 517                         qFatal("At least one optional component failed to initialize:\n%s", errorMsg);
 518                         return -1.0;
 519                 }
 520                 qFatal("At least one optional component failed to initialize!");
 521                 return -1.0;
 522         }
 523
 524         m_bSuccess = true;
 525         delay();
 526
 527         return delayExtract;
 528 }
 529
 530 ////////////////////////////////////////////////////////////
 531 // INTERNAL FUNCTIONS
 532 ////////////////////////////////////////////////////////////
 533
 534 void InitializationThread::delay(void)
 535 {
 536         MUtils::OS::sleep_ms(333);
 537 }
 538
 539 ////////////////////////////////////////////////////////////
 540 // Translation Support
 541 ////////////////////////////////////////////////////////////
 542
 543 void InitializationThread::initTranslations(void)
 544 {
 545         //Search for language files
 546         const QDir qmDirectory(":/localization");
 547         const QStringList qmFiles = qmDirectory.entryList(QStringList() << "LameXP_??.qm", QDir::Files, QDir::Name);
 548
 549         //Make sure we found at least one translation
 550         if(qmFiles.count() < 1)
 551         {
 552                 qFatal("Could not find any translation files!");
 553                 return;
 554         }
 555
 556         //Initialize variables
 557         const QString langResTemplate(":/localization/%1.txt");
 558         QRegExp langIdExp("^LameXP_(\\w\\w)\\.qm$", Qt::CaseInsensitive);
 559
 560         //Add all available translations
 561         for(QStringList::ConstIterator iter = qmFiles.constBegin(); iter != qmFiles.constEnd(); iter++)
 562         {
 563                 const QString langFile = qmDirectory.absoluteFilePath(*iter);
 564                 QString langId, langName;
 565                 unsigned int systemId = 0, country = 0;
 566
 567                 if(QFileInfo(langFile).isFile() && (langIdExp.indexIn(*iter) >= 0))
 568                 {
 569                         langId = langIdExp.cap(1).toLower();
 570                         QScopedPointer<QResource> langRes(new QResource(langResTemplate.arg(*iter)));
 571                         if(langRes->isValid() && langRes->size() > 0)
 572                         {
 573                                 QByteArray data = QByteArray::fromRawData(reinterpret_cast<const char*>(langRes->data()), langRes->size());
 574                                 QTextStream stream(&data, QIODevice::ReadOnly);
 575                                 stream.setAutoDetectUnicode(false); stream.setCodec("UTF-8");
 576
 577                                 while(!(stream.atEnd() || (stream.status() != QTextStream::Ok)))
 578                                 {
 579                                         QStringList langInfo = stream.readLine().simplified().split(",", QString::SkipEmptyParts);
 580                                         if(langInfo.count() >= 3)
 581                                         {
 582                                                 systemId = langInfo.at(0).trimmed().toUInt();
 583                                                 country  = langInfo.at(1).trimmed().toUInt();
 584                                                 langName = langInfo.at(2).trimmed();
 585                                                 break;
 586                                         }
 587                                 }
 588                         }
 589                 }
 590
 591                 if(!(langId.isEmpty() || langName.isEmpty() || (systemId == 0)))
 592                 {
 593                         if(MUtils::Translation::insert(langId, langFile, langName, systemId, country))
 594                         {
 595                                 qDebug("Registering translation: %s = %s (%u) [%u]", MUTILS_UTF8(*iter), MUTILS_UTF8(langName), systemId, country);
 596                         }
 597                         else
 598                         {
 599                                 qWarning("Failed to register: %s", langFile.toLatin1().constData());
 600                         }
 601                 }
 602         }
 603
 604         qDebug("All registered.\n");
 605 }
 606
 607 ////////////////////////////////////////////////////////////
 608 // AAC Encoder Detection
 609 ////////////////////////////////////////////////////////////
 610
 611 void InitAacEncTask::initAacEncImpl(const char *const toolName, const char *const fileNames[], const quint32 &toolMinVersion, const quint32 &verDigits, const quint32 &verShift, const char *const verStr, QRegExp &regExpVer, QRegExp &regExpSig)
 612 {
 613         static const size_t MAX_FILES = 8;
 614         const QString appPath = QDir(QCoreApplication::applicationDirPath()).canonicalPath();
 615
 616         QFileInfoList fileInfo;
 617         for(size_t i = 0; fileNames[i] && (fileInfo.count() < MAX_FILES); i++)
 618         {
 619                 fileInfo.append(QFileInfo(QString("%1/%2").arg(appPath, QString::fromLatin1(fileNames[i]))));
 620         }
 621
 622         for(QFileInfoList::ConstIterator iter = fileInfo.constBegin(); iter != fileInfo.constEnd(); iter++)
 623         {
 624                 if(!(iter->exists() && iter->isFile()))
 625                 {
 626                         qDebug("%s encoder binaries not found -> Encoding support will be disabled!\n", toolName);
 627                         return;
 628                 }
 629                 if((iter->suffix().compare("exe", Qt::CaseInsensitive) == 0) && (!MUtils::OS::is_executable_file(iter->canonicalFilePath())))
 630                 {
 631                         qDebug("%s executable is invalid -> %s support will be disabled!\n", MUTILS_UTF8(iter->fileName()), toolName);
 632                         return;
 633                 }
 634         }
 635
 636         qDebug("Found %s encoder binary:\n%s\n", toolName, MUTILS_UTF8(fileInfo.first().canonicalFilePath()));
 637
 638         //Lock the encoder binaries
 639         QScopedPointer<LockedFile> binaries[MAX_FILES];
 640         try
 641         {
 642                 size_t index = 0;
 643                 for(QFileInfoList::ConstIterator iter = fileInfo.constBegin(); iter != fileInfo.constEnd(); iter++)
 644                 {
 645                         binaries[index++].reset(new LockedFile(iter->canonicalFilePath()));
 646                 }
 647         }
 648         catch(...)
 649         {
 650                 qWarning("Failed to get excluive lock to encoder binary -> %s support will be disabled!", toolName);
 651                 return;
 652         }
 653
 654         QProcess process;
 655         MUtils::init_process(process, fileInfo.first().absolutePath());
 656
 657         process.start(fileInfo.first().canonicalFilePath(), QStringList() << "-help");
 658
 659         if(!process.waitForStarted())
 660         {
 661                 qWarning("%s process failed to create!", toolName);
 662                 qWarning("Error message: \"%s\"\n", process.errorString().toLatin1().constData());
 663                 process.kill();
 664                 process.waitForFinished(-1);
 665                 return;
 666         }
 667
 668         quint32 toolVersion = 0;
 669         bool sigFound = regExpSig.isEmpty() ? true : false;
 670
 671         while(process.state() != QProcess::NotRunning)
 672         {
 673                 if(!process.waitForReadyRead())
 674                 {
 675                         if(process.state() == QProcess::Running)
 676                         {
 677                                 qWarning("%s process time out -> killing!", toolName);
 678                                 process.kill();
 679                                 process.waitForFinished(-1);
 680                                 return;
 681                         }
 682                 }
 683                 while(process.canReadLine())
 684                 {
 685                         QString line = QString::fromUtf8(process.readLine().constData()).simplified();
 686                         if((!sigFound) && regExpSig.lastIndexIn(line) >= 0)
 687                         {
 688                                 sigFound = true;
 689                                 continue;
 690                         }
 691                         if(sigFound && (regExpVer.lastIndexIn(line) >= 0))
 692                         {
 693                                 quint32 tmp[8];
 694                                 if(MUtils::regexp_parse_uint32(regExpVer, tmp, qMin(verDigits, 8U)))
 695                                 {
 696                                         toolVersion = 0;
 697                                         for(quint32 i = 0; i < verDigits; i++)
 698                                         {
 699                                                 toolVersion = (toolVersion * verShift) + qBound(0U, tmp[i], (verShift - 1));
 700                                         }
 701                                 }
 702                         }
 703                 }
 704         }
 705
 706         if(toolVersion <= 0)
 707         {
 708                 qWarning("%s version could not be determined -> Encoding support will be disabled!", toolName);
 709                 return;
 710         }
 711         else if(toolVersion < toolMinVersion)
 712         {
 713                 qWarning("%s version is too much outdated (%s) -> Encoding support will be disabled!", toolName, MUTILS_UTF8(lamexp_version2string(verStr, toolVersion,    "N/A")));
 714                 qWarning("Minimum required %s version currently is: %s\n",                             toolName, MUTILS_UTF8(lamexp_version2string(verStr, toolMinVersion, "N/A")));
 715                 return;
 716         }
 717
 718         qDebug("Enabled %s encoder %s.\n", toolName, MUTILS_UTF8(lamexp_version2string(verStr, toolVersion, "N/A")));
 719
 720         size_t index = 0;
 721         for(QFileInfoList::ConstIterator iter = fileInfo.constBegin(); iter != fileInfo.constEnd(); iter++)
 722         {
 723                 lamexp_tools_register(iter->fileName(), binaries[index++].take(), toolVersion);
 724         }
 725 }
 726
 727 ////////////////////////////////////////////////////////////
 728 // Self-Test Function
 729 ////////////////////////////////////////////////////////////
 730
 731 void InitializationThread::selfTest(void)
 732 {
 733         const unsigned int cpu[4] = {CPU_TYPE_X86_GEN, CPU_TYPE_X86_SSE, CPU_TYPE_X64_GEN, CPU_TYPE_X64_SSE};
 734
 735         FileHash::selfTest();
 736
 737         for(size_t k = 0; k < 4; k++)
 738         {
 739                 qDebug("[TEST]");
 740                 switch(cpu[k])
 741                 {
 742                         PRINT_CPU_TYPE(CPU_TYPE_X86_GEN); break;
 743                         PRINT_CPU_TYPE(CPU_TYPE_X86_SSE); break;
 744                         PRINT_CPU_TYPE(CPU_TYPE_X64_GEN); break;
 745                         PRINT_CPU_TYPE(CPU_TYPE_X64_SSE); break;
 746                         default: MUTILS_THROW("CPU support undefined!");
 747                 }
 748                 unsigned int n = 0;
 749                 for(int i = 0; true; i++)
 750                 {
 751                         if(!(g_lamexp_tools[i].pcName || g_lamexp_tools[i].pcHash  || g_lamexp_tools[i].uiVersion))
 752                         {
 753                                 break;
 754                         }
 755                         else if(g_lamexp_tools[i].pcName && g_lamexp_tools[i].pcHash && g_lamexp_tools[i].uiVersion)
 756                         {
 757                                 const QString toolName = QString::fromLatin1(g_lamexp_tools[i].pcName);
 758                                 const QByteArray expectedHash = QByteArray(g_lamexp_tools[i].pcHash);
 759                                 if(g_lamexp_tools[i].uiCpuType & cpu[k])
 760                                 {
 761                                         qDebug("%02i -> %s", ++n, MUTILS_UTF8(toolName));
 762                                         QFile resource(QString(":/tools/%1").arg(toolName));
 763                                         if(!resource.open(QIODevice::ReadOnly))
 764                                         {
 765                                                 qFatal("The resource for \"%s\" could not be opened!", MUTILS_UTF8(toolName));
 766                                                 break;
 767                                         }
 768                                         QByteArray hash = FileHash::computeHash(resource);
 769                                         if(hash.isNull() || _stricmp(hash.constData(), expectedHash.constData()))
 770                                         {
 771                                                 qFatal("Hash check for tool \"%s\" has failed!", MUTILS_UTF8(toolName));
 772                                                 break;
 773                                         }
 774                                         resource.close();
 775                                 }
 776                         }
 777                         else
 778                         {
 779                                 qFatal("Inconsistent checksum data detected. Take care!");
 780                         }
 781                 }
 782                 if(n != EXPECTED_TOOL_COUNT)
 783                 {
 784                         qFatal("Tool count mismatch for CPU type %u !!!", cpu[k]);
 785                 }
 786                 qDebug("Done.\n");
 787         }
 788 }
 789
 790 ////////////////////////////////////////////////////////////
 791 // EVENTS
 792 ////////////////////////////////////////////////////////////
 793
 794 /*NONE*/