ocaml/toplevel/topdirs.ml

848 lines
27 KiB
OCaml

(**************************************************************************)
(* *)
(* OCaml *)
(* *)
(* Xavier Leroy, projet Cristal, INRIA Rocquencourt *)
(* *)
(* Copyright 1996 Institut National de Recherche en Informatique et *)
(* en Automatique. *)
(* *)
(* All rights reserved. This file is distributed under the terms of *)
(* the GNU Lesser General Public License version 2.1, with the *)
(* special exception on linking described in the file LICENSE. *)
(* *)
(**************************************************************************)
(* Toplevel directives *)
open Format
open Misc
open Longident
open Types
open Cmo_format
open Trace
open Toploop
(* The standard output formatter *)
let std_out = std_formatter
(* Directive sections (used in #help) *)
let section_general = "General"
let section_run = "Loading code"
let section_env = "Environment queries"
let section_print = "Pretty-printing"
let section_trace = "Tracing"
let section_options = "Compiler options"
let section_undocumented = "Undocumented"
(* we will print the sections in the first list,
then all user-defined sections,
then the sections in the second list,
then all undocumented directives *)
let order_of_sections =
([
section_general;
section_run;
section_env;
], [
section_print;
section_trace;
section_options;
section_undocumented;
])
(* Do not forget to keep the directives synchronized with the manual in
manual/manual/cmds/top.etex *)
(* To quit *)
let dir_quit () = raise (Compenv.Exit_with_status 0)
let _ = add_directive "quit" (Directive_none dir_quit)
{
section = section_general;
doc = "Exit the toplevel.";
}
(* To add a directory to the load path *)
let dir_directory s =
let d = expand_directory Config.standard_library s in
Dll.add_path [d];
let dir = Load_path.Dir.create d in
Load_path.add dir;
toplevel_env :=
Stdlib.String.Set.fold
(fun name env ->
Env.add_persistent_structure (Ident.create_persistent name) env)
(Env.persistent_structures_of_dir dir)
!toplevel_env
let _ = add_directive "directory" (Directive_string dir_directory)
{
section = section_run;
doc = "Add the given directory to search path for source and compiled \
files.";
}
(* To remove a directory from the load path *)
let dir_remove_directory s =
let d = expand_directory Config.standard_library s in
let keep id =
match Load_path.find_uncap (Ident.name id ^ ".cmi") with
| exception Not_found -> true
| fn -> Filename.dirname fn <> d
in
toplevel_env := Env.filter_non_loaded_persistent keep !toplevel_env;
Load_path.remove_dir s;
Dll.remove_path [d]
let _ = add_directive "remove_directory" (Directive_string dir_remove_directory)
{
section = section_run;
doc = "Remove the given directory from the search path.";
}
(* To change the current directory *)
let dir_cd s = Sys.chdir s
let _ = add_directive "cd" (Directive_string dir_cd)
{
section = section_run;
doc = "Change the current working directory.";
}
(* Load in-core a .cmo file *)
exception Load_failed
let check_consistency ppf filename cu =
try Env.import_crcs ~source:filename cu.cu_imports
with Persistent_env.Consistbl.Inconsistency {
unit_name = name;
inconsistent_source = user;
original_source = auth;
} ->
fprintf ppf "@[<hv 0>The files %s@ and %s@ \
disagree over interface %s@]@."
user auth name;
raise Load_failed
let load_compunit ic filename ppf compunit =
check_consistency ppf filename compunit;
seek_in ic compunit.cu_pos;
let code_size = compunit.cu_codesize + 8 in
let code = LongString.create code_size in
LongString.input_bytes_into code ic compunit.cu_codesize;
LongString.set code compunit.cu_codesize (Char.chr Opcodes.opRETURN);
LongString.blit_string "\000\000\000\001\000\000\000" 0
code (compunit.cu_codesize + 1) 7;
let initial_symtable = Symtable.current_state() in
Symtable.patch_object code compunit.cu_reloc;
Symtable.update_global_table();
let events =
if compunit.cu_debug = 0 then [| |]
else begin
seek_in ic compunit.cu_debug;
[| input_value ic |]
end in
begin try
may_trace := true;
let _bytecode, closure = Meta.reify_bytecode code events None in
ignore (closure ());
may_trace := false;
with exn ->
record_backtrace ();
may_trace := false;
Symtable.restore_state initial_symtable;
print_exception_outcome ppf exn;
raise Load_failed
end
let rec load_file recursive ppf name =
let filename =
try Some (Load_path.find name) with Not_found -> None
in
match filename with
| None -> fprintf ppf "Cannot find file %s.@." name; false
| Some filename ->
let ic = open_in_bin filename in
Misc.try_finally
~always:(fun () -> close_in ic)
(fun () -> really_load_file recursive ppf name filename ic)
and really_load_file recursive ppf name filename ic =
let buffer = really_input_string ic (String.length Config.cmo_magic_number) in
try
if buffer = Config.cmo_magic_number then begin
let compunit_pos = input_binary_int ic in (* Go to descriptor *)
seek_in ic compunit_pos;
let cu : compilation_unit = input_value ic in
if recursive then
List.iter
(function
| (Reloc_getglobal id, _)
when not (Symtable.is_global_defined id) ->
let file = Ident.name id ^ ".cmo" in
begin match Load_path.find_uncap file with
| exception Not_found -> ()
| file ->
if not (load_file recursive ppf file) then raise Load_failed
end
| _ -> ()
)
cu.cu_reloc;
load_compunit ic filename ppf cu;
true
end else
if buffer = Config.cma_magic_number then begin
let toc_pos = input_binary_int ic in (* Go to table of contents *)
seek_in ic toc_pos;
let lib = (input_value ic : library) in
List.iter
(fun dllib ->
let name = Dll.extract_dll_name dllib in
try Dll.open_dlls Dll.For_execution [name]
with Failure reason ->
fprintf ppf
"Cannot load required shared library %s.@.Reason: %s.@."
name reason;
raise Load_failed)
lib.lib_dllibs;
List.iter (load_compunit ic filename ppf) lib.lib_units;
true
end else begin
fprintf ppf "File %s is not a bytecode object file.@." name;
false
end
with Load_failed -> false
let dir_load ppf name = ignore (load_file false ppf name)
let _ = add_directive "load" (Directive_string (dir_load std_out))
{
section = section_run;
doc = "Load in memory a bytecode object, produced by ocamlc.";
}
let dir_load_rec ppf name = ignore (load_file true ppf name)
let _ = add_directive "load_rec"
(Directive_string (dir_load_rec std_out))
{
section = section_run;
doc = "As #load, but loads dependencies recursively.";
}
let load_file = load_file false
(* Load commands from a file *)
let dir_use ppf name = ignore(Toploop.use_file ppf name)
let dir_use_output ppf name = ignore(Toploop.use_output ppf name)
let dir_mod_use ppf name = ignore(Toploop.mod_use_file ppf name)
let _ = add_directive "use" (Directive_string (dir_use std_out))
{
section = section_run;
doc = "Read, compile and execute source phrases from the given file.";
}
let _ = add_directive "use_output" (Directive_string (dir_use_output std_out))
{
section = section_run;
doc = "Execute a command and read, compile and execute source phrases \
from its output.";
}
let _ = add_directive "mod_use" (Directive_string (dir_mod_use std_out))
{
section = section_run;
doc = "Usage is identical to #use but #mod_use \
wraps the contents in a module.";
}
(* Install, remove a printer *)
let filter_arrow ty =
let ty = Ctype.expand_head !toplevel_env ty in
match ty.desc with
| Tarrow (lbl, l, r, _) when not (Btype.is_optional lbl) -> Some (l, r)
| _ -> None
let rec extract_last_arrow desc =
match filter_arrow desc with
| None -> raise (Ctype.Unify [])
| Some (_, r as res) ->
try extract_last_arrow r
with Ctype.Unify _ -> res
let extract_target_type ty = fst (extract_last_arrow ty)
let extract_target_parameters ty =
let ty = extract_target_type ty |> Ctype.expand_head !toplevel_env in
match ty.desc with
| Tconstr (path, (_ :: _ as args), _)
when Ctype.all_distinct_vars !toplevel_env args -> Some (path, args)
| _ -> None
type 'a printer_type_new = Format.formatter -> 'a -> unit
type 'a printer_type_old = 'a -> unit
let printer_type ppf typename =
let printer_type =
match
Env.find_type_by_name
(Ldot(Lident "Topdirs", typename)) !toplevel_env
with
| path, _ -> path
| exception Not_found ->
fprintf ppf "Cannot find type Topdirs.%s.@." typename;
raise Exit
in
printer_type
let match_simple_printer_type desc printer_type =
Ctype.begin_def();
let ty_arg = Ctype.newvar() in
Ctype.unify !toplevel_env
(Ctype.newconstr printer_type [ty_arg])
(Ctype.instance desc.val_type);
Ctype.end_def();
Ctype.generalize ty_arg;
(ty_arg, None)
let match_generic_printer_type desc path args printer_type =
Ctype.begin_def();
let args = List.map (fun _ -> Ctype.newvar ()) args in
let ty_target = Ctype.newty (Tconstr (path, args, ref Mnil)) in
let ty_args =
List.map (fun ty_var -> Ctype.newconstr printer_type [ty_var]) args in
let ty_expected =
List.fold_right
(fun ty_arg ty -> Ctype.newty (Tarrow (Asttypes.Nolabel, ty_arg, ty,
Cunknown)))
ty_args (Ctype.newconstr printer_type [ty_target]) in
Ctype.unify !toplevel_env
ty_expected
(Ctype.instance desc.val_type);
Ctype.end_def();
Ctype.generalize ty_expected;
if not (Ctype.all_distinct_vars !toplevel_env args) then
raise (Ctype.Unify []);
(ty_expected, Some (path, ty_args))
let match_printer_type ppf desc =
let printer_type_new = printer_type ppf "printer_type_new" in
let printer_type_old = printer_type ppf "printer_type_old" in
try
(match_simple_printer_type desc printer_type_new, false)
with Ctype.Unify _ ->
try
(match_simple_printer_type desc printer_type_old, true)
with Ctype.Unify _ as exn ->
match extract_target_parameters desc.val_type with
| None -> raise exn
| Some (path, args) ->
(match_generic_printer_type desc path args printer_type_new,
false)
let find_printer_type ppf lid =
match Env.find_value_by_name lid !toplevel_env with
| (path, desc) -> begin
match match_printer_type ppf desc with
| (ty_arg, is_old_style) -> (ty_arg, path, is_old_style)
| exception Ctype.Unify _ ->
fprintf ppf "%a has a wrong type for a printing function.@."
Printtyp.longident lid;
raise Exit
end
| exception Not_found ->
fprintf ppf "Unbound value %a.@." Printtyp.longident lid;
raise Exit
let dir_install_printer ppf lid =
try
let ((ty_arg, ty), path, is_old_style) =
find_printer_type ppf lid in
let v = eval_value_path !toplevel_env path in
match ty with
| None ->
let print_function =
if is_old_style then
(fun _formatter repr -> Obj.obj v (Obj.obj repr))
else
(fun formatter repr -> Obj.obj v formatter (Obj.obj repr)) in
install_printer path ty_arg print_function
| Some (ty_path, ty_args) ->
let rec build v = function
| [] ->
let print_function =
if is_old_style then
(fun _formatter repr -> Obj.obj v (Obj.obj repr))
else
(fun formatter repr -> Obj.obj v formatter (Obj.obj repr)) in
Zero print_function
| _ :: args ->
Succ
(fun fn -> build ((Obj.obj v : _ -> Obj.t) fn) args) in
install_generic_printer' path ty_path (build v ty_args)
with Exit -> ()
let dir_remove_printer ppf lid =
try
let (_ty_arg, path, _is_old_style) = find_printer_type ppf lid in
begin try
remove_printer path
with Not_found ->
fprintf ppf "No printer named %a.@." Printtyp.longident lid
end
with Exit -> ()
let _ = add_directive "install_printer"
(Directive_ident (dir_install_printer std_out))
{
section = section_print;
doc = "Registers a printer for values of a certain type.";
}
let _ = add_directive "remove_printer"
(Directive_ident (dir_remove_printer std_out))
{
section = section_print;
doc = "Remove the named function from the table of toplevel printers.";
}
(* The trace *)
external current_environment: unit -> Obj.t = "caml_get_current_environment"
let tracing_function_ptr =
get_code_pointer
(Obj.repr (fun arg -> Trace.print_trace (current_environment()) arg))
let dir_trace ppf lid =
match Env.find_value_by_name lid !toplevel_env with
| (path, desc) -> begin
(* Check if this is a primitive *)
match desc.val_kind with
| Val_prim _ ->
fprintf ppf "%a is an external function and cannot be traced.@."
Printtyp.longident lid
| _ ->
let clos = eval_value_path !toplevel_env path in
(* Nothing to do if it's not a closure *)
if Obj.is_block clos
&& (Obj.tag clos = Obj.closure_tag || Obj.tag clos = Obj.infix_tag)
&& (match Ctype.(repr (expand_head !toplevel_env desc.val_type))
with {desc=Tarrow _} -> true | _ -> false)
then begin
match is_traced clos with
| Some opath ->
fprintf ppf "%a is already traced (under the name %a).@."
Printtyp.path path
Printtyp.path opath
| None ->
(* Instrument the old closure *)
traced_functions :=
{ path = path;
closure = clos;
actual_code = get_code_pointer clos;
instrumented_fun =
instrument_closure !toplevel_env lid ppf desc.val_type }
:: !traced_functions;
(* Redirect the code field of the closure to point
to the instrumentation function *)
set_code_pointer clos tracing_function_ptr;
fprintf ppf "%a is now traced.@." Printtyp.longident lid
end else fprintf ppf "%a is not a function.@." Printtyp.longident lid
end
| exception Not_found ->
fprintf ppf "Unbound value %a.@." Printtyp.longident lid
let dir_untrace ppf lid =
match Env.find_value_by_name lid !toplevel_env with
| (path, _desc) ->
let rec remove = function
| [] ->
fprintf ppf "%a was not traced.@." Printtyp.longident lid;
[]
| f :: rem ->
if Path.same f.path path then begin
set_code_pointer f.closure f.actual_code;
fprintf ppf "%a is no longer traced.@." Printtyp.longident lid;
rem
end else f :: remove rem in
traced_functions := remove !traced_functions
| exception Not_found ->
fprintf ppf "Unbound value %a.@." Printtyp.longident lid
let dir_untrace_all ppf () =
List.iter
(fun f ->
set_code_pointer f.closure f.actual_code;
fprintf ppf "%a is no longer traced.@." Printtyp.path f.path)
!traced_functions;
traced_functions := []
let parse_warnings ppf iserr s =
try Warnings.parse_options iserr s
with Arg.Bad err -> fprintf ppf "%s.@." err
(* Typing information *)
let trim_signature = function
Mty_signature sg ->
Mty_signature
(List.map
(function
Sig_module (id, pres, md, rs, priv) ->
let attribute =
Ast_helper.Attr.mk
(Location.mknoloc "...")
(Parsetree.PStr [])
in
Sig_module (id, pres, {md with md_attributes =
attribute :: md.md_attributes},
rs, priv)
(*| Sig_modtype (id, Modtype_manifest mty) ->
Sig_modtype (id, Modtype_manifest (trim_modtype mty))*)
| item -> item)
sg)
| mty -> mty
let show_prim to_sig ppf lid =
let env = !Toploop.toplevel_env in
let loc = Location.none in
try
let s =
match lid with
| Longident.Lident s -> s
| Longident.Ldot (_,s) -> s
| Longident.Lapply _ ->
fprintf ppf "Invalid path %a@." Printtyp.longident lid;
raise Exit
in
let id = Ident.create_persistent s in
let sg = to_sig env loc id lid in
Printtyp.wrap_printing_env ~error:false env
(fun () -> fprintf ppf "@[%a@]@." Printtyp.signature sg)
with
| Not_found ->
fprintf ppf "@[Unknown element.@]@."
| Exit -> ()
let all_show_funs = ref []
let reg_show_prim name to_sig doc =
all_show_funs := to_sig :: !all_show_funs;
add_directive
name
(Directive_ident (show_prim to_sig std_out))
{
section = section_env;
doc;
}
let () =
reg_show_prim "show_val"
(fun env loc id lid ->
let _path, desc = Env.lookup_value ~loc lid env in
[ Sig_value (id, desc, Exported) ]
)
"Print the signature of the corresponding value."
let () =
reg_show_prim "show_type"
(fun env loc id lid ->
let _path, desc = Env.lookup_type ~loc lid env in
[ Sig_type (id, desc, Trec_not, Exported) ]
)
"Print the signature of the corresponding type constructor."
(* Each registered show_prim function is called in turn
* and any output produced is sent to std_out.
* Two show_prim functions are needed for constructors,
* one for exception constructors and another for
* non-exception constructors (normal and extensible variants). *)
let is_exception_constructor env type_expr =
Ctype.equal env true [type_expr] [Predef.type_exn]
let is_extension_constructor = function
| Cstr_extension _ -> true
| _ -> false
let () =
(* This show_prim function will only show constructor types
* that are not also exception types. *)
reg_show_prim "show_constructor"
(fun env loc id lid ->
let desc = Env.lookup_constructor ~loc Env.Positive lid env in
if is_exception_constructor env desc.cstr_res then
raise Not_found;
let path =
match Ctype.repr desc.cstr_res with
| {desc=Tconstr(path, _, _)} -> path
| _ -> raise Not_found
in
let type_decl = Env.find_type path env in
if is_extension_constructor desc.cstr_tag then
let ret_type =
if desc.cstr_generalized then Some desc.cstr_res
else None
in
let ext =
{ ext_type_path = path;
ext_type_params = type_decl.type_params;
ext_args = Cstr_tuple desc.cstr_args;
ext_ret_type = ret_type;
ext_private = Asttypes.Public;
ext_loc = desc.cstr_loc;
ext_attributes = desc.cstr_attributes;
ext_uid = desc.cstr_uid; }
in
[Sig_typext (id, ext, Text_first, Exported)]
else
(* make up a fake Ident.t as type_decl : Types.type_declaration
* does not have an Ident.t yet. Ident.create_presistent is a
* good choice because it has no side-effects.
* *)
let type_id = Ident.create_persistent (Path.name path) in
[ Sig_type (type_id, type_decl, Trec_first, Exported) ]
)
"Print the signature of the corresponding value constructor."
let () =
reg_show_prim "show_exception"
(fun env loc id lid ->
let desc = Env.lookup_constructor ~loc Env.Positive lid env in
if not (is_exception_constructor env desc.cstr_res) then
raise Not_found;
let ret_type =
if desc.cstr_generalized then Some Predef.type_exn
else None
in
let ext =
{ ext_type_path = Predef.path_exn;
ext_type_params = [];
ext_args = Cstr_tuple desc.cstr_args;
ext_ret_type = ret_type;
ext_private = Asttypes.Public;
ext_loc = desc.cstr_loc;
ext_attributes = desc.cstr_attributes;
ext_uid = desc.cstr_uid;
}
in
[Sig_typext (id, ext, Text_exception, Exported)]
)
"Print the signature of the corresponding exception."
let () =
reg_show_prim "show_module"
(fun env loc id lid ->
let rec accum_aliases md acc =
let acc =
Sig_module (id, Mp_present,
{md with md_type = trim_signature md.md_type},
Trec_not, Exported) :: acc in
match md.md_type with
| Mty_alias path ->
let md = Env.find_module path env in
accum_aliases md acc
| Mty_ident _ | Mty_signature _ | Mty_functor _ ->
List.rev acc
in
let _, md = Env.lookup_module ~loc lid env in
accum_aliases md []
)
"Print the signature of the corresponding module."
let () =
reg_show_prim "show_module_type"
(fun env loc id lid ->
let _path, desc = Env.lookup_modtype ~loc lid env in
[ Sig_modtype (id, desc, Exported) ]
)
"Print the signature of the corresponding module type."
let () =
reg_show_prim "show_class"
(fun env loc id lid ->
let _path, desc = Env.lookup_class ~loc lid env in
[ Sig_class (id, desc, Trec_not, Exported) ]
)
"Print the signature of the corresponding class."
let () =
reg_show_prim "show_class_type"
(fun env loc id lid ->
let _path, desc = Env.lookup_cltype ~loc lid env in
[ Sig_class_type (id, desc, Trec_not, Exported) ]
)
"Print the signature of the corresponding class type."
let show env loc id lid =
let sg =
List.fold_left
(fun sg f -> try (f env loc id lid) @ sg with _ -> sg)
[] !all_show_funs
in
if sg = [] then raise Not_found else sg
let () =
add_directive "show" (Directive_ident (show_prim show std_out))
{
section = section_env;
doc = "Print the signatures of components \
from any of the categories below.";
}
let _ = add_directive "trace"
(Directive_ident (dir_trace std_out))
{
section = section_trace;
doc = "All calls to the function \
named function-name will be traced.";
}
let _ = add_directive "untrace"
(Directive_ident (dir_untrace std_out))
{
section = section_trace;
doc = "Stop tracing the given function.";
}
let _ = add_directive "untrace_all"
(Directive_none (dir_untrace_all std_out))
{
section = section_trace;
doc = "Stop tracing all functions traced so far.";
}
(* Control the printing of values *)
let _ = add_directive "print_depth"
(Directive_int(fun n -> max_printer_depth := n))
{
section = section_print;
doc = "Limit the printing of values to a maximal depth of n.";
}
let _ = add_directive "print_length"
(Directive_int(fun n -> max_printer_steps := n))
{
section = section_print;
doc = "Limit the number of value nodes printed to at most n.";
}
(* Set various compiler flags *)
let _ = add_directive "labels"
(Directive_bool(fun b -> Clflags.classic := not b))
{
section = section_options;
doc = "Choose whether to ignore labels in function types.";
}
let _ = add_directive "principal"
(Directive_bool(fun b -> Clflags.principal := b))
{
section = section_options;
doc = "Make sure that all types are derived in a principal way.";
}
let _ = add_directive "rectypes"
(Directive_none(fun () -> Clflags.recursive_types := true))
{
section = section_options;
doc = "Allow arbitrary recursive types during type-checking.";
}
let _ = add_directive "ppx"
(Directive_string(fun s -> Clflags.all_ppx := s :: !Clflags.all_ppx))
{
section = section_options;
doc = "After parsing, pipe the abstract \
syntax tree through the preprocessor command.";
}
let _ = add_directive "warnings"
(Directive_string (parse_warnings std_out false))
{
section = section_options;
doc = "Enable or disable warnings according to the argument.";
}
let _ = add_directive "warn_error"
(Directive_string (parse_warnings std_out true))
{
section = section_options;
doc = "Treat as errors the warnings enabled by the argument.";
}
(* #help directive *)
let directive_sections () =
let sections = Hashtbl.create 10 in
let add_dir name dir =
let section, doc =
match Hashtbl.find directive_info_table name with
| { section; doc } -> section, Some doc
| exception Not_found -> "Undocumented", None
in
Hashtbl.replace sections section
((name, dir, doc)
:: (try Hashtbl.find sections section with Not_found -> []))
in
Hashtbl.iter add_dir directive_table;
let take_section section =
if not (Hashtbl.mem sections section) then (section, [])
else begin
let section_dirs =
Hashtbl.find sections section
|> List.sort (fun (n1, _, _) (n2, _, _) -> String.compare n1 n2) in
Hashtbl.remove sections section;
(section, section_dirs)
end
in
let before, after = order_of_sections in
let sections_before = List.map take_section before in
let sections_after = List.map take_section after in
let sections_user =
Hashtbl.fold (fun section _ acc -> section::acc) sections []
|> List.sort String.compare
|> List.map take_section in
sections_before @ sections_user @ sections_after
let print_directive ppf (name, directive, doc) =
let param = match directive with
| Directive_none _ -> ""
| Directive_string _ -> " <str>"
| Directive_int _ -> " <int>"
| Directive_bool _ -> " <bool>"
| Directive_ident _ -> " <ident>" in
match doc with
| None -> fprintf ppf "#%s%s@." name param
| Some doc ->
fprintf ppf "@[<hov 2>#%s%s@\n%a@]@."
name param
Format.pp_print_text doc
let print_section ppf (section, directives) =
if directives <> [] then begin
fprintf ppf "%30s%s@." "" section;
List.iter (print_directive ppf) directives;
fprintf ppf "@.";
end
let print_directives ppf () =
List.iter (print_section ppf) (directive_sections ())
let _ = add_directive "help"
(Directive_none (print_directives std_out))
{
section = section_general;
doc = "Prints a list of all available directives, with \
corresponding argument type if appropriate.";
}