6024e5c395
clang 20.0.0 (based on r547379) from build 12806354. Bug: http://b/379133546 Test: N/A Change-Id: I2eb8938af55d809de674be63cb30cf27e801862b Upstream-Commit: ad834e67b1105d15ef907f6255d4c96e8e733f57
129 lines
4.5 KiB
C++
129 lines
4.5 KiB
C++
//===---- EPCGenericRTDyldMemoryManager.h - EPC-based MemMgr ----*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Defines a RuntimeDyld::MemoryManager that uses EPC and the ORC runtime
|
|
// bootstrap functions.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_EXECUTIONENGINE_ORC_EPCGENERICRTDYLDMEMORYMANAGER_H
|
|
#define LLVM_EXECUTIONENGINE_ORC_EPCGENERICRTDYLDMEMORYMANAGER_H
|
|
|
|
#include "llvm/ExecutionEngine/Orc/ExecutorProcessControl.h"
|
|
#include "llvm/ExecutionEngine/RuntimeDyld.h"
|
|
|
|
#define DEBUG_TYPE "orc"
|
|
|
|
namespace llvm {
|
|
namespace orc {
|
|
|
|
/// Remote-mapped RuntimeDyld-compatible memory manager.
|
|
class EPCGenericRTDyldMemoryManager : public RuntimeDyld::MemoryManager {
|
|
public:
|
|
/// Symbol addresses for memory access.
|
|
struct SymbolAddrs {
|
|
ExecutorAddr Instance;
|
|
ExecutorAddr Reserve;
|
|
ExecutorAddr Finalize;
|
|
ExecutorAddr Deallocate;
|
|
ExecutorAddr RegisterEHFrame;
|
|
ExecutorAddr DeregisterEHFrame;
|
|
};
|
|
|
|
/// Create an EPCGenericRTDyldMemoryManager using the given EPC, looking up
|
|
/// the default symbol names in the bootstrap symbol set.
|
|
static Expected<std::unique_ptr<EPCGenericRTDyldMemoryManager>>
|
|
CreateWithDefaultBootstrapSymbols(ExecutorProcessControl &EPC);
|
|
|
|
/// Create an EPCGenericRTDyldMemoryManager using the given EPC and symbol
|
|
/// addrs.
|
|
EPCGenericRTDyldMemoryManager(ExecutorProcessControl &EPC, SymbolAddrs SAs);
|
|
|
|
EPCGenericRTDyldMemoryManager(const EPCGenericRTDyldMemoryManager &) = delete;
|
|
EPCGenericRTDyldMemoryManager &
|
|
operator=(const EPCGenericRTDyldMemoryManager &) = delete;
|
|
EPCGenericRTDyldMemoryManager(EPCGenericRTDyldMemoryManager &&) = delete;
|
|
EPCGenericRTDyldMemoryManager &
|
|
operator=(EPCGenericRTDyldMemoryManager &&) = delete;
|
|
~EPCGenericRTDyldMemoryManager();
|
|
|
|
uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment,
|
|
unsigned SectionID,
|
|
StringRef SectionName) override;
|
|
|
|
uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment,
|
|
unsigned SectionID, StringRef SectionName,
|
|
bool IsReadOnly) override;
|
|
|
|
void reserveAllocationSpace(uintptr_t CodeSize, Align CodeAlign,
|
|
uintptr_t RODataSize, Align RODataAlign,
|
|
uintptr_t RWDataSize, Align RWDataAlign) override;
|
|
|
|
bool needsToReserveAllocationSpace() override;
|
|
|
|
void registerEHFrames(uint8_t *Addr, uint64_t LoadAddr, size_t Size) override;
|
|
|
|
void deregisterEHFrames() override;
|
|
|
|
void notifyObjectLoaded(RuntimeDyld &Dyld,
|
|
const object::ObjectFile &Obj) override;
|
|
|
|
bool finalizeMemory(std::string *ErrMsg = nullptr) override;
|
|
|
|
private:
|
|
struct SectionAlloc {
|
|
public:
|
|
SectionAlloc(uint64_t Size, unsigned Align)
|
|
: Size(Size), Align(Align),
|
|
Contents(std::make_unique<uint8_t[]>(Size + Align - 1)) {}
|
|
|
|
uint64_t Size;
|
|
unsigned Align;
|
|
std::unique_ptr<uint8_t[]> Contents;
|
|
ExecutorAddr RemoteAddr;
|
|
};
|
|
|
|
// Group of section allocations to be allocated together in the executor. The
|
|
// RemoteCodeAddr will stand in as the id of the group for deallocation
|
|
// purposes.
|
|
struct SectionAllocGroup {
|
|
SectionAllocGroup() = default;
|
|
SectionAllocGroup(const SectionAllocGroup &) = delete;
|
|
SectionAllocGroup &operator=(const SectionAllocGroup &) = delete;
|
|
SectionAllocGroup(SectionAllocGroup &&) = default;
|
|
SectionAllocGroup &operator=(SectionAllocGroup &&) = default;
|
|
|
|
ExecutorAddrRange RemoteCode;
|
|
ExecutorAddrRange RemoteROData;
|
|
ExecutorAddrRange RemoteRWData;
|
|
std::vector<ExecutorAddrRange> UnfinalizedEHFrames;
|
|
std::vector<SectionAlloc> CodeAllocs, RODataAllocs, RWDataAllocs;
|
|
};
|
|
|
|
// Maps all allocations in SectionAllocs to aligned blocks
|
|
void mapAllocsToRemoteAddrs(RuntimeDyld &Dyld,
|
|
std::vector<SectionAlloc> &SecAllocs,
|
|
ExecutorAddr NextAddr);
|
|
|
|
ExecutorProcessControl &EPC;
|
|
SymbolAddrs SAs;
|
|
|
|
std::mutex M;
|
|
std::vector<SectionAllocGroup> Unmapped;
|
|
std::vector<SectionAllocGroup> Unfinalized;
|
|
std::vector<ExecutorAddr> FinalizedAllocs;
|
|
std::string ErrMsg;
|
|
};
|
|
|
|
} // end namespace orc
|
|
} // end namespace llvm
|
|
|
|
#undef DEBUG_TYPE
|
|
|
|
#endif // LLVM_EXECUTIONENGINE_ORC_EPCGENERICRTDYLDMEMORYMANAGER_H
|