llvm/lib/ExecutionEngine/Orc/EPCDebugObjectRegistrar.cpp

   1 //===----- EPCDebugObjectRegistrar.cpp - EPC-based debug registration -----===//
   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/EPCDebugObjectRegistrar.h"
  10
  11 #include "llvm/ExecutionEngine/Orc/Core.h"
  12
  13 namespace llvm {
  14 namespace orc {
  15
  16 Expected<std::unique_ptr<EPCDebugObjectRegistrar>> createJITLoaderGDBRegistrar(
  17     ExecutionSession &ES,
  18     std::optional<ExecutorAddr> RegistrationFunctionDylib) {
  19   auto &EPC = ES.getExecutorProcessControl();
  20
  21   if (!RegistrationFunctionDylib) {
  22     if (auto D = EPC.getDylibMgr().loadDylib(nullptr))
  23       RegistrationFunctionDylib = *D;
  24     else
  25       return D.takeError();
  26   }
  27
  28   SymbolStringPtr RegisterFn =
  29       EPC.getTargetTriple().isOSBinFormatMachO()
  30           ? EPC.intern("_llvm_orc_registerJITLoaderGDBWrapper")
  31           : EPC.intern("llvm_orc_registerJITLoaderGDBWrapper");
  32
  33   SymbolLookupSet RegistrationSymbols;
  34   RegistrationSymbols.add(RegisterFn);
  35
  36   auto Result = EPC.getDylibMgr().lookupSymbols(
  37       {{*RegistrationFunctionDylib, RegistrationSymbols}});
  38   if (!Result)
  39     return Result.takeError();
  40
  41   assert(Result->size() == 1 && "Unexpected number of dylibs in result");
  42   assert((*Result)[0].size() == 1 &&
  43          "Unexpected number of addresses in result");
  44
  45   ExecutorAddr RegisterAddr = (*Result)[0][0].getAddress();
  46   return std::make_unique<EPCDebugObjectRegistrar>(ES, RegisterAddr);
  47 }
  48
  49 Error EPCDebugObjectRegistrar::registerDebugObject(ExecutorAddrRange TargetMem,
  50                                                    bool AutoRegisterCode) {
  51   return ES.callSPSWrapper<void(shared::SPSExecutorAddrRange, bool)>(
  52       RegisterFn, TargetMem, AutoRegisterCode);
  53 }
  54
  55 } // namespace orc
  56 } // namespace llvm