feat(iris): limit judge result output (#2256)

* feat(iris): limit judge result output * feat(iris): set limit 1MB
skkuding · Dec 7, 2024 · e92dd4d · e92dd4d
1 parent 9c6e9ef
commit e92dd4d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/apps/iris/src/common/constants/constants.go b/apps/iris/src/common/constants/constants.go
@@ -53,3 +53,5 @@ const (
 	EXCHANGE   = "judger-exchange"
 	RESULT_KEY = "result"
 )
+
+const MAX_OUTPUT = 1048576 // 1MB
diff --git a/apps/iris/src/handler/judge-handler.go b/apps/iris/src/handler/judge-handler.go
@@ -347,6 +347,10 @@ func (j *JudgeHandler) judgeTestcase(idx int, dir string, validReq *Request,
 	res.SetJudgeExecResult(runResult.ExecResult)
 	res.Output = string(runResult.Output)
 
+	if len(res.Output) > constants.MAX_OUTPUT {
+		res.Output = res.Output[:constants.MAX_OUTPUT]
+	}
+
 	if runResult.ExecResult.ResultCode != sandbox.RUN_SUCCESS {
 		res.SetJudgeResultCode(SandboxResultCodeToJudgeResultCode(runResult.ExecResult.ResultCode))
 		goto Send