Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 1 | #!/usr/bin/env python |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 2 | # Capstone Python bindings, by Nguyen Anh Quynnh <aquynh@gmail.com> |
| 3 | |
Nguyen Anh Quynh | 1098329 | 2014-05-17 09:51:15 +0800 | [diff] [blame] | 4 | from __future__ import print_function |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 5 | from capstone import * |
Joxean | 114df0e | 2013-12-04 07:11:32 +0100 | [diff] [blame] | 6 | import binascii |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 7 | import sys |
Nguyen Anh Quynh | 1098329 | 2014-05-17 09:51:15 +0800 | [diff] [blame] | 8 | |
Nguyen Anh Quynh | 03d1e1f | 2015-04-27 11:51:48 +0800 | [diff] [blame] | 9 | from xprint import to_hex |
Nguyen Anh Quynh | 1098329 | 2014-05-17 09:51:15 +0800 | [diff] [blame] | 10 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 11 | _python3 = sys.version_info.major == 3 |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 12 | |
Nguyen Anh Quynh | 749046b | 2014-04-12 01:15:10 +0800 | [diff] [blame] | 13 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 14 | X86_CODE16 = b"\x8d\x4c\x32\x08\x01\xd8\x81\xc6\x34\x12\x00\x00" |
Nguyen Anh Quynh | 4d7d5c1 | 2015-08-09 10:52:43 -0700 | [diff] [blame] | 15 | X86_CODE32 = b"\xba\xcd\xab\x00\x00\x8d\x4c\x32\x08\x01\xd8\x81\xc6\x34\x12\x00\x00" |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 16 | X86_CODE64 = b"\x55\x48\x8b\x05\xb8\x13\x00\x00" |
| 17 | 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" |
| 18 | ARM_CODE2 = b"\x10\xf1\x10\xe7\x11\xf2\x31\xe7\xdc\xa1\x2e\xf3\xe8\x4e\x62\xf3" |
| 19 | THUMB_CODE = b"\x70\x47\xeb\x46\x83\xb0\xc9\x68" |
| 20 | THUMB_CODE2 = b"\x4f\xf0\x00\x01\xbd\xe8\x00\x88\xd1\xe8\x00\xf0" |
Nguyen Anh Quynh | 2e40e69 | 2014-11-11 21:54:22 +0800 | [diff] [blame] | 21 | THUMB_MCLASS = b"\xef\xf3\x02\x80" |
| 22 | ARMV8 = b"\xe0\x3b\xb2\xee\x42\x00\x01\xe1\x51\xf0\x7f\xf5" |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 23 | MIPS_CODE = b"\x0C\x10\x00\x97\x00\x00\x00\x00\x24\x02\x00\x0c\x8f\xa2\x00\x00\x34\x21\x34\x56" |
| 24 | MIPS_CODE2 = b"\x56\x34\x21\x34\xc2\x17\x01\x00" |
Nguyen Anh Quynh | 248519e | 2014-11-09 14:07:07 +0800 | [diff] [blame] | 25 | MIPS_32R6M = b"\x00\x07\x00\x07\x00\x11\x93\x7c\x01\x8c\x8b\x7c\x00\xc7\x48\xd0" |
| 26 | MIPS_32R6 = b"\xec\x80\x00\x19\x7c\x43\x22\xa0" |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 27 | ARM64_CODE = b"\x21\x7c\x02\x9b\x21\x7c\x00\x53\x00\x40\x21\x4b\xe1\x0b\x40\xb9" |
| 28 | 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 | cac770a | 2015-03-12 17:03:33 +0800 | [diff] [blame] | 29 | PPC_CODE2 = b"\x10\x60\x2a\x10\x10\x64\x28\x88\x7c\x4a\x5d\x0f" |
Nguyen Anh Quynh | 749046b | 2014-04-12 01:15:10 +0800 | [diff] [blame] | 30 | SPARC_CODE = b"\x80\xa0\x40\x02\x85\xc2\x60\x08\x85\xe8\x20\x01\x81\xe8\x00\x00\x90\x10\x20\x01\xd5\xf6\x10\x16\x21\x00\x00\x0a\x86\x00\x40\x02\x01\x00\x00\x00\x12\xbf\xff\xff\x10\xbf\xff\xff\xa0\x02\x00\x09\x0d\xbf\xff\xff\xd4\x20\x60\x00\xd4\x4e\x00\x16\x2a\xc2\x80\x03" |
| 31 | SPARCV9_CODE = b"\x81\xa8\x0a\x24\x89\xa0\x10\x20\x89\xa0\x1a\x60\x89\xa0\x00\xe0" |
| 32 | SYSZ_CODE = b"\xed\x00\x00\x00\x00\x1a\x5a\x0f\x1f\xff\xc2\x09\x80\x00\x00\x00\x07\xf7\xeb\x2a\xff\xff\x7f\x57\xe3\x01\xff\xff\x7f\x57\xeb\x00\xf0\x00\x00\x24\xb2\x4f\x00\x78" |
Nguyen Anh Quynh | 553bb48 | 2014-05-26 23:47:04 +0800 | [diff] [blame] | 33 | XCORE_CODE = b"\xfe\x0f\xfe\x17\x13\x17\xc6\xfe\xec\x17\x97\xf8\xec\x4f\x1f\xfd\xec\x37\x07\xf2\x45\x5b\xf9\xfa\x02\x06\x1b\x10" |
Niels Boehm | 0e407ea | 2016-06-14 13:07:57 +0200 | [diff] [blame] | 34 | M68K_CODE = b"\xd4\x40\x87\x5a\x4e\x71\x02\xb4\xc0\xde\xc0\xde\x5c\x00\x1d\x80\x71\x12\x01\x23\xf2\x3c\x44\x22\x40\x49\x0e\x56\x54\xc5\xf2\x3c\x44\x00\x44\x7a\x00\x00\xf2\x00\x0a\x28\x4E\xB9\x00\x00\x00\x12\x4E\x75" |
Fotis Loukos | 0a159f1 | 2016-05-03 16:37:52 +0300 | [diff] [blame] | 35 | TMS320C64X_CODE = b"\x01\xac\x88\x40\x81\xac\x88\x43\x00\x00\x00\x00\x02\x90\x32\x96\x02\x80\x46\x9e\x05\x3c\x83\xe6\x0b\x0c\x8b\x24" |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 36 | |
| 37 | all_tests = ( |
Niels Boehm | 32b6346 | 2016-06-13 12:25:24 +0200 | [diff] [blame] | 38 | (CS_ARCH_X86, CS_MODE_16, X86_CODE16, "X86 16bit (Intel syntax)", None), |
Nguyen Anh Quynh | 4649071 | 2013-12-06 00:44:44 +0800 | [diff] [blame] | 39 | (CS_ARCH_X86, CS_MODE_32, X86_CODE32, "X86 32bit (ATT syntax)", CS_OPT_SYNTAX_ATT), |
Niels Boehm | 32b6346 | 2016-06-13 12:25:24 +0200 | [diff] [blame] | 40 | (CS_ARCH_X86, CS_MODE_32, X86_CODE32, "X86 32 (Intel syntax)", None), |
Nguyen Anh Quynh | 4d7d5c1 | 2015-08-09 10:52:43 -0700 | [diff] [blame] | 41 | (CS_ARCH_X86, CS_MODE_32, X86_CODE32, "X86 32 (MASM syntax)", CS_OPT_SYNTAX_MASM), |
Niels Boehm | 32b6346 | 2016-06-13 12:25:24 +0200 | [diff] [blame] | 42 | (CS_ARCH_X86, CS_MODE_64, X86_CODE64, "X86 64 (Intel syntax)", None), |
| 43 | (CS_ARCH_ARM, CS_MODE_ARM, ARM_CODE, "ARM", None), |
| 44 | (CS_ARCH_ARM, CS_MODE_THUMB, THUMB_CODE2, "THUMB-2", None), |
| 45 | (CS_ARCH_ARM, CS_MODE_ARM, ARM_CODE2, "ARM: Cortex-A15 + NEON", None), |
| 46 | (CS_ARCH_ARM, CS_MODE_THUMB, THUMB_CODE, "THUMB", None), |
| 47 | (CS_ARCH_ARM, CS_MODE_THUMB + CS_MODE_MCLASS, THUMB_MCLASS, "Thumb-MClass", None), |
| 48 | (CS_ARCH_ARM, CS_MODE_ARM + CS_MODE_V8, ARMV8, "Arm-V8", None), |
| 49 | (CS_ARCH_MIPS, CS_MODE_MIPS32 + CS_MODE_BIG_ENDIAN, MIPS_CODE, "MIPS-32 (Big-endian)", None), |
| 50 | (CS_ARCH_MIPS, CS_MODE_MIPS64 + CS_MODE_LITTLE_ENDIAN, MIPS_CODE2, "MIPS-64-EL (Little-endian)", None), |
| 51 | (CS_ARCH_MIPS, CS_MODE_MIPS32R6 + CS_MODE_MICRO + CS_MODE_BIG_ENDIAN, MIPS_32R6M, "MIPS-32R6 | Micro (Big-endian)", None), |
| 52 | (CS_ARCH_MIPS, CS_MODE_MIPS32R6 + CS_MODE_BIG_ENDIAN, MIPS_32R6, "MIPS-32R6 (Big-endian)", None), |
| 53 | (CS_ARCH_ARM64, CS_MODE_ARM, ARM64_CODE, "ARM-64", None), |
| 54 | (CS_ARCH_PPC, CS_MODE_BIG_ENDIAN, PPC_CODE, "PPC-64", None), |
Nguyen Anh Quynh | fc4bc12 | 2014-01-08 10:21:04 +0800 | [diff] [blame] | 55 | (CS_ARCH_PPC, CS_MODE_BIG_ENDIAN, PPC_CODE, "PPC-64, print register with number only", CS_OPT_SYNTAX_NOREGNAME), |
Nguyen Anh Quynh | cac770a | 2015-03-12 17:03:33 +0800 | [diff] [blame] | 56 | (CS_ARCH_PPC, CS_MODE_BIG_ENDIAN + CS_MODE_QPX, PPC_CODE2, "PPC-64 + QPX", CS_OPT_SYNTAX_NOREGNAME), |
Niels Boehm | 32b6346 | 2016-06-13 12:25:24 +0200 | [diff] [blame] | 57 | (CS_ARCH_SPARC, CS_MODE_BIG_ENDIAN, SPARC_CODE, "Sparc", None), |
| 58 | (CS_ARCH_SPARC, CS_MODE_BIG_ENDIAN + CS_MODE_V9, SPARCV9_CODE, "SparcV9", None), |
| 59 | (CS_ARCH_SYSZ, 0, SYSZ_CODE, "SystemZ", None), |
| 60 | (CS_ARCH_XCORE, 0, XCORE_CODE, "XCore", None), |
Niels Boehm | 0e407ea | 2016-06-14 13:07:57 +0200 | [diff] [blame] | 61 | (CS_ARCH_M68K, CS_MODE_BIG_ENDIAN | CS_MODE_M68K_040, M68K_CODE, "M68K (68040)", None), |
Fotis Loukos | 0a159f1 | 2016-05-03 16:37:52 +0300 | [diff] [blame] | 62 | (CS_ARCH_TMS320C64X, 0, TMS320C64X_CODE, "TMS320C64x", None), |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 63 | ) |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 64 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 65 | # ## Test cs_disasm_quick() |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 66 | def test_cs_disasm_quick(): |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 67 | for arch, mode, code, comment, syntax in all_tests: |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 68 | print('*' * 40) |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 69 | print("Platform: %s" % comment) |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 70 | print("Disasm:"), |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 71 | print(to_hex(code)) |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 72 | for insn in cs_disasm_quick(arch, mode, code, 0x1000): |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 73 | print("0x%x:\t%s\t%s" % (insn.address, insn.mnemonic, insn.op_str)) |
| 74 | print() |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 75 | |
| 76 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 77 | # ## Test class Cs |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 78 | def test_class(): |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 79 | for arch, mode, code, comment, syntax in all_tests: |
danghvu | 1a7c449 | 2013-11-27 22:51:11 -0600 | [diff] [blame] | 80 | print('*' * 16) |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 81 | print("Platform: %s" % comment) |
danghvu | 1a7c449 | 2013-11-27 22:51:11 -0600 | [diff] [blame] | 82 | print("Code: %s" % to_hex(code)) |
| 83 | print("Disasm:") |
| 84 | |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 85 | try: |
Nguyen Anh Quynh | 520c361 | 2013-12-06 15:26:07 +0800 | [diff] [blame] | 86 | md = Cs(arch, mode) |
Nguyen Anh Quynh | daaed13 | 2013-12-03 22:18:28 +0800 | [diff] [blame] | 87 | |
Niels Boehm | 32b6346 | 2016-06-13 12:25:24 +0200 | [diff] [blame] | 88 | if syntax is not None: |
Nguyen Anh Quynh | 4649071 | 2013-12-06 00:44:44 +0800 | [diff] [blame] | 89 | md.syntax = syntax |
Nguyen Anh Quynh | daaed13 | 2013-12-03 22:18:28 +0800 | [diff] [blame] | 90 | |
Nguyen Anh Quynh | e099ede | 2013-12-06 18:06:11 +0800 | [diff] [blame] | 91 | for insn in md.disasm(code, 0x1000): |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 92 | # bytes = binascii.hexlify(insn.bytes) |
| 93 | # print("0x%x:\t%s\t%s\t// hex-code: %s" %(insn.address, insn.mnemonic, insn.op_str, bytes)) |
| 94 | print("0x%x:\t%s\t%s" % (insn.address, insn.mnemonic, insn.op_str)) |
danghvu | 1a7c449 | 2013-11-27 22:51:11 -0600 | [diff] [blame] | 95 | |
Nguyen Anh Quynh | e099ede | 2013-12-06 18:06:11 +0800 | [diff] [blame] | 96 | print("0x%x:" % (insn.address + insn.size)) |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 97 | print() |
Nguyen Anh Quynh | f1618bc | 2013-12-06 20:58:04 +0800 | [diff] [blame] | 98 | except CsError as e: |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 99 | print("ERROR: %s" % e) |
Nguyen Anh Quynh | 26ee41a | 2013-11-27 12:11:31 +0800 | [diff] [blame] | 100 | |
| 101 | |
fenuks | 110ab1d | 2014-04-11 11:00:33 +0200 | [diff] [blame] | 102 | # test_cs_disasm_quick() |
| 103 | # print ("*" * 40) |
| 104 | if __name__ == '__main__': |
| 105 | test_class() |