This commit is contained in:
Yacine Elhamer
2023-07-21 14:41:07 +01:00
parent c4ba5afe6b
commit 830bad54bd
3 changed files with 9 additions and 0 deletions

View File

@@ -93,6 +93,9 @@ class DnfileFeatureExtractor(StaticFeatureExtractor):
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

View File

@@ -91,6 +91,9 @@ class DnfileFeatureExtractor(StaticFeatureExtractor):
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

View File

@@ -175,6 +175,9 @@ class DotnetFileFeatureExtractor(StaticFeatureExtractor):
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