162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0+
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * NILFS dat/inode allocator
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2006-2008 Nippon Telegraph and Telephone Corporation.
662306a36Sopenharmony_ci *
762306a36Sopenharmony_ci * Originally written by Koji Sato.
862306a36Sopenharmony_ci * Two allocators were unified by Ryusuke Konishi and Amagai Yoshiji.
962306a36Sopenharmony_ci */
1062306a36Sopenharmony_ci
1162306a36Sopenharmony_ci#include <linux/types.h>
1262306a36Sopenharmony_ci#include <linux/buffer_head.h>
1362306a36Sopenharmony_ci#include <linux/fs.h>
1462306a36Sopenharmony_ci#include <linux/bitops.h>
1562306a36Sopenharmony_ci#include <linux/slab.h>
1662306a36Sopenharmony_ci#include "mdt.h"
1762306a36Sopenharmony_ci#include "alloc.h"
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_ci/**
2162306a36Sopenharmony_ci * nilfs_palloc_groups_per_desc_block - get the number of groups that a group
2262306a36Sopenharmony_ci *					descriptor block can maintain
2362306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
2462306a36Sopenharmony_ci */
2562306a36Sopenharmony_cistatic inline unsigned long
2662306a36Sopenharmony_cinilfs_palloc_groups_per_desc_block(const struct inode *inode)
2762306a36Sopenharmony_ci{
2862306a36Sopenharmony_ci	return i_blocksize(inode) /
2962306a36Sopenharmony_ci		sizeof(struct nilfs_palloc_group_desc);
3062306a36Sopenharmony_ci}
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_ci/**
3362306a36Sopenharmony_ci * nilfs_palloc_groups_count - get maximum number of groups
3462306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
3562306a36Sopenharmony_ci */
3662306a36Sopenharmony_cistatic inline unsigned long
3762306a36Sopenharmony_cinilfs_palloc_groups_count(const struct inode *inode)
3862306a36Sopenharmony_ci{
3962306a36Sopenharmony_ci	return 1UL << (BITS_PER_LONG - (inode->i_blkbits + 3 /* log2(8) */));
4062306a36Sopenharmony_ci}
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_ci/**
4362306a36Sopenharmony_ci * nilfs_palloc_init_blockgroup - initialize private variables for allocator
4462306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
4562306a36Sopenharmony_ci * @entry_size: size of the persistent object
4662306a36Sopenharmony_ci */
4762306a36Sopenharmony_ciint nilfs_palloc_init_blockgroup(struct inode *inode, unsigned int entry_size)
4862306a36Sopenharmony_ci{
4962306a36Sopenharmony_ci	struct nilfs_mdt_info *mi = NILFS_MDT(inode);
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	mi->mi_bgl = kmalloc(sizeof(*mi->mi_bgl), GFP_NOFS);
5262306a36Sopenharmony_ci	if (!mi->mi_bgl)
5362306a36Sopenharmony_ci		return -ENOMEM;
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	bgl_lock_init(mi->mi_bgl);
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	nilfs_mdt_set_entry_size(inode, entry_size, 0);
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	mi->mi_blocks_per_group =
6062306a36Sopenharmony_ci		DIV_ROUND_UP(nilfs_palloc_entries_per_group(inode),
6162306a36Sopenharmony_ci			     mi->mi_entries_per_block) + 1;
6262306a36Sopenharmony_ci		/*
6362306a36Sopenharmony_ci		 * Number of blocks in a group including entry blocks
6462306a36Sopenharmony_ci		 * and a bitmap block
6562306a36Sopenharmony_ci		 */
6662306a36Sopenharmony_ci	mi->mi_blocks_per_desc_block =
6762306a36Sopenharmony_ci		nilfs_palloc_groups_per_desc_block(inode) *
6862306a36Sopenharmony_ci		mi->mi_blocks_per_group + 1;
6962306a36Sopenharmony_ci		/*
7062306a36Sopenharmony_ci		 * Number of blocks per descriptor including the
7162306a36Sopenharmony_ci		 * descriptor block
7262306a36Sopenharmony_ci		 */
7362306a36Sopenharmony_ci	return 0;
7462306a36Sopenharmony_ci}
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci/**
7762306a36Sopenharmony_ci * nilfs_palloc_group - get group number and offset from an entry number
7862306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
7962306a36Sopenharmony_ci * @nr: serial number of the entry (e.g. inode number)
8062306a36Sopenharmony_ci * @offset: pointer to store offset number in the group
8162306a36Sopenharmony_ci */
8262306a36Sopenharmony_cistatic unsigned long nilfs_palloc_group(const struct inode *inode, __u64 nr,
8362306a36Sopenharmony_ci					unsigned long *offset)
8462306a36Sopenharmony_ci{
8562306a36Sopenharmony_ci	__u64 group = nr;
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	*offset = do_div(group, nilfs_palloc_entries_per_group(inode));
8862306a36Sopenharmony_ci	return group;
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_ci/**
9262306a36Sopenharmony_ci * nilfs_palloc_desc_blkoff - get block offset of a group descriptor block
9362306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
9462306a36Sopenharmony_ci * @group: group number
9562306a36Sopenharmony_ci *
9662306a36Sopenharmony_ci * nilfs_palloc_desc_blkoff() returns block offset of the descriptor
9762306a36Sopenharmony_ci * block which contains a descriptor of the specified group.
9862306a36Sopenharmony_ci */
9962306a36Sopenharmony_cistatic unsigned long
10062306a36Sopenharmony_cinilfs_palloc_desc_blkoff(const struct inode *inode, unsigned long group)
10162306a36Sopenharmony_ci{
10262306a36Sopenharmony_ci	unsigned long desc_block =
10362306a36Sopenharmony_ci		group / nilfs_palloc_groups_per_desc_block(inode);
10462306a36Sopenharmony_ci	return desc_block * NILFS_MDT(inode)->mi_blocks_per_desc_block;
10562306a36Sopenharmony_ci}
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci/**
10862306a36Sopenharmony_ci * nilfs_palloc_bitmap_blkoff - get block offset of a bitmap block
10962306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
11062306a36Sopenharmony_ci * @group: group number
11162306a36Sopenharmony_ci *
11262306a36Sopenharmony_ci * nilfs_palloc_bitmap_blkoff() returns block offset of the bitmap
11362306a36Sopenharmony_ci * block used to allocate/deallocate entries in the specified group.
11462306a36Sopenharmony_ci */
11562306a36Sopenharmony_cistatic unsigned long
11662306a36Sopenharmony_cinilfs_palloc_bitmap_blkoff(const struct inode *inode, unsigned long group)
11762306a36Sopenharmony_ci{
11862306a36Sopenharmony_ci	unsigned long desc_offset =
11962306a36Sopenharmony_ci		group % nilfs_palloc_groups_per_desc_block(inode);
12062306a36Sopenharmony_ci	return nilfs_palloc_desc_blkoff(inode, group) + 1 +
12162306a36Sopenharmony_ci		desc_offset * NILFS_MDT(inode)->mi_blocks_per_group;
12262306a36Sopenharmony_ci}
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci/**
12562306a36Sopenharmony_ci * nilfs_palloc_group_desc_nfrees - get the number of free entries in a group
12662306a36Sopenharmony_ci * @desc: pointer to descriptor structure for the group
12762306a36Sopenharmony_ci * @lock: spin lock protecting @desc
12862306a36Sopenharmony_ci */
12962306a36Sopenharmony_cistatic unsigned long
13062306a36Sopenharmony_cinilfs_palloc_group_desc_nfrees(const struct nilfs_palloc_group_desc *desc,
13162306a36Sopenharmony_ci			       spinlock_t *lock)
13262306a36Sopenharmony_ci{
13362306a36Sopenharmony_ci	unsigned long nfree;
13462306a36Sopenharmony_ci
13562306a36Sopenharmony_ci	spin_lock(lock);
13662306a36Sopenharmony_ci	nfree = le32_to_cpu(desc->pg_nfrees);
13762306a36Sopenharmony_ci	spin_unlock(lock);
13862306a36Sopenharmony_ci	return nfree;
13962306a36Sopenharmony_ci}
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci/**
14262306a36Sopenharmony_ci * nilfs_palloc_group_desc_add_entries - adjust count of free entries
14362306a36Sopenharmony_ci * @desc: pointer to descriptor structure for the group
14462306a36Sopenharmony_ci * @lock: spin lock protecting @desc
14562306a36Sopenharmony_ci * @n: delta to be added
14662306a36Sopenharmony_ci */
14762306a36Sopenharmony_cistatic u32
14862306a36Sopenharmony_cinilfs_palloc_group_desc_add_entries(struct nilfs_palloc_group_desc *desc,
14962306a36Sopenharmony_ci				    spinlock_t *lock, u32 n)
15062306a36Sopenharmony_ci{
15162306a36Sopenharmony_ci	u32 nfree;
15262306a36Sopenharmony_ci
15362306a36Sopenharmony_ci	spin_lock(lock);
15462306a36Sopenharmony_ci	le32_add_cpu(&desc->pg_nfrees, n);
15562306a36Sopenharmony_ci	nfree = le32_to_cpu(desc->pg_nfrees);
15662306a36Sopenharmony_ci	spin_unlock(lock);
15762306a36Sopenharmony_ci	return nfree;
15862306a36Sopenharmony_ci}
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ci/**
16162306a36Sopenharmony_ci * nilfs_palloc_entry_blkoff - get block offset of an entry block
16262306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
16362306a36Sopenharmony_ci * @nr: serial number of the entry (e.g. inode number)
16462306a36Sopenharmony_ci */
16562306a36Sopenharmony_cistatic unsigned long
16662306a36Sopenharmony_cinilfs_palloc_entry_blkoff(const struct inode *inode, __u64 nr)
16762306a36Sopenharmony_ci{
16862306a36Sopenharmony_ci	unsigned long group, group_offset;
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	group = nilfs_palloc_group(inode, nr, &group_offset);
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_ci	return nilfs_palloc_bitmap_blkoff(inode, group) + 1 +
17362306a36Sopenharmony_ci		group_offset / NILFS_MDT(inode)->mi_entries_per_block;
17462306a36Sopenharmony_ci}
17562306a36Sopenharmony_ci
17662306a36Sopenharmony_ci/**
17762306a36Sopenharmony_ci * nilfs_palloc_desc_block_init - initialize buffer of a group descriptor block
17862306a36Sopenharmony_ci * @inode: inode of metadata file
17962306a36Sopenharmony_ci * @bh: buffer head of the buffer to be initialized
18062306a36Sopenharmony_ci * @kaddr: kernel address mapped for the page including the buffer
18162306a36Sopenharmony_ci */
18262306a36Sopenharmony_cistatic void nilfs_palloc_desc_block_init(struct inode *inode,
18362306a36Sopenharmony_ci					 struct buffer_head *bh, void *kaddr)
18462306a36Sopenharmony_ci{
18562306a36Sopenharmony_ci	struct nilfs_palloc_group_desc *desc = kaddr + bh_offset(bh);
18662306a36Sopenharmony_ci	unsigned long n = nilfs_palloc_groups_per_desc_block(inode);
18762306a36Sopenharmony_ci	__le32 nfrees;
18862306a36Sopenharmony_ci
18962306a36Sopenharmony_ci	nfrees = cpu_to_le32(nilfs_palloc_entries_per_group(inode));
19062306a36Sopenharmony_ci	while (n-- > 0) {
19162306a36Sopenharmony_ci		desc->pg_nfrees = nfrees;
19262306a36Sopenharmony_ci		desc++;
19362306a36Sopenharmony_ci	}
19462306a36Sopenharmony_ci}
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_cistatic int nilfs_palloc_get_block(struct inode *inode, unsigned long blkoff,
19762306a36Sopenharmony_ci				  int create,
19862306a36Sopenharmony_ci				  void (*init_block)(struct inode *,
19962306a36Sopenharmony_ci						     struct buffer_head *,
20062306a36Sopenharmony_ci						     void *),
20162306a36Sopenharmony_ci				  struct buffer_head **bhp,
20262306a36Sopenharmony_ci				  struct nilfs_bh_assoc *prev,
20362306a36Sopenharmony_ci				  spinlock_t *lock)
20462306a36Sopenharmony_ci{
20562306a36Sopenharmony_ci	int ret;
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_ci	spin_lock(lock);
20862306a36Sopenharmony_ci	if (prev->bh && blkoff == prev->blkoff &&
20962306a36Sopenharmony_ci	    likely(buffer_uptodate(prev->bh))) {
21062306a36Sopenharmony_ci		get_bh(prev->bh);
21162306a36Sopenharmony_ci		*bhp = prev->bh;
21262306a36Sopenharmony_ci		spin_unlock(lock);
21362306a36Sopenharmony_ci		return 0;
21462306a36Sopenharmony_ci	}
21562306a36Sopenharmony_ci	spin_unlock(lock);
21662306a36Sopenharmony_ci
21762306a36Sopenharmony_ci	ret = nilfs_mdt_get_block(inode, blkoff, create, init_block, bhp);
21862306a36Sopenharmony_ci	if (!ret) {
21962306a36Sopenharmony_ci		spin_lock(lock);
22062306a36Sopenharmony_ci		/*
22162306a36Sopenharmony_ci		 * The following code must be safe for change of the
22262306a36Sopenharmony_ci		 * cache contents during the get block call.
22362306a36Sopenharmony_ci		 */
22462306a36Sopenharmony_ci		brelse(prev->bh);
22562306a36Sopenharmony_ci		get_bh(*bhp);
22662306a36Sopenharmony_ci		prev->bh = *bhp;
22762306a36Sopenharmony_ci		prev->blkoff = blkoff;
22862306a36Sopenharmony_ci		spin_unlock(lock);
22962306a36Sopenharmony_ci	}
23062306a36Sopenharmony_ci	return ret;
23162306a36Sopenharmony_ci}
23262306a36Sopenharmony_ci
23362306a36Sopenharmony_ci/**
23462306a36Sopenharmony_ci * nilfs_palloc_delete_block - delete a block on the persistent allocator file
23562306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
23662306a36Sopenharmony_ci * @blkoff: block offset
23762306a36Sopenharmony_ci * @prev: nilfs_bh_assoc struct of the last used buffer
23862306a36Sopenharmony_ci * @lock: spin lock protecting @prev
23962306a36Sopenharmony_ci */
24062306a36Sopenharmony_cistatic int nilfs_palloc_delete_block(struct inode *inode, unsigned long blkoff,
24162306a36Sopenharmony_ci				     struct nilfs_bh_assoc *prev,
24262306a36Sopenharmony_ci				     spinlock_t *lock)
24362306a36Sopenharmony_ci{
24462306a36Sopenharmony_ci	spin_lock(lock);
24562306a36Sopenharmony_ci	if (prev->bh && blkoff == prev->blkoff) {
24662306a36Sopenharmony_ci		brelse(prev->bh);
24762306a36Sopenharmony_ci		prev->bh = NULL;
24862306a36Sopenharmony_ci	}
24962306a36Sopenharmony_ci	spin_unlock(lock);
25062306a36Sopenharmony_ci	return nilfs_mdt_delete_block(inode, blkoff);
25162306a36Sopenharmony_ci}
25262306a36Sopenharmony_ci
25362306a36Sopenharmony_ci/**
25462306a36Sopenharmony_ci * nilfs_palloc_get_desc_block - get buffer head of a group descriptor block
25562306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
25662306a36Sopenharmony_ci * @group: group number
25762306a36Sopenharmony_ci * @create: create flag
25862306a36Sopenharmony_ci * @bhp: pointer to store the resultant buffer head
25962306a36Sopenharmony_ci */
26062306a36Sopenharmony_cistatic int nilfs_palloc_get_desc_block(struct inode *inode,
26162306a36Sopenharmony_ci				       unsigned long group,
26262306a36Sopenharmony_ci				       int create, struct buffer_head **bhp)
26362306a36Sopenharmony_ci{
26462306a36Sopenharmony_ci	struct nilfs_palloc_cache *cache = NILFS_MDT(inode)->mi_palloc_cache;
26562306a36Sopenharmony_ci
26662306a36Sopenharmony_ci	return nilfs_palloc_get_block(inode,
26762306a36Sopenharmony_ci				      nilfs_palloc_desc_blkoff(inode, group),
26862306a36Sopenharmony_ci				      create, nilfs_palloc_desc_block_init,
26962306a36Sopenharmony_ci				      bhp, &cache->prev_desc, &cache->lock);
27062306a36Sopenharmony_ci}
27162306a36Sopenharmony_ci
27262306a36Sopenharmony_ci/**
27362306a36Sopenharmony_ci * nilfs_palloc_get_bitmap_block - get buffer head of a bitmap block
27462306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
27562306a36Sopenharmony_ci * @group: group number
27662306a36Sopenharmony_ci * @create: create flag
27762306a36Sopenharmony_ci * @bhp: pointer to store the resultant buffer head
27862306a36Sopenharmony_ci */
27962306a36Sopenharmony_cistatic int nilfs_palloc_get_bitmap_block(struct inode *inode,
28062306a36Sopenharmony_ci					 unsigned long group,
28162306a36Sopenharmony_ci					 int create, struct buffer_head **bhp)
28262306a36Sopenharmony_ci{
28362306a36Sopenharmony_ci	struct nilfs_palloc_cache *cache = NILFS_MDT(inode)->mi_palloc_cache;
28462306a36Sopenharmony_ci
28562306a36Sopenharmony_ci	return nilfs_palloc_get_block(inode,
28662306a36Sopenharmony_ci				      nilfs_palloc_bitmap_blkoff(inode, group),
28762306a36Sopenharmony_ci				      create, NULL, bhp,
28862306a36Sopenharmony_ci				      &cache->prev_bitmap, &cache->lock);
28962306a36Sopenharmony_ci}
29062306a36Sopenharmony_ci
29162306a36Sopenharmony_ci/**
29262306a36Sopenharmony_ci * nilfs_palloc_delete_bitmap_block - delete a bitmap block
29362306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
29462306a36Sopenharmony_ci * @group: group number
29562306a36Sopenharmony_ci */
29662306a36Sopenharmony_cistatic int nilfs_palloc_delete_bitmap_block(struct inode *inode,
29762306a36Sopenharmony_ci					    unsigned long group)
29862306a36Sopenharmony_ci{
29962306a36Sopenharmony_ci	struct nilfs_palloc_cache *cache = NILFS_MDT(inode)->mi_palloc_cache;
30062306a36Sopenharmony_ci
30162306a36Sopenharmony_ci	return nilfs_palloc_delete_block(inode,
30262306a36Sopenharmony_ci					 nilfs_palloc_bitmap_blkoff(inode,
30362306a36Sopenharmony_ci								    group),
30462306a36Sopenharmony_ci					 &cache->prev_bitmap, &cache->lock);
30562306a36Sopenharmony_ci}
30662306a36Sopenharmony_ci
30762306a36Sopenharmony_ci/**
30862306a36Sopenharmony_ci * nilfs_palloc_get_entry_block - get buffer head of an entry block
30962306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
31062306a36Sopenharmony_ci * @nr: serial number of the entry (e.g. inode number)
31162306a36Sopenharmony_ci * @create: create flag
31262306a36Sopenharmony_ci * @bhp: pointer to store the resultant buffer head
31362306a36Sopenharmony_ci */
31462306a36Sopenharmony_ciint nilfs_palloc_get_entry_block(struct inode *inode, __u64 nr,
31562306a36Sopenharmony_ci				 int create, struct buffer_head **bhp)
31662306a36Sopenharmony_ci{
31762306a36Sopenharmony_ci	struct nilfs_palloc_cache *cache = NILFS_MDT(inode)->mi_palloc_cache;
31862306a36Sopenharmony_ci
31962306a36Sopenharmony_ci	return nilfs_palloc_get_block(inode,
32062306a36Sopenharmony_ci				      nilfs_palloc_entry_blkoff(inode, nr),
32162306a36Sopenharmony_ci				      create, NULL, bhp,
32262306a36Sopenharmony_ci				      &cache->prev_entry, &cache->lock);
32362306a36Sopenharmony_ci}
32462306a36Sopenharmony_ci
32562306a36Sopenharmony_ci/**
32662306a36Sopenharmony_ci * nilfs_palloc_delete_entry_block - delete an entry block
32762306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
32862306a36Sopenharmony_ci * @nr: serial number of the entry
32962306a36Sopenharmony_ci */
33062306a36Sopenharmony_cistatic int nilfs_palloc_delete_entry_block(struct inode *inode, __u64 nr)
33162306a36Sopenharmony_ci{
33262306a36Sopenharmony_ci	struct nilfs_palloc_cache *cache = NILFS_MDT(inode)->mi_palloc_cache;
33362306a36Sopenharmony_ci
33462306a36Sopenharmony_ci	return nilfs_palloc_delete_block(inode,
33562306a36Sopenharmony_ci					 nilfs_palloc_entry_blkoff(inode, nr),
33662306a36Sopenharmony_ci					 &cache->prev_entry, &cache->lock);
33762306a36Sopenharmony_ci}
33862306a36Sopenharmony_ci
33962306a36Sopenharmony_ci/**
34062306a36Sopenharmony_ci * nilfs_palloc_block_get_group_desc - get kernel address of a group descriptor
34162306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
34262306a36Sopenharmony_ci * @group: group number
34362306a36Sopenharmony_ci * @bh: buffer head of the buffer storing the group descriptor block
34462306a36Sopenharmony_ci * @kaddr: kernel address mapped for the page including the buffer
34562306a36Sopenharmony_ci */
34662306a36Sopenharmony_cistatic struct nilfs_palloc_group_desc *
34762306a36Sopenharmony_cinilfs_palloc_block_get_group_desc(const struct inode *inode,
34862306a36Sopenharmony_ci				  unsigned long group,
34962306a36Sopenharmony_ci				  const struct buffer_head *bh, void *kaddr)
35062306a36Sopenharmony_ci{
35162306a36Sopenharmony_ci	return (struct nilfs_palloc_group_desc *)(kaddr + bh_offset(bh)) +
35262306a36Sopenharmony_ci		group % nilfs_palloc_groups_per_desc_block(inode);
35362306a36Sopenharmony_ci}
35462306a36Sopenharmony_ci
35562306a36Sopenharmony_ci/**
35662306a36Sopenharmony_ci * nilfs_palloc_block_get_entry - get kernel address of an entry
35762306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
35862306a36Sopenharmony_ci * @nr: serial number of the entry (e.g. inode number)
35962306a36Sopenharmony_ci * @bh: buffer head of the buffer storing the entry block
36062306a36Sopenharmony_ci * @kaddr: kernel address mapped for the page including the buffer
36162306a36Sopenharmony_ci */
36262306a36Sopenharmony_civoid *nilfs_palloc_block_get_entry(const struct inode *inode, __u64 nr,
36362306a36Sopenharmony_ci				   const struct buffer_head *bh, void *kaddr)
36462306a36Sopenharmony_ci{
36562306a36Sopenharmony_ci	unsigned long entry_offset, group_offset;
36662306a36Sopenharmony_ci
36762306a36Sopenharmony_ci	nilfs_palloc_group(inode, nr, &group_offset);
36862306a36Sopenharmony_ci	entry_offset = group_offset % NILFS_MDT(inode)->mi_entries_per_block;
36962306a36Sopenharmony_ci
37062306a36Sopenharmony_ci	return kaddr + bh_offset(bh) +
37162306a36Sopenharmony_ci		entry_offset * NILFS_MDT(inode)->mi_entry_size;
37262306a36Sopenharmony_ci}
37362306a36Sopenharmony_ci
37462306a36Sopenharmony_ci/**
37562306a36Sopenharmony_ci * nilfs_palloc_find_available_slot - find available slot in a group
37662306a36Sopenharmony_ci * @bitmap: bitmap of the group
37762306a36Sopenharmony_ci * @target: offset number of an entry in the group (start point)
37862306a36Sopenharmony_ci * @bsize: size in bits
37962306a36Sopenharmony_ci * @lock: spin lock protecting @bitmap
38062306a36Sopenharmony_ci */
38162306a36Sopenharmony_cistatic int nilfs_palloc_find_available_slot(unsigned char *bitmap,
38262306a36Sopenharmony_ci					    unsigned long target,
38362306a36Sopenharmony_ci					    unsigned int bsize,
38462306a36Sopenharmony_ci					    spinlock_t *lock)
38562306a36Sopenharmony_ci{
38662306a36Sopenharmony_ci	int pos, end = bsize;
38762306a36Sopenharmony_ci
38862306a36Sopenharmony_ci	if (likely(target < bsize)) {
38962306a36Sopenharmony_ci		pos = target;
39062306a36Sopenharmony_ci		do {
39162306a36Sopenharmony_ci			pos = nilfs_find_next_zero_bit(bitmap, end, pos);
39262306a36Sopenharmony_ci			if (pos >= end)
39362306a36Sopenharmony_ci				break;
39462306a36Sopenharmony_ci			if (!nilfs_set_bit_atomic(lock, pos, bitmap))
39562306a36Sopenharmony_ci				return pos;
39662306a36Sopenharmony_ci		} while (++pos < end);
39762306a36Sopenharmony_ci
39862306a36Sopenharmony_ci		end = target;
39962306a36Sopenharmony_ci	}
40062306a36Sopenharmony_ci
40162306a36Sopenharmony_ci	/* wrap around */
40262306a36Sopenharmony_ci	for (pos = 0; pos < end; pos++) {
40362306a36Sopenharmony_ci		pos = nilfs_find_next_zero_bit(bitmap, end, pos);
40462306a36Sopenharmony_ci		if (pos >= end)
40562306a36Sopenharmony_ci			break;
40662306a36Sopenharmony_ci		if (!nilfs_set_bit_atomic(lock, pos, bitmap))
40762306a36Sopenharmony_ci			return pos;
40862306a36Sopenharmony_ci	}
40962306a36Sopenharmony_ci
41062306a36Sopenharmony_ci	return -ENOSPC;
41162306a36Sopenharmony_ci}
41262306a36Sopenharmony_ci
41362306a36Sopenharmony_ci/**
41462306a36Sopenharmony_ci * nilfs_palloc_rest_groups_in_desc_block - get the remaining number of groups
41562306a36Sopenharmony_ci *					    in a group descriptor block
41662306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
41762306a36Sopenharmony_ci * @curr: current group number
41862306a36Sopenharmony_ci * @max: maximum number of groups
41962306a36Sopenharmony_ci */
42062306a36Sopenharmony_cistatic unsigned long
42162306a36Sopenharmony_cinilfs_palloc_rest_groups_in_desc_block(const struct inode *inode,
42262306a36Sopenharmony_ci				       unsigned long curr, unsigned long max)
42362306a36Sopenharmony_ci{
42462306a36Sopenharmony_ci	return min_t(unsigned long,
42562306a36Sopenharmony_ci		     nilfs_palloc_groups_per_desc_block(inode) -
42662306a36Sopenharmony_ci		     curr % nilfs_palloc_groups_per_desc_block(inode),
42762306a36Sopenharmony_ci		     max - curr + 1);
42862306a36Sopenharmony_ci}
42962306a36Sopenharmony_ci
43062306a36Sopenharmony_ci/**
43162306a36Sopenharmony_ci * nilfs_palloc_count_desc_blocks - count descriptor blocks number
43262306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
43362306a36Sopenharmony_ci * @desc_blocks: descriptor blocks number [out]
43462306a36Sopenharmony_ci */
43562306a36Sopenharmony_cistatic int nilfs_palloc_count_desc_blocks(struct inode *inode,
43662306a36Sopenharmony_ci					    unsigned long *desc_blocks)
43762306a36Sopenharmony_ci{
43862306a36Sopenharmony_ci	__u64 blknum;
43962306a36Sopenharmony_ci	int ret;
44062306a36Sopenharmony_ci
44162306a36Sopenharmony_ci	ret = nilfs_bmap_last_key(NILFS_I(inode)->i_bmap, &blknum);
44262306a36Sopenharmony_ci	if (likely(!ret))
44362306a36Sopenharmony_ci		*desc_blocks = DIV_ROUND_UP(
44462306a36Sopenharmony_ci			(unsigned long)blknum,
44562306a36Sopenharmony_ci			NILFS_MDT(inode)->mi_blocks_per_desc_block);
44662306a36Sopenharmony_ci	return ret;
44762306a36Sopenharmony_ci}
44862306a36Sopenharmony_ci
44962306a36Sopenharmony_ci/**
45062306a36Sopenharmony_ci * nilfs_palloc_mdt_file_can_grow - check potential opportunity for
45162306a36Sopenharmony_ci *					MDT file growing
45262306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
45362306a36Sopenharmony_ci * @desc_blocks: known current descriptor blocks count
45462306a36Sopenharmony_ci */
45562306a36Sopenharmony_cistatic inline bool nilfs_palloc_mdt_file_can_grow(struct inode *inode,
45662306a36Sopenharmony_ci						    unsigned long desc_blocks)
45762306a36Sopenharmony_ci{
45862306a36Sopenharmony_ci	return (nilfs_palloc_groups_per_desc_block(inode) * desc_blocks) <
45962306a36Sopenharmony_ci			nilfs_palloc_groups_count(inode);
46062306a36Sopenharmony_ci}
46162306a36Sopenharmony_ci
46262306a36Sopenharmony_ci/**
46362306a36Sopenharmony_ci * nilfs_palloc_count_max_entries - count max number of entries that can be
46462306a36Sopenharmony_ci *					described by descriptor blocks count
46562306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
46662306a36Sopenharmony_ci * @nused: current number of used entries
46762306a36Sopenharmony_ci * @nmaxp: max number of entries [out]
46862306a36Sopenharmony_ci */
46962306a36Sopenharmony_ciint nilfs_palloc_count_max_entries(struct inode *inode, u64 nused, u64 *nmaxp)
47062306a36Sopenharmony_ci{
47162306a36Sopenharmony_ci	unsigned long desc_blocks = 0;
47262306a36Sopenharmony_ci	u64 entries_per_desc_block, nmax;
47362306a36Sopenharmony_ci	int err;
47462306a36Sopenharmony_ci
47562306a36Sopenharmony_ci	err = nilfs_palloc_count_desc_blocks(inode, &desc_blocks);
47662306a36Sopenharmony_ci	if (unlikely(err))
47762306a36Sopenharmony_ci		return err;
47862306a36Sopenharmony_ci
47962306a36Sopenharmony_ci	entries_per_desc_block = (u64)nilfs_palloc_entries_per_group(inode) *
48062306a36Sopenharmony_ci				nilfs_palloc_groups_per_desc_block(inode);
48162306a36Sopenharmony_ci	nmax = entries_per_desc_block * desc_blocks;
48262306a36Sopenharmony_ci
48362306a36Sopenharmony_ci	if (nused == nmax &&
48462306a36Sopenharmony_ci			nilfs_palloc_mdt_file_can_grow(inode, desc_blocks))
48562306a36Sopenharmony_ci		nmax += entries_per_desc_block;
48662306a36Sopenharmony_ci
48762306a36Sopenharmony_ci	if (nused > nmax)
48862306a36Sopenharmony_ci		return -ERANGE;
48962306a36Sopenharmony_ci
49062306a36Sopenharmony_ci	*nmaxp = nmax;
49162306a36Sopenharmony_ci	return 0;
49262306a36Sopenharmony_ci}
49362306a36Sopenharmony_ci
49462306a36Sopenharmony_ci/**
49562306a36Sopenharmony_ci * nilfs_palloc_prepare_alloc_entry - prepare to allocate a persistent object
49662306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
49762306a36Sopenharmony_ci * @req: nilfs_palloc_req structure exchanged for the allocation
49862306a36Sopenharmony_ci */
49962306a36Sopenharmony_ciint nilfs_palloc_prepare_alloc_entry(struct inode *inode,
50062306a36Sopenharmony_ci				     struct nilfs_palloc_req *req)
50162306a36Sopenharmony_ci{
50262306a36Sopenharmony_ci	struct buffer_head *desc_bh, *bitmap_bh;
50362306a36Sopenharmony_ci	struct nilfs_palloc_group_desc *desc;
50462306a36Sopenharmony_ci	unsigned char *bitmap;
50562306a36Sopenharmony_ci	void *desc_kaddr, *bitmap_kaddr;
50662306a36Sopenharmony_ci	unsigned long group, maxgroup, ngroups;
50762306a36Sopenharmony_ci	unsigned long group_offset, maxgroup_offset;
50862306a36Sopenharmony_ci	unsigned long n, entries_per_group;
50962306a36Sopenharmony_ci	unsigned long i, j;
51062306a36Sopenharmony_ci	spinlock_t *lock;
51162306a36Sopenharmony_ci	int pos, ret;
51262306a36Sopenharmony_ci
51362306a36Sopenharmony_ci	ngroups = nilfs_palloc_groups_count(inode);
51462306a36Sopenharmony_ci	maxgroup = ngroups - 1;
51562306a36Sopenharmony_ci	group = nilfs_palloc_group(inode, req->pr_entry_nr, &group_offset);
51662306a36Sopenharmony_ci	entries_per_group = nilfs_palloc_entries_per_group(inode);
51762306a36Sopenharmony_ci
51862306a36Sopenharmony_ci	for (i = 0; i < ngroups; i += n) {
51962306a36Sopenharmony_ci		if (group >= ngroups) {
52062306a36Sopenharmony_ci			/* wrap around */
52162306a36Sopenharmony_ci			group = 0;
52262306a36Sopenharmony_ci			maxgroup = nilfs_palloc_group(inode, req->pr_entry_nr,
52362306a36Sopenharmony_ci						      &maxgroup_offset) - 1;
52462306a36Sopenharmony_ci		}
52562306a36Sopenharmony_ci		ret = nilfs_palloc_get_desc_block(inode, group, 1, &desc_bh);
52662306a36Sopenharmony_ci		if (ret < 0)
52762306a36Sopenharmony_ci			return ret;
52862306a36Sopenharmony_ci		desc_kaddr = kmap(desc_bh->b_page);
52962306a36Sopenharmony_ci		desc = nilfs_palloc_block_get_group_desc(
53062306a36Sopenharmony_ci			inode, group, desc_bh, desc_kaddr);
53162306a36Sopenharmony_ci		n = nilfs_palloc_rest_groups_in_desc_block(inode, group,
53262306a36Sopenharmony_ci							   maxgroup);
53362306a36Sopenharmony_ci		for (j = 0; j < n; j++, desc++, group++) {
53462306a36Sopenharmony_ci			lock = nilfs_mdt_bgl_lock(inode, group);
53562306a36Sopenharmony_ci			if (nilfs_palloc_group_desc_nfrees(desc, lock) > 0) {
53662306a36Sopenharmony_ci				ret = nilfs_palloc_get_bitmap_block(
53762306a36Sopenharmony_ci					inode, group, 1, &bitmap_bh);
53862306a36Sopenharmony_ci				if (ret < 0)
53962306a36Sopenharmony_ci					goto out_desc;
54062306a36Sopenharmony_ci				bitmap_kaddr = kmap(bitmap_bh->b_page);
54162306a36Sopenharmony_ci				bitmap = bitmap_kaddr + bh_offset(bitmap_bh);
54262306a36Sopenharmony_ci				pos = nilfs_palloc_find_available_slot(
54362306a36Sopenharmony_ci					bitmap, group_offset,
54462306a36Sopenharmony_ci					entries_per_group, lock);
54562306a36Sopenharmony_ci				if (pos >= 0) {
54662306a36Sopenharmony_ci					/* found a free entry */
54762306a36Sopenharmony_ci					nilfs_palloc_group_desc_add_entries(
54862306a36Sopenharmony_ci						desc, lock, -1);
54962306a36Sopenharmony_ci					req->pr_entry_nr =
55062306a36Sopenharmony_ci						entries_per_group * group + pos;
55162306a36Sopenharmony_ci					kunmap(desc_bh->b_page);
55262306a36Sopenharmony_ci					kunmap(bitmap_bh->b_page);
55362306a36Sopenharmony_ci
55462306a36Sopenharmony_ci					req->pr_desc_bh = desc_bh;
55562306a36Sopenharmony_ci					req->pr_bitmap_bh = bitmap_bh;
55662306a36Sopenharmony_ci					return 0;
55762306a36Sopenharmony_ci				}
55862306a36Sopenharmony_ci				kunmap(bitmap_bh->b_page);
55962306a36Sopenharmony_ci				brelse(bitmap_bh);
56062306a36Sopenharmony_ci			}
56162306a36Sopenharmony_ci
56262306a36Sopenharmony_ci			group_offset = 0;
56362306a36Sopenharmony_ci		}
56462306a36Sopenharmony_ci
56562306a36Sopenharmony_ci		kunmap(desc_bh->b_page);
56662306a36Sopenharmony_ci		brelse(desc_bh);
56762306a36Sopenharmony_ci	}
56862306a36Sopenharmony_ci
56962306a36Sopenharmony_ci	/* no entries left */
57062306a36Sopenharmony_ci	return -ENOSPC;
57162306a36Sopenharmony_ci
57262306a36Sopenharmony_ci out_desc:
57362306a36Sopenharmony_ci	kunmap(desc_bh->b_page);
57462306a36Sopenharmony_ci	brelse(desc_bh);
57562306a36Sopenharmony_ci	return ret;
57662306a36Sopenharmony_ci}
57762306a36Sopenharmony_ci
57862306a36Sopenharmony_ci/**
57962306a36Sopenharmony_ci * nilfs_palloc_commit_alloc_entry - finish allocation of a persistent object
58062306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
58162306a36Sopenharmony_ci * @req: nilfs_palloc_req structure exchanged for the allocation
58262306a36Sopenharmony_ci */
58362306a36Sopenharmony_civoid nilfs_palloc_commit_alloc_entry(struct inode *inode,
58462306a36Sopenharmony_ci				     struct nilfs_palloc_req *req)
58562306a36Sopenharmony_ci{
58662306a36Sopenharmony_ci	mark_buffer_dirty(req->pr_bitmap_bh);
58762306a36Sopenharmony_ci	mark_buffer_dirty(req->pr_desc_bh);
58862306a36Sopenharmony_ci	nilfs_mdt_mark_dirty(inode);
58962306a36Sopenharmony_ci
59062306a36Sopenharmony_ci	brelse(req->pr_bitmap_bh);
59162306a36Sopenharmony_ci	brelse(req->pr_desc_bh);
59262306a36Sopenharmony_ci}
59362306a36Sopenharmony_ci
59462306a36Sopenharmony_ci/**
59562306a36Sopenharmony_ci * nilfs_palloc_commit_free_entry - finish deallocating a persistent object
59662306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
59762306a36Sopenharmony_ci * @req: nilfs_palloc_req structure exchanged for the removal
59862306a36Sopenharmony_ci */
59962306a36Sopenharmony_civoid nilfs_palloc_commit_free_entry(struct inode *inode,
60062306a36Sopenharmony_ci				    struct nilfs_palloc_req *req)
60162306a36Sopenharmony_ci{
60262306a36Sopenharmony_ci	struct nilfs_palloc_group_desc *desc;
60362306a36Sopenharmony_ci	unsigned long group, group_offset;
60462306a36Sopenharmony_ci	unsigned char *bitmap;
60562306a36Sopenharmony_ci	void *desc_kaddr, *bitmap_kaddr;
60662306a36Sopenharmony_ci	spinlock_t *lock;
60762306a36Sopenharmony_ci
60862306a36Sopenharmony_ci	group = nilfs_palloc_group(inode, req->pr_entry_nr, &group_offset);
60962306a36Sopenharmony_ci	desc_kaddr = kmap(req->pr_desc_bh->b_page);
61062306a36Sopenharmony_ci	desc = nilfs_palloc_block_get_group_desc(inode, group,
61162306a36Sopenharmony_ci						 req->pr_desc_bh, desc_kaddr);
61262306a36Sopenharmony_ci	bitmap_kaddr = kmap(req->pr_bitmap_bh->b_page);
61362306a36Sopenharmony_ci	bitmap = bitmap_kaddr + bh_offset(req->pr_bitmap_bh);
61462306a36Sopenharmony_ci	lock = nilfs_mdt_bgl_lock(inode, group);
61562306a36Sopenharmony_ci
61662306a36Sopenharmony_ci	if (!nilfs_clear_bit_atomic(lock, group_offset, bitmap))
61762306a36Sopenharmony_ci		nilfs_warn(inode->i_sb,
61862306a36Sopenharmony_ci			   "%s (ino=%lu): entry number %llu already freed",
61962306a36Sopenharmony_ci			   __func__, inode->i_ino,
62062306a36Sopenharmony_ci			   (unsigned long long)req->pr_entry_nr);
62162306a36Sopenharmony_ci	else
62262306a36Sopenharmony_ci		nilfs_palloc_group_desc_add_entries(desc, lock, 1);
62362306a36Sopenharmony_ci
62462306a36Sopenharmony_ci	kunmap(req->pr_bitmap_bh->b_page);
62562306a36Sopenharmony_ci	kunmap(req->pr_desc_bh->b_page);
62662306a36Sopenharmony_ci
62762306a36Sopenharmony_ci	mark_buffer_dirty(req->pr_desc_bh);
62862306a36Sopenharmony_ci	mark_buffer_dirty(req->pr_bitmap_bh);
62962306a36Sopenharmony_ci	nilfs_mdt_mark_dirty(inode);
63062306a36Sopenharmony_ci
63162306a36Sopenharmony_ci	brelse(req->pr_bitmap_bh);
63262306a36Sopenharmony_ci	brelse(req->pr_desc_bh);
63362306a36Sopenharmony_ci}
63462306a36Sopenharmony_ci
63562306a36Sopenharmony_ci/**
63662306a36Sopenharmony_ci * nilfs_palloc_abort_alloc_entry - cancel allocation of a persistent object
63762306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
63862306a36Sopenharmony_ci * @req: nilfs_palloc_req structure exchanged for the allocation
63962306a36Sopenharmony_ci */
64062306a36Sopenharmony_civoid nilfs_palloc_abort_alloc_entry(struct inode *inode,
64162306a36Sopenharmony_ci				    struct nilfs_palloc_req *req)
64262306a36Sopenharmony_ci{
64362306a36Sopenharmony_ci	struct nilfs_palloc_group_desc *desc;
64462306a36Sopenharmony_ci	void *desc_kaddr, *bitmap_kaddr;
64562306a36Sopenharmony_ci	unsigned char *bitmap;
64662306a36Sopenharmony_ci	unsigned long group, group_offset;
64762306a36Sopenharmony_ci	spinlock_t *lock;
64862306a36Sopenharmony_ci
64962306a36Sopenharmony_ci	group = nilfs_palloc_group(inode, req->pr_entry_nr, &group_offset);
65062306a36Sopenharmony_ci	desc_kaddr = kmap(req->pr_desc_bh->b_page);
65162306a36Sopenharmony_ci	desc = nilfs_palloc_block_get_group_desc(inode, group,
65262306a36Sopenharmony_ci						 req->pr_desc_bh, desc_kaddr);
65362306a36Sopenharmony_ci	bitmap_kaddr = kmap(req->pr_bitmap_bh->b_page);
65462306a36Sopenharmony_ci	bitmap = bitmap_kaddr + bh_offset(req->pr_bitmap_bh);
65562306a36Sopenharmony_ci	lock = nilfs_mdt_bgl_lock(inode, group);
65662306a36Sopenharmony_ci
65762306a36Sopenharmony_ci	if (!nilfs_clear_bit_atomic(lock, group_offset, bitmap))
65862306a36Sopenharmony_ci		nilfs_warn(inode->i_sb,
65962306a36Sopenharmony_ci			   "%s (ino=%lu): entry number %llu already freed",
66062306a36Sopenharmony_ci			   __func__, inode->i_ino,
66162306a36Sopenharmony_ci			   (unsigned long long)req->pr_entry_nr);
66262306a36Sopenharmony_ci	else
66362306a36Sopenharmony_ci		nilfs_palloc_group_desc_add_entries(desc, lock, 1);
66462306a36Sopenharmony_ci
66562306a36Sopenharmony_ci	kunmap(req->pr_bitmap_bh->b_page);
66662306a36Sopenharmony_ci	kunmap(req->pr_desc_bh->b_page);
66762306a36Sopenharmony_ci
66862306a36Sopenharmony_ci	brelse(req->pr_bitmap_bh);
66962306a36Sopenharmony_ci	brelse(req->pr_desc_bh);
67062306a36Sopenharmony_ci
67162306a36Sopenharmony_ci	req->pr_entry_nr = 0;
67262306a36Sopenharmony_ci	req->pr_bitmap_bh = NULL;
67362306a36Sopenharmony_ci	req->pr_desc_bh = NULL;
67462306a36Sopenharmony_ci}
67562306a36Sopenharmony_ci
67662306a36Sopenharmony_ci/**
67762306a36Sopenharmony_ci * nilfs_palloc_prepare_free_entry - prepare to deallocate a persistent object
67862306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
67962306a36Sopenharmony_ci * @req: nilfs_palloc_req structure exchanged for the removal
68062306a36Sopenharmony_ci */
68162306a36Sopenharmony_ciint nilfs_palloc_prepare_free_entry(struct inode *inode,
68262306a36Sopenharmony_ci				    struct nilfs_palloc_req *req)
68362306a36Sopenharmony_ci{
68462306a36Sopenharmony_ci	struct buffer_head *desc_bh, *bitmap_bh;
68562306a36Sopenharmony_ci	unsigned long group, group_offset;
68662306a36Sopenharmony_ci	int ret;
68762306a36Sopenharmony_ci
68862306a36Sopenharmony_ci	group = nilfs_palloc_group(inode, req->pr_entry_nr, &group_offset);
68962306a36Sopenharmony_ci	ret = nilfs_palloc_get_desc_block(inode, group, 1, &desc_bh);
69062306a36Sopenharmony_ci	if (ret < 0)
69162306a36Sopenharmony_ci		return ret;
69262306a36Sopenharmony_ci	ret = nilfs_palloc_get_bitmap_block(inode, group, 1, &bitmap_bh);
69362306a36Sopenharmony_ci	if (ret < 0) {
69462306a36Sopenharmony_ci		brelse(desc_bh);
69562306a36Sopenharmony_ci		return ret;
69662306a36Sopenharmony_ci	}
69762306a36Sopenharmony_ci
69862306a36Sopenharmony_ci	req->pr_desc_bh = desc_bh;
69962306a36Sopenharmony_ci	req->pr_bitmap_bh = bitmap_bh;
70062306a36Sopenharmony_ci	return 0;
70162306a36Sopenharmony_ci}
70262306a36Sopenharmony_ci
70362306a36Sopenharmony_ci/**
70462306a36Sopenharmony_ci * nilfs_palloc_abort_free_entry - cancel deallocating a persistent object
70562306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
70662306a36Sopenharmony_ci * @req: nilfs_palloc_req structure exchanged for the removal
70762306a36Sopenharmony_ci */
70862306a36Sopenharmony_civoid nilfs_palloc_abort_free_entry(struct inode *inode,
70962306a36Sopenharmony_ci				   struct nilfs_palloc_req *req)
71062306a36Sopenharmony_ci{
71162306a36Sopenharmony_ci	brelse(req->pr_bitmap_bh);
71262306a36Sopenharmony_ci	brelse(req->pr_desc_bh);
71362306a36Sopenharmony_ci
71462306a36Sopenharmony_ci	req->pr_entry_nr = 0;
71562306a36Sopenharmony_ci	req->pr_bitmap_bh = NULL;
71662306a36Sopenharmony_ci	req->pr_desc_bh = NULL;
71762306a36Sopenharmony_ci}
71862306a36Sopenharmony_ci
71962306a36Sopenharmony_ci/**
72062306a36Sopenharmony_ci * nilfs_palloc_freev - deallocate a set of persistent objects
72162306a36Sopenharmony_ci * @inode: inode of metadata file using this allocator
72262306a36Sopenharmony_ci * @entry_nrs: array of entry numbers to be deallocated
72362306a36Sopenharmony_ci * @nitems: number of entries stored in @entry_nrs
72462306a36Sopenharmony_ci */
72562306a36Sopenharmony_ciint nilfs_palloc_freev(struct inode *inode, __u64 *entry_nrs, size_t nitems)
72662306a36Sopenharmony_ci{
72762306a36Sopenharmony_ci	struct buffer_head *desc_bh, *bitmap_bh;
72862306a36Sopenharmony_ci	struct nilfs_palloc_group_desc *desc;
72962306a36Sopenharmony_ci	unsigned char *bitmap;
73062306a36Sopenharmony_ci	void *desc_kaddr, *bitmap_kaddr;
73162306a36Sopenharmony_ci	unsigned long group, group_offset;
73262306a36Sopenharmony_ci	__u64 group_min_nr, last_nrs[8];
73362306a36Sopenharmony_ci	const unsigned long epg = nilfs_palloc_entries_per_group(inode);
73462306a36Sopenharmony_ci	const unsigned int epb = NILFS_MDT(inode)->mi_entries_per_block;
73562306a36Sopenharmony_ci	unsigned int entry_start, end, pos;
73662306a36Sopenharmony_ci	spinlock_t *lock;
73762306a36Sopenharmony_ci	int i, j, k, ret;
73862306a36Sopenharmony_ci	u32 nfree;
73962306a36Sopenharmony_ci
74062306a36Sopenharmony_ci	for (i = 0; i < nitems; i = j) {
74162306a36Sopenharmony_ci		int change_group = false;
74262306a36Sopenharmony_ci		int nempties = 0, n = 0;
74362306a36Sopenharmony_ci
74462306a36Sopenharmony_ci		group = nilfs_palloc_group(inode, entry_nrs[i], &group_offset);
74562306a36Sopenharmony_ci		ret = nilfs_palloc_get_desc_block(inode, group, 0, &desc_bh);
74662306a36Sopenharmony_ci		if (ret < 0)
74762306a36Sopenharmony_ci			return ret;
74862306a36Sopenharmony_ci		ret = nilfs_palloc_get_bitmap_block(inode, group, 0,
74962306a36Sopenharmony_ci						    &bitmap_bh);
75062306a36Sopenharmony_ci		if (ret < 0) {
75162306a36Sopenharmony_ci			brelse(desc_bh);
75262306a36Sopenharmony_ci			return ret;
75362306a36Sopenharmony_ci		}
75462306a36Sopenharmony_ci
75562306a36Sopenharmony_ci		/* Get the first entry number of the group */
75662306a36Sopenharmony_ci		group_min_nr = (__u64)group * epg;
75762306a36Sopenharmony_ci
75862306a36Sopenharmony_ci		bitmap_kaddr = kmap(bitmap_bh->b_page);
75962306a36Sopenharmony_ci		bitmap = bitmap_kaddr + bh_offset(bitmap_bh);
76062306a36Sopenharmony_ci		lock = nilfs_mdt_bgl_lock(inode, group);
76162306a36Sopenharmony_ci
76262306a36Sopenharmony_ci		j = i;
76362306a36Sopenharmony_ci		entry_start = rounddown(group_offset, epb);
76462306a36Sopenharmony_ci		do {
76562306a36Sopenharmony_ci			if (!nilfs_clear_bit_atomic(lock, group_offset,
76662306a36Sopenharmony_ci						    bitmap)) {
76762306a36Sopenharmony_ci				nilfs_warn(inode->i_sb,
76862306a36Sopenharmony_ci					   "%s (ino=%lu): entry number %llu already freed",
76962306a36Sopenharmony_ci					   __func__, inode->i_ino,
77062306a36Sopenharmony_ci					   (unsigned long long)entry_nrs[j]);
77162306a36Sopenharmony_ci			} else {
77262306a36Sopenharmony_ci				n++;
77362306a36Sopenharmony_ci			}
77462306a36Sopenharmony_ci
77562306a36Sopenharmony_ci			j++;
77662306a36Sopenharmony_ci			if (j >= nitems || entry_nrs[j] < group_min_nr ||
77762306a36Sopenharmony_ci			    entry_nrs[j] >= group_min_nr + epg) {
77862306a36Sopenharmony_ci				change_group = true;
77962306a36Sopenharmony_ci			} else {
78062306a36Sopenharmony_ci				group_offset = entry_nrs[j] - group_min_nr;
78162306a36Sopenharmony_ci				if (group_offset >= entry_start &&
78262306a36Sopenharmony_ci				    group_offset < entry_start + epb) {
78362306a36Sopenharmony_ci					/* This entry is in the same block */
78462306a36Sopenharmony_ci					continue;
78562306a36Sopenharmony_ci				}
78662306a36Sopenharmony_ci			}
78762306a36Sopenharmony_ci
78862306a36Sopenharmony_ci			/* Test if the entry block is empty or not */
78962306a36Sopenharmony_ci			end = entry_start + epb;
79062306a36Sopenharmony_ci			pos = nilfs_find_next_bit(bitmap, end, entry_start);
79162306a36Sopenharmony_ci			if (pos >= end) {
79262306a36Sopenharmony_ci				last_nrs[nempties++] = entry_nrs[j - 1];
79362306a36Sopenharmony_ci				if (nempties >= ARRAY_SIZE(last_nrs))
79462306a36Sopenharmony_ci					break;
79562306a36Sopenharmony_ci			}
79662306a36Sopenharmony_ci
79762306a36Sopenharmony_ci			if (change_group)
79862306a36Sopenharmony_ci				break;
79962306a36Sopenharmony_ci
80062306a36Sopenharmony_ci			/* Go on to the next entry block */
80162306a36Sopenharmony_ci			entry_start = rounddown(group_offset, epb);
80262306a36Sopenharmony_ci		} while (true);
80362306a36Sopenharmony_ci
80462306a36Sopenharmony_ci		kunmap(bitmap_bh->b_page);
80562306a36Sopenharmony_ci		mark_buffer_dirty(bitmap_bh);
80662306a36Sopenharmony_ci		brelse(bitmap_bh);
80762306a36Sopenharmony_ci
80862306a36Sopenharmony_ci		for (k = 0; k < nempties; k++) {
80962306a36Sopenharmony_ci			ret = nilfs_palloc_delete_entry_block(inode,
81062306a36Sopenharmony_ci							      last_nrs[k]);
81162306a36Sopenharmony_ci			if (ret && ret != -ENOENT)
81262306a36Sopenharmony_ci				nilfs_warn(inode->i_sb,
81362306a36Sopenharmony_ci					   "error %d deleting block that object (entry=%llu, ino=%lu) belongs to",
81462306a36Sopenharmony_ci					   ret, (unsigned long long)last_nrs[k],
81562306a36Sopenharmony_ci					   inode->i_ino);
81662306a36Sopenharmony_ci		}
81762306a36Sopenharmony_ci
81862306a36Sopenharmony_ci		desc_kaddr = kmap_atomic(desc_bh->b_page);
81962306a36Sopenharmony_ci		desc = nilfs_palloc_block_get_group_desc(
82062306a36Sopenharmony_ci			inode, group, desc_bh, desc_kaddr);
82162306a36Sopenharmony_ci		nfree = nilfs_palloc_group_desc_add_entries(desc, lock, n);
82262306a36Sopenharmony_ci		kunmap_atomic(desc_kaddr);
82362306a36Sopenharmony_ci		mark_buffer_dirty(desc_bh);
82462306a36Sopenharmony_ci		nilfs_mdt_mark_dirty(inode);
82562306a36Sopenharmony_ci		brelse(desc_bh);
82662306a36Sopenharmony_ci
82762306a36Sopenharmony_ci		if (nfree == nilfs_palloc_entries_per_group(inode)) {
82862306a36Sopenharmony_ci			ret = nilfs_palloc_delete_bitmap_block(inode, group);
82962306a36Sopenharmony_ci			if (ret && ret != -ENOENT)
83062306a36Sopenharmony_ci				nilfs_warn(inode->i_sb,
83162306a36Sopenharmony_ci					   "error %d deleting bitmap block of group=%lu, ino=%lu",
83262306a36Sopenharmony_ci					   ret, group, inode->i_ino);
83362306a36Sopenharmony_ci		}
83462306a36Sopenharmony_ci	}
83562306a36Sopenharmony_ci	return 0;
83662306a36Sopenharmony_ci}
83762306a36Sopenharmony_ci
83862306a36Sopenharmony_civoid nilfs_palloc_setup_cache(struct inode *inode,
83962306a36Sopenharmony_ci			      struct nilfs_palloc_cache *cache)
84062306a36Sopenharmony_ci{
84162306a36Sopenharmony_ci	NILFS_MDT(inode)->mi_palloc_cache = cache;
84262306a36Sopenharmony_ci	spin_lock_init(&cache->lock);
84362306a36Sopenharmony_ci}
84462306a36Sopenharmony_ci
84562306a36Sopenharmony_civoid nilfs_palloc_clear_cache(struct inode *inode)
84662306a36Sopenharmony_ci{
84762306a36Sopenharmony_ci	struct nilfs_palloc_cache *cache = NILFS_MDT(inode)->mi_palloc_cache;
84862306a36Sopenharmony_ci
84962306a36Sopenharmony_ci	spin_lock(&cache->lock);
85062306a36Sopenharmony_ci	brelse(cache->prev_desc.bh);
85162306a36Sopenharmony_ci	brelse(cache->prev_bitmap.bh);
85262306a36Sopenharmony_ci	brelse(cache->prev_entry.bh);
85362306a36Sopenharmony_ci	cache->prev_desc.bh = NULL;
85462306a36Sopenharmony_ci	cache->prev_bitmap.bh = NULL;
85562306a36Sopenharmony_ci	cache->prev_entry.bh = NULL;
85662306a36Sopenharmony_ci	spin_unlock(&cache->lock);
85762306a36Sopenharmony_ci}
85862306a36Sopenharmony_ci
85962306a36Sopenharmony_civoid nilfs_palloc_destroy_cache(struct inode *inode)
86062306a36Sopenharmony_ci{
86162306a36Sopenharmony_ci	nilfs_palloc_clear_cache(inode);
86262306a36Sopenharmony_ci	NILFS_MDT(inode)->mi_palloc_cache = NULL;
86362306a36Sopenharmony_ci}
864