Lang Hames
ac0f0407a8
NativeDylibManager::load now handles an empty path by returning the process's global lookup handle (RTLD_DEFAULT on POSIX) directly, bypassing dlopen and the shutdown-time dlclose registration. This matches the behavior of OrcTargetProcess's SimpleExecutorDylibManager.
100 lines
3.4 KiB
C++
100 lines
3.4 KiB
C++
//===- NativeDylibManager.cpp ---------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// NativeDylibManager and related APIs.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "orc-rt/NativeDylibManager.h"
|
|
#include "orc-rt/Session.h"
|
|
|
|
#include <sstream> // For NativeDylibAPIs.inc.
|
|
|
|
#if defined(__APPLE__) || defined(__linux__)
|
|
#include "Unix/NativeDylibAPIs.inc"
|
|
#else
|
|
#error "Target OS dylib APIs unsupported"
|
|
#endif
|
|
|
|
namespace orc_rt {
|
|
|
|
Expected<std::unique_ptr<NativeDylibManager>>
|
|
NativeDylibManager::Create(Session &S, SimpleSymbolTable &ST,
|
|
const char *InstanceName,
|
|
SimpleSymbolTable::MutatorFn AddInterface) {
|
|
|
|
std::unique_ptr<NativeDylibManager> Instance(new NativeDylibManager(S));
|
|
|
|
SimpleSymbolTable NDMST;
|
|
if (auto Err = AddInterface(NDMST))
|
|
return Err;
|
|
std::pair<const char *, const void *> InstanceSym[] = {
|
|
{InstanceName, static_cast<const void *>(Instance.get())}};
|
|
if (auto Err = NDMST.addUnique(InstanceSym))
|
|
return std::move(Err);
|
|
|
|
if (auto Err = ST.addUnique(NDMST))
|
|
return std::move(Err);
|
|
|
|
return std::move(Instance);
|
|
}
|
|
|
|
void NativeDylibManager::load(OnLoadCompleteFn &&OnComplete, std::string Path) {
|
|
// Empty path -> global handle; no shutdown callback (RTLD_DEFAULT
|
|
// mustn't be dlclose'd).
|
|
if (Path.empty())
|
|
return OnComplete(hostOSGetGlobalLookupHandle());
|
|
|
|
auto H = hostOSLoadLibrary(Path);
|
|
if (!H)
|
|
return OnComplete(H.takeError());
|
|
|
|
// Capture S by reference, rather than this, so that the callback remains
|
|
// valid even if the NativeDylibManager is destroyed prior to shutdown.
|
|
S.addOnShutdown([&S = this->S, Handle = *H]() {
|
|
if (auto Err = hostOSUnloadLibrary(Handle))
|
|
S.reportError(std::move(Err));
|
|
});
|
|
OnComplete(std::move(H));
|
|
}
|
|
|
|
void NativeDylibManager::lookup(OnLookupCompleteFn &&OnLookupComplete,
|
|
void *Handle, SymbolLookupSet Symbols) {
|
|
std::vector<std::string> Names;
|
|
Names.reserve(Symbols.size());
|
|
for (auto &S : Symbols)
|
|
Names.push_back(std::move(S.first));
|
|
|
|
auto Addrs = hostOSLibraryLookup(Handle, Names);
|
|
|
|
// Convert weak-missing entries (empty optional from hostOSLibraryLookup)
|
|
// to a present zero address. This matches the resolve semantics of
|
|
// llvm::orc::rt_bootstrap::SimpleExecutorDylibManager: an empty optional
|
|
// in the result signals a missing required symbol, while a missing
|
|
// weakly-referenced symbol is reported as a zero address.
|
|
for (size_t I = 0, E = Symbols.size(); I != E; ++I)
|
|
if (!Addrs[I] && Symbols[I].second == WeaklyReferencedSymbol)
|
|
Addrs[I] = nullptr;
|
|
|
|
OnLookupComplete(std::move(Addrs));
|
|
}
|
|
|
|
void NativeDylibManager::onDetach(Service::OnCompleteFn OnComplete,
|
|
bool ShutdownRequested) {
|
|
// Detach is a noop for now. If/when we add bloom-filter support this will be
|
|
// a good time to update filters.
|
|
OnComplete();
|
|
}
|
|
|
|
void NativeDylibManager::onShutdown(Service::OnCompleteFn OnComplete) {
|
|
// Unloads happen via Session shutdown callbacks registered in load().
|
|
OnComplete();
|
|
}
|
|
|
|
} // namespace orc_rt
|