xref: /onnv-gate/usr/src/uts/common/fs/zfs/dnode.c (revision 1544)
1789Sahrens /*
2789Sahrens  * CDDL HEADER START
3789Sahrens  *
4789Sahrens  * The contents of this file are subject to the terms of the
51491Sahrens  * Common Development and Distribution License (the "License").
61491Sahrens  * You may not use this file except in compliance with the License.
7789Sahrens  *
8789Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9789Sahrens  * or http://www.opensolaris.org/os/licensing.
10789Sahrens  * See the License for the specific language governing permissions
11789Sahrens  * and limitations under the License.
12789Sahrens  *
13789Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14789Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15789Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16789Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17789Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18789Sahrens  *
19789Sahrens  * CDDL HEADER END
20789Sahrens  */
21789Sahrens /*
221402Sahrens  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
23789Sahrens  * Use is subject to license terms.
24789Sahrens  */
25789Sahrens 
26789Sahrens #pragma ident	"%Z%%M%	%I%	%E% SMI"
27789Sahrens 
28789Sahrens #include <sys/zfs_context.h>
29789Sahrens #include <sys/dbuf.h>
30789Sahrens #include <sys/dnode.h>
31789Sahrens #include <sys/dmu.h>
32789Sahrens #include <sys/dmu_impl.h>
33789Sahrens #include <sys/dmu_tx.h>
34789Sahrens #include <sys/dmu_objset.h>
35789Sahrens #include <sys/dsl_dir.h>
36789Sahrens #include <sys/dsl_dataset.h>
37789Sahrens #include <sys/spa.h>
38789Sahrens #include <sys/zio.h>
39789Sahrens #include <sys/dmu_zfetch.h>
40789Sahrens 
41789Sahrens static int free_range_compar(const void *node1, const void *node2);
42789Sahrens 
43789Sahrens static kmem_cache_t *dnode_cache;
44789Sahrens 
45789Sahrens static dnode_phys_t dnode_phys_zero;
46789Sahrens 
47789Sahrens int zfs_default_bs = SPA_MINBLOCKSHIFT;
48789Sahrens int zfs_default_ibs = DN_MAX_INDBLKSHIFT;
49789Sahrens 
50789Sahrens /* ARGSUSED */
51789Sahrens static int
52789Sahrens dnode_cons(void *arg, void *unused, int kmflag)
53789Sahrens {
54789Sahrens 	int i;
55789Sahrens 	dnode_t *dn = arg;
56789Sahrens 	bzero(dn, sizeof (dnode_t));
57789Sahrens 
58789Sahrens 	rw_init(&dn->dn_struct_rwlock, NULL, RW_DEFAULT, NULL);
59789Sahrens 	mutex_init(&dn->dn_mtx, NULL, MUTEX_DEFAULT, NULL);
60789Sahrens 	mutex_init(&dn->dn_dbufs_mtx, NULL, MUTEX_DEFAULT, NULL);
61789Sahrens 	refcount_create(&dn->dn_holds);
62789Sahrens 	refcount_create(&dn->dn_tx_holds);
63789Sahrens 
64789Sahrens 	for (i = 0; i < TXG_SIZE; i++) {
65789Sahrens 		avl_create(&dn->dn_ranges[i], free_range_compar,
66789Sahrens 		    sizeof (free_range_t),
67789Sahrens 		    offsetof(struct free_range, fr_node));
68789Sahrens 		list_create(&dn->dn_dirty_dbufs[i],
69789Sahrens 		    sizeof (dmu_buf_impl_t),
70789Sahrens 		    offsetof(dmu_buf_impl_t, db_dirty_node[i]));
71789Sahrens 	}
72789Sahrens 
73789Sahrens 	list_create(&dn->dn_dbufs, sizeof (dmu_buf_impl_t),
74789Sahrens 	    offsetof(dmu_buf_impl_t, db_link));
75789Sahrens 
76789Sahrens 	return (0);
77789Sahrens }
78789Sahrens 
79789Sahrens /* ARGSUSED */
80789Sahrens static void
81789Sahrens dnode_dest(void *arg, void *unused)
82789Sahrens {
83789Sahrens 	int i;
84789Sahrens 	dnode_t *dn = arg;
85789Sahrens 
86789Sahrens 	rw_destroy(&dn->dn_struct_rwlock);
87789Sahrens 	mutex_destroy(&dn->dn_mtx);
88789Sahrens 	mutex_destroy(&dn->dn_dbufs_mtx);
89789Sahrens 	refcount_destroy(&dn->dn_holds);
90789Sahrens 	refcount_destroy(&dn->dn_tx_holds);
91789Sahrens 
92789Sahrens 	for (i = 0; i < TXG_SIZE; i++) {
93789Sahrens 		avl_destroy(&dn->dn_ranges[i]);
94789Sahrens 		list_destroy(&dn->dn_dirty_dbufs[i]);
95789Sahrens 	}
96789Sahrens 
97789Sahrens 	list_destroy(&dn->dn_dbufs);
98789Sahrens }
99789Sahrens 
100789Sahrens void
101789Sahrens dnode_init(void)
102789Sahrens {
103789Sahrens 	dnode_cache = kmem_cache_create("dnode_t",
104789Sahrens 	    sizeof (dnode_t),
105789Sahrens 	    0, dnode_cons, dnode_dest, NULL, NULL, NULL, 0);
106789Sahrens }
107789Sahrens 
108789Sahrens void
109789Sahrens dnode_fini(void)
110789Sahrens {
111789Sahrens 	kmem_cache_destroy(dnode_cache);
112789Sahrens }
113789Sahrens 
114789Sahrens 
115873Sek110237 #ifdef ZFS_DEBUG
116789Sahrens void
117789Sahrens dnode_verify(dnode_t *dn)
118789Sahrens {
119789Sahrens 	int drop_struct_lock = FALSE;
120789Sahrens 
121789Sahrens 	ASSERT(dn->dn_phys);
122789Sahrens 	ASSERT(dn->dn_objset);
123789Sahrens 
124789Sahrens 	ASSERT(dn->dn_phys->dn_type < DMU_OT_NUMTYPES);
125789Sahrens 
126789Sahrens 	if (!(zfs_flags & ZFS_DEBUG_DNODE_VERIFY))
127789Sahrens 		return;
128789Sahrens 
129789Sahrens 	if (!RW_WRITE_HELD(&dn->dn_struct_rwlock)) {
130789Sahrens 		rw_enter(&dn->dn_struct_rwlock, RW_READER);
131789Sahrens 		drop_struct_lock = TRUE;
132789Sahrens 	}
133789Sahrens 	if (dn->dn_phys->dn_type != DMU_OT_NONE || dn->dn_allocated_txg != 0) {
134789Sahrens 		int i;
135789Sahrens 		ASSERT3U(dn->dn_indblkshift, >=, 0);
136789Sahrens 		ASSERT3U(dn->dn_indblkshift, <=, SPA_MAXBLOCKSHIFT);
137789Sahrens 		if (dn->dn_datablkshift) {
138789Sahrens 			ASSERT3U(dn->dn_datablkshift, >=, SPA_MINBLOCKSHIFT);
139789Sahrens 			ASSERT3U(dn->dn_datablkshift, <=, SPA_MAXBLOCKSHIFT);
140789Sahrens 			ASSERT3U(1<<dn->dn_datablkshift, ==, dn->dn_datablksz);
141789Sahrens 		}
142789Sahrens 		ASSERT3U(dn->dn_nlevels, <=, 30);
143789Sahrens 		ASSERT3U(dn->dn_type, <=, DMU_OT_NUMTYPES);
144789Sahrens 		ASSERT3U(dn->dn_nblkptr, >=, 1);
145789Sahrens 		ASSERT3U(dn->dn_nblkptr, <=, DN_MAX_NBLKPTR);
146789Sahrens 		ASSERT3U(dn->dn_bonuslen, <=, DN_MAX_BONUSLEN);
147789Sahrens 		ASSERT3U(dn->dn_datablksz, ==,
148789Sahrens 		    dn->dn_datablkszsec << SPA_MINBLOCKSHIFT);
149789Sahrens 		ASSERT3U(ISP2(dn->dn_datablksz), ==, dn->dn_datablkshift != 0);
150789Sahrens 		ASSERT3U((dn->dn_nblkptr - 1) * sizeof (blkptr_t) +
151789Sahrens 		    dn->dn_bonuslen, <=, DN_MAX_BONUSLEN);
152789Sahrens 		for (i = 0; i < TXG_SIZE; i++) {
153789Sahrens 			ASSERT3U(dn->dn_next_nlevels[i], <=, dn->dn_nlevels);
154789Sahrens 		}
155789Sahrens 	}
156789Sahrens 	if (dn->dn_phys->dn_type != DMU_OT_NONE)
157789Sahrens 		ASSERT3U(dn->dn_phys->dn_nlevels, <=, dn->dn_nlevels);
158*1544Seschrock 	ASSERT(dn->dn_object == DMU_META_DNODE_OBJECT || dn->dn_dbuf != NULL);
159789Sahrens 	if (dn->dn_dbuf != NULL) {
160789Sahrens 		ASSERT3P(dn->dn_phys, ==,
161789Sahrens 		    (dnode_phys_t *)dn->dn_dbuf->db.db_data +
162789Sahrens 		    (dn->dn_object % (dn->dn_dbuf->db.db_size >> DNODE_SHIFT)));
163789Sahrens 	}
164789Sahrens 	if (drop_struct_lock)
165789Sahrens 		rw_exit(&dn->dn_struct_rwlock);
166873Sek110237 }
167789Sahrens #endif
168789Sahrens 
169789Sahrens void
170789Sahrens dnode_byteswap(dnode_phys_t *dnp)
171789Sahrens {
172789Sahrens 	uint64_t *buf64 = (void*)&dnp->dn_blkptr;
173789Sahrens 	int i;
174789Sahrens 
175789Sahrens 	if (dnp->dn_type == DMU_OT_NONE) {
176789Sahrens 		bzero(dnp, sizeof (dnode_phys_t));
177789Sahrens 		return;
178789Sahrens 	}
179789Sahrens 
180789Sahrens 	dnp->dn_type = BSWAP_8(dnp->dn_type);
181789Sahrens 	dnp->dn_indblkshift = BSWAP_8(dnp->dn_indblkshift);
182789Sahrens 	dnp->dn_nlevels = BSWAP_8(dnp->dn_nlevels);
183789Sahrens 	dnp->dn_nblkptr = BSWAP_8(dnp->dn_nblkptr);
184789Sahrens 	dnp->dn_bonustype = BSWAP_8(dnp->dn_bonustype);
185789Sahrens 	dnp->dn_checksum = BSWAP_8(dnp->dn_checksum);
186789Sahrens 	dnp->dn_compress = BSWAP_8(dnp->dn_compress);
187789Sahrens 	dnp->dn_datablkszsec = BSWAP_16(dnp->dn_datablkszsec);
188789Sahrens 	dnp->dn_bonuslen = BSWAP_16(dnp->dn_bonuslen);
189789Sahrens 	dnp->dn_maxblkid = BSWAP_64(dnp->dn_maxblkid);
190789Sahrens 	dnp->dn_secphys = BSWAP_64(dnp->dn_secphys);
191789Sahrens 
192789Sahrens 	/*
193789Sahrens 	 * dn_nblkptr is only one byte, so it's OK to read it in either
194789Sahrens 	 * byte order.  We can't read dn_bouslen.
195789Sahrens 	 */
196789Sahrens 	ASSERT(dnp->dn_indblkshift <= SPA_MAXBLOCKSHIFT);
197789Sahrens 	ASSERT(dnp->dn_nblkptr <= DN_MAX_NBLKPTR);
198789Sahrens 	for (i = 0; i < dnp->dn_nblkptr * sizeof (blkptr_t)/8; i++)
199789Sahrens 		buf64[i] = BSWAP_64(buf64[i]);
200789Sahrens 
201789Sahrens 	/*
202789Sahrens 	 * OK to check dn_bonuslen for zero, because it won't matter if
203789Sahrens 	 * we have the wrong byte order.  This is necessary because the
204789Sahrens 	 * dnode dnode is smaller than a regular dnode.
205789Sahrens 	 */
206789Sahrens 	if (dnp->dn_bonuslen != 0) {
207789Sahrens 		/*
208789Sahrens 		 * Note that the bonus length calculated here may be
209789Sahrens 		 * longer than the actual bonus buffer.  This is because
210789Sahrens 		 * we always put the bonus buffer after the last block
211789Sahrens 		 * pointer (instead of packing it against the end of the
212789Sahrens 		 * dnode buffer).
213789Sahrens 		 */
214789Sahrens 		int off = (dnp->dn_nblkptr-1) * sizeof (blkptr_t);
215789Sahrens 		size_t len = DN_MAX_BONUSLEN - off;
216789Sahrens 		dmu_ot[dnp->dn_bonustype].ot_byteswap(dnp->dn_bonus + off, len);
217789Sahrens 	}
218789Sahrens }
219789Sahrens 
220789Sahrens void
221789Sahrens dnode_buf_byteswap(void *vbuf, size_t size)
222789Sahrens {
223789Sahrens 	dnode_phys_t *buf = vbuf;
224789Sahrens 	int i;
225789Sahrens 
226789Sahrens 	ASSERT3U(sizeof (dnode_phys_t), ==, (1<<DNODE_SHIFT));
227789Sahrens 	ASSERT((size & (sizeof (dnode_phys_t)-1)) == 0);
228789Sahrens 
229789Sahrens 	size >>= DNODE_SHIFT;
230789Sahrens 	for (i = 0; i < size; i++) {
231789Sahrens 		dnode_byteswap(buf);
232789Sahrens 		buf++;
233789Sahrens 	}
234789Sahrens }
235789Sahrens 
236789Sahrens static int
237789Sahrens free_range_compar(const void *node1, const void *node2)
238789Sahrens {
239789Sahrens 	const free_range_t *rp1 = node1;
240789Sahrens 	const free_range_t *rp2 = node2;
241789Sahrens 
242789Sahrens 	if (rp1->fr_blkid < rp2->fr_blkid)
243789Sahrens 		return (-1);
244789Sahrens 	else if (rp1->fr_blkid > rp2->fr_blkid)
245789Sahrens 		return (1);
246789Sahrens 	else return (0);
247789Sahrens }
248789Sahrens 
249789Sahrens static void
250789Sahrens dnode_setdblksz(dnode_t *dn, int size)
251789Sahrens {
252789Sahrens 	ASSERT3U(P2PHASE(size, SPA_MINBLOCKSIZE), ==, 0);
253789Sahrens 	ASSERT3U(size, <=, SPA_MAXBLOCKSIZE);
254789Sahrens 	ASSERT3U(size, >=, SPA_MINBLOCKSIZE);
255789Sahrens 	ASSERT3U(size >> SPA_MINBLOCKSHIFT, <,
256789Sahrens 	    1<<(sizeof (dn->dn_phys->dn_datablkszsec) * 8));
257789Sahrens 	dn->dn_datablksz = size;
258789Sahrens 	dn->dn_datablkszsec = size >> SPA_MINBLOCKSHIFT;
259789Sahrens 	dn->dn_datablkshift = ISP2(size) ? highbit(size - 1) : 0;
260789Sahrens }
261789Sahrens 
262789Sahrens static dnode_t *
263789Sahrens dnode_create(objset_impl_t *os, dnode_phys_t *dnp, dmu_buf_impl_t *db,
264789Sahrens     uint64_t object)
265789Sahrens {
266789Sahrens 	dnode_t *dn = kmem_cache_alloc(dnode_cache, KM_SLEEP);
267789Sahrens 	(void) dnode_cons(dn, NULL, 0); /* XXX */
268789Sahrens 
269789Sahrens 	dn->dn_objset = os;
270789Sahrens 	dn->dn_object = object;
271789Sahrens 	dn->dn_dbuf = db;
272789Sahrens 	dn->dn_phys = dnp;
273789Sahrens 
274789Sahrens 	if (dnp->dn_datablkszsec)
275789Sahrens 		dnode_setdblksz(dn, dnp->dn_datablkszsec << SPA_MINBLOCKSHIFT);
276789Sahrens 	dn->dn_indblkshift = dnp->dn_indblkshift;
277789Sahrens 	dn->dn_nlevels = dnp->dn_nlevels;
278789Sahrens 	dn->dn_type = dnp->dn_type;
279789Sahrens 	dn->dn_nblkptr = dnp->dn_nblkptr;
280789Sahrens 	dn->dn_checksum = dnp->dn_checksum;
281789Sahrens 	dn->dn_compress = dnp->dn_compress;
282789Sahrens 	dn->dn_bonustype = dnp->dn_bonustype;
283789Sahrens 	dn->dn_bonuslen = dnp->dn_bonuslen;
284789Sahrens 	dn->dn_maxblkid = dnp->dn_maxblkid;
285789Sahrens 
286789Sahrens 	dmu_zfetch_init(&dn->dn_zfetch, dn);
287789Sahrens 
288789Sahrens 	ASSERT(dn->dn_phys->dn_type < DMU_OT_NUMTYPES);
289789Sahrens 	mutex_enter(&os->os_lock);
290789Sahrens 	list_insert_head(&os->os_dnodes, dn);
291789Sahrens 	mutex_exit(&os->os_lock);
292789Sahrens 
293789Sahrens 	return (dn);
294789Sahrens }
295789Sahrens 
296789Sahrens static void
297789Sahrens dnode_destroy(dnode_t *dn)
298789Sahrens {
299789Sahrens 	objset_impl_t *os = dn->dn_objset;
300789Sahrens 
301789Sahrens 	mutex_enter(&os->os_lock);
302789Sahrens 	list_remove(&os->os_dnodes, dn);
303789Sahrens 	mutex_exit(&os->os_lock);
304789Sahrens 
305789Sahrens 	if (dn->dn_dirtyctx_firstset) {
306789Sahrens 		kmem_free(dn->dn_dirtyctx_firstset, 1);
307789Sahrens 		dn->dn_dirtyctx_firstset = NULL;
308789Sahrens 	}
309789Sahrens 	dmu_zfetch_rele(&dn->dn_zfetch);
310*1544Seschrock 	if (dn->dn_bonus) {
311*1544Seschrock 		mutex_enter(&dn->dn_bonus->db_mtx);
312*1544Seschrock 		dbuf_evict(dn->dn_bonus);
313*1544Seschrock 		dn->dn_bonus = NULL;
314*1544Seschrock 	}
315789Sahrens 	kmem_cache_free(dnode_cache, dn);
316789Sahrens }
317789Sahrens 
318789Sahrens void
319789Sahrens dnode_allocate(dnode_t *dn, dmu_object_type_t ot, int blocksize, int ibs,
320789Sahrens 	dmu_object_type_t bonustype, int bonuslen, dmu_tx_t *tx)
321789Sahrens {
322789Sahrens 	int i;
323789Sahrens 
324789Sahrens 	if (blocksize == 0)
325789Sahrens 		blocksize = 1 << zfs_default_bs;
3261402Sahrens 	else if (blocksize > SPA_MAXBLOCKSIZE)
3271402Sahrens 		blocksize = SPA_MAXBLOCKSIZE;
3281402Sahrens 	else
3291402Sahrens 		blocksize = P2ROUNDUP(blocksize, SPA_MINBLOCKSIZE);
330789Sahrens 
331789Sahrens 	if (ibs == 0)
332789Sahrens 		ibs = zfs_default_ibs;
333789Sahrens 
334789Sahrens 	ibs = MIN(MAX(ibs, DN_MIN_INDBLKSHIFT), DN_MAX_INDBLKSHIFT);
335789Sahrens 
336789Sahrens 	dprintf("os=%p obj=%llu txg=%llu blocksize=%d ibs=%d\n", dn->dn_objset,
337789Sahrens 	    dn->dn_object, tx->tx_txg, blocksize, ibs);
338789Sahrens 
339789Sahrens 	ASSERT(dn->dn_type == DMU_OT_NONE);
340789Sahrens 	ASSERT(bcmp(dn->dn_phys, &dnode_phys_zero, sizeof (dnode_phys_t)) == 0);
341789Sahrens 	ASSERT(dn->dn_phys->dn_type == DMU_OT_NONE);
342789Sahrens 	ASSERT(ot != DMU_OT_NONE);
343789Sahrens 	ASSERT3U(ot, <, DMU_OT_NUMTYPES);
344789Sahrens 	ASSERT((bonustype == DMU_OT_NONE && bonuslen == 0) ||
345789Sahrens 	    (bonustype != DMU_OT_NONE && bonuslen != 0));
346789Sahrens 	ASSERT3U(bonustype, <, DMU_OT_NUMTYPES);
347789Sahrens 	ASSERT3U(bonuslen, <=, DN_MAX_BONUSLEN);
348789Sahrens 	ASSERT(dn->dn_type == DMU_OT_NONE);
349789Sahrens 	ASSERT3U(dn->dn_maxblkid, ==, 0);
350789Sahrens 	ASSERT3U(dn->dn_allocated_txg, ==, 0);
351789Sahrens 	ASSERT3U(dn->dn_assigned_txg, ==, 0);
352789Sahrens 	ASSERT(refcount_is_zero(&dn->dn_tx_holds));
353789Sahrens 	ASSERT3U(refcount_count(&dn->dn_holds), <=, 1);
354789Sahrens 	ASSERT3P(list_head(&dn->dn_dbufs), ==, NULL);
355789Sahrens 
356789Sahrens 	for (i = 0; i < TXG_SIZE; i++) {
357789Sahrens 		ASSERT3U(dn->dn_next_nlevels[i], ==, 0);
358789Sahrens 		ASSERT3U(dn->dn_next_indblkshift[i], ==, 0);
359789Sahrens 		ASSERT3U(dn->dn_dirtyblksz[i], ==, 0);
360789Sahrens 		ASSERT3P(list_head(&dn->dn_dirty_dbufs[i]), ==, NULL);
361789Sahrens 		ASSERT3U(avl_numnodes(&dn->dn_ranges[i]), ==, 0);
362789Sahrens 	}
363789Sahrens 
364789Sahrens 	dn->dn_type = ot;
365789Sahrens 	dnode_setdblksz(dn, blocksize);
366789Sahrens 	dn->dn_indblkshift = ibs;
367789Sahrens 	dn->dn_nlevels = 1;
368789Sahrens 	dn->dn_nblkptr = 1 + ((DN_MAX_BONUSLEN - bonuslen) >> SPA_BLKPTRSHIFT);
369789Sahrens 	dn->dn_bonustype = bonustype;
370789Sahrens 	dn->dn_bonuslen = bonuslen;
371789Sahrens 	dn->dn_checksum = ZIO_CHECKSUM_INHERIT;
372789Sahrens 	dn->dn_compress = ZIO_COMPRESS_INHERIT;
373789Sahrens 	dn->dn_dirtyctx = 0;
374789Sahrens 
375789Sahrens 	dn->dn_free_txg = 0;
376789Sahrens 	if (dn->dn_dirtyctx_firstset) {
377789Sahrens 		kmem_free(dn->dn_dirtyctx_firstset, 1);
378789Sahrens 		dn->dn_dirtyctx_firstset = NULL;
379789Sahrens 	}
380789Sahrens 
381789Sahrens 	dn->dn_allocated_txg = tx->tx_txg;
382789Sahrens 	dnode_setdirty(dn, tx);
383789Sahrens }
384789Sahrens 
385789Sahrens void
386789Sahrens dnode_reallocate(dnode_t *dn, dmu_object_type_t ot, int blocksize,
387789Sahrens     dmu_object_type_t bonustype, int bonuslen, dmu_tx_t *tx)
388789Sahrens {
389789Sahrens 	ASSERT3U(blocksize, >=, SPA_MINBLOCKSIZE);
390789Sahrens 	ASSERT3U(blocksize, <=, SPA_MAXBLOCKSIZE);
391789Sahrens 	ASSERT3U(blocksize % SPA_MINBLOCKSIZE, ==, 0);
392*1544Seschrock 	ASSERT(dn->dn_object != DMU_META_DNODE_OBJECT || dmu_tx_private_ok(tx));
393789Sahrens 	ASSERT(tx->tx_txg != 0);
394789Sahrens 	ASSERT((bonustype == DMU_OT_NONE && bonuslen == 0) ||
395789Sahrens 	    (bonustype != DMU_OT_NONE && bonuslen != 0));
396789Sahrens 	ASSERT3U(bonustype, <, DMU_OT_NUMTYPES);
397789Sahrens 	ASSERT3U(bonuslen, <=, DN_MAX_BONUSLEN);
398789Sahrens 	ASSERT(dn->dn_dirtyblksz[0] == 0);
399789Sahrens 	ASSERT(dn->dn_dirtyblksz[1] == 0);
400789Sahrens 	ASSERT(dn->dn_dirtyblksz[2] == 0);
401789Sahrens 	ASSERT(dn->dn_dirtyblksz[3] == 0);
402789Sahrens 
403*1544Seschrock 	/* clean up any unreferenced dbufs */
404*1544Seschrock 	dnode_evict_dbufs(dn);
405*1544Seschrock 	ASSERT3P(list_head(&dn->dn_dbufs), ==, NULL);
406*1544Seschrock 
407789Sahrens 	/*
408789Sahrens 	 * XXX I should really have a generation number to tell if we
409789Sahrens 	 * need to do this...
410789Sahrens 	 */
411789Sahrens 	if (blocksize != dn->dn_datablksz ||
412789Sahrens 	    dn->dn_bonustype != bonustype || dn->dn_bonuslen != bonuslen) {
413789Sahrens 		/* free all old data */
414789Sahrens 		dnode_free_range(dn, 0, -1ULL, tx);
415789Sahrens 	}
416789Sahrens 
417789Sahrens 	/* change blocksize */
418789Sahrens 	rw_enter(&dn->dn_struct_rwlock, RW_WRITER);
419789Sahrens 	dnode_setdblksz(dn, blocksize);
420789Sahrens 	dnode_setdirty(dn, tx);
421789Sahrens 	/* don't need dd_dirty_mtx, dnode is already dirty */
422789Sahrens 	ASSERT(dn->dn_dirtyblksz[tx->tx_txg&TXG_MASK] != 0);
423789Sahrens 	dn->dn_dirtyblksz[tx->tx_txg&TXG_MASK] = blocksize;
424789Sahrens 	rw_exit(&dn->dn_struct_rwlock);
425789Sahrens 
426789Sahrens 	/* change type */
427789Sahrens 	dn->dn_type = ot;
428789Sahrens 
429789Sahrens 	if (dn->dn_bonuslen != bonuslen) {
430*1544Seschrock 		dmu_buf_impl_t *db = NULL;
431*1544Seschrock 
432789Sahrens 		/* change bonus size */
433789Sahrens 		if (bonuslen == 0)
434789Sahrens 			bonuslen = 1; /* XXX */
435*1544Seschrock 		rw_enter(&dn->dn_struct_rwlock, RW_WRITER);
436*1544Seschrock 		if (dn->dn_bonus == NULL)
437*1544Seschrock 			dn->dn_bonus = dbuf_create_bonus(dn);
438*1544Seschrock 		db = dn->dn_bonus;
439*1544Seschrock 		rw_exit(&dn->dn_struct_rwlock);
440*1544Seschrock 		if (refcount_add(&db->db_holds, FTAG) == 1)
441*1544Seschrock 			dnode_add_ref(dn, db);
442789Sahrens 		mutex_enter(&db->db_mtx);
443789Sahrens 		ASSERT3U(db->db.db_size, ==, dn->dn_bonuslen);
444789Sahrens 		ASSERT(db->db.db_data != NULL);
445789Sahrens 		db->db.db_size = bonuslen;
446789Sahrens 		mutex_exit(&db->db_mtx);
447789Sahrens 		dbuf_dirty(db, tx);
448*1544Seschrock 		dbuf_rele(db, FTAG);
449789Sahrens 	}
450789Sahrens 
451789Sahrens 	/* change bonus size and type */
452789Sahrens 	mutex_enter(&dn->dn_mtx);
453789Sahrens 	dn->dn_bonustype = bonustype;
454789Sahrens 	dn->dn_bonuslen = bonuslen;
455789Sahrens 	dn->dn_nblkptr = 1 + ((DN_MAX_BONUSLEN - bonuslen) >> SPA_BLKPTRSHIFT);
456789Sahrens 	dn->dn_checksum = ZIO_CHECKSUM_INHERIT;
457789Sahrens 	dn->dn_compress = ZIO_COMPRESS_INHERIT;
458789Sahrens 	ASSERT3U(dn->dn_nblkptr, <=, DN_MAX_NBLKPTR);
459789Sahrens 
460789Sahrens 	dn->dn_allocated_txg = tx->tx_txg;
461789Sahrens 	mutex_exit(&dn->dn_mtx);
462789Sahrens }
463789Sahrens 
464789Sahrens void
465789Sahrens dnode_special_close(dnode_t *dn)
466789Sahrens {
467*1544Seschrock 	/*
468*1544Seschrock 	 * Wait for final references to the dnode to clear.  This can
469*1544Seschrock 	 * only happen if the arc is asyncronously evicting state that
470*1544Seschrock 	 * has a hold on this dnode while we are trying to evict this
471*1544Seschrock 	 * dnode.
472*1544Seschrock 	 */
473*1544Seschrock 	while (refcount_count(&dn->dn_holds) > 0)
474*1544Seschrock 		delay(1);
475789Sahrens 	dnode_destroy(dn);
476789Sahrens }
477789Sahrens 
478789Sahrens dnode_t *
479789Sahrens dnode_special_open(objset_impl_t *os, dnode_phys_t *dnp, uint64_t object)
480789Sahrens {
481789Sahrens 	dnode_t *dn = dnode_create(os, dnp, NULL, object);
482873Sek110237 	DNODE_VERIFY(dn);
483789Sahrens 	return (dn);
484789Sahrens }
485789Sahrens 
486789Sahrens static void
487789Sahrens dnode_buf_pageout(dmu_buf_t *db, void *arg)
488789Sahrens {
489789Sahrens 	dnode_t **children_dnodes = arg;
490789Sahrens 	int i;
491789Sahrens 	int epb = db->db_size >> DNODE_SHIFT;
492789Sahrens 
493789Sahrens 	for (i = 0; i < epb; i++) {
494789Sahrens 		dnode_t *dn = children_dnodes[i];
495789Sahrens 		int n;
496789Sahrens 
497789Sahrens 		if (dn == NULL)
498789Sahrens 			continue;
499789Sahrens #ifdef ZFS_DEBUG
500789Sahrens 		/*
501789Sahrens 		 * If there are holds on this dnode, then there should
502789Sahrens 		 * be holds on the dnode's containing dbuf as well; thus
503789Sahrens 		 * it wouldn't be eligable for eviction and this function
504789Sahrens 		 * would not have been called.
505789Sahrens 		 */
506789Sahrens 		ASSERT(refcount_is_zero(&dn->dn_holds));
507789Sahrens 		ASSERT(list_head(&dn->dn_dbufs) == NULL);
508789Sahrens 		ASSERT(refcount_is_zero(&dn->dn_tx_holds));
509789Sahrens 
510789Sahrens 		for (n = 0; n < TXG_SIZE; n++)
511789Sahrens 			ASSERT(dn->dn_dirtyblksz[n] == 0);
512789Sahrens #endif
513789Sahrens 		children_dnodes[i] = NULL;
514789Sahrens 		dnode_destroy(dn);
515789Sahrens 	}
516789Sahrens 	kmem_free(children_dnodes, epb * sizeof (dnode_t *));
517789Sahrens }
518789Sahrens 
519789Sahrens /*
520*1544Seschrock  * errors:
521*1544Seschrock  * EINVAL - invalid object number.
522*1544Seschrock  * EIO - i/o error.
523*1544Seschrock  * succeeds even for free dnodes.
524789Sahrens  */
525*1544Seschrock int
526*1544Seschrock dnode_hold_impl(objset_impl_t *os, uint64_t object, int flag,
527*1544Seschrock     void *tag, dnode_t **dnp)
528789Sahrens {
529*1544Seschrock 	int epb, idx, err;
530789Sahrens 	int drop_struct_lock = FALSE;
531*1544Seschrock 	int type;
532789Sahrens 	uint64_t blk;
533789Sahrens 	dnode_t *mdn, *dn;
534789Sahrens 	dmu_buf_impl_t *db;
535789Sahrens 	dnode_t **children_dnodes;
536789Sahrens 
537789Sahrens 	if (object == 0 || object >= DN_MAX_OBJECT)
538*1544Seschrock 		return (EINVAL);
539789Sahrens 
540789Sahrens 	mdn = os->os_meta_dnode;
541789Sahrens 
542873Sek110237 	DNODE_VERIFY(mdn);
543789Sahrens 
544789Sahrens 	if (!RW_WRITE_HELD(&mdn->dn_struct_rwlock)) {
545789Sahrens 		rw_enter(&mdn->dn_struct_rwlock, RW_READER);
546789Sahrens 		drop_struct_lock = TRUE;
547789Sahrens 	}
548789Sahrens 
549789Sahrens 	blk = dbuf_whichblock(mdn, object * sizeof (dnode_phys_t));
550789Sahrens 
551*1544Seschrock 	db = dbuf_hold(mdn, blk, FTAG);
552789Sahrens 	if (drop_struct_lock)
553789Sahrens 		rw_exit(&mdn->dn_struct_rwlock);
554*1544Seschrock 	if (db == NULL)
555*1544Seschrock 		return (EIO);
556*1544Seschrock 	err = dbuf_read(db, NULL, DB_RF_CANFAIL);
557*1544Seschrock 	if (err) {
558*1544Seschrock 		dbuf_rele(db, FTAG);
559*1544Seschrock 		return (err);
560*1544Seschrock 	}
561789Sahrens 
562789Sahrens 	ASSERT3U(db->db.db_size, >=, 1<<DNODE_SHIFT);
563789Sahrens 	epb = db->db.db_size >> DNODE_SHIFT;
564789Sahrens 
565789Sahrens 	idx = object & (epb-1);
566789Sahrens 
567789Sahrens 	children_dnodes = dmu_buf_get_user(&db->db);
568789Sahrens 	if (children_dnodes == NULL) {
569789Sahrens 		dnode_t **winner;
570789Sahrens 		children_dnodes = kmem_zalloc(epb * sizeof (dnode_t *),
571789Sahrens 		    KM_SLEEP);
572789Sahrens 		if (winner = dmu_buf_set_user(&db->db, children_dnodes, NULL,
573789Sahrens 		    dnode_buf_pageout)) {
574789Sahrens 			kmem_free(children_dnodes, epb * sizeof (dnode_t *));
575789Sahrens 			children_dnodes = winner;
576789Sahrens 		}
577789Sahrens 	}
578789Sahrens 
579789Sahrens 	if ((dn = children_dnodes[idx]) == NULL) {
580789Sahrens 		dnode_t *winner;
581789Sahrens 		dn = dnode_create(os, (dnode_phys_t *)db->db.db_data+idx,
582789Sahrens 			db, object);
583789Sahrens 		winner = atomic_cas_ptr(&children_dnodes[idx], NULL, dn);
584789Sahrens 		if (winner != NULL) {
585789Sahrens 			dnode_destroy(dn);
586789Sahrens 			dn = winner;
587789Sahrens 		}
588789Sahrens 	}
589789Sahrens 
590789Sahrens 	mutex_enter(&dn->dn_mtx);
591*1544Seschrock 	type = dn->dn_type;
592789Sahrens 	if (dn->dn_free_txg ||
593*1544Seschrock 	    ((flag & DNODE_MUST_BE_ALLOCATED) && type == DMU_OT_NONE) ||
594*1544Seschrock 	    ((flag & DNODE_MUST_BE_FREE) && type != DMU_OT_NONE)) {
595789Sahrens 		mutex_exit(&dn->dn_mtx);
596*1544Seschrock 		dbuf_rele(db, FTAG);
597*1544Seschrock 		return (type == DMU_OT_NONE ? ENOENT : EEXIST);
598789Sahrens 	}
599789Sahrens 	mutex_exit(&dn->dn_mtx);
600789Sahrens 
601*1544Seschrock 	if (refcount_add(&dn->dn_holds, tag) == 1)
602789Sahrens 		dbuf_add_ref(db, dn);
603789Sahrens 
604873Sek110237 	DNODE_VERIFY(dn);
605789Sahrens 	ASSERT3P(dn->dn_dbuf, ==, db);
606789Sahrens 	ASSERT3U(dn->dn_object, ==, object);
607*1544Seschrock 	dbuf_rele(db, FTAG);
608789Sahrens 
609*1544Seschrock 	*dnp = dn;
610*1544Seschrock 	return (0);
611789Sahrens }
612789Sahrens 
613789Sahrens /*
614789Sahrens  * Return held dnode if the object is allocated, NULL if not.
615789Sahrens  */
616*1544Seschrock int
617*1544Seschrock dnode_hold(objset_impl_t *os, uint64_t object, void *tag, dnode_t **dnp)
618789Sahrens {
619*1544Seschrock 	return (dnode_hold_impl(os, object, DNODE_MUST_BE_ALLOCATED, tag, dnp));
620789Sahrens }
621789Sahrens 
622789Sahrens void
623*1544Seschrock dnode_add_ref(dnode_t *dn, void *tag)
624789Sahrens {
625789Sahrens 	ASSERT(refcount_count(&dn->dn_holds) > 0);
626*1544Seschrock 	(void) refcount_add(&dn->dn_holds, tag);
627789Sahrens }
628789Sahrens 
629789Sahrens void
630*1544Seschrock dnode_rele(dnode_t *dn, void *tag)
631789Sahrens {
632789Sahrens 	uint64_t refs;
633789Sahrens 
634*1544Seschrock 	refs = refcount_remove(&dn->dn_holds, tag);
635789Sahrens 	/* NOTE: the DNODE_DNODE does not have a dn_dbuf */
636789Sahrens 	if (refs == 0 && dn->dn_dbuf)
637*1544Seschrock 		dbuf_rele(dn->dn_dbuf, dn);
638789Sahrens }
639789Sahrens 
640789Sahrens void
641789Sahrens dnode_setdirty(dnode_t *dn, dmu_tx_t *tx)
642789Sahrens {
643789Sahrens 	objset_impl_t *os = dn->dn_objset;
644789Sahrens 	uint64_t txg = tx->tx_txg;
645789Sahrens 
646*1544Seschrock 	if (dn->dn_object == DMU_META_DNODE_OBJECT)
647789Sahrens 		return;
648789Sahrens 
649873Sek110237 	DNODE_VERIFY(dn);
650789Sahrens 
651789Sahrens #ifdef ZFS_DEBUG
652789Sahrens 	mutex_enter(&dn->dn_mtx);
653789Sahrens 	ASSERT(dn->dn_phys->dn_type || dn->dn_allocated_txg);
654789Sahrens 	/* ASSERT(dn->dn_free_txg == 0 || dn->dn_free_txg >= txg); */
655789Sahrens 	mutex_exit(&dn->dn_mtx);
656789Sahrens #endif
657789Sahrens 
658789Sahrens 	mutex_enter(&os->os_lock);
659789Sahrens 
660789Sahrens 	/*
661789Sahrens 	 * If we are already marked dirty, we're done.
662789Sahrens 	 */
663789Sahrens 	if (dn->dn_dirtyblksz[txg&TXG_MASK] > 0) {
664789Sahrens 		mutex_exit(&os->os_lock);
665789Sahrens 		return;
666789Sahrens 	}
667789Sahrens 
668789Sahrens 	ASSERT(!refcount_is_zero(&dn->dn_holds) || list_head(&dn->dn_dbufs));
669789Sahrens 	ASSERT(dn->dn_datablksz != 0);
670789Sahrens 	dn->dn_dirtyblksz[txg&TXG_MASK] = dn->dn_datablksz;
671789Sahrens 
672789Sahrens 	dprintf_ds(os->os_dsl_dataset, "obj=%llu txg=%llu\n",
673789Sahrens 	    dn->dn_object, txg);
674789Sahrens 
675789Sahrens 	if (dn->dn_free_txg > 0 && dn->dn_free_txg <= txg) {
676789Sahrens 		list_insert_tail(&os->os_free_dnodes[txg&TXG_MASK], dn);
677789Sahrens 	} else {
678789Sahrens 		list_insert_tail(&os->os_dirty_dnodes[txg&TXG_MASK], dn);
679789Sahrens 	}
680789Sahrens 
681789Sahrens 	mutex_exit(&os->os_lock);
682789Sahrens 
683789Sahrens 	/*
684789Sahrens 	 * The dnode maintains a hold on its containing dbuf as
685789Sahrens 	 * long as there are holds on it.  Each instantiated child
686789Sahrens 	 * dbuf maintaines a hold on the dnode.  When the last child
687789Sahrens 	 * drops its hold, the dnode will drop its hold on the
688789Sahrens 	 * containing dbuf. We add a "dirty hold" here so that the
689789Sahrens 	 * dnode will hang around after we finish processing its
690789Sahrens 	 * children.
691789Sahrens 	 */
692*1544Seschrock 	dnode_add_ref(dn, (void *)(uintptr_t)tx->tx_txg);
693789Sahrens 
694789Sahrens 	dbuf_dirty(dn->dn_dbuf, tx);
695789Sahrens 
696789Sahrens 	dsl_dataset_dirty(os->os_dsl_dataset, tx);
697789Sahrens }
698789Sahrens 
699789Sahrens void
700789Sahrens dnode_free(dnode_t *dn, dmu_tx_t *tx)
701789Sahrens {
702789Sahrens 	dprintf("dn=%p txg=%llu\n", dn, tx->tx_txg);
703789Sahrens 
704789Sahrens 	/* we should be the only holder... hopefully */
705789Sahrens 	/* ASSERT3U(refcount_count(&dn->dn_holds), ==, 1); */
706789Sahrens 
707789Sahrens 	mutex_enter(&dn->dn_mtx);
708789Sahrens 	if (dn->dn_type == DMU_OT_NONE || dn->dn_free_txg) {
709789Sahrens 		mutex_exit(&dn->dn_mtx);
710789Sahrens 		return;
711789Sahrens 	}
712789Sahrens 	dn->dn_free_txg = tx->tx_txg;
713789Sahrens 	mutex_exit(&dn->dn_mtx);
714789Sahrens 
715789Sahrens 	/*
716789Sahrens 	 * If the dnode is already dirty, it needs to be moved from
717789Sahrens 	 * the dirty list to the free list.
718789Sahrens 	 */
719789Sahrens 	mutex_enter(&dn->dn_objset->os_lock);
720789Sahrens 	if (dn->dn_dirtyblksz[tx->tx_txg&TXG_MASK] > 0) {
721789Sahrens 		list_remove(
722789Sahrens 		    &dn->dn_objset->os_dirty_dnodes[tx->tx_txg&TXG_MASK], dn);
723789Sahrens 		list_insert_tail(
724789Sahrens 		    &dn->dn_objset->os_free_dnodes[tx->tx_txg&TXG_MASK], dn);
725789Sahrens 		mutex_exit(&dn->dn_objset->os_lock);
726789Sahrens 	} else {
727789Sahrens 		mutex_exit(&dn->dn_objset->os_lock);
728789Sahrens 		dnode_setdirty(dn, tx);
729789Sahrens 	}
730789Sahrens }
731789Sahrens 
732789Sahrens /*
733789Sahrens  * Try to change the block size for the indicated dnode.  This can only
734789Sahrens  * succeed if there are no blocks allocated or dirty beyond first block
735789Sahrens  */
736789Sahrens int
737789Sahrens dnode_set_blksz(dnode_t *dn, uint64_t size, int ibs, dmu_tx_t *tx)
738789Sahrens {
739789Sahrens 	dmu_buf_impl_t *db, *db_next;
740789Sahrens 	int have_db0 = FALSE;
741789Sahrens 	int err = ENOTSUP;
742789Sahrens 
743789Sahrens 	if (size == 0)
744789Sahrens 		size = SPA_MINBLOCKSIZE;
745789Sahrens 	if (size > SPA_MAXBLOCKSIZE)
746789Sahrens 		size = SPA_MAXBLOCKSIZE;
747789Sahrens 	else
748789Sahrens 		size = P2ROUNDUP(size, SPA_MINBLOCKSIZE);
749789Sahrens 
750789Sahrens 	if (ibs == 0)
751789Sahrens 		ibs = dn->dn_indblkshift;
752789Sahrens 
753789Sahrens 	if (size >> SPA_MINBLOCKSHIFT == dn->dn_datablkszsec &&
754789Sahrens 	    ibs == dn->dn_indblkshift)
755789Sahrens 		return (0);
756789Sahrens 
757789Sahrens 	rw_enter(&dn->dn_struct_rwlock, RW_WRITER);
758789Sahrens 
759789Sahrens 	/* Check for any allocated blocks beyond the first */
760789Sahrens 	if (dn->dn_phys->dn_maxblkid != 0)
761789Sahrens 		goto end;
762789Sahrens 
763789Sahrens 	/*
764789Sahrens 	 * Any buffers allocated for blocks beyond the first
765789Sahrens 	 * must be evictable/evicted, because they're the wrong size.
766789Sahrens 	 */
767789Sahrens 	mutex_enter(&dn->dn_dbufs_mtx);
768789Sahrens 	/*
769789Sahrens 	 * Since we have the dn_dbufs_mtx, nothing can be
770789Sahrens 	 * removed from dn_dbufs.  Since we have dn_struct_rwlock/w,
771789Sahrens 	 * nothing can be added to dn_dbufs.
772789Sahrens 	 */
773789Sahrens 	for (db = list_head(&dn->dn_dbufs); db; db = db_next) {
774789Sahrens 		db_next = list_next(&dn->dn_dbufs, db);
775789Sahrens 
776789Sahrens 		if (db->db_blkid == 0) {
777789Sahrens 			have_db0 = TRUE;
778789Sahrens 		} else if (db->db_blkid != DB_BONUS_BLKID) {
779789Sahrens 			mutex_exit(&dn->dn_dbufs_mtx);
780789Sahrens 			goto end;
781789Sahrens 		}
782789Sahrens 	}
783789Sahrens 	mutex_exit(&dn->dn_dbufs_mtx);
784789Sahrens 
785789Sahrens 	/* Fast-track if there is no data in the file */
786789Sahrens 	if (BP_IS_HOLE(&dn->dn_phys->dn_blkptr[0]) && !have_db0) {
787789Sahrens 		dnode_setdblksz(dn, size);
788789Sahrens 		dn->dn_indblkshift = ibs;
789789Sahrens 		dnode_setdirty(dn, tx);
790789Sahrens 		/* don't need dd_dirty_mtx, dnode is already dirty */
791789Sahrens 		dn->dn_dirtyblksz[tx->tx_txg&TXG_MASK] = size;
792789Sahrens 		dn->dn_next_indblkshift[tx->tx_txg&TXG_MASK] = ibs;
793789Sahrens 		rw_exit(&dn->dn_struct_rwlock);
794789Sahrens 		return (0);
795789Sahrens 	}
796789Sahrens 
797789Sahrens 	/* obtain the old block */
798*1544Seschrock 	db = dbuf_hold(dn, 0, FTAG);
799789Sahrens 
800789Sahrens 	dbuf_new_size(db, size, tx);
801789Sahrens 
802789Sahrens 	dnode_setdblksz(dn, size);
803789Sahrens 	dn->dn_indblkshift = ibs;
804789Sahrens 	/* don't need dd_dirty_mtx, dnode is already dirty */
805789Sahrens 	dn->dn_dirtyblksz[tx->tx_txg&TXG_MASK] = size;
806789Sahrens 	dn->dn_next_indblkshift[tx->tx_txg&TXG_MASK] = ibs;
807*1544Seschrock 	dbuf_rele(db, FTAG);
808789Sahrens 
809789Sahrens 	err = 0;
810789Sahrens end:
811789Sahrens 	rw_exit(&dn->dn_struct_rwlock);
812789Sahrens 	return (err);
813789Sahrens }
814789Sahrens 
815789Sahrens uint64_t
816789Sahrens dnode_max_nonzero_offset(dnode_t *dn)
817789Sahrens {
818789Sahrens 	if (dn->dn_phys->dn_maxblkid == 0 &&
819789Sahrens 	    BP_IS_HOLE(&dn->dn_phys->dn_blkptr[0]))
820789Sahrens 		return (0);
821789Sahrens 	else
822789Sahrens 		return ((dn->dn_phys->dn_maxblkid+1) * dn->dn_datablksz);
823789Sahrens }
824789Sahrens 
825789Sahrens void
826789Sahrens dnode_new_blkid(dnode_t *dn, uint64_t blkid, dmu_tx_t *tx)
827789Sahrens {
828789Sahrens 	uint64_t txgoff = tx->tx_txg & TXG_MASK;
829789Sahrens 	int drop_struct_lock = FALSE;
830789Sahrens 	int epbs, old_nlevels, new_nlevels;
831789Sahrens 	uint64_t sz;
832789Sahrens 
833789Sahrens 	if (blkid == DB_BONUS_BLKID)
834789Sahrens 		return;
835789Sahrens 
836789Sahrens 	if (!RW_WRITE_HELD(&dn->dn_struct_rwlock)) {
837789Sahrens 		rw_enter(&dn->dn_struct_rwlock, RW_WRITER);
838789Sahrens 		drop_struct_lock = TRUE;
839789Sahrens 	}
840789Sahrens 
841789Sahrens 	if (blkid > dn->dn_maxblkid)
842789Sahrens 		dn->dn_maxblkid = blkid;
843789Sahrens 
844789Sahrens 	/*
845789Sahrens 	 * Compute the number of levels necessary to support the
846789Sahrens 	 * new blkid.
847789Sahrens 	 */
848789Sahrens 	new_nlevels = 1;
849789Sahrens 	epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
850789Sahrens 
851789Sahrens 	for (sz = dn->dn_nblkptr; sz <= blkid && sz >= dn->dn_nblkptr;
852789Sahrens 	    sz <<= epbs)
853789Sahrens 		new_nlevels++;
854789Sahrens 	old_nlevels = dn->dn_nlevels;
855789Sahrens 
856789Sahrens 	if (new_nlevels > dn->dn_next_nlevels[txgoff])
857789Sahrens 		dn->dn_next_nlevels[txgoff] = new_nlevels;
858789Sahrens 
859789Sahrens 	if (new_nlevels > old_nlevels) {
860789Sahrens 		dprintf("dn %p increasing nlevels from %u to %u\n",
861789Sahrens 		    dn, dn->dn_nlevels, new_nlevels);
862789Sahrens 		dn->dn_nlevels = new_nlevels;
863789Sahrens 	}
864789Sahrens 
865789Sahrens 	/*
866789Sahrens 	 * Dirty the left indirects.
867789Sahrens 	 * Note: the caller should have just dnode_use_space()'d one
868789Sahrens 	 * data block's worth, so we could subtract that out of
869789Sahrens 	 * dn_inflight_data to determine if there is any dirty data
870789Sahrens 	 * besides this block.
871789Sahrens 	 * We don't strictly need to dirty them unless there's
872789Sahrens 	 * *something* in the object (eg. on disk or dirty)...
873789Sahrens 	 */
874789Sahrens 	if (new_nlevels > old_nlevels) {
875789Sahrens 		dmu_buf_impl_t *db = dbuf_hold_level(dn, old_nlevels, 0, FTAG);
876789Sahrens 		dprintf("dn %p dirtying left indirects\n", dn);
877789Sahrens 		dbuf_dirty(db, tx);
878*1544Seschrock 		dbuf_rele(db, FTAG);
879789Sahrens 	}
880789Sahrens #ifdef ZFS_DEBUG
881789Sahrens 	else if (old_nlevels > 1 && new_nlevels > old_nlevels) {
882789Sahrens 		dmu_buf_impl_t *db;
883789Sahrens 		int i;
884789Sahrens 
885789Sahrens 		for (i = 0; i < dn->dn_nblkptr; i++) {
886789Sahrens 			db = dbuf_hold_level(dn, old_nlevels-1, i, FTAG);
887789Sahrens 			ASSERT(!
888789Sahrens 			    list_link_active(&db->db_dirty_node[txgoff]));
889*1544Seschrock 			dbuf_rele(db, FTAG);
890789Sahrens 		}
891789Sahrens 	}
892789Sahrens #endif
893789Sahrens 
894789Sahrens 	dprintf("dn %p done\n", dn);
895789Sahrens 
896789Sahrens out:
897789Sahrens 	if (drop_struct_lock)
898789Sahrens 		rw_exit(&dn->dn_struct_rwlock);
899789Sahrens }
900789Sahrens 
901789Sahrens void
902789Sahrens dnode_clear_range(dnode_t *dn, uint64_t blkid, uint64_t nblks, dmu_tx_t *tx)
903789Sahrens {
904789Sahrens 	avl_tree_t *tree = &dn->dn_ranges[tx->tx_txg&TXG_MASK];
905789Sahrens 	avl_index_t where;
906789Sahrens 	free_range_t *rp;
907789Sahrens 	free_range_t rp_tofind;
908789Sahrens 	uint64_t endblk = blkid + nblks;
909789Sahrens 
910789Sahrens 	ASSERT(MUTEX_HELD(&dn->dn_mtx));
911789Sahrens 	ASSERT(nblks <= UINT64_MAX - blkid); /* no overflow */
912789Sahrens 
913789Sahrens 	dprintf_dnode(dn, "blkid=%llu nblks=%llu txg=%llu\n",
914789Sahrens 	    blkid, nblks, tx->tx_txg);
915789Sahrens 	rp_tofind.fr_blkid = blkid;
916789Sahrens 	rp = avl_find(tree, &rp_tofind, &where);
917789Sahrens 	if (rp == NULL)
918789Sahrens 		rp = avl_nearest(tree, where, AVL_BEFORE);
919789Sahrens 	if (rp == NULL)
920789Sahrens 		rp = avl_nearest(tree, where, AVL_AFTER);
921789Sahrens 
922789Sahrens 	while (rp && (rp->fr_blkid <= blkid + nblks)) {
923789Sahrens 		uint64_t fr_endblk = rp->fr_blkid + rp->fr_nblks;
924789Sahrens 		free_range_t *nrp = AVL_NEXT(tree, rp);
925789Sahrens 
926789Sahrens 		if (blkid <= rp->fr_blkid && endblk >= fr_endblk) {
927789Sahrens 			/* clear this entire range */
928789Sahrens 			avl_remove(tree, rp);
929789Sahrens 			kmem_free(rp, sizeof (free_range_t));
930789Sahrens 		} else if (blkid <= rp->fr_blkid &&
931789Sahrens 		    endblk > rp->fr_blkid && endblk < fr_endblk) {
932789Sahrens 			/* clear the beginning of this range */
933789Sahrens 			rp->fr_blkid = endblk;
934789Sahrens 			rp->fr_nblks = fr_endblk - endblk;
935789Sahrens 		} else if (blkid > rp->fr_blkid && blkid < fr_endblk &&
936789Sahrens 		    endblk >= fr_endblk) {
937789Sahrens 			/* clear the end of this range */
938789Sahrens 			rp->fr_nblks = blkid - rp->fr_blkid;
939789Sahrens 		} else if (blkid > rp->fr_blkid && endblk < fr_endblk) {
940789Sahrens 			/* clear a chunk out of this range */
941789Sahrens 			free_range_t *new_rp =
942789Sahrens 			    kmem_alloc(sizeof (free_range_t), KM_SLEEP);
943789Sahrens 
944789Sahrens 			new_rp->fr_blkid = endblk;
945789Sahrens 			new_rp->fr_nblks = fr_endblk - endblk;
946789Sahrens 			avl_insert_here(tree, new_rp, rp, AVL_AFTER);
947789Sahrens 			rp->fr_nblks = blkid - rp->fr_blkid;
948789Sahrens 		}
949789Sahrens 		/* there may be no overlap */
950789Sahrens 		rp = nrp;
951789Sahrens 	}
952789Sahrens }
953789Sahrens 
954789Sahrens void
955789Sahrens dnode_free_range(dnode_t *dn, uint64_t off, uint64_t len, dmu_tx_t *tx)
956789Sahrens {
957789Sahrens 	dmu_buf_impl_t *db;
958789Sahrens 	uint64_t start, objsize, blkid, nblks;
959789Sahrens 	int blkshift, blksz, tail, head, epbs;
960789Sahrens 	int trunc = FALSE;
961789Sahrens 
962789Sahrens 	rw_enter(&dn->dn_struct_rwlock, RW_WRITER);
963789Sahrens 	blksz = dn->dn_datablksz;
964789Sahrens 	blkshift = dn->dn_datablkshift;
965789Sahrens 	epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
966789Sahrens 
967789Sahrens 	/* If the range is past the end of the file, this is a no-op */
968789Sahrens 	objsize = blksz * (dn->dn_maxblkid+1);
969789Sahrens 	if (off >= objsize)
970789Sahrens 		goto out;
971789Sahrens 	if (len == -1ULL) {
972789Sahrens 		len = UINT64_MAX - off;
973789Sahrens 		trunc = TRUE;
974789Sahrens 	}
975789Sahrens 
976789Sahrens 	/*
977789Sahrens 	 * First, block align the region to free:
978789Sahrens 	 */
979789Sahrens 	if (dn->dn_maxblkid == 0) {
980789Sahrens 		if (off == 0) {
981789Sahrens 			head = 0;
982789Sahrens 		} else {
983789Sahrens 			head = blksz - off;
984789Sahrens 			ASSERT3U(head, >, 0);
985789Sahrens 		}
986789Sahrens 		start = off;
987789Sahrens 	} else {
988789Sahrens 		ASSERT(ISP2(blksz));
989789Sahrens 		head = P2NPHASE(off, blksz);
990789Sahrens 		start = P2PHASE(off, blksz);
991789Sahrens 	}
992789Sahrens 	/* zero out any partial block data at the start of the range */
993789Sahrens 	if (head) {
994789Sahrens 		ASSERT3U(start + head, ==, blksz);
995789Sahrens 		if (len < head)
996789Sahrens 			head = len;
997789Sahrens 		if (dbuf_hold_impl(dn, 0, dbuf_whichblock(dn, off), TRUE,
998789Sahrens 		    FTAG, &db) == 0) {
999789Sahrens 			caddr_t data;
1000789Sahrens 
1001789Sahrens 			/* don't dirty if it isn't on disk and isn't dirty */
1002789Sahrens 			if (db->db_dirtied ||
1003789Sahrens 			    (db->db_blkptr && !BP_IS_HOLE(db->db_blkptr))) {
1004789Sahrens 				rw_exit(&dn->dn_struct_rwlock);
1005789Sahrens 				dbuf_will_dirty(db, tx);
1006789Sahrens 				rw_enter(&dn->dn_struct_rwlock, RW_WRITER);
1007789Sahrens 				data = db->db.db_data;
1008789Sahrens 				bzero(data + start, head);
1009789Sahrens 			}
1010*1544Seschrock 			dbuf_rele(db, FTAG);
1011789Sahrens 		}
1012789Sahrens 		off += head;
1013789Sahrens 		len -= head;
1014789Sahrens 	}
1015789Sahrens 	/* If the range was less than one block, we are done */
1016789Sahrens 	if (len == 0)
1017789Sahrens 		goto out;
1018789Sahrens 
1019789Sahrens 	/* If the remaining range is past the end of the file, we are done */
1020789Sahrens 	if (off > dn->dn_maxblkid << blkshift)
1021789Sahrens 		goto out;
1022789Sahrens 
1023789Sahrens 	if (off + len == UINT64_MAX)
1024789Sahrens 		tail = 0;
1025789Sahrens 	else
1026789Sahrens 		tail = P2PHASE(len, blksz);
1027789Sahrens 
1028789Sahrens 	ASSERT3U(P2PHASE(off, blksz), ==, 0);
1029789Sahrens 	/* zero out any partial block data at the end of the range */
1030789Sahrens 	if (tail) {
1031789Sahrens 		if (len < tail)
1032789Sahrens 			tail = len;
1033789Sahrens 		if (dbuf_hold_impl(dn, 0, dbuf_whichblock(dn, off+len),
1034789Sahrens 		    TRUE, FTAG, &db) == 0) {
1035789Sahrens 			/* don't dirty if it isn't on disk and isn't dirty */
1036789Sahrens 			if (db->db_dirtied ||
1037789Sahrens 			    (db->db_blkptr && !BP_IS_HOLE(db->db_blkptr))) {
1038789Sahrens 				rw_exit(&dn->dn_struct_rwlock);
1039789Sahrens 				dbuf_will_dirty(db, tx);
1040789Sahrens 				rw_enter(&dn->dn_struct_rwlock, RW_WRITER);
1041789Sahrens 				bzero(db->db.db_data, tail);
1042789Sahrens 			}
1043*1544Seschrock 			dbuf_rele(db, FTAG);
1044789Sahrens 		}
1045789Sahrens 		len -= tail;
1046789Sahrens 	}
1047789Sahrens 	/* If the range did not include a full block, we are done */
1048789Sahrens 	if (len == 0)
1049789Sahrens 		goto out;
1050789Sahrens 
1051789Sahrens 	/* dirty the left indirects */
1052789Sahrens 	if (dn->dn_nlevels > 1 && off != 0) {
1053789Sahrens 		db = dbuf_hold_level(dn, 1,
1054789Sahrens 		    (off - head) >> (blkshift + epbs), FTAG);
1055789Sahrens 		dbuf_will_dirty(db, tx);
1056*1544Seschrock 		dbuf_rele(db, FTAG);
1057789Sahrens 	}
1058789Sahrens 
1059789Sahrens 	/* dirty the right indirects */
1060789Sahrens 	if (dn->dn_nlevels > 1 && !trunc) {
1061789Sahrens 		db = dbuf_hold_level(dn, 1,
1062789Sahrens 		    (off + len + tail - 1) >> (blkshift + epbs), FTAG);
1063789Sahrens 		dbuf_will_dirty(db, tx);
1064*1544Seschrock 		dbuf_rele(db, FTAG);
1065789Sahrens 	}
1066789Sahrens 
1067789Sahrens 	/*
1068789Sahrens 	 * Finally, add this range to the dnode range list, we
1069789Sahrens 	 * will finish up this free operation in the syncing phase.
1070789Sahrens 	 */
1071789Sahrens 	ASSERT(IS_P2ALIGNED(off, 1<<blkshift));
1072789Sahrens 	ASSERT(off + len == UINT64_MAX || IS_P2ALIGNED(len, 1<<blkshift));
1073789Sahrens 	blkid = off >> blkshift;
1074789Sahrens 	nblks = len >> blkshift;
1075789Sahrens 
1076789Sahrens 	if (trunc)
1077789Sahrens 		dn->dn_maxblkid = (blkid ? blkid - 1 : 0);
1078789Sahrens 
1079789Sahrens 	mutex_enter(&dn->dn_mtx);
1080789Sahrens 	dnode_clear_range(dn, blkid, nblks, tx);
1081789Sahrens 	{
1082789Sahrens 		free_range_t *rp, *found;
1083789Sahrens 		avl_index_t where;
1084789Sahrens 		avl_tree_t *tree = &dn->dn_ranges[tx->tx_txg&TXG_MASK];
1085789Sahrens 
1086789Sahrens 		/* Add new range to dn_ranges */
1087789Sahrens 		rp = kmem_alloc(sizeof (free_range_t), KM_SLEEP);
1088789Sahrens 		rp->fr_blkid = blkid;
1089789Sahrens 		rp->fr_nblks = nblks;
1090789Sahrens 		found = avl_find(tree, rp, &where);
1091789Sahrens 		ASSERT(found == NULL);
1092789Sahrens 		avl_insert(tree, rp, where);
1093789Sahrens 		dprintf_dnode(dn, "blkid=%llu nblks=%llu txg=%llu\n",
1094789Sahrens 		    blkid, nblks, tx->tx_txg);
1095789Sahrens 	}
1096789Sahrens 	mutex_exit(&dn->dn_mtx);
1097789Sahrens 
1098789Sahrens 	dbuf_free_range(dn, blkid, nblks, tx);
1099789Sahrens 	dnode_setdirty(dn, tx);
1100789Sahrens out:
1101789Sahrens 	rw_exit(&dn->dn_struct_rwlock);
1102789Sahrens }
1103789Sahrens 
1104789Sahrens /* return TRUE if this blkid was freed in a recent txg, or FALSE if it wasn't */
1105789Sahrens uint64_t
1106789Sahrens dnode_block_freed(dnode_t *dn, uint64_t blkid)
1107789Sahrens {
1108789Sahrens 	free_range_t range_tofind;
1109789Sahrens 	void *dp = spa_get_dsl(dn->dn_objset->os_spa);
1110789Sahrens 	int i;
1111789Sahrens 
1112789Sahrens 	if (blkid == DB_BONUS_BLKID)
1113789Sahrens 		return (FALSE);
1114789Sahrens 
1115789Sahrens 	/*
1116789Sahrens 	 * If we're in the process of opening the pool, dp will not be
1117789Sahrens 	 * set yet, but there shouldn't be anything dirty.
1118789Sahrens 	 */
1119789Sahrens 	if (dp == NULL)
1120789Sahrens 		return (FALSE);
1121789Sahrens 
1122789Sahrens 	if (dn->dn_free_txg)
1123789Sahrens 		return (TRUE);
1124789Sahrens 
1125789Sahrens 	/*
1126789Sahrens 	 * If dn_datablkshift is not set, then there's only a single
1127789Sahrens 	 * block, in which case there will never be a free range so it
1128789Sahrens 	 * won't matter.
1129789Sahrens 	 */
1130789Sahrens 	range_tofind.fr_blkid = blkid;
1131789Sahrens 	mutex_enter(&dn->dn_mtx);
1132789Sahrens 	for (i = 0; i < TXG_SIZE; i++) {
1133789Sahrens 		free_range_t *range_found;
1134789Sahrens 		avl_index_t idx;
1135789Sahrens 
1136789Sahrens 		range_found = avl_find(&dn->dn_ranges[i], &range_tofind, &idx);
1137789Sahrens 		if (range_found) {
1138789Sahrens 			ASSERT(range_found->fr_nblks > 0);
1139789Sahrens 			break;
1140789Sahrens 		}
1141789Sahrens 		range_found = avl_nearest(&dn->dn_ranges[i], idx, AVL_BEFORE);
1142789Sahrens 		if (range_found &&
1143789Sahrens 		    range_found->fr_blkid + range_found->fr_nblks > blkid)
1144789Sahrens 			break;
1145789Sahrens 	}
1146789Sahrens 	mutex_exit(&dn->dn_mtx);
1147789Sahrens 	return (i < TXG_SIZE);
1148789Sahrens }
1149789Sahrens 
1150789Sahrens /* call from syncing context when we actually write/free space for this dnode */
1151789Sahrens void
1152789Sahrens dnode_diduse_space(dnode_t *dn, int64_t space)
1153789Sahrens {
1154789Sahrens 	uint64_t sectors;
1155789Sahrens 
1156789Sahrens 	dprintf_dnode(dn, "dn=%p dnp=%p secphys=%llu space=%lld\n",
1157789Sahrens 	    dn, dn->dn_phys,
1158789Sahrens 	    (u_longlong_t)dn->dn_phys->dn_secphys,
1159789Sahrens 	    (longlong_t)space);
1160789Sahrens 
1161789Sahrens 	ASSERT(P2PHASE(space, 1<<DEV_BSHIFT) == 0);
1162789Sahrens 
1163789Sahrens 	mutex_enter(&dn->dn_mtx);
1164789Sahrens 	if (space > 0) {
1165789Sahrens 		sectors = space >> DEV_BSHIFT;
1166789Sahrens 		ASSERT3U(dn->dn_phys->dn_secphys + sectors, >=,
1167789Sahrens 		    dn->dn_phys->dn_secphys);
1168789Sahrens 		dn->dn_phys->dn_secphys += sectors;
1169789Sahrens 	} else {
1170789Sahrens 		sectors = -space >> DEV_BSHIFT;
1171789Sahrens 		ASSERT3U(dn->dn_phys->dn_secphys, >=, sectors);
1172789Sahrens 		dn->dn_phys->dn_secphys -= sectors;
1173789Sahrens 	}
1174789Sahrens 	mutex_exit(&dn->dn_mtx);
1175789Sahrens }
1176789Sahrens 
1177789Sahrens /*
1178789Sahrens  * Call when we think we're going to write/free space in open context.
1179789Sahrens  * Be conservative (ie. OK to write less than this or free more than
1180789Sahrens  * this, but don't write more or free less).
1181789Sahrens  */
1182789Sahrens void
1183789Sahrens dnode_willuse_space(dnode_t *dn, int64_t space, dmu_tx_t *tx)
1184789Sahrens {
1185789Sahrens 	objset_impl_t *os = dn->dn_objset;
1186789Sahrens 	dsl_dataset_t *ds = os->os_dsl_dataset;
1187789Sahrens 
1188789Sahrens 	if (space > 0)
1189789Sahrens 		space = spa_get_asize(os->os_spa, space);
1190789Sahrens 
1191789Sahrens 	if (ds)
1192789Sahrens 		dsl_dir_willuse_space(ds->ds_dir, space, tx);
1193789Sahrens 
1194789Sahrens 	dmu_tx_willuse_space(tx, space);
1195789Sahrens }
1196789Sahrens 
1197789Sahrens static int
1198789Sahrens dnode_next_offset_level(dnode_t *dn, boolean_t hole, uint64_t *offset,
1199789Sahrens 	int lvl, uint64_t blkfill)
1200789Sahrens {
1201789Sahrens 	dmu_buf_impl_t *db = NULL;
1202789Sahrens 	void *data = NULL;
1203789Sahrens 	uint64_t epbs = dn->dn_phys->dn_indblkshift - SPA_BLKPTRSHIFT;
1204789Sahrens 	uint64_t epb = 1ULL << epbs;
1205789Sahrens 	uint64_t minfill, maxfill;
1206789Sahrens 	int i, error, span;
1207789Sahrens 
1208789Sahrens 	dprintf("probing object %llu offset %llx level %d of %u\n",
1209789Sahrens 	    dn->dn_object, *offset, lvl, dn->dn_phys->dn_nlevels);
1210789Sahrens 
1211789Sahrens 	if (lvl == dn->dn_phys->dn_nlevels) {
1212789Sahrens 		error = 0;
1213789Sahrens 		epb = dn->dn_phys->dn_nblkptr;
1214789Sahrens 		data = dn->dn_phys->dn_blkptr;
1215789Sahrens 	} else {
1216789Sahrens 		uint64_t blkid = dbuf_whichblock(dn, *offset) >> (epbs * lvl);
1217789Sahrens 		error = dbuf_hold_impl(dn, lvl, blkid, TRUE, FTAG, &db);
1218789Sahrens 		if (error) {
1219789Sahrens 			if (error == ENOENT)
1220789Sahrens 				return (hole ? 0 : ESRCH);
1221789Sahrens 			return (error);
1222789Sahrens 		}
1223*1544Seschrock 		(void) dbuf_read(db, NULL,
1224*1544Seschrock 		    DB_RF_MUST_SUCCEED | DB_RF_HAVESTRUCT);
1225789Sahrens 		data = db->db.db_data;
1226789Sahrens 	}
1227789Sahrens 
1228789Sahrens 	if (lvl == 0) {
1229789Sahrens 		dnode_phys_t *dnp = data;
1230789Sahrens 		span = DNODE_SHIFT;
1231789Sahrens 		ASSERT(dn->dn_type == DMU_OT_DNODE);
1232789Sahrens 
1233789Sahrens 		for (i = (*offset >> span) & (blkfill - 1); i < blkfill; i++) {
1234789Sahrens 			if (!dnp[i].dn_type == hole)
1235789Sahrens 				break;
1236789Sahrens 			*offset += 1ULL << span;
1237789Sahrens 		}
1238789Sahrens 		if (i == blkfill)
1239789Sahrens 			error = ESRCH;
1240789Sahrens 	} else {
1241789Sahrens 		blkptr_t *bp = data;
1242789Sahrens 		span = (lvl - 1) * epbs + dn->dn_datablkshift;
1243789Sahrens 		minfill = 0;
1244789Sahrens 		maxfill = blkfill << ((lvl - 1) * epbs);
1245789Sahrens 
1246789Sahrens 		if (hole)
1247789Sahrens 			maxfill--;
1248789Sahrens 		else
1249789Sahrens 			minfill++;
1250789Sahrens 
1251789Sahrens 		for (i = (*offset >> span) & ((1ULL << epbs) - 1);
1252789Sahrens 		    i < epb; i++) {
1253789Sahrens 			if (bp[i].blk_fill >= minfill &&
1254789Sahrens 			    bp[i].blk_fill <= maxfill)
1255789Sahrens 				break;
1256789Sahrens 			*offset += 1ULL << span;
1257789Sahrens 		}
1258789Sahrens 		if (i >= epb)
1259789Sahrens 			error = ESRCH;
1260789Sahrens 	}
1261789Sahrens 
1262789Sahrens 	if (db)
1263*1544Seschrock 		dbuf_rele(db, FTAG);
1264789Sahrens 
1265789Sahrens 	return (error);
1266789Sahrens }
1267789Sahrens 
1268789Sahrens /*
1269789Sahrens  * Find the next hole, data, or sparse region at or after *offset.
1270789Sahrens  * The value 'blkfill' tells us how many items we expect to find
1271789Sahrens  * in an L0 data block; this value is 1 for normal objects,
1272789Sahrens  * DNODES_PER_BLOCK for the meta dnode, and some fraction of
1273789Sahrens  * DNODES_PER_BLOCK when searching for sparse regions thereof.
1274789Sahrens  * Examples:
1275789Sahrens  *
1276789Sahrens  * dnode_next_offset(dn, hole, offset, 1, 1);
1277789Sahrens  *	Finds the next hole/data in a file.
1278789Sahrens  *	Used in dmu_offset_next().
1279789Sahrens  *
1280789Sahrens  * dnode_next_offset(mdn, hole, offset, 0, DNODES_PER_BLOCK);
1281789Sahrens  *	Finds the next free/allocated dnode an objset's meta-dnode.
1282789Sahrens  *	Used in dmu_object_next().
1283789Sahrens  *
1284789Sahrens  * dnode_next_offset(mdn, TRUE, offset, 2, DNODES_PER_BLOCK >> 2);
1285789Sahrens  *	Finds the next L2 meta-dnode bp that's at most 1/4 full.
1286789Sahrens  *	Used in dmu_object_alloc().
1287789Sahrens  */
1288789Sahrens int
1289789Sahrens dnode_next_offset(dnode_t *dn, boolean_t hole, uint64_t *offset,
1290789Sahrens     int minlvl, uint64_t blkfill)
1291789Sahrens {
1292789Sahrens 	int lvl, maxlvl;
1293789Sahrens 	int error = 0;
1294789Sahrens 	uint64_t initial_offset = *offset;
1295789Sahrens 
1296789Sahrens 	rw_enter(&dn->dn_struct_rwlock, RW_READER);
1297789Sahrens 
1298789Sahrens 	if (dn->dn_phys->dn_nlevels == 0) {
1299789Sahrens 		rw_exit(&dn->dn_struct_rwlock);
1300789Sahrens 		return (ESRCH);
1301789Sahrens 	}
1302789Sahrens 
1303789Sahrens 	if (dn->dn_datablkshift == 0) {
1304789Sahrens 		if (*offset < dn->dn_datablksz) {
1305789Sahrens 			if (hole)
1306789Sahrens 				*offset = dn->dn_datablksz;
1307789Sahrens 		} else {
1308789Sahrens 			error = ESRCH;
1309789Sahrens 		}
1310789Sahrens 		rw_exit(&dn->dn_struct_rwlock);
1311789Sahrens 		return (error);
1312789Sahrens 	}
1313789Sahrens 
1314789Sahrens 	maxlvl = dn->dn_phys->dn_nlevels;
1315789Sahrens 
1316789Sahrens 	for (lvl = minlvl; lvl <= maxlvl; lvl++) {
1317789Sahrens 		error = dnode_next_offset_level(dn, hole, offset, lvl, blkfill);
1318789Sahrens 		if (error == 0)
1319789Sahrens 			break;
1320789Sahrens 	}
1321789Sahrens 
1322789Sahrens 	while (--lvl >= minlvl && error == 0)
1323789Sahrens 		error = dnode_next_offset_level(dn, hole, offset, lvl, blkfill);
1324789Sahrens 
1325789Sahrens 	rw_exit(&dn->dn_struct_rwlock);
1326789Sahrens 
1327789Sahrens 	if (initial_offset > *offset)
1328789Sahrens 		return (ESRCH);
1329789Sahrens 
1330789Sahrens 	return (error);
1331789Sahrens }
1332