Linux Audio

Check our new training course

Loading...
v5.4
  1// SPDX-License-Identifier: GPL-2.0
  2
  3#include "misc.h"
  4#include "ctree.h"
  5#include "block-rsv.h"
  6#include "space-info.h"
  7#include "transaction.h"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  8
  9static u64 block_rsv_release_bytes(struct btrfs_fs_info *fs_info,
 10				    struct btrfs_block_rsv *block_rsv,
 11				    struct btrfs_block_rsv *dest, u64 num_bytes,
 12				    u64 *qgroup_to_release_ret)
 13{
 14	struct btrfs_space_info *space_info = block_rsv->space_info;
 15	u64 qgroup_to_release = 0;
 16	u64 ret;
 17
 18	spin_lock(&block_rsv->lock);
 19	if (num_bytes == (u64)-1) {
 20		num_bytes = block_rsv->size;
 21		qgroup_to_release = block_rsv->qgroup_rsv_size;
 22	}
 23	block_rsv->size -= num_bytes;
 24	if (block_rsv->reserved >= block_rsv->size) {
 25		num_bytes = block_rsv->reserved - block_rsv->size;
 26		block_rsv->reserved = block_rsv->size;
 27		block_rsv->full = 1;
 28	} else {
 29		num_bytes = 0;
 30	}
 31	if (block_rsv->qgroup_rsv_reserved >= block_rsv->qgroup_rsv_size) {
 
 32		qgroup_to_release = block_rsv->qgroup_rsv_reserved -
 33				    block_rsv->qgroup_rsv_size;
 34		block_rsv->qgroup_rsv_reserved = block_rsv->qgroup_rsv_size;
 35	} else {
 36		qgroup_to_release = 0;
 37	}
 38	spin_unlock(&block_rsv->lock);
 39
 40	ret = num_bytes;
 41	if (num_bytes > 0) {
 42		if (dest) {
 43			spin_lock(&dest->lock);
 44			if (!dest->full) {
 45				u64 bytes_to_add;
 46
 47				bytes_to_add = dest->size - dest->reserved;
 48				bytes_to_add = min(num_bytes, bytes_to_add);
 49				dest->reserved += bytes_to_add;
 50				if (dest->reserved >= dest->size)
 51					dest->full = 1;
 52				num_bytes -= bytes_to_add;
 53			}
 54			spin_unlock(&dest->lock);
 55		}
 56		if (num_bytes)
 57			btrfs_space_info_free_bytes_may_use(fs_info,
 58							    space_info,
 59							    num_bytes);
 60	}
 61	if (qgroup_to_release_ret)
 62		*qgroup_to_release_ret = qgroup_to_release;
 63	return ret;
 64}
 65
 66int btrfs_block_rsv_migrate(struct btrfs_block_rsv *src,
 67			    struct btrfs_block_rsv *dst, u64 num_bytes,
 68			    bool update_size)
 69{
 70	int ret;
 71
 72	ret = btrfs_block_rsv_use_bytes(src, num_bytes);
 73	if (ret)
 74		return ret;
 75
 76	btrfs_block_rsv_add_bytes(dst, num_bytes, update_size);
 77	return 0;
 78}
 79
 80void btrfs_init_block_rsv(struct btrfs_block_rsv *rsv, unsigned short type)
 81{
 82	memset(rsv, 0, sizeof(*rsv));
 83	spin_lock_init(&rsv->lock);
 84	rsv->type = type;
 85}
 86
 87void btrfs_init_metadata_block_rsv(struct btrfs_fs_info *fs_info,
 88				   struct btrfs_block_rsv *rsv,
 89				   unsigned short type)
 90{
 91	btrfs_init_block_rsv(rsv, type);
 92	rsv->space_info = btrfs_find_space_info(fs_info,
 93					    BTRFS_BLOCK_GROUP_METADATA);
 94}
 95
 96struct btrfs_block_rsv *btrfs_alloc_block_rsv(struct btrfs_fs_info *fs_info,
 97					      unsigned short type)
 98{
 99	struct btrfs_block_rsv *block_rsv;
100
101	block_rsv = kmalloc(sizeof(*block_rsv), GFP_NOFS);
102	if (!block_rsv)
103		return NULL;
104
105	btrfs_init_metadata_block_rsv(fs_info, block_rsv, type);
106	return block_rsv;
107}
108
109void btrfs_free_block_rsv(struct btrfs_fs_info *fs_info,
110			  struct btrfs_block_rsv *rsv)
111{
112	if (!rsv)
113		return;
114	btrfs_block_rsv_release(fs_info, rsv, (u64)-1);
115	kfree(rsv);
116}
117
118int btrfs_block_rsv_add(struct btrfs_root *root,
119			struct btrfs_block_rsv *block_rsv, u64 num_bytes,
120			enum btrfs_reserve_flush_enum flush)
121{
122	int ret;
123
124	if (num_bytes == 0)
125		return 0;
126
127	ret = btrfs_reserve_metadata_bytes(root, block_rsv, num_bytes, flush);
 
128	if (!ret)
129		btrfs_block_rsv_add_bytes(block_rsv, num_bytes, true);
130
131	return ret;
132}
133
134int btrfs_block_rsv_check(struct btrfs_block_rsv *block_rsv, int min_factor)
135{
136	u64 num_bytes = 0;
137	int ret = -ENOSPC;
138
139	if (!block_rsv)
140		return 0;
141
142	spin_lock(&block_rsv->lock);
143	num_bytes = div_factor(block_rsv->size, min_factor);
144	if (block_rsv->reserved >= num_bytes)
145		ret = 0;
146	spin_unlock(&block_rsv->lock);
147
148	return ret;
149}
150
151int btrfs_block_rsv_refill(struct btrfs_root *root,
152			   struct btrfs_block_rsv *block_rsv, u64 min_reserved,
153			   enum btrfs_reserve_flush_enum flush)
154{
155	u64 num_bytes = 0;
156	int ret = -ENOSPC;
157
158	if (!block_rsv)
159		return 0;
160
161	spin_lock(&block_rsv->lock);
162	num_bytes = min_reserved;
163	if (block_rsv->reserved >= num_bytes)
164		ret = 0;
165	else
166		num_bytes -= block_rsv->reserved;
167	spin_unlock(&block_rsv->lock);
168
169	if (!ret)
170		return 0;
171
172	ret = btrfs_reserve_metadata_bytes(root, block_rsv, num_bytes, flush);
 
173	if (!ret) {
174		btrfs_block_rsv_add_bytes(block_rsv, num_bytes, false);
175		return 0;
176	}
177
178	return ret;
179}
180
181u64 __btrfs_block_rsv_release(struct btrfs_fs_info *fs_info,
182			      struct btrfs_block_rsv *block_rsv,
183			      u64 num_bytes, u64 *qgroup_to_release)
184{
185	struct btrfs_block_rsv *global_rsv = &fs_info->global_block_rsv;
186	struct btrfs_block_rsv *delayed_rsv = &fs_info->delayed_refs_rsv;
187	struct btrfs_block_rsv *target = NULL;
188
189	/*
190	 * If we are the delayed_rsv then push to the global rsv, otherwise dump
191	 * into the delayed rsv if it is not full.
192	 */
193	if (block_rsv == delayed_rsv)
194		target = global_rsv;
195	else if (block_rsv != global_rsv && !delayed_rsv->full)
196		target = delayed_rsv;
197
198	if (target && block_rsv->space_info != target->space_info)
199		target = NULL;
200
201	return block_rsv_release_bytes(fs_info, block_rsv, target, num_bytes,
202				       qgroup_to_release);
203}
204
205int btrfs_block_rsv_use_bytes(struct btrfs_block_rsv *block_rsv, u64 num_bytes)
206{
207	int ret = -ENOSPC;
208
209	spin_lock(&block_rsv->lock);
210	if (block_rsv->reserved >= num_bytes) {
211		block_rsv->reserved -= num_bytes;
212		if (block_rsv->reserved < block_rsv->size)
213			block_rsv->full = 0;
214		ret = 0;
215	}
216	spin_unlock(&block_rsv->lock);
217	return ret;
218}
219
220void btrfs_block_rsv_add_bytes(struct btrfs_block_rsv *block_rsv,
221			       u64 num_bytes, bool update_size)
222{
223	spin_lock(&block_rsv->lock);
224	block_rsv->reserved += num_bytes;
225	if (update_size)
226		block_rsv->size += num_bytes;
227	else if (block_rsv->reserved >= block_rsv->size)
228		block_rsv->full = 1;
229	spin_unlock(&block_rsv->lock);
230}
231
232int btrfs_cond_migrate_bytes(struct btrfs_fs_info *fs_info,
233			     struct btrfs_block_rsv *dest, u64 num_bytes,
234			     int min_factor)
235{
236	struct btrfs_block_rsv *global_rsv = &fs_info->global_block_rsv;
237	u64 min_bytes;
238
239	if (global_rsv->space_info != dest->space_info)
240		return -ENOSPC;
241
242	spin_lock(&global_rsv->lock);
243	min_bytes = div_factor(global_rsv->size, min_factor);
244	if (global_rsv->reserved < min_bytes + num_bytes) {
245		spin_unlock(&global_rsv->lock);
246		return -ENOSPC;
247	}
248	global_rsv->reserved -= num_bytes;
249	if (global_rsv->reserved < global_rsv->size)
250		global_rsv->full = 0;
251	spin_unlock(&global_rsv->lock);
252
253	btrfs_block_rsv_add_bytes(dest, num_bytes, true);
254	return 0;
255}
256
257void btrfs_update_global_block_rsv(struct btrfs_fs_info *fs_info)
258{
259	struct btrfs_block_rsv *block_rsv = &fs_info->global_block_rsv;
260	struct btrfs_space_info *sinfo = block_rsv->space_info;
261	u64 num_bytes;
262	unsigned min_items;
 
263
264	/*
265	 * The global block rsv is based on the size of the extent tree, the
266	 * checksum tree and the root tree.  If the fs is empty we want to set
267	 * it to a minimal amount for safety.
 
 
 
268	 */
269	num_bytes = btrfs_root_used(&fs_info->extent_root->root_item) +
270		btrfs_root_used(&fs_info->csum_root->root_item) +
271		btrfs_root_used(&fs_info->tree_root->root_item);
 
 
 
 
 
 
 
 
272
273	/*
274	 * We at a minimum are going to modify the csum root, the tree root, and
275	 * the extent root.
276	 */
277	min_items = 3;
 
 
 
 
278
279	/*
280	 * But we also want to reserve enough space so we can do the fallback
281	 * global reserve for an unlink, which is an additional 5 items (see the
282	 * comment in __unlink_start_trans for what we're modifying.)
283	 *
284	 * But we also need space for the delayed ref updates from the unlink,
285	 * so its 10, 5 for the actual operation, and 5 for the delayed ref
286	 * updates.
287	 */
288	min_items += 10;
289
290	num_bytes = max_t(u64, num_bytes,
291			  btrfs_calc_insert_metadata_size(fs_info, min_items));
 
 
292
293	spin_lock(&sinfo->lock);
294	spin_lock(&block_rsv->lock);
295
296	block_rsv->size = min_t(u64, num_bytes, SZ_512M);
297
298	if (block_rsv->reserved < block_rsv->size) {
299		num_bytes = block_rsv->size - block_rsv->reserved;
300		block_rsv->reserved += num_bytes;
301		btrfs_space_info_update_bytes_may_use(fs_info, sinfo,
302						      num_bytes);
 
303	} else if (block_rsv->reserved > block_rsv->size) {
304		num_bytes = block_rsv->reserved - block_rsv->size;
305		btrfs_space_info_update_bytes_may_use(fs_info, sinfo,
306						      -num_bytes);
307		block_rsv->reserved = block_rsv->size;
308		btrfs_try_granting_tickets(fs_info, sinfo);
309	}
310
311	if (block_rsv->reserved == block_rsv->size)
312		block_rsv->full = 1;
313	else
314		block_rsv->full = 0;
315
 
 
316	spin_unlock(&block_rsv->lock);
317	spin_unlock(&sinfo->lock);
318}
319
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
320void btrfs_init_global_block_rsv(struct btrfs_fs_info *fs_info)
321{
322	struct btrfs_space_info *space_info;
323
324	space_info = btrfs_find_space_info(fs_info, BTRFS_BLOCK_GROUP_SYSTEM);
325	fs_info->chunk_block_rsv.space_info = space_info;
326
327	space_info = btrfs_find_space_info(fs_info, BTRFS_BLOCK_GROUP_METADATA);
328	fs_info->global_block_rsv.space_info = space_info;
329	fs_info->trans_block_rsv.space_info = space_info;
330	fs_info->empty_block_rsv.space_info = space_info;
331	fs_info->delayed_block_rsv.space_info = space_info;
332	fs_info->delayed_refs_rsv.space_info = space_info;
333
334	fs_info->extent_root->block_rsv = &fs_info->delayed_refs_rsv;
335	fs_info->csum_root->block_rsv = &fs_info->delayed_refs_rsv;
336	fs_info->dev_root->block_rsv = &fs_info->global_block_rsv;
337	fs_info->tree_root->block_rsv = &fs_info->global_block_rsv;
338	if (fs_info->quota_root)
339		fs_info->quota_root->block_rsv = &fs_info->global_block_rsv;
340	fs_info->chunk_root->block_rsv = &fs_info->chunk_block_rsv;
341
342	btrfs_update_global_block_rsv(fs_info);
343}
344
345void btrfs_release_global_block_rsv(struct btrfs_fs_info *fs_info)
346{
347	btrfs_block_rsv_release(fs_info, &fs_info->global_block_rsv, (u64)-1);
 
348	WARN_ON(fs_info->trans_block_rsv.size > 0);
349	WARN_ON(fs_info->trans_block_rsv.reserved > 0);
350	WARN_ON(fs_info->chunk_block_rsv.size > 0);
351	WARN_ON(fs_info->chunk_block_rsv.reserved > 0);
352	WARN_ON(fs_info->delayed_block_rsv.size > 0);
353	WARN_ON(fs_info->delayed_block_rsv.reserved > 0);
354	WARN_ON(fs_info->delayed_refs_rsv.reserved > 0);
355	WARN_ON(fs_info->delayed_refs_rsv.size > 0);
356}
357
358static struct btrfs_block_rsv *get_block_rsv(
359					const struct btrfs_trans_handle *trans,
360					const struct btrfs_root *root)
361{
362	struct btrfs_fs_info *fs_info = root->fs_info;
363	struct btrfs_block_rsv *block_rsv = NULL;
364
365	if (test_bit(BTRFS_ROOT_REF_COWS, &root->state) ||
366	    (root == fs_info->csum_root && trans->adding_csums) ||
367	    (root == fs_info->uuid_root))
 
368		block_rsv = trans->block_rsv;
369
370	if (!block_rsv)
371		block_rsv = root->block_rsv;
372
373	if (!block_rsv)
374		block_rsv = &fs_info->empty_block_rsv;
375
376	return block_rsv;
377}
378
379struct btrfs_block_rsv *btrfs_use_block_rsv(struct btrfs_trans_handle *trans,
380					    struct btrfs_root *root,
381					    u32 blocksize)
382{
383	struct btrfs_fs_info *fs_info = root->fs_info;
384	struct btrfs_block_rsv *block_rsv;
385	struct btrfs_block_rsv *global_rsv = &fs_info->global_block_rsv;
386	int ret;
387	bool global_updated = false;
388
389	block_rsv = get_block_rsv(trans, root);
390
391	if (unlikely(block_rsv->size == 0))
392		goto try_reserve;
393again:
394	ret = btrfs_block_rsv_use_bytes(block_rsv, blocksize);
395	if (!ret)
396		return block_rsv;
397
398	if (block_rsv->failfast)
399		return ERR_PTR(ret);
400
401	if (block_rsv->type == BTRFS_BLOCK_RSV_GLOBAL && !global_updated) {
402		global_updated = true;
403		btrfs_update_global_block_rsv(fs_info);
404		goto again;
405	}
406
407	/*
408	 * The global reserve still exists to save us from ourselves, so don't
409	 * warn_on if we are short on our delayed refs reserve.
410	 */
411	if (block_rsv->type != BTRFS_BLOCK_RSV_DELREFS &&
412	    btrfs_test_opt(fs_info, ENOSPC_DEBUG)) {
413		static DEFINE_RATELIMIT_STATE(_rs,
414				DEFAULT_RATELIMIT_INTERVAL * 10,
415				/*DEFAULT_RATELIMIT_BURST*/ 1);
416		if (__ratelimit(&_rs))
417			WARN(1, KERN_DEBUG
418				"BTRFS: block rsv returned %d\n", ret);
 
419	}
420try_reserve:
421	ret = btrfs_reserve_metadata_bytes(root, block_rsv, blocksize,
422					   BTRFS_RESERVE_NO_FLUSH);
423	if (!ret)
424		return block_rsv;
425	/*
426	 * If we couldn't reserve metadata bytes try and use some from
427	 * the global reserve if its space type is the same as the global
428	 * reservation.
429	 */
430	if (block_rsv->type != BTRFS_BLOCK_RSV_GLOBAL &&
431	    block_rsv->space_info == global_rsv->space_info) {
432		ret = btrfs_block_rsv_use_bytes(global_rsv, blocksize);
433		if (!ret)
434			return global_rsv;
435	}
 
 
 
 
 
 
 
 
 
 
 
 
436	return ERR_PTR(ret);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
437}
v6.8
  1// SPDX-License-Identifier: GPL-2.0
  2
  3#include "misc.h"
  4#include "ctree.h"
  5#include "block-rsv.h"
  6#include "space-info.h"
  7#include "transaction.h"
  8#include "block-group.h"
  9#include "disk-io.h"
 10#include "fs.h"
 11#include "accessors.h"
 12
 13/*
 14 * HOW DO BLOCK RESERVES WORK
 15 *
 16 *   Think of block_rsv's as buckets for logically grouped metadata
 17 *   reservations.  Each block_rsv has a ->size and a ->reserved.  ->size is
 18 *   how large we want our block rsv to be, ->reserved is how much space is
 19 *   currently reserved for this block reserve.
 20 *
 21 *   ->failfast exists for the truncate case, and is described below.
 22 *
 23 * NORMAL OPERATION
 24 *
 25 *   -> Reserve
 26 *     Entrance: btrfs_block_rsv_add, btrfs_block_rsv_refill
 27 *
 28 *     We call into btrfs_reserve_metadata_bytes() with our bytes, which is
 29 *     accounted for in space_info->bytes_may_use, and then add the bytes to
 30 *     ->reserved, and ->size in the case of btrfs_block_rsv_add.
 31 *
 32 *     ->size is an over-estimation of how much we may use for a particular
 33 *     operation.
 34 *
 35 *   -> Use
 36 *     Entrance: btrfs_use_block_rsv
 37 *
 38 *     When we do a btrfs_alloc_tree_block() we call into btrfs_use_block_rsv()
 39 *     to determine the appropriate block_rsv to use, and then verify that
 40 *     ->reserved has enough space for our tree block allocation.  Once
 41 *     successful we subtract fs_info->nodesize from ->reserved.
 42 *
 43 *   -> Finish
 44 *     Entrance: btrfs_block_rsv_release
 45 *
 46 *     We are finished with our operation, subtract our individual reservation
 47 *     from ->size, and then subtract ->size from ->reserved and free up the
 48 *     excess if there is any.
 49 *
 50 *     There is some logic here to refill the delayed refs rsv or the global rsv
 51 *     as needed, otherwise the excess is subtracted from
 52 *     space_info->bytes_may_use.
 53 *
 54 * TYPES OF BLOCK RESERVES
 55 *
 56 * BLOCK_RSV_TRANS, BLOCK_RSV_DELOPS, BLOCK_RSV_CHUNK
 57 *   These behave normally, as described above, just within the confines of the
 58 *   lifetime of their particular operation (transaction for the whole trans
 59 *   handle lifetime, for example).
 60 *
 61 * BLOCK_RSV_GLOBAL
 62 *   It is impossible to properly account for all the space that may be required
 63 *   to make our extent tree updates.  This block reserve acts as an overflow
 64 *   buffer in case our delayed refs reserve does not reserve enough space to
 65 *   update the extent tree.
 66 *
 67 *   We can steal from this in some cases as well, notably on evict() or
 68 *   truncate() in order to help users recover from ENOSPC conditions.
 69 *
 70 * BLOCK_RSV_DELALLOC
 71 *   The individual item sizes are determined by the per-inode size
 72 *   calculations, which are described with the delalloc code.  This is pretty
 73 *   straightforward, it's just the calculation of ->size encodes a lot of
 74 *   different items, and thus it gets used when updating inodes, inserting file
 75 *   extents, and inserting checksums.
 76 *
 77 * BLOCK_RSV_DELREFS
 78 *   We keep a running tally of how many delayed refs we have on the system.
 79 *   We assume each one of these delayed refs are going to use a full
 80 *   reservation.  We use the transaction items and pre-reserve space for every
 81 *   operation, and use this reservation to refill any gap between ->size and
 82 *   ->reserved that may exist.
 83 *
 84 *   From there it's straightforward, removing a delayed ref means we remove its
 85 *   count from ->size and free up reservations as necessary.  Since this is
 86 *   the most dynamic block reserve in the system, we will try to refill this
 87 *   block reserve first with any excess returned by any other block reserve.
 88 *
 89 * BLOCK_RSV_EMPTY
 90 *   This is the fallback block reserve to make us try to reserve space if we
 91 *   don't have a specific bucket for this allocation.  It is mostly used for
 92 *   updating the device tree and such, since that is a separate pool we're
 93 *   content to just reserve space from the space_info on demand.
 94 *
 95 * BLOCK_RSV_TEMP
 96 *   This is used by things like truncate and iput.  We will temporarily
 97 *   allocate a block reserve, set it to some size, and then truncate bytes
 98 *   until we have no space left.  With ->failfast set we'll simply return
 99 *   ENOSPC from btrfs_use_block_rsv() to signal that we need to unwind and try
100 *   to make a new reservation.  This is because these operations are
101 *   unbounded, so we want to do as much work as we can, and then back off and
102 *   re-reserve.
103 */
104
105static u64 block_rsv_release_bytes(struct btrfs_fs_info *fs_info,
106				    struct btrfs_block_rsv *block_rsv,
107				    struct btrfs_block_rsv *dest, u64 num_bytes,
108				    u64 *qgroup_to_release_ret)
109{
110	struct btrfs_space_info *space_info = block_rsv->space_info;
111	u64 qgroup_to_release = 0;
112	u64 ret;
113
114	spin_lock(&block_rsv->lock);
115	if (num_bytes == (u64)-1) {
116		num_bytes = block_rsv->size;
117		qgroup_to_release = block_rsv->qgroup_rsv_size;
118	}
119	block_rsv->size -= num_bytes;
120	if (block_rsv->reserved >= block_rsv->size) {
121		num_bytes = block_rsv->reserved - block_rsv->size;
122		block_rsv->reserved = block_rsv->size;
123		block_rsv->full = true;
124	} else {
125		num_bytes = 0;
126	}
127	if (qgroup_to_release_ret &&
128	    block_rsv->qgroup_rsv_reserved >= block_rsv->qgroup_rsv_size) {
129		qgroup_to_release = block_rsv->qgroup_rsv_reserved -
130				    block_rsv->qgroup_rsv_size;
131		block_rsv->qgroup_rsv_reserved = block_rsv->qgroup_rsv_size;
132	} else {
133		qgroup_to_release = 0;
134	}
135	spin_unlock(&block_rsv->lock);
136
137	ret = num_bytes;
138	if (num_bytes > 0) {
139		if (dest) {
140			spin_lock(&dest->lock);
141			if (!dest->full) {
142				u64 bytes_to_add;
143
144				bytes_to_add = dest->size - dest->reserved;
145				bytes_to_add = min(num_bytes, bytes_to_add);
146				dest->reserved += bytes_to_add;
147				if (dest->reserved >= dest->size)
148					dest->full = true;
149				num_bytes -= bytes_to_add;
150			}
151			spin_unlock(&dest->lock);
152		}
153		if (num_bytes)
154			btrfs_space_info_free_bytes_may_use(fs_info,
155							    space_info,
156							    num_bytes);
157	}
158	if (qgroup_to_release_ret)
159		*qgroup_to_release_ret = qgroup_to_release;
160	return ret;
161}
162
163int btrfs_block_rsv_migrate(struct btrfs_block_rsv *src,
164			    struct btrfs_block_rsv *dst, u64 num_bytes,
165			    bool update_size)
166{
167	int ret;
168
169	ret = btrfs_block_rsv_use_bytes(src, num_bytes);
170	if (ret)
171		return ret;
172
173	btrfs_block_rsv_add_bytes(dst, num_bytes, update_size);
174	return 0;
175}
176
177void btrfs_init_block_rsv(struct btrfs_block_rsv *rsv, enum btrfs_rsv_type type)
178{
179	memset(rsv, 0, sizeof(*rsv));
180	spin_lock_init(&rsv->lock);
181	rsv->type = type;
182}
183
184void btrfs_init_metadata_block_rsv(struct btrfs_fs_info *fs_info,
185				   struct btrfs_block_rsv *rsv,
186				   enum btrfs_rsv_type type)
187{
188	btrfs_init_block_rsv(rsv, type);
189	rsv->space_info = btrfs_find_space_info(fs_info,
190					    BTRFS_BLOCK_GROUP_METADATA);
191}
192
193struct btrfs_block_rsv *btrfs_alloc_block_rsv(struct btrfs_fs_info *fs_info,
194					      enum btrfs_rsv_type type)
195{
196	struct btrfs_block_rsv *block_rsv;
197
198	block_rsv = kmalloc(sizeof(*block_rsv), GFP_NOFS);
199	if (!block_rsv)
200		return NULL;
201
202	btrfs_init_metadata_block_rsv(fs_info, block_rsv, type);
203	return block_rsv;
204}
205
206void btrfs_free_block_rsv(struct btrfs_fs_info *fs_info,
207			  struct btrfs_block_rsv *rsv)
208{
209	if (!rsv)
210		return;
211	btrfs_block_rsv_release(fs_info, rsv, (u64)-1, NULL);
212	kfree(rsv);
213}
214
215int btrfs_block_rsv_add(struct btrfs_fs_info *fs_info,
216			struct btrfs_block_rsv *block_rsv, u64 num_bytes,
217			enum btrfs_reserve_flush_enum flush)
218{
219	int ret;
220
221	if (num_bytes == 0)
222		return 0;
223
224	ret = btrfs_reserve_metadata_bytes(fs_info, block_rsv->space_info,
225					   num_bytes, flush);
226	if (!ret)
227		btrfs_block_rsv_add_bytes(block_rsv, num_bytes, true);
228
229	return ret;
230}
231
232int btrfs_block_rsv_check(struct btrfs_block_rsv *block_rsv, int min_percent)
233{
234	u64 num_bytes = 0;
235	int ret = -ENOSPC;
236
 
 
 
237	spin_lock(&block_rsv->lock);
238	num_bytes = mult_perc(block_rsv->size, min_percent);
239	if (block_rsv->reserved >= num_bytes)
240		ret = 0;
241	spin_unlock(&block_rsv->lock);
242
243	return ret;
244}
245
246int btrfs_block_rsv_refill(struct btrfs_fs_info *fs_info,
247			   struct btrfs_block_rsv *block_rsv, u64 num_bytes,
248			   enum btrfs_reserve_flush_enum flush)
249{
 
250	int ret = -ENOSPC;
251
252	if (!block_rsv)
253		return 0;
254
255	spin_lock(&block_rsv->lock);
 
256	if (block_rsv->reserved >= num_bytes)
257		ret = 0;
258	else
259		num_bytes -= block_rsv->reserved;
260	spin_unlock(&block_rsv->lock);
261
262	if (!ret)
263		return 0;
264
265	ret = btrfs_reserve_metadata_bytes(fs_info, block_rsv->space_info,
266					   num_bytes, flush);
267	if (!ret) {
268		btrfs_block_rsv_add_bytes(block_rsv, num_bytes, false);
269		return 0;
270	}
271
272	return ret;
273}
274
275u64 btrfs_block_rsv_release(struct btrfs_fs_info *fs_info,
276			    struct btrfs_block_rsv *block_rsv, u64 num_bytes,
277			    u64 *qgroup_to_release)
278{
279	struct btrfs_block_rsv *global_rsv = &fs_info->global_block_rsv;
280	struct btrfs_block_rsv *delayed_rsv = &fs_info->delayed_refs_rsv;
281	struct btrfs_block_rsv *target = NULL;
282
283	/*
284	 * If we are a delayed block reserve then push to the global rsv,
285	 * otherwise dump into the global delayed reserve if it is not full.
286	 */
287	if (block_rsv->type == BTRFS_BLOCK_RSV_DELOPS)
288		target = global_rsv;
289	else if (block_rsv != global_rsv && !btrfs_block_rsv_full(delayed_rsv))
290		target = delayed_rsv;
291
292	if (target && block_rsv->space_info != target->space_info)
293		target = NULL;
294
295	return block_rsv_release_bytes(fs_info, block_rsv, target, num_bytes,
296				       qgroup_to_release);
297}
298
299int btrfs_block_rsv_use_bytes(struct btrfs_block_rsv *block_rsv, u64 num_bytes)
300{
301	int ret = -ENOSPC;
302
303	spin_lock(&block_rsv->lock);
304	if (block_rsv->reserved >= num_bytes) {
305		block_rsv->reserved -= num_bytes;
306		if (block_rsv->reserved < block_rsv->size)
307			block_rsv->full = false;
308		ret = 0;
309	}
310	spin_unlock(&block_rsv->lock);
311	return ret;
312}
313
314void btrfs_block_rsv_add_bytes(struct btrfs_block_rsv *block_rsv,
315			       u64 num_bytes, bool update_size)
316{
317	spin_lock(&block_rsv->lock);
318	block_rsv->reserved += num_bytes;
319	if (update_size)
320		block_rsv->size += num_bytes;
321	else if (block_rsv->reserved >= block_rsv->size)
322		block_rsv->full = true;
323	spin_unlock(&block_rsv->lock);
324}
325
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
326void btrfs_update_global_block_rsv(struct btrfs_fs_info *fs_info)
327{
328	struct btrfs_block_rsv *block_rsv = &fs_info->global_block_rsv;
329	struct btrfs_space_info *sinfo = block_rsv->space_info;
330	struct btrfs_root *root, *tmp;
331	u64 num_bytes = btrfs_root_used(&fs_info->tree_root->root_item);
332	unsigned int min_items = 1;
333
334	/*
335	 * The global block rsv is based on the size of the extent tree, the
336	 * checksum tree and the root tree.  If the fs is empty we want to set
337	 * it to a minimal amount for safety.
338	 *
339	 * We also are going to need to modify the minimum of the tree root and
340	 * any global roots we could touch.
341	 */
342	read_lock(&fs_info->global_root_lock);
343	rbtree_postorder_for_each_entry_safe(root, tmp, &fs_info->global_root_tree,
344					     rb_node) {
345		if (root->root_key.objectid == BTRFS_EXTENT_TREE_OBJECTID ||
346		    root->root_key.objectid == BTRFS_CSUM_TREE_OBJECTID ||
347		    root->root_key.objectid == BTRFS_FREE_SPACE_TREE_OBJECTID) {
348			num_bytes += btrfs_root_used(&root->root_item);
349			min_items++;
350		}
351	}
352	read_unlock(&fs_info->global_root_lock);
353
354	if (btrfs_fs_compat_ro(fs_info, BLOCK_GROUP_TREE)) {
355		num_bytes += btrfs_root_used(&fs_info->block_group_root->root_item);
356		min_items++;
357	}
358
359	if (btrfs_fs_incompat(fs_info, RAID_STRIPE_TREE)) {
360		num_bytes += btrfs_root_used(&fs_info->stripe_root->root_item);
361		min_items++;
362	}
363
364	/*
365	 * But we also want to reserve enough space so we can do the fallback
366	 * global reserve for an unlink, which is an additional
367	 * BTRFS_UNLINK_METADATA_UNITS items.
368	 *
369	 * But we also need space for the delayed ref updates from the unlink,
370	 * so add BTRFS_UNLINK_METADATA_UNITS units for delayed refs, one for
371	 * each unlink metadata item.
372	 */
373	min_items += BTRFS_UNLINK_METADATA_UNITS;
374
375	num_bytes = max_t(u64, num_bytes,
376			  btrfs_calc_insert_metadata_size(fs_info, min_items) +
377			  btrfs_calc_delayed_ref_bytes(fs_info,
378					       BTRFS_UNLINK_METADATA_UNITS));
379
380	spin_lock(&sinfo->lock);
381	spin_lock(&block_rsv->lock);
382
383	block_rsv->size = min_t(u64, num_bytes, SZ_512M);
384
385	if (block_rsv->reserved < block_rsv->size) {
386		num_bytes = block_rsv->size - block_rsv->reserved;
 
387		btrfs_space_info_update_bytes_may_use(fs_info, sinfo,
388						      num_bytes);
389		block_rsv->reserved = block_rsv->size;
390	} else if (block_rsv->reserved > block_rsv->size) {
391		num_bytes = block_rsv->reserved - block_rsv->size;
392		btrfs_space_info_update_bytes_may_use(fs_info, sinfo,
393						      -num_bytes);
394		block_rsv->reserved = block_rsv->size;
395		btrfs_try_granting_tickets(fs_info, sinfo);
396	}
397
398	block_rsv->full = (block_rsv->reserved == block_rsv->size);
 
 
 
399
400	if (block_rsv->size >= sinfo->total_bytes)
401		sinfo->force_alloc = CHUNK_ALLOC_FORCE;
402	spin_unlock(&block_rsv->lock);
403	spin_unlock(&sinfo->lock);
404}
405
406void btrfs_init_root_block_rsv(struct btrfs_root *root)
407{
408	struct btrfs_fs_info *fs_info = root->fs_info;
409
410	switch (root->root_key.objectid) {
411	case BTRFS_CSUM_TREE_OBJECTID:
412	case BTRFS_EXTENT_TREE_OBJECTID:
413	case BTRFS_FREE_SPACE_TREE_OBJECTID:
414	case BTRFS_BLOCK_GROUP_TREE_OBJECTID:
415	case BTRFS_RAID_STRIPE_TREE_OBJECTID:
416		root->block_rsv = &fs_info->delayed_refs_rsv;
417		break;
418	case BTRFS_ROOT_TREE_OBJECTID:
419	case BTRFS_DEV_TREE_OBJECTID:
420	case BTRFS_QUOTA_TREE_OBJECTID:
421		root->block_rsv = &fs_info->global_block_rsv;
422		break;
423	case BTRFS_CHUNK_TREE_OBJECTID:
424		root->block_rsv = &fs_info->chunk_block_rsv;
425		break;
426	default:
427		root->block_rsv = NULL;
428		break;
429	}
430}
431
432void btrfs_init_global_block_rsv(struct btrfs_fs_info *fs_info)
433{
434	struct btrfs_space_info *space_info;
435
436	space_info = btrfs_find_space_info(fs_info, BTRFS_BLOCK_GROUP_SYSTEM);
437	fs_info->chunk_block_rsv.space_info = space_info;
438
439	space_info = btrfs_find_space_info(fs_info, BTRFS_BLOCK_GROUP_METADATA);
440	fs_info->global_block_rsv.space_info = space_info;
441	fs_info->trans_block_rsv.space_info = space_info;
442	fs_info->empty_block_rsv.space_info = space_info;
443	fs_info->delayed_block_rsv.space_info = space_info;
444	fs_info->delayed_refs_rsv.space_info = space_info;
445
 
 
 
 
 
 
 
 
446	btrfs_update_global_block_rsv(fs_info);
447}
448
449void btrfs_release_global_block_rsv(struct btrfs_fs_info *fs_info)
450{
451	btrfs_block_rsv_release(fs_info, &fs_info->global_block_rsv, (u64)-1,
452				NULL);
453	WARN_ON(fs_info->trans_block_rsv.size > 0);
454	WARN_ON(fs_info->trans_block_rsv.reserved > 0);
455	WARN_ON(fs_info->chunk_block_rsv.size > 0);
456	WARN_ON(fs_info->chunk_block_rsv.reserved > 0);
457	WARN_ON(fs_info->delayed_block_rsv.size > 0);
458	WARN_ON(fs_info->delayed_block_rsv.reserved > 0);
459	WARN_ON(fs_info->delayed_refs_rsv.reserved > 0);
460	WARN_ON(fs_info->delayed_refs_rsv.size > 0);
461}
462
463static struct btrfs_block_rsv *get_block_rsv(
464					const struct btrfs_trans_handle *trans,
465					const struct btrfs_root *root)
466{
467	struct btrfs_fs_info *fs_info = root->fs_info;
468	struct btrfs_block_rsv *block_rsv = NULL;
469
470	if (test_bit(BTRFS_ROOT_SHAREABLE, &root->state) ||
471	    (root == fs_info->uuid_root) ||
472	    (trans->adding_csums &&
473	     root->root_key.objectid == BTRFS_CSUM_TREE_OBJECTID))
474		block_rsv = trans->block_rsv;
475
476	if (!block_rsv)
477		block_rsv = root->block_rsv;
478
479	if (!block_rsv)
480		block_rsv = &fs_info->empty_block_rsv;
481
482	return block_rsv;
483}
484
485struct btrfs_block_rsv *btrfs_use_block_rsv(struct btrfs_trans_handle *trans,
486					    struct btrfs_root *root,
487					    u32 blocksize)
488{
489	struct btrfs_fs_info *fs_info = root->fs_info;
490	struct btrfs_block_rsv *block_rsv;
491	struct btrfs_block_rsv *global_rsv = &fs_info->global_block_rsv;
492	int ret;
493	bool global_updated = false;
494
495	block_rsv = get_block_rsv(trans, root);
496
497	if (unlikely(btrfs_block_rsv_size(block_rsv) == 0))
498		goto try_reserve;
499again:
500	ret = btrfs_block_rsv_use_bytes(block_rsv, blocksize);
501	if (!ret)
502		return block_rsv;
503
504	if (block_rsv->failfast)
505		return ERR_PTR(ret);
506
507	if (block_rsv->type == BTRFS_BLOCK_RSV_GLOBAL && !global_updated) {
508		global_updated = true;
509		btrfs_update_global_block_rsv(fs_info);
510		goto again;
511	}
512
513	/*
514	 * The global reserve still exists to save us from ourselves, so don't
515	 * warn_on if we are short on our delayed refs reserve.
516	 */
517	if (block_rsv->type != BTRFS_BLOCK_RSV_DELREFS &&
518	    btrfs_test_opt(fs_info, ENOSPC_DEBUG)) {
519		static DEFINE_RATELIMIT_STATE(_rs,
520				DEFAULT_RATELIMIT_INTERVAL * 10,
521				/*DEFAULT_RATELIMIT_BURST*/ 1);
522		if (__ratelimit(&_rs))
523			WARN(1, KERN_DEBUG
524				"BTRFS: block rsv %d returned %d\n",
525				block_rsv->type, ret);
526	}
527try_reserve:
528	ret = btrfs_reserve_metadata_bytes(fs_info, block_rsv->space_info,
529					   blocksize, BTRFS_RESERVE_NO_FLUSH);
530	if (!ret)
531		return block_rsv;
532	/*
533	 * If we couldn't reserve metadata bytes try and use some from
534	 * the global reserve if its space type is the same as the global
535	 * reservation.
536	 */
537	if (block_rsv->type != BTRFS_BLOCK_RSV_GLOBAL &&
538	    block_rsv->space_info == global_rsv->space_info) {
539		ret = btrfs_block_rsv_use_bytes(global_rsv, blocksize);
540		if (!ret)
541			return global_rsv;
542	}
543
544	/*
545	 * All hope is lost, but of course our reservations are overly
546	 * pessimistic, so instead of possibly having an ENOSPC abort here, try
547	 * one last time to force a reservation if there's enough actual space
548	 * on disk to make the reservation.
549	 */
550	ret = btrfs_reserve_metadata_bytes(fs_info, block_rsv->space_info, blocksize,
551					   BTRFS_RESERVE_FLUSH_EMERGENCY);
552	if (!ret)
553		return block_rsv;
554
555	return ERR_PTR(ret);
556}
557
558int btrfs_check_trunc_cache_free_space(struct btrfs_fs_info *fs_info,
559				       struct btrfs_block_rsv *rsv)
560{
561	u64 needed_bytes;
562	int ret;
563
564	/* 1 for slack space, 1 for updating the inode */
565	needed_bytes = btrfs_calc_insert_metadata_size(fs_info, 1) +
566		btrfs_calc_metadata_size(fs_info, 1);
567
568	spin_lock(&rsv->lock);
569	if (rsv->reserved < needed_bytes)
570		ret = -ENOSPC;
571	else
572		ret = 0;
573	spin_unlock(&rsv->lock);
574	return ret;
575}