added ValidateSaveRestoreSupport in Save/Restore methods

Code-Hex · Oct 22, 2023 · f02f617 · f02f617
1 parent 877edb3
commit f02f617
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/virtualization.go b/virtualization.go
@@ -92,6 +92,8 @@ type VirtualMachine struct {
 	machineState  *machineState
 
 	finalizeOnce sync.Once
+
+	config *VirtualMachineConfiguration
 }
 
 type machineState struct {
@@ -134,6 +136,7 @@ func NewVirtualMachine(config *VirtualMachineConfiguration) (*VirtualMachine, er
 		),
 		dispatchQueue: dispatchQueue,
 		machineState:  machineState,
+		config:        config,
 	}
 
 	objc.SetFinalizer(v, func(self *VirtualMachine) {

diff --git a/virtualization_arm64.go b/virtualization_arm64.go
@@ -598,6 +598,9 @@ func (v *VirtualMachine) SaveMachineStateToPath(saveFilePath string) error {
 	if err := macOSAvailable(14); err != nil {
 		return err
 	}
+	if _, err := v.config.ValidateSaveRestoreSupport(); err != nil {
+		return err
+	}
 	cs := charWithGoString(saveFilePath)
 	defer cs.Free()
 	h, errCh := makeHandler()
@@ -629,6 +632,9 @@ func (v *VirtualMachine) RestoreMachineStateFromURL(saveFilePath string) error {
 	if err := macOSAvailable(14); err != nil {
 		return err
 	}
+	if _, err := v.config.ValidateSaveRestoreSupport(); err != nil {
+		return err
+	}
 	cs := charWithGoString(saveFilePath)
 	defer cs.Free()
 	h, errCh := makeHandler()