[thirdparty/e2fsprogs.git] / lib / ext2fs / bmap.c

/*
 * bmap.c --- logical to physical block mapping
 *
 * Copyright (C) 1997 Theodore Ts'o.
 *
 * %Begin-Header%
 * This file may be redistributed under the terms of the GNU Library
 * General Public License, version 2.
 * %End-Header%
 */

#include <stdio.h>
#include <string.h>
#if HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <errno.h>

#include "ext2_fs.h"
#include "ext2fs.h"

#if defined(__GNUC__) && !defined(NO_INLINE_FUNCS)
#define _BMAP_INLINE_	__inline__
#else
#define _BMAP_INLINE_
#endif

extern errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino,
			     struct ext2_inode *inode,
			     char *block_buf, int bmap_flags,
			     blk_t block, blk_t *phys_blk);

#define inode_bmap(inode, nr) ((inode)->i_block[(nr)])

static _BMAP_INLINE_ errcode_t block_ind_bmap(ext2_filsys fs, int flags,
					      blk_t ind, char *block_buf,
					      int *blocks_alloc,
					      blk_t nr, blk_t *ret_blk)
{
	errcode_t	retval;
	blk_t		b;

	if (!ind) {
		if (flags & BMAP_SET)
			return EXT2_ET_SET_BMAP_NO_IND;
		*ret_blk = 0;
		return 0;
	}
	retval = io_channel_read_blk(fs->io, ind, 1, block_buf);
	if (retval)
		return retval;

	if (flags & BMAP_SET) {
		b = *ret_blk;
#ifdef WORDS_BIGENDIAN
		b = ext2fs_swab32(b);
#endif
		((blk_t *) block_buf)[nr] = b;
		return io_channel_write_blk(fs->io, ind, 1, block_buf);
	}

	b = ((blk_t *) block_buf)[nr];

#ifdef WORDS_BIGENDIAN
	b = ext2fs_swab32(b);
#endif

	if (!b && (flags & BMAP_ALLOC)) {
		b = nr ? ((blk_t *) block_buf)[nr-1] : 0;
		retval = ext2fs_alloc_block(fs, b,
					    block_buf + fs->blocksize, &b);
		if (retval)
			return retval;

#ifdef WORDS_BIGENDIAN
		((blk_t *) block_buf)[nr] = ext2fs_swab32(b);
#else
		((blk_t *) block_buf)[nr] = b;
#endif

		retval = io_channel_write_blk(fs->io, ind, 1, block_buf);
		if (retval)
			return retval;

		(*blocks_alloc)++;
	}

	*ret_blk = b;
	return 0;
}

static _BMAP_INLINE_ errcode_t block_dind_bmap(ext2_filsys fs, int flags,
					       blk_t dind, char *block_buf,
					       int *blocks_alloc,
					       blk_t nr, blk_t *ret_blk)
{
	blk_t		b;
	errcode_t	retval;
	blk_t		addr_per_block;

	addr_per_block = (blk_t) fs->blocksize >> 2;

	retval = block_ind_bmap(fs, flags & ~BMAP_SET, dind, block_buf,
				blocks_alloc, nr / addr_per_block, &b);
	if (retval)
		return retval;
	retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
				nr % addr_per_block, ret_blk);
	return retval;
}

static _BMAP_INLINE_ errcode_t block_tind_bmap(ext2_filsys fs, int flags,
					       blk_t tind, char *block_buf,
					       int *blocks_alloc,
					       blk_t nr, blk_t *ret_blk)
{
	blk_t		b;
	errcode_t	retval;
	blk_t		addr_per_block;

	addr_per_block = (blk_t) fs->blocksize >> 2;

	retval = block_dind_bmap(fs, flags & ~BMAP_SET, tind, block_buf,
				 blocks_alloc, nr / addr_per_block, &b);
	if (retval)
		return retval;
	retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
				nr % addr_per_block, ret_blk);
	return retval;
}

static errcode_t extent_bmap(ext2_filsys fs, ext2_ino_t ino,
			     struct ext2_inode *inode,
			     ext2_extent_handle_t handle,
			     char *block_buf, int bmap_flags, blk64_t block,
			     int *ret_flags, int *blocks_alloc,
			     blk64_t *phys_blk);

static errcode_t implied_cluster_alloc(ext2_filsys fs, ext2_ino_t ino,
				       struct ext2_inode *inode,
				       ext2_extent_handle_t handle,
				       blk64_t block, blk64_t *phys_blk)
{
	blk64_t	base_block, pblock = 0;
	int i;

	if (!EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
					EXT4_FEATURE_RO_COMPAT_BIGALLOC))
		return 0;

	base_block = block & ~EXT2FS_CLUSTER_MASK(fs);
	for (i = 0; i < EXT2FS_CLUSTER_RATIO(fs); i++) {
		if (block == base_block)
			return 0;
		extent_bmap(fs, ino, inode, handle, 0, 0,
			    base_block + i, 0, 0, &pblock);
		if (pblock)
			break;
	}
	if (pblock == 0)
		return 0;
	*phys_blk = pblock - i + (block - base_block);
	return 0;
}

static errcode_t extent_bmap(ext2_filsys fs, ext2_ino_t ino,
			     struct ext2_inode *inode,
			     ext2_extent_handle_t handle,
			     char *block_buf, int bmap_flags, blk64_t block,
			     int *ret_flags, int *blocks_alloc,
			     blk64_t *phys_blk)
{
	struct ext2fs_extent	extent;
	unsigned int		offset;
	errcode_t		retval = 0;
	blk64_t			blk64 = 0;
	int			alloc = 0;

	if (bmap_flags & BMAP_SET) {
		retval = ext2fs_extent_set_bmap(handle, block,
						*phys_blk, 0);
		return retval;
	}
	retval = ext2fs_extent_goto(handle, block);
	if (retval) {
		/* If the extent is not found, return phys_blk = 0 */
		if (retval == EXT2_ET_EXTENT_NOT_FOUND)
			goto got_block;
		return retval;
	}
	retval = ext2fs_extent_get(handle, EXT2_EXTENT_CURRENT, &extent);
	if (retval)
		return retval;
	offset = block - extent.e_lblk;
	if (block >= extent.e_lblk && (offset <= extent.e_len)) {
		*phys_blk = extent.e_pblk + offset;
		if (ret_flags && extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT)
			*ret_flags |= BMAP_RET_UNINIT;
	}
got_block:
	if ((*phys_blk == 0) && (bmap_flags & BMAP_ALLOC)) {
		implied_cluster_alloc(fs, ino, inode, handle, block, &blk64);
		if (blk64)
			goto set_extent;
		retval = extent_bmap(fs, ino, inode, handle, block_buf,
				     0, block-1, 0, blocks_alloc, &blk64);
		if (retval)
			blk64 = 0;
		retval = ext2fs_alloc_block2(fs, blk64, block_buf,
					     &blk64);
		if (retval)
			return retval;
		blk64 &= ~EXT2FS_CLUSTER_MASK(fs);
		blk64 += EXT2FS_CLUSTER_MASK(fs) & block;
		alloc++;
	set_extent:
		retval = ext2fs_extent_set_bmap(handle, block,
						blk64, 0);
		if (retval)
			return retval;
		/* Update inode after setting extent */
		retval = ext2fs_read_inode(fs, ino, inode);
		if (retval)
			return retval;
		*blocks_alloc += alloc;
		*phys_blk = blk64;
	}
	return 0;
}


errcode_t ext2fs_bmap2(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
		       char *block_buf, int bmap_flags, blk64_t block,
		       int *ret_flags, blk64_t *phys_blk)
{
	struct ext2_inode inode_buf;
	ext2_extent_handle_t handle = 0;
	blk_t addr_per_block;
	blk_t	b, blk32;
	char	*buf = 0;
	errcode_t	retval = 0;
	int		blocks_alloc = 0, inode_dirty = 0;

	if (!(bmap_flags & BMAP_SET))
		*phys_blk = 0;

	if (ret_flags)
		*ret_flags = 0;

	/* Read inode structure if necessary */
	if (!inode) {
		retval = ext2fs_read_inode(fs, ino, &inode_buf);
		if (retval)
			return retval;
		inode = &inode_buf;
	}
	addr_per_block = (blk_t) fs->blocksize >> 2;

	if (!block_buf) {
		retval = ext2fs_get_array(2, fs->blocksize, &buf);
		if (retval)
			return retval;
		block_buf = buf;
	}

	if (inode->i_flags & EXT4_EXTENTS_FL) {
		retval = ext2fs_extent_open2(fs, ino, inode, &handle);
		if (retval)
			goto done;
		retval = extent_bmap(fs, ino, inode, handle, block_buf,
				     bmap_flags, block, ret_flags,
				     &blocks_alloc, phys_blk);
		goto done;
	}

	if (block < EXT2_NDIR_BLOCKS) {
		if (bmap_flags & BMAP_SET) {
			b = *phys_blk;
			inode_bmap(inode, block) = b;
			inode_dirty++;
			goto done;
		}

		*phys_blk = inode_bmap(inode, block);
		b = block ? inode_bmap(inode, block-1) : 0;

		if ((*phys_blk == 0) && (bmap_flags & BMAP_ALLOC)) {
			retval = ext2fs_alloc_block(fs, b, block_buf, &b);
			if (retval)
				goto done;
			inode_bmap(inode, block) = b;
			blocks_alloc++;
			*phys_blk = b;
		}
		goto done;
	}

	/* Indirect block */
	block -= EXT2_NDIR_BLOCKS;
	blk32 = *phys_blk;
	if (block < addr_per_block) {
		b = inode_bmap(inode, EXT2_IND_BLOCK);
		if (!b) {
			if (!(bmap_flags & BMAP_ALLOC)) {
				if (bmap_flags & BMAP_SET)
					retval = EXT2_ET_SET_BMAP_NO_IND;
				goto done;
			}

			b = inode_bmap(inode, EXT2_IND_BLOCK-1);
 			retval = ext2fs_alloc_block(fs, b, block_buf, &b);
			if (retval)
				goto done;
			inode_bmap(inode, EXT2_IND_BLOCK) = b;
			blocks_alloc++;
		}
		retval = block_ind_bmap(fs, bmap_flags, b, block_buf,
					&blocks_alloc, block, &blk32);
		if (retval == 0)
			*phys_blk = blk32;
		goto done;
	}

	/* Doubly indirect block  */
	block -= addr_per_block;
	if (block < addr_per_block * addr_per_block) {
		b = inode_bmap(inode, EXT2_DIND_BLOCK);
		if (!b) {
			if (!(bmap_flags & BMAP_ALLOC)) {
				if (bmap_flags & BMAP_SET)
					retval = EXT2_ET_SET_BMAP_NO_IND;
				goto done;
			}

			b = inode_bmap(inode, EXT2_IND_BLOCK);
 			retval = ext2fs_alloc_block(fs, b, block_buf, &b);
			if (retval)
				goto done;
			inode_bmap(inode, EXT2_DIND_BLOCK) = b;
			blocks_alloc++;
		}
		retval = block_dind_bmap(fs, bmap_flags, b, block_buf,
					 &blocks_alloc, block, &blk32);
		if (retval == 0)
			*phys_blk = blk32;
		goto done;
	}

	/* Triply indirect block */
	block -= addr_per_block * addr_per_block;
	b = inode_bmap(inode, EXT2_TIND_BLOCK);
	if (!b) {
		if (!(bmap_flags & BMAP_ALLOC)) {
			if (bmap_flags & BMAP_SET)
				retval = EXT2_ET_SET_BMAP_NO_IND;
			goto done;
		}

		b = inode_bmap(inode, EXT2_DIND_BLOCK);
		retval = ext2fs_alloc_block(fs, b, block_buf, &b);
		if (retval)
			goto done;
		inode_bmap(inode, EXT2_TIND_BLOCK) = b;
		blocks_alloc++;
	}
	retval = block_tind_bmap(fs, bmap_flags, b, block_buf,
				 &blocks_alloc, block, &blk32);
	if (retval == 0)
		*phys_blk = blk32;
done:
	if (buf)
		ext2fs_free_mem(&buf);
	if (handle)
		ext2fs_extent_free(handle);
	if ((retval == 0) && (blocks_alloc || inode_dirty)) {
		ext2fs_iblk_add_blocks(fs, inode, blocks_alloc);
		retval = ext2fs_write_inode(fs, ino, inode);
	}
	return retval;
}

errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
		      char *block_buf, int bmap_flags, blk_t block,
		      blk_t *phys_blk)
{
	errcode_t ret;
	blk64_t	ret_blk = *phys_blk;

	ret = ext2fs_bmap2(fs, ino, inode, block_buf, bmap_flags, block,
			    0, &ret_blk);
	if (ret)
		return ret;
	if (ret_blk >= ((long long) 1 << 32))
		return EOVERFLOW;
	*phys_blk = ret_blk;
	return 0;
}
Commit	Line	Data
30fab293	1	/*
80e808fc	2	* bmap.c --- logical to physical block mapping
30fab293 TT	3	*
	4	* Copyright (C) 1997 Theodore Ts'o.
	5	*
	6	* %Begin-Header%
543547a5 TT	7	* This file may be redistributed under the terms of the GNU Library
543547a5 TT	8	* General Public License, version 2.
30fab293 TT	9	* %End-Header%
	10	*/
	11
	12	#include <stdio.h>
	13	#include <string.h>
	14	#if HAVE_UNISTD_H
	15	#include <unistd.h>
	16	#endif
cc9bf5d2	17	#include <errno.h>
30fab293	18
b5abe6fa	19	#include "ext2_fs.h"
30fab293 TT	20	#include "ext2fs.h"
30fab293 TT	21
78d8f90f	22	#if defined(__GNUC__) && !defined(NO_INLINE_FUNCS)
30fab293 TT	23	#define _BMAP_INLINE_ __inline__
	24	#else
	25	#define _BMAP_INLINE_
	26	#endif
	27
31dbecd4	28	extern errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino,
efc6f628	29	struct ext2_inode *inode,
30fab293 TT	30	char *block_buf, int bmap_flags,
	31	blk_t block, blk_t *phys_blk);
	32
30fab293 TT	33	#define inode_bmap(inode, nr) ((inode)->i_block[(nr)])
30fab293 TT	34
efc6f628 TT	35	static _BMAP_INLINE_ errcode_t block_ind_bmap(ext2_filsys fs, int flags,
efc6f628 TT	36	blk_t ind, char *block_buf,
30fab293 TT	37	int *blocks_alloc,
	38	blk_t nr, blk_t *ret_blk)
	39	{
	40	errcode_t retval;
	41	blk_t b;
	42
	43	if (!ind) {
1d667534 TT	44	if (flags & BMAP_SET)
1d667534 TT	45	return EXT2_ET_SET_BMAP_NO_IND;
30fab293 TT	46	*ret_blk = 0;
	47	return 0;
	48	}
	49	retval = io_channel_read_blk(fs->io, ind, 1, block_buf);
	50	if (retval)
	51	return retval;
	52
1d667534 TT	53	if (flags & BMAP_SET) {
1d667534 TT	54	b = *ret_blk;
126a291c TT	55	#ifdef WORDS_BIGENDIAN
126a291c TT	56	b = ext2fs_swab32(b);
1d667534 TT	57	#endif
	58	((blk_t *) block_buf)[nr] = b;
	59	return io_channel_write_blk(fs->io, ind, 1, block_buf);
	60	}
	61
30fab293 TT	62	b = ((blk_t *) block_buf)[nr];
30fab293 TT	63
126a291c TT	64	#ifdef WORDS_BIGENDIAN
126a291c TT	65	b = ext2fs_swab32(b);
5df55d7f	66	#endif
30fab293 TT	67
	68	if (!b && (flags & BMAP_ALLOC)) {
	69	b = nr ? ((blk_t *) block_buf)[nr-1] : 0;
	70	retval = ext2fs_alloc_block(fs, b,
	71	block_buf + fs->blocksize, &b);
	72	if (retval)
	73	return retval;
	74
126a291c TT	75	#ifdef WORDS_BIGENDIAN
	76	((blk_t *) block_buf)[nr] = ext2fs_swab32(b);
	77	#else
	78	((blk_t *) block_buf)[nr] = b;
5df55d7f	79	#endif
30fab293 TT	80
	81	retval = io_channel_write_blk(fs->io, ind, 1, block_buf);
	82	if (retval)
	83	return retval;
	84
	85	(*blocks_alloc)++;
	86	}
	87
	88	*ret_blk = b;
	89	return 0;
	90	}
	91
546a1ff1	92	static _BMAP_INLINE_ errcode_t block_dind_bmap(ext2_filsys fs, int flags,
efc6f628	93	blk_t dind, char *block_buf,
30fab293 TT	94	int *blocks_alloc,
	95	blk_t nr, blk_t *ret_blk)
	96	{
	97	blk_t b;
	98	errcode_t retval;
2eb374c9	99	blk_t addr_per_block;
efc6f628	100
2eb374c9	101	addr_per_block = (blk_t) fs->blocksize >> 2;
30fab293	102
efc6f628	103	retval = block_ind_bmap(fs, flags & ~BMAP_SET, dind, block_buf,
1d667534	104	blocks_alloc, nr / addr_per_block, &b);
30fab293 TT	105	if (retval)
	106	return retval;
	107	retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
	108	nr % addr_per_block, ret_blk);
	109	return retval;
	110	}
	111
546a1ff1	112	static _BMAP_INLINE_ errcode_t block_tind_bmap(ext2_filsys fs, int flags,
efc6f628	113	blk_t tind, char *block_buf,
30fab293 TT	114	int *blocks_alloc,
	115	blk_t nr, blk_t *ret_blk)
	116	{
	117	blk_t b;
	118	errcode_t retval;
2eb374c9	119	blk_t addr_per_block;
efc6f628	120
2eb374c9	121	addr_per_block = (blk_t) fs->blocksize >> 2;
30fab293	122
efc6f628	123	retval = block_dind_bmap(fs, flags & ~BMAP_SET, tind, block_buf,
1d667534	124	blocks_alloc, nr / addr_per_block, &b);
30fab293 TT	125	if (retval)
	126	return retval;
	127	retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
	128	nr % addr_per_block, ret_blk);
	129	return retval;
	130	}
	131
8e8a190f TT	132	static errcode_t extent_bmap(ext2_filsys fs, ext2_ino_t ino,
	133	struct ext2_inode *inode,
	134	ext2_extent_handle_t handle,
	135	char *block_buf, int bmap_flags, blk64_t block,
	136	int ret_flags, int blocks_alloc,
	137	blk64_t *phys_blk);
	138
	139	static errcode_t implied_cluster_alloc(ext2_filsys fs, ext2_ino_t ino,
	140	struct ext2_inode *inode,
	141	ext2_extent_handle_t handle,
	142	blk64_t block, blk64_t *phys_blk)
	143	{
	144	blk64_t base_block, pblock = 0;
	145	int i;
	146
	147	if (!EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
	148	EXT4_FEATURE_RO_COMPAT_BIGALLOC))
	149	return 0;
	150
	151	base_block = block & ~EXT2FS_CLUSTER_MASK(fs);
	152	for (i = 0; i < EXT2FS_CLUSTER_RATIO(fs); i++) {
	153	if (block == base_block)
	154	return 0;
	155	extent_bmap(fs, ino, inode, handle, 0, 0,
	156	base_block + i, 0, 0, &pblock);
	157	if (pblock)
	158	break;
	159	}
	160	if (pblock == 0)
	161	return 0;
	162	*phys_blk = pblock - i + (block - base_block);
	163	return 0;
	164	}
	165
551e2e46 TT	166	static errcode_t extent_bmap(ext2_filsys fs, ext2_ino_t ino,
	167	struct ext2_inode *inode,
	168	ext2_extent_handle_t handle,
	169	char *block_buf, int bmap_flags, blk64_t block,
	170	int ret_flags, int blocks_alloc,
	171	blk64_t *phys_blk)
	172	{
	173	struct ext2fs_extent extent;
	174	unsigned int offset;
	175	errcode_t retval = 0;
8e8a190f TT	176	blk64_t blk64 = 0;
8e8a190f TT	177	int alloc = 0;
551e2e46 TT	178
	179	if (bmap_flags & BMAP_SET) {
	180	retval = ext2fs_extent_set_bmap(handle, block,
	181	*phys_blk, 0);
	182	return retval;
	183	}
	184	retval = ext2fs_extent_goto(handle, block);
	185	if (retval) {
	186	/* If the extent is not found, return phys_blk = 0 */
	187	if (retval == EXT2_ET_EXTENT_NOT_FOUND)
	188	goto got_block;
	189	return retval;
	190	}
	191	retval = ext2fs_extent_get(handle, EXT2_EXTENT_CURRENT, &extent);
	192	if (retval)
	193	return retval;
	194	offset = block - extent.e_lblk;
	195	if (block >= extent.e_lblk && (offset <= extent.e_len)) {
	196	*phys_blk = extent.e_pblk + offset;
	197	if (ret_flags && extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT)
	198	*ret_flags \|= BMAP_RET_UNINIT;
	199	}
	200	got_block:
	201	if ((*phys_blk == 0) && (bmap_flags & BMAP_ALLOC)) {
8e8a190f TT	202	implied_cluster_alloc(fs, ino, inode, handle, block, &blk64);
	203	if (blk64)
	204	goto set_extent;
551e2e46 TT	205	retval = extent_bmap(fs, ino, inode, handle, block_buf,
	206	0, block-1, 0, blocks_alloc, &blk64);
	207	if (retval)
	208	blk64 = 0;
	209	retval = ext2fs_alloc_block2(fs, blk64, block_buf,
	210	&blk64);
	211	if (retval)
	212	return retval;
8e8a190f TT	213	blk64 &= ~EXT2FS_CLUSTER_MASK(fs);
	214	blk64 += EXT2FS_CLUSTER_MASK(fs) & block;
	215	alloc++;
	216	set_extent:
551e2e46 TT	217	retval = ext2fs_extent_set_bmap(handle, block,
	218	blk64, 0);
	219	if (retval)
	220	return retval;
	221	/* Update inode after setting extent */
	222	retval = ext2fs_read_inode(fs, ino, inode);
	223	if (retval)
	224	return retval;
8e8a190f	225	*blocks_alloc += alloc;
551e2e46 TT	226	*phys_blk = blk64;
	227	}
	228	return 0;
	229	}
	230
	231
cc9bf5d2 TT	232	errcode_t ext2fs_bmap2(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
	233	char *block_buf, int bmap_flags, blk64_t block,
	234	int ret_flags, blk64_t phys_blk)
30fab293 TT	235	{
30fab293 TT	236	struct ext2_inode inode_buf;
cc9bf5d2	237	ext2_extent_handle_t handle = 0;
2eb374c9	238	blk_t addr_per_block;
cc9bf5d2	239	blk_t b, blk32;
30fab293 TT	240	char *buf = 0;
30fab293 TT	241	errcode_t retval = 0;
1d667534	242	int blocks_alloc = 0, inode_dirty = 0;
30fab293	243
1d667534 TT	244	if (!(bmap_flags & BMAP_SET))
1d667534 TT	245	*phys_blk = 0;
30fab293	246
cc9bf5d2 TT	247	if (ret_flags)
	248	*ret_flags = 0;
	249
30fab293 TT	250	/* Read inode structure if necessary */
	251	if (!inode) {
	252	retval = ext2fs_read_inode(fs, ino, &inode_buf);
b38cd283	253	if (retval)
30fab293 TT	254	return retval;
	255	inode = &inode_buf;
	256	}
2eb374c9	257	addr_per_block = (blk_t) fs->blocksize >> 2;
30fab293 TT	258
30fab293 TT	259	if (!block_buf) {
ee01079a	260	retval = ext2fs_get_array(2, fs->blocksize, &buf);
7b4e4534 TT	261	if (retval)
7b4e4534 TT	262	return retval;
30fab293 TT	263	block_buf = buf;
	264	}
	265
551e2e46 TT	266	if (inode->i_flags & EXT4_EXTENTS_FL) {
	267	retval = ext2fs_extent_open2(fs, ino, inode, &handle);
	268	if (retval)
	269	goto done;
	270	retval = extent_bmap(fs, ino, inode, handle, block_buf,
	271	bmap_flags, block, ret_flags,
	272	&blocks_alloc, phys_blk);
	273	goto done;
	274	}
	275
30fab293	276	if (block < EXT2_NDIR_BLOCKS) {
1d667534 TT	277	if (bmap_flags & BMAP_SET) {
1d667534 TT	278	b = *phys_blk;
1d667534 TT	279	inode_bmap(inode, block) = b;
	280	inode_dirty++;
	281	goto done;
	282	}
	283
30fab293 TT	284	*phys_blk = inode_bmap(inode, block);
30fab293 TT	285	b = block ? inode_bmap(inode, block-1) : 0;
efc6f628	286
30fab293 TT	287	if ((*phys_blk == 0) && (bmap_flags & BMAP_ALLOC)) {
	288	retval = ext2fs_alloc_block(fs, b, block_buf, &b);
	289	if (retval)
	290	goto done;
	291	inode_bmap(inode, block) = b;
	292	blocks_alloc++;
	293	*phys_blk = b;
	294	}
	295	goto done;
	296	}
efc6f628	297
30fab293 TT	298	/* Indirect block */
30fab293 TT	299	block -= EXT2_NDIR_BLOCKS;
cc9bf5d2	300	blk32 = *phys_blk;
30fab293 TT	301	if (block < addr_per_block) {
	302	b = inode_bmap(inode, EXT2_IND_BLOCK);
	303	if (!b) {
1d667534 TT	304	if (!(bmap_flags & BMAP_ALLOC)) {
	305	if (bmap_flags & BMAP_SET)
	306	retval = EXT2_ET_SET_BMAP_NO_IND;
	307	goto done;
	308	}
30fab293 TT	309
	310	b = inode_bmap(inode, EXT2_IND_BLOCK-1);
	311	retval = ext2fs_alloc_block(fs, b, block_buf, &b);
	312	if (retval)
	313	goto done;
	314	inode_bmap(inode, EXT2_IND_BLOCK) = b;
	315	blocks_alloc++;
	316	}
efc6f628	317	retval = block_ind_bmap(fs, bmap_flags, b, block_buf,
cc9bf5d2 TT	318	&blocks_alloc, block, &blk32);
	319	if (retval == 0)
	320	*phys_blk = blk32;
30fab293 TT	321	goto done;
30fab293 TT	322	}
efc6f628	323
30fab293 TT	324	/* Doubly indirect block */
	325	block -= addr_per_block;
	326	if (block < addr_per_block * addr_per_block) {
	327	b = inode_bmap(inode, EXT2_DIND_BLOCK);
	328	if (!b) {
1d667534 TT	329	if (!(bmap_flags & BMAP_ALLOC)) {
	330	if (bmap_flags & BMAP_SET)
	331	retval = EXT2_ET_SET_BMAP_NO_IND;
	332	goto done;
	333	}
30fab293 TT	334
	335	b = inode_bmap(inode, EXT2_IND_BLOCK);
	336	retval = ext2fs_alloc_block(fs, b, block_buf, &b);
	337	if (retval)
	338	goto done;
	339	inode_bmap(inode, EXT2_DIND_BLOCK) = b;
	340	blocks_alloc++;
	341	}
efc6f628	342	retval = block_dind_bmap(fs, bmap_flags, b, block_buf,
cc9bf5d2 TT	343	&blocks_alloc, block, &blk32);
	344	if (retval == 0)
	345	*phys_blk = blk32;
30fab293 TT	346	goto done;
	347	}
	348
	349	/* Triply indirect block */
	350	block -= addr_per_block * addr_per_block;
	351	b = inode_bmap(inode, EXT2_TIND_BLOCK);
	352	if (!b) {
1d667534 TT	353	if (!(bmap_flags & BMAP_ALLOC)) {
	354	if (bmap_flags & BMAP_SET)
	355	retval = EXT2_ET_SET_BMAP_NO_IND;
30fab293	356	goto done;
1d667534	357	}
30fab293 TT	358
	359	b = inode_bmap(inode, EXT2_DIND_BLOCK);
	360	retval = ext2fs_alloc_block(fs, b, block_buf, &b);
	361	if (retval)
	362	goto done;
	363	inode_bmap(inode, EXT2_TIND_BLOCK) = b;
	364	blocks_alloc++;
	365	}
efc6f628	366	retval = block_tind_bmap(fs, bmap_flags, b, block_buf,
cc9bf5d2 TT	367	&blocks_alloc, block, &blk32);
	368	if (retval == 0)
	369	*phys_blk = blk32;
30fab293 TT	370	done:
30fab293 TT	371	if (buf)
c4e3d3f3	372	ext2fs_free_mem(&buf);
cc9bf5d2 TT	373	if (handle)
cc9bf5d2 TT	374	ext2fs_extent_free(handle);
1d667534	375	if ((retval == 0) && (blocks_alloc \|\| inode_dirty)) {
1ca1059f	376	ext2fs_iblk_add_blocks(fs, inode, blocks_alloc);
30fab293 TT	377	retval = ext2fs_write_inode(fs, ino, inode);
	378	}
	379	return retval;
	380	}
	381
cc9bf5d2 TT	382	errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
	383	char *block_buf, int bmap_flags, blk_t block,
	384	blk_t *phys_blk)
	385	{
	386	errcode_t ret;
9c9e1d5f	387	blk64_t ret_blk = *phys_blk;
cc9bf5d2 TT	388
	389	ret = ext2fs_bmap2(fs, ino, inode, block_buf, bmap_flags, block,
	390	0, &ret_blk);
	391	if (ret)
	392	return ret;
	393	if (ret_blk >= ((long long) 1 << 32))
	394	return EOVERFLOW;
	395	*phys_blk = ret_blk;
	396	return 0;
	397	}