281 lines
9.1 KiB
C++
281 lines
9.1 KiB
C++
/*
|
|
* Copyright (C) 2020-2022 Roy Qu (royqh1979@gmail.com)
|
|
*
|
|
* 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
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
#include "ojproblemcasesrunner.h"
|
|
#include "../utils.h"
|
|
#include "../settings.h"
|
|
#include "../systemconsts.h"
|
|
#include <QElapsedTimer>
|
|
#include <QProcess>
|
|
#ifdef Q_OS_WINDOWS
|
|
#include <psapi.h>
|
|
#endif
|
|
|
|
|
|
OJProblemCasesRunner::OJProblemCasesRunner(const QString& filename, const QStringList& arguments, const QString& workDir,
|
|
const QVector<POJProblemCase>& problemCases, QObject *parent):
|
|
Runner(filename,arguments,workDir,parent),
|
|
mExecTimeout(0),
|
|
mMemoryLimit(0)
|
|
{
|
|
mProblemCases = problemCases;
|
|
mBufferSize = 8192;
|
|
mOutputRefreshTime = 1000;
|
|
setWaitForFinishTime(100);
|
|
}
|
|
|
|
OJProblemCasesRunner::OJProblemCasesRunner(const QString& filename, const QStringList& arguments, const QString& workDir,
|
|
POJProblemCase problemCase, QObject *parent):
|
|
Runner(filename,arguments,workDir,parent),
|
|
mExecTimeout(0),
|
|
mMemoryLimit(0)
|
|
{
|
|
mProblemCases.append(problemCase);
|
|
mBufferSize = 8192;
|
|
mOutputRefreshTime = 1000;
|
|
setWaitForFinishTime(100);
|
|
}
|
|
|
|
void OJProblemCasesRunner::runCase(int index,POJProblemCase problemCase)
|
|
{
|
|
emit caseStarted(problemCase->getId(),index, mProblemCases.count());
|
|
auto action = finally([this,&index, &problemCase]{
|
|
emit caseFinished(problemCase->getId(), index, mProblemCases.count());
|
|
});
|
|
QProcess process;
|
|
bool errorOccurred = false;
|
|
QByteArray readed;
|
|
QByteArray buffer;
|
|
QByteArray output;
|
|
int noOutputTime = 0;
|
|
QElapsedTimer elapsedTimer;
|
|
bool execTimeouted = false;
|
|
process.setProgram(mFilename);
|
|
process.setArguments(mArguments);
|
|
process.setWorkingDirectory(mWorkDir);
|
|
QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
|
|
QString path = env.value("PATH");
|
|
QStringList pathAdded;
|
|
bool writeChannelClosed = false;
|
|
if (pSettings->compilerSets().defaultSet()) {
|
|
foreach(const QString& dir, pSettings->compilerSets().defaultSet()->binDirs()) {
|
|
pathAdded.append(dir);
|
|
}
|
|
}
|
|
pathAdded.append(pSettings->dirs().appDir());
|
|
if (!path.isEmpty()) {
|
|
path= pathAdded.join(PATH_SEPARATOR) + PATH_SEPARATOR + path;
|
|
} else {
|
|
path = pathAdded.join(PATH_SEPARATOR);
|
|
}
|
|
env.insert("PATH",path);
|
|
process.setProcessEnvironment(env);
|
|
if (pSettings->executor().redirectStderrToToolLog()) {
|
|
emit logStderrOutput("\n");
|
|
} else {
|
|
process.setProcessChannelMode(QProcess::MergedChannels);
|
|
process.setReadChannel(QProcess::StandardOutput);
|
|
}
|
|
process.connect(
|
|
&process, &QProcess::errorOccurred,
|
|
[&](){
|
|
errorOccurred= true;
|
|
});
|
|
problemCase->output.clear();
|
|
process.start();
|
|
process.waitForStarted(5000);
|
|
#ifdef Q_OS_WIN
|
|
HANDLE hProcess = NULL;
|
|
if (process.processId()!=0) {
|
|
hProcess = OpenProcess(PROCESS_ALL_ACCESS,FALSE,process.processId());
|
|
}
|
|
#endif
|
|
if (process.state()==QProcess::Running) {
|
|
if (fileExists(problemCase->inputFileName))
|
|
process.write(readFileToByteArray(problemCase->inputFileName));
|
|
else
|
|
process.write(problemCase->input.toUtf8());
|
|
process.waitForFinished(0);
|
|
}
|
|
|
|
elapsedTimer.start();
|
|
while (true) {
|
|
if (process.bytesToWrite()==0 && !writeChannelClosed) {
|
|
writeChannelClosed = true;
|
|
process.closeWriteChannel();
|
|
}
|
|
process.waitForFinished(mWaitForFinishTime);
|
|
if (process.state()!=QProcess::Running) {
|
|
break;
|
|
}
|
|
if (mExecTimeout>0) {
|
|
int msec = elapsedTimer.elapsed();
|
|
if (msec>mExecTimeout) {
|
|
execTimeouted=true;
|
|
}
|
|
}
|
|
if (mStop || execTimeouted) {
|
|
process.terminate();
|
|
process.kill();
|
|
break;
|
|
}
|
|
if (errorOccurred)
|
|
break;
|
|
if (pSettings->executor().redirectStderrToToolLog()) {
|
|
QString s = QString::fromLocal8Bit(process.readAllStandardError());
|
|
if (!s.isEmpty())
|
|
emit logStderrOutput(s);
|
|
}
|
|
readed = process.read(mBufferSize);
|
|
buffer += readed;
|
|
if (buffer.length()>=mBufferSize || noOutputTime > mOutputRefreshTime) {
|
|
if (!buffer.isEmpty()) {
|
|
emit newOutputGetted(problemCase->getId(),QString::fromLocal8Bit(buffer));
|
|
output.append(buffer);
|
|
buffer.clear();
|
|
}
|
|
noOutputTime = 0;
|
|
} else {
|
|
noOutputTime += mWaitForFinishTime;
|
|
}
|
|
}
|
|
problemCase->runningTime=elapsedTimer.elapsed();
|
|
problemCase->runningMemory = 0;
|
|
#ifdef Q_OS_WIN
|
|
if (hProcess!=NULL) {
|
|
PROCESS_MEMORY_COUNTERS counter{0};
|
|
counter.cb = sizeof(counter);
|
|
if (GetProcessMemoryInfo(hProcess,&counter,
|
|
sizeof(counter))){
|
|
problemCase->runningMemory = counter.PeakWorkingSetSize;
|
|
}
|
|
FILETIME creationTime;
|
|
FILETIME exitTime;
|
|
FILETIME kernelTime;
|
|
FILETIME userTime;
|
|
if (GetProcessTimes(hProcess,&creationTime,&exitTime,&kernelTime,&userTime)) {
|
|
LONGLONG t=((LONGLONG)kernelTime.dwHighDateTime<<32)
|
|
+((LONGLONG)userTime.dwHighDateTime<<32)
|
|
+(kernelTime.dwLowDateTime)+(userTime.dwLowDateTime);
|
|
problemCase->runningTime=(double)t/10000;
|
|
}
|
|
}
|
|
#endif
|
|
if (execTimeouted) {
|
|
problemCase->output = tr("Time limit exceeded!");
|
|
emit resetOutput(problemCase->getId(), problemCase->output);
|
|
} else if (mMemoryLimit>0 && problemCase->runningMemory>mMemoryLimit) {
|
|
problemCase->output = tr("Memory limit exceeded!");
|
|
emit resetOutput(problemCase->getId(), problemCase->output);
|
|
} else {
|
|
if (pSettings->executor().redirectStderrToToolLog()) {
|
|
QString s = QString::fromLocal8Bit(process.readAllStandardError());
|
|
if (!s.isEmpty())
|
|
emit logStderrOutput(s);
|
|
}
|
|
if (process.state() == QProcess::ProcessState::NotRunning)
|
|
buffer += process.readAll();
|
|
emit newOutputGetted(problemCase->getId(),QString::fromLocal8Bit(buffer));
|
|
output.append(buffer);
|
|
problemCase->output = QString::fromLocal8Bit(output);
|
|
|
|
if (errorOccurred) {
|
|
//qDebug()<<"process error:"<<process.error();
|
|
switch (process.error()) {
|
|
case QProcess::FailedToStart:
|
|
emit runErrorOccurred(tr("The runner process '%1' failed to start.").arg(mFilename));
|
|
break;
|
|
// case QProcess::Crashed:
|
|
// if (!mStop)
|
|
// emit runErrorOccurred(tr("The runner process crashed after starting successfully."));
|
|
// break;
|
|
case QProcess::Timedout:
|
|
emit runErrorOccurred(tr("The last waitFor...() function timed out."));
|
|
break;
|
|
case QProcess::WriteError:
|
|
emit runErrorOccurred(tr("An error occurred when attempting to write to the runner process."));
|
|
break;
|
|
case QProcess::ReadError:
|
|
emit runErrorOccurred(tr("An error occurred when attempting to read from the runner process."));
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void OJProblemCasesRunner::run()
|
|
{
|
|
emit started();
|
|
auto action = finally([this]{
|
|
emit terminated();
|
|
});
|
|
for (int i=0; i < mProblemCases.size(); i++) {
|
|
if (mStop)
|
|
break;
|
|
POJProblemCase problemCase = mProblemCases[i];
|
|
runCase(i,problemCase);
|
|
}
|
|
}
|
|
|
|
int OJProblemCasesRunner::execTimeout() const
|
|
{
|
|
return mExecTimeout;
|
|
}
|
|
|
|
void OJProblemCasesRunner::setExecTimeout(int newExecTimeout)
|
|
{
|
|
mExecTimeout = newExecTimeout;
|
|
}
|
|
|
|
void OJProblemCasesRunner::setMemoryLimit(size_t limit)
|
|
{
|
|
mMemoryLimit = limit;
|
|
}
|
|
|
|
int OJProblemCasesRunner::waitForFinishTime() const
|
|
{
|
|
return mWaitForFinishTime;
|
|
}
|
|
|
|
void OJProblemCasesRunner::setWaitForFinishTime(int newWaitForFinishTime)
|
|
{
|
|
mWaitForFinishTime = newWaitForFinishTime;
|
|
}
|
|
|
|
int OJProblemCasesRunner::outputRefreshTime() const
|
|
{
|
|
return mOutputRefreshTime;
|
|
}
|
|
|
|
void OJProblemCasesRunner::setOutputRefreshTime(int newOutputRefreshTime)
|
|
{
|
|
mOutputRefreshTime = newOutputRefreshTime;
|
|
}
|
|
|
|
int OJProblemCasesRunner::bufferSize() const
|
|
{
|
|
return mBufferSize;
|
|
}
|
|
|
|
void OJProblemCasesRunner::setBufferSize(int newBufferSize)
|
|
{
|
|
mBufferSize = newBufferSize;
|
|
}
|
|
|
|
|