blob: 43b63d51d30797aa8432e4b2cfe62bf373e4563a [file] [log] [blame]
Nguyen Anh Quynha6519b02014-04-16 20:03:55 +08001#!/usr/bin/env python
2
3# Capstone Python bindings, by Nguyen Anh Quynnh <aquynh@gmail.com>
4from __future__ import print_function
5import sys
6from capstone import *
7
Nguyen Anh Quynh4171e482014-04-16 20:44:10 +08008CODE32 = b"\xc0\xe0\x02"
Nguyen Anh Quynhd7110602014-04-23 12:40:58 +08009CODE32 += b"\xc0\xf6\x02" # sal dh, 0
10CODE32 += b"\xc1\xf6\x00" # sal esi, 0
Nguyen Anh Quynh33e16362014-04-20 11:32:00 +080011CODE32 += b"\x82\xc0\x00"
Nguyen Anh Quynhd7110602014-04-23 12:40:58 +080012CODE32 += b"\x0f\x1a\x00" # nop dword ptr [eax]
13CODE32 += b"\xf7\xc0\x11\x22\x33\x44" # test eax, 0x44332211
14CODE32 += b"\xf7\xc8\x11\x22\x33\x44" # test eax, 0x44332211
15CODE32 += b"\xf7\x88\x00\x00\x00\x00\x00\x00\x00\x00" # test dword ptr [eax], 0
16CODE32 += b"\xf6\x88\x00\x00\x00\x00\x00" # test byte ptr [eax], 0
17
Nguyen Anh Quynh191c0702014-04-24 22:50:54 +080018CODE32 += b"\xd9\xd8" # fstpnce st(0), st(0)
19CODE32 += b"\xdf\xdf" # fstp st(7), st(0)
20
21CODE32 += b"\x0f\x20\x00" # mov eax, cr0
22CODE32 += b"\x0f\x20\x40" # mov eax, cr0
23CODE32 += b"\x0f\x20\x80" # mov eax, cr0
24
25CODE32 += b"\x0f\x22\x00" # mov cr0, eax
26CODE32 += b"\x0f\x22\x40" # mov cr0, eax
27CODE32 += b"\x0f\x22\x80" # mov cr0, eax
28
29CODE32 += b"\x0f\x21\x00" # mov eax, dr0
30CODE32 += b"\x0f\x21\x40" # mov eax, dr0
31CODE32 += b"\x0f\x21\x80" # mov eax, dr0
32
33CODE32 += b"\x0f\x23\x00" # mov dr0, eax
34CODE32 += b"\x0f\x23\x40" # mov dr0, eax
35CODE32 += b"\x0f\x23\x80" # mov dr0, eax
36
Nguyen Anh Quynh668b96c2014-11-02 23:15:18 +080037CODE32_MEMREF = b"\x8b\x84\x91\x23\x01\x00\x00"
38CODE32_MEMREF += b"\x8b\x04\x95\x23\x01\x00\x00"
39CODE32_MEMREF += b"\x8b\x04\x95\xdd\xfe\xff\xff"
40CODE32_MEMREF += b"\xa1\x23\x01\x00\x00"
41CODE32_MEMREF += b"\xa1\x00\x00\x00\x00"
Nguyen Anh Quynhed0fbce2014-11-02 23:32:39 +080042CODE32_MEMREF += b"\xa1\xdd\xfe\xff\xff"
Nguyen Anh Quynhb87f8552014-11-02 23:38:35 +080043CODE32_MEMREF += b"\x8b\x04\x91"
Nguyen Anh Quynh668b96c2014-11-02 23:15:18 +080044
Nguyen Anh Quynhc3528972014-11-04 11:04:32 +080045CODE64_MEMREF = b"\xa3\x0b\x00\x00\x0f\xbe\xc0\x48\x83"
46CODE64_MEMREF += b"\xa0\x71\xfa\xff\x48\x85\xc0\x48\x89"
Nguyen Anh Quynha6519b02014-04-16 20:03:55 +080047
Nguyen Anh Quynhff7bba32014-11-03 16:32:06 +080048CODE32_ARITH = b"\x83\xe0\xf7"
49CODE32_ARITH += b"\x83\xe0\x10"
50CODE32_ARITH += b"\x83\xe0\x00"
51CODE32_ARITH += b"\x80\x23\x10"
52
53CODE64_ARITH = b"\x41\x83\xe0\xfa"
54CODE64_ARITH += b"\x48\x83\xe4\xf0"
55
Nguyen Anh Quynhc3528972014-11-04 11:04:32 +080056CODE32_IMM = b"\xc2\xb8\xc0"
57CODE32_IMM += b"\xc2\x0f\x92"
58CODE32_IMM += b"\x02\x2d\x00\x00\x00\x83"
59
Nguyen Anh Quynhff7bba32014-11-03 16:32:06 +080060
61
Nguyen Anh Quynha6519b02014-04-16 20:03:55 +080062_python3 = sys.version_info.major == 3
63
64all_tests = (
65 (CS_ARCH_X86, CS_MODE_32, CODE32, "X86 32 (Intel syntax)", 0),
Nguyen Anh Quynh668b96c2014-11-02 23:15:18 +080066 (CS_ARCH_X86, CS_MODE_32, CODE32, "X86 32 (ATT syntax)", CS_OPT_SYNTAX_ATT),
Nguyen Anh Quynhff7bba32014-11-03 16:32:06 +080067
Nguyen Anh Quynh668b96c2014-11-02 23:15:18 +080068 (CS_ARCH_X86, CS_MODE_32, CODE32_MEMREF, "X86 32 MemRef (Intel syntax)", 0),
69 (CS_ARCH_X86, CS_MODE_32, CODE32_MEMREF, "X86 32 MemRef (ATT syntax)", CS_OPT_SYNTAX_ATT),
Nguyen Anh Quynhc3528972014-11-04 11:04:32 +080070 (CS_ARCH_X86, CS_MODE_64, CODE64_MEMREF, "X86 64 (Intel syntax)", 0),
Nguyen Anh Quynhff7bba32014-11-03 16:32:06 +080071
72 (CS_ARCH_X86, CS_MODE_32, CODE32_ARITH, "X86 32 (Intel syntax)", 0),
Nguyen Anh Quynhff7bba32014-11-03 16:32:06 +080073 (CS_ARCH_X86, CS_MODE_64, CODE64_ARITH, "X86 64 (Intel syntax)", 0),
Nguyen Anh Quynhc3528972014-11-04 11:04:32 +080074
75 (CS_ARCH_X86, CS_MODE_32, CODE32_IMM, "X86 32 (Intel syntax)", 0),
76 (CS_ARCH_X86, CS_MODE_32, CODE32_IMM, "X86 32 (Intel syntax)", CS_OPT_SYNTAX_ATT),
Nguyen Anh Quynha6519b02014-04-16 20:03:55 +080077)
78
79
80def to_hex(s):
81 if _python3:
82 return " ".join("0x{0:02x}".format(c) for c in s) # <-- Python 3 is OK
83 else:
84 return " ".join("0x{0:02x}".format(ord(c)) for c in s)
85
86# ## Test cs_disasm_quick()
87def test_cs_disasm_quick():
88 for (arch, mode, code, comment, syntax) in all_tests:
89 print("Platform: %s" % comment)
90 print("Code: %s" %(to_hex(code))),
91 print("Disasm:")
Nguyen Anh Quynh668b96c2014-11-02 23:15:18 +080092 md = Cs(arch, mode)
93 if syntax != 0:
94 md.syntax = syntax
95 for insn in md.disasm(code, 0x1000):
96 print("0x%x:\t%s\t%s" % (insn.address, insn.mnemonic, insn.op_str))
97 print("--------")
Nguyen Anh Quynha6519b02014-04-16 20:03:55 +080098
99
100if __name__ == '__main__':
101 test_cs_disasm_quick()