limayaml: add a .saveOnStop field to control whether to use suspend

default `false`

Signed-off-by: Norio Nomura <norio.nomura@gmail.com>

Author: norio-nomura

pkg/limayaml/defaults.go

@@ -855,6 +855,32 @@ func FillDefault(y, d, o *LimaYAML, filePath string, warn bool) {
 		y.NestedVirtualization = ptr.Of(false)
 	}
 
+	if y.SaveOnStop == nil {
+		y.SaveOnStop = d.SaveOnStop
+	}
+	if o.SaveOnStop != nil {
+		y.SaveOnStop = o.SaveOnStop
+	}
+	if y.SaveOnStop == nil {
+		y.SaveOnStop = ptr.Of(false)
+	}
+	if *y.SaveOnStop {
+		y.SaveOnStop = ptr.Of(false)
+		if *y.VMType != VZ {
+			logrus.Warn("saveOnStop is only supported for VM type vz")
+		} else if runtime.GOARCH != "arm64" {
+			logrus.Warn("saveOnStop is only supported for arm64 VM type vz")
+		} else if runtime.GOOS != "darwin" {
+			logrus.Warn("saveOnStop is only supported on macOS")
+		} else if macOSProductVersion, err := osutil.ProductVersion(); err != nil {
+			logrus.WithError(err).Warn("Failed to get macOS product version")
+		} else if macOSProductVersion.LessThan(*semver.New("14.0.0")) {
+			logrus.Warn("saveOnStop is not supported on macOS prior to 14.0")
+		} else {
+			y.SaveOnStop = ptr.Of(true)
+		}
+	}
+
 	if y.Plain == nil {
 		y.Plain = d.Plain
 	}
@@ -878,6 +904,7 @@ func fixUpForPlainMode(y *LimaYAML) {
 	y.Containerd.User = ptr.Of(false)
 	y.Rosetta.BinFmt = ptr.Of(false)
 	y.Rosetta.Enabled = ptr.Of(false)
+	y.SaveOnStop = ptr.Of(false)
 	y.TimeZone = ptr.Of("")
 }
 

pkg/limayaml/defaults_test.go

@@ -112,6 +112,7 @@ func TestFillDefault(t *testing.T) {
 			RemoveDefaults: ptr.Of(false),
 		},
 		NestedVirtualization: ptr.Of(false),
+		SaveOnStop:           ptr.Of(false),
 		Plain:                ptr.Of(false),
 		User: User{
 			Name:    ptr.Of(user.Username),
@@ -437,6 +438,7 @@ func TestFillDefault(t *testing.T) {
 			BinFmt:  ptr.Of(true),
 		},
 		NestedVirtualization: ptr.Of(true),
+		SaveOnStop:           ptr.Of(true),
 		User: User{
 			Name:    ptr.Of("xxx"),
 			Comment: ptr.Of("Foo Bar"),
@@ -474,11 +476,13 @@ func TestFillDefault(t *testing.T) {
 			Enabled: ptr.Of(true),
 			BinFmt:  ptr.Of(true),
 		}
+		expect.SaveOnStop = ptr.Of(true)
 	} else {
 		expect.Rosetta = Rosetta{
 			Enabled: ptr.Of(false),
 			BinFmt:  ptr.Of(true),
 		}
+		expect.SaveOnStop = ptr.Of(false)
 	}
 	expect.Plain = ptr.Of(false)
 
@@ -660,6 +664,7 @@ func TestFillDefault(t *testing.T) {
 			BinFmt:  ptr.Of(false),
 		},
 		NestedVirtualization: ptr.Of(false),
+		SaveOnStop:           ptr.Of(false),
 		User: User{
 			Name:    ptr.Of("foo"),
 			Comment: ptr.Of("foo bar baz"),
@@ -723,6 +728,7 @@ func TestFillDefault(t *testing.T) {
 	expect.Plain = ptr.Of(false)
 
 	expect.NestedVirtualization = ptr.Of(false)
+	expect.SaveOnStop = ptr.Of(false)
 
 	FillDefault(&y, &d, &o, filePath, false)
 	assert.DeepEqual(t, &y, &expect, opts...)

pkg/limayaml/limayaml.go

@@ -48,6 +48,7 @@ type LimaYAML struct {
 	TimeZone             *string        `yaml:"timezone,omitempty" json:"timezone,omitempty" jsonschema:"nullable"`
 	NestedVirtualization *bool          `yaml:"nestedVirtualization,omitempty" json:"nestedVirtualization,omitempty" jsonschema:"nullable"`
 	User                 User           `yaml:"user,omitempty" json:"user,omitempty"`
+	SaveOnStop           *bool          `yaml:"saveOnStop,omitempty" json:"saveOnStop,omitempty" jsonschema:"nullable"`
 }
 
 type (

pkg/vz/vz_driver_darwin.go

@@ -53,6 +53,7 @@ var knownYamlProperties = []string{
 	"PropagateProxyEnv",
 	"Provision",
 	"Rosetta",
+	"SaveOnStop",
 	"SSH",
 	"TimeZone",
 	"UpgradePackages",
@@ -67,11 +68,22 @@ type LimaVzDriver struct {
 	*driver.BaseDriver
 
 	machine *virtualMachineWrapper
+
+	// Runtime configuration
+	config LimaVzDriverRuntimeConfig
+}
+
+type LimaVzDriverRuntimeConfig struct {
+	// SaveOnStop is a flag to save the VM state on stop
+	SaveOnStop bool `json:"saveOnStop"`
 }
 
 func New(driver *driver.BaseDriver) *LimaVzDriver {
 	return &LimaVzDriver{
 		BaseDriver: driver,
+		config: LimaVzDriverRuntimeConfig{
+			SaveOnStop: *driver.Instance.Config.SaveOnStop,
+		},
 	}
 }
 
@@ -193,11 +205,13 @@ func (l *LimaVzDriver) RunGUI() error {
 }
 
 func (l *LimaVzDriver) Stop(_ context.Context) error {
-	machineStatePath := filepath.Join(l.Instance.Dir, filenames.VzMachineState)
-	if err := saveVM(l.machine.VirtualMachine, machineStatePath); err != nil {
-		logrus.WithError(err).Warn("Failed to save VZ. Falling back to shutdown")
-	} else {
-		return nil
+	if l.config.SaveOnStop {
+		machineStatePath := filepath.Join(l.Instance.Dir, filenames.VzMachineState)
+		if err := saveVM(l.machine.VirtualMachine, machineStatePath); err != nil {
+			logrus.WithError(err).Warn("Failed to save VZ. Falling back to shutdown")
+		} else {
+			return nil
+		}
 	}
 
 	logrus.Info("Shutting down VZ")

templates/default.yaml

@@ -525,6 +525,11 @@ hostResolver:
 # 🟢 Builtin default: /usr/local
 guestInstallPrefix: null
 
+# When "saveOnStop" is enabled, Lima saves the VM instead of shutting it down.
+# The VM can be restored with `limactl start` if the saved state is available.
+# 🟢 Builtin default: false
+saveOnStop: null
+
 # When the "plain" mode is enabled:
 # - the YAML properties for mounts, port forwarding, containerd, etc. will be ignored
 # - guest agent will not be running