96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
formolv1alpha1 "github.com/desmo999r/formol/api/v1alpha1"
|
|
"github.com/go-logr/logr"
|
|
"github.com/go-logr/zapr"
|
|
"go.uber.org/zap"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strconv"
|
|
)
|
|
|
|
var logger logr.Logger
|
|
|
|
func init() {
|
|
zapLog, _ := zap.NewDevelopment()
|
|
logger = zapr.NewLogger(zapLog)
|
|
}
|
|
|
|
func Run(runCmd string, args []string) error {
|
|
log := logger.WithValues("Run", runCmd, "Args", args)
|
|
cmd := exec.Command(runCmd, args...)
|
|
output, err := cmd.CombinedOutput()
|
|
log.V(1).Info("result", "output", string(output))
|
|
if err != nil {
|
|
log.Error(err, "something went wrong")
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func RunChroot(lookForTag bool, runCmd string, args ...string) error {
|
|
log := logger.WithValues("RunChroot", runCmd, "Args", args)
|
|
root := regexp.MustCompile(`/proc/[0-9]+/root`)
|
|
env := regexp.MustCompile(`/proc/[0-9]+/environ`)
|
|
pid := strconv.Itoa(os.Getpid())
|
|
skip := false
|
|
if err := filepath.Walk("/proc", func(path string, info os.FileInfo, err error) error {
|
|
if skip {
|
|
return filepath.SkipDir
|
|
}
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
if info.IsDir() && (info.Name() == "1" || info.Name() == pid) {
|
|
return filepath.SkipDir
|
|
}
|
|
if lookForTag && env.MatchString(path) {
|
|
log.V(0).Info("Looking for tag", "file", path)
|
|
content, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return filepath.SkipDir
|
|
}
|
|
|
|
var matched bool
|
|
for _, envVar := range bytes.Split(content, []byte{'\000'}) {
|
|
matched, err = regexp.Match(formolv1alpha1.TARGETCONTAINER_TAG, envVar)
|
|
if err != nil {
|
|
log.Error(err, "cannot regexp")
|
|
return err
|
|
}
|
|
if matched {
|
|
log.V(0).Info("Found the target tag", "file", path)
|
|
break
|
|
}
|
|
}
|
|
if matched == false {
|
|
return filepath.SkipDir
|
|
}
|
|
}
|
|
if root.MatchString(path) {
|
|
if _, err := filepath.EvalSymlinks(path); err != nil {
|
|
return filepath.SkipDir
|
|
}
|
|
log.V(0).Info("running chroot in", "path", path)
|
|
cmd := exec.Command("chroot", append([]string{path, runCmd}, args...)...)
|
|
output, err := cmd.CombinedOutput()
|
|
log.V(0).Info("result", "output", string(output))
|
|
if err != nil {
|
|
log.Error(err, "something went wrong")
|
|
return err
|
|
}
|
|
skip = true
|
|
return filepath.SkipDir
|
|
}
|
|
return nil
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|