1 /* SPDX-License-Identifier: GPL-2.0 */
2
3 #ifndef BTRFS_EXTENT_IO_H
4 #define BTRFS_EXTENT_IO_H
5
6 #include <linux/rbtree.h>
7 #include <linux/refcount.h>
8 #include <linux/fiemap.h>
9 #include <linux/btrfs_tree.h>
10 #include "ulist.h"
11
12 /*
13 * flags for bio submission. The high bits indicate the compression
14 * type for this bio
15 */
16 #define EXTENT_BIO_COMPRESSED 1
17 #define EXTENT_BIO_FLAG_SHIFT 16
18
19 enum {
20 EXTENT_BUFFER_UPTODATE,
21 EXTENT_BUFFER_DIRTY,
22 EXTENT_BUFFER_CORRUPT,
23 /* this got triggered by readahead */
24 EXTENT_BUFFER_READAHEAD,
25 EXTENT_BUFFER_TREE_REF,
26 EXTENT_BUFFER_STALE,
27 EXTENT_BUFFER_WRITEBACK,
28 /* read IO error */
29 EXTENT_BUFFER_READ_ERR,
30 EXTENT_BUFFER_UNMAPPED,
31 EXTENT_BUFFER_IN_TREE,
32 /* write IO error */
33 EXTENT_BUFFER_WRITE_ERR,
34 EXTENT_BUFFER_NO_CHECK,
35 };
36
37 /* these are flags for __process_pages_contig */
38 #define PAGE_UNLOCK (1 << 0)
39 /* Page starts writeback, clear dirty bit and set writeback bit */
40 #define PAGE_START_WRITEBACK (1 << 1)
41 #define PAGE_END_WRITEBACK (1 << 2)
42 #define PAGE_SET_ORDERED (1 << 3)
43 #define PAGE_SET_ERROR (1 << 4)
44 #define PAGE_LOCK (1 << 5)
45
46 /*
47 * page->private values. Every page that is controlled by the extent
48 * map has page->private set to one.
49 */
50 #define EXTENT_PAGE_PRIVATE 1
51
52 /*
53 * The extent buffer bitmap operations are done with byte granularity instead of
54 * word granularity for two reasons:
55 * 1. The bitmaps must be little-endian on disk.
56 * 2. Bitmap items are not guaranteed to be aligned to a word and therefore a
57 * single word in a bitmap may straddle two pages in the extent buffer.
58 */
59 #define BIT_BYTE(nr) ((nr) / BITS_PER_BYTE)
60 #define BYTE_MASK ((1 << BITS_PER_BYTE) - 1)
61 #define BITMAP_FIRST_BYTE_MASK(start) \
62 ((BYTE_MASK << ((start) & (BITS_PER_BYTE - 1))) & BYTE_MASK)
63 #define BITMAP_LAST_BYTE_MASK(nbits) \
64 (BYTE_MASK >> (-(nbits) & (BITS_PER_BYTE - 1)))
65
66 struct btrfs_root;
67 struct btrfs_inode;
68 struct btrfs_io_bio;
69 struct btrfs_fs_info;
70 struct io_failure_record;
71 struct extent_io_tree;
72
73 typedef blk_status_t (submit_bio_hook_t)(struct inode *inode, struct bio *bio,
74 int mirror_num,
75 unsigned long bio_flags);
76
77 typedef blk_status_t (extent_submit_bio_start_t)(struct inode *inode,
78 struct bio *bio, u64 dio_file_offset);
79
80 #define INLINE_EXTENT_BUFFER_PAGES (BTRFS_MAX_METADATA_BLOCKSIZE / PAGE_SIZE)
81 struct extent_buffer {
82 u64 start;
83 unsigned long len;
84 unsigned long bflags;
85 struct btrfs_fs_info *fs_info;
86 spinlock_t refs_lock;
87 atomic_t refs;
88 atomic_t io_pages;
89 int read_mirror;
90 struct rcu_head rcu_head;
91 pid_t lock_owner;
92 /* >= 0 if eb belongs to a log tree, -1 otherwise */
93 s8 log_index;
94
95 struct rw_semaphore lock;
96
97 struct page *pages[INLINE_EXTENT_BUFFER_PAGES];
98 struct list_head release_list;
99 #ifdef CONFIG_BTRFS_DEBUG
100 struct list_head leak_list;
101 #endif
102 };
103
104 /*
105 * Structure to record info about the bio being assembled, and other info like
106 * how many bytes are there before stripe/ordered extent boundary.
107 */
108 struct btrfs_bio_ctrl {
109 struct bio *bio;
110 unsigned long bio_flags;
111 u32 len_to_stripe_boundary;
112 u32 len_to_oe_boundary;
113 };
114
115 /*
116 * Structure to record how many bytes and which ranges are set/cleared
117 */
118 struct extent_changeset {
119 /* How many bytes are set/cleared in this operation */
120 u64 bytes_changed;
121
122 /* Changed ranges */
123 struct ulist range_changed;
124 };
125
extent_changeset_init(struct extent_changeset * changeset)126 static inline void extent_changeset_init(struct extent_changeset *changeset)
127 {
128 changeset->bytes_changed = 0;
129 ulist_init(&changeset->range_changed);
130 }
131
extent_changeset_alloc(void)132 static inline struct extent_changeset *extent_changeset_alloc(void)
133 {
134 struct extent_changeset *ret;
135
136 ret = kmalloc(sizeof(*ret), GFP_KERNEL);
137 if (!ret)
138 return NULL;
139
140 extent_changeset_init(ret);
141 return ret;
142 }
143
extent_changeset_release(struct extent_changeset * changeset)144 static inline void extent_changeset_release(struct extent_changeset *changeset)
145 {
146 if (!changeset)
147 return;
148 changeset->bytes_changed = 0;
149 ulist_release(&changeset->range_changed);
150 }
151
extent_changeset_free(struct extent_changeset * changeset)152 static inline void extent_changeset_free(struct extent_changeset *changeset)
153 {
154 if (!changeset)
155 return;
156 extent_changeset_release(changeset);
157 kfree(changeset);
158 }
159
extent_set_compress_type(unsigned long * bio_flags,int compress_type)160 static inline void extent_set_compress_type(unsigned long *bio_flags,
161 int compress_type)
162 {
163 *bio_flags |= compress_type << EXTENT_BIO_FLAG_SHIFT;
164 }
165
extent_compress_type(unsigned long bio_flags)166 static inline int extent_compress_type(unsigned long bio_flags)
167 {
168 return bio_flags >> EXTENT_BIO_FLAG_SHIFT;
169 }
170
171 struct extent_map_tree;
172
173 typedef struct extent_map *(get_extent_t)(struct btrfs_inode *inode,
174 struct page *page, size_t pg_offset,
175 u64 start, u64 len);
176
177 int try_release_extent_mapping(struct page *page, gfp_t mask);
178 int try_release_extent_buffer(struct page *page);
179
180 int __must_check submit_one_bio(struct bio *bio, int mirror_num,
181 unsigned long bio_flags);
182 int btrfs_do_readpage(struct page *page, struct extent_map **em_cached,
183 struct btrfs_bio_ctrl *bio_ctrl,
184 unsigned int read_flags, u64 *prev_em_start);
185 int extent_write_full_page(struct page *page, struct writeback_control *wbc);
186 int extent_write_locked_range(struct inode *inode, u64 start, u64 end,
187 int mode);
188 int extent_writepages(struct address_space *mapping,
189 struct writeback_control *wbc);
190 int btree_write_cache_pages(struct address_space *mapping,
191 struct writeback_control *wbc);
192 void extent_readahead(struct readahead_control *rac);
193 int extent_fiemap(struct btrfs_inode *inode, struct fiemap_extent_info *fieinfo,
194 u64 start, u64 len);
195 int set_page_extent_mapped(struct page *page);
196 void clear_page_extent_mapped(struct page *page);
197
198 struct extent_buffer *alloc_extent_buffer(struct btrfs_fs_info *fs_info,
199 u64 start, u64 owner_root, int level);
200 struct extent_buffer *__alloc_dummy_extent_buffer(struct btrfs_fs_info *fs_info,
201 u64 start, unsigned long len);
202 struct extent_buffer *alloc_dummy_extent_buffer(struct btrfs_fs_info *fs_info,
203 u64 start);
204 struct extent_buffer *btrfs_clone_extent_buffer(const struct extent_buffer *src);
205 struct extent_buffer *find_extent_buffer(struct btrfs_fs_info *fs_info,
206 u64 start);
207 void free_extent_buffer(struct extent_buffer *eb);
208 void free_extent_buffer_stale(struct extent_buffer *eb);
209 #define WAIT_NONE 0
210 #define WAIT_COMPLETE 1
211 #define WAIT_PAGE_LOCK 2
212 int read_extent_buffer_pages(struct extent_buffer *eb, int wait,
213 int mirror_num);
214 void wait_on_extent_buffer_writeback(struct extent_buffer *eb);
215 void btrfs_readahead_tree_block(struct btrfs_fs_info *fs_info,
216 u64 bytenr, u64 owner_root, u64 gen, int level);
217 void btrfs_readahead_node_child(struct extent_buffer *node, int slot);
218
num_extent_pages(const struct extent_buffer * eb)219 static inline int num_extent_pages(const struct extent_buffer *eb)
220 {
221 /*
222 * For sectorsize == PAGE_SIZE case, since nodesize is always aligned to
223 * sectorsize, it's just eb->len >> PAGE_SHIFT.
224 *
225 * For sectorsize < PAGE_SIZE case, we could have nodesize < PAGE_SIZE,
226 * thus have to ensure we get at least one page.
227 */
228 return (eb->len >> PAGE_SHIFT) ?: 1;
229 }
230
extent_buffer_uptodate(const struct extent_buffer * eb)231 static inline int extent_buffer_uptodate(const struct extent_buffer *eb)
232 {
233 return test_bit(EXTENT_BUFFER_UPTODATE, &eb->bflags);
234 }
235
236 int memcmp_extent_buffer(const struct extent_buffer *eb, const void *ptrv,
237 unsigned long start, unsigned long len);
238 void read_extent_buffer(const struct extent_buffer *eb, void *dst,
239 unsigned long start,
240 unsigned long len);
241 int read_extent_buffer_to_user_nofault(const struct extent_buffer *eb,
242 void __user *dst, unsigned long start,
243 unsigned long len);
244 void write_extent_buffer_fsid(const struct extent_buffer *eb, const void *src);
245 void write_extent_buffer_chunk_tree_uuid(const struct extent_buffer *eb,
246 const void *src);
247 void write_extent_buffer(const struct extent_buffer *eb, const void *src,
248 unsigned long start, unsigned long len);
249 void copy_extent_buffer_full(const struct extent_buffer *dst,
250 const struct extent_buffer *src);
251 void copy_extent_buffer(const struct extent_buffer *dst,
252 const struct extent_buffer *src,
253 unsigned long dst_offset, unsigned long src_offset,
254 unsigned long len);
255 void memcpy_extent_buffer(const struct extent_buffer *dst,
256 unsigned long dst_offset, unsigned long src_offset,
257 unsigned long len);
258 void memmove_extent_buffer(const struct extent_buffer *dst,
259 unsigned long dst_offset, unsigned long src_offset,
260 unsigned long len);
261 void memzero_extent_buffer(const struct extent_buffer *eb, unsigned long start,
262 unsigned long len);
263 int extent_buffer_test_bit(const struct extent_buffer *eb, unsigned long start,
264 unsigned long pos);
265 void extent_buffer_bitmap_set(const struct extent_buffer *eb, unsigned long start,
266 unsigned long pos, unsigned long len);
267 void extent_buffer_bitmap_clear(const struct extent_buffer *eb,
268 unsigned long start, unsigned long pos,
269 unsigned long len);
270 void clear_extent_buffer_dirty(const struct extent_buffer *eb);
271 bool set_extent_buffer_dirty(struct extent_buffer *eb);
272 void set_extent_buffer_uptodate(struct extent_buffer *eb);
273 void clear_extent_buffer_uptodate(struct extent_buffer *eb);
274 int extent_buffer_under_io(const struct extent_buffer *eb);
275 void extent_range_clear_dirty_for_io(struct inode *inode, u64 start, u64 end);
276 void extent_range_redirty_for_io(struct inode *inode, u64 start, u64 end);
277 void extent_clear_unlock_delalloc(struct btrfs_inode *inode, u64 start, u64 end,
278 struct page *locked_page,
279 u32 bits_to_clear, unsigned long page_ops);
280 struct bio *btrfs_bio_alloc(u64 first_byte);
281 struct bio *btrfs_io_bio_alloc(unsigned int nr_iovecs);
282 struct bio *btrfs_bio_clone(struct bio *bio);
283 struct bio *btrfs_bio_clone_partial(struct bio *orig, u64 offset, u64 size);
284
285 int repair_io_failure(struct btrfs_fs_info *fs_info, u64 ino, u64 start,
286 u64 length, u64 logical, struct page *page,
287 unsigned int pg_offset, int mirror_num);
288 void end_extent_writepage(struct page *page, int err, u64 start, u64 end);
289 int btrfs_repair_eb_io_failure(const struct extent_buffer *eb, int mirror_num);
290
291 /*
292 * When IO fails, either with EIO or csum verification fails, we
293 * try other mirrors that might have a good copy of the data. This
294 * io_failure_record is used to record state as we go through all the
295 * mirrors. If another mirror has good data, the sector is set up to date
296 * and things continue. If a good mirror can't be found, the original
297 * bio end_io callback is called to indicate things have failed.
298 */
299 struct io_failure_record {
300 struct page *page;
301 u64 start;
302 u64 len;
303 u64 logical;
304 unsigned long bio_flags;
305 int this_mirror;
306 int failed_mirror;
307 };
308
309 int btrfs_repair_one_sector(struct inode *inode,
310 struct bio *failed_bio, u32 bio_offset,
311 struct page *page, unsigned int pgoff,
312 u64 start, int failed_mirror,
313 submit_bio_hook_t *submit_bio_hook);
314
315 #ifdef CONFIG_BTRFS_FS_RUN_SANITY_TESTS
316 bool find_lock_delalloc_range(struct inode *inode,
317 struct page *locked_page, u64 *start,
318 u64 *end);
319 #endif
320 struct extent_buffer *alloc_test_extent_buffer(struct btrfs_fs_info *fs_info,
321 u64 start);
322
323 #ifdef CONFIG_BTRFS_DEBUG
324 void btrfs_extent_buffer_leak_debug_check(struct btrfs_fs_info *fs_info);
325 #else
326 #define btrfs_extent_buffer_leak_debug_check(fs_info) do {} while (0)
327 #endif
328
329 #endif
330