storage: reset storage_bmap_dict after writing

Signed-off-by: Ding Meng <[email protected]>
siemens · Nov 22, 2023 · e4b0f93 · e4b0f93
1 parent 2125eaf
commit e4b0f93
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mtda/client.py b/mtda/client.py
@@ -270,6 +270,7 @@ def storage_write_image(self, path, callback=None):
         finally:
             # Storage may be closed now
             self.storage_close()
+            self._impl.storage_bmap_dict(None, self._session)
 
     def parseBmap(self, bmap, bmap_path):
         try:

diff --git a/mtda/storage/helpers/image.py b/mtda/storage/helpers/image.py
@@ -298,7 +298,8 @@ def setBmap(self, bmapDict):
         self.crtBlockRange = 0
         self.writtenBytes = 0
         self.overlap = 0
-        self.rangeChkSum = self._get_hasher_by_name()
+        if bmapDict is not None:
+            self.rangeChkSum = self._get_hasher_by_name()
 
     def supports_hotplug(self):
         return False