compiler.go

package compiler

import (
	"errors"
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"strings"

	"github.com/MarkLux/JudgeServer/config"
	"github.com/MarkLux/Judger_GO"
)

/*
* compile the program in the judger sandbox.
* return the executable file path
 */

func Compile(compileConfig config.CompileConfig, srcPath string, outputDir string) (string, error) {
	exePath := filepath.Join(outputDir, compileConfig.ExeName)
	// replace param then build the real compile command.
	replacements := map[string]string{
		"{src_path}": srcPath,
		"{exe_dir}":  outputDir,
		"{exe_path}": exePath,
	}
	command := compileConfig.CompileCommand.FillWith(replacements)
	// log.Println("compile command: ", command)
	// file compiler.out contains the ouput of compile progarm's output (rewrite).
	compilerOut := filepath.Join(outputDir, "compiler.out")

	// split the command into execute path and args.
	args := strings.Split(command, " ")

	//parse args

	result, err := judger.JudgerRun(judger.Config{
		MaxCpuTime:       compileConfig.MaxCpuTime,
		MaxRealTime:      compileConfig.MaxRealTime,
		MaxMemory:        compileConfig.MaxMemory,
		MaxStack:         128 * 1024 * 1024,
		MaxOutPutSize:    judger.UNLIMITED,
		MaxProcessNumber: judger.UNLIMITED,
		ExePath:          args[0],
		InputPath:        srcPath,
		OutputPath:       compilerOut,
		ErrorPath:        compilerOut,
		Args:             args[1:], // suit new Judger,ignore the exepath
		Env:              []string{"PATH=" + os.Getenv("PATH")},
		LogPath:          config.COMPILER_LOG_PATH,
		SecCompRuleName:  "none",
		Uid:              config.COMPILER_USER_UID,
		Gid:              config.COMPILER_GROUP_UID,
	})

	// debug output

	if err != nil {
		return "", err
	}

	if result.Result != judger.SUCCESS {
		// log.Printf("Compile Result\n")
		// log.Printf("%#v\n", result)
		// read the compiler output and
		_, err := os.Stat(compilerOut)
		var errOut string
		if err == nil {
			errByte, _ := ioutil.ReadFile(compilerOut)
			errOut = string(errByte[:])
			log.Println(errOut)
			//os.Remove(compilerOut)
		} else {
			errOut = fmt.Sprintf("Compiler Runtime Error , info %#v", result)
		}
		return exePath, errors.New(errOut)
	}

	// os.Remove(compilerOut)
	return exePath, nil
}