/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _ASM_S390_KMSAN_H
#define _ASM_S390_KMSAN_H

#include <asm/lowcore.h>
#include <asm/page.h>
#include <linux/kmsan.h>
#include <linux/mmzone.h>
#include <linux/stddef.h>

#ifndef MODULE

static inline bool is_lowcore_addr(void *addr)
{
	return addr >= (void *)get_lowcore() &&
	       addr < (void *)(get_lowcore() + 1);
}

static inline void *arch_kmsan_get_meta_or_null(void *addr, bool is_origin)
{
	if (is_lowcore_addr(addr)) {
		/*
		 * Different lowcores accessed via S390_lowcore are described
		 * by the same struct page. Resolve the prefix manually in
		 * order to get a distinct struct page.
		 */
		addr += (void *)lowcore_ptr[raw_smp_processor_id()] -
			(void *)get_lowcore();
		if (KMSAN_WARN_ON(is_lowcore_addr(addr)))
			return NULL;
		return kmsan_get_metadata(addr, is_origin);
	}
	return NULL;
}

static inline bool kmsan_virt_addr_valid(void *addr)
{
	bool ret;

	/*
	 * pfn_valid() relies on RCU, and may call into the scheduler on exiting
	 * the critical section. However, this would result in recursion with
	 * KMSAN. Therefore, disable preemption here, and re-enable preemption
	 * below while suppressing reschedules to avoid recursion.
	 *
	 * Note, this sacrifices occasionally breaking scheduling guarantees.
	 * Although, a kernel compiled with KMSAN has already given up on any
	 * performance guarantees due to being heavily instrumented.
	 */
	preempt_disable();
	ret = virt_addr_valid(addr);
	preempt_enable_no_resched();

	return ret;
}

#endif /* !MODULE */

#endif /* _ASM_S390_KMSAN_H */