Implement GDB JIT interface

Author: weliveindetail

JitFromScratch.cpp

@@ -16,8 +16,10 @@ using namespace llvm::orc;
 JitFromScratch::JitFromScratch(ExitOnError ExitOnErr)
     : ES(std::make_unique<ExecutionSession>()),
       TM(createTargetMachine(ExitOnErr)),
+      GDBListener(JITEventListener::createGDBRegistrationListener()),
       ObjLinkingLayer(*ES, createMemoryManagerFtor()),
       CompileLayer(*ES, ObjLinkingLayer, SimpleCompiler(*TM)) {
+  ObjLinkingLayer.setNotifyLoaded(createNotifyLoadedFtor());
   if (auto R = createHostProcessResolver())
     ES->getMainJITDylib().setGenerator(std::move(R));
 }
@@ -57,6 +59,13 @@ GetMemoryManagerFunction JitFromScratch::createMemoryManagerFtor() {
   };
 }
 
+RTDyldObjectLinkingLayer::NotifyLoadedFunction
+JitFromScratch::createNotifyLoadedFtor() {
+  using namespace std::placeholders;
+  return std::bind(&JITEventListener::notifyObjectLoaded,
+                   GDBListener, _1, _2, _3);
+}
+
 std::string JitFromScratch::mangle(StringRef UnmangledName) {
   std::string MangledName;
   {

JitFromScratch.h

@@ -1,6 +1,7 @@
 #pragma once
 
 #include <llvm/ADT/StringRef.h>
+#include <llvm/ExecutionEngine/JITEventListener.h>
 #include <llvm/ExecutionEngine/JITSymbol.h>
 #include <llvm/ExecutionEngine/Orc/Core.h>
 #include <llvm/ExecutionEngine/Orc/IRCompileLayer.h>
@@ -49,6 +50,7 @@ class JitFromScratch {
 private:
   std::unique_ptr<llvm::orc::ExecutionSession> ES;
   std::unique_ptr<llvm::TargetMachine> TM;
+  llvm::JITEventListener *GDBListener;
 
   llvm::orc::RTDyldObjectLinkingLayer ObjLinkingLayer;
   llvm::orc::IRCompileLayer CompileLayer;
@@ -61,6 +63,9 @@ class JitFromScratch {
   llvm::orc::RTDyldObjectLinkingLayer::GetMemoryManagerFunction
   createMemoryManagerFtor();
 
+  llvm::orc::RTDyldObjectLinkingLayer::NotifyLoadedFunction
+  createNotifyLoadedFtor();
+
   std::string mangle(llvm::StringRef UnmangledName);
   llvm::Error applyDataLayout(llvm::Module &M);