Respect script limits when building executables.

meisterT · Feb 22, 2024 · 9d237fd · 9d237fd
1 parent 89fe096
commit 9d237fd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/judge/build_executable.sh b/judge/build_executable.sh
@@ -76,7 +76,9 @@ logmsg $LOG_INFO "starting build"
 
 exitcode=0
 $GAINROOT "$RUNGUARD" ${DEBUG:+-v} -u "$RUNUSER" -g "$RUNGROUP" \
-	-r "$CHROOTDIR" -d '/build' --no-core -- \
+	-r "$CHROOTDIR" -d '/build' \
+	-m $SCRIPTMEMLIMIT -t $SCRIPTTIMELIMIT --no-core -f $SCRIPTFILELIMIT -s $SCRIPTFILELIMIT \
+	-- \
 	'./build' > 'build.log' 2>&1 || \
 	exitcode=$?
 

diff --git a/judge/judgedaemon.main.php b/judge/judgedaemon.main.php
@@ -439,6 +439,11 @@ function fetch_executable_internal(
 
         if ($do_compile) {
             logmsg(LOG_DEBUG, "Building executable in $execdir, under 'build/'");
+
+            putenv('SCRIPTTIMELIMIT=' . djconfig_get_value('script_timelimit'));
+            putenv('SCRIPTMEMLIMIT='  . djconfig_get_value('script_memory_limit'));
+            putenv('SCRIPTFILELIMIT=' . djconfig_get_value('script_filesize_limit'));
+
             system(LIBJUDGEDIR . '/build_executable.sh ' . dj_escapeshellarg($execdir), $retval);
             if ($retval !== 0) {
                 return [null, "Failed to build executable in $execdir.", "$execdir/build.log"];