summary refs log tree commit diff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/plugins.rs7
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/plugins.rs b/tests/plugins.rs
index 7d2d8ff..987b033 100644
--- a/tests/plugins.rs
+++ b/tests/plugins.rs
@@ -411,6 +411,7 @@ fn test_msrs() {
             #define KILL_ADDRESS 0x3000
 
             int g_kill_evt;
+            uint32_t g_msr2_count;
             struct kvm_msr_entry g_msr2;
 
             int setup_vm(struct crosvm *crosvm, void *mem) {
@@ -445,7 +446,7 @@ fn test_msrs() {
                 {
                     uint64_t dummy = 1;
                     g_msr2.index = MSR2_INDEX;
-                    crosvm_vcpu_get_msrs(vcpu, 1, &g_msr2);
+                    crosvm_vcpu_get_msrs(vcpu, 1, &g_msr2, &g_msr2_count);
                     write(g_kill_evt, &dummy, sizeof(dummy));
                     return 1;
                 }
@@ -458,6 +459,10 @@ fn test_msrs() {
                     fprintf(stderr, "msr1 has unexpected value: 0x%x\n", msr1_data);
                     return 1;
                 }
+                if (g_msr2_count != 1) {
+                    fprintf(stderr, "incorrect number of returned MSRSs: %d\n", g_msr2_count);
+                    return 1;
+                }
                 if (g_msr2.data != MSR2_DATA) {
                     fprintf(stderr, "msr2 has unexpected value: 0x%x\n", g_msr2.data);
                     return 1;