cap.rs - source

From 69f1b8f726fb30b3628c2449bee7bee939801210 Mon Sep 17 00:00:00 2001 From: johannst Date: Thu, 1 Jul 2021 21:12:07 +0000 Subject: deploy: 9f025e829ab6ed468cfb51d72ac105624afc3851 --- src/kvm_rs/cap.rs.html | 75 +++++++++++++++++++++++++++++++++++++++++ src/kvm_rs/fmt.rs.html | 38 ++++++++++----------- src/kvm_rs/kvm.rs.html | 82 +++++++++++++++++++++++++++++++++----------- src/kvm_rs/kvm_sys.rs.html | 46 ++++++++++++------------- src/kvm_rs/lib.rs.html | 84 ++++++++++++++++++++++++---------------------- src/kvm_rs/vcpu.rs.html | 60 ++++++++++++++++----------------- src/kvm_rs/vm.rs.html | 32 +++++++++--------- src/kvm_rs/x86_64.rs.html | 4 +-- 8 files changed, 270 insertions(+), 151 deletions(-) create mode 100644 src/kvm_rs/cap.rs.html (limited to 'src/kvm_rs') diff --git a/src/kvm_rs/cap.rs.html b/src/kvm_rs/cap.rs.html new file mode 100644 index 0000000..72b076d --- /dev/null +++ b/src/kvm_rs/cap.rs.html @@ -0,0 +1,75 @@ +cap.rs - source + +

 1
+ 2
+ 3
+ 4
+ 5
+ 6
+ 7
+ 8
+ 9
+10
+11
+12
+13
+14
+15
+16
+17
+18
+19
+20
+21
+22
+23
+24
+25
+26
+27
+28
+29
+30
+31
+32
+33
+34
+

+//! Definitions of KVM capabilities.
+
+use crate::kvm_sys;
+use std::convert::Into;
+
+/// Definition of capabilities that return a bool value indicating whether the capability is
+/// supported or not.
+#[repr(u64)]
+pub enum CapBool {
+    /// Check if capabilities can be queried on VM fds (`KVM_CAP_CHECK_EXTENSION_VM`).
+    CheckExtensionVm = kvm_sys::KVM_CAP_CHECK_EXTENSION_VM,
+}
+
+impl Into<u64> for CapBool {
+    fn into(self) -> u64 {
+        self as u64
+    }
+}
+
+/// Definition of capabilities that return an integer value indicating the amount of the queried
+/// capability.
+#[repr(u64)]
+pub enum CapInt {
+    /// Get the recommended max VPCUs (`KVM_CAP_NR_VCPUS`).
+    NrVcpus = kvm_sys::KVM_CAP_NR_VCPUS,
+    /// Get the possible max VPCUs (`KVM_CAP_MAX_VCPUS`).
+    MaxVcpus = kvm_sys::KVM_CAP_MAX_VCPUS,
+}
+
+impl Into<u64> for CapInt {
+    fn into(self) -> u64 {
+        self as u64
+    }
+}
+

+ \ No newline at end of file diff --git a/src/kvm_rs/fmt.rs.html b/src/kvm_rs/fmt.rs.html index ec871ef..006e8c1 100644 --- a/src/kvm_rs/fmt.rs.html +++ b/src/kvm_rs/fmt.rs.html @@ -1,6 +1,6 @@ fmt.rs - source -

  1
   2
   3
@@ -131,14 +131,14 @@
 128
 129

-use std::fmt;
+use std::fmt;
 
-use crate::kvm_sys::{kvm_dtable, kvm_regs, kvm_segment, kvm_sregs};
-use crate::x86_64::*;
+use crate::kvm_sys::{kvm_dtable, kvm_regs, kvm_segment, kvm_sregs};
+use crate::x86_64::*;
 
-impl fmt::Display for kvm_regs {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        write!(
+impl fmt::Display for kvm_regs {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(
             f,
             "rax: {:#018x} rbx: {:#018x} rcx: {:#018x} rdx: {:#018x}\n\
              rsi: {:#018x} rdi: {:#018x}\n\
@@ -175,12 +175,12 @@
     }
 }
 
-impl fmt::Display for kvm_segment {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+impl fmt::Display for kvm_segment {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
         if self.present == 0 {
-            write!(f, "{:#04x} P(0)", self.selector)
+            write!(f, "{:#04x} P(0)", self.selector)
         } else {
-            write!(
+            write!(
                 f,
                 "{:#06x} T({}) RPL({}) BASE({:#010x}) LIMIT({:#07x}) P(1) S({}) DPL({}) DB({}) L({}) TYPE({})",
                 self.selector,
@@ -213,22 +213,22 @@
                     13 => "C:X-AC",
                     14 => "C:XR-C",
                     15 => "C:XRAC",
-                    _ => unreachable!(),
+                    _ => unreachable!(),
                 }
             )
         }
     }
 }
 
-impl fmt::Display for kvm_dtable {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        write!(f, "BASE({:#018x}) LIMIT({:#07x})", self.base, self.limit)
+impl fmt::Display for kvm_dtable {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "BASE({:#018x}) LIMIT({:#07x})", self.base, self.limit)
     }
 }
 
-impl fmt::Display for kvm_sregs {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        write!(
+impl fmt::Display for kvm_sregs {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(
             f,
             "cs  : {}\n\
              ds  : {}\n\
@@ -261,5 +261,5 @@
     }
 }

\ No newline at end of file diff --git a/src/kvm_rs/kvm.rs.html b/src/kvm_rs/kvm.rs.html index ccd6c37..43a3f1e 100644 --- a/src/kvm_rs/kvm.rs.html +++ b/src/kvm_rs/kvm.rs.html @@ -1,6 +1,6 @@ kvm.rs - source -

 1
  2
  3
@@ -53,15 +53,37 @@
 50
 51
 52
+53
+54
+55
+56
+57
+58
+59
+60
+61
+62
+63
+64
+65
+66
+67
+68
+69
+70
+71
+72
+73

 //! KVM system ioctls.
 
-use std::fs;
-use std::io;
-use std::os::unix::io::FromRawFd;
+use std::fs;
+use std::io;
+use std::os::unix::io::FromRawFd;
 
-use crate::{libcret, ioctl, kvm_sys};
-use crate::vm::Vm;
+use crate::cap::{CapBool, CapInt};
+use crate::vm::Vm;
+use crate::{ioctl, kvm_sys, libcret};
 
 /// Wrapper for `/dev/kvm` ioctls.
 ///
@@ -70,42 +92,62 @@
 ///
 /// [kvm]: https://www.kernel.org/doc/html/latest/virt/kvm/api.html#general-description
 pub struct Kvm {
-    kvm: fs::File,
+    kvm: fs::File,
 }
 
 impl Kvm {
     /// Open the `/dev/kvm` device.
-    pub fn new() -> io::Result<Kvm> {
+    pub fn new() -> io::Result<Kvm> {
         let kvm = libcret(unsafe {
-            libc::open("/dev/kvm\0".as_ptr().cast(), libc::O_RDWR | libc::O_CLOEXEC)
+            libc::open("/dev/kvm\0".as_ptr().cast(), libc::O_RDWR | libc::O_CLOEXEC)
         })
-        .map(|fd| unsafe { fs::File::from_raw_fd(fd) })?;
+        .map(|fd| unsafe { fs::File::from_raw_fd(fd) })?;
 
-        assert_eq!(
-            kvm_sys::KVM_API_VERSION,
-            ioctl(&kvm, kvm_sys::KVM_GET_API_VERSION, 0)?
+        assert_eq!(
+            kvm_sys::KVM_API_VERSION,
+            ioctl(&kvm, kvm_sys::KVM_GET_API_VERSION, 0)?
         );
 
         Ok(Kvm { kvm })
     }
 
-    fn get_vpcu_mmap_size(&self) -> io::Result<usize> {
-        ioctl(&self.kvm, kvm_sys::KVM_GET_VCPU_MMAP_SIZE, 0).map(|size| size as usize)
+    fn get_vpcu_mmap_size(&self) -> io::Result<usize> {
+        ioctl(&self.kvm, kvm_sys::KVM_GET_VCPU_MMAP_SIZE, 0).map(|size| size as usize)
     }
 
     /// Create a new virtual machine with the [`KVM_CREATE_VM`][kvm-create-vm] ioctl.
     /// Returns a wrapper [`vm::Vm`][crate::vm::Vm] representing the VM.
     ///
     /// [kvm-create-vm]: https://www.kernel.org/doc/html/latest/virt/kvm/api.html#kvm-create-vm
-    pub fn create_vm(&self) -> io::Result<Vm> {
-        let vm = ioctl(&self.kvm, kvm_sys::KVM_CREATE_VM, 0 /* machine id */)
-            .map(|fd| unsafe { fs::File::from_raw_fd(fd) })?;
+    pub fn create_vm(&self) -> io::Result<Vm> {
+        let vm = ioctl(&self.kvm, kvm_sys::KVM_CREATE_VM, 0 /* machine id */)
+            .map(|fd| unsafe { fs::File::from_raw_fd(fd) })?;
 
         let vcpu_mmap_size = self.get_vpcu_mmap_size()?;
 
-        Ok(Vm::new(vm, vcpu_mmap_size))
+        Ok(Vm::new(vm, vcpu_mmap_size))
+    }
+
+    /// Check availability of an extension with the [`KVM_CHECK_EXTENSION`][kvm-check-extension]
+    /// ioctl.
+    ///
+    /// [kvm-check-extension]: https://www.kernel.org/doc/html/latest/virt/kvm/api.html#kvm-check-extension
+    pub fn check_extenstion(&self, cap: CapBool) -> bool {
+        let ret = ioctl(&self.kvm, kvm_sys::KVM_CHECK_EXTENSION, cap.into());
+
+        matches!(ret, Ok(ret) if ret > 0)
+    }
+
+    /// Check availability of an extension with the [`KVM_CHECK_EXTENSION`][kvm-check-extension]
+    /// ioctl.
+    ///
+    /// [kvm-check-extension]: https://www.kernel.org/doc/html/latest/virt/kvm/api.html#kvm-check-extension
+    pub fn check_extenstion_int(&self, cap: CapInt) -> i32 {
+        let ret = ioctl(&self.kvm, kvm_sys::KVM_CHECK_EXTENSION, cap.into());
+
+        ret.unwrap_or(0)
     }
 }

\ No newline at end of file diff --git a/src/kvm_rs/kvm_sys.rs.html b/src/kvm_rs/kvm_sys.rs.html index 44136b6..9fbbf41 100644 --- a/src/kvm_rs/kvm_sys.rs.html +++ b/src/kvm_rs/kvm_sys.rs.html @@ -1,6 +1,6 @@ kvm_sys.rs - source -

  1
   2
   3
@@ -208,7 +208,7 @@
 #![allow(dead_code)]
 
 // Generated by `build.rs`.
-include!(concat!(env!("OUT_DIR"), "/kvm_constants.rs"));
+include!(concat!(env!("OUT_DIR"), "/kvm_constants.rs"));
 
 #[repr(C)]
 #[derive(Default, Debug)]
@@ -345,57 +345,57 @@
 #[cfg(test)]
 mod tests {
     use super::*;
-    use std::mem;
+    use std::mem;
 
     #[test]
     fn check_kvm_regs() {
-        assert_eq!(mem::size_of::<kvm_regs>(), TEST_KVM_REGS_SIZE);
-        assert_eq!(mem::align_of::<kvm_regs>(), TEST_KVM_REGS_ALIGN);
+        assert_eq!(mem::size_of::<kvm_regs>(), TEST_KVM_REGS_SIZE);
+        assert_eq!(mem::align_of::<kvm_regs>(), TEST_KVM_REGS_ALIGN);
     }
 
     #[test]
     fn check_kvm_segment() {
-        assert_eq!(mem::size_of::<kvm_segment>(), TEST_KVM_SEGMENT_SIZE);
-        assert_eq!(mem::align_of::<kvm_segment>(), TEST_KVM_SEGMENT_ALIGN);
+        assert_eq!(mem::size_of::<kvm_segment>(), TEST_KVM_SEGMENT_SIZE);
+        assert_eq!(mem::align_of::<kvm_segment>(), TEST_KVM_SEGMENT_ALIGN);
     }
 
     #[test]
     fn check_kvm_dtable() {
-        assert_eq!(mem::size_of::<kvm_dtable>(), TEST_KVM_DTABLE_SIZE);
-        assert_eq!(mem::align_of::<kvm_dtable>(), TEST_KVM_DTABLE_ALIGN);
+        assert_eq!(mem::size_of::<kvm_dtable>(), TEST_KVM_DTABLE_SIZE);
+        assert_eq!(mem::align_of::<kvm_dtable>(), TEST_KVM_DTABLE_ALIGN);
     }
 
     #[test]
     fn check_kvm_sregs() {
-        assert_eq!(mem::size_of::<kvm_sregs>(), TEST_KVM_SREGS_SIZE);
-        assert_eq!(mem::align_of::<kvm_sregs>(), TEST_KVM_SREGS_ALIGN);
-        assert_eq!(
-            mem::size_of_val(&kvm_sregs::default().interrupt_bitmap),
+        assert_eq!(mem::size_of::<kvm_sregs>(), TEST_KVM_SREGS_SIZE);
+        assert_eq!(mem::align_of::<kvm_sregs>(), TEST_KVM_SREGS_ALIGN);
+        assert_eq!(
+            mem::size_of_val(&kvm_sregs::default().interrupt_bitmap),
             TEST_KVM_SREGS_INTERRTUP_BITMAP_SIZE
         );
     }
 
     #[test]
     fn check_kvm_userspace_memory_region() {
-        assert_eq!(
-            mem::size_of::<kvm_userspace_memory_region>(),
+        assert_eq!(
+            mem::size_of::<kvm_userspace_memory_region>(),
             TEST_KVM_USERSPACE_MEMORY_REGION_SIZE
         );
-        assert_eq!(
-            mem::align_of::<kvm_userspace_memory_region>(),
+        assert_eq!(
+            mem::align_of::<kvm_userspace_memory_region>(),
             TEST_KVM_USERSPACE_MEMORY_REGION_ALIGN
         );
     }
 
     #[test]
     fn check_kvm_run() {
-        assert_eq!(mem::size_of::<kvm_run>(), TEST_KVM_RUN_SIZE);
-        assert_eq!(mem::align_of::<kvm_run>(), TEST_KVM_RUN_ALIGN);
-        assert_eq!(mem::size_of::<kvm_run_io>(), TEST_KVM_RUN_IO_SIZE);
-        assert_eq!(mem::size_of::<kvm_run_mmio>(), TEST_KVM_RUN_MMIO_SIZE);
-        assert_eq!(mem::size_of::<kvm_run_union_s>(), TEST_KVM_RUN_UNION_S_SIZE);
+        assert_eq!(mem::size_of::<kvm_run>(), TEST_KVM_RUN_SIZE);
+        assert_eq!(mem::align_of::<kvm_run>(), TEST_KVM_RUN_ALIGN);
+        assert_eq!(mem::size_of::<kvm_run_io>(), TEST_KVM_RUN_IO_SIZE);
+        assert_eq!(mem::size_of::<kvm_run_mmio>(), TEST_KVM_RUN_MMIO_SIZE);
+        assert_eq!(mem::size_of::<kvm_run_union_s>(), TEST_KVM_RUN_UNION_S_SIZE);
     }
 }

\ No newline at end of file diff --git a/src/kvm_rs/lib.rs.html b/src/kvm_rs/lib.rs.html index 8d74487..2e03eed 100644 --- a/src/kvm_rs/lib.rs.html +++ b/src/kvm_rs/lib.rs.html @@ -1,6 +1,6 @@ lib.rs - source -