本文为从零开始写 Docker 系列第十五篇，实现 mydocker run -e, 支持在启动容器时指定环境变量，让容器内运行的程序可以使用外部传递的环境变量。
完整代码见：https://github.com/lixd/mydocker
欢迎 Star
推荐阅读以下文章对 docker 基本实现有一个大致认识：
- 核心原理：深入理解 Docker 核心原理：Namespace、Cgroups 和 Rootfs
- 基于 namespace 的视图隔离：探索 Linux Namespace：Docker 隔离的神奇背后
- 基于 cgroups 的资源限制
- 基于 overlayfs 的文件系统：Docker 魔法解密：探索 UnionFS 与 OverlayFS
- 基于 veth pair、bridge、iptables 等等技术的 Docker 网络：揭秘 Docker 网络：手动实现 Docker 桥接网络
开发环境如下：
root@mydocker:~# lsb_release -a
No LSB modules are available.
Distributor ID: Ubuntu
Description: Ubuntu 20.04.2 LTS
Release: 20.04
Codename: focal
root@mydocker:~# uname -r
5.4.0-74-generic
注意：需要使用 root 用户
1. 概述
本节实现mydocker run -e flag,支持在启动容器时指定环境变量，让容器内运行的程序可以使用外部传递的环境变量。
2. 实现
实现也比较简单，就是在构建 cmd 的时候指定 Env 参数。
- 1）run 命令增加 -e 参数
- 2）cmd 中指定 Env 参数
run 命令增加 -e flag
在原来的基础上,增加 -e 选项指定环境变量，由于可能存在多个环境变量，因此允许用户通过多次使用 -e 选项来传递多个环境变量。
var runCommand = cli.Command{
Name: "run",
Usage: `Create a container with namespace and cgroups limit
mydocker run -it [command]
mydocker run -d -name [containerName] [imageName] [command]`,
Flags: []cli.Flag{
// 省略其他内容
cli.StringSliceFlag{ // 增加 -e flag
Name: "e",
Usage: "set environment,e.g. -e name=mydocker",
},
},
Action: func(context *cli.Context) error {
if len(context.Args()) < 1 {
return fmt.Errorf("missing container command")
}
envSlice := context.StringSlice("e") // 获取 env 并传递
Run(tty, cmdArray, envSlice, resConf, volume, containerName, imageName)
return nil
},
}
注意到这里的类型是cli. StringSliceFlag,即字符串数组参数，因为这是针对传入多个环境变量的情况。
然后增加对环境变量的解析，并且传递给 Run 函数。
cmd 对象指定 Env 参数
由于原来的 command 实际就是容器启动的进程，所以只需要在原来的基础上，增加一下环境变量的配置即可。
默认情况下，新启动进程的环境变量都是继承于原来父进程的环境变量，但是如果手动指定了环境变量，那么这里就会覆盖掉原来继承自父进程的变量。
由于在容器的进程中，有时候还需要使用原来父进程的环境变量，比如 PATH 等，因此这里会使用 os.Environ() 来获取宿主机的环境变量，然后把自定义的变量加进去。
func NewParentProcess(tty bool, volume, containerId, imageName string, envSlice []string) (*exec.Cmd, *os.File) {
// 省略其他内容
cmd.Env = append(os.Environ(), envSlice...)
return cmd, writePipe
}
到此，环境变量的实现就完成了。
3. 测试
通过 -e 注入两个环境变量测试一下
$ go build .
./mydocker run -it -name c1 -e user=17x -e name=mydocker busybox sh
然后在容器中查看环境变量
/ # env|grep user
user=17x
/ # env|grep name
name=mydocker
这里可以看到，手动指定的环境变量 user=17x 和name=mydocker 都已经可以在容器内可见了。
说明，-e flag 基本 ok。
下面创建一个后台运行的容器，查看一下是否可以。
root@mydocker:~/feat-run-e/mydocker# ./mydocker run -d -name c2 -e user=17x -e name=mydocker busybox top
查看 ID
root@mydocker:~/feat-run-e/mydocker# ./mydocker ps
ID NAME PID STATUS COMMAND CREATED
9250006592 c2 228185 running top 2024-02-27 10:37:09
然后通过 exec 命令进入容器，查看环境变量
root@mydocker:~/feat-run-e/mydocker# ./mydocker exec 9250006592 sh
# 容器内
/ # env|grep user
/ #
可以发现，并没有看到创建时指定的环境变量。
这里看不到环境变量的原因是：exec 命令其实是 mydocker 创建
的另外一个进程，这个进程的父进程其实是宿主机的的进程，并不是容器进程的。
因为在 Cgo 里面使用了 setns 系统调用，才使得这个进程进入到了容器内的命名空间
由于环境变量是继承自父进程的，因此这个 exec 进程的环境变量其实是继承自宿主机的，所以在 exec 进程内看到的环境变量其实是宿主机的环境变量。
因此需要修改一下 exec 命令实现，使其能够看到容器中的环境变量。
4. 修改 mydocker exec 命令
首先提供了一个函数，可以根据指定的 PID 来获取对应进程的环境变量。
// getEnvsByPid 读取指定PID进程的环境变量
func getEnvsByPid(pid string) []string {
path := fmt.Sprintf("/proc/%s/environ", pid)
contentBytes, err := os.ReadFile(path)
if err != nil {
log.Errorf("Read file %s error %v", path, err)
return nil
}
// env split by \u0000
envs := strings.Split(string(contentBytes), "\u0000")
return envs
}
由于进程存放环境变量的位置是/proc/，因此根据给定的 PID 去读取这个文件，便可以获取环境变量。
在文件的内容中，每个环境变量之间是通过\u0000分割的，因此以此为标记来获取环境变量数组。
然后再启动 exec 进程时把容器中的环境变量也一并带上：
func ExecContainer(containerName string, comArray []string) {
// 省略其他内容
// 把指定PID进程的环境变量传递给新启动的进程，实现通过exec命令也能查询到容器的环境变量
containerEnvs := getEnvsByPid(pid)
cmd.Env = append(os.Environ(), containerEnvs...)
}
这样，exec 到容器内之后就可以看到所有的环境变量了。
再次测试一下，使用通用的 exec 命令进入容器，查看能否看到环境变量
root@mydocker:~/feat-run-e/mydocker# ./mydocker exec 9250006592 sh
/ # env|grep user
user=17x
/ # env|grep name
name=mydocker
ok，mydocker exec 已经可以获取到容器中的环境变量了。
5. 小结
本章实现了mydocker run -e flag 的添加，支持启动容器时传递环境变量到容器中。
核心实现就是启动 cmd 时指定 Env 参数，具体如下：
cmd := exec.Command("/proc/self/exe", "init")
cmd.Env = append(os.Environ(), envSlice...)
同时修改了 exec 命令，将容器中的环境变量append 到 exec 进程，便于查看。
【从零开始写 Docker 系列】持续更新中，搜索公众号【探索云原生】订阅，阅读更多文章。
完整代码见：https://github.com/lixd/mydocker
欢迎关注~
相关代码见 refactor-isolate-rootfs 分支,测试脚本如下：
需要提前在 /var/lib/mydocker/image 目录准备好 busybox.tar 文件，具体见第四篇第二节。
# 克隆代码
git clone -b feat-run-e https://github.com/lixd/mydocker.git
cd mydocker
# 拉取依赖并编译
go mod tidy
go build .
# 测试
./mydocker run -d -name c1 -e user=17x -e name=mydocker busybox top
# 查看容器 Id
./mydocker ps
# stop 停止指定容器
./mydocker exec ${containerId} sh