Correctly handle error from check_extension_int()

Kvm::check_extension_int() may return negative value as error code, so Kvm::get_max_vcpus() and Kvm::get_max_vcpu_id() should handle the error cases. Signed-off-by: Liu Jiang <[email protected]>
rust-vmm · Jun 5, 2021 · 2fd74d8 · 2fd74d8
1 parent 577d4eb
commit 2fd74d8
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/src/ioctls/system.rs b/src/ioctls/system.rs
@@ -221,9 +221,12 @@ impl Kvm {
     /// ```
     ///
     pub fn get_max_vcpus(&self) -> usize {
-        match self.check_extension_int(Cap::MaxVcpus) {
-            0 => self.get_nr_vcpus(),
-            x => x as usize,
+        let v = self.check_extension_int(Cap::MaxVcpus);
+
+        if v <= 0 {
+            self.get_nr_vcpus()
+        } else {
+            v as usize
         }
     }
 
@@ -242,9 +245,12 @@ impl Kvm {
     /// ```
     ///
     pub fn get_max_vcpu_id(&self) -> usize {
-        match self.check_extension_int(Cap::MaxVcpuId) {
-            0 => self.get_max_vcpus(),
-            x => x as usize,
+        let v = self.check_extension_int(Cap::MaxVcpuId);
+
+        if v <= 0 {
+            self.get_max_vcpus()
+        } else {
+            v as usize
         }
     }