InstanceSymbolGenerator.cpp

Go to the documentation of this file.

//===- InstanceSymbolGenerator.cpp - Executor instance symbols ------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
 
#include "llvm/ExecutionEngine/Orc/InstanceSymbolGenerator.h"
 
#include "llvm/ExecutionEngine/Orc/AbsoluteSymbols.h"
#include "llvm/ExecutionEngine/Orc/DebugUtils.h"
 
#define DEBUG_TYPE "orc"
 
namespace llvm::orc {
 
Error InstanceSymbolGenerator::tryToGenerate(LookupState &LS, LookupKind K,
                                             JITDylib &JD,
                                             JITDylibLookupFlags JDLookupFlags,
                                             const SymbolLookupSet &Symbols) {
 
  if (Symbols.empty())
    return Error::success();
 
  LLVM_DEBUG({
    dbgs() << "InstanceSymbolGenerator trying to generate " << Symbols << "\n";
  });
 
  // Look the symbols up as required, so that names that aren't instance
  // symbols come back as std::nullopt rather than as null addresses. This
  // doesn't fail the lookup: those names are just left undefined.
  SymbolLookupSet LookupSymbols;
  for (auto &[Name, Flags] : Symbols)
    LookupSymbols.add(Name, SymbolLookupFlags::RequiredSymbol);
 
  Lookup(
      [&JD, LS = std::move(LS),
       LookupSymbols](Expected<SymbolLookupResult> Result) mutable {
        if (!Result)
          return LS.continueLookup(Result.takeError());
 
        if (Result->size() != LookupSymbols.size())
          return LS.continueLookup(make_error<StringError>(
              "Instance symbol lookup returned " + Twine(Result->size()) +
                  " results for " + Twine(LookupSymbols.size()) + " symbols",
              inconvertibleErrorCode()));
 
        SymbolMap NewSymbols;
        auto ResultIt = Result->begin();
        for (auto &[Name, Flags] : LookupSymbols) {
          const auto &Addr = *ResultIt++;
          if (Addr)
            NewSymbols[Name] = {*Addr, JITSymbolFlags::Exported};
        }
 
        LLVM_DEBUG({
          dbgs() << "InstanceSymbolGenerator lookup returned " << NewSymbols
                 << "\n";
        });
 
        if (NewSymbols.empty())
          return LS.continueLookup(Error::success());
 
        LS.continueLookup(JD.define(absoluteSymbols(std::move(NewSymbols))));
      },
      JD.getExecutionSession(), LookupSymbols);
 
  return Error::success();
}
 
} // namespace llvm::orc