[RFC PATCH v1 17/37] KVM: selftests: Update framework to use KVM_SET_MEMORY_ATTRIBUTES2

[Ackerley Tng]

Update KVM selftest framework to use KVM_SET_MEMORY_ATTRIBUTES2 and the
accompanying struct kvm_memory_attributes2.

Signed-off-by: Ackerley Tng <ackerley...@google.com>
---
 tools/testing/selftests/kvm/include/kvm_util.h | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/tools/testing/selftests/kvm/include/kvm_util.h 
b/tools/testing/selftests/kvm/include/kvm_util.h
index de8ae9be19067..019ffcec4510f 100644
--- a/tools/testing/selftests/kvm/include/kvm_util.h
+++ b/tools/testing/selftests/kvm/include/kvm_util.h
@@ -394,24 +394,30 @@ static inline void vm_enable_cap(struct kvm_vm *vm, 
uint32_t cap, uint64_t arg0)
        vm_ioctl(vm, KVM_ENABLE_CAP, &enable_cap);
 }
 
+#define TEST_REQUIRE_SET_MEMORY_ATTRIBUTES2()                          \
+       __TEST_REQUIRE(kvm_has_cap(KVM_CAP_MEMORY_ATTRIBUTES2),         \
+                      "KVM selftests now require KVM_SET_MEMORY_ATTRIBUTES2")
+
 static inline void vm_set_memory_attributes(struct kvm_vm *vm, uint64_t gpa,
                                            uint64_t size, uint64_t attributes)
 {
-       struct kvm_memory_attributes attr = {
+       struct kvm_memory_attributes2 attr = {
                .attributes = attributes,
                .address = gpa,
                .size = size,
                .flags = 0,
        };
 
+       TEST_REQUIRE_SET_MEMORY_ATTRIBUTES2();
+
        /*
-        * KVM_SET_MEMORY_ATTRIBUTES overwrites _all_ attributes.  These flows
+        * KVM_SET_MEMORY_ATTRIBUTES2 overwrites _all_ attributes.  These flows
         * need significant enhancements to support multiple attributes.
         */
        TEST_ASSERT(!attributes || attributes == KVM_MEMORY_ATTRIBUTE_PRIVATE,
                    "Update me to support multiple attributes!");
 
-       vm_ioctl(vm, KVM_SET_MEMORY_ATTRIBUTES, &attr);
+       vm_ioctl(vm, KVM_SET_MEMORY_ATTRIBUTES2, &attr);
 }
 
 
-- 
2.51.0.858.gf9c4a03a3a-goog