翻译自:https://sj14.gitlab.io/post/2...
介绍
在本文中,我们将使用 Go 语言,编写一个最小的 UNIX(-like)操作系统 SHELL,它只需要大概 60 行代码。你需要稍微了解一些 Go 语言(知道如何编译简单的项目),以及简单使用 UNIX Shell。
UNIX 非常简单,简单到一个天才都能理解它的简单性 - Dennis Ritchie
当然,我并非天才,我也不太确定 Dennis Ritchie 所说的,是否也包括运行于用户空间的工具。Shell 只是完整操作系统的一小部分(相较于内核,它真的是一个简单的部分),但我希望在本文的结尾,你可以感到吃惊,吃惊于编写一个 SHELL,所用到的知识如此少。
什么是SHELL
给 SHELL 下定义有点困难。我认为 SHELL 可以理解为你所使用的操作系统,基本的用户界面。你可以在 SHELL 中输入命令,然后接收一些反馈输出。如果想了解更多信息,或者更明确的定义,请查阅 维基百科)。
一些 SHELL 的例子:
有像 Windows 和 GNOME 这种图形界面 SHELL,但大多数 IT 相关人员(至少我是),当谈论起 SHELL,指的是基于文本的 SHELL(上面列表的头两项)。当然,也可以简化的定义为非图形界面 SHELL。
事实上,SHELL 的功能可以定义为输入命令,然后接收该命令的输出。想看个例子?运行 ls
命令,输出目录的内容。
Input:
ls
Output:
Applications etc
Library home
...
就是这样,十分简单。让我们开始吧!
输入循环
要执行一个命令,我们必须接收输入。而输入来自我们人类,使用键盘进行的。
键盘是我们的标准输入设备(os.Stdin),我们可以访问并读取它。当我们按下回车键的时候,会创建新的一行。这行新的文本以 \n
结尾。当敲击回车键的时候,所有存储在输入区的内容将被输入。
reader := bufio.NewReader(os.Stdin)
input, err := reader.ReadString('\n')
让我们将这些代码输入进我们的 main.go
文件,ReadingString 方法被嵌套在 for 循环中,所以我们可以反复输入命令。当读取输入,发生错误时,我们可以将错误信息输出到标准错误处理设备(os.Stderr)。如果我们使用 fmt.Println
,但并没有指定输出设备,这个错误信息还是会输出到标准输出设备中(os.Stdout)。这并不会改变 SHELL 的功能,但是输出到单独的设备,可以方便过滤输出,以进行下一步处理。
func main() {
reader := bufio.NewReader(os.Stdin)
for {
// Read the keyboad input.
input, err := reader.ReadString('\n')
if err != nil {
fmt.Fprintln(os.Stderr, err)
}
}
}
执行命令
现在,我们打算执行输入的命令。增加一个名为 execInput
的新的函数,他接收输入的字符串作为参数。首先,我们移除输入结尾的换行符。接下来,通过 exec.Command(input)
来准备执行命令,设置参数,以及捕获输出的结果和错误。最后,通过 cmd.Run()
来执行。
func execInput(input string) error {
// Remove the newline character.
input = strings.TrimSuffix(input, "\n")
// Prepare the command to execute.
cmd := exec.Command(input)
// Set the correct output device.
cmd.Stderr = os.Stderr
cmd.Stdout = os.Stdout
// Execute the command and return the error.
return cmd.Run()
}
原型
接着,在循环语句上面,添加一个美化作用的指示器(>),在循环语句下面,添加新的 execInput
函数,此时,主要功能就完成了。
func main() {
reader := bufio.NewReader(os.Stdin)
for {
fmt.Print("> ")
// Read the keyboad input.
input, err := reader.ReadString('\n')
if err != nil {
fmt.Fprintln(os.Stderr, err)
}
// Handle the execution of the input.
if err = execInput(input); err != nil {
fmt.Fprintln(os.Stderr, err)
}
}
}
是时候执行一次测试了。使用 go run main.go
构建并运行咱们的 SHELL。你将看到输入标识符 >
,此时可以接受输入。举个例子,我们可以执行 ls
命令。
> ls
LICENSE
main.go
main_test.go
不错,可以运行!咱们的程序此时可以执行 ls
命令,并输出当前目录的内容。你可以像退出其他程序一样,使用 ctrl+c
,退出它。
参数
让我们命令后面加个参数,如 ls -l
。
> ls -l
执此时执行会报错:exec: "ls -l": executable file not found in $PATH
。
这是因为我们的 SHELL 尝试执行 ls -l
,但是并没有找到叫这个名字的程序。我们的意思是执行 ls
,带上 -l
的参数。当前,我们的程序还不支持接受命令参数。要修复这个问题,需要修改 execLine
函数,将要执行的命令以空格拆分。
func execInput(input string) error {
// Remove the newline character.
input = strings.TrimSuffix(input, "\n")
// Split the input to separate the command and the arguments.
args := strings.Split(input, " ")
// Pass the program and the arguments separately.
cmd := exec.Command(args[0], args[1:]...)
...
}
程序的名字现在存储在 args[0] 中,程序执行的参数存储在数组其他索引中。执行 ls -l
现在可以得到预期的结果。
> ls -l
total 24
-rw-r--r-- 1 simon staff 1076 30 Jun 09:49 LICENSE
-rw-r--r-- 1 simon staff 1058 30 Jun 10:10 main.go
-rw-r--r-- 1 simon staff 897 30 Jun 09:49 main_test.go
更改目录(cd)
现在,我们已经可以带着参数执行命令了。现有的这些功能,距离达到一个最小的可用性,只差了一点点。你也许在使用我们的 Shell 时候,已经注意到了:你无法通过 cd
改变当前命令执行的目录。
> cd /
> ls
LICENSE
main.go
main_test.go
不,这不是我们根目录的内容。那为什么 cd
命令不起作用呢?要理解这点很容易:没有真正的 cd
程序,该功能是 SHELL
的内置命令。
我们必须对 execInput
函数再次进行修改。在 Split
方法后面,我们添加 switch
结构语句,并将 args[0] 作为它的参数。当这个命令是 cd
,我们检查它后面是否还有参数,如果没有指定参数,我们无法改变当前目录(在大多数 SHELL 中,不指定参数,将跳转到主目录)。当 args[1]
中有一个后续参数时(存储路径的参数),我们使用 os.Chdir(args[1])
更改目录。在 case
块的末尾,我们返回 execInput
函数以停止其他处理。
因为如此简单,我们在 cd
块后面,再添加一个 exit
命令,exit 可以用来退出当前SHELL(另一个退出方法是 CTRL+C
)。
// Split the input to separate the command and the arguments.
args := strings.Split(input, " ")
// Check for built-in commands.
switch args[0] {
case "cd":
// 'cd' to home dir with empty path not yet supported.
if len(args) < 2 {
return errors.New("path required")
}
// Change the directory and return the error.
return os.Chdir(args[1])
case "exit":
os.Exit(0)
}
...
可以看到,此时输出的内容,相较于之前的输出结果,更像是我们的根目录。
> cd /
> ls
Applications
Library
Network
System
...
至此,我们已经完成了这个简单的 SHEEL 的编写。
考虑改善的地方
此时,如果你觉得有些无聊,你可以尝试改进这个 SHELL。下面是一些可以改善的点:
修改光标所在行的显示:
- 增加当前目录
- 增加机器名称
- 增加当前用户
- 通过输入 up/down 键,来翻阅输入的历史
结尾
至此,本文已接近尾声,我希望你读的愉快。如你所见,SHELL 背后的概念十分简单。
Go 同样是一门十分简单的编程语言,它帮助我们更快的得到想要的结果。我们无需关心内存管理。Rob Pike 和 Ken Thompson,以及 Robert Griesemer 共同创造了 Go,他们也创造了 Unix,所以,使用 Go 编写 SHELL 是个很好的选择。
我也一直在学习,如果你发现本文有哪些可改进的地方,请联系我。
有疑问加站长微信联系(非本文作者)