Improved the support of some unusual endiannesses.

1 files changed, 129 insertions, 0 deletions

diff --git a/tests/analysis/contents/endian.py b/tests/analysis/contents/endian.py
new file mode 100644
index 0000000..209600a
--- /dev/null
+++ b/tests/analysis/contents/endian.py
@@ -0,0 +1,129 @@
+#!/usr/bin/python3-dbg
+# -*- coding: utf-8 -*-
+
+
+# Tests minimalistes pour valider le boutisme des accès mémoire.
+
+
+from chrysacase import ChrysalideTestCase
+from pychrysalide.analysis.contents import FileContent, RestrictedContent
+from pychrysalide.arch import vmpa
+import tempfile
+
+
+class TestEndianness(ChrysalideTestCase):
+    """TestCase for analysis.BinContent."""
+
+    @classmethod
+    def setUpClass(cls):
+
+        super(TestEndianness, cls).setUpClass()
+
+        cls._out = tempfile.NamedTemporaryFile()
+
+        cls._out.write(b'\x01\x02\x03\x04')
+        cls._out.write(b'\x05\x06\x07\x08')
+        cls._out.write(b'\x11\x12\x13\x14')
+        cls._out.write(b'\x15\x16\x17\x18')
+        cls._out.write(b'\x21\x22\x23\x24')
+        cls._out.write(b'\x25\x26\x27\x28')
+        cls._out.write(b'\x31\x32\x33\x34')
+        cls._out.write(b'\x35\x36\x37\x38')
+
+        cls._out.flush()
+
+        cls.log('Using temporary file "%s"' % cls._out.name)
+
+
+    @classmethod
+    def tearDownClass(cls):
+
+        super(TestEndianness, cls).tearDownClass()
+
+        cls.log('Delete file "%s"' % cls._out.name)
+
+        cls._out.close()
+
+
+    def testMiddleEndianness(self):
+        """Test some old endianness."""
+
+        fcnt = FileContent(self._out.name)
+
+        # 16 bits
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u16(start, vmpa.SRE_LITTLE_WORD)
+        self.assertEqual(val, 0x1516)
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u16(start, vmpa.SRE_BIG_WORD)
+        self.assertEqual(val, 0x1615)
+
+        # 32 bits
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u32(start, vmpa.SRE_LITTLE_WORD)
+        self.assertEqual(val, 0x17181516)
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u32(start, vmpa.SRE_BIG_WORD)
+        self.assertEqual(val, 0x16151817)
+
+        # 64 bits
+
+        start = vmpa(0, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u64(start, vmpa.SRE_LITTLE_WORD)
+        self.assertEqual(val, 0x0708050603040102)
+
+        start = vmpa(0, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u64(start, vmpa.SRE_BIG_WORD)
+        self.assertEqual(val, 0x0201040306050807)
+
+
+    def testEndianness(self):
+        """Test usual endianness."""
+
+        fcnt = FileContent(self._out.name)
+
+        # 16 bits
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u16(start, vmpa.SRE_LITTLE)
+        self.assertEqual(val, 0x1615)
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u16(start, vmpa.SRE_BIG)
+        self.assertEqual(val, 0x1516)
+
+        # 32 bits
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u32(start, vmpa.SRE_LITTLE)
+        self.assertEqual(val, 0x18171615)
+
+        start = vmpa(12, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u32(start, vmpa.SRE_BIG)
+        self.assertEqual(val, 0x15161718)
+
+        # 64 bits
+
+        start = vmpa(0, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u64(start, vmpa.SRE_LITTLE)
+        self.assertEqual(val, 0x0807060504030201)
+
+        start = vmpa(0, vmpa.VMPA_NO_VIRTUAL)
+
+        val = fcnt.read_u64(start, vmpa.SRE_BIG)
+        self.assertEqual(val, 0x0102030405060708)