scripts: match on exe type over str in security-check.py

Author: fanquake

contrib/devtools/security-check.py

@@ -180,22 +180,22 @@ def check_control_flow(binary) -> bool:
 
 
 CHECKS = {
-'ELF': [
+lief.EXE_FORMATS.ELF: [
     ('PIE', check_PIE),
     ('NX', check_NX),
     ('RELRO', check_ELF_RELRO),
     ('Canary', check_ELF_Canary),
     ('separate_code', check_ELF_separate_code),
 ],
-'PE': [
+lief.EXE_FORMATS.PE: [
     ('PIE', check_PIE),
     ('DYNAMIC_BASE', check_PE_DYNAMIC_BASE),
     ('HIGH_ENTROPY_VA', check_PE_HIGH_ENTROPY_VA),
     ('NX', check_NX),
     ('RELOC_SECTION', check_PE_RELOC_SECTION),
     ('CONTROL_FLOW', check_PE_control_flow),
 ],
-'MACHO': [
+lief.EXE_FORMATS.MACHO: [
     ('PIE', check_PIE),
     ('NOUNDEFS', check_MACHO_NOUNDEFS),
     ('NX', check_NX),
@@ -210,7 +210,7 @@ def check_control_flow(binary) -> bool:
     for filename in sys.argv[1:]:
         try:
             binary = lief.parse(filename)
-            etype = binary.format.name
+            etype = binary.format
             if etype == lief.EXE_FORMATS.UNKNOWN:
                 print(f'{filename}: unknown executable format')
                 retval = 1