2487db1f28
Wrapper function call and dispatch handler helpers are moved to ExecutionSession, and existing EPC-based tools are re-written to take an ExecutionSession argument instead. Requiring an ExecutorProcessControl instance simplifies existing EPC based utilities (which only need to take an ES now), and should encourage more utilities to use the EPC interface. It also simplifies process termination, since the session can automatically call ExecutorProcessControl::disconnect (previously this had to be done manually, and carefully ordered with the rest of JIT tear-down to work correctly).
74 lines
2.7 KiB
C++
74 lines
2.7 KiB
C++
//===------ EPCEHFrameRegistrar.cpp - EPC-based eh-frame registration -----===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ExecutionEngine/Orc/EPCEHFrameRegistrar.h"
|
|
|
|
#include "llvm/ExecutionEngine/Orc/Core.h"
|
|
#include "llvm/Support/BinaryStreamWriter.h"
|
|
|
|
using namespace llvm::orc::shared;
|
|
|
|
namespace llvm {
|
|
namespace orc {
|
|
|
|
Expected<std::unique_ptr<EPCEHFrameRegistrar>>
|
|
EPCEHFrameRegistrar::Create(ExecutionSession &ES) {
|
|
// FIXME: Proper mangling here -- we really need to decouple linker mangling
|
|
// from DataLayout.
|
|
|
|
// Find the addresses of the registration/deregistration functions in the
|
|
// executor process.
|
|
auto &EPC = ES.getExecutorProcessControl();
|
|
auto ProcessHandle = EPC.loadDylib(nullptr);
|
|
if (!ProcessHandle)
|
|
return ProcessHandle.takeError();
|
|
|
|
std::string RegisterWrapperName, DeregisterWrapperName;
|
|
if (EPC.getTargetTriple().isOSBinFormatMachO()) {
|
|
RegisterWrapperName += '_';
|
|
DeregisterWrapperName += '_';
|
|
}
|
|
RegisterWrapperName += "llvm_orc_registerEHFrameSectionWrapper";
|
|
DeregisterWrapperName += "llvm_orc_deregisterEHFrameSectionWrapper";
|
|
|
|
SymbolLookupSet RegistrationSymbols;
|
|
RegistrationSymbols.add(EPC.intern(RegisterWrapperName));
|
|
RegistrationSymbols.add(EPC.intern(DeregisterWrapperName));
|
|
|
|
auto Result = EPC.lookupSymbols({{*ProcessHandle, RegistrationSymbols}});
|
|
if (!Result)
|
|
return Result.takeError();
|
|
|
|
assert(Result->size() == 1 && "Unexpected number of dylibs in result");
|
|
assert((*Result)[0].size() == 2 &&
|
|
"Unexpected number of addresses in result");
|
|
|
|
auto RegisterEHFrameWrapperFnAddr = (*Result)[0][0];
|
|
auto DeregisterEHFrameWrapperFnAddr = (*Result)[0][1];
|
|
|
|
return std::make_unique<EPCEHFrameRegistrar>(ES, RegisterEHFrameWrapperFnAddr,
|
|
DeregisterEHFrameWrapperFnAddr);
|
|
}
|
|
|
|
Error EPCEHFrameRegistrar::registerEHFrames(JITTargetAddress EHFrameSectionAddr,
|
|
size_t EHFrameSectionSize) {
|
|
return ES.callSPSWrapper<void(SPSExecutorAddress, uint64_t)>(
|
|
RegisterEHFrameWrapperFnAddr, EHFrameSectionAddr,
|
|
static_cast<uint64_t>(EHFrameSectionSize));
|
|
}
|
|
|
|
Error EPCEHFrameRegistrar::deregisterEHFrames(
|
|
JITTargetAddress EHFrameSectionAddr, size_t EHFrameSectionSize) {
|
|
return ES.callSPSWrapper<void(SPSExecutorAddress, uint64_t)>(
|
|
DeregisterEHFrameWrapperFnAddr, EHFrameSectionAddr,
|
|
static_cast<uint64_t>(EHFrameSectionSize));
|
|
}
|
|
|
|
} // end namespace orc
|
|
} // end namespace llvm
|