From aafea6afbeb212c373b94732db2b709e5d4a65ff Mon Sep 17 00:00:00 2001 From: Kevin Waldock Date: Thu, 2 Nov 2023 13:20:58 +0100 Subject: [PATCH] Make has_metadata_pointer work with newer lief versions Checking if the type is equal or a subclass of V4 should be sufficient --- chpe_scanner/main.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/chpe_scanner/main.py b/chpe_scanner/main.py index b85d5bb..9f7f27d 100644 --- a/chpe_scanner/main.py +++ b/chpe_scanner/main.py @@ -31,12 +31,7 @@ def is_chpe_type(bin_type: "BinType") -> bool: @staticmethod def has_metadata_pointer(load_config: lief.PE.LoadConfiguration) -> bool: - return ( - type(load_config) == lief.PE.LoadConfigurationV4 - or type(load_config) == lief.PE.LoadConfigurationV5 - or type(load_config) == lief.PE.LoadConfigurationV6 - or type(load_config) == lief.PE.LoadConfigurationV7 - ) + return type(load_config) == lief.PE.LoadConfigurationV4 or issubclass(type(load_config), lief.PE.LoadConfigurationV4) @staticmethod def get_chpe_bintype(bin_: lief.PE) -> "BinType":