264 lines
8.5 KiB
OCaml
264 lines
8.5 KiB
OCaml
(***********************************************************************)
|
|
(* *)
|
|
(* Caml Special Light *)
|
|
(* *)
|
|
(* Xavier Leroy, projet Cristal, INRIA Rocquencourt *)
|
|
(* *)
|
|
(* Copyright 1995 Institut National de Recherche en Informatique et *)
|
|
(* Automatique. Distributed only by permission. *)
|
|
(* *)
|
|
(***********************************************************************)
|
|
|
|
(* $Id$ *)
|
|
|
|
(* Toplevel directives *)
|
|
|
|
open Format
|
|
open Misc
|
|
open Longident
|
|
open Path
|
|
open Typedtree
|
|
open Emitcode
|
|
open Printval
|
|
open Trace
|
|
open Toploop
|
|
|
|
(* Temporary assignment to a reference *)
|
|
|
|
let protect r newval body =
|
|
let oldval = !r in
|
|
try
|
|
r := newval;
|
|
let res = body() in
|
|
r := oldval;
|
|
res
|
|
with x ->
|
|
r := oldval;
|
|
raise x
|
|
|
|
(* Return the value referred to by a path *)
|
|
|
|
let rec eval_path = function
|
|
Pident id -> Symtable.get_global_value id
|
|
| Pdot(p, s, pos) -> Obj.field (eval_path p) pos
|
|
| Papply(p1, p2) -> fatal_error "Topdirs.eval_path"
|
|
|
|
(* To quit *)
|
|
|
|
let dir_quit () = exit 0; ()
|
|
|
|
let _ = Hashtbl.add directive_table "quit" (Directive_none dir_quit)
|
|
|
|
(* To add a directory to the load path *)
|
|
|
|
let dir_directory s =
|
|
Config.load_path := s :: !Config.load_path;
|
|
Env.reset_cache()
|
|
|
|
let _ = Hashtbl.add directive_table "directory" (Directive_string dir_directory)
|
|
|
|
(* To change the current directory *)
|
|
|
|
let dir_cd s =
|
|
Sys.chdir s
|
|
|
|
let _ = Hashtbl.add directive_table "cd" (Directive_string dir_cd)
|
|
|
|
(* Load in-core a .cmo file *)
|
|
|
|
let dir_load name =
|
|
try
|
|
let filename = find_in_path !Config.load_path name in
|
|
let ic = open_in_bin filename in
|
|
let buffer = String.create (String.length Config.cmo_magic_number) in
|
|
really_input ic buffer 0 (String.length Config.cmo_magic_number);
|
|
if buffer <> Config.cmo_magic_number then begin
|
|
print_string "File "; print_string name;
|
|
print_string " is not a bytecode object file."; print_newline()
|
|
end else begin
|
|
let compunit_pos = input_binary_int ic in (* Go to descriptor *)
|
|
seek_in ic compunit_pos;
|
|
let compunit = (input_value ic : compilation_unit) in
|
|
Bytelink.check_consistency filename compunit;
|
|
seek_in ic compunit.cu_pos;
|
|
let code_size = compunit.cu_codesize + 4 in
|
|
let code = Meta.static_alloc code_size in
|
|
unsafe_really_input ic code 0 compunit.cu_codesize;
|
|
String.unsafe_set code compunit.cu_codesize
|
|
(Char.chr Opcodes.opSTOP);
|
|
String.unsafe_set code (compunit.cu_codesize + 1) '\000';
|
|
String.unsafe_set code (compunit.cu_codesize + 2) '\000';
|
|
String.unsafe_set code (compunit.cu_codesize + 3) '\000';
|
|
Symtable.patch_object code compunit.cu_reloc;
|
|
Symtable.update_global_table();
|
|
begin try
|
|
Meta.execute_bytecode code code_size; ()
|
|
with exn ->
|
|
print_exception_outcome exn
|
|
end
|
|
end;
|
|
close_in ic
|
|
with Not_found ->
|
|
print_string "Cannot find file "; print_string name; print_newline()
|
|
|
|
let _ = Hashtbl.add directive_table "load" (Directive_string dir_load)
|
|
|
|
(* Load commands from a file *)
|
|
|
|
let dir_use name =
|
|
try
|
|
let filename = find_in_path !Config.load_path name in
|
|
let ic = open_in_bin filename in
|
|
let lb = Lexing.from_channel ic in
|
|
protect Location.input_name filename (fun () ->
|
|
try
|
|
List.iter
|
|
(fun ph -> if execute_phrase ph then () else raise Exit)
|
|
(Parse.use_file lb)
|
|
with
|
|
Exit -> ()
|
|
| Sys.Break ->
|
|
print_string "Interrupted."; print_newline()
|
|
| x ->
|
|
Errors.report_error x);
|
|
close_in ic
|
|
with Not_found ->
|
|
print_string "Cannot find file "; print_string name; print_newline()
|
|
|
|
let _ = Hashtbl.add directive_table "use" (Directive_string dir_use)
|
|
|
|
(* Install, remove a printer *)
|
|
|
|
let find_printer_type lid =
|
|
try
|
|
let (path, desc) = Env.lookup_value lid !toplevel_env in
|
|
Ctype.begin_def();
|
|
let ty_arg = Ctype.newvar() in
|
|
Ctype.unify !toplevel_env (Tarrow(ty_arg, Predef.type_unit))
|
|
(Ctype.instance desc.val_type);
|
|
Ctype.end_def();
|
|
Ctype.generalize ty_arg;
|
|
(ty_arg, path)
|
|
with
|
|
Not_found ->
|
|
print_string "Unbound value "; Printtyp.longident lid;
|
|
print_newline(); raise Exit
|
|
| Ctype.Unify ->
|
|
Printtyp.longident lid;
|
|
print_string " has the wrong type for a printing function";
|
|
print_newline(); raise Exit
|
|
|
|
let dir_install_printer lid =
|
|
try
|
|
let (ty_arg, path) = find_printer_type lid in
|
|
let v = eval_path path in
|
|
Printval.printers :=
|
|
(path, ty_arg, (Obj.magic v : Obj.t -> unit)) :: !Printval.printers
|
|
with Exit ->
|
|
()
|
|
|
|
let dir_remove_printer lid =
|
|
try
|
|
let (ty_arg, path) = find_printer_type lid in
|
|
let rec remove = function
|
|
[] ->
|
|
print_string "No printer named "; Printtyp.longident lid;
|
|
print_newline();
|
|
[]
|
|
| (p, ty, fn as printer) :: rem ->
|
|
if Path.same p path then rem else printer :: remove rem in
|
|
Printval.printers := remove !Printval.printers
|
|
with Exit ->
|
|
()
|
|
|
|
let _ = Hashtbl.add directive_table "install_printer"
|
|
(Directive_ident dir_install_printer)
|
|
let _ = Hashtbl.add directive_table "remove_printer"
|
|
(Directive_ident dir_remove_printer)
|
|
|
|
(* The trace *)
|
|
|
|
external current_environment: unit -> Obj.t = "get_current_environment"
|
|
|
|
let dir_trace lid =
|
|
try
|
|
let (path, desc) = Env.lookup_value lid !toplevel_env in
|
|
if is_traced path then begin
|
|
Printtyp.path path; print_string " is already traced.";
|
|
print_newline()
|
|
end else begin
|
|
let clos = eval_path path in
|
|
(* Nothing to do if it's not a closure *)
|
|
if Obj.is_block clos & Obj.tag clos = 250 then begin
|
|
let old_clos = copy_closure clos in
|
|
(* Instrument the old closure *)
|
|
traced_functions :=
|
|
{ path = path;
|
|
closure = clos;
|
|
initial_closure = old_clos;
|
|
instrumented_fun =
|
|
instrument_closure lid (Ctype.instance desc.val_type) old_clos}
|
|
:: !traced_functions;
|
|
(* Redirect the code field of the old closure *)
|
|
overwrite_closure clos
|
|
(Obj.repr (fun arg -> Trace.print_trace (current_environment()) arg));
|
|
(* Warn if this is a primitive *)
|
|
match desc.val_prim with
|
|
None ->
|
|
Printtyp.longident lid; print_string " is now traced.";
|
|
print_newline()
|
|
| Some p ->
|
|
open_hovbox 0;
|
|
print_string "Warning: "; Printtyp.longident lid;
|
|
print_string " is an external function."; print_space();
|
|
print_string "Inlined calls will not be traced.";
|
|
close_box(); print_newline()
|
|
end else begin
|
|
Printtyp.longident lid; print_string " is not a function.";
|
|
print_newline()
|
|
end
|
|
end
|
|
with Not_found ->
|
|
print_string "Unbound value "; Printtyp.longident lid;
|
|
print_newline()
|
|
|
|
let dir_untrace lid =
|
|
try
|
|
let (path, desc) = Env.lookup_value lid !toplevel_env in
|
|
let rec remove = function
|
|
[] ->
|
|
Printtyp.longident lid; print_string " was not traced.";
|
|
print_newline();
|
|
[]
|
|
| f :: rem ->
|
|
if Path.same f.path path then begin
|
|
overwrite_closure (eval_path path) f.initial_closure;
|
|
Printtyp.longident lid; print_string " is no longer traced.";
|
|
print_newline();
|
|
rem
|
|
end else f :: remove rem in
|
|
traced_functions := remove !traced_functions
|
|
with Not_found ->
|
|
print_string "Unbound value "; Printtyp.longident lid;
|
|
print_newline()
|
|
|
|
let dir_untrace_all () =
|
|
List.iter
|
|
(fun f ->
|
|
overwrite_closure (eval_path f.path) f.initial_closure;
|
|
Printtyp.path f.path; print_string " is no longer traced.";
|
|
print_newline())
|
|
!traced_functions;
|
|
traced_functions := []
|
|
|
|
let _ = Hashtbl.add directive_table "trace" (Directive_ident dir_trace)
|
|
let _ = Hashtbl.add directive_table "untrace" (Directive_ident dir_untrace)
|
|
let _ = Hashtbl.add directive_table "untrace_all" (Directive_none dir_untrace_all)
|
|
|
|
(* Control the printing of values *)
|
|
|
|
let _ = Hashtbl.add directive_table "print_depth"
|
|
(Directive_int(fun n -> max_printer_depth := n))
|
|
let _ = Hashtbl.add directive_table "print_length"
|
|
(Directive_int(fun n -> max_printer_steps := n))
|