diff --git a/Makefile b/Makefile index dec9b18..1cb844e 100644 --- a/Makefile +++ b/Makefile @@ -12,6 +12,7 @@ SOURCE=\ metadata_checker.cc \ metadata_dumper.cc \ metadata_disk_structures.cc \ + restore_emitter.cc \ space_map_disk.cc \ thin_pool.cc \ transaction_manager.cc \ diff --git a/restore_emitter.cc b/restore_emitter.cc new file mode 100644 index 0000000..3fe2046 --- /dev/null +++ b/restore_emitter.cc @@ -0,0 +1,124 @@ +#include "restore_emitter.h" + +using namespace boost; +using namespace std; +using namespace thin_provisioning; + +//---------------------------------------------------------------- + +namespace { + class restorer : public emitter { + public: + restorer(metadata::ptr md) + : md_(md), + in_superblock_(false) { + } + + virtual ~restorer() { + if (in_superblock_) { + throw runtime_error("still in superblock"); + } + } + + virtual void begin_superblock(std::string const &uuid, + uint64_t time, + uint64_t trans_id, + uint32_t data_block_size) { + in_superblock_ = true; + + superblock &sb = md_->sb_; + memcpy(&sb.uuid_, &uuid, sizeof(&sb.uuid_)); + sb.time_ = time; + sb.trans_id_ = trans_id; + sb.data_block_size_ = data_block_size; + } + + virtual void end_superblock() { + if (!in_superblock_) + throw runtime_error("missing superblock"); + + md_->commit(); + + in_superblock_ = false; + } + + virtual void begin_device(uint32_t dev, + uint64_t mapped_blocks, + uint64_t trans_id, + uint64_t creation_time, + uint64_t snap_time) { + if (!in_superblock_) + throw runtime_error("missing superblock"); + + if (device_exists(dev)) + throw std::runtime_error("Device already exists"); + + // Add entry to the details tree + uint64_t key[1] = {dev}; + device_details details = {mapped_blocks, trans_id, creation_time, snap_time}; + md_->details_.insert(key, details); + + // Insert an empty mapping tree + single_mapping_tree::ptr new_tree( + new single_mapping_tree(md_->tm_, + block_time_ref_counter(md_->data_sm_))); + md_->mappings_top_level_.insert(key, new_tree->get_root()); + md_->mappings_.set_root(md_->mappings_top_level_.get_root()); // FIXME: ugly + + current_device_ = optional(dev); + } + + virtual void end_device() { + current_device_ = optional(); + } + + virtual void begin_named_mapping(std::string const &name) { + throw runtime_error("not implemented"); + } + + virtual void end_named_mapping() { + throw runtime_error("not implemented"); + } + + virtual void identifier(std::string const &name) { + throw runtime_error("not implemented"); + } + + virtual void range_map(uint64_t origin_begin, uint64_t data_begin, uint32_t time, uint64_t len) { + for (uint64_t i = 0; i < len; i++) + single_map(origin_begin++, data_begin++, time); + } + + virtual void single_map(uint64_t origin_block, uint64_t data_block, uint32_t time) { + if (!current_device_) + throw runtime_error("not in device"); + + uint64_t key[2] = {*current_device_, origin_block}; + block_time bt; + bt.block_ = data_block; + bt.time_ = time; + md_->mappings_.insert(key, bt); + md_->mappings_top_level_.set_root(md_->mappings_.get_root()); + } + + private: + bool device_exists(thin_dev_t dev) const { + uint64_t key[1] = {dev}; + return md_->details_.lookup(key); + } + + metadata::ptr md_; + bool in_superblock_; + optional current_device_; + }; +} + +//---------------------------------------------------------------- + +emitter::ptr +thin_provisioning::create_restore_emitter(metadata::ptr md) +{ + return emitter::ptr(new restorer(md)); +} + +//---------------------------------------------------------------- diff --git a/restore_emitter.h b/restore_emitter.h new file mode 100644 index 0000000..749c75f --- /dev/null +++ b/restore_emitter.h @@ -0,0 +1,15 @@ +#ifndef RESTORE_EMITTER_H +#define RESTORE_EMITTER_H + +#include "emitter.h" +#include "metadata.h" + +//---------------------------------------------------------------- + +namespace thin_provisioning { + emitter::ptr create_restore_emitter(metadata::ptr md); +} + +//---------------------------------------------------------------- + +#endif