程序源码
https://gitee.com/not-a-stupid-child/online-judge
compile_server
整体思路
这个服务要对oj_server 发送过来的代码进行编译和运行,最后把结果返回给oj_server。
所以我们可以把这个模块在分为
- 编译模块(compile.hpp):只负责对代码进行编译,以及差错处理。
- 运行模块(runner.hpp):只负责代码的运行,以及差错处理。
- 编译运行模块(compile_run.hpp):整合编译和运行模块,以及适配用户的需求,形成唯一的文件名。
- 编译运行服务(compile_server.hpp):引入网络,与编译和运行模块适配。
编译(compile.hpp)
这个模块主要是为了完成对代码的编译,编译可能成功也可能失败,如果编译成功就很完美,但是如果失败了,我们需要形成临时文件保存失败的信息,当然我们进行编译一定不能让我们主进程去编译,不然失败了整个服务就挂掉了,所以要创建子进程去编译,g++编译失败了会想标准错误打印消息,所以在编译之前我们可以进行重定向功能。我们这个函数传递进来的文件名是只存在文件名的,我们形成的临时文件也不能都在自己的当前目录,所以我们可以通过文件名来把路径拼接一下,然后形成各种的后缀就好了。这个拼接路径的工作可能多个模块都需要,所以需要放在comm中。
运行模块
该模块主要实现对可执行程序的运行,当然也是需要创建子进程的,但是对于一个进程来说,一定会存在标准输入、标准输出、标准错误。所以我们就需要根据文件名来形成这三个文件,然后进行重定向,如果这个程序需要输入就从对应的文件中读取,当然最后执行完之后需要的结果也会在标准错误对应的文件中,我们只需要将这三个文件创建好,然后只关心运行成功与否就可以了,因此对于父进程来说,一定是要关心子进程的运行状态的,如果程序运行成功了,那就返回就好了,如果出现异常运行失败了,一定会收到某种信号,就把收到的对应信号进行返回,所以对于这个函数来说,返回值>0 运行异常, 返回值 = 0 运行成功,结果不确定, 返回值 <0 程序内部出错了(打开文件等)。
当然对于用户提交的程序,我们需要对它有时间和空间的限制,此时就需要使用setrlimit 函数,进行时间和空间的限制,防止恶意程序消耗我们CPU资源。
编译运行模块
编译运行模块主要是对运行和编译模块的一个汇总,以及功能的补充,我们的编译和运行模块主要提供了一个Start的方法,这个方法需要外部提供一个Json串,然后我们把程序运行的结果在包装成一个Json串返回回去。因为我们这个服务是在后端让oj_server调用的,所以一定要通过序列化的方式将结果发送回去。
对于输入的Json 串
{"code":"#inclde...", "input":"1 2", "cpu_limit":"1", "mem_limit":"2"}
包含这些字段。code为需要编译运行的代码,input是用户的输入,cpu_limit就是时间复杂度的控制,mem_limit就是空间复杂度的控制。
我们输出的Json串
_outjson {"status":1, "reason":"...", "stdout":"..", "stderr":".."}
这些字段stdou和stderr分别表示标准出入和标准输出的内容,他俩不是必须得,只有程序运行成功才会有,而status表示标识程序是否成功运行了,reason是程序出错的原因。
通过用户传入的Json串可以分析出来代码等很多内容,然后我们需要形成很多的临时文件,比如可执行等等,这是就需要形成一个唯一的文件名,这个可以使用时间戳+递增的值的方式可以大概率的形成唯一的文件名,有了唯一文件名,就需要把代码写入源文件中,然后进行调用编译木块和运行模块,进行代码的编译和运行,然后通过status来分析程序运行的结果。根据status是否是零来分析是否需要填写stdout和stderr。不管程序怎么运行的,编译失败和运行成功的结果一定保存在文件中的,所以这期间也会存在很多读文件的操作,所以可以把读写文件封装在comm部分。
编译运行服务
这个服务就很简单了,但是需要引入一个httplib的库,然后通过网络将请求的Json串拿到,交给Start函数,拿到结果返回给oj_server就可以了。
核心代码:
Server svr;
svr.Post("/compile_run", [](const Request &req, Response &resp)
{
std::string in = req.body;
std::string out;
Compile_run::Start(in,&out);
resp.set_content(out,"application/json;charset=utf-8");
});
svr.listen("0.0.0.0", std::stoi(argv[1]));
本文章主要描述主体思路,细节需要自己补充。