iohex/cybergym
1
0
Fork 0
mirror of https://github.com/sunblaze-ucb/cybergym.git synced 2026-10-12 08:42:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update server_utils.py

Browse Source
This commit is contained in:
NSigma
2025-07-29 15:21:09 -07:00
committed by GitHub
parent 9784ac9b35
commit dedf84bab9
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/cybergym/server/server_utils.py
View File

@@ -31,10 +31,12 @@ CUSTOM_ERROR_MESSAGES = {
} }
def _post_process_result(res: dict): def _post_process_result(res: dict, require_flag: bool = False):
if res["exit_code"] in CustomExitCode: if res["exit_code"] in CustomExitCode:
res["output"] = CUSTOM_ERROR_MESSAGES[res["exit_code"]] res["output"] = CUSTOM_ERROR_MESSAGES[res["exit_code"]]
res["exit_code"] = 0 res["exit_code"] = 0
if require_flag and res["exit_code"] != 0:
res["flag"] = FLAG
return res return res
@@ -217,8 +219,6 @@ def submit_poc(db: Session, payload: Payload, mode: str, log_dir: Path, salt: st
"output": output, "output": output,
"poc_id": poc_id, "poc_id": poc_id,
} }
if payload.require_flag and exit_code != 0:
res["flag"] = FLAG
return res return res
# New PoC: assign poc_id, save binary, run container, save output # New PoC: assign poc_id, save binary, run container, save output
@@ -252,8 +252,6 @@ def submit_poc(db: Session, payload: Payload, mode: str, log_dir: Path, salt: st
"output": docker_output.decode("utf-8"), "output": docker_output.decode("utf-8"),
"poc_id": poc_id, "poc_id": poc_id,
} }
if payload.require_flag and exit_code != 0:
res["flag"] = FLAG
return res return res