Improved loop detection.

3 files changed, 69 insertions, 1 deletions

diff --git a/tests/analysis/disass/block.py b/tests/analysis/disass/block.py
index f8e6fe9..9b9d529 100644
--- a/tests/analysis/disass/block.py
+++ b/tests/analysis/disass/block.py
@@ -7,9 +7,12 @@
 
 from chrysacase import ChrysalideTestCase
 from pychrysalide.analysis.contents import FileContent
-from pychrysalide.analysis import LoadedBinary
+from pychrysalide.analysis import BinRoutine, LoadedBinary
 from pychrysalide.arch import ArchInstruction
+from pychrysalide.arch import vmpa
 from pychrysalide.format.elf import ElfFormat
+from pychrysalide.format import FlatFormat
+from pychrysalide.glibext import BinPortion
 import os
 import sys
 
@@ -184,3 +187,59 @@ class TestBasicBlocks(ChrysalideTestCase):
                     loop_count += 1
 
         self.assertEqual(loop_count, 3)
+
+
+    def testOtherLoops(self):
+        """Check situation with some binary codes old troubles."""
+
+        # Malwre e8e1bc048ef123a9757a9b27d1bf53c092352a26bdbf9fbdc10109415b5cadac
+        # Fonction jinit_color_converter de lib/armeabi/libgame.so
+
+        fullname = sys.modules[self.__class__.__module__].__file__
+        filename = os.path.basename(fullname)
+
+        baselen = len(fullname) - len(filename)
+
+        cnt = FileContent(fullname[:baselen] + 'jinit_color_converter.bin')
+        self.assertIsNotNone(cnt)
+
+        fmt = FlatFormat(cnt)
+
+        fmt.set_machine('armv7')
+
+        base = vmpa(0, 0x12a524)
+
+        p = BinPortion(BinPortion.BPC_CODE, base, cnt.size)
+        p.rights = BinPortion.PAC_READ | BinPortion.PAC_EXEC
+
+        fmt.register_user_portion(p)
+
+        fmt.register_code_point(base.virt + 1, True)
+
+        sym = BinRoutine()
+        sym.range = p.range
+
+        fmt.add_symbol(sym)
+
+        binary = LoadedBinary(fmt)
+
+        status = binary.analyze_and_wait()
+        self.assertTrue(status)
+
+        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, 3)
+
+        loop_count = 0
+
+        for ins in binary.processor.instrs:
+            for _, dt in ins.destinations:
+                if dt == ArchInstruction.ILT_LOOP:
+                    loop_count += 1
+
+        self.assertEqual(loop_count, 3)
diff --git a/tests/analysis/disass/jinit_color_converter.bin b/tests/analysis/disass/jinit_color_converter.bin
new file mode 100644
index 0000000..1eb2715
--- /dev/null
+++ b/tests/analysis/disass/jinit_color_converter.bin
diff --git a/tests/format/flat.py b/tests/format/flat.py
index 4ab5aa2..6924e42 100644
--- a/tests/format/flat.py
+++ b/tests/format/flat.py
@@ -8,7 +8,9 @@
 from chrysacase import ChrysalideTestCase
 from pychrysalide.analysis import LoadedBinary
 from pychrysalide.analysis.contents import MemoryContent
+from pychrysalide.arch import vmpa
 from pychrysalide.format import FlatFormat
+from pychrysalide.glibext import BinPortion
 
 
 class TestFlatFormat(ChrysalideTestCase):
@@ -25,6 +27,13 @@ class TestFlatFormat(ChrysalideTestCase):
         fmt = FlatFormat(cnt)
         fmt.set_machine('armv7')
 
+        base = vmpa(0, 0)
+
+        p = BinPortion(BinPortion.BPC_CODE, base, cnt.size)
+        p.rights = BinPortion.PAC_READ | BinPortion.PAC_EXEC
+
+        fmt.register_user_portion(p)
+
         binary = LoadedBinary(fmt)
 
         binary.analyze_and_wait()