src/Thread_Initialization.cpp

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