40b4ac278e
This migrates the dylib manager lookup and related APIs to replace ExecutorAddress with ExecutorSymbolDef so that in the future we can model JITSymbolFlags for these symbols. The current change should be NFC as we are only setting the Exported symbol flag.
60 lines
2.1 KiB
C++
60 lines
2.1 KiB
C++
//===----- EPCDebugObjectRegistrar.cpp - EPC-based debug 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/EPCDebugObjectRegistrar.h"
|
|
|
|
#include "llvm/ExecutionEngine/Orc/Core.h"
|
|
#include "llvm/ExecutionEngine/Orc/Shared/SimplePackedSerialization.h"
|
|
#include "llvm/ExecutionEngine/Orc/TargetProcess/JITLoaderGDB.h"
|
|
#include "llvm/Support/BinaryStreamWriter.h"
|
|
|
|
namespace llvm {
|
|
namespace orc {
|
|
|
|
Expected<std::unique_ptr<EPCDebugObjectRegistrar>> createJITLoaderGDBRegistrar(
|
|
ExecutionSession &ES,
|
|
std::optional<ExecutorAddr> RegistrationFunctionDylib) {
|
|
auto &EPC = ES.getExecutorProcessControl();
|
|
|
|
if (!RegistrationFunctionDylib) {
|
|
if (auto D = EPC.loadDylib(nullptr))
|
|
RegistrationFunctionDylib = *D;
|
|
else
|
|
return D.takeError();
|
|
}
|
|
|
|
SymbolStringPtr RegisterFn =
|
|
EPC.getTargetTriple().isOSBinFormatMachO()
|
|
? EPC.intern("_llvm_orc_registerJITLoaderGDBWrapper")
|
|
: EPC.intern("llvm_orc_registerJITLoaderGDBWrapper");
|
|
|
|
SymbolLookupSet RegistrationSymbols;
|
|
RegistrationSymbols.add(RegisterFn);
|
|
|
|
auto Result =
|
|
EPC.lookupSymbols({{*RegistrationFunctionDylib, RegistrationSymbols}});
|
|
if (!Result)
|
|
return Result.takeError();
|
|
|
|
assert(Result->size() == 1 && "Unexpected number of dylibs in result");
|
|
assert((*Result)[0].size() == 1 &&
|
|
"Unexpected number of addresses in result");
|
|
|
|
ExecutorAddr RegisterAddr = (*Result)[0][0].getAddress();
|
|
return std::make_unique<EPCDebugObjectRegistrar>(ES, RegisterAddr);
|
|
}
|
|
|
|
Error EPCDebugObjectRegistrar::registerDebugObject(ExecutorAddrRange TargetMem,
|
|
bool AutoRegisterCode) {
|
|
return ES.callSPSWrapper<void(shared::SPSExecutorAddrRange, bool)>(
|
|
RegisterFn, TargetMem, AutoRegisterCode);
|
|
}
|
|
|
|
} // namespace orc
|
|
} // namespace llvm
|