Fix: .image tries to load and image=True expects to load

CERT-Polska · Oct 11, 2023 · bc4fcc0 · bc4fcc0
1 parent 543eff3
commit bc4fcc0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/malduck/procmem/binmem.py b/malduck/procmem/binmem.py
@@ -78,8 +78,8 @@ def load_binaries_from_memory(cls: Type[T], procmem: ProcessMemory) -> Iterator[
             binary_procmem_dmp = cls.from_memory(procmem, base=binary_va)
             if binary_procmem_dmp.is_valid():
                 yield binary_procmem_dmp
-            binary_procmem_img = cls.from_memory(procmem, base=binary_va, image=True)
-            if binary_procmem_img.is_valid():
+            binary_procmem_img = binary_procmem_dmp.image
+            if binary_procmem_img and binary_procmem_img.is_valid():
                 yield binary_procmem_img
 
     @abstractmethod

diff --git a/setup.cfg b/setup.cfg
@@ -23,3 +23,6 @@ ignore_missing_imports = True
 
 [mypy-ida_bytes.*]
 ignore_missing_imports = True
+
+[mypy-dnfile.*]
+ignore_missing_imports = True