Add read and write operations handling

IO manager is responsible for handling IO operation when VMExit.
It works out the specific device according to the address range and
hand over to DeviceIo trait.

Signed-off-by: Jing Liu <[email protected]>

Author: Jing Liu

src/device_manager.rs

@@ -12,8 +12,9 @@
 //! devices IO ranges, and finally set resources to virtual device.
 
 use crate::resources::Resource;
-use crate::DeviceIo;
+use crate::{DeviceIo, IoAddress, IoSize};
 
+use std::cmp::{Ord, Ordering, PartialEq, PartialOrd};
 use std::collections::btree_map::BTreeMap;
 use std::result;
 use std::sync::Arc;
@@ -23,18 +24,62 @@ use std::sync::Arc;
 pub enum Error {
     /// The inserting device overlaps with a current device.
     DeviceOverlap,
+    /// The device doesn't exist.
+    NoDevice,
 }
 
 /// Simplify the `Result` type.
 pub type Result<T> = result::Result<T, Error>;
 
+// Structure describing an IO range.
+#[derive(Debug, Copy, Clone)]
+struct IoRange {
+    base: IoAddress,
+    size: IoSize,
+}
+
+impl IoRange {
+    fn new_pio_range(base: u16, size: u16) -> Self {
+        IoRange {
+            base: IoAddress::Pio(base),
+            size: IoSize::Pio(size),
+        }
+    }
+    fn new_mmio_range(base: u64, size: u64) -> Self {
+        IoRange {
+            base: IoAddress::Mmio(base),
+            size: IoSize::Mmio(size),
+        }
+    }
+}
+
+impl Eq for IoRange {}
+
+impl PartialEq for IoRange {
+    fn eq(&self, other: &IoRange) -> bool {
+        self.base == other.base
+    }
+}
+
+impl Ord for IoRange {
+    fn cmp(&self, other: &IoRange) -> Ordering {
+        self.base.cmp(&other.base)
+    }
+}
+
+impl PartialOrd for IoRange {
+    fn partial_cmp(&self, other: &IoRange) -> Option<Ordering> {
+        self.base.partial_cmp(&other.base)
+    }
+}
+
 /// System IO manager serving for all devices management and VM exit handling.
 #[derive(Default)]
 pub struct IoManager {
     /// Range mapping for VM exit pio operations.
-    pio_bus: BTreeMap<(u16, u16), Arc<dyn DeviceIo>>,
+    pio_bus: BTreeMap<IoRange, Arc<dyn DeviceIo>>,
     /// Range mapping for VM exit mmio operations.
-    mmio_bus: BTreeMap<(u64, u64), Arc<dyn DeviceIo>>,
+    mmio_bus: BTreeMap<IoRange, Arc<dyn DeviceIo>>,
 }
 
 impl IoManager {
@@ -60,7 +105,11 @@ impl IoManager {
         for (idx, res) in resources.iter().enumerate() {
             match *res {
                 Resource::PioAddressRange { base, size } => {
-                    if self.pio_bus.insert((base, size), device.clone()).is_some() {
+                    if self
+                        .pio_bus
+                        .insert(IoRange::new_pio_range(base, size), device.clone())
+                        .is_some()
+                    {
                         // Unregister registered resources.
                         self.unregister_device_io(&resources[0..idx])
                             .expect("failed to unregister devices");
@@ -69,7 +118,11 @@ impl IoManager {
                     }
                 }
                 Resource::MmioAddressRange { base, size } => {
-                    if self.mmio_bus.insert((base, size), device.clone()).is_some() {
+                    if self
+                        .mmio_bus
+                        .insert(IoRange::new_mmio_range(base, size), device.clone())
+                        .is_some()
+                    {
                         // Unregister registered resources.
                         self.unregister_device_io(&resources[0..idx])
                             .expect("failed to unregister devices");
@@ -95,14 +148,89 @@ impl IoManager {
         for res in resources.iter() {
             match *res {
                 Resource::PioAddressRange { base, size } => {
-                    self.pio_bus.remove(&(base, size));
+                    self.pio_bus.remove(&IoRange::new_pio_range(base, size));
                 }
                 Resource::MmioAddressRange { base, size } => {
-                    self.mmio_bus.remove(&(base, size));
+                    self.mmio_bus.remove(&IoRange::new_mmio_range(base, size));
                 }
                 _ => continue,
             }
         }
         Ok(())
     }
+
+    fn get_entry(&self, addr: IoAddress) -> Option<(&IoRange, &Arc<dyn DeviceIo>)> {
+        match addr {
+            IoAddress::Pio(a) => self
+                .pio_bus
+                .range(..=&IoRange::new_pio_range(a, 0))
+                .nth_back(0),
+            IoAddress::Mmio(a) => self
+                .mmio_bus
+                .range(..=&IoRange::new_mmio_range(a, 0))
+                .nth_back(0),
+        }
+    }
+
+    // Return the Device mapped `addr` and the base address.
+    fn get_device(&self, addr: IoAddress) -> Option<(&Arc<dyn DeviceIo>, IoAddress)> {
+        if let Some((range, dev)) = self.get_entry(addr) {
+            if (addr.raw_value() - range.base.raw_value()) < range.size.raw_value() {
+                return Some((dev, range.base));
+            }
+        }
+        None
+    }
+
+    /// A helper function handling PIO read command during VM exit.
+    /// The virtual device itself provides mutable ability and thead-safe protection.
+    ///
+    /// Return error if failed to get the device.
+    pub fn pio_read(&self, addr: u16, data: &mut [u8]) -> Result<()> {
+        if let Some((device, base)) = self.get_device(IoAddress::Pio(addr)) {
+            device.read(base, IoAddress::Pio(addr - (base.raw_value() as u16)), data);
+            Ok(())
+        } else {
+            Err(Error::NoDevice)
+        }
+    }
+
+    /// A helper function handling PIO write command during VM exit.
+    /// The virtual device itself provides mutable ability and thead-safe protection.
+    ///
+    /// Return error if failed to get the device.
+    pub fn pio_write(&self, addr: u16, data: &[u8]) -> Result<()> {
+        if let Some((device, base)) = self.get_device(IoAddress::Pio(addr)) {
+            device.write(base, IoAddress::Pio(addr - (base.raw_value() as u16)), data);
+            Ok(())
+        } else {
+            Err(Error::NoDevice)
+        }
+    }
+
+    /// A helper function handling MMIO read command during VM exit.
+    /// The virtual device itself provides mutable ability and thead-safe protection.
+    ///
+    /// Return error if failed to get the device.
+    pub fn mmio_read(&self, addr: u64, data: &mut [u8]) -> Result<()> {
+        if let Some((device, base)) = self.get_device(IoAddress::Mmio(addr)) {
+            device.read(base, IoAddress::Mmio(addr - base.raw_value()), data);
+            Ok(())
+        } else {
+            Err(Error::NoDevice)
+        }
+    }
+
+    /// A helper function handling MMIO write command during VM exit.
+    /// The virtual device itself provides mutable ability and thead-safe protection.
+    ///
+    /// Return error if failed to get the device.
+    pub fn mmio_write(&self, addr: u64, data: &[u8]) -> Result<()> {
+        if let Some((device, base)) = self.get_device(IoAddress::Mmio(addr)) {
+            device.write(base, IoAddress::Mmio(addr - base.raw_value()), data);
+            Ok(())
+        } else {
+            Err(Error::NoDevice)
+        }
+    }
 }

src/lib.rs

@@ -3,9 +3,31 @@
 
 //! rust-vmm device model.
 
+use std::cmp::{Ord, Ordering, PartialOrd};
+
 pub mod device_manager;
 pub mod resources;
 
+// IO Size.
+#[derive(Debug, Copy, Clone)]
+enum IoSize {
+    // Port I/O size.
+    Pio(u16),
+
+    // Memory mapped I/O size.
+    Mmio(u64),
+}
+
+impl IoSize {
+    // Get the raw value as u64 to make operation simple.
+    fn raw_value(&self) -> u64 {
+        match *self {
+            IoSize::Pio(p) => u64::from(p),
+            IoSize::Mmio(m) => m,
+        }
+    }
+}
+
 /// IO Addresses.
 #[derive(Debug, Copy, Clone)]
 pub enum IoAddress {
@@ -16,6 +38,36 @@ pub enum IoAddress {
     Mmio(u64),
 }
 
+impl IoAddress {
+    // Get the raw value of IO Address to make operation simple.
+    fn raw_value(&self) -> u64 {
+        match *self {
+            IoAddress::Pio(p) => u64::from(p),
+            IoAddress::Mmio(m) => m,
+        }
+    }
+}
+
+impl Eq for IoAddress {}
+
+impl PartialEq for IoAddress {
+    fn eq(&self, other: &IoAddress) -> bool {
+        self.raw_value() == other.raw_value()
+    }
+}
+
+impl Ord for IoAddress {
+    fn cmp(&self, other: &IoAddress) -> Ordering {
+        self.raw_value().cmp(&other.raw_value())
+    }
+}
+
+impl PartialOrd for IoAddress {
+    fn partial_cmp(&self, other: &IoAddress) -> Option<Ordering> {
+        self.raw_value().partial_cmp(&other.raw_value())
+    }
+}
+
 /// Device IO trait.
 /// A device supporting memory based I/O should implement this trait, then
 /// register itself against the different IO type ranges it handles.