d27d0a673c
Libomptarget grew out of a project that was originally not in LLVM. As we develop libomptarget this has led to an increasingly large clash between the naming conventions used. This patch fixes most of the variable names that did not confrom to the LLVM standard, that is `VariableName` for variables and `functionName` for functions. This patch was primarily done using my editor's linting messages, if there are any issues I missed arising from the automation let me know. Reviewed By: saiislam Differential Revision: https://reviews.llvm.org/D128997
89 lines
3.0 KiB
C++
89 lines
3.0 KiB
C++
//===-- elf_common.cpp - Common ELF functionality -------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Common ELF functionality for target plugins.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#include "elf_common.h"
|
|
#include "Debug.h"
|
|
|
|
#include "llvm/BinaryFormat/Magic.h"
|
|
#include "llvm/Object/Binary.h"
|
|
#include "llvm/Object/ELFObjectFile.h"
|
|
#include "llvm/Object/ELFTypes.h"
|
|
#include "llvm/Object/ObjectFile.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
|
|
#ifndef TARGET_NAME
|
|
#define TARGET_NAME ELF Common
|
|
#endif
|
|
#define DEBUG_PREFIX "TARGET " GETNAME(TARGET_NAME)
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::ELF;
|
|
using namespace llvm::object;
|
|
|
|
/// If the given range of bytes [\p BytesBegin, \p BytesEnd) represents
|
|
/// a valid ELF, then invoke \p Callback on the ELFObjectFileBase
|
|
/// created from this range, otherwise, return 0.
|
|
/// If \p Callback is invoked, then return whatever value \p Callback returns.
|
|
template <typename F>
|
|
static int32_t withBytesAsElf(char *BytesBegin, char *BytesEnd, F Callback) {
|
|
size_t Size = BytesEnd - BytesBegin;
|
|
StringRef StrBuf(BytesBegin, Size);
|
|
|
|
auto Magic = identify_magic(StrBuf);
|
|
if (Magic != file_magic::elf && Magic != file_magic::elf_relocatable &&
|
|
Magic != file_magic::elf_executable &&
|
|
Magic != file_magic::elf_shared_object && Magic != file_magic::elf_core) {
|
|
DP("Not an ELF image!\n");
|
|
return 0;
|
|
}
|
|
|
|
std::unique_ptr<MemoryBuffer> MemBuf =
|
|
MemoryBuffer::getMemBuffer(StrBuf, "", false);
|
|
Expected<std::unique_ptr<ObjectFile>> BinOrErr =
|
|
ObjectFile::createELFObjectFile(MemBuf->getMemBufferRef(),
|
|
/*InitContent=*/false);
|
|
if (!BinOrErr) {
|
|
DP("Unable to get ELF handle: %s!\n",
|
|
toString(BinOrErr.takeError()).c_str());
|
|
return 0;
|
|
}
|
|
|
|
auto *Object = dyn_cast<const ELFObjectFileBase>(BinOrErr->get());
|
|
|
|
if (!Object) {
|
|
DP("Unknown ELF format!\n");
|
|
return 0;
|
|
}
|
|
|
|
return Callback(Object);
|
|
}
|
|
|
|
// Check whether an image is valid for execution on target_id
|
|
int32_t elf_check_machine(__tgt_device_image *Image, uint16_t TargetId) {
|
|
auto CheckMachine = [TargetId](const ELFObjectFileBase *Object) {
|
|
return TargetId == Object->getEMachine();
|
|
};
|
|
return withBytesAsElf(reinterpret_cast<char *>(Image->ImageStart),
|
|
reinterpret_cast<char *>(Image->ImageEnd),
|
|
CheckMachine);
|
|
}
|
|
|
|
int32_t elf_is_dynamic(__tgt_device_image *Image) {
|
|
auto CheckDynType = [](const ELFObjectFileBase *Object) {
|
|
uint16_t Type = Object->getEType();
|
|
DP("ELF Type: %d\n", Type);
|
|
return Type == ET_DYN;
|
|
};
|
|
return withBytesAsElf(reinterpret_cast<char *>(Image->ImageStart),
|
|
reinterpret_cast<char *>(Image->ImageEnd),
|
|
CheckDynType);
|
|
}
|