1 //===-- RemoteJITUtils.h - Utilities for remote-JITing with LLI -*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // Utilities for remote-JITing with LLI.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_TOOLS_LLI_REMOTEJITUTILS_H
15 #define LLVM_TOOLS_LLI_REMOTEJITUTILS_H
16 
17 #include "llvm/ExecutionEngine/Orc/RPCChannel.h"
18 #include "llvm/ExecutionEngine/RTDyldMemoryManager.h"
19 #include <mutex>
20 
21 #if !defined(_MSC_VER) && !defined(__MINGW32__)
22 #include <unistd.h>
23 #else
24 #include <io.h>
25 #endif
26 
27 /// RPC channel that reads from and writes from file descriptors.
28 class FDRPCChannel final : public llvm::orc::remote::RPCChannel {
29 public:
FDRPCChannel(int InFD,int OutFD)30   FDRPCChannel(int InFD, int OutFD) : InFD(InFD), OutFD(OutFD) {}
31 
readBytes(char * Dst,unsigned Size)32   llvm::Error readBytes(char *Dst, unsigned Size) override {
33     assert(Dst && "Attempt to read into null.");
34     ssize_t Completed = 0;
35     while (Completed < static_cast<ssize_t>(Size)) {
36       ssize_t Read = ::read(InFD, Dst + Completed, Size - Completed);
37       if (Read <= 0) {
38         auto ErrNo = errno;
39         if (ErrNo == EAGAIN || ErrNo == EINTR)
40           continue;
41         else
42           return llvm::errorCodeToError(
43                    std::error_code(errno, std::generic_category()));
44       }
45       Completed += Read;
46     }
47     return llvm::Error::success();
48   }
49 
appendBytes(const char * Src,unsigned Size)50   llvm::Error appendBytes(const char *Src, unsigned Size) override {
51     assert(Src && "Attempt to append from null.");
52     ssize_t Completed = 0;
53     while (Completed < static_cast<ssize_t>(Size)) {
54       ssize_t Written = ::write(OutFD, Src + Completed, Size - Completed);
55       if (Written < 0) {
56         auto ErrNo = errno;
57         if (ErrNo == EAGAIN || ErrNo == EINTR)
58           continue;
59         else
60           return llvm::errorCodeToError(
61                    std::error_code(errno, std::generic_category()));
62       }
63       Completed += Written;
64     }
65     return llvm::Error::success();
66   }
67 
send()68   llvm::Error send() override { return llvm::Error::success(); }
69 
70 private:
71   int InFD, OutFD;
72 };
73 
74 #endif
75