xref: /openbmc/linux/fs/ext4/mballoc.c (revision 648f5879)
1c9de560dSAlex Tomas /*
2c9de560dSAlex Tomas  * Copyright (c) 2003-2006, Cluster File Systems, Inc, info@clusterfs.com
3c9de560dSAlex Tomas  * Written by Alex Tomas <alex@clusterfs.com>
4c9de560dSAlex Tomas  *
5c9de560dSAlex Tomas  * This program is free software; you can redistribute it and/or modify
6c9de560dSAlex Tomas  * it under the terms of the GNU General Public License version 2 as
7c9de560dSAlex Tomas  * published by the Free Software Foundation.
8c9de560dSAlex Tomas  *
9c9de560dSAlex Tomas  * This program is distributed in the hope that it will be useful,
10c9de560dSAlex Tomas  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11c9de560dSAlex Tomas  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12c9de560dSAlex Tomas  * GNU General Public License for more details.
13c9de560dSAlex Tomas  *
14c9de560dSAlex Tomas  * You should have received a copy of the GNU General Public Licens
15c9de560dSAlex Tomas  * along with this program; if not, write to the Free Software
16c9de560dSAlex Tomas  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-
17c9de560dSAlex Tomas  */
18c9de560dSAlex Tomas 
19c9de560dSAlex Tomas 
20c9de560dSAlex Tomas /*
21c9de560dSAlex Tomas  * mballoc.c contains the multiblocks allocation routines
22c9de560dSAlex Tomas  */
23c9de560dSAlex Tomas 
248f6e39a7SMingming Cao #include "mballoc.h"
25c9de560dSAlex Tomas /*
26c9de560dSAlex Tomas  * MUSTDO:
27c9de560dSAlex Tomas  *   - test ext4_ext_search_left() and ext4_ext_search_right()
28c9de560dSAlex Tomas  *   - search for metadata in few groups
29c9de560dSAlex Tomas  *
30c9de560dSAlex Tomas  * TODO v4:
31c9de560dSAlex Tomas  *   - normalization should take into account whether file is still open
32c9de560dSAlex Tomas  *   - discard preallocations if no free space left (policy?)
33c9de560dSAlex Tomas  *   - don't normalize tails
34c9de560dSAlex Tomas  *   - quota
35c9de560dSAlex Tomas  *   - reservation for superuser
36c9de560dSAlex Tomas  *
37c9de560dSAlex Tomas  * TODO v3:
38c9de560dSAlex Tomas  *   - bitmap read-ahead (proposed by Oleg Drokin aka green)
39c9de560dSAlex Tomas  *   - track min/max extents in each group for better group selection
40c9de560dSAlex Tomas  *   - mb_mark_used() may allocate chunk right after splitting buddy
41c9de560dSAlex Tomas  *   - tree of groups sorted by number of free blocks
42c9de560dSAlex Tomas  *   - error handling
43c9de560dSAlex Tomas  */
44c9de560dSAlex Tomas 
45c9de560dSAlex Tomas /*
46c9de560dSAlex Tomas  * The allocation request involve request for multiple number of blocks
47c9de560dSAlex Tomas  * near to the goal(block) value specified.
48c9de560dSAlex Tomas  *
49c9de560dSAlex Tomas  * During initialization phase of the allocator we decide to use the group
50c9de560dSAlex Tomas  * preallocation or inode preallocation depending on the size file. The
51c9de560dSAlex Tomas  * size of the file could be the resulting file size we would have after
52c9de560dSAlex Tomas  * allocation or the current file size which ever is larger. If the size is
53c9de560dSAlex Tomas  * less that sbi->s_mb_stream_request we select the group
54c9de560dSAlex Tomas  * preallocation. The default value of s_mb_stream_request is 16
55c9de560dSAlex Tomas  * blocks. This can also be tuned via
56c9de560dSAlex Tomas  * /proc/fs/ext4/<partition>/stream_req. The value is represented in terms
57c9de560dSAlex Tomas  * of number of blocks.
58c9de560dSAlex Tomas  *
59c9de560dSAlex Tomas  * The main motivation for having small file use group preallocation is to
60c9de560dSAlex Tomas  * ensure that we have small file closer in the disk.
61c9de560dSAlex Tomas  *
62c9de560dSAlex Tomas  * First stage the allocator looks at the inode prealloc list
63c9de560dSAlex Tomas  * ext4_inode_info->i_prealloc_list contain list of prealloc spaces for
64c9de560dSAlex Tomas  * this particular inode. The inode prealloc space is represented as:
65c9de560dSAlex Tomas  *
66c9de560dSAlex Tomas  * pa_lstart -> the logical start block for this prealloc space
67c9de560dSAlex Tomas  * pa_pstart -> the physical start block for this prealloc space
68c9de560dSAlex Tomas  * pa_len    -> lenght for this prealloc space
69c9de560dSAlex Tomas  * pa_free   ->  free space available in this prealloc space
70c9de560dSAlex Tomas  *
71c9de560dSAlex Tomas  * The inode preallocation space is used looking at the _logical_ start
72c9de560dSAlex Tomas  * block. If only the logical file block falls within the range of prealloc
73c9de560dSAlex Tomas  * space we will consume the particular prealloc space. This make sure that
74c9de560dSAlex Tomas  * that the we have contiguous physical blocks representing the file blocks
75c9de560dSAlex Tomas  *
76c9de560dSAlex Tomas  * The important thing to be noted in case of inode prealloc space is that
77c9de560dSAlex Tomas  * we don't modify the values associated to inode prealloc space except
78c9de560dSAlex Tomas  * pa_free.
79c9de560dSAlex Tomas  *
80c9de560dSAlex Tomas  * If we are not able to find blocks in the inode prealloc space and if we
81c9de560dSAlex Tomas  * have the group allocation flag set then we look at the locality group
82c9de560dSAlex Tomas  * prealloc space. These are per CPU prealloc list repreasented as
83c9de560dSAlex Tomas  *
84c9de560dSAlex Tomas  * ext4_sb_info.s_locality_groups[smp_processor_id()]
85c9de560dSAlex Tomas  *
86c9de560dSAlex Tomas  * The reason for having a per cpu locality group is to reduce the contention
87c9de560dSAlex Tomas  * between CPUs. It is possible to get scheduled at this point.
88c9de560dSAlex Tomas  *
89c9de560dSAlex Tomas  * The locality group prealloc space is used looking at whether we have
90c9de560dSAlex Tomas  * enough free space (pa_free) withing the prealloc space.
91c9de560dSAlex Tomas  *
92c9de560dSAlex Tomas  * If we can't allocate blocks via inode prealloc or/and locality group
93c9de560dSAlex Tomas  * prealloc then we look at the buddy cache. The buddy cache is represented
94c9de560dSAlex Tomas  * by ext4_sb_info.s_buddy_cache (struct inode) whose file offset gets
95c9de560dSAlex Tomas  * mapped to the buddy and bitmap information regarding different
96c9de560dSAlex Tomas  * groups. The buddy information is attached to buddy cache inode so that
97c9de560dSAlex Tomas  * we can access them through the page cache. The information regarding
98c9de560dSAlex Tomas  * each group is loaded via ext4_mb_load_buddy.  The information involve
99c9de560dSAlex Tomas  * block bitmap and buddy information. The information are stored in the
100c9de560dSAlex Tomas  * inode as:
101c9de560dSAlex Tomas  *
102c9de560dSAlex Tomas  *  {                        page                        }
103c3a326a6SAneesh Kumar K.V  *  [ group 0 bitmap][ group 0 buddy] [group 1][ group 1]...
104c9de560dSAlex Tomas  *
105c9de560dSAlex Tomas  *
106c9de560dSAlex Tomas  * one block each for bitmap and buddy information.  So for each group we
107c9de560dSAlex Tomas  * take up 2 blocks. A page can contain blocks_per_page (PAGE_CACHE_SIZE /
108c9de560dSAlex Tomas  * blocksize) blocks.  So it can have information regarding groups_per_page
109c9de560dSAlex Tomas  * which is blocks_per_page/2
110c9de560dSAlex Tomas  *
111c9de560dSAlex Tomas  * The buddy cache inode is not stored on disk. The inode is thrown
112c9de560dSAlex Tomas  * away when the filesystem is unmounted.
113c9de560dSAlex Tomas  *
114c9de560dSAlex Tomas  * We look for count number of blocks in the buddy cache. If we were able
115c9de560dSAlex Tomas  * to locate that many free blocks we return with additional information
116c9de560dSAlex Tomas  * regarding rest of the contiguous physical block available
117c9de560dSAlex Tomas  *
118c9de560dSAlex Tomas  * Before allocating blocks via buddy cache we normalize the request
119c9de560dSAlex Tomas  * blocks. This ensure we ask for more blocks that we needed. The extra
120c9de560dSAlex Tomas  * blocks that we get after allocation is added to the respective prealloc
121c9de560dSAlex Tomas  * list. In case of inode preallocation we follow a list of heuristics
122c9de560dSAlex Tomas  * based on file size. This can be found in ext4_mb_normalize_request. If
123c9de560dSAlex Tomas  * we are doing a group prealloc we try to normalize the request to
124c9de560dSAlex Tomas  * sbi->s_mb_group_prealloc. Default value of s_mb_group_prealloc is set to
125c9de560dSAlex Tomas  * 512 blocks. This can be tuned via
126c9de560dSAlex Tomas  * /proc/fs/ext4/<partition/group_prealloc. The value is represented in
127c9de560dSAlex Tomas  * terms of number of blocks. If we have mounted the file system with -O
128c9de560dSAlex Tomas  * stripe=<value> option the group prealloc request is normalized to the
129c9de560dSAlex Tomas  * stripe value (sbi->s_stripe)
130c9de560dSAlex Tomas  *
131c9de560dSAlex Tomas  * The regular allocator(using the buddy cache) support few tunables.
132c9de560dSAlex Tomas  *
133c9de560dSAlex Tomas  * /proc/fs/ext4/<partition>/min_to_scan
134c9de560dSAlex Tomas  * /proc/fs/ext4/<partition>/max_to_scan
135c9de560dSAlex Tomas  * /proc/fs/ext4/<partition>/order2_req
136c9de560dSAlex Tomas  *
137c9de560dSAlex Tomas  * The regular allocator use buddy scan only if the request len is power of
138c9de560dSAlex Tomas  * 2 blocks and the order of allocation is >= sbi->s_mb_order2_reqs. The
139c9de560dSAlex Tomas  * value of s_mb_order2_reqs can be tuned via
140c9de560dSAlex Tomas  * /proc/fs/ext4/<partition>/order2_req.  If the request len is equal to
141c9de560dSAlex Tomas  * stripe size (sbi->s_stripe), we try to search for contigous block in
142c9de560dSAlex Tomas  * stripe size. This should result in better allocation on RAID setup. If
143c9de560dSAlex Tomas  * not we search in the specific group using bitmap for best extents. The
144c9de560dSAlex Tomas  * tunable min_to_scan and max_to_scan controll the behaviour here.
145c9de560dSAlex Tomas  * min_to_scan indicate how long the mballoc __must__ look for a best
146c9de560dSAlex Tomas  * extent and max_to_scanindicate how long the mballoc __can__ look for a
147c9de560dSAlex Tomas  * best extent in the found extents. Searching for the blocks starts with
148c9de560dSAlex Tomas  * the group specified as the goal value in allocation context via
149c9de560dSAlex Tomas  * ac_g_ex. Each group is first checked based on the criteria whether it
150c9de560dSAlex Tomas  * can used for allocation. ext4_mb_good_group explains how the groups are
151c9de560dSAlex Tomas  * checked.
152c9de560dSAlex Tomas  *
153c9de560dSAlex Tomas  * Both the prealloc space are getting populated as above. So for the first
154c9de560dSAlex Tomas  * request we will hit the buddy cache which will result in this prealloc
155c9de560dSAlex Tomas  * space getting filled. The prealloc space is then later used for the
156c9de560dSAlex Tomas  * subsequent request.
157c9de560dSAlex Tomas  */
158c9de560dSAlex Tomas 
159c9de560dSAlex Tomas /*
160c9de560dSAlex Tomas  * mballoc operates on the following data:
161c9de560dSAlex Tomas  *  - on-disk bitmap
162c9de560dSAlex Tomas  *  - in-core buddy (actually includes buddy and bitmap)
163c9de560dSAlex Tomas  *  - preallocation descriptors (PAs)
164c9de560dSAlex Tomas  *
165c9de560dSAlex Tomas  * there are two types of preallocations:
166c9de560dSAlex Tomas  *  - inode
167c9de560dSAlex Tomas  *    assiged to specific inode and can be used for this inode only.
168c9de560dSAlex Tomas  *    it describes part of inode's space preallocated to specific
169c9de560dSAlex Tomas  *    physical blocks. any block from that preallocated can be used
170c9de560dSAlex Tomas  *    independent. the descriptor just tracks number of blocks left
171c9de560dSAlex Tomas  *    unused. so, before taking some block from descriptor, one must
172c9de560dSAlex Tomas  *    make sure corresponded logical block isn't allocated yet. this
173c9de560dSAlex Tomas  *    also means that freeing any block within descriptor's range
174c9de560dSAlex Tomas  *    must discard all preallocated blocks.
175c9de560dSAlex Tomas  *  - locality group
176c9de560dSAlex Tomas  *    assigned to specific locality group which does not translate to
177c9de560dSAlex Tomas  *    permanent set of inodes: inode can join and leave group. space
178c9de560dSAlex Tomas  *    from this type of preallocation can be used for any inode. thus
179c9de560dSAlex Tomas  *    it's consumed from the beginning to the end.
180c9de560dSAlex Tomas  *
181c9de560dSAlex Tomas  * relation between them can be expressed as:
182c9de560dSAlex Tomas  *    in-core buddy = on-disk bitmap + preallocation descriptors
183c9de560dSAlex Tomas  *
184c9de560dSAlex Tomas  * this mean blocks mballoc considers used are:
185c9de560dSAlex Tomas  *  - allocated blocks (persistent)
186c9de560dSAlex Tomas  *  - preallocated blocks (non-persistent)
187c9de560dSAlex Tomas  *
188c9de560dSAlex Tomas  * consistency in mballoc world means that at any time a block is either
189c9de560dSAlex Tomas  * free or used in ALL structures. notice: "any time" should not be read
190c9de560dSAlex Tomas  * literally -- time is discrete and delimited by locks.
191c9de560dSAlex Tomas  *
192c9de560dSAlex Tomas  *  to keep it simple, we don't use block numbers, instead we count number of
193c9de560dSAlex Tomas  *  blocks: how many blocks marked used/free in on-disk bitmap, buddy and PA.
194c9de560dSAlex Tomas  *
195c9de560dSAlex Tomas  * all operations can be expressed as:
196c9de560dSAlex Tomas  *  - init buddy:			buddy = on-disk + PAs
197c9de560dSAlex Tomas  *  - new PA:				buddy += N; PA = N
198c9de560dSAlex Tomas  *  - use inode PA:			on-disk += N; PA -= N
199c9de560dSAlex Tomas  *  - discard inode PA			buddy -= on-disk - PA; PA = 0
200c9de560dSAlex Tomas  *  - use locality group PA		on-disk += N; PA -= N
201c9de560dSAlex Tomas  *  - discard locality group PA		buddy -= PA; PA = 0
202c9de560dSAlex Tomas  *  note: 'buddy -= on-disk - PA' is used to show that on-disk bitmap
203c9de560dSAlex Tomas  *        is used in real operation because we can't know actual used
204c9de560dSAlex Tomas  *        bits from PA, only from on-disk bitmap
205c9de560dSAlex Tomas  *
206c9de560dSAlex Tomas  * if we follow this strict logic, then all operations above should be atomic.
207c9de560dSAlex Tomas  * given some of them can block, we'd have to use something like semaphores
208c9de560dSAlex Tomas  * killing performance on high-end SMP hardware. let's try to relax it using
209c9de560dSAlex Tomas  * the following knowledge:
210c9de560dSAlex Tomas  *  1) if buddy is referenced, it's already initialized
211c9de560dSAlex Tomas  *  2) while block is used in buddy and the buddy is referenced,
212c9de560dSAlex Tomas  *     nobody can re-allocate that block
213c9de560dSAlex Tomas  *  3) we work on bitmaps and '+' actually means 'set bits'. if on-disk has
214c9de560dSAlex Tomas  *     bit set and PA claims same block, it's OK. IOW, one can set bit in
215c9de560dSAlex Tomas  *     on-disk bitmap if buddy has same bit set or/and PA covers corresponded
216c9de560dSAlex Tomas  *     block
217c9de560dSAlex Tomas  *
218c9de560dSAlex Tomas  * so, now we're building a concurrency table:
219c9de560dSAlex Tomas  *  - init buddy vs.
220c9de560dSAlex Tomas  *    - new PA
221c9de560dSAlex Tomas  *      blocks for PA are allocated in the buddy, buddy must be referenced
222c9de560dSAlex Tomas  *      until PA is linked to allocation group to avoid concurrent buddy init
223c9de560dSAlex Tomas  *    - use inode PA
224c9de560dSAlex Tomas  *      we need to make sure that either on-disk bitmap or PA has uptodate data
225c9de560dSAlex Tomas  *      given (3) we care that PA-=N operation doesn't interfere with init
226c9de560dSAlex Tomas  *    - discard inode PA
227c9de560dSAlex Tomas  *      the simplest way would be to have buddy initialized by the discard
228c9de560dSAlex Tomas  *    - use locality group PA
229c9de560dSAlex Tomas  *      again PA-=N must be serialized with init
230c9de560dSAlex Tomas  *    - discard locality group PA
231c9de560dSAlex Tomas  *      the simplest way would be to have buddy initialized by the discard
232c9de560dSAlex Tomas  *  - new PA vs.
233c9de560dSAlex Tomas  *    - use inode PA
234c9de560dSAlex Tomas  *      i_data_sem serializes them
235c9de560dSAlex Tomas  *    - discard inode PA
236c9de560dSAlex Tomas  *      discard process must wait until PA isn't used by another process
237c9de560dSAlex Tomas  *    - use locality group PA
238c9de560dSAlex Tomas  *      some mutex should serialize them
239c9de560dSAlex Tomas  *    - discard locality group PA
240c9de560dSAlex Tomas  *      discard process must wait until PA isn't used by another process
241c9de560dSAlex Tomas  *  - use inode PA
242c9de560dSAlex Tomas  *    - use inode PA
243c9de560dSAlex Tomas  *      i_data_sem or another mutex should serializes them
244c9de560dSAlex Tomas  *    - discard inode PA
245c9de560dSAlex Tomas  *      discard process must wait until PA isn't used by another process
246c9de560dSAlex Tomas  *    - use locality group PA
247c9de560dSAlex Tomas  *      nothing wrong here -- they're different PAs covering different blocks
248c9de560dSAlex Tomas  *    - discard locality group PA
249c9de560dSAlex Tomas  *      discard process must wait until PA isn't used by another process
250c9de560dSAlex Tomas  *
251c9de560dSAlex Tomas  * now we're ready to make few consequences:
252c9de560dSAlex Tomas  *  - PA is referenced and while it is no discard is possible
253c9de560dSAlex Tomas  *  - PA is referenced until block isn't marked in on-disk bitmap
254c9de560dSAlex Tomas  *  - PA changes only after on-disk bitmap
255c9de560dSAlex Tomas  *  - discard must not compete with init. either init is done before
256c9de560dSAlex Tomas  *    any discard or they're serialized somehow
257c9de560dSAlex Tomas  *  - buddy init as sum of on-disk bitmap and PAs is done atomically
258c9de560dSAlex Tomas  *
259c9de560dSAlex Tomas  * a special case when we've used PA to emptiness. no need to modify buddy
260c9de560dSAlex Tomas  * in this case, but we should care about concurrent init
261c9de560dSAlex Tomas  *
262c9de560dSAlex Tomas  */
263c9de560dSAlex Tomas 
264c9de560dSAlex Tomas  /*
265c9de560dSAlex Tomas  * Logic in few words:
266c9de560dSAlex Tomas  *
267c9de560dSAlex Tomas  *  - allocation:
268c9de560dSAlex Tomas  *    load group
269c9de560dSAlex Tomas  *    find blocks
270c9de560dSAlex Tomas  *    mark bits in on-disk bitmap
271c9de560dSAlex Tomas  *    release group
272c9de560dSAlex Tomas  *
273c9de560dSAlex Tomas  *  - use preallocation:
274c9de560dSAlex Tomas  *    find proper PA (per-inode or group)
275c9de560dSAlex Tomas  *    load group
276c9de560dSAlex Tomas  *    mark bits in on-disk bitmap
277c9de560dSAlex Tomas  *    release group
278c9de560dSAlex Tomas  *    release PA
279c9de560dSAlex Tomas  *
280c9de560dSAlex Tomas  *  - free:
281c9de560dSAlex Tomas  *    load group
282c9de560dSAlex Tomas  *    mark bits in on-disk bitmap
283c9de560dSAlex Tomas  *    release group
284c9de560dSAlex Tomas  *
285c9de560dSAlex Tomas  *  - discard preallocations in group:
286c9de560dSAlex Tomas  *    mark PAs deleted
287c9de560dSAlex Tomas  *    move them onto local list
288c9de560dSAlex Tomas  *    load on-disk bitmap
289c9de560dSAlex Tomas  *    load group
290c9de560dSAlex Tomas  *    remove PA from object (inode or locality group)
291c9de560dSAlex Tomas  *    mark free blocks in-core
292c9de560dSAlex Tomas  *
293c9de560dSAlex Tomas  *  - discard inode's preallocations:
294c9de560dSAlex Tomas  */
295c9de560dSAlex Tomas 
296c9de560dSAlex Tomas /*
297c9de560dSAlex Tomas  * Locking rules
298c9de560dSAlex Tomas  *
299c9de560dSAlex Tomas  * Locks:
300c9de560dSAlex Tomas  *  - bitlock on a group	(group)
301c9de560dSAlex Tomas  *  - object (inode/locality)	(object)
302c9de560dSAlex Tomas  *  - per-pa lock		(pa)
303c9de560dSAlex Tomas  *
304c9de560dSAlex Tomas  * Paths:
305c9de560dSAlex Tomas  *  - new pa
306c9de560dSAlex Tomas  *    object
307c9de560dSAlex Tomas  *    group
308c9de560dSAlex Tomas  *
309c9de560dSAlex Tomas  *  - find and use pa:
310c9de560dSAlex Tomas  *    pa
311c9de560dSAlex Tomas  *
312c9de560dSAlex Tomas  *  - release consumed pa:
313c9de560dSAlex Tomas  *    pa
314c9de560dSAlex Tomas  *    group
315c9de560dSAlex Tomas  *    object
316c9de560dSAlex Tomas  *
317c9de560dSAlex Tomas  *  - generate in-core bitmap:
318c9de560dSAlex Tomas  *    group
319c9de560dSAlex Tomas  *        pa
320c9de560dSAlex Tomas  *
321c9de560dSAlex Tomas  *  - discard all for given object (inode, locality group):
322c9de560dSAlex Tomas  *    object
323c9de560dSAlex Tomas  *        pa
324c9de560dSAlex Tomas  *    group
325c9de560dSAlex Tomas  *
326c9de560dSAlex Tomas  *  - discard all for given group:
327c9de560dSAlex Tomas  *    group
328c9de560dSAlex Tomas  *        pa
329c9de560dSAlex Tomas  *    group
330c9de560dSAlex Tomas  *        object
331c9de560dSAlex Tomas  *
332c9de560dSAlex Tomas  */
333c3a326a6SAneesh Kumar K.V static struct kmem_cache *ext4_pspace_cachep;
334c3a326a6SAneesh Kumar K.V static struct kmem_cache *ext4_ac_cachep;
335c3a326a6SAneesh Kumar K.V static struct kmem_cache *ext4_free_ext_cachep;
336c3a326a6SAneesh Kumar K.V static void ext4_mb_generate_from_pa(struct super_block *sb, void *bitmap,
337c3a326a6SAneesh Kumar K.V 					ext4_group_t group);
3387a2fcbf7SAneesh Kumar K.V static void ext4_mb_generate_from_freelist(struct super_block *sb, void *bitmap,
3397a2fcbf7SAneesh Kumar K.V 						ext4_group_t group);
340c3a326a6SAneesh Kumar K.V static int ext4_mb_init_per_dev_proc(struct super_block *sb);
341c3a326a6SAneesh Kumar K.V static int ext4_mb_destroy_per_dev_proc(struct super_block *sb);
342c3a326a6SAneesh Kumar K.V static void release_blocks_on_commit(journal_t *journal, transaction_t *txn);
343c3a326a6SAneesh Kumar K.V 
344c3a326a6SAneesh Kumar K.V 
345c9de560dSAlex Tomas 
346ffad0a44SAneesh Kumar K.V static inline void *mb_correct_addr_and_bit(int *bit, void *addr)
347ffad0a44SAneesh Kumar K.V {
348c9de560dSAlex Tomas #if BITS_PER_LONG == 64
349ffad0a44SAneesh Kumar K.V 	*bit += ((unsigned long) addr & 7UL) << 3;
350ffad0a44SAneesh Kumar K.V 	addr = (void *) ((unsigned long) addr & ~7UL);
351c9de560dSAlex Tomas #elif BITS_PER_LONG == 32
352ffad0a44SAneesh Kumar K.V 	*bit += ((unsigned long) addr & 3UL) << 3;
353ffad0a44SAneesh Kumar K.V 	addr = (void *) ((unsigned long) addr & ~3UL);
354c9de560dSAlex Tomas #else
355c9de560dSAlex Tomas #error "how many bits you are?!"
356c9de560dSAlex Tomas #endif
357ffad0a44SAneesh Kumar K.V 	return addr;
358ffad0a44SAneesh Kumar K.V }
359c9de560dSAlex Tomas 
360c9de560dSAlex Tomas static inline int mb_test_bit(int bit, void *addr)
361c9de560dSAlex Tomas {
362c9de560dSAlex Tomas 	/*
363c9de560dSAlex Tomas 	 * ext4_test_bit on architecture like powerpc
364c9de560dSAlex Tomas 	 * needs unsigned long aligned address
365c9de560dSAlex Tomas 	 */
366ffad0a44SAneesh Kumar K.V 	addr = mb_correct_addr_and_bit(&bit, addr);
367c9de560dSAlex Tomas 	return ext4_test_bit(bit, addr);
368c9de560dSAlex Tomas }
369c9de560dSAlex Tomas 
370c9de560dSAlex Tomas static inline void mb_set_bit(int bit, void *addr)
371c9de560dSAlex Tomas {
372ffad0a44SAneesh Kumar K.V 	addr = mb_correct_addr_and_bit(&bit, addr);
373c9de560dSAlex Tomas 	ext4_set_bit(bit, addr);
374c9de560dSAlex Tomas }
375c9de560dSAlex Tomas 
376c9de560dSAlex Tomas static inline void mb_set_bit_atomic(spinlock_t *lock, int bit, void *addr)
377c9de560dSAlex Tomas {
378ffad0a44SAneesh Kumar K.V 	addr = mb_correct_addr_and_bit(&bit, addr);
379c9de560dSAlex Tomas 	ext4_set_bit_atomic(lock, bit, addr);
380c9de560dSAlex Tomas }
381c9de560dSAlex Tomas 
382c9de560dSAlex Tomas static inline void mb_clear_bit(int bit, void *addr)
383c9de560dSAlex Tomas {
384ffad0a44SAneesh Kumar K.V 	addr = mb_correct_addr_and_bit(&bit, addr);
385c9de560dSAlex Tomas 	ext4_clear_bit(bit, addr);
386c9de560dSAlex Tomas }
387c9de560dSAlex Tomas 
388c9de560dSAlex Tomas static inline void mb_clear_bit_atomic(spinlock_t *lock, int bit, void *addr)
389c9de560dSAlex Tomas {
390ffad0a44SAneesh Kumar K.V 	addr = mb_correct_addr_and_bit(&bit, addr);
391c9de560dSAlex Tomas 	ext4_clear_bit_atomic(lock, bit, addr);
392c9de560dSAlex Tomas }
393c9de560dSAlex Tomas 
394ffad0a44SAneesh Kumar K.V static inline int mb_find_next_zero_bit(void *addr, int max, int start)
395ffad0a44SAneesh Kumar K.V {
396e7dfb246SAneesh Kumar K.V 	int fix = 0, ret, tmpmax;
397ffad0a44SAneesh Kumar K.V 	addr = mb_correct_addr_and_bit(&fix, addr);
398e7dfb246SAneesh Kumar K.V 	tmpmax = max + fix;
399ffad0a44SAneesh Kumar K.V 	start += fix;
400ffad0a44SAneesh Kumar K.V 
401e7dfb246SAneesh Kumar K.V 	ret = ext4_find_next_zero_bit(addr, tmpmax, start) - fix;
402e7dfb246SAneesh Kumar K.V 	if (ret > max)
403e7dfb246SAneesh Kumar K.V 		return max;
404e7dfb246SAneesh Kumar K.V 	return ret;
405ffad0a44SAneesh Kumar K.V }
406ffad0a44SAneesh Kumar K.V 
407ffad0a44SAneesh Kumar K.V static inline int mb_find_next_bit(void *addr, int max, int start)
408ffad0a44SAneesh Kumar K.V {
409e7dfb246SAneesh Kumar K.V 	int fix = 0, ret, tmpmax;
410ffad0a44SAneesh Kumar K.V 	addr = mb_correct_addr_and_bit(&fix, addr);
411e7dfb246SAneesh Kumar K.V 	tmpmax = max + fix;
412ffad0a44SAneesh Kumar K.V 	start += fix;
413ffad0a44SAneesh Kumar K.V 
414e7dfb246SAneesh Kumar K.V 	ret = ext4_find_next_bit(addr, tmpmax, start) - fix;
415e7dfb246SAneesh Kumar K.V 	if (ret > max)
416e7dfb246SAneesh Kumar K.V 		return max;
417e7dfb246SAneesh Kumar K.V 	return ret;
418ffad0a44SAneesh Kumar K.V }
419ffad0a44SAneesh Kumar K.V 
420c9de560dSAlex Tomas static void *mb_find_buddy(struct ext4_buddy *e4b, int order, int *max)
421c9de560dSAlex Tomas {
422c9de560dSAlex Tomas 	char *bb;
423c9de560dSAlex Tomas 
424c9de560dSAlex Tomas 	BUG_ON(EXT4_MB_BITMAP(e4b) == EXT4_MB_BUDDY(e4b));
425c9de560dSAlex Tomas 	BUG_ON(max == NULL);
426c9de560dSAlex Tomas 
427c9de560dSAlex Tomas 	if (order > e4b->bd_blkbits + 1) {
428c9de560dSAlex Tomas 		*max = 0;
429c9de560dSAlex Tomas 		return NULL;
430c9de560dSAlex Tomas 	}
431c9de560dSAlex Tomas 
432c9de560dSAlex Tomas 	/* at order 0 we see each particular block */
433c9de560dSAlex Tomas 	*max = 1 << (e4b->bd_blkbits + 3);
434c9de560dSAlex Tomas 	if (order == 0)
435c9de560dSAlex Tomas 		return EXT4_MB_BITMAP(e4b);
436c9de560dSAlex Tomas 
437c9de560dSAlex Tomas 	bb = EXT4_MB_BUDDY(e4b) + EXT4_SB(e4b->bd_sb)->s_mb_offsets[order];
438c9de560dSAlex Tomas 	*max = EXT4_SB(e4b->bd_sb)->s_mb_maxs[order];
439c9de560dSAlex Tomas 
440c9de560dSAlex Tomas 	return bb;
441c9de560dSAlex Tomas }
442c9de560dSAlex Tomas 
443c9de560dSAlex Tomas #ifdef DOUBLE_CHECK
444c9de560dSAlex Tomas static void mb_free_blocks_double(struct inode *inode, struct ext4_buddy *e4b,
445c9de560dSAlex Tomas 			   int first, int count)
446c9de560dSAlex Tomas {
447c9de560dSAlex Tomas 	int i;
448c9de560dSAlex Tomas 	struct super_block *sb = e4b->bd_sb;
449c9de560dSAlex Tomas 
450c9de560dSAlex Tomas 	if (unlikely(e4b->bd_info->bb_bitmap == NULL))
451c9de560dSAlex Tomas 		return;
452c9de560dSAlex Tomas 	BUG_ON(!ext4_is_group_locked(sb, e4b->bd_group));
453c9de560dSAlex Tomas 	for (i = 0; i < count; i++) {
454c9de560dSAlex Tomas 		if (!mb_test_bit(first + i, e4b->bd_info->bb_bitmap)) {
455c9de560dSAlex Tomas 			ext4_fsblk_t blocknr;
456c9de560dSAlex Tomas 			blocknr = e4b->bd_group * EXT4_BLOCKS_PER_GROUP(sb);
457c9de560dSAlex Tomas 			blocknr += first + i;
458c9de560dSAlex Tomas 			blocknr +=
459c9de560dSAlex Tomas 			    le32_to_cpu(EXT4_SB(sb)->s_es->s_first_data_block);
4605d1b1b3fSAneesh Kumar K.V 			ext4_grp_locked_error(sb, e4b->bd_group,
4615d1b1b3fSAneesh Kumar K.V 				   __func__, "double-free of inode"
462a9df9a49STheodore Ts'o 				   " %lu's block %llu(bit %u in group %u)",
463c9de560dSAlex Tomas 				   inode ? inode->i_ino : 0, blocknr,
464c9de560dSAlex Tomas 				   first + i, e4b->bd_group);
465c9de560dSAlex Tomas 		}
466c9de560dSAlex Tomas 		mb_clear_bit(first + i, e4b->bd_info->bb_bitmap);
467c9de560dSAlex Tomas 	}
468c9de560dSAlex Tomas }
469c9de560dSAlex Tomas 
470c9de560dSAlex Tomas static void mb_mark_used_double(struct ext4_buddy *e4b, int first, int count)
471c9de560dSAlex Tomas {
472c9de560dSAlex Tomas 	int i;
473c9de560dSAlex Tomas 
474c9de560dSAlex Tomas 	if (unlikely(e4b->bd_info->bb_bitmap == NULL))
475c9de560dSAlex Tomas 		return;
476c9de560dSAlex Tomas 	BUG_ON(!ext4_is_group_locked(e4b->bd_sb, e4b->bd_group));
477c9de560dSAlex Tomas 	for (i = 0; i < count; i++) {
478c9de560dSAlex Tomas 		BUG_ON(mb_test_bit(first + i, e4b->bd_info->bb_bitmap));
479c9de560dSAlex Tomas 		mb_set_bit(first + i, e4b->bd_info->bb_bitmap);
480c9de560dSAlex Tomas 	}
481c9de560dSAlex Tomas }
482c9de560dSAlex Tomas 
483c9de560dSAlex Tomas static void mb_cmp_bitmaps(struct ext4_buddy *e4b, void *bitmap)
484c9de560dSAlex Tomas {
485c9de560dSAlex Tomas 	if (memcmp(e4b->bd_info->bb_bitmap, bitmap, e4b->bd_sb->s_blocksize)) {
486c9de560dSAlex Tomas 		unsigned char *b1, *b2;
487c9de560dSAlex Tomas 		int i;
488c9de560dSAlex Tomas 		b1 = (unsigned char *) e4b->bd_info->bb_bitmap;
489c9de560dSAlex Tomas 		b2 = (unsigned char *) bitmap;
490c9de560dSAlex Tomas 		for (i = 0; i < e4b->bd_sb->s_blocksize; i++) {
491c9de560dSAlex Tomas 			if (b1[i] != b2[i]) {
492a9df9a49STheodore Ts'o 				printk(KERN_ERR "corruption in group %u "
4934776004fSTheodore Ts'o 				       "at byte %u(%u): %x in copy != %x "
4944776004fSTheodore Ts'o 				       "on disk/prealloc\n",
495c9de560dSAlex Tomas 				       e4b->bd_group, i, i * 8, b1[i], b2[i]);
496c9de560dSAlex Tomas 				BUG();
497c9de560dSAlex Tomas 			}
498c9de560dSAlex Tomas 		}
499c9de560dSAlex Tomas 	}
500c9de560dSAlex Tomas }
501c9de560dSAlex Tomas 
502c9de560dSAlex Tomas #else
503c9de560dSAlex Tomas static inline void mb_free_blocks_double(struct inode *inode,
504c9de560dSAlex Tomas 				struct ext4_buddy *e4b, int first, int count)
505c9de560dSAlex Tomas {
506c9de560dSAlex Tomas 	return;
507c9de560dSAlex Tomas }
508c9de560dSAlex Tomas static inline void mb_mark_used_double(struct ext4_buddy *e4b,
509c9de560dSAlex Tomas 						int first, int count)
510c9de560dSAlex Tomas {
511c9de560dSAlex Tomas 	return;
512c9de560dSAlex Tomas }
513c9de560dSAlex Tomas static inline void mb_cmp_bitmaps(struct ext4_buddy *e4b, void *bitmap)
514c9de560dSAlex Tomas {
515c9de560dSAlex Tomas 	return;
516c9de560dSAlex Tomas }
517c9de560dSAlex Tomas #endif
518c9de560dSAlex Tomas 
519c9de560dSAlex Tomas #ifdef AGGRESSIVE_CHECK
520c9de560dSAlex Tomas 
521c9de560dSAlex Tomas #define MB_CHECK_ASSERT(assert)						\
522c9de560dSAlex Tomas do {									\
523c9de560dSAlex Tomas 	if (!(assert)) {						\
524c9de560dSAlex Tomas 		printk(KERN_EMERG					\
525c9de560dSAlex Tomas 			"Assertion failure in %s() at %s:%d: \"%s\"\n",	\
526c9de560dSAlex Tomas 			function, file, line, # assert);		\
527c9de560dSAlex Tomas 		BUG();							\
528c9de560dSAlex Tomas 	}								\
529c9de560dSAlex Tomas } while (0)
530c9de560dSAlex Tomas 
531c9de560dSAlex Tomas static int __mb_check_buddy(struct ext4_buddy *e4b, char *file,
532c9de560dSAlex Tomas 				const char *function, int line)
533c9de560dSAlex Tomas {
534c9de560dSAlex Tomas 	struct super_block *sb = e4b->bd_sb;
535c9de560dSAlex Tomas 	int order = e4b->bd_blkbits + 1;
536c9de560dSAlex Tomas 	int max;
537c9de560dSAlex Tomas 	int max2;
538c9de560dSAlex Tomas 	int i;
539c9de560dSAlex Tomas 	int j;
540c9de560dSAlex Tomas 	int k;
541c9de560dSAlex Tomas 	int count;
542c9de560dSAlex Tomas 	struct ext4_group_info *grp;
543c9de560dSAlex Tomas 	int fragments = 0;
544c9de560dSAlex Tomas 	int fstart;
545c9de560dSAlex Tomas 	struct list_head *cur;
546c9de560dSAlex Tomas 	void *buddy;
547c9de560dSAlex Tomas 	void *buddy2;
548c9de560dSAlex Tomas 
549c9de560dSAlex Tomas 	{
550c9de560dSAlex Tomas 		static int mb_check_counter;
551c9de560dSAlex Tomas 		if (mb_check_counter++ % 100 != 0)
552c9de560dSAlex Tomas 			return 0;
553c9de560dSAlex Tomas 	}
554c9de560dSAlex Tomas 
555c9de560dSAlex Tomas 	while (order > 1) {
556c9de560dSAlex Tomas 		buddy = mb_find_buddy(e4b, order, &max);
557c9de560dSAlex Tomas 		MB_CHECK_ASSERT(buddy);
558c9de560dSAlex Tomas 		buddy2 = mb_find_buddy(e4b, order - 1, &max2);
559c9de560dSAlex Tomas 		MB_CHECK_ASSERT(buddy2);
560c9de560dSAlex Tomas 		MB_CHECK_ASSERT(buddy != buddy2);
561c9de560dSAlex Tomas 		MB_CHECK_ASSERT(max * 2 == max2);
562c9de560dSAlex Tomas 
563c9de560dSAlex Tomas 		count = 0;
564c9de560dSAlex Tomas 		for (i = 0; i < max; i++) {
565c9de560dSAlex Tomas 
566c9de560dSAlex Tomas 			if (mb_test_bit(i, buddy)) {
567c9de560dSAlex Tomas 				/* only single bit in buddy2 may be 1 */
568c9de560dSAlex Tomas 				if (!mb_test_bit(i << 1, buddy2)) {
569c9de560dSAlex Tomas 					MB_CHECK_ASSERT(
570c9de560dSAlex Tomas 						mb_test_bit((i<<1)+1, buddy2));
571c9de560dSAlex Tomas 				} else if (!mb_test_bit((i << 1) + 1, buddy2)) {
572c9de560dSAlex Tomas 					MB_CHECK_ASSERT(
573c9de560dSAlex Tomas 						mb_test_bit(i << 1, buddy2));
574c9de560dSAlex Tomas 				}
575c9de560dSAlex Tomas 				continue;
576c9de560dSAlex Tomas 			}
577c9de560dSAlex Tomas 
578c9de560dSAlex Tomas 			/* both bits in buddy2 must be 0 */
579c9de560dSAlex Tomas 			MB_CHECK_ASSERT(mb_test_bit(i << 1, buddy2));
580c9de560dSAlex Tomas 			MB_CHECK_ASSERT(mb_test_bit((i << 1) + 1, buddy2));
581c9de560dSAlex Tomas 
582c9de560dSAlex Tomas 			for (j = 0; j < (1 << order); j++) {
583c9de560dSAlex Tomas 				k = (i * (1 << order)) + j;
584c9de560dSAlex Tomas 				MB_CHECK_ASSERT(
585c9de560dSAlex Tomas 					!mb_test_bit(k, EXT4_MB_BITMAP(e4b)));
586c9de560dSAlex Tomas 			}
587c9de560dSAlex Tomas 			count++;
588c9de560dSAlex Tomas 		}
589c9de560dSAlex Tomas 		MB_CHECK_ASSERT(e4b->bd_info->bb_counters[order] == count);
590c9de560dSAlex Tomas 		order--;
591c9de560dSAlex Tomas 	}
592c9de560dSAlex Tomas 
593c9de560dSAlex Tomas 	fstart = -1;
594c9de560dSAlex Tomas 	buddy = mb_find_buddy(e4b, 0, &max);
595c9de560dSAlex Tomas 	for (i = 0; i < max; i++) {
596c9de560dSAlex Tomas 		if (!mb_test_bit(i, buddy)) {
597c9de560dSAlex Tomas 			MB_CHECK_ASSERT(i >= e4b->bd_info->bb_first_free);
598c9de560dSAlex Tomas 			if (fstart == -1) {
599c9de560dSAlex Tomas 				fragments++;
600c9de560dSAlex Tomas 				fstart = i;
601c9de560dSAlex Tomas 			}
602c9de560dSAlex Tomas 			continue;
603c9de560dSAlex Tomas 		}
604c9de560dSAlex Tomas 		fstart = -1;
605c9de560dSAlex Tomas 		/* check used bits only */
606c9de560dSAlex Tomas 		for (j = 0; j < e4b->bd_blkbits + 1; j++) {
607c9de560dSAlex Tomas 			buddy2 = mb_find_buddy(e4b, j, &max2);
608c9de560dSAlex Tomas 			k = i >> j;
609c9de560dSAlex Tomas 			MB_CHECK_ASSERT(k < max2);
610c9de560dSAlex Tomas 			MB_CHECK_ASSERT(mb_test_bit(k, buddy2));
611c9de560dSAlex Tomas 		}
612c9de560dSAlex Tomas 	}
613c9de560dSAlex Tomas 	MB_CHECK_ASSERT(!EXT4_MB_GRP_NEED_INIT(e4b->bd_info));
614c9de560dSAlex Tomas 	MB_CHECK_ASSERT(e4b->bd_info->bb_fragments == fragments);
615c9de560dSAlex Tomas 
616c9de560dSAlex Tomas 	grp = ext4_get_group_info(sb, e4b->bd_group);
617c9de560dSAlex Tomas 	buddy = mb_find_buddy(e4b, 0, &max);
618c9de560dSAlex Tomas 	list_for_each(cur, &grp->bb_prealloc_list) {
619c9de560dSAlex Tomas 		ext4_group_t groupnr;
620c9de560dSAlex Tomas 		struct ext4_prealloc_space *pa;
62160bd63d1SSolofo Ramangalahy 		pa = list_entry(cur, struct ext4_prealloc_space, pa_group_list);
62260bd63d1SSolofo Ramangalahy 		ext4_get_group_no_and_offset(sb, pa->pa_pstart, &groupnr, &k);
623c9de560dSAlex Tomas 		MB_CHECK_ASSERT(groupnr == e4b->bd_group);
62460bd63d1SSolofo Ramangalahy 		for (i = 0; i < pa->pa_len; i++)
625c9de560dSAlex Tomas 			MB_CHECK_ASSERT(mb_test_bit(k + i, buddy));
626c9de560dSAlex Tomas 	}
627c9de560dSAlex Tomas 	return 0;
628c9de560dSAlex Tomas }
629c9de560dSAlex Tomas #undef MB_CHECK_ASSERT
630c9de560dSAlex Tomas #define mb_check_buddy(e4b) __mb_check_buddy(e4b,	\
63146e665e9SHarvey Harrison 					__FILE__, __func__, __LINE__)
632c9de560dSAlex Tomas #else
633c9de560dSAlex Tomas #define mb_check_buddy(e4b)
634c9de560dSAlex Tomas #endif
635c9de560dSAlex Tomas 
636c9de560dSAlex Tomas /* FIXME!! need more doc */
637c9de560dSAlex Tomas static void ext4_mb_mark_free_simple(struct super_block *sb,
638c9de560dSAlex Tomas 				void *buddy, unsigned first, int len,
639c9de560dSAlex Tomas 					struct ext4_group_info *grp)
640c9de560dSAlex Tomas {
641c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
642c9de560dSAlex Tomas 	unsigned short min;
643c9de560dSAlex Tomas 	unsigned short max;
644c9de560dSAlex Tomas 	unsigned short chunk;
645c9de560dSAlex Tomas 	unsigned short border;
646c9de560dSAlex Tomas 
647b73fce69SValerie Clement 	BUG_ON(len > EXT4_BLOCKS_PER_GROUP(sb));
648c9de560dSAlex Tomas 
649c9de560dSAlex Tomas 	border = 2 << sb->s_blocksize_bits;
650c9de560dSAlex Tomas 
651c9de560dSAlex Tomas 	while (len > 0) {
652c9de560dSAlex Tomas 		/* find how many blocks can be covered since this position */
653c9de560dSAlex Tomas 		max = ffs(first | border) - 1;
654c9de560dSAlex Tomas 
655c9de560dSAlex Tomas 		/* find how many blocks of power 2 we need to mark */
656c9de560dSAlex Tomas 		min = fls(len) - 1;
657c9de560dSAlex Tomas 
658c9de560dSAlex Tomas 		if (max < min)
659c9de560dSAlex Tomas 			min = max;
660c9de560dSAlex Tomas 		chunk = 1 << min;
661c9de560dSAlex Tomas 
662c9de560dSAlex Tomas 		/* mark multiblock chunks only */
663c9de560dSAlex Tomas 		grp->bb_counters[min]++;
664c9de560dSAlex Tomas 		if (min > 0)
665c9de560dSAlex Tomas 			mb_clear_bit(first >> min,
666c9de560dSAlex Tomas 				     buddy + sbi->s_mb_offsets[min]);
667c9de560dSAlex Tomas 
668c9de560dSAlex Tomas 		len -= chunk;
669c9de560dSAlex Tomas 		first += chunk;
670c9de560dSAlex Tomas 	}
671c9de560dSAlex Tomas }
672c9de560dSAlex Tomas 
673c9de560dSAlex Tomas static void ext4_mb_generate_buddy(struct super_block *sb,
674c9de560dSAlex Tomas 				void *buddy, void *bitmap, ext4_group_t group)
675c9de560dSAlex Tomas {
676c9de560dSAlex Tomas 	struct ext4_group_info *grp = ext4_get_group_info(sb, group);
677c9de560dSAlex Tomas 	unsigned short max = EXT4_BLOCKS_PER_GROUP(sb);
678c9de560dSAlex Tomas 	unsigned short i = 0;
679c9de560dSAlex Tomas 	unsigned short first;
680c9de560dSAlex Tomas 	unsigned short len;
681c9de560dSAlex Tomas 	unsigned free = 0;
682c9de560dSAlex Tomas 	unsigned fragments = 0;
683c9de560dSAlex Tomas 	unsigned long long period = get_cycles();
684c9de560dSAlex Tomas 
685c9de560dSAlex Tomas 	/* initialize buddy from bitmap which is aggregation
686c9de560dSAlex Tomas 	 * of on-disk bitmap and preallocations */
687ffad0a44SAneesh Kumar K.V 	i = mb_find_next_zero_bit(bitmap, max, 0);
688c9de560dSAlex Tomas 	grp->bb_first_free = i;
689c9de560dSAlex Tomas 	while (i < max) {
690c9de560dSAlex Tomas 		fragments++;
691c9de560dSAlex Tomas 		first = i;
692ffad0a44SAneesh Kumar K.V 		i = mb_find_next_bit(bitmap, max, i);
693c9de560dSAlex Tomas 		len = i - first;
694c9de560dSAlex Tomas 		free += len;
695c9de560dSAlex Tomas 		if (len > 1)
696c9de560dSAlex Tomas 			ext4_mb_mark_free_simple(sb, buddy, first, len, grp);
697c9de560dSAlex Tomas 		else
698c9de560dSAlex Tomas 			grp->bb_counters[0]++;
699c9de560dSAlex Tomas 		if (i < max)
700ffad0a44SAneesh Kumar K.V 			i = mb_find_next_zero_bit(bitmap, max, i);
701c9de560dSAlex Tomas 	}
702c9de560dSAlex Tomas 	grp->bb_fragments = fragments;
703c9de560dSAlex Tomas 
704c9de560dSAlex Tomas 	if (free != grp->bb_free) {
7055d1b1b3fSAneesh Kumar K.V 		ext4_grp_locked_error(sb, group,  __func__,
706a9df9a49STheodore Ts'o 			"EXT4-fs: group %u: %u blocks in bitmap, %u in gd",
707c9de560dSAlex Tomas 			group, free, grp->bb_free);
708e56eb659SAneesh Kumar K.V 		/*
709e56eb659SAneesh Kumar K.V 		 * If we intent to continue, we consider group descritor
710e56eb659SAneesh Kumar K.V 		 * corrupt and update bb_free using bitmap value
711e56eb659SAneesh Kumar K.V 		 */
712c9de560dSAlex Tomas 		grp->bb_free = free;
713c9de560dSAlex Tomas 	}
714c9de560dSAlex Tomas 
715c9de560dSAlex Tomas 	clear_bit(EXT4_GROUP_INFO_NEED_INIT_BIT, &(grp->bb_state));
716c9de560dSAlex Tomas 
717c9de560dSAlex Tomas 	period = get_cycles() - period;
718c9de560dSAlex Tomas 	spin_lock(&EXT4_SB(sb)->s_bal_lock);
719c9de560dSAlex Tomas 	EXT4_SB(sb)->s_mb_buddies_generated++;
720c9de560dSAlex Tomas 	EXT4_SB(sb)->s_mb_generation_time += period;
721c9de560dSAlex Tomas 	spin_unlock(&EXT4_SB(sb)->s_bal_lock);
722c9de560dSAlex Tomas }
723c9de560dSAlex Tomas 
724c9de560dSAlex Tomas /* The buddy information is attached the buddy cache inode
725c9de560dSAlex Tomas  * for convenience. The information regarding each group
726c9de560dSAlex Tomas  * is loaded via ext4_mb_load_buddy. The information involve
727c9de560dSAlex Tomas  * block bitmap and buddy information. The information are
728c9de560dSAlex Tomas  * stored in the inode as
729c9de560dSAlex Tomas  *
730c9de560dSAlex Tomas  * {                        page                        }
731c3a326a6SAneesh Kumar K.V  * [ group 0 bitmap][ group 0 buddy] [group 1][ group 1]...
732c9de560dSAlex Tomas  *
733c9de560dSAlex Tomas  *
734c9de560dSAlex Tomas  * one block each for bitmap and buddy information.
735c9de560dSAlex Tomas  * So for each group we take up 2 blocks. A page can
736c9de560dSAlex Tomas  * contain blocks_per_page (PAGE_CACHE_SIZE / blocksize)  blocks.
737c9de560dSAlex Tomas  * So it can have information regarding groups_per_page which
738c9de560dSAlex Tomas  * is blocks_per_page/2
739c9de560dSAlex Tomas  */
740c9de560dSAlex Tomas 
741c9de560dSAlex Tomas static int ext4_mb_init_cache(struct page *page, char *incore)
742c9de560dSAlex Tomas {
743c9de560dSAlex Tomas 	int blocksize;
744c9de560dSAlex Tomas 	int blocks_per_page;
745c9de560dSAlex Tomas 	int groups_per_page;
746c9de560dSAlex Tomas 	int err = 0;
747c9de560dSAlex Tomas 	int i;
748c9de560dSAlex Tomas 	ext4_group_t first_group;
749c9de560dSAlex Tomas 	int first_block;
750c9de560dSAlex Tomas 	struct super_block *sb;
751c9de560dSAlex Tomas 	struct buffer_head *bhs;
752c9de560dSAlex Tomas 	struct buffer_head **bh;
753c9de560dSAlex Tomas 	struct inode *inode;
754c9de560dSAlex Tomas 	char *data;
755c9de560dSAlex Tomas 	char *bitmap;
756c9de560dSAlex Tomas 
757c9de560dSAlex Tomas 	mb_debug("init page %lu\n", page->index);
758c9de560dSAlex Tomas 
759c9de560dSAlex Tomas 	inode = page->mapping->host;
760c9de560dSAlex Tomas 	sb = inode->i_sb;
761c9de560dSAlex Tomas 	blocksize = 1 << inode->i_blkbits;
762c9de560dSAlex Tomas 	blocks_per_page = PAGE_CACHE_SIZE / blocksize;
763c9de560dSAlex Tomas 
764c9de560dSAlex Tomas 	groups_per_page = blocks_per_page >> 1;
765c9de560dSAlex Tomas 	if (groups_per_page == 0)
766c9de560dSAlex Tomas 		groups_per_page = 1;
767c9de560dSAlex Tomas 
768c9de560dSAlex Tomas 	/* allocate buffer_heads to read bitmaps */
769c9de560dSAlex Tomas 	if (groups_per_page > 1) {
770c9de560dSAlex Tomas 		err = -ENOMEM;
771c9de560dSAlex Tomas 		i = sizeof(struct buffer_head *) * groups_per_page;
772c9de560dSAlex Tomas 		bh = kzalloc(i, GFP_NOFS);
773c9de560dSAlex Tomas 		if (bh == NULL)
774c9de560dSAlex Tomas 			goto out;
775c9de560dSAlex Tomas 	} else
776c9de560dSAlex Tomas 		bh = &bhs;
777c9de560dSAlex Tomas 
778c9de560dSAlex Tomas 	first_group = page->index * blocks_per_page / 2;
779c9de560dSAlex Tomas 
780c9de560dSAlex Tomas 	/* read all groups the page covers into the cache */
781c9de560dSAlex Tomas 	for (i = 0; i < groups_per_page; i++) {
782c9de560dSAlex Tomas 		struct ext4_group_desc *desc;
783c9de560dSAlex Tomas 
784c9de560dSAlex Tomas 		if (first_group + i >= EXT4_SB(sb)->s_groups_count)
785c9de560dSAlex Tomas 			break;
786c9de560dSAlex Tomas 
787c9de560dSAlex Tomas 		err = -EIO;
788c9de560dSAlex Tomas 		desc = ext4_get_group_desc(sb, first_group + i, NULL);
789c9de560dSAlex Tomas 		if (desc == NULL)
790c9de560dSAlex Tomas 			goto out;
791c9de560dSAlex Tomas 
792c9de560dSAlex Tomas 		err = -ENOMEM;
793c9de560dSAlex Tomas 		bh[i] = sb_getblk(sb, ext4_block_bitmap(sb, desc));
794c9de560dSAlex Tomas 		if (bh[i] == NULL)
795c9de560dSAlex Tomas 			goto out;
796c9de560dSAlex Tomas 
7972ccb5fb9SAneesh Kumar K.V 		if (bitmap_uptodate(bh[i]))
798c9de560dSAlex Tomas 			continue;
799c9de560dSAlex Tomas 
800c806e68fSFrederic Bohe 		lock_buffer(bh[i]);
8012ccb5fb9SAneesh Kumar K.V 		if (bitmap_uptodate(bh[i])) {
8022ccb5fb9SAneesh Kumar K.V 			unlock_buffer(bh[i]);
8032ccb5fb9SAneesh Kumar K.V 			continue;
8042ccb5fb9SAneesh Kumar K.V 		}
805b5f10eedSEric Sandeen 		spin_lock(sb_bgl_lock(EXT4_SB(sb), first_group + i));
806c9de560dSAlex Tomas 		if (desc->bg_flags & cpu_to_le16(EXT4_BG_BLOCK_UNINIT)) {
807c9de560dSAlex Tomas 			ext4_init_block_bitmap(sb, bh[i],
808c9de560dSAlex Tomas 						first_group + i, desc);
8092ccb5fb9SAneesh Kumar K.V 			set_bitmap_uptodate(bh[i]);
810c9de560dSAlex Tomas 			set_buffer_uptodate(bh[i]);
811b5f10eedSEric Sandeen 			spin_unlock(sb_bgl_lock(EXT4_SB(sb), first_group + i));
8123300bedaSAneesh Kumar K.V 			unlock_buffer(bh[i]);
813c9de560dSAlex Tomas 			continue;
814c9de560dSAlex Tomas 		}
815b5f10eedSEric Sandeen 		spin_unlock(sb_bgl_lock(EXT4_SB(sb), first_group + i));
8162ccb5fb9SAneesh Kumar K.V 		if (buffer_uptodate(bh[i])) {
8172ccb5fb9SAneesh Kumar K.V 			/*
8182ccb5fb9SAneesh Kumar K.V 			 * if not uninit if bh is uptodate,
8192ccb5fb9SAneesh Kumar K.V 			 * bitmap is also uptodate
8202ccb5fb9SAneesh Kumar K.V 			 */
8212ccb5fb9SAneesh Kumar K.V 			set_bitmap_uptodate(bh[i]);
8222ccb5fb9SAneesh Kumar K.V 			unlock_buffer(bh[i]);
8232ccb5fb9SAneesh Kumar K.V 			continue;
8242ccb5fb9SAneesh Kumar K.V 		}
825c9de560dSAlex Tomas 		get_bh(bh[i]);
8262ccb5fb9SAneesh Kumar K.V 		/*
8272ccb5fb9SAneesh Kumar K.V 		 * submit the buffer_head for read. We can
8282ccb5fb9SAneesh Kumar K.V 		 * safely mark the bitmap as uptodate now.
8292ccb5fb9SAneesh Kumar K.V 		 * We do it here so the bitmap uptodate bit
8302ccb5fb9SAneesh Kumar K.V 		 * get set with buffer lock held.
8312ccb5fb9SAneesh Kumar K.V 		 */
8322ccb5fb9SAneesh Kumar K.V 		set_bitmap_uptodate(bh[i]);
833c9de560dSAlex Tomas 		bh[i]->b_end_io = end_buffer_read_sync;
834c9de560dSAlex Tomas 		submit_bh(READ, bh[i]);
835a9df9a49STheodore Ts'o 		mb_debug("read bitmap for group %u\n", first_group + i);
836c9de560dSAlex Tomas 	}
837c9de560dSAlex Tomas 
838c9de560dSAlex Tomas 	/* wait for I/O completion */
839c9de560dSAlex Tomas 	for (i = 0; i < groups_per_page && bh[i]; i++)
840c9de560dSAlex Tomas 		wait_on_buffer(bh[i]);
841c9de560dSAlex Tomas 
842c9de560dSAlex Tomas 	err = -EIO;
843c9de560dSAlex Tomas 	for (i = 0; i < groups_per_page && bh[i]; i++)
844c9de560dSAlex Tomas 		if (!buffer_uptodate(bh[i]))
845c9de560dSAlex Tomas 			goto out;
846c9de560dSAlex Tomas 
84731b481dcSMingming Cao 	err = 0;
848c9de560dSAlex Tomas 	first_block = page->index * blocks_per_page;
849c9de560dSAlex Tomas 	for (i = 0; i < blocks_per_page; i++) {
850c9de560dSAlex Tomas 		int group;
851c9de560dSAlex Tomas 		struct ext4_group_info *grinfo;
852c9de560dSAlex Tomas 
853c9de560dSAlex Tomas 		group = (first_block + i) >> 1;
854c9de560dSAlex Tomas 		if (group >= EXT4_SB(sb)->s_groups_count)
855c9de560dSAlex Tomas 			break;
856c9de560dSAlex Tomas 
857c9de560dSAlex Tomas 		/*
858c9de560dSAlex Tomas 		 * data carry information regarding this
859c9de560dSAlex Tomas 		 * particular group in the format specified
860c9de560dSAlex Tomas 		 * above
861c9de560dSAlex Tomas 		 *
862c9de560dSAlex Tomas 		 */
863c9de560dSAlex Tomas 		data = page_address(page) + (i * blocksize);
864c9de560dSAlex Tomas 		bitmap = bh[group - first_group]->b_data;
865c9de560dSAlex Tomas 
866c9de560dSAlex Tomas 		/*
867c9de560dSAlex Tomas 		 * We place the buddy block and bitmap block
868c9de560dSAlex Tomas 		 * close together
869c9de560dSAlex Tomas 		 */
870c9de560dSAlex Tomas 		if ((first_block + i) & 1) {
871c9de560dSAlex Tomas 			/* this is block of buddy */
872c9de560dSAlex Tomas 			BUG_ON(incore == NULL);
873c9de560dSAlex Tomas 			mb_debug("put buddy for group %u in page %lu/%x\n",
874c9de560dSAlex Tomas 				group, page->index, i * blocksize);
875c9de560dSAlex Tomas 			memset(data, 0xff, blocksize);
876c9de560dSAlex Tomas 			grinfo = ext4_get_group_info(sb, group);
877c9de560dSAlex Tomas 			grinfo->bb_fragments = 0;
878c9de560dSAlex Tomas 			memset(grinfo->bb_counters, 0,
879c9de560dSAlex Tomas 			       sizeof(unsigned short)*(sb->s_blocksize_bits+2));
880c9de560dSAlex Tomas 			/*
881c9de560dSAlex Tomas 			 * incore got set to the group block bitmap below
882c9de560dSAlex Tomas 			 */
8837a2fcbf7SAneesh Kumar K.V 			ext4_lock_group(sb, group);
884c9de560dSAlex Tomas 			ext4_mb_generate_buddy(sb, data, incore, group);
8857a2fcbf7SAneesh Kumar K.V 			ext4_unlock_group(sb, group);
886c9de560dSAlex Tomas 			incore = NULL;
887c9de560dSAlex Tomas 		} else {
888c9de560dSAlex Tomas 			/* this is block of bitmap */
889c9de560dSAlex Tomas 			BUG_ON(incore != NULL);
890c9de560dSAlex Tomas 			mb_debug("put bitmap for group %u in page %lu/%x\n",
891c9de560dSAlex Tomas 				group, page->index, i * blocksize);
892c9de560dSAlex Tomas 
893c9de560dSAlex Tomas 			/* see comments in ext4_mb_put_pa() */
894c9de560dSAlex Tomas 			ext4_lock_group(sb, group);
895c9de560dSAlex Tomas 			memcpy(data, bitmap, blocksize);
896c9de560dSAlex Tomas 
897c9de560dSAlex Tomas 			/* mark all preallocated blks used in in-core bitmap */
898c9de560dSAlex Tomas 			ext4_mb_generate_from_pa(sb, data, group);
8997a2fcbf7SAneesh Kumar K.V 			ext4_mb_generate_from_freelist(sb, data, group);
900c9de560dSAlex Tomas 			ext4_unlock_group(sb, group);
901c9de560dSAlex Tomas 
902c9de560dSAlex Tomas 			/* set incore so that the buddy information can be
903c9de560dSAlex Tomas 			 * generated using this
904c9de560dSAlex Tomas 			 */
905c9de560dSAlex Tomas 			incore = data;
906c9de560dSAlex Tomas 		}
907c9de560dSAlex Tomas 	}
908c9de560dSAlex Tomas 	SetPageUptodate(page);
909c9de560dSAlex Tomas 
910c9de560dSAlex Tomas out:
911c9de560dSAlex Tomas 	if (bh) {
912c9de560dSAlex Tomas 		for (i = 0; i < groups_per_page && bh[i]; i++)
913c9de560dSAlex Tomas 			brelse(bh[i]);
914c9de560dSAlex Tomas 		if (bh != &bhs)
915c9de560dSAlex Tomas 			kfree(bh);
916c9de560dSAlex Tomas 	}
917c9de560dSAlex Tomas 	return err;
918c9de560dSAlex Tomas }
919c9de560dSAlex Tomas 
9204ddfef7bSEric Sandeen static noinline_for_stack int
9214ddfef7bSEric Sandeen ext4_mb_load_buddy(struct super_block *sb, ext4_group_t group,
922c9de560dSAlex Tomas 					struct ext4_buddy *e4b)
923c9de560dSAlex Tomas {
924c9de560dSAlex Tomas 	int blocks_per_page;
925c9de560dSAlex Tomas 	int block;
926c9de560dSAlex Tomas 	int pnum;
927c9de560dSAlex Tomas 	int poff;
928c9de560dSAlex Tomas 	struct page *page;
929fdf6c7a7SShen Feng 	int ret;
930920313a7SAneesh Kumar K.V 	struct ext4_group_info *grp;
931920313a7SAneesh Kumar K.V 	struct ext4_sb_info *sbi = EXT4_SB(sb);
932920313a7SAneesh Kumar K.V 	struct inode *inode = sbi->s_buddy_cache;
933c9de560dSAlex Tomas 
934a9df9a49STheodore Ts'o 	mb_debug("load group %u\n", group);
935c9de560dSAlex Tomas 
936c9de560dSAlex Tomas 	blocks_per_page = PAGE_CACHE_SIZE / sb->s_blocksize;
937920313a7SAneesh Kumar K.V 	grp = ext4_get_group_info(sb, group);
938c9de560dSAlex Tomas 
939c9de560dSAlex Tomas 	e4b->bd_blkbits = sb->s_blocksize_bits;
940c9de560dSAlex Tomas 	e4b->bd_info = ext4_get_group_info(sb, group);
941c9de560dSAlex Tomas 	e4b->bd_sb = sb;
942c9de560dSAlex Tomas 	e4b->bd_group = group;
943c9de560dSAlex Tomas 	e4b->bd_buddy_page = NULL;
944c9de560dSAlex Tomas 	e4b->bd_bitmap_page = NULL;
945920313a7SAneesh Kumar K.V 	e4b->alloc_semp = &grp->alloc_sem;
946920313a7SAneesh Kumar K.V 
947920313a7SAneesh Kumar K.V 	/* Take the read lock on the group alloc
948920313a7SAneesh Kumar K.V 	 * sem. This would make sure a parallel
949920313a7SAneesh Kumar K.V 	 * ext4_mb_init_group happening on other
950920313a7SAneesh Kumar K.V 	 * groups mapped by the page is blocked
951920313a7SAneesh Kumar K.V 	 * till we are done with allocation
952920313a7SAneesh Kumar K.V 	 */
953920313a7SAneesh Kumar K.V 	down_read(e4b->alloc_semp);
954c9de560dSAlex Tomas 
955c9de560dSAlex Tomas 	/*
956c9de560dSAlex Tomas 	 * the buddy cache inode stores the block bitmap
957c9de560dSAlex Tomas 	 * and buddy information in consecutive blocks.
958c9de560dSAlex Tomas 	 * So for each group we need two blocks.
959c9de560dSAlex Tomas 	 */
960c9de560dSAlex Tomas 	block = group * 2;
961c9de560dSAlex Tomas 	pnum = block / blocks_per_page;
962c9de560dSAlex Tomas 	poff = block % blocks_per_page;
963c9de560dSAlex Tomas 
964c9de560dSAlex Tomas 	/* we could use find_or_create_page(), but it locks page
965c9de560dSAlex Tomas 	 * what we'd like to avoid in fast path ... */
966c9de560dSAlex Tomas 	page = find_get_page(inode->i_mapping, pnum);
967c9de560dSAlex Tomas 	if (page == NULL || !PageUptodate(page)) {
968c9de560dSAlex Tomas 		if (page)
969920313a7SAneesh Kumar K.V 			/*
970920313a7SAneesh Kumar K.V 			 * drop the page reference and try
971920313a7SAneesh Kumar K.V 			 * to get the page with lock. If we
972920313a7SAneesh Kumar K.V 			 * are not uptodate that implies
973920313a7SAneesh Kumar K.V 			 * somebody just created the page but
974920313a7SAneesh Kumar K.V 			 * is yet to initialize the same. So
975920313a7SAneesh Kumar K.V 			 * wait for it to initialize.
976920313a7SAneesh Kumar K.V 			 */
977c9de560dSAlex Tomas 			page_cache_release(page);
978c9de560dSAlex Tomas 		page = find_or_create_page(inode->i_mapping, pnum, GFP_NOFS);
979c9de560dSAlex Tomas 		if (page) {
980c9de560dSAlex Tomas 			BUG_ON(page->mapping != inode->i_mapping);
981c9de560dSAlex Tomas 			if (!PageUptodate(page)) {
982fdf6c7a7SShen Feng 				ret = ext4_mb_init_cache(page, NULL);
983fdf6c7a7SShen Feng 				if (ret) {
984fdf6c7a7SShen Feng 					unlock_page(page);
985fdf6c7a7SShen Feng 					goto err;
986fdf6c7a7SShen Feng 				}
987c9de560dSAlex Tomas 				mb_cmp_bitmaps(e4b, page_address(page) +
988c9de560dSAlex Tomas 					       (poff * sb->s_blocksize));
989c9de560dSAlex Tomas 			}
990c9de560dSAlex Tomas 			unlock_page(page);
991c9de560dSAlex Tomas 		}
992c9de560dSAlex Tomas 	}
993fdf6c7a7SShen Feng 	if (page == NULL || !PageUptodate(page)) {
994fdf6c7a7SShen Feng 		ret = -EIO;
995c9de560dSAlex Tomas 		goto err;
996fdf6c7a7SShen Feng 	}
997c9de560dSAlex Tomas 	e4b->bd_bitmap_page = page;
998c9de560dSAlex Tomas 	e4b->bd_bitmap = page_address(page) + (poff * sb->s_blocksize);
999c9de560dSAlex Tomas 	mark_page_accessed(page);
1000c9de560dSAlex Tomas 
1001c9de560dSAlex Tomas 	block++;
1002c9de560dSAlex Tomas 	pnum = block / blocks_per_page;
1003c9de560dSAlex Tomas 	poff = block % blocks_per_page;
1004c9de560dSAlex Tomas 
1005c9de560dSAlex Tomas 	page = find_get_page(inode->i_mapping, pnum);
1006c9de560dSAlex Tomas 	if (page == NULL || !PageUptodate(page)) {
1007c9de560dSAlex Tomas 		if (page)
1008c9de560dSAlex Tomas 			page_cache_release(page);
1009c9de560dSAlex Tomas 		page = find_or_create_page(inode->i_mapping, pnum, GFP_NOFS);
1010c9de560dSAlex Tomas 		if (page) {
1011c9de560dSAlex Tomas 			BUG_ON(page->mapping != inode->i_mapping);
1012fdf6c7a7SShen Feng 			if (!PageUptodate(page)) {
1013fdf6c7a7SShen Feng 				ret = ext4_mb_init_cache(page, e4b->bd_bitmap);
1014fdf6c7a7SShen Feng 				if (ret) {
1015fdf6c7a7SShen Feng 					unlock_page(page);
1016fdf6c7a7SShen Feng 					goto err;
1017fdf6c7a7SShen Feng 				}
1018fdf6c7a7SShen Feng 			}
1019c9de560dSAlex Tomas 			unlock_page(page);
1020c9de560dSAlex Tomas 		}
1021c9de560dSAlex Tomas 	}
1022fdf6c7a7SShen Feng 	if (page == NULL || !PageUptodate(page)) {
1023fdf6c7a7SShen Feng 		ret = -EIO;
1024c9de560dSAlex Tomas 		goto err;
1025fdf6c7a7SShen Feng 	}
1026c9de560dSAlex Tomas 	e4b->bd_buddy_page = page;
1027c9de560dSAlex Tomas 	e4b->bd_buddy = page_address(page) + (poff * sb->s_blocksize);
1028c9de560dSAlex Tomas 	mark_page_accessed(page);
1029c9de560dSAlex Tomas 
1030c9de560dSAlex Tomas 	BUG_ON(e4b->bd_bitmap_page == NULL);
1031c9de560dSAlex Tomas 	BUG_ON(e4b->bd_buddy_page == NULL);
1032c9de560dSAlex Tomas 
1033c9de560dSAlex Tomas 	return 0;
1034c9de560dSAlex Tomas 
1035c9de560dSAlex Tomas err:
1036c9de560dSAlex Tomas 	if (e4b->bd_bitmap_page)
1037c9de560dSAlex Tomas 		page_cache_release(e4b->bd_bitmap_page);
1038c9de560dSAlex Tomas 	if (e4b->bd_buddy_page)
1039c9de560dSAlex Tomas 		page_cache_release(e4b->bd_buddy_page);
1040c9de560dSAlex Tomas 	e4b->bd_buddy = NULL;
1041c9de560dSAlex Tomas 	e4b->bd_bitmap = NULL;
1042920313a7SAneesh Kumar K.V 
1043920313a7SAneesh Kumar K.V 	/* Done with the buddy cache */
1044920313a7SAneesh Kumar K.V 	up_read(e4b->alloc_semp);
1045fdf6c7a7SShen Feng 	return ret;
1046c9de560dSAlex Tomas }
1047c9de560dSAlex Tomas 
1048c9de560dSAlex Tomas static void ext4_mb_release_desc(struct ext4_buddy *e4b)
1049c9de560dSAlex Tomas {
1050c9de560dSAlex Tomas 	if (e4b->bd_bitmap_page)
1051c9de560dSAlex Tomas 		page_cache_release(e4b->bd_bitmap_page);
1052c9de560dSAlex Tomas 	if (e4b->bd_buddy_page)
1053c9de560dSAlex Tomas 		page_cache_release(e4b->bd_buddy_page);
1054920313a7SAneesh Kumar K.V 	/* Done with the buddy cache */
1055920313a7SAneesh Kumar K.V 	up_read(e4b->alloc_semp);
1056c9de560dSAlex Tomas }
1057c9de560dSAlex Tomas 
1058c9de560dSAlex Tomas 
1059c9de560dSAlex Tomas static int mb_find_order_for_block(struct ext4_buddy *e4b, int block)
1060c9de560dSAlex Tomas {
1061c9de560dSAlex Tomas 	int order = 1;
1062c9de560dSAlex Tomas 	void *bb;
1063c9de560dSAlex Tomas 
1064c9de560dSAlex Tomas 	BUG_ON(EXT4_MB_BITMAP(e4b) == EXT4_MB_BUDDY(e4b));
1065c9de560dSAlex Tomas 	BUG_ON(block >= (1 << (e4b->bd_blkbits + 3)));
1066c9de560dSAlex Tomas 
1067c9de560dSAlex Tomas 	bb = EXT4_MB_BUDDY(e4b);
1068c9de560dSAlex Tomas 	while (order <= e4b->bd_blkbits + 1) {
1069c9de560dSAlex Tomas 		block = block >> 1;
1070c9de560dSAlex Tomas 		if (!mb_test_bit(block, bb)) {
1071c9de560dSAlex Tomas 			/* this block is part of buddy of order 'order' */
1072c9de560dSAlex Tomas 			return order;
1073c9de560dSAlex Tomas 		}
1074c9de560dSAlex Tomas 		bb += 1 << (e4b->bd_blkbits - order);
1075c9de560dSAlex Tomas 		order++;
1076c9de560dSAlex Tomas 	}
1077c9de560dSAlex Tomas 	return 0;
1078c9de560dSAlex Tomas }
1079c9de560dSAlex Tomas 
1080c9de560dSAlex Tomas static void mb_clear_bits(spinlock_t *lock, void *bm, int cur, int len)
1081c9de560dSAlex Tomas {
1082c9de560dSAlex Tomas 	__u32 *addr;
1083c9de560dSAlex Tomas 
1084c9de560dSAlex Tomas 	len = cur + len;
1085c9de560dSAlex Tomas 	while (cur < len) {
1086c9de560dSAlex Tomas 		if ((cur & 31) == 0 && (len - cur) >= 32) {
1087c9de560dSAlex Tomas 			/* fast path: clear whole word at once */
1088c9de560dSAlex Tomas 			addr = bm + (cur >> 3);
1089c9de560dSAlex Tomas 			*addr = 0;
1090c9de560dSAlex Tomas 			cur += 32;
1091c9de560dSAlex Tomas 			continue;
1092c9de560dSAlex Tomas 		}
1093e8134b27SAneesh Kumar K.V 		if (lock)
1094c9de560dSAlex Tomas 			mb_clear_bit_atomic(lock, cur, bm);
1095e8134b27SAneesh Kumar K.V 		else
1096e8134b27SAneesh Kumar K.V 			mb_clear_bit(cur, bm);
1097c9de560dSAlex Tomas 		cur++;
1098c9de560dSAlex Tomas 	}
1099c9de560dSAlex Tomas }
1100c9de560dSAlex Tomas 
1101c9de560dSAlex Tomas static void mb_set_bits(spinlock_t *lock, void *bm, int cur, int len)
1102c9de560dSAlex Tomas {
1103c9de560dSAlex Tomas 	__u32 *addr;
1104c9de560dSAlex Tomas 
1105c9de560dSAlex Tomas 	len = cur + len;
1106c9de560dSAlex Tomas 	while (cur < len) {
1107c9de560dSAlex Tomas 		if ((cur & 31) == 0 && (len - cur) >= 32) {
1108c9de560dSAlex Tomas 			/* fast path: set whole word at once */
1109c9de560dSAlex Tomas 			addr = bm + (cur >> 3);
1110c9de560dSAlex Tomas 			*addr = 0xffffffff;
1111c9de560dSAlex Tomas 			cur += 32;
1112c9de560dSAlex Tomas 			continue;
1113c9de560dSAlex Tomas 		}
1114e8134b27SAneesh Kumar K.V 		if (lock)
1115c9de560dSAlex Tomas 			mb_set_bit_atomic(lock, cur, bm);
1116e8134b27SAneesh Kumar K.V 		else
1117e8134b27SAneesh Kumar K.V 			mb_set_bit(cur, bm);
1118c9de560dSAlex Tomas 		cur++;
1119c9de560dSAlex Tomas 	}
1120c9de560dSAlex Tomas }
1121c9de560dSAlex Tomas 
11227e5a8cddSShen Feng static void mb_free_blocks(struct inode *inode, struct ext4_buddy *e4b,
1123c9de560dSAlex Tomas 			  int first, int count)
1124c9de560dSAlex Tomas {
1125c9de560dSAlex Tomas 	int block = 0;
1126c9de560dSAlex Tomas 	int max = 0;
1127c9de560dSAlex Tomas 	int order;
1128c9de560dSAlex Tomas 	void *buddy;
1129c9de560dSAlex Tomas 	void *buddy2;
1130c9de560dSAlex Tomas 	struct super_block *sb = e4b->bd_sb;
1131c9de560dSAlex Tomas 
1132c9de560dSAlex Tomas 	BUG_ON(first + count > (sb->s_blocksize << 3));
1133c9de560dSAlex Tomas 	BUG_ON(!ext4_is_group_locked(sb, e4b->bd_group));
1134c9de560dSAlex Tomas 	mb_check_buddy(e4b);
1135c9de560dSAlex Tomas 	mb_free_blocks_double(inode, e4b, first, count);
1136c9de560dSAlex Tomas 
1137c9de560dSAlex Tomas 	e4b->bd_info->bb_free += count;
1138c9de560dSAlex Tomas 	if (first < e4b->bd_info->bb_first_free)
1139c9de560dSAlex Tomas 		e4b->bd_info->bb_first_free = first;
1140c9de560dSAlex Tomas 
1141c9de560dSAlex Tomas 	/* let's maintain fragments counter */
1142c9de560dSAlex Tomas 	if (first != 0)
1143c9de560dSAlex Tomas 		block = !mb_test_bit(first - 1, EXT4_MB_BITMAP(e4b));
1144c9de560dSAlex Tomas 	if (first + count < EXT4_SB(sb)->s_mb_maxs[0])
1145c9de560dSAlex Tomas 		max = !mb_test_bit(first + count, EXT4_MB_BITMAP(e4b));
1146c9de560dSAlex Tomas 	if (block && max)
1147c9de560dSAlex Tomas 		e4b->bd_info->bb_fragments--;
1148c9de560dSAlex Tomas 	else if (!block && !max)
1149c9de560dSAlex Tomas 		e4b->bd_info->bb_fragments++;
1150c9de560dSAlex Tomas 
1151c9de560dSAlex Tomas 	/* let's maintain buddy itself */
1152c9de560dSAlex Tomas 	while (count-- > 0) {
1153c9de560dSAlex Tomas 		block = first++;
1154c9de560dSAlex Tomas 		order = 0;
1155c9de560dSAlex Tomas 
1156c9de560dSAlex Tomas 		if (!mb_test_bit(block, EXT4_MB_BITMAP(e4b))) {
1157c9de560dSAlex Tomas 			ext4_fsblk_t blocknr;
1158c9de560dSAlex Tomas 			blocknr = e4b->bd_group * EXT4_BLOCKS_PER_GROUP(sb);
1159c9de560dSAlex Tomas 			blocknr += block;
1160c9de560dSAlex Tomas 			blocknr +=
1161c9de560dSAlex Tomas 			    le32_to_cpu(EXT4_SB(sb)->s_es->s_first_data_block);
11625d1b1b3fSAneesh Kumar K.V 			ext4_grp_locked_error(sb, e4b->bd_group,
11635d1b1b3fSAneesh Kumar K.V 				   __func__, "double-free of inode"
1164a9df9a49STheodore Ts'o 				   " %lu's block %llu(bit %u in group %u)",
1165c9de560dSAlex Tomas 				   inode ? inode->i_ino : 0, blocknr, block,
1166c9de560dSAlex Tomas 				   e4b->bd_group);
1167c9de560dSAlex Tomas 		}
1168c9de560dSAlex Tomas 		mb_clear_bit(block, EXT4_MB_BITMAP(e4b));
1169c9de560dSAlex Tomas 		e4b->bd_info->bb_counters[order]++;
1170c9de560dSAlex Tomas 
1171c9de560dSAlex Tomas 		/* start of the buddy */
1172c9de560dSAlex Tomas 		buddy = mb_find_buddy(e4b, order, &max);
1173c9de560dSAlex Tomas 
1174c9de560dSAlex Tomas 		do {
1175c9de560dSAlex Tomas 			block &= ~1UL;
1176c9de560dSAlex Tomas 			if (mb_test_bit(block, buddy) ||
1177c9de560dSAlex Tomas 					mb_test_bit(block + 1, buddy))
1178c9de560dSAlex Tomas 				break;
1179c9de560dSAlex Tomas 
1180c9de560dSAlex Tomas 			/* both the buddies are free, try to coalesce them */
1181c9de560dSAlex Tomas 			buddy2 = mb_find_buddy(e4b, order + 1, &max);
1182c9de560dSAlex Tomas 
1183c9de560dSAlex Tomas 			if (!buddy2)
1184c9de560dSAlex Tomas 				break;
1185c9de560dSAlex Tomas 
1186c9de560dSAlex Tomas 			if (order > 0) {
1187c9de560dSAlex Tomas 				/* for special purposes, we don't set
1188c9de560dSAlex Tomas 				 * free bits in bitmap */
1189c9de560dSAlex Tomas 				mb_set_bit(block, buddy);
1190c9de560dSAlex Tomas 				mb_set_bit(block + 1, buddy);
1191c9de560dSAlex Tomas 			}
1192c9de560dSAlex Tomas 			e4b->bd_info->bb_counters[order]--;
1193c9de560dSAlex Tomas 			e4b->bd_info->bb_counters[order]--;
1194c9de560dSAlex Tomas 
1195c9de560dSAlex Tomas 			block = block >> 1;
1196c9de560dSAlex Tomas 			order++;
1197c9de560dSAlex Tomas 			e4b->bd_info->bb_counters[order]++;
1198c9de560dSAlex Tomas 
1199c9de560dSAlex Tomas 			mb_clear_bit(block, buddy2);
1200c9de560dSAlex Tomas 			buddy = buddy2;
1201c9de560dSAlex Tomas 		} while (1);
1202c9de560dSAlex Tomas 	}
1203c9de560dSAlex Tomas 	mb_check_buddy(e4b);
1204c9de560dSAlex Tomas }
1205c9de560dSAlex Tomas 
1206c9de560dSAlex Tomas static int mb_find_extent(struct ext4_buddy *e4b, int order, int block,
1207c9de560dSAlex Tomas 				int needed, struct ext4_free_extent *ex)
1208c9de560dSAlex Tomas {
1209c9de560dSAlex Tomas 	int next = block;
1210c9de560dSAlex Tomas 	int max;
1211c9de560dSAlex Tomas 	int ord;
1212c9de560dSAlex Tomas 	void *buddy;
1213c9de560dSAlex Tomas 
1214c9de560dSAlex Tomas 	BUG_ON(!ext4_is_group_locked(e4b->bd_sb, e4b->bd_group));
1215c9de560dSAlex Tomas 	BUG_ON(ex == NULL);
1216c9de560dSAlex Tomas 
1217c9de560dSAlex Tomas 	buddy = mb_find_buddy(e4b, order, &max);
1218c9de560dSAlex Tomas 	BUG_ON(buddy == NULL);
1219c9de560dSAlex Tomas 	BUG_ON(block >= max);
1220c9de560dSAlex Tomas 	if (mb_test_bit(block, buddy)) {
1221c9de560dSAlex Tomas 		ex->fe_len = 0;
1222c9de560dSAlex Tomas 		ex->fe_start = 0;
1223c9de560dSAlex Tomas 		ex->fe_group = 0;
1224c9de560dSAlex Tomas 		return 0;
1225c9de560dSAlex Tomas 	}
1226c9de560dSAlex Tomas 
1227c9de560dSAlex Tomas 	/* FIXME dorp order completely ? */
1228c9de560dSAlex Tomas 	if (likely(order == 0)) {
1229c9de560dSAlex Tomas 		/* find actual order */
1230c9de560dSAlex Tomas 		order = mb_find_order_for_block(e4b, block);
1231c9de560dSAlex Tomas 		block = block >> order;
1232c9de560dSAlex Tomas 	}
1233c9de560dSAlex Tomas 
1234c9de560dSAlex Tomas 	ex->fe_len = 1 << order;
1235c9de560dSAlex Tomas 	ex->fe_start = block << order;
1236c9de560dSAlex Tomas 	ex->fe_group = e4b->bd_group;
1237c9de560dSAlex Tomas 
1238c9de560dSAlex Tomas 	/* calc difference from given start */
1239c9de560dSAlex Tomas 	next = next - ex->fe_start;
1240c9de560dSAlex Tomas 	ex->fe_len -= next;
1241c9de560dSAlex Tomas 	ex->fe_start += next;
1242c9de560dSAlex Tomas 
1243c9de560dSAlex Tomas 	while (needed > ex->fe_len &&
1244c9de560dSAlex Tomas 	       (buddy = mb_find_buddy(e4b, order, &max))) {
1245c9de560dSAlex Tomas 
1246c9de560dSAlex Tomas 		if (block + 1 >= max)
1247c9de560dSAlex Tomas 			break;
1248c9de560dSAlex Tomas 
1249c9de560dSAlex Tomas 		next = (block + 1) * (1 << order);
1250c9de560dSAlex Tomas 		if (mb_test_bit(next, EXT4_MB_BITMAP(e4b)))
1251c9de560dSAlex Tomas 			break;
1252c9de560dSAlex Tomas 
1253c9de560dSAlex Tomas 		ord = mb_find_order_for_block(e4b, next);
1254c9de560dSAlex Tomas 
1255c9de560dSAlex Tomas 		order = ord;
1256c9de560dSAlex Tomas 		block = next >> order;
1257c9de560dSAlex Tomas 		ex->fe_len += 1 << order;
1258c9de560dSAlex Tomas 	}
1259c9de560dSAlex Tomas 
1260c9de560dSAlex Tomas 	BUG_ON(ex->fe_start + ex->fe_len > (1 << (e4b->bd_blkbits + 3)));
1261c9de560dSAlex Tomas 	return ex->fe_len;
1262c9de560dSAlex Tomas }
1263c9de560dSAlex Tomas 
1264c9de560dSAlex Tomas static int mb_mark_used(struct ext4_buddy *e4b, struct ext4_free_extent *ex)
1265c9de560dSAlex Tomas {
1266c9de560dSAlex Tomas 	int ord;
1267c9de560dSAlex Tomas 	int mlen = 0;
1268c9de560dSAlex Tomas 	int max = 0;
1269c9de560dSAlex Tomas 	int cur;
1270c9de560dSAlex Tomas 	int start = ex->fe_start;
1271c9de560dSAlex Tomas 	int len = ex->fe_len;
1272c9de560dSAlex Tomas 	unsigned ret = 0;
1273c9de560dSAlex Tomas 	int len0 = len;
1274c9de560dSAlex Tomas 	void *buddy;
1275c9de560dSAlex Tomas 
1276c9de560dSAlex Tomas 	BUG_ON(start + len > (e4b->bd_sb->s_blocksize << 3));
1277c9de560dSAlex Tomas 	BUG_ON(e4b->bd_group != ex->fe_group);
1278c9de560dSAlex Tomas 	BUG_ON(!ext4_is_group_locked(e4b->bd_sb, e4b->bd_group));
1279c9de560dSAlex Tomas 	mb_check_buddy(e4b);
1280c9de560dSAlex Tomas 	mb_mark_used_double(e4b, start, len);
1281c9de560dSAlex Tomas 
1282c9de560dSAlex Tomas 	e4b->bd_info->bb_free -= len;
1283c9de560dSAlex Tomas 	if (e4b->bd_info->bb_first_free == start)
1284c9de560dSAlex Tomas 		e4b->bd_info->bb_first_free += len;
1285c9de560dSAlex Tomas 
1286c9de560dSAlex Tomas 	/* let's maintain fragments counter */
1287c9de560dSAlex Tomas 	if (start != 0)
1288c9de560dSAlex Tomas 		mlen = !mb_test_bit(start - 1, EXT4_MB_BITMAP(e4b));
1289c9de560dSAlex Tomas 	if (start + len < EXT4_SB(e4b->bd_sb)->s_mb_maxs[0])
1290c9de560dSAlex Tomas 		max = !mb_test_bit(start + len, EXT4_MB_BITMAP(e4b));
1291c9de560dSAlex Tomas 	if (mlen && max)
1292c9de560dSAlex Tomas 		e4b->bd_info->bb_fragments++;
1293c9de560dSAlex Tomas 	else if (!mlen && !max)
1294c9de560dSAlex Tomas 		e4b->bd_info->bb_fragments--;
1295c9de560dSAlex Tomas 
1296c9de560dSAlex Tomas 	/* let's maintain buddy itself */
1297c9de560dSAlex Tomas 	while (len) {
1298c9de560dSAlex Tomas 		ord = mb_find_order_for_block(e4b, start);
1299c9de560dSAlex Tomas 
1300c9de560dSAlex Tomas 		if (((start >> ord) << ord) == start && len >= (1 << ord)) {
1301c9de560dSAlex Tomas 			/* the whole chunk may be allocated at once! */
1302c9de560dSAlex Tomas 			mlen = 1 << ord;
1303c9de560dSAlex Tomas 			buddy = mb_find_buddy(e4b, ord, &max);
1304c9de560dSAlex Tomas 			BUG_ON((start >> ord) >= max);
1305c9de560dSAlex Tomas 			mb_set_bit(start >> ord, buddy);
1306c9de560dSAlex Tomas 			e4b->bd_info->bb_counters[ord]--;
1307c9de560dSAlex Tomas 			start += mlen;
1308c9de560dSAlex Tomas 			len -= mlen;
1309c9de560dSAlex Tomas 			BUG_ON(len < 0);
1310c9de560dSAlex Tomas 			continue;
1311c9de560dSAlex Tomas 		}
1312c9de560dSAlex Tomas 
1313c9de560dSAlex Tomas 		/* store for history */
1314c9de560dSAlex Tomas 		if (ret == 0)
1315c9de560dSAlex Tomas 			ret = len | (ord << 16);
1316c9de560dSAlex Tomas 
1317c9de560dSAlex Tomas 		/* we have to split large buddy */
1318c9de560dSAlex Tomas 		BUG_ON(ord <= 0);
1319c9de560dSAlex Tomas 		buddy = mb_find_buddy(e4b, ord, &max);
1320c9de560dSAlex Tomas 		mb_set_bit(start >> ord, buddy);
1321c9de560dSAlex Tomas 		e4b->bd_info->bb_counters[ord]--;
1322c9de560dSAlex Tomas 
1323c9de560dSAlex Tomas 		ord--;
1324c9de560dSAlex Tomas 		cur = (start >> ord) & ~1U;
1325c9de560dSAlex Tomas 		buddy = mb_find_buddy(e4b, ord, &max);
1326c9de560dSAlex Tomas 		mb_clear_bit(cur, buddy);
1327c9de560dSAlex Tomas 		mb_clear_bit(cur + 1, buddy);
1328c9de560dSAlex Tomas 		e4b->bd_info->bb_counters[ord]++;
1329c9de560dSAlex Tomas 		e4b->bd_info->bb_counters[ord]++;
1330c9de560dSAlex Tomas 	}
1331c9de560dSAlex Tomas 
1332c9de560dSAlex Tomas 	mb_set_bits(sb_bgl_lock(EXT4_SB(e4b->bd_sb), ex->fe_group),
1333c9de560dSAlex Tomas 			EXT4_MB_BITMAP(e4b), ex->fe_start, len0);
1334c9de560dSAlex Tomas 	mb_check_buddy(e4b);
1335c9de560dSAlex Tomas 
1336c9de560dSAlex Tomas 	return ret;
1337c9de560dSAlex Tomas }
1338c9de560dSAlex Tomas 
1339c9de560dSAlex Tomas /*
1340c9de560dSAlex Tomas  * Must be called under group lock!
1341c9de560dSAlex Tomas  */
1342c9de560dSAlex Tomas static void ext4_mb_use_best_found(struct ext4_allocation_context *ac,
1343c9de560dSAlex Tomas 					struct ext4_buddy *e4b)
1344c9de560dSAlex Tomas {
1345c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(ac->ac_sb);
1346c9de560dSAlex Tomas 	int ret;
1347c9de560dSAlex Tomas 
1348c9de560dSAlex Tomas 	BUG_ON(ac->ac_b_ex.fe_group != e4b->bd_group);
1349c9de560dSAlex Tomas 	BUG_ON(ac->ac_status == AC_STATUS_FOUND);
1350c9de560dSAlex Tomas 
1351c9de560dSAlex Tomas 	ac->ac_b_ex.fe_len = min(ac->ac_b_ex.fe_len, ac->ac_g_ex.fe_len);
1352c9de560dSAlex Tomas 	ac->ac_b_ex.fe_logical = ac->ac_g_ex.fe_logical;
1353c9de560dSAlex Tomas 	ret = mb_mark_used(e4b, &ac->ac_b_ex);
1354c9de560dSAlex Tomas 
1355c9de560dSAlex Tomas 	/* preallocation can change ac_b_ex, thus we store actually
1356c9de560dSAlex Tomas 	 * allocated blocks for history */
1357c9de560dSAlex Tomas 	ac->ac_f_ex = ac->ac_b_ex;
1358c9de560dSAlex Tomas 
1359c9de560dSAlex Tomas 	ac->ac_status = AC_STATUS_FOUND;
1360c9de560dSAlex Tomas 	ac->ac_tail = ret & 0xffff;
1361c9de560dSAlex Tomas 	ac->ac_buddy = ret >> 16;
1362c9de560dSAlex Tomas 
1363c3a326a6SAneesh Kumar K.V 	/*
1364c3a326a6SAneesh Kumar K.V 	 * take the page reference. We want the page to be pinned
1365c3a326a6SAneesh Kumar K.V 	 * so that we don't get a ext4_mb_init_cache_call for this
1366c3a326a6SAneesh Kumar K.V 	 * group until we update the bitmap. That would mean we
1367c3a326a6SAneesh Kumar K.V 	 * double allocate blocks. The reference is dropped
1368c3a326a6SAneesh Kumar K.V 	 * in ext4_mb_release_context
1369c3a326a6SAneesh Kumar K.V 	 */
1370c9de560dSAlex Tomas 	ac->ac_bitmap_page = e4b->bd_bitmap_page;
1371c9de560dSAlex Tomas 	get_page(ac->ac_bitmap_page);
1372c9de560dSAlex Tomas 	ac->ac_buddy_page = e4b->bd_buddy_page;
1373c9de560dSAlex Tomas 	get_page(ac->ac_buddy_page);
1374c9de560dSAlex Tomas 
1375c9de560dSAlex Tomas 	/* store last allocated for subsequent stream allocation */
1376c9de560dSAlex Tomas 	if ((ac->ac_flags & EXT4_MB_HINT_DATA)) {
1377c9de560dSAlex Tomas 		spin_lock(&sbi->s_md_lock);
1378c9de560dSAlex Tomas 		sbi->s_mb_last_group = ac->ac_f_ex.fe_group;
1379c9de560dSAlex Tomas 		sbi->s_mb_last_start = ac->ac_f_ex.fe_start;
1380c9de560dSAlex Tomas 		spin_unlock(&sbi->s_md_lock);
1381c9de560dSAlex Tomas 	}
1382c9de560dSAlex Tomas }
1383c9de560dSAlex Tomas 
1384c9de560dSAlex Tomas /*
1385c9de560dSAlex Tomas  * regular allocator, for general purposes allocation
1386c9de560dSAlex Tomas  */
1387c9de560dSAlex Tomas 
1388c9de560dSAlex Tomas static void ext4_mb_check_limits(struct ext4_allocation_context *ac,
1389c9de560dSAlex Tomas 					struct ext4_buddy *e4b,
1390c9de560dSAlex Tomas 					int finish_group)
1391c9de560dSAlex Tomas {
1392c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(ac->ac_sb);
1393c9de560dSAlex Tomas 	struct ext4_free_extent *bex = &ac->ac_b_ex;
1394c9de560dSAlex Tomas 	struct ext4_free_extent *gex = &ac->ac_g_ex;
1395c9de560dSAlex Tomas 	struct ext4_free_extent ex;
1396c9de560dSAlex Tomas 	int max;
1397c9de560dSAlex Tomas 
1398032115fcSAneesh Kumar K.V 	if (ac->ac_status == AC_STATUS_FOUND)
1399032115fcSAneesh Kumar K.V 		return;
1400c9de560dSAlex Tomas 	/*
1401c9de560dSAlex Tomas 	 * We don't want to scan for a whole year
1402c9de560dSAlex Tomas 	 */
1403c9de560dSAlex Tomas 	if (ac->ac_found > sbi->s_mb_max_to_scan &&
1404c9de560dSAlex Tomas 			!(ac->ac_flags & EXT4_MB_HINT_FIRST)) {
1405c9de560dSAlex Tomas 		ac->ac_status = AC_STATUS_BREAK;
1406c9de560dSAlex Tomas 		return;
1407c9de560dSAlex Tomas 	}
1408c9de560dSAlex Tomas 
1409c9de560dSAlex Tomas 	/*
1410c9de560dSAlex Tomas 	 * Haven't found good chunk so far, let's continue
1411c9de560dSAlex Tomas 	 */
1412c9de560dSAlex Tomas 	if (bex->fe_len < gex->fe_len)
1413c9de560dSAlex Tomas 		return;
1414c9de560dSAlex Tomas 
1415c9de560dSAlex Tomas 	if ((finish_group || ac->ac_found > sbi->s_mb_min_to_scan)
1416c9de560dSAlex Tomas 			&& bex->fe_group == e4b->bd_group) {
1417c9de560dSAlex Tomas 		/* recheck chunk's availability - we don't know
1418c9de560dSAlex Tomas 		 * when it was found (within this lock-unlock
1419c9de560dSAlex Tomas 		 * period or not) */
1420c9de560dSAlex Tomas 		max = mb_find_extent(e4b, 0, bex->fe_start, gex->fe_len, &ex);
1421c9de560dSAlex Tomas 		if (max >= gex->fe_len) {
1422c9de560dSAlex Tomas 			ext4_mb_use_best_found(ac, e4b);
1423c9de560dSAlex Tomas 			return;
1424c9de560dSAlex Tomas 		}
1425c9de560dSAlex Tomas 	}
1426c9de560dSAlex Tomas }
1427c9de560dSAlex Tomas 
1428c9de560dSAlex Tomas /*
1429c9de560dSAlex Tomas  * The routine checks whether found extent is good enough. If it is,
1430c9de560dSAlex Tomas  * then the extent gets marked used and flag is set to the context
1431c9de560dSAlex Tomas  * to stop scanning. Otherwise, the extent is compared with the
1432c9de560dSAlex Tomas  * previous found extent and if new one is better, then it's stored
1433c9de560dSAlex Tomas  * in the context. Later, the best found extent will be used, if
1434c9de560dSAlex Tomas  * mballoc can't find good enough extent.
1435c9de560dSAlex Tomas  *
1436c9de560dSAlex Tomas  * FIXME: real allocation policy is to be designed yet!
1437c9de560dSAlex Tomas  */
1438c9de560dSAlex Tomas static void ext4_mb_measure_extent(struct ext4_allocation_context *ac,
1439c9de560dSAlex Tomas 					struct ext4_free_extent *ex,
1440c9de560dSAlex Tomas 					struct ext4_buddy *e4b)
1441c9de560dSAlex Tomas {
1442c9de560dSAlex Tomas 	struct ext4_free_extent *bex = &ac->ac_b_ex;
1443c9de560dSAlex Tomas 	struct ext4_free_extent *gex = &ac->ac_g_ex;
1444c9de560dSAlex Tomas 
1445c9de560dSAlex Tomas 	BUG_ON(ex->fe_len <= 0);
1446c9de560dSAlex Tomas 	BUG_ON(ex->fe_len >= EXT4_BLOCKS_PER_GROUP(ac->ac_sb));
1447c9de560dSAlex Tomas 	BUG_ON(ex->fe_start >= EXT4_BLOCKS_PER_GROUP(ac->ac_sb));
1448c9de560dSAlex Tomas 	BUG_ON(ac->ac_status != AC_STATUS_CONTINUE);
1449c9de560dSAlex Tomas 
1450c9de560dSAlex Tomas 	ac->ac_found++;
1451c9de560dSAlex Tomas 
1452c9de560dSAlex Tomas 	/*
1453c9de560dSAlex Tomas 	 * The special case - take what you catch first
1454c9de560dSAlex Tomas 	 */
1455c9de560dSAlex Tomas 	if (unlikely(ac->ac_flags & EXT4_MB_HINT_FIRST)) {
1456c9de560dSAlex Tomas 		*bex = *ex;
1457c9de560dSAlex Tomas 		ext4_mb_use_best_found(ac, e4b);
1458c9de560dSAlex Tomas 		return;
1459c9de560dSAlex Tomas 	}
1460c9de560dSAlex Tomas 
1461c9de560dSAlex Tomas 	/*
1462c9de560dSAlex Tomas 	 * Let's check whether the chuck is good enough
1463c9de560dSAlex Tomas 	 */
1464c9de560dSAlex Tomas 	if (ex->fe_len == gex->fe_len) {
1465c9de560dSAlex Tomas 		*bex = *ex;
1466c9de560dSAlex Tomas 		ext4_mb_use_best_found(ac, e4b);
1467c9de560dSAlex Tomas 		return;
1468c9de560dSAlex Tomas 	}
1469c9de560dSAlex Tomas 
1470c9de560dSAlex Tomas 	/*
1471c9de560dSAlex Tomas 	 * If this is first found extent, just store it in the context
1472c9de560dSAlex Tomas 	 */
1473c9de560dSAlex Tomas 	if (bex->fe_len == 0) {
1474c9de560dSAlex Tomas 		*bex = *ex;
1475c9de560dSAlex Tomas 		return;
1476c9de560dSAlex Tomas 	}
1477c9de560dSAlex Tomas 
1478c9de560dSAlex Tomas 	/*
1479c9de560dSAlex Tomas 	 * If new found extent is better, store it in the context
1480c9de560dSAlex Tomas 	 */
1481c9de560dSAlex Tomas 	if (bex->fe_len < gex->fe_len) {
1482c9de560dSAlex Tomas 		/* if the request isn't satisfied, any found extent
1483c9de560dSAlex Tomas 		 * larger than previous best one is better */
1484c9de560dSAlex Tomas 		if (ex->fe_len > bex->fe_len)
1485c9de560dSAlex Tomas 			*bex = *ex;
1486c9de560dSAlex Tomas 	} else if (ex->fe_len > gex->fe_len) {
1487c9de560dSAlex Tomas 		/* if the request is satisfied, then we try to find
1488c9de560dSAlex Tomas 		 * an extent that still satisfy the request, but is
1489c9de560dSAlex Tomas 		 * smaller than previous one */
1490c9de560dSAlex Tomas 		if (ex->fe_len < bex->fe_len)
1491c9de560dSAlex Tomas 			*bex = *ex;
1492c9de560dSAlex Tomas 	}
1493c9de560dSAlex Tomas 
1494c9de560dSAlex Tomas 	ext4_mb_check_limits(ac, e4b, 0);
1495c9de560dSAlex Tomas }
1496c9de560dSAlex Tomas 
1497c9de560dSAlex Tomas static int ext4_mb_try_best_found(struct ext4_allocation_context *ac,
1498c9de560dSAlex Tomas 					struct ext4_buddy *e4b)
1499c9de560dSAlex Tomas {
1500c9de560dSAlex Tomas 	struct ext4_free_extent ex = ac->ac_b_ex;
1501c9de560dSAlex Tomas 	ext4_group_t group = ex.fe_group;
1502c9de560dSAlex Tomas 	int max;
1503c9de560dSAlex Tomas 	int err;
1504c9de560dSAlex Tomas 
1505c9de560dSAlex Tomas 	BUG_ON(ex.fe_len <= 0);
1506c9de560dSAlex Tomas 	err = ext4_mb_load_buddy(ac->ac_sb, group, e4b);
1507c9de560dSAlex Tomas 	if (err)
1508c9de560dSAlex Tomas 		return err;
1509c9de560dSAlex Tomas 
1510c9de560dSAlex Tomas 	ext4_lock_group(ac->ac_sb, group);
1511c9de560dSAlex Tomas 	max = mb_find_extent(e4b, 0, ex.fe_start, ex.fe_len, &ex);
1512c9de560dSAlex Tomas 
1513c9de560dSAlex Tomas 	if (max > 0) {
1514c9de560dSAlex Tomas 		ac->ac_b_ex = ex;
1515c9de560dSAlex Tomas 		ext4_mb_use_best_found(ac, e4b);
1516c9de560dSAlex Tomas 	}
1517c9de560dSAlex Tomas 
1518c9de560dSAlex Tomas 	ext4_unlock_group(ac->ac_sb, group);
1519c9de560dSAlex Tomas 	ext4_mb_release_desc(e4b);
1520c9de560dSAlex Tomas 
1521c9de560dSAlex Tomas 	return 0;
1522c9de560dSAlex Tomas }
1523c9de560dSAlex Tomas 
1524c9de560dSAlex Tomas static int ext4_mb_find_by_goal(struct ext4_allocation_context *ac,
1525c9de560dSAlex Tomas 				struct ext4_buddy *e4b)
1526c9de560dSAlex Tomas {
1527c9de560dSAlex Tomas 	ext4_group_t group = ac->ac_g_ex.fe_group;
1528c9de560dSAlex Tomas 	int max;
1529c9de560dSAlex Tomas 	int err;
1530c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(ac->ac_sb);
1531c9de560dSAlex Tomas 	struct ext4_super_block *es = sbi->s_es;
1532c9de560dSAlex Tomas 	struct ext4_free_extent ex;
1533c9de560dSAlex Tomas 
1534c9de560dSAlex Tomas 	if (!(ac->ac_flags & EXT4_MB_HINT_TRY_GOAL))
1535c9de560dSAlex Tomas 		return 0;
1536c9de560dSAlex Tomas 
1537c9de560dSAlex Tomas 	err = ext4_mb_load_buddy(ac->ac_sb, group, e4b);
1538c9de560dSAlex Tomas 	if (err)
1539c9de560dSAlex Tomas 		return err;
1540c9de560dSAlex Tomas 
1541c9de560dSAlex Tomas 	ext4_lock_group(ac->ac_sb, group);
1542c9de560dSAlex Tomas 	max = mb_find_extent(e4b, 0, ac->ac_g_ex.fe_start,
1543c9de560dSAlex Tomas 			     ac->ac_g_ex.fe_len, &ex);
1544c9de560dSAlex Tomas 
1545c9de560dSAlex Tomas 	if (max >= ac->ac_g_ex.fe_len && ac->ac_g_ex.fe_len == sbi->s_stripe) {
1546c9de560dSAlex Tomas 		ext4_fsblk_t start;
1547c9de560dSAlex Tomas 
1548c9de560dSAlex Tomas 		start = (e4b->bd_group * EXT4_BLOCKS_PER_GROUP(ac->ac_sb)) +
1549c9de560dSAlex Tomas 			ex.fe_start + le32_to_cpu(es->s_first_data_block);
1550c9de560dSAlex Tomas 		/* use do_div to get remainder (would be 64-bit modulo) */
1551c9de560dSAlex Tomas 		if (do_div(start, sbi->s_stripe) == 0) {
1552c9de560dSAlex Tomas 			ac->ac_found++;
1553c9de560dSAlex Tomas 			ac->ac_b_ex = ex;
1554c9de560dSAlex Tomas 			ext4_mb_use_best_found(ac, e4b);
1555c9de560dSAlex Tomas 		}
1556c9de560dSAlex Tomas 	} else if (max >= ac->ac_g_ex.fe_len) {
1557c9de560dSAlex Tomas 		BUG_ON(ex.fe_len <= 0);
1558c9de560dSAlex Tomas 		BUG_ON(ex.fe_group != ac->ac_g_ex.fe_group);
1559c9de560dSAlex Tomas 		BUG_ON(ex.fe_start != ac->ac_g_ex.fe_start);
1560c9de560dSAlex Tomas 		ac->ac_found++;
1561c9de560dSAlex Tomas 		ac->ac_b_ex = ex;
1562c9de560dSAlex Tomas 		ext4_mb_use_best_found(ac, e4b);
1563c9de560dSAlex Tomas 	} else if (max > 0 && (ac->ac_flags & EXT4_MB_HINT_MERGE)) {
1564c9de560dSAlex Tomas 		/* Sometimes, caller may want to merge even small
1565c9de560dSAlex Tomas 		 * number of blocks to an existing extent */
1566c9de560dSAlex Tomas 		BUG_ON(ex.fe_len <= 0);
1567c9de560dSAlex Tomas 		BUG_ON(ex.fe_group != ac->ac_g_ex.fe_group);
1568c9de560dSAlex Tomas 		BUG_ON(ex.fe_start != ac->ac_g_ex.fe_start);
1569c9de560dSAlex Tomas 		ac->ac_found++;
1570c9de560dSAlex Tomas 		ac->ac_b_ex = ex;
1571c9de560dSAlex Tomas 		ext4_mb_use_best_found(ac, e4b);
1572c9de560dSAlex Tomas 	}
1573c9de560dSAlex Tomas 	ext4_unlock_group(ac->ac_sb, group);
1574c9de560dSAlex Tomas 	ext4_mb_release_desc(e4b);
1575c9de560dSAlex Tomas 
1576c9de560dSAlex Tomas 	return 0;
1577c9de560dSAlex Tomas }
1578c9de560dSAlex Tomas 
1579c9de560dSAlex Tomas /*
1580c9de560dSAlex Tomas  * The routine scans buddy structures (not bitmap!) from given order
1581c9de560dSAlex Tomas  * to max order and tries to find big enough chunk to satisfy the req
1582c9de560dSAlex Tomas  */
1583c9de560dSAlex Tomas static void ext4_mb_simple_scan_group(struct ext4_allocation_context *ac,
1584c9de560dSAlex Tomas 					struct ext4_buddy *e4b)
1585c9de560dSAlex Tomas {
1586c9de560dSAlex Tomas 	struct super_block *sb = ac->ac_sb;
1587c9de560dSAlex Tomas 	struct ext4_group_info *grp = e4b->bd_info;
1588c9de560dSAlex Tomas 	void *buddy;
1589c9de560dSAlex Tomas 	int i;
1590c9de560dSAlex Tomas 	int k;
1591c9de560dSAlex Tomas 	int max;
1592c9de560dSAlex Tomas 
1593c9de560dSAlex Tomas 	BUG_ON(ac->ac_2order <= 0);
1594c9de560dSAlex Tomas 	for (i = ac->ac_2order; i <= sb->s_blocksize_bits + 1; i++) {
1595c9de560dSAlex Tomas 		if (grp->bb_counters[i] == 0)
1596c9de560dSAlex Tomas 			continue;
1597c9de560dSAlex Tomas 
1598c9de560dSAlex Tomas 		buddy = mb_find_buddy(e4b, i, &max);
1599c9de560dSAlex Tomas 		BUG_ON(buddy == NULL);
1600c9de560dSAlex Tomas 
1601ffad0a44SAneesh Kumar K.V 		k = mb_find_next_zero_bit(buddy, max, 0);
1602c9de560dSAlex Tomas 		BUG_ON(k >= max);
1603c9de560dSAlex Tomas 
1604c9de560dSAlex Tomas 		ac->ac_found++;
1605c9de560dSAlex Tomas 
1606c9de560dSAlex Tomas 		ac->ac_b_ex.fe_len = 1 << i;
1607c9de560dSAlex Tomas 		ac->ac_b_ex.fe_start = k << i;
1608c9de560dSAlex Tomas 		ac->ac_b_ex.fe_group = e4b->bd_group;
1609c9de560dSAlex Tomas 
1610c9de560dSAlex Tomas 		ext4_mb_use_best_found(ac, e4b);
1611c9de560dSAlex Tomas 
1612c9de560dSAlex Tomas 		BUG_ON(ac->ac_b_ex.fe_len != ac->ac_g_ex.fe_len);
1613c9de560dSAlex Tomas 
1614c9de560dSAlex Tomas 		if (EXT4_SB(sb)->s_mb_stats)
1615c9de560dSAlex Tomas 			atomic_inc(&EXT4_SB(sb)->s_bal_2orders);
1616c9de560dSAlex Tomas 
1617c9de560dSAlex Tomas 		break;
1618c9de560dSAlex Tomas 	}
1619c9de560dSAlex Tomas }
1620c9de560dSAlex Tomas 
1621c9de560dSAlex Tomas /*
1622c9de560dSAlex Tomas  * The routine scans the group and measures all found extents.
1623c9de560dSAlex Tomas  * In order to optimize scanning, caller must pass number of
1624c9de560dSAlex Tomas  * free blocks in the group, so the routine can know upper limit.
1625c9de560dSAlex Tomas  */
1626c9de560dSAlex Tomas static void ext4_mb_complex_scan_group(struct ext4_allocation_context *ac,
1627c9de560dSAlex Tomas 					struct ext4_buddy *e4b)
1628c9de560dSAlex Tomas {
1629c9de560dSAlex Tomas 	struct super_block *sb = ac->ac_sb;
1630c9de560dSAlex Tomas 	void *bitmap = EXT4_MB_BITMAP(e4b);
1631c9de560dSAlex Tomas 	struct ext4_free_extent ex;
1632c9de560dSAlex Tomas 	int i;
1633c9de560dSAlex Tomas 	int free;
1634c9de560dSAlex Tomas 
1635c9de560dSAlex Tomas 	free = e4b->bd_info->bb_free;
1636c9de560dSAlex Tomas 	BUG_ON(free <= 0);
1637c9de560dSAlex Tomas 
1638c9de560dSAlex Tomas 	i = e4b->bd_info->bb_first_free;
1639c9de560dSAlex Tomas 
1640c9de560dSAlex Tomas 	while (free && ac->ac_status == AC_STATUS_CONTINUE) {
1641ffad0a44SAneesh Kumar K.V 		i = mb_find_next_zero_bit(bitmap,
1642c9de560dSAlex Tomas 						EXT4_BLOCKS_PER_GROUP(sb), i);
1643c9de560dSAlex Tomas 		if (i >= EXT4_BLOCKS_PER_GROUP(sb)) {
164426346ff6SAneesh Kumar K.V 			/*
1645e56eb659SAneesh Kumar K.V 			 * IF we have corrupt bitmap, we won't find any
164626346ff6SAneesh Kumar K.V 			 * free blocks even though group info says we
164726346ff6SAneesh Kumar K.V 			 * we have free blocks
164826346ff6SAneesh Kumar K.V 			 */
16495d1b1b3fSAneesh Kumar K.V 			ext4_grp_locked_error(sb, e4b->bd_group,
16505d1b1b3fSAneesh Kumar K.V 					__func__, "%d free blocks as per "
1651fde4d95aSTheodore Ts'o 					"group info. But bitmap says 0",
165226346ff6SAneesh Kumar K.V 					free);
1653c9de560dSAlex Tomas 			break;
1654c9de560dSAlex Tomas 		}
1655c9de560dSAlex Tomas 
1656c9de560dSAlex Tomas 		mb_find_extent(e4b, 0, i, ac->ac_g_ex.fe_len, &ex);
1657c9de560dSAlex Tomas 		BUG_ON(ex.fe_len <= 0);
165826346ff6SAneesh Kumar K.V 		if (free < ex.fe_len) {
16595d1b1b3fSAneesh Kumar K.V 			ext4_grp_locked_error(sb, e4b->bd_group,
16605d1b1b3fSAneesh Kumar K.V 					__func__, "%d free blocks as per "
1661fde4d95aSTheodore Ts'o 					"group info. But got %d blocks",
166226346ff6SAneesh Kumar K.V 					free, ex.fe_len);
1663e56eb659SAneesh Kumar K.V 			/*
1664e56eb659SAneesh Kumar K.V 			 * The number of free blocks differs. This mostly
1665e56eb659SAneesh Kumar K.V 			 * indicate that the bitmap is corrupt. So exit
1666e56eb659SAneesh Kumar K.V 			 * without claiming the space.
1667e56eb659SAneesh Kumar K.V 			 */
1668e56eb659SAneesh Kumar K.V 			break;
166926346ff6SAneesh Kumar K.V 		}
1670c9de560dSAlex Tomas 
1671c9de560dSAlex Tomas 		ext4_mb_measure_extent(ac, &ex, e4b);
1672c9de560dSAlex Tomas 
1673c9de560dSAlex Tomas 		i += ex.fe_len;
1674c9de560dSAlex Tomas 		free -= ex.fe_len;
1675c9de560dSAlex Tomas 	}
1676c9de560dSAlex Tomas 
1677c9de560dSAlex Tomas 	ext4_mb_check_limits(ac, e4b, 1);
1678c9de560dSAlex Tomas }
1679c9de560dSAlex Tomas 
1680c9de560dSAlex Tomas /*
1681c9de560dSAlex Tomas  * This is a special case for storages like raid5
1682c9de560dSAlex Tomas  * we try to find stripe-aligned chunks for stripe-size requests
1683c9de560dSAlex Tomas  * XXX should do so at least for multiples of stripe size as well
1684c9de560dSAlex Tomas  */
1685c9de560dSAlex Tomas static void ext4_mb_scan_aligned(struct ext4_allocation_context *ac,
1686c9de560dSAlex Tomas 				 struct ext4_buddy *e4b)
1687c9de560dSAlex Tomas {
1688c9de560dSAlex Tomas 	struct super_block *sb = ac->ac_sb;
1689c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
1690c9de560dSAlex Tomas 	void *bitmap = EXT4_MB_BITMAP(e4b);
1691c9de560dSAlex Tomas 	struct ext4_free_extent ex;
1692c9de560dSAlex Tomas 	ext4_fsblk_t first_group_block;
1693c9de560dSAlex Tomas 	ext4_fsblk_t a;
1694c9de560dSAlex Tomas 	ext4_grpblk_t i;
1695c9de560dSAlex Tomas 	int max;
1696c9de560dSAlex Tomas 
1697c9de560dSAlex Tomas 	BUG_ON(sbi->s_stripe == 0);
1698c9de560dSAlex Tomas 
1699c9de560dSAlex Tomas 	/* find first stripe-aligned block in group */
1700c9de560dSAlex Tomas 	first_group_block = e4b->bd_group * EXT4_BLOCKS_PER_GROUP(sb)
1701c9de560dSAlex Tomas 		+ le32_to_cpu(sbi->s_es->s_first_data_block);
1702c9de560dSAlex Tomas 	a = first_group_block + sbi->s_stripe - 1;
1703c9de560dSAlex Tomas 	do_div(a, sbi->s_stripe);
1704c9de560dSAlex Tomas 	i = (a * sbi->s_stripe) - first_group_block;
1705c9de560dSAlex Tomas 
1706c9de560dSAlex Tomas 	while (i < EXT4_BLOCKS_PER_GROUP(sb)) {
1707c9de560dSAlex Tomas 		if (!mb_test_bit(i, bitmap)) {
1708c9de560dSAlex Tomas 			max = mb_find_extent(e4b, 0, i, sbi->s_stripe, &ex);
1709c9de560dSAlex Tomas 			if (max >= sbi->s_stripe) {
1710c9de560dSAlex Tomas 				ac->ac_found++;
1711c9de560dSAlex Tomas 				ac->ac_b_ex = ex;
1712c9de560dSAlex Tomas 				ext4_mb_use_best_found(ac, e4b);
1713c9de560dSAlex Tomas 				break;
1714c9de560dSAlex Tomas 			}
1715c9de560dSAlex Tomas 		}
1716c9de560dSAlex Tomas 		i += sbi->s_stripe;
1717c9de560dSAlex Tomas 	}
1718c9de560dSAlex Tomas }
1719c9de560dSAlex Tomas 
1720c9de560dSAlex Tomas static int ext4_mb_good_group(struct ext4_allocation_context *ac,
1721c9de560dSAlex Tomas 				ext4_group_t group, int cr)
1722c9de560dSAlex Tomas {
1723c9de560dSAlex Tomas 	unsigned free, fragments;
1724c9de560dSAlex Tomas 	unsigned i, bits;
1725c9de560dSAlex Tomas 	struct ext4_group_desc *desc;
1726c9de560dSAlex Tomas 	struct ext4_group_info *grp = ext4_get_group_info(ac->ac_sb, group);
1727c9de560dSAlex Tomas 
1728c9de560dSAlex Tomas 	BUG_ON(cr < 0 || cr >= 4);
1729c9de560dSAlex Tomas 	BUG_ON(EXT4_MB_GRP_NEED_INIT(grp));
1730c9de560dSAlex Tomas 
1731c9de560dSAlex Tomas 	free = grp->bb_free;
1732c9de560dSAlex Tomas 	fragments = grp->bb_fragments;
1733c9de560dSAlex Tomas 	if (free == 0)
1734c9de560dSAlex Tomas 		return 0;
1735c9de560dSAlex Tomas 	if (fragments == 0)
1736c9de560dSAlex Tomas 		return 0;
1737c9de560dSAlex Tomas 
1738c9de560dSAlex Tomas 	switch (cr) {
1739c9de560dSAlex Tomas 	case 0:
1740c9de560dSAlex Tomas 		BUG_ON(ac->ac_2order == 0);
1741c9de560dSAlex Tomas 		/* If this group is uninitialized, skip it initially */
1742c9de560dSAlex Tomas 		desc = ext4_get_group_desc(ac->ac_sb, group, NULL);
1743c9de560dSAlex Tomas 		if (desc->bg_flags & cpu_to_le16(EXT4_BG_BLOCK_UNINIT))
1744c9de560dSAlex Tomas 			return 0;
1745c9de560dSAlex Tomas 
1746c9de560dSAlex Tomas 		bits = ac->ac_sb->s_blocksize_bits + 1;
1747c9de560dSAlex Tomas 		for (i = ac->ac_2order; i <= bits; i++)
1748c9de560dSAlex Tomas 			if (grp->bb_counters[i] > 0)
1749c9de560dSAlex Tomas 				return 1;
1750c9de560dSAlex Tomas 		break;
1751c9de560dSAlex Tomas 	case 1:
1752c9de560dSAlex Tomas 		if ((free / fragments) >= ac->ac_g_ex.fe_len)
1753c9de560dSAlex Tomas 			return 1;
1754c9de560dSAlex Tomas 		break;
1755c9de560dSAlex Tomas 	case 2:
1756c9de560dSAlex Tomas 		if (free >= ac->ac_g_ex.fe_len)
1757c9de560dSAlex Tomas 			return 1;
1758c9de560dSAlex Tomas 		break;
1759c9de560dSAlex Tomas 	case 3:
1760c9de560dSAlex Tomas 		return 1;
1761c9de560dSAlex Tomas 	default:
1762c9de560dSAlex Tomas 		BUG();
1763c9de560dSAlex Tomas 	}
1764c9de560dSAlex Tomas 
1765c9de560dSAlex Tomas 	return 0;
1766c9de560dSAlex Tomas }
1767c9de560dSAlex Tomas 
1768920313a7SAneesh Kumar K.V /*
1769920313a7SAneesh Kumar K.V  * lock the group_info alloc_sem of all the groups
1770920313a7SAneesh Kumar K.V  * belonging to the same buddy cache page. This
1771920313a7SAneesh Kumar K.V  * make sure other parallel operation on the buddy
1772920313a7SAneesh Kumar K.V  * cache doesn't happen  whild holding the buddy cache
1773920313a7SAneesh Kumar K.V  * lock
1774920313a7SAneesh Kumar K.V  */
1775920313a7SAneesh Kumar K.V int ext4_mb_get_buddy_cache_lock(struct super_block *sb, ext4_group_t group)
1776920313a7SAneesh Kumar K.V {
1777920313a7SAneesh Kumar K.V 	int i;
1778920313a7SAneesh Kumar K.V 	int block, pnum;
1779920313a7SAneesh Kumar K.V 	int blocks_per_page;
1780920313a7SAneesh Kumar K.V 	int groups_per_page;
1781920313a7SAneesh Kumar K.V 	ext4_group_t first_group;
1782920313a7SAneesh Kumar K.V 	struct ext4_group_info *grp;
1783920313a7SAneesh Kumar K.V 
1784920313a7SAneesh Kumar K.V 	blocks_per_page = PAGE_CACHE_SIZE / sb->s_blocksize;
1785920313a7SAneesh Kumar K.V 	/*
1786920313a7SAneesh Kumar K.V 	 * the buddy cache inode stores the block bitmap
1787920313a7SAneesh Kumar K.V 	 * and buddy information in consecutive blocks.
1788920313a7SAneesh Kumar K.V 	 * So for each group we need two blocks.
1789920313a7SAneesh Kumar K.V 	 */
1790920313a7SAneesh Kumar K.V 	block = group * 2;
1791920313a7SAneesh Kumar K.V 	pnum = block / blocks_per_page;
1792920313a7SAneesh Kumar K.V 	first_group = pnum * blocks_per_page / 2;
1793920313a7SAneesh Kumar K.V 
1794920313a7SAneesh Kumar K.V 	groups_per_page = blocks_per_page >> 1;
1795920313a7SAneesh Kumar K.V 	if (groups_per_page == 0)
1796920313a7SAneesh Kumar K.V 		groups_per_page = 1;
1797920313a7SAneesh Kumar K.V 	/* read all groups the page covers into the cache */
1798920313a7SAneesh Kumar K.V 	for (i = 0; i < groups_per_page; i++) {
1799920313a7SAneesh Kumar K.V 
1800920313a7SAneesh Kumar K.V 		if ((first_group + i) >= EXT4_SB(sb)->s_groups_count)
1801920313a7SAneesh Kumar K.V 			break;
1802920313a7SAneesh Kumar K.V 		grp = ext4_get_group_info(sb, first_group + i);
1803920313a7SAneesh Kumar K.V 		/* take all groups write allocation
1804920313a7SAneesh Kumar K.V 		 * semaphore. This make sure there is
1805920313a7SAneesh Kumar K.V 		 * no block allocation going on in any
1806920313a7SAneesh Kumar K.V 		 * of that groups
1807920313a7SAneesh Kumar K.V 		 */
1808b7be019eSAneesh Kumar K.V 		down_write_nested(&grp->alloc_sem, i);
1809920313a7SAneesh Kumar K.V 	}
1810920313a7SAneesh Kumar K.V 	return i;
1811920313a7SAneesh Kumar K.V }
1812920313a7SAneesh Kumar K.V 
1813920313a7SAneesh Kumar K.V void ext4_mb_put_buddy_cache_lock(struct super_block *sb,
1814920313a7SAneesh Kumar K.V 					ext4_group_t group, int locked_group)
1815920313a7SAneesh Kumar K.V {
1816920313a7SAneesh Kumar K.V 	int i;
1817920313a7SAneesh Kumar K.V 	int block, pnum;
1818920313a7SAneesh Kumar K.V 	int blocks_per_page;
1819920313a7SAneesh Kumar K.V 	ext4_group_t first_group;
1820920313a7SAneesh Kumar K.V 	struct ext4_group_info *grp;
1821920313a7SAneesh Kumar K.V 
1822920313a7SAneesh Kumar K.V 	blocks_per_page = PAGE_CACHE_SIZE / sb->s_blocksize;
1823920313a7SAneesh Kumar K.V 	/*
1824920313a7SAneesh Kumar K.V 	 * the buddy cache inode stores the block bitmap
1825920313a7SAneesh Kumar K.V 	 * and buddy information in consecutive blocks.
1826920313a7SAneesh Kumar K.V 	 * So for each group we need two blocks.
1827920313a7SAneesh Kumar K.V 	 */
1828920313a7SAneesh Kumar K.V 	block = group * 2;
1829920313a7SAneesh Kumar K.V 	pnum = block / blocks_per_page;
1830920313a7SAneesh Kumar K.V 	first_group = pnum * blocks_per_page / 2;
1831920313a7SAneesh Kumar K.V 	/* release locks on all the groups */
1832920313a7SAneesh Kumar K.V 	for (i = 0; i < locked_group; i++) {
1833920313a7SAneesh Kumar K.V 
1834920313a7SAneesh Kumar K.V 		grp = ext4_get_group_info(sb, first_group + i);
1835920313a7SAneesh Kumar K.V 		/* take all groups write allocation
1836920313a7SAneesh Kumar K.V 		 * semaphore. This make sure there is
1837920313a7SAneesh Kumar K.V 		 * no block allocation going on in any
1838920313a7SAneesh Kumar K.V 		 * of that groups
1839920313a7SAneesh Kumar K.V 		 */
1840920313a7SAneesh Kumar K.V 		up_write(&grp->alloc_sem);
1841920313a7SAneesh Kumar K.V 	}
1842920313a7SAneesh Kumar K.V 
1843920313a7SAneesh Kumar K.V }
1844920313a7SAneesh Kumar K.V 
1845920313a7SAneesh Kumar K.V static int ext4_mb_init_group(struct super_block *sb, ext4_group_t group)
1846920313a7SAneesh Kumar K.V {
1847920313a7SAneesh Kumar K.V 
1848920313a7SAneesh Kumar K.V 	int ret;
1849920313a7SAneesh Kumar K.V 	void *bitmap;
1850920313a7SAneesh Kumar K.V 	int blocks_per_page;
1851920313a7SAneesh Kumar K.V 	int block, pnum, poff;
1852920313a7SAneesh Kumar K.V 	int num_grp_locked = 0;
1853920313a7SAneesh Kumar K.V 	struct ext4_group_info *this_grp;
1854920313a7SAneesh Kumar K.V 	struct ext4_sb_info *sbi = EXT4_SB(sb);
1855920313a7SAneesh Kumar K.V 	struct inode *inode = sbi->s_buddy_cache;
1856920313a7SAneesh Kumar K.V 	struct page *page = NULL, *bitmap_page = NULL;
1857920313a7SAneesh Kumar K.V 
1858920313a7SAneesh Kumar K.V 	mb_debug("init group %lu\n", group);
1859920313a7SAneesh Kumar K.V 	blocks_per_page = PAGE_CACHE_SIZE / sb->s_blocksize;
1860920313a7SAneesh Kumar K.V 	this_grp = ext4_get_group_info(sb, group);
1861920313a7SAneesh Kumar K.V 	/*
1862920313a7SAneesh Kumar K.V 	 * This ensures we don't add group
1863920313a7SAneesh Kumar K.V 	 * to this buddy cache via resize
1864920313a7SAneesh Kumar K.V 	 */
1865920313a7SAneesh Kumar K.V 	num_grp_locked =  ext4_mb_get_buddy_cache_lock(sb, group);
1866920313a7SAneesh Kumar K.V 	if (!EXT4_MB_GRP_NEED_INIT(this_grp)) {
1867920313a7SAneesh Kumar K.V 		/*
1868920313a7SAneesh Kumar K.V 		 * somebody initialized the group
1869920313a7SAneesh Kumar K.V 		 * return without doing anything
1870920313a7SAneesh Kumar K.V 		 */
1871920313a7SAneesh Kumar K.V 		ret = 0;
1872920313a7SAneesh Kumar K.V 		goto err;
1873920313a7SAneesh Kumar K.V 	}
1874920313a7SAneesh Kumar K.V 	/*
1875920313a7SAneesh Kumar K.V 	 * the buddy cache inode stores the block bitmap
1876920313a7SAneesh Kumar K.V 	 * and buddy information in consecutive blocks.
1877920313a7SAneesh Kumar K.V 	 * So for each group we need two blocks.
1878920313a7SAneesh Kumar K.V 	 */
1879920313a7SAneesh Kumar K.V 	block = group * 2;
1880920313a7SAneesh Kumar K.V 	pnum = block / blocks_per_page;
1881920313a7SAneesh Kumar K.V 	poff = block % blocks_per_page;
1882920313a7SAneesh Kumar K.V 	page = find_or_create_page(inode->i_mapping, pnum, GFP_NOFS);
1883920313a7SAneesh Kumar K.V 	if (page) {
1884920313a7SAneesh Kumar K.V 		BUG_ON(page->mapping != inode->i_mapping);
1885920313a7SAneesh Kumar K.V 		ret = ext4_mb_init_cache(page, NULL);
1886920313a7SAneesh Kumar K.V 		if (ret) {
1887920313a7SAneesh Kumar K.V 			unlock_page(page);
1888920313a7SAneesh Kumar K.V 			goto err;
1889920313a7SAneesh Kumar K.V 		}
1890920313a7SAneesh Kumar K.V 		unlock_page(page);
1891920313a7SAneesh Kumar K.V 	}
1892920313a7SAneesh Kumar K.V 	if (page == NULL || !PageUptodate(page)) {
1893920313a7SAneesh Kumar K.V 		ret = -EIO;
1894920313a7SAneesh Kumar K.V 		goto err;
1895920313a7SAneesh Kumar K.V 	}
1896920313a7SAneesh Kumar K.V 	mark_page_accessed(page);
1897920313a7SAneesh Kumar K.V 	bitmap_page = page;
1898920313a7SAneesh Kumar K.V 	bitmap = page_address(page) + (poff * sb->s_blocksize);
1899920313a7SAneesh Kumar K.V 
1900920313a7SAneesh Kumar K.V 	/* init buddy cache */
1901920313a7SAneesh Kumar K.V 	block++;
1902920313a7SAneesh Kumar K.V 	pnum = block / blocks_per_page;
1903920313a7SAneesh Kumar K.V 	poff = block % blocks_per_page;
1904920313a7SAneesh Kumar K.V 	page = find_or_create_page(inode->i_mapping, pnum, GFP_NOFS);
1905920313a7SAneesh Kumar K.V 	if (page == bitmap_page) {
1906920313a7SAneesh Kumar K.V 		/*
1907920313a7SAneesh Kumar K.V 		 * If both the bitmap and buddy are in
1908920313a7SAneesh Kumar K.V 		 * the same page we don't need to force
1909920313a7SAneesh Kumar K.V 		 * init the buddy
1910920313a7SAneesh Kumar K.V 		 */
1911920313a7SAneesh Kumar K.V 		unlock_page(page);
1912920313a7SAneesh Kumar K.V 	} else if (page) {
1913920313a7SAneesh Kumar K.V 		BUG_ON(page->mapping != inode->i_mapping);
1914920313a7SAneesh Kumar K.V 		ret = ext4_mb_init_cache(page, bitmap);
1915920313a7SAneesh Kumar K.V 		if (ret) {
1916920313a7SAneesh Kumar K.V 			unlock_page(page);
1917920313a7SAneesh Kumar K.V 			goto err;
1918920313a7SAneesh Kumar K.V 		}
1919920313a7SAneesh Kumar K.V 		unlock_page(page);
1920920313a7SAneesh Kumar K.V 	}
1921920313a7SAneesh Kumar K.V 	if (page == NULL || !PageUptodate(page)) {
1922920313a7SAneesh Kumar K.V 		ret = -EIO;
1923920313a7SAneesh Kumar K.V 		goto err;
1924920313a7SAneesh Kumar K.V 	}
1925920313a7SAneesh Kumar K.V 	mark_page_accessed(page);
1926920313a7SAneesh Kumar K.V err:
1927920313a7SAneesh Kumar K.V 	ext4_mb_put_buddy_cache_lock(sb, group, num_grp_locked);
1928920313a7SAneesh Kumar K.V 	if (bitmap_page)
1929920313a7SAneesh Kumar K.V 		page_cache_release(bitmap_page);
1930920313a7SAneesh Kumar K.V 	if (page)
1931920313a7SAneesh Kumar K.V 		page_cache_release(page);
1932920313a7SAneesh Kumar K.V 	return ret;
1933920313a7SAneesh Kumar K.V }
1934920313a7SAneesh Kumar K.V 
19354ddfef7bSEric Sandeen static noinline_for_stack int
19364ddfef7bSEric Sandeen ext4_mb_regular_allocator(struct ext4_allocation_context *ac)
1937c9de560dSAlex Tomas {
1938c9de560dSAlex Tomas 	ext4_group_t group;
1939c9de560dSAlex Tomas 	ext4_group_t i;
1940c9de560dSAlex Tomas 	int cr;
1941c9de560dSAlex Tomas 	int err = 0;
1942c9de560dSAlex Tomas 	int bsbits;
1943c9de560dSAlex Tomas 	struct ext4_sb_info *sbi;
1944c9de560dSAlex Tomas 	struct super_block *sb;
1945c9de560dSAlex Tomas 	struct ext4_buddy e4b;
1946c9de560dSAlex Tomas 	loff_t size, isize;
1947c9de560dSAlex Tomas 
1948c9de560dSAlex Tomas 	sb = ac->ac_sb;
1949c9de560dSAlex Tomas 	sbi = EXT4_SB(sb);
1950c9de560dSAlex Tomas 	BUG_ON(ac->ac_status == AC_STATUS_FOUND);
1951c9de560dSAlex Tomas 
1952c9de560dSAlex Tomas 	/* first, try the goal */
1953c9de560dSAlex Tomas 	err = ext4_mb_find_by_goal(ac, &e4b);
1954c9de560dSAlex Tomas 	if (err || ac->ac_status == AC_STATUS_FOUND)
1955c9de560dSAlex Tomas 		goto out;
1956c9de560dSAlex Tomas 
1957c9de560dSAlex Tomas 	if (unlikely(ac->ac_flags & EXT4_MB_HINT_GOAL_ONLY))
1958c9de560dSAlex Tomas 		goto out;
1959c9de560dSAlex Tomas 
1960c9de560dSAlex Tomas 	/*
1961c9de560dSAlex Tomas 	 * ac->ac2_order is set only if the fe_len is a power of 2
1962c9de560dSAlex Tomas 	 * if ac2_order is set we also set criteria to 0 so that we
1963c9de560dSAlex Tomas 	 * try exact allocation using buddy.
1964c9de560dSAlex Tomas 	 */
1965c9de560dSAlex Tomas 	i = fls(ac->ac_g_ex.fe_len);
1966c9de560dSAlex Tomas 	ac->ac_2order = 0;
1967c9de560dSAlex Tomas 	/*
1968c9de560dSAlex Tomas 	 * We search using buddy data only if the order of the request
1969c9de560dSAlex Tomas 	 * is greater than equal to the sbi_s_mb_order2_reqs
1970c9de560dSAlex Tomas 	 * You can tune it via /proc/fs/ext4/<partition>/order2_req
1971c9de560dSAlex Tomas 	 */
1972c9de560dSAlex Tomas 	if (i >= sbi->s_mb_order2_reqs) {
1973c9de560dSAlex Tomas 		/*
1974c9de560dSAlex Tomas 		 * This should tell if fe_len is exactly power of 2
1975c9de560dSAlex Tomas 		 */
1976c9de560dSAlex Tomas 		if ((ac->ac_g_ex.fe_len & (~(1 << (i - 1)))) == 0)
1977c9de560dSAlex Tomas 			ac->ac_2order = i - 1;
1978c9de560dSAlex Tomas 	}
1979c9de560dSAlex Tomas 
1980c9de560dSAlex Tomas 	bsbits = ac->ac_sb->s_blocksize_bits;
1981c9de560dSAlex Tomas 	/* if stream allocation is enabled, use global goal */
1982c9de560dSAlex Tomas 	size = ac->ac_o_ex.fe_logical + ac->ac_o_ex.fe_len;
1983c9de560dSAlex Tomas 	isize = i_size_read(ac->ac_inode) >> bsbits;
1984c9de560dSAlex Tomas 	if (size < isize)
1985c9de560dSAlex Tomas 		size = isize;
1986c9de560dSAlex Tomas 
1987c9de560dSAlex Tomas 	if (size < sbi->s_mb_stream_request &&
1988c9de560dSAlex Tomas 			(ac->ac_flags & EXT4_MB_HINT_DATA)) {
1989c9de560dSAlex Tomas 		/* TBD: may be hot point */
1990c9de560dSAlex Tomas 		spin_lock(&sbi->s_md_lock);
1991c9de560dSAlex Tomas 		ac->ac_g_ex.fe_group = sbi->s_mb_last_group;
1992c9de560dSAlex Tomas 		ac->ac_g_ex.fe_start = sbi->s_mb_last_start;
1993c9de560dSAlex Tomas 		spin_unlock(&sbi->s_md_lock);
1994c9de560dSAlex Tomas 	}
1995c9de560dSAlex Tomas 	/* Let's just scan groups to find more-less suitable blocks */
1996c9de560dSAlex Tomas 	cr = ac->ac_2order ? 0 : 1;
1997c9de560dSAlex Tomas 	/*
1998c9de560dSAlex Tomas 	 * cr == 0 try to get exact allocation,
1999c9de560dSAlex Tomas 	 * cr == 3  try to get anything
2000c9de560dSAlex Tomas 	 */
2001c9de560dSAlex Tomas repeat:
2002c9de560dSAlex Tomas 	for (; cr < 4 && ac->ac_status == AC_STATUS_CONTINUE; cr++) {
2003c9de560dSAlex Tomas 		ac->ac_criteria = cr;
2004ed8f9c75SAneesh Kumar K.V 		/*
2005ed8f9c75SAneesh Kumar K.V 		 * searching for the right group start
2006ed8f9c75SAneesh Kumar K.V 		 * from the goal value specified
2007ed8f9c75SAneesh Kumar K.V 		 */
2008ed8f9c75SAneesh Kumar K.V 		group = ac->ac_g_ex.fe_group;
2009ed8f9c75SAneesh Kumar K.V 
2010c9de560dSAlex Tomas 		for (i = 0; i < EXT4_SB(sb)->s_groups_count; group++, i++) {
2011c9de560dSAlex Tomas 			struct ext4_group_info *grp;
2012c9de560dSAlex Tomas 			struct ext4_group_desc *desc;
2013c9de560dSAlex Tomas 
2014c9de560dSAlex Tomas 			if (group == EXT4_SB(sb)->s_groups_count)
2015c9de560dSAlex Tomas 				group = 0;
2016c9de560dSAlex Tomas 
2017c9de560dSAlex Tomas 			/* quick check to skip empty groups */
2018920313a7SAneesh Kumar K.V 			grp = ext4_get_group_info(sb, group);
2019c9de560dSAlex Tomas 			if (grp->bb_free == 0)
2020c9de560dSAlex Tomas 				continue;
2021c9de560dSAlex Tomas 
2022c9de560dSAlex Tomas 			/*
2023c9de560dSAlex Tomas 			 * if the group is already init we check whether it is
2024c9de560dSAlex Tomas 			 * a good group and if not we don't load the buddy
2025c9de560dSAlex Tomas 			 */
2026c9de560dSAlex Tomas 			if (EXT4_MB_GRP_NEED_INIT(grp)) {
2027c9de560dSAlex Tomas 				/*
2028c9de560dSAlex Tomas 				 * we need full data about the group
2029c9de560dSAlex Tomas 				 * to make a good selection
2030c9de560dSAlex Tomas 				 */
2031920313a7SAneesh Kumar K.V 				err = ext4_mb_init_group(sb, group);
2032c9de560dSAlex Tomas 				if (err)
2033c9de560dSAlex Tomas 					goto out;
2034c9de560dSAlex Tomas 			}
2035c9de560dSAlex Tomas 
2036c9de560dSAlex Tomas 			/*
2037c9de560dSAlex Tomas 			 * If the particular group doesn't satisfy our
2038c9de560dSAlex Tomas 			 * criteria we continue with the next group
2039c9de560dSAlex Tomas 			 */
2040c9de560dSAlex Tomas 			if (!ext4_mb_good_group(ac, group, cr))
2041c9de560dSAlex Tomas 				continue;
2042c9de560dSAlex Tomas 
2043c9de560dSAlex Tomas 			err = ext4_mb_load_buddy(sb, group, &e4b);
2044c9de560dSAlex Tomas 			if (err)
2045c9de560dSAlex Tomas 				goto out;
2046c9de560dSAlex Tomas 
2047c9de560dSAlex Tomas 			ext4_lock_group(sb, group);
2048c9de560dSAlex Tomas 			if (!ext4_mb_good_group(ac, group, cr)) {
2049c9de560dSAlex Tomas 				/* someone did allocation from this group */
2050c9de560dSAlex Tomas 				ext4_unlock_group(sb, group);
2051c9de560dSAlex Tomas 				ext4_mb_release_desc(&e4b);
2052c9de560dSAlex Tomas 				continue;
2053c9de560dSAlex Tomas 			}
2054c9de560dSAlex Tomas 
2055c9de560dSAlex Tomas 			ac->ac_groups_scanned++;
2056c9de560dSAlex Tomas 			desc = ext4_get_group_desc(sb, group, NULL);
2057c9de560dSAlex Tomas 			if (cr == 0 || (desc->bg_flags &
2058c9de560dSAlex Tomas 					cpu_to_le16(EXT4_BG_BLOCK_UNINIT) &&
2059c9de560dSAlex Tomas 					ac->ac_2order != 0))
2060c9de560dSAlex Tomas 				ext4_mb_simple_scan_group(ac, &e4b);
2061c9de560dSAlex Tomas 			else if (cr == 1 &&
2062c9de560dSAlex Tomas 					ac->ac_g_ex.fe_len == sbi->s_stripe)
2063c9de560dSAlex Tomas 				ext4_mb_scan_aligned(ac, &e4b);
2064c9de560dSAlex Tomas 			else
2065c9de560dSAlex Tomas 				ext4_mb_complex_scan_group(ac, &e4b);
2066c9de560dSAlex Tomas 
2067c9de560dSAlex Tomas 			ext4_unlock_group(sb, group);
2068c9de560dSAlex Tomas 			ext4_mb_release_desc(&e4b);
2069c9de560dSAlex Tomas 
2070c9de560dSAlex Tomas 			if (ac->ac_status != AC_STATUS_CONTINUE)
2071c9de560dSAlex Tomas 				break;
2072c9de560dSAlex Tomas 		}
2073c9de560dSAlex Tomas 	}
2074c9de560dSAlex Tomas 
2075c9de560dSAlex Tomas 	if (ac->ac_b_ex.fe_len > 0 && ac->ac_status != AC_STATUS_FOUND &&
2076c9de560dSAlex Tomas 	    !(ac->ac_flags & EXT4_MB_HINT_FIRST)) {
2077c9de560dSAlex Tomas 		/*
2078c9de560dSAlex Tomas 		 * We've been searching too long. Let's try to allocate
2079c9de560dSAlex Tomas 		 * the best chunk we've found so far
2080c9de560dSAlex Tomas 		 */
2081c9de560dSAlex Tomas 
2082c9de560dSAlex Tomas 		ext4_mb_try_best_found(ac, &e4b);
2083c9de560dSAlex Tomas 		if (ac->ac_status != AC_STATUS_FOUND) {
2084c9de560dSAlex Tomas 			/*
2085c9de560dSAlex Tomas 			 * Someone more lucky has already allocated it.
2086c9de560dSAlex Tomas 			 * The only thing we can do is just take first
2087c9de560dSAlex Tomas 			 * found block(s)
2088c9de560dSAlex Tomas 			printk(KERN_DEBUG "EXT4-fs: someone won our chunk\n");
2089c9de560dSAlex Tomas 			 */
2090c9de560dSAlex Tomas 			ac->ac_b_ex.fe_group = 0;
2091c9de560dSAlex Tomas 			ac->ac_b_ex.fe_start = 0;
2092c9de560dSAlex Tomas 			ac->ac_b_ex.fe_len = 0;
2093c9de560dSAlex Tomas 			ac->ac_status = AC_STATUS_CONTINUE;
2094c9de560dSAlex Tomas 			ac->ac_flags |= EXT4_MB_HINT_FIRST;
2095c9de560dSAlex Tomas 			cr = 3;
2096c9de560dSAlex Tomas 			atomic_inc(&sbi->s_mb_lost_chunks);
2097c9de560dSAlex Tomas 			goto repeat;
2098c9de560dSAlex Tomas 		}
2099c9de560dSAlex Tomas 	}
2100c9de560dSAlex Tomas out:
2101c9de560dSAlex Tomas 	return err;
2102c9de560dSAlex Tomas }
2103c9de560dSAlex Tomas 
2104c9de560dSAlex Tomas #ifdef EXT4_MB_HISTORY
2105c9de560dSAlex Tomas struct ext4_mb_proc_session {
2106c9de560dSAlex Tomas 	struct ext4_mb_history *history;
2107c9de560dSAlex Tomas 	struct super_block *sb;
2108c9de560dSAlex Tomas 	int start;
2109c9de560dSAlex Tomas 	int max;
2110c9de560dSAlex Tomas };
2111c9de560dSAlex Tomas 
2112c9de560dSAlex Tomas static void *ext4_mb_history_skip_empty(struct ext4_mb_proc_session *s,
2113c9de560dSAlex Tomas 					struct ext4_mb_history *hs,
2114c9de560dSAlex Tomas 					int first)
2115c9de560dSAlex Tomas {
2116c9de560dSAlex Tomas 	if (hs == s->history + s->max)
2117c9de560dSAlex Tomas 		hs = s->history;
2118c9de560dSAlex Tomas 	if (!first && hs == s->history + s->start)
2119c9de560dSAlex Tomas 		return NULL;
2120c9de560dSAlex Tomas 	while (hs->orig.fe_len == 0) {
2121c9de560dSAlex Tomas 		hs++;
2122c9de560dSAlex Tomas 		if (hs == s->history + s->max)
2123c9de560dSAlex Tomas 			hs = s->history;
2124c9de560dSAlex Tomas 		if (hs == s->history + s->start)
2125c9de560dSAlex Tomas 			return NULL;
2126c9de560dSAlex Tomas 	}
2127c9de560dSAlex Tomas 	return hs;
2128c9de560dSAlex Tomas }
2129c9de560dSAlex Tomas 
2130c9de560dSAlex Tomas static void *ext4_mb_seq_history_start(struct seq_file *seq, loff_t *pos)
2131c9de560dSAlex Tomas {
2132c9de560dSAlex Tomas 	struct ext4_mb_proc_session *s = seq->private;
2133c9de560dSAlex Tomas 	struct ext4_mb_history *hs;
2134c9de560dSAlex Tomas 	int l = *pos;
2135c9de560dSAlex Tomas 
2136c9de560dSAlex Tomas 	if (l == 0)
2137c9de560dSAlex Tomas 		return SEQ_START_TOKEN;
2138c9de560dSAlex Tomas 	hs = ext4_mb_history_skip_empty(s, s->history + s->start, 1);
2139c9de560dSAlex Tomas 	if (!hs)
2140c9de560dSAlex Tomas 		return NULL;
2141c9de560dSAlex Tomas 	while (--l && (hs = ext4_mb_history_skip_empty(s, ++hs, 0)) != NULL);
2142c9de560dSAlex Tomas 	return hs;
2143c9de560dSAlex Tomas }
2144c9de560dSAlex Tomas 
2145c9de560dSAlex Tomas static void *ext4_mb_seq_history_next(struct seq_file *seq, void *v,
2146c9de560dSAlex Tomas 				      loff_t *pos)
2147c9de560dSAlex Tomas {
2148c9de560dSAlex Tomas 	struct ext4_mb_proc_session *s = seq->private;
2149c9de560dSAlex Tomas 	struct ext4_mb_history *hs = v;
2150c9de560dSAlex Tomas 
2151c9de560dSAlex Tomas 	++*pos;
2152c9de560dSAlex Tomas 	if (v == SEQ_START_TOKEN)
2153c9de560dSAlex Tomas 		return ext4_mb_history_skip_empty(s, s->history + s->start, 1);
2154c9de560dSAlex Tomas 	else
2155c9de560dSAlex Tomas 		return ext4_mb_history_skip_empty(s, ++hs, 0);
2156c9de560dSAlex Tomas }
2157c9de560dSAlex Tomas 
2158c9de560dSAlex Tomas static int ext4_mb_seq_history_show(struct seq_file *seq, void *v)
2159c9de560dSAlex Tomas {
2160c9de560dSAlex Tomas 	char buf[25], buf2[25], buf3[25], *fmt;
2161c9de560dSAlex Tomas 	struct ext4_mb_history *hs = v;
2162c9de560dSAlex Tomas 
2163c9de560dSAlex Tomas 	if (v == SEQ_START_TOKEN) {
2164c9de560dSAlex Tomas 		seq_printf(seq, "%-5s %-8s %-23s %-23s %-23s %-5s "
2165c9de560dSAlex Tomas 				"%-5s %-2s %-5s %-5s %-5s %-6s\n",
2166c9de560dSAlex Tomas 			  "pid", "inode", "original", "goal", "result", "found",
2167c9de560dSAlex Tomas 			   "grps", "cr", "flags", "merge", "tail", "broken");
2168c9de560dSAlex Tomas 		return 0;
2169c9de560dSAlex Tomas 	}
2170c9de560dSAlex Tomas 
2171c9de560dSAlex Tomas 	if (hs->op == EXT4_MB_HISTORY_ALLOC) {
2172c9de560dSAlex Tomas 		fmt = "%-5u %-8u %-23s %-23s %-23s %-5u %-5u %-2u "
2173c9de560dSAlex Tomas 			"%-5u %-5s %-5u %-6u\n";
2174a9df9a49STheodore Ts'o 		sprintf(buf2, "%u/%d/%u@%u", hs->result.fe_group,
2175c9de560dSAlex Tomas 			hs->result.fe_start, hs->result.fe_len,
2176c9de560dSAlex Tomas 			hs->result.fe_logical);
2177a9df9a49STheodore Ts'o 		sprintf(buf, "%u/%d/%u@%u", hs->orig.fe_group,
2178c9de560dSAlex Tomas 			hs->orig.fe_start, hs->orig.fe_len,
2179c9de560dSAlex Tomas 			hs->orig.fe_logical);
2180a9df9a49STheodore Ts'o 		sprintf(buf3, "%u/%d/%u@%u", hs->goal.fe_group,
2181c9de560dSAlex Tomas 			hs->goal.fe_start, hs->goal.fe_len,
2182c9de560dSAlex Tomas 			hs->goal.fe_logical);
2183c9de560dSAlex Tomas 		seq_printf(seq, fmt, hs->pid, hs->ino, buf, buf3, buf2,
2184c9de560dSAlex Tomas 				hs->found, hs->groups, hs->cr, hs->flags,
2185c9de560dSAlex Tomas 				hs->merged ? "M" : "", hs->tail,
2186c9de560dSAlex Tomas 				hs->buddy ? 1 << hs->buddy : 0);
2187c9de560dSAlex Tomas 	} else if (hs->op == EXT4_MB_HISTORY_PREALLOC) {
2188c9de560dSAlex Tomas 		fmt = "%-5u %-8u %-23s %-23s %-23s\n";
2189a9df9a49STheodore Ts'o 		sprintf(buf2, "%u/%d/%u@%u", hs->result.fe_group,
2190c9de560dSAlex Tomas 			hs->result.fe_start, hs->result.fe_len,
2191c9de560dSAlex Tomas 			hs->result.fe_logical);
2192a9df9a49STheodore Ts'o 		sprintf(buf, "%u/%d/%u@%u", hs->orig.fe_group,
2193c9de560dSAlex Tomas 			hs->orig.fe_start, hs->orig.fe_len,
2194c9de560dSAlex Tomas 			hs->orig.fe_logical);
2195c9de560dSAlex Tomas 		seq_printf(seq, fmt, hs->pid, hs->ino, buf, "", buf2);
2196c9de560dSAlex Tomas 	} else if (hs->op == EXT4_MB_HISTORY_DISCARD) {
2197a9df9a49STheodore Ts'o 		sprintf(buf2, "%u/%d/%u", hs->result.fe_group,
2198c9de560dSAlex Tomas 			hs->result.fe_start, hs->result.fe_len);
2199c9de560dSAlex Tomas 		seq_printf(seq, "%-5u %-8u %-23s discard\n",
2200c9de560dSAlex Tomas 				hs->pid, hs->ino, buf2);
2201c9de560dSAlex Tomas 	} else if (hs->op == EXT4_MB_HISTORY_FREE) {
2202a9df9a49STheodore Ts'o 		sprintf(buf2, "%u/%d/%u", hs->result.fe_group,
2203c9de560dSAlex Tomas 			hs->result.fe_start, hs->result.fe_len);
2204c9de560dSAlex Tomas 		seq_printf(seq, "%-5u %-8u %-23s free\n",
2205c9de560dSAlex Tomas 				hs->pid, hs->ino, buf2);
2206c9de560dSAlex Tomas 	}
2207c9de560dSAlex Tomas 	return 0;
2208c9de560dSAlex Tomas }
2209c9de560dSAlex Tomas 
2210c9de560dSAlex Tomas static void ext4_mb_seq_history_stop(struct seq_file *seq, void *v)
2211c9de560dSAlex Tomas {
2212c9de560dSAlex Tomas }
2213c9de560dSAlex Tomas 
2214c9de560dSAlex Tomas static struct seq_operations ext4_mb_seq_history_ops = {
2215c9de560dSAlex Tomas 	.start  = ext4_mb_seq_history_start,
2216c9de560dSAlex Tomas 	.next   = ext4_mb_seq_history_next,
2217c9de560dSAlex Tomas 	.stop   = ext4_mb_seq_history_stop,
2218c9de560dSAlex Tomas 	.show   = ext4_mb_seq_history_show,
2219c9de560dSAlex Tomas };
2220c9de560dSAlex Tomas 
2221c9de560dSAlex Tomas static int ext4_mb_seq_history_open(struct inode *inode, struct file *file)
2222c9de560dSAlex Tomas {
2223c9de560dSAlex Tomas 	struct super_block *sb = PDE(inode)->data;
2224c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2225c9de560dSAlex Tomas 	struct ext4_mb_proc_session *s;
2226c9de560dSAlex Tomas 	int rc;
2227c9de560dSAlex Tomas 	int size;
2228c9de560dSAlex Tomas 
222974767c5aSShen Feng 	if (unlikely(sbi->s_mb_history == NULL))
223074767c5aSShen Feng 		return -ENOMEM;
2231c9de560dSAlex Tomas 	s = kmalloc(sizeof(*s), GFP_KERNEL);
2232c9de560dSAlex Tomas 	if (s == NULL)
2233c9de560dSAlex Tomas 		return -ENOMEM;
2234c9de560dSAlex Tomas 	s->sb = sb;
2235c9de560dSAlex Tomas 	size = sizeof(struct ext4_mb_history) * sbi->s_mb_history_max;
2236c9de560dSAlex Tomas 	s->history = kmalloc(size, GFP_KERNEL);
2237c9de560dSAlex Tomas 	if (s->history == NULL) {
2238c9de560dSAlex Tomas 		kfree(s);
2239c9de560dSAlex Tomas 		return -ENOMEM;
2240c9de560dSAlex Tomas 	}
2241c9de560dSAlex Tomas 
2242c9de560dSAlex Tomas 	spin_lock(&sbi->s_mb_history_lock);
2243c9de560dSAlex Tomas 	memcpy(s->history, sbi->s_mb_history, size);
2244c9de560dSAlex Tomas 	s->max = sbi->s_mb_history_max;
2245c9de560dSAlex Tomas 	s->start = sbi->s_mb_history_cur % s->max;
2246c9de560dSAlex Tomas 	spin_unlock(&sbi->s_mb_history_lock);
2247c9de560dSAlex Tomas 
2248c9de560dSAlex Tomas 	rc = seq_open(file, &ext4_mb_seq_history_ops);
2249c9de560dSAlex Tomas 	if (rc == 0) {
2250c9de560dSAlex Tomas 		struct seq_file *m = (struct seq_file *)file->private_data;
2251c9de560dSAlex Tomas 		m->private = s;
2252c9de560dSAlex Tomas 	} else {
2253c9de560dSAlex Tomas 		kfree(s->history);
2254c9de560dSAlex Tomas 		kfree(s);
2255c9de560dSAlex Tomas 	}
2256c9de560dSAlex Tomas 	return rc;
2257c9de560dSAlex Tomas 
2258c9de560dSAlex Tomas }
2259c9de560dSAlex Tomas 
2260c9de560dSAlex Tomas static int ext4_mb_seq_history_release(struct inode *inode, struct file *file)
2261c9de560dSAlex Tomas {
2262c9de560dSAlex Tomas 	struct seq_file *seq = (struct seq_file *)file->private_data;
2263c9de560dSAlex Tomas 	struct ext4_mb_proc_session *s = seq->private;
2264c9de560dSAlex Tomas 	kfree(s->history);
2265c9de560dSAlex Tomas 	kfree(s);
2266c9de560dSAlex Tomas 	return seq_release(inode, file);
2267c9de560dSAlex Tomas }
2268c9de560dSAlex Tomas 
2269c9de560dSAlex Tomas static ssize_t ext4_mb_seq_history_write(struct file *file,
2270c9de560dSAlex Tomas 				const char __user *buffer,
2271c9de560dSAlex Tomas 				size_t count, loff_t *ppos)
2272c9de560dSAlex Tomas {
2273c9de560dSAlex Tomas 	struct seq_file *seq = (struct seq_file *)file->private_data;
2274c9de560dSAlex Tomas 	struct ext4_mb_proc_session *s = seq->private;
2275c9de560dSAlex Tomas 	struct super_block *sb = s->sb;
2276c9de560dSAlex Tomas 	char str[32];
2277c9de560dSAlex Tomas 	int value;
2278c9de560dSAlex Tomas 
2279c9de560dSAlex Tomas 	if (count >= sizeof(str)) {
2280c9de560dSAlex Tomas 		printk(KERN_ERR "EXT4-fs: %s string too long, max %u bytes\n",
2281c9de560dSAlex Tomas 				"mb_history", (int)sizeof(str));
2282c9de560dSAlex Tomas 		return -EOVERFLOW;
2283c9de560dSAlex Tomas 	}
2284c9de560dSAlex Tomas 
2285c9de560dSAlex Tomas 	if (copy_from_user(str, buffer, count))
2286c9de560dSAlex Tomas 		return -EFAULT;
2287c9de560dSAlex Tomas 
2288c9de560dSAlex Tomas 	value = simple_strtol(str, NULL, 0);
2289c9de560dSAlex Tomas 	if (value < 0)
2290c9de560dSAlex Tomas 		return -ERANGE;
2291c9de560dSAlex Tomas 	EXT4_SB(sb)->s_mb_history_filter = value;
2292c9de560dSAlex Tomas 
2293c9de560dSAlex Tomas 	return count;
2294c9de560dSAlex Tomas }
2295c9de560dSAlex Tomas 
2296c9de560dSAlex Tomas static struct file_operations ext4_mb_seq_history_fops = {
2297c9de560dSAlex Tomas 	.owner		= THIS_MODULE,
2298c9de560dSAlex Tomas 	.open		= ext4_mb_seq_history_open,
2299c9de560dSAlex Tomas 	.read		= seq_read,
2300c9de560dSAlex Tomas 	.write		= ext4_mb_seq_history_write,
2301c9de560dSAlex Tomas 	.llseek		= seq_lseek,
2302c9de560dSAlex Tomas 	.release	= ext4_mb_seq_history_release,
2303c9de560dSAlex Tomas };
2304c9de560dSAlex Tomas 
2305c9de560dSAlex Tomas static void *ext4_mb_seq_groups_start(struct seq_file *seq, loff_t *pos)
2306c9de560dSAlex Tomas {
2307c9de560dSAlex Tomas 	struct super_block *sb = seq->private;
2308c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2309c9de560dSAlex Tomas 	ext4_group_t group;
2310c9de560dSAlex Tomas 
2311c9de560dSAlex Tomas 	if (*pos < 0 || *pos >= sbi->s_groups_count)
2312c9de560dSAlex Tomas 		return NULL;
2313c9de560dSAlex Tomas 
2314c9de560dSAlex Tomas 	group = *pos + 1;
2315a9df9a49STheodore Ts'o 	return (void *) ((unsigned long) group);
2316c9de560dSAlex Tomas }
2317c9de560dSAlex Tomas 
2318c9de560dSAlex Tomas static void *ext4_mb_seq_groups_next(struct seq_file *seq, void *v, loff_t *pos)
2319c9de560dSAlex Tomas {
2320c9de560dSAlex Tomas 	struct super_block *sb = seq->private;
2321c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2322c9de560dSAlex Tomas 	ext4_group_t group;
2323c9de560dSAlex Tomas 
2324c9de560dSAlex Tomas 	++*pos;
2325c9de560dSAlex Tomas 	if (*pos < 0 || *pos >= sbi->s_groups_count)
2326c9de560dSAlex Tomas 		return NULL;
2327c9de560dSAlex Tomas 	group = *pos + 1;
2328a9df9a49STheodore Ts'o 	return (void *) ((unsigned long) group);
2329c9de560dSAlex Tomas }
2330c9de560dSAlex Tomas 
2331c9de560dSAlex Tomas static int ext4_mb_seq_groups_show(struct seq_file *seq, void *v)
2332c9de560dSAlex Tomas {
2333c9de560dSAlex Tomas 	struct super_block *sb = seq->private;
2334a9df9a49STheodore Ts'o 	ext4_group_t group = (ext4_group_t) ((unsigned long) v);
2335c9de560dSAlex Tomas 	int i;
2336c9de560dSAlex Tomas 	int err;
2337c9de560dSAlex Tomas 	struct ext4_buddy e4b;
2338c9de560dSAlex Tomas 	struct sg {
2339c9de560dSAlex Tomas 		struct ext4_group_info info;
2340c9de560dSAlex Tomas 		unsigned short counters[16];
2341c9de560dSAlex Tomas 	} sg;
2342c9de560dSAlex Tomas 
2343c9de560dSAlex Tomas 	group--;
2344c9de560dSAlex Tomas 	if (group == 0)
2345c9de560dSAlex Tomas 		seq_printf(seq, "#%-5s: %-5s %-5s %-5s "
2346c9de560dSAlex Tomas 				"[ %-5s %-5s %-5s %-5s %-5s %-5s %-5s "
2347c9de560dSAlex Tomas 				  "%-5s %-5s %-5s %-5s %-5s %-5s %-5s ]\n",
2348c9de560dSAlex Tomas 			   "group", "free", "frags", "first",
2349c9de560dSAlex Tomas 			   "2^0", "2^1", "2^2", "2^3", "2^4", "2^5", "2^6",
2350c9de560dSAlex Tomas 			   "2^7", "2^8", "2^9", "2^10", "2^11", "2^12", "2^13");
2351c9de560dSAlex Tomas 
2352c9de560dSAlex Tomas 	i = (sb->s_blocksize_bits + 2) * sizeof(sg.info.bb_counters[0]) +
2353c9de560dSAlex Tomas 		sizeof(struct ext4_group_info);
2354c9de560dSAlex Tomas 	err = ext4_mb_load_buddy(sb, group, &e4b);
2355c9de560dSAlex Tomas 	if (err) {
2356a9df9a49STheodore Ts'o 		seq_printf(seq, "#%-5u: I/O error\n", group);
2357c9de560dSAlex Tomas 		return 0;
2358c9de560dSAlex Tomas 	}
2359c9de560dSAlex Tomas 	ext4_lock_group(sb, group);
2360c9de560dSAlex Tomas 	memcpy(&sg, ext4_get_group_info(sb, group), i);
2361c9de560dSAlex Tomas 	ext4_unlock_group(sb, group);
2362c9de560dSAlex Tomas 	ext4_mb_release_desc(&e4b);
2363c9de560dSAlex Tomas 
2364a9df9a49STheodore Ts'o 	seq_printf(seq, "#%-5u: %-5u %-5u %-5u [", group, sg.info.bb_free,
2365c9de560dSAlex Tomas 			sg.info.bb_fragments, sg.info.bb_first_free);
2366c9de560dSAlex Tomas 	for (i = 0; i <= 13; i++)
2367c9de560dSAlex Tomas 		seq_printf(seq, " %-5u", i <= sb->s_blocksize_bits + 1 ?
2368c9de560dSAlex Tomas 				sg.info.bb_counters[i] : 0);
2369c9de560dSAlex Tomas 	seq_printf(seq, " ]\n");
2370c9de560dSAlex Tomas 
2371c9de560dSAlex Tomas 	return 0;
2372c9de560dSAlex Tomas }
2373c9de560dSAlex Tomas 
2374c9de560dSAlex Tomas static void ext4_mb_seq_groups_stop(struct seq_file *seq, void *v)
2375c9de560dSAlex Tomas {
2376c9de560dSAlex Tomas }
2377c9de560dSAlex Tomas 
2378c9de560dSAlex Tomas static struct seq_operations ext4_mb_seq_groups_ops = {
2379c9de560dSAlex Tomas 	.start  = ext4_mb_seq_groups_start,
2380c9de560dSAlex Tomas 	.next   = ext4_mb_seq_groups_next,
2381c9de560dSAlex Tomas 	.stop   = ext4_mb_seq_groups_stop,
2382c9de560dSAlex Tomas 	.show   = ext4_mb_seq_groups_show,
2383c9de560dSAlex Tomas };
2384c9de560dSAlex Tomas 
2385c9de560dSAlex Tomas static int ext4_mb_seq_groups_open(struct inode *inode, struct file *file)
2386c9de560dSAlex Tomas {
2387c9de560dSAlex Tomas 	struct super_block *sb = PDE(inode)->data;
2388c9de560dSAlex Tomas 	int rc;
2389c9de560dSAlex Tomas 
2390c9de560dSAlex Tomas 	rc = seq_open(file, &ext4_mb_seq_groups_ops);
2391c9de560dSAlex Tomas 	if (rc == 0) {
2392c9de560dSAlex Tomas 		struct seq_file *m = (struct seq_file *)file->private_data;
2393c9de560dSAlex Tomas 		m->private = sb;
2394c9de560dSAlex Tomas 	}
2395c9de560dSAlex Tomas 	return rc;
2396c9de560dSAlex Tomas 
2397c9de560dSAlex Tomas }
2398c9de560dSAlex Tomas 
2399c9de560dSAlex Tomas static struct file_operations ext4_mb_seq_groups_fops = {
2400c9de560dSAlex Tomas 	.owner		= THIS_MODULE,
2401c9de560dSAlex Tomas 	.open		= ext4_mb_seq_groups_open,
2402c9de560dSAlex Tomas 	.read		= seq_read,
2403c9de560dSAlex Tomas 	.llseek		= seq_lseek,
2404c9de560dSAlex Tomas 	.release	= seq_release,
2405c9de560dSAlex Tomas };
2406c9de560dSAlex Tomas 
2407c9de560dSAlex Tomas static void ext4_mb_history_release(struct super_block *sb)
2408c9de560dSAlex Tomas {
2409c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2410c9de560dSAlex Tomas 
24119f6200bbSTheodore Ts'o 	if (sbi->s_proc != NULL) {
24129f6200bbSTheodore Ts'o 		remove_proc_entry("mb_groups", sbi->s_proc);
24139f6200bbSTheodore Ts'o 		remove_proc_entry("mb_history", sbi->s_proc);
24149f6200bbSTheodore Ts'o 	}
2415c9de560dSAlex Tomas 	kfree(sbi->s_mb_history);
2416c9de560dSAlex Tomas }
2417c9de560dSAlex Tomas 
2418c9de560dSAlex Tomas static void ext4_mb_history_init(struct super_block *sb)
2419c9de560dSAlex Tomas {
2420c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2421c9de560dSAlex Tomas 	int i;
2422c9de560dSAlex Tomas 
24239f6200bbSTheodore Ts'o 	if (sbi->s_proc != NULL) {
24249f6200bbSTheodore Ts'o 		proc_create_data("mb_history", S_IRUGO, sbi->s_proc,
242546fe74f2SDenis V. Lunev 				 &ext4_mb_seq_history_fops, sb);
24269f6200bbSTheodore Ts'o 		proc_create_data("mb_groups", S_IRUGO, sbi->s_proc,
242746fe74f2SDenis V. Lunev 				 &ext4_mb_seq_groups_fops, sb);
2428c9de560dSAlex Tomas 	}
2429c9de560dSAlex Tomas 
2430c9de560dSAlex Tomas 	sbi->s_mb_history_max = 1000;
2431c9de560dSAlex Tomas 	sbi->s_mb_history_cur = 0;
2432c9de560dSAlex Tomas 	spin_lock_init(&sbi->s_mb_history_lock);
2433c9de560dSAlex Tomas 	i = sbi->s_mb_history_max * sizeof(struct ext4_mb_history);
243474767c5aSShen Feng 	sbi->s_mb_history = kzalloc(i, GFP_KERNEL);
2435c9de560dSAlex Tomas 	/* if we can't allocate history, then we simple won't use it */
2436c9de560dSAlex Tomas }
2437c9de560dSAlex Tomas 
24384ddfef7bSEric Sandeen static noinline_for_stack void
24394ddfef7bSEric Sandeen ext4_mb_store_history(struct ext4_allocation_context *ac)
2440c9de560dSAlex Tomas {
2441c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(ac->ac_sb);
2442c9de560dSAlex Tomas 	struct ext4_mb_history h;
2443c9de560dSAlex Tomas 
2444c9de560dSAlex Tomas 	if (unlikely(sbi->s_mb_history == NULL))
2445c9de560dSAlex Tomas 		return;
2446c9de560dSAlex Tomas 
2447c9de560dSAlex Tomas 	if (!(ac->ac_op & sbi->s_mb_history_filter))
2448c9de560dSAlex Tomas 		return;
2449c9de560dSAlex Tomas 
2450c9de560dSAlex Tomas 	h.op = ac->ac_op;
2451c9de560dSAlex Tomas 	h.pid = current->pid;
2452c9de560dSAlex Tomas 	h.ino = ac->ac_inode ? ac->ac_inode->i_ino : 0;
2453c9de560dSAlex Tomas 	h.orig = ac->ac_o_ex;
2454c9de560dSAlex Tomas 	h.result = ac->ac_b_ex;
2455c9de560dSAlex Tomas 	h.flags = ac->ac_flags;
2456c9de560dSAlex Tomas 	h.found = ac->ac_found;
2457c9de560dSAlex Tomas 	h.groups = ac->ac_groups_scanned;
2458c9de560dSAlex Tomas 	h.cr = ac->ac_criteria;
2459c9de560dSAlex Tomas 	h.tail = ac->ac_tail;
2460c9de560dSAlex Tomas 	h.buddy = ac->ac_buddy;
2461c9de560dSAlex Tomas 	h.merged = 0;
2462c9de560dSAlex Tomas 	if (ac->ac_op == EXT4_MB_HISTORY_ALLOC) {
2463c9de560dSAlex Tomas 		if (ac->ac_g_ex.fe_start == ac->ac_b_ex.fe_start &&
2464c9de560dSAlex Tomas 				ac->ac_g_ex.fe_group == ac->ac_b_ex.fe_group)
2465c9de560dSAlex Tomas 			h.merged = 1;
2466c9de560dSAlex Tomas 		h.goal = ac->ac_g_ex;
2467c9de560dSAlex Tomas 		h.result = ac->ac_f_ex;
2468c9de560dSAlex Tomas 	}
2469c9de560dSAlex Tomas 
2470c9de560dSAlex Tomas 	spin_lock(&sbi->s_mb_history_lock);
2471c9de560dSAlex Tomas 	memcpy(sbi->s_mb_history + sbi->s_mb_history_cur, &h, sizeof(h));
2472c9de560dSAlex Tomas 	if (++sbi->s_mb_history_cur >= sbi->s_mb_history_max)
2473c9de560dSAlex Tomas 		sbi->s_mb_history_cur = 0;
2474c9de560dSAlex Tomas 	spin_unlock(&sbi->s_mb_history_lock);
2475c9de560dSAlex Tomas }
2476c9de560dSAlex Tomas 
2477c9de560dSAlex Tomas #else
2478c9de560dSAlex Tomas #define ext4_mb_history_release(sb)
2479c9de560dSAlex Tomas #define ext4_mb_history_init(sb)
2480c9de560dSAlex Tomas #endif
2481c9de560dSAlex Tomas 
24825f21b0e6SFrederic Bohe 
24835f21b0e6SFrederic Bohe /* Create and initialize ext4_group_info data for the given group. */
2484920313a7SAneesh Kumar K.V int ext4_mb_add_groupinfo(struct super_block *sb, ext4_group_t group,
24855f21b0e6SFrederic Bohe 			  struct ext4_group_desc *desc)
24865f21b0e6SFrederic Bohe {
24875f21b0e6SFrederic Bohe 	int i, len;
24885f21b0e6SFrederic Bohe 	int metalen = 0;
24895f21b0e6SFrederic Bohe 	struct ext4_sb_info *sbi = EXT4_SB(sb);
24905f21b0e6SFrederic Bohe 	struct ext4_group_info **meta_group_info;
24915f21b0e6SFrederic Bohe 
24925f21b0e6SFrederic Bohe 	/*
24935f21b0e6SFrederic Bohe 	 * First check if this group is the first of a reserved block.
24945f21b0e6SFrederic Bohe 	 * If it's true, we have to allocate a new table of pointers
24955f21b0e6SFrederic Bohe 	 * to ext4_group_info structures
24965f21b0e6SFrederic Bohe 	 */
24975f21b0e6SFrederic Bohe 	if (group % EXT4_DESC_PER_BLOCK(sb) == 0) {
24985f21b0e6SFrederic Bohe 		metalen = sizeof(*meta_group_info) <<
24995f21b0e6SFrederic Bohe 			EXT4_DESC_PER_BLOCK_BITS(sb);
25005f21b0e6SFrederic Bohe 		meta_group_info = kmalloc(metalen, GFP_KERNEL);
25015f21b0e6SFrederic Bohe 		if (meta_group_info == NULL) {
25025f21b0e6SFrederic Bohe 			printk(KERN_ERR "EXT4-fs: can't allocate mem for a "
25035f21b0e6SFrederic Bohe 			       "buddy group\n");
25045f21b0e6SFrederic Bohe 			goto exit_meta_group_info;
25055f21b0e6SFrederic Bohe 		}
25065f21b0e6SFrederic Bohe 		sbi->s_group_info[group >> EXT4_DESC_PER_BLOCK_BITS(sb)] =
25075f21b0e6SFrederic Bohe 			meta_group_info;
25085f21b0e6SFrederic Bohe 	}
25095f21b0e6SFrederic Bohe 
25105f21b0e6SFrederic Bohe 	/*
25115f21b0e6SFrederic Bohe 	 * calculate needed size. if change bb_counters size,
25125f21b0e6SFrederic Bohe 	 * don't forget about ext4_mb_generate_buddy()
25135f21b0e6SFrederic Bohe 	 */
25145f21b0e6SFrederic Bohe 	len = offsetof(typeof(**meta_group_info),
25155f21b0e6SFrederic Bohe 		       bb_counters[sb->s_blocksize_bits + 2]);
25165f21b0e6SFrederic Bohe 
25175f21b0e6SFrederic Bohe 	meta_group_info =
25185f21b0e6SFrederic Bohe 		sbi->s_group_info[group >> EXT4_DESC_PER_BLOCK_BITS(sb)];
25195f21b0e6SFrederic Bohe 	i = group & (EXT4_DESC_PER_BLOCK(sb) - 1);
25205f21b0e6SFrederic Bohe 
25215f21b0e6SFrederic Bohe 	meta_group_info[i] = kzalloc(len, GFP_KERNEL);
25225f21b0e6SFrederic Bohe 	if (meta_group_info[i] == NULL) {
25235f21b0e6SFrederic Bohe 		printk(KERN_ERR "EXT4-fs: can't allocate buddy mem\n");
25245f21b0e6SFrederic Bohe 		goto exit_group_info;
25255f21b0e6SFrederic Bohe 	}
25265f21b0e6SFrederic Bohe 	set_bit(EXT4_GROUP_INFO_NEED_INIT_BIT,
25275f21b0e6SFrederic Bohe 		&(meta_group_info[i]->bb_state));
25285f21b0e6SFrederic Bohe 
25295f21b0e6SFrederic Bohe 	/*
25305f21b0e6SFrederic Bohe 	 * initialize bb_free to be able to skip
25315f21b0e6SFrederic Bohe 	 * empty groups without initialization
25325f21b0e6SFrederic Bohe 	 */
25335f21b0e6SFrederic Bohe 	if (desc->bg_flags & cpu_to_le16(EXT4_BG_BLOCK_UNINIT)) {
25345f21b0e6SFrederic Bohe 		meta_group_info[i]->bb_free =
25355f21b0e6SFrederic Bohe 			ext4_free_blocks_after_init(sb, group, desc);
25365f21b0e6SFrederic Bohe 	} else {
25375f21b0e6SFrederic Bohe 		meta_group_info[i]->bb_free =
2538560671a0SAneesh Kumar K.V 			ext4_free_blks_count(sb, desc);
25395f21b0e6SFrederic Bohe 	}
25405f21b0e6SFrederic Bohe 
25415f21b0e6SFrederic Bohe 	INIT_LIST_HEAD(&meta_group_info[i]->bb_prealloc_list);
2542920313a7SAneesh Kumar K.V 	init_rwsem(&meta_group_info[i]->alloc_sem);
2543c894058dSAneesh Kumar K.V 	meta_group_info[i]->bb_free_root.rb_node = NULL;;
25445f21b0e6SFrederic Bohe 
25455f21b0e6SFrederic Bohe #ifdef DOUBLE_CHECK
25465f21b0e6SFrederic Bohe 	{
25475f21b0e6SFrederic Bohe 		struct buffer_head *bh;
25485f21b0e6SFrederic Bohe 		meta_group_info[i]->bb_bitmap =
25495f21b0e6SFrederic Bohe 			kmalloc(sb->s_blocksize, GFP_KERNEL);
25505f21b0e6SFrederic Bohe 		BUG_ON(meta_group_info[i]->bb_bitmap == NULL);
25515f21b0e6SFrederic Bohe 		bh = ext4_read_block_bitmap(sb, group);
25525f21b0e6SFrederic Bohe 		BUG_ON(bh == NULL);
25535f21b0e6SFrederic Bohe 		memcpy(meta_group_info[i]->bb_bitmap, bh->b_data,
25545f21b0e6SFrederic Bohe 			sb->s_blocksize);
25555f21b0e6SFrederic Bohe 		put_bh(bh);
25565f21b0e6SFrederic Bohe 	}
25575f21b0e6SFrederic Bohe #endif
25585f21b0e6SFrederic Bohe 
25595f21b0e6SFrederic Bohe 	return 0;
25605f21b0e6SFrederic Bohe 
25615f21b0e6SFrederic Bohe exit_group_info:
25625f21b0e6SFrederic Bohe 	/* If a meta_group_info table has been allocated, release it now */
25635f21b0e6SFrederic Bohe 	if (group % EXT4_DESC_PER_BLOCK(sb) == 0)
25645f21b0e6SFrederic Bohe 		kfree(sbi->s_group_info[group >> EXT4_DESC_PER_BLOCK_BITS(sb)]);
25655f21b0e6SFrederic Bohe exit_meta_group_info:
25665f21b0e6SFrederic Bohe 	return -ENOMEM;
25675f21b0e6SFrederic Bohe } /* ext4_mb_add_groupinfo */
25685f21b0e6SFrederic Bohe 
25695f21b0e6SFrederic Bohe /*
25705f21b0e6SFrederic Bohe  * Update an existing group.
25715f21b0e6SFrederic Bohe  * This function is used for online resize
25725f21b0e6SFrederic Bohe  */
25735f21b0e6SFrederic Bohe void ext4_mb_update_group_info(struct ext4_group_info *grp, ext4_grpblk_t add)
25745f21b0e6SFrederic Bohe {
25755f21b0e6SFrederic Bohe 	grp->bb_free += add;
25765f21b0e6SFrederic Bohe }
25775f21b0e6SFrederic Bohe 
2578c9de560dSAlex Tomas static int ext4_mb_init_backend(struct super_block *sb)
2579c9de560dSAlex Tomas {
2580c9de560dSAlex Tomas 	ext4_group_t i;
25815f21b0e6SFrederic Bohe 	int metalen;
2582c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
25835f21b0e6SFrederic Bohe 	struct ext4_super_block *es = sbi->s_es;
25845f21b0e6SFrederic Bohe 	int num_meta_group_infos;
25855f21b0e6SFrederic Bohe 	int num_meta_group_infos_max;
25865f21b0e6SFrederic Bohe 	int array_size;
2587c9de560dSAlex Tomas 	struct ext4_group_info **meta_group_info;
25885f21b0e6SFrederic Bohe 	struct ext4_group_desc *desc;
2589c9de560dSAlex Tomas 
25905f21b0e6SFrederic Bohe 	/* This is the number of blocks used by GDT */
25915f21b0e6SFrederic Bohe 	num_meta_group_infos = (sbi->s_groups_count + EXT4_DESC_PER_BLOCK(sb) -
25925f21b0e6SFrederic Bohe 				1) >> EXT4_DESC_PER_BLOCK_BITS(sb);
25935f21b0e6SFrederic Bohe 
25945f21b0e6SFrederic Bohe 	/*
25955f21b0e6SFrederic Bohe 	 * This is the total number of blocks used by GDT including
25965f21b0e6SFrederic Bohe 	 * the number of reserved blocks for GDT.
25975f21b0e6SFrederic Bohe 	 * The s_group_info array is allocated with this value
25985f21b0e6SFrederic Bohe 	 * to allow a clean online resize without a complex
25995f21b0e6SFrederic Bohe 	 * manipulation of pointer.
26005f21b0e6SFrederic Bohe 	 * The drawback is the unused memory when no resize
26015f21b0e6SFrederic Bohe 	 * occurs but it's very low in terms of pages
26025f21b0e6SFrederic Bohe 	 * (see comments below)
26035f21b0e6SFrederic Bohe 	 * Need to handle this properly when META_BG resizing is allowed
26045f21b0e6SFrederic Bohe 	 */
26055f21b0e6SFrederic Bohe 	num_meta_group_infos_max = num_meta_group_infos +
26065f21b0e6SFrederic Bohe 				le16_to_cpu(es->s_reserved_gdt_blocks);
26075f21b0e6SFrederic Bohe 
26085f21b0e6SFrederic Bohe 	/*
26095f21b0e6SFrederic Bohe 	 * array_size is the size of s_group_info array. We round it
26105f21b0e6SFrederic Bohe 	 * to the next power of two because this approximation is done
26115f21b0e6SFrederic Bohe 	 * internally by kmalloc so we can have some more memory
26125f21b0e6SFrederic Bohe 	 * for free here (e.g. may be used for META_BG resize).
26135f21b0e6SFrederic Bohe 	 */
26145f21b0e6SFrederic Bohe 	array_size = 1;
26155f21b0e6SFrederic Bohe 	while (array_size < sizeof(*sbi->s_group_info) *
26165f21b0e6SFrederic Bohe 	       num_meta_group_infos_max)
26175f21b0e6SFrederic Bohe 		array_size = array_size << 1;
2618c9de560dSAlex Tomas 	/* An 8TB filesystem with 64-bit pointers requires a 4096 byte
2619c9de560dSAlex Tomas 	 * kmalloc. A 128kb malloc should suffice for a 256TB filesystem.
2620c9de560dSAlex Tomas 	 * So a two level scheme suffices for now. */
26215f21b0e6SFrederic Bohe 	sbi->s_group_info = kmalloc(array_size, GFP_KERNEL);
2622c9de560dSAlex Tomas 	if (sbi->s_group_info == NULL) {
2623c9de560dSAlex Tomas 		printk(KERN_ERR "EXT4-fs: can't allocate buddy meta group\n");
2624c9de560dSAlex Tomas 		return -ENOMEM;
2625c9de560dSAlex Tomas 	}
2626c9de560dSAlex Tomas 	sbi->s_buddy_cache = new_inode(sb);
2627c9de560dSAlex Tomas 	if (sbi->s_buddy_cache == NULL) {
2628c9de560dSAlex Tomas 		printk(KERN_ERR "EXT4-fs: can't get new inode\n");
2629c9de560dSAlex Tomas 		goto err_freesgi;
2630c9de560dSAlex Tomas 	}
2631c9de560dSAlex Tomas 	EXT4_I(sbi->s_buddy_cache)->i_disksize = 0;
2632c9de560dSAlex Tomas 
2633c9de560dSAlex Tomas 	metalen = sizeof(*meta_group_info) << EXT4_DESC_PER_BLOCK_BITS(sb);
2634c9de560dSAlex Tomas 	for (i = 0; i < num_meta_group_infos; i++) {
2635c9de560dSAlex Tomas 		if ((i + 1) == num_meta_group_infos)
2636c9de560dSAlex Tomas 			metalen = sizeof(*meta_group_info) *
2637c9de560dSAlex Tomas 				(sbi->s_groups_count -
2638c9de560dSAlex Tomas 					(i << EXT4_DESC_PER_BLOCK_BITS(sb)));
2639c9de560dSAlex Tomas 		meta_group_info = kmalloc(metalen, GFP_KERNEL);
2640c9de560dSAlex Tomas 		if (meta_group_info == NULL) {
2641c9de560dSAlex Tomas 			printk(KERN_ERR "EXT4-fs: can't allocate mem for a "
2642c9de560dSAlex Tomas 			       "buddy group\n");
2643c9de560dSAlex Tomas 			goto err_freemeta;
2644c9de560dSAlex Tomas 		}
2645c9de560dSAlex Tomas 		sbi->s_group_info[i] = meta_group_info;
2646c9de560dSAlex Tomas 	}
2647c9de560dSAlex Tomas 
2648c9de560dSAlex Tomas 	for (i = 0; i < sbi->s_groups_count; i++) {
2649c9de560dSAlex Tomas 		desc = ext4_get_group_desc(sb, i, NULL);
2650c9de560dSAlex Tomas 		if (desc == NULL) {
2651c9de560dSAlex Tomas 			printk(KERN_ERR
2652a9df9a49STheodore Ts'o 				"EXT4-fs: can't read descriptor %u\n", i);
2653c9de560dSAlex Tomas 			goto err_freebuddy;
2654c9de560dSAlex Tomas 		}
26555f21b0e6SFrederic Bohe 		if (ext4_mb_add_groupinfo(sb, i, desc) != 0)
26565f21b0e6SFrederic Bohe 			goto err_freebuddy;
2657c9de560dSAlex Tomas 	}
2658c9de560dSAlex Tomas 
2659c9de560dSAlex Tomas 	return 0;
2660c9de560dSAlex Tomas 
2661c9de560dSAlex Tomas err_freebuddy:
2662f1fa3342SRoel Kluin 	while (i-- > 0)
2663c9de560dSAlex Tomas 		kfree(ext4_get_group_info(sb, i));
2664c9de560dSAlex Tomas 	i = num_meta_group_infos;
2665c9de560dSAlex Tomas err_freemeta:
2666f1fa3342SRoel Kluin 	while (i-- > 0)
2667c9de560dSAlex Tomas 		kfree(sbi->s_group_info[i]);
2668c9de560dSAlex Tomas 	iput(sbi->s_buddy_cache);
2669c9de560dSAlex Tomas err_freesgi:
2670c9de560dSAlex Tomas 	kfree(sbi->s_group_info);
2671c9de560dSAlex Tomas 	return -ENOMEM;
2672c9de560dSAlex Tomas }
2673c9de560dSAlex Tomas 
2674c9de560dSAlex Tomas int ext4_mb_init(struct super_block *sb, int needs_recovery)
2675c9de560dSAlex Tomas {
2676c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
26776be2ded1SAneesh Kumar K.V 	unsigned i, j;
2678c9de560dSAlex Tomas 	unsigned offset;
2679c9de560dSAlex Tomas 	unsigned max;
268074767c5aSShen Feng 	int ret;
2681c9de560dSAlex Tomas 
2682c9de560dSAlex Tomas 	i = (sb->s_blocksize_bits + 2) * sizeof(unsigned short);
2683c9de560dSAlex Tomas 
2684c9de560dSAlex Tomas 	sbi->s_mb_offsets = kmalloc(i, GFP_KERNEL);
2685c9de560dSAlex Tomas 	if (sbi->s_mb_offsets == NULL) {
2686c9de560dSAlex Tomas 		return -ENOMEM;
2687c9de560dSAlex Tomas 	}
2688ff7ef329SYasunori Goto 
2689ff7ef329SYasunori Goto 	i = (sb->s_blocksize_bits + 2) * sizeof(unsigned int);
2690c9de560dSAlex Tomas 	sbi->s_mb_maxs = kmalloc(i, GFP_KERNEL);
2691c9de560dSAlex Tomas 	if (sbi->s_mb_maxs == NULL) {
2692c9de560dSAlex Tomas 		kfree(sbi->s_mb_maxs);
2693c9de560dSAlex Tomas 		return -ENOMEM;
2694c9de560dSAlex Tomas 	}
2695c9de560dSAlex Tomas 
2696c9de560dSAlex Tomas 	/* order 0 is regular bitmap */
2697c9de560dSAlex Tomas 	sbi->s_mb_maxs[0] = sb->s_blocksize << 3;
2698c9de560dSAlex Tomas 	sbi->s_mb_offsets[0] = 0;
2699c9de560dSAlex Tomas 
2700c9de560dSAlex Tomas 	i = 1;
2701c9de560dSAlex Tomas 	offset = 0;
2702c9de560dSAlex Tomas 	max = sb->s_blocksize << 2;
2703c9de560dSAlex Tomas 	do {
2704c9de560dSAlex Tomas 		sbi->s_mb_offsets[i] = offset;
2705c9de560dSAlex Tomas 		sbi->s_mb_maxs[i] = max;
2706c9de560dSAlex Tomas 		offset += 1 << (sb->s_blocksize_bits - i);
2707c9de560dSAlex Tomas 		max = max >> 1;
2708c9de560dSAlex Tomas 		i++;
2709c9de560dSAlex Tomas 	} while (i <= sb->s_blocksize_bits + 1);
2710c9de560dSAlex Tomas 
2711c9de560dSAlex Tomas 	/* init file for buddy data */
271274767c5aSShen Feng 	ret = ext4_mb_init_backend(sb);
271374767c5aSShen Feng 	if (ret != 0) {
2714c9de560dSAlex Tomas 		kfree(sbi->s_mb_offsets);
2715c9de560dSAlex Tomas 		kfree(sbi->s_mb_maxs);
271674767c5aSShen Feng 		return ret;
2717c9de560dSAlex Tomas 	}
2718c9de560dSAlex Tomas 
2719c9de560dSAlex Tomas 	spin_lock_init(&sbi->s_md_lock);
2720c9de560dSAlex Tomas 	spin_lock_init(&sbi->s_bal_lock);
2721c9de560dSAlex Tomas 
2722c9de560dSAlex Tomas 	sbi->s_mb_max_to_scan = MB_DEFAULT_MAX_TO_SCAN;
2723c9de560dSAlex Tomas 	sbi->s_mb_min_to_scan = MB_DEFAULT_MIN_TO_SCAN;
2724c9de560dSAlex Tomas 	sbi->s_mb_stats = MB_DEFAULT_STATS;
2725c9de560dSAlex Tomas 	sbi->s_mb_stream_request = MB_DEFAULT_STREAM_THRESHOLD;
2726c9de560dSAlex Tomas 	sbi->s_mb_order2_reqs = MB_DEFAULT_ORDER2_REQS;
2727c9de560dSAlex Tomas 	sbi->s_mb_history_filter = EXT4_MB_HISTORY_DEFAULT;
2728c9de560dSAlex Tomas 	sbi->s_mb_group_prealloc = MB_DEFAULT_GROUP_PREALLOC;
2729c9de560dSAlex Tomas 
2730730c213cSEric Sandeen 	sbi->s_locality_groups = alloc_percpu(struct ext4_locality_group);
2731c9de560dSAlex Tomas 	if (sbi->s_locality_groups == NULL) {
2732c9de560dSAlex Tomas 		kfree(sbi->s_mb_offsets);
2733c9de560dSAlex Tomas 		kfree(sbi->s_mb_maxs);
2734c9de560dSAlex Tomas 		return -ENOMEM;
2735c9de560dSAlex Tomas 	}
2736730c213cSEric Sandeen 	for_each_possible_cpu(i) {
2737c9de560dSAlex Tomas 		struct ext4_locality_group *lg;
2738730c213cSEric Sandeen 		lg = per_cpu_ptr(sbi->s_locality_groups, i);
2739c9de560dSAlex Tomas 		mutex_init(&lg->lg_mutex);
27406be2ded1SAneesh Kumar K.V 		for (j = 0; j < PREALLOC_TB_SIZE; j++)
27416be2ded1SAneesh Kumar K.V 			INIT_LIST_HEAD(&lg->lg_prealloc_list[j]);
2742c9de560dSAlex Tomas 		spin_lock_init(&lg->lg_prealloc_lock);
2743c9de560dSAlex Tomas 	}
2744c9de560dSAlex Tomas 
2745c9de560dSAlex Tomas 	ext4_mb_init_per_dev_proc(sb);
2746c9de560dSAlex Tomas 	ext4_mb_history_init(sb);
2747c9de560dSAlex Tomas 
27480390131bSFrank Mayhar 	if (sbi->s_journal)
27493e624fc7STheodore Ts'o 		sbi->s_journal->j_commit_callback = release_blocks_on_commit;
27503e624fc7STheodore Ts'o 
27514776004fSTheodore Ts'o 	printk(KERN_INFO "EXT4-fs: mballoc enabled\n");
2752c9de560dSAlex Tomas 	return 0;
2753c9de560dSAlex Tomas }
2754c9de560dSAlex Tomas 
2755c9de560dSAlex Tomas /* need to called with ext4 group lock (ext4_lock_group) */
2756c9de560dSAlex Tomas static void ext4_mb_cleanup_pa(struct ext4_group_info *grp)
2757c9de560dSAlex Tomas {
2758c9de560dSAlex Tomas 	struct ext4_prealloc_space *pa;
2759c9de560dSAlex Tomas 	struct list_head *cur, *tmp;
2760c9de560dSAlex Tomas 	int count = 0;
2761c9de560dSAlex Tomas 
2762c9de560dSAlex Tomas 	list_for_each_safe(cur, tmp, &grp->bb_prealloc_list) {
2763c9de560dSAlex Tomas 		pa = list_entry(cur, struct ext4_prealloc_space, pa_group_list);
2764c9de560dSAlex Tomas 		list_del(&pa->pa_group_list);
2765c9de560dSAlex Tomas 		count++;
2766688f05a0SAneesh Kumar K.V 		kmem_cache_free(ext4_pspace_cachep, pa);
2767c9de560dSAlex Tomas 	}
2768c9de560dSAlex Tomas 	if (count)
2769c9de560dSAlex Tomas 		mb_debug("mballoc: %u PAs left\n", count);
2770c9de560dSAlex Tomas 
2771c9de560dSAlex Tomas }
2772c9de560dSAlex Tomas 
2773c9de560dSAlex Tomas int ext4_mb_release(struct super_block *sb)
2774c9de560dSAlex Tomas {
2775c9de560dSAlex Tomas 	ext4_group_t i;
2776c9de560dSAlex Tomas 	int num_meta_group_infos;
2777c9de560dSAlex Tomas 	struct ext4_group_info *grinfo;
2778c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2779c9de560dSAlex Tomas 
2780c9de560dSAlex Tomas 	if (sbi->s_group_info) {
2781c9de560dSAlex Tomas 		for (i = 0; i < sbi->s_groups_count; i++) {
2782c9de560dSAlex Tomas 			grinfo = ext4_get_group_info(sb, i);
2783c9de560dSAlex Tomas #ifdef DOUBLE_CHECK
2784c9de560dSAlex Tomas 			kfree(grinfo->bb_bitmap);
2785c9de560dSAlex Tomas #endif
2786c9de560dSAlex Tomas 			ext4_lock_group(sb, i);
2787c9de560dSAlex Tomas 			ext4_mb_cleanup_pa(grinfo);
2788c9de560dSAlex Tomas 			ext4_unlock_group(sb, i);
2789c9de560dSAlex Tomas 			kfree(grinfo);
2790c9de560dSAlex Tomas 		}
2791c9de560dSAlex Tomas 		num_meta_group_infos = (sbi->s_groups_count +
2792c9de560dSAlex Tomas 				EXT4_DESC_PER_BLOCK(sb) - 1) >>
2793c9de560dSAlex Tomas 			EXT4_DESC_PER_BLOCK_BITS(sb);
2794c9de560dSAlex Tomas 		for (i = 0; i < num_meta_group_infos; i++)
2795c9de560dSAlex Tomas 			kfree(sbi->s_group_info[i]);
2796c9de560dSAlex Tomas 		kfree(sbi->s_group_info);
2797c9de560dSAlex Tomas 	}
2798c9de560dSAlex Tomas 	kfree(sbi->s_mb_offsets);
2799c9de560dSAlex Tomas 	kfree(sbi->s_mb_maxs);
2800c9de560dSAlex Tomas 	if (sbi->s_buddy_cache)
2801c9de560dSAlex Tomas 		iput(sbi->s_buddy_cache);
2802c9de560dSAlex Tomas 	if (sbi->s_mb_stats) {
2803c9de560dSAlex Tomas 		printk(KERN_INFO
2804c9de560dSAlex Tomas 		       "EXT4-fs: mballoc: %u blocks %u reqs (%u success)\n",
2805c9de560dSAlex Tomas 				atomic_read(&sbi->s_bal_allocated),
2806c9de560dSAlex Tomas 				atomic_read(&sbi->s_bal_reqs),
2807c9de560dSAlex Tomas 				atomic_read(&sbi->s_bal_success));
2808c9de560dSAlex Tomas 		printk(KERN_INFO
2809c9de560dSAlex Tomas 		      "EXT4-fs: mballoc: %u extents scanned, %u goal hits, "
2810c9de560dSAlex Tomas 				"%u 2^N hits, %u breaks, %u lost\n",
2811c9de560dSAlex Tomas 				atomic_read(&sbi->s_bal_ex_scanned),
2812c9de560dSAlex Tomas 				atomic_read(&sbi->s_bal_goals),
2813c9de560dSAlex Tomas 				atomic_read(&sbi->s_bal_2orders),
2814c9de560dSAlex Tomas 				atomic_read(&sbi->s_bal_breaks),
2815c9de560dSAlex Tomas 				atomic_read(&sbi->s_mb_lost_chunks));
2816c9de560dSAlex Tomas 		printk(KERN_INFO
2817c9de560dSAlex Tomas 		       "EXT4-fs: mballoc: %lu generated and it took %Lu\n",
2818c9de560dSAlex Tomas 				sbi->s_mb_buddies_generated++,
2819c9de560dSAlex Tomas 				sbi->s_mb_generation_time);
2820c9de560dSAlex Tomas 		printk(KERN_INFO
2821c9de560dSAlex Tomas 		       "EXT4-fs: mballoc: %u preallocated, %u discarded\n",
2822c9de560dSAlex Tomas 				atomic_read(&sbi->s_mb_preallocated),
2823c9de560dSAlex Tomas 				atomic_read(&sbi->s_mb_discarded));
2824c9de560dSAlex Tomas 	}
2825c9de560dSAlex Tomas 
2826730c213cSEric Sandeen 	free_percpu(sbi->s_locality_groups);
2827c9de560dSAlex Tomas 	ext4_mb_history_release(sb);
2828c9de560dSAlex Tomas 	ext4_mb_destroy_per_dev_proc(sb);
2829c9de560dSAlex Tomas 
2830c9de560dSAlex Tomas 	return 0;
2831c9de560dSAlex Tomas }
2832c9de560dSAlex Tomas 
28333e624fc7STheodore Ts'o /*
28343e624fc7STheodore Ts'o  * This function is called by the jbd2 layer once the commit has finished,
28353e624fc7STheodore Ts'o  * so we know we can free the blocks that were released with that commit.
28363e624fc7STheodore Ts'o  */
28373e624fc7STheodore Ts'o static void release_blocks_on_commit(journal_t *journal, transaction_t *txn)
2838c9de560dSAlex Tomas {
28393e624fc7STheodore Ts'o 	struct super_block *sb = journal->j_private;
2840c9de560dSAlex Tomas 	struct ext4_buddy e4b;
2841c894058dSAneesh Kumar K.V 	struct ext4_group_info *db;
2842c894058dSAneesh Kumar K.V 	int err, count = 0, count2 = 0;
2843c894058dSAneesh Kumar K.V 	struct ext4_free_data *entry;
28448a0aba73STheodore Ts'o 	ext4_fsblk_t discard_block;
28453e624fc7STheodore Ts'o 	struct list_head *l, *ltmp;
2846c9de560dSAlex Tomas 
28473e624fc7STheodore Ts'o 	list_for_each_safe(l, ltmp, &txn->t_private_list) {
28483e624fc7STheodore Ts'o 		entry = list_entry(l, struct ext4_free_data, list);
2849c9de560dSAlex Tomas 
2850a9df9a49STheodore Ts'o 		mb_debug("gonna free %u blocks in group %u (0x%p):",
2851c894058dSAneesh Kumar K.V 			 entry->count, entry->group, entry);
2852c9de560dSAlex Tomas 
2853c894058dSAneesh Kumar K.V 		err = ext4_mb_load_buddy(sb, entry->group, &e4b);
2854c9de560dSAlex Tomas 		/* we expect to find existing buddy because it's pinned */
2855c9de560dSAlex Tomas 		BUG_ON(err != 0);
2856c9de560dSAlex Tomas 
2857c894058dSAneesh Kumar K.V 		db = e4b.bd_info;
2858c9de560dSAlex Tomas 		/* there are blocks to put in buddy to make them really free */
2859c894058dSAneesh Kumar K.V 		count += entry->count;
2860c9de560dSAlex Tomas 		count2++;
2861c894058dSAneesh Kumar K.V 		ext4_lock_group(sb, entry->group);
2862c894058dSAneesh Kumar K.V 		/* Take it out of per group rb tree */
2863c894058dSAneesh Kumar K.V 		rb_erase(&entry->node, &(db->bb_free_root));
2864c894058dSAneesh Kumar K.V 		mb_free_blocks(NULL, &e4b, entry->start_blk, entry->count);
2865c9de560dSAlex Tomas 
2866c894058dSAneesh Kumar K.V 		if (!db->bb_free_root.rb_node) {
2867c894058dSAneesh Kumar K.V 			/* No more items in the per group rb tree
2868c894058dSAneesh Kumar K.V 			 * balance refcounts from ext4_mb_free_metadata()
2869c894058dSAneesh Kumar K.V 			 */
2870c9de560dSAlex Tomas 			page_cache_release(e4b.bd_buddy_page);
2871c9de560dSAlex Tomas 			page_cache_release(e4b.bd_bitmap_page);
2872c894058dSAneesh Kumar K.V 		}
2873c894058dSAneesh Kumar K.V 		ext4_unlock_group(sb, entry->group);
28748a0aba73STheodore Ts'o 		discard_block = (ext4_fsblk_t) entry->group * EXT4_BLOCKS_PER_GROUP(sb)
28758a0aba73STheodore Ts'o 			+ entry->start_blk
28768a0aba73STheodore Ts'o 			+ le32_to_cpu(EXT4_SB(sb)->s_es->s_first_data_block);
28778a0aba73STheodore Ts'o 		trace_mark(ext4_discard_blocks, "dev %s blk %llu count %u", sb->s_id,
28788a0aba73STheodore Ts'o 			   (unsigned long long) discard_block, entry->count);
28798a0aba73STheodore Ts'o 		sb_issue_discard(sb, discard_block, entry->count);
2880c9de560dSAlex Tomas 
2881c894058dSAneesh Kumar K.V 		kmem_cache_free(ext4_free_ext_cachep, entry);
2882c9de560dSAlex Tomas 		ext4_mb_release_desc(&e4b);
28833e624fc7STheodore Ts'o 	}
2884c9de560dSAlex Tomas 
2885c9de560dSAlex Tomas 	mb_debug("freed %u blocks in %u structures\n", count, count2);
2886c9de560dSAlex Tomas }
2887c9de560dSAlex Tomas 
2888c9de560dSAlex Tomas #define EXT4_MB_STATS_NAME		"stats"
2889c9de560dSAlex Tomas #define EXT4_MB_MAX_TO_SCAN_NAME	"max_to_scan"
2890c9de560dSAlex Tomas #define EXT4_MB_MIN_TO_SCAN_NAME	"min_to_scan"
2891c9de560dSAlex Tomas #define EXT4_MB_ORDER2_REQ		"order2_req"
2892c9de560dSAlex Tomas #define EXT4_MB_STREAM_REQ		"stream_req"
2893c9de560dSAlex Tomas #define EXT4_MB_GROUP_PREALLOC		"group_prealloc"
2894c9de560dSAlex Tomas 
2895c9de560dSAlex Tomas static int ext4_mb_init_per_dev_proc(struct super_block *sb)
2896c9de560dSAlex Tomas {
28970b09923eSManish Katiyar #ifdef CONFIG_PROC_FS
2898c9de560dSAlex Tomas 	mode_t mode = S_IFREG | S_IRUGO | S_IWUSR;
2899c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2900c9de560dSAlex Tomas 	struct proc_dir_entry *proc;
2901c9de560dSAlex Tomas 
29029f6200bbSTheodore Ts'o 	if (sbi->s_proc == NULL)
2903cfbe7e4fSShen Feng 		return -EINVAL;
2904c9de560dSAlex Tomas 
29055e8814f2STheodore Ts'o 	EXT4_PROC_HANDLER(EXT4_MB_STATS_NAME, mb_stats);
29065e8814f2STheodore Ts'o 	EXT4_PROC_HANDLER(EXT4_MB_MAX_TO_SCAN_NAME, mb_max_to_scan);
29075e8814f2STheodore Ts'o 	EXT4_PROC_HANDLER(EXT4_MB_MIN_TO_SCAN_NAME, mb_min_to_scan);
29085e8814f2STheodore Ts'o 	EXT4_PROC_HANDLER(EXT4_MB_ORDER2_REQ, mb_order2_reqs);
29095e8814f2STheodore Ts'o 	EXT4_PROC_HANDLER(EXT4_MB_STREAM_REQ, mb_stream_request);
29105e8814f2STheodore Ts'o 	EXT4_PROC_HANDLER(EXT4_MB_GROUP_PREALLOC, mb_group_prealloc);
2911c9de560dSAlex Tomas 	return 0;
2912c9de560dSAlex Tomas 
2913c9de560dSAlex Tomas err_out:
29149f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_GROUP_PREALLOC, sbi->s_proc);
29159f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_STREAM_REQ, sbi->s_proc);
29169f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_ORDER2_REQ, sbi->s_proc);
29179f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_MIN_TO_SCAN_NAME, sbi->s_proc);
29189f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_MAX_TO_SCAN_NAME, sbi->s_proc);
29199f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_STATS_NAME, sbi->s_proc);
2920c9de560dSAlex Tomas 	return -ENOMEM;
29210b09923eSManish Katiyar #else
29220b09923eSManish Katiyar 	return 0;
29230b09923eSManish Katiyar #endif
2924c9de560dSAlex Tomas }
2925c9de560dSAlex Tomas 
2926c9de560dSAlex Tomas static int ext4_mb_destroy_per_dev_proc(struct super_block *sb)
2927c9de560dSAlex Tomas {
29280b09923eSManish Katiyar #ifdef CONFIG_PROC_FS
2929c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
2930c9de560dSAlex Tomas 
29319f6200bbSTheodore Ts'o 	if (sbi->s_proc == NULL)
2932c9de560dSAlex Tomas 		return -EINVAL;
2933c9de560dSAlex Tomas 
29349f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_GROUP_PREALLOC, sbi->s_proc);
29359f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_STREAM_REQ, sbi->s_proc);
29369f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_ORDER2_REQ, sbi->s_proc);
29379f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_MIN_TO_SCAN_NAME, sbi->s_proc);
29389f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_MAX_TO_SCAN_NAME, sbi->s_proc);
29399f6200bbSTheodore Ts'o 	remove_proc_entry(EXT4_MB_STATS_NAME, sbi->s_proc);
29400b09923eSManish Katiyar #endif
2941c9de560dSAlex Tomas 	return 0;
2942c9de560dSAlex Tomas }
2943c9de560dSAlex Tomas 
2944c9de560dSAlex Tomas int __init init_ext4_mballoc(void)
2945c9de560dSAlex Tomas {
2946c9de560dSAlex Tomas 	ext4_pspace_cachep =
2947c9de560dSAlex Tomas 		kmem_cache_create("ext4_prealloc_space",
2948c9de560dSAlex Tomas 				     sizeof(struct ext4_prealloc_space),
2949c9de560dSAlex Tomas 				     0, SLAB_RECLAIM_ACCOUNT, NULL);
2950c9de560dSAlex Tomas 	if (ext4_pspace_cachep == NULL)
2951c9de560dSAlex Tomas 		return -ENOMEM;
2952c9de560dSAlex Tomas 
2953256bdb49SEric Sandeen 	ext4_ac_cachep =
2954256bdb49SEric Sandeen 		kmem_cache_create("ext4_alloc_context",
2955256bdb49SEric Sandeen 				     sizeof(struct ext4_allocation_context),
2956256bdb49SEric Sandeen 				     0, SLAB_RECLAIM_ACCOUNT, NULL);
2957256bdb49SEric Sandeen 	if (ext4_ac_cachep == NULL) {
2958256bdb49SEric Sandeen 		kmem_cache_destroy(ext4_pspace_cachep);
2959256bdb49SEric Sandeen 		return -ENOMEM;
2960256bdb49SEric Sandeen 	}
2961c894058dSAneesh Kumar K.V 
2962c894058dSAneesh Kumar K.V 	ext4_free_ext_cachep =
2963c894058dSAneesh Kumar K.V 		kmem_cache_create("ext4_free_block_extents",
2964c894058dSAneesh Kumar K.V 				     sizeof(struct ext4_free_data),
2965c894058dSAneesh Kumar K.V 				     0, SLAB_RECLAIM_ACCOUNT, NULL);
2966c894058dSAneesh Kumar K.V 	if (ext4_free_ext_cachep == NULL) {
2967c894058dSAneesh Kumar K.V 		kmem_cache_destroy(ext4_pspace_cachep);
2968c894058dSAneesh Kumar K.V 		kmem_cache_destroy(ext4_ac_cachep);
2969c894058dSAneesh Kumar K.V 		return -ENOMEM;
2970c894058dSAneesh Kumar K.V 	}
2971c9de560dSAlex Tomas 	return 0;
2972c9de560dSAlex Tomas }
2973c9de560dSAlex Tomas 
2974c9de560dSAlex Tomas void exit_ext4_mballoc(void)
2975c9de560dSAlex Tomas {
2976c9de560dSAlex Tomas 	/* XXX: synchronize_rcu(); */
2977c9de560dSAlex Tomas 	kmem_cache_destroy(ext4_pspace_cachep);
2978256bdb49SEric Sandeen 	kmem_cache_destroy(ext4_ac_cachep);
2979c894058dSAneesh Kumar K.V 	kmem_cache_destroy(ext4_free_ext_cachep);
2980c9de560dSAlex Tomas }
2981c9de560dSAlex Tomas 
2982c9de560dSAlex Tomas 
2983c9de560dSAlex Tomas /*
2984c9de560dSAlex Tomas  * Check quota and mark choosed space (ac->ac_b_ex) non-free in bitmaps
2985c9de560dSAlex Tomas  * Returns 0 if success or error code
2986c9de560dSAlex Tomas  */
29874ddfef7bSEric Sandeen static noinline_for_stack int
29884ddfef7bSEric Sandeen ext4_mb_mark_diskspace_used(struct ext4_allocation_context *ac,
2989498e5f24STheodore Ts'o 				handle_t *handle, unsigned int reserv_blks)
2990c9de560dSAlex Tomas {
2991c9de560dSAlex Tomas 	struct buffer_head *bitmap_bh = NULL;
2992c9de560dSAlex Tomas 	struct ext4_super_block *es;
2993c9de560dSAlex Tomas 	struct ext4_group_desc *gdp;
2994c9de560dSAlex Tomas 	struct buffer_head *gdp_bh;
2995c9de560dSAlex Tomas 	struct ext4_sb_info *sbi;
2996c9de560dSAlex Tomas 	struct super_block *sb;
2997c9de560dSAlex Tomas 	ext4_fsblk_t block;
2998519deca0SAneesh Kumar K.V 	int err, len;
2999c9de560dSAlex Tomas 
3000c9de560dSAlex Tomas 	BUG_ON(ac->ac_status != AC_STATUS_FOUND);
3001c9de560dSAlex Tomas 	BUG_ON(ac->ac_b_ex.fe_len <= 0);
3002c9de560dSAlex Tomas 
3003c9de560dSAlex Tomas 	sb = ac->ac_sb;
3004c9de560dSAlex Tomas 	sbi = EXT4_SB(sb);
3005c9de560dSAlex Tomas 	es = sbi->s_es;
3006c9de560dSAlex Tomas 
3007c9de560dSAlex Tomas 
3008c9de560dSAlex Tomas 	err = -EIO;
3009574ca174STheodore Ts'o 	bitmap_bh = ext4_read_block_bitmap(sb, ac->ac_b_ex.fe_group);
3010c9de560dSAlex Tomas 	if (!bitmap_bh)
3011c9de560dSAlex Tomas 		goto out_err;
3012c9de560dSAlex Tomas 
3013c9de560dSAlex Tomas 	err = ext4_journal_get_write_access(handle, bitmap_bh);
3014c9de560dSAlex Tomas 	if (err)
3015c9de560dSAlex Tomas 		goto out_err;
3016c9de560dSAlex Tomas 
3017c9de560dSAlex Tomas 	err = -EIO;
3018c9de560dSAlex Tomas 	gdp = ext4_get_group_desc(sb, ac->ac_b_ex.fe_group, &gdp_bh);
3019c9de560dSAlex Tomas 	if (!gdp)
3020c9de560dSAlex Tomas 		goto out_err;
3021c9de560dSAlex Tomas 
3022a9df9a49STheodore Ts'o 	ext4_debug("using block group %u(%d)\n", ac->ac_b_ex.fe_group,
302303cddb80SAneesh Kumar K.V 			gdp->bg_free_blocks_count);
302403cddb80SAneesh Kumar K.V 
3025c9de560dSAlex Tomas 	err = ext4_journal_get_write_access(handle, gdp_bh);
3026c9de560dSAlex Tomas 	if (err)
3027c9de560dSAlex Tomas 		goto out_err;
3028c9de560dSAlex Tomas 
3029c9de560dSAlex Tomas 	block = ac->ac_b_ex.fe_group * EXT4_BLOCKS_PER_GROUP(sb)
3030c9de560dSAlex Tomas 		+ ac->ac_b_ex.fe_start
3031c9de560dSAlex Tomas 		+ le32_to_cpu(es->s_first_data_block);
3032c9de560dSAlex Tomas 
3033519deca0SAneesh Kumar K.V 	len = ac->ac_b_ex.fe_len;
3034519deca0SAneesh Kumar K.V 	if (in_range(ext4_block_bitmap(sb, gdp), block, len) ||
3035519deca0SAneesh Kumar K.V 	    in_range(ext4_inode_bitmap(sb, gdp), block, len) ||
3036c9de560dSAlex Tomas 	    in_range(block, ext4_inode_table(sb, gdp),
3037519deca0SAneesh Kumar K.V 		     EXT4_SB(sb)->s_itb_per_group) ||
3038519deca0SAneesh Kumar K.V 	    in_range(block + len - 1, ext4_inode_table(sb, gdp),
3039c9de560dSAlex Tomas 		     EXT4_SB(sb)->s_itb_per_group)) {
304046e665e9SHarvey Harrison 		ext4_error(sb, __func__,
3041648f5879SAneesh Kumar K.V 			   "Allocating block %llu in system zone of %d group\n",
3042648f5879SAneesh Kumar K.V 			   block, ac->ac_b_ex.fe_group);
3043519deca0SAneesh Kumar K.V 		/* File system mounted not to panic on error
3044519deca0SAneesh Kumar K.V 		 * Fix the bitmap and repeat the block allocation
3045519deca0SAneesh Kumar K.V 		 * We leak some of the blocks here.
3046519deca0SAneesh Kumar K.V 		 */
3047519deca0SAneesh Kumar K.V 		mb_set_bits(sb_bgl_lock(sbi, ac->ac_b_ex.fe_group),
3048519deca0SAneesh Kumar K.V 				bitmap_bh->b_data, ac->ac_b_ex.fe_start,
3049519deca0SAneesh Kumar K.V 				ac->ac_b_ex.fe_len);
30500390131bSFrank Mayhar 		err = ext4_handle_dirty_metadata(handle, NULL, bitmap_bh);
3051519deca0SAneesh Kumar K.V 		if (!err)
3052519deca0SAneesh Kumar K.V 			err = -EAGAIN;
3053519deca0SAneesh Kumar K.V 		goto out_err;
3054c9de560dSAlex Tomas 	}
3055c9de560dSAlex Tomas #ifdef AGGRESSIVE_CHECK
3056c9de560dSAlex Tomas 	{
3057c9de560dSAlex Tomas 		int i;
3058c9de560dSAlex Tomas 		for (i = 0; i < ac->ac_b_ex.fe_len; i++) {
3059c9de560dSAlex Tomas 			BUG_ON(mb_test_bit(ac->ac_b_ex.fe_start + i,
3060c9de560dSAlex Tomas 						bitmap_bh->b_data));
3061c9de560dSAlex Tomas 		}
3062c9de560dSAlex Tomas 	}
3063c9de560dSAlex Tomas #endif
3064c9de560dSAlex Tomas 	spin_lock(sb_bgl_lock(sbi, ac->ac_b_ex.fe_group));
3065e8134b27SAneesh Kumar K.V 	mb_set_bits(NULL, bitmap_bh->b_data,
3066e8134b27SAneesh Kumar K.V 				ac->ac_b_ex.fe_start, ac->ac_b_ex.fe_len);
3067c9de560dSAlex Tomas 	if (gdp->bg_flags & cpu_to_le16(EXT4_BG_BLOCK_UNINIT)) {
3068c9de560dSAlex Tomas 		gdp->bg_flags &= cpu_to_le16(~EXT4_BG_BLOCK_UNINIT);
3069560671a0SAneesh Kumar K.V 		ext4_free_blks_set(sb, gdp,
3070560671a0SAneesh Kumar K.V 					ext4_free_blocks_after_init(sb,
3071560671a0SAneesh Kumar K.V 					ac->ac_b_ex.fe_group, gdp));
3072c9de560dSAlex Tomas 	}
3073560671a0SAneesh Kumar K.V 	len = ext4_free_blks_count(sb, gdp) - ac->ac_b_ex.fe_len;
3074560671a0SAneesh Kumar K.V 	ext4_free_blks_set(sb, gdp, len);
3075c9de560dSAlex Tomas 	gdp->bg_checksum = ext4_group_desc_csum(sbi, ac->ac_b_ex.fe_group, gdp);
3076c9de560dSAlex Tomas 	spin_unlock(sb_bgl_lock(sbi, ac->ac_b_ex.fe_group));
30776bc6e63fSAneesh Kumar K.V 	percpu_counter_sub(&sbi->s_freeblocks_counter, ac->ac_b_ex.fe_len);
3078d2a17637SMingming Cao 	/*
30796bc6e63fSAneesh Kumar K.V 	 * Now reduce the dirty block count also. Should not go negative
3080d2a17637SMingming Cao 	 */
30816bc6e63fSAneesh Kumar K.V 	if (!(ac->ac_flags & EXT4_MB_DELALLOC_RESERVED))
30826bc6e63fSAneesh Kumar K.V 		/* release all the reserved blocks if non delalloc */
30836bc6e63fSAneesh Kumar K.V 		percpu_counter_sub(&sbi->s_dirtyblocks_counter, reserv_blks);
30846bc6e63fSAneesh Kumar K.V 	else
30856bc6e63fSAneesh Kumar K.V 		percpu_counter_sub(&sbi->s_dirtyblocks_counter,
30866bc6e63fSAneesh Kumar K.V 						ac->ac_b_ex.fe_len);
3087c9de560dSAlex Tomas 
3088772cb7c8SJose R. Santos 	if (sbi->s_log_groups_per_flex) {
3089772cb7c8SJose R. Santos 		ext4_group_t flex_group = ext4_flex_group(sbi,
3090772cb7c8SJose R. Santos 							  ac->ac_b_ex.fe_group);
3091772cb7c8SJose R. Santos 		spin_lock(sb_bgl_lock(sbi, flex_group));
3092772cb7c8SJose R. Santos 		sbi->s_flex_groups[flex_group].free_blocks -= ac->ac_b_ex.fe_len;
3093772cb7c8SJose R. Santos 		spin_unlock(sb_bgl_lock(sbi, flex_group));
3094772cb7c8SJose R. Santos 	}
3095772cb7c8SJose R. Santos 
30960390131bSFrank Mayhar 	err = ext4_handle_dirty_metadata(handle, NULL, bitmap_bh);
3097c9de560dSAlex Tomas 	if (err)
3098c9de560dSAlex Tomas 		goto out_err;
30990390131bSFrank Mayhar 	err = ext4_handle_dirty_metadata(handle, NULL, gdp_bh);
3100c9de560dSAlex Tomas 
3101c9de560dSAlex Tomas out_err:
3102c9de560dSAlex Tomas 	sb->s_dirt = 1;
310342a10addSAneesh Kumar K.V 	brelse(bitmap_bh);
3104c9de560dSAlex Tomas 	return err;
3105c9de560dSAlex Tomas }
3106c9de560dSAlex Tomas 
3107c9de560dSAlex Tomas /*
3108c9de560dSAlex Tomas  * here we normalize request for locality group
3109c9de560dSAlex Tomas  * Group request are normalized to s_strip size if we set the same via mount
3110c9de560dSAlex Tomas  * option. If not we set it to s_mb_group_prealloc which can be configured via
3111c9de560dSAlex Tomas  * /proc/fs/ext4/<partition>/group_prealloc
3112c9de560dSAlex Tomas  *
3113c9de560dSAlex Tomas  * XXX: should we try to preallocate more than the group has now?
3114c9de560dSAlex Tomas  */
3115c9de560dSAlex Tomas static void ext4_mb_normalize_group_request(struct ext4_allocation_context *ac)
3116c9de560dSAlex Tomas {
3117c9de560dSAlex Tomas 	struct super_block *sb = ac->ac_sb;
3118c9de560dSAlex Tomas 	struct ext4_locality_group *lg = ac->ac_lg;
3119c9de560dSAlex Tomas 
3120c9de560dSAlex Tomas 	BUG_ON(lg == NULL);
3121c9de560dSAlex Tomas 	if (EXT4_SB(sb)->s_stripe)
3122c9de560dSAlex Tomas 		ac->ac_g_ex.fe_len = EXT4_SB(sb)->s_stripe;
3123c9de560dSAlex Tomas 	else
3124c9de560dSAlex Tomas 		ac->ac_g_ex.fe_len = EXT4_SB(sb)->s_mb_group_prealloc;
312560bd63d1SSolofo Ramangalahy 	mb_debug("#%u: goal %u blocks for locality group\n",
3126c9de560dSAlex Tomas 		current->pid, ac->ac_g_ex.fe_len);
3127c9de560dSAlex Tomas }
3128c9de560dSAlex Tomas 
3129c9de560dSAlex Tomas /*
3130c9de560dSAlex Tomas  * Normalization means making request better in terms of
3131c9de560dSAlex Tomas  * size and alignment
3132c9de560dSAlex Tomas  */
31334ddfef7bSEric Sandeen static noinline_for_stack void
31344ddfef7bSEric Sandeen ext4_mb_normalize_request(struct ext4_allocation_context *ac,
3135c9de560dSAlex Tomas 				struct ext4_allocation_request *ar)
3136c9de560dSAlex Tomas {
3137c9de560dSAlex Tomas 	int bsbits, max;
3138c9de560dSAlex Tomas 	ext4_lblk_t end;
3139c9de560dSAlex Tomas 	loff_t size, orig_size, start_off;
3140c9de560dSAlex Tomas 	ext4_lblk_t start, orig_start;
3141c9de560dSAlex Tomas 	struct ext4_inode_info *ei = EXT4_I(ac->ac_inode);
31429a0762c5SAneesh Kumar K.V 	struct ext4_prealloc_space *pa;
3143c9de560dSAlex Tomas 
3144c9de560dSAlex Tomas 	/* do normalize only data requests, metadata requests
3145c9de560dSAlex Tomas 	   do not need preallocation */
3146c9de560dSAlex Tomas 	if (!(ac->ac_flags & EXT4_MB_HINT_DATA))
3147c9de560dSAlex Tomas 		return;
3148c9de560dSAlex Tomas 
3149c9de560dSAlex Tomas 	/* sometime caller may want exact blocks */
3150c9de560dSAlex Tomas 	if (unlikely(ac->ac_flags & EXT4_MB_HINT_GOAL_ONLY))
3151c9de560dSAlex Tomas 		return;
3152c9de560dSAlex Tomas 
3153c9de560dSAlex Tomas 	/* caller may indicate that preallocation isn't
3154c9de560dSAlex Tomas 	 * required (it's a tail, for example) */
3155c9de560dSAlex Tomas 	if (ac->ac_flags & EXT4_MB_HINT_NOPREALLOC)
3156c9de560dSAlex Tomas 		return;
3157c9de560dSAlex Tomas 
3158c9de560dSAlex Tomas 	if (ac->ac_flags & EXT4_MB_HINT_GROUP_ALLOC) {
3159c9de560dSAlex Tomas 		ext4_mb_normalize_group_request(ac);
3160c9de560dSAlex Tomas 		return ;
3161c9de560dSAlex Tomas 	}
3162c9de560dSAlex Tomas 
3163c9de560dSAlex Tomas 	bsbits = ac->ac_sb->s_blocksize_bits;
3164c9de560dSAlex Tomas 
3165c9de560dSAlex Tomas 	/* first, let's learn actual file size
3166c9de560dSAlex Tomas 	 * given current request is allocated */
3167c9de560dSAlex Tomas 	size = ac->ac_o_ex.fe_logical + ac->ac_o_ex.fe_len;
3168c9de560dSAlex Tomas 	size = size << bsbits;
3169c9de560dSAlex Tomas 	if (size < i_size_read(ac->ac_inode))
3170c9de560dSAlex Tomas 		size = i_size_read(ac->ac_inode);
3171c9de560dSAlex Tomas 
31721930479cSValerie Clement 	/* max size of free chunks */
31731930479cSValerie Clement 	max = 2 << bsbits;
3174c9de560dSAlex Tomas 
31751930479cSValerie Clement #define NRL_CHECK_SIZE(req, size, max, chunk_size)	\
31761930479cSValerie Clement 		(req <= (size) || max <= (chunk_size))
3177c9de560dSAlex Tomas 
3178c9de560dSAlex Tomas 	/* first, try to predict filesize */
3179c9de560dSAlex Tomas 	/* XXX: should this table be tunable? */
3180c9de560dSAlex Tomas 	start_off = 0;
3181c9de560dSAlex Tomas 	if (size <= 16 * 1024) {
3182c9de560dSAlex Tomas 		size = 16 * 1024;
3183c9de560dSAlex Tomas 	} else if (size <= 32 * 1024) {
3184c9de560dSAlex Tomas 		size = 32 * 1024;
3185c9de560dSAlex Tomas 	} else if (size <= 64 * 1024) {
3186c9de560dSAlex Tomas 		size = 64 * 1024;
3187c9de560dSAlex Tomas 	} else if (size <= 128 * 1024) {
3188c9de560dSAlex Tomas 		size = 128 * 1024;
3189c9de560dSAlex Tomas 	} else if (size <= 256 * 1024) {
3190c9de560dSAlex Tomas 		size = 256 * 1024;
3191c9de560dSAlex Tomas 	} else if (size <= 512 * 1024) {
3192c9de560dSAlex Tomas 		size = 512 * 1024;
3193c9de560dSAlex Tomas 	} else if (size <= 1024 * 1024) {
3194c9de560dSAlex Tomas 		size = 1024 * 1024;
31951930479cSValerie Clement 	} else if (NRL_CHECK_SIZE(size, 4 * 1024 * 1024, max, 2 * 1024)) {
3196c9de560dSAlex Tomas 		start_off = ((loff_t)ac->ac_o_ex.fe_logical >>
31971930479cSValerie Clement 						(21 - bsbits)) << 21;
31981930479cSValerie Clement 		size = 2 * 1024 * 1024;
31991930479cSValerie Clement 	} else if (NRL_CHECK_SIZE(size, 8 * 1024 * 1024, max, 4 * 1024)) {
3200c9de560dSAlex Tomas 		start_off = ((loff_t)ac->ac_o_ex.fe_logical >>
3201c9de560dSAlex Tomas 							(22 - bsbits)) << 22;
3202c9de560dSAlex Tomas 		size = 4 * 1024 * 1024;
3203c9de560dSAlex Tomas 	} else if (NRL_CHECK_SIZE(ac->ac_o_ex.fe_len,
32041930479cSValerie Clement 					(8<<20)>>bsbits, max, 8 * 1024)) {
3205c9de560dSAlex Tomas 		start_off = ((loff_t)ac->ac_o_ex.fe_logical >>
3206c9de560dSAlex Tomas 							(23 - bsbits)) << 23;
3207c9de560dSAlex Tomas 		size = 8 * 1024 * 1024;
3208c9de560dSAlex Tomas 	} else {
3209c9de560dSAlex Tomas 		start_off = (loff_t)ac->ac_o_ex.fe_logical << bsbits;
3210c9de560dSAlex Tomas 		size	  = ac->ac_o_ex.fe_len << bsbits;
3211c9de560dSAlex Tomas 	}
3212c9de560dSAlex Tomas 	orig_size = size = size >> bsbits;
3213c9de560dSAlex Tomas 	orig_start = start = start_off >> bsbits;
3214c9de560dSAlex Tomas 
3215c9de560dSAlex Tomas 	/* don't cover already allocated blocks in selected range */
3216c9de560dSAlex Tomas 	if (ar->pleft && start <= ar->lleft) {
3217c9de560dSAlex Tomas 		size -= ar->lleft + 1 - start;
3218c9de560dSAlex Tomas 		start = ar->lleft + 1;
3219c9de560dSAlex Tomas 	}
3220c9de560dSAlex Tomas 	if (ar->pright && start + size - 1 >= ar->lright)
3221c9de560dSAlex Tomas 		size -= start + size - ar->lright;
3222c9de560dSAlex Tomas 
3223c9de560dSAlex Tomas 	end = start + size;
3224c9de560dSAlex Tomas 
3225c9de560dSAlex Tomas 	/* check we don't cross already preallocated blocks */
3226c9de560dSAlex Tomas 	rcu_read_lock();
32279a0762c5SAneesh Kumar K.V 	list_for_each_entry_rcu(pa, &ei->i_prealloc_list, pa_inode_list) {
3228498e5f24STheodore Ts'o 		ext4_lblk_t pa_end;
3229c9de560dSAlex Tomas 
3230c9de560dSAlex Tomas 		if (pa->pa_deleted)
3231c9de560dSAlex Tomas 			continue;
3232c9de560dSAlex Tomas 		spin_lock(&pa->pa_lock);
3233c9de560dSAlex Tomas 		if (pa->pa_deleted) {
3234c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
3235c9de560dSAlex Tomas 			continue;
3236c9de560dSAlex Tomas 		}
3237c9de560dSAlex Tomas 
3238c9de560dSAlex Tomas 		pa_end = pa->pa_lstart + pa->pa_len;
3239c9de560dSAlex Tomas 
3240c9de560dSAlex Tomas 		/* PA must not overlap original request */
3241c9de560dSAlex Tomas 		BUG_ON(!(ac->ac_o_ex.fe_logical >= pa_end ||
3242c9de560dSAlex Tomas 			ac->ac_o_ex.fe_logical < pa->pa_lstart));
3243c9de560dSAlex Tomas 
3244c9de560dSAlex Tomas 		/* skip PA normalized request doesn't overlap with */
3245c9de560dSAlex Tomas 		if (pa->pa_lstart >= end) {
3246c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
3247c9de560dSAlex Tomas 			continue;
3248c9de560dSAlex Tomas 		}
3249c9de560dSAlex Tomas 		if (pa_end <= start) {
3250c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
3251c9de560dSAlex Tomas 			continue;
3252c9de560dSAlex Tomas 		}
3253c9de560dSAlex Tomas 		BUG_ON(pa->pa_lstart <= start && pa_end >= end);
3254c9de560dSAlex Tomas 
3255c9de560dSAlex Tomas 		if (pa_end <= ac->ac_o_ex.fe_logical) {
3256c9de560dSAlex Tomas 			BUG_ON(pa_end < start);
3257c9de560dSAlex Tomas 			start = pa_end;
3258c9de560dSAlex Tomas 		}
3259c9de560dSAlex Tomas 
3260c9de560dSAlex Tomas 		if (pa->pa_lstart > ac->ac_o_ex.fe_logical) {
3261c9de560dSAlex Tomas 			BUG_ON(pa->pa_lstart > end);
3262c9de560dSAlex Tomas 			end = pa->pa_lstart;
3263c9de560dSAlex Tomas 		}
3264c9de560dSAlex Tomas 		spin_unlock(&pa->pa_lock);
3265c9de560dSAlex Tomas 	}
3266c9de560dSAlex Tomas 	rcu_read_unlock();
3267c9de560dSAlex Tomas 	size = end - start;
3268c9de560dSAlex Tomas 
3269c9de560dSAlex Tomas 	/* XXX: extra loop to check we really don't overlap preallocations */
3270c9de560dSAlex Tomas 	rcu_read_lock();
32719a0762c5SAneesh Kumar K.V 	list_for_each_entry_rcu(pa, &ei->i_prealloc_list, pa_inode_list) {
3272498e5f24STheodore Ts'o 		ext4_lblk_t pa_end;
3273c9de560dSAlex Tomas 		spin_lock(&pa->pa_lock);
3274c9de560dSAlex Tomas 		if (pa->pa_deleted == 0) {
3275c9de560dSAlex Tomas 			pa_end = pa->pa_lstart + pa->pa_len;
3276c9de560dSAlex Tomas 			BUG_ON(!(start >= pa_end || end <= pa->pa_lstart));
3277c9de560dSAlex Tomas 		}
3278c9de560dSAlex Tomas 		spin_unlock(&pa->pa_lock);
3279c9de560dSAlex Tomas 	}
3280c9de560dSAlex Tomas 	rcu_read_unlock();
3281c9de560dSAlex Tomas 
3282c9de560dSAlex Tomas 	if (start + size <= ac->ac_o_ex.fe_logical &&
3283c9de560dSAlex Tomas 			start > ac->ac_o_ex.fe_logical) {
3284c9de560dSAlex Tomas 		printk(KERN_ERR "start %lu, size %lu, fe_logical %lu\n",
3285c9de560dSAlex Tomas 			(unsigned long) start, (unsigned long) size,
3286c9de560dSAlex Tomas 			(unsigned long) ac->ac_o_ex.fe_logical);
3287c9de560dSAlex Tomas 	}
3288c9de560dSAlex Tomas 	BUG_ON(start + size <= ac->ac_o_ex.fe_logical &&
3289c9de560dSAlex Tomas 			start > ac->ac_o_ex.fe_logical);
3290c9de560dSAlex Tomas 	BUG_ON(size <= 0 || size >= EXT4_BLOCKS_PER_GROUP(ac->ac_sb));
3291c9de560dSAlex Tomas 
3292c9de560dSAlex Tomas 	/* now prepare goal request */
3293c9de560dSAlex Tomas 
3294c9de560dSAlex Tomas 	/* XXX: is it better to align blocks WRT to logical
3295c9de560dSAlex Tomas 	 * placement or satisfy big request as is */
3296c9de560dSAlex Tomas 	ac->ac_g_ex.fe_logical = start;
3297c9de560dSAlex Tomas 	ac->ac_g_ex.fe_len = size;
3298c9de560dSAlex Tomas 
3299c9de560dSAlex Tomas 	/* define goal start in order to merge */
3300c9de560dSAlex Tomas 	if (ar->pright && (ar->lright == (start + size))) {
3301c9de560dSAlex Tomas 		/* merge to the right */
3302c9de560dSAlex Tomas 		ext4_get_group_no_and_offset(ac->ac_sb, ar->pright - size,
3303c9de560dSAlex Tomas 						&ac->ac_f_ex.fe_group,
3304c9de560dSAlex Tomas 						&ac->ac_f_ex.fe_start);
3305c9de560dSAlex Tomas 		ac->ac_flags |= EXT4_MB_HINT_TRY_GOAL;
3306c9de560dSAlex Tomas 	}
3307c9de560dSAlex Tomas 	if (ar->pleft && (ar->lleft + 1 == start)) {
3308c9de560dSAlex Tomas 		/* merge to the left */
3309c9de560dSAlex Tomas 		ext4_get_group_no_and_offset(ac->ac_sb, ar->pleft + 1,
3310c9de560dSAlex Tomas 						&ac->ac_f_ex.fe_group,
3311c9de560dSAlex Tomas 						&ac->ac_f_ex.fe_start);
3312c9de560dSAlex Tomas 		ac->ac_flags |= EXT4_MB_HINT_TRY_GOAL;
3313c9de560dSAlex Tomas 	}
3314c9de560dSAlex Tomas 
3315c9de560dSAlex Tomas 	mb_debug("goal: %u(was %u) blocks at %u\n", (unsigned) size,
3316c9de560dSAlex Tomas 		(unsigned) orig_size, (unsigned) start);
3317c9de560dSAlex Tomas }
3318c9de560dSAlex Tomas 
3319c9de560dSAlex Tomas static void ext4_mb_collect_stats(struct ext4_allocation_context *ac)
3320c9de560dSAlex Tomas {
3321c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(ac->ac_sb);
3322c9de560dSAlex Tomas 
3323c9de560dSAlex Tomas 	if (sbi->s_mb_stats && ac->ac_g_ex.fe_len > 1) {
3324c9de560dSAlex Tomas 		atomic_inc(&sbi->s_bal_reqs);
3325c9de560dSAlex Tomas 		atomic_add(ac->ac_b_ex.fe_len, &sbi->s_bal_allocated);
3326c9de560dSAlex Tomas 		if (ac->ac_o_ex.fe_len >= ac->ac_g_ex.fe_len)
3327c9de560dSAlex Tomas 			atomic_inc(&sbi->s_bal_success);
3328c9de560dSAlex Tomas 		atomic_add(ac->ac_found, &sbi->s_bal_ex_scanned);
3329c9de560dSAlex Tomas 		if (ac->ac_g_ex.fe_start == ac->ac_b_ex.fe_start &&
3330c9de560dSAlex Tomas 				ac->ac_g_ex.fe_group == ac->ac_b_ex.fe_group)
3331c9de560dSAlex Tomas 			atomic_inc(&sbi->s_bal_goals);
3332c9de560dSAlex Tomas 		if (ac->ac_found > sbi->s_mb_max_to_scan)
3333c9de560dSAlex Tomas 			atomic_inc(&sbi->s_bal_breaks);
3334c9de560dSAlex Tomas 	}
3335c9de560dSAlex Tomas 
3336c9de560dSAlex Tomas 	ext4_mb_store_history(ac);
3337c9de560dSAlex Tomas }
3338c9de560dSAlex Tomas 
3339c9de560dSAlex Tomas /*
3340c9de560dSAlex Tomas  * use blocks preallocated to inode
3341c9de560dSAlex Tomas  */
3342c9de560dSAlex Tomas static void ext4_mb_use_inode_pa(struct ext4_allocation_context *ac,
3343c9de560dSAlex Tomas 				struct ext4_prealloc_space *pa)
3344c9de560dSAlex Tomas {
3345c9de560dSAlex Tomas 	ext4_fsblk_t start;
3346c9de560dSAlex Tomas 	ext4_fsblk_t end;
3347c9de560dSAlex Tomas 	int len;
3348c9de560dSAlex Tomas 
3349c9de560dSAlex Tomas 	/* found preallocated blocks, use them */
3350c9de560dSAlex Tomas 	start = pa->pa_pstart + (ac->ac_o_ex.fe_logical - pa->pa_lstart);
3351c9de560dSAlex Tomas 	end = min(pa->pa_pstart + pa->pa_len, start + ac->ac_o_ex.fe_len);
3352c9de560dSAlex Tomas 	len = end - start;
3353c9de560dSAlex Tomas 	ext4_get_group_no_and_offset(ac->ac_sb, start, &ac->ac_b_ex.fe_group,
3354c9de560dSAlex Tomas 					&ac->ac_b_ex.fe_start);
3355c9de560dSAlex Tomas 	ac->ac_b_ex.fe_len = len;
3356c9de560dSAlex Tomas 	ac->ac_status = AC_STATUS_FOUND;
3357c9de560dSAlex Tomas 	ac->ac_pa = pa;
3358c9de560dSAlex Tomas 
3359c9de560dSAlex Tomas 	BUG_ON(start < pa->pa_pstart);
3360c9de560dSAlex Tomas 	BUG_ON(start + len > pa->pa_pstart + pa->pa_len);
3361c9de560dSAlex Tomas 	BUG_ON(pa->pa_free < len);
3362c9de560dSAlex Tomas 	pa->pa_free -= len;
3363c9de560dSAlex Tomas 
336460bd63d1SSolofo Ramangalahy 	mb_debug("use %llu/%u from inode pa %p\n", start, len, pa);
3365c9de560dSAlex Tomas }
3366c9de560dSAlex Tomas 
3367c9de560dSAlex Tomas /*
3368c9de560dSAlex Tomas  * use blocks preallocated to locality group
3369c9de560dSAlex Tomas  */
3370c9de560dSAlex Tomas static void ext4_mb_use_group_pa(struct ext4_allocation_context *ac,
3371c9de560dSAlex Tomas 				struct ext4_prealloc_space *pa)
3372c9de560dSAlex Tomas {
337303cddb80SAneesh Kumar K.V 	unsigned int len = ac->ac_o_ex.fe_len;
33746be2ded1SAneesh Kumar K.V 
3375c9de560dSAlex Tomas 	ext4_get_group_no_and_offset(ac->ac_sb, pa->pa_pstart,
3376c9de560dSAlex Tomas 					&ac->ac_b_ex.fe_group,
3377c9de560dSAlex Tomas 					&ac->ac_b_ex.fe_start);
3378c9de560dSAlex Tomas 	ac->ac_b_ex.fe_len = len;
3379c9de560dSAlex Tomas 	ac->ac_status = AC_STATUS_FOUND;
3380c9de560dSAlex Tomas 	ac->ac_pa = pa;
3381c9de560dSAlex Tomas 
3382c9de560dSAlex Tomas 	/* we don't correct pa_pstart or pa_plen here to avoid
338326346ff6SAneesh Kumar K.V 	 * possible race when the group is being loaded concurrently
3384c9de560dSAlex Tomas 	 * instead we correct pa later, after blocks are marked
338526346ff6SAneesh Kumar K.V 	 * in on-disk bitmap -- see ext4_mb_release_context()
338626346ff6SAneesh Kumar K.V 	 * Other CPUs are prevented from allocating from this pa by lg_mutex
3387c9de560dSAlex Tomas 	 */
3388c9de560dSAlex Tomas 	mb_debug("use %u/%u from group pa %p\n", pa->pa_lstart-len, len, pa);
3389c9de560dSAlex Tomas }
3390c9de560dSAlex Tomas 
3391c9de560dSAlex Tomas /*
33925e745b04SAneesh Kumar K.V  * Return the prealloc space that have minimal distance
33935e745b04SAneesh Kumar K.V  * from the goal block. @cpa is the prealloc
33945e745b04SAneesh Kumar K.V  * space that is having currently known minimal distance
33955e745b04SAneesh Kumar K.V  * from the goal block.
33965e745b04SAneesh Kumar K.V  */
33975e745b04SAneesh Kumar K.V static struct ext4_prealloc_space *
33985e745b04SAneesh Kumar K.V ext4_mb_check_group_pa(ext4_fsblk_t goal_block,
33995e745b04SAneesh Kumar K.V 			struct ext4_prealloc_space *pa,
34005e745b04SAneesh Kumar K.V 			struct ext4_prealloc_space *cpa)
34015e745b04SAneesh Kumar K.V {
34025e745b04SAneesh Kumar K.V 	ext4_fsblk_t cur_distance, new_distance;
34035e745b04SAneesh Kumar K.V 
34045e745b04SAneesh Kumar K.V 	if (cpa == NULL) {
34055e745b04SAneesh Kumar K.V 		atomic_inc(&pa->pa_count);
34065e745b04SAneesh Kumar K.V 		return pa;
34075e745b04SAneesh Kumar K.V 	}
34085e745b04SAneesh Kumar K.V 	cur_distance = abs(goal_block - cpa->pa_pstart);
34095e745b04SAneesh Kumar K.V 	new_distance = abs(goal_block - pa->pa_pstart);
34105e745b04SAneesh Kumar K.V 
34115e745b04SAneesh Kumar K.V 	if (cur_distance < new_distance)
34125e745b04SAneesh Kumar K.V 		return cpa;
34135e745b04SAneesh Kumar K.V 
34145e745b04SAneesh Kumar K.V 	/* drop the previous reference */
34155e745b04SAneesh Kumar K.V 	atomic_dec(&cpa->pa_count);
34165e745b04SAneesh Kumar K.V 	atomic_inc(&pa->pa_count);
34175e745b04SAneesh Kumar K.V 	return pa;
34185e745b04SAneesh Kumar K.V }
34195e745b04SAneesh Kumar K.V 
34205e745b04SAneesh Kumar K.V /*
3421c9de560dSAlex Tomas  * search goal blocks in preallocated space
3422c9de560dSAlex Tomas  */
34234ddfef7bSEric Sandeen static noinline_for_stack int
34244ddfef7bSEric Sandeen ext4_mb_use_preallocated(struct ext4_allocation_context *ac)
3425c9de560dSAlex Tomas {
34266be2ded1SAneesh Kumar K.V 	int order, i;
3427c9de560dSAlex Tomas 	struct ext4_inode_info *ei = EXT4_I(ac->ac_inode);
3428c9de560dSAlex Tomas 	struct ext4_locality_group *lg;
34295e745b04SAneesh Kumar K.V 	struct ext4_prealloc_space *pa, *cpa = NULL;
34305e745b04SAneesh Kumar K.V 	ext4_fsblk_t goal_block;
3431c9de560dSAlex Tomas 
3432c9de560dSAlex Tomas 	/* only data can be preallocated */
3433c9de560dSAlex Tomas 	if (!(ac->ac_flags & EXT4_MB_HINT_DATA))
3434c9de560dSAlex Tomas 		return 0;
3435c9de560dSAlex Tomas 
3436c9de560dSAlex Tomas 	/* first, try per-file preallocation */
3437c9de560dSAlex Tomas 	rcu_read_lock();
34389a0762c5SAneesh Kumar K.V 	list_for_each_entry_rcu(pa, &ei->i_prealloc_list, pa_inode_list) {
3439c9de560dSAlex Tomas 
3440c9de560dSAlex Tomas 		/* all fields in this condition don't change,
3441c9de560dSAlex Tomas 		 * so we can skip locking for them */
3442c9de560dSAlex Tomas 		if (ac->ac_o_ex.fe_logical < pa->pa_lstart ||
3443c9de560dSAlex Tomas 			ac->ac_o_ex.fe_logical >= pa->pa_lstart + pa->pa_len)
3444c9de560dSAlex Tomas 			continue;
3445c9de560dSAlex Tomas 
3446c9de560dSAlex Tomas 		/* found preallocated blocks, use them */
3447c9de560dSAlex Tomas 		spin_lock(&pa->pa_lock);
3448c9de560dSAlex Tomas 		if (pa->pa_deleted == 0 && pa->pa_free) {
3449c9de560dSAlex Tomas 			atomic_inc(&pa->pa_count);
3450c9de560dSAlex Tomas 			ext4_mb_use_inode_pa(ac, pa);
3451c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
3452c9de560dSAlex Tomas 			ac->ac_criteria = 10;
3453c9de560dSAlex Tomas 			rcu_read_unlock();
3454c9de560dSAlex Tomas 			return 1;
3455c9de560dSAlex Tomas 		}
3456c9de560dSAlex Tomas 		spin_unlock(&pa->pa_lock);
3457c9de560dSAlex Tomas 	}
3458c9de560dSAlex Tomas 	rcu_read_unlock();
3459c9de560dSAlex Tomas 
3460c9de560dSAlex Tomas 	/* can we use group allocation? */
3461c9de560dSAlex Tomas 	if (!(ac->ac_flags & EXT4_MB_HINT_GROUP_ALLOC))
3462c9de560dSAlex Tomas 		return 0;
3463c9de560dSAlex Tomas 
3464c9de560dSAlex Tomas 	/* inode may have no locality group for some reason */
3465c9de560dSAlex Tomas 	lg = ac->ac_lg;
3466c9de560dSAlex Tomas 	if (lg == NULL)
3467c9de560dSAlex Tomas 		return 0;
34686be2ded1SAneesh Kumar K.V 	order  = fls(ac->ac_o_ex.fe_len) - 1;
34696be2ded1SAneesh Kumar K.V 	if (order > PREALLOC_TB_SIZE - 1)
34706be2ded1SAneesh Kumar K.V 		/* The max size of hash table is PREALLOC_TB_SIZE */
34716be2ded1SAneesh Kumar K.V 		order = PREALLOC_TB_SIZE - 1;
3472c9de560dSAlex Tomas 
34735e745b04SAneesh Kumar K.V 	goal_block = ac->ac_g_ex.fe_group * EXT4_BLOCKS_PER_GROUP(ac->ac_sb) +
34745e745b04SAneesh Kumar K.V 		     ac->ac_g_ex.fe_start +
34755e745b04SAneesh Kumar K.V 		     le32_to_cpu(EXT4_SB(ac->ac_sb)->s_es->s_first_data_block);
34765e745b04SAneesh Kumar K.V 	/*
34775e745b04SAneesh Kumar K.V 	 * search for the prealloc space that is having
34785e745b04SAneesh Kumar K.V 	 * minimal distance from the goal block.
34795e745b04SAneesh Kumar K.V 	 */
34806be2ded1SAneesh Kumar K.V 	for (i = order; i < PREALLOC_TB_SIZE; i++) {
3481c9de560dSAlex Tomas 		rcu_read_lock();
34826be2ded1SAneesh Kumar K.V 		list_for_each_entry_rcu(pa, &lg->lg_prealloc_list[i],
34836be2ded1SAneesh Kumar K.V 					pa_inode_list) {
3484c9de560dSAlex Tomas 			spin_lock(&pa->pa_lock);
34856be2ded1SAneesh Kumar K.V 			if (pa->pa_deleted == 0 &&
34866be2ded1SAneesh Kumar K.V 					pa->pa_free >= ac->ac_o_ex.fe_len) {
34875e745b04SAneesh Kumar K.V 
34885e745b04SAneesh Kumar K.V 				cpa = ext4_mb_check_group_pa(goal_block,
34895e745b04SAneesh Kumar K.V 								pa, cpa);
34905e745b04SAneesh Kumar K.V 			}
3491c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
34925e745b04SAneesh Kumar K.V 		}
34935e745b04SAneesh Kumar K.V 		rcu_read_unlock();
34945e745b04SAneesh Kumar K.V 	}
34955e745b04SAneesh Kumar K.V 	if (cpa) {
34965e745b04SAneesh Kumar K.V 		ext4_mb_use_group_pa(ac, cpa);
3497c9de560dSAlex Tomas 		ac->ac_criteria = 20;
3498c9de560dSAlex Tomas 		return 1;
3499c9de560dSAlex Tomas 	}
3500c9de560dSAlex Tomas 	return 0;
3501c9de560dSAlex Tomas }
3502c9de560dSAlex Tomas 
3503c9de560dSAlex Tomas /*
35047a2fcbf7SAneesh Kumar K.V  * the function goes through all block freed in the group
35057a2fcbf7SAneesh Kumar K.V  * but not yet committed and marks them used in in-core bitmap.
35067a2fcbf7SAneesh Kumar K.V  * buddy must be generated from this bitmap
35077a2fcbf7SAneesh Kumar K.V  * Need to be called with ext4 group lock (ext4_lock_group)
35087a2fcbf7SAneesh Kumar K.V  */
35097a2fcbf7SAneesh Kumar K.V static void ext4_mb_generate_from_freelist(struct super_block *sb, void *bitmap,
35107a2fcbf7SAneesh Kumar K.V 						ext4_group_t group)
35117a2fcbf7SAneesh Kumar K.V {
35127a2fcbf7SAneesh Kumar K.V 	struct rb_node *n;
35137a2fcbf7SAneesh Kumar K.V 	struct ext4_group_info *grp;
35147a2fcbf7SAneesh Kumar K.V 	struct ext4_free_data *entry;
35157a2fcbf7SAneesh Kumar K.V 
35167a2fcbf7SAneesh Kumar K.V 	grp = ext4_get_group_info(sb, group);
35177a2fcbf7SAneesh Kumar K.V 	n = rb_first(&(grp->bb_free_root));
35187a2fcbf7SAneesh Kumar K.V 
35197a2fcbf7SAneesh Kumar K.V 	while (n) {
35207a2fcbf7SAneesh Kumar K.V 		entry = rb_entry(n, struct ext4_free_data, node);
35217a2fcbf7SAneesh Kumar K.V 		mb_set_bits(sb_bgl_lock(EXT4_SB(sb), group),
35227a2fcbf7SAneesh Kumar K.V 				bitmap, entry->start_blk,
35237a2fcbf7SAneesh Kumar K.V 				entry->count);
35247a2fcbf7SAneesh Kumar K.V 		n = rb_next(n);
35257a2fcbf7SAneesh Kumar K.V 	}
35267a2fcbf7SAneesh Kumar K.V 	return;
35277a2fcbf7SAneesh Kumar K.V }
35287a2fcbf7SAneesh Kumar K.V 
35297a2fcbf7SAneesh Kumar K.V /*
3530c9de560dSAlex Tomas  * the function goes through all preallocation in this group and marks them
3531c9de560dSAlex Tomas  * used in in-core bitmap. buddy must be generated from this bitmap
3532c9de560dSAlex Tomas  * Need to be called with ext4 group lock (ext4_lock_group)
3533c9de560dSAlex Tomas  */
3534c9de560dSAlex Tomas static void ext4_mb_generate_from_pa(struct super_block *sb, void *bitmap,
3535c9de560dSAlex Tomas 					ext4_group_t group)
3536c9de560dSAlex Tomas {
3537c9de560dSAlex Tomas 	struct ext4_group_info *grp = ext4_get_group_info(sb, group);
3538c9de560dSAlex Tomas 	struct ext4_prealloc_space *pa;
3539c9de560dSAlex Tomas 	struct list_head *cur;
3540c9de560dSAlex Tomas 	ext4_group_t groupnr;
3541c9de560dSAlex Tomas 	ext4_grpblk_t start;
3542c9de560dSAlex Tomas 	int preallocated = 0;
3543c9de560dSAlex Tomas 	int count = 0;
3544c9de560dSAlex Tomas 	int len;
3545c9de560dSAlex Tomas 
3546c9de560dSAlex Tomas 	/* all form of preallocation discards first load group,
3547c9de560dSAlex Tomas 	 * so the only competing code is preallocation use.
3548c9de560dSAlex Tomas 	 * we don't need any locking here
3549c9de560dSAlex Tomas 	 * notice we do NOT ignore preallocations with pa_deleted
3550c9de560dSAlex Tomas 	 * otherwise we could leave used blocks available for
3551c9de560dSAlex Tomas 	 * allocation in buddy when concurrent ext4_mb_put_pa()
3552c9de560dSAlex Tomas 	 * is dropping preallocation
3553c9de560dSAlex Tomas 	 */
3554c9de560dSAlex Tomas 	list_for_each(cur, &grp->bb_prealloc_list) {
3555c9de560dSAlex Tomas 		pa = list_entry(cur, struct ext4_prealloc_space, pa_group_list);
3556c9de560dSAlex Tomas 		spin_lock(&pa->pa_lock);
3557c9de560dSAlex Tomas 		ext4_get_group_no_and_offset(sb, pa->pa_pstart,
3558c9de560dSAlex Tomas 					     &groupnr, &start);
3559c9de560dSAlex Tomas 		len = pa->pa_len;
3560c9de560dSAlex Tomas 		spin_unlock(&pa->pa_lock);
3561c9de560dSAlex Tomas 		if (unlikely(len == 0))
3562c9de560dSAlex Tomas 			continue;
3563c9de560dSAlex Tomas 		BUG_ON(groupnr != group);
3564c9de560dSAlex Tomas 		mb_set_bits(sb_bgl_lock(EXT4_SB(sb), group),
3565c9de560dSAlex Tomas 						bitmap, start, len);
3566c9de560dSAlex Tomas 		preallocated += len;
3567c9de560dSAlex Tomas 		count++;
3568c9de560dSAlex Tomas 	}
3569a9df9a49STheodore Ts'o 	mb_debug("prellocated %u for group %u\n", preallocated, group);
3570c9de560dSAlex Tomas }
3571c9de560dSAlex Tomas 
3572c9de560dSAlex Tomas static void ext4_mb_pa_callback(struct rcu_head *head)
3573c9de560dSAlex Tomas {
3574c9de560dSAlex Tomas 	struct ext4_prealloc_space *pa;
3575c9de560dSAlex Tomas 	pa = container_of(head, struct ext4_prealloc_space, u.pa_rcu);
3576c9de560dSAlex Tomas 	kmem_cache_free(ext4_pspace_cachep, pa);
3577c9de560dSAlex Tomas }
3578c9de560dSAlex Tomas 
3579c9de560dSAlex Tomas /*
3580c9de560dSAlex Tomas  * drops a reference to preallocated space descriptor
3581c9de560dSAlex Tomas  * if this was the last reference and the space is consumed
3582c9de560dSAlex Tomas  */
3583c9de560dSAlex Tomas static void ext4_mb_put_pa(struct ext4_allocation_context *ac,
3584c9de560dSAlex Tomas 			struct super_block *sb, struct ext4_prealloc_space *pa)
3585c9de560dSAlex Tomas {
3586a9df9a49STheodore Ts'o 	ext4_group_t grp;
3587c9de560dSAlex Tomas 
3588c9de560dSAlex Tomas 	if (!atomic_dec_and_test(&pa->pa_count) || pa->pa_free != 0)
3589c9de560dSAlex Tomas 		return;
3590c9de560dSAlex Tomas 
3591c9de560dSAlex Tomas 	/* in this short window concurrent discard can set pa_deleted */
3592c9de560dSAlex Tomas 	spin_lock(&pa->pa_lock);
3593c9de560dSAlex Tomas 	if (pa->pa_deleted == 1) {
3594c9de560dSAlex Tomas 		spin_unlock(&pa->pa_lock);
3595c9de560dSAlex Tomas 		return;
3596c9de560dSAlex Tomas 	}
3597c9de560dSAlex Tomas 
3598c9de560dSAlex Tomas 	pa->pa_deleted = 1;
3599c9de560dSAlex Tomas 	spin_unlock(&pa->pa_lock);
3600c9de560dSAlex Tomas 
3601c9de560dSAlex Tomas 	/* -1 is to protect from crossing allocation group */
3602c9de560dSAlex Tomas 	ext4_get_group_no_and_offset(sb, pa->pa_pstart - 1, &grp, NULL);
3603c9de560dSAlex Tomas 
3604c9de560dSAlex Tomas 	/*
3605c9de560dSAlex Tomas 	 * possible race:
3606c9de560dSAlex Tomas 	 *
3607c9de560dSAlex Tomas 	 *  P1 (buddy init)			P2 (regular allocation)
3608c9de560dSAlex Tomas 	 *					find block B in PA
3609c9de560dSAlex Tomas 	 *  copy on-disk bitmap to buddy
3610c9de560dSAlex Tomas 	 *  					mark B in on-disk bitmap
3611c9de560dSAlex Tomas 	 *					drop PA from group
3612c9de560dSAlex Tomas 	 *  mark all PAs in buddy
3613c9de560dSAlex Tomas 	 *
3614c9de560dSAlex Tomas 	 * thus, P1 initializes buddy with B available. to prevent this
3615c9de560dSAlex Tomas 	 * we make "copy" and "mark all PAs" atomic and serialize "drop PA"
3616c9de560dSAlex Tomas 	 * against that pair
3617c9de560dSAlex Tomas 	 */
3618c9de560dSAlex Tomas 	ext4_lock_group(sb, grp);
3619c9de560dSAlex Tomas 	list_del(&pa->pa_group_list);
3620c9de560dSAlex Tomas 	ext4_unlock_group(sb, grp);
3621c9de560dSAlex Tomas 
3622c9de560dSAlex Tomas 	spin_lock(pa->pa_obj_lock);
3623c9de560dSAlex Tomas 	list_del_rcu(&pa->pa_inode_list);
3624c9de560dSAlex Tomas 	spin_unlock(pa->pa_obj_lock);
3625c9de560dSAlex Tomas 
3626c9de560dSAlex Tomas 	call_rcu(&(pa)->u.pa_rcu, ext4_mb_pa_callback);
3627c9de560dSAlex Tomas }
3628c9de560dSAlex Tomas 
3629c9de560dSAlex Tomas /*
3630c9de560dSAlex Tomas  * creates new preallocated space for given inode
3631c9de560dSAlex Tomas  */
36324ddfef7bSEric Sandeen static noinline_for_stack int
36334ddfef7bSEric Sandeen ext4_mb_new_inode_pa(struct ext4_allocation_context *ac)
3634c9de560dSAlex Tomas {
3635c9de560dSAlex Tomas 	struct super_block *sb = ac->ac_sb;
3636c9de560dSAlex Tomas 	struct ext4_prealloc_space *pa;
3637c9de560dSAlex Tomas 	struct ext4_group_info *grp;
3638c9de560dSAlex Tomas 	struct ext4_inode_info *ei;
3639c9de560dSAlex Tomas 
3640c9de560dSAlex Tomas 	/* preallocate only when found space is larger then requested */
3641c9de560dSAlex Tomas 	BUG_ON(ac->ac_o_ex.fe_len >= ac->ac_b_ex.fe_len);
3642c9de560dSAlex Tomas 	BUG_ON(ac->ac_status != AC_STATUS_FOUND);
3643c9de560dSAlex Tomas 	BUG_ON(!S_ISREG(ac->ac_inode->i_mode));
3644c9de560dSAlex Tomas 
3645c9de560dSAlex Tomas 	pa = kmem_cache_alloc(ext4_pspace_cachep, GFP_NOFS);
3646c9de560dSAlex Tomas 	if (pa == NULL)
3647c9de560dSAlex Tomas 		return -ENOMEM;
3648c9de560dSAlex Tomas 
3649c9de560dSAlex Tomas 	if (ac->ac_b_ex.fe_len < ac->ac_g_ex.fe_len) {
3650c9de560dSAlex Tomas 		int winl;
3651c9de560dSAlex Tomas 		int wins;
3652c9de560dSAlex Tomas 		int win;
3653c9de560dSAlex Tomas 		int offs;
3654c9de560dSAlex Tomas 
3655c9de560dSAlex Tomas 		/* we can't allocate as much as normalizer wants.
3656c9de560dSAlex Tomas 		 * so, found space must get proper lstart
3657c9de560dSAlex Tomas 		 * to cover original request */
3658c9de560dSAlex Tomas 		BUG_ON(ac->ac_g_ex.fe_logical > ac->ac_o_ex.fe_logical);
3659c9de560dSAlex Tomas 		BUG_ON(ac->ac_g_ex.fe_len < ac->ac_o_ex.fe_len);
3660c9de560dSAlex Tomas 
3661c9de560dSAlex Tomas 		/* we're limited by original request in that
3662c9de560dSAlex Tomas 		 * logical block must be covered any way
3663c9de560dSAlex Tomas 		 * winl is window we can move our chunk within */
3664c9de560dSAlex Tomas 		winl = ac->ac_o_ex.fe_logical - ac->ac_g_ex.fe_logical;
3665c9de560dSAlex Tomas 
3666c9de560dSAlex Tomas 		/* also, we should cover whole original request */
3667c9de560dSAlex Tomas 		wins = ac->ac_b_ex.fe_len - ac->ac_o_ex.fe_len;
3668c9de560dSAlex Tomas 
3669c9de560dSAlex Tomas 		/* the smallest one defines real window */
3670c9de560dSAlex Tomas 		win = min(winl, wins);
3671c9de560dSAlex Tomas 
3672c9de560dSAlex Tomas 		offs = ac->ac_o_ex.fe_logical % ac->ac_b_ex.fe_len;
3673c9de560dSAlex Tomas 		if (offs && offs < win)
3674c9de560dSAlex Tomas 			win = offs;
3675c9de560dSAlex Tomas 
3676c9de560dSAlex Tomas 		ac->ac_b_ex.fe_logical = ac->ac_o_ex.fe_logical - win;
3677c9de560dSAlex Tomas 		BUG_ON(ac->ac_o_ex.fe_logical < ac->ac_b_ex.fe_logical);
3678c9de560dSAlex Tomas 		BUG_ON(ac->ac_o_ex.fe_len > ac->ac_b_ex.fe_len);
3679c9de560dSAlex Tomas 	}
3680c9de560dSAlex Tomas 
3681c9de560dSAlex Tomas 	/* preallocation can change ac_b_ex, thus we store actually
3682c9de560dSAlex Tomas 	 * allocated blocks for history */
3683c9de560dSAlex Tomas 	ac->ac_f_ex = ac->ac_b_ex;
3684c9de560dSAlex Tomas 
3685c9de560dSAlex Tomas 	pa->pa_lstart = ac->ac_b_ex.fe_logical;
3686c9de560dSAlex Tomas 	pa->pa_pstart = ext4_grp_offs_to_block(sb, &ac->ac_b_ex);
3687c9de560dSAlex Tomas 	pa->pa_len = ac->ac_b_ex.fe_len;
3688c9de560dSAlex Tomas 	pa->pa_free = pa->pa_len;
3689c9de560dSAlex Tomas 	atomic_set(&pa->pa_count, 1);
3690c9de560dSAlex Tomas 	spin_lock_init(&pa->pa_lock);
3691c9de560dSAlex Tomas 	pa->pa_deleted = 0;
3692c9de560dSAlex Tomas 	pa->pa_linear = 0;
3693c9de560dSAlex Tomas 
3694c9de560dSAlex Tomas 	mb_debug("new inode pa %p: %llu/%u for %u\n", pa,
3695c9de560dSAlex Tomas 			pa->pa_pstart, pa->pa_len, pa->pa_lstart);
3696c9de560dSAlex Tomas 
3697c9de560dSAlex Tomas 	ext4_mb_use_inode_pa(ac, pa);
3698c9de560dSAlex Tomas 	atomic_add(pa->pa_free, &EXT4_SB(sb)->s_mb_preallocated);
3699c9de560dSAlex Tomas 
3700c9de560dSAlex Tomas 	ei = EXT4_I(ac->ac_inode);
3701c9de560dSAlex Tomas 	grp = ext4_get_group_info(sb, ac->ac_b_ex.fe_group);
3702c9de560dSAlex Tomas 
3703c9de560dSAlex Tomas 	pa->pa_obj_lock = &ei->i_prealloc_lock;
3704c9de560dSAlex Tomas 	pa->pa_inode = ac->ac_inode;
3705c9de560dSAlex Tomas 
3706c9de560dSAlex Tomas 	ext4_lock_group(sb, ac->ac_b_ex.fe_group);
3707c9de560dSAlex Tomas 	list_add(&pa->pa_group_list, &grp->bb_prealloc_list);
3708c9de560dSAlex Tomas 	ext4_unlock_group(sb, ac->ac_b_ex.fe_group);
3709c9de560dSAlex Tomas 
3710c9de560dSAlex Tomas 	spin_lock(pa->pa_obj_lock);
3711c9de560dSAlex Tomas 	list_add_rcu(&pa->pa_inode_list, &ei->i_prealloc_list);
3712c9de560dSAlex Tomas 	spin_unlock(pa->pa_obj_lock);
3713c9de560dSAlex Tomas 
3714c9de560dSAlex Tomas 	return 0;
3715c9de560dSAlex Tomas }
3716c9de560dSAlex Tomas 
3717c9de560dSAlex Tomas /*
3718c9de560dSAlex Tomas  * creates new preallocated space for locality group inodes belongs to
3719c9de560dSAlex Tomas  */
37204ddfef7bSEric Sandeen static noinline_for_stack int
37214ddfef7bSEric Sandeen ext4_mb_new_group_pa(struct ext4_allocation_context *ac)
3722c9de560dSAlex Tomas {
3723c9de560dSAlex Tomas 	struct super_block *sb = ac->ac_sb;
3724c9de560dSAlex Tomas 	struct ext4_locality_group *lg;
3725c9de560dSAlex Tomas 	struct ext4_prealloc_space *pa;
3726c9de560dSAlex Tomas 	struct ext4_group_info *grp;
3727c9de560dSAlex Tomas 
3728c9de560dSAlex Tomas 	/* preallocate only when found space is larger then requested */
3729c9de560dSAlex Tomas 	BUG_ON(ac->ac_o_ex.fe_len >= ac->ac_b_ex.fe_len);
3730c9de560dSAlex Tomas 	BUG_ON(ac->ac_status != AC_STATUS_FOUND);
3731c9de560dSAlex Tomas 	BUG_ON(!S_ISREG(ac->ac_inode->i_mode));
3732c9de560dSAlex Tomas 
3733c9de560dSAlex Tomas 	BUG_ON(ext4_pspace_cachep == NULL);
3734c9de560dSAlex Tomas 	pa = kmem_cache_alloc(ext4_pspace_cachep, GFP_NOFS);
3735c9de560dSAlex Tomas 	if (pa == NULL)
3736c9de560dSAlex Tomas 		return -ENOMEM;
3737c9de560dSAlex Tomas 
3738c9de560dSAlex Tomas 	/* preallocation can change ac_b_ex, thus we store actually
3739c9de560dSAlex Tomas 	 * allocated blocks for history */
3740c9de560dSAlex Tomas 	ac->ac_f_ex = ac->ac_b_ex;
3741c9de560dSAlex Tomas 
3742c9de560dSAlex Tomas 	pa->pa_pstart = ext4_grp_offs_to_block(sb, &ac->ac_b_ex);
3743c9de560dSAlex Tomas 	pa->pa_lstart = pa->pa_pstart;
3744c9de560dSAlex Tomas 	pa->pa_len = ac->ac_b_ex.fe_len;
3745c9de560dSAlex Tomas 	pa->pa_free = pa->pa_len;
3746c9de560dSAlex Tomas 	atomic_set(&pa->pa_count, 1);
3747c9de560dSAlex Tomas 	spin_lock_init(&pa->pa_lock);
37486be2ded1SAneesh Kumar K.V 	INIT_LIST_HEAD(&pa->pa_inode_list);
3749c9de560dSAlex Tomas 	pa->pa_deleted = 0;
3750c9de560dSAlex Tomas 	pa->pa_linear = 1;
3751c9de560dSAlex Tomas 
3752c9de560dSAlex Tomas 	mb_debug("new group pa %p: %llu/%u for %u\n", pa,
3753c9de560dSAlex Tomas 			pa->pa_pstart, pa->pa_len, pa->pa_lstart);
3754c9de560dSAlex Tomas 
3755c9de560dSAlex Tomas 	ext4_mb_use_group_pa(ac, pa);
3756c9de560dSAlex Tomas 	atomic_add(pa->pa_free, &EXT4_SB(sb)->s_mb_preallocated);
3757c9de560dSAlex Tomas 
3758c9de560dSAlex Tomas 	grp = ext4_get_group_info(sb, ac->ac_b_ex.fe_group);
3759c9de560dSAlex Tomas 	lg = ac->ac_lg;
3760c9de560dSAlex Tomas 	BUG_ON(lg == NULL);
3761c9de560dSAlex Tomas 
3762c9de560dSAlex Tomas 	pa->pa_obj_lock = &lg->lg_prealloc_lock;
3763c9de560dSAlex Tomas 	pa->pa_inode = NULL;
3764c9de560dSAlex Tomas 
3765c9de560dSAlex Tomas 	ext4_lock_group(sb, ac->ac_b_ex.fe_group);
3766c9de560dSAlex Tomas 	list_add(&pa->pa_group_list, &grp->bb_prealloc_list);
3767c9de560dSAlex Tomas 	ext4_unlock_group(sb, ac->ac_b_ex.fe_group);
3768c9de560dSAlex Tomas 
37696be2ded1SAneesh Kumar K.V 	/*
37706be2ded1SAneesh Kumar K.V 	 * We will later add the new pa to the right bucket
37716be2ded1SAneesh Kumar K.V 	 * after updating the pa_free in ext4_mb_release_context
37726be2ded1SAneesh Kumar K.V 	 */
3773c9de560dSAlex Tomas 	return 0;
3774c9de560dSAlex Tomas }
3775c9de560dSAlex Tomas 
3776c9de560dSAlex Tomas static int ext4_mb_new_preallocation(struct ext4_allocation_context *ac)
3777c9de560dSAlex Tomas {
3778c9de560dSAlex Tomas 	int err;
3779c9de560dSAlex Tomas 
3780c9de560dSAlex Tomas 	if (ac->ac_flags & EXT4_MB_HINT_GROUP_ALLOC)
3781c9de560dSAlex Tomas 		err = ext4_mb_new_group_pa(ac);
3782c9de560dSAlex Tomas 	else
3783c9de560dSAlex Tomas 		err = ext4_mb_new_inode_pa(ac);
3784c9de560dSAlex Tomas 	return err;
3785c9de560dSAlex Tomas }
3786c9de560dSAlex Tomas 
3787c9de560dSAlex Tomas /*
3788c9de560dSAlex Tomas  * finds all unused blocks in on-disk bitmap, frees them in
3789c9de560dSAlex Tomas  * in-core bitmap and buddy.
3790c9de560dSAlex Tomas  * @pa must be unlinked from inode and group lists, so that
3791c9de560dSAlex Tomas  * nobody else can find/use it.
3792c9de560dSAlex Tomas  * the caller MUST hold group/inode locks.
3793c9de560dSAlex Tomas  * TODO: optimize the case when there are no in-core structures yet
3794c9de560dSAlex Tomas  */
37954ddfef7bSEric Sandeen static noinline_for_stack int
37964ddfef7bSEric Sandeen ext4_mb_release_inode_pa(struct ext4_buddy *e4b, struct buffer_head *bitmap_bh,
3797c83617dbSAneesh Kumar K.V 			struct ext4_prealloc_space *pa,
3798c83617dbSAneesh Kumar K.V 			struct ext4_allocation_context *ac)
3799c9de560dSAlex Tomas {
3800c9de560dSAlex Tomas 	struct super_block *sb = e4b->bd_sb;
3801c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
3802498e5f24STheodore Ts'o 	unsigned int end;
3803498e5f24STheodore Ts'o 	unsigned int next;
3804c9de560dSAlex Tomas 	ext4_group_t group;
3805c9de560dSAlex Tomas 	ext4_grpblk_t bit;
3806c9de560dSAlex Tomas 	sector_t start;
3807c9de560dSAlex Tomas 	int err = 0;
3808c9de560dSAlex Tomas 	int free = 0;
3809c9de560dSAlex Tomas 
3810c9de560dSAlex Tomas 	BUG_ON(pa->pa_deleted == 0);
3811c9de560dSAlex Tomas 	ext4_get_group_no_and_offset(sb, pa->pa_pstart, &group, &bit);
3812c9de560dSAlex Tomas 	BUG_ON(group != e4b->bd_group && pa->pa_len != 0);
3813c9de560dSAlex Tomas 	end = bit + pa->pa_len;
3814c9de560dSAlex Tomas 
3815256bdb49SEric Sandeen 	if (ac) {
3816256bdb49SEric Sandeen 		ac->ac_sb = sb;
3817256bdb49SEric Sandeen 		ac->ac_inode = pa->pa_inode;
3818256bdb49SEric Sandeen 		ac->ac_op = EXT4_MB_HISTORY_DISCARD;
3819256bdb49SEric Sandeen 	}
3820c9de560dSAlex Tomas 
3821c9de560dSAlex Tomas 	while (bit < end) {
3822ffad0a44SAneesh Kumar K.V 		bit = mb_find_next_zero_bit(bitmap_bh->b_data, end, bit);
3823c9de560dSAlex Tomas 		if (bit >= end)
3824c9de560dSAlex Tomas 			break;
3825ffad0a44SAneesh Kumar K.V 		next = mb_find_next_bit(bitmap_bh->b_data, end, bit);
3826c9de560dSAlex Tomas 		start = group * EXT4_BLOCKS_PER_GROUP(sb) + bit +
3827c9de560dSAlex Tomas 				le32_to_cpu(sbi->s_es->s_first_data_block);
3828c9de560dSAlex Tomas 		mb_debug("    free preallocated %u/%u in group %u\n",
3829c9de560dSAlex Tomas 				(unsigned) start, (unsigned) next - bit,
3830c9de560dSAlex Tomas 				(unsigned) group);
3831c9de560dSAlex Tomas 		free += next - bit;
3832c9de560dSAlex Tomas 
3833256bdb49SEric Sandeen 		if (ac) {
3834256bdb49SEric Sandeen 			ac->ac_b_ex.fe_group = group;
3835256bdb49SEric Sandeen 			ac->ac_b_ex.fe_start = bit;
3836256bdb49SEric Sandeen 			ac->ac_b_ex.fe_len = next - bit;
3837256bdb49SEric Sandeen 			ac->ac_b_ex.fe_logical = 0;
3838256bdb49SEric Sandeen 			ext4_mb_store_history(ac);
3839256bdb49SEric Sandeen 		}
3840c9de560dSAlex Tomas 
3841c9de560dSAlex Tomas 		mb_free_blocks(pa->pa_inode, e4b, bit, next - bit);
3842c9de560dSAlex Tomas 		bit = next + 1;
3843c9de560dSAlex Tomas 	}
3844c9de560dSAlex Tomas 	if (free != pa->pa_free) {
384526346ff6SAneesh Kumar K.V 		printk(KERN_CRIT "pa %p: logic %lu, phys. %lu, len %lu\n",
3846c9de560dSAlex Tomas 			pa, (unsigned long) pa->pa_lstart,
3847c9de560dSAlex Tomas 			(unsigned long) pa->pa_pstart,
3848c9de560dSAlex Tomas 			(unsigned long) pa->pa_len);
38495d1b1b3fSAneesh Kumar K.V 		ext4_grp_locked_error(sb, group,
38505d1b1b3fSAneesh Kumar K.V 					__func__, "free %u, pa_free %u",
385126346ff6SAneesh Kumar K.V 					free, pa->pa_free);
3852e56eb659SAneesh Kumar K.V 		/*
3853e56eb659SAneesh Kumar K.V 		 * pa is already deleted so we use the value obtained
3854e56eb659SAneesh Kumar K.V 		 * from the bitmap and continue.
3855e56eb659SAneesh Kumar K.V 		 */
3856c9de560dSAlex Tomas 	}
3857c9de560dSAlex Tomas 	atomic_add(free, &sbi->s_mb_discarded);
3858c9de560dSAlex Tomas 
3859c9de560dSAlex Tomas 	return err;
3860c9de560dSAlex Tomas }
3861c9de560dSAlex Tomas 
38624ddfef7bSEric Sandeen static noinline_for_stack int
38634ddfef7bSEric Sandeen ext4_mb_release_group_pa(struct ext4_buddy *e4b,
3864c83617dbSAneesh Kumar K.V 				struct ext4_prealloc_space *pa,
3865c83617dbSAneesh Kumar K.V 				struct ext4_allocation_context *ac)
3866c9de560dSAlex Tomas {
3867c9de560dSAlex Tomas 	struct super_block *sb = e4b->bd_sb;
3868c9de560dSAlex Tomas 	ext4_group_t group;
3869c9de560dSAlex Tomas 	ext4_grpblk_t bit;
3870c9de560dSAlex Tomas 
3871256bdb49SEric Sandeen 	if (ac)
3872256bdb49SEric Sandeen 		ac->ac_op = EXT4_MB_HISTORY_DISCARD;
3873c9de560dSAlex Tomas 
3874c9de560dSAlex Tomas 	BUG_ON(pa->pa_deleted == 0);
3875c9de560dSAlex Tomas 	ext4_get_group_no_and_offset(sb, pa->pa_pstart, &group, &bit);
3876c9de560dSAlex Tomas 	BUG_ON(group != e4b->bd_group && pa->pa_len != 0);
3877c9de560dSAlex Tomas 	mb_free_blocks(pa->pa_inode, e4b, bit, pa->pa_len);
3878c9de560dSAlex Tomas 	atomic_add(pa->pa_len, &EXT4_SB(sb)->s_mb_discarded);
3879c9de560dSAlex Tomas 
3880256bdb49SEric Sandeen 	if (ac) {
3881256bdb49SEric Sandeen 		ac->ac_sb = sb;
3882256bdb49SEric Sandeen 		ac->ac_inode = NULL;
3883256bdb49SEric Sandeen 		ac->ac_b_ex.fe_group = group;
3884256bdb49SEric Sandeen 		ac->ac_b_ex.fe_start = bit;
3885256bdb49SEric Sandeen 		ac->ac_b_ex.fe_len = pa->pa_len;
3886256bdb49SEric Sandeen 		ac->ac_b_ex.fe_logical = 0;
3887256bdb49SEric Sandeen 		ext4_mb_store_history(ac);
3888256bdb49SEric Sandeen 	}
3889c9de560dSAlex Tomas 
3890c9de560dSAlex Tomas 	return 0;
3891c9de560dSAlex Tomas }
3892c9de560dSAlex Tomas 
3893c9de560dSAlex Tomas /*
3894c9de560dSAlex Tomas  * releases all preallocations in given group
3895c9de560dSAlex Tomas  *
3896c9de560dSAlex Tomas  * first, we need to decide discard policy:
3897c9de560dSAlex Tomas  * - when do we discard
3898c9de560dSAlex Tomas  *   1) ENOSPC
3899c9de560dSAlex Tomas  * - how many do we discard
3900c9de560dSAlex Tomas  *   1) how many requested
3901c9de560dSAlex Tomas  */
39024ddfef7bSEric Sandeen static noinline_for_stack int
39034ddfef7bSEric Sandeen ext4_mb_discard_group_preallocations(struct super_block *sb,
3904c9de560dSAlex Tomas 					ext4_group_t group, int needed)
3905c9de560dSAlex Tomas {
3906c9de560dSAlex Tomas 	struct ext4_group_info *grp = ext4_get_group_info(sb, group);
3907c9de560dSAlex Tomas 	struct buffer_head *bitmap_bh = NULL;
3908c9de560dSAlex Tomas 	struct ext4_prealloc_space *pa, *tmp;
3909c83617dbSAneesh Kumar K.V 	struct ext4_allocation_context *ac;
3910c9de560dSAlex Tomas 	struct list_head list;
3911c9de560dSAlex Tomas 	struct ext4_buddy e4b;
3912c9de560dSAlex Tomas 	int err;
3913c9de560dSAlex Tomas 	int busy = 0;
3914c9de560dSAlex Tomas 	int free = 0;
3915c9de560dSAlex Tomas 
3916a9df9a49STheodore Ts'o 	mb_debug("discard preallocation for group %u\n", group);
3917c9de560dSAlex Tomas 
3918c9de560dSAlex Tomas 	if (list_empty(&grp->bb_prealloc_list))
3919c9de560dSAlex Tomas 		return 0;
3920c9de560dSAlex Tomas 
3921574ca174STheodore Ts'o 	bitmap_bh = ext4_read_block_bitmap(sb, group);
3922c9de560dSAlex Tomas 	if (bitmap_bh == NULL) {
3923ce89f46cSAneesh Kumar K.V 		ext4_error(sb, __func__, "Error in reading block "
3924a9df9a49STheodore Ts'o 				"bitmap for %u", group);
3925ce89f46cSAneesh Kumar K.V 		return 0;
3926c9de560dSAlex Tomas 	}
3927c9de560dSAlex Tomas 
3928c9de560dSAlex Tomas 	err = ext4_mb_load_buddy(sb, group, &e4b);
3929ce89f46cSAneesh Kumar K.V 	if (err) {
3930ce89f46cSAneesh Kumar K.V 		ext4_error(sb, __func__, "Error in loading buddy "
3931a9df9a49STheodore Ts'o 				"information for %u", group);
3932ce89f46cSAneesh Kumar K.V 		put_bh(bitmap_bh);
3933ce89f46cSAneesh Kumar K.V 		return 0;
3934ce89f46cSAneesh Kumar K.V 	}
3935c9de560dSAlex Tomas 
3936c9de560dSAlex Tomas 	if (needed == 0)
3937c9de560dSAlex Tomas 		needed = EXT4_BLOCKS_PER_GROUP(sb) + 1;
3938c9de560dSAlex Tomas 
3939c9de560dSAlex Tomas 	INIT_LIST_HEAD(&list);
3940c83617dbSAneesh Kumar K.V 	ac = kmem_cache_alloc(ext4_ac_cachep, GFP_NOFS);
3941c9de560dSAlex Tomas repeat:
3942c9de560dSAlex Tomas 	ext4_lock_group(sb, group);
3943c9de560dSAlex Tomas 	list_for_each_entry_safe(pa, tmp,
3944c9de560dSAlex Tomas 				&grp->bb_prealloc_list, pa_group_list) {
3945c9de560dSAlex Tomas 		spin_lock(&pa->pa_lock);
3946c9de560dSAlex Tomas 		if (atomic_read(&pa->pa_count)) {
3947c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
3948c9de560dSAlex Tomas 			busy = 1;
3949c9de560dSAlex Tomas 			continue;
3950c9de560dSAlex Tomas 		}
3951c9de560dSAlex Tomas 		if (pa->pa_deleted) {
3952c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
3953c9de560dSAlex Tomas 			continue;
3954c9de560dSAlex Tomas 		}
3955c9de560dSAlex Tomas 
3956c9de560dSAlex Tomas 		/* seems this one can be freed ... */
3957c9de560dSAlex Tomas 		pa->pa_deleted = 1;
3958c9de560dSAlex Tomas 
3959c9de560dSAlex Tomas 		/* we can trust pa_free ... */
3960c9de560dSAlex Tomas 		free += pa->pa_free;
3961c9de560dSAlex Tomas 
3962c9de560dSAlex Tomas 		spin_unlock(&pa->pa_lock);
3963c9de560dSAlex Tomas 
3964c9de560dSAlex Tomas 		list_del(&pa->pa_group_list);
3965c9de560dSAlex Tomas 		list_add(&pa->u.pa_tmp_list, &list);
3966c9de560dSAlex Tomas 	}
3967c9de560dSAlex Tomas 
3968c9de560dSAlex Tomas 	/* if we still need more blocks and some PAs were used, try again */
3969c9de560dSAlex Tomas 	if (free < needed && busy) {
3970c9de560dSAlex Tomas 		busy = 0;
3971c9de560dSAlex Tomas 		ext4_unlock_group(sb, group);
3972c9de560dSAlex Tomas 		/*
3973c9de560dSAlex Tomas 		 * Yield the CPU here so that we don't get soft lockup
3974c9de560dSAlex Tomas 		 * in non preempt case.
3975c9de560dSAlex Tomas 		 */
3976c9de560dSAlex Tomas 		yield();
3977c9de560dSAlex Tomas 		goto repeat;
3978c9de560dSAlex Tomas 	}
3979c9de560dSAlex Tomas 
3980c9de560dSAlex Tomas 	/* found anything to free? */
3981c9de560dSAlex Tomas 	if (list_empty(&list)) {
3982c9de560dSAlex Tomas 		BUG_ON(free != 0);
3983c9de560dSAlex Tomas 		goto out;
3984c9de560dSAlex Tomas 	}
3985c9de560dSAlex Tomas 
3986c9de560dSAlex Tomas 	/* now free all selected PAs */
3987c9de560dSAlex Tomas 	list_for_each_entry_safe(pa, tmp, &list, u.pa_tmp_list) {
3988c9de560dSAlex Tomas 
3989c9de560dSAlex Tomas 		/* remove from object (inode or locality group) */
3990c9de560dSAlex Tomas 		spin_lock(pa->pa_obj_lock);
3991c9de560dSAlex Tomas 		list_del_rcu(&pa->pa_inode_list);
3992c9de560dSAlex Tomas 		spin_unlock(pa->pa_obj_lock);
3993c9de560dSAlex Tomas 
3994c9de560dSAlex Tomas 		if (pa->pa_linear)
3995c83617dbSAneesh Kumar K.V 			ext4_mb_release_group_pa(&e4b, pa, ac);
3996c9de560dSAlex Tomas 		else
3997c83617dbSAneesh Kumar K.V 			ext4_mb_release_inode_pa(&e4b, bitmap_bh, pa, ac);
3998c9de560dSAlex Tomas 
3999c9de560dSAlex Tomas 		list_del(&pa->u.pa_tmp_list);
4000c9de560dSAlex Tomas 		call_rcu(&(pa)->u.pa_rcu, ext4_mb_pa_callback);
4001c9de560dSAlex Tomas 	}
4002c9de560dSAlex Tomas 
4003c9de560dSAlex Tomas out:
4004c9de560dSAlex Tomas 	ext4_unlock_group(sb, group);
4005c83617dbSAneesh Kumar K.V 	if (ac)
4006c83617dbSAneesh Kumar K.V 		kmem_cache_free(ext4_ac_cachep, ac);
4007c9de560dSAlex Tomas 	ext4_mb_release_desc(&e4b);
4008c9de560dSAlex Tomas 	put_bh(bitmap_bh);
4009c9de560dSAlex Tomas 	return free;
4010c9de560dSAlex Tomas }
4011c9de560dSAlex Tomas 
4012c9de560dSAlex Tomas /*
4013c9de560dSAlex Tomas  * releases all non-used preallocated blocks for given inode
4014c9de560dSAlex Tomas  *
4015c9de560dSAlex Tomas  * It's important to discard preallocations under i_data_sem
4016c9de560dSAlex Tomas  * We don't want another block to be served from the prealloc
4017c9de560dSAlex Tomas  * space when we are discarding the inode prealloc space.
4018c9de560dSAlex Tomas  *
4019c9de560dSAlex Tomas  * FIXME!! Make sure it is valid at all the call sites
4020c9de560dSAlex Tomas  */
4021c2ea3fdeSTheodore Ts'o void ext4_discard_preallocations(struct inode *inode)
4022c9de560dSAlex Tomas {
4023c9de560dSAlex Tomas 	struct ext4_inode_info *ei = EXT4_I(inode);
4024c9de560dSAlex Tomas 	struct super_block *sb = inode->i_sb;
4025c9de560dSAlex Tomas 	struct buffer_head *bitmap_bh = NULL;
4026c9de560dSAlex Tomas 	struct ext4_prealloc_space *pa, *tmp;
4027c83617dbSAneesh Kumar K.V 	struct ext4_allocation_context *ac;
4028c9de560dSAlex Tomas 	ext4_group_t group = 0;
4029c9de560dSAlex Tomas 	struct list_head list;
4030c9de560dSAlex Tomas 	struct ext4_buddy e4b;
4031c9de560dSAlex Tomas 	int err;
4032c9de560dSAlex Tomas 
4033c2ea3fdeSTheodore Ts'o 	if (!S_ISREG(inode->i_mode)) {
4034c9de560dSAlex Tomas 		/*BUG_ON(!list_empty(&ei->i_prealloc_list));*/
4035c9de560dSAlex Tomas 		return;
4036c9de560dSAlex Tomas 	}
4037c9de560dSAlex Tomas 
4038c9de560dSAlex Tomas 	mb_debug("discard preallocation for inode %lu\n", inode->i_ino);
4039c9de560dSAlex Tomas 
4040c9de560dSAlex Tomas 	INIT_LIST_HEAD(&list);
4041c9de560dSAlex Tomas 
4042c83617dbSAneesh Kumar K.V 	ac = kmem_cache_alloc(ext4_ac_cachep, GFP_NOFS);
4043c9de560dSAlex Tomas repeat:
4044c9de560dSAlex Tomas 	/* first, collect all pa's in the inode */
4045c9de560dSAlex Tomas 	spin_lock(&ei->i_prealloc_lock);
4046c9de560dSAlex Tomas 	while (!list_empty(&ei->i_prealloc_list)) {
4047c9de560dSAlex Tomas 		pa = list_entry(ei->i_prealloc_list.next,
4048c9de560dSAlex Tomas 				struct ext4_prealloc_space, pa_inode_list);
4049c9de560dSAlex Tomas 		BUG_ON(pa->pa_obj_lock != &ei->i_prealloc_lock);
4050c9de560dSAlex Tomas 		spin_lock(&pa->pa_lock);
4051c9de560dSAlex Tomas 		if (atomic_read(&pa->pa_count)) {
4052c9de560dSAlex Tomas 			/* this shouldn't happen often - nobody should
4053c9de560dSAlex Tomas 			 * use preallocation while we're discarding it */
4054c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
4055c9de560dSAlex Tomas 			spin_unlock(&ei->i_prealloc_lock);
4056c9de560dSAlex Tomas 			printk(KERN_ERR "uh-oh! used pa while discarding\n");
4057c9de560dSAlex Tomas 			WARN_ON(1);
4058c9de560dSAlex Tomas 			schedule_timeout_uninterruptible(HZ);
4059c9de560dSAlex Tomas 			goto repeat;
4060c9de560dSAlex Tomas 
4061c9de560dSAlex Tomas 		}
4062c9de560dSAlex Tomas 		if (pa->pa_deleted == 0) {
4063c9de560dSAlex Tomas 			pa->pa_deleted = 1;
4064c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
4065c9de560dSAlex Tomas 			list_del_rcu(&pa->pa_inode_list);
4066c9de560dSAlex Tomas 			list_add(&pa->u.pa_tmp_list, &list);
4067c9de560dSAlex Tomas 			continue;
4068c9de560dSAlex Tomas 		}
4069c9de560dSAlex Tomas 
4070c9de560dSAlex Tomas 		/* someone is deleting pa right now */
4071c9de560dSAlex Tomas 		spin_unlock(&pa->pa_lock);
4072c9de560dSAlex Tomas 		spin_unlock(&ei->i_prealloc_lock);
4073c9de560dSAlex Tomas 
4074c9de560dSAlex Tomas 		/* we have to wait here because pa_deleted
4075c9de560dSAlex Tomas 		 * doesn't mean pa is already unlinked from
4076c9de560dSAlex Tomas 		 * the list. as we might be called from
4077c9de560dSAlex Tomas 		 * ->clear_inode() the inode will get freed
4078c9de560dSAlex Tomas 		 * and concurrent thread which is unlinking
4079c9de560dSAlex Tomas 		 * pa from inode's list may access already
4080c9de560dSAlex Tomas 		 * freed memory, bad-bad-bad */
4081c9de560dSAlex Tomas 
4082c9de560dSAlex Tomas 		/* XXX: if this happens too often, we can
4083c9de560dSAlex Tomas 		 * add a flag to force wait only in case
4084c9de560dSAlex Tomas 		 * of ->clear_inode(), but not in case of
4085c9de560dSAlex Tomas 		 * regular truncate */
4086c9de560dSAlex Tomas 		schedule_timeout_uninterruptible(HZ);
4087c9de560dSAlex Tomas 		goto repeat;
4088c9de560dSAlex Tomas 	}
4089c9de560dSAlex Tomas 	spin_unlock(&ei->i_prealloc_lock);
4090c9de560dSAlex Tomas 
4091c9de560dSAlex Tomas 	list_for_each_entry_safe(pa, tmp, &list, u.pa_tmp_list) {
4092c9de560dSAlex Tomas 		BUG_ON(pa->pa_linear != 0);
4093c9de560dSAlex Tomas 		ext4_get_group_no_and_offset(sb, pa->pa_pstart, &group, NULL);
4094c9de560dSAlex Tomas 
4095c9de560dSAlex Tomas 		err = ext4_mb_load_buddy(sb, group, &e4b);
4096ce89f46cSAneesh Kumar K.V 		if (err) {
4097ce89f46cSAneesh Kumar K.V 			ext4_error(sb, __func__, "Error in loading buddy "
4098a9df9a49STheodore Ts'o 					"information for %u", group);
4099ce89f46cSAneesh Kumar K.V 			continue;
4100ce89f46cSAneesh Kumar K.V 		}
4101c9de560dSAlex Tomas 
4102574ca174STheodore Ts'o 		bitmap_bh = ext4_read_block_bitmap(sb, group);
4103c9de560dSAlex Tomas 		if (bitmap_bh == NULL) {
4104ce89f46cSAneesh Kumar K.V 			ext4_error(sb, __func__, "Error in reading block "
4105a9df9a49STheodore Ts'o 					"bitmap for %u", group);
4106c9de560dSAlex Tomas 			ext4_mb_release_desc(&e4b);
4107ce89f46cSAneesh Kumar K.V 			continue;
4108c9de560dSAlex Tomas 		}
4109c9de560dSAlex Tomas 
4110c9de560dSAlex Tomas 		ext4_lock_group(sb, group);
4111c9de560dSAlex Tomas 		list_del(&pa->pa_group_list);
4112c83617dbSAneesh Kumar K.V 		ext4_mb_release_inode_pa(&e4b, bitmap_bh, pa, ac);
4113c9de560dSAlex Tomas 		ext4_unlock_group(sb, group);
4114c9de560dSAlex Tomas 
4115c9de560dSAlex Tomas 		ext4_mb_release_desc(&e4b);
4116c9de560dSAlex Tomas 		put_bh(bitmap_bh);
4117c9de560dSAlex Tomas 
4118c9de560dSAlex Tomas 		list_del(&pa->u.pa_tmp_list);
4119c9de560dSAlex Tomas 		call_rcu(&(pa)->u.pa_rcu, ext4_mb_pa_callback);
4120c9de560dSAlex Tomas 	}
4121c83617dbSAneesh Kumar K.V 	if (ac)
4122c83617dbSAneesh Kumar K.V 		kmem_cache_free(ext4_ac_cachep, ac);
4123c9de560dSAlex Tomas }
4124c9de560dSAlex Tomas 
4125c9de560dSAlex Tomas /*
4126c9de560dSAlex Tomas  * finds all preallocated spaces and return blocks being freed to them
4127c9de560dSAlex Tomas  * if preallocated space becomes full (no block is used from the space)
4128c9de560dSAlex Tomas  * then the function frees space in buddy
4129c9de560dSAlex Tomas  * XXX: at the moment, truncate (which is the only way to free blocks)
4130c9de560dSAlex Tomas  * discards all preallocations
4131c9de560dSAlex Tomas  */
4132c9de560dSAlex Tomas static void ext4_mb_return_to_preallocation(struct inode *inode,
4133c9de560dSAlex Tomas 					struct ext4_buddy *e4b,
4134c9de560dSAlex Tomas 					sector_t block, int count)
4135c9de560dSAlex Tomas {
4136c9de560dSAlex Tomas 	BUG_ON(!list_empty(&EXT4_I(inode)->i_prealloc_list));
4137c9de560dSAlex Tomas }
4138c9de560dSAlex Tomas #ifdef MB_DEBUG
4139c9de560dSAlex Tomas static void ext4_mb_show_ac(struct ext4_allocation_context *ac)
4140c9de560dSAlex Tomas {
4141c9de560dSAlex Tomas 	struct super_block *sb = ac->ac_sb;
4142c9de560dSAlex Tomas 	ext4_group_t i;
4143c9de560dSAlex Tomas 
4144c9de560dSAlex Tomas 	printk(KERN_ERR "EXT4-fs: Can't allocate:"
4145c9de560dSAlex Tomas 			" Allocation context details:\n");
4146c9de560dSAlex Tomas 	printk(KERN_ERR "EXT4-fs: status %d flags %d\n",
4147c9de560dSAlex Tomas 			ac->ac_status, ac->ac_flags);
4148c9de560dSAlex Tomas 	printk(KERN_ERR "EXT4-fs: orig %lu/%lu/%lu@%lu, goal %lu/%lu/%lu@%lu, "
4149c9de560dSAlex Tomas 			"best %lu/%lu/%lu@%lu cr %d\n",
4150c9de560dSAlex Tomas 			(unsigned long)ac->ac_o_ex.fe_group,
4151c9de560dSAlex Tomas 			(unsigned long)ac->ac_o_ex.fe_start,
4152c9de560dSAlex Tomas 			(unsigned long)ac->ac_o_ex.fe_len,
4153c9de560dSAlex Tomas 			(unsigned long)ac->ac_o_ex.fe_logical,
4154c9de560dSAlex Tomas 			(unsigned long)ac->ac_g_ex.fe_group,
4155c9de560dSAlex Tomas 			(unsigned long)ac->ac_g_ex.fe_start,
4156c9de560dSAlex Tomas 			(unsigned long)ac->ac_g_ex.fe_len,
4157c9de560dSAlex Tomas 			(unsigned long)ac->ac_g_ex.fe_logical,
4158c9de560dSAlex Tomas 			(unsigned long)ac->ac_b_ex.fe_group,
4159c9de560dSAlex Tomas 			(unsigned long)ac->ac_b_ex.fe_start,
4160c9de560dSAlex Tomas 			(unsigned long)ac->ac_b_ex.fe_len,
4161c9de560dSAlex Tomas 			(unsigned long)ac->ac_b_ex.fe_logical,
4162c9de560dSAlex Tomas 			(int)ac->ac_criteria);
4163c9de560dSAlex Tomas 	printk(KERN_ERR "EXT4-fs: %lu scanned, %d found\n", ac->ac_ex_scanned,
4164c9de560dSAlex Tomas 		ac->ac_found);
4165c9de560dSAlex Tomas 	printk(KERN_ERR "EXT4-fs: groups: \n");
4166c9de560dSAlex Tomas 	for (i = 0; i < EXT4_SB(sb)->s_groups_count; i++) {
4167c9de560dSAlex Tomas 		struct ext4_group_info *grp = ext4_get_group_info(sb, i);
4168c9de560dSAlex Tomas 		struct ext4_prealloc_space *pa;
4169c9de560dSAlex Tomas 		ext4_grpblk_t start;
4170c9de560dSAlex Tomas 		struct list_head *cur;
4171c9de560dSAlex Tomas 		ext4_lock_group(sb, i);
4172c9de560dSAlex Tomas 		list_for_each(cur, &grp->bb_prealloc_list) {
4173c9de560dSAlex Tomas 			pa = list_entry(cur, struct ext4_prealloc_space,
4174c9de560dSAlex Tomas 					pa_group_list);
4175c9de560dSAlex Tomas 			spin_lock(&pa->pa_lock);
4176c9de560dSAlex Tomas 			ext4_get_group_no_and_offset(sb, pa->pa_pstart,
4177c9de560dSAlex Tomas 						     NULL, &start);
4178c9de560dSAlex Tomas 			spin_unlock(&pa->pa_lock);
4179c9de560dSAlex Tomas 			printk(KERN_ERR "PA:%lu:%d:%u \n", i,
4180c9de560dSAlex Tomas 							start, pa->pa_len);
4181c9de560dSAlex Tomas 		}
418260bd63d1SSolofo Ramangalahy 		ext4_unlock_group(sb, i);
4183c9de560dSAlex Tomas 
4184c9de560dSAlex Tomas 		if (grp->bb_free == 0)
4185c9de560dSAlex Tomas 			continue;
4186c9de560dSAlex Tomas 		printk(KERN_ERR "%lu: %d/%d \n",
4187c9de560dSAlex Tomas 		       i, grp->bb_free, grp->bb_fragments);
4188c9de560dSAlex Tomas 	}
4189c9de560dSAlex Tomas 	printk(KERN_ERR "\n");
4190c9de560dSAlex Tomas }
4191c9de560dSAlex Tomas #else
4192c9de560dSAlex Tomas static inline void ext4_mb_show_ac(struct ext4_allocation_context *ac)
4193c9de560dSAlex Tomas {
4194c9de560dSAlex Tomas 	return;
4195c9de560dSAlex Tomas }
4196c9de560dSAlex Tomas #endif
4197c9de560dSAlex Tomas 
4198c9de560dSAlex Tomas /*
4199c9de560dSAlex Tomas  * We use locality group preallocation for small size file. The size of the
4200c9de560dSAlex Tomas  * file is determined by the current size or the resulting size after
4201c9de560dSAlex Tomas  * allocation which ever is larger
4202c9de560dSAlex Tomas  *
4203c9de560dSAlex Tomas  * One can tune this size via /proc/fs/ext4/<partition>/stream_req
4204c9de560dSAlex Tomas  */
4205c9de560dSAlex Tomas static void ext4_mb_group_or_file(struct ext4_allocation_context *ac)
4206c9de560dSAlex Tomas {
4207c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(ac->ac_sb);
4208c9de560dSAlex Tomas 	int bsbits = ac->ac_sb->s_blocksize_bits;
4209c9de560dSAlex Tomas 	loff_t size, isize;
4210c9de560dSAlex Tomas 
4211c9de560dSAlex Tomas 	if (!(ac->ac_flags & EXT4_MB_HINT_DATA))
4212c9de560dSAlex Tomas 		return;
4213c9de560dSAlex Tomas 
4214c9de560dSAlex Tomas 	size = ac->ac_o_ex.fe_logical + ac->ac_o_ex.fe_len;
4215c9de560dSAlex Tomas 	isize = i_size_read(ac->ac_inode) >> bsbits;
4216c9de560dSAlex Tomas 	size = max(size, isize);
4217c9de560dSAlex Tomas 
4218c9de560dSAlex Tomas 	/* don't use group allocation for large files */
4219c9de560dSAlex Tomas 	if (size >= sbi->s_mb_stream_request)
4220c9de560dSAlex Tomas 		return;
4221c9de560dSAlex Tomas 
4222c9de560dSAlex Tomas 	if (unlikely(ac->ac_flags & EXT4_MB_HINT_GOAL_ONLY))
4223c9de560dSAlex Tomas 		return;
4224c9de560dSAlex Tomas 
4225c9de560dSAlex Tomas 	BUG_ON(ac->ac_lg != NULL);
4226c9de560dSAlex Tomas 	/*
4227c9de560dSAlex Tomas 	 * locality group prealloc space are per cpu. The reason for having
4228c9de560dSAlex Tomas 	 * per cpu locality group is to reduce the contention between block
4229c9de560dSAlex Tomas 	 * request from multiple CPUs.
4230c9de560dSAlex Tomas 	 */
4231730c213cSEric Sandeen 	ac->ac_lg = per_cpu_ptr(sbi->s_locality_groups, raw_smp_processor_id());
4232c9de560dSAlex Tomas 
4233c9de560dSAlex Tomas 	/* we're going to use group allocation */
4234c9de560dSAlex Tomas 	ac->ac_flags |= EXT4_MB_HINT_GROUP_ALLOC;
4235c9de560dSAlex Tomas 
4236c9de560dSAlex Tomas 	/* serialize all allocations in the group */
4237c9de560dSAlex Tomas 	mutex_lock(&ac->ac_lg->lg_mutex);
4238c9de560dSAlex Tomas }
4239c9de560dSAlex Tomas 
42404ddfef7bSEric Sandeen static noinline_for_stack int
42414ddfef7bSEric Sandeen ext4_mb_initialize_context(struct ext4_allocation_context *ac,
4242c9de560dSAlex Tomas 				struct ext4_allocation_request *ar)
4243c9de560dSAlex Tomas {
4244c9de560dSAlex Tomas 	struct super_block *sb = ar->inode->i_sb;
4245c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
4246c9de560dSAlex Tomas 	struct ext4_super_block *es = sbi->s_es;
4247c9de560dSAlex Tomas 	ext4_group_t group;
4248498e5f24STheodore Ts'o 	unsigned int len;
4249498e5f24STheodore Ts'o 	ext4_fsblk_t goal;
4250c9de560dSAlex Tomas 	ext4_grpblk_t block;
4251c9de560dSAlex Tomas 
4252c9de560dSAlex Tomas 	/* we can't allocate > group size */
4253c9de560dSAlex Tomas 	len = ar->len;
4254c9de560dSAlex Tomas 
4255c9de560dSAlex Tomas 	/* just a dirty hack to filter too big requests  */
4256c9de560dSAlex Tomas 	if (len >= EXT4_BLOCKS_PER_GROUP(sb) - 10)
4257c9de560dSAlex Tomas 		len = EXT4_BLOCKS_PER_GROUP(sb) - 10;
4258c9de560dSAlex Tomas 
4259c9de560dSAlex Tomas 	/* start searching from the goal */
4260c9de560dSAlex Tomas 	goal = ar->goal;
4261c9de560dSAlex Tomas 	if (goal < le32_to_cpu(es->s_first_data_block) ||
4262c9de560dSAlex Tomas 			goal >= ext4_blocks_count(es))
4263c9de560dSAlex Tomas 		goal = le32_to_cpu(es->s_first_data_block);
4264c9de560dSAlex Tomas 	ext4_get_group_no_and_offset(sb, goal, &group, &block);
4265c9de560dSAlex Tomas 
4266c9de560dSAlex Tomas 	/* set up allocation goals */
4267c9de560dSAlex Tomas 	ac->ac_b_ex.fe_logical = ar->logical;
4268c9de560dSAlex Tomas 	ac->ac_b_ex.fe_group = 0;
4269c9de560dSAlex Tomas 	ac->ac_b_ex.fe_start = 0;
4270c9de560dSAlex Tomas 	ac->ac_b_ex.fe_len = 0;
4271c9de560dSAlex Tomas 	ac->ac_status = AC_STATUS_CONTINUE;
4272c9de560dSAlex Tomas 	ac->ac_groups_scanned = 0;
4273c9de560dSAlex Tomas 	ac->ac_ex_scanned = 0;
4274c9de560dSAlex Tomas 	ac->ac_found = 0;
4275c9de560dSAlex Tomas 	ac->ac_sb = sb;
4276c9de560dSAlex Tomas 	ac->ac_inode = ar->inode;
4277c9de560dSAlex Tomas 	ac->ac_o_ex.fe_logical = ar->logical;
4278c9de560dSAlex Tomas 	ac->ac_o_ex.fe_group = group;
4279c9de560dSAlex Tomas 	ac->ac_o_ex.fe_start = block;
4280c9de560dSAlex Tomas 	ac->ac_o_ex.fe_len = len;
4281c9de560dSAlex Tomas 	ac->ac_g_ex.fe_logical = ar->logical;
4282c9de560dSAlex Tomas 	ac->ac_g_ex.fe_group = group;
4283c9de560dSAlex Tomas 	ac->ac_g_ex.fe_start = block;
4284c9de560dSAlex Tomas 	ac->ac_g_ex.fe_len = len;
4285c9de560dSAlex Tomas 	ac->ac_f_ex.fe_len = 0;
4286c9de560dSAlex Tomas 	ac->ac_flags = ar->flags;
4287c9de560dSAlex Tomas 	ac->ac_2order = 0;
4288c9de560dSAlex Tomas 	ac->ac_criteria = 0;
4289c9de560dSAlex Tomas 	ac->ac_pa = NULL;
4290c9de560dSAlex Tomas 	ac->ac_bitmap_page = NULL;
4291c9de560dSAlex Tomas 	ac->ac_buddy_page = NULL;
4292c9de560dSAlex Tomas 	ac->ac_lg = NULL;
4293c9de560dSAlex Tomas 
4294c9de560dSAlex Tomas 	/* we have to define context: we'll we work with a file or
4295c9de560dSAlex Tomas 	 * locality group. this is a policy, actually */
4296c9de560dSAlex Tomas 	ext4_mb_group_or_file(ac);
4297c9de560dSAlex Tomas 
4298c9de560dSAlex Tomas 	mb_debug("init ac: %u blocks @ %u, goal %u, flags %x, 2^%d, "
4299c9de560dSAlex Tomas 			"left: %u/%u, right %u/%u to %swritable\n",
4300c9de560dSAlex Tomas 			(unsigned) ar->len, (unsigned) ar->logical,
4301c9de560dSAlex Tomas 			(unsigned) ar->goal, ac->ac_flags, ac->ac_2order,
4302c9de560dSAlex Tomas 			(unsigned) ar->lleft, (unsigned) ar->pleft,
4303c9de560dSAlex Tomas 			(unsigned) ar->lright, (unsigned) ar->pright,
4304c9de560dSAlex Tomas 			atomic_read(&ar->inode->i_writecount) ? "" : "non-");
4305c9de560dSAlex Tomas 	return 0;
4306c9de560dSAlex Tomas 
4307c9de560dSAlex Tomas }
4308c9de560dSAlex Tomas 
43096be2ded1SAneesh Kumar K.V static noinline_for_stack void
43106be2ded1SAneesh Kumar K.V ext4_mb_discard_lg_preallocations(struct super_block *sb,
43116be2ded1SAneesh Kumar K.V 					struct ext4_locality_group *lg,
43126be2ded1SAneesh Kumar K.V 					int order, int total_entries)
43136be2ded1SAneesh Kumar K.V {
43146be2ded1SAneesh Kumar K.V 	ext4_group_t group = 0;
43156be2ded1SAneesh Kumar K.V 	struct ext4_buddy e4b;
43166be2ded1SAneesh Kumar K.V 	struct list_head discard_list;
43176be2ded1SAneesh Kumar K.V 	struct ext4_prealloc_space *pa, *tmp;
43186be2ded1SAneesh Kumar K.V 	struct ext4_allocation_context *ac;
43196be2ded1SAneesh Kumar K.V 
43206be2ded1SAneesh Kumar K.V 	mb_debug("discard locality group preallocation\n");
43216be2ded1SAneesh Kumar K.V 
43226be2ded1SAneesh Kumar K.V 	INIT_LIST_HEAD(&discard_list);
43236be2ded1SAneesh Kumar K.V 	ac = kmem_cache_alloc(ext4_ac_cachep, GFP_NOFS);
43246be2ded1SAneesh Kumar K.V 
43256be2ded1SAneesh Kumar K.V 	spin_lock(&lg->lg_prealloc_lock);
43266be2ded1SAneesh Kumar K.V 	list_for_each_entry_rcu(pa, &lg->lg_prealloc_list[order],
43276be2ded1SAneesh Kumar K.V 						pa_inode_list) {
43286be2ded1SAneesh Kumar K.V 		spin_lock(&pa->pa_lock);
43296be2ded1SAneesh Kumar K.V 		if (atomic_read(&pa->pa_count)) {
43306be2ded1SAneesh Kumar K.V 			/*
43316be2ded1SAneesh Kumar K.V 			 * This is the pa that we just used
43326be2ded1SAneesh Kumar K.V 			 * for block allocation. So don't
43336be2ded1SAneesh Kumar K.V 			 * free that
43346be2ded1SAneesh Kumar K.V 			 */
43356be2ded1SAneesh Kumar K.V 			spin_unlock(&pa->pa_lock);
43366be2ded1SAneesh Kumar K.V 			continue;
43376be2ded1SAneesh Kumar K.V 		}
43386be2ded1SAneesh Kumar K.V 		if (pa->pa_deleted) {
43396be2ded1SAneesh Kumar K.V 			spin_unlock(&pa->pa_lock);
43406be2ded1SAneesh Kumar K.V 			continue;
43416be2ded1SAneesh Kumar K.V 		}
43426be2ded1SAneesh Kumar K.V 		/* only lg prealloc space */
43436be2ded1SAneesh Kumar K.V 		BUG_ON(!pa->pa_linear);
43446be2ded1SAneesh Kumar K.V 
43456be2ded1SAneesh Kumar K.V 		/* seems this one can be freed ... */
43466be2ded1SAneesh Kumar K.V 		pa->pa_deleted = 1;
43476be2ded1SAneesh Kumar K.V 		spin_unlock(&pa->pa_lock);
43486be2ded1SAneesh Kumar K.V 
43496be2ded1SAneesh Kumar K.V 		list_del_rcu(&pa->pa_inode_list);
43506be2ded1SAneesh Kumar K.V 		list_add(&pa->u.pa_tmp_list, &discard_list);
43516be2ded1SAneesh Kumar K.V 
43526be2ded1SAneesh Kumar K.V 		total_entries--;
43536be2ded1SAneesh Kumar K.V 		if (total_entries <= 5) {
43546be2ded1SAneesh Kumar K.V 			/*
43556be2ded1SAneesh Kumar K.V 			 * we want to keep only 5 entries
43566be2ded1SAneesh Kumar K.V 			 * allowing it to grow to 8. This
43576be2ded1SAneesh Kumar K.V 			 * mak sure we don't call discard
43586be2ded1SAneesh Kumar K.V 			 * soon for this list.
43596be2ded1SAneesh Kumar K.V 			 */
43606be2ded1SAneesh Kumar K.V 			break;
43616be2ded1SAneesh Kumar K.V 		}
43626be2ded1SAneesh Kumar K.V 	}
43636be2ded1SAneesh Kumar K.V 	spin_unlock(&lg->lg_prealloc_lock);
43646be2ded1SAneesh Kumar K.V 
43656be2ded1SAneesh Kumar K.V 	list_for_each_entry_safe(pa, tmp, &discard_list, u.pa_tmp_list) {
43666be2ded1SAneesh Kumar K.V 
43676be2ded1SAneesh Kumar K.V 		ext4_get_group_no_and_offset(sb, pa->pa_pstart, &group, NULL);
43686be2ded1SAneesh Kumar K.V 		if (ext4_mb_load_buddy(sb, group, &e4b)) {
43696be2ded1SAneesh Kumar K.V 			ext4_error(sb, __func__, "Error in loading buddy "
4370a9df9a49STheodore Ts'o 					"information for %u", group);
43716be2ded1SAneesh Kumar K.V 			continue;
43726be2ded1SAneesh Kumar K.V 		}
43736be2ded1SAneesh Kumar K.V 		ext4_lock_group(sb, group);
43746be2ded1SAneesh Kumar K.V 		list_del(&pa->pa_group_list);
43756be2ded1SAneesh Kumar K.V 		ext4_mb_release_group_pa(&e4b, pa, ac);
43766be2ded1SAneesh Kumar K.V 		ext4_unlock_group(sb, group);
43776be2ded1SAneesh Kumar K.V 
43786be2ded1SAneesh Kumar K.V 		ext4_mb_release_desc(&e4b);
43796be2ded1SAneesh Kumar K.V 		list_del(&pa->u.pa_tmp_list);
43806be2ded1SAneesh Kumar K.V 		call_rcu(&(pa)->u.pa_rcu, ext4_mb_pa_callback);
43816be2ded1SAneesh Kumar K.V 	}
43826be2ded1SAneesh Kumar K.V 	if (ac)
43836be2ded1SAneesh Kumar K.V 		kmem_cache_free(ext4_ac_cachep, ac);
43846be2ded1SAneesh Kumar K.V }
43856be2ded1SAneesh Kumar K.V 
43866be2ded1SAneesh Kumar K.V /*
43876be2ded1SAneesh Kumar K.V  * We have incremented pa_count. So it cannot be freed at this
43886be2ded1SAneesh Kumar K.V  * point. Also we hold lg_mutex. So no parallel allocation is
43896be2ded1SAneesh Kumar K.V  * possible from this lg. That means pa_free cannot be updated.
43906be2ded1SAneesh Kumar K.V  *
43916be2ded1SAneesh Kumar K.V  * A parallel ext4_mb_discard_group_preallocations is possible.
43926be2ded1SAneesh Kumar K.V  * which can cause the lg_prealloc_list to be updated.
43936be2ded1SAneesh Kumar K.V  */
43946be2ded1SAneesh Kumar K.V 
43956be2ded1SAneesh Kumar K.V static void ext4_mb_add_n_trim(struct ext4_allocation_context *ac)
43966be2ded1SAneesh Kumar K.V {
43976be2ded1SAneesh Kumar K.V 	int order, added = 0, lg_prealloc_count = 1;
43986be2ded1SAneesh Kumar K.V 	struct super_block *sb = ac->ac_sb;
43996be2ded1SAneesh Kumar K.V 	struct ext4_locality_group *lg = ac->ac_lg;
44006be2ded1SAneesh Kumar K.V 	struct ext4_prealloc_space *tmp_pa, *pa = ac->ac_pa;
44016be2ded1SAneesh Kumar K.V 
44026be2ded1SAneesh Kumar K.V 	order = fls(pa->pa_free) - 1;
44036be2ded1SAneesh Kumar K.V 	if (order > PREALLOC_TB_SIZE - 1)
44046be2ded1SAneesh Kumar K.V 		/* The max size of hash table is PREALLOC_TB_SIZE */
44056be2ded1SAneesh Kumar K.V 		order = PREALLOC_TB_SIZE - 1;
44066be2ded1SAneesh Kumar K.V 	/* Add the prealloc space to lg */
44076be2ded1SAneesh Kumar K.V 	rcu_read_lock();
44086be2ded1SAneesh Kumar K.V 	list_for_each_entry_rcu(tmp_pa, &lg->lg_prealloc_list[order],
44096be2ded1SAneesh Kumar K.V 						pa_inode_list) {
44106be2ded1SAneesh Kumar K.V 		spin_lock(&tmp_pa->pa_lock);
44116be2ded1SAneesh Kumar K.V 		if (tmp_pa->pa_deleted) {
44126be2ded1SAneesh Kumar K.V 			spin_unlock(&pa->pa_lock);
44136be2ded1SAneesh Kumar K.V 			continue;
44146be2ded1SAneesh Kumar K.V 		}
44156be2ded1SAneesh Kumar K.V 		if (!added && pa->pa_free < tmp_pa->pa_free) {
44166be2ded1SAneesh Kumar K.V 			/* Add to the tail of the previous entry */
44176be2ded1SAneesh Kumar K.V 			list_add_tail_rcu(&pa->pa_inode_list,
44186be2ded1SAneesh Kumar K.V 						&tmp_pa->pa_inode_list);
44196be2ded1SAneesh Kumar K.V 			added = 1;
44206be2ded1SAneesh Kumar K.V 			/*
44216be2ded1SAneesh Kumar K.V 			 * we want to count the total
44226be2ded1SAneesh Kumar K.V 			 * number of entries in the list
44236be2ded1SAneesh Kumar K.V 			 */
44246be2ded1SAneesh Kumar K.V 		}
44256be2ded1SAneesh Kumar K.V 		spin_unlock(&tmp_pa->pa_lock);
44266be2ded1SAneesh Kumar K.V 		lg_prealloc_count++;
44276be2ded1SAneesh Kumar K.V 	}
44286be2ded1SAneesh Kumar K.V 	if (!added)
44296be2ded1SAneesh Kumar K.V 		list_add_tail_rcu(&pa->pa_inode_list,
44306be2ded1SAneesh Kumar K.V 					&lg->lg_prealloc_list[order]);
44316be2ded1SAneesh Kumar K.V 	rcu_read_unlock();
44326be2ded1SAneesh Kumar K.V 
44336be2ded1SAneesh Kumar K.V 	/* Now trim the list to be not more than 8 elements */
44346be2ded1SAneesh Kumar K.V 	if (lg_prealloc_count > 8) {
44356be2ded1SAneesh Kumar K.V 		ext4_mb_discard_lg_preallocations(sb, lg,
44366be2ded1SAneesh Kumar K.V 						order, lg_prealloc_count);
44376be2ded1SAneesh Kumar K.V 		return;
44386be2ded1SAneesh Kumar K.V 	}
44396be2ded1SAneesh Kumar K.V 	return ;
44406be2ded1SAneesh Kumar K.V }
44416be2ded1SAneesh Kumar K.V 
4442c9de560dSAlex Tomas /*
4443c9de560dSAlex Tomas  * release all resource we used in allocation
4444c9de560dSAlex Tomas  */
4445c9de560dSAlex Tomas static int ext4_mb_release_context(struct ext4_allocation_context *ac)
4446c9de560dSAlex Tomas {
44476be2ded1SAneesh Kumar K.V 	struct ext4_prealloc_space *pa = ac->ac_pa;
44486be2ded1SAneesh Kumar K.V 	if (pa) {
44496be2ded1SAneesh Kumar K.V 		if (pa->pa_linear) {
4450c9de560dSAlex Tomas 			/* see comment in ext4_mb_use_group_pa() */
44516be2ded1SAneesh Kumar K.V 			spin_lock(&pa->pa_lock);
44526be2ded1SAneesh Kumar K.V 			pa->pa_pstart += ac->ac_b_ex.fe_len;
44536be2ded1SAneesh Kumar K.V 			pa->pa_lstart += ac->ac_b_ex.fe_len;
44546be2ded1SAneesh Kumar K.V 			pa->pa_free -= ac->ac_b_ex.fe_len;
44556be2ded1SAneesh Kumar K.V 			pa->pa_len -= ac->ac_b_ex.fe_len;
44566be2ded1SAneesh Kumar K.V 			spin_unlock(&pa->pa_lock);
44576be2ded1SAneesh Kumar K.V 			/*
44586be2ded1SAneesh Kumar K.V 			 * We want to add the pa to the right bucket.
44596be2ded1SAneesh Kumar K.V 			 * Remove it from the list and while adding
44606be2ded1SAneesh Kumar K.V 			 * make sure the list to which we are adding
44616be2ded1SAneesh Kumar K.V 			 * doesn't grow big.
44626be2ded1SAneesh Kumar K.V 			 */
44636be2ded1SAneesh Kumar K.V 			if (likely(pa->pa_free)) {
44646be2ded1SAneesh Kumar K.V 				spin_lock(pa->pa_obj_lock);
44656be2ded1SAneesh Kumar K.V 				list_del_rcu(&pa->pa_inode_list);
44666be2ded1SAneesh Kumar K.V 				spin_unlock(pa->pa_obj_lock);
44676be2ded1SAneesh Kumar K.V 				ext4_mb_add_n_trim(ac);
4468c9de560dSAlex Tomas 			}
44696be2ded1SAneesh Kumar K.V 		}
44706be2ded1SAneesh Kumar K.V 		ext4_mb_put_pa(ac, ac->ac_sb, pa);
4471c9de560dSAlex Tomas 	}
4472c9de560dSAlex Tomas 	if (ac->ac_bitmap_page)
4473c9de560dSAlex Tomas 		page_cache_release(ac->ac_bitmap_page);
4474c9de560dSAlex Tomas 	if (ac->ac_buddy_page)
4475c9de560dSAlex Tomas 		page_cache_release(ac->ac_buddy_page);
4476c9de560dSAlex Tomas 	if (ac->ac_flags & EXT4_MB_HINT_GROUP_ALLOC)
4477c9de560dSAlex Tomas 		mutex_unlock(&ac->ac_lg->lg_mutex);
4478c9de560dSAlex Tomas 	ext4_mb_collect_stats(ac);
4479c9de560dSAlex Tomas 	return 0;
4480c9de560dSAlex Tomas }
4481c9de560dSAlex Tomas 
4482c9de560dSAlex Tomas static int ext4_mb_discard_preallocations(struct super_block *sb, int needed)
4483c9de560dSAlex Tomas {
4484c9de560dSAlex Tomas 	ext4_group_t i;
4485c9de560dSAlex Tomas 	int ret;
4486c9de560dSAlex Tomas 	int freed = 0;
4487c9de560dSAlex Tomas 
4488c9de560dSAlex Tomas 	for (i = 0; i < EXT4_SB(sb)->s_groups_count && needed > 0; i++) {
4489c9de560dSAlex Tomas 		ret = ext4_mb_discard_group_preallocations(sb, i, needed);
4490c9de560dSAlex Tomas 		freed += ret;
4491c9de560dSAlex Tomas 		needed -= ret;
4492c9de560dSAlex Tomas 	}
4493c9de560dSAlex Tomas 
4494c9de560dSAlex Tomas 	return freed;
4495c9de560dSAlex Tomas }
4496c9de560dSAlex Tomas 
4497c9de560dSAlex Tomas /*
4498c9de560dSAlex Tomas  * Main entry point into mballoc to allocate blocks
4499c9de560dSAlex Tomas  * it tries to use preallocation first, then falls back
4500c9de560dSAlex Tomas  * to usual allocation
4501c9de560dSAlex Tomas  */
4502c9de560dSAlex Tomas ext4_fsblk_t ext4_mb_new_blocks(handle_t *handle,
4503c9de560dSAlex Tomas 				 struct ext4_allocation_request *ar, int *errp)
4504c9de560dSAlex Tomas {
45056bc6e63fSAneesh Kumar K.V 	int freed;
4506256bdb49SEric Sandeen 	struct ext4_allocation_context *ac = NULL;
4507c9de560dSAlex Tomas 	struct ext4_sb_info *sbi;
4508c9de560dSAlex Tomas 	struct super_block *sb;
4509c9de560dSAlex Tomas 	ext4_fsblk_t block = 0;
4510498e5f24STheodore Ts'o 	unsigned int inquota;
4511498e5f24STheodore Ts'o 	unsigned int reserv_blks = 0;
4512c9de560dSAlex Tomas 
4513c9de560dSAlex Tomas 	sb = ar->inode->i_sb;
4514c9de560dSAlex Tomas 	sbi = EXT4_SB(sb);
4515c9de560dSAlex Tomas 
4516d2a17637SMingming Cao 	if (!EXT4_I(ar->inode)->i_delalloc_reserved_flag) {
4517d2a17637SMingming Cao 		/*
4518d2a17637SMingming Cao 		 * With delalloc we already reserved the blocks
4519d2a17637SMingming Cao 		 */
4520030ba6bcSAneesh Kumar K.V 		while (ar->len && ext4_claim_free_blocks(sbi, ar->len)) {
4521030ba6bcSAneesh Kumar K.V 			/* let others to free the space */
4522030ba6bcSAneesh Kumar K.V 			yield();
4523030ba6bcSAneesh Kumar K.V 			ar->len = ar->len >> 1;
4524030ba6bcSAneesh Kumar K.V 		}
4525030ba6bcSAneesh Kumar K.V 		if (!ar->len) {
452607031431SMingming Cao 			*errp = -ENOSPC;
452707031431SMingming Cao 			return 0;
452807031431SMingming Cao 		}
45296bc6e63fSAneesh Kumar K.V 		reserv_blks = ar->len;
4530a30d542aSAneesh Kumar K.V 	}
4531c9de560dSAlex Tomas 	while (ar->len && DQUOT_ALLOC_BLOCK(ar->inode, ar->len)) {
4532c9de560dSAlex Tomas 		ar->flags |= EXT4_MB_HINT_NOPREALLOC;
4533c9de560dSAlex Tomas 		ar->len--;
4534c9de560dSAlex Tomas 	}
4535c9de560dSAlex Tomas 	if (ar->len == 0) {
4536c9de560dSAlex Tomas 		*errp = -EDQUOT;
4537c9de560dSAlex Tomas 		return 0;
4538c9de560dSAlex Tomas 	}
4539c9de560dSAlex Tomas 	inquota = ar->len;
4540c9de560dSAlex Tomas 
4541d2a17637SMingming Cao 	if (EXT4_I(ar->inode)->i_delalloc_reserved_flag)
4542d2a17637SMingming Cao 		ar->flags |= EXT4_MB_DELALLOC_RESERVED;
4543d2a17637SMingming Cao 
4544256bdb49SEric Sandeen 	ac = kmem_cache_alloc(ext4_ac_cachep, GFP_NOFS);
4545256bdb49SEric Sandeen 	if (!ac) {
4546363d4251SShen Feng 		ar->len = 0;
4547256bdb49SEric Sandeen 		*errp = -ENOMEM;
4548363d4251SShen Feng 		goto out1;
4549256bdb49SEric Sandeen 	}
4550256bdb49SEric Sandeen 
4551256bdb49SEric Sandeen 	*errp = ext4_mb_initialize_context(ac, ar);
4552c9de560dSAlex Tomas 	if (*errp) {
4553c9de560dSAlex Tomas 		ar->len = 0;
4554363d4251SShen Feng 		goto out2;
4555c9de560dSAlex Tomas 	}
4556c9de560dSAlex Tomas 
4557256bdb49SEric Sandeen 	ac->ac_op = EXT4_MB_HISTORY_PREALLOC;
4558256bdb49SEric Sandeen 	if (!ext4_mb_use_preallocated(ac)) {
4559256bdb49SEric Sandeen 		ac->ac_op = EXT4_MB_HISTORY_ALLOC;
4560256bdb49SEric Sandeen 		ext4_mb_normalize_request(ac, ar);
4561c9de560dSAlex Tomas repeat:
4562c9de560dSAlex Tomas 		/* allocate space in core */
4563256bdb49SEric Sandeen 		ext4_mb_regular_allocator(ac);
4564c9de560dSAlex Tomas 
4565c9de560dSAlex Tomas 		/* as we've just preallocated more space than
4566c9de560dSAlex Tomas 		 * user requested orinally, we store allocated
4567c9de560dSAlex Tomas 		 * space in a special descriptor */
4568256bdb49SEric Sandeen 		if (ac->ac_status == AC_STATUS_FOUND &&
4569256bdb49SEric Sandeen 				ac->ac_o_ex.fe_len < ac->ac_b_ex.fe_len)
4570256bdb49SEric Sandeen 			ext4_mb_new_preallocation(ac);
4571c9de560dSAlex Tomas 	}
4572c9de560dSAlex Tomas 
4573256bdb49SEric Sandeen 	if (likely(ac->ac_status == AC_STATUS_FOUND)) {
45746bc6e63fSAneesh Kumar K.V 		*errp = ext4_mb_mark_diskspace_used(ac, handle, reserv_blks);
4575519deca0SAneesh Kumar K.V 		if (*errp ==  -EAGAIN) {
4576519deca0SAneesh Kumar K.V 			ac->ac_b_ex.fe_group = 0;
4577519deca0SAneesh Kumar K.V 			ac->ac_b_ex.fe_start = 0;
4578519deca0SAneesh Kumar K.V 			ac->ac_b_ex.fe_len = 0;
4579519deca0SAneesh Kumar K.V 			ac->ac_status = AC_STATUS_CONTINUE;
4580519deca0SAneesh Kumar K.V 			goto repeat;
4581519deca0SAneesh Kumar K.V 		} else if (*errp) {
4582519deca0SAneesh Kumar K.V 			ac->ac_b_ex.fe_len = 0;
4583519deca0SAneesh Kumar K.V 			ar->len = 0;
4584519deca0SAneesh Kumar K.V 			ext4_mb_show_ac(ac);
4585519deca0SAneesh Kumar K.V 		} else {
4586256bdb49SEric Sandeen 			block = ext4_grp_offs_to_block(sb, &ac->ac_b_ex);
4587256bdb49SEric Sandeen 			ar->len = ac->ac_b_ex.fe_len;
4588519deca0SAneesh Kumar K.V 		}
4589c9de560dSAlex Tomas 	} else {
4590256bdb49SEric Sandeen 		freed  = ext4_mb_discard_preallocations(sb, ac->ac_o_ex.fe_len);
4591c9de560dSAlex Tomas 		if (freed)
4592c9de560dSAlex Tomas 			goto repeat;
4593c9de560dSAlex Tomas 		*errp = -ENOSPC;
4594256bdb49SEric Sandeen 		ac->ac_b_ex.fe_len = 0;
4595c9de560dSAlex Tomas 		ar->len = 0;
4596256bdb49SEric Sandeen 		ext4_mb_show_ac(ac);
4597c9de560dSAlex Tomas 	}
4598c9de560dSAlex Tomas 
4599256bdb49SEric Sandeen 	ext4_mb_release_context(ac);
4600c9de560dSAlex Tomas 
4601363d4251SShen Feng out2:
4602363d4251SShen Feng 	kmem_cache_free(ext4_ac_cachep, ac);
4603363d4251SShen Feng out1:
4604c9de560dSAlex Tomas 	if (ar->len < inquota)
4605c9de560dSAlex Tomas 		DQUOT_FREE_BLOCK(ar->inode, inquota - ar->len);
4606c9de560dSAlex Tomas 
4607c9de560dSAlex Tomas 	return block;
4608c9de560dSAlex Tomas }
4609c9de560dSAlex Tomas 
4610c894058dSAneesh Kumar K.V /*
4611c894058dSAneesh Kumar K.V  * We can merge two free data extents only if the physical blocks
4612c894058dSAneesh Kumar K.V  * are contiguous, AND the extents were freed by the same transaction,
4613c894058dSAneesh Kumar K.V  * AND the blocks are associated with the same group.
4614c894058dSAneesh Kumar K.V  */
4615c894058dSAneesh Kumar K.V static int can_merge(struct ext4_free_data *entry1,
4616c894058dSAneesh Kumar K.V 			struct ext4_free_data *entry2)
4617c894058dSAneesh Kumar K.V {
4618c894058dSAneesh Kumar K.V 	if ((entry1->t_tid == entry2->t_tid) &&
4619c894058dSAneesh Kumar K.V 	    (entry1->group == entry2->group) &&
4620c894058dSAneesh Kumar K.V 	    ((entry1->start_blk + entry1->count) == entry2->start_blk))
4621c894058dSAneesh Kumar K.V 		return 1;
4622c894058dSAneesh Kumar K.V 	return 0;
4623c894058dSAneesh Kumar K.V }
4624c894058dSAneesh Kumar K.V 
46254ddfef7bSEric Sandeen static noinline_for_stack int
46264ddfef7bSEric Sandeen ext4_mb_free_metadata(handle_t *handle, struct ext4_buddy *e4b,
46277a2fcbf7SAneesh Kumar K.V 		      struct ext4_free_data *new_entry)
4628c9de560dSAlex Tomas {
46297a2fcbf7SAneesh Kumar K.V 	ext4_grpblk_t block;
46307a2fcbf7SAneesh Kumar K.V 	struct ext4_free_data *entry;
4631c9de560dSAlex Tomas 	struct ext4_group_info *db = e4b->bd_info;
4632c9de560dSAlex Tomas 	struct super_block *sb = e4b->bd_sb;
4633c9de560dSAlex Tomas 	struct ext4_sb_info *sbi = EXT4_SB(sb);
4634c894058dSAneesh Kumar K.V 	struct rb_node **n = &db->bb_free_root.rb_node, *node;
4635c894058dSAneesh Kumar K.V 	struct rb_node *parent = NULL, *new_node;
4636c894058dSAneesh Kumar K.V 
46370390131bSFrank Mayhar 	BUG_ON(!ext4_handle_valid(handle));
4638c9de560dSAlex Tomas 	BUG_ON(e4b->bd_bitmap_page == NULL);
4639c9de560dSAlex Tomas 	BUG_ON(e4b->bd_buddy_page == NULL);
4640c9de560dSAlex Tomas 
4641c894058dSAneesh Kumar K.V 	new_node = &new_entry->node;
46427a2fcbf7SAneesh Kumar K.V 	block = new_entry->start_blk;
4643c9de560dSAlex Tomas 
4644c894058dSAneesh Kumar K.V 	if (!*n) {
4645c894058dSAneesh Kumar K.V 		/* first free block exent. We need to
4646c894058dSAneesh Kumar K.V 		   protect buddy cache from being freed,
4647c9de560dSAlex Tomas 		 * otherwise we'll refresh it from
4648c9de560dSAlex Tomas 		 * on-disk bitmap and lose not-yet-available
4649c9de560dSAlex Tomas 		 * blocks */
4650c9de560dSAlex Tomas 		page_cache_get(e4b->bd_buddy_page);
4651c9de560dSAlex Tomas 		page_cache_get(e4b->bd_bitmap_page);
4652c894058dSAneesh Kumar K.V 	}
4653c894058dSAneesh Kumar K.V 	while (*n) {
4654c894058dSAneesh Kumar K.V 		parent = *n;
4655c894058dSAneesh Kumar K.V 		entry = rb_entry(parent, struct ext4_free_data, node);
4656c894058dSAneesh Kumar K.V 		if (block < entry->start_blk)
4657c894058dSAneesh Kumar K.V 			n = &(*n)->rb_left;
4658c894058dSAneesh Kumar K.V 		else if (block >= (entry->start_blk + entry->count))
4659c894058dSAneesh Kumar K.V 			n = &(*n)->rb_right;
4660c894058dSAneesh Kumar K.V 		else {
46615d1b1b3fSAneesh Kumar K.V 			ext4_grp_locked_error(sb, e4b->bd_group, __func__,
4662fde4d95aSTheodore Ts'o 					"Double free of blocks %d (%d %d)",
4663c894058dSAneesh Kumar K.V 					block, entry->start_blk, entry->count);
4664c894058dSAneesh Kumar K.V 			return 0;
4665c9de560dSAlex Tomas 		}
4666c9de560dSAlex Tomas 	}
4667c9de560dSAlex Tomas 
4668c894058dSAneesh Kumar K.V 	rb_link_node(new_node, parent, n);
4669c894058dSAneesh Kumar K.V 	rb_insert_color(new_node, &db->bb_free_root);
4670c894058dSAneesh Kumar K.V 
4671c894058dSAneesh Kumar K.V 	/* Now try to see the extent can be merged to left and right */
4672c894058dSAneesh Kumar K.V 	node = rb_prev(new_node);
4673c894058dSAneesh Kumar K.V 	if (node) {
4674c894058dSAneesh Kumar K.V 		entry = rb_entry(node, struct ext4_free_data, node);
4675c894058dSAneesh Kumar K.V 		if (can_merge(entry, new_entry)) {
4676c894058dSAneesh Kumar K.V 			new_entry->start_blk = entry->start_blk;
4677c894058dSAneesh Kumar K.V 			new_entry->count += entry->count;
4678c894058dSAneesh Kumar K.V 			rb_erase(node, &(db->bb_free_root));
4679c894058dSAneesh Kumar K.V 			spin_lock(&sbi->s_md_lock);
4680c894058dSAneesh Kumar K.V 			list_del(&entry->list);
4681c894058dSAneesh Kumar K.V 			spin_unlock(&sbi->s_md_lock);
4682c894058dSAneesh Kumar K.V 			kmem_cache_free(ext4_free_ext_cachep, entry);
4683c9de560dSAlex Tomas 		}
4684c9de560dSAlex Tomas 	}
4685c894058dSAneesh Kumar K.V 
4686c894058dSAneesh Kumar K.V 	node = rb_next(new_node);
4687c894058dSAneesh Kumar K.V 	if (node) {
4688c894058dSAneesh Kumar K.V 		entry = rb_entry(node, struct ext4_free_data, node);
4689c894058dSAneesh Kumar K.V 		if (can_merge(new_entry, entry)) {
4690c894058dSAneesh Kumar K.V 			new_entry->count += entry->count;
4691c894058dSAneesh Kumar K.V 			rb_erase(node, &(db->bb_free_root));
4692c894058dSAneesh Kumar K.V 			spin_lock(&sbi->s_md_lock);
4693c894058dSAneesh Kumar K.V 			list_del(&entry->list);
4694c894058dSAneesh Kumar K.V 			spin_unlock(&sbi->s_md_lock);
4695c894058dSAneesh Kumar K.V 			kmem_cache_free(ext4_free_ext_cachep, entry);
4696c894058dSAneesh Kumar K.V 		}
4697c894058dSAneesh Kumar K.V 	}
46983e624fc7STheodore Ts'o 	/* Add the extent to transaction's private list */
4699c894058dSAneesh Kumar K.V 	spin_lock(&sbi->s_md_lock);
47003e624fc7STheodore Ts'o 	list_add(&new_entry->list, &handle->h_transaction->t_private_list);
4701c894058dSAneesh Kumar K.V 	spin_unlock(&sbi->s_md_lock);
4702c9de560dSAlex Tomas 	return 0;
4703c9de560dSAlex Tomas }
4704c9de560dSAlex Tomas 
4705c9de560dSAlex Tomas /*
4706c9de560dSAlex Tomas  * Main entry point into mballoc to free blocks
4707c9de560dSAlex Tomas  */
4708c9de560dSAlex Tomas void ext4_mb_free_blocks(handle_t *handle, struct inode *inode,
4709c9de560dSAlex Tomas 			unsigned long block, unsigned long count,
4710c9de560dSAlex Tomas 			int metadata, unsigned long *freed)
4711c9de560dSAlex Tomas {
471226346ff6SAneesh Kumar K.V 	struct buffer_head *bitmap_bh = NULL;
4713c9de560dSAlex Tomas 	struct super_block *sb = inode->i_sb;
4714256bdb49SEric Sandeen 	struct ext4_allocation_context *ac = NULL;
4715c9de560dSAlex Tomas 	struct ext4_group_desc *gdp;
4716c9de560dSAlex Tomas 	struct ext4_super_block *es;
4717498e5f24STheodore Ts'o 	unsigned int overflow;
4718c9de560dSAlex Tomas 	ext4_grpblk_t bit;
4719c9de560dSAlex Tomas 	struct buffer_head *gd_bh;
4720c9de560dSAlex Tomas 	ext4_group_t block_group;
4721c9de560dSAlex Tomas 	struct ext4_sb_info *sbi;
4722c9de560dSAlex Tomas 	struct ext4_buddy e4b;
4723c9de560dSAlex Tomas 	int err = 0;
4724c9de560dSAlex Tomas 	int ret;
4725c9de560dSAlex Tomas 
4726c9de560dSAlex Tomas 	*freed = 0;
4727c9de560dSAlex Tomas 
4728c9de560dSAlex Tomas 	sbi = EXT4_SB(sb);
4729c9de560dSAlex Tomas 	es = EXT4_SB(sb)->s_es;
4730c9de560dSAlex Tomas 	if (block < le32_to_cpu(es->s_first_data_block) ||
4731c9de560dSAlex Tomas 	    block + count < block ||
4732c9de560dSAlex Tomas 	    block + count > ext4_blocks_count(es)) {
473346e665e9SHarvey Harrison 		ext4_error(sb, __func__,
4734c9de560dSAlex Tomas 			    "Freeing blocks not in datazone - "
4735c9de560dSAlex Tomas 			    "block = %lu, count = %lu", block, count);
4736c9de560dSAlex Tomas 		goto error_return;
4737c9de560dSAlex Tomas 	}
4738c9de560dSAlex Tomas 
4739c9de560dSAlex Tomas 	ext4_debug("freeing block %lu\n", block);
4740c9de560dSAlex Tomas 
4741256bdb49SEric Sandeen 	ac = kmem_cache_alloc(ext4_ac_cachep, GFP_NOFS);
4742256bdb49SEric Sandeen 	if (ac) {
4743256bdb49SEric Sandeen 		ac->ac_op = EXT4_MB_HISTORY_FREE;
4744256bdb49SEric Sandeen 		ac->ac_inode = inode;
4745256bdb49SEric Sandeen 		ac->ac_sb = sb;
4746256bdb49SEric Sandeen 	}
4747c9de560dSAlex Tomas 
4748c9de560dSAlex Tomas do_more:
4749c9de560dSAlex Tomas 	overflow = 0;
4750c9de560dSAlex Tomas 	ext4_get_group_no_and_offset(sb, block, &block_group, &bit);
4751c9de560dSAlex Tomas 
4752c9de560dSAlex Tomas 	/*
4753c9de560dSAlex Tomas 	 * Check to see if we are freeing blocks across a group
4754c9de560dSAlex Tomas 	 * boundary.
4755c9de560dSAlex Tomas 	 */
4756c9de560dSAlex Tomas 	if (bit + count > EXT4_BLOCKS_PER_GROUP(sb)) {
4757c9de560dSAlex Tomas 		overflow = bit + count - EXT4_BLOCKS_PER_GROUP(sb);
4758c9de560dSAlex Tomas 		count -= overflow;
4759c9de560dSAlex Tomas 	}
4760574ca174STheodore Ts'o 	bitmap_bh = ext4_read_block_bitmap(sb, block_group);
4761ce89f46cSAneesh Kumar K.V 	if (!bitmap_bh) {
4762ce89f46cSAneesh Kumar K.V 		err = -EIO;
4763c9de560dSAlex Tomas 		goto error_return;
4764ce89f46cSAneesh Kumar K.V 	}
4765c9de560dSAlex Tomas 	gdp = ext4_get_group_desc(sb, block_group, &gd_bh);
4766ce89f46cSAneesh Kumar K.V 	if (!gdp) {
4767ce89f46cSAneesh Kumar K.V 		err = -EIO;
4768c9de560dSAlex Tomas 		goto error_return;
4769ce89f46cSAneesh Kumar K.V 	}
4770c9de560dSAlex Tomas 
4771c9de560dSAlex Tomas 	if (in_range(ext4_block_bitmap(sb, gdp), block, count) ||
4772c9de560dSAlex Tomas 	    in_range(ext4_inode_bitmap(sb, gdp), block, count) ||
4773c9de560dSAlex Tomas 	    in_range(block, ext4_inode_table(sb, gdp),
4774c9de560dSAlex Tomas 		      EXT4_SB(sb)->s_itb_per_group) ||
4775c9de560dSAlex Tomas 	    in_range(block + count - 1, ext4_inode_table(sb, gdp),
4776c9de560dSAlex Tomas 		      EXT4_SB(sb)->s_itb_per_group)) {
4777c9de560dSAlex Tomas 
477846e665e9SHarvey Harrison 		ext4_error(sb, __func__,
4779c9de560dSAlex Tomas 			   "Freeing blocks in system zone - "
4780c9de560dSAlex Tomas 			   "Block = %lu, count = %lu", block, count);
4781519deca0SAneesh Kumar K.V 		/* err = 0. ext4_std_error should be a no op */
4782519deca0SAneesh Kumar K.V 		goto error_return;
4783c9de560dSAlex Tomas 	}
4784c9de560dSAlex Tomas 
4785c9de560dSAlex Tomas 	BUFFER_TRACE(bitmap_bh, "getting write access");
4786c9de560dSAlex Tomas 	err = ext4_journal_get_write_access(handle, bitmap_bh);
4787c9de560dSAlex Tomas 	if (err)
4788c9de560dSAlex Tomas 		goto error_return;
4789c9de560dSAlex Tomas 
4790c9de560dSAlex Tomas 	/*
4791c9de560dSAlex Tomas 	 * We are about to modify some metadata.  Call the journal APIs
4792c9de560dSAlex Tomas 	 * to unshare ->b_data if a currently-committing transaction is
4793c9de560dSAlex Tomas 	 * using it
4794c9de560dSAlex Tomas 	 */
4795c9de560dSAlex Tomas 	BUFFER_TRACE(gd_bh, "get_write_access");
4796c9de560dSAlex Tomas 	err = ext4_journal_get_write_access(handle, gd_bh);
4797c9de560dSAlex Tomas 	if (err)
4798c9de560dSAlex Tomas 		goto error_return;
4799c9de560dSAlex Tomas #ifdef AGGRESSIVE_CHECK
4800c9de560dSAlex Tomas 	{
4801c9de560dSAlex Tomas 		int i;
4802c9de560dSAlex Tomas 		for (i = 0; i < count; i++)
4803c9de560dSAlex Tomas 			BUG_ON(!mb_test_bit(bit + i, bitmap_bh->b_data));
4804c9de560dSAlex Tomas 	}
4805c9de560dSAlex Tomas #endif
4806256bdb49SEric Sandeen 	if (ac) {
4807256bdb49SEric Sandeen 		ac->ac_b_ex.fe_group = block_group;
4808256bdb49SEric Sandeen 		ac->ac_b_ex.fe_start = bit;
4809256bdb49SEric Sandeen 		ac->ac_b_ex.fe_len = count;
4810256bdb49SEric Sandeen 		ext4_mb_store_history(ac);
4811256bdb49SEric Sandeen 	}
4812c9de560dSAlex Tomas 
4813920313a7SAneesh Kumar K.V 	err = ext4_mb_load_buddy(sb, block_group, &e4b);
4814920313a7SAneesh Kumar K.V 	if (err)
4815920313a7SAneesh Kumar K.V 		goto error_return;
48160390131bSFrank Mayhar 	if (metadata && ext4_handle_valid(handle)) {
48177a2fcbf7SAneesh Kumar K.V 		struct ext4_free_data *new_entry;
48187a2fcbf7SAneesh Kumar K.V 		/*
48197a2fcbf7SAneesh Kumar K.V 		 * blocks being freed are metadata. these blocks shouldn't
48207a2fcbf7SAneesh Kumar K.V 		 * be used until this transaction is committed
48217a2fcbf7SAneesh Kumar K.V 		 */
48227a2fcbf7SAneesh Kumar K.V 		new_entry  = kmem_cache_alloc(ext4_free_ext_cachep, GFP_NOFS);
48237a2fcbf7SAneesh Kumar K.V 		new_entry->start_blk = bit;
48247a2fcbf7SAneesh Kumar K.V 		new_entry->group  = block_group;
48257a2fcbf7SAneesh Kumar K.V 		new_entry->count = count;
48267a2fcbf7SAneesh Kumar K.V 		new_entry->t_tid = handle->h_transaction->t_tid;
48277a2fcbf7SAneesh Kumar K.V 		ext4_lock_group(sb, block_group);
48287a2fcbf7SAneesh Kumar K.V 		mb_clear_bits(sb_bgl_lock(sbi, block_group), bitmap_bh->b_data,
48297a2fcbf7SAneesh Kumar K.V 				bit, count);
48307a2fcbf7SAneesh Kumar K.V 		ext4_mb_free_metadata(handle, &e4b, new_entry);
48317a2fcbf7SAneesh Kumar K.V 		ext4_unlock_group(sb, block_group);
4832c9de560dSAlex Tomas 	} else {
4833c9de560dSAlex Tomas 		ext4_lock_group(sb, block_group);
48347a2fcbf7SAneesh Kumar K.V 		/* need to update group_info->bb_free and bitmap
48357a2fcbf7SAneesh Kumar K.V 		 * with group lock held. generate_buddy look at
48367a2fcbf7SAneesh Kumar K.V 		 * them with group lock_held
48377a2fcbf7SAneesh Kumar K.V 		 */
48387a2fcbf7SAneesh Kumar K.V 		mb_clear_bits(sb_bgl_lock(sbi, block_group), bitmap_bh->b_data,
48397a2fcbf7SAneesh Kumar K.V 				bit, count);
48407e5a8cddSShen Feng 		mb_free_blocks(inode, &e4b, bit, count);
4841c9de560dSAlex Tomas 		ext4_mb_return_to_preallocation(inode, &e4b, block, count);
4842c9de560dSAlex Tomas 		ext4_unlock_group(sb, block_group);
4843c9de560dSAlex Tomas 	}
4844c9de560dSAlex Tomas 
4845c9de560dSAlex Tomas 	spin_lock(sb_bgl_lock(sbi, block_group));
4846560671a0SAneesh Kumar K.V 	ret = ext4_free_blks_count(sb, gdp) + count;
4847560671a0SAneesh Kumar K.V 	ext4_free_blks_set(sb, gdp, ret);
4848c9de560dSAlex Tomas 	gdp->bg_checksum = ext4_group_desc_csum(sbi, block_group, gdp);
4849c9de560dSAlex Tomas 	spin_unlock(sb_bgl_lock(sbi, block_group));
4850c9de560dSAlex Tomas 	percpu_counter_add(&sbi->s_freeblocks_counter, count);
4851c9de560dSAlex Tomas 
4852772cb7c8SJose R. Santos 	if (sbi->s_log_groups_per_flex) {
4853772cb7c8SJose R. Santos 		ext4_group_t flex_group = ext4_flex_group(sbi, block_group);
4854772cb7c8SJose R. Santos 		spin_lock(sb_bgl_lock(sbi, flex_group));
4855772cb7c8SJose R. Santos 		sbi->s_flex_groups[flex_group].free_blocks += count;
4856772cb7c8SJose R. Santos 		spin_unlock(sb_bgl_lock(sbi, flex_group));
4857772cb7c8SJose R. Santos 	}
4858772cb7c8SJose R. Santos 
4859c9de560dSAlex Tomas 	ext4_mb_release_desc(&e4b);
4860c9de560dSAlex Tomas 
4861c9de560dSAlex Tomas 	*freed += count;
4862c9de560dSAlex Tomas 
48637a2fcbf7SAneesh Kumar K.V 	/* We dirtied the bitmap block */
48647a2fcbf7SAneesh Kumar K.V 	BUFFER_TRACE(bitmap_bh, "dirtied bitmap block");
48657a2fcbf7SAneesh Kumar K.V 	err = ext4_handle_dirty_metadata(handle, NULL, bitmap_bh);
48667a2fcbf7SAneesh Kumar K.V 
4867c9de560dSAlex Tomas 	/* And the group descriptor block */
4868c9de560dSAlex Tomas 	BUFFER_TRACE(gd_bh, "dirtied group descriptor block");
48690390131bSFrank Mayhar 	ret = ext4_handle_dirty_metadata(handle, NULL, gd_bh);
4870c9de560dSAlex Tomas 	if (!err)
4871c9de560dSAlex Tomas 		err = ret;
4872c9de560dSAlex Tomas 
4873c9de560dSAlex Tomas 	if (overflow && !err) {
4874c9de560dSAlex Tomas 		block += count;
4875c9de560dSAlex Tomas 		count = overflow;
4876c9de560dSAlex Tomas 		put_bh(bitmap_bh);
4877c9de560dSAlex Tomas 		goto do_more;
4878c9de560dSAlex Tomas 	}
4879c9de560dSAlex Tomas 	sb->s_dirt = 1;
4880c9de560dSAlex Tomas error_return:
4881c9de560dSAlex Tomas 	brelse(bitmap_bh);
4882c9de560dSAlex Tomas 	ext4_std_error(sb, err);
4883256bdb49SEric Sandeen 	if (ac)
4884256bdb49SEric Sandeen 		kmem_cache_free(ext4_ac_cachep, ac);
4885c9de560dSAlex Tomas 	return;
4886c9de560dSAlex Tomas }
4887