///////////////////////////////////////////////////////////////////////////////
// MuldeR's Utilities for Qt
-// Copyright (C) 2004-2017 LoRd_MuldeR <MuldeR2@GMX.de>
+// Copyright (C) 2004-2018 LoRd_MuldeR <MuldeR2@GMX.de>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
#include <QStringList>
#include <QFile>
#include <QFileInfo>
+#include <QDir>
#include <QProcess>
#include <QUrl>
#include <QEventLoop>
#include <QTimer>
#include <QElapsedTimer>
#include <QSet>
+#include <QHash>
+#include <QQueue>
#include "Mirrors.h"
// CONSTANTS
///////////////////////////////////////////////////////////////////////////////
-static const char *header_id = "!Update";
+static const char *GLOBALHEADER_ID = "!Update";
-static const char *mirror_url_postfix[] =
+static const char *MIRROR_URL_POSTFIX[] =
{
"update.ver",
"update_beta.ver",
static const int DOWNLOAD_TIMEOUT = 30000;
static const int VERSION_INFO_EXPIRES_MONTHS = 6;
-static char *USER_AGENT_STR = "Mozilla/5.0 (X11; Linux i686; rv:10.0) Gecko/20100101 Firefox/10.0"; /*use something innocuous*/
+static char *USER_AGENT_STR = "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:52.0) Gecko/20100101 Firefox/52.0"; /*use something innocuous*/
+
+////////////////////////////////////////////////////////////
+// Utility Macros
+////////////////////////////////////////////////////////////
#define CHECK_CANCELLED() do \
{ \
} \
while(0)
+#define LOG_MESSAGE_HELPER(X) do \
+{ \
+ if (!(X).isNull()) \
+ { \
+ emit messageLogged((X)); \
+ } \
+} \
+while(0)
+
+#define STRICMP(X,Y) ((X).compare((Y), Qt::CaseInsensitive) == 0)
+
////////////////////////////////////////////////////////////
// Helper Functions
////////////////////////////////////////////////////////////
-static int getMaxProgress(void)
+static QQueue<QString> buildRandomList(const char *const *values)
{
- int counter = 0;
- while (update_mirrors[counter])
+ QQueue<QString> list;
+ while(*values)
{
- counter++;
+ list.insert(MUtils::next_rand_u32(list.size() + 1), QString::fromLatin1(*(values++)));
}
- counter += MIN_CONNSCORE + QUICK_MIRRORS + 2;
- return counter; ;
+ return list;
}
-static QStringList buildRandomList(const char *const values[])
+static const QHash<QString, QString> *initEnvVars(void)
{
- QStringList list;
- for (int index = 0; values[index]; index++)
- {
- const int pos = MUtils::next_rand_u32() % (index + 1);
- list.insert(pos, QString::fromLatin1(values[index]));
- }
- return list;
+ QHash<QString, QString> *const environment = new QHash<QString, QString>();
+ environment->insert(QLatin1String("CURL_HOME"), QDir::toNativeSeparators(MUtils::temp_folder()));
+ return environment;
}
////////////////////////////////////////////////////////////
bool MUtils::UpdateCheckerInfo::isComplete(void)
{
- if(this->m_buildNo < 1) return false;
- if(this->m_buildDate.year() < 2010) return false;
- if(this->m_downloadSite.isEmpty()) return false;
- if(this->m_downloadAddress.isEmpty()) return false;
- if(this->m_downloadFilename.isEmpty()) return false;
- if(this->m_downloadFilecode.isEmpty()) return false;
- if(this->m_downloadChecksum.isEmpty()) return false;
-
- return true;
+ return (this->m_buildNo > 0) &&
+ (this->m_buildDate.year() >= 2010) &&
+ (!this->m_downloadSite.isEmpty()) &&
+ (!this->m_downloadAddress.isEmpty()) &&
+ (!this->m_downloadFilename.isEmpty()) &&
+ (!this->m_downloadFilecode.isEmpty()) &&
+ (!this->m_downloadChecksum.isEmpty());
}
////////////////////////////////////////////////////////////
// Constructor & Destructor
////////////////////////////////////////////////////////////
-MUtils::UpdateChecker::UpdateChecker(const QString &binWGet, const QString &binMCat, const QString &binGnuPG, const QString &binKeys, const QString &applicationId, const quint32 &installedBuildNo, const bool betaUpdates, const bool testMode)
+MUtils::UpdateChecker::UpdateChecker(const QString &binCurl, const QString &binGnuPG, const QString &binKeys, const QString &applicationId, const quint32 &installedBuildNo, const bool betaUpdates, const bool testMode)
:
m_updateInfo(new UpdateCheckerInfo()),
- m_binaryWGet(binWGet),
- m_binaryMCat(binMCat),
+ m_binaryCurl(binCurl),
m_binaryGnuPG(binGnuPG),
m_binaryKeys(binKeys),
m_applicationId(applicationId),
m_installedBuildNo(installedBuildNo),
m_betaUpdates(betaUpdates),
m_testMode(testMode),
- m_maxProgress(getMaxProgress())
+ m_maxProgress(MIN_CONNSCORE + 5),
+ m_environment(initEnvVars())
{
m_status = UpdateStatus_NotStartedYet;
m_progress = 0;
- if(m_binaryWGet.isEmpty() || m_binaryGnuPG.isEmpty() || m_binaryKeys.isEmpty())
+ if(m_binaryCurl.isEmpty() || m_binaryGnuPG.isEmpty() || m_binaryKeys.isEmpty())
{
MUTILS_THROW("Tools not initialized correctly!");
}
// ----- Test Internet Connection ----- //
- log("Checking internet connection...", "");
+ log("Checking your Internet connection...", "");
setStatus(UpdateStatus_CheckingConnection);
const int networkStatus = OS::network_status();
if(networkStatus == OS::NETWORK_TYPE_NON)
{
- log("Operating system reports that the computer is currently offline !!!");
- setProgress(m_maxProgress);
- setStatus(UpdateStatus_ErrorNoConnection);
- return;
+ if (!MUtils::OS::arguments().contains("ignore-network-status"))
+ {
+ log("Operating system reports that the computer is currently offline !!!");
+ setProgress(m_maxProgress);
+ setStatus(UpdateStatus_ErrorNoConnection);
+ return;
+ }
}
+ msleep(333);
setProgress(1);
// ----- Test Known Hosts Connectivity ----- //
int connectionScore = 0;
- QStringList mirrorList = buildRandomList(known_hosts);
+ QQueue<QString> mirrorList = buildRandomList(known_hosts);
- for(int connectionTimout = 500; connectionTimout <= MAX_CONN_TIMEOUT; connectionTimout *= 2)
+ for(int connectionTimeout = 1000; connectionTimeout <= MAX_CONN_TIMEOUT; connectionTimeout *= 2)
{
QElapsedTimer elapsedTimer;
elapsedTimer.start();
- const int globalTimout = 2 * MIN_CONNSCORE * connectionTimout;
- while (!elapsedTimer.hasExpired(globalTimout))
+ const qint64 globalTimeout = 2 * MIN_CONNSCORE * connectionTimeout;
+ forever
{
- const QString hostName = mirrorList.takeFirst();
- if (tryContactHost(hostName, connectionTimout))
+ if (mirrorList.isEmpty())
{
- connectionScore += 1;
- setProgress(qBound(1, connectionScore + 1, MIN_CONNSCORE + 1));
- elapsedTimer.restart();
+ goto endLoop; /*depleted!*/
+ }
+ const QString hostName = mirrorList.dequeue();
+ if (tryContactHost(hostName, connectionTimeout))
+ {
+ setProgress(1 + (++connectionScore));
if (connectionScore >= MIN_CONNSCORE)
{
goto endLoop; /*success*/
}
else
{
- mirrorList.append(hostName); /*re-schedule*/
+ mirrorList.enqueue(hostName);
+ if(elapsedTimer.hasExpired(globalTimeout))
+ {
+ break; /*timer expired*/
+ }
}
CHECK_CANCELLED();
- msleep(1);
}
}
// ----- Fetch Update Info From Server ----- //
- log("----", "", "Checking for updates online...");
+ log("----", "", "Internet connection is operational, checking for updates online...");
setStatus(UpdateStatus_FetchingUpdates);
int mirrorCount = 0;
while(!mirrorList.isEmpty())
{
- setProgress(m_progress + 1);
const QString currentMirror = mirrorList.takeFirst();
const bool isQuick = (mirrorCount++ < QUICK_MIRRORS);
if(tryUpdateMirror(m_updateInfo.data(), currentMirror, isQuick))
CHECK_CANCELLED();
msleep(1);
}
-
- while (m_progress < m_maxProgress)
- {
- msleep(16);
- setProgress(m_progress + 1);
- CHECK_CANCELLED();
- }
+
+ msleep(333);
+ setProgress(MIN_CONNSCORE + 5);
// ----- Generate final result ----- //
void MUtils::UpdateChecker::testMirrorsList(void)
{
- // ----- Test update mirrors ----- //
-
QStringList mirrorList;
for(int i = 0; update_mirrors[i]; i++)
{
mirrorList << QString::fromLatin1(update_mirrors[i]);
}
+ // ----- Test update mirrors ----- //
+
qDebug("\n[Mirror Sites]");
log("Testing all known mirror sites...", "", "---");
qDebug("\n[Known Hosts]");
log("Testing all known hosts...", "", "---");
- QSet<quint32> ipAddrSet;
- quint32 ipAddr;
while(!knownHostList.isEmpty())
{
const QString currentHost = knownHostList.takeFirst();
qDebug("Testing: %s", MUTILS_L1STR(currentHost));
log(QLatin1String(""), "Testing:", currentHost, "");
- if (tryContactHost(currentHost, DOWNLOAD_TIMEOUT, &ipAddr))
- {
- if (ipAddrSet.contains(ipAddr))
- {
- qWarning("Duplicate IP-address 0x%08X was encountered!", ipAddr);
- }
- else
- {
- ipAddrSet << ipAddr; /*not encountered yet*/
- }
- }
- else
+ if (!tryContactHost(currentHost, DOWNLOAD_TIMEOUT))
{
qWarning("\nConnectivity test FAILED on the following host:\n%s\n", MUTILS_L1STR(currentHost));
}
- log("", "---");
+ log("---");
}
}
void MUtils::UpdateChecker::setProgress(const int progress)
{
- if(m_progress != progress)
+ const int value = qBound(0, progress, m_maxProgress);
+ if(m_progress != value)
{
- m_progress = progress;
- emit progressChanged(progress);
+ emit progressChanged(m_progress = value);
}
}
void MUtils::UpdateChecker::log(const QString &str1, const QString &str2, const QString &str3, const QString &str4)
{
- if(!str1.isNull()) emit messageLogged(str1);
- if(!str2.isNull()) emit messageLogged(str2);
- if(!str3.isNull()) emit messageLogged(str3);
- if(!str4.isNull()) emit messageLogged(str4);
+ LOG_MESSAGE_HELPER(str1);
+ LOG_MESSAGE_HELPER(str2);
+ LOG_MESSAGE_HELPER(str3);
+ LOG_MESSAGE_HELPER(str4);
}
bool MUtils::UpdateChecker::tryUpdateMirror(UpdateCheckerInfo *updateInfo, const QString &url, const bool &quick)
{
bool success = false;
- log("", "Trying mirror:", url, "");
+ log("", "Trying update mirror:", url, "");
if (quick)
{
- if (!tryContactHost(QUrl(url).host(), (MAX_CONN_TIMEOUT / 10)))
+ setProgress(MIN_CONNSCORE + 1);
+ if (!tryContactHost(QUrl(url).host(), (MAX_CONN_TIMEOUT / 8)))
{
log("", "Mirror is too slow, skipping!");
return false;
const QString outFileVers = QString("%1/%2.ver").arg(temp_folder(), randPart);
const QString outFileSign = QString("%1/%2.sig").arg(temp_folder(), randPart);
- if (getUpdateInfo(url, outFileVers, outFileSign))
+ if (!getUpdateInfo(url, outFileVers, outFileSign))
{
- log("", "Download okay, checking signature:");
- if (checkSignature(outFileVers, outFileSign))
- {
- log("", "Signature okay, parsing info:", "");
- success = parseVersionInfo(outFileVers, updateInfo);
- }
- else
- {
- log("", "Bad signature, take care!");
- }
+ log("", "Oops: Download of update information has failed!");
+ goto cleanUp;
}
- else
+
+ log("Download completed, verifying signature:", "");
+ setProgress(MIN_CONNSCORE + 4);
+ if (!checkSignature(outFileVers, outFileSign))
{
- log("", "Download has failed!");
+ log("", "Bad signature detected, take care !!!");
+ goto cleanUp;
}
+ log("", "Signature is valid, parsing update information:", "");
+ success = parseVersionInfo(outFileVers, updateInfo);
+
+cleanUp:
QFile::remove(outFileVers);
QFile::remove(outFileSign);
-
return success;
}
bool MUtils::UpdateChecker::getUpdateInfo(const QString &url, const QString &outFileVers, const QString &outFileSign)
{
- log("Downloading update info:", "");
- if(getFile(QString("%1%2").arg(url, mirror_url_postfix[m_betaUpdates ? 1 : 0]), outFileVers))
+ log("Downloading update information:", "");
+ setProgress(MIN_CONNSCORE + 2);
+ if(getFile(QUrl(QString("%1%2").arg(url, MIRROR_URL_POSTFIX[m_betaUpdates ? 1 : 0])), outFileVers))
{
if (!m_cancelled)
{
- log("", "Downloading signature:", "");
- if (getFile(QString("%1%2.sig2").arg(url, mirror_url_postfix[m_betaUpdates ? 1 : 0]), outFileSign))
+ log( "Downloading signature file:", "");
+ setProgress(MIN_CONNSCORE + 3);
+ if (getFile(QUrl(QString("%1%2.sig2").arg(url, MIRROR_URL_POSTFIX[m_betaUpdates ? 1 : 0])), outFileSign))
{
- return true;
+ return true; /*completed*/
}
}
}
return false;
}
-bool MUtils::UpdateChecker::parseVersionInfo(const QString &file, UpdateCheckerInfo *updateInfo)
-{
- QRegExp value("^(\\w+)=(.+)$");
- QRegExp section("^\\[(.+)\\]$");
+//----------------------------------------------------------
+// PARSE UPDATE INFO
+//----------------------------------------------------------
- QDate updateInfoDate;
+#define _CHECK_HEADER(ID,NAME) \
+ if (STRICMP(name, (NAME))) \
+ { \
+ sectionId = (ID); \
+ continue; \
+ }
+
+#define _PARSE_TEXT(OUT,KEY) \
+ if (STRICMP(key, (KEY))) \
+ { \
+ (OUT) = val; \
+ break; \
+ }
+
+#define _PARSE_UINT(OUT,KEY) \
+ if (STRICMP(key, (KEY))) \
+ { \
+ bool _ok = false; \
+ const unsigned int _tmp = val.toUInt(&_ok); \
+ if (_ok) \
+ { \
+ (OUT) = _tmp; \
+ break; \
+ } \
+ }
+
+#define _PARSE_DATE(OUT,KEY) \
+ if (STRICMP(key, (KEY))) \
+ { \
+ const QDate _tmp = QDate::fromString(val, Qt::ISODate); \
+ if (_tmp.isValid()) \
+ { \
+ (OUT) = _tmp; \
+ break; \
+ } \
+ }
+
+bool MUtils::UpdateChecker::parseVersionInfo(const QString &file, UpdateCheckerInfo *const updateInfo)
+{
updateInfo->resetInfo();
QFile data(file);
qWarning("Cannot open update info file for reading!");
return false;
}
-
- bool inHdr = false;
- bool inSec = false;
-
+
+ QDate updateInfoDate;
+ int sectionId = 0;
+ QRegExp regex_sec("^\\[(.+)\\]$"), regex_val("^([^=]+)=(.+)$");
+
while(!data.atEnd())
{
QString line = QString::fromLatin1(data.readLine()).trimmed();
- if(section.indexIn(line) >= 0)
+ if (regex_sec.indexIn(line) >= 0)
{
- log(QString("Sec: [%1]").arg(section.cap(1)));
- inSec = (section.cap(1).compare(m_applicationId, Qt::CaseInsensitive) == 0);
- inHdr = (section.cap(1).compare(QString::fromLatin1(header_id), Qt::CaseInsensitive) == 0);
+ sectionId = 0; /*unknown section*/
+ const QString name = regex_sec.cap(1).trimmed();
+ log(QString("Sec: [%1]").arg(name));
+ _CHECK_HEADER(1, GLOBALHEADER_ID)
+ _CHECK_HEADER(2, m_applicationId)
continue;
}
- if(inSec && (value.indexIn(line) >= 0))
+ if (regex_val.indexIn(line) >= 0)
{
- log(QString("Val: '%1' ==> '%2").arg(value.cap(1), value.cap(2)));
- if(value.cap(1).compare("BuildNo", Qt::CaseInsensitive) == 0)
+ const QString key = regex_val.cap(1).trimmed();
+ const QString val = regex_val.cap(2).trimmed();
+ log(QString("Val: \"%1\" = \"%2\"").arg(key, val));
+ switch (sectionId)
{
- bool ok = false;
- const unsigned int temp = value.cap(2).toUInt(&ok);
- if(ok) updateInfo->m_buildNo = temp;
- }
- else if(value.cap(1).compare("BuildDate", Qt::CaseInsensitive) == 0)
- {
- const QDate temp = QDate::fromString(value.cap(2).trimmed(), Qt::ISODate);
- if(temp.isValid()) updateInfo->m_buildDate = temp;
- }
- else if(value.cap(1).compare("DownloadSite", Qt::CaseInsensitive) == 0)
- {
- updateInfo->m_downloadSite = value.cap(2).trimmed();
- }
- else if(value.cap(1).compare("DownloadAddress", Qt::CaseInsensitive) == 0)
- {
- updateInfo->m_downloadAddress = value.cap(2).trimmed();
- }
- else if(value.cap(1).compare("DownloadFilename", Qt::CaseInsensitive) == 0)
- {
- updateInfo->m_downloadFilename = value.cap(2).trimmed();
- }
- else if(value.cap(1).compare("DownloadFilecode", Qt::CaseInsensitive) == 0)
- {
- updateInfo->m_downloadFilecode = value.cap(2).trimmed();
- }
- else if(value.cap(1).compare("DownloadChecksum", Qt::CaseInsensitive) == 0)
- {
- updateInfo->m_downloadChecksum = value.cap(2).trimmed();
- }
- }
- if(inHdr && (value.indexIn(line) >= 0))
- {
- log(QString("Val: '%1' ==> '%2").arg(value.cap(1), value.cap(2)));
- if(value.cap(1).compare("TimestampCreated", Qt::CaseInsensitive) == 0)
- {
- QDate temp = QDate::fromString(value.cap(2).trimmed(), Qt::ISODate);
- if(temp.isValid()) updateInfoDate = temp;
+ case 1:
+ _PARSE_DATE(updateInfoDate, "TimestampCreated")
+ break;
+ case 2:
+ _PARSE_UINT(updateInfo->m_buildNo, "BuildNo")
+ _PARSE_DATE(updateInfo->m_buildDate, "BuildDate")
+ _PARSE_TEXT(updateInfo->m_downloadSite, "DownloadSite")
+ _PARSE_TEXT(updateInfo->m_downloadAddress, "DownloadAddress")
+ _PARSE_TEXT(updateInfo->m_downloadFilename, "DownloadFilename")
+ _PARSE_TEXT(updateInfo->m_downloadFilecode, "DownloadFilecode")
+ _PARSE_TEXT(updateInfo->m_downloadChecksum, "DownloadChecksum")
+ break;
}
}
}
- if(!updateInfoDate.isValid())
- {
- updateInfo->resetInfo();
- log("WARNING: Version info timestamp is missing!");
- return false;
- }
-
- const QDate currentDate = OS::current_date();
- if(updateInfoDate.addMonths(VERSION_INFO_EXPIRES_MONTHS) < currentDate)
+ if (!updateInfo->isComplete())
{
- updateInfo->resetInfo();
- log(QString::fromLatin1("WARNING: This version info has expired at %1!").arg(updateInfoDate.addMonths(VERSION_INFO_EXPIRES_MONTHS).toString(Qt::ISODate)));
- return false;
+ log("", "WARNING: Update information is incomplete!");
+ goto failure;
}
- else if(currentDate < updateInfoDate)
+
+ if(updateInfoDate.isValid())
{
- log("Version info is from the future, take care!");
- qWarning("Version info is from the future, take care!");
+ const QDate expiredDate = updateInfoDate.addMonths(VERSION_INFO_EXPIRES_MONTHS);
+ if (expiredDate < OS::current_date())
+ {
+ log("", QString("WARNING: Update information has expired at %1!").arg(expiredDate.toString(Qt::ISODate)));
+ goto failure;
+ }
}
-
- if(!updateInfo->isComplete())
+ else
{
- log("WARNING: Version info is incomplete!");
- return false;
+ log("", "WARNING: Timestamp is missing from update information header!");
+ goto failure;
}
- return true;
+ log("", "Success: Update information is complete.");
+ return true; /*success*/
+
+failure:
+ updateInfo->resetInfo();
+ return false;
}
//----------------------------------------------------------
// EXTERNAL TOOLS
//----------------------------------------------------------
-bool MUtils::UpdateChecker::getFile(const QString &url, const QString &outFile, const unsigned int maxRedir)
-{
- for (int i = 0; i < 2; i++)
- {
- if (getFile(url, (i > 0), outFile, maxRedir))
- {
- return true;
- }
- if (MUTILS_BOOLIFY(m_cancelled))
- {
- break; /*cancelled*/
- }
- }
- return false;
-}
-
-bool MUtils::UpdateChecker::getFile(const QString &url, const bool forceIp4, const QString &outFile, const unsigned int maxRedir)
+bool MUtils::UpdateChecker::getFile(const QUrl &url, const QString &outFile, const unsigned int maxRedir)
{
QFileInfo output(outFile);
output.setCaching(false);
QFile::remove(output.canonicalFilePath());
if (output.exists())
{
+ qWarning("Existing output file could not be found!");
return false;
}
}
+
+ QStringList args(QLatin1String("-vsSNqkfL"));
+ args << "-m" << QString::number(DOWNLOAD_TIMEOUT / 1000);
+ args << "--max-redirs" << QString::number(maxRedir);
+ args << "-A" << USER_AGENT_STR;
+ args << "-e" << QString("%1://%2/;auto").arg(url.scheme(), url.host());
+ args << "-o" << output.fileName() << url.toString();
+
+ return execCurl(args, output.absolutePath(), DOWNLOAD_TIMEOUT);
+}
- QProcess process;
- init_process(process, output.absolutePath());
+bool MUtils::UpdateChecker::tryContactHost(const QString &hostname, const int &timeoutMsec)
+{
+ log(QString("Connecting to host: %1").arg(hostname), "");
- QStringList args;
- if (forceIp4)
+ QStringList args(QLatin1String("-vsSNqkI"));
+ args << "-m" << QString::number(qMax(1, timeoutMsec / 1000));
+ args << "-A" << USER_AGENT_STR;
+ args << "-o" << OS::null_device() << QString("http://%1/").arg(hostname);
+
+ return execCurl(args, temp_folder(), timeoutMsec);
+}
+
+bool MUtils::UpdateChecker::checkSignature(const QString &file, const QString &signature)
+{
+ if (QFileInfo(file).absolutePath().compare(QFileInfo(signature).absolutePath(), Qt::CaseInsensitive) != 0)
{
- args << "-4";
+ qWarning("CheckSignature: File and signature should be in same folder!");
+ return false;
}
- args << "--no-config" << "--no-cache" << "--no-dns-cache" << "--no-check-certificate" << "--no-hsts";
- args << QString().sprintf("--max-redirect=%u", maxRedir) << QString().sprintf("--timeout=%u", DOWNLOAD_TIMEOUT / 1000);
- args << QString("--referer=%1://%2/").arg(QUrl(url).scheme(), QUrl(url).host()) << "-U" << USER_AGENT_STR;
- args << "-O" << output.fileName() << url;
-
- QEventLoop loop;
- connect(&process, SIGNAL(error(QProcess::ProcessError)), &loop, SLOT(quit()));
- connect(&process, SIGNAL(finished(int, QProcess::ExitStatus)), &loop, SLOT(quit()));
- connect(&process, SIGNAL(readyRead()), &loop, SLOT(quit()));
-
- QTimer timer;
- timer.setSingleShot(true);
- connect(&timer, SIGNAL(timeout()), &loop, SLOT(quit()));
+ QString keyRingPath(m_binaryKeys);
+ bool removeKeyring = false;
+ if (QFileInfo(file).absolutePath().compare(QFileInfo(m_binaryKeys).absolutePath(), Qt::CaseInsensitive) != 0)
+ {
+ keyRingPath = make_temp_file(QFileInfo(file).absolutePath(), "gpg");
+ removeKeyring = true;
+ if (!QFile::copy(m_binaryKeys, keyRingPath))
+ {
+ qWarning("CheckSignature: Failed to copy the key-ring file!");
+ return false;
+ }
+ }
- const QRegExp httpResponseOK("200 OK$");
+ QStringList args;
+ args << QStringList() << "--homedir" << ".";
+ args << "--keyring" << QFileInfo(keyRingPath).fileName();
+ args << QFileInfo(signature).fileName();
+ args << QFileInfo(file).fileName();
- process.start(m_binaryWGet, args);
+ const int exitCode = execProcess(m_binaryGnuPG, args, QFileInfo(file).absolutePath(), DOWNLOAD_TIMEOUT);
+ if (exitCode != INT_MAX)
+ {
+ log(QString().sprintf("Exited with code %d", exitCode));
+ }
- if (!process.waitForStarted())
+ if (removeKeyring)
{
- return false;
+ remove_file(keyRingPath);
}
- timer.start(DOWNLOAD_TIMEOUT);
+ return (exitCode == 0); /*completed*/
+}
- while (process.state() != QProcess::NotRunning)
+bool MUtils::UpdateChecker::execCurl(const QStringList &args, const QString &workingDir, const int timeout)
+{
+ const int exitCode = execProcess(m_binaryCurl, args, workingDir, timeout + (timeout / 2));
+ if (exitCode != INT_MAX)
{
- loop.exec();
- const bool bTimeOut = (!timer.isActive());
- while (process.canReadLine())
- {
- const QString line = QString::fromLatin1(process.readLine()).simplified();
- log(line);
- }
- if (bTimeOut || MUTILS_BOOLIFY(m_cancelled))
+ switch (exitCode)
{
- qWarning("WGet process timed out <-- killing!");
- process.kill();
- process.waitForFinished();
- log(bTimeOut ? "!!! TIMEOUT !!!": "!!! CANCELLED !!!");
- return false;
+ case 0: log(QLatin1String("DONE: Transfer completed successfully."), ""); break;
+ case 6: log(QLatin1String("ERROR: Remote host could not be resolved!"), ""); break;
+ case 7: log(QLatin1String("ERROR: Connection to remote host could not be established!"), ""); break;
+ case 22: log(QLatin1String("ERROR: Requested URL was not found or returned an error!"), ""); break;
+ case 28: log(QLatin1String("ERROR: Operation timed out !!!"), ""); break;
+ default: log(QString().sprintf("ERROR: Terminated with unknown code %d", exitCode), ""); break;
}
}
- timer.stop();
- timer.disconnect(&timer, SIGNAL(timeout()), &loop, SLOT(quit()));
-
- log(QString().sprintf("Exited with code %d", process.exitCode()));
- return (process.exitCode() == 0) && output.exists() && output.isFile();
+ return (exitCode == 0); /*completed*/
}
-bool MUtils::UpdateChecker::tryContactHost(const QString &hostname, const int &timeoutMsec, quint32 *const ipAddrOut)
+int MUtils::UpdateChecker::execProcess(const QString &programFile, const QStringList &args, const QString &workingDir, const int timeout)
{
- log(QString("Connecting to host: %1").arg(hostname), "");
-
QProcess process;
- init_process(process, temp_folder());
-
- QStringList args;
- args << "--retry" << QString::number(3) << hostname << QString::number(80);
+ init_process(process, workingDir, true, NULL, m_environment.data());
QEventLoop loop;
connect(&process, SIGNAL(error(QProcess::ProcessError)), &loop, SLOT(quit()));
timer.setSingleShot(true);
connect(&timer, SIGNAL(timeout()), &loop, SLOT(quit()));
- QScopedPointer<QRegExp> ipAddr;
- if (ipAddrOut)
- {
- *ipAddrOut = 0;
- ipAddr.reset(new QRegExp("Connecting\\s+to\\s+(\\d+)\\.(\\d+)\\.(\\d+)\\.(\\d+):(\\d+)", Qt::CaseInsensitive));
- }
-
- process.start(m_binaryMCat, args);
-
+ process.start(programFile, args);
if (!process.waitForStarted())
{
- return false;
+ log("PROCESS FAILED TO START !!!", "");
+ qWarning("WARNING: %s process could not be created!", MUTILS_UTF8(QFileInfo(programFile).fileName()));
+ return INT_MAX; /*failed to start*/
}
- timer.start(timeoutMsec);
+ bool bAborted = false;
+ timer.start(qMax(timeout, 1500));
while (process.state() != QProcess::NotRunning)
{
loop.exec();
- const bool bTimeOut = (!timer.isActive());
while (process.canReadLine())
{
const QString line = QString::fromLatin1(process.readLine()).simplified();
- if (!ipAddr.isNull())
+ if (line.length() > 1)
{
- if (ipAddr->indexIn(line) >= 0)
- {
- quint32 values[4];
- if (MUtils::regexp_parse_uint32((*ipAddr), values, 4))
- {
- *ipAddrOut |= ((values[0] & 0xFF) << 0x18);
- *ipAddrOut |= ((values[1] & 0xFF) << 0x10);
- *ipAddrOut |= ((values[2] & 0xFF) << 0x08);
- *ipAddrOut |= ((values[3] & 0xFF) << 0x00);
- }
- }
+ log(line);
}
- log(line);
}
- if (bTimeOut || MUTILS_BOOLIFY(m_cancelled))
+ const bool bCancelled = MUTILS_BOOLIFY(m_cancelled);
+ if (bAborted = (bCancelled || ((!timer.isActive()) && (!process.waitForFinished(125)))))
{
- qWarning("MCat process timed out <-- killing!");
- process.kill();
- process.waitForFinished();
- log(bTimeOut ? "!!! TIMEOUT !!!" : "!!! CANCELLED !!!");
- return false;
+ log(bCancelled ? "CANCELLED BY USER !!!" : "PROCESS TIMEOUT !!!", "");
+ qWarning("WARNING: %s process %s!", MUTILS_UTF8(QFileInfo(programFile).fileName()), bCancelled ? "cancelled" : "timed out");
+ break; /*abort process*/
}
}
timer.stop();
timer.disconnect(&timer, SIGNAL(timeout()), &loop, SLOT(quit()));
- if (process.exitCode() != 0)
- {
- log("Connection has failed!");
- }
-
- log(QString().sprintf("Exited with code %d", process.exitCode()), "");
- return (process.exitCode() == 0);
-}
-
-bool MUtils::UpdateChecker::checkSignature(const QString &file, const QString &signature)
-{
- if (QFileInfo(file).absolutePath().compare(QFileInfo(signature).absolutePath(), Qt::CaseInsensitive) != 0)
- {
- qWarning("CheckSignature: File and signature should be in same folder!");
- return false;
- }
-
- QString keyRingPath(m_binaryKeys);
- bool removeKeyring = false;
- if (QFileInfo(file).absolutePath().compare(QFileInfo(m_binaryKeys).absolutePath(), Qt::CaseInsensitive) != 0)
- {
- keyRingPath = make_temp_file(QFileInfo(file).absolutePath(), "gpg");
- removeKeyring = true;
- if (!QFile::copy(m_binaryKeys, keyRingPath))
- {
- qWarning("CheckSignature: Failed to copy the key-ring file!");
- return false;
- }
- }
-
- QProcess process;
- init_process(process, QFileInfo(file).absolutePath());
-
- QEventLoop loop;
- connect(&process, SIGNAL(error(QProcess::ProcessError)), &loop, SLOT(quit()));
- connect(&process, SIGNAL(finished(int, QProcess::ExitStatus)), &loop, SLOT(quit()));
- connect(&process, SIGNAL(readyRead()), &loop, SLOT(quit()));
-
- process.start(m_binaryGnuPG, QStringList() << "--homedir" << "." << "--keyring" << QFileInfo(keyRingPath).fileName() << QFileInfo(signature).fileName() << QFileInfo(file).fileName());
-
- if (!process.waitForStarted())
+ if (bAborted)
{
- if (removeKeyring)
- {
- remove_file(keyRingPath);
- }
- return false;
+ process.kill();
+ process.waitForFinished(-1);
}
- while (process.state() == QProcess::Running)
+ while (process.canReadLine())
{
- loop.exec();
- while (process.canReadLine())
+ const QString line = QString::fromLatin1(process.readLine()).simplified();
+ if (line.length() > 1)
{
- log(QString::fromLatin1(process.readLine()).simplified());
+ log(line);
}
}
- if (removeKeyring)
- {
- remove_file(keyRingPath);
- }
-
- log(QString().sprintf("Exited with code %d", process.exitCode()));
- return (process.exitCode() == 0);
+ return bAborted ? INT_MAX : process.exitCode();
}
////////////////////////////////////////////////////////////
OSDN Git Service
