| /* Capstone Disassembler Engine */ |
| /* By Nguyen Anh Quynh <aquynh@gmail.com>, 2013 */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| |
| #include <capstone/platform.h> |
| #include <capstone/capstone.h> |
| |
| static csh handle; |
| |
| struct platform { |
| cs_arch arch; |
| cs_mode mode; |
| unsigned char *code; |
| size_t size; |
| const char *comment; |
| cs_opt_type opt_type; |
| cs_opt_value opt_value; |
| }; |
| |
| static void print_string_hex(const char *comment, unsigned char *str, size_t len) |
| { |
| unsigned char *c; |
| |
| printf("%s", comment); |
| for (c = str; c < str + len; c++) { |
| printf("0x%02x ", *c & 0xff); |
| } |
| |
| printf("\n"); |
| } |
| |
| static const char *get_eflag_name(uint64_t flag) |
| { |
| switch(flag) { |
| default: |
| return NULL; |
| case X86_EFLAGS_UNDEFINED_OF: |
| return "UNDEF_OF"; |
| case X86_EFLAGS_UNDEFINED_SF: |
| return "UNDEF_SF"; |
| case X86_EFLAGS_UNDEFINED_ZF: |
| return "UNDEF_ZF"; |
| case X86_EFLAGS_MODIFY_AF: |
| return "MOD_AF"; |
| case X86_EFLAGS_UNDEFINED_PF: |
| return "UNDEF_PF"; |
| case X86_EFLAGS_MODIFY_CF: |
| return "MOD_CF"; |
| case X86_EFLAGS_MODIFY_SF: |
| return "MOD_SF"; |
| case X86_EFLAGS_MODIFY_ZF: |
| return "MOD_ZF"; |
| case X86_EFLAGS_UNDEFINED_AF: |
| return "UNDEF_AF"; |
| case X86_EFLAGS_MODIFY_PF: |
| return "MOD_PF"; |
| case X86_EFLAGS_UNDEFINED_CF: |
| return "UNDEF_CF"; |
| case X86_EFLAGS_MODIFY_OF: |
| return "MOD_OF"; |
| case X86_EFLAGS_RESET_OF: |
| return "RESET_OF"; |
| case X86_EFLAGS_RESET_CF: |
| return "RESET_CF"; |
| case X86_EFLAGS_RESET_DF: |
| return "RESET_DF"; |
| case X86_EFLAGS_RESET_IF: |
| return "RESET_IF"; |
| case X86_EFLAGS_TEST_OF: |
| return "TEST_OF"; |
| case X86_EFLAGS_TEST_SF: |
| return "TEST_SF"; |
| case X86_EFLAGS_TEST_ZF: |
| return "TEST_ZF"; |
| case X86_EFLAGS_TEST_PF: |
| return "TEST_PF"; |
| case X86_EFLAGS_TEST_CF: |
| return "TEST_CF"; |
| case X86_EFLAGS_RESET_SF: |
| return "RESET_SF"; |
| case X86_EFLAGS_RESET_AF: |
| return "RESET_AF"; |
| case X86_EFLAGS_RESET_TF: |
| return "RESET_TF"; |
| case X86_EFLAGS_RESET_NT: |
| return "RESET_NT"; |
| case X86_EFLAGS_PRIOR_OF: |
| return "PRIOR_OF"; |
| case X86_EFLAGS_PRIOR_SF: |
| return "PRIOR_SF"; |
| case X86_EFLAGS_PRIOR_ZF: |
| return "PRIOR_ZF"; |
| case X86_EFLAGS_PRIOR_AF: |
| return "PRIOR_AF"; |
| case X86_EFLAGS_PRIOR_PF: |
| return "PRIOR_PF"; |
| case X86_EFLAGS_PRIOR_CF: |
| return "PRIOR_CF"; |
| case X86_EFLAGS_PRIOR_TF: |
| return "PRIOR_TF"; |
| case X86_EFLAGS_PRIOR_IF: |
| return "PRIOR_IF"; |
| case X86_EFLAGS_PRIOR_DF: |
| return "PRIOR_DF"; |
| case X86_EFLAGS_TEST_NT: |
| return "TEST_NT"; |
| case X86_EFLAGS_TEST_DF: |
| return "TEST_DF"; |
| case X86_EFLAGS_RESET_PF: |
| return "RESET_PF"; |
| case X86_EFLAGS_PRIOR_NT: |
| return "PRIOR_NT"; |
| case X86_EFLAGS_MODIFY_TF: |
| return "MOD_TF"; |
| case X86_EFLAGS_MODIFY_IF: |
| return "MOD_IF"; |
| case X86_EFLAGS_MODIFY_DF: |
| return "MOD_DF"; |
| case X86_EFLAGS_MODIFY_NT: |
| return "MOD_NT"; |
| case X86_EFLAGS_MODIFY_RF: |
| return "MOD_RF"; |
| case X86_EFLAGS_SET_CF: |
| return "SET_CF"; |
| case X86_EFLAGS_SET_DF: |
| return "SET_DF"; |
| case X86_EFLAGS_SET_IF: |
| return "SET_IF"; |
| } |
| } |
| |
| static void print_insn_detail(csh ud, cs_mode mode, cs_insn *ins) |
| { |
| int count, i; |
| cs_x86 *x86; |
| cs_regs regs_read, regs_write; |
| uint8_t regs_read_count, regs_write_count; |
| |
| // detail can be NULL on "data" instruction if SKIPDATA option is turned ON |
| if (ins->detail == NULL) |
| return; |
| |
| x86 = &(ins->detail->x86); |
| |
| print_string_hex("\tPrefix:", x86->prefix, 4); |
| |
| print_string_hex("\tOpcode:", x86->opcode, 4); |
| |
| printf("\trex: 0x%x\n", x86->rex); |
| |
| printf("\taddr_size: %u\n", x86->addr_size); |
| printf("\tmodrm: 0x%x\n", x86->modrm); |
| if (x86->encoding.modrm_offset != 0) { |
| printf("\tmodrm_offset: 0x%x\n", x86->encoding.modrm_offset); |
| } |
| |
| printf("\tdisp: 0x%" PRIx64 "\n", x86->disp); |
| if (x86->encoding.disp_offset != 0) { |
| printf("\tdisp_offset: 0x%x\n", x86->encoding.disp_offset); |
| } |
| |
| if (x86->encoding.disp_size != 0) { |
| printf("\tdisp_size: 0x%x\n", x86->encoding.disp_size); |
| } |
| |
| // SIB is not available in 16-bit mode |
| if ((mode & CS_MODE_16) == 0) { |
| printf("\tsib: 0x%x\n", x86->sib); |
| if (x86->sib_base != X86_REG_INVALID) |
| printf("\t\tsib_base: %s\n", cs_reg_name(handle, x86->sib_base)); |
| if (x86->sib_index != X86_REG_INVALID) |
| printf("\t\tsib_index: %s\n", cs_reg_name(handle, x86->sib_index)); |
| if (x86->sib_scale != 0) |
| printf("\t\tsib_scale: %d\n", x86->sib_scale); |
| } |
| |
| // XOP code condition |
| if (x86->xop_cc != X86_XOP_CC_INVALID) { |
| printf("\txop_cc: %u\n", x86->xop_cc); |
| } |
| |
| // SSE code condition |
| if (x86->sse_cc != X86_SSE_CC_INVALID) { |
| printf("\tsse_cc: %u\n", x86->sse_cc); |
| } |
| |
| // AVX code condition |
| if (x86->avx_cc != X86_AVX_CC_INVALID) { |
| printf("\tavx_cc: %u\n", x86->avx_cc); |
| } |
| |
| // AVX Suppress All Exception |
| if (x86->avx_sae) { |
| printf("\tavx_sae: %u\n", x86->avx_sae); |
| } |
| |
| // AVX Rounding Mode |
| if (x86->avx_rm != X86_AVX_RM_INVALID) { |
| printf("\tavx_rm: %u\n", x86->avx_rm); |
| } |
| |
| // Print out all immediate operands |
| count = cs_op_count(ud, ins, X86_OP_IMM); |
| if (count) { |
| printf("\timm_count: %u\n", count); |
| for (i = 1; i < count + 1; i++) { |
| int index = cs_op_index(ud, ins, X86_OP_IMM, i); |
| printf("\t\timms[%u]: 0x%" PRIx64 "\n", i, x86->operands[index].imm); |
| if (x86->encoding.imm_offset != 0) { |
| printf("\timm_offset: 0x%x\n", x86->encoding.imm_offset); |
| } |
| |
| if (x86->encoding.imm_size != 0) { |
| printf("\timm_size: 0x%x\n", x86->encoding.imm_size); |
| } |
| } |
| } |
| |
| if (x86->op_count) |
| printf("\top_count: %u\n", x86->op_count); |
| |
| // Print out all operands |
| for (i = 0; i < x86->op_count; i++) { |
| cs_x86_op *op = &(x86->operands[i]); |
| |
| switch((int)op->type) { |
| case X86_OP_REG: |
| printf("\t\toperands[%u].type: REG = %s\n", i, cs_reg_name(handle, op->reg)); |
| break; |
| case X86_OP_IMM: |
| printf("\t\toperands[%u].type: IMM = 0x%" PRIx64 "\n", i, op->imm); |
| break; |
| case X86_OP_MEM: |
| printf("\t\toperands[%u].type: MEM\n", i); |
| if (op->mem.segment != X86_REG_INVALID) |
| printf("\t\t\toperands[%u].mem.segment: REG = %s\n", i, cs_reg_name(handle, op->mem.segment)); |
| if (op->mem.base != X86_REG_INVALID) |
| printf("\t\t\toperands[%u].mem.base: REG = %s\n", i, cs_reg_name(handle, op->mem.base)); |
| if (op->mem.index != X86_REG_INVALID) |
| printf("\t\t\toperands[%u].mem.index: REG = %s\n", i, cs_reg_name(handle, op->mem.index)); |
| if (op->mem.scale != 1) |
| printf("\t\t\toperands[%u].mem.scale: %u\n", i, op->mem.scale); |
| if (op->mem.disp != 0) |
| printf("\t\t\toperands[%u].mem.disp: 0x%" PRIx64 "\n", i, op->mem.disp); |
| break; |
| default: |
| break; |
| } |
| |
| // AVX broadcast type |
| if (op->avx_bcast != X86_AVX_BCAST_INVALID) |
| printf("\t\toperands[%u].avx_bcast: %u\n", i, op->avx_bcast); |
| |
| // AVX zero opmask {z} |
| if (op->avx_zero_opmask != false) |
| printf("\t\toperands[%u].avx_zero_opmask: TRUE\n", i); |
| |
| printf("\t\toperands[%u].size: %u\n", i, op->size); |
| |
| switch(op->access) { |
| default: |
| break; |
| case CS_AC_READ: |
| printf("\t\toperands[%u].access: READ\n", i); |
| break; |
| case CS_AC_WRITE: |
| printf("\t\toperands[%u].access: WRITE\n", i); |
| break; |
| case CS_AC_READ | CS_AC_WRITE: |
| printf("\t\toperands[%u].access: READ | WRITE\n", i); |
| break; |
| } |
| } |
| |
| // Print out all registers accessed by this instruction (either implicit or explicit) |
| if (!cs_regs_access(ud, ins, |
| regs_read, ®s_read_count, |
| regs_write, ®s_write_count)) { |
| if (regs_read_count) { |
| printf("\tRegisters read:"); |
| for(i = 0; i < regs_read_count; i++) { |
| printf(" %s", cs_reg_name(handle, regs_read[i])); |
| } |
| printf("\n"); |
| } |
| |
| if (regs_write_count) { |
| printf("\tRegisters modified:"); |
| for(i = 0; i < regs_write_count; i++) { |
| printf(" %s", cs_reg_name(handle, regs_write[i])); |
| } |
| printf("\n"); |
| } |
| } |
| |
| if (x86->eflags) { |
| printf("\tEFLAGS:"); |
| for(i = 0; i <= 45; i++) |
| if (x86->eflags & ((uint64_t)1 << i)) { |
| printf(" %s", get_eflag_name((uint64_t)1 << i)); |
| } |
| printf("\n"); |
| } |
| |
| printf("\n"); |
| } |
| |
| static void test() |
| { |
| //#define X86_CODE32 "\x01\xd8\x81\xc6\x34\x12\x00\x00\x05\x78\x56\x00\x00" |
| //#define X86_CODE32 "\x05\x78\x56\x00\x00" |
| //#define X86_CODE32 "\x01\xd8" |
| //#define X86_CODE32 "\x05\x23\x01\x00\x00" |
| //#define X86_CODE32 "\x8d\x87\x89\x67\x00\x00" |
| //#define X86_CODE32 "\xa1\x13\x48\x6d\x3a\x8b\x81\x23\x01\x00\x00\x8b\x84\x39\x23\x01\x00\x00" |
| //#define X86_CODE32 "\xb4\xc6" // mov ah, 0x6c |
| //#define X86_CODE32 "\x77\x04" // ja +6 |
| #define X86_CODE64 "\x55\x48\x8b\x05\xb8\x13\x00\x00\xe9\xea\xbe\xad\xde\xff\x25\x23\x01\x00\x00\xe8\xdf\xbe\xad\xde\x74\xff" |
| //#define X86_CODE64 "\xe9\x79\xff\xff\xff" // jmp 0xf7e |
| |
| #define X86_CODE16 "\x8d\x4c\x32\x08\x01\xd8\x81\xc6\x34\x12\x00\x00\x05\x23\x01\x00\x00\x36\x8b\x84\x91\x23\x01\x00\x00\x41\x8d\x84\x39\x89\x67\x00\x00\x8d\x87\x89\x67\x00\x00\xb4\xc6\x66\xe9\xb8\x00\x00\x00\x67\xff\xa0\x23\x01\x00\x00\x66\xe8\xcb\x00\x00\x00\x74\xfc" |
| #define X86_CODE32 "\x8d\x4c\x32\x08\x01\xd8\x81\xc6\x34\x12\x00\x00\x05\x23\x01\x00\x00\x36\x8b\x84\x91\x23\x01\x00\x00\x41\x8d\x84\x39\x89\x67\x00\x00\x8d\x87\x89\x67\x00\x00\xb4\xc6\xe9\xea\xbe\xad\xde\xff\xa0\x23\x01\x00\x00\xe8\xdf\xbe\xad\xde\x74\xff" |
| //#define X86_CODE32 "\x05\x23\x01\x00\x00\x0f\x01\xda" |
| //#define X86_CODE32 "\x0f\xa7\xc0" // xstorerng |
| //#define X86_CODE32 "\x64\xa1\x18\x00\x00\x00" // mov eax, dword ptr fs:[18] |
| //#define X86_CODE32 "\x64\xa3\x00\x00\x00\x00" // mov [fs:0x0], eax |
| //#define X86_CODE32 "\xd1\xe1" // shl ecx, 1 |
| //#define X86_CODE32 "\xd1\xc8" // ror eax, 1 |
| //#define X86_CODE32 "\x83\xC0\x80" // add eax, -x80 |
| //#define X86_CODE32 "\xe8\x26\xfe\xff\xff" // call 0xe2b |
| //#define X86_CODE32 "\xcd\x80" // int 0x80 |
| //#define X86_CODE32 "\x24\xb8" // and $0xb8,%al |
| //#define X86_CODE32 "\xf0\x01\xd8" // lock add eax,ebx |
| //#define X86_CODE32 "\xf3\xaa" // rep stosb |
| //#define X86_CODE32 "\x81\xc6\x23\x01\x00\x00" |
| |
| struct platform platforms[] = { |
| { |
| CS_ARCH_X86, |
| CS_MODE_16, |
| (unsigned char *)X86_CODE16, |
| sizeof(X86_CODE16) - 1, |
| "X86 16bit (Intel syntax)" |
| }, |
| { |
| CS_ARCH_X86, |
| CS_MODE_32, |
| (unsigned char *)X86_CODE32, |
| sizeof(X86_CODE32) - 1, |
| "X86 32 (AT&T syntax)", |
| CS_OPT_SYNTAX, |
| CS_OPT_SYNTAX_ATT, |
| }, |
| { |
| CS_ARCH_X86, |
| CS_MODE_32, |
| (unsigned char *)X86_CODE32, |
| sizeof(X86_CODE32) - 1, |
| "X86 32 (Intel syntax)" |
| }, |
| { |
| CS_ARCH_X86, |
| CS_MODE_64, |
| (unsigned char *)X86_CODE64, |
| sizeof(X86_CODE64) - 1, |
| "X86 64 (Intel syntax)" |
| }, |
| }; |
| |
| uint64_t address = 0x1000; |
| cs_insn *insn; |
| int i; |
| size_t count; |
| |
| for (i = 0; i < sizeof(platforms)/sizeof(platforms[0]); i++) { |
| cs_err err = cs_open(platforms[i].arch, platforms[i].mode, &handle); |
| if (err) { |
| printf("Failed on cs_open() with error returned: %u\n", err); |
| abort(); |
| } |
| |
| if (platforms[i].opt_type) |
| cs_option(handle, platforms[i].opt_type, platforms[i].opt_value); |
| |
| cs_option(handle, CS_OPT_DETAIL, CS_OPT_ON); |
| |
| count = cs_disasm(handle, platforms[i].code, platforms[i].size, address, 0, &insn); |
| if (count) { |
| size_t j; |
| |
| printf("****************\n"); |
| printf("Platform: %s\n", platforms[i].comment); |
| print_string_hex("Code:", platforms[i].code, platforms[i].size); |
| printf("Disasm:\n"); |
| |
| for (j = 0; j < count; j++) { |
| printf("0x%" PRIx64 ":\t%s\t%s\n\n", insn[j].address, insn[j].mnemonic, insn[j].op_str); |
| print_insn_detail(handle, platforms[i].mode, &insn[j]); |
| } |
| printf("0x%" PRIx64 ":\n", insn[j-1].address + insn[j-1].size); |
| |
| // free memory allocated by cs_disasm() |
| cs_free(insn, count); |
| } else { |
| printf("****************\n"); |
| printf("Platform: %s\n", platforms[i].comment); |
| print_string_hex("Code:", platforms[i].code, platforms[i].size); |
| printf("ERROR: Failed to disasm given code!\n"); |
| abort(); |
| } |
| |
| printf("\n"); |
| |
| cs_close(&handle); |
| } |
| } |
| |
| int main() |
| { |
| test(); |
| |
| return 0; |
| } |