1//===---------------- EPCDynamicLibrarySearchGenerator.cpp ----------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "llvm/ExecutionEngine/Orc/EPCDynamicLibrarySearchGenerator.h"
10
11#include "llvm/ExecutionEngine/Orc/AbsoluteSymbols.h"
12#include "llvm/ExecutionEngine/Orc/DebugUtils.h"
13#include "llvm/Support/Error.h"
14
15#define DEBUG_TYPE "orc"
16
17namespace llvm {
18namespace orc {
19
20Expected<std::unique_ptr<EPCDynamicLibrarySearchGenerator>>
21EPCDynamicLibrarySearchGenerator::Load(
22 ExecutionSession &ES, DylibManager &DylibMgr, const char *LibraryPath,
23 SymbolPredicate Allow, AddAbsoluteSymbolsFn AddAbsoluteSymbols) {
24 auto Handle = DylibMgr.loadDylib(DylibPath: LibraryPath);
25 if (!Handle)
26 return Handle.takeError();
27
28 return std::make_unique<EPCDynamicLibrarySearchGenerator>(
29 args&: ES, args&: DylibMgr, args&: *Handle, args: std::move(Allow), args: std::move(AddAbsoluteSymbols));
30}
31
32Error EPCDynamicLibrarySearchGenerator::tryToGenerate(
33 LookupState &LS, LookupKind K, JITDylib &JD,
34 JITDylibLookupFlags JDLookupFlags, const SymbolLookupSet &Symbols) {
35
36 if (Symbols.empty())
37 return Error::success();
38
39 LLVM_DEBUG({
40 dbgs() << "EPCDynamicLibrarySearchGenerator trying to generate "
41 << Symbols << "\n";
42 });
43
44 // If there's no handle then resolve all requested symbols to null.
45 if (!H) {
46 assert(Allow && "No handle or filter?");
47 SymbolMap Nulls;
48 for (auto &[Name, LookupFlags] : Symbols) {
49 if (Allow(Name))
50 Nulls[Name] = {};
51 }
52 return addAbsolutes(JD, Symbols: std::move(Nulls));
53 }
54
55 // Otherwise proceed with lookup in the remote.
56 SymbolLookupSet LookupSymbols;
57
58 for (auto &KV : Symbols) {
59 // Skip symbols that don't match the filter.
60 if (Allow && !Allow(KV.first))
61 continue;
62 LookupSymbols.add(Name: KV.first, Flags: SymbolLookupFlags::WeaklyReferencedSymbol);
63 }
64
65 DylibMgr.lookupSymbolsAsync(
66 H: *H, Symbols: LookupSymbols,
67 F: [this, &JD, LS = std::move(LS), LookupSymbols](auto Result) mutable {
68 if (!Result) {
69 LLVM_DEBUG({
70 dbgs() << "EPCDynamicLibrarySearchGenerator lookup failed due to "
71 "error";
72 });
73 return LS.continueLookup(Err: Result.takeError());
74 }
75
76 assert(Result->size() == LookupSymbols.size() &&
77 "Result has incorrect number of elements");
78
79 auto SymsIt = Result->begin();
80 SymbolNameSet MissingSymbols;
81 SymbolMap NewSymbols;
82 for (auto &[Name, Flags] : LookupSymbols) {
83 const auto &Sym = *SymsIt++;
84 if (Sym && *Sym)
85 NewSymbols[Name] = {*Sym, JITSymbolFlags::Exported};
86 else if (LLVM_UNLIKELY(!Sym &&
87 Flags == SymbolLookupFlags::RequiredSymbol))
88 MissingSymbols.insert(V: Name);
89 }
90
91 LLVM_DEBUG({
92 dbgs() << "EPCDynamicLibrarySearchGenerator lookup returned "
93 << NewSymbols << "\n";
94 });
95
96 // If there were no resolved symbols bail out.
97 if (NewSymbols.empty())
98 return LS.continueLookup(Err: Error::success());
99
100 if (LLVM_UNLIKELY(!MissingSymbols.empty()))
101 return LS.continueLookup(Err: make_error<SymbolsNotFound>(
102 Args: this->ES.getSymbolStringPool(), Args: std::move(MissingSymbols)));
103
104 // Define resolved symbols.
105 Error Err = addAbsolutes(JD, Symbols: std::move(NewSymbols));
106
107 LS.continueLookup(Err: std::move(Err));
108 });
109
110 return Error::success();
111}
112
113Error EPCDynamicLibrarySearchGenerator::addAbsolutes(JITDylib &JD,
114 SymbolMap Symbols) {
115 return AddAbsoluteSymbols ? AddAbsoluteSymbols(JD, std::move(Symbols))
116 : JD.define(MU: absoluteSymbols(Symbols: std::move(Symbols)));
117}
118
119} // end namespace orc
120} // end namespace llvm
121