1 files changed, 119 insertions, 0 deletions
diff --git a/llvm/examples/Kaleidoscope/BuildingAJIT/Chapter5/Server/server.cpp b/llvm/examples/Kaleidoscope/BuildingAJIT/Chapter5/Server/server.cpp
new file mode 100644
index 00000000000..c53e22fe83a
--- /dev/null
+++ b/llvm/examples/Kaleidoscope/BuildingAJIT/Chapter5/Server/server.cpp
@@ -0,0 +1,119 @@
+#include "llvm/Support/CommandLine.h"
+#include "llvm/Support/DynamicLibrary.h"
+#include "llvm/Support/TargetSelect.h"
+#include "llvm/ExecutionEngine/Orc/OrcRemoteTargetServer.h"
+#include "llvm/ExecutionEngine/Orc/OrcABISupport.h"
+
+#include "../RemoteJITUtils.h"
+
+#include <cstring>
+#include <unistd.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+
+
+using namespace llvm;
+using namespace llvm::orc;
+
+// Command line argument for TCP port.
+cl::opt<uint32_t> Port("port",
+                       cl::desc("TCP port to listen on"),
+                       cl::init(20000));
+
+ExitOnError ExitOnErr;
+
+typedef int (*MainFun)(int, const char*[]);
+
+template <typename NativePtrT>
+NativePtrT MakeNative(uint64_t P) {
+  return reinterpret_cast<NativePtrT>(static_cast<uintptr_t>(P));
+}
+
+extern "C"
+void printExprResult(double Val) {
+  printf("Expression evaluated to: %f\n", Val);
+}
+
+// --- LAZY COMPILE TEST ---
+int main(int argc, char* argv[]) {
+
+  if (argc == 0)
+    ExitOnErr.setBanner("jit_server: ");
+  else
+    ExitOnErr.setBanner(std::string(argv[0]) + ": ");
+
+  // --- Initialize LLVM ---
+  cl::ParseCommandLineOptions(argc, argv, "LLVM lazy JIT example.\n");
+
+  InitializeNativeTarget();
+  InitializeNativeTargetAsmPrinter();
+  InitializeNativeTargetAsmParser();
+
+  if (sys::DynamicLibrary::LoadLibraryPermanently(nullptr)) {
+    errs() << "Error loading program symbols.\n";
+    return 1;
+  }
+
+  // --- Initialize remote connection ---
+
+  int sockfd = socket(PF_INET, SOCK_STREAM, 0);
+  sockaddr_in servAddr, clientAddr;
+  socklen_t clientAddrLen = sizeof(clientAddr);
+  bzero(&servAddr, sizeof(servAddr));
+  servAddr.sin_family = PF_INET;
+  servAddr.sin_family = INADDR_ANY;
+  servAddr.sin_port = htons(Port);
+
+  {
+    // avoid "Address already in use" error.
+    int yes=1;
+    if (setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR,&yes,sizeof(int)) == -1) {
+      errs() << "Error calling setsockopt.\n";
+      return 1;
+    }
+  }
+
+  if (bind(sockfd, reinterpret_cast<sockaddr*>(&servAddr),
+           sizeof(servAddr)) < 0) {
+    errs() << "Error on binding.\n";
+    return 1;
+  }
+  listen(sockfd, 1);
+  int newsockfd = accept(sockfd, reinterpret_cast<sockaddr*>(&clientAddr),
+                         &clientAddrLen);
+
+  auto SymbolLookup =
+    [](const std::string &Name) {
+      return RTDyldMemoryManager::getSymbolAddressInProcess(Name);
+    };
+
+  auto RegisterEHFrames =
+    [](uint8_t *Addr, uint32_t Size) {
+      RTDyldMemoryManager::registerEHFramesInProcess(Addr, Size);
+    };
+
+  auto DeregisterEHFrames =
+    [](uint8_t *Addr, uint32_t Size) {
+      RTDyldMemoryManager::deregisterEHFramesInProcess(Addr, Size);
+    };
+
+  FDRPCChannel TCPChannel(newsockfd, newsockfd);
+  typedef remote::OrcRemoteTargetServer<FDRPCChannel, OrcX86_64_SysV> MyServerT;
+
+  MyServerT Server(TCPChannel, SymbolLookup, RegisterEHFrames, DeregisterEHFrames);
+
+  while (1) {
+    MyServerT::JITFuncId Id = MyServerT::InvalidId;
+    ExitOnErr(Server.startReceivingFunction(TCPChannel, (uint32_t&)Id));
+    switch (Id) {
+    case MyServerT::TerminateSessionId:
+      ExitOnErr(Server.handleTerminateSession());
+      return 0;
+    default:
+      ExitOnErr(Server.handleKnownFunction(Id));
+      break;
+    }
+  }
+
+  llvm_unreachable("Fell through server command loop.");
+}