Add VhostUserRng device implementation

Implements a vhost-user RNG device. The
VMM now switches between the standard RNG device
and vhost-user RNG depending on whether a socket
path is configured via krun_add_vhost_user_device()

This allows us to use the RNG device from the
rust-vmm vhost-device running in a separate
process for better isolation and flexibility.

Signed-off-by: Dorinda Bassey <dbassey@redhat.com>

Author: Dorinda Bassey

src/vmm/src/builder.rs

@@ -976,7 +976,28 @@ pub fn build_microvm(
     #[cfg(not(feature = "tee"))]
     attach_balloon_device(&mut vmm, event_manager, intc.clone())?;
     #[cfg(not(feature = "tee"))]
-    attach_rng_device(&mut vmm, event_manager, intc.clone())?;
+    {
+        #[cfg(feature = "vhost-user")]
+        {
+            const VIRTIO_ID_RNG: u32 = 4;
+
+            let has_vhost_user_rng = vm_resources
+                .vhost_user_devices
+                .iter()
+                .any(|dev| dev.device_type == VIRTIO_ID_RNG);
+
+            if has_vhost_user_rng {
+                for device_config in &vm_resources.vhost_user_devices {
+                    attach_vhost_user_device(&mut vmm, intc.clone(), device_config)?;
+                }
+            } else {
+                attach_rng_device(&mut vmm, event_manager, intc.clone())?;
+            }
+        }
+
+        #[cfg(not(feature = "vhost-user"))]
+        attach_rng_device(&mut vmm, event_manager, intc.clone())?;
+    }
     let mut console_id = 0;
     if !vm_resources.disable_implicit_console {
         attach_console_devices(
@@ -2388,6 +2409,36 @@ fn attach_rng_device(
     Ok(())
 }
 
+#[cfg(not(feature = "tee"))]
+#[cfg(feature = "vhost-user")]
+fn attach_vhost_user_device(
+    vmm: &mut Vmm,
+    intc: IrqChip,
+    device_config: &VhostUserDeviceConfig,
+) -> std::result::Result<(), StartMicrovmError> {
+    use self::StartMicrovmError::*;
+
+    let device_name = device_config
+        .name
+        .clone()
+        .unwrap_or_else(|| format!("vhost-user-{}", device_config.device_type));
+
+    let device = Arc::new(Mutex::new(
+        devices::virtio::VhostUserDevice::new(
+            &device_config.socket_path,
+            device_config.device_type,
+            device_name.clone(),
+            device_config.num_queues,
+            &device_config.queue_sizes,
+        )
+        .map_err(|e| RegisterRngDevice(device_manager::mmio::Error::VhostUserDevice(e)))?,
+    ));
+
+    attach_mmio_device(vmm, device_name, intc.clone(), device).map_err(RegisterRngDevice)?;
+
+    Ok(())
+}
+
 #[cfg(feature = "gpu")]
 #[allow(clippy::too_many_arguments)]
 fn attach_gpu_device(

src/vmm/src/device_manager/kvm/mmio.rs

@@ -41,6 +41,9 @@ pub enum Error {
     DeviceNotFound,
     /// Failed to update the mmio device.
     UpdateFailed,
+    /// Failed to create vhost-user device.
+    #[cfg(feature = "vhost-user")]
+    VhostUserDevice(io::Error),
 }
 
 impl fmt::Display for Error {
@@ -59,6 +62,8 @@ impl fmt::Display for Error {
             Error::RegisterIrqFd(ref e) => write!(f, "failed to register irqfd: {e}"),
             Error::DeviceNotFound => write!(f, "the device couldn't be found"),
             Error::UpdateFailed => write!(f, "failed to update the mmio device"),
+            #[cfg(feature = "vhost-user")]
+            Error::VhostUserDevice(ref e) => write!(f, "failed to create vhost-user device: {e}"),
         }
     }
 }