2ccf7ed277
Use SymbolStringPtr for Symbol names in LinkGraph. This reduces string interning on the boundary between JITLink and ORC, and allows pointer comparisons (rather than string comparisons) between Symbol names. This should improve the performance and readability of code that bridges between JITLink and ORC (e.g. ObjectLinkingLayer and ObjectLinkingLayer::Plugins). To enable use of SymbolStringPtr a std::shared_ptr<SymbolStringPool> is added to LinkGraph and threaded through to its construction sites in LLVM and Bolt. All LinkGraphs that are to have symbol names compared by pointer equality must point to the same SymbolStringPool instance, which in ORC sessions should be the pool attached to the ExecutionSession. --------- Co-authored-by: Lang Hames <lhames@gmail.com>
91 lines
3.1 KiB
C++
91 lines
3.1 KiB
C++
//===-------------- MachO.cpp - JIT linker function for MachO -------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// MachO jit-link function.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ExecutionEngine/JITLink/MachO.h"
|
|
|
|
#include "llvm/BinaryFormat/MachO.h"
|
|
#include "llvm/ExecutionEngine/JITLink/MachO_arm64.h"
|
|
#include "llvm/ExecutionEngine/JITLink/MachO_x86_64.h"
|
|
#include "llvm/Support/Format.h"
|
|
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "jitlink"
|
|
|
|
namespace llvm {
|
|
namespace jitlink {
|
|
|
|
Expected<std::unique_ptr<LinkGraph>>
|
|
createLinkGraphFromMachOObject(MemoryBufferRef ObjectBuffer,
|
|
std::shared_ptr<orc::SymbolStringPool> SSP) {
|
|
StringRef Data = ObjectBuffer.getBuffer();
|
|
if (Data.size() < 4)
|
|
return make_error<JITLinkError>("Truncated MachO buffer \"" +
|
|
ObjectBuffer.getBufferIdentifier() + "\"");
|
|
|
|
uint32_t Magic;
|
|
memcpy(&Magic, Data.data(), sizeof(uint32_t));
|
|
LLVM_DEBUG({
|
|
dbgs() << "jitLink_MachO: magic = " << format("0x%08" PRIx32, Magic)
|
|
<< ", identifier = \"" << ObjectBuffer.getBufferIdentifier()
|
|
<< "\"\n";
|
|
});
|
|
|
|
if (Magic == MachO::MH_MAGIC || Magic == MachO::MH_CIGAM)
|
|
return make_error<JITLinkError>("MachO 32-bit platforms not supported");
|
|
else if (Magic == MachO::MH_MAGIC_64 || Magic == MachO::MH_CIGAM_64) {
|
|
|
|
if (Data.size() < sizeof(MachO::mach_header_64))
|
|
return make_error<JITLinkError>("Truncated MachO buffer \"" +
|
|
ObjectBuffer.getBufferIdentifier() +
|
|
"\"");
|
|
|
|
// Read the CPU type from the header.
|
|
uint32_t CPUType;
|
|
memcpy(&CPUType, Data.data() + 4, sizeof(uint32_t));
|
|
if (Magic == MachO::MH_CIGAM_64)
|
|
CPUType = llvm::byteswap<uint32_t>(CPUType);
|
|
|
|
LLVM_DEBUG({
|
|
dbgs() << "jitLink_MachO: cputype = " << format("0x%08" PRIx32, CPUType)
|
|
<< "\n";
|
|
});
|
|
|
|
switch (CPUType) {
|
|
case MachO::CPU_TYPE_ARM64:
|
|
return createLinkGraphFromMachOObject_arm64(ObjectBuffer, std::move(SSP));
|
|
case MachO::CPU_TYPE_X86_64:
|
|
return createLinkGraphFromMachOObject_x86_64(ObjectBuffer,
|
|
std::move(SSP));
|
|
}
|
|
return make_error<JITLinkError>("MachO-64 CPU type not valid");
|
|
} else
|
|
return make_error<JITLinkError>("Unrecognized MachO magic value");
|
|
}
|
|
|
|
void link_MachO(std::unique_ptr<LinkGraph> G,
|
|
std::unique_ptr<JITLinkContext> Ctx) {
|
|
|
|
switch (G->getTargetTriple().getArch()) {
|
|
case Triple::aarch64:
|
|
return link_MachO_arm64(std::move(G), std::move(Ctx));
|
|
case Triple::x86_64:
|
|
return link_MachO_x86_64(std::move(G), std::move(Ctx));
|
|
default:
|
|
Ctx->notifyFailed(make_error<JITLinkError>("MachO-64 CPU type not valid"));
|
|
return;
|
|
}
|
|
}
|
|
|
|
} // end namespace jitlink
|
|
} // end namespace llvm
|