[sandbox] Add cpp build

This commit is contained in:
orosmatthew 2024-02-17 15:01:29 -05:00
parent 245b987db9
commit 6e673cf390
3 changed files with 147 additions and 3 deletions

View File

@ -12,7 +12,7 @@ ENV NODE_MAJOR=18
RUN echo "deb [signed-by=/etc/apt/keyrings/nodesource.gpg] https://deb.nodesource.com/node_$NODE_MAJOR.x nodistro main" | tee /etc/apt/sources.list.d/nodesource.list
RUN apt-get update
RUN apt-get install nodejs git openjdk-17-jdk-headless dotnet-sdk-7.0 -y
RUN apt-get install nodejs git openjdk-17-jdk-headless dotnet-sdk-7.0 build-essential cmake -y
ENV DOTNET_NOLOGO=true
ENV DOTNET_SKIP_FIRST_TIME_EXPERIENCE=true

View File

@ -7,10 +7,11 @@ import { join } from 'path';
import { simpleGit, SimpleGit } from 'simple-git';
import { runJava } from './run/java.js';
import { runCSharp } from './run/csharp.js';
import { runCpp } from './run/cpp.js';
export const timeoutSeconds = 30;
const RunResultKind = z.enum(['CompileFailed', 'TimeLimitExceeded', 'Completed', 'SandboxError']);
const RunResultKind = z.enum(['CompileFailed', 'TimeLimitExceeded', 'Completed', 'SandboxError', 'RunError']);
export type RunResultKind = z.infer<typeof RunResultKind>;
const RunResult = z
@ -40,7 +41,7 @@ const submissionGetData = z
contestName: z.string(),
teamId: z.number(),
teamName: z.string(),
teamLanguage: z.enum(['Java', 'CSharp']),
teamLanguage: z.enum(['Java', 'CSharp', 'CPP']),
problem: z.object({
id: z.number(),
pascalName: z.string(),
@ -127,6 +128,18 @@ async function cloneAndRun(submissionData: SubmissionGetData) {
} else {
runResult = res.runResult;
}
} else if (submissionData.submission.teamLanguage === 'CPP') {
let res = await runCpp({
srcDir: repoDir,
input: submissionData.submission.problem.realInput,
cppPlatform: 'GCC',
problemName: submissionData.submission.problem.pascalName
});
if (res.success === true) {
runResult = await res.runResult;
} else {
runResult = res.runResult;
}
}
} catch (error) {
runResult = {

131
sandbox/src/run/cpp.ts Normal file
View File

@ -0,0 +1,131 @@
import { join } from 'path';
import { exec, spawn } from 'child_process';
import util from 'util';
import { RunResult, timeoutSeconds } from '../index.js';
import { IRunner, IRunnerParams, IRunnerReturn } from './types.js';
import kill from 'tree-kill';
import * as fs from 'fs';
const execPromise = util.promisify(exec);
export type CppPlatform = 'VisualStudio' | 'GCC';
interface IRunnerParamsCpp extends IRunnerParams {
srcDir: string;
problemName: string;
input: string;
cppPlatform: CppPlatform;
outputCallback?: (data: string) => void;
}
export const runCpp: IRunner<IRunnerParamsCpp> = async function (
params: IRunnerParamsCpp
): IRunnerReturn {
console.log(`- BUILD: ${params.problemName}`);
const configureCommand = `cmake -S ${params.srcDir} -B ${join(params.srcDir, 'build')}`;
try {
await execPromise(configureCommand);
} catch (e) {
const buildErrorText = e?.toString() ?? 'Unknown build errors.';
console.log('Build errors: ' + buildErrorText);
return {
success: false,
runResult: { kind: 'CompileFailed', resultKindReason: buildErrorText }
};
}
const compileCommand = `cmake --build ${join(params.srcDir, 'build')} --target ${params.problemName}`;
try {
await execPromise(compileCommand);
} catch (e) {
const buildErrorText = e?.toString() ?? 'Unknown build errors.';
console.log('Build errors: ' + buildErrorText);
return {
success: false,
runResult: { kind: 'CompileFailed', resultKindReason: buildErrorText }
};
}
console.log(`- RUN: ${params.problemName}`);
let runCommand = '';
if (params.cppPlatform === 'VisualStudio') {
runCommand = `${join(params.srcDir, 'build', 'Debug', `${params.problemName}.exe`)}`;
} else {
runCommand = `${join(params.srcDir, 'build', params.problemName)}`;
}
try {
let outputBuffer = '';
const child = spawn(runCommand, { shell: true });
child.stdout.setEncoding('utf8');
child.stdout.on('data', (data) => {
outputBuffer += data.toString();
});
child.stderr.setEncoding('utf8');
child.stderr.on('data', (data) => {
outputBuffer += data.toString();
});
let runStartTime = performance.now();
child.stdin.write(params.input);
child.stdin.end();
let timeLimitExceeded = false;
let completedNormally = false;
return {
success: true,
runResult: new Promise<RunResult>(async (resolve) => {
child.on('close', () => {
completedNormally = !timeLimitExceeded;
let runEndTime = performance.now();
const runtimeMilliseconds = Math.floor(runEndTime - runStartTime);
if (completedNormally) {
clearTimeout(timeoutHandle);
resolve({
kind: 'Completed',
output: outputBuffer,
exitCode: child.exitCode!,
runtimeMilliseconds
});
} else {
console.log(`Process terminated, total sandbox time: ${runtimeMilliseconds}ms`);
resolve({
kind: 'TimeLimitExceeded',
output: outputBuffer,
resultKindReason: `Timeout after ${timeoutSeconds} seconds`
});
}
});
let timeoutHandle = setTimeout(() => {
if (completedNormally) {
return;
}
console.log(`Run timed out after ${timeoutSeconds} seconds, killing process...`);
timeLimitExceeded = true;
child.stdin.end();
child.stdin.destroy();
child.stdout.destroy();
child.stderr.destroy();
child.kill('SIGKILL');
}, timeoutSeconds * 1000);
}),
killFunc() {
if (child.pid !== undefined) {
if (!completedNormally && !timeLimitExceeded) {
kill(child.pid);
params.outputCallback?.('\n[Manually stopped]');
}
}
}
};
} catch (error) {
return { success: false, runResult: { kind: 'RunError' } };
}
};