1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#-*- coding:utf-8 -*-
from miasm2.expression.expression import ExprAff, ExprOp, ExprId
from miasm2.core.graph import DiGraph
from miasm2.core.asmbloc import expr_is_label
from miasm2.ir.ir import ir, irbloc, AssignBlock
from miasm2.ir.analysis import ira
from miasm2.arch.x86.sem import ir_x86_16, ir_x86_32, ir_x86_64
class ir_a_x86_16(ir_x86_16, ira):
def __init__(self, symbol_pool=None):
ir_x86_16.__init__(self, symbol_pool)
self.ret_reg = self.arch.regs.AX
# for test XXX TODO
def set_dead_regs(self, b):
b.rw[-1][1].add(self.arch.regs.zf)
b.rw[-1][1].add(self.arch.regs.of)
b.rw[-1][1].add(self.arch.regs.pf)
b.rw[-1][1].add(self.arch.regs.cf)
b.rw[-1][1].add(self.arch.regs.nf)
b.rw[-1][1].add(self.arch.regs.af)
def get_out_regs(self, b):
return set([self.ret_reg, self.sp])
def add_unused_regs(self):
leaves = [self.blocs[n] for n in self.g.leafs()]
for b in leaves:
self.set_dead_regs(b)
def pre_add_instr(self, block, instr, irb_cur, ir_blocks_all, gen_pc_update):
if not instr.is_subcall():
return irb_cur
call_effects = self.call_effects(instr.args[0], instr)
for assignblk in call_effects:
irb_cur.irs.append(assignblk)
irb_cur.lines.append(instr)
return None
class ir_a_x86_32(ir_x86_32, ir_a_x86_16):
def __init__(self, symbol_pool=None):
ir_x86_32.__init__(self, symbol_pool)
self.ret_reg = self.arch.regs.EAX
def sizeof_char(self):
return 8
def sizeof_short(self):
return 16
def sizeof_int(self):
return 32
def sizeof_long(self):
return 32
def sizeof_pointer(self):
return 32
class ir_a_x86_64(ir_x86_64, ir_a_x86_16):
def __init__(self, symbol_pool=None):
ir_x86_64.__init__(self, symbol_pool)
self.ret_reg = self.arch.regs.RAX
def call_effects(self, ad, instr):
return [AssignBlock([ExprAff(self.ret_reg, ExprOp('call_func_ret', ad,
self.sp,
self.arch.regs.RCX,
self.arch.regs.RDX,
self.arch.regs.R8,
self.arch.regs.R9,
)),
ExprAff(self.sp, ExprOp('call_func_stack',
ad, self.sp)),
])]
def sizeof_char(self):
return 8
def sizeof_short(self):
return 16
def sizeof_int(self):
return 32
def sizeof_long(self):
return 64
def sizeof_pointer(self):
return 64
|