src/Thread_Initialization.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2011 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 "Global.h"
  25 #include "LockedFile.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
  36 #include <Windows.h>
  37
  38 ////////////////////////////////////////////////////////////
  39 // TOOLS
  40 ////////////////////////////////////////////////////////////
  41
  42 static const struct
  43 {
  44         char *pcHash;
  45         char *pcName;
  46         unsigned int uiVersion;
  47 }
  48 g_lamexp_tools[] =
  49 {
  50         {"3b41f85dde8d4a5a0f4cd5f461099d0db24610ba", "alac.exe", UINT_MAX},
  51         {"fb74ac8b73ad8cba2c3b4e6e61f23401d630dc22", "elevator.exe", UINT_MAX},
  52         {"3c647950bccfcc75d0746c0772e7115684be4dc5", "faad.exe", UINT_MAX},
  53         {"d33cd86f04bd4067e244d2804466583c7b90a4e2", "flac.exe", 121},
  54         {"9328a50e89b54ec065637496d9681a7e3eebf915", "gpgv.exe", 1411},
  55         {"d837bf6ee4dab557d8b02d46c75a24e58980fffa", "gpgv.gpg", UINT_MAX},
  56         {"143fc001a2f6c56fe1b9e6f8a2eb2b53b9e1e504", "lame.exe", 39910},
  57         {"a4e929cfaa42fa2e61a3d0c6434c77a06d45aef3", "mac.exe", 406},
  58         {"ec235c6404caa31e7975e9bbf9ba59599258ae1d", "mediainfo_i386.exe", 741},
  59         {"72d7efdcafc2dee3b534f27ab6e01916d02ba470", "mediainfo_x64.exe", 741},
  60         {"55c293a80475f7aeccf449ac9487a4626e5139cb", "mpcdec.exe", UINT_MAX},
  61         {"6b6913a54cac08b22d5b468aaed83550fc9ae5b4", "mpg123.exe", 1131},
  62         {"8dd7138714c3bcb39f5a3213413addba13d06f1e", "oggdec.exe", UINT_MAX},
  63         {"ecd15abe103184aca96e406f5f1c82c6fb2e665d", "oggenc2_i386.exe", 287},
  64         {"ffe0fbd73352396dc3752ac9d484dbfc754a226d", "oggenc2_sse2.exe", 287},
  65         {"a8c50872e544a55495a824426e9378984f2ae01d", "oggenc2_x64.exe", 287},
  66         {"cd95369051f96b9ca3a997658771c5ea52bc874d", "selfdelete.exe", UINT_MAX},
  67         {"ffeaa70bd6321185eafcb067ab2dc441650038bf", "shorten.exe", UINT_MAX},
  68         {"2d08c3586f9cf99f2e4c89ac54eeb595f63aef61", "sox.exe", 1431},
  69         {"346ce516281c97e92e1b8957ddeca52edcf2d056", "speexdec.exe", UINT_MAX},
  70         {"8a74b767cfe88bf88c068fdae0de02d65589d25e", "takc.exe", UINT_MAX},
  71         {"d6e0de1e7a2d9dee10d06ae0b6b4f93b63205920", "ttaenc.exe", 341},
  72         {"8c842eef65248b46fa6cb9a9e5714f575672d999", "valdec.exe", 31},
  73         {"8159f4e824b3e343ece95ba6dbb5e16da9c4866e", "volumax.exe", UINT_MAX},
  74         {"62e2805d1b2eb2a4d86a5ca6e6ea58010d05d2a7", "wget.exe", UINT_MAX},
  75         {"a7e8aad52213e339ad985829722f35eab62be182", "wupdate.exe", UINT_MAX},
  76         {"b7d14b3540d24df13119a55d97623a61412de6e3", "wvunpack.exe", 4601},
  77         {NULL, NULL, NULL}
  78 };
  79
  80 ////////////////////////////////////////////////////////////
  81 // Constructor
  82 ////////////////////////////////////////////////////////////
  83
  84 InitializationThread::InitializationThread(void)
  85 {
  86         m_bSuccess = false;
  87 }
  88
  89 ////////////////////////////////////////////////////////////
  90 // Thread Main
  91 ////////////////////////////////////////////////////////////
  92
  93 void InitializationThread::run()
  94 {
  95         m_bSuccess = false;
  96         delay();
  97
  98         QMap<QString, QString> checksum;
  99         QMap<QString, unsigned int> version;
 100
 101         //Init checksums
 102         for(int i = 0; i < INT_MAX; i++)
 103         {
 104                 if(!g_lamexp_tools[i].pcName && !g_lamexp_tools[i].pcHash && !g_lamexp_tools[i].uiVersion)
 105                 {
 106                         break;
 107                 }
 108                 else if(g_lamexp_tools[i].pcName && g_lamexp_tools[i].pcHash && g_lamexp_tools[i].uiVersion)
 109                 {
 110                         const QString currentTool = QString::fromLatin1(g_lamexp_tools[i].pcName);
 111                         checksum.insert(currentTool, QString::fromLatin1(g_lamexp_tools[i].pcHash));
 112                         version.insert(currentTool, g_lamexp_tools[i].uiVersion);
 113                 }
 114                 else
 115                 {
 116                         qFatal("Inconsistent checksum data detected. Take care!");
 117                 }
 118         }
 119
 120         QDir toolsDir(":/tools/");
 121         QList<QFileInfo> toolsList = toolsDir.entryInfoList(QStringList("*.*"), QDir::Files, QDir::Name);
 122
 123         QTime timer;
 124         timer.start();
 125
 126         //Extract all files
 127         for(int i = 0; i < toolsList.count(); i++)
 128         {
 129                 try
 130                 {
 131                         QString toolName = toolsList.at(i).fileName().toLower();
 132                         qDebug("Extracting file: %s", toolName.toLatin1().constData());
 133                         QByteArray toolHash = checksum.take(toolName).toLatin1();
 134                         unsigned int toolVersion = version.take(toolName);
 135                         if(toolHash.size() != 40)
 136                         {
 137                                 throw "The required checksum is missing, take care!";
 138                         }
 139                         LockedFile *lockedFile = new LockedFile(QString(":/tools/%1").arg(toolName), QString(lamexp_temp_folder()).append(QString("/tool_%1").arg(toolName)), toolHash);
 140                         lamexp_register_tool(toolName, lockedFile, toolVersion);
 141                 }
 142                 catch(char *errorMsg)
 143                 {
 144                         qFatal("At least one required tool could not be extracted:\n%s", errorMsg);
 145                         return;
 146                 }
 147         }
 148
 149         if(!checksum.isEmpty())
 150         {
 151                 qFatal("At least one required tool could not be found:\n%s", toolsDir.filePath(checksum.keys().first()).toLatin1().constData());
 152                 return;
 153         }
 154
 155         qDebug("All extracted.\n");
 156
 157         //Check delay
 158         double delayExtract = static_cast<double>(timer.elapsed()) / 1000.0;
 159         if(delayExtract > 8.0)
 160         {
 161                 qWarning("Extracting tools took %.3f seconds -> probably slow realtime virus scanner.", delayExtract);
 162                 qWarning("Please report performance problems to your anti-virus developer !!!\n");
 163         }
 164
 165         //Register all translations
 166         initTranslations();
 167
 168         //Look for Nero encoder
 169         initNeroAac();
 170
 171         //Look for WMA File decoder
 172         initWmaDec();
 173
 174         delay();
 175         m_bSuccess = true;
 176 }
 177
 178 ////////////////////////////////////////////////////////////
 179 // PUBLIC FUNCTIONS
 180 ////////////////////////////////////////////////////////////
 181
 182 void InitializationThread::delay(void)
 183 {
 184         const char *temp = "|/-\\";
 185         printf("Thread is doing something important... ?\b", temp[4]);
 186
 187         for(int i = 0; i < 20; i++)
 188         {
 189                 printf("%c\b", temp[i%4]);
 190                 msleep(25);
 191         }
 192
 193         printf("Done\n\n");
 194 }
 195
 196 void InitializationThread::initTranslations(void)
 197 {
 198         //Search for language files
 199         QStringList qmFiles = QDir(":/localization").entryList(QStringList() << "LameXP_??.qm", QDir::Files, QDir::Name);
 200
 201         //Make sure we found at least one translation
 202         if(qmFiles.count() < 1)
 203         {
 204                 qFatal("Could not find any translation files!");
 205                 return;
 206         }
 207
 208         //Add all available translations
 209         while(!qmFiles.isEmpty())
 210         {
 211                 QString langId, langName;
 212                 unsigned int systemId = 0;
 213                 QString qmFile = qmFiles.takeFirst();
 214
 215                 QRegExp langIdExp("LameXP_(\\w\\w)\\.qm", Qt::CaseInsensitive);
 216                 if(langIdExp.indexIn(qmFile) >= 0)
 217                 {
 218                         langId = langIdExp.cap(1).toLower();
 219                 }
 220
 221                 QResource langRes = (QString(":/localization/%1.txt").arg(qmFile));
 222                 if(langRes.isValid() && langRes.size() > 0)
 223                 {
 224                         QStringList langInfo = QString::fromUtf8(reinterpret_cast<const char*>(langRes.data()), langRes.size()).simplified().split(",", QString::SkipEmptyParts);
 225                         if(langInfo.count() == 2)
 226                         {
 227                                 systemId = langInfo.at(0).toUInt();
 228                                 langName = langInfo.at(1);
 229                         }
 230                 }
 231
 232                 if(lamexp_translation_register(langId, qmFile, langName, systemId))
 233                 {
 234                         qDebug("Registering translation: %s = %s (%u)", qmFile.toLatin1().constData(), langName.toLatin1().constData(), systemId);
 235                 }
 236                 else
 237                 {
 238                         qWarning("Failed to register: %s", qmFile.toLatin1().constData());
 239                 }
 240         }
 241
 242         qDebug("All registered.\n");
 243 }
 244
 245 void InitializationThread::initNeroAac(void)
 246 {
 247         QFileInfo neroFileInfo[3];
 248         neroFileInfo[0] = QFileInfo(QString("%1/neroAacEnc.exe").arg(QCoreApplication::applicationDirPath()));
 249         neroFileInfo[1] = QFileInfo(QString("%1/neroAacDec.exe").arg(QCoreApplication::applicationDirPath()));
 250         neroFileInfo[2] = QFileInfo(QString("%1/neroAacTag.exe").arg(QCoreApplication::applicationDirPath()));
 251
 252         bool neroFilesFound = true;
 253         for(int i = 0; i < 3; i++)      { if(!neroFileInfo[i].exists()) neroFilesFound = false; }
 254
 255         //Lock the Nero binaries
 256         if(!neroFilesFound)
 257         {
 258                 qDebug("Nero encoder binaries not found -> AAC encoding support will be disabled!\n");
 259                 return;
 260         }
 261
 262         qDebug("Found Nero AAC encoder binary:\n%s\n", neroFileInfo[0].canonicalFilePath().toUtf8().constData());
 263
 264         LockedFile *neroBin[3];
 265         for(int i = 0; i < 3; i++) neroBin[i] = NULL;
 266
 267         try
 268         {
 269                 for(int i = 0; i < 3; i++)
 270                 {
 271                         neroBin[i] = new LockedFile(neroFileInfo[i].canonicalFilePath());
 272                 }
 273         }
 274         catch(...)
 275         {
 276                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
 277                 qWarning("Failed to get excluive lock to Nero encoder binary -> AAC encoding support will be disabled!");
 278                 return;
 279         }
 280
 281         QProcess process;
 282         process.setProcessChannelMode(QProcess::MergedChannels);
 283         process.setReadChannel(QProcess::StandardOutput);
 284         process.start(neroFileInfo[0].canonicalFilePath(), QStringList() << "-help");
 285
 286         if(!process.waitForStarted())
 287         {
 288                 qWarning("Nero process failed to create!");
 289                 qWarning("Error message: \"%s\"\n", process.errorString().toLatin1().constData());
 290                 process.kill();
 291                 process.waitForFinished(-1);
 292                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
 293                 return;
 294         }
 295
 296         unsigned int neroVersion = 0;
 297
 298         while(process.state() != QProcess::NotRunning)
 299         {
 300                 if(!process.waitForReadyRead())
 301                 {
 302                         if(process.state() == QProcess::Running)
 303                         {
 304                                 qWarning("Nero process time out -> killing!");
 305                                 process.kill();
 306                                 process.waitForFinished(-1);
 307                                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
 308                                 return;
 309                         }
 310                 }
 311
 312                 while(process.canReadLine())
 313                 {
 314                         QString line = QString::fromUtf8(process.readLine().constData()).simplified();
 315                         QStringList tokens = line.split(" ", QString::SkipEmptyParts, Qt::CaseInsensitive);
 316                         int index1 = tokens.indexOf("Package");
 317                         int index2 = tokens.indexOf("version:");
 318                         if(index1 >= 0 && index2 >= 0 && index1 + 1 == index2 && index2 < tokens.count() - 1)
 319                         {
 320                                 QStringList versionTokens = tokens.at(index2 + 1).split(".", QString::SkipEmptyParts, Qt::CaseInsensitive);
 321                                 if(versionTokens.count() == 4)
 322                                 {
 323                                         neroVersion = 0;
 324                                         neroVersion += min(9, max(0, versionTokens.at(3).toInt()));
 325                                         neroVersion += min(9, max(0, versionTokens.at(2).toInt())) * 10;
 326                                         neroVersion += min(9, max(0, versionTokens.at(1).toInt())) * 100;
 327                                         neroVersion += min(9, max(0, versionTokens.at(0).toInt())) * 1000;
 328                                 }
 329                         }
 330                 }
 331         }
 332
 333         if(!(neroVersion > 0))
 334         {
 335                 qWarning("Nero AAC version could not be determined -> AAC encoding support will be disabled!");
 336                 for(int i = 0; i < 3; i++) LAMEXP_DELETE(neroBin[i]);
 337                 return;
 338         }
 339
 340         for(int i = 0; i < 3; i++)
 341         {
 342                 lamexp_register_tool(neroFileInfo[i].fileName(), neroBin[i], neroVersion);
 343         }
 344 }
 345
 346
 347 void InitializationThread::initWmaDec(void)
 348 {
 349         static const char* wmaDecoderComponentPath = "NCH Software/Components/wmawav/wmawav.exe";
 350
 351         LockedFile *wmaFileBin = NULL;
 352         QFileInfo wmaFileInfo = QFileInfo(QString("%1/%2").arg(lamexp_known_folder(lamexp_folder_programfiles), wmaDecoderComponentPath));
 353
 354         if(!wmaFileInfo.exists())
 355         {
 356                 qDebug("WMA File Decoder not found -> WMA decoding support will be disabled!\n");
 357                 return;
 358         }
 359
 360         try
 361         {
 362                 wmaFileBin = new LockedFile(wmaFileInfo.canonicalFilePath());
 363         }
 364         catch(...)
 365         {
 366                 qWarning("Failed to get excluive lock to WMA File Decoder binary -> WMA decoding support will be disabled!");
 367                 return;
 368         }
 369
 370         QProcess process;
 371         process.setProcessChannelMode(QProcess::MergedChannels);
 372         process.setReadChannel(QProcess::StandardOutput);
 373         process.start(wmaFileInfo.canonicalFilePath(), QStringList());
 374
 375         if(!process.waitForStarted())
 376         {
 377                 qWarning("WmaWav process failed to create!");
 378                 qWarning("Error message: \"%s\"\n", process.errorString().toLatin1().constData());
 379                 process.kill();
 380                 process.waitForFinished(-1);
 381                 return;
 382         }
 383
 384         bool b_wmaWavFound = false;
 385
 386         while(process.state() != QProcess::NotRunning)
 387         {
 388                 if(!process.waitForReadyRead())
 389                 {
 390                         if(process.state() == QProcess::Running)
 391                         {
 392                                 qWarning("WmaWav process time out -> killing!");
 393                                 process.kill();
 394                                 process.waitForFinished(-1);
 395                                 return;
 396                         }
 397                 }
 398                 while(process.canReadLine())
 399                 {
 400                         QString line = QString::fromUtf8(process.readLine().constData()).simplified();
 401                         if(line.contains("Usage: wmatowav.exe WMAFileSpec WAVFileSpec", Qt::CaseInsensitive))
 402                         {
 403                                 b_wmaWavFound = true;
 404                         }
 405                 }
 406         }
 407
 408         if(!b_wmaWavFound)
 409         {
 410                 qWarning("WmaWav could not be identified -> WMA decoding support will be disabled!\n");
 411                 LAMEXP_DELETE(wmaFileBin);
 412                 return;
 413         }
 414
 415         qDebug("Found WMA File Decoder binary:\n%s\n", wmaFileInfo.canonicalFilePath().toUtf8().constData());
 416
 417         if(wmaFileBin)
 418         {
 419                 lamexp_register_tool(wmaFileInfo.fileName(), wmaFileBin);
 420         }
 421 }
 422
 423 ////////////////////////////////////////////////////////////
 424 // EVENTS
 425 ////////////////////////////////////////////////////////////
 426
 427 /*NONE*/