1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Copyright 2005, Paul Mackerras, IBM Corporation.
4  * Copyright 2009, Benjamin Herrenschmidt, IBM Corporation.
5  * Copyright 2015-2016, Aneesh Kumar K.V, IBM Corporation.
6  */
7 
8 #include <linux/sched.h>
9 #include <linux/mm_types.h>
10 #include <linux/mm.h>
11 #include <linux/stop_machine.h>
12 
13 #include <asm/sections.h>
14 #include <asm/mmu.h>
15 #include <asm/tlb.h>
16 #include <asm/firmware.h>
17 
18 #include <mm/mmu_decl.h>
19 
20 #include <trace/events/thp.h>
21 
22 #if H_PGTABLE_RANGE > (USER_VSID_RANGE * (TASK_SIZE_USER64 / TASK_CONTEXT_SIZE))
23 #warning Limited user VSID range means pagetable space is wasted
24 #endif
25 
26 #ifdef CONFIG_SPARSEMEM_VMEMMAP
27 /*
28  * vmemmap is the starting address of the virtual address space where
29  * struct pages are allocated for all possible PFNs present on the system
30  * including holes and bad memory (hence sparse). These virtual struct
31  * pages are stored in sequence in this virtual address space irrespective
32  * of the fact whether the corresponding PFN is valid or not. This achieves
33  * constant relationship between address of struct page and its PFN.
34  *
35  * During boot or memory hotplug operation when a new memory section is
36  * added, physical memory allocation (including hash table bolting) will
37  * be performed for the set of struct pages which are part of the memory
38  * section. This saves memory by not allocating struct pages for PFNs
39  * which are not valid.
40  *
41  *		----------------------------------------------
42  *		| PHYSICAL ALLOCATION OF VIRTUAL STRUCT PAGES|
43  *		----------------------------------------------
44  *
45  *	   f000000000000000                  c000000000000000
46  * vmemmap +--------------+                  +--------------+
47  *  +      |  page struct | +--------------> |  page struct |
48  *  |      +--------------+                  +--------------+
49  *  |      |  page struct | +--------------> |  page struct |
50  *  |      +--------------+ |                +--------------+
51  *  |      |  page struct | +       +------> |  page struct |
52  *  |      +--------------+         |        +--------------+
53  *  |      |  page struct |         |   +--> |  page struct |
54  *  |      +--------------+         |   |    +--------------+
55  *  |      |  page struct |         |   |
56  *  |      +--------------+         |   |
57  *  |      |  page struct |         |   |
58  *  |      +--------------+         |   |
59  *  |      |  page struct |         |   |
60  *  |      +--------------+         |   |
61  *  |      |  page struct |         |   |
62  *  |      +--------------+         |   |
63  *  |      |  page struct | +-------+   |
64  *  |      +--------------+             |
65  *  |      |  page struct | +-----------+
66  *  |      +--------------+
67  *  |      |  page struct | No mapping
68  *  |      +--------------+
69  *  |      |  page struct | No mapping
70  *  v      +--------------+
71  *
72  *		-----------------------------------------
73  *		| RELATION BETWEEN STRUCT PAGES AND PFNS|
74  *		-----------------------------------------
75  *
76  * vmemmap +--------------+                 +---------------+
77  *  +      |  page struct | +-------------> |      PFN      |
78  *  |      +--------------+                 +---------------+
79  *  |      |  page struct | +-------------> |      PFN      |
80  *  |      +--------------+                 +---------------+
81  *  |      |  page struct | +-------------> |      PFN      |
82  *  |      +--------------+                 +---------------+
83  *  |      |  page struct | +-------------> |      PFN      |
84  *  |      +--------------+                 +---------------+
85  *  |      |              |
86  *  |      +--------------+
87  *  |      |              |
88  *  |      +--------------+
89  *  |      |              |
90  *  |      +--------------+                 +---------------+
91  *  |      |  page struct | +-------------> |      PFN      |
92  *  |      +--------------+                 +---------------+
93  *  |      |              |
94  *  |      +--------------+
95  *  |      |              |
96  *  |      +--------------+                 +---------------+
97  *  |      |  page struct | +-------------> |      PFN      |
98  *  |      +--------------+                 +---------------+
99  *  |      |  page struct | +-------------> |      PFN      |
100  *  v      +--------------+                 +---------------+
101  */
102 /*
103  * On hash-based CPUs, the vmemmap is bolted in the hash table.
104  *
105  */
106 int __meminit hash__vmemmap_create_mapping(unsigned long start,
107 				       unsigned long page_size,
108 				       unsigned long phys)
109 {
110 	int rc;
111 
112 	if ((start + page_size) >= H_VMEMMAP_END) {
113 		pr_warn("Outside the supported range\n");
114 		return -1;
115 	}
116 
117 	rc = htab_bolt_mapping(start, start + page_size, phys,
118 			       pgprot_val(PAGE_KERNEL),
119 			       mmu_vmemmap_psize, mmu_kernel_ssize);
120 	if (rc < 0) {
121 		int rc2 = htab_remove_mapping(start, start + page_size,
122 					      mmu_vmemmap_psize,
123 					      mmu_kernel_ssize);
124 		BUG_ON(rc2 && (rc2 != -ENOENT));
125 	}
126 	return rc;
127 }
128 
129 #ifdef CONFIG_MEMORY_HOTPLUG
130 void hash__vmemmap_remove_mapping(unsigned long start,
131 			      unsigned long page_size)
132 {
133 	int rc = htab_remove_mapping(start, start + page_size,
134 				     mmu_vmemmap_psize,
135 				     mmu_kernel_ssize);
136 	BUG_ON((rc < 0) && (rc != -ENOENT));
137 	WARN_ON(rc == -ENOENT);
138 }
139 #endif
140 #endif /* CONFIG_SPARSEMEM_VMEMMAP */
141 
142 /*
143  * map_kernel_page currently only called by __ioremap
144  * map_kernel_page adds an entry to the ioremap page table
145  * and adds an entry to the HPT, possibly bolting it
146  */
147 int hash__map_kernel_page(unsigned long ea, unsigned long pa, pgprot_t prot)
148 {
149 	pgd_t *pgdp;
150 	p4d_t *p4dp;
151 	pud_t *pudp;
152 	pmd_t *pmdp;
153 	pte_t *ptep;
154 
155 	BUILD_BUG_ON(TASK_SIZE_USER64 > H_PGTABLE_RANGE);
156 	if (slab_is_available()) {
157 		pgdp = pgd_offset_k(ea);
158 		p4dp = p4d_offset(pgdp, ea);
159 		pudp = pud_alloc(&init_mm, p4dp, ea);
160 		if (!pudp)
161 			return -ENOMEM;
162 		pmdp = pmd_alloc(&init_mm, pudp, ea);
163 		if (!pmdp)
164 			return -ENOMEM;
165 		ptep = pte_alloc_kernel(pmdp, ea);
166 		if (!ptep)
167 			return -ENOMEM;
168 		set_pte_at(&init_mm, ea, ptep, pfn_pte(pa >> PAGE_SHIFT, prot));
169 	} else {
170 		/*
171 		 * If the mm subsystem is not fully up, we cannot create a
172 		 * linux page table entry for this mapping.  Simply bolt an
173 		 * entry in the hardware page table.
174 		 *
175 		 */
176 		if (htab_bolt_mapping(ea, ea + PAGE_SIZE, pa, pgprot_val(prot),
177 				      mmu_io_psize, mmu_kernel_ssize)) {
178 			printk(KERN_ERR "Failed to do bolted mapping IO "
179 			       "memory at %016lx !\n", pa);
180 			return -ENOMEM;
181 		}
182 	}
183 
184 	smp_wmb();
185 	return 0;
186 }
187 
188 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
189 
190 unsigned long hash__pmd_hugepage_update(struct mm_struct *mm, unsigned long addr,
191 				    pmd_t *pmdp, unsigned long clr,
192 				    unsigned long set)
193 {
194 	__be64 old_be, tmp;
195 	unsigned long old;
196 
197 #ifdef CONFIG_DEBUG_VM
198 	WARN_ON(!hash__pmd_trans_huge(*pmdp) && !pmd_devmap(*pmdp));
199 	assert_spin_locked(pmd_lockptr(mm, pmdp));
200 #endif
201 
202 	__asm__ __volatile__(
203 	"1:	ldarx	%0,0,%3\n\
204 		and.	%1,%0,%6\n\
205 		bne-	1b \n\
206 		andc	%1,%0,%4 \n\
207 		or	%1,%1,%7\n\
208 		stdcx.	%1,0,%3 \n\
209 		bne-	1b"
210 	: "=&r" (old_be), "=&r" (tmp), "=m" (*pmdp)
211 	: "r" (pmdp), "r" (cpu_to_be64(clr)), "m" (*pmdp),
212 	  "r" (cpu_to_be64(H_PAGE_BUSY)), "r" (cpu_to_be64(set))
213 	: "cc" );
214 
215 	old = be64_to_cpu(old_be);
216 
217 	trace_hugepage_update(addr, old, clr, set);
218 	if (old & H_PAGE_HASHPTE)
219 		hpte_do_hugepage_flush(mm, addr, pmdp, old);
220 	return old;
221 }
222 
223 pmd_t hash__pmdp_collapse_flush(struct vm_area_struct *vma, unsigned long address,
224 			    pmd_t *pmdp)
225 {
226 	pmd_t pmd;
227 
228 	VM_BUG_ON(address & ~HPAGE_PMD_MASK);
229 	VM_BUG_ON(pmd_trans_huge(*pmdp));
230 	VM_BUG_ON(pmd_devmap(*pmdp));
231 
232 	pmd = *pmdp;
233 	pmd_clear(pmdp);
234 	/*
235 	 * Wait for all pending hash_page to finish. This is needed
236 	 * in case of subpage collapse. When we collapse normal pages
237 	 * to hugepage, we first clear the pmd, then invalidate all
238 	 * the PTE entries. The assumption here is that any low level
239 	 * page fault will see a none pmd and take the slow path that
240 	 * will wait on mmap_lock. But we could very well be in a
241 	 * hash_page with local ptep pointer value. Such a hash page
242 	 * can result in adding new HPTE entries for normal subpages.
243 	 * That means we could be modifying the page content as we
244 	 * copy them to a huge page. So wait for parallel hash_page
245 	 * to finish before invalidating HPTE entries. We can do this
246 	 * by sending an IPI to all the cpus and executing a dummy
247 	 * function there.
248 	 */
249 	serialize_against_pte_lookup(vma->vm_mm);
250 	/*
251 	 * Now invalidate the hpte entries in the range
252 	 * covered by pmd. This make sure we take a
253 	 * fault and will find the pmd as none, which will
254 	 * result in a major fault which takes mmap_lock and
255 	 * hence wait for collapse to complete. Without this
256 	 * the __collapse_huge_page_copy can result in copying
257 	 * the old content.
258 	 */
259 	flush_hash_table_pmd_range(vma->vm_mm, &pmd, address);
260 	return pmd;
261 }
262 
263 /*
264  * We want to put the pgtable in pmd and use pgtable for tracking
265  * the base page size hptes
266  */
267 void hash__pgtable_trans_huge_deposit(struct mm_struct *mm, pmd_t *pmdp,
268 				  pgtable_t pgtable)
269 {
270 	pgtable_t *pgtable_slot;
271 
272 	assert_spin_locked(pmd_lockptr(mm, pmdp));
273 	/*
274 	 * we store the pgtable in the second half of PMD
275 	 */
276 	pgtable_slot = (pgtable_t *)pmdp + PTRS_PER_PMD;
277 	*pgtable_slot = pgtable;
278 	/*
279 	 * expose the deposited pgtable to other cpus.
280 	 * before we set the hugepage PTE at pmd level
281 	 * hash fault code looks at the deposted pgtable
282 	 * to store hash index values.
283 	 */
284 	smp_wmb();
285 }
286 
287 pgtable_t hash__pgtable_trans_huge_withdraw(struct mm_struct *mm, pmd_t *pmdp)
288 {
289 	pgtable_t pgtable;
290 	pgtable_t *pgtable_slot;
291 
292 	assert_spin_locked(pmd_lockptr(mm, pmdp));
293 
294 	pgtable_slot = (pgtable_t *)pmdp + PTRS_PER_PMD;
295 	pgtable = *pgtable_slot;
296 	/*
297 	 * Once we withdraw, mark the entry NULL.
298 	 */
299 	*pgtable_slot = NULL;
300 	/*
301 	 * We store HPTE information in the deposited PTE fragment.
302 	 * zero out the content on withdraw.
303 	 */
304 	memset(pgtable, 0, PTE_FRAG_SIZE);
305 	return pgtable;
306 }
307 
308 /*
309  * A linux hugepage PMD was changed and the corresponding hash table entries
310  * neesd to be flushed.
311  */
312 void hpte_do_hugepage_flush(struct mm_struct *mm, unsigned long addr,
313 			    pmd_t *pmdp, unsigned long old_pmd)
314 {
315 	int ssize;
316 	unsigned int psize;
317 	unsigned long vsid;
318 	unsigned long flags = 0;
319 
320 	/* get the base page size,vsid and segment size */
321 #ifdef CONFIG_DEBUG_VM
322 	psize = get_slice_psize(mm, addr);
323 	BUG_ON(psize == MMU_PAGE_16M);
324 #endif
325 	if (old_pmd & H_PAGE_COMBO)
326 		psize = MMU_PAGE_4K;
327 	else
328 		psize = MMU_PAGE_64K;
329 
330 	if (!is_kernel_addr(addr)) {
331 		ssize = user_segment_size(addr);
332 		vsid = get_user_vsid(&mm->context, addr, ssize);
333 		WARN_ON(vsid == 0);
334 	} else {
335 		vsid = get_kernel_vsid(addr, mmu_kernel_ssize);
336 		ssize = mmu_kernel_ssize;
337 	}
338 
339 	if (mm_is_thread_local(mm))
340 		flags |= HPTE_LOCAL_UPDATE;
341 
342 	return flush_hash_hugepage(vsid, addr, pmdp, psize, ssize, flags);
343 }
344 
345 pmd_t hash__pmdp_huge_get_and_clear(struct mm_struct *mm,
346 				unsigned long addr, pmd_t *pmdp)
347 {
348 	pmd_t old_pmd;
349 	pgtable_t pgtable;
350 	unsigned long old;
351 	pgtable_t *pgtable_slot;
352 
353 	old = pmd_hugepage_update(mm, addr, pmdp, ~0UL, 0);
354 	old_pmd = __pmd(old);
355 	/*
356 	 * We have pmd == none and we are holding page_table_lock.
357 	 * So we can safely go and clear the pgtable hash
358 	 * index info.
359 	 */
360 	pgtable_slot = (pgtable_t *)pmdp + PTRS_PER_PMD;
361 	pgtable = *pgtable_slot;
362 	/*
363 	 * Let's zero out old valid and hash index details
364 	 * hash fault look at them.
365 	 */
366 	memset(pgtable, 0, PTE_FRAG_SIZE);
367 	return old_pmd;
368 }
369 
370 int hash__has_transparent_hugepage(void)
371 {
372 
373 	if (!mmu_has_feature(MMU_FTR_16M_PAGE))
374 		return 0;
375 	/*
376 	 * We support THP only if PMD_SIZE is 16MB.
377 	 */
378 	if (mmu_psize_defs[MMU_PAGE_16M].shift != PMD_SHIFT)
379 		return 0;
380 	/*
381 	 * We need to make sure that we support 16MB hugepage in a segment
382 	 * with base page size 64K or 4K. We only enable THP with a PAGE_SIZE
383 	 * of 64K.
384 	 */
385 	/*
386 	 * If we have 64K HPTE, we will be using that by default
387 	 */
388 	if (mmu_psize_defs[MMU_PAGE_64K].shift &&
389 	    (mmu_psize_defs[MMU_PAGE_64K].penc[MMU_PAGE_16M] == -1))
390 		return 0;
391 	/*
392 	 * Ok we only have 4K HPTE
393 	 */
394 	if (mmu_psize_defs[MMU_PAGE_4K].penc[MMU_PAGE_16M] == -1)
395 		return 0;
396 
397 	return 1;
398 }
399 EXPORT_SYMBOL_GPL(hash__has_transparent_hugepage);
400 
401 #endif /* CONFIG_TRANSPARENT_HUGEPAGE */
402 
403 #ifdef CONFIG_STRICT_KERNEL_RWX
404 
405 struct change_memory_parms {
406 	unsigned long start, end, newpp;
407 	unsigned int step, nr_cpus, master_cpu;
408 	atomic_t cpu_counter;
409 };
410 
411 // We'd rather this was on the stack but it has to be in the RMO
412 static struct change_memory_parms chmem_parms;
413 
414 // And therefore we need a lock to protect it from concurrent use
415 static DEFINE_MUTEX(chmem_lock);
416 
417 static void change_memory_range(unsigned long start, unsigned long end,
418 				unsigned int step, unsigned long newpp)
419 {
420 	unsigned long idx;
421 
422 	pr_debug("Changing page protection on range 0x%lx-0x%lx, to 0x%lx, step 0x%x\n",
423 		 start, end, newpp, step);
424 
425 	for (idx = start; idx < end; idx += step)
426 		/* Not sure if we can do much with the return value */
427 		mmu_hash_ops.hpte_updateboltedpp(newpp, idx, mmu_linear_psize,
428 							mmu_kernel_ssize);
429 }
430 
431 static int notrace chmem_secondary_loop(struct change_memory_parms *parms)
432 {
433 	unsigned long msr, tmp, flags;
434 	int *p;
435 
436 	p = &parms->cpu_counter.counter;
437 
438 	local_irq_save(flags);
439 	hard_irq_disable();
440 
441 	asm volatile (
442 	// Switch to real mode and leave interrupts off
443 	"mfmsr	%[msr]			;"
444 	"li	%[tmp], %[MSR_IR_DR]	;"
445 	"andc	%[tmp], %[msr], %[tmp]	;"
446 	"mtmsrd %[tmp]			;"
447 
448 	// Tell the master we are in real mode
449 	"1:				"
450 	"lwarx	%[tmp], 0, %[p]		;"
451 	"addic	%[tmp], %[tmp], -1	;"
452 	"stwcx.	%[tmp], 0, %[p]		;"
453 	"bne-	1b			;"
454 
455 	// Spin until the counter goes to zero
456 	"2:				;"
457 	"lwz	%[tmp], 0(%[p])		;"
458 	"cmpwi	%[tmp], 0		;"
459 	"bne-	2b			;"
460 
461 	// Switch back to virtual mode
462 	"mtmsrd %[msr]			;"
463 
464 	: // outputs
465 	  [msr] "=&r" (msr), [tmp] "=&b" (tmp), "+m" (*p)
466 	: // inputs
467 	  [p] "b" (p), [MSR_IR_DR] "i" (MSR_IR | MSR_DR)
468 	: // clobbers
469 	  "cc", "xer"
470 	);
471 
472 	local_irq_restore(flags);
473 
474 	return 0;
475 }
476 
477 static int change_memory_range_fn(void *data)
478 {
479 	struct change_memory_parms *parms = data;
480 
481 	if (parms->master_cpu != smp_processor_id())
482 		return chmem_secondary_loop(parms);
483 
484 	// Wait for all but one CPU (this one) to call-in
485 	while (atomic_read(&parms->cpu_counter) > 1)
486 		barrier();
487 
488 	change_memory_range(parms->start, parms->end, parms->step, parms->newpp);
489 
490 	mb();
491 
492 	// Signal the other CPUs that we're done
493 	atomic_dec(&parms->cpu_counter);
494 
495 	return 0;
496 }
497 
498 static bool hash__change_memory_range(unsigned long start, unsigned long end,
499 				      unsigned long newpp)
500 {
501 	unsigned int step, shift;
502 
503 	shift = mmu_psize_defs[mmu_linear_psize].shift;
504 	step = 1 << shift;
505 
506 	start = ALIGN_DOWN(start, step);
507 	end = ALIGN(end, step); // aligns up
508 
509 	if (start >= end)
510 		return false;
511 
512 	if (firmware_has_feature(FW_FEATURE_LPAR)) {
513 		mutex_lock(&chmem_lock);
514 
515 		chmem_parms.start = start;
516 		chmem_parms.end = end;
517 		chmem_parms.step = step;
518 		chmem_parms.newpp = newpp;
519 		chmem_parms.master_cpu = smp_processor_id();
520 
521 		cpus_read_lock();
522 
523 		atomic_set(&chmem_parms.cpu_counter, num_online_cpus());
524 
525 		// Ensure state is consistent before we call the other CPUs
526 		mb();
527 
528 		stop_machine_cpuslocked(change_memory_range_fn, &chmem_parms,
529 					cpu_online_mask);
530 
531 		cpus_read_unlock();
532 		mutex_unlock(&chmem_lock);
533 	} else
534 		change_memory_range(start, end, step, newpp);
535 
536 	return true;
537 }
538 
539 void hash__mark_rodata_ro(void)
540 {
541 	unsigned long start, end, pp;
542 
543 	start = (unsigned long)_stext;
544 	end = (unsigned long)__end_rodata;
545 
546 	pp = htab_convert_pte_flags(pgprot_val(PAGE_KERNEL_ROX), HPTE_USE_KERNEL_KEY);
547 
548 	WARN_ON(!hash__change_memory_range(start, end, pp));
549 }
550 
551 void hash__mark_initmem_nx(void)
552 {
553 	unsigned long start, end, pp;
554 
555 	start = (unsigned long)__init_begin;
556 	end = (unsigned long)__init_end;
557 
558 	pp = htab_convert_pte_flags(pgprot_val(PAGE_KERNEL), HPTE_USE_KERNEL_KEY);
559 
560 	WARN_ON(!hash__change_memory_range(start, end, pp));
561 }
562 #endif
563