1 files changed, 113 insertions, 0 deletions
diff --git a/judge/jmod_test_check.cpp b/judge/jmod_test_check.cpp
new file mode 100644
index 0000000..ae5fe89
--- /dev/null
+++ b/judge/jmod_test_check.cpp
@@ -0,0 +1,113 @@
+#ifdef HAVE_CONFIG_H
+# include<judge_conf.h>
+#endif
+
+#include<stdio.h>
+#include<stdlib.h>
+#include<string.h>
+#include<limits.h>
+#include<unistd.h>
+#include<fcntl.h>
+#include<errno.h>
+
+#include<judge_def.h>
+#include"judgm_lib.h"
+#include"jmod_test.h"
+#include"jmod_test_check.h"
+
+static int ansfd;
+static judgm_hyperio *hyperio;
+static int tty_idx;
+
+int p[2];
+
+DLL_PUBLIC int init(int judgk_modfd,char *datapath,char *runpath){
+    char tpath[PATH_MAX + 1];
+    char dstpath[PATH_MAX + 1];
+
+    pipe(p);
+
+    snprintf(tpath,sizeof(tpath),"%s/in",datapath);
+    snprintf(dstpath,sizeof(dstpath),"%s/in",runpath);
+    if(link(tpath,dstpath)){
+	return -1;
+    }
+    snprintf(tpath,sizeof(tpath),"%s/ans",datapath);
+    if((ansfd = open(tpath,O_RDONLY)) == -1){
+	return -1;
+    }
+
+    //hyperio = new judgm_hyperio(judgk_modfd);
+    //tty_idx = hyperio->tty_idx;
+
+    return 0;
+}
+DLL_PUBLIC int run(int &status,char *err_msg){
+    int ret;
+
+    char *inbuf;
+    char *ansbuf;
+
+    status = JUDGE_AC;
+    inbuf = new char[65536];
+    ansbuf = new char[65536];
+
+    /*while((ret = read(ansfd,ansbuf,65536)) > 0){
+	if(hyperio->compare(ansbuf,ret)){
+	    status = JUDGE_WA;
+	    break;
+	}
+    }
+    if(status == JUDGE_AC && hyperio->wait() > 0){
+	status = JUDGE_WA;
+    }
+	
+    delete inbuf;
+    delete ansbuf;
+    close(ansfd);
+    delete hyperio;*/
+
+    close(p[1]);
+    int c_read=-1,pre_status=status;
+    while((ret = read(p[0],inbuf,65536)) > 0){
+	if((c_read=read(ansfd,ansbuf,ret)) != ret){
+	    status = JUDGE_WA;
+	    break;
+	}
+	if(memcmp(ansbuf,inbuf,ret)){
+	    status = JUDGE_WA;
+	    break;
+	}
+    }
+//    if(c_read - ret == 1 && ansbuf[c_read-1]=='\n'){
+//	status = pre_status;
+//    }
+	    
+    if(status == JUDGE_AC && ((c_read = read(ansfd,ansbuf,1)) > 1 || c_read == 1 && ansbuf[0]!='\n')){
+	status = JUDGE_WA;
+    }
+
+    return 0;
+}
+DLL_PUBLIC int proc(){
+    int infd;
+    int outfd;
+
+    if((infd = open("in",O_RDONLY)) == -1){
+	return -1;
+    }
+    /*if((outfd = judgm_hyperio::get_ttyfd(tty_idx)) == -1){
+	return -1;
+    }*/
+
+    close(p[0]);
+    outfd = p[1];
+    dup2(infd,0);
+    dup2(outfd,1);
+    dup2(outfd,2);
+
+    return 0;
+}
+DLL_PUBLIC int stop(){
+    return 0;
+}