/**
* Executes a command as {@code user}. Saves the stdout/stderr in the corresponding fields and
* returns the return code of the child process.
*
* @param cmd The command to execute
* @param data The data to send to the stdin of the process
* @param timelimt How many seconds the command can run
* @param memlimit How many megabytes the command can use
*/
private int exec(String cmdS, String data, int timelimit, int memlimit) throws Exception {
ArrayList<String> cmd = new ArrayList<String>(cmdPrefix);
cmd.add("-m");
cmd.add("" + memlimit);
cmd.add("-c");
cmd.add("" + timelimit);
cmd.add("-w");
cmd.add("" + (3 * timelimit + 1));
cmd.add("-x");
cmd.add(cmdS);
String tmp = "";
for (String cs : cmd) tmp += " \"" + cs + "\"";
log.fine("exec " + tmp);
ProcessBuilder pb = new ProcessBuilder(cmd);
pb.directory(workDir);
Process p = pb.start();
OutputStreamWriter writer = new OutputStreamWriter(p.getOutputStream());
StreamReader rOut = new StreamReader(p.getInputStream());
StreamReader rErr = new StreamReader(p.getErrorStream());
rOut.start();
rErr.start();
// TODO I think this may block for big tests. Check and Fix.
// send and receive data "in parallel"
writer.write(data);
writer.flush();
writer.close();
rOut.join();
rErr.join();
stdout = rOut.result;
stderr = rErr.result;
if (rOut.exception != null) throw rOut.exception;
if (rErr.exception != null) throw rErr.exception;
// log.finer("out: " + stdout);
// log.finer("err: " + stderr);
// log.finer("done exec");
return p.waitFor();
}
public boolean ok(String out, String reference) {
try {
File fo = File.createTempFile("homeworkeval", ".out");
File fr = File.createTempFile("homeworkeval", ".ref");
log.fine("Running custom validator.");
save(out, fo);
save(reference, fr);
ArrayList<String> c = new ArrayList<String>();
c.add(command);
c.add(fo.getPath());
c.add(fr.getPath());
ProcessBuilder pb = new ProcessBuilder(c);
Process p = pb.start();
int rc = p.waitFor();
fo.delete();
fr.delete();
log.fine("Custom validator " + command + " says " + rc);
return rc == 0;
} catch (Exception e) {
// Assume NOK. Don't propagate info to the client.
log.finer("exc: " + UtilSrv.describe(e));
return false;
}
}