+++ /dev/null
-// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
-// vim: ts=8 sw=2 smarttab
-/*
- * Bitmap based in-memory allocator unit test cases.
- * Author: Ramesh Chander, Ramesh.Chander@sandisk.com
- */
-
-#include "include/Context.h"
-#include "os/bluestore/BitAllocator.h"
-#include <stdio.h>
-#include <assert.h>
-#include <math.h>
-#include <sstream>
-#include <gtest/gtest.h>
-
-
-//#define bmap_test_assert(x) ASSERT_EQ(true, (x))
-#define bmap_test_assert(x) assert((x))
-#define NUM_THREADS 16
-#define MAX_BLOCKS (1024 * 1024 * 1)
-
-TEST(BitAllocator, test_bmap_iter)
-{
- int num_items = 5;
- int off = 2;
-
- class BmapEntityTmp {
- int64_t m_num;
- int64_t m_len;
- public:
- void init(int index) {
- m_num = index;
- }
- BmapEntityTmp() {
-
- }
- BmapEntityTmp(int num) {
- m_num = num;
- m_len = num;
- }
-
- int64_t get_index() {
- return m_num;
- }
- bool is_allocated(int64_t s, int64_t num)
- {
- return true;
- }
- };
- BmapEntityTmp *obj = NULL;
- int i = 0;
- mempool::bluestore_alloc::vector<BmapEntityTmp> *arr = new mempool::bluestore_alloc::vector<BmapEntityTmp>(num_items);
- for (i = 0; i < num_items; i++) {
- (*arr)[i].init(i);
- }
- BitMapEntityIter<BmapEntityTmp> iter = BitMapEntityIter<BmapEntityTmp>(arr, off, false);
-
- i = off;
- int count = 0;
- int64_t last_idx = off;
- while ((obj = iter.next())) {
- bmap_test_assert(obj->get_index() == last_idx);
- bmap_test_assert(obj->get_index() == i);
- bmap_test_assert(obj == &(*arr)[i]);
- last_idx = iter.index();
- i++;
- count++;
- }
- bmap_test_assert(i == num_items);
- bmap_test_assert(count == num_items - off);
-
- iter = BitMapEntityIter<BmapEntityTmp>(arr, off, true);
-
- i = off;
- last_idx = off;
- count = 0;
- while ((obj = iter.next())) {
- bmap_test_assert(obj->get_index() == last_idx);
- bmap_test_assert(obj->get_index() == i);
- bmap_test_assert(obj == &(*arr)[i]);
- last_idx = iter.index();
-
- i = (i + 1) % num_items;
- count++;
- }
- bmap_test_assert(i == off + 1);
- bmap_test_assert(count == num_items + 1);
-
- delete arr;
-
- num_items = 4;
- off = num_items - 1;
-
- arr = new mempool::bluestore_alloc::vector<BmapEntityTmp>(num_items);
- for (i = 0; i < num_items; i++) {
- (*arr)[i].init(i);
- }
- iter = BitMapEntityIter<BmapEntityTmp>(arr, off, true);
- i = off;
- last_idx = off;
- count = 0;
- while ((obj = static_cast<BmapEntityTmp*>(iter.next()))) {
- bmap_test_assert(obj->get_index() == last_idx);
- bmap_test_assert(obj->get_index() == i);
- bmap_test_assert(obj == &(*arr)[i]);
- last_idx = iter.index();
- i = (i + 1) % num_items;
- count++;
- }
- bmap_test_assert(i == (off + 1)%num_items);
- bmap_test_assert(count == num_items + 1);
-
- delete arr;
-
- /*
- * BitMapArea Iter tests.
- */
- BitMapArea *area = nullptr;
- std::vector<BitMapArea*> children;
- children.reserve(num_items);
- for (i = 0; i < num_items; i++) {
- children.emplace_back(new BitMapAreaLeaf(
- g_ceph_context,
- BitMapArea::get_span_size(g_ceph_context), i, false));
- }
-
- off = 0;
- BitMapAreaList *area_list = \
- new BitMapAreaList(std::vector<BitMapArea*>(children));
- BmapEntityListIter area_iter = BmapEntityListIter(
- area_list, (int64_t) 0);
- i = off;
- last_idx = off;
- count = 0;
- while ((area = area_iter.next())) {
- bmap_test_assert(area->get_index() == last_idx);
- bmap_test_assert(area->get_index() == i);
- bmap_test_assert(area == children[i]);
- last_idx = area_iter.index();
- i = (i + 1) % num_items;
- count++;
- }
- bmap_test_assert(i == off);
- bmap_test_assert(count == num_items);
-
- off = 0;
- area_iter = BmapEntityListIter(area_list, off, true);
- i = off;
- last_idx = off;
- count = 0;
- while ((area = area_iter.next())) {
- bmap_test_assert(area->get_index() == last_idx);
- bmap_test_assert(area->get_index() == i);
- bmap_test_assert(area == children[i]);
- last_idx = area_iter.index();
- i = (i + 1) % num_items;
- count++;
- }
- bmap_test_assert(i == (off + 1)%num_items);
- bmap_test_assert(count == num_items + 1);
-
- for (i = 0; i < num_items; i++)
- delete children[i];
-
- delete area_list;
-}
-
-TEST(BitAllocator, test_bmap_entry)
-{
- int i = 0;
- int start = 0;
- int64_t scanned = 0;
- int64_t allocated = 0;
- int size = BmapEntry::size();
-
- BmapEntry *bmap = new BmapEntry(g_ceph_context, true);
-
- // Clear bits one by one and check they are cleared
- for (i = 0; i < size; i++) {
- bmap->clear_bit(i);
- bmap_test_assert(!bmap->check_bit(i));
- }
-
- // Set all bits again using set_bits
- bmap->set_bits(0, size);
-
- // clear 4 bits at a time and then check allocated
- for (i = 0; i < size/4; i++) {
- bmap->clear_bits(i * 4, 4);
- bmap_test_assert(!bmap->is_allocated(i * 4, 4));
- }
-
- // set all bits again
- bmap->set_bits(0, size);
-
- // clear alternate bits, check and set those bits
- for (i = 0; i < size/2; i++) {
- bmap->clear_bit(i * 2 + 1);
- bmap_test_assert(!bmap->check_bit(i * 2 + 1));
- bmap_test_assert(bmap->check_n_set_bit(i * 2 + 1));
- }
-
- // free 1, 2 and size bits at a time and try to find n cont bits
- for (i = 0; i < size / 4; i++) {
- bmap->clear_bits(i * 2 + 1, i + 1);
- bmap_test_assert(!bmap->check_bit(i * 2 + 1));
- bmap_test_assert(bmap->find_n_cont_bits(i * 2 + 1, i + 1) ==
- i + 1);
- }
-
- // free 1, 2 and size bits at a time and try to find any cont bits
- for (i = 0; i < size / 4; i++) {
- bmap->clear_bits(i * 2 + 1, i + 1);
- bmap_test_assert(!bmap->is_allocated(i * 2 + 1, i + 1));
- }
-
- for (i = 0; i < size / 4; i++) {
- bmap->clear_bits(i * 2 + 1, i + 1);
- allocated = bmap->find_first_set_bits(i + 1, 0, &start, &scanned);
-
- bmap_test_assert(allocated == i + 1);
- bmap_test_assert(scanned == ((i * 2 + 1) + (i + 1)));
- bmap_test_assert(start == i * 2 + 1);
- bmap->set_bits(0, BmapEntry::size());
-
- }
-
-
-
- // Find few bits at end of bitmap and find those
- bmap->clear_bits(0, 4);
- bmap->clear_bits(BmapEntry::size() - 12, 5);
- bmap->clear_bits(BmapEntry::size() - 6, 6);
- allocated = bmap->find_first_set_bits(6, 0, &start, &scanned);
-
- bmap_test_assert(allocated == 6);
- bmap_test_assert(scanned == BmapEntry::size() - 6 + 6);
- bmap_test_assert(start == BmapEntry::size() - 6);
- bmap_test_assert(bmap->is_allocated(start, 6));
-
- delete bmap;
-
- {
-
- bmap = new BmapEntry(g_ceph_context, false);
- start = -1;
- scanned = 0;
- allocated = 0;
- allocated = bmap->find_first_set_bits(1, 1, &start, &scanned);
- bmap_test_assert(allocated == 1);
- bmap_test_assert(start == 1);
-
- allocated = bmap->find_first_set_bits(1, BmapEntry::size() - 2, &start, &scanned);
- bmap_test_assert(allocated == 1);
- bmap_test_assert(start == BmapEntry::size() - 2);
-
- bmap->clear_bits(0, BmapEntry::size());
- bmap->set_bits(0, BmapEntry::size() / 4);
- allocated = bmap->find_first_set_bits(4, 2, &start, &scanned);
- bmap_test_assert(allocated == 4);
- bmap_test_assert(start == BmapEntry::size() / 4);
- delete bmap;
- }
-
- bmap = new BmapEntry(g_ceph_context, false);
- bmap->set_bits(4, BmapEntry::size() - 4);
- bmap_test_assert(bmap->is_allocated(4, BmapEntry::size() - 4));
- bmap_test_assert(!bmap->is_allocated(0, 4));
- bmap->set_bits(0, 4);
- bmap_test_assert(bmap->is_allocated(0, BmapEntry::size()));
- delete bmap;
-
-}
-
-TEST(BitAllocator, test_zone_alloc)
-{
- int total_blocks = 1024;
- int64_t allocated = 0;
-
- BitMapZone *zone = new BitMapZone(g_ceph_context, total_blocks, 0);
-
- // Allocate all blocks and see that it is allocating in order.
- bool lock = zone->lock_excl_try();
- bmap_test_assert(lock);
-
- int64_t blk_size = 1024;
- AllocExtentVector extents;
- ExtentList *block_list = new ExtentList(&extents, blk_size);
- allocated = zone->alloc_blocks_dis(zone->size() / 2, 1, 0, 0, block_list);
- bmap_test_assert(allocated == zone->size() / 2);
-
-
- {
- int64_t blk_size = 1024;
- AllocExtentVector extents;
- ExtentList *block_list = new ExtentList(&extents, blk_size);
-
- zone = new BitMapZone(g_ceph_context, total_blocks, 0);
- lock = zone->lock_excl_try();
- bmap_test_assert(lock);
- for (int i = 0; i < zone->size(); i += 4) {
- block_list->reset();
- allocated = zone->alloc_blocks_dis(1, 1, i, 0, block_list);
- bmap_test_assert(allocated == 1);
- EXPECT_EQ(extents[0].offset, (uint64_t) i * blk_size);
- }
-
- for (int i = 0; i < zone->size(); i += 4) {
- zone->free_blocks(i, 1);
- }
- }
-
- /*
- * Min alloc size cases.
- */
- {
- int64_t blk_size = 1;
- AllocExtentVector extents;
-
- for (int i = 1; i <= total_blocks - BmapEntry::size(); i = i << 1) {
- for (int64_t j = 0; j <= BmapEntry::size(); j = 1 << j) {
- extents.clear();
- ExtentList *block_list = new ExtentList(&extents, blk_size);
- zone = new BitMapZone(g_ceph_context, total_blocks, 0);
- lock = zone->lock_excl_try();
- bmap_test_assert(lock);
-
- block_list->reset();
- int64_t need_blks = (((total_blocks - j) / i) * i);
- allocated = zone->alloc_blocks_dis(need_blks, i, j, 0, block_list);
- bmap_test_assert(allocated == need_blks);
- bmap_test_assert(extents[0].offset == (uint64_t) j);
- delete block_list;
- delete zone;
- }
- }
-
- //allocation in loop
- {
- extents.clear();
- ExtentList *block_list = new ExtentList(&extents, blk_size);
- zone = new BitMapZone(g_ceph_context, total_blocks, 0);
- lock = zone->lock_excl_try();
-
- for (int iter = 1; iter < 5; iter++) {
- for (int i = 1; i <= total_blocks; i = i << 1) {
- for (int j = 0; j < total_blocks; j +=i) {
- bmap_test_assert(lock);
- block_list->reset();
- int64_t need_blks = i;
- allocated = zone->alloc_blocks_dis(need_blks, i, 0, 0, block_list);
- bmap_test_assert(allocated == need_blks);
- bmap_test_assert(extents[0].offset == (uint64_t) j);
- block_list->reset();
- }
- {
- allocated = zone->alloc_blocks_dis(1, 1, 0, 0, block_list);
- bmap_test_assert(allocated == 0);
- block_list->reset();
- }
-
- for (int j = 0; j < total_blocks; j +=i) {
- zone->free_blocks(j, i);
- }
- }
- }
- delete block_list;
- delete zone;
- }
-
- {
- extents.clear();
- ExtentList *block_list = new ExtentList(&extents, blk_size);
- zone = new BitMapZone(g_ceph_context, total_blocks, 0);
- lock = zone->lock_excl_try();
- bmap_test_assert(lock);
-
- block_list->reset();
- allocated = zone->alloc_blocks_dis(total_blocks + 1, total_blocks + 1, 0, 1024, block_list);
- bmap_test_assert(allocated == 0);
-
- block_list->reset();
- allocated = zone->alloc_blocks_dis(total_blocks, total_blocks, 1, 1024, block_list);
- bmap_test_assert(allocated == 0);
-
- block_list->reset();
- allocated = zone->alloc_blocks_dis(total_blocks, total_blocks, 0, 0, block_list);
- bmap_test_assert(allocated == total_blocks);
- bmap_test_assert(extents[0].offset == 0);
-
- zone->free_blocks(extents[0].offset, allocated);
-
- delete block_list;
- extents.clear();
- block_list = new ExtentList(&extents, blk_size, total_blocks / 4 * blk_size);
- allocated = zone->alloc_blocks_dis(total_blocks, total_blocks / 4, 0, 0, block_list);
- bmap_test_assert(allocated == total_blocks);
- for (int i = 0; i < 4; i++) {
- bmap_test_assert(extents[i].offset == (uint64_t) i * (total_blocks / 4));
- }
- }
- }
-}
-
-TEST(BitAllocator, test_bmap_alloc)
-{
- const int max_iter = 3;
-
- for (int round = 0; round < 3; round++) {
- // Test zone of different sizes: 512, 1024, 2048
- int64_t zone_size = 512ull << round;
- ostringstream val;
- val << zone_size;
- g_conf->set_val("bluestore_bitmapallocator_blocks_per_zone", val.str());
-
- // choose randomized span_size
- int64_t span_size = 512ull << (rand() % 4);
- val.str("");
- val << span_size;
- g_conf->set_val("bluestore_bitmapallocator_span_size", val.str());
- g_ceph_context->_conf->apply_changes(NULL);
-
- int64_t total_blocks = zone_size * 4;
- int64_t allocated = 0;
-
- BitAllocator *alloc = new BitAllocator(g_ceph_context, total_blocks,
- zone_size, CONCURRENT);
- int64_t alloc_size = 2;
- for (int64_t iter = 0; iter < max_iter; iter++) {
- for (int64_t j = 0; alloc_size <= total_blocks; j++) {
- int64_t blk_size = 1024;
- AllocExtentVector extents;
- ExtentList *block_list = new ExtentList(&extents, blk_size, alloc_size);
- for (int64_t i = 0; i < total_blocks; i += alloc_size) {
- bmap_test_assert(alloc->reserve_blocks(alloc_size) == true);
- allocated = alloc->alloc_blocks_dis_res(alloc_size, MIN(alloc_size, zone_size),
- 0, block_list);
- bmap_test_assert(alloc_size == allocated);
- bmap_test_assert(block_list->get_extent_count() ==
- (alloc_size > zone_size? alloc_size / zone_size: 1));
- bmap_test_assert(extents[0].offset == (uint64_t) i * blk_size);
- bmap_test_assert((int64_t) extents[0].length ==
- ((alloc_size > zone_size? zone_size: alloc_size) * blk_size));
- block_list->reset();
- }
- for (int64_t i = 0; i < total_blocks; i += alloc_size) {
- alloc->free_blocks(i, alloc_size);
- }
- alloc_size = 2 << j;
- }
- }
-
- int64_t blk_size = 1024;
- AllocExtentVector extents;
-
- ExtentList *block_list = new ExtentList(&extents, blk_size);
-
- ASSERT_EQ(alloc->reserve_blocks(alloc->size() / 2), true);
- allocated = alloc->alloc_blocks_dis_res(alloc->size()/2, 1, 0, block_list);
- ASSERT_EQ(alloc->size()/2, allocated);
-
- block_list->reset();
- ASSERT_EQ(alloc->reserve_blocks(1), true);
- allocated = alloc->alloc_blocks_dis_res(1, 1, 0, block_list);
- bmap_test_assert(allocated == 1);
-
- alloc->free_blocks(alloc->size()/2, 1);
-
- block_list->reset();
- ASSERT_EQ(alloc->reserve_blocks(1), true);
- allocated = alloc->alloc_blocks_dis_res(1, 1, 0, block_list);
- bmap_test_assert(allocated == 1);
-
- bmap_test_assert((int64_t) extents[0].offset == alloc->size()/2 * blk_size);
-
- delete block_list;
- delete alloc;
-
- }
-
- // restore to typical value
- g_conf->set_val("bluestore_bitmapallocator_blocks_per_zone", "1024");
- g_conf->set_val("bluestore_bitmapallocator_span_size", "1024");
- g_ceph_context->_conf->apply_changes(NULL);
-}
-
-bool alloc_extents_max_block(BitAllocator *alloc,
- int64_t max_alloc,
- int64_t total_alloc)
-{
- int64_t blk_size = 1;
- int64_t allocated = 0;
- int64_t verified = 0;
- int64_t count = 0;
- AllocExtentVector extents;
-
- ExtentList *block_list = new ExtentList(&extents, blk_size, max_alloc);
-
- EXPECT_EQ(alloc->reserve_blocks(total_alloc), true);
- allocated = alloc->alloc_blocks_dis_res(total_alloc, blk_size, 0, block_list);
- EXPECT_EQ(allocated, total_alloc);
-
- max_alloc = total_alloc > max_alloc? max_alloc: total_alloc;
-
- for (auto &p: extents) {
- count++;
- EXPECT_EQ(p.length, max_alloc);
- verified += p.length;
- if (verified >= total_alloc) {
- break;
- }
- }
-
- EXPECT_EQ(total_alloc / max_alloc, count);
- return true;
-}
-
-TEST(BitAllocator, test_bmap_alloc2)
-{
- int64_t total_blocks = 1024 * 4;
- int64_t zone_size = 1024;
- BitAllocator *alloc = new BitAllocator(g_ceph_context, total_blocks,
- zone_size, CONCURRENT);
-
- alloc_extents_max_block(alloc, 1, 16);
- alloc_extents_max_block(alloc, 4, 16);
- alloc_extents_max_block(alloc, 16, 16);
- alloc_extents_max_block(alloc, 32, 16);
-}
-
-__thread int my_tid;
-
-void
-do_work_dis(BitAllocator *alloc)
-{
- int num_iters = 10;
- int64_t alloced = 0;
- int64_t num_blocks = alloc->size() / NUM_THREADS;
-
- AllocExtentVector extents;
- ExtentList *block_list = new ExtentList(&extents, 4096);
-
- while (num_iters--) {
- alloc_assert(alloc->reserve_blocks(num_blocks));
- alloced = alloc->alloc_blocks_dis_res(num_blocks, 1, 0, block_list);
- alloc_assert(alloced == num_blocks);
-
- alloc_assert(alloc->is_allocated_dis(block_list, num_blocks));
- alloc->free_blocks_dis(num_blocks, block_list);
- block_list->reset();
- }
-}
-
-int tid = 0;
-static bool cont = true;
-
-void *
-worker(void *args)
-{
- my_tid = __sync_fetch_and_add(&tid, 1);
- BitAllocator *alloc = (BitAllocator *) args;
- printf("Starting thread %d", my_tid);
- do_work_dis(alloc);
-
- return NULL;
-}
-
-TEST(BitAllocator, test_bmap_alloc_concurrent)
-{
- int64_t total_blocks = MAX_BLOCKS;
- int64_t zone_size = 1024;
- pthread_t pthreads[NUM_THREADS] = {0};
-
- bmap_test_assert(total_blocks <= MAX_BLOCKS);
-
- BitAllocator *alloc = new BitAllocator(g_ceph_context, total_blocks,
- zone_size, CONCURRENT);
-
- for (int k = 0; k < 2; k++) {
- cont = k;
- printf("Spawning %d threads for parallel test. Mode Cont = %d.....\n", NUM_THREADS, cont);
- for (int j = 0; j < NUM_THREADS; j++) {
- if (pthread_create(&pthreads[j], NULL, worker, alloc)) {
- printf("Unable to create worker thread.\n");
- exit(0);
- }
- }
-
- for (int j = 0; j < NUM_THREADS; j++) {
- pthread_join(pthreads[j], NULL);
- }
- }
-}