From c8f4346609f4b51474481d9e6b869076e5a96534 Mon Sep 17 00:00:00 2001 From: Brandon Miller Date: Mon, 9 Jun 2025 09:40:18 -0400 Subject: More None checks when querying IL internal to API --- python/basicblock.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'python/basicblock.py') diff --git a/python/basicblock.py b/python/basicblock.py index b7182c7a..b5000a2f 100644 --- a/python/basicblock.py +++ b/python/basicblock.py @@ -252,19 +252,23 @@ class BasicBlock: elif il_type == _function.FunctionGraphType.LiftedILFunctionGraph: return func.lifted_il elif il_type == _function.FunctionGraphType.LowLevelILSSAFormFunctionGraph: - return func.low_level_il.ssa_form + func_llil = func.low_level_il + return func_llil.ssa_form if func_llil is not None else None elif il_type == _function.FunctionGraphType.MediumLevelILFunctionGraph: return func.medium_level_il elif il_type == _function.FunctionGraphType.MediumLevelILSSAFormFunctionGraph: - return func.medium_level_il.ssa_form + func_mlil = func.medium_level_il + return func_mlil.ssa_form if func_mlil is not None else None elif il_type == _function.FunctionGraphType.MappedMediumLevelILFunctionGraph: return func.mapped_medium_level_il elif il_type == _function.FunctionGraphType.MappedMediumLevelILSSAFormFunctionGraph: - return func.mapped_medium_level_il.ssa_form + func_mmlil = func.mapped_medium_level_il + return func_mmlil.ssa_form if func_mmlil is not None else None elif il_type == _function.FunctionGraphType.HighLevelILFunctionGraph: return func.high_level_il elif il_type == _function.FunctionGraphType.HighLevelILSSAFormFunctionGraph: - return func.high_level_il.ssa_form + func_hlil = func.high_level_il + return func_hlil.ssa_form if func_hlil is not None else None elif il_type == _function.FunctionGraphType.HighLevelLanguageRepresentationFunctionGraph: return func.high_level_il else: -- cgit v1.3.1