Add support for packed virtqueue in message.rs and mod.rs

Author: prakash-shekhar

vhost/src/vhost_kern/mod.rs

@@ -26,6 +26,9 @@ use super::{
     VringConfigData, VHOST_MAX_MEMORY_REGIONS,
 };
 
+#[cfg(feature = "vhost-user")]
+use super::vhost_user::message::VhostUserVringAddrFlags;
+
 pub mod vhost_binding;
 use self::vhost_binding::*;
 
@@ -73,26 +76,63 @@ pub trait VhostKernBackend: AsRawFd {
         }
 
         let m = self.mem().memory();
-        let desc_table_size = 16 * u64::from(queue_size) as GuestUsize;
-        let avail_ring_size = 6 + 2 * u64::from(queue_size) as GuestUsize;
-        let used_ring_size = 6 + 8 * u64::from(queue_size) as GuestUsize;
-        if GuestAddress(config_data.desc_table_addr)
-            .checked_add(desc_table_size)
-            .is_none_or(|v| !m.address_in_range(v))
-        {
-            return false;
-        }
-        if GuestAddress(config_data.avail_ring_addr)
-            .checked_add(avail_ring_size)
-            .is_none_or(|v| !m.address_in_range(v))
-        {
-            return false;
-        }
-        if GuestAddress(config_data.used_ring_addr)
-            .checked_add(used_ring_size)
-            .is_none_or(|v| !m.address_in_range(v))
-        {
-            return false;
+
+        // Check if packed ring format is being used
+        #[cfg(feature = "vhost-user")]
+        let is_packed = config_data.flags & VhostUserVringAddrFlags::VHOST_VRING_F_PACKED.bits() != 0;
+        #[cfg(not(feature = "vhost-user"))]
+        let is_packed = false;
+
+        if is_packed {
+            // Packed ring: single descriptor ring layout
+            let desc_ring_size = 16 * u64::from(queue_size) as GuestUsize;
+            let driver_event_size = 4; // 4 bytes for driver event suppression
+            let device_event_size = 4; // 4 bytes for device event suppression
+
+            // Validate packed descriptor ring
+            if GuestAddress(config_data.desc_table_addr)
+                .checked_add(desc_ring_size)
+                .is_none_or(|v| !m.address_in_range(v))
+            {
+                return false;
+            }
+            // Validate driver event suppression area (available)
+            if GuestAddress(config_data.avail_ring_addr)
+                .checked_add(driver_event_size)
+                .is_none_or(|v| !m.address_in_range(v))
+            {
+                return false;
+            }
+            // Validate device event suppression area (used)
+            if GuestAddress(config_data.used_ring_addr)
+                .checked_add(device_event_size)
+                .is_none_or(|v| !m.address_in_range(v))
+            {
+                return false;
+            }
+        } else {
+            // Split ring validation (existing logic)
+            let desc_table_size = 16 * u64::from(queue_size) as GuestUsize;
+            let avail_ring_size = 6 + 2 * u64::from(queue_size) as GuestUsize;
+            let used_ring_size = 6 + 8 * u64::from(queue_size) as GuestUsize;
+            if GuestAddress(config_data.desc_table_addr)
+                .checked_add(desc_table_size)
+                .is_none_or(|v| !m.address_in_range(v))
+            {
+                return false;
+            }
+            if GuestAddress(config_data.avail_ring_addr)
+                .checked_add(avail_ring_size)
+                .is_none_or(|v| !m.address_in_range(v))
+            {
+                return false;
+            }
+            if GuestAddress(config_data.used_ring_addr)
+                .checked_add(used_ring_size)
+                .is_none_or(|v| !m.address_in_range(v))
+            {
+                return false;
+            }
         }
 
         config_data.is_log_addr_valid()

vhost/src/vhost_user/message.rs

@@ -386,6 +386,8 @@ bitflags! {
         const LOG_ALL = 0x400_0000;
         /// Feature flag for the protocol feature.
         const PROTOCOL_FEATURES = 0x4000_0000;
+        /// Feature flag for packed virtqueue support (VIRTIO_F_RING_PACKED, bit 34).
+        const RING_PACKED = 0x4_0000_0000;
     }
 }
 
@@ -713,6 +715,9 @@ bitflags! {
         /// Support log of vring operations.
         /// Modifications to "used" vring should be logged.
         const VHOST_VRING_F_LOG = 0x1;
+        /// Indicates packed virtqueue format.
+        /// When set, the vring uses packed layout instead of split layout.
+        const VHOST_VRING_F_PACKED = 0x2;
     }
 }
 
@@ -777,12 +782,28 @@ impl VhostUserMsgValidator for VhostUserVringAddr {
     fn is_valid(&self) -> bool {
         if (self.flags & !VhostUserVringAddrFlags::all().bits()) != 0 {
             return false;
-        } else if self.descriptor & 0xf != 0 {
-            return false;
-        } else if self.available & 0x1 != 0 {
-            return false;
-        } else if self.used & 0x3 != 0 {
-            return false;
+        }
+        
+        // Check if packed ring format
+        if self.flags & VhostUserVringAddrFlags::VHOST_VRING_F_PACKED.bits() != 0 {
+            // Packed ring validation:
+            // - descriptor ring must be 16-byte aligned
+            if self.descriptor & 0xf != 0 {
+                return false;
+            }
+            // - driver/device event suppression areas must be 4-byte aligned
+            if self.available & 0x3 != 0 || self.used & 0x3 != 0 {
+                return false;
+            }
+        } else {
+            // Split ring validation (current logic)
+            if self.descriptor & 0xf != 0 {
+                return false;
+            } else if self.available & 0x1 != 0 {
+                return false;
+            } else if self.used & 0x3 != 0 {
+                return false;
+            }
         }
         true
     }