update app judge

Author: limstash

module/judge/judge_compileUnix.go

@@ -6,7 +6,7 @@ import (
 )
 
 func CompileUnixC11(SubmissionID int, lang string) bool {
-	cmd := exec.Command("gcc", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.c", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++11", "-lm")
+	cmd := exec.Command("gcc", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.c", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++11", "-lm", "-DONLINE_JUDGE")
 	err := cmd.Run()
 
 	if err != nil {
@@ -17,7 +17,7 @@ func CompileUnixC11(SubmissionID int, lang string) bool {
 }
 
 func CompileUnixCPP11(SubmissionID int, lang string) bool {
-	cmd := exec.Command("g++", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.cpp", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++11", "-lm")
+	cmd := exec.Command("g++", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.cpp", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++11", "-lm", "-DONLINE_JUDGE")
 	err := cmd.Run()
 
 	if err != nil {
@@ -28,7 +28,7 @@ func CompileUnixCPP11(SubmissionID int, lang string) bool {
 }
 
 func CompileUnixCPP14(SubmissionID int, lang string) bool {
-	cmd := exec.Command("g++", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.cpp", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++14", "-lm")
+	cmd := exec.Command("g++", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.cpp", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++14", "-lm", "-DONLINE_JUDGE")
 	err := cmd.Run()
 
 	if err != nil {
@@ -39,7 +39,7 @@ func CompileUnixCPP14(SubmissionID int, lang string) bool {
 }
 
 func CompileUnixCPP17(SubmissionID int, lang string) bool {
-	cmd := exec.Command("g++", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.cpp", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++17", "-lm")
+	cmd := exec.Command("g++", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main.cpp", "-o", getPath()+"/src/"+strconv.Itoa(SubmissionID)+"/main", "--std=c++17", "-lm", "-DONLINE_JUDGE")
 	err := cmd.Run()
 
 	if err != nil {

module/judge/judge_runUnix.go

@@ -17,18 +17,21 @@ func RunUnixBin(SubmissionID int, SubmissionPath string) bool {
 		return false
 	}
 
-	defer stdin.Close()
-
 	stdout, err := os.OpenFile(SubmissionPath+"/data.out", os.O_CREATE|os.O_WRONLY, 0600)
 
 	if err != nil {
 		return false
 	}
 
-	defer stdout.Close()
-
 	subProcess.Stdout = stdout
-	subProcess.Stderr = os.Stderr
+
+	stderr, err := os.OpenFile(SubmissionPath+"/data.err", os.O_CREATE|os.O_WRONLY, 0600)
+
+	if err != nil {
+		return false
+	}
+
+	subProcess.Stderr = stderr
 
 	if err = subProcess.Start(); err != nil {
 		return false
@@ -40,20 +43,22 @@ func RunUnixBin(SubmissionID int, SubmissionPath string) bool {
 		return false
 	}
 
-	defer fi.Close()
-
 	inputBuff := bufio.NewReader(fi)
 
 	for {
 		a, _, c := inputBuff.ReadLine()
 		if c == io.EOF {
+			stdin.Close()
 			break
 		}
 		io.WriteString(stdin, string(a)+"\n")
 	}
 
 	subProcess.Wait()
 
+	stdout.Close()
+	fi.Close()
+
 	return true
 }
 
@@ -66,18 +71,22 @@ func RunPython2(SubmissionID int, SubmissionPath string) bool {
 		return false
 	}
 
-	defer stdin.Close()
-
 	stdout, err := os.OpenFile(SubmissionPath+"/data.out", os.O_CREATE|os.O_WRONLY, 0600)
 
 	if err != nil {
 		return false
 	}
 
-	defer stdout.Close()
-
 	subProcess.Stdout = stdout
 
+	stderr, err := os.OpenFile(SubmissionPath+"/data.err", os.O_CREATE|os.O_WRONLY, 0600)
+
+	if err != nil {
+		return false
+	}
+
+	subProcess.Stderr = stderr
+
 	if err = subProcess.Start(); err != nil {
 		return false
 	}
@@ -88,20 +97,22 @@ func RunPython2(SubmissionID int, SubmissionPath string) bool {
 		return false
 	}
 
-	defer fi.Close()
-
 	inputBuff := bufio.NewReader(fi)
 
 	for {
 		a, _, c := inputBuff.ReadLine()
 		if c == io.EOF {
+			stdin.Close()
 			break
 		}
 		io.WriteString(stdin, string(a)+"\n")
 	}
 
 	subProcess.Wait()
 
+	stdout.Close()
+	fi.Close()
+
 	return true
 }
 
@@ -114,18 +125,22 @@ func RunPython3(SubmissionID int, SubmissionPath string) bool {
 		return false
 	}
 
-	defer stdin.Close()
-
 	stdout, err := os.OpenFile(SubmissionPath+"/data.out", os.O_CREATE|os.O_WRONLY, 0600)
 
 	if err != nil {
 		return false
 	}
 
-	defer stdout.Close()
-
 	subProcess.Stdout = stdout
 
+	stderr, err := os.OpenFile(SubmissionPath+"/data.err", os.O_CREATE|os.O_WRONLY, 0600)
+
+	if err != nil {
+		return false
+	}
+
+	subProcess.Stderr = stderr
+
 	if err = subProcess.Start(); err != nil {
 		return false
 	}
@@ -136,8 +151,6 @@ func RunPython3(SubmissionID int, SubmissionPath string) bool {
 		return false
 	}
 
-	defer fi.Close()
-
 	inputBuff := bufio.NewReader(fi)
 
 	for {
@@ -150,6 +163,9 @@ func RunPython3(SubmissionID int, SubmissionPath string) bool {
 
 	subProcess.Wait()
 
+	stdout.Close()
+	fi.Close()
+
 	return true
 }