Load ELF sections to virtual memory addresses

Roughly 110KB/s (~5 seconds to load "hello" test).

1 files changed, 62 insertions, 19 deletions

diff --git a/src/test/run_power.ml b/src/test/run_power.ml
index b06f79e3..28345ae9 100644
--- a/src/test/run_power.ml
+++ b/src/test/run_power.ml
@@ -4,6 +4,25 @@ open Interp_ast ;;
 open Interp_lib ;;
 open Run_interp ;;
 
+let startaddr = ref 0 ;;
+let mainaddr  = ref 0 ;;
+let sections = ref [] ;;
+let file = ref "" ;;
+
+let add_section s =
+  match Str.split (Str.regexp ",") s with
+  | [name;offset;size;addr] ->
+      begin try
+        sections := (
+          int_of_string offset,
+          int_of_string size,
+          Big_int.big_int_of_int64 (Int64.of_string addr)) ::
+            !sections
+      with Failure msg -> raise (Arg.Bad (msg ^ ": " ^ s))
+      end
+  | _ -> raise (Arg.Bad ("Wrong section format: "^s))
+;;
+
 let rec foldli f acc ?(i=0) = function
   | [] -> acc
   | x::xs -> foldli f (f i acc x) ~i:(i+1) xs
@@ -12,30 +31,54 @@ let rec foldli f acc ?(i=0) = function
 (* POWER is big-endian *)
 let little_endian = false ;;
 
-let binary file =
-  let ic = open_in_bin file in
-  let rec read acc =
-    match try Some (input_byte ic) with End_of_file -> None
-    with
-    | Some b -> read ((V_lit (L_aux (L_num (Big_int.big_int_of_int b), Unknown))) :: acc)
-    | None -> close_in ic; acc in
-  let bytes = read [] in
-  List.rev_map ((if little_endian then to_vec_inc else to_vec_dec) 8) bytes
+let mem = ref Mem.empty ;;
+
+let add_mem byte addr =
+  assert(byte >= 0 && byte < 256);
+  let lit_byte = V_lit (L_aux (L_num (Big_int.big_int_of_int byte), Unknown)) in
+  let vector = (if little_endian then to_vec_inc else to_vec_dec) 8 lit_byte in
+  let key = Id_aux (Id "MEM", Unknown), addr in
+  mem := Mem.add key vector !mem
 ;;
 
-let init_mem bytes =
-  foldli
-    (fun i m b -> Mem.add (Id_aux (Id "MEM", Unknown), Big_int.big_int_of_int i) b m)
-    Mem.empty bytes
+let load_section ic (offset,size,addr) =
+  seek_in ic offset;
+  for i = 0 to size - 1 do
+    add_mem (input_byte ic) (Big_int.add_int_big_int i addr);
+  done
+;;
+
+let args = [
+  ("--file", Arg.Set_string file, "filename binary code to load in memory");
+  ("--data", Arg.String add_section, "name,offset,size,addr add a data section");
+  ("--code", Arg.String add_section, "name,offset,size,addr add a code section");
+  ("--startaddr", Arg.Set_int startaddr, "addr initial address");
+  ("--mainaddr", Arg.Set_int mainaddr, "addr address of the main section");
+] ;;
+
+let time_it action arg =
+  let start_time = Sys.time () in
+  ignore (action arg);
+  let finish_time = Sys.time () in
+  finish_time -. start_time
 ;;
 
 let run () =
-  let file = Sys.argv.(1) in
-  let bytes = binary file in
-  eprintf "bytes read\n";
-  let mem = init_mem bytes in
-  eprintf "memory initialized\n";
-  let r = Run_interp.run ~mem (file, Power.defs) in
+  Arg.parse args (fun _ -> raise (Arg.Bad "anonymous parameter")) "" ;
+  if !file = "" then begin
+    Arg.usage args "";
+    exit 1;
+  end;
+  let ic = open_in_bin !file in
+  if !sections = [] then begin
+    sections := [(0, in_channel_length ic, Big_int.zero_big_int)];
+  end;
+  let total_size = List.fold_left (fun n (_,s,_) -> n+s) 0 !sections in
+  eprintf "Loading binary into memory (%d bytes)... %!" total_size;
+  let t = time_it (List.iter (load_section ic)) !sections in
+  eprintf "done. (%f seconds)\n%!" t;
+  close_in ic;
+  let r = Run_interp.run ~mem:!mem (!file, Power.defs) in
   eprintf "%s\n" (if r then "SUCCESS" else "FAILURE")
 ;;