// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "dynlink.h"

__NO_SAFESTACK NO_ASAN static inline void apply_relr(ElfW(Addr) base, const ElfW(Addr) * relr,
                                                     size_t relrsz) {
  const size_t nrelr = relrsz / sizeof(relr[0]);
  size_t i = 0;
  while (i < nrelr) {
    ElfW(Addr)* addr = (void*)(base + relr[i++]);
    *addr++ += base;
    while (i < nrelr && (relr[i] & 1)) {
      // Each bitmask word covers the next 63 possible relocations.
      // Each bit is one if run[bit] should be relocated.

      ElfW(Addr)* run = addr;
      addr += (sizeof(relr[0]) * CHAR_BIT) - 1;

      int skip;
      for (ElfW(Addr) bitmask = relr[i++] >> 1; bitmask != 0; bitmask >>= skip) {
        skip = __builtin_ctzl(bitmask) + 1;
        run += skip;
        run[-1] += base;
      }
    }
  }
}