arch/arm64/kvm/guest.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Copyright (C) 2012,2013 - ARM Ltd
   4  * Author: Marc Zyngier <marc.zyngier@arm.com>
   5  *
   6  * Derived from arch/arm/kvm/guest.c:
   7  * Copyright (C) 2012 - Virtual Open Systems and Columbia University
   8  * Author: Christoffer Dall <c.dall@virtualopensystems.com>
   9  */
  10
  11 #include <linux/bits.h>
  12 #include <linux/errno.h>
  13 #include <linux/err.h>
  14 #include <linux/nospec.h>
  15 #include <linux/kvm_host.h>
  16 #include <linux/module.h>
  17 #include <linux/stddef.h>
  18 #include <linux/string.h>
  19 #include <linux/vmalloc.h>
  20 #include <linux/fs.h>
  21 #include <kvm/arm_psci.h>
  22 #include <asm/cputype.h>
  23 #include <linux/uaccess.h>
  24 #include <asm/fpsimd.h>
  25 #include <asm/kvm.h>
  26 #include <asm/kvm_emulate.h>
  27 #include <asm/kvm_coproc.h>
  28 #include <asm/sigcontext.h>
  29
  30 #include "trace.h"
  31
  32 #define VM_STAT(x) { #x, offsetof(struct kvm, stat.x), KVM_STAT_VM }
  33 #define VCPU_STAT(x) { #x, offsetof(struct kvm_vcpu, stat.x), KVM_STAT_VCPU }
  34
  35 struct kvm_stats_debugfs_item debugfs_entries[] = {
  36         VCPU_STAT(halt_successful_poll),
  37         VCPU_STAT(halt_attempted_poll),
  38         VCPU_STAT(halt_poll_invalid),
  39         VCPU_STAT(halt_wakeup),
  40         VCPU_STAT(hvc_exit_stat),
  41         VCPU_STAT(wfe_exit_stat),
  42         VCPU_STAT(wfi_exit_stat),
  43         VCPU_STAT(mmio_exit_user),
  44         VCPU_STAT(mmio_exit_kernel),
  45         VCPU_STAT(exits),
  46         { NULL }
  47 };
  48
  49 static bool core_reg_offset_is_vreg(u64 off)
  50 {
  51         return off >= KVM_REG_ARM_CORE_REG(fp_regs.vregs) &&
  52                 off < KVM_REG_ARM_CORE_REG(fp_regs.fpsr);
  53 }
  54
  55 static u64 core_reg_offset_from_id(u64 id)
  56 {
  57         return id & ~(KVM_REG_ARCH_MASK | KVM_REG_SIZE_MASK | KVM_REG_ARM_CORE);
  58 }
  59
  60 static int core_reg_size_from_offset(const struct kvm_vcpu *vcpu, u64 off)
  61 {
  62         int size;
  63
  64         switch (off) {
  65         case KVM_REG_ARM_CORE_REG(regs.regs[0]) ...
  66              KVM_REG_ARM_CORE_REG(regs.regs[30]):
  67         case KVM_REG_ARM_CORE_REG(regs.sp):
  68         case KVM_REG_ARM_CORE_REG(regs.pc):
  69         case KVM_REG_ARM_CORE_REG(regs.pstate):
  70         case KVM_REG_ARM_CORE_REG(sp_el1):
  71         case KVM_REG_ARM_CORE_REG(elr_el1):
  72         case KVM_REG_ARM_CORE_REG(spsr[0]) ...
  73              KVM_REG_ARM_CORE_REG(spsr[KVM_NR_SPSR - 1]):
  74                 size = sizeof(__u64);
  75                 break;
  76
  77         case KVM_REG_ARM_CORE_REG(fp_regs.vregs[0]) ...
  78              KVM_REG_ARM_CORE_REG(fp_regs.vregs[31]):
  79                 size = sizeof(__uint128_t);
  80                 break;
  81
  82         case KVM_REG_ARM_CORE_REG(fp_regs.fpsr):
  83         case KVM_REG_ARM_CORE_REG(fp_regs.fpcr):
  84                 size = sizeof(__u32);
  85                 break;
  86
  87         default:
  88                 return -EINVAL;
  89         }
  90
  91         if (!IS_ALIGNED(off, size / sizeof(__u32)))
  92                 return -EINVAL;
  93
  94         /*
  95          * The KVM_REG_ARM64_SVE regs must be used instead of
  96          * KVM_REG_ARM_CORE for accessing the FPSIMD V-registers on
  97          * SVE-enabled vcpus:
  98          */
  99         if (vcpu_has_sve(vcpu) && core_reg_offset_is_vreg(off))
 100                 return -EINVAL;
 101
 102         return size;
 103 }
 104
 105 static int validate_core_offset(const struct kvm_vcpu *vcpu,
 106                                 const struct kvm_one_reg *reg)
 107 {
 108         u64 off = core_reg_offset_from_id(reg->id);
 109         int size = core_reg_size_from_offset(vcpu, off);
 110
 111         if (size < 0)
 112                 return -EINVAL;
 113
 114         if (KVM_REG_SIZE(reg->id) != size)
 115                 return -EINVAL;
 116
 117         return 0;
 118 }
 119
 120 static int get_core_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 121 {
 122         /*
 123          * Because the kvm_regs structure is a mix of 32, 64 and
 124          * 128bit fields, we index it as if it was a 32bit
 125          * array. Hence below, nr_regs is the number of entries, and
 126          * off the index in the "array".
 127          */
 128         __u32 __user *uaddr = (__u32 __user *)(unsigned long)reg->addr;
 129         struct kvm_regs *regs = vcpu_gp_regs(vcpu);
 130         int nr_regs = sizeof(*regs) / sizeof(__u32);
 131         u32 off;
 132
 133         /* Our ID is an index into the kvm_regs struct. */
 134         off = core_reg_offset_from_id(reg->id);
 135         if (off >= nr_regs ||
 136             (off + (KVM_REG_SIZE(reg->id) / sizeof(__u32))) >= nr_regs)
 137                 return -ENOENT;
 138
 139         if (validate_core_offset(vcpu, reg))
 140                 return -EINVAL;
 141
 142         if (copy_to_user(uaddr, ((u32 *)regs) + off, KVM_REG_SIZE(reg->id)))
 143                 return -EFAULT;
 144
 145         return 0;
 146 }
 147
 148 static int set_core_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 149 {
 150         __u32 __user *uaddr = (__u32 __user *)(unsigned long)reg->addr;
 151         struct kvm_regs *regs = vcpu_gp_regs(vcpu);
 152         int nr_regs = sizeof(*regs) / sizeof(__u32);
 153         __uint128_t tmp;
 154         void *valp = &tmp;
 155         u64 off;
 156         int err = 0;
 157
 158         /* Our ID is an index into the kvm_regs struct. */
 159         off = core_reg_offset_from_id(reg->id);
 160         if (off >= nr_regs ||
 161             (off + (KVM_REG_SIZE(reg->id) / sizeof(__u32))) >= nr_regs)
 162                 return -ENOENT;
 163
 164         if (validate_core_offset(vcpu, reg))
 165                 return -EINVAL;
 166
 167         if (KVM_REG_SIZE(reg->id) > sizeof(tmp))
 168                 return -EINVAL;
 169
 170         if (copy_from_user(valp, uaddr, KVM_REG_SIZE(reg->id))) {
 171                 err = -EFAULT;
 172                 goto out;
 173         }
 174
 175         if (off == KVM_REG_ARM_CORE_REG(regs.pstate)) {
 176                 u64 mode = (*(u64 *)valp) & PSR_AA32_MODE_MASK;
 177                 switch (mode) {
 178                 case PSR_AA32_MODE_USR:
 179                         if (!system_supports_32bit_el0())
 180                                 return -EINVAL;
 181                         break;
 182                 case PSR_AA32_MODE_FIQ:
 183                 case PSR_AA32_MODE_IRQ:
 184                 case PSR_AA32_MODE_SVC:
 185                 case PSR_AA32_MODE_ABT:
 186                 case PSR_AA32_MODE_UND:
 187                         if (!vcpu_el1_is_32bit(vcpu))
 188                                 return -EINVAL;
 189                         break;
 190                 case PSR_MODE_EL0t:
 191                 case PSR_MODE_EL1t:
 192                 case PSR_MODE_EL1h:
 193                         if (vcpu_el1_is_32bit(vcpu))
 194                                 return -EINVAL;
 195                         break;
 196                 default:
 197                         err = -EINVAL;
 198                         goto out;
 199                 }
 200         }
 201
 202         memcpy((u32 *)regs + off, valp, KVM_REG_SIZE(reg->id));
 203
 204         if (*vcpu_cpsr(vcpu) & PSR_MODE32_BIT) {
 205                 int i;
 206
 207                 for (i = 0; i < 16; i++)
 208                         *vcpu_reg32(vcpu, i) = (u32)*vcpu_reg32(vcpu, i);
 209         }
 210 out:
 211         return err;
 212 }
 213
 214 #define vq_word(vq) (((vq) - SVE_VQ_MIN) / 64)
 215 #define vq_mask(vq) ((u64)1 << ((vq) - SVE_VQ_MIN) % 64)
 216 #define vq_present(vqs, vq) (!!((vqs)[vq_word(vq)] & vq_mask(vq)))
 217
 218 static int get_sve_vls(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 219 {
 220         unsigned int max_vq, vq;
 221         u64 vqs[KVM_ARM64_SVE_VLS_WORDS];
 222
 223         if (!vcpu_has_sve(vcpu))
 224                 return -ENOENT;
 225
 226         if (WARN_ON(!sve_vl_valid(vcpu->arch.sve_max_vl)))
 227                 return -EINVAL;
 228
 229         memset(vqs, 0, sizeof(vqs));
 230
 231         max_vq = sve_vq_from_vl(vcpu->arch.sve_max_vl);
 232         for (vq = SVE_VQ_MIN; vq <= max_vq; ++vq)
 233                 if (sve_vq_available(vq))
 234                         vqs[vq_word(vq)] |= vq_mask(vq);
 235
 236         if (copy_to_user((void __user *)reg->addr, vqs, sizeof(vqs)))
 237                 return -EFAULT;
 238
 239         return 0;
 240 }
 241
 242 static int set_sve_vls(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 243 {
 244         unsigned int max_vq, vq;
 245         u64 vqs[KVM_ARM64_SVE_VLS_WORDS];
 246
 247         if (!vcpu_has_sve(vcpu))
 248                 return -ENOENT;
 249
 250         if (kvm_arm_vcpu_sve_finalized(vcpu))
 251                 return -EPERM; /* too late! */
 252
 253         if (WARN_ON(vcpu->arch.sve_state))
 254                 return -EINVAL;
 255
 256         if (copy_from_user(vqs, (const void __user *)reg->addr, sizeof(vqs)))
 257                 return -EFAULT;
 258
 259         max_vq = 0;
 260         for (vq = SVE_VQ_MIN; vq <= SVE_VQ_MAX; ++vq)
 261                 if (vq_present(vqs, vq))
 262                         max_vq = vq;
 263
 264         if (max_vq > sve_vq_from_vl(kvm_sve_max_vl))
 265                 return -EINVAL;
 266
 267         /*
 268          * Vector lengths supported by the host can't currently be
 269          * hidden from the guest individually: instead we can only set a
 270          * maxmium via ZCR_EL2.LEN.  So, make sure the available vector
 271          * lengths match the set requested exactly up to the requested
 272          * maximum:
 273          */
 274         for (vq = SVE_VQ_MIN; vq <= max_vq; ++vq)
 275                 if (vq_present(vqs, vq) != sve_vq_available(vq))
 276                         return -EINVAL;
 277
 278         /* Can't run with no vector lengths at all: */
 279         if (max_vq < SVE_VQ_MIN)
 280                 return -EINVAL;
 281
 282         /* vcpu->arch.sve_state will be alloc'd by kvm_vcpu_finalize_sve() */
 283         vcpu->arch.sve_max_vl = sve_vl_from_vq(max_vq);
 284
 285         return 0;
 286 }
 287
 288 #define SVE_REG_SLICE_SHIFT     0
 289 #define SVE_REG_SLICE_BITS      5
 290 #define SVE_REG_ID_SHIFT        (SVE_REG_SLICE_SHIFT + SVE_REG_SLICE_BITS)
 291 #define SVE_REG_ID_BITS         5
 292
 293 #define SVE_REG_SLICE_MASK                                      \
 294         GENMASK(SVE_REG_SLICE_SHIFT + SVE_REG_SLICE_BITS - 1,   \
 295                 SVE_REG_SLICE_SHIFT)
 296 #define SVE_REG_ID_MASK                                                 \
 297         GENMASK(SVE_REG_ID_SHIFT + SVE_REG_ID_BITS - 1, SVE_REG_ID_SHIFT)
 298
 299 #define SVE_NUM_SLICES (1 << SVE_REG_SLICE_BITS)
 300
 301 #define KVM_SVE_ZREG_SIZE KVM_REG_SIZE(KVM_REG_ARM64_SVE_ZREG(0, 0))
 302 #define KVM_SVE_PREG_SIZE KVM_REG_SIZE(KVM_REG_ARM64_SVE_PREG(0, 0))
 303
 304 /*
 305  * Number of register slices required to cover each whole SVE register.
 306  * NOTE: Only the first slice every exists, for now.
 307  * If you are tempted to modify this, you must also rework sve_reg_to_region()
 308  * to match:
 309  */
 310 #define vcpu_sve_slices(vcpu) 1
 311
 312 /* Bounds of a single SVE register slice within vcpu->arch.sve_state */
 313 struct sve_state_reg_region {
 314         unsigned int koffset;   /* offset into sve_state in kernel memory */
 315         unsigned int klen;      /* length in kernel memory */
 316         unsigned int upad;      /* extra trailing padding in user memory */
 317 };
 318
 319 /*
 320  * Validate SVE register ID and get sanitised bounds for user/kernel SVE
 321  * register copy
 322  */
 323 static int sve_reg_to_region(struct sve_state_reg_region *region,
 324                              struct kvm_vcpu *vcpu,
 325                              const struct kvm_one_reg *reg)
 326 {
 327         /* reg ID ranges for Z- registers */
 328         const u64 zreg_id_min = KVM_REG_ARM64_SVE_ZREG(0, 0);
 329         const u64 zreg_id_max = KVM_REG_ARM64_SVE_ZREG(SVE_NUM_ZREGS - 1,
 330                                                        SVE_NUM_SLICES - 1);
 331
 332         /* reg ID ranges for P- registers and FFR (which are contiguous) */
 333         const u64 preg_id_min = KVM_REG_ARM64_SVE_PREG(0, 0);
 334         const u64 preg_id_max = KVM_REG_ARM64_SVE_FFR(SVE_NUM_SLICES - 1);
 335
 336         unsigned int vq;
 337         unsigned int reg_num;
 338
 339         unsigned int reqoffset, reqlen; /* User-requested offset and length */
 340         unsigned int maxlen; /* Maxmimum permitted length */
 341
 342         size_t sve_state_size;
 343
 344         const u64 last_preg_id = KVM_REG_ARM64_SVE_PREG(SVE_NUM_PREGS - 1,
 345                                                         SVE_NUM_SLICES - 1);
 346
 347         /* Verify that the P-regs and FFR really do have contiguous IDs: */
 348         BUILD_BUG_ON(KVM_REG_ARM64_SVE_FFR(0) != last_preg_id + 1);
 349
 350         /* Verify that we match the UAPI header: */
 351         BUILD_BUG_ON(SVE_NUM_SLICES != KVM_ARM64_SVE_MAX_SLICES);
 352
 353         reg_num = (reg->id & SVE_REG_ID_MASK) >> SVE_REG_ID_SHIFT;
 354
 355         if (reg->id >= zreg_id_min && reg->id <= zreg_id_max) {
 356                 if (!vcpu_has_sve(vcpu) || (reg->id & SVE_REG_SLICE_MASK) > 0)
 357                         return -ENOENT;
 358
 359                 vq = sve_vq_from_vl(vcpu->arch.sve_max_vl);
 360
 361                 reqoffset = SVE_SIG_ZREG_OFFSET(vq, reg_num) -
 362                                 SVE_SIG_REGS_OFFSET;
 363                 reqlen = KVM_SVE_ZREG_SIZE;
 364                 maxlen = SVE_SIG_ZREG_SIZE(vq);
 365         } else if (reg->id >= preg_id_min && reg->id <= preg_id_max) {
 366                 if (!vcpu_has_sve(vcpu) || (reg->id & SVE_REG_SLICE_MASK) > 0)
 367                         return -ENOENT;
 368
 369                 vq = sve_vq_from_vl(vcpu->arch.sve_max_vl);
 370
 371                 reqoffset = SVE_SIG_PREG_OFFSET(vq, reg_num) -
 372                                 SVE_SIG_REGS_OFFSET;
 373                 reqlen = KVM_SVE_PREG_SIZE;
 374                 maxlen = SVE_SIG_PREG_SIZE(vq);
 375         } else {
 376                 return -EINVAL;
 377         }
 378
 379         sve_state_size = vcpu_sve_state_size(vcpu);
 380         if (WARN_ON(!sve_state_size))
 381                 return -EINVAL;
 382
 383         region->koffset = array_index_nospec(reqoffset, sve_state_size);
 384         region->klen = min(maxlen, reqlen);
 385         region->upad = reqlen - region->klen;
 386
 387         return 0;
 388 }
 389
 390 static int get_sve_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 391 {
 392         int ret;
 393         struct sve_state_reg_region region;
 394         char __user *uptr = (char __user *)reg->addr;
 395
 396         /* Handle the KVM_REG_ARM64_SVE_VLS pseudo-reg as a special case: */
 397         if (reg->id == KVM_REG_ARM64_SVE_VLS)
 398                 return get_sve_vls(vcpu, reg);
 399
 400         /* Try to interpret reg ID as an architectural SVE register... */
 401         ret = sve_reg_to_region(&region, vcpu, reg);
 402         if (ret)
 403                 return ret;
 404
 405         if (!kvm_arm_vcpu_sve_finalized(vcpu))
 406                 return -EPERM;
 407
 408         if (copy_to_user(uptr, vcpu->arch.sve_state + region.koffset,
 409                          region.klen) ||
 410             clear_user(uptr + region.klen, region.upad))
 411                 return -EFAULT;
 412
 413         return 0;
 414 }
 415
 416 static int set_sve_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 417 {
 418         int ret;
 419         struct sve_state_reg_region region;
 420         const char __user *uptr = (const char __user *)reg->addr;
 421
 422         /* Handle the KVM_REG_ARM64_SVE_VLS pseudo-reg as a special case: */
 423         if (reg->id == KVM_REG_ARM64_SVE_VLS)
 424                 return set_sve_vls(vcpu, reg);
 425
 426         /* Try to interpret reg ID as an architectural SVE register... */
 427         ret = sve_reg_to_region(&region, vcpu, reg);
 428         if (ret)
 429                 return ret;
 430
 431         if (!kvm_arm_vcpu_sve_finalized(vcpu))
 432                 return -EPERM;
 433
 434         if (copy_from_user(vcpu->arch.sve_state + region.koffset, uptr,
 435                            region.klen))
 436                 return -EFAULT;
 437
 438         return 0;
 439 }
 440
 441 int kvm_arch_vcpu_ioctl_get_regs(struct kvm_vcpu *vcpu, struct kvm_regs *regs)
 442 {
 443         return -EINVAL;
 444 }
 445
 446 int kvm_arch_vcpu_ioctl_set_regs(struct kvm_vcpu *vcpu, struct kvm_regs *regs)
 447 {
 448         return -EINVAL;
 449 }
 450
 451 static int copy_core_reg_indices(const struct kvm_vcpu *vcpu,
 452                                  u64 __user *uindices)
 453 {
 454         unsigned int i;
 455         int n = 0;
 456
 457         for (i = 0; i < sizeof(struct kvm_regs) / sizeof(__u32); i++) {
 458                 u64 reg = KVM_REG_ARM64 | KVM_REG_ARM_CORE | i;
 459                 int size = core_reg_size_from_offset(vcpu, i);
 460
 461                 if (size < 0)
 462                         continue;
 463
 464                 switch (size) {
 465                 case sizeof(__u32):
 466                         reg |= KVM_REG_SIZE_U32;
 467                         break;
 468
 469                 case sizeof(__u64):
 470                         reg |= KVM_REG_SIZE_U64;
 471                         break;
 472
 473                 case sizeof(__uint128_t):
 474                         reg |= KVM_REG_SIZE_U128;
 475                         break;
 476
 477                 default:
 478                         WARN_ON(1);
 479                         continue;
 480                 }
 481
 482                 if (uindices) {
 483                         if (put_user(reg, uindices))
 484                                 return -EFAULT;
 485                         uindices++;
 486                 }
 487
 488                 n++;
 489         }
 490
 491         return n;
 492 }
 493
 494 static unsigned long num_core_regs(const struct kvm_vcpu *vcpu)
 495 {
 496         return copy_core_reg_indices(vcpu, NULL);
 497 }
 498
 499 /**
 500  * ARM64 versions of the TIMER registers, always available on arm64
 501  */
 502
 503 #define NUM_TIMER_REGS 3
 504
 505 static bool is_timer_reg(u64 index)
 506 {
 507         switch (index) {
 508         case KVM_REG_ARM_TIMER_CTL:
 509         case KVM_REG_ARM_TIMER_CNT:
 510         case KVM_REG_ARM_TIMER_CVAL:
 511                 return true;
 512         }
 513         return false;
 514 }
 515
 516 static int copy_timer_indices(struct kvm_vcpu *vcpu, u64 __user *uindices)
 517 {
 518         if (put_user(KVM_REG_ARM_TIMER_CTL, uindices))
 519                 return -EFAULT;
 520         uindices++;
 521         if (put_user(KVM_REG_ARM_TIMER_CNT, uindices))
 522                 return -EFAULT;
 523         uindices++;
 524         if (put_user(KVM_REG_ARM_TIMER_CVAL, uindices))
 525                 return -EFAULT;
 526
 527         return 0;
 528 }
 529
 530 static int set_timer_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 531 {
 532         void __user *uaddr = (void __user *)(long)reg->addr;
 533         u64 val;
 534         int ret;
 535
 536         ret = copy_from_user(&val, uaddr, KVM_REG_SIZE(reg->id));
 537         if (ret != 0)
 538                 return -EFAULT;
 539
 540         return kvm_arm_timer_set_reg(vcpu, reg->id, val);
 541 }
 542
 543 static int get_timer_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 544 {
 545         void __user *uaddr = (void __user *)(long)reg->addr;
 546         u64 val;
 547
 548         val = kvm_arm_timer_get_reg(vcpu, reg->id);
 549         return copy_to_user(uaddr, &val, KVM_REG_SIZE(reg->id)) ? -EFAULT : 0;
 550 }
 551
 552 static unsigned long num_sve_regs(const struct kvm_vcpu *vcpu)
 553 {
 554         const unsigned int slices = vcpu_sve_slices(vcpu);
 555
 556         if (!vcpu_has_sve(vcpu))
 557                 return 0;
 558
 559         /* Policed by KVM_GET_REG_LIST: */
 560         WARN_ON(!kvm_arm_vcpu_sve_finalized(vcpu));
 561
 562         return slices * (SVE_NUM_PREGS + SVE_NUM_ZREGS + 1 /* FFR */)
 563                 + 1; /* KVM_REG_ARM64_SVE_VLS */
 564 }
 565
 566 static int copy_sve_reg_indices(const struct kvm_vcpu *vcpu,
 567                                 u64 __user *uindices)
 568 {
 569         const unsigned int slices = vcpu_sve_slices(vcpu);
 570         u64 reg;
 571         unsigned int i, n;
 572         int num_regs = 0;
 573
 574         if (!vcpu_has_sve(vcpu))
 575                 return 0;
 576
 577         /* Policed by KVM_GET_REG_LIST: */
 578         WARN_ON(!kvm_arm_vcpu_sve_finalized(vcpu));
 579
 580         /*
 581          * Enumerate this first, so that userspace can save/restore in
 582          * the order reported by KVM_GET_REG_LIST:
 583          */
 584         reg = KVM_REG_ARM64_SVE_VLS;
 585         if (put_user(reg, uindices++))
 586                 return -EFAULT;
 587         ++num_regs;
 588
 589         for (i = 0; i < slices; i++) {
 590                 for (n = 0; n < SVE_NUM_ZREGS; n++) {
 591                         reg = KVM_REG_ARM64_SVE_ZREG(n, i);
 592                         if (put_user(reg, uindices++))
 593                                 return -EFAULT;
 594                         num_regs++;
 595                 }
 596
 597                 for (n = 0; n < SVE_NUM_PREGS; n++) {
 598                         reg = KVM_REG_ARM64_SVE_PREG(n, i);
 599                         if (put_user(reg, uindices++))
 600                                 return -EFAULT;
 601                         num_regs++;
 602                 }
 603
 604                 reg = KVM_REG_ARM64_SVE_FFR(i);
 605                 if (put_user(reg, uindices++))
 606                         return -EFAULT;
 607                 num_regs++;
 608         }
 609
 610         return num_regs;
 611 }
 612
 613 /**
 614  * kvm_arm_num_regs - how many registers do we present via KVM_GET_ONE_REG
 615  *
 616  * This is for all registers.
 617  */
 618 unsigned long kvm_arm_num_regs(struct kvm_vcpu *vcpu)
 619 {
 620         unsigned long res = 0;
 621
 622         res += num_core_regs(vcpu);
 623         res += num_sve_regs(vcpu);
 624         res += kvm_arm_num_sys_reg_descs(vcpu);
 625         res += kvm_arm_get_fw_num_regs(vcpu);
 626         res += NUM_TIMER_REGS;
 627
 628         return res;
 629 }
 630
 631 /**
 632  * kvm_arm_copy_reg_indices - get indices of all registers.
 633  *
 634  * We do core registers right here, then we append system regs.
 635  */
 636 int kvm_arm_copy_reg_indices(struct kvm_vcpu *vcpu, u64 __user *uindices)
 637 {
 638         int ret;
 639
 640         ret = copy_core_reg_indices(vcpu, uindices);
 641         if (ret < 0)
 642                 return ret;
 643         uindices += ret;
 644
 645         ret = copy_sve_reg_indices(vcpu, uindices);
 646         if (ret < 0)
 647                 return ret;
 648         uindices += ret;
 649
 650         ret = kvm_arm_copy_fw_reg_indices(vcpu, uindices);
 651         if (ret < 0)
 652                 return ret;
 653         uindices += kvm_arm_get_fw_num_regs(vcpu);
 654
 655         ret = copy_timer_indices(vcpu, uindices);
 656         if (ret < 0)
 657                 return ret;
 658         uindices += NUM_TIMER_REGS;
 659
 660         return kvm_arm_copy_sys_reg_indices(vcpu, uindices);
 661 }
 662
 663 int kvm_arm_get_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 664 {
 665         /* We currently use nothing arch-specific in upper 32 bits */
 666         if ((reg->id & ~KVM_REG_SIZE_MASK) >> 32 != KVM_REG_ARM64 >> 32)
 667                 return -EINVAL;
 668
 669         switch (reg->id & KVM_REG_ARM_COPROC_MASK) {
 670         case KVM_REG_ARM_CORE:  return get_core_reg(vcpu, reg);
 671         case KVM_REG_ARM_FW:    return kvm_arm_get_fw_reg(vcpu, reg);
 672         case KVM_REG_ARM64_SVE: return get_sve_reg(vcpu, reg);
 673         }
 674
 675         if (is_timer_reg(reg->id))
 676                 return get_timer_reg(vcpu, reg);
 677
 678         return kvm_arm_sys_reg_get_reg(vcpu, reg);
 679 }
 680
 681 int kvm_arm_set_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
 682 {
 683         /* We currently use nothing arch-specific in upper 32 bits */
 684         if ((reg->id & ~KVM_REG_SIZE_MASK) >> 32 != KVM_REG_ARM64 >> 32)
 685                 return -EINVAL;
 686
 687         switch (reg->id & KVM_REG_ARM_COPROC_MASK) {
 688         case KVM_REG_ARM_CORE:  return set_core_reg(vcpu, reg);
 689         case KVM_REG_ARM_FW:    return kvm_arm_set_fw_reg(vcpu, reg);
 690         case KVM_REG_ARM64_SVE: return set_sve_reg(vcpu, reg);
 691         }
 692
 693         if (is_timer_reg(reg->id))
 694                 return set_timer_reg(vcpu, reg);
 695
 696         return kvm_arm_sys_reg_set_reg(vcpu, reg);
 697 }
 698
 699 int kvm_arch_vcpu_ioctl_get_sregs(struct kvm_vcpu *vcpu,
 700                                   struct kvm_sregs *sregs)
 701 {
 702         return -EINVAL;
 703 }
 704
 705 int kvm_arch_vcpu_ioctl_set_sregs(struct kvm_vcpu *vcpu,
 706                                   struct kvm_sregs *sregs)
 707 {
 708         return -EINVAL;
 709 }
 710
 711 int __kvm_arm_vcpu_get_events(struct kvm_vcpu *vcpu,
 712                               struct kvm_vcpu_events *events)
 713 {
 714         events->exception.serror_pending = !!(vcpu->arch.hcr_el2 & HCR_VSE);
 715         events->exception.serror_has_esr = cpus_have_const_cap(ARM64_HAS_RAS_EXTN);
 716
 717         if (events->exception.serror_pending && events->exception.serror_has_esr)
 718                 events->exception.serror_esr = vcpu_get_vsesr(vcpu);
 719
 720         /*
 721          * We never return a pending ext_dabt here because we deliver it to
 722          * the virtual CPU directly when setting the event and it's no longer
 723          * 'pending' at this point.
 724          */
 725
 726         return 0;
 727 }
 728
 729 int __kvm_arm_vcpu_set_events(struct kvm_vcpu *vcpu,
 730                               struct kvm_vcpu_events *events)
 731 {
 732         bool serror_pending = events->exception.serror_pending;
 733         bool has_esr = events->exception.serror_has_esr;
 734         bool ext_dabt_pending = events->exception.ext_dabt_pending;
 735
 736         if (serror_pending && has_esr) {
 737                 if (!cpus_have_const_cap(ARM64_HAS_RAS_EXTN))
 738                         return -EINVAL;
 739
 740                 if (!((events->exception.serror_esr) & ~ESR_ELx_ISS_MASK))
 741                         kvm_set_sei_esr(vcpu, events->exception.serror_esr);
 742                 else
 743                         return -EINVAL;
 744         } else if (serror_pending) {
 745                 kvm_inject_vabt(vcpu);
 746         }
 747
 748         if (ext_dabt_pending)
 749                 kvm_inject_dabt(vcpu, kvm_vcpu_get_hfar(vcpu));
 750
 751         return 0;
 752 }
 753
 754 int __attribute_const__ kvm_target_cpu(void)
 755 {
 756         unsigned long implementor = read_cpuid_implementor();
 757         unsigned long part_number = read_cpuid_part_number();
 758
 759         switch (implementor) {
 760         case ARM_CPU_IMP_ARM:
 761                 switch (part_number) {
 762                 case ARM_CPU_PART_AEM_V8:
 763                         return KVM_ARM_TARGET_AEM_V8;
 764                 case ARM_CPU_PART_FOUNDATION:
 765                         return KVM_ARM_TARGET_FOUNDATION_V8;
 766                 case ARM_CPU_PART_CORTEX_A53:
 767                         return KVM_ARM_TARGET_CORTEX_A53;
 768                 case ARM_CPU_PART_CORTEX_A57:
 769                         return KVM_ARM_TARGET_CORTEX_A57;
 770                 }
 771                 break;
 772         case ARM_CPU_IMP_APM:
 773                 switch (part_number) {
 774                 case APM_CPU_PART_POTENZA:
 775                         return KVM_ARM_TARGET_XGENE_POTENZA;
 776                 }
 777                 break;
 778         }
 779
 780         /* Return a default generic target */
 781         return KVM_ARM_TARGET_GENERIC_V8;
 782 }
 783
 784 int kvm_vcpu_preferred_target(struct kvm_vcpu_init *init)
 785 {
 786         int target = kvm_target_cpu();
 787
 788         if (target < 0)
 789                 return -ENODEV;
 790
 791         memset(init, 0, sizeof(*init));
 792
 793         /*
 794          * For now, we don't return any features.
 795          * In future, we might use features to return target
 796          * specific features available for the preferred
 797          * target type.
 798          */
 799         init->target = (__u32)target;
 800
 801         return 0;
 802 }
 803
 804 int kvm_arch_vcpu_ioctl_get_fpu(struct kvm_vcpu *vcpu, struct kvm_fpu *fpu)
 805 {
 806         return -EINVAL;
 807 }
 808
 809 int kvm_arch_vcpu_ioctl_set_fpu(struct kvm_vcpu *vcpu, struct kvm_fpu *fpu)
 810 {
 811         return -EINVAL;
 812 }
 813
 814 int kvm_arch_vcpu_ioctl_translate(struct kvm_vcpu *vcpu,
 815                                   struct kvm_translation *tr)
 816 {
 817         return -EINVAL;
 818 }
 819
 820 #define KVM_GUESTDBG_VALID_MASK (KVM_GUESTDBG_ENABLE |    \
 821                             KVM_GUESTDBG_USE_SW_BP | \
 822                             KVM_GUESTDBG_USE_HW | \
 823                             KVM_GUESTDBG_SINGLESTEP)
 824
 825 /**
 826  * kvm_arch_vcpu_ioctl_set_guest_debug - set up guest debugging
 827  * @kvm:        pointer to the KVM struct
 828  * @kvm_guest_debug: the ioctl data buffer
 829  *
 830  * This sets up and enables the VM for guest debugging. Userspace
 831  * passes in a control flag to enable different debug types and
 832  * potentially other architecture specific information in the rest of
 833  * the structure.
 834  */
 835 int kvm_arch_vcpu_ioctl_set_guest_debug(struct kvm_vcpu *vcpu,
 836                                         struct kvm_guest_debug *dbg)
 837 {
 838         int ret = 0;
 839
 840         trace_kvm_set_guest_debug(vcpu, dbg->control);
 841
 842         if (dbg->control & ~KVM_GUESTDBG_VALID_MASK) {
 843                 ret = -EINVAL;
 844                 goto out;
 845         }
 846
 847         if (dbg->control & KVM_GUESTDBG_ENABLE) {
 848                 vcpu->guest_debug = dbg->control;
 849
 850                 /* Hardware assisted Break and Watch points */
 851                 if (vcpu->guest_debug & KVM_GUESTDBG_USE_HW) {
 852                         vcpu->arch.external_debug_state = dbg->arch;
 853                 }
 854
 855         } else {
 856                 /* If not enabled clear all flags */
 857                 vcpu->guest_debug = 0;
 858         }
 859
 860 out:
 861         return ret;
 862 }
 863
 864 int kvm_arm_vcpu_arch_set_attr(struct kvm_vcpu *vcpu,
 865                                struct kvm_device_attr *attr)
 866 {
 867         int ret;
 868
 869         switch (attr->group) {
 870         case KVM_ARM_VCPU_PMU_V3_CTRL:
 871                 ret = kvm_arm_pmu_v3_set_attr(vcpu, attr);
 872                 break;
 873         case KVM_ARM_VCPU_TIMER_CTRL:
 874                 ret = kvm_arm_timer_set_attr(vcpu, attr);
 875                 break;
 876         case KVM_ARM_VCPU_PVTIME_CTRL:
 877                 ret = kvm_arm_pvtime_set_attr(vcpu, attr);
 878                 break;
 879         default:
 880                 ret = -ENXIO;
 881                 break;
 882         }
 883
 884         return ret;
 885 }
 886
 887 int kvm_arm_vcpu_arch_get_attr(struct kvm_vcpu *vcpu,
 888                                struct kvm_device_attr *attr)
 889 {
 890         int ret;
 891
 892         switch (attr->group) {
 893         case KVM_ARM_VCPU_PMU_V3_CTRL:
 894                 ret = kvm_arm_pmu_v3_get_attr(vcpu, attr);
 895                 break;
 896         case KVM_ARM_VCPU_TIMER_CTRL:
 897                 ret = kvm_arm_timer_get_attr(vcpu, attr);
 898                 break;
 899         case KVM_ARM_VCPU_PVTIME_CTRL:
 900                 ret = kvm_arm_pvtime_get_attr(vcpu, attr);
 901                 break;
 902         default:
 903                 ret = -ENXIO;
 904                 break;
 905         }
 906
 907         return ret;
 908 }
 909
 910 int kvm_arm_vcpu_arch_has_attr(struct kvm_vcpu *vcpu,
 911                                struct kvm_device_attr *attr)
 912 {
 913         int ret;
 914
 915         switch (attr->group) {
 916         case KVM_ARM_VCPU_PMU_V3_CTRL:
 917                 ret = kvm_arm_pmu_v3_has_attr(vcpu, attr);
 918                 break;
 919         case KVM_ARM_VCPU_TIMER_CTRL:
 920                 ret = kvm_arm_timer_has_attr(vcpu, attr);
 921                 break;
 922         case KVM_ARM_VCPU_PVTIME_CTRL:
 923                 ret = kvm_arm_pvtime_has_attr(vcpu, attr);
 924                 break;
 925         default:
 926                 ret = -ENXIO;
 927                 break;
 928         }
 929
 930         return ret;
 931 }