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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
type register =
| ZERO | A0 | A1 | A2 | V0 | T0 | T1 | T2 | S0 | RA | SP | FP
type address =
| Alab of string
| Areg of int * register
type operand =
| Oimm of int
| Oreg of register
type arith = Add | Sub | Mul | Div | Rem
type condition = Eq | Ne | Le | Lt | Ge | Gt
type label = string
type instruction =
| Move of register * register
| Li of register * int
| Li32 of register * int32
| La of register * label
| Lw of register * address
| Sw of register * address
| Lb of register * address
| Sb of register * address
| Arith of arith * register * register * operand
| Neg of register * register
| Set of condition * register * register * operand
| B of label
| Beq of register * register * label
| Beqz of register * label
| Bnez of register * label
| J of string
| Jal of string
| Jr of register
| Jalr of register
| Syscall
| Label of string
| Inline of string
type word = Wint of int | Waddr of string
type data =
| Asciiz of string * string
| Word of string * word list
| Space of string * int
| Align of int
type code =
| Clist of instruction list
| Capp of code * code
let nop = Clist []
let mips l = Clist l
let inline s = Clist [Inline s]
let (++) c1 c2 = Capp (c1, c2)
type program = {
text : code;
data : data list;
}
open Format
let print_register fmt = function
| ZERO -> pp_print_string fmt "$0"
| A0 -> pp_print_string fmt "$a0"
| A1 -> pp_print_string fmt "$a1"
| A2 -> pp_print_string fmt "$a2"
| V0 -> pp_print_string fmt "$v0"
| T0 -> pp_print_string fmt "$t0"
| T1 -> pp_print_string fmt "$t1"
| T2 -> pp_print_string fmt "$t2"
| S0 -> pp_print_string fmt "$s0"
| RA -> pp_print_string fmt "$ra"
| SP -> pp_print_string fmt "$sp"
| FP -> pp_print_string fmt "$fp"
let print_arith fmt = function
| Add -> pp_print_string fmt "add"
| Sub -> pp_print_string fmt "sub"
| Mul -> pp_print_string fmt "mul"
| Div -> pp_print_string fmt "div"
| Rem -> pp_print_string fmt "rem"
let print_condition fmt = function
| Eq -> pp_print_string fmt "seq"
| Ne -> pp_print_string fmt "sne"
| Lt -> pp_print_string fmt "slt"
| Le -> pp_print_string fmt "sle"
| Gt -> pp_print_string fmt "sgt"
| Ge -> pp_print_string fmt "sge"
let print_address fmt = function
| Alab s -> pp_print_string fmt s
| Areg (ofs, r) -> fprintf fmt "%d(%a)" ofs print_register r
let print_operand fmt = function
| Oimm i -> pp_print_int fmt i
| Oreg r -> print_register fmt r
let print_instruction fmt = function
| Move (dst, src) ->
fprintf fmt "\tmove %a, %a\n" print_register dst print_register src
| Li (r, i) ->
fprintf fmt "\tli %a, %d\n" print_register r i
| Li32 (r, i) ->
fprintf fmt "\tli %a, %ld\n" print_register r i
| La (r, s) ->
fprintf fmt "\tla %a, %s\n" print_register r s
| Lw (r, a) ->
fprintf fmt "\tlw %a, %a\n" print_register r print_address a
| Sw (r, a) ->
fprintf fmt "\tsw %a, %a\n" print_register r print_address a
| Lb (r, a) ->
fprintf fmt "\tlb %a, %a\n" print_register r print_address a
| Sb (r, a) ->
fprintf fmt "\tsb %a, %a\n" print_register r print_address a
| Arith (a, dst, src, op) ->
fprintf fmt "\t%a %a, %a, %a\n"
print_arith a print_register dst print_register src print_operand op
| Neg (dst, src) ->
fprintf fmt "\tneg %a, %a\n" print_register dst print_register src
| Set (cond, dst, src, op) ->
fprintf fmt "\t%a %a, %a, %a\n"
print_condition cond print_register dst print_register src
print_operand op
| B l ->
fprintf fmt "\tb %s\n" l
| Beq (r1, r2, l) ->
fprintf fmt "\tbeq %a, %a, %s\n" print_register r1 print_register r2 l
| Beqz (r, l) ->
fprintf fmt "\tbeqz %a, %s\n" print_register r l
| Bnez (r, l) ->
fprintf fmt "\tbnez %a, %s\n" print_register r l
| J s ->
fprintf fmt "\tj %s\n" s
| Jal s ->
fprintf fmt "\tjal %s\n" s
| Jalr r ->
fprintf fmt "\tjalr %a\n" print_register r
| Jr r ->
fprintf fmt "\tjr %a\n" print_register r
| Syscall ->
fprintf fmt "\tsyscall\n"
| Label s ->
fprintf fmt "%s:\n" s
| Inline s ->
fprintf fmt "%s" s
let rec print_code fmt = function
| Clist l -> List.iter (print_instruction fmt) l
| Capp (c1, c2) -> print_code fmt c1; print_code fmt c2
let print_word fmt = function
| Wint n -> pp_print_int fmt n
| Waddr s -> pp_print_string fmt s
let rec print_list print fmt = function
| [] -> ()
| [x] -> print fmt x
| x :: r -> fprintf fmt "%a, %a" print x (print_list print) r
let print_data fmt = function
| Asciiz (l, s) ->
fprintf fmt "%s:\n\t.asciiz %S\n" l s
| Word (l, n) ->
fprintf fmt "%s:\n\t.word %a\n" l (print_list print_word) n
| Space (l, n) ->
fprintf fmt "%s:\n\t.space %d\n" l n
| Align n ->
fprintf fmt "\t.align %d\n" n
let print_program fmt p =
fprintf fmt "\t.text\n";
print_code fmt p.text;
fprintf fmt "\t.data\n";
List.iter (print_data fmt) p.data;
fprintf fmt "@."
|
