Add compile timeout

oj-lab · Oct 1, 2024 · 473034f · 473034f
1 parent af19251
commit 473034f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/judge-core/src/compiler.rs b/judge-core/src/compiler.rs
@@ -10,8 +10,7 @@ const TEMPLATE_ARG_SRC_PATH: &str = "{src_path}";
 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";
+const CPP_COMPILE_COMMAND_TEMPLATE: &str = "g++ {src_path} -o {target_path} -O2 -static";
 const PYTHON_COMPILE_COMMAND_TEMPLATE: &str = "cp {src_path} {target_path}";
 
 #[derive(Clone)]
@@ -163,7 +162,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