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