judge.go

package controller

import (
	"context"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"

	log "github.com/Sirupsen/logrus"
	"github.com/VOID001/D-judge/config"
	"github.com/VOID001/D-judge/request"
	"github.com/docker/engine-api/client"
	"github.com/pkg/errors"
)

const (
	ExitAC = 42
	ExitWA = 43
)

func (w *Worker) judge(ctx context.Context, rank int64, tid int64) (err error) {
	// Create testcase dir, use to store result
	execdir := filepath.Join(w.WorkDir, "execdir")
	testcasedir := filepath.Join(w.WorkDir, fmt.Sprintf("testcase%03d", rank))
	err = os.Mkdir(testcasedir, DirPerm)
	if err != nil {
		err = errors.Wrap(err, fmt.Sprintf("Judge error on Run#%d case %d", w.JudgeInfo.SubmitID, rank))
		return
	}
	// Build the judge script
	cli, er := client.NewClient(config.GlobalConfig.DockerServer, config.GlobalConfig.DockerVersion, nil, nil)
	if er != nil {
		err = errors.Wrap(er, fmt.Sprintf("Judge error on Run#%d case %d", w.JudgeInfo.SubmitID, rank))
		return
	}

	cmd := fmt.Sprintf("compare/run execdir/testcase.in execdir/testcase.out testcase001 < execdir/program.out 2> compare.err >compare.out")
	log.Debugf("executing command %s", cmd)
	info, err := w.execcmdAttach(ctx, cli, "root", cmd)
	//	time.Sleep(time.Second * 10)
	code := info.ExitCode
	if err != nil {
		err = errors.Wrap(err, fmt.Sprintf("Judge error on Run#%d case %d", w.JudgeInfo.SubmitID, rank))
		return
	}

	res := config.RunResult{}
	res.RunTime = 0 // This is uneccessary
	res.TestcaseID = tid
	res.JudgingID = w.JudgeInfo.JudgingID

	// Parse system meta and send to output system
	// For Domjudge compability
	// Save for Judge use
	data, er := ioutil.ReadFile(filepath.Join(execdir, "program.meta"))
	if er != nil {
		err = errors.Wrap(er, "judge error")
		return
	}
	res.OutputSystem = fmt.Sprintf("%s", data)

	switch code {
	case ExitWA:
		res.RunResult = config.ResWA
		// Report Accepted
	case ExitAC:
		res.RunResult = config.ResAC
		// Report Wrong Answer
	default:
		err = errors.New(fmt.Sprintf("Judge return unexpected exit code %d", code))
		return
	}

	// Remove execdir for next time use
	oldexecdir := fmt.Sprintf("%s%03d", execdir, rank)
	err = request.PostResult(ctx, res)
	if err != nil {
		err = errors.Wrap(err, "Judge error")
		return
	}
	err = os.Rename(execdir, oldexecdir)
	if err != nil {
		err = errors.Wrap(err, fmt.Sprintf("Judge error on Run#%d case %d", w.JudgeInfo.SubmitID, rank))
		return
	}
	return
}