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/loop.py | |
parent | d992b4ae5a35954d4fda460735f6dc861ae7bd26 (diff) |
Fixed a little mistake introduced in 7fc86b40.
Diffstat (limited to 'tests/analysis/disass/loop.py')
-rw-r--r-- | tests/analysis/disass/loop.py | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/tests/analysis/disass/loop.py b/tests/analysis/disass/loop.py new file mode 100644 index 0000000..6f7f405 --- /dev/null +++ b/tests/analysis/disass/loop.py @@ -0,0 +1,75 @@ +#!/usr/bin/python3 +# -*- coding: utf-8 -*- + + +# S'assure du bon fonctionnement des blocs basiques + + +from chrysacase import ChrysalideTestCase +from pychrysalide.analysis.contents import FileContent +from pychrysalide.analysis import LoadedBinary +from pychrysalide.arch import ArchInstruction +from pychrysalide.format.elf import ElfFormat +import os +import sys + + +class TestDisassLoop(ChrysalideTestCase): + """TestCase for loop detection.""" + + @classmethod + def setUpClass(cls): + + super(TestDisassLoop, cls).setUpClass() + + cls.log('Compile binary "h2" if needed...') + + fullname = sys.modules[cls.__module__].__file__ + dirpath = os.path.dirname(fullname) + + os.system('make -C %s h2 > /dev/null 2>&1' % dirpath) + + + @classmethod + def tearDownClass(cls): + + super(TestDisassLoop, cls).tearDownClass() + + cls.log('Delete built binaries...') + + fullname = sys.modules[cls.__module__].__file__ + dirpath = os.path.dirname(fullname) + + os.system('make -C %s clean > /dev/null 2>&1' % dirpath) + + + def testLoopOverflow(self): + """Ensure loop number does't not get overestimated.""" + + fullname = sys.modules[self.__class__.__module__].__file__ + filename = os.path.basename(fullname) + + baselen = len(fullname) - len(filename) + + cnt = FileContent(fullname[:baselen] + 'h2') + self.assertIsNotNone(cnt) + + fmt = ElfFormat(cnt) + self.assertIsNotNone(fmt) + + binary = LoadedBinary(fmt) + self.assertIsNotNone(binary) + + binary.analyze_and_wait() + + sym = fmt.find_symbol_by_label('main') + self.assertIsNotNone(sym) + + loop_count = 0 + + for blk in sym.basic_blocks: + for _, dt in blk.destinations: + if dt == ArchInstruction.ILT_LOOP: + loop_count += 1 + + self.assertEqual(loop_count, 2) |