1 //===---------------- TPCDynamicLibrarySearchGenerator.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/TPCDynamicLibrarySearchGenerator.h"
10
11 namespace llvm {
12 namespace orc {
13
14 Expected<std::unique_ptr<TPCDynamicLibrarySearchGenerator>>
Load(TargetProcessControl & TPC,const char * LibraryPath,SymbolPredicate Allow)15 TPCDynamicLibrarySearchGenerator::Load(TargetProcessControl &TPC,
16 const char *LibraryPath,
17 SymbolPredicate Allow) {
18 auto Handle = TPC.loadDylib(LibraryPath);
19 if (!Handle)
20 return Handle.takeError();
21
22 return std::make_unique<TPCDynamicLibrarySearchGenerator>(TPC, *Handle,
23 std::move(Allow));
24 }
25
tryToGenerate(LookupState & LS,LookupKind K,JITDylib & JD,JITDylibLookupFlags JDLookupFlags,const SymbolLookupSet & Symbols)26 Error TPCDynamicLibrarySearchGenerator::tryToGenerate(
27 LookupState &LS, LookupKind K, JITDylib &JD,
28 JITDylibLookupFlags JDLookupFlags, const SymbolLookupSet &Symbols) {
29
30 if (Symbols.empty())
31 return Error::success();
32
33 SymbolLookupSet LookupSymbols;
34
35 for (auto &KV : Symbols) {
36 // Skip symbols that don't match the filter.
37 if (Allow && !Allow(KV.first))
38 continue;
39 LookupSymbols.add(KV.first, SymbolLookupFlags::WeaklyReferencedSymbol);
40 }
41
42 SymbolMap NewSymbols;
43
44 tpctypes::LookupRequest Request(H, LookupSymbols);
45 auto Result = TPC.lookupSymbols(Request);
46 if (!Result)
47 return Result.takeError();
48
49 assert(Result->size() == 1 && "Results for more than one library returned");
50 assert(Result->front().size() == LookupSymbols.size() &&
51 "Result has incorrect number of elements");
52
53 auto ResultI = Result->front().begin();
54 for (auto &KV : LookupSymbols) {
55 if (*ResultI)
56 NewSymbols[KV.first] =
57 JITEvaluatedSymbol(*ResultI, JITSymbolFlags::Exported);
58 ResultI++;
59 }
60
61 // If there were no resolved symbols bail out.
62 if (NewSymbols.empty())
63 return Error::success();
64
65 // Define resolved symbols.
66 return JD.define(absoluteSymbols(std::move(NewSymbols)));
67 }
68
69 } // end namespace orc
70 } // end namespace llvm
71