// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2025 Chen Miao
 *
 * Based on arch/arm/kernel/jump_label.c
 */
#include <linux/jump_label.h>
#include <linux/kernel.h>
#include <linux/memory.h>
#include <asm/bug.h>
#include <asm/cacheflush.h>
#include <asm/text-patching.h>

bool arch_jump_label_transform_queue(struct jump_entry *entry,
				     enum jump_label_type type)
{
	void *addr = (void *)jump_entry_code(entry);
	u32 insn;

	if (type == JUMP_LABEL_JMP) {
		long offset;

		offset = jump_entry_target(entry) - jump_entry_code(entry);
		/*
		 * The actual maximum range of the l.j instruction's offset is -134,217,728
		 * ~ 134,217,724 (sign 26-bit imm).
		 * For the original jump range, we need to right-shift N by 2 to obtain the
		 * instruction's offset.
		 */
		WARN_ON_ONCE(offset < -134217728 || offset > 134217724);

		/* 26bit imm mask */
		offset = (offset >> 2) & 0x03ffffff;

		insn = offset;
	} else {
		insn = OPENRISC_INSN_NOP;
	}

	if (early_boot_irqs_disabled)
		copy_to_kernel_nofault(addr, &insn, sizeof(insn));
	else
		patch_insn_write(addr, insn);

	return true;
}

void arch_jump_label_transform_apply(void)
{
	kick_all_cpus_sync();
}