From af3a59aa330025967095d8bfd6419007fc99d5a1 Mon Sep 17 00:00:00 2001
From: Sebastien Boeuf <sebastien.boeuf@intel.com>
Date: Wed, 22 Sep 2021 17:34:59 +0200
Subject: [PATCH] virtio-devices: mem: Add constructor for BlocksState

Signed-off-by: Sebastien Boeuf <sebastien.boeuf@intel.com>
---
 virtio-devices/src/mem.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/virtio-devices/src/mem.rs b/virtio-devices/src/mem.rs
index d98175e2c..8f28bd1e4 100644
--- a/virtio-devices/src/mem.rs
+++ b/virtio-devices/src/mem.rs
@@ -392,6 +392,12 @@ pub struct BlocksState {
 }
 
 impl BlocksState {
+    fn new(region_size: u64) -> Self {
+        BlocksState {
+            bitmap: vec![false; (region_size / VIRTIO_MEM_DEFAULT_BLOCK_SIZE) as usize],
+        }
+    }
+
     fn is_range_state(&self, first_block_index: usize, nb_blocks: u16, plug: bool) -> bool {
         for state in self
             .bitmap
@@ -855,9 +861,7 @@ impl Mem {
             seccomp_action,
             hugepages,
             dma_mapping_handlers: Arc::new(Mutex::new(BTreeMap::new())),
-            blocks_state: Arc::new(Mutex::new(BlocksState {
-                bitmap: vec![false; (config.region_size / config.block_size) as usize],
-            })),
+            blocks_state: Arc::new(Mutex::new(BlocksState::new(config.region_size))),
             exit_evt,
         })
     }