crosvm/tests/plugin_vm_state.c
Dmitry Torokhov 3e40b51a62 plugin: allow retrieving and setting VM and VCPU states
This change allows plugin to retrieve and set various VM and VCPU states:
interrupt controller, PIT, LAPIC and MP state.

BUG=b:76083711
TEST=cargo test -p kvm

Change-Id: Ie32a67b0cd4a1f0a19ccd826a6e1c9dc25670f95
Signed-off-by: Dmitry Torokhov <dtor@chromium.org>
Reviewed-on: https://chromium-review.googlesource.com/986511
Reviewed-by: Zach Reizner <zachr@chromium.org>
2018-03-30 00:07:07 -07:00

120 lines
3.2 KiB
C

/*
* Copyright 2018 The Chromium OS Authors. All rights reserved.
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "crosvm.h"
int main(int argc, char** argv) {
struct crosvm *crosvm;
int ret = crosvm_connect(&crosvm);
if (ret) {
fprintf(stderr, "failed to connect to crosvm: %d\n", ret);
return 1;
}
struct kvm_pic_state pic_state;
ret = crosvm_get_pic_state(crosvm, false, &pic_state);
if (ret < 0) {
fprintf(stderr, "failed to get initial PIC1 state: %d\n", ret);
return 1;
}
if (pic_state.auto_eoi) {
fprintf(stderr, "unexpected value of auto_eoi flag\n");
return 1;
}
pic_state.auto_eoi = true;
ret = crosvm_set_pic_state(crosvm, false, &pic_state);
if (ret < 0) {
fprintf(stderr, "failed to update PIC1 state: %d\n", ret);
return 1;
}
ret = crosvm_get_pic_state(crosvm, false, &pic_state);
if (ret < 0) {
fprintf(stderr, "failed to get updated PIC1 state: %d\n", ret);
return 1;
}
if (!pic_state.auto_eoi) {
fprintf(stderr, "unexpected value of auto_eoi flag after update\n");
return 1;
}
// Test retrieving and setting IOAPIC state.
struct kvm_ioapic_state ioapic_state;
ret = crosvm_get_ioapic_state(crosvm, &ioapic_state);
if (ret < 0) {
fprintf(stderr, "failed to get initial PIC1 state: %d\n", ret);
return 1;
}
fprintf(stderr, "IOAPIC ID: %d\n", ioapic_state.id);
if (ioapic_state.id != 0) {
fprintf(stderr, "unexpected value of IOAPIC ID: %d\n", ioapic_state.id);
return 1;
}
ioapic_state.id = 1;
ret = crosvm_set_ioapic_state(crosvm, &ioapic_state);
if (ret < 0) {
fprintf(stderr, "failed to update PIC1 state: %d\n", ret);
return 1;
}
ret = crosvm_get_ioapic_state(crosvm, &ioapic_state);
if (ret < 0) {
fprintf(stderr, "failed to get updated PIC1 state: %d\n", ret);
return 1;
}
if (ioapic_state.id != 1) {
fprintf(stderr, "unexpected value of IOAPIC ID after update: %d\n",
ioapic_state.id);
return 1;
}
// Test retrieving and setting PIT state.
struct kvm_pit_state2 pit_state;
ret = crosvm_get_pit_state(crosvm, &pit_state);
if (ret < 0) {
fprintf(stderr, "failed to get initial PIT state: %d\n", ret);
return 1;
}
if (pit_state.flags & KVM_PIT_FLAGS_HPET_LEGACY) {
fprintf(stderr, "unexpected value of KVM_PIT_FLAGS_HPET_LEGACY flag\n");
return 1;
}
pit_state.flags |= KVM_PIT_FLAGS_HPET_LEGACY;
ret = crosvm_set_pit_state(crosvm, &pit_state);
if (ret < 0) {
fprintf(stderr, "failed to update PIT state: %d\n", ret);
return 1;
}
ret = crosvm_get_pit_state(crosvm, &pit_state);
if (ret < 0) {
fprintf(stderr, "failed to get updated PIT state: %d\n", ret);
return 1;
}
if (!(pit_state.flags & KVM_PIT_FLAGS_HPET_LEGACY)) {
fprintf(stderr,
"unexpected value of KVM_PIT_FLAGS_HPET_LEGACY after update\n");
return 1;
}
return 0;
}