tools/lli/ChildTarget/ChildTarget.cpp

   1 #include "llvm/ExecutionEngine/Orc/OrcArchitectureSupport.h"
   2 #include "llvm/ExecutionEngine/Orc/OrcRemoteTargetServer.h"
   3 #include "llvm/Support/Debug.h"
   4 #include "llvm/Support/DynamicLibrary.h"
   5 #include "llvm/Support/Process.h"
   6 #include <sstream>
   7
   8 #include "../RemoteJITUtils.h"
   9
  10 using namespace llvm;
  11 using namespace llvm::orc;
  12 using namespace llvm::sys;
  13
  14 #ifdef __x86_64__
  15 typedef OrcX86_64 HostOrcArch;
  16 #else
  17 typedef OrcGenericArchitecture HostOrcArch;
  18 #endif
  19
  20 int main(int argc, char *argv[]) {
  21
  22   if (argc != 3) {
  23     errs() << "Usage: " << argv[0] << " <input fd> <output fd>\n";
  24     return 1;
  25   }
  26
  27   int InFD;
  28   int OutFD;
  29   {
  30     std::istringstream InFDStream(argv[1]), OutFDStream(argv[2]);
  31     InFDStream >> InFD;
  32     OutFDStream >> OutFD;
  33   }
  34
  35   if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr)) {
  36     errs() << "Error loading program symbols.\n";
  37     return 1;
  38   }
  39
  40   auto SymbolLookup = [](const std::string &Name) {
  41     return RTDyldMemoryManager::getSymbolAddressInProcess(Name);
  42   };
  43
  44   FDRPCChannel Channel(InFD, OutFD);
  45   typedef remote::OrcRemoteTargetServer<FDRPCChannel, HostOrcArch> JITServer;
  46   JITServer Server(Channel, SymbolLookup);
  47
  48   while (1) {
  49     JITServer::JITProcId Id = JITServer::InvalidId;
  50     if (auto EC = Server.getNextProcId(Id)) {
  51       errs() << "Error: " << EC.message() << "\n";
  52       return 1;
  53     }
  54     switch (Id) {
  55     case JITServer::TerminateSessionId:
  56       return 0;
  57     default:
  58       if (auto EC = Server.handleKnownProcedure(Id)) {
  59         errs() << "Error: " << EC.message() << "\n";
  60         return 1;
  61       }
  62     }
  63   }
  64
  65   close(InFD);
  66   close(OutFD);
  67
  68   return 0;
  69 }