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
|
(****************************************************************************)
(* the diy toolsuite *)
(* *)
(* Jade Alglave, University College London, UK. *)
(* Luc Maranget, INRIA Paris-Rocquencourt, France. *)
(* *)
(* Copyright 2010-present Institut National de Recherche en Informatique et *)
(* en Automatique, ARM Ltd and the authors. All rights reserved. *)
(* *)
(* This software is governed by the CeCILL-B license under French law and *)
(* abiding by the rules of distribution of free software. You can use, *)
(* modify and/ or redistribute the software under the terms of the CeCILL-B *)
(* license as circulated by CEA, CNRS and INRIA at the following URL *)
(* "http://www.cecill.info". We also give a copy in LICENSE.txt. *)
(****************************************************************************)
(************************************************)
(* "load" program in memory, somehow abstracted *)
(************************************************)
let func_size = Pseudo.func_size
let proc_size = 10000
let func_start_addr proc = function
| MiscParser.Main -> (proc + 1) * proc_size
| MiscParser.FaultHandler -> (proc + 1) * proc_size + func_size
module type S = sig
type nice_prog
type program
type start_points
type code_segment
val load : nice_prog -> program * start_points * code_segment
end
module Make(A:Arch_herd.S) =
struct
type nice_prog = A.nice_prog
type program = A.program
type start_points = A.start_points
type code_segment = A.code_segment
let preload_labels proc =
let add_label lbl addr m =
if Label.Map.mem lbl m then
Warn.user_error
"Label %s occurs more that once" lbl ;
Label.Map.add lbl (proc,addr) m in
A.fold_label_addr add_label
let preload =
List.fold_left
(fun m ((proc,_,func),code) ->
let addr = func_start_addr proc func in
preload_labels proc m addr code)
Label.Map.empty
let convert_lbl_to_offset proc pc mem instr =
let labelmap =
let open BranchTarget in
function
| Lbl l ->
let tgt_proc, tgt_addr =
try Label.Map.find l mem
with Not_found ->
Warn.user_error
"Label %s not found on %s, although used in the instruction %s"
(Label.pp l)
(Proc.pp proc)
(A.dump_instruction instr) in
if Proc.equal tgt_proc proc then
Offset (tgt_addr - pc)
else
Warn.user_error
"%s cannot refer to %s defined by %s, use register with initial value %s"
(Proc.pp proc) (Label.pp l)
(Proc.pp tgt_proc) (Label.Full.pp (tgt_proc,l))
| Offset _ as x -> x in
A.map_labels_base labelmap instr
let rec load_code proc addr mem rets = function
| [] ->
[],IntMap.add addr (proc,[]) rets
| ins::code ->
load_ins proc addr mem rets code ins
and load_ins proc addr mem rets code = fun x ->
match x with
| A.Nop ->
load_code proc addr mem rets code
| A.Instruction ins ->
let start,new_rets =
load_code proc (addr+A.size_of_ins ins) mem rets code in
let new_ins =
convert_lbl_to_offset proc addr mem ins in
let new_start = (addr,new_ins)::start in
let newer_rets = IntMap.add addr (proc,new_start) new_rets in
new_start,newer_rets
| A.Label (_,ins) ->
let start,new_rets = load_ins proc addr mem rets code ins in
start,new_rets
| A.Symbolic _
| A.Macro (_,_) -> assert false
let load prog =
let mem = preload prog in
let rec load_iter = function
| [] -> [],IntMap.empty
| ((proc,_,func),code)::prog ->
let starts,rets = load_iter prog in
let addr = func_start_addr proc func in
let start,fin_rets = load_code proc addr mem rets code in
(proc,func,start)::starts,fin_rets in
let starts,codes = load_iter prog in
let mains,fhandlers =
List.partition (fun (_,func,_) -> func=MiscParser.Main) starts in
let add_fhandler (proc,_,start) =
let fhandler =
List.find_opt (fun (p,_,_) -> Proc.equal p proc) fhandlers in
match fhandler with
| Some (_,_,fh_start) ->
(proc,start,Some fh_start)
| None -> (proc,start,None) in
Label.Map.map snd mem,List.map add_fhandler mains,codes
end
|
