1 //===- LambdaResolverMM - Redirect symbol lookup via a functor --*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // Defines a RuntimeDyld::SymbolResolver subclass that uses a user-supplied 11 // functor for symbol resolution. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #ifndef LLVM_EXECUTIONENGINE_ORC_LAMBDARESOLVER_H 16 #define LLVM_EXECUTIONENGINE_ORC_LAMBDARESOLVER_H 17 18 #include "llvm/ADT/STLExtras.h" 19 #include "llvm/ExecutionEngine/JITSymbol.h" 20 #include <memory> 21 22 namespace llvm { 23 namespace orc { 24 25 template <typename DylibLookupFtorT, typename ExternalLookupFtorT> 26 class LambdaResolver : public JITSymbolResolver { 27 public: 28 LambdaResolver(DylibLookupFtorT DylibLookupFtor, 29 ExternalLookupFtorT ExternalLookupFtor) 30 : DylibLookupFtor(DylibLookupFtor), 31 ExternalLookupFtor(ExternalLookupFtor) {} 32 33 JITSymbol findSymbolInLogicalDylib(const std::string &Name) final { 34 return DylibLookupFtor(Name); 35 } 36 37 JITSymbol findSymbol(const std::string &Name) final { 38 return ExternalLookupFtor(Name); 39 } 40 41 private: 42 DylibLookupFtorT DylibLookupFtor; 43 ExternalLookupFtorT ExternalLookupFtor; 44 }; 45 46 template <typename DylibLookupFtorT, 47 typename ExternalLookupFtorT> 48 std::shared_ptr<LambdaResolver<DylibLookupFtorT, ExternalLookupFtorT>> 49 createLambdaResolver(DylibLookupFtorT DylibLookupFtor, 50 ExternalLookupFtorT ExternalLookupFtor) { 51 using LR = LambdaResolver<DylibLookupFtorT, ExternalLookupFtorT>; 52 return make_unique<LR>(std::move(DylibLookupFtor), 53 std::move(ExternalLookupFtor)); 54 } 55 56 } // end namespace orc 57 } // end namespace llvm 58 59 #endif // LLVM_EXECUTIONENGINE_ORC_LAMBDARESOLVER_H 60