xref: /illumos-gate/usr/src/uts/common/fs/zfs/dbuf.c (revision 9c9dc39aa72ac40bb2558d54adfa596d217135d9)
1fa9e4066Sahrens /*
2fa9e4066Sahrens  * CDDL HEADER START
3fa9e4066Sahrens  *
4fa9e4066Sahrens  * The contents of this file are subject to the terms of the
5fa9e4066Sahrens  * Common Development and Distribution License, Version 1.0 only
6fa9e4066Sahrens  * (the "License").  You may not use this file except in compliance
7fa9e4066Sahrens  * with the License.
8fa9e4066Sahrens  *
9fa9e4066Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10fa9e4066Sahrens  * or http://www.opensolaris.org/os/licensing.
11fa9e4066Sahrens  * See the License for the specific language governing permissions
12fa9e4066Sahrens  * and limitations under the License.
13fa9e4066Sahrens  *
14fa9e4066Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
15fa9e4066Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16fa9e4066Sahrens  * If applicable, add the following below this CDDL HEADER, with the
17fa9e4066Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
18fa9e4066Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
19fa9e4066Sahrens  *
20fa9e4066Sahrens  * CDDL HEADER END
21fa9e4066Sahrens  */
22fa9e4066Sahrens /*
23fa9e4066Sahrens  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
24fa9e4066Sahrens  * Use is subject to license terms.
25fa9e4066Sahrens  */
26fa9e4066Sahrens 
27fa9e4066Sahrens #pragma ident	"%Z%%M%	%I%	%E% SMI"
28fa9e4066Sahrens 
29fa9e4066Sahrens #include <sys/zfs_context.h>
30fa9e4066Sahrens #include <sys/dmu.h>
31fa9e4066Sahrens #include <sys/dmu_impl.h>
32fa9e4066Sahrens #include <sys/dbuf.h>
33fa9e4066Sahrens #include <sys/dmu_objset.h>
34fa9e4066Sahrens #include <sys/dsl_dataset.h>
35fa9e4066Sahrens #include <sys/dsl_dir.h>
36fa9e4066Sahrens #include <sys/dmu_tx.h>
37fa9e4066Sahrens #include <sys/spa.h>
38fa9e4066Sahrens #include <sys/zio.h>
39fa9e4066Sahrens #include <sys/dmu_zfetch.h>
40fa9e4066Sahrens 
41fa9e4066Sahrens static void dbuf_destroy(dmu_buf_impl_t *db);
42fa9e4066Sahrens static int dbuf_undirty(dmu_buf_impl_t *db, dmu_tx_t *tx);
43fa9e4066Sahrens static arc_done_func_t dbuf_write_done;
44fa9e4066Sahrens 
45fa9e4066Sahrens /*
46fa9e4066Sahrens  * Global data structures and functions for the dbuf cache.
47fa9e4066Sahrens  */
48fa9e4066Sahrens taskq_t *dbuf_tq;
49fa9e4066Sahrens static kmem_cache_t *dbuf_cache;
50fa9e4066Sahrens 
51fa9e4066Sahrens /* ARGSUSED */
52fa9e4066Sahrens static int
53fa9e4066Sahrens dbuf_cons(void *vdb, void *unused, int kmflag)
54fa9e4066Sahrens {
55fa9e4066Sahrens 	dmu_buf_impl_t *db = vdb;
56fa9e4066Sahrens 	bzero(db, sizeof (dmu_buf_impl_t));
57fa9e4066Sahrens 
58fa9e4066Sahrens 	mutex_init(&db->db_mtx, NULL, MUTEX_DEFAULT, NULL);
59fa9e4066Sahrens 	cv_init(&db->db_changed, NULL, CV_DEFAULT, NULL);
60fa9e4066Sahrens 	refcount_create(&db->db_holds);
61fa9e4066Sahrens 	return (0);
62fa9e4066Sahrens }
63fa9e4066Sahrens 
64fa9e4066Sahrens /* ARGSUSED */
65fa9e4066Sahrens static void
66fa9e4066Sahrens dbuf_dest(void *vdb, void *unused)
67fa9e4066Sahrens {
68fa9e4066Sahrens 	dmu_buf_impl_t *db = vdb;
69fa9e4066Sahrens 	mutex_destroy(&db->db_mtx);
70fa9e4066Sahrens 	cv_destroy(&db->db_changed);
71fa9e4066Sahrens 	refcount_destroy(&db->db_holds);
72fa9e4066Sahrens }
73fa9e4066Sahrens 
74fa9e4066Sahrens /*
75fa9e4066Sahrens  * dbuf hash table routines
76fa9e4066Sahrens  */
77fa9e4066Sahrens static dbuf_hash_table_t dbuf_hash_table;
78fa9e4066Sahrens 
79fa9e4066Sahrens static uint64_t dbuf_hash_count;
80fa9e4066Sahrens 
81fa9e4066Sahrens static uint64_t
82fa9e4066Sahrens dbuf_hash(void *os, uint64_t obj, uint8_t lvl, uint64_t blkid)
83fa9e4066Sahrens {
84fa9e4066Sahrens 	uintptr_t osv = (uintptr_t)os;
85fa9e4066Sahrens 	uint64_t crc = -1ULL;
86fa9e4066Sahrens 
87fa9e4066Sahrens 	ASSERT(zfs_crc64_table[128] == ZFS_CRC64_POLY);
88fa9e4066Sahrens 	crc = (crc >> 8) ^ zfs_crc64_table[(crc ^ (lvl)) & 0xFF];
89fa9e4066Sahrens 	crc = (crc >> 8) ^ zfs_crc64_table[(crc ^ (osv >> 6)) & 0xFF];
90fa9e4066Sahrens 	crc = (crc >> 8) ^ zfs_crc64_table[(crc ^ (obj >> 0)) & 0xFF];
91fa9e4066Sahrens 	crc = (crc >> 8) ^ zfs_crc64_table[(crc ^ (obj >> 8)) & 0xFF];
92fa9e4066Sahrens 	crc = (crc >> 8) ^ zfs_crc64_table[(crc ^ (blkid >> 0)) & 0xFF];
93fa9e4066Sahrens 	crc = (crc >> 8) ^ zfs_crc64_table[(crc ^ (blkid >> 8)) & 0xFF];
94fa9e4066Sahrens 
95fa9e4066Sahrens 	crc ^= (osv>>14) ^ (obj>>16) ^ (blkid>>16);
96fa9e4066Sahrens 
97fa9e4066Sahrens 	return (crc);
98fa9e4066Sahrens }
99fa9e4066Sahrens 
100fa9e4066Sahrens #define	DBUF_HASH(os, obj, level, blkid) dbuf_hash(os, obj, level, blkid);
101fa9e4066Sahrens 
102fa9e4066Sahrens #define	DBUF_EQUAL(dbuf, os, obj, level, blkid)		\
103fa9e4066Sahrens 	((dbuf)->db.db_object == (obj) &&		\
104fa9e4066Sahrens 	(dbuf)->db_objset == (os) &&			\
105fa9e4066Sahrens 	(dbuf)->db_level == (level) &&			\
106fa9e4066Sahrens 	(dbuf)->db_blkid == (blkid))
107fa9e4066Sahrens 
108fa9e4066Sahrens dmu_buf_impl_t *
109fa9e4066Sahrens dbuf_find(dnode_t *dn, uint8_t level, uint64_t blkid)
110fa9e4066Sahrens {
111fa9e4066Sahrens 	dbuf_hash_table_t *h = &dbuf_hash_table;
112fa9e4066Sahrens 	objset_impl_t *os = dn->dn_objset;
113fa9e4066Sahrens 	uint64_t obj = dn->dn_object;
114fa9e4066Sahrens 	uint64_t hv = DBUF_HASH(os, obj, level, blkid);
115fa9e4066Sahrens 	uint64_t idx = hv & h->hash_table_mask;
116fa9e4066Sahrens 	dmu_buf_impl_t *db;
117fa9e4066Sahrens 
118fa9e4066Sahrens 	mutex_enter(DBUF_HASH_MUTEX(h, idx));
119fa9e4066Sahrens 	for (db = h->hash_table[idx]; db != NULL; db = db->db_hash_next) {
120fa9e4066Sahrens 		if (DBUF_EQUAL(db, os, obj, level, blkid)) {
121fa9e4066Sahrens 			mutex_enter(&db->db_mtx);
122fa9e4066Sahrens 			if (!refcount_is_zero(&db->db_holds)) {
123fa9e4066Sahrens 				mutex_exit(DBUF_HASH_MUTEX(h, idx));
124fa9e4066Sahrens 				return (db);
125fa9e4066Sahrens 			}
126fa9e4066Sahrens 			mutex_exit(&db->db_mtx);
127fa9e4066Sahrens 		}
128fa9e4066Sahrens 	}
129fa9e4066Sahrens 	mutex_exit(DBUF_HASH_MUTEX(h, idx));
130fa9e4066Sahrens 	return (NULL);
131fa9e4066Sahrens }
132fa9e4066Sahrens 
133fa9e4066Sahrens /*
134fa9e4066Sahrens  * Insert an entry into the hash table.  If there is already an element
135fa9e4066Sahrens  * equal to elem in the hash table, then the already existing element
136fa9e4066Sahrens  * will be returned and the new element will not be inserted.
137fa9e4066Sahrens  * Otherwise returns NULL.
138fa9e4066Sahrens  */
139fa9e4066Sahrens static dmu_buf_impl_t *
140fa9e4066Sahrens dbuf_hash_insert(dmu_buf_impl_t *db)
141fa9e4066Sahrens {
142fa9e4066Sahrens 	dbuf_hash_table_t *h = &dbuf_hash_table;
143fa9e4066Sahrens 	objset_impl_t *os = db->db_objset;
144fa9e4066Sahrens 	uint64_t obj = db->db.db_object;
145fa9e4066Sahrens 	int level = db->db_level;
146fa9e4066Sahrens 	uint64_t blkid = db->db_blkid;
147fa9e4066Sahrens 	uint64_t hv = DBUF_HASH(os, obj, level, blkid);
148fa9e4066Sahrens 	uint64_t idx = hv & h->hash_table_mask;
149fa9e4066Sahrens 	dmu_buf_impl_t *dbf;
150fa9e4066Sahrens 
151fa9e4066Sahrens 	mutex_enter(DBUF_HASH_MUTEX(h, idx));
152fa9e4066Sahrens 	for (dbf = h->hash_table[idx]; dbf != NULL; dbf = dbf->db_hash_next) {
153fa9e4066Sahrens 		if (DBUF_EQUAL(dbf, os, obj, level, blkid)) {
154fa9e4066Sahrens 			mutex_enter(&dbf->db_mtx);
155fa9e4066Sahrens 			if (!refcount_is_zero(&dbf->db_holds)) {
156fa9e4066Sahrens 				mutex_exit(DBUF_HASH_MUTEX(h, idx));
157fa9e4066Sahrens 				return (dbf);
158fa9e4066Sahrens 			}
159fa9e4066Sahrens 			mutex_exit(&dbf->db_mtx);
160fa9e4066Sahrens 		}
161fa9e4066Sahrens 	}
162fa9e4066Sahrens 
163fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
164fa9e4066Sahrens 	db->db_hash_next = h->hash_table[idx];
165fa9e4066Sahrens 	h->hash_table[idx] = db;
166fa9e4066Sahrens 	mutex_exit(DBUF_HASH_MUTEX(h, idx));
167fa9e4066Sahrens 	atomic_add_64(&dbuf_hash_count, 1);
168fa9e4066Sahrens 
169fa9e4066Sahrens 	return (NULL);
170fa9e4066Sahrens }
171fa9e4066Sahrens 
172fa9e4066Sahrens /*
173fa9e4066Sahrens  * Remove an entry from the hash table.  This operation will
174fa9e4066Sahrens  * fail if there are any existing holds on the db.
175fa9e4066Sahrens  */
176fa9e4066Sahrens static void
177fa9e4066Sahrens dbuf_hash_remove(dmu_buf_impl_t *db)
178fa9e4066Sahrens {
179fa9e4066Sahrens 	dbuf_hash_table_t *h = &dbuf_hash_table;
180fa9e4066Sahrens 	uint64_t hv = DBUF_HASH(db->db_objset, db->db.db_object,
181fa9e4066Sahrens 	    db->db_level, db->db_blkid);
182fa9e4066Sahrens 	uint64_t idx = hv & h->hash_table_mask;
183fa9e4066Sahrens 	dmu_buf_impl_t *dbf, **dbp;
184fa9e4066Sahrens 
185fa9e4066Sahrens 	/*
186fa9e4066Sahrens 	 * We musn't hold db_mtx to maintin lock ordering:
187fa9e4066Sahrens 	 * DBUF_HASH_MUTEX > db_mtx.
188fa9e4066Sahrens 	 */
189fa9e4066Sahrens 	ASSERT(refcount_is_zero(&db->db_holds));
190fa9e4066Sahrens 	ASSERT(db->db_dnode != NULL);
191fa9e4066Sahrens 	ASSERT(!MUTEX_HELD(&db->db_mtx));
192fa9e4066Sahrens 
193fa9e4066Sahrens 	mutex_enter(DBUF_HASH_MUTEX(h, idx));
194fa9e4066Sahrens 	dbp = &h->hash_table[idx];
195fa9e4066Sahrens 	while ((dbf = *dbp) != db) {
196fa9e4066Sahrens 		dbp = &dbf->db_hash_next;
197fa9e4066Sahrens 		ASSERT(dbf != NULL);
198fa9e4066Sahrens 	}
199fa9e4066Sahrens 	*dbp = db->db_hash_next;
200fa9e4066Sahrens 	db->db_hash_next = NULL;
201fa9e4066Sahrens 	mutex_exit(DBUF_HASH_MUTEX(h, idx));
202fa9e4066Sahrens 	atomic_add_64(&dbuf_hash_count, -1);
203fa9e4066Sahrens }
204fa9e4066Sahrens 
205fa9e4066Sahrens static int dbuf_evictable(dmu_buf_impl_t *db);
206fa9e4066Sahrens static void dbuf_clear(dmu_buf_impl_t *db);
207fa9e4066Sahrens 
208fa9e4066Sahrens void
209fa9e4066Sahrens dbuf_evict(dmu_buf_impl_t *db)
210fa9e4066Sahrens {
211fa9e4066Sahrens 	int err;
212fa9e4066Sahrens 
213fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
214fa9e4066Sahrens 	err = dbuf_evictable(db);
215fa9e4066Sahrens 	ASSERT(err == TRUE);
216fa9e4066Sahrens 	dbuf_clear(db);
217fa9e4066Sahrens 	dbuf_destroy(db);
218fa9e4066Sahrens }
219fa9e4066Sahrens 
220fa9e4066Sahrens static void
221fa9e4066Sahrens dbuf_evict_user(dmu_buf_impl_t *db)
222fa9e4066Sahrens {
223fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
224fa9e4066Sahrens 
225fa9e4066Sahrens 	if (db->db_level != 0 || db->db_d.db_evict_func == NULL)
226fa9e4066Sahrens 		return;
227fa9e4066Sahrens 
228fa9e4066Sahrens 	if (db->db_d.db_user_data_ptr_ptr)
229fa9e4066Sahrens 		*db->db_d.db_user_data_ptr_ptr = db->db.db_data;
230fa9e4066Sahrens 	db->db_d.db_evict_func(&db->db, db->db_d.db_user_ptr);
231fa9e4066Sahrens 	db->db_d.db_user_ptr = NULL;
232fa9e4066Sahrens 	db->db_d.db_user_data_ptr_ptr = NULL;
233fa9e4066Sahrens 	db->db_d.db_evict_func = NULL;
234fa9e4066Sahrens }
235fa9e4066Sahrens 
236fa9e4066Sahrens void
237fa9e4066Sahrens dbuf_init(void)
238fa9e4066Sahrens {
239fa9e4066Sahrens 	uint64_t hsize = 1;
240fa9e4066Sahrens 	dbuf_hash_table_t *h = &dbuf_hash_table;
241fa9e4066Sahrens 	int i;
242fa9e4066Sahrens 
243fa9e4066Sahrens 	/*
244fa9e4066Sahrens 	 * The hash table is big enough to fill all of physical memory
245fa9e4066Sahrens 	 * with an average 64k block size.  The table will take up
246fa9e4066Sahrens 	 * totalmem*sizeof(void*)/64k bytes (i.e. 128KB/GB with 8-byte
247fa9e4066Sahrens 	 * pointers).
248fa9e4066Sahrens 	 */
249fa9e4066Sahrens 	while (hsize * 65536 < physmem * PAGESIZE)
250fa9e4066Sahrens 		hsize <<= 1;
251fa9e4066Sahrens 
252fa9e4066Sahrens 	h->hash_table_mask = hsize - 1;
253fa9e4066Sahrens 	h->hash_table = kmem_zalloc(hsize * sizeof (void *), KM_SLEEP);
254fa9e4066Sahrens 
255fa9e4066Sahrens 	dbuf_cache = kmem_cache_create("dmu_buf_impl_t",
256fa9e4066Sahrens 	    sizeof (dmu_buf_impl_t),
257fa9e4066Sahrens 	    0, dbuf_cons, dbuf_dest, NULL, NULL, NULL, 0);
258fa9e4066Sahrens 	dbuf_tq = taskq_create("dbuf_tq", 8, maxclsyspri, 50, INT_MAX,
259fa9e4066Sahrens 	    TASKQ_PREPOPULATE);
260fa9e4066Sahrens 
261fa9e4066Sahrens 	for (i = 0; i < DBUF_MUTEXES; i++)
262fa9e4066Sahrens 		mutex_init(&h->hash_mutexes[i], NULL, MUTEX_DEFAULT, NULL);
263fa9e4066Sahrens }
264fa9e4066Sahrens 
265fa9e4066Sahrens void
266fa9e4066Sahrens dbuf_fini(void)
267fa9e4066Sahrens {
268fa9e4066Sahrens 	dbuf_hash_table_t *h = &dbuf_hash_table;
269fa9e4066Sahrens 	int i;
270fa9e4066Sahrens 
271fa9e4066Sahrens 	taskq_destroy(dbuf_tq);
272fa9e4066Sahrens 	dbuf_tq = NULL;
273fa9e4066Sahrens 
274fa9e4066Sahrens 	for (i = 0; i < DBUF_MUTEXES; i++)
275fa9e4066Sahrens 		mutex_destroy(&h->hash_mutexes[i]);
276fa9e4066Sahrens 	kmem_free(h->hash_table, (h->hash_table_mask + 1) * sizeof (void *));
277fa9e4066Sahrens 	kmem_cache_destroy(dbuf_cache);
278fa9e4066Sahrens }
279fa9e4066Sahrens 
280fa9e4066Sahrens /*
281fa9e4066Sahrens  * Other stuff.
282fa9e4066Sahrens  */
283fa9e4066Sahrens 
284*9c9dc39aSek #ifdef ZFS_DEBUG
285fa9e4066Sahrens static void
286fa9e4066Sahrens dbuf_verify(dmu_buf_impl_t *db)
287fa9e4066Sahrens {
288fa9e4066Sahrens 	int i;
289fa9e4066Sahrens 	dnode_t *dn = db->db_dnode;
290fa9e4066Sahrens 
291fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
292fa9e4066Sahrens 
293fa9e4066Sahrens 	if (!(zfs_flags & ZFS_DEBUG_DBUF_VERIFY))
294fa9e4066Sahrens 		return;
295fa9e4066Sahrens 
296fa9e4066Sahrens 	ASSERT(db->db_objset != NULL);
297fa9e4066Sahrens 	if (dn == NULL) {
298fa9e4066Sahrens 		ASSERT(db->db_parent == NULL);
299fa9e4066Sahrens 		ASSERT(db->db_blkptr == NULL);
300fa9e4066Sahrens 	} else {
301fa9e4066Sahrens 		ASSERT3U(db->db.db_object, ==, dn->dn_object);
302fa9e4066Sahrens 		ASSERT3P(db->db_objset, ==, dn->dn_objset);
303fa9e4066Sahrens 		ASSERT(list_head(&dn->dn_dbufs));
304fa9e4066Sahrens 		ASSERT3U(db->db_level, <, dn->dn_nlevels);
305fa9e4066Sahrens 	}
306fa9e4066Sahrens 	if (db->db_blkid == DB_BONUS_BLKID) {
307fa9e4066Sahrens 		ASSERT(dn != NULL);
308fa9e4066Sahrens 		ASSERT3U(db->db.db_size, ==, dn->dn_bonuslen);
309fa9e4066Sahrens 		ASSERT3U(db->db.db_offset, ==, DB_BONUS_BLKID);
310fa9e4066Sahrens 	} else {
311fa9e4066Sahrens 		ASSERT3U(db->db.db_offset, ==, db->db_blkid * db->db.db_size);
312fa9e4066Sahrens 	}
313fa9e4066Sahrens 
314fa9e4066Sahrens 	if (db->db_level == 0) {
315fa9e4066Sahrens 		void **udpp = db->db_d.db_user_data_ptr_ptr;
316fa9e4066Sahrens 		/* we can be momentarily larger in dnode_set_blksz() */
317fa9e4066Sahrens 		if (db->db_blkid != DB_BONUS_BLKID && dn) {
318fa9e4066Sahrens 			ASSERT3U(db->db.db_size, >=, dn->dn_datablksz);
319fa9e4066Sahrens 		}
320fa9e4066Sahrens 		if (udpp) {
321fa9e4066Sahrens 			ASSERT((refcount_is_zero(&db->db_holds) &&
322fa9e4066Sahrens 			    *udpp == NULL) ||
323fa9e4066Sahrens 			    (!refcount_is_zero(&db->db_holds) &&
324fa9e4066Sahrens 			    *udpp == db->db.db_data));
325fa9e4066Sahrens 		}
326fa9e4066Sahrens 
327fa9e4066Sahrens 		if (IS_DNODE_DNODE(db->db.db_object)) {
328fa9e4066Sahrens 			for (i = 0; i < TXG_SIZE; i++) {
329fa9e4066Sahrens 				/*
330fa9e4066Sahrens 				 * it should only be modified in syncing
331fa9e4066Sahrens 				 * context, so make sure we only have
332fa9e4066Sahrens 				 * one copy of the data.
333fa9e4066Sahrens 				 */
334fa9e4066Sahrens 				ASSERT(db->db_d.db_data_old[i] == NULL ||
335fa9e4066Sahrens 				    db->db_d.db_data_old[i] == db->db_buf);
336fa9e4066Sahrens 			}
337fa9e4066Sahrens 		}
338fa9e4066Sahrens 	}
339fa9e4066Sahrens 
340fa9e4066Sahrens 	/* verify db->db_blkptr */
341fa9e4066Sahrens 	if (db->db_blkptr) {
342fa9e4066Sahrens 		if (db->db_parent == dn->dn_dbuf) {
343fa9e4066Sahrens 			/* db is pointed to by the dnode */
344fa9e4066Sahrens 			/* ASSERT3U(db->db_blkid, <, dn->dn_nblkptr); */
345fa9e4066Sahrens 			if (IS_DNODE_DNODE(db->db.db_object))
346fa9e4066Sahrens 				ASSERT(db->db_parent == NULL);
347fa9e4066Sahrens 			else
348fa9e4066Sahrens 				ASSERT(db->db_parent != NULL);
349fa9e4066Sahrens 			ASSERT3P(db->db_blkptr, ==,
350fa9e4066Sahrens 			    &dn->dn_phys->dn_blkptr[db->db_blkid]);
351fa9e4066Sahrens 		} else {
352fa9e4066Sahrens 			/* db is pointed to by an indirect block */
353fa9e4066Sahrens 			int epb = db->db_parent->db.db_size >> SPA_BLKPTRSHIFT;
354fa9e4066Sahrens 			ASSERT3U(db->db_parent->db_level, ==, db->db_level+1);
355fa9e4066Sahrens 			ASSERT3U(db->db_parent->db.db_object, ==,
356fa9e4066Sahrens 			    db->db.db_object);
357fa9e4066Sahrens 			/*
358fa9e4066Sahrens 			 * dnode_grow_indblksz() can make this fail if we don't
359fa9e4066Sahrens 			 * have the struct_rwlock.  XXX indblksz no longer
360fa9e4066Sahrens 			 * grows.  safe to do this now?
361fa9e4066Sahrens 			 */
362fa9e4066Sahrens 			if (RW_WRITE_HELD(&db->db_dnode->dn_struct_rwlock)) {
363fa9e4066Sahrens 				ASSERT3P(db->db_blkptr, ==,
364fa9e4066Sahrens 				    ((blkptr_t *)db->db_parent->db.db_data +
365fa9e4066Sahrens 				    db->db_blkid % epb));
366fa9e4066Sahrens 			}
367fa9e4066Sahrens 		}
368fa9e4066Sahrens 	}
369fa9e4066Sahrens 	if ((db->db_blkptr == NULL || BP_IS_HOLE(db->db_blkptr)) &&
370fa9e4066Sahrens 	    db->db.db_data && db->db_blkid != DB_BONUS_BLKID &&
371fa9e4066Sahrens 	    db->db_state != DB_FILL && !dn->dn_free_txg) {
372fa9e4066Sahrens 		/*
373fa9e4066Sahrens 		 * If the blkptr isn't set but they have nonzero data,
374fa9e4066Sahrens 		 * it had better be dirty, otherwise we'll lose that
375fa9e4066Sahrens 		 * data when we evict this buffer.
376fa9e4066Sahrens 		 */
377fa9e4066Sahrens 		if (db->db_dirtycnt == 0) {
378fa9e4066Sahrens 			uint64_t *buf = db->db.db_data;
379fa9e4066Sahrens 			int i;
380fa9e4066Sahrens 
381fa9e4066Sahrens 			for (i = 0; i < db->db.db_size >> 3; i++) {
382fa9e4066Sahrens 				ASSERT(buf[i] == 0);
383fa9e4066Sahrens 			}
384fa9e4066Sahrens 		}
385fa9e4066Sahrens 	}
386fa9e4066Sahrens }
387*9c9dc39aSek #endif
388fa9e4066Sahrens 
389fa9e4066Sahrens static void
390fa9e4066Sahrens dbuf_update_data(dmu_buf_impl_t *db)
391fa9e4066Sahrens {
392fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
393fa9e4066Sahrens 	if (db->db_level == 0 && db->db_d.db_user_data_ptr_ptr) {
394fa9e4066Sahrens 		ASSERT(!refcount_is_zero(&db->db_holds));
395fa9e4066Sahrens 		*db->db_d.db_user_data_ptr_ptr = db->db.db_data;
396fa9e4066Sahrens 	}
397fa9e4066Sahrens }
398fa9e4066Sahrens 
399fa9e4066Sahrens static void
400fa9e4066Sahrens dbuf_set_data(dmu_buf_impl_t *db, arc_buf_t *buf)
401fa9e4066Sahrens {
402fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
403fa9e4066Sahrens 	ASSERT(buf->b_data != NULL);
404fa9e4066Sahrens 	db->db_buf = buf;
405fa9e4066Sahrens 	db->db.db_data = buf->b_data;
406fa9e4066Sahrens 	dbuf_update_data(db);
407fa9e4066Sahrens }
408fa9e4066Sahrens 
409fa9e4066Sahrens uint64_t
410fa9e4066Sahrens dbuf_whichblock(dnode_t *dn, uint64_t offset)
411fa9e4066Sahrens {
412fa9e4066Sahrens 	if (dn->dn_datablkshift) {
413fa9e4066Sahrens 		return (offset >> dn->dn_datablkshift);
414fa9e4066Sahrens 	} else {
415fa9e4066Sahrens 		ASSERT3U(offset, <, dn->dn_datablksz);
416fa9e4066Sahrens 		return (0);
417fa9e4066Sahrens 	}
418fa9e4066Sahrens }
419fa9e4066Sahrens 
420fa9e4066Sahrens static void
421fa9e4066Sahrens dbuf_read_done(zio_t *zio, arc_buf_t *buf, void *vdb)
422fa9e4066Sahrens {
423fa9e4066Sahrens 	dmu_buf_impl_t *db = vdb;
424fa9e4066Sahrens 
425fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
426fa9e4066Sahrens 	ASSERT3U(db->db_state, ==, DB_READ);
427fa9e4066Sahrens 	/*
428fa9e4066Sahrens 	 * All reads are synchronous, so we must have a hold on the dbuf
429fa9e4066Sahrens 	 */
430fa9e4066Sahrens 	ASSERT(refcount_count(&db->db_holds) > 0);
431fa9e4066Sahrens 	ASSERT(db->db.db_data == NULL);
432fa9e4066Sahrens 	if (db->db_level == 0 && db->db_d.db_freed_in_flight) {
433fa9e4066Sahrens 		/* we were freed in flight; disregard any error */
434fa9e4066Sahrens 		arc_release(buf, db);
435fa9e4066Sahrens 		bzero(buf->b_data, db->db.db_size);
436fa9e4066Sahrens 		db->db_d.db_freed_in_flight = FALSE;
437fa9e4066Sahrens 		dbuf_set_data(db, buf);
438fa9e4066Sahrens 		db->db_state = DB_CACHED;
439fa9e4066Sahrens 	} else if (zio == NULL || zio->io_error == 0) {
440fa9e4066Sahrens 		dbuf_set_data(db, buf);
441fa9e4066Sahrens 		db->db_state = DB_CACHED;
442fa9e4066Sahrens 	} else {
443fa9e4066Sahrens 		ASSERT(db->db_blkid != DB_BONUS_BLKID);
444fa9e4066Sahrens 		arc_buf_free(buf, db);
445fa9e4066Sahrens 		db->db_state = DB_UNCACHED;
446fa9e4066Sahrens 		ASSERT3P(db->db_buf, ==, NULL);
447fa9e4066Sahrens 	}
448fa9e4066Sahrens 	cv_broadcast(&db->db_changed);
449fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
450fa9e4066Sahrens }
451fa9e4066Sahrens 
452fa9e4066Sahrens void
453fa9e4066Sahrens dbuf_read_impl(dmu_buf_impl_t *db, zio_t *zio, uint32_t flags)
454fa9e4066Sahrens {
455fa9e4066Sahrens 	arc_buf_t *buf;
456fa9e4066Sahrens 	blkptr_t *bp;
457fa9e4066Sahrens 
458fa9e4066Sahrens 	ASSERT(!refcount_is_zero(&db->db_holds));
459fa9e4066Sahrens 	/* We need the struct_rwlock to prevent db_blkptr from changing. */
460fa9e4066Sahrens 	ASSERT(RW_LOCK_HELD(&db->db_dnode->dn_struct_rwlock));
461fa9e4066Sahrens 
462fa9e4066Sahrens 	/*
463fa9e4066Sahrens 	 * prefetch only data blocks (level 0) -- don't prefetch indirect
464fa9e4066Sahrens 	 * blocks
465fa9e4066Sahrens 	 */
466fa9e4066Sahrens 	if ((db->db_level > 0) || (db->db_blkid == DB_BONUS_BLKID)) {
467fa9e4066Sahrens 		flags |= DB_RF_NOPREFETCH;
468fa9e4066Sahrens 	}
469fa9e4066Sahrens 
470fa9e4066Sahrens 	if (((flags & DB_RF_NOPREFETCH) == 0) && (db->db_dnode != NULL)) {
471fa9e4066Sahrens 		dmu_zfetch(&db->db_dnode->dn_zfetch, db->db.db_offset,
472fa9e4066Sahrens 		    db->db.db_size);
473fa9e4066Sahrens 	}
474fa9e4066Sahrens 
475fa9e4066Sahrens 	if (db->db_state == DB_CACHED) {
476fa9e4066Sahrens 		ASSERT(db->db.db_data != NULL);
477fa9e4066Sahrens 		return;
478fa9e4066Sahrens 	}
479fa9e4066Sahrens 
480fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
481fa9e4066Sahrens 
482fa9e4066Sahrens 	if (db->db_state != DB_UNCACHED) {
483fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
484fa9e4066Sahrens 		return;
485fa9e4066Sahrens 	}
486fa9e4066Sahrens 
487fa9e4066Sahrens 	ASSERT3U(db->db_state, ==, DB_UNCACHED);
488fa9e4066Sahrens 
489fa9e4066Sahrens 	if (db->db_blkid == DB_BONUS_BLKID) {
490fa9e4066Sahrens 		ASSERT3U(db->db_dnode->dn_bonuslen, ==, db->db.db_size);
491fa9e4066Sahrens 		buf = arc_buf_alloc(db->db_dnode->dn_objset->os_spa,
492fa9e4066Sahrens 		    DN_MAX_BONUSLEN, db);
493fa9e4066Sahrens 		if (db->db.db_size < DN_MAX_BONUSLEN)
494fa9e4066Sahrens 			bzero(buf->b_data, DN_MAX_BONUSLEN);
495fa9e4066Sahrens 		bcopy(DN_BONUS(db->db_dnode->dn_phys), buf->b_data,
496fa9e4066Sahrens 		    db->db.db_size);
497fa9e4066Sahrens 		dbuf_set_data(db, buf);
498fa9e4066Sahrens 		db->db_state = DB_CACHED;
499fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
500fa9e4066Sahrens 		return;
501fa9e4066Sahrens 	}
502fa9e4066Sahrens 
503fa9e4066Sahrens 	if (db->db_level == 0 && dnode_block_freed(db->db_dnode, db->db_blkid))
504fa9e4066Sahrens 		bp = NULL;
505fa9e4066Sahrens 	else
506fa9e4066Sahrens 		bp = db->db_blkptr;
507fa9e4066Sahrens 
508fa9e4066Sahrens 	if (bp == NULL)
509fa9e4066Sahrens 		dprintf_dbuf(db, "blkptr: %s\n", "NULL");
510fa9e4066Sahrens 	else
511fa9e4066Sahrens 		dprintf_dbuf_bp(db, bp, "%s", "blkptr:");
512fa9e4066Sahrens 
513fa9e4066Sahrens 	if (bp == NULL || BP_IS_HOLE(bp)) {
514fa9e4066Sahrens 		ASSERT(bp == NULL || BP_IS_HOLE(bp));
515fa9e4066Sahrens 		dbuf_set_data(db, arc_buf_alloc(db->db_dnode->dn_objset->os_spa,
516fa9e4066Sahrens 		    db->db.db_size, db));
517fa9e4066Sahrens 		bzero(db->db.db_data, db->db.db_size);
518fa9e4066Sahrens 		db->db_state = DB_CACHED;
519fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
520fa9e4066Sahrens 		return;
521fa9e4066Sahrens 	}
522fa9e4066Sahrens 
523fa9e4066Sahrens 	db->db_state = DB_READ;
524fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
525fa9e4066Sahrens 
526fa9e4066Sahrens 	/* ZIO_FLAG_CANFAIL callers have to check the parent zio's error */
527fa9e4066Sahrens 	(void) arc_read(zio, db->db_dnode->dn_objset->os_spa, bp,
528fa9e4066Sahrens 	    db->db_level > 0 ? byteswap_uint64_array :
529fa9e4066Sahrens 	    dmu_ot[db->db_dnode->dn_type].ot_byteswap,
530fa9e4066Sahrens 	    dbuf_read_done, db, ZIO_PRIORITY_SYNC_READ,
531fa9e4066Sahrens 	    (flags & DB_RF_CANFAIL) ? ZIO_FLAG_CANFAIL : ZIO_FLAG_MUSTSUCCEED,
532fa9e4066Sahrens 	    ARC_NOWAIT);
533fa9e4066Sahrens }
534fa9e4066Sahrens 
535fa9e4066Sahrens static int
536fa9e4066Sahrens dbuf_read_generic(dmu_buf_impl_t *db, uint32_t flags)
537fa9e4066Sahrens {
538fa9e4066Sahrens 	zio_t *zio;
539fa9e4066Sahrens 	int err;
540fa9e4066Sahrens 
541fa9e4066Sahrens 	/*
542fa9e4066Sahrens 	 * We don't have to hold the mutex to check db_state because it
543fa9e4066Sahrens 	 * can't be freed while we have a hold on the buffer.
544fa9e4066Sahrens 	 */
545fa9e4066Sahrens 	ASSERT(!refcount_is_zero(&db->db_holds));
546fa9e4066Sahrens 	if (db->db_state == DB_CACHED)
547fa9e4066Sahrens 		return (0);
548fa9e4066Sahrens 
549fa9e4066Sahrens 	if (db->db_state == DB_UNCACHED) {
550fa9e4066Sahrens 		zio = zio_root(db->db_dnode->dn_objset->os_spa, NULL, NULL,
551fa9e4066Sahrens 		    ZIO_FLAG_CANFAIL);
552fa9e4066Sahrens 		if ((flags & DB_RF_HAVESTRUCT) == 0)
553fa9e4066Sahrens 			rw_enter(&db->db_dnode->dn_struct_rwlock, RW_READER);
554fa9e4066Sahrens 		dbuf_read_impl(db, zio, flags);
555fa9e4066Sahrens 		if ((flags & DB_RF_HAVESTRUCT) == 0)
556fa9e4066Sahrens 			rw_exit(&db->db_dnode->dn_struct_rwlock);
557fa9e4066Sahrens 		err = zio_wait(zio);
558fa9e4066Sahrens 		if (err)
559fa9e4066Sahrens 			return (err);
560fa9e4066Sahrens 	}
561fa9e4066Sahrens 
562fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
563fa9e4066Sahrens 	while (db->db_state == DB_READ || db->db_state == DB_FILL) {
564fa9e4066Sahrens 		ASSERT(db->db_state == DB_READ ||
565fa9e4066Sahrens 		    (flags & DB_RF_HAVESTRUCT) == 0);
566fa9e4066Sahrens 		cv_wait(&db->db_changed, &db->db_mtx);
567fa9e4066Sahrens 	}
568fa9e4066Sahrens 	ASSERT3U(db->db_state, ==, DB_CACHED);
569fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
570fa9e4066Sahrens 
571fa9e4066Sahrens 	return (0);
572fa9e4066Sahrens }
573fa9e4066Sahrens 
574fa9e4066Sahrens #pragma weak dmu_buf_read = dbuf_read
575fa9e4066Sahrens void
576fa9e4066Sahrens dbuf_read(dmu_buf_impl_t *db)
577fa9e4066Sahrens {
578fa9e4066Sahrens 	int err;
579fa9e4066Sahrens 
580fa9e4066Sahrens 	err = dbuf_read_generic(db, DB_RF_MUST_SUCCEED);
581fa9e4066Sahrens 	ASSERT(err == 0);
582fa9e4066Sahrens }
583fa9e4066Sahrens 
584fa9e4066Sahrens #pragma weak dmu_buf_read_canfail = dbuf_read_canfail
585fa9e4066Sahrens int
586fa9e4066Sahrens dbuf_read_canfail(dmu_buf_impl_t *db)
587fa9e4066Sahrens {
588fa9e4066Sahrens 	return (dbuf_read_generic(db, DB_RF_CANFAIL));
589fa9e4066Sahrens }
590fa9e4066Sahrens 
591fa9e4066Sahrens void
592fa9e4066Sahrens dbuf_read_havestruct(dmu_buf_impl_t *db)
593fa9e4066Sahrens {
594fa9e4066Sahrens 	int err;
595fa9e4066Sahrens 
596fa9e4066Sahrens 	ASSERT(RW_LOCK_HELD(&db->db_dnode->dn_struct_rwlock));
597fa9e4066Sahrens 	err = dbuf_read_generic(db, (DB_RF_HAVESTRUCT | DB_RF_NOPREFETCH));
598fa9e4066Sahrens 	ASSERT(err == 0);
599fa9e4066Sahrens }
600fa9e4066Sahrens 
601fa9e4066Sahrens static void
602fa9e4066Sahrens dbuf_noread(dmu_buf_impl_t *db)
603fa9e4066Sahrens {
604fa9e4066Sahrens 	ASSERT(!refcount_is_zero(&db->db_holds));
605fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
606fa9e4066Sahrens 	while (db->db_state == DB_READ || db->db_state == DB_FILL)
607fa9e4066Sahrens 		cv_wait(&db->db_changed, &db->db_mtx);
608fa9e4066Sahrens 	if (db->db_state == DB_UNCACHED) {
609fa9e4066Sahrens 		int blksz = (db->db_blkid == DB_BONUS_BLKID) ?
610fa9e4066Sahrens 		    DN_MAX_BONUSLEN : db->db.db_size;
611fa9e4066Sahrens 		ASSERT(db->db.db_data == NULL);
612fa9e4066Sahrens 		dbuf_set_data(db, arc_buf_alloc(db->db_dnode->dn_objset->os_spa,
613fa9e4066Sahrens 		    blksz, db));
614fa9e4066Sahrens 		db->db_state = DB_FILL;
615fa9e4066Sahrens 	} else {
616fa9e4066Sahrens 		ASSERT3U(db->db_state, ==, DB_CACHED);
617fa9e4066Sahrens 	}
618fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
619fa9e4066Sahrens }
620fa9e4066Sahrens 
621fa9e4066Sahrens /*
622fa9e4066Sahrens  * This is our just-in-time copy function.  It makes a copy of
623fa9e4066Sahrens  * buffers, that have been modified in a previous transaction
624fa9e4066Sahrens  * group, before we modify them in the current active group.
625fa9e4066Sahrens  *
626fa9e4066Sahrens  * This function is used in two places: when we are dirtying a
627fa9e4066Sahrens  * buffer for the first time in a txg, and when we are freeing
628fa9e4066Sahrens  * a range in a dnode that includes this buffer.
629fa9e4066Sahrens  *
630fa9e4066Sahrens  * Note that when we are called from dbuf_free_range() we do
631fa9e4066Sahrens  * not put a hold on the buffer, we just traverse the active
632fa9e4066Sahrens  * dbuf list for the dnode.
633fa9e4066Sahrens  */
634fa9e4066Sahrens static void
635fa9e4066Sahrens dbuf_fix_old_data(dmu_buf_impl_t *db, uint64_t txg)
636fa9e4066Sahrens {
637fa9e4066Sahrens 	arc_buf_t **quiescing, **syncing;
638fa9e4066Sahrens 	int size = (db->db_blkid == DB_BONUS_BLKID) ?
639fa9e4066Sahrens 	    DN_MAX_BONUSLEN : db->db.db_size;
640fa9e4066Sahrens 
641fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
642fa9e4066Sahrens 	ASSERT(db->db.db_data != NULL);
643fa9e4066Sahrens 
644fa9e4066Sahrens 	quiescing = &db->db_d.db_data_old[(txg-1)&TXG_MASK];
645fa9e4066Sahrens 	syncing = &db->db_d.db_data_old[(txg-2)&TXG_MASK];
646fa9e4066Sahrens 
647fa9e4066Sahrens 	/*
648fa9e4066Sahrens 	 * If this buffer is referenced from the current quiescing
649fa9e4066Sahrens 	 * transaction group: either make a copy and reset the reference
650fa9e4066Sahrens 	 * to point to the copy, or (if there a no active holders) just
651fa9e4066Sahrens 	 * null out the current db_data pointer.
652fa9e4066Sahrens 	 */
653fa9e4066Sahrens 	if (*quiescing == db->db_buf) {
654fa9e4066Sahrens 		/*
655fa9e4066Sahrens 		 * If the quiescing txg is "dirty", then we better not
656fa9e4066Sahrens 		 * be referencing the same buffer from the syncing txg.
657fa9e4066Sahrens 		 */
658fa9e4066Sahrens 		ASSERT(*syncing != db->db_buf);
659fa9e4066Sahrens 		if (refcount_count(&db->db_holds) > db->db_dirtycnt) {
660fa9e4066Sahrens 			*quiescing = arc_buf_alloc(
661fa9e4066Sahrens 			    db->db_dnode->dn_objset->os_spa, size, db);
662fa9e4066Sahrens 			bcopy(db->db.db_data, (*quiescing)->b_data, size);
663fa9e4066Sahrens 		} else {
664fa9e4066Sahrens 			db->db.db_data = NULL;
665fa9e4066Sahrens 			db->db_buf = NULL;
666fa9e4066Sahrens 			db->db_state = DB_UNCACHED;
667fa9e4066Sahrens 		}
668fa9e4066Sahrens 		return;
669fa9e4066Sahrens 	}
670fa9e4066Sahrens 
671fa9e4066Sahrens 	/*
672fa9e4066Sahrens 	 * If this buffer is referenced from the current syncing
673fa9e4066Sahrens 	 * transaction group: either
674fa9e4066Sahrens 	 *	1 - make a copy and reset the reference, or
675fa9e4066Sahrens 	 *	2 - if there are no holders, just null the current db_data.
676fa9e4066Sahrens 	 */
677fa9e4066Sahrens 	if (*syncing == db->db_buf) {
678fa9e4066Sahrens 		ASSERT3P(*quiescing, ==, NULL);
679fa9e4066Sahrens 		ASSERT3U(db->db_dirtycnt, ==, 1);
680fa9e4066Sahrens 		if (refcount_count(&db->db_holds) > db->db_dirtycnt) {
681fa9e4066Sahrens 			/* we can't copy if we have already started a write */
682fa9e4066Sahrens 			ASSERT(*syncing != db->db_data_pending);
683fa9e4066Sahrens 			*syncing = arc_buf_alloc(
684fa9e4066Sahrens 			    db->db_dnode->dn_objset->os_spa, size, db);
685fa9e4066Sahrens 			bcopy(db->db.db_data, (*syncing)->b_data, size);
686fa9e4066Sahrens 		} else {
687fa9e4066Sahrens 			db->db.db_data = NULL;
688fa9e4066Sahrens 			db->db_buf = NULL;
689fa9e4066Sahrens 			db->db_state = DB_UNCACHED;
690fa9e4066Sahrens 		}
691fa9e4066Sahrens 	}
692fa9e4066Sahrens }
693fa9e4066Sahrens 
694fa9e4066Sahrens void
695fa9e4066Sahrens dbuf_unoverride(dmu_buf_impl_t *db, uint64_t txg)
696fa9e4066Sahrens {
697fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
698fa9e4066Sahrens 	if (db->db_d.db_overridden_by[txg&TXG_MASK] == IN_DMU_SYNC) {
699fa9e4066Sahrens 		db->db_d.db_overridden_by[txg&TXG_MASK] = NULL;
700fa9e4066Sahrens 	} else if (db->db_d.db_overridden_by[txg&TXG_MASK] != NULL) {
701fa9e4066Sahrens 		/* free this block */
702fa9e4066Sahrens 		ASSERT(list_link_active(&db->db_dirty_node[txg&TXG_MASK]) ||
703fa9e4066Sahrens 		    db->db_dnode->dn_free_txg == txg);
704fa9e4066Sahrens 		if (!BP_IS_HOLE(db->db_d.db_overridden_by[txg&TXG_MASK])) {
705fa9e4066Sahrens 			/* XXX can get silent EIO here */
706fa9e4066Sahrens 			(void) arc_free(NULL, db->db_dnode->dn_objset->os_spa,
707fa9e4066Sahrens 			    txg, db->db_d.db_overridden_by[txg&TXG_MASK],
708fa9e4066Sahrens 			    NULL, NULL, ARC_WAIT);
709fa9e4066Sahrens 		}
710fa9e4066Sahrens 		kmem_free(db->db_d.db_overridden_by[txg&TXG_MASK],
711fa9e4066Sahrens 		    sizeof (blkptr_t));
712fa9e4066Sahrens 		db->db_d.db_overridden_by[txg&TXG_MASK] = NULL;
713fa9e4066Sahrens 		/* release the already-written buffer */
714fa9e4066Sahrens 		arc_release(db->db_d.db_data_old[txg&TXG_MASK], db);
715fa9e4066Sahrens 	}
716fa9e4066Sahrens }
717fa9e4066Sahrens 
718fa9e4066Sahrens void
719fa9e4066Sahrens dbuf_free_range(dnode_t *dn, uint64_t blkid, uint64_t nblks, dmu_tx_t *tx)
720fa9e4066Sahrens {
721fa9e4066Sahrens 	dmu_buf_impl_t *db, *db_next;
722fa9e4066Sahrens 	uint64_t txg = tx->tx_txg;
723fa9e4066Sahrens 
724fa9e4066Sahrens 	dprintf_dnode(dn, "blkid=%llu nblks=%llu\n", blkid, nblks);
725fa9e4066Sahrens 	mutex_enter(&dn->dn_dbufs_mtx);
726fa9e4066Sahrens 	for (db = list_head(&dn->dn_dbufs); db; db = db_next) {
727fa9e4066Sahrens 		db_next = list_next(&dn->dn_dbufs, db);
728fa9e4066Sahrens 		if ((db->db_level != 0) || (db->db_blkid == DB_BONUS_BLKID))
729fa9e4066Sahrens 			continue;
730fa9e4066Sahrens 		dprintf_dbuf(db, "found buf %s\n", "");
731fa9e4066Sahrens 		if (db->db_blkid < blkid ||
732fa9e4066Sahrens 		    db->db_blkid >= blkid+nblks)
733fa9e4066Sahrens 			continue;
734fa9e4066Sahrens 
735fa9e4066Sahrens 		/* found a level 0 buffer in the range */
736fa9e4066Sahrens 		if (dbuf_undirty(db, tx))
737fa9e4066Sahrens 			continue;
738fa9e4066Sahrens 
739fa9e4066Sahrens 		mutex_enter(&db->db_mtx);
740fa9e4066Sahrens 		if (db->db_state == DB_UNCACHED) {
741fa9e4066Sahrens 			ASSERT(db->db.db_data == NULL);
742fa9e4066Sahrens 			mutex_exit(&db->db_mtx);
743fa9e4066Sahrens 			continue;
744fa9e4066Sahrens 		}
745fa9e4066Sahrens 		if (db->db_state == DB_READ) {
746fa9e4066Sahrens 			/* this will be handled in dbuf_read_done() */
747fa9e4066Sahrens 			db->db_d.db_freed_in_flight = TRUE;
748fa9e4066Sahrens 			mutex_exit(&db->db_mtx);
749fa9e4066Sahrens 			continue;
750fa9e4066Sahrens 		}
751fa9e4066Sahrens 		if (db->db_state == DB_FILL) {
752fa9e4066Sahrens 			/* this will be handled in dbuf_rele() */
753fa9e4066Sahrens 			db->db_d.db_freed_in_flight = TRUE;
754fa9e4066Sahrens 			mutex_exit(&db->db_mtx);
755fa9e4066Sahrens 			continue;
756fa9e4066Sahrens 		}
757fa9e4066Sahrens 
758fa9e4066Sahrens 		/* make a copy of the data if necessary */
759fa9e4066Sahrens 		dbuf_fix_old_data(db, txg);
760fa9e4066Sahrens 
761fa9e4066Sahrens 		if (db->db.db_data) {
762fa9e4066Sahrens 			/* fill in with appropriate data */
763fa9e4066Sahrens 			arc_release(db->db_buf, db);
764fa9e4066Sahrens 			bzero(db->db.db_data, db->db.db_size);
765fa9e4066Sahrens 		}
766fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
767fa9e4066Sahrens 	}
768fa9e4066Sahrens 	mutex_exit(&dn->dn_dbufs_mtx);
769fa9e4066Sahrens }
770fa9e4066Sahrens 
771fa9e4066Sahrens static int
772fa9e4066Sahrens dbuf_new_block(dmu_buf_impl_t *db, dmu_tx_t *tx)
773fa9e4066Sahrens {
774fa9e4066Sahrens 	dsl_dataset_t *ds = db->db_objset->os_dsl_dataset;
775fa9e4066Sahrens 	uint64_t birth_txg = 0;
776fa9e4066Sahrens 
777fa9e4066Sahrens 	/* Don't count meta-objects */
778fa9e4066Sahrens 	if (ds == NULL)
779fa9e4066Sahrens 		return (FALSE);
780fa9e4066Sahrens 
781fa9e4066Sahrens 	/*
782fa9e4066Sahrens 	 * We don't need any locking to protect db_blkptr:
783fa9e4066Sahrens 	 * If it's syncing, then db_dirtied will be set so we'll
784fa9e4066Sahrens 	 * ignore db_blkptr.
785fa9e4066Sahrens 	 */
786fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx)); /* XXX strictly necessary? */
787fa9e4066Sahrens 	/* If we have been dirtied since the last snapshot, its not new */
788fa9e4066Sahrens 	if (db->db_dirtied)
789fa9e4066Sahrens 		birth_txg = db->db_dirtied;
790fa9e4066Sahrens 	else if (db->db_blkptr)
791fa9e4066Sahrens 		birth_txg = db->db_blkptr->blk_birth;
792fa9e4066Sahrens 
793fa9e4066Sahrens 	if (birth_txg)
794fa9e4066Sahrens 		return (!dsl_dataset_block_freeable(ds, birth_txg, tx));
795fa9e4066Sahrens 	else
796fa9e4066Sahrens 		return (TRUE);
797fa9e4066Sahrens }
798fa9e4066Sahrens 
799fa9e4066Sahrens void
800fa9e4066Sahrens dbuf_new_size(dmu_buf_impl_t *db, int size, dmu_tx_t *tx)
801fa9e4066Sahrens {
802fa9e4066Sahrens 	arc_buf_t *buf, *obuf;
803fa9e4066Sahrens 	int osize = db->db.db_size;
804fa9e4066Sahrens 
805fa9e4066Sahrens 	/* XXX does *this* func really need the lock? */
806fa9e4066Sahrens 	ASSERT(RW_WRITE_HELD(&db->db_dnode->dn_struct_rwlock));
807fa9e4066Sahrens 
808fa9e4066Sahrens 	ASSERT3U(osize, <=, size);
809fa9e4066Sahrens 	if (osize == size)
810fa9e4066Sahrens 		return;
811fa9e4066Sahrens 
812fa9e4066Sahrens 	/*
813fa9e4066Sahrens 	 * This call to dbuf_will_dirty() with the dn_struct_rwlock held
814fa9e4066Sahrens 	 * is OK, because there can be no other references to the db
815fa9e4066Sahrens 	 * when we are changing its size, so no concurrent DB_FILL can
816fa9e4066Sahrens 	 * be happening.
817fa9e4066Sahrens 	 */
818fa9e4066Sahrens 	/* Make a copy of the data if necessary */
819fa9e4066Sahrens 	dbuf_will_dirty(db, tx);
820fa9e4066Sahrens 
821fa9e4066Sahrens 	/* create the data buffer for the new block */
822fa9e4066Sahrens 	buf = arc_buf_alloc(db->db_dnode->dn_objset->os_spa, size, db);
823fa9e4066Sahrens 
824fa9e4066Sahrens 	/* copy old block data to the new block */
825fa9e4066Sahrens 	obuf = db->db_buf;
826fa9e4066Sahrens 	bcopy(obuf->b_data, buf->b_data, osize);
827fa9e4066Sahrens 	/* zero the remainder */
828fa9e4066Sahrens 	bzero((uint8_t *)buf->b_data + osize, size - osize);
829fa9e4066Sahrens 
830fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
831fa9e4066Sahrens 	/* ASSERT3U(refcount_count(&db->db_holds), ==, 1); */
832fa9e4066Sahrens 	dbuf_set_data(db, buf);
833fa9e4066Sahrens 	arc_buf_free(obuf, db);
834fa9e4066Sahrens 	db->db.db_size = size;
835fa9e4066Sahrens 
836fa9e4066Sahrens 	/* fix up the dirty info */
837fa9e4066Sahrens 	if (db->db_level == 0)
838fa9e4066Sahrens 		db->db_d.db_data_old[tx->tx_txg&TXG_MASK] = buf;
839fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
840fa9e4066Sahrens 
841fa9e4066Sahrens 	dnode_willuse_space(db->db_dnode, size-osize, tx);
842fa9e4066Sahrens }
843fa9e4066Sahrens 
844fa9e4066Sahrens void
845fa9e4066Sahrens dbuf_dirty(dmu_buf_impl_t *db, dmu_tx_t *tx)
846fa9e4066Sahrens {
847fa9e4066Sahrens 	dnode_t *dn = db->db_dnode;
848fa9e4066Sahrens 	objset_impl_t *os = dn->dn_objset;
849fa9e4066Sahrens 	int drop_struct_lock = FALSE;
850fa9e4066Sahrens 	int txgoff = tx->tx_txg & TXG_MASK;
851fa9e4066Sahrens 
852fa9e4066Sahrens 	ASSERT(tx->tx_txg != 0);
853fa9e4066Sahrens 	ASSERT(!refcount_is_zero(&db->db_holds));
854*9c9dc39aSek 	DMU_TX_DIRTY_BUF(tx, db);
855fa9e4066Sahrens 
856fa9e4066Sahrens 	/*
857fa9e4066Sahrens 	 * Shouldn't dirty a regular buffer in syncing context.  Private
858fa9e4066Sahrens 	 * objects may be dirtied in syncing context, but only if they
859fa9e4066Sahrens 	 * were already pre-dirtied in open context.
860fa9e4066Sahrens 	 * XXX We may want to prohibit dirtying in syncing context even
861fa9e4066Sahrens 	 * if they did pre-dirty.
862fa9e4066Sahrens 	 */
863fa9e4066Sahrens 	ASSERT(!(dmu_tx_is_syncing(tx) &&
864fa9e4066Sahrens 	    !BP_IS_HOLE(&dn->dn_objset->os_rootbp) &&
865fa9e4066Sahrens 	    !(dn->dn_object & DMU_PRIVATE_OBJECT) &&
866fa9e4066Sahrens 	    dn->dn_objset->os_dsl_dataset != NULL &&
867fa9e4066Sahrens 	    !dsl_dir_is_private(
868fa9e4066Sahrens 	    dn->dn_objset->os_dsl_dataset->ds_dir)));
869fa9e4066Sahrens 
870fa9e4066Sahrens 	/*
871fa9e4066Sahrens 	 * We make this assert for private objects as well, but after we
872fa9e4066Sahrens 	 * check if we're already dirty.  They are allowed to re-dirty
873fa9e4066Sahrens 	 * in syncing context.
874fa9e4066Sahrens 	 */
875fa9e4066Sahrens 	ASSERT(dn->dn_object & DMU_PRIVATE_OBJECT ||
876fa9e4066Sahrens 	    dn->dn_dirtyctx == DN_UNDIRTIED ||
877fa9e4066Sahrens 	    dn->dn_dirtyctx ==
878fa9e4066Sahrens 	    (dmu_tx_is_syncing(tx) ? DN_DIRTY_SYNC : DN_DIRTY_OPEN));
879fa9e4066Sahrens 
880fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
881fa9e4066Sahrens 	/* XXX make this true for indirects too? */
882fa9e4066Sahrens 	ASSERT(db->db_level != 0 || db->db_state == DB_CACHED ||
883fa9e4066Sahrens 	    db->db_state == DB_FILL);
884fa9e4066Sahrens 
885fa9e4066Sahrens 	/*
886fa9e4066Sahrens 	 * If this buffer is currently part of an "overridden" region,
887fa9e4066Sahrens 	 * we now need to remove it from that region.
888fa9e4066Sahrens 	 */
889fa9e4066Sahrens 	if (db->db_level == 0 && db->db_blkid != DB_BONUS_BLKID &&
890fa9e4066Sahrens 	    db->db_d.db_overridden_by[txgoff] != NULL) {
891fa9e4066Sahrens 		dbuf_unoverride(db, tx->tx_txg);
892fa9e4066Sahrens 	}
893fa9e4066Sahrens 
894fa9e4066Sahrens 	mutex_enter(&dn->dn_mtx);
895fa9e4066Sahrens 	/*
896fa9e4066Sahrens 	 * Don't set dirtyctx to SYNC if we're just modifying this as we
897fa9e4066Sahrens 	 * initialize the objset.
898fa9e4066Sahrens 	 */
899fa9e4066Sahrens 	if (dn->dn_dirtyctx == DN_UNDIRTIED &&
900fa9e4066Sahrens 	    !BP_IS_HOLE(&dn->dn_objset->os_rootbp)) {
901fa9e4066Sahrens 		dn->dn_dirtyctx =
902fa9e4066Sahrens 		    (dmu_tx_is_syncing(tx) ? DN_DIRTY_SYNC : DN_DIRTY_OPEN);
903fa9e4066Sahrens 		ASSERT(dn->dn_dirtyctx_firstset == NULL);
904fa9e4066Sahrens 		dn->dn_dirtyctx_firstset = kmem_alloc(1, KM_SLEEP);
905fa9e4066Sahrens 	}
906fa9e4066Sahrens 	mutex_exit(&dn->dn_mtx);
907fa9e4066Sahrens 
908fa9e4066Sahrens 	/*
909fa9e4066Sahrens 	 * If this buffer is already dirty, we're done.
910fa9e4066Sahrens 	 */
911fa9e4066Sahrens 	if (list_link_active(&db->db_dirty_node[txgoff])) {
912fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
913fa9e4066Sahrens 		return;
914fa9e4066Sahrens 	}
915fa9e4066Sahrens 
916fa9e4066Sahrens 	/*
917fa9e4066Sahrens 	 * Only valid if not already dirty.
918fa9e4066Sahrens 	 */
919fa9e4066Sahrens 	ASSERT(dn->dn_dirtyctx == DN_UNDIRTIED || dn->dn_dirtyctx ==
920fa9e4066Sahrens 	    (dmu_tx_is_syncing(tx) ? DN_DIRTY_SYNC : DN_DIRTY_OPEN));
921fa9e4066Sahrens 
922fa9e4066Sahrens 	ASSERT3U(dn->dn_nlevels, >, db->db_level);
923fa9e4066Sahrens 	ASSERT((dn->dn_phys->dn_nlevels == 0 && db->db_level == 0) ||
924fa9e4066Sahrens 	    dn->dn_phys->dn_nlevels > db->db_level ||
925fa9e4066Sahrens 	    dn->dn_next_nlevels[txgoff] > db->db_level ||
926fa9e4066Sahrens 	    dn->dn_next_nlevels[(tx->tx_txg-1) & TXG_MASK] > db->db_level ||
927fa9e4066Sahrens 	    dn->dn_next_nlevels[(tx->tx_txg-2) & TXG_MASK] > db->db_level);
928fa9e4066Sahrens 
929fa9e4066Sahrens 	/*
930fa9e4066Sahrens 	 * We should only be dirtying in syncing context if it's the
931fa9e4066Sahrens 	 * mos, a spa os, or we're initializing the os.  However, we are
932fa9e4066Sahrens 	 * allowed to dirty in syncing context provided we already
933fa9e4066Sahrens 	 * dirtied it in open context.  Hence we must make this
934fa9e4066Sahrens 	 * assertion only if we're not already dirty.
935fa9e4066Sahrens 	 */
936fa9e4066Sahrens 	ASSERT(!dmu_tx_is_syncing(tx) ||
937fa9e4066Sahrens 	    os->os_dsl_dataset == NULL ||
938fa9e4066Sahrens 	    !dsl_dir_is_private(os->os_dsl_dataset->ds_dir) ||
939fa9e4066Sahrens 	    !BP_IS_HOLE(&os->os_rootbp));
940fa9e4066Sahrens 	ASSERT(db->db.db_size != 0);
941fa9e4066Sahrens 
942fa9e4066Sahrens 	dprintf_dbuf(db, "size=%llx\n", (u_longlong_t)db->db.db_size);
943fa9e4066Sahrens 
944fa9e4066Sahrens 	if (db->db_level == 0) {
945fa9e4066Sahrens 		/*
946fa9e4066Sahrens 		 * Release the data buffer from the cache so that we
947fa9e4066Sahrens 		 * can modify it without impacting possible other users
948fa9e4066Sahrens 		 * of this cached data block.  Note that indirect blocks
949fa9e4066Sahrens 		 * and private objects are not released until the syncing
950fa9e4066Sahrens 		 * state (since they are only modified then).
951fa9e4066Sahrens 		 *
952fa9e4066Sahrens 		 * If this buffer is dirty in an old transaction group we need
953fa9e4066Sahrens 		 * to make a copy of it so that the changes we make in this
954fa9e4066Sahrens 		 * transaction group won't leak out when we sync the older txg.
955fa9e4066Sahrens 		 */
956fa9e4066Sahrens 		ASSERT(db->db_buf != NULL);
957fa9e4066Sahrens 		ASSERT(db->db.db_data != NULL);
958fa9e4066Sahrens 		ASSERT(db->db_d.db_data_old[txgoff] == NULL);
959fa9e4066Sahrens 		if (!(db->db.db_object & DMU_PRIVATE_OBJECT)) {
960fa9e4066Sahrens 			arc_release(db->db_buf, db);
961fa9e4066Sahrens 			dbuf_fix_old_data(db, tx->tx_txg);
962fa9e4066Sahrens 			ASSERT(db->db_buf != NULL);
963fa9e4066Sahrens 		}
964fa9e4066Sahrens 		db->db_d.db_data_old[txgoff] = db->db_buf;
965fa9e4066Sahrens 	}
966fa9e4066Sahrens 
967fa9e4066Sahrens 	mutex_enter(&dn->dn_mtx);
968fa9e4066Sahrens 	/*
969fa9e4066Sahrens 	 * We could have been freed_in_flight between the dbuf_noread
970fa9e4066Sahrens 	 * and dbuf_dirty.  We win, as though the dbuf_noread() had
971fa9e4066Sahrens 	 * happened after the free.
972fa9e4066Sahrens 	 */
973fa9e4066Sahrens 	if (db->db_level == 0 && db->db_blkid != DB_BONUS_BLKID) {
974fa9e4066Sahrens 		dnode_clear_range(dn, db->db_blkid, 1, tx);
975fa9e4066Sahrens 		db->db_d.db_freed_in_flight = FALSE;
976fa9e4066Sahrens 	}
977fa9e4066Sahrens 
978fa9e4066Sahrens 	db->db_dirtied = tx->tx_txg;
979fa9e4066Sahrens 	list_insert_tail(&dn->dn_dirty_dbufs[txgoff], db);
980fa9e4066Sahrens 	mutex_exit(&dn->dn_mtx);
981fa9e4066Sahrens 
982fa9e4066Sahrens 	/*
983fa9e4066Sahrens 	 * If writting this buffer will consume a new block on disk,
984fa9e4066Sahrens 	 * then update the accounting.
985fa9e4066Sahrens 	 */
986fa9e4066Sahrens 	if (db->db_blkid != DB_BONUS_BLKID) {
987fa9e4066Sahrens 		if (!dbuf_new_block(db, tx) && db->db_blkptr) {
988fa9e4066Sahrens 			/*
989fa9e4066Sahrens 			 * This is only a guess -- if the dbuf is dirty
990fa9e4066Sahrens 			 * in a previous txg, we don't know how much
991fa9e4066Sahrens 			 * space it will use on disk yet.  We should
992fa9e4066Sahrens 			 * really have the struct_rwlock to access
993fa9e4066Sahrens 			 * db_blkptr, but since this is just a guess,
994fa9e4066Sahrens 			 * it's OK if we get an odd answer.
995fa9e4066Sahrens 			 */
996fa9e4066Sahrens 			dnode_willuse_space(dn,
997fa9e4066Sahrens 			    -BP_GET_ASIZE(db->db_blkptr), tx);
998fa9e4066Sahrens 		}
999fa9e4066Sahrens 		dnode_willuse_space(dn, db->db.db_size, tx);
1000fa9e4066Sahrens 	}
1001fa9e4066Sahrens 
1002fa9e4066Sahrens 	/*
1003fa9e4066Sahrens 	 * This buffer is now part of this txg
1004fa9e4066Sahrens 	 */
1005fa9e4066Sahrens 	dbuf_add_ref(db, (void *)(uintptr_t)tx->tx_txg);
1006fa9e4066Sahrens 	db->db_dirtycnt += 1;
1007fa9e4066Sahrens 	ASSERT3U(db->db_dirtycnt, <=, 3);
1008fa9e4066Sahrens 
1009fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
1010fa9e4066Sahrens 
1011fa9e4066Sahrens 	if (db->db_blkid == DB_BONUS_BLKID) {
1012fa9e4066Sahrens 		dnode_setdirty(dn, tx);
1013fa9e4066Sahrens 		return;
1014fa9e4066Sahrens 	}
1015fa9e4066Sahrens 
1016fa9e4066Sahrens 	if (db->db_level == 0)
1017fa9e4066Sahrens 		dnode_new_blkid(dn, db->db_blkid, tx);
1018fa9e4066Sahrens 
1019fa9e4066Sahrens 	if (!RW_WRITE_HELD(&dn->dn_struct_rwlock)) {
1020fa9e4066Sahrens 		rw_enter(&dn->dn_struct_rwlock, RW_READER);
1021fa9e4066Sahrens 		drop_struct_lock = TRUE;
1022fa9e4066Sahrens 	}
1023fa9e4066Sahrens 
1024fa9e4066Sahrens 	if (db->db_level < dn->dn_nlevels-1) {
1025fa9e4066Sahrens 		int epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
1026fa9e4066Sahrens 		dmu_buf_impl_t *parent;
1027fa9e4066Sahrens 		parent = dbuf_hold_level(dn, db->db_level+1,
1028fa9e4066Sahrens 		    db->db_blkid >> epbs, FTAG);
1029fa9e4066Sahrens 		if (drop_struct_lock)
1030fa9e4066Sahrens 			rw_exit(&dn->dn_struct_rwlock);
1031fa9e4066Sahrens 		dbuf_dirty(parent, tx);
1032fa9e4066Sahrens 		dbuf_remove_ref(parent, FTAG);
1033fa9e4066Sahrens 	} else {
1034fa9e4066Sahrens 		if (drop_struct_lock)
1035fa9e4066Sahrens 			rw_exit(&dn->dn_struct_rwlock);
1036fa9e4066Sahrens 	}
1037fa9e4066Sahrens 
1038fa9e4066Sahrens 	dnode_setdirty(dn, tx);
1039fa9e4066Sahrens }
1040fa9e4066Sahrens 
1041fa9e4066Sahrens static int
1042fa9e4066Sahrens dbuf_undirty(dmu_buf_impl_t *db, dmu_tx_t *tx)
1043fa9e4066Sahrens {
1044fa9e4066Sahrens 	dnode_t *dn = db->db_dnode;
1045fa9e4066Sahrens 	int txgoff = tx->tx_txg & TXG_MASK;
1046fa9e4066Sahrens 
1047fa9e4066Sahrens 	ASSERT(tx->tx_txg != 0);
1048fa9e4066Sahrens 
1049fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
1050fa9e4066Sahrens 
1051fa9e4066Sahrens 	/*
1052fa9e4066Sahrens 	 * If this buffer is not dirty, we're done.
1053fa9e4066Sahrens 	 */
1054fa9e4066Sahrens 	if (!list_link_active(&db->db_dirty_node[txgoff])) {
1055fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1056fa9e4066Sahrens 		return (0);
1057fa9e4066Sahrens 	}
1058fa9e4066Sahrens 
1059fa9e4066Sahrens 	/*
1060fa9e4066Sahrens 	 * If this buffer is currently held, we cannot undirty
1061fa9e4066Sahrens 	 * it, since one of the current holders may be in the
1062fa9e4066Sahrens 	 * middle of an update.  Note that users of dbuf_undirty()
1063fa9e4066Sahrens 	 * should not place a hold on the dbuf before the call.
1064fa9e4066Sahrens 	 * XXX - this check assumes we are being called from
1065fa9e4066Sahrens 	 * dbuf_free_range(), perhaps we should move it there?
1066fa9e4066Sahrens 	 */
1067fa9e4066Sahrens 	if (refcount_count(&db->db_holds) > db->db_dirtycnt) {
1068fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1069fa9e4066Sahrens 		mutex_enter(&dn->dn_mtx);
1070fa9e4066Sahrens 		dnode_clear_range(dn, db->db_blkid, 1, tx);
1071fa9e4066Sahrens 		mutex_exit(&dn->dn_mtx);
1072fa9e4066Sahrens 		return (0);
1073fa9e4066Sahrens 	}
1074fa9e4066Sahrens 
1075fa9e4066Sahrens 	dprintf_dbuf(db, "size=%llx\n", (u_longlong_t)db->db.db_size);
1076fa9e4066Sahrens 
1077fa9e4066Sahrens 	dbuf_unoverride(db, tx->tx_txg);
1078fa9e4066Sahrens 
1079fa9e4066Sahrens 	ASSERT(db->db.db_size != 0);
1080fa9e4066Sahrens 	if (db->db_level == 0) {
1081fa9e4066Sahrens 		ASSERT(db->db_buf != NULL);
1082fa9e4066Sahrens 		ASSERT(db->db_d.db_data_old[txgoff] != NULL);
1083fa9e4066Sahrens 		if (db->db_d.db_data_old[txgoff] != db->db_buf)
1084fa9e4066Sahrens 			arc_buf_free(db->db_d.db_data_old[txgoff], db);
1085fa9e4066Sahrens 		db->db_d.db_data_old[txgoff] = NULL;
1086fa9e4066Sahrens 	}
1087fa9e4066Sahrens 
1088fa9e4066Sahrens 	/* XXX would be nice to fix up dn_towrite_space[] */
1089fa9e4066Sahrens 	/* XXX undo db_dirtied? but how? */
1090fa9e4066Sahrens 	/* db->db_dirtied = tx->tx_txg; */
1091fa9e4066Sahrens 
1092fa9e4066Sahrens 	mutex_enter(&dn->dn_mtx);
1093fa9e4066Sahrens 	list_remove(&dn->dn_dirty_dbufs[txgoff], db);
1094fa9e4066Sahrens 	mutex_exit(&dn->dn_mtx);
1095fa9e4066Sahrens 
1096fa9e4066Sahrens 	ASSERT(db->db_dirtycnt > 0);
1097fa9e4066Sahrens 	db->db_dirtycnt -= 1;
1098fa9e4066Sahrens 
1099fa9e4066Sahrens 	if (refcount_remove(&db->db_holds,
1100fa9e4066Sahrens 	    (void *)(uintptr_t)tx->tx_txg) == 0) {
1101fa9e4066Sahrens 		/* make duf_verify() happy */
1102fa9e4066Sahrens 		if (db->db.db_data)
1103fa9e4066Sahrens 			bzero(db->db.db_data, db->db.db_size);
1104fa9e4066Sahrens 
1105fa9e4066Sahrens 		dbuf_evict(db);
1106fa9e4066Sahrens 		return (1);
1107fa9e4066Sahrens 	}
1108fa9e4066Sahrens 
1109fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
1110fa9e4066Sahrens 	return (0);
1111fa9e4066Sahrens }
1112fa9e4066Sahrens 
1113fa9e4066Sahrens #pragma weak dmu_buf_will_dirty = dbuf_will_dirty
1114fa9e4066Sahrens void
1115fa9e4066Sahrens dbuf_will_dirty(dmu_buf_impl_t *db, dmu_tx_t *tx)
1116fa9e4066Sahrens {
1117fa9e4066Sahrens 	int rf = DB_RF_MUST_SUCCEED;
1118fa9e4066Sahrens 
1119fa9e4066Sahrens 	ASSERT(tx->tx_txg != 0);
1120fa9e4066Sahrens 	ASSERT(!refcount_is_zero(&db->db_holds));
1121fa9e4066Sahrens 
1122fa9e4066Sahrens 	if (RW_WRITE_HELD(&db->db_dnode->dn_struct_rwlock))
1123fa9e4066Sahrens 		rf |= DB_RF_HAVESTRUCT;
1124fa9e4066Sahrens 	(void) dbuf_read_generic(db, rf);
1125fa9e4066Sahrens 	dbuf_dirty(db, tx);
1126fa9e4066Sahrens }
1127fa9e4066Sahrens 
1128fa9e4066Sahrens #pragma weak dmu_buf_will_fill = dbuf_will_fill
1129fa9e4066Sahrens void
1130fa9e4066Sahrens dbuf_will_fill(dmu_buf_impl_t *db, dmu_tx_t *tx)
1131fa9e4066Sahrens {
1132fa9e4066Sahrens 	ASSERT(tx->tx_txg != 0);
1133fa9e4066Sahrens 	ASSERT(db->db_level == 0);
1134fa9e4066Sahrens 	ASSERT(!refcount_is_zero(&db->db_holds));
1135fa9e4066Sahrens 
1136fa9e4066Sahrens 	ASSERT(!(db->db.db_object & DMU_PRIVATE_OBJECT) ||
1137fa9e4066Sahrens 	    dmu_tx_private_ok(tx));
1138fa9e4066Sahrens 
1139fa9e4066Sahrens 	dbuf_noread(db);
1140fa9e4066Sahrens 	dbuf_dirty(db, tx);
1141fa9e4066Sahrens }
1142fa9e4066Sahrens 
1143fa9e4066Sahrens #pragma weak dmu_buf_fill_done = dbuf_fill_done
1144fa9e4066Sahrens /* ARGSUSED */
1145fa9e4066Sahrens void
1146fa9e4066Sahrens dbuf_fill_done(dmu_buf_impl_t *db, dmu_tx_t *tx)
1147fa9e4066Sahrens {
1148fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
1149*9c9dc39aSek 	DBUF_VERIFY(db);
1150fa9e4066Sahrens 
1151fa9e4066Sahrens 	if (db->db_state == DB_FILL) {
1152fa9e4066Sahrens 		if (db->db_level == 0 && db->db_d.db_freed_in_flight) {
1153fa9e4066Sahrens 			/* we were freed while filling */
1154fa9e4066Sahrens 			/* XXX dbuf_undirty? */
1155fa9e4066Sahrens 			bzero(db->db.db_data, db->db.db_size);
1156fa9e4066Sahrens 			db->db_d.db_freed_in_flight = FALSE;
1157fa9e4066Sahrens 		}
1158fa9e4066Sahrens 		db->db_state = DB_CACHED;
1159fa9e4066Sahrens 		cv_broadcast(&db->db_changed);
1160fa9e4066Sahrens 	}
1161fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
1162fa9e4066Sahrens }
1163fa9e4066Sahrens 
1164fa9e4066Sahrens 
1165fa9e4066Sahrens static void
1166fa9e4066Sahrens dbuf_clear(dmu_buf_impl_t *db)
1167fa9e4066Sahrens {
1168fa9e4066Sahrens 	dnode_t *dn = db->db_dnode;
1169fa9e4066Sahrens 
1170fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&dn->dn_dbufs_mtx));
1171fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
1172fa9e4066Sahrens 	ASSERT(refcount_is_zero(&db->db_holds));
1173fa9e4066Sahrens 
1174fa9e4066Sahrens 	if (db->db_state == DB_CACHED) {
1175fa9e4066Sahrens 		ASSERT(db->db_buf != NULL);
1176fa9e4066Sahrens 		arc_buf_free(db->db_buf, db);
1177fa9e4066Sahrens 		db->db.db_data = NULL;
1178fa9e4066Sahrens 		db->db_buf = NULL;
1179fa9e4066Sahrens 		db->db_state = DB_UNCACHED;
1180fa9e4066Sahrens 	}
1181fa9e4066Sahrens 
1182fa9e4066Sahrens 	ASSERT3U(db->db_state, ==, DB_UNCACHED);
1183fa9e4066Sahrens 	ASSERT(db->db_buf == NULL);
1184fa9e4066Sahrens 	ASSERT(db->db_data_pending == NULL);
1185fa9e4066Sahrens 
1186fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
1187fa9e4066Sahrens 
1188fa9e4066Sahrens 	/*
1189fa9e4066Sahrens 	 * If this dbuf is referened from an indirect dbuf,
1190fa9e4066Sahrens 	 * decrement the ref count on the indirect dbuf.
1191fa9e4066Sahrens 	 */
1192fa9e4066Sahrens 	if (db->db_parent && db->db_parent != dn->dn_dbuf)
1193fa9e4066Sahrens 		dbuf_remove_ref(db->db_parent, db);
1194fa9e4066Sahrens 
1195fa9e4066Sahrens 	/* remove from dn_dbufs */
1196fa9e4066Sahrens 	list_remove(&dn->dn_dbufs, db);
1197fa9e4066Sahrens 
1198fa9e4066Sahrens 	dnode_rele(dn, db);
1199fa9e4066Sahrens 
1200fa9e4066Sahrens 	dbuf_hash_remove(db);
1201fa9e4066Sahrens 
1202fa9e4066Sahrens 	db->db_dnode = NULL;
1203fa9e4066Sahrens 	db->db_parent = NULL;
1204fa9e4066Sahrens 	db->db_blkptr = NULL;
1205fa9e4066Sahrens }
1206fa9e4066Sahrens 
1207fa9e4066Sahrens static int
1208fa9e4066Sahrens dbuf_findbp(dnode_t *dn, int level, uint64_t blkid, int fail_sparse,
1209fa9e4066Sahrens     dmu_buf_impl_t **parentp, blkptr_t **bpp)
1210fa9e4066Sahrens {
1211fa9e4066Sahrens 	int nlevels, epbs;
1212fa9e4066Sahrens 
1213fa9e4066Sahrens 	if (dn->dn_phys->dn_nlevels == 0)
1214fa9e4066Sahrens 		nlevels = 1;
1215fa9e4066Sahrens 	else
1216fa9e4066Sahrens 		nlevels = dn->dn_phys->dn_nlevels;
1217fa9e4066Sahrens 
1218fa9e4066Sahrens 	epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
1219fa9e4066Sahrens 
1220fa9e4066Sahrens 	ASSERT3U(level * epbs, <, 64);
1221fa9e4066Sahrens 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
1222fa9e4066Sahrens 	if (blkid == DB_BONUS_BLKID) {
1223fa9e4066Sahrens 		/* this is the bonus buffer */
1224fa9e4066Sahrens 		*parentp = NULL;
1225fa9e4066Sahrens 		*bpp = NULL;
1226fa9e4066Sahrens 		return (0);
1227fa9e4066Sahrens 	} else if (level >= nlevels ||
1228fa9e4066Sahrens 	    (blkid > (dn->dn_phys->dn_maxblkid >> (level * epbs)))) {
1229fa9e4066Sahrens 		/* the buffer has no parent yet */
1230fa9e4066Sahrens 		*parentp = NULL;
1231fa9e4066Sahrens 		*bpp = NULL;
1232fa9e4066Sahrens 		return (ENOENT);
1233fa9e4066Sahrens 	} else if (level < nlevels-1) {
1234fa9e4066Sahrens 		/* this block is referenced from an indirect block */
1235fa9e4066Sahrens 		int err = dbuf_hold_impl(dn, level+1,
1236fa9e4066Sahrens 		    blkid >> epbs, fail_sparse, NULL, parentp);
1237fa9e4066Sahrens 		if (err)
1238fa9e4066Sahrens 			return (err);
1239fa9e4066Sahrens 		dbuf_read_havestruct(*parentp);
1240fa9e4066Sahrens 		*bpp = ((blkptr_t *)(*parentp)->db.db_data) +
1241fa9e4066Sahrens 		    (blkid & ((1ULL << epbs) - 1));
1242fa9e4066Sahrens 		return (0);
1243fa9e4066Sahrens 	} else {
1244fa9e4066Sahrens 		/* the block is referenced from the dnode */
1245fa9e4066Sahrens 		ASSERT3U(level, ==, nlevels-1);
1246fa9e4066Sahrens 		ASSERT(dn->dn_phys->dn_nblkptr == 0 ||
1247fa9e4066Sahrens 		    blkid < dn->dn_phys->dn_nblkptr);
1248fa9e4066Sahrens 		*parentp = dn->dn_dbuf;
1249fa9e4066Sahrens 		*bpp = &dn->dn_phys->dn_blkptr[blkid];
1250fa9e4066Sahrens 		return (0);
1251fa9e4066Sahrens 	}
1252fa9e4066Sahrens }
1253fa9e4066Sahrens 
1254fa9e4066Sahrens static dmu_buf_impl_t *
1255fa9e4066Sahrens dbuf_create(dnode_t *dn, uint8_t level, uint64_t blkid,
1256fa9e4066Sahrens     dmu_buf_impl_t *parent, blkptr_t *blkptr)
1257fa9e4066Sahrens {
1258fa9e4066Sahrens 	objset_impl_t *os = dn->dn_objset;
1259fa9e4066Sahrens 	dmu_buf_impl_t *db, *odb;
1260fa9e4066Sahrens 
1261fa9e4066Sahrens 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
1262fa9e4066Sahrens 	ASSERT(dn->dn_type != DMU_OT_NONE);
1263fa9e4066Sahrens 
1264fa9e4066Sahrens 	db = kmem_cache_alloc(dbuf_cache, KM_SLEEP);
1265fa9e4066Sahrens 
1266fa9e4066Sahrens 	db->db_objset = os;
1267fa9e4066Sahrens 	db->db.db_object = dn->dn_object;
1268fa9e4066Sahrens 	db->db_level = level;
1269fa9e4066Sahrens 	db->db_blkid = blkid;
1270fa9e4066Sahrens 	db->db_state = DB_UNCACHED;
1271fa9e4066Sahrens 
1272fa9e4066Sahrens 	if (db->db_blkid == DB_BONUS_BLKID) {
1273fa9e4066Sahrens 		db->db.db_size = dn->dn_bonuslen;
1274fa9e4066Sahrens 		db->db.db_offset = DB_BONUS_BLKID;
1275fa9e4066Sahrens 	} else {
1276fa9e4066Sahrens 		int blocksize =
1277fa9e4066Sahrens 		    db->db_level ? 1<<dn->dn_indblkshift :  dn->dn_datablksz;
1278fa9e4066Sahrens 		db->db.db_size = blocksize;
1279fa9e4066Sahrens 		db->db.db_offset = db->db_blkid * blocksize;
1280fa9e4066Sahrens 	}
1281fa9e4066Sahrens 
1282fa9e4066Sahrens 	db->db_dirtied = 0;
1283fa9e4066Sahrens 	db->db_dirtycnt = 0;
1284fa9e4066Sahrens 
1285fa9e4066Sahrens 	bzero(&db->db_d, sizeof (db->db_d));
1286fa9e4066Sahrens 
1287fa9e4066Sahrens 	/*
1288fa9e4066Sahrens 	 * Hold the dn_dbufs_mtx while we get the new dbuf
1289fa9e4066Sahrens 	 * in the hash table *and* added to the dbufs list.
1290fa9e4066Sahrens 	 * This prevents a possible deadlock with someone
1291fa9e4066Sahrens 	 * trying to look up this dbuf before its added to the
1292fa9e4066Sahrens 	 * dn_dbufs list.
1293fa9e4066Sahrens 	 */
1294fa9e4066Sahrens 	mutex_enter(&dn->dn_dbufs_mtx);
1295fa9e4066Sahrens 	if ((odb = dbuf_hash_insert(db)) != NULL) {
1296fa9e4066Sahrens 		/* someone else inserted it first */
1297fa9e4066Sahrens 		kmem_cache_free(dbuf_cache, db);
1298fa9e4066Sahrens 		mutex_exit(&dn->dn_dbufs_mtx);
1299fa9e4066Sahrens 		return (odb);
1300fa9e4066Sahrens 	}
1301fa9e4066Sahrens 	list_insert_head(&dn->dn_dbufs, db);
1302fa9e4066Sahrens 	mutex_exit(&dn->dn_dbufs_mtx);
1303fa9e4066Sahrens 
1304fa9e4066Sahrens 	if (parent && parent != dn->dn_dbuf)
1305fa9e4066Sahrens 		dbuf_add_ref(parent, db);
1306fa9e4066Sahrens 
1307fa9e4066Sahrens 	(void) refcount_add(&dn->dn_holds, db);
1308fa9e4066Sahrens 
1309fa9e4066Sahrens 	db->db_dnode = dn;
1310fa9e4066Sahrens 	db->db_parent = parent;
1311fa9e4066Sahrens 	db->db_blkptr = blkptr;
1312fa9e4066Sahrens 
1313fa9e4066Sahrens 	dprintf_dbuf(db, "db=%p\n", db);
1314fa9e4066Sahrens 
1315fa9e4066Sahrens 	return (db);
1316fa9e4066Sahrens }
1317fa9e4066Sahrens 
1318fa9e4066Sahrens static int
1319fa9e4066Sahrens dbuf_evictable(dmu_buf_impl_t *db)
1320fa9e4066Sahrens {
1321fa9e4066Sahrens 	int i;
1322fa9e4066Sahrens 
1323fa9e4066Sahrens 	ASSERT(MUTEX_HELD(&db->db_mtx));
1324*9c9dc39aSek 	DBUF_VERIFY(db);
1325fa9e4066Sahrens 
1326fa9e4066Sahrens 	if (db->db_state != DB_UNCACHED && db->db_state != DB_CACHED)
1327fa9e4066Sahrens 		return (FALSE);
1328fa9e4066Sahrens 
1329fa9e4066Sahrens 	if (!refcount_is_zero(&db->db_holds))
1330fa9e4066Sahrens 		return (FALSE);
1331fa9e4066Sahrens 
1332fa9e4066Sahrens #ifdef ZFS_DEBUG
1333fa9e4066Sahrens 	for (i = 0; i < TXG_SIZE; i++) {
1334fa9e4066Sahrens 		ASSERT(!list_link_active(&db->db_dirty_node[i]));
1335fa9e4066Sahrens 		ASSERT(db->db_level != 0 || db->db_d.db_data_old[i] == NULL);
1336fa9e4066Sahrens 	}
1337fa9e4066Sahrens #endif
1338fa9e4066Sahrens 
1339fa9e4066Sahrens 	/*
1340fa9e4066Sahrens 	 * Now we know we want to free it.
1341fa9e4066Sahrens 	 * This call must be done last, since it has side effects -
1342fa9e4066Sahrens 	 * calling the db_evict_func().
1343fa9e4066Sahrens 	 */
1344fa9e4066Sahrens 	dbuf_evict_user(db);
1345fa9e4066Sahrens 	return (TRUE);
1346fa9e4066Sahrens }
1347fa9e4066Sahrens 
1348fa9e4066Sahrens static void
1349fa9e4066Sahrens dbuf_destroy(dmu_buf_impl_t *db)
1350fa9e4066Sahrens {
1351fa9e4066Sahrens 	ASSERT(refcount_is_zero(&db->db_holds));
1352fa9e4066Sahrens 
1353fa9e4066Sahrens 	ASSERT(db->db.db_data == NULL);
1354fa9e4066Sahrens 	ASSERT(db->db_dnode == NULL);
1355fa9e4066Sahrens 	ASSERT(db->db_parent == NULL);
1356fa9e4066Sahrens 	ASSERT(db->db_hash_next == NULL);
1357fa9e4066Sahrens 	ASSERT(db->db_blkptr == NULL);
1358fa9e4066Sahrens 	ASSERT(db->db_data_pending == NULL);
1359fa9e4066Sahrens 
1360fa9e4066Sahrens 	kmem_cache_free(dbuf_cache, db);
1361fa9e4066Sahrens }
1362fa9e4066Sahrens 
1363fa9e4066Sahrens void
1364fa9e4066Sahrens dbuf_prefetch(dnode_t *dn, uint64_t blkid)
1365fa9e4066Sahrens {
1366fa9e4066Sahrens 	dmu_buf_impl_t *db, *parent = NULL;
1367fa9e4066Sahrens 	blkptr_t *bp = NULL;
1368fa9e4066Sahrens 
1369fa9e4066Sahrens 	ASSERT(blkid != DB_BONUS_BLKID);
1370fa9e4066Sahrens 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
1371fa9e4066Sahrens 
1372fa9e4066Sahrens 	if (dnode_block_freed(dn, blkid))
1373fa9e4066Sahrens 		return;
1374fa9e4066Sahrens 
1375fa9e4066Sahrens 	/* dbuf_find() returns with db_mtx held */
1376fa9e4066Sahrens 	if (db = dbuf_find(dn, 0, blkid)) {
1377fa9e4066Sahrens 		/*
1378fa9e4066Sahrens 		 * This dbuf is already in the cache.  We assume that
1379fa9e4066Sahrens 		 * it is already CACHED, or else about to be either
1380fa9e4066Sahrens 		 * read or filled.
1381fa9e4066Sahrens 		 */
1382fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1383fa9e4066Sahrens 		return;
1384fa9e4066Sahrens 	}
1385fa9e4066Sahrens 
1386fa9e4066Sahrens 	if (dbuf_findbp(dn, 0, blkid, TRUE, &parent, &bp) == 0) {
1387fa9e4066Sahrens 		if (bp && !BP_IS_HOLE(bp)) {
1388fa9e4066Sahrens 			(void) arc_read(NULL, dn->dn_objset->os_spa, bp,
1389fa9e4066Sahrens 			    dmu_ot[dn->dn_type].ot_byteswap,
1390fa9e4066Sahrens 			    NULL, NULL, ZIO_PRIORITY_ASYNC_READ,
1391fa9e4066Sahrens 			    ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE,
1392fa9e4066Sahrens 			    (ARC_NOWAIT | ARC_PREFETCH));
1393fa9e4066Sahrens 		}
1394fa9e4066Sahrens 		if (parent && parent != dn->dn_dbuf)
1395fa9e4066Sahrens 			dbuf_rele(parent);
1396fa9e4066Sahrens 	}
1397fa9e4066Sahrens }
1398fa9e4066Sahrens 
1399fa9e4066Sahrens /*
1400fa9e4066Sahrens  * Returns with db_holds incremented, and db_mtx not held.
1401fa9e4066Sahrens  * Note: dn_struct_rwlock must be held.
1402fa9e4066Sahrens  */
1403fa9e4066Sahrens int
1404fa9e4066Sahrens dbuf_hold_impl(dnode_t *dn, uint8_t level, uint64_t blkid, int fail_sparse,
1405fa9e4066Sahrens     void *tag, dmu_buf_impl_t **dbp)
1406fa9e4066Sahrens {
1407fa9e4066Sahrens 	dmu_buf_impl_t *db, *parent = NULL;
1408fa9e4066Sahrens 
1409fa9e4066Sahrens 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
1410fa9e4066Sahrens 	ASSERT3U(dn->dn_nlevels, >, level);
1411fa9e4066Sahrens 
1412fa9e4066Sahrens 	*dbp = NULL;
1413fa9e4066Sahrens 
1414fa9e4066Sahrens 	/* dbuf_find() returns with db_mtx held */
1415fa9e4066Sahrens 	db = dbuf_find(dn, level, blkid);
1416fa9e4066Sahrens 
1417fa9e4066Sahrens 	if (db == NULL) {
1418fa9e4066Sahrens 		blkptr_t *bp = NULL;
1419fa9e4066Sahrens 		int err;
1420fa9e4066Sahrens 
1421fa9e4066Sahrens 		err = dbuf_findbp(dn, level, blkid, fail_sparse, &parent, &bp);
1422fa9e4066Sahrens 		if (fail_sparse) {
1423fa9e4066Sahrens 			if (err == 0 && bp && BP_IS_HOLE(bp))
1424fa9e4066Sahrens 				err = ENOENT;
1425fa9e4066Sahrens 			if (err) {
1426fa9e4066Sahrens 				if (parent && parent != dn->dn_dbuf)
1427fa9e4066Sahrens 					dbuf_rele(parent);
1428fa9e4066Sahrens 				return (err);
1429fa9e4066Sahrens 			}
1430fa9e4066Sahrens 		}
1431fa9e4066Sahrens 		db = dbuf_create(dn, level, blkid, parent, bp);
1432fa9e4066Sahrens 	}
1433fa9e4066Sahrens 
1434fa9e4066Sahrens 	/*
1435fa9e4066Sahrens 	 * If this buffer is currently syncing out, and we are
1436fa9e4066Sahrens 	 * are still referencing it from db_data, we need to make
1437fa9e4066Sahrens 	 * a copy of it in case we decide we want to dirty it
1438fa9e4066Sahrens 	 * again in this txg.
1439fa9e4066Sahrens 	 */
1440fa9e4066Sahrens 	if (db->db_level == 0 && db->db_state == DB_CACHED &&
1441fa9e4066Sahrens 	    !(dn->dn_object & DMU_PRIVATE_OBJECT) &&
1442fa9e4066Sahrens 	    db->db_data_pending == db->db_buf) {
1443fa9e4066Sahrens 		int size = (db->db_blkid == DB_BONUS_BLKID) ?
1444fa9e4066Sahrens 		    DN_MAX_BONUSLEN : db->db.db_size;
1445fa9e4066Sahrens 
1446fa9e4066Sahrens 		dbuf_set_data(db, arc_buf_alloc(db->db_dnode->dn_objset->os_spa,
1447fa9e4066Sahrens 		    size, db));
1448fa9e4066Sahrens 		bcopy(db->db_data_pending->b_data, db->db.db_data,
1449fa9e4066Sahrens 		    db->db.db_size);
1450fa9e4066Sahrens 	}
1451fa9e4066Sahrens 
1452fa9e4066Sahrens 	dbuf_add_ref(db, tag);
1453fa9e4066Sahrens 	dbuf_update_data(db);
1454*9c9dc39aSek 	DBUF_VERIFY(db);
1455fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
1456fa9e4066Sahrens 
1457fa9e4066Sahrens 	/* NOTE: we can't rele the parent until after we drop the db_mtx */
1458fa9e4066Sahrens 	if (parent && parent != dn->dn_dbuf)
1459fa9e4066Sahrens 		dbuf_rele(parent);
1460fa9e4066Sahrens 
1461fa9e4066Sahrens 	ASSERT3P(db->db_dnode, ==, dn);
1462fa9e4066Sahrens 	ASSERT3U(db->db_blkid, ==, blkid);
1463fa9e4066Sahrens 	ASSERT3U(db->db_level, ==, level);
1464fa9e4066Sahrens 	*dbp = db;
1465fa9e4066Sahrens 
1466fa9e4066Sahrens 	return (0);
1467fa9e4066Sahrens }
1468fa9e4066Sahrens 
1469fa9e4066Sahrens dmu_buf_impl_t *
1470fa9e4066Sahrens dbuf_hold(dnode_t *dn, uint64_t blkid)
1471fa9e4066Sahrens {
1472fa9e4066Sahrens 	dmu_buf_impl_t *db;
1473fa9e4066Sahrens 	(void) dbuf_hold_impl(dn, 0, blkid, FALSE, NULL, &db);
1474fa9e4066Sahrens 	return (db);
1475fa9e4066Sahrens }
1476fa9e4066Sahrens 
1477fa9e4066Sahrens dmu_buf_impl_t *
1478fa9e4066Sahrens dbuf_hold_level(dnode_t *dn, int level, uint64_t blkid, void *tag)
1479fa9e4066Sahrens {
1480fa9e4066Sahrens 	dmu_buf_impl_t *db;
1481fa9e4066Sahrens 	(void) dbuf_hold_impl(dn, level, blkid, FALSE, tag, &db);
1482fa9e4066Sahrens 	return (db);
1483fa9e4066Sahrens }
1484fa9e4066Sahrens 
1485fa9e4066Sahrens dmu_buf_impl_t *
1486fa9e4066Sahrens dbuf_hold_bonus(dnode_t *dn, void *tag)
1487fa9e4066Sahrens {
1488fa9e4066Sahrens 	dmu_buf_impl_t *db;
1489fa9e4066Sahrens 	rw_enter(&dn->dn_struct_rwlock, RW_READER);
1490fa9e4066Sahrens 	(void) dbuf_hold_impl(dn, 0, DB_BONUS_BLKID, FALSE, tag, &db);
1491fa9e4066Sahrens 	rw_exit(&dn->dn_struct_rwlock);
1492fa9e4066Sahrens 	return (db);
1493fa9e4066Sahrens }
1494fa9e4066Sahrens 
1495fa9e4066Sahrens void
1496fa9e4066Sahrens dbuf_add_ref(dmu_buf_impl_t *db, void *tag)
1497fa9e4066Sahrens {
1498fa9e4066Sahrens 	(void) refcount_add(&db->db_holds, tag);
1499fa9e4066Sahrens 	/* dprintf_dbuf(db, "adding ref %p; holds up to %lld\n", tag, holds); */
1500fa9e4066Sahrens }
1501fa9e4066Sahrens 
1502fa9e4066Sahrens void
1503fa9e4066Sahrens dbuf_remove_ref(dmu_buf_impl_t *db, void *tag)
1504fa9e4066Sahrens {
1505fa9e4066Sahrens 	int64_t holds;
1506fa9e4066Sahrens 	dnode_t *dn = db->db_dnode;
1507fa9e4066Sahrens 	int need_mutex;
1508fa9e4066Sahrens 
1509fa9e4066Sahrens 	ASSERT(dn != NULL);
1510fa9e4066Sahrens 	need_mutex = !MUTEX_HELD(&dn->dn_dbufs_mtx);
1511fa9e4066Sahrens 
1512fa9e4066Sahrens 	if (need_mutex) {
1513fa9e4066Sahrens 		dnode_add_ref(dn, FTAG);
1514fa9e4066Sahrens 		mutex_enter(&dn->dn_dbufs_mtx);
1515fa9e4066Sahrens 	}
1516fa9e4066Sahrens 
1517fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
1518*9c9dc39aSek 	DBUF_VERIFY(db);
1519fa9e4066Sahrens 
1520fa9e4066Sahrens 	holds = refcount_remove(&db->db_holds, tag);
1521fa9e4066Sahrens 
1522fa9e4066Sahrens 	if (holds == 0) {
1523fa9e4066Sahrens 		ASSERT3U(db->db_state, !=, DB_FILL);
1524fa9e4066Sahrens 		if (db->db_level == 0 &&
1525fa9e4066Sahrens 		    db->db_d.db_user_data_ptr_ptr != NULL)
1526fa9e4066Sahrens 			*db->db_d.db_user_data_ptr_ptr = NULL;
1527fa9e4066Sahrens 		dbuf_evict(db);
1528fa9e4066Sahrens 	} else {
1529fa9e4066Sahrens 		if (holds == db->db_dirtycnt &&
1530fa9e4066Sahrens 		    db->db_level == 0 && db->db_d.db_immediate_evict)
1531fa9e4066Sahrens 			dbuf_evict_user(db);
1532fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1533fa9e4066Sahrens 	}
1534fa9e4066Sahrens 
1535fa9e4066Sahrens 	if (need_mutex) {
1536fa9e4066Sahrens 		mutex_exit(&dn->dn_dbufs_mtx);
1537fa9e4066Sahrens 		dnode_rele(dn, FTAG);
1538fa9e4066Sahrens 	}
1539fa9e4066Sahrens }
1540fa9e4066Sahrens 
1541fa9e4066Sahrens void
1542fa9e4066Sahrens dbuf_rele(dmu_buf_impl_t *db)
1543fa9e4066Sahrens {
1544fa9e4066Sahrens 	dbuf_remove_ref(db, NULL);
1545fa9e4066Sahrens }
1546fa9e4066Sahrens 
1547fa9e4066Sahrens #pragma weak dmu_buf_refcount = dbuf_refcount
1548fa9e4066Sahrens uint64_t
1549fa9e4066Sahrens dbuf_refcount(dmu_buf_impl_t *db)
1550fa9e4066Sahrens {
1551fa9e4066Sahrens 	return (refcount_count(&db->db_holds));
1552fa9e4066Sahrens }
1553fa9e4066Sahrens 
1554fa9e4066Sahrens void *
1555fa9e4066Sahrens dmu_buf_set_user(dmu_buf_t *db_fake, void *user_ptr, void *user_data_ptr_ptr,
1556fa9e4066Sahrens     dmu_buf_evict_func_t *evict_func)
1557fa9e4066Sahrens {
1558fa9e4066Sahrens 	return (dmu_buf_update_user(db_fake, NULL, user_ptr,
1559fa9e4066Sahrens 	    user_data_ptr_ptr, evict_func));
1560fa9e4066Sahrens }
1561fa9e4066Sahrens 
1562fa9e4066Sahrens void *
1563fa9e4066Sahrens dmu_buf_set_user_ie(dmu_buf_t *db_fake, void *user_ptr, void *user_data_ptr_ptr,
1564fa9e4066Sahrens     dmu_buf_evict_func_t *evict_func)
1565fa9e4066Sahrens {
1566fa9e4066Sahrens 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
1567fa9e4066Sahrens 
1568fa9e4066Sahrens 	db->db_d.db_immediate_evict = TRUE;
1569fa9e4066Sahrens 	return (dmu_buf_update_user(db_fake, NULL, user_ptr,
1570fa9e4066Sahrens 	    user_data_ptr_ptr, evict_func));
1571fa9e4066Sahrens }
1572fa9e4066Sahrens 
1573fa9e4066Sahrens void *
1574fa9e4066Sahrens dmu_buf_update_user(dmu_buf_t *db_fake, void *old_user_ptr, void *user_ptr,
1575fa9e4066Sahrens     void *user_data_ptr_ptr, dmu_buf_evict_func_t *evict_func)
1576fa9e4066Sahrens {
1577fa9e4066Sahrens 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
1578fa9e4066Sahrens 	ASSERT(db->db_level == 0);
1579fa9e4066Sahrens 
1580fa9e4066Sahrens 	ASSERT((user_ptr == NULL) == (evict_func == NULL));
1581fa9e4066Sahrens 
1582fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
1583fa9e4066Sahrens 
1584fa9e4066Sahrens 	if (db->db_d.db_user_ptr == old_user_ptr) {
1585fa9e4066Sahrens 		db->db_d.db_user_ptr = user_ptr;
1586fa9e4066Sahrens 		db->db_d.db_user_data_ptr_ptr = user_data_ptr_ptr;
1587fa9e4066Sahrens 		db->db_d.db_evict_func = evict_func;
1588fa9e4066Sahrens 
1589fa9e4066Sahrens 		dbuf_update_data(db);
1590fa9e4066Sahrens 	} else {
1591fa9e4066Sahrens 		old_user_ptr = db->db_d.db_user_ptr;
1592fa9e4066Sahrens 	}
1593fa9e4066Sahrens 
1594fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
1595fa9e4066Sahrens 	return (old_user_ptr);
1596fa9e4066Sahrens }
1597fa9e4066Sahrens 
1598fa9e4066Sahrens void *
1599fa9e4066Sahrens dmu_buf_get_user(dmu_buf_t *db_fake)
1600fa9e4066Sahrens {
1601fa9e4066Sahrens 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
1602fa9e4066Sahrens 	ASSERT(!refcount_is_zero(&db->db_holds));
1603fa9e4066Sahrens 
1604fa9e4066Sahrens 	return (db->db_d.db_user_ptr);
1605fa9e4066Sahrens }
1606fa9e4066Sahrens 
1607fa9e4066Sahrens void
1608fa9e4066Sahrens dbuf_sync(dmu_buf_impl_t *db, zio_t *zio, dmu_tx_t *tx)
1609fa9e4066Sahrens {
1610fa9e4066Sahrens 	arc_buf_t **data;
1611fa9e4066Sahrens 	uint64_t txg = tx->tx_txg;
1612fa9e4066Sahrens 	dnode_t *dn = db->db_dnode;
1613fa9e4066Sahrens 	objset_impl_t *os = dn->dn_objset;
1614fa9e4066Sahrens 	int blksz;
1615fa9e4066Sahrens 
1616fa9e4066Sahrens 	ASSERT(dmu_tx_is_syncing(tx));
1617fa9e4066Sahrens 
1618fa9e4066Sahrens 	dprintf_dbuf_bp(db, db->db_blkptr, "blkptr=%p", db->db_blkptr);
1619fa9e4066Sahrens 
1620fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
1621fa9e4066Sahrens 	/*
1622fa9e4066Sahrens 	 * To be synced, we must be dirtied.  But we
1623fa9e4066Sahrens 	 * might have been freed after the dirty.
1624fa9e4066Sahrens 	 */
1625fa9e4066Sahrens 	if (db->db_state == DB_UNCACHED) {
1626fa9e4066Sahrens 		/* This buffer has been freed since it was dirtied */
1627fa9e4066Sahrens 		ASSERT(db->db.db_data == NULL);
1628fa9e4066Sahrens 	} else if (db->db_state == DB_FILL) {
1629fa9e4066Sahrens 		/* This buffer was freed and is now being re-filled */
1630fa9e4066Sahrens 		ASSERT(db->db.db_data != db->db_d.db_data_old[txg&TXG_MASK]);
1631fa9e4066Sahrens 	} else {
1632fa9e4066Sahrens 		ASSERT3U(db->db_state, ==, DB_CACHED);
1633fa9e4066Sahrens 	}
1634*9c9dc39aSek 	DBUF_VERIFY(db);
1635fa9e4066Sahrens 
1636fa9e4066Sahrens 	/*
1637fa9e4066Sahrens 	 * Don't need a lock on db_dirty (dn_mtx), because it can't
1638fa9e4066Sahrens 	 * be modified yet.
1639fa9e4066Sahrens 	 */
1640fa9e4066Sahrens 
1641fa9e4066Sahrens 	if (db->db_level == 0) {
1642fa9e4066Sahrens 		data = &db->db_d.db_data_old[txg&TXG_MASK];
1643fa9e4066Sahrens 		blksz = arc_buf_size(*data);
1644fa9e4066Sahrens 		/*
1645fa9e4066Sahrens 		 * If this buffer is currently "in use" (i.e., there are
1646fa9e4066Sahrens 		 * active holds and db_data still references it), then make
1647fa9e4066Sahrens 		 * a copy before we start the write so that any modifications
1648fa9e4066Sahrens 		 * from the open txg will not leak into this write.
1649fa9e4066Sahrens 		 *
1650fa9e4066Sahrens 		 * NOTE: this copy does not need to be made for objects only
1651fa9e4066Sahrens 		 * modified in the syncing context (e.g. DNONE_DNODE blocks)
1652fa9e4066Sahrens 		 * or if there is no actual write involved (bonus blocks).
1653fa9e4066Sahrens 		 */
1654fa9e4066Sahrens 		if (!(dn->dn_object & DMU_PRIVATE_OBJECT) &&
1655fa9e4066Sahrens 		    db->db_d.db_overridden_by[txg&TXG_MASK] == NULL &&
1656fa9e4066Sahrens 		    db->db_blkid != DB_BONUS_BLKID) {
1657fa9e4066Sahrens 			if (refcount_count(&db->db_holds) > 1 &&
1658fa9e4066Sahrens 			    *data == db->db_buf) {
1659fa9e4066Sahrens 				*data = arc_buf_alloc(
1660fa9e4066Sahrens 				    db->db_dnode->dn_objset->os_spa, blksz, db);
1661fa9e4066Sahrens 				bcopy(db->db.db_data, (*data)->b_data, blksz);
1662fa9e4066Sahrens 			}
1663fa9e4066Sahrens 			db->db_data_pending = *data;
1664fa9e4066Sahrens 		} else if (dn->dn_object & DMU_PRIVATE_OBJECT) {
1665fa9e4066Sahrens 			/*
1666fa9e4066Sahrens 			 * Private object buffers are released here rather
1667fa9e4066Sahrens 			 * than in dbuf_dirty() since they are only modified
1668fa9e4066Sahrens 			 * in the syncing context and we don't want the
1669fa9e4066Sahrens 			 * overhead of making multiple copies of the data.
1670fa9e4066Sahrens 			 */
1671fa9e4066Sahrens 			arc_release(db->db_buf, db);
1672fa9e4066Sahrens 		}
1673fa9e4066Sahrens 	} else {
1674fa9e4066Sahrens 		data = &db->db_buf;
1675fa9e4066Sahrens 		if (*data == NULL) {
1676fa9e4066Sahrens 			/*
1677fa9e4066Sahrens 			 * This can happen if we dirty and then free
1678fa9e4066Sahrens 			 * the level-0 data blocks in the same txg. So
1679fa9e4066Sahrens 			 * this indirect remains unchanged.
1680fa9e4066Sahrens 			 */
1681fa9e4066Sahrens 			if (db->db_dirtied == txg)
1682fa9e4066Sahrens 				db->db_dirtied = 0;
1683fa9e4066Sahrens 			ASSERT(db->db_dirtycnt > 0);
1684fa9e4066Sahrens 			db->db_dirtycnt -= 1;
1685fa9e4066Sahrens 			mutex_exit(&db->db_mtx);
1686fa9e4066Sahrens 			dbuf_remove_ref(db, (void *)(uintptr_t)txg);
1687fa9e4066Sahrens 			return;
1688fa9e4066Sahrens 		}
1689fa9e4066Sahrens 		blksz = db->db.db_size;
1690fa9e4066Sahrens 		ASSERT3U(blksz, ==, 1<<dn->dn_phys->dn_indblkshift);
1691fa9e4066Sahrens 	}
1692fa9e4066Sahrens 
1693fa9e4066Sahrens 	ASSERT(*data != NULL);
1694fa9e4066Sahrens 
1695fa9e4066Sahrens 	if (db->db_blkid == DB_BONUS_BLKID) {
1696fa9e4066Sahrens 		/*
1697fa9e4066Sahrens 		 * Simply copy the bonus data into the dnode.  It will
1698fa9e4066Sahrens 		 * be written out when the dnode is synced (and it will
1699fa9e4066Sahrens 		 * be synced, since it must have been dirty for dbuf_sync
1700fa9e4066Sahrens 		 * to be called).  The bonus data will be byte swapped
1701fa9e4066Sahrens 		 * in dnode_byteswap.
1702fa9e4066Sahrens 		 */
1703fa9e4066Sahrens 		/*
1704fa9e4066Sahrens 		 * Use dn_phys->dn_bonuslen since db.db_size is the length
1705fa9e4066Sahrens 		 * of the bonus buffer in the open transaction rather than
1706fa9e4066Sahrens 		 * the syncing transaction.
1707fa9e4066Sahrens 		 */
1708fa9e4066Sahrens 		ASSERT3U(db->db_level, ==, 0);
1709fa9e4066Sahrens 		ASSERT3U(dn->dn_phys->dn_bonuslen, <=, blksz);
1710fa9e4066Sahrens 		bcopy((*data)->b_data, DN_BONUS(dn->dn_phys),
1711fa9e4066Sahrens 		    dn->dn_phys->dn_bonuslen);
1712fa9e4066Sahrens 		if (*data != db->db_buf)
1713fa9e4066Sahrens 			arc_buf_free(*data, db);
1714fa9e4066Sahrens 		db->db_d.db_data_old[txg&TXG_MASK] = NULL;
1715fa9e4066Sahrens 		db->db_data_pending = NULL;
1716fa9e4066Sahrens 		if (db->db_dirtied == txg)
1717fa9e4066Sahrens 			db->db_dirtied = 0;
1718fa9e4066Sahrens 		ASSERT(db->db_dirtycnt > 0);
1719fa9e4066Sahrens 		db->db_dirtycnt -= 1;
1720fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1721fa9e4066Sahrens 		dbuf_remove_ref(db, (void *)(uintptr_t)txg);
1722fa9e4066Sahrens 		return;
1723fa9e4066Sahrens 	} else if (db->db_level > 0 && !arc_released(db->db_buf)) {
1724fa9e4066Sahrens 		/*
1725fa9e4066Sahrens 		 * This indirect buffer was marked dirty, but
1726fa9e4066Sahrens 		 * never modified (if it had been modified, then
1727fa9e4066Sahrens 		 * we would have released the buffer).  There is
1728fa9e4066Sahrens 		 * no reason to write anything.
1729fa9e4066Sahrens 		 */
1730fa9e4066Sahrens 		db->db_data_pending = NULL;
1731fa9e4066Sahrens 		if (db->db_dirtied == txg)
1732fa9e4066Sahrens 			db->db_dirtied = 0;
1733fa9e4066Sahrens 		ASSERT(db->db_dirtycnt > 0);
1734fa9e4066Sahrens 		db->db_dirtycnt -= 1;
1735fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1736fa9e4066Sahrens 		dbuf_remove_ref(db, (void *)(uintptr_t)txg);
1737fa9e4066Sahrens 		return;
1738fa9e4066Sahrens 	} else if (db->db_blkptr == NULL &&
1739fa9e4066Sahrens 	    db->db_level == dn->dn_phys->dn_nlevels-1 &&
1740fa9e4066Sahrens 	    db->db_blkid < dn->dn_phys->dn_nblkptr) {
1741fa9e4066Sahrens 		/*
1742fa9e4066Sahrens 		 * This buffer was allocated at a time when there was
1743fa9e4066Sahrens 		 * no available blkptrs from the dnode, or it was
1744fa9e4066Sahrens 		 * inappropriate to hook it in (i.e., nlevels mis-match).
1745fa9e4066Sahrens 		 */
1746fa9e4066Sahrens 		ASSERT(db->db_blkptr == NULL);
1747fa9e4066Sahrens 		ASSERT(db->db_parent == NULL);
1748fa9e4066Sahrens 		db->db_parent = dn->dn_dbuf;
1749fa9e4066Sahrens 		db->db_blkptr = &dn->dn_phys->dn_blkptr[db->db_blkid];
1750*9c9dc39aSek 		DBUF_VERIFY(db);
1751fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1752fa9e4066Sahrens 	} else if (db->db_blkptr == NULL) {
1753fa9e4066Sahrens 		dmu_buf_impl_t *parent = db->db_parent;
1754fa9e4066Sahrens 		int epbs = dn->dn_phys->dn_indblkshift - SPA_BLKPTRSHIFT;
1755fa9e4066Sahrens 
1756fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1757fa9e4066Sahrens 		ASSERT(dn->dn_phys->dn_nlevels > 1);
1758fa9e4066Sahrens 		if (parent == NULL) {
1759fa9e4066Sahrens 			rw_enter(&dn->dn_struct_rwlock, RW_READER);
1760fa9e4066Sahrens 			(void) dbuf_hold_impl(dn, db->db_level+1,
1761fa9e4066Sahrens 			    db->db_blkid >> epbs, FALSE, NULL, &parent);
1762fa9e4066Sahrens 			rw_exit(&dn->dn_struct_rwlock);
1763fa9e4066Sahrens 			dbuf_add_ref(parent, db);
1764fa9e4066Sahrens 			db->db_parent = parent;
1765fa9e4066Sahrens 			dbuf_rele(parent);
1766fa9e4066Sahrens 		}
1767fa9e4066Sahrens 		dbuf_read(parent);
1768fa9e4066Sahrens 	} else {
1769fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1770fa9e4066Sahrens 	}
1771fa9e4066Sahrens 
1772fa9e4066Sahrens 	ASSERT(IS_DNODE_DNODE(dn->dn_object) || db->db_parent != NULL);
1773fa9e4066Sahrens 
1774fa9e4066Sahrens 	if (db->db_parent != dn->dn_dbuf) {
1775fa9e4066Sahrens 		dmu_buf_impl_t *parent = db->db_parent;
1776fa9e4066Sahrens 		int epbs = dn->dn_phys->dn_indblkshift - SPA_BLKPTRSHIFT;
1777fa9e4066Sahrens 
1778fa9e4066Sahrens 		mutex_enter(&db->db_mtx);
1779fa9e4066Sahrens 		ASSERT(db->db_level == parent->db_level-1);
1780fa9e4066Sahrens 		ASSERT(list_link_active(&parent->db_dirty_node[txg&TXG_MASK]));
1781fa9e4066Sahrens 		/*
1782fa9e4066Sahrens 		 * We may have read this block after we dirtied it,
1783fa9e4066Sahrens 		 * so never released it from the cache.
1784fa9e4066Sahrens 		 */
1785fa9e4066Sahrens 		arc_release(parent->db_buf, parent);
1786fa9e4066Sahrens 
1787fa9e4066Sahrens 		db->db_blkptr = (blkptr_t *)parent->db.db_data +
1788fa9e4066Sahrens 		    (db->db_blkid & ((1ULL << epbs) - 1));
1789*9c9dc39aSek 		DBUF_VERIFY(db);
1790fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1791fa9e4066Sahrens 	}
1792fa9e4066Sahrens 	ASSERT(db->db_parent == NULL || arc_released(db->db_parent->db_buf));
1793fa9e4066Sahrens 
1794fa9e4066Sahrens #ifdef ZFS_DEBUG
1795fa9e4066Sahrens 	if (db->db_parent == dn->dn_dbuf) {
1796fa9e4066Sahrens 		/*
1797fa9e4066Sahrens 		 * We don't need to dnode_setdirty(dn) because if we got
1798fa9e4066Sahrens 		 * here then the parent is already dirty.
1799fa9e4066Sahrens 		 */
1800fa9e4066Sahrens 		ASSERT(db->db_level == dn->dn_phys->dn_nlevels-1);
1801fa9e4066Sahrens 		ASSERT3P(db->db_blkptr, ==,
1802fa9e4066Sahrens 		    &dn->dn_phys->dn_blkptr[db->db_blkid]);
1803fa9e4066Sahrens 	}
1804fa9e4066Sahrens #endif
1805fa9e4066Sahrens 	if (db->db_level == 0 &&
1806fa9e4066Sahrens 	    db->db_d.db_overridden_by[txg&TXG_MASK] != NULL) {
1807fa9e4066Sahrens 		arc_buf_t **old = &db->db_d.db_data_old[txg&TXG_MASK];
1808fa9e4066Sahrens 		blkptr_t **bpp = &db->db_d.db_overridden_by[txg&TXG_MASK];
1809fa9e4066Sahrens 		int old_size = BP_GET_ASIZE(db->db_blkptr);
1810fa9e4066Sahrens 		int new_size = BP_GET_ASIZE(*bpp);
1811fa9e4066Sahrens 
1812fa9e4066Sahrens 		ASSERT(db->db_blkid != DB_BONUS_BLKID);
1813fa9e4066Sahrens 
1814fa9e4066Sahrens 		dnode_diduse_space(dn, new_size-old_size);
1815fa9e4066Sahrens 		mutex_enter(&dn->dn_mtx);
1816fa9e4066Sahrens 		if (db->db_blkid > dn->dn_phys->dn_maxblkid)
1817fa9e4066Sahrens 			dn->dn_phys->dn_maxblkid = db->db_blkid;
1818fa9e4066Sahrens 		mutex_exit(&dn->dn_mtx);
1819fa9e4066Sahrens 
1820fa9e4066Sahrens 		dsl_dataset_block_born(os->os_dsl_dataset, *bpp, tx);
1821fa9e4066Sahrens 		if (!BP_IS_HOLE(db->db_blkptr))
1822fa9e4066Sahrens 			dsl_dataset_block_kill(os->os_dsl_dataset,
1823fa9e4066Sahrens 			    db->db_blkptr, os->os_synctx);
1824fa9e4066Sahrens 
1825fa9e4066Sahrens 		mutex_enter(&db->db_mtx);
1826fa9e4066Sahrens 		*db->db_blkptr = **bpp;
1827fa9e4066Sahrens 		kmem_free(*bpp, sizeof (blkptr_t));
1828fa9e4066Sahrens 		*bpp = NULL;
1829fa9e4066Sahrens 
1830fa9e4066Sahrens 		if (*old != db->db_buf)
1831fa9e4066Sahrens 			arc_buf_free(*old, db);
1832fa9e4066Sahrens 		*old = NULL;
1833fa9e4066Sahrens 		db->db_data_pending = NULL;
1834fa9e4066Sahrens 
1835fa9e4066Sahrens 		cv_broadcast(&db->db_changed);
1836fa9e4066Sahrens 
1837fa9e4066Sahrens 		ASSERT(db->db_dirtycnt > 0);
1838fa9e4066Sahrens 		db->db_dirtycnt -= 1;
1839fa9e4066Sahrens 		mutex_exit(&db->db_mtx);
1840fa9e4066Sahrens 		dbuf_remove_ref(db, (void *)(uintptr_t)txg);
1841fa9e4066Sahrens 	} else {
1842fa9e4066Sahrens 		int checksum, compress;
1843fa9e4066Sahrens 
1844fa9e4066Sahrens 		if (db->db_level > 0) {
1845fa9e4066Sahrens 			/*
1846fa9e4066Sahrens 			 * XXX -- we should design a compression algorithm
1847fa9e4066Sahrens 			 * that specializes in arrays of bps.
1848fa9e4066Sahrens 			 */
1849fa9e4066Sahrens 			checksum = ZIO_CHECKSUM_FLETCHER_4;
1850fa9e4066Sahrens 			compress = ZIO_COMPRESS_LZJB;
1851fa9e4066Sahrens 		} else {
1852fa9e4066Sahrens 			/*
1853fa9e4066Sahrens 			 * Allow dnode settings to override objset settings,
1854fa9e4066Sahrens 			 * except for metadata checksums.
1855fa9e4066Sahrens 			 */
1856fa9e4066Sahrens 			if (dmu_ot[dn->dn_type].ot_metadata) {
1857fa9e4066Sahrens 				checksum = os->os_md_checksum;
1858fa9e4066Sahrens 				compress = zio_compress_select(dn->dn_compress,
1859fa9e4066Sahrens 				    os->os_md_compress);
1860fa9e4066Sahrens 			} else {
1861fa9e4066Sahrens 				checksum = zio_checksum_select(dn->dn_checksum,
1862fa9e4066Sahrens 				    os->os_checksum);
1863fa9e4066Sahrens 				compress = zio_compress_select(dn->dn_compress,
1864fa9e4066Sahrens 				    os->os_compress);
1865fa9e4066Sahrens 			}
1866fa9e4066Sahrens 		}
1867fa9e4066Sahrens #ifdef ZFS_DEBUG
1868fa9e4066Sahrens 		if (db->db_parent) {
1869fa9e4066Sahrens 			ASSERT(list_link_active(
1870fa9e4066Sahrens 			    &db->db_parent->db_dirty_node[txg&TXG_MASK]));
1871fa9e4066Sahrens 			ASSERT(db->db_parent == dn->dn_dbuf ||
1872fa9e4066Sahrens 			    db->db_parent->db_level > 0);
1873fa9e4066Sahrens 			if (dn->dn_object & DMU_PRIVATE_OBJECT ||
1874fa9e4066Sahrens 			    db->db_level > 0)
1875fa9e4066Sahrens 				ASSERT(*data == db->db_buf);
1876fa9e4066Sahrens 		}
1877fa9e4066Sahrens #endif
1878fa9e4066Sahrens 		ASSERT3U(db->db_blkptr->blk_birth, <=, tx->tx_txg);
1879fa9e4066Sahrens 		(void) arc_write(zio, os->os_spa, checksum, compress, txg,
1880fa9e4066Sahrens 		    db->db_blkptr, *data, dbuf_write_done, db,
1881fa9e4066Sahrens 		    ZIO_PRIORITY_ASYNC_WRITE, ZIO_FLAG_MUSTSUCCEED, ARC_NOWAIT);
1882fa9e4066Sahrens 		/*
1883fa9e4066Sahrens 		 * We can't access db after arc_write, since it could finish
1884fa9e4066Sahrens 		 * and be freed, and we have no locks on it.
1885fa9e4066Sahrens 		 */
1886fa9e4066Sahrens 	}
1887fa9e4066Sahrens }
1888fa9e4066Sahrens 
1889fa9e4066Sahrens struct dbuf_arg {
1890fa9e4066Sahrens 	objset_impl_t *os;
1891fa9e4066Sahrens 	blkptr_t bp;
1892fa9e4066Sahrens };
1893fa9e4066Sahrens 
1894fa9e4066Sahrens static void
1895fa9e4066Sahrens dbuf_do_born(void *arg)
1896fa9e4066Sahrens {
1897fa9e4066Sahrens 	struct dbuf_arg *da = arg;
1898fa9e4066Sahrens 	dsl_dataset_block_born(da->os->os_dsl_dataset,
1899fa9e4066Sahrens 	    &da->bp, da->os->os_synctx);
1900fa9e4066Sahrens 	kmem_free(da, sizeof (struct dbuf_arg));
1901fa9e4066Sahrens }
1902fa9e4066Sahrens 
1903fa9e4066Sahrens static void
1904fa9e4066Sahrens dbuf_do_kill(void *arg)
1905fa9e4066Sahrens {
1906fa9e4066Sahrens 	struct dbuf_arg *da = arg;
1907fa9e4066Sahrens 	dsl_dataset_block_kill(da->os->os_dsl_dataset,
1908fa9e4066Sahrens 	    &da->bp, da->os->os_synctx);
1909fa9e4066Sahrens 	kmem_free(da, sizeof (struct dbuf_arg));
1910fa9e4066Sahrens }
1911fa9e4066Sahrens 
1912fa9e4066Sahrens /* ARGSUSED */
1913fa9e4066Sahrens static void
1914fa9e4066Sahrens dbuf_write_done(zio_t *zio, arc_buf_t *buf, void *vdb)
1915fa9e4066Sahrens {
1916fa9e4066Sahrens 	dmu_buf_impl_t *db = vdb;
1917fa9e4066Sahrens 	dnode_t *dn = db->db_dnode;
1918fa9e4066Sahrens 	objset_impl_t *os = dn->dn_objset;
1919fa9e4066Sahrens 	uint64_t txg = zio->io_txg;
1920fa9e4066Sahrens 	uint64_t fill = 0;
1921fa9e4066Sahrens 	int i;
1922fa9e4066Sahrens 	int old_size, new_size;
1923fa9e4066Sahrens 
1924fa9e4066Sahrens 	ASSERT3U(zio->io_error, ==, 0);
1925fa9e4066Sahrens 
1926fa9e4066Sahrens 	dprintf_dbuf_bp(db, &zio->io_bp_orig, "bp_orig: %s", "");
1927fa9e4066Sahrens 
1928fa9e4066Sahrens 	old_size = BP_GET_ASIZE(&zio->io_bp_orig);
1929fa9e4066Sahrens 	new_size = BP_GET_ASIZE(zio->io_bp);
1930fa9e4066Sahrens 
1931fa9e4066Sahrens 	dnode_diduse_space(dn, new_size-old_size);
1932fa9e4066Sahrens 
1933fa9e4066Sahrens 	mutex_enter(&db->db_mtx);
1934fa9e4066Sahrens 
1935fa9e4066Sahrens 	if (db->db_dirtied == txg)
1936fa9e4066Sahrens 		db->db_dirtied = 0;
1937fa9e4066Sahrens 
1938fa9e4066Sahrens 	if (db->db_level == 0) {
1939fa9e4066Sahrens 		arc_buf_t **old = &db->db_d.db_data_old[txg&TXG_MASK];
1940fa9e4066Sahrens 
1941fa9e4066Sahrens 		ASSERT(db->db_blkid != DB_BONUS_BLKID);
1942fa9e4066Sahrens 
1943fa9e4066Sahrens 		if (*old != db->db_buf)
1944fa9e4066Sahrens 			arc_buf_free(*old, db);
1945fa9e4066Sahrens 		*old = NULL;
1946fa9e4066Sahrens 		db->db_data_pending = NULL;
1947fa9e4066Sahrens 
1948fa9e4066Sahrens 		mutex_enter(&dn->dn_mtx);
1949fa9e4066Sahrens 		if (db->db_blkid > dn->dn_phys->dn_maxblkid &&
1950fa9e4066Sahrens 		    !BP_IS_HOLE(db->db_blkptr))
1951fa9e4066Sahrens 			dn->dn_phys->dn_maxblkid = db->db_blkid;
1952fa9e4066Sahrens 		mutex_exit(&dn->dn_mtx);
1953fa9e4066Sahrens 
1954fa9e4066Sahrens 		if (dn->dn_type == DMU_OT_DNODE) {
1955fa9e4066Sahrens 			dnode_phys_t *dnp = db->db.db_data;
1956fa9e4066Sahrens 			for (i = db->db.db_size >> DNODE_SHIFT; i > 0;
1957fa9e4066Sahrens 			    i--, dnp++) {
1958fa9e4066Sahrens 				if (dnp->dn_type != DMU_OT_NONE)
1959fa9e4066Sahrens 					fill++;
1960fa9e4066Sahrens 			}
1961fa9e4066Sahrens 		} else {
1962fa9e4066Sahrens 			if (!BP_IS_HOLE(db->db_blkptr))
1963fa9e4066Sahrens 				fill = 1;
1964fa9e4066Sahrens 		}
1965fa9e4066Sahrens 	} else {
1966fa9e4066Sahrens 		blkptr_t *bp = db->db.db_data;
1967fa9e4066Sahrens 		ASSERT3U(db->db.db_size, ==, 1<<dn->dn_phys->dn_indblkshift);
1968fa9e4066Sahrens 		if (!BP_IS_HOLE(db->db_blkptr)) {
1969fa9e4066Sahrens 			ASSERT3U(BP_GET_LSIZE(zio->io_bp), ==, db->db.db_size);
1970fa9e4066Sahrens 			ASSERT3U(BP_GET_LSIZE(db->db_blkptr), ==,
1971fa9e4066Sahrens 			    db->db.db_size);
1972fa9e4066Sahrens 		}
1973fa9e4066Sahrens 		for (i = db->db.db_size >> SPA_BLKPTRSHIFT; i > 0; i--, bp++) {
1974fa9e4066Sahrens 			if (BP_IS_HOLE(bp))
1975fa9e4066Sahrens 				continue;
1976fa9e4066Sahrens 			ASSERT3U(BP_GET_LSIZE(bp), ==,
1977fa9e4066Sahrens 			    db->db_level == 1 ? dn->dn_datablksz :
1978fa9e4066Sahrens 			    (1<<dn->dn_phys->dn_indblkshift));
1979fa9e4066Sahrens 			fill += bp->blk_fill;
1980fa9e4066Sahrens 		}
1981fa9e4066Sahrens 	}
1982fa9e4066Sahrens 
1983fa9e4066Sahrens 	if (!BP_IS_HOLE(db->db_blkptr)) {
1984fa9e4066Sahrens 		db->db_blkptr->blk_fill = fill;
1985fa9e4066Sahrens 		BP_SET_TYPE(db->db_blkptr, dn->dn_type);
1986fa9e4066Sahrens 		BP_SET_LEVEL(db->db_blkptr, db->db_level);
1987fa9e4066Sahrens 	} else {
1988fa9e4066Sahrens 		ASSERT3U(fill, ==, 0);
1989fa9e4066Sahrens 		ASSERT3U(db->db_blkptr->blk_fill, ==, 0);
1990fa9e4066Sahrens 	}
1991fa9e4066Sahrens 
1992fa9e4066Sahrens 	dprintf_dbuf_bp(db, db->db_blkptr,
1993fa9e4066Sahrens 	    "wrote %llu bytes to blkptr:", zio->io_size);
1994fa9e4066Sahrens 
1995fa9e4066Sahrens 	ASSERT(db->db_parent == NULL ||
1996fa9e4066Sahrens 	    list_link_active(&db->db_parent->db_dirty_node[txg&TXG_MASK]));
1997fa9e4066Sahrens 	cv_broadcast(&db->db_changed);
1998fa9e4066Sahrens 	ASSERT(db->db_dirtycnt > 0);
1999fa9e4066Sahrens 	db->db_dirtycnt -= 1;
2000fa9e4066Sahrens 	mutex_exit(&db->db_mtx);
2001fa9e4066Sahrens 
2002fa9e4066Sahrens 	/* We must do this after we've set the bp's type and level */
2003fa9e4066Sahrens 	if (!DVA_EQUAL(BP_IDENTITY(zio->io_bp),
2004fa9e4066Sahrens 	    BP_IDENTITY(&zio->io_bp_orig))) {
2005fa9e4066Sahrens 		struct dbuf_arg *da;
2006fa9e4066Sahrens 		da = kmem_alloc(sizeof (struct dbuf_arg), KM_SLEEP);
2007fa9e4066Sahrens 		da->os = os;
2008fa9e4066Sahrens 		da->bp = *zio->io_bp;
2009fa9e4066Sahrens 		(void) taskq_dispatch(dbuf_tq, dbuf_do_born, da, 0);
2010fa9e4066Sahrens 		if (!BP_IS_HOLE(&zio->io_bp_orig)) {
2011fa9e4066Sahrens 			da = kmem_alloc(sizeof (struct dbuf_arg), KM_SLEEP);
2012fa9e4066Sahrens 			da->os = os;
2013fa9e4066Sahrens 			da->bp = zio->io_bp_orig;
2014fa9e4066Sahrens 			(void) taskq_dispatch(dbuf_tq, dbuf_do_kill, da, 0);
2015fa9e4066Sahrens 		}
2016fa9e4066Sahrens 	}
2017fa9e4066Sahrens 
2018fa9e4066Sahrens 	dbuf_remove_ref(db, (void *)(uintptr_t)txg);
2019fa9e4066Sahrens }
2020