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
x86_insn_is_valid(x86_insn_t * insn)11 int 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
x86_get_address(x86_insn_t * insn)19 uint32_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
x86_get_rel_offset(x86_insn_t * insn)40 int32_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
x86_get_branch_target(x86_insn_t * insn)57 x86_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 }
x86_get_imm(x86_insn_t * insn)71 x86_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 */
x86_get_raw_imm(x86_insn_t * insn)92 unsigned 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
x86_operand_size(x86_op_t * op)123 unsigned 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
x86_set_insn_addr(x86_insn_t * insn,uint32_t addr)155 void x86_set_insn_addr( x86_insn_t *insn, uint32_t addr ) {
156 if ( insn ) insn->addr = addr;
157 }
158
x86_set_insn_offset(x86_insn_t * insn,unsigned int offset)159 void x86_set_insn_offset( x86_insn_t *insn, unsigned int offset ){
160 if ( insn ) insn->offset = offset;
161 }
162
x86_set_insn_function(x86_insn_t * insn,void * func)163 void x86_set_insn_function( x86_insn_t *insn, void * func ){
164 if ( insn ) insn->function = func;
165 }
166
x86_set_insn_block(x86_insn_t * insn,void * block)167 void x86_set_insn_block( x86_insn_t *insn, void * block ){
168 if ( insn ) insn->block = block;
169 }
170
x86_tag_insn(x86_insn_t * insn)171 void x86_tag_insn( x86_insn_t *insn ){
172 if ( insn ) insn->tag = 1;
173 }
174
x86_untag_insn(x86_insn_t * insn)175 void x86_untag_insn( x86_insn_t *insn ){
176 if ( insn ) insn->tag = 0;
177 }
178
x86_insn_is_tagged(x86_insn_t * insn)179 int x86_insn_is_tagged( x86_insn_t *insn ){
180 return insn->tag;
181 }
182
183