160 lines
3.7 KiB
C++
160 lines
3.7 KiB
C++
#include "JavaChecker.h"
|
|
#include <FileSystem.h>
|
|
#include <Commandline.h>
|
|
#include <QFile>
|
|
#include <QProcess>
|
|
#include <QMap>
|
|
#include <QCoreApplication>
|
|
#include <QDebug>
|
|
|
|
JavaChecker::JavaChecker(QObject *parent) : QObject(parent)
|
|
{
|
|
}
|
|
|
|
void JavaChecker::performCheck()
|
|
{
|
|
QString checkerJar = FS::PathCombine(QCoreApplication::applicationDirPath(), "jars", "JavaCheck.jar");
|
|
|
|
QStringList args;
|
|
|
|
process.reset(new QProcess());
|
|
if(m_args.size())
|
|
{
|
|
auto extraArgs = Commandline::splitArgs(m_args);
|
|
args.append(extraArgs);
|
|
}
|
|
if(m_minMem != 0)
|
|
{
|
|
args << QString("-Xms%1m").arg(m_minMem);
|
|
}
|
|
if(m_maxMem != 0)
|
|
{
|
|
args << QString("-Xmx%1m").arg(m_maxMem);
|
|
}
|
|
if(m_permGen != 64)
|
|
{
|
|
args << QString("-XX:PermSize=%1m").arg(m_permGen);
|
|
}
|
|
|
|
args.append({"-jar", checkerJar});
|
|
process->setArguments(args);
|
|
process->setProgram(m_path);
|
|
process->setProcessChannelMode(QProcess::SeparateChannels);
|
|
qDebug() << "Running java checker: " + m_path + args.join(" ");;
|
|
|
|
connect(process.get(), SIGNAL(finished(int, QProcess::ExitStatus)), this, SLOT(finished(int, QProcess::ExitStatus)));
|
|
connect(process.get(), SIGNAL(error(QProcess::ProcessError)), this, SLOT(error(QProcess::ProcessError)));
|
|
connect(process.get(), SIGNAL(readyReadStandardOutput()), this, SLOT(stdoutReady()));
|
|
connect(process.get(), SIGNAL(readyReadStandardError()), this, SLOT(stderrReady()));
|
|
connect(&killTimer, SIGNAL(timeout()), SLOT(timeout()));
|
|
killTimer.setSingleShot(true);
|
|
killTimer.start(15000);
|
|
process->start();
|
|
}
|
|
|
|
void JavaChecker::stdoutReady()
|
|
{
|
|
QByteArray data = process->readAllStandardOutput();
|
|
QString added = QString::fromLocal8Bit(data);
|
|
added.remove('\r');
|
|
m_stdout += added;
|
|
}
|
|
|
|
void JavaChecker::stderrReady()
|
|
{
|
|
QByteArray data = process->readAllStandardError();
|
|
QString added = QString::fromLocal8Bit(data);
|
|
added.remove('\r');
|
|
m_stderr += added;
|
|
}
|
|
|
|
void JavaChecker::finished(int exitcode, QProcess::ExitStatus status)
|
|
{
|
|
killTimer.stop();
|
|
QProcessPtr _process;
|
|
_process.swap(process);
|
|
|
|
JavaCheckResult result;
|
|
{
|
|
result.path = m_path;
|
|
result.id = m_id;
|
|
}
|
|
result.errorLog = m_stderr;
|
|
qDebug() << "STDOUT" << m_stdout;
|
|
qWarning() << "STDERR" << m_stderr;
|
|
qDebug() << "Java checker finished with status " << status << " exit code " << exitcode;
|
|
|
|
if (status == QProcess::CrashExit || exitcode == 1)
|
|
{
|
|
qDebug() << "Java checker failed!";
|
|
emit checkFinished(result);
|
|
return;
|
|
}
|
|
|
|
bool success = true;
|
|
|
|
QMap<QString, QString> results;
|
|
QStringList lines = m_stdout.split("\n", QString::SkipEmptyParts);
|
|
for(QString line : lines)
|
|
{
|
|
line = line.trimmed();
|
|
|
|
auto parts = line.split('=', QString::SkipEmptyParts);
|
|
if(parts.size() != 2 || parts[0].isEmpty() || parts[1].isEmpty())
|
|
{
|
|
success = false;
|
|
}
|
|
else
|
|
{
|
|
results.insert(parts[0], parts[1]);
|
|
}
|
|
}
|
|
|
|
if(!results.contains("os.arch") || !results.contains("java.version") || !success)
|
|
{
|
|
qDebug() << "Java checker failed - couldn't extract required information.";
|
|
emit checkFinished(result);
|
|
return;
|
|
}
|
|
|
|
auto os_arch = results["os.arch"];
|
|
auto java_version = results["java.version"];
|
|
bool is_64 = os_arch == "x86_64" || os_arch == "amd64";
|
|
|
|
|
|
result.valid = true;
|
|
result.is_64bit = is_64;
|
|
result.mojangPlatform = is_64 ? "64" : "32";
|
|
result.realPlatform = os_arch;
|
|
result.javaVersion = java_version;
|
|
qDebug() << "Java checker succeeded.";
|
|
emit checkFinished(result);
|
|
}
|
|
|
|
void JavaChecker::error(QProcess::ProcessError err)
|
|
{
|
|
if(err == QProcess::FailedToStart)
|
|
{
|
|
killTimer.stop();
|
|
qDebug() << "Java checker has failed to start.";
|
|
JavaCheckResult result;
|
|
{
|
|
result.path = m_path;
|
|
result.id = m_id;
|
|
}
|
|
|
|
emit checkFinished(result);
|
|
return;
|
|
}
|
|
}
|
|
|
|
void JavaChecker::timeout()
|
|
{
|
|
// NO MERCY. NO ABUSE.
|
|
if(process)
|
|
{
|
|
qDebug() << "Java checker has been killed by timeout.";
|
|
process->kill();
|
|
}
|
|
}
|