diff options
Diffstat (limited to 'suite/testcommon.py')
| -rw-r--r-- | suite/testcommon.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/suite/testcommon.py b/suite/testcommon.py index 6941f8b3..a70ec09f 100644 --- a/suite/testcommon.py +++ b/suite/testcommon.py @@ -1302,6 +1302,26 @@ class VerifyBuilder(Builder): finally: self.delete_package("helloworld") + def test_get_il_vars(self): + file_name = self.unpackage_file("helloworld") + try: + with binja.BinaryViewType.get_view_of_file(file_name) as bv: + main_func = bv.get_functions_by_name("main")[0] + + mlil_vars = main_func.get_il_vars(FunctionGraphType.MediumLevelILFunctionGraph) + mlil_vars = list(map(lambda v: v.name, mlil_vars)) + mlil_vars.sort() + + hlil_vars = main_func.get_il_vars(FunctionGraphType.HighLevelILFunctionGraph) + hlil_vars = list(map(lambda v: v.name, hlil_vars)) + hlil_vars.sort() + + assert mlil_vars == ['argc', 'argv', 'envp', 'r0', 'r3', 'var_10', 'var_c'] + assert hlil_vars == ['argc', 'argv', 'envp'] + return True + finally: + self.delete_package("helloworld") + def test_verify_BNDB_round_trip(self): """Binary Ninja Database output doesn't match its input""" # This will test Binja's ability to save and restore databases |