mirror of
https://github.com/sunblaze-ucb/cybergym.git
synced 2026-10-12 08:42:24 +00:00
Update __main__.py
This commit is contained in:
@@ -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
|
raise HTTPException(status_code=400, detail="Invalid metadata format") from None
|
||||||
payload.data = file.file.read()
|
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 = 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
|
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
|
raise HTTPException(status_code=400, detail="Invalid metadata format") from None
|
||||||
payload.data = file.file.read()
|
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 = 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
|
return res
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user