1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun // Copyright (C) 2018 Hangzhou C-SKY Microsystems co.,ltd.
3*4882a593Smuzhiyun
4*4882a593Smuzhiyun #include <linux/bitops.h>
5*4882a593Smuzhiyun #include <linux/sched.h>
6*4882a593Smuzhiyun #include <linux/slab.h>
7*4882a593Smuzhiyun #include <linux/mm.h>
8*4882a593Smuzhiyun
9*4882a593Smuzhiyun #include <asm/asid.h>
10*4882a593Smuzhiyun #include <asm/mmu_context.h>
11*4882a593Smuzhiyun #include <asm/smp.h>
12*4882a593Smuzhiyun #include <asm/tlbflush.h>
13*4882a593Smuzhiyun
14*4882a593Smuzhiyun static DEFINE_PER_CPU(atomic64_t, active_asids);
15*4882a593Smuzhiyun static DEFINE_PER_CPU(u64, reserved_asids);
16*4882a593Smuzhiyun
17*4882a593Smuzhiyun struct asid_info asid_info;
18*4882a593Smuzhiyun
check_and_switch_context(struct mm_struct * mm,unsigned int cpu)19*4882a593Smuzhiyun void check_and_switch_context(struct mm_struct *mm, unsigned int cpu)
20*4882a593Smuzhiyun {
21*4882a593Smuzhiyun asid_check_context(&asid_info, &mm->context.asid, cpu, mm);
22*4882a593Smuzhiyun }
23*4882a593Smuzhiyun
asid_flush_cpu_ctxt(void)24*4882a593Smuzhiyun static void asid_flush_cpu_ctxt(void)
25*4882a593Smuzhiyun {
26*4882a593Smuzhiyun local_tlb_invalid_all();
27*4882a593Smuzhiyun }
28*4882a593Smuzhiyun
asids_init(void)29*4882a593Smuzhiyun static int asids_init(void)
30*4882a593Smuzhiyun {
31*4882a593Smuzhiyun BUG_ON(((1 << CONFIG_CPU_ASID_BITS) - 1) <= num_possible_cpus());
32*4882a593Smuzhiyun
33*4882a593Smuzhiyun if (asid_allocator_init(&asid_info, CONFIG_CPU_ASID_BITS, 1,
34*4882a593Smuzhiyun asid_flush_cpu_ctxt))
35*4882a593Smuzhiyun panic("Unable to initialize ASID allocator for %lu ASIDs\n",
36*4882a593Smuzhiyun NUM_ASIDS(&asid_info));
37*4882a593Smuzhiyun
38*4882a593Smuzhiyun asid_info.active = &active_asids;
39*4882a593Smuzhiyun asid_info.reserved = &reserved_asids;
40*4882a593Smuzhiyun
41*4882a593Smuzhiyun pr_info("ASID allocator initialised with %lu entries\n",
42*4882a593Smuzhiyun NUM_CTXT_ASIDS(&asid_info));
43*4882a593Smuzhiyun
44*4882a593Smuzhiyun return 0;
45*4882a593Smuzhiyun }
46*4882a593Smuzhiyun early_initcall(asids_init);
47