ExecutorResolutionGenerator.h

Go to the documentation of this file.

//===----- ExecutorResolver.h - Resolve symbols in executor -----*- C++ -*-===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// Declares ExecutorResolutionGenerator for symbol resolution,
// dynamic library loading, and lookup in an executor process via
// ExecutorResolver.
//
//===----------------------------------------------------------------------===//
 
#ifndef LLVM_EXECUTIONENGINE_ORC_EXECUTORRESOLUTIONGENERATOR_H
#define LLVM_EXECUTIONENGINE_ORC_EXECUTORRESOLUTIONGENERATOR_H
 
#include "llvm/ADT/FunctionExtras.h"
#include "llvm/ExecutionEngine/Orc/AbsoluteSymbols.h"
#include "llvm/ExecutionEngine/Orc/Core.h"
#include "llvm/ExecutionEngine/Orc/DylibManager.h"
 
namespace llvm::orc {
 
class ExecutorResolutionGenerator : public DefinitionGenerator {
public:
  using SymbolPredicate = unique_function<bool(const SymbolStringPtr &)>;
  using AbsoluteSymbolsFn =
      unique_function<std::unique_ptr<MaterializationUnit>(SymbolMap)>;
 
  ExecutorResolutionGenerator(
      ExecutionSession &ES, DylibManager &DylibMgr, tpctypes::ResolverHandle H,
      SymbolPredicate Allow = SymbolPredicate(),
      AbsoluteSymbolsFn AbsoluteSymbols = absoluteSymbols)
      : ES(ES), DylibMgr(DylibMgr), H(H), Allow(std::move(Allow)),
        AbsoluteSymbols(std::move(AbsoluteSymbols)) {}
 
  ExecutorResolutionGenerator(
      ExecutionSession &ES, DylibManager &DylibMgr,
      SymbolPredicate Allow = SymbolPredicate(),
      AbsoluteSymbolsFn AbsoluteSymbols = absoluteSymbols)
      : ES(ES), DylibMgr(DylibMgr), Allow(std::move(Allow)),
        AbsoluteSymbols(std::move(AbsoluteSymbols)) {}
 
  /// Permanently loads the library at the given path and, on success, returns
  /// an ExecutorResolutionGenerator that will search it for symbol
  /// definitions in the library. On failure returns the reason the library
  /// failed to load.
  static Expected<std::unique_ptr<ExecutorResolutionGenerator>>
  Load(ExecutionSession &ES, DylibManager &DylibMgr, const char *LibraryPath,
       SymbolPredicate Allow = SymbolPredicate(),
       AbsoluteSymbolsFn AbsoluteSymbols = absoluteSymbols);
 
  /// Creates a ExecutorResolutionGenerator that searches for symbols in
  /// the target process.
  static Expected<std::unique_ptr<ExecutorResolutionGenerator>>
  GetForTargetProcess(ExecutionSession &ES, DylibManager &DylibMgr,
                      SymbolPredicate Allow = SymbolPredicate(),
                      AbsoluteSymbolsFn AbsoluteSymbols = absoluteSymbols) {
    return Load(ES, DylibMgr, nullptr, std::move(Allow),
                std::move(AbsoluteSymbols));
  }
 
  Error tryToGenerate(LookupState &LS, LookupKind K, JITDylib &JD,
                      JITDylibLookupFlags JDLookupFlags,
                      const SymbolLookupSet &LookupSet) override;
 
private:
  ExecutionSession &ES;
  DylibManager &DylibMgr;
  tpctypes::ResolverHandle H;
  SymbolPredicate Allow;
  AbsoluteSymbolsFn AbsoluteSymbols;
};
 
} // namespace llvm::orc
 
#endif // LLVM_EXECUTIONENGINE_ORC_EXECUTORRESOLUTIONGENERATOR_H