xref: /aosp_15_r20/external/e2fsprogs/lib/ext2fs/bmap.c (revision 6a54128f25917bfc36a8a6e9d722c04a0b4641b6)
1*6a54128fSAndroid Build Coastguard Worker /*
2*6a54128fSAndroid Build Coastguard Worker  * bmap.c --- logical to physical block mapping
3*6a54128fSAndroid Build Coastguard Worker  *
4*6a54128fSAndroid Build Coastguard Worker  * Copyright (C) 1997 Theodore Ts'o.
5*6a54128fSAndroid Build Coastguard Worker  *
6*6a54128fSAndroid Build Coastguard Worker  * %Begin-Header%
7*6a54128fSAndroid Build Coastguard Worker  * This file may be redistributed under the terms of the GNU Library
8*6a54128fSAndroid Build Coastguard Worker  * General Public License, version 2.
9*6a54128fSAndroid Build Coastguard Worker  * %End-Header%
10*6a54128fSAndroid Build Coastguard Worker  */
11*6a54128fSAndroid Build Coastguard Worker 
12*6a54128fSAndroid Build Coastguard Worker #include "config.h"
13*6a54128fSAndroid Build Coastguard Worker #include <stdio.h>
14*6a54128fSAndroid Build Coastguard Worker #include <string.h>
15*6a54128fSAndroid Build Coastguard Worker #if HAVE_UNISTD_H
16*6a54128fSAndroid Build Coastguard Worker #include <unistd.h>
17*6a54128fSAndroid Build Coastguard Worker #endif
18*6a54128fSAndroid Build Coastguard Worker #include <errno.h>
19*6a54128fSAndroid Build Coastguard Worker 
20*6a54128fSAndroid Build Coastguard Worker #include "ext2_fs.h"
21*6a54128fSAndroid Build Coastguard Worker #include "ext2fsP.h"
22*6a54128fSAndroid Build Coastguard Worker 
23*6a54128fSAndroid Build Coastguard Worker #if defined(__GNUC__) && !defined(NO_INLINE_FUNCS)
24*6a54128fSAndroid Build Coastguard Worker #define _BMAP_INLINE_	__inline__
25*6a54128fSAndroid Build Coastguard Worker #else
26*6a54128fSAndroid Build Coastguard Worker #define _BMAP_INLINE_
27*6a54128fSAndroid Build Coastguard Worker #endif
28*6a54128fSAndroid Build Coastguard Worker 
29*6a54128fSAndroid Build Coastguard Worker extern errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino,
30*6a54128fSAndroid Build Coastguard Worker 			     struct ext2_inode *inode,
31*6a54128fSAndroid Build Coastguard Worker 			     char *block_buf, int bmap_flags,
32*6a54128fSAndroid Build Coastguard Worker 			     blk_t block, blk_t *phys_blk);
33*6a54128fSAndroid Build Coastguard Worker 
34*6a54128fSAndroid Build Coastguard Worker #define inode_bmap(inode, nr) ((inode)->i_block[(nr)])
35*6a54128fSAndroid Build Coastguard Worker 
block_ind_bmap(ext2_filsys fs,int flags,blk_t ind,char * block_buf,int * blocks_alloc,blk_t nr,blk_t * ret_blk)36*6a54128fSAndroid Build Coastguard Worker static _BMAP_INLINE_ errcode_t block_ind_bmap(ext2_filsys fs, int flags,
37*6a54128fSAndroid Build Coastguard Worker 					      blk_t ind, char *block_buf,
38*6a54128fSAndroid Build Coastguard Worker 					      int *blocks_alloc,
39*6a54128fSAndroid Build Coastguard Worker 					      blk_t nr, blk_t *ret_blk)
40*6a54128fSAndroid Build Coastguard Worker {
41*6a54128fSAndroid Build Coastguard Worker 	errcode_t	retval;
42*6a54128fSAndroid Build Coastguard Worker 	blk_t		b;
43*6a54128fSAndroid Build Coastguard Worker 
44*6a54128fSAndroid Build Coastguard Worker 	if (!ind) {
45*6a54128fSAndroid Build Coastguard Worker 		if (flags & BMAP_SET)
46*6a54128fSAndroid Build Coastguard Worker 			return EXT2_ET_SET_BMAP_NO_IND;
47*6a54128fSAndroid Build Coastguard Worker 		*ret_blk = 0;
48*6a54128fSAndroid Build Coastguard Worker 		return 0;
49*6a54128fSAndroid Build Coastguard Worker 	}
50*6a54128fSAndroid Build Coastguard Worker 	retval = io_channel_read_blk(fs->io, ind, 1, block_buf);
51*6a54128fSAndroid Build Coastguard Worker 	if (retval)
52*6a54128fSAndroid Build Coastguard Worker 		return retval;
53*6a54128fSAndroid Build Coastguard Worker 
54*6a54128fSAndroid Build Coastguard Worker 	if (flags & BMAP_SET) {
55*6a54128fSAndroid Build Coastguard Worker 		b = *ret_blk;
56*6a54128fSAndroid Build Coastguard Worker #ifdef WORDS_BIGENDIAN
57*6a54128fSAndroid Build Coastguard Worker 		b = ext2fs_swab32(b);
58*6a54128fSAndroid Build Coastguard Worker #endif
59*6a54128fSAndroid Build Coastguard Worker 		((blk_t *) block_buf)[nr] = b;
60*6a54128fSAndroid Build Coastguard Worker 		return io_channel_write_blk(fs->io, ind, 1, block_buf);
61*6a54128fSAndroid Build Coastguard Worker 	}
62*6a54128fSAndroid Build Coastguard Worker 
63*6a54128fSAndroid Build Coastguard Worker 	b = ((blk_t *) block_buf)[nr];
64*6a54128fSAndroid Build Coastguard Worker 
65*6a54128fSAndroid Build Coastguard Worker #ifdef WORDS_BIGENDIAN
66*6a54128fSAndroid Build Coastguard Worker 	b = ext2fs_swab32(b);
67*6a54128fSAndroid Build Coastguard Worker #endif
68*6a54128fSAndroid Build Coastguard Worker 
69*6a54128fSAndroid Build Coastguard Worker 	if (!b && (flags & BMAP_ALLOC)) {
70*6a54128fSAndroid Build Coastguard Worker 		b = nr ? ext2fs_le32_to_cpu(((blk_t *)block_buf)[nr - 1]) : ind;
71*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_alloc_block(fs, b,
72*6a54128fSAndroid Build Coastguard Worker 					    block_buf + fs->blocksize, &b);
73*6a54128fSAndroid Build Coastguard Worker 		if (retval)
74*6a54128fSAndroid Build Coastguard Worker 			return retval;
75*6a54128fSAndroid Build Coastguard Worker 
76*6a54128fSAndroid Build Coastguard Worker #ifdef WORDS_BIGENDIAN
77*6a54128fSAndroid Build Coastguard Worker 		((blk_t *) block_buf)[nr] = ext2fs_swab32(b);
78*6a54128fSAndroid Build Coastguard Worker #else
79*6a54128fSAndroid Build Coastguard Worker 		((blk_t *) block_buf)[nr] = b;
80*6a54128fSAndroid Build Coastguard Worker #endif
81*6a54128fSAndroid Build Coastguard Worker 
82*6a54128fSAndroid Build Coastguard Worker 		retval = io_channel_write_blk(fs->io, ind, 1, block_buf);
83*6a54128fSAndroid Build Coastguard Worker 		if (retval)
84*6a54128fSAndroid Build Coastguard Worker 			return retval;
85*6a54128fSAndroid Build Coastguard Worker 
86*6a54128fSAndroid Build Coastguard Worker 		(*blocks_alloc)++;
87*6a54128fSAndroid Build Coastguard Worker 	}
88*6a54128fSAndroid Build Coastguard Worker 
89*6a54128fSAndroid Build Coastguard Worker 	*ret_blk = b;
90*6a54128fSAndroid Build Coastguard Worker 	return 0;
91*6a54128fSAndroid Build Coastguard Worker }
92*6a54128fSAndroid Build Coastguard Worker 
block_dind_bmap(ext2_filsys fs,int flags,blk_t dind,char * block_buf,int * blocks_alloc,blk_t nr,blk_t * ret_blk)93*6a54128fSAndroid Build Coastguard Worker static _BMAP_INLINE_ errcode_t block_dind_bmap(ext2_filsys fs, int flags,
94*6a54128fSAndroid Build Coastguard Worker 					       blk_t dind, char *block_buf,
95*6a54128fSAndroid Build Coastguard Worker 					       int *blocks_alloc,
96*6a54128fSAndroid Build Coastguard Worker 					       blk_t nr, blk_t *ret_blk)
97*6a54128fSAndroid Build Coastguard Worker {
98*6a54128fSAndroid Build Coastguard Worker 	blk_t		b = 0;
99*6a54128fSAndroid Build Coastguard Worker 	errcode_t	retval;
100*6a54128fSAndroid Build Coastguard Worker 	blk_t		addr_per_block;
101*6a54128fSAndroid Build Coastguard Worker 
102*6a54128fSAndroid Build Coastguard Worker 	addr_per_block = (blk_t) fs->blocksize >> 2;
103*6a54128fSAndroid Build Coastguard Worker 
104*6a54128fSAndroid Build Coastguard Worker 	retval = block_ind_bmap(fs, flags & ~BMAP_SET, dind, block_buf,
105*6a54128fSAndroid Build Coastguard Worker 				blocks_alloc, nr / addr_per_block, &b);
106*6a54128fSAndroid Build Coastguard Worker 	if (retval)
107*6a54128fSAndroid Build Coastguard Worker 		return retval;
108*6a54128fSAndroid Build Coastguard Worker 	retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
109*6a54128fSAndroid Build Coastguard Worker 				nr % addr_per_block, ret_blk);
110*6a54128fSAndroid Build Coastguard Worker 	return retval;
111*6a54128fSAndroid Build Coastguard Worker }
112*6a54128fSAndroid Build Coastguard Worker 
block_tind_bmap(ext2_filsys fs,int flags,blk_t tind,char * block_buf,int * blocks_alloc,blk_t nr,blk_t * ret_blk)113*6a54128fSAndroid Build Coastguard Worker static _BMAP_INLINE_ errcode_t block_tind_bmap(ext2_filsys fs, int flags,
114*6a54128fSAndroid Build Coastguard Worker 					       blk_t tind, char *block_buf,
115*6a54128fSAndroid Build Coastguard Worker 					       int *blocks_alloc,
116*6a54128fSAndroid Build Coastguard Worker 					       blk_t nr, blk_t *ret_blk)
117*6a54128fSAndroid Build Coastguard Worker {
118*6a54128fSAndroid Build Coastguard Worker 	blk_t		b = 0;
119*6a54128fSAndroid Build Coastguard Worker 	errcode_t	retval;
120*6a54128fSAndroid Build Coastguard Worker 	blk_t		addr_per_block;
121*6a54128fSAndroid Build Coastguard Worker 
122*6a54128fSAndroid Build Coastguard Worker 	addr_per_block = (blk_t) fs->blocksize >> 2;
123*6a54128fSAndroid Build Coastguard Worker 
124*6a54128fSAndroid Build Coastguard Worker 	retval = block_dind_bmap(fs, flags & ~BMAP_SET, tind, block_buf,
125*6a54128fSAndroid Build Coastguard Worker 				 blocks_alloc, nr / addr_per_block, &b);
126*6a54128fSAndroid Build Coastguard Worker 	if (retval)
127*6a54128fSAndroid Build Coastguard Worker 		return retval;
128*6a54128fSAndroid Build Coastguard Worker 	retval = block_ind_bmap(fs, flags, b, block_buf, blocks_alloc,
129*6a54128fSAndroid Build Coastguard Worker 				nr % addr_per_block, ret_blk);
130*6a54128fSAndroid Build Coastguard Worker 	return retval;
131*6a54128fSAndroid Build Coastguard Worker }
132*6a54128fSAndroid Build Coastguard Worker 
133*6a54128fSAndroid Build Coastguard Worker static errcode_t extent_bmap(ext2_filsys fs, ext2_ino_t ino,
134*6a54128fSAndroid Build Coastguard Worker 			     struct ext2_inode *inode,
135*6a54128fSAndroid Build Coastguard Worker 			     ext2_extent_handle_t handle,
136*6a54128fSAndroid Build Coastguard Worker 			     char *block_buf, int bmap_flags, blk64_t block,
137*6a54128fSAndroid Build Coastguard Worker 			     int *ret_flags, int *blocks_alloc,
138*6a54128fSAndroid Build Coastguard Worker 			     blk64_t *phys_blk);
139*6a54128fSAndroid Build Coastguard Worker 
implied_cluster_alloc(ext2_filsys fs,ext2_ino_t ino,struct ext2_inode * inode,ext2_extent_handle_t handle,blk64_t lblk,blk64_t * phys_blk)140*6a54128fSAndroid Build Coastguard Worker static errcode_t implied_cluster_alloc(ext2_filsys fs, ext2_ino_t ino,
141*6a54128fSAndroid Build Coastguard Worker 				       struct ext2_inode *inode,
142*6a54128fSAndroid Build Coastguard Worker 				       ext2_extent_handle_t handle,
143*6a54128fSAndroid Build Coastguard Worker 				       blk64_t lblk, blk64_t *phys_blk)
144*6a54128fSAndroid Build Coastguard Worker {
145*6a54128fSAndroid Build Coastguard Worker 	blk64_t	base_block, pblock = 0;
146*6a54128fSAndroid Build Coastguard Worker 	int i;
147*6a54128fSAndroid Build Coastguard Worker 
148*6a54128fSAndroid Build Coastguard Worker 	if (!ext2fs_has_feature_bigalloc(fs->super))
149*6a54128fSAndroid Build Coastguard Worker 		return 0;
150*6a54128fSAndroid Build Coastguard Worker 
151*6a54128fSAndroid Build Coastguard Worker 	base_block = lblk & ~EXT2FS_CLUSTER_MASK(fs);
152*6a54128fSAndroid Build Coastguard Worker 	/*
153*6a54128fSAndroid Build Coastguard Worker 	 * Except for the logical block (lblk) that was passed in, search all
154*6a54128fSAndroid Build Coastguard Worker 	 * blocks in this logical cluster for a mapping to a physical cluster.
155*6a54128fSAndroid Build Coastguard Worker 	 * If any such map exists, calculate the physical block that maps to
156*6a54128fSAndroid Build Coastguard Worker 	 * the logical block and return that.
157*6a54128fSAndroid Build Coastguard Worker 	 *
158*6a54128fSAndroid Build Coastguard Worker 	 * The old code wouldn't even look if (block % cluster_ratio) == 0;
159*6a54128fSAndroid Build Coastguard Worker 	 * this is incorrect if we're allocating blocks in reverse order.
160*6a54128fSAndroid Build Coastguard Worker 	 */
161*6a54128fSAndroid Build Coastguard Worker 	for (i = 0; i < EXT2FS_CLUSTER_RATIO(fs); i++) {
162*6a54128fSAndroid Build Coastguard Worker 		if (base_block + i == lblk)
163*6a54128fSAndroid Build Coastguard Worker 			continue;
164*6a54128fSAndroid Build Coastguard Worker 		extent_bmap(fs, ino, inode, handle, 0, 0,
165*6a54128fSAndroid Build Coastguard Worker 			    base_block + i, 0, 0, &pblock);
166*6a54128fSAndroid Build Coastguard Worker 		if (pblock)
167*6a54128fSAndroid Build Coastguard Worker 			break;
168*6a54128fSAndroid Build Coastguard Worker 	}
169*6a54128fSAndroid Build Coastguard Worker 	if (pblock == 0)
170*6a54128fSAndroid Build Coastguard Worker 		return 0;
171*6a54128fSAndroid Build Coastguard Worker 	*phys_blk = pblock - i + (lblk - base_block);
172*6a54128fSAndroid Build Coastguard Worker 	return 0;
173*6a54128fSAndroid Build Coastguard Worker }
174*6a54128fSAndroid Build Coastguard Worker 
175*6a54128fSAndroid Build Coastguard Worker /* Try to map a logical block to an already-allocated physical cluster. */
ext2fs_map_cluster_block(ext2_filsys fs,ext2_ino_t ino,struct ext2_inode * inode,blk64_t lblk,blk64_t * pblk)176*6a54128fSAndroid Build Coastguard Worker errcode_t ext2fs_map_cluster_block(ext2_filsys fs, ext2_ino_t ino,
177*6a54128fSAndroid Build Coastguard Worker 				   struct ext2_inode *inode, blk64_t lblk,
178*6a54128fSAndroid Build Coastguard Worker 				   blk64_t *pblk)
179*6a54128fSAndroid Build Coastguard Worker {
180*6a54128fSAndroid Build Coastguard Worker 	ext2_extent_handle_t handle;
181*6a54128fSAndroid Build Coastguard Worker 	errcode_t retval;
182*6a54128fSAndroid Build Coastguard Worker 
183*6a54128fSAndroid Build Coastguard Worker 	/* Need bigalloc and extents to be enabled */
184*6a54128fSAndroid Build Coastguard Worker 	*pblk = 0;
185*6a54128fSAndroid Build Coastguard Worker 	if (!ext2fs_has_feature_bigalloc(fs->super) ||
186*6a54128fSAndroid Build Coastguard Worker 	    !(inode->i_flags & EXT4_EXTENTS_FL))
187*6a54128fSAndroid Build Coastguard Worker 		return 0;
188*6a54128fSAndroid Build Coastguard Worker 
189*6a54128fSAndroid Build Coastguard Worker 	retval = ext2fs_extent_open2(fs, ino, inode, &handle);
190*6a54128fSAndroid Build Coastguard Worker 	if (retval)
191*6a54128fSAndroid Build Coastguard Worker 		goto out;
192*6a54128fSAndroid Build Coastguard Worker 
193*6a54128fSAndroid Build Coastguard Worker 	retval = implied_cluster_alloc(fs, ino, inode, handle, lblk, pblk);
194*6a54128fSAndroid Build Coastguard Worker 	if (retval)
195*6a54128fSAndroid Build Coastguard Worker 		goto out2;
196*6a54128fSAndroid Build Coastguard Worker 
197*6a54128fSAndroid Build Coastguard Worker out2:
198*6a54128fSAndroid Build Coastguard Worker 	ext2fs_extent_free(handle);
199*6a54128fSAndroid Build Coastguard Worker out:
200*6a54128fSAndroid Build Coastguard Worker 	return retval;
201*6a54128fSAndroid Build Coastguard Worker }
202*6a54128fSAndroid Build Coastguard Worker 
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)203*6a54128fSAndroid Build Coastguard Worker static errcode_t extent_bmap(ext2_filsys fs, ext2_ino_t ino,
204*6a54128fSAndroid Build Coastguard Worker 			     struct ext2_inode *inode,
205*6a54128fSAndroid Build Coastguard Worker 			     ext2_extent_handle_t handle,
206*6a54128fSAndroid Build Coastguard Worker 			     char *block_buf, int bmap_flags, blk64_t block,
207*6a54128fSAndroid Build Coastguard Worker 			     int *ret_flags, int *blocks_alloc,
208*6a54128fSAndroid Build Coastguard Worker 			     blk64_t *phys_blk)
209*6a54128fSAndroid Build Coastguard Worker {
210*6a54128fSAndroid Build Coastguard Worker 	struct blk_alloc_ctx	alloc_ctx;
211*6a54128fSAndroid Build Coastguard Worker 	struct ext2fs_extent	extent;
212*6a54128fSAndroid Build Coastguard Worker 	unsigned int		offset;
213*6a54128fSAndroid Build Coastguard Worker 	errcode_t		retval = 0;
214*6a54128fSAndroid Build Coastguard Worker 	blk64_t			blk64 = 0;
215*6a54128fSAndroid Build Coastguard Worker 	int			alloc = 0;
216*6a54128fSAndroid Build Coastguard Worker 	int			set_flags;
217*6a54128fSAndroid Build Coastguard Worker 
218*6a54128fSAndroid Build Coastguard Worker 	set_flags = bmap_flags & BMAP_UNINIT ? EXT2_EXTENT_SET_BMAP_UNINIT : 0;
219*6a54128fSAndroid Build Coastguard Worker 
220*6a54128fSAndroid Build Coastguard Worker 	if (bmap_flags & BMAP_SET) {
221*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_extent_set_bmap(handle, block,
222*6a54128fSAndroid Build Coastguard Worker 						*phys_blk, set_flags);
223*6a54128fSAndroid Build Coastguard Worker 		return retval;
224*6a54128fSAndroid Build Coastguard Worker 	}
225*6a54128fSAndroid Build Coastguard Worker 	retval = ext2fs_extent_goto(handle, block);
226*6a54128fSAndroid Build Coastguard Worker 	if (retval) {
227*6a54128fSAndroid Build Coastguard Worker 		/* If the extent is not found, return phys_blk = 0 */
228*6a54128fSAndroid Build Coastguard Worker 		if (retval == EXT2_ET_EXTENT_NOT_FOUND) {
229*6a54128fSAndroid Build Coastguard Worker 			extent.e_lblk = block;
230*6a54128fSAndroid Build Coastguard Worker 			goto got_block;
231*6a54128fSAndroid Build Coastguard Worker 		}
232*6a54128fSAndroid Build Coastguard Worker 		return retval;
233*6a54128fSAndroid Build Coastguard Worker 	}
234*6a54128fSAndroid Build Coastguard Worker 	retval = ext2fs_extent_get(handle, EXT2_EXTENT_CURRENT, &extent);
235*6a54128fSAndroid Build Coastguard Worker 	if (retval)
236*6a54128fSAndroid Build Coastguard Worker 		return retval;
237*6a54128fSAndroid Build Coastguard Worker 	offset = block - extent.e_lblk;
238*6a54128fSAndroid Build Coastguard Worker 	if (block >= extent.e_lblk && (offset <= extent.e_len)) {
239*6a54128fSAndroid Build Coastguard Worker 		*phys_blk = extent.e_pblk + offset;
240*6a54128fSAndroid Build Coastguard Worker 		if (ret_flags && extent.e_flags & EXT2_EXTENT_FLAGS_UNINIT)
241*6a54128fSAndroid Build Coastguard Worker 			*ret_flags |= BMAP_RET_UNINIT;
242*6a54128fSAndroid Build Coastguard Worker 	}
243*6a54128fSAndroid Build Coastguard Worker got_block:
244*6a54128fSAndroid Build Coastguard Worker 	if ((*phys_blk == 0) && (bmap_flags & BMAP_ALLOC)) {
245*6a54128fSAndroid Build Coastguard Worker 		implied_cluster_alloc(fs, ino, inode, handle, block, &blk64);
246*6a54128fSAndroid Build Coastguard Worker 		if (blk64)
247*6a54128fSAndroid Build Coastguard Worker 			goto set_extent;
248*6a54128fSAndroid Build Coastguard Worker 		retval = extent_bmap(fs, ino, inode, handle, block_buf,
249*6a54128fSAndroid Build Coastguard Worker 				     0, block-1, 0, blocks_alloc, &blk64);
250*6a54128fSAndroid Build Coastguard Worker 		if (retval)
251*6a54128fSAndroid Build Coastguard Worker 			blk64 = ext2fs_find_inode_goal(fs, ino, inode, block);
252*6a54128fSAndroid Build Coastguard Worker 		alloc_ctx.ino = ino;
253*6a54128fSAndroid Build Coastguard Worker 		alloc_ctx.inode = inode;
254*6a54128fSAndroid Build Coastguard Worker 		alloc_ctx.lblk = extent.e_lblk;
255*6a54128fSAndroid Build Coastguard Worker 		alloc_ctx.flags = BLOCK_ALLOC_DATA;
256*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_alloc_block3(fs, blk64, block_buf, &blk64,
257*6a54128fSAndroid Build Coastguard Worker 					     &alloc_ctx);
258*6a54128fSAndroid Build Coastguard Worker 		if (retval)
259*6a54128fSAndroid Build Coastguard Worker 			return retval;
260*6a54128fSAndroid Build Coastguard Worker 		blk64 &= ~EXT2FS_CLUSTER_MASK(fs);
261*6a54128fSAndroid Build Coastguard Worker 		blk64 += EXT2FS_CLUSTER_MASK(fs) & block;
262*6a54128fSAndroid Build Coastguard Worker 		alloc++;
263*6a54128fSAndroid Build Coastguard Worker 	set_extent:
264*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_extent_set_bmap(handle, block,
265*6a54128fSAndroid Build Coastguard Worker 						blk64, set_flags);
266*6a54128fSAndroid Build Coastguard Worker 		if (retval) {
267*6a54128fSAndroid Build Coastguard Worker 			ext2fs_block_alloc_stats2(fs, blk64, -1);
268*6a54128fSAndroid Build Coastguard Worker 			return retval;
269*6a54128fSAndroid Build Coastguard Worker 		}
270*6a54128fSAndroid Build Coastguard Worker 		/* Update inode after setting extent */
271*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_read_inode(fs, ino, inode);
272*6a54128fSAndroid Build Coastguard Worker 		if (retval)
273*6a54128fSAndroid Build Coastguard Worker 			return retval;
274*6a54128fSAndroid Build Coastguard Worker 		*blocks_alloc += alloc;
275*6a54128fSAndroid Build Coastguard Worker 		*phys_blk = blk64;
276*6a54128fSAndroid Build Coastguard Worker 	}
277*6a54128fSAndroid Build Coastguard Worker 	return 0;
278*6a54128fSAndroid Build Coastguard Worker }
279*6a54128fSAndroid Build Coastguard Worker 
ext2fs_file_block_offset_too_big(ext2_filsys fs,struct ext2_inode * inode,blk64_t offset)280*6a54128fSAndroid Build Coastguard Worker int ext2fs_file_block_offset_too_big(ext2_filsys fs,
281*6a54128fSAndroid Build Coastguard Worker 				     struct ext2_inode *inode,
282*6a54128fSAndroid Build Coastguard Worker 				     blk64_t offset)
283*6a54128fSAndroid Build Coastguard Worker {
284*6a54128fSAndroid Build Coastguard Worker 	blk64_t addr_per_block, max_map_block;
285*6a54128fSAndroid Build Coastguard Worker 
286*6a54128fSAndroid Build Coastguard Worker 	/* Kernel seems to cut us off at 4294967294 blocks */
287*6a54128fSAndroid Build Coastguard Worker 	if (offset >= (1ULL << 32) - 1)
288*6a54128fSAndroid Build Coastguard Worker 		return 1;
289*6a54128fSAndroid Build Coastguard Worker 
290*6a54128fSAndroid Build Coastguard Worker 	if (inode->i_flags & EXT4_EXTENTS_FL)
291*6a54128fSAndroid Build Coastguard Worker 		return 0;
292*6a54128fSAndroid Build Coastguard Worker 
293*6a54128fSAndroid Build Coastguard Worker 	addr_per_block = fs->blocksize >> 2;
294*6a54128fSAndroid Build Coastguard Worker 	max_map_block = addr_per_block;
295*6a54128fSAndroid Build Coastguard Worker 	max_map_block += addr_per_block * addr_per_block;
296*6a54128fSAndroid Build Coastguard Worker 	max_map_block += addr_per_block * addr_per_block * addr_per_block;
297*6a54128fSAndroid Build Coastguard Worker 	max_map_block += 12;
298*6a54128fSAndroid Build Coastguard Worker 
299*6a54128fSAndroid Build Coastguard Worker 	return offset >= max_map_block;
300*6a54128fSAndroid Build Coastguard Worker }
301*6a54128fSAndroid Build Coastguard Worker 
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)302*6a54128fSAndroid Build Coastguard Worker errcode_t ext2fs_bmap2(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
303*6a54128fSAndroid Build Coastguard Worker 		       char *block_buf, int bmap_flags, blk64_t block,
304*6a54128fSAndroid Build Coastguard Worker 		       int *ret_flags, blk64_t *phys_blk)
305*6a54128fSAndroid Build Coastguard Worker {
306*6a54128fSAndroid Build Coastguard Worker 	struct ext2_inode inode_buf;
307*6a54128fSAndroid Build Coastguard Worker 	ext2_extent_handle_t handle = 0;
308*6a54128fSAndroid Build Coastguard Worker 	blk_t addr_per_block;
309*6a54128fSAndroid Build Coastguard Worker 	blk_t	b, blk32;
310*6a54128fSAndroid Build Coastguard Worker 	blk64_t b64;
311*6a54128fSAndroid Build Coastguard Worker 	char	*buf = 0;
312*6a54128fSAndroid Build Coastguard Worker 	errcode_t	retval = 0;
313*6a54128fSAndroid Build Coastguard Worker 	int		blocks_alloc = 0, inode_dirty = 0;
314*6a54128fSAndroid Build Coastguard Worker 	struct blk_alloc_ctx alloc_ctx = {
315*6a54128fSAndroid Build Coastguard Worker 		.ino	= ino,
316*6a54128fSAndroid Build Coastguard Worker 		.inode	= inode,
317*6a54128fSAndroid Build Coastguard Worker 		.lblk	= 0,
318*6a54128fSAndroid Build Coastguard Worker 		.flags	= BLOCK_ALLOC_DATA,
319*6a54128fSAndroid Build Coastguard Worker 	};
320*6a54128fSAndroid Build Coastguard Worker 
321*6a54128fSAndroid Build Coastguard Worker 	if (!(bmap_flags & BMAP_SET))
322*6a54128fSAndroid Build Coastguard Worker 		*phys_blk = 0;
323*6a54128fSAndroid Build Coastguard Worker 
324*6a54128fSAndroid Build Coastguard Worker 	if (ret_flags)
325*6a54128fSAndroid Build Coastguard Worker 		*ret_flags = 0;
326*6a54128fSAndroid Build Coastguard Worker 
327*6a54128fSAndroid Build Coastguard Worker 	/* Read inode structure if necessary */
328*6a54128fSAndroid Build Coastguard Worker 	if (!inode) {
329*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_read_inode(fs, ino, &inode_buf);
330*6a54128fSAndroid Build Coastguard Worker 		if (retval)
331*6a54128fSAndroid Build Coastguard Worker 			return retval;
332*6a54128fSAndroid Build Coastguard Worker 		inode = &inode_buf;
333*6a54128fSAndroid Build Coastguard Worker 	}
334*6a54128fSAndroid Build Coastguard Worker 	addr_per_block = (blk_t) fs->blocksize >> 2;
335*6a54128fSAndroid Build Coastguard Worker 
336*6a54128fSAndroid Build Coastguard Worker 	if (ext2fs_file_block_offset_too_big(fs, inode, block))
337*6a54128fSAndroid Build Coastguard Worker 		return EXT2_ET_FILE_TOO_BIG;
338*6a54128fSAndroid Build Coastguard Worker 
339*6a54128fSAndroid Build Coastguard Worker 	/*
340*6a54128fSAndroid Build Coastguard Worker 	 * If an inode has inline data, that means that it doesn't have
341*6a54128fSAndroid Build Coastguard Worker 	 * any blocks and we shouldn't map any blocks for it.
342*6a54128fSAndroid Build Coastguard Worker 	 */
343*6a54128fSAndroid Build Coastguard Worker 	if (inode->i_flags & EXT4_INLINE_DATA_FL)
344*6a54128fSAndroid Build Coastguard Worker 		return EXT2_ET_INLINE_DATA_NO_BLOCK;
345*6a54128fSAndroid Build Coastguard Worker 
346*6a54128fSAndroid Build Coastguard Worker 	if (!block_buf) {
347*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_get_array(2, fs->blocksize, &buf);
348*6a54128fSAndroid Build Coastguard Worker 		if (retval)
349*6a54128fSAndroid Build Coastguard Worker 			return retval;
350*6a54128fSAndroid Build Coastguard Worker 		block_buf = buf;
351*6a54128fSAndroid Build Coastguard Worker 	}
352*6a54128fSAndroid Build Coastguard Worker 
353*6a54128fSAndroid Build Coastguard Worker 	if (inode->i_flags & EXT4_EXTENTS_FL) {
354*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_extent_open2(fs, ino, inode, &handle);
355*6a54128fSAndroid Build Coastguard Worker 		if (retval)
356*6a54128fSAndroid Build Coastguard Worker 			goto done;
357*6a54128fSAndroid Build Coastguard Worker 		retval = extent_bmap(fs, ino, inode, handle, block_buf,
358*6a54128fSAndroid Build Coastguard Worker 				     bmap_flags, block, ret_flags,
359*6a54128fSAndroid Build Coastguard Worker 				     &blocks_alloc, phys_blk);
360*6a54128fSAndroid Build Coastguard Worker 		goto done;
361*6a54128fSAndroid Build Coastguard Worker 	}
362*6a54128fSAndroid Build Coastguard Worker 
363*6a54128fSAndroid Build Coastguard Worker 	if (block < EXT2_NDIR_BLOCKS) {
364*6a54128fSAndroid Build Coastguard Worker 		if (bmap_flags & BMAP_SET) {
365*6a54128fSAndroid Build Coastguard Worker 			b = *phys_blk;
366*6a54128fSAndroid Build Coastguard Worker 			inode_bmap(inode, block) = b;
367*6a54128fSAndroid Build Coastguard Worker 			inode_dirty++;
368*6a54128fSAndroid Build Coastguard Worker 			goto done;
369*6a54128fSAndroid Build Coastguard Worker 		}
370*6a54128fSAndroid Build Coastguard Worker 
371*6a54128fSAndroid Build Coastguard Worker 		*phys_blk = inode_bmap(inode, block);
372*6a54128fSAndroid Build Coastguard Worker 		b = block ? inode_bmap(inode, block - 1) :
373*6a54128fSAndroid Build Coastguard Worker 			    ext2fs_find_inode_goal(fs, ino, inode, block);
374*6a54128fSAndroid Build Coastguard Worker 
375*6a54128fSAndroid Build Coastguard Worker 		if ((*phys_blk == 0) && (bmap_flags & BMAP_ALLOC)) {
376*6a54128fSAndroid Build Coastguard Worker 			b64 = b;
377*6a54128fSAndroid Build Coastguard Worker 			retval = ext2fs_alloc_block3(fs, b64, block_buf, &b64,
378*6a54128fSAndroid Build Coastguard Worker 						     &alloc_ctx);
379*6a54128fSAndroid Build Coastguard Worker 			b = b64;
380*6a54128fSAndroid Build Coastguard Worker 			if (retval)
381*6a54128fSAndroid Build Coastguard Worker 				goto done;
382*6a54128fSAndroid Build Coastguard Worker 			inode_bmap(inode, block) = b;
383*6a54128fSAndroid Build Coastguard Worker 			blocks_alloc++;
384*6a54128fSAndroid Build Coastguard Worker 			*phys_blk = b;
385*6a54128fSAndroid Build Coastguard Worker 		}
386*6a54128fSAndroid Build Coastguard Worker 		goto done;
387*6a54128fSAndroid Build Coastguard Worker 	}
388*6a54128fSAndroid Build Coastguard Worker 
389*6a54128fSAndroid Build Coastguard Worker 	/* Indirect block */
390*6a54128fSAndroid Build Coastguard Worker 	block -= EXT2_NDIR_BLOCKS;
391*6a54128fSAndroid Build Coastguard Worker 	blk32 = *phys_blk;
392*6a54128fSAndroid Build Coastguard Worker 	if (block < addr_per_block) {
393*6a54128fSAndroid Build Coastguard Worker 		b = inode_bmap(inode, EXT2_IND_BLOCK);
394*6a54128fSAndroid Build Coastguard Worker 		if (!b) {
395*6a54128fSAndroid Build Coastguard Worker 			if (!(bmap_flags & BMAP_ALLOC)) {
396*6a54128fSAndroid Build Coastguard Worker 				if (bmap_flags & BMAP_SET)
397*6a54128fSAndroid Build Coastguard Worker 					retval = EXT2_ET_SET_BMAP_NO_IND;
398*6a54128fSAndroid Build Coastguard Worker 				goto done;
399*6a54128fSAndroid Build Coastguard Worker 			}
400*6a54128fSAndroid Build Coastguard Worker 
401*6a54128fSAndroid Build Coastguard Worker 			b = inode_bmap(inode, EXT2_IND_BLOCK-1);
402*6a54128fSAndroid Build Coastguard Worker 			b64 = b;
403*6a54128fSAndroid Build Coastguard Worker 			retval = ext2fs_alloc_block3(fs, b64, block_buf, &b64,
404*6a54128fSAndroid Build Coastguard Worker 						     &alloc_ctx);
405*6a54128fSAndroid Build Coastguard Worker 			b = b64;
406*6a54128fSAndroid Build Coastguard Worker 			if (retval)
407*6a54128fSAndroid Build Coastguard Worker 				goto done;
408*6a54128fSAndroid Build Coastguard Worker 			inode_bmap(inode, EXT2_IND_BLOCK) = b;
409*6a54128fSAndroid Build Coastguard Worker 			blocks_alloc++;
410*6a54128fSAndroid Build Coastguard Worker 		}
411*6a54128fSAndroid Build Coastguard Worker 		retval = block_ind_bmap(fs, bmap_flags, b, block_buf,
412*6a54128fSAndroid Build Coastguard Worker 					&blocks_alloc, block, &blk32);
413*6a54128fSAndroid Build Coastguard Worker 		if (retval == 0)
414*6a54128fSAndroid Build Coastguard Worker 			*phys_blk = blk32;
415*6a54128fSAndroid Build Coastguard Worker 		goto done;
416*6a54128fSAndroid Build Coastguard Worker 	}
417*6a54128fSAndroid Build Coastguard Worker 
418*6a54128fSAndroid Build Coastguard Worker 	/* Doubly indirect block  */
419*6a54128fSAndroid Build Coastguard Worker 	block -= addr_per_block;
420*6a54128fSAndroid Build Coastguard Worker 	if (block < addr_per_block * addr_per_block) {
421*6a54128fSAndroid Build Coastguard Worker 		b = inode_bmap(inode, EXT2_DIND_BLOCK);
422*6a54128fSAndroid Build Coastguard Worker 		if (!b) {
423*6a54128fSAndroid Build Coastguard Worker 			if (!(bmap_flags & BMAP_ALLOC)) {
424*6a54128fSAndroid Build Coastguard Worker 				if (bmap_flags & BMAP_SET)
425*6a54128fSAndroid Build Coastguard Worker 					retval = EXT2_ET_SET_BMAP_NO_IND;
426*6a54128fSAndroid Build Coastguard Worker 				goto done;
427*6a54128fSAndroid Build Coastguard Worker 			}
428*6a54128fSAndroid Build Coastguard Worker 
429*6a54128fSAndroid Build Coastguard Worker 			b = inode_bmap(inode, EXT2_IND_BLOCK);
430*6a54128fSAndroid Build Coastguard Worker 			b64 = b;
431*6a54128fSAndroid Build Coastguard Worker 			retval = ext2fs_alloc_block3(fs, b64, block_buf, &b64,
432*6a54128fSAndroid Build Coastguard Worker 						     &alloc_ctx);
433*6a54128fSAndroid Build Coastguard Worker 			b = b64;
434*6a54128fSAndroid Build Coastguard Worker 			if (retval)
435*6a54128fSAndroid Build Coastguard Worker 				goto done;
436*6a54128fSAndroid Build Coastguard Worker 			inode_bmap(inode, EXT2_DIND_BLOCK) = b;
437*6a54128fSAndroid Build Coastguard Worker 			blocks_alloc++;
438*6a54128fSAndroid Build Coastguard Worker 		}
439*6a54128fSAndroid Build Coastguard Worker 		retval = block_dind_bmap(fs, bmap_flags, b, block_buf,
440*6a54128fSAndroid Build Coastguard Worker 					 &blocks_alloc, block, &blk32);
441*6a54128fSAndroid Build Coastguard Worker 		if (retval == 0)
442*6a54128fSAndroid Build Coastguard Worker 			*phys_blk = blk32;
443*6a54128fSAndroid Build Coastguard Worker 		goto done;
444*6a54128fSAndroid Build Coastguard Worker 	}
445*6a54128fSAndroid Build Coastguard Worker 
446*6a54128fSAndroid Build Coastguard Worker 	/* Triply indirect block */
447*6a54128fSAndroid Build Coastguard Worker 	block -= addr_per_block * addr_per_block;
448*6a54128fSAndroid Build Coastguard Worker 	b = inode_bmap(inode, EXT2_TIND_BLOCK);
449*6a54128fSAndroid Build Coastguard Worker 	if (!b) {
450*6a54128fSAndroid Build Coastguard Worker 		if (!(bmap_flags & BMAP_ALLOC)) {
451*6a54128fSAndroid Build Coastguard Worker 			if (bmap_flags & BMAP_SET)
452*6a54128fSAndroid Build Coastguard Worker 				retval = EXT2_ET_SET_BMAP_NO_IND;
453*6a54128fSAndroid Build Coastguard Worker 			goto done;
454*6a54128fSAndroid Build Coastguard Worker 		}
455*6a54128fSAndroid Build Coastguard Worker 
456*6a54128fSAndroid Build Coastguard Worker 		b = inode_bmap(inode, EXT2_DIND_BLOCK);
457*6a54128fSAndroid Build Coastguard Worker 		b64 = b;
458*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_alloc_block3(fs, b64, block_buf, &b64,
459*6a54128fSAndroid Build Coastguard Worker 					     &alloc_ctx);
460*6a54128fSAndroid Build Coastguard Worker 		b = b64;
461*6a54128fSAndroid Build Coastguard Worker 		if (retval)
462*6a54128fSAndroid Build Coastguard Worker 			goto done;
463*6a54128fSAndroid Build Coastguard Worker 		inode_bmap(inode, EXT2_TIND_BLOCK) = b;
464*6a54128fSAndroid Build Coastguard Worker 		blocks_alloc++;
465*6a54128fSAndroid Build Coastguard Worker 	}
466*6a54128fSAndroid Build Coastguard Worker 	retval = block_tind_bmap(fs, bmap_flags, b, block_buf,
467*6a54128fSAndroid Build Coastguard Worker 				 &blocks_alloc, block, &blk32);
468*6a54128fSAndroid Build Coastguard Worker 	if (retval == 0)
469*6a54128fSAndroid Build Coastguard Worker 		*phys_blk = blk32;
470*6a54128fSAndroid Build Coastguard Worker done:
471*6a54128fSAndroid Build Coastguard Worker 	if (*phys_blk && retval == 0 && (bmap_flags & BMAP_ZERO))
472*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_zero_blocks2(fs, *phys_blk, 1, NULL, NULL);
473*6a54128fSAndroid Build Coastguard Worker 	if (buf)
474*6a54128fSAndroid Build Coastguard Worker 		ext2fs_free_mem(&buf);
475*6a54128fSAndroid Build Coastguard Worker 	if (handle)
476*6a54128fSAndroid Build Coastguard Worker 		ext2fs_extent_free(handle);
477*6a54128fSAndroid Build Coastguard Worker 	if ((retval == 0) && (blocks_alloc || inode_dirty)) {
478*6a54128fSAndroid Build Coastguard Worker 		ext2fs_iblk_add_blocks(fs, inode, blocks_alloc);
479*6a54128fSAndroid Build Coastguard Worker 		retval = ext2fs_write_inode(fs, ino, inode);
480*6a54128fSAndroid Build Coastguard Worker 	}
481*6a54128fSAndroid Build Coastguard Worker 	return retval;
482*6a54128fSAndroid Build Coastguard Worker }
483*6a54128fSAndroid Build Coastguard Worker 
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)484*6a54128fSAndroid Build Coastguard Worker errcode_t ext2fs_bmap(ext2_filsys fs, ext2_ino_t ino, struct ext2_inode *inode,
485*6a54128fSAndroid Build Coastguard Worker 		      char *block_buf, int bmap_flags, blk_t block,
486*6a54128fSAndroid Build Coastguard Worker 		      blk_t *phys_blk)
487*6a54128fSAndroid Build Coastguard Worker {
488*6a54128fSAndroid Build Coastguard Worker 	errcode_t ret;
489*6a54128fSAndroid Build Coastguard Worker 	blk64_t	ret_blk = *phys_blk;
490*6a54128fSAndroid Build Coastguard Worker 
491*6a54128fSAndroid Build Coastguard Worker 	ret = ext2fs_bmap2(fs, ino, inode, block_buf, bmap_flags, block,
492*6a54128fSAndroid Build Coastguard Worker 			    0, &ret_blk);
493*6a54128fSAndroid Build Coastguard Worker 	if (ret)
494*6a54128fSAndroid Build Coastguard Worker 		return ret;
495*6a54128fSAndroid Build Coastguard Worker 	if (ret_blk >= ((long long) 1 << 32))
496*6a54128fSAndroid Build Coastguard Worker 		return EOVERFLOW;
497*6a54128fSAndroid Build Coastguard Worker 	*phys_blk = ret_blk;
498*6a54128fSAndroid Build Coastguard Worker 	return 0;
499*6a54128fSAndroid Build Coastguard Worker }
500