1 //===------------------------ OrcRTBootstrap.cpp --------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "OrcRTBootstrap.h" 10 11 #include "llvm/ExecutionEngine/Orc/Shared/OrcRTBridge.h" 12 #include "llvm/ExecutionEngine/Orc/Shared/WrapperFunctionUtils.h" 13 #include "llvm/ExecutionEngine/Orc/TargetProcess/RegisterEHFrames.h" 14 #include "llvm/ExecutionEngine/Orc/TargetProcess/TargetExecutionUtils.h" 15 16 #define DEBUG_TYPE "orc" 17 18 using namespace llvm::orc::shared; 19 20 namespace llvm { 21 namespace orc { 22 namespace rt_bootstrap { 23 24 template <typename WriteT, typename SPSWriteT> 25 static llvm::orc::shared::CWrapperFunctionResult writeUIntsWrapper(const char * ArgData,size_t ArgSize)26 writeUIntsWrapper(const char *ArgData, size_t ArgSize) { 27 return WrapperFunction<void(SPSSequence<SPSWriteT>)>::handle( 28 ArgData, ArgSize, 29 [](std::vector<WriteT> Ws) { 30 for (auto &W : Ws) 31 *W.Addr.template toPtr<decltype(W.Value) *>() = W.Value; 32 }) 33 .release(); 34 } 35 36 static llvm::orc::shared::CWrapperFunctionResult writeBuffersWrapper(const char * ArgData,size_t ArgSize)37 writeBuffersWrapper(const char *ArgData, size_t ArgSize) { 38 return WrapperFunction<void(SPSSequence<SPSMemoryAccessBufferWrite>)>::handle( 39 ArgData, ArgSize, 40 [](std::vector<tpctypes::BufferWrite> Ws) { 41 for (auto &W : Ws) 42 memcpy(W.Addr.template toPtr<char *>(), W.Buffer.data(), 43 W.Buffer.size()); 44 }) 45 .release(); 46 } 47 48 static llvm::orc::shared::CWrapperFunctionResult runAsMainWrapper(const char * ArgData,size_t ArgSize)49 runAsMainWrapper(const char *ArgData, size_t ArgSize) { 50 return WrapperFunction<rt::SPSRunAsMainSignature>::handle( 51 ArgData, ArgSize, 52 [](ExecutorAddr MainAddr, 53 std::vector<std::string> Args) -> int64_t { 54 return runAsMain(MainAddr.toPtr<int (*)(int, char *[])>(), Args); 55 }) 56 .release(); 57 } 58 59 static llvm::orc::shared::CWrapperFunctionResult runAsVoidFunctionWrapper(const char * ArgData,size_t ArgSize)60 runAsVoidFunctionWrapper(const char *ArgData, size_t ArgSize) { 61 return WrapperFunction<rt::SPSRunAsVoidFunctionSignature>::handle( 62 ArgData, ArgSize, 63 [](ExecutorAddr MainAddr) -> int32_t { 64 return runAsVoidFunction(MainAddr.toPtr<int32_t (*)(void)>()); 65 }) 66 .release(); 67 } 68 69 static llvm::orc::shared::CWrapperFunctionResult runAsIntFunctionWrapper(const char * ArgData,size_t ArgSize)70 runAsIntFunctionWrapper(const char *ArgData, size_t ArgSize) { 71 return WrapperFunction<rt::SPSRunAsIntFunctionSignature>::handle( 72 ArgData, ArgSize, 73 [](ExecutorAddr MainAddr, int32_t Arg) -> int32_t { 74 return runAsIntFunction(MainAddr.toPtr<int32_t (*)(int32_t)>(), 75 Arg); 76 }) 77 .release(); 78 } 79 addTo(StringMap<ExecutorAddr> & M)80 void addTo(StringMap<ExecutorAddr> &M) { 81 M[rt::MemoryWriteUInt8sWrapperName] = ExecutorAddr::fromPtr( 82 &writeUIntsWrapper<tpctypes::UInt8Write, 83 shared::SPSMemoryAccessUInt8Write>); 84 M[rt::MemoryWriteUInt16sWrapperName] = ExecutorAddr::fromPtr( 85 &writeUIntsWrapper<tpctypes::UInt16Write, 86 shared::SPSMemoryAccessUInt16Write>); 87 M[rt::MemoryWriteUInt32sWrapperName] = ExecutorAddr::fromPtr( 88 &writeUIntsWrapper<tpctypes::UInt32Write, 89 shared::SPSMemoryAccessUInt32Write>); 90 M[rt::MemoryWriteUInt64sWrapperName] = ExecutorAddr::fromPtr( 91 &writeUIntsWrapper<tpctypes::UInt64Write, 92 shared::SPSMemoryAccessUInt64Write>); 93 M[rt::MemoryWriteBuffersWrapperName] = 94 ExecutorAddr::fromPtr(&writeBuffersWrapper); 95 M[rt::RegisterEHFrameSectionWrapperName] = 96 ExecutorAddr::fromPtr(&llvm_orc_registerEHFrameSectionWrapper); 97 M[rt::DeregisterEHFrameSectionWrapperName] = 98 ExecutorAddr::fromPtr(&llvm_orc_deregisterEHFrameSectionWrapper); 99 M[rt::RunAsMainWrapperName] = ExecutorAddr::fromPtr(&runAsMainWrapper); 100 M[rt::RunAsVoidFunctionWrapperName] = 101 ExecutorAddr::fromPtr(&runAsVoidFunctionWrapper); 102 M[rt::RunAsIntFunctionWrapperName] = 103 ExecutorAddr::fromPtr(&runAsIntFunctionWrapper); 104 } 105 106 } // end namespace rt_bootstrap 107 } // end namespace orc 108 } // end namespace llvm 109