repro-run/runner/runner.go

146 lines
3.5 KiB
Go

package runner
import (
"embed"
"fmt"
"io"
"log"
"os"
"os/exec"
"path"
"strings"
)
//go:embed alpine/keys
var alpineKeys embed.FS
func Run(config Config, rootfs string, cache string, src string, w io.Writer, dontDeleteTemp bool, env map[string]string) (err error) {
if err = os.RemoveAll(rootfs); err != nil {
return
}
if err = os.MkdirAll(path.Join(rootfs, "etc/apk/keys"), 0700); err != nil {
return
}
if err = os.WriteFile(path.Join(rootfs, "etc/apk/repositories"), []byte(
`https://dl-cdn.alpinelinux.org/alpine/v3.17/main
https://dl-cdn.alpinelinux.org/alpine/v3.17/community`), 0600); err != nil {
return
}
if err = os.WriteFile(path.Join(rootfs, "etc/resolv.conf"), []byte(`nameserver 8.8.8.8`), 0644); err != nil {
return
}
keys, err := alpineKeys.ReadDir("alpine/keys")
if err != nil {
return
}
for _, entry := range keys {
o, err := os.Create(path.Join(rootfs, "etc/apk/keys/", entry.Name()))
if err != nil {
return err
}
s, err := alpineKeys.Open(path.Join("alpine/keys", entry.Name()))
if err != nil {
return err
}
if _, err = io.Copy(o, s); err != nil {
return err
}
}
if err = os.MkdirAll(path.Join(cache, "_var_cache_apk"), 0700); err != nil {
return
}
if err = os.Symlink("/var/cache/apk", path.Join(rootfs, "etc/apk/cache")); err != nil {
return
}
cmd := exec.Command("apk", "add",
"--root", rootfs,
"--initdb",
"--cache-dir", path.Join("..", cache, "_var_cache_apk"),
"apk-tools")
if err = cmd.Run(); err != nil {
return fmt.Errorf("apk add apk-tools: %w", err)
}
cached := append(config.Cache, "/var/cache/apk")
var cachedParams []string
for _, c := range cached {
cacheDir := path.Join(cache, strings.ReplaceAll(c, "/", "_"))
if err = os.MkdirAll(cacheDir, 0700); err != nil {
return
}
cachedParams = append(cachedParams, "--bind", cacheDir, c)
}
tmp, err := os.MkdirTemp("", "repro-run-")
if err != nil {
return
}
defer removeTemp(tmp, dontDeleteTemp)
log.Println("/work = " + tmp)
params := append(
[]string{"--bind", rootfs, "/",
"--ro-bind", src, "/src",
"--dev-bind", "/dev", "/dev",
"--dev-bind", "/proc", "/proc",
"--bind", tmp, "/work",
"--unshare-user", "--uid", "1000", "--gid", "1000",
"--setenv", "HOME", "/home/repro",
"--chdir", "/work",
"--setenv", "PATH", "/usr/bin:/usr/sbin:/bin:/sbin",
},
cachedParams...)
for k, v := range env {
params = append(params, "--setenv", k, v)
}
cmd = exec.Command("bwrap", append(params,
"apk", "add", "--quiet", "busybox", "busybox-suid", "libc-utils", "alpine-baselayout", "alpine-conf", "alpine-release")...)
cmd.Stdout = os.Stderr
cmd.Stderr = os.Stderr
if err = cmd.Run(); err != nil {
return fmt.Errorf("bwrap apk add: %w", err)
}
cmd = exec.Command("bwrap", append(params,
"--uid", "0",
"adduser", "-u", "1000", "-D", "repro")...)
cmd.Stdout = os.Stderr
cmd.Stderr = os.Stderr
if err = cmd.Run(); err != nil {
return fmt.Errorf("bwrap adduser repro: %w", err)
}
cmd = exec.Command("bwrap", append(params, strings.Split(config.Command, " ")...)...)
// cmd.Stdin = os.Stdin
if w == nil {
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
} else {
cmd.Stdout = w
cmd.Stderr = w
}
if err = cmd.Run(); err != nil {
return fmt.Errorf("running Command: %w", err)
}
return
}
type Config struct {
Command string
Cache []string
}
func removeTemp(name string, dontRemoveTemp bool) {
if !dontRemoveTemp {
err := os.RemoveAll(name)
if err != nil {
log.Printf("No pude borrar %s: %v", name, err)
}
}
}