Add capabilities for pointer authentication

Add the capabilities for pointer authentication in Aarch64,
KVM_CAP_ARM_PTRAUTH_GENERIC and KVM_CAP_ARM_PTRAUTH_ADDRESS. Also, add a
unit test for checking the existence of capabilities and try to enable
the feature in the vcpu initialization if present.

Signed-off-by: Babis Chalios <[email protected]>

Author: bchalios

CHANGELOG.md

@@ -6,6 +6,8 @@
   of `Cap` enum.
 - [[#228](https://github.com/rust-vmm/kvm-ioctls/pull/228)] arm64: add
 support for vCPU SVE feature.
+- [[#219](https://github.com/rust-vmm/kvm-ioctls/pull/226)] Add `Cap::ArmPtrAuthAddress`
+  and `Cap::ArmPtrAuthGeneric` capabilities.
 
 # v0.14.0
 

src/cap.rs

@@ -150,4 +150,8 @@ pub enum Cap {
     GetMsrFeatures = KVM_CAP_GET_MSR_FEATURES,
     #[cfg(target_arch = "aarch64")]
     ArmSve = KVM_CAP_ARM_SVE,
+    #[cfg(any(target_arch = "arm", target_arch = "aarch64"))]
+    ArmPtrAuthAddress = KVM_CAP_ARM_PTRAUTH_ADDRESS,
+    #[cfg(any(target_arch = "arm", target_arch = "aarch64"))]
+    ArmPtrAuthGeneric = KVM_CAP_ARM_PTRAUTH_GENERIC,
 }

src/ioctls/vcpu.rs

@@ -2785,4 +2785,23 @@ mod tests {
         assert!(vcpu.has_device_attr(&dist_attr).is_ok());
         assert!(vcpu.set_device_attr(&dist_attr).is_ok());
     }
+
+    #[test]
+    #[cfg(target_arch = "aarch64")]
+    fn test_pointer_authentication() {
+        let kvm = Kvm::new().unwrap();
+        let vm = kvm.create_vm().unwrap();
+        let vcpu = vm.create_vcpu(0).unwrap();
+
+        let mut kvi: kvm_bindings::kvm_vcpu_init = kvm_bindings::kvm_vcpu_init::default();
+        vm.get_preferred_target(&mut kvi)
+            .expect("Cannot get preferred target");
+        if kvm.check_extension(Cap::ArmPtrAuthAddress) {
+            kvi.features[0] |= 1 << kvm_bindings::KVM_ARM_VCPU_PTRAUTH_ADDRESS;
+        }
+        if kvm.check_extension(Cap::ArmPtrAuthGeneric) {
+            kvi.features[0] |= 1 << kvm_bindings::KVM_ARM_VCPU_PTRAUTH_GENERIC;
+        }
+        assert!(vcpu.vcpu_init(&kvi).is_ok());
+    }
 }