runv cli refactor

Signed-off-by: Lai Jiangshan <jiangshanlai@gmail.com>

Author: laijs

container.go

@@ -0,0 +1,200 @@
+package main
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"syscall"
+
+	"github.com/golang/glog"
+	"github.com/hyperhq/runv/api"
+	"github.com/hyperhq/runv/hypervisor"
+	"github.com/opencontainers/runtime-spec/specs-go"
+)
+
+func startContainer(vm *hypervisor.Vm, container string, spec *specs.Spec, state *specs.State) error {
+
+	shim, err := os.FindProcess(state.Pid)
+	if err != nil {
+		return err
+	}
+
+	err = vm.StartContainer(container)
+	if err != nil {
+		glog.V(1).Infof("Start Container fail: fail to start container with err: %#v\n", err)
+		return err
+	}
+
+	err = shim.Signal(syscall.SIGUSR1)
+	if err != nil {
+		// cli refactor todo: remove the container
+		return err
+	}
+
+	err = execPoststartHooks(spec, state)
+	if err != nil {
+		glog.V(1).Infof("execute Poststart hooks failed %s\n", err.Error())
+	}
+
+	return nil
+}
+
+func createContainer(options runvOptions, vm *hypervisor.Vm, container, bundle, stateRoot string, spec *specs.Spec) (shim *os.Process, err error) {
+	containerSharedFs := filepath.Join(hypervisor.BaseDir, vm.Id, hypervisor.ShareDirTag, container)
+	if err = setupFs(bundle, containerSharedFs, spec); err != nil {
+		return nil, err
+	}
+
+	glog.V(3).Infof("vm.AddContainer()")
+	config := api.ContainerDescriptionFromOCF(container, spec)
+	r := vm.AddContainer(config)
+	if !r.IsSuccess() {
+		return nil, fmt.Errorf("add container %s failed: %s", container, r.Message())
+	}
+
+	// Prepare container state directory
+	stateDir := filepath.Join(stateRoot, container)
+	_, err = os.Stat(stateDir)
+	if err == nil {
+		glog.Errorf("Container %s exists\n", container)
+		return nil, fmt.Errorf("Container %s exists", container)
+	}
+	err = os.MkdirAll(stateDir, 0644)
+	if err != nil {
+		glog.V(1).Infof("%s\n", err.Error())
+		return nil, err
+	}
+	// Create sandbox dir symbol link in container root dir
+	vmRootLinkPath := filepath.Join(stateDir, "sandbox")
+	vmRootPath := sandboxPath(vm)
+	if err = os.Symlink(vmRootPath, vmRootLinkPath); err != nil {
+		return nil, fmt.Errorf("failed to create symbol link %q: %v", vmRootLinkPath, err)
+	}
+
+	// create shim and save the state
+	shim, err = createShim(options, container, "init")
+	if err != nil {
+		return nil, err
+	}
+	state := &specs.State{
+		Version: spec.Version,
+		ID:      container,
+		Pid:     shim.Pid,
+		Bundle:  bundle,
+	}
+	glog.V(3).Infof("save state id %s, boundle %s", container, bundle)
+	if err = saveStateFile(filepath.Join(stateDir, "state.json"), state); err != nil {
+		return nil, err
+	}
+
+	err = execPrestartHooks(spec, state)
+	if err != nil {
+		// cli refator todo stop container
+		glog.V(1).Infof("execute Prestart hooks failed, %s\n", err.Error())
+		return nil, err
+	}
+
+	// If runv is launched via docker/containerd, we start netlistener to watch/collect network changes.
+	// TODO: if runv is launched by cni compatible tools, the cni script can use `runv cni` cmdline to update the network.
+	// Create the listener process which will enters into the netns of the shim
+	options.withContainer = state
+	if err = startNsListener(options, vm); err != nil {
+		// cli refator todo stop container
+		glog.Errorf("start ns listener fail: %v", err)
+		return nil, err
+	}
+
+	return shim, nil
+}
+
+func deleteContainer(vm *hypervisor.Vm, container string, spec *specs.Spec, state *specs.State) error {
+	err := execPoststopHooks(spec, state)
+	if err != nil {
+		glog.V(1).Infof("execute Poststop hooks failed %s\n", err.Error())
+		return err
+	}
+	return nil
+}
+
+func addProcess(options runvOptions, vm *hypervisor.Vm, container, process string, spec *specs.Process) (shim *os.Process, err error) {
+	err = vm.AddProcess(&api.Process{
+		Container: container,
+		Id:        process,
+		Terminal:  spec.Terminal,
+		Args:      spec.Args,
+		Envs:      spec.Env,
+		Workdir:   spec.Cwd}, nil)
+
+	if err != nil {
+		glog.V(1).Infof("add process to container failed: %v\n", err)
+	}
+
+	shim, err = createShim(options, container, process)
+	if err != nil {
+		// cli refactor todo: remove the process
+		return nil, err
+	}
+
+	// cli refactor todo (for the purpose of 'runv ps` command) save <container, process, shim-pid, spec> to persist file.
+
+	return shim, nil
+}
+
+func execHook(hook specs.Hook, state *specs.State) error {
+	b, err := json.Marshal(state)
+	if err != nil {
+		return err
+	}
+	cmd := exec.Cmd{
+		Path:  hook.Path,
+		Args:  hook.Args,
+		Env:   hook.Env,
+		Stdin: bytes.NewReader(b),
+	}
+	return cmd.Run()
+}
+
+func execPrestartHooks(rt *specs.Spec, state *specs.State) error {
+	if rt.Hooks == nil {
+		return nil
+	}
+	for _, hook := range rt.Hooks.Prestart {
+		err := execHook(hook, state)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func execPoststartHooks(rt *specs.Spec, state *specs.State) error {
+	if rt.Hooks == nil {
+		return nil
+	}
+	for _, hook := range rt.Hooks.Poststart {
+		err := execHook(hook, state)
+		if err != nil {
+			glog.V(1).Infof("exec Poststart hook %s failed %s", hook.Path, err.Error())
+		}
+	}
+
+	return nil
+}
+
+func execPoststopHooks(rt *specs.Spec, state *specs.State) error {
+	if rt.Hooks == nil {
+		return nil
+	}
+	for _, hook := range rt.Hooks.Poststop {
+		err := execHook(hook, state)
+		if err != nil {
+			glog.V(1).Infof("exec Poststop hook %s failed %s", hook.Path, err.Error())
+		}
+	}
+
+	return nil
+}