///////////////////////////////////////////////////////////////////////////////
// Simple x264 Launcher
-// Copyright (C) 2004-2016 LoRd_MuldeR <MuldeR2@GMX.de>
+// Copyright (C) 2004-2020 LoRd_MuldeR <MuldeR2@GMX.de>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
//MUtils
#include <MUtils/UpdateChecker.h>
-#include <MUtils/Hash_Blake2.h>
+#include <MUtils/Hash.h>
#include <MUtils/GUI.h>
#include <MUtils/OSSupport.h>
#include <MUtils/Exception.h>
#include <QFileInfo>
#include <QDir>
#include <QMap>
+#include <QElapsedTimer>
///////////////////////////////////////////////////////////////////////////////
+static const char *const DIGEST_KEY = "~Dv/bW3/7t>6?RXVwkaZk-hmS0#O4JS/5YQAO>\\8hvr0B~7[n!X~KMYruemu:MDq";
+
const UpdaterDialog::binary_t UpdaterDialog::BINARIES[] =
{
- { "wget.exe", "7b522345239bcb95b5b0f7f50a883ba5957894a1feb769763e38ed789a8a0f63fead0155f54b9ffd0f1cdc5dfd855d207a6e7a8e4fd192589a8838ce646c504e", 1 },
- { "netc.exe", "c199ea12d761fa3191006da250f8f600ad426265fdf4a43e551cdf04a451a105692efd3ef82ac621c0799394aa21ac65bfbb4bab90c3fbb1f557e93f490fcb75", 1 },
- { "gpgv.exe", "18c5456cbb9ebf5cb9012a939b199d9eaa71c92a39f574f1e032babad0bbd9e72a064af96ca9d3d01f2892b064ec239fd61f27bac2eb9a64f7b2ece7beea3158", 1 },
- { "gpgv.gpg", "745c7a9c040196d9d322b1580e0046ff26ec13238cfd04325ceb3d4c8948294c593c027f895dc8ec427295175003e75d34f083019b706b0f4f06f81cce8df47d", 0 },
- { "wupd.exe", "28a5fd1515e27180a4659a9c1eee10e7483d25c556a6a142bce88176d0ac287105c0bd315fa5ece0a1a145fa64465378997a484f7a9ffffb28743c80f2c12ccd", 1 },
+ { "curl.exe", "4d51862b5df757e177a578fd6fabf631b78d9eccedfb71249db0dfb5c1b721f59f4eb7f7046e1c1054b76b9af7a0bed2479b09b7599df51236aee42c3625e67c", 1 },
+ { "gpgv.exe", "d0869bd858294520c992b66e1c7594176ebfb51bd64c4f50f782a4749118498a29c9fc70da5ed08cda9837ff7099d950428ca9f1968fa69883929bd0dba8c9e5", 1 },
+ { "gpgv.gpg", "1a2f528e551b9abfb064f08674fdd421d3abe403469ddfee2beafd007775a6c684212a6274dc2b41a0b20dd5c2200021c91320e737f7a90b2ac5a40a6221d93f", 0 },
+ { "wupd.exe", "018a8d0d848407fb0cb530b4540c6f025fd4c280885becd37f83feed8aeb3af6f8e8e0d45066a36549efac7e64706ac1ef09aaa5c75ab8d12c4a70f41518a894", 1 },
{ NULL, NULL, 0 }
};
} \
while(0)
+static inline QString getBin(const QMap<QString, QSharedPointer<QFile>> &binaries, const QString &nameName)
+{
+ const QSharedPointer<QFile> file = binaries.value(nameName);
+ return file.isNull() ? QString() : file->fileName();
+}
+
+static void qFileDeleter(QFile *const file)
+{
+ if(file)
+ {
+ file->close();
+ delete file;
+ }
+}
///////////////////////////////////////////////////////////////////////////////
// Constructor & Destructor
ui->setupUi(this);
setWindowFlags(windowFlags() & (~Qt::WindowContextHelpButtonHint));
- //Fix size
+ //Scale and fix size
+ MUtils::GUI::scale_widget(this);
setFixedSize(size());
//Enable buttons
ui->buttonCancel->setEnabled(false);
ui->buttonRetry->hide();
ui->buttonDownload->hide();
+ ui->labelCancel->hide();
//Start animation
SHOW_ANIMATION(true);
m_thread->wait();
}
}
-
- cleanFiles();
delete ui;
}
///////////////////////////////////////////////////////////////////////////////
-// Public Functions
-///////////////////////////////////////////////////////////////////////////////
-
-/*None yet*/
-
-///////////////////////////////////////////////////////////////////////////////
// Events
///////////////////////////////////////////////////////////////////////////////
void UpdaterDialog::keyPressEvent(QKeyEvent *event)
{
- if(event->key() == Qt::Key_F11)
+ switch (event->key())
{
- QFile logFile(QString("%1/%2.log").arg(MUtils::temp_folder(), MUtils::rand_str()));
- if(logFile.open(QIODevice::WriteOnly | QIODevice::Truncate))
+ case Qt::Key_Escape:
+ if ((!m_thread.isNull()) && m_thread->isRunning())
{
- logFile.write("\xEF\xBB\xBF");
- for(QStringList::ConstIterator iter = m_logFile.constBegin(); iter != m_logFile.constEnd(); iter++)
+ if (m_status >= MUtils::UpdateChecker::UpdateStatus_FetchingUpdates)
{
- logFile.write(iter->toUtf8());
- logFile.write("\r\n");
+ UPDATE_TEXT(2, tr("Cancellation requested..."));
+ }
+ else
+ {
+ UPDATE_TEXT(1, tr("Cancellation requested..."));
+ }
+ m_thread->cancel();
+ }
+ break;
+ case Qt::Key_F11:
+ {
+ const QString logFilePath = MUtils::make_temp_file(MUtils::temp_folder(), "txt", true);
+ if (!logFilePath.isEmpty())
+ {
+ qWarning("Write log to: '%s'", MUTILS_UTF8(logFilePath));
+ QFile logFile(logFilePath);
+ if (logFile.open(QIODevice::WriteOnly | QIODevice::Truncate))
+ {
+ logFile.write("\xEF\xBB\xBF");
+ for (QStringList::ConstIterator iter = m_logFile.constBegin(); iter != m_logFile.constEnd(); iter++)
+ {
+ logFile.write(iter->toUtf8());
+ logFile.write("\r\n");
+ }
+ logFile.close();
+ QDesktopServices::openUrl(QUrl::fromLocalFile(logFile.fileName()));
+ }
}
- logFile.close();
- QDesktopServices::openUrl(QUrl::fromLocalFile(logFile.fileName()));
}
+ break;
+ default:
+ QDialog::keyPressEvent(event);
}
}
void UpdaterDialog::initUpdate(void)
{
- //Clean up files from previous attempt
- if(!m_binaries.isEmpty())
- {
- cleanFiles();
- }
-
//Check binary files
if(!checkBinaries())
{
{
QDesktopServices::openUrl(QUrl(QString::fromLatin1(m_updateUrl)));
}
- close(); return;
+ close();
+ return;
}
//Make sure user does have admin access
if(QMessageBox::critical(this, this->windowTitle(), message, tr("Discard"), tr("Ignore")) != 1)
{
ui->buttonCancel->setEnabled(true);
- close(); return;
+ close();
+ return;
}
}
//Create and setup thread
if(!m_thread)
{
- m_thread.reset(new MUtils::UpdateChecker(m_binaries.value("wget.exe"), m_binaries.value("netc.exe"), m_binaries.value("gpgv.exe"), m_binaries.value("gpgv.gpg"), "Simple x264 Launcher", x264_version_build(), false));
+ m_thread.reset(new MUtils::UpdateChecker(getBin(m_binaries, "curl.exe"), getBin(m_binaries, "gpgv.exe"), getBin(m_binaries, "gpgv.gpg"), "Simple x264 Launcher", x264_version_build(), false));
connect(m_thread.data(), SIGNAL(statusChanged(int)), this, SLOT(threadStatusChanged(int)));
connect(m_thread.data(), SIGNAL(finished()), this, SLOT(threadFinished()));
connect(m_thread.data(), SIGNAL(terminated()), this, SLOT(threadFinished()));
//Hide labels
ui->labelInfo->hide();
ui->labelUrl->hide();
+ ui->labelCancel->show();
//Update status
threadStatusChanged(MUtils::UpdateChecker::UpdateStatus_NotStartedYet);
//Clear log
m_logFile.clear();
+ //Init timer
+ m_elapsed.reset(new QElapsedTimer());
+ m_elapsed->start();
+
//Start the updater thread
- QTimer::singleShot(250, m_thread.data(), SLOT(start()));
+ QTimer::singleShot(125, m_thread.data(), SLOT(start()));
}
void UpdaterDialog::threadStatusChanged(int status)
{
+ const int prevStatus = m_status;
switch(m_status = status)
{
case MUtils::UpdateChecker::UpdateStatus_NotStartedYet:
UPDATE_TEXT(2, tr("Update information received successfully."));
UPDATE_ICON(3, "play");
break;
+ case MUtils::UpdateChecker::UpdateStatus_CancelledByUser:
+ if (prevStatus >= MUtils::UpdateChecker::UpdateStatus_FetchingUpdates)
+ {
+ UPDATE_ICON(2, "shield_error");
+ UPDATE_TEXT(2, tr("Operation was cancelled by the user!"));
+ UPDATE_ICON(3, "shield_grey");
+ }
+ else
+ {
+ UPDATE_ICON(1, "shield_error");
+ UPDATE_TEXT(1, tr("Operation was cancelled by the user!"));
+ UPDATE_ICON(2, "shield_grey");
+ UPDATE_ICON(3, "shield_grey");
+ }
+ break;
default:
MUTILS_THROW("Unknown status code!");
}
void UpdaterDialog::threadFinished(void)
{
m_success = m_thread->getSuccess();
- QTimer::singleShot((m_success ? 1000 : 0), this, SLOT(updateFinished()));
+ ui->labelCancel->hide();
+ QTimer::singleShot((m_success ? 500 : 0), this, SLOT(updateFinished()));
}
void UpdaterDialog::updateFinished(void)
{
+ //Query the timer, if available
+ if (!m_elapsed.isNull())
+ {
+ const quint64 elapsed = m_elapsed->restart();
+ qDebug("Update check completed after %.2f seconds.", double(elapsed) / 1000.0);
+ }
+
//Restore cursor
QApplication::restoreOverrideCursor();
case MUtils::UpdateChecker::UpdateStatus_ErrorNoConnection:
case MUtils::UpdateChecker::UpdateStatus_ErrorConnectionTestFailed:
case MUtils::UpdateChecker::UpdateStatus_ErrorFetchUpdateInfo:
+ case MUtils::UpdateChecker::UpdateStatus_CancelledByUser:
m_animator->stop();
ui->buttonRetry->show();
break;
args << QString("/ToExFile=%1.exe").arg(QFileInfo(QFileInfo(QApplication::applicationFilePath()).canonicalFilePath()).completeBaseName());
args << QString("/AppTitle=Simple x264 Launcher (Build #%1)").arg(QString::number(updateInfo->getBuildNo()));
- process.start(m_binaries.value("wupd.exe"), args);
+ process.start(getBin(m_binaries, "wupd.exe"), args);
if(!process.waitForStarted())
{
QApplication::restoreOverrideCursor();
bool UpdaterDialog::checkBinaries(void)
{
qDebug("[File Verification]");
- m_binaries.clear();
-
- //Validate hashes first
- const QString tempPath = MUtils::temp_folder();
for(size_t i = 0; BINARIES[i].name; i++)
{
- const QString orgName = QString::fromLatin1(BINARIES[i].name);
- const QString binPath = QString("%1/toolset/common/%2").arg(m_sysinfo->getAppPath(), orgName);
- const QString outPath = QString("%1/%2_%3.%4").arg(tempPath, QFileInfo(orgName).baseName(), MUtils::rand_str(), QFileInfo(orgName).suffix());
- if(!checkFileHash(binPath, BINARIES[i].hash))
- {
- qWarning("Verification of '%s' has failed!", MUTILS_UTF8(orgName));
- return false;
- }
- if(!QFile::copy(binPath, outPath))
+ const QString name = QString::fromLatin1(BINARIES[i].name);
+ if (!m_binaries.contains(name))
{
- qWarning("Copying of '%s' has failed!", MUTILS_UTF8(orgName));
- return false;
+ QScopedPointer<QFile> binary(new QFile(QString("%1/toolset/common/%2").arg(m_sysinfo->getAppPath(), name)));
+ if (binary->open(QIODevice::ReadOnly))
+ {
+ if (checkFileHash(binary->fileName(), BINARIES[i].hash))
+ {
+ QApplication::processEvents(QEventLoop::ExcludeUserInputEvents);
+ m_binaries.insert(name, QSharedPointer<QFile>(binary.take(), qFileDeleter));
+ }
+ else
+ {
+ qWarning("Verification of '%s' has failed!", MUTILS_UTF8(name));
+ binary->close();
+ return false;
+ }
+ }
+ else
+ {
+ qWarning("File '%s' could not be opened!", MUTILS_UTF8(name));
+ return false;
+ }
}
- QFile::setPermissions(outPath, QFile::ReadOwner);
- m_binaries.insert(BINARIES[i].name, outPath);
- QApplication::processEvents(QEventLoop::ExcludeUserInputEvents);
}
-
+ qDebug("File check completed.\n");
return true;
}
bool UpdaterDialog::checkFileHash(const QString &filePath, const char *expectedHash)
{
- qDebug("Checking file: %s", filePath.toUtf8().constData());
- MUtils::Hash::Blake2 checksum2;
+ qDebug("Checking file: %s", MUTILS_UTF8(filePath));
+ QScopedPointer<MUtils::Hash::Hash> checksum(MUtils::Hash::create(MUtils::Hash::HASH_BLAKE2_512, DIGEST_KEY));
QFile file(filePath);
if(file.open(QIODevice::ReadOnly))
{
- checksum2.update(file);
- const QByteArray fileHash = checksum2.finalize();
+ checksum->update(file);
+ const QByteArray fileHash = checksum->digest();
if((strlen(expectedHash) != fileHash.size()) || (memcmp(fileHash.constData(), expectedHash, fileHash.size()) != 0))
{
qWarning("\nFile appears to be corrupted:\n%s\n", filePath.toUtf8().constData());
return false;
}
}
-
-void UpdaterDialog::cleanFiles(void)
-{
- const QStringList keys = m_binaries.keys();
- foreach(const QString &key, keys)
- {
- const QString fileName = m_binaries.value(key);
- QFile::setPermissions(fileName, QFile::ReadOwner | QFile::WriteOwner);
- if(!QFile::remove(fileName))
- {
- qWarning("Failed to remove file: %s", MUTILS_UTF8(fileName));
- }
- m_binaries.remove(key);
- }
-}