TheRock/compiler/amd-llvm/libc/startup/linux/riscv/irelative.cpp

//===-- Implementation of apply_irelative_relocs (RISC-V) -----------------===//
//
// 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
//
//===----------------------------------------------------------------------===//

#include "startup/linux/irelative.h"
#include "hdr/elf_macros.h"
#include "hdr/elf_proxy.h"
#include "hdr/link_macros.h"
#include "src/__support/macros/config.h"

namespace LIBC_NAMESPACE_DECL {

// RISC-V may be 32-bit or 64-bit. ElfW(Rela) handles the struct type,
// but we need the correct R_TYPE extraction macro.
static constexpr unsigned get_r_type(uintptr_t info) {
#ifdef __LP64__
  return ELF64_R_TYPE(info);
#else
  return ELF32_R_TYPE(info);
#endif
}

void apply_irelative_relocs(intptr_t base, unsigned long /*hwcap*/,
                            unsigned long /*hwcap2*/) {
  for (const ElfW(Rela) *rela = __rela_iplt_start; rela != __rela_iplt_end;
       ++rela) {
    if (get_r_type(rela->r_info) != R_RISCV_IRELATIVE)
      continue;

    // RISC-V resolvers take no arguments (same as x86_64).
    // Use unsigned arithmetic to avoid undefined behavior on signed overflow,
    // which can occur with very large binaries or high load addresses.
    uintptr_t resolver_addr =
        static_cast<uintptr_t>(base) + static_cast<uintptr_t>(rela->r_addend);
    auto resolver = reinterpret_cast<uintptr_t (*)(void)>(resolver_addr);
    uintptr_t result = resolver();

    uintptr_t target_addr = static_cast<uintptr_t>(base) + rela->r_offset;
    *reinterpret_cast<uintptr_t *>(target_addr) = result;
  }
}

} // namespace LIBC_NAMESPACE_DECL