OLD | NEW |
| (Empty) |
1 %module x86disasm | |
2 %{ | |
3 #ifdef _MSC_VER | |
4 typedef __int64 qword; | |
5 #else | |
6 typedef long long qword; | |
7 #endif | |
8 | |
9 #include <sys/types.h> | |
10 | |
11 #define MAX_REGNAME 8 | |
12 #define MAX_PREFIX_STR 32 | |
13 #define MAX_MNEM_STR 16 | |
14 #define MAX_INSN_SIZE 20 | |
15 #define MAX_OP_STRING 32 | |
16 #define MAX_OP_RAW_STRING 64 | |
17 #define MAX_OP_XML_STRING 256 | |
18 #define MAX_NUM_OPERANDS 8 | |
19 #define MAX_INSN_STRING 512 | |
20 #define MAX_INSN_RAW_STRING 1024 | |
21 #define MAX_INSN_XML_STRING 4096 | |
22 | |
23 #include "../../../config.h" | |
24 | |
25 | |
26 const char * version_string( void ) { | |
27 return PACKAGE_VERSION; | |
28 } | |
29 | |
30 %} | |
31 | |
32 const char * version_string( void ); | |
33 | |
34 %rename(X86_Register) x86_reg_t; | |
35 %rename(X86_EAddr) x86_ea_t; | |
36 %rename(X86_Operand) x86_op_t; | |
37 //%rename(X86_OpList) x86_oplist_t; | |
38 %rename(X86_Insn) x86_insn_t; | |
39 %rename(X86_InvOperand) x86_invariant_op_t; | |
40 %rename(X86_Invariant) x86_invariant_t; | |
41 | |
42 %include "carrays.i" | |
43 | |
44 %array_class( unsigned char, byteArray ); | |
45 | |
46 | |
47 %apply (unsigned char *STRING, int LENGTH) { | |
48 (unsigned char *buf, size_t buf_len) | |
49 }; | |
50 | |
51 | |
52 %inline %{ | |
53 | |
54 | |
55 enum x86_asm_format { | |
56 unknown_syntax = 0, /* never use! */ | |
57 native_syntax, /* header: 35 bytes */ | |
58 intel_syntax, /* header: 23 bytes */ | |
59 att_syntax, /* header: 23 bytes */ | |
60 xml_syntax, /* header: 679 bytes */ | |
61 raw_syntax /* header: 172 bytes */ | |
62 }; | |
63 %} | |
64 | |
65 /* ================================================================== */ | |
66 /* operand class */ | |
67 %inline %{ | |
68 enum x86_reg_type { | |
69 reg_gen = 0x00001, reg_in = 0x00002, | |
70 reg_out = 0x00004, reg_local = 0x00008, | |
71 reg_fpu = 0x00010, reg_seg = 0x00020, | |
72 reg_simd = 0x00040, reg_sys = 0x00080, | |
73 reg_sp = 0x00100, reg_fp = 0x00200, | |
74 reg_pc = 0x00400, reg_retaddr = 0x00800, | |
75 reg_cond = 0x01000, reg_zero = 0x02000, | |
76 reg_ret = 0x04000, reg_src = 0x10000, | |
77 reg_dest = 0x20000, reg_count = 0x40000 | |
78 }; | |
79 | |
80 typedef struct { | |
81 char name[MAX_REGNAME]; | |
82 enum x86_reg_type type; | |
83 unsigned int size; | |
84 unsigned int id; | |
85 unsigned int alias; | |
86 unsigned int shift; | |
87 } x86_reg_t; | |
88 | |
89 void x86_reg_from_id( unsigned int id, x86_reg_t * reg ); | |
90 | |
91 typedef struct { | |
92 unsigned int scale; | |
93 x86_reg_t index, base; | |
94 long disp; | |
95 char disp_sign; | |
96 char disp_size; | |
97 } x86_ea_t; | |
98 | |
99 enum x86_op_type { | |
100 op_unused = 0, | |
101 op_register = 1, | |
102 op_immediate = 2, | |
103 op_relative_near = 3, | |
104 op_relative_far = 4, | |
105 op_absolute = 5, | |
106 op_expression = 6, | |
107 op_offset = 7, | |
108 op_unknown | |
109 }; | |
110 | |
111 enum x86_op_datatype { | |
112 op_byte = 1, op_word = 2, | |
113 op_dword = 3, op_qword = 4, | |
114 op_dqword = 5, op_sreal = 6, | |
115 op_dreal = 7, op_extreal = 8, | |
116 op_bcd = 9, op_ssimd = 10, | |
117 op_dsimd = 11, op_sssimd = 12, | |
118 op_sdsimd = 13, op_descr32 = 14, | |
119 op_descr16 = 15, op_pdescr32 = 16, | |
120 op_pdescr16 = 17, op_fpuenv = 18, | |
121 op_fpregset = 19, | |
122 }; | |
123 | |
124 enum x86_op_access { | |
125 op_read = 1, | |
126 op_write = 2, | |
127 op_execute = 4 | |
128 }; | |
129 | |
130 enum x86_op_flags { | |
131 op_signed = 1, op_string = 2, | |
132 op_constant = 4, op_pointer = 8, | |
133 op_sysref = 0x010, op_implied = 0x020, | |
134 op_hardcode = 0x40, op_es_seg = 0x100, | |
135 op_cs_seg = 0x200, op_ss_seg = 0x300, | |
136 op_ds_seg = 0x400, op_fs_seg = 0x500, | |
137 op_gs_seg = 0x600 | |
138 }; | |
139 | |
140 typedef struct { | |
141 enum x86_op_type type; | |
142 enum x86_op_datatype datatype; | |
143 enum x86_op_access access; | |
144 enum x86_op_flags flags; | |
145 union { | |
146 char sbyte; | |
147 short sword; | |
148 long sdword; | |
149 qword sqword; | |
150 unsigned char byte; | |
151 unsigned short word; | |
152 unsigned long dword; | |
153 qword qword; | |
154 float sreal; | |
155 double dreal; | |
156 unsigned char extreal[10]; | |
157 unsigned char bcd[10]; | |
158 qword dqword[2]; | |
159 unsigned char simd[16]; | |
160 unsigned char fpuenv[28]; | |
161 void * address; | |
162 unsigned long offset; | |
163 x86_reg_t reg; | |
164 char relative_near; | |
165 long relative_far; | |
166 x86_ea_t expression; | |
167 } data; | |
168 void * insn; | |
169 } x86_op_t; | |
170 | |
171 unsigned int x86_operand_size( x86_op_t *op ); | |
172 | |
173 int x86_format_operand(x86_op_t *op, char *buf, int len, | |
174 enum x86_asm_format format); | |
175 %} | |
176 | |
177 %extend x86_reg_t{ | |
178 x86_reg_t * aliased_reg( ) { | |
179 x86_reg_t * reg = (x86_reg_t * ) | |
180 calloc( sizeof(x86_reg_t), 1 ); | |
181 x86_reg_from_id( self->id, reg ); | |
182 return reg; | |
183 } | |
184 } | |
185 | |
186 %extend x86_op_t{ | |
187 size_t size() { | |
188 return x86_operand_size( self ); | |
189 } | |
190 char * format( enum x86_asm_format format ) { | |
191 char *buf, *str; | |
192 size_t len; | |
193 | |
194 switch ( format ) { | |
195 case xml_syntax: | |
196 len = MAX_OP_XML_STRING; | |
197 break; | |
198 case raw_syntax: | |
199 len = MAX_OP_RAW_STRING; | |
200 break; | |
201 case native_syntax: | |
202 case intel_syntax: | |
203 case att_syntax: | |
204 case unknown_syntax: | |
205 default: | |
206 len = MAX_OP_STRING; | |
207 break; | |
208 } | |
209 | |
210 buf = (char * ) calloc( len + 1, 1 ); | |
211 x86_format_operand( self, buf, len, format ); | |
212 | |
213 /* drop buffer down to a reasonable size */ | |
214 str = strdup( buf ); | |
215 free(buf); | |
216 return str; | |
217 } | |
218 | |
219 int is_address( ) { | |
220 if ( self->type == op_absolute || | |
221 self->type == op_offset ) { | |
222 return 1; | |
223 } | |
224 | |
225 return 0; | |
226 } | |
227 | |
228 int is_relative( ) { | |
229 if ( self->type == op_relative_near || | |
230 self->type == op_relative_far ) { | |
231 return 1; | |
232 } | |
233 | |
234 return 0; | |
235 } | |
236 | |
237 %newobject copy; | |
238 x86_op_t * copy() { | |
239 x86_op_t *op = (x86_op_t *) calloc( sizeof(x86_op_t), 1 ); | |
240 | |
241 if ( op ) { | |
242 memcpy( op, self, sizeof(x86_op_t) ); | |
243 } | |
244 | |
245 return op; | |
246 } | |
247 } | |
248 | |
249 /* ================================================================== */ | |
250 /* operand list class */ | |
251 %inline %{ | |
252 typedef struct X86_OpListNode { | |
253 x86_op_t *op; | |
254 struct X86_OpListNode *next, *prev; | |
255 } X86_OpListNode; | |
256 | |
257 typedef struct X86_OpList { | |
258 size_t count; | |
259 X86_OpListNode *head, *tail, *curr; | |
260 } X86_OpList; | |
261 %} | |
262 | |
263 %extend X86_OpList { | |
264 X86_OpList () { | |
265 X86_OpList *list = (X86_OpList *) | |
266 calloc( sizeof(X86_OpList), 1 ); | |
267 list->count = 0; | |
268 return list; | |
269 } | |
270 | |
271 ~X86_OpList() { | |
272 X86_OpListNode *node, *next; | |
273 | |
274 node = self->head; | |
275 while ( node ) { | |
276 next = node->next; | |
277 /* free( node->insn ); */ | |
278 free( node ); | |
279 node = next; | |
280 } | |
281 | |
282 free( self ); | |
283 } | |
284 | |
285 X86_OpListNode * first() { | |
286 self->curr = self->head; | |
287 return self->head; | |
288 } | |
289 | |
290 X86_OpListNode * last() { | |
291 self->curr = self->tail; | |
292 return self->tail; | |
293 } | |
294 | |
295 X86_OpListNode * next() { | |
296 if (! self->curr ) { | |
297 self->curr = self->head; | |
298 return self->head; | |
299 } | |
300 | |
301 self->curr = self->curr->next; | |
302 return self->curr; | |
303 } | |
304 | |
305 X86_OpListNode * prev() { | |
306 if (! self->curr ) { | |
307 self->curr = self->tail; | |
308 return self->tail; | |
309 } | |
310 | |
311 self->curr = self->curr->prev; | |
312 return self->curr; | |
313 } | |
314 | |
315 %newobject append; | |
316 void append( x86_op_t *op ) { | |
317 X86_OpListNode *node = (X86_OpListNode *) | |
318 calloc( sizeof(X86_OpListNode) , 1 ); | |
319 if (! node ) { | |
320 return; | |
321 } | |
322 | |
323 self->count++; | |
324 if ( ! self->tail ) { | |
325 self->head = self->tail = node; | |
326 } else { | |
327 self->tail->next = node; | |
328 node->prev = self->tail; | |
329 self->tail = node; | |
330 } | |
331 | |
332 node->op = x86_op_t_copy( op ); | |
333 } | |
334 } | |
335 | |
336 %inline %{ | |
337 typedef struct x86_operand_list { | |
338 x86_op_t op; | |
339 struct x86_operand_list *next; | |
340 } x86_oplist_t; | |
341 %} | |
342 | |
343 %extend x86_oplist_t { | |
344 %newobject x86_oplist_node_copy; | |
345 } | |
346 | |
347 /* ================================================================== */ | |
348 /* instruction class */ | |
349 %inline %{ | |
350 x86_oplist_t * x86_oplist_node_copy( x86_oplist_t * list ) { | |
351 x86_oplist_t *ptr; | |
352 ptr = (x86_oplist_t *) calloc( sizeof(x86_oplist_t), 1 ); | |
353 if ( ptr ) { | |
354 memcpy( &ptr->op, &list->op, sizeof(x86_op_t) ); | |
355 } | |
356 | |
357 return ptr; | |
358 } | |
359 | |
360 enum x86_insn_group { | |
361 insn_none = 0, insn_controlflow = 1, | |
362 insn_arithmetic = 2, insn_logic = 3, | |
363 insn_stack = 4, insn_comparison = 5, | |
364 insn_move = 6, insn_string = 7, | |
365 insn_bit_manip = 8, insn_flag_manip = 9, | |
366 insn_fpu = 10, insn_interrupt = 13, | |
367 insn_system = 14, insn_other = 15 | |
368 }; | |
369 | |
370 enum x86_insn_type { | |
371 insn_invalid = 0, insn_jmp = 0x1001, | |
372 insn_jcc = 0x1002, insn_call = 0x1003, | |
373 insn_callcc = 0x1004, insn_return = 0x1005, | |
374 insn_add = 0x2001, insn_sub = 0x2002, | |
375 insn_mul = 0x2003, insn_div = 0x2004, | |
376 insn_inc = 0x2005, insn_dec = 0x2006, | |
377 insn_shl = 0x2007, insn_shr = 0x2008, | |
378 insn_rol = 0x2009, insn_ror = 0x200A, | |
379 insn_and = 0x3001, insn_or = 0x3002, | |
380 insn_xor = 0x3003, insn_not = 0x3004, | |
381 insn_neg = 0x3005, insn_push = 0x4001, | |
382 insn_pop = 0x4002, insn_pushregs = 0x4003, | |
383 insn_popregs = 0x4004, insn_pushflags = 0x4005, | |
384 insn_popflags = 0x4006, insn_enter = 0x4007, | |
385 insn_leave = 0x4008, insn_test = 0x5001, | |
386 insn_cmp = 0x5002, insn_mov = 0x6001, | |
387 insn_movcc = 0x6002, insn_xchg = 0x6003, | |
388 insn_xchgcc = 0x6004, insn_strcmp = 0x7001, | |
389 insn_strload = 0x7002, insn_strmov = 0x7003, | |
390 insn_strstore = 0x7004, insn_translate = 0x7005, | |
391 insn_bittest = 0x8001, insn_bitset = 0x8002, | |
392 insn_bitclear = 0x8003, insn_clear_carry = 0x9001, | |
393 insn_clear_zero = 0x9002, insn_clear_oflow = 0x9003, | |
394 insn_clear_dir = 0x9004, insn_clear_sign = 0x9005, | |
395 insn_clear_parity = 0x9006, insn_set_carry = 0x9007, | |
396 insn_set_zero = 0x9008, insn_set_oflow = 0x9009, | |
397 insn_set_dir = 0x900A, insn_set_sign = 0x900B, | |
398 insn_set_parity = 0x900C, insn_tog_carry = 0x9010, | |
399 insn_tog_zero = 0x9020, insn_tog_oflow = 0x9030, | |
400 insn_tog_dir = 0x9040, insn_tog_sign = 0x9050, | |
401 insn_tog_parity = 0x9060, insn_fmov = 0xA001, | |
402 insn_fmovcc = 0xA002, insn_fneg = 0xA003, | |
403 insn_fabs = 0xA004, insn_fadd = 0xA005, | |
404 insn_fsub = 0xA006, insn_fmul = 0xA007, | |
405 insn_fdiv = 0xA008, insn_fsqrt = 0xA009, | |
406 insn_fcmp = 0xA00A, insn_fcos = 0xA00C, | |
407 insn_fldpi = 0xA00D, insn_fldz = 0xA00E, | |
408 insn_ftan = 0xA00F, insn_fsine = 0xA010, | |
409 insn_fsys = 0xA020, insn_int = 0xD001, | |
410 insn_intcc = 0xD002, insn_iret = 0xD003, | |
411 insn_bound = 0xD004, insn_debug = 0xD005, | |
412 insn_trace = 0xD006, insn_invalid_op = 0xD007, | |
413 insn_oflow = 0xD008, insn_halt = 0xE001, | |
414 insn_in = 0xE002, insn_out = 0xE003, | |
415 insn_cpuid = 0xE004, insn_nop = 0xF001, | |
416 insn_bcdconv = 0xF002, insn_szconv = 0xF003 | |
417 }; | |
418 | |
419 enum x86_insn_note { | |
420 insn_note_ring0 = 1, | |
421 insn_note_smm = 2, | |
422 insn_note_serial = 4 | |
423 }; | |
424 | |
425 enum x86_flag_status { | |
426 insn_carry_set = 0x1, | |
427 insn_zero_set = 0x2, | |
428 insn_oflow_set = 0x4, | |
429 insn_dir_set = 0x8, | |
430 insn_sign_set = 0x10, | |
431 insn_parity_set = 0x20, | |
432 insn_carry_or_zero_set = 0x40, | |
433 insn_zero_set_or_sign_ne_oflow = 0x80, | |
434 insn_carry_clear = 0x100, | |
435 insn_zero_clear = 0x200, | |
436 insn_oflow_clear = 0x400, | |
437 insn_dir_clear = 0x800, | |
438 insn_sign_clear = 0x1000, | |
439 insn_parity_clear = 0x2000, | |
440 insn_sign_eq_oflow = 0x4000, | |
441 insn_sign_ne_oflow = 0x8000 | |
442 }; | |
443 | |
444 enum x86_insn_cpu { | |
445 cpu_8086 = 1, cpu_80286 = 2, | |
446 cpu_80386 = 3, cpu_80387 = 4, | |
447 cpu_80486 = 5, cpu_pentium = 6, | |
448 cpu_pentiumpro = 7, cpu_pentium2 = 8, | |
449 cpu_pentium3 = 9, cpu_pentium4 = 10, | |
450 cpu_k6 = 16, cpu_k7 = 32, | |
451 cpu_athlon = 48 | |
452 }; | |
453 | |
454 enum x86_insn_isa { | |
455 isa_gp = 1, isa_fp = 2, | |
456 isa_fpumgt = 3, isa_mmx = 4, | |
457 isa_sse1 = 5, isa_sse2 = 6, | |
458 isa_sse3 = 7, isa_3dnow = 8, | |
459 isa_sys = 9 | |
460 }; | |
461 | |
462 enum x86_insn_prefix { | |
463 insn_no_prefix = 0, | |
464 insn_rep_zero = 1, | |
465 insn_rep_notzero = 2, | |
466 insn_lock = 4 | |
467 }; | |
468 | |
469 | |
470 typedef struct { | |
471 unsigned long addr; | |
472 unsigned long offset; | |
473 enum x86_insn_group group; | |
474 enum x86_insn_type type; | |
475 enum x86_insn_note note; | |
476 unsigned char bytes[MAX_INSN_SIZE]; | |
477 unsigned char size; | |
478 unsigned char addr_size; | |
479 unsigned char op_size; | |
480 enum x86_insn_cpu cpu; | |
481 enum x86_insn_isa isa; | |
482 enum x86_flag_status flags_set; | |
483 enum x86_flag_status flags_tested; | |
484 unsigned char stack_mod; | |
485 long stack_mod_val; | |
486 enum x86_insn_prefix prefix; | |
487 char prefix_string[MAX_PREFIX_STR]; | |
488 char mnemonic[MAX_MNEM_STR]; | |
489 x86_oplist_t *operands; | |
490 size_t operand_count; | |
491 size_t explicit_count; | |
492 void *block; | |
493 void *function; | |
494 int tag; | |
495 } x86_insn_t; | |
496 | |
497 typedef void (*x86_operand_fn)(x86_op_t *op, x86_insn_t *insn, | |
498 void *arg); | |
499 | |
500 enum x86_op_foreach_type { | |
501 op_any = 0, | |
502 op_dest = 1, | |
503 op_src = 2, | |
504 op_ro = 3, | |
505 op_wo = 4, | |
506 op_xo = 5, | |
507 op_rw = 6, | |
508 op_implicit = 0x10, | |
509 op_explicit = 0x20 | |
510 }; | |
511 | |
512 size_t x86_operand_count( x86_insn_t *insn, | |
513 enum x86_op_foreach_type type ); | |
514 x86_op_t * x86_operand_1st( x86_insn_t *insn ); | |
515 x86_op_t * x86_operand_2nd( x86_insn_t *insn ); | |
516 x86_op_t * x86_operand_3rd( x86_insn_t *insn ); | |
517 long x86_get_rel_offset( x86_insn_t *insn ); | |
518 x86_op_t * x86_get_branch_target( x86_insn_t *insn ); | |
519 x86_op_t * x86_get_imm( x86_insn_t *insn ); | |
520 unsigned char * x86_get_raw_imm( x86_insn_t *insn ); | |
521 void x86_set_insn_addr( x86_insn_t *insn, unsigned long addr ); | |
522 int x86_format_mnemonic(x86_insn_t *insn, char *buf, int len, | |
523 enum x86_asm_format format); | |
524 int x86_format_insn(x86_insn_t *insn, char *buf, int len, | |
525 enum x86_asm_format); | |
526 void x86_oplist_free( x86_insn_t *insn ); | |
527 int x86_insn_is_valid( x86_insn_t *insn ); | |
528 %} | |
529 | |
530 %extend x86_insn_t { | |
531 x86_insn_t() { | |
532 x86_insn_t *insn = (x86_insn_t *) | |
533 calloc( sizeof(x86_insn_t), 1 ); | |
534 return insn; | |
535 } | |
536 ~x86_insn_t() { | |
537 x86_oplist_free( self ); | |
538 free( self ); | |
539 } | |
540 | |
541 int is_valid( ) { | |
542 return x86_insn_is_valid( self ); | |
543 } | |
544 | |
545 x86_op_t * operand_1st() { | |
546 return x86_operand_1st( self ); | |
547 } | |
548 | |
549 x86_op_t * operand_2nd() { | |
550 return x86_operand_2nd( self ); | |
551 } | |
552 | |
553 x86_op_t * operand_3rd() { | |
554 return x86_operand_3rd( self ); | |
555 } | |
556 | |
557 x86_op_t * operand_dest() { | |
558 return x86_operand_1st( self ); | |
559 } | |
560 | |
561 x86_op_t * operand_src() { | |
562 return x86_operand_2nd( self ); | |
563 } | |
564 | |
565 size_t num_operands( enum x86_op_foreach_type type ) { | |
566 return x86_operand_count( self, type ); | |
567 } | |
568 | |
569 long rel_offset() { | |
570 return x86_get_rel_offset( self ); | |
571 } | |
572 | |
573 x86_op_t * branch_target() { | |
574 return x86_get_branch_target( self ); | |
575 } | |
576 | |
577 x86_op_t * imm() { | |
578 return x86_get_imm( self ); | |
579 } | |
580 | |
581 unsigned char * raw_imm() { | |
582 return x86_get_raw_imm( self ); | |
583 } | |
584 | |
585 %newobject format; | |
586 char * format( enum x86_asm_format format ) { | |
587 char *buf, *str; | |
588 size_t len; | |
589 | |
590 switch ( format ) { | |
591 case xml_syntax: | |
592 len = MAX_INSN_XML_STRING; | |
593 break; | |
594 case raw_syntax: | |
595 len = MAX_INSN_RAW_STRING; | |
596 break; | |
597 case native_syntax: | |
598 case intel_syntax: | |
599 case att_syntax: | |
600 case unknown_syntax: | |
601 default: | |
602 len = MAX_INSN_STRING; | |
603 break; | |
604 } | |
605 | |
606 buf = (char * ) calloc( len + 1, 1 ); | |
607 x86_format_insn( self, buf, len, format ); | |
608 | |
609 /* drop buffer down to a reasonable size */ | |
610 str = strdup( buf ); | |
611 free(buf); | |
612 return str; | |
613 } | |
614 | |
615 %newobject format_mnemonic; | |
616 char * format_mnemonic( enum x86_asm_format format ) { | |
617 char *buf, *str; | |
618 size_t len = MAX_MNEM_STR + MAX_PREFIX_STR + 4; | |
619 | |
620 buf = (char * ) calloc( len, 1 ); | |
621 x86_format_mnemonic( self, buf, len, format ); | |
622 | |
623 /* drop buffer down to a reasonable size */ | |
624 str = strdup( buf ); | |
625 free(buf); | |
626 | |
627 return str; | |
628 } | |
629 | |
630 %newobject copy; | |
631 x86_insn_t * copy() { | |
632 x86_oplist_t *ptr, *list, *last = NULL; | |
633 x86_insn_t *insn = (x86_insn_t *) | |
634 calloc( sizeof(x86_insn_t), 1 ); | |
635 | |
636 if ( insn ) { | |
637 memcpy( insn, self, sizeof(x86_insn_t) ); | |
638 insn->operands = NULL; | |
639 insn->block = NULL; | |
640 insn->function = NULL; | |
641 | |
642 /* copy operand list */ | |
643 for ( list = self->operands; list; list = list->next ) { | |
644 ptr = x86_oplist_node_copy( list ); | |
645 | |
646 if (! ptr ) { | |
647 continue; | |
648 } | |
649 | |
650 if ( insn->operands ) { | |
651 last->next = ptr; | |
652 } else { | |
653 insn->operands = ptr; | |
654 } | |
655 last = ptr; | |
656 } | |
657 } | |
658 | |
659 return insn; | |
660 } | |
661 | |
662 X86_OpList * operand_list( ) { | |
663 x86_oplist_t *list = self->operands; | |
664 X86_OpList *op_list = new_X86_OpList(); | |
665 | |
666 for ( list = self->operands; list; list = list->next ) { | |
667 X86_OpList_append( op_list, &list->op ); | |
668 } | |
669 | |
670 return op_list; | |
671 } | |
672 } | |
673 | |
674 /* ================================================================== */ | |
675 /* invariant instruction class */ | |
676 %inline %{ | |
677 #define X86_WILDCARD_BYTE 0xF4 | |
678 | |
679 typedef struct { | |
680 enum x86_op_type type; | |
681 enum x86_op_datatype datatype; | |
682 enum x86_op_access access; | |
683 enum x86_op_flags flags; | |
684 } x86_invariant_op_t; | |
685 | |
686 typedef struct { | |
687 unsigned char bytes[64]; | |
688 unsigned int size; | |
689 enum x86_insn_group group; | |
690 enum x86_insn_type type; | |
691 x86_invariant_op_t operands[3]; | |
692 } x86_invariant_t; | |
693 %} | |
694 | |
695 %extend x86_invariant_t { | |
696 | |
697 x86_invariant_t() { | |
698 x86_invariant_t *inv = (x86_invariant_t *) | |
699 calloc( sizeof(x86_invariant_t), 1 ); | |
700 return inv; | |
701 } | |
702 | |
703 ~x86_invariant_t() { | |
704 free( self ); | |
705 } | |
706 } | |
707 | |
708 /* ================================================================== */ | |
709 /* instruction list class */ | |
710 %inline %{ | |
711 typedef struct X86_InsnListNode { | |
712 x86_insn_t *insn; | |
713 struct X86_InsnListNode *next, *prev; | |
714 } X86_InsnListNode; | |
715 | |
716 typedef struct X86_InsnList { | |
717 size_t count; | |
718 X86_InsnListNode *head, *tail, *curr; | |
719 } X86_InsnList; | |
720 %} | |
721 | |
722 %extend X86_InsnList { | |
723 X86_InsnList () { | |
724 X86_InsnList *list = (X86_InsnList *) | |
725 calloc( sizeof(X86_InsnList), 1 ); | |
726 list->count = 0; | |
727 return list; | |
728 } | |
729 | |
730 ~X86_InsnList() { | |
731 X86_InsnListNode *node, *next; | |
732 | |
733 node = self->head; | |
734 while ( node ) { | |
735 next = node->next; | |
736 /* free( node->insn ); */ | |
737 free( node ); | |
738 node = next; | |
739 } | |
740 | |
741 free( self ); | |
742 } | |
743 | |
744 X86_InsnListNode * first() { return self->head; } | |
745 | |
746 X86_InsnListNode * last() { return self->tail; } | |
747 | |
748 X86_InsnListNode * next() { | |
749 if (! self->curr ) { | |
750 self->curr = self->head; | |
751 return self->head; | |
752 } | |
753 | |
754 self->curr = self->curr->next; | |
755 return self->curr; | |
756 } | |
757 | |
758 X86_InsnListNode * prev() { | |
759 if (! self->curr ) { | |
760 self->curr = self->tail; | |
761 return self->tail; | |
762 } | |
763 | |
764 self->curr = self->curr->prev; | |
765 return self->curr; | |
766 } | |
767 | |
768 %newobject append; | |
769 void append( x86_insn_t *insn ) { | |
770 X86_InsnListNode *node = (X86_InsnListNode *) | |
771 calloc( sizeof(X86_InsnListNode) , 1 ); | |
772 if (! node ) { | |
773 return; | |
774 } | |
775 | |
776 self->count++; | |
777 if ( ! self->tail ) { | |
778 self->head = self->tail = node; | |
779 } else { | |
780 self->tail->next = node; | |
781 node->prev = self->tail; | |
782 self->tail = node; | |
783 } | |
784 | |
785 node->insn = x86_insn_t_copy( insn ); | |
786 } | |
787 } | |
788 | |
789 /* ================================================================== */ | |
790 /* address table class */ | |
791 /* slight TODO */ | |
792 | |
793 /* ================================================================== */ | |
794 /* Main disassembler class */ | |
795 %inline %{ | |
796 | |
797 enum x86_options { | |
798 opt_none= 0, | |
799 opt_ignore_nulls=1, | |
800 opt_16_bit=2 | |
801 }; | |
802 enum x86_report_codes { | |
803 report_disasm_bounds, | |
804 report_insn_bounds, | |
805 report_invalid_insn, | |
806 report_unknown | |
807 }; | |
808 | |
809 | |
810 typedef struct { | |
811 enum x86_report_codes last_error; | |
812 void * last_error_data; | |
813 void * disasm_callback; | |
814 void * disasm_resolver; | |
815 } X86_Disasm; | |
816 | |
817 typedef void (*DISASM_REPORTER)( enum x86_report_codes code, | |
818 void *data, void *arg ); | |
819 typedef void (*DISASM_CALLBACK)( x86_insn_t *insn, void * arg ); | |
820 typedef long (*DISASM_RESOLVER)( x86_op_t *op, | |
821 x86_insn_t * current_insn, | |
822 void *arg ); | |
823 | |
824 void x86_report_error( enum x86_report_codes code, void *data ); | |
825 int x86_init( enum x86_options options, DISASM_REPORTER reporter, | |
826 void *arg); | |
827 void x86_set_reporter( DISASM_REPORTER reporter, void *arg); | |
828 void x86_set_options( enum x86_options options ); | |
829 enum x86_options x86_get_options( void ); | |
830 int x86_cleanup(void); | |
831 int x86_format_header( char *buf, int len, enum x86_asm_format format); | |
832 unsigned int x86_endian(void); | |
833 unsigned int x86_addr_size(void); | |
834 unsigned int x86_op_size(void); | |
835 unsigned int x86_word_size(void); | |
836 unsigned int x86_max_insn_size(void); | |
837 unsigned int x86_sp_reg(void); | |
838 unsigned int x86_fp_reg(void); | |
839 unsigned int x86_ip_reg(void); | |
840 size_t x86_invariant_disasm( unsigned char *buf, int buf_len, | |
841 x86_invariant_t *inv ); | |
842 size_t x86_size_disasm( unsigned char *buf, unsigned int buf_len ); | |
843 int x86_disasm( unsigned char *buf, unsigned int buf_len, | |
844 unsigned long buf_rva, unsigned int offset, | |
845 x86_insn_t * insn ); | |
846 int x86_disasm_range( unsigned char *buf, unsigned long buf_rva, | |
847 unsigned int offset, unsigned int len, | |
848 DISASM_CALLBACK func, void *arg ); | |
849 int x86_disasm_forward( unsigned char *buf, unsigned int buf_len, | |
850 unsigned long buf_rva, unsigned int offset, | |
851 DISASM_CALLBACK func, void *arg, | |
852 DISASM_RESOLVER resolver, void *r_arg ); | |
853 | |
854 void x86_default_reporter( enum x86_report_codes code, | |
855 void *data, void *arg ) { | |
856 X86_Disasm *dis = (X86_Disasm *) arg; | |
857 if ( dis ) { | |
858 dis->last_error = code; | |
859 dis->last_error_data = data; | |
860 } | |
861 } | |
862 | |
863 void x86_default_callback( x86_insn_t *insn, void *arg ) { | |
864 X86_InsnList *list = (X86_InsnList *) arg; | |
865 if ( list ) { | |
866 X86_InsnList_append( list, insn ); | |
867 } | |
868 } | |
869 | |
870 /* TODO: resolver stack, maybe a callback */ | |
871 long x86_default_resolver( x86_op_t *op, x86_insn_t *insn, void *arg ) { | |
872 X86_Disasm *dis = (X86_Disasm *) arg; | |
873 if ( dis ) { | |
874 //return dis->resolver( op, insn ); | |
875 return 0; | |
876 } | |
877 | |
878 return 0; | |
879 } | |
880 | |
881 %} | |
882 | |
883 %extend X86_Disasm { | |
884 | |
885 X86_Disasm( ) { | |
886 X86_Disasm * dis = (X86_Disasm *) | |
887 calloc( sizeof( X86_Disasm ), 1 ); | |
888 x86_init( opt_none, x86_default_reporter, dis ); | |
889 return dis; | |
890 } | |
891 | |
892 X86_Disasm( enum x86_options options ) { | |
893 X86_Disasm * dis = (X86_Disasm *) | |
894 calloc( sizeof( X86_Disasm ), 1 ); | |
895 x86_init( options, x86_default_reporter, dis ); | |
896 return dis; | |
897 } | |
898 | |
899 X86_Disasm( enum x86_options options, DISASM_REPORTER reporter ) { | |
900 X86_Disasm * dis = (X86_Disasm *) | |
901 calloc( sizeof( X86_Disasm ), 1 ); | |
902 x86_init( options, reporter, NULL ); | |
903 return dis; | |
904 } | |
905 | |
906 X86_Disasm( enum x86_options options, DISASM_REPORTER reporter, | |
907 void * arg ) { | |
908 X86_Disasm * dis = (X86_Disasm *) | |
909 calloc( sizeof( X86_Disasm ), 1 ); | |
910 x86_init( options, reporter, arg ); | |
911 return dis; | |
912 } | |
913 | |
914 ~X86_Disasm() { | |
915 x86_cleanup(); | |
916 free( self ); | |
917 } | |
918 | |
919 void set_options( enum x86_options options ) { | |
920 return x86_set_options( options ); | |
921 } | |
922 | |
923 enum x86_options options() { | |
924 return x86_get_options(); | |
925 } | |
926 | |
927 void set_callback( void * callback ) { | |
928 self->disasm_callback = callback; | |
929 } | |
930 | |
931 void set_resolver( void * callback ) { | |
932 self->disasm_resolver = callback; | |
933 } | |
934 | |
935 void report_error( enum x86_report_codes code ) { | |
936 x86_report_error( code, NULL ); | |
937 } | |
938 | |
939 %newobject disasm; | |
940 x86_insn_t * disasm( unsigned char *buf, size_t buf_len, | |
941 unsigned long buf_rva, unsigned int offset ) { | |
942 x86_insn_t *insn = calloc( sizeof( x86_insn_t ), 1 ); | |
943 x86_disasm( buf, buf_len, buf_rva, offset, insn ); | |
944 return insn; | |
945 } | |
946 | |
947 int disasm_range( unsigned char *buf, size_t buf_len, | |
948 unsigned long buf_rva, unsigned int offset, | |
949 unsigned int len ) { | |
950 | |
951 X86_InsnList *list = new_X86_InsnList(); | |
952 | |
953 if ( len > buf_len ) { | |
954 len = buf_len; | |
955 } | |
956 | |
957 return x86_disasm_range( buf, buf_rva, offset, len, | |
958 x86_default_callback, list ); | |
959 } | |
960 | |
961 int disasm_forward( unsigned char *buf, size_t buf_len, | |
962 unsigned long buf_rva, unsigned int offset ) { | |
963 X86_InsnList *list = new_X86_InsnList(); | |
964 | |
965 /* use default resolver: damn SWIG callbacks! */ | |
966 return x86_disasm_forward( buf, buf_len, buf_rva, offset, | |
967 x86_default_callback, list, | |
968 x86_default_resolver, NULL ); | |
969 } | |
970 | |
971 size_t disasm_invariant( unsigned char *buf, size_t buf_len, | |
972 x86_invariant_t *inv ) { | |
973 return x86_invariant_disasm( buf, buf_len, inv ); | |
974 } | |
975 | |
976 size_t disasm_size( unsigned char *buf, size_t buf_len ) { | |
977 return x86_size_disasm( buf, buf_len ); | |
978 } | |
979 | |
980 %newobject format_header; | |
981 char * format_header( enum x86_asm_format format) { | |
982 char *buf, *str; | |
983 size_t len; | |
984 | |
985 switch ( format ) { | |
986 /* these were obtained from x86_format.c */ | |
987 case xml_syntax: | |
988 len = 679; break; | |
989 case raw_syntax: | |
990 len = 172; break; | |
991 case native_syntax: | |
992 len = 35; break; | |
993 case intel_syntax: | |
994 len = 23; break; | |
995 case att_syntax: | |
996 len = 23; break; | |
997 case unknown_syntax: | |
998 default: | |
999 len = 23; break; | |
1000 } | |
1001 | |
1002 buf = (char * ) calloc( len + 1, 1 ); | |
1003 x86_format_header( buf, len, format ); | |
1004 | |
1005 return buf; | |
1006 } | |
1007 | |
1008 unsigned int endian() { | |
1009 return x86_endian(); | |
1010 } | |
1011 | |
1012 unsigned int addr_size() { | |
1013 return x86_addr_size(); | |
1014 } | |
1015 | |
1016 unsigned int op_size() { | |
1017 return x86_op_size(); | |
1018 } | |
1019 | |
1020 unsigned int word_size() { | |
1021 return x86_word_size(); | |
1022 } | |
1023 | |
1024 unsigned int max_insn_size() { | |
1025 return x86_max_insn_size(); | |
1026 } | |
1027 | |
1028 unsigned int sp_reg() { | |
1029 return x86_sp_reg(); | |
1030 } | |
1031 | |
1032 unsigned int fp_reg() { | |
1033 return x86_fp_reg(); | |
1034 } | |
1035 | |
1036 unsigned int ip_reg() { | |
1037 return x86_ip_reg(); | |
1038 } | |
1039 | |
1040 %newobject reg_from_id; | |
1041 x86_reg_t * reg_from_id( unsigned int id ) { | |
1042 x86_reg_t * reg = calloc( sizeof(x86_reg_t), 1 ); | |
1043 x86_reg_from_id( id, reg ); | |
1044 return reg; | |
1045 } | |
1046 | |
1047 unsigned char wildcard_byte() { return X86_WILDCARD_BYTE; } | |
1048 | |
1049 int max_register_string() { return MAX_REGNAME; } | |
1050 | |
1051 int max_prefix_string() { return MAX_PREFIX_STR; } | |
1052 | |
1053 int max_mnemonic_string() { return MAX_MNEM_STR; } | |
1054 | |
1055 int max_operand_string( enum x86_asm_format format ) { | |
1056 switch ( format ) { | |
1057 case xml_syntax: | |
1058 return MAX_OP_XML_STRING; | |
1059 break; | |
1060 case raw_syntax: | |
1061 return MAX_OP_RAW_STRING; | |
1062 break; | |
1063 case native_syntax: | |
1064 case intel_syntax: | |
1065 case att_syntax: | |
1066 case unknown_syntax: | |
1067 default: | |
1068 return MAX_OP_STRING; | |
1069 break; | |
1070 } | |
1071 } | |
1072 | |
1073 | |
1074 int max_insn_string( enum x86_asm_format format ) { | |
1075 switch ( format ) { | |
1076 case xml_syntax: | |
1077 return MAX_INSN_XML_STRING; | |
1078 break; | |
1079 case raw_syntax: | |
1080 return MAX_INSN_RAW_STRING; | |
1081 break; | |
1082 case native_syntax: | |
1083 case intel_syntax: | |
1084 case att_syntax: | |
1085 case unknown_syntax: | |
1086 default: | |
1087 return MAX_INSN_STRING; | |
1088 break; | |
1089 } | |
1090 } | |
1091 | |
1092 int max_num_operands( ) { return MAX_NUM_OPERANDS; } | |
1093 } | |
1094 | |
1095 /* python callback, per the manual */ | |
1096 /*%typemap(python,in) PyObject *pyfunc { | |
1097 if (!PyCallable_Check($source)) { | |
1098 PyErr_SetString(PyExc_TypeError, "Need a callable object!"); | |
1099 return NULL; | |
1100 } | |
1101 $target = $source; | |
1102 }*/ | |
1103 | |
1104 /* python FILE * callback, per the manual */ | |
1105 /* | |
1106 %typemap(python,in) FILE * { | |
1107 if (!PyFile_Check($source)) { | |
1108 PyErr_SetString(PyExc_TypeError, "Need a file!"); | |
1109 return NULL; | |
1110 } | |
1111 $target = PyFile_AsFile($source); | |
1112 }*/ | |
1113 | |
1114 | |
OLD | NEW |