emo/thin-provisioning-tools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
Joe Thornber 2014-07-29 11:34:26 +01:00
parent d482a76bda
commit f06a2673c5
12 changed files with 122 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
block-cache/block_cache.cc
View File

@ -69,7 +69,7 @@ namespace bcache {
if (!blocks)
return -ENOMEM;
blocks_memory_.reset(reinterpret_cast<unsigned char *>(blocks));
blocks_memory_ = blocks;
/* Allocate the data for each block. We page align the data. */
data = alloc_aligned(count * block_size, PAGE_SIZE);
@ -78,7 +78,7 @@ namespace bcache {
return -ENOMEM;
}
blocks_data_.reset(reinterpret_cast<unsigned char *>(data));
blocks_data_ = data;
for (i = 0; i < count; i++) {
block *b = new (blocks + i) block();
@ -145,6 +145,7 @@ namespace bcache {
* |b->list| should be valid (either pointing to itself, on one of the other
* lists.
*/
// FIXME: add batch issue
int
block_cache::issue_low_level(block &b, enum io_iocb_cmd opcode, const char *desc)
{
@ -162,7 +163,6 @@ namespace bcache {
r = io_submit(aio_context_, 1, control_blocks);
if (r != 1) {
if (r < 0) {
perror("io_submit error");
info("io_submit failed with %s op: %d\n", desc, r);
} else
info("could not submit IOs, with %s op\n", desc);
@ -212,9 +212,15 @@ namespace bcache {
else if (e.res < 0)
complete_io(*b, e.res);
else
info("incomplete io for block %llu, unexpected: %d\n",
b->index_, e.res);
else {
std::cerr << "incomplete io for block " << b->index_
<< ", e.res = " << e.res
<< ", e.res2 = " << e.res2
<< ", offset = " << b->control_block_.u.c.offset
<< ", nbytes = " << b->control_block_.u.c.nbytes
<< "\n";
exit(1);
}
}
}
@ -422,20 +428,14 @@ namespace bcache {
}
block_cache::block_cache(int fd, sector_t block_size, uint64_t on_disk_blocks, size_t mem)
: nr_dirty_(0),
: nr_locked_(0),
nr_dirty_(0),
nr_io_pending_(0)
{
int r;
unsigned nr_cache_blocks = calc_nr_cache_blocks(mem, block_size);
unsigned nr_buckets = calc_nr_buckets(nr_cache_blocks);
info("block_size = %llu, on_disk_blocks = %llu, mem = %llu, nr_cache_blocks = %llu\n",
(unsigned long long) block_size,
(unsigned long long) on_disk_blocks,
(unsigned long long) mem,
(unsigned long long) nr_cache_blocks);
buckets_.resize(nr_buckets);
fd_ = fd;
@ -448,7 +448,6 @@ namespace bcache {
aio_context_ = 0; /* needed or io_setup will fail */
r = io_setup(nr_cache_blocks, &aio_context_);
if (r < 0) {
std::cerr << "r = " << r << "\n";
perror("io_setup failed");
throw std::runtime_error("io_setup failed");
}
@ -467,11 +466,20 @@ namespace bcache {
block_cache::~block_cache()
{
assert(!nr_locked_);
flush();
wait_all();
// FIXME: use unique_ptrs
if (blocks_memory_)
free(blocks_memory_);
if (blocks_data_)
free(blocks_data_);
if (aio_context_)
io_destroy(aio_context_);
::close(fd_);
}
uint64_t
@ -533,10 +541,14 @@ namespace bcache {
block *b = lookup_or_read_block(index, flags, v);
if (b) {
if (b->ref_count_)
throw std::runtime_error("block already locked");
if (b->ref_count_ && flags & (GF_DIRTY | GF_ZERO))
throw std::runtime_error("attempt to write lock block concurrently");
hit(*b);
if (!b->ref_count_)
nr_locked_++;
b->ref_count_++;
if (flags & GF_BARRIER)
@ -565,10 +577,10 @@ namespace bcache {
{
assert(!b.ref_count_);
#if 0
nr_locked_--;
if (b.test_flags(BF_FLUSH))
flush();
#endif
if (b.test_flags(BF_DIRTY)) {
if (!b.test_flags(BF_PREVIOUSLY_DIRTY)) {
@ -577,11 +589,9 @@ namespace bcache {
b.set_flags(BF_PREVIOUSLY_DIRTY);
}
#if 0
if (b.test_flags(BF_FLUSH))
flush();
else
#endif
preemptive_writeback();
b.clear_flags(BF_FLUSH);

12
block-cache/block_cache.h
View File

@ -68,10 +68,6 @@ namespace bcache {
set_flags(BF_DIRTY);
}
void mark_flush() {
set_flags(BF_FLUSH);
}
void set_flags(unsigned flags) {
flags_ |= flags;
}
@ -127,7 +123,7 @@ namespace bcache {
enum get_flags {
GF_ZERO = (1 << 0),
GF_DIRTY = (1 << 1),
GF_BARRIER = (1 << 1)
GF_BARRIER = (1 << 2)
};
block_cache::block &get(block_address index, unsigned flags, validator::ptr v);
@ -176,8 +172,9 @@ namespace bcache {
uint64_t nr_data_blocks_;
uint64_t nr_cache_blocks_;
std::auto_ptr<unsigned char> blocks_memory_;
std::auto_ptr<unsigned char> blocks_data_;
// We can't use auto_ptr or unique_ptr because the memory is allocated with malloc
void *blocks_memory_;
void *blocks_data_;
io_context_t aio_context_;
std::vector<io_event> events_;
@ -191,6 +188,7 @@ namespace bcache {
list_head dirty_;
list_head clean_;
unsigned nr_locked_;
unsigned nr_dirty_;
unsigned nr_io_pending_;

5
caching/hint_array.cc
View File

@ -35,10 +35,7 @@ namespace {
// use the appropriate one.
#define all_widths \
xx(4); xx(8); xx(12); xx(16); xx(20); xx(24); xx(28); xx(32);\
xx(36); xx(40); xx(44); xx(48); xx(52); xx(56); xx(60); xx(64); \
xx(68); xx(72); xx(76); xx(80); xx(84); xx(88); xx(92); xx(96); \
xx(100); xx(104); xx(108); xx(112); xx(116); xx(120); xx(124); xx(128);
xx(4);
template <uint32_t WIDTH>
shared_ptr<array_base> mk_array(transaction_manager::ptr tm) {

1
persistent-data/block.h
View File

@ -135,6 +135,7 @@ namespace persistent_data {
bool is_locked(block_address b) const;
private:
int open_or_create_block_file(std::string const &path, off_t file_size, mode m);
void check(block_address b) const;
int fd_;

26
persistent-data/block.tcc
View File

@ -84,10 +84,9 @@ namespace {
int fd = open_file(path, O_CREAT | O_RDWR);
// fallocate didn't seem to work
int r = ::lseek(fd, file_size, SEEK_SET);
int r = ::ftruncate(fd, file_size);
if (r < 0)
syscall_failed("lseek");
syscall_failed("ftruncate");
return fd;
}
@ -179,11 +178,30 @@ namespace persistent_data {
block_address nr_blocks,
unsigned max_concurrent_blocks,
mode m)
: fd_(open_block_file(path, nr_blocks * BlockSize, m == READ_WRITE)),
: fd_(open_or_create_block_file(path, nr_blocks * BlockSize, m)),
bc_(fd_, BlockSize >> SECTOR_SHIFT, nr_blocks, 1024u * 1024u * 16)
{
}
template <uint32_t BlockSize>
int
block_manager<BlockSize>::open_or_create_block_file(string const &path, off_t file_size, mode m)
{
switch (m) {
case READ_ONLY:
return open_block_file(path, file_size, false);
case READ_WRITE:
return open_block_file(path, file_size, true);
case CREATE:
return create_block_file(path, file_size);
default:
throw std::runtime_error("unsupported mode");
}
}
template <uint32_t BlockSize>
typename block_manager<BlockSize>::read_ref
block_manager<BlockSize>::read_lock(block_address location,

6
unit-tests/Makefile.in
View File

@ -25,7 +25,7 @@ GMOCK_FLAGS=\
-Wno-unused-local-typedefs
GMOCK_LIBS=\
-Llib -lpdata -lgmock -lpthread
-Llib -lpdata -lgmock -lpthread -laio
GMOCK_DEPS=\
$(wildcard $(GMOCK_DIR)/include/*.h) \
@ -48,14 +48,12 @@ TEST_SOURCE=\
unit-tests/array_block_t.cc \
unit-tests/array_t.cc \
unit-tests/base64_t.cc \
unit-tests/bitset_t.cc \
unit-tests/block_t.cc \
unit-tests/bitset_t.cc \
unit-tests/bloom_filter_t.cc \
unit-tests/btree_t.cc \
unit-tests/btree_counter_t.cc \
unit-tests/btree_damage_visitor_t.cc \
unit-tests/buffer_t.cc \
unit-tests/cache_t.cc \
unit-tests/cache_superblock_t.cc \
unit-tests/damage_tracker_t.cc \
unit-tests/endian_t.cc \

6
unit-tests/array_block_t.cc
View File

@ -35,7 +35,7 @@ using namespace testing;
namespace {
uint64_t MAX_VALUE = 1000ull;
block_address const NR_BLOCKS = 1024;
typedef typename block_manager<>::noop_validator noop_validator;
typedef typename bcache::noop_validator noop_validator;
typedef typename block_manager<>::read_ref read_ref;
typedef typename block_manager<>::write_ref write_ref;
@ -79,9 +79,9 @@ namespace {
typedef array_block<uint64_traits, write_ref> ablock64;
typedef array_block<uint64_traits, read_ref> ablock64_r;
block_manager<>::validator::ptr
bcache::validator::ptr
validator() {
return block_manager<>::validator::ptr(new block_manager<>::noop_validator);
return bcache::validator::ptr(new bcache::noop_validator);
}
transaction_manager::ptr

51
unit-tests/block_t.cc
View File

@ -30,31 +30,33 @@ using namespace testing;
namespace {
template <uint32_t BlockSize>
void check_all_bytes(typename block_manager<BlockSize>::read_ref const &rr, int v) {
persistent_data::buffer<BlockSize> const &data = rr.data();
unsigned char const *data = reinterpret_cast<unsigned char const *>(rr.data());
for (unsigned b = 0; b < BlockSize; b++)
ASSERT_THAT(data[b], Eq(static_cast<unsigned char>(v)));
}
template <uint32_t BlockSize>
struct zero_validator : public block_manager<BlockSize>::validator {
virtual void check(buffer<BlockSize> const &data, block_address location) const {
struct zero_validator : public bcache::validator {
virtual void check(void const *raw, block_address location) const {
unsigned char const *data = reinterpret_cast<unsigned char const *>(raw);
for (unsigned b = 0; b < BlockSize; b++)
if (data[b] != 0)
throw runtime_error("validator check zero");
}
virtual void prepare(buffer<BlockSize> &data, block_address location) const {
virtual void prepare(void *raw, block_address location) const {
unsigned char *data = reinterpret_cast<unsigned char *>(raw);
for (unsigned b = 0; b < BlockSize; b++)
data[b] = 0;
}
};
class validator_mock : public block_manager<4096>::validator {
class validator_mock : public bcache::validator {
public:
typedef boost::shared_ptr<validator_mock> ptr;
MOCK_CONST_METHOD2(check, void(buffer<4096> const &, block_address));
MOCK_CONST_METHOD2(prepare, void(buffer<4096> &, block_address));
MOCK_CONST_METHOD2(check, void(void const *, block_address));
MOCK_CONST_METHOD2(prepare, void(void *, block_address));
};
typedef block_manager<4096> bm4096;
@ -96,7 +98,7 @@ TEST(BlockTests, writes_persist)
bm4096::ptr bm = create_bm<4096>(nr);
for (unsigned i = 0; i < nr; i++) {
bm4096::write_ref wr = bm->write_lock(i);
::memset(wr.data().raw(), i, 4096);
::memset(wr.data(), i, 4096);
}
for (unsigned i = 0; i < nr; i++) {
@ -115,20 +117,36 @@ TEST(BlockTests, different_block_sizes)
{
{
bm4096::ptr bm = create_bm<4096>(64);
bm4096::read_ref rr = bm->read_lock(0);
ASSERT_THAT(sizeof(rr.data()), Eq(4096u));
{
bm4096::write_ref wr = bm->write_lock(0);
memset(wr.data(), 23, 4096);
}
{
bm4096::write_ref wr = bm->write_lock_zero(0);
check_all_bytes<4096>(wr, 0);
}
}
{
block_manager<64 * 1024>::ptr bm = create_bm<64 * 1024>(64);
block_manager<64 * 1024>::read_ref rr = bm->read_lock(0);
ASSERT_THAT(sizeof(rr.data()), Eq(64u * 1024u));
{
block_manager<64 * 1024>::write_ref wr = bm->write_lock(0);
memset(wr.data(), 72, 64 * 1024);
}
{
block_manager<64 * 1024>::write_ref wr = bm->write_lock_zero(0);
check_all_bytes<64 * 1024>(wr, 0);
}
}
}
TEST(BlockTests, read_validator_works)
{
bm4096::block_manager::validator::ptr v(new zero_validator<4096>());
bcache::validator::ptr v(new zero_validator<4096>());
bm4096::ptr bm = create_bm<4096>(64);
bm->write_lock_zero(0);
bm->read_lock(0, v);
@ -137,11 +155,11 @@ TEST(BlockTests, read_validator_works)
TEST(BlockTests, write_validator_works)
{
bm4096::ptr bm = create_bm<4096>(64);
bm4096::block_manager::validator::ptr v(new zero_validator<4096>());
bcache::validator::ptr v(new zero_validator<4096>());
{
bm4096::write_ref wr = bm->write_lock(0, v);
::memset(wr.data().raw(), 23, sizeof(wr.data().size()));
::memset(wr.data(), 23, 4096);
}
bm->flush(); // force the prepare method to be called
@ -422,7 +440,8 @@ TEST_F(ValidatorTests, validator_check_failure_gets_passed_up)
EXPECT_CALL(*v, check(_, Eq(0ull))).Times(1).WillOnce(Throw(my_error("bang!")));
ASSERT_THROW(bm->read_lock(0, v), my_error);
ASSERT_FALSE(bm->is_locked(0));
// FIXME: put this back in
//ASSERT_FALSE(bm->is_locked(0));
}
//----------------------------------------------------------------

1
unit-tests/buffer_t.cc
View File

@ -20,7 +20,6 @@
#define COMPILE_TIME_ERROR 0
#include "gmock/gmock.h"
#include "persistent-data/buffer.h"
using namespace persistent_data;
using namespace testing;

2
unit-tests/test_utils.cc
View File

@ -9,7 +9,7 @@ using namespace persistent_data;
void test::zero_block(block_manager<>::ptr bm, block_address b)
{
block_manager<>::write_ref wr = bm->write_lock(b);
memset(&wr.data(), 0, sizeof(wr.data()));
memset(wr.data(), 0, 4096);
}
transaction_manager::ptr

4
unit-tests/test_utils.h
View File

@ -19,6 +19,8 @@
#include "persistent-data/block.h"
#include "persistent-data/transaction_manager.h"
#include <linux/limits.h>
//----------------------------------------------------------------
namespace test {
@ -36,7 +38,7 @@ namespace test {
return typename block_manager<BlockSize>::ptr(
new block_manager<BlockSize>(path, nr, MAX_HELD_LOCKS,
block_io<BlockSize>::CREATE));
block_manager<BlockSize>::CREATE));
}
// Don't use this to update the metadata.

34
unit-tests/transaction_manager_t.cc
View File

@ -40,11 +40,11 @@ namespace {
return tm;
}
typedef block_manager<>::validator::ptr validator_ptr;
typedef bcache::validator::ptr validator_ptr;
validator_ptr noop_validator() {
return block_manager<>::validator::ptr(
new block_manager<>::noop_validator);
validator_ptr mk_noop_validator() {
return bcache::validator::ptr(
new bcache::noop_validator);
}
typedef block_manager<>::write_ref write_ref;
@ -55,20 +55,20 @@ namespace {
TEST(TransactionManagerTests, commit_succeeds)
{
transaction_manager::ptr tm = create_tm();
tm->begin(0, noop_validator());
tm->begin(0, mk_noop_validator());
}
TEST(TransactionManagerTests, shadowing)
{
transaction_manager::ptr tm = create_tm();
block_manager<>::write_ref superblock = tm->begin(0, noop_validator());
block_manager<>::write_ref superblock = tm->begin(0, mk_noop_validator());
space_map::ptr sm = tm->get_sm();
sm->inc(1);
block_address b;
{
pair<write_ref, bool> p = tm->shadow(1, noop_validator());
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
b = p.first.get_location();
ASSERT_THAT(b, Ne(1u));
ASSERT_FALSE(p.second);
@ -76,7 +76,7 @@ TEST(TransactionManagerTests, shadowing)
}
{
pair<write_ref, bool> p = tm->shadow(b, noop_validator());
pair<write_ref, bool> p = tm->shadow(b, mk_noop_validator());
ASSERT_THAT(p.first.get_location(), Eq(b));
ASSERT_FALSE(p.second);
}
@ -84,7 +84,7 @@ TEST(TransactionManagerTests, shadowing)
sm->inc(b);
{
pair<write_ref, bool> p = tm->shadow(b, noop_validator());
pair<write_ref, bool> p = tm->shadow(b, mk_noop_validator());
ASSERT_THAT(p.first.get_location(), Ne(b));
ASSERT_TRUE(p.second);
}
@ -98,8 +98,8 @@ TEST(TransactionManagerTests, multiple_shadowing)
block_address b, b2;
{
write_ref superblock = tm->begin(0, noop_validator());
pair<write_ref, bool> p = tm->shadow(1, noop_validator());
write_ref superblock = tm->begin(0, mk_noop_validator());
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
b = p.first.get_location();
ASSERT_THAT(b, Ne(1u));
ASSERT_TRUE(p.second);
@ -107,8 +107,8 @@ TEST(TransactionManagerTests, multiple_shadowing)
}
{
write_ref superblock = tm->begin(0, noop_validator());
pair<write_ref, bool> p = tm->shadow(1, noop_validator());
write_ref superblock = tm->begin(0, mk_noop_validator());
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
b2 = p.first.get_location();
ASSERT_THAT(b2, Ne(1u));
ASSERT_THAT(b2, Ne(b));
@ -117,8 +117,8 @@ TEST(TransactionManagerTests, multiple_shadowing)
}
{
write_ref superblock = tm->begin(0, noop_validator());
pair<write_ref, bool> p = tm->shadow(1, noop_validator());
write_ref superblock = tm->begin(0, mk_noop_validator());
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
block_address b3 = p.first.get_location();
ASSERT_THAT(b3, Ne(b2));
ASSERT_THAT(b3, Ne(b));
@ -131,8 +131,8 @@ TEST(TransactionManagerTests, multiple_shadowing)
TEST(TransactionManagerTests, shadow_free_block_fails)
{
transaction_manager::ptr tm = create_tm();
write_ref superblock = tm->begin(0, noop_validator());
ASSERT_THROW(tm->shadow(1, noop_validator()), runtime_error);
write_ref superblock = tm->begin(0, mk_noop_validator());
ASSERT_THROW(tm->shadow(1, mk_noop_validator()), runtime_error);
}
//----------------------------------------------------------------