basicblock.py

class BasicBlock:
    """
    A basic block is a set of instructions, that has a single entry and single exit.
    Execution begins from the top and ends at the bottom. There can be no branching
    in between.
    """

    def __init__(self):
        self.address = 0
        self.instructions = []
        self.has_xrefs_to = False
        # Instructions which xreference this basic block
        self.xref_instructions = []

        # b_seen is used to perform a DFS of basicblocks
        self.b_seen = False

    def add_instruction(self, ins):
        self.instructions.append(ins)

    def instruction_iter(self):
        """
        An iterator for traversing over the instructions.

        :return: An iterator for iterating over the instruction
        """
        for ins in self.instructions:
            yield ins

    def size(self):
        """
        Calculates the size of the basic block
        :return:
        """
        return reduce(lambda x, ins: x + ins.size, self.instructions, 0)