| 1 | #ifndef __EXTENTIO__ |
| 2 | #define __EXTENTIO__ |
| 3 | |
| 4 | #include <linux/rbtree.h> |
| 5 | |
| 6 | /* bits for the extent state */ |
| 7 | #define EXTENT_DIRTY 1 |
| 8 | #define EXTENT_WRITEBACK (1 << 1) |
| 9 | #define EXTENT_UPTODATE (1 << 2) |
| 10 | #define EXTENT_LOCKED (1 << 3) |
| 11 | #define EXTENT_NEW (1 << 4) |
| 12 | #define EXTENT_DELALLOC (1 << 5) |
| 13 | #define EXTENT_DEFRAG (1 << 6) |
| 14 | #define EXTENT_DEFRAG_DONE (1 << 7) |
| 15 | #define EXTENT_BUFFER_FILLED (1 << 8) |
| 16 | #define EXTENT_BOUNDARY (1 << 9) |
| 17 | #define EXTENT_NODATASUM (1 << 10) |
| 18 | #define EXTENT_DO_ACCOUNTING (1 << 11) |
| 19 | #define EXTENT_FIRST_DELALLOC (1 << 12) |
| 20 | #define EXTENT_NEED_WAIT (1 << 13) |
| 21 | #define EXTENT_DAMAGED (1 << 14) |
| 22 | #define EXTENT_NORESERVE (1 << 15) |
| 23 | #define EXTENT_IOBITS (EXTENT_LOCKED | EXTENT_WRITEBACK) |
| 24 | #define EXTENT_CTLBITS (EXTENT_DO_ACCOUNTING | EXTENT_FIRST_DELALLOC) |
| 25 | |
| 26 | /* |
| 27 | * flags for bio submission. The high bits indicate the compression |
| 28 | * type for this bio |
| 29 | */ |
| 30 | #define EXTENT_BIO_COMPRESSED 1 |
| 31 | #define EXTENT_BIO_TREE_LOG 2 |
| 32 | #define EXTENT_BIO_FLAG_SHIFT 16 |
| 33 | |
| 34 | /* these are bit numbers for test/set bit */ |
| 35 | #define EXTENT_BUFFER_UPTODATE 0 |
| 36 | #define EXTENT_BUFFER_BLOCKING 1 |
| 37 | #define EXTENT_BUFFER_DIRTY 2 |
| 38 | #define EXTENT_BUFFER_CORRUPT 3 |
| 39 | #define EXTENT_BUFFER_READAHEAD 4 /* this got triggered by readahead */ |
| 40 | #define EXTENT_BUFFER_TREE_REF 5 |
| 41 | #define EXTENT_BUFFER_STALE 6 |
| 42 | #define EXTENT_BUFFER_WRITEBACK 7 |
| 43 | #define EXTENT_BUFFER_IOERR 8 |
| 44 | #define EXTENT_BUFFER_DUMMY 9 |
| 45 | |
| 46 | /* these are flags for extent_clear_unlock_delalloc */ |
| 47 | #define PAGE_UNLOCK (1 << 0) |
| 48 | #define PAGE_CLEAR_DIRTY (1 << 1) |
| 49 | #define PAGE_SET_WRITEBACK (1 << 2) |
| 50 | #define PAGE_END_WRITEBACK (1 << 3) |
| 51 | #define PAGE_SET_PRIVATE2 (1 << 4) |
| 52 | |
| 53 | /* |
| 54 | * page->private values. Every page that is controlled by the extent |
| 55 | * map has page->private set to one. |
| 56 | */ |
| 57 | #define EXTENT_PAGE_PRIVATE 1 |
| 58 | #define EXTENT_PAGE_PRIVATE_FIRST_PAGE 3 |
| 59 | |
| 60 | struct extent_state; |
| 61 | struct btrfs_root; |
| 62 | struct btrfs_io_bio; |
| 63 | |
| 64 | typedef int (extent_submit_bio_hook_t)(struct inode *inode, int rw, |
| 65 | struct bio *bio, int mirror_num, |
| 66 | unsigned long bio_flags, u64 bio_offset); |
| 67 | struct extent_io_ops { |
| 68 | int (*fill_delalloc)(struct inode *inode, struct page *locked_page, |
| 69 | u64 start, u64 end, int *page_started, |
| 70 | unsigned long *nr_written); |
| 71 | int (*writepage_start_hook)(struct page *page, u64 start, u64 end); |
| 72 | int (*writepage_io_hook)(struct page *page, u64 start, u64 end); |
| 73 | extent_submit_bio_hook_t *submit_bio_hook; |
| 74 | int (*merge_bio_hook)(int rw, struct page *page, unsigned long offset, |
| 75 | size_t size, struct bio *bio, |
| 76 | unsigned long bio_flags); |
| 77 | int (*readpage_io_failed_hook)(struct page *page, int failed_mirror); |
| 78 | int (*readpage_end_io_hook)(struct btrfs_io_bio *io_bio, u64 phy_offset, |
| 79 | struct page *page, u64 start, u64 end, |
| 80 | int mirror); |
| 81 | int (*writepage_end_io_hook)(struct page *page, u64 start, u64 end, |
| 82 | struct extent_state *state, int uptodate); |
| 83 | void (*set_bit_hook)(struct inode *inode, struct extent_state *state, |
| 84 | unsigned long *bits); |
| 85 | void (*clear_bit_hook)(struct inode *inode, struct extent_state *state, |
| 86 | unsigned long *bits); |
| 87 | void (*merge_extent_hook)(struct inode *inode, |
| 88 | struct extent_state *new, |
| 89 | struct extent_state *other); |
| 90 | void (*split_extent_hook)(struct inode *inode, |
| 91 | struct extent_state *orig, u64 split); |
| 92 | }; |
| 93 | |
| 94 | struct extent_io_tree { |
| 95 | struct rb_root state; |
| 96 | struct radix_tree_root buffer; |
| 97 | struct address_space *mapping; |
| 98 | u64 dirty_bytes; |
| 99 | int track_uptodate; |
| 100 | spinlock_t lock; |
| 101 | spinlock_t buffer_lock; |
| 102 | struct extent_io_ops *ops; |
| 103 | }; |
| 104 | |
| 105 | struct extent_state { |
| 106 | u64 start; |
| 107 | u64 end; /* inclusive */ |
| 108 | struct rb_node rb_node; |
| 109 | |
| 110 | /* ADD NEW ELEMENTS AFTER THIS */ |
| 111 | struct extent_io_tree *tree; |
| 112 | wait_queue_head_t wq; |
| 113 | atomic_t refs; |
| 114 | unsigned long state; |
| 115 | |
| 116 | /* for use by the FS */ |
| 117 | u64 private; |
| 118 | |
| 119 | #ifdef CONFIG_BTRFS_DEBUG |
| 120 | struct list_head leak_list; |
| 121 | #endif |
| 122 | }; |
| 123 | |
| 124 | #define INLINE_EXTENT_BUFFER_PAGES 16 |
| 125 | #define MAX_INLINE_EXTENT_BUFFER_SIZE (INLINE_EXTENT_BUFFER_PAGES * PAGE_CACHE_SIZE) |
| 126 | struct extent_buffer { |
| 127 | u64 start; |
| 128 | unsigned long len; |
| 129 | unsigned long map_start; |
| 130 | unsigned long map_len; |
| 131 | unsigned long bflags; |
| 132 | struct extent_io_tree *tree; |
| 133 | spinlock_t refs_lock; |
| 134 | atomic_t refs; |
| 135 | atomic_t io_pages; |
| 136 | int read_mirror; |
| 137 | struct rcu_head rcu_head; |
| 138 | pid_t lock_owner; |
| 139 | |
| 140 | /* count of read lock holders on the extent buffer */ |
| 141 | atomic_t write_locks; |
| 142 | atomic_t read_locks; |
| 143 | atomic_t blocking_writers; |
| 144 | atomic_t blocking_readers; |
| 145 | atomic_t spinning_readers; |
| 146 | atomic_t spinning_writers; |
| 147 | int lock_nested; |
| 148 | |
| 149 | /* protects write locks */ |
| 150 | rwlock_t lock; |
| 151 | |
| 152 | /* readers use lock_wq while they wait for the write |
| 153 | * lock holders to unlock |
| 154 | */ |
| 155 | wait_queue_head_t write_lock_wq; |
| 156 | |
| 157 | /* writers use read_lock_wq while they wait for readers |
| 158 | * to unlock |
| 159 | */ |
| 160 | wait_queue_head_t read_lock_wq; |
| 161 | wait_queue_head_t lock_wq; |
| 162 | struct page *pages[INLINE_EXTENT_BUFFER_PAGES]; |
| 163 | #ifdef CONFIG_BTRFS_DEBUG |
| 164 | struct list_head leak_list; |
| 165 | #endif |
| 166 | }; |
| 167 | |
| 168 | static inline void extent_set_compress_type(unsigned long *bio_flags, |
| 169 | int compress_type) |
| 170 | { |
| 171 | *bio_flags |= compress_type << EXTENT_BIO_FLAG_SHIFT; |
| 172 | } |
| 173 | |
| 174 | static inline int extent_compress_type(unsigned long bio_flags) |
| 175 | { |
| 176 | return bio_flags >> EXTENT_BIO_FLAG_SHIFT; |
| 177 | } |
| 178 | |
| 179 | struct extent_map_tree; |
| 180 | |
| 181 | typedef struct extent_map *(get_extent_t)(struct inode *inode, |
| 182 | struct page *page, |
| 183 | size_t pg_offset, |
| 184 | u64 start, u64 len, |
| 185 | int create); |
| 186 | |
| 187 | void extent_io_tree_init(struct extent_io_tree *tree, |
| 188 | struct address_space *mapping); |
| 189 | int try_release_extent_mapping(struct extent_map_tree *map, |
| 190 | struct extent_io_tree *tree, struct page *page, |
| 191 | gfp_t mask); |
| 192 | int try_release_extent_buffer(struct page *page); |
| 193 | int lock_extent(struct extent_io_tree *tree, u64 start, u64 end); |
| 194 | int lock_extent_bits(struct extent_io_tree *tree, u64 start, u64 end, |
| 195 | unsigned long bits, struct extent_state **cached); |
| 196 | int unlock_extent(struct extent_io_tree *tree, u64 start, u64 end); |
| 197 | int unlock_extent_cached(struct extent_io_tree *tree, u64 start, u64 end, |
| 198 | struct extent_state **cached, gfp_t mask); |
| 199 | int try_lock_extent(struct extent_io_tree *tree, u64 start, u64 end); |
| 200 | int extent_read_full_page(struct extent_io_tree *tree, struct page *page, |
| 201 | get_extent_t *get_extent, int mirror_num); |
| 202 | int __init extent_io_init(void); |
| 203 | void extent_io_exit(void); |
| 204 | |
| 205 | u64 count_range_bits(struct extent_io_tree *tree, |
| 206 | u64 *start, u64 search_end, |
| 207 | u64 max_bytes, unsigned long bits, int contig); |
| 208 | |
| 209 | void free_extent_state(struct extent_state *state); |
| 210 | int test_range_bit(struct extent_io_tree *tree, u64 start, u64 end, |
| 211 | unsigned long bits, int filled, |
| 212 | struct extent_state *cached_state); |
| 213 | int clear_extent_bits(struct extent_io_tree *tree, u64 start, u64 end, |
| 214 | unsigned long bits, gfp_t mask); |
| 215 | int clear_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, |
| 216 | unsigned long bits, int wake, int delete, |
| 217 | struct extent_state **cached, gfp_t mask); |
| 218 | int set_extent_bits(struct extent_io_tree *tree, u64 start, u64 end, |
| 219 | unsigned long bits, gfp_t mask); |
| 220 | int set_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, |
| 221 | unsigned long bits, u64 *failed_start, |
| 222 | struct extent_state **cached_state, gfp_t mask); |
| 223 | int set_extent_uptodate(struct extent_io_tree *tree, u64 start, u64 end, |
| 224 | struct extent_state **cached_state, gfp_t mask); |
| 225 | int clear_extent_uptodate(struct extent_io_tree *tree, u64 start, u64 end, |
| 226 | struct extent_state **cached_state, gfp_t mask); |
| 227 | int set_extent_new(struct extent_io_tree *tree, u64 start, u64 end, |
| 228 | gfp_t mask); |
| 229 | int set_extent_dirty(struct extent_io_tree *tree, u64 start, u64 end, |
| 230 | gfp_t mask); |
| 231 | int clear_extent_dirty(struct extent_io_tree *tree, u64 start, u64 end, |
| 232 | gfp_t mask); |
| 233 | int convert_extent_bit(struct extent_io_tree *tree, u64 start, u64 end, |
| 234 | unsigned long bits, unsigned long clear_bits, |
| 235 | struct extent_state **cached_state, gfp_t mask); |
| 236 | int set_extent_delalloc(struct extent_io_tree *tree, u64 start, u64 end, |
| 237 | struct extent_state **cached_state, gfp_t mask); |
| 238 | int set_extent_defrag(struct extent_io_tree *tree, u64 start, u64 end, |
| 239 | struct extent_state **cached_state, gfp_t mask); |
| 240 | int find_first_extent_bit(struct extent_io_tree *tree, u64 start, |
| 241 | u64 *start_ret, u64 *end_ret, unsigned long bits, |
| 242 | struct extent_state **cached_state); |
| 243 | int extent_invalidatepage(struct extent_io_tree *tree, |
| 244 | struct page *page, unsigned long offset); |
| 245 | int extent_write_full_page(struct extent_io_tree *tree, struct page *page, |
| 246 | get_extent_t *get_extent, |
| 247 | struct writeback_control *wbc); |
| 248 | int extent_write_locked_range(struct extent_io_tree *tree, struct inode *inode, |
| 249 | u64 start, u64 end, get_extent_t *get_extent, |
| 250 | int mode); |
| 251 | int extent_writepages(struct extent_io_tree *tree, |
| 252 | struct address_space *mapping, |
| 253 | get_extent_t *get_extent, |
| 254 | struct writeback_control *wbc); |
| 255 | int btree_write_cache_pages(struct address_space *mapping, |
| 256 | struct writeback_control *wbc); |
| 257 | int extent_readpages(struct extent_io_tree *tree, |
| 258 | struct address_space *mapping, |
| 259 | struct list_head *pages, unsigned nr_pages, |
| 260 | get_extent_t get_extent); |
| 261 | int extent_fiemap(struct inode *inode, struct fiemap_extent_info *fieinfo, |
| 262 | __u64 start, __u64 len, get_extent_t *get_extent); |
| 263 | int set_state_private(struct extent_io_tree *tree, u64 start, u64 private); |
| 264 | int get_state_private(struct extent_io_tree *tree, u64 start, u64 *private); |
| 265 | void set_page_extent_mapped(struct page *page); |
| 266 | |
| 267 | struct extent_buffer *alloc_extent_buffer(struct extent_io_tree *tree, |
| 268 | u64 start, unsigned long len); |
| 269 | struct extent_buffer *alloc_dummy_extent_buffer(u64 start, unsigned long len); |
| 270 | struct extent_buffer *btrfs_clone_extent_buffer(struct extent_buffer *src); |
| 271 | struct extent_buffer *find_extent_buffer(struct extent_io_tree *tree, |
| 272 | u64 start, unsigned long len); |
| 273 | void free_extent_buffer(struct extent_buffer *eb); |
| 274 | void free_extent_buffer_stale(struct extent_buffer *eb); |
| 275 | #define WAIT_NONE 0 |
| 276 | #define WAIT_COMPLETE 1 |
| 277 | #define WAIT_PAGE_LOCK 2 |
| 278 | int read_extent_buffer_pages(struct extent_io_tree *tree, |
| 279 | struct extent_buffer *eb, u64 start, int wait, |
| 280 | get_extent_t *get_extent, int mirror_num); |
| 281 | void wait_on_extent_buffer_writeback(struct extent_buffer *eb); |
| 282 | |
| 283 | static inline unsigned long num_extent_pages(u64 start, u64 len) |
| 284 | { |
| 285 | return ((start + len + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT) - |
| 286 | (start >> PAGE_CACHE_SHIFT); |
| 287 | } |
| 288 | |
| 289 | static inline struct page *extent_buffer_page(struct extent_buffer *eb, |
| 290 | unsigned long i) |
| 291 | { |
| 292 | return eb->pages[i]; |
| 293 | } |
| 294 | |
| 295 | static inline void extent_buffer_get(struct extent_buffer *eb) |
| 296 | { |
| 297 | atomic_inc(&eb->refs); |
| 298 | } |
| 299 | |
| 300 | int memcmp_extent_buffer(struct extent_buffer *eb, const void *ptrv, |
| 301 | unsigned long start, |
| 302 | unsigned long len); |
| 303 | void read_extent_buffer(struct extent_buffer *eb, void *dst, |
| 304 | unsigned long start, |
| 305 | unsigned long len); |
| 306 | void write_extent_buffer(struct extent_buffer *eb, const void *src, |
| 307 | unsigned long start, unsigned long len); |
| 308 | void copy_extent_buffer(struct extent_buffer *dst, struct extent_buffer *src, |
| 309 | unsigned long dst_offset, unsigned long src_offset, |
| 310 | unsigned long len); |
| 311 | void memcpy_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset, |
| 312 | unsigned long src_offset, unsigned long len); |
| 313 | void memmove_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset, |
| 314 | unsigned long src_offset, unsigned long len); |
| 315 | void memset_extent_buffer(struct extent_buffer *eb, char c, |
| 316 | unsigned long start, unsigned long len); |
| 317 | void clear_extent_buffer_dirty(struct extent_buffer *eb); |
| 318 | int set_extent_buffer_dirty(struct extent_buffer *eb); |
| 319 | int set_extent_buffer_uptodate(struct extent_buffer *eb); |
| 320 | int clear_extent_buffer_uptodate(struct extent_buffer *eb); |
| 321 | int extent_buffer_uptodate(struct extent_buffer *eb); |
| 322 | int map_private_extent_buffer(struct extent_buffer *eb, unsigned long offset, |
| 323 | unsigned long min_len, char **map, |
| 324 | unsigned long *map_start, |
| 325 | unsigned long *map_len); |
| 326 | int extent_range_clear_dirty_for_io(struct inode *inode, u64 start, u64 end); |
| 327 | int extent_range_redirty_for_io(struct inode *inode, u64 start, u64 end); |
| 328 | int extent_clear_unlock_delalloc(struct inode *inode, u64 start, u64 end, |
| 329 | struct page *locked_page, |
| 330 | unsigned long bits_to_clear, |
| 331 | unsigned long page_ops); |
| 332 | struct bio * |
| 333 | btrfs_bio_alloc(struct block_device *bdev, u64 first_sector, int nr_vecs, |
| 334 | gfp_t gfp_flags); |
| 335 | struct bio *btrfs_io_bio_alloc(gfp_t gfp_mask, unsigned int nr_iovecs); |
| 336 | struct bio *btrfs_bio_clone(struct bio *bio, gfp_t gfp_mask); |
| 337 | |
| 338 | struct btrfs_fs_info; |
| 339 | |
| 340 | int repair_io_failure(struct btrfs_fs_info *fs_info, u64 start, |
| 341 | u64 length, u64 logical, struct page *page, |
| 342 | int mirror_num); |
| 343 | int end_extent_writepage(struct page *page, int err, u64 start, u64 end); |
| 344 | int repair_eb_io_failure(struct btrfs_root *root, struct extent_buffer *eb, |
| 345 | int mirror_num); |
| 346 | #endif |