xref: /openbmc/linux/arch/powerpc/kvm/book3s_hv.c (revision b9ccfda2)
1 /*
2  * Copyright 2011 Paul Mackerras, IBM Corp. <paulus@au1.ibm.com>
3  * Copyright (C) 2009. SUSE Linux Products GmbH. All rights reserved.
4  *
5  * Authors:
6  *    Paul Mackerras <paulus@au1.ibm.com>
7  *    Alexander Graf <agraf@suse.de>
8  *    Kevin Wolf <mail@kevin-wolf.de>
9  *
10  * Description: KVM functions specific to running on Book 3S
11  * processors in hypervisor mode (specifically POWER7 and later).
12  *
13  * This file is derived from arch/powerpc/kvm/book3s.c,
14  * by Alexander Graf <agraf@suse.de>.
15  *
16  * This program is free software; you can redistribute it and/or modify
17  * it under the terms of the GNU General Public License, version 2, as
18  * published by the Free Software Foundation.
19  */
20 
21 #include <linux/kvm_host.h>
22 #include <linux/err.h>
23 #include <linux/slab.h>
24 #include <linux/preempt.h>
25 #include <linux/sched.h>
26 #include <linux/delay.h>
27 #include <linux/export.h>
28 #include <linux/fs.h>
29 #include <linux/anon_inodes.h>
30 #include <linux/cpumask.h>
31 #include <linux/spinlock.h>
32 #include <linux/page-flags.h>
33 
34 #include <asm/reg.h>
35 #include <asm/cputable.h>
36 #include <asm/cacheflush.h>
37 #include <asm/tlbflush.h>
38 #include <asm/uaccess.h>
39 #include <asm/io.h>
40 #include <asm/kvm_ppc.h>
41 #include <asm/kvm_book3s.h>
42 #include <asm/mmu_context.h>
43 #include <asm/lppaca.h>
44 #include <asm/processor.h>
45 #include <asm/cputhreads.h>
46 #include <asm/page.h>
47 #include <asm/hvcall.h>
48 #include <asm/switch_to.h>
49 #include <linux/gfp.h>
50 #include <linux/vmalloc.h>
51 #include <linux/highmem.h>
52 #include <linux/hugetlb.h>
53 
54 /* #define EXIT_DEBUG */
55 /* #define EXIT_DEBUG_SIMPLE */
56 /* #define EXIT_DEBUG_INT */
57 
58 static void kvmppc_end_cede(struct kvm_vcpu *vcpu);
59 static int kvmppc_hv_setup_htab_rma(struct kvm_vcpu *vcpu);
60 
61 void kvmppc_core_vcpu_load(struct kvm_vcpu *vcpu, int cpu)
62 {
63 	struct kvmppc_vcore *vc = vcpu->arch.vcore;
64 
65 	local_paca->kvm_hstate.kvm_vcpu = vcpu;
66 	local_paca->kvm_hstate.kvm_vcore = vc;
67 	if (vc->runner == vcpu && vc->vcore_state != VCORE_INACTIVE)
68 		vc->stolen_tb += mftb() - vc->preempt_tb;
69 }
70 
71 void kvmppc_core_vcpu_put(struct kvm_vcpu *vcpu)
72 {
73 	struct kvmppc_vcore *vc = vcpu->arch.vcore;
74 
75 	if (vc->runner == vcpu && vc->vcore_state != VCORE_INACTIVE)
76 		vc->preempt_tb = mftb();
77 }
78 
79 void kvmppc_set_msr(struct kvm_vcpu *vcpu, u64 msr)
80 {
81 	vcpu->arch.shregs.msr = msr;
82 	kvmppc_end_cede(vcpu);
83 }
84 
85 void kvmppc_set_pvr(struct kvm_vcpu *vcpu, u32 pvr)
86 {
87 	vcpu->arch.pvr = pvr;
88 }
89 
90 void kvmppc_dump_regs(struct kvm_vcpu *vcpu)
91 {
92 	int r;
93 
94 	pr_err("vcpu %p (%d):\n", vcpu, vcpu->vcpu_id);
95 	pr_err("pc  = %.16lx  msr = %.16llx  trap = %x\n",
96 	       vcpu->arch.pc, vcpu->arch.shregs.msr, vcpu->arch.trap);
97 	for (r = 0; r < 16; ++r)
98 		pr_err("r%2d = %.16lx  r%d = %.16lx\n",
99 		       r, kvmppc_get_gpr(vcpu, r),
100 		       r+16, kvmppc_get_gpr(vcpu, r+16));
101 	pr_err("ctr = %.16lx  lr  = %.16lx\n",
102 	       vcpu->arch.ctr, vcpu->arch.lr);
103 	pr_err("srr0 = %.16llx srr1 = %.16llx\n",
104 	       vcpu->arch.shregs.srr0, vcpu->arch.shregs.srr1);
105 	pr_err("sprg0 = %.16llx sprg1 = %.16llx\n",
106 	       vcpu->arch.shregs.sprg0, vcpu->arch.shregs.sprg1);
107 	pr_err("sprg2 = %.16llx sprg3 = %.16llx\n",
108 	       vcpu->arch.shregs.sprg2, vcpu->arch.shregs.sprg3);
109 	pr_err("cr = %.8x  xer = %.16lx  dsisr = %.8x\n",
110 	       vcpu->arch.cr, vcpu->arch.xer, vcpu->arch.shregs.dsisr);
111 	pr_err("dar = %.16llx\n", vcpu->arch.shregs.dar);
112 	pr_err("fault dar = %.16lx dsisr = %.8x\n",
113 	       vcpu->arch.fault_dar, vcpu->arch.fault_dsisr);
114 	pr_err("SLB (%d entries):\n", vcpu->arch.slb_max);
115 	for (r = 0; r < vcpu->arch.slb_max; ++r)
116 		pr_err("  ESID = %.16llx VSID = %.16llx\n",
117 		       vcpu->arch.slb[r].orige, vcpu->arch.slb[r].origv);
118 	pr_err("lpcr = %.16lx sdr1 = %.16lx last_inst = %.8x\n",
119 	       vcpu->kvm->arch.lpcr, vcpu->kvm->arch.sdr1,
120 	       vcpu->arch.last_inst);
121 }
122 
123 struct kvm_vcpu *kvmppc_find_vcpu(struct kvm *kvm, int id)
124 {
125 	int r;
126 	struct kvm_vcpu *v, *ret = NULL;
127 
128 	mutex_lock(&kvm->lock);
129 	kvm_for_each_vcpu(r, v, kvm) {
130 		if (v->vcpu_id == id) {
131 			ret = v;
132 			break;
133 		}
134 	}
135 	mutex_unlock(&kvm->lock);
136 	return ret;
137 }
138 
139 static void init_vpa(struct kvm_vcpu *vcpu, struct lppaca *vpa)
140 {
141 	vpa->shared_proc = 1;
142 	vpa->yield_count = 1;
143 }
144 
145 /* Length for a per-processor buffer is passed in at offset 4 in the buffer */
146 struct reg_vpa {
147 	u32 dummy;
148 	union {
149 		u16 hword;
150 		u32 word;
151 	} length;
152 };
153 
154 static int vpa_is_registered(struct kvmppc_vpa *vpap)
155 {
156 	if (vpap->update_pending)
157 		return vpap->next_gpa != 0;
158 	return vpap->pinned_addr != NULL;
159 }
160 
161 static unsigned long do_h_register_vpa(struct kvm_vcpu *vcpu,
162 				       unsigned long flags,
163 				       unsigned long vcpuid, unsigned long vpa)
164 {
165 	struct kvm *kvm = vcpu->kvm;
166 	unsigned long len, nb;
167 	void *va;
168 	struct kvm_vcpu *tvcpu;
169 	int err;
170 	int subfunc;
171 	struct kvmppc_vpa *vpap;
172 
173 	tvcpu = kvmppc_find_vcpu(kvm, vcpuid);
174 	if (!tvcpu)
175 		return H_PARAMETER;
176 
177 	subfunc = (flags >> H_VPA_FUNC_SHIFT) & H_VPA_FUNC_MASK;
178 	if (subfunc == H_VPA_REG_VPA || subfunc == H_VPA_REG_DTL ||
179 	    subfunc == H_VPA_REG_SLB) {
180 		/* Registering new area - address must be cache-line aligned */
181 		if ((vpa & (L1_CACHE_BYTES - 1)) || !vpa)
182 			return H_PARAMETER;
183 
184 		/* convert logical addr to kernel addr and read length */
185 		va = kvmppc_pin_guest_page(kvm, vpa, &nb);
186 		if (va == NULL)
187 			return H_PARAMETER;
188 		if (subfunc == H_VPA_REG_VPA)
189 			len = ((struct reg_vpa *)va)->length.hword;
190 		else
191 			len = ((struct reg_vpa *)va)->length.word;
192 		kvmppc_unpin_guest_page(kvm, va);
193 
194 		/* Check length */
195 		if (len > nb || len < sizeof(struct reg_vpa))
196 			return H_PARAMETER;
197 	} else {
198 		vpa = 0;
199 		len = 0;
200 	}
201 
202 	err = H_PARAMETER;
203 	vpap = NULL;
204 	spin_lock(&tvcpu->arch.vpa_update_lock);
205 
206 	switch (subfunc) {
207 	case H_VPA_REG_VPA:		/* register VPA */
208 		if (len < sizeof(struct lppaca))
209 			break;
210 		vpap = &tvcpu->arch.vpa;
211 		err = 0;
212 		break;
213 
214 	case H_VPA_REG_DTL:		/* register DTL */
215 		if (len < sizeof(struct dtl_entry))
216 			break;
217 		len -= len % sizeof(struct dtl_entry);
218 
219 		/* Check that they have previously registered a VPA */
220 		err = H_RESOURCE;
221 		if (!vpa_is_registered(&tvcpu->arch.vpa))
222 			break;
223 
224 		vpap = &tvcpu->arch.dtl;
225 		err = 0;
226 		break;
227 
228 	case H_VPA_REG_SLB:		/* register SLB shadow buffer */
229 		/* Check that they have previously registered a VPA */
230 		err = H_RESOURCE;
231 		if (!vpa_is_registered(&tvcpu->arch.vpa))
232 			break;
233 
234 		vpap = &tvcpu->arch.slb_shadow;
235 		err = 0;
236 		break;
237 
238 	case H_VPA_DEREG_VPA:		/* deregister VPA */
239 		/* Check they don't still have a DTL or SLB buf registered */
240 		err = H_RESOURCE;
241 		if (vpa_is_registered(&tvcpu->arch.dtl) ||
242 		    vpa_is_registered(&tvcpu->arch.slb_shadow))
243 			break;
244 
245 		vpap = &tvcpu->arch.vpa;
246 		err = 0;
247 		break;
248 
249 	case H_VPA_DEREG_DTL:		/* deregister DTL */
250 		vpap = &tvcpu->arch.dtl;
251 		err = 0;
252 		break;
253 
254 	case H_VPA_DEREG_SLB:		/* deregister SLB shadow buffer */
255 		vpap = &tvcpu->arch.slb_shadow;
256 		err = 0;
257 		break;
258 	}
259 
260 	if (vpap) {
261 		vpap->next_gpa = vpa;
262 		vpap->len = len;
263 		vpap->update_pending = 1;
264 	}
265 
266 	spin_unlock(&tvcpu->arch.vpa_update_lock);
267 
268 	return err;
269 }
270 
271 static void kvmppc_update_vpa(struct kvm_vcpu *vcpu, struct kvmppc_vpa *vpap)
272 {
273 	struct kvm *kvm = vcpu->kvm;
274 	void *va;
275 	unsigned long nb;
276 	unsigned long gpa;
277 
278 	/*
279 	 * We need to pin the page pointed to by vpap->next_gpa,
280 	 * but we can't call kvmppc_pin_guest_page under the lock
281 	 * as it does get_user_pages() and down_read().  So we
282 	 * have to drop the lock, pin the page, then get the lock
283 	 * again and check that a new area didn't get registered
284 	 * in the meantime.
285 	 */
286 	for (;;) {
287 		gpa = vpap->next_gpa;
288 		spin_unlock(&vcpu->arch.vpa_update_lock);
289 		va = NULL;
290 		nb = 0;
291 		if (gpa)
292 			va = kvmppc_pin_guest_page(kvm, vpap->next_gpa, &nb);
293 		spin_lock(&vcpu->arch.vpa_update_lock);
294 		if (gpa == vpap->next_gpa)
295 			break;
296 		/* sigh... unpin that one and try again */
297 		if (va)
298 			kvmppc_unpin_guest_page(kvm, va);
299 	}
300 
301 	vpap->update_pending = 0;
302 	if (va && nb < vpap->len) {
303 		/*
304 		 * If it's now too short, it must be that userspace
305 		 * has changed the mappings underlying guest memory,
306 		 * so unregister the region.
307 		 */
308 		kvmppc_unpin_guest_page(kvm, va);
309 		va = NULL;
310 	}
311 	if (vpap->pinned_addr)
312 		kvmppc_unpin_guest_page(kvm, vpap->pinned_addr);
313 	vpap->pinned_addr = va;
314 	if (va)
315 		vpap->pinned_end = va + vpap->len;
316 }
317 
318 static void kvmppc_update_vpas(struct kvm_vcpu *vcpu)
319 {
320 	spin_lock(&vcpu->arch.vpa_update_lock);
321 	if (vcpu->arch.vpa.update_pending) {
322 		kvmppc_update_vpa(vcpu, &vcpu->arch.vpa);
323 		init_vpa(vcpu, vcpu->arch.vpa.pinned_addr);
324 	}
325 	if (vcpu->arch.dtl.update_pending) {
326 		kvmppc_update_vpa(vcpu, &vcpu->arch.dtl);
327 		vcpu->arch.dtl_ptr = vcpu->arch.dtl.pinned_addr;
328 		vcpu->arch.dtl_index = 0;
329 	}
330 	if (vcpu->arch.slb_shadow.update_pending)
331 		kvmppc_update_vpa(vcpu, &vcpu->arch.slb_shadow);
332 	spin_unlock(&vcpu->arch.vpa_update_lock);
333 }
334 
335 static void kvmppc_create_dtl_entry(struct kvm_vcpu *vcpu,
336 				    struct kvmppc_vcore *vc)
337 {
338 	struct dtl_entry *dt;
339 	struct lppaca *vpa;
340 	unsigned long old_stolen;
341 
342 	dt = vcpu->arch.dtl_ptr;
343 	vpa = vcpu->arch.vpa.pinned_addr;
344 	old_stolen = vcpu->arch.stolen_logged;
345 	vcpu->arch.stolen_logged = vc->stolen_tb;
346 	if (!dt || !vpa)
347 		return;
348 	memset(dt, 0, sizeof(struct dtl_entry));
349 	dt->dispatch_reason = 7;
350 	dt->processor_id = vc->pcpu + vcpu->arch.ptid;
351 	dt->timebase = mftb();
352 	dt->enqueue_to_dispatch_time = vc->stolen_tb - old_stolen;
353 	dt->srr0 = kvmppc_get_pc(vcpu);
354 	dt->srr1 = vcpu->arch.shregs.msr;
355 	++dt;
356 	if (dt == vcpu->arch.dtl.pinned_end)
357 		dt = vcpu->arch.dtl.pinned_addr;
358 	vcpu->arch.dtl_ptr = dt;
359 	/* order writing *dt vs. writing vpa->dtl_idx */
360 	smp_wmb();
361 	vpa->dtl_idx = ++vcpu->arch.dtl_index;
362 }
363 
364 int kvmppc_pseries_do_hcall(struct kvm_vcpu *vcpu)
365 {
366 	unsigned long req = kvmppc_get_gpr(vcpu, 3);
367 	unsigned long target, ret = H_SUCCESS;
368 	struct kvm_vcpu *tvcpu;
369 
370 	switch (req) {
371 	case H_ENTER:
372 		ret = kvmppc_virtmode_h_enter(vcpu, kvmppc_get_gpr(vcpu, 4),
373 					      kvmppc_get_gpr(vcpu, 5),
374 					      kvmppc_get_gpr(vcpu, 6),
375 					      kvmppc_get_gpr(vcpu, 7));
376 		break;
377 	case H_CEDE:
378 		break;
379 	case H_PROD:
380 		target = kvmppc_get_gpr(vcpu, 4);
381 		tvcpu = kvmppc_find_vcpu(vcpu->kvm, target);
382 		if (!tvcpu) {
383 			ret = H_PARAMETER;
384 			break;
385 		}
386 		tvcpu->arch.prodded = 1;
387 		smp_mb();
388 		if (vcpu->arch.ceded) {
389 			if (waitqueue_active(&vcpu->wq)) {
390 				wake_up_interruptible(&vcpu->wq);
391 				vcpu->stat.halt_wakeup++;
392 			}
393 		}
394 		break;
395 	case H_CONFER:
396 		break;
397 	case H_REGISTER_VPA:
398 		ret = do_h_register_vpa(vcpu, kvmppc_get_gpr(vcpu, 4),
399 					kvmppc_get_gpr(vcpu, 5),
400 					kvmppc_get_gpr(vcpu, 6));
401 		break;
402 	default:
403 		return RESUME_HOST;
404 	}
405 	kvmppc_set_gpr(vcpu, 3, ret);
406 	vcpu->arch.hcall_needed = 0;
407 	return RESUME_GUEST;
408 }
409 
410 static int kvmppc_handle_exit(struct kvm_run *run, struct kvm_vcpu *vcpu,
411 			      struct task_struct *tsk)
412 {
413 	int r = RESUME_HOST;
414 
415 	vcpu->stat.sum_exits++;
416 
417 	run->exit_reason = KVM_EXIT_UNKNOWN;
418 	run->ready_for_interrupt_injection = 1;
419 	switch (vcpu->arch.trap) {
420 	/* We're good on these - the host merely wanted to get our attention */
421 	case BOOK3S_INTERRUPT_HV_DECREMENTER:
422 		vcpu->stat.dec_exits++;
423 		r = RESUME_GUEST;
424 		break;
425 	case BOOK3S_INTERRUPT_EXTERNAL:
426 		vcpu->stat.ext_intr_exits++;
427 		r = RESUME_GUEST;
428 		break;
429 	case BOOK3S_INTERRUPT_PERFMON:
430 		r = RESUME_GUEST;
431 		break;
432 	case BOOK3S_INTERRUPT_PROGRAM:
433 	{
434 		ulong flags;
435 		/*
436 		 * Normally program interrupts are delivered directly
437 		 * to the guest by the hardware, but we can get here
438 		 * as a result of a hypervisor emulation interrupt
439 		 * (e40) getting turned into a 700 by BML RTAS.
440 		 */
441 		flags = vcpu->arch.shregs.msr & 0x1f0000ull;
442 		kvmppc_core_queue_program(vcpu, flags);
443 		r = RESUME_GUEST;
444 		break;
445 	}
446 	case BOOK3S_INTERRUPT_SYSCALL:
447 	{
448 		/* hcall - punt to userspace */
449 		int i;
450 
451 		if (vcpu->arch.shregs.msr & MSR_PR) {
452 			/* sc 1 from userspace - reflect to guest syscall */
453 			kvmppc_book3s_queue_irqprio(vcpu, BOOK3S_INTERRUPT_SYSCALL);
454 			r = RESUME_GUEST;
455 			break;
456 		}
457 		run->papr_hcall.nr = kvmppc_get_gpr(vcpu, 3);
458 		for (i = 0; i < 9; ++i)
459 			run->papr_hcall.args[i] = kvmppc_get_gpr(vcpu, 4 + i);
460 		run->exit_reason = KVM_EXIT_PAPR_HCALL;
461 		vcpu->arch.hcall_needed = 1;
462 		r = RESUME_HOST;
463 		break;
464 	}
465 	/*
466 	 * We get these next two if the guest accesses a page which it thinks
467 	 * it has mapped but which is not actually present, either because
468 	 * it is for an emulated I/O device or because the corresonding
469 	 * host page has been paged out.  Any other HDSI/HISI interrupts
470 	 * have been handled already.
471 	 */
472 	case BOOK3S_INTERRUPT_H_DATA_STORAGE:
473 		r = kvmppc_book3s_hv_page_fault(run, vcpu,
474 				vcpu->arch.fault_dar, vcpu->arch.fault_dsisr);
475 		break;
476 	case BOOK3S_INTERRUPT_H_INST_STORAGE:
477 		r = kvmppc_book3s_hv_page_fault(run, vcpu,
478 				kvmppc_get_pc(vcpu), 0);
479 		break;
480 	/*
481 	 * This occurs if the guest executes an illegal instruction.
482 	 * We just generate a program interrupt to the guest, since
483 	 * we don't emulate any guest instructions at this stage.
484 	 */
485 	case BOOK3S_INTERRUPT_H_EMUL_ASSIST:
486 		kvmppc_core_queue_program(vcpu, 0x80000);
487 		r = RESUME_GUEST;
488 		break;
489 	default:
490 		kvmppc_dump_regs(vcpu);
491 		printk(KERN_EMERG "trap=0x%x | pc=0x%lx | msr=0x%llx\n",
492 			vcpu->arch.trap, kvmppc_get_pc(vcpu),
493 			vcpu->arch.shregs.msr);
494 		r = RESUME_HOST;
495 		BUG();
496 		break;
497 	}
498 
499 	return r;
500 }
501 
502 int kvm_arch_vcpu_ioctl_get_sregs(struct kvm_vcpu *vcpu,
503                                   struct kvm_sregs *sregs)
504 {
505 	int i;
506 
507 	sregs->pvr = vcpu->arch.pvr;
508 
509 	memset(sregs, 0, sizeof(struct kvm_sregs));
510 	for (i = 0; i < vcpu->arch.slb_max; i++) {
511 		sregs->u.s.ppc64.slb[i].slbe = vcpu->arch.slb[i].orige;
512 		sregs->u.s.ppc64.slb[i].slbv = vcpu->arch.slb[i].origv;
513 	}
514 
515 	return 0;
516 }
517 
518 int kvm_arch_vcpu_ioctl_set_sregs(struct kvm_vcpu *vcpu,
519                                   struct kvm_sregs *sregs)
520 {
521 	int i, j;
522 
523 	kvmppc_set_pvr(vcpu, sregs->pvr);
524 
525 	j = 0;
526 	for (i = 0; i < vcpu->arch.slb_nr; i++) {
527 		if (sregs->u.s.ppc64.slb[i].slbe & SLB_ESID_V) {
528 			vcpu->arch.slb[j].orige = sregs->u.s.ppc64.slb[i].slbe;
529 			vcpu->arch.slb[j].origv = sregs->u.s.ppc64.slb[i].slbv;
530 			++j;
531 		}
532 	}
533 	vcpu->arch.slb_max = j;
534 
535 	return 0;
536 }
537 
538 int kvm_vcpu_ioctl_get_one_reg(struct kvm_vcpu *vcpu, struct kvm_one_reg *reg)
539 {
540 	int r = -EINVAL;
541 
542 	switch (reg->id) {
543 	case KVM_REG_PPC_HIOR:
544 		r = put_user(0, (u64 __user *)reg->addr);
545 		break;
546 	default:
547 		break;
548 	}
549 
550 	return r;
551 }
552 
553 int kvm_vcpu_ioctl_set_one_reg(struct kvm_vcpu *vcpu, struct kvm_one_reg *reg)
554 {
555 	int r = -EINVAL;
556 
557 	switch (reg->id) {
558 	case KVM_REG_PPC_HIOR:
559 	{
560 		u64 hior;
561 		/* Only allow this to be set to zero */
562 		r = get_user(hior, (u64 __user *)reg->addr);
563 		if (!r && (hior != 0))
564 			r = -EINVAL;
565 		break;
566 	}
567 	default:
568 		break;
569 	}
570 
571 	return r;
572 }
573 
574 int kvmppc_core_check_processor_compat(void)
575 {
576 	if (cpu_has_feature(CPU_FTR_HVMODE))
577 		return 0;
578 	return -EIO;
579 }
580 
581 struct kvm_vcpu *kvmppc_core_vcpu_create(struct kvm *kvm, unsigned int id)
582 {
583 	struct kvm_vcpu *vcpu;
584 	int err = -EINVAL;
585 	int core;
586 	struct kvmppc_vcore *vcore;
587 
588 	core = id / threads_per_core;
589 	if (core >= KVM_MAX_VCORES)
590 		goto out;
591 
592 	err = -ENOMEM;
593 	vcpu = kmem_cache_zalloc(kvm_vcpu_cache, GFP_KERNEL);
594 	if (!vcpu)
595 		goto out;
596 
597 	err = kvm_vcpu_init(vcpu, kvm, id);
598 	if (err)
599 		goto free_vcpu;
600 
601 	vcpu->arch.shared = &vcpu->arch.shregs;
602 	vcpu->arch.last_cpu = -1;
603 	vcpu->arch.mmcr[0] = MMCR0_FC;
604 	vcpu->arch.ctrl = CTRL_RUNLATCH;
605 	/* default to host PVR, since we can't spoof it */
606 	vcpu->arch.pvr = mfspr(SPRN_PVR);
607 	kvmppc_set_pvr(vcpu, vcpu->arch.pvr);
608 	spin_lock_init(&vcpu->arch.vpa_update_lock);
609 
610 	kvmppc_mmu_book3s_hv_init(vcpu);
611 
612 	/*
613 	 * We consider the vcpu stopped until we see the first run ioctl for it.
614 	 */
615 	vcpu->arch.state = KVMPPC_VCPU_STOPPED;
616 
617 	init_waitqueue_head(&vcpu->arch.cpu_run);
618 
619 	mutex_lock(&kvm->lock);
620 	vcore = kvm->arch.vcores[core];
621 	if (!vcore) {
622 		vcore = kzalloc(sizeof(struct kvmppc_vcore), GFP_KERNEL);
623 		if (vcore) {
624 			INIT_LIST_HEAD(&vcore->runnable_threads);
625 			spin_lock_init(&vcore->lock);
626 			init_waitqueue_head(&vcore->wq);
627 			vcore->preempt_tb = mftb();
628 		}
629 		kvm->arch.vcores[core] = vcore;
630 	}
631 	mutex_unlock(&kvm->lock);
632 
633 	if (!vcore)
634 		goto free_vcpu;
635 
636 	spin_lock(&vcore->lock);
637 	++vcore->num_threads;
638 	spin_unlock(&vcore->lock);
639 	vcpu->arch.vcore = vcore;
640 	vcpu->arch.stolen_logged = vcore->stolen_tb;
641 
642 	vcpu->arch.cpu_type = KVM_CPU_3S_64;
643 	kvmppc_sanity_check(vcpu);
644 
645 	return vcpu;
646 
647 free_vcpu:
648 	kmem_cache_free(kvm_vcpu_cache, vcpu);
649 out:
650 	return ERR_PTR(err);
651 }
652 
653 void kvmppc_core_vcpu_free(struct kvm_vcpu *vcpu)
654 {
655 	spin_lock(&vcpu->arch.vpa_update_lock);
656 	if (vcpu->arch.dtl.pinned_addr)
657 		kvmppc_unpin_guest_page(vcpu->kvm, vcpu->arch.dtl.pinned_addr);
658 	if (vcpu->arch.slb_shadow.pinned_addr)
659 		kvmppc_unpin_guest_page(vcpu->kvm, vcpu->arch.slb_shadow.pinned_addr);
660 	if (vcpu->arch.vpa.pinned_addr)
661 		kvmppc_unpin_guest_page(vcpu->kvm, vcpu->arch.vpa.pinned_addr);
662 	spin_unlock(&vcpu->arch.vpa_update_lock);
663 	kvm_vcpu_uninit(vcpu);
664 	kmem_cache_free(kvm_vcpu_cache, vcpu);
665 }
666 
667 static void kvmppc_set_timer(struct kvm_vcpu *vcpu)
668 {
669 	unsigned long dec_nsec, now;
670 
671 	now = get_tb();
672 	if (now > vcpu->arch.dec_expires) {
673 		/* decrementer has already gone negative */
674 		kvmppc_core_queue_dec(vcpu);
675 		kvmppc_core_prepare_to_enter(vcpu);
676 		return;
677 	}
678 	dec_nsec = (vcpu->arch.dec_expires - now) * NSEC_PER_SEC
679 		   / tb_ticks_per_sec;
680 	hrtimer_start(&vcpu->arch.dec_timer, ktime_set(0, dec_nsec),
681 		      HRTIMER_MODE_REL);
682 	vcpu->arch.timer_running = 1;
683 }
684 
685 static void kvmppc_end_cede(struct kvm_vcpu *vcpu)
686 {
687 	vcpu->arch.ceded = 0;
688 	if (vcpu->arch.timer_running) {
689 		hrtimer_try_to_cancel(&vcpu->arch.dec_timer);
690 		vcpu->arch.timer_running = 0;
691 	}
692 }
693 
694 extern int __kvmppc_vcore_entry(struct kvm_run *kvm_run, struct kvm_vcpu *vcpu);
695 extern void xics_wake_cpu(int cpu);
696 
697 static void kvmppc_remove_runnable(struct kvmppc_vcore *vc,
698 				   struct kvm_vcpu *vcpu)
699 {
700 	struct kvm_vcpu *v;
701 
702 	if (vcpu->arch.state != KVMPPC_VCPU_RUNNABLE)
703 		return;
704 	vcpu->arch.state = KVMPPC_VCPU_BUSY_IN_HOST;
705 	--vc->n_runnable;
706 	++vc->n_busy;
707 	/* decrement the physical thread id of each following vcpu */
708 	v = vcpu;
709 	list_for_each_entry_continue(v, &vc->runnable_threads, arch.run_list)
710 		--v->arch.ptid;
711 	list_del(&vcpu->arch.run_list);
712 }
713 
714 static int kvmppc_grab_hwthread(int cpu)
715 {
716 	struct paca_struct *tpaca;
717 	long timeout = 1000;
718 
719 	tpaca = &paca[cpu];
720 
721 	/* Ensure the thread won't go into the kernel if it wakes */
722 	tpaca->kvm_hstate.hwthread_req = 1;
723 
724 	/*
725 	 * If the thread is already executing in the kernel (e.g. handling
726 	 * a stray interrupt), wait for it to get back to nap mode.
727 	 * The smp_mb() is to ensure that our setting of hwthread_req
728 	 * is visible before we look at hwthread_state, so if this
729 	 * races with the code at system_reset_pSeries and the thread
730 	 * misses our setting of hwthread_req, we are sure to see its
731 	 * setting of hwthread_state, and vice versa.
732 	 */
733 	smp_mb();
734 	while (tpaca->kvm_hstate.hwthread_state == KVM_HWTHREAD_IN_KERNEL) {
735 		if (--timeout <= 0) {
736 			pr_err("KVM: couldn't grab cpu %d\n", cpu);
737 			return -EBUSY;
738 		}
739 		udelay(1);
740 	}
741 	return 0;
742 }
743 
744 static void kvmppc_release_hwthread(int cpu)
745 {
746 	struct paca_struct *tpaca;
747 
748 	tpaca = &paca[cpu];
749 	tpaca->kvm_hstate.hwthread_req = 0;
750 	tpaca->kvm_hstate.kvm_vcpu = NULL;
751 }
752 
753 static void kvmppc_start_thread(struct kvm_vcpu *vcpu)
754 {
755 	int cpu;
756 	struct paca_struct *tpaca;
757 	struct kvmppc_vcore *vc = vcpu->arch.vcore;
758 
759 	if (vcpu->arch.timer_running) {
760 		hrtimer_try_to_cancel(&vcpu->arch.dec_timer);
761 		vcpu->arch.timer_running = 0;
762 	}
763 	cpu = vc->pcpu + vcpu->arch.ptid;
764 	tpaca = &paca[cpu];
765 	tpaca->kvm_hstate.kvm_vcpu = vcpu;
766 	tpaca->kvm_hstate.kvm_vcore = vc;
767 	tpaca->kvm_hstate.napping = 0;
768 	vcpu->cpu = vc->pcpu;
769 	smp_wmb();
770 #if defined(CONFIG_PPC_ICP_NATIVE) && defined(CONFIG_SMP)
771 	if (vcpu->arch.ptid) {
772 		kvmppc_grab_hwthread(cpu);
773 		xics_wake_cpu(cpu);
774 		++vc->n_woken;
775 	}
776 #endif
777 }
778 
779 static void kvmppc_wait_for_nap(struct kvmppc_vcore *vc)
780 {
781 	int i;
782 
783 	HMT_low();
784 	i = 0;
785 	while (vc->nap_count < vc->n_woken) {
786 		if (++i >= 1000000) {
787 			pr_err("kvmppc_wait_for_nap timeout %d %d\n",
788 			       vc->nap_count, vc->n_woken);
789 			break;
790 		}
791 		cpu_relax();
792 	}
793 	HMT_medium();
794 }
795 
796 /*
797  * Check that we are on thread 0 and that any other threads in
798  * this core are off-line.
799  */
800 static int on_primary_thread(void)
801 {
802 	int cpu = smp_processor_id();
803 	int thr = cpu_thread_in_core(cpu);
804 
805 	if (thr)
806 		return 0;
807 	while (++thr < threads_per_core)
808 		if (cpu_online(cpu + thr))
809 			return 0;
810 	return 1;
811 }
812 
813 /*
814  * Run a set of guest threads on a physical core.
815  * Called with vc->lock held.
816  */
817 static int kvmppc_run_core(struct kvmppc_vcore *vc)
818 {
819 	struct kvm_vcpu *vcpu, *vcpu0, *vnext;
820 	long ret;
821 	u64 now;
822 	int ptid, i, need_vpa_update;
823 
824 	/* don't start if any threads have a signal pending */
825 	need_vpa_update = 0;
826 	list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
827 		if (signal_pending(vcpu->arch.run_task))
828 			return 0;
829 		need_vpa_update |= vcpu->arch.vpa.update_pending |
830 			vcpu->arch.slb_shadow.update_pending |
831 			vcpu->arch.dtl.update_pending;
832 	}
833 
834 	/*
835 	 * Initialize *vc, in particular vc->vcore_state, so we can
836 	 * drop the vcore lock if necessary.
837 	 */
838 	vc->n_woken = 0;
839 	vc->nap_count = 0;
840 	vc->entry_exit_count = 0;
841 	vc->vcore_state = VCORE_RUNNING;
842 	vc->in_guest = 0;
843 	vc->napping_threads = 0;
844 
845 	/*
846 	 * Updating any of the vpas requires calling kvmppc_pin_guest_page,
847 	 * which can't be called with any spinlocks held.
848 	 */
849 	if (need_vpa_update) {
850 		spin_unlock(&vc->lock);
851 		list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list)
852 			kvmppc_update_vpas(vcpu);
853 		spin_lock(&vc->lock);
854 	}
855 
856 	/*
857 	 * Make sure we are running on thread 0, and that
858 	 * secondary threads are offline.
859 	 * XXX we should also block attempts to bring any
860 	 * secondary threads online.
861 	 */
862 	if (threads_per_core > 1 && !on_primary_thread()) {
863 		list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list)
864 			vcpu->arch.ret = -EBUSY;
865 		goto out;
866 	}
867 
868 	/*
869 	 * Assign physical thread IDs, first to non-ceded vcpus
870 	 * and then to ceded ones.
871 	 */
872 	ptid = 0;
873 	vcpu0 = NULL;
874 	list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
875 		if (!vcpu->arch.ceded) {
876 			if (!ptid)
877 				vcpu0 = vcpu;
878 			vcpu->arch.ptid = ptid++;
879 		}
880 	}
881 	if (!vcpu0)
882 		return 0;		/* nothing to run */
883 	list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list)
884 		if (vcpu->arch.ceded)
885 			vcpu->arch.ptid = ptid++;
886 
887 	vc->stolen_tb += mftb() - vc->preempt_tb;
888 	vc->pcpu = smp_processor_id();
889 	list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
890 		kvmppc_start_thread(vcpu);
891 		kvmppc_create_dtl_entry(vcpu, vc);
892 	}
893 	/* Grab any remaining hw threads so they can't go into the kernel */
894 	for (i = ptid; i < threads_per_core; ++i)
895 		kvmppc_grab_hwthread(vc->pcpu + i);
896 
897 	preempt_disable();
898 	spin_unlock(&vc->lock);
899 
900 	kvm_guest_enter();
901 	__kvmppc_vcore_entry(NULL, vcpu0);
902 	for (i = 0; i < threads_per_core; ++i)
903 		kvmppc_release_hwthread(vc->pcpu + i);
904 
905 	spin_lock(&vc->lock);
906 	/* disable sending of IPIs on virtual external irqs */
907 	list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list)
908 		vcpu->cpu = -1;
909 	/* wait for secondary threads to finish writing their state to memory */
910 	if (vc->nap_count < vc->n_woken)
911 		kvmppc_wait_for_nap(vc);
912 	/* prevent other vcpu threads from doing kvmppc_start_thread() now */
913 	vc->vcore_state = VCORE_EXITING;
914 	spin_unlock(&vc->lock);
915 
916 	/* make sure updates to secondary vcpu structs are visible now */
917 	smp_mb();
918 	kvm_guest_exit();
919 
920 	preempt_enable();
921 	kvm_resched(vcpu);
922 
923 	now = get_tb();
924 	list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
925 		/* cancel pending dec exception if dec is positive */
926 		if (now < vcpu->arch.dec_expires &&
927 		    kvmppc_core_pending_dec(vcpu))
928 			kvmppc_core_dequeue_dec(vcpu);
929 
930 		ret = RESUME_GUEST;
931 		if (vcpu->arch.trap)
932 			ret = kvmppc_handle_exit(vcpu->arch.kvm_run, vcpu,
933 						 vcpu->arch.run_task);
934 
935 		vcpu->arch.ret = ret;
936 		vcpu->arch.trap = 0;
937 
938 		if (vcpu->arch.ceded) {
939 			if (ret != RESUME_GUEST)
940 				kvmppc_end_cede(vcpu);
941 			else
942 				kvmppc_set_timer(vcpu);
943 		}
944 	}
945 
946 	spin_lock(&vc->lock);
947  out:
948 	vc->vcore_state = VCORE_INACTIVE;
949 	vc->preempt_tb = mftb();
950 	list_for_each_entry_safe(vcpu, vnext, &vc->runnable_threads,
951 				 arch.run_list) {
952 		if (vcpu->arch.ret != RESUME_GUEST) {
953 			kvmppc_remove_runnable(vc, vcpu);
954 			wake_up(&vcpu->arch.cpu_run);
955 		}
956 	}
957 
958 	return 1;
959 }
960 
961 /*
962  * Wait for some other vcpu thread to execute us, and
963  * wake us up when we need to handle something in the host.
964  */
965 static void kvmppc_wait_for_exec(struct kvm_vcpu *vcpu, int wait_state)
966 {
967 	DEFINE_WAIT(wait);
968 
969 	prepare_to_wait(&vcpu->arch.cpu_run, &wait, wait_state);
970 	if (vcpu->arch.state == KVMPPC_VCPU_RUNNABLE)
971 		schedule();
972 	finish_wait(&vcpu->arch.cpu_run, &wait);
973 }
974 
975 /*
976  * All the vcpus in this vcore are idle, so wait for a decrementer
977  * or external interrupt to one of the vcpus.  vc->lock is held.
978  */
979 static void kvmppc_vcore_blocked(struct kvmppc_vcore *vc)
980 {
981 	DEFINE_WAIT(wait);
982 	struct kvm_vcpu *v;
983 	int all_idle = 1;
984 
985 	prepare_to_wait(&vc->wq, &wait, TASK_INTERRUPTIBLE);
986 	vc->vcore_state = VCORE_SLEEPING;
987 	spin_unlock(&vc->lock);
988 	list_for_each_entry(v, &vc->runnable_threads, arch.run_list) {
989 		if (!v->arch.ceded || v->arch.pending_exceptions) {
990 			all_idle = 0;
991 			break;
992 		}
993 	}
994 	if (all_idle)
995 		schedule();
996 	finish_wait(&vc->wq, &wait);
997 	spin_lock(&vc->lock);
998 	vc->vcore_state = VCORE_INACTIVE;
999 }
1000 
1001 static int kvmppc_run_vcpu(struct kvm_run *kvm_run, struct kvm_vcpu *vcpu)
1002 {
1003 	int n_ceded;
1004 	int prev_state;
1005 	struct kvmppc_vcore *vc;
1006 	struct kvm_vcpu *v, *vn;
1007 
1008 	kvm_run->exit_reason = 0;
1009 	vcpu->arch.ret = RESUME_GUEST;
1010 	vcpu->arch.trap = 0;
1011 
1012 	/*
1013 	 * Synchronize with other threads in this virtual core
1014 	 */
1015 	vc = vcpu->arch.vcore;
1016 	spin_lock(&vc->lock);
1017 	vcpu->arch.ceded = 0;
1018 	vcpu->arch.run_task = current;
1019 	vcpu->arch.kvm_run = kvm_run;
1020 	prev_state = vcpu->arch.state;
1021 	vcpu->arch.state = KVMPPC_VCPU_RUNNABLE;
1022 	list_add_tail(&vcpu->arch.run_list, &vc->runnable_threads);
1023 	++vc->n_runnable;
1024 
1025 	/*
1026 	 * This happens the first time this is called for a vcpu.
1027 	 * If the vcore is already running, we may be able to start
1028 	 * this thread straight away and have it join in.
1029 	 */
1030 	if (prev_state == KVMPPC_VCPU_STOPPED) {
1031 		if (vc->vcore_state == VCORE_RUNNING &&
1032 		    VCORE_EXIT_COUNT(vc) == 0) {
1033 			vcpu->arch.ptid = vc->n_runnable - 1;
1034 			kvmppc_start_thread(vcpu);
1035 		}
1036 
1037 	} else if (prev_state == KVMPPC_VCPU_BUSY_IN_HOST)
1038 		--vc->n_busy;
1039 
1040 	while (vcpu->arch.state == KVMPPC_VCPU_RUNNABLE &&
1041 	       !signal_pending(current)) {
1042 		if (vc->n_busy || vc->vcore_state != VCORE_INACTIVE) {
1043 			spin_unlock(&vc->lock);
1044 			kvmppc_wait_for_exec(vcpu, TASK_INTERRUPTIBLE);
1045 			spin_lock(&vc->lock);
1046 			continue;
1047 		}
1048 		vc->runner = vcpu;
1049 		n_ceded = 0;
1050 		list_for_each_entry(v, &vc->runnable_threads, arch.run_list)
1051 			n_ceded += v->arch.ceded;
1052 		if (n_ceded == vc->n_runnable)
1053 			kvmppc_vcore_blocked(vc);
1054 		else
1055 			kvmppc_run_core(vc);
1056 
1057 		list_for_each_entry_safe(v, vn, &vc->runnable_threads,
1058 					 arch.run_list) {
1059 			kvmppc_core_prepare_to_enter(v);
1060 			if (signal_pending(v->arch.run_task)) {
1061 				kvmppc_remove_runnable(vc, v);
1062 				v->stat.signal_exits++;
1063 				v->arch.kvm_run->exit_reason = KVM_EXIT_INTR;
1064 				v->arch.ret = -EINTR;
1065 				wake_up(&v->arch.cpu_run);
1066 			}
1067 		}
1068 		vc->runner = NULL;
1069 	}
1070 
1071 	if (signal_pending(current)) {
1072 		if (vc->vcore_state == VCORE_RUNNING ||
1073 		    vc->vcore_state == VCORE_EXITING) {
1074 			spin_unlock(&vc->lock);
1075 			kvmppc_wait_for_exec(vcpu, TASK_UNINTERRUPTIBLE);
1076 			spin_lock(&vc->lock);
1077 		}
1078 		if (vcpu->arch.state == KVMPPC_VCPU_RUNNABLE) {
1079 			kvmppc_remove_runnable(vc, vcpu);
1080 			vcpu->stat.signal_exits++;
1081 			kvm_run->exit_reason = KVM_EXIT_INTR;
1082 			vcpu->arch.ret = -EINTR;
1083 		}
1084 	}
1085 
1086 	spin_unlock(&vc->lock);
1087 	return vcpu->arch.ret;
1088 }
1089 
1090 int kvmppc_vcpu_run(struct kvm_run *run, struct kvm_vcpu *vcpu)
1091 {
1092 	int r;
1093 
1094 	if (!vcpu->arch.sane) {
1095 		run->exit_reason = KVM_EXIT_INTERNAL_ERROR;
1096 		return -EINVAL;
1097 	}
1098 
1099 	kvmppc_core_prepare_to_enter(vcpu);
1100 
1101 	/* No need to go into the guest when all we'll do is come back out */
1102 	if (signal_pending(current)) {
1103 		run->exit_reason = KVM_EXIT_INTR;
1104 		return -EINTR;
1105 	}
1106 
1107 	atomic_inc(&vcpu->kvm->arch.vcpus_running);
1108 	/* Order vcpus_running vs. rma_setup_done, see kvmppc_alloc_reset_hpt */
1109 	smp_mb();
1110 
1111 	/* On the first time here, set up HTAB and VRMA or RMA */
1112 	if (!vcpu->kvm->arch.rma_setup_done) {
1113 		r = kvmppc_hv_setup_htab_rma(vcpu);
1114 		if (r)
1115 			goto out;
1116 	}
1117 
1118 	flush_fp_to_thread(current);
1119 	flush_altivec_to_thread(current);
1120 	flush_vsx_to_thread(current);
1121 	vcpu->arch.wqp = &vcpu->arch.vcore->wq;
1122 	vcpu->arch.pgdir = current->mm->pgd;
1123 
1124 	do {
1125 		r = kvmppc_run_vcpu(run, vcpu);
1126 
1127 		if (run->exit_reason == KVM_EXIT_PAPR_HCALL &&
1128 		    !(vcpu->arch.shregs.msr & MSR_PR)) {
1129 			r = kvmppc_pseries_do_hcall(vcpu);
1130 			kvmppc_core_prepare_to_enter(vcpu);
1131 		}
1132 	} while (r == RESUME_GUEST);
1133 
1134  out:
1135 	atomic_dec(&vcpu->kvm->arch.vcpus_running);
1136 	return r;
1137 }
1138 
1139 
1140 /* Work out RMLS (real mode limit selector) field value for a given RMA size.
1141    Assumes POWER7 or PPC970. */
1142 static inline int lpcr_rmls(unsigned long rma_size)
1143 {
1144 	switch (rma_size) {
1145 	case 32ul << 20:	/* 32 MB */
1146 		if (cpu_has_feature(CPU_FTR_ARCH_206))
1147 			return 8;	/* only supported on POWER7 */
1148 		return -1;
1149 	case 64ul << 20:	/* 64 MB */
1150 		return 3;
1151 	case 128ul << 20:	/* 128 MB */
1152 		return 7;
1153 	case 256ul << 20:	/* 256 MB */
1154 		return 4;
1155 	case 1ul << 30:		/* 1 GB */
1156 		return 2;
1157 	case 16ul << 30:	/* 16 GB */
1158 		return 1;
1159 	case 256ul << 30:	/* 256 GB */
1160 		return 0;
1161 	default:
1162 		return -1;
1163 	}
1164 }
1165 
1166 static int kvm_rma_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
1167 {
1168 	struct kvmppc_linear_info *ri = vma->vm_file->private_data;
1169 	struct page *page;
1170 
1171 	if (vmf->pgoff >= ri->npages)
1172 		return VM_FAULT_SIGBUS;
1173 
1174 	page = pfn_to_page(ri->base_pfn + vmf->pgoff);
1175 	get_page(page);
1176 	vmf->page = page;
1177 	return 0;
1178 }
1179 
1180 static const struct vm_operations_struct kvm_rma_vm_ops = {
1181 	.fault = kvm_rma_fault,
1182 };
1183 
1184 static int kvm_rma_mmap(struct file *file, struct vm_area_struct *vma)
1185 {
1186 	vma->vm_flags |= VM_RESERVED;
1187 	vma->vm_ops = &kvm_rma_vm_ops;
1188 	return 0;
1189 }
1190 
1191 static int kvm_rma_release(struct inode *inode, struct file *filp)
1192 {
1193 	struct kvmppc_linear_info *ri = filp->private_data;
1194 
1195 	kvm_release_rma(ri);
1196 	return 0;
1197 }
1198 
1199 static struct file_operations kvm_rma_fops = {
1200 	.mmap           = kvm_rma_mmap,
1201 	.release	= kvm_rma_release,
1202 };
1203 
1204 long kvm_vm_ioctl_allocate_rma(struct kvm *kvm, struct kvm_allocate_rma *ret)
1205 {
1206 	struct kvmppc_linear_info *ri;
1207 	long fd;
1208 
1209 	ri = kvm_alloc_rma();
1210 	if (!ri)
1211 		return -ENOMEM;
1212 
1213 	fd = anon_inode_getfd("kvm-rma", &kvm_rma_fops, ri, O_RDWR);
1214 	if (fd < 0)
1215 		kvm_release_rma(ri);
1216 
1217 	ret->rma_size = ri->npages << PAGE_SHIFT;
1218 	return fd;
1219 }
1220 
1221 static void kvmppc_add_seg_page_size(struct kvm_ppc_one_seg_page_size **sps,
1222 				     int linux_psize)
1223 {
1224 	struct mmu_psize_def *def = &mmu_psize_defs[linux_psize];
1225 
1226 	if (!def->shift)
1227 		return;
1228 	(*sps)->page_shift = def->shift;
1229 	(*sps)->slb_enc = def->sllp;
1230 	(*sps)->enc[0].page_shift = def->shift;
1231 	(*sps)->enc[0].pte_enc = def->penc;
1232 	(*sps)++;
1233 }
1234 
1235 int kvm_vm_ioctl_get_smmu_info(struct kvm *kvm, struct kvm_ppc_smmu_info *info)
1236 {
1237 	struct kvm_ppc_one_seg_page_size *sps;
1238 
1239 	info->flags = KVM_PPC_PAGE_SIZES_REAL;
1240 	if (mmu_has_feature(MMU_FTR_1T_SEGMENT))
1241 		info->flags |= KVM_PPC_1T_SEGMENTS;
1242 	info->slb_size = mmu_slb_size;
1243 
1244 	/* We only support these sizes for now, and no muti-size segments */
1245 	sps = &info->sps[0];
1246 	kvmppc_add_seg_page_size(&sps, MMU_PAGE_4K);
1247 	kvmppc_add_seg_page_size(&sps, MMU_PAGE_64K);
1248 	kvmppc_add_seg_page_size(&sps, MMU_PAGE_16M);
1249 
1250 	return 0;
1251 }
1252 
1253 /*
1254  * Get (and clear) the dirty memory log for a memory slot.
1255  */
1256 int kvm_vm_ioctl_get_dirty_log(struct kvm *kvm, struct kvm_dirty_log *log)
1257 {
1258 	struct kvm_memory_slot *memslot;
1259 	int r;
1260 	unsigned long n;
1261 
1262 	mutex_lock(&kvm->slots_lock);
1263 
1264 	r = -EINVAL;
1265 	if (log->slot >= KVM_MEMORY_SLOTS)
1266 		goto out;
1267 
1268 	memslot = id_to_memslot(kvm->memslots, log->slot);
1269 	r = -ENOENT;
1270 	if (!memslot->dirty_bitmap)
1271 		goto out;
1272 
1273 	n = kvm_dirty_bitmap_bytes(memslot);
1274 	memset(memslot->dirty_bitmap, 0, n);
1275 
1276 	r = kvmppc_hv_get_dirty_log(kvm, memslot);
1277 	if (r)
1278 		goto out;
1279 
1280 	r = -EFAULT;
1281 	if (copy_to_user(log->dirty_bitmap, memslot->dirty_bitmap, n))
1282 		goto out;
1283 
1284 	r = 0;
1285 out:
1286 	mutex_unlock(&kvm->slots_lock);
1287 	return r;
1288 }
1289 
1290 static unsigned long slb_pgsize_encoding(unsigned long psize)
1291 {
1292 	unsigned long senc = 0;
1293 
1294 	if (psize > 0x1000) {
1295 		senc = SLB_VSID_L;
1296 		if (psize == 0x10000)
1297 			senc |= SLB_VSID_LP_01;
1298 	}
1299 	return senc;
1300 }
1301 
1302 int kvmppc_core_prepare_memory_region(struct kvm *kvm,
1303 				struct kvm_userspace_memory_region *mem)
1304 {
1305 	unsigned long npages;
1306 	unsigned long *phys;
1307 
1308 	/* Allocate a slot_phys array */
1309 	phys = kvm->arch.slot_phys[mem->slot];
1310 	if (!kvm->arch.using_mmu_notifiers && !phys) {
1311 		npages = mem->memory_size >> PAGE_SHIFT;
1312 		phys = vzalloc(npages * sizeof(unsigned long));
1313 		if (!phys)
1314 			return -ENOMEM;
1315 		kvm->arch.slot_phys[mem->slot] = phys;
1316 		kvm->arch.slot_npages[mem->slot] = npages;
1317 	}
1318 
1319 	return 0;
1320 }
1321 
1322 static void unpin_slot(struct kvm *kvm, int slot_id)
1323 {
1324 	unsigned long *physp;
1325 	unsigned long j, npages, pfn;
1326 	struct page *page;
1327 
1328 	physp = kvm->arch.slot_phys[slot_id];
1329 	npages = kvm->arch.slot_npages[slot_id];
1330 	if (physp) {
1331 		spin_lock(&kvm->arch.slot_phys_lock);
1332 		for (j = 0; j < npages; j++) {
1333 			if (!(physp[j] & KVMPPC_GOT_PAGE))
1334 				continue;
1335 			pfn = physp[j] >> PAGE_SHIFT;
1336 			page = pfn_to_page(pfn);
1337 			SetPageDirty(page);
1338 			put_page(page);
1339 		}
1340 		kvm->arch.slot_phys[slot_id] = NULL;
1341 		spin_unlock(&kvm->arch.slot_phys_lock);
1342 		vfree(physp);
1343 	}
1344 }
1345 
1346 void kvmppc_core_commit_memory_region(struct kvm *kvm,
1347 				struct kvm_userspace_memory_region *mem)
1348 {
1349 }
1350 
1351 static int kvmppc_hv_setup_htab_rma(struct kvm_vcpu *vcpu)
1352 {
1353 	int err = 0;
1354 	struct kvm *kvm = vcpu->kvm;
1355 	struct kvmppc_linear_info *ri = NULL;
1356 	unsigned long hva;
1357 	struct kvm_memory_slot *memslot;
1358 	struct vm_area_struct *vma;
1359 	unsigned long lpcr, senc;
1360 	unsigned long psize, porder;
1361 	unsigned long rma_size;
1362 	unsigned long rmls;
1363 	unsigned long *physp;
1364 	unsigned long i, npages;
1365 
1366 	mutex_lock(&kvm->lock);
1367 	if (kvm->arch.rma_setup_done)
1368 		goto out;	/* another vcpu beat us to it */
1369 
1370 	/* Allocate hashed page table (if not done already) and reset it */
1371 	if (!kvm->arch.hpt_virt) {
1372 		err = kvmppc_alloc_hpt(kvm, NULL);
1373 		if (err) {
1374 			pr_err("KVM: Couldn't alloc HPT\n");
1375 			goto out;
1376 		}
1377 	}
1378 
1379 	/* Look up the memslot for guest physical address 0 */
1380 	memslot = gfn_to_memslot(kvm, 0);
1381 
1382 	/* We must have some memory at 0 by now */
1383 	err = -EINVAL;
1384 	if (!memslot || (memslot->flags & KVM_MEMSLOT_INVALID))
1385 		goto out;
1386 
1387 	/* Look up the VMA for the start of this memory slot */
1388 	hva = memslot->userspace_addr;
1389 	down_read(&current->mm->mmap_sem);
1390 	vma = find_vma(current->mm, hva);
1391 	if (!vma || vma->vm_start > hva || (vma->vm_flags & VM_IO))
1392 		goto up_out;
1393 
1394 	psize = vma_kernel_pagesize(vma);
1395 	porder = __ilog2(psize);
1396 
1397 	/* Is this one of our preallocated RMAs? */
1398 	if (vma->vm_file && vma->vm_file->f_op == &kvm_rma_fops &&
1399 	    hva == vma->vm_start)
1400 		ri = vma->vm_file->private_data;
1401 
1402 	up_read(&current->mm->mmap_sem);
1403 
1404 	if (!ri) {
1405 		/* On POWER7, use VRMA; on PPC970, give up */
1406 		err = -EPERM;
1407 		if (cpu_has_feature(CPU_FTR_ARCH_201)) {
1408 			pr_err("KVM: CPU requires an RMO\n");
1409 			goto out;
1410 		}
1411 
1412 		/* We can handle 4k, 64k or 16M pages in the VRMA */
1413 		err = -EINVAL;
1414 		if (!(psize == 0x1000 || psize == 0x10000 ||
1415 		      psize == 0x1000000))
1416 			goto out;
1417 
1418 		/* Update VRMASD field in the LPCR */
1419 		senc = slb_pgsize_encoding(psize);
1420 		kvm->arch.vrma_slb_v = senc | SLB_VSID_B_1T |
1421 			(VRMA_VSID << SLB_VSID_SHIFT_1T);
1422 		lpcr = kvm->arch.lpcr & ~LPCR_VRMASD;
1423 		lpcr |= senc << (LPCR_VRMASD_SH - 4);
1424 		kvm->arch.lpcr = lpcr;
1425 
1426 		/* Create HPTEs in the hash page table for the VRMA */
1427 		kvmppc_map_vrma(vcpu, memslot, porder);
1428 
1429 	} else {
1430 		/* Set up to use an RMO region */
1431 		rma_size = ri->npages;
1432 		if (rma_size > memslot->npages)
1433 			rma_size = memslot->npages;
1434 		rma_size <<= PAGE_SHIFT;
1435 		rmls = lpcr_rmls(rma_size);
1436 		err = -EINVAL;
1437 		if (rmls < 0) {
1438 			pr_err("KVM: Can't use RMA of 0x%lx bytes\n", rma_size);
1439 			goto out;
1440 		}
1441 		atomic_inc(&ri->use_count);
1442 		kvm->arch.rma = ri;
1443 
1444 		/* Update LPCR and RMOR */
1445 		lpcr = kvm->arch.lpcr;
1446 		if (cpu_has_feature(CPU_FTR_ARCH_201)) {
1447 			/* PPC970; insert RMLS value (split field) in HID4 */
1448 			lpcr &= ~((1ul << HID4_RMLS0_SH) |
1449 				  (3ul << HID4_RMLS2_SH));
1450 			lpcr |= ((rmls >> 2) << HID4_RMLS0_SH) |
1451 				((rmls & 3) << HID4_RMLS2_SH);
1452 			/* RMOR is also in HID4 */
1453 			lpcr |= ((ri->base_pfn >> (26 - PAGE_SHIFT)) & 0xffff)
1454 				<< HID4_RMOR_SH;
1455 		} else {
1456 			/* POWER7 */
1457 			lpcr &= ~(LPCR_VPM0 | LPCR_VRMA_L);
1458 			lpcr |= rmls << LPCR_RMLS_SH;
1459 			kvm->arch.rmor = kvm->arch.rma->base_pfn << PAGE_SHIFT;
1460 		}
1461 		kvm->arch.lpcr = lpcr;
1462 		pr_info("KVM: Using RMO at %lx size %lx (LPCR = %lx)\n",
1463 			ri->base_pfn << PAGE_SHIFT, rma_size, lpcr);
1464 
1465 		/* Initialize phys addrs of pages in RMO */
1466 		npages = ri->npages;
1467 		porder = __ilog2(npages);
1468 		physp = kvm->arch.slot_phys[memslot->id];
1469 		spin_lock(&kvm->arch.slot_phys_lock);
1470 		for (i = 0; i < npages; ++i)
1471 			physp[i] = ((ri->base_pfn + i) << PAGE_SHIFT) + porder;
1472 		spin_unlock(&kvm->arch.slot_phys_lock);
1473 	}
1474 
1475 	/* Order updates to kvm->arch.lpcr etc. vs. rma_setup_done */
1476 	smp_wmb();
1477 	kvm->arch.rma_setup_done = 1;
1478 	err = 0;
1479  out:
1480 	mutex_unlock(&kvm->lock);
1481 	return err;
1482 
1483  up_out:
1484 	up_read(&current->mm->mmap_sem);
1485 	goto out;
1486 }
1487 
1488 int kvmppc_core_init_vm(struct kvm *kvm)
1489 {
1490 	unsigned long lpcr, lpid;
1491 
1492 	/* Allocate the guest's logical partition ID */
1493 
1494 	lpid = kvmppc_alloc_lpid();
1495 	if (lpid < 0)
1496 		return -ENOMEM;
1497 	kvm->arch.lpid = lpid;
1498 
1499 	INIT_LIST_HEAD(&kvm->arch.spapr_tce_tables);
1500 
1501 	kvm->arch.rma = NULL;
1502 
1503 	kvm->arch.host_sdr1 = mfspr(SPRN_SDR1);
1504 
1505 	if (cpu_has_feature(CPU_FTR_ARCH_201)) {
1506 		/* PPC970; HID4 is effectively the LPCR */
1507 		kvm->arch.host_lpid = 0;
1508 		kvm->arch.host_lpcr = lpcr = mfspr(SPRN_HID4);
1509 		lpcr &= ~((3 << HID4_LPID1_SH) | (0xful << HID4_LPID5_SH));
1510 		lpcr |= ((lpid >> 4) << HID4_LPID1_SH) |
1511 			((lpid & 0xf) << HID4_LPID5_SH);
1512 	} else {
1513 		/* POWER7; init LPCR for virtual RMA mode */
1514 		kvm->arch.host_lpid = mfspr(SPRN_LPID);
1515 		kvm->arch.host_lpcr = lpcr = mfspr(SPRN_LPCR);
1516 		lpcr &= LPCR_PECE | LPCR_LPES;
1517 		lpcr |= (4UL << LPCR_DPFD_SH) | LPCR_HDICE |
1518 			LPCR_VPM0 | LPCR_VPM1;
1519 		kvm->arch.vrma_slb_v = SLB_VSID_B_1T |
1520 			(VRMA_VSID << SLB_VSID_SHIFT_1T);
1521 	}
1522 	kvm->arch.lpcr = lpcr;
1523 
1524 	kvm->arch.using_mmu_notifiers = !!cpu_has_feature(CPU_FTR_ARCH_206);
1525 	spin_lock_init(&kvm->arch.slot_phys_lock);
1526 	return 0;
1527 }
1528 
1529 void kvmppc_core_destroy_vm(struct kvm *kvm)
1530 {
1531 	unsigned long i;
1532 
1533 	if (!kvm->arch.using_mmu_notifiers)
1534 		for (i = 0; i < KVM_MEM_SLOTS_NUM; i++)
1535 			unpin_slot(kvm, i);
1536 
1537 	if (kvm->arch.rma) {
1538 		kvm_release_rma(kvm->arch.rma);
1539 		kvm->arch.rma = NULL;
1540 	}
1541 
1542 	kvmppc_free_hpt(kvm);
1543 	WARN_ON(!list_empty(&kvm->arch.spapr_tce_tables));
1544 }
1545 
1546 /* These are stubs for now */
1547 void kvmppc_mmu_pte_pflush(struct kvm_vcpu *vcpu, ulong pa_start, ulong pa_end)
1548 {
1549 }
1550 
1551 /* We don't need to emulate any privileged instructions or dcbz */
1552 int kvmppc_core_emulate_op(struct kvm_run *run, struct kvm_vcpu *vcpu,
1553                            unsigned int inst, int *advance)
1554 {
1555 	return EMULATE_FAIL;
1556 }
1557 
1558 int kvmppc_core_emulate_mtspr(struct kvm_vcpu *vcpu, int sprn, ulong spr_val)
1559 {
1560 	return EMULATE_FAIL;
1561 }
1562 
1563 int kvmppc_core_emulate_mfspr(struct kvm_vcpu *vcpu, int sprn, ulong *spr_val)
1564 {
1565 	return EMULATE_FAIL;
1566 }
1567 
1568 static int kvmppc_book3s_hv_init(void)
1569 {
1570 	int r;
1571 
1572 	r = kvm_init(NULL, sizeof(struct kvm_vcpu), 0, THIS_MODULE);
1573 
1574 	if (r)
1575 		return r;
1576 
1577 	r = kvmppc_mmu_hv_init();
1578 
1579 	return r;
1580 }
1581 
1582 static void kvmppc_book3s_hv_exit(void)
1583 {
1584 	kvm_exit();
1585 }
1586 
1587 module_init(kvmppc_book3s_hv_init);
1588 module_exit(kvmppc_book3s_hv_exit);
1589