arch/powerpc/kvm/e500_mmu.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Copyright (C) 2008-2013 Freescale Semiconductor, Inc. All rights reserved.
   4  *
   5  * Author: Yu Liu, yu.liu@freescale.com
   6  *         Scott Wood, scottwood@freescale.com
   7  *         Ashish Kalra, ashish.kalra@freescale.com
   8  *         Varun Sethi, varun.sethi@freescale.com
   9  *         Alexander Graf, agraf@suse.de
  10  *
  11  * Description:
  12  * This file is based on arch/powerpc/kvm/44x_tlb.c,
  13  * by Hollis Blanchard <hollisb@us.ibm.com>.
  14  */
  15
  16 #include <linux/kernel.h>
  17 #include <linux/types.h>
  18 #include <linux/slab.h>
  19 #include <linux/string.h>
  20 #include <linux/kvm.h>
  21 #include <linux/kvm_host.h>
  22 #include <linux/highmem.h>
  23 #include <linux/log2.h>
  24 #include <linux/uaccess.h>
  25 #include <linux/sched.h>
  26 #include <linux/rwsem.h>
  27 #include <linux/vmalloc.h>
  28 #include <linux/hugetlb.h>
  29 #include <asm/kvm_ppc.h>
  30
  31 #include "e500.h"
  32 #include "trace_booke.h"
  33 #include "timing.h"
  34 #include "e500_mmu_host.h"
  35
  36 static inline unsigned int gtlb0_get_next_victim(
  37                 struct kvmppc_vcpu_e500 *vcpu_e500)
  38 {
  39         unsigned int victim;
  40
  41         victim = vcpu_e500->gtlb_nv[0]++;
  42         if (unlikely(vcpu_e500->gtlb_nv[0] >= vcpu_e500->gtlb_params[0].ways))
  43                 vcpu_e500->gtlb_nv[0] = 0;
  44
  45         return victim;
  46 }
  47
  48 static int tlb0_set_base(gva_t addr, int sets, int ways)
  49 {
  50         int set_base;
  51
  52         set_base = (addr >> PAGE_SHIFT) & (sets - 1);
  53         set_base *= ways;
  54
  55         return set_base;
  56 }
  57
  58 static int gtlb0_set_base(struct kvmppc_vcpu_e500 *vcpu_e500, gva_t addr)
  59 {
  60         return tlb0_set_base(addr, vcpu_e500->gtlb_params[0].sets,
  61                              vcpu_e500->gtlb_params[0].ways);
  62 }
  63
  64 static unsigned int get_tlb_esel(struct kvm_vcpu *vcpu, int tlbsel)
  65 {
  66         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
  67         int esel = get_tlb_esel_bit(vcpu);
  68
  69         if (tlbsel == 0) {
  70                 esel &= vcpu_e500->gtlb_params[0].ways - 1;
  71                 esel += gtlb0_set_base(vcpu_e500, vcpu->arch.shared->mas2);
  72         } else {
  73                 esel &= vcpu_e500->gtlb_params[tlbsel].entries - 1;
  74         }
  75
  76         return esel;
  77 }
  78
  79 /* Search the guest TLB for a matching entry. */
  80 static int kvmppc_e500_tlb_index(struct kvmppc_vcpu_e500 *vcpu_e500,
  81                 gva_t eaddr, int tlbsel, unsigned int pid, int as)
  82 {
  83         int size = vcpu_e500->gtlb_params[tlbsel].entries;
  84         unsigned int set_base, offset;
  85         int i;
  86
  87         if (tlbsel == 0) {
  88                 set_base = gtlb0_set_base(vcpu_e500, eaddr);
  89                 size = vcpu_e500->gtlb_params[0].ways;
  90         } else {
  91                 if (eaddr < vcpu_e500->tlb1_min_eaddr ||
  92                                 eaddr > vcpu_e500->tlb1_max_eaddr)
  93                         return -1;
  94                 set_base = 0;
  95         }
  96
  97         offset = vcpu_e500->gtlb_offset[tlbsel];
  98
  99         for (i = 0; i < size; i++) {
 100                 struct kvm_book3e_206_tlb_entry *tlbe =
 101                         &vcpu_e500->gtlb_arch[offset + set_base + i];
 102                 unsigned int tid;
 103
 104                 if (eaddr < get_tlb_eaddr(tlbe))
 105                         continue;
 106
 107                 if (eaddr > get_tlb_end(tlbe))
 108                         continue;
 109
 110                 tid = get_tlb_tid(tlbe);
 111                 if (tid && (tid != pid))
 112                         continue;
 113
 114                 if (!get_tlb_v(tlbe))
 115                         continue;
 116
 117                 if (get_tlb_ts(tlbe) != as && as != -1)
 118                         continue;
 119
 120                 return set_base + i;
 121         }
 122
 123         return -1;
 124 }
 125
 126 static inline void kvmppc_e500_deliver_tlb_miss(struct kvm_vcpu *vcpu,
 127                 gva_t eaddr, int as)
 128 {
 129         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 130         unsigned int victim, tsized;
 131         int tlbsel;
 132
 133         /* since we only have two TLBs, only lower bit is used. */
 134         tlbsel = (vcpu->arch.shared->mas4 >> 28) & 0x1;
 135         victim = (tlbsel == 0) ? gtlb0_get_next_victim(vcpu_e500) : 0;
 136         tsized = (vcpu->arch.shared->mas4 >> 7) & 0x1f;
 137
 138         vcpu->arch.shared->mas0 = MAS0_TLBSEL(tlbsel) | MAS0_ESEL(victim)
 139                 | MAS0_NV(vcpu_e500->gtlb_nv[tlbsel]);
 140         vcpu->arch.shared->mas1 = MAS1_VALID | (as ? MAS1_TS : 0)
 141                 | MAS1_TID(get_tlbmiss_tid(vcpu))
 142                 | MAS1_TSIZE(tsized);
 143         vcpu->arch.shared->mas2 = (eaddr & MAS2_EPN)
 144                 | (vcpu->arch.shared->mas4 & MAS2_ATTRIB_MASK);
 145         vcpu->arch.shared->mas7_3 &= MAS3_U0 | MAS3_U1 | MAS3_U2 | MAS3_U3;
 146         vcpu->arch.shared->mas6 = (vcpu->arch.shared->mas6 & MAS6_SPID1)
 147                 | (get_cur_pid(vcpu) << 16)
 148                 | (as ? MAS6_SAS : 0);
 149 }
 150
 151 static void kvmppc_recalc_tlb1map_range(struct kvmppc_vcpu_e500 *vcpu_e500)
 152 {
 153         int size = vcpu_e500->gtlb_params[1].entries;
 154         unsigned int offset;
 155         gva_t eaddr;
 156         int i;
 157
 158         vcpu_e500->tlb1_min_eaddr = ~0UL;
 159         vcpu_e500->tlb1_max_eaddr = 0;
 160         offset = vcpu_e500->gtlb_offset[1];
 161
 162         for (i = 0; i < size; i++) {
 163                 struct kvm_book3e_206_tlb_entry *tlbe =
 164                         &vcpu_e500->gtlb_arch[offset + i];
 165
 166                 if (!get_tlb_v(tlbe))
 167                         continue;
 168
 169                 eaddr = get_tlb_eaddr(tlbe);
 170                 vcpu_e500->tlb1_min_eaddr =
 171                                 min(vcpu_e500->tlb1_min_eaddr, eaddr);
 172
 173                 eaddr = get_tlb_end(tlbe);
 174                 vcpu_e500->tlb1_max_eaddr =
 175                                 max(vcpu_e500->tlb1_max_eaddr, eaddr);
 176         }
 177 }
 178
 179 static int kvmppc_need_recalc_tlb1map_range(struct kvmppc_vcpu_e500 *vcpu_e500,
 180                                 struct kvm_book3e_206_tlb_entry *gtlbe)
 181 {
 182         unsigned long start, end, size;
 183
 184         size = get_tlb_bytes(gtlbe);
 185         start = get_tlb_eaddr(gtlbe) & ~(size - 1);
 186         end = start + size - 1;
 187
 188         return vcpu_e500->tlb1_min_eaddr == start ||
 189                         vcpu_e500->tlb1_max_eaddr == end;
 190 }
 191
 192 /* This function is supposed to be called for a adding a new valid tlb entry */
 193 static void kvmppc_set_tlb1map_range(struct kvm_vcpu *vcpu,
 194                                 struct kvm_book3e_206_tlb_entry *gtlbe)
 195 {
 196         unsigned long start, end, size;
 197         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 198
 199         if (!get_tlb_v(gtlbe))
 200                 return;
 201
 202         size = get_tlb_bytes(gtlbe);
 203         start = get_tlb_eaddr(gtlbe) & ~(size - 1);
 204         end = start + size - 1;
 205
 206         vcpu_e500->tlb1_min_eaddr = min(vcpu_e500->tlb1_min_eaddr, start);
 207         vcpu_e500->tlb1_max_eaddr = max(vcpu_e500->tlb1_max_eaddr, end);
 208 }
 209
 210 static inline int kvmppc_e500_gtlbe_invalidate(
 211                                 struct kvmppc_vcpu_e500 *vcpu_e500,
 212                                 int tlbsel, int esel)
 213 {
 214         struct kvm_book3e_206_tlb_entry *gtlbe =
 215                 get_entry(vcpu_e500, tlbsel, esel);
 216
 217         if (unlikely(get_tlb_iprot(gtlbe)))
 218                 return -1;
 219
 220         if (tlbsel == 1 && kvmppc_need_recalc_tlb1map_range(vcpu_e500, gtlbe))
 221                 kvmppc_recalc_tlb1map_range(vcpu_e500);
 222
 223         gtlbe->mas1 = 0;
 224
 225         return 0;
 226 }
 227
 228 int kvmppc_e500_emul_mt_mmucsr0(struct kvmppc_vcpu_e500 *vcpu_e500, ulong value)
 229 {
 230         int esel;
 231
 232         if (value & MMUCSR0_TLB0FI)
 233                 for (esel = 0; esel < vcpu_e500->gtlb_params[0].entries; esel++)
 234                         kvmppc_e500_gtlbe_invalidate(vcpu_e500, 0, esel);
 235         if (value & MMUCSR0_TLB1FI)
 236                 for (esel = 0; esel < vcpu_e500->gtlb_params[1].entries; esel++)
 237                         kvmppc_e500_gtlbe_invalidate(vcpu_e500, 1, esel);
 238
 239         /* Invalidate all host shadow mappings */
 240         kvmppc_core_flush_tlb(&vcpu_e500->vcpu);
 241
 242         return EMULATE_DONE;
 243 }
 244
 245 int kvmppc_e500_emul_tlbivax(struct kvm_vcpu *vcpu, gva_t ea)
 246 {
 247         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 248         unsigned int ia;
 249         int esel, tlbsel;
 250
 251         ia = (ea >> 2) & 0x1;
 252
 253         /* since we only have two TLBs, only lower bit is used. */
 254         tlbsel = (ea >> 3) & 0x1;
 255
 256         if (ia) {
 257                 /* invalidate all entries */
 258                 for (esel = 0; esel < vcpu_e500->gtlb_params[tlbsel].entries;
 259                      esel++)
 260                         kvmppc_e500_gtlbe_invalidate(vcpu_e500, tlbsel, esel);
 261         } else {
 262                 ea &= 0xfffff000;
 263                 esel = kvmppc_e500_tlb_index(vcpu_e500, ea, tlbsel,
 264                                 get_cur_pid(vcpu), -1);
 265                 if (esel >= 0)
 266                         kvmppc_e500_gtlbe_invalidate(vcpu_e500, tlbsel, esel);
 267         }
 268
 269         /* Invalidate all host shadow mappings */
 270         kvmppc_core_flush_tlb(&vcpu_e500->vcpu);
 271
 272         return EMULATE_DONE;
 273 }
 274
 275 static void tlbilx_all(struct kvmppc_vcpu_e500 *vcpu_e500, int tlbsel,
 276                        int pid, int type)
 277 {
 278         struct kvm_book3e_206_tlb_entry *tlbe;
 279         int tid, esel;
 280
 281         /* invalidate all entries */
 282         for (esel = 0; esel < vcpu_e500->gtlb_params[tlbsel].entries; esel++) {
 283                 tlbe = get_entry(vcpu_e500, tlbsel, esel);
 284                 tid = get_tlb_tid(tlbe);
 285                 if (type == 0 || tid == pid) {
 286                         inval_gtlbe_on_host(vcpu_e500, tlbsel, esel);
 287                         kvmppc_e500_gtlbe_invalidate(vcpu_e500, tlbsel, esel);
 288                 }
 289         }
 290 }
 291
 292 static void tlbilx_one(struct kvmppc_vcpu_e500 *vcpu_e500, int pid,
 293                        gva_t ea)
 294 {
 295         int tlbsel, esel;
 296
 297         for (tlbsel = 0; tlbsel < 2; tlbsel++) {
 298                 esel = kvmppc_e500_tlb_index(vcpu_e500, ea, tlbsel, pid, -1);
 299                 if (esel >= 0) {
 300                         inval_gtlbe_on_host(vcpu_e500, tlbsel, esel);
 301                         kvmppc_e500_gtlbe_invalidate(vcpu_e500, tlbsel, esel);
 302                         break;
 303                 }
 304         }
 305 }
 306
 307 int kvmppc_e500_emul_tlbilx(struct kvm_vcpu *vcpu, int type, gva_t ea)
 308 {
 309         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 310         int pid = get_cur_spid(vcpu);
 311
 312         if (type == 0 || type == 1) {
 313                 tlbilx_all(vcpu_e500, 0, pid, type);
 314                 tlbilx_all(vcpu_e500, 1, pid, type);
 315         } else if (type == 3) {
 316                 tlbilx_one(vcpu_e500, pid, ea);
 317         }
 318
 319         return EMULATE_DONE;
 320 }
 321
 322 int kvmppc_e500_emul_tlbre(struct kvm_vcpu *vcpu)
 323 {
 324         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 325         int tlbsel, esel;
 326         struct kvm_book3e_206_tlb_entry *gtlbe;
 327
 328         tlbsel = get_tlb_tlbsel(vcpu);
 329         esel = get_tlb_esel(vcpu, tlbsel);
 330
 331         gtlbe = get_entry(vcpu_e500, tlbsel, esel);
 332         vcpu->arch.shared->mas0 &= ~MAS0_NV(~0);
 333         vcpu->arch.shared->mas0 |= MAS0_NV(vcpu_e500->gtlb_nv[tlbsel]);
 334         vcpu->arch.shared->mas1 = gtlbe->mas1;
 335         vcpu->arch.shared->mas2 = gtlbe->mas2;
 336         vcpu->arch.shared->mas7_3 = gtlbe->mas7_3;
 337
 338         return EMULATE_DONE;
 339 }
 340
 341 int kvmppc_e500_emul_tlbsx(struct kvm_vcpu *vcpu, gva_t ea)
 342 {
 343         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 344         int as = !!get_cur_sas(vcpu);
 345         unsigned int pid = get_cur_spid(vcpu);
 346         int esel, tlbsel;
 347         struct kvm_book3e_206_tlb_entry *gtlbe = NULL;
 348
 349         for (tlbsel = 0; tlbsel < 2; tlbsel++) {
 350                 esel = kvmppc_e500_tlb_index(vcpu_e500, ea, tlbsel, pid, as);
 351                 if (esel >= 0) {
 352                         gtlbe = get_entry(vcpu_e500, tlbsel, esel);
 353                         break;
 354                 }
 355         }
 356
 357         if (gtlbe) {
 358                 esel &= vcpu_e500->gtlb_params[tlbsel].ways - 1;
 359
 360                 vcpu->arch.shared->mas0 = MAS0_TLBSEL(tlbsel) | MAS0_ESEL(esel)
 361                         | MAS0_NV(vcpu_e500->gtlb_nv[tlbsel]);
 362                 vcpu->arch.shared->mas1 = gtlbe->mas1;
 363                 vcpu->arch.shared->mas2 = gtlbe->mas2;
 364                 vcpu->arch.shared->mas7_3 = gtlbe->mas7_3;
 365         } else {
 366                 int victim;
 367
 368                 /* since we only have two TLBs, only lower bit is used. */
 369                 tlbsel = vcpu->arch.shared->mas4 >> 28 & 0x1;
 370                 victim = (tlbsel == 0) ? gtlb0_get_next_victim(vcpu_e500) : 0;
 371
 372                 vcpu->arch.shared->mas0 = MAS0_TLBSEL(tlbsel)
 373                         | MAS0_ESEL(victim)
 374                         | MAS0_NV(vcpu_e500->gtlb_nv[tlbsel]);
 375                 vcpu->arch.shared->mas1 =
 376                           (vcpu->arch.shared->mas6 & MAS6_SPID0)
 377                         | ((vcpu->arch.shared->mas6 & MAS6_SAS) ? MAS1_TS : 0)
 378                         | (vcpu->arch.shared->mas4 & MAS4_TSIZED(~0));
 379                 vcpu->arch.shared->mas2 &= MAS2_EPN;
 380                 vcpu->arch.shared->mas2 |= vcpu->arch.shared->mas4 &
 381                                            MAS2_ATTRIB_MASK;
 382                 vcpu->arch.shared->mas7_3 &= MAS3_U0 | MAS3_U1 |
 383                                              MAS3_U2 | MAS3_U3;
 384         }
 385
 386         kvmppc_set_exit_type(vcpu, EMULATED_TLBSX_EXITS);
 387         return EMULATE_DONE;
 388 }
 389
 390 int kvmppc_e500_emul_tlbwe(struct kvm_vcpu *vcpu)
 391 {
 392         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 393         struct kvm_book3e_206_tlb_entry *gtlbe;
 394         int tlbsel, esel;
 395         int recal = 0;
 396         int idx;
 397
 398         tlbsel = get_tlb_tlbsel(vcpu);
 399         esel = get_tlb_esel(vcpu, tlbsel);
 400
 401         gtlbe = get_entry(vcpu_e500, tlbsel, esel);
 402
 403         if (get_tlb_v(gtlbe)) {
 404                 inval_gtlbe_on_host(vcpu_e500, tlbsel, esel);
 405                 if ((tlbsel == 1) &&
 406                         kvmppc_need_recalc_tlb1map_range(vcpu_e500, gtlbe))
 407                         recal = 1;
 408         }
 409
 410         gtlbe->mas1 = vcpu->arch.shared->mas1;
 411         gtlbe->mas2 = vcpu->arch.shared->mas2;
 412         if (!(vcpu->arch.shared->msr & MSR_CM))
 413                 gtlbe->mas2 &= 0xffffffffUL;
 414         gtlbe->mas7_3 = vcpu->arch.shared->mas7_3;
 415
 416         trace_kvm_booke206_gtlb_write(vcpu->arch.shared->mas0, gtlbe->mas1,
 417                                       gtlbe->mas2, gtlbe->mas7_3);
 418
 419         if (tlbsel == 1) {
 420                 /*
 421                  * If a valid tlb1 entry is overwritten then recalculate the
 422                  * min/max TLB1 map address range otherwise no need to look
 423                  * in tlb1 array.
 424                  */
 425                 if (recal)
 426                         kvmppc_recalc_tlb1map_range(vcpu_e500);
 427                 else
 428                         kvmppc_set_tlb1map_range(vcpu, gtlbe);
 429         }
 430
 431         idx = srcu_read_lock(&vcpu->kvm->srcu);
 432
 433         /* Invalidate shadow mappings for the about-to-be-clobbered TLBE. */
 434         if (tlbe_is_host_safe(vcpu, gtlbe)) {
 435                 u64 eaddr = get_tlb_eaddr(gtlbe);
 436                 u64 raddr = get_tlb_raddr(gtlbe);
 437
 438                 if (tlbsel == 0) {
 439                         gtlbe->mas1 &= ~MAS1_TSIZE(~0);
 440                         gtlbe->mas1 |= MAS1_TSIZE(BOOK3E_PAGESZ_4K);
 441                 }
 442
 443                 /* Premap the faulting page */
 444                 kvmppc_mmu_map(vcpu, eaddr, raddr, index_of(tlbsel, esel));
 445         }
 446
 447         srcu_read_unlock(&vcpu->kvm->srcu, idx);
 448
 449         kvmppc_set_exit_type(vcpu, EMULATED_TLBWE_EXITS);
 450         return EMULATE_DONE;
 451 }
 452
 453 static int kvmppc_e500_tlb_search(struct kvm_vcpu *vcpu,
 454                                   gva_t eaddr, unsigned int pid, int as)
 455 {
 456         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 457         int esel, tlbsel;
 458
 459         for (tlbsel = 0; tlbsel < 2; tlbsel++) {
 460                 esel = kvmppc_e500_tlb_index(vcpu_e500, eaddr, tlbsel, pid, as);
 461                 if (esel >= 0)
 462                         return index_of(tlbsel, esel);
 463         }
 464
 465         return -1;
 466 }
 467
 468 /* 'linear_address' is actually an encoding of AS|PID|EADDR . */
 469 int kvmppc_core_vcpu_translate(struct kvm_vcpu *vcpu,
 470                                struct kvm_translation *tr)
 471 {
 472         int index;
 473         gva_t eaddr;
 474         u8 pid;
 475         u8 as;
 476
 477         eaddr = tr->linear_address;
 478         pid = (tr->linear_address >> 32) & 0xff;
 479         as = (tr->linear_address >> 40) & 0x1;
 480
 481         index = kvmppc_e500_tlb_search(vcpu, eaddr, pid, as);
 482         if (index < 0) {
 483                 tr->valid = 0;
 484                 return 0;
 485         }
 486
 487         tr->physical_address = kvmppc_mmu_xlate(vcpu, index, eaddr);
 488         /* XXX what does "writeable" and "usermode" even mean? */
 489         tr->valid = 1;
 490
 491         return 0;
 492 }
 493
 494
 495 int kvmppc_mmu_itlb_index(struct kvm_vcpu *vcpu, gva_t eaddr)
 496 {
 497         unsigned int as = !!(vcpu->arch.shared->msr & MSR_IS);
 498
 499         return kvmppc_e500_tlb_search(vcpu, eaddr, get_cur_pid(vcpu), as);
 500 }
 501
 502 int kvmppc_mmu_dtlb_index(struct kvm_vcpu *vcpu, gva_t eaddr)
 503 {
 504         unsigned int as = !!(vcpu->arch.shared->msr & MSR_DS);
 505
 506         return kvmppc_e500_tlb_search(vcpu, eaddr, get_cur_pid(vcpu), as);
 507 }
 508
 509 void kvmppc_mmu_itlb_miss(struct kvm_vcpu *vcpu)
 510 {
 511         unsigned int as = !!(vcpu->arch.shared->msr & MSR_IS);
 512
 513         kvmppc_e500_deliver_tlb_miss(vcpu, vcpu->arch.regs.nip, as);
 514 }
 515
 516 void kvmppc_mmu_dtlb_miss(struct kvm_vcpu *vcpu)
 517 {
 518         unsigned int as = !!(vcpu->arch.shared->msr & MSR_DS);
 519
 520         kvmppc_e500_deliver_tlb_miss(vcpu, vcpu->arch.fault_dear, as);
 521 }
 522
 523 gpa_t kvmppc_mmu_xlate(struct kvm_vcpu *vcpu, unsigned int index,
 524                         gva_t eaddr)
 525 {
 526         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 527         struct kvm_book3e_206_tlb_entry *gtlbe;
 528         u64 pgmask;
 529
 530         gtlbe = get_entry(vcpu_e500, tlbsel_of(index), esel_of(index));
 531         pgmask = get_tlb_bytes(gtlbe) - 1;
 532
 533         return get_tlb_raddr(gtlbe) | (eaddr & pgmask);
 534 }
 535
 536 void kvmppc_mmu_destroy_e500(struct kvm_vcpu *vcpu)
 537 {
 538 }
 539
 540 /*****************************************/
 541
 542 static void free_gtlb(struct kvmppc_vcpu_e500 *vcpu_e500)
 543 {
 544         int i;
 545
 546         kvmppc_core_flush_tlb(&vcpu_e500->vcpu);
 547         kfree(vcpu_e500->g2h_tlb1_map);
 548         kfree(vcpu_e500->gtlb_priv[0]);
 549         kfree(vcpu_e500->gtlb_priv[1]);
 550
 551         if (vcpu_e500->shared_tlb_pages) {
 552                 vfree((void *)(round_down((uintptr_t)vcpu_e500->gtlb_arch,
 553                                           PAGE_SIZE)));
 554
 555                 for (i = 0; i < vcpu_e500->num_shared_tlb_pages; i++) {
 556                         set_page_dirty_lock(vcpu_e500->shared_tlb_pages[i]);
 557                         put_page(vcpu_e500->shared_tlb_pages[i]);
 558                 }
 559
 560                 vcpu_e500->num_shared_tlb_pages = 0;
 561
 562                 kfree(vcpu_e500->shared_tlb_pages);
 563                 vcpu_e500->shared_tlb_pages = NULL;
 564         } else {
 565                 kfree(vcpu_e500->gtlb_arch);
 566         }
 567
 568         vcpu_e500->gtlb_arch = NULL;
 569 }
 570
 571 void kvmppc_get_sregs_e500_tlb(struct kvm_vcpu *vcpu, struct kvm_sregs *sregs)
 572 {
 573         sregs->u.e.mas0 = vcpu->arch.shared->mas0;
 574         sregs->u.e.mas1 = vcpu->arch.shared->mas1;
 575         sregs->u.e.mas2 = vcpu->arch.shared->mas2;
 576         sregs->u.e.mas7_3 = vcpu->arch.shared->mas7_3;
 577         sregs->u.e.mas4 = vcpu->arch.shared->mas4;
 578         sregs->u.e.mas6 = vcpu->arch.shared->mas6;
 579
 580         sregs->u.e.mmucfg = vcpu->arch.mmucfg;
 581         sregs->u.e.tlbcfg[0] = vcpu->arch.tlbcfg[0];
 582         sregs->u.e.tlbcfg[1] = vcpu->arch.tlbcfg[1];
 583         sregs->u.e.tlbcfg[2] = 0;
 584         sregs->u.e.tlbcfg[3] = 0;
 585 }
 586
 587 int kvmppc_set_sregs_e500_tlb(struct kvm_vcpu *vcpu, struct kvm_sregs *sregs)
 588 {
 589         if (sregs->u.e.features & KVM_SREGS_E_ARCH206_MMU) {
 590                 vcpu->arch.shared->mas0 = sregs->u.e.mas0;
 591                 vcpu->arch.shared->mas1 = sregs->u.e.mas1;
 592                 vcpu->arch.shared->mas2 = sregs->u.e.mas2;
 593                 vcpu->arch.shared->mas7_3 = sregs->u.e.mas7_3;
 594                 vcpu->arch.shared->mas4 = sregs->u.e.mas4;
 595                 vcpu->arch.shared->mas6 = sregs->u.e.mas6;
 596         }
 597
 598         return 0;
 599 }
 600
 601 int kvmppc_get_one_reg_e500_tlb(struct kvm_vcpu *vcpu, u64 id,
 602                                 union kvmppc_one_reg *val)
 603 {
 604         int r = 0;
 605         long int i;
 606
 607         switch (id) {
 608         case KVM_REG_PPC_MAS0:
 609                 *val = get_reg_val(id, vcpu->arch.shared->mas0);
 610                 break;
 611         case KVM_REG_PPC_MAS1:
 612                 *val = get_reg_val(id, vcpu->arch.shared->mas1);
 613                 break;
 614         case KVM_REG_PPC_MAS2:
 615                 *val = get_reg_val(id, vcpu->arch.shared->mas2);
 616                 break;
 617         case KVM_REG_PPC_MAS7_3:
 618                 *val = get_reg_val(id, vcpu->arch.shared->mas7_3);
 619                 break;
 620         case KVM_REG_PPC_MAS4:
 621                 *val = get_reg_val(id, vcpu->arch.shared->mas4);
 622                 break;
 623         case KVM_REG_PPC_MAS6:
 624                 *val = get_reg_val(id, vcpu->arch.shared->mas6);
 625                 break;
 626         case KVM_REG_PPC_MMUCFG:
 627                 *val = get_reg_val(id, vcpu->arch.mmucfg);
 628                 break;
 629         case KVM_REG_PPC_EPTCFG:
 630                 *val = get_reg_val(id, vcpu->arch.eptcfg);
 631                 break;
 632         case KVM_REG_PPC_TLB0CFG:
 633         case KVM_REG_PPC_TLB1CFG:
 634         case KVM_REG_PPC_TLB2CFG:
 635         case KVM_REG_PPC_TLB3CFG:
 636                 i = id - KVM_REG_PPC_TLB0CFG;
 637                 *val = get_reg_val(id, vcpu->arch.tlbcfg[i]);
 638                 break;
 639         case KVM_REG_PPC_TLB0PS:
 640         case KVM_REG_PPC_TLB1PS:
 641         case KVM_REG_PPC_TLB2PS:
 642         case KVM_REG_PPC_TLB3PS:
 643                 i = id - KVM_REG_PPC_TLB0PS;
 644                 *val = get_reg_val(id, vcpu->arch.tlbps[i]);
 645                 break;
 646         default:
 647                 r = -EINVAL;
 648                 break;
 649         }
 650
 651         return r;
 652 }
 653
 654 int kvmppc_set_one_reg_e500_tlb(struct kvm_vcpu *vcpu, u64 id,
 655                                union kvmppc_one_reg *val)
 656 {
 657         int r = 0;
 658         long int i;
 659
 660         switch (id) {
 661         case KVM_REG_PPC_MAS0:
 662                 vcpu->arch.shared->mas0 = set_reg_val(id, *val);
 663                 break;
 664         case KVM_REG_PPC_MAS1:
 665                 vcpu->arch.shared->mas1 = set_reg_val(id, *val);
 666                 break;
 667         case KVM_REG_PPC_MAS2:
 668                 vcpu->arch.shared->mas2 = set_reg_val(id, *val);
 669                 break;
 670         case KVM_REG_PPC_MAS7_3:
 671                 vcpu->arch.shared->mas7_3 = set_reg_val(id, *val);
 672                 break;
 673         case KVM_REG_PPC_MAS4:
 674                 vcpu->arch.shared->mas4 = set_reg_val(id, *val);
 675                 break;
 676         case KVM_REG_PPC_MAS6:
 677                 vcpu->arch.shared->mas6 = set_reg_val(id, *val);
 678                 break;
 679         /* Only allow MMU registers to be set to the config supported by KVM */
 680         case KVM_REG_PPC_MMUCFG: {
 681                 u32 reg = set_reg_val(id, *val);
 682                 if (reg != vcpu->arch.mmucfg)
 683                         r = -EINVAL;
 684                 break;
 685         }
 686         case KVM_REG_PPC_EPTCFG: {
 687                 u32 reg = set_reg_val(id, *val);
 688                 if (reg != vcpu->arch.eptcfg)
 689                         r = -EINVAL;
 690                 break;
 691         }
 692         case KVM_REG_PPC_TLB0CFG:
 693         case KVM_REG_PPC_TLB1CFG:
 694         case KVM_REG_PPC_TLB2CFG:
 695         case KVM_REG_PPC_TLB3CFG: {
 696                 /* MMU geometry (N_ENTRY/ASSOC) can be set only using SW_TLB */
 697                 u32 reg = set_reg_val(id, *val);
 698                 i = id - KVM_REG_PPC_TLB0CFG;
 699                 if (reg != vcpu->arch.tlbcfg[i])
 700                         r = -EINVAL;
 701                 break;
 702         }
 703         case KVM_REG_PPC_TLB0PS:
 704         case KVM_REG_PPC_TLB1PS:
 705         case KVM_REG_PPC_TLB2PS:
 706         case KVM_REG_PPC_TLB3PS: {
 707                 u32 reg = set_reg_val(id, *val);
 708                 i = id - KVM_REG_PPC_TLB0PS;
 709                 if (reg != vcpu->arch.tlbps[i])
 710                         r = -EINVAL;
 711                 break;
 712         }
 713         default:
 714                 r = -EINVAL;
 715                 break;
 716         }
 717
 718         return r;
 719 }
 720
 721 static int vcpu_mmu_geometry_update(struct kvm_vcpu *vcpu,
 722                 struct kvm_book3e_206_tlb_params *params)
 723 {
 724         vcpu->arch.tlbcfg[0] &= ~(TLBnCFG_N_ENTRY | TLBnCFG_ASSOC);
 725         if (params->tlb_sizes[0] <= 2048)
 726                 vcpu->arch.tlbcfg[0] |= params->tlb_sizes[0];
 727         vcpu->arch.tlbcfg[0] |= params->tlb_ways[0] << TLBnCFG_ASSOC_SHIFT;
 728
 729         vcpu->arch.tlbcfg[1] &= ~(TLBnCFG_N_ENTRY | TLBnCFG_ASSOC);
 730         vcpu->arch.tlbcfg[1] |= params->tlb_sizes[1];
 731         vcpu->arch.tlbcfg[1] |= params->tlb_ways[1] << TLBnCFG_ASSOC_SHIFT;
 732         return 0;
 733 }
 734
 735 int kvm_vcpu_ioctl_config_tlb(struct kvm_vcpu *vcpu,
 736                               struct kvm_config_tlb *cfg)
 737 {
 738         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 739         struct kvm_book3e_206_tlb_params params;
 740         char *virt;
 741         struct page **pages;
 742         struct tlbe_priv *privs[2] = {};
 743         u64 *g2h_bitmap;
 744         size_t array_len;
 745         u32 sets;
 746         int num_pages, ret, i;
 747
 748         if (cfg->mmu_type != KVM_MMU_FSL_BOOKE_NOHV)
 749                 return -EINVAL;
 750
 751         if (copy_from_user(&params, (void __user *)(uintptr_t)cfg->params,
 752                            sizeof(params)))
 753                 return -EFAULT;
 754
 755         if (params.tlb_sizes[1] > 64)
 756                 return -EINVAL;
 757         if (params.tlb_ways[1] != params.tlb_sizes[1])
 758                 return -EINVAL;
 759         if (params.tlb_sizes[2] != 0 || params.tlb_sizes[3] != 0)
 760                 return -EINVAL;
 761         if (params.tlb_ways[2] != 0 || params.tlb_ways[3] != 0)
 762                 return -EINVAL;
 763
 764         if (!is_power_of_2(params.tlb_ways[0]))
 765                 return -EINVAL;
 766
 767         sets = params.tlb_sizes[0] >> ilog2(params.tlb_ways[0]);
 768         if (!is_power_of_2(sets))
 769                 return -EINVAL;
 770
 771         array_len = params.tlb_sizes[0] + params.tlb_sizes[1];
 772         array_len *= sizeof(struct kvm_book3e_206_tlb_entry);
 773
 774         if (cfg->array_len < array_len)
 775                 return -EINVAL;
 776
 777         num_pages = DIV_ROUND_UP(cfg->array + array_len - 1, PAGE_SIZE) -
 778                     cfg->array / PAGE_SIZE;
 779         pages = kmalloc_array(num_pages, sizeof(*pages), GFP_KERNEL);
 780         if (!pages)
 781                 return -ENOMEM;
 782
 783         ret = get_user_pages_fast(cfg->array, num_pages, FOLL_WRITE, pages);
 784         if (ret < 0)
 785                 goto free_pages;
 786
 787         if (ret != num_pages) {
 788                 num_pages = ret;
 789                 ret = -EFAULT;
 790                 goto put_pages;
 791         }
 792
 793         virt = vmap(pages, num_pages, VM_MAP, PAGE_KERNEL);
 794         if (!virt) {
 795                 ret = -ENOMEM;
 796                 goto put_pages;
 797         }
 798
 799         privs[0] = kcalloc(params.tlb_sizes[0], sizeof(*privs[0]), GFP_KERNEL);
 800         if (!privs[0]) {
 801                 ret = -ENOMEM;
 802                 goto put_pages;
 803         }
 804
 805         privs[1] = kcalloc(params.tlb_sizes[1], sizeof(*privs[1]), GFP_KERNEL);
 806         if (!privs[1]) {
 807                 ret = -ENOMEM;
 808                 goto free_privs_first;
 809         }
 810
 811         g2h_bitmap = kcalloc(params.tlb_sizes[1],
 812                              sizeof(*g2h_bitmap),
 813                              GFP_KERNEL);
 814         if (!g2h_bitmap) {
 815                 ret = -ENOMEM;
 816                 goto free_privs_second;
 817         }
 818
 819         free_gtlb(vcpu_e500);
 820
 821         vcpu_e500->gtlb_priv[0] = privs[0];
 822         vcpu_e500->gtlb_priv[1] = privs[1];
 823         vcpu_e500->g2h_tlb1_map = g2h_bitmap;
 824
 825         vcpu_e500->gtlb_arch = (struct kvm_book3e_206_tlb_entry *)
 826                 (virt + (cfg->array & (PAGE_SIZE - 1)));
 827
 828         vcpu_e500->gtlb_params[0].entries = params.tlb_sizes[0];
 829         vcpu_e500->gtlb_params[1].entries = params.tlb_sizes[1];
 830
 831         vcpu_e500->gtlb_offset[0] = 0;
 832         vcpu_e500->gtlb_offset[1] = params.tlb_sizes[0];
 833
 834         /* Update vcpu's MMU geometry based on SW_TLB input */
 835         vcpu_mmu_geometry_update(vcpu, &params);
 836
 837         vcpu_e500->shared_tlb_pages = pages;
 838         vcpu_e500->num_shared_tlb_pages = num_pages;
 839
 840         vcpu_e500->gtlb_params[0].ways = params.tlb_ways[0];
 841         vcpu_e500->gtlb_params[0].sets = sets;
 842
 843         vcpu_e500->gtlb_params[1].ways = params.tlb_sizes[1];
 844         vcpu_e500->gtlb_params[1].sets = 1;
 845
 846         kvmppc_recalc_tlb1map_range(vcpu_e500);
 847         return 0;
 848  free_privs_second:
 849         kfree(privs[1]);
 850  free_privs_first:
 851         kfree(privs[0]);
 852  put_pages:
 853         for (i = 0; i < num_pages; i++)
 854                 put_page(pages[i]);
 855  free_pages:
 856         kfree(pages);
 857         return ret;
 858 }
 859
 860 int kvm_vcpu_ioctl_dirty_tlb(struct kvm_vcpu *vcpu,
 861                              struct kvm_dirty_tlb *dirty)
 862 {
 863         struct kvmppc_vcpu_e500 *vcpu_e500 = to_e500(vcpu);
 864         kvmppc_recalc_tlb1map_range(vcpu_e500);
 865         kvmppc_core_flush_tlb(vcpu);
 866         return 0;
 867 }
 868
 869 /* Vcpu's MMU default configuration */
 870 static int vcpu_mmu_init(struct kvm_vcpu *vcpu,
 871                        struct kvmppc_e500_tlb_params *params)
 872 {
 873         /* Initialize RASIZE, PIDSIZE, NTLBS and MAVN fields with host values*/
 874         vcpu->arch.mmucfg = mfspr(SPRN_MMUCFG) & ~MMUCFG_LPIDSIZE;
 875
 876         /* Initialize TLBnCFG fields with host values and SW_TLB geometry*/
 877         vcpu->arch.tlbcfg[0] = mfspr(SPRN_TLB0CFG) &
 878                              ~(TLBnCFG_N_ENTRY | TLBnCFG_ASSOC);
 879         vcpu->arch.tlbcfg[0] |= params[0].entries;
 880         vcpu->arch.tlbcfg[0] |= params[0].ways << TLBnCFG_ASSOC_SHIFT;
 881
 882         vcpu->arch.tlbcfg[1] = mfspr(SPRN_TLB1CFG) &
 883                              ~(TLBnCFG_N_ENTRY | TLBnCFG_ASSOC);
 884         vcpu->arch.tlbcfg[1] |= params[1].entries;
 885         vcpu->arch.tlbcfg[1] |= params[1].ways << TLBnCFG_ASSOC_SHIFT;
 886
 887         if (has_feature(vcpu, VCPU_FTR_MMU_V2)) {
 888                 vcpu->arch.tlbps[0] = mfspr(SPRN_TLB0PS);
 889                 vcpu->arch.tlbps[1] = mfspr(SPRN_TLB1PS);
 890
 891                 vcpu->arch.mmucfg &= ~MMUCFG_LRAT;
 892
 893                 /* Guest mmu emulation currently doesn't handle E.PT */
 894                 vcpu->arch.eptcfg = 0;
 895                 vcpu->arch.tlbcfg[0] &= ~TLBnCFG_PT;
 896                 vcpu->arch.tlbcfg[1] &= ~TLBnCFG_IND;
 897         }
 898
 899         return 0;
 900 }
 901
 902 int kvmppc_e500_tlb_init(struct kvmppc_vcpu_e500 *vcpu_e500)
 903 {
 904         struct kvm_vcpu *vcpu = &vcpu_e500->vcpu;
 905
 906         if (e500_mmu_host_init(vcpu_e500))
 907                 goto free_vcpu;
 908
 909         vcpu_e500->gtlb_params[0].entries = KVM_E500_TLB0_SIZE;
 910         vcpu_e500->gtlb_params[1].entries = KVM_E500_TLB1_SIZE;
 911
 912         vcpu_e500->gtlb_params[0].ways = KVM_E500_TLB0_WAY_NUM;
 913         vcpu_e500->gtlb_params[0].sets =
 914                 KVM_E500_TLB0_SIZE / KVM_E500_TLB0_WAY_NUM;
 915
 916         vcpu_e500->gtlb_params[1].ways = KVM_E500_TLB1_SIZE;
 917         vcpu_e500->gtlb_params[1].sets = 1;
 918
 919         vcpu_e500->gtlb_arch = kmalloc_array(KVM_E500_TLB0_SIZE +
 920                                              KVM_E500_TLB1_SIZE,
 921                                              sizeof(*vcpu_e500->gtlb_arch),
 922                                              GFP_KERNEL);
 923         if (!vcpu_e500->gtlb_arch)
 924                 return -ENOMEM;
 925
 926         vcpu_e500->gtlb_offset[0] = 0;
 927         vcpu_e500->gtlb_offset[1] = KVM_E500_TLB0_SIZE;
 928
 929         vcpu_e500->gtlb_priv[0] = kcalloc(vcpu_e500->gtlb_params[0].entries,
 930                                           sizeof(struct tlbe_ref),
 931                                           GFP_KERNEL);
 932         if (!vcpu_e500->gtlb_priv[0])
 933                 goto free_vcpu;
 934
 935         vcpu_e500->gtlb_priv[1] = kcalloc(vcpu_e500->gtlb_params[1].entries,
 936                                           sizeof(struct tlbe_ref),
 937                                           GFP_KERNEL);
 938         if (!vcpu_e500->gtlb_priv[1])
 939                 goto free_vcpu;
 940
 941         vcpu_e500->g2h_tlb1_map = kcalloc(vcpu_e500->gtlb_params[1].entries,
 942                                           sizeof(*vcpu_e500->g2h_tlb1_map),
 943                                           GFP_KERNEL);
 944         if (!vcpu_e500->g2h_tlb1_map)
 945                 goto free_vcpu;
 946
 947         vcpu_mmu_init(vcpu, vcpu_e500->gtlb_params);
 948
 949         kvmppc_recalc_tlb1map_range(vcpu_e500);
 950         return 0;
 951  free_vcpu:
 952         free_gtlb(vcpu_e500);
 953         return -1;
 954 }
 955
 956 void kvmppc_e500_tlb_uninit(struct kvmppc_vcpu_e500 *vcpu_e500)
 957 {
 958         free_gtlb(vcpu_e500);
 959         e500_mmu_host_uninit(vcpu_e500);
 960 }