summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/analysis/disass/loop.c10
-rw-r--r--tests/analysis/disass/block.py61
-rw-r--r--tests/analysis/disass/jinit_color_converter.binbin0 -> 456 bytes
-rw-r--r--tests/format/flat.py9
4 files changed, 78 insertions, 2 deletions
diff --git a/src/analysis/disass/loop.c b/src/analysis/disass/loop.c
index c37c7f6..b001992 100644
--- a/src/analysis/disass/loop.c
+++ b/src/analysis/disass/loop.c
@@ -409,7 +409,15 @@ static void define_basic_blocks_loops(GBlockList *list, bblock_info_t *info)
links = get_block_predecessors(block, info, &count);
for (k = 0; k < count; k++)
- if (should_be_natural_loop_link(iter->iloop_header, links[k].info->iloop_header))
+ if (should_be_natural_loop_link(iter->iloop_header, links[k].info->iloop_header)
+ /**
+ * Il se peut qu'un bloc fasse référence à lui même !
+ *
+ * Cf. tests/analysis/disass/jinit_color_converter.bin
+ *
+ * On évite ici une boucle sans fin en officialisant cette boucle.
+ */
+ || links[k].info == iter->iloop_header)
{
g_basic_block_get_boundaries(G_BASIC_BLOCK(links[k].linked), NULL, &last);
g_basic_block_get_boundaries(G_BASIC_BLOCK(block), &first, NULL);
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
Binary files differ
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()