struct inode *inode = page->mapping->host;
        struct btrfs_fs_info *fs_info = btrfs_sb(inode->i_sb);
        u64 logical = bio->bi_iter.bi_sector << 9;
+       struct extent_map *em;
        u64 length = 0;
        u64 map_length;
-       int ret;
+       int ret = 0;
        struct btrfs_io_geometry geom;
 
        if (bio_flags & EXTENT_BIO_COMPRESSED)
 
        length = bio->bi_iter.bi_size;
        map_length = length;
-       ret = btrfs_get_io_geometry(fs_info, btrfs_op(bio), logical, map_length,
-                                   &geom);
+       em = btrfs_get_chunk_map(fs_info, logical, map_length);
+       if (IS_ERR(em))
+               return PTR_ERR(em);
+       ret = btrfs_get_io_geometry(fs_info, em, btrfs_op(bio), logical,
+                                   map_length, &geom);
        if (ret < 0)
-               return ret;
+               goto out;
 
        if (geom.len < length + size)
-               return 1;
-       return 0;
+               ret = 1;
+out:
+       free_extent_map(em);
+       return ret;
 }
 
 /*
        u64 submit_len;
        int clone_offset = 0;
        int clone_len;
+       u64 logical;
        int ret;
        blk_status_t status;
        struct btrfs_io_geometry geom;
        struct btrfs_dio_data *dio_data = iomap->private;
+       struct extent_map *em = NULL;
 
        dip = btrfs_create_dio_private(dio_bio, inode, file_offset);
        if (!dip) {
        submit_len = dio_bio->bi_iter.bi_size;
 
        do {
-               ret = btrfs_get_io_geometry(fs_info, btrfs_op(dio_bio),
-                                           start_sector << 9, submit_len,
-                                           &geom);
+               logical = start_sector << 9;
+               em = btrfs_get_chunk_map(fs_info, logical, submit_len);
+               if (IS_ERR(em)) {
+                       status = errno_to_blk_status(PTR_ERR(em));
+                       em = NULL;
+                       goto out_err_em;
+               }
+               ret = btrfs_get_io_geometry(fs_info, em, btrfs_op(dio_bio),
+                                           logical, submit_len, &geom);
                if (ret) {
                        status = errno_to_blk_status(ret);
-                       goto out_err;
+                       goto out_err_em;
                }
                ASSERT(geom.len <= INT_MAX);
 
                        bio_put(bio);
                        if (submit_len > 0)
                                refcount_dec(&dip->refs);
-                       goto out_err;
+                       goto out_err_em;
                }
 
                dio_data->submitted += clone_len;
                clone_offset += clone_len;
                start_sector += clone_len >> 9;
                file_offset += clone_len;
+
+               free_extent_map(em);
        } while (submit_len > 0);
        return BLK_QC_T_NONE;
 
+out_err_em:
+       free_extent_map(em);
 out_err:
        dip->dio_bio->bi_status = status;
        btrfs_dio_private_put(dip);
+
        return BLK_QC_T_NONE;
 }
 
 
 }
 
 /*
- * 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.
+ * Calculate the geometry 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 filesystem
- * @logical - address that we want to figure out the geometry of
- * @len            - the length of IO we are going to perform, starting at @logical
- * @op      - type of operation - write or read
- * @io_geom - pointer used to return values
+ * @fs_info: the filesystem
+ * @em:      mapping containing the logical extent
+ * @op:      type of operation - write or read
+ * @logical: address that we want to figure out the geometry of
+ * @len:     the length of IO we are going to perform, starting at @logical
+ * @io_geom: pointer used to return values
  *
  * Returns < 0 in case a chunk for the given logical address cannot be found,
  * usually shouldn't happen unless @logical is corrupted, 0 otherwise.
  */
-int btrfs_get_io_geometry(struct btrfs_fs_info *fs_info, enum btrfs_map_op op,
-                       u64 logical, u64 len, struct btrfs_io_geometry *io_geom)
+int btrfs_get_io_geometry(struct btrfs_fs_info *fs_info, struct extent_map *em,
+                         enum btrfs_map_op op, u64 logical, u64 len,
+                         struct btrfs_io_geometry *io_geom)
 {
-       struct extent_map *em;
        struct map_lookup *map;
        u64 offset;
        u64 stripe_offset;
        u64 stripe_len;
        u64 raid56_full_stripe_start = (u64)-1;
        int data_stripes;
-       int ret = 0;
 
        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;
                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);
-               ret = -EINVAL;
-               goto out;
+               return -EINVAL;
        }
 
        /* stripe_offset is the offset of this block in its stripe */
        io_geom->stripe_offset = stripe_offset;
        io_geom->raid56_stripe_offset = raid56_full_stripe_start;
 
-out:
-       /* once for us */
-       free_extent_map(em);
-       return ret;
+       return 0;
 }
 
 static int __btrfs_map_block(struct btrfs_fs_info *fs_info,
        ASSERT(bbio_ret);
        ASSERT(op != BTRFS_MAP_DISCARD);
 
-       ret = btrfs_get_io_geometry(fs_info, op, logical, *length, &geom);
+       em = btrfs_get_chunk_map(fs_info, logical, *length);
+       ASSERT(!IS_ERR(em));
+
+       ret = btrfs_get_io_geometry(fs_info, em, op, logical, *length, &geom);
        if (ret < 0)
                return ret;
 
-       em = btrfs_get_chunk_map(fs_info, logical, *length);
-       ASSERT(!IS_ERR(em));
        map = em->map_lookup;
 
        *length = geom.len;
 
 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_io_geometry(struct btrfs_fs_info *fs_info, enum btrfs_map_op op,
-               u64 logical, u64 len, struct btrfs_io_geometry *io_geom);
+int btrfs_get_io_geometry(struct btrfs_fs_info *fs_info, struct extent_map *map,
+                         enum btrfs_map_op op, u64 logical, u64 len,
+                         struct btrfs_io_geometry *io_geom);
 int btrfs_read_sys_array(struct btrfs_fs_info *fs_info);
 int btrfs_read_chunk_tree(struct btrfs_fs_info *fs_info);
 int btrfs_alloc_chunk(struct btrfs_trans_handle *trans, u64 type);