Add compile timeout

oj-lab · Oct 1, 2024 · 18c1fd0 · 18c1fd0
1 parent af19251
commit 18c1fd0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/judge-core/src/compiler.rs b/judge-core/src/compiler.rs
@@ -11,7 +11,7 @@ const TEMPLATE_ARG_TARGET_PATH: &str = "{target_path}";
 
 const RUST_COMPILE_COMMAND_TEMPLATE: &str = "rustc {src_path} -o {target_path}";
 const CPP_COMPILE_COMMAND_TEMPLATE: &str =
-    "g++ -fmax-include-depth=10 {src_path} -o {target_path} -O2 -static";
+    "g++ {src_path} -o {target_path} -O2 -static";
 const PYTHON_COMPILE_COMMAND_TEMPLATE: &str = "cp {src_path} {target_path}";
 
 #[derive(Clone)]
@@ -163,7 +163,9 @@ impl Compiler {
             std::fs::remove_file(target_path)?;
         }
 
-        let output = Command::new("sh")
+        let output = Command::new("timeout")
+            .arg("30s") // TODO: make timeout configurable
+            .arg("sh")
             .arg("-c")
             .arg(
                 self.command_builder