%% -*- erlang-indent-level: 2 -*-
%% =======================================================================
%% Filename : hipe_unified_loader.erl
%% Module : hipe_unified_loader
%% Purpose : To load code into memory and link it to the system.
%% Notes : See hipe_ext_format.hrl for description of the external
%% format.
%% =======================================================================
%% TODO:
%% Problems with the order in which things are done.
%% export_funs should atomically patch references to make fe and
%% make beam stubs. !!
%%
%% Each function should have two proper databases.
%% Describe the patch algorithm:
%% For each function MFA that is (re)compiled to Address:
%% 1. For the old MFA
%% a. RefsTo = MFA->refers_to
%% b. for each {F,Adr} in RefsTo: remove Adr from F->is_referred
%% c. RefsFrom = MFA->is_referred
%% d. For each {Adr,Type} in RefsFrom:
%% update instr at Adr to refer to Address instead.
%% 2. For the new MFA
%% a. MFA->is_referred=RefsFrom
%% 3. For each function F referenced in the code at Offset:
%% add {Address+Offset,Type} to F->is_referred
%% add {F,Address+Offset} to MFA->refers_to
%% 4. Make Address the entrypoint for MFA
%%
%% Add exporting of exported constants.
%% Add freeing of old code.
%% Inline hipe_sparc_ext_format somehow.
%% =======================================================================
-module(hipe_unified_loader).
-compile(no_native).
% 'no_native' is a workaround to avoid "The code server called unloaded module"
% caused by Mod:module_info(exports) in patch_to_emu_step1() called by post_beam_load.
% Reproducable with hipelibs and asn1_SUITE.
% I think the real solution would be to let BIF erlang:load_module/2 redirect all
% hipe calls to the module and thereby remove post_beam_load.
-export([chunk_name/1,
%% Only the code and code_server modules may call the entries below!
load_native_code/2,
post_beam_load/1,
load_module/3,
load/2]).
%%-define(DEBUG,true).
-define(DO_ASSERT,true).
-define(HIPE_LOGGING,true).
-include("../../hipe/main/hipe.hrl").
-include("hipe_ext_format.hrl").
%% Currently, there is no need to expose these to the outside world.
-define(HS8P_TAG,"HS8P").
-define(HPPC_TAG,"HPPC").
-define(HP64_TAG,"HP64").
-define(HARM_TAG,"HARM").
-define(HX86_TAG,"HX86").
-define(HA64_TAG,"HA64").
%%========================================================================
-spec chunk_name(hipe_architecture()) -> string().
%% @doc
%% Returns the native code chunk name of the Architecture.
%% (On which presumably we are running.)
chunk_name(Architecture) ->
case Architecture of
amd64 -> ?HA64_TAG; %% HiPE, x86_64, (implicit: 64-bit, Unix)
arm -> ?HARM_TAG; %% HiPE, arm, v5 (implicit: 32-bit, Linux)
powerpc -> ?HPPC_TAG; %% HiPE, PowerPC (implicit: 32-bit, Linux)
ppc64 -> ?HP64_TAG; %% HiPE, ppc64 (implicit: 64-bit, Linux)
ultrasparc -> ?HS8P_TAG; %% HiPE, SPARC, V8+ (implicit: 32-bit)
x86 -> ?HX86_TAG %% HiPE, x86, (implicit: Unix)
%% Future: HSV9 %% HiPE, SPARC, V9 (implicit: 64-bit)
%% HW32 %% HiPE, x86, Win32
end.
%%========================================================================
-spec load_native_code(Mod, binary()) -> 'no_native' | {'module', Mod}
when Mod :: atom().
%% @doc
%% Loads the native code of a module Mod.
%% Returns {module,Mod} on success (for compatibility with
%% code:load_file/1) and the atom `no_native' on failure.
load_native_code(Mod, Bin) when is_atom(Mod), is_binary(Bin) ->
Architecture = erlang:system_info(hipe_architecture),
try chunk_name(Architecture) of
ChunkTag ->
%% patch_to_emu(Mod),
case code:get_chunk(Bin, ChunkTag) of
undefined -> no_native;
NativeCode when is_binary(NativeCode) ->
erlang:system_flag(multi_scheduling, block),
try
OldReferencesToPatch = patch_to_emu_step1(Mod),
case load_module(Mod, NativeCode, Bin, OldReferencesToPatch) of
bad_crc -> no_native;
Result -> Result
end
after
erlang:system_flag(multi_scheduling, unblock)
end
end
catch
_:_ ->
%% Unknown HiPE architecture. Can't happen (in principle).
no_native
end.
%%========================================================================
-spec post_beam_load(atom()) -> 'ok'.
post_beam_load(Mod) when is_atom(Mod) ->
Architecture = erlang:system_info(hipe_architecture),
try chunk_name(Architecture) of
_ChunkTag ->
erlang:system_flag(multi_scheduling, block),
try
patch_to_emu(Mod)
after
erlang:system_flag(multi_scheduling, unblock)
end
catch
_:_ ->
ok
end.
%%========================================================================
version_check(Version, Mod) when is_atom(Mod) ->
Ver = ?VERSION_STRING(),
case Version < Ver of
true ->
?msg("WARNING: Module ~w was compiled with HiPE version ~s\n",
[Mod, Version]);
_ -> ok
end.
%%========================================================================
-spec load_module(Mod, binary(), _) -> 'bad_crc' | {'module', Mod}
when Mod :: atom().
load_module(Mod, Bin, Beam) ->
erlang:system_flag(multi_scheduling, block),
try
load_module_nosmp(Mod, Bin, Beam)
after
erlang:system_flag(multi_scheduling, unblock)
end.
load_module_nosmp(Mod, Bin, Beam) ->
load_module(Mod, Bin, Beam, []).
load_module(Mod, Bin, Beam, OldReferencesToPatch) ->
?debug_msg("************ Loading Module ~w ************\n",[Mod]),
%% Loading a whole module, let the BEAM loader patch closures.
put(hipe_patch_closures, false),
load_common(Mod, Bin, Beam, OldReferencesToPatch).
%%========================================================================
-spec load(Mod, binary()) -> 'bad_crc' | {'module', Mod} when Mod :: atom().
load(Mod, Bin) ->
erlang:system_flag(multi_scheduling, block),
try
load_nosmp(Mod, Bin)
after
erlang:system_flag(multi_scheduling, unblock)
end.
load_nosmp(Mod, Bin) ->
?debug_msg("********* Loading funs in module ~w *********\n",[Mod]),
%% Loading just some functions in a module; patch closures separately.
put(hipe_patch_closures, true),
load_common(Mod, Bin, [], []).
%%------------------------------------------------------------------------
load_common(Mod, Bin, Beam, OldReferencesToPatch) ->
%% Unpack the binary.
[{Version, CheckSum},
ConstAlign, ConstSize, ConstMap, LabelMap, ExportMap,
CodeSize, CodeBinary, Refs,
0,[] % ColdSize, CRrefs
] = binary_to_term(Bin),
%% Check that we are loading up-to-date code.
version_check(Version, Mod),
case hipe_bifs:check_crc(CheckSum) of
false ->
?msg("Warning: not loading native code for module ~w: "
"it was compiled for an incompatible runtime system; "
"please regenerate native code for this runtime system\n", [Mod]),
bad_crc;
true ->
%% Create data segment
{ConstAddr,ConstMap2} =
create_data_segment(ConstAlign, ConstSize, ConstMap),
%% Find callees for which we may need trampolines.
CalleeMFAs = find_callee_mfas(Refs),
%% Write the code to memory.
{CodeAddress,Trampolines} =
enter_code(CodeSize, CodeBinary, CalleeMFAs, Mod, Beam),
%% Construct CalleeMFA-to-trampoline mapping.
TrampolineMap = mk_trampoline_map(CalleeMFAs, Trampolines),
%% Patch references to code labels in data seg.
ok = patch_consts(LabelMap, ConstAddr, CodeAddress),
%% Find out which functions are being loaded (and where).
%% Note: Addresses are sorted descending.
{MFAs,Addresses} = exports(ExportMap, CodeAddress),
%% Remove references to old versions of the module.
ReferencesToPatch = get_refs_from(MFAs, []),
ok = remove_refs_from(MFAs),
%% Patch all dynamic references in the code.
%% Function calls, Atoms, Constants, System calls
ok = patch(Refs, CodeAddress, ConstMap2, Addresses, TrampolineMap),
%% Tell the system where the loaded funs are.
%% (patches the BEAM code to redirect to native.)
case Beam of
[] ->
export_funs(Addresses),
ok;
BeamBinary when is_binary(BeamBinary) ->
%% Find all closures in the code.
ClosurePatches = find_closure_patches(Refs),
AddressesOfClosuresToPatch =
calculate_addresses(ClosurePatches, CodeAddress, Addresses),
export_funs(Addresses),
export_funs(Mod, BeamBinary, Addresses, AddressesOfClosuresToPatch),
ok
end,
%% Redirect references to the old module to the new module's BEAM stub.
patch_to_emu_step2(OldReferencesToPatch),
%% Patch referring functions to call the new function
%% The call to export_funs/1 above updated the native addresses
%% for the targets, so passing 'Addresses' is not needed.
redirect(ReferencesToPatch),
?debug_msg("****************Loader Finished****************\n", []),
{module,Mod} % for compatibility with code:load_file/1
end.
%%----------------------------------------------------------------
%% Scan the list of patches and build a set (returned as a tuple)
%% of the callees for which we may need trampolines.
%%
find_callee_mfas(Patches) when is_list(Patches) ->
case erlang:system_info(hipe_architecture) of
amd64 -> [];
arm -> find_callee_mfas(Patches, gb_sets:empty(), false);
powerpc -> find_callee_mfas(Patches, gb_sets:empty(), true);
ppc64 -> find_callee_mfas(Patches, gb_sets:empty(), true);
ultrasparc -> [];
x86 -> []
end.
find_callee_mfas([{Type,Data}|Patches], MFAs, SkipErtsSyms) ->
NewMFAs =
case ?EXT2PATCH_TYPE(Type) of
call_local -> add_callee_mfas(Data, MFAs, SkipErtsSyms);
call_remote -> add_callee_mfas(Data, MFAs, SkipErtsSyms);
%% load_address(function) deliberately ignored
_ -> MFAs
end,
find_callee_mfas(Patches, NewMFAs, SkipErtsSyms);
find_callee_mfas([], MFAs, _SkipErtsSyms) ->
list_to_tuple(gb_sets:to_list(MFAs)).
add_callee_mfas([{DestMFA,_Offsets}|Refs], MFAs, SkipErtsSyms) ->
NewMFAs =
case SkipErtsSyms of
true ->
%% On PowerPC we put the runtime system below the
%% 32M boundary, which allows BIFs and primops to
%% be called with ba/bla instructions. Hence we do
%% not need trampolines for BIFs or primops.
case bif_address(DestMFA) of
false -> gb_sets:add_element(DestMFA, MFAs);
BifAddress when is_integer(BifAddress) -> MFAs
end;
false ->
%% On ARM we also need trampolines for BIFs and primops.
gb_sets:add_element(DestMFA, MFAs)
end,
add_callee_mfas(Refs, NewMFAs, SkipErtsSyms);
add_callee_mfas([], MFAs, _SkipErtsSyms) -> MFAs.
%%----------------------------------------------------------------
%%
mk_trampoline_map([], []) -> []; % archs not using trampolines
mk_trampoline_map(CalleeMFAs, Trampolines) ->
SizeofLong =
case erlang:system_info(hipe_architecture) of
amd64 -> 8;
ppc64 -> 8;
_ -> 4
end,
mk_trampoline_map(tuple_size(CalleeMFAs), CalleeMFAs,
Trampolines, SizeofLong, gb_trees:empty()).
mk_trampoline_map(I, CalleeMFAs, Trampolines, SizeofLong, Map) when I >= 1 ->
MFA = element(I, CalleeMFAs),
%% Trampoline = element(I, Trampolines),
Skip = (I-1)*SizeofLong,
<<_:Skip/binary-unit:8,
Trampoline:SizeofLong/integer-unsigned-native-unit:8,
_/binary>> = Trampolines,
NewMap = gb_trees:insert(MFA, Trampoline, Map),
mk_trampoline_map(I-1, CalleeMFAs, Trampolines, SizeofLong, NewMap);
mk_trampoline_map(0, _, _, _, Map) -> Map.
%%----------------------------------------------------------------
%%
trampoline_map_get(_, []) -> []; % archs not using trampolines
trampoline_map_get(MFA, Map) -> gb_trees:get(MFA, Map).
trampoline_map_lookup(_, []) -> []; % archs not using trampolines
trampoline_map_lookup(Primop, Map) ->
case gb_trees:lookup(Primop, Map) of
{value, X} -> X;
_ -> []
end.
%%------------------------------------------------------------------------
-record(fundef, {address :: integer(),
mfa :: mfa(),
is_closure :: boolean(),
is_exported :: boolean()}).
exports(ExportMap, BaseAddress) ->
exports(ExportMap, BaseAddress, [], []).
exports([Offset,M,F,A,IsClosure,IsExported|Rest], BaseAddress, MFAs, Addresses) ->
case IsExported andalso erlang:is_builtin(M, F, A) of
true ->
exports(Rest, BaseAddress, MFAs, Addresses);
_false ->
MFA = {M,F,A},
Address = BaseAddress + Offset,
FunDef = #fundef{address=Address, mfa=MFA, is_closure=IsClosure,
is_exported=IsExported},
exports(Rest, BaseAddress, [MFA|MFAs], [FunDef|Addresses])
end;
exports([], _, MFAs, Addresses) ->
{MFAs, Addresses}.
mod({M,_F,_A}) -> M.
%%------------------------------------------------------------------------
calculate_addresses(PatchOffsets, Base, Addresses) ->
RemoteOrLocal = local, % closure code refs are local
[{Data,
offsets_to_addresses(Offsets, Base),
get_native_address(DestMFA, Addresses, RemoteOrLocal)} ||
{{DestMFA,_,_}=Data,Offsets} <- PatchOffsets].
offsets_to_addresses(Os, Base) ->
[{O+Base,load_fe} || O <- Os].
%%------------------------------------------------------------------------
find_closure_patches([{Type,Refs} | Rest]) ->
case ?EXT2PATCH_TYPE(Type) of
load_address ->
find_closure_refs(Refs, Rest);
_ ->
find_closure_patches(Rest)
end;
find_closure_patches([]) -> [].
find_closure_refs([{Dest,Offsets} | Rest], Refs) ->
case Dest of
{closure,Data} ->
[{Data,Offsets}|find_closure_refs(Rest,Refs)];
_ ->
find_closure_refs(Rest,Refs)
end;
find_closure_refs([], Refs) ->
find_closure_patches(Refs).
%%------------------------------------------------------------------------
export_funs([FunDef | Addresses]) ->
#fundef{address=Address, mfa=MFA, is_closure=IsClosure,
is_exported=IsExported} = FunDef,
?IF_DEBUG({M,F,A} = MFA, no_debug),
?IF_DEBUG(
case IsClosure of
false ->
?debug_msg("LINKING: ~w:~w/~w to (0x~.16b)\n",
[M,F,A, Address]);
true ->
?debug_msg("LINKING: ~w:~w/~w to closure (0x~.16b)\n",
[M,F,A, Address])
end, no_debug),
hipe_bifs:set_funinfo_native_address(MFA, Address, IsExported),
hipe_bifs:set_native_address(MFA, Address, IsClosure),
export_funs(Addresses);
export_funs([]) ->
ok.
export_funs(Mod, Beam, Addresses, ClosuresToPatch) ->
Fs = [{F,A,Address} || #fundef{address=Address, mfa={_M,F,A}} <- Addresses],
Mod = code:make_stub_module(Mod, Beam, {Fs,ClosuresToPatch}),
ok.
%%========================================================================
%% Patching
%% @spec patch(refs(), BaseAddress::integer(), ConstAndZone::term(),
%% Addresses::term(), TrampolineMap::term()) -> 'ok'.
%% @type refs()=[{RefType::integer(), Reflist::reflist()} | refs()]
%%
%% @type reflist()= [{Data::term(), Offsets::offests()}|reflist()]
%% @type offsets()= [Offset::integer() | offsets()]
%% @doc
%% The patchlist is a list of lists of patches of a type.
%% For each type the list of references is sorted so that several
%% references to the same type of data come after each other
%% (we use this to look up the address of a referred function only once).
%%
patch([{Type,SortedRefs}|Rest], CodeAddress, ConstMap2, Addresses, TrampolineMap) ->
?debug_msg("Patching ~w at [~w+offset] with ~w\n",
[Type,CodeAddress,SortedRefs]),
case ?EXT2PATCH_TYPE(Type) of
call_local ->
patch_call(SortedRefs, CodeAddress, Addresses, 'local', TrampolineMap);
call_remote ->
patch_call(SortedRefs, CodeAddress, Addresses, 'remote', TrampolineMap);
Other ->
patch_all(Other, SortedRefs, CodeAddress, {ConstMap2,CodeAddress}, Addresses)
end,
patch(Rest, CodeAddress, ConstMap2, Addresses, TrampolineMap);
patch([], _, _, _, _) -> ok.
%%----------------------------------------------------------------
%% Handle a 'call_local' or 'call_remote' patch.
%%
patch_call([{DestMFA,Offsets}|SortedRefs], BaseAddress, Addresses, RemoteOrLocal, TrampolineMap) ->
case bif_address(DestMFA) of
false ->
%% Previous code used mfa_to_address(DestMFA, Addresses)
%% here for local calls. That is wrong because even local
%% destinations may not be present in Addresses: they may
%% not have been compiled yet, or they may be BEAM-only
%% functions (e.g. module_info).
DestAddress = get_native_address(DestMFA, Addresses, RemoteOrLocal),
Trampoline = trampoline_map_get(DestMFA, TrampolineMap),
patch_mfa_call_list(Offsets, BaseAddress, DestMFA, DestAddress, Addresses, RemoteOrLocal, Trampoline);
BifAddress when is_integer(BifAddress) ->
Trampoline = trampoline_map_lookup(DestMFA, TrampolineMap),
patch_bif_call_list(Offsets, BaseAddress, BifAddress, Trampoline)
end,
patch_call(SortedRefs, BaseAddress, Addresses, RemoteOrLocal, TrampolineMap);
patch_call([], _, _, _, _) ->
ok.
patch_bif_call_list([Offset|Offsets], BaseAddress, BifAddress, Trampoline) ->
CallAddress = BaseAddress+Offset,
?ASSERT(assert_local_patch(CallAddress)),
patch_call_insn(CallAddress, BifAddress, Trampoline),
patch_bif_call_list(Offsets, BaseAddress, BifAddress, Trampoline);
patch_bif_call_list([], _, _, _) -> ok.
patch_mfa_call_list([Offset|Offsets], BaseAddress, DestMFA, DestAddress, Addresses, RemoteOrLocal, Trampoline) ->
CallAddress = BaseAddress+Offset,
add_ref(DestMFA, CallAddress, Addresses, 'call', Trampoline, RemoteOrLocal),
?ASSERT(assert_local_patch(CallAddress)),
patch_call_insn(CallAddress, DestAddress, Trampoline),
patch_mfa_call_list(Offsets, BaseAddress, DestMFA, DestAddress, Addresses, RemoteOrLocal, Trampoline);
patch_mfa_call_list([], _, _, _, _, _, _) -> ok.
patch_call_insn(CallAddress, DestAddress, Trampoline) ->
%% This assertion is false when we're called from redirect/2.
%% ?ASSERT(assert_local_patch(CallAddress)),
hipe_bifs:patch_call(CallAddress, DestAddress, Trampoline).
%% ____________________________________________________________________
%%
patch_all(Type, [{Dest,Offsets}|Rest], BaseAddress, ConstAndZone, Addresses)->
patch_all_offsets(Type, Dest, Offsets, BaseAddress, ConstAndZone, Addresses),
patch_all(Type, Rest, BaseAddress, ConstAndZone, Addresses);
patch_all(_, [], _, _, _) -> ok.
patch_all_offsets(Type, Data, [Offset|Offsets], BaseAddress,
ConstAndZone, Addresses) ->
?debug_msg("Patching ~w at [~w+~w] with ~w\n",
[Type,BaseAddress,Offset, Data]),
Address = BaseAddress + Offset,
patch_offset(Type, Data, Address, ConstAndZone, Addresses),
?debug_msg("Patching done\n",[]),
patch_all_offsets(Type, Data, Offsets, BaseAddress, ConstAndZone, Addresses);
patch_all_offsets(_, _, [], _, _, _) -> ok.
%%----------------------------------------------------------------
%% Handle any patch type except 'call_local' or 'call_remote'.
%%
patch_offset(Type, Data, Address, ConstAndZone, Addresses) ->
case Type of
load_address ->
patch_load_address(Data, Address, ConstAndZone, Addresses);
load_atom ->
Atom = Data,
patch_atom(Address, Atom);
sdesc ->
patch_sdesc(Data, Address, ConstAndZone, Addresses);
x86_abs_pcrel ->
patch_instr(Address, Data, x86_abs_pcrel)
%% _ ->
%% ?error_msg("Unknown ref ~w ~w ~w\n", [Type, Address, Data]),
%% exit({unknown_reference, Type, Address, Data})
end.
patch_atom(Address, Atom) ->
?ASSERT(assert_local_patch(Address)),
patch_instr(Address, hipe_bifs:atom_to_word(Atom), atom).
patch_sdesc(?STACK_DESC(SymExnRA, FSize, Arity, Live),
Address, {_ConstMap2,CodeAddress}, _Addresses) ->
ExnRA =
case SymExnRA of
[] -> 0; % No catch
LabelOffset -> CodeAddress + LabelOffset
end,
?ASSERT(assert_local_patch(Address)),
DBG_MFA = ?IF_DEBUG(address_to_mfa_lth(Address, _Addresses), {undefined,undefined,0}),
hipe_bifs:enter_sdesc({Address, ExnRA, FSize, Arity, Live, DBG_MFA}).
%%----------------------------------------------------------------
%% Handle a 'load_address'-type patch.
%%
patch_load_address(Data, Address, ConstAndZone, Addresses) ->
case Data of
{local_function,DestMFA} ->
patch_load_mfa(Address, DestMFA, Addresses, 'local');
{remote_function,DestMFA} ->
patch_load_mfa(Address, DestMFA, Addresses, 'remote');
{constant,Name} ->
{ConstMap2,_CodeAddress} = ConstAndZone,
ConstAddress = find_const(Name, ConstMap2),
patch_instr(Address, ConstAddress, constant);
{closure,{DestMFA,Uniq,Index}} ->
patch_closure(DestMFA, Uniq, Index, Address, Addresses);
{c_const,CConst} ->
patch_instr(Address, bif_address(CConst), c_const)
end.
patch_closure(DestMFA, Uniq, Index, Address, Addresses) ->
case get(hipe_patch_closures) of
false ->
[]; % This is taken care of when registering the module.
true -> % We are not loading a module patch these closures
RemoteOrLocal = local, % closure code refs are local
DestAddress = get_native_address(DestMFA, Addresses, RemoteOrLocal),
BEAMAddress = hipe_bifs:fun_to_address(DestMFA),
FE = hipe_bifs:get_fe(mod(DestMFA), {Uniq, Index, BEAMAddress}),
hipe_bifs:set_native_address_in_fe(FE, DestAddress),
?debug_msg("Patch FE(~w) to 0x~.16b->0x~.16b (emu:0x~.16b)\n",
[DestMFA, FE, DestAddress, BEAMAddress]),
?ASSERT(assert_local_patch(Address)),
patch_instr(Address, FE, closure)
end.
%%----------------------------------------------------------------
%% Patch an instruction loading the address of an MFA.
%% RemoteOrLocal ::= 'remote' | 'local'
%%
patch_load_mfa(CodeAddress, DestMFA, Addresses, RemoteOrLocal) ->
DestAddress =
case bif_address(DestMFA) of
false ->
NativeAddress = get_native_address(DestMFA, Addresses, RemoteOrLocal),
add_ref(DestMFA, CodeAddress, Addresses, 'load_mfa', [], RemoteOrLocal),
NativeAddress;
BifAddress when is_integer(BifAddress) ->
BifAddress
end,
?ASSERT(assert_local_patch(CodeAddress)),
patch_instr(CodeAddress, DestAddress, 'load_mfa').
%%----------------------------------------------------------------
%% Patch references to code labels in the data segment.
%%
patch_consts(Labels, DataAddress, CodeAddress) ->
lists:foreach(fun (L) ->
patch_label_or_labels(L, DataAddress, CodeAddress)
end, Labels).
patch_label_or_labels({Pos,Offset}, DataAddress, CodeAddress) ->
?ASSERT(assert_local_patch(CodeAddress+Offset)),
write_word(DataAddress+Pos, CodeAddress+Offset);
patch_label_or_labels({sorted,Base,UnOrderdList}, DataAddress, CodeAddress) ->
sort_and_write(UnOrderdList, Base, DataAddress, CodeAddress).
sort_and_write(UnOrderdList, Base, DataAddress, CodeAddress) ->
WriteAndInc =
fun ({_, Offset}, DataPos) ->
?ASSERT(assert_local_patch(CodeAddress+Offset)),
write_word(DataPos, CodeAddress+Offset)
end,
lists:foldl(WriteAndInc, DataAddress+Base, sort_on_representation(UnOrderdList)).
sort_on_representation(List) ->
lists:sort([{hipe_bifs:term_to_word(Term), Offset} ||
{Term, Offset} <- List]).
%%--------------------------------------------------------------------
%% Update an instruction to refer to a value of a given type.
%%
%% Type ::= 'call' | 'load_mfa' | 'x86_abs_pcrel' | 'atom'
%% | 'constant' | 'c_const' | 'closure'
%%
%% Note: the values of this Type are hard-coded in file erl_bif_types.erl
%%
patch_instr(Address, Value, Type) ->
hipe_bifs:patch_insn(Address, Value, Type).
%%--------------------------------------------------------------------
%% Write a data word of the machine's natural word size.
%% Returns the address of the next word.
%%
%% XXX: It appears this is used for inserting both code addresses
%% and other data. In HiPE, code addresses are still 32-bit on
%% some 64-bit machines.
write_word(DataAddress, DataWord) ->
case erlang:system_info(hipe_architecture) of
amd64 ->
hipe_bifs:write_u64(DataAddress, DataWord),
DataAddress+8;
ppc64 ->
hipe_bifs:write_u64(DataAddress, DataWord),
DataAddress+8;
_ ->
hipe_bifs:write_u32(DataAddress, DataWord),
DataAddress+4
end.
%%--------------------------------------------------------------------
bif_address({M,F,A}) ->
hipe_bifs:bif_address(M,F,A);
bif_address(Name) when is_atom(Name) ->
hipe_bifs:primop_address(Name).
%%--------------------------------------------------------------------
%% create_data_segment/3 takes an object file ConstMap, as produced by
%% hipe_pack_constants:slim_constmap/1, loads the constants into
%% memory, and produces a ConstMap2 mapping each constant's ConstNo to
%% its runtime address, tagged if the constant is a term.
%%
create_data_segment(DataAlign, DataSize, DataList) ->
%%io:format("create_data_segment: \nDataAlign: ~p\nDataSize: ~p\nDataList: ~p\n",[DataAlign,DataSize,DataList]),
DataAddress = hipe_bifs:alloc_data(DataAlign, DataSize),
enter_data(DataList, [], DataAddress, DataSize).
enter_data(List, ConstMap2, DataAddress, DataSize) ->
case List of
[ConstNo,Offset,Type,Data|Rest] when is_integer(Offset) ->
%%?msg("Const ~w\n",[[ConstNo,Offset,Type,Data]]),
?ASSERT((Offset >= 0) and (Offset =< DataSize)),
Res = enter_datum(Type, Data, DataAddress+Offset),
enter_data(Rest, [{ConstNo,Res}|ConstMap2], DataAddress, DataSize);
[] ->
{DataAddress, ConstMap2}
end.
enter_datum(Type, Data, Address) ->
case ?EXT2CONST_TYPE(Type) of
term ->
%% Address is unused for terms
hipe_bifs:term_to_word(hipe_bifs:merge_term(Data));
sorted_block ->
L = lists:sort([hipe_bifs:term_to_word(Term) || Term <- Data]),
write_words(L, Address),
Address;
block ->
case Data of
{Lbls, []} ->
write_bytes(Lbls, Address);
{Lbls, SortOrder} ->
SortedLbls = [Lbl || {_,Lbl} <- lists:sort(group(Lbls, SortOrder))],
write_words(SortedLbls, Address);
Lbls ->
write_bytes(Lbls, Address)
end,
Address
end.
group([], []) ->
[];
group([B1,B2,B3,B4|Ls], [O|Os]) ->
[{hipe_bifs:term_to_word(O),bytes_to_32(B4,B3,B2,B1)}|group(Ls,Os)].
bytes_to_32(B4,B3,B2,B1) ->
(B4 bsl 24) bor (B3 bsl 16) bor (B2 bsl 8) bor B1.
write_words([W|Rest], Addr) ->
write_words(Rest, write_word(Addr, W));
write_words([], Addr) when is_integer(Addr) -> true.
write_bytes([B|Rest], Addr) ->
hipe_bifs:write_u8(Addr, B),
write_bytes(Rest, Addr+1);
write_bytes([], Addr) when is_integer(Addr) -> true.
%%% lists:keysearch/3 conses a useless wrapper around the found tuple :-(
%%% otherwise it would have been a good replacement for this loop
find_const(ConstNo, [{ConstNo,Addr}|_ConstMap2]) ->
Addr;
find_const(ConstNo, [_|ConstMap2]) ->
find_const(ConstNo, ConstMap2);
find_const(ConstNo, []) ->
?error_msg("Constant not found ~w\n",[ConstNo]),
exit({constant_not_found,ConstNo}).
%%----------------------------------------------------------------
%% Record that the code at address 'Address' has a reference
%% of type 'RefType' ('call' or 'load_mfa') to 'CalleeMFA'.
%% 'Addresses' must be an address-descending list from exports/2.
%%
%% If 'RefType' is 'call', then 'Trampoline' may be the address
%% of a stub branching to 'CalleeMFA', where the stub is reachable
%% from 'Address' via a normal call or tailcall instruction.
%%
%% RemoteOrLocal ::= 'remote' | 'local'.
%%
%%
%% -record(ref, {caller_mfa, address, ref_type, trampoline, remote_or_local}).
%%
add_ref(CalleeMFA, Address, Addresses, RefType, Trampoline, RemoteOrLocal) ->
CallerMFA = address_to_mfa_lth(Address, Addresses),
%% just a sanity assertion below
true = case RemoteOrLocal of
local ->
{M1,_,_} = CalleeMFA,
{M2,_,_} = CallerMFA,
M1 =:= M2;
remote ->
true
end,
%% io:format("Adding ref ~w\n",[{CallerMFA, CalleeMFA, Address, RefType}]),
hipe_bifs:add_ref(CalleeMFA, {CallerMFA,Address,RefType,Trampoline,RemoteOrLocal}).
% For FunDefs sorted from low to high addresses
address_to_mfa_lth(Address, FunDefs) ->
case address_to_mfa_lth(Address, FunDefs, false) of
false ->
?error_msg("Local adddress not found ~w\n",[Address]),
exit({?MODULE, local_address_not_found});
MFA ->
MFA
end.
address_to_mfa_lth(Address, [#fundef{address=Adr, mfa=MFA}|Rest], Prev) ->
if Address < Adr ->
Prev;
true ->
address_to_mfa_lth(Address, Rest, MFA)
end;
address_to_mfa_lth(_Address, [], Prev) ->
Prev.
% For FunDefs sorted from high to low addresses
%% address_to_mfa_htl(Address, [#fundef{address=Adr, mfa=MFA}|_Rest]) when Address >= Adr -> MFA;
%% address_to_mfa_htl(Address, [_ | Rest]) -> address_to_mfa_htl(Address, Rest);
%% address_to_mfa_htl(Address, []) ->
%% ?error_msg("Local adddress not found ~w\n",[Address]),
%% exit({?MODULE, local_address_not_found}).
%%----------------------------------------------------------------
%% Change callers of the given module to instead trap to BEAM.
%% load_native_code/2 calls this just before loading native code.
%%
patch_to_emu(Mod) ->
patch_to_emu_step2(patch_to_emu_step1(Mod)).
%% Step 1 must occur before the loading of native code updates
%% references information or creates a new BEAM stub module.
patch_to_emu_step1(Mod) ->
case is_loaded(Mod) of
true ->
%% Get exported functions
MFAs = [{Mod,Fun,Arity} || {Fun,Arity} <- Mod:module_info(exports)],
%% get_refs_from/2 only finds references from compiled static
%% call sites to the module, but some native address entries
%% were added as the result of dynamic apply calls. We must
%% purge them too, but we have no explicit record of them.
%% Therefore invalidate all native addresses for the module.
%% emu_make_stubs/1 will repair the ones for compiled static calls.
hipe_bifs:invalidate_funinfo_native_addresses(MFAs),
%% Find all call sites that call these MFAs. As a side-effect,
%% create native stubs for any MFAs that are referred.
ReferencesToPatch = get_refs_from(MFAs, []),
ok = remove_refs_from(MFAs),
ReferencesToPatch;
false ->
%% The first time we load the module, no redirection needs to be done.
[]
end.
%% Step 2 must occur after the new BEAM stub module is created.
patch_to_emu_step2(ReferencesToPatch) ->
emu_make_stubs(ReferencesToPatch),
redirect(ReferencesToPatch).
-spec is_loaded(Module::atom()) -> boolean().
%% @doc Checks whether a module is loaded or not.
is_loaded(M) when is_atom(M) ->
try hipe_bifs:fun_to_address({M,module_info,0}) of
I when is_integer(I) -> true
catch _:_ -> false
end.
-ifdef(notdef).
emu_make_stubs([{MFA,_Refs}|Rest]) ->
make_stub(MFA),
emu_make_stubs(Rest);
emu_make_stubs([]) ->
[].
make_stub({_,_,A} = MFA) ->
EmuAddress = hipe_bifs:get_emu_address(MFA),
StubAddress = hipe_bifs:make_native_stub(EmuAddress, A),
hipe_bifs:set_funinfo_native_address(MFA, StubAddress).
-else.
emu_make_stubs(_) -> [].
-endif.
%%--------------------------------------------------------------------
%% Given a list of MFAs, tag them with their referred_from references.
%% The resulting {MFA,Refs} list is later passed to redirect/1, once
%% the MFAs have been bound to (possibly new) native-code addresses.
%%
get_refs_from(MFAs, []) ->
mark_referred_from(MFAs),
MFAs.
mark_referred_from(MFAs) ->
lists:foreach(fun(MFA) -> hipe_bifs:mark_referred_from(MFA) end, MFAs).
%%--------------------------------------------------------------------
%% Given a list of MFAs with referred_from references, update their
%% callers to refer to their new native-code addresses.
%%
%% The {MFA,Refs} list must come from get_refs_from/2.
%%
redirect(MFAs) ->
lists:foreach(fun(MFA) -> hipe_bifs:redirect_referred_from(MFA) end, MFAs).
%%--------------------------------------------------------------------
%% Given a list of MFAs, remove all referred_from references having
%% any of them as CallerMFA.
%%
%% This is the only place using refers_to. Whenever a reference is
%% added from CallerMFA to CalleeMFA, CallerMFA is added to CalleeMFA's
%% referred_from list, and CalleeMFA is added to CallerMFA's refers_to
%% list. The refers_to list is used here to find the CalleeMFAs whose
%% referred_from lists should be updated.
%%
remove_refs_from(MFAs) ->
lists:foreach(fun(MFA) -> hipe_bifs:remove_refs_from(MFA) end, MFAs).
%%--------------------------------------------------------------------
%% To find the native code of an MFA we need to look in 3 places:
%% 1. If it is compiled now look in the Addresses data structure.
%% 2. Then look in native_addresses from module info.
%% 3. Then (the function might have been singled compiled) look in
%% hipe_funinfo
%% If all else fails create a native stub for the MFA
get_native_address(MFA, Addresses, RemoteOrLocal) ->
case mfa_to_address(MFA, Addresses, RemoteOrLocal) of
Adr when is_integer(Adr) -> Adr;
false ->
IsRemote =
case RemoteOrLocal of
remote -> true;
local -> false
end,
hipe_bifs:find_na_or_make_stub(MFA, IsRemote)
end.
mfa_to_address(MFA, [#fundef{address=Adr, mfa=MFA,
is_exported=IsExported}|_Rest], RemoteOrLocal) ->
case RemoteOrLocal of
local ->
Adr;
remote ->
case IsExported of
true ->
Adr;
false ->
false
end
end;
mfa_to_address(MFA, [_|Rest], RemoteOrLocal) ->
mfa_to_address(MFA, Rest, RemoteOrLocal);
mfa_to_address(_, [], _) -> false.
%% ____________________________________________________________________
%%
-ifdef(DO_ASSERT).
-define(init_assert_patch(Base, Size), put(hipe_assert_code_area,{Base,Base+Size})).
assert_local_patch(Address) when is_integer(Address) ->
{First,Last} = get(hipe_assert_code_area),
Address >= First andalso Address < (Last).
-else.
-define(init_assert_patch(Base, Size), ok).
-endif.
%% ____________________________________________________________________
%%
%% Beam: nil() | binary() (used as a flag)
enter_code(CodeSize, CodeBinary, CalleeMFAs, Mod, Beam) ->
true = byte_size(CodeBinary) =:= CodeSize,
hipe_bifs:update_code_size(Mod, Beam, CodeSize),
{CodeAddress,Trampolines} = hipe_bifs:enter_code(CodeBinary, CalleeMFAs),
?init_assert_patch(CodeAddress, byte_size(CodeBinary)),
{CodeAddress,Trampolines}.
