|
261
|
1 import struct
|
|
342
|
2 from ..basetarget import Register, Instruction, Target, Label, Alignment
|
|
346
|
3 from .instructions import Add2, Sub, Sub3, Add3, Cmp, Lsl, Orr, Add, Cmp2, Sub2, Add2, Mul, And
|
|
342
|
4 from .instructions import Dcd, Pop, Push, Yield, Mov2, Mov3
|
|
|
5 from .instructions import B, Bl, Bne, Beq, Blt, Bgt
|
|
346
|
6 from .instructions import Ldr, Str2, Ldr2, Str1, Ldr1, Ldr3
|
|
202
|
7
|
|
346
|
8 from .frame import ArmFrame
|
|
322
|
9 from .arminstructionselector import ArmInstructionSelector
|
|
342
|
10 from ..arm.registers import R0, R1, R2, R3, R4, R5, R6, R7, SP, LR, PC
|
|
|
11 from ..arm.registers import register_range
|
|
381
|
12 from ...assembler import BaseAssembler
|
|
341
|
13
|
|
290
|
14
|
|
|
15 """ ARM target description. """
|
|
287
|
16
|
|
218
|
17 # TODO: encode this in DSL (domain specific language)
|
|
275
|
18 # TBD: is this required?
|
|
292
|
19 # TODO: make a difference between armv7 and armv5?
|
|
219
|
20
|
|
381
|
21
|
|
|
22 class ThumbAssembler(BaseAssembler):
|
|
|
23 def __init__(self, target):
|
|
|
24 super().__init__(target)
|
|
|
25 self.make_parser()
|
|
277
|
26
|
|
341
|
27
|
|
|
28 class ThumbTarget(Target):
|
|
292
|
29 def __init__(self):
|
|
341
|
30 super().__init__('thumb')
|
|
322
|
31 self.ins_sel = ArmInstructionSelector()
|
|
|
32 self.FrameClass = ArmFrame
|
|
341
|
33 self.add_rules()
|
|
|
34
|
|
346
|
35 # Add lowering options:
|
|
|
36 self.add_lowering(Str2, lambda im: Str2(im.src[1], im.src[0], im.others[0]))
|
|
|
37 self.add_lowering(Ldr2, lambda im: Ldr2(im.dst[0], im.src[0], im.others[0]))
|
|
|
38 self.add_lowering(Ldr3, lambda im: Ldr3(im.dst[0], im.others[0]))
|
|
|
39 self.add_lowering(Mov3, lambda im: Mov3(im.dst[0], im.others[0]))
|
|
|
40 self.add_lowering(Add2, lambda im: Add2(im.dst[0], im.src[0], im.others[0]))
|
|
|
41 self.add_lowering(Sub2, lambda im: Sub2(im.dst[0], im.src[0], im.others[0]))
|
|
|
42 self.add_lowering(Mov2, lambda im: Mov2(im.dst[0], im.src[0]))
|
|
|
43 self.add_lowering(Add3, lambda im: Add3(im.dst[0], im.src[0], im.src[1]))
|
|
|
44 self.add_lowering(Sub3, lambda im: Sub3(im.dst[0], im.src[0], im.src[1]))
|
|
|
45 self.add_lowering(Mul, lambda im: Mul(im.src[0], im.dst[0]))
|
|
|
46 self.add_lowering(And, lambda im: And(im.src[0], im.src[1]))
|
|
|
47 self.add_lowering(Orr, lambda im: Orr(im.src[0], im.src[1]))
|
|
|
48 self.add_lowering(Lsl, lambda im: Lsl(im.src[0], im.src[1]))
|
|
|
49 self.add_lowering(Cmp, lambda im: Cmp(im.src[0], im.src[1]))
|
|
|
50
|
|
381
|
51 self.assembler = ThumbAssembler(self)
|
|
|
52
|
|
341
|
53 def add_rules(self):
|
|
|
54
|
|
|
55 # Add instructions:
|
|
|
56 self.add_keyword('dcd')
|
|
|
57 self.add_instruction(['dcd', 'imm32'], lambda rhs: Dcd(rhs[1]))
|
|
|
58
|
|
|
59 self.add_keyword('mov')
|
|
|
60 self.add_instruction(['mov', 'reg8', ',', 'reg8'],
|
|
|
61 lambda rhs: Mov2(rhs[1], rhs[3]))
|
|
|
62
|
|
|
63 self.add_instruction(['mov', 'reg8', ',', 'imm8'],
|
|
|
64 lambda rhs: Mov3(rhs[1], rhs[3]))
|
|
|
65
|
|
|
66 self.add_keyword('add')
|
|
|
67 self.add_instruction(['add', 'reg8', ',', 'reg8', ',', 'imm3'],
|
|
|
68 lambda rhs: Add2(rhs[1], rhs[3], rhs[5]))
|
|
|
69
|
|
|
70 self.add_instruction(['add', 'reg8', ',', 'reg8', ',', 'reg8'],
|
|
|
71 lambda rhs: Add3(rhs[1], rhs[3], rhs[5]))
|
|
|
72
|
|
|
73 self.add_keyword('sub')
|
|
|
74 self.add_instruction(['sub', 'reg8', ',', 'reg8', ',', 'imm3'],
|
|
|
75 lambda rhs: Sub(rhs[1], rhs[3], rhs[5]))
|
|
|
76
|
|
|
77 self.add_instruction(['sub', 'sp', ',', 'sp', ',', 'imm8'],
|
|
|
78 lambda rhs: Sub(SP, SP, rhs[5]))
|
|
|
79
|
|
|
80 self.add_instruction(['add', 'sp', ',', 'sp', ',', 'imm8'],
|
|
342
|
81 lambda rhs: Add(SP, SP, rhs[5]))
|
|
341
|
82
|
|
|
83 self.add_keyword('cmp')
|
|
|
84 self.add_instruction(['cmp', 'reg8', ',', 'reg8'],
|
|
|
85 lambda rhs: Cmp(rhs[1], rhs[3]))
|
|
342
|
86 self.add_instruction(['cmp', 'reg8', ',', 'imm8'],
|
|
|
87 lambda rhs: Cmp2(rhs[1], rhs[3]))
|
|
341
|
88
|
|
|
89 self.add_keyword('lsl')
|
|
|
90 self.add_instruction(['lsl', 'reg8', ',', 'reg8'],
|
|
|
91 lambda rhs: Lsl(rhs[1], rhs[3]))
|
|
|
92
|
|
|
93 self.add_keyword('str')
|
|
|
94 self.add_instruction(['str', 'reg8', ',', '[', 'reg8', '+', 'imm5', ']'],
|
|
|
95 lambda rhs: Str2(rhs[1], rhs[4], rhs[6]))
|
|
|
96
|
|
|
97 self.add_keyword('ldr')
|
|
|
98 self.add_instruction(['ldr', 'reg8', ',', '[', 'reg8', '+', 'imm5', ']'],
|
|
|
99 lambda rhs: Ldr2(rhs[1], rhs[4], rhs[6]))
|
|
|
100
|
|
|
101 self.add_instruction(['str', 'reg8', ',', '[', 'sp', '+', 'imm8', ']'],
|
|
|
102 lambda rhs: Str1(rhs[1], rhs[6]))
|
|
|
103
|
|
|
104 self.add_instruction(['ldr', 'reg8', ',', '[', 'sp', '+', 'imm8', ']'],
|
|
|
105 lambda rhs: Ldr1(rhs[1], rhs[6]))
|
|
|
106
|
|
|
107 self.add_keyword('pop')
|
|
|
108 self.add_instruction(['pop', 'reg_list'], lambda rhs: Pop(rhs[1]))
|
|
|
109 self.add_keyword('push')
|
|
|
110 self.add_instruction(['push', 'reg_list'], lambda rhs: Push(rhs[1]))
|
|
|
111
|
|
|
112 self.add_keyword('yield')
|
|
|
113 self.add_instruction(['yield'], lambda rhs: Yield())
|
|
|
114
|
|
|
115 self.add_keyword('b')
|
|
|
116 self.add_keyword('bl')
|
|
|
117 self.add_instruction(['b', 'ID'], lambda rhs: B(rhs[1].val))
|
|
|
118 self.add_instruction(['bl', 'ID'], lambda rhs: Bl(rhs[1].val))
|
|
|
119 self.add_keyword('beq')
|
|
|
120 self.add_keyword('bne')
|
|
|
121 self.add_keyword('blt')
|
|
|
122 self.add_keyword('bgt')
|
|
|
123 self.add_instruction(['beq', 'ID'], lambda rhs: Beq(rhs[1].val))
|
|
|
124 self.add_instruction(['bne', 'ID'], lambda rhs: Bne(rhs[1].val))
|
|
|
125 self.add_instruction(['blt', 'ID'], lambda rhs: Blt(rhs[1].val))
|
|
|
126 self.add_instruction(['bgt', 'ID'], lambda rhs: Bgt(rhs[1].val))
|
|
|
127
|
|
|
128 self.add_keyword('align')
|
|
|
129 self.add_instruction(['align', 'imm8'], lambda rhs: Alignment(rhs[1]))
|
|
|
130
|
|
|
131 self.add_instruction(['ldr', 'reg8', ',', 'ID'],
|
|
|
132 lambda rhs: Ldr(rhs[1], rhs[3].val))
|
|
|
133
|
|
|
134 # Additional rules:
|
|
|
135
|
|
|
136 # Register list grammar:
|
|
|
137 self.add_rule('reg_list', ['{', 'reg_list_inner', '}'],
|
|
|
138 lambda rhs: rhs[1])
|
|
|
139 self.add_rule('reg_list_inner', ['reg_or_range'],
|
|
|
140 lambda rhs: rhs[0])
|
|
|
141 self.add_rule('reg_list_inner', ['reg_or_range', ',', 'reg_list_inner'],
|
|
|
142 lambda rhs: rhs[0] | rhs[2])
|
|
|
143 self.add_rule('reg_or_range', ['reg8'], lambda rhs: {rhs[0]})
|
|
|
144 self.add_rule('reg_or_range', ['lr'], lambda rhs: {LR})
|
|
|
145 self.add_rule('reg_or_range', ['pc'], lambda rhs: {PC})
|
|
|
146
|
|
|
147 self.add_rule('reg_or_range', ['reg8', '-', 'reg8'],
|
|
|
148 lambda rhs: register_range(rhs[0], rhs[2]))
|
|
|
149
|
|
|
150 self.add_keyword('r0')
|
|
|
151 self.add_keyword('r1')
|
|
|
152 self.add_keyword('r2')
|
|
|
153 self.add_keyword('r3')
|
|
|
154 self.add_keyword('r4')
|
|
|
155 self.add_keyword('r5')
|
|
|
156 self.add_keyword('r6')
|
|
|
157 self.add_keyword('r7')
|
|
|
158 self.add_keyword('sp')
|
|
|
159 self.add_keyword('lr')
|
|
|
160 self.add_keyword('pc')
|
|
|
161 self.add_rule('reg8', ['r0'], lambda rhs: R0)
|
|
|
162 self.add_rule('reg8', ['r1'], lambda rhs: R1)
|
|
|
163 self.add_rule('reg8', ['r2'], lambda rhs: R2)
|
|
|
164 self.add_rule('reg8', ['r3'], lambda rhs: R3)
|
|
|
165 self.add_rule('reg8', ['r4'], lambda rhs: R4)
|
|
|
166 self.add_rule('reg8', ['r5'], lambda rhs: R5)
|
|
|
167 self.add_rule('reg8', ['r6'], lambda rhs: R6)
|
|
|
168 self.add_rule('reg8', ['r7'], lambda rhs: R7)
|
|
|
169
|
