]> Pileus Git - ~andy/linux/blob - fs/btrfs/extent_map.h
Btrfs: Fix extent_buffer and extent_state leaks
[~andy/linux] / fs / btrfs / extent_map.h
1 #ifndef __EXTENTMAP__
2 #define __EXTENTMAP__
3
4 #include <linux/rbtree.h>
5
6 #define EXTENT_MAP_HOLE (u64)-3
7 #define EXTENT_MAP_INLINE (u64)-2
8 #define EXTENT_MAP_DELALLOC (u64)-1
9
10 /* bits for the extent state */
11 #define EXTENT_DIRTY 1
12 #define EXTENT_WRITEBACK (1 << 1)
13 #define EXTENT_UPTODATE (1 << 2)
14 #define EXTENT_LOCKED (1 << 3)
15 #define EXTENT_NEW (1 << 4)
16 #define EXTENT_DELALLOC (1 << 5)
17 #define EXTENT_IOBITS (EXTENT_LOCKED | EXTENT_WRITEBACK)
18
19
20 struct extent_map_ops {
21         int (*fill_delalloc)(struct inode *inode, u64 start, u64 end);
22         int (*writepage_io_hook)(struct page *page, u64 start, u64 end);
23         int (*readpage_io_hook)(struct page *page, u64 start, u64 end);
24         int (*readpage_end_io_hook)(struct page *page, u64 start, u64 end);
25         void (*writepage_end_io_hook)(struct page *page, u64 start, u64 end);
26 };
27
28 struct extent_map_tree {
29         struct rb_root map;
30         struct rb_root state;
31         struct address_space *mapping;
32         rwlock_t lock;
33         struct extent_map_ops *ops;
34 };
35
36 /* note, this must start with the same fields as fs/extent_map.c:tree_entry */
37 struct extent_map {
38         u64 start;
39         u64 end; /* inclusive */
40         int in_tree;
41         struct rb_node rb_node;
42         /* block_start and block_end are in bytes */
43         u64 block_start;
44         u64 block_end; /* inclusive */
45         struct block_device *bdev;
46         atomic_t refs;
47 };
48
49 /* note, this must start with the same fields as fs/extent_map.c:tree_entry */
50 struct extent_state {
51         u64 start;
52         u64 end; /* inclusive */
53         int in_tree;
54         struct rb_node rb_node;
55         wait_queue_head_t wq;
56         atomic_t refs;
57         unsigned long state;
58
59         /* for use by the FS */
60         u64 private;
61
62         struct list_head list;
63 };
64
65 struct extent_buffer {
66         u64 start;
67         unsigned long len;
68         atomic_t refs;
69         int flags;
70         struct list_head list;
71         struct list_head leak_list;
72         struct page *first_page;
73         unsigned long alloc_addr;
74 };
75
76 typedef struct extent_map *(get_extent_t)(struct inode *inode,
77                                           struct page *page,
78                                           size_t page_offset,
79                                           u64 start, u64 end,
80                                           int create);
81
82 void extent_map_tree_init(struct extent_map_tree *tree,
83                           struct address_space *mapping, gfp_t mask);
84 struct extent_map *lookup_extent_mapping(struct extent_map_tree *tree,
85                                          u64 start, u64 end);
86 int add_extent_mapping(struct extent_map_tree *tree,
87                        struct extent_map *em);
88 int remove_extent_mapping(struct extent_map_tree *tree, struct extent_map *em);
89 int try_release_extent_mapping(struct extent_map_tree *tree, struct page *page);
90 int lock_extent(struct extent_map_tree *tree, u64 start, u64 end, gfp_t mask);
91 int unlock_extent(struct extent_map_tree *tree, u64 start, u64 end, gfp_t mask);
92 struct extent_map *alloc_extent_map(gfp_t mask);
93 void free_extent_map(struct extent_map *em);
94 int extent_read_full_page(struct extent_map_tree *tree, struct page *page,
95                           get_extent_t *get_extent);
96 void __init extent_map_init(void);
97 void __exit extent_map_exit(void);
98 int extent_clean_all_trees(struct extent_map_tree *tree);
99 int set_extent_uptodate(struct extent_map_tree *tree, u64 start, u64 end,
100                         gfp_t mask);
101 int set_extent_new(struct extent_map_tree *tree, u64 start, u64 end,
102                    gfp_t mask);
103 int set_extent_dirty(struct extent_map_tree *tree, u64 start, u64 end,
104                      gfp_t mask);
105 int clear_extent_dirty(struct extent_map_tree *tree, u64 start, u64 end,
106                        gfp_t mask);
107 int set_extent_delalloc(struct extent_map_tree *tree, u64 start, u64 end,
108                      gfp_t mask);
109 int find_first_extent_bit(struct extent_map_tree *tree, u64 start,
110                           u64 *start_ret, u64 *end_ret, int bits);
111 int extent_invalidatepage(struct extent_map_tree *tree,
112                           struct page *page, unsigned long offset);
113 int extent_write_full_page(struct extent_map_tree *tree, struct page *page,
114                           get_extent_t *get_extent,
115                           struct writeback_control *wbc);
116 int extent_prepare_write(struct extent_map_tree *tree,
117                          struct inode *inode, struct page *page,
118                          unsigned from, unsigned to, get_extent_t *get_extent);
119 int extent_commit_write(struct extent_map_tree *tree,
120                         struct inode *inode, struct page *page,
121                         unsigned from, unsigned to);
122 sector_t extent_bmap(struct address_space *mapping, sector_t iblock,
123                 get_extent_t *get_extent);
124 int set_range_dirty(struct extent_map_tree *tree, u64 start, u64 end);
125 int set_state_private(struct extent_map_tree *tree, u64 start, u64 private);
126 int get_state_private(struct extent_map_tree *tree, u64 start, u64 *private);
127 void set_page_extent_mapped(struct page *page);
128
129 struct extent_buffer *alloc_extent_buffer(struct extent_map_tree *tree,
130                                           u64 start, unsigned long len,
131                                           gfp_t mask);
132 struct extent_buffer *find_extent_buffer(struct extent_map_tree *tree,
133                                          u64 start, unsigned long len,
134                                           gfp_t mask);
135 void free_extent_buffer(struct extent_buffer *eb);
136 int read_extent_buffer_pages(struct extent_map_tree *tree,
137                              struct extent_buffer *eb, int wait);
138
139 static inline void extent_buffer_get(struct extent_buffer *eb)
140 {
141         atomic_inc(&eb->refs);
142 }
143
144 static inline u64 extent_buffer_blocknr(struct extent_buffer *eb)
145 {
146         return eb->start / 4096;
147 }
148
149 int memcmp_extent_buffer(struct extent_buffer *eb, const void *ptrv,
150                           unsigned long start,
151                           unsigned long len);
152 void read_extent_buffer(struct extent_buffer *eb, void *dst,
153                         unsigned long start,
154                         unsigned long len);
155 void write_extent_buffer(struct extent_buffer *eb, const void *src,
156                          unsigned long start, unsigned long len);
157 void copy_extent_buffer(struct extent_buffer *dst, struct extent_buffer *src,
158                         unsigned long dst_offset, unsigned long src_offset,
159                         unsigned long len);
160 void memcpy_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset,
161                            unsigned long src_offset, unsigned long len);
162 void memmove_extent_buffer(struct extent_buffer *dst, unsigned long dst_offset,
163                            unsigned long src_offset, unsigned long len);
164 void memset_extent_buffer(struct extent_buffer *eb, char c,
165                           unsigned long start, unsigned long len);
166 int wait_on_extent_buffer_writeback(struct extent_map_tree *tree,
167                                     struct extent_buffer *eb);
168 int clear_extent_buffer_dirty(struct extent_map_tree *tree,
169                               struct extent_buffer *eb);
170 int set_extent_buffer_dirty(struct extent_map_tree *tree,
171                              struct extent_buffer *eb);
172 int set_extent_buffer_uptodate(struct extent_map_tree *tree,
173                                struct extent_buffer *eb);
174 int extent_buffer_uptodate(struct extent_map_tree *tree,
175                            struct extent_buffer *eb);
176 int map_extent_buffer(struct extent_buffer *eb, unsigned long offset,
177                       unsigned long min_len, char **token, char **map,
178                       unsigned long *map_start,
179                       unsigned long *map_len, int km);
180 void unmap_extent_buffer(struct extent_buffer *eb, char *token, int km);
181 #endif