Hi Dave,

On 19/03/2019 17:52, Dave Martin wrote:
> This patch adds the following registers for access via the
> KVM_{GET,SET}_ONE_REG interface:
> 
>  * KVM_REG_ARM64_SVE_ZREG(n, i) (n = 0..31) (in 2048-bit slices)
>  * KVM_REG_ARM64_SVE_PREG(n, i) (n = 0..15) (in 256-bit slices)
>  * KVM_REG_ARM64_SVE_FFR(i) (in 256-bit slices)
> 
> In order to adapt gracefully to future architectural extensions,
> the registers are logically divided up into slices as noted above:
> the i parameter denotes the slice index.
> 
> This allows us to reserve space in the ABI for future expansion of
> these registers.  However, as of today the architecture does not
> permit registers to be larger than a single slice, so no code is
> needed in the kernel to expose additional slices, for now.  The
> code can be extended later as needed to expose them up to a maximum
> of 32 slices (as carved out in the architecture itself) if they
> really exist someday.
> 
> The registers are only visible for vcpus that have SVE enabled.
> They are not enumerated by KVM_GET_REG_LIST on vcpus that do not
> have SVE.
> 
> Accesses to the FPSIMD registers via KVM_REG_ARM_CORE is not
> allowed for SVE-enabled vcpus: SVE-aware userspace can use the
> KVM_REG_ARM64_SVE_ZREG() interface instead to access the same
> register state.  This avoids some complex and pointless emulation
> in the kernel to convert between the two views of these aliased
> registers.
> 
> Signed-off-by: Dave Martin <[email protected]>
> 

Maybe it's because I already had reviewed the previous iteration, but
this time things do seem a bit clearer.

Reviewed-by: Julien Thierry <[email protected]>

Thanks,

Julien

> ---
> 
> Changes since v5:
> 
>  * [Julien Thierry] rename sve_reg_region() to sve_reg_to_region() to
>    make its purpose a bit clearer.
> 
>  * [Julien Thierry] rename struct sve_state_region to
>    sve_state_reg_region to make it clearer this this struct only
>    describes the bounds of (part of) a single register within
>    sve_state.
> 
>  * [Julien Thierry] Add a comment to clarify the purpose of struct
>    sve_state_reg_region.
> ---
>  arch/arm64/include/asm/kvm_host.h |  14 ++++
>  arch/arm64/include/uapi/asm/kvm.h |  17 +++++
>  arch/arm64/kvm/guest.c            | 139 
> ++++++++++++++++++++++++++++++++++----
>  3 files changed, 158 insertions(+), 12 deletions(-)
> 
> diff --git a/arch/arm64/include/asm/kvm_host.h 
> b/arch/arm64/include/asm/kvm_host.h
> index 4fabfd2..205438a 100644
> --- a/arch/arm64/include/asm/kvm_host.h
> +++ b/arch/arm64/include/asm/kvm_host.h
> @@ -329,6 +329,20 @@ struct kvm_vcpu_arch {
>  #define vcpu_sve_pffr(vcpu) ((void *)((char *)((vcpu)->arch.sve_state) + \
>                                     sve_ffr_offset((vcpu)->arch.sve_max_vl)))
>  
> +#define vcpu_sve_state_size(vcpu) ({                                 \
> +     size_t __size_ret;                                              \
> +     unsigned int __vcpu_vq;                                         \
> +                                                                     \
> +     if (WARN_ON(!sve_vl_valid((vcpu)->arch.sve_max_vl))) {          \
> +             __size_ret = 0;                                         \
> +     } else {                                                        \
> +             __vcpu_vq = sve_vq_from_vl((vcpu)->arch.sve_max_vl);    \
> +             __size_ret = SVE_SIG_REGS_SIZE(__vcpu_vq);              \
> +     }                                                               \
> +                                                                     \
> +     __size_ret;                                                     \
> +})
> +
>  /* vcpu_arch flags field values: */
>  #define KVM_ARM64_DEBUG_DIRTY                (1 << 0)
>  #define KVM_ARM64_FP_ENABLED         (1 << 1) /* guest FP regs loaded */
> diff --git a/arch/arm64/include/uapi/asm/kvm.h 
> b/arch/arm64/include/uapi/asm/kvm.h
> index 97c3478..ced760c 100644
> --- a/arch/arm64/include/uapi/asm/kvm.h
> +++ b/arch/arm64/include/uapi/asm/kvm.h
> @@ -226,6 +226,23 @@ struct kvm_vcpu_events {
>                                        KVM_REG_ARM_FW | ((r) & 0xffff))
>  #define KVM_REG_ARM_PSCI_VERSION     KVM_REG_ARM_FW_REG(0)
>  
> +/* SVE registers */
> +#define KVM_REG_ARM64_SVE            (0x15 << KVM_REG_ARM_COPROC_SHIFT)
> +
> +/* Z- and P-regs occupy blocks at the following offsets within this range: */
> +#define KVM_REG_ARM64_SVE_ZREG_BASE  0
> +#define KVM_REG_ARM64_SVE_PREG_BASE  0x400
> +
> +#define KVM_REG_ARM64_SVE_ZREG(n, i) (KVM_REG_ARM64 | KVM_REG_ARM64_SVE | \
> +                                      KVM_REG_ARM64_SVE_ZREG_BASE |  \
> +                                      KVM_REG_SIZE_U2048 |           \
> +                                      ((n) << 5) | (i))
> +#define KVM_REG_ARM64_SVE_PREG(n, i) (KVM_REG_ARM64 | KVM_REG_ARM64_SVE | \
> +                                      KVM_REG_ARM64_SVE_PREG_BASE |  \
> +                                      KVM_REG_SIZE_U256 |            \
> +                                      ((n) << 5) | (i))
> +#define KVM_REG_ARM64_SVE_FFR(i)     KVM_REG_ARM64_SVE_PREG(16, i)
> +
>  /* Device Control API: ARM VGIC */
>  #define KVM_DEV_ARM_VGIC_GRP_ADDR    0
>  #define KVM_DEV_ARM_VGIC_GRP_DIST_REGS       1
> diff --git a/arch/arm64/kvm/guest.c b/arch/arm64/kvm/guest.c
> index 756d0d6..736d8cb 100644
> --- a/arch/arm64/kvm/guest.c
> +++ b/arch/arm64/kvm/guest.c
> @@ -19,8 +19,11 @@
>   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
>   */
>  
> +#include <linux/bits.h>
>  #include <linux/errno.h>
>  #include <linux/err.h>
> +#include <linux/nospec.h>
> +#include <linux/kernel.h>
>  #include <linux/kvm_host.h>
>  #include <linux/module.h>
>  #include <linux/stddef.h>
> @@ -30,9 +33,12 @@
>  #include <kvm/arm_psci.h>
>  #include <asm/cputype.h>
>  #include <linux/uaccess.h>
> +#include <asm/fpsimd.h>
>  #include <asm/kvm.h>
>  #include <asm/kvm_emulate.h>
>  #include <asm/kvm_coproc.h>
> +#include <asm/kvm_host.h>
> +#include <asm/sigcontext.h>
>  
>  #include "trace.h"
>  
> @@ -200,6 +206,115 @@ static int set_core_reg(struct kvm_vcpu *vcpu, const 
> struct kvm_one_reg *reg)
>       return err;
>  }
>  
> +#define SVE_REG_SLICE_SHIFT  0
> +#define SVE_REG_SLICE_BITS   5
> +#define SVE_REG_ID_SHIFT     (SVE_REG_SLICE_SHIFT + SVE_REG_SLICE_BITS)
> +#define SVE_REG_ID_BITS              5
> +
> +#define SVE_REG_SLICE_MASK                                   \
> +     GENMASK(SVE_REG_SLICE_SHIFT + SVE_REG_SLICE_BITS - 1,   \
> +             SVE_REG_SLICE_SHIFT)
> +#define SVE_REG_ID_MASK                                                      
> \
> +     GENMASK(SVE_REG_ID_SHIFT + SVE_REG_ID_BITS - 1, SVE_REG_ID_SHIFT)
> +
> +#define SVE_NUM_SLICES (1 << SVE_REG_SLICE_BITS)
> +
> +#define KVM_SVE_ZREG_SIZE KVM_REG_SIZE(KVM_REG_ARM64_SVE_ZREG(0, 0))
> +#define KVM_SVE_PREG_SIZE KVM_REG_SIZE(KVM_REG_ARM64_SVE_PREG(0, 0))
> +
> +/* Bounds of a single SVE register slice within vcpu->arch.sve_state */
> +struct sve_state_reg_region {
> +     unsigned int koffset;   /* offset into sve_state in kernel memory */
> +     unsigned int klen;      /* length in kernel memory */
> +     unsigned int upad;      /* extra trailing padding in user memory */
> +};
> +
> +/* Get sanitised bounds for user/kernel SVE register copy */
> +static int sve_reg_to_region(struct sve_state_reg_region *region,
> +                          struct kvm_vcpu *vcpu,
> +                          const struct kvm_one_reg *reg)
> +{
> +     /* reg ID ranges for Z- registers */
> +     const u64 zreg_id_min = KVM_REG_ARM64_SVE_ZREG(0, 0);
> +     const u64 zreg_id_max = KVM_REG_ARM64_SVE_ZREG(SVE_NUM_ZREGS - 1,
> +                                                    SVE_NUM_SLICES - 1);
> +
> +     /* reg ID ranges for P- registers and FFR (which are contiguous) */
> +     const u64 preg_id_min = KVM_REG_ARM64_SVE_PREG(0, 0);
> +     const u64 preg_id_max = KVM_REG_ARM64_SVE_FFR(SVE_NUM_SLICES - 1);
> +
> +     unsigned int vq;
> +     unsigned int reg_num;
> +
> +     unsigned int reqoffset, reqlen; /* User-requested offset and length */
> +     unsigned int maxlen; /* Maxmimum permitted length */
> +
> +     size_t sve_state_size;
> +
> +     /* Only the first slice ever exists, for now: */
> +     if ((reg->id & SVE_REG_SLICE_MASK) != 0)
> +             return -ENOENT;
> +
> +     vq = sve_vq_from_vl(vcpu->arch.sve_max_vl);
> +
> +     reg_num = (reg->id & SVE_REG_ID_MASK) >> SVE_REG_ID_SHIFT;
> +
> +     if (reg->id >= zreg_id_min && reg->id <= zreg_id_max) {
> +             reqoffset = SVE_SIG_ZREG_OFFSET(vq, reg_num) -
> +                             SVE_SIG_REGS_OFFSET;
> +             reqlen = KVM_SVE_ZREG_SIZE;
> +             maxlen = SVE_SIG_ZREG_SIZE(vq);
> +     } else if (reg->id >= preg_id_min && reg->id <= preg_id_max) {
> +             reqoffset = SVE_SIG_PREG_OFFSET(vq, reg_num) -
> +                             SVE_SIG_REGS_OFFSET;
> +             reqlen = KVM_SVE_PREG_SIZE;
> +             maxlen = SVE_SIG_PREG_SIZE(vq);
> +     } else {
> +             return -ENOENT;
> +     }
> +
> +     sve_state_size = vcpu_sve_state_size(vcpu);
> +     if (!sve_state_size)
> +             return -EINVAL;
> +
> +     region->koffset = array_index_nospec(reqoffset, sve_state_size);
> +     region->klen = min(maxlen, reqlen);
> +     region->upad = reqlen - region->klen;
> +
> +     return 0;
> +}
> +
> +static int get_sve_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
> +{
> +     struct sve_state_reg_region region;
> +     char __user *uptr = (char __user *)reg->addr;
> +
> +     if (!vcpu_has_sve(vcpu) || sve_reg_to_region(&region, vcpu, reg))
> +             return -ENOENT;
> +
> +     if (copy_to_user(uptr, vcpu->arch.sve_state + region.koffset,
> +                      region.klen) ||
> +         clear_user(uptr + region.klen, region.upad))
> +             return -EFAULT;
> +
> +     return 0;
> +}
> +
> +static int set_sve_reg(struct kvm_vcpu *vcpu, const struct kvm_one_reg *reg)
> +{
> +     struct sve_state_reg_region region;
> +     const char __user *uptr = (const char __user *)reg->addr;
> +
> +     if (!vcpu_has_sve(vcpu) || sve_reg_to_region(&region, vcpu, reg))
> +             return -ENOENT;
> +
> +     if (copy_from_user(vcpu->arch.sve_state + region.koffset, uptr,
> +                        region.klen))
> +             return -EFAULT;
> +
> +     return 0;
> +}
> +
>  int kvm_arch_vcpu_ioctl_get_regs(struct kvm_vcpu *vcpu, struct kvm_regs 
> *regs)
>  {
>       return -EINVAL;
> @@ -346,12 +461,12 @@ int kvm_arm_get_reg(struct kvm_vcpu *vcpu, const struct 
> kvm_one_reg *reg)
>       if ((reg->id & ~KVM_REG_SIZE_MASK) >> 32 != KVM_REG_ARM64 >> 32)
>               return -EINVAL;
>  
> -     /* Register group 16 means we want a core register. */
> -     if ((reg->id & KVM_REG_ARM_COPROC_MASK) == KVM_REG_ARM_CORE)
> -             return get_core_reg(vcpu, reg);
> -
> -     if ((reg->id & KVM_REG_ARM_COPROC_MASK) == KVM_REG_ARM_FW)
> -             return kvm_arm_get_fw_reg(vcpu, reg);
> +     switch (reg->id & KVM_REG_ARM_COPROC_MASK) {
> +     case KVM_REG_ARM_CORE:  return get_core_reg(vcpu, reg);
> +     case KVM_REG_ARM_FW:    return kvm_arm_get_fw_reg(vcpu, reg);
> +     case KVM_REG_ARM64_SVE: return get_sve_reg(vcpu, reg);
> +     default: break; /* fall through */
> +     }
>  
>       if (is_timer_reg(reg->id))
>               return get_timer_reg(vcpu, reg);
> @@ -365,12 +480,12 @@ int kvm_arm_set_reg(struct kvm_vcpu *vcpu, const struct 
> kvm_one_reg *reg)
>       if ((reg->id & ~KVM_REG_SIZE_MASK) >> 32 != KVM_REG_ARM64 >> 32)
>               return -EINVAL;
>  
> -     /* Register group 16 means we set a core register. */
> -     if ((reg->id & KVM_REG_ARM_COPROC_MASK) == KVM_REG_ARM_CORE)
> -             return set_core_reg(vcpu, reg);
> -
> -     if ((reg->id & KVM_REG_ARM_COPROC_MASK) == KVM_REG_ARM_FW)
> -             return kvm_arm_set_fw_reg(vcpu, reg);
> +     switch (reg->id & KVM_REG_ARM_COPROC_MASK) {
> +     case KVM_REG_ARM_CORE:  return set_core_reg(vcpu, reg);
> +     case KVM_REG_ARM_FW:    return kvm_arm_set_fw_reg(vcpu, reg);
> +     case KVM_REG_ARM64_SVE: return set_sve_reg(vcpu, reg);
> +     default: break; /* fall through */
> +     }
>  
>       if (is_timer_reg(reg->id))
>               return set_timer_reg(vcpu, reg);
> 

-- 
Julien Thierry
_______________________________________________
kvmarm mailing list
[email protected]
https://lists.cs.columbia.edu/mailman/listinfo/kvmarm

Reply via email to