vmm: keep virtio activation alive in migration

Live migration can deadlock if the guest triggers a virtio device
activation while the migration worker owns the VM.

The failure shows up when starting live migrations during boot and
firmware startup, where the guest can reset and reinitialize virtio
devices while precopy is running. In the failing case, the source log
shows a pending virtio activation that never completes:

    8.115833s _virtio-pci-net_0: Needs activation; returning barrier
    8.115854s vmm/src/vm.rs:464 -- Waiting for barrier
    24.875452s Entering downtime phase
    24.875481s stopping vcpu throttling thread
    ...
    vCPU thread did not respond in 10ms to signal - retrying
    vCPU thread did not respond in 20ms to signal - retrying
    ...
    thread 'throttle-vcpu' (1029) panicked
    ...
    Pause(Error signalling vCPUs: Timeout when waiting for signal
        to be acknowledged)

The vCPU blocks on the activation barrier and never reaches the normal
pause checkpoint. Later, migration enters downtime and stops the vCPU
throttle thread. In the failing case, that thread is still inside a
CpuManager::pause() call, which waits for every vCPU to acknowledge
the signal. The blocked vCPU never does, so the pause times out.

Fix this by storing the DeviceManager inside VmOwnership::Migration.
This keeps just enough state on the VMM thread to drain pending virtio
activations while the migration worker owns the Vm. The barrier logic
stays unchanged. The VMM now releases the same activation barrier during
migration that it already released before migration started.

This keeps the guest from getting stuck in the activation wait and
lets the later pause succeed.

Co-authored-by: Leander Kohler <leander.kohler@cyberus-technology.de>
On-behalf-of: SAP leander.kohler@sap.com
Signed-off-by: Leander Kohler <leander.kohler@cyberus-technology.de>

On-behalf-of: SAP philipp.schuster@sap.com
Signed-off-by: Philipp Schuster <philipp.schuster@cyberus-technology.de>

(cherry picked from commit 4d727c4)
Signed-off-by: Julian Schindel <julian.schindel@cyberus-technology.de>

Author: phip1611

vmm/src/lib.rs

@@ -20,7 +20,7 @@ use std::panic::AssertUnwindSafe;
 use std::path::PathBuf;
 use std::sync::atomic::{AtomicBool, Ordering};
 use std::sync::mpsc::{Receiver, RecvError, SendError, Sender};
-use std::sync::{Arc, Mutex};
+use std::sync::{Arc, Mutex, Weak};
 use std::time::Duration;
 #[cfg(not(target_arch = "riscv64"))]
 use std::time::Instant;
@@ -70,6 +70,7 @@ use crate::config::{MemoryRestoreMode, RestoreConfig, add_to_config};
 use crate::coredump::GuestDebuggable;
 #[cfg(feature = "kvm")]
 use crate::cpu::IS_IN_SHUTDOWN;
+use crate::device_manager::DeviceManager;
 use crate::landlock::Landlock;
 use crate::memory_manager::MemoryManager;
 use crate::migration::transport::{
@@ -643,13 +644,18 @@ pub struct VmmThreadHandle {
 
 /// Models the current ownership and associated state of the VM from the
 /// perspective of the VMM.
-pub enum VmOwnership {
+enum VmOwnership {
     Owned(Vm),
     /// The VM is temporarily owned by an ongoing migration worker.
+    ///
+    /// We deliberately do not use shared access to the VM to prevent a whole
+    /// class of race conditions.
     Migration {
         migration_worker_handle: MigrationWorkerHandle,
         /// Snapshot returned while the VMM cannot inspect the worker-owned VM.
         vm_info_response: VmInfoResponse,
+        /// Access to VM state needed during migration.
+        device_manager: Weak<Mutex<DeviceManager>>,
     },
     None,
 }
@@ -2225,12 +2231,27 @@ impl Vmm {
                         }
                     }
                     EpollDispatch::ActivateVirtioDevices => {
-                        // TODO: Future follow-up must resolve virtio activation handling while migrating.
                         let count = self.activate_evt.read().map_err(Error::EventFdRead)?;
-                        if let VmOwnership::Owned(ref vm) = self.vm {
-                            info!("Trying to activate pending virtio devices: count = {count}");
-                            vm.activate_virtio_devices()
-                                .map_err(Error::ActivateVirtioDevices)?;
+                        info!("Trying to activate pending virtio devices: count = {count}");
+                        match &self.vm {
+                            VmOwnership::Owned(vm) => {
+                                vm.activate_virtio_devices()
+                                    .map_err(Error::ActivateVirtioDevices)?;
+                            }
+                            VmOwnership::Migration { device_manager, .. } => {
+                                // If the VM (and thus the device manager) were
+                                // dropped at this point, we'd have a serious
+                                // programming bug.
+                                let device_manager = device_manager
+                                    .upgrade()
+                                    .expect("DeviceManager should remain alive during a migration");
+                                let device_manager = device_manager.lock().unwrap();
+                                device_manager
+                                    .activate_virtio_devices()
+                                    .map_err(VmError::ActivateVirtioDevices)
+                                    .map_err(Error::ActivateVirtioDevices)?;
+                            }
+                            VmOwnership::None => {}
                         }
                     }
                     EpollDispatch::Api => {
@@ -3323,6 +3344,8 @@ impl RequestHandler for Vmm {
             .take_owned_or(VmError::VmNotRunning)
             .expect("should have VM ownership as we just checked it");
 
+        let device_manager = Arc::downgrade(vm.device_manager());
+
         match MigrationWorker::spawn(
             vm,
             check_migration_evt,
@@ -3336,6 +3359,7 @@ impl RequestHandler for Vmm {
                 self.vm = VmOwnership::Migration {
                     migration_worker_handle: handle,
                     vm_info_response: vm_info_snapshot,
+                    device_manager,
                 };
                 Ok(())
             }

vmm/src/vm.rs

@@ -3485,6 +3485,10 @@ impl Vm {
             .unwrap()
             .post_migration_announce();
     }
+
+    pub fn device_manager(&self) -> &Arc<Mutex<DeviceManager>> {
+        &self.device_manager
+    }
 }
 
 impl Pausable for Vm {