Introduce a new function, scrub_tree_mirror(), to scrub mirror based
tree blocks (Single/DUP/RAID0/1/10)

This function can also be used on in-memory tree blocks using @data
parameter.
This is very handy for RAID5/6 case, either checking the data stripe
tree block by @bytenr and 0 as @mirror, or using @data parameter for
recovered in-memory data.

Signed-off-by: Qu Wenruo <quwen...@cn.fujitsu.com>
---
 disk-io.c |  4 ++--
 disk-io.h |  2 ++
 scrub.c   | 72 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 76 insertions(+), 2 deletions(-)

diff --git a/disk-io.c b/disk-io.c
index 9140a81b..d5011572 100644
--- a/disk-io.c
+++ b/disk-io.c
@@ -51,8 +51,8 @@ static u32 max_nritems(u8 level, u32 nodesize)
                sizeof(struct btrfs_key_ptr));
 }
 
-static int check_tree_block(struct btrfs_fs_info *fs_info,
-                           struct extent_buffer *buf)
+int check_tree_block(struct btrfs_fs_info *fs_info,
+                    struct extent_buffer *buf)
 {
 
        struct btrfs_fs_devices *fs_devices;
diff --git a/disk-io.h b/disk-io.h
index 4de9fef7..db883d57 100644
--- a/disk-io.h
+++ b/disk-io.h
@@ -119,6 +119,8 @@ static inline struct extent_buffer* read_tree_block(
                        parent_transid);
 }
 
+int check_tree_block(struct btrfs_fs_info *fs_info,
+                    struct extent_buffer *buf);
 int read_extent_data(struct btrfs_root *root, char *data, u64 logical,
                     u64 *len, int mirror);
 void readahead_tree_block(struct btrfs_root *root, u64 bytenr, u32 blocksize,
diff --git a/scrub.c b/scrub.c
index c9ca817e..4cf678fb 100644
--- a/scrub.c
+++ b/scrub.c
@@ -118,3 +118,75 @@ static struct scrub_full_stripe *alloc_full_stripe(int 
nr_stripes,
        }
        return ret;
 }
+
+static inline int is_data_stripe(struct scrub_stripe *stripe)
+{
+       u64 bytenr = stripe->logical;
+
+       if (bytenr == BTRFS_RAID5_P_STRIPE || bytenr == BTRFS_RAID6_Q_STRIPE)
+               return 0;
+       return 1;
+}
+
+/*
+ * Scrub one tree mirror given by @bytenr and @mirror, or @data.
+ * If @data is not given(NULL), the function will try to read out tree block
+ * using @bytenr and @mirror.
+ * If @data is given, use data directly, won't try to read from disk.
+ *
+ * The extra @data prameter is handy for RAID5/6 recovery code to verify
+ * the recovered data.
+ *
+ * Return 0 if everything is OK.
+ * Return <0 something goes wrong, and @scrub_ctx accounting will be updated
+ * if it's a data corruption.
+ */
+static int scrub_tree_mirror(struct btrfs_fs_info *fs_info,
+                            struct btrfs_scrub_progress *scrub_ctx,
+                            char *data, u64 bytenr, int mirror)
+{
+       struct extent_buffer *eb;
+       u32 nodesize = fs_info->tree_root->nodesize;
+       int ret;
+
+       if (!IS_ALIGNED(bytenr, fs_info->tree_root->sectorsize)) {
+               /* Such error will be reported by check_tree_block() */
+               scrub_ctx->verify_errors++;
+               return -EIO;
+       }
+
+       eb = btrfs_find_create_tree_block(fs_info, bytenr, nodesize);
+       if (!eb)
+               return -ENOMEM;
+       if (data) {
+               memcpy(eb->data, data, nodesize);
+       } else {
+               ret = read_whole_eb(fs_info, eb, mirror);
+               if (ret) {
+                       scrub_ctx->read_errors++;
+                       error("failed to read tree block %llu mirror %d",
+                             bytenr, mirror);
+                       goto out;
+               }
+       }
+
+       scrub_ctx->tree_bytes_scrubbed += nodesize;
+       if (csum_tree_block(fs_info->tree_root, eb, 1)) {
+               error("tree block %llu mirror %d checksum mismatch", bytenr,
+                       mirror);
+               scrub_ctx->csum_errors++;
+               ret = -EIO;
+               goto out;
+       }
+       ret = check_tree_block(fs_info, eb);
+       if (ret < 0) {
+               error("tree block %llu mirror %d is invalid", bytenr, mirror);
+               scrub_ctx->verify_errors++;
+               goto out;
+       }
+
+       scrub_ctx->tree_extents_scrubbed++;
+out:
+       free_extent_buffer(eb);
+       return ret;
+}
-- 
2.11.0



--
To unsubscribe from this list: send the line "unsubscribe linux-btrfs" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to