Add time limit language wise (#22)

Eluminatis-of-LU · May 9, 2024 · 904b164 · 904b164
1 parent dd67b1f
commit 904b164
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/sojj/Dtos/Language.cs b/sojj/Dtos/Language.cs
@@ -18,4 +18,7 @@ public class Language
 
     [JsonPropertyName("execute")]
     public string[]? Execute { get; set; }
+
+    [JsonPropertyName("cpuLimit")]
+    public long? CpuLimit { get; set; }
 }
diff --git a/sojj/Services/SandBoxService.cs b/sojj/Services/SandBoxService.cs
@@ -82,6 +82,7 @@ public async Task<CompileResult> CompileAsync(string sourceCode, string runId, s
         }
 
         logger.LogInformation("Language {language} found", language);
+        long timeLimitInNs = (languageInfo.CpuLimit ?? this.cpuLimitForRuns) * Constants.NanoSecondInSecond;
         var request = new SandboxRunRequest
         {
             Commands =
@@ -107,8 +108,8 @@ public async Task<CompileResult> CompileAsync(string sourceCode, string runId, s
                                 Max = this.outputLimitForRuns * Constants.ByteInKiloByte,
                             },
                         ],
-                        CpuLimit = this.cpuLimitForRuns * Constants.NanoSecondInSecond,
-                        ClockLimit = (this.cpuLimitForRuns * Constants.NanoSecondInSecond) * 3,
+                        CpuLimit = timeLimitInNs,
+                        ClockLimit = timeLimitInNs * 3,
                         MemoryLimit = this.memoryLimitForRuns * Constants.ByteInMegaByte,
                         ProcessLimit = this.processLimitForRuns,
                         CopyIn = new Dictionary<string, SandboxFile>

diff --git a/sojj/languages.json b/sojj/languages.json
@@ -88,6 +88,7 @@
     "compile": ["kotlinc", "foo.kt", "-include-runtime", "-d", "foo.jar"],
     "codeFile": "foo.kt",
     "outputFile": "foo.jar",
-    "execute": ["java", "-jar", "foo.jar"]
+    "execute": ["java", "-jar", "foo.jar"],
+    "cpuLimit": 25
   }
 }