VILLASnode/tests/unit/pool.cpp

/* Unit tests for memory pool.
 *
 * Author: Steffen Vogel <post@steffenvogel.de>
 * SPDX-FileCopyrightText: 2014-2023 Institute for Automation of Complex Power Systems, RWTH Aachen University
 * SPDX-License-Identifier: Apache-2.0
 */

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

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

#include <villas/log.hpp>
#include <villas/pool.hpp>
#include <villas/utils.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");
}