// SPDX-License-Identifier: GPL-2.0
#include <linux/static_call.h>
#include <linux/memory.h>
#include <asm/text-patching.h>

void arch_static_call_transform(void *site, void *tramp, void *func, bool tail)
{
        u64 literal;
        int ret;

        if (!func)
                func = __static_call_return0;

        /* decode the instructions to discover the literal address */
        literal = ALIGN_DOWN((u64)tramp + 4, SZ_4K) +
                  aarch64_insn_adrp_get_offset(le32_to_cpup(tramp + 4)) +
                  8 * aarch64_insn_decode_immediate(AARCH64_INSN_IMM_12,
                                                    le32_to_cpup(tramp + 8));

        ret = aarch64_insn_write_literal_u64((void *)literal, (u64)func);
        WARN_ON_ONCE(ret);
}
EXPORT_SYMBOL_GPL(arch_static_call_transform);