process.go

// Copyright (c) 2018 Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0

package tests

import (
	"errors"
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"regexp"
)

const procPath = "/proc"

var errFound = errors.New("found")

// processRunning looks for a process in /proc that matches with the regexps
func processRunning(regexps []string) bool {
	err := filepath.Walk(procPath, func(path string, _ os.FileInfo, _ error) error {
		if path == "" {
			return filepath.SkipDir
		}

		info, err := os.Stat(path)
		if err != nil {
			return filepath.SkipDir
		}

		if !info.IsDir() {
			return filepath.SkipDir
		}

		content, err := ioutil.ReadFile(filepath.Join(path, "cmdline"))
		if err != nil {
			return filepath.SkipDir
		}

		for _, r := range regexps {
			matcher := regexp.MustCompile(r)
			if matcher.MatchString(string(content)) {
				return errFound
			}
		}

		return nil
	})

	return err == errFound
}

// HypervisorRunning returns true if the hypervisor is still running, otherwise false
func HypervisorRunning(containerID string) bool {
	var typeHypervisor = map[string]string{
		DefaultHypervisor:     (".*-name.*" + containerID + ".*-qmp.*unix:.*/" + containerID + "/.*"),
		FirecrackerHypervisor: (".*--api-sock.*" + containerID + ".*firecracker.sock.*"),
	}
	for h, r := range typeHypervisor {
		config, ok := KataConfig.Hypervisor[h]
		if ok {
			return processRunning([]string{config.Path + r})
		}
	}
	log.Fatal("Could not determine if hypervisor is running")
	return false
}

// ProxyRunning returns true if the proxy is still running, otherwise false
func ProxyRunning(containerID string) bool {
	if _, ok := KataConfig.Hypervisor[FirecrackerHypervisor]; ok {
		return false
	}
	proxyPath := KataConfig.Proxy[DefaultProxy].Path
	if proxyPath == "" {
		log.Fatal("Could not determine if proxy is running: proxy path is empty")
		return false
	}
	proxyRegexps := []string{proxyPath + ".*-listen-socket.*unix:.*/" + containerID + "/.*"}
	return processRunning(proxyRegexps)
}

// ShimRunning returns true if the shim is still running, otherwise false
func ShimRunning(containerID string) bool {
	shimPath := KataConfig.Shim[DefaultShim].Path
	if shimPath == "" {
		log.Fatal("Could not determine if shim is running: shim path is empty")
		return false
	}
	shimRegexps := []string{shimPath + ".*-agent.*unix:.*/" + containerID + "/.*-container.*" + containerID + ".*"}
	return processRunning(shimRegexps)
}