【sduoj】让程序执行外部命令

2021SC@SDUSC

引言

与选择题、填空题不同的是，编程题很难用比较字符串是否相同的方式来判断正误。对于某些稍微复杂点的题目，一千个人可能提交一千份不同的代码。评判某份代码是否是正确的，只要看它能不能跑出正确的结果就行了。

以下有一份代码。

package main

import "fmt"

func main() {
   
   
	fmt.Println("Hello World!")
}

对于人来说，我们可以创建一个main.go文件，将上述代码复制进这个文件中，然后用以下指令来编译并运行这个程序。

go run main.go

如果想要让判题机替代人来完成这个工作，就需要写一段程序来控制计算机完成以上操作，暂令该程序的文件名为judge.go，代码如下：

package main

import (
	"bytes"
	"os/exec"
)

func main() {
   
   
	in := bytes.NewBuffer(nil)
	cmd := exec.Command("sh")
	cmd.Stdin = in
	go func() {
   
   
		in.WriteString("go run Main.go > output.txt\n")
		in.WriteString("exit\n")
	}()
	_ = cmd.Run()
}

运行判断程序。

go run judge.go

我们会看到，在当前目录下，多了一个output.txt文件，里面保存着Main.go的运行结果。

源码分析

exec.Commond

os/exec包用于执行外部的命令，其下的Commond方法初始化并返回一个Cmd对象。

filepath.Base返回路径的最后一个元素，比如将/home/user/path.go传进去，会返回一个path.go。LookPath会在环境变量指定的目录中搜索可执行文件，返回值lp是完整路径或者相对于当前目录的一个相对路径，如传入ls会得到/usr/bin/ls。

func Command(name string, arg ...string) *Cmd {
   
   
	cmd := &Cmd{
   
   
		Path: name,
		Args: append([]string{
   
   name}, arg...),
	}
	if filepath.Base(name) == name {
   
   
		if lp, err := LookPath(name); err != nil {
   
   
			cmd.lookPathErr = err
		} else {
   
   
			cmd.Path = lp
		}
	}
	return cmd
}

以下是Cmd结构体的全部属性。

type Cmd struct {
   
   

	// 将要执行的命令的路径
	Path string

	// 命令的参数
	Args []string