feat(disk): map a source code disk into vm /dev/vdX

.golangci.yml

@@ -53,6 +53,7 @@ linters:
     - revive
     - funcorder
     - noinlineerr
+    - godoclint
   settings:
     errcheck:
      check-blank: false

cmd/init.go

@@ -65,7 +65,6 @@ var initCmd = cli.Command{
 			Value:    "v1.0",
 			Required: true,
 		},
-
 		&cli.StringFlag{
 			Name:  "vmm",
 			Usage: "vm provider, support: krunkit, vfkit",

pkg/machine/disk/disk.go

@@ -0,0 +1,43 @@
+//  SPDX-FileCopyrightText: 2024-2026 OOMOL, Inc. <https://www.oomol.com>
+//  SPDX-License-Identifier: MPL-2.0
+
+package disk
+
+import (
+	"bytes"
+	"context"
+	_ "embed"
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+
+	"github.com/sirupsen/logrus"
+)
+
+//go:embed source.ext4.tar
+var sourceCodeExt4Disk []byte
+
+func ExtractSourceCodeDisk(ctx context.Context, targetDirPath string, overwrite bool) error {
+	_, err := os.Stat(filepath.Join(targetDirPath, "source.ext4"))
+	if err == nil && !overwrite {
+		logrus.Infof("source code disk already exists, skip extraction")
+		return nil
+	}
+
+	if err := os.MkdirAll(targetDirPath, 0755); err != nil {
+		return fmt.Errorf("failed to create target directory: %w", err)
+	}
+	cmd := exec.CommandContext(ctx, "tar", "-xaS", "-C", targetDirPath, "-f", "-", "source.ext4")
+	cmd.Stdin = bytes.NewReader(sourceCodeExt4Disk)
+
+	var stderr bytes.Buffer
+	cmd.Stderr = &stderr
+
+	logrus.Infof("cmdline: %q", cmd.Args)
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("tar extraction failed: %w, stderr: %s", err, stderr.String())
+	}
+
+	return nil
+}

pkg/machine/machine_common.go

@@ -110,10 +110,11 @@ const applehvMACAddress = "5a:94:ef:e4:0c:ee"
 func SetupDevices(mc *vmconfig.MachineConfig) ([]vfConfig.VirtioDevice, error) {
 	var devices []vfConfig.VirtioDevice
 
-	disk, err := vfConfig.VirtioBlkNew(mc.Bootable.Path)
+	bootableDisk, err := vfConfig.VirtioBlkNew(mc.Bootable.Path)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create bootable disk device: %w", err)
 	}
+
 	rng, err := vfConfig.VirtioRngNew()
 	if err != nil {
 		return nil, fmt.Errorf("failed to create rng device: %w", err)
@@ -125,6 +126,11 @@ func SetupDevices(mc *vmconfig.MachineConfig) ([]vfConfig.VirtioDevice, error) {
 		return nil, fmt.Errorf("failed to create externalDisk device: %w", err)
 	}
 
+	sourceDisk, err := vfConfig.VirtioBlkNew(mc.GetSourceDiskPath())
+	if err != nil {
+		return nil, fmt.Errorf("failed to create source disk device: %w", err)
+	}
+
 	// using gvproxy as network backend
 	netDevice, err := vfConfig.VirtioNetNew(applehvMACAddress)
 	if err != nil {
@@ -133,8 +139,8 @@ func SetupDevices(mc *vmconfig.MachineConfig) ([]vfConfig.VirtioDevice, error) {
 
 	netDevice.SetUnixSocketPath(mc.GetNetworkStackEndpoint())
 
-	// externalDisk **must** be at the end of the device
-	devices = append(devices, disk, rng, netDevice, externalDisk)
+	// The externalDisk must be added to the devices queue after the bootableDisk
+	devices = append(devices, bootableDisk, rng, netDevice, externalDisk, sourceDisk)
 
 	VirtIOMounts, err := VirtIOFsToVFKitVirtIODevice(mc.Mounts)
 	if err != nil {

pkg/machine/shim/host.go

@@ -7,10 +7,12 @@ import (
 	"context"
 	"fmt"
 	"os/exec"
+	"path/filepath"
 
 	"bauklotze/pkg/decompress"
 	"bauklotze/pkg/machine"
 	"bauklotze/pkg/machine/define"
+	"bauklotze/pkg/machine/disk"
 	"bauklotze/pkg/machine/events"
 	"bauklotze/pkg/machine/krunkit"
 	"bauklotze/pkg/machine/ssh/service"
@@ -95,14 +97,20 @@ func Start(parentCtx context.Context, mc *vmconfig.MachineConfig, vmp vmconfig.V
 
 		cancel(context.Cause(parentCtx))
 	})
+
 	// 1. Start the network stack
 	if err := vmp.StartNetworkProvider(ctx, mc); err != nil {
 		return fmt.Errorf("failed to start network stack: %w", err)
 	}
 
-	// 2. Start the VM provider
+	// 2. extract the source code disk
+	if err := disk.ExtractSourceCodeDisk(ctx, filepath.Dir(mc.GetSourceDiskPath()), false); err != nil {
+		return fmt.Errorf("failed to extract source code disk: %w", err)
+	}
+
+	// 3. Start the VM provider
 	if err := vmp.StartVMProvider(ctx, mc); err != nil {
-		return fmt.Errorf("failed to start network stack: %w", err)
+		return fmt.Errorf("failed to start vm provider: %w", err)
 	}
 
 	// Optional services are placed in separation go routines, these services will not crash the VM even if they fail

pkg/machine/vmconfig/vmconfig.go

@@ -291,6 +291,10 @@ func (mc *MachineConfig) getBinDir() (string, error) {
 	return binDir, nil
 }
 
+func (mc *MachineConfig) GetSourceDiskPath() string {
+	return filepath.Join(mc.Dirs.DataDir, "source.ext4")
+}
+
 func (mc *MachineConfig) GetKrunkitBin() (string, error) {
 	dir, err := mc.getBinDir()
 	if err != nil {