cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

get_msr_index_features.c (3125B)

      1// SPDX-License-Identifier: GPL-2.0
      2/*
      3 * Test that KVM_GET_MSR_INDEX_LIST and
      4 * KVM_GET_MSR_FEATURE_INDEX_LIST work as intended
      5 *
      6 * Copyright (C) 2020, Red Hat, Inc.
      7 */
      8#include <fcntl.h>
      9#include <stdio.h>
     10#include <stdlib.h>
     11#include <string.h>
     12#include <sys/ioctl.h>
     13
     14#include "test_util.h"
     15#include "kvm_util.h"
     16#include "processor.h"
     17
     18static int kvm_num_index_msrs(int kvm_fd, int nmsrs)
     19{
     20	struct kvm_msr_list *list;
     21	int r;
     22
     23	list = malloc(sizeof(*list) + nmsrs * sizeof(list->indices[0]));
     24	list->nmsrs = nmsrs;
     25	r = ioctl(kvm_fd, KVM_GET_MSR_INDEX_LIST, list);
     26	TEST_ASSERT(r == -1 && errno == E2BIG,
     27				"Unexpected result from KVM_GET_MSR_INDEX_LIST probe, r: %i",
     28				r);
     29
     30	r = list->nmsrs;
     31	free(list);
     32	return r;
     33}
     34
     35static void test_get_msr_index(void)
     36{
     37	int old_res, res, kvm_fd, r;
     38	struct kvm_msr_list *list;
     39
     40	kvm_fd = open_kvm_dev_path_or_exit();
     41
     42	old_res = kvm_num_index_msrs(kvm_fd, 0);
     43	TEST_ASSERT(old_res != 0, "Expecting nmsrs to be > 0");
     44
     45	if (old_res != 1) {
     46		res = kvm_num_index_msrs(kvm_fd, 1);
     47		TEST_ASSERT(res > 1, "Expecting nmsrs to be > 1");
     48		TEST_ASSERT(res == old_res, "Expecting nmsrs to be identical");
     49	}
     50
     51	list = malloc(sizeof(*list) + old_res * sizeof(list->indices[0]));
     52	list->nmsrs = old_res;
     53	r = ioctl(kvm_fd, KVM_GET_MSR_INDEX_LIST, list);
     54
     55	TEST_ASSERT(r == 0,
     56		    "Unexpected result from KVM_GET_MSR_FEATURE_INDEX_LIST, r: %i",
     57		    r);
     58	TEST_ASSERT(list->nmsrs == old_res, "Expecting nmsrs to be identical");
     59	free(list);
     60
     61	close(kvm_fd);
     62}
     63
     64static int kvm_num_feature_msrs(int kvm_fd, int nmsrs)
     65{
     66	struct kvm_msr_list *list;
     67	int r;
     68
     69	list = malloc(sizeof(*list) + nmsrs * sizeof(list->indices[0]));
     70	list->nmsrs = nmsrs;
     71	r = ioctl(kvm_fd, KVM_GET_MSR_FEATURE_INDEX_LIST, list);
     72	TEST_ASSERT(r == -1 && errno == E2BIG,
     73		"Unexpected result from KVM_GET_MSR_FEATURE_INDEX_LIST probe, r: %i",
     74				r);
     75
     76	r = list->nmsrs;
     77	free(list);
     78	return r;
     79}
     80
     81struct kvm_msr_list *kvm_get_msr_feature_list(int kvm_fd, int nmsrs)
     82{
     83	struct kvm_msr_list *list;
     84	int r;
     85
     86	list = malloc(sizeof(*list) + nmsrs * sizeof(list->indices[0]));
     87	list->nmsrs = nmsrs;
     88	r = ioctl(kvm_fd, KVM_GET_MSR_FEATURE_INDEX_LIST, list);
     89
     90	TEST_ASSERT(r == 0,
     91		"Unexpected result from KVM_GET_MSR_FEATURE_INDEX_LIST, r: %i",
     92		r);
     93
     94	return list;
     95}
     96
     97static void test_get_msr_feature(void)
     98{
     99	int res, old_res, i, kvm_fd;
    100	struct kvm_msr_list *feature_list;
    101
    102	kvm_fd = open_kvm_dev_path_or_exit();
    103
    104	old_res = kvm_num_feature_msrs(kvm_fd, 0);
    105	TEST_ASSERT(old_res != 0, "Expecting nmsrs to be > 0");
    106
    107	if (old_res != 1) {
    108		res = kvm_num_feature_msrs(kvm_fd, 1);
    109		TEST_ASSERT(res > 1, "Expecting nmsrs to be > 1");
    110		TEST_ASSERT(res == old_res, "Expecting nmsrs to be identical");
    111	}
    112
    113	feature_list = kvm_get_msr_feature_list(kvm_fd, old_res);
    114	TEST_ASSERT(old_res == feature_list->nmsrs,
    115				"Unmatching number of msr indexes");
    116
    117	for (i = 0; i < feature_list->nmsrs; i++)
    118		kvm_get_feature_msr(feature_list->indices[i]);
    119
    120	free(feature_list);
    121	close(kvm_fd);
    122}
    123
    124int main(int argc, char *argv[])
    125{
    126	if (kvm_check_cap(KVM_CAP_GET_MSR_FEATURES))
    127		test_get_msr_feature();
    128
    129	test_get_msr_index();
    130}