diff --git a/vm-migration/src/lib.rs b/vm-migration/src/lib.rs index 7d0e52d2d..b426b6e14 100644 --- a/vm-migration/src/lib.rs +++ b/vm-migration/src/lib.rs @@ -154,7 +154,7 @@ impl Snapshot { T: Serialize, { let mut snapshot_data = Snapshot::default(); - snapshot_data.add_data_section(SnapshotData::new_from_state(state)?); + snapshot_data.add_data(SnapshotData::new_from_state(state)?); Ok(snapshot_data) } @@ -165,7 +165,7 @@ impl Snapshot { T: Versionize + VersionMapped, { let mut snapshot_data = Snapshot::default(); - snapshot_data.add_data_section(SnapshotData::new_from_versioned_state(state)?); + snapshot_data.add_data(SnapshotData::new_from_versioned_state(state)?); Ok(snapshot_data) } @@ -176,7 +176,7 @@ impl Snapshot { } /// Add a SnapshotData to the component snapshot data. - pub fn add_data_section(&mut self, section: SnapshotData) { + pub fn add_data(&mut self, section: SnapshotData) { self.snapshot_data = Some(section); } diff --git a/vmm/src/cpu.rs b/vmm/src/cpu.rs index d07dbbe55..3db6b8334 100644 --- a/vmm/src/cpu.rs +++ b/vmm/src/cpu.rs @@ -400,7 +400,7 @@ impl Snapshottable for Vcpu { .map_err(|e| MigratableError::Pause(anyhow!("Could not get vCPU state {:?}", e)))?; let mut vcpu_snapshot = Snapshot::default(); - vcpu_snapshot.add_data_section(SnapshotData::new_from_state(&saved_state)?); + vcpu_snapshot.add_data(SnapshotData::new_from_state(&saved_state)?); self.saved_state = Some(saved_state); diff --git a/vmm/src/device_manager.rs b/vmm/src/device_manager.rs index 95543605b..fe74c11e1 100644 --- a/vmm/src/device_manager.rs +++ b/vmm/src/device_manager.rs @@ -4456,7 +4456,7 @@ impl Snapshottable for DeviceManager { } // Then we store the DeviceManager state. - snapshot.add_data_section(SnapshotData::new_from_state(&self.state())?); + snapshot.add_data(SnapshotData::new_from_state(&self.state())?); Ok(snapshot) } diff --git a/vmm/src/memory_manager.rs b/vmm/src/memory_manager.rs index c074024fc..96c369da9 100644 --- a/vmm/src/memory_manager.rs +++ b/vmm/src/memory_manager.rs @@ -2442,7 +2442,7 @@ impl Snapshottable for MemoryManager { // memory range content for the ranges requiring it. self.snapshot_memory_ranges = memory_ranges; - memory_manager_snapshot.add_data_section(SnapshotData::new_from_versioned_state( + memory_manager_snapshot.add_data(SnapshotData::new_from_versioned_state( &self.snapshot_data(), )?); diff --git a/vmm/src/vm.rs b/vmm/src/vm.rs index 3c751080e..2e90010e6 100644 --- a/vmm/src/vm.rs +++ b/vmm/src/vm.rs @@ -2532,7 +2532,7 @@ impl Snapshottable for Vm { (device_manager.id(), device_manager.snapshot()?) }; vm_snapshot.add_snapshot(id, snapshot); - vm_snapshot.add_data_section(SnapshotData(vm_snapshot_data)); + vm_snapshot.add_data(SnapshotData(vm_snapshot_data)); event!("vm", "snapshotted"); Ok(vm_snapshot)