]> Pileus Git - ~andy/linux/blob - arch/powerpc/kvm/book3s_hv.c
KVM: PPC: Book3S HV: Store LPCR value for each virtual core
[~andy/linux] / arch / powerpc / kvm / book3s_hv.c
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 #include <linux/srcu.h>
34
35 #include <asm/reg.h>
36 #include <asm/cputable.h>
37 #include <asm/cacheflush.h>
38 #include <asm/tlbflush.h>
39 #include <asm/uaccess.h>
40 #include <asm/io.h>
41 #include <asm/kvm_ppc.h>
42 #include <asm/kvm_book3s.h>
43 #include <asm/mmu_context.h>
44 #include <asm/lppaca.h>
45 #include <asm/processor.h>
46 #include <asm/cputhreads.h>
47 #include <asm/page.h>
48 #include <asm/hvcall.h>
49 #include <asm/switch_to.h>
50 #include <asm/smp.h>
51 #include <linux/gfp.h>
52 #include <linux/vmalloc.h>
53 #include <linux/highmem.h>
54 #include <linux/hugetlb.h>
55
56 /* #define EXIT_DEBUG */
57 /* #define EXIT_DEBUG_SIMPLE */
58 /* #define EXIT_DEBUG_INT */
59
60 /* Used to indicate that a guest page fault needs to be handled */
61 #define RESUME_PAGE_FAULT       (RESUME_GUEST | RESUME_FLAG_ARCH1)
62
63 /* Used as a "null" value for timebase values */
64 #define TB_NIL  (~(u64)0)
65
66 static void kvmppc_end_cede(struct kvm_vcpu *vcpu);
67 static int kvmppc_hv_setup_htab_rma(struct kvm_vcpu *vcpu);
68
69 void kvmppc_fast_vcpu_kick(struct kvm_vcpu *vcpu)
70 {
71         int me;
72         int cpu = vcpu->cpu;
73         wait_queue_head_t *wqp;
74
75         wqp = kvm_arch_vcpu_wq(vcpu);
76         if (waitqueue_active(wqp)) {
77                 wake_up_interruptible(wqp);
78                 ++vcpu->stat.halt_wakeup;
79         }
80
81         me = get_cpu();
82
83         /* CPU points to the first thread of the core */
84         if (cpu != me && cpu >= 0 && cpu < nr_cpu_ids) {
85                 int real_cpu = cpu + vcpu->arch.ptid;
86                 if (paca[real_cpu].kvm_hstate.xics_phys)
87                         xics_wake_cpu(real_cpu);
88                 else if (cpu_online(cpu))
89                         smp_send_reschedule(cpu);
90         }
91         put_cpu();
92 }
93
94 /*
95  * We use the vcpu_load/put functions to measure stolen time.
96  * Stolen time is counted as time when either the vcpu is able to
97  * run as part of a virtual core, but the task running the vcore
98  * is preempted or sleeping, or when the vcpu needs something done
99  * in the kernel by the task running the vcpu, but that task is
100  * preempted or sleeping.  Those two things have to be counted
101  * separately, since one of the vcpu tasks will take on the job
102  * of running the core, and the other vcpu tasks in the vcore will
103  * sleep waiting for it to do that, but that sleep shouldn't count
104  * as stolen time.
105  *
106  * Hence we accumulate stolen time when the vcpu can run as part of
107  * a vcore using vc->stolen_tb, and the stolen time when the vcpu
108  * needs its task to do other things in the kernel (for example,
109  * service a page fault) in busy_stolen.  We don't accumulate
110  * stolen time for a vcore when it is inactive, or for a vcpu
111  * when it is in state RUNNING or NOTREADY.  NOTREADY is a bit of
112  * a misnomer; it means that the vcpu task is not executing in
113  * the KVM_VCPU_RUN ioctl, i.e. it is in userspace or elsewhere in
114  * the kernel.  We don't have any way of dividing up that time
115  * between time that the vcpu is genuinely stopped, time that
116  * the task is actively working on behalf of the vcpu, and time
117  * that the task is preempted, so we don't count any of it as
118  * stolen.
119  *
120  * Updates to busy_stolen are protected by arch.tbacct_lock;
121  * updates to vc->stolen_tb are protected by the arch.tbacct_lock
122  * of the vcpu that has taken responsibility for running the vcore
123  * (i.e. vc->runner).  The stolen times are measured in units of
124  * timebase ticks.  (Note that the != TB_NIL checks below are
125  * purely defensive; they should never fail.)
126  */
127
128 void kvmppc_core_vcpu_load(struct kvm_vcpu *vcpu, int cpu)
129 {
130         struct kvmppc_vcore *vc = vcpu->arch.vcore;
131
132         spin_lock(&vcpu->arch.tbacct_lock);
133         if (vc->runner == vcpu && vc->vcore_state != VCORE_INACTIVE &&
134             vc->preempt_tb != TB_NIL) {
135                 vc->stolen_tb += mftb() - vc->preempt_tb;
136                 vc->preempt_tb = TB_NIL;
137         }
138         if (vcpu->arch.state == KVMPPC_VCPU_BUSY_IN_HOST &&
139             vcpu->arch.busy_preempt != TB_NIL) {
140                 vcpu->arch.busy_stolen += mftb() - vcpu->arch.busy_preempt;
141                 vcpu->arch.busy_preempt = TB_NIL;
142         }
143         spin_unlock(&vcpu->arch.tbacct_lock);
144 }
145
146 void kvmppc_core_vcpu_put(struct kvm_vcpu *vcpu)
147 {
148         struct kvmppc_vcore *vc = vcpu->arch.vcore;
149
150         spin_lock(&vcpu->arch.tbacct_lock);
151         if (vc->runner == vcpu && vc->vcore_state != VCORE_INACTIVE)
152                 vc->preempt_tb = mftb();
153         if (vcpu->arch.state == KVMPPC_VCPU_BUSY_IN_HOST)
154                 vcpu->arch.busy_preempt = mftb();
155         spin_unlock(&vcpu->arch.tbacct_lock);
156 }
157
158 void kvmppc_set_msr(struct kvm_vcpu *vcpu, u64 msr)
159 {
160         vcpu->arch.shregs.msr = msr;
161         kvmppc_end_cede(vcpu);
162 }
163
164 void kvmppc_set_pvr(struct kvm_vcpu *vcpu, u32 pvr)
165 {
166         vcpu->arch.pvr = pvr;
167 }
168
169 void kvmppc_dump_regs(struct kvm_vcpu *vcpu)
170 {
171         int r;
172
173         pr_err("vcpu %p (%d):\n", vcpu, vcpu->vcpu_id);
174         pr_err("pc  = %.16lx  msr = %.16llx  trap = %x\n",
175                vcpu->arch.pc, vcpu->arch.shregs.msr, vcpu->arch.trap);
176         for (r = 0; r < 16; ++r)
177                 pr_err("r%2d = %.16lx  r%d = %.16lx\n",
178                        r, kvmppc_get_gpr(vcpu, r),
179                        r+16, kvmppc_get_gpr(vcpu, r+16));
180         pr_err("ctr = %.16lx  lr  = %.16lx\n",
181                vcpu->arch.ctr, vcpu->arch.lr);
182         pr_err("srr0 = %.16llx srr1 = %.16llx\n",
183                vcpu->arch.shregs.srr0, vcpu->arch.shregs.srr1);
184         pr_err("sprg0 = %.16llx sprg1 = %.16llx\n",
185                vcpu->arch.shregs.sprg0, vcpu->arch.shregs.sprg1);
186         pr_err("sprg2 = %.16llx sprg3 = %.16llx\n",
187                vcpu->arch.shregs.sprg2, vcpu->arch.shregs.sprg3);
188         pr_err("cr = %.8x  xer = %.16lx  dsisr = %.8x\n",
189                vcpu->arch.cr, vcpu->arch.xer, vcpu->arch.shregs.dsisr);
190         pr_err("dar = %.16llx\n", vcpu->arch.shregs.dar);
191         pr_err("fault dar = %.16lx dsisr = %.8x\n",
192                vcpu->arch.fault_dar, vcpu->arch.fault_dsisr);
193         pr_err("SLB (%d entries):\n", vcpu->arch.slb_max);
194         for (r = 0; r < vcpu->arch.slb_max; ++r)
195                 pr_err("  ESID = %.16llx VSID = %.16llx\n",
196                        vcpu->arch.slb[r].orige, vcpu->arch.slb[r].origv);
197         pr_err("lpcr = %.16lx sdr1 = %.16lx last_inst = %.8x\n",
198                vcpu->arch.vcore->lpcr, vcpu->kvm->arch.sdr1,
199                vcpu->arch.last_inst);
200 }
201
202 struct kvm_vcpu *kvmppc_find_vcpu(struct kvm *kvm, int id)
203 {
204         int r;
205         struct kvm_vcpu *v, *ret = NULL;
206
207         mutex_lock(&kvm->lock);
208         kvm_for_each_vcpu(r, v, kvm) {
209                 if (v->vcpu_id == id) {
210                         ret = v;
211                         break;
212                 }
213         }
214         mutex_unlock(&kvm->lock);
215         return ret;
216 }
217
218 static void init_vpa(struct kvm_vcpu *vcpu, struct lppaca *vpa)
219 {
220         vpa->__old_status |= LPPACA_OLD_SHARED_PROC;
221         vpa->yield_count = 1;
222 }
223
224 static int set_vpa(struct kvm_vcpu *vcpu, struct kvmppc_vpa *v,
225                    unsigned long addr, unsigned long len)
226 {
227         /* check address is cacheline aligned */
228         if (addr & (L1_CACHE_BYTES - 1))
229                 return -EINVAL;
230         spin_lock(&vcpu->arch.vpa_update_lock);
231         if (v->next_gpa != addr || v->len != len) {
232                 v->next_gpa = addr;
233                 v->len = addr ? len : 0;
234                 v->update_pending = 1;
235         }
236         spin_unlock(&vcpu->arch.vpa_update_lock);
237         return 0;
238 }
239
240 /* Length for a per-processor buffer is passed in at offset 4 in the buffer */
241 struct reg_vpa {
242         u32 dummy;
243         union {
244                 u16 hword;
245                 u32 word;
246         } length;
247 };
248
249 static int vpa_is_registered(struct kvmppc_vpa *vpap)
250 {
251         if (vpap->update_pending)
252                 return vpap->next_gpa != 0;
253         return vpap->pinned_addr != NULL;
254 }
255
256 static unsigned long do_h_register_vpa(struct kvm_vcpu *vcpu,
257                                        unsigned long flags,
258                                        unsigned long vcpuid, unsigned long vpa)
259 {
260         struct kvm *kvm = vcpu->kvm;
261         unsigned long len, nb;
262         void *va;
263         struct kvm_vcpu *tvcpu;
264         int err;
265         int subfunc;
266         struct kvmppc_vpa *vpap;
267
268         tvcpu = kvmppc_find_vcpu(kvm, vcpuid);
269         if (!tvcpu)
270                 return H_PARAMETER;
271
272         subfunc = (flags >> H_VPA_FUNC_SHIFT) & H_VPA_FUNC_MASK;
273         if (subfunc == H_VPA_REG_VPA || subfunc == H_VPA_REG_DTL ||
274             subfunc == H_VPA_REG_SLB) {
275                 /* Registering new area - address must be cache-line aligned */
276                 if ((vpa & (L1_CACHE_BYTES - 1)) || !vpa)
277                         return H_PARAMETER;
278
279                 /* convert logical addr to kernel addr and read length */
280                 va = kvmppc_pin_guest_page(kvm, vpa, &nb);
281                 if (va == NULL)
282                         return H_PARAMETER;
283                 if (subfunc == H_VPA_REG_VPA)
284                         len = ((struct reg_vpa *)va)->length.hword;
285                 else
286                         len = ((struct reg_vpa *)va)->length.word;
287                 kvmppc_unpin_guest_page(kvm, va, vpa, false);
288
289                 /* Check length */
290                 if (len > nb || len < sizeof(struct reg_vpa))
291                         return H_PARAMETER;
292         } else {
293                 vpa = 0;
294                 len = 0;
295         }
296
297         err = H_PARAMETER;
298         vpap = NULL;
299         spin_lock(&tvcpu->arch.vpa_update_lock);
300
301         switch (subfunc) {
302         case H_VPA_REG_VPA:             /* register VPA */
303                 if (len < sizeof(struct lppaca))
304                         break;
305                 vpap = &tvcpu->arch.vpa;
306                 err = 0;
307                 break;
308
309         case H_VPA_REG_DTL:             /* register DTL */
310                 if (len < sizeof(struct dtl_entry))
311                         break;
312                 len -= len % sizeof(struct dtl_entry);
313
314                 /* Check that they have previously registered a VPA */
315                 err = H_RESOURCE;
316                 if (!vpa_is_registered(&tvcpu->arch.vpa))
317                         break;
318
319                 vpap = &tvcpu->arch.dtl;
320                 err = 0;
321                 break;
322
323         case H_VPA_REG_SLB:             /* register SLB shadow buffer */
324                 /* Check that they have previously registered a VPA */
325                 err = H_RESOURCE;
326                 if (!vpa_is_registered(&tvcpu->arch.vpa))
327                         break;
328
329                 vpap = &tvcpu->arch.slb_shadow;
330                 err = 0;
331                 break;
332
333         case H_VPA_DEREG_VPA:           /* deregister VPA */
334                 /* Check they don't still have a DTL or SLB buf registered */
335                 err = H_RESOURCE;
336                 if (vpa_is_registered(&tvcpu->arch.dtl) ||
337                     vpa_is_registered(&tvcpu->arch.slb_shadow))
338                         break;
339
340                 vpap = &tvcpu->arch.vpa;
341                 err = 0;
342                 break;
343
344         case H_VPA_DEREG_DTL:           /* deregister DTL */
345                 vpap = &tvcpu->arch.dtl;
346                 err = 0;
347                 break;
348
349         case H_VPA_DEREG_SLB:           /* deregister SLB shadow buffer */
350                 vpap = &tvcpu->arch.slb_shadow;
351                 err = 0;
352                 break;
353         }
354
355         if (vpap) {
356                 vpap->next_gpa = vpa;
357                 vpap->len = len;
358                 vpap->update_pending = 1;
359         }
360
361         spin_unlock(&tvcpu->arch.vpa_update_lock);
362
363         return err;
364 }
365
366 static void kvmppc_update_vpa(struct kvm_vcpu *vcpu, struct kvmppc_vpa *vpap)
367 {
368         struct kvm *kvm = vcpu->kvm;
369         void *va;
370         unsigned long nb;
371         unsigned long gpa;
372
373         /*
374          * We need to pin the page pointed to by vpap->next_gpa,
375          * but we can't call kvmppc_pin_guest_page under the lock
376          * as it does get_user_pages() and down_read().  So we
377          * have to drop the lock, pin the page, then get the lock
378          * again and check that a new area didn't get registered
379          * in the meantime.
380          */
381         for (;;) {
382                 gpa = vpap->next_gpa;
383                 spin_unlock(&vcpu->arch.vpa_update_lock);
384                 va = NULL;
385                 nb = 0;
386                 if (gpa)
387                         va = kvmppc_pin_guest_page(kvm, gpa, &nb);
388                 spin_lock(&vcpu->arch.vpa_update_lock);
389                 if (gpa == vpap->next_gpa)
390                         break;
391                 /* sigh... unpin that one and try again */
392                 if (va)
393                         kvmppc_unpin_guest_page(kvm, va, gpa, false);
394         }
395
396         vpap->update_pending = 0;
397         if (va && nb < vpap->len) {
398                 /*
399                  * If it's now too short, it must be that userspace
400                  * has changed the mappings underlying guest memory,
401                  * so unregister the region.
402                  */
403                 kvmppc_unpin_guest_page(kvm, va, gpa, false);
404                 va = NULL;
405         }
406         if (vpap->pinned_addr)
407                 kvmppc_unpin_guest_page(kvm, vpap->pinned_addr, vpap->gpa,
408                                         vpap->dirty);
409         vpap->gpa = gpa;
410         vpap->pinned_addr = va;
411         vpap->dirty = false;
412         if (va)
413                 vpap->pinned_end = va + vpap->len;
414 }
415
416 static void kvmppc_update_vpas(struct kvm_vcpu *vcpu)
417 {
418         if (!(vcpu->arch.vpa.update_pending ||
419               vcpu->arch.slb_shadow.update_pending ||
420               vcpu->arch.dtl.update_pending))
421                 return;
422
423         spin_lock(&vcpu->arch.vpa_update_lock);
424         if (vcpu->arch.vpa.update_pending) {
425                 kvmppc_update_vpa(vcpu, &vcpu->arch.vpa);
426                 if (vcpu->arch.vpa.pinned_addr)
427                         init_vpa(vcpu, vcpu->arch.vpa.pinned_addr);
428         }
429         if (vcpu->arch.dtl.update_pending) {
430                 kvmppc_update_vpa(vcpu, &vcpu->arch.dtl);
431                 vcpu->arch.dtl_ptr = vcpu->arch.dtl.pinned_addr;
432                 vcpu->arch.dtl_index = 0;
433         }
434         if (vcpu->arch.slb_shadow.update_pending)
435                 kvmppc_update_vpa(vcpu, &vcpu->arch.slb_shadow);
436         spin_unlock(&vcpu->arch.vpa_update_lock);
437 }
438
439 /*
440  * Return the accumulated stolen time for the vcore up until `now'.
441  * The caller should hold the vcore lock.
442  */
443 static u64 vcore_stolen_time(struct kvmppc_vcore *vc, u64 now)
444 {
445         u64 p;
446
447         /*
448          * If we are the task running the vcore, then since we hold
449          * the vcore lock, we can't be preempted, so stolen_tb/preempt_tb
450          * can't be updated, so we don't need the tbacct_lock.
451          * If the vcore is inactive, it can't become active (since we
452          * hold the vcore lock), so the vcpu load/put functions won't
453          * update stolen_tb/preempt_tb, and we don't need tbacct_lock.
454          */
455         if (vc->vcore_state != VCORE_INACTIVE &&
456             vc->runner->arch.run_task != current) {
457                 spin_lock(&vc->runner->arch.tbacct_lock);
458                 p = vc->stolen_tb;
459                 if (vc->preempt_tb != TB_NIL)
460                         p += now - vc->preempt_tb;
461                 spin_unlock(&vc->runner->arch.tbacct_lock);
462         } else {
463                 p = vc->stolen_tb;
464         }
465         return p;
466 }
467
468 static void kvmppc_create_dtl_entry(struct kvm_vcpu *vcpu,
469                                     struct kvmppc_vcore *vc)
470 {
471         struct dtl_entry *dt;
472         struct lppaca *vpa;
473         unsigned long stolen;
474         unsigned long core_stolen;
475         u64 now;
476
477         dt = vcpu->arch.dtl_ptr;
478         vpa = vcpu->arch.vpa.pinned_addr;
479         now = mftb();
480         core_stolen = vcore_stolen_time(vc, now);
481         stolen = core_stolen - vcpu->arch.stolen_logged;
482         vcpu->arch.stolen_logged = core_stolen;
483         spin_lock(&vcpu->arch.tbacct_lock);
484         stolen += vcpu->arch.busy_stolen;
485         vcpu->arch.busy_stolen = 0;
486         spin_unlock(&vcpu->arch.tbacct_lock);
487         if (!dt || !vpa)
488                 return;
489         memset(dt, 0, sizeof(struct dtl_entry));
490         dt->dispatch_reason = 7;
491         dt->processor_id = vc->pcpu + vcpu->arch.ptid;
492         dt->timebase = now + vc->tb_offset;
493         dt->enqueue_to_dispatch_time = stolen;
494         dt->srr0 = kvmppc_get_pc(vcpu);
495         dt->srr1 = vcpu->arch.shregs.msr;
496         ++dt;
497         if (dt == vcpu->arch.dtl.pinned_end)
498                 dt = vcpu->arch.dtl.pinned_addr;
499         vcpu->arch.dtl_ptr = dt;
500         /* order writing *dt vs. writing vpa->dtl_idx */
501         smp_wmb();
502         vpa->dtl_idx = ++vcpu->arch.dtl_index;
503         vcpu->arch.dtl.dirty = true;
504 }
505
506 int kvmppc_pseries_do_hcall(struct kvm_vcpu *vcpu)
507 {
508         unsigned long req = kvmppc_get_gpr(vcpu, 3);
509         unsigned long target, ret = H_SUCCESS;
510         struct kvm_vcpu *tvcpu;
511         int idx, rc;
512
513         switch (req) {
514         case H_ENTER:
515                 idx = srcu_read_lock(&vcpu->kvm->srcu);
516                 ret = kvmppc_virtmode_h_enter(vcpu, kvmppc_get_gpr(vcpu, 4),
517                                               kvmppc_get_gpr(vcpu, 5),
518                                               kvmppc_get_gpr(vcpu, 6),
519                                               kvmppc_get_gpr(vcpu, 7));
520                 srcu_read_unlock(&vcpu->kvm->srcu, idx);
521                 break;
522         case H_CEDE:
523                 break;
524         case H_PROD:
525                 target = kvmppc_get_gpr(vcpu, 4);
526                 tvcpu = kvmppc_find_vcpu(vcpu->kvm, target);
527                 if (!tvcpu) {
528                         ret = H_PARAMETER;
529                         break;
530                 }
531                 tvcpu->arch.prodded = 1;
532                 smp_mb();
533                 if (vcpu->arch.ceded) {
534                         if (waitqueue_active(&vcpu->wq)) {
535                                 wake_up_interruptible(&vcpu->wq);
536                                 vcpu->stat.halt_wakeup++;
537                         }
538                 }
539                 break;
540         case H_CONFER:
541                 target = kvmppc_get_gpr(vcpu, 4);
542                 if (target == -1)
543                         break;
544                 tvcpu = kvmppc_find_vcpu(vcpu->kvm, target);
545                 if (!tvcpu) {
546                         ret = H_PARAMETER;
547                         break;
548                 }
549                 kvm_vcpu_yield_to(tvcpu);
550                 break;
551         case H_REGISTER_VPA:
552                 ret = do_h_register_vpa(vcpu, kvmppc_get_gpr(vcpu, 4),
553                                         kvmppc_get_gpr(vcpu, 5),
554                                         kvmppc_get_gpr(vcpu, 6));
555                 break;
556         case H_RTAS:
557                 if (list_empty(&vcpu->kvm->arch.rtas_tokens))
558                         return RESUME_HOST;
559
560                 rc = kvmppc_rtas_hcall(vcpu);
561
562                 if (rc == -ENOENT)
563                         return RESUME_HOST;
564                 else if (rc == 0)
565                         break;
566
567                 /* Send the error out to userspace via KVM_RUN */
568                 return rc;
569
570         case H_XIRR:
571         case H_CPPR:
572         case H_EOI:
573         case H_IPI:
574         case H_IPOLL:
575         case H_XIRR_X:
576                 if (kvmppc_xics_enabled(vcpu)) {
577                         ret = kvmppc_xics_hcall(vcpu, req);
578                         break;
579                 } /* fallthrough */
580         default:
581                 return RESUME_HOST;
582         }
583         kvmppc_set_gpr(vcpu, 3, ret);
584         vcpu->arch.hcall_needed = 0;
585         return RESUME_GUEST;
586 }
587
588 static int kvmppc_handle_exit(struct kvm_run *run, struct kvm_vcpu *vcpu,
589                               struct task_struct *tsk)
590 {
591         int r = RESUME_HOST;
592
593         vcpu->stat.sum_exits++;
594
595         run->exit_reason = KVM_EXIT_UNKNOWN;
596         run->ready_for_interrupt_injection = 1;
597         switch (vcpu->arch.trap) {
598         /* We're good on these - the host merely wanted to get our attention */
599         case BOOK3S_INTERRUPT_HV_DECREMENTER:
600                 vcpu->stat.dec_exits++;
601                 r = RESUME_GUEST;
602                 break;
603         case BOOK3S_INTERRUPT_EXTERNAL:
604                 vcpu->stat.ext_intr_exits++;
605                 r = RESUME_GUEST;
606                 break;
607         case BOOK3S_INTERRUPT_PERFMON:
608                 r = RESUME_GUEST;
609                 break;
610         case BOOK3S_INTERRUPT_MACHINE_CHECK:
611                 /*
612                  * Deliver a machine check interrupt to the guest.
613                  * We have to do this, even if the host has handled the
614                  * machine check, because machine checks use SRR0/1 and
615                  * the interrupt might have trashed guest state in them.
616                  */
617                 kvmppc_book3s_queue_irqprio(vcpu,
618                                             BOOK3S_INTERRUPT_MACHINE_CHECK);
619                 r = RESUME_GUEST;
620                 break;
621         case BOOK3S_INTERRUPT_PROGRAM:
622         {
623                 ulong flags;
624                 /*
625                  * Normally program interrupts are delivered directly
626                  * to the guest by the hardware, but we can get here
627                  * as a result of a hypervisor emulation interrupt
628                  * (e40) getting turned into a 700 by BML RTAS.
629                  */
630                 flags = vcpu->arch.shregs.msr & 0x1f0000ull;
631                 kvmppc_core_queue_program(vcpu, flags);
632                 r = RESUME_GUEST;
633                 break;
634         }
635         case BOOK3S_INTERRUPT_SYSCALL:
636         {
637                 /* hcall - punt to userspace */
638                 int i;
639
640                 if (vcpu->arch.shregs.msr & MSR_PR) {
641                         /* sc 1 from userspace - reflect to guest syscall */
642                         kvmppc_book3s_queue_irqprio(vcpu, BOOK3S_INTERRUPT_SYSCALL);
643                         r = RESUME_GUEST;
644                         break;
645                 }
646                 run->papr_hcall.nr = kvmppc_get_gpr(vcpu, 3);
647                 for (i = 0; i < 9; ++i)
648                         run->papr_hcall.args[i] = kvmppc_get_gpr(vcpu, 4 + i);
649                 run->exit_reason = KVM_EXIT_PAPR_HCALL;
650                 vcpu->arch.hcall_needed = 1;
651                 r = RESUME_HOST;
652                 break;
653         }
654         /*
655          * We get these next two if the guest accesses a page which it thinks
656          * it has mapped but which is not actually present, either because
657          * it is for an emulated I/O device or because the corresonding
658          * host page has been paged out.  Any other HDSI/HISI interrupts
659          * have been handled already.
660          */
661         case BOOK3S_INTERRUPT_H_DATA_STORAGE:
662                 r = RESUME_PAGE_FAULT;
663                 break;
664         case BOOK3S_INTERRUPT_H_INST_STORAGE:
665                 vcpu->arch.fault_dar = kvmppc_get_pc(vcpu);
666                 vcpu->arch.fault_dsisr = 0;
667                 r = RESUME_PAGE_FAULT;
668                 break;
669         /*
670          * This occurs if the guest executes an illegal instruction.
671          * We just generate a program interrupt to the guest, since
672          * we don't emulate any guest instructions at this stage.
673          */
674         case BOOK3S_INTERRUPT_H_EMUL_ASSIST:
675                 kvmppc_core_queue_program(vcpu, 0x80000);
676                 r = RESUME_GUEST;
677                 break;
678         default:
679                 kvmppc_dump_regs(vcpu);
680                 printk(KERN_EMERG "trap=0x%x | pc=0x%lx | msr=0x%llx\n",
681                         vcpu->arch.trap, kvmppc_get_pc(vcpu),
682                         vcpu->arch.shregs.msr);
683                 r = RESUME_HOST;
684                 BUG();
685                 break;
686         }
687
688         return r;
689 }
690
691 int kvm_arch_vcpu_ioctl_get_sregs(struct kvm_vcpu *vcpu,
692                                   struct kvm_sregs *sregs)
693 {
694         int i;
695
696         memset(sregs, 0, sizeof(struct kvm_sregs));
697         sregs->pvr = vcpu->arch.pvr;
698         for (i = 0; i < vcpu->arch.slb_max; i++) {
699                 sregs->u.s.ppc64.slb[i].slbe = vcpu->arch.slb[i].orige;
700                 sregs->u.s.ppc64.slb[i].slbv = vcpu->arch.slb[i].origv;
701         }
702
703         return 0;
704 }
705
706 int kvm_arch_vcpu_ioctl_set_sregs(struct kvm_vcpu *vcpu,
707                                   struct kvm_sregs *sregs)
708 {
709         int i, j;
710
711         kvmppc_set_pvr(vcpu, sregs->pvr);
712
713         j = 0;
714         for (i = 0; i < vcpu->arch.slb_nr; i++) {
715                 if (sregs->u.s.ppc64.slb[i].slbe & SLB_ESID_V) {
716                         vcpu->arch.slb[j].orige = sregs->u.s.ppc64.slb[i].slbe;
717                         vcpu->arch.slb[j].origv = sregs->u.s.ppc64.slb[i].slbv;
718                         ++j;
719                 }
720         }
721         vcpu->arch.slb_max = j;
722
723         return 0;
724 }
725
726 static void kvmppc_set_lpcr(struct kvm_vcpu *vcpu, u64 new_lpcr)
727 {
728         struct kvmppc_vcore *vc = vcpu->arch.vcore;
729         u64 mask;
730
731         spin_lock(&vc->lock);
732         /*
733          * Userspace can only modify DPFD (default prefetch depth),
734          * ILE (interrupt little-endian) and TC (translation control).
735          */
736         mask = LPCR_DPFD | LPCR_ILE | LPCR_TC;
737         vc->lpcr = (vc->lpcr & ~mask) | (new_lpcr & mask);
738         spin_unlock(&vc->lock);
739 }
740
741 int kvmppc_get_one_reg(struct kvm_vcpu *vcpu, u64 id, union kvmppc_one_reg *val)
742 {
743         int r = 0;
744         long int i;
745
746         switch (id) {
747         case KVM_REG_PPC_HIOR:
748                 *val = get_reg_val(id, 0);
749                 break;
750         case KVM_REG_PPC_DABR:
751                 *val = get_reg_val(id, vcpu->arch.dabr);
752                 break;
753         case KVM_REG_PPC_DSCR:
754                 *val = get_reg_val(id, vcpu->arch.dscr);
755                 break;
756         case KVM_REG_PPC_PURR:
757                 *val = get_reg_val(id, vcpu->arch.purr);
758                 break;
759         case KVM_REG_PPC_SPURR:
760                 *val = get_reg_val(id, vcpu->arch.spurr);
761                 break;
762         case KVM_REG_PPC_AMR:
763                 *val = get_reg_val(id, vcpu->arch.amr);
764                 break;
765         case KVM_REG_PPC_UAMOR:
766                 *val = get_reg_val(id, vcpu->arch.uamor);
767                 break;
768         case KVM_REG_PPC_MMCR0 ... KVM_REG_PPC_MMCRA:
769                 i = id - KVM_REG_PPC_MMCR0;
770                 *val = get_reg_val(id, vcpu->arch.mmcr[i]);
771                 break;
772         case KVM_REG_PPC_PMC1 ... KVM_REG_PPC_PMC8:
773                 i = id - KVM_REG_PPC_PMC1;
774                 *val = get_reg_val(id, vcpu->arch.pmc[i]);
775                 break;
776         case KVM_REG_PPC_SIAR:
777                 *val = get_reg_val(id, vcpu->arch.siar);
778                 break;
779         case KVM_REG_PPC_SDAR:
780                 *val = get_reg_val(id, vcpu->arch.sdar);
781                 break;
782 #ifdef CONFIG_VSX
783         case KVM_REG_PPC_FPR0 ... KVM_REG_PPC_FPR31:
784                 if (cpu_has_feature(CPU_FTR_VSX)) {
785                         /* VSX => FP reg i is stored in arch.vsr[2*i] */
786                         long int i = id - KVM_REG_PPC_FPR0;
787                         *val = get_reg_val(id, vcpu->arch.vsr[2 * i]);
788                 } else {
789                         /* let generic code handle it */
790                         r = -EINVAL;
791                 }
792                 break;
793         case KVM_REG_PPC_VSR0 ... KVM_REG_PPC_VSR31:
794                 if (cpu_has_feature(CPU_FTR_VSX)) {
795                         long int i = id - KVM_REG_PPC_VSR0;
796                         val->vsxval[0] = vcpu->arch.vsr[2 * i];
797                         val->vsxval[1] = vcpu->arch.vsr[2 * i + 1];
798                 } else {
799                         r = -ENXIO;
800                 }
801                 break;
802 #endif /* CONFIG_VSX */
803         case KVM_REG_PPC_VPA_ADDR:
804                 spin_lock(&vcpu->arch.vpa_update_lock);
805                 *val = get_reg_val(id, vcpu->arch.vpa.next_gpa);
806                 spin_unlock(&vcpu->arch.vpa_update_lock);
807                 break;
808         case KVM_REG_PPC_VPA_SLB:
809                 spin_lock(&vcpu->arch.vpa_update_lock);
810                 val->vpaval.addr = vcpu->arch.slb_shadow.next_gpa;
811                 val->vpaval.length = vcpu->arch.slb_shadow.len;
812                 spin_unlock(&vcpu->arch.vpa_update_lock);
813                 break;
814         case KVM_REG_PPC_VPA_DTL:
815                 spin_lock(&vcpu->arch.vpa_update_lock);
816                 val->vpaval.addr = vcpu->arch.dtl.next_gpa;
817                 val->vpaval.length = vcpu->arch.dtl.len;
818                 spin_unlock(&vcpu->arch.vpa_update_lock);
819                 break;
820         case KVM_REG_PPC_TB_OFFSET:
821                 *val = get_reg_val(id, vcpu->arch.vcore->tb_offset);
822                 break;
823         case KVM_REG_PPC_LPCR:
824                 *val = get_reg_val(id, vcpu->arch.vcore->lpcr);
825                 break;
826         default:
827                 r = -EINVAL;
828                 break;
829         }
830
831         return r;
832 }
833
834 int kvmppc_set_one_reg(struct kvm_vcpu *vcpu, u64 id, union kvmppc_one_reg *val)
835 {
836         int r = 0;
837         long int i;
838         unsigned long addr, len;
839
840         switch (id) {
841         case KVM_REG_PPC_HIOR:
842                 /* Only allow this to be set to zero */
843                 if (set_reg_val(id, *val))
844                         r = -EINVAL;
845                 break;
846         case KVM_REG_PPC_DABR:
847                 vcpu->arch.dabr = set_reg_val(id, *val);
848                 break;
849         case KVM_REG_PPC_DSCR:
850                 vcpu->arch.dscr = set_reg_val(id, *val);
851                 break;
852         case KVM_REG_PPC_PURR:
853                 vcpu->arch.purr = set_reg_val(id, *val);
854                 break;
855         case KVM_REG_PPC_SPURR:
856                 vcpu->arch.spurr = set_reg_val(id, *val);
857                 break;
858         case KVM_REG_PPC_AMR:
859                 vcpu->arch.amr = set_reg_val(id, *val);
860                 break;
861         case KVM_REG_PPC_UAMOR:
862                 vcpu->arch.uamor = set_reg_val(id, *val);
863                 break;
864         case KVM_REG_PPC_MMCR0 ... KVM_REG_PPC_MMCRA:
865                 i = id - KVM_REG_PPC_MMCR0;
866                 vcpu->arch.mmcr[i] = set_reg_val(id, *val);
867                 break;
868         case KVM_REG_PPC_PMC1 ... KVM_REG_PPC_PMC8:
869                 i = id - KVM_REG_PPC_PMC1;
870                 vcpu->arch.pmc[i] = set_reg_val(id, *val);
871                 break;
872         case KVM_REG_PPC_SIAR:
873                 vcpu->arch.siar = set_reg_val(id, *val);
874                 break;
875         case KVM_REG_PPC_SDAR:
876                 vcpu->arch.sdar = set_reg_val(id, *val);
877                 break;
878 #ifdef CONFIG_VSX
879         case KVM_REG_PPC_FPR0 ... KVM_REG_PPC_FPR31:
880                 if (cpu_has_feature(CPU_FTR_VSX)) {
881                         /* VSX => FP reg i is stored in arch.vsr[2*i] */
882                         long int i = id - KVM_REG_PPC_FPR0;
883                         vcpu->arch.vsr[2 * i] = set_reg_val(id, *val);
884                 } else {
885                         /* let generic code handle it */
886                         r = -EINVAL;
887                 }
888                 break;
889         case KVM_REG_PPC_VSR0 ... KVM_REG_PPC_VSR31:
890                 if (cpu_has_feature(CPU_FTR_VSX)) {
891                         long int i = id - KVM_REG_PPC_VSR0;
892                         vcpu->arch.vsr[2 * i] = val->vsxval[0];
893                         vcpu->arch.vsr[2 * i + 1] = val->vsxval[1];
894                 } else {
895                         r = -ENXIO;
896                 }
897                 break;
898 #endif /* CONFIG_VSX */
899         case KVM_REG_PPC_VPA_ADDR:
900                 addr = set_reg_val(id, *val);
901                 r = -EINVAL;
902                 if (!addr && (vcpu->arch.slb_shadow.next_gpa ||
903                               vcpu->arch.dtl.next_gpa))
904                         break;
905                 r = set_vpa(vcpu, &vcpu->arch.vpa, addr, sizeof(struct lppaca));
906                 break;
907         case KVM_REG_PPC_VPA_SLB:
908                 addr = val->vpaval.addr;
909                 len = val->vpaval.length;
910                 r = -EINVAL;
911                 if (addr && !vcpu->arch.vpa.next_gpa)
912                         break;
913                 r = set_vpa(vcpu, &vcpu->arch.slb_shadow, addr, len);
914                 break;
915         case KVM_REG_PPC_VPA_DTL:
916                 addr = val->vpaval.addr;
917                 len = val->vpaval.length;
918                 r = -EINVAL;
919                 if (addr && (len < sizeof(struct dtl_entry) ||
920                              !vcpu->arch.vpa.next_gpa))
921                         break;
922                 len -= len % sizeof(struct dtl_entry);
923                 r = set_vpa(vcpu, &vcpu->arch.dtl, addr, len);
924                 break;
925         case KVM_REG_PPC_TB_OFFSET:
926                 /* round up to multiple of 2^24 */
927                 vcpu->arch.vcore->tb_offset =
928                         ALIGN(set_reg_val(id, *val), 1UL << 24);
929                 break;
930         case KVM_REG_PPC_LPCR:
931                 kvmppc_set_lpcr(vcpu, set_reg_val(id, *val));
932                 break;
933         default:
934                 r = -EINVAL;
935                 break;
936         }
937
938         return r;
939 }
940
941 int kvmppc_core_check_processor_compat(void)
942 {
943         if (cpu_has_feature(CPU_FTR_HVMODE))
944                 return 0;
945         return -EIO;
946 }
947
948 struct kvm_vcpu *kvmppc_core_vcpu_create(struct kvm *kvm, unsigned int id)
949 {
950         struct kvm_vcpu *vcpu;
951         int err = -EINVAL;
952         int core;
953         struct kvmppc_vcore *vcore;
954
955         core = id / threads_per_core;
956         if (core >= KVM_MAX_VCORES)
957                 goto out;
958
959         err = -ENOMEM;
960         vcpu = kmem_cache_zalloc(kvm_vcpu_cache, GFP_KERNEL);
961         if (!vcpu)
962                 goto out;
963
964         err = kvm_vcpu_init(vcpu, kvm, id);
965         if (err)
966                 goto free_vcpu;
967
968         vcpu->arch.shared = &vcpu->arch.shregs;
969         vcpu->arch.mmcr[0] = MMCR0_FC;
970         vcpu->arch.ctrl = CTRL_RUNLATCH;
971         /* default to host PVR, since we can't spoof it */
972         vcpu->arch.pvr = mfspr(SPRN_PVR);
973         kvmppc_set_pvr(vcpu, vcpu->arch.pvr);
974         spin_lock_init(&vcpu->arch.vpa_update_lock);
975         spin_lock_init(&vcpu->arch.tbacct_lock);
976         vcpu->arch.busy_preempt = TB_NIL;
977
978         kvmppc_mmu_book3s_hv_init(vcpu);
979
980         vcpu->arch.state = KVMPPC_VCPU_NOTREADY;
981
982         init_waitqueue_head(&vcpu->arch.cpu_run);
983
984         mutex_lock(&kvm->lock);
985         vcore = kvm->arch.vcores[core];
986         if (!vcore) {
987                 vcore = kzalloc(sizeof(struct kvmppc_vcore), GFP_KERNEL);
988                 if (vcore) {
989                         INIT_LIST_HEAD(&vcore->runnable_threads);
990                         spin_lock_init(&vcore->lock);
991                         init_waitqueue_head(&vcore->wq);
992                         vcore->preempt_tb = TB_NIL;
993                         vcore->lpcr = kvm->arch.lpcr;
994                 }
995                 kvm->arch.vcores[core] = vcore;
996                 kvm->arch.online_vcores++;
997         }
998         mutex_unlock(&kvm->lock);
999
1000         if (!vcore)
1001                 goto free_vcpu;
1002
1003         spin_lock(&vcore->lock);
1004         ++vcore->num_threads;
1005         spin_unlock(&vcore->lock);
1006         vcpu->arch.vcore = vcore;
1007
1008         vcpu->arch.cpu_type = KVM_CPU_3S_64;
1009         kvmppc_sanity_check(vcpu);
1010
1011         return vcpu;
1012
1013 free_vcpu:
1014         kmem_cache_free(kvm_vcpu_cache, vcpu);
1015 out:
1016         return ERR_PTR(err);
1017 }
1018
1019 static void unpin_vpa(struct kvm *kvm, struct kvmppc_vpa *vpa)
1020 {
1021         if (vpa->pinned_addr)
1022                 kvmppc_unpin_guest_page(kvm, vpa->pinned_addr, vpa->gpa,
1023                                         vpa->dirty);
1024 }
1025
1026 void kvmppc_core_vcpu_free(struct kvm_vcpu *vcpu)
1027 {
1028         spin_lock(&vcpu->arch.vpa_update_lock);
1029         unpin_vpa(vcpu->kvm, &vcpu->arch.dtl);
1030         unpin_vpa(vcpu->kvm, &vcpu->arch.slb_shadow);
1031         unpin_vpa(vcpu->kvm, &vcpu->arch.vpa);
1032         spin_unlock(&vcpu->arch.vpa_update_lock);
1033         kvm_vcpu_uninit(vcpu);
1034         kmem_cache_free(kvm_vcpu_cache, vcpu);
1035 }
1036
1037 static void kvmppc_set_timer(struct kvm_vcpu *vcpu)
1038 {
1039         unsigned long dec_nsec, now;
1040
1041         now = get_tb();
1042         if (now > vcpu->arch.dec_expires) {
1043                 /* decrementer has already gone negative */
1044                 kvmppc_core_queue_dec(vcpu);
1045                 kvmppc_core_prepare_to_enter(vcpu);
1046                 return;
1047         }
1048         dec_nsec = (vcpu->arch.dec_expires - now) * NSEC_PER_SEC
1049                    / tb_ticks_per_sec;
1050         hrtimer_start(&vcpu->arch.dec_timer, ktime_set(0, dec_nsec),
1051                       HRTIMER_MODE_REL);
1052         vcpu->arch.timer_running = 1;
1053 }
1054
1055 static void kvmppc_end_cede(struct kvm_vcpu *vcpu)
1056 {
1057         vcpu->arch.ceded = 0;
1058         if (vcpu->arch.timer_running) {
1059                 hrtimer_try_to_cancel(&vcpu->arch.dec_timer);
1060                 vcpu->arch.timer_running = 0;
1061         }
1062 }
1063
1064 extern int __kvmppc_vcore_entry(struct kvm_run *kvm_run, struct kvm_vcpu *vcpu);
1065
1066 static void kvmppc_remove_runnable(struct kvmppc_vcore *vc,
1067                                    struct kvm_vcpu *vcpu)
1068 {
1069         u64 now;
1070
1071         if (vcpu->arch.state != KVMPPC_VCPU_RUNNABLE)
1072                 return;
1073         spin_lock(&vcpu->arch.tbacct_lock);
1074         now = mftb();
1075         vcpu->arch.busy_stolen += vcore_stolen_time(vc, now) -
1076                 vcpu->arch.stolen_logged;
1077         vcpu->arch.busy_preempt = now;
1078         vcpu->arch.state = KVMPPC_VCPU_BUSY_IN_HOST;
1079         spin_unlock(&vcpu->arch.tbacct_lock);
1080         --vc->n_runnable;
1081         list_del(&vcpu->arch.run_list);
1082 }
1083
1084 static int kvmppc_grab_hwthread(int cpu)
1085 {
1086         struct paca_struct *tpaca;
1087         long timeout = 1000;
1088
1089         tpaca = &paca[cpu];
1090
1091         /* Ensure the thread won't go into the kernel if it wakes */
1092         tpaca->kvm_hstate.hwthread_req = 1;
1093         tpaca->kvm_hstate.kvm_vcpu = NULL;
1094
1095         /*
1096          * If the thread is already executing in the kernel (e.g. handling
1097          * a stray interrupt), wait for it to get back to nap mode.
1098          * The smp_mb() is to ensure that our setting of hwthread_req
1099          * is visible before we look at hwthread_state, so if this
1100          * races with the code at system_reset_pSeries and the thread
1101          * misses our setting of hwthread_req, we are sure to see its
1102          * setting of hwthread_state, and vice versa.
1103          */
1104         smp_mb();
1105         while (tpaca->kvm_hstate.hwthread_state == KVM_HWTHREAD_IN_KERNEL) {
1106                 if (--timeout <= 0) {
1107                         pr_err("KVM: couldn't grab cpu %d\n", cpu);
1108                         return -EBUSY;
1109                 }
1110                 udelay(1);
1111         }
1112         return 0;
1113 }
1114
1115 static void kvmppc_release_hwthread(int cpu)
1116 {
1117         struct paca_struct *tpaca;
1118
1119         tpaca = &paca[cpu];
1120         tpaca->kvm_hstate.hwthread_req = 0;
1121         tpaca->kvm_hstate.kvm_vcpu = NULL;
1122 }
1123
1124 static void kvmppc_start_thread(struct kvm_vcpu *vcpu)
1125 {
1126         int cpu;
1127         struct paca_struct *tpaca;
1128         struct kvmppc_vcore *vc = vcpu->arch.vcore;
1129
1130         if (vcpu->arch.timer_running) {
1131                 hrtimer_try_to_cancel(&vcpu->arch.dec_timer);
1132                 vcpu->arch.timer_running = 0;
1133         }
1134         cpu = vc->pcpu + vcpu->arch.ptid;
1135         tpaca = &paca[cpu];
1136         tpaca->kvm_hstate.kvm_vcpu = vcpu;
1137         tpaca->kvm_hstate.kvm_vcore = vc;
1138         tpaca->kvm_hstate.napping = 0;
1139         vcpu->cpu = vc->pcpu;
1140         smp_wmb();
1141 #if defined(CONFIG_PPC_ICP_NATIVE) && defined(CONFIG_SMP)
1142         if (vcpu->arch.ptid) {
1143                 xics_wake_cpu(cpu);
1144                 ++vc->n_woken;
1145         }
1146 #endif
1147 }
1148
1149 static void kvmppc_wait_for_nap(struct kvmppc_vcore *vc)
1150 {
1151         int i;
1152
1153         HMT_low();
1154         i = 0;
1155         while (vc->nap_count < vc->n_woken) {
1156                 if (++i >= 1000000) {
1157                         pr_err("kvmppc_wait_for_nap timeout %d %d\n",
1158                                vc->nap_count, vc->n_woken);
1159                         break;
1160                 }
1161                 cpu_relax();
1162         }
1163         HMT_medium();
1164 }
1165
1166 /*
1167  * Check that we are on thread 0 and that any other threads in
1168  * this core are off-line.  Then grab the threads so they can't
1169  * enter the kernel.
1170  */
1171 static int on_primary_thread(void)
1172 {
1173         int cpu = smp_processor_id();
1174         int thr = cpu_thread_in_core(cpu);
1175
1176         if (thr)
1177                 return 0;
1178         while (++thr < threads_per_core)
1179                 if (cpu_online(cpu + thr))
1180                         return 0;
1181
1182         /* Grab all hw threads so they can't go into the kernel */
1183         for (thr = 1; thr < threads_per_core; ++thr) {
1184                 if (kvmppc_grab_hwthread(cpu + thr)) {
1185                         /* Couldn't grab one; let the others go */
1186                         do {
1187                                 kvmppc_release_hwthread(cpu + thr);
1188                         } while (--thr > 0);
1189                         return 0;
1190                 }
1191         }
1192         return 1;
1193 }
1194
1195 /*
1196  * Run a set of guest threads on a physical core.
1197  * Called with vc->lock held.
1198  */
1199 static void kvmppc_run_core(struct kvmppc_vcore *vc)
1200 {
1201         struct kvm_vcpu *vcpu, *vcpu0, *vnext;
1202         long ret;
1203         u64 now;
1204         int ptid, i, need_vpa_update;
1205         int srcu_idx;
1206         struct kvm_vcpu *vcpus_to_update[threads_per_core];
1207
1208         /* don't start if any threads have a signal pending */
1209         need_vpa_update = 0;
1210         list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
1211                 if (signal_pending(vcpu->arch.run_task))
1212                         return;
1213                 if (vcpu->arch.vpa.update_pending ||
1214                     vcpu->arch.slb_shadow.update_pending ||
1215                     vcpu->arch.dtl.update_pending)
1216                         vcpus_to_update[need_vpa_update++] = vcpu;
1217         }
1218
1219         /*
1220          * Initialize *vc, in particular vc->vcore_state, so we can
1221          * drop the vcore lock if necessary.
1222          */
1223         vc->n_woken = 0;
1224         vc->nap_count = 0;
1225         vc->entry_exit_count = 0;
1226         vc->vcore_state = VCORE_STARTING;
1227         vc->in_guest = 0;
1228         vc->napping_threads = 0;
1229
1230         /*
1231          * Updating any of the vpas requires calling kvmppc_pin_guest_page,
1232          * which can't be called with any spinlocks held.
1233          */
1234         if (need_vpa_update) {
1235                 spin_unlock(&vc->lock);
1236                 for (i = 0; i < need_vpa_update; ++i)
1237                         kvmppc_update_vpas(vcpus_to_update[i]);
1238                 spin_lock(&vc->lock);
1239         }
1240
1241         /*
1242          * Assign physical thread IDs, first to non-ceded vcpus
1243          * and then to ceded ones.
1244          */
1245         ptid = 0;
1246         vcpu0 = NULL;
1247         list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
1248                 if (!vcpu->arch.ceded) {
1249                         if (!ptid)
1250                                 vcpu0 = vcpu;
1251                         vcpu->arch.ptid = ptid++;
1252                 }
1253         }
1254         if (!vcpu0)
1255                 goto out;       /* nothing to run; should never happen */
1256         list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list)
1257                 if (vcpu->arch.ceded)
1258                         vcpu->arch.ptid = ptid++;
1259
1260         /*
1261          * Make sure we are running on thread 0, and that
1262          * secondary threads are offline.
1263          */
1264         if (threads_per_core > 1 && !on_primary_thread()) {
1265                 list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list)
1266                         vcpu->arch.ret = -EBUSY;
1267                 goto out;
1268         }
1269
1270         vc->pcpu = smp_processor_id();
1271         list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
1272                 kvmppc_start_thread(vcpu);
1273                 kvmppc_create_dtl_entry(vcpu, vc);
1274         }
1275
1276         vc->vcore_state = VCORE_RUNNING;
1277         preempt_disable();
1278         spin_unlock(&vc->lock);
1279
1280         kvm_guest_enter();
1281
1282         srcu_idx = srcu_read_lock(&vcpu0->kvm->srcu);
1283
1284         __kvmppc_vcore_entry(NULL, vcpu0);
1285
1286         spin_lock(&vc->lock);
1287         /* disable sending of IPIs on virtual external irqs */
1288         list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list)
1289                 vcpu->cpu = -1;
1290         /* wait for secondary threads to finish writing their state to memory */
1291         if (vc->nap_count < vc->n_woken)
1292                 kvmppc_wait_for_nap(vc);
1293         for (i = 0; i < threads_per_core; ++i)
1294                 kvmppc_release_hwthread(vc->pcpu + i);
1295         /* prevent other vcpu threads from doing kvmppc_start_thread() now */
1296         vc->vcore_state = VCORE_EXITING;
1297         spin_unlock(&vc->lock);
1298
1299         srcu_read_unlock(&vcpu0->kvm->srcu, srcu_idx);
1300
1301         /* make sure updates to secondary vcpu structs are visible now */
1302         smp_mb();
1303         kvm_guest_exit();
1304
1305         preempt_enable();
1306         kvm_resched(vcpu);
1307
1308         spin_lock(&vc->lock);
1309         now = get_tb();
1310         list_for_each_entry(vcpu, &vc->runnable_threads, arch.run_list) {
1311                 /* cancel pending dec exception if dec is positive */
1312                 if (now < vcpu->arch.dec_expires &&
1313                     kvmppc_core_pending_dec(vcpu))
1314                         kvmppc_core_dequeue_dec(vcpu);
1315
1316                 ret = RESUME_GUEST;
1317                 if (vcpu->arch.trap)
1318                         ret = kvmppc_handle_exit(vcpu->arch.kvm_run, vcpu,
1319                                                  vcpu->arch.run_task);
1320
1321                 vcpu->arch.ret = ret;
1322                 vcpu->arch.trap = 0;
1323
1324                 if (vcpu->arch.ceded) {
1325                         if (ret != RESUME_GUEST)
1326                                 kvmppc_end_cede(vcpu);
1327                         else
1328                                 kvmppc_set_timer(vcpu);
1329                 }
1330         }
1331
1332  out:
1333         vc->vcore_state = VCORE_INACTIVE;
1334         list_for_each_entry_safe(vcpu, vnext, &vc->runnable_threads,
1335                                  arch.run_list) {
1336                 if (vcpu->arch.ret != RESUME_GUEST) {
1337                         kvmppc_remove_runnable(vc, vcpu);
1338                         wake_up(&vcpu->arch.cpu_run);
1339                 }
1340         }
1341 }
1342
1343 /*
1344  * Wait for some other vcpu thread to execute us, and
1345  * wake us up when we need to handle something in the host.
1346  */
1347 static void kvmppc_wait_for_exec(struct kvm_vcpu *vcpu, int wait_state)
1348 {
1349         DEFINE_WAIT(wait);
1350
1351         prepare_to_wait(&vcpu->arch.cpu_run, &wait, wait_state);
1352         if (vcpu->arch.state == KVMPPC_VCPU_RUNNABLE)
1353                 schedule();
1354         finish_wait(&vcpu->arch.cpu_run, &wait);
1355 }
1356
1357 /*
1358  * All the vcpus in this vcore are idle, so wait for a decrementer
1359  * or external interrupt to one of the vcpus.  vc->lock is held.
1360  */
1361 static void kvmppc_vcore_blocked(struct kvmppc_vcore *vc)
1362 {
1363         DEFINE_WAIT(wait);
1364
1365         prepare_to_wait(&vc->wq, &wait, TASK_INTERRUPTIBLE);
1366         vc->vcore_state = VCORE_SLEEPING;
1367         spin_unlock(&vc->lock);
1368         schedule();
1369         finish_wait(&vc->wq, &wait);
1370         spin_lock(&vc->lock);
1371         vc->vcore_state = VCORE_INACTIVE;
1372 }
1373
1374 static int kvmppc_run_vcpu(struct kvm_run *kvm_run, struct kvm_vcpu *vcpu)
1375 {
1376         int n_ceded;
1377         struct kvmppc_vcore *vc;
1378         struct kvm_vcpu *v, *vn;
1379
1380         kvm_run->exit_reason = 0;
1381         vcpu->arch.ret = RESUME_GUEST;
1382         vcpu->arch.trap = 0;
1383         kvmppc_update_vpas(vcpu);
1384
1385         /*
1386          * Synchronize with other threads in this virtual core
1387          */
1388         vc = vcpu->arch.vcore;
1389         spin_lock(&vc->lock);
1390         vcpu->arch.ceded = 0;
1391         vcpu->arch.run_task = current;
1392         vcpu->arch.kvm_run = kvm_run;
1393         vcpu->arch.stolen_logged = vcore_stolen_time(vc, mftb());
1394         vcpu->arch.state = KVMPPC_VCPU_RUNNABLE;
1395         vcpu->arch.busy_preempt = TB_NIL;
1396         list_add_tail(&vcpu->arch.run_list, &vc->runnable_threads);
1397         ++vc->n_runnable;
1398
1399         /*
1400          * This happens the first time this is called for a vcpu.
1401          * If the vcore is already running, we may be able to start
1402          * this thread straight away and have it join in.
1403          */
1404         if (!signal_pending(current)) {
1405                 if (vc->vcore_state == VCORE_RUNNING &&
1406                     VCORE_EXIT_COUNT(vc) == 0) {
1407                         vcpu->arch.ptid = vc->n_runnable - 1;
1408                         kvmppc_create_dtl_entry(vcpu, vc);
1409                         kvmppc_start_thread(vcpu);
1410                 } else if (vc->vcore_state == VCORE_SLEEPING) {
1411                         wake_up(&vc->wq);
1412                 }
1413
1414         }
1415
1416         while (vcpu->arch.state == KVMPPC_VCPU_RUNNABLE &&
1417                !signal_pending(current)) {
1418                 if (vc->vcore_state != VCORE_INACTIVE) {
1419                         spin_unlock(&vc->lock);
1420                         kvmppc_wait_for_exec(vcpu, TASK_INTERRUPTIBLE);
1421                         spin_lock(&vc->lock);
1422                         continue;
1423                 }
1424                 list_for_each_entry_safe(v, vn, &vc->runnable_threads,
1425                                          arch.run_list) {
1426                         kvmppc_core_prepare_to_enter(v);
1427                         if (signal_pending(v->arch.run_task)) {
1428                                 kvmppc_remove_runnable(vc, v);
1429                                 v->stat.signal_exits++;
1430                                 v->arch.kvm_run->exit_reason = KVM_EXIT_INTR;
1431                                 v->arch.ret = -EINTR;
1432                                 wake_up(&v->arch.cpu_run);
1433                         }
1434                 }
1435                 if (!vc->n_runnable || vcpu->arch.state != KVMPPC_VCPU_RUNNABLE)
1436                         break;
1437                 vc->runner = vcpu;
1438                 n_ceded = 0;
1439                 list_for_each_entry(v, &vc->runnable_threads, arch.run_list) {
1440                         if (!v->arch.pending_exceptions)
1441                                 n_ceded += v->arch.ceded;
1442                         else
1443                                 v->arch.ceded = 0;
1444                 }
1445                 if (n_ceded == vc->n_runnable)
1446                         kvmppc_vcore_blocked(vc);
1447                 else
1448                         kvmppc_run_core(vc);
1449                 vc->runner = NULL;
1450         }
1451
1452         while (vcpu->arch.state == KVMPPC_VCPU_RUNNABLE &&
1453                (vc->vcore_state == VCORE_RUNNING ||
1454                 vc->vcore_state == VCORE_EXITING)) {
1455                 spin_unlock(&vc->lock);
1456                 kvmppc_wait_for_exec(vcpu, TASK_UNINTERRUPTIBLE);
1457                 spin_lock(&vc->lock);
1458         }
1459
1460         if (vcpu->arch.state == KVMPPC_VCPU_RUNNABLE) {
1461                 kvmppc_remove_runnable(vc, vcpu);
1462                 vcpu->stat.signal_exits++;
1463                 kvm_run->exit_reason = KVM_EXIT_INTR;
1464                 vcpu->arch.ret = -EINTR;
1465         }
1466
1467         if (vc->n_runnable && vc->vcore_state == VCORE_INACTIVE) {
1468                 /* Wake up some vcpu to run the core */
1469                 v = list_first_entry(&vc->runnable_threads,
1470                                      struct kvm_vcpu, arch.run_list);
1471                 wake_up(&v->arch.cpu_run);
1472         }
1473
1474         spin_unlock(&vc->lock);
1475         return vcpu->arch.ret;
1476 }
1477
1478 int kvmppc_vcpu_run(struct kvm_run *run, struct kvm_vcpu *vcpu)
1479 {
1480         int r;
1481         int srcu_idx;
1482
1483         if (!vcpu->arch.sane) {
1484                 run->exit_reason = KVM_EXIT_INTERNAL_ERROR;
1485                 return -EINVAL;
1486         }
1487
1488         kvmppc_core_prepare_to_enter(vcpu);
1489
1490         /* No need to go into the guest when all we'll do is come back out */
1491         if (signal_pending(current)) {
1492                 run->exit_reason = KVM_EXIT_INTR;
1493                 return -EINTR;
1494         }
1495
1496         atomic_inc(&vcpu->kvm->arch.vcpus_running);
1497         /* Order vcpus_running vs. rma_setup_done, see kvmppc_alloc_reset_hpt */
1498         smp_mb();
1499
1500         /* On the first time here, set up HTAB and VRMA or RMA */
1501         if (!vcpu->kvm->arch.rma_setup_done) {
1502                 r = kvmppc_hv_setup_htab_rma(vcpu);
1503                 if (r)
1504                         goto out;
1505         }
1506
1507         flush_fp_to_thread(current);
1508         flush_altivec_to_thread(current);
1509         flush_vsx_to_thread(current);
1510         vcpu->arch.wqp = &vcpu->arch.vcore->wq;
1511         vcpu->arch.pgdir = current->mm->pgd;
1512         vcpu->arch.state = KVMPPC_VCPU_BUSY_IN_HOST;
1513
1514         do {
1515                 r = kvmppc_run_vcpu(run, vcpu);
1516
1517                 if (run->exit_reason == KVM_EXIT_PAPR_HCALL &&
1518                     !(vcpu->arch.shregs.msr & MSR_PR)) {
1519                         r = kvmppc_pseries_do_hcall(vcpu);
1520                         kvmppc_core_prepare_to_enter(vcpu);
1521                 } else if (r == RESUME_PAGE_FAULT) {
1522                         srcu_idx = srcu_read_lock(&vcpu->kvm->srcu);
1523                         r = kvmppc_book3s_hv_page_fault(run, vcpu,
1524                                 vcpu->arch.fault_dar, vcpu->arch.fault_dsisr);
1525                         srcu_read_unlock(&vcpu->kvm->srcu, srcu_idx);
1526                 }
1527         } while (r == RESUME_GUEST);
1528
1529  out:
1530         vcpu->arch.state = KVMPPC_VCPU_NOTREADY;
1531         atomic_dec(&vcpu->kvm->arch.vcpus_running);
1532         return r;
1533 }
1534
1535
1536 /* Work out RMLS (real mode limit selector) field value for a given RMA size.
1537    Assumes POWER7 or PPC970. */
1538 static inline int lpcr_rmls(unsigned long rma_size)
1539 {
1540         switch (rma_size) {
1541         case 32ul << 20:        /* 32 MB */
1542                 if (cpu_has_feature(CPU_FTR_ARCH_206))
1543                         return 8;       /* only supported on POWER7 */
1544                 return -1;
1545         case 64ul << 20:        /* 64 MB */
1546                 return 3;
1547         case 128ul << 20:       /* 128 MB */
1548                 return 7;
1549         case 256ul << 20:       /* 256 MB */
1550                 return 4;
1551         case 1ul << 30:         /* 1 GB */
1552                 return 2;
1553         case 16ul << 30:        /* 16 GB */
1554                 return 1;
1555         case 256ul << 30:       /* 256 GB */
1556                 return 0;
1557         default:
1558                 return -1;
1559         }
1560 }
1561
1562 static int kvm_rma_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
1563 {
1564         struct page *page;
1565         struct kvm_rma_info *ri = vma->vm_file->private_data;
1566
1567         if (vmf->pgoff >= kvm_rma_pages)
1568                 return VM_FAULT_SIGBUS;
1569
1570         page = pfn_to_page(ri->base_pfn + vmf->pgoff);
1571         get_page(page);
1572         vmf->page = page;
1573         return 0;
1574 }
1575
1576 static const struct vm_operations_struct kvm_rma_vm_ops = {
1577         .fault = kvm_rma_fault,
1578 };
1579
1580 static int kvm_rma_mmap(struct file *file, struct vm_area_struct *vma)
1581 {
1582         vma->vm_flags |= VM_DONTEXPAND | VM_DONTDUMP;
1583         vma->vm_ops = &kvm_rma_vm_ops;
1584         return 0;
1585 }
1586
1587 static int kvm_rma_release(struct inode *inode, struct file *filp)
1588 {
1589         struct kvm_rma_info *ri = filp->private_data;
1590
1591         kvm_release_rma(ri);
1592         return 0;
1593 }
1594
1595 static const struct file_operations kvm_rma_fops = {
1596         .mmap           = kvm_rma_mmap,
1597         .release        = kvm_rma_release,
1598 };
1599
1600 long kvm_vm_ioctl_allocate_rma(struct kvm *kvm, struct kvm_allocate_rma *ret)
1601 {
1602         long fd;
1603         struct kvm_rma_info *ri;
1604         /*
1605          * Only do this on PPC970 in HV mode
1606          */
1607         if (!cpu_has_feature(CPU_FTR_HVMODE) ||
1608             !cpu_has_feature(CPU_FTR_ARCH_201))
1609                 return -EINVAL;
1610
1611         if (!kvm_rma_pages)
1612                 return -EINVAL;
1613
1614         ri = kvm_alloc_rma();
1615         if (!ri)
1616                 return -ENOMEM;
1617
1618         fd = anon_inode_getfd("kvm-rma", &kvm_rma_fops, ri, O_RDWR | O_CLOEXEC);
1619         if (fd < 0)
1620                 kvm_release_rma(ri);
1621
1622         ret->rma_size = kvm_rma_pages << PAGE_SHIFT;
1623         return fd;
1624 }
1625
1626 static void kvmppc_add_seg_page_size(struct kvm_ppc_one_seg_page_size **sps,
1627                                      int linux_psize)
1628 {
1629         struct mmu_psize_def *def = &mmu_psize_defs[linux_psize];
1630
1631         if (!def->shift)
1632                 return;
1633         (*sps)->page_shift = def->shift;
1634         (*sps)->slb_enc = def->sllp;
1635         (*sps)->enc[0].page_shift = def->shift;
1636         /*
1637          * Only return base page encoding. We don't want to return
1638          * all the supporting pte_enc, because our H_ENTER doesn't
1639          * support MPSS yet. Once they do, we can start passing all
1640          * support pte_enc here
1641          */
1642         (*sps)->enc[0].pte_enc = def->penc[linux_psize];
1643         (*sps)++;
1644 }
1645
1646 int kvm_vm_ioctl_get_smmu_info(struct kvm *kvm, struct kvm_ppc_smmu_info *info)
1647 {
1648         struct kvm_ppc_one_seg_page_size *sps;
1649
1650         info->flags = KVM_PPC_PAGE_SIZES_REAL;
1651         if (mmu_has_feature(MMU_FTR_1T_SEGMENT))
1652                 info->flags |= KVM_PPC_1T_SEGMENTS;
1653         info->slb_size = mmu_slb_size;
1654
1655         /* We only support these sizes for now, and no muti-size segments */
1656         sps = &info->sps[0];
1657         kvmppc_add_seg_page_size(&sps, MMU_PAGE_4K);
1658         kvmppc_add_seg_page_size(&sps, MMU_PAGE_64K);
1659         kvmppc_add_seg_page_size(&sps, MMU_PAGE_16M);
1660
1661         return 0;
1662 }
1663
1664 /*
1665  * Get (and clear) the dirty memory log for a memory slot.
1666  */
1667 int kvm_vm_ioctl_get_dirty_log(struct kvm *kvm, struct kvm_dirty_log *log)
1668 {
1669         struct kvm_memory_slot *memslot;
1670         int r;
1671         unsigned long n;
1672
1673         mutex_lock(&kvm->slots_lock);
1674
1675         r = -EINVAL;
1676         if (log->slot >= KVM_USER_MEM_SLOTS)
1677                 goto out;
1678
1679         memslot = id_to_memslot(kvm->memslots, log->slot);
1680         r = -ENOENT;
1681         if (!memslot->dirty_bitmap)
1682                 goto out;
1683
1684         n = kvm_dirty_bitmap_bytes(memslot);
1685         memset(memslot->dirty_bitmap, 0, n);
1686
1687         r = kvmppc_hv_get_dirty_log(kvm, memslot, memslot->dirty_bitmap);
1688         if (r)
1689                 goto out;
1690
1691         r = -EFAULT;
1692         if (copy_to_user(log->dirty_bitmap, memslot->dirty_bitmap, n))
1693                 goto out;
1694
1695         r = 0;
1696 out:
1697         mutex_unlock(&kvm->slots_lock);
1698         return r;
1699 }
1700
1701 static void unpin_slot(struct kvm_memory_slot *memslot)
1702 {
1703         unsigned long *physp;
1704         unsigned long j, npages, pfn;
1705         struct page *page;
1706
1707         physp = memslot->arch.slot_phys;
1708         npages = memslot->npages;
1709         if (!physp)
1710                 return;
1711         for (j = 0; j < npages; j++) {
1712                 if (!(physp[j] & KVMPPC_GOT_PAGE))
1713                         continue;
1714                 pfn = physp[j] >> PAGE_SHIFT;
1715                 page = pfn_to_page(pfn);
1716                 SetPageDirty(page);
1717                 put_page(page);
1718         }
1719 }
1720
1721 void kvmppc_core_free_memslot(struct kvm_memory_slot *free,
1722                               struct kvm_memory_slot *dont)
1723 {
1724         if (!dont || free->arch.rmap != dont->arch.rmap) {
1725                 vfree(free->arch.rmap);
1726                 free->arch.rmap = NULL;
1727         }
1728         if (!dont || free->arch.slot_phys != dont->arch.slot_phys) {
1729                 unpin_slot(free);
1730                 vfree(free->arch.slot_phys);
1731                 free->arch.slot_phys = NULL;
1732         }
1733 }
1734
1735 int kvmppc_core_create_memslot(struct kvm_memory_slot *slot,
1736                                unsigned long npages)
1737 {
1738         slot->arch.rmap = vzalloc(npages * sizeof(*slot->arch.rmap));
1739         if (!slot->arch.rmap)
1740                 return -ENOMEM;
1741         slot->arch.slot_phys = NULL;
1742
1743         return 0;
1744 }
1745
1746 int kvmppc_core_prepare_memory_region(struct kvm *kvm,
1747                                       struct kvm_memory_slot *memslot,
1748                                       struct kvm_userspace_memory_region *mem)
1749 {
1750         unsigned long *phys;
1751
1752         /* Allocate a slot_phys array if needed */
1753         phys = memslot->arch.slot_phys;
1754         if (!kvm->arch.using_mmu_notifiers && !phys && memslot->npages) {
1755                 phys = vzalloc(memslot->npages * sizeof(unsigned long));
1756                 if (!phys)
1757                         return -ENOMEM;
1758                 memslot->arch.slot_phys = phys;
1759         }
1760
1761         return 0;
1762 }
1763
1764 void kvmppc_core_commit_memory_region(struct kvm *kvm,
1765                                       struct kvm_userspace_memory_region *mem,
1766                                       const struct kvm_memory_slot *old)
1767 {
1768         unsigned long npages = mem->memory_size >> PAGE_SHIFT;
1769         struct kvm_memory_slot *memslot;
1770
1771         if (npages && old->npages) {
1772                 /*
1773                  * If modifying a memslot, reset all the rmap dirty bits.
1774                  * If this is a new memslot, we don't need to do anything
1775                  * since the rmap array starts out as all zeroes,
1776                  * i.e. no pages are dirty.
1777                  */
1778                 memslot = id_to_memslot(kvm->memslots, mem->slot);
1779                 kvmppc_hv_get_dirty_log(kvm, memslot, NULL);
1780         }
1781 }
1782
1783 /*
1784  * Update LPCR values in kvm->arch and in vcores.
1785  * Caller must hold kvm->lock.
1786  */
1787 void kvmppc_update_lpcr(struct kvm *kvm, unsigned long lpcr, unsigned long mask)
1788 {
1789         long int i;
1790         u32 cores_done = 0;
1791
1792         if ((kvm->arch.lpcr & mask) == lpcr)
1793                 return;
1794
1795         kvm->arch.lpcr = (kvm->arch.lpcr & ~mask) | lpcr;
1796
1797         for (i = 0; i < KVM_MAX_VCORES; ++i) {
1798                 struct kvmppc_vcore *vc = kvm->arch.vcores[i];
1799                 if (!vc)
1800                         continue;
1801                 spin_lock(&vc->lock);
1802                 vc->lpcr = (vc->lpcr & ~mask) | lpcr;
1803                 spin_unlock(&vc->lock);
1804                 if (++cores_done >= kvm->arch.online_vcores)
1805                         break;
1806         }
1807 }
1808
1809 static int kvmppc_hv_setup_htab_rma(struct kvm_vcpu *vcpu)
1810 {
1811         int err = 0;
1812         struct kvm *kvm = vcpu->kvm;
1813         struct kvm_rma_info *ri = NULL;
1814         unsigned long hva;
1815         struct kvm_memory_slot *memslot;
1816         struct vm_area_struct *vma;
1817         unsigned long lpcr = 0, senc;
1818         unsigned long lpcr_mask = 0;
1819         unsigned long psize, porder;
1820         unsigned long rma_size;
1821         unsigned long rmls;
1822         unsigned long *physp;
1823         unsigned long i, npages;
1824         int srcu_idx;
1825
1826         mutex_lock(&kvm->lock);
1827         if (kvm->arch.rma_setup_done)
1828                 goto out;       /* another vcpu beat us to it */
1829
1830         /* Allocate hashed page table (if not done already) and reset it */
1831         if (!kvm->arch.hpt_virt) {
1832                 err = kvmppc_alloc_hpt(kvm, NULL);
1833                 if (err) {
1834                         pr_err("KVM: Couldn't alloc HPT\n");
1835                         goto out;
1836                 }
1837         }
1838
1839         /* Look up the memslot for guest physical address 0 */
1840         srcu_idx = srcu_read_lock(&kvm->srcu);
1841         memslot = gfn_to_memslot(kvm, 0);
1842
1843         /* We must have some memory at 0 by now */
1844         err = -EINVAL;
1845         if (!memslot || (memslot->flags & KVM_MEMSLOT_INVALID))
1846                 goto out_srcu;
1847
1848         /* Look up the VMA for the start of this memory slot */
1849         hva = memslot->userspace_addr;
1850         down_read(&current->mm->mmap_sem);
1851         vma = find_vma(current->mm, hva);
1852         if (!vma || vma->vm_start > hva || (vma->vm_flags & VM_IO))
1853                 goto up_out;
1854
1855         psize = vma_kernel_pagesize(vma);
1856         porder = __ilog2(psize);
1857
1858         /* Is this one of our preallocated RMAs? */
1859         if (vma->vm_file && vma->vm_file->f_op == &kvm_rma_fops &&
1860             hva == vma->vm_start)
1861                 ri = vma->vm_file->private_data;
1862
1863         up_read(&current->mm->mmap_sem);
1864
1865         if (!ri) {
1866                 /* On POWER7, use VRMA; on PPC970, give up */
1867                 err = -EPERM;
1868                 if (cpu_has_feature(CPU_FTR_ARCH_201)) {
1869                         pr_err("KVM: CPU requires an RMO\n");
1870                         goto out_srcu;
1871                 }
1872
1873                 /* We can handle 4k, 64k or 16M pages in the VRMA */
1874                 err = -EINVAL;
1875                 if (!(psize == 0x1000 || psize == 0x10000 ||
1876                       psize == 0x1000000))
1877                         goto out_srcu;
1878
1879                 /* Update VRMASD field in the LPCR */
1880                 senc = slb_pgsize_encoding(psize);
1881                 kvm->arch.vrma_slb_v = senc | SLB_VSID_B_1T |
1882                         (VRMA_VSID << SLB_VSID_SHIFT_1T);
1883                 lpcr_mask = LPCR_VRMASD;
1884                 /* the -4 is to account for senc values starting at 0x10 */
1885                 lpcr = senc << (LPCR_VRMASD_SH - 4);
1886
1887                 /* Create HPTEs in the hash page table for the VRMA */
1888                 kvmppc_map_vrma(vcpu, memslot, porder);
1889
1890         } else {
1891                 /* Set up to use an RMO region */
1892                 rma_size = kvm_rma_pages;
1893                 if (rma_size > memslot->npages)
1894                         rma_size = memslot->npages;
1895                 rma_size <<= PAGE_SHIFT;
1896                 rmls = lpcr_rmls(rma_size);
1897                 err = -EINVAL;
1898                 if ((long)rmls < 0) {
1899                         pr_err("KVM: Can't use RMA of 0x%lx bytes\n", rma_size);
1900                         goto out_srcu;
1901                 }
1902                 atomic_inc(&ri->use_count);
1903                 kvm->arch.rma = ri;
1904
1905                 /* Update LPCR and RMOR */
1906                 if (cpu_has_feature(CPU_FTR_ARCH_201)) {
1907                         /* PPC970; insert RMLS value (split field) in HID4 */
1908                         lpcr_mask = (1ul << HID4_RMLS0_SH) |
1909                                 (3ul << HID4_RMLS2_SH) | HID4_RMOR;
1910                         lpcr = ((rmls >> 2) << HID4_RMLS0_SH) |
1911                                 ((rmls & 3) << HID4_RMLS2_SH);
1912                         /* RMOR is also in HID4 */
1913                         lpcr |= ((ri->base_pfn >> (26 - PAGE_SHIFT)) & 0xffff)
1914                                 << HID4_RMOR_SH;
1915                 } else {
1916                         /* POWER7 */
1917                         lpcr_mask = LPCR_VPM0 | LPCR_VRMA_L | LPCR_RMLS;
1918                         lpcr = rmls << LPCR_RMLS_SH;
1919                         kvm->arch.rmor = ri->base_pfn << PAGE_SHIFT;
1920                 }
1921                 pr_info("KVM: Using RMO at %lx size %lx (LPCR = %lx)\n",
1922                         ri->base_pfn << PAGE_SHIFT, rma_size, lpcr);
1923
1924                 /* Initialize phys addrs of pages in RMO */
1925                 npages = kvm_rma_pages;
1926                 porder = __ilog2(npages);
1927                 physp = memslot->arch.slot_phys;
1928                 if (physp) {
1929                         if (npages > memslot->npages)
1930                                 npages = memslot->npages;
1931                         spin_lock(&kvm->arch.slot_phys_lock);
1932                         for (i = 0; i < npages; ++i)
1933                                 physp[i] = ((ri->base_pfn + i) << PAGE_SHIFT) +
1934                                         porder;
1935                         spin_unlock(&kvm->arch.slot_phys_lock);
1936                 }
1937         }
1938
1939         kvmppc_update_lpcr(kvm, lpcr, lpcr_mask);
1940
1941         /* Order updates to kvm->arch.lpcr etc. vs. rma_setup_done */
1942         smp_wmb();
1943         kvm->arch.rma_setup_done = 1;
1944         err = 0;
1945  out_srcu:
1946         srcu_read_unlock(&kvm->srcu, srcu_idx);
1947  out:
1948         mutex_unlock(&kvm->lock);
1949         return err;
1950
1951  up_out:
1952         up_read(&current->mm->mmap_sem);
1953         goto out_srcu;
1954 }
1955
1956 int kvmppc_core_init_vm(struct kvm *kvm)
1957 {
1958         unsigned long lpcr, lpid;
1959
1960         /* Allocate the guest's logical partition ID */
1961
1962         lpid = kvmppc_alloc_lpid();
1963         if ((long)lpid < 0)
1964                 return -ENOMEM;
1965         kvm->arch.lpid = lpid;
1966
1967         /*
1968          * Since we don't flush the TLB when tearing down a VM,
1969          * and this lpid might have previously been used,
1970          * make sure we flush on each core before running the new VM.
1971          */
1972         cpumask_setall(&kvm->arch.need_tlb_flush);
1973
1974         INIT_LIST_HEAD(&kvm->arch.spapr_tce_tables);
1975         INIT_LIST_HEAD(&kvm->arch.rtas_tokens);
1976
1977         kvm->arch.rma = NULL;
1978
1979         kvm->arch.host_sdr1 = mfspr(SPRN_SDR1);
1980
1981         if (cpu_has_feature(CPU_FTR_ARCH_201)) {
1982                 /* PPC970; HID4 is effectively the LPCR */
1983                 kvm->arch.host_lpid = 0;
1984                 kvm->arch.host_lpcr = lpcr = mfspr(SPRN_HID4);
1985                 lpcr &= ~((3 << HID4_LPID1_SH) | (0xful << HID4_LPID5_SH));
1986                 lpcr |= ((lpid >> 4) << HID4_LPID1_SH) |
1987                         ((lpid & 0xf) << HID4_LPID5_SH);
1988         } else {
1989                 /* POWER7; init LPCR for virtual RMA mode */
1990                 kvm->arch.host_lpid = mfspr(SPRN_LPID);
1991                 kvm->arch.host_lpcr = lpcr = mfspr(SPRN_LPCR);
1992                 lpcr &= LPCR_PECE | LPCR_LPES;
1993                 lpcr |= (4UL << LPCR_DPFD_SH) | LPCR_HDICE |
1994                         LPCR_VPM0 | LPCR_VPM1;
1995                 kvm->arch.vrma_slb_v = SLB_VSID_B_1T |
1996                         (VRMA_VSID << SLB_VSID_SHIFT_1T);
1997         }
1998         kvm->arch.lpcr = lpcr;
1999
2000         kvm->arch.using_mmu_notifiers = !!cpu_has_feature(CPU_FTR_ARCH_206);
2001         spin_lock_init(&kvm->arch.slot_phys_lock);
2002
2003         /*
2004          * Don't allow secondary CPU threads to come online
2005          * while any KVM VMs exist.
2006          */
2007         inhibit_secondary_onlining();
2008
2009         return 0;
2010 }
2011
2012 void kvmppc_core_destroy_vm(struct kvm *kvm)
2013 {
2014         uninhibit_secondary_onlining();
2015
2016         if (kvm->arch.rma) {
2017                 kvm_release_rma(kvm->arch.rma);
2018                 kvm->arch.rma = NULL;
2019         }
2020
2021         kvmppc_rtas_tokens_free(kvm);
2022
2023         kvmppc_free_hpt(kvm);
2024         WARN_ON(!list_empty(&kvm->arch.spapr_tce_tables));
2025 }
2026
2027 /* These are stubs for now */
2028 void kvmppc_mmu_pte_pflush(struct kvm_vcpu *vcpu, ulong pa_start, ulong pa_end)
2029 {
2030 }
2031
2032 /* We don't need to emulate any privileged instructions or dcbz */
2033 int kvmppc_core_emulate_op(struct kvm_run *run, struct kvm_vcpu *vcpu,
2034                            unsigned int inst, int *advance)
2035 {
2036         return EMULATE_FAIL;
2037 }
2038
2039 int kvmppc_core_emulate_mtspr(struct kvm_vcpu *vcpu, int sprn, ulong spr_val)
2040 {
2041         return EMULATE_FAIL;
2042 }
2043
2044 int kvmppc_core_emulate_mfspr(struct kvm_vcpu *vcpu, int sprn, ulong *spr_val)
2045 {
2046         return EMULATE_FAIL;
2047 }
2048
2049 static int kvmppc_book3s_hv_init(void)
2050 {
2051         int r;
2052
2053         r = kvm_init(NULL, sizeof(struct kvm_vcpu), 0, THIS_MODULE);
2054
2055         if (r)
2056                 return r;
2057
2058         r = kvmppc_mmu_hv_init();
2059
2060         return r;
2061 }
2062
2063 static void kvmppc_book3s_hv_exit(void)
2064 {
2065         kvm_exit();
2066 }
2067
2068 module_init(kvmppc_book3s_hv_init);
2069 module_exit(kvmppc_book3s_hv_exit);