1#include <stdio.h>
2#include <stdlib.h>
3
4#include "libdis.h"
5
6#ifdef _MSC_VER
7        #define snprintf        _snprintf
8        #define inline          __inline
9#endif
10
11int x86_insn_is_valid( x86_insn_t *insn ) {
12	if ( insn && insn->type != insn_invalid && insn->size > 0 ) {
13		return 1;
14	}
15
16	return 0;
17}
18
19uint32_t x86_get_address( x86_insn_t *insn ) {
20	x86_oplist_t *op_lst;
21        if (! insn || ! insn->operands ) {
22        	return 0;
23        }
24
25	for (op_lst = insn->operands; op_lst; op_lst = op_lst->next ) {
26		if ( op_lst->op.type == op_offset ) {
27			return op_lst->op.data.offset;
28		} else if ( op_lst->op.type == op_absolute ) {
29			if ( op_lst->op.datatype == op_descr16 ) {
30				return (uint32_t)
31					op_lst->op.data.absolute.offset.off16;
32			}
33			return op_lst->op.data.absolute.offset.off32;
34		}
35	}
36
37	return 0;
38}
39
40int32_t x86_get_rel_offset( x86_insn_t *insn ) {
41	x86_oplist_t *op_lst;
42        if (! insn || ! insn->operands ) {
43        	return 0;
44        }
45
46	for (op_lst = insn->operands; op_lst; op_lst = op_lst->next ) {
47		if ( op_lst->op.type == op_relative_near ) {
48			return (int32_t) op_lst->op.data.relative_near;
49		} else if ( op_lst->op.type == op_relative_far ) {
50			return op_lst->op.data.relative_far;
51		}
52	}
53
54	return 0;
55}
56
57x86_op_t * x86_get_branch_target( x86_insn_t *insn ) {
58	x86_oplist_t *op_lst;
59        if (! insn || ! insn->operands ) {
60        	return NULL;
61        }
62
63	for (op_lst = insn->operands; op_lst; op_lst = op_lst->next ) {
64		if ( op_lst->op.access & op_execute ) {
65			return &(op_lst->op);
66		}
67	}
68
69	return NULL;
70}
71x86_op_t * x86_get_imm( x86_insn_t *insn ) {
72	x86_oplist_t *op_lst;
73        if (! insn || ! insn->operands ) {
74        	return NULL;
75        }
76
77	for (op_lst = insn->operands; op_lst; op_lst = op_lst->next ) {
78		if ( op_lst->op.type == op_immediate ) {
79			return &(op_lst->op);
80		}
81	}
82
83	return NULL;
84}
85
86#define IS_PROPER_IMM( x ) \
87	x->op.type == op_immediate && ! (x->op.flags & op_hardcode)
88
89
90/* if there is an immediate value in the instruction, return a pointer to
91 * it */
92unsigned char * x86_get_raw_imm( x86_insn_t *insn ) {
93        int size, offset;
94        x86_op_t *op  = NULL;
95
96        if (! insn || ! insn->operands ) {
97        	return(NULL);
98        }
99
100	/* a bit inelegant, but oh well... */
101	if ( IS_PROPER_IMM( insn->operands ) ) {
102		op = &insn->operands->op;
103	} else if ( insn->operands->next ) {
104		if ( IS_PROPER_IMM( insn->operands->next ) ) {
105			op = &insn->operands->next->op;
106		} else if ( insn->operands->next->next &&
107			    IS_PROPER_IMM( insn->operands->next->next ) ) {
108			op = &insn->operands->next->next->op;
109		}
110	}
111
112	if (! op ) {
113		return( NULL );
114	}
115
116	/* immediate data is at the end of the insn */
117	size = x86_operand_size( op );
118	offset = insn->size - size;
119	return( &insn->bytes[offset] );
120}
121
122
123unsigned int x86_operand_size( x86_op_t *op ) {
124        switch (op->datatype ) {
125                case op_byte:    return 1;
126                case op_word:    return 2;
127                case op_dword:   return 4;
128                case op_qword:   return 8;
129                case op_dqword:  return 16;
130                case op_sreal:   return 4;
131                case op_dreal:   return 8;
132                case op_extreal: return 10;
133                case op_bcd:     return 10;
134                case op_ssimd:   return 16;
135                case op_dsimd:   return 16;
136                case op_sssimd:  return 4;
137                case op_sdsimd:  return 8;
138                case op_descr32: return 6;
139                case op_descr16: return 4;
140                case op_pdescr32: return 6;
141                case op_pdescr16: return 6;
142		case op_bounds16: return 4;
143		case op_bounds32: return 8;
144                case op_fpuenv16:  return 14;
145                case op_fpuenv32:  return 28;
146                case op_fpustate16:  return 94;
147                case op_fpustate32:  return 108;
148                case op_fpregset: return 512;
149		case op_fpreg: return 10;
150		case op_none: return 0;
151        }
152        return(4);      /* default size */
153}
154
155void x86_set_insn_addr( x86_insn_t *insn, uint32_t addr ) {
156        if ( insn ) insn->addr = addr;
157}
158
159void x86_set_insn_offset( x86_insn_t *insn, unsigned int offset ){
160        if ( insn ) insn->offset = offset;
161}
162
163void x86_set_insn_function( x86_insn_t *insn, void * func ){
164        if ( insn ) insn->function = func;
165}
166
167void x86_set_insn_block( x86_insn_t *insn, void * block ){
168        if ( insn ) insn->block = block;
169}
170
171void x86_tag_insn( x86_insn_t *insn ){
172        if ( insn ) insn->tag = 1;
173}
174
175void x86_untag_insn( x86_insn_t *insn ){
176        if ( insn ) insn->tag = 0;
177}
178
179int x86_insn_is_tagged( x86_insn_t *insn ){
180        return insn->tag;
181}
182
183