Merge tag 'block-5.14-2021-08-07' of git://git.kernel.dk/linux-block
[linux-2.6-microblaze.git] / arch / csky / mm / tcm.c
1 // SPDX-License-Identifier: GPL-2.0
2
3 #include <linux/highmem.h>
4 #include <linux/genalloc.h>
5 #include <asm/tlbflush.h>
6 #include <asm/fixmap.h>
7
8 #if (CONFIG_ITCM_RAM_BASE == 0xffffffff)
9 #error "You should define ITCM_RAM_BASE"
10 #endif
11
12 #ifdef CONFIG_HAVE_DTCM
13 #if (CONFIG_DTCM_RAM_BASE == 0xffffffff)
14 #error "You should define DTCM_RAM_BASE"
15 #endif
16
17 #if (CONFIG_DTCM_RAM_BASE == CONFIG_ITCM_RAM_BASE)
18 #error "You should define correct DTCM_RAM_BASE"
19 #endif
20 #endif
21
22 extern char __tcm_start, __tcm_end, __dtcm_start;
23
24 static struct gen_pool *tcm_pool;
25
26 static void __init tcm_mapping_init(void)
27 {
28         pte_t *tcm_pte;
29         unsigned long vaddr, paddr;
30         int i;
31
32         paddr = CONFIG_ITCM_RAM_BASE;
33
34         if (pfn_valid(PFN_DOWN(CONFIG_ITCM_RAM_BASE)))
35                 goto panic;
36
37 #ifndef CONFIG_HAVE_DTCM
38         for (i = 0; i < TCM_NR_PAGES; i++) {
39 #else
40         for (i = 0; i < CONFIG_ITCM_NR_PAGES; i++) {
41 #endif
42                 vaddr = __fix_to_virt(FIX_TCM - i);
43
44                 tcm_pte =
45                         pte_offset_kernel((pmd_t *)pgd_offset_k(vaddr), vaddr);
46
47                 set_pte(tcm_pte, pfn_pte(__phys_to_pfn(paddr), PAGE_KERNEL));
48
49                 flush_tlb_one(vaddr);
50
51                 paddr = paddr + PAGE_SIZE;
52         }
53
54 #ifdef CONFIG_HAVE_DTCM
55         if (pfn_valid(PFN_DOWN(CONFIG_DTCM_RAM_BASE)))
56                 goto panic;
57
58         paddr = CONFIG_DTCM_RAM_BASE;
59
60         for (i = 0; i < CONFIG_DTCM_NR_PAGES; i++) {
61                 vaddr = __fix_to_virt(FIX_TCM - CONFIG_ITCM_NR_PAGES - i);
62
63                 tcm_pte =
64                         pte_offset_kernel((pmd_t *) pgd_offset_k(vaddr), vaddr);
65
66                 set_pte(tcm_pte, pfn_pte(__phys_to_pfn(paddr), PAGE_KERNEL));
67
68                 flush_tlb_one(vaddr);
69
70                 paddr = paddr + PAGE_SIZE;
71         }
72 #endif
73
74 #ifndef CONFIG_HAVE_DTCM
75         memcpy((void *)__fix_to_virt(FIX_TCM),
76                                 &__tcm_start, &__tcm_end - &__tcm_start);
77
78         pr_info("%s: mapping tcm va:0x%08lx to pa:0x%08x\n",
79                         __func__, __fix_to_virt(FIX_TCM), CONFIG_ITCM_RAM_BASE);
80
81         pr_info("%s: __tcm_start va:0x%08lx size:%d\n",
82                         __func__, (unsigned long)&__tcm_start, &__tcm_end - &__tcm_start);
83 #else
84         memcpy((void *)__fix_to_virt(FIX_TCM),
85                                 &__tcm_start, &__dtcm_start - &__tcm_start);
86
87         pr_info("%s: mapping itcm va:0x%08lx to pa:0x%08x\n",
88                         __func__, __fix_to_virt(FIX_TCM), CONFIG_ITCM_RAM_BASE);
89
90         pr_info("%s: __itcm_start va:0x%08lx size:%d\n",
91                         __func__, (unsigned long)&__tcm_start, &__dtcm_start - &__tcm_start);
92
93         memcpy((void *)__fix_to_virt(FIX_TCM - CONFIG_ITCM_NR_PAGES),
94                                 &__dtcm_start, &__tcm_end - &__dtcm_start);
95
96         pr_info("%s: mapping dtcm va:0x%08lx to pa:0x%08x\n",
97                         __func__, __fix_to_virt(FIX_TCM - CONFIG_ITCM_NR_PAGES),
98                                                 CONFIG_DTCM_RAM_BASE);
99
100         pr_info("%s: __dtcm_start va:0x%08lx size:%d\n",
101                         __func__, (unsigned long)&__dtcm_start, &__tcm_end - &__dtcm_start);
102
103 #endif
104         return;
105 panic:
106         panic("TCM init error");
107 }
108
109 void *tcm_alloc(size_t len)
110 {
111         unsigned long vaddr;
112
113         if (!tcm_pool)
114                 return NULL;
115
116         vaddr = gen_pool_alloc(tcm_pool, len);
117         if (!vaddr)
118                 return NULL;
119
120         return (void *) vaddr;
121 }
122 EXPORT_SYMBOL(tcm_alloc);
123
124 void tcm_free(void *addr, size_t len)
125 {
126         gen_pool_free(tcm_pool, (unsigned long) addr, len);
127 }
128 EXPORT_SYMBOL(tcm_free);
129
130 static int __init tcm_setup_pool(void)
131 {
132 #ifndef CONFIG_HAVE_DTCM
133         u32 pool_size = (u32) (TCM_NR_PAGES * PAGE_SIZE)
134                                 - (u32) (&__tcm_end - &__tcm_start);
135
136         u32 tcm_pool_start = __fix_to_virt(FIX_TCM)
137                                 + (u32) (&__tcm_end - &__tcm_start);
138 #else
139         u32 pool_size = (u32) (CONFIG_DTCM_NR_PAGES * PAGE_SIZE)
140                                 - (u32) (&__tcm_end - &__dtcm_start);
141
142         u32 tcm_pool_start = __fix_to_virt(FIX_TCM - CONFIG_ITCM_NR_PAGES)
143                                 + (u32) (&__tcm_end - &__dtcm_start);
144 #endif
145         int ret;
146
147         tcm_pool = gen_pool_create(2, -1);
148
149         ret = gen_pool_add(tcm_pool, tcm_pool_start, pool_size, -1);
150         if (ret) {
151                 pr_err("%s: gen_pool add failed!\n", __func__);
152                 return ret;
153         }
154
155         pr_info("%s: Added %d bytes @ 0x%08x to memory pool\n",
156                 __func__, pool_size, tcm_pool_start);
157
158         return 0;
159 }
160
161 static int __init tcm_init(void)
162 {
163         tcm_mapping_init();
164
165         tcm_setup_pool();
166
167         return 0;
168 }
169 arch_initcall(tcm_init);