Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
10 kB · 357 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358//go:build linux
package main
import ( "context" "database/sql" "errors" "fmt" "log/slog" "net" "os" "time"
"github.com/mdlayher/vsock" "github.com/urfave/cli/v3" agentv1 "tangled.org/core/spindle/agentproto/gen" "tangled.org/core/spindle/db" "tangled.org/core/spindle/engines/microvm")
func SpindleMicroVMRunCommand() *cli.Command { return &cli.Command{ Name: "spindle-microvm-run", Usage: "launch the Spindle base microVM and run one command over vsock", Flags: []cli.Flag{ &cli.StringFlag{ Name: "image-spec", Sources: cli.EnvVars("SPINDLE_MICROVM_IMAGE_SPEC"), Usage: "path to microVM image spec JSON", }, &cli.StringFlag{ Name: "mkfs-ext4", Usage: "override mkfs.ext4 binary", }, &cli.StringFlag{ Name: "work-dir", Usage: "directory for per-run disks and sockets", }, &cli.UintFlag{ Name: "cid", Usage: "guest vsock CID; defaults to a random high CID", }, &cli.UintFlag{ Name: "port", Value: 10240, Usage: "host vsock port to listen on", }, &cli.UintFlag{ Name: "memory-mib", Usage: "override the guest memory size in MiB (defaults to the image spec)", }, &cli.BoolFlag{ Name: "disable-kvm", Usage: "run without -enable-kvm even if /dev/kvm is available", }, &cli.BoolFlag{ Name: "dev", Usage: "enable dev mode (allows host network access, disables SSL verification)", }, &cli.DurationFlag{ Name: "qmp-timeout", Value: 10 * time.Second, Usage: "how long to wait for qmp to become ready", }, &cli.DurationFlag{ Name: "accept-timeout", Value: 15 * time.Second, Usage: "how long to wait for the guest agent after qemu starts", }, &cli.DurationFlag{ Name: "exec-timeout", Value: 30 * time.Second, Usage: "timeout for the guest command", }, &cli.DurationFlag{ Name: "cache-drain-timeout", Value: 5 * time.Minute, Usage: "how long to wait for queued cache uploads after the guest command exits", }, &cli.DurationFlag{ Name: "shutdown-timeout", Value: 10 * time.Second, Usage: "how long to wait for qemu to exit after guest powerdown", }, &cli.StringFlag{ Name: "cwd", Usage: "guest working directory", }, &cli.StringSliceFlag{ Name: "cache-read-url", Sources: cli.EnvVars("SPINDLE_NIX_CACHE_READ_URLS"), Usage: "Nix binary cache URL to pass to the guest; repeatable", }, &cli.StringSliceFlag{ Name: "cache-trusted-public-key", Sources: cli.EnvVars("SPINDLE_NIX_CACHE_TRUSTED_PUBLIC_KEYS"), Usage: "Nix binary cache public key to trust in the guest; repeatable", }, &cli.StringFlag{ Name: "cache-upload-url", Sources: cli.EnvVars("SPINDLE_NIX_CACHE_UPLOAD_URL"), Usage: "optional cache upload URL for guest-built store paths", }, &cli.StringFlag{ Name: "activate-config", Usage: "JSON user config to activate before exec (e.g. '{\"services\":{\"openssh\":{\"enable\":true}}}')", }, &cli.StringFlag{ Name: "db", Usage: "path to sqlite database for config cache", }, }, Action: runMicroVMRunDev, }}
func runMicroVMRunDev(ctx context.Context, cmd *cli.Command) error { imageSpecPath := cmd.String("image-spec") if imageSpecPath == "" { return fmt.Errorf("--image-spec or SPINDLE_MICROVM_IMAGE_SPEC is required") }
imageSpec, err := microvm.LoadImageSpec(imageSpecPath) if err != nil { return err }
port := uint32(cmd.Uint("port")) // tell the guest which host vsock port to dial back on. shuttle reads the // cmdline params this is so we can run multiple of this process // concurrently, because otherwise it listens on a specific vsock port, and // we cant bind to the same port twice... imageSpec.BootArgs = fmt.Sprintf("%s shuttle.vsock_port=%d", imageSpec.BootArgs, port) if mib := cmd.Uint("memory-mib"); mib > 0 { imageSpec.MemoryMiB = int(mib) } ln, err := vsock.Listen(port, nil) if err != nil { return fmt.Errorf("listen on vsock port %d: %w", port, err) } defer ln.Close()
vm, err := microvm.StartVM(ctx, microvm.VMConfig{ Image: imageSpec, BootTimeout: cmd.Duration("qmp-timeout"), CID: uint32(cmd.Uint("cid")), EnableKVM: !cmd.Bool("disable-kvm"), MkfsExt4: cmd.String("mkfs-ext4"), WorkDir: cmd.String("work-dir"), Dev: cmd.Bool("dev"), }, slog.Default()) if err != nil { return err } defer vm.Close()
logs := vm.Logs() fmt.Fprintf(os.Stderr, "microvm started: cid=%d work-dir=%s serial-log=%s qemu-log=%s\n", vm.CID(), vm.WorkDir(), logs.Serial, logs.Extra["qemu"], )
logger := slog.Default()
if cmd.Duration("accept-timeout") > 0 { if err := ln.SetDeadline(time.Now().Add(cmd.Duration("accept-timeout"))); err != nil { return fmt.Errorf("set accept deadline: %w", err) } }
argv := cmd.Args().Slice() if len(argv) == 0 { argv = []string{"/run/current-system/sw/bin/echo", "hello-from-spindle"} } jobID := "spindle-microvm-run" execID := "dev-1"
fmt.Fprintf(os.Stderr, "listening for agent on %s\n", ln.Addr()) conn, err := acceptExpectedVsockConn(ln, vm.CID(), logger) if err != nil { return fmt.Errorf("accept agent connection: %w", err) } defer conn.Close()
upstreams, err := microvm.BuildCacheUpstreams(cmd.StringSlice("cache-read-url"), nil) if err != nil { return fmt.Errorf("build cache upstreams: %w", err) }
var readCache *microvm.ReadCacheProxy if len(cmd.StringSlice("cache-read-url")) > 0 { var err error readCache, err = microvm.StartReadCacheProxy(ctx, vm.CID(), upstreams, logger) if err != nil { return fmt.Errorf("start read cache proxy: %w", err) } defer readCache.Close() }
var uploadCache *microvm.UploadCacheProxy if cmd.String("cache-upload-url") != "" { var err error uploadCache, err = microvm.StartUploadCacheProxy(ctx, vm.CID(), cmd.String("cache-upload-url"), upstreams, logger) if err != nil { return fmt.Errorf("start upload cache proxy: %w", err) } defer uploadCache.Close() } dnsProxy, err := microvm.StartDNSProxy(ctx, vm.CID(), logger) if err != nil { return fmt.Errorf("start dns proxy: %w", err) } defer dnsProxy.Close()
session := microvm.NewAgentSession(conn, logger)
initCtx, cancelInit := context.WithTimeout(ctx, 30*time.Second) defer cancelInit() if err := session.Init(initCtx, &agentv1.Init{ JobId: jobID, CacheTrustedPublicKeys: cmd.StringSlice("cache-trusted-public-key"), CacheReadProxyPort: readCache.Port(), CacheUploadProxyPort: uploadCache.Port(), DnsProxyPort: dnsProxy.Port(), }); err != nil { return fmt.Errorf("init agent: %w", err) }
execCtx := ctx if cmd.Duration("exec-timeout") > 0 { var cancel context.CancelFunc execCtx, cancel = context.WithTimeout(ctx, cmd.Duration("exec-timeout")) defer cancel() }
if cmd.String("activate-config") != "" { actCtx := execCtx baseHash, err := microvm.BaseConfigHash(imageSpec) if err != nil { return fmt.Errorf("calculate base config hash: %w", err) }
var d *db.DB var configKey string var cachedToplevel string if cmd.String("db") != "" { d, err = db.Make(ctx, cmd.String("db")) if err != nil { return fmt.Errorf("failed to open database: %w", err) } defer d.Close()
configKey, err = microvm.BuildConfigKey(imageSpec, cmd.String("activate-config")) if err != nil { return fmt.Errorf("calculate config key: %w", err) }
record, err := d.GetNixOSToplevelCacheRecord(configKey) if err != nil { if !errors.Is(err, sql.ErrNoRows) { return fmt.Errorf("lookup config cache: %w", err) } } else { cachedToplevel = record.Toplevel fmt.Printf("realizing cached NixOS config %s\n", cachedToplevel) } }
result, err := session.ActivateConfig(actCtx, "dev-activate", &agentv1.ActivateConfig{ ConfigKey: configKey, BaseConfigHash: baseHash, UserConfig: cmd.String("activate-config"), Toplevel: cachedToplevel, }) if err != nil { return fmt.Errorf("activate config: %w", err) } fmt.Fprintf(os.Stderr, "activated config toplevel: %s\n", result.Toplevel)
if d != nil && cachedToplevel == "" && result.Toplevel != "" && configKey != "" { err = d.SaveNixOSToplevelCacheRecord(configKey, result.Toplevel) if err != nil { return fmt.Errorf("save config cache: %w", err) } } }
exitCode, err := session.Exec(execCtx, microvm.AgentExec{ ID: execID, ExecStart: &agentv1.ExecStart{ Argv: argv, Cwd: cmd.String("cwd"), }, Stdout: os.Stdout, Stderr: os.Stderr, }) if err != nil { return err }
if uploadCache != nil { drainCtx := ctx if cmd.Duration("cache-drain-timeout") > 0 { var cancel context.CancelFunc drainCtx, cancel = context.WithTimeout(ctx, cmd.Duration("cache-drain-timeout")) defer cancel() } uploaded, err := session.Drain(drainCtx) if err != nil { return err } fmt.Printf("cache uploaded: %d\n", uploaded) }
// mirror the engine shutdown order: ask the agent to power off first, // then fall back to qemu powerdown / kill shutdownCtx, cancel := context.WithTimeout(context.Background(), cmd.Duration("shutdown-timeout")) defer cancel() poweredOff := false if err := session.Poweroff(shutdownCtx); err != nil { fmt.Fprintf(os.Stderr, "agent poweroff: %s\n", err) } else if err := vm.WaitContext(shutdownCtx); err == nil { poweredOff = true } if !poweredOff { if err := vm.Shutdown(shutdownCtx); err != nil { fmt.Fprintf(os.Stderr, "microvm shutdown fallback: %s\n", err) } }
if exitCode != 0 { return fmt.Errorf("guest command exited with code %d", exitCode) } return nil}
func acceptExpectedVsockConn(ln *vsock.Listener, allowedCID uint32, logger *slog.Logger) (net.Conn, error) { for { conn, err := ln.Accept() if err != nil { return nil, err } if allowedCID == 0 { return conn, nil } addr, ok := conn.RemoteAddr().(*vsock.Addr) if ok && addr.ContextID == allowedCID { return conn, nil } remote := conn.RemoteAddr() _ = conn.Close() logger.Warn("dropped agent connection from unexpected cid", "remote", remote, "expected", allowedCID) }}