最近无聊看了看gty的Gena感觉非常 的高大上。
其实我非常早的就想自己弄一个自动评测软件了，于是学习了以下开始动手。
然后发现有非常多的问题，大部分问题都是用黑科技解决，比如fork之后堆栈不再共用没办法互相传递信息，那么我们就开一个文件来传递信息。
不过还是不少地方借鉴了以下，比如得到内存信息，和时间记录的方式(clock函数会出现一些奇怪的问题)
//coder : davidwang
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <sys/types.h>
#include <unistd.h>
#include <sys/timeb.h>
#include <ctime>
#include <sys/stat.h>
#include <sys/wait.h>
#include <vector>
#include "My_checker.hpp"
using namespace std;
/*
this program is to check a OI program with specific testcases
./checker [-t Time_limit(ms)]{default : 1000} [-T Time_limit(s)] [-m memory_limit(b)]{default:256Mb} [-M memory_limit(mb)] [-f path of programm and testcases]{default: current path} [-F initial file] [-I input files]{default: a.in} [-O output files] {default: a.out}
*/
void init_file(char *s1);
void show_help();
void work();
vector<string>list;
int init(int argv,char *argc[]){
for (int i=1;i<argv;i+=2){
if (!strcmp(argc[i],"--help")){
show_help();
return -1;
}
if (!strcmp(argc[i],"-t")) sscanf(argc[i+1],"%d",&Time_limit);
else if (!strcmp(argc[i],"-T")) sscanf(argc[i+1],"%d",&Time_limit),Time_limit*=1000;
else if (!strcmp(argc[i],"-m")) sscanf(argc[i+1],"%d",&Memory_limit);
else if (!strcmp(argc[i],"-M")) sscanf(argc[i+1],"%d",&Memory_limit),Memory_limit*=1024*1024;
else if (!strcmp(argc[i],"-f")){
sscanf(argc[i+1],"%s",path);
used_path=1;
}else if (!strcmp(argc[i],"-F")){
init_file(argc[i+1]);
return 1;
}else if (!strcmp(argc[i],"-I") || !strcmp(argc[i],"-i")){
Infile=1;
sscanf(argc[i+1],"%s",input_file);
}else if (!strcmp(argc[i],"-O") || !strcmp(argc[i],"-o")){
Outfile=1;
sscanf(argc[i+1],"%s",output_file);
}
}
return 0;
}
int type=-1;
int make_file(){
sprintf(cmd,"ls %s >cur_files_list.ckr",path);
system(cmd);
f1=fopen("cur_files_list.ckr","r");
while (fscanf(f1,"%s",cur)!=EOF){
list.push_back(string(cur));
}
fclose(f1);
system("rm -r cur_files_list.ckr");
sort(list.begin(),list.end());
//0 for cpp 1 for c 2 for pas
char tmp[110];
for (int i=0;i<list.size();i++){
int pos=-1;
if (list[i]==string("cur_files_list.ckr")){
list.erase(list.begin()+i);
i--;
continue;
}
if (list[i]==string("My_checker.cpp")){
list.erase(list.begin()+i);
i--;
continue;
}
if (list[i]==string("My_checker")){
list.erase(list.begin()+i);
i--;
continue;
}
for (int j=0;j<list[i].length();j++){
if (list[i][j]=='.'){
pos=j;
memset(tmp,0,sizeof(tmp));
for (int k=pos+1;k<list[i].length();k++){
tmp[k-pos-1]=list[i][k];
}
if (!strcmp(tmp,"cpp")){
if (type!=-1){
printf("Error : multiple program included\nautochecker failed,please use file style");
return -1;
}
printf("%d\n",pos);
for (int k=0;k<pos;k++) program_name[k]=list[i][k];
type=0;
list.erase(list.begin()+i);
i--;
}else if (!strcmp(tmp,"pas")){
if (type!=-1){
printf("Error : multiple program included\nautochecker failed,please use file style");
return -1;
}
for (int k=0;k<pos;k++) program_name[k]=list[i][k];
type=1;
list.erase(list.begin()+i);
i--;
}else if (!strcmp(tmp,"c")){
if (type!=-1){
printf("Error : multiple program included\nautochecker failed,please use file style");
return -1;
}
for (int k=0;k<pos;k++) program_name[k]=list[i][k];
type=2;
list.erase(list.begin()+i);
i--;
}else if (strcmp(tmp,"in") && strcmp(tmp,"out")){
list.erase(list.begin()+i);
i--;
}
break;
}
}
if (pos==-1){
list.erase(list.begin()+i);
i--;
}
}
#ifdef DEBUG
printf("list begin! with %d\n",list.size());
for (int i=0;i<list.size();i++){
cout<<list[i]<<endl;
}
puts("list end");
#endif
if (type==-1){
printf("Error: No program found!\n checker failed");
return -1;
}
}
int Result,Running;
inline int testcase(string input,string output,string program_name){
#ifdef DEBUG
printf("check(%s,%s)\n",input.c_str(),output.c_str());
#endif
if (used_path) sprintf(cmd,"cat %s/%s > %s",path,input.c_str(),input_file);
else sprintf(cmd,"cat %s > %s",input.c_str(),input_file);
#ifdef DEBUGcmd
puts(cmd);
#endif
system(cmd);
timeb times,timet;
ftime(×);
int stat;
int pid=fork();
if (pid<0){
printf("checker failed!\n");
return 5;
}
if (pid==0){
char In[22],Out[22];
// if (!Infile) sprintf(In," <%s.in",program_name.c_str());
// if (!Outfile) sprintf(Out," >%s.out",program_name.c_str());
if (used_path) sprintf(cmd,"cd %s && ./%s",path,program_name.c_str());
else sprintf(cmd,"./%s",program_name.c_str());
#ifdef DEBUGcmd
puts(cmd);
#endif
int id=system(cmd);
f2=fopen("finished_running_time.ckr","w");
fprintf(f2,"finished!");
fclose(f2);
#ifdef DEBUG
printf("son progress return ed normally");
#endif
while (1);
return id;
}else{
mem=0; Rtime=0;
while (1){
ftime(&timet);
mem=max(mem,getmemory(pid));
Rtime=(timet.time - times.time) * 1000 + (timet.millitm - times.millitm);
if (Rtime>Time_limit){
Result=2;
kill_judge(pid);
Running=0;
#ifdef DEBUG
printf("parent progress return ed TLE");
#endif
return 0;
}else if (mem>Memory_limit){
// Result=3;
// kill_judge(pid);
// Running=0;
#ifdef DEBUG
printf("parent progress return ed MLE");
#endif
// return 0;
}
f1=fopen("finished_running_time.ckr","r");
if (f1==NULL) continue;
kill_judge(pid);
Running=0;
system("rm -r finished_running_time.ckr");
break;
}
#ifdef DEBUG
printf("parent progress return ed nomally");
#endif
wait(&stat);
int i=WEXITSTATUS(stat);
// printf("son program returned %d\n",i);
if (i){
Running=0;
Result=4;
return 0;
}
Running=0;
Result=check(output);
return 0;
}
}
inline void sleep_with_limit(int T){
timeb s,t;
ftime(&s);
while (1){
if (Running==0) return;
ftime(&t);
if (t.time-s.time>T+1 || t.time-s.time) return;
}
}
int work_auto()
{
int tmp=make_file();
if (tmp==-1){
return -1;
}
int Compile_id;
if (type==0){
if (used_path)
sprintf(cmd,"g++ -o2 %s/%s.cpp -o %s/%s",path,program_name.c_str(),path,program_name.c_str());
else
sprintf(cmd,"g++ -o2 %s.cpp -o %s",program_name.c_str(),program_name.c_str());
Compile_id=system(cmd);
}else if (type==1){
if (used_path)
sprintf(cmd,"fpc -o2 %s/%s.pas",path,program_name.c_str());
else sprintf(cmd,"fpc -o2 %s.pas",program_name.c_str());
Compile_id=system(cmd);
}else if (type==2){
if (used_path)
sprintf(cmd,"gcc -o2 %s/%s.c -o %s/%s",path,program_name.c_str(),path,program_name.c_str());
else
sprintf(cmd,"gcc -o2 %s.c -o %s",program_name.c_str(),program_name.c_str());
Compile_id=system(cmd);
}
if (Compile_id){
puts(result[5]);
return 5;
}
int Id=0,now;
if (!Infile){
if (used_path) sprintf(input_file,"%s/%s.in",path,program_name.c_str());
else sprintf(input_file,"%s.in",program_name.c_str());
}
if (!Outfile){
if (used_path) sprintf(output_file,"%s/%s.out",path,program_name.c_str());
sprintf(output_file,"%s.out",program_name.c_str());
}
int ip;
int pre;
for (ip=0;ip<list.size();ip=pre+2){
pre=ip;
Running=1;
testcase(list[ip],list[ip+1],program_name);
sleep_with_limit(Time_limit/1000);
if (!Id) Id=Result;
printf("%s time:%d memory:%d\n",result[Result],Rtime,mem);
sprintf(cmd,"rm -r %s",input_file);
system(cmd);
if (used_path) sprintf(cmd,"rm -r %s/%s",path,output_file);
else sprintf(cmd,"rm -r %s",output_file);
system(cmd);
}
return Id;
}
int main(int argv,char *argc[])
{
int tmp=init(argv,argc);
if (tmp==-1) return 0;
else if (tmp==1){
printf("Error!\n"),show_help();
return 0;
}else if (tmp==2){
work();
}else{
int ret=work_auto();
}
return 0;
}
//coder : davidwang
#include <string>
using namespace std;
const int DefaultTime_limit=1000;
const int DefaultMemory_limit=256*2048*2048;
int Time_limit=DefaultTime_limit,Memory_limit=DefaultMemory_limit;
char cmd[1000],cur[110];
char input_file[1000],output_file[1000];
string program_name;
int Infile,Outfile;
FILE *a1,*f1,*f2;
int mem,Rtime;
char path[2500];
int used_path;
char result[100][100]={"Accepted","Wrong Answer","Time Limit Exceeded","Memory Limit Exceeded","Runtime Error","Compile Error"};
// 0: Accepted 1: Wrong Answer 2: TLE 3: MLE 4: RE 5: CE
inline int getmemory(int pid)
{
sprintf(cmd, "ps -eo pid,rss | grep %d", pid);
FILE *f = popen(cmd, "r");
int a;
fscanf(f, "%d%d", &a, &a);
pclose(f);
return a;
}
void kill_judge(int pid)
{
sprintf(cmd, "kill -s 9 %d", pid);
system(cmd);
}
void init_file(char *s1){
}
void show_help(){
printf("this program is to check a OI program with specific testcases\n./checker [-t Time_limit(ms)]{default : 1000} [-T Time_limit(s)] [-m memory_limit(b)]{default:256Mb} [-M memory_limit(mb)] [-f path of programm and testcases]{default: current path} [-F initial file] [-I input files]{default: (s).in} [-O output files] {default: (s).out}\n");
return;
}
void work()
{
}
char s1[1000010],s2[1000010];
char o1[1000],o2[1000];
inline int check(string output,string checker=string("cmp"))
{
FILE *fout,*fstd;
// printf("checker runned!");
if (used_path) sprintf(o1,"%s/%s",path,output.c_str());
else sprintf(o1,"%s",output.c_str());
if (used_path) sprintf(o2,"%s/%s",path,output_file);
else sprintf(o2,"%s",output_file);
fstd=fopen(o1,"r");
fout=fopen(o2,"r");
// puts(output_file);
// puts(output.c_str());
int t=0;
while (fscanf(fstd,"%s",s1)!=EOF){
if (fscanf(fout,"%s",s2)==EOF){
printf("checker log: your output is shorter than standard output\n");
return 1;
}
t++;
// puts(s1);
// puts(s2);
if (strcmp(s1,s2)){
printf("checker log: Wrong with the %d segment!\n",t);
printf("%s %s\n",s1,s2);
return 1;
}
}
if (fscanf(fout,"%s",s2)!=EOF){
printf("checker log: Your output is longer than standard output!\n");
return 1;
}
return 0;
}
放在一个目录下编译就可以用了,暂时只支持linux……
