add docker image and env

Author: pibigstar

README.md

@@ -1,32 +1,14 @@
 # go-docker
 > 用go写一个docker
 
-**注意**
+## 环境配置
+- 开发环境： windows
+- 运行环境： CentOS
+
+### windows中goland配置
 > windows下要修改goland的OS环境为 linux,不然只会引用`exec_windows.go`而不会引用`exec_linxu_go`
 > 在Setting->Go->Build Tags & Vendoring -> OS=linux
 
-## namespace
-- uts : 隔离主机名
-- pid : 隔离进程pid
-- user : 隔离用户
-- network : 隔离网络
-- mount : 隔离挂载点
-- ipc : 隔离System VIPC和POSIX message queues
-
-## cgroup
-> 主要是使用三个组件相互协作实现的，分别是：subsystem, hierarchy, cgroup,
-
-- cgroup: 是对进程分组管理的一种机制
-- subsystem: 是一组资源控制的模块
-- hierarchy: 把一组cgroup串成一个树状结构(可让其实现继承)
-
-### 实现方式
-> 主要实现方式是在`/sys/fs/cgroup/` 文件夹下，根据限制的不同，创建一个新的文件夹即可，kernel会将这个文件夹
-> 标记为它的`子cgroup`, 比如要限制内存使用，则在`/sys/fs/cgroup/memory/` 下创建`test-limit-memory`文件夹即可，将
-> 内存限制数写到该文件夹里面的 `memory.limit_in_bytes`即可
-
-
-## 环境配置
 ### 设置CentOS支持aufs
 查看是否支持
 ```bash
@@ -50,6 +32,67 @@ grub2-mkconfig -o /boot/grub2/grub.cfg
 # 重启计算机
 reboot
 ```
+### 配置busybox
+```bash
+# 下载 busybox
+docker pull busybox
+# 运行
+docker run -d busybox top -b
+# 导出
+docker export -o busybox.tar (容器ID)
+# 解压到 /root文件夹下
+cd /root
+mkdir busybox
+tar -xvf busybox.tar -C busybox/
+```
+
+## 使用指南
+```bash
+# 编译
+go build .
+
+# 启动一个容器
+./go-docker run -ti --name test sh
+
+# 后台启动
+./go-docker run -d --name test sh
+
+# 进入容器
+./go-docker exec test sh
+
+# 查看容器日志
+./go-docker logs test
+
+# 查看容器列表
+./go-docker ps
+
+# 停止容器
+./go-docker stop test
+
+# 删除容器
+./go-docker rm test
+```
+
+## docker核心技术
+### namespace
+- uts : 隔离主机名
+- pid : 隔离进程pid
+- user : 隔离用户
+- network : 隔离网络
+- mount : 隔离挂载点
+- ipc : 隔离System VIPC和POSIX message queues
+
+### cgroup
+> 主要是使用三个组件相互协作实现的，分别是：subsystem, hierarchy, cgroup,
+
+- cgroup: 是对进程分组管理的一种机制
+- subsystem: 是一组资源控制的模块
+- hierarchy: 把一组cgroup串成一个树状结构(可让其实现继承)
+
+#### 实现方式
+> 主要实现方式是在`/sys/fs/cgroup/` 文件夹下，根据限制的不同，创建一个新的文件夹即可，kernel会将这个文件夹
+> 标记为它的`子cgroup`, 比如要限制内存使用，则在`/sys/fs/cgroup/memory/` 下创建`test-limit-memory`文件夹即可，将
+> 内存限制数写到该文件夹里面的 `memory.limit_in_bytes`即可
 
 ## 指令小记
 

command.go

@@ -45,6 +45,10 @@ var runCommand = cli.Command{
 			Name:  "name",
 			Usage: "container name",
 		},
+		cli.StringSliceFlag{
+			Name:  "e",
+			Usage: "docker env",
+		},
 	},
 	Action: func(ctx *cli.Context) error {
 		if len(ctx.Args()) < 1 {
@@ -57,22 +61,26 @@ var runCommand = cli.Command{
 			CpuShare:    ctx.String("cpushare"),
 		}
 		// cmdArray 为容器运行后，执行的第一个命令信息
-		// cmdArray[0] 为命令内容, 后面的为命令参数
+		// cmdArray[0] 为镜像名, .Tail() 是去掉第一个后的全部参数
 		var cmdArray []string
-		for _, arg := range ctx.Args() {
+		for _, arg := range ctx.Args().Tail() {
 			cmdArray = append(cmdArray, arg)
 		}
 
 		tty := ctx.Bool("ti")
-		volume := ctx.String("v")
 		detach := ctx.Bool("d")
 
 		if tty && detach {
 			return fmt.Errorf("ti and d paramter can not both provided")
 		}
 
 		containerName := ctx.String("name")
-		Run(cmdArray, tty, res, volume, containerName)
+		volume := ctx.String("v")
+		// 要运行的镜像名
+		imageName := ctx.Args().Get(0)
+		envs := ctx.StringSlice("e")
+
+		Run(cmdArray, tty, res, containerName, imageName, volume, envs)
 		return nil
 	},
 }
@@ -124,7 +132,7 @@ var logCommand = cli.Command{
 			return fmt.Errorf("missing container name")
 		}
 		containerName := ctx.Args().Get(0)
-		container.LookContainerName(containerName)
+		container.LookContainerLog(containerName)
 		return nil
 	},
 }

common/common.go

@@ -3,8 +3,6 @@ package common
 const (
 	RootPath   = "/root/"
 	MntPath    = "/root/mnt/"
-	BusyBox    = "busybox"
-	BusyBoxTar = "busybox.tar"
 	WriteLayer = "writeLayer"
 )
 

container/exec.go

@@ -1,6 +1,8 @@
 package container
 
 import (
+	"fmt"
+	"io/ioutil"
 	"os"
 	"os/exec"
 	"strings"
@@ -24,7 +26,25 @@ func ExecContainer(containerName string, cmdArray []string) {
 	cmd.Stderr = os.Stderr
 	err = os.Setenv(common.EnvExecPid, info.Pid)
 	err = os.Setenv(common.EnvExecCmd, strings.Join(cmdArray, " "))
+	// 设置环境变量
+	envs := getEnvsByPid(info.Pid)
+	cmd.Env = append(os.Environ(), envs...)
+
 	if err = cmd.Run(); err != nil {
 		logrus.Errorf("exec cmd run, err: %v", err)
 	}
 }
+
+func getEnvsByPid(pid string) []string {
+	envFilePath := fmt.Sprintf("/proc/%s/environ", pid)
+	file, err := os.Open(envFilePath)
+	if err != nil {
+		logrus.Errorf("open env file, path: %s, err: %v", envFilePath, err)
+		return nil
+	}
+	bs, err := ioutil.ReadAll(file)
+	if err != nil {
+		logrus.Errorf("read env file, err: %v", err)
+	}
+	return strings.Split(string(bs), "\u0000")
+}

container/info.go

@@ -24,6 +24,7 @@ type ContainerInfo struct {
 	Status     string `json:"status"`
 }
 
+// 记录容器信息
 func RecordContainerInfo(containerPID int, cmdArray []string, containerName, containerID string) error {
 	info := &ContainerInfo{
 		Pid:        strconv.Itoa(containerPID),

container/log.go

@@ -11,7 +11,8 @@ import (
 	"go-docker/common"
 )
 
-func LookContainerName(containerName string) {
+// 查看容器内日志信息
+func LookContainerLog(containerName string) {
 	logFileName := path.Join(common.DefaultContainerInfoPath, containerName, common.ContainerLogFileName)
 	file, err := os.Open(logFileName)
 	if err != nil {

container/process.go

@@ -12,7 +12,7 @@ import (
 )
 
 // 创建一个会隔离namespace进程的Command
-func NewParentProcess(tty bool, volume, containerName string) (*exec.Cmd, *os.File) {
+func NewParentProcess(tty bool, volume, containerName, imageName string, envs []string) (*exec.Cmd, *os.File) {
 	readPipe, writePipe, _ := os.Pipe()
 	// 调用自身，传入 init 参数，也就是执行 initCommand
 	cmd := exec.Command("/proc/self/exe", "init")
@@ -40,10 +40,13 @@ func NewParentProcess(tty bool, volume, containerName string) (*exec.Cmd, *os.Fi
 		}
 		cmd.Stdout = file
 	}
+	// 设置额外文件句柄
 	cmd.ExtraFiles = []*os.File{
 		readPipe,
 	}
-	err := NewWorkSpace(common.RootPath, common.MntPath, volume)
+	// 设置环境变量
+	cmd.Env = append(os.Environ(), envs...)
+	err := NewWorkSpace(volume, containerName, imageName)
 	if err != nil {
 		logrus.Errorf("new work space, err: %v", err)
 	}

container/remove.go

@@ -9,6 +9,7 @@ import (
 	"go-docker/common"
 )
 
+// 删除容器
 func RemoveContainer(containerName string) {
 	info, err := getContainerInfo(containerName)
 	if err != nil {