@@ -5915,6 +5915,125 @@ static bool need_full_stripe(enum btrfs_map_op op)
return (op == BTRFS_MAP_WRITE || op == BTRFS_MAP_GET_READ_MIRRORS);
}
+/*
+ * btrfs_get_io_geometry - calculates the geomery of a particular (address, len)
+ * tuple. This information is used to calculate how big a
+ * particular bio can get before it straddles a stripe.
+ *
+ * @fs_info - The omnipresent btrfs structure
+ * @logical - Address that we want to figure out the geometry of
+ * @len - Size of IO we are going to perform, starting at @logical
+ * @op - Type of operation - Write or Read
+ * @out_len - length before we cross btrfs' internal stripe boundary
+ * @out_offset - offset of @logical in chunk
+ * @out_stripe_len - length of a single stripe
+ * @out_stripe_nr - the number of stripe @logical falls within
+ * @out_stripe_offset - offset of @logical in stripe
+ * @out_raid56_stripe_offset - offset of raid56 stripe in chunk
+ *
+ * Returns < 0 in case a chunk for the given logical address cannot be found,
+ * usually shouldn't happen unless @logical is corrupted, 0 otherwise.
+ */
+static int btrfs_get_io_geometry(struct btrfs_fs_info *fs_info,
+ enum btrfs_map_op op, u64 logical, u64 len,
+ u64 *out_len, u64 *out_offset, u64 *out_stripe_len,
+ u64 *out_stripe_nr, u64 *out_stripe_offset,
+ u64 *out_raid56_stripe_offset)
+
+{
+ struct extent_map *em;
+ struct map_lookup *map;
+ u64 offset;
+ u64 stripe_offset;
+ u64 stripe_nr;
+ u64 stripe_len;
+ u64 raid56_full_stripe_start = (u64)-1;
+ int data_stripes;
+
+ ASSERT(op != BTRFS_MAP_DISCARD);
+
+ em = btrfs_get_chunk_map(fs_info, logical, len);
+ if (IS_ERR(em))
+ return PTR_ERR(em);
+
+ map = em->map_lookup;
+ /* Offset of this logical address in the chunk */
+ offset = logical - em->start;
+ /* Len of a stripe in a chunk */
+ stripe_len = map->stripe_len;
+ /* Stripe wher this block falls in */
+ stripe_nr = div64_u64(offset, stripe_len);
+ /* Offset of stripe in the chunk */
+ stripe_offset = stripe_nr * stripe_len;
+ if (offset < stripe_offset) {
+ btrfs_crit(fs_info,
+ "stripe math has gone wrong, stripe_offset=%llu, offset=%llu, start=%llu, logical=%llu, stripe_len=%llu",
+ stripe_offset, offset, em->start, logical,
+ stripe_len);
+ free_extent_map(em);
+ return -EINVAL;
+ }
+
+ /* stripe_offset is the offset of this block in its stripe*/
+ stripe_offset = offset - stripe_offset;
+ data_stripes = nr_data_stripes(map);
+
+ if (map->type & BTRFS_BLOCK_GROUP_PROFILE_MASK) {
+ u64 max_len = stripe_len - stripe_offset;
+
+ /*
+ * In case of raid56, we need to know the stripe aligned start
+ */
+ if (map->type & BTRFS_BLOCK_GROUP_RAID56_MASK) {
+ unsigned long full_stripe_len = stripe_len * data_stripes;
+ raid56_full_stripe_start = offset;
+
+ /*
+ * Allow a write of a full stripe, but make sure we
+ * don't allow straddling of stripes
+ */
+ raid56_full_stripe_start = div64_u64(raid56_full_stripe_start,
+ full_stripe_len);
+ raid56_full_stripe_start *= full_stripe_len;
+
+ /*
+ * For writes to RAID[56], allow a full stripeset across
+ * all disks. For other RAID types and for RAID[56]
+ * reads, just allow a single stripe (on a single disk).
+ */
+ if (op == BTRFS_MAP_WRITE) {
+ max_len = stripe_len * data_stripes -
+ (offset - raid56_full_stripe_start);
+ }
+ }
+ len = min_t(u64, em->len - offset, max_len);
+ } else {
+ len = em->len - offset;
+ }
+
+ if (out_len)
+ *out_len = len;
+ if (out_offset)
+ *out_offset = offset;
+ if (out_stripe_len)
+ *out_stripe_len = stripe_len;
+ if (out_stripe_nr)
+ *out_stripe_nr = stripe_nr;
+ if (out_stripe_offset)
+ *out_stripe_offset = stripe_offset;
+ if (out_raid56_stripe_offset)
+ *out_raid56_stripe_offset = raid56_full_stripe_start;
+
+ return 0;
+}
+
+int btrfs_get_stripe_rem(struct btrfs_fs_info *fs_info, enum btrfs_map_op op,
+ u64 logical, u64 len, u64 *out_len)
+{
+ return btrfs_get_io_geometry(fs_info, op, logical, len, out_len, NULL,
+ NULL, NULL, NULL, NULL);
+}
+
static int __btrfs_map_block(struct btrfs_fs_info *fs_info,
enum btrfs_map_op op,
u64 logical, u64 *length,
@@ -413,6 +413,8 @@ int btrfs_map_block(struct btrfs_fs_info *fs_info, enum btrfs_map_op op,
int btrfs_map_sblock(struct btrfs_fs_info *fs_info, enum btrfs_map_op op,
u64 logical, u64 *length,
struct btrfs_bio **bbio_ret);
+int btrfs_get_stripe_rem(struct btrfs_fs_info *fs_info, enum btrfs_map_op op,
+ u64 logical, u64 len, u64 *out_len);
int btrfs_rmap_block(struct btrfs_fs_info *fs_info, u64 chunk_start,
u64 physical, u64 **logical, int *naddrs, int *stripe_len);
int btrfs_read_sys_array(struct btrfs_fs_info *fs_info);
Prepare for refactoring and simplifying the code in __btrfs_map_block by factoring out common code into a separete function - btrfs_get_io_geometry. It perform necessary actions to map a (logical addr, len) pair to the underlying physical disk. This code is also necessary to figure if particular IO req will span a btrfs stripe, to that effect __btrfs_map_block was called with bio_ret parameter set to NULL as a way to indicate we only need the "mapping" logic and not submit. Further it introduces a simple wrapper over its numerous return parameters. Having multiple return params rather than a single struct is a deliberate choice so as not to bloat the stack. Signed-off-by: Nikolay Borisov <nborisov@suse.com> --- fs/btrfs/volumes.c | 119 +++++++++++++++++++++++++++++++++++++++++++++ fs/btrfs/volumes.h | 2 + 2 files changed, 121 insertions(+)