From 2937ed2d2bd063187884904ecb2ad15f1e0a2b51 Mon Sep 17 00:00:00 2001
From: Joe Thornber <ejt@redhat.com>
Date: Fri, 28 Oct 2011 14:01:44 +0100
Subject: [PATCH] restore_emitter

---
 Makefile           |   1 +
 restore_emitter.cc | 124 +++++++++++++++++++++++++++++++++++++++++++++
 restore_emitter.h  |  15 ++++++
 3 files changed, 140 insertions(+)
 create mode 100644 restore_emitter.cc
 create mode 100644 restore_emitter.h

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<uint32_t>(dev);
+		}
+
+		virtual void end_device() {
+			current_device_ = optional<uint32_t>();
+		}
+
+		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<uint32_t> 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