Exemple #1
0
 def __init__(self,
              opcodes,
              imageFileName,
              symbols,
              indirectAddresses,
              directAddresses=()):
     self.opcodes = opcodes
     self.memory = ReadWriteMemory(file(imageFileName, "rb"))
     self.memoryExplorer = MemoryExplorer(size=len(self.memory),
                                          offset=0x0000)
     self.getByte = self.memory.createGetter(1, self.memory.BIG_ENDIAN)
     self.getWord = self.memory.createGetter(2, self.memory.BIG_ENDIAN)
     self.symbols = symbols
     Operation.symbols = self.symbols
     self.pc = 0x0000
     self.callTargets = set(
         tuple([self.getWord(e)
                for e in indirectAddresses])).union(set(directAddresses))
     self.jumpTargets = set()
     self.processed = set()
Exemple #2
0
 def __init__(self, opcodes, imageFileName, symbols, indirectAddresses, directAddresses = ()):
     self.opcodes = opcodes
     self.memory = ReadWriteMemory(file(imageFileName, "rb"))
     self.memoryExplorer = MemoryExplorer(size = len(self.memory), offset = 0x0000)
     self.getByte = self.memory.createGetter(1, self.memory.BIG_ENDIAN)
     self.getWord = self.memory.createGetter(2, self.memory.BIG_ENDIAN)
     self.symbols = symbols
     Operation.symbols = self.symbols
     self.pc = 0x0000
     self.callTargets = set(tuple([self.getWord(e) for e in indirectAddresses])).union(set(directAddresses))
     self.jumpTargets = set()
     self.processed = set()
Exemple #3
0
 def __init__(self, image):
     self.memory = ReadWriteMemory(StringIO(image))
     self.getByte = self.memory.createGetter(1, self.memory.BIG_ENDIAN)
     self.getWord = self.memory.createGetter(2, self.memory.BIG_ENDIAN)
Exemple #4
0
class Disassembler(object):
    def __init__(self, opcodes, imageFileName, symbols, indirectAddresses, directAddresses = ()):
        self.opcodes = opcodes
        self.memory = ReadWriteMemory(file(imageFileName, "rb"))
        self.memoryExplorer = MemoryExplorer(size = len(self.memory), offset = 0x0000)
        self.getByte = self.memory.createGetter(1, self.memory.BIG_ENDIAN)
        self.getWord = self.memory.createGetter(2, self.memory.BIG_ENDIAN)
        self.symbols = symbols
        Operation.symbols = self.symbols
        self.pc = 0x0000
        self.callTargets = set(tuple([self.getWord(e) for e in indirectAddresses])).union(set(directAddresses))
        self.jumpTargets = set()
        self.processed = set()
        #print "Call-Targets: ", [hex(x) for x in self.callTargets]
        #print sorted([hex(x) for x in self.jumpTargets])

    def disassemble(self):
        lines = set()
        while self.callTargets:
            target = self.callTargets.pop()
            result = self.disassembleTillReturn(target)
            #lines.extend(result)
            lines = lines.union(result)
        #return sorted(set(lines), key = lambda o: o.address)
        return dict((e.address, e) for e in lines)

    def disassembleTillReturn(self, address):
        lines = set()
        origAddress = address
        if address == 0xaf4:
            print ""
        while True:
            try:
                line = self.disassembleLine(address)
            except IndexError as e:
                    print "Index-Error while disassembling line: %s" % e
                    break
            except IllegalOpcode as e:
                    print "Illegal Opcode while disassembling line: %s" % e
                    break
            address += line.size
            #if not self.memoryExplorer.isExplored(origAddress): # and line not in lines:
            if origAddress not in self.processed:
                lines.add(line) #lines.append(line)
            if line.opcode in RETURNS:
                break
        self.processed.add(origAddress)
        while self.jumpTargets:
            target = self.jumpTargets.pop()
            if not self.memoryExplorer.isExplored(target):
                lines = lines.union(self.disassembleTillReturn(target))#lines.extend(self.disassembleTillReturn(target))
        return lines

    def disassembleLine(self, address):
        op = self.getByte(address)
        opcode = self.opcodes[op]
        opcodeName = opcode.name
        addressingMode = opcode.addressingMode
        opcodeSize = opcode.size
        operandData = self.getWord(address + 1) if opcodeSize == 3 else self.getByte(address + 1) if opcodeSize == 2 else None
        operand = ""
        if op in ILLEGAL_OPCODES:
            raise IllegalOpcode("0x%02x [Address: 0x%04x]" % (op, address, ))
        else:
            if address in Operation.symbols:
                #print "0x%04x ==> %s" % (address, Operation.symbols[address])
                label = Operation.symbols[address]
            else:
                label = ''
            operation = Operation(address, opcodeSize, label, op, opcodeName, operand, operandData, addressingMode, '')
            #print "$%04x %02x %s"   % (address, op, operation.formatOperandData()),
            #print opcodeName,

            displayString, destination = operation.processes()
            #print displayString

            self.memoryExplorer.setExplored(address, opcodeSize)
            newAddress = address + opcodeSize
        ##
        ## Code above is almost identical with simulator.
        ##
        ## START SPECIAL.


        ## END SPECIAL.
        ##
        ## Code below is almost identical with simulator.
        ##
        if op in JUMPS:
            if destination and not self.memoryExplorer.isExplored(destination):  # in self.processed:
                self.jumpTargets.add(destination)
        elif op in CALLS:
            if destination and not self.memoryExplorer.isExplored(destination): #destination in self.processed:
                if destination < len(self.memory):       ### CHECK!!!
                    self.callTargets.add(destination)
                else:
                    print "Uups!?"
        return operation
Exemple #5
0
 def __init__(self, image):
     self.memory = ReadWriteMemory(StringIO(image))
     self.getByte = self.memory.createGetter(1, self.memory.BIG_ENDIAN)
     self.getWord = self.memory.createGetter(2, self.memory.BIG_ENDIAN)
Exemple #6
0
class Disassembler(object):
    def __init__(self,
                 opcodes,
                 imageFileName,
                 symbols,
                 indirectAddresses,
                 directAddresses=()):
        self.opcodes = opcodes
        self.memory = ReadWriteMemory(file(imageFileName, "rb"))
        self.memoryExplorer = MemoryExplorer(size=len(self.memory),
                                             offset=0x0000)
        self.getByte = self.memory.createGetter(1, self.memory.BIG_ENDIAN)
        self.getWord = self.memory.createGetter(2, self.memory.BIG_ENDIAN)
        self.symbols = symbols
        Operation.symbols = self.symbols
        self.pc = 0x0000
        self.callTargets = set(
            tuple([self.getWord(e)
                   for e in indirectAddresses])).union(set(directAddresses))
        self.jumpTargets = set()
        self.processed = set()
        #print "Call-Targets: ", [hex(x) for x in self.callTargets]
        #print sorted([hex(x) for x in self.jumpTargets])

    def disassemble(self):
        lines = set()
        while self.callTargets:
            target = self.callTargets.pop()
            result = self.disassembleTillReturn(target)
            #lines.extend(result)
            lines = lines.union(result)
        #return sorted(set(lines), key = lambda o: o.address)
        return dict((e.address, e) for e in lines)

    def disassembleTillReturn(self, address):
        lines = set()
        origAddress = address
        if address == 0xaf4:
            print ""
        while True:
            try:
                line = self.disassembleLine(address)
            except IndexError as e:
                print "Index-Error while disassembling line: {0!s}".format(e)
                break
            except IllegalOpcode as e:
                print "Illegal Opcode while disassembling line: {0!s}".format(
                    e)
                break
            address += line.size
            #if not self.memoryExplorer.isExplored(origAddress): # and line not in lines:
            if origAddress not in self.processed:
                lines.add(line)  #lines.append(line)
            if line.opcode in RETURNS:
                break
        self.processed.add(origAddress)
        while self.jumpTargets:
            target = self.jumpTargets.pop()
            if not self.memoryExplorer.isExplored(target):
                lines = lines.union(self.disassembleTillReturn(
                    target))  #lines.extend(self.disassembleTillReturn(target))
        return lines

    def disassembleLine(self, address):
        op = self.getByte(address)
        opcode = self.opcodes[op]
        opcodeName = opcode.name
        addressingMode = opcode.addressingMode
        opcodeSize = opcode.size
        operandData = self.getWord(address +
                                   1) if opcodeSize == 3 else self.getByte(
                                       address +
                                       1) if opcodeSize == 2 else None
        operand = ""
        if op in ILLEGAL_OPCODES:
            raise IllegalOpcode("0x{0:02x} [Address: 0x{1:04x}]".format(
                op, address))
        else:
            if address in Operation.symbols:
                #print "0x%04x ==> %s" % (address, Operation.symbols[address])
                label = Operation.symbols[address]
            else:
                label = ''
            operation = Operation(address, opcodeSize, label, op, opcodeName,
                                  operand, operandData, addressingMode, '')
            #print "$%04x %02x %s"   % (address, op, operation.formatOperandData()),
            #print opcodeName,

            displayString, destination = operation.processes()
            #print displayString

            self.memoryExplorer.setExplored(address, opcodeSize)
            newAddress = address + opcodeSize
        ##
        ## Code above is almost identical with simulator.
        ##
        ## START SPECIAL.

        ## END SPECIAL.
        ##
        ## Code below is almost identical with simulator.
        ##
        if op in JUMPS:
            if destination and not self.memoryExplorer.isExplored(
                    destination):  # in self.processed:
                self.jumpTargets.add(destination)
        elif op in CALLS:
            if destination and not self.memoryExplorer.isExplored(
                    destination):  #destination in self.processed:
                if destination < len(self.memory):  ### CHECK!!!
                    self.callTargets.add(destination)
                else:
                    print "Uups!?"
        return operation