arch/arm64/kvm/sys_regs.c

   1 /*
   2  * Copyright (C) 2012,2013 - ARM Ltd
   3  * Author: Marc Zyngier <marc.zyngier@arm.com>
   4  *
   5  * Derived from arch/arm/kvm/coproc.c:
   6  * Copyright (C) 2012 - Virtual Open Systems and Columbia University
   7  * Authors: Rusty Russell <rusty@rustcorp.com.au>
   8  *          Christoffer Dall <c.dall@virtualopensystems.com>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License, version 2, as
  12  * published by the Free Software Foundation.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  */
  22
  23 #include <linux/bsearch.h>
  24 #include <linux/kvm_host.h>
  25 #include <linux/mm.h>
  26 #include <linux/uaccess.h>
  27
  28 #include <asm/cacheflush.h>
  29 #include <asm/cputype.h>
  30 #include <asm/debug-monitors.h>
  31 #include <asm/esr.h>
  32 #include <asm/kvm_arm.h>
  33 #include <asm/kvm_asm.h>
  34 #include <asm/kvm_coproc.h>
  35 #include <asm/kvm_emulate.h>
  36 #include <asm/kvm_host.h>
  37 #include <asm/kvm_mmu.h>
  38 #include <asm/perf_event.h>
  39 #include <asm/sysreg.h>
  40
  41 #include <trace/events/kvm.h>
  42
  43 #include "sys_regs.h"
  44
  45 #include "trace.h"
  46
  47 /*
  48  * All of this file is extremly similar to the ARM coproc.c, but the
  49  * types are different. My gut feeling is that it should be pretty
  50  * easy to merge, but that would be an ABI breakage -- again. VFP
  51  * would also need to be abstracted.
  52  *
  53  * For AArch32, we only take care of what is being trapped. Anything
  54  * that has to do with init and userspace access has to go via the
  55  * 64bit interface.
  56  */
  57
  58 /* 3 bits per cache level, as per CLIDR, but non-existent caches always 0 */
  59 static u32 cache_levels;
  60
  61 /* CSSELR values; used to index KVM_REG_ARM_DEMUX_ID_CCSIDR */
  62 #define CSSELR_MAX 12
  63
  64 /* Which cache CCSIDR represents depends on CSSELR value. */
  65 static u32 get_ccsidr(u32 csselr)
  66 {
  67         u32 ccsidr;
  68
  69         /* Make sure noone else changes CSSELR during this! */
  70         local_irq_disable();
  71         write_sysreg(csselr, csselr_el1);
  72         isb();
  73         ccsidr = read_sysreg(ccsidr_el1);
  74         local_irq_enable();
  75
  76         return ccsidr;
  77 }
  78
  79 /*
  80  * See note at ARMv7 ARM B1.14.4 (TL;DR: S/W ops are not easily virtualized).
  81  */
  82 static bool access_dcsw(struct kvm_vcpu *vcpu,
  83                         struct sys_reg_params *p,
  84                         const struct sys_reg_desc *r)
  85 {
  86         if (!p->is_write)
  87                 return read_from_write_only(vcpu, p);
  88
  89         kvm_set_way_flush(vcpu);
  90         return true;
  91 }
  92
  93 /*
  94  * Generic accessor for VM registers. Only called as long as HCR_TVM
  95  * is set. If the guest enables the MMU, we stop trapping the VM
  96  * sys_regs and leave it in complete control of the caches.
  97  */
  98 static bool access_vm_reg(struct kvm_vcpu *vcpu,
  99                           struct sys_reg_params *p,
 100                           const struct sys_reg_desc *r)
 101 {
 102         bool was_enabled = vcpu_has_cache_enabled(vcpu);
 103
 104         BUG_ON(!p->is_write);
 105
 106         if (!p->is_aarch32) {
 107                 vcpu_sys_reg(vcpu, r->reg) = p->regval;
 108         } else {
 109                 if (!p->is_32bit)
 110                         vcpu_cp15_64_high(vcpu, r->reg) = upper_32_bits(p->regval);
 111                 vcpu_cp15_64_low(vcpu, r->reg) = lower_32_bits(p->regval);
 112         }
 113
 114         kvm_toggle_cache(vcpu, was_enabled);
 115         return true;
 116 }
 117
 118 /*
 119  * Trap handler for the GICv3 SGI generation system register.
 120  * Forward the request to the VGIC emulation.
 121  * The cp15_64 code makes sure this automatically works
 122  * for both AArch64 and AArch32 accesses.
 123  */
 124 static bool access_gic_sgi(struct kvm_vcpu *vcpu,
 125                            struct sys_reg_params *p,
 126                            const struct sys_reg_desc *r)
 127 {
 128         if (!p->is_write)
 129                 return read_from_write_only(vcpu, p);
 130
 131         vgic_v3_dispatch_sgi(vcpu, p->regval);
 132
 133         return true;
 134 }
 135
 136 static bool access_gic_sre(struct kvm_vcpu *vcpu,
 137                            struct sys_reg_params *p,
 138                            const struct sys_reg_desc *r)
 139 {
 140         if (p->is_write)
 141                 return ignore_write(vcpu, p);
 142
 143         p->regval = vcpu->arch.vgic_cpu.vgic_v3.vgic_sre;
 144         return true;
 145 }
 146
 147 static bool trap_raz_wi(struct kvm_vcpu *vcpu,
 148                         struct sys_reg_params *p,
 149                         const struct sys_reg_desc *r)
 150 {
 151         if (p->is_write)
 152                 return ignore_write(vcpu, p);
 153         else
 154                 return read_zero(vcpu, p);
 155 }
 156
 157 static bool trap_oslsr_el1(struct kvm_vcpu *vcpu,
 158                            struct sys_reg_params *p,
 159                            const struct sys_reg_desc *r)
 160 {
 161         if (p->is_write) {
 162                 return ignore_write(vcpu, p);
 163         } else {
 164                 p->regval = (1 << 3);
 165                 return true;
 166         }
 167 }
 168
 169 static bool trap_dbgauthstatus_el1(struct kvm_vcpu *vcpu,
 170                                    struct sys_reg_params *p,
 171                                    const struct sys_reg_desc *r)
 172 {
 173         if (p->is_write) {
 174                 return ignore_write(vcpu, p);
 175         } else {
 176                 p->regval = read_sysreg(dbgauthstatus_el1);
 177                 return true;
 178         }
 179 }
 180
 181 /*
 182  * We want to avoid world-switching all the DBG registers all the
 183  * time:
 184  *
 185  * - If we've touched any debug register, it is likely that we're
 186  *   going to touch more of them. It then makes sense to disable the
 187  *   traps and start doing the save/restore dance
 188  * - If debug is active (DBG_MDSCR_KDE or DBG_MDSCR_MDE set), it is
 189  *   then mandatory to save/restore the registers, as the guest
 190  *   depends on them.
 191  *
 192  * For this, we use a DIRTY bit, indicating the guest has modified the
 193  * debug registers, used as follow:
 194  *
 195  * On guest entry:
 196  * - If the dirty bit is set (because we're coming back from trapping),
 197  *   disable the traps, save host registers, restore guest registers.
 198  * - If debug is actively in use (DBG_MDSCR_KDE or DBG_MDSCR_MDE set),
 199  *   set the dirty bit, disable the traps, save host registers,
 200  *   restore guest registers.
 201  * - Otherwise, enable the traps
 202  *
 203  * On guest exit:
 204  * - If the dirty bit is set, save guest registers, restore host
 205  *   registers and clear the dirty bit. This ensure that the host can
 206  *   now use the debug registers.
 207  */
 208 static bool trap_debug_regs(struct kvm_vcpu *vcpu,
 209                             struct sys_reg_params *p,
 210                             const struct sys_reg_desc *r)
 211 {
 212         if (p->is_write) {
 213                 vcpu_sys_reg(vcpu, r->reg) = p->regval;
 214                 vcpu->arch.debug_flags |= KVM_ARM64_DEBUG_DIRTY;
 215         } else {
 216                 p->regval = vcpu_sys_reg(vcpu, r->reg);
 217         }
 218
 219         trace_trap_reg(__func__, r->reg, p->is_write, p->regval);
 220
 221         return true;
 222 }
 223
 224 /*
 225  * reg_to_dbg/dbg_to_reg
 226  *
 227  * A 32 bit write to a debug register leave top bits alone
 228  * A 32 bit read from a debug register only returns the bottom bits
 229  *
 230  * All writes will set the KVM_ARM64_DEBUG_DIRTY flag to ensure the
 231  * hyp.S code switches between host and guest values in future.
 232  */
 233 static void reg_to_dbg(struct kvm_vcpu *vcpu,
 234                        struct sys_reg_params *p,
 235                        u64 *dbg_reg)
 236 {
 237         u64 val = p->regval;
 238
 239         if (p->is_32bit) {
 240                 val &= 0xffffffffUL;
 241                 val |= ((*dbg_reg >> 32) << 32);
 242         }
 243
 244         *dbg_reg = val;
 245         vcpu->arch.debug_flags |= KVM_ARM64_DEBUG_DIRTY;
 246 }
 247
 248 static void dbg_to_reg(struct kvm_vcpu *vcpu,
 249                        struct sys_reg_params *p,
 250                        u64 *dbg_reg)
 251 {
 252         p->regval = *dbg_reg;
 253         if (p->is_32bit)
 254                 p->regval &= 0xffffffffUL;
 255 }
 256
 257 static bool trap_bvr(struct kvm_vcpu *vcpu,
 258                      struct sys_reg_params *p,
 259                      const struct sys_reg_desc *rd)
 260 {
 261         u64 *dbg_reg = &vcpu->arch.vcpu_debug_state.dbg_bvr[rd->reg];
 262
 263         if (p->is_write)
 264                 reg_to_dbg(vcpu, p, dbg_reg);
 265         else
 266                 dbg_to_reg(vcpu, p, dbg_reg);
 267
 268         trace_trap_reg(__func__, rd->reg, p->is_write, *dbg_reg);
 269
 270         return true;
 271 }
 272
 273 static int set_bvr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 274                 const struct kvm_one_reg *reg, void __user *uaddr)
 275 {
 276         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_bvr[rd->reg];
 277
 278         if (copy_from_user(r, uaddr, KVM_REG_SIZE(reg->id)) != 0)
 279                 return -EFAULT;
 280         return 0;
 281 }
 282
 283 static int get_bvr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 284         const struct kvm_one_reg *reg, void __user *uaddr)
 285 {
 286         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_bvr[rd->reg];
 287
 288         if (copy_to_user(uaddr, r, KVM_REG_SIZE(reg->id)) != 0)
 289                 return -EFAULT;
 290         return 0;
 291 }
 292
 293 static void reset_bvr(struct kvm_vcpu *vcpu,
 294                       const struct sys_reg_desc *rd)
 295 {
 296         vcpu->arch.vcpu_debug_state.dbg_bvr[rd->reg] = rd->val;
 297 }
 298
 299 static bool trap_bcr(struct kvm_vcpu *vcpu,
 300                      struct sys_reg_params *p,
 301                      const struct sys_reg_desc *rd)
 302 {
 303         u64 *dbg_reg = &vcpu->arch.vcpu_debug_state.dbg_bcr[rd->reg];
 304
 305         if (p->is_write)
 306                 reg_to_dbg(vcpu, p, dbg_reg);
 307         else
 308                 dbg_to_reg(vcpu, p, dbg_reg);
 309
 310         trace_trap_reg(__func__, rd->reg, p->is_write, *dbg_reg);
 311
 312         return true;
 313 }
 314
 315 static int set_bcr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 316                 const struct kvm_one_reg *reg, void __user *uaddr)
 317 {
 318         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_bcr[rd->reg];
 319
 320         if (copy_from_user(r, uaddr, KVM_REG_SIZE(reg->id)) != 0)
 321                 return -EFAULT;
 322
 323         return 0;
 324 }
 325
 326 static int get_bcr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 327         const struct kvm_one_reg *reg, void __user *uaddr)
 328 {
 329         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_bcr[rd->reg];
 330
 331         if (copy_to_user(uaddr, r, KVM_REG_SIZE(reg->id)) != 0)
 332                 return -EFAULT;
 333         return 0;
 334 }
 335
 336 static void reset_bcr(struct kvm_vcpu *vcpu,
 337                       const struct sys_reg_desc *rd)
 338 {
 339         vcpu->arch.vcpu_debug_state.dbg_bcr[rd->reg] = rd->val;
 340 }
 341
 342 static bool trap_wvr(struct kvm_vcpu *vcpu,
 343                      struct sys_reg_params *p,
 344                      const struct sys_reg_desc *rd)
 345 {
 346         u64 *dbg_reg = &vcpu->arch.vcpu_debug_state.dbg_wvr[rd->reg];
 347
 348         if (p->is_write)
 349                 reg_to_dbg(vcpu, p, dbg_reg);
 350         else
 351                 dbg_to_reg(vcpu, p, dbg_reg);
 352
 353         trace_trap_reg(__func__, rd->reg, p->is_write,
 354                 vcpu->arch.vcpu_debug_state.dbg_wvr[rd->reg]);
 355
 356         return true;
 357 }
 358
 359 static int set_wvr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 360                 const struct kvm_one_reg *reg, void __user *uaddr)
 361 {
 362         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_wvr[rd->reg];
 363
 364         if (copy_from_user(r, uaddr, KVM_REG_SIZE(reg->id)) != 0)
 365                 return -EFAULT;
 366         return 0;
 367 }
 368
 369 static int get_wvr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 370         const struct kvm_one_reg *reg, void __user *uaddr)
 371 {
 372         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_wvr[rd->reg];
 373
 374         if (copy_to_user(uaddr, r, KVM_REG_SIZE(reg->id)) != 0)
 375                 return -EFAULT;
 376         return 0;
 377 }
 378
 379 static void reset_wvr(struct kvm_vcpu *vcpu,
 380                       const struct sys_reg_desc *rd)
 381 {
 382         vcpu->arch.vcpu_debug_state.dbg_wvr[rd->reg] = rd->val;
 383 }
 384
 385 static bool trap_wcr(struct kvm_vcpu *vcpu,
 386                      struct sys_reg_params *p,
 387                      const struct sys_reg_desc *rd)
 388 {
 389         u64 *dbg_reg = &vcpu->arch.vcpu_debug_state.dbg_wcr[rd->reg];
 390
 391         if (p->is_write)
 392                 reg_to_dbg(vcpu, p, dbg_reg);
 393         else
 394                 dbg_to_reg(vcpu, p, dbg_reg);
 395
 396         trace_trap_reg(__func__, rd->reg, p->is_write, *dbg_reg);
 397
 398         return true;
 399 }
 400
 401 static int set_wcr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 402                 const struct kvm_one_reg *reg, void __user *uaddr)
 403 {
 404         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_wcr[rd->reg];
 405
 406         if (copy_from_user(r, uaddr, KVM_REG_SIZE(reg->id)) != 0)
 407                 return -EFAULT;
 408         return 0;
 409 }
 410
 411 static int get_wcr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *rd,
 412         const struct kvm_one_reg *reg, void __user *uaddr)
 413 {
 414         __u64 *r = &vcpu->arch.vcpu_debug_state.dbg_wcr[rd->reg];
 415
 416         if (copy_to_user(uaddr, r, KVM_REG_SIZE(reg->id)) != 0)
 417                 return -EFAULT;
 418         return 0;
 419 }
 420
 421 static void reset_wcr(struct kvm_vcpu *vcpu,
 422                       const struct sys_reg_desc *rd)
 423 {
 424         vcpu->arch.vcpu_debug_state.dbg_wcr[rd->reg] = rd->val;
 425 }
 426
 427 static void reset_amair_el1(struct kvm_vcpu *vcpu, const struct sys_reg_desc *r)
 428 {
 429         vcpu_sys_reg(vcpu, AMAIR_EL1) = read_sysreg(amair_el1);
 430 }
 431
 432 static void reset_mpidr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *r)
 433 {
 434         u64 mpidr;
 435
 436         /*
 437          * Map the vcpu_id into the first three affinity level fields of
 438          * the MPIDR. We limit the number of VCPUs in level 0 due to a
 439          * limitation to 16 CPUs in that level in the ICC_SGIxR registers
 440          * of the GICv3 to be able to address each CPU directly when
 441          * sending IPIs.
 442          */
 443         mpidr = (vcpu->vcpu_id & 0x0f) << MPIDR_LEVEL_SHIFT(0);
 444         mpidr |= ((vcpu->vcpu_id >> 4) & 0xff) << MPIDR_LEVEL_SHIFT(1);
 445         mpidr |= ((vcpu->vcpu_id >> 12) & 0xff) << MPIDR_LEVEL_SHIFT(2);
 446         vcpu_sys_reg(vcpu, MPIDR_EL1) = (1ULL << 31) | mpidr;
 447 }
 448
 449 static void reset_pmcr(struct kvm_vcpu *vcpu, const struct sys_reg_desc *r)
 450 {
 451         u64 pmcr, val;
 452
 453         pmcr = read_sysreg(pmcr_el0);
 454         /*
 455          * Writable bits of PMCR_EL0 (ARMV8_PMU_PMCR_MASK) are reset to UNKNOWN
 456          * except PMCR.E resetting to zero.
 457          */
 458         val = ((pmcr & ~ARMV8_PMU_PMCR_MASK)
 459                | (ARMV8_PMU_PMCR_MASK & 0xdecafbad)) & (~ARMV8_PMU_PMCR_E);
 460         vcpu_sys_reg(vcpu, PMCR_EL0) = val;
 461 }
 462
 463 static bool pmu_access_el0_disabled(struct kvm_vcpu *vcpu)
 464 {
 465         u64 reg = vcpu_sys_reg(vcpu, PMUSERENR_EL0);
 466
 467         return !((reg & ARMV8_PMU_USERENR_EN) || vcpu_mode_priv(vcpu));
 468 }
 469
 470 static bool pmu_write_swinc_el0_disabled(struct kvm_vcpu *vcpu)
 471 {
 472         u64 reg = vcpu_sys_reg(vcpu, PMUSERENR_EL0);
 473
 474         return !((reg & (ARMV8_PMU_USERENR_SW | ARMV8_PMU_USERENR_EN))
 475                  || vcpu_mode_priv(vcpu));
 476 }
 477
 478 static bool pmu_access_cycle_counter_el0_disabled(struct kvm_vcpu *vcpu)
 479 {
 480         u64 reg = vcpu_sys_reg(vcpu, PMUSERENR_EL0);
 481
 482         return !((reg & (ARMV8_PMU_USERENR_CR | ARMV8_PMU_USERENR_EN))
 483                  || vcpu_mode_priv(vcpu));
 484 }
 485
 486 static bool pmu_access_event_counter_el0_disabled(struct kvm_vcpu *vcpu)
 487 {
 488         u64 reg = vcpu_sys_reg(vcpu, PMUSERENR_EL0);
 489
 490         return !((reg & (ARMV8_PMU_USERENR_ER | ARMV8_PMU_USERENR_EN))
 491                  || vcpu_mode_priv(vcpu));
 492 }
 493
 494 static bool access_pmcr(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 495                         const struct sys_reg_desc *r)
 496 {
 497         u64 val;
 498
 499         if (!kvm_arm_pmu_v3_ready(vcpu))
 500                 return trap_raz_wi(vcpu, p, r);
 501
 502         if (pmu_access_el0_disabled(vcpu))
 503                 return false;
 504
 505         if (p->is_write) {
 506                 /* Only update writeable bits of PMCR */
 507                 val = vcpu_sys_reg(vcpu, PMCR_EL0);
 508                 val &= ~ARMV8_PMU_PMCR_MASK;
 509                 val |= p->regval & ARMV8_PMU_PMCR_MASK;
 510                 vcpu_sys_reg(vcpu, PMCR_EL0) = val;
 511                 kvm_pmu_handle_pmcr(vcpu, val);
 512         } else {
 513                 /* PMCR.P & PMCR.C are RAZ */
 514                 val = vcpu_sys_reg(vcpu, PMCR_EL0)
 515                       & ~(ARMV8_PMU_PMCR_P | ARMV8_PMU_PMCR_C);
 516                 p->regval = val;
 517         }
 518
 519         return true;
 520 }
 521
 522 static bool access_pmselr(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 523                           const struct sys_reg_desc *r)
 524 {
 525         if (!kvm_arm_pmu_v3_ready(vcpu))
 526                 return trap_raz_wi(vcpu, p, r);
 527
 528         if (pmu_access_event_counter_el0_disabled(vcpu))
 529                 return false;
 530
 531         if (p->is_write)
 532                 vcpu_sys_reg(vcpu, PMSELR_EL0) = p->regval;
 533         else
 534                 /* return PMSELR.SEL field */
 535                 p->regval = vcpu_sys_reg(vcpu, PMSELR_EL0)
 536                             & ARMV8_PMU_COUNTER_MASK;
 537
 538         return true;
 539 }
 540
 541 static bool access_pmceid(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 542                           const struct sys_reg_desc *r)
 543 {
 544         u64 pmceid;
 545
 546         if (!kvm_arm_pmu_v3_ready(vcpu))
 547                 return trap_raz_wi(vcpu, p, r);
 548
 549         BUG_ON(p->is_write);
 550
 551         if (pmu_access_el0_disabled(vcpu))
 552                 return false;
 553
 554         if (!(p->Op2 & 1))
 555                 pmceid = read_sysreg(pmceid0_el0);
 556         else
 557                 pmceid = read_sysreg(pmceid1_el0);
 558
 559         p->regval = pmceid;
 560
 561         return true;
 562 }
 563
 564 static bool pmu_counter_idx_valid(struct kvm_vcpu *vcpu, u64 idx)
 565 {
 566         u64 pmcr, val;
 567
 568         pmcr = vcpu_sys_reg(vcpu, PMCR_EL0);
 569         val = (pmcr >> ARMV8_PMU_PMCR_N_SHIFT) & ARMV8_PMU_PMCR_N_MASK;
 570         if (idx >= val && idx != ARMV8_PMU_CYCLE_IDX)
 571                 return false;
 572
 573         return true;
 574 }
 575
 576 static bool access_pmu_evcntr(struct kvm_vcpu *vcpu,
 577                               struct sys_reg_params *p,
 578                               const struct sys_reg_desc *r)
 579 {
 580         u64 idx;
 581
 582         if (!kvm_arm_pmu_v3_ready(vcpu))
 583                 return trap_raz_wi(vcpu, p, r);
 584
 585         if (r->CRn == 9 && r->CRm == 13) {
 586                 if (r->Op2 == 2) {
 587                         /* PMXEVCNTR_EL0 */
 588                         if (pmu_access_event_counter_el0_disabled(vcpu))
 589                                 return false;
 590
 591                         idx = vcpu_sys_reg(vcpu, PMSELR_EL0)
 592                               & ARMV8_PMU_COUNTER_MASK;
 593                 } else if (r->Op2 == 0) {
 594                         /* PMCCNTR_EL0 */
 595                         if (pmu_access_cycle_counter_el0_disabled(vcpu))
 596                                 return false;
 597
 598                         idx = ARMV8_PMU_CYCLE_IDX;
 599                 } else {
 600                         return false;
 601                 }
 602         } else if (r->CRn == 0 && r->CRm == 9) {
 603                 /* PMCCNTR */
 604                 if (pmu_access_event_counter_el0_disabled(vcpu))
 605                         return false;
 606
 607                 idx = ARMV8_PMU_CYCLE_IDX;
 608         } else if (r->CRn == 14 && (r->CRm & 12) == 8) {
 609                 /* PMEVCNTRn_EL0 */
 610                 if (pmu_access_event_counter_el0_disabled(vcpu))
 611                         return false;
 612
 613                 idx = ((r->CRm & 3) << 3) | (r->Op2 & 7);
 614         } else {
 615                 return false;
 616         }
 617
 618         if (!pmu_counter_idx_valid(vcpu, idx))
 619                 return false;
 620
 621         if (p->is_write) {
 622                 if (pmu_access_el0_disabled(vcpu))
 623                         return false;
 624
 625                 kvm_pmu_set_counter_value(vcpu, idx, p->regval);
 626         } else {
 627                 p->regval = kvm_pmu_get_counter_value(vcpu, idx);
 628         }
 629
 630         return true;
 631 }
 632
 633 static bool access_pmu_evtyper(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 634                                const struct sys_reg_desc *r)
 635 {
 636         u64 idx, reg;
 637
 638         if (!kvm_arm_pmu_v3_ready(vcpu))
 639                 return trap_raz_wi(vcpu, p, r);
 640
 641         if (pmu_access_el0_disabled(vcpu))
 642                 return false;
 643
 644         if (r->CRn == 9 && r->CRm == 13 && r->Op2 == 1) {
 645                 /* PMXEVTYPER_EL0 */
 646                 idx = vcpu_sys_reg(vcpu, PMSELR_EL0) & ARMV8_PMU_COUNTER_MASK;
 647                 reg = PMEVTYPER0_EL0 + idx;
 648         } else if (r->CRn == 14 && (r->CRm & 12) == 12) {
 649                 idx = ((r->CRm & 3) << 3) | (r->Op2 & 7);
 650                 if (idx == ARMV8_PMU_CYCLE_IDX)
 651                         reg = PMCCFILTR_EL0;
 652                 else
 653                         /* PMEVTYPERn_EL0 */
 654                         reg = PMEVTYPER0_EL0 + idx;
 655         } else {
 656                 BUG();
 657         }
 658
 659         if (!pmu_counter_idx_valid(vcpu, idx))
 660                 return false;
 661
 662         if (p->is_write) {
 663                 kvm_pmu_set_counter_event_type(vcpu, p->regval, idx);
 664                 vcpu_sys_reg(vcpu, reg) = p->regval & ARMV8_PMU_EVTYPE_MASK;
 665         } else {
 666                 p->regval = vcpu_sys_reg(vcpu, reg) & ARMV8_PMU_EVTYPE_MASK;
 667         }
 668
 669         return true;
 670 }
 671
 672 static bool access_pmcnten(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 673                            const struct sys_reg_desc *r)
 674 {
 675         u64 val, mask;
 676
 677         if (!kvm_arm_pmu_v3_ready(vcpu))
 678                 return trap_raz_wi(vcpu, p, r);
 679
 680         if (pmu_access_el0_disabled(vcpu))
 681                 return false;
 682
 683         mask = kvm_pmu_valid_counter_mask(vcpu);
 684         if (p->is_write) {
 685                 val = p->regval & mask;
 686                 if (r->Op2 & 0x1) {
 687                         /* accessing PMCNTENSET_EL0 */
 688                         vcpu_sys_reg(vcpu, PMCNTENSET_EL0) |= val;
 689                         kvm_pmu_enable_counter(vcpu, val);
 690                 } else {
 691                         /* accessing PMCNTENCLR_EL0 */
 692                         vcpu_sys_reg(vcpu, PMCNTENSET_EL0) &= ~val;
 693                         kvm_pmu_disable_counter(vcpu, val);
 694                 }
 695         } else {
 696                 p->regval = vcpu_sys_reg(vcpu, PMCNTENSET_EL0) & mask;
 697         }
 698
 699         return true;
 700 }
 701
 702 static bool access_pminten(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 703                            const struct sys_reg_desc *r)
 704 {
 705         u64 mask = kvm_pmu_valid_counter_mask(vcpu);
 706
 707         if (!kvm_arm_pmu_v3_ready(vcpu))
 708                 return trap_raz_wi(vcpu, p, r);
 709
 710         if (!vcpu_mode_priv(vcpu))
 711                 return false;
 712
 713         if (p->is_write) {
 714                 u64 val = p->regval & mask;
 715
 716                 if (r->Op2 & 0x1)
 717                         /* accessing PMINTENSET_EL1 */
 718                         vcpu_sys_reg(vcpu, PMINTENSET_EL1) |= val;
 719                 else
 720                         /* accessing PMINTENCLR_EL1 */
 721                         vcpu_sys_reg(vcpu, PMINTENSET_EL1) &= ~val;
 722         } else {
 723                 p->regval = vcpu_sys_reg(vcpu, PMINTENSET_EL1) & mask;
 724         }
 725
 726         return true;
 727 }
 728
 729 static bool access_pmovs(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 730                          const struct sys_reg_desc *r)
 731 {
 732         u64 mask = kvm_pmu_valid_counter_mask(vcpu);
 733
 734         if (!kvm_arm_pmu_v3_ready(vcpu))
 735                 return trap_raz_wi(vcpu, p, r);
 736
 737         if (pmu_access_el0_disabled(vcpu))
 738                 return false;
 739
 740         if (p->is_write) {
 741                 if (r->CRm & 0x2)
 742                         /* accessing PMOVSSET_EL0 */
 743                         kvm_pmu_overflow_set(vcpu, p->regval & mask);
 744                 else
 745                         /* accessing PMOVSCLR_EL0 */
 746                         vcpu_sys_reg(vcpu, PMOVSSET_EL0) &= ~(p->regval & mask);
 747         } else {
 748                 p->regval = vcpu_sys_reg(vcpu, PMOVSSET_EL0) & mask;
 749         }
 750
 751         return true;
 752 }
 753
 754 static bool access_pmswinc(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 755                            const struct sys_reg_desc *r)
 756 {
 757         u64 mask;
 758
 759         if (!kvm_arm_pmu_v3_ready(vcpu))
 760                 return trap_raz_wi(vcpu, p, r);
 761
 762         if (pmu_write_swinc_el0_disabled(vcpu))
 763                 return false;
 764
 765         if (p->is_write) {
 766                 mask = kvm_pmu_valid_counter_mask(vcpu);
 767                 kvm_pmu_software_increment(vcpu, p->regval & mask);
 768                 return true;
 769         }
 770
 771         return false;
 772 }
 773
 774 static bool access_pmuserenr(struct kvm_vcpu *vcpu, struct sys_reg_params *p,
 775                              const struct sys_reg_desc *r)
 776 {
 777         if (!kvm_arm_pmu_v3_ready(vcpu))
 778                 return trap_raz_wi(vcpu, p, r);
 779
 780         if (p->is_write) {
 781                 if (!vcpu_mode_priv(vcpu))
 782                         return false;
 783
 784                 vcpu_sys_reg(vcpu, PMUSERENR_EL0) = p->regval
 785                                                     & ARMV8_PMU_USERENR_MASK;
 786         } else {
 787                 p->regval = vcpu_sys_reg(vcpu, PMUSERENR_EL0)
 788                             & ARMV8_PMU_USERENR_MASK;
 789         }
 790
 791         return true;
 792 }
 793
 794 /* Silly macro to expand the DBG{BCR,BVR,WVR,WCR}n_EL1 registers in one go */
 795 #define DBG_BCR_BVR_WCR_WVR_EL1(n)                                      \
 796         { SYS_DESC(SYS_DBGBVRn_EL1(n)),                                 \
 797           trap_bvr, reset_bvr, n, 0, get_bvr, set_bvr },                \
 798         { SYS_DESC(SYS_DBGBCRn_EL1(n)),                                 \
 799           trap_bcr, reset_bcr, n, 0, get_bcr, set_bcr },                \
 800         { SYS_DESC(SYS_DBGWVRn_EL1(n)),                                 \
 801           trap_wvr, reset_wvr, n, 0,  get_wvr, set_wvr },               \
 802         { SYS_DESC(SYS_DBGWCRn_EL1(n)),                                 \
 803           trap_wcr, reset_wcr, n, 0,  get_wcr, set_wcr }
 804
 805 /* Macro to expand the PMEVCNTRn_EL0 register */
 806 #define PMU_PMEVCNTR_EL0(n)                                             \
 807         { SYS_DESC(SYS_PMEVCNTRn_EL0(n)),                                       \
 808           access_pmu_evcntr, reset_unknown, (PMEVCNTR0_EL0 + n), }
 809
 810 /* Macro to expand the PMEVTYPERn_EL0 register */
 811 #define PMU_PMEVTYPER_EL0(n)                                            \
 812         { SYS_DESC(SYS_PMEVTYPERn_EL0(n)),                                      \
 813           access_pmu_evtyper, reset_unknown, (PMEVTYPER0_EL0 + n), }
 814
 815 static bool access_cntp_tval(struct kvm_vcpu *vcpu,
 816                 struct sys_reg_params *p,
 817                 const struct sys_reg_desc *r)
 818 {
 819         struct arch_timer_context *ptimer = vcpu_ptimer(vcpu);
 820         u64 now = kvm_phys_timer_read();
 821
 822         if (p->is_write)
 823                 ptimer->cnt_cval = p->regval + now;
 824         else
 825                 p->regval = ptimer->cnt_cval - now;
 826
 827         return true;
 828 }
 829
 830 static bool access_cntp_ctl(struct kvm_vcpu *vcpu,
 831                 struct sys_reg_params *p,
 832                 const struct sys_reg_desc *r)
 833 {
 834         struct arch_timer_context *ptimer = vcpu_ptimer(vcpu);
 835
 836         if (p->is_write) {
 837                 /* ISTATUS bit is read-only */
 838                 ptimer->cnt_ctl = p->regval & ~ARCH_TIMER_CTRL_IT_STAT;
 839         } else {
 840                 u64 now = kvm_phys_timer_read();
 841
 842                 p->regval = ptimer->cnt_ctl;
 843                 /*
 844                  * Set ISTATUS bit if it's expired.
 845                  * Note that according to ARMv8 ARM Issue A.k, ISTATUS bit is
 846                  * UNKNOWN when ENABLE bit is 0, so we chose to set ISTATUS bit
 847                  * regardless of ENABLE bit for our implementation convenience.
 848                  */
 849                 if (ptimer->cnt_cval <= now)
 850                         p->regval |= ARCH_TIMER_CTRL_IT_STAT;
 851         }
 852
 853         return true;
 854 }
 855
 856 static bool access_cntp_cval(struct kvm_vcpu *vcpu,
 857                 struct sys_reg_params *p,
 858                 const struct sys_reg_desc *r)
 859 {
 860         struct arch_timer_context *ptimer = vcpu_ptimer(vcpu);
 861
 862         if (p->is_write)
 863                 ptimer->cnt_cval = p->regval;
 864         else
 865                 p->regval = ptimer->cnt_cval;
 866
 867         return true;
 868 }
 869
 870 /*
 871  * Architected system registers.
 872  * Important: Must be sorted ascending by Op0, Op1, CRn, CRm, Op2
 873  *
 874  * Debug handling: We do trap most, if not all debug related system
 875  * registers. The implementation is good enough to ensure that a guest
 876  * can use these with minimal performance degradation. The drawback is
 877  * that we don't implement any of the external debug, none of the
 878  * OSlock protocol. This should be revisited if we ever encounter a
 879  * more demanding guest...
 880  */
 881 static const struct sys_reg_desc sys_reg_descs[] = {
 882         /* DC ISW */
 883         { Op0(0b01), Op1(0b000), CRn(0b0111), CRm(0b0110), Op2(0b010),
 884           access_dcsw },
 885         /* DC CSW */
 886         { Op0(0b01), Op1(0b000), CRn(0b0111), CRm(0b1010), Op2(0b010),
 887           access_dcsw },
 888         /* DC CISW */
 889         { Op0(0b01), Op1(0b000), CRn(0b0111), CRm(0b1110), Op2(0b010),
 890           access_dcsw },
 891
 892         DBG_BCR_BVR_WCR_WVR_EL1(0),
 893         DBG_BCR_BVR_WCR_WVR_EL1(1),
 894         { SYS_DESC(SYS_MDCCINT_EL1), trap_debug_regs, reset_val, MDCCINT_EL1, 0 },
 895         { SYS_DESC(SYS_MDSCR_EL1), trap_debug_regs, reset_val, MDSCR_EL1, 0 },
 896         DBG_BCR_BVR_WCR_WVR_EL1(2),
 897         DBG_BCR_BVR_WCR_WVR_EL1(3),
 898         DBG_BCR_BVR_WCR_WVR_EL1(4),
 899         DBG_BCR_BVR_WCR_WVR_EL1(5),
 900         DBG_BCR_BVR_WCR_WVR_EL1(6),
 901         DBG_BCR_BVR_WCR_WVR_EL1(7),
 902         DBG_BCR_BVR_WCR_WVR_EL1(8),
 903         DBG_BCR_BVR_WCR_WVR_EL1(9),
 904         DBG_BCR_BVR_WCR_WVR_EL1(10),
 905         DBG_BCR_BVR_WCR_WVR_EL1(11),
 906         DBG_BCR_BVR_WCR_WVR_EL1(12),
 907         DBG_BCR_BVR_WCR_WVR_EL1(13),
 908         DBG_BCR_BVR_WCR_WVR_EL1(14),
 909         DBG_BCR_BVR_WCR_WVR_EL1(15),
 910
 911         { SYS_DESC(SYS_MDRAR_EL1), trap_raz_wi },
 912         { SYS_DESC(SYS_OSLAR_EL1), trap_raz_wi },
 913         { SYS_DESC(SYS_OSLSR_EL1), trap_oslsr_el1 },
 914         { SYS_DESC(SYS_OSDLR_EL1), trap_raz_wi },
 915         { SYS_DESC(SYS_DBGPRCR_EL1), trap_raz_wi },
 916         { SYS_DESC(SYS_DBGCLAIMSET_EL1), trap_raz_wi },
 917         { SYS_DESC(SYS_DBGCLAIMCLR_EL1), trap_raz_wi },
 918         { SYS_DESC(SYS_DBGAUTHSTATUS_EL1), trap_dbgauthstatus_el1 },
 919
 920         { SYS_DESC(SYS_MDCCSR_EL0), trap_raz_wi },
 921         { SYS_DESC(SYS_DBGDTR_EL0), trap_raz_wi },
 922         // DBGDTR[TR]X_EL0 share the same encoding
 923         { SYS_DESC(SYS_DBGDTRTX_EL0), trap_raz_wi },
 924
 925         { SYS_DESC(SYS_DBGVCR32_EL2), NULL, reset_val, DBGVCR32_EL2, 0 },
 926
 927         /* MPIDR_EL1 */
 928         { Op0(0b11), Op1(0b000), CRn(0b0000), CRm(0b0000), Op2(0b101),
 929           NULL, reset_mpidr, MPIDR_EL1 },
 930         /* SCTLR_EL1 */
 931         { Op0(0b11), Op1(0b000), CRn(0b0001), CRm(0b0000), Op2(0b000),
 932           access_vm_reg, reset_val, SCTLR_EL1, 0x00C50078 },
 933         /* CPACR_EL1 */
 934         { Op0(0b11), Op1(0b000), CRn(0b0001), CRm(0b0000), Op2(0b010),
 935           NULL, reset_val, CPACR_EL1, 0 },
 936         /* TTBR0_EL1 */
 937         { Op0(0b11), Op1(0b000), CRn(0b0010), CRm(0b0000), Op2(0b000),
 938           access_vm_reg, reset_unknown, TTBR0_EL1 },
 939         /* TTBR1_EL1 */
 940         { Op0(0b11), Op1(0b000), CRn(0b0010), CRm(0b0000), Op2(0b001),
 941           access_vm_reg, reset_unknown, TTBR1_EL1 },
 942         /* TCR_EL1 */
 943         { Op0(0b11), Op1(0b000), CRn(0b0010), CRm(0b0000), Op2(0b010),
 944           access_vm_reg, reset_val, TCR_EL1, 0 },
 945
 946         /* AFSR0_EL1 */
 947         { Op0(0b11), Op1(0b000), CRn(0b0101), CRm(0b0001), Op2(0b000),
 948           access_vm_reg, reset_unknown, AFSR0_EL1 },
 949         /* AFSR1_EL1 */
 950         { Op0(0b11), Op1(0b000), CRn(0b0101), CRm(0b0001), Op2(0b001),
 951           access_vm_reg, reset_unknown, AFSR1_EL1 },
 952         /* ESR_EL1 */
 953         { Op0(0b11), Op1(0b000), CRn(0b0101), CRm(0b0010), Op2(0b000),
 954           access_vm_reg, reset_unknown, ESR_EL1 },
 955         /* FAR_EL1 */
 956         { Op0(0b11), Op1(0b000), CRn(0b0110), CRm(0b0000), Op2(0b000),
 957           access_vm_reg, reset_unknown, FAR_EL1 },
 958         /* PAR_EL1 */
 959         { Op0(0b11), Op1(0b000), CRn(0b0111), CRm(0b0100), Op2(0b000),
 960           NULL, reset_unknown, PAR_EL1 },
 961
 962         { SYS_DESC(SYS_PMINTENSET_EL1), access_pminten, reset_unknown, PMINTENSET_EL1 },
 963         { SYS_DESC(SYS_PMINTENCLR_EL1), access_pminten, NULL, PMINTENSET_EL1 },
 964
 965         /* MAIR_EL1 */
 966         { Op0(0b11), Op1(0b000), CRn(0b1010), CRm(0b0010), Op2(0b000),
 967           access_vm_reg, reset_unknown, MAIR_EL1 },
 968         /* AMAIR_EL1 */
 969         { Op0(0b11), Op1(0b000), CRn(0b1010), CRm(0b0011), Op2(0b000),
 970           access_vm_reg, reset_amair_el1, AMAIR_EL1 },
 971
 972         /* VBAR_EL1 */
 973         { Op0(0b11), Op1(0b000), CRn(0b1100), CRm(0b0000), Op2(0b000),
 974           NULL, reset_val, VBAR_EL1, 0 },
 975
 976         { SYS_DESC(SYS_ICC_SGI1R_EL1), access_gic_sgi },
 977         { SYS_DESC(SYS_ICC_SRE_EL1), access_gic_sre },
 978
 979         /* CONTEXTIDR_EL1 */
 980         { Op0(0b11), Op1(0b000), CRn(0b1101), CRm(0b0000), Op2(0b001),
 981           access_vm_reg, reset_val, CONTEXTIDR_EL1, 0 },
 982         /* TPIDR_EL1 */
 983         { Op0(0b11), Op1(0b000), CRn(0b1101), CRm(0b0000), Op2(0b100),
 984           NULL, reset_unknown, TPIDR_EL1 },
 985
 986         /* CNTKCTL_EL1 */
 987         { Op0(0b11), Op1(0b000), CRn(0b1110), CRm(0b0001), Op2(0b000),
 988           NULL, reset_val, CNTKCTL_EL1, 0},
 989
 990         /* CSSELR_EL1 */
 991         { Op0(0b11), Op1(0b010), CRn(0b0000), CRm(0b0000), Op2(0b000),
 992           NULL, reset_unknown, CSSELR_EL1 },
 993
 994         { SYS_DESC(SYS_PMCR_EL0), access_pmcr, reset_pmcr, },
 995         { SYS_DESC(SYS_PMCNTENSET_EL0), access_pmcnten, reset_unknown, PMCNTENSET_EL0 },
 996         { SYS_DESC(SYS_PMCNTENCLR_EL0), access_pmcnten, NULL, PMCNTENSET_EL0 },
 997         { SYS_DESC(SYS_PMOVSCLR_EL0), access_pmovs, NULL, PMOVSSET_EL0 },
 998         { SYS_DESC(SYS_PMSWINC_EL0), access_pmswinc, reset_unknown, PMSWINC_EL0 },
 999         { SYS_DESC(SYS_PMSELR_EL0), access_pmselr, reset_unknown, PMSELR_EL0 },
1000         { SYS_DESC(SYS_PMCEID0_EL0), access_pmceid },
1001         { SYS_DESC(SYS_PMCEID1_EL0), access_pmceid },
1002         { SYS_DESC(SYS_PMCCNTR_EL0), access_pmu_evcntr, reset_unknown, PMCCNTR_EL0 },
1003         { SYS_DESC(SYS_PMXEVTYPER_EL0), access_pmu_evtyper },
1004         { SYS_DESC(SYS_PMXEVCNTR_EL0), access_pmu_evcntr },
1005         /*
1006          * PMUSERENR_EL0 resets as unknown in 64bit mode while it resets as zero
1007          * in 32bit mode. Here we choose to reset it as zero for consistency.
1008          */
1009         { SYS_DESC(SYS_PMUSERENR_EL0), access_pmuserenr, reset_val, PMUSERENR_EL0, 0 },
1010         { SYS_DESC(SYS_PMOVSSET_EL0), access_pmovs, reset_unknown, PMOVSSET_EL0 },
1011
1012         /* TPIDR_EL0 */
1013         { Op0(0b11), Op1(0b011), CRn(0b1101), CRm(0b0000), Op2(0b010),
1014           NULL, reset_unknown, TPIDR_EL0 },
1015         /* TPIDRRO_EL0 */
1016         { Op0(0b11), Op1(0b011), CRn(0b1101), CRm(0b0000), Op2(0b011),
1017           NULL, reset_unknown, TPIDRRO_EL0 },
1018
1019         { SYS_DESC(SYS_CNTP_TVAL_EL0), access_cntp_tval },
1020         { SYS_DESC(SYS_CNTP_CTL_EL0), access_cntp_ctl },
1021         { SYS_DESC(SYS_CNTP_CVAL_EL0), access_cntp_cval },
1022
1023         /* PMEVCNTRn_EL0 */
1024         PMU_PMEVCNTR_EL0(0),
1025         PMU_PMEVCNTR_EL0(1),
1026         PMU_PMEVCNTR_EL0(2),
1027         PMU_PMEVCNTR_EL0(3),
1028         PMU_PMEVCNTR_EL0(4),
1029         PMU_PMEVCNTR_EL0(5),
1030         PMU_PMEVCNTR_EL0(6),
1031         PMU_PMEVCNTR_EL0(7),
1032         PMU_PMEVCNTR_EL0(8),
1033         PMU_PMEVCNTR_EL0(9),
1034         PMU_PMEVCNTR_EL0(10),
1035         PMU_PMEVCNTR_EL0(11),
1036         PMU_PMEVCNTR_EL0(12),
1037         PMU_PMEVCNTR_EL0(13),
1038         PMU_PMEVCNTR_EL0(14),
1039         PMU_PMEVCNTR_EL0(15),
1040         PMU_PMEVCNTR_EL0(16),
1041         PMU_PMEVCNTR_EL0(17),
1042         PMU_PMEVCNTR_EL0(18),
1043         PMU_PMEVCNTR_EL0(19),
1044         PMU_PMEVCNTR_EL0(20),
1045         PMU_PMEVCNTR_EL0(21),
1046         PMU_PMEVCNTR_EL0(22),
1047         PMU_PMEVCNTR_EL0(23),
1048         PMU_PMEVCNTR_EL0(24),
1049         PMU_PMEVCNTR_EL0(25),
1050         PMU_PMEVCNTR_EL0(26),
1051         PMU_PMEVCNTR_EL0(27),
1052         PMU_PMEVCNTR_EL0(28),
1053         PMU_PMEVCNTR_EL0(29),
1054         PMU_PMEVCNTR_EL0(30),
1055         /* PMEVTYPERn_EL0 */
1056         PMU_PMEVTYPER_EL0(0),
1057         PMU_PMEVTYPER_EL0(1),
1058         PMU_PMEVTYPER_EL0(2),
1059         PMU_PMEVTYPER_EL0(3),
1060         PMU_PMEVTYPER_EL0(4),
1061         PMU_PMEVTYPER_EL0(5),
1062         PMU_PMEVTYPER_EL0(6),
1063         PMU_PMEVTYPER_EL0(7),
1064         PMU_PMEVTYPER_EL0(8),
1065         PMU_PMEVTYPER_EL0(9),
1066         PMU_PMEVTYPER_EL0(10),
1067         PMU_PMEVTYPER_EL0(11),
1068         PMU_PMEVTYPER_EL0(12),
1069         PMU_PMEVTYPER_EL0(13),
1070         PMU_PMEVTYPER_EL0(14),
1071         PMU_PMEVTYPER_EL0(15),
1072         PMU_PMEVTYPER_EL0(16),
1073         PMU_PMEVTYPER_EL0(17),
1074         PMU_PMEVTYPER_EL0(18),
1075         PMU_PMEVTYPER_EL0(19),
1076         PMU_PMEVTYPER_EL0(20),
1077         PMU_PMEVTYPER_EL0(21),
1078         PMU_PMEVTYPER_EL0(22),
1079         PMU_PMEVTYPER_EL0(23),
1080         PMU_PMEVTYPER_EL0(24),
1081         PMU_PMEVTYPER_EL0(25),
1082         PMU_PMEVTYPER_EL0(26),
1083         PMU_PMEVTYPER_EL0(27),
1084         PMU_PMEVTYPER_EL0(28),
1085         PMU_PMEVTYPER_EL0(29),
1086         PMU_PMEVTYPER_EL0(30),
1087         /*
1088          * PMCCFILTR_EL0 resets as unknown in 64bit mode while it resets as zero
1089          * in 32bit mode. Here we choose to reset it as zero for consistency.
1090          */
1091         { SYS_DESC(SYS_PMCCFILTR_EL0), access_pmu_evtyper, reset_val, PMCCFILTR_EL0, 0 },
1092
1093         /* DACR32_EL2 */
1094         { Op0(0b11), Op1(0b100), CRn(0b0011), CRm(0b0000), Op2(0b000),
1095           NULL, reset_unknown, DACR32_EL2 },
1096         /* IFSR32_EL2 */
1097         { Op0(0b11), Op1(0b100), CRn(0b0101), CRm(0b0000), Op2(0b001),
1098           NULL, reset_unknown, IFSR32_EL2 },
1099         /* FPEXC32_EL2 */
1100         { Op0(0b11), Op1(0b100), CRn(0b0101), CRm(0b0011), Op2(0b000),
1101           NULL, reset_val, FPEXC32_EL2, 0x70 },
1102 };
1103
1104 static bool trap_dbgidr(struct kvm_vcpu *vcpu,
1105                         struct sys_reg_params *p,
1106                         const struct sys_reg_desc *r)
1107 {
1108         if (p->is_write) {
1109                 return ignore_write(vcpu, p);
1110         } else {
1111                 u64 dfr = read_system_reg(SYS_ID_AA64DFR0_EL1);
1112                 u64 pfr = read_system_reg(SYS_ID_AA64PFR0_EL1);
1113                 u32 el3 = !!cpuid_feature_extract_unsigned_field(pfr, ID_AA64PFR0_EL3_SHIFT);
1114
1115                 p->regval = ((((dfr >> ID_AA64DFR0_WRPS_SHIFT) & 0xf) << 28) |
1116                              (((dfr >> ID_AA64DFR0_BRPS_SHIFT) & 0xf) << 24) |
1117                              (((dfr >> ID_AA64DFR0_CTX_CMPS_SHIFT) & 0xf) << 20)
1118                              | (6 << 16) | (el3 << 14) | (el3 << 12));
1119                 return true;
1120         }
1121 }
1122
1123 static bool trap_debug32(struct kvm_vcpu *vcpu,
1124                          struct sys_reg_params *p,
1125                          const struct sys_reg_desc *r)
1126 {
1127         if (p->is_write) {
1128                 vcpu_cp14(vcpu, r->reg) = p->regval;
1129                 vcpu->arch.debug_flags |= KVM_ARM64_DEBUG_DIRTY;
1130         } else {
1131                 p->regval = vcpu_cp14(vcpu, r->reg);
1132         }
1133
1134         return true;
1135 }
1136
1137 /* AArch32 debug register mappings
1138  *
1139  * AArch32 DBGBVRn is mapped to DBGBVRn_EL1[31:0]
1140  * AArch32 DBGBXVRn is mapped to DBGBVRn_EL1[63:32]
1141  *
1142  * All control registers and watchpoint value registers are mapped to
1143  * the lower 32 bits of their AArch64 equivalents. We share the trap
1144  * handlers with the above AArch64 code which checks what mode the
1145  * system is in.
1146  */
1147
1148 static bool trap_xvr(struct kvm_vcpu *vcpu,
1149                      struct sys_reg_params *p,
1150                      const struct sys_reg_desc *rd)
1151 {
1152         u64 *dbg_reg = &vcpu->arch.vcpu_debug_state.dbg_bvr[rd->reg];
1153
1154         if (p->is_write) {
1155                 u64 val = *dbg_reg;
1156
1157                 val &= 0xffffffffUL;
1158                 val |= p->regval << 32;
1159                 *dbg_reg = val;
1160
1161                 vcpu->arch.debug_flags |= KVM_ARM64_DEBUG_DIRTY;
1162         } else {
1163                 p->regval = *dbg_reg >> 32;
1164         }
1165
1166         trace_trap_reg(__func__, rd->reg, p->is_write, *dbg_reg);
1167
1168         return true;
1169 }
1170
1171 #define DBG_BCR_BVR_WCR_WVR(n)                                          \
1172         /* DBGBVRn */                                                   \
1173         { Op1( 0), CRn( 0), CRm((n)), Op2( 4), trap_bvr, NULL, n },     \
1174         /* DBGBCRn */                                                   \
1175         { Op1( 0), CRn( 0), CRm((n)), Op2( 5), trap_bcr, NULL, n },     \
1176         /* DBGWVRn */                                                   \
1177         { Op1( 0), CRn( 0), CRm((n)), Op2( 6), trap_wvr, NULL, n },     \
1178         /* DBGWCRn */                                                   \
1179         { Op1( 0), CRn( 0), CRm((n)), Op2( 7), trap_wcr, NULL, n }
1180
1181 #define DBGBXVR(n)                                                      \
1182         { Op1( 0), CRn( 1), CRm((n)), Op2( 1), trap_xvr, NULL, n }
1183
1184 /*
1185  * Trapped cp14 registers. We generally ignore most of the external
1186  * debug, on the principle that they don't really make sense to a
1187  * guest. Revisit this one day, would this principle change.
1188  */
1189 static const struct sys_reg_desc cp14_regs[] = {
1190         /* DBGIDR */
1191         { Op1( 0), CRn( 0), CRm( 0), Op2( 0), trap_dbgidr },
1192         /* DBGDTRRXext */
1193         { Op1( 0), CRn( 0), CRm( 0), Op2( 2), trap_raz_wi },
1194
1195         DBG_BCR_BVR_WCR_WVR(0),
1196         /* DBGDSCRint */
1197         { Op1( 0), CRn( 0), CRm( 1), Op2( 0), trap_raz_wi },
1198         DBG_BCR_BVR_WCR_WVR(1),
1199         /* DBGDCCINT */
1200         { Op1( 0), CRn( 0), CRm( 2), Op2( 0), trap_debug32 },
1201         /* DBGDSCRext */
1202         { Op1( 0), CRn( 0), CRm( 2), Op2( 2), trap_debug32 },
1203         DBG_BCR_BVR_WCR_WVR(2),
1204         /* DBGDTR[RT]Xint */
1205         { Op1( 0), CRn( 0), CRm( 3), Op2( 0), trap_raz_wi },
1206         /* DBGDTR[RT]Xext */
1207         { Op1( 0), CRn( 0), CRm( 3), Op2( 2), trap_raz_wi },
1208         DBG_BCR_BVR_WCR_WVR(3),
1209         DBG_BCR_BVR_WCR_WVR(4),
1210         DBG_BCR_BVR_WCR_WVR(5),
1211         /* DBGWFAR */
1212         { Op1( 0), CRn( 0), CRm( 6), Op2( 0), trap_raz_wi },
1213         /* DBGOSECCR */
1214         { Op1( 0), CRn( 0), CRm( 6), Op2( 2), trap_raz_wi },
1215         DBG_BCR_BVR_WCR_WVR(6),
1216         /* DBGVCR */
1217         { Op1( 0), CRn( 0), CRm( 7), Op2( 0), trap_debug32 },
1218         DBG_BCR_BVR_WCR_WVR(7),
1219         DBG_BCR_BVR_WCR_WVR(8),
1220         DBG_BCR_BVR_WCR_WVR(9),
1221         DBG_BCR_BVR_WCR_WVR(10),
1222         DBG_BCR_BVR_WCR_WVR(11),
1223         DBG_BCR_BVR_WCR_WVR(12),
1224         DBG_BCR_BVR_WCR_WVR(13),
1225         DBG_BCR_BVR_WCR_WVR(14),
1226         DBG_BCR_BVR_WCR_WVR(15),
1227
1228         /* DBGDRAR (32bit) */
1229         { Op1( 0), CRn( 1), CRm( 0), Op2( 0), trap_raz_wi },
1230
1231         DBGBXVR(0),
1232         /* DBGOSLAR */
1233         { Op1( 0), CRn( 1), CRm( 0), Op2( 4), trap_raz_wi },
1234         DBGBXVR(1),
1235         /* DBGOSLSR */
1236         { Op1( 0), CRn( 1), CRm( 1), Op2( 4), trap_oslsr_el1 },
1237         DBGBXVR(2),
1238         DBGBXVR(3),
1239         /* DBGOSDLR */
1240         { Op1( 0), CRn( 1), CRm( 3), Op2( 4), trap_raz_wi },
1241         DBGBXVR(4),
1242         /* DBGPRCR */
1243         { Op1( 0), CRn( 1), CRm( 4), Op2( 4), trap_raz_wi },
1244         DBGBXVR(5),
1245         DBGBXVR(6),
1246         DBGBXVR(7),
1247         DBGBXVR(8),
1248         DBGBXVR(9),
1249         DBGBXVR(10),
1250         DBGBXVR(11),
1251         DBGBXVR(12),
1252         DBGBXVR(13),
1253         DBGBXVR(14),
1254         DBGBXVR(15),
1255
1256         /* DBGDSAR (32bit) */
1257         { Op1( 0), CRn( 2), CRm( 0), Op2( 0), trap_raz_wi },
1258
1259         /* DBGDEVID2 */
1260         { Op1( 0), CRn( 7), CRm( 0), Op2( 7), trap_raz_wi },
1261         /* DBGDEVID1 */
1262         { Op1( 0), CRn( 7), CRm( 1), Op2( 7), trap_raz_wi },
1263         /* DBGDEVID */
1264         { Op1( 0), CRn( 7), CRm( 2), Op2( 7), trap_raz_wi },
1265         /* DBGCLAIMSET */
1266         { Op1( 0), CRn( 7), CRm( 8), Op2( 6), trap_raz_wi },
1267         /* DBGCLAIMCLR */
1268         { Op1( 0), CRn( 7), CRm( 9), Op2( 6), trap_raz_wi },
1269         /* DBGAUTHSTATUS */
1270         { Op1( 0), CRn( 7), CRm(14), Op2( 6), trap_dbgauthstatus_el1 },
1271 };
1272
1273 /* Trapped cp14 64bit registers */
1274 static const struct sys_reg_desc cp14_64_regs[] = {
1275         /* DBGDRAR (64bit) */
1276         { Op1( 0), CRm( 1), .access = trap_raz_wi },
1277
1278         /* DBGDSAR (64bit) */
1279         { Op1( 0), CRm( 2), .access = trap_raz_wi },
1280 };
1281
1282 /* Macro to expand the PMEVCNTRn register */
1283 #define PMU_PMEVCNTR(n)                                                 \
1284         /* PMEVCNTRn */                                                 \
1285         { Op1(0), CRn(0b1110),                                          \
1286           CRm((0b1000 | (((n) >> 3) & 0x3))), Op2(((n) & 0x7)),         \
1287           access_pmu_evcntr }
1288
1289 /* Macro to expand the PMEVTYPERn register */
1290 #define PMU_PMEVTYPER(n)                                                \
1291         /* PMEVTYPERn */                                                \
1292         { Op1(0), CRn(0b1110),                                          \
1293           CRm((0b1100 | (((n) >> 3) & 0x3))), Op2(((n) & 0x7)),         \
1294           access_pmu_evtyper }
1295
1296 /*
1297  * Trapped cp15 registers. TTBR0/TTBR1 get a double encoding,
1298  * depending on the way they are accessed (as a 32bit or a 64bit
1299  * register).
1300  */
1301 static const struct sys_reg_desc cp15_regs[] = {
1302         { Op1( 0), CRn( 0), CRm(12), Op2( 0), access_gic_sgi },
1303
1304         { Op1( 0), CRn( 1), CRm( 0), Op2( 0), access_vm_reg, NULL, c1_SCTLR },
1305         { Op1( 0), CRn( 2), CRm( 0), Op2( 0), access_vm_reg, NULL, c2_TTBR0 },
1306         { Op1( 0), CRn( 2), CRm( 0), Op2( 1), access_vm_reg, NULL, c2_TTBR1 },
1307         { Op1( 0), CRn( 2), CRm( 0), Op2( 2), access_vm_reg, NULL, c2_TTBCR },
1308         { Op1( 0), CRn( 3), CRm( 0), Op2( 0), access_vm_reg, NULL, c3_DACR },
1309         { Op1( 0), CRn( 5), CRm( 0), Op2( 0), access_vm_reg, NULL, c5_DFSR },
1310         { Op1( 0), CRn( 5), CRm( 0), Op2( 1), access_vm_reg, NULL, c5_IFSR },
1311         { Op1( 0), CRn( 5), CRm( 1), Op2( 0), access_vm_reg, NULL, c5_ADFSR },
1312         { Op1( 0), CRn( 5), CRm( 1), Op2( 1), access_vm_reg, NULL, c5_AIFSR },
1313         { Op1( 0), CRn( 6), CRm( 0), Op2( 0), access_vm_reg, NULL, c6_DFAR },
1314         { Op1( 0), CRn( 6), CRm( 0), Op2( 2), access_vm_reg, NULL, c6_IFAR },
1315
1316         /*
1317          * DC{C,I,CI}SW operations:
1318          */
1319         { Op1( 0), CRn( 7), CRm( 6), Op2( 2), access_dcsw },
1320         { Op1( 0), CRn( 7), CRm(10), Op2( 2), access_dcsw },
1321         { Op1( 0), CRn( 7), CRm(14), Op2( 2), access_dcsw },
1322
1323         /* PMU */
1324         { Op1( 0), CRn( 9), CRm(12), Op2( 0), access_pmcr },
1325         { Op1( 0), CRn( 9), CRm(12), Op2( 1), access_pmcnten },
1326         { Op1( 0), CRn( 9), CRm(12), Op2( 2), access_pmcnten },
1327         { Op1( 0), CRn( 9), CRm(12), Op2( 3), access_pmovs },
1328         { Op1( 0), CRn( 9), CRm(12), Op2( 4), access_pmswinc },
1329         { Op1( 0), CRn( 9), CRm(12), Op2( 5), access_pmselr },
1330         { Op1( 0), CRn( 9), CRm(12), Op2( 6), access_pmceid },
1331         { Op1( 0), CRn( 9), CRm(12), Op2( 7), access_pmceid },
1332         { Op1( 0), CRn( 9), CRm(13), Op2( 0), access_pmu_evcntr },
1333         { Op1( 0), CRn( 9), CRm(13), Op2( 1), access_pmu_evtyper },
1334         { Op1( 0), CRn( 9), CRm(13), Op2( 2), access_pmu_evcntr },
1335         { Op1( 0), CRn( 9), CRm(14), Op2( 0), access_pmuserenr },
1336         { Op1( 0), CRn( 9), CRm(14), Op2( 1), access_pminten },
1337         { Op1( 0), CRn( 9), CRm(14), Op2( 2), access_pminten },
1338         { Op1( 0), CRn( 9), CRm(14), Op2( 3), access_pmovs },
1339
1340         { Op1( 0), CRn(10), CRm( 2), Op2( 0), access_vm_reg, NULL, c10_PRRR },
1341         { Op1( 0), CRn(10), CRm( 2), Op2( 1), access_vm_reg, NULL, c10_NMRR },
1342         { Op1( 0), CRn(10), CRm( 3), Op2( 0), access_vm_reg, NULL, c10_AMAIR0 },
1343         { Op1( 0), CRn(10), CRm( 3), Op2( 1), access_vm_reg, NULL, c10_AMAIR1 },
1344
1345         /* ICC_SRE */
1346         { Op1( 0), CRn(12), CRm(12), Op2( 5), access_gic_sre },
1347
1348         { Op1( 0), CRn(13), CRm( 0), Op2( 1), access_vm_reg, NULL, c13_CID },
1349
1350         /* PMEVCNTRn */
1351         PMU_PMEVCNTR(0),
1352         PMU_PMEVCNTR(1),
1353         PMU_PMEVCNTR(2),
1354         PMU_PMEVCNTR(3),
1355         PMU_PMEVCNTR(4),
1356         PMU_PMEVCNTR(5),
1357         PMU_PMEVCNTR(6),
1358         PMU_PMEVCNTR(7),
1359         PMU_PMEVCNTR(8),
1360         PMU_PMEVCNTR(9),
1361         PMU_PMEVCNTR(10),
1362         PMU_PMEVCNTR(11),
1363         PMU_PMEVCNTR(12),
1364         PMU_PMEVCNTR(13),
1365         PMU_PMEVCNTR(14),
1366         PMU_PMEVCNTR(15),
1367         PMU_PMEVCNTR(16),
1368         PMU_PMEVCNTR(17),
1369         PMU_PMEVCNTR(18),
1370         PMU_PMEVCNTR(19),
1371         PMU_PMEVCNTR(20),
1372         PMU_PMEVCNTR(21),
1373         PMU_PMEVCNTR(22),
1374         PMU_PMEVCNTR(23),
1375         PMU_PMEVCNTR(24),
1376         PMU_PMEVCNTR(25),
1377         PMU_PMEVCNTR(26),
1378         PMU_PMEVCNTR(27),
1379         PMU_PMEVCNTR(28),
1380         PMU_PMEVCNTR(29),
1381         PMU_PMEVCNTR(30),
1382         /* PMEVTYPERn */
1383         PMU_PMEVTYPER(0),
1384         PMU_PMEVTYPER(1),
1385         PMU_PMEVTYPER(2),
1386         PMU_PMEVTYPER(3),
1387         PMU_PMEVTYPER(4),
1388         PMU_PMEVTYPER(5),
1389         PMU_PMEVTYPER(6),
1390         PMU_PMEVTYPER(7),
1391         PMU_PMEVTYPER(8),
1392         PMU_PMEVTYPER(9),
1393         PMU_PMEVTYPER(10),
1394         PMU_PMEVTYPER(11),
1395         PMU_PMEVTYPER(12),
1396         PMU_PMEVTYPER(13),
1397         PMU_PMEVTYPER(14),
1398         PMU_PMEVTYPER(15),
1399         PMU_PMEVTYPER(16),
1400         PMU_PMEVTYPER(17),
1401         PMU_PMEVTYPER(18),
1402         PMU_PMEVTYPER(19),
1403         PMU_PMEVTYPER(20),
1404         PMU_PMEVTYPER(21),
1405         PMU_PMEVTYPER(22),
1406         PMU_PMEVTYPER(23),
1407         PMU_PMEVTYPER(24),
1408         PMU_PMEVTYPER(25),
1409         PMU_PMEVTYPER(26),
1410         PMU_PMEVTYPER(27),
1411         PMU_PMEVTYPER(28),
1412         PMU_PMEVTYPER(29),
1413         PMU_PMEVTYPER(30),
1414         /* PMCCFILTR */
1415         { Op1(0), CRn(14), CRm(15), Op2(7), access_pmu_evtyper },
1416 };
1417
1418 static const struct sys_reg_desc cp15_64_regs[] = {
1419         { Op1( 0), CRn( 0), CRm( 2), Op2( 0), access_vm_reg, NULL, c2_TTBR0 },
1420         { Op1( 0), CRn( 0), CRm( 9), Op2( 0), access_pmu_evcntr },
1421         { Op1( 0), CRn( 0), CRm(12), Op2( 0), access_gic_sgi },
1422         { Op1( 1), CRn( 0), CRm( 2), Op2( 0), access_vm_reg, NULL, c2_TTBR1 },
1423 };
1424
1425 /* Target specific emulation tables */
1426 static struct kvm_sys_reg_target_table *target_tables[KVM_ARM_NUM_TARGETS];
1427
1428 void kvm_register_target_sys_reg_table(unsigned int target,
1429                                        struct kvm_sys_reg_target_table *table)
1430 {
1431         target_tables[target] = table;
1432 }
1433
1434 /* Get specific register table for this target. */
1435 static const struct sys_reg_desc *get_target_table(unsigned target,
1436                                                    bool mode_is_64,
1437                                                    size_t *num)
1438 {
1439         struct kvm_sys_reg_target_table *table;
1440
1441         table = target_tables[target];
1442         if (mode_is_64) {
1443                 *num = table->table64.num;
1444                 return table->table64.table;
1445         } else {
1446                 *num = table->table32.num;
1447                 return table->table32.table;
1448         }
1449 }
1450
1451 #define reg_to_match_value(x)                                           \
1452         ({                                                              \
1453                 unsigned long val;                                      \
1454                 val  = (x)->Op0 << 14;                                  \
1455                 val |= (x)->Op1 << 11;                                  \
1456                 val |= (x)->CRn << 7;                                   \
1457                 val |= (x)->CRm << 3;                                   \
1458                 val |= (x)->Op2;                                        \
1459                 val;                                                    \
1460          })
1461
1462 static int match_sys_reg(const void *key, const void *elt)
1463 {
1464         const unsigned long pval = (unsigned long)key;
1465         const struct sys_reg_desc *r = elt;
1466
1467         return pval - reg_to_match_value(r);
1468 }
1469
1470 static const struct sys_reg_desc *find_reg(const struct sys_reg_params *params,
1471                                          const struct sys_reg_desc table[],
1472                                          unsigned int num)
1473 {
1474         unsigned long pval = reg_to_match_value(params);
1475
1476         return bsearch((void *)pval, table, num, sizeof(table[0]), match_sys_reg);
1477 }
1478
1479 int kvm_handle_cp14_load_store(struct kvm_vcpu *vcpu, struct kvm_run *run)
1480 {
1481         kvm_inject_undefined(vcpu);
1482         return 1;
1483 }
1484
1485 /*
1486  * emulate_cp --  tries to match a sys_reg access in a handling table, and
1487  *                call the corresponding trap handler.
1488  *
1489  * @params: pointer to the descriptor of the access
1490  * @table: array of trap descriptors
1491  * @num: size of the trap descriptor array
1492  *
1493  * Return 0 if the access has been handled, and -1 if not.
1494  */
1495 static int emulate_cp(struct kvm_vcpu *vcpu,
1496                       struct sys_reg_params *params,
1497                       const struct sys_reg_desc *table,
1498                       size_t num)
1499 {
1500         const struct sys_reg_desc *r;
1501
1502         if (!table)
1503                 return -1;      /* Not handled */
1504
1505         r = find_reg(params, table, num);
1506
1507         if (r) {
1508                 /*
1509                  * Not having an accessor means that we have
1510                  * configured a trap that we don't know how to
1511                  * handle. This certainly qualifies as a gross bug
1512                  * that should be fixed right away.
1513                  */
1514                 BUG_ON(!r->access);
1515
1516                 if (likely(r->access(vcpu, params, r))) {
1517                         /* Skip instruction, since it was emulated */
1518                         kvm_skip_instr(vcpu, kvm_vcpu_trap_il_is32bit(vcpu));
1519                         /* Handled */
1520                         return 0;
1521                 }
1522         }
1523
1524         /* Not handled */
1525         return -1;
1526 }
1527
1528 static void unhandled_cp_access(struct kvm_vcpu *vcpu,
1529                                 struct sys_reg_params *params)
1530 {
1531         u8 hsr_ec = kvm_vcpu_trap_get_class(vcpu);
1532         int cp = -1;
1533
1534         switch(hsr_ec) {
1535         case ESR_ELx_EC_CP15_32:
1536         case ESR_ELx_EC_CP15_64:
1537                 cp = 15;
1538                 break;
1539         case ESR_ELx_EC_CP14_MR:
1540         case ESR_ELx_EC_CP14_64:
1541                 cp = 14;
1542                 break;
1543         default:
1544                 WARN_ON(1);
1545         }
1546
1547         kvm_err("Unsupported guest CP%d access at: %08lx\n",
1548                 cp, *vcpu_pc(vcpu));
1549         print_sys_reg_instr(params);
1550         kvm_inject_undefined(vcpu);
1551 }
1552
1553 /**
1554  * kvm_handle_cp_64 -- handles a mrrc/mcrr trap on a guest CP14/CP15 access
1555  * @vcpu: The VCPU pointer
1556  * @run:  The kvm_run struct
1557  */
1558 static int kvm_handle_cp_64(struct kvm_vcpu *vcpu,
1559                             const struct sys_reg_desc *global,
1560                             size_t nr_global,
1561                             const struct sys_reg_desc *target_specific,
1562                             size_t nr_specific)
1563 {
1564         struct sys_reg_params params;
1565         u32 hsr = kvm_vcpu_get_hsr(vcpu);
1566         int Rt = (hsr >> 5) & 0xf;
1567         int Rt2 = (hsr >> 10) & 0xf;
1568
1569         params.is_aarch32 = true;
1570         params.is_32bit = false;
1571         params.CRm = (hsr >> 1) & 0xf;
1572         params.is_write = ((hsr & 1) == 0);
1573
1574         params.Op0 = 0;
1575         params.Op1 = (hsr >> 16) & 0xf;
1576         params.Op2 = 0;
1577         params.CRn = 0;
1578
1579         /*
1580          * Make a 64-bit value out of Rt and Rt2. As we use the same trap
1581          * backends between AArch32 and AArch64, we get away with it.
1582          */
1583         if (params.is_write) {
1584                 params.regval = vcpu_get_reg(vcpu, Rt) & 0xffffffff;
1585                 params.regval |= vcpu_get_reg(vcpu, Rt2) << 32;
1586         }
1587
1588         if (!emulate_cp(vcpu, &params, target_specific, nr_specific))
1589                 goto out;
1590         if (!emulate_cp(vcpu, &params, global, nr_global))
1591                 goto out;
1592
1593         unhandled_cp_access(vcpu, &params);
1594
1595 out:
1596         /* Split up the value between registers for the read side */
1597         if (!params.is_write) {
1598                 vcpu_set_reg(vcpu, Rt, lower_32_bits(params.regval));
1599                 vcpu_set_reg(vcpu, Rt2, upper_32_bits(params.regval));
1600         }
1601
1602         return 1;
1603 }
1604
1605 /**
1606  * kvm_handle_cp_32 -- handles a mrc/mcr trap on a guest CP14/CP15 access
1607  * @vcpu: The VCPU pointer
1608  * @run:  The kvm_run struct
1609  */
1610 static int kvm_handle_cp_32(struct kvm_vcpu *vcpu,
1611                             const struct sys_reg_desc *global,
1612                             size_t nr_global,
1613                             const struct sys_reg_desc *target_specific,
1614                             size_t nr_specific)
1615 {
1616         struct sys_reg_params params;
1617         u32 hsr = kvm_vcpu_get_hsr(vcpu);
1618         int Rt  = (hsr >> 5) & 0xf;
1619
1620         params.is_aarch32 = true;
1621         params.is_32bit = true;
1622         params.CRm = (hsr >> 1) & 0xf;
1623         params.regval = vcpu_get_reg(vcpu, Rt);
1624         params.is_write = ((hsr & 1) == 0);
1625         params.CRn = (hsr >> 10) & 0xf;
1626         params.Op0 = 0;
1627         params.Op1 = (hsr >> 14) & 0x7;
1628         params.Op2 = (hsr >> 17) & 0x7;
1629
1630         if (!emulate_cp(vcpu, &params, target_specific, nr_specific) ||
1631             !emulate_cp(vcpu, &params, global, nr_global)) {
1632                 if (!params.is_write)
1633                         vcpu_set_reg(vcpu, Rt, params.regval);
1634                 return 1;
1635         }
1636
1637         unhandled_cp_access(vcpu, &params);
1638         return 1;
1639 }
1640
1641 int kvm_handle_cp15_64(struct kvm_vcpu *vcpu, struct kvm_run *run)
1642 {
1643         const struct sys_reg_desc *target_specific;
1644         size_t num;
1645
1646         target_specific = get_target_table(vcpu->arch.target, false, &num);
1647         return kvm_handle_cp_64(vcpu,
1648                                 cp15_64_regs, ARRAY_SIZE(cp15_64_regs),
1649                                 target_specific, num);
1650 }
1651
1652 int kvm_handle_cp15_32(struct kvm_vcpu *vcpu, struct kvm_run *run)
1653 {
1654         const struct sys_reg_desc *target_specific;
1655         size_t num;
1656
1657         target_specific = get_target_table(vcpu->arch.target, false, &num);
1658         return kvm_handle_cp_32(vcpu,
1659                                 cp15_regs, ARRAY_SIZE(cp15_regs),
1660                                 target_specific, num);
1661 }
1662
1663 int kvm_handle_cp14_64(struct kvm_vcpu *vcpu, struct kvm_run *run)
1664 {
1665         return kvm_handle_cp_64(vcpu,
1666                                 cp14_64_regs, ARRAY_SIZE(cp14_64_regs),
1667                                 NULL, 0);
1668 }
1669
1670 int kvm_handle_cp14_32(struct kvm_vcpu *vcpu, struct kvm_run *run)
1671 {
1672         return kvm_handle_cp_32(vcpu,
1673                                 cp14_regs, ARRAY_SIZE(cp14_regs),
1674                                 NULL, 0);
1675 }
1676
1677 static int emulate_sys_reg(struct kvm_vcpu *vcpu,
1678                            struct sys_reg_params *params)
1679 {
1680         size_t num;
1681         const struct sys_reg_desc *table, *r;
1682
1683         table = get_target_table(vcpu->arch.target, true, &num);
1684
1685         /* Search target-specific then generic table. */
1686         r = find_reg(params, table, num);
1687         if (!r)
1688                 r = find_reg(params, sys_reg_descs, ARRAY_SIZE(sys_reg_descs));
1689
1690         if (likely(r)) {
1691                 /*
1692                  * Not having an accessor means that we have
1693                  * configured a trap that we don't know how to
1694                  * handle. This certainly qualifies as a gross bug
1695                  * that should be fixed right away.
1696                  */
1697                 BUG_ON(!r->access);
1698
1699                 if (likely(r->access(vcpu, params, r))) {
1700                         /* Skip instruction, since it was emulated */
1701                         kvm_skip_instr(vcpu, kvm_vcpu_trap_il_is32bit(vcpu));
1702                         return 1;
1703                 }
1704                 /* If access function fails, it should complain. */
1705         } else {
1706                 kvm_err("Unsupported guest sys_reg access at: %lx\n",
1707                         *vcpu_pc(vcpu));
1708                 print_sys_reg_instr(params);
1709         }
1710         kvm_inject_undefined(vcpu);
1711         return 1;
1712 }
1713
1714 static void reset_sys_reg_descs(struct kvm_vcpu *vcpu,
1715                               const struct sys_reg_desc *table, size_t num)
1716 {
1717         unsigned long i;
1718
1719         for (i = 0; i < num; i++)
1720                 if (table[i].reset)
1721                         table[i].reset(vcpu, &table[i]);
1722 }
1723
1724 /**
1725  * kvm_handle_sys_reg -- handles a mrs/msr trap on a guest sys_reg access
1726  * @vcpu: The VCPU pointer
1727  * @run:  The kvm_run struct
1728  */
1729 int kvm_handle_sys_reg(struct kvm_vcpu *vcpu, struct kvm_run *run)
1730 {
1731         struct sys_reg_params params;
1732         unsigned long esr = kvm_vcpu_get_hsr(vcpu);
1733         int Rt = (esr >> 5) & 0x1f;
1734         int ret;
1735
1736         trace_kvm_handle_sys_reg(esr);
1737
1738         params.is_aarch32 = false;
1739         params.is_32bit = false;
1740         params.Op0 = (esr >> 20) & 3;
1741         params.Op1 = (esr >> 14) & 0x7;
1742         params.CRn = (esr >> 10) & 0xf;
1743         params.CRm = (esr >> 1) & 0xf;
1744         params.Op2 = (esr >> 17) & 0x7;
1745         params.regval = vcpu_get_reg(vcpu, Rt);
1746         params.is_write = !(esr & 1);
1747
1748         ret = emulate_sys_reg(vcpu, &params);
1749
1750         if (!params.is_write)
1751                 vcpu_set_reg(vcpu, Rt, params.regval);
1752         return ret;
1753 }
1754
1755 /******************************************************************************
1756  * Userspace API
1757  *****************************************************************************/
1758
1759 static bool index_to_params(u64 id, struct sys_reg_params *params)
1760 {
1761         switch (id & KVM_REG_SIZE_MASK) {
1762         case KVM_REG_SIZE_U64:
1763                 /* Any unused index bits means it's not valid. */
1764                 if (id & ~(KVM_REG_ARCH_MASK | KVM_REG_SIZE_MASK
1765                               | KVM_REG_ARM_COPROC_MASK
1766                               | KVM_REG_ARM64_SYSREG_OP0_MASK
1767                               | KVM_REG_ARM64_SYSREG_OP1_MASK
1768                               | KVM_REG_ARM64_SYSREG_CRN_MASK
1769                               | KVM_REG_ARM64_SYSREG_CRM_MASK
1770                               | KVM_REG_ARM64_SYSREG_OP2_MASK))
1771                         return false;
1772                 params->Op0 = ((id & KVM_REG_ARM64_SYSREG_OP0_MASK)
1773                                >> KVM_REG_ARM64_SYSREG_OP0_SHIFT);
1774                 params->Op1 = ((id & KVM_REG_ARM64_SYSREG_OP1_MASK)
1775                                >> KVM_REG_ARM64_SYSREG_OP1_SHIFT);
1776                 params->CRn = ((id & KVM_REG_ARM64_SYSREG_CRN_MASK)
1777                                >> KVM_REG_ARM64_SYSREG_CRN_SHIFT);
1778                 params->CRm = ((id & KVM_REG_ARM64_SYSREG_CRM_MASK)
1779                                >> KVM_REG_ARM64_SYSREG_CRM_SHIFT);
1780                 params->Op2 = ((id & KVM_REG_ARM64_SYSREG_OP2_MASK)
1781                                >> KVM_REG_ARM64_SYSREG_OP2_SHIFT);
1782                 return true;
1783         default:
1784                 return false;
1785         }
1786 }
1787
1788 const struct sys_reg_desc *find_reg_by_id(u64 id,
1789                                           struct sys_reg_params *params,
1790                                           const struct sys_reg_desc table[],
1791                                           unsigned int num)
1792 {
1793         if (!index_to_params(id, params))
1794                 return NULL;
1795
1796         return find_reg(params, table, num);
1797 }
1798
1799 /* Decode an index value, and find the sys_reg_desc entry. */
1800 static const struct sys_reg_desc *index_to_sys_reg_desc(struct kvm_vcpu *vcpu,
1801                                                     u64 id)
1802 {
1803         size_t num;
1804         const struct sys_reg_desc *table, *r;
1805         struct sys_reg_params params;
1806
1807         /* We only do sys_reg for now. */
1808         if ((id & KVM_REG_ARM_COPROC_MASK) != KVM_REG_ARM64_SYSREG)
1809                 return NULL;
1810
1811         table = get_target_table(vcpu->arch.target, true, &num);
1812         r = find_reg_by_id(id, &params, table, num);
1813         if (!r)
1814                 r = find_reg(&params, sys_reg_descs, ARRAY_SIZE(sys_reg_descs));
1815
1816         /* Not saved in the sys_reg array? */
1817         if (r && !r->reg)
1818                 r = NULL;
1819
1820         return r;
1821 }
1822
1823 /*
1824  * These are the invariant sys_reg registers: we let the guest see the
1825  * host versions of these, so they're part of the guest state.
1826  *
1827  * A future CPU may provide a mechanism to present different values to
1828  * the guest, or a future kvm may trap them.
1829  */
1830
1831 #define FUNCTION_INVARIANT(reg)                                         \
1832         static void get_##reg(struct kvm_vcpu *v,                       \
1833                               const struct sys_reg_desc *r)             \
1834         {                                                               \
1835                 ((struct sys_reg_desc *)r)->val = read_sysreg(reg);     \
1836         }
1837
1838 FUNCTION_INVARIANT(midr_el1)
1839 FUNCTION_INVARIANT(ctr_el0)
1840 FUNCTION_INVARIANT(revidr_el1)
1841 FUNCTION_INVARIANT(id_pfr0_el1)
1842 FUNCTION_INVARIANT(id_pfr1_el1)
1843 FUNCTION_INVARIANT(id_dfr0_el1)
1844 FUNCTION_INVARIANT(id_afr0_el1)
1845 FUNCTION_INVARIANT(id_mmfr0_el1)
1846 FUNCTION_INVARIANT(id_mmfr1_el1)
1847 FUNCTION_INVARIANT(id_mmfr2_el1)
1848 FUNCTION_INVARIANT(id_mmfr3_el1)
1849 FUNCTION_INVARIANT(id_isar0_el1)
1850 FUNCTION_INVARIANT(id_isar1_el1)
1851 FUNCTION_INVARIANT(id_isar2_el1)
1852 FUNCTION_INVARIANT(id_isar3_el1)
1853 FUNCTION_INVARIANT(id_isar4_el1)
1854 FUNCTION_INVARIANT(id_isar5_el1)
1855 FUNCTION_INVARIANT(clidr_el1)
1856 FUNCTION_INVARIANT(aidr_el1)
1857
1858 /* ->val is filled in by kvm_sys_reg_table_init() */
1859 static struct sys_reg_desc invariant_sys_regs[] = {
1860         { SYS_DESC(SYS_MIDR_EL1), NULL, get_midr_el1 },
1861         { SYS_DESC(SYS_REVIDR_EL1), NULL, get_revidr_el1 },
1862         { SYS_DESC(SYS_ID_PFR0_EL1), NULL, get_id_pfr0_el1 },
1863         { SYS_DESC(SYS_ID_PFR1_EL1), NULL, get_id_pfr1_el1 },
1864         { SYS_DESC(SYS_ID_DFR0_EL1), NULL, get_id_dfr0_el1 },
1865         { SYS_DESC(SYS_ID_AFR0_EL1), NULL, get_id_afr0_el1 },
1866         { SYS_DESC(SYS_ID_MMFR0_EL1), NULL, get_id_mmfr0_el1 },
1867         { SYS_DESC(SYS_ID_MMFR1_EL1), NULL, get_id_mmfr1_el1 },
1868         { SYS_DESC(SYS_ID_MMFR2_EL1), NULL, get_id_mmfr2_el1 },
1869         { SYS_DESC(SYS_ID_MMFR3_EL1), NULL, get_id_mmfr3_el1 },
1870         { SYS_DESC(SYS_ID_ISAR0_EL1), NULL, get_id_isar0_el1 },
1871         { SYS_DESC(SYS_ID_ISAR1_EL1), NULL, get_id_isar1_el1 },
1872         { SYS_DESC(SYS_ID_ISAR2_EL1), NULL, get_id_isar2_el1 },
1873         { SYS_DESC(SYS_ID_ISAR3_EL1), NULL, get_id_isar3_el1 },
1874         { SYS_DESC(SYS_ID_ISAR4_EL1), NULL, get_id_isar4_el1 },
1875         { SYS_DESC(SYS_ID_ISAR5_EL1), NULL, get_id_isar5_el1 },
1876         { SYS_DESC(SYS_CLIDR_EL1), NULL, get_clidr_el1 },
1877         { SYS_DESC(SYS_AIDR_EL1), NULL, get_aidr_el1 },
1878         { SYS_DESC(SYS_CTR_EL0), NULL, get_ctr_el0 },
1879 };
1880
1881 static int reg_from_user(u64 *val, const void __user *uaddr, u64 id)
1882 {
1883         if (copy_from_user(val, uaddr, KVM_REG_SIZE(id)) != 0)
1884                 return -EFAULT;
1885         return 0;
1886 }
1887
1888 static int reg_to_user(void __user *uaddr, const u64 *val, u64 id)
1889 {
1890         if (copy_to_user(uaddr, val, KVM_REG_SIZE(id)) != 0)
1891                 return -EFAULT;
1892         return 0;
1893 }
1894
1895 static int get_invariant_sys_reg(u64 id, void __user *uaddr)
1896 {
1897         struct sys_reg_params params;
1898         const struct sys_reg_desc *r;
1899
1900         r = find_reg_by_id(id, &params, invariant_sys_regs,
1901                            ARRAY_SIZE(invariant_sys_regs));
1902         if (!r)
1903                 return -ENOENT;
1904
1905         return reg_to_user(uaddr, &r->val, id);
1906 }
1907
1908 static int set_invariant_sys_reg(u64 id, void __user *uaddr)
1909 {
1910         struct sys_reg_params params;
1911         const struct sys_reg_desc *r;
1912         int err;
1913         u64 val = 0; /* Make sure high bits are 0 for 32-bit regs */
1914
1915         r = find_reg_by_id(id, &params, invariant_sys_regs,
1916                            ARRAY_SIZE(invariant_sys_regs));
1917         if (!r)
1918                 return -ENOENT;
1919
1920         err = reg_from_user(&val, uaddr, id);
1921         if (err)
1922                 return err;
1923
1924         /* This is what we mean by invariant: you can't change it. */
1925         if (r->val != val)
1926                 return -EINVAL;
1927
1928         return 0;
1929 }
1930
1931 static bool is_valid_cache(u32 val)
1932 {
1933         u32 level, ctype;
1934
1935         if (val >= CSSELR_MAX)
1936                 return false;
1937
1938         /* Bottom bit is Instruction or Data bit.  Next 3 bits are level. */
1939         level = (val >> 1);
1940         ctype = (cache_levels >> (level * 3)) & 7;
1941
1942         switch (ctype) {
1943         case 0: /* No cache */
1944                 return false;
1945         case 1: /* Instruction cache only */
1946                 return (val & 1);
1947         case 2: /* Data cache only */
1948         case 4: /* Unified cache */
1949                 return !(val & 1);
1950         case 3: /* Separate instruction and data caches */
1951                 return true;
1952         default: /* Reserved: we can't know instruction or data. */
1953                 return false;
1954         }
1955 }
1956
1957 static int demux_c15_get(u64 id, void __user *uaddr)
1958 {
1959         u32 val;
1960         u32 __user *uval = uaddr;
1961
1962         /* Fail if we have unknown bits set. */
1963         if (id & ~(KVM_REG_ARCH_MASK|KVM_REG_SIZE_MASK|KVM_REG_ARM_COPROC_MASK
1964                    | ((1 << KVM_REG_ARM_COPROC_SHIFT)-1)))
1965                 return -ENOENT;
1966
1967         switch (id & KVM_REG_ARM_DEMUX_ID_MASK) {
1968         case KVM_REG_ARM_DEMUX_ID_CCSIDR:
1969                 if (KVM_REG_SIZE(id) != 4)
1970                         return -ENOENT;
1971                 val = (id & KVM_REG_ARM_DEMUX_VAL_MASK)
1972                         >> KVM_REG_ARM_DEMUX_VAL_SHIFT;
1973                 if (!is_valid_cache(val))
1974                         return -ENOENT;
1975
1976                 return put_user(get_ccsidr(val), uval);
1977         default:
1978                 return -ENOENT;
1979         }
1980 }
1981
1982 static int demux_c15_set(u64 id, void __user *uaddr)
1983 {
1984         u32 val, newval;
1985         u32 __user *uval = uaddr;
1986
1987         /* Fail if we have unknown bits set. */
1988         if (id & ~(KVM_REG_ARCH_MASK|KVM_REG_SIZE_MASK|KVM_REG_ARM_COPROC_MASK
1989                    | ((1 << KVM_REG_ARM_COPROC_SHIFT)-1)))
1990                 return -ENOENT;
1991
1992         switch (id & KVM_REG_ARM_DEMUX_ID_MASK) {
1993         case KVM_REG_ARM_DEMUX_ID_CCSIDR:
1994                 if (KVM_REG_SIZE(id) != 4)
1995                         return -ENOENT;
1996                 val = (id & KVM_REG_ARM_DEMUX_VAL_MASK)
1997                         >> KVM_REG_ARM_DEMUX_VAL_SHIFT;
1998                 if (!is_valid_cache(val))
1999                         return -ENOENT;
2000
2001                 if (get_user(newval, uval))
2002                         return -EFAULT;
2003
2004                 /* This is also invariant: you can't change it. */
2005                 if (newval != get_ccsidr(val))
2006                         return -EINVAL;
2007                 return 0;
2008         default:
2009                 return -ENOENT;
2010         }
2011 }
2012
2013 int kvm_arm_sys_reg_get_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
2014 {
2015         const struct sys_reg_desc *r;
2016         void __user *uaddr = (void __user *)(unsigned long)reg->addr;
2017
2018         if ((reg->id & KVM_REG_ARM_COPROC_MASK) == KVM_REG_ARM_DEMUX)
2019                 return demux_c15_get(reg->id, uaddr);
2020
2021         if (KVM_REG_SIZE(reg->id) != sizeof(__u64))
2022                 return -ENOENT;
2023
2024         r = index_to_sys_reg_desc(vcpu, reg->id);
2025         if (!r)
2026                 return get_invariant_sys_reg(reg->id, uaddr);
2027
2028         if (r->get_user)
2029                 return (r->get_user)(vcpu, r, reg, uaddr);
2030
2031         return reg_to_user(uaddr, &vcpu_sys_reg(vcpu, r->reg), reg->id);
2032 }
2033
2034 int kvm_arm_sys_reg_set_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
2035 {
2036         const struct sys_reg_desc *r;
2037         void __user *uaddr = (void __user *)(unsigned long)reg->addr;
2038
2039         if ((reg->id & KVM_REG_ARM_COPROC_MASK) == KVM_REG_ARM_DEMUX)
2040                 return demux_c15_set(reg->id, uaddr);
2041
2042         if (KVM_REG_SIZE(reg->id) != sizeof(__u64))
2043                 return -ENOENT;
2044
2045         r = index_to_sys_reg_desc(vcpu, reg->id);
2046         if (!r)
2047                 return set_invariant_sys_reg(reg->id, uaddr);
2048
2049         if (r->set_user)
2050                 return (r->set_user)(vcpu, r, reg, uaddr);
2051
2052         return reg_from_user(&vcpu_sys_reg(vcpu, r->reg), uaddr, reg->id);
2053 }
2054
2055 static unsigned int num_demux_regs(void)
2056 {
2057         unsigned int i, count = 0;
2058
2059         for (i = 0; i < CSSELR_MAX; i++)
2060                 if (is_valid_cache(i))
2061                         count++;
2062
2063         return count;
2064 }
2065
2066 static int write_demux_regids(u64 __user *uindices)
2067 {
2068         u64 val = KVM_REG_ARM64 | KVM_REG_SIZE_U32 | KVM_REG_ARM_DEMUX;
2069         unsigned int i;
2070
2071         val |= KVM_REG_ARM_DEMUX_ID_CCSIDR;
2072         for (i = 0; i < CSSELR_MAX; i++) {
2073                 if (!is_valid_cache(i))
2074                         continue;
2075                 if (put_user(val | i, uindices))
2076                         return -EFAULT;
2077                 uindices++;
2078         }
2079         return 0;
2080 }
2081
2082 static u64 sys_reg_to_index(const struct sys_reg_desc *reg)
2083 {
2084         return (KVM_REG_ARM64 | KVM_REG_SIZE_U64 |
2085                 KVM_REG_ARM64_SYSREG |
2086                 (reg->Op0 << KVM_REG_ARM64_SYSREG_OP0_SHIFT) |
2087                 (reg->Op1 << KVM_REG_ARM64_SYSREG_OP1_SHIFT) |
2088                 (reg->CRn << KVM_REG_ARM64_SYSREG_CRN_SHIFT) |
2089                 (reg->CRm << KVM_REG_ARM64_SYSREG_CRM_SHIFT) |
2090                 (reg->Op2 << KVM_REG_ARM64_SYSREG_OP2_SHIFT));
2091 }
2092
2093 static bool copy_reg_to_user(const struct sys_reg_desc *reg, u64 __user **uind)
2094 {
2095         if (!*uind)
2096                 return true;
2097
2098         if (put_user(sys_reg_to_index(reg), *uind))
2099                 return false;
2100
2101         (*uind)++;
2102         return true;
2103 }
2104
2105 /* Assumed ordered tables, see kvm_sys_reg_table_init. */
2106 static int walk_sys_regs(struct kvm_vcpu *vcpu, u64 __user *uind)
2107 {
2108         const struct sys_reg_desc *i1, *i2, *end1, *end2;
2109         unsigned int total = 0;
2110         size_t num;
2111
2112         /* We check for duplicates here, to allow arch-specific overrides. */
2113         i1 = get_target_table(vcpu->arch.target, true, &num);
2114         end1 = i1 + num;
2115         i2 = sys_reg_descs;
2116         end2 = sys_reg_descs + ARRAY_SIZE(sys_reg_descs);
2117
2118         BUG_ON(i1 == end1 || i2 == end2);
2119
2120         /* Walk carefully, as both tables may refer to the same register. */
2121         while (i1 || i2) {
2122                 int cmp = cmp_sys_reg(i1, i2);
2123                 /* target-specific overrides generic entry. */
2124                 if (cmp <= 0) {
2125                         /* Ignore registers we trap but don't save. */
2126                         if (i1->reg) {
2127                                 if (!copy_reg_to_user(i1, &uind))
2128                                         return -EFAULT;
2129                                 total++;
2130                         }
2131                 } else {
2132                         /* Ignore registers we trap but don't save. */
2133                         if (i2->reg) {
2134                                 if (!copy_reg_to_user(i2, &uind))
2135                                         return -EFAULT;
2136                                 total++;
2137                         }
2138                 }
2139
2140                 if (cmp <= 0 && ++i1 == end1)
2141                         i1 = NULL;
2142                 if (cmp >= 0 && ++i2 == end2)
2143                         i2 = NULL;
2144         }
2145         return total;
2146 }
2147
2148 unsigned long kvm_arm_num_sys_reg_descs(struct kvm_vcpu *vcpu)
2149 {
2150         return ARRAY_SIZE(invariant_sys_regs)
2151                 + num_demux_regs()
2152                 + walk_sys_regs(vcpu, (u64 __user *)NULL);
2153 }
2154
2155 int kvm_arm_copy_sys_reg_indices(struct kvm_vcpu *vcpu, u64 __user *uindices)
2156 {
2157         unsigned int i;
2158         int err;
2159
2160         /* Then give them all the invariant registers' indices. */
2161         for (i = 0; i < ARRAY_SIZE(invariant_sys_regs); i++) {
2162                 if (put_user(sys_reg_to_index(&invariant_sys_regs[i]), uindices))
2163                         return -EFAULT;
2164                 uindices++;
2165         }
2166
2167         err = walk_sys_regs(vcpu, uindices);
2168         if (err < 0)
2169                 return err;
2170         uindices += err;
2171
2172         return write_demux_regids(uindices);
2173 }
2174
2175 static int check_sysreg_table(const struct sys_reg_desc *table, unsigned int n)
2176 {
2177         unsigned int i;
2178
2179         for (i = 1; i < n; i++) {
2180                 if (cmp_sys_reg(&table[i-1], &table[i]) >= 0) {
2181                         kvm_err("sys_reg table %p out of order (%d)\n", table, i - 1);
2182                         return 1;
2183                 }
2184         }
2185
2186         return 0;
2187 }
2188
2189 void kvm_sys_reg_table_init(void)
2190 {
2191         unsigned int i;
2192         struct sys_reg_desc clidr;
2193
2194         /* Make sure tables are unique and in order. */
2195         BUG_ON(check_sysreg_table(sys_reg_descs, ARRAY_SIZE(sys_reg_descs)));
2196         BUG_ON(check_sysreg_table(cp14_regs, ARRAY_SIZE(cp14_regs)));
2197         BUG_ON(check_sysreg_table(cp14_64_regs, ARRAY_SIZE(cp14_64_regs)));
2198         BUG_ON(check_sysreg_table(cp15_regs, ARRAY_SIZE(cp15_regs)));
2199         BUG_ON(check_sysreg_table(cp15_64_regs, ARRAY_SIZE(cp15_64_regs)));
2200         BUG_ON(check_sysreg_table(invariant_sys_regs, ARRAY_SIZE(invariant_sys_regs)));
2201
2202         /* We abuse the reset function to overwrite the table itself. */
2203         for (i = 0; i < ARRAY_SIZE(invariant_sys_regs); i++)
2204                 invariant_sys_regs[i].reset(NULL, &invariant_sys_regs[i]);
2205
2206         /*
2207          * CLIDR format is awkward, so clean it up.  See ARM B4.1.20:
2208          *
2209          *   If software reads the Cache Type fields from Ctype1
2210          *   upwards, once it has seen a value of 0b000, no caches
2211          *   exist at further-out levels of the hierarchy. So, for
2212          *   example, if Ctype3 is the first Cache Type field with a
2213          *   value of 0b000, the values of Ctype4 to Ctype7 must be
2214          *   ignored.
2215          */
2216         get_clidr_el1(NULL, &clidr); /* Ugly... */
2217         cache_levels = clidr.val;
2218         for (i = 0; i < 7; i++)
2219                 if (((cache_levels >> (i*3)) & 7) == 0)
2220                         break;
2221         /* Clear all higher bits. */
2222         cache_levels &= (1 << (i*3))-1;
2223 }
2224
2225 /**
2226  * kvm_reset_sys_regs - sets system registers to reset value
2227  * @vcpu: The VCPU pointer
2228  *
2229  * This function finds the right table above and sets the registers on the
2230  * virtual CPU struct to their architecturally defined reset values.
2231  */
2232 void kvm_reset_sys_regs(struct kvm_vcpu *vcpu)
2233 {
2234         size_t num;
2235         const struct sys_reg_desc *table;
2236
2237         /* Catch someone adding a register without putting in reset entry. */
2238         memset(&vcpu->arch.ctxt.sys_regs, 0x42, sizeof(vcpu->arch.ctxt.sys_regs));
2239
2240         /* Generic chip reset first (so target could override). */
2241         reset_sys_reg_descs(vcpu, sys_reg_descs, ARRAY_SIZE(sys_reg_descs));
2242
2243         table = get_target_table(vcpu->arch.target, true, &num);
2244         reset_sys_reg_descs(vcpu, table, num);
2245
2246         for (num = 1; num < NR_SYS_REGS; num++)
2247                 if (vcpu_sys_reg(vcpu, num) == 0x4242424242424242)
2248                         panic("Didn't reset vcpu_sys_reg(%zi)", num);
2249 }