Merge pull request #149 from Code-Hex/add/validation-save-restore

added ValidateSaveRestoreSupport in Save/Restore methods
Code-Hex · Oct 22, 2023 · bd29a7e · bd29a7e
2 parents b9079bb + f02f617
commit bd29a7e
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()