From e0a0971b5d319936e3c2caccc8fd1b2c52f91c81 Mon Sep 17 00:00:00 2001 From: NSigma Date: Tue, 29 Jul 2025 15:21:58 -0700 Subject: [PATCH] Update __main__.py --- src/cybergym/server/__main__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cybergym/server/__main__.py b/src/cybergym/server/__main__.py index 8392e72..d171dd6 100644 --- a/src/cybergym/server/__main__.py +++ b/src/cybergym/server/__main__.py @@ -67,7 +67,7 @@ def submit_vul(db: SessionDep, metadata: Annotated[str, Form()], file: Annotated raise HTTPException(status_code=400, detail="Invalid metadata format") from None payload.data = file.file.read() res = submit_poc(db, payload, mode="vul", log_dir=LOG_DIR, salt=SALT, oss_fuzz_path=OSS_FUZZ_PATH) - res = _post_process_result(res) + res = _post_process_result(res, payload.require_flag) return res @@ -79,7 +79,7 @@ def submit_fix(db: SessionDep, metadata: Annotated[str, Form()], file: Annotated raise HTTPException(status_code=400, detail="Invalid metadata format") from None payload.data = file.file.read() res = submit_poc(db, payload, mode="fix", log_dir=LOG_DIR, salt=SALT, oss_fuzz_path=OSS_FUZZ_PATH) - res = _post_process_result(res) + res = _post_process_result(res, payload.require_flag) return res