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/RawByteChannel.h"
18 #include "llvm/Support/Error.h"
19 #include <cassert>
20 #include <cerrno>
21 #include <system_error>
22 
23 #if !defined(_MSC_VER) && !defined(__MINGW32__)
24 #include <unistd.h>
25 #else
26 #include <io.h>
27 #endif
28 
29 /// RPC channel that reads from and writes from file descriptors.
30 class FDRPCChannel final : public llvm::orc::rpc::RawByteChannel {
31 public:
FDRPCChannel(int InFD,int OutFD)32   FDRPCChannel(int InFD, int OutFD) : InFD(InFD), OutFD(OutFD) {}
33 
readBytes(char * Dst,unsigned Size)34   llvm::Error readBytes(char *Dst, unsigned Size) override {
35     assert(Dst && "Attempt to read into null.");
36     ssize_t Completed = 0;
37     while (Completed < static_cast<ssize_t>(Size)) {
38       ssize_t Read = ::read(InFD, Dst + Completed, Size - Completed);
39       if (Read <= 0) {
40         auto ErrNo = errno;
41         if (ErrNo == EAGAIN || ErrNo == EINTR)
42           continue;
43         else
44           return llvm::errorCodeToError(
45                    std::error_code(errno, std::generic_category()));
46       }
47       Completed += Read;
48     }
49     return llvm::Error::success();
50   }
51 
appendBytes(const char * Src,unsigned Size)52   llvm::Error appendBytes(const char *Src, unsigned Size) override {
53     assert(Src && "Attempt to append from null.");
54     ssize_t Completed = 0;
55     while (Completed < static_cast<ssize_t>(Size)) {
56       ssize_t Written = ::write(OutFD, Src + Completed, Size - Completed);
57       if (Written < 0) {
58         auto ErrNo = errno;
59         if (ErrNo == EAGAIN || ErrNo == EINTR)
60           continue;
61         else
62           return llvm::errorCodeToError(
63                    std::error_code(errno, std::generic_category()));
64       }
65       Completed += Written;
66     }
67     return llvm::Error::success();
68   }
69 
send()70   llvm::Error send() override { return llvm::Error::success(); }
71 
72 private:
73   int InFD, OutFD;
74 };
75 
76 #endif // LLVM_TOOLS_LLI_REMOTEJITUTILS_H
77