ELF.cpp 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. //===-------------- ELF.cpp - JIT linker function for ELF -------------===//
  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. // ELF jit-link function.
  10. //
  11. //===----------------------------------------------------------------------===//
  12. #include "llvm/ExecutionEngine/JITLink/ELF.h"
  13. #include "llvm/BinaryFormat/ELF.h"
  14. #include "llvm/ExecutionEngine/JITLink/ELF_aarch64.h"
  15. #include "llvm/ExecutionEngine/JITLink/ELF_riscv.h"
  16. #include "llvm/ExecutionEngine/JITLink/ELF_x86_64.h"
  17. #include "llvm/Object/ELF.h"
  18. #include "llvm/Support/Endian.h"
  19. #include "llvm/Support/Format.h"
  20. #include "llvm/Support/MemoryBuffer.h"
  21. #include <cstring>
  22. using namespace llvm;
  23. #define DEBUG_TYPE "jitlink"
  24. namespace llvm {
  25. namespace jitlink {
  26. Expected<uint16_t> readTargetMachineArch(StringRef Buffer) {
  27. const char *Data = Buffer.data();
  28. if (Data[ELF::EI_DATA] == ELF::ELFDATA2LSB) {
  29. if (Data[ELF::EI_CLASS] == ELF::ELFCLASS64) {
  30. if (auto File = llvm::object::ELF64LEFile::create(Buffer)) {
  31. return File->getHeader().e_machine;
  32. } else {
  33. return File.takeError();
  34. }
  35. } else if (Data[ELF::EI_CLASS] == ELF::ELFCLASS32) {
  36. if (auto File = llvm::object::ELF32LEFile::create(Buffer)) {
  37. return File->getHeader().e_machine;
  38. } else {
  39. return File.takeError();
  40. }
  41. }
  42. }
  43. return ELF::EM_NONE;
  44. }
  45. Expected<std::unique_ptr<LinkGraph>>
  46. createLinkGraphFromELFObject(MemoryBufferRef ObjectBuffer) {
  47. StringRef Buffer = ObjectBuffer.getBuffer();
  48. if (Buffer.size() < ELF::EI_MAG3 + 1)
  49. return make_error<JITLinkError>("Truncated ELF buffer");
  50. if (memcmp(Buffer.data(), ELF::ElfMagic, strlen(ELF::ElfMagic)) != 0)
  51. return make_error<JITLinkError>("ELF magic not valid");
  52. Expected<uint16_t> TargetMachineArch = readTargetMachineArch(Buffer);
  53. if (!TargetMachineArch)
  54. return TargetMachineArch.takeError();
  55. switch (*TargetMachineArch) {
  56. case ELF::EM_AARCH64:
  57. return createLinkGraphFromELFObject_aarch64(ObjectBuffer);
  58. case ELF::EM_RISCV:
  59. return createLinkGraphFromELFObject_riscv(ObjectBuffer);
  60. case ELF::EM_X86_64:
  61. return createLinkGraphFromELFObject_x86_64(ObjectBuffer);
  62. default:
  63. return make_error<JITLinkError>(
  64. "Unsupported target machine architecture in ELF object " +
  65. ObjectBuffer.getBufferIdentifier());
  66. }
  67. }
  68. void link_ELF(std::unique_ptr<LinkGraph> G,
  69. std::unique_ptr<JITLinkContext> Ctx) {
  70. switch (G->getTargetTriple().getArch()) {
  71. case Triple::aarch64:
  72. link_ELF_aarch64(std::move(G), std::move(Ctx));
  73. return;
  74. case Triple::riscv32:
  75. case Triple::riscv64:
  76. link_ELF_riscv(std::move(G), std::move(Ctx));
  77. return;
  78. case Triple::x86_64:
  79. link_ELF_x86_64(std::move(G), std::move(Ctx));
  80. return;
  81. default:
  82. Ctx->notifyFailed(make_error<JITLinkError>(
  83. "Unsupported target machine architecture in ELF link graph " +
  84. G->getName()));
  85. return;
  86. }
  87. }
  88. } // end namespace jitlink
  89. } // end namespace llvm