import time
from pdb import pm
from miasm2.arch.msp430.arch import *
from miasm2.core.asmblock import AsmSymbolPool

symbol_pool = AsmSymbolPool()

def h2i(s):
    return s.replace(' ', '').decode('hex')


def u16swap(i):
    return struct.unpack('<H', struct.pack('>H', i))[0]


reg_tests_msp = [
    ("4456    mov.w      SP, R4",
     "0441"),
    ("4d4f    mov.b      R13, R15",
     "4f4d"),
    ("49fe    mov.w      @R13, R9",
     "294d"),
    ("4982    mov.w      0x10(R14), R13",
     "1d4e1000"),
    ("4972    mov.w      R14, 0x0(SP)",
     "814e0000"),
    ("46de    mov.w      0x2(R14), 0x2(R13)",
     "9d4e02000200"),
    ("469e    mov.w      @0x2400, R11",
     "1b420024"),
    ("4c14    mov.w      0x4A96, R15",
     "3f40964a"),
    ("47c0    mov.w      0x1, R8",
     "1843"),
    ("48fc    mov.w      0x2, R10",
     "2a43"),
    ("44fe    mov.w      0x4, R7",
     "2742"),
    ("4a28    mov.w      0xFFFF, R15",
     "3f43"),
    ("4416    mov.w      R5, @0x15C",
     "82455c01"),

    ("4a22    add.w      R11, R15",
     "0f5b"),
    ("448e    sub.w      R15, SP",
     "018f"),
    ("4474    cmp.b      @R15, R13",
     "6d9f"),
    ("46a8    bit.w      0x1, R13",
     "1db3"),
    ("440a    bis.w      0x5A08, R5",
     "35d0085a"),
    ("4c1a    xor.w      R15, R10",
     "0aef"),
    ("4408    and.b      0xFF, R5",
     "75f3"),


    ("4cf0    push.w     SR",
     "0212"),
    ("4d6e    push.w     0x0",
     "0312"),
    ("45dc    push.w     0x2(R11)",
     "1b120200"),
    ("49cc    push.w     R11",
     "0b12"),

    ("443a    call       0x4B66",
     "b012664b"),

    ("4442    jmp        0xFFFC",
     "fd3f"),
    ("4422    jnz        0xFFF4",
     "f923"),

    ("xxxx    mov.b      @R13+, 0x0(R14)",
     "fe4d0000"),

    ("4a36    mov.w      @SP+, PC",
     "3041"),


]

ts = time.time()

for s, l in reg_tests_msp:
    print "-" * 80
    s = s[8:]
    b = h2i((l))
    print repr(b)
    mn = mn_msp430.dis(b, None)
    print [str(x) for x in mn.args]
    print s
    print mn
    assert(str(mn) == s)
    # print hex(b)
    # print [str(x.get()) for x in mn.args]
    l = mn_msp430.fromstring(s, symbol_pool, None)
    # print l
    assert(str(l) == s)
    a = mn_msp430.asm(l)
    print [x for x in a]
    print repr(b)
    # print mn.args
    assert(b in a)