diff options
author | Cyrille Bagard <nocbos@gmail.com> | 2019-02-21 20:33:20 (GMT) |
---|---|---|
committer | Cyrille Bagard <nocbos@gmail.com> | 2019-02-21 20:33:20 (GMT) |
commit | d6bf56e4a75be0fe0946189a42dd7baecb399903 (patch) | |
tree | 4fe89b83cd353c894a9214c063551ca225740c31 /tests/analysis/disass/links.py | |
parent | d992b4ae5a35954d4fda460735f6dc861ae7bd26 (diff) |
Fixed a little mistake introduced in 7fc86b40.
Diffstat (limited to 'tests/analysis/disass/links.py')
-rw-r--r-- | tests/analysis/disass/links.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/tests/analysis/disass/links.py b/tests/analysis/disass/links.py index e0c9ec9..e4c9946 100644 --- a/tests/analysis/disass/links.py +++ b/tests/analysis/disass/links.py @@ -62,11 +62,14 @@ class TestDisassLinks(ChrysalideTestCase): binary.analyze_and_wait() + sym = fmt.find_symbol_by_label('main') + self.assertIsNotNone(sym) + nat_count = 0 - for ins in binary.processor.instrs: - for _, dt in ins.destinations: - if dt == ArchInstruction.ILT_EXEC_FLOW: + for blk in sym.basic_blocks: + for _, dt in blk.destinations: + if dt == ArchInstruction.ILT_LOOP: nat_count += 1 - self.assertEqual(nat_count, 3) + self.assertEqual(nat_count, 1) |