mirror of
https://github.com/cloud-hypervisor/cloud-hypervisor.git
synced 2026-08-05 02:19:16 +00:00
8c71a0d821
Keep the original error sources while adding operation context to the straightforward migration send and receive paths. This keeps up a chain of errors that can be printed nicely and in a helpful way. This commit does that for all MigratableError-occurences where this change is easily applicable. Co-authored-by: Sebastian Eydam <sebastian.eydam@cyberus-technology.de> On-behalf-of: SAP philipp.schuster@sap.com Signed-off-by: Philipp Schuster <philipp.schuster@cyberus-technology.de>
102 lines
3.2 KiB
Rust
102 lines
3.2 KiB
Rust
// Copyright © 2020 Intel Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use std::fs::File;
|
|
use std::io::Read;
|
|
use std::path::PathBuf;
|
|
use std::result;
|
|
|
|
use anyhow::{Context, anyhow};
|
|
use vm_migration::{MigratableError, Snapshot};
|
|
|
|
#[cfg(all(target_arch = "x86_64", feature = "guest_debug"))]
|
|
use crate::coredump::GuestDebuggableError;
|
|
use crate::vm::VmSnapshot;
|
|
use crate::vm_config::VmConfig;
|
|
|
|
pub(crate) mod transport;
|
|
pub(crate) mod worker;
|
|
|
|
pub const SNAPSHOT_STATE_FILE: &str = "state.json";
|
|
pub const SNAPSHOT_CONFIG_FILE: &str = "config.json";
|
|
|
|
pub fn url_to_path(url: &str) -> result::Result<PathBuf, MigratableError> {
|
|
let path: PathBuf = url
|
|
.strip_prefix("file://")
|
|
.ok_or_else(|| {
|
|
MigratableError::MigrateSend(anyhow!("Could not extract path from URL: {url}"))
|
|
})
|
|
.map(|s| s.into())?;
|
|
|
|
if !path.is_dir() {
|
|
return Err(MigratableError::MigrateSend(anyhow!(
|
|
"Destination is not a directory: {path:?}"
|
|
)));
|
|
}
|
|
|
|
Ok(path)
|
|
}
|
|
|
|
#[cfg(all(target_arch = "x86_64", feature = "guest_debug"))]
|
|
pub fn url_to_file(url: &str) -> result::Result<PathBuf, GuestDebuggableError> {
|
|
let file: PathBuf = url
|
|
.strip_prefix("file://")
|
|
.ok_or_else(|| {
|
|
GuestDebuggableError::Coredump(anyhow!("Could not extract file from URL: {url}"))
|
|
})
|
|
.map(|s| s.into())?;
|
|
|
|
Ok(file)
|
|
}
|
|
|
|
pub fn recv_vm_config(source_url: &str) -> result::Result<VmConfig, MigratableError> {
|
|
let mut vm_config_path = url_to_path(source_url)?;
|
|
|
|
vm_config_path.push(SNAPSHOT_CONFIG_FILE);
|
|
|
|
// Try opening the snapshot file
|
|
let mut vm_config_file = File::open(&vm_config_path)
|
|
.with_context(|| format!("Error opening VM config snapshot file {vm_config_path:?}"))
|
|
.map_err(MigratableError::MigrateReceive)?;
|
|
let mut bytes = Vec::new();
|
|
vm_config_file
|
|
.read_to_end(&mut bytes)
|
|
.with_context(|| format!("Error reading VM config snapshot file {vm_config_path:?}"))
|
|
.map_err(MigratableError::MigrateReceive)?;
|
|
|
|
serde_json::from_slice(&bytes)
|
|
.context("Error deserialising VM config snapshot")
|
|
.map_err(MigratableError::MigrateReceive)
|
|
}
|
|
|
|
pub fn recv_vm_state(source_url: &str) -> result::Result<Snapshot, MigratableError> {
|
|
let mut vm_state_path = url_to_path(source_url)?;
|
|
|
|
vm_state_path.push(SNAPSHOT_STATE_FILE);
|
|
|
|
// Try opening the snapshot file
|
|
let mut vm_state_file = File::open(&vm_state_path)
|
|
.with_context(|| format!("Error opening VM state snapshot file {vm_state_path:?}"))
|
|
.map_err(MigratableError::MigrateReceive)?;
|
|
let mut bytes = Vec::new();
|
|
vm_state_file
|
|
.read_to_end(&mut bytes)
|
|
.with_context(|| format!("Error reading VM state snapshot file {vm_state_path:?}"))
|
|
.map_err(MigratableError::MigrateReceive)?;
|
|
|
|
serde_json::from_slice(&bytes)
|
|
.context("Error deserialising VM state snapshot")
|
|
.map_err(MigratableError::MigrateReceive)
|
|
}
|
|
|
|
pub fn get_vm_snapshot(snapshot: &Snapshot) -> result::Result<VmSnapshot, MigratableError> {
|
|
if let Some(snapshot_data) = snapshot.snapshot_data.as_ref() {
|
|
return snapshot_data.to_state();
|
|
}
|
|
|
|
Err(MigratableError::Restore(anyhow!(
|
|
"Could not find VM config snapshot section"
|
|
)))
|
|
}
|