gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2026-02-04 11:07:53 -08:00
Code Issues Packages Projects Releases Wiki Activity

add sample_hashes attribute to the base extractors

Browse Source
This commit is contained in:
Yacine Elhamer
2023-07-20 21:51:37 +01:00
parent b4cf50fb6e
commit ab092cb536
1 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
capa/features/extractors/base_extractor.py
View File

@@ -116,6 +116,8 @@ class StaticFeatureExtractor:
# this base class doesn't know what to do with that info, though.
#
super().__init__()
# all extractors must be able to provide a samples hashes
self.sample_hashes: SampleHashes
@abc.abstractmethod
def get_base_address(self) -> Union[AbsoluteVirtualAddress, capa.features.address._NoAddress]:
@@ -131,10 +133,8 @@ class StaticFeatureExtractor:
def get_sample_hashes(self) -> SampleHashes:
"""
fetch the hashes for the sample contained within the extractor.
the order of the hashes is: md5, sha1, sha256
"""
raise NotImplementedError()
return self.sample_hashes
@abc.abstractmethod
def extract_global_features(self) -> Iterator[Tuple[Feature, Address]]:
@@ -341,13 +341,23 @@ class DynamicFeatureExtractor:
This class is not instantiated directly; it is the base class for other implementations.
"""
def get_sample_hashes(self) -> Tuple[str, str, str]:
__metaclass__ = abc.ABCMeta
def __init__(self):
#
# note: a subclass should define ctor parameters for its own use.
# for example, the Vivisect feature extract might require the vw and/or path.
# this base class doesn't know what to do with that info, though.
#
super().__init__()
# all extractors must be able to provide a samples hashes
self.sample_hashes: SampleHashes
def get_sample_hashes(self) -> SampleHashes:
"""
fetch the hashes for the sample contained within the extractor.
the order of the hashes is: md5, sha1, sha256
"""
raise NotImplementedError()
return self.sample_hashes
@abc.abstractmethod
def extract_global_features(self) -> Iterator[Tuple[Feature, Address]]: