fix bugs

mandiant · Jul 21, 2023 · 830bad5 · 830bad5
1 parent c4ba5af
commit 830bad5
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/capa/features/extractors/dnfile/extractor.py b/capa/features/extractors/dnfile/extractor.py
@@ -93,6 +93,9 @@ def __init__(self, path: Path):
     def get_base_address(self):
         return NO_ADDRESS
 
+    def get_sample_hashes(self) -> SampleHashes:
+        return self.sample_hashes
+
     def extract_global_features(self):
         yield from self.global_features
 

diff --git a/capa/features/extractors/dnfile_.py b/capa/features/extractors/dnfile_.py
@@ -91,6 +91,9 @@ def __init__(self, path: Path):
     def get_base_address(self) -> AbsoluteVirtualAddress:
         return AbsoluteVirtualAddress(0x0)
 
+    def get_sample_hashes(self) -> SampleHashes:
+        return self.sample_hashes
+
     def get_entry_point(self) -> int:
         # self.pe.net.Flags.CLT_NATIVE_ENTRYPOINT
         #  True: native EP: Token

diff --git a/capa/features/extractors/dotnetfile.py b/capa/features/extractors/dotnetfile.py
@@ -175,6 +175,9 @@ def __init__(self, path: Path):
     def get_base_address(self):
         return NO_ADDRESS
 
+    def get_sample_hashes(self) -> SampleHashes:
+        return self.sample_hashes
+
     def get_entry_point(self) -> int:
         # self.pe.net.Flags.CLT_NATIVE_ENTRYPOINT
         #  True: native EP: Token