thin-provisioning-tools/unit-tests/bcache_t.cc
Ming-Hung Tsai 955e11bc28 [block-cache] fix potential file descriptor leak
Encapsulate file descriptor into an object, to ensure that an fd will be
closed properly while exception raised, e.g., the block_cache throws
exception during the block_manager's construction.
2020-02-23 15:41:16 +08:00

71 lines
2.0 KiB
C++

// Copyright (C) 2017 Red Hat, Inc. All rights reserved.
//
// This file is part of the thin-provisioning-tools source.
//
// thin-provisioning-tools is free software: you can redistribute it
// and/or modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation, either version 3 of
// the License, or (at your option) any later version.
//
// thin-provisioning-tools is distributed in the hope that it will be
// useful, but WITHOUT ANY WARRANTY; without even the implied warranty
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with thin-provisioning-tools. If not, see
// <http://www.gnu.org/licenses/>.
#include "gmock/gmock.h"
#include "block-cache/block_cache.h"
#include "test_utils.h"
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
using namespace std;
using namespace test;
using namespace testing;
//----------------------------------------------------------------
TEST(BCacheTests, cleaned_on_demand)
{
using namespace bcache;
unsigned const NR_BLOCKS = 16;
temp_file tmp("bcache_t", 1);
file_utils::file_descriptor fd(tmp.get_path().c_str(), O_RDWR | O_DIRECT);
uint64_t bs = 8;
block_cache bc(fd, bs, 64, (bs << SECTOR_SHIFT) * NR_BLOCKS);
validator::ptr v(new bcache::noop_validator());
for (unsigned i = 0; i < NR_BLOCKS; i++) {
block_cache::block &b = bc.get(i, block_cache::GF_DIRTY | block_cache::GF_ZERO, v);
b.put();
}
// Default WRITEBACK_LOW_THRESHOLD_PERCENT == 33
// For NR_BLOCKS cache entires, only 6 dirty blocks will be submitted,
// so the 7th get() doesn't work.
list<block_cache::block *> blocks;
try {
for (unsigned i = 0; i < NR_BLOCKS; i++)
blocks.push_back(&bc.get(NR_BLOCKS + i, 0, v));
} catch (...) {
for (auto b : blocks)
b->put();
throw;
}
for (auto b : blocks)
b->put();
}