from miasm2.core.asmblock import disasmEngine
from miasm2.arch.x86.arch import mn_x86


cb_x86_funcs = []


def cb_x86_disasm(*args, **kwargs):
    for func in cb_x86_funcs:
        func(*args, **kwargs)


class dis_x86(disasmEngine):
    attrib = None

    def __init__(self, bs=None, **kwargs):
        super(dis_x86, self).__init__(mn_x86, self.attrib, bs, **kwargs)
        self.dis_block_callback = cb_x86_disasm


class dis_x86_16(dis_x86):
    attrib = 16


class dis_x86_32(dis_x86):
    attrib = 32


class dis_x86_64(dis_x86):
    attrib = 64