Home | History | Annotate | Download | only in ocaml
      1 (* Capstone Disassembly Engine
      2 * By Nguyen Anh Quynh <aquynh (a] gmail.com>, 2013-2014 *)
      3 
      4 open Printf
      5 open Capstone
      6 open Arm
      7 open Arm_const
      8 
      9 
     10 let print_string_hex comment str =
     11 	printf "%s" comment;
     12 	for i = 0 to (Array.length str - 1) do
     13 		printf "0x%02x " str.(i)
     14 	done;
     15 	printf "\n"
     16 
     17 
     18 let _ARM_CODE = "\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";;
     19 let _ARM_CODE2 = "\xd1\xe8\x00\xf0\xf0\x24\x04\x07\x1f\x3c\xf2\xc0\x00\x00\x4f\xf0\x00\x01\x46\x6c";;
     20 let _THUMB_CODE2  = "\x4f\xf0\x00\x01\xbd\xe8\x00\x88\xd1\xe8\x00\xf0";;
     21 let _THUMB_CODE  = "\x70\x47\xeb\x46\x83\xb0\xc9\x68\x1f\xb1";;
     22 
     23 
     24 let all_tests = [
     25         (CS_ARCH_ARM, [CS_MODE_ARM], _ARM_CODE, "ARM");
     26         (CS_ARCH_ARM, [CS_MODE_THUMB], _THUMB_CODE, "Thumb");
     27         (CS_ARCH_ARM, [CS_MODE_THUMB], _ARM_CODE2, "Thumb-mixed");
     28         (CS_ARCH_ARM, [CS_MODE_THUMB], _THUMB_CODE2, "Thumb-2");
     29 ];;
     30 
     31 
     32 let print_op handle i op =
     33 	( match op.value with
     34 	| ARM_OP_INVALID _ -> ();	(* this would never happens *)
     35 	| ARM_OP_REG reg -> printf "\t\top[%d]: REG = %s\n" i (cs_reg_name handle reg);
     36 	| ARM_OP_CIMM imm -> printf "\t\top[%d]: C-IMM = %u\n" i imm;
     37 	| ARM_OP_PIMM imm -> printf "\t\top[%d]: P-IMM = %u\n" i imm;
     38 	| ARM_OP_IMM imm -> printf "\t\top[%d]: IMM = 0x%x\n" i imm;
     39 	| ARM_OP_FP fp -> printf "\t\top[%d]: FP = %f\n" i fp;
     40 	| ARM_OP_MEM mem -> ( printf "\t\top[%d]: MEM\n" i;
     41 		if mem.base != 0 then
     42 			printf "\t\t\toperands[%u].mem.base: REG = %s\n" i (cs_reg_name handle mem.base);
     43 		if mem.index != 0 then
     44 			printf "\t\t\toperands[%u].mem.index: REG = %s\n" i (cs_reg_name handle mem.index);
     45 		if mem.scale != 1 then
     46 			printf "\t\t\toperands[%u].mem.scale: %d\n" i mem.scale;
     47 		if mem.disp != 0 then
     48 			printf "\t\t\toperands[%u].mem.disp: 0x%x\n" i mem.disp;
     49 		);
     50 	| ARM_OP_SETEND sd -> printf "\t\top[%d]: SETEND = %u\n" i sd;
     51 	);
     52 
     53 	if op.shift.shift_type != _ARM_SFT_INVALID && op.shift.shift_value > 0 then
     54 		printf "\t\t\tShift: type = %u, value = %u\n"
     55                 op.shift.shift_type op.shift.shift_value;
     56 	();;
     57 
     58 
     59 let print_detail handle insn =
     60 	match insn.arch with
     61 	| CS_INFO_ARM arm -> (
     62 			if arm.cc != _ARM_CC_AL && arm.cc != _ARM_CC_INVALID then
     63 			printf "\tCode condition: %u\n" arm.cc;
     64 
     65 			if arm.update_flags then
     66 			printf "\tUpdate-flags: True\n";
     67 
     68 			if arm.writeback then
     69 			printf "\tWriteback: True\n";
     70 
     71 			(* print all operands info (type & value) *)
     72 			if (Array.length arm.operands) > 0 then (
     73 				printf "\top_count: %d\n" (Array.length arm.operands);
     74 				Array.iteri (print_op handle) arm.operands;
     75 			);
     76 			printf "\n";
     77 		);
     78 	| _ -> ();
     79 	;;
     80 
     81 
     82 let print_insn handle insn =
     83 	printf "0x%x\t%s\t%s\n" insn.address insn.mnemonic insn.op_str;
     84 	print_detail handle insn
     85 
     86 
     87 let print_arch x =
     88 	let (arch, mode, code, comment) = x in
     89 		let handle = cs_open arch mode in
     90 		let err = cs_option handle CS_OPT_DETAIL _CS_OPT_ON in
     91 		match err with
     92 		| _ -> ();
     93 		let insns = cs_disasm handle code 0x1000L 0L in
     94 			printf "*************\n";
     95 			printf "Platform: %s\n" comment;
     96 			List.iter (print_insn handle) insns;
     97 		match cs_close handle with
     98 		| 0 -> ();
     99 		| _ -> printf "Failed to close handle";
    100 		;;
    101 
    102 
    103 List.iter print_arch all_tests;;
    104