OnlineJudge/judge/compiler.py
virusdefender cf2fc9df1a 指定使用nobody用户进行编译,防止#include</etc/shadow>等情况的发生
部分情况下,include敏感文件可能造成信息泄露。这里设置一下用户权限。
2016-03-05 11:42:17 +08:00

42 lines
1.5 KiB
Python

# coding=utf-8
import time
import os
import judger
from judge_exceptions import CompileError, JudgeClientError
from logger import logger
from settings import judger_workspace
def compile_(language_item, src_path, exe_path, judge_base_path):
compile_command = language_item["compile_command"].format(src_path=src_path, exe_path=exe_path).split(" ")
compiler = compile_command[0]
compile_args = compile_command[1:]
compiler_output_file = os.path.join(judge_base_path, "compiler.out")
compile_result = judger.run(path=compiler,
in_file="/dev/null",
out_file=compiler_output_file,
max_cpu_time=2000,
max_memory=2000000000,
args=compile_args,
env=["PATH=" + os.environ["PATH"]],
use_sandbox=False,
use_nobody=True)
compile_output_handler = open(compiler_output_file)
compile_output = compile_output_handler.read().strip()
compile_output_handler.close()
if compile_result["flag"] != 0:
logger.error("Compiler error")
logger.error(compile_output)
logger.error(str(compile_result))
if compile_output:
raise CompileError(compile_output)
else:
raise CompileError("Compile error, info: " + str(compile_result))
else:
if "error" in compile_output:
raise CompileError(compile_output)
return exe_path