Mascali
/
piston
mirror of https://github.com/engineer-man/piston.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Stop appending to output if longer than limit

This commit is contained in:
Vrganj 2021-01-23 20:10:10 +01:00
parent 563cf19068
commit b64598763d
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
shared/execute.js
View File

@ -1,6 +1,8 @@
const { writeFileSync, unlinkSync } = require('fs'); const { writeFileSync, unlinkSync } = require('fs');
const { spawn } = require('child_process'); const { spawn } = require('child_process');
const OUTPUT_LIMIT = 65535;
function execute(language, source, stdin = '', args = []) { function execute(language, source, stdin = '', args = []) {
return new Promise(resolve => { return new Promise(resolve => {
const stamp = new Date().getTime(); const stamp = new Date().getTime();
@ -20,11 +22,15 @@ function execute(language, source, stdin = '', args = []) {
let output = ''; let output = '';
process.stderr.on('data', chunk => { process.stderr.on('data', chunk => {
if (stderr.length >= OUTPUT_LIMIT) return;
stderr += chunk; stderr += chunk;
output += chunk; output += chunk;
}); });
process.stdout.on('data', chunk => { process.stdout.on('data', chunk => {
if (stdout.length >= OUTPUT_LIMIT) return;
stdout += chunk; stdout += chunk;
output += chunk; output += chunk;
}); });
@ -32,9 +38,9 @@ function execute(language, source, stdin = '', args = []) {
process.on('exit', code => { process.on('exit', code => {
unlinkSync(sourceFile); unlinkSync(sourceFile);
stderr = stderr.trim().substring(0, 65535); stderr = stderr.trim().substring(0, OUTPUT_LIMIT);
stdout = stdout.trim().substring(0, 65535); stdout = stdout.trim().substring(0, OUTPUT_LIMIT);
output = output.trim().substring(0, 65535); output = output.trim().substring(0, OUTPUT_LIMIT);
resolve({ resolve({
stdout, stdout,