JUDGER_AND_CLIENT/Judger/src/kernel/Judger.java

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package kernel;

import MyCache.Shared;
import common.Config;
import resultData.RunInfo;
import common.Const;
//import gui.Control;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import log.Log;
import resultData.CompileInfo;
import resultData.Result;
import tool.ThreadTool;
import log.Log;

/**
 *
 * @author Administrator
 */
public class Judger {

    String sourceFile;
    ExeCommand exe;
    Boolean isFound;
    String mainClassName; //
    String sourceDir;
    
    public Judger() {
        Config.freshConfig();
        sourceFile = "";
        exe = new ExeCommand();
        isFound = true;
        //checkCompiler();
    }
    
    public Boolean compileFound(){
        return isFound;
    }

//    private void checkCompiler() {
//        //for c
//        File file1 = new File(Config.getCompilerDir("c") + File.separator + "gcc.exe");
//        File file2 = new File(Config.getCompilerDir("c") + File.separator + "gcc.exe");
//        if (!file1.exists() || !file2.exists()) {
//            System.out.println("<22><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>δ<EFBFBD>ҵ<EFBFBD>");
//            isFound = false;
//        }
//    }
    
    public Boolean checkForCompiler(){
        
        File file1 = new File(Config.getCompilerDir("c",null) + File.separator + "gcc.exe");
        File file2 = new File(Config.getCompilerDir("c",null) + File.separator + "gcc.exe");
        if (!file1.exists() || !file2.exists()) {
            System.out.println("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>δ<EFBFBD>ҵ<EFBFBD>");
            isFound = false;
            return false;
        }
        isFound = true;
        return true;
    }

    private void saveSourceCodeFile(String language, String sourceCode) {
        File dir = new File(Config.getSourcePath());
        if (!dir.exists()) {
            dir.mkdirs();
        }
        dir = new File(Config.getTargetPath());
        if (!dir.exists()) {
            dir.mkdirs();
        }
        language = language.toLowerCase();
        try {
            //String s= Config.getTargetPath();
            sourceDir =Config.getSourcePath()+File.separator+"output" ;
            sourceFile =sourceDir+ File.separator + "Main"
                    + Const.getLatterSuffix(language);
            
            if(language.equals("java")){
                mainClassName=findMainClassName(sourceCode);
                sourceFile = sourceDir+ File.separator + findMainClassName(sourceCode) //TODO
                    + Const.getLatterSuffix(language);
                //System.out.println(sourceFile);
            }
            File fl =new File(sourceDir);
            if(!fl.exists()) {
                 fl.mkdirs();
            }          
            BufferedWriter out = new BufferedWriter(new FileWriter(sourceFile));
            out.write(sourceCode);
            out.close();
        } catch (IOException e) {
            e.printStackTrace();
            Log.writeExceptionLog(e.getMessage());         
        }

    }

    private String linkCommand(String language) {
        //String language = "c";
        String linkCommand = Config.getCompilerDir(language) + File.separator + "g++ " +"\""+ Config.getTargetPath() +File.separator+"output"+File.separator+  "Main"+".o"+"\"" + " -o " +"\""+ Config.getTargetPath()+File.separator+"output"+File.separator + "Main"+".exe"+"\"\n";      
        return linkCommand;
    }
//mingw32-g++.exe -Wall -g  -c E:\Downloads\aaa\aa.cpp -o obj\Debug\aa.o
//mingw32-g++.exe  -o bin\Debug\aaa.exe obj\Debug\aa.o   
    private String compileCommand(String language) {
        String compileCommand = "";
        language = language.toLowerCase();
        if (language.equals("c")) {
            compileCommand += "\"" + Config.getCompilerDir(language) + File.separator + "gcc\" -c " + "\""+sourceFile +"\""+ " -o " +"\""+Config.getTargetPath()+ File.separator+"output"+ File.separator + "Main"+".o"+"\"\n";
        } else if (language.equals("java")) {
            compileCommand += "\"" + Config.getCompilerDir(language) + File.separator + "javac\" " + sourceFile; //todo<64>ļ<EFBFBD>·<EFBFBD><C2B7>
        } else if (language.equals("cpp")||language.equals("c++")) {
            compileCommand += "\"" + Config.getCompilerDir(language) + File.separator + "g++\" -Wall -g -std=c++14 -c "
                    + "\""+sourceFile+"\"" + " -o " + "\""+Config.getTargetPath()+ File.separator+"output"+ File.separator + "Main"+".o"+"\"\n";
        } else {
            CompileInfo.info = "this programing language is not support!!!";
        }
        return compileCommand;
    }

    private String runCommand(String language) {
        String runCommand = "";
        if (language.equals("c")) {
            runCommand +="\""+ Config.getTargetPath()+ File.separator+"output"+File.separator + "Main"+"\"";
            
        } else if (language.equals("java")) {
        runCommand += Config.getCompilerDir(language) + File.separator + "java"+ " -cp " + Config.getSourcePath()+ File.separator+"output"+File.separator+ " "+mainClassName;  // TODO <20>ļ<EFBFBD>·<EFBFBD><C2B7> start            
        //System.err.println(runCommand);
        } else if (language.equals("cpp")||language.equals("c++")) {
            runCommand += "\""+ Config.getTargetPath()+ File.separator+"output"+File.separator  + "Main"+"\"";
        }
//        System.out.println(runCommand);
        return runCommand;
    }

    public int compile(String sourceCode, String language) {
        int result = -1;
        //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƿ<EFBFBD><C7B7>ڷ<EFBFBD>Χ<EFBFBD><CEA7>
        language = language.toLowerCase();
        if (language.equals("c") || language.equals("cpp")|| language.equals("c++") || language.equals("java")) {
            //ok
        } else {
            Result.status = Const.CE;
//            CompileInfo.remark = "<22><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>಻<EFBFBD><E0B2BB><EFBFBD><EFBFBD>Ҫ<EFBFBD><D2AA>";
            return result;
        }
        try {
            if (Shared.PID!=-1&&ThreadTool.findProcess(Shared.PID)) {  //tore0
                Runtime.getRuntime().exec("taskkill /f /t /PID "+Shared.PID).waitFor();
               // System.out.println("<22><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ɱ<EFBFBD><C9B1>");    
//                System.out.println("ɱ<><C9B1><EFBFBD><EFBFBD>"+"taskkill /f /t /PID "+Shared.PID);
            }
        } catch (IOException | InterruptedException e) {
            System.out.println(e.getMessage());
        }

        saveSourceCodeFile(language, sourceCode);

        int repeatTime = 3;
        String compileCom = compileCommand(language);
        for (int i = 0; i < repeatTime; i++) {
            result = exe.exeCompile(compileCom,"Path="+Config.getCompilerDir(language));
            if (result == 0) {
                if (language.equals("c") || language.equals("cpp")||language.equals("c++")) {
                    result = exe.exeLink(linkCommand(language),"Path="+Config.getCompilerDir(language));
                    if(result==0){
                        break;
                    }
                } else {
                    break;
                }
            }
        }
        return result;
    }

    public int run(String language, String input, int timeLimit) {
        language = language.toLowerCase();
        return exe.exeRun(runCommand(language), "Path="+Config.getCompilerDir(language), input, timeLimit);
    }

    public boolean check(String stdAns) {
        String output= preProcess(RunInfo.info);
        String stdans = preProcess(stdAns);

        int status = Const.WA;
        if (output.equals(stdans)) {
            status = Const.AC;
        } else {
            String output1 = removeSP(output);
            String ans1 = removeSP(stdans);
            if (output1.equals(ans1)) {
                status = Const.PE;
            }

        }
        Result.status = status;
        return (status == Const.AC);
    }

    private String removeSP(String s) {
        StringBuilder temp = new StringBuilder();
        for (int i = 0; i < s.length();i++) {
            if (s.charAt(i) != '\n' && s.charAt(i) != '\t' && s.charAt(i) != '\r' && s.charAt(i) != '\f' && s.charAt(i)!=' '&& s.charAt(i)!='\000') {
                temp.append(s.charAt(i));
            }
        }
     //System.out.println((int)temp.charAt(temp.length()-1));
        return temp.toString();
    }
    
     public static String replaceBlank(String str) {
        String dest = "";
        if (str!=null) {
            Pattern p = Pattern.compile("\\s*|\t|\r|\n");
            Matcher m = p.matcher(str);
            dest = m.replaceAll("");
        }
        char chars[]=dest.toCharArray();
        return dest;
    }

    private String preProcess(String s) {
        StringBuilder temp = new StringBuilder();
        int len = s.length();
        //<2F><>\r\n<>滻Ϊ\n
        s = s.replace("\r\n", "\n");
//        for (int i = 0; i < len;) {
//            if ((i < len - 1) && s.charAt(i + 1) == '\n' && s.charAt(i) == '\r') {
//                temp.append('\n');
//                i = i + 2;
//            } else {
//                temp.append(s.charAt(i));
//                i = i + 1;
//            }
//        }
       char[] chars=s.toCharArray();
       int lastIndex=chars.length-1;
       //ȥ<><C8A5>ĩβ<C4A9><CEB2>\n
       for(int i=chars.length-1;i>=0;i--){
           if(chars[i]=='\n'){
               lastIndex=i-1;
           }else{
               break;
           }
       }
        return new String(chars,0,lastIndex+1);
    }

     /*
    <EFBFBD>ҵ<EFBFBD>һ<EFBFBD><EFBFBD>java<EFBFBD>ļ<EFBFBD><EFBFBD>а<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ĺ<EFBFBD><EFBFBD><EFBFBD>main<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
    */
    static String findMainClassName(String code){
        //1.<2E><><EFBFBD><EFBFBD>Ѱ<EFBFBD><D1B0>һ<EFBFBD><D2BB>java<76>ļ<EFBFBD><C4BC>й<EFBFBD><D0B9><EFBFBD><EFBFBD><EFBFBD>
        int index=code.indexOf("public class");
        if(index==-1){
            //2.<2E><><EFBFBD>Һ<EFBFBD><D2BA><EFBFBD>main<69><6E><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
            int startIndex=code.indexOf("public static void main");
           if(startIndex==-1){
               return null;
           }
           startIndex=code.lastIndexOf("class",startIndex);
            startIndex+=(new String("class")).length();
            int endIndex=code.indexOf('{',startIndex);
            String mainClassName=code.substring(startIndex,endIndex);
            return mainClassName;
        }
            
        index+=(new String("public class")).length();
        int index1=code.indexOf('{',index);
        String mainClassName=code.substring(index, index1);
        mainClassName=mainClassName.trim();
        
        return mainClassName;
    }
}