xref: /freebsd/sys/contrib/openzfs/module/zfs/dbuf.c (revision 0d8fe2373503aeac48492f28073049a8bfa4feb5)
1eda14cbcSMatt Macy /*
2eda14cbcSMatt Macy  * CDDL HEADER START
3eda14cbcSMatt Macy  *
4eda14cbcSMatt Macy  * The contents of this file are subject to the terms of the
5eda14cbcSMatt Macy  * Common Development and Distribution License (the "License").
6eda14cbcSMatt Macy  * You may not use this file except in compliance with the License.
7eda14cbcSMatt Macy  *
8eda14cbcSMatt Macy  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9eda14cbcSMatt Macy  * or http://www.opensolaris.org/os/licensing.
10eda14cbcSMatt Macy  * See the License for the specific language governing permissions
11eda14cbcSMatt Macy  * and limitations under the License.
12eda14cbcSMatt Macy  *
13eda14cbcSMatt Macy  * When distributing Covered Code, include this CDDL HEADER in each
14eda14cbcSMatt Macy  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15eda14cbcSMatt Macy  * If applicable, add the following below this CDDL HEADER, with the
16eda14cbcSMatt Macy  * fields enclosed by brackets "[]" replaced with your own identifying
17eda14cbcSMatt Macy  * information: Portions Copyright [yyyy] [name of copyright owner]
18eda14cbcSMatt Macy  *
19eda14cbcSMatt Macy  * CDDL HEADER END
20eda14cbcSMatt Macy  */
21eda14cbcSMatt Macy /*
22eda14cbcSMatt Macy  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
23eda14cbcSMatt Macy  * Copyright 2011 Nexenta Systems, Inc.  All rights reserved.
247877fdebSMatt Macy  * Copyright (c) 2012, 2020 by Delphix. All rights reserved.
25eda14cbcSMatt Macy  * Copyright (c) 2013 by Saso Kiselkov. All rights reserved.
26eda14cbcSMatt Macy  * Copyright (c) 2014 Spectra Logic Corporation, All rights reserved.
27eda14cbcSMatt Macy  * Copyright (c) 2019, Klara Inc.
28eda14cbcSMatt Macy  * Copyright (c) 2019, Allan Jude
29eda14cbcSMatt Macy  */
30eda14cbcSMatt Macy 
31eda14cbcSMatt Macy #include <sys/zfs_context.h>
32eda14cbcSMatt Macy #include <sys/arc.h>
33eda14cbcSMatt Macy #include <sys/dmu.h>
34eda14cbcSMatt Macy #include <sys/dmu_send.h>
35eda14cbcSMatt Macy #include <sys/dmu_impl.h>
36eda14cbcSMatt Macy #include <sys/dbuf.h>
37eda14cbcSMatt Macy #include <sys/dmu_objset.h>
38eda14cbcSMatt Macy #include <sys/dsl_dataset.h>
39eda14cbcSMatt Macy #include <sys/dsl_dir.h>
40eda14cbcSMatt Macy #include <sys/dmu_tx.h>
41eda14cbcSMatt Macy #include <sys/spa.h>
42eda14cbcSMatt Macy #include <sys/zio.h>
43eda14cbcSMatt Macy #include <sys/dmu_zfetch.h>
44eda14cbcSMatt Macy #include <sys/sa.h>
45eda14cbcSMatt Macy #include <sys/sa_impl.h>
46eda14cbcSMatt Macy #include <sys/zfeature.h>
47eda14cbcSMatt Macy #include <sys/blkptr.h>
48eda14cbcSMatt Macy #include <sys/range_tree.h>
49eda14cbcSMatt Macy #include <sys/trace_zfs.h>
50eda14cbcSMatt Macy #include <sys/callb.h>
51eda14cbcSMatt Macy #include <sys/abd.h>
52eda14cbcSMatt Macy #include <sys/vdev.h>
53eda14cbcSMatt Macy #include <cityhash.h>
54eda14cbcSMatt Macy #include <sys/spa_impl.h>
55*0d8fe237SMartin Matuska #include <sys/wmsum.h>
56eda14cbcSMatt Macy 
57eda14cbcSMatt Macy kstat_t *dbuf_ksp;
58eda14cbcSMatt Macy 
59eda14cbcSMatt Macy typedef struct dbuf_stats {
60eda14cbcSMatt Macy 	/*
61eda14cbcSMatt Macy 	 * Various statistics about the size of the dbuf cache.
62eda14cbcSMatt Macy 	 */
63eda14cbcSMatt Macy 	kstat_named_t cache_count;
64eda14cbcSMatt Macy 	kstat_named_t cache_size_bytes;
65eda14cbcSMatt Macy 	kstat_named_t cache_size_bytes_max;
66eda14cbcSMatt Macy 	/*
67eda14cbcSMatt Macy 	 * Statistics regarding the bounds on the dbuf cache size.
68eda14cbcSMatt Macy 	 */
69eda14cbcSMatt Macy 	kstat_named_t cache_target_bytes;
70eda14cbcSMatt Macy 	kstat_named_t cache_lowater_bytes;
71eda14cbcSMatt Macy 	kstat_named_t cache_hiwater_bytes;
72eda14cbcSMatt Macy 	/*
73eda14cbcSMatt Macy 	 * Total number of dbuf cache evictions that have occurred.
74eda14cbcSMatt Macy 	 */
75eda14cbcSMatt Macy 	kstat_named_t cache_total_evicts;
76eda14cbcSMatt Macy 	/*
77eda14cbcSMatt Macy 	 * The distribution of dbuf levels in the dbuf cache and
78eda14cbcSMatt Macy 	 * the total size of all dbufs at each level.
79eda14cbcSMatt Macy 	 */
80eda14cbcSMatt Macy 	kstat_named_t cache_levels[DN_MAX_LEVELS];
81eda14cbcSMatt Macy 	kstat_named_t cache_levels_bytes[DN_MAX_LEVELS];
82eda14cbcSMatt Macy 	/*
83eda14cbcSMatt Macy 	 * Statistics about the dbuf hash table.
84eda14cbcSMatt Macy 	 */
85eda14cbcSMatt Macy 	kstat_named_t hash_hits;
86eda14cbcSMatt Macy 	kstat_named_t hash_misses;
87eda14cbcSMatt Macy 	kstat_named_t hash_collisions;
88eda14cbcSMatt Macy 	kstat_named_t hash_elements;
89eda14cbcSMatt Macy 	kstat_named_t hash_elements_max;
90eda14cbcSMatt Macy 	/*
91eda14cbcSMatt Macy 	 * Number of sublists containing more than one dbuf in the dbuf
92eda14cbcSMatt Macy 	 * hash table. Keep track of the longest hash chain.
93eda14cbcSMatt Macy 	 */
94eda14cbcSMatt Macy 	kstat_named_t hash_chains;
95eda14cbcSMatt Macy 	kstat_named_t hash_chain_max;
96eda14cbcSMatt Macy 	/*
97eda14cbcSMatt Macy 	 * Number of times a dbuf_create() discovers that a dbuf was
98eda14cbcSMatt Macy 	 * already created and in the dbuf hash table.
99eda14cbcSMatt Macy 	 */
100eda14cbcSMatt Macy 	kstat_named_t hash_insert_race;
101eda14cbcSMatt Macy 	/*
102eda14cbcSMatt Macy 	 * Statistics about the size of the metadata dbuf cache.
103eda14cbcSMatt Macy 	 */
104eda14cbcSMatt Macy 	kstat_named_t metadata_cache_count;
105eda14cbcSMatt Macy 	kstat_named_t metadata_cache_size_bytes;
106eda14cbcSMatt Macy 	kstat_named_t metadata_cache_size_bytes_max;
107eda14cbcSMatt Macy 	/*
108eda14cbcSMatt Macy 	 * For diagnostic purposes, this is incremented whenever we can't add
109eda14cbcSMatt Macy 	 * something to the metadata cache because it's full, and instead put
110eda14cbcSMatt Macy 	 * the data in the regular dbuf cache.
111eda14cbcSMatt Macy 	 */
112eda14cbcSMatt Macy 	kstat_named_t metadata_cache_overflow;
113eda14cbcSMatt Macy } dbuf_stats_t;
114eda14cbcSMatt Macy 
115eda14cbcSMatt Macy dbuf_stats_t dbuf_stats = {
116eda14cbcSMatt Macy 	{ "cache_count",			KSTAT_DATA_UINT64 },
117eda14cbcSMatt Macy 	{ "cache_size_bytes",			KSTAT_DATA_UINT64 },
118eda14cbcSMatt Macy 	{ "cache_size_bytes_max",		KSTAT_DATA_UINT64 },
119eda14cbcSMatt Macy 	{ "cache_target_bytes",			KSTAT_DATA_UINT64 },
120eda14cbcSMatt Macy 	{ "cache_lowater_bytes",		KSTAT_DATA_UINT64 },
121eda14cbcSMatt Macy 	{ "cache_hiwater_bytes",		KSTAT_DATA_UINT64 },
122eda14cbcSMatt Macy 	{ "cache_total_evicts",			KSTAT_DATA_UINT64 },
123eda14cbcSMatt Macy 	{ { "cache_levels_N",			KSTAT_DATA_UINT64 } },
124eda14cbcSMatt Macy 	{ { "cache_levels_bytes_N",		KSTAT_DATA_UINT64 } },
125eda14cbcSMatt Macy 	{ "hash_hits",				KSTAT_DATA_UINT64 },
126eda14cbcSMatt Macy 	{ "hash_misses",			KSTAT_DATA_UINT64 },
127eda14cbcSMatt Macy 	{ "hash_collisions",			KSTAT_DATA_UINT64 },
128eda14cbcSMatt Macy 	{ "hash_elements",			KSTAT_DATA_UINT64 },
129eda14cbcSMatt Macy 	{ "hash_elements_max",			KSTAT_DATA_UINT64 },
130eda14cbcSMatt Macy 	{ "hash_chains",			KSTAT_DATA_UINT64 },
131eda14cbcSMatt Macy 	{ "hash_chain_max",			KSTAT_DATA_UINT64 },
132eda14cbcSMatt Macy 	{ "hash_insert_race",			KSTAT_DATA_UINT64 },
133eda14cbcSMatt Macy 	{ "metadata_cache_count",		KSTAT_DATA_UINT64 },
134eda14cbcSMatt Macy 	{ "metadata_cache_size_bytes",		KSTAT_DATA_UINT64 },
135eda14cbcSMatt Macy 	{ "metadata_cache_size_bytes_max",	KSTAT_DATA_UINT64 },
136eda14cbcSMatt Macy 	{ "metadata_cache_overflow",		KSTAT_DATA_UINT64 }
137eda14cbcSMatt Macy };
138eda14cbcSMatt Macy 
139*0d8fe237SMartin Matuska struct {
140*0d8fe237SMartin Matuska 	wmsum_t cache_count;
141*0d8fe237SMartin Matuska 	wmsum_t cache_total_evicts;
142*0d8fe237SMartin Matuska 	wmsum_t cache_levels[DN_MAX_LEVELS];
143*0d8fe237SMartin Matuska 	wmsum_t cache_levels_bytes[DN_MAX_LEVELS];
144*0d8fe237SMartin Matuska 	wmsum_t hash_hits;
145*0d8fe237SMartin Matuska 	wmsum_t hash_misses;
146*0d8fe237SMartin Matuska 	wmsum_t hash_collisions;
147*0d8fe237SMartin Matuska 	wmsum_t hash_chains;
148*0d8fe237SMartin Matuska 	wmsum_t hash_insert_race;
149*0d8fe237SMartin Matuska 	wmsum_t metadata_cache_count;
150*0d8fe237SMartin Matuska 	wmsum_t metadata_cache_overflow;
151*0d8fe237SMartin Matuska } dbuf_sums;
152*0d8fe237SMartin Matuska 
153eda14cbcSMatt Macy #define	DBUF_STAT_INCR(stat, val)	\
154*0d8fe237SMartin Matuska 	wmsum_add(&dbuf_sums.stat, val);
155eda14cbcSMatt Macy #define	DBUF_STAT_DECR(stat, val)	\
156eda14cbcSMatt Macy 	DBUF_STAT_INCR(stat, -(val));
157eda14cbcSMatt Macy #define	DBUF_STAT_BUMP(stat)		\
158eda14cbcSMatt Macy 	DBUF_STAT_INCR(stat, 1);
159eda14cbcSMatt Macy #define	DBUF_STAT_BUMPDOWN(stat)	\
160eda14cbcSMatt Macy 	DBUF_STAT_INCR(stat, -1);
161eda14cbcSMatt Macy #define	DBUF_STAT_MAX(stat, v) {					\
162eda14cbcSMatt Macy 	uint64_t _m;							\
163eda14cbcSMatt Macy 	while ((v) > (_m = dbuf_stats.stat.value.ui64) &&		\
164eda14cbcSMatt Macy 	    (_m != atomic_cas_64(&dbuf_stats.stat.value.ui64, _m, (v))))\
165eda14cbcSMatt Macy 		continue;						\
166eda14cbcSMatt Macy }
167eda14cbcSMatt Macy 
168eda14cbcSMatt Macy static boolean_t dbuf_undirty(dmu_buf_impl_t *db, dmu_tx_t *tx);
169eda14cbcSMatt Macy static void dbuf_write(dbuf_dirty_record_t *dr, arc_buf_t *data, dmu_tx_t *tx);
170eda14cbcSMatt Macy static void dbuf_sync_leaf_verify_bonus_dnode(dbuf_dirty_record_t *dr);
171eda14cbcSMatt Macy static int dbuf_read_verify_dnode_crypt(dmu_buf_impl_t *db, uint32_t flags);
172eda14cbcSMatt Macy 
173eda14cbcSMatt Macy extern inline void dmu_buf_init_user(dmu_buf_user_t *dbu,
174eda14cbcSMatt Macy     dmu_buf_evict_func_t *evict_func_sync,
175eda14cbcSMatt Macy     dmu_buf_evict_func_t *evict_func_async,
176eda14cbcSMatt Macy     dmu_buf_t **clear_on_evict_dbufp);
177eda14cbcSMatt Macy 
178eda14cbcSMatt Macy /*
179eda14cbcSMatt Macy  * Global data structures and functions for the dbuf cache.
180eda14cbcSMatt Macy  */
181eda14cbcSMatt Macy static kmem_cache_t *dbuf_kmem_cache;
182eda14cbcSMatt Macy static taskq_t *dbu_evict_taskq;
183eda14cbcSMatt Macy 
184eda14cbcSMatt Macy static kthread_t *dbuf_cache_evict_thread;
185eda14cbcSMatt Macy static kmutex_t dbuf_evict_lock;
186eda14cbcSMatt Macy static kcondvar_t dbuf_evict_cv;
187eda14cbcSMatt Macy static boolean_t dbuf_evict_thread_exit;
188eda14cbcSMatt Macy 
189eda14cbcSMatt Macy /*
190eda14cbcSMatt Macy  * There are two dbuf caches; each dbuf can only be in one of them at a time.
191eda14cbcSMatt Macy  *
192eda14cbcSMatt Macy  * 1. Cache of metadata dbufs, to help make read-heavy administrative commands
193eda14cbcSMatt Macy  *    from /sbin/zfs run faster. The "metadata cache" specifically stores dbufs
194eda14cbcSMatt Macy  *    that represent the metadata that describes filesystems/snapshots/
195eda14cbcSMatt Macy  *    bookmarks/properties/etc. We only evict from this cache when we export a
196eda14cbcSMatt Macy  *    pool, to short-circuit as much I/O as possible for all administrative
197eda14cbcSMatt Macy  *    commands that need the metadata. There is no eviction policy for this
198eda14cbcSMatt Macy  *    cache, because we try to only include types in it which would occupy a
199eda14cbcSMatt Macy  *    very small amount of space per object but create a large impact on the
200eda14cbcSMatt Macy  *    performance of these commands. Instead, after it reaches a maximum size
201eda14cbcSMatt Macy  *    (which should only happen on very small memory systems with a very large
202eda14cbcSMatt Macy  *    number of filesystem objects), we stop taking new dbufs into the
203eda14cbcSMatt Macy  *    metadata cache, instead putting them in the normal dbuf cache.
204eda14cbcSMatt Macy  *
205eda14cbcSMatt Macy  * 2. LRU cache of dbufs. The dbuf cache maintains a list of dbufs that
206eda14cbcSMatt Macy  *    are not currently held but have been recently released. These dbufs
207eda14cbcSMatt Macy  *    are not eligible for arc eviction until they are aged out of the cache.
208eda14cbcSMatt Macy  *    Dbufs that are aged out of the cache will be immediately destroyed and
209eda14cbcSMatt Macy  *    become eligible for arc eviction.
210eda14cbcSMatt Macy  *
211eda14cbcSMatt Macy  * Dbufs are added to these caches once the last hold is released. If a dbuf is
212eda14cbcSMatt Macy  * later accessed and still exists in the dbuf cache, then it will be removed
213eda14cbcSMatt Macy  * from the cache and later re-added to the head of the cache.
214eda14cbcSMatt Macy  *
215eda14cbcSMatt Macy  * If a given dbuf meets the requirements for the metadata cache, it will go
216eda14cbcSMatt Macy  * there, otherwise it will be considered for the generic LRU dbuf cache. The
217eda14cbcSMatt Macy  * caches and the refcounts tracking their sizes are stored in an array indexed
218eda14cbcSMatt Macy  * by those caches' matching enum values (from dbuf_cached_state_t).
219eda14cbcSMatt Macy  */
220eda14cbcSMatt Macy typedef struct dbuf_cache {
2213ff01b23SMartin Matuska 	multilist_t cache;
2223ff01b23SMartin Matuska 	zfs_refcount_t size ____cacheline_aligned;
223eda14cbcSMatt Macy } dbuf_cache_t;
224eda14cbcSMatt Macy dbuf_cache_t dbuf_caches[DB_CACHE_MAX];
225eda14cbcSMatt Macy 
226eda14cbcSMatt Macy /* Size limits for the caches */
227eda14cbcSMatt Macy unsigned long dbuf_cache_max_bytes = ULONG_MAX;
228eda14cbcSMatt Macy unsigned long dbuf_metadata_cache_max_bytes = ULONG_MAX;
229eda14cbcSMatt Macy 
230eda14cbcSMatt Macy /* Set the default sizes of the caches to log2 fraction of arc size */
231eda14cbcSMatt Macy int dbuf_cache_shift = 5;
232eda14cbcSMatt Macy int dbuf_metadata_cache_shift = 6;
233eda14cbcSMatt Macy 
234eda14cbcSMatt Macy static unsigned long dbuf_cache_target_bytes(void);
235eda14cbcSMatt Macy static unsigned long dbuf_metadata_cache_target_bytes(void);
236eda14cbcSMatt Macy 
237eda14cbcSMatt Macy /*
238eda14cbcSMatt Macy  * The LRU dbuf cache uses a three-stage eviction policy:
239eda14cbcSMatt Macy  *	- A low water marker designates when the dbuf eviction thread
240eda14cbcSMatt Macy  *	should stop evicting from the dbuf cache.
241eda14cbcSMatt Macy  *	- When we reach the maximum size (aka mid water mark), we
242eda14cbcSMatt Macy  *	signal the eviction thread to run.
243eda14cbcSMatt Macy  *	- The high water mark indicates when the eviction thread
244eda14cbcSMatt Macy  *	is unable to keep up with the incoming load and eviction must
245eda14cbcSMatt Macy  *	happen in the context of the calling thread.
246eda14cbcSMatt Macy  *
247eda14cbcSMatt Macy  * The dbuf cache:
248eda14cbcSMatt Macy  *                                                 (max size)
249eda14cbcSMatt Macy  *                                      low water   mid water   hi water
250eda14cbcSMatt Macy  * +----------------------------------------+----------+----------+
251eda14cbcSMatt Macy  * |                                        |          |          |
252eda14cbcSMatt Macy  * |                                        |          |          |
253eda14cbcSMatt Macy  * |                                        |          |          |
254eda14cbcSMatt Macy  * |                                        |          |          |
255eda14cbcSMatt Macy  * +----------------------------------------+----------+----------+
256eda14cbcSMatt Macy  *                                        stop        signal     evict
257eda14cbcSMatt Macy  *                                      evicting     eviction   directly
258eda14cbcSMatt Macy  *                                                    thread
259eda14cbcSMatt Macy  *
260eda14cbcSMatt Macy  * The high and low water marks indicate the operating range for the eviction
261eda14cbcSMatt Macy  * thread. The low water mark is, by default, 90% of the total size of the
262eda14cbcSMatt Macy  * cache and the high water mark is at 110% (both of these percentages can be
263eda14cbcSMatt Macy  * changed by setting dbuf_cache_lowater_pct and dbuf_cache_hiwater_pct,
264eda14cbcSMatt Macy  * respectively). The eviction thread will try to ensure that the cache remains
265eda14cbcSMatt Macy  * within this range by waking up every second and checking if the cache is
266eda14cbcSMatt Macy  * above the low water mark. The thread can also be woken up by callers adding
267eda14cbcSMatt Macy  * elements into the cache if the cache is larger than the mid water (i.e max
268eda14cbcSMatt Macy  * cache size). Once the eviction thread is woken up and eviction is required,
269eda14cbcSMatt Macy  * it will continue evicting buffers until it's able to reduce the cache size
270eda14cbcSMatt Macy  * to the low water mark. If the cache size continues to grow and hits the high
271eda14cbcSMatt Macy  * water mark, then callers adding elements to the cache will begin to evict
272eda14cbcSMatt Macy  * directly from the cache until the cache is no longer above the high water
273eda14cbcSMatt Macy  * mark.
274eda14cbcSMatt Macy  */
275eda14cbcSMatt Macy 
276eda14cbcSMatt Macy /*
277eda14cbcSMatt Macy  * The percentage above and below the maximum cache size.
278eda14cbcSMatt Macy  */
279eda14cbcSMatt Macy uint_t dbuf_cache_hiwater_pct = 10;
280eda14cbcSMatt Macy uint_t dbuf_cache_lowater_pct = 10;
281eda14cbcSMatt Macy 
282eda14cbcSMatt Macy /* ARGSUSED */
283eda14cbcSMatt Macy static int
284eda14cbcSMatt Macy dbuf_cons(void *vdb, void *unused, int kmflag)
285eda14cbcSMatt Macy {
286eda14cbcSMatt Macy 	dmu_buf_impl_t *db = vdb;
287eda14cbcSMatt Macy 	bzero(db, sizeof (dmu_buf_impl_t));
288eda14cbcSMatt Macy 
289eda14cbcSMatt Macy 	mutex_init(&db->db_mtx, NULL, MUTEX_DEFAULT, NULL);
290eda14cbcSMatt Macy 	rw_init(&db->db_rwlock, NULL, RW_DEFAULT, NULL);
291eda14cbcSMatt Macy 	cv_init(&db->db_changed, NULL, CV_DEFAULT, NULL);
292eda14cbcSMatt Macy 	multilist_link_init(&db->db_cache_link);
293eda14cbcSMatt Macy 	zfs_refcount_create(&db->db_holds);
294eda14cbcSMatt Macy 
295eda14cbcSMatt Macy 	return (0);
296eda14cbcSMatt Macy }
297eda14cbcSMatt Macy 
298eda14cbcSMatt Macy /* ARGSUSED */
299eda14cbcSMatt Macy static void
300eda14cbcSMatt Macy dbuf_dest(void *vdb, void *unused)
301eda14cbcSMatt Macy {
302eda14cbcSMatt Macy 	dmu_buf_impl_t *db = vdb;
303eda14cbcSMatt Macy 	mutex_destroy(&db->db_mtx);
304eda14cbcSMatt Macy 	rw_destroy(&db->db_rwlock);
305eda14cbcSMatt Macy 	cv_destroy(&db->db_changed);
306eda14cbcSMatt Macy 	ASSERT(!multilist_link_active(&db->db_cache_link));
307eda14cbcSMatt Macy 	zfs_refcount_destroy(&db->db_holds);
308eda14cbcSMatt Macy }
309eda14cbcSMatt Macy 
310eda14cbcSMatt Macy /*
311eda14cbcSMatt Macy  * dbuf hash table routines
312eda14cbcSMatt Macy  */
313eda14cbcSMatt Macy static dbuf_hash_table_t dbuf_hash_table;
314eda14cbcSMatt Macy 
315eda14cbcSMatt Macy /*
316eda14cbcSMatt Macy  * We use Cityhash for this. It's fast, and has good hash properties without
317eda14cbcSMatt Macy  * requiring any large static buffers.
318eda14cbcSMatt Macy  */
319eda14cbcSMatt Macy static uint64_t
320eda14cbcSMatt Macy dbuf_hash(void *os, uint64_t obj, uint8_t lvl, uint64_t blkid)
321eda14cbcSMatt Macy {
322eda14cbcSMatt Macy 	return (cityhash4((uintptr_t)os, obj, (uint64_t)lvl, blkid));
323eda14cbcSMatt Macy }
324eda14cbcSMatt Macy 
325eda14cbcSMatt Macy #define	DTRACE_SET_STATE(db, why) \
326eda14cbcSMatt Macy 	DTRACE_PROBE2(dbuf__state_change, dmu_buf_impl_t *, db,	\
327eda14cbcSMatt Macy 	    const char *, why)
328eda14cbcSMatt Macy 
329eda14cbcSMatt Macy #define	DBUF_EQUAL(dbuf, os, obj, level, blkid)		\
330eda14cbcSMatt Macy 	((dbuf)->db.db_object == (obj) &&		\
331eda14cbcSMatt Macy 	(dbuf)->db_objset == (os) &&			\
332eda14cbcSMatt Macy 	(dbuf)->db_level == (level) &&			\
333eda14cbcSMatt Macy 	(dbuf)->db_blkid == (blkid))
334eda14cbcSMatt Macy 
335eda14cbcSMatt Macy dmu_buf_impl_t *
336eda14cbcSMatt Macy dbuf_find(objset_t *os, uint64_t obj, uint8_t level, uint64_t blkid)
337eda14cbcSMatt Macy {
338eda14cbcSMatt Macy 	dbuf_hash_table_t *h = &dbuf_hash_table;
339eda14cbcSMatt Macy 	uint64_t hv;
340eda14cbcSMatt Macy 	uint64_t idx;
341eda14cbcSMatt Macy 	dmu_buf_impl_t *db;
342eda14cbcSMatt Macy 
343eda14cbcSMatt Macy 	hv = dbuf_hash(os, obj, level, blkid);
344eda14cbcSMatt Macy 	idx = hv & h->hash_table_mask;
345eda14cbcSMatt Macy 
346eda14cbcSMatt Macy 	mutex_enter(DBUF_HASH_MUTEX(h, idx));
347eda14cbcSMatt Macy 	for (db = h->hash_table[idx]; db != NULL; db = db->db_hash_next) {
348eda14cbcSMatt Macy 		if (DBUF_EQUAL(db, os, obj, level, blkid)) {
349eda14cbcSMatt Macy 			mutex_enter(&db->db_mtx);
350eda14cbcSMatt Macy 			if (db->db_state != DB_EVICTING) {
351eda14cbcSMatt Macy 				mutex_exit(DBUF_HASH_MUTEX(h, idx));
352eda14cbcSMatt Macy 				return (db);
353eda14cbcSMatt Macy 			}
354eda14cbcSMatt Macy 			mutex_exit(&db->db_mtx);
355eda14cbcSMatt Macy 		}
356eda14cbcSMatt Macy 	}
357eda14cbcSMatt Macy 	mutex_exit(DBUF_HASH_MUTEX(h, idx));
358eda14cbcSMatt Macy 	return (NULL);
359eda14cbcSMatt Macy }
360eda14cbcSMatt Macy 
361eda14cbcSMatt Macy static dmu_buf_impl_t *
362eda14cbcSMatt Macy dbuf_find_bonus(objset_t *os, uint64_t object)
363eda14cbcSMatt Macy {
364eda14cbcSMatt Macy 	dnode_t *dn;
365eda14cbcSMatt Macy 	dmu_buf_impl_t *db = NULL;
366eda14cbcSMatt Macy 
367eda14cbcSMatt Macy 	if (dnode_hold(os, object, FTAG, &dn) == 0) {
368eda14cbcSMatt Macy 		rw_enter(&dn->dn_struct_rwlock, RW_READER);
369eda14cbcSMatt Macy 		if (dn->dn_bonus != NULL) {
370eda14cbcSMatt Macy 			db = dn->dn_bonus;
371eda14cbcSMatt Macy 			mutex_enter(&db->db_mtx);
372eda14cbcSMatt Macy 		}
373eda14cbcSMatt Macy 		rw_exit(&dn->dn_struct_rwlock);
374eda14cbcSMatt Macy 		dnode_rele(dn, FTAG);
375eda14cbcSMatt Macy 	}
376eda14cbcSMatt Macy 	return (db);
377eda14cbcSMatt Macy }
378eda14cbcSMatt Macy 
379eda14cbcSMatt Macy /*
380eda14cbcSMatt Macy  * Insert an entry into the hash table.  If there is already an element
381eda14cbcSMatt Macy  * equal to elem in the hash table, then the already existing element
382eda14cbcSMatt Macy  * will be returned and the new element will not be inserted.
383eda14cbcSMatt Macy  * Otherwise returns NULL.
384eda14cbcSMatt Macy  */
385eda14cbcSMatt Macy static dmu_buf_impl_t *
386eda14cbcSMatt Macy dbuf_hash_insert(dmu_buf_impl_t *db)
387eda14cbcSMatt Macy {
388eda14cbcSMatt Macy 	dbuf_hash_table_t *h = &dbuf_hash_table;
389eda14cbcSMatt Macy 	objset_t *os = db->db_objset;
390eda14cbcSMatt Macy 	uint64_t obj = db->db.db_object;
391eda14cbcSMatt Macy 	int level = db->db_level;
392eda14cbcSMatt Macy 	uint64_t blkid, hv, idx;
393eda14cbcSMatt Macy 	dmu_buf_impl_t *dbf;
394eda14cbcSMatt Macy 	uint32_t i;
395eda14cbcSMatt Macy 
396eda14cbcSMatt Macy 	blkid = db->db_blkid;
397eda14cbcSMatt Macy 	hv = dbuf_hash(os, obj, level, blkid);
398eda14cbcSMatt Macy 	idx = hv & h->hash_table_mask;
399eda14cbcSMatt Macy 
400eda14cbcSMatt Macy 	mutex_enter(DBUF_HASH_MUTEX(h, idx));
401eda14cbcSMatt Macy 	for (dbf = h->hash_table[idx], i = 0; dbf != NULL;
402eda14cbcSMatt Macy 	    dbf = dbf->db_hash_next, i++) {
403eda14cbcSMatt Macy 		if (DBUF_EQUAL(dbf, os, obj, level, blkid)) {
404eda14cbcSMatt Macy 			mutex_enter(&dbf->db_mtx);
405eda14cbcSMatt Macy 			if (dbf->db_state != DB_EVICTING) {
406eda14cbcSMatt Macy 				mutex_exit(DBUF_HASH_MUTEX(h, idx));
407eda14cbcSMatt Macy 				return (dbf);
408eda14cbcSMatt Macy 			}
409eda14cbcSMatt Macy 			mutex_exit(&dbf->db_mtx);
410eda14cbcSMatt Macy 		}
411eda14cbcSMatt Macy 	}
412eda14cbcSMatt Macy 
413eda14cbcSMatt Macy 	if (i > 0) {
414eda14cbcSMatt Macy 		DBUF_STAT_BUMP(hash_collisions);
415eda14cbcSMatt Macy 		if (i == 1)
416eda14cbcSMatt Macy 			DBUF_STAT_BUMP(hash_chains);
417eda14cbcSMatt Macy 
418eda14cbcSMatt Macy 		DBUF_STAT_MAX(hash_chain_max, i);
419eda14cbcSMatt Macy 	}
420eda14cbcSMatt Macy 
421eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
422eda14cbcSMatt Macy 	db->db_hash_next = h->hash_table[idx];
423eda14cbcSMatt Macy 	h->hash_table[idx] = db;
424eda14cbcSMatt Macy 	mutex_exit(DBUF_HASH_MUTEX(h, idx));
425*0d8fe237SMartin Matuska 	uint64_t he = atomic_inc_64_nv(&dbuf_stats.hash_elements.value.ui64);
426*0d8fe237SMartin Matuska 	DBUF_STAT_MAX(hash_elements_max, he);
427eda14cbcSMatt Macy 
428eda14cbcSMatt Macy 	return (NULL);
429eda14cbcSMatt Macy }
430eda14cbcSMatt Macy 
431eda14cbcSMatt Macy /*
432eda14cbcSMatt Macy  * This returns whether this dbuf should be stored in the metadata cache, which
433eda14cbcSMatt Macy  * is based on whether it's from one of the dnode types that store data related
434eda14cbcSMatt Macy  * to traversing dataset hierarchies.
435eda14cbcSMatt Macy  */
436eda14cbcSMatt Macy static boolean_t
437eda14cbcSMatt Macy dbuf_include_in_metadata_cache(dmu_buf_impl_t *db)
438eda14cbcSMatt Macy {
439eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
440eda14cbcSMatt Macy 	dmu_object_type_t type = DB_DNODE(db)->dn_type;
441eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
442eda14cbcSMatt Macy 
443eda14cbcSMatt Macy 	/* Check if this dbuf is one of the types we care about */
444eda14cbcSMatt Macy 	if (DMU_OT_IS_METADATA_CACHED(type)) {
445eda14cbcSMatt Macy 		/* If we hit this, then we set something up wrong in dmu_ot */
446eda14cbcSMatt Macy 		ASSERT(DMU_OT_IS_METADATA(type));
447eda14cbcSMatt Macy 
448eda14cbcSMatt Macy 		/*
449eda14cbcSMatt Macy 		 * Sanity check for small-memory systems: don't allocate too
450eda14cbcSMatt Macy 		 * much memory for this purpose.
451eda14cbcSMatt Macy 		 */
452eda14cbcSMatt Macy 		if (zfs_refcount_count(
453eda14cbcSMatt Macy 		    &dbuf_caches[DB_DBUF_METADATA_CACHE].size) >
454eda14cbcSMatt Macy 		    dbuf_metadata_cache_target_bytes()) {
455eda14cbcSMatt Macy 			DBUF_STAT_BUMP(metadata_cache_overflow);
456eda14cbcSMatt Macy 			return (B_FALSE);
457eda14cbcSMatt Macy 		}
458eda14cbcSMatt Macy 
459eda14cbcSMatt Macy 		return (B_TRUE);
460eda14cbcSMatt Macy 	}
461eda14cbcSMatt Macy 
462eda14cbcSMatt Macy 	return (B_FALSE);
463eda14cbcSMatt Macy }
464eda14cbcSMatt Macy 
465eda14cbcSMatt Macy /*
466eda14cbcSMatt Macy  * Remove an entry from the hash table.  It must be in the EVICTING state.
467eda14cbcSMatt Macy  */
468eda14cbcSMatt Macy static void
469eda14cbcSMatt Macy dbuf_hash_remove(dmu_buf_impl_t *db)
470eda14cbcSMatt Macy {
471eda14cbcSMatt Macy 	dbuf_hash_table_t *h = &dbuf_hash_table;
472eda14cbcSMatt Macy 	uint64_t hv, idx;
473eda14cbcSMatt Macy 	dmu_buf_impl_t *dbf, **dbp;
474eda14cbcSMatt Macy 
475eda14cbcSMatt Macy 	hv = dbuf_hash(db->db_objset, db->db.db_object,
476eda14cbcSMatt Macy 	    db->db_level, db->db_blkid);
477eda14cbcSMatt Macy 	idx = hv & h->hash_table_mask;
478eda14cbcSMatt Macy 
479eda14cbcSMatt Macy 	/*
480eda14cbcSMatt Macy 	 * We mustn't hold db_mtx to maintain lock ordering:
481eda14cbcSMatt Macy 	 * DBUF_HASH_MUTEX > db_mtx.
482eda14cbcSMatt Macy 	 */
483eda14cbcSMatt Macy 	ASSERT(zfs_refcount_is_zero(&db->db_holds));
484eda14cbcSMatt Macy 	ASSERT(db->db_state == DB_EVICTING);
485eda14cbcSMatt Macy 	ASSERT(!MUTEX_HELD(&db->db_mtx));
486eda14cbcSMatt Macy 
487eda14cbcSMatt Macy 	mutex_enter(DBUF_HASH_MUTEX(h, idx));
488eda14cbcSMatt Macy 	dbp = &h->hash_table[idx];
489eda14cbcSMatt Macy 	while ((dbf = *dbp) != db) {
490eda14cbcSMatt Macy 		dbp = &dbf->db_hash_next;
491eda14cbcSMatt Macy 		ASSERT(dbf != NULL);
492eda14cbcSMatt Macy 	}
493eda14cbcSMatt Macy 	*dbp = db->db_hash_next;
494eda14cbcSMatt Macy 	db->db_hash_next = NULL;
495eda14cbcSMatt Macy 	if (h->hash_table[idx] &&
496eda14cbcSMatt Macy 	    h->hash_table[idx]->db_hash_next == NULL)
497eda14cbcSMatt Macy 		DBUF_STAT_BUMPDOWN(hash_chains);
498eda14cbcSMatt Macy 	mutex_exit(DBUF_HASH_MUTEX(h, idx));
499*0d8fe237SMartin Matuska 	atomic_dec_64(&dbuf_stats.hash_elements.value.ui64);
500eda14cbcSMatt Macy }
501eda14cbcSMatt Macy 
502eda14cbcSMatt Macy typedef enum {
503eda14cbcSMatt Macy 	DBVU_EVICTING,
504eda14cbcSMatt Macy 	DBVU_NOT_EVICTING
505eda14cbcSMatt Macy } dbvu_verify_type_t;
506eda14cbcSMatt Macy 
507eda14cbcSMatt Macy static void
508eda14cbcSMatt Macy dbuf_verify_user(dmu_buf_impl_t *db, dbvu_verify_type_t verify_type)
509eda14cbcSMatt Macy {
510eda14cbcSMatt Macy #ifdef ZFS_DEBUG
511eda14cbcSMatt Macy 	int64_t holds;
512eda14cbcSMatt Macy 
513eda14cbcSMatt Macy 	if (db->db_user == NULL)
514eda14cbcSMatt Macy 		return;
515eda14cbcSMatt Macy 
516eda14cbcSMatt Macy 	/* Only data blocks support the attachment of user data. */
517eda14cbcSMatt Macy 	ASSERT(db->db_level == 0);
518eda14cbcSMatt Macy 
519eda14cbcSMatt Macy 	/* Clients must resolve a dbuf before attaching user data. */
520eda14cbcSMatt Macy 	ASSERT(db->db.db_data != NULL);
521eda14cbcSMatt Macy 	ASSERT3U(db->db_state, ==, DB_CACHED);
522eda14cbcSMatt Macy 
523eda14cbcSMatt Macy 	holds = zfs_refcount_count(&db->db_holds);
524eda14cbcSMatt Macy 	if (verify_type == DBVU_EVICTING) {
525eda14cbcSMatt Macy 		/*
526eda14cbcSMatt Macy 		 * Immediate eviction occurs when holds == dirtycnt.
527eda14cbcSMatt Macy 		 * For normal eviction buffers, holds is zero on
528eda14cbcSMatt Macy 		 * eviction, except when dbuf_fix_old_data() calls
529eda14cbcSMatt Macy 		 * dbuf_clear_data().  However, the hold count can grow
530eda14cbcSMatt Macy 		 * during eviction even though db_mtx is held (see
531eda14cbcSMatt Macy 		 * dmu_bonus_hold() for an example), so we can only
532eda14cbcSMatt Macy 		 * test the generic invariant that holds >= dirtycnt.
533eda14cbcSMatt Macy 		 */
534eda14cbcSMatt Macy 		ASSERT3U(holds, >=, db->db_dirtycnt);
535eda14cbcSMatt Macy 	} else {
536eda14cbcSMatt Macy 		if (db->db_user_immediate_evict == TRUE)
537eda14cbcSMatt Macy 			ASSERT3U(holds, >=, db->db_dirtycnt);
538eda14cbcSMatt Macy 		else
539eda14cbcSMatt Macy 			ASSERT3U(holds, >, 0);
540eda14cbcSMatt Macy 	}
541eda14cbcSMatt Macy #endif
542eda14cbcSMatt Macy }
543eda14cbcSMatt Macy 
544eda14cbcSMatt Macy static void
545eda14cbcSMatt Macy dbuf_evict_user(dmu_buf_impl_t *db)
546eda14cbcSMatt Macy {
547eda14cbcSMatt Macy 	dmu_buf_user_t *dbu = db->db_user;
548eda14cbcSMatt Macy 
549eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
550eda14cbcSMatt Macy 
551eda14cbcSMatt Macy 	if (dbu == NULL)
552eda14cbcSMatt Macy 		return;
553eda14cbcSMatt Macy 
554eda14cbcSMatt Macy 	dbuf_verify_user(db, DBVU_EVICTING);
555eda14cbcSMatt Macy 	db->db_user = NULL;
556eda14cbcSMatt Macy 
557eda14cbcSMatt Macy #ifdef ZFS_DEBUG
558eda14cbcSMatt Macy 	if (dbu->dbu_clear_on_evict_dbufp != NULL)
559eda14cbcSMatt Macy 		*dbu->dbu_clear_on_evict_dbufp = NULL;
560eda14cbcSMatt Macy #endif
561eda14cbcSMatt Macy 
562eda14cbcSMatt Macy 	/*
563eda14cbcSMatt Macy 	 * There are two eviction callbacks - one that we call synchronously
564eda14cbcSMatt Macy 	 * and one that we invoke via a taskq.  The async one is useful for
565eda14cbcSMatt Macy 	 * avoiding lock order reversals and limiting stack depth.
566eda14cbcSMatt Macy 	 *
567eda14cbcSMatt Macy 	 * Note that if we have a sync callback but no async callback,
568eda14cbcSMatt Macy 	 * it's likely that the sync callback will free the structure
569eda14cbcSMatt Macy 	 * containing the dbu.  In that case we need to take care to not
570eda14cbcSMatt Macy 	 * dereference dbu after calling the sync evict func.
571eda14cbcSMatt Macy 	 */
572eda14cbcSMatt Macy 	boolean_t has_async = (dbu->dbu_evict_func_async != NULL);
573eda14cbcSMatt Macy 
574eda14cbcSMatt Macy 	if (dbu->dbu_evict_func_sync != NULL)
575eda14cbcSMatt Macy 		dbu->dbu_evict_func_sync(dbu);
576eda14cbcSMatt Macy 
577eda14cbcSMatt Macy 	if (has_async) {
578eda14cbcSMatt Macy 		taskq_dispatch_ent(dbu_evict_taskq, dbu->dbu_evict_func_async,
579eda14cbcSMatt Macy 		    dbu, 0, &dbu->dbu_tqent);
580eda14cbcSMatt Macy 	}
581eda14cbcSMatt Macy }
582eda14cbcSMatt Macy 
583eda14cbcSMatt Macy boolean_t
584eda14cbcSMatt Macy dbuf_is_metadata(dmu_buf_impl_t *db)
585eda14cbcSMatt Macy {
586eda14cbcSMatt Macy 	/*
587eda14cbcSMatt Macy 	 * Consider indirect blocks and spill blocks to be meta data.
588eda14cbcSMatt Macy 	 */
589eda14cbcSMatt Macy 	if (db->db_level > 0 || db->db_blkid == DMU_SPILL_BLKID) {
590eda14cbcSMatt Macy 		return (B_TRUE);
591eda14cbcSMatt Macy 	} else {
592eda14cbcSMatt Macy 		boolean_t is_metadata;
593eda14cbcSMatt Macy 
594eda14cbcSMatt Macy 		DB_DNODE_ENTER(db);
595eda14cbcSMatt Macy 		is_metadata = DMU_OT_IS_METADATA(DB_DNODE(db)->dn_type);
596eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
597eda14cbcSMatt Macy 
598eda14cbcSMatt Macy 		return (is_metadata);
599eda14cbcSMatt Macy 	}
600eda14cbcSMatt Macy }
601eda14cbcSMatt Macy 
602eda14cbcSMatt Macy 
603eda14cbcSMatt Macy /*
604eda14cbcSMatt Macy  * This function *must* return indices evenly distributed between all
605eda14cbcSMatt Macy  * sublists of the multilist. This is needed due to how the dbuf eviction
606eda14cbcSMatt Macy  * code is laid out; dbuf_evict_thread() assumes dbufs are evenly
607eda14cbcSMatt Macy  * distributed between all sublists and uses this assumption when
608eda14cbcSMatt Macy  * deciding which sublist to evict from and how much to evict from it.
609eda14cbcSMatt Macy  */
610eda14cbcSMatt Macy static unsigned int
611eda14cbcSMatt Macy dbuf_cache_multilist_index_func(multilist_t *ml, void *obj)
612eda14cbcSMatt Macy {
613eda14cbcSMatt Macy 	dmu_buf_impl_t *db = obj;
614eda14cbcSMatt Macy 
615eda14cbcSMatt Macy 	/*
616eda14cbcSMatt Macy 	 * The assumption here, is the hash value for a given
617eda14cbcSMatt Macy 	 * dmu_buf_impl_t will remain constant throughout it's lifetime
618eda14cbcSMatt Macy 	 * (i.e. it's objset, object, level and blkid fields don't change).
619eda14cbcSMatt Macy 	 * Thus, we don't need to store the dbuf's sublist index
620eda14cbcSMatt Macy 	 * on insertion, as this index can be recalculated on removal.
621eda14cbcSMatt Macy 	 *
622eda14cbcSMatt Macy 	 * Also, the low order bits of the hash value are thought to be
623eda14cbcSMatt Macy 	 * distributed evenly. Otherwise, in the case that the multilist
624eda14cbcSMatt Macy 	 * has a power of two number of sublists, each sublists' usage
625eda14cbcSMatt Macy 	 * would not be evenly distributed.
626eda14cbcSMatt Macy 	 */
627eda14cbcSMatt Macy 	return (dbuf_hash(db->db_objset, db->db.db_object,
628eda14cbcSMatt Macy 	    db->db_level, db->db_blkid) %
629eda14cbcSMatt Macy 	    multilist_get_num_sublists(ml));
630eda14cbcSMatt Macy }
631eda14cbcSMatt Macy 
632eda14cbcSMatt Macy /*
633eda14cbcSMatt Macy  * The target size of the dbuf cache can grow with the ARC target,
634eda14cbcSMatt Macy  * unless limited by the tunable dbuf_cache_max_bytes.
635eda14cbcSMatt Macy  */
636eda14cbcSMatt Macy static inline unsigned long
637eda14cbcSMatt Macy dbuf_cache_target_bytes(void)
638eda14cbcSMatt Macy {
639eda14cbcSMatt Macy 	return (MIN(dbuf_cache_max_bytes,
640eda14cbcSMatt Macy 	    arc_target_bytes() >> dbuf_cache_shift));
641eda14cbcSMatt Macy }
642eda14cbcSMatt Macy 
643eda14cbcSMatt Macy /*
644eda14cbcSMatt Macy  * The target size of the dbuf metadata cache can grow with the ARC target,
645eda14cbcSMatt Macy  * unless limited by the tunable dbuf_metadata_cache_max_bytes.
646eda14cbcSMatt Macy  */
647eda14cbcSMatt Macy static inline unsigned long
648eda14cbcSMatt Macy dbuf_metadata_cache_target_bytes(void)
649eda14cbcSMatt Macy {
650eda14cbcSMatt Macy 	return (MIN(dbuf_metadata_cache_max_bytes,
651eda14cbcSMatt Macy 	    arc_target_bytes() >> dbuf_metadata_cache_shift));
652eda14cbcSMatt Macy }
653eda14cbcSMatt Macy 
654eda14cbcSMatt Macy static inline uint64_t
655eda14cbcSMatt Macy dbuf_cache_hiwater_bytes(void)
656eda14cbcSMatt Macy {
657eda14cbcSMatt Macy 	uint64_t dbuf_cache_target = dbuf_cache_target_bytes();
658eda14cbcSMatt Macy 	return (dbuf_cache_target +
659eda14cbcSMatt Macy 	    (dbuf_cache_target * dbuf_cache_hiwater_pct) / 100);
660eda14cbcSMatt Macy }
661eda14cbcSMatt Macy 
662eda14cbcSMatt Macy static inline uint64_t
663eda14cbcSMatt Macy dbuf_cache_lowater_bytes(void)
664eda14cbcSMatt Macy {
665eda14cbcSMatt Macy 	uint64_t dbuf_cache_target = dbuf_cache_target_bytes();
666eda14cbcSMatt Macy 	return (dbuf_cache_target -
667eda14cbcSMatt Macy 	    (dbuf_cache_target * dbuf_cache_lowater_pct) / 100);
668eda14cbcSMatt Macy }
669eda14cbcSMatt Macy 
670eda14cbcSMatt Macy static inline boolean_t
671eda14cbcSMatt Macy dbuf_cache_above_lowater(void)
672eda14cbcSMatt Macy {
673eda14cbcSMatt Macy 	return (zfs_refcount_count(&dbuf_caches[DB_DBUF_CACHE].size) >
674eda14cbcSMatt Macy 	    dbuf_cache_lowater_bytes());
675eda14cbcSMatt Macy }
676eda14cbcSMatt Macy 
677eda14cbcSMatt Macy /*
678eda14cbcSMatt Macy  * Evict the oldest eligible dbuf from the dbuf cache.
679eda14cbcSMatt Macy  */
680eda14cbcSMatt Macy static void
681eda14cbcSMatt Macy dbuf_evict_one(void)
682eda14cbcSMatt Macy {
6833ff01b23SMartin Matuska 	int idx = multilist_get_random_index(&dbuf_caches[DB_DBUF_CACHE].cache);
684eda14cbcSMatt Macy 	multilist_sublist_t *mls = multilist_sublist_lock(
6853ff01b23SMartin Matuska 	    &dbuf_caches[DB_DBUF_CACHE].cache, idx);
686eda14cbcSMatt Macy 
687eda14cbcSMatt Macy 	ASSERT(!MUTEX_HELD(&dbuf_evict_lock));
688eda14cbcSMatt Macy 
689eda14cbcSMatt Macy 	dmu_buf_impl_t *db = multilist_sublist_tail(mls);
690eda14cbcSMatt Macy 	while (db != NULL && mutex_tryenter(&db->db_mtx) == 0) {
691eda14cbcSMatt Macy 		db = multilist_sublist_prev(mls, db);
692eda14cbcSMatt Macy 	}
693eda14cbcSMatt Macy 
694eda14cbcSMatt Macy 	DTRACE_PROBE2(dbuf__evict__one, dmu_buf_impl_t *, db,
695eda14cbcSMatt Macy 	    multilist_sublist_t *, mls);
696eda14cbcSMatt Macy 
697eda14cbcSMatt Macy 	if (db != NULL) {
698eda14cbcSMatt Macy 		multilist_sublist_remove(mls, db);
699eda14cbcSMatt Macy 		multilist_sublist_unlock(mls);
700eda14cbcSMatt Macy 		(void) zfs_refcount_remove_many(
701eda14cbcSMatt Macy 		    &dbuf_caches[DB_DBUF_CACHE].size, db->db.db_size, db);
702eda14cbcSMatt Macy 		DBUF_STAT_BUMPDOWN(cache_levels[db->db_level]);
703eda14cbcSMatt Macy 		DBUF_STAT_BUMPDOWN(cache_count);
704eda14cbcSMatt Macy 		DBUF_STAT_DECR(cache_levels_bytes[db->db_level],
705eda14cbcSMatt Macy 		    db->db.db_size);
706eda14cbcSMatt Macy 		ASSERT3U(db->db_caching_status, ==, DB_DBUF_CACHE);
707eda14cbcSMatt Macy 		db->db_caching_status = DB_NO_CACHE;
708eda14cbcSMatt Macy 		dbuf_destroy(db);
709eda14cbcSMatt Macy 		DBUF_STAT_BUMP(cache_total_evicts);
710eda14cbcSMatt Macy 	} else {
711eda14cbcSMatt Macy 		multilist_sublist_unlock(mls);
712eda14cbcSMatt Macy 	}
713eda14cbcSMatt Macy }
714eda14cbcSMatt Macy 
715eda14cbcSMatt Macy /*
716eda14cbcSMatt Macy  * The dbuf evict thread is responsible for aging out dbufs from the
717eda14cbcSMatt Macy  * cache. Once the cache has reached it's maximum size, dbufs are removed
718eda14cbcSMatt Macy  * and destroyed. The eviction thread will continue running until the size
719eda14cbcSMatt Macy  * of the dbuf cache is at or below the maximum size. Once the dbuf is aged
720eda14cbcSMatt Macy  * out of the cache it is destroyed and becomes eligible for arc eviction.
721eda14cbcSMatt Macy  */
722eda14cbcSMatt Macy /* ARGSUSED */
723eda14cbcSMatt Macy static void
724eda14cbcSMatt Macy dbuf_evict_thread(void *unused)
725eda14cbcSMatt Macy {
726eda14cbcSMatt Macy 	callb_cpr_t cpr;
727eda14cbcSMatt Macy 
728eda14cbcSMatt Macy 	CALLB_CPR_INIT(&cpr, &dbuf_evict_lock, callb_generic_cpr, FTAG);
729eda14cbcSMatt Macy 
730eda14cbcSMatt Macy 	mutex_enter(&dbuf_evict_lock);
731eda14cbcSMatt Macy 	while (!dbuf_evict_thread_exit) {
732eda14cbcSMatt Macy 		while (!dbuf_cache_above_lowater() && !dbuf_evict_thread_exit) {
733eda14cbcSMatt Macy 			CALLB_CPR_SAFE_BEGIN(&cpr);
7342c48331dSMatt Macy 			(void) cv_timedwait_idle_hires(&dbuf_evict_cv,
735eda14cbcSMatt Macy 			    &dbuf_evict_lock, SEC2NSEC(1), MSEC2NSEC(1), 0);
736eda14cbcSMatt Macy 			CALLB_CPR_SAFE_END(&cpr, &dbuf_evict_lock);
737eda14cbcSMatt Macy 		}
738eda14cbcSMatt Macy 		mutex_exit(&dbuf_evict_lock);
739eda14cbcSMatt Macy 
740eda14cbcSMatt Macy 		/*
741eda14cbcSMatt Macy 		 * Keep evicting as long as we're above the low water mark
742eda14cbcSMatt Macy 		 * for the cache. We do this without holding the locks to
743eda14cbcSMatt Macy 		 * minimize lock contention.
744eda14cbcSMatt Macy 		 */
745eda14cbcSMatt Macy 		while (dbuf_cache_above_lowater() && !dbuf_evict_thread_exit) {
746eda14cbcSMatt Macy 			dbuf_evict_one();
747eda14cbcSMatt Macy 		}
748eda14cbcSMatt Macy 
749eda14cbcSMatt Macy 		mutex_enter(&dbuf_evict_lock);
750eda14cbcSMatt Macy 	}
751eda14cbcSMatt Macy 
752eda14cbcSMatt Macy 	dbuf_evict_thread_exit = B_FALSE;
753eda14cbcSMatt Macy 	cv_broadcast(&dbuf_evict_cv);
754eda14cbcSMatt Macy 	CALLB_CPR_EXIT(&cpr);	/* drops dbuf_evict_lock */
755eda14cbcSMatt Macy 	thread_exit();
756eda14cbcSMatt Macy }
757eda14cbcSMatt Macy 
758eda14cbcSMatt Macy /*
759eda14cbcSMatt Macy  * Wake up the dbuf eviction thread if the dbuf cache is at its max size.
760eda14cbcSMatt Macy  * If the dbuf cache is at its high water mark, then evict a dbuf from the
761eda14cbcSMatt Macy  * dbuf cache using the callers context.
762eda14cbcSMatt Macy  */
763eda14cbcSMatt Macy static void
764eda14cbcSMatt Macy dbuf_evict_notify(uint64_t size)
765eda14cbcSMatt Macy {
766eda14cbcSMatt Macy 	/*
767eda14cbcSMatt Macy 	 * We check if we should evict without holding the dbuf_evict_lock,
768eda14cbcSMatt Macy 	 * because it's OK to occasionally make the wrong decision here,
769eda14cbcSMatt Macy 	 * and grabbing the lock results in massive lock contention.
770eda14cbcSMatt Macy 	 */
771eda14cbcSMatt Macy 	if (size > dbuf_cache_target_bytes()) {
772eda14cbcSMatt Macy 		if (size > dbuf_cache_hiwater_bytes())
773eda14cbcSMatt Macy 			dbuf_evict_one();
774eda14cbcSMatt Macy 		cv_signal(&dbuf_evict_cv);
775eda14cbcSMatt Macy 	}
776eda14cbcSMatt Macy }
777eda14cbcSMatt Macy 
778eda14cbcSMatt Macy static int
779eda14cbcSMatt Macy dbuf_kstat_update(kstat_t *ksp, int rw)
780eda14cbcSMatt Macy {
781eda14cbcSMatt Macy 	dbuf_stats_t *ds = ksp->ks_data;
782eda14cbcSMatt Macy 
783*0d8fe237SMartin Matuska 	if (rw == KSTAT_WRITE)
784eda14cbcSMatt Macy 		return (SET_ERROR(EACCES));
785*0d8fe237SMartin Matuska 
786*0d8fe237SMartin Matuska 	ds->cache_count.value.ui64 =
787*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.cache_count);
788eda14cbcSMatt Macy 	ds->cache_size_bytes.value.ui64 =
789eda14cbcSMatt Macy 	    zfs_refcount_count(&dbuf_caches[DB_DBUF_CACHE].size);
790eda14cbcSMatt Macy 	ds->cache_target_bytes.value.ui64 = dbuf_cache_target_bytes();
791eda14cbcSMatt Macy 	ds->cache_hiwater_bytes.value.ui64 = dbuf_cache_hiwater_bytes();
792eda14cbcSMatt Macy 	ds->cache_lowater_bytes.value.ui64 = dbuf_cache_lowater_bytes();
793*0d8fe237SMartin Matuska 	ds->cache_total_evicts.value.ui64 =
794*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.cache_total_evicts);
795*0d8fe237SMartin Matuska 	for (int i = 0; i < DN_MAX_LEVELS; i++) {
796*0d8fe237SMartin Matuska 		ds->cache_levels[i].value.ui64 =
797*0d8fe237SMartin Matuska 		    wmsum_value(&dbuf_sums.cache_levels[i]);
798*0d8fe237SMartin Matuska 		ds->cache_levels_bytes[i].value.ui64 =
799*0d8fe237SMartin Matuska 		    wmsum_value(&dbuf_sums.cache_levels_bytes[i]);
800eda14cbcSMatt Macy 	}
801*0d8fe237SMartin Matuska 	ds->hash_hits.value.ui64 =
802*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.hash_hits);
803*0d8fe237SMartin Matuska 	ds->hash_misses.value.ui64 =
804*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.hash_misses);
805*0d8fe237SMartin Matuska 	ds->hash_collisions.value.ui64 =
806*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.hash_collisions);
807*0d8fe237SMartin Matuska 	ds->hash_chains.value.ui64 =
808*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.hash_chains);
809*0d8fe237SMartin Matuska 	ds->hash_insert_race.value.ui64 =
810*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.hash_insert_race);
811*0d8fe237SMartin Matuska 	ds->metadata_cache_count.value.ui64 =
812*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.metadata_cache_count);
813*0d8fe237SMartin Matuska 	ds->metadata_cache_size_bytes.value.ui64 = zfs_refcount_count(
814*0d8fe237SMartin Matuska 	    &dbuf_caches[DB_DBUF_METADATA_CACHE].size);
815*0d8fe237SMartin Matuska 	ds->metadata_cache_overflow.value.ui64 =
816*0d8fe237SMartin Matuska 	    wmsum_value(&dbuf_sums.metadata_cache_overflow);
817eda14cbcSMatt Macy 	return (0);
818eda14cbcSMatt Macy }
819eda14cbcSMatt Macy 
820eda14cbcSMatt Macy void
821eda14cbcSMatt Macy dbuf_init(void)
822eda14cbcSMatt Macy {
823eda14cbcSMatt Macy 	uint64_t hsize = 1ULL << 16;
824eda14cbcSMatt Macy 	dbuf_hash_table_t *h = &dbuf_hash_table;
825eda14cbcSMatt Macy 	int i;
826eda14cbcSMatt Macy 
827eda14cbcSMatt Macy 	/*
828eda14cbcSMatt Macy 	 * The hash table is big enough to fill all of physical memory
829eda14cbcSMatt Macy 	 * with an average block size of zfs_arc_average_blocksize (default 8K).
830eda14cbcSMatt Macy 	 * By default, the table will take up
831eda14cbcSMatt Macy 	 * totalmem * sizeof(void*) / 8K (1MB per GB with 8-byte pointers).
832eda14cbcSMatt Macy 	 */
833eda14cbcSMatt Macy 	while (hsize * zfs_arc_average_blocksize < physmem * PAGESIZE)
834eda14cbcSMatt Macy 		hsize <<= 1;
835eda14cbcSMatt Macy 
836eda14cbcSMatt Macy retry:
837eda14cbcSMatt Macy 	h->hash_table_mask = hsize - 1;
838eda14cbcSMatt Macy #if defined(_KERNEL)
839eda14cbcSMatt Macy 	/*
840eda14cbcSMatt Macy 	 * Large allocations which do not require contiguous pages
841eda14cbcSMatt Macy 	 * should be using vmem_alloc() in the linux kernel
842eda14cbcSMatt Macy 	 */
843eda14cbcSMatt Macy 	h->hash_table = vmem_zalloc(hsize * sizeof (void *), KM_SLEEP);
844eda14cbcSMatt Macy #else
845eda14cbcSMatt Macy 	h->hash_table = kmem_zalloc(hsize * sizeof (void *), KM_NOSLEEP);
846eda14cbcSMatt Macy #endif
847eda14cbcSMatt Macy 	if (h->hash_table == NULL) {
848eda14cbcSMatt Macy 		/* XXX - we should really return an error instead of assert */
849eda14cbcSMatt Macy 		ASSERT(hsize > (1ULL << 10));
850eda14cbcSMatt Macy 		hsize >>= 1;
851eda14cbcSMatt Macy 		goto retry;
852eda14cbcSMatt Macy 	}
853eda14cbcSMatt Macy 
854eda14cbcSMatt Macy 	dbuf_kmem_cache = kmem_cache_create("dmu_buf_impl_t",
855eda14cbcSMatt Macy 	    sizeof (dmu_buf_impl_t),
856eda14cbcSMatt Macy 	    0, dbuf_cons, dbuf_dest, NULL, NULL, NULL, 0);
857eda14cbcSMatt Macy 
858eda14cbcSMatt Macy 	for (i = 0; i < DBUF_MUTEXES; i++)
859eda14cbcSMatt Macy 		mutex_init(&h->hash_mutexes[i], NULL, MUTEX_DEFAULT, NULL);
860eda14cbcSMatt Macy 
861eda14cbcSMatt Macy 	dbuf_stats_init(h);
862eda14cbcSMatt Macy 
863eda14cbcSMatt Macy 	/*
864eda14cbcSMatt Macy 	 * All entries are queued via taskq_dispatch_ent(), so min/maxalloc
865eda14cbcSMatt Macy 	 * configuration is not required.
866eda14cbcSMatt Macy 	 */
867eda14cbcSMatt Macy 	dbu_evict_taskq = taskq_create("dbu_evict", 1, defclsyspri, 0, 0, 0);
868eda14cbcSMatt Macy 
869eda14cbcSMatt Macy 	for (dbuf_cached_state_t dcs = 0; dcs < DB_CACHE_MAX; dcs++) {
8703ff01b23SMartin Matuska 		multilist_create(&dbuf_caches[dcs].cache,
8713ff01b23SMartin Matuska 		    sizeof (dmu_buf_impl_t),
872eda14cbcSMatt Macy 		    offsetof(dmu_buf_impl_t, db_cache_link),
873eda14cbcSMatt Macy 		    dbuf_cache_multilist_index_func);
874eda14cbcSMatt Macy 		zfs_refcount_create(&dbuf_caches[dcs].size);
875eda14cbcSMatt Macy 	}
876eda14cbcSMatt Macy 
877eda14cbcSMatt Macy 	dbuf_evict_thread_exit = B_FALSE;
878eda14cbcSMatt Macy 	mutex_init(&dbuf_evict_lock, NULL, MUTEX_DEFAULT, NULL);
879eda14cbcSMatt Macy 	cv_init(&dbuf_evict_cv, NULL, CV_DEFAULT, NULL);
880eda14cbcSMatt Macy 	dbuf_cache_evict_thread = thread_create(NULL, 0, dbuf_evict_thread,
881eda14cbcSMatt Macy 	    NULL, 0, &p0, TS_RUN, minclsyspri);
882eda14cbcSMatt Macy 
883*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.cache_count, 0);
884*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.cache_total_evicts, 0);
885*0d8fe237SMartin Matuska 	for (i = 0; i < DN_MAX_LEVELS; i++) {
886*0d8fe237SMartin Matuska 		wmsum_init(&dbuf_sums.cache_levels[i], 0);
887*0d8fe237SMartin Matuska 		wmsum_init(&dbuf_sums.cache_levels_bytes[i], 0);
888*0d8fe237SMartin Matuska 	}
889*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.hash_hits, 0);
890*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.hash_misses, 0);
891*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.hash_collisions, 0);
892*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.hash_chains, 0);
893*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.hash_insert_race, 0);
894*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.metadata_cache_count, 0);
895*0d8fe237SMartin Matuska 	wmsum_init(&dbuf_sums.metadata_cache_overflow, 0);
896*0d8fe237SMartin Matuska 
897eda14cbcSMatt Macy 	dbuf_ksp = kstat_create("zfs", 0, "dbufstats", "misc",
898eda14cbcSMatt Macy 	    KSTAT_TYPE_NAMED, sizeof (dbuf_stats) / sizeof (kstat_named_t),
899eda14cbcSMatt Macy 	    KSTAT_FLAG_VIRTUAL);
900eda14cbcSMatt Macy 	if (dbuf_ksp != NULL) {
901eda14cbcSMatt Macy 		for (i = 0; i < DN_MAX_LEVELS; i++) {
902eda14cbcSMatt Macy 			snprintf(dbuf_stats.cache_levels[i].name,
903eda14cbcSMatt Macy 			    KSTAT_STRLEN, "cache_level_%d", i);
904eda14cbcSMatt Macy 			dbuf_stats.cache_levels[i].data_type =
905eda14cbcSMatt Macy 			    KSTAT_DATA_UINT64;
906eda14cbcSMatt Macy 			snprintf(dbuf_stats.cache_levels_bytes[i].name,
907eda14cbcSMatt Macy 			    KSTAT_STRLEN, "cache_level_%d_bytes", i);
908eda14cbcSMatt Macy 			dbuf_stats.cache_levels_bytes[i].data_type =
909eda14cbcSMatt Macy 			    KSTAT_DATA_UINT64;
910eda14cbcSMatt Macy 		}
911eda14cbcSMatt Macy 		dbuf_ksp->ks_data = &dbuf_stats;
912eda14cbcSMatt Macy 		dbuf_ksp->ks_update = dbuf_kstat_update;
913eda14cbcSMatt Macy 		kstat_install(dbuf_ksp);
914eda14cbcSMatt Macy 	}
915eda14cbcSMatt Macy }
916eda14cbcSMatt Macy 
917eda14cbcSMatt Macy void
918eda14cbcSMatt Macy dbuf_fini(void)
919eda14cbcSMatt Macy {
920eda14cbcSMatt Macy 	dbuf_hash_table_t *h = &dbuf_hash_table;
921eda14cbcSMatt Macy 	int i;
922eda14cbcSMatt Macy 
923eda14cbcSMatt Macy 	dbuf_stats_destroy();
924eda14cbcSMatt Macy 
925eda14cbcSMatt Macy 	for (i = 0; i < DBUF_MUTEXES; i++)
926eda14cbcSMatt Macy 		mutex_destroy(&h->hash_mutexes[i]);
927eda14cbcSMatt Macy #if defined(_KERNEL)
928eda14cbcSMatt Macy 	/*
929eda14cbcSMatt Macy 	 * Large allocations which do not require contiguous pages
930eda14cbcSMatt Macy 	 * should be using vmem_free() in the linux kernel
931eda14cbcSMatt Macy 	 */
932eda14cbcSMatt Macy 	vmem_free(h->hash_table, (h->hash_table_mask + 1) * sizeof (void *));
933eda14cbcSMatt Macy #else
934eda14cbcSMatt Macy 	kmem_free(h->hash_table, (h->hash_table_mask + 1) * sizeof (void *));
935eda14cbcSMatt Macy #endif
936eda14cbcSMatt Macy 	kmem_cache_destroy(dbuf_kmem_cache);
937eda14cbcSMatt Macy 	taskq_destroy(dbu_evict_taskq);
938eda14cbcSMatt Macy 
939eda14cbcSMatt Macy 	mutex_enter(&dbuf_evict_lock);
940eda14cbcSMatt Macy 	dbuf_evict_thread_exit = B_TRUE;
941eda14cbcSMatt Macy 	while (dbuf_evict_thread_exit) {
942eda14cbcSMatt Macy 		cv_signal(&dbuf_evict_cv);
943eda14cbcSMatt Macy 		cv_wait(&dbuf_evict_cv, &dbuf_evict_lock);
944eda14cbcSMatt Macy 	}
945eda14cbcSMatt Macy 	mutex_exit(&dbuf_evict_lock);
946eda14cbcSMatt Macy 
947eda14cbcSMatt Macy 	mutex_destroy(&dbuf_evict_lock);
948eda14cbcSMatt Macy 	cv_destroy(&dbuf_evict_cv);
949eda14cbcSMatt Macy 
950eda14cbcSMatt Macy 	for (dbuf_cached_state_t dcs = 0; dcs < DB_CACHE_MAX; dcs++) {
951eda14cbcSMatt Macy 		zfs_refcount_destroy(&dbuf_caches[dcs].size);
9523ff01b23SMartin Matuska 		multilist_destroy(&dbuf_caches[dcs].cache);
953eda14cbcSMatt Macy 	}
954eda14cbcSMatt Macy 
955eda14cbcSMatt Macy 	if (dbuf_ksp != NULL) {
956eda14cbcSMatt Macy 		kstat_delete(dbuf_ksp);
957eda14cbcSMatt Macy 		dbuf_ksp = NULL;
958eda14cbcSMatt Macy 	}
959*0d8fe237SMartin Matuska 
960*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.cache_count);
961*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.cache_total_evicts);
962*0d8fe237SMartin Matuska 	for (i = 0; i < DN_MAX_LEVELS; i++) {
963*0d8fe237SMartin Matuska 		wmsum_fini(&dbuf_sums.cache_levels[i]);
964*0d8fe237SMartin Matuska 		wmsum_fini(&dbuf_sums.cache_levels_bytes[i]);
965*0d8fe237SMartin Matuska 	}
966*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.hash_hits);
967*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.hash_misses);
968*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.hash_collisions);
969*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.hash_chains);
970*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.hash_insert_race);
971*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.metadata_cache_count);
972*0d8fe237SMartin Matuska 	wmsum_fini(&dbuf_sums.metadata_cache_overflow);
973eda14cbcSMatt Macy }
974eda14cbcSMatt Macy 
975eda14cbcSMatt Macy /*
976eda14cbcSMatt Macy  * Other stuff.
977eda14cbcSMatt Macy  */
978eda14cbcSMatt Macy 
979eda14cbcSMatt Macy #ifdef ZFS_DEBUG
980eda14cbcSMatt Macy static void
981eda14cbcSMatt Macy dbuf_verify(dmu_buf_impl_t *db)
982eda14cbcSMatt Macy {
983eda14cbcSMatt Macy 	dnode_t *dn;
984eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
985eda14cbcSMatt Macy 	uint32_t txg_prev;
986eda14cbcSMatt Macy 
987eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
988eda14cbcSMatt Macy 
989eda14cbcSMatt Macy 	if (!(zfs_flags & ZFS_DEBUG_DBUF_VERIFY))
990eda14cbcSMatt Macy 		return;
991eda14cbcSMatt Macy 
992eda14cbcSMatt Macy 	ASSERT(db->db_objset != NULL);
993eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
994eda14cbcSMatt Macy 	dn = DB_DNODE(db);
995eda14cbcSMatt Macy 	if (dn == NULL) {
996eda14cbcSMatt Macy 		ASSERT(db->db_parent == NULL);
997eda14cbcSMatt Macy 		ASSERT(db->db_blkptr == NULL);
998eda14cbcSMatt Macy 	} else {
999eda14cbcSMatt Macy 		ASSERT3U(db->db.db_object, ==, dn->dn_object);
1000eda14cbcSMatt Macy 		ASSERT3P(db->db_objset, ==, dn->dn_objset);
1001eda14cbcSMatt Macy 		ASSERT3U(db->db_level, <, dn->dn_nlevels);
1002eda14cbcSMatt Macy 		ASSERT(db->db_blkid == DMU_BONUS_BLKID ||
1003eda14cbcSMatt Macy 		    db->db_blkid == DMU_SPILL_BLKID ||
1004eda14cbcSMatt Macy 		    !avl_is_empty(&dn->dn_dbufs));
1005eda14cbcSMatt Macy 	}
1006eda14cbcSMatt Macy 	if (db->db_blkid == DMU_BONUS_BLKID) {
1007eda14cbcSMatt Macy 		ASSERT(dn != NULL);
1008eda14cbcSMatt Macy 		ASSERT3U(db->db.db_size, >=, dn->dn_bonuslen);
1009eda14cbcSMatt Macy 		ASSERT3U(db->db.db_offset, ==, DMU_BONUS_BLKID);
1010eda14cbcSMatt Macy 	} else if (db->db_blkid == DMU_SPILL_BLKID) {
1011eda14cbcSMatt Macy 		ASSERT(dn != NULL);
1012eda14cbcSMatt Macy 		ASSERT0(db->db.db_offset);
1013eda14cbcSMatt Macy 	} else {
1014eda14cbcSMatt Macy 		ASSERT3U(db->db.db_offset, ==, db->db_blkid * db->db.db_size);
1015eda14cbcSMatt Macy 	}
1016eda14cbcSMatt Macy 
1017eda14cbcSMatt Macy 	if ((dr = list_head(&db->db_dirty_records)) != NULL) {
1018eda14cbcSMatt Macy 		ASSERT(dr->dr_dbuf == db);
1019eda14cbcSMatt Macy 		txg_prev = dr->dr_txg;
1020eda14cbcSMatt Macy 		for (dr = list_next(&db->db_dirty_records, dr); dr != NULL;
1021eda14cbcSMatt Macy 		    dr = list_next(&db->db_dirty_records, dr)) {
1022eda14cbcSMatt Macy 			ASSERT(dr->dr_dbuf == db);
1023eda14cbcSMatt Macy 			ASSERT(txg_prev > dr->dr_txg);
1024eda14cbcSMatt Macy 			txg_prev = dr->dr_txg;
1025eda14cbcSMatt Macy 		}
1026eda14cbcSMatt Macy 	}
1027eda14cbcSMatt Macy 
1028eda14cbcSMatt Macy 	/*
1029eda14cbcSMatt Macy 	 * We can't assert that db_size matches dn_datablksz because it
1030eda14cbcSMatt Macy 	 * can be momentarily different when another thread is doing
1031eda14cbcSMatt Macy 	 * dnode_set_blksz().
1032eda14cbcSMatt Macy 	 */
1033eda14cbcSMatt Macy 	if (db->db_level == 0 && db->db.db_object == DMU_META_DNODE_OBJECT) {
1034eda14cbcSMatt Macy 		dr = db->db_data_pending;
1035eda14cbcSMatt Macy 		/*
1036eda14cbcSMatt Macy 		 * It should only be modified in syncing context, so
1037eda14cbcSMatt Macy 		 * make sure we only have one copy of the data.
1038eda14cbcSMatt Macy 		 */
1039eda14cbcSMatt Macy 		ASSERT(dr == NULL || dr->dt.dl.dr_data == db->db_buf);
1040eda14cbcSMatt Macy 	}
1041eda14cbcSMatt Macy 
1042eda14cbcSMatt Macy 	/* verify db->db_blkptr */
1043eda14cbcSMatt Macy 	if (db->db_blkptr) {
1044eda14cbcSMatt Macy 		if (db->db_parent == dn->dn_dbuf) {
1045eda14cbcSMatt Macy 			/* db is pointed to by the dnode */
1046eda14cbcSMatt Macy 			/* ASSERT3U(db->db_blkid, <, dn->dn_nblkptr); */
1047eda14cbcSMatt Macy 			if (DMU_OBJECT_IS_SPECIAL(db->db.db_object))
1048eda14cbcSMatt Macy 				ASSERT(db->db_parent == NULL);
1049eda14cbcSMatt Macy 			else
1050eda14cbcSMatt Macy 				ASSERT(db->db_parent != NULL);
1051eda14cbcSMatt Macy 			if (db->db_blkid != DMU_SPILL_BLKID)
1052eda14cbcSMatt Macy 				ASSERT3P(db->db_blkptr, ==,
1053eda14cbcSMatt Macy 				    &dn->dn_phys->dn_blkptr[db->db_blkid]);
1054eda14cbcSMatt Macy 		} else {
1055eda14cbcSMatt Macy 			/* db is pointed to by an indirect block */
1056eda14cbcSMatt Macy 			int epb __maybe_unused = db->db_parent->db.db_size >>
1057eda14cbcSMatt Macy 			    SPA_BLKPTRSHIFT;
1058eda14cbcSMatt Macy 			ASSERT3U(db->db_parent->db_level, ==, db->db_level+1);
1059eda14cbcSMatt Macy 			ASSERT3U(db->db_parent->db.db_object, ==,
1060eda14cbcSMatt Macy 			    db->db.db_object);
1061eda14cbcSMatt Macy 			/*
1062eda14cbcSMatt Macy 			 * dnode_grow_indblksz() can make this fail if we don't
1063eda14cbcSMatt Macy 			 * have the parent's rwlock.  XXX indblksz no longer
1064eda14cbcSMatt Macy 			 * grows.  safe to do this now?
1065eda14cbcSMatt Macy 			 */
1066eda14cbcSMatt Macy 			if (RW_LOCK_HELD(&db->db_parent->db_rwlock)) {
1067eda14cbcSMatt Macy 				ASSERT3P(db->db_blkptr, ==,
1068eda14cbcSMatt Macy 				    ((blkptr_t *)db->db_parent->db.db_data +
1069eda14cbcSMatt Macy 				    db->db_blkid % epb));
1070eda14cbcSMatt Macy 			}
1071eda14cbcSMatt Macy 		}
1072eda14cbcSMatt Macy 	}
1073eda14cbcSMatt Macy 	if ((db->db_blkptr == NULL || BP_IS_HOLE(db->db_blkptr)) &&
1074eda14cbcSMatt Macy 	    (db->db_buf == NULL || db->db_buf->b_data) &&
1075eda14cbcSMatt Macy 	    db->db.db_data && db->db_blkid != DMU_BONUS_BLKID &&
1076eda14cbcSMatt Macy 	    db->db_state != DB_FILL && !dn->dn_free_txg) {
1077eda14cbcSMatt Macy 		/*
1078eda14cbcSMatt Macy 		 * If the blkptr isn't set but they have nonzero data,
1079eda14cbcSMatt Macy 		 * it had better be dirty, otherwise we'll lose that
1080eda14cbcSMatt Macy 		 * data when we evict this buffer.
1081eda14cbcSMatt Macy 		 *
1082eda14cbcSMatt Macy 		 * There is an exception to this rule for indirect blocks; in
1083eda14cbcSMatt Macy 		 * this case, if the indirect block is a hole, we fill in a few
1084eda14cbcSMatt Macy 		 * fields on each of the child blocks (importantly, birth time)
1085eda14cbcSMatt Macy 		 * to prevent hole birth times from being lost when you
1086eda14cbcSMatt Macy 		 * partially fill in a hole.
1087eda14cbcSMatt Macy 		 */
1088eda14cbcSMatt Macy 		if (db->db_dirtycnt == 0) {
1089eda14cbcSMatt Macy 			if (db->db_level == 0) {
1090eda14cbcSMatt Macy 				uint64_t *buf = db->db.db_data;
1091eda14cbcSMatt Macy 				int i;
1092eda14cbcSMatt Macy 
1093eda14cbcSMatt Macy 				for (i = 0; i < db->db.db_size >> 3; i++) {
1094eda14cbcSMatt Macy 					ASSERT(buf[i] == 0);
1095eda14cbcSMatt Macy 				}
1096eda14cbcSMatt Macy 			} else {
1097eda14cbcSMatt Macy 				blkptr_t *bps = db->db.db_data;
1098eda14cbcSMatt Macy 				ASSERT3U(1 << DB_DNODE(db)->dn_indblkshift, ==,
1099eda14cbcSMatt Macy 				    db->db.db_size);
1100eda14cbcSMatt Macy 				/*
1101eda14cbcSMatt Macy 				 * We want to verify that all the blkptrs in the
1102eda14cbcSMatt Macy 				 * indirect block are holes, but we may have
1103eda14cbcSMatt Macy 				 * automatically set up a few fields for them.
1104eda14cbcSMatt Macy 				 * We iterate through each blkptr and verify
1105eda14cbcSMatt Macy 				 * they only have those fields set.
1106eda14cbcSMatt Macy 				 */
1107eda14cbcSMatt Macy 				for (int i = 0;
1108eda14cbcSMatt Macy 				    i < db->db.db_size / sizeof (blkptr_t);
1109eda14cbcSMatt Macy 				    i++) {
1110eda14cbcSMatt Macy 					blkptr_t *bp = &bps[i];
1111eda14cbcSMatt Macy 					ASSERT(ZIO_CHECKSUM_IS_ZERO(
1112eda14cbcSMatt Macy 					    &bp->blk_cksum));
1113eda14cbcSMatt Macy 					ASSERT(
1114eda14cbcSMatt Macy 					    DVA_IS_EMPTY(&bp->blk_dva[0]) &&
1115eda14cbcSMatt Macy 					    DVA_IS_EMPTY(&bp->blk_dva[1]) &&
1116eda14cbcSMatt Macy 					    DVA_IS_EMPTY(&bp->blk_dva[2]));
1117eda14cbcSMatt Macy 					ASSERT0(bp->blk_fill);
1118eda14cbcSMatt Macy 					ASSERT0(bp->blk_pad[0]);
1119eda14cbcSMatt Macy 					ASSERT0(bp->blk_pad[1]);
1120eda14cbcSMatt Macy 					ASSERT(!BP_IS_EMBEDDED(bp));
1121eda14cbcSMatt Macy 					ASSERT(BP_IS_HOLE(bp));
1122eda14cbcSMatt Macy 					ASSERT0(bp->blk_phys_birth);
1123eda14cbcSMatt Macy 				}
1124eda14cbcSMatt Macy 			}
1125eda14cbcSMatt Macy 		}
1126eda14cbcSMatt Macy 	}
1127eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
1128eda14cbcSMatt Macy }
1129eda14cbcSMatt Macy #endif
1130eda14cbcSMatt Macy 
1131eda14cbcSMatt Macy static void
1132eda14cbcSMatt Macy dbuf_clear_data(dmu_buf_impl_t *db)
1133eda14cbcSMatt Macy {
1134eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1135eda14cbcSMatt Macy 	dbuf_evict_user(db);
1136eda14cbcSMatt Macy 	ASSERT3P(db->db_buf, ==, NULL);
1137eda14cbcSMatt Macy 	db->db.db_data = NULL;
1138eda14cbcSMatt Macy 	if (db->db_state != DB_NOFILL) {
1139eda14cbcSMatt Macy 		db->db_state = DB_UNCACHED;
1140eda14cbcSMatt Macy 		DTRACE_SET_STATE(db, "clear data");
1141eda14cbcSMatt Macy 	}
1142eda14cbcSMatt Macy }
1143eda14cbcSMatt Macy 
1144eda14cbcSMatt Macy static void
1145eda14cbcSMatt Macy dbuf_set_data(dmu_buf_impl_t *db, arc_buf_t *buf)
1146eda14cbcSMatt Macy {
1147eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1148eda14cbcSMatt Macy 	ASSERT(buf != NULL);
1149eda14cbcSMatt Macy 
1150eda14cbcSMatt Macy 	db->db_buf = buf;
1151eda14cbcSMatt Macy 	ASSERT(buf->b_data != NULL);
1152eda14cbcSMatt Macy 	db->db.db_data = buf->b_data;
1153eda14cbcSMatt Macy }
1154eda14cbcSMatt Macy 
1155eda14cbcSMatt Macy static arc_buf_t *
1156eda14cbcSMatt Macy dbuf_alloc_arcbuf_from_arcbuf(dmu_buf_impl_t *db, arc_buf_t *data)
1157eda14cbcSMatt Macy {
1158eda14cbcSMatt Macy 	objset_t *os = db->db_objset;
1159eda14cbcSMatt Macy 	spa_t *spa = os->os_spa;
1160eda14cbcSMatt Macy 	arc_buf_contents_t type = DBUF_GET_BUFC_TYPE(db);
1161eda14cbcSMatt Macy 	enum zio_compress compress_type;
1162eda14cbcSMatt Macy 	uint8_t complevel;
1163eda14cbcSMatt Macy 	int psize, lsize;
1164eda14cbcSMatt Macy 
1165eda14cbcSMatt Macy 	psize = arc_buf_size(data);
1166eda14cbcSMatt Macy 	lsize = arc_buf_lsize(data);
1167eda14cbcSMatt Macy 	compress_type = arc_get_compression(data);
1168eda14cbcSMatt Macy 	complevel = arc_get_complevel(data);
1169eda14cbcSMatt Macy 
1170eda14cbcSMatt Macy 	if (arc_is_encrypted(data)) {
1171eda14cbcSMatt Macy 		boolean_t byteorder;
1172eda14cbcSMatt Macy 		uint8_t salt[ZIO_DATA_SALT_LEN];
1173eda14cbcSMatt Macy 		uint8_t iv[ZIO_DATA_IV_LEN];
1174eda14cbcSMatt Macy 		uint8_t mac[ZIO_DATA_MAC_LEN];
1175eda14cbcSMatt Macy 		dnode_t *dn = DB_DNODE(db);
1176eda14cbcSMatt Macy 
1177eda14cbcSMatt Macy 		arc_get_raw_params(data, &byteorder, salt, iv, mac);
1178eda14cbcSMatt Macy 		data = arc_alloc_raw_buf(spa, db, dmu_objset_id(os),
1179eda14cbcSMatt Macy 		    byteorder, salt, iv, mac, dn->dn_type, psize, lsize,
1180eda14cbcSMatt Macy 		    compress_type, complevel);
1181eda14cbcSMatt Macy 	} else if (compress_type != ZIO_COMPRESS_OFF) {
1182eda14cbcSMatt Macy 		ASSERT3U(type, ==, ARC_BUFC_DATA);
1183eda14cbcSMatt Macy 		data = arc_alloc_compressed_buf(spa, db,
1184eda14cbcSMatt Macy 		    psize, lsize, compress_type, complevel);
1185eda14cbcSMatt Macy 	} else {
1186eda14cbcSMatt Macy 		data = arc_alloc_buf(spa, db, type, psize);
1187eda14cbcSMatt Macy 	}
1188eda14cbcSMatt Macy 	return (data);
1189eda14cbcSMatt Macy }
1190eda14cbcSMatt Macy 
1191eda14cbcSMatt Macy static arc_buf_t *
1192eda14cbcSMatt Macy dbuf_alloc_arcbuf(dmu_buf_impl_t *db)
1193eda14cbcSMatt Macy {
1194eda14cbcSMatt Macy 	spa_t *spa = db->db_objset->os_spa;
1195eda14cbcSMatt Macy 
1196eda14cbcSMatt Macy 	return (arc_alloc_buf(spa, db, DBUF_GET_BUFC_TYPE(db), db->db.db_size));
1197eda14cbcSMatt Macy }
1198eda14cbcSMatt Macy 
1199eda14cbcSMatt Macy /*
1200eda14cbcSMatt Macy  * Loan out an arc_buf for read.  Return the loaned arc_buf.
1201eda14cbcSMatt Macy  */
1202eda14cbcSMatt Macy arc_buf_t *
1203eda14cbcSMatt Macy dbuf_loan_arcbuf(dmu_buf_impl_t *db)
1204eda14cbcSMatt Macy {
1205eda14cbcSMatt Macy 	arc_buf_t *abuf;
1206eda14cbcSMatt Macy 
1207eda14cbcSMatt Macy 	ASSERT(db->db_blkid != DMU_BONUS_BLKID);
1208eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
1209eda14cbcSMatt Macy 	if (arc_released(db->db_buf) || zfs_refcount_count(&db->db_holds) > 1) {
1210eda14cbcSMatt Macy 		int blksz = db->db.db_size;
1211eda14cbcSMatt Macy 		spa_t *spa = db->db_objset->os_spa;
1212eda14cbcSMatt Macy 
1213eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
1214eda14cbcSMatt Macy 		abuf = arc_loan_buf(spa, B_FALSE, blksz);
1215eda14cbcSMatt Macy 		bcopy(db->db.db_data, abuf->b_data, blksz);
1216eda14cbcSMatt Macy 	} else {
1217eda14cbcSMatt Macy 		abuf = db->db_buf;
1218eda14cbcSMatt Macy 		arc_loan_inuse_buf(abuf, db);
1219eda14cbcSMatt Macy 		db->db_buf = NULL;
1220eda14cbcSMatt Macy 		dbuf_clear_data(db);
1221eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
1222eda14cbcSMatt Macy 	}
1223eda14cbcSMatt Macy 	return (abuf);
1224eda14cbcSMatt Macy }
1225eda14cbcSMatt Macy 
1226eda14cbcSMatt Macy /*
1227eda14cbcSMatt Macy  * Calculate which level n block references the data at the level 0 offset
1228eda14cbcSMatt Macy  * provided.
1229eda14cbcSMatt Macy  */
1230eda14cbcSMatt Macy uint64_t
1231eda14cbcSMatt Macy dbuf_whichblock(const dnode_t *dn, const int64_t level, const uint64_t offset)
1232eda14cbcSMatt Macy {
1233eda14cbcSMatt Macy 	if (dn->dn_datablkshift != 0 && dn->dn_indblkshift != 0) {
1234eda14cbcSMatt Macy 		/*
1235eda14cbcSMatt Macy 		 * The level n blkid is equal to the level 0 blkid divided by
1236eda14cbcSMatt Macy 		 * the number of level 0s in a level n block.
1237eda14cbcSMatt Macy 		 *
1238eda14cbcSMatt Macy 		 * The level 0 blkid is offset >> datablkshift =
1239eda14cbcSMatt Macy 		 * offset / 2^datablkshift.
1240eda14cbcSMatt Macy 		 *
1241eda14cbcSMatt Macy 		 * The number of level 0s in a level n is the number of block
1242eda14cbcSMatt Macy 		 * pointers in an indirect block, raised to the power of level.
1243eda14cbcSMatt Macy 		 * This is 2^(indblkshift - SPA_BLKPTRSHIFT)^level =
1244eda14cbcSMatt Macy 		 * 2^(level*(indblkshift - SPA_BLKPTRSHIFT)).
1245eda14cbcSMatt Macy 		 *
1246eda14cbcSMatt Macy 		 * Thus, the level n blkid is: offset /
1247eda14cbcSMatt Macy 		 * ((2^datablkshift)*(2^(level*(indblkshift-SPA_BLKPTRSHIFT))))
1248eda14cbcSMatt Macy 		 * = offset / 2^(datablkshift + level *
1249eda14cbcSMatt Macy 		 *   (indblkshift - SPA_BLKPTRSHIFT))
1250eda14cbcSMatt Macy 		 * = offset >> (datablkshift + level *
1251eda14cbcSMatt Macy 		 *   (indblkshift - SPA_BLKPTRSHIFT))
1252eda14cbcSMatt Macy 		 */
1253eda14cbcSMatt Macy 
1254eda14cbcSMatt Macy 		const unsigned exp = dn->dn_datablkshift +
1255eda14cbcSMatt Macy 		    level * (dn->dn_indblkshift - SPA_BLKPTRSHIFT);
1256eda14cbcSMatt Macy 
1257eda14cbcSMatt Macy 		if (exp >= 8 * sizeof (offset)) {
1258eda14cbcSMatt Macy 			/* This only happens on the highest indirection level */
1259eda14cbcSMatt Macy 			ASSERT3U(level, ==, dn->dn_nlevels - 1);
1260eda14cbcSMatt Macy 			return (0);
1261eda14cbcSMatt Macy 		}
1262eda14cbcSMatt Macy 
1263eda14cbcSMatt Macy 		ASSERT3U(exp, <, 8 * sizeof (offset));
1264eda14cbcSMatt Macy 
1265eda14cbcSMatt Macy 		return (offset >> exp);
1266eda14cbcSMatt Macy 	} else {
1267eda14cbcSMatt Macy 		ASSERT3U(offset, <, dn->dn_datablksz);
1268eda14cbcSMatt Macy 		return (0);
1269eda14cbcSMatt Macy 	}
1270eda14cbcSMatt Macy }
1271eda14cbcSMatt Macy 
1272eda14cbcSMatt Macy /*
1273eda14cbcSMatt Macy  * This function is used to lock the parent of the provided dbuf. This should be
1274eda14cbcSMatt Macy  * used when modifying or reading db_blkptr.
1275eda14cbcSMatt Macy  */
1276eda14cbcSMatt Macy db_lock_type_t
1277eda14cbcSMatt Macy dmu_buf_lock_parent(dmu_buf_impl_t *db, krw_t rw, void *tag)
1278eda14cbcSMatt Macy {
1279eda14cbcSMatt Macy 	enum db_lock_type ret = DLT_NONE;
1280eda14cbcSMatt Macy 	if (db->db_parent != NULL) {
1281eda14cbcSMatt Macy 		rw_enter(&db->db_parent->db_rwlock, rw);
1282eda14cbcSMatt Macy 		ret = DLT_PARENT;
1283eda14cbcSMatt Macy 	} else if (dmu_objset_ds(db->db_objset) != NULL) {
1284eda14cbcSMatt Macy 		rrw_enter(&dmu_objset_ds(db->db_objset)->ds_bp_rwlock, rw,
1285eda14cbcSMatt Macy 		    tag);
1286eda14cbcSMatt Macy 		ret = DLT_OBJSET;
1287eda14cbcSMatt Macy 	}
1288eda14cbcSMatt Macy 	/*
1289eda14cbcSMatt Macy 	 * We only return a DLT_NONE lock when it's the top-most indirect block
1290eda14cbcSMatt Macy 	 * of the meta-dnode of the MOS.
1291eda14cbcSMatt Macy 	 */
1292eda14cbcSMatt Macy 	return (ret);
1293eda14cbcSMatt Macy }
1294eda14cbcSMatt Macy 
1295eda14cbcSMatt Macy /*
1296eda14cbcSMatt Macy  * We need to pass the lock type in because it's possible that the block will
1297eda14cbcSMatt Macy  * move from being the topmost indirect block in a dnode (and thus, have no
1298eda14cbcSMatt Macy  * parent) to not the top-most via an indirection increase. This would cause a
1299eda14cbcSMatt Macy  * panic if we didn't pass the lock type in.
1300eda14cbcSMatt Macy  */
1301eda14cbcSMatt Macy void
1302eda14cbcSMatt Macy dmu_buf_unlock_parent(dmu_buf_impl_t *db, db_lock_type_t type, void *tag)
1303eda14cbcSMatt Macy {
1304eda14cbcSMatt Macy 	if (type == DLT_PARENT)
1305eda14cbcSMatt Macy 		rw_exit(&db->db_parent->db_rwlock);
1306eda14cbcSMatt Macy 	else if (type == DLT_OBJSET)
1307eda14cbcSMatt Macy 		rrw_exit(&dmu_objset_ds(db->db_objset)->ds_bp_rwlock, tag);
1308eda14cbcSMatt Macy }
1309eda14cbcSMatt Macy 
1310eda14cbcSMatt Macy static void
1311eda14cbcSMatt Macy dbuf_read_done(zio_t *zio, const zbookmark_phys_t *zb, const blkptr_t *bp,
1312eda14cbcSMatt Macy     arc_buf_t *buf, void *vdb)
1313eda14cbcSMatt Macy {
1314eda14cbcSMatt Macy 	dmu_buf_impl_t *db = vdb;
1315eda14cbcSMatt Macy 
1316eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
1317eda14cbcSMatt Macy 	ASSERT3U(db->db_state, ==, DB_READ);
1318eda14cbcSMatt Macy 	/*
1319eda14cbcSMatt Macy 	 * All reads are synchronous, so we must have a hold on the dbuf
1320eda14cbcSMatt Macy 	 */
1321eda14cbcSMatt Macy 	ASSERT(zfs_refcount_count(&db->db_holds) > 0);
1322eda14cbcSMatt Macy 	ASSERT(db->db_buf == NULL);
1323eda14cbcSMatt Macy 	ASSERT(db->db.db_data == NULL);
1324eda14cbcSMatt Macy 	if (buf == NULL) {
1325eda14cbcSMatt Macy 		/* i/o error */
1326eda14cbcSMatt Macy 		ASSERT(zio == NULL || zio->io_error != 0);
1327eda14cbcSMatt Macy 		ASSERT(db->db_blkid != DMU_BONUS_BLKID);
1328eda14cbcSMatt Macy 		ASSERT3P(db->db_buf, ==, NULL);
1329eda14cbcSMatt Macy 		db->db_state = DB_UNCACHED;
1330eda14cbcSMatt Macy 		DTRACE_SET_STATE(db, "i/o error");
1331eda14cbcSMatt Macy 	} else if (db->db_level == 0 && db->db_freed_in_flight) {
1332eda14cbcSMatt Macy 		/* freed in flight */
1333eda14cbcSMatt Macy 		ASSERT(zio == NULL || zio->io_error == 0);
1334eda14cbcSMatt Macy 		arc_release(buf, db);
1335eda14cbcSMatt Macy 		bzero(buf->b_data, db->db.db_size);
1336eda14cbcSMatt Macy 		arc_buf_freeze(buf);
1337eda14cbcSMatt Macy 		db->db_freed_in_flight = FALSE;
1338eda14cbcSMatt Macy 		dbuf_set_data(db, buf);
1339eda14cbcSMatt Macy 		db->db_state = DB_CACHED;
1340eda14cbcSMatt Macy 		DTRACE_SET_STATE(db, "freed in flight");
1341eda14cbcSMatt Macy 	} else {
1342eda14cbcSMatt Macy 		/* success */
1343eda14cbcSMatt Macy 		ASSERT(zio == NULL || zio->io_error == 0);
1344eda14cbcSMatt Macy 		dbuf_set_data(db, buf);
1345eda14cbcSMatt Macy 		db->db_state = DB_CACHED;
1346eda14cbcSMatt Macy 		DTRACE_SET_STATE(db, "successful read");
1347eda14cbcSMatt Macy 	}
1348eda14cbcSMatt Macy 	cv_broadcast(&db->db_changed);
1349eda14cbcSMatt Macy 	dbuf_rele_and_unlock(db, NULL, B_FALSE);
1350eda14cbcSMatt Macy }
1351eda14cbcSMatt Macy 
1352eda14cbcSMatt Macy /*
1353eda14cbcSMatt Macy  * Shortcut for performing reads on bonus dbufs.  Returns
1354eda14cbcSMatt Macy  * an error if we fail to verify the dnode associated with
1355eda14cbcSMatt Macy  * a decrypted block. Otherwise success.
1356eda14cbcSMatt Macy  */
1357eda14cbcSMatt Macy static int
1358eda14cbcSMatt Macy dbuf_read_bonus(dmu_buf_impl_t *db, dnode_t *dn, uint32_t flags)
1359eda14cbcSMatt Macy {
1360eda14cbcSMatt Macy 	int bonuslen, max_bonuslen, err;
1361eda14cbcSMatt Macy 
1362eda14cbcSMatt Macy 	err = dbuf_read_verify_dnode_crypt(db, flags);
1363eda14cbcSMatt Macy 	if (err)
1364eda14cbcSMatt Macy 		return (err);
1365eda14cbcSMatt Macy 
1366eda14cbcSMatt Macy 	bonuslen = MIN(dn->dn_bonuslen, dn->dn_phys->dn_bonuslen);
1367eda14cbcSMatt Macy 	max_bonuslen = DN_SLOTS_TO_BONUSLEN(dn->dn_num_slots);
1368eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1369eda14cbcSMatt Macy 	ASSERT(DB_DNODE_HELD(db));
1370eda14cbcSMatt Macy 	ASSERT3U(bonuslen, <=, db->db.db_size);
1371eda14cbcSMatt Macy 	db->db.db_data = kmem_alloc(max_bonuslen, KM_SLEEP);
1372eda14cbcSMatt Macy 	arc_space_consume(max_bonuslen, ARC_SPACE_BONUS);
1373eda14cbcSMatt Macy 	if (bonuslen < max_bonuslen)
1374eda14cbcSMatt Macy 		bzero(db->db.db_data, max_bonuslen);
1375eda14cbcSMatt Macy 	if (bonuslen)
1376eda14cbcSMatt Macy 		bcopy(DN_BONUS(dn->dn_phys), db->db.db_data, bonuslen);
1377eda14cbcSMatt Macy 	db->db_state = DB_CACHED;
1378eda14cbcSMatt Macy 	DTRACE_SET_STATE(db, "bonus buffer filled");
1379eda14cbcSMatt Macy 	return (0);
1380eda14cbcSMatt Macy }
1381eda14cbcSMatt Macy 
1382eda14cbcSMatt Macy static void
1383eda14cbcSMatt Macy dbuf_handle_indirect_hole(dmu_buf_impl_t *db, dnode_t *dn)
1384eda14cbcSMatt Macy {
1385eda14cbcSMatt Macy 	blkptr_t *bps = db->db.db_data;
1386eda14cbcSMatt Macy 	uint32_t indbs = 1ULL << dn->dn_indblkshift;
1387eda14cbcSMatt Macy 	int n_bps = indbs >> SPA_BLKPTRSHIFT;
1388eda14cbcSMatt Macy 
1389eda14cbcSMatt Macy 	for (int i = 0; i < n_bps; i++) {
1390eda14cbcSMatt Macy 		blkptr_t *bp = &bps[i];
1391eda14cbcSMatt Macy 
1392eda14cbcSMatt Macy 		ASSERT3U(BP_GET_LSIZE(db->db_blkptr), ==, indbs);
1393eda14cbcSMatt Macy 		BP_SET_LSIZE(bp, BP_GET_LEVEL(db->db_blkptr) == 1 ?
1394eda14cbcSMatt Macy 		    dn->dn_datablksz : BP_GET_LSIZE(db->db_blkptr));
1395eda14cbcSMatt Macy 		BP_SET_TYPE(bp, BP_GET_TYPE(db->db_blkptr));
1396eda14cbcSMatt Macy 		BP_SET_LEVEL(bp, BP_GET_LEVEL(db->db_blkptr) - 1);
1397eda14cbcSMatt Macy 		BP_SET_BIRTH(bp, db->db_blkptr->blk_birth, 0);
1398eda14cbcSMatt Macy 	}
1399eda14cbcSMatt Macy }
1400eda14cbcSMatt Macy 
1401eda14cbcSMatt Macy /*
1402eda14cbcSMatt Macy  * Handle reads on dbufs that are holes, if necessary.  This function
1403eda14cbcSMatt Macy  * requires that the dbuf's mutex is held. Returns success (0) if action
1404eda14cbcSMatt Macy  * was taken, ENOENT if no action was taken.
1405eda14cbcSMatt Macy  */
1406eda14cbcSMatt Macy static int
1407eda14cbcSMatt Macy dbuf_read_hole(dmu_buf_impl_t *db, dnode_t *dn, uint32_t flags)
1408eda14cbcSMatt Macy {
1409eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1410eda14cbcSMatt Macy 
1411eda14cbcSMatt Macy 	int is_hole = db->db_blkptr == NULL || BP_IS_HOLE(db->db_blkptr);
1412eda14cbcSMatt Macy 	/*
1413eda14cbcSMatt Macy 	 * For level 0 blocks only, if the above check fails:
1414eda14cbcSMatt Macy 	 * Recheck BP_IS_HOLE() after dnode_block_freed() in case dnode_sync()
1415eda14cbcSMatt Macy 	 * processes the delete record and clears the bp while we are waiting
1416eda14cbcSMatt Macy 	 * for the dn_mtx (resulting in a "no" from block_freed).
1417eda14cbcSMatt Macy 	 */
1418eda14cbcSMatt Macy 	if (!is_hole && db->db_level == 0) {
1419eda14cbcSMatt Macy 		is_hole = dnode_block_freed(dn, db->db_blkid) ||
1420eda14cbcSMatt Macy 		    BP_IS_HOLE(db->db_blkptr);
1421eda14cbcSMatt Macy 	}
1422eda14cbcSMatt Macy 
1423eda14cbcSMatt Macy 	if (is_hole) {
1424eda14cbcSMatt Macy 		dbuf_set_data(db, dbuf_alloc_arcbuf(db));
1425eda14cbcSMatt Macy 		bzero(db->db.db_data, db->db.db_size);
1426eda14cbcSMatt Macy 
1427eda14cbcSMatt Macy 		if (db->db_blkptr != NULL && db->db_level > 0 &&
1428eda14cbcSMatt Macy 		    BP_IS_HOLE(db->db_blkptr) &&
1429eda14cbcSMatt Macy 		    db->db_blkptr->blk_birth != 0) {
1430eda14cbcSMatt Macy 			dbuf_handle_indirect_hole(db, dn);
1431eda14cbcSMatt Macy 		}
1432eda14cbcSMatt Macy 		db->db_state = DB_CACHED;
1433eda14cbcSMatt Macy 		DTRACE_SET_STATE(db, "hole read satisfied");
1434eda14cbcSMatt Macy 		return (0);
1435eda14cbcSMatt Macy 	}
1436eda14cbcSMatt Macy 	return (ENOENT);
1437eda14cbcSMatt Macy }
1438eda14cbcSMatt Macy 
1439eda14cbcSMatt Macy /*
1440eda14cbcSMatt Macy  * This function ensures that, when doing a decrypting read of a block,
1441eda14cbcSMatt Macy  * we make sure we have decrypted the dnode associated with it. We must do
1442eda14cbcSMatt Macy  * this so that we ensure we are fully authenticating the checksum-of-MACs
1443eda14cbcSMatt Macy  * tree from the root of the objset down to this block. Indirect blocks are
1444eda14cbcSMatt Macy  * always verified against their secure checksum-of-MACs assuming that the
1445eda14cbcSMatt Macy  * dnode containing them is correct. Now that we are doing a decrypting read,
1446eda14cbcSMatt Macy  * we can be sure that the key is loaded and verify that assumption. This is
1447eda14cbcSMatt Macy  * especially important considering that we always read encrypted dnode
1448eda14cbcSMatt Macy  * blocks as raw data (without verifying their MACs) to start, and
1449eda14cbcSMatt Macy  * decrypt / authenticate them when we need to read an encrypted bonus buffer.
1450eda14cbcSMatt Macy  */
1451eda14cbcSMatt Macy static int
1452eda14cbcSMatt Macy dbuf_read_verify_dnode_crypt(dmu_buf_impl_t *db, uint32_t flags)
1453eda14cbcSMatt Macy {
1454eda14cbcSMatt Macy 	int err = 0;
1455eda14cbcSMatt Macy 	objset_t *os = db->db_objset;
1456eda14cbcSMatt Macy 	arc_buf_t *dnode_abuf;
1457eda14cbcSMatt Macy 	dnode_t *dn;
1458eda14cbcSMatt Macy 	zbookmark_phys_t zb;
1459eda14cbcSMatt Macy 
1460eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1461eda14cbcSMatt Macy 
1462eda14cbcSMatt Macy 	if (!os->os_encrypted || os->os_raw_receive ||
1463eda14cbcSMatt Macy 	    (flags & DB_RF_NO_DECRYPT) != 0)
1464eda14cbcSMatt Macy 		return (0);
1465eda14cbcSMatt Macy 
1466eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
1467eda14cbcSMatt Macy 	dn = DB_DNODE(db);
1468eda14cbcSMatt Macy 	dnode_abuf = (dn->dn_dbuf != NULL) ? dn->dn_dbuf->db_buf : NULL;
1469eda14cbcSMatt Macy 
1470eda14cbcSMatt Macy 	if (dnode_abuf == NULL || !arc_is_encrypted(dnode_abuf)) {
1471eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
1472eda14cbcSMatt Macy 		return (0);
1473eda14cbcSMatt Macy 	}
1474eda14cbcSMatt Macy 
1475eda14cbcSMatt Macy 	SET_BOOKMARK(&zb, dmu_objset_id(os),
1476eda14cbcSMatt Macy 	    DMU_META_DNODE_OBJECT, 0, dn->dn_dbuf->db_blkid);
1477eda14cbcSMatt Macy 	err = arc_untransform(dnode_abuf, os->os_spa, &zb, B_TRUE);
1478eda14cbcSMatt Macy 
1479eda14cbcSMatt Macy 	/*
1480eda14cbcSMatt Macy 	 * An error code of EACCES tells us that the key is still not
1481eda14cbcSMatt Macy 	 * available. This is ok if we are only reading authenticated
1482eda14cbcSMatt Macy 	 * (and therefore non-encrypted) blocks.
1483eda14cbcSMatt Macy 	 */
1484eda14cbcSMatt Macy 	if (err == EACCES && ((db->db_blkid != DMU_BONUS_BLKID &&
1485eda14cbcSMatt Macy 	    !DMU_OT_IS_ENCRYPTED(dn->dn_type)) ||
1486eda14cbcSMatt Macy 	    (db->db_blkid == DMU_BONUS_BLKID &&
1487eda14cbcSMatt Macy 	    !DMU_OT_IS_ENCRYPTED(dn->dn_bonustype))))
1488eda14cbcSMatt Macy 		err = 0;
1489eda14cbcSMatt Macy 
1490eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
1491eda14cbcSMatt Macy 
1492eda14cbcSMatt Macy 	return (err);
1493eda14cbcSMatt Macy }
1494eda14cbcSMatt Macy 
1495eda14cbcSMatt Macy /*
1496eda14cbcSMatt Macy  * Drops db_mtx and the parent lock specified by dblt and tag before
1497eda14cbcSMatt Macy  * returning.
1498eda14cbcSMatt Macy  */
1499eda14cbcSMatt Macy static int
1500eda14cbcSMatt Macy dbuf_read_impl(dmu_buf_impl_t *db, zio_t *zio, uint32_t flags,
1501eda14cbcSMatt Macy     db_lock_type_t dblt, void *tag)
1502eda14cbcSMatt Macy {
1503eda14cbcSMatt Macy 	dnode_t *dn;
1504eda14cbcSMatt Macy 	zbookmark_phys_t zb;
1505eda14cbcSMatt Macy 	uint32_t aflags = ARC_FLAG_NOWAIT;
1506eda14cbcSMatt Macy 	int err, zio_flags;
1507eda14cbcSMatt Macy 
1508eda14cbcSMatt Macy 	err = zio_flags = 0;
1509eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
1510eda14cbcSMatt Macy 	dn = DB_DNODE(db);
1511eda14cbcSMatt Macy 	ASSERT(!zfs_refcount_is_zero(&db->db_holds));
1512eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1513eda14cbcSMatt Macy 	ASSERT(db->db_state == DB_UNCACHED);
1514eda14cbcSMatt Macy 	ASSERT(db->db_buf == NULL);
1515eda14cbcSMatt Macy 	ASSERT(db->db_parent == NULL ||
1516eda14cbcSMatt Macy 	    RW_LOCK_HELD(&db->db_parent->db_rwlock));
1517eda14cbcSMatt Macy 
1518eda14cbcSMatt Macy 	if (db->db_blkid == DMU_BONUS_BLKID) {
1519eda14cbcSMatt Macy 		err = dbuf_read_bonus(db, dn, flags);
1520eda14cbcSMatt Macy 		goto early_unlock;
1521eda14cbcSMatt Macy 	}
1522eda14cbcSMatt Macy 
1523eda14cbcSMatt Macy 	err = dbuf_read_hole(db, dn, flags);
1524eda14cbcSMatt Macy 	if (err == 0)
1525eda14cbcSMatt Macy 		goto early_unlock;
1526eda14cbcSMatt Macy 
1527eda14cbcSMatt Macy 	/*
1528eda14cbcSMatt Macy 	 * Any attempt to read a redacted block should result in an error. This
1529eda14cbcSMatt Macy 	 * will never happen under normal conditions, but can be useful for
1530eda14cbcSMatt Macy 	 * debugging purposes.
1531eda14cbcSMatt Macy 	 */
1532eda14cbcSMatt Macy 	if (BP_IS_REDACTED(db->db_blkptr)) {
1533eda14cbcSMatt Macy 		ASSERT(dsl_dataset_feature_is_active(
1534eda14cbcSMatt Macy 		    db->db_objset->os_dsl_dataset,
1535eda14cbcSMatt Macy 		    SPA_FEATURE_REDACTED_DATASETS));
1536eda14cbcSMatt Macy 		err = SET_ERROR(EIO);
1537eda14cbcSMatt Macy 		goto early_unlock;
1538eda14cbcSMatt Macy 	}
1539eda14cbcSMatt Macy 
1540eda14cbcSMatt Macy 	SET_BOOKMARK(&zb, dmu_objset_id(db->db_objset),
1541eda14cbcSMatt Macy 	    db->db.db_object, db->db_level, db->db_blkid);
1542eda14cbcSMatt Macy 
1543eda14cbcSMatt Macy 	/*
1544eda14cbcSMatt Macy 	 * All bps of an encrypted os should have the encryption bit set.
1545eda14cbcSMatt Macy 	 * If this is not true it indicates tampering and we report an error.
1546eda14cbcSMatt Macy 	 */
1547eda14cbcSMatt Macy 	if (db->db_objset->os_encrypted && !BP_USES_CRYPT(db->db_blkptr)) {
1548eda14cbcSMatt Macy 		spa_log_error(db->db_objset->os_spa, &zb);
1549eda14cbcSMatt Macy 		zfs_panic_recover("unencrypted block in encrypted "
1550eda14cbcSMatt Macy 		    "object set %llu", dmu_objset_id(db->db_objset));
1551eda14cbcSMatt Macy 		err = SET_ERROR(EIO);
1552eda14cbcSMatt Macy 		goto early_unlock;
1553eda14cbcSMatt Macy 	}
1554eda14cbcSMatt Macy 
1555eda14cbcSMatt Macy 	err = dbuf_read_verify_dnode_crypt(db, flags);
1556eda14cbcSMatt Macy 	if (err != 0)
1557eda14cbcSMatt Macy 		goto early_unlock;
1558eda14cbcSMatt Macy 
1559eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
1560eda14cbcSMatt Macy 
1561eda14cbcSMatt Macy 	db->db_state = DB_READ;
1562eda14cbcSMatt Macy 	DTRACE_SET_STATE(db, "read issued");
1563eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
1564eda14cbcSMatt Macy 
1565eda14cbcSMatt Macy 	if (DBUF_IS_L2CACHEABLE(db))
1566eda14cbcSMatt Macy 		aflags |= ARC_FLAG_L2CACHE;
1567eda14cbcSMatt Macy 
1568eda14cbcSMatt Macy 	dbuf_add_ref(db, NULL);
1569eda14cbcSMatt Macy 
1570eda14cbcSMatt Macy 	zio_flags = (flags & DB_RF_CANFAIL) ?
1571eda14cbcSMatt Macy 	    ZIO_FLAG_CANFAIL : ZIO_FLAG_MUSTSUCCEED;
1572eda14cbcSMatt Macy 
1573eda14cbcSMatt Macy 	if ((flags & DB_RF_NO_DECRYPT) && BP_IS_PROTECTED(db->db_blkptr))
1574eda14cbcSMatt Macy 		zio_flags |= ZIO_FLAG_RAW;
1575eda14cbcSMatt Macy 	/*
1576eda14cbcSMatt Macy 	 * The zio layer will copy the provided blkptr later, but we need to
1577eda14cbcSMatt Macy 	 * do this now so that we can release the parent's rwlock. We have to
1578eda14cbcSMatt Macy 	 * do that now so that if dbuf_read_done is called synchronously (on
1579eda14cbcSMatt Macy 	 * an l1 cache hit) we don't acquire the db_mtx while holding the
1580eda14cbcSMatt Macy 	 * parent's rwlock, which would be a lock ordering violation.
1581eda14cbcSMatt Macy 	 */
1582eda14cbcSMatt Macy 	blkptr_t bp = *db->db_blkptr;
1583eda14cbcSMatt Macy 	dmu_buf_unlock_parent(db, dblt, tag);
1584eda14cbcSMatt Macy 	(void) arc_read(zio, db->db_objset->os_spa, &bp,
1585eda14cbcSMatt Macy 	    dbuf_read_done, db, ZIO_PRIORITY_SYNC_READ, zio_flags,
1586eda14cbcSMatt Macy 	    &aflags, &zb);
1587eda14cbcSMatt Macy 	return (err);
1588eda14cbcSMatt Macy early_unlock:
1589eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
1590eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
1591eda14cbcSMatt Macy 	dmu_buf_unlock_parent(db, dblt, tag);
1592eda14cbcSMatt Macy 	return (err);
1593eda14cbcSMatt Macy }
1594eda14cbcSMatt Macy 
1595eda14cbcSMatt Macy /*
1596eda14cbcSMatt Macy  * This is our just-in-time copy function.  It makes a copy of buffers that
1597eda14cbcSMatt Macy  * have been modified in a previous transaction group before we access them in
1598eda14cbcSMatt Macy  * the current active group.
1599eda14cbcSMatt Macy  *
1600eda14cbcSMatt Macy  * This function is used in three places: when we are dirtying a buffer for the
1601eda14cbcSMatt Macy  * first time in a txg, when we are freeing a range in a dnode that includes
1602eda14cbcSMatt Macy  * this buffer, and when we are accessing a buffer which was received compressed
1603eda14cbcSMatt Macy  * and later referenced in a WRITE_BYREF record.
1604eda14cbcSMatt Macy  *
1605eda14cbcSMatt Macy  * Note that when we are called from dbuf_free_range() we do not put a hold on
1606eda14cbcSMatt Macy  * the buffer, we just traverse the active dbuf list for the dnode.
1607eda14cbcSMatt Macy  */
1608eda14cbcSMatt Macy static void
1609eda14cbcSMatt Macy dbuf_fix_old_data(dmu_buf_impl_t *db, uint64_t txg)
1610eda14cbcSMatt Macy {
1611eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr = list_head(&db->db_dirty_records);
1612eda14cbcSMatt Macy 
1613eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1614eda14cbcSMatt Macy 	ASSERT(db->db.db_data != NULL);
1615eda14cbcSMatt Macy 	ASSERT(db->db_level == 0);
1616eda14cbcSMatt Macy 	ASSERT(db->db.db_object != DMU_META_DNODE_OBJECT);
1617eda14cbcSMatt Macy 
1618eda14cbcSMatt Macy 	if (dr == NULL ||
1619eda14cbcSMatt Macy 	    (dr->dt.dl.dr_data !=
1620eda14cbcSMatt Macy 	    ((db->db_blkid  == DMU_BONUS_BLKID) ? db->db.db_data : db->db_buf)))
1621eda14cbcSMatt Macy 		return;
1622eda14cbcSMatt Macy 
1623eda14cbcSMatt Macy 	/*
1624eda14cbcSMatt Macy 	 * If the last dirty record for this dbuf has not yet synced
1625eda14cbcSMatt Macy 	 * and its referencing the dbuf data, either:
1626eda14cbcSMatt Macy 	 *	reset the reference to point to a new copy,
1627eda14cbcSMatt Macy 	 * or (if there a no active holders)
1628eda14cbcSMatt Macy 	 *	just null out the current db_data pointer.
1629eda14cbcSMatt Macy 	 */
1630eda14cbcSMatt Macy 	ASSERT3U(dr->dr_txg, >=, txg - 2);
1631eda14cbcSMatt Macy 	if (db->db_blkid == DMU_BONUS_BLKID) {
1632eda14cbcSMatt Macy 		dnode_t *dn = DB_DNODE(db);
1633eda14cbcSMatt Macy 		int bonuslen = DN_SLOTS_TO_BONUSLEN(dn->dn_num_slots);
1634eda14cbcSMatt Macy 		dr->dt.dl.dr_data = kmem_alloc(bonuslen, KM_SLEEP);
1635eda14cbcSMatt Macy 		arc_space_consume(bonuslen, ARC_SPACE_BONUS);
1636eda14cbcSMatt Macy 		bcopy(db->db.db_data, dr->dt.dl.dr_data, bonuslen);
1637eda14cbcSMatt Macy 	} else if (zfs_refcount_count(&db->db_holds) > db->db_dirtycnt) {
1638eda14cbcSMatt Macy 		arc_buf_t *buf = dbuf_alloc_arcbuf_from_arcbuf(db, db->db_buf);
1639eda14cbcSMatt Macy 		dr->dt.dl.dr_data = buf;
1640eda14cbcSMatt Macy 		bcopy(db->db.db_data, buf->b_data, arc_buf_size(buf));
1641eda14cbcSMatt Macy 	} else {
1642eda14cbcSMatt Macy 		db->db_buf = NULL;
1643eda14cbcSMatt Macy 		dbuf_clear_data(db);
1644eda14cbcSMatt Macy 	}
1645eda14cbcSMatt Macy }
1646eda14cbcSMatt Macy 
1647eda14cbcSMatt Macy int
1648eda14cbcSMatt Macy dbuf_read(dmu_buf_impl_t *db, zio_t *zio, uint32_t flags)
1649eda14cbcSMatt Macy {
1650eda14cbcSMatt Macy 	int err = 0;
1651eda14cbcSMatt Macy 	boolean_t prefetch;
1652eda14cbcSMatt Macy 	dnode_t *dn;
1653eda14cbcSMatt Macy 
1654eda14cbcSMatt Macy 	/*
1655eda14cbcSMatt Macy 	 * We don't have to hold the mutex to check db_state because it
1656eda14cbcSMatt Macy 	 * can't be freed while we have a hold on the buffer.
1657eda14cbcSMatt Macy 	 */
1658eda14cbcSMatt Macy 	ASSERT(!zfs_refcount_is_zero(&db->db_holds));
1659eda14cbcSMatt Macy 
1660eda14cbcSMatt Macy 	if (db->db_state == DB_NOFILL)
1661eda14cbcSMatt Macy 		return (SET_ERROR(EIO));
1662eda14cbcSMatt Macy 
1663eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
1664eda14cbcSMatt Macy 	dn = DB_DNODE(db);
1665eda14cbcSMatt Macy 
1666eda14cbcSMatt Macy 	prefetch = db->db_level == 0 && db->db_blkid != DMU_BONUS_BLKID &&
1667eda14cbcSMatt Macy 	    (flags & DB_RF_NOPREFETCH) == 0 && dn != NULL &&
1668eda14cbcSMatt Macy 	    DBUF_IS_CACHEABLE(db);
1669eda14cbcSMatt Macy 
1670eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
1671eda14cbcSMatt Macy 	if (db->db_state == DB_CACHED) {
1672eda14cbcSMatt Macy 		spa_t *spa = dn->dn_objset->os_spa;
1673eda14cbcSMatt Macy 
1674eda14cbcSMatt Macy 		/*
1675eda14cbcSMatt Macy 		 * Ensure that this block's dnode has been decrypted if
1676eda14cbcSMatt Macy 		 * the caller has requested decrypted data.
1677eda14cbcSMatt Macy 		 */
1678eda14cbcSMatt Macy 		err = dbuf_read_verify_dnode_crypt(db, flags);
1679eda14cbcSMatt Macy 
1680eda14cbcSMatt Macy 		/*
1681eda14cbcSMatt Macy 		 * If the arc buf is compressed or encrypted and the caller
1682eda14cbcSMatt Macy 		 * requested uncompressed data, we need to untransform it
1683eda14cbcSMatt Macy 		 * before returning. We also call arc_untransform() on any
1684eda14cbcSMatt Macy 		 * unauthenticated blocks, which will verify their MAC if
1685eda14cbcSMatt Macy 		 * the key is now available.
1686eda14cbcSMatt Macy 		 */
1687eda14cbcSMatt Macy 		if (err == 0 && db->db_buf != NULL &&
1688eda14cbcSMatt Macy 		    (flags & DB_RF_NO_DECRYPT) == 0 &&
1689eda14cbcSMatt Macy 		    (arc_is_encrypted(db->db_buf) ||
1690eda14cbcSMatt Macy 		    arc_is_unauthenticated(db->db_buf) ||
1691eda14cbcSMatt Macy 		    arc_get_compression(db->db_buf) != ZIO_COMPRESS_OFF)) {
1692eda14cbcSMatt Macy 			zbookmark_phys_t zb;
1693eda14cbcSMatt Macy 
1694eda14cbcSMatt Macy 			SET_BOOKMARK(&zb, dmu_objset_id(db->db_objset),
1695eda14cbcSMatt Macy 			    db->db.db_object, db->db_level, db->db_blkid);
1696eda14cbcSMatt Macy 			dbuf_fix_old_data(db, spa_syncing_txg(spa));
1697eda14cbcSMatt Macy 			err = arc_untransform(db->db_buf, spa, &zb, B_FALSE);
1698eda14cbcSMatt Macy 			dbuf_set_data(db, db->db_buf);
1699eda14cbcSMatt Macy 		}
1700eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
1701eda14cbcSMatt Macy 		if (err == 0 && prefetch) {
1702eda14cbcSMatt Macy 			dmu_zfetch(&dn->dn_zfetch, db->db_blkid, 1, B_TRUE,
1703f9693befSMartin Matuska 			    B_FALSE, flags & DB_RF_HAVESTRUCT);
1704eda14cbcSMatt Macy 		}
1705eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
1706eda14cbcSMatt Macy 		DBUF_STAT_BUMP(hash_hits);
1707eda14cbcSMatt Macy 	} else if (db->db_state == DB_UNCACHED) {
1708eda14cbcSMatt Macy 		spa_t *spa = dn->dn_objset->os_spa;
1709eda14cbcSMatt Macy 		boolean_t need_wait = B_FALSE;
1710eda14cbcSMatt Macy 
1711eda14cbcSMatt Macy 		db_lock_type_t dblt = dmu_buf_lock_parent(db, RW_READER, FTAG);
1712eda14cbcSMatt Macy 
1713eda14cbcSMatt Macy 		if (zio == NULL &&
1714eda14cbcSMatt Macy 		    db->db_blkptr != NULL && !BP_IS_HOLE(db->db_blkptr)) {
1715eda14cbcSMatt Macy 			zio = zio_root(spa, NULL, NULL, ZIO_FLAG_CANFAIL);
1716eda14cbcSMatt Macy 			need_wait = B_TRUE;
1717eda14cbcSMatt Macy 		}
1718eda14cbcSMatt Macy 		err = dbuf_read_impl(db, zio, flags, dblt, FTAG);
1719eda14cbcSMatt Macy 		/*
1720eda14cbcSMatt Macy 		 * dbuf_read_impl has dropped db_mtx and our parent's rwlock
1721eda14cbcSMatt Macy 		 * for us
1722eda14cbcSMatt Macy 		 */
1723eda14cbcSMatt Macy 		if (!err && prefetch) {
1724eda14cbcSMatt Macy 			dmu_zfetch(&dn->dn_zfetch, db->db_blkid, 1, B_TRUE,
1725f9693befSMartin Matuska 			    db->db_state != DB_CACHED,
1726eda14cbcSMatt Macy 			    flags & DB_RF_HAVESTRUCT);
1727eda14cbcSMatt Macy 		}
1728eda14cbcSMatt Macy 
1729eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
1730eda14cbcSMatt Macy 		DBUF_STAT_BUMP(hash_misses);
1731eda14cbcSMatt Macy 
1732eda14cbcSMatt Macy 		/*
1733eda14cbcSMatt Macy 		 * If we created a zio_root we must execute it to avoid
1734eda14cbcSMatt Macy 		 * leaking it, even if it isn't attached to any work due
1735eda14cbcSMatt Macy 		 * to an error in dbuf_read_impl().
1736eda14cbcSMatt Macy 		 */
1737eda14cbcSMatt Macy 		if (need_wait) {
1738eda14cbcSMatt Macy 			if (err == 0)
1739eda14cbcSMatt Macy 				err = zio_wait(zio);
1740eda14cbcSMatt Macy 			else
1741eda14cbcSMatt Macy 				VERIFY0(zio_wait(zio));
1742eda14cbcSMatt Macy 		}
1743eda14cbcSMatt Macy 	} else {
1744eda14cbcSMatt Macy 		/*
1745eda14cbcSMatt Macy 		 * Another reader came in while the dbuf was in flight
1746eda14cbcSMatt Macy 		 * between UNCACHED and CACHED.  Either a writer will finish
1747eda14cbcSMatt Macy 		 * writing the buffer (sending the dbuf to CACHED) or the
1748eda14cbcSMatt Macy 		 * first reader's request will reach the read_done callback
1749eda14cbcSMatt Macy 		 * and send the dbuf to CACHED.  Otherwise, a failure
1750eda14cbcSMatt Macy 		 * occurred and the dbuf went to UNCACHED.
1751eda14cbcSMatt Macy 		 */
1752eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
1753eda14cbcSMatt Macy 		if (prefetch) {
1754eda14cbcSMatt Macy 			dmu_zfetch(&dn->dn_zfetch, db->db_blkid, 1, B_TRUE,
1755f9693befSMartin Matuska 			    B_TRUE, flags & DB_RF_HAVESTRUCT);
1756eda14cbcSMatt Macy 		}
1757eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
1758eda14cbcSMatt Macy 		DBUF_STAT_BUMP(hash_misses);
1759eda14cbcSMatt Macy 
1760eda14cbcSMatt Macy 		/* Skip the wait per the caller's request. */
1761eda14cbcSMatt Macy 		if ((flags & DB_RF_NEVERWAIT) == 0) {
1762eda14cbcSMatt Macy 			mutex_enter(&db->db_mtx);
1763eda14cbcSMatt Macy 			while (db->db_state == DB_READ ||
1764eda14cbcSMatt Macy 			    db->db_state == DB_FILL) {
1765eda14cbcSMatt Macy 				ASSERT(db->db_state == DB_READ ||
1766eda14cbcSMatt Macy 				    (flags & DB_RF_HAVESTRUCT) == 0);
1767eda14cbcSMatt Macy 				DTRACE_PROBE2(blocked__read, dmu_buf_impl_t *,
1768eda14cbcSMatt Macy 				    db, zio_t *, zio);
1769eda14cbcSMatt Macy 				cv_wait(&db->db_changed, &db->db_mtx);
1770eda14cbcSMatt Macy 			}
1771eda14cbcSMatt Macy 			if (db->db_state == DB_UNCACHED)
1772eda14cbcSMatt Macy 				err = SET_ERROR(EIO);
1773eda14cbcSMatt Macy 			mutex_exit(&db->db_mtx);
1774eda14cbcSMatt Macy 		}
1775eda14cbcSMatt Macy 	}
1776eda14cbcSMatt Macy 
1777eda14cbcSMatt Macy 	return (err);
1778eda14cbcSMatt Macy }
1779eda14cbcSMatt Macy 
1780eda14cbcSMatt Macy static void
1781eda14cbcSMatt Macy dbuf_noread(dmu_buf_impl_t *db)
1782eda14cbcSMatt Macy {
1783eda14cbcSMatt Macy 	ASSERT(!zfs_refcount_is_zero(&db->db_holds));
1784eda14cbcSMatt Macy 	ASSERT(db->db_blkid != DMU_BONUS_BLKID);
1785eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
1786eda14cbcSMatt Macy 	while (db->db_state == DB_READ || db->db_state == DB_FILL)
1787eda14cbcSMatt Macy 		cv_wait(&db->db_changed, &db->db_mtx);
1788eda14cbcSMatt Macy 	if (db->db_state == DB_UNCACHED) {
1789eda14cbcSMatt Macy 		ASSERT(db->db_buf == NULL);
1790eda14cbcSMatt Macy 		ASSERT(db->db.db_data == NULL);
1791eda14cbcSMatt Macy 		dbuf_set_data(db, dbuf_alloc_arcbuf(db));
1792eda14cbcSMatt Macy 		db->db_state = DB_FILL;
1793eda14cbcSMatt Macy 		DTRACE_SET_STATE(db, "assigning filled buffer");
1794eda14cbcSMatt Macy 	} else if (db->db_state == DB_NOFILL) {
1795eda14cbcSMatt Macy 		dbuf_clear_data(db);
1796eda14cbcSMatt Macy 	} else {
1797eda14cbcSMatt Macy 		ASSERT3U(db->db_state, ==, DB_CACHED);
1798eda14cbcSMatt Macy 	}
1799eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
1800eda14cbcSMatt Macy }
1801eda14cbcSMatt Macy 
1802eda14cbcSMatt Macy void
1803eda14cbcSMatt Macy dbuf_unoverride(dbuf_dirty_record_t *dr)
1804eda14cbcSMatt Macy {
1805eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
1806eda14cbcSMatt Macy 	blkptr_t *bp = &dr->dt.dl.dr_overridden_by;
1807eda14cbcSMatt Macy 	uint64_t txg = dr->dr_txg;
1808eda14cbcSMatt Macy 
1809eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
1810eda14cbcSMatt Macy 	/*
1811eda14cbcSMatt Macy 	 * This assert is valid because dmu_sync() expects to be called by
1812eda14cbcSMatt Macy 	 * a zilog's get_data while holding a range lock.  This call only
1813eda14cbcSMatt Macy 	 * comes from dbuf_dirty() callers who must also hold a range lock.
1814eda14cbcSMatt Macy 	 */
1815eda14cbcSMatt Macy 	ASSERT(dr->dt.dl.dr_override_state != DR_IN_DMU_SYNC);
1816eda14cbcSMatt Macy 	ASSERT(db->db_level == 0);
1817eda14cbcSMatt Macy 
1818eda14cbcSMatt Macy 	if (db->db_blkid == DMU_BONUS_BLKID ||
1819eda14cbcSMatt Macy 	    dr->dt.dl.dr_override_state == DR_NOT_OVERRIDDEN)
1820eda14cbcSMatt Macy 		return;
1821eda14cbcSMatt Macy 
1822eda14cbcSMatt Macy 	ASSERT(db->db_data_pending != dr);
1823eda14cbcSMatt Macy 
1824eda14cbcSMatt Macy 	/* free this block */
1825eda14cbcSMatt Macy 	if (!BP_IS_HOLE(bp) && !dr->dt.dl.dr_nopwrite)
1826eda14cbcSMatt Macy 		zio_free(db->db_objset->os_spa, txg, bp);
1827eda14cbcSMatt Macy 
1828eda14cbcSMatt Macy 	dr->dt.dl.dr_override_state = DR_NOT_OVERRIDDEN;
1829eda14cbcSMatt Macy 	dr->dt.dl.dr_nopwrite = B_FALSE;
1830eda14cbcSMatt Macy 	dr->dt.dl.dr_has_raw_params = B_FALSE;
1831eda14cbcSMatt Macy 
1832eda14cbcSMatt Macy 	/*
1833eda14cbcSMatt Macy 	 * Release the already-written buffer, so we leave it in
1834eda14cbcSMatt Macy 	 * a consistent dirty state.  Note that all callers are
1835eda14cbcSMatt Macy 	 * modifying the buffer, so they will immediately do
1836eda14cbcSMatt Macy 	 * another (redundant) arc_release().  Therefore, leave
1837eda14cbcSMatt Macy 	 * the buf thawed to save the effort of freezing &
1838eda14cbcSMatt Macy 	 * immediately re-thawing it.
1839eda14cbcSMatt Macy 	 */
1840eda14cbcSMatt Macy 	arc_release(dr->dt.dl.dr_data, db);
1841eda14cbcSMatt Macy }
1842eda14cbcSMatt Macy 
1843eda14cbcSMatt Macy /*
1844eda14cbcSMatt Macy  * Evict (if its unreferenced) or clear (if its referenced) any level-0
1845eda14cbcSMatt Macy  * data blocks in the free range, so that any future readers will find
1846eda14cbcSMatt Macy  * empty blocks.
1847eda14cbcSMatt Macy  */
1848eda14cbcSMatt Macy void
1849eda14cbcSMatt Macy dbuf_free_range(dnode_t *dn, uint64_t start_blkid, uint64_t end_blkid,
1850eda14cbcSMatt Macy     dmu_tx_t *tx)
1851eda14cbcSMatt Macy {
1852eda14cbcSMatt Macy 	dmu_buf_impl_t *db_search;
1853eda14cbcSMatt Macy 	dmu_buf_impl_t *db, *db_next;
1854eda14cbcSMatt Macy 	uint64_t txg = tx->tx_txg;
1855eda14cbcSMatt Macy 	avl_index_t where;
1856eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
1857eda14cbcSMatt Macy 
1858eda14cbcSMatt Macy 	if (end_blkid > dn->dn_maxblkid &&
1859eda14cbcSMatt Macy 	    !(start_blkid == DMU_SPILL_BLKID || end_blkid == DMU_SPILL_BLKID))
1860eda14cbcSMatt Macy 		end_blkid = dn->dn_maxblkid;
1861eda14cbcSMatt Macy 	dprintf_dnode(dn, "start=%llu end=%llu\n", start_blkid, end_blkid);
1862eda14cbcSMatt Macy 
1863eda14cbcSMatt Macy 	db_search = kmem_alloc(sizeof (dmu_buf_impl_t), KM_SLEEP);
1864eda14cbcSMatt Macy 	db_search->db_level = 0;
1865eda14cbcSMatt Macy 	db_search->db_blkid = start_blkid;
1866eda14cbcSMatt Macy 	db_search->db_state = DB_SEARCH;
1867eda14cbcSMatt Macy 
1868eda14cbcSMatt Macy 	mutex_enter(&dn->dn_dbufs_mtx);
1869eda14cbcSMatt Macy 	db = avl_find(&dn->dn_dbufs, db_search, &where);
1870eda14cbcSMatt Macy 	ASSERT3P(db, ==, NULL);
1871eda14cbcSMatt Macy 
1872eda14cbcSMatt Macy 	db = avl_nearest(&dn->dn_dbufs, where, AVL_AFTER);
1873eda14cbcSMatt Macy 
1874eda14cbcSMatt Macy 	for (; db != NULL; db = db_next) {
1875eda14cbcSMatt Macy 		db_next = AVL_NEXT(&dn->dn_dbufs, db);
1876eda14cbcSMatt Macy 		ASSERT(db->db_blkid != DMU_BONUS_BLKID);
1877eda14cbcSMatt Macy 
1878eda14cbcSMatt Macy 		if (db->db_level != 0 || db->db_blkid > end_blkid) {
1879eda14cbcSMatt Macy 			break;
1880eda14cbcSMatt Macy 		}
1881eda14cbcSMatt Macy 		ASSERT3U(db->db_blkid, >=, start_blkid);
1882eda14cbcSMatt Macy 
1883eda14cbcSMatt Macy 		/* found a level 0 buffer in the range */
1884eda14cbcSMatt Macy 		mutex_enter(&db->db_mtx);
1885eda14cbcSMatt Macy 		if (dbuf_undirty(db, tx)) {
1886eda14cbcSMatt Macy 			/* mutex has been dropped and dbuf destroyed */
1887eda14cbcSMatt Macy 			continue;
1888eda14cbcSMatt Macy 		}
1889eda14cbcSMatt Macy 
1890eda14cbcSMatt Macy 		if (db->db_state == DB_UNCACHED ||
1891eda14cbcSMatt Macy 		    db->db_state == DB_NOFILL ||
1892eda14cbcSMatt Macy 		    db->db_state == DB_EVICTING) {
1893eda14cbcSMatt Macy 			ASSERT(db->db.db_data == NULL);
1894eda14cbcSMatt Macy 			mutex_exit(&db->db_mtx);
1895eda14cbcSMatt Macy 			continue;
1896eda14cbcSMatt Macy 		}
1897eda14cbcSMatt Macy 		if (db->db_state == DB_READ || db->db_state == DB_FILL) {
1898eda14cbcSMatt Macy 			/* will be handled in dbuf_read_done or dbuf_rele */
1899eda14cbcSMatt Macy 			db->db_freed_in_flight = TRUE;
1900eda14cbcSMatt Macy 			mutex_exit(&db->db_mtx);
1901eda14cbcSMatt Macy 			continue;
1902eda14cbcSMatt Macy 		}
1903eda14cbcSMatt Macy 		if (zfs_refcount_count(&db->db_holds) == 0) {
1904eda14cbcSMatt Macy 			ASSERT(db->db_buf);
1905eda14cbcSMatt Macy 			dbuf_destroy(db);
1906eda14cbcSMatt Macy 			continue;
1907eda14cbcSMatt Macy 		}
1908eda14cbcSMatt Macy 		/* The dbuf is referenced */
1909eda14cbcSMatt Macy 
1910eda14cbcSMatt Macy 		dr = list_head(&db->db_dirty_records);
1911eda14cbcSMatt Macy 		if (dr != NULL) {
1912eda14cbcSMatt Macy 			if (dr->dr_txg == txg) {
1913eda14cbcSMatt Macy 				/*
1914eda14cbcSMatt Macy 				 * This buffer is "in-use", re-adjust the file
1915eda14cbcSMatt Macy 				 * size to reflect that this buffer may
1916eda14cbcSMatt Macy 				 * contain new data when we sync.
1917eda14cbcSMatt Macy 				 */
1918eda14cbcSMatt Macy 				if (db->db_blkid != DMU_SPILL_BLKID &&
1919eda14cbcSMatt Macy 				    db->db_blkid > dn->dn_maxblkid)
1920eda14cbcSMatt Macy 					dn->dn_maxblkid = db->db_blkid;
1921eda14cbcSMatt Macy 				dbuf_unoverride(dr);
1922eda14cbcSMatt Macy 			} else {
1923eda14cbcSMatt Macy 				/*
1924eda14cbcSMatt Macy 				 * This dbuf is not dirty in the open context.
1925eda14cbcSMatt Macy 				 * Either uncache it (if its not referenced in
1926eda14cbcSMatt Macy 				 * the open context) or reset its contents to
1927eda14cbcSMatt Macy 				 * empty.
1928eda14cbcSMatt Macy 				 */
1929eda14cbcSMatt Macy 				dbuf_fix_old_data(db, txg);
1930eda14cbcSMatt Macy 			}
1931eda14cbcSMatt Macy 		}
1932eda14cbcSMatt Macy 		/* clear the contents if its cached */
1933eda14cbcSMatt Macy 		if (db->db_state == DB_CACHED) {
1934eda14cbcSMatt Macy 			ASSERT(db->db.db_data != NULL);
1935eda14cbcSMatt Macy 			arc_release(db->db_buf, db);
1936eda14cbcSMatt Macy 			rw_enter(&db->db_rwlock, RW_WRITER);
1937eda14cbcSMatt Macy 			bzero(db->db.db_data, db->db.db_size);
1938eda14cbcSMatt Macy 			rw_exit(&db->db_rwlock);
1939eda14cbcSMatt Macy 			arc_buf_freeze(db->db_buf);
1940eda14cbcSMatt Macy 		}
1941eda14cbcSMatt Macy 
1942eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
1943eda14cbcSMatt Macy 	}
1944eda14cbcSMatt Macy 
1945eda14cbcSMatt Macy 	kmem_free(db_search, sizeof (dmu_buf_impl_t));
1946eda14cbcSMatt Macy 	mutex_exit(&dn->dn_dbufs_mtx);
1947eda14cbcSMatt Macy }
1948eda14cbcSMatt Macy 
1949eda14cbcSMatt Macy void
1950eda14cbcSMatt Macy dbuf_new_size(dmu_buf_impl_t *db, int size, dmu_tx_t *tx)
1951eda14cbcSMatt Macy {
1952eda14cbcSMatt Macy 	arc_buf_t *buf, *old_buf;
1953eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
1954eda14cbcSMatt Macy 	int osize = db->db.db_size;
1955eda14cbcSMatt Macy 	arc_buf_contents_t type = DBUF_GET_BUFC_TYPE(db);
1956eda14cbcSMatt Macy 	dnode_t *dn;
1957eda14cbcSMatt Macy 
1958eda14cbcSMatt Macy 	ASSERT(db->db_blkid != DMU_BONUS_BLKID);
1959eda14cbcSMatt Macy 
1960eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
1961eda14cbcSMatt Macy 	dn = DB_DNODE(db);
1962eda14cbcSMatt Macy 
1963eda14cbcSMatt Macy 	/*
1964eda14cbcSMatt Macy 	 * XXX we should be doing a dbuf_read, checking the return
1965eda14cbcSMatt Macy 	 * value and returning that up to our callers
1966eda14cbcSMatt Macy 	 */
1967eda14cbcSMatt Macy 	dmu_buf_will_dirty(&db->db, tx);
1968eda14cbcSMatt Macy 
1969eda14cbcSMatt Macy 	/* create the data buffer for the new block */
1970eda14cbcSMatt Macy 	buf = arc_alloc_buf(dn->dn_objset->os_spa, db, type, size);
1971eda14cbcSMatt Macy 
1972eda14cbcSMatt Macy 	/* copy old block data to the new block */
1973eda14cbcSMatt Macy 	old_buf = db->db_buf;
1974eda14cbcSMatt Macy 	bcopy(old_buf->b_data, buf->b_data, MIN(osize, size));
1975eda14cbcSMatt Macy 	/* zero the remainder */
1976eda14cbcSMatt Macy 	if (size > osize)
1977eda14cbcSMatt Macy 		bzero((uint8_t *)buf->b_data + osize, size - osize);
1978eda14cbcSMatt Macy 
1979eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
1980eda14cbcSMatt Macy 	dbuf_set_data(db, buf);
1981eda14cbcSMatt Macy 	arc_buf_destroy(old_buf, db);
1982eda14cbcSMatt Macy 	db->db.db_size = size;
1983eda14cbcSMatt Macy 
1984eda14cbcSMatt Macy 	dr = list_head(&db->db_dirty_records);
1985eda14cbcSMatt Macy 	/* dirty record added by dmu_buf_will_dirty() */
1986eda14cbcSMatt Macy 	VERIFY(dr != NULL);
1987eda14cbcSMatt Macy 	if (db->db_level == 0)
1988eda14cbcSMatt Macy 		dr->dt.dl.dr_data = buf;
1989eda14cbcSMatt Macy 	ASSERT3U(dr->dr_txg, ==, tx->tx_txg);
1990eda14cbcSMatt Macy 	ASSERT3U(dr->dr_accounted, ==, osize);
1991eda14cbcSMatt Macy 	dr->dr_accounted = size;
1992eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
1993eda14cbcSMatt Macy 
1994eda14cbcSMatt Macy 	dmu_objset_willuse_space(dn->dn_objset, size - osize, tx);
1995eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
1996eda14cbcSMatt Macy }
1997eda14cbcSMatt Macy 
1998eda14cbcSMatt Macy void
1999eda14cbcSMatt Macy dbuf_release_bp(dmu_buf_impl_t *db)
2000eda14cbcSMatt Macy {
2001eda14cbcSMatt Macy 	objset_t *os __maybe_unused = db->db_objset;
2002eda14cbcSMatt Macy 
2003eda14cbcSMatt Macy 	ASSERT(dsl_pool_sync_context(dmu_objset_pool(os)));
2004eda14cbcSMatt Macy 	ASSERT(arc_released(os->os_phys_buf) ||
2005eda14cbcSMatt Macy 	    list_link_active(&os->os_dsl_dataset->ds_synced_link));
2006eda14cbcSMatt Macy 	ASSERT(db->db_parent == NULL || arc_released(db->db_parent->db_buf));
2007eda14cbcSMatt Macy 
2008eda14cbcSMatt Macy 	(void) arc_release(db->db_buf, db);
2009eda14cbcSMatt Macy }
2010eda14cbcSMatt Macy 
2011eda14cbcSMatt Macy /*
2012eda14cbcSMatt Macy  * We already have a dirty record for this TXG, and we are being
2013eda14cbcSMatt Macy  * dirtied again.
2014eda14cbcSMatt Macy  */
2015eda14cbcSMatt Macy static void
2016eda14cbcSMatt Macy dbuf_redirty(dbuf_dirty_record_t *dr)
2017eda14cbcSMatt Macy {
2018eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
2019eda14cbcSMatt Macy 
2020eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
2021eda14cbcSMatt Macy 
2022eda14cbcSMatt Macy 	if (db->db_level == 0 && db->db_blkid != DMU_BONUS_BLKID) {
2023eda14cbcSMatt Macy 		/*
2024eda14cbcSMatt Macy 		 * If this buffer has already been written out,
2025eda14cbcSMatt Macy 		 * we now need to reset its state.
2026eda14cbcSMatt Macy 		 */
2027eda14cbcSMatt Macy 		dbuf_unoverride(dr);
2028eda14cbcSMatt Macy 		if (db->db.db_object != DMU_META_DNODE_OBJECT &&
2029eda14cbcSMatt Macy 		    db->db_state != DB_NOFILL) {
2030eda14cbcSMatt Macy 			/* Already released on initial dirty, so just thaw. */
2031eda14cbcSMatt Macy 			ASSERT(arc_released(db->db_buf));
2032eda14cbcSMatt Macy 			arc_buf_thaw(db->db_buf);
2033eda14cbcSMatt Macy 		}
2034eda14cbcSMatt Macy 	}
2035eda14cbcSMatt Macy }
2036eda14cbcSMatt Macy 
2037eda14cbcSMatt Macy dbuf_dirty_record_t *
20387877fdebSMatt Macy dbuf_dirty_lightweight(dnode_t *dn, uint64_t blkid, dmu_tx_t *tx)
20397877fdebSMatt Macy {
20407877fdebSMatt Macy 	rw_enter(&dn->dn_struct_rwlock, RW_READER);
20417877fdebSMatt Macy 	IMPLY(dn->dn_objset->os_raw_receive, dn->dn_maxblkid >= blkid);
20427877fdebSMatt Macy 	dnode_new_blkid(dn, blkid, tx, B_TRUE, B_FALSE);
20437877fdebSMatt Macy 	ASSERT(dn->dn_maxblkid >= blkid);
20447877fdebSMatt Macy 
20457877fdebSMatt Macy 	dbuf_dirty_record_t *dr = kmem_zalloc(sizeof (*dr), KM_SLEEP);
20467877fdebSMatt Macy 	list_link_init(&dr->dr_dirty_node);
20477877fdebSMatt Macy 	list_link_init(&dr->dr_dbuf_node);
20487877fdebSMatt Macy 	dr->dr_dnode = dn;
20497877fdebSMatt Macy 	dr->dr_txg = tx->tx_txg;
20507877fdebSMatt Macy 	dr->dt.dll.dr_blkid = blkid;
20517877fdebSMatt Macy 	dr->dr_accounted = dn->dn_datablksz;
20527877fdebSMatt Macy 
20537877fdebSMatt Macy 	/*
20547877fdebSMatt Macy 	 * There should not be any dbuf for the block that we're dirtying.
20557877fdebSMatt Macy 	 * Otherwise the buffer contents could be inconsistent between the
20567877fdebSMatt Macy 	 * dbuf and the lightweight dirty record.
20577877fdebSMatt Macy 	 */
20587877fdebSMatt Macy 	ASSERT3P(NULL, ==, dbuf_find(dn->dn_objset, dn->dn_object, 0, blkid));
20597877fdebSMatt Macy 
20607877fdebSMatt Macy 	mutex_enter(&dn->dn_mtx);
20617877fdebSMatt Macy 	int txgoff = tx->tx_txg & TXG_MASK;
20627877fdebSMatt Macy 	if (dn->dn_free_ranges[txgoff] != NULL) {
20637877fdebSMatt Macy 		range_tree_clear(dn->dn_free_ranges[txgoff], blkid, 1);
20647877fdebSMatt Macy 	}
20657877fdebSMatt Macy 
20667877fdebSMatt Macy 	if (dn->dn_nlevels == 1) {
20677877fdebSMatt Macy 		ASSERT3U(blkid, <, dn->dn_nblkptr);
20687877fdebSMatt Macy 		list_insert_tail(&dn->dn_dirty_records[txgoff], dr);
20697877fdebSMatt Macy 		mutex_exit(&dn->dn_mtx);
20707877fdebSMatt Macy 		rw_exit(&dn->dn_struct_rwlock);
20717877fdebSMatt Macy 		dnode_setdirty(dn, tx);
20727877fdebSMatt Macy 	} else {
20737877fdebSMatt Macy 		mutex_exit(&dn->dn_mtx);
20747877fdebSMatt Macy 
20757877fdebSMatt Macy 		int epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
20767877fdebSMatt Macy 		dmu_buf_impl_t *parent_db = dbuf_hold_level(dn,
20777877fdebSMatt Macy 		    1, blkid >> epbs, FTAG);
20787877fdebSMatt Macy 		rw_exit(&dn->dn_struct_rwlock);
20797877fdebSMatt Macy 		if (parent_db == NULL) {
20807877fdebSMatt Macy 			kmem_free(dr, sizeof (*dr));
20817877fdebSMatt Macy 			return (NULL);
20827877fdebSMatt Macy 		}
20837877fdebSMatt Macy 		int err = dbuf_read(parent_db, NULL,
20847877fdebSMatt Macy 		    (DB_RF_NOPREFETCH | DB_RF_CANFAIL));
20857877fdebSMatt Macy 		if (err != 0) {
20867877fdebSMatt Macy 			dbuf_rele(parent_db, FTAG);
20877877fdebSMatt Macy 			kmem_free(dr, sizeof (*dr));
20887877fdebSMatt Macy 			return (NULL);
20897877fdebSMatt Macy 		}
20907877fdebSMatt Macy 
20917877fdebSMatt Macy 		dbuf_dirty_record_t *parent_dr = dbuf_dirty(parent_db, tx);
20927877fdebSMatt Macy 		dbuf_rele(parent_db, FTAG);
20937877fdebSMatt Macy 		mutex_enter(&parent_dr->dt.di.dr_mtx);
20947877fdebSMatt Macy 		ASSERT3U(parent_dr->dr_txg, ==, tx->tx_txg);
20957877fdebSMatt Macy 		list_insert_tail(&parent_dr->dt.di.dr_children, dr);
20967877fdebSMatt Macy 		mutex_exit(&parent_dr->dt.di.dr_mtx);
20977877fdebSMatt Macy 		dr->dr_parent = parent_dr;
20987877fdebSMatt Macy 	}
20997877fdebSMatt Macy 
21007877fdebSMatt Macy 	dmu_objset_willuse_space(dn->dn_objset, dr->dr_accounted, tx);
21017877fdebSMatt Macy 
21027877fdebSMatt Macy 	return (dr);
21037877fdebSMatt Macy }
21047877fdebSMatt Macy 
21057877fdebSMatt Macy dbuf_dirty_record_t *
2106eda14cbcSMatt Macy dbuf_dirty(dmu_buf_impl_t *db, dmu_tx_t *tx)
2107eda14cbcSMatt Macy {
2108eda14cbcSMatt Macy 	dnode_t *dn;
2109eda14cbcSMatt Macy 	objset_t *os;
2110eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr, *dr_next, *dr_head;
2111eda14cbcSMatt Macy 	int txgoff = tx->tx_txg & TXG_MASK;
2112eda14cbcSMatt Macy 	boolean_t drop_struct_rwlock = B_FALSE;
2113eda14cbcSMatt Macy 
2114eda14cbcSMatt Macy 	ASSERT(tx->tx_txg != 0);
2115eda14cbcSMatt Macy 	ASSERT(!zfs_refcount_is_zero(&db->db_holds));
2116eda14cbcSMatt Macy 	DMU_TX_DIRTY_BUF(tx, db);
2117eda14cbcSMatt Macy 
2118eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
2119eda14cbcSMatt Macy 	dn = DB_DNODE(db);
2120eda14cbcSMatt Macy 	/*
2121eda14cbcSMatt Macy 	 * Shouldn't dirty a regular buffer in syncing context.  Private
2122eda14cbcSMatt Macy 	 * objects may be dirtied in syncing context, but only if they
2123eda14cbcSMatt Macy 	 * were already pre-dirtied in open context.
2124eda14cbcSMatt Macy 	 */
2125eda14cbcSMatt Macy #ifdef ZFS_DEBUG
2126eda14cbcSMatt Macy 	if (dn->dn_objset->os_dsl_dataset != NULL) {
2127eda14cbcSMatt Macy 		rrw_enter(&dn->dn_objset->os_dsl_dataset->ds_bp_rwlock,
2128eda14cbcSMatt Macy 		    RW_READER, FTAG);
2129eda14cbcSMatt Macy 	}
2130eda14cbcSMatt Macy 	ASSERT(!dmu_tx_is_syncing(tx) ||
2131eda14cbcSMatt Macy 	    BP_IS_HOLE(dn->dn_objset->os_rootbp) ||
2132eda14cbcSMatt Macy 	    DMU_OBJECT_IS_SPECIAL(dn->dn_object) ||
2133eda14cbcSMatt Macy 	    dn->dn_objset->os_dsl_dataset == NULL);
2134eda14cbcSMatt Macy 	if (dn->dn_objset->os_dsl_dataset != NULL)
2135eda14cbcSMatt Macy 		rrw_exit(&dn->dn_objset->os_dsl_dataset->ds_bp_rwlock, FTAG);
2136eda14cbcSMatt Macy #endif
2137eda14cbcSMatt Macy 	/*
2138eda14cbcSMatt Macy 	 * We make this assert for private objects as well, but after we
2139eda14cbcSMatt Macy 	 * check if we're already dirty.  They are allowed to re-dirty
2140eda14cbcSMatt Macy 	 * in syncing context.
2141eda14cbcSMatt Macy 	 */
2142eda14cbcSMatt Macy 	ASSERT(dn->dn_object == DMU_META_DNODE_OBJECT ||
2143eda14cbcSMatt Macy 	    dn->dn_dirtyctx == DN_UNDIRTIED || dn->dn_dirtyctx ==
2144eda14cbcSMatt Macy 	    (dmu_tx_is_syncing(tx) ? DN_DIRTY_SYNC : DN_DIRTY_OPEN));
2145eda14cbcSMatt Macy 
2146eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
2147eda14cbcSMatt Macy 	/*
2148eda14cbcSMatt Macy 	 * XXX make this true for indirects too?  The problem is that
2149eda14cbcSMatt Macy 	 * transactions created with dmu_tx_create_assigned() from
2150eda14cbcSMatt Macy 	 * syncing context don't bother holding ahead.
2151eda14cbcSMatt Macy 	 */
2152eda14cbcSMatt Macy 	ASSERT(db->db_level != 0 ||
2153eda14cbcSMatt Macy 	    db->db_state == DB_CACHED || db->db_state == DB_FILL ||
2154eda14cbcSMatt Macy 	    db->db_state == DB_NOFILL);
2155eda14cbcSMatt Macy 
2156eda14cbcSMatt Macy 	mutex_enter(&dn->dn_mtx);
2157eda14cbcSMatt Macy 	dnode_set_dirtyctx(dn, tx, db);
2158eda14cbcSMatt Macy 	if (tx->tx_txg > dn->dn_dirty_txg)
2159eda14cbcSMatt Macy 		dn->dn_dirty_txg = tx->tx_txg;
2160eda14cbcSMatt Macy 	mutex_exit(&dn->dn_mtx);
2161eda14cbcSMatt Macy 
2162eda14cbcSMatt Macy 	if (db->db_blkid == DMU_SPILL_BLKID)
2163eda14cbcSMatt Macy 		dn->dn_have_spill = B_TRUE;
2164eda14cbcSMatt Macy 
2165eda14cbcSMatt Macy 	/*
2166eda14cbcSMatt Macy 	 * If this buffer is already dirty, we're done.
2167eda14cbcSMatt Macy 	 */
2168eda14cbcSMatt Macy 	dr_head = list_head(&db->db_dirty_records);
2169eda14cbcSMatt Macy 	ASSERT(dr_head == NULL || dr_head->dr_txg <= tx->tx_txg ||
2170eda14cbcSMatt Macy 	    db->db.db_object == DMU_META_DNODE_OBJECT);
2171eda14cbcSMatt Macy 	dr_next = dbuf_find_dirty_lte(db, tx->tx_txg);
2172eda14cbcSMatt Macy 	if (dr_next && dr_next->dr_txg == tx->tx_txg) {
2173eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
2174eda14cbcSMatt Macy 
2175eda14cbcSMatt Macy 		dbuf_redirty(dr_next);
2176eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
2177eda14cbcSMatt Macy 		return (dr_next);
2178eda14cbcSMatt Macy 	}
2179eda14cbcSMatt Macy 
2180eda14cbcSMatt Macy 	/*
2181eda14cbcSMatt Macy 	 * Only valid if not already dirty.
2182eda14cbcSMatt Macy 	 */
2183eda14cbcSMatt Macy 	ASSERT(dn->dn_object == 0 ||
2184eda14cbcSMatt Macy 	    dn->dn_dirtyctx == DN_UNDIRTIED || dn->dn_dirtyctx ==
2185eda14cbcSMatt Macy 	    (dmu_tx_is_syncing(tx) ? DN_DIRTY_SYNC : DN_DIRTY_OPEN));
2186eda14cbcSMatt Macy 
2187eda14cbcSMatt Macy 	ASSERT3U(dn->dn_nlevels, >, db->db_level);
2188eda14cbcSMatt Macy 
2189eda14cbcSMatt Macy 	/*
2190eda14cbcSMatt Macy 	 * We should only be dirtying in syncing context if it's the
2191eda14cbcSMatt Macy 	 * mos or we're initializing the os or it's a special object.
2192eda14cbcSMatt Macy 	 * However, we are allowed to dirty in syncing context provided
2193eda14cbcSMatt Macy 	 * we already dirtied it in open context.  Hence we must make
2194eda14cbcSMatt Macy 	 * this assertion only if we're not already dirty.
2195eda14cbcSMatt Macy 	 */
2196eda14cbcSMatt Macy 	os = dn->dn_objset;
2197eda14cbcSMatt Macy 	VERIFY3U(tx->tx_txg, <=, spa_final_dirty_txg(os->os_spa));
2198eda14cbcSMatt Macy #ifdef ZFS_DEBUG
2199eda14cbcSMatt Macy 	if (dn->dn_objset->os_dsl_dataset != NULL)
2200eda14cbcSMatt Macy 		rrw_enter(&os->os_dsl_dataset->ds_bp_rwlock, RW_READER, FTAG);
2201eda14cbcSMatt Macy 	ASSERT(!dmu_tx_is_syncing(tx) || DMU_OBJECT_IS_SPECIAL(dn->dn_object) ||
2202eda14cbcSMatt Macy 	    os->os_dsl_dataset == NULL || BP_IS_HOLE(os->os_rootbp));
2203eda14cbcSMatt Macy 	if (dn->dn_objset->os_dsl_dataset != NULL)
2204eda14cbcSMatt Macy 		rrw_exit(&os->os_dsl_dataset->ds_bp_rwlock, FTAG);
2205eda14cbcSMatt Macy #endif
2206eda14cbcSMatt Macy 	ASSERT(db->db.db_size != 0);
2207eda14cbcSMatt Macy 
2208eda14cbcSMatt Macy 	dprintf_dbuf(db, "size=%llx\n", (u_longlong_t)db->db.db_size);
2209eda14cbcSMatt Macy 
2210eda14cbcSMatt Macy 	if (db->db_blkid != DMU_BONUS_BLKID) {
2211eda14cbcSMatt Macy 		dmu_objset_willuse_space(os, db->db.db_size, tx);
2212eda14cbcSMatt Macy 	}
2213eda14cbcSMatt Macy 
2214eda14cbcSMatt Macy 	/*
2215eda14cbcSMatt Macy 	 * If this buffer is dirty in an old transaction group we need
2216eda14cbcSMatt Macy 	 * to make a copy of it so that the changes we make in this
2217eda14cbcSMatt Macy 	 * transaction group won't leak out when we sync the older txg.
2218eda14cbcSMatt Macy 	 */
2219eda14cbcSMatt Macy 	dr = kmem_zalloc(sizeof (dbuf_dirty_record_t), KM_SLEEP);
2220eda14cbcSMatt Macy 	list_link_init(&dr->dr_dirty_node);
2221eda14cbcSMatt Macy 	list_link_init(&dr->dr_dbuf_node);
22227877fdebSMatt Macy 	dr->dr_dnode = dn;
2223eda14cbcSMatt Macy 	if (db->db_level == 0) {
2224eda14cbcSMatt Macy 		void *data_old = db->db_buf;
2225eda14cbcSMatt Macy 
2226eda14cbcSMatt Macy 		if (db->db_state != DB_NOFILL) {
2227eda14cbcSMatt Macy 			if (db->db_blkid == DMU_BONUS_BLKID) {
2228eda14cbcSMatt Macy 				dbuf_fix_old_data(db, tx->tx_txg);
2229eda14cbcSMatt Macy 				data_old = db->db.db_data;
2230eda14cbcSMatt Macy 			} else if (db->db.db_object != DMU_META_DNODE_OBJECT) {
2231eda14cbcSMatt Macy 				/*
2232eda14cbcSMatt Macy 				 * Release the data buffer from the cache so
2233eda14cbcSMatt Macy 				 * that we can modify it without impacting
2234eda14cbcSMatt Macy 				 * possible other users of this cached data
2235eda14cbcSMatt Macy 				 * block.  Note that indirect blocks and
2236eda14cbcSMatt Macy 				 * private objects are not released until the
2237eda14cbcSMatt Macy 				 * syncing state (since they are only modified
2238eda14cbcSMatt Macy 				 * then).
2239eda14cbcSMatt Macy 				 */
2240eda14cbcSMatt Macy 				arc_release(db->db_buf, db);
2241eda14cbcSMatt Macy 				dbuf_fix_old_data(db, tx->tx_txg);
2242eda14cbcSMatt Macy 				data_old = db->db_buf;
2243eda14cbcSMatt Macy 			}
2244eda14cbcSMatt Macy 			ASSERT(data_old != NULL);
2245eda14cbcSMatt Macy 		}
2246eda14cbcSMatt Macy 		dr->dt.dl.dr_data = data_old;
2247eda14cbcSMatt Macy 	} else {
2248eda14cbcSMatt Macy 		mutex_init(&dr->dt.di.dr_mtx, NULL, MUTEX_NOLOCKDEP, NULL);
2249eda14cbcSMatt Macy 		list_create(&dr->dt.di.dr_children,
2250eda14cbcSMatt Macy 		    sizeof (dbuf_dirty_record_t),
2251eda14cbcSMatt Macy 		    offsetof(dbuf_dirty_record_t, dr_dirty_node));
2252eda14cbcSMatt Macy 	}
2253eda14cbcSMatt Macy 	if (db->db_blkid != DMU_BONUS_BLKID)
2254eda14cbcSMatt Macy 		dr->dr_accounted = db->db.db_size;
2255eda14cbcSMatt Macy 	dr->dr_dbuf = db;
2256eda14cbcSMatt Macy 	dr->dr_txg = tx->tx_txg;
2257eda14cbcSMatt Macy 	list_insert_before(&db->db_dirty_records, dr_next, dr);
2258eda14cbcSMatt Macy 
2259eda14cbcSMatt Macy 	/*
2260eda14cbcSMatt Macy 	 * We could have been freed_in_flight between the dbuf_noread
2261eda14cbcSMatt Macy 	 * and dbuf_dirty.  We win, as though the dbuf_noread() had
2262eda14cbcSMatt Macy 	 * happened after the free.
2263eda14cbcSMatt Macy 	 */
2264eda14cbcSMatt Macy 	if (db->db_level == 0 && db->db_blkid != DMU_BONUS_BLKID &&
2265eda14cbcSMatt Macy 	    db->db_blkid != DMU_SPILL_BLKID) {
2266eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
2267eda14cbcSMatt Macy 		if (dn->dn_free_ranges[txgoff] != NULL) {
2268eda14cbcSMatt Macy 			range_tree_clear(dn->dn_free_ranges[txgoff],
2269eda14cbcSMatt Macy 			    db->db_blkid, 1);
2270eda14cbcSMatt Macy 		}
2271eda14cbcSMatt Macy 		mutex_exit(&dn->dn_mtx);
2272eda14cbcSMatt Macy 		db->db_freed_in_flight = FALSE;
2273eda14cbcSMatt Macy 	}
2274eda14cbcSMatt Macy 
2275eda14cbcSMatt Macy 	/*
2276eda14cbcSMatt Macy 	 * This buffer is now part of this txg
2277eda14cbcSMatt Macy 	 */
2278eda14cbcSMatt Macy 	dbuf_add_ref(db, (void *)(uintptr_t)tx->tx_txg);
2279eda14cbcSMatt Macy 	db->db_dirtycnt += 1;
2280eda14cbcSMatt Macy 	ASSERT3U(db->db_dirtycnt, <=, 3);
2281eda14cbcSMatt Macy 
2282eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
2283eda14cbcSMatt Macy 
2284eda14cbcSMatt Macy 	if (db->db_blkid == DMU_BONUS_BLKID ||
2285eda14cbcSMatt Macy 	    db->db_blkid == DMU_SPILL_BLKID) {
2286eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
2287eda14cbcSMatt Macy 		ASSERT(!list_link_active(&dr->dr_dirty_node));
2288eda14cbcSMatt Macy 		list_insert_tail(&dn->dn_dirty_records[txgoff], dr);
2289eda14cbcSMatt Macy 		mutex_exit(&dn->dn_mtx);
2290eda14cbcSMatt Macy 		dnode_setdirty(dn, tx);
2291eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
2292eda14cbcSMatt Macy 		return (dr);
2293eda14cbcSMatt Macy 	}
2294eda14cbcSMatt Macy 
2295eda14cbcSMatt Macy 	if (!RW_WRITE_HELD(&dn->dn_struct_rwlock)) {
2296eda14cbcSMatt Macy 		rw_enter(&dn->dn_struct_rwlock, RW_READER);
2297eda14cbcSMatt Macy 		drop_struct_rwlock = B_TRUE;
2298eda14cbcSMatt Macy 	}
2299eda14cbcSMatt Macy 
2300eda14cbcSMatt Macy 	/*
2301eda14cbcSMatt Macy 	 * If we are overwriting a dedup BP, then unless it is snapshotted,
2302eda14cbcSMatt Macy 	 * when we get to syncing context we will need to decrement its
2303eda14cbcSMatt Macy 	 * refcount in the DDT.  Prefetch the relevant DDT block so that
2304eda14cbcSMatt Macy 	 * syncing context won't have to wait for the i/o.
2305eda14cbcSMatt Macy 	 */
2306eda14cbcSMatt Macy 	if (db->db_blkptr != NULL) {
2307eda14cbcSMatt Macy 		db_lock_type_t dblt = dmu_buf_lock_parent(db, RW_READER, FTAG);
2308eda14cbcSMatt Macy 		ddt_prefetch(os->os_spa, db->db_blkptr);
2309eda14cbcSMatt Macy 		dmu_buf_unlock_parent(db, dblt, FTAG);
2310eda14cbcSMatt Macy 	}
2311eda14cbcSMatt Macy 
2312eda14cbcSMatt Macy 	/*
2313eda14cbcSMatt Macy 	 * We need to hold the dn_struct_rwlock to make this assertion,
2314eda14cbcSMatt Macy 	 * because it protects dn_phys / dn_next_nlevels from changing.
2315eda14cbcSMatt Macy 	 */
2316eda14cbcSMatt Macy 	ASSERT((dn->dn_phys->dn_nlevels == 0 && db->db_level == 0) ||
2317eda14cbcSMatt Macy 	    dn->dn_phys->dn_nlevels > db->db_level ||
2318eda14cbcSMatt Macy 	    dn->dn_next_nlevels[txgoff] > db->db_level ||
2319eda14cbcSMatt Macy 	    dn->dn_next_nlevels[(tx->tx_txg-1) & TXG_MASK] > db->db_level ||
2320eda14cbcSMatt Macy 	    dn->dn_next_nlevels[(tx->tx_txg-2) & TXG_MASK] > db->db_level);
2321eda14cbcSMatt Macy 
2322eda14cbcSMatt Macy 
2323eda14cbcSMatt Macy 	if (db->db_level == 0) {
2324eda14cbcSMatt Macy 		ASSERT(!db->db_objset->os_raw_receive ||
2325eda14cbcSMatt Macy 		    dn->dn_maxblkid >= db->db_blkid);
2326eda14cbcSMatt Macy 		dnode_new_blkid(dn, db->db_blkid, tx,
2327eda14cbcSMatt Macy 		    drop_struct_rwlock, B_FALSE);
2328eda14cbcSMatt Macy 		ASSERT(dn->dn_maxblkid >= db->db_blkid);
2329eda14cbcSMatt Macy 	}
2330eda14cbcSMatt Macy 
2331eda14cbcSMatt Macy 	if (db->db_level+1 < dn->dn_nlevels) {
2332eda14cbcSMatt Macy 		dmu_buf_impl_t *parent = db->db_parent;
2333eda14cbcSMatt Macy 		dbuf_dirty_record_t *di;
2334eda14cbcSMatt Macy 		int parent_held = FALSE;
2335eda14cbcSMatt Macy 
2336eda14cbcSMatt Macy 		if (db->db_parent == NULL || db->db_parent == dn->dn_dbuf) {
2337eda14cbcSMatt Macy 			int epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
2338eda14cbcSMatt Macy 			parent = dbuf_hold_level(dn, db->db_level + 1,
2339eda14cbcSMatt Macy 			    db->db_blkid >> epbs, FTAG);
2340eda14cbcSMatt Macy 			ASSERT(parent != NULL);
2341eda14cbcSMatt Macy 			parent_held = TRUE;
2342eda14cbcSMatt Macy 		}
2343eda14cbcSMatt Macy 		if (drop_struct_rwlock)
2344eda14cbcSMatt Macy 			rw_exit(&dn->dn_struct_rwlock);
2345eda14cbcSMatt Macy 		ASSERT3U(db->db_level + 1, ==, parent->db_level);
2346eda14cbcSMatt Macy 		di = dbuf_dirty(parent, tx);
2347eda14cbcSMatt Macy 		if (parent_held)
2348eda14cbcSMatt Macy 			dbuf_rele(parent, FTAG);
2349eda14cbcSMatt Macy 
2350eda14cbcSMatt Macy 		mutex_enter(&db->db_mtx);
2351eda14cbcSMatt Macy 		/*
2352eda14cbcSMatt Macy 		 * Since we've dropped the mutex, it's possible that
2353eda14cbcSMatt Macy 		 * dbuf_undirty() might have changed this out from under us.
2354eda14cbcSMatt Macy 		 */
2355eda14cbcSMatt Macy 		if (list_head(&db->db_dirty_records) == dr ||
2356eda14cbcSMatt Macy 		    dn->dn_object == DMU_META_DNODE_OBJECT) {
2357eda14cbcSMatt Macy 			mutex_enter(&di->dt.di.dr_mtx);
2358eda14cbcSMatt Macy 			ASSERT3U(di->dr_txg, ==, tx->tx_txg);
2359eda14cbcSMatt Macy 			ASSERT(!list_link_active(&dr->dr_dirty_node));
2360eda14cbcSMatt Macy 			list_insert_tail(&di->dt.di.dr_children, dr);
2361eda14cbcSMatt Macy 			mutex_exit(&di->dt.di.dr_mtx);
2362eda14cbcSMatt Macy 			dr->dr_parent = di;
2363eda14cbcSMatt Macy 		}
2364eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
2365eda14cbcSMatt Macy 	} else {
2366eda14cbcSMatt Macy 		ASSERT(db->db_level + 1 == dn->dn_nlevels);
2367eda14cbcSMatt Macy 		ASSERT(db->db_blkid < dn->dn_nblkptr);
2368eda14cbcSMatt Macy 		ASSERT(db->db_parent == NULL || db->db_parent == dn->dn_dbuf);
2369eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
2370eda14cbcSMatt Macy 		ASSERT(!list_link_active(&dr->dr_dirty_node));
2371eda14cbcSMatt Macy 		list_insert_tail(&dn->dn_dirty_records[txgoff], dr);
2372eda14cbcSMatt Macy 		mutex_exit(&dn->dn_mtx);
2373eda14cbcSMatt Macy 		if (drop_struct_rwlock)
2374eda14cbcSMatt Macy 			rw_exit(&dn->dn_struct_rwlock);
2375eda14cbcSMatt Macy 	}
2376eda14cbcSMatt Macy 
2377eda14cbcSMatt Macy 	dnode_setdirty(dn, tx);
2378eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
2379eda14cbcSMatt Macy 	return (dr);
2380eda14cbcSMatt Macy }
2381eda14cbcSMatt Macy 
2382eda14cbcSMatt Macy static void
2383eda14cbcSMatt Macy dbuf_undirty_bonus(dbuf_dirty_record_t *dr)
2384eda14cbcSMatt Macy {
2385eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
2386eda14cbcSMatt Macy 
2387eda14cbcSMatt Macy 	if (dr->dt.dl.dr_data != db->db.db_data) {
23887877fdebSMatt Macy 		struct dnode *dn = dr->dr_dnode;
2389eda14cbcSMatt Macy 		int max_bonuslen = DN_SLOTS_TO_BONUSLEN(dn->dn_num_slots);
2390eda14cbcSMatt Macy 
2391eda14cbcSMatt Macy 		kmem_free(dr->dt.dl.dr_data, max_bonuslen);
2392eda14cbcSMatt Macy 		arc_space_return(max_bonuslen, ARC_SPACE_BONUS);
2393eda14cbcSMatt Macy 	}
2394eda14cbcSMatt Macy 	db->db_data_pending = NULL;
2395eda14cbcSMatt Macy 	ASSERT(list_next(&db->db_dirty_records, dr) == NULL);
2396eda14cbcSMatt Macy 	list_remove(&db->db_dirty_records, dr);
2397eda14cbcSMatt Macy 	if (dr->dr_dbuf->db_level != 0) {
2398eda14cbcSMatt Macy 		mutex_destroy(&dr->dt.di.dr_mtx);
2399eda14cbcSMatt Macy 		list_destroy(&dr->dt.di.dr_children);
2400eda14cbcSMatt Macy 	}
2401eda14cbcSMatt Macy 	kmem_free(dr, sizeof (dbuf_dirty_record_t));
2402eda14cbcSMatt Macy 	ASSERT3U(db->db_dirtycnt, >, 0);
2403eda14cbcSMatt Macy 	db->db_dirtycnt -= 1;
2404eda14cbcSMatt Macy }
2405eda14cbcSMatt Macy 
2406eda14cbcSMatt Macy /*
2407eda14cbcSMatt Macy  * Undirty a buffer in the transaction group referenced by the given
2408eda14cbcSMatt Macy  * transaction.  Return whether this evicted the dbuf.
2409eda14cbcSMatt Macy  */
2410eda14cbcSMatt Macy static boolean_t
2411eda14cbcSMatt Macy dbuf_undirty(dmu_buf_impl_t *db, dmu_tx_t *tx)
2412eda14cbcSMatt Macy {
2413eda14cbcSMatt Macy 	uint64_t txg = tx->tx_txg;
2414eda14cbcSMatt Macy 
2415eda14cbcSMatt Macy 	ASSERT(txg != 0);
2416eda14cbcSMatt Macy 
2417eda14cbcSMatt Macy 	/*
2418eda14cbcSMatt Macy 	 * Due to our use of dn_nlevels below, this can only be called
2419eda14cbcSMatt Macy 	 * in open context, unless we are operating on the MOS.
2420eda14cbcSMatt Macy 	 * From syncing context, dn_nlevels may be different from the
2421eda14cbcSMatt Macy 	 * dn_nlevels used when dbuf was dirtied.
2422eda14cbcSMatt Macy 	 */
2423eda14cbcSMatt Macy 	ASSERT(db->db_objset ==
2424eda14cbcSMatt Macy 	    dmu_objset_pool(db->db_objset)->dp_meta_objset ||
2425eda14cbcSMatt Macy 	    txg != spa_syncing_txg(dmu_objset_spa(db->db_objset)));
2426eda14cbcSMatt Macy 	ASSERT(db->db_blkid != DMU_BONUS_BLKID);
2427eda14cbcSMatt Macy 	ASSERT0(db->db_level);
2428eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
2429eda14cbcSMatt Macy 
2430eda14cbcSMatt Macy 	/*
2431eda14cbcSMatt Macy 	 * If this buffer is not dirty, we're done.
2432eda14cbcSMatt Macy 	 */
24337877fdebSMatt Macy 	dbuf_dirty_record_t *dr = dbuf_find_dirty_eq(db, txg);
2434eda14cbcSMatt Macy 	if (dr == NULL)
2435eda14cbcSMatt Macy 		return (B_FALSE);
2436eda14cbcSMatt Macy 	ASSERT(dr->dr_dbuf == db);
2437eda14cbcSMatt Macy 
24387877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
2439eda14cbcSMatt Macy 
2440eda14cbcSMatt Macy 	dprintf_dbuf(db, "size=%llx\n", (u_longlong_t)db->db.db_size);
2441eda14cbcSMatt Macy 
2442eda14cbcSMatt Macy 	ASSERT(db->db.db_size != 0);
2443eda14cbcSMatt Macy 
2444eda14cbcSMatt Macy 	dsl_pool_undirty_space(dmu_objset_pool(dn->dn_objset),
2445eda14cbcSMatt Macy 	    dr->dr_accounted, txg);
2446eda14cbcSMatt Macy 
2447eda14cbcSMatt Macy 	list_remove(&db->db_dirty_records, dr);
2448eda14cbcSMatt Macy 
2449eda14cbcSMatt Macy 	/*
2450eda14cbcSMatt Macy 	 * Note that there are three places in dbuf_dirty()
2451eda14cbcSMatt Macy 	 * where this dirty record may be put on a list.
2452eda14cbcSMatt Macy 	 * Make sure to do a list_remove corresponding to
2453eda14cbcSMatt Macy 	 * every one of those list_insert calls.
2454eda14cbcSMatt Macy 	 */
2455eda14cbcSMatt Macy 	if (dr->dr_parent) {
2456eda14cbcSMatt Macy 		mutex_enter(&dr->dr_parent->dt.di.dr_mtx);
2457eda14cbcSMatt Macy 		list_remove(&dr->dr_parent->dt.di.dr_children, dr);
2458eda14cbcSMatt Macy 		mutex_exit(&dr->dr_parent->dt.di.dr_mtx);
2459eda14cbcSMatt Macy 	} else if (db->db_blkid == DMU_SPILL_BLKID ||
2460eda14cbcSMatt Macy 	    db->db_level + 1 == dn->dn_nlevels) {
2461eda14cbcSMatt Macy 		ASSERT(db->db_blkptr == NULL || db->db_parent == dn->dn_dbuf);
2462eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
2463eda14cbcSMatt Macy 		list_remove(&dn->dn_dirty_records[txg & TXG_MASK], dr);
2464eda14cbcSMatt Macy 		mutex_exit(&dn->dn_mtx);
2465eda14cbcSMatt Macy 	}
2466eda14cbcSMatt Macy 
2467eda14cbcSMatt Macy 	if (db->db_state != DB_NOFILL) {
2468eda14cbcSMatt Macy 		dbuf_unoverride(dr);
2469eda14cbcSMatt Macy 
2470eda14cbcSMatt Macy 		ASSERT(db->db_buf != NULL);
2471eda14cbcSMatt Macy 		ASSERT(dr->dt.dl.dr_data != NULL);
2472eda14cbcSMatt Macy 		if (dr->dt.dl.dr_data != db->db_buf)
2473eda14cbcSMatt Macy 			arc_buf_destroy(dr->dt.dl.dr_data, db);
2474eda14cbcSMatt Macy 	}
2475eda14cbcSMatt Macy 
2476eda14cbcSMatt Macy 	kmem_free(dr, sizeof (dbuf_dirty_record_t));
2477eda14cbcSMatt Macy 
2478eda14cbcSMatt Macy 	ASSERT(db->db_dirtycnt > 0);
2479eda14cbcSMatt Macy 	db->db_dirtycnt -= 1;
2480eda14cbcSMatt Macy 
2481eda14cbcSMatt Macy 	if (zfs_refcount_remove(&db->db_holds, (void *)(uintptr_t)txg) == 0) {
2482eda14cbcSMatt Macy 		ASSERT(db->db_state == DB_NOFILL || arc_released(db->db_buf));
2483eda14cbcSMatt Macy 		dbuf_destroy(db);
2484eda14cbcSMatt Macy 		return (B_TRUE);
2485eda14cbcSMatt Macy 	}
2486eda14cbcSMatt Macy 
2487eda14cbcSMatt Macy 	return (B_FALSE);
2488eda14cbcSMatt Macy }
2489eda14cbcSMatt Macy 
2490eda14cbcSMatt Macy static void
2491eda14cbcSMatt Macy dmu_buf_will_dirty_impl(dmu_buf_t *db_fake, int flags, dmu_tx_t *tx)
2492eda14cbcSMatt Macy {
2493eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
2494eda14cbcSMatt Macy 
2495eda14cbcSMatt Macy 	ASSERT(tx->tx_txg != 0);
2496eda14cbcSMatt Macy 	ASSERT(!zfs_refcount_is_zero(&db->db_holds));
2497eda14cbcSMatt Macy 
2498eda14cbcSMatt Macy 	/*
2499eda14cbcSMatt Macy 	 * Quick check for dirtiness.  For already dirty blocks, this
2500eda14cbcSMatt Macy 	 * reduces runtime of this function by >90%, and overall performance
2501eda14cbcSMatt Macy 	 * by 50% for some workloads (e.g. file deletion with indirect blocks
2502eda14cbcSMatt Macy 	 * cached).
2503eda14cbcSMatt Macy 	 */
2504eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
2505eda14cbcSMatt Macy 
2506eda14cbcSMatt Macy 	if (db->db_state == DB_CACHED) {
2507eda14cbcSMatt Macy 		dbuf_dirty_record_t *dr = dbuf_find_dirty_eq(db, tx->tx_txg);
2508eda14cbcSMatt Macy 		/*
2509eda14cbcSMatt Macy 		 * It's possible that it is already dirty but not cached,
2510eda14cbcSMatt Macy 		 * because there are some calls to dbuf_dirty() that don't
2511eda14cbcSMatt Macy 		 * go through dmu_buf_will_dirty().
2512eda14cbcSMatt Macy 		 */
2513eda14cbcSMatt Macy 		if (dr != NULL) {
2514eda14cbcSMatt Macy 			/* This dbuf is already dirty and cached. */
2515eda14cbcSMatt Macy 			dbuf_redirty(dr);
2516eda14cbcSMatt Macy 			mutex_exit(&db->db_mtx);
2517eda14cbcSMatt Macy 			return;
2518eda14cbcSMatt Macy 		}
2519eda14cbcSMatt Macy 	}
2520eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
2521eda14cbcSMatt Macy 
2522eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
2523eda14cbcSMatt Macy 	if (RW_WRITE_HELD(&DB_DNODE(db)->dn_struct_rwlock))
2524eda14cbcSMatt Macy 		flags |= DB_RF_HAVESTRUCT;
2525eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
2526eda14cbcSMatt Macy 	(void) dbuf_read(db, NULL, flags);
2527eda14cbcSMatt Macy 	(void) dbuf_dirty(db, tx);
2528eda14cbcSMatt Macy }
2529eda14cbcSMatt Macy 
2530eda14cbcSMatt Macy void
2531eda14cbcSMatt Macy dmu_buf_will_dirty(dmu_buf_t *db_fake, dmu_tx_t *tx)
2532eda14cbcSMatt Macy {
2533eda14cbcSMatt Macy 	dmu_buf_will_dirty_impl(db_fake,
2534eda14cbcSMatt Macy 	    DB_RF_MUST_SUCCEED | DB_RF_NOPREFETCH, tx);
2535eda14cbcSMatt Macy }
2536eda14cbcSMatt Macy 
2537eda14cbcSMatt Macy boolean_t
2538eda14cbcSMatt Macy dmu_buf_is_dirty(dmu_buf_t *db_fake, dmu_tx_t *tx)
2539eda14cbcSMatt Macy {
2540eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
2541eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
2542eda14cbcSMatt Macy 
2543eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
2544eda14cbcSMatt Macy 	dr = dbuf_find_dirty_eq(db, tx->tx_txg);
2545eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
2546eda14cbcSMatt Macy 	return (dr != NULL);
2547eda14cbcSMatt Macy }
2548eda14cbcSMatt Macy 
2549eda14cbcSMatt Macy void
2550eda14cbcSMatt Macy dmu_buf_will_not_fill(dmu_buf_t *db_fake, dmu_tx_t *tx)
2551eda14cbcSMatt Macy {
2552eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
2553eda14cbcSMatt Macy 
2554eda14cbcSMatt Macy 	db->db_state = DB_NOFILL;
2555eda14cbcSMatt Macy 	DTRACE_SET_STATE(db, "allocating NOFILL buffer");
2556eda14cbcSMatt Macy 	dmu_buf_will_fill(db_fake, tx);
2557eda14cbcSMatt Macy }
2558eda14cbcSMatt Macy 
2559eda14cbcSMatt Macy void
2560eda14cbcSMatt Macy dmu_buf_will_fill(dmu_buf_t *db_fake, dmu_tx_t *tx)
2561eda14cbcSMatt Macy {
2562eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
2563eda14cbcSMatt Macy 
2564eda14cbcSMatt Macy 	ASSERT(db->db_blkid != DMU_BONUS_BLKID);
2565eda14cbcSMatt Macy 	ASSERT(tx->tx_txg != 0);
2566eda14cbcSMatt Macy 	ASSERT(db->db_level == 0);
2567eda14cbcSMatt Macy 	ASSERT(!zfs_refcount_is_zero(&db->db_holds));
2568eda14cbcSMatt Macy 
2569eda14cbcSMatt Macy 	ASSERT(db->db.db_object != DMU_META_DNODE_OBJECT ||
2570eda14cbcSMatt Macy 	    dmu_tx_private_ok(tx));
2571eda14cbcSMatt Macy 
2572eda14cbcSMatt Macy 	dbuf_noread(db);
2573eda14cbcSMatt Macy 	(void) dbuf_dirty(db, tx);
2574eda14cbcSMatt Macy }
2575eda14cbcSMatt Macy 
2576eda14cbcSMatt Macy /*
2577eda14cbcSMatt Macy  * This function is effectively the same as dmu_buf_will_dirty(), but
2578eda14cbcSMatt Macy  * indicates the caller expects raw encrypted data in the db, and provides
2579eda14cbcSMatt Macy  * the crypt params (byteorder, salt, iv, mac) which should be stored in the
2580eda14cbcSMatt Macy  * blkptr_t when this dbuf is written.  This is only used for blocks of
2581eda14cbcSMatt Macy  * dnodes, during raw receive.
2582eda14cbcSMatt Macy  */
2583eda14cbcSMatt Macy void
2584eda14cbcSMatt Macy dmu_buf_set_crypt_params(dmu_buf_t *db_fake, boolean_t byteorder,
2585eda14cbcSMatt Macy     const uint8_t *salt, const uint8_t *iv, const uint8_t *mac, dmu_tx_t *tx)
2586eda14cbcSMatt Macy {
2587eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
2588eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
2589eda14cbcSMatt Macy 
2590eda14cbcSMatt Macy 	/*
2591eda14cbcSMatt Macy 	 * dr_has_raw_params is only processed for blocks of dnodes
2592eda14cbcSMatt Macy 	 * (see dbuf_sync_dnode_leaf_crypt()).
2593eda14cbcSMatt Macy 	 */
2594eda14cbcSMatt Macy 	ASSERT3U(db->db.db_object, ==, DMU_META_DNODE_OBJECT);
2595eda14cbcSMatt Macy 	ASSERT3U(db->db_level, ==, 0);
2596eda14cbcSMatt Macy 	ASSERT(db->db_objset->os_raw_receive);
2597eda14cbcSMatt Macy 
2598eda14cbcSMatt Macy 	dmu_buf_will_dirty_impl(db_fake,
2599eda14cbcSMatt Macy 	    DB_RF_MUST_SUCCEED | DB_RF_NOPREFETCH | DB_RF_NO_DECRYPT, tx);
2600eda14cbcSMatt Macy 
2601eda14cbcSMatt Macy 	dr = dbuf_find_dirty_eq(db, tx->tx_txg);
2602eda14cbcSMatt Macy 
2603eda14cbcSMatt Macy 	ASSERT3P(dr, !=, NULL);
2604eda14cbcSMatt Macy 
2605eda14cbcSMatt Macy 	dr->dt.dl.dr_has_raw_params = B_TRUE;
2606eda14cbcSMatt Macy 	dr->dt.dl.dr_byteorder = byteorder;
2607eda14cbcSMatt Macy 	bcopy(salt, dr->dt.dl.dr_salt, ZIO_DATA_SALT_LEN);
2608eda14cbcSMatt Macy 	bcopy(iv, dr->dt.dl.dr_iv, ZIO_DATA_IV_LEN);
2609eda14cbcSMatt Macy 	bcopy(mac, dr->dt.dl.dr_mac, ZIO_DATA_MAC_LEN);
2610eda14cbcSMatt Macy }
2611eda14cbcSMatt Macy 
2612eda14cbcSMatt Macy static void
2613eda14cbcSMatt Macy dbuf_override_impl(dmu_buf_impl_t *db, const blkptr_t *bp, dmu_tx_t *tx)
2614eda14cbcSMatt Macy {
2615eda14cbcSMatt Macy 	struct dirty_leaf *dl;
2616eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
2617eda14cbcSMatt Macy 
2618eda14cbcSMatt Macy 	dr = list_head(&db->db_dirty_records);
2619eda14cbcSMatt Macy 	ASSERT3U(dr->dr_txg, ==, tx->tx_txg);
2620eda14cbcSMatt Macy 	dl = &dr->dt.dl;
2621eda14cbcSMatt Macy 	dl->dr_overridden_by = *bp;
2622eda14cbcSMatt Macy 	dl->dr_override_state = DR_OVERRIDDEN;
2623eda14cbcSMatt Macy 	dl->dr_overridden_by.blk_birth = dr->dr_txg;
2624eda14cbcSMatt Macy }
2625eda14cbcSMatt Macy 
2626eda14cbcSMatt Macy /* ARGSUSED */
2627eda14cbcSMatt Macy void
2628eda14cbcSMatt Macy dmu_buf_fill_done(dmu_buf_t *dbuf, dmu_tx_t *tx)
2629eda14cbcSMatt Macy {
2630eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)dbuf;
2631eda14cbcSMatt Macy 	dbuf_states_t old_state;
2632eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
2633eda14cbcSMatt Macy 	DBUF_VERIFY(db);
2634eda14cbcSMatt Macy 
2635eda14cbcSMatt Macy 	old_state = db->db_state;
2636eda14cbcSMatt Macy 	db->db_state = DB_CACHED;
2637eda14cbcSMatt Macy 	if (old_state == DB_FILL) {
2638eda14cbcSMatt Macy 		if (db->db_level == 0 && db->db_freed_in_flight) {
2639eda14cbcSMatt Macy 			ASSERT(db->db_blkid != DMU_BONUS_BLKID);
2640eda14cbcSMatt Macy 			/* we were freed while filling */
2641eda14cbcSMatt Macy 			/* XXX dbuf_undirty? */
2642eda14cbcSMatt Macy 			bzero(db->db.db_data, db->db.db_size);
2643eda14cbcSMatt Macy 			db->db_freed_in_flight = FALSE;
2644eda14cbcSMatt Macy 			DTRACE_SET_STATE(db,
2645eda14cbcSMatt Macy 			    "fill done handling freed in flight");
2646eda14cbcSMatt Macy 		} else {
2647eda14cbcSMatt Macy 			DTRACE_SET_STATE(db, "fill done");
2648eda14cbcSMatt Macy 		}
2649eda14cbcSMatt Macy 		cv_broadcast(&db->db_changed);
2650eda14cbcSMatt Macy 	}
2651eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
2652eda14cbcSMatt Macy }
2653eda14cbcSMatt Macy 
2654eda14cbcSMatt Macy void
2655eda14cbcSMatt Macy dmu_buf_write_embedded(dmu_buf_t *dbuf, void *data,
2656eda14cbcSMatt Macy     bp_embedded_type_t etype, enum zio_compress comp,
2657eda14cbcSMatt Macy     int uncompressed_size, int compressed_size, int byteorder,
2658eda14cbcSMatt Macy     dmu_tx_t *tx)
2659eda14cbcSMatt Macy {
2660eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)dbuf;
2661eda14cbcSMatt Macy 	struct dirty_leaf *dl;
2662eda14cbcSMatt Macy 	dmu_object_type_t type;
2663eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
2664eda14cbcSMatt Macy 
2665eda14cbcSMatt Macy 	if (etype == BP_EMBEDDED_TYPE_DATA) {
2666eda14cbcSMatt Macy 		ASSERT(spa_feature_is_active(dmu_objset_spa(db->db_objset),
2667eda14cbcSMatt Macy 		    SPA_FEATURE_EMBEDDED_DATA));
2668eda14cbcSMatt Macy 	}
2669eda14cbcSMatt Macy 
2670eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
2671eda14cbcSMatt Macy 	type = DB_DNODE(db)->dn_type;
2672eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
2673eda14cbcSMatt Macy 
2674eda14cbcSMatt Macy 	ASSERT0(db->db_level);
2675eda14cbcSMatt Macy 	ASSERT(db->db_blkid != DMU_BONUS_BLKID);
2676eda14cbcSMatt Macy 
2677eda14cbcSMatt Macy 	dmu_buf_will_not_fill(dbuf, tx);
2678eda14cbcSMatt Macy 
2679eda14cbcSMatt Macy 	dr = list_head(&db->db_dirty_records);
2680eda14cbcSMatt Macy 	ASSERT3U(dr->dr_txg, ==, tx->tx_txg);
2681eda14cbcSMatt Macy 	dl = &dr->dt.dl;
2682eda14cbcSMatt Macy 	encode_embedded_bp_compressed(&dl->dr_overridden_by,
2683eda14cbcSMatt Macy 	    data, comp, uncompressed_size, compressed_size);
2684eda14cbcSMatt Macy 	BPE_SET_ETYPE(&dl->dr_overridden_by, etype);
2685eda14cbcSMatt Macy 	BP_SET_TYPE(&dl->dr_overridden_by, type);
2686eda14cbcSMatt Macy 	BP_SET_LEVEL(&dl->dr_overridden_by, 0);
2687eda14cbcSMatt Macy 	BP_SET_BYTEORDER(&dl->dr_overridden_by, byteorder);
2688eda14cbcSMatt Macy 
2689eda14cbcSMatt Macy 	dl->dr_override_state = DR_OVERRIDDEN;
2690eda14cbcSMatt Macy 	dl->dr_overridden_by.blk_birth = dr->dr_txg;
2691eda14cbcSMatt Macy }
2692eda14cbcSMatt Macy 
2693eda14cbcSMatt Macy void
2694eda14cbcSMatt Macy dmu_buf_redact(dmu_buf_t *dbuf, dmu_tx_t *tx)
2695eda14cbcSMatt Macy {
2696eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)dbuf;
2697eda14cbcSMatt Macy 	dmu_object_type_t type;
2698eda14cbcSMatt Macy 	ASSERT(dsl_dataset_feature_is_active(db->db_objset->os_dsl_dataset,
2699eda14cbcSMatt Macy 	    SPA_FEATURE_REDACTED_DATASETS));
2700eda14cbcSMatt Macy 
2701eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
2702eda14cbcSMatt Macy 	type = DB_DNODE(db)->dn_type;
2703eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
2704eda14cbcSMatt Macy 
2705eda14cbcSMatt Macy 	ASSERT0(db->db_level);
2706eda14cbcSMatt Macy 	dmu_buf_will_not_fill(dbuf, tx);
2707eda14cbcSMatt Macy 
2708eda14cbcSMatt Macy 	blkptr_t bp = { { { {0} } } };
2709eda14cbcSMatt Macy 	BP_SET_TYPE(&bp, type);
2710eda14cbcSMatt Macy 	BP_SET_LEVEL(&bp, 0);
2711eda14cbcSMatt Macy 	BP_SET_BIRTH(&bp, tx->tx_txg, 0);
2712eda14cbcSMatt Macy 	BP_SET_REDACTED(&bp);
2713eda14cbcSMatt Macy 	BPE_SET_LSIZE(&bp, dbuf->db_size);
2714eda14cbcSMatt Macy 
2715eda14cbcSMatt Macy 	dbuf_override_impl(db, &bp, tx);
2716eda14cbcSMatt Macy }
2717eda14cbcSMatt Macy 
2718eda14cbcSMatt Macy /*
2719eda14cbcSMatt Macy  * Directly assign a provided arc buf to a given dbuf if it's not referenced
2720eda14cbcSMatt Macy  * by anybody except our caller. Otherwise copy arcbuf's contents to dbuf.
2721eda14cbcSMatt Macy  */
2722eda14cbcSMatt Macy void
2723eda14cbcSMatt Macy dbuf_assign_arcbuf(dmu_buf_impl_t *db, arc_buf_t *buf, dmu_tx_t *tx)
2724eda14cbcSMatt Macy {
2725eda14cbcSMatt Macy 	ASSERT(!zfs_refcount_is_zero(&db->db_holds));
2726eda14cbcSMatt Macy 	ASSERT(db->db_blkid != DMU_BONUS_BLKID);
2727eda14cbcSMatt Macy 	ASSERT(db->db_level == 0);
2728eda14cbcSMatt Macy 	ASSERT3U(dbuf_is_metadata(db), ==, arc_is_metadata(buf));
2729eda14cbcSMatt Macy 	ASSERT(buf != NULL);
2730eda14cbcSMatt Macy 	ASSERT3U(arc_buf_lsize(buf), ==, db->db.db_size);
2731eda14cbcSMatt Macy 	ASSERT(tx->tx_txg != 0);
2732eda14cbcSMatt Macy 
2733eda14cbcSMatt Macy 	arc_return_buf(buf, db);
2734eda14cbcSMatt Macy 	ASSERT(arc_released(buf));
2735eda14cbcSMatt Macy 
2736eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
2737eda14cbcSMatt Macy 
2738eda14cbcSMatt Macy 	while (db->db_state == DB_READ || db->db_state == DB_FILL)
2739eda14cbcSMatt Macy 		cv_wait(&db->db_changed, &db->db_mtx);
2740eda14cbcSMatt Macy 
2741eda14cbcSMatt Macy 	ASSERT(db->db_state == DB_CACHED || db->db_state == DB_UNCACHED);
2742eda14cbcSMatt Macy 
2743eda14cbcSMatt Macy 	if (db->db_state == DB_CACHED &&
2744eda14cbcSMatt Macy 	    zfs_refcount_count(&db->db_holds) - 1 > db->db_dirtycnt) {
2745eda14cbcSMatt Macy 		/*
2746eda14cbcSMatt Macy 		 * In practice, we will never have a case where we have an
2747eda14cbcSMatt Macy 		 * encrypted arc buffer while additional holds exist on the
2748eda14cbcSMatt Macy 		 * dbuf. We don't handle this here so we simply assert that
2749eda14cbcSMatt Macy 		 * fact instead.
2750eda14cbcSMatt Macy 		 */
2751eda14cbcSMatt Macy 		ASSERT(!arc_is_encrypted(buf));
2752eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
2753eda14cbcSMatt Macy 		(void) dbuf_dirty(db, tx);
2754eda14cbcSMatt Macy 		bcopy(buf->b_data, db->db.db_data, db->db.db_size);
2755eda14cbcSMatt Macy 		arc_buf_destroy(buf, db);
2756eda14cbcSMatt Macy 		return;
2757eda14cbcSMatt Macy 	}
2758eda14cbcSMatt Macy 
2759eda14cbcSMatt Macy 	if (db->db_state == DB_CACHED) {
2760eda14cbcSMatt Macy 		dbuf_dirty_record_t *dr = list_head(&db->db_dirty_records);
2761eda14cbcSMatt Macy 
2762eda14cbcSMatt Macy 		ASSERT(db->db_buf != NULL);
2763eda14cbcSMatt Macy 		if (dr != NULL && dr->dr_txg == tx->tx_txg) {
2764eda14cbcSMatt Macy 			ASSERT(dr->dt.dl.dr_data == db->db_buf);
2765eda14cbcSMatt Macy 
2766eda14cbcSMatt Macy 			if (!arc_released(db->db_buf)) {
2767eda14cbcSMatt Macy 				ASSERT(dr->dt.dl.dr_override_state ==
2768eda14cbcSMatt Macy 				    DR_OVERRIDDEN);
2769eda14cbcSMatt Macy 				arc_release(db->db_buf, db);
2770eda14cbcSMatt Macy 			}
2771eda14cbcSMatt Macy 			dr->dt.dl.dr_data = buf;
2772eda14cbcSMatt Macy 			arc_buf_destroy(db->db_buf, db);
2773eda14cbcSMatt Macy 		} else if (dr == NULL || dr->dt.dl.dr_data != db->db_buf) {
2774eda14cbcSMatt Macy 			arc_release(db->db_buf, db);
2775eda14cbcSMatt Macy 			arc_buf_destroy(db->db_buf, db);
2776eda14cbcSMatt Macy 		}
2777eda14cbcSMatt Macy 		db->db_buf = NULL;
2778eda14cbcSMatt Macy 	}
2779eda14cbcSMatt Macy 	ASSERT(db->db_buf == NULL);
2780eda14cbcSMatt Macy 	dbuf_set_data(db, buf);
2781eda14cbcSMatt Macy 	db->db_state = DB_FILL;
2782eda14cbcSMatt Macy 	DTRACE_SET_STATE(db, "filling assigned arcbuf");
2783eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
2784eda14cbcSMatt Macy 	(void) dbuf_dirty(db, tx);
2785eda14cbcSMatt Macy 	dmu_buf_fill_done(&db->db, tx);
2786eda14cbcSMatt Macy }
2787eda14cbcSMatt Macy 
2788eda14cbcSMatt Macy void
2789eda14cbcSMatt Macy dbuf_destroy(dmu_buf_impl_t *db)
2790eda14cbcSMatt Macy {
2791eda14cbcSMatt Macy 	dnode_t *dn;
2792eda14cbcSMatt Macy 	dmu_buf_impl_t *parent = db->db_parent;
2793eda14cbcSMatt Macy 	dmu_buf_impl_t *dndb;
2794eda14cbcSMatt Macy 
2795eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
2796eda14cbcSMatt Macy 	ASSERT(zfs_refcount_is_zero(&db->db_holds));
2797eda14cbcSMatt Macy 
2798eda14cbcSMatt Macy 	if (db->db_buf != NULL) {
2799eda14cbcSMatt Macy 		arc_buf_destroy(db->db_buf, db);
2800eda14cbcSMatt Macy 		db->db_buf = NULL;
2801eda14cbcSMatt Macy 	}
2802eda14cbcSMatt Macy 
2803eda14cbcSMatt Macy 	if (db->db_blkid == DMU_BONUS_BLKID) {
2804eda14cbcSMatt Macy 		int slots = DB_DNODE(db)->dn_num_slots;
2805eda14cbcSMatt Macy 		int bonuslen = DN_SLOTS_TO_BONUSLEN(slots);
2806eda14cbcSMatt Macy 		if (db->db.db_data != NULL) {
2807eda14cbcSMatt Macy 			kmem_free(db->db.db_data, bonuslen);
2808eda14cbcSMatt Macy 			arc_space_return(bonuslen, ARC_SPACE_BONUS);
2809eda14cbcSMatt Macy 			db->db_state = DB_UNCACHED;
2810eda14cbcSMatt Macy 			DTRACE_SET_STATE(db, "buffer cleared");
2811eda14cbcSMatt Macy 		}
2812eda14cbcSMatt Macy 	}
2813eda14cbcSMatt Macy 
2814eda14cbcSMatt Macy 	dbuf_clear_data(db);
2815eda14cbcSMatt Macy 
2816eda14cbcSMatt Macy 	if (multilist_link_active(&db->db_cache_link)) {
2817eda14cbcSMatt Macy 		ASSERT(db->db_caching_status == DB_DBUF_CACHE ||
2818eda14cbcSMatt Macy 		    db->db_caching_status == DB_DBUF_METADATA_CACHE);
2819eda14cbcSMatt Macy 
28203ff01b23SMartin Matuska 		multilist_remove(&dbuf_caches[db->db_caching_status].cache, db);
2821eda14cbcSMatt Macy 		(void) zfs_refcount_remove_many(
2822eda14cbcSMatt Macy 		    &dbuf_caches[db->db_caching_status].size,
2823eda14cbcSMatt Macy 		    db->db.db_size, db);
2824eda14cbcSMatt Macy 
2825eda14cbcSMatt Macy 		if (db->db_caching_status == DB_DBUF_METADATA_CACHE) {
2826eda14cbcSMatt Macy 			DBUF_STAT_BUMPDOWN(metadata_cache_count);
2827eda14cbcSMatt Macy 		} else {
2828eda14cbcSMatt Macy 			DBUF_STAT_BUMPDOWN(cache_levels[db->db_level]);
2829eda14cbcSMatt Macy 			DBUF_STAT_BUMPDOWN(cache_count);
2830eda14cbcSMatt Macy 			DBUF_STAT_DECR(cache_levels_bytes[db->db_level],
2831eda14cbcSMatt Macy 			    db->db.db_size);
2832eda14cbcSMatt Macy 		}
2833eda14cbcSMatt Macy 		db->db_caching_status = DB_NO_CACHE;
2834eda14cbcSMatt Macy 	}
2835eda14cbcSMatt Macy 
2836eda14cbcSMatt Macy 	ASSERT(db->db_state == DB_UNCACHED || db->db_state == DB_NOFILL);
2837eda14cbcSMatt Macy 	ASSERT(db->db_data_pending == NULL);
2838eda14cbcSMatt Macy 	ASSERT(list_is_empty(&db->db_dirty_records));
2839eda14cbcSMatt Macy 
2840eda14cbcSMatt Macy 	db->db_state = DB_EVICTING;
2841eda14cbcSMatt Macy 	DTRACE_SET_STATE(db, "buffer eviction started");
2842eda14cbcSMatt Macy 	db->db_blkptr = NULL;
2843eda14cbcSMatt Macy 
2844eda14cbcSMatt Macy 	/*
2845eda14cbcSMatt Macy 	 * Now that db_state is DB_EVICTING, nobody else can find this via
2846eda14cbcSMatt Macy 	 * the hash table.  We can now drop db_mtx, which allows us to
2847eda14cbcSMatt Macy 	 * acquire the dn_dbufs_mtx.
2848eda14cbcSMatt Macy 	 */
2849eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
2850eda14cbcSMatt Macy 
2851eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
2852eda14cbcSMatt Macy 	dn = DB_DNODE(db);
2853eda14cbcSMatt Macy 	dndb = dn->dn_dbuf;
2854eda14cbcSMatt Macy 	if (db->db_blkid != DMU_BONUS_BLKID) {
2855eda14cbcSMatt Macy 		boolean_t needlock = !MUTEX_HELD(&dn->dn_dbufs_mtx);
2856eda14cbcSMatt Macy 		if (needlock)
2857eda14cbcSMatt Macy 			mutex_enter_nested(&dn->dn_dbufs_mtx,
2858eda14cbcSMatt Macy 			    NESTED_SINGLE);
2859eda14cbcSMatt Macy 		avl_remove(&dn->dn_dbufs, db);
2860eda14cbcSMatt Macy 		membar_producer();
2861eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
2862eda14cbcSMatt Macy 		if (needlock)
2863eda14cbcSMatt Macy 			mutex_exit(&dn->dn_dbufs_mtx);
2864eda14cbcSMatt Macy 		/*
2865eda14cbcSMatt Macy 		 * Decrementing the dbuf count means that the hold corresponding
2866eda14cbcSMatt Macy 		 * to the removed dbuf is no longer discounted in dnode_move(),
2867eda14cbcSMatt Macy 		 * so the dnode cannot be moved until after we release the hold.
2868eda14cbcSMatt Macy 		 * The membar_producer() ensures visibility of the decremented
2869eda14cbcSMatt Macy 		 * value in dnode_move(), since DB_DNODE_EXIT doesn't actually
2870eda14cbcSMatt Macy 		 * release any lock.
2871eda14cbcSMatt Macy 		 */
2872eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
2873eda14cbcSMatt Macy 		dnode_rele_and_unlock(dn, db, B_TRUE);
2874eda14cbcSMatt Macy 		db->db_dnode_handle = NULL;
2875eda14cbcSMatt Macy 
2876eda14cbcSMatt Macy 		dbuf_hash_remove(db);
2877eda14cbcSMatt Macy 	} else {
2878eda14cbcSMatt Macy 		DB_DNODE_EXIT(db);
2879eda14cbcSMatt Macy 	}
2880eda14cbcSMatt Macy 
2881eda14cbcSMatt Macy 	ASSERT(zfs_refcount_is_zero(&db->db_holds));
2882eda14cbcSMatt Macy 
2883eda14cbcSMatt Macy 	db->db_parent = NULL;
2884eda14cbcSMatt Macy 
2885eda14cbcSMatt Macy 	ASSERT(db->db_buf == NULL);
2886eda14cbcSMatt Macy 	ASSERT(db->db.db_data == NULL);
2887eda14cbcSMatt Macy 	ASSERT(db->db_hash_next == NULL);
2888eda14cbcSMatt Macy 	ASSERT(db->db_blkptr == NULL);
2889eda14cbcSMatt Macy 	ASSERT(db->db_data_pending == NULL);
2890eda14cbcSMatt Macy 	ASSERT3U(db->db_caching_status, ==, DB_NO_CACHE);
2891eda14cbcSMatt Macy 	ASSERT(!multilist_link_active(&db->db_cache_link));
2892eda14cbcSMatt Macy 
2893eda14cbcSMatt Macy 	kmem_cache_free(dbuf_kmem_cache, db);
2894eda14cbcSMatt Macy 	arc_space_return(sizeof (dmu_buf_impl_t), ARC_SPACE_DBUF);
2895eda14cbcSMatt Macy 
2896eda14cbcSMatt Macy 	/*
2897eda14cbcSMatt Macy 	 * If this dbuf is referenced from an indirect dbuf,
2898eda14cbcSMatt Macy 	 * decrement the ref count on the indirect dbuf.
2899eda14cbcSMatt Macy 	 */
2900eda14cbcSMatt Macy 	if (parent && parent != dndb) {
2901eda14cbcSMatt Macy 		mutex_enter(&parent->db_mtx);
2902eda14cbcSMatt Macy 		dbuf_rele_and_unlock(parent, db, B_TRUE);
2903eda14cbcSMatt Macy 	}
2904eda14cbcSMatt Macy }
2905eda14cbcSMatt Macy 
2906eda14cbcSMatt Macy /*
2907eda14cbcSMatt Macy  * Note: While bpp will always be updated if the function returns success,
2908eda14cbcSMatt Macy  * parentp will not be updated if the dnode does not have dn_dbuf filled in;
2909eda14cbcSMatt Macy  * this happens when the dnode is the meta-dnode, or {user|group|project}used
2910eda14cbcSMatt Macy  * object.
2911eda14cbcSMatt Macy  */
2912eda14cbcSMatt Macy __attribute__((always_inline))
2913eda14cbcSMatt Macy static inline int
2914eda14cbcSMatt Macy dbuf_findbp(dnode_t *dn, int level, uint64_t blkid, int fail_sparse,
2915eda14cbcSMatt Macy     dmu_buf_impl_t **parentp, blkptr_t **bpp)
2916eda14cbcSMatt Macy {
2917eda14cbcSMatt Macy 	*parentp = NULL;
2918eda14cbcSMatt Macy 	*bpp = NULL;
2919eda14cbcSMatt Macy 
2920eda14cbcSMatt Macy 	ASSERT(blkid != DMU_BONUS_BLKID);
2921eda14cbcSMatt Macy 
2922eda14cbcSMatt Macy 	if (blkid == DMU_SPILL_BLKID) {
2923eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
2924eda14cbcSMatt Macy 		if (dn->dn_have_spill &&
2925eda14cbcSMatt Macy 		    (dn->dn_phys->dn_flags & DNODE_FLAG_SPILL_BLKPTR))
2926eda14cbcSMatt Macy 			*bpp = DN_SPILL_BLKPTR(dn->dn_phys);
2927eda14cbcSMatt Macy 		else
2928eda14cbcSMatt Macy 			*bpp = NULL;
2929eda14cbcSMatt Macy 		dbuf_add_ref(dn->dn_dbuf, NULL);
2930eda14cbcSMatt Macy 		*parentp = dn->dn_dbuf;
2931eda14cbcSMatt Macy 		mutex_exit(&dn->dn_mtx);
2932eda14cbcSMatt Macy 		return (0);
2933eda14cbcSMatt Macy 	}
2934eda14cbcSMatt Macy 
2935eda14cbcSMatt Macy 	int nlevels =
2936eda14cbcSMatt Macy 	    (dn->dn_phys->dn_nlevels == 0) ? 1 : dn->dn_phys->dn_nlevels;
2937eda14cbcSMatt Macy 	int epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
2938eda14cbcSMatt Macy 
2939eda14cbcSMatt Macy 	ASSERT3U(level * epbs, <, 64);
2940eda14cbcSMatt Macy 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
2941eda14cbcSMatt Macy 	/*
2942eda14cbcSMatt Macy 	 * This assertion shouldn't trip as long as the max indirect block size
2943eda14cbcSMatt Macy 	 * is less than 1M.  The reason for this is that up to that point,
2944eda14cbcSMatt Macy 	 * the number of levels required to address an entire object with blocks
2945eda14cbcSMatt Macy 	 * of size SPA_MINBLOCKSIZE satisfies nlevels * epbs + 1 <= 64.	 In
2946eda14cbcSMatt Macy 	 * other words, if N * epbs + 1 > 64, then if (N-1) * epbs + 1 > 55
2947eda14cbcSMatt Macy 	 * (i.e. we can address the entire object), objects will all use at most
2948eda14cbcSMatt Macy 	 * N-1 levels and the assertion won't overflow.	 However, once epbs is
2949eda14cbcSMatt Macy 	 * 13, 4 * 13 + 1 = 53, but 5 * 13 + 1 = 66.  Then, 4 levels will not be
2950eda14cbcSMatt Macy 	 * enough to address an entire object, so objects will have 5 levels,
2951eda14cbcSMatt Macy 	 * but then this assertion will overflow.
2952eda14cbcSMatt Macy 	 *
2953eda14cbcSMatt Macy 	 * All this is to say that if we ever increase DN_MAX_INDBLKSHIFT, we
2954eda14cbcSMatt Macy 	 * need to redo this logic to handle overflows.
2955eda14cbcSMatt Macy 	 */
2956eda14cbcSMatt Macy 	ASSERT(level >= nlevels ||
2957eda14cbcSMatt Macy 	    ((nlevels - level - 1) * epbs) +
2958eda14cbcSMatt Macy 	    highbit64(dn->dn_phys->dn_nblkptr) <= 64);
2959eda14cbcSMatt Macy 	if (level >= nlevels ||
2960eda14cbcSMatt Macy 	    blkid >= ((uint64_t)dn->dn_phys->dn_nblkptr <<
2961eda14cbcSMatt Macy 	    ((nlevels - level - 1) * epbs)) ||
2962eda14cbcSMatt Macy 	    (fail_sparse &&
2963eda14cbcSMatt Macy 	    blkid > (dn->dn_phys->dn_maxblkid >> (level * epbs)))) {
2964eda14cbcSMatt Macy 		/* the buffer has no parent yet */
2965eda14cbcSMatt Macy 		return (SET_ERROR(ENOENT));
2966eda14cbcSMatt Macy 	} else if (level < nlevels-1) {
2967eda14cbcSMatt Macy 		/* this block is referenced from an indirect block */
2968eda14cbcSMatt Macy 		int err;
2969eda14cbcSMatt Macy 
2970eda14cbcSMatt Macy 		err = dbuf_hold_impl(dn, level + 1,
2971eda14cbcSMatt Macy 		    blkid >> epbs, fail_sparse, FALSE, NULL, parentp);
2972eda14cbcSMatt Macy 
2973eda14cbcSMatt Macy 		if (err)
2974eda14cbcSMatt Macy 			return (err);
2975eda14cbcSMatt Macy 		err = dbuf_read(*parentp, NULL,
2976eda14cbcSMatt Macy 		    (DB_RF_HAVESTRUCT | DB_RF_NOPREFETCH | DB_RF_CANFAIL));
2977eda14cbcSMatt Macy 		if (err) {
2978eda14cbcSMatt Macy 			dbuf_rele(*parentp, NULL);
2979eda14cbcSMatt Macy 			*parentp = NULL;
2980eda14cbcSMatt Macy 			return (err);
2981eda14cbcSMatt Macy 		}
2982eda14cbcSMatt Macy 		rw_enter(&(*parentp)->db_rwlock, RW_READER);
2983eda14cbcSMatt Macy 		*bpp = ((blkptr_t *)(*parentp)->db.db_data) +
2984eda14cbcSMatt Macy 		    (blkid & ((1ULL << epbs) - 1));
2985eda14cbcSMatt Macy 		if (blkid > (dn->dn_phys->dn_maxblkid >> (level * epbs)))
2986eda14cbcSMatt Macy 			ASSERT(BP_IS_HOLE(*bpp));
2987eda14cbcSMatt Macy 		rw_exit(&(*parentp)->db_rwlock);
2988eda14cbcSMatt Macy 		return (0);
2989eda14cbcSMatt Macy 	} else {
2990eda14cbcSMatt Macy 		/* the block is referenced from the dnode */
2991eda14cbcSMatt Macy 		ASSERT3U(level, ==, nlevels-1);
2992eda14cbcSMatt Macy 		ASSERT(dn->dn_phys->dn_nblkptr == 0 ||
2993eda14cbcSMatt Macy 		    blkid < dn->dn_phys->dn_nblkptr);
2994eda14cbcSMatt Macy 		if (dn->dn_dbuf) {
2995eda14cbcSMatt Macy 			dbuf_add_ref(dn->dn_dbuf, NULL);
2996eda14cbcSMatt Macy 			*parentp = dn->dn_dbuf;
2997eda14cbcSMatt Macy 		}
2998eda14cbcSMatt Macy 		*bpp = &dn->dn_phys->dn_blkptr[blkid];
2999eda14cbcSMatt Macy 		return (0);
3000eda14cbcSMatt Macy 	}
3001eda14cbcSMatt Macy }
3002eda14cbcSMatt Macy 
3003eda14cbcSMatt Macy static dmu_buf_impl_t *
3004eda14cbcSMatt Macy dbuf_create(dnode_t *dn, uint8_t level, uint64_t blkid,
3005eda14cbcSMatt Macy     dmu_buf_impl_t *parent, blkptr_t *blkptr)
3006eda14cbcSMatt Macy {
3007eda14cbcSMatt Macy 	objset_t *os = dn->dn_objset;
3008eda14cbcSMatt Macy 	dmu_buf_impl_t *db, *odb;
3009eda14cbcSMatt Macy 
3010eda14cbcSMatt Macy 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
3011eda14cbcSMatt Macy 	ASSERT(dn->dn_type != DMU_OT_NONE);
3012eda14cbcSMatt Macy 
3013eda14cbcSMatt Macy 	db = kmem_cache_alloc(dbuf_kmem_cache, KM_SLEEP);
3014eda14cbcSMatt Macy 
3015eda14cbcSMatt Macy 	list_create(&db->db_dirty_records, sizeof (dbuf_dirty_record_t),
3016eda14cbcSMatt Macy 	    offsetof(dbuf_dirty_record_t, dr_dbuf_node));
3017eda14cbcSMatt Macy 
3018eda14cbcSMatt Macy 	db->db_objset = os;
3019eda14cbcSMatt Macy 	db->db.db_object = dn->dn_object;
3020eda14cbcSMatt Macy 	db->db_level = level;
3021eda14cbcSMatt Macy 	db->db_blkid = blkid;
3022eda14cbcSMatt Macy 	db->db_dirtycnt = 0;
3023eda14cbcSMatt Macy 	db->db_dnode_handle = dn->dn_handle;
3024eda14cbcSMatt Macy 	db->db_parent = parent;
3025eda14cbcSMatt Macy 	db->db_blkptr = blkptr;
3026eda14cbcSMatt Macy 
3027eda14cbcSMatt Macy 	db->db_user = NULL;
3028eda14cbcSMatt Macy 	db->db_user_immediate_evict = FALSE;
3029eda14cbcSMatt Macy 	db->db_freed_in_flight = FALSE;
3030eda14cbcSMatt Macy 	db->db_pending_evict = FALSE;
3031eda14cbcSMatt Macy 
3032eda14cbcSMatt Macy 	if (blkid == DMU_BONUS_BLKID) {
3033eda14cbcSMatt Macy 		ASSERT3P(parent, ==, dn->dn_dbuf);
3034eda14cbcSMatt Macy 		db->db.db_size = DN_SLOTS_TO_BONUSLEN(dn->dn_num_slots) -
3035eda14cbcSMatt Macy 		    (dn->dn_nblkptr-1) * sizeof (blkptr_t);
3036eda14cbcSMatt Macy 		ASSERT3U(db->db.db_size, >=, dn->dn_bonuslen);
3037eda14cbcSMatt Macy 		db->db.db_offset = DMU_BONUS_BLKID;
3038eda14cbcSMatt Macy 		db->db_state = DB_UNCACHED;
3039eda14cbcSMatt Macy 		DTRACE_SET_STATE(db, "bonus buffer created");
3040eda14cbcSMatt Macy 		db->db_caching_status = DB_NO_CACHE;
3041eda14cbcSMatt Macy 		/* the bonus dbuf is not placed in the hash table */
3042eda14cbcSMatt Macy 		arc_space_consume(sizeof (dmu_buf_impl_t), ARC_SPACE_DBUF);
3043eda14cbcSMatt Macy 		return (db);
3044eda14cbcSMatt Macy 	} else if (blkid == DMU_SPILL_BLKID) {
3045eda14cbcSMatt Macy 		db->db.db_size = (blkptr != NULL) ?
3046eda14cbcSMatt Macy 		    BP_GET_LSIZE(blkptr) : SPA_MINBLOCKSIZE;
3047eda14cbcSMatt Macy 		db->db.db_offset = 0;
3048eda14cbcSMatt Macy 	} else {
3049eda14cbcSMatt Macy 		int blocksize =
3050eda14cbcSMatt Macy 		    db->db_level ? 1 << dn->dn_indblkshift : dn->dn_datablksz;
3051eda14cbcSMatt Macy 		db->db.db_size = blocksize;
3052eda14cbcSMatt Macy 		db->db.db_offset = db->db_blkid * blocksize;
3053eda14cbcSMatt Macy 	}
3054eda14cbcSMatt Macy 
3055eda14cbcSMatt Macy 	/*
3056eda14cbcSMatt Macy 	 * Hold the dn_dbufs_mtx while we get the new dbuf
3057eda14cbcSMatt Macy 	 * in the hash table *and* added to the dbufs list.
3058eda14cbcSMatt Macy 	 * This prevents a possible deadlock with someone
3059eda14cbcSMatt Macy 	 * trying to look up this dbuf before it's added to the
3060eda14cbcSMatt Macy 	 * dn_dbufs list.
3061eda14cbcSMatt Macy 	 */
3062eda14cbcSMatt Macy 	mutex_enter(&dn->dn_dbufs_mtx);
3063eda14cbcSMatt Macy 	db->db_state = DB_EVICTING; /* not worth logging this state change */
3064eda14cbcSMatt Macy 	if ((odb = dbuf_hash_insert(db)) != NULL) {
3065eda14cbcSMatt Macy 		/* someone else inserted it first */
3066eda14cbcSMatt Macy 		kmem_cache_free(dbuf_kmem_cache, db);
3067eda14cbcSMatt Macy 		mutex_exit(&dn->dn_dbufs_mtx);
3068eda14cbcSMatt Macy 		DBUF_STAT_BUMP(hash_insert_race);
3069eda14cbcSMatt Macy 		return (odb);
3070eda14cbcSMatt Macy 	}
3071eda14cbcSMatt Macy 	avl_add(&dn->dn_dbufs, db);
3072eda14cbcSMatt Macy 
3073eda14cbcSMatt Macy 	db->db_state = DB_UNCACHED;
3074eda14cbcSMatt Macy 	DTRACE_SET_STATE(db, "regular buffer created");
3075eda14cbcSMatt Macy 	db->db_caching_status = DB_NO_CACHE;
3076eda14cbcSMatt Macy 	mutex_exit(&dn->dn_dbufs_mtx);
3077eda14cbcSMatt Macy 	arc_space_consume(sizeof (dmu_buf_impl_t), ARC_SPACE_DBUF);
3078eda14cbcSMatt Macy 
3079eda14cbcSMatt Macy 	if (parent && parent != dn->dn_dbuf)
3080eda14cbcSMatt Macy 		dbuf_add_ref(parent, db);
3081eda14cbcSMatt Macy 
3082eda14cbcSMatt Macy 	ASSERT(dn->dn_object == DMU_META_DNODE_OBJECT ||
3083eda14cbcSMatt Macy 	    zfs_refcount_count(&dn->dn_holds) > 0);
3084eda14cbcSMatt Macy 	(void) zfs_refcount_add(&dn->dn_holds, db);
3085eda14cbcSMatt Macy 
3086eda14cbcSMatt Macy 	dprintf_dbuf(db, "db=%p\n", db);
3087eda14cbcSMatt Macy 
3088eda14cbcSMatt Macy 	return (db);
3089eda14cbcSMatt Macy }
3090eda14cbcSMatt Macy 
3091eda14cbcSMatt Macy /*
3092eda14cbcSMatt Macy  * This function returns a block pointer and information about the object,
3093eda14cbcSMatt Macy  * given a dnode and a block.  This is a publicly accessible version of
3094eda14cbcSMatt Macy  * dbuf_findbp that only returns some information, rather than the
3095eda14cbcSMatt Macy  * dbuf.  Note that the dnode passed in must be held, and the dn_struct_rwlock
3096eda14cbcSMatt Macy  * should be locked as (at least) a reader.
3097eda14cbcSMatt Macy  */
3098eda14cbcSMatt Macy int
3099eda14cbcSMatt Macy dbuf_dnode_findbp(dnode_t *dn, uint64_t level, uint64_t blkid,
3100eda14cbcSMatt Macy     blkptr_t *bp, uint16_t *datablkszsec, uint8_t *indblkshift)
3101eda14cbcSMatt Macy {
3102eda14cbcSMatt Macy 	dmu_buf_impl_t *dbp = NULL;
3103eda14cbcSMatt Macy 	blkptr_t *bp2;
3104eda14cbcSMatt Macy 	int err = 0;
3105eda14cbcSMatt Macy 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
3106eda14cbcSMatt Macy 
3107eda14cbcSMatt Macy 	err = dbuf_findbp(dn, level, blkid, B_FALSE, &dbp, &bp2);
3108eda14cbcSMatt Macy 	if (err == 0) {
3109eda14cbcSMatt Macy 		*bp = *bp2;
3110eda14cbcSMatt Macy 		if (dbp != NULL)
3111eda14cbcSMatt Macy 			dbuf_rele(dbp, NULL);
3112eda14cbcSMatt Macy 		if (datablkszsec != NULL)
3113eda14cbcSMatt Macy 			*datablkszsec = dn->dn_phys->dn_datablkszsec;
3114eda14cbcSMatt Macy 		if (indblkshift != NULL)
3115eda14cbcSMatt Macy 			*indblkshift = dn->dn_phys->dn_indblkshift;
3116eda14cbcSMatt Macy 	}
3117eda14cbcSMatt Macy 
3118eda14cbcSMatt Macy 	return (err);
3119eda14cbcSMatt Macy }
3120eda14cbcSMatt Macy 
3121eda14cbcSMatt Macy typedef struct dbuf_prefetch_arg {
3122eda14cbcSMatt Macy 	spa_t *dpa_spa;	/* The spa to issue the prefetch in. */
3123eda14cbcSMatt Macy 	zbookmark_phys_t dpa_zb; /* The target block to prefetch. */
3124eda14cbcSMatt Macy 	int dpa_epbs; /* Entries (blkptr_t's) Per Block Shift. */
3125eda14cbcSMatt Macy 	int dpa_curlevel; /* The current level that we're reading */
3126eda14cbcSMatt Macy 	dnode_t *dpa_dnode; /* The dnode associated with the prefetch */
3127eda14cbcSMatt Macy 	zio_priority_t dpa_prio; /* The priority I/Os should be issued at. */
3128eda14cbcSMatt Macy 	zio_t *dpa_zio; /* The parent zio_t for all prefetches. */
3129eda14cbcSMatt Macy 	arc_flags_t dpa_aflags; /* Flags to pass to the final prefetch. */
31307877fdebSMatt Macy 	dbuf_prefetch_fn dpa_cb; /* prefetch completion callback */
31317877fdebSMatt Macy 	void *dpa_arg; /* prefetch completion arg */
3132eda14cbcSMatt Macy } dbuf_prefetch_arg_t;
3133eda14cbcSMatt Macy 
31347877fdebSMatt Macy static void
31357877fdebSMatt Macy dbuf_prefetch_fini(dbuf_prefetch_arg_t *dpa, boolean_t io_done)
31367877fdebSMatt Macy {
31377877fdebSMatt Macy 	if (dpa->dpa_cb != NULL)
31387877fdebSMatt Macy 		dpa->dpa_cb(dpa->dpa_arg, io_done);
31397877fdebSMatt Macy 	kmem_free(dpa, sizeof (*dpa));
31407877fdebSMatt Macy }
31417877fdebSMatt Macy 
31427877fdebSMatt Macy static void
31437877fdebSMatt Macy dbuf_issue_final_prefetch_done(zio_t *zio, const zbookmark_phys_t *zb,
31447877fdebSMatt Macy     const blkptr_t *iobp, arc_buf_t *abuf, void *private)
31457877fdebSMatt Macy {
31467877fdebSMatt Macy 	dbuf_prefetch_arg_t *dpa = private;
31477877fdebSMatt Macy 
31487877fdebSMatt Macy 	dbuf_prefetch_fini(dpa, B_TRUE);
31497877fdebSMatt Macy 	if (abuf != NULL)
31507877fdebSMatt Macy 		arc_buf_destroy(abuf, private);
31517877fdebSMatt Macy }
31527877fdebSMatt Macy 
3153eda14cbcSMatt Macy /*
3154eda14cbcSMatt Macy  * Actually issue the prefetch read for the block given.
3155eda14cbcSMatt Macy  */
3156eda14cbcSMatt Macy static void
3157eda14cbcSMatt Macy dbuf_issue_final_prefetch(dbuf_prefetch_arg_t *dpa, blkptr_t *bp)
3158eda14cbcSMatt Macy {
3159eda14cbcSMatt Macy 	ASSERT(!BP_IS_REDACTED(bp) ||
3160eda14cbcSMatt Macy 	    dsl_dataset_feature_is_active(
3161eda14cbcSMatt Macy 	    dpa->dpa_dnode->dn_objset->os_dsl_dataset,
3162eda14cbcSMatt Macy 	    SPA_FEATURE_REDACTED_DATASETS));
3163eda14cbcSMatt Macy 
3164eda14cbcSMatt Macy 	if (BP_IS_HOLE(bp) || BP_IS_EMBEDDED(bp) || BP_IS_REDACTED(bp))
31657877fdebSMatt Macy 		return (dbuf_prefetch_fini(dpa, B_FALSE));
3166eda14cbcSMatt Macy 
3167eda14cbcSMatt Macy 	int zio_flags = ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE;
3168eda14cbcSMatt Macy 	arc_flags_t aflags =
31697877fdebSMatt Macy 	    dpa->dpa_aflags | ARC_FLAG_NOWAIT | ARC_FLAG_PREFETCH |
31707877fdebSMatt Macy 	    ARC_FLAG_NO_BUF;
3171eda14cbcSMatt Macy 
3172eda14cbcSMatt Macy 	/* dnodes are always read as raw and then converted later */
3173eda14cbcSMatt Macy 	if (BP_GET_TYPE(bp) == DMU_OT_DNODE && BP_IS_PROTECTED(bp) &&
3174eda14cbcSMatt Macy 	    dpa->dpa_curlevel == 0)
3175eda14cbcSMatt Macy 		zio_flags |= ZIO_FLAG_RAW;
3176eda14cbcSMatt Macy 
3177eda14cbcSMatt Macy 	ASSERT3U(dpa->dpa_curlevel, ==, BP_GET_LEVEL(bp));
3178eda14cbcSMatt Macy 	ASSERT3U(dpa->dpa_curlevel, ==, dpa->dpa_zb.zb_level);
3179eda14cbcSMatt Macy 	ASSERT(dpa->dpa_zio != NULL);
31807877fdebSMatt Macy 	(void) arc_read(dpa->dpa_zio, dpa->dpa_spa, bp,
31817877fdebSMatt Macy 	    dbuf_issue_final_prefetch_done, dpa,
3182eda14cbcSMatt Macy 	    dpa->dpa_prio, zio_flags, &aflags, &dpa->dpa_zb);
3183eda14cbcSMatt Macy }
3184eda14cbcSMatt Macy 
3185eda14cbcSMatt Macy /*
3186eda14cbcSMatt Macy  * Called when an indirect block above our prefetch target is read in.  This
3187eda14cbcSMatt Macy  * will either read in the next indirect block down the tree or issue the actual
3188eda14cbcSMatt Macy  * prefetch if the next block down is our target.
3189eda14cbcSMatt Macy  */
3190eda14cbcSMatt Macy static void
3191eda14cbcSMatt Macy dbuf_prefetch_indirect_done(zio_t *zio, const zbookmark_phys_t *zb,
3192eda14cbcSMatt Macy     const blkptr_t *iobp, arc_buf_t *abuf, void *private)
3193eda14cbcSMatt Macy {
3194eda14cbcSMatt Macy 	dbuf_prefetch_arg_t *dpa = private;
3195eda14cbcSMatt Macy 
3196eda14cbcSMatt Macy 	ASSERT3S(dpa->dpa_zb.zb_level, <, dpa->dpa_curlevel);
3197eda14cbcSMatt Macy 	ASSERT3S(dpa->dpa_curlevel, >, 0);
3198eda14cbcSMatt Macy 
3199eda14cbcSMatt Macy 	if (abuf == NULL) {
3200eda14cbcSMatt Macy 		ASSERT(zio == NULL || zio->io_error != 0);
32017877fdebSMatt Macy 		return (dbuf_prefetch_fini(dpa, B_TRUE));
3202eda14cbcSMatt Macy 	}
3203eda14cbcSMatt Macy 	ASSERT(zio == NULL || zio->io_error == 0);
3204eda14cbcSMatt Macy 
3205eda14cbcSMatt Macy 	/*
3206eda14cbcSMatt Macy 	 * The dpa_dnode is only valid if we are called with a NULL
3207eda14cbcSMatt Macy 	 * zio. This indicates that the arc_read() returned without
3208eda14cbcSMatt Macy 	 * first calling zio_read() to issue a physical read. Once
3209eda14cbcSMatt Macy 	 * a physical read is made the dpa_dnode must be invalidated
3210eda14cbcSMatt Macy 	 * as the locks guarding it may have been dropped. If the
3211eda14cbcSMatt Macy 	 * dpa_dnode is still valid, then we want to add it to the dbuf
3212eda14cbcSMatt Macy 	 * cache. To do so, we must hold the dbuf associated with the block
3213eda14cbcSMatt Macy 	 * we just prefetched, read its contents so that we associate it
3214eda14cbcSMatt Macy 	 * with an arc_buf_t, and then release it.
3215eda14cbcSMatt Macy 	 */
3216eda14cbcSMatt Macy 	if (zio != NULL) {
3217eda14cbcSMatt Macy 		ASSERT3S(BP_GET_LEVEL(zio->io_bp), ==, dpa->dpa_curlevel);
3218eda14cbcSMatt Macy 		if (zio->io_flags & ZIO_FLAG_RAW_COMPRESS) {
3219eda14cbcSMatt Macy 			ASSERT3U(BP_GET_PSIZE(zio->io_bp), ==, zio->io_size);
3220eda14cbcSMatt Macy 		} else {
3221eda14cbcSMatt Macy 			ASSERT3U(BP_GET_LSIZE(zio->io_bp), ==, zio->io_size);
3222eda14cbcSMatt Macy 		}
3223eda14cbcSMatt Macy 		ASSERT3P(zio->io_spa, ==, dpa->dpa_spa);
3224eda14cbcSMatt Macy 
3225eda14cbcSMatt Macy 		dpa->dpa_dnode = NULL;
3226eda14cbcSMatt Macy 	} else if (dpa->dpa_dnode != NULL) {
3227eda14cbcSMatt Macy 		uint64_t curblkid = dpa->dpa_zb.zb_blkid >>
3228eda14cbcSMatt Macy 		    (dpa->dpa_epbs * (dpa->dpa_curlevel -
3229eda14cbcSMatt Macy 		    dpa->dpa_zb.zb_level));
3230eda14cbcSMatt Macy 		dmu_buf_impl_t *db = dbuf_hold_level(dpa->dpa_dnode,
3231eda14cbcSMatt Macy 		    dpa->dpa_curlevel, curblkid, FTAG);
3232eda14cbcSMatt Macy 		if (db == NULL) {
3233eda14cbcSMatt Macy 			arc_buf_destroy(abuf, private);
32347877fdebSMatt Macy 			return (dbuf_prefetch_fini(dpa, B_TRUE));
3235eda14cbcSMatt Macy 		}
3236eda14cbcSMatt Macy 		(void) dbuf_read(db, NULL,
3237eda14cbcSMatt Macy 		    DB_RF_MUST_SUCCEED | DB_RF_NOPREFETCH | DB_RF_HAVESTRUCT);
3238eda14cbcSMatt Macy 		dbuf_rele(db, FTAG);
3239eda14cbcSMatt Macy 	}
3240eda14cbcSMatt Macy 
3241eda14cbcSMatt Macy 	dpa->dpa_curlevel--;
3242eda14cbcSMatt Macy 	uint64_t nextblkid = dpa->dpa_zb.zb_blkid >>
3243eda14cbcSMatt Macy 	    (dpa->dpa_epbs * (dpa->dpa_curlevel - dpa->dpa_zb.zb_level));
3244eda14cbcSMatt Macy 	blkptr_t *bp = ((blkptr_t *)abuf->b_data) +
3245eda14cbcSMatt Macy 	    P2PHASE(nextblkid, 1ULL << dpa->dpa_epbs);
3246eda14cbcSMatt Macy 
3247eda14cbcSMatt Macy 	ASSERT(!BP_IS_REDACTED(bp) ||
3248eda14cbcSMatt Macy 	    dsl_dataset_feature_is_active(
3249eda14cbcSMatt Macy 	    dpa->dpa_dnode->dn_objset->os_dsl_dataset,
3250eda14cbcSMatt Macy 	    SPA_FEATURE_REDACTED_DATASETS));
3251eda14cbcSMatt Macy 	if (BP_IS_HOLE(bp) || BP_IS_REDACTED(bp)) {
32527877fdebSMatt Macy 		dbuf_prefetch_fini(dpa, B_TRUE);
3253eda14cbcSMatt Macy 	} else if (dpa->dpa_curlevel == dpa->dpa_zb.zb_level) {
3254eda14cbcSMatt Macy 		ASSERT3U(nextblkid, ==, dpa->dpa_zb.zb_blkid);
3255eda14cbcSMatt Macy 		dbuf_issue_final_prefetch(dpa, bp);
3256eda14cbcSMatt Macy 	} else {
3257eda14cbcSMatt Macy 		arc_flags_t iter_aflags = ARC_FLAG_NOWAIT;
3258eda14cbcSMatt Macy 		zbookmark_phys_t zb;
3259eda14cbcSMatt Macy 
3260eda14cbcSMatt Macy 		/* flag if L2ARC eligible, l2arc_noprefetch then decides */
3261eda14cbcSMatt Macy 		if (dpa->dpa_aflags & ARC_FLAG_L2CACHE)
3262eda14cbcSMatt Macy 			iter_aflags |= ARC_FLAG_L2CACHE;
3263eda14cbcSMatt Macy 
3264eda14cbcSMatt Macy 		ASSERT3U(dpa->dpa_curlevel, ==, BP_GET_LEVEL(bp));
3265eda14cbcSMatt Macy 
3266eda14cbcSMatt Macy 		SET_BOOKMARK(&zb, dpa->dpa_zb.zb_objset,
3267eda14cbcSMatt Macy 		    dpa->dpa_zb.zb_object, dpa->dpa_curlevel, nextblkid);
3268eda14cbcSMatt Macy 
3269eda14cbcSMatt Macy 		(void) arc_read(dpa->dpa_zio, dpa->dpa_spa,
3270eda14cbcSMatt Macy 		    bp, dbuf_prefetch_indirect_done, dpa, dpa->dpa_prio,
3271eda14cbcSMatt Macy 		    ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE,
3272eda14cbcSMatt Macy 		    &iter_aflags, &zb);
3273eda14cbcSMatt Macy 	}
3274eda14cbcSMatt Macy 
3275eda14cbcSMatt Macy 	arc_buf_destroy(abuf, private);
3276eda14cbcSMatt Macy }
3277eda14cbcSMatt Macy 
3278eda14cbcSMatt Macy /*
3279eda14cbcSMatt Macy  * Issue prefetch reads for the given block on the given level.  If the indirect
3280eda14cbcSMatt Macy  * blocks above that block are not in memory, we will read them in
3281eda14cbcSMatt Macy  * asynchronously.  As a result, this call never blocks waiting for a read to
3282eda14cbcSMatt Macy  * complete. Note that the prefetch might fail if the dataset is encrypted and
3283eda14cbcSMatt Macy  * the encryption key is unmapped before the IO completes.
3284eda14cbcSMatt Macy  */
32857877fdebSMatt Macy int
32867877fdebSMatt Macy dbuf_prefetch_impl(dnode_t *dn, int64_t level, uint64_t blkid,
32877877fdebSMatt Macy     zio_priority_t prio, arc_flags_t aflags, dbuf_prefetch_fn cb,
32887877fdebSMatt Macy     void *arg)
3289eda14cbcSMatt Macy {
3290eda14cbcSMatt Macy 	blkptr_t bp;
3291eda14cbcSMatt Macy 	int epbs, nlevels, curlevel;
3292eda14cbcSMatt Macy 	uint64_t curblkid;
3293eda14cbcSMatt Macy 
3294eda14cbcSMatt Macy 	ASSERT(blkid != DMU_BONUS_BLKID);
3295eda14cbcSMatt Macy 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
3296eda14cbcSMatt Macy 
3297eda14cbcSMatt Macy 	if (blkid > dn->dn_maxblkid)
32987877fdebSMatt Macy 		goto no_issue;
3299eda14cbcSMatt Macy 
3300eda14cbcSMatt Macy 	if (level == 0 && dnode_block_freed(dn, blkid))
33017877fdebSMatt Macy 		goto no_issue;
3302eda14cbcSMatt Macy 
3303eda14cbcSMatt Macy 	/*
3304eda14cbcSMatt Macy 	 * This dnode hasn't been written to disk yet, so there's nothing to
3305eda14cbcSMatt Macy 	 * prefetch.
3306eda14cbcSMatt Macy 	 */
3307eda14cbcSMatt Macy 	nlevels = dn->dn_phys->dn_nlevels;
3308eda14cbcSMatt Macy 	if (level >= nlevels || dn->dn_phys->dn_nblkptr == 0)
33097877fdebSMatt Macy 		goto no_issue;
3310eda14cbcSMatt Macy 
3311eda14cbcSMatt Macy 	epbs = dn->dn_phys->dn_indblkshift - SPA_BLKPTRSHIFT;
3312eda14cbcSMatt Macy 	if (dn->dn_phys->dn_maxblkid < blkid << (epbs * level))
33137877fdebSMatt Macy 		goto no_issue;
3314eda14cbcSMatt Macy 
3315eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dbuf_find(dn->dn_objset, dn->dn_object,
3316eda14cbcSMatt Macy 	    level, blkid);
3317eda14cbcSMatt Macy 	if (db != NULL) {
3318eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
3319eda14cbcSMatt Macy 		/*
3320eda14cbcSMatt Macy 		 * This dbuf already exists.  It is either CACHED, or
3321eda14cbcSMatt Macy 		 * (we assume) about to be read or filled.
3322eda14cbcSMatt Macy 		 */
33237877fdebSMatt Macy 		goto no_issue;
3324eda14cbcSMatt Macy 	}
3325eda14cbcSMatt Macy 
3326eda14cbcSMatt Macy 	/*
3327eda14cbcSMatt Macy 	 * Find the closest ancestor (indirect block) of the target block
3328eda14cbcSMatt Macy 	 * that is present in the cache.  In this indirect block, we will
3329eda14cbcSMatt Macy 	 * find the bp that is at curlevel, curblkid.
3330eda14cbcSMatt Macy 	 */
3331eda14cbcSMatt Macy 	curlevel = level;
3332eda14cbcSMatt Macy 	curblkid = blkid;
3333eda14cbcSMatt Macy 	while (curlevel < nlevels - 1) {
3334eda14cbcSMatt Macy 		int parent_level = curlevel + 1;
3335eda14cbcSMatt Macy 		uint64_t parent_blkid = curblkid >> epbs;
3336eda14cbcSMatt Macy 		dmu_buf_impl_t *db;
3337eda14cbcSMatt Macy 
3338eda14cbcSMatt Macy 		if (dbuf_hold_impl(dn, parent_level, parent_blkid,
3339eda14cbcSMatt Macy 		    FALSE, TRUE, FTAG, &db) == 0) {
3340eda14cbcSMatt Macy 			blkptr_t *bpp = db->db_buf->b_data;
3341eda14cbcSMatt Macy 			bp = bpp[P2PHASE(curblkid, 1 << epbs)];
3342eda14cbcSMatt Macy 			dbuf_rele(db, FTAG);
3343eda14cbcSMatt Macy 			break;
3344eda14cbcSMatt Macy 		}
3345eda14cbcSMatt Macy 
3346eda14cbcSMatt Macy 		curlevel = parent_level;
3347eda14cbcSMatt Macy 		curblkid = parent_blkid;
3348eda14cbcSMatt Macy 	}
3349eda14cbcSMatt Macy 
3350eda14cbcSMatt Macy 	if (curlevel == nlevels - 1) {
3351eda14cbcSMatt Macy 		/* No cached indirect blocks found. */
3352eda14cbcSMatt Macy 		ASSERT3U(curblkid, <, dn->dn_phys->dn_nblkptr);
3353eda14cbcSMatt Macy 		bp = dn->dn_phys->dn_blkptr[curblkid];
3354eda14cbcSMatt Macy 	}
3355eda14cbcSMatt Macy 	ASSERT(!BP_IS_REDACTED(&bp) ||
3356eda14cbcSMatt Macy 	    dsl_dataset_feature_is_active(dn->dn_objset->os_dsl_dataset,
3357eda14cbcSMatt Macy 	    SPA_FEATURE_REDACTED_DATASETS));
3358eda14cbcSMatt Macy 	if (BP_IS_HOLE(&bp) || BP_IS_REDACTED(&bp))
33597877fdebSMatt Macy 		goto no_issue;
3360eda14cbcSMatt Macy 
3361eda14cbcSMatt Macy 	ASSERT3U(curlevel, ==, BP_GET_LEVEL(&bp));
3362eda14cbcSMatt Macy 
3363eda14cbcSMatt Macy 	zio_t *pio = zio_root(dmu_objset_spa(dn->dn_objset), NULL, NULL,
3364eda14cbcSMatt Macy 	    ZIO_FLAG_CANFAIL);
3365eda14cbcSMatt Macy 
3366eda14cbcSMatt Macy 	dbuf_prefetch_arg_t *dpa = kmem_zalloc(sizeof (*dpa), KM_SLEEP);
3367eda14cbcSMatt Macy 	dsl_dataset_t *ds = dn->dn_objset->os_dsl_dataset;
3368eda14cbcSMatt Macy 	SET_BOOKMARK(&dpa->dpa_zb, ds != NULL ? ds->ds_object : DMU_META_OBJSET,
3369eda14cbcSMatt Macy 	    dn->dn_object, level, blkid);
3370eda14cbcSMatt Macy 	dpa->dpa_curlevel = curlevel;
3371eda14cbcSMatt Macy 	dpa->dpa_prio = prio;
3372eda14cbcSMatt Macy 	dpa->dpa_aflags = aflags;
3373eda14cbcSMatt Macy 	dpa->dpa_spa = dn->dn_objset->os_spa;
3374eda14cbcSMatt Macy 	dpa->dpa_dnode = dn;
3375eda14cbcSMatt Macy 	dpa->dpa_epbs = epbs;
3376eda14cbcSMatt Macy 	dpa->dpa_zio = pio;
33777877fdebSMatt Macy 	dpa->dpa_cb = cb;
33787877fdebSMatt Macy 	dpa->dpa_arg = arg;
3379eda14cbcSMatt Macy 
3380eda14cbcSMatt Macy 	/* flag if L2ARC eligible, l2arc_noprefetch then decides */
3381eda14cbcSMatt Macy 	if (DNODE_LEVEL_IS_L2CACHEABLE(dn, level))
3382eda14cbcSMatt Macy 		dpa->dpa_aflags |= ARC_FLAG_L2CACHE;
3383eda14cbcSMatt Macy 
3384eda14cbcSMatt Macy 	/*
3385eda14cbcSMatt Macy 	 * If we have the indirect just above us, no need to do the asynchronous
3386eda14cbcSMatt Macy 	 * prefetch chain; we'll just run the last step ourselves.  If we're at
3387eda14cbcSMatt Macy 	 * a higher level, though, we want to issue the prefetches for all the
3388eda14cbcSMatt Macy 	 * indirect blocks asynchronously, so we can go on with whatever we were
3389eda14cbcSMatt Macy 	 * doing.
3390eda14cbcSMatt Macy 	 */
3391eda14cbcSMatt Macy 	if (curlevel == level) {
3392eda14cbcSMatt Macy 		ASSERT3U(curblkid, ==, blkid);
3393eda14cbcSMatt Macy 		dbuf_issue_final_prefetch(dpa, &bp);
3394eda14cbcSMatt Macy 	} else {
3395eda14cbcSMatt Macy 		arc_flags_t iter_aflags = ARC_FLAG_NOWAIT;
3396eda14cbcSMatt Macy 		zbookmark_phys_t zb;
3397eda14cbcSMatt Macy 
3398eda14cbcSMatt Macy 		/* flag if L2ARC eligible, l2arc_noprefetch then decides */
3399eda14cbcSMatt Macy 		if (DNODE_LEVEL_IS_L2CACHEABLE(dn, level))
3400eda14cbcSMatt Macy 			iter_aflags |= ARC_FLAG_L2CACHE;
3401eda14cbcSMatt Macy 
3402eda14cbcSMatt Macy 		SET_BOOKMARK(&zb, ds != NULL ? ds->ds_object : DMU_META_OBJSET,
3403eda14cbcSMatt Macy 		    dn->dn_object, curlevel, curblkid);
3404eda14cbcSMatt Macy 		(void) arc_read(dpa->dpa_zio, dpa->dpa_spa,
3405eda14cbcSMatt Macy 		    &bp, dbuf_prefetch_indirect_done, dpa, prio,
3406eda14cbcSMatt Macy 		    ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE,
3407eda14cbcSMatt Macy 		    &iter_aflags, &zb);
3408eda14cbcSMatt Macy 	}
3409eda14cbcSMatt Macy 	/*
3410eda14cbcSMatt Macy 	 * We use pio here instead of dpa_zio since it's possible that
3411eda14cbcSMatt Macy 	 * dpa may have already been freed.
3412eda14cbcSMatt Macy 	 */
3413eda14cbcSMatt Macy 	zio_nowait(pio);
34147877fdebSMatt Macy 	return (1);
34157877fdebSMatt Macy no_issue:
34167877fdebSMatt Macy 	if (cb != NULL)
34177877fdebSMatt Macy 		cb(arg, B_FALSE);
34187877fdebSMatt Macy 	return (0);
34197877fdebSMatt Macy }
34207877fdebSMatt Macy 
34217877fdebSMatt Macy int
34227877fdebSMatt Macy dbuf_prefetch(dnode_t *dn, int64_t level, uint64_t blkid, zio_priority_t prio,
34237877fdebSMatt Macy     arc_flags_t aflags)
34247877fdebSMatt Macy {
34257877fdebSMatt Macy 
34267877fdebSMatt Macy 	return (dbuf_prefetch_impl(dn, level, blkid, prio, aflags, NULL, NULL));
3427eda14cbcSMatt Macy }
3428eda14cbcSMatt Macy 
3429eda14cbcSMatt Macy /*
3430eda14cbcSMatt Macy  * Helper function for dbuf_hold_impl() to copy a buffer. Handles
3431eda14cbcSMatt Macy  * the case of encrypted, compressed and uncompressed buffers by
3432eda14cbcSMatt Macy  * allocating the new buffer, respectively, with arc_alloc_raw_buf(),
3433eda14cbcSMatt Macy  * arc_alloc_compressed_buf() or arc_alloc_buf().*
3434eda14cbcSMatt Macy  *
3435eda14cbcSMatt Macy  * NOTE: Declared noinline to avoid stack bloat in dbuf_hold_impl().
3436eda14cbcSMatt Macy  */
3437eda14cbcSMatt Macy noinline static void
3438eda14cbcSMatt Macy dbuf_hold_copy(dnode_t *dn, dmu_buf_impl_t *db)
3439eda14cbcSMatt Macy {
3440eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr = db->db_data_pending;
3441eda14cbcSMatt Macy 	arc_buf_t *newdata, *data = dr->dt.dl.dr_data;
3442eda14cbcSMatt Macy 
3443eda14cbcSMatt Macy 	newdata = dbuf_alloc_arcbuf_from_arcbuf(db, data);
3444eda14cbcSMatt Macy 	dbuf_set_data(db, newdata);
3445eda14cbcSMatt Macy 	rw_enter(&db->db_rwlock, RW_WRITER);
3446eda14cbcSMatt Macy 	bcopy(data->b_data, db->db.db_data, arc_buf_size(data));
3447eda14cbcSMatt Macy 	rw_exit(&db->db_rwlock);
3448eda14cbcSMatt Macy }
3449eda14cbcSMatt Macy 
3450eda14cbcSMatt Macy /*
3451eda14cbcSMatt Macy  * Returns with db_holds incremented, and db_mtx not held.
3452eda14cbcSMatt Macy  * Note: dn_struct_rwlock must be held.
3453eda14cbcSMatt Macy  */
3454eda14cbcSMatt Macy int
3455eda14cbcSMatt Macy dbuf_hold_impl(dnode_t *dn, uint8_t level, uint64_t blkid,
3456eda14cbcSMatt Macy     boolean_t fail_sparse, boolean_t fail_uncached,
3457eda14cbcSMatt Macy     void *tag, dmu_buf_impl_t **dbp)
3458eda14cbcSMatt Macy {
3459eda14cbcSMatt Macy 	dmu_buf_impl_t *db, *parent = NULL;
3460eda14cbcSMatt Macy 
3461eda14cbcSMatt Macy 	/* If the pool has been created, verify the tx_sync_lock is not held */
3462eda14cbcSMatt Macy 	spa_t *spa = dn->dn_objset->os_spa;
3463eda14cbcSMatt Macy 	dsl_pool_t *dp = spa->spa_dsl_pool;
3464eda14cbcSMatt Macy 	if (dp != NULL) {
3465eda14cbcSMatt Macy 		ASSERT(!MUTEX_HELD(&dp->dp_tx.tx_sync_lock));
3466eda14cbcSMatt Macy 	}
3467eda14cbcSMatt Macy 
3468eda14cbcSMatt Macy 	ASSERT(blkid != DMU_BONUS_BLKID);
3469eda14cbcSMatt Macy 	ASSERT(RW_LOCK_HELD(&dn->dn_struct_rwlock));
3470eda14cbcSMatt Macy 	ASSERT3U(dn->dn_nlevels, >, level);
3471eda14cbcSMatt Macy 
3472eda14cbcSMatt Macy 	*dbp = NULL;
3473eda14cbcSMatt Macy 
3474eda14cbcSMatt Macy 	/* dbuf_find() returns with db_mtx held */
3475eda14cbcSMatt Macy 	db = dbuf_find(dn->dn_objset, dn->dn_object, level, blkid);
3476eda14cbcSMatt Macy 
3477eda14cbcSMatt Macy 	if (db == NULL) {
3478eda14cbcSMatt Macy 		blkptr_t *bp = NULL;
3479eda14cbcSMatt Macy 		int err;
3480eda14cbcSMatt Macy 
3481eda14cbcSMatt Macy 		if (fail_uncached)
3482eda14cbcSMatt Macy 			return (SET_ERROR(ENOENT));
3483eda14cbcSMatt Macy 
3484eda14cbcSMatt Macy 		ASSERT3P(parent, ==, NULL);
3485eda14cbcSMatt Macy 		err = dbuf_findbp(dn, level, blkid, fail_sparse, &parent, &bp);
3486eda14cbcSMatt Macy 		if (fail_sparse) {
3487eda14cbcSMatt Macy 			if (err == 0 && bp && BP_IS_HOLE(bp))
3488eda14cbcSMatt Macy 				err = SET_ERROR(ENOENT);
3489eda14cbcSMatt Macy 			if (err) {
3490eda14cbcSMatt Macy 				if (parent)
3491eda14cbcSMatt Macy 					dbuf_rele(parent, NULL);
3492eda14cbcSMatt Macy 				return (err);
3493eda14cbcSMatt Macy 			}
3494eda14cbcSMatt Macy 		}
3495eda14cbcSMatt Macy 		if (err && err != ENOENT)
3496eda14cbcSMatt Macy 			return (err);
3497eda14cbcSMatt Macy 		db = dbuf_create(dn, level, blkid, parent, bp);
3498eda14cbcSMatt Macy 	}
3499eda14cbcSMatt Macy 
3500eda14cbcSMatt Macy 	if (fail_uncached && db->db_state != DB_CACHED) {
3501eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
3502eda14cbcSMatt Macy 		return (SET_ERROR(ENOENT));
3503eda14cbcSMatt Macy 	}
3504eda14cbcSMatt Macy 
3505eda14cbcSMatt Macy 	if (db->db_buf != NULL) {
3506eda14cbcSMatt Macy 		arc_buf_access(db->db_buf);
3507eda14cbcSMatt Macy 		ASSERT3P(db->db.db_data, ==, db->db_buf->b_data);
3508eda14cbcSMatt Macy 	}
3509eda14cbcSMatt Macy 
3510eda14cbcSMatt Macy 	ASSERT(db->db_buf == NULL || arc_referenced(db->db_buf));
3511eda14cbcSMatt Macy 
3512eda14cbcSMatt Macy 	/*
3513eda14cbcSMatt Macy 	 * If this buffer is currently syncing out, and we are
3514eda14cbcSMatt Macy 	 * still referencing it from db_data, we need to make a copy
3515eda14cbcSMatt Macy 	 * of it in case we decide we want to dirty it again in this txg.
3516eda14cbcSMatt Macy 	 */
3517eda14cbcSMatt Macy 	if (db->db_level == 0 && db->db_blkid != DMU_BONUS_BLKID &&
3518eda14cbcSMatt Macy 	    dn->dn_object != DMU_META_DNODE_OBJECT &&
3519eda14cbcSMatt Macy 	    db->db_state == DB_CACHED && db->db_data_pending) {
3520eda14cbcSMatt Macy 		dbuf_dirty_record_t *dr = db->db_data_pending;
3521eda14cbcSMatt Macy 		if (dr->dt.dl.dr_data == db->db_buf)
3522eda14cbcSMatt Macy 			dbuf_hold_copy(dn, db);
3523eda14cbcSMatt Macy 	}
3524eda14cbcSMatt Macy 
3525eda14cbcSMatt Macy 	if (multilist_link_active(&db->db_cache_link)) {
3526eda14cbcSMatt Macy 		ASSERT(zfs_refcount_is_zero(&db->db_holds));
3527eda14cbcSMatt Macy 		ASSERT(db->db_caching_status == DB_DBUF_CACHE ||
3528eda14cbcSMatt Macy 		    db->db_caching_status == DB_DBUF_METADATA_CACHE);
3529eda14cbcSMatt Macy 
35303ff01b23SMartin Matuska 		multilist_remove(&dbuf_caches[db->db_caching_status].cache, db);
3531eda14cbcSMatt Macy 		(void) zfs_refcount_remove_many(
3532eda14cbcSMatt Macy 		    &dbuf_caches[db->db_caching_status].size,
3533eda14cbcSMatt Macy 		    db->db.db_size, db);
3534eda14cbcSMatt Macy 
3535eda14cbcSMatt Macy 		if (db->db_caching_status == DB_DBUF_METADATA_CACHE) {
3536eda14cbcSMatt Macy 			DBUF_STAT_BUMPDOWN(metadata_cache_count);
3537eda14cbcSMatt Macy 		} else {
3538eda14cbcSMatt Macy 			DBUF_STAT_BUMPDOWN(cache_levels[db->db_level]);
3539eda14cbcSMatt Macy 			DBUF_STAT_BUMPDOWN(cache_count);
3540eda14cbcSMatt Macy 			DBUF_STAT_DECR(cache_levels_bytes[db->db_level],
3541eda14cbcSMatt Macy 			    db->db.db_size);
3542eda14cbcSMatt Macy 		}
3543eda14cbcSMatt Macy 		db->db_caching_status = DB_NO_CACHE;
3544eda14cbcSMatt Macy 	}
3545eda14cbcSMatt Macy 	(void) zfs_refcount_add(&db->db_holds, tag);
3546eda14cbcSMatt Macy 	DBUF_VERIFY(db);
3547eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
3548eda14cbcSMatt Macy 
3549eda14cbcSMatt Macy 	/* NOTE: we can't rele the parent until after we drop the db_mtx */
3550eda14cbcSMatt Macy 	if (parent)
3551eda14cbcSMatt Macy 		dbuf_rele(parent, NULL);
3552eda14cbcSMatt Macy 
3553eda14cbcSMatt Macy 	ASSERT3P(DB_DNODE(db), ==, dn);
3554eda14cbcSMatt Macy 	ASSERT3U(db->db_blkid, ==, blkid);
3555eda14cbcSMatt Macy 	ASSERT3U(db->db_level, ==, level);
3556eda14cbcSMatt Macy 	*dbp = db;
3557eda14cbcSMatt Macy 
3558eda14cbcSMatt Macy 	return (0);
3559eda14cbcSMatt Macy }
3560eda14cbcSMatt Macy 
3561eda14cbcSMatt Macy dmu_buf_impl_t *
3562eda14cbcSMatt Macy dbuf_hold(dnode_t *dn, uint64_t blkid, void *tag)
3563eda14cbcSMatt Macy {
3564eda14cbcSMatt Macy 	return (dbuf_hold_level(dn, 0, blkid, tag));
3565eda14cbcSMatt Macy }
3566eda14cbcSMatt Macy 
3567eda14cbcSMatt Macy dmu_buf_impl_t *
3568eda14cbcSMatt Macy dbuf_hold_level(dnode_t *dn, int level, uint64_t blkid, void *tag)
3569eda14cbcSMatt Macy {
3570eda14cbcSMatt Macy 	dmu_buf_impl_t *db;
3571eda14cbcSMatt Macy 	int err = dbuf_hold_impl(dn, level, blkid, FALSE, FALSE, tag, &db);
3572eda14cbcSMatt Macy 	return (err ? NULL : db);
3573eda14cbcSMatt Macy }
3574eda14cbcSMatt Macy 
3575eda14cbcSMatt Macy void
3576eda14cbcSMatt Macy dbuf_create_bonus(dnode_t *dn)
3577eda14cbcSMatt Macy {
3578eda14cbcSMatt Macy 	ASSERT(RW_WRITE_HELD(&dn->dn_struct_rwlock));
3579eda14cbcSMatt Macy 
3580eda14cbcSMatt Macy 	ASSERT(dn->dn_bonus == NULL);
3581eda14cbcSMatt Macy 	dn->dn_bonus = dbuf_create(dn, 0, DMU_BONUS_BLKID, dn->dn_dbuf, NULL);
3582eda14cbcSMatt Macy }
3583eda14cbcSMatt Macy 
3584eda14cbcSMatt Macy int
3585eda14cbcSMatt Macy dbuf_spill_set_blksz(dmu_buf_t *db_fake, uint64_t blksz, dmu_tx_t *tx)
3586eda14cbcSMatt Macy {
3587eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
3588eda14cbcSMatt Macy 
3589eda14cbcSMatt Macy 	if (db->db_blkid != DMU_SPILL_BLKID)
3590eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
3591eda14cbcSMatt Macy 	if (blksz == 0)
3592eda14cbcSMatt Macy 		blksz = SPA_MINBLOCKSIZE;
3593eda14cbcSMatt Macy 	ASSERT3U(blksz, <=, spa_maxblocksize(dmu_objset_spa(db->db_objset)));
3594eda14cbcSMatt Macy 	blksz = P2ROUNDUP(blksz, SPA_MINBLOCKSIZE);
3595eda14cbcSMatt Macy 
3596eda14cbcSMatt Macy 	dbuf_new_size(db, blksz, tx);
3597eda14cbcSMatt Macy 
3598eda14cbcSMatt Macy 	return (0);
3599eda14cbcSMatt Macy }
3600eda14cbcSMatt Macy 
3601eda14cbcSMatt Macy void
3602eda14cbcSMatt Macy dbuf_rm_spill(dnode_t *dn, dmu_tx_t *tx)
3603eda14cbcSMatt Macy {
3604eda14cbcSMatt Macy 	dbuf_free_range(dn, DMU_SPILL_BLKID, DMU_SPILL_BLKID, tx);
3605eda14cbcSMatt Macy }
3606eda14cbcSMatt Macy 
3607eda14cbcSMatt Macy #pragma weak dmu_buf_add_ref = dbuf_add_ref
3608eda14cbcSMatt Macy void
3609eda14cbcSMatt Macy dbuf_add_ref(dmu_buf_impl_t *db, void *tag)
3610eda14cbcSMatt Macy {
3611eda14cbcSMatt Macy 	int64_t holds = zfs_refcount_add(&db->db_holds, tag);
3612eda14cbcSMatt Macy 	VERIFY3S(holds, >, 1);
3613eda14cbcSMatt Macy }
3614eda14cbcSMatt Macy 
3615eda14cbcSMatt Macy #pragma weak dmu_buf_try_add_ref = dbuf_try_add_ref
3616eda14cbcSMatt Macy boolean_t
3617eda14cbcSMatt Macy dbuf_try_add_ref(dmu_buf_t *db_fake, objset_t *os, uint64_t obj, uint64_t blkid,
3618eda14cbcSMatt Macy     void *tag)
3619eda14cbcSMatt Macy {
3620eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
3621eda14cbcSMatt Macy 	dmu_buf_impl_t *found_db;
3622eda14cbcSMatt Macy 	boolean_t result = B_FALSE;
3623eda14cbcSMatt Macy 
3624eda14cbcSMatt Macy 	if (blkid == DMU_BONUS_BLKID)
3625eda14cbcSMatt Macy 		found_db = dbuf_find_bonus(os, obj);
3626eda14cbcSMatt Macy 	else
3627eda14cbcSMatt Macy 		found_db = dbuf_find(os, obj, 0, blkid);
3628eda14cbcSMatt Macy 
3629eda14cbcSMatt Macy 	if (found_db != NULL) {
3630eda14cbcSMatt Macy 		if (db == found_db && dbuf_refcount(db) > db->db_dirtycnt) {
3631eda14cbcSMatt Macy 			(void) zfs_refcount_add(&db->db_holds, tag);
3632eda14cbcSMatt Macy 			result = B_TRUE;
3633eda14cbcSMatt Macy 		}
3634eda14cbcSMatt Macy 		mutex_exit(&found_db->db_mtx);
3635eda14cbcSMatt Macy 	}
3636eda14cbcSMatt Macy 	return (result);
3637eda14cbcSMatt Macy }
3638eda14cbcSMatt Macy 
3639eda14cbcSMatt Macy /*
3640eda14cbcSMatt Macy  * If you call dbuf_rele() you had better not be referencing the dnode handle
3641eda14cbcSMatt Macy  * unless you have some other direct or indirect hold on the dnode. (An indirect
3642eda14cbcSMatt Macy  * hold is a hold on one of the dnode's dbufs, including the bonus buffer.)
3643eda14cbcSMatt Macy  * Without that, the dbuf_rele() could lead to a dnode_rele() followed by the
3644eda14cbcSMatt Macy  * dnode's parent dbuf evicting its dnode handles.
3645eda14cbcSMatt Macy  */
3646eda14cbcSMatt Macy void
3647eda14cbcSMatt Macy dbuf_rele(dmu_buf_impl_t *db, void *tag)
3648eda14cbcSMatt Macy {
3649eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
3650eda14cbcSMatt Macy 	dbuf_rele_and_unlock(db, tag, B_FALSE);
3651eda14cbcSMatt Macy }
3652eda14cbcSMatt Macy 
3653eda14cbcSMatt Macy void
3654eda14cbcSMatt Macy dmu_buf_rele(dmu_buf_t *db, void *tag)
3655eda14cbcSMatt Macy {
3656eda14cbcSMatt Macy 	dbuf_rele((dmu_buf_impl_t *)db, tag);
3657eda14cbcSMatt Macy }
3658eda14cbcSMatt Macy 
3659eda14cbcSMatt Macy /*
3660eda14cbcSMatt Macy  * dbuf_rele() for an already-locked dbuf.  This is necessary to allow
3661eda14cbcSMatt Macy  * db_dirtycnt and db_holds to be updated atomically.  The 'evicting'
3662eda14cbcSMatt Macy  * argument should be set if we are already in the dbuf-evicting code
3663eda14cbcSMatt Macy  * path, in which case we don't want to recursively evict.  This allows us to
3664eda14cbcSMatt Macy  * avoid deeply nested stacks that would have a call flow similar to this:
3665eda14cbcSMatt Macy  *
3666eda14cbcSMatt Macy  * dbuf_rele()-->dbuf_rele_and_unlock()-->dbuf_evict_notify()
3667eda14cbcSMatt Macy  *	^						|
3668eda14cbcSMatt Macy  *	|						|
3669eda14cbcSMatt Macy  *	+-----dbuf_destroy()<--dbuf_evict_one()<--------+
3670eda14cbcSMatt Macy  *
3671eda14cbcSMatt Macy  */
3672eda14cbcSMatt Macy void
3673eda14cbcSMatt Macy dbuf_rele_and_unlock(dmu_buf_impl_t *db, void *tag, boolean_t evicting)
3674eda14cbcSMatt Macy {
3675eda14cbcSMatt Macy 	int64_t holds;
3676eda14cbcSMatt Macy 	uint64_t size;
3677eda14cbcSMatt Macy 
3678eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
3679eda14cbcSMatt Macy 	DBUF_VERIFY(db);
3680eda14cbcSMatt Macy 
3681eda14cbcSMatt Macy 	/*
3682eda14cbcSMatt Macy 	 * Remove the reference to the dbuf before removing its hold on the
3683eda14cbcSMatt Macy 	 * dnode so we can guarantee in dnode_move() that a referenced bonus
3684eda14cbcSMatt Macy 	 * buffer has a corresponding dnode hold.
3685eda14cbcSMatt Macy 	 */
3686eda14cbcSMatt Macy 	holds = zfs_refcount_remove(&db->db_holds, tag);
3687eda14cbcSMatt Macy 	ASSERT(holds >= 0);
3688eda14cbcSMatt Macy 
3689eda14cbcSMatt Macy 	/*
3690eda14cbcSMatt Macy 	 * We can't freeze indirects if there is a possibility that they
3691eda14cbcSMatt Macy 	 * may be modified in the current syncing context.
3692eda14cbcSMatt Macy 	 */
3693eda14cbcSMatt Macy 	if (db->db_buf != NULL &&
3694eda14cbcSMatt Macy 	    holds == (db->db_level == 0 ? db->db_dirtycnt : 0)) {
3695eda14cbcSMatt Macy 		arc_buf_freeze(db->db_buf);
3696eda14cbcSMatt Macy 	}
3697eda14cbcSMatt Macy 
3698eda14cbcSMatt Macy 	if (holds == db->db_dirtycnt &&
3699eda14cbcSMatt Macy 	    db->db_level == 0 && db->db_user_immediate_evict)
3700eda14cbcSMatt Macy 		dbuf_evict_user(db);
3701eda14cbcSMatt Macy 
3702eda14cbcSMatt Macy 	if (holds == 0) {
3703eda14cbcSMatt Macy 		if (db->db_blkid == DMU_BONUS_BLKID) {
3704eda14cbcSMatt Macy 			dnode_t *dn;
3705eda14cbcSMatt Macy 			boolean_t evict_dbuf = db->db_pending_evict;
3706eda14cbcSMatt Macy 
3707eda14cbcSMatt Macy 			/*
3708eda14cbcSMatt Macy 			 * If the dnode moves here, we cannot cross this
3709eda14cbcSMatt Macy 			 * barrier until the move completes.
3710eda14cbcSMatt Macy 			 */
3711eda14cbcSMatt Macy 			DB_DNODE_ENTER(db);
3712eda14cbcSMatt Macy 
3713eda14cbcSMatt Macy 			dn = DB_DNODE(db);
3714eda14cbcSMatt Macy 			atomic_dec_32(&dn->dn_dbufs_count);
3715eda14cbcSMatt Macy 
3716eda14cbcSMatt Macy 			/*
3717eda14cbcSMatt Macy 			 * Decrementing the dbuf count means that the bonus
3718eda14cbcSMatt Macy 			 * buffer's dnode hold is no longer discounted in
3719eda14cbcSMatt Macy 			 * dnode_move(). The dnode cannot move until after
3720eda14cbcSMatt Macy 			 * the dnode_rele() below.
3721eda14cbcSMatt Macy 			 */
3722eda14cbcSMatt Macy 			DB_DNODE_EXIT(db);
3723eda14cbcSMatt Macy 
3724eda14cbcSMatt Macy 			/*
3725eda14cbcSMatt Macy 			 * Do not reference db after its lock is dropped.
3726eda14cbcSMatt Macy 			 * Another thread may evict it.
3727eda14cbcSMatt Macy 			 */
3728eda14cbcSMatt Macy 			mutex_exit(&db->db_mtx);
3729eda14cbcSMatt Macy 
3730eda14cbcSMatt Macy 			if (evict_dbuf)
3731eda14cbcSMatt Macy 				dnode_evict_bonus(dn);
3732eda14cbcSMatt Macy 
3733eda14cbcSMatt Macy 			dnode_rele(dn, db);
3734eda14cbcSMatt Macy 		} else if (db->db_buf == NULL) {
3735eda14cbcSMatt Macy 			/*
3736eda14cbcSMatt Macy 			 * This is a special case: we never associated this
3737eda14cbcSMatt Macy 			 * dbuf with any data allocated from the ARC.
3738eda14cbcSMatt Macy 			 */
3739eda14cbcSMatt Macy 			ASSERT(db->db_state == DB_UNCACHED ||
3740eda14cbcSMatt Macy 			    db->db_state == DB_NOFILL);
3741eda14cbcSMatt Macy 			dbuf_destroy(db);
3742eda14cbcSMatt Macy 		} else if (arc_released(db->db_buf)) {
3743eda14cbcSMatt Macy 			/*
3744eda14cbcSMatt Macy 			 * This dbuf has anonymous data associated with it.
3745eda14cbcSMatt Macy 			 */
3746eda14cbcSMatt Macy 			dbuf_destroy(db);
3747eda14cbcSMatt Macy 		} else {
3748eda14cbcSMatt Macy 			boolean_t do_arc_evict = B_FALSE;
3749eda14cbcSMatt Macy 			blkptr_t bp;
3750eda14cbcSMatt Macy 			spa_t *spa = dmu_objset_spa(db->db_objset);
3751eda14cbcSMatt Macy 
3752eda14cbcSMatt Macy 			if (!DBUF_IS_CACHEABLE(db) &&
3753eda14cbcSMatt Macy 			    db->db_blkptr != NULL &&
3754eda14cbcSMatt Macy 			    !BP_IS_HOLE(db->db_blkptr) &&
3755eda14cbcSMatt Macy 			    !BP_IS_EMBEDDED(db->db_blkptr)) {
3756eda14cbcSMatt Macy 				do_arc_evict = B_TRUE;
3757eda14cbcSMatt Macy 				bp = *db->db_blkptr;
3758eda14cbcSMatt Macy 			}
3759eda14cbcSMatt Macy 
3760eda14cbcSMatt Macy 			if (!DBUF_IS_CACHEABLE(db) ||
3761eda14cbcSMatt Macy 			    db->db_pending_evict) {
3762eda14cbcSMatt Macy 				dbuf_destroy(db);
3763eda14cbcSMatt Macy 			} else if (!multilist_link_active(&db->db_cache_link)) {
3764eda14cbcSMatt Macy 				ASSERT3U(db->db_caching_status, ==,
3765eda14cbcSMatt Macy 				    DB_NO_CACHE);
3766eda14cbcSMatt Macy 
3767eda14cbcSMatt Macy 				dbuf_cached_state_t dcs =
3768eda14cbcSMatt Macy 				    dbuf_include_in_metadata_cache(db) ?
3769eda14cbcSMatt Macy 				    DB_DBUF_METADATA_CACHE : DB_DBUF_CACHE;
3770eda14cbcSMatt Macy 				db->db_caching_status = dcs;
3771eda14cbcSMatt Macy 
37723ff01b23SMartin Matuska 				multilist_insert(&dbuf_caches[dcs].cache, db);
3773*0d8fe237SMartin Matuska 				uint64_t db_size = db->db.db_size;
3774eda14cbcSMatt Macy 				size = zfs_refcount_add_many(
3775*0d8fe237SMartin Matuska 				    &dbuf_caches[dcs].size, db_size, db);
3776*0d8fe237SMartin Matuska 				uint8_t db_level = db->db_level;
3777*0d8fe237SMartin Matuska 				mutex_exit(&db->db_mtx);
3778eda14cbcSMatt Macy 
3779eda14cbcSMatt Macy 				if (dcs == DB_DBUF_METADATA_CACHE) {
3780eda14cbcSMatt Macy 					DBUF_STAT_BUMP(metadata_cache_count);
3781eda14cbcSMatt Macy 					DBUF_STAT_MAX(
3782eda14cbcSMatt Macy 					    metadata_cache_size_bytes_max,
3783eda14cbcSMatt Macy 					    size);
3784eda14cbcSMatt Macy 				} else {
3785eda14cbcSMatt Macy 					DBUF_STAT_BUMP(cache_count);
3786eda14cbcSMatt Macy 					DBUF_STAT_MAX(cache_size_bytes_max,
3787eda14cbcSMatt Macy 					    size);
3788*0d8fe237SMartin Matuska 					DBUF_STAT_BUMP(cache_levels[db_level]);
3789*0d8fe237SMartin Matuska 					DBUF_STAT_INCR(
3790*0d8fe237SMartin Matuska 					    cache_levels_bytes[db_level],
3791*0d8fe237SMartin Matuska 					    db_size);
3792eda14cbcSMatt Macy 				}
3793eda14cbcSMatt Macy 
3794eda14cbcSMatt Macy 				if (dcs == DB_DBUF_CACHE && !evicting)
3795eda14cbcSMatt Macy 					dbuf_evict_notify(size);
3796eda14cbcSMatt Macy 			}
3797eda14cbcSMatt Macy 
3798eda14cbcSMatt Macy 			if (do_arc_evict)
3799eda14cbcSMatt Macy 				arc_freed(spa, &bp);
3800eda14cbcSMatt Macy 		}
3801eda14cbcSMatt Macy 	} else {
3802eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
3803eda14cbcSMatt Macy 	}
3804eda14cbcSMatt Macy 
3805eda14cbcSMatt Macy }
3806eda14cbcSMatt Macy 
3807eda14cbcSMatt Macy #pragma weak dmu_buf_refcount = dbuf_refcount
3808eda14cbcSMatt Macy uint64_t
3809eda14cbcSMatt Macy dbuf_refcount(dmu_buf_impl_t *db)
3810eda14cbcSMatt Macy {
3811eda14cbcSMatt Macy 	return (zfs_refcount_count(&db->db_holds));
3812eda14cbcSMatt Macy }
3813eda14cbcSMatt Macy 
3814eda14cbcSMatt Macy uint64_t
3815eda14cbcSMatt Macy dmu_buf_user_refcount(dmu_buf_t *db_fake)
3816eda14cbcSMatt Macy {
3817eda14cbcSMatt Macy 	uint64_t holds;
3818eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
3819eda14cbcSMatt Macy 
3820eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
3821eda14cbcSMatt Macy 	ASSERT3U(zfs_refcount_count(&db->db_holds), >=, db->db_dirtycnt);
3822eda14cbcSMatt Macy 	holds = zfs_refcount_count(&db->db_holds) - db->db_dirtycnt;
3823eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
3824eda14cbcSMatt Macy 
3825eda14cbcSMatt Macy 	return (holds);
3826eda14cbcSMatt Macy }
3827eda14cbcSMatt Macy 
3828eda14cbcSMatt Macy void *
3829eda14cbcSMatt Macy dmu_buf_replace_user(dmu_buf_t *db_fake, dmu_buf_user_t *old_user,
3830eda14cbcSMatt Macy     dmu_buf_user_t *new_user)
3831eda14cbcSMatt Macy {
3832eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
3833eda14cbcSMatt Macy 
3834eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
3835eda14cbcSMatt Macy 	dbuf_verify_user(db, DBVU_NOT_EVICTING);
3836eda14cbcSMatt Macy 	if (db->db_user == old_user)
3837eda14cbcSMatt Macy 		db->db_user = new_user;
3838eda14cbcSMatt Macy 	else
3839eda14cbcSMatt Macy 		old_user = db->db_user;
3840eda14cbcSMatt Macy 	dbuf_verify_user(db, DBVU_NOT_EVICTING);
3841eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
3842eda14cbcSMatt Macy 
3843eda14cbcSMatt Macy 	return (old_user);
3844eda14cbcSMatt Macy }
3845eda14cbcSMatt Macy 
3846eda14cbcSMatt Macy void *
3847eda14cbcSMatt Macy dmu_buf_set_user(dmu_buf_t *db_fake, dmu_buf_user_t *user)
3848eda14cbcSMatt Macy {
3849eda14cbcSMatt Macy 	return (dmu_buf_replace_user(db_fake, NULL, user));
3850eda14cbcSMatt Macy }
3851eda14cbcSMatt Macy 
3852eda14cbcSMatt Macy void *
3853eda14cbcSMatt Macy dmu_buf_set_user_ie(dmu_buf_t *db_fake, dmu_buf_user_t *user)
3854eda14cbcSMatt Macy {
3855eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
3856eda14cbcSMatt Macy 
3857eda14cbcSMatt Macy 	db->db_user_immediate_evict = TRUE;
3858eda14cbcSMatt Macy 	return (dmu_buf_set_user(db_fake, user));
3859eda14cbcSMatt Macy }
3860eda14cbcSMatt Macy 
3861eda14cbcSMatt Macy void *
3862eda14cbcSMatt Macy dmu_buf_remove_user(dmu_buf_t *db_fake, dmu_buf_user_t *user)
3863eda14cbcSMatt Macy {
3864eda14cbcSMatt Macy 	return (dmu_buf_replace_user(db_fake, user, NULL));
3865eda14cbcSMatt Macy }
3866eda14cbcSMatt Macy 
3867eda14cbcSMatt Macy void *
3868eda14cbcSMatt Macy dmu_buf_get_user(dmu_buf_t *db_fake)
3869eda14cbcSMatt Macy {
3870eda14cbcSMatt Macy 	dmu_buf_impl_t *db = (dmu_buf_impl_t *)db_fake;
3871eda14cbcSMatt Macy 
3872eda14cbcSMatt Macy 	dbuf_verify_user(db, DBVU_NOT_EVICTING);
3873eda14cbcSMatt Macy 	return (db->db_user);
3874eda14cbcSMatt Macy }
3875eda14cbcSMatt Macy 
3876eda14cbcSMatt Macy void
3877eda14cbcSMatt Macy dmu_buf_user_evict_wait()
3878eda14cbcSMatt Macy {
3879eda14cbcSMatt Macy 	taskq_wait(dbu_evict_taskq);
3880eda14cbcSMatt Macy }
3881eda14cbcSMatt Macy 
3882eda14cbcSMatt Macy blkptr_t *
3883eda14cbcSMatt Macy dmu_buf_get_blkptr(dmu_buf_t *db)
3884eda14cbcSMatt Macy {
3885eda14cbcSMatt Macy 	dmu_buf_impl_t *dbi = (dmu_buf_impl_t *)db;
3886eda14cbcSMatt Macy 	return (dbi->db_blkptr);
3887eda14cbcSMatt Macy }
3888eda14cbcSMatt Macy 
3889eda14cbcSMatt Macy objset_t *
3890eda14cbcSMatt Macy dmu_buf_get_objset(dmu_buf_t *db)
3891eda14cbcSMatt Macy {
3892eda14cbcSMatt Macy 	dmu_buf_impl_t *dbi = (dmu_buf_impl_t *)db;
3893eda14cbcSMatt Macy 	return (dbi->db_objset);
3894eda14cbcSMatt Macy }
3895eda14cbcSMatt Macy 
3896eda14cbcSMatt Macy dnode_t *
3897eda14cbcSMatt Macy dmu_buf_dnode_enter(dmu_buf_t *db)
3898eda14cbcSMatt Macy {
3899eda14cbcSMatt Macy 	dmu_buf_impl_t *dbi = (dmu_buf_impl_t *)db;
3900eda14cbcSMatt Macy 	DB_DNODE_ENTER(dbi);
3901eda14cbcSMatt Macy 	return (DB_DNODE(dbi));
3902eda14cbcSMatt Macy }
3903eda14cbcSMatt Macy 
3904eda14cbcSMatt Macy void
3905eda14cbcSMatt Macy dmu_buf_dnode_exit(dmu_buf_t *db)
3906eda14cbcSMatt Macy {
3907eda14cbcSMatt Macy 	dmu_buf_impl_t *dbi = (dmu_buf_impl_t *)db;
3908eda14cbcSMatt Macy 	DB_DNODE_EXIT(dbi);
3909eda14cbcSMatt Macy }
3910eda14cbcSMatt Macy 
3911eda14cbcSMatt Macy static void
3912eda14cbcSMatt Macy dbuf_check_blkptr(dnode_t *dn, dmu_buf_impl_t *db)
3913eda14cbcSMatt Macy {
3914eda14cbcSMatt Macy 	/* ASSERT(dmu_tx_is_syncing(tx) */
3915eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
3916eda14cbcSMatt Macy 
3917eda14cbcSMatt Macy 	if (db->db_blkptr != NULL)
3918eda14cbcSMatt Macy 		return;
3919eda14cbcSMatt Macy 
3920eda14cbcSMatt Macy 	if (db->db_blkid == DMU_SPILL_BLKID) {
3921eda14cbcSMatt Macy 		db->db_blkptr = DN_SPILL_BLKPTR(dn->dn_phys);
3922eda14cbcSMatt Macy 		BP_ZERO(db->db_blkptr);
3923eda14cbcSMatt Macy 		return;
3924eda14cbcSMatt Macy 	}
3925eda14cbcSMatt Macy 	if (db->db_level == dn->dn_phys->dn_nlevels-1) {
3926eda14cbcSMatt Macy 		/*
3927eda14cbcSMatt Macy 		 * This buffer was allocated at a time when there was
3928eda14cbcSMatt Macy 		 * no available blkptrs from the dnode, or it was
3929eda14cbcSMatt Macy 		 * inappropriate to hook it in (i.e., nlevels mismatch).
3930eda14cbcSMatt Macy 		 */
3931eda14cbcSMatt Macy 		ASSERT(db->db_blkid < dn->dn_phys->dn_nblkptr);
3932eda14cbcSMatt Macy 		ASSERT(db->db_parent == NULL);
3933eda14cbcSMatt Macy 		db->db_parent = dn->dn_dbuf;
3934eda14cbcSMatt Macy 		db->db_blkptr = &dn->dn_phys->dn_blkptr[db->db_blkid];
3935eda14cbcSMatt Macy 		DBUF_VERIFY(db);
3936eda14cbcSMatt Macy 	} else {
3937eda14cbcSMatt Macy 		dmu_buf_impl_t *parent = db->db_parent;
3938eda14cbcSMatt Macy 		int epbs = dn->dn_phys->dn_indblkshift - SPA_BLKPTRSHIFT;
3939eda14cbcSMatt Macy 
3940eda14cbcSMatt Macy 		ASSERT(dn->dn_phys->dn_nlevels > 1);
3941eda14cbcSMatt Macy 		if (parent == NULL) {
3942eda14cbcSMatt Macy 			mutex_exit(&db->db_mtx);
3943eda14cbcSMatt Macy 			rw_enter(&dn->dn_struct_rwlock, RW_READER);
3944eda14cbcSMatt Macy 			parent = dbuf_hold_level(dn, db->db_level + 1,
3945eda14cbcSMatt Macy 			    db->db_blkid >> epbs, db);
3946eda14cbcSMatt Macy 			rw_exit(&dn->dn_struct_rwlock);
3947eda14cbcSMatt Macy 			mutex_enter(&db->db_mtx);
3948eda14cbcSMatt Macy 			db->db_parent = parent;
3949eda14cbcSMatt Macy 		}
3950eda14cbcSMatt Macy 		db->db_blkptr = (blkptr_t *)parent->db.db_data +
3951eda14cbcSMatt Macy 		    (db->db_blkid & ((1ULL << epbs) - 1));
3952eda14cbcSMatt Macy 		DBUF_VERIFY(db);
3953eda14cbcSMatt Macy 	}
3954eda14cbcSMatt Macy }
3955eda14cbcSMatt Macy 
3956eda14cbcSMatt Macy static void
3957eda14cbcSMatt Macy dbuf_sync_bonus(dbuf_dirty_record_t *dr, dmu_tx_t *tx)
3958eda14cbcSMatt Macy {
3959eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
3960eda14cbcSMatt Macy 	void *data = dr->dt.dl.dr_data;
3961eda14cbcSMatt Macy 
3962eda14cbcSMatt Macy 	ASSERT0(db->db_level);
3963eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
3964eda14cbcSMatt Macy 	ASSERT(db->db_blkid == DMU_BONUS_BLKID);
3965eda14cbcSMatt Macy 	ASSERT(data != NULL);
3966eda14cbcSMatt Macy 
39677877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
3968eda14cbcSMatt Macy 	ASSERT3U(DN_MAX_BONUS_LEN(dn->dn_phys), <=,
3969eda14cbcSMatt Macy 	    DN_SLOTS_TO_BONUSLEN(dn->dn_phys->dn_extra_slots + 1));
3970eda14cbcSMatt Macy 	bcopy(data, DN_BONUS(dn->dn_phys), DN_MAX_BONUS_LEN(dn->dn_phys));
3971eda14cbcSMatt Macy 
3972eda14cbcSMatt Macy 	dbuf_sync_leaf_verify_bonus_dnode(dr);
3973eda14cbcSMatt Macy 
3974eda14cbcSMatt Macy 	dbuf_undirty_bonus(dr);
3975eda14cbcSMatt Macy 	dbuf_rele_and_unlock(db, (void *)(uintptr_t)tx->tx_txg, B_FALSE);
3976eda14cbcSMatt Macy }
3977eda14cbcSMatt Macy 
3978eda14cbcSMatt Macy /*
3979eda14cbcSMatt Macy  * When syncing out a blocks of dnodes, adjust the block to deal with
3980eda14cbcSMatt Macy  * encryption.  Normally, we make sure the block is decrypted before writing
3981eda14cbcSMatt Macy  * it.  If we have crypt params, then we are writing a raw (encrypted) block,
3982eda14cbcSMatt Macy  * from a raw receive.  In this case, set the ARC buf's crypt params so
3983eda14cbcSMatt Macy  * that the BP will be filled with the correct byteorder, salt, iv, and mac.
3984eda14cbcSMatt Macy  */
3985eda14cbcSMatt Macy static void
3986eda14cbcSMatt Macy dbuf_prepare_encrypted_dnode_leaf(dbuf_dirty_record_t *dr)
3987eda14cbcSMatt Macy {
3988eda14cbcSMatt Macy 	int err;
3989eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
3990eda14cbcSMatt Macy 
3991eda14cbcSMatt Macy 	ASSERT(MUTEX_HELD(&db->db_mtx));
3992eda14cbcSMatt Macy 	ASSERT3U(db->db.db_object, ==, DMU_META_DNODE_OBJECT);
3993eda14cbcSMatt Macy 	ASSERT3U(db->db_level, ==, 0);
3994eda14cbcSMatt Macy 
3995eda14cbcSMatt Macy 	if (!db->db_objset->os_raw_receive && arc_is_encrypted(db->db_buf)) {
3996eda14cbcSMatt Macy 		zbookmark_phys_t zb;
3997eda14cbcSMatt Macy 
3998eda14cbcSMatt Macy 		/*
3999eda14cbcSMatt Macy 		 * Unfortunately, there is currently no mechanism for
4000eda14cbcSMatt Macy 		 * syncing context to handle decryption errors. An error
4001eda14cbcSMatt Macy 		 * here is only possible if an attacker maliciously
4002eda14cbcSMatt Macy 		 * changed a dnode block and updated the associated
4003eda14cbcSMatt Macy 		 * checksums going up the block tree.
4004eda14cbcSMatt Macy 		 */
4005eda14cbcSMatt Macy 		SET_BOOKMARK(&zb, dmu_objset_id(db->db_objset),
4006eda14cbcSMatt Macy 		    db->db.db_object, db->db_level, db->db_blkid);
4007eda14cbcSMatt Macy 		err = arc_untransform(db->db_buf, db->db_objset->os_spa,
4008eda14cbcSMatt Macy 		    &zb, B_TRUE);
4009eda14cbcSMatt Macy 		if (err)
4010eda14cbcSMatt Macy 			panic("Invalid dnode block MAC");
4011eda14cbcSMatt Macy 	} else if (dr->dt.dl.dr_has_raw_params) {
4012eda14cbcSMatt Macy 		(void) arc_release(dr->dt.dl.dr_data, db);
4013eda14cbcSMatt Macy 		arc_convert_to_raw(dr->dt.dl.dr_data,
4014eda14cbcSMatt Macy 		    dmu_objset_id(db->db_objset),
4015eda14cbcSMatt Macy 		    dr->dt.dl.dr_byteorder, DMU_OT_DNODE,
4016eda14cbcSMatt Macy 		    dr->dt.dl.dr_salt, dr->dt.dl.dr_iv, dr->dt.dl.dr_mac);
4017eda14cbcSMatt Macy 	}
4018eda14cbcSMatt Macy }
4019eda14cbcSMatt Macy 
4020eda14cbcSMatt Macy /*
4021eda14cbcSMatt Macy  * dbuf_sync_indirect() is called recursively from dbuf_sync_list() so it
4022eda14cbcSMatt Macy  * is critical the we not allow the compiler to inline this function in to
4023eda14cbcSMatt Macy  * dbuf_sync_list() thereby drastically bloating the stack usage.
4024eda14cbcSMatt Macy  */
4025eda14cbcSMatt Macy noinline static void
4026eda14cbcSMatt Macy dbuf_sync_indirect(dbuf_dirty_record_t *dr, dmu_tx_t *tx)
4027eda14cbcSMatt Macy {
4028eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
40297877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
4030eda14cbcSMatt Macy 
4031eda14cbcSMatt Macy 	ASSERT(dmu_tx_is_syncing(tx));
4032eda14cbcSMatt Macy 
4033eda14cbcSMatt Macy 	dprintf_dbuf_bp(db, db->db_blkptr, "blkptr=%p", db->db_blkptr);
4034eda14cbcSMatt Macy 
4035eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
4036eda14cbcSMatt Macy 
4037eda14cbcSMatt Macy 	ASSERT(db->db_level > 0);
4038eda14cbcSMatt Macy 	DBUF_VERIFY(db);
4039eda14cbcSMatt Macy 
4040eda14cbcSMatt Macy 	/* Read the block if it hasn't been read yet. */
4041eda14cbcSMatt Macy 	if (db->db_buf == NULL) {
4042eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
4043eda14cbcSMatt Macy 		(void) dbuf_read(db, NULL, DB_RF_MUST_SUCCEED);
4044eda14cbcSMatt Macy 		mutex_enter(&db->db_mtx);
4045eda14cbcSMatt Macy 	}
4046eda14cbcSMatt Macy 	ASSERT3U(db->db_state, ==, DB_CACHED);
4047eda14cbcSMatt Macy 	ASSERT(db->db_buf != NULL);
4048eda14cbcSMatt Macy 
4049eda14cbcSMatt Macy 	/* Indirect block size must match what the dnode thinks it is. */
4050eda14cbcSMatt Macy 	ASSERT3U(db->db.db_size, ==, 1<<dn->dn_phys->dn_indblkshift);
4051eda14cbcSMatt Macy 	dbuf_check_blkptr(dn, db);
4052eda14cbcSMatt Macy 
4053eda14cbcSMatt Macy 	/* Provide the pending dirty record to child dbufs */
4054eda14cbcSMatt Macy 	db->db_data_pending = dr;
4055eda14cbcSMatt Macy 
4056eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
4057eda14cbcSMatt Macy 
4058eda14cbcSMatt Macy 	dbuf_write(dr, db->db_buf, tx);
4059eda14cbcSMatt Macy 
40607877fdebSMatt Macy 	zio_t *zio = dr->dr_zio;
4061eda14cbcSMatt Macy 	mutex_enter(&dr->dt.di.dr_mtx);
4062eda14cbcSMatt Macy 	dbuf_sync_list(&dr->dt.di.dr_children, db->db_level - 1, tx);
4063eda14cbcSMatt Macy 	ASSERT(list_head(&dr->dt.di.dr_children) == NULL);
4064eda14cbcSMatt Macy 	mutex_exit(&dr->dt.di.dr_mtx);
4065eda14cbcSMatt Macy 	zio_nowait(zio);
4066eda14cbcSMatt Macy }
4067eda14cbcSMatt Macy 
4068eda14cbcSMatt Macy /*
4069eda14cbcSMatt Macy  * Verify that the size of the data in our bonus buffer does not exceed
4070eda14cbcSMatt Macy  * its recorded size.
4071eda14cbcSMatt Macy  *
4072eda14cbcSMatt Macy  * The purpose of this verification is to catch any cases in development
4073eda14cbcSMatt Macy  * where the size of a phys structure (i.e space_map_phys_t) grows and,
4074eda14cbcSMatt Macy  * due to incorrect feature management, older pools expect to read more
4075eda14cbcSMatt Macy  * data even though they didn't actually write it to begin with.
4076eda14cbcSMatt Macy  *
4077eda14cbcSMatt Macy  * For a example, this would catch an error in the feature logic where we
4078eda14cbcSMatt Macy  * open an older pool and we expect to write the space map histogram of
4079eda14cbcSMatt Macy  * a space map with size SPACE_MAP_SIZE_V0.
4080eda14cbcSMatt Macy  */
4081eda14cbcSMatt Macy static void
4082eda14cbcSMatt Macy dbuf_sync_leaf_verify_bonus_dnode(dbuf_dirty_record_t *dr)
4083eda14cbcSMatt Macy {
4084eda14cbcSMatt Macy #ifdef ZFS_DEBUG
40857877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
4086eda14cbcSMatt Macy 
4087eda14cbcSMatt Macy 	/*
4088eda14cbcSMatt Macy 	 * Encrypted bonus buffers can have data past their bonuslen.
4089eda14cbcSMatt Macy 	 * Skip the verification of these blocks.
4090eda14cbcSMatt Macy 	 */
4091eda14cbcSMatt Macy 	if (DMU_OT_IS_ENCRYPTED(dn->dn_bonustype))
4092eda14cbcSMatt Macy 		return;
4093eda14cbcSMatt Macy 
4094eda14cbcSMatt Macy 	uint16_t bonuslen = dn->dn_phys->dn_bonuslen;
4095eda14cbcSMatt Macy 	uint16_t maxbonuslen = DN_SLOTS_TO_BONUSLEN(dn->dn_num_slots);
4096eda14cbcSMatt Macy 	ASSERT3U(bonuslen, <=, maxbonuslen);
4097eda14cbcSMatt Macy 
4098eda14cbcSMatt Macy 	arc_buf_t *datap = dr->dt.dl.dr_data;
4099eda14cbcSMatt Macy 	char *datap_end = ((char *)datap) + bonuslen;
4100eda14cbcSMatt Macy 	char *datap_max = ((char *)datap) + maxbonuslen;
4101eda14cbcSMatt Macy 
4102eda14cbcSMatt Macy 	/* ensure that everything is zero after our data */
4103eda14cbcSMatt Macy 	for (; datap_end < datap_max; datap_end++)
4104eda14cbcSMatt Macy 		ASSERT(*datap_end == 0);
4105eda14cbcSMatt Macy #endif
4106eda14cbcSMatt Macy }
4107eda14cbcSMatt Macy 
41087877fdebSMatt Macy static blkptr_t *
41097877fdebSMatt Macy dbuf_lightweight_bp(dbuf_dirty_record_t *dr)
41107877fdebSMatt Macy {
41117877fdebSMatt Macy 	/* This must be a lightweight dirty record. */
41127877fdebSMatt Macy 	ASSERT3P(dr->dr_dbuf, ==, NULL);
41137877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
41147877fdebSMatt Macy 
41157877fdebSMatt Macy 	if (dn->dn_phys->dn_nlevels == 1) {
41167877fdebSMatt Macy 		VERIFY3U(dr->dt.dll.dr_blkid, <, dn->dn_phys->dn_nblkptr);
41177877fdebSMatt Macy 		return (&dn->dn_phys->dn_blkptr[dr->dt.dll.dr_blkid]);
41187877fdebSMatt Macy 	} else {
41197877fdebSMatt Macy 		dmu_buf_impl_t *parent_db = dr->dr_parent->dr_dbuf;
41207877fdebSMatt Macy 		int epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
41217877fdebSMatt Macy 		VERIFY3U(parent_db->db_level, ==, 1);
41227877fdebSMatt Macy 		VERIFY3P(parent_db->db_dnode_handle->dnh_dnode, ==, dn);
41237877fdebSMatt Macy 		VERIFY3U(dr->dt.dll.dr_blkid >> epbs, ==, parent_db->db_blkid);
41247877fdebSMatt Macy 		blkptr_t *bp = parent_db->db.db_data;
41257877fdebSMatt Macy 		return (&bp[dr->dt.dll.dr_blkid & ((1 << epbs) - 1)]);
41267877fdebSMatt Macy 	}
41277877fdebSMatt Macy }
41287877fdebSMatt Macy 
41297877fdebSMatt Macy static void
41307877fdebSMatt Macy dbuf_lightweight_ready(zio_t *zio)
41317877fdebSMatt Macy {
41327877fdebSMatt Macy 	dbuf_dirty_record_t *dr = zio->io_private;
41337877fdebSMatt Macy 	blkptr_t *bp = zio->io_bp;
41347877fdebSMatt Macy 
41357877fdebSMatt Macy 	if (zio->io_error != 0)
41367877fdebSMatt Macy 		return;
41377877fdebSMatt Macy 
41387877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
41397877fdebSMatt Macy 
41407877fdebSMatt Macy 	blkptr_t *bp_orig = dbuf_lightweight_bp(dr);
41417877fdebSMatt Macy 	spa_t *spa = dmu_objset_spa(dn->dn_objset);
41427877fdebSMatt Macy 	int64_t delta = bp_get_dsize_sync(spa, bp) -
41437877fdebSMatt Macy 	    bp_get_dsize_sync(spa, bp_orig);
41447877fdebSMatt Macy 	dnode_diduse_space(dn, delta);
41457877fdebSMatt Macy 
41467877fdebSMatt Macy 	uint64_t blkid = dr->dt.dll.dr_blkid;
41477877fdebSMatt Macy 	mutex_enter(&dn->dn_mtx);
41487877fdebSMatt Macy 	if (blkid > dn->dn_phys->dn_maxblkid) {
41497877fdebSMatt Macy 		ASSERT0(dn->dn_objset->os_raw_receive);
41507877fdebSMatt Macy 		dn->dn_phys->dn_maxblkid = blkid;
41517877fdebSMatt Macy 	}
41527877fdebSMatt Macy 	mutex_exit(&dn->dn_mtx);
41537877fdebSMatt Macy 
41547877fdebSMatt Macy 	if (!BP_IS_EMBEDDED(bp)) {
41557877fdebSMatt Macy 		uint64_t fill = BP_IS_HOLE(bp) ? 0 : 1;
41567877fdebSMatt Macy 		BP_SET_FILL(bp, fill);
41577877fdebSMatt Macy 	}
41587877fdebSMatt Macy 
41597877fdebSMatt Macy 	dmu_buf_impl_t *parent_db;
41607877fdebSMatt Macy 	EQUIV(dr->dr_parent == NULL, dn->dn_phys->dn_nlevels == 1);
41617877fdebSMatt Macy 	if (dr->dr_parent == NULL) {
41627877fdebSMatt Macy 		parent_db = dn->dn_dbuf;
41637877fdebSMatt Macy 	} else {
41647877fdebSMatt Macy 		parent_db = dr->dr_parent->dr_dbuf;
41657877fdebSMatt Macy 	}
41667877fdebSMatt Macy 	rw_enter(&parent_db->db_rwlock, RW_WRITER);
41677877fdebSMatt Macy 	*bp_orig = *bp;
41687877fdebSMatt Macy 	rw_exit(&parent_db->db_rwlock);
41697877fdebSMatt Macy }
41707877fdebSMatt Macy 
41717877fdebSMatt Macy static void
41727877fdebSMatt Macy dbuf_lightweight_physdone(zio_t *zio)
41737877fdebSMatt Macy {
41747877fdebSMatt Macy 	dbuf_dirty_record_t *dr = zio->io_private;
41757877fdebSMatt Macy 	dsl_pool_t *dp = spa_get_dsl(zio->io_spa);
41767877fdebSMatt Macy 	ASSERT3U(dr->dr_txg, ==, zio->io_txg);
41777877fdebSMatt Macy 
41787877fdebSMatt Macy 	/*
41797877fdebSMatt Macy 	 * The callback will be called io_phys_children times.  Retire one
41807877fdebSMatt Macy 	 * portion of our dirty space each time we are called.  Any rounding
41817877fdebSMatt Macy 	 * error will be cleaned up by dbuf_lightweight_done().
41827877fdebSMatt Macy 	 */
41837877fdebSMatt Macy 	int delta = dr->dr_accounted / zio->io_phys_children;
41847877fdebSMatt Macy 	dsl_pool_undirty_space(dp, delta, zio->io_txg);
41857877fdebSMatt Macy }
41867877fdebSMatt Macy 
41877877fdebSMatt Macy static void
41887877fdebSMatt Macy dbuf_lightweight_done(zio_t *zio)
41897877fdebSMatt Macy {
41907877fdebSMatt Macy 	dbuf_dirty_record_t *dr = zio->io_private;
41917877fdebSMatt Macy 
41927877fdebSMatt Macy 	VERIFY0(zio->io_error);
41937877fdebSMatt Macy 
41947877fdebSMatt Macy 	objset_t *os = dr->dr_dnode->dn_objset;
41957877fdebSMatt Macy 	dmu_tx_t *tx = os->os_synctx;
41967877fdebSMatt Macy 
41977877fdebSMatt Macy 	if (zio->io_flags & (ZIO_FLAG_IO_REWRITE | ZIO_FLAG_NOPWRITE)) {
41987877fdebSMatt Macy 		ASSERT(BP_EQUAL(zio->io_bp, &zio->io_bp_orig));
41997877fdebSMatt Macy 	} else {
42007877fdebSMatt Macy 		dsl_dataset_t *ds = os->os_dsl_dataset;
42017877fdebSMatt Macy 		(void) dsl_dataset_block_kill(ds, &zio->io_bp_orig, tx, B_TRUE);
42027877fdebSMatt Macy 		dsl_dataset_block_born(ds, zio->io_bp, tx);
42037877fdebSMatt Macy 	}
42047877fdebSMatt Macy 
42057877fdebSMatt Macy 	/*
42067877fdebSMatt Macy 	 * See comment in dbuf_write_done().
42077877fdebSMatt Macy 	 */
42087877fdebSMatt Macy 	if (zio->io_phys_children == 0) {
42097877fdebSMatt Macy 		dsl_pool_undirty_space(dmu_objset_pool(os),
42107877fdebSMatt Macy 		    dr->dr_accounted, zio->io_txg);
42117877fdebSMatt Macy 	} else {
42127877fdebSMatt Macy 		dsl_pool_undirty_space(dmu_objset_pool(os),
42137877fdebSMatt Macy 		    dr->dr_accounted % zio->io_phys_children, zio->io_txg);
42147877fdebSMatt Macy 	}
42157877fdebSMatt Macy 
42167877fdebSMatt Macy 	abd_free(dr->dt.dll.dr_abd);
42177877fdebSMatt Macy 	kmem_free(dr, sizeof (*dr));
42187877fdebSMatt Macy }
42197877fdebSMatt Macy 
42207877fdebSMatt Macy noinline static void
42217877fdebSMatt Macy dbuf_sync_lightweight(dbuf_dirty_record_t *dr, dmu_tx_t *tx)
42227877fdebSMatt Macy {
42237877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
42247877fdebSMatt Macy 	zio_t *pio;
42257877fdebSMatt Macy 	if (dn->dn_phys->dn_nlevels == 1) {
42267877fdebSMatt Macy 		pio = dn->dn_zio;
42277877fdebSMatt Macy 	} else {
42287877fdebSMatt Macy 		pio = dr->dr_parent->dr_zio;
42297877fdebSMatt Macy 	}
42307877fdebSMatt Macy 
42317877fdebSMatt Macy 	zbookmark_phys_t zb = {
42327877fdebSMatt Macy 		.zb_objset = dmu_objset_id(dn->dn_objset),
42337877fdebSMatt Macy 		.zb_object = dn->dn_object,
42347877fdebSMatt Macy 		.zb_level = 0,
42357877fdebSMatt Macy 		.zb_blkid = dr->dt.dll.dr_blkid,
42367877fdebSMatt Macy 	};
42377877fdebSMatt Macy 
42387877fdebSMatt Macy 	/*
42397877fdebSMatt Macy 	 * See comment in dbuf_write().  This is so that zio->io_bp_orig
42407877fdebSMatt Macy 	 * will have the old BP in dbuf_lightweight_done().
42417877fdebSMatt Macy 	 */
42427877fdebSMatt Macy 	dr->dr_bp_copy = *dbuf_lightweight_bp(dr);
42437877fdebSMatt Macy 
42447877fdebSMatt Macy 	dr->dr_zio = zio_write(pio, dmu_objset_spa(dn->dn_objset),
42457877fdebSMatt Macy 	    dmu_tx_get_txg(tx), &dr->dr_bp_copy, dr->dt.dll.dr_abd,
42467877fdebSMatt Macy 	    dn->dn_datablksz, abd_get_size(dr->dt.dll.dr_abd),
42477877fdebSMatt Macy 	    &dr->dt.dll.dr_props, dbuf_lightweight_ready, NULL,
42487877fdebSMatt Macy 	    dbuf_lightweight_physdone, dbuf_lightweight_done, dr,
42497877fdebSMatt Macy 	    ZIO_PRIORITY_ASYNC_WRITE,
42507877fdebSMatt Macy 	    ZIO_FLAG_MUSTSUCCEED | dr->dt.dll.dr_flags, &zb);
42517877fdebSMatt Macy 
42527877fdebSMatt Macy 	zio_nowait(dr->dr_zio);
42537877fdebSMatt Macy }
42547877fdebSMatt Macy 
4255eda14cbcSMatt Macy /*
4256eda14cbcSMatt Macy  * dbuf_sync_leaf() is called recursively from dbuf_sync_list() so it is
4257eda14cbcSMatt Macy  * critical the we not allow the compiler to inline this function in to
4258eda14cbcSMatt Macy  * dbuf_sync_list() thereby drastically bloating the stack usage.
4259eda14cbcSMatt Macy  */
4260eda14cbcSMatt Macy noinline static void
4261eda14cbcSMatt Macy dbuf_sync_leaf(dbuf_dirty_record_t *dr, dmu_tx_t *tx)
4262eda14cbcSMatt Macy {
4263eda14cbcSMatt Macy 	arc_buf_t **datap = &dr->dt.dl.dr_data;
4264eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
42657877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
4266eda14cbcSMatt Macy 	objset_t *os;
4267eda14cbcSMatt Macy 	uint64_t txg = tx->tx_txg;
4268eda14cbcSMatt Macy 
4269eda14cbcSMatt Macy 	ASSERT(dmu_tx_is_syncing(tx));
4270eda14cbcSMatt Macy 
4271eda14cbcSMatt Macy 	dprintf_dbuf_bp(db, db->db_blkptr, "blkptr=%p", db->db_blkptr);
4272eda14cbcSMatt Macy 
4273eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
4274eda14cbcSMatt Macy 	/*
4275eda14cbcSMatt Macy 	 * To be synced, we must be dirtied.  But we
4276eda14cbcSMatt Macy 	 * might have been freed after the dirty.
4277eda14cbcSMatt Macy 	 */
4278eda14cbcSMatt Macy 	if (db->db_state == DB_UNCACHED) {
4279eda14cbcSMatt Macy 		/* This buffer has been freed since it was dirtied */
4280eda14cbcSMatt Macy 		ASSERT(db->db.db_data == NULL);
4281eda14cbcSMatt Macy 	} else if (db->db_state == DB_FILL) {
4282eda14cbcSMatt Macy 		/* This buffer was freed and is now being re-filled */
4283eda14cbcSMatt Macy 		ASSERT(db->db.db_data != dr->dt.dl.dr_data);
4284eda14cbcSMatt Macy 	} else {
4285eda14cbcSMatt Macy 		ASSERT(db->db_state == DB_CACHED || db->db_state == DB_NOFILL);
4286eda14cbcSMatt Macy 	}
4287eda14cbcSMatt Macy 	DBUF_VERIFY(db);
4288eda14cbcSMatt Macy 
4289eda14cbcSMatt Macy 	if (db->db_blkid == DMU_SPILL_BLKID) {
4290eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
4291eda14cbcSMatt Macy 		if (!(dn->dn_phys->dn_flags & DNODE_FLAG_SPILL_BLKPTR)) {
4292eda14cbcSMatt Macy 			/*
4293eda14cbcSMatt Macy 			 * In the previous transaction group, the bonus buffer
4294eda14cbcSMatt Macy 			 * was entirely used to store the attributes for the
4295eda14cbcSMatt Macy 			 * dnode which overrode the dn_spill field.  However,
4296eda14cbcSMatt Macy 			 * when adding more attributes to the file a spill
4297eda14cbcSMatt Macy 			 * block was required to hold the extra attributes.
4298eda14cbcSMatt Macy 			 *
4299eda14cbcSMatt Macy 			 * Make sure to clear the garbage left in the dn_spill
4300eda14cbcSMatt Macy 			 * field from the previous attributes in the bonus
4301eda14cbcSMatt Macy 			 * buffer.  Otherwise, after writing out the spill
4302eda14cbcSMatt Macy 			 * block to the new allocated dva, it will free
4303eda14cbcSMatt Macy 			 * the old block pointed to by the invalid dn_spill.
4304eda14cbcSMatt Macy 			 */
4305eda14cbcSMatt Macy 			db->db_blkptr = NULL;
4306eda14cbcSMatt Macy 		}
4307eda14cbcSMatt Macy 		dn->dn_phys->dn_flags |= DNODE_FLAG_SPILL_BLKPTR;
4308eda14cbcSMatt Macy 		mutex_exit(&dn->dn_mtx);
4309eda14cbcSMatt Macy 	}
4310eda14cbcSMatt Macy 
4311eda14cbcSMatt Macy 	/*
4312eda14cbcSMatt Macy 	 * If this is a bonus buffer, simply copy the bonus data into the
4313eda14cbcSMatt Macy 	 * dnode.  It will be written out when the dnode is synced (and it
4314eda14cbcSMatt Macy 	 * will be synced, since it must have been dirty for dbuf_sync to
4315eda14cbcSMatt Macy 	 * be called).
4316eda14cbcSMatt Macy 	 */
4317eda14cbcSMatt Macy 	if (db->db_blkid == DMU_BONUS_BLKID) {
4318eda14cbcSMatt Macy 		ASSERT(dr->dr_dbuf == db);
4319eda14cbcSMatt Macy 		dbuf_sync_bonus(dr, tx);
4320eda14cbcSMatt Macy 		return;
4321eda14cbcSMatt Macy 	}
4322eda14cbcSMatt Macy 
4323eda14cbcSMatt Macy 	os = dn->dn_objset;
4324eda14cbcSMatt Macy 
4325eda14cbcSMatt Macy 	/*
4326eda14cbcSMatt Macy 	 * This function may have dropped the db_mtx lock allowing a dmu_sync
4327eda14cbcSMatt Macy 	 * operation to sneak in. As a result, we need to ensure that we
4328eda14cbcSMatt Macy 	 * don't check the dr_override_state until we have returned from
4329eda14cbcSMatt Macy 	 * dbuf_check_blkptr.
4330eda14cbcSMatt Macy 	 */
4331eda14cbcSMatt Macy 	dbuf_check_blkptr(dn, db);
4332eda14cbcSMatt Macy 
4333eda14cbcSMatt Macy 	/*
4334eda14cbcSMatt Macy 	 * If this buffer is in the middle of an immediate write,
4335eda14cbcSMatt Macy 	 * wait for the synchronous IO to complete.
4336eda14cbcSMatt Macy 	 */
4337eda14cbcSMatt Macy 	while (dr->dt.dl.dr_override_state == DR_IN_DMU_SYNC) {
4338eda14cbcSMatt Macy 		ASSERT(dn->dn_object != DMU_META_DNODE_OBJECT);
4339eda14cbcSMatt Macy 		cv_wait(&db->db_changed, &db->db_mtx);
4340eda14cbcSMatt Macy 		ASSERT(dr->dt.dl.dr_override_state != DR_NOT_OVERRIDDEN);
4341eda14cbcSMatt Macy 	}
4342eda14cbcSMatt Macy 
4343eda14cbcSMatt Macy 	/*
4344eda14cbcSMatt Macy 	 * If this is a dnode block, ensure it is appropriately encrypted
4345eda14cbcSMatt Macy 	 * or decrypted, depending on what we are writing to it this txg.
4346eda14cbcSMatt Macy 	 */
4347eda14cbcSMatt Macy 	if (os->os_encrypted && dn->dn_object == DMU_META_DNODE_OBJECT)
4348eda14cbcSMatt Macy 		dbuf_prepare_encrypted_dnode_leaf(dr);
4349eda14cbcSMatt Macy 
4350eda14cbcSMatt Macy 	if (db->db_state != DB_NOFILL &&
4351eda14cbcSMatt Macy 	    dn->dn_object != DMU_META_DNODE_OBJECT &&
4352eda14cbcSMatt Macy 	    zfs_refcount_count(&db->db_holds) > 1 &&
4353eda14cbcSMatt Macy 	    dr->dt.dl.dr_override_state != DR_OVERRIDDEN &&
4354eda14cbcSMatt Macy 	    *datap == db->db_buf) {
4355eda14cbcSMatt Macy 		/*
4356eda14cbcSMatt Macy 		 * If this buffer is currently "in use" (i.e., there
4357eda14cbcSMatt Macy 		 * are active holds and db_data still references it),
4358eda14cbcSMatt Macy 		 * then make a copy before we start the write so that
4359eda14cbcSMatt Macy 		 * any modifications from the open txg will not leak
4360eda14cbcSMatt Macy 		 * into this write.
4361eda14cbcSMatt Macy 		 *
4362eda14cbcSMatt Macy 		 * NOTE: this copy does not need to be made for
4363eda14cbcSMatt Macy 		 * objects only modified in the syncing context (e.g.
4364eda14cbcSMatt Macy 		 * DNONE_DNODE blocks).
4365eda14cbcSMatt Macy 		 */
4366eda14cbcSMatt Macy 		*datap = dbuf_alloc_arcbuf_from_arcbuf(db, db->db_buf);
4367eda14cbcSMatt Macy 		bcopy(db->db.db_data, (*datap)->b_data, arc_buf_size(*datap));
4368eda14cbcSMatt Macy 	}
4369eda14cbcSMatt Macy 	db->db_data_pending = dr;
4370eda14cbcSMatt Macy 
4371eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
4372eda14cbcSMatt Macy 
4373eda14cbcSMatt Macy 	dbuf_write(dr, *datap, tx);
4374eda14cbcSMatt Macy 
4375eda14cbcSMatt Macy 	ASSERT(!list_link_active(&dr->dr_dirty_node));
4376eda14cbcSMatt Macy 	if (dn->dn_object == DMU_META_DNODE_OBJECT) {
4377eda14cbcSMatt Macy 		list_insert_tail(&dn->dn_dirty_records[txg & TXG_MASK], dr);
4378eda14cbcSMatt Macy 	} else {
4379eda14cbcSMatt Macy 		zio_nowait(dr->dr_zio);
4380eda14cbcSMatt Macy 	}
4381eda14cbcSMatt Macy }
4382eda14cbcSMatt Macy 
4383eda14cbcSMatt Macy void
4384eda14cbcSMatt Macy dbuf_sync_list(list_t *list, int level, dmu_tx_t *tx)
4385eda14cbcSMatt Macy {
4386eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
4387eda14cbcSMatt Macy 
4388eda14cbcSMatt Macy 	while ((dr = list_head(list))) {
4389eda14cbcSMatt Macy 		if (dr->dr_zio != NULL) {
4390eda14cbcSMatt Macy 			/*
4391eda14cbcSMatt Macy 			 * If we find an already initialized zio then we
4392eda14cbcSMatt Macy 			 * are processing the meta-dnode, and we have finished.
4393eda14cbcSMatt Macy 			 * The dbufs for all dnodes are put back on the list
4394eda14cbcSMatt Macy 			 * during processing, so that we can zio_wait()
4395eda14cbcSMatt Macy 			 * these IOs after initiating all child IOs.
4396eda14cbcSMatt Macy 			 */
4397eda14cbcSMatt Macy 			ASSERT3U(dr->dr_dbuf->db.db_object, ==,
4398eda14cbcSMatt Macy 			    DMU_META_DNODE_OBJECT);
4399eda14cbcSMatt Macy 			break;
4400eda14cbcSMatt Macy 		}
44017877fdebSMatt Macy 		list_remove(list, dr);
44027877fdebSMatt Macy 		if (dr->dr_dbuf == NULL) {
44037877fdebSMatt Macy 			dbuf_sync_lightweight(dr, tx);
44047877fdebSMatt Macy 		} else {
4405eda14cbcSMatt Macy 			if (dr->dr_dbuf->db_blkid != DMU_BONUS_BLKID &&
4406eda14cbcSMatt Macy 			    dr->dr_dbuf->db_blkid != DMU_SPILL_BLKID) {
4407eda14cbcSMatt Macy 				VERIFY3U(dr->dr_dbuf->db_level, ==, level);
4408eda14cbcSMatt Macy 			}
4409eda14cbcSMatt Macy 			if (dr->dr_dbuf->db_level > 0)
4410eda14cbcSMatt Macy 				dbuf_sync_indirect(dr, tx);
4411eda14cbcSMatt Macy 			else
4412eda14cbcSMatt Macy 				dbuf_sync_leaf(dr, tx);
4413eda14cbcSMatt Macy 		}
4414eda14cbcSMatt Macy 	}
44157877fdebSMatt Macy }
4416eda14cbcSMatt Macy 
4417eda14cbcSMatt Macy /* ARGSUSED */
4418eda14cbcSMatt Macy static void
4419eda14cbcSMatt Macy dbuf_write_ready(zio_t *zio, arc_buf_t *buf, void *vdb)
4420eda14cbcSMatt Macy {
4421eda14cbcSMatt Macy 	dmu_buf_impl_t *db = vdb;
4422eda14cbcSMatt Macy 	dnode_t *dn;
4423eda14cbcSMatt Macy 	blkptr_t *bp = zio->io_bp;
4424eda14cbcSMatt Macy 	blkptr_t *bp_orig = &zio->io_bp_orig;
4425eda14cbcSMatt Macy 	spa_t *spa = zio->io_spa;
4426eda14cbcSMatt Macy 	int64_t delta;
4427eda14cbcSMatt Macy 	uint64_t fill = 0;
4428eda14cbcSMatt Macy 	int i;
4429eda14cbcSMatt Macy 
4430eda14cbcSMatt Macy 	ASSERT3P(db->db_blkptr, !=, NULL);
4431eda14cbcSMatt Macy 	ASSERT3P(&db->db_data_pending->dr_bp_copy, ==, bp);
4432eda14cbcSMatt Macy 
4433eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
4434eda14cbcSMatt Macy 	dn = DB_DNODE(db);
4435eda14cbcSMatt Macy 	delta = bp_get_dsize_sync(spa, bp) - bp_get_dsize_sync(spa, bp_orig);
4436eda14cbcSMatt Macy 	dnode_diduse_space(dn, delta - zio->io_prev_space_delta);
4437eda14cbcSMatt Macy 	zio->io_prev_space_delta = delta;
4438eda14cbcSMatt Macy 
4439eda14cbcSMatt Macy 	if (bp->blk_birth != 0) {
4440eda14cbcSMatt Macy 		ASSERT((db->db_blkid != DMU_SPILL_BLKID &&
4441eda14cbcSMatt Macy 		    BP_GET_TYPE(bp) == dn->dn_type) ||
4442eda14cbcSMatt Macy 		    (db->db_blkid == DMU_SPILL_BLKID &&
4443eda14cbcSMatt Macy 		    BP_GET_TYPE(bp) == dn->dn_bonustype) ||
4444eda14cbcSMatt Macy 		    BP_IS_EMBEDDED(bp));
4445eda14cbcSMatt Macy 		ASSERT(BP_GET_LEVEL(bp) == db->db_level);
4446eda14cbcSMatt Macy 	}
4447eda14cbcSMatt Macy 
4448eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
4449eda14cbcSMatt Macy 
4450eda14cbcSMatt Macy #ifdef ZFS_DEBUG
4451eda14cbcSMatt Macy 	if (db->db_blkid == DMU_SPILL_BLKID) {
4452eda14cbcSMatt Macy 		ASSERT(dn->dn_phys->dn_flags & DNODE_FLAG_SPILL_BLKPTR);
4453eda14cbcSMatt Macy 		ASSERT(!(BP_IS_HOLE(bp)) &&
4454eda14cbcSMatt Macy 		    db->db_blkptr == DN_SPILL_BLKPTR(dn->dn_phys));
4455eda14cbcSMatt Macy 	}
4456eda14cbcSMatt Macy #endif
4457eda14cbcSMatt Macy 
4458eda14cbcSMatt Macy 	if (db->db_level == 0) {
4459eda14cbcSMatt Macy 		mutex_enter(&dn->dn_mtx);
4460eda14cbcSMatt Macy 		if (db->db_blkid > dn->dn_phys->dn_maxblkid &&
4461eda14cbcSMatt Macy 		    db->db_blkid != DMU_SPILL_BLKID) {
4462eda14cbcSMatt Macy 			ASSERT0(db->db_objset->os_raw_receive);
4463eda14cbcSMatt Macy 			dn->dn_phys->dn_maxblkid = db->db_blkid;
4464eda14cbcSMatt Macy 		}
4465eda14cbcSMatt Macy 		mutex_exit(&dn->dn_mtx);
4466eda14cbcSMatt Macy 
4467eda14cbcSMatt Macy 		if (dn->dn_type == DMU_OT_DNODE) {
4468eda14cbcSMatt Macy 			i = 0;
4469eda14cbcSMatt Macy 			while (i < db->db.db_size) {
4470eda14cbcSMatt Macy 				dnode_phys_t *dnp =
4471eda14cbcSMatt Macy 				    (void *)(((char *)db->db.db_data) + i);
4472eda14cbcSMatt Macy 
4473eda14cbcSMatt Macy 				i += DNODE_MIN_SIZE;
4474eda14cbcSMatt Macy 				if (dnp->dn_type != DMU_OT_NONE) {
4475eda14cbcSMatt Macy 					fill++;
4476eda14cbcSMatt Macy 					i += dnp->dn_extra_slots *
4477eda14cbcSMatt Macy 					    DNODE_MIN_SIZE;
4478eda14cbcSMatt Macy 				}
4479eda14cbcSMatt Macy 			}
4480eda14cbcSMatt Macy 		} else {
4481eda14cbcSMatt Macy 			if (BP_IS_HOLE(bp)) {
4482eda14cbcSMatt Macy 				fill = 0;
4483eda14cbcSMatt Macy 			} else {
4484eda14cbcSMatt Macy 				fill = 1;
4485eda14cbcSMatt Macy 			}
4486eda14cbcSMatt Macy 		}
4487eda14cbcSMatt Macy 	} else {
4488eda14cbcSMatt Macy 		blkptr_t *ibp = db->db.db_data;
4489eda14cbcSMatt Macy 		ASSERT3U(db->db.db_size, ==, 1<<dn->dn_phys->dn_indblkshift);
4490eda14cbcSMatt Macy 		for (i = db->db.db_size >> SPA_BLKPTRSHIFT; i > 0; i--, ibp++) {
4491eda14cbcSMatt Macy 			if (BP_IS_HOLE(ibp))
4492eda14cbcSMatt Macy 				continue;
4493eda14cbcSMatt Macy 			fill += BP_GET_FILL(ibp);
4494eda14cbcSMatt Macy 		}
4495eda14cbcSMatt Macy 	}
4496eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
4497eda14cbcSMatt Macy 
4498eda14cbcSMatt Macy 	if (!BP_IS_EMBEDDED(bp))
4499eda14cbcSMatt Macy 		BP_SET_FILL(bp, fill);
4500eda14cbcSMatt Macy 
4501eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
4502eda14cbcSMatt Macy 
4503eda14cbcSMatt Macy 	db_lock_type_t dblt = dmu_buf_lock_parent(db, RW_WRITER, FTAG);
4504eda14cbcSMatt Macy 	*db->db_blkptr = *bp;
4505eda14cbcSMatt Macy 	dmu_buf_unlock_parent(db, dblt, FTAG);
4506eda14cbcSMatt Macy }
4507eda14cbcSMatt Macy 
4508eda14cbcSMatt Macy /* ARGSUSED */
4509eda14cbcSMatt Macy /*
4510eda14cbcSMatt Macy  * This function gets called just prior to running through the compression
4511eda14cbcSMatt Macy  * stage of the zio pipeline. If we're an indirect block comprised of only
4512eda14cbcSMatt Macy  * holes, then we want this indirect to be compressed away to a hole. In
4513eda14cbcSMatt Macy  * order to do that we must zero out any information about the holes that
4514eda14cbcSMatt Macy  * this indirect points to prior to before we try to compress it.
4515eda14cbcSMatt Macy  */
4516eda14cbcSMatt Macy static void
4517eda14cbcSMatt Macy dbuf_write_children_ready(zio_t *zio, arc_buf_t *buf, void *vdb)
4518eda14cbcSMatt Macy {
4519eda14cbcSMatt Macy 	dmu_buf_impl_t *db = vdb;
4520eda14cbcSMatt Macy 	dnode_t *dn;
4521eda14cbcSMatt Macy 	blkptr_t *bp;
4522eda14cbcSMatt Macy 	unsigned int epbs, i;
4523eda14cbcSMatt Macy 
4524eda14cbcSMatt Macy 	ASSERT3U(db->db_level, >, 0);
4525eda14cbcSMatt Macy 	DB_DNODE_ENTER(db);
4526eda14cbcSMatt Macy 	dn = DB_DNODE(db);
4527eda14cbcSMatt Macy 	epbs = dn->dn_phys->dn_indblkshift - SPA_BLKPTRSHIFT;
4528eda14cbcSMatt Macy 	ASSERT3U(epbs, <, 31);
4529eda14cbcSMatt Macy 
4530eda14cbcSMatt Macy 	/* Determine if all our children are holes */
4531eda14cbcSMatt Macy 	for (i = 0, bp = db->db.db_data; i < 1ULL << epbs; i++, bp++) {
4532eda14cbcSMatt Macy 		if (!BP_IS_HOLE(bp))
4533eda14cbcSMatt Macy 			break;
4534eda14cbcSMatt Macy 	}
4535eda14cbcSMatt Macy 
4536eda14cbcSMatt Macy 	/*
4537eda14cbcSMatt Macy 	 * If all the children are holes, then zero them all out so that
4538eda14cbcSMatt Macy 	 * we may get compressed away.
4539eda14cbcSMatt Macy 	 */
4540eda14cbcSMatt Macy 	if (i == 1ULL << epbs) {
4541eda14cbcSMatt Macy 		/*
4542eda14cbcSMatt Macy 		 * We only found holes. Grab the rwlock to prevent
4543eda14cbcSMatt Macy 		 * anybody from reading the blocks we're about to
4544eda14cbcSMatt Macy 		 * zero out.
4545eda14cbcSMatt Macy 		 */
4546eda14cbcSMatt Macy 		rw_enter(&db->db_rwlock, RW_WRITER);
4547eda14cbcSMatt Macy 		bzero(db->db.db_data, db->db.db_size);
4548eda14cbcSMatt Macy 		rw_exit(&db->db_rwlock);
4549eda14cbcSMatt Macy 	}
4550eda14cbcSMatt Macy 	DB_DNODE_EXIT(db);
4551eda14cbcSMatt Macy }
4552eda14cbcSMatt Macy 
4553eda14cbcSMatt Macy /*
4554eda14cbcSMatt Macy  * The SPA will call this callback several times for each zio - once
4555eda14cbcSMatt Macy  * for every physical child i/o (zio->io_phys_children times).  This
4556eda14cbcSMatt Macy  * allows the DMU to monitor the progress of each logical i/o.  For example,
4557eda14cbcSMatt Macy  * there may be 2 copies of an indirect block, or many fragments of a RAID-Z
4558eda14cbcSMatt Macy  * block.  There may be a long delay before all copies/fragments are completed,
4559eda14cbcSMatt Macy  * so this callback allows us to retire dirty space gradually, as the physical
4560eda14cbcSMatt Macy  * i/os complete.
4561eda14cbcSMatt Macy  */
4562eda14cbcSMatt Macy /* ARGSUSED */
4563eda14cbcSMatt Macy static void
4564eda14cbcSMatt Macy dbuf_write_physdone(zio_t *zio, arc_buf_t *buf, void *arg)
4565eda14cbcSMatt Macy {
4566eda14cbcSMatt Macy 	dmu_buf_impl_t *db = arg;
4567eda14cbcSMatt Macy 	objset_t *os = db->db_objset;
4568eda14cbcSMatt Macy 	dsl_pool_t *dp = dmu_objset_pool(os);
4569eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr;
4570eda14cbcSMatt Macy 	int delta = 0;
4571eda14cbcSMatt Macy 
4572eda14cbcSMatt Macy 	dr = db->db_data_pending;
4573eda14cbcSMatt Macy 	ASSERT3U(dr->dr_txg, ==, zio->io_txg);
4574eda14cbcSMatt Macy 
4575eda14cbcSMatt Macy 	/*
4576eda14cbcSMatt Macy 	 * The callback will be called io_phys_children times.  Retire one
4577eda14cbcSMatt Macy 	 * portion of our dirty space each time we are called.  Any rounding
4578eda14cbcSMatt Macy 	 * error will be cleaned up by dbuf_write_done().
4579eda14cbcSMatt Macy 	 */
4580eda14cbcSMatt Macy 	delta = dr->dr_accounted / zio->io_phys_children;
4581eda14cbcSMatt Macy 	dsl_pool_undirty_space(dp, delta, zio->io_txg);
4582eda14cbcSMatt Macy }
4583eda14cbcSMatt Macy 
4584eda14cbcSMatt Macy /* ARGSUSED */
4585eda14cbcSMatt Macy static void
4586eda14cbcSMatt Macy dbuf_write_done(zio_t *zio, arc_buf_t *buf, void *vdb)
4587eda14cbcSMatt Macy {
4588eda14cbcSMatt Macy 	dmu_buf_impl_t *db = vdb;
4589eda14cbcSMatt Macy 	blkptr_t *bp_orig = &zio->io_bp_orig;
4590eda14cbcSMatt Macy 	blkptr_t *bp = db->db_blkptr;
4591eda14cbcSMatt Macy 	objset_t *os = db->db_objset;
4592eda14cbcSMatt Macy 	dmu_tx_t *tx = os->os_synctx;
4593eda14cbcSMatt Macy 
4594eda14cbcSMatt Macy 	ASSERT0(zio->io_error);
4595eda14cbcSMatt Macy 	ASSERT(db->db_blkptr == bp);
4596eda14cbcSMatt Macy 
4597eda14cbcSMatt Macy 	/*
4598eda14cbcSMatt Macy 	 * For nopwrites and rewrites we ensure that the bp matches our
4599eda14cbcSMatt Macy 	 * original and bypass all the accounting.
4600eda14cbcSMatt Macy 	 */
4601eda14cbcSMatt Macy 	if (zio->io_flags & (ZIO_FLAG_IO_REWRITE | ZIO_FLAG_NOPWRITE)) {
4602eda14cbcSMatt Macy 		ASSERT(BP_EQUAL(bp, bp_orig));
4603eda14cbcSMatt Macy 	} else {
4604eda14cbcSMatt Macy 		dsl_dataset_t *ds = os->os_dsl_dataset;
4605eda14cbcSMatt Macy 		(void) dsl_dataset_block_kill(ds, bp_orig, tx, B_TRUE);
4606eda14cbcSMatt Macy 		dsl_dataset_block_born(ds, bp, tx);
4607eda14cbcSMatt Macy 	}
4608eda14cbcSMatt Macy 
4609eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
4610eda14cbcSMatt Macy 
4611eda14cbcSMatt Macy 	DBUF_VERIFY(db);
4612eda14cbcSMatt Macy 
46137877fdebSMatt Macy 	dbuf_dirty_record_t *dr = db->db_data_pending;
46147877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
4615eda14cbcSMatt Macy 	ASSERT(!list_link_active(&dr->dr_dirty_node));
4616eda14cbcSMatt Macy 	ASSERT(dr->dr_dbuf == db);
4617eda14cbcSMatt Macy 	ASSERT(list_next(&db->db_dirty_records, dr) == NULL);
4618eda14cbcSMatt Macy 	list_remove(&db->db_dirty_records, dr);
4619eda14cbcSMatt Macy 
4620eda14cbcSMatt Macy #ifdef ZFS_DEBUG
4621eda14cbcSMatt Macy 	if (db->db_blkid == DMU_SPILL_BLKID) {
4622eda14cbcSMatt Macy 		ASSERT(dn->dn_phys->dn_flags & DNODE_FLAG_SPILL_BLKPTR);
4623eda14cbcSMatt Macy 		ASSERT(!(BP_IS_HOLE(db->db_blkptr)) &&
4624eda14cbcSMatt Macy 		    db->db_blkptr == DN_SPILL_BLKPTR(dn->dn_phys));
4625eda14cbcSMatt Macy 	}
4626eda14cbcSMatt Macy #endif
4627eda14cbcSMatt Macy 
4628eda14cbcSMatt Macy 	if (db->db_level == 0) {
4629eda14cbcSMatt Macy 		ASSERT(db->db_blkid != DMU_BONUS_BLKID);
4630eda14cbcSMatt Macy 		ASSERT(dr->dt.dl.dr_override_state == DR_NOT_OVERRIDDEN);
4631eda14cbcSMatt Macy 		if (db->db_state != DB_NOFILL) {
4632eda14cbcSMatt Macy 			if (dr->dt.dl.dr_data != db->db_buf)
4633eda14cbcSMatt Macy 				arc_buf_destroy(dr->dt.dl.dr_data, db);
4634eda14cbcSMatt Macy 		}
4635eda14cbcSMatt Macy 	} else {
4636eda14cbcSMatt Macy 		ASSERT(list_head(&dr->dt.di.dr_children) == NULL);
4637eda14cbcSMatt Macy 		ASSERT3U(db->db.db_size, ==, 1 << dn->dn_phys->dn_indblkshift);
4638eda14cbcSMatt Macy 		if (!BP_IS_HOLE(db->db_blkptr)) {
4639eda14cbcSMatt Macy 			int epbs __maybe_unused = dn->dn_phys->dn_indblkshift -
4640eda14cbcSMatt Macy 			    SPA_BLKPTRSHIFT;
4641eda14cbcSMatt Macy 			ASSERT3U(db->db_blkid, <=,
4642eda14cbcSMatt Macy 			    dn->dn_phys->dn_maxblkid >> (db->db_level * epbs));
4643eda14cbcSMatt Macy 			ASSERT3U(BP_GET_LSIZE(db->db_blkptr), ==,
4644eda14cbcSMatt Macy 			    db->db.db_size);
4645eda14cbcSMatt Macy 		}
4646eda14cbcSMatt Macy 		mutex_destroy(&dr->dt.di.dr_mtx);
4647eda14cbcSMatt Macy 		list_destroy(&dr->dt.di.dr_children);
4648eda14cbcSMatt Macy 	}
4649eda14cbcSMatt Macy 
4650eda14cbcSMatt Macy 	cv_broadcast(&db->db_changed);
4651eda14cbcSMatt Macy 	ASSERT(db->db_dirtycnt > 0);
4652eda14cbcSMatt Macy 	db->db_dirtycnt -= 1;
4653eda14cbcSMatt Macy 	db->db_data_pending = NULL;
4654eda14cbcSMatt Macy 	dbuf_rele_and_unlock(db, (void *)(uintptr_t)tx->tx_txg, B_FALSE);
4655eda14cbcSMatt Macy 
4656eda14cbcSMatt Macy 	/*
4657eda14cbcSMatt Macy 	 * If we didn't do a physical write in this ZIO and we
4658eda14cbcSMatt Macy 	 * still ended up here, it means that the space of the
4659eda14cbcSMatt Macy 	 * dbuf that we just released (and undirtied) above hasn't
4660eda14cbcSMatt Macy 	 * been marked as undirtied in the pool's accounting.
4661eda14cbcSMatt Macy 	 *
4662eda14cbcSMatt Macy 	 * Thus, we undirty that space in the pool's view of the
4663eda14cbcSMatt Macy 	 * world here. For physical writes this type of update
4664eda14cbcSMatt Macy 	 * happens in dbuf_write_physdone().
4665eda14cbcSMatt Macy 	 *
4666eda14cbcSMatt Macy 	 * If we did a physical write, cleanup any rounding errors
4667eda14cbcSMatt Macy 	 * that came up due to writing multiple copies of a block
4668eda14cbcSMatt Macy 	 * on disk [see dbuf_write_physdone()].
4669eda14cbcSMatt Macy 	 */
4670eda14cbcSMatt Macy 	if (zio->io_phys_children == 0) {
4671eda14cbcSMatt Macy 		dsl_pool_undirty_space(dmu_objset_pool(os),
4672eda14cbcSMatt Macy 		    dr->dr_accounted, zio->io_txg);
4673eda14cbcSMatt Macy 	} else {
4674eda14cbcSMatt Macy 		dsl_pool_undirty_space(dmu_objset_pool(os),
4675eda14cbcSMatt Macy 		    dr->dr_accounted % zio->io_phys_children, zio->io_txg);
4676eda14cbcSMatt Macy 	}
4677eda14cbcSMatt Macy 
4678eda14cbcSMatt Macy 	kmem_free(dr, sizeof (dbuf_dirty_record_t));
4679eda14cbcSMatt Macy }
4680eda14cbcSMatt Macy 
4681eda14cbcSMatt Macy static void
4682eda14cbcSMatt Macy dbuf_write_nofill_ready(zio_t *zio)
4683eda14cbcSMatt Macy {
4684eda14cbcSMatt Macy 	dbuf_write_ready(zio, NULL, zio->io_private);
4685eda14cbcSMatt Macy }
4686eda14cbcSMatt Macy 
4687eda14cbcSMatt Macy static void
4688eda14cbcSMatt Macy dbuf_write_nofill_done(zio_t *zio)
4689eda14cbcSMatt Macy {
4690eda14cbcSMatt Macy 	dbuf_write_done(zio, NULL, zio->io_private);
4691eda14cbcSMatt Macy }
4692eda14cbcSMatt Macy 
4693eda14cbcSMatt Macy static void
4694eda14cbcSMatt Macy dbuf_write_override_ready(zio_t *zio)
4695eda14cbcSMatt Macy {
4696eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr = zio->io_private;
4697eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
4698eda14cbcSMatt Macy 
4699eda14cbcSMatt Macy 	dbuf_write_ready(zio, NULL, db);
4700eda14cbcSMatt Macy }
4701eda14cbcSMatt Macy 
4702eda14cbcSMatt Macy static void
4703eda14cbcSMatt Macy dbuf_write_override_done(zio_t *zio)
4704eda14cbcSMatt Macy {
4705eda14cbcSMatt Macy 	dbuf_dirty_record_t *dr = zio->io_private;
4706eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
4707eda14cbcSMatt Macy 	blkptr_t *obp = &dr->dt.dl.dr_overridden_by;
4708eda14cbcSMatt Macy 
4709eda14cbcSMatt Macy 	mutex_enter(&db->db_mtx);
4710eda14cbcSMatt Macy 	if (!BP_EQUAL(zio->io_bp, obp)) {
4711eda14cbcSMatt Macy 		if (!BP_IS_HOLE(obp))
4712eda14cbcSMatt Macy 			dsl_free(spa_get_dsl(zio->io_spa), zio->io_txg, obp);
4713eda14cbcSMatt Macy 		arc_release(dr->dt.dl.dr_data, db);
4714eda14cbcSMatt Macy 	}
4715eda14cbcSMatt Macy 	mutex_exit(&db->db_mtx);
4716eda14cbcSMatt Macy 
4717eda14cbcSMatt Macy 	dbuf_write_done(zio, NULL, db);
4718eda14cbcSMatt Macy 
4719eda14cbcSMatt Macy 	if (zio->io_abd != NULL)
4720184c1b94SMartin Matuska 		abd_free(zio->io_abd);
4721eda14cbcSMatt Macy }
4722eda14cbcSMatt Macy 
4723eda14cbcSMatt Macy typedef struct dbuf_remap_impl_callback_arg {
4724eda14cbcSMatt Macy 	objset_t	*drica_os;
4725eda14cbcSMatt Macy 	uint64_t	drica_blk_birth;
4726eda14cbcSMatt Macy 	dmu_tx_t	*drica_tx;
4727eda14cbcSMatt Macy } dbuf_remap_impl_callback_arg_t;
4728eda14cbcSMatt Macy 
4729eda14cbcSMatt Macy static void
4730eda14cbcSMatt Macy dbuf_remap_impl_callback(uint64_t vdev, uint64_t offset, uint64_t size,
4731eda14cbcSMatt Macy     void *arg)
4732eda14cbcSMatt Macy {
4733eda14cbcSMatt Macy 	dbuf_remap_impl_callback_arg_t *drica = arg;
4734eda14cbcSMatt Macy 	objset_t *os = drica->drica_os;
4735eda14cbcSMatt Macy 	spa_t *spa = dmu_objset_spa(os);
4736eda14cbcSMatt Macy 	dmu_tx_t *tx = drica->drica_tx;
4737eda14cbcSMatt Macy 
4738eda14cbcSMatt Macy 	ASSERT(dsl_pool_sync_context(spa_get_dsl(spa)));
4739eda14cbcSMatt Macy 
4740eda14cbcSMatt Macy 	if (os == spa_meta_objset(spa)) {
4741eda14cbcSMatt Macy 		spa_vdev_indirect_mark_obsolete(spa, vdev, offset, size, tx);
4742eda14cbcSMatt Macy 	} else {
4743eda14cbcSMatt Macy 		dsl_dataset_block_remapped(dmu_objset_ds(os), vdev, offset,
4744eda14cbcSMatt Macy 		    size, drica->drica_blk_birth, tx);
4745eda14cbcSMatt Macy 	}
4746eda14cbcSMatt Macy }
4747eda14cbcSMatt Macy 
4748eda14cbcSMatt Macy static void
4749eda14cbcSMatt Macy dbuf_remap_impl(dnode_t *dn, blkptr_t *bp, krwlock_t *rw, dmu_tx_t *tx)
4750eda14cbcSMatt Macy {
4751eda14cbcSMatt Macy 	blkptr_t bp_copy = *bp;
4752eda14cbcSMatt Macy 	spa_t *spa = dmu_objset_spa(dn->dn_objset);
4753eda14cbcSMatt Macy 	dbuf_remap_impl_callback_arg_t drica;
4754eda14cbcSMatt Macy 
4755eda14cbcSMatt Macy 	ASSERT(dsl_pool_sync_context(spa_get_dsl(spa)));
4756eda14cbcSMatt Macy 
4757eda14cbcSMatt Macy 	drica.drica_os = dn->dn_objset;
4758eda14cbcSMatt Macy 	drica.drica_blk_birth = bp->blk_birth;
4759eda14cbcSMatt Macy 	drica.drica_tx = tx;
4760eda14cbcSMatt Macy 	if (spa_remap_blkptr(spa, &bp_copy, dbuf_remap_impl_callback,
4761eda14cbcSMatt Macy 	    &drica)) {
4762eda14cbcSMatt Macy 		/*
4763eda14cbcSMatt Macy 		 * If the blkptr being remapped is tracked by a livelist,
4764eda14cbcSMatt Macy 		 * then we need to make sure the livelist reflects the update.
4765eda14cbcSMatt Macy 		 * First, cancel out the old blkptr by appending a 'FREE'
4766eda14cbcSMatt Macy 		 * entry. Next, add an 'ALLOC' to track the new version. This
4767eda14cbcSMatt Macy 		 * way we avoid trying to free an inaccurate blkptr at delete.
4768eda14cbcSMatt Macy 		 * Note that embedded blkptrs are not tracked in livelists.
4769eda14cbcSMatt Macy 		 */
4770eda14cbcSMatt Macy 		if (dn->dn_objset != spa_meta_objset(spa)) {
4771eda14cbcSMatt Macy 			dsl_dataset_t *ds = dmu_objset_ds(dn->dn_objset);
4772eda14cbcSMatt Macy 			if (dsl_deadlist_is_open(&ds->ds_dir->dd_livelist) &&
4773eda14cbcSMatt Macy 			    bp->blk_birth > ds->ds_dir->dd_origin_txg) {
4774eda14cbcSMatt Macy 				ASSERT(!BP_IS_EMBEDDED(bp));
4775eda14cbcSMatt Macy 				ASSERT(dsl_dir_is_clone(ds->ds_dir));
4776eda14cbcSMatt Macy 				ASSERT(spa_feature_is_enabled(spa,
4777eda14cbcSMatt Macy 				    SPA_FEATURE_LIVELIST));
4778eda14cbcSMatt Macy 				bplist_append(&ds->ds_dir->dd_pending_frees,
4779eda14cbcSMatt Macy 				    bp);
4780eda14cbcSMatt Macy 				bplist_append(&ds->ds_dir->dd_pending_allocs,
4781eda14cbcSMatt Macy 				    &bp_copy);
4782eda14cbcSMatt Macy 			}
4783eda14cbcSMatt Macy 		}
4784eda14cbcSMatt Macy 
4785eda14cbcSMatt Macy 		/*
4786eda14cbcSMatt Macy 		 * The db_rwlock prevents dbuf_read_impl() from
4787eda14cbcSMatt Macy 		 * dereferencing the BP while we are changing it.  To
4788eda14cbcSMatt Macy 		 * avoid lock contention, only grab it when we are actually
4789eda14cbcSMatt Macy 		 * changing the BP.
4790eda14cbcSMatt Macy 		 */
4791eda14cbcSMatt Macy 		if (rw != NULL)
4792eda14cbcSMatt Macy 			rw_enter(rw, RW_WRITER);
4793eda14cbcSMatt Macy 		*bp = bp_copy;
4794eda14cbcSMatt Macy 		if (rw != NULL)
4795eda14cbcSMatt Macy 			rw_exit(rw);
4796eda14cbcSMatt Macy 	}
4797eda14cbcSMatt Macy }
4798eda14cbcSMatt Macy 
4799eda14cbcSMatt Macy /*
4800eda14cbcSMatt Macy  * Remap any existing BP's to concrete vdevs, if possible.
4801eda14cbcSMatt Macy  */
4802eda14cbcSMatt Macy static void
4803eda14cbcSMatt Macy dbuf_remap(dnode_t *dn, dmu_buf_impl_t *db, dmu_tx_t *tx)
4804eda14cbcSMatt Macy {
4805eda14cbcSMatt Macy 	spa_t *spa = dmu_objset_spa(db->db_objset);
4806eda14cbcSMatt Macy 	ASSERT(dsl_pool_sync_context(spa_get_dsl(spa)));
4807eda14cbcSMatt Macy 
4808eda14cbcSMatt Macy 	if (!spa_feature_is_active(spa, SPA_FEATURE_DEVICE_REMOVAL))
4809eda14cbcSMatt Macy 		return;
4810eda14cbcSMatt Macy 
4811eda14cbcSMatt Macy 	if (db->db_level > 0) {
4812eda14cbcSMatt Macy 		blkptr_t *bp = db->db.db_data;
4813eda14cbcSMatt Macy 		for (int i = 0; i < db->db.db_size >> SPA_BLKPTRSHIFT; i++) {
4814eda14cbcSMatt Macy 			dbuf_remap_impl(dn, &bp[i], &db->db_rwlock, tx);
4815eda14cbcSMatt Macy 		}
4816eda14cbcSMatt Macy 	} else if (db->db.db_object == DMU_META_DNODE_OBJECT) {
4817eda14cbcSMatt Macy 		dnode_phys_t *dnp = db->db.db_data;
4818eda14cbcSMatt Macy 		ASSERT3U(db->db_dnode_handle->dnh_dnode->dn_type, ==,
4819eda14cbcSMatt Macy 		    DMU_OT_DNODE);
4820eda14cbcSMatt Macy 		for (int i = 0; i < db->db.db_size >> DNODE_SHIFT;
4821eda14cbcSMatt Macy 		    i += dnp[i].dn_extra_slots + 1) {
4822eda14cbcSMatt Macy 			for (int j = 0; j < dnp[i].dn_nblkptr; j++) {
4823eda14cbcSMatt Macy 				krwlock_t *lock = (dn->dn_dbuf == NULL ? NULL :
4824eda14cbcSMatt Macy 				    &dn->dn_dbuf->db_rwlock);
4825eda14cbcSMatt Macy 				dbuf_remap_impl(dn, &dnp[i].dn_blkptr[j], lock,
4826eda14cbcSMatt Macy 				    tx);
4827eda14cbcSMatt Macy 			}
4828eda14cbcSMatt Macy 		}
4829eda14cbcSMatt Macy 	}
4830eda14cbcSMatt Macy }
4831eda14cbcSMatt Macy 
4832eda14cbcSMatt Macy 
4833eda14cbcSMatt Macy /* Issue I/O to commit a dirty buffer to disk. */
4834eda14cbcSMatt Macy static void
4835eda14cbcSMatt Macy dbuf_write(dbuf_dirty_record_t *dr, arc_buf_t *data, dmu_tx_t *tx)
4836eda14cbcSMatt Macy {
4837eda14cbcSMatt Macy 	dmu_buf_impl_t *db = dr->dr_dbuf;
48387877fdebSMatt Macy 	dnode_t *dn = dr->dr_dnode;
4839eda14cbcSMatt Macy 	objset_t *os;
4840eda14cbcSMatt Macy 	dmu_buf_impl_t *parent = db->db_parent;
4841eda14cbcSMatt Macy 	uint64_t txg = tx->tx_txg;
4842eda14cbcSMatt Macy 	zbookmark_phys_t zb;
4843eda14cbcSMatt Macy 	zio_prop_t zp;
4844eda14cbcSMatt Macy 	zio_t *pio; /* parent I/O */
4845eda14cbcSMatt Macy 	int wp_flag = 0;
4846eda14cbcSMatt Macy 
4847eda14cbcSMatt Macy 	ASSERT(dmu_tx_is_syncing(tx));
4848eda14cbcSMatt Macy 
4849eda14cbcSMatt Macy 	os = dn->dn_objset;
4850eda14cbcSMatt Macy 
4851eda14cbcSMatt Macy 	if (db->db_state != DB_NOFILL) {
4852eda14cbcSMatt Macy 		if (db->db_level > 0 || dn->dn_type == DMU_OT_DNODE) {
4853eda14cbcSMatt Macy 			/*
4854eda14cbcSMatt Macy 			 * Private object buffers are released here rather
4855eda14cbcSMatt Macy 			 * than in dbuf_dirty() since they are only modified
4856eda14cbcSMatt Macy 			 * in the syncing context and we don't want the
4857eda14cbcSMatt Macy 			 * overhead of making multiple copies of the data.
4858eda14cbcSMatt Macy 			 */
4859eda14cbcSMatt Macy 			if (BP_IS_HOLE(db->db_blkptr)) {
4860eda14cbcSMatt Macy 				arc_buf_thaw(data);
4861eda14cbcSMatt Macy 			} else {
4862eda14cbcSMatt Macy 				dbuf_release_bp(db);
4863eda14cbcSMatt Macy 			}
4864eda14cbcSMatt Macy 			dbuf_remap(dn, db, tx);
4865eda14cbcSMatt Macy 		}
4866eda14cbcSMatt Macy 	}
4867eda14cbcSMatt Macy 
4868eda14cbcSMatt Macy 	if (parent != dn->dn_dbuf) {
4869eda14cbcSMatt Macy 		/* Our parent is an indirect block. */
4870eda14cbcSMatt Macy 		/* We have a dirty parent that has been scheduled for write. */
4871eda14cbcSMatt Macy 		ASSERT(parent && parent->db_data_pending);
4872eda14cbcSMatt Macy 		/* Our parent's buffer is one level closer to the dnode. */
4873eda14cbcSMatt Macy 		ASSERT(db->db_level == parent->db_level-1);
4874eda14cbcSMatt Macy 		/*
4875eda14cbcSMatt Macy 		 * We're about to modify our parent's db_data by modifying
4876eda14cbcSMatt Macy 		 * our block pointer, so the parent must be released.
4877eda14cbcSMatt Macy 		 */
4878eda14cbcSMatt Macy 		ASSERT(arc_released(parent->db_buf));
4879eda14cbcSMatt Macy 		pio = parent->db_data_pending->dr_zio;
4880eda14cbcSMatt Macy 	} else {
4881eda14cbcSMatt Macy 		/* Our parent is the dnode itself. */
4882eda14cbcSMatt Macy 		ASSERT((db->db_level == dn->dn_phys->dn_nlevels-1 &&
4883eda14cbcSMatt Macy 		    db->db_blkid != DMU_SPILL_BLKID) ||
4884eda14cbcSMatt Macy 		    (db->db_blkid == DMU_SPILL_BLKID && db->db_level == 0));
4885eda14cbcSMatt Macy 		if (db->db_blkid != DMU_SPILL_BLKID)
4886eda14cbcSMatt Macy 			ASSERT3P(db->db_blkptr, ==,
4887eda14cbcSMatt Macy 			    &dn->dn_phys->dn_blkptr[db->db_blkid]);
4888eda14cbcSMatt Macy 		pio = dn->dn_zio;
4889eda14cbcSMatt Macy 	}
4890eda14cbcSMatt Macy 
4891eda14cbcSMatt Macy 	ASSERT(db->db_level == 0 || data == db->db_buf);
4892eda14cbcSMatt Macy 	ASSERT3U(db->db_blkptr->blk_birth, <=, txg);
4893eda14cbcSMatt Macy 	ASSERT(pio);
4894eda14cbcSMatt Macy 
4895eda14cbcSMatt Macy 	SET_BOOKMARK(&zb, os->os_dsl_dataset ?
4896eda14cbcSMatt Macy 	    os->os_dsl_dataset->ds_object : DMU_META_OBJSET,
4897eda14cbcSMatt Macy 	    db->db.db_object, db->db_level, db->db_blkid);
4898eda14cbcSMatt Macy 
4899eda14cbcSMatt Macy 	if (db->db_blkid == DMU_SPILL_BLKID)
4900eda14cbcSMatt Macy 		wp_flag = WP_SPILL;
4901eda14cbcSMatt Macy 	wp_flag |= (db->db_state == DB_NOFILL) ? WP_NOFILL : 0;
4902eda14cbcSMatt Macy 
4903eda14cbcSMatt Macy 	dmu_write_policy(os, dn, db->db_level, wp_flag, &zp);
4904eda14cbcSMatt Macy 
4905eda14cbcSMatt Macy 	/*
4906eda14cbcSMatt Macy 	 * We copy the blkptr now (rather than when we instantiate the dirty
4907eda14cbcSMatt Macy 	 * record), because its value can change between open context and
4908eda14cbcSMatt Macy 	 * syncing context. We do not need to hold dn_struct_rwlock to read
4909eda14cbcSMatt Macy 	 * db_blkptr because we are in syncing context.
4910eda14cbcSMatt Macy 	 */
4911eda14cbcSMatt Macy 	dr->dr_bp_copy = *db->db_blkptr;
4912eda14cbcSMatt Macy 
4913eda14cbcSMatt Macy 	if (db->db_level == 0 &&
4914eda14cbcSMatt Macy 	    dr->dt.dl.dr_override_state == DR_OVERRIDDEN) {
4915eda14cbcSMatt Macy 		/*
4916eda14cbcSMatt Macy 		 * The BP for this block has been provided by open context
4917eda14cbcSMatt Macy 		 * (by dmu_sync() or dmu_buf_write_embedded()).
4918eda14cbcSMatt Macy 		 */
4919eda14cbcSMatt Macy 		abd_t *contents = (data != NULL) ?
4920eda14cbcSMatt Macy 		    abd_get_from_buf(data->b_data, arc_buf_size(data)) : NULL;
4921eda14cbcSMatt Macy 
4922eda14cbcSMatt Macy 		dr->dr_zio = zio_write(pio, os->os_spa, txg, &dr->dr_bp_copy,
4923eda14cbcSMatt Macy 		    contents, db->db.db_size, db->db.db_size, &zp,
4924eda14cbcSMatt Macy 		    dbuf_write_override_ready, NULL, NULL,
4925eda14cbcSMatt Macy 		    dbuf_write_override_done,
4926eda14cbcSMatt Macy 		    dr, ZIO_PRIORITY_ASYNC_WRITE, ZIO_FLAG_MUSTSUCCEED, &zb);
4927eda14cbcSMatt Macy 		mutex_enter(&db->db_mtx);
4928eda14cbcSMatt Macy 		dr->dt.dl.dr_override_state = DR_NOT_OVERRIDDEN;
4929eda14cbcSMatt Macy 		zio_write_override(dr->dr_zio, &dr->dt.dl.dr_overridden_by,
4930eda14cbcSMatt Macy 		    dr->dt.dl.dr_copies, dr->dt.dl.dr_nopwrite);
4931eda14cbcSMatt Macy 		mutex_exit(&db->db_mtx);
4932eda14cbcSMatt Macy 	} else if (db->db_state == DB_NOFILL) {
4933eda14cbcSMatt Macy 		ASSERT(zp.zp_checksum == ZIO_CHECKSUM_OFF ||
4934eda14cbcSMatt Macy 		    zp.zp_checksum == ZIO_CHECKSUM_NOPARITY);
4935eda14cbcSMatt Macy 		dr->dr_zio = zio_write(pio, os->os_spa, txg,
4936eda14cbcSMatt Macy 		    &dr->dr_bp_copy, NULL, db->db.db_size, db->db.db_size, &zp,
4937eda14cbcSMatt Macy 		    dbuf_write_nofill_ready, NULL, NULL,
4938eda14cbcSMatt Macy 		    dbuf_write_nofill_done, db,
4939eda14cbcSMatt Macy 		    ZIO_PRIORITY_ASYNC_WRITE,
4940eda14cbcSMatt Macy 		    ZIO_FLAG_MUSTSUCCEED | ZIO_FLAG_NODATA, &zb);
4941eda14cbcSMatt Macy 	} else {
4942eda14cbcSMatt Macy 		ASSERT(arc_released(data));
4943eda14cbcSMatt Macy 
4944eda14cbcSMatt Macy 		/*
4945eda14cbcSMatt Macy 		 * For indirect blocks, we want to setup the children
4946eda14cbcSMatt Macy 		 * ready callback so that we can properly handle an indirect
4947eda14cbcSMatt Macy 		 * block that only contains holes.
4948eda14cbcSMatt Macy 		 */
4949eda14cbcSMatt Macy 		arc_write_done_func_t *children_ready_cb = NULL;
4950eda14cbcSMatt Macy 		if (db->db_level != 0)
4951eda14cbcSMatt Macy 			children_ready_cb = dbuf_write_children_ready;
4952eda14cbcSMatt Macy 
4953eda14cbcSMatt Macy 		dr->dr_zio = arc_write(pio, os->os_spa, txg,
4954eda14cbcSMatt Macy 		    &dr->dr_bp_copy, data, DBUF_IS_L2CACHEABLE(db),
4955eda14cbcSMatt Macy 		    &zp, dbuf_write_ready,
4956eda14cbcSMatt Macy 		    children_ready_cb, dbuf_write_physdone,
4957eda14cbcSMatt Macy 		    dbuf_write_done, db, ZIO_PRIORITY_ASYNC_WRITE,
4958eda14cbcSMatt Macy 		    ZIO_FLAG_MUSTSUCCEED, &zb);
4959eda14cbcSMatt Macy 	}
4960eda14cbcSMatt Macy }
4961eda14cbcSMatt Macy 
4962eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_find);
4963eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_is_metadata);
4964eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_destroy);
4965eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_loan_arcbuf);
4966eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_whichblock);
4967eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_read);
4968eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_unoverride);
4969eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_free_range);
4970eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_new_size);
4971eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_release_bp);
4972eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_dirty);
4973eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_set_crypt_params);
4974eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_will_dirty);
4975eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_is_dirty);
4976eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_will_not_fill);
4977eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_will_fill);
4978eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_fill_done);
4979eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_rele);
4980eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_assign_arcbuf);
4981eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_prefetch);
4982eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_hold_impl);
4983eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_hold);
4984eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_hold_level);
4985eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_create_bonus);
4986eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_spill_set_blksz);
4987eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_rm_spill);
4988eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_add_ref);
4989eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_rele);
4990eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_rele_and_unlock);
4991eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_refcount);
4992eda14cbcSMatt Macy EXPORT_SYMBOL(dbuf_sync_list);
4993eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_set_user);
4994eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_set_user_ie);
4995eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_get_user);
4996eda14cbcSMatt Macy EXPORT_SYMBOL(dmu_buf_get_blkptr);
4997eda14cbcSMatt Macy 
4998eda14cbcSMatt Macy /* BEGIN CSTYLED */
4999eda14cbcSMatt Macy ZFS_MODULE_PARAM(zfs_dbuf_cache, dbuf_cache_, max_bytes, ULONG, ZMOD_RW,
5000eda14cbcSMatt Macy 	"Maximum size in bytes of the dbuf cache.");
5001eda14cbcSMatt Macy 
5002eda14cbcSMatt Macy ZFS_MODULE_PARAM(zfs_dbuf_cache, dbuf_cache_, hiwater_pct, UINT, ZMOD_RW,
5003eda14cbcSMatt Macy 	"Percentage over dbuf_cache_max_bytes when dbufs must be evicted "
5004eda14cbcSMatt Macy 	"directly.");
5005eda14cbcSMatt Macy 
5006eda14cbcSMatt Macy ZFS_MODULE_PARAM(zfs_dbuf_cache, dbuf_cache_, lowater_pct, UINT, ZMOD_RW,
5007eda14cbcSMatt Macy 	"Percentage below dbuf_cache_max_bytes when the evict thread stops "
5008eda14cbcSMatt Macy 	"evicting dbufs.");
5009eda14cbcSMatt Macy 
5010eda14cbcSMatt Macy ZFS_MODULE_PARAM(zfs_dbuf, dbuf_, metadata_cache_max_bytes, ULONG, ZMOD_RW,
5011eda14cbcSMatt Macy 	"Maximum size in bytes of the dbuf metadata cache.");
5012eda14cbcSMatt Macy 
5013eda14cbcSMatt Macy ZFS_MODULE_PARAM(zfs_dbuf, dbuf_, cache_shift, INT, ZMOD_RW,
5014eda14cbcSMatt Macy 	"Set the size of the dbuf cache to a log2 fraction of arc size.");
5015eda14cbcSMatt Macy 
5016eda14cbcSMatt Macy ZFS_MODULE_PARAM(zfs_dbuf, dbuf_, metadata_cache_shift, INT, ZMOD_RW,
5017eda14cbcSMatt Macy 	"Set the size of the dbuf metadata cache to a log2 fraction of arc "
5018eda14cbcSMatt Macy 	"size.");
5019eda14cbcSMatt Macy /* END CSTYLED */
5020