diff options
Diffstat (limited to 'arch/x86/lib/insn.c')
| -rw-r--r-- | arch/x86/lib/insn.c | 114 | 
1 files changed, 89 insertions, 25 deletions
diff --git a/arch/x86/lib/insn.c b/arch/x86/lib/insn.c index 9f33b984d0e..54fcffed28e 100644 --- a/arch/x86/lib/insn.c +++ b/arch/x86/lib/insn.c @@ -18,18 +18,31 @@   * Copyright (C) IBM Corporation, 2002, 2004, 2009   */ +#ifdef __KERNEL__  #include <linux/string.h> +#else +#include <string.h> +#endif  #include <asm/inat.h>  #include <asm/insn.h> -#define get_next(t, insn)	\ -	({t r; r = *(t*)insn->next_byte; insn->next_byte += sizeof(t); r; }) +/* Verify next sizeof(t) bytes can be on the same instruction */ +#define validate_next(t, insn, n)	\ +	((insn)->next_byte + sizeof(t) + n - (insn)->kaddr <= MAX_INSN_SIZE) + +#define __get_next(t, insn)	\ +	({ t r = *(t*)insn->next_byte; insn->next_byte += sizeof(t); r; }) -#define peek_next(t, insn)	\ -	({t r; r = *(t*)insn->next_byte; r; }) +#define __peek_nbyte_next(t, insn, n)	\ +	({ t r = *(t*)((insn)->next_byte + n); r; }) + +#define get_next(t, insn)	\ +	({ if (unlikely(!validate_next(t, insn, 0))) goto err_out; __get_next(t, insn); })  #define peek_nbyte_next(t, insn, n)	\ -	({t r; r = *(t*)((insn)->next_byte + n); r; }) +	({ if (unlikely(!validate_next(t, insn, n))) goto err_out; __peek_nbyte_next(t, insn, n); }) + +#define peek_next(t, insn)	peek_nbyte_next(t, insn, 0)  /**   * insn_init() - initialize struct insn @@ -158,6 +171,8 @@ vex_end:  	insn->vex_prefix.got = 1;  	prefixes->got = 1; + +err_out:  	return;  } @@ -174,7 +189,8 @@ vex_end:  void insn_get_opcode(struct insn *insn)  {  	struct insn_field *opcode = &insn->opcode; -	insn_byte_t op, pfx; +	insn_byte_t op; +	int pfx_id;  	if (opcode->got)  		return;  	if (!insn->prefixes.got) @@ -191,7 +207,7 @@ void insn_get_opcode(struct insn *insn)  		m = insn_vex_m_bits(insn);  		p = insn_vex_p_bits(insn);  		insn->attr = inat_get_avx_attribute(op, m, p); -		if (!inat_accept_vex(insn->attr)) +		if (!inat_accept_vex(insn->attr) && !inat_is_group(insn->attr))  			insn->attr = 0;	/* This instruction is bad */  		goto end;	/* VEX has only 1 byte for opcode */  	} @@ -201,13 +217,16 @@ void insn_get_opcode(struct insn *insn)  		/* Get escaped opcode */  		op = get_next(insn_byte_t, insn);  		opcode->bytes[opcode->nbytes++] = op; -		pfx = insn_last_prefix(insn); -		insn->attr = inat_get_escape_attribute(op, pfx, insn->attr); +		pfx_id = insn_last_prefix_id(insn); +		insn->attr = inat_get_escape_attribute(op, pfx_id, insn->attr);  	}  	if (inat_must_vex(insn->attr))  		insn->attr = 0;	/* This instruction is bad */  end:  	opcode->got = 1; + +err_out: +	return;  }  /** @@ -221,7 +240,7 @@ end:  void insn_get_modrm(struct insn *insn)  {  	struct insn_field *modrm = &insn->modrm; -	insn_byte_t pfx, mod; +	insn_byte_t pfx_id, mod;  	if (modrm->got)  		return;  	if (!insn->opcode.got) @@ -232,15 +251,20 @@ void insn_get_modrm(struct insn *insn)  		modrm->value = mod;  		modrm->nbytes = 1;  		if (inat_is_group(insn->attr)) { -			pfx = insn_last_prefix(insn); -			insn->attr = inat_get_group_attribute(mod, pfx, +			pfx_id = insn_last_prefix_id(insn); +			insn->attr = inat_get_group_attribute(mod, pfx_id,  							      insn->attr); +			if (insn_is_avx(insn) && !inat_accept_vex(insn->attr)) +				insn->attr = 0;	/* This is bad */  		}  	}  	if (insn->x86_64 && inat_is_force64(insn->attr))  		insn->opnd_bytes = 8;  	modrm->got = 1; + +err_out: +	return;  } @@ -290,6 +314,9 @@ void insn_get_sib(struct insn *insn)  		}  	}  	insn->sib.got = 1; + +err_out: +	return;  } @@ -351,10 +378,13 @@ void insn_get_displacement(struct insn *insn)  	}  out:  	insn->displacement.got = 1; + +err_out: +	return;  } -/* Decode moffset16/32/64 */ -static void __get_moffset(struct insn *insn) +/* Decode moffset16/32/64. Return 0 if failed */ +static int __get_moffset(struct insn *insn)  {  	switch (insn->addr_bytes) {  	case 2: @@ -371,12 +401,19 @@ static void __get_moffset(struct insn *insn)  		insn->moffset2.value = get_next(int, insn);  		insn->moffset2.nbytes = 4;  		break; +	default:	/* opnd_bytes must be modified manually */ +		goto err_out;  	}  	insn->moffset1.got = insn->moffset2.got = 1; + +	return 1; + +err_out: +	return 0;  } -/* Decode imm v32(Iz) */ -static void __get_immv32(struct insn *insn) +/* Decode imm v32(Iz). Return 0 if failed */ +static int __get_immv32(struct insn *insn)  {  	switch (insn->opnd_bytes) {  	case 2: @@ -388,11 +425,18 @@ static void __get_immv32(struct insn *insn)  		insn->immediate.value = get_next(int, insn);  		insn->immediate.nbytes = 4;  		break; +	default:	/* opnd_bytes must be modified manually */ +		goto err_out;  	} + +	return 1; + +err_out: +	return 0;  } -/* Decode imm v64(Iv/Ov) */ -static void __get_immv(struct insn *insn) +/* Decode imm v64(Iv/Ov), Return 0 if failed */ +static int __get_immv(struct insn *insn)  {  	switch (insn->opnd_bytes) {  	case 2: @@ -409,12 +453,18 @@ static void __get_immv(struct insn *insn)  		insn->immediate2.value = get_next(int, insn);  		insn->immediate2.nbytes = 4;  		break; +	default:	/* opnd_bytes must be modified manually */ +		goto err_out;  	}  	insn->immediate1.got = insn->immediate2.got = 1; + +	return 1; +err_out: +	return 0;  }  /* Decode ptr16:16/32(Ap) */ -static void __get_immptr(struct insn *insn) +static int __get_immptr(struct insn *insn)  {  	switch (insn->opnd_bytes) {  	case 2: @@ -427,11 +477,17 @@ static void __get_immptr(struct insn *insn)  		break;  	case 8:  		/* ptr16:64 is not exist (no segment) */ -		return; +		return 0; +	default:	/* opnd_bytes must be modified manually */ +		goto err_out;  	}  	insn->immediate2.value = get_next(unsigned short, insn);  	insn->immediate2.nbytes = 2;  	insn->immediate1.got = insn->immediate2.got = 1; + +	return 1; +err_out: +	return 0;  }  /** @@ -451,7 +507,8 @@ void insn_get_immediate(struct insn *insn)  		insn_get_displacement(insn);  	if (inat_has_moffset(insn->attr)) { -		__get_moffset(insn); +		if (!__get_moffset(insn)) +			goto err_out;  		goto done;  	} @@ -479,16 +536,20 @@ void insn_get_immediate(struct insn *insn)  		insn->immediate2.nbytes = 4;  		break;  	case INAT_IMM_PTR: -		__get_immptr(insn); +		if (!__get_immptr(insn)) +			goto err_out;  		break;  	case INAT_IMM_VWORD32: -		__get_immv32(insn); +		if (!__get_immv32(insn)) +			goto err_out;  		break;  	case INAT_IMM_VWORD: -		__get_immv(insn); +		if (!__get_immv(insn)) +			goto err_out;  		break;  	default: -		break; +		/* Here, insn must have an immediate, but failed */ +		goto err_out;  	}  	if (inat_has_second_immediate(insn->attr)) {  		insn->immediate2.value = get_next(char, insn); @@ -496,6 +557,9 @@ void insn_get_immediate(struct insn *insn)  	}  done:  	insn->immediate.got = 1; + +err_out: +	return;  }  /**  | 
