tests/test_arm64.c - third_party/capstone - Git at Google

 /* Capstone Disassembler Engine */
 /* By Nguyen Anh Quynh <aquynh@gmail.com>, 2013-2019 */

 #include <assert.h>
 #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;
 };

 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 void print_insn_detail(cs_insn *ins)
 {
 	cs_arm64 *arm64;
 	int i;
 	cs_regs regs_read, regs_write;
 	unsigned char regs_read_count, regs_write_count;
 	unsigned char access;

 	// detail can be NULL if SKIPDATA option is turned ON
 	if (ins->detail == NULL)
 		return;

 	arm64 = &(ins->detail->arm64);
 	if (arm64->op_count)
 		printf("\top_count: %u\n", arm64->op_count);

 	for (i = 0; i < arm64->op_count; i++) {
 		cs_arm64_op *op = &(arm64->operands[i]);
 		switch(op->type) {
 			default:
 				break;
 			case ARM64_OP_REG:
 				printf("\t\toperands[%u].type: REG = %s\n", i, cs_reg_name(handle, op->reg));
 				break;
 			case ARM64_OP_IMM:
 				printf("\t\toperands[%u].type: IMM = 0x%" PRIx64 "\n", i, op->imm);
 				break;
 			case ARM64_OP_FP:
 #if defined(_KERNEL_MODE)
 				// Issue #681: Windows kernel does not support formatting float point
 				printf("\t\toperands[%u].type: FP = <float_point_unsupported>\n", i);
 #else
 				printf("\t\toperands[%u].type: FP = %f\n", i, op->fp);
 #endif
 				break;
 			case ARM64_OP_MEM:
 				printf("\t\toperands[%u].type: MEM\n", i);
 				if (op->mem.base != ARM64_REG_INVALID)
 					printf("\t\t\toperands[%u].mem.base: REG = %s\n", i, cs_reg_name(handle, op->mem.base));
 				if (op->mem.index != ARM64_REG_INVALID)
 					printf("\t\t\toperands[%u].mem.index: REG = %s\n", i, cs_reg_name(handle, op->mem.index));
 				if (op->mem.disp != 0)
 					printf("\t\t\toperands[%u].mem.disp: 0x%x\n", i, op->mem.disp);

 				break;
 			case ARM64_OP_CIMM:
 				printf("\t\toperands[%u].type: C-IMM = %u\n", i, (int)op->imm);
 				break;
 			case ARM64_OP_REG_MRS:
 				printf("\t\toperands[%u].type: REG_MRS = 0x%x\n", i, op->reg);
 				break;
 			case ARM64_OP_REG_MSR:
 				printf("\t\toperands[%u].type: REG_MSR = 0x%x\n", i, op->reg);
 				break;
 			case ARM64_OP_PSTATE:
 				printf("\t\toperands[%u].type: PSTATE = 0x%x\n", i, op->pstate);
 				break;
 			case ARM64_OP_SYS:
 				printf("\t\toperands[%u].type: SYS = 0x%x\n", i, op->sys);
 				break;
 			case ARM64_OP_PREFETCH:
 				printf("\t\toperands[%u].type: PREFETCH = 0x%x\n", i, op->prefetch);
 				break;
 			case ARM64_OP_BARRIER:
 				printf("\t\toperands[%u].type: BARRIER = 0x%x\n", i, op->barrier);
 				break;
 		}

 		access = op->access;
 		switch(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;
 		}

 		if (op->shift.type != ARM64_SFT_INVALID &&
 				op->shift.value)
 			printf("\t\t\tShift: type = %u, value = %u\n",
 					op->shift.type, op->shift.value);

 		if (op->ext != ARM64_EXT_INVALID)
 			printf("\t\t\tExt: %u\n", op->ext);

 		if (op->vas != ARM64_VAS_INVALID)
 			printf("\t\t\tVector Arrangement Specifier: 0x%x\n", op->vas);

 		if (op->vess != ARM64_VESS_INVALID)
 			printf("\t\t\tVector Element Size Specifier: %u\n", op->vess);

 		if (op->vector_index != -1)
 			printf("\t\t\tVector Index: %u\n", op->vector_index);
 	}

 	if (arm64->update_flags)
 		printf("\tUpdate-flags: True\n");

 	if (arm64->writeback)
 		printf("\tWrite-back: True\n");

 	if (arm64->cc)
 		printf("\tCode-condition: %u\n", arm64->cc);

 	// Print out all registers accessed by this instruction (either implicit or explicit)
 	if (!cs_regs_access(handle, ins,
 				regs_read, &regs_read_count,
 				regs_write, &regs_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");
 		}
 	}

 	printf("\n");
 }

 static void test()
 {
 #define ARM64_CODE "\x09\x00\x38\xd5\xbf\x40\x00\xd5\x0c\x05\x13\xd5\x20\x50\x02\x0e\x20\xe4\x3d\x0f\x00\x18\xa0\x5f\xa2\x00\xae\x9e\x9f\x37\x03\xd5\xbf\x33\x03\xd5\xdf\x3f\x03\xd5\x21\x7c\x02\x9b\x21\x7c\x00\x53\x00\x40\x21\x4b\xe1\x0b\x40\xb9\x20\x04\x81\xda\x20\x08\x02\x8b\x10\x5b\xe8\x3c"

 	struct platform platforms[] = {
 		{
 			CS_ARCH_ARM64,
 			CS_MODE_ARM,
 			(unsigned char *)ARM64_CODE,
 			sizeof(ARM64_CODE) - 1,
 			"ARM-64"
 		},
 	};

 	uint64_t address = 0x2c;
 	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();
 		}

 		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", insn[j].address, insn[j].mnemonic, insn[j].op_str);
 				print_insn_detail(&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);
 	}
 }

 void test_macros() {
 	assert(CS_AARCH64(_INS_BL) == ARM64_INS_BL);
 	assert(CS_AARCH64pre(CS_ARCH_) == CS_ARCH_ARM64);
 	assert(CS_AARCH64CC(_AL) == ARM64_CC_AL);
 	assert(CS_AARCH64_VL_(16B) == ARM64_VAS_16B);
 	cs_detail detail = { 0 };
 	CS_cs_aarch64() arm64_detail = { 0 };
 	detail.arm64 = arm64_detail;
 	CS_aarch64_op() op = { 0 };
 	detail.CS_aarch64_.operands[0] = op;
 	CS_aarch64_reg() reg = 1;
 	CS_aarch64_cc() cc = ARM64_CC_AL;
 	CS_aarch64_extender() arm64_extender = ARM64_EXT_SXTB;
 	CS_aarch64_shifter() arm64_shifter = ARM64_SFT_LSL;
 	CS_aarch64_vas() arm64_vas = ARM64_VAS_16B;
 }

 int main()
 {
 	test();
 	test_macros();

 	return 0;
 }
	/* Capstone Disassembler Engine */
	/* By Nguyen Anh Quynh <aquynh@gmail.com>, 2013-2019 */

	#include <assert.h>
	#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;
	};

	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 void print_insn_detail(cs_insn *ins)
	{
	cs_arm64 *arm64;
	int i;
	cs_regs regs_read, regs_write;
	unsigned char regs_read_count, regs_write_count;
	unsigned char access;

	// detail can be NULL if SKIPDATA option is turned ON
	if (ins->detail == NULL)
	return;

	arm64 = &(ins->detail->arm64);
	if (arm64->op_count)
	printf("\top_count: %u\n", arm64->op_count);

	for (i = 0; i < arm64->op_count; i++) {
	cs_arm64_op *op = &(arm64->operands[i]);
	switch(op->type) {
	default:
	break;
	case ARM64_OP_REG:
	printf("\t\toperands[%u].type: REG = %s\n", i, cs_reg_name(handle, op->reg));
	break;
	case ARM64_OP_IMM:
	printf("\t\toperands[%u].type: IMM = 0x%" PRIx64 "\n", i, op->imm);
	break;
	case ARM64_OP_FP:
	#if defined(_KERNEL_MODE)
	// Issue #681: Windows kernel does not support formatting float point
	printf("\t\toperands[%u].type: FP = <float_point_unsupported>\n", i);
	#else
	printf("\t\toperands[%u].type: FP = %f\n", i, op->fp);
	#endif
	break;
	case ARM64_OP_MEM:
	printf("\t\toperands[%u].type: MEM\n", i);
	if (op->mem.base != ARM64_REG_INVALID)
	printf("\t\t\toperands[%u].mem.base: REG = %s\n", i, cs_reg_name(handle, op->mem.base));
	if (op->mem.index != ARM64_REG_INVALID)
	printf("\t\t\toperands[%u].mem.index: REG = %s\n", i, cs_reg_name(handle, op->mem.index));
	if (op->mem.disp != 0)
	printf("\t\t\toperands[%u].mem.disp: 0x%x\n", i, op->mem.disp);

	break;
	case ARM64_OP_CIMM:
	printf("\t\toperands[%u].type: C-IMM = %u\n", i, (int)op->imm);
	break;
	case ARM64_OP_REG_MRS:
	printf("\t\toperands[%u].type: REG_MRS = 0x%x\n", i, op->reg);
	break;
	case ARM64_OP_REG_MSR:
	printf("\t\toperands[%u].type: REG_MSR = 0x%x\n", i, op->reg);
	break;
	case ARM64_OP_PSTATE:
	printf("\t\toperands[%u].type: PSTATE = 0x%x\n", i, op->pstate);
	break;
	case ARM64_OP_SYS:
	printf("\t\toperands[%u].type: SYS = 0x%x\n", i, op->sys);
	break;
	case ARM64_OP_PREFETCH:
	printf("\t\toperands[%u].type: PREFETCH = 0x%x\n", i, op->prefetch);
	break;
	case ARM64_OP_BARRIER:
	printf("\t\toperands[%u].type: BARRIER = 0x%x\n", i, op->barrier);
	break;
	}

	access = op->access;
	switch(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;
	}

	if (op->shift.type != ARM64_SFT_INVALID &&
	op->shift.value)
	printf("\t\t\tShift: type = %u, value = %u\n",
	op->shift.type, op->shift.value);

	if (op->ext != ARM64_EXT_INVALID)
	printf("\t\t\tExt: %u\n", op->ext);

	if (op->vas != ARM64_VAS_INVALID)
	printf("\t\t\tVector Arrangement Specifier: 0x%x\n", op->vas);

	if (op->vess != ARM64_VESS_INVALID)
	printf("\t\t\tVector Element Size Specifier: %u\n", op->vess);

	if (op->vector_index != -1)
	printf("\t\t\tVector Index: %u\n", op->vector_index);
	}

	if (arm64->update_flags)
	printf("\tUpdate-flags: True\n");

	if (arm64->writeback)
	printf("\tWrite-back: True\n");

	if (arm64->cc)
	printf("\tCode-condition: %u\n", arm64->cc);

	// Print out all registers accessed by this instruction (either implicit or explicit)
	if (!cs_regs_access(handle, ins,
	regs_read, &regs_read_count,
	regs_write, &regs_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");
	}
	}

	printf("\n");
	}

	static void test()
	{
	#define ARM64_CODE "\x09\x00\x38\xd5\xbf\x40\x00\xd5\x0c\x05\x13\xd5\x20\x50\x02\x0e\x20\xe4\x3d\x0f\x00\x18\xa0\x5f\xa2\x00\xae\x9e\x9f\x37\x03\xd5\xbf\x33\x03\xd5\xdf\x3f\x03\xd5\x21\x7c\x02\x9b\x21\x7c\x00\x53\x00\x40\x21\x4b\xe1\x0b\x40\xb9\x20\x04\x81\xda\x20\x08\x02\x8b\x10\x5b\xe8\x3c"

	struct platform platforms[] = {
	{
	CS_ARCH_ARM64,
	CS_MODE_ARM,
	(unsigned char *)ARM64_CODE,
	sizeof(ARM64_CODE) - 1,
	"ARM-64"
	},
	};

	uint64_t address = 0x2c;
	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();
	}

	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", insn[j].address, insn[j].mnemonic, insn[j].op_str);
	print_insn_detail(&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);
	}
	}

	void test_macros() {
	assert(CS_AARCH64(_INS_BL) == ARM64_INS_BL);
	assert(CS_AARCH64pre(CS_ARCH_) == CS_ARCH_ARM64);
	assert(CS_AARCH64CC(_AL) == ARM64_CC_AL);
	assert(CS_AARCH64_VL_(16B) == ARM64_VAS_16B);
	cs_detail detail = { 0 };
	CS_cs_aarch64() arm64_detail = { 0 };
	detail.arm64 = arm64_detail;
	CS_aarch64_op() op = { 0 };
	detail.CS_aarch64_.operands[0] = op;
	CS_aarch64_reg() reg = 1;
	CS_aarch64_cc() cc = ARM64_CC_AL;
	CS_aarch64_extender() arm64_extender = ARM64_EXT_SXTB;
	CS_aarch64_shifter() arm64_shifter = ARM64_SFT_LSL;
	CS_aarch64_vas() arm64_vas = ARM64_VAS_16B;
	}

	int main()
	{
	test();
	test_macros();

	return 0;
	}