增加判题服务器 token 验证

This commit is contained in:
virusdefender 2015-12-07 17:35:03 +08:00
parent d61dae9be0
commit efee635173

View File

@ -11,14 +11,15 @@ from settings import judger_workspace
class JudgeInstanceRunner(object):
def __init__(self):
pass
def run(self, submission_id, language_code, code, time_limit, memory_limit, test_case_id):
def run(self, token, submission_id, language_code, code, time_limit, memory_limit, test_case_id):
language = languages[language_code]
host_name = socket.gethostname()
judge_base_path = os.path.join(judger_workspace, "run", submission_id)
if not token or token != os.environ.get("rpc_token"):
return {"code": 2, "data": {"error": "Invalid token", "server": host_name}}
try:
os.mkdir(judge_base_path)
os.chmod(judge_base_path, 0777)