162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2020 - Google LLC
462306a36Sopenharmony_ci * Author: David Brazdil <dbrazdil@google.com>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#include <asm/kvm_asm.h>
862306a36Sopenharmony_ci#include <asm/kvm_hyp.h>
962306a36Sopenharmony_ci#include <asm/kvm_mmu.h>
1062306a36Sopenharmony_ci
1162306a36Sopenharmony_ci/*
1262306a36Sopenharmony_ci * nVHE copy of data structures tracking available CPU cores.
1362306a36Sopenharmony_ci * Only entries for CPUs that were online at KVM init are populated.
1462306a36Sopenharmony_ci * Other CPUs should not be allowed to boot because their features were
1562306a36Sopenharmony_ci * not checked against the finalized system capabilities.
1662306a36Sopenharmony_ci */
1762306a36Sopenharmony_ciu64 __ro_after_init hyp_cpu_logical_map[NR_CPUS] = { [0 ... NR_CPUS-1] = INVALID_HWID };
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ciu64 cpu_logical_map(unsigned int cpu)
2062306a36Sopenharmony_ci{
2162306a36Sopenharmony_ci	BUG_ON(cpu >= ARRAY_SIZE(hyp_cpu_logical_map));
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci	return hyp_cpu_logical_map[cpu];
2462306a36Sopenharmony_ci}
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ciunsigned long __ro_after_init kvm_arm_hyp_percpu_base[NR_CPUS];
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ciunsigned long __hyp_per_cpu_offset(unsigned int cpu)
2962306a36Sopenharmony_ci{
3062306a36Sopenharmony_ci	unsigned long *cpu_base_array;
3162306a36Sopenharmony_ci	unsigned long this_cpu_base;
3262306a36Sopenharmony_ci	unsigned long elf_base;
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci	BUG_ON(cpu >= ARRAY_SIZE(kvm_arm_hyp_percpu_base));
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	cpu_base_array = (unsigned long *)&kvm_arm_hyp_percpu_base;
3762306a36Sopenharmony_ci	this_cpu_base = kern_hyp_va(cpu_base_array[cpu]);
3862306a36Sopenharmony_ci	elf_base = (unsigned long)&__per_cpu_start;
3962306a36Sopenharmony_ci	return this_cpu_base - elf_base;
4062306a36Sopenharmony_ci}
41