VILLASnode/tests/unit/pool.cpp

/** Unit tests for memory pool
 *
 * @author Steffen Vogel <post@steffenvogel.de>
 * @copyright 2014-2022, Institute for Automation of Complex Power Systems, EONERC
 * @license Apache 2.0
 *********************************************************************************/

#include <criterion/criterion.h>
#include <criterion/parameterized.h>

#include <signal.h>
#include <vector>

#include <villas/pool.hpp>
#include <villas/utils.hpp>
#include <villas/log.hpp>

using namespace villas;

using namespace villas::node;

extern void init_memory();

struct param {
	int thread_count;
	int pool_size;
	size_t block_size;
	struct memory::Type *mt;
};

ParameterizedTestParameters(pool, basic)
{
	static std::vector<struct param> params;

	params.clear();
	params.push_back({ 1, 4096,    150,  &memory::heap });
	params.push_back({ 1, 128,     8,    &memory::mmap });
	params.push_back({ 1, 4,       8192, &memory::mmap_hugetlb });
	params.push_back({ 1, 1 << 13, 4,    &memory::mmap_hugetlb });

	return cr_make_param_array(struct param, params.data(), params.size());
}

// cppcheck-suppress unknownMacro
ParameterizedTest(struct param *p, pool, basic, .init = init_memory)
{
	int ret;
	struct Pool pool;
	void *ptr, *ptrs[p->pool_size];

	logging.setLevel("trace");

	if (!utils::isPrivileged() && p->mt == &memory::mmap_hugetlb)
		cr_skip_test("Skipping memory_mmap_hugetlb tests allocatpr because we are running in an unprivileged environment.");

	ret = pool_init(&pool, p->pool_size, p->block_size, p->mt);
	cr_assert_eq(ret, 0, "Failed to create pool");

	ptr = pool_get(&pool);
	cr_assert_neq(ptr, nullptr);

	memset(ptr, 1, p->block_size); /* check that we dont get a seg fault */

	int i;
	for (i = 1; i < p->pool_size; i++) {
		ptrs[i] = pool_get(&pool);

		if (ptrs[i] == nullptr)
			break;
	}

	if (i < p->pool_size)
		cr_assert_neq(ptrs[i], nullptr);

	ptr = pool_get(&pool);
	cr_assert_eq(ptr, nullptr);

	ret = pool_destroy(&pool);
	cr_assert_eq(ret, 0, "Failed to destroy pool");
}