test_common_vmcall.cpp
Go to the documentation of this file.
1 //
2 // Bareflank Hypervisor
3 //
4 // Copyright (C) 2015 Assured Information Security, Inc.
5 // Author: Rian Quinn <quinnr@ainfosec.com>
6 // Author: Brendan Kerrigan <kerriganb@ainfosec.com>
7 //
8 // This library is free software; you can redistribute it and/or
9 // modify it under the terms of the GNU Lesser General Public
10 // License as published by the Free Software Foundation; either
11 // version 2.1 of the License, or (at your option) any later version.
12 //
13 // This library is distributed in the hope that it will be useful,
14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 // Lesser General Public License for more details.
17 //
18 // You should have received a copy of the GNU Lesser General Public
19 // License along with this library; if not, write to the Free Software
20 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 
22 #include <gsl/gsl>
23 
24 #include <test.h>
25 
26 #include <common.h>
27 #include <platform.h>
28 
29 extern uint64_t g_set_afinity_fails;
30 
31 void
32 driver_entry_ut::test_common_vmcall_invalid_args()
33 {
34  vmcall_registers_t regs{};
35 
36  this->expect_true(common_add_module(m_dummy_start_vmm_success.get(), m_dummy_start_vmm_success_length) == BF_SUCCESS);
37  this->expect_true(common_add_module(m_dummy_stop_vmm_success.get(), m_dummy_stop_vmm_success_length) == BF_SUCCESS);
38  this->expect_true(common_add_module(m_dummy_add_md_success.get(), m_dummy_add_md_success_length) == BF_SUCCESS);
39  this->expect_true(common_add_module(m_dummy_misc.get(), m_dummy_misc_length) == BF_SUCCESS);
40  this->expect_true(common_load_vmm() == BF_SUCCESS);
41  this->expect_true(common_start_vmm() == BF_SUCCESS);
42 
43  this->expect_true(common_vmcall(nullptr, 0) == BF_ERROR_INVALID_ARG);
44  this->expect_true(common_vmcall(&regs, 10) == BF_ERROR_INVALID_ARG);
45 
46  this->expect_true(common_fini() == BF_SUCCESS);
47 }
48 
49 void
50 driver_entry_ut::test_common_vmcall_set_affinity_failure()
51 {
52  vmcall_registers_t regs{};
53 
54  this->expect_true(common_add_module(m_dummy_start_vmm_success.get(), m_dummy_start_vmm_success_length) == BF_SUCCESS);
55  this->expect_true(common_add_module(m_dummy_stop_vmm_success.get(), m_dummy_stop_vmm_success_length) == BF_SUCCESS);
56  this->expect_true(common_add_module(m_dummy_add_md_success.get(), m_dummy_add_md_success_length) == BF_SUCCESS);
57  this->expect_true(common_add_module(m_dummy_misc.get(), m_dummy_misc_length) == BF_SUCCESS);
58  this->expect_true(common_load_vmm() == BF_SUCCESS);
59  this->expect_true(common_start_vmm() == BF_SUCCESS);
60 
61  g_set_afinity_fails = 1;
62  auto ___ = gsl::finally([&] { g_set_afinity_fails = 0; });
63 
64  this->expect_true(common_vmcall(&regs, 0) == -1);
65 
66  g_set_afinity_fails = 0;
67 
68  this->expect_true(common_fini() == BF_SUCCESS);
69 }
70 
71 void
72 driver_entry_ut::test_common_vmcall_success()
73 {
74  vmcall_registers_t regs{};
75 
76  this->expect_true(common_add_module(m_dummy_start_vmm_success.get(), m_dummy_start_vmm_success_length) == BF_SUCCESS);
77  this->expect_true(common_add_module(m_dummy_stop_vmm_success.get(), m_dummy_stop_vmm_success_length) == BF_SUCCESS);
78  this->expect_true(common_add_module(m_dummy_add_md_success.get(), m_dummy_add_md_success_length) == BF_SUCCESS);
79  this->expect_true(common_add_module(m_dummy_misc.get(), m_dummy_misc_length) == BF_SUCCESS);
80  this->expect_true(common_load_vmm() == BF_SUCCESS);
81  this->expect_true(common_start_vmm() == BF_SUCCESS);
82 
83  this->expect_true(common_vmcall(&regs, 0) == BF_SUCCESS);
84  this->expect_true(common_vmcall(&regs, 0xFFFFFFFFFFFFFFFF) == BF_SUCCESS);
85 
86  this->expect_true(common_fini() == BF_SUCCESS);
87 }
88 
89 void
90 driver_entry_ut::test_common_vmcall_success_event()
91 {
92  vmcall_registers_t regs{};
93  regs.r00 = VMCALL_EVENT;
94 
95  this->expect_true(common_add_module(m_dummy_start_vmm_success.get(), m_dummy_start_vmm_success_length) == BF_SUCCESS);
96  this->expect_true(common_add_module(m_dummy_stop_vmm_success.get(), m_dummy_stop_vmm_success_length) == BF_SUCCESS);
97  this->expect_true(common_add_module(m_dummy_add_md_success.get(), m_dummy_add_md_success_length) == BF_SUCCESS);
98  this->expect_true(common_add_module(m_dummy_misc.get(), m_dummy_misc_length) == BF_SUCCESS);
99  this->expect_true(common_load_vmm() == BF_SUCCESS);
100  this->expect_true(common_start_vmm() == BF_SUCCESS);
101 
102  this->expect_true(common_vmcall(&regs, 0) == BF_SUCCESS);
103  this->expect_true(common_vmcall(&regs, 0xFFFFFFFFFFFFFFFF) == BF_SUCCESS);
104 
105  this->expect_true(common_fini() == BF_SUCCESS);
106 }
107 
108 void
109 driver_entry_ut::test_common_vmcall_vmcall_when_unloaded()
110 {
111  vmcall_registers_t regs{};
112 
113  this->expect_true(common_add_module(m_dummy_start_vmm_success.get(), m_dummy_start_vmm_success_length) == BF_SUCCESS);
114  this->expect_true(common_add_module(m_dummy_stop_vmm_success.get(), m_dummy_stop_vmm_success_length) == BF_SUCCESS);
115  this->expect_true(common_add_module(m_dummy_add_md_success.get(), m_dummy_add_md_success_length) == BF_SUCCESS);
116  this->expect_true(common_add_module(m_dummy_get_drr_success.get(), m_dummy_get_drr_success_length) == BF_SUCCESS);
117  this->expect_true(common_add_module(m_dummy_misc.get(), m_dummy_misc_length) == BF_SUCCESS);
118  this->expect_true(common_vmcall(&regs, 0) == BF_ERROR_VMM_INVALID_STATE);
119  this->expect_true(common_fini() == BF_SUCCESS);
120 }
121 
122 void
123 driver_entry_ut::test_common_vmcall_vmcall_when_corrupt()
124 {
125  vmcall_registers_t regs{};
126 
127  this->expect_true(common_add_module(m_dummy_start_vmm_success.get(), m_dummy_start_vmm_success_length) == BF_SUCCESS);
128  this->expect_true(common_add_module(m_dummy_stop_vmm_failure.get(), m_dummy_stop_vmm_failure_length) == BF_SUCCESS);
129  this->expect_true(common_add_module(m_dummy_add_md_success.get(), m_dummy_add_md_success_length) == BF_SUCCESS);
130  this->expect_true(common_add_module(m_dummy_get_drr_success.get(), m_dummy_get_drr_success_length) == BF_SUCCESS);
131  this->expect_true(common_add_module(m_dummy_misc.get(), m_dummy_misc_length) == BF_SUCCESS);
132  this->expect_true(common_load_vmm() == BF_SUCCESS);
133  this->expect_true(common_start_vmm() == BF_SUCCESS);
134  this->expect_true(common_stop_vmm() == ENTRY_ERROR_VMM_STOP_FAILED);
135  this->expect_true(common_vmcall(&regs, 0) == BF_ERROR_VMM_CORRUPTED);
136  this->expect_true(common_fini() == BF_ERROR_VMM_CORRUPTED);
137 
138  common_reset();
139 }
140 
141 void
142 driver_entry_ut::test_common_vmcall_vmcall_when_loaded()
143 {
144  vmcall_registers_t regs{};
145 
146  this->expect_true(common_add_module(m_dummy_start_vmm_success.get(), m_dummy_start_vmm_success_length) == BF_SUCCESS);
147  this->expect_true(common_add_module(m_dummy_stop_vmm_success.get(), m_dummy_stop_vmm_success_length) == BF_SUCCESS);
148  this->expect_true(common_add_module(m_dummy_add_md_success.get(), m_dummy_add_md_success_length) == BF_SUCCESS);
149  this->expect_true(common_add_module(m_dummy_get_drr_success.get(), m_dummy_get_drr_success_length) == BF_SUCCESS);
150  this->expect_true(common_add_module(m_dummy_misc.get(), m_dummy_misc_length) == BF_SUCCESS);
151  this->expect_true(common_load_vmm() == BF_SUCCESS);
152  this->expect_true(common_vmcall(&regs, 0) == BF_ERROR_VMM_INVALID_STATE);
153  this->expect_true(common_fini() == BF_SUCCESS);
154 }
BF_SUCCESS
#define BF_SUCCESS
Definition: error_codes.h:105
common_add_module
int64_t common_add_module(const char *file, uint64_t fsize)
Definition: common.c:305
vmcall_registers_t
Definition: vmcall_interface.h:303
common_load_vmm
int64_t common_load_vmm(void)
Definition: common.c:357
common_fini
int64_t common_fini(void)
Definition: common.c:273
BF_ERROR_VMM_INVALID_STATE
#define BF_ERROR_VMM_INVALID_STATE
Definition: error_codes.h:110
VMCALL_EVENT
Definition: vmcall_interface.h:185
common_vmcall
int64_t common_vmcall(struct vmcall_registers_t *regs, uint64_t cpuid)
Definition: common.c:610
vmcall_registers_t::r00
uintptr_t r00
Definition: vmcall_interface.h:305
common_stop_vmm
int64_t common_stop_vmm(void)
Definition: common.c:543
common_start_vmm
int64_t common_start_vmm(void)
Definition: common.c:490
ENTRY_ERROR_VMM_STOP_FAILED
#define ENTRY_ERROR_VMM_STOP_FAILED
Definition: error_codes.h:52
g_set_afinity_fails
uint64_t g_set_afinity_fails
Definition: platform.c:38
BF_ERROR_INVALID_ARG
#define BF_ERROR_INVALID_ARG
Definition: error_codes.h:106
BF_ERROR_VMM_CORRUPTED
#define BF_ERROR_VMM_CORRUPTED
Definition: error_codes.h:114
expect_true
#define expect_true(a)
Definition: exit_handler_intel_x64_unittests.h:38
common_reset
int64_t common_reset(void)
Definition: common.c:233
Generated on Fri Apr 28 2017 22:11:38 by   doxygen 1.8.14