Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | # Capstone Python bindings, by Nguyen Anh Quynnh <aquynh@gmail.com> |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 4 | from __future__ import print_function |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 5 | from capstone import * |
| 6 | import binascii |
Nguyen Anh Quynh | 1098329 | 2014-05-17 09:51:15 +0800 | [diff] [blame] | 7 | from xprint import to_hex |
| 8 | |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 9 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 10 | X86_CODE16 = b"\x8d\x4c\x32\x08\x01\xd8\x81\xc6\x34\x12\x00\x00" |
| 11 | X86_CODE32 = b"\x8d\x4c\x32\x08\x01\xd8\x81\xc6\x34\x12\x00\x00" |
| 12 | X86_CODE64 = b"\x55\x48\x8b\x05\xb8\x13\x00\x00" |
| 13 | ARM_CODE = b"\xED\xFF\xFF\xEB\x04\xe0\x2d\xe5\x00\x00\x00\x00\xe0\x83\x22\xe5\xf1\x02\x03\x0e\x00\x00\xa0\xe3\x02\x30\xc1\xe7\x00\x00\x53\xe3" |
| 14 | ARM_CODE2 = b"\x10\xf1\x10\xe7\x11\xf2\x31\xe7\xdc\xa1\x2e\xf3\xe8\x4e\x62\xf3" |
| 15 | THUMB_CODE = b"\x70\x47\xeb\x46\x83\xb0\xc9\x68" |
| 16 | THUMB_CODE2 = b"\x4f\xf0\x00\x01\xbd\xe8\x00\x88\xd1\xe8\x00\xf0" |
| 17 | MIPS_CODE = b"\x0C\x10\x00\x97\x00\x00\x00\x00\x24\x02\x00\x0c\x8f\xa2\x00\x00\x34\x21\x34\x56" |
| 18 | MIPS_CODE2 = b"\x56\x34\x21\x34\xc2\x17\x01\x00" |
| 19 | ARM64_CODE = b"\x21\x7c\x02\x9b\x21\x7c\x00\x53\x00\x40\x21\x4b\xe1\x0b\x40\xb9" |
| 20 | PPC_CODE = b"\x80\x20\x00\x00\x80\x3f\x00\x00\x10\x43\x23\x0e\xd0\x44\x00\x80\x4c\x43\x22\x02\x2d\x03\x00\x80\x7c\x43\x20\x14\x7c\x43\x20\x93\x4f\x20\x00\x21\x4c\xc8\x00\x21" |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 21 | |
| 22 | all_tests = ( |
| 23 | (CS_ARCH_X86, CS_MODE_16, X86_CODE16, "X86 16bit (Intel syntax)", 0), |
| 24 | (CS_ARCH_X86, CS_MODE_32, X86_CODE32, "X86 32bit (ATT syntax)", CS_OPT_SYNTAX_ATT), |
| 25 | (CS_ARCH_X86, CS_MODE_32, X86_CODE32, "X86 32 (Intel syntax)", 0), |
| 26 | (CS_ARCH_X86, CS_MODE_64, X86_CODE64, "X86 64 (Intel syntax)", 0), |
| 27 | (CS_ARCH_ARM, CS_MODE_ARM, ARM_CODE, "ARM", 0), |
| 28 | (CS_ARCH_ARM, CS_MODE_THUMB, THUMB_CODE2, "THUMB-2", 0), |
| 29 | (CS_ARCH_ARM, CS_MODE_ARM, ARM_CODE2, "ARM: Cortex-A15 + NEON", 0), |
| 30 | (CS_ARCH_ARM, CS_MODE_THUMB, THUMB_CODE, "THUMB", 0), |
Nguyen Anh Quynh | ec58a02 | 2014-11-13 11:42:38 +0800 | [diff] [blame^] | 31 | (CS_ARCH_MIPS, CS_MODE_MIPS32 + CS_MODE_BIG_ENDIAN, MIPS_CODE, "MIPS-32 (Big-endian)", 0), |
| 32 | (CS_ARCH_MIPS, CS_MODE_MIPS64 + CS_MODE_LITTLE_ENDIAN, MIPS_CODE2, "MIPS-64-EL (Little-endian)", 0), |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 33 | (CS_ARCH_ARM64, CS_MODE_ARM, ARM64_CODE, "ARM-64", 0), |
| 34 | (CS_ARCH_PPC, CS_MODE_BIG_ENDIAN, PPC_CODE, "PPC-64", 0), |
| 35 | (CS_ARCH_PPC, CS_MODE_BIG_ENDIAN, PPC_CODE, "PPC-64, print register with number only", CS_OPT_SYNTAX_NOREGNAME), |
| 36 | ) |
| 37 | |
| 38 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 39 | # ## Test cs_disasm_quick() |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 40 | def test_cs_disasm_quick(): |
Nguyen Anh Quynh | 27a35fe | 2014-02-25 17:07:25 +0800 | [diff] [blame] | 41 | for (arch, mode, code, comment, syntax) in all_tests: |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 42 | print('*' * 40) |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 43 | print("Platform: %s" % comment) |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 44 | print("Disasm:"), |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 45 | print(to_hex(code)) |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 46 | for (addr, size, mnemonic, op_str) in cs_disasm_lite(arch, mode, code, 0x1000): |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 47 | print("0x%x:\t%s\t%s" % (addr, mnemonic, op_str)) |
| 48 | print() |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 49 | |
| 50 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 51 | # ## Test class Cs |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 52 | def test_class(): |
| 53 | for (arch, mode, code, comment, syntax) in all_tests: |
| 54 | print('*' * 16) |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 55 | print("Platform: %s" % comment) |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 56 | print("Code: %s" % to_hex(code)) |
| 57 | print("Disasm:") |
| 58 | |
| 59 | try: |
| 60 | md = Cs(arch, mode) |
| 61 | |
| 62 | if syntax != 0: |
| 63 | md.syntax = syntax |
| 64 | |
| 65 | for (addr, size, mnemonic, op_str) in md.disasm_lite(code, 0x1000): |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 66 | print("0x%x:\t%s\t%s" % (addr, mnemonic, op_str)) |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 67 | |
| 68 | print("0x%x:" % (addr + size)) |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 69 | print() |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 70 | except CsError as e: |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 71 | print("ERROR: %s" % e) |
Nguyen Anh Quynh | 5d0428e | 2014-02-21 18:06:04 +0800 | [diff] [blame] | 72 | |
| 73 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 74 | # test_cs_disasm_quick() |
| 75 | # print "*" * 40 |
| 76 | if __name__ == '__main__': |
| 77 | test_class() |