static long
ioctl_unload_vmm(void)
{
int i;
int64_t ret;
long status = BF_IOCTL_SUCCESS;
ret = common_unload_vmm();
if (ret != BF_SUCCESS)
{
ALERT("IOCTL_UNLOAD_VMM: common_unload_vmm failed: %p - %s\n", \
(void *)ret, ec_to_str(ret));
status = BF_IOCTL_FAILURE;
}
for (i = 0; i < g_num_files; i++)
{
vfree(files[i]);
files[i] = 0;
files_size[i] = 0;
}
g_num_files = 0;
if (status == BF_IOCTL_SUCCESS)
DEBUG("IOCTL_UNLOAD_VMM: succeeded\n");
return status;
}
static int64_t
ioctl_unload_vmm(void)
{
int64_t i;
int64_t ret;
int64_t status = BF_IOCTL_SUCCESS;
ret = common_unload_vmm();
if (ret != BF_SUCCESS)
{
ALERT("IOCTL_UNLOAD_VMM: common_unload_vmm failed: %p - %s\n",
(void *)ret, ec_to_str(ret));
status = BF_IOCTL_FAILURE;
}
for (i = 0; i < g_num_pmodules; i++)
platform_free_rwe(pmodules[i].data, pmodules[i].size);
g_num_pmodules = 0;
platform_memset(&pmodules, 0, sizeof(pmodules));
if (status == BF_IOCTL_SUCCESS)
DEBUG("IOCTL_UNLOAD_VMM: succeeded\n");
return status;
}
void
driver_entry_ut::test_common_dump_resolve_symbol_failed()
{
EXPECT_TRUE(common_add_module(m_dummy1, m_dummy1_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy2, m_dummy2_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy3, m_dummy3_length) == BF_SUCCESS);
EXPECT_TRUE(common_load_vmm() == BF_SUCCESS);
// We need to wrap the mocks variable in a temporary stack frame as the
// current implementation of HippoMocks does not have a function to undo
// it's hooks into c-style functions as the functionality to do this is
// in the destructor.
{
MockRepository mocks;
mocks.OnCallFunc(resolve_symbol).Return(-1);
RUN_UNITTEST_WITH_MOCKS(mocks, [&]
{
EXPECT_TRUE(common_dump_vmm() == BF_ERROR_FAILED_TO_DUMP_DR);
});
}
EXPECT_TRUE(common_unload_vmm() == BF_SUCCESS);
}
void
driver_entry_ut::test_common_load_success()
{
EXPECT_TRUE(common_add_module(m_dummy1, m_dummy1_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy2, m_dummy2_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy3, m_dummy3_length) == BF_SUCCESS);
EXPECT_TRUE(common_load_vmm() == BF_SUCCESS);
EXPECT_TRUE(common_unload_vmm() == BF_SUCCESS);
}
void
driver_entry_ut::test_common_unload_unload_when_running()
{
EXPECT_TRUE(common_add_module(m_dummy_start_vmm_success, m_dummy_start_vmm_success_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy_stop_vmm_success, m_dummy_stop_vmm_success_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy_add_md_success, m_dummy_add_md_success_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy_misc, m_dummy_misc_length) == BF_SUCCESS);
EXPECT_TRUE(common_load_vmm() == BF_SUCCESS);
EXPECT_TRUE(common_start_vmm() == BF_SUCCESS);
EXPECT_TRUE(common_unload_vmm() == BF_ERROR_VMM_INVALID_STATE);
EXPECT_TRUE(common_fini() == BF_SUCCESS);
}
void
driver_entry_ut::test_common_unload_unload_when_corrupt()
{
EXPECT_TRUE(common_add_module(m_dummy_start_vmm_success, m_dummy_start_vmm_success_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy_stop_vmm_failure, m_dummy_stop_vmm_failure_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy_add_md_success, m_dummy_add_md_success_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy_misc, m_dummy_misc_length) == BF_SUCCESS);
EXPECT_TRUE(common_load_vmm() == BF_SUCCESS);
EXPECT_TRUE(common_start_vmm() == BF_SUCCESS);
EXPECT_TRUE(common_fini() == BF_ERROR_VMM_CORRUPTED);
EXPECT_TRUE(common_vmm_status() == VMM_CORRUPT);
EXPECT_TRUE(common_unload_vmm() == BF_ERROR_VMM_CORRUPTED);
common_reset();
}
void
driver_entry_ut::test_common_load_loader_add_file_failed()
{
MockRepository mocks;
mocks.OnCallFunc(bfelf_loader_add).Return(-1);
RUN_UNITTEST_WITH_MOCKS(mocks, [&]
{
EXPECT_TRUE(common_add_module(m_dummy1, m_dummy1_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy2, m_dummy2_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy3, m_dummy3_length) == BF_SUCCESS);
EXPECT_TRUE(common_load_vmm() == -1);
EXPECT_TRUE(common_unload_vmm() == BF_SUCCESS);
});
}
void
driver_entry_ut::test_common_dump_platform_alloc_failed()
{
MockRepository mocks;
mocks.OnCallFunc(platform_alloc).Return(0);
RUN_UNITTEST_WITH_MOCKS(mocks, [&]
{
EXPECT_TRUE(common_add_module(m_dummy1, m_dummy1_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy2, m_dummy2_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy3, m_dummy3_length) == BF_SUCCESS);
EXPECT_TRUE(common_load_vmm() == BF_SUCCESS);
EXPECT_TRUE(common_dump_vmm() == BF_ERROR_OUT_OF_MEMORY);
EXPECT_TRUE(common_unload_vmm() == BF_SUCCESS);
});
}
void
driver_entry_ut::test_common_dump_debug_ring_read_failed()
{
MockRepository mocks;
mocks.OnCallFunc(debug_ring_read).Return(-1);
RUN_UNITTEST_WITH_MOCKS(mocks, [&]
{
EXPECT_TRUE(common_add_module(m_dummy1, m_dummy1_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy2, m_dummy2_length) == BF_SUCCESS);
EXPECT_TRUE(common_add_module(m_dummy3, m_dummy3_length) == BF_SUCCESS);
EXPECT_TRUE(common_load_vmm() == BF_SUCCESS);
EXPECT_TRUE(common_dump_vmm() == BF_ERROR_FAILED_TO_DUMP_DR);
EXPECT_TRUE(common_unload_vmm() == BF_SUCCESS);
});
}
static long
ioctl_unload_vmm(void)
{
int i;
int64_t ret;
long status = BF_IOCTL_SUCCESS;
ret = common_unload_vmm();
if (ret != BF_SUCCESS)
{
ALERT("IOCTL_UNLOAD_VMM: failed to unload vmm: %lld\n", ret);
status = BF_IOCTL_FAILURE;
}
for (i = 0; i < g_num_files; i++)
platform_free(files[i], files_sizes[i]);
g_num_files = 0;
if (status == BF_IOCTL_SUCCESS)
DEBUG("IOCTL_UNLOAD_VMM: succeeded\n");
return status;
}
void
driver_entry_ut::test_common_unload_unload_when_already_unloaded()
{
EXPECT_TRUE(common_unload_vmm() == BF_SUCCESS);
}
