refactor(virtio/mmio): reuse ComCfg::get_max_queue_size

Signed-off-by: Martin Kröning <[email protected]>
hermit-os · May 27, 2024 · 50d63aa · 50d63aa
1 parent d07e22a
commit 50d63aa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/drivers/virtio/transport/mmio.rs b/src/drivers/virtio/transport/mmio.rs
@@ -122,7 +122,7 @@ impl ComCfg {
 	///
 	/// INFO: The queue size is automatically bounded by constant `src::config:VIRTIO_MAX_QUEUE_SIZE`.
 	pub fn select_vq(&mut self, index: u16) -> Option<VqCfgHandler<'_>> {
-		if self.com_cfg.get_max_queue_size(u32::from(index)) == 0 {
+		if self.get_max_queue_size(index) == 0 {
 			None
 		} else {
 			Some(VqCfgHandler {
@@ -132,8 +132,8 @@ impl ComCfg {
 		}
 	}
 
-	pub fn get_max_queue_size(&mut self, sel: u32) -> u32 {
-		self.com_cfg.get_max_queue_size(sel)
+	pub fn get_max_queue_size(&mut self, sel: u16) -> u16 {
+		self.com_cfg.get_max_queue_size(u32::from(sel)).try_into().unwrap()
 	}
 
 	pub fn get_queue_ready(&mut self, sel: u32) -> bool {