michael@0: #include <stdlib.h>
michael@0: #include <string.h>
michael@0: 
michael@0: #include "ia32_reg.h"
michael@0: #include "ia32_insn.h"
michael@0: 
michael@0: #define NUM_X86_REGS	92
michael@0: 
michael@0: /* register sizes */
michael@0: #define REG_DWORD_SIZE 4
michael@0: #define REG_WORD_SIZE 2
michael@0: #define REG_BYTE_SIZE 1
michael@0: #define REG_MMX_SIZE 8
michael@0: #define REG_SIMD_SIZE 16
michael@0: #define REG_DEBUG_SIZE 4
michael@0: #define REG_CTRL_SIZE 4
michael@0: #define REG_TEST_SIZE 4
michael@0: #define REG_SEG_SIZE 2
michael@0: #define REG_FPU_SIZE 10
michael@0: #define REG_FLAGS_SIZE 4
michael@0: #define REG_FPCTRL_SIZE 2
michael@0: #define REG_FPSTATUS_SIZE 2
michael@0: #define REG_FPTAG_SIZE 2
michael@0: #define REG_EIP_SIZE 4
michael@0: #define REG_IP_SIZE 2
michael@0: 
michael@0: /* REGISTER ALIAS TABLE:
michael@0:  *
michael@0:  * NOTE: the MMX register mapping is fixed to the physical registers
michael@0:  * used by the FPU. The floating FP stack does not effect the location
michael@0:  * of the MMX registers, so this aliasing is not 100% accurate.
michael@0:  * */
michael@0: static struct {
michael@0: 	unsigned char alias;	/* id of register this is an alias for */
michael@0: 	unsigned char shift;	/* # of bits register must be shifted */
michael@0: } ia32_reg_aliases[] = {
michael@0: 	{ 0,0 },
michael@0: 	{ REG_DWORD_OFFSET,	0 }, 	/* al :  1 */
michael@0: 	{ REG_DWORD_OFFSET,	8 }, 	/* ah :  2 */
michael@0: 	{ REG_DWORD_OFFSET,	0 }, 	/* ax :  3 */
michael@0: 	{ REG_DWORD_OFFSET + 1,	0 }, 	/* cl :  4 */
michael@0: 	{ REG_DWORD_OFFSET + 1,	8 }, 	/* ch :  5 */
michael@0: 	{ REG_DWORD_OFFSET + 1,	0 }, 	/* cx :  6 */
michael@0: 	{ REG_DWORD_OFFSET + 2,	0 }, 	/* dl :  7 */
michael@0: 	{ REG_DWORD_OFFSET + 2,	8 }, 	/* dh :  8 */
michael@0: 	{ REG_DWORD_OFFSET + 2,	0 }, 	/* dx :  9 */
michael@0: 	{ REG_DWORD_OFFSET + 3,	0 }, 	/* bl : 10 */
michael@0: 	{ REG_DWORD_OFFSET + 3,	8 }, 	/* bh : 11 */
michael@0: 	{ REG_DWORD_OFFSET + 3,	0 }, 	/* bx : 12 */
michael@0: 	{ REG_DWORD_OFFSET + 4,	0 }, 	/* sp : 13 */
michael@0: 	{ REG_DWORD_OFFSET + 5,	0 }, 	/* bp : 14 */
michael@0: 	{ REG_DWORD_OFFSET + 6,	0 }, 	/* si : 15 */
michael@0: 	{ REG_DWORD_OFFSET + 7,	0 }, 	/* di : 16 */
michael@0: 	{ REG_EIP_INDEX,	0 }, 	/* ip : 17 */
michael@0: 	{ REG_FPU_OFFSET,	0 }, 	/* mm0 : 18 */
michael@0: 	{ REG_FPU_OFFSET + 1,	0 }, 	/* mm1 : 19 */
michael@0: 	{ REG_FPU_OFFSET + 2,	0 }, 	/* mm2 : 20 */
michael@0: 	{ REG_FPU_OFFSET + 3,	0 }, 	/* mm3 : 21 */
michael@0: 	{ REG_FPU_OFFSET + 4,	0 }, 	/* mm4 : 22 */
michael@0: 	{ REG_FPU_OFFSET + 5,	0 }, 	/* mm5 : 23 */
michael@0: 	{ REG_FPU_OFFSET + 6,	0 }, 	/* mm6 : 24 */
michael@0: 	{ REG_FPU_OFFSET + 7,	0 } 	/* mm7 : 25 */
michael@0:  };
michael@0: 
michael@0: /* REGISTER TABLE: size, type, and name of every register in the 
michael@0:  *                 CPU. Does not include MSRs since the are, after all,
michael@0:  *                 model specific. */
michael@0: static struct {
michael@0: 	unsigned int size;
michael@0: 	enum x86_reg_type type;
michael@0: 	unsigned int alias;
michael@0: 	char mnemonic[8];
michael@0: } ia32_reg_table[NUM_X86_REGS + 2] = {
michael@0: 	{ 0, 0, 0, "" },
michael@0: 	/* REG_DWORD_OFFSET */
michael@0: 	{ REG_DWORD_SIZE, reg_gen | reg_ret, 0, "eax" },
michael@0: 	{ REG_DWORD_SIZE, reg_gen | reg_count, 0, "ecx" },
michael@0: 	{ REG_DWORD_SIZE, reg_gen, 0, "edx" },
michael@0: 	{ REG_DWORD_SIZE, reg_gen, 0, "ebx" },
michael@0: 	/* REG_ESP_INDEX */
michael@0: 	{ REG_DWORD_SIZE, reg_gen | reg_sp, 0, "esp" },
michael@0: 	{ REG_DWORD_SIZE, reg_gen | reg_fp, 0, "ebp" },
michael@0: 	{ REG_DWORD_SIZE, reg_gen | reg_src, 0, "esi" },
michael@0: 	{ REG_DWORD_SIZE, reg_gen | reg_dest, 0, "edi" },
michael@0: 	/* REG_WORD_OFFSET */
michael@0: 	{ REG_WORD_SIZE, reg_gen | reg_ret, 3, "ax" },
michael@0: 	{ REG_WORD_SIZE, reg_gen | reg_count, 6, "cx" },
michael@0: 	{ REG_WORD_SIZE, reg_gen, 9, "dx" },
michael@0: 	{ REG_WORD_SIZE, reg_gen, 12, "bx" },
michael@0: 	{ REG_WORD_SIZE, reg_gen | reg_sp, 13, "sp" },
michael@0: 	{ REG_WORD_SIZE, reg_gen | reg_fp, 14, "bp" },
michael@0: 	{ REG_WORD_SIZE, reg_gen | reg_src, 15, "si" },
michael@0: 	{ REG_WORD_SIZE, reg_gen | reg_dest, 16, "di" },
michael@0: 	/* REG_BYTE_OFFSET */
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 1, "al" },
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 4, "cl" },
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 7, "dl" },
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 10, "bl" },
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 2, "ah" },
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 5, "ch" },
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 8, "dh" },
michael@0: 	{ REG_BYTE_SIZE, reg_gen, 11, "bh" },
michael@0: 	/* REG_MMX_OFFSET */
michael@0: 	{ REG_MMX_SIZE, reg_simd, 18, "mm0" },
michael@0: 	{ REG_MMX_SIZE, reg_simd, 19, "mm1" },
michael@0: 	{ REG_MMX_SIZE, reg_simd, 20, "mm2" },
michael@0: 	{ REG_MMX_SIZE, reg_simd, 21, "mm3" },
michael@0: 	{ REG_MMX_SIZE, reg_simd, 22, "mm4" },
michael@0: 	{ REG_MMX_SIZE, reg_simd, 23, "mm5" },
michael@0: 	{ REG_MMX_SIZE, reg_simd, 24, "mm6" },
michael@0: 	{ REG_MMX_SIZE, reg_simd, 25, "mm7" },
michael@0: 	/* REG_SIMD_OFFSET */
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm0" },
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm1" },
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm2" },
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm3" },
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm4" },
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm5" },
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm6" },
michael@0: 	{ REG_SIMD_SIZE, reg_simd, 0, "xmm7" },
michael@0: 	/* REG_DEBUG_OFFSET */
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr0" },
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr1" },
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr2" },
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr3" },
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr4" },
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr5" },
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr6" },
michael@0: 	{ REG_DEBUG_SIZE, reg_sys, 0, "dr7" },
michael@0: 	/* REG_CTRL_OFFSET */
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr0" },
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr1" },
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr2" },
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr3" },
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr4" },
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr5" },
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr6" },
michael@0: 	{ REG_CTRL_SIZE, reg_sys, 0, "cr7" },
michael@0: 	/* REG_TEST_OFFSET */
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr0" },
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr1" },
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr2" },
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr3" },
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr4" },
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr5" },
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr6" },
michael@0: 	{ REG_TEST_SIZE, reg_sys, 0, "tr7" },
michael@0: 	/* REG_SEG_OFFSET */
michael@0: 	{ REG_SEG_SIZE, reg_seg, 0, "es" },
michael@0: 	{ REG_SEG_SIZE, reg_seg, 0, "cs" },
michael@0: 	{ REG_SEG_SIZE, reg_seg, 0, "ss" },
michael@0: 	{ REG_SEG_SIZE, reg_seg, 0, "ds" },
michael@0: 	{ REG_SEG_SIZE, reg_seg, 0, "fs" },
michael@0: 	{ REG_SEG_SIZE, reg_seg, 0, "gs" },
michael@0: 	/* REG_LDTR_INDEX */
michael@0: 	{ REG_DWORD_SIZE, reg_sys, 0, "ldtr" },
michael@0: 	/* REG_GDTR_INDEX */
michael@0: 	{ REG_DWORD_SIZE, reg_sys, 0, "gdtr" },
michael@0: 	/* REG_FPU_OFFSET */
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(0)" },
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(1)" },
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(2)" },
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(3)" },
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(4)" },
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(5)" },
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(6)" },
michael@0: 	{ REG_FPU_SIZE, reg_fpu, 0, "st(7)" },
michael@0: 	/* REG_FLAGS_INDEX : 81 */
michael@0: 	{ REG_FLAGS_SIZE, reg_cond, 0, "eflags" }, 
michael@0: 	/* REG_FPCTRL_INDEX  : 82*/
michael@0: 	{ REG_FPCTRL_SIZE, reg_fpu | reg_sys, 0, "fpctrl" }, 
michael@0: 	/* REG_FPSTATUS_INDEX : 83*/
michael@0: 	{ REG_FPSTATUS_SIZE, reg_fpu | reg_sys, 0, "fpstat" },
michael@0: 	/* REG_FPTAG_INDEX : 84 */
michael@0: 	{ REG_FPTAG_SIZE, reg_fpu | reg_sys, 0, "fptag" }, 
michael@0: 	/* REG_EIP_INDEX : 85 */
michael@0: 	{ REG_EIP_SIZE, reg_pc, 0, "eip" },
michael@0: 	/* REG_IP_INDEX : 86 */
michael@0: 	{ REG_IP_SIZE, reg_pc, 17, "ip" },
michael@0: 	/* REG_IDTR_INDEX : 87 */
michael@0: 	{ REG_DWORD_SIZE, reg_sys, 0, "idtr" },
michael@0: 	/* REG_MXCSG_INDEX : SSE Control Reg : 88 */
michael@0: 	{ REG_DWORD_SIZE, reg_sys | reg_simd, 0, "mxcsr" },
michael@0: 	/* REG_TR_INDEX : Task Register : 89 */
michael@0: 	{ 16 + 64, reg_sys, 0, "tr" },
michael@0: 	/* REG_CSMSR_INDEX : SYSENTER_CS_MSR : 90 */
michael@0: 	{ REG_DWORD_SIZE, reg_sys, 0, "cs_msr" },
michael@0: 	/* REG_ESPMSR_INDEX : SYSENTER_ESP_MSR : 91 */
michael@0: 	{ REG_DWORD_SIZE, reg_sys, 0, "esp_msr" },
michael@0: 	/* REG_EIPMSR_INDEX : SYSENTER_EIP_MSR : 92 */
michael@0: 	{ REG_DWORD_SIZE, reg_sys, 0, "eip_msr" },
michael@0: 	{ 0 }
michael@0:  };
michael@0: 
michael@0: 
michael@0: static size_t sz_regtable = NUM_X86_REGS + 1;
michael@0: 
michael@0: 
michael@0: void ia32_handle_register( x86_reg_t *reg, size_t id ) {
michael@0: 	unsigned int alias;
michael@0: 	if (! id || id > sz_regtable ) {
michael@0: 		return;
michael@0: 	}
michael@0: 
michael@0: 	memset( reg, 0, sizeof(x86_reg_t) );
michael@0: 
michael@0:         strncpy( reg->name, ia32_reg_table[id].mnemonic, MAX_REGNAME );
michael@0: 
michael@0:         reg->type = ia32_reg_table[id].type;
michael@0:         reg->size = ia32_reg_table[id].size;
michael@0: 
michael@0: 	alias = ia32_reg_table[id].alias;
michael@0: 	if ( alias ) {
michael@0: 		reg->alias = ia32_reg_aliases[alias].alias;
michael@0: 		reg->shift = ia32_reg_aliases[alias].shift;
michael@0: 	}
michael@0:         reg->id = id;
michael@0: 
michael@0: 	return;
michael@0: }
michael@0: 
michael@0: size_t ia32_true_register_id( size_t id ) {
michael@0: 	size_t reg;
michael@0: 
michael@0: 	if (! id || id > sz_regtable ) {
michael@0: 		return 0;
michael@0: 	}
michael@0: 
michael@0: 	reg = id;
michael@0: 	if (ia32_reg_table[reg].alias) {
michael@0: 		reg = ia32_reg_aliases[ia32_reg_table[reg].alias].alias;
michael@0: 	}
michael@0: 	return reg;
michael@0: }