<?php
////////////////////////////////////////////////////////////////////////////////
//BOCA Online Contest Administrator
//    Copyright (C) 2003-2012 by BOCA Development Team (bocasystem@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 <http://www.gnu.org/licenses/>.
////////////////////////////////////////////////////////////////////////////////
// Last modified 05/aug/2012 by cassio@ime.usp.br
$ds = DIRECTORY_SEPARATOR;
if($ds=="") $ds = "/";
if(is_readable(getcwd() . $ds . '..' .$ds . 'db.php')) {
	require_once(getcwd() . $ds . '..' .$ds . 'db.php');
	require_once(getcwd() . $ds . '..' .$ds . 'version.php');
} else {
  if(is_readable(getcwd() . $ds . 'db.php')) {
	require_once(getcwd() . $ds . 'db.php');
	require_once(getcwd() . $ds . 'version.php');
  } else {
	  echo "unable to find db.php";
	  exit;
  }
}
if (getIP()!="UNKNOWN" || php_sapi_name()!=="cli") exit;
if(system('test "`id -u`" -eq "0"',$retval)===false || $retval!=0) {
	echo "Must be run as root\n";
	exit;
}

ini_set('memory_limit','600M');
ini_set('output_buffering','off');
ini_set('implicit_flush','on');
@ob_end_flush();
echo "max memory set to " . ini_get('memory_limit'). "\n";

$tmpdir = getenv("TMP");
if($tmpdir=="") $tmpdir = getenv("TMPDIR");
if($tmpdir[0] != '/') $tmdir = "/tmp";
if($tmpdir=="") $tmpdir = "/tmp";

$basdir=$ds;
if(file_exists($ds . 'bocajail' . $tmpdir)) {
  $tmpdir=$ds . 'bocajail' . $tmpdir;
  $basdir=$ds . 'bocajail' . $ds;
  echo "bocajail environment seems to exist - trying to use it\n";
} else {
  echo "bocajail not found - trying to proceed without using it\n";
}

if($ds=='/') {
	system("find $basdir -user bocajail -delete >/dev/null 2>/dev/null");
	system("find $basdir -user nobody -delete >/dev/null 2>/dev/null");
	system("find $basdir -group users -exec chgrp root '{}' \\; 2>/dev/null");
	system("find $basdir -perm /1002 -type d > /tmp/boca.writabledirs.tmp 2>/dev/null");
	system('chmod 400 /tmp/boca.writabledirs.tmp 2>/dev/null');
}
umask(0022);

$cache = $tmpdir . $ds . "bocacache.d";
cleardir($cache);
@mkdir($cache);
$key=md5(mt_rand() . rand() . mt_rand());

$cf = globalconf();
$ip = $cf["ip"];
$activecontest=DBGetActiveContest();
$prevsleep=0;
while(42) {
if(($run = DBGetRunToAutojudging($activecontest["contestnumber"], $ip)) === false) {
  if($prevsleep==0)
    echo "Nothing to do. Sleeping...";
  else
    echo ".";
  flush();
  sleep(10);
  $prevsleep=1;
  continue;
}
echo "\n";
flush();
$prevsleep=0;

$number=$run["number"];
$site=$run["site"];
$contest=$run["contest"];

echo "Removing possible files from previous runs\n";
$dirs=file('/tmp/boca.writabledirs.tmp');
for($dir=0;$dir<count($dirs);$dir++) {
	$dirn=trim($dirs[$dir]) . $ds;
	if($dirn[0] != '/') continue;
	system("find \"$dirn\" -user bocajail -delete >/dev/null 2>/dev/null");
	system("find \"$dirn\" -user nobody -delete >/dev/null 2>/dev/null");
}

echo "Entering directory $tmpdir (contest=$contest, site=$site, run=$number)\n";
chdir($tmpdir);
for($i=0; $i<5; $i++) {
	  $name = tempnam($tmpdir, "boca");
	  $dir = $name . ".d";
	  if(@mkdir($dir, 0755)) break;
	  @unlink($name);
	  @rmdir($dir);
}
if($i>=5) {
	  echo "It was not possible to create a unique temporary directory\n";
  	  LogLevel("Autojuging: Unable to create temp directory (run=$number, site=$site, contest=$contest)",1);
	  DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem creating temp directory");
	  continue;
}
chdir($dir);

echo "Using directory $dir (contest=$contest, site=$site, run=$number)\n";

/*
if($run["scriptoid"]=="" || $run["scriptname"]=="") {
	LogLevel("Autojuging: Script file not defined (run=$number, site=$site, contest=$contest)",1);
        echo "Compiling/running script file not defined (contest=$contest, site=$site, run=$number)\n";
	cleardir($dir);
	unlink($name);
        DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: compiling/running script not defined");
	continue;
}
if($run["compscriptoid"]=="" || $run["compscriptname"]=="") {
	LogLevel("Autojuging: Comparing script file not defined (run=$number, site=$site, contest=$contest)",1);
        echo "Comparing script file not defined (contest=$contest, site=$site, run=$number)\n";
	cleardir($dir);
	unlink($name);
        DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: comparing script not defined");
	continue;
}
*/
if($run["sourceoid"]=="" || $run["sourcename"]=="") {
	LogLevel("Autojuging: Source file not defined (run=$number, site=$site, contest=$contest)",1);
        echo "Source file not defined (contest=$contest, site=$site, run=$number)\n";
	cleardir($dir);
	unlink($name);
        DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: source file not defined");
	continue;
}
if($run["inputoid"]=="" || $run["inputname"]=="") {
	LogLevel("Autojuging: problem package not defined (run=$number, site=$site, contest=$contest)",1);
        echo "Package file not defined (contest=$contest, site=$site, run=$number)\n";
	cleardir($dir);
	unlink($name);
        DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file not defined");
	continue;
}
/*
if($run["soloid"]=="" || $run["solname"]=="") {
	LogLevel("Autojuging: sol file not defined (run=$number, site=$site, contest=$contest)",1);
        echo "solfile not defined (contest=$contest, site=$site, run=$number)\n";
	cleardir($dir);
	unlink($name);
        DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: correct output file not defined");
	continue;
}
*/
$c = DBConnect();
DBExec($c, "begin work", "Autojudging(exporttransaction)");
/*
if(DB_lo_export($c, $run["scriptoid"], $dir . $ds . $run["scriptname"]) === false) {
        DBExec($c, "rollback work", "Autojudging(rollback-script)");
	LogLevel("Autojuging: Unable to export script file (run=$number, site=$site, contest=$contest)",1);
        echo "Error exporting compiling/running script file ${run["scriptname"]} (contest=$contest, site=$site, run=$number)\n";
	cleardir($dir);
	unlink($name);
        DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: unable to export compiling/running script");
	continue;
}
*/
if(DB_lo_export($c, $run["sourceoid"], $dir . $ds . $run["sourcename"]) === false) {
        DBExec($c, "rollback work", "Autojudging(rollback-source)");
	LogLevel("Autojudging: Unable to export source file (run=$number, site=$site, contest=$contest)",1);
        echo "Error exporting source file ${run["sourcename"]} (contest=$contest, site=$site, run=$number)\n";
	cleardir($dir);
	unlink($name);
	DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: unable to export source file");
	DBExec($c, "commit", "Autojudging(exportcommit)");
	continue;
}
if(is_readable($cache . $ds . $run["inputoid"] . "." . $run["inputname"])) {
	DBExec($c, "commit", "Autojudging(exportcommit)");
	echo "Getting problem package file from local cache\n";
	$s = file_get_contents($cache .	$ds . $run["inputoid"]	. "." . $run["inputname"]);
	file_put_contents($dir . $ds . $run["inputname"], decryptData($s,$key));
} else {
	echo "Downloading problem package file from db\n";
	if(DB_lo_export($c, $run["inputoid"], $dir . $ds . $run["inputname"]) === false) {
        DBExec($c, "rollback work", "Autojudging(rollback-input)");
		LogLevel("Autojudging: Unable to export problem package file (run=$number, site=$site, contest=$contest)",1);
        echo "Error exporting problem package file ${run["inputname"]} (contest=$contest, site=$site, run=$number)\n";
		cleardir($dir);
		unlink($name);
		DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: unable to export problem package file");
		DBExec($c, "commit", "Autojudging(exportcommit)");
		continue;
	}
	DBExec($c, "commit", "Autojudging(exportcommit)");

	echo "Problem package downloaded -- running init scripts to obtain limits and other information\n";
	$zip = new ZipArchive;
	if ($zip->open($dir . $ds . $run["inputname"]) === true) {
		$zip->extractTo($dir . $ds . "problemdata");
		$zip->close();
	} else {
		echo 'Failed to unzip the file -- please check the problem package\n';
		DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
		continue;
	}
	if(($info=file($dir . $ds . "problemdata" . $ds . "description" . $ds . 'problem.info'))===false) {
		echo 'Problem content missing (description/problem.info) -- please check the problem package\n';
		DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
		continue;
	}
	$basename='';
	foreach($info as $line) {
		$a=explode('=',$line);
		if(trim($a[0])=='basename') {
			$basename=trim($a[1]);
			break;
		}
	}
	if($basename=='') {
		echo 'Problem content missing (description/problem.info) -- please check the problem package\n';
		DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
		continue;
	}
	if(!is_dir($dir . $ds . "problemdata" . $ds . "limits")) {
		echo 'Problem content missing (limits) -- please check the problem package\n';
		DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
		continue;
	}
	chdir($dir . $ds . "problemdata" . $ds . "limits");
	foreach(glob($dir . $ds . "problemdata" . $ds . "limits" .$ds . '*') as $file) {
		chmod($file,0700);
		$ex = escapeshellcmd($file);
		$ex .= " >stdout 2>stderr";
		echo "Executing INIT SCRIPT " . $ex . "\n";
		if(system($ex, $retval)===false) $retval=-1;
		if($retval != 0) {
			echo 'Error running script -- please check the problem package\n';
			DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
			continue;
		}
		$limits[$basename] = file('stdout');
	}

	foreach(glob($dir . $ds . "problemdata" . $ds . "tests" .$ds . '*') as $file) {
		chdir($dir . $ds . "problemdata" . $ds . "tests");
		chmod($file,0700);
		$ex = escapeshellcmd($file);
		$ex .= " >stdout 2>stderr";
		echo "Executing TEST SCRIPT " . $ex . "\n";
		if(system($ex, $retval)===false) $retval=-1;
		if($retval != 0) {
			echo 'Error running test script -- please check the problem package\n';
			DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: internal test script failed");
			continue;
		}
	}
	cleardir($dir . $ds . "problemdata");
	$s = file_get_contents($dir . $ds . $run["inputname"]);
	file_put_contents($cache . $ds . $run["inputoid"] . "." . $run["inputname"], encryptData($s,$key));
}

function exitmsg($retval) {
/* FROM SAFEEXEC
# 0 ok
# 1 compile error
# 2 runtime error
# 3 timelimit exceeded
# 4 internal error
# 5 parameter error
# 6 internal error
# 7 memory limit exceeded
# 8 security threat
# 9 runtime error
*/
/*
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 0, 'Not answered yet', 'f', 't')", "DBNewContest(insert fake answer)");
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 1, 'YES', 't', 'f')", "DBNewContest(insert YES answer)");
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 2, 'NO - Compilation error', 'f', 'f')", "DBNewContest(insert CE answer)");
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 3, 'NO - Runtime error', 'f', 'f')", "DBNewContest(insert RE answer)");
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 4, 'NO - Time limit exceeded', 'f', 'f')", "DBNewContest(insert TLE answer)");
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 5, 'NO - Presentation error', 'f', 'f')", "DBNewContest(insert PE answer)");
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 6, 'NO - Wrong answer', 'f', 'f')", "DBNewContest(insert WA answer)");
	DBExec($c, "insert into answertable (contestnumber, answernumber, runanswer, yes, fake) values ".
			"($n, 7, 'NO - Contact staff', 'f', 'f')", "DBNewContest(insert CS answer)");
*/
	if($retval==-1) {
		$answer="Internal error while executing run command";
		$retval = 7; // contact staff
	}
	else if($retval==1) {
		$answer="Compilation error";
		$retval = 2; // compilation error
	}
	else if($retval==2) {
		$answer="Runtime error";
		$retval = 3; // runtime error
	}
	else if($retval==3) {
		$answer="Time limit exceeded";
		$retval = 4; // timelimit exceeded
	}
	else if($retval==4) {
		$answer="safeexec internal error (4)";
		$retval = 7; // contact staff
	}
	else if($retval==5) {
		$answer="safeexec error: parameter problem";
		$retval = 7; // contact staff
	}
	else if($retval==6) {
		$answer="safeexec internal error (6)";
		$retval = 7; // contact staff
	}
	else if($retval==7) {
		$answer="Runtime error (memory-limit)";
		$retval = 3; // runtime error
	}
	else if($retval==8) {
		$answer="Code generates security threat";
		$retval = 3; // runtime error
	}
	else if($retval==9) {
		$answer="Runtime error";
		$retval = 3; // runtime error
	}
	return array($retval,$answer);
}

// COMPILATION
//# parameters are:
//# $1 source_file
//# $2 exe_file (default ../run.exe)
//# $3 timelimit (optional, limit to run all the repetitions, by default only one repetition)
//# $4 maximum allowed memory (in MBytes)

$zip = new ZipArchive;
if ($zip->open($dir . $ds . $run["inputname"]) === true) {
	$zip->extractTo($dir, "compile" . $ds . $run["extension"]);
	$zip->close();
} else {
	echo 'Failed to unzip the file -- please check the problem package\n';
	DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
	continue;
}

$script = $dir . $ds . 'compile' . $ds . $run["extension"];
if(!is_file($script)) {
	echo 'Error (not found) compile script for '.$run["extension"].' -- please check the problem package\n';
	DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: compile script failed (".$run["extension"].")");
	continue;
}

chmod($script, 0700);
$ex = escapeshellcmd($script) ." ".
	escapeshellarg($run["sourcename"])." ".
	"run.exe ".
	escapeshellarg(trim($limits[$run["extension"]][0]))." ".
	escapeshellarg(trim($limits[$run["extension"]][2]));
$ex .= " >stdout 2>stderr";
echo "Executing " . $ex . "\n";
if(system($ex, $retval)===false) $retval=-1;

if($retval != 0) {
	list($retval,$answer) = exitmsg($retval);
	$answer = "(WHILE COMPILING) " . $answer;
} else {
//# parameters are:
//# $1 exe_file
//# $2 input_file
//# $3 timelimit (limit to run all the repetitions, by default only one repetition)
//# $4 number_of_repetitions_to_run (optional, can be used for better tuning the timelimit)
//# $5 maximum allowed memory (in KBytes)

	$zip = new ZipArchive;
	if ($zip->open($dir . $ds . $run["inputname"]) === true) {
		$zip->extractTo($dir, array("run" . $ds . $run["extension"],"input" . $ds . '*'));
		$zip->close();
	} else {
		echo 'Failed to unzip the file (inputs) -- please check the problem package\n';
		DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
		continue;
	}
	chdir($dir);
	@unlink('allout');
	@unlink('allerr');
	$retval = 0;
	$script = $dir . $ds . 'run' . $ds . $run["extension"];
	chmod($script, 0700);
	mkdir('team', 0755);
	$d = opendir($dir . $ds . "input");
	while (($file = readdir($d)) !== false) {
		if(is_file($dir . $ds . "input" . $ds . $file)) {
			$ex = escapeshellcmd($script) ." ".
				"run.exe ".
				escapeshellarg($dir . $ds . "input" . $ds . $file)." ".
				escapeshellarg(trim($limit[0]))." ".
				escapeshellarg(trim($limit[1]))." ".
				escapeshellarg(trim($limit[2]));
			$ex .= " >stdout 2>stderr";
			echo "Executing " . $ex . " for input " . $file . "\n";
			if(system($ex, $retval)===false) $retval=-1;
			foreach (glob($dir . $ds . '*') as $fne) {
				@chown($fne,"nobody");
				@chmod($fne,0755);
			}
			if(is_readable('stderr0'))
				system('cat stderr0 >> stderr');
			system('echo ##### STDERR FOR FILE ' . escapeshellarg($file) . ' >> allerr');
			system('cat stderr >> allerr');
			system('cat stdout > team' . $ds . escapeshellarg($file));
			system('echo ##### STDOUT FOR FILE ' . escapeshellarg($file) . ' >> allout');
			system('cat stdout >> allout');
			if($retval != 0) {
				list($retval,$answer) = exitmsg($retval);
				$answer = "(WHILE RUNNING) " . $answer;
				break;
			}
		}
	}
	if($retval==0) {
		echo "Processing results\n";
		$zip = new ZipArchive;
		if ($zip->open($dir . $ds . $run["inputname"]) === true) {
			$zip->extractTo($dir, array("compare" . $ds . $run["extension"],"output" . $ds . '*'));
			$zip->close();
		} else {
			echo 'Failed to unzip the file (outputs) -- please check the problem package\n';
			DBGiveUpRunAutojudging($contest, $site, $number, $ip, "Autojuging error: problem package file is invalid");
			continue;
		}
		$script = $dir . $ds . 'compare' . $ds . $run["extension"];
		$retval = 0;
		chmod($script, 0700);
		$d = opendir($dir . $ds . "output");
		while (($file = readdir($d)) !== false) {
			if(is_file($dir . $ds . "output" . $ds . $file)) {
				$ex = escapeshellcmd($script) ." ".
					escapeshellarg("team" . $ds . $file)." ".
					escapeshellarg("output" . $ds . $file)." ".
					escapeshellarg("input" . $ds . $file) . " >compout";
				echo "Executing " . $ex . "\n";
				$answer = system($ex, $localretval);

				$fp = fopen($dir . $ds . "allerr", "a+");
				fwrite($fp, "\n\n===OUTPUT OF COMPARING SCRIPT FOLLOWS FOR FILE " .$file ." (EMPTY MEANS NO DIFF)===\n");
				$dif = file($dir . $ds . "compout");
				$difi = 0;
				for(; $difi < count($dif)-1 && $difi < 5000; $difi++)
					fwrite($fp, $dif[$difi]);
				if($difi >= 5000) fwrite($fp, "===OUTPUT OF COMPARING SCRIPT TOO LONG - TRUNCATED===\n");
				else fwrite($fp, "===OUTPUT OF COMPARING SCRIPT ENDS HERE===\n");
				$answer = trim($dif[count($dif)-1]);
				fclose($fp);
				foreach (glob($dir . $ds . '*') as $fne) {
					@chown($fne,"nobody");
					@chmod($fne,0755);
				}
				// retval 5 (presentation) and retval 6 (wronganswer) are already compatible with the compare script
				if($localretval < 4 || $localretval > 6) {
					// contact staff
					$retval = 7;
					$answer='Contact staff';
				}
				else if($retval==0) {
					if($localretval==4) {
						// YES!
						$answer='YES';
						$retval = 1;
					} else $retval=$localretval;
				}
				else if($retval==1) {
					if($localretval!=4) {
						$retval=$localretval;
						$answer='Presentation error';
					}
				}
			}
		}
	}
}
if($retval > 9) {
	$ans = file("allout");
	$anstmp = trim(escape_string($ans[count($ans)-1]));
	unset($ans);
	LogLevel("Autojudging: Script returned unusual code: $retval ($anstmp)".
			 "(run=$number, site=$site, contest=$contest)",1);
	echo "Autojudging script returned unusual code $retval ($anstmp)".
		"(contest=$contest, site=$site, run=$number)\n";
//  cleardir($dir);
//  unlink($name);
//  DBGiveUpRunAutojudging($contest, $site, $number, $ip, "(unusual code: $retval) " . $anstmp);
//  continue;
	$answer = "(check output files - unusual code: $retval) " . $anstmp;
	
	// contact staff
	$retval = 7;
}

echo "Sending results to server\n";
DBUpdateRunAutojudging($contest, $site, $number, $ip, $answer, $dir.$ds.'allout', $dir.$ds.'allerr', $retval);
LogLevel("Autojudging: answered '$answer' (run=$number, site=$site, contest=$contest)",3);
echo "Autojudging answered '$answer' (contest=$contest, site=$site, run=$number)\n";
if(!isset($dodebug)) {
	cleardir($dir);
	unlink($name);
}
}
?>