From 6fe92d4a6315d4b2f19681979e5375425b654096 Mon Sep 17 00:00:00 2001
From: Joe Thornber <ejt@redhat.com>
Date: Wed, 22 May 2013 14:55:28 +0100
Subject: [PATCH] [thin_check] wire up mapping tree checker

---
 Makefile.in                     |  1 +
 features/thin_check.feature     |  8 ++++--
 thin-provisioning/thin_check.cc | 45 +++++++++++++++++++++++++++++++--
 3 files changed, 50 insertions(+), 4 deletions(-)

diff --git a/Makefile.in b/Makefile.in
index 6b83df4..e4e35f7 100644
--- a/Makefile.in
+++ b/Makefile.in
@@ -136,6 +136,7 @@ THIN_CHECK_SOURCE=\
 	persistent-data/transaction_manager.cc \
 	thin-provisioning/file_utils.cc \
 	thin-provisioning/device_tree.cc \
+	thin-provisioning/mapping_tree.cc \
 	thin-provisioning/metadata.cc \
 	thin-provisioning/metadata_checker.cc \
 	thin-provisioning/superblock.cc
diff --git a/features/thin_check.feature b/features/thin_check.feature
index 526ec1a..69f9c74 100644
--- a/features/thin_check.feature
+++ b/features/thin_check.feature
@@ -49,8 +49,12 @@ Feature: thin_check
     When I run thin_check with --super-block-only
     Then it should pass
 
-  @announce
   Scenario: --super-block-only check fails with corrupt superblock
     Given a corrupt superblock
     When I run thin_check with --super-block-only
-    Then it should fail
\ No newline at end of file
+    Then it should fail with:
+    """
+    examining superblock
+      superblock is corrupt
+        bad checksum in superblock
+    """
\ No newline at end of file
diff --git a/thin-provisioning/thin_check.cc b/thin-provisioning/thin_check.cc
index 0c59606..011bff5 100644
--- a/thin-provisioning/thin_check.cc
+++ b/thin-provisioning/thin_check.cc
@@ -198,12 +198,44 @@ namespace {
 
 	//--------------------------------
 
+	class mapping_reporter : public mapping_tree_detail::damage_visitor {
+	public:
+		mapping_reporter(nested_output &out)
+		: out_(out) {
+		}
+
+		virtual void visit(mapping_tree_detail::missing_devices const &d) {
+			out_ << "missing all mappings for devices: " << d.keys_ << end_message();
+			{
+				auto _ = out_.push();
+				out_ << d.desc_ << end_message();
+			}
+			err_ = combine_errors(err_, FATAL);
+		}
+
+		virtual void visit(mapping_tree_detail::missing_mappings const &d) {
+			out_ << "thin device " << d.thin_dev_ << " is missing mappings " << d.keys_ << end_message();
+			{
+				auto _ = out_.push();
+				out_ << d.desc_ << end_message();
+			}
+			err_ = combine_errors(err_, FATAL);
+		}
+
+	private:
+		nested_output &out_;
+		error_state err_;
+	};
+
+	//--------------------------------
+
 	error_state metadata_check(string const &path) {
 		block_manager<>::ptr bm = open_bm(path);
 
 		nested_output out(cerr, 2);
 		superblock_reporter sb_rep(out);
 		devices_reporter dev_rep(out);
+		mapping_reporter mapping_rep(out);
 
 		out << "examining superblock" << end_message();
 		{
@@ -214,16 +246,25 @@ namespace {
 		if (sb_rep.get_error() == FATAL)
 			return FATAL;
 
+		superblock_detail::superblock sb = read_superblock(bm);
+		transaction_manager::ptr tm = open_tm(bm);
+
 		out << "examining devices tree" << end_message();
 		{
 			auto _ = out.push();
-			superblock_detail::superblock sb = read_superblock(bm);
-			transaction_manager::ptr tm = open_tm(bm);
 			device_tree dtree(tm, sb.device_details_root_,
 					  device_tree_detail::device_details_traits::ref_counter());
 			check_device_tree(dtree, dev_rep);
 		}
 
+		out << "examining mapping tree" << end_message();
+		{
+			auto _ = out.push();
+			mapping_tree mtree(tm, sb.data_mapping_root_,
+					   mapping_tree_detail::block_traits::ref_counter(tm->get_sm()));
+			check_mapping_tree(mtree, mapping_rep);
+		}
+
 		return combine_errors(sb_rep.get_error(),
 				      dev_rep.get_error());
 	}