e05d80ec7b
This patch adds the new **executor-side resolver API** as suggested by @lhames. It introduces a `DylibSymbolResolver` that helps resolve symbols for each loaded dylib. Previously, we returned a `DylibHandle` to the controller. Now, we wrap the native handle inside `DylibSymbolResolver` and return a `ResolverHandle` instead. This makes the code cleaner and separates the symbol resolution logic from raw handle management.
47 lines
1.4 KiB
C++
47 lines
1.4 KiB
C++
|
|
#include "llvm/ExecutionEngine/Orc/TargetProcess/ExecutorResolver.h"
|
|
|
|
#include "llvm/Support/DynamicLibrary.h"
|
|
#include "llvm/Support/Error.h"
|
|
|
|
namespace llvm::orc {
|
|
|
|
void DylibSymbolResolver::resolveAsync(
|
|
const RemoteSymbolLookupSet &L,
|
|
ExecutorResolver::YieldResolveResultFn &&OnResolve) {
|
|
std::vector<std::optional<ExecutorSymbolDef>> Result;
|
|
auto DL = sys::DynamicLibrary(Handle.toPtr<void *>());
|
|
|
|
for (const auto &E : L) {
|
|
if (E.Name.empty()) {
|
|
if (E.Required)
|
|
OnResolve(
|
|
make_error<StringError>("Required address for empty symbol \"\"",
|
|
inconvertibleErrorCode()));
|
|
else
|
|
Result.emplace_back();
|
|
} else {
|
|
|
|
const char *DemangledSymName = E.Name.c_str();
|
|
#ifdef __APPLE__
|
|
if (E.Name.front() != '_')
|
|
OnResolve(make_error<StringError>(Twine("MachO symbol \"") + E.Name +
|
|
"\" missing leading '_'",
|
|
inconvertibleErrorCode()));
|
|
++DemangledSymName;
|
|
#endif
|
|
|
|
void *Addr = DL.getAddressOfSymbol(DemangledSymName);
|
|
if (!Addr && E.Required)
|
|
Result.emplace_back();
|
|
else
|
|
// FIXME: determine accurate JITSymbolFlags.
|
|
Result.emplace_back(ExecutorSymbolDef(ExecutorAddr::fromPtr(Addr),
|
|
JITSymbolFlags::Exported));
|
|
}
|
|
}
|
|
|
|
OnResolve(std::move(Result));
|
|
}
|
|
|
|
} // end namespace llvm::orc
|