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 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 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 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 void addTo(StringMap<ExecutorAddr> &M) { 60 M[rt::MemoryWriteUInt8sWrapperName] = ExecutorAddr::fromPtr( 61 &writeUIntsWrapper<tpctypes::UInt8Write, 62 shared::SPSMemoryAccessUInt8Write>); 63 M[rt::MemoryWriteUInt16sWrapperName] = ExecutorAddr::fromPtr( 64 &writeUIntsWrapper<tpctypes::UInt16Write, 65 shared::SPSMemoryAccessUInt16Write>); 66 M[rt::MemoryWriteUInt32sWrapperName] = ExecutorAddr::fromPtr( 67 &writeUIntsWrapper<tpctypes::UInt32Write, 68 shared::SPSMemoryAccessUInt32Write>); 69 M[rt::MemoryWriteUInt64sWrapperName] = ExecutorAddr::fromPtr( 70 &writeUIntsWrapper<tpctypes::UInt64Write, 71 shared::SPSMemoryAccessUInt64Write>); 72 M[rt::MemoryWriteBuffersWrapperName] = 73 ExecutorAddr::fromPtr(&writeBuffersWrapper); 74 M[rt::RegisterEHFrameSectionWrapperName] = 75 ExecutorAddr::fromPtr(&llvm_orc_registerEHFrameSectionWrapper); 76 M[rt::DeregisterEHFrameSectionWrapperName] = 77 ExecutorAddr::fromPtr(&llvm_orc_deregisterEHFrameSectionWrapper); 78 M[rt::RunAsMainWrapperName] = ExecutorAddr::fromPtr(&runAsMainWrapper); 79 } 80 81 } // end namespace rt_bootstrap 82 } // end namespace orc 83 } // end namespace llvm 84