thin-provisioning-tools/persistent-data/block.tcc

242 lines
6.2 KiB
Plaintext
Raw Normal View History

2011-12-16 00:04:31 +05:30
// Copyright (C) 2011 Red Hat, Inc. All rights reserved.
2011-12-06 19:23:05 +05:30
//
2011-12-06 19:13:56 +05:30
// 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 "block.h"
#include "base/error_string.h"
#include "base/file_utils.h"
#include "block-cache/io_engine.h"
2011-06-27 15:15:30 +05:30
#include <boost/bind.hpp>
#include <stdexcept>
//----------------------------------------------------------------
2013-05-28 18:18:10 +05:30
namespace persistent_data {
template <uint32_t BlockSize>
2014-07-28 18:43:28 +05:30
block_manager<BlockSize>::read_ref::read_ref(block_cache::block &b)
: b_(b)
2013-05-28 18:18:10 +05:30
{
2011-10-24 22:34:19 +05:30
}
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
block_manager<BlockSize>::read_ref::read_ref(read_ref const &rhs)
2014-07-28 18:43:28 +05:30
: b_(rhs.b_)
2013-05-28 18:18:10 +05:30
{
2014-07-28 18:43:28 +05:30
b_.get();
2011-10-24 22:34:19 +05:30
}
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
block_manager<BlockSize>::read_ref::~read_ref()
{
2014-07-28 18:43:28 +05:30
b_.put();
2013-05-28 18:18:10 +05:30
}
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
typename block_manager<BlockSize>::read_ref const &
block_manager<BlockSize>::read_ref::operator =(read_ref const &rhs)
{
if (this != &rhs) {
2014-07-25 20:44:24 +05:30
b_ = rhs.b_;
2014-07-28 18:43:28 +05:30
b_.get();
2013-05-28 18:18:10 +05:30
}
2013-05-28 18:18:10 +05:30
return *this;
}
template <uint32_t BlockSize>
block_address
block_manager<BlockSize>::read_ref::get_location() const
{
2014-07-25 20:44:24 +05:30
return b_.get_index();
2013-05-28 18:18:10 +05:30
}
template <uint32_t BlockSize>
2014-07-25 19:16:51 +05:30
void const *
2013-05-28 18:18:10 +05:30
block_manager<BlockSize>::read_ref::data() const
{
2014-07-25 20:44:24 +05:30
return b_.get_data();
2013-05-28 18:18:10 +05:30
}
//--------------------------------
2011-10-24 22:34:19 +05:30
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
2014-07-28 18:43:28 +05:30
block_manager<BlockSize>::write_ref::write_ref(block_cache::block &b)
: read_ref(b),
ref_count_(NULL)
2013-05-28 18:18:10 +05:30
{
}
2011-06-27 15:15:30 +05:30
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
block_manager<BlockSize>::write_ref::write_ref(block_cache::block &b, unsigned &ref_count)
: read_ref(b),
ref_count_(&ref_count) {
if (*ref_count_)
throw std::runtime_error("superblock already locked");
(*ref_count_)++;
}
template <uint32_t BlockSize>
block_manager<BlockSize>::write_ref::write_ref(write_ref const &rhs)
: read_ref(rhs),
ref_count_(rhs.ref_count_) {
if (ref_count_)
(*ref_count_)++;
}
template <uint32_t BlockSize>
block_manager<BlockSize>::write_ref::~write_ref()
2013-05-28 18:18:10 +05:30
{
if (ref_count_) {
if (!*ref_count_) {
std::cerr << "write_ref ref_count going below zero";
::exit(1);
}
(*ref_count_)--;
}
2014-07-25 20:44:24 +05:30
}
template <uint32_t BlockSize>
typename block_manager<BlockSize>::write_ref const &
block_manager<BlockSize>::write_ref::operator =(write_ref const &rhs)
{
if (&rhs != this) {
read_ref::operator =(rhs);
ref_count_ = rhs.ref_count_;
if (ref_count_)
(*ref_count_)++;
}
}
2014-07-25 20:44:24 +05:30
template <uint32_t BlockSize>
void *
block_manager<BlockSize>::write_ref::data()
2014-07-28 18:43:28 +05:30
{
return read_ref::b_.get_data();
2013-05-28 18:18:10 +05:30
}
//----------------------------------------------------------------
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
block_manager<BlockSize>::block_manager(std::string const &path,
block_address nr_blocks,
unsigned max_concurrent_blocks,
mode m,
bool excl)
2015-08-19 17:02:57 +05:30
// FIXME: * BlockSize ?
: fd_(open_or_create_block_file(path, nr_blocks * BlockSize, m, excl)),
bc_(fd_, BlockSize >> SECTOR_SHIFT, nr_blocks, 1024u * 1024u * 16),
superblock_ref_count_(0)
2013-05-28 18:18:10 +05:30
{
}
2014-07-29 16:04:26 +05:30
template <uint32_t BlockSize>
int
block_manager<BlockSize>::open_or_create_block_file(std::string const &path, off_t file_size, mode m, bool excl)
2014-07-29 16:04:26 +05:30
{
switch (m) {
case READ_ONLY:
return file_utils::open_block_file(path, file_size, false, excl);
2014-07-29 16:04:26 +05:30
case READ_WRITE:
return file_utils::open_block_file(path, file_size, true, excl);
2014-07-29 16:04:26 +05:30
case CREATE:
return file_utils::create_block_file(path, file_size);
2014-07-29 16:04:26 +05:30
default:
throw std::runtime_error("unsupported mode");
}
}
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
typename block_manager<BlockSize>::read_ref
block_manager<BlockSize>::read_lock(block_address location,
2014-07-25 19:16:51 +05:30
typename bcache::validator::ptr v) const
2013-05-28 18:18:10 +05:30
{
2014-07-28 18:43:28 +05:30
block_cache::block &b = bc_.get(location, 0, v);
return read_ref(b);
2013-05-28 18:18:10 +05:30
}
2011-07-13 19:39:33 +05:30
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
typename block_manager<BlockSize>::write_ref
block_manager<BlockSize>::write_lock(block_address location,
2014-07-25 19:16:51 +05:30
typename bcache::validator::ptr v)
2013-05-28 18:18:10 +05:30
{
2014-07-28 18:43:28 +05:30
block_cache::block &b = bc_.get(location, block_cache::GF_DIRTY, v);
return write_ref(b);
2011-10-24 22:34:19 +05:30
}
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
typename block_manager<BlockSize>::write_ref
block_manager<BlockSize>::write_lock_zero(block_address location,
2014-07-25 19:16:51 +05:30
typename bcache::validator::ptr v)
2013-05-28 18:18:10 +05:30
{
2014-07-28 18:43:28 +05:30
block_cache::block &b = bc_.get(location, block_cache::GF_ZERO, v);
return write_ref(b);
}
2011-10-24 22:34:19 +05:30
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
typename block_manager<BlockSize>::write_ref
block_manager<BlockSize>::superblock(block_address location,
2014-07-25 19:16:51 +05:30
typename bcache::validator::ptr v)
2013-05-28 18:18:10 +05:30
{
if (bc_.get_nr_locked() > 0)
throw std::runtime_error("attempt to lock superblock while other locks are still held");
block_cache::block &b = bc_.get(location, block_cache::GF_DIRTY | block_cache::GF_BARRIER, v);
return write_ref(b, superblock_ref_count_);
}
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
typename block_manager<BlockSize>::write_ref
block_manager<BlockSize>::superblock_zero(block_address location,
2014-07-25 19:16:51 +05:30
typename bcache::validator::ptr v)
2013-05-28 18:18:10 +05:30
{
if (bc_.get_nr_locked() > 0)
throw std::runtime_error("attempt to lock superblock while other locks are still held");
2014-07-28 18:43:28 +05:30
block_cache::block &b = bc_.get(location, block_cache::GF_ZERO | block_cache::GF_BARRIER, v);
return write_ref(b, superblock_ref_count_);
2013-05-28 18:18:10 +05:30
}
2011-07-13 19:39:33 +05:30
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
block_address
block_manager<BlockSize>::get_nr_blocks() const
{
2014-07-25 15:05:04 +05:30
return bc_.get_nr_blocks();
2013-05-28 18:18:10 +05:30
}
2011-07-22 20:39:56 +05:30
template <uint32_t BlockSize>
void
block_manager<BlockSize>::prefetch(block_address b) const
{
bc_.prefetch(b);
}
2013-05-28 18:18:10 +05:30
template <uint32_t BlockSize>
void
block_manager<BlockSize>::flush() const
{
2014-07-25 15:05:04 +05:30
bc_.flush();
2013-05-28 18:18:10 +05:30
}
}
//----------------------------------------------------------------