๐Ÿ“ฆ wagoodman / mini-x86-disassembler

๐Ÿ“„ recursiveDescent.py ยท 99 lines
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99import logging
import shutil

import utils
from error import *
from strategy.decoderStrategy import DecoderStrategy

CALL_INST = ('CALL',)
FUNC_END = ('RET','RETN','RETF')
JUMP_INST = ('JMP','JZ','JNZ')
JCC_INST = ('JZ','JNZ')

class RecursiveDescent(DecoderStrategy):

    addressDefer = None

    def __init__(self, *args, **kwargs):
        super(RecursiveDescent, self).__init__(*args, **kwargs)
        self.addressDefer = []

    def decode(self, continueOnError=True,  verbose=False, detail=False):

        if not verbose:
            utils.logger.setLevel(logging.INFO)
        else:
            utils.logger.setLevel(logging.DEBUG)
            terminalSize = shutil.get_terminal_size((80, 20))

        instCount = 1

        while not self.decoder.state.isRecursiveDescentComplete():
            try:
                if verbose:
                    title = "Instruction %d" % instCount
                    utils.logger.debug(utils.colors.INVERT+(title + " "*(terminalSize.columns-len(title)))+utils.colors.NORMAL)

                operator, targetAddr = self.decoder.decodeSingleInstruction()

                instCount += 1
                if verbose:
                    self.decoder.state.showDecodeProgress(detail)

                self.decoder.state.doRecursiveDescent(operator, targetAddr)

            except InvalidTranslationValue:
                location = self.decoder.state.getCurIdx()
                try:
                    location = hex(location)
                except:
                    location = repr(location)

                try:
                    theByte = hex(self.decoder.state.contents[self.decoder.state.getCurIdx()])
                except:
                    theByte = repr("???")

                message = 'Unable to parse byte as an operand @ position %s (byte:%s).' % (location, theByte)
                utils.logger.info(utils.colors.RED+utils.colors.BOLD +message+utils.colors.NORMAL)
                self.decoder.state.markError()

                if not continueOnError:
                    break

            except InvalidOpcode:
                location = self.decoder.state.getCurIdx()
                try:
                    location = hex(location)
                except:
                    location = repr(location)

                try:
                    theByte = hex(self.decoder.state.contents[self.decoder.state.getCurIdx()])
                except:
                    theByte = repr("???")

                message = 'Unable to parse byte as an opcode @ position %s (byte:%s).' % (location, theByte)
                utils.logger.info(utils.colors.RED+utils.colors.BOLD +message+utils.colors.NORMAL)
                self.decoder.state.markError()

                if not continueOnError:
                    break
            except:
                location = self.decoder.state.getCurIdx()
                try:
                    location = hex(location)
                except:
                    location = repr(location)

                try:
                    theByte = hex(self.decoder.state.contents[self.decoder.state.getCurIdx()])
                except:
                    theByte = repr("???")

                message = 'Unrecoverable Error: Unable to parse byte @ position %s (byte:%s).' % (location, theByte)
                utils.logger.info(utils.colors.RED+utils.colors.BOLD +message+utils.colors.NORMAL)
                break

        return self.decoder.state.isComplete()