bytecodes.py

from opcodes import mono_ops
from opcodes import bin_ops


class ByteCode(object):
    def __init__(self, opcode, raw_byte, major_ad, minor_ad=0):
        self.opcode = opcode
        self.raw_byte = raw_byte
        self.__major_ad = major_ad
        self.__minor_ad = minor_ad
        self.dependencies = dict()
        # this is bad, but this is here
        self.__block_id = -1

    def set_container_block_id(self, block_id):
        self.__block_id = block_id

    def get_container_block_id(self):
        return self.__block_id

    def get_str_address(self):
        str_ad = hex(self.__major_ad)[2:]
        if self.__minor_ad > 0:
            str_ad += "+" + str(self.__minor_ad)
        elif self.__minor_ad < 0:
            str_ad += str(self.__minor_ad)
        return str_ad

    def get_address(self):
        return self.__major_ad

    def reset_dependencies(self):
        self.dependencies = dict()

    def set_dependency(self, pos, node):
        self.dependencies[pos] = node

    def is_jumpdest(self):
        return self.opcode == "JUMPDEST"

    def is_jump(self):
        return self.opcode == "JUMP"

    def __str__(self):
        return "0x%s\t%s" % (self.get_str_address(), self.opcode)


class PushByteCode(ByteCode):
    def __init__(self, opcode, raw_byte, major_ad, minor_ad=0):
        ByteCode.__init__(self, opcode, raw_byte, major_ad, minor_ad)

    def reset_dependencies(self):
        # override to do nothing
        pass

    def get_value(self):
        return self.dependencies[0]

    def __str__(self):
        value_str = "\t\t0x%x" % self.dependencies[0]
        return ByteCode.__str__(self) + value_str


class MonoOpByteCode(ByteCode):
    def __init__(self, opcode, raw_byte, major_ad):
        ByteCode.__init__(self, opcode, raw_byte, major_ad)
        self.operator = mono_ops[self.opcode]


class BinOpByteCode(ByteCode):
    def __init__(self, opcode, raw_byte, major_ad):
        ByteCode.__init__(self, opcode, raw_byte, major_ad)
        self.operator = bin_ops[self.opcode]