From a5531990ae81f171cdb32cbdfad26282416ef90a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Fri, 25 Aug 2017 17:41:58 +0200 Subject: [PATCH] Do not try to run code requiring lief --- pymisp/tools/create_misp_object.py | 41 +++++++++++++++--------------- 1 file changed, 21 insertions(+), 20 deletions(-) diff --git a/pymisp/tools/create_misp_object.py b/pymisp/tools/create_misp_object.py index 8c9b304..9e2d7fb 100644 --- a/pymisp/tools/create_misp_object.py +++ b/pymisp/tools/create_misp_object.py @@ -45,24 +45,25 @@ def make_macho_objects(lief_parsed, misp_file): def make_binary_objects(filepath): misp_file = FileObject(filepath) - try: - lief_parsed = lief.parse(filepath) - if isinstance(lief_parsed, lief.PE.Binary): - return make_pe_objects(lief_parsed, misp_file) - elif isinstance(lief_parsed, lief.ELF.Binary): - return make_elf_objects(lief_parsed, misp_file) - elif isinstance(lief_parsed, lief.MachO.Binary): - return make_macho_objects(lief_parsed, misp_file) - except lief.bad_format as e: - # print('\tBad format: ', e) - pass - except lief.bad_file as e: - # print('\tBad file: ', e) - pass - except lief.parser_error as e: - # print('\tParser error: ', e) - pass - except FileTypeNotImplemented as e: # noqa - # print(e) - pass + if HAS_LIEF: + try: + lief_parsed = lief.parse(filepath) + if isinstance(lief_parsed, lief.PE.Binary): + return make_pe_objects(lief_parsed, misp_file) + elif isinstance(lief_parsed, lief.ELF.Binary): + return make_elf_objects(lief_parsed, misp_file) + elif isinstance(lief_parsed, lief.MachO.Binary): + return make_macho_objects(lief_parsed, misp_file) + except lief.bad_format as e: + # print('\tBad format: ', e) + pass + except lief.bad_file as e: + # print('\tBad file: ', e) + pass + except lief.parser_error as e: + # print('\tParser error: ', e) + pass + except FileTypeNotImplemented as e: # noqa + # print(e) + pass return misp_file, None, None