summaryrefslogtreecommitdiff
path: root/tests/format/errors.py
blob: 36b7129bdc92df831705e4336a3488cc1f7741c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#!/usr/bin/python3-dbg
# -*- coding: utf-8 -*-


# Tests minimalistes pour valider la gestion des erreurs relevées.


from chrysacase import ChrysalideTestCase
from pychrysalide.analysis.contents import FileContent
from pychrysalide.arch import vmpa
from pychrysalide.format import BinFormat
from pychrysalide.format.elf import ElfFormat
import os
import sys


class TestFormatErrors(ChrysalideTestCase):
    """TestCase for format.BinFormat errors."""

    @classmethod
    def setUpClass(cls):

        super(TestFormatErrors, cls).setUpClass()

        cls.log('Compile binary "strings" if needed...')

        fullname = sys.modules[cls.__module__].__file__
        dirpath = os.path.dirname(fullname)

        os.system('make -C %s%self strings 2>&1 > /dev/null' % (dirpath, os.sep))


    def testBasic(self):
        """Perform some sanity tests on format error handling."""

        errlen = 3

        pattern = []

        for i in range(errlen):

            addr = vmpa(vmpa.VMPA_NO_PHYSICAL, 0x100 + i * 0x10)

            pattern.append([BinFormat.BFE_STRUCTURE, addr, 'random desc #%d' % i])

        fullname = sys.modules[self.__class__.__module__].__file__
        filename = os.path.basename(fullname)

        baselen = len(fullname) - len(filename)

        cnt = FileContent(fullname[:baselen] + 'elf' + os.sep + 'strings')
        fmt = ElfFormat(cnt, None, None)

        for i in range(errlen):

            fmt.add_error(pattern[i][0], pattern[i][1], pattern[i][2])

        self.assertEqual(len(fmt.errors), errlen)