xref: /linux/arch/powerpc/kvm/book3s_hv_rm_mmu.c (revision e0bf6c5ca2d3281f231c5f0c9bf145e9513644de)
1 /*
2  * This program is free software; you can redistribute it and/or modify
3  * it under the terms of the GNU General Public License, version 2, as
4  * published by the Free Software Foundation.
5  *
6  * Copyright 2010-2011 Paul Mackerras, IBM Corp. <paulus@au1.ibm.com>
7  */
8 
9 #include <linux/types.h>
10 #include <linux/string.h>
11 #include <linux/kvm.h>
12 #include <linux/kvm_host.h>
13 #include <linux/hugetlb.h>
14 #include <linux/module.h>
15 
16 #include <asm/tlbflush.h>
17 #include <asm/kvm_ppc.h>
18 #include <asm/kvm_book3s.h>
19 #include <asm/mmu-hash64.h>
20 #include <asm/hvcall.h>
21 #include <asm/synch.h>
22 #include <asm/ppc-opcode.h>
23 
24 /* Translate address of a vmalloc'd thing to a linear map address */
25 static void *real_vmalloc_addr(void *x)
26 {
27 	unsigned long addr = (unsigned long) x;
28 	pte_t *p;
29 
30 	p = find_linux_pte_or_hugepte(swapper_pg_dir, addr, NULL);
31 	if (!p || !pte_present(*p))
32 		return NULL;
33 	/* assume we don't have huge pages in vmalloc space... */
34 	addr = (pte_pfn(*p) << PAGE_SHIFT) | (addr & ~PAGE_MASK);
35 	return __va(addr);
36 }
37 
38 /* Return 1 if we need to do a global tlbie, 0 if we can use tlbiel */
39 static int global_invalidates(struct kvm *kvm, unsigned long flags)
40 {
41 	int global;
42 
43 	/*
44 	 * If there is only one vcore, and it's currently running,
45 	 * as indicated by local_paca->kvm_hstate.kvm_vcpu being set,
46 	 * we can use tlbiel as long as we mark all other physical
47 	 * cores as potentially having stale TLB entries for this lpid.
48 	 * Otherwise, don't use tlbiel.
49 	 */
50 	if (kvm->arch.online_vcores == 1 && local_paca->kvm_hstate.kvm_vcpu)
51 		global = 0;
52 	else
53 		global = 1;
54 
55 	if (!global) {
56 		/* any other core might now have stale TLB entries... */
57 		smp_wmb();
58 		cpumask_setall(&kvm->arch.need_tlb_flush);
59 		cpumask_clear_cpu(local_paca->kvm_hstate.kvm_vcore->pcpu,
60 				  &kvm->arch.need_tlb_flush);
61 	}
62 
63 	return global;
64 }
65 
66 /*
67  * Add this HPTE into the chain for the real page.
68  * Must be called with the chain locked; it unlocks the chain.
69  */
70 void kvmppc_add_revmap_chain(struct kvm *kvm, struct revmap_entry *rev,
71 			     unsigned long *rmap, long pte_index, int realmode)
72 {
73 	struct revmap_entry *head, *tail;
74 	unsigned long i;
75 
76 	if (*rmap & KVMPPC_RMAP_PRESENT) {
77 		i = *rmap & KVMPPC_RMAP_INDEX;
78 		head = &kvm->arch.revmap[i];
79 		if (realmode)
80 			head = real_vmalloc_addr(head);
81 		tail = &kvm->arch.revmap[head->back];
82 		if (realmode)
83 			tail = real_vmalloc_addr(tail);
84 		rev->forw = i;
85 		rev->back = head->back;
86 		tail->forw = pte_index;
87 		head->back = pte_index;
88 	} else {
89 		rev->forw = rev->back = pte_index;
90 		*rmap = (*rmap & ~KVMPPC_RMAP_INDEX) |
91 			pte_index | KVMPPC_RMAP_PRESENT;
92 	}
93 	unlock_rmap(rmap);
94 }
95 EXPORT_SYMBOL_GPL(kvmppc_add_revmap_chain);
96 
97 /* Remove this HPTE from the chain for a real page */
98 static void remove_revmap_chain(struct kvm *kvm, long pte_index,
99 				struct revmap_entry *rev,
100 				unsigned long hpte_v, unsigned long hpte_r)
101 {
102 	struct revmap_entry *next, *prev;
103 	unsigned long gfn, ptel, head;
104 	struct kvm_memory_slot *memslot;
105 	unsigned long *rmap;
106 	unsigned long rcbits;
107 
108 	rcbits = hpte_r & (HPTE_R_R | HPTE_R_C);
109 	ptel = rev->guest_rpte |= rcbits;
110 	gfn = hpte_rpn(ptel, hpte_page_size(hpte_v, ptel));
111 	memslot = __gfn_to_memslot(kvm_memslots_raw(kvm), gfn);
112 	if (!memslot)
113 		return;
114 
115 	rmap = real_vmalloc_addr(&memslot->arch.rmap[gfn - memslot->base_gfn]);
116 	lock_rmap(rmap);
117 
118 	head = *rmap & KVMPPC_RMAP_INDEX;
119 	next = real_vmalloc_addr(&kvm->arch.revmap[rev->forw]);
120 	prev = real_vmalloc_addr(&kvm->arch.revmap[rev->back]);
121 	next->back = rev->back;
122 	prev->forw = rev->forw;
123 	if (head == pte_index) {
124 		head = rev->forw;
125 		if (head == pte_index)
126 			*rmap &= ~(KVMPPC_RMAP_PRESENT | KVMPPC_RMAP_INDEX);
127 		else
128 			*rmap = (*rmap & ~KVMPPC_RMAP_INDEX) | head;
129 	}
130 	*rmap |= rcbits << KVMPPC_RMAP_RC_SHIFT;
131 	unlock_rmap(rmap);
132 }
133 
134 static pte_t lookup_linux_pte_and_update(pgd_t *pgdir, unsigned long hva,
135 			      int writing, unsigned long *pte_sizep)
136 {
137 	pte_t *ptep;
138 	unsigned long ps = *pte_sizep;
139 	unsigned int hugepage_shift;
140 
141 	ptep = find_linux_pte_or_hugepte(pgdir, hva, &hugepage_shift);
142 	if (!ptep)
143 		return __pte(0);
144 	if (hugepage_shift)
145 		*pte_sizep = 1ul << hugepage_shift;
146 	else
147 		*pte_sizep = PAGE_SIZE;
148 	if (ps > *pte_sizep)
149 		return __pte(0);
150 	return kvmppc_read_update_linux_pte(ptep, writing, hugepage_shift);
151 }
152 
153 static inline void unlock_hpte(__be64 *hpte, unsigned long hpte_v)
154 {
155 	asm volatile(PPC_RELEASE_BARRIER "" : : : "memory");
156 	hpte[0] = cpu_to_be64(hpte_v);
157 }
158 
159 long kvmppc_do_h_enter(struct kvm *kvm, unsigned long flags,
160 		       long pte_index, unsigned long pteh, unsigned long ptel,
161 		       pgd_t *pgdir, bool realmode, unsigned long *pte_idx_ret)
162 {
163 	unsigned long i, pa, gpa, gfn, psize;
164 	unsigned long slot_fn, hva;
165 	__be64 *hpte;
166 	struct revmap_entry *rev;
167 	unsigned long g_ptel;
168 	struct kvm_memory_slot *memslot;
169 	unsigned long pte_size;
170 	unsigned long is_io;
171 	unsigned long *rmap;
172 	pte_t pte;
173 	unsigned int writing;
174 	unsigned long mmu_seq;
175 	unsigned long rcbits;
176 
177 	psize = hpte_page_size(pteh, ptel);
178 	if (!psize)
179 		return H_PARAMETER;
180 	writing = hpte_is_writable(ptel);
181 	pteh &= ~(HPTE_V_HVLOCK | HPTE_V_ABSENT | HPTE_V_VALID);
182 	ptel &= ~HPTE_GR_RESERVED;
183 	g_ptel = ptel;
184 
185 	/* used later to detect if we might have been invalidated */
186 	mmu_seq = kvm->mmu_notifier_seq;
187 	smp_rmb();
188 
189 	/* Find the memslot (if any) for this address */
190 	gpa = (ptel & HPTE_R_RPN) & ~(psize - 1);
191 	gfn = gpa >> PAGE_SHIFT;
192 	memslot = __gfn_to_memslot(kvm_memslots_raw(kvm), gfn);
193 	pa = 0;
194 	is_io = ~0ul;
195 	rmap = NULL;
196 	if (!(memslot && !(memslot->flags & KVM_MEMSLOT_INVALID))) {
197 		/* Emulated MMIO - mark this with key=31 */
198 		pteh |= HPTE_V_ABSENT;
199 		ptel |= HPTE_R_KEY_HI | HPTE_R_KEY_LO;
200 		goto do_insert;
201 	}
202 
203 	/* Check if the requested page fits entirely in the memslot. */
204 	if (!slot_is_aligned(memslot, psize))
205 		return H_PARAMETER;
206 	slot_fn = gfn - memslot->base_gfn;
207 	rmap = &memslot->arch.rmap[slot_fn];
208 
209 	/* Translate to host virtual address */
210 	hva = __gfn_to_hva_memslot(memslot, gfn);
211 
212 	/* Look up the Linux PTE for the backing page */
213 	pte_size = psize;
214 	pte = lookup_linux_pte_and_update(pgdir, hva, writing, &pte_size);
215 	if (pte_present(pte) && !pte_protnone(pte)) {
216 		if (writing && !pte_write(pte))
217 			/* make the actual HPTE be read-only */
218 			ptel = hpte_make_readonly(ptel);
219 		is_io = hpte_cache_bits(pte_val(pte));
220 		pa = pte_pfn(pte) << PAGE_SHIFT;
221 		pa |= hva & (pte_size - 1);
222 		pa |= gpa & ~PAGE_MASK;
223 	}
224 
225 	if (pte_size < psize)
226 		return H_PARAMETER;
227 
228 	ptel &= ~(HPTE_R_PP0 - psize);
229 	ptel |= pa;
230 
231 	if (pa)
232 		pteh |= HPTE_V_VALID;
233 	else
234 		pteh |= HPTE_V_ABSENT;
235 
236 	/* Check WIMG */
237 	if (is_io != ~0ul && !hpte_cache_flags_ok(ptel, is_io)) {
238 		if (is_io)
239 			return H_PARAMETER;
240 		/*
241 		 * Allow guest to map emulated device memory as
242 		 * uncacheable, but actually make it cacheable.
243 		 */
244 		ptel &= ~(HPTE_R_W|HPTE_R_I|HPTE_R_G);
245 		ptel |= HPTE_R_M;
246 	}
247 
248 	/* Find and lock the HPTEG slot to use */
249  do_insert:
250 	if (pte_index >= kvm->arch.hpt_npte)
251 		return H_PARAMETER;
252 	if (likely((flags & H_EXACT) == 0)) {
253 		pte_index &= ~7UL;
254 		hpte = (__be64 *)(kvm->arch.hpt_virt + (pte_index << 4));
255 		for (i = 0; i < 8; ++i) {
256 			if ((be64_to_cpu(*hpte) & HPTE_V_VALID) == 0 &&
257 			    try_lock_hpte(hpte, HPTE_V_HVLOCK | HPTE_V_VALID |
258 					  HPTE_V_ABSENT))
259 				break;
260 			hpte += 2;
261 		}
262 		if (i == 8) {
263 			/*
264 			 * Since try_lock_hpte doesn't retry (not even stdcx.
265 			 * failures), it could be that there is a free slot
266 			 * but we transiently failed to lock it.  Try again,
267 			 * actually locking each slot and checking it.
268 			 */
269 			hpte -= 16;
270 			for (i = 0; i < 8; ++i) {
271 				u64 pte;
272 				while (!try_lock_hpte(hpte, HPTE_V_HVLOCK))
273 					cpu_relax();
274 				pte = be64_to_cpu(*hpte);
275 				if (!(pte & (HPTE_V_VALID | HPTE_V_ABSENT)))
276 					break;
277 				*hpte &= ~cpu_to_be64(HPTE_V_HVLOCK);
278 				hpte += 2;
279 			}
280 			if (i == 8)
281 				return H_PTEG_FULL;
282 		}
283 		pte_index += i;
284 	} else {
285 		hpte = (__be64 *)(kvm->arch.hpt_virt + (pte_index << 4));
286 		if (!try_lock_hpte(hpte, HPTE_V_HVLOCK | HPTE_V_VALID |
287 				   HPTE_V_ABSENT)) {
288 			/* Lock the slot and check again */
289 			u64 pte;
290 
291 			while (!try_lock_hpte(hpte, HPTE_V_HVLOCK))
292 				cpu_relax();
293 			pte = be64_to_cpu(*hpte);
294 			if (pte & (HPTE_V_VALID | HPTE_V_ABSENT)) {
295 				*hpte &= ~cpu_to_be64(HPTE_V_HVLOCK);
296 				return H_PTEG_FULL;
297 			}
298 		}
299 	}
300 
301 	/* Save away the guest's idea of the second HPTE dword */
302 	rev = &kvm->arch.revmap[pte_index];
303 	if (realmode)
304 		rev = real_vmalloc_addr(rev);
305 	if (rev) {
306 		rev->guest_rpte = g_ptel;
307 		note_hpte_modification(kvm, rev);
308 	}
309 
310 	/* Link HPTE into reverse-map chain */
311 	if (pteh & HPTE_V_VALID) {
312 		if (realmode)
313 			rmap = real_vmalloc_addr(rmap);
314 		lock_rmap(rmap);
315 		/* Check for pending invalidations under the rmap chain lock */
316 		if (mmu_notifier_retry(kvm, mmu_seq)) {
317 			/* inval in progress, write a non-present HPTE */
318 			pteh |= HPTE_V_ABSENT;
319 			pteh &= ~HPTE_V_VALID;
320 			unlock_rmap(rmap);
321 		} else {
322 			kvmppc_add_revmap_chain(kvm, rev, rmap, pte_index,
323 						realmode);
324 			/* Only set R/C in real HPTE if already set in *rmap */
325 			rcbits = *rmap >> KVMPPC_RMAP_RC_SHIFT;
326 			ptel &= rcbits | ~(HPTE_R_R | HPTE_R_C);
327 		}
328 	}
329 
330 	hpte[1] = cpu_to_be64(ptel);
331 
332 	/* Write the first HPTE dword, unlocking the HPTE and making it valid */
333 	eieio();
334 	hpte[0] = cpu_to_be64(pteh);
335 	asm volatile("ptesync" : : : "memory");
336 
337 	*pte_idx_ret = pte_index;
338 	return H_SUCCESS;
339 }
340 EXPORT_SYMBOL_GPL(kvmppc_do_h_enter);
341 
342 long kvmppc_h_enter(struct kvm_vcpu *vcpu, unsigned long flags,
343 		    long pte_index, unsigned long pteh, unsigned long ptel)
344 {
345 	return kvmppc_do_h_enter(vcpu->kvm, flags, pte_index, pteh, ptel,
346 				 vcpu->arch.pgdir, true, &vcpu->arch.gpr[4]);
347 }
348 
349 #ifdef __BIG_ENDIAN__
350 #define LOCK_TOKEN	(*(u32 *)(&get_paca()->lock_token))
351 #else
352 #define LOCK_TOKEN	(*(u32 *)(&get_paca()->paca_index))
353 #endif
354 
355 static inline int try_lock_tlbie(unsigned int *lock)
356 {
357 	unsigned int tmp, old;
358 	unsigned int token = LOCK_TOKEN;
359 
360 	asm volatile("1:lwarx	%1,0,%2\n"
361 		     "	cmpwi	cr0,%1,0\n"
362 		     "	bne	2f\n"
363 		     "  stwcx.	%3,0,%2\n"
364 		     "	bne-	1b\n"
365 		     "  isync\n"
366 		     "2:"
367 		     : "=&r" (tmp), "=&r" (old)
368 		     : "r" (lock), "r" (token)
369 		     : "cc", "memory");
370 	return old == 0;
371 }
372 
373 static void do_tlbies(struct kvm *kvm, unsigned long *rbvalues,
374 		      long npages, int global, bool need_sync)
375 {
376 	long i;
377 
378 	if (global) {
379 		while (!try_lock_tlbie(&kvm->arch.tlbie_lock))
380 			cpu_relax();
381 		if (need_sync)
382 			asm volatile("ptesync" : : : "memory");
383 		for (i = 0; i < npages; ++i)
384 			asm volatile(PPC_TLBIE(%1,%0) : :
385 				     "r" (rbvalues[i]), "r" (kvm->arch.lpid));
386 		asm volatile("eieio; tlbsync; ptesync" : : : "memory");
387 		kvm->arch.tlbie_lock = 0;
388 	} else {
389 		if (need_sync)
390 			asm volatile("ptesync" : : : "memory");
391 		for (i = 0; i < npages; ++i)
392 			asm volatile("tlbiel %0" : : "r" (rbvalues[i]));
393 		asm volatile("ptesync" : : : "memory");
394 	}
395 }
396 
397 long kvmppc_do_h_remove(struct kvm *kvm, unsigned long flags,
398 			unsigned long pte_index, unsigned long avpn,
399 			unsigned long *hpret)
400 {
401 	__be64 *hpte;
402 	unsigned long v, r, rb;
403 	struct revmap_entry *rev;
404 	u64 pte;
405 
406 	if (pte_index >= kvm->arch.hpt_npte)
407 		return H_PARAMETER;
408 	hpte = (__be64 *)(kvm->arch.hpt_virt + (pte_index << 4));
409 	while (!try_lock_hpte(hpte, HPTE_V_HVLOCK))
410 		cpu_relax();
411 	pte = be64_to_cpu(hpte[0]);
412 	if ((pte & (HPTE_V_ABSENT | HPTE_V_VALID)) == 0 ||
413 	    ((flags & H_AVPN) && (pte & ~0x7fUL) != avpn) ||
414 	    ((flags & H_ANDCOND) && (pte & avpn) != 0)) {
415 		hpte[0] &= ~cpu_to_be64(HPTE_V_HVLOCK);
416 		return H_NOT_FOUND;
417 	}
418 
419 	rev = real_vmalloc_addr(&kvm->arch.revmap[pte_index]);
420 	v = pte & ~HPTE_V_HVLOCK;
421 	if (v & HPTE_V_VALID) {
422 		u64 pte1;
423 
424 		pte1 = be64_to_cpu(hpte[1]);
425 		hpte[0] &= ~cpu_to_be64(HPTE_V_VALID);
426 		rb = compute_tlbie_rb(v, pte1, pte_index);
427 		do_tlbies(kvm, &rb, 1, global_invalidates(kvm, flags), true);
428 		/* Read PTE low word after tlbie to get final R/C values */
429 		remove_revmap_chain(kvm, pte_index, rev, v, pte1);
430 	}
431 	r = rev->guest_rpte & ~HPTE_GR_RESERVED;
432 	note_hpte_modification(kvm, rev);
433 	unlock_hpte(hpte, 0);
434 
435 	hpret[0] = v;
436 	hpret[1] = r;
437 	return H_SUCCESS;
438 }
439 EXPORT_SYMBOL_GPL(kvmppc_do_h_remove);
440 
441 long kvmppc_h_remove(struct kvm_vcpu *vcpu, unsigned long flags,
442 		     unsigned long pte_index, unsigned long avpn)
443 {
444 	return kvmppc_do_h_remove(vcpu->kvm, flags, pte_index, avpn,
445 				  &vcpu->arch.gpr[4]);
446 }
447 
448 long kvmppc_h_bulk_remove(struct kvm_vcpu *vcpu)
449 {
450 	struct kvm *kvm = vcpu->kvm;
451 	unsigned long *args = &vcpu->arch.gpr[4];
452 	__be64 *hp, *hptes[4];
453 	unsigned long tlbrb[4];
454 	long int i, j, k, n, found, indexes[4];
455 	unsigned long flags, req, pte_index, rcbits;
456 	int global;
457 	long int ret = H_SUCCESS;
458 	struct revmap_entry *rev, *revs[4];
459 	u64 hp0;
460 
461 	global = global_invalidates(kvm, 0);
462 	for (i = 0; i < 4 && ret == H_SUCCESS; ) {
463 		n = 0;
464 		for (; i < 4; ++i) {
465 			j = i * 2;
466 			pte_index = args[j];
467 			flags = pte_index >> 56;
468 			pte_index &= ((1ul << 56) - 1);
469 			req = flags >> 6;
470 			flags &= 3;
471 			if (req == 3) {		/* no more requests */
472 				i = 4;
473 				break;
474 			}
475 			if (req != 1 || flags == 3 ||
476 			    pte_index >= kvm->arch.hpt_npte) {
477 				/* parameter error */
478 				args[j] = ((0xa0 | flags) << 56) + pte_index;
479 				ret = H_PARAMETER;
480 				break;
481 			}
482 			hp = (__be64 *) (kvm->arch.hpt_virt + (pte_index << 4));
483 			/* to avoid deadlock, don't spin except for first */
484 			if (!try_lock_hpte(hp, HPTE_V_HVLOCK)) {
485 				if (n)
486 					break;
487 				while (!try_lock_hpte(hp, HPTE_V_HVLOCK))
488 					cpu_relax();
489 			}
490 			found = 0;
491 			hp0 = be64_to_cpu(hp[0]);
492 			if (hp0 & (HPTE_V_ABSENT | HPTE_V_VALID)) {
493 				switch (flags & 3) {
494 				case 0:		/* absolute */
495 					found = 1;
496 					break;
497 				case 1:		/* andcond */
498 					if (!(hp0 & args[j + 1]))
499 						found = 1;
500 					break;
501 				case 2:		/* AVPN */
502 					if ((hp0 & ~0x7fUL) == args[j + 1])
503 						found = 1;
504 					break;
505 				}
506 			}
507 			if (!found) {
508 				hp[0] &= ~cpu_to_be64(HPTE_V_HVLOCK);
509 				args[j] = ((0x90 | flags) << 56) + pte_index;
510 				continue;
511 			}
512 
513 			args[j] = ((0x80 | flags) << 56) + pte_index;
514 			rev = real_vmalloc_addr(&kvm->arch.revmap[pte_index]);
515 			note_hpte_modification(kvm, rev);
516 
517 			if (!(hp0 & HPTE_V_VALID)) {
518 				/* insert R and C bits from PTE */
519 				rcbits = rev->guest_rpte & (HPTE_R_R|HPTE_R_C);
520 				args[j] |= rcbits << (56 - 5);
521 				hp[0] = 0;
522 				continue;
523 			}
524 
525 			/* leave it locked */
526 			hp[0] &= ~cpu_to_be64(HPTE_V_VALID);
527 			tlbrb[n] = compute_tlbie_rb(be64_to_cpu(hp[0]),
528 				be64_to_cpu(hp[1]), pte_index);
529 			indexes[n] = j;
530 			hptes[n] = hp;
531 			revs[n] = rev;
532 			++n;
533 		}
534 
535 		if (!n)
536 			break;
537 
538 		/* Now that we've collected a batch, do the tlbies */
539 		do_tlbies(kvm, tlbrb, n, global, true);
540 
541 		/* Read PTE low words after tlbie to get final R/C values */
542 		for (k = 0; k < n; ++k) {
543 			j = indexes[k];
544 			pte_index = args[j] & ((1ul << 56) - 1);
545 			hp = hptes[k];
546 			rev = revs[k];
547 			remove_revmap_chain(kvm, pte_index, rev,
548 				be64_to_cpu(hp[0]), be64_to_cpu(hp[1]));
549 			rcbits = rev->guest_rpte & (HPTE_R_R|HPTE_R_C);
550 			args[j] |= rcbits << (56 - 5);
551 			hp[0] = 0;
552 		}
553 	}
554 
555 	return ret;
556 }
557 
558 long kvmppc_h_protect(struct kvm_vcpu *vcpu, unsigned long flags,
559 		      unsigned long pte_index, unsigned long avpn,
560 		      unsigned long va)
561 {
562 	struct kvm *kvm = vcpu->kvm;
563 	__be64 *hpte;
564 	struct revmap_entry *rev;
565 	unsigned long v, r, rb, mask, bits;
566 	u64 pte;
567 
568 	if (pte_index >= kvm->arch.hpt_npte)
569 		return H_PARAMETER;
570 
571 	hpte = (__be64 *)(kvm->arch.hpt_virt + (pte_index << 4));
572 	while (!try_lock_hpte(hpte, HPTE_V_HVLOCK))
573 		cpu_relax();
574 	pte = be64_to_cpu(hpte[0]);
575 	if ((pte & (HPTE_V_ABSENT | HPTE_V_VALID)) == 0 ||
576 	    ((flags & H_AVPN) && (pte & ~0x7fUL) != avpn)) {
577 		hpte[0] &= ~cpu_to_be64(HPTE_V_HVLOCK);
578 		return H_NOT_FOUND;
579 	}
580 
581 	v = pte;
582 	bits = (flags << 55) & HPTE_R_PP0;
583 	bits |= (flags << 48) & HPTE_R_KEY_HI;
584 	bits |= flags & (HPTE_R_PP | HPTE_R_N | HPTE_R_KEY_LO);
585 
586 	/* Update guest view of 2nd HPTE dword */
587 	mask = HPTE_R_PP0 | HPTE_R_PP | HPTE_R_N |
588 		HPTE_R_KEY_HI | HPTE_R_KEY_LO;
589 	rev = real_vmalloc_addr(&kvm->arch.revmap[pte_index]);
590 	if (rev) {
591 		r = (rev->guest_rpte & ~mask) | bits;
592 		rev->guest_rpte = r;
593 		note_hpte_modification(kvm, rev);
594 	}
595 
596 	/* Update HPTE */
597 	if (v & HPTE_V_VALID) {
598 		/*
599 		 * If the page is valid, don't let it transition from
600 		 * readonly to writable.  If it should be writable, we'll
601 		 * take a trap and let the page fault code sort it out.
602 		 */
603 		pte = be64_to_cpu(hpte[1]);
604 		r = (pte & ~mask) | bits;
605 		if (hpte_is_writable(r) && !hpte_is_writable(pte))
606 			r = hpte_make_readonly(r);
607 		/* If the PTE is changing, invalidate it first */
608 		if (r != pte) {
609 			rb = compute_tlbie_rb(v, r, pte_index);
610 			hpte[0] = cpu_to_be64((v & ~HPTE_V_VALID) |
611 					      HPTE_V_ABSENT);
612 			do_tlbies(kvm, &rb, 1, global_invalidates(kvm, flags),
613 				  true);
614 			hpte[1] = cpu_to_be64(r);
615 		}
616 	}
617 	unlock_hpte(hpte, v & ~HPTE_V_HVLOCK);
618 	asm volatile("ptesync" : : : "memory");
619 	return H_SUCCESS;
620 }
621 
622 long kvmppc_h_read(struct kvm_vcpu *vcpu, unsigned long flags,
623 		   unsigned long pte_index)
624 {
625 	struct kvm *kvm = vcpu->kvm;
626 	__be64 *hpte;
627 	unsigned long v, r;
628 	int i, n = 1;
629 	struct revmap_entry *rev = NULL;
630 
631 	if (pte_index >= kvm->arch.hpt_npte)
632 		return H_PARAMETER;
633 	if (flags & H_READ_4) {
634 		pte_index &= ~3;
635 		n = 4;
636 	}
637 	rev = real_vmalloc_addr(&kvm->arch.revmap[pte_index]);
638 	for (i = 0; i < n; ++i, ++pte_index) {
639 		hpte = (__be64 *)(kvm->arch.hpt_virt + (pte_index << 4));
640 		v = be64_to_cpu(hpte[0]) & ~HPTE_V_HVLOCK;
641 		r = be64_to_cpu(hpte[1]);
642 		if (v & HPTE_V_ABSENT) {
643 			v &= ~HPTE_V_ABSENT;
644 			v |= HPTE_V_VALID;
645 		}
646 		if (v & HPTE_V_VALID) {
647 			r = rev[i].guest_rpte | (r & (HPTE_R_R | HPTE_R_C));
648 			r &= ~HPTE_GR_RESERVED;
649 		}
650 		vcpu->arch.gpr[4 + i * 2] = v;
651 		vcpu->arch.gpr[5 + i * 2] = r;
652 	}
653 	return H_SUCCESS;
654 }
655 
656 void kvmppc_invalidate_hpte(struct kvm *kvm, __be64 *hptep,
657 			unsigned long pte_index)
658 {
659 	unsigned long rb;
660 
661 	hptep[0] &= ~cpu_to_be64(HPTE_V_VALID);
662 	rb = compute_tlbie_rb(be64_to_cpu(hptep[0]), be64_to_cpu(hptep[1]),
663 			      pte_index);
664 	do_tlbies(kvm, &rb, 1, 1, true);
665 }
666 EXPORT_SYMBOL_GPL(kvmppc_invalidate_hpte);
667 
668 void kvmppc_clear_ref_hpte(struct kvm *kvm, __be64 *hptep,
669 			   unsigned long pte_index)
670 {
671 	unsigned long rb;
672 	unsigned char rbyte;
673 
674 	rb = compute_tlbie_rb(be64_to_cpu(hptep[0]), be64_to_cpu(hptep[1]),
675 			      pte_index);
676 	rbyte = (be64_to_cpu(hptep[1]) & ~HPTE_R_R) >> 8;
677 	/* modify only the second-last byte, which contains the ref bit */
678 	*((char *)hptep + 14) = rbyte;
679 	do_tlbies(kvm, &rb, 1, 1, false);
680 }
681 EXPORT_SYMBOL_GPL(kvmppc_clear_ref_hpte);
682 
683 static int slb_base_page_shift[4] = {
684 	24,	/* 16M */
685 	16,	/* 64k */
686 	34,	/* 16G */
687 	20,	/* 1M, unsupported */
688 };
689 
690 /* When called from virtmode, this func should be protected by
691  * preempt_disable(), otherwise, the holding of HPTE_V_HVLOCK
692  * can trigger deadlock issue.
693  */
694 long kvmppc_hv_find_lock_hpte(struct kvm *kvm, gva_t eaddr, unsigned long slb_v,
695 			      unsigned long valid)
696 {
697 	unsigned int i;
698 	unsigned int pshift;
699 	unsigned long somask;
700 	unsigned long vsid, hash;
701 	unsigned long avpn;
702 	__be64 *hpte;
703 	unsigned long mask, val;
704 	unsigned long v, r;
705 
706 	/* Get page shift, work out hash and AVPN etc. */
707 	mask = SLB_VSID_B | HPTE_V_AVPN | HPTE_V_SECONDARY;
708 	val = 0;
709 	pshift = 12;
710 	if (slb_v & SLB_VSID_L) {
711 		mask |= HPTE_V_LARGE;
712 		val |= HPTE_V_LARGE;
713 		pshift = slb_base_page_shift[(slb_v & SLB_VSID_LP) >> 4];
714 	}
715 	if (slb_v & SLB_VSID_B_1T) {
716 		somask = (1UL << 40) - 1;
717 		vsid = (slb_v & ~SLB_VSID_B) >> SLB_VSID_SHIFT_1T;
718 		vsid ^= vsid << 25;
719 	} else {
720 		somask = (1UL << 28) - 1;
721 		vsid = (slb_v & ~SLB_VSID_B) >> SLB_VSID_SHIFT;
722 	}
723 	hash = (vsid ^ ((eaddr & somask) >> pshift)) & kvm->arch.hpt_mask;
724 	avpn = slb_v & ~(somask >> 16);	/* also includes B */
725 	avpn |= (eaddr & somask) >> 16;
726 
727 	if (pshift >= 24)
728 		avpn &= ~((1UL << (pshift - 16)) - 1);
729 	else
730 		avpn &= ~0x7fUL;
731 	val |= avpn;
732 
733 	for (;;) {
734 		hpte = (__be64 *)(kvm->arch.hpt_virt + (hash << 7));
735 
736 		for (i = 0; i < 16; i += 2) {
737 			/* Read the PTE racily */
738 			v = be64_to_cpu(hpte[i]) & ~HPTE_V_HVLOCK;
739 
740 			/* Check valid/absent, hash, segment size and AVPN */
741 			if (!(v & valid) || (v & mask) != val)
742 				continue;
743 
744 			/* Lock the PTE and read it under the lock */
745 			while (!try_lock_hpte(&hpte[i], HPTE_V_HVLOCK))
746 				cpu_relax();
747 			v = be64_to_cpu(hpte[i]) & ~HPTE_V_HVLOCK;
748 			r = be64_to_cpu(hpte[i+1]);
749 
750 			/*
751 			 * Check the HPTE again, including base page size
752 			 */
753 			if ((v & valid) && (v & mask) == val &&
754 			    hpte_base_page_size(v, r) == (1ul << pshift))
755 				/* Return with the HPTE still locked */
756 				return (hash << 3) + (i >> 1);
757 
758 			/* Unlock and move on */
759 			hpte[i] = cpu_to_be64(v);
760 		}
761 
762 		if (val & HPTE_V_SECONDARY)
763 			break;
764 		val |= HPTE_V_SECONDARY;
765 		hash = hash ^ kvm->arch.hpt_mask;
766 	}
767 	return -1;
768 }
769 EXPORT_SYMBOL(kvmppc_hv_find_lock_hpte);
770 
771 /*
772  * Called in real mode to check whether an HPTE not found fault
773  * is due to accessing a paged-out page or an emulated MMIO page,
774  * or if a protection fault is due to accessing a page that the
775  * guest wanted read/write access to but which we made read-only.
776  * Returns a possibly modified status (DSISR) value if not
777  * (i.e. pass the interrupt to the guest),
778  * -1 to pass the fault up to host kernel mode code, -2 to do that
779  * and also load the instruction word (for MMIO emulation),
780  * or 0 if we should make the guest retry the access.
781  */
782 long kvmppc_hpte_hv_fault(struct kvm_vcpu *vcpu, unsigned long addr,
783 			  unsigned long slb_v, unsigned int status, bool data)
784 {
785 	struct kvm *kvm = vcpu->kvm;
786 	long int index;
787 	unsigned long v, r, gr;
788 	__be64 *hpte;
789 	unsigned long valid;
790 	struct revmap_entry *rev;
791 	unsigned long pp, key;
792 
793 	/* For protection fault, expect to find a valid HPTE */
794 	valid = HPTE_V_VALID;
795 	if (status & DSISR_NOHPTE)
796 		valid |= HPTE_V_ABSENT;
797 
798 	index = kvmppc_hv_find_lock_hpte(kvm, addr, slb_v, valid);
799 	if (index < 0) {
800 		if (status & DSISR_NOHPTE)
801 			return status;	/* there really was no HPTE */
802 		return 0;		/* for prot fault, HPTE disappeared */
803 	}
804 	hpte = (__be64 *)(kvm->arch.hpt_virt + (index << 4));
805 	v = be64_to_cpu(hpte[0]) & ~HPTE_V_HVLOCK;
806 	r = be64_to_cpu(hpte[1]);
807 	rev = real_vmalloc_addr(&kvm->arch.revmap[index]);
808 	gr = rev->guest_rpte;
809 
810 	unlock_hpte(hpte, v);
811 
812 	/* For not found, if the HPTE is valid by now, retry the instruction */
813 	if ((status & DSISR_NOHPTE) && (v & HPTE_V_VALID))
814 		return 0;
815 
816 	/* Check access permissions to the page */
817 	pp = gr & (HPTE_R_PP0 | HPTE_R_PP);
818 	key = (vcpu->arch.shregs.msr & MSR_PR) ? SLB_VSID_KP : SLB_VSID_KS;
819 	status &= ~DSISR_NOHPTE;	/* DSISR_NOHPTE == SRR1_ISI_NOPT */
820 	if (!data) {
821 		if (gr & (HPTE_R_N | HPTE_R_G))
822 			return status | SRR1_ISI_N_OR_G;
823 		if (!hpte_read_permission(pp, slb_v & key))
824 			return status | SRR1_ISI_PROT;
825 	} else if (status & DSISR_ISSTORE) {
826 		/* check write permission */
827 		if (!hpte_write_permission(pp, slb_v & key))
828 			return status | DSISR_PROTFAULT;
829 	} else {
830 		if (!hpte_read_permission(pp, slb_v & key))
831 			return status | DSISR_PROTFAULT;
832 	}
833 
834 	/* Check storage key, if applicable */
835 	if (data && (vcpu->arch.shregs.msr & MSR_DR)) {
836 		unsigned int perm = hpte_get_skey_perm(gr, vcpu->arch.amr);
837 		if (status & DSISR_ISSTORE)
838 			perm >>= 1;
839 		if (perm & 1)
840 			return status | DSISR_KEYFAULT;
841 	}
842 
843 	/* Save HPTE info for virtual-mode handler */
844 	vcpu->arch.pgfault_addr = addr;
845 	vcpu->arch.pgfault_index = index;
846 	vcpu->arch.pgfault_hpte[0] = v;
847 	vcpu->arch.pgfault_hpte[1] = r;
848 
849 	/* Check the storage key to see if it is possibly emulated MMIO */
850 	if (data && (vcpu->arch.shregs.msr & MSR_IR) &&
851 	    (r & (HPTE_R_KEY_HI | HPTE_R_KEY_LO)) ==
852 	    (HPTE_R_KEY_HI | HPTE_R_KEY_LO))
853 		return -2;	/* MMIO emulation - load instr word */
854 
855 	return -1;		/* send fault up to host kernel mode */
856 }
857