xref: /illumos-gate/usr/src/uts/common/fs/zfs/zio.c (revision f6ef4223)
1fa9e4066Sahrens /*
2fa9e4066Sahrens  * CDDL HEADER START
3fa9e4066Sahrens  *
4fa9e4066Sahrens  * The contents of this file are subject to the terms of the
5ea8dc4b6Seschrock  * Common Development and Distribution License (the "License").
6ea8dc4b6Seschrock  * You may not use this file except in compliance with the License.
7fa9e4066Sahrens  *
8fa9e4066Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9fa9e4066Sahrens  * or http://www.opensolaris.org/os/licensing.
10fa9e4066Sahrens  * See the License for the specific language governing permissions
11fa9e4066Sahrens  * and limitations under the License.
12fa9e4066Sahrens  *
13fa9e4066Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14fa9e4066Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15fa9e4066Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16fa9e4066Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17fa9e4066Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18fa9e4066Sahrens  *
19fa9e4066Sahrens  * CDDL HEADER END
20fa9e4066Sahrens  */
21fa9e4066Sahrens /*
223f9d6ad7SLin Ling  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
23f78cdc34SPaul Dagnelie  * Copyright (c) 2011, 2018 by Delphix. All rights reserved.
245aeb9474SGarrett D'Amore  * Copyright (c) 2011 Nexenta Systems, Inc. All rights reserved.
25c3d26abcSMatthew Ahrens  * Copyright (c) 2014 Integros [integros.com]
26663207adSDon Brady  * Copyright (c) 2017, Intel Corporation.
278548ec78SJohn Levon  * Copyright 2020 Joyent, Inc.
28fa9e4066Sahrens  */
29fa9e4066Sahrens 
30de710d24SJosef 'Jeff' Sipek #include <sys/sysmacros.h>
31fa9e4066Sahrens #include <sys/zfs_context.h>
32ea8dc4b6Seschrock #include <sys/fm/fs/zfs.h>
33fa9e4066Sahrens #include <sys/spa.h>
34fa9e4066Sahrens #include <sys/txg.h>
35fa9e4066Sahrens #include <sys/spa_impl.h>
36fa9e4066Sahrens #include <sys/vdev_impl.h>
37084fd14fSBrian Behlendorf #include <sys/vdev_trim.h>
38fa9e4066Sahrens #include <sys/zio_impl.h>
39fa9e4066Sahrens #include <sys/zio_compress.h>
40fa9e4066Sahrens #include <sys/zio_checksum.h>
41b24ab676SJeff Bonwick #include <sys/dmu_objset.h>
42b24ab676SJeff Bonwick #include <sys/arc.h>
43b24ab676SJeff Bonwick #include <sys/ddt.h>
445d7b4d43SMatthew Ahrens #include <sys/blkptr.h>
4543466aaeSMax Grossman #include <sys/zfeature.h>
46dd50e0ccSTony Hutter #include <sys/time.h>
47a3874b8bSToomas Soome #include <sys/dsl_scan.h>
480f7643c7SGeorge Wilson #include <sys/metaslab_impl.h>
49770499e1SDan Kimmel #include <sys/abd.h>
50f78cdc34SPaul Dagnelie #include <sys/cityhash.h>
51eb633035STom Caputi #include <sys/dsl_crypt.h>
52*f6ef4223SJoshua M. Clulow #include <sys/stdbool.h>
53fa9e4066Sahrens 
54fa9e4066Sahrens /*
55fa9e4066Sahrens  * ==========================================================================
56fa9e4066Sahrens  * I/O type descriptions
57fa9e4066Sahrens  * ==========================================================================
58fa9e4066Sahrens  */
5969962b56SMatthew Ahrens const char *zio_type_name[ZIO_TYPES] = {
6080eb36f2SGeorge Wilson 	"zio_null", "zio_read", "zio_write", "zio_free", "zio_claim",
61084fd14fSBrian Behlendorf 	"zio_ioctl", "z_trim"
6280eb36f2SGeorge Wilson };
63fa9e4066Sahrens 
640f7643c7SGeorge Wilson boolean_t zio_dva_throttle_enabled = B_TRUE;
650f7643c7SGeorge Wilson 
66fa9e4066Sahrens /*
67fa9e4066Sahrens  * ==========================================================================
68fa9e4066Sahrens  * I/O kmem caches
69fa9e4066Sahrens  * ==========================================================================
70fa9e4066Sahrens  */
71ccae0b50Seschrock kmem_cache_t *zio_cache;
72a3f829aeSBill Moore kmem_cache_t *zio_link_cache;
73fa9e4066Sahrens kmem_cache_t *zio_buf_cache[SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT];
74ad23a2dbSjohansen kmem_cache_t *zio_data_buf_cache[SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT];
75ad23a2dbSjohansen 
76ad23a2dbSjohansen #ifdef _KERNEL
77ad23a2dbSjohansen extern vmem_t *zio_alloc_arena;
78ad23a2dbSjohansen #endif
79fa9e4066Sahrens 
80738f37bcSGeorge Wilson #define	ZIO_PIPELINE_CONTINUE		0x100
81738f37bcSGeorge Wilson #define	ZIO_PIPELINE_STOP		0x101
82738f37bcSGeorge Wilson 
83dd50e0ccSTony Hutter /* Mark IOs as "slow" if they take longer than 30 seconds */
84dd50e0ccSTony Hutter int zio_slow_io_ms = (30 * MILLISEC);
85dd50e0ccSTony Hutter 
86a2cdcdd2SPaul Dagnelie #define	BP_SPANB(indblkshift, level) \
87a2cdcdd2SPaul Dagnelie 	(((uint64_t)1) << ((level) * ((indblkshift) - SPA_BLKPTRSHIFT)))
88a2cdcdd2SPaul Dagnelie #define	COMPARE_META_LEVEL	0x80000000ul
8901f55e48SGeorge Wilson /*
9001f55e48SGeorge Wilson  * The following actions directly effect the spa's sync-to-convergence logic.
9101f55e48SGeorge Wilson  * The values below define the sync pass when we start performing the action.
9201f55e48SGeorge Wilson  * Care should be taken when changing these values as they directly impact
9301f55e48SGeorge Wilson  * spa_sync() performance. Tuning these values may introduce subtle performance
9401f55e48SGeorge Wilson  * pathologies and should only be done in the context of performance analysis.
9501f55e48SGeorge Wilson  * These tunables will eventually be removed and replaced with #defines once
9601f55e48SGeorge Wilson  * enough analysis has been done to determine optimal values.
9701f55e48SGeorge Wilson  *
9801f55e48SGeorge Wilson  * The 'zfs_sync_pass_deferred_free' pass must be greater than 1 to ensure that
9901f55e48SGeorge Wilson  * regular blocks are not deferred.
10001f55e48SGeorge Wilson  */
10101f55e48SGeorge Wilson int zfs_sync_pass_deferred_free = 2; /* defer frees starting in this pass */
10201f55e48SGeorge Wilson int zfs_sync_pass_dont_compress = 5; /* don't compress starting in this pass */
10301f55e48SGeorge Wilson int zfs_sync_pass_rewrite = 2; /* rewrite new bps starting in this pass */
10401f55e48SGeorge Wilson 
1050a4e9518Sgw /*
106e14bb325SJeff Bonwick  * An allocating zio is one that either currently has the DVA allocate
107e14bb325SJeff Bonwick  * stage set or will have it later in its lifetime.
1080a4e9518Sgw  */
109b24ab676SJeff Bonwick #define	IO_IS_ALLOCATING(zio) ((zio)->io_orig_pipeline & ZIO_STAGE_DVA_ALLOCATE)
110b24ab676SJeff Bonwick 
11135a5a358SJonathan Adams boolean_t	zio_requeue_io_start_cut_in_line = B_TRUE;
11235a5a358SJonathan Adams 
113b24ab676SJeff Bonwick #ifdef ZFS_DEBUG
114b24ab676SJeff Bonwick int zio_buf_debug_limit = 16384;
115b24ab676SJeff Bonwick #else
116b24ab676SJeff Bonwick int zio_buf_debug_limit = 0;
117b24ab676SJeff Bonwick #endif
1180a4e9518Sgw 
1190f7643c7SGeorge Wilson static void zio_taskq_dispatch(zio_t *, zio_taskq_type_t, boolean_t);
1200f7643c7SGeorge Wilson 
121fa9e4066Sahrens void
122fa9e4066Sahrens zio_init(void)
123fa9e4066Sahrens {
124fa9e4066Sahrens 	size_t c;
125ad23a2dbSjohansen 	vmem_t *data_alloc_arena = NULL;
126ad23a2dbSjohansen 
127ad23a2dbSjohansen #ifdef _KERNEL
128ad23a2dbSjohansen 	data_alloc_arena = zio_alloc_arena;
129ad23a2dbSjohansen #endif
130a3f829aeSBill Moore 	zio_cache = kmem_cache_create("zio_cache",
131a3f829aeSBill Moore 	    sizeof (zio_t), 0, NULL, NULL, NULL, NULL, NULL, 0);
132a3f829aeSBill Moore 	zio_link_cache = kmem_cache_create("zio_link_cache",
133a3f829aeSBill Moore 	    sizeof (zio_link_t), 0, NULL, NULL, NULL, NULL, NULL, 0);
134ccae0b50Seschrock 
135fa9e4066Sahrens 	/*
136fa9e4066Sahrens 	 * For small buffers, we want a cache for each multiple of
137b5152584SMatthew Ahrens 	 * SPA_MINBLOCKSIZE.  For larger buffers, we want a cache
138b5152584SMatthew Ahrens 	 * for each quarter-power of 2.
139fa9e4066Sahrens 	 */
140fa9e4066Sahrens 	for (c = 0; c < SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT; c++) {
141fa9e4066Sahrens 		size_t size = (c + 1) << SPA_MINBLOCKSHIFT;
142fa9e4066Sahrens 		size_t p2 = size;
143fa9e4066Sahrens 		size_t align = 0;
144e291592aSJonathan Adams 		size_t cflags = (size > zio_buf_debug_limit) ? KMC_NODEBUG : 0;
145fa9e4066Sahrens 
146de710d24SJosef 'Jeff' Sipek 		while (!ISP2(p2))
147fa9e4066Sahrens 			p2 &= p2 - 1;
148fa9e4066Sahrens 
149cd1c8b85SMatthew Ahrens #ifndef _KERNEL
150cd1c8b85SMatthew Ahrens 		/*
151cd1c8b85SMatthew Ahrens 		 * If we are using watchpoints, put each buffer on its own page,
152cd1c8b85SMatthew Ahrens 		 * to eliminate the performance overhead of trapping to the
153cd1c8b85SMatthew Ahrens 		 * kernel when modifying a non-watched buffer that shares the
154cd1c8b85SMatthew Ahrens 		 * page with a watched buffer.
155cd1c8b85SMatthew Ahrens 		 */
156cd1c8b85SMatthew Ahrens 		if (arc_watch && !IS_P2ALIGNED(size, PAGESIZE))
157cd1c8b85SMatthew Ahrens 			continue;
158cd1c8b85SMatthew Ahrens #endif
159fa9e4066Sahrens 		if (size <= 4 * SPA_MINBLOCKSIZE) {
160fa9e4066Sahrens 			align = SPA_MINBLOCKSIZE;
161cd1c8b85SMatthew Ahrens 		} else if (IS_P2ALIGNED(size, p2 >> 2)) {
162b5152584SMatthew Ahrens 			align = MIN(p2 >> 2, PAGESIZE);
163fa9e4066Sahrens 		}
164fa9e4066Sahrens 
165fa9e4066Sahrens 		if (align != 0) {
166ad23a2dbSjohansen 			char name[36];
1675ad82045Snd 			(void) sprintf(name, "zio_buf_%lu", (ulong_t)size);
168fa9e4066Sahrens 			zio_buf_cache[c] = kmem_cache_create(name, size,
169e291592aSJonathan Adams 			    align, NULL, NULL, NULL, NULL, NULL, cflags);
170ad23a2dbSjohansen 
171e291592aSJonathan Adams 			/*
172e291592aSJonathan Adams 			 * Since zio_data bufs do not appear in crash dumps, we
173e291592aSJonathan Adams 			 * pass KMC_NOTOUCH so that no allocator metadata is
174e291592aSJonathan Adams 			 * stored with the buffers.
175e291592aSJonathan Adams 			 */
176ad23a2dbSjohansen 			(void) sprintf(name, "zio_data_buf_%lu", (ulong_t)size);
177ad23a2dbSjohansen 			zio_data_buf_cache[c] = kmem_cache_create(name, size,
178ad23a2dbSjohansen 			    align, NULL, NULL, NULL, NULL, data_alloc_arena,
179e291592aSJonathan Adams 			    cflags | KMC_NOTOUCH);
180fa9e4066Sahrens 		}
181fa9e4066Sahrens 	}
182fa9e4066Sahrens 
183fa9e4066Sahrens 	while (--c != 0) {
184fa9e4066Sahrens 		ASSERT(zio_buf_cache[c] != NULL);
185fa9e4066Sahrens 		if (zio_buf_cache[c - 1] == NULL)
186fa9e4066Sahrens 			zio_buf_cache[c - 1] = zio_buf_cache[c];
187ad23a2dbSjohansen 
188ad23a2dbSjohansen 		ASSERT(zio_data_buf_cache[c] != NULL);
189ad23a2dbSjohansen 		if (zio_data_buf_cache[c - 1] == NULL)
190ad23a2dbSjohansen 			zio_data_buf_cache[c - 1] = zio_data_buf_cache[c];
191fa9e4066Sahrens 	}
192ea8dc4b6Seschrock 
193ea8dc4b6Seschrock 	zio_inject_init();
194fa9e4066Sahrens }
195fa9e4066Sahrens 
196fa9e4066Sahrens void
197fa9e4066Sahrens zio_fini(void)
198fa9e4066Sahrens {
199fa9e4066Sahrens 	size_t c;
200fa9e4066Sahrens 	kmem_cache_t *last_cache = NULL;
201ad23a2dbSjohansen 	kmem_cache_t *last_data_cache = NULL;
202fa9e4066Sahrens 
203fa9e4066Sahrens 	for (c = 0; c < SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT; c++) {
204fa9e4066Sahrens 		if (zio_buf_cache[c] != last_cache) {
205fa9e4066Sahrens 			last_cache = zio_buf_cache[c];
206fa9e4066Sahrens 			kmem_cache_destroy(zio_buf_cache[c]);
207fa9e4066Sahrens 		}
208fa9e4066Sahrens 		zio_buf_cache[c] = NULL;
209ad23a2dbSjohansen 
210ad23a2dbSjohansen 		if (zio_data_buf_cache[c] != last_data_cache) {
211ad23a2dbSjohansen 			last_data_cache = zio_data_buf_cache[c];
212ad23a2dbSjohansen 			kmem_cache_destroy(zio_data_buf_cache[c]);
213ad23a2dbSjohansen 		}
214ad23a2dbSjohansen 		zio_data_buf_cache[c] = NULL;
215fa9e4066Sahrens 	}
216ea8dc4b6Seschrock 
217a3f829aeSBill Moore 	kmem_cache_destroy(zio_link_cache);
218ccae0b50Seschrock 	kmem_cache_destroy(zio_cache);
219ccae0b50Seschrock 
220ea8dc4b6Seschrock 	zio_inject_fini();
221fa9e4066Sahrens }
222fa9e4066Sahrens 
223fa9e4066Sahrens /*
224fa9e4066Sahrens  * ==========================================================================
225fa9e4066Sahrens  * Allocate and free I/O buffers
226fa9e4066Sahrens  * ==========================================================================
227fa9e4066Sahrens  */
228ad23a2dbSjohansen 
229ad23a2dbSjohansen /*
230ad23a2dbSjohansen  * Use zio_buf_alloc to allocate ZFS metadata.  This data will appear in a
231ad23a2dbSjohansen  * crashdump if the kernel panics, so use it judiciously.  Obviously, it's
232ad23a2dbSjohansen  * useful to inspect ZFS metadata, but if possible, we should avoid keeping
233ad23a2dbSjohansen  * excess / transient data in-core during a crashdump.
234ad23a2dbSjohansen  */
235fa9e4066Sahrens void *
236fa9e4066Sahrens zio_buf_alloc(size_t size)
237fa9e4066Sahrens {
238fa9e4066Sahrens 	size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
239fa9e4066Sahrens 
240f63ab3d5SMatthew Ahrens 	VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
241fa9e4066Sahrens 
2421ab7f2deSmaybee 	return (kmem_cache_alloc(zio_buf_cache[c], KM_PUSHPAGE));
243fa9e4066Sahrens }
244fa9e4066Sahrens 
245ad23a2dbSjohansen /*
246ad23a2dbSjohansen  * Use zio_data_buf_alloc to allocate data.  The data will not appear in a
247ad23a2dbSjohansen  * crashdump if the kernel panics.  This exists so that we will limit the amount
248ad23a2dbSjohansen  * of ZFS data that shows up in a kernel crashdump.  (Thus reducing the amount
249ad23a2dbSjohansen  * of kernel heap dumped to disk when the kernel panics)
250ad23a2dbSjohansen  */
251ad23a2dbSjohansen void *
252ad23a2dbSjohansen zio_data_buf_alloc(size_t size)
253ad23a2dbSjohansen {
254ad23a2dbSjohansen 	size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
255ad23a2dbSjohansen 
256f63ab3d5SMatthew Ahrens 	VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
257ad23a2dbSjohansen 
2581ab7f2deSmaybee 	return (kmem_cache_alloc(zio_data_buf_cache[c], KM_PUSHPAGE));
259ad23a2dbSjohansen }
260ad23a2dbSjohansen 
261fa9e4066Sahrens void
262fa9e4066Sahrens zio_buf_free(void *buf, size_t size)
263fa9e4066Sahrens {
264fa9e4066Sahrens 	size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
265fa9e4066Sahrens 
266f63ab3d5SMatthew Ahrens 	VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
267fa9e4066Sahrens 
268fa9e4066Sahrens 	kmem_cache_free(zio_buf_cache[c], buf);
269fa9e4066Sahrens }
270fa9e4066Sahrens 
271ad23a2dbSjohansen void
272ad23a2dbSjohansen zio_data_buf_free(void *buf, size_t size)
273ad23a2dbSjohansen {
274ad23a2dbSjohansen 	size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
275ad23a2dbSjohansen 
276f63ab3d5SMatthew Ahrens 	VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
277ad23a2dbSjohansen 
278ad23a2dbSjohansen 	kmem_cache_free(zio_data_buf_cache[c], buf);
279ad23a2dbSjohansen }
280b3995adbSahrens 
281eb633035STom Caputi /* ARGSUSED */
282eb633035STom Caputi static void
283eb633035STom Caputi zio_abd_free(void *abd, size_t size)
284eb633035STom Caputi {
285eb633035STom Caputi 	abd_free((abd_t *)abd);
286eb633035STom Caputi }
287eb633035STom Caputi 
288fa9e4066Sahrens /*
289fa9e4066Sahrens  * ==========================================================================
290fa9e4066Sahrens  * Push and pop I/O transform buffers
291fa9e4066Sahrens  * ==========================================================================
292fa9e4066Sahrens  */
293dcbf3bd6SGeorge Wilson void
294770499e1SDan Kimmel zio_push_transform(zio_t *zio, abd_t *data, uint64_t size, uint64_t bufsize,
2959a686fbcSPaul Dagnelie     zio_transform_func_t *transform)
296fa9e4066Sahrens {
297fa9e4066Sahrens 	zio_transform_t *zt = kmem_alloc(sizeof (zio_transform_t), KM_SLEEP);
298fa9e4066Sahrens 
299770499e1SDan Kimmel 	/*
300770499e1SDan Kimmel 	 * Ensure that anyone expecting this zio to contain a linear ABD isn't
301770499e1SDan Kimmel 	 * going to get a nasty surprise when they try to access the data.
302770499e1SDan Kimmel 	 */
303770499e1SDan Kimmel 	IMPLY(abd_is_linear(zio->io_abd), abd_is_linear(data));
304770499e1SDan Kimmel 
305770499e1SDan Kimmel 	zt->zt_orig_abd = zio->io_abd;
306e14bb325SJeff Bonwick 	zt->zt_orig_size = zio->io_size;
307fa9e4066Sahrens 	zt->zt_bufsize = bufsize;
308e14bb325SJeff Bonwick 	zt->zt_transform = transform;
309fa9e4066Sahrens 
310fa9e4066Sahrens 	zt->zt_next = zio->io_transform_stack;
311fa9e4066Sahrens 	zio->io_transform_stack = zt;
312fa9e4066Sahrens 
313770499e1SDan Kimmel 	zio->io_abd = data;
314fa9e4066Sahrens 	zio->io_size = size;
315fa9e4066Sahrens }
316fa9e4066Sahrens 
317dcbf3bd6SGeorge Wilson void
318e14bb325SJeff Bonwick zio_pop_transforms(zio_t *zio)
319fa9e4066Sahrens {
320e14bb325SJeff Bonwick 	zio_transform_t *zt;
321e14bb325SJeff Bonwick 
322e14bb325SJeff Bonwick 	while ((zt = zio->io_transform_stack) != NULL) {
323e14bb325SJeff Bonwick 		if (zt->zt_transform != NULL)
324e14bb325SJeff Bonwick 			zt->zt_transform(zio,
325770499e1SDan Kimmel 			    zt->zt_orig_abd, zt->zt_orig_size);
326fa9e4066Sahrens 
327b24ab676SJeff Bonwick 		if (zt->zt_bufsize != 0)
328770499e1SDan Kimmel 			abd_free(zio->io_abd);
329fa9e4066Sahrens 
330770499e1SDan Kimmel 		zio->io_abd = zt->zt_orig_abd;
331e14bb325SJeff Bonwick 		zio->io_size = zt->zt_orig_size;
332e14bb325SJeff Bonwick 		zio->io_transform_stack = zt->zt_next;
333fa9e4066Sahrens 
334e14bb325SJeff Bonwick 		kmem_free(zt, sizeof (zio_transform_t));
335fa9e4066Sahrens 	}
336fa9e4066Sahrens }
337fa9e4066Sahrens 
338e14bb325SJeff Bonwick /*
339e14bb325SJeff Bonwick  * ==========================================================================
340eb633035STom Caputi  * I/O transform callbacks for subblocks, decompression, and decryption
341e14bb325SJeff Bonwick  * ==========================================================================
342e14bb325SJeff Bonwick  */
343e14bb325SJeff Bonwick static void
344770499e1SDan Kimmel zio_subblock(zio_t *zio, abd_t *data, uint64_t size)
345e14bb325SJeff Bonwick {
346e14bb325SJeff Bonwick 	ASSERT(zio->io_size > size);
347e14bb325SJeff Bonwick 
348e14bb325SJeff Bonwick 	if (zio->io_type == ZIO_TYPE_READ)
349770499e1SDan Kimmel 		abd_copy(data, zio->io_abd, size);
350e14bb325SJeff Bonwick }
351e14bb325SJeff Bonwick 
352e14bb325SJeff Bonwick static void
353770499e1SDan Kimmel zio_decompress(zio_t *zio, abd_t *data, uint64_t size)
354e14bb325SJeff Bonwick {
355770499e1SDan Kimmel 	if (zio->io_error == 0) {
356770499e1SDan Kimmel 		void *tmp = abd_borrow_buf(data, size);
357770499e1SDan Kimmel 		int ret = zio_decompress_data(BP_GET_COMPRESS(zio->io_bp),
358770499e1SDan Kimmel 		    zio->io_abd, tmp, zio->io_size, size);
359770499e1SDan Kimmel 		abd_return_buf_copy(data, tmp, size);
360770499e1SDan Kimmel 
361770499e1SDan Kimmel 		if (ret != 0)
362770499e1SDan Kimmel 			zio->io_error = SET_ERROR(EIO);
363770499e1SDan Kimmel 	}
364e14bb325SJeff Bonwick }
365e14bb325SJeff Bonwick 
366eb633035STom Caputi static void
367eb633035STom Caputi zio_decrypt(zio_t *zio, abd_t *data, uint64_t size)
368eb633035STom Caputi {
369eb633035STom Caputi 	int ret;
370eb633035STom Caputi 	void *tmp;
371eb633035STom Caputi 	blkptr_t *bp = zio->io_bp;
372eb633035STom Caputi 	spa_t *spa = zio->io_spa;
373eb633035STom Caputi 	uint64_t dsobj = zio->io_bookmark.zb_objset;
374eb633035STom Caputi 	uint64_t lsize = BP_GET_LSIZE(bp);
375eb633035STom Caputi 	dmu_object_type_t ot = BP_GET_TYPE(bp);
376eb633035STom Caputi 	uint8_t salt[ZIO_DATA_SALT_LEN];
377eb633035STom Caputi 	uint8_t iv[ZIO_DATA_IV_LEN];
378eb633035STom Caputi 	uint8_t mac[ZIO_DATA_MAC_LEN];
379eb633035STom Caputi 	boolean_t no_crypt = B_FALSE;
380eb633035STom Caputi 
381eb633035STom Caputi 	ASSERT(BP_USES_CRYPT(bp));
382eb633035STom Caputi 	ASSERT3U(size, !=, 0);
383eb633035STom Caputi 
384eb633035STom Caputi 	if (zio->io_error != 0)
385eb633035STom Caputi 		return;
386eb633035STom Caputi 
387eb633035STom Caputi 	/*
388eb633035STom Caputi 	 * Verify the cksum of MACs stored in an indirect bp. It will always
389eb633035STom Caputi 	 * be possible to verify this since it does not require an encryption
390eb633035STom Caputi 	 * key.
391eb633035STom Caputi 	 */
392eb633035STom Caputi 	if (BP_HAS_INDIRECT_MAC_CKSUM(bp)) {
393eb633035STom Caputi 		zio_crypt_decode_mac_bp(bp, mac);
394eb633035STom Caputi 
395eb633035STom Caputi 		if (BP_GET_COMPRESS(bp) != ZIO_COMPRESS_OFF) {
396eb633035STom Caputi 			/*
397eb633035STom Caputi 			 * We haven't decompressed the data yet, but
398eb633035STom Caputi 			 * zio_crypt_do_indirect_mac_checksum() requires
399eb633035STom Caputi 			 * decompressed data to be able to parse out the MACs
400eb633035STom Caputi 			 * from the indirect block. We decompress it now and
401eb633035STom Caputi 			 * throw away the result after we are finished.
402eb633035STom Caputi 			 */
403eb633035STom Caputi 			tmp = zio_buf_alloc(lsize);
404eb633035STom Caputi 			ret = zio_decompress_data(BP_GET_COMPRESS(bp),
405eb633035STom Caputi 			    zio->io_abd, tmp, zio->io_size, lsize);
406eb633035STom Caputi 			if (ret != 0) {
407eb633035STom Caputi 				ret = SET_ERROR(EIO);
408eb633035STom Caputi 				goto error;
409eb633035STom Caputi 			}
410eb633035STom Caputi 			ret = zio_crypt_do_indirect_mac_checksum(B_FALSE,
411eb633035STom Caputi 			    tmp, lsize, BP_SHOULD_BYTESWAP(bp), mac);
412eb633035STom Caputi 			zio_buf_free(tmp, lsize);
413eb633035STom Caputi 		} else {
414eb633035STom Caputi 			ret = zio_crypt_do_indirect_mac_checksum_abd(B_FALSE,
415eb633035STom Caputi 			    zio->io_abd, size, BP_SHOULD_BYTESWAP(bp), mac);
416eb633035STom Caputi 		}
417eb633035STom Caputi 		abd_copy(data, zio->io_abd, size);
418eb633035STom Caputi 
419eb633035STom Caputi 		if (ret != 0)
420eb633035STom Caputi 			goto error;
421eb633035STom Caputi 
422eb633035STom Caputi 		return;
423eb633035STom Caputi 	}
424eb633035STom Caputi 
425eb633035STom Caputi 	/*
426eb633035STom Caputi 	 * If this is an authenticated block, just check the MAC. It would be
427eb633035STom Caputi 	 * nice to separate this out into its own flag, but for the moment
428eb633035STom Caputi 	 * enum zio_flag is out of bits.
429eb633035STom Caputi 	 */
430eb633035STom Caputi 	if (BP_IS_AUTHENTICATED(bp)) {
431eb633035STom Caputi 		if (ot == DMU_OT_OBJSET) {
432eb633035STom Caputi 			ret = spa_do_crypt_objset_mac_abd(B_FALSE, spa,
433eb633035STom Caputi 			    dsobj, zio->io_abd, size, BP_SHOULD_BYTESWAP(bp));
434eb633035STom Caputi 		} else {
435eb633035STom Caputi 			zio_crypt_decode_mac_bp(bp, mac);
436eb633035STom Caputi 			ret = spa_do_crypt_mac_abd(B_FALSE, spa, dsobj,
437eb633035STom Caputi 			    zio->io_abd, size, mac);
438eb633035STom Caputi 		}
439eb633035STom Caputi 		abd_copy(data, zio->io_abd, size);
440eb633035STom Caputi 
441eb633035STom Caputi 		if (zio_injection_enabled && ot != DMU_OT_DNODE && ret == 0) {
442eb633035STom Caputi 			ret = zio_handle_decrypt_injection(spa,
443eb633035STom Caputi 			    &zio->io_bookmark, ot, ECKSUM);
444eb633035STom Caputi 		}
445eb633035STom Caputi 		if (ret != 0)
446eb633035STom Caputi 			goto error;
447eb633035STom Caputi 
448eb633035STom Caputi 		return;
449eb633035STom Caputi 	}
450eb633035STom Caputi 
451eb633035STom Caputi 	zio_crypt_decode_params_bp(bp, salt, iv);
452eb633035STom Caputi 
453eb633035STom Caputi 	if (ot == DMU_OT_INTENT_LOG) {
454eb633035STom Caputi 		tmp = abd_borrow_buf_copy(zio->io_abd, sizeof (zil_chain_t));
455eb633035STom Caputi 		zio_crypt_decode_mac_zil(tmp, mac);
456eb633035STom Caputi 		abd_return_buf(zio->io_abd, tmp, sizeof (zil_chain_t));
457eb633035STom Caputi 	} else {
458eb633035STom Caputi 		zio_crypt_decode_mac_bp(bp, mac);
459eb633035STom Caputi 	}
460eb633035STom Caputi 
461eb633035STom Caputi 	ret = spa_do_crypt_abd(B_FALSE, spa, &zio->io_bookmark, BP_GET_TYPE(bp),
462eb633035STom Caputi 	    BP_GET_DEDUP(bp), BP_SHOULD_BYTESWAP(bp), salt, iv, mac, size, data,
463eb633035STom Caputi 	    zio->io_abd, &no_crypt);
464eb633035STom Caputi 	if (no_crypt)
465eb633035STom Caputi 		abd_copy(data, zio->io_abd, size);
466eb633035STom Caputi 
467eb633035STom Caputi 	if (ret != 0)
468eb633035STom Caputi 		goto error;
469eb633035STom Caputi 
470eb633035STom Caputi 	return;
471eb633035STom Caputi 
472eb633035STom Caputi error:
473eb633035STom Caputi 	/* assert that the key was found unless this was speculative */
474eb633035STom Caputi 	ASSERT(ret != EACCES || (zio->io_flags & ZIO_FLAG_SPECULATIVE));
475eb633035STom Caputi 
476eb633035STom Caputi 	/*
477eb633035STom Caputi 	 * If there was a decryption / authentication error return EIO as
478eb633035STom Caputi 	 * the io_error. If this was not a speculative zio, create an ereport.
479eb633035STom Caputi 	 */
480eb633035STom Caputi 	if (ret == ECKSUM) {
481eb633035STom Caputi 		zio->io_error = SET_ERROR(EIO);
482eb633035STom Caputi 		if ((zio->io_flags & ZIO_FLAG_SPECULATIVE) == 0) {
483eb633035STom Caputi 			spa_log_error(spa, &zio->io_bookmark);
4849b088140SToomas Soome 			(void) zfs_ereport_post(FM_EREPORT_ZFS_AUTHENTICATION,
485eb633035STom Caputi 			    spa, NULL, &zio->io_bookmark, zio, 0, 0);
486eb633035STom Caputi 		}
487eb633035STom Caputi 	} else {
488eb633035STom Caputi 		zio->io_error = ret;
489eb633035STom Caputi 	}
490eb633035STom Caputi }
491eb633035STom Caputi 
492e14bb325SJeff Bonwick /*
493e14bb325SJeff Bonwick  * ==========================================================================
494e14bb325SJeff Bonwick  * I/O parent/child relationships and pipeline interlocks
495e14bb325SJeff Bonwick  * ==========================================================================
496e14bb325SJeff Bonwick  */
497a3f829aeSBill Moore zio_t *
4980f7643c7SGeorge Wilson zio_walk_parents(zio_t *cio, zio_link_t **zl)
499a3f829aeSBill Moore {
500a3f829aeSBill Moore 	list_t *pl = &cio->io_parent_list;
501e14bb325SJeff Bonwick 
5020f7643c7SGeorge Wilson 	*zl = (*zl == NULL) ? list_head(pl) : list_next(pl, *zl);
5030f7643c7SGeorge Wilson 	if (*zl == NULL)
504a3f829aeSBill Moore 		return (NULL);
505a3f829aeSBill Moore 
5060f7643c7SGeorge Wilson 	ASSERT((*zl)->zl_child == cio);
5070f7643c7SGeorge Wilson 	return ((*zl)->zl_parent);
508a3f829aeSBill Moore }
509a3f829aeSBill Moore 
510a3f829aeSBill Moore zio_t *
5110f7643c7SGeorge Wilson zio_walk_children(zio_t *pio, zio_link_t **zl)
512a3f829aeSBill Moore {
513a3f829aeSBill Moore 	list_t *cl = &pio->io_child_list;
514a3f829aeSBill Moore 
515a3874b8bSToomas Soome 	ASSERT(MUTEX_HELD(&pio->io_lock));
516a3874b8bSToomas Soome 
5170f7643c7SGeorge Wilson 	*zl = (*zl == NULL) ? list_head(cl) : list_next(cl, *zl);
5180f7643c7SGeorge Wilson 	if (*zl == NULL)
519a3f829aeSBill Moore 		return (NULL);
520a3f829aeSBill Moore 
5210f7643c7SGeorge Wilson 	ASSERT((*zl)->zl_parent == pio);
5220f7643c7SGeorge Wilson 	return ((*zl)->zl_child);
523a3f829aeSBill Moore }
524a3f829aeSBill Moore 
525a3f829aeSBill Moore zio_t *
526a3f829aeSBill Moore zio_unique_parent(zio_t *cio)
527a3f829aeSBill Moore {
5280f7643c7SGeorge Wilson 	zio_link_t *zl = NULL;
5290f7643c7SGeorge Wilson 	zio_t *pio = zio_walk_parents(cio, &zl);
530a3f829aeSBill Moore 
5310f7643c7SGeorge Wilson 	VERIFY3P(zio_walk_parents(cio, &zl), ==, NULL);
532a3f829aeSBill Moore 	return (pio);
533a3f829aeSBill Moore }
534a3f829aeSBill Moore 
535a3f829aeSBill Moore void
536a3f829aeSBill Moore zio_add_child(zio_t *pio, zio_t *cio)
537e14bb325SJeff Bonwick {
538a3f829aeSBill Moore 	zio_link_t *zl = kmem_cache_alloc(zio_link_cache, KM_SLEEP);
539a3f829aeSBill Moore 
540a3f829aeSBill Moore 	/*
541a3f829aeSBill Moore 	 * Logical I/Os can have logical, gang, or vdev children.
542a3f829aeSBill Moore 	 * Gang I/Os can have gang or vdev children.
543a3f829aeSBill Moore 	 * Vdev I/Os can only have vdev children.
544a3f829aeSBill Moore 	 * The following ASSERT captures all of these constraints.
545a3f829aeSBill Moore 	 */
5461271e4b1SPrakash Surya 	ASSERT3S(cio->io_child_type, <=, pio->io_child_type);
547a3f829aeSBill Moore 
548a3f829aeSBill Moore 	zl->zl_parent = pio;
549a3f829aeSBill Moore 	zl->zl_child = cio;
550a3f829aeSBill Moore 
551e14bb325SJeff Bonwick 	mutex_enter(&pio->io_lock);
552a3874b8bSToomas Soome 	mutex_enter(&cio->io_lock);
553a3f829aeSBill Moore 
554a3f829aeSBill Moore 	ASSERT(pio->io_state[ZIO_WAIT_DONE] == 0);
555a3f829aeSBill Moore 
556a3f829aeSBill Moore 	for (int w = 0; w < ZIO_WAIT_TYPES; w++)
557a3f829aeSBill Moore 		pio->io_children[cio->io_child_type][w] += !cio->io_state[w];
558a3f829aeSBill Moore 
559a3f829aeSBill Moore 	list_insert_head(&pio->io_child_list, zl);
560a3f829aeSBill Moore 	list_insert_head(&cio->io_parent_list, zl);
561a3f829aeSBill Moore 
562b24ab676SJeff Bonwick 	pio->io_child_count++;
563b24ab676SJeff Bonwick 	cio->io_parent_count++;
564b24ab676SJeff Bonwick 
565a3f829aeSBill Moore 	mutex_exit(&cio->io_lock);
566a3874b8bSToomas Soome 	mutex_exit(&pio->io_lock);
567e14bb325SJeff Bonwick }
568e14bb325SJeff Bonwick 
569fa9e4066Sahrens static void
570a3f829aeSBill Moore zio_remove_child(zio_t *pio, zio_t *cio, zio_link_t *zl)
571e14bb325SJeff Bonwick {
572a3f829aeSBill Moore 	ASSERT(zl->zl_parent == pio);
573a3f829aeSBill Moore 	ASSERT(zl->zl_child == cio);
574e14bb325SJeff Bonwick 
575e14bb325SJeff Bonwick 	mutex_enter(&pio->io_lock);
576a3874b8bSToomas Soome 	mutex_enter(&cio->io_lock);
577a3f829aeSBill Moore 
578a3f829aeSBill Moore 	list_remove(&pio->io_child_list, zl);
579a3f829aeSBill Moore 	list_remove(&cio->io_parent_list, zl);
580a3f829aeSBill Moore 
581b24ab676SJeff Bonwick 	pio->io_child_count--;
582b24ab676SJeff Bonwick 	cio->io_parent_count--;
583b24ab676SJeff Bonwick 
584a3f829aeSBill Moore 	mutex_exit(&cio->io_lock);
585a3874b8bSToomas Soome 	mutex_exit(&pio->io_lock);
586a3f829aeSBill Moore 
587a3f829aeSBill Moore 	kmem_cache_free(zio_link_cache, zl);
588e14bb325SJeff Bonwick }
589e14bb325SJeff Bonwick 
590e14bb325SJeff Bonwick static boolean_t
591d6e1c446SGeorge Wilson zio_wait_for_children(zio_t *zio, uint8_t childbits, enum zio_wait_type wait)
592fa9e4066Sahrens {
593e14bb325SJeff Bonwick 	boolean_t waiting = B_FALSE;
594e14bb325SJeff Bonwick 
595e14bb325SJeff Bonwick 	mutex_enter(&zio->io_lock);
596e14bb325SJeff Bonwick 	ASSERT(zio->io_stall == NULL);
597d6e1c446SGeorge Wilson 	for (int c = 0; c < ZIO_CHILD_TYPES; c++) {
598d6e1c446SGeorge Wilson 		if (!(ZIO_CHILD_BIT_IS_SET(childbits, c)))
599d6e1c446SGeorge Wilson 			continue;
600d6e1c446SGeorge Wilson 
601d6e1c446SGeorge Wilson 		uint64_t *countp = &zio->io_children[c][wait];
602d6e1c446SGeorge Wilson 		if (*countp != 0) {
603d6e1c446SGeorge Wilson 			zio->io_stage >>= 1;
604d6e1c446SGeorge Wilson 			ASSERT3U(zio->io_stage, !=, ZIO_STAGE_OPEN);
605d6e1c446SGeorge Wilson 			zio->io_stall = countp;
606d6e1c446SGeorge Wilson 			waiting = B_TRUE;
607d6e1c446SGeorge Wilson 			break;
608d6e1c446SGeorge Wilson 		}
609e14bb325SJeff Bonwick 	}
610e14bb325SJeff Bonwick 	mutex_exit(&zio->io_lock);
611e14bb325SJeff Bonwick 	return (waiting);
612e14bb325SJeff Bonwick }
613fa9e4066Sahrens 
614e14bb325SJeff Bonwick static void
615e14bb325SJeff Bonwick zio_notify_parent(zio_t *pio, zio_t *zio, enum zio_wait_type wait)
616e14bb325SJeff Bonwick {
617e14bb325SJeff Bonwick 	uint64_t *countp = &pio->io_children[zio->io_child_type][wait];
618e14bb325SJeff Bonwick 	int *errorp = &pio->io_child_error[zio->io_child_type];
619fa9e4066Sahrens 
620e14bb325SJeff Bonwick 	mutex_enter(&pio->io_lock);
621e14bb325SJeff Bonwick 	if (zio->io_error && !(zio->io_flags & ZIO_FLAG_DONT_PROPAGATE))
622e14bb325SJeff Bonwick 		*errorp = zio_worst_error(*errorp, zio->io_error);
623e14bb325SJeff Bonwick 	pio->io_reexecute |= zio->io_reexecute;
624e14bb325SJeff Bonwick 	ASSERT3U(*countp, >, 0);
62569962b56SMatthew Ahrens 
62669962b56SMatthew Ahrens 	(*countp)--;
62769962b56SMatthew Ahrens 
62869962b56SMatthew Ahrens 	if (*countp == 0 && pio->io_stall == countp) {
6290f7643c7SGeorge Wilson 		zio_taskq_type_t type =
6300f7643c7SGeorge Wilson 		    pio->io_stage < ZIO_STAGE_VDEV_IO_START ? ZIO_TASKQ_ISSUE :
6310f7643c7SGeorge Wilson 		    ZIO_TASKQ_INTERRUPT;
632e14bb325SJeff Bonwick 		pio->io_stall = NULL;
633e14bb325SJeff Bonwick 		mutex_exit(&pio->io_lock);
6340f7643c7SGeorge Wilson 		/*
6350f7643c7SGeorge Wilson 		 * Dispatch the parent zio in its own taskq so that
6360f7643c7SGeorge Wilson 		 * the child can continue to make progress. This also
6370f7643c7SGeorge Wilson 		 * prevents overflowing the stack when we have deeply nested
6380f7643c7SGeorge Wilson 		 * parent-child relationships.
6390f7643c7SGeorge Wilson 		 */
6400f7643c7SGeorge Wilson 		zio_taskq_dispatch(pio, type, B_FALSE);
641e14bb325SJeff Bonwick 	} else {
642e14bb325SJeff Bonwick 		mutex_exit(&pio->io_lock);
643fa9e4066Sahrens 	}
644fa9e4066Sahrens }
645fa9e4066Sahrens 
646e14bb325SJeff Bonwick static void
647e14bb325SJeff Bonwick zio_inherit_child_errors(zio_t *zio, enum zio_child c)
648e14bb325SJeff Bonwick {
649e14bb325SJeff Bonwick 	if (zio->io_child_error[c] != 0 && zio->io_error == 0)
650e14bb325SJeff Bonwick 		zio->io_error = zio->io_child_error[c];
651e14bb325SJeff Bonwick }
652e14bb325SJeff Bonwick 
6530f7643c7SGeorge Wilson int
65494c2d0ebSMatthew Ahrens zio_bookmark_compare(const void *x1, const void *x2)
6550f7643c7SGeorge Wilson {
6560f7643c7SGeorge Wilson 	const zio_t *z1 = x1;
6570f7643c7SGeorge Wilson 	const zio_t *z2 = x2;
6580f7643c7SGeorge Wilson 
65994c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_objset < z2->io_bookmark.zb_objset)
6600f7643c7SGeorge Wilson 		return (-1);
66194c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_objset > z2->io_bookmark.zb_objset)
6620f7643c7SGeorge Wilson 		return (1);
6630f7643c7SGeorge Wilson 
66494c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_object < z2->io_bookmark.zb_object)
6650f7643c7SGeorge Wilson 		return (-1);
66694c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_object > z2->io_bookmark.zb_object)
66794c2d0ebSMatthew Ahrens 		return (1);
66894c2d0ebSMatthew Ahrens 
66994c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_level < z2->io_bookmark.zb_level)
67094c2d0ebSMatthew Ahrens 		return (-1);
67194c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_level > z2->io_bookmark.zb_level)
67294c2d0ebSMatthew Ahrens 		return (1);
67394c2d0ebSMatthew Ahrens 
67494c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_blkid < z2->io_bookmark.zb_blkid)
67594c2d0ebSMatthew Ahrens 		return (-1);
67694c2d0ebSMatthew Ahrens 	if (z1->io_bookmark.zb_blkid > z2->io_bookmark.zb_blkid)
6770f7643c7SGeorge Wilson 		return (1);
6780f7643c7SGeorge Wilson 
6790f7643c7SGeorge Wilson 	if (z1 < z2)
6800f7643c7SGeorge Wilson 		return (-1);
6810f7643c7SGeorge Wilson 	if (z1 > z2)
6820f7643c7SGeorge Wilson 		return (1);
6830f7643c7SGeorge Wilson 
6840f7643c7SGeorge Wilson 	return (0);
6850f7643c7SGeorge Wilson }
6860f7643c7SGeorge Wilson 
687fa9e4066Sahrens /*
688fa9e4066Sahrens  * ==========================================================================
689e14bb325SJeff Bonwick  * Create the various types of I/O (read, write, free, etc)
690fa9e4066Sahrens  * ==========================================================================
691fa9e4066Sahrens  */
692fa9e4066Sahrens static zio_t *
693b24ab676SJeff Bonwick zio_create(zio_t *pio, spa_t *spa, uint64_t txg, const blkptr_t *bp,
694770499e1SDan Kimmel     abd_t *data, uint64_t lsize, uint64_t psize, zio_done_func_t *done,
6955602294fSDan Kimmel     void *private, zio_type_t type, zio_priority_t priority,
6965602294fSDan Kimmel     enum zio_flag flags, vdev_t *vd, uint64_t offset,
6975602294fSDan Kimmel     const zbookmark_phys_t *zb, enum zio_stage stage, enum zio_stage pipeline)
698fa9e4066Sahrens {
699fa9e4066Sahrens 	zio_t *zio;
700fa9e4066Sahrens 
701084fd14fSBrian Behlendorf 	IMPLY(type != ZIO_TYPE_TRIM, psize <= SPA_MAXBLOCKSIZE);
7025602294fSDan Kimmel 	ASSERT(P2PHASE(psize, SPA_MINBLOCKSIZE) == 0);
703e14bb325SJeff Bonwick 	ASSERT(P2PHASE(offset, SPA_MINBLOCKSIZE) == 0);
704fa9e4066Sahrens 
705e14bb325SJeff Bonwick 	ASSERT(!vd || spa_config_held(spa, SCL_STATE_ALL, RW_READER));
706e14bb325SJeff Bonwick 	ASSERT(!bp || !(flags & ZIO_FLAG_CONFIG_WRITER));
707e14bb325SJeff Bonwick 	ASSERT(vd || stage == ZIO_STAGE_OPEN);
708088f3894Sahrens 
709eb633035STom Caputi 	IMPLY(lsize != psize, (flags & ZIO_FLAG_RAW_COMPRESS) != 0);
7105602294fSDan Kimmel 
711ccae0b50Seschrock 	zio = kmem_cache_alloc(zio_cache, KM_SLEEP);
712ccae0b50Seschrock 	bzero(zio, sizeof (zio_t));
713e14bb325SJeff Bonwick 
714e14bb325SJeff Bonwick 	mutex_init(&zio->io_lock, NULL, MUTEX_DEFAULT, NULL);
715e14bb325SJeff Bonwick 	cv_init(&zio->io_cv, NULL, CV_DEFAULT, NULL);
716e14bb325SJeff Bonwick 
717a3f829aeSBill Moore 	list_create(&zio->io_parent_list, sizeof (zio_link_t),
718a3f829aeSBill Moore 	    offsetof(zio_link_t, zl_parent_node));
719a3f829aeSBill Moore 	list_create(&zio->io_child_list, sizeof (zio_link_t),
720a3f829aeSBill Moore 	    offsetof(zio_link_t, zl_child_node));
7218363e80aSGeorge Wilson 	metaslab_trace_init(&zio->io_alloc_list);
722a3f829aeSBill Moore 
723e14bb325SJeff Bonwick 	if (vd != NULL)
724e14bb325SJeff Bonwick 		zio->io_child_type = ZIO_CHILD_VDEV;
725e14bb325SJeff Bonwick 	else if (flags & ZIO_FLAG_GANG_CHILD)
726e14bb325SJeff Bonwick 		zio->io_child_type = ZIO_CHILD_GANG;
727b24ab676SJeff Bonwick 	else if (flags & ZIO_FLAG_DDT_CHILD)
728b24ab676SJeff Bonwick 		zio->io_child_type = ZIO_CHILD_DDT;
729e14bb325SJeff Bonwick 	else
730e14bb325SJeff Bonwick 		zio->io_child_type = ZIO_CHILD_LOGICAL;
731e14bb325SJeff Bonwick 
732fa9e4066Sahrens 	if (bp != NULL) {
733b24ab676SJeff Bonwick 		zio->io_bp = (blkptr_t *)bp;
734fa9e4066Sahrens 		zio->io_bp_copy = *bp;
735fa9e4066Sahrens 		zio->io_bp_orig = *bp;
736b24ab676SJeff Bonwick 		if (type != ZIO_TYPE_WRITE ||
737b24ab676SJeff Bonwick 		    zio->io_child_type == ZIO_CHILD_DDT)
738e14bb325SJeff Bonwick 			zio->io_bp = &zio->io_bp_copy;	/* so caller can free */
739f5383399SBill Moore 		if (zio->io_child_type == ZIO_CHILD_LOGICAL)
740e14bb325SJeff Bonwick 			zio->io_logical = zio;
741f5383399SBill Moore 		if (zio->io_child_type > ZIO_CHILD_GANG && BP_IS_GANG(bp))
742f5383399SBill Moore 			pipeline |= ZIO_GANG_STAGES;
743fa9e4066Sahrens 	}
744e14bb325SJeff Bonwick 
745e14bb325SJeff Bonwick 	zio->io_spa = spa;
746e14bb325SJeff Bonwick 	zio->io_txg = txg;
747fa9e4066Sahrens 	zio->io_done = done;
748fa9e4066Sahrens 	zio->io_private = private;
749fa9e4066Sahrens 	zio->io_type = type;
750fa9e4066Sahrens 	zio->io_priority = priority;
751e14bb325SJeff Bonwick 	zio->io_vd = vd;
752e14bb325SJeff Bonwick 	zio->io_offset = offset;
753770499e1SDan Kimmel 	zio->io_orig_abd = zio->io_abd = data;
7545602294fSDan Kimmel 	zio->io_orig_size = zio->io_size = psize;
7555602294fSDan Kimmel 	zio->io_lsize = lsize;
756e14bb325SJeff Bonwick 	zio->io_orig_flags = zio->io_flags = flags;
757e14bb325SJeff Bonwick 	zio->io_orig_stage = zio->io_stage = stage;
758e14bb325SJeff Bonwick 	zio->io_orig_pipeline = zio->io_pipeline = pipeline;
7590f7643c7SGeorge Wilson 	zio->io_pipeline_trace = ZIO_STAGE_OPEN;
760fa9e4066Sahrens 
761a3f829aeSBill Moore 	zio->io_state[ZIO_WAIT_READY] = (stage >= ZIO_STAGE_READY);
762a3f829aeSBill Moore 	zio->io_state[ZIO_WAIT_DONE] = (stage >= ZIO_STAGE_DONE);
763a3f829aeSBill Moore 
764e14bb325SJeff Bonwick 	if (zb != NULL)
765e14bb325SJeff Bonwick 		zio->io_bookmark = *zb;
766e14bb325SJeff Bonwick 
767e14bb325SJeff Bonwick 	if (pio != NULL) {
768663207adSDon Brady 		if (zio->io_metaslab_class == NULL)
769663207adSDon Brady 			zio->io_metaslab_class = pio->io_metaslab_class;
770e14bb325SJeff Bonwick 		if (zio->io_logical == NULL)
771ea8dc4b6Seschrock 			zio->io_logical = pio->io_logical;
772f5383399SBill Moore 		if (zio->io_child_type == ZIO_CHILD_GANG)
773f5383399SBill Moore 			zio->io_gang_leader = pio->io_gang_leader;
774e14bb325SJeff Bonwick 		zio_add_child(pio, zio);
775fa9e4066Sahrens 	}
776fa9e4066Sahrens 
777fa9e4066Sahrens 	return (zio);
778fa9e4066Sahrens }
779fa9e4066Sahrens 
7800a4e9518Sgw static void
781e14bb325SJeff Bonwick zio_destroy(zio_t *zio)
7820a4e9518Sgw {
7838363e80aSGeorge Wilson 	metaslab_trace_fini(&zio->io_alloc_list);
784a3f829aeSBill Moore 	list_destroy(&zio->io_parent_list);
785a3f829aeSBill Moore 	list_destroy(&zio->io_child_list);
786e14bb325SJeff Bonwick 	mutex_destroy(&zio->io_lock);
787e14bb325SJeff Bonwick 	cv_destroy(&zio->io_cv);
788e14bb325SJeff Bonwick 	kmem_cache_free(zio_cache, zio);
7890a4e9518Sgw }
7900a4e9518Sgw 
791fa9e4066Sahrens zio_t *
792a3f829aeSBill Moore zio_null(zio_t *pio, spa_t *spa, vdev_t *vd, zio_done_func_t *done,
793b24ab676SJeff Bonwick     void *private, enum zio_flag flags)
794fa9e4066Sahrens {
795fa9e4066Sahrens 	zio_t *zio;
796fa9e4066Sahrens 
7975602294fSDan Kimmel 	zio = zio_create(pio, spa, 0, NULL, NULL, 0, 0, done, private,
798a3f829aeSBill Moore 	    ZIO_TYPE_NULL, ZIO_PRIORITY_NOW, flags, vd, 0, NULL,
799e14bb325SJeff Bonwick 	    ZIO_STAGE_OPEN, ZIO_INTERLOCK_PIPELINE);
800fa9e4066Sahrens 
801fa9e4066Sahrens 	return (zio);
802fa9e4066Sahrens }
803fa9e4066Sahrens 
804fa9e4066Sahrens zio_t *
805b24ab676SJeff Bonwick zio_root(spa_t *spa, zio_done_func_t *done, void *private, enum zio_flag flags)
806fa9e4066Sahrens {
807a3f829aeSBill Moore 	return (zio_null(NULL, spa, NULL, done, private, flags));
808fa9e4066Sahrens }
809fa9e4066Sahrens 
810f63ab3d5SMatthew Ahrens void
811f63ab3d5SMatthew Ahrens zfs_blkptr_verify(spa_t *spa, const blkptr_t *bp)
812f63ab3d5SMatthew Ahrens {
813f63ab3d5SMatthew Ahrens 	if (!DMU_OT_IS_VALID(BP_GET_TYPE(bp))) {
814f63ab3d5SMatthew Ahrens 		zfs_panic_recover("blkptr at %p has invalid TYPE %llu",
815f63ab3d5SMatthew Ahrens 		    bp, (longlong_t)BP_GET_TYPE(bp));
816f63ab3d5SMatthew Ahrens 	}
817f63ab3d5SMatthew Ahrens 	if (BP_GET_CHECKSUM(bp) >= ZIO_CHECKSUM_FUNCTIONS ||
818f63ab3d5SMatthew Ahrens 	    BP_GET_CHECKSUM(bp) <= ZIO_CHECKSUM_ON) {
819f63ab3d5SMatthew Ahrens 		zfs_panic_recover("blkptr at %p has invalid CHECKSUM %llu",
820f63ab3d5SMatthew Ahrens 		    bp, (longlong_t)BP_GET_CHECKSUM(bp));
821f63ab3d5SMatthew Ahrens 	}
822f63ab3d5SMatthew Ahrens 	if (BP_GET_COMPRESS(bp) >= ZIO_COMPRESS_FUNCTIONS ||
823f63ab3d5SMatthew Ahrens 	    BP_GET_COMPRESS(bp) <= ZIO_COMPRESS_ON) {
824f63ab3d5SMatthew Ahrens 		zfs_panic_recover("blkptr at %p has invalid COMPRESS %llu",
825f63ab3d5SMatthew Ahrens 		    bp, (longlong_t)BP_GET_COMPRESS(bp));
826f63ab3d5SMatthew Ahrens 	}
827f63ab3d5SMatthew Ahrens 	if (BP_GET_LSIZE(bp) > SPA_MAXBLOCKSIZE) {
828f63ab3d5SMatthew Ahrens 		zfs_panic_recover("blkptr at %p has invalid LSIZE %llu",
829f63ab3d5SMatthew Ahrens 		    bp, (longlong_t)BP_GET_LSIZE(bp));
830f63ab3d5SMatthew Ahrens 	}
831f63ab3d5SMatthew Ahrens 	if (BP_GET_PSIZE(bp) > SPA_MAXBLOCKSIZE) {
832f63ab3d5SMatthew Ahrens 		zfs_panic_recover("blkptr at %p has invalid PSIZE %llu",
833f63ab3d5SMatthew Ahrens 		    bp, (longlong_t)BP_GET_PSIZE(bp));
834f63ab3d5SMatthew Ahrens 	}
835f63ab3d5SMatthew Ahrens 
836f63ab3d5SMatthew Ahrens 	if (BP_IS_EMBEDDED(bp)) {
837f63ab3d5SMatthew Ahrens 		if (BPE_GET_ETYPE(bp) > NUM_BP_EMBEDDED_TYPES) {
838f63ab3d5SMatthew Ahrens 			zfs_panic_recover("blkptr at %p has invalid ETYPE %llu",
839f63ab3d5SMatthew Ahrens 			    bp, (longlong_t)BPE_GET_ETYPE(bp));
840f63ab3d5SMatthew Ahrens 		}
841f63ab3d5SMatthew Ahrens 	}
842f63ab3d5SMatthew Ahrens 
8436f793812SPavel Zakharov 	/*
8446f793812SPavel Zakharov 	 * Do not verify individual DVAs if the config is not trusted. This
8456f793812SPavel Zakharov 	 * will be done once the zio is executed in vdev_mirror_map_alloc.
8466f793812SPavel Zakharov 	 */
8476f793812SPavel Zakharov 	if (!spa->spa_trust_config)
8486f793812SPavel Zakharov 		return;
8496f793812SPavel Zakharov 
850f63ab3d5SMatthew Ahrens 	/*
851f63ab3d5SMatthew Ahrens 	 * Pool-specific checks.
852f63ab3d5SMatthew Ahrens 	 *
853f63ab3d5SMatthew Ahrens 	 * Note: it would be nice to verify that the blk_birth and
854f63ab3d5SMatthew Ahrens 	 * BP_PHYSICAL_BIRTH() are not too large.  However, spa_freeze()
855f63ab3d5SMatthew Ahrens 	 * allows the birth time of log blocks (and dmu_sync()-ed blocks
856f63ab3d5SMatthew Ahrens 	 * that are in the log) to be arbitrarily large.
857f63ab3d5SMatthew Ahrens 	 */
858f63ab3d5SMatthew Ahrens 	for (int i = 0; i < BP_GET_NDVAS(bp); i++) {
859f63ab3d5SMatthew Ahrens 		uint64_t vdevid = DVA_GET_VDEV(&bp->blk_dva[i]);
860f63ab3d5SMatthew Ahrens 		if (vdevid >= spa->spa_root_vdev->vdev_children) {
861f63ab3d5SMatthew Ahrens 			zfs_panic_recover("blkptr at %p DVA %u has invalid "
862f63ab3d5SMatthew Ahrens 			    "VDEV %llu",
863f63ab3d5SMatthew Ahrens 			    bp, i, (longlong_t)vdevid);
8645897eb49SJustin Gibbs 			continue;
865f63ab3d5SMatthew Ahrens 		}
866f63ab3d5SMatthew Ahrens 		vdev_t *vd = spa->spa_root_vdev->vdev_child[vdevid];
867f63ab3d5SMatthew Ahrens 		if (vd == NULL) {
868f63ab3d5SMatthew Ahrens 			zfs_panic_recover("blkptr at %p DVA %u has invalid "
869f63ab3d5SMatthew Ahrens 			    "VDEV %llu",
870f63ab3d5SMatthew Ahrens 			    bp, i, (longlong_t)vdevid);
8715897eb49SJustin Gibbs 			continue;
872f63ab3d5SMatthew Ahrens 		}
873f63ab3d5SMatthew Ahrens 		if (vd->vdev_ops == &vdev_hole_ops) {
874f63ab3d5SMatthew Ahrens 			zfs_panic_recover("blkptr at %p DVA %u has hole "
875f63ab3d5SMatthew Ahrens 			    "VDEV %llu",
876f63ab3d5SMatthew Ahrens 			    bp, i, (longlong_t)vdevid);
8775897eb49SJustin Gibbs 			continue;
878f63ab3d5SMatthew Ahrens 		}
879f63ab3d5SMatthew Ahrens 		if (vd->vdev_ops == &vdev_missing_ops) {
880f63ab3d5SMatthew Ahrens 			/*
881f63ab3d5SMatthew Ahrens 			 * "missing" vdevs are valid during import, but we
882f63ab3d5SMatthew Ahrens 			 * don't have their detailed info (e.g. asize), so
883f63ab3d5SMatthew Ahrens 			 * we can't perform any more checks on them.
884f63ab3d5SMatthew Ahrens 			 */
885f63ab3d5SMatthew Ahrens 			continue;
886f63ab3d5SMatthew Ahrens 		}
887f63ab3d5SMatthew Ahrens 		uint64_t offset = DVA_GET_OFFSET(&bp->blk_dva[i]);
888f63ab3d5SMatthew Ahrens 		uint64_t asize = DVA_GET_ASIZE(&bp->blk_dva[i]);
889f63ab3d5SMatthew Ahrens 		if (BP_IS_GANG(bp))
890f63ab3d5SMatthew Ahrens 			asize = vdev_psize_to_asize(vd, SPA_GANGBLOCKSIZE);
891f63ab3d5SMatthew Ahrens 		if (offset + asize > vd->vdev_asize) {
892f63ab3d5SMatthew Ahrens 			zfs_panic_recover("blkptr at %p DVA %u has invalid "
893f63ab3d5SMatthew Ahrens 			    "OFFSET %llu",
894f63ab3d5SMatthew Ahrens 			    bp, i, (longlong_t)offset);
895f63ab3d5SMatthew Ahrens 		}
896f63ab3d5SMatthew Ahrens 	}
897f63ab3d5SMatthew Ahrens }
898f63ab3d5SMatthew Ahrens 
8996f793812SPavel Zakharov boolean_t
9006f793812SPavel Zakharov zfs_dva_valid(spa_t *spa, const dva_t *dva, const blkptr_t *bp)
9016f793812SPavel Zakharov {
9026f793812SPavel Zakharov 	uint64_t vdevid = DVA_GET_VDEV(dva);
9036f793812SPavel Zakharov 
9046f793812SPavel Zakharov 	if (vdevid >= spa->spa_root_vdev->vdev_children)
9056f793812SPavel Zakharov 		return (B_FALSE);
9066f793812SPavel Zakharov 
9076f793812SPavel Zakharov 	vdev_t *vd = spa->spa_root_vdev->vdev_child[vdevid];
9086f793812SPavel Zakharov 	if (vd == NULL)
9096f793812SPavel Zakharov 		return (B_FALSE);
9106f793812SPavel Zakharov 
9116f793812SPavel Zakharov 	if (vd->vdev_ops == &vdev_hole_ops)
9126f793812SPavel Zakharov 		return (B_FALSE);
9136f793812SPavel Zakharov 
9146f793812SPavel Zakharov 	if (vd->vdev_ops == &vdev_missing_ops) {
9156f793812SPavel Zakharov 		return (B_FALSE);
9166f793812SPavel Zakharov 	}
9176f793812SPavel Zakharov 
9186f793812SPavel Zakharov 	uint64_t offset = DVA_GET_OFFSET(dva);
9196f793812SPavel Zakharov 	uint64_t asize = DVA_GET_ASIZE(dva);
9206f793812SPavel Zakharov 
9216f793812SPavel Zakharov 	if (BP_IS_GANG(bp))
9226f793812SPavel Zakharov 		asize = vdev_psize_to_asize(vd, SPA_GANGBLOCKSIZE);
9236f793812SPavel Zakharov 	if (offset + asize > vd->vdev_asize)
9246f793812SPavel Zakharov 		return (B_FALSE);
9256f793812SPavel Zakharov 
9266f793812SPavel Zakharov 	return (B_TRUE);
9276f793812SPavel Zakharov }
9286f793812SPavel Zakharov 
929fa9e4066Sahrens zio_t *
930e14bb325SJeff Bonwick zio_read(zio_t *pio, spa_t *spa, const blkptr_t *bp,
931770499e1SDan Kimmel     abd_t *data, uint64_t size, zio_done_func_t *done, void *private,
9327802d7bfSMatthew Ahrens     zio_priority_t priority, enum zio_flag flags, const zbookmark_phys_t *zb)
933fa9e4066Sahrens {
934fa9e4066Sahrens 	zio_t *zio;
935fa9e4066Sahrens 
936f63ab3d5SMatthew Ahrens 	zfs_blkptr_verify(spa, bp);
937f63ab3d5SMatthew Ahrens 
938b24ab676SJeff Bonwick 	zio = zio_create(pio, spa, BP_PHYSICAL_BIRTH(bp), bp,
9395602294fSDan Kimmel 	    data, size, size, done, private,
940e14bb325SJeff Bonwick 	    ZIO_TYPE_READ, priority, flags, NULL, 0, zb,
941b24ab676SJeff Bonwick 	    ZIO_STAGE_OPEN, (flags & ZIO_FLAG_DDT_CHILD) ?
942b24ab676SJeff Bonwick 	    ZIO_DDT_CHILD_READ_PIPELINE : ZIO_READ_PIPELINE);
943fa9e4066Sahrens 
944fa9e4066Sahrens 	return (zio);
945fa9e4066Sahrens }
946fa9e4066Sahrens 
947fa9e4066Sahrens zio_t *
948e14bb325SJeff Bonwick zio_write(zio_t *pio, spa_t *spa, uint64_t txg, blkptr_t *bp,
949770499e1SDan Kimmel     abd_t *data, uint64_t lsize, uint64_t psize, const zio_prop_t *zp,
9508df0bcf0SPaul Dagnelie     zio_done_func_t *ready, zio_done_func_t *children_ready,
9518df0bcf0SPaul Dagnelie     zio_done_func_t *physdone, zio_done_func_t *done,
9528df0bcf0SPaul Dagnelie     void *private, zio_priority_t priority, enum zio_flag flags,
9538df0bcf0SPaul Dagnelie     const zbookmark_phys_t *zb)
954fa9e4066Sahrens {
955fa9e4066Sahrens 	zio_t *zio;
956fa9e4066Sahrens 
957e14bb325SJeff Bonwick 	ASSERT(zp->zp_checksum >= ZIO_CHECKSUM_OFF &&
958e14bb325SJeff Bonwick 	    zp->zp_checksum < ZIO_CHECKSUM_FUNCTIONS &&
959e14bb325SJeff Bonwick 	    zp->zp_compress >= ZIO_COMPRESS_OFF &&
960e14bb325SJeff Bonwick 	    zp->zp_compress < ZIO_COMPRESS_FUNCTIONS &&
961ad135b5dSChristopher Siden 	    DMU_OT_IS_VALID(zp->zp_type) &&
962e14bb325SJeff Bonwick 	    zp->zp_level < 32 &&
963b24ab676SJeff Bonwick 	    zp->zp_copies > 0 &&
96480901aeaSGeorge Wilson 	    zp->zp_copies <= spa_max_replication(spa));
9650a4e9518Sgw 
9665602294fSDan Kimmel 	zio = zio_create(pio, spa, txg, bp, data, lsize, psize, done, private,
967e14bb325SJeff Bonwick 	    ZIO_TYPE_WRITE, priority, flags, NULL, 0, zb,
968b24ab676SJeff Bonwick 	    ZIO_STAGE_OPEN, (flags & ZIO_FLAG_DDT_CHILD) ?
969b24ab676SJeff Bonwick 	    ZIO_DDT_CHILD_WRITE_PIPELINE : ZIO_WRITE_PIPELINE);
970fa9e4066Sahrens 
971c717a561Smaybee 	zio->io_ready = ready;
9728df0bcf0SPaul Dagnelie 	zio->io_children_ready = children_ready;
97369962b56SMatthew Ahrens 	zio->io_physdone = physdone;
974e14bb325SJeff Bonwick 	zio->io_prop = *zp;
975fa9e4066Sahrens 
9765d7b4d43SMatthew Ahrens 	/*
9775d7b4d43SMatthew Ahrens 	 * Data can be NULL if we are going to call zio_write_override() to
9785d7b4d43SMatthew Ahrens 	 * provide the already-allocated BP.  But we may need the data to
9795d7b4d43SMatthew Ahrens 	 * verify a dedup hit (if requested).  In this case, don't try to
980eb633035STom Caputi 	 * dedup (just take the already-allocated BP verbatim). Encrypted
981eb633035STom Caputi 	 * dedup blocks need data as well so we also disable dedup in this
982eb633035STom Caputi 	 * case.
9835d7b4d43SMatthew Ahrens 	 */
984eb633035STom Caputi 	if (data == NULL &&
985eb633035STom Caputi 	    (zio->io_prop.zp_dedup_verify || zio->io_prop.zp_encrypt)) {
9865d7b4d43SMatthew Ahrens 		zio->io_prop.zp_dedup = zio->io_prop.zp_dedup_verify = B_FALSE;
9875d7b4d43SMatthew Ahrens 	}
9885d7b4d43SMatthew Ahrens 
989fa9e4066Sahrens 	return (zio);
990fa9e4066Sahrens }
991fa9e4066Sahrens 
992fa9e4066Sahrens zio_t *
993770499e1SDan Kimmel zio_rewrite(zio_t *pio, spa_t *spa, uint64_t txg, blkptr_t *bp, abd_t *data,
99469962b56SMatthew Ahrens     uint64_t size, zio_done_func_t *done, void *private,
9957802d7bfSMatthew Ahrens     zio_priority_t priority, enum zio_flag flags, zbookmark_phys_t *zb)
996fa9e4066Sahrens {
997fa9e4066Sahrens 	zio_t *zio;
998fa9e4066Sahrens 
9995602294fSDan Kimmel 	zio = zio_create(pio, spa, txg, bp, data, size, size, done, private,
10000f7643c7SGeorge Wilson 	    ZIO_TYPE_WRITE, priority, flags | ZIO_FLAG_IO_REWRITE, NULL, 0, zb,
1001e14bb325SJeff Bonwick 	    ZIO_STAGE_OPEN, ZIO_REWRITE_PIPELINE);
1002fa9e4066Sahrens 
1003fa9e4066Sahrens 	return (zio);
1004fa9e4066Sahrens }
1005fa9e4066Sahrens 
1006b24ab676SJeff Bonwick void
100780901aeaSGeorge Wilson zio_write_override(zio_t *zio, blkptr_t *bp, int copies, boolean_t nopwrite)
1008b24ab676SJeff Bonwick {
1009b24ab676SJeff Bonwick 	ASSERT(zio->io_type == ZIO_TYPE_WRITE);
1010b24ab676SJeff Bonwick 	ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
1011b24ab676SJeff Bonwick 	ASSERT(zio->io_stage == ZIO_STAGE_OPEN);
1012b24ab676SJeff Bonwick 	ASSERT(zio->io_txg == spa_syncing_txg(zio->io_spa));
1013b24ab676SJeff Bonwick 
101480901aeaSGeorge Wilson 	/*
101580901aeaSGeorge Wilson 	 * We must reset the io_prop to match the values that existed
101680901aeaSGeorge Wilson 	 * when the bp was first written by dmu_sync() keeping in mind
101780901aeaSGeorge Wilson 	 * that nopwrite and dedup are mutually exclusive.
101880901aeaSGeorge Wilson 	 */
101980901aeaSGeorge Wilson 	zio->io_prop.zp_dedup = nopwrite ? B_FALSE : zio->io_prop.zp_dedup;
102080901aeaSGeorge Wilson 	zio->io_prop.zp_nopwrite = nopwrite;
1021b24ab676SJeff Bonwick 	zio->io_prop.zp_copies = copies;
1022b24ab676SJeff Bonwick 	zio->io_bp_override = bp;
1023b24ab676SJeff Bonwick }
1024b24ab676SJeff Bonwick 
1025b24ab676SJeff Bonwick void
1026b24ab676SJeff Bonwick zio_free(spa_t *spa, uint64_t txg, const blkptr_t *bp)
1027b24ab676SJeff Bonwick {
10285d7b4d43SMatthew Ahrens 
10295cabbc6bSPrashanth Sreenivasa 	zfs_blkptr_verify(spa, bp);
10305cabbc6bSPrashanth Sreenivasa 
10315d7b4d43SMatthew Ahrens 	/*
10325d7b4d43SMatthew Ahrens 	 * The check for EMBEDDED is a performance optimization.  We
10335d7b4d43SMatthew Ahrens 	 * process the free here (by ignoring it) rather than
10345d7b4d43SMatthew Ahrens 	 * putting it on the list and then processing it in zio_free_sync().
10355d7b4d43SMatthew Ahrens 	 */
10365d7b4d43SMatthew Ahrens 	if (BP_IS_EMBEDDED(bp))
10375d7b4d43SMatthew Ahrens 		return;
10383b2aab18SMatthew Ahrens 	metaslab_check_free(spa, bp);
10399cb154a3SMatthew Ahrens 
10409cb154a3SMatthew Ahrens 	/*
10419cb154a3SMatthew Ahrens 	 * Frees that are for the currently-syncing txg, are not going to be
10429cb154a3SMatthew Ahrens 	 * deferred, and which will not need to do a read (i.e. not GANG or
10439cb154a3SMatthew Ahrens 	 * DEDUP), can be processed immediately.  Otherwise, put them on the
10449cb154a3SMatthew Ahrens 	 * in-memory list for later processing.
1045814dcd43SSerapheim Dimitropoulos 	 *
1046814dcd43SSerapheim Dimitropoulos 	 * Note that we only defer frees after zfs_sync_pass_deferred_free
1047814dcd43SSerapheim Dimitropoulos 	 * when the log space map feature is disabled. [see relevant comment
1048814dcd43SSerapheim Dimitropoulos 	 * in spa_sync_iterate_to_convergence()]
10499cb154a3SMatthew Ahrens 	 */
1050814dcd43SSerapheim Dimitropoulos 	if (BP_IS_GANG(bp) ||
1051814dcd43SSerapheim Dimitropoulos 	    BP_GET_DEDUP(bp) ||
10529cb154a3SMatthew Ahrens 	    txg != spa->spa_syncing_txg ||
1053814dcd43SSerapheim Dimitropoulos 	    (spa_sync_pass(spa) >= zfs_sync_pass_deferred_free &&
1054814dcd43SSerapheim Dimitropoulos 	    !spa_feature_is_active(spa, SPA_FEATURE_LOG_SPACEMAP))) {
10559cb154a3SMatthew Ahrens 		bplist_append(&spa->spa_free_bplist[txg & TXG_MASK], bp);
10569cb154a3SMatthew Ahrens 	} else {
10579cb154a3SMatthew Ahrens 		VERIFY0(zio_wait(zio_free_sync(NULL, spa, txg, bp, 0)));
10589cb154a3SMatthew Ahrens 	}
1059b24ab676SJeff Bonwick }
1060b24ab676SJeff Bonwick 
1061fa9e4066Sahrens zio_t *
1062b24ab676SJeff Bonwick zio_free_sync(zio_t *pio, spa_t *spa, uint64_t txg, const blkptr_t *bp,
1063b24ab676SJeff Bonwick     enum zio_flag flags)
1064fa9e4066Sahrens {
1065fa9e4066Sahrens 	zio_t *zio;
10669cb154a3SMatthew Ahrens 	enum zio_stage stage = ZIO_FREE_PIPELINE;
1067fa9e4066Sahrens 
1068fa9e4066Sahrens 	ASSERT(!BP_IS_HOLE(bp));
1069b24ab676SJeff Bonwick 	ASSERT(spa_syncing_txg(spa) == txg);
1070fa9e4066Sahrens 
10715d7b4d43SMatthew Ahrens 	if (BP_IS_EMBEDDED(bp))
10725d7b4d43SMatthew Ahrens 		return (zio_null(pio, spa, NULL, NULL, NULL, 0));
10735d7b4d43SMatthew Ahrens 
10743b2aab18SMatthew Ahrens 	metaslab_check_free(spa, bp);
10756e6d5868SMatthew Ahrens 	arc_freed(spa, bp);
1076a3874b8bSToomas Soome 	dsl_scan_freed(spa, bp);
10773b2aab18SMatthew Ahrens 
10789cb154a3SMatthew Ahrens 	/*
10799cb154a3SMatthew Ahrens 	 * GANG and DEDUP blocks can induce a read (for the gang block header,
10809cb154a3SMatthew Ahrens 	 * or the DDT), so issue them asynchronously so that this thread is
10819cb154a3SMatthew Ahrens 	 * not tied up.
10829cb154a3SMatthew Ahrens 	 */
10839cb154a3SMatthew Ahrens 	if (BP_IS_GANG(bp) || BP_GET_DEDUP(bp))
10849cb154a3SMatthew Ahrens 		stage |= ZIO_STAGE_ISSUE_ASYNC;
10859cb154a3SMatthew Ahrens 
1086e14bb325SJeff Bonwick 	zio = zio_create(pio, spa, txg, bp, NULL, BP_GET_PSIZE(bp),
10875602294fSDan Kimmel 	    BP_GET_PSIZE(bp), NULL, NULL, ZIO_TYPE_FREE, ZIO_PRIORITY_NOW,
10885602294fSDan Kimmel 	    flags, NULL, 0, NULL, ZIO_STAGE_OPEN, stage);
10899cb154a3SMatthew Ahrens 
1090fa9e4066Sahrens 	return (zio);
1091fa9e4066Sahrens }
1092fa9e4066Sahrens 
1093fa9e4066Sahrens zio_t *
1094b24ab676SJeff Bonwick zio_claim(zio_t *pio, spa_t *spa, uint64_t txg, const blkptr_t *bp,
1095b24ab676SJeff Bonwick     zio_done_func_t *done, void *private, enum zio_flag flags)
1096fa9e4066Sahrens {
1097fa9e4066Sahrens 	zio_t *zio;
1098fa9e4066Sahrens 
10995cabbc6bSPrashanth Sreenivasa 	zfs_blkptr_verify(spa, bp);
11005d7b4d43SMatthew Ahrens 
11015d7b4d43SMatthew Ahrens 	if (BP_IS_EMBEDDED(bp))
11025d7b4d43SMatthew Ahrens 		return (zio_null(pio, spa, NULL, NULL, NULL, 0));
11035d7b4d43SMatthew Ahrens 
1104fa9e4066Sahrens 	/*
1105fa9e4066Sahrens 	 * A claim is an allocation of a specific block.  Claims are needed
1106fa9e4066Sahrens 	 * to support immediate writes in the intent log.  The issue is that
1107fa9e4066Sahrens 	 * immediate writes contain committed data, but in a txg that was
1108fa9e4066Sahrens 	 * *not* committed.  Upon opening the pool after an unclean shutdown,
1109fa9e4066Sahrens 	 * the intent log claims all blocks that contain immediate write data
1110fa9e4066Sahrens 	 * so that the SPA knows they're in use.
1111fa9e4066Sahrens 	 *
1112fa9e4066Sahrens 	 * All claims *must* be resolved in the first txg -- before the SPA
1113fa9e4066Sahrens 	 * starts allocating blocks -- so that nothing is allocated twice.
1114b24ab676SJeff Bonwick 	 * If txg == 0 we just verify that the block is claimable.
1115fa9e4066Sahrens 	 */
111686714001SSerapheim Dimitropoulos 	ASSERT3U(spa->spa_uberblock.ub_rootbp.blk_birth, <,
111786714001SSerapheim Dimitropoulos 	    spa_min_claim_txg(spa));
111886714001SSerapheim Dimitropoulos 	ASSERT(txg == spa_min_claim_txg(spa) || txg == 0);
1119b24ab676SJeff Bonwick 	ASSERT(!BP_GET_DEDUP(bp) || !spa_writeable(spa));	/* zdb(1M) */
1120fa9e4066Sahrens 
1121e14bb325SJeff Bonwick 	zio = zio_create(pio, spa, txg, bp, NULL, BP_GET_PSIZE(bp),
11225602294fSDan Kimmel 	    BP_GET_PSIZE(bp), done, private, ZIO_TYPE_CLAIM, ZIO_PRIORITY_NOW,
11235602294fSDan Kimmel 	    flags, NULL, 0, NULL, ZIO_STAGE_OPEN, ZIO_CLAIM_PIPELINE);
11240f7643c7SGeorge Wilson 	ASSERT0(zio->io_queued_timestamp);
1125fa9e4066Sahrens 
1126fa9e4066Sahrens 	return (zio);
1127fa9e4066Sahrens }
1128fa9e4066Sahrens 
1129fa9e4066Sahrens zio_t *
1130fa9e4066Sahrens zio_ioctl(zio_t *pio, spa_t *spa, vdev_t *vd, int cmd,
113169962b56SMatthew Ahrens     zio_done_func_t *done, void *private, enum zio_flag flags)
1132fa9e4066Sahrens {
1133fa9e4066Sahrens 	zio_t *zio;
1134fa9e4066Sahrens 	int c;
1135fa9e4066Sahrens 
1136fa9e4066Sahrens 	if (vd->vdev_children == 0) {
11375602294fSDan Kimmel 		zio = zio_create(pio, spa, 0, NULL, NULL, 0, 0, done, private,
113869962b56SMatthew Ahrens 		    ZIO_TYPE_IOCTL, ZIO_PRIORITY_NOW, flags, vd, 0, NULL,
1139fa9e4066Sahrens 		    ZIO_STAGE_OPEN, ZIO_IOCTL_PIPELINE);
1140fa9e4066Sahrens 
1141fa9e4066Sahrens 		zio->io_cmd = cmd;
1142fa9e4066Sahrens 	} else {
1143a3f829aeSBill Moore 		zio = zio_null(pio, spa, NULL, NULL, NULL, flags);
1144fa9e4066Sahrens 
1145fa9e4066Sahrens 		for (c = 0; c < vd->vdev_children; c++)
1146fa9e4066Sahrens 			zio_nowait(zio_ioctl(zio, spa, vd->vdev_child[c], cmd,
114769962b56SMatthew Ahrens 			    done, private, flags));
1148fa9e4066Sahrens 	}
1149fa9e4066Sahrens 
1150fa9e4066Sahrens 	return (zio);
1151fa9e4066Sahrens }
1152fa9e4066Sahrens 
1153084fd14fSBrian Behlendorf zio_t *
1154084fd14fSBrian Behlendorf zio_trim(zio_t *pio, vdev_t *vd, uint64_t offset, uint64_t size,
1155084fd14fSBrian Behlendorf     zio_done_func_t *done, void *private, zio_priority_t priority,
1156084fd14fSBrian Behlendorf     enum zio_flag flags, enum trim_flag trim_flags)
1157084fd14fSBrian Behlendorf {
1158084fd14fSBrian Behlendorf 	zio_t *zio;
1159084fd14fSBrian Behlendorf 
1160084fd14fSBrian Behlendorf 	ASSERT0(vd->vdev_children);
1161084fd14fSBrian Behlendorf 	ASSERT0(P2PHASE(offset, 1ULL << vd->vdev_ashift));
1162084fd14fSBrian Behlendorf 	ASSERT0(P2PHASE(size, 1ULL << vd->vdev_ashift));
1163084fd14fSBrian Behlendorf 	ASSERT3U(size, !=, 0);
1164084fd14fSBrian Behlendorf 
1165084fd14fSBrian Behlendorf 	zio = zio_create(pio, vd->vdev_spa, 0, NULL, NULL, size, size, done,
1166084fd14fSBrian Behlendorf 	    private, ZIO_TYPE_TRIM, priority, flags | ZIO_FLAG_PHYSICAL,
1167084fd14fSBrian Behlendorf 	    vd, offset, NULL, ZIO_STAGE_OPEN, ZIO_TRIM_PIPELINE);
1168084fd14fSBrian Behlendorf 	zio->io_trim_flags = trim_flags;
1169084fd14fSBrian Behlendorf 
1170084fd14fSBrian Behlendorf 	return (zio);
1171084fd14fSBrian Behlendorf }
1172084fd14fSBrian Behlendorf 
1173fa9e4066Sahrens zio_t *
1174fa9e4066Sahrens zio_read_phys(zio_t *pio, vdev_t *vd, uint64_t offset, uint64_t size,
1175770499e1SDan Kimmel     abd_t *data, int checksum, zio_done_func_t *done, void *private,
117669962b56SMatthew Ahrens     zio_priority_t priority, enum zio_flag flags, boolean_t labels)
1177fa9e4066Sahrens {
1178fa9e4066Sahrens 	zio_t *zio;
11790a4e9518Sgw 
1180e14bb325SJeff Bonwick 	ASSERT(vd->vdev_children == 0);
1181e14bb325SJeff Bonwick 	ASSERT(!labels || offset + size <= VDEV_LABEL_START_SIZE ||
1182e14bb325SJeff Bonwick 	    offset >= vd->vdev_psize - VDEV_LABEL_END_SIZE);
1183e14bb325SJeff Bonwick 	ASSERT3U(offset + size, <=, vd->vdev_psize);
1184fa9e4066Sahrens 
11855602294fSDan Kimmel 	zio = zio_create(pio, vd->vdev_spa, 0, NULL, data, size, size, done,
11865602294fSDan Kimmel 	    private, ZIO_TYPE_READ, priority, flags | ZIO_FLAG_PHYSICAL, vd,
11875602294fSDan Kimmel 	    offset, NULL, ZIO_STAGE_OPEN, ZIO_READ_PHYS_PIPELINE);
1188fa9e4066Sahrens 
1189e14bb325SJeff Bonwick 	zio->io_prop.zp_checksum = checksum;
1190fa9e4066Sahrens 
1191fa9e4066Sahrens 	return (zio);
1192fa9e4066Sahrens }
1193fa9e4066Sahrens 
1194fa9e4066Sahrens zio_t *
1195fa9e4066Sahrens zio_write_phys(zio_t *pio, vdev_t *vd, uint64_t offset, uint64_t size,
1196770499e1SDan Kimmel     abd_t *data, int checksum, zio_done_func_t *done, void *private,
119769962b56SMatthew Ahrens     zio_priority_t priority, enum zio_flag flags, boolean_t labels)
1198fa9e4066Sahrens {
1199fa9e4066Sahrens 	zio_t *zio;
12000a4e9518Sgw 
1201e14bb325SJeff Bonwick 	ASSERT(vd->vdev_children == 0);
1202e14bb325SJeff Bonwick 	ASSERT(!labels || offset + size <= VDEV_LABEL_START_SIZE ||
1203e14bb325SJeff Bonwick 	    offset >= vd->vdev_psize - VDEV_LABEL_END_SIZE);
1204e14bb325SJeff Bonwick 	ASSERT3U(offset + size, <=, vd->vdev_psize);
1205fa9e4066Sahrens 
12065602294fSDan Kimmel 	zio = zio_create(pio, vd->vdev_spa, 0, NULL, data, size, size, done,
12075602294fSDan Kimmel 	    private, ZIO_TYPE_WRITE, priority, flags | ZIO_FLAG_PHYSICAL, vd,
12085602294fSDan Kimmel 	    offset, NULL, ZIO_STAGE_OPEN, ZIO_WRITE_PHYS_PIPELINE);
1209fa9e4066Sahrens 
1210e14bb325SJeff Bonwick 	zio->io_prop.zp_checksum = checksum;
1211fa9e4066Sahrens 
121245818ee1SMatthew Ahrens 	if (zio_checksum_table[checksum].ci_flags & ZCHECKSUM_FLAG_EMBEDDED) {
1213fa9e4066Sahrens 		/*
12146e1f5caaSNeil Perrin 		 * zec checksums are necessarily destructive -- they modify
1215e14bb325SJeff Bonwick 		 * the end of the write buffer to hold the verifier/checksum.
1216fa9e4066Sahrens 		 * Therefore, we must make a local copy in case the data is
1217e14bb325SJeff Bonwick 		 * being written to multiple places in parallel.
1218fa9e4066Sahrens 		 */
1219770499e1SDan Kimmel 		abd_t *wbuf = abd_alloc_sametype(data, size);
1220770499e1SDan Kimmel 		abd_copy(wbuf, data, size);
1221770499e1SDan Kimmel 
1222e14bb325SJeff Bonwick 		zio_push_transform(zio, wbuf, size, size, NULL);
1223fa9e4066Sahrens 	}
1224fa9e4066Sahrens 
1225fa9e4066Sahrens 	return (zio);
1226fa9e4066Sahrens }
1227fa9e4066Sahrens 
1228fa9e4066Sahrens /*
1229e14bb325SJeff Bonwick  * Create a child I/O to do some work for us.
1230fa9e4066Sahrens  */
1231fa9e4066Sahrens zio_t *
1232e14bb325SJeff Bonwick zio_vdev_child_io(zio_t *pio, blkptr_t *bp, vdev_t *vd, uint64_t offset,
1233770499e1SDan Kimmel     abd_t *data, uint64_t size, int type, zio_priority_t priority,
1234dcbf3bd6SGeorge Wilson     enum zio_flag flags, zio_done_func_t *done, void *private)
1235fa9e4066Sahrens {
1236b24ab676SJeff Bonwick 	enum zio_stage pipeline = ZIO_VDEV_CHILD_PIPELINE;
1237e14bb325SJeff Bonwick 	zio_t *zio;
1238e14bb325SJeff Bonwick 
12395cabbc6bSPrashanth Sreenivasa 	/*
12405cabbc6bSPrashanth Sreenivasa 	 * vdev child I/Os do not propagate their error to the parent.
12415cabbc6bSPrashanth Sreenivasa 	 * Therefore, for correct operation the caller *must* check for
12425cabbc6bSPrashanth Sreenivasa 	 * and handle the error in the child i/o's done callback.
12435cabbc6bSPrashanth Sreenivasa 	 * The only exceptions are i/os that we don't care about
12445cabbc6bSPrashanth Sreenivasa 	 * (OPTIONAL or REPAIR).
12455cabbc6bSPrashanth Sreenivasa 	 */
12465cabbc6bSPrashanth Sreenivasa 	ASSERT((flags & ZIO_FLAG_OPTIONAL) || (flags & ZIO_FLAG_IO_REPAIR) ||
12475cabbc6bSPrashanth Sreenivasa 	    done != NULL);
12485cabbc6bSPrashanth Sreenivasa 
1249fa9e4066Sahrens 	if (type == ZIO_TYPE_READ && bp != NULL) {
1250fa9e4066Sahrens 		/*
1251fa9e4066Sahrens 		 * If we have the bp, then the child should perform the
1252fa9e4066Sahrens 		 * checksum and the parent need not.  This pushes error
1253fa9e4066Sahrens 		 * detection as close to the leaves as possible and
1254fa9e4066Sahrens 		 * eliminates redundant checksums in the interior nodes.
1255fa9e4066Sahrens 		 */
1256b24ab676SJeff Bonwick 		pipeline |= ZIO_STAGE_CHECKSUM_VERIFY;
1257b24ab676SJeff Bonwick 		pio->io_pipeline &= ~ZIO_STAGE_CHECKSUM_VERIFY;
1258fa9e4066Sahrens 	}
1259fa9e4066Sahrens 
12605cabbc6bSPrashanth Sreenivasa 	if (vd->vdev_ops->vdev_op_leaf) {
12615cabbc6bSPrashanth Sreenivasa 		ASSERT0(vd->vdev_children);
1262e14bb325SJeff Bonwick 		offset += VDEV_LABEL_START_SIZE;
12635cabbc6bSPrashanth Sreenivasa 	}
1264e14bb325SJeff Bonwick 
12655cabbc6bSPrashanth Sreenivasa 	flags |= ZIO_VDEV_CHILD_FLAGS(pio);
1266b24ab676SJeff Bonwick 
1267b24ab676SJeff Bonwick 	/*
1268b24ab676SJeff Bonwick 	 * If we've decided to do a repair, the write is not speculative --
1269b24ab676SJeff Bonwick 	 * even if the original read was.
1270b24ab676SJeff Bonwick 	 */
1271b24ab676SJeff Bonwick 	if (flags & ZIO_FLAG_IO_REPAIR)
1272b24ab676SJeff Bonwick 		flags &= ~ZIO_FLAG_SPECULATIVE;
1273b24ab676SJeff Bonwick 
12740f7643c7SGeorge Wilson 	/*
12750f7643c7SGeorge Wilson 	 * If we're creating a child I/O that is not associated with a
12760f7643c7SGeorge Wilson 	 * top-level vdev, then the child zio is not an allocating I/O.
12770f7643c7SGeorge Wilson 	 * If this is a retried I/O then we ignore it since we will
12780f7643c7SGeorge Wilson 	 * have already processed the original allocating I/O.
12790f7643c7SGeorge Wilson 	 */
12800f7643c7SGeorge Wilson 	if (flags & ZIO_FLAG_IO_ALLOCATING &&
12810f7643c7SGeorge Wilson 	    (vd != vd->vdev_top || (flags & ZIO_FLAG_IO_RETRY))) {
1282663207adSDon Brady 		ASSERT(pio->io_metaslab_class != NULL);
1283663207adSDon Brady 		ASSERT(pio->io_metaslab_class->mc_alloc_throttle_enabled);
12840f7643c7SGeorge Wilson 		ASSERT(type == ZIO_TYPE_WRITE);
12850f7643c7SGeorge Wilson 		ASSERT(priority == ZIO_PRIORITY_ASYNC_WRITE);
12860f7643c7SGeorge Wilson 		ASSERT(!(flags & ZIO_FLAG_IO_REPAIR));
12870f7643c7SGeorge Wilson 		ASSERT(!(pio->io_flags & ZIO_FLAG_IO_REWRITE) ||
12880f7643c7SGeorge Wilson 		    pio->io_child_type == ZIO_CHILD_GANG);
12890f7643c7SGeorge Wilson 
12900f7643c7SGeorge Wilson 		flags &= ~ZIO_FLAG_IO_ALLOCATING;
12910f7643c7SGeorge Wilson 	}
12920f7643c7SGeorge Wilson 
12935602294fSDan Kimmel 	zio = zio_create(pio, pio->io_spa, pio->io_txg, bp, data, size, size,
1294b24ab676SJeff Bonwick 	    done, private, type, priority, flags, vd, offset, &pio->io_bookmark,
1295b24ab676SJeff Bonwick 	    ZIO_STAGE_VDEV_IO_START >> 1, pipeline);
12960f7643c7SGeorge Wilson 	ASSERT3U(zio->io_child_type, ==, ZIO_CHILD_VDEV);
1297fa9e4066Sahrens 
129869962b56SMatthew Ahrens 	zio->io_physdone = pio->io_physdone;
129969962b56SMatthew Ahrens 	if (vd->vdev_ops->vdev_op_leaf && zio->io_logical != NULL)
130069962b56SMatthew Ahrens 		zio->io_logical->io_phys_children++;
130169962b56SMatthew Ahrens 
1302e14bb325SJeff Bonwick 	return (zio);
130332b87932Sek }
130432b87932Sek 
1305e14bb325SJeff Bonwick zio_t *
1306770499e1SDan Kimmel zio_vdev_delegated_io(vdev_t *vd, uint64_t offset, abd_t *data, uint64_t size,
13073a4b1be9SMatthew Ahrens     zio_type_t type, zio_priority_t priority, enum zio_flag flags,
13089a686fbcSPaul Dagnelie     zio_done_func_t *done, void *private)
1309fa9e4066Sahrens {
1310e14bb325SJeff Bonwick 	zio_t *zio;
1311fa9e4066Sahrens 
1312e14bb325SJeff Bonwick 	ASSERT(vd->vdev_ops->vdev_op_leaf);
1313fa9e4066Sahrens 
1314e14bb325SJeff Bonwick 	zio = zio_create(NULL, vd->vdev_spa, 0, NULL,
13155602294fSDan Kimmel 	    data, size, size, done, private, type, priority,
131669962b56SMatthew Ahrens 	    flags | ZIO_FLAG_CANFAIL | ZIO_FLAG_DONT_RETRY | ZIO_FLAG_DELEGATED,
1317e14bb325SJeff Bonwick 	    vd, offset, NULL,
1318b24ab676SJeff Bonwick 	    ZIO_STAGE_VDEV_IO_START >> 1, ZIO_VDEV_CHILD_PIPELINE);
1319fa9e4066Sahrens 
1320e14bb325SJeff Bonwick 	return (zio);
1321e05725b1Sbonwick }
1322e05725b1Sbonwick 
1323e05725b1Sbonwick void
1324e14bb325SJeff Bonwick zio_flush(zio_t *zio, vdev_t *vd)
1325e05725b1Sbonwick {
1326e14bb325SJeff Bonwick 	zio_nowait(zio_ioctl(zio, zio->io_spa, vd, DKIOCFLUSHWRITECACHE,
132769962b56SMatthew Ahrens 	    NULL, NULL,
1328e14bb325SJeff Bonwick 	    ZIO_FLAG_CANFAIL | ZIO_FLAG_DONT_PROPAGATE | ZIO_FLAG_DONT_RETRY));
1329fa9e4066Sahrens }
1330fa9e4066Sahrens 
13316e1f5caaSNeil Perrin void
13326e1f5caaSNeil Perrin zio_shrink(zio_t *zio, uint64_t size)
13336e1f5caaSNeil Perrin {
13341271e4b1SPrakash Surya 	ASSERT3P(zio->io_executor, ==, NULL);
13351271e4b1SPrakash Surya 	ASSERT3P(zio->io_orig_size, ==, zio->io_size);
13361271e4b1SPrakash Surya 	ASSERT3U(size, <=, zio->io_size);
13376e1f5caaSNeil Perrin 
13386e1f5caaSNeil Perrin 	/*
13396e1f5caaSNeil Perrin 	 * We don't shrink for raidz because of problems with the
13406e1f5caaSNeil Perrin 	 * reconstruction when reading back less than the block size.
13416e1f5caaSNeil Perrin 	 * Note, BP_IS_RAIDZ() assumes no compression.
13426e1f5caaSNeil Perrin 	 */
13436e1f5caaSNeil Perrin 	ASSERT(BP_GET_COMPRESS(zio->io_bp) == ZIO_COMPRESS_OFF);
13445602294fSDan Kimmel 	if (!BP_IS_RAIDZ(zio->io_bp)) {
13455602294fSDan Kimmel 		/* we are not doing a raw write */
13465602294fSDan Kimmel 		ASSERT3U(zio->io_size, ==, zio->io_lsize);
13475602294fSDan Kimmel 		zio->io_orig_size = zio->io_size = zio->io_lsize = size;
13485602294fSDan Kimmel 	}
13496e1f5caaSNeil Perrin }
13506e1f5caaSNeil Perrin 
1351fa9e4066Sahrens /*
1352fa9e4066Sahrens  * ==========================================================================
1353e14bb325SJeff Bonwick  * Prepare to read and write logical blocks
1354fa9e4066Sahrens  * ==========================================================================
1355fa9e4066Sahrens  */
1356e14bb325SJeff Bonwick 
1357e05725b1Sbonwick static int
1358e14bb325SJeff Bonwick zio_read_bp_init(zio_t *zio)
1359fa9e4066Sahrens {
1360e14bb325SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
1361eb633035STom Caputi 	uint64_t psize =
1362eb633035STom Caputi 	    BP_IS_EMBEDDED(bp) ? BPE_GET_PSIZE(bp) : BP_GET_PSIZE(bp);
1363e05725b1Sbonwick 
13645cabbc6bSPrashanth Sreenivasa 	ASSERT3P(zio->io_bp, ==, &zio->io_bp_copy);
13655cabbc6bSPrashanth Sreenivasa 
136603361682SJeff Bonwick 	if (BP_GET_COMPRESS(bp) != ZIO_COMPRESS_OFF &&
1367f5383399SBill Moore 	    zio->io_child_type == ZIO_CHILD_LOGICAL &&
1368eb633035STom Caputi 	    !(zio->io_flags & ZIO_FLAG_RAW_COMPRESS)) {
1369770499e1SDan Kimmel 		zio_push_transform(zio, abd_alloc_sametype(zio->io_abd, psize),
1370770499e1SDan Kimmel 		    psize, psize, zio_decompress);
1371e14bb325SJeff Bonwick 	}
1372fa9e4066Sahrens 
1373eb633035STom Caputi 	if (((BP_IS_PROTECTED(bp) && !(zio->io_flags & ZIO_FLAG_RAW_ENCRYPT)) ||
1374eb633035STom Caputi 	    BP_HAS_INDIRECT_MAC_CKSUM(bp)) &&
1375eb633035STom Caputi 	    zio->io_child_type == ZIO_CHILD_LOGICAL) {
1376eb633035STom Caputi 		zio_push_transform(zio, abd_alloc_sametype(zio->io_abd, psize),
1377eb633035STom Caputi 		    psize, psize, zio_decrypt);
1378eb633035STom Caputi 	}
1379770499e1SDan Kimmel 
1380eb633035STom Caputi 	if (BP_IS_EMBEDDED(bp) && BPE_GET_ETYPE(bp) == BP_EMBEDDED_TYPE_DATA) {
1381770499e1SDan Kimmel 		int psize = BPE_GET_PSIZE(bp);
1382770499e1SDan Kimmel 		void *data = abd_borrow_buf(zio->io_abd, psize);
1383eb633035STom Caputi 
1384eb633035STom Caputi 		zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
1385770499e1SDan Kimmel 		decode_embedded_bp_compressed(bp, data);
1386770499e1SDan Kimmel 		abd_return_buf_copy(zio->io_abd, data, psize);
13875d7b4d43SMatthew Ahrens 	} else {
13885d7b4d43SMatthew Ahrens 		ASSERT(!BP_IS_EMBEDDED(bp));
13895cabbc6bSPrashanth Sreenivasa 		ASSERT3P(zio->io_bp, ==, &zio->io_bp_copy);
13905d7b4d43SMatthew Ahrens 	}
13915d7b4d43SMatthew Ahrens 
1392ad135b5dSChristopher Siden 	if (!DMU_OT_IS_METADATA(BP_GET_TYPE(bp)) && BP_GET_LEVEL(bp) == 0)
1393e14bb325SJeff Bonwick 		zio->io_flags |= ZIO_FLAG_DONT_CACHE;
1394fa9e4066Sahrens 
1395bbfd46c4SJeff Bonwick 	if (BP_GET_TYPE(bp) == DMU_OT_DDT_ZAP)
1396bbfd46c4SJeff Bonwick 		zio->io_flags |= ZIO_FLAG_DONT_CACHE;
1397bbfd46c4SJeff Bonwick 
1398b24ab676SJeff Bonwick 	if (BP_GET_DEDUP(bp) && zio->io_child_type == ZIO_CHILD_LOGICAL)
1399b24ab676SJeff Bonwick 		zio->io_pipeline = ZIO_DDT_READ_PIPELINE;
1400b24ab676SJeff Bonwick 
1401e14bb325SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
1402fa9e4066Sahrens }
1403fa9e4066Sahrens 
1404e05725b1Sbonwick static int
1405e14bb325SJeff Bonwick zio_write_bp_init(zio_t *zio)
14060a4e9518Sgw {
1407e14bb325SJeff Bonwick 	if (!IO_IS_ALLOCATING(zio))
1408e14bb325SJeff Bonwick 		return (ZIO_PIPELINE_CONTINUE);
14090a4e9518Sgw 
1410b24ab676SJeff Bonwick 	ASSERT(zio->io_child_type != ZIO_CHILD_DDT);
1411b24ab676SJeff Bonwick 
1412b24ab676SJeff Bonwick 	if (zio->io_bp_override) {
14130f7643c7SGeorge Wilson 		blkptr_t *bp = zio->io_bp;
14140f7643c7SGeorge Wilson 		zio_prop_t *zp = &zio->io_prop;
14150f7643c7SGeorge Wilson 
1416b24ab676SJeff Bonwick 		ASSERT(bp->blk_birth != zio->io_txg);
1417b24ab676SJeff Bonwick 		ASSERT(BP_GET_DEDUP(zio->io_bp_override) == 0);
1418b24ab676SJeff Bonwick 
1419b24ab676SJeff Bonwick 		*bp = *zio->io_bp_override;
1420b24ab676SJeff Bonwick 		zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
1421b24ab676SJeff Bonwick 
14225d7b4d43SMatthew Ahrens 		if (BP_IS_EMBEDDED(bp))
14235d7b4d43SMatthew Ahrens 			return (ZIO_PIPELINE_CONTINUE);
14245d7b4d43SMatthew Ahrens 
142580901aeaSGeorge Wilson 		/*
142680901aeaSGeorge Wilson 		 * If we've been overridden and nopwrite is set then
142780901aeaSGeorge Wilson 		 * set the flag accordingly to indicate that a nopwrite
142880901aeaSGeorge Wilson 		 * has already occurred.
142980901aeaSGeorge Wilson 		 */
143080901aeaSGeorge Wilson 		if (!BP_IS_HOLE(bp) && zp->zp_nopwrite) {
143180901aeaSGeorge Wilson 			ASSERT(!zp->zp_dedup);
14320f7643c7SGeorge Wilson 			ASSERT3U(BP_GET_CHECKSUM(bp), ==, zp->zp_checksum);
143380901aeaSGeorge Wilson 			zio->io_flags |= ZIO_FLAG_NOPWRITE;
143480901aeaSGeorge Wilson 			return (ZIO_PIPELINE_CONTINUE);
143580901aeaSGeorge Wilson 		}
143680901aeaSGeorge Wilson 
143780901aeaSGeorge Wilson 		ASSERT(!zp->zp_nopwrite);
143880901aeaSGeorge Wilson 
1439b24ab676SJeff Bonwick 		if (BP_IS_HOLE(bp) || !zp->zp_dedup)
1440b24ab676SJeff Bonwick 			return (ZIO_PIPELINE_CONTINUE);
1441b24ab676SJeff Bonwick 
144245818ee1SMatthew Ahrens 		ASSERT((zio_checksum_table[zp->zp_checksum].ci_flags &
144345818ee1SMatthew Ahrens 		    ZCHECKSUM_FLAG_DEDUP) || zp->zp_dedup_verify);
1444b24ab676SJeff Bonwick 
1445eb633035STom Caputi 		if (BP_GET_CHECKSUM(bp) == zp->zp_checksum &&
1446eb633035STom Caputi 		    !zp->zp_encrypt) {
1447b24ab676SJeff Bonwick 			BP_SET_DEDUP(bp, 1);
1448b24ab676SJeff Bonwick 			zio->io_pipeline |= ZIO_STAGE_DDT_WRITE;
1449b24ab676SJeff Bonwick 			return (ZIO_PIPELINE_CONTINUE);
1450b24ab676SJeff Bonwick 		}
14510f7643c7SGeorge Wilson 
14520f7643c7SGeorge Wilson 		/*
14530f7643c7SGeorge Wilson 		 * We were unable to handle this as an override bp, treat
14540f7643c7SGeorge Wilson 		 * it as a regular write I/O.
14550f7643c7SGeorge Wilson 		 */
1456b39b744bSMatthew Ahrens 		zio->io_bp_override = NULL;
14570f7643c7SGeorge Wilson 		*bp = zio->io_bp_orig;
14580f7643c7SGeorge Wilson 		zio->io_pipeline = zio->io_orig_pipeline;
1459b24ab676SJeff Bonwick 	}
14600a4e9518Sgw 
14610f7643c7SGeorge Wilson 	return (ZIO_PIPELINE_CONTINUE);
14620f7643c7SGeorge Wilson }
14630f7643c7SGeorge Wilson 
14640f7643c7SGeorge Wilson static int
14650f7643c7SGeorge Wilson zio_write_compress(zio_t *zio)
14660f7643c7SGeorge Wilson {
14670f7643c7SGeorge Wilson 	spa_t *spa = zio->io_spa;
14680f7643c7SGeorge Wilson 	zio_prop_t *zp = &zio->io_prop;
14690f7643c7SGeorge Wilson 	enum zio_compress compress = zp->zp_compress;
14700f7643c7SGeorge Wilson 	blkptr_t *bp = zio->io_bp;
14715602294fSDan Kimmel 	uint64_t lsize = zio->io_lsize;
14725602294fSDan Kimmel 	uint64_t psize = zio->io_size;
14730f7643c7SGeorge Wilson 	int pass = 1;
14740f7643c7SGeorge Wilson 
14750f7643c7SGeorge Wilson 	/*
14760f7643c7SGeorge Wilson 	 * If our children haven't all reached the ready stage,
14770f7643c7SGeorge Wilson 	 * wait for them and then repeat this pipeline stage.
14780f7643c7SGeorge Wilson 	 */
1479d6e1c446SGeorge Wilson 	if (zio_wait_for_children(zio, ZIO_CHILD_LOGICAL_BIT |
1480d6e1c446SGeorge Wilson 	    ZIO_CHILD_GANG_BIT, ZIO_WAIT_READY)) {
14810f7643c7SGeorge Wilson 		return (ZIO_PIPELINE_STOP);
1482d6e1c446SGeorge Wilson 	}
14830f7643c7SGeorge Wilson 
14840f7643c7SGeorge Wilson 	if (!IO_IS_ALLOCATING(zio))
14850f7643c7SGeorge Wilson 		return (ZIO_PIPELINE_CONTINUE);
14860f7643c7SGeorge Wilson 
14870f7643c7SGeorge Wilson 	if (zio->io_children_ready != NULL) {
14880f7643c7SGeorge Wilson 		/*
14890f7643c7SGeorge Wilson 		 * Now that all our children are ready, run the callback
14900f7643c7SGeorge Wilson 		 * associated with this zio in case it wants to modify the
14910f7643c7SGeorge Wilson 		 * data to be written.
14920f7643c7SGeorge Wilson 		 */
14930f7643c7SGeorge Wilson 		ASSERT3U(zp->zp_level, >, 0);
14940f7643c7SGeorge Wilson 		zio->io_children_ready(zio);
14950f7643c7SGeorge Wilson 	}
14960f7643c7SGeorge Wilson 
14970f7643c7SGeorge Wilson 	ASSERT(zio->io_child_type != ZIO_CHILD_DDT);
14980f7643c7SGeorge Wilson 	ASSERT(zio->io_bp_override == NULL);
14990f7643c7SGeorge Wilson 
150043466aaeSMax Grossman 	if (!BP_IS_HOLE(bp) && bp->blk_birth == zio->io_txg) {
1501e14bb325SJeff Bonwick 		/*
1502e14bb325SJeff Bonwick 		 * We're rewriting an existing block, which means we're
1503e14bb325SJeff Bonwick 		 * working on behalf of spa_sync().  For spa_sync() to
1504e14bb325SJeff Bonwick 		 * converge, it must eventually be the case that we don't
1505e14bb325SJeff Bonwick 		 * have to allocate new blocks.  But compression changes
1506e14bb325SJeff Bonwick 		 * the blocksize, which forces a reallocate, and makes
1507e14bb325SJeff Bonwick 		 * convergence take longer.  Therefore, after the first
1508e14bb325SJeff Bonwick 		 * few passes, stop compressing to ensure convergence.
1509e14bb325SJeff Bonwick 		 */
1510b24ab676SJeff Bonwick 		pass = spa_sync_pass(spa);
1511b24ab676SJeff Bonwick 
1512b24ab676SJeff Bonwick 		ASSERT(zio->io_txg == spa_syncing_txg(spa));
1513b24ab676SJeff Bonwick 		ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
1514b24ab676SJeff Bonwick 		ASSERT(!BP_GET_DEDUP(bp));
1515e05725b1Sbonwick 
151601f55e48SGeorge Wilson 		if (pass >= zfs_sync_pass_dont_compress)
1517e14bb325SJeff Bonwick 			compress = ZIO_COMPRESS_OFF;
1518e05725b1Sbonwick 
1519e14bb325SJeff Bonwick 		/* Make sure someone doesn't change their mind on overwrites */
15205d7b4d43SMatthew Ahrens 		ASSERT(BP_IS_EMBEDDED(bp) || MIN(zp->zp_copies + BP_IS_GANG(bp),
1521b24ab676SJeff Bonwick 		    spa_max_replication(spa)) == BP_GET_NDVAS(bp));
1522e14bb325SJeff Bonwick 	}
1523fa9e4066Sahrens 
15245602294fSDan Kimmel 	/* If it's a compressed write that is not raw, compress the buffer. */
1525eb633035STom Caputi 	if (compress != ZIO_COMPRESS_OFF &&
1526eb633035STom Caputi 	    !(zio->io_flags & ZIO_FLAG_RAW_COMPRESS)) {
1527b24ab676SJeff Bonwick 		void *cbuf = zio_buf_alloc(lsize);
1528770499e1SDan Kimmel 		psize = zio_compress_data(compress, zio->io_abd, cbuf, lsize);
1529b24ab676SJeff Bonwick 		if (psize == 0 || psize == lsize) {
1530e14bb325SJeff Bonwick 			compress = ZIO_COMPRESS_OFF;
1531b24ab676SJeff Bonwick 			zio_buf_free(cbuf, lsize);
1532eb633035STom Caputi 		} else if (!zp->zp_dedup && !zp->zp_encrypt &&
1533eb633035STom Caputi 		    psize <= BPE_PAYLOAD_SIZE &&
15345d7b4d43SMatthew Ahrens 		    zp->zp_level == 0 && !DMU_OT_HAS_FILL(zp->zp_type) &&
15355d7b4d43SMatthew Ahrens 		    spa_feature_is_enabled(spa, SPA_FEATURE_EMBEDDED_DATA)) {
15365d7b4d43SMatthew Ahrens 			encode_embedded_bp_compressed(bp,
15375d7b4d43SMatthew Ahrens 			    cbuf, compress, lsize, psize);
15385d7b4d43SMatthew Ahrens 			BPE_SET_ETYPE(bp, BP_EMBEDDED_TYPE_DATA);
15395d7b4d43SMatthew Ahrens 			BP_SET_TYPE(bp, zio->io_prop.zp_type);
15405d7b4d43SMatthew Ahrens 			BP_SET_LEVEL(bp, zio->io_prop.zp_level);
15415d7b4d43SMatthew Ahrens 			zio_buf_free(cbuf, lsize);
15425d7b4d43SMatthew Ahrens 			bp->blk_birth = zio->io_txg;
15435d7b4d43SMatthew Ahrens 			zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
15445d7b4d43SMatthew Ahrens 			ASSERT(spa_feature_is_active(spa,
15455d7b4d43SMatthew Ahrens 			    SPA_FEATURE_EMBEDDED_DATA));
15465d7b4d43SMatthew Ahrens 			return (ZIO_PIPELINE_CONTINUE);
1547b24ab676SJeff Bonwick 		} else {
15485d7b4d43SMatthew Ahrens 			/*
154981cd5c55SMatthew Ahrens 			 * Round up compressed size up to the ashift
155081cd5c55SMatthew Ahrens 			 * of the smallest-ashift device, and zero the tail.
155181cd5c55SMatthew Ahrens 			 * This ensures that the compressed size of the BP
155281cd5c55SMatthew Ahrens 			 * (and thus compressratio property) are correct,
155381cd5c55SMatthew Ahrens 			 * in that we charge for the padding used to fill out
155481cd5c55SMatthew Ahrens 			 * the last sector.
15555d7b4d43SMatthew Ahrens 			 */
155681cd5c55SMatthew Ahrens 			ASSERT3U(spa->spa_min_ashift, >=, SPA_MINBLOCKSHIFT);
155781cd5c55SMatthew Ahrens 			size_t rounded = (size_t)P2ROUNDUP(psize,
155881cd5c55SMatthew Ahrens 			    1ULL << spa->spa_min_ashift);
155981cd5c55SMatthew Ahrens 			if (rounded >= lsize) {
15605d7b4d43SMatthew Ahrens 				compress = ZIO_COMPRESS_OFF;
15615d7b4d43SMatthew Ahrens 				zio_buf_free(cbuf, lsize);
156281cd5c55SMatthew Ahrens 				psize = lsize;
15635d7b4d43SMatthew Ahrens 			} else {
1564770499e1SDan Kimmel 				abd_t *cdata = abd_get_from_buf(cbuf, lsize);
1565770499e1SDan Kimmel 				abd_take_ownership_of_buf(cdata, B_TRUE);
1566770499e1SDan Kimmel 				abd_zero_off(cdata, psize, rounded - psize);
156781cd5c55SMatthew Ahrens 				psize = rounded;
1568770499e1SDan Kimmel 				zio_push_transform(zio, cdata,
15695d7b4d43SMatthew Ahrens 				    psize, lsize, NULL);
15705d7b4d43SMatthew Ahrens 			}
1571e14bb325SJeff Bonwick 		}
15720f7643c7SGeorge Wilson 
15730f7643c7SGeorge Wilson 		/*
15740f7643c7SGeorge Wilson 		 * We were unable to handle this as an override bp, treat
15750f7643c7SGeorge Wilson 		 * it as a regular write I/O.
15760f7643c7SGeorge Wilson 		 */
15770f7643c7SGeorge Wilson 		zio->io_bp_override = NULL;
15780f7643c7SGeorge Wilson 		*bp = zio->io_bp_orig;
15790f7643c7SGeorge Wilson 		zio->io_pipeline = zio->io_orig_pipeline;
1580eb633035STom Caputi 
1581eb633035STom Caputi 	} else if ((zio->io_flags & ZIO_FLAG_RAW_ENCRYPT) != 0 &&
1582eb633035STom Caputi 	    zp->zp_type == DMU_OT_DNODE) {
1583eb633035STom Caputi 		/*
1584eb633035STom Caputi 		 * The DMU actually relies on the zio layer's compression
1585eb633035STom Caputi 		 * to free metadnode blocks that have had all contained
1586eb633035STom Caputi 		 * dnodes freed. As a result, even when doing a raw
1587eb633035STom Caputi 		 * receive, we must check whether the block can be compressed
1588eb633035STom Caputi 		 * to a hole.
1589eb633035STom Caputi 		 */
1590eb633035STom Caputi 		psize = zio_compress_data(ZIO_COMPRESS_EMPTY,
1591eb633035STom Caputi 		    zio->io_abd, NULL, lsize);
1592eb633035STom Caputi 		if (psize == 0)
1593eb633035STom Caputi 			compress = ZIO_COMPRESS_OFF;
15945602294fSDan Kimmel 	} else {
15955602294fSDan Kimmel 		ASSERT3U(psize, !=, 0);
1596e14bb325SJeff Bonwick 	}
1597c717a561Smaybee 
1598e14bb325SJeff Bonwick 	/*
1599e14bb325SJeff Bonwick 	 * The final pass of spa_sync() must be all rewrites, but the first
1600e14bb325SJeff Bonwick 	 * few passes offer a trade-off: allocating blocks defers convergence,
1601e14bb325SJeff Bonwick 	 * but newly allocated blocks are sequential, so they can be written
1602e14bb325SJeff Bonwick 	 * to disk faster.  Therefore, we allow the first few passes of
1603e14bb325SJeff Bonwick 	 * spa_sync() to allocate new blocks, but force rewrites after that.
1604e14bb325SJeff Bonwick 	 * There should only be a handful of blocks after pass 1 in any case.
1605e14bb325SJeff Bonwick 	 */
160643466aaeSMax Grossman 	if (!BP_IS_HOLE(bp) && bp->blk_birth == zio->io_txg &&
160743466aaeSMax Grossman 	    BP_GET_PSIZE(bp) == psize &&
160801f55e48SGeorge Wilson 	    pass >= zfs_sync_pass_rewrite) {
1609663207adSDon Brady 		VERIFY3U(psize, !=, 0);
1610b24ab676SJeff Bonwick 		enum zio_stage gang_stages = zio->io_pipeline & ZIO_GANG_STAGES;
1611e14bb325SJeff Bonwick 		zio->io_pipeline = ZIO_REWRITE_PIPELINE | gang_stages;
1612e14bb325SJeff Bonwick 		zio->io_flags |= ZIO_FLAG_IO_REWRITE;
1613e14bb325SJeff Bonwick 	} else {
1614e14bb325SJeff Bonwick 		BP_ZERO(bp);
1615e14bb325SJeff Bonwick 		zio->io_pipeline = ZIO_WRITE_PIPELINE;
1616e14bb325SJeff Bonwick 	}
1617fa9e4066Sahrens 
1618b24ab676SJeff Bonwick 	if (psize == 0) {
161943466aaeSMax Grossman 		if (zio->io_bp_orig.blk_birth != 0 &&
162043466aaeSMax Grossman 		    spa_feature_is_active(spa, SPA_FEATURE_HOLE_BIRTH)) {
162143466aaeSMax Grossman 			BP_SET_LSIZE(bp, lsize);
162243466aaeSMax Grossman 			BP_SET_TYPE(bp, zp->zp_type);
162343466aaeSMax Grossman 			BP_SET_LEVEL(bp, zp->zp_level);
162443466aaeSMax Grossman 			BP_SET_BIRTH(bp, zio->io_txg, 0);
162543466aaeSMax Grossman 		}
1626e14bb325SJeff Bonwick 		zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
1627e14bb325SJeff Bonwick 	} else {
1628e14bb325SJeff Bonwick 		ASSERT(zp->zp_checksum != ZIO_CHECKSUM_GANG_HEADER);
1629e14bb325SJeff Bonwick 		BP_SET_LSIZE(bp, lsize);
163043466aaeSMax Grossman 		BP_SET_TYPE(bp, zp->zp_type);
163143466aaeSMax Grossman 		BP_SET_LEVEL(bp, zp->zp_level);
1632b24ab676SJeff Bonwick 		BP_SET_PSIZE(bp, psize);
1633e14bb325SJeff Bonwick 		BP_SET_COMPRESS(bp, compress);
1634e14bb325SJeff Bonwick 		BP_SET_CHECKSUM(bp, zp->zp_checksum);
1635b24ab676SJeff Bonwick 		BP_SET_DEDUP(bp, zp->zp_dedup);
1636e14bb325SJeff Bonwick 		BP_SET_BYTEORDER(bp, ZFS_HOST_BYTEORDER);
1637b24ab676SJeff Bonwick 		if (zp->zp_dedup) {
1638b24ab676SJeff Bonwick 			ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
1639b24ab676SJeff Bonwick 			ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REWRITE));
1640eb633035STom Caputi 			ASSERT(!zp->zp_encrypt ||
1641eb633035STom Caputi 			    DMU_OT_IS_ENCRYPTED(zp->zp_type));
1642b24ab676SJeff Bonwick 			zio->io_pipeline = ZIO_DDT_WRITE_PIPELINE;
1643b24ab676SJeff Bonwick 		}
164480901aeaSGeorge Wilson 		if (zp->zp_nopwrite) {
164580901aeaSGeorge Wilson 			ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
164680901aeaSGeorge Wilson 			ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REWRITE));
164780901aeaSGeorge Wilson 			zio->io_pipeline |= ZIO_STAGE_NOP_WRITE;
164880901aeaSGeorge Wilson 		}
1649b24ab676SJeff Bonwick 	}
1650b24ab676SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
1651b24ab676SJeff Bonwick }
1652b24ab676SJeff Bonwick 
1653b24ab676SJeff Bonwick static int
1654b24ab676SJeff Bonwick zio_free_bp_init(zio_t *zio)
1655b24ab676SJeff Bonwick {
1656b24ab676SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
1657b24ab676SJeff Bonwick 
1658b24ab676SJeff Bonwick 	if (zio->io_child_type == ZIO_CHILD_LOGICAL) {
1659b24ab676SJeff Bonwick 		if (BP_GET_DEDUP(bp))
1660b24ab676SJeff Bonwick 			zio->io_pipeline = ZIO_DDT_FREE_PIPELINE;
1661e14bb325SJeff Bonwick 	}
1662fa9e4066Sahrens 
16635cabbc6bSPrashanth Sreenivasa 	ASSERT3P(zio->io_bp, ==, &zio->io_bp_copy);
16645cabbc6bSPrashanth Sreenivasa 
1665e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
1666fa9e4066Sahrens }
1667fa9e4066Sahrens 
1668e14bb325SJeff Bonwick /*
1669e14bb325SJeff Bonwick  * ==========================================================================
1670e14bb325SJeff Bonwick  * Execute the I/O pipeline
1671e14bb325SJeff Bonwick  * ==========================================================================
1672e14bb325SJeff Bonwick  */
1673e14bb325SJeff Bonwick 
1674e14bb325SJeff Bonwick static void
1675ec94d322SAdam Leventhal zio_taskq_dispatch(zio_t *zio, zio_taskq_type_t q, boolean_t cutinline)
1676fa9e4066Sahrens {
167780eb36f2SGeorge Wilson 	spa_t *spa = zio->io_spa;
1678e14bb325SJeff Bonwick 	zio_type_t t = zio->io_type;
16795aeb9474SGarrett D'Amore 	int flags = (cutinline ? TQ_FRONT : 0);
16800a4e9518Sgw 
16810a4e9518Sgw 	/*
1682bbe36defSGeorge Wilson 	 * If we're a config writer or a probe, the normal issue and
1683bbe36defSGeorge Wilson 	 * interrupt threads may all be blocked waiting for the config lock.
1684bbe36defSGeorge Wilson 	 * In this case, select the otherwise-unused taskq for ZIO_TYPE_NULL.
16850a4e9518Sgw 	 */
1686bbe36defSGeorge Wilson 	if (zio->io_flags & (ZIO_FLAG_CONFIG_WRITER | ZIO_FLAG_PROBE))
1687e14bb325SJeff Bonwick 		t = ZIO_TYPE_NULL;
16880a4e9518Sgw 
16890a4e9518Sgw 	/*
1690e14bb325SJeff Bonwick 	 * A similar issue exists for the L2ARC write thread until L2ARC 2.0.
16910a4e9518Sgw 	 */
1692e14bb325SJeff Bonwick 	if (t == ZIO_TYPE_WRITE && zio->io_vd && zio->io_vd->vdev_aux)
1693e14bb325SJeff Bonwick 		t = ZIO_TYPE_NULL;
16940a4e9518Sgw 
169580eb36f2SGeorge Wilson 	/*
1696ec94d322SAdam Leventhal 	 * If this is a high priority I/O, then use the high priority taskq if
1697ec94d322SAdam Leventhal 	 * available.
169880eb36f2SGeorge Wilson 	 */
16992258ad0bSGeorge Wilson 	if ((zio->io_priority == ZIO_PRIORITY_NOW ||
17002258ad0bSGeorge Wilson 	    zio->io_priority == ZIO_PRIORITY_SYNC_WRITE) &&
1701ec94d322SAdam Leventhal 	    spa->spa_zio_taskq[t][q + 1].stqs_count != 0)
170280eb36f2SGeorge Wilson 		q++;
170380eb36f2SGeorge Wilson 
170480eb36f2SGeorge Wilson 	ASSERT3U(q, <, ZIO_TASKQ_TYPES);
17055aeb9474SGarrett D'Amore 
17065aeb9474SGarrett D'Amore 	/*
17075aeb9474SGarrett D'Amore 	 * NB: We are assuming that the zio can only be dispatched
17085aeb9474SGarrett D'Amore 	 * to a single taskq at a time.  It would be a grievous error
17095aeb9474SGarrett D'Amore 	 * to dispatch the zio to another taskq at the same time.
17105aeb9474SGarrett D'Amore 	 */
17115aeb9474SGarrett D'Amore 	ASSERT(zio->io_tqent.tqent_next == NULL);
1712ec94d322SAdam Leventhal 	spa_taskq_dispatch_ent(spa, t, q, (task_func_t *)zio_execute, zio,
1713ec94d322SAdam Leventhal 	    flags, &zio->io_tqent);
1714e14bb325SJeff Bonwick }
17150a4e9518Sgw 
1716e14bb325SJeff Bonwick static boolean_t
1717ec94d322SAdam Leventhal zio_taskq_member(zio_t *zio, zio_taskq_type_t q)
1718e14bb325SJeff Bonwick {
1719e14bb325SJeff Bonwick 	kthread_t *executor = zio->io_executor;
1720e14bb325SJeff Bonwick 	spa_t *spa = zio->io_spa;
17210a4e9518Sgw 
1722ec94d322SAdam Leventhal 	for (zio_type_t t = 0; t < ZIO_TYPES; t++) {
1723ec94d322SAdam Leventhal 		spa_taskqs_t *tqs = &spa->spa_zio_taskq[t][q];
1724ec94d322SAdam Leventhal 		uint_t i;
1725ec94d322SAdam Leventhal 		for (i = 0; i < tqs->stqs_count; i++) {
1726ec94d322SAdam Leventhal 			if (taskq_member(tqs->stqs_taskq[i], executor))
1727ec94d322SAdam Leventhal 				return (B_TRUE);
1728ec94d322SAdam Leventhal 		}
1729ec94d322SAdam Leventhal 	}
17300a4e9518Sgw 
1731e14bb325SJeff Bonwick 	return (B_FALSE);
1732e14bb325SJeff Bonwick }
1733e05725b1Sbonwick 
1734e14bb325SJeff Bonwick static int
1735e14bb325SJeff Bonwick zio_issue_async(zio_t *zio)
1736e14bb325SJeff Bonwick {
173735a5a358SJonathan Adams 	zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_FALSE);
1738e14bb325SJeff Bonwick 
1739e14bb325SJeff Bonwick 	return (ZIO_PIPELINE_STOP);
17400a4e9518Sgw }
17410a4e9518Sgw 
1742e14bb325SJeff Bonwick void
1743e14bb325SJeff Bonwick zio_interrupt(zio_t *zio)
17440a4e9518Sgw {
174535a5a358SJonathan Adams 	zio_taskq_dispatch(zio, ZIO_TASKQ_INTERRUPT, B_FALSE);
1746e14bb325SJeff Bonwick }
17470a4e9518Sgw 
174897e81309SPrakash Surya void
174997e81309SPrakash Surya zio_delay_interrupt(zio_t *zio)
175097e81309SPrakash Surya {
175197e81309SPrakash Surya 	/*
175297e81309SPrakash Surya 	 * The timeout_generic() function isn't defined in userspace, so
175397e81309SPrakash Surya 	 * rather than trying to implement the function, the zio delay
175497e81309SPrakash Surya 	 * functionality has been disabled for userspace builds.
175597e81309SPrakash Surya 	 */
175697e81309SPrakash Surya 
175797e81309SPrakash Surya #ifdef _KERNEL
175897e81309SPrakash Surya 	/*
175997e81309SPrakash Surya 	 * If io_target_timestamp is zero, then no delay has been registered
176097e81309SPrakash Surya 	 * for this IO, thus jump to the end of this function and "skip" the
176197e81309SPrakash Surya 	 * delay; issuing it directly to the zio layer.
176297e81309SPrakash Surya 	 */
176397e81309SPrakash Surya 	if (zio->io_target_timestamp != 0) {
176497e81309SPrakash Surya 		hrtime_t now = gethrtime();
176597e81309SPrakash Surya 
176697e81309SPrakash Surya 		if (now >= zio->io_target_timestamp) {
176797e81309SPrakash Surya 			/*
176897e81309SPrakash Surya 			 * This IO has already taken longer than the target
176997e81309SPrakash Surya 			 * delay to complete, so we don't want to delay it
177097e81309SPrakash Surya 			 * any longer; we "miss" the delay and issue it
177197e81309SPrakash Surya 			 * directly to the zio layer. This is likely due to
177297e81309SPrakash Surya 			 * the target latency being set to a value less than
177397e81309SPrakash Surya 			 * the underlying hardware can satisfy (e.g. delay
177497e81309SPrakash Surya 			 * set to 1ms, but the disks take 10ms to complete an
177597e81309SPrakash Surya 			 * IO request).
177697e81309SPrakash Surya 			 */
177797e81309SPrakash Surya 
177897e81309SPrakash Surya 			DTRACE_PROBE2(zio__delay__miss, zio_t *, zio,
177997e81309SPrakash Surya 			    hrtime_t, now);
178097e81309SPrakash Surya 
178197e81309SPrakash Surya 			zio_interrupt(zio);
178297e81309SPrakash Surya 		} else {
178397e81309SPrakash Surya 			hrtime_t diff = zio->io_target_timestamp - now;
178497e81309SPrakash Surya 
178597e81309SPrakash Surya 			DTRACE_PROBE3(zio__delay__hit, zio_t *, zio,
178697e81309SPrakash Surya 			    hrtime_t, now, hrtime_t, diff);
178797e81309SPrakash Surya 
178897e81309SPrakash Surya 			(void) timeout_generic(CALLOUT_NORMAL,
178997e81309SPrakash Surya 			    (void (*)(void *))zio_interrupt, zio, diff, 1, 0);
179097e81309SPrakash Surya 		}
179197e81309SPrakash Surya 
179297e81309SPrakash Surya 		return;
179397e81309SPrakash Surya 	}
179497e81309SPrakash Surya #endif
179597e81309SPrakash Surya 
179697e81309SPrakash Surya 	DTRACE_PROBE1(zio__delay__skip, zio_t *, zio);
179797e81309SPrakash Surya 	zio_interrupt(zio);
179897e81309SPrakash Surya }
179997e81309SPrakash Surya 
1800e14bb325SJeff Bonwick /*
1801e14bb325SJeff Bonwick  * Execute the I/O pipeline until one of the following occurs:
1802f7170741SWill Andrews  *
1803f7170741SWill Andrews  *	(1) the I/O completes
1804f7170741SWill Andrews  *	(2) the pipeline stalls waiting for dependent child I/Os
1805f7170741SWill Andrews  *	(3) the I/O issues, so we're waiting for an I/O completion interrupt
1806f7170741SWill Andrews  *	(4) the I/O is delegated by vdev-level caching or aggregation
1807f7170741SWill Andrews  *	(5) the I/O is deferred due to vdev-level queueing
1808f7170741SWill Andrews  *	(6) the I/O is handed off to another thread.
1809f7170741SWill Andrews  *
1810f7170741SWill Andrews  * In all cases, the pipeline stops whenever there's no CPU work; it never
1811f7170741SWill Andrews  * burns a thread in cv_wait().
1812e14bb325SJeff Bonwick  *
1813e14bb325SJeff Bonwick  * There's no locking on io_stage because there's no legitimate way
1814e14bb325SJeff Bonwick  * for multiple threads to be attempting to process the same I/O.
1815e14bb325SJeff Bonwick  */
1816b24ab676SJeff Bonwick static zio_pipe_stage_t *zio_pipeline[];
18170a4e9518Sgw 
1818e14bb325SJeff Bonwick void
1819e14bb325SJeff Bonwick zio_execute(zio_t *zio)
1820e14bb325SJeff Bonwick {
1821e14bb325SJeff Bonwick 	zio->io_executor = curthread;
18220a4e9518Sgw 
18230f7643c7SGeorge Wilson 	ASSERT3U(zio->io_queued_timestamp, >, 0);
18240f7643c7SGeorge Wilson 
1825e14bb325SJeff Bonwick 	while (zio->io_stage < ZIO_STAGE_DONE) {
1826b24ab676SJeff Bonwick 		enum zio_stage pipeline = zio->io_pipeline;
1827b24ab676SJeff Bonwick 		enum zio_stage stage = zio->io_stage;
1828e14bb325SJeff Bonwick 		int rv;
18290a4e9518Sgw 
1830e14bb325SJeff Bonwick 		ASSERT(!MUTEX_HELD(&zio->io_lock));
1831b24ab676SJeff Bonwick 		ASSERT(ISP2(stage));
1832b24ab676SJeff Bonwick 		ASSERT(zio->io_stall == NULL);
18330a4e9518Sgw 
1834b24ab676SJeff Bonwick 		do {
1835b24ab676SJeff Bonwick 			stage <<= 1;
1836b24ab676SJeff Bonwick 		} while ((stage & pipeline) == 0);
1837e14bb325SJeff Bonwick 
1838e14bb325SJeff Bonwick 		ASSERT(stage <= ZIO_STAGE_DONE);
18390a4e9518Sgw 
18400a4e9518Sgw 		/*
1841e14bb325SJeff Bonwick 		 * If we are in interrupt context and this pipeline stage
1842e14bb325SJeff Bonwick 		 * will grab a config lock that is held across I/O,
1843b24ab676SJeff Bonwick 		 * or may wait for an I/O that needs an interrupt thread
1844b24ab676SJeff Bonwick 		 * to complete, issue async to avoid deadlock.
184535a5a358SJonathan Adams 		 *
184635a5a358SJonathan Adams 		 * For VDEV_IO_START, we cut in line so that the io will
184735a5a358SJonathan Adams 		 * be sent to disk promptly.
18480a4e9518Sgw 		 */
1849b24ab676SJeff Bonwick 		if ((stage & ZIO_BLOCKING_STAGES) && zio->io_vd == NULL &&
1850e14bb325SJeff Bonwick 		    zio_taskq_member(zio, ZIO_TASKQ_INTERRUPT)) {
185135a5a358SJonathan Adams 			boolean_t cut = (stage == ZIO_STAGE_VDEV_IO_START) ?
185235a5a358SJonathan Adams 			    zio_requeue_io_start_cut_in_line : B_FALSE;
185335a5a358SJonathan Adams 			zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, cut);
1854e14bb325SJeff Bonwick 			return;
18550a4e9518Sgw 		}
18560a4e9518Sgw 
1857*f6ef4223SJoshua M. Clulow #ifdef _KERNEL
1858*f6ef4223SJoshua M. Clulow 		/*
1859*f6ef4223SJoshua M. Clulow 		 * The I/O pipeline is a part of the machinery responsible for
1860*f6ef4223SJoshua M. Clulow 		 * evacuation of memory pages to disk when we are under
1861*f6ef4223SJoshua M. Clulow 		 * sufficient memory pressure for pageout to run.  By setting
1862*f6ef4223SJoshua M. Clulow 		 * this flag, allocations may dip into pages in the pageout
1863*f6ef4223SJoshua M. Clulow 		 * reserved pool in order to try to make forward progress.
1864*f6ef4223SJoshua M. Clulow 		 */
1865*f6ef4223SJoshua M. Clulow 		bool set_pushpage = false;
1866*f6ef4223SJoshua M. Clulow 		if (!(curthread->t_flag & T_PUSHPAGE)) {
1867*f6ef4223SJoshua M. Clulow 			/*
1868*f6ef4223SJoshua M. Clulow 			 * We can be called recursively, so we need to remember
1869*f6ef4223SJoshua M. Clulow 			 * if this frame was the one that first set the flag or
1870*f6ef4223SJoshua M. Clulow 			 * not.
1871*f6ef4223SJoshua M. Clulow 			 */
1872*f6ef4223SJoshua M. Clulow 			set_pushpage = true;
1873*f6ef4223SJoshua M. Clulow 			curthread->t_flag |= T_PUSHPAGE;
1874*f6ef4223SJoshua M. Clulow 		}
1875*f6ef4223SJoshua M. Clulow #endif
1876*f6ef4223SJoshua M. Clulow 
1877e14bb325SJeff Bonwick 		zio->io_stage = stage;
18780f7643c7SGeorge Wilson 		zio->io_pipeline_trace |= zio->io_stage;
1879bf16b11eSMatthew Ahrens 		rv = zio_pipeline[highbit64(stage) - 1](zio);
18800a4e9518Sgw 
1881*f6ef4223SJoshua M. Clulow #ifdef _KERNEL
1882*f6ef4223SJoshua M. Clulow 		if (set_pushpage) {
1883*f6ef4223SJoshua M. Clulow 			curthread->t_flag &= ~T_PUSHPAGE;
1884*f6ef4223SJoshua M. Clulow 		}
1885*f6ef4223SJoshua M. Clulow #endif
1886*f6ef4223SJoshua M. Clulow 
1887e14bb325SJeff Bonwick 		if (rv == ZIO_PIPELINE_STOP)
1888e14bb325SJeff Bonwick 			return;
18890a4e9518Sgw 
1890e14bb325SJeff Bonwick 		ASSERT(rv == ZIO_PIPELINE_CONTINUE);
1891e14bb325SJeff Bonwick 	}
18920a4e9518Sgw }
18930a4e9518Sgw 
1894e14bb325SJeff Bonwick /*
1895e14bb325SJeff Bonwick  * ==========================================================================
1896e14bb325SJeff Bonwick  * Initiate I/O, either sync or async
1897e14bb325SJeff Bonwick  * ==========================================================================
1898e14bb325SJeff Bonwick  */
1899e14bb325SJeff Bonwick int
1900e14bb325SJeff Bonwick zio_wait(zio_t *zio)
19010a4e9518Sgw {
1902e14bb325SJeff Bonwick 	int error;
19030a4e9518Sgw 
19041271e4b1SPrakash Surya 	ASSERT3P(zio->io_stage, ==, ZIO_STAGE_OPEN);
19051271e4b1SPrakash Surya 	ASSERT3P(zio->io_executor, ==, NULL);
19060a4e9518Sgw 
1907e14bb325SJeff Bonwick 	zio->io_waiter = curthread;
19080f7643c7SGeorge Wilson 	ASSERT0(zio->io_queued_timestamp);
19090f7643c7SGeorge Wilson 	zio->io_queued_timestamp = gethrtime();
1910e05725b1Sbonwick 
1911e14bb325SJeff Bonwick 	zio_execute(zio);
19120a4e9518Sgw 
1913e14bb325SJeff Bonwick 	mutex_enter(&zio->io_lock);
1914e14bb325SJeff Bonwick 	while (zio->io_executor != NULL)
1915e14bb325SJeff Bonwick 		cv_wait(&zio->io_cv, &zio->io_lock);
1916e14bb325SJeff Bonwick 	mutex_exit(&zio->io_lock);
191732b87932Sek 
1918e14bb325SJeff Bonwick 	error = zio->io_error;
1919e14bb325SJeff Bonwick 	zio_destroy(zio);
192032b87932Sek 
1921e14bb325SJeff Bonwick 	return (error);
192232b87932Sek }
192332b87932Sek 
1924e14bb325SJeff Bonwick void
1925e14bb325SJeff Bonwick zio_nowait(zio_t *zio)
19260a4e9518Sgw {
19271271e4b1SPrakash Surya 	ASSERT3P(zio->io_executor, ==, NULL);
1928fa9e4066Sahrens 
1929a3f829aeSBill Moore 	if (zio->io_child_type == ZIO_CHILD_LOGICAL &&
1930a3f829aeSBill Moore 	    zio_unique_parent(zio) == NULL) {
1931ea8dc4b6Seschrock 		/*
1932e14bb325SJeff Bonwick 		 * This is a logical async I/O with no parent to wait for it.
193354d692b7SGeorge Wilson 		 * We add it to the spa_async_root_zio "Godfather" I/O which
193454d692b7SGeorge Wilson 		 * will ensure they complete prior to unloading the pool.
1935ea8dc4b6Seschrock 		 */
1936e14bb325SJeff Bonwick 		spa_t *spa = zio->io_spa;
193754d692b7SGeorge Wilson 
19386f834bc1SMatthew Ahrens 		zio_add_child(spa->spa_async_zio_root[CPU_SEQID], zio);
1939e14bb325SJeff Bonwick 	}
1940ea8dc4b6Seschrock 
19410f7643c7SGeorge Wilson 	ASSERT0(zio->io_queued_timestamp);
19420f7643c7SGeorge Wilson 	zio->io_queued_timestamp = gethrtime();
1943e14bb325SJeff Bonwick 	zio_execute(zio);
1944e14bb325SJeff Bonwick }
1945ea8dc4b6Seschrock 
1946e14bb325SJeff Bonwick /*
1947e14bb325SJeff Bonwick  * ==========================================================================
19481271e4b1SPrakash Surya  * Reexecute, cancel, or suspend/resume failed I/O
1949e14bb325SJeff Bonwick  * ==========================================================================
1950e14bb325SJeff Bonwick  */
1951fa9e4066Sahrens 
1952e14bb325SJeff Bonwick static void
1953e14bb325SJeff Bonwick zio_reexecute(zio_t *pio)
1954e14bb325SJeff Bonwick {
1955a3f829aeSBill Moore 	zio_t *cio, *cio_next;
1956a3f829aeSBill Moore 
1957a3f829aeSBill Moore 	ASSERT(pio->io_child_type == ZIO_CHILD_LOGICAL);
1958a3f829aeSBill Moore 	ASSERT(pio->io_orig_stage == ZIO_STAGE_OPEN);
1959f5383399SBill Moore 	ASSERT(pio->io_gang_leader == NULL);
1960f5383399SBill Moore 	ASSERT(pio->io_gang_tree == NULL);
1961e05725b1Sbonwick 
1962e14bb325SJeff Bonwick 	pio->io_flags = pio->io_orig_flags;
1963e14bb325SJeff Bonwick 	pio->io_stage = pio->io_orig_stage;
1964e14bb325SJeff Bonwick 	pio->io_pipeline = pio->io_orig_pipeline;
1965e14bb325SJeff Bonwick 	pio->io_reexecute = 0;
196680901aeaSGeorge Wilson 	pio->io_flags |= ZIO_FLAG_REEXECUTED;
19670f7643c7SGeorge Wilson 	pio->io_pipeline_trace = 0;
1968e14bb325SJeff Bonwick 	pio->io_error = 0;
1969a3f829aeSBill Moore 	for (int w = 0; w < ZIO_WAIT_TYPES; w++)
1970a3f829aeSBill Moore 		pio->io_state[w] = 0;
1971e14bb325SJeff Bonwick 	for (int c = 0; c < ZIO_CHILD_TYPES; c++)
1972e14bb325SJeff Bonwick 		pio->io_child_error[c] = 0;
19730a4e9518Sgw 
1974b24ab676SJeff Bonwick 	if (IO_IS_ALLOCATING(pio))
1975b24ab676SJeff Bonwick 		BP_ZERO(pio->io_bp);
1976d58459f4Sek 
1977e14bb325SJeff Bonwick 	/*
1978e14bb325SJeff Bonwick 	 * As we reexecute pio's children, new children could be created.
1979a3f829aeSBill Moore 	 * New children go to the head of pio's io_child_list, however,
1980e14bb325SJeff Bonwick 	 * so we will (correctly) not reexecute them.  The key is that
1981a3f829aeSBill Moore 	 * the remainder of pio's io_child_list, from 'cio_next' onward,
1982a3f829aeSBill Moore 	 * cannot be affected by any side effects of reexecuting 'cio'.
1983e14bb325SJeff Bonwick 	 */
19840f7643c7SGeorge Wilson 	zio_link_t *zl = NULL;
1985a3874b8bSToomas Soome 	mutex_enter(&pio->io_lock);
19860f7643c7SGeorge Wilson 	for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
19870f7643c7SGeorge Wilson 		cio_next = zio_walk_children(pio, &zl);
1988a3f829aeSBill Moore 		for (int w = 0; w < ZIO_WAIT_TYPES; w++)
1989a3f829aeSBill Moore 			pio->io_children[cio->io_child_type][w]++;
1990e14bb325SJeff Bonwick 		mutex_exit(&pio->io_lock);
1991a3f829aeSBill Moore 		zio_reexecute(cio);
1992a3874b8bSToomas Soome 		mutex_enter(&pio->io_lock);
1993fa9e4066Sahrens 	}
1994a3874b8bSToomas Soome 	mutex_exit(&pio->io_lock);
1995e05725b1Sbonwick 
1996e14bb325SJeff Bonwick 	/*
1997e14bb325SJeff Bonwick 	 * Now that all children have been reexecuted, execute the parent.
199854d692b7SGeorge Wilson 	 * We don't reexecute "The Godfather" I/O here as it's the
199948bbca81SDaniel Hoffman 	 * responsibility of the caller to wait on it.
2000e14bb325SJeff Bonwick 	 */
20010f7643c7SGeorge Wilson 	if (!(pio->io_flags & ZIO_FLAG_GODFATHER)) {
20020f7643c7SGeorge Wilson 		pio->io_queued_timestamp = gethrtime();
200354d692b7SGeorge Wilson 		zio_execute(pio);
20040f7643c7SGeorge Wilson 	}
20050a4e9518Sgw }
20060a4e9518Sgw 
2007e14bb325SJeff Bonwick void
2008e0f1c0afSOlaf Faaland zio_suspend(spa_t *spa, zio_t *zio, zio_suspend_reason_t reason)
20090a4e9518Sgw {
2010e14bb325SJeff Bonwick 	if (spa_get_failmode(spa) == ZIO_FAILURE_MODE_PANIC)
2011e14bb325SJeff Bonwick 		fm_panic("Pool '%s' has encountered an uncorrectable I/O "
2012e14bb325SJeff Bonwick 		    "failure and the failure mode property for this pool "
2013e14bb325SJeff Bonwick 		    "is set to panic.", spa_name(spa));
20140a4e9518Sgw 
20158548ec78SJohn Levon 	cmn_err(CE_WARN, "Pool '%s' has encountered an uncorrectable I/O "
20168548ec78SJohn Levon 	    "failure and has been suspended; `zpool clear` will be required "
20178548ec78SJohn Levon 	    "before the pool can be written to.", spa_name(spa));
20188548ec78SJohn Levon 
20199b088140SToomas Soome 	(void) zfs_ereport_post(FM_EREPORT_ZFS_IO_FAILURE, spa, NULL,
2020eb633035STom Caputi 	    NULL, NULL, 0, 0);
20210a4e9518Sgw 
2022e14bb325SJeff Bonwick 	mutex_enter(&spa->spa_suspend_lock);
2023fa9e4066Sahrens 
2024e14bb325SJeff Bonwick 	if (spa->spa_suspend_zio_root == NULL)
202554d692b7SGeorge Wilson 		spa->spa_suspend_zio_root = zio_root(spa, NULL, NULL,
202654d692b7SGeorge Wilson 		    ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE |
202754d692b7SGeorge Wilson 		    ZIO_FLAG_GODFATHER);
2028fa9e4066Sahrens 
2029e0f1c0afSOlaf Faaland 	spa->spa_suspended = reason;
2030fa9e4066Sahrens 
2031e14bb325SJeff Bonwick 	if (zio != NULL) {
203254d692b7SGeorge Wilson 		ASSERT(!(zio->io_flags & ZIO_FLAG_GODFATHER));
2033e14bb325SJeff Bonwick 		ASSERT(zio != spa->spa_suspend_zio_root);
2034e14bb325SJeff Bonwick 		ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
2035a3f829aeSBill Moore 		ASSERT(zio_unique_parent(zio) == NULL);
2036e14bb325SJeff Bonwick 		ASSERT(zio->io_stage == ZIO_STAGE_DONE);
2037e14bb325SJeff Bonwick 		zio_add_child(spa->spa_suspend_zio_root, zio);
2038e14bb325SJeff Bonwick 	}
2039fa9e4066Sahrens 
2040e14bb325SJeff Bonwick 	mutex_exit(&spa->spa_suspend_lock);
2041e14bb325SJeff Bonwick }
2042fa9e4066Sahrens 
204354d692b7SGeorge Wilson int
2044e14bb325SJeff Bonwick zio_resume(spa_t *spa)
2045e14bb325SJeff Bonwick {
204654d692b7SGeorge Wilson 	zio_t *pio;
2047fa9e4066Sahrens 
2048b3995adbSahrens 	/*
2049e14bb325SJeff Bonwick 	 * Reexecute all previously suspended i/o.
2050b3995adbSahrens 	 */
2051e14bb325SJeff Bonwick 	mutex_enter(&spa->spa_suspend_lock);
2052e0f1c0afSOlaf Faaland 	spa->spa_suspended = ZIO_SUSPEND_NONE;
2053e14bb325SJeff Bonwick 	cv_broadcast(&spa->spa_suspend_cv);
2054e14bb325SJeff Bonwick 	pio = spa->spa_suspend_zio_root;
2055e14bb325SJeff Bonwick 	spa->spa_suspend_zio_root = NULL;
2056e14bb325SJeff Bonwick 	mutex_exit(&spa->spa_suspend_lock);
2057e14bb325SJeff Bonwick 
2058e14bb325SJeff Bonwick 	if (pio == NULL)
205954d692b7SGeorge Wilson 		return (0);
2060e14bb325SJeff Bonwick 
206154d692b7SGeorge Wilson 	zio_reexecute(pio);
206254d692b7SGeorge Wilson 	return (zio_wait(pio));
2063e14bb325SJeff Bonwick }
2064e14bb325SJeff Bonwick 
2065e14bb325SJeff Bonwick void
2066e14bb325SJeff Bonwick zio_resume_wait(spa_t *spa)
2067e14bb325SJeff Bonwick {
2068e14bb325SJeff Bonwick 	mutex_enter(&spa->spa_suspend_lock);
2069e14bb325SJeff Bonwick 	while (spa_suspended(spa))
2070e14bb325SJeff Bonwick 		cv_wait(&spa->spa_suspend_cv, &spa->spa_suspend_lock);
2071e14bb325SJeff Bonwick 	mutex_exit(&spa->spa_suspend_lock);
2072fa9e4066Sahrens }
2073fa9e4066Sahrens 
2074fa9e4066Sahrens /*
2075fa9e4066Sahrens  * ==========================================================================
2076e14bb325SJeff Bonwick  * Gang blocks.
2077e14bb325SJeff Bonwick  *
2078e14bb325SJeff Bonwick  * A gang block is a collection of small blocks that looks to the DMU
2079e14bb325SJeff Bonwick  * like one large block.  When zio_dva_allocate() cannot find a block
2080e14bb325SJeff Bonwick  * of the requested size, due to either severe fragmentation or the pool
2081e14bb325SJeff Bonwick  * being nearly full, it calls zio_write_gang_block() to construct the
2082e14bb325SJeff Bonwick  * block from smaller fragments.
2083e14bb325SJeff Bonwick  *
2084e14bb325SJeff Bonwick  * A gang block consists of a gang header (zio_gbh_phys_t) and up to
2085e14bb325SJeff Bonwick  * three (SPA_GBH_NBLKPTRS) gang members.  The gang header is just like
2086e14bb325SJeff Bonwick  * an indirect block: it's an array of block pointers.  It consumes
2087e14bb325SJeff Bonwick  * only one sector and hence is allocatable regardless of fragmentation.
2088e14bb325SJeff Bonwick  * The gang header's bps point to its gang members, which hold the data.
2089e14bb325SJeff Bonwick  *
2090e14bb325SJeff Bonwick  * Gang blocks are self-checksumming, using the bp's <vdev, offset, txg>
2091e14bb325SJeff Bonwick  * as the verifier to ensure uniqueness of the SHA256 checksum.
2092e14bb325SJeff Bonwick  * Critically, the gang block bp's blk_cksum is the checksum of the data,
2093e14bb325SJeff Bonwick  * not the gang header.  This ensures that data block signatures (needed for
2094e14bb325SJeff Bonwick  * deduplication) are independent of how the block is physically stored.
2095e14bb325SJeff Bonwick  *
2096e14bb325SJeff Bonwick  * Gang blocks can be nested: a gang member may itself be a gang block.
2097e14bb325SJeff Bonwick  * Thus every gang block is a tree in which root and all interior nodes are
2098e14bb325SJeff Bonwick  * gang headers, and the leaves are normal blocks that contain user data.
2099e14bb325SJeff Bonwick  * The root of the gang tree is called the gang leader.
2100e14bb325SJeff Bonwick  *
2101e14bb325SJeff Bonwick  * To perform any operation (read, rewrite, free, claim) on a gang block,
2102e14bb325SJeff Bonwick  * zio_gang_assemble() first assembles the gang tree (minus data leaves)
2103e14bb325SJeff Bonwick  * in the io_gang_tree field of the original logical i/o by recursively
2104e14bb325SJeff Bonwick  * reading the gang leader and all gang headers below it.  This yields
2105e14bb325SJeff Bonwick  * an in-core tree containing the contents of every gang header and the
2106e14bb325SJeff Bonwick  * bps for every constituent of the gang block.
2107e14bb325SJeff Bonwick  *
2108e14bb325SJeff Bonwick  * With the gang tree now assembled, zio_gang_issue() just walks the gang tree
2109e14bb325SJeff Bonwick  * and invokes a callback on each bp.  To free a gang block, zio_gang_issue()
2110e14bb325SJeff Bonwick  * calls zio_free_gang() -- a trivial wrapper around zio_free() -- for each bp.
2111e14bb325SJeff Bonwick  * zio_claim_gang() provides a similarly trivial wrapper for zio_claim().
2112e14bb325SJeff Bonwick  * zio_read_gang() is a wrapper around zio_read() that omits reading gang
2113e14bb325SJeff Bonwick  * headers, since we already have those in io_gang_tree.  zio_rewrite_gang()
2114e14bb325SJeff Bonwick  * performs a zio_rewrite() of the data or, for gang headers, a zio_rewrite()
2115e14bb325SJeff Bonwick  * of the gang header plus zio_checksum_compute() of the data to update the
2116e14bb325SJeff Bonwick  * gang header's blk_cksum as described above.
2117e14bb325SJeff Bonwick  *
2118e14bb325SJeff Bonwick  * The two-phase assemble/issue model solves the problem of partial failure --
2119e14bb325SJeff Bonwick  * what if you'd freed part of a gang block but then couldn't read the
2120e14bb325SJeff Bonwick  * gang header for another part?  Assembling the entire gang tree first
2121e14bb325SJeff Bonwick  * ensures that all the necessary gang header I/O has succeeded before
2122e14bb325SJeff Bonwick  * starting the actual work of free, claim, or write.  Once the gang tree
2123e14bb325SJeff Bonwick  * is assembled, free and claim are in-memory operations that cannot fail.
2124e14bb325SJeff Bonwick  *
2125e14bb325SJeff Bonwick  * In the event that a gang write fails, zio_dva_unallocate() walks the
2126e14bb325SJeff Bonwick  * gang tree to immediately free (i.e. insert back into the space map)
2127e14bb325SJeff Bonwick  * everything we've allocated.  This ensures that we don't get ENOSPC
2128e14bb325SJeff Bonwick  * errors during repeated suspend/resume cycles due to a flaky device.
2129e14bb325SJeff Bonwick  *
2130e14bb325SJeff Bonwick  * Gang rewrites only happen during sync-to-convergence.  If we can't assemble
2131e14bb325SJeff Bonwick  * the gang tree, we won't modify the block, so we can safely defer the free
2132e14bb325SJeff Bonwick  * (knowing that the block is still intact).  If we *can* assemble the gang
2133e14bb325SJeff Bonwick  * tree, then even if some of the rewrites fail, zio_dva_unallocate() will free
2134e14bb325SJeff Bonwick  * each constituent bp and we can allocate a new block on the next sync pass.
2135e14bb325SJeff Bonwick  *
2136e14bb325SJeff Bonwick  * In all cases, the gang tree allows complete recovery from partial failure.
2137fa9e4066Sahrens  * ==========================================================================
2138fa9e4066Sahrens  */
2139e14bb325SJeff Bonwick 
2140770499e1SDan Kimmel static void
2141770499e1SDan Kimmel zio_gang_issue_func_done(zio_t *zio)
2142770499e1SDan Kimmel {
2143770499e1SDan Kimmel 	abd_put(zio->io_abd);
2144770499e1SDan Kimmel }
2145770499e1SDan Kimmel 
2146e14bb325SJeff Bonwick static zio_t *
2147770499e1SDan Kimmel zio_read_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2148770499e1SDan Kimmel     uint64_t offset)
2149fa9e4066Sahrens {
2150e14bb325SJeff Bonwick 	if (gn != NULL)
2151e14bb325SJeff Bonwick 		return (pio);
2152fa9e4066Sahrens 
2153770499e1SDan Kimmel 	return (zio_read(pio, pio->io_spa, bp, abd_get_offset(data, offset),
2154770499e1SDan Kimmel 	    BP_GET_PSIZE(bp), zio_gang_issue_func_done,
2155770499e1SDan Kimmel 	    NULL, pio->io_priority, ZIO_GANG_CHILD_FLAGS(pio),
2156e14bb325SJeff Bonwick 	    &pio->io_bookmark));
2157e14bb325SJeff Bonwick }
2158e14bb325SJeff Bonwick 
2159770499e1SDan Kimmel static zio_t *
2160770499e1SDan Kimmel zio_rewrite_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2161770499e1SDan Kimmel     uint64_t offset)
2162e14bb325SJeff Bonwick {
2163e14bb325SJeff Bonwick 	zio_t *zio;
2164e14bb325SJeff Bonwick 
2165e14bb325SJeff Bonwick 	if (gn != NULL) {
2166770499e1SDan Kimmel 		abd_t *gbh_abd =
2167770499e1SDan Kimmel 		    abd_get_from_buf(gn->gn_gbh, SPA_GANGBLOCKSIZE);
2168e14bb325SJeff Bonwick 		zio = zio_rewrite(pio, pio->io_spa, pio->io_txg, bp,
2169770499e1SDan Kimmel 		    gbh_abd, SPA_GANGBLOCKSIZE, zio_gang_issue_func_done, NULL,
2170770499e1SDan Kimmel 		    pio->io_priority, ZIO_GANG_CHILD_FLAGS(pio),
2171770499e1SDan Kimmel 		    &pio->io_bookmark);
2172fa9e4066Sahrens 		/*
2173e14bb325SJeff Bonwick 		 * As we rewrite each gang header, the pipeline will compute
2174e14bb325SJeff Bonwick 		 * a new gang block header checksum for it; but no one will
2175e14bb325SJeff Bonwick 		 * compute a new data checksum, so we do that here.  The one
2176e14bb325SJeff Bonwick 		 * exception is the gang leader: the pipeline already computed
2177e14bb325SJeff Bonwick 		 * its data checksum because that stage precedes gang assembly.
2178e14bb325SJeff Bonwick 		 * (Presently, nothing actually uses interior data checksums;
2179e14bb325SJeff Bonwick 		 * this is just good hygiene.)
2180fa9e4066Sahrens 		 */
2181f5383399SBill Moore 		if (gn != pio->io_gang_leader->io_gang_tree) {
2182770499e1SDan Kimmel 			abd_t *buf = abd_get_offset(data, offset);
2183770499e1SDan Kimmel 
2184e14bb325SJeff Bonwick 			zio_checksum_compute(zio, BP_GET_CHECKSUM(bp),
2185770499e1SDan Kimmel 			    buf, BP_GET_PSIZE(bp));
2186770499e1SDan Kimmel 
2187770499e1SDan Kimmel 			abd_put(buf);
2188e14bb325SJeff Bonwick 		}
2189b24ab676SJeff Bonwick 		/*
2190b24ab676SJeff Bonwick 		 * If we are here to damage data for testing purposes,
2191b24ab676SJeff Bonwick 		 * leave the GBH alone so that we can detect the damage.
2192b24ab676SJeff Bonwick 		 */
2193b24ab676SJeff Bonwick 		if (pio->io_gang_leader->io_flags & ZIO_FLAG_INDUCE_DAMAGE)
2194b24ab676SJeff Bonwick 			zio->io_pipeline &= ~ZIO_VDEV_IO_STAGES;
2195fa9e4066Sahrens 	} else {
2196e14bb325SJeff Bonwick 		zio = zio_rewrite(pio, pio->io_spa, pio->io_txg, bp,
2197770499e1SDan Kimmel 		    abd_get_offset(data, offset), BP_GET_PSIZE(bp),
2198770499e1SDan Kimmel 		    zio_gang_issue_func_done, NULL, pio->io_priority,
2199e14bb325SJeff Bonwick 		    ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
2200fa9e4066Sahrens 	}
2201fa9e4066Sahrens 
2202e14bb325SJeff Bonwick 	return (zio);
2203e14bb325SJeff Bonwick }
2204fa9e4066Sahrens 
2205e14bb325SJeff Bonwick /* ARGSUSED */
2206770499e1SDan Kimmel static zio_t *
2207770499e1SDan Kimmel zio_free_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2208770499e1SDan Kimmel     uint64_t offset)
2209e14bb325SJeff Bonwick {
2210b24ab676SJeff Bonwick 	return (zio_free_sync(pio, pio->io_spa, pio->io_txg, bp,
2211b24ab676SJeff Bonwick 	    ZIO_GANG_CHILD_FLAGS(pio)));
2212fa9e4066Sahrens }
2213fa9e4066Sahrens 
2214e14bb325SJeff Bonwick /* ARGSUSED */
2215770499e1SDan Kimmel static zio_t *
2216770499e1SDan Kimmel zio_claim_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2217770499e1SDan Kimmel     uint64_t offset)
2218fa9e4066Sahrens {
2219e14bb325SJeff Bonwick 	return (zio_claim(pio, pio->io_spa, pio->io_txg, bp,
2220e14bb325SJeff Bonwick 	    NULL, NULL, ZIO_GANG_CHILD_FLAGS(pio)));
2221e14bb325SJeff Bonwick }
2222fa9e4066Sahrens 
2223e14bb325SJeff Bonwick static zio_gang_issue_func_t *zio_gang_issue_func[ZIO_TYPES] = {
2224e14bb325SJeff Bonwick 	NULL,
2225e14bb325SJeff Bonwick 	zio_read_gang,
2226e14bb325SJeff Bonwick 	zio_rewrite_gang,
2227e14bb325SJeff Bonwick 	zio_free_gang,
2228e14bb325SJeff Bonwick 	zio_claim_gang,
2229e14bb325SJeff Bonwick 	NULL
2230e14bb325SJeff Bonwick };
2231fa9e4066Sahrens 
2232e14bb325SJeff Bonwick static void zio_gang_tree_assemble_done(zio_t *zio);
2233fa9e4066Sahrens 
2234e14bb325SJeff Bonwick static zio_gang_node_t *
2235e14bb325SJeff Bonwick zio_gang_node_alloc(zio_gang_node_t **gnpp)
2236e14bb325SJeff Bonwick {
2237e14bb325SJeff Bonwick 	zio_gang_node_t *gn;
2238fa9e4066Sahrens 
2239e14bb325SJeff Bonwick 	ASSERT(*gnpp == NULL);
2240fa9e4066Sahrens 
2241e14bb325SJeff Bonwick 	gn = kmem_zalloc(sizeof (*gn), KM_SLEEP);
2242e14bb325SJeff Bonwick 	gn->gn_gbh = zio_buf_alloc(SPA_GANGBLOCKSIZE);
2243e14bb325SJeff Bonwick 	*gnpp = gn;
2244e14bb325SJeff Bonwick 
2245e14bb325SJeff Bonwick 	return (gn);
2246fa9e4066Sahrens }
2247fa9e4066Sahrens 
2248fa9e4066Sahrens static void
2249e14bb325SJeff Bonwick zio_gang_node_free(zio_gang_node_t **gnpp)
2250fa9e4066Sahrens {
2251e14bb325SJeff Bonwick 	zio_gang_node_t *gn = *gnpp;
2252fa9e4066Sahrens 
2253e14bb325SJeff Bonwick 	for (int g = 0; g < SPA_GBH_NBLKPTRS; g++)
2254e14bb325SJeff Bonwick 		ASSERT(gn->gn_child[g] == NULL);
2255e14bb325SJeff Bonwick 
2256e14bb325SJeff Bonwick 	zio_buf_free(gn->gn_gbh, SPA_GANGBLOCKSIZE);
2257e14bb325SJeff Bonwick 	kmem_free(gn, sizeof (*gn));
2258e14bb325SJeff Bonwick 	*gnpp = NULL;
2259fa9e4066Sahrens }
2260fa9e4066Sahrens 
2261e14bb325SJeff Bonwick static void
2262e14bb325SJeff Bonwick zio_gang_tree_free(zio_gang_node_t **gnpp)
2263fa9e4066Sahrens {
2264e14bb325SJeff Bonwick 	zio_gang_node_t *gn = *gnpp;
2265fa9e4066Sahrens 
2266e14bb325SJeff Bonwick 	if (gn == NULL)
2267e14bb325SJeff Bonwick 		return;
2268fa9e4066Sahrens 
2269e14bb325SJeff Bonwick 	for (int g = 0; g < SPA_GBH_NBLKPTRS; g++)
2270e14bb325SJeff Bonwick 		zio_gang_tree_free(&gn->gn_child[g]);
2271fa9e4066Sahrens 
2272e14bb325SJeff Bonwick 	zio_gang_node_free(gnpp);
2273fa9e4066Sahrens }
2274fa9e4066Sahrens 
2275e14bb325SJeff Bonwick static void
2276f5383399SBill Moore zio_gang_tree_assemble(zio_t *gio, blkptr_t *bp, zio_gang_node_t **gnpp)
2277fa9e4066Sahrens {
2278e14bb325SJeff Bonwick 	zio_gang_node_t *gn = zio_gang_node_alloc(gnpp);
2279770499e1SDan Kimmel 	abd_t *gbh_abd = abd_get_from_buf(gn->gn_gbh, SPA_GANGBLOCKSIZE);
2280e14bb325SJeff Bonwick 
2281f5383399SBill Moore 	ASSERT(gio->io_gang_leader == gio);
2282e14bb325SJeff Bonwick 	ASSERT(BP_IS_GANG(bp));
2283fa9e4066Sahrens 
2284770499e1SDan Kimmel 	zio_nowait(zio_read(gio, gio->io_spa, bp, gbh_abd, SPA_GANGBLOCKSIZE,
2285770499e1SDan Kimmel 	    zio_gang_tree_assemble_done, gn, gio->io_priority,
2286770499e1SDan Kimmel 	    ZIO_GANG_CHILD_FLAGS(gio), &gio->io_bookmark));
2287e14bb325SJeff Bonwick }
2288fa9e4066Sahrens 
2289e14bb325SJeff Bonwick static void
2290e14bb325SJeff Bonwick zio_gang_tree_assemble_done(zio_t *zio)
2291e14bb325SJeff Bonwick {
2292f5383399SBill Moore 	zio_t *gio = zio->io_gang_leader;
2293e14bb325SJeff Bonwick 	zio_gang_node_t *gn = zio->io_private;
2294e14bb325SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2295fa9e4066Sahrens 
2296f5383399SBill Moore 	ASSERT(gio == zio_unique_parent(zio));
2297b24ab676SJeff Bonwick 	ASSERT(zio->io_child_count == 0);
2298fa9e4066Sahrens 
2299e14bb325SJeff Bonwick 	if (zio->io_error)
2300e14bb325SJeff Bonwick 		return;
2301fa9e4066Sahrens 
2302770499e1SDan Kimmel 	/* this ABD was created from a linear buf in zio_gang_tree_assemble */
2303e14bb325SJeff Bonwick 	if (BP_SHOULD_BYTESWAP(bp))
2304770499e1SDan Kimmel 		byteswap_uint64_array(abd_to_buf(zio->io_abd), zio->io_size);
2305fa9e4066Sahrens 
2306770499e1SDan Kimmel 	ASSERT3P(abd_to_buf(zio->io_abd), ==, gn->gn_gbh);
2307e14bb325SJeff Bonwick 	ASSERT(zio->io_size == SPA_GANGBLOCKSIZE);
23086e1f5caaSNeil Perrin 	ASSERT(gn->gn_gbh->zg_tail.zec_magic == ZEC_MAGIC);
2309e05725b1Sbonwick 
2310770499e1SDan Kimmel 	abd_put(zio->io_abd);
2311770499e1SDan Kimmel 
2312e14bb325SJeff Bonwick 	for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
2313e14bb325SJeff Bonwick 		blkptr_t *gbp = &gn->gn_gbh->zg_blkptr[g];
2314e14bb325SJeff Bonwick 		if (!BP_IS_GANG(gbp))
2315e14bb325SJeff Bonwick 			continue;
2316f5383399SBill Moore 		zio_gang_tree_assemble(gio, gbp, &gn->gn_child[g]);
2317e14bb325SJeff Bonwick 	}
2318fa9e4066Sahrens }
2319fa9e4066Sahrens 
2320e14bb325SJeff Bonwick static void
2321770499e1SDan Kimmel zio_gang_tree_issue(zio_t *pio, zio_gang_node_t *gn, blkptr_t *bp, abd_t *data,
2322770499e1SDan Kimmel     uint64_t offset)
2323fa9e4066Sahrens {
2324f5383399SBill Moore 	zio_t *gio = pio->io_gang_leader;
2325e14bb325SJeff Bonwick 	zio_t *zio;
2326fa9e4066Sahrens 
2327e14bb325SJeff Bonwick 	ASSERT(BP_IS_GANG(bp) == !!gn);
2328f5383399SBill Moore 	ASSERT(BP_GET_CHECKSUM(bp) == BP_GET_CHECKSUM(gio->io_bp));
2329f5383399SBill Moore 	ASSERT(BP_GET_LSIZE(bp) == BP_GET_PSIZE(bp) || gn == gio->io_gang_tree);
2330fa9e4066Sahrens 
2331e14bb325SJeff Bonwick 	/*
2332e14bb325SJeff Bonwick 	 * If you're a gang header, your data is in gn->gn_gbh.
2333e14bb325SJeff Bonwick 	 * If you're a gang member, your data is in 'data' and gn == NULL.
2334e14bb325SJeff Bonwick 	 */
2335770499e1SDan Kimmel 	zio = zio_gang_issue_func[gio->io_type](pio, bp, gn, data, offset);
2336fa9e4066Sahrens 
2337e14bb325SJeff Bonwick 	if (gn != NULL) {
23386e1f5caaSNeil Perrin 		ASSERT(gn->gn_gbh->zg_tail.zec_magic == ZEC_MAGIC);
2339fa9e4066Sahrens 
2340e14bb325SJeff Bonwick 		for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
2341e14bb325SJeff Bonwick 			blkptr_t *gbp = &gn->gn_gbh->zg_blkptr[g];
2342e14bb325SJeff Bonwick 			if (BP_IS_HOLE(gbp))
2343e14bb325SJeff Bonwick 				continue;
2344770499e1SDan Kimmel 			zio_gang_tree_issue(zio, gn->gn_child[g], gbp, data,
2345770499e1SDan Kimmel 			    offset);
2346770499e1SDan Kimmel 			offset += BP_GET_PSIZE(gbp);
2347e14bb325SJeff Bonwick 		}
2348fa9e4066Sahrens 	}
2349fa9e4066Sahrens 
2350f5383399SBill Moore 	if (gn == gio->io_gang_tree)
2351770499e1SDan Kimmel 		ASSERT3U(gio->io_size, ==, offset);
2352e05725b1Sbonwick 
2353e14bb325SJeff Bonwick 	if (zio != pio)
2354e14bb325SJeff Bonwick 		zio_nowait(zio);
2355fa9e4066Sahrens }
2356fa9e4066Sahrens 
2357e05725b1Sbonwick static int
2358e14bb325SJeff Bonwick zio_gang_assemble(zio_t *zio)
2359fa9e4066Sahrens {
2360e14bb325SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2361fa9e4066Sahrens 
2362f5383399SBill Moore 	ASSERT(BP_IS_GANG(bp) && zio->io_gang_leader == NULL);
2363f5383399SBill Moore 	ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
2364f5383399SBill Moore 
2365f5383399SBill Moore 	zio->io_gang_leader = zio;
2366fa9e4066Sahrens 
2367e14bb325SJeff Bonwick 	zio_gang_tree_assemble(zio, bp, &zio->io_gang_tree);
2368e05725b1Sbonwick 
2369e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
2370fa9e4066Sahrens }
2371fa9e4066Sahrens 
2372e05725b1Sbonwick static int
2373e14bb325SJeff Bonwick zio_gang_issue(zio_t *zio)
2374fa9e4066Sahrens {
2375e14bb325SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2376fa9e4066Sahrens 
2377d6e1c446SGeorge Wilson 	if (zio_wait_for_children(zio, ZIO_CHILD_GANG_BIT, ZIO_WAIT_DONE)) {
2378e14bb325SJeff Bonwick 		return (ZIO_PIPELINE_STOP);
2379d6e1c446SGeorge Wilson 	}
2380fa9e4066Sahrens 
2381f5383399SBill Moore 	ASSERT(BP_IS_GANG(bp) && zio->io_gang_leader == zio);
2382f5383399SBill Moore 	ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
2383fa9e4066Sahrens 
2384e14bb325SJeff Bonwick 	if (zio->io_child_error[ZIO_CHILD_GANG] == 0)
2385770499e1SDan Kimmel 		zio_gang_tree_issue(zio, zio->io_gang_tree, bp, zio->io_abd,
2386770499e1SDan Kimmel 		    0);
2387e14bb325SJeff Bonwick 	else
2388f5383399SBill Moore 		zio_gang_tree_free(&zio->io_gang_tree);
2389fa9e4066Sahrens 
2390e14bb325SJeff Bonwick 	zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
2391e05725b1Sbonwick 
2392e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
2393fa9e4066Sahrens }
2394fa9e4066Sahrens 
2395fa9e4066Sahrens static void
2396e14bb325SJeff Bonwick zio_write_gang_member_ready(zio_t *zio)
2397fa9e4066Sahrens {
2398a3f829aeSBill Moore 	zio_t *pio = zio_unique_parent(zio);
2399f5383399SBill Moore 	zio_t *gio = zio->io_gang_leader;
240044cd46caSbillm 	dva_t *cdva = zio->io_bp->blk_dva;
240144cd46caSbillm 	dva_t *pdva = pio->io_bp->blk_dva;
2402fa9e4066Sahrens 	uint64_t asize;
2403fa9e4066Sahrens 
2404e14bb325SJeff Bonwick 	if (BP_IS_HOLE(zio->io_bp))
2405e14bb325SJeff Bonwick 		return;
2406e14bb325SJeff Bonwick 
2407e14bb325SJeff Bonwick 	ASSERT(BP_IS_HOLE(&zio->io_bp_orig));
2408e14bb325SJeff Bonwick 
2409e14bb325SJeff Bonwick 	ASSERT(zio->io_child_type == ZIO_CHILD_GANG);
2410b24ab676SJeff Bonwick 	ASSERT3U(zio->io_prop.zp_copies, ==, gio->io_prop.zp_copies);
2411b24ab676SJeff Bonwick 	ASSERT3U(zio->io_prop.zp_copies, <=, BP_GET_NDVAS(zio->io_bp));
2412b24ab676SJeff Bonwick 	ASSERT3U(pio->io_prop.zp_copies, <=, BP_GET_NDVAS(pio->io_bp));
241344cd46caSbillm 	ASSERT3U(BP_GET_NDVAS(zio->io_bp), <=, BP_GET_NDVAS(pio->io_bp));
2414fa9e4066Sahrens 
2415fa9e4066Sahrens 	mutex_enter(&pio->io_lock);
2416e14bb325SJeff Bonwick 	for (int d = 0; d < BP_GET_NDVAS(zio->io_bp); d++) {
241744cd46caSbillm 		ASSERT(DVA_GET_GANG(&pdva[d]));
241844cd46caSbillm 		asize = DVA_GET_ASIZE(&pdva[d]);
241944cd46caSbillm 		asize += DVA_GET_ASIZE(&cdva[d]);
242044cd46caSbillm 		DVA_SET_ASIZE(&pdva[d], asize);
242144cd46caSbillm 	}
2422fa9e4066Sahrens 	mutex_exit(&pio->io_lock);
2423fa9e4066Sahrens }
2424fa9e4066Sahrens 
2425770499e1SDan Kimmel static void
2426770499e1SDan Kimmel zio_write_gang_done(zio_t *zio)
2427770499e1SDan Kimmel {
24287341a7deSBrad Lewis 	/*
24297341a7deSBrad Lewis 	 * The io_abd field will be NULL for a zio with no data.  The io_flags
24307341a7deSBrad Lewis 	 * will initially have the ZIO_FLAG_NODATA bit flag set, but we can't
24317341a7deSBrad Lewis 	 * check for it here as it is cleared in zio_ready.
24327341a7deSBrad Lewis 	 */
24337341a7deSBrad Lewis 	if (zio->io_abd != NULL)
24347341a7deSBrad Lewis 		abd_put(zio->io_abd);
2435770499e1SDan Kimmel }
2436770499e1SDan Kimmel 
24370a4e9518Sgw static int
2438e14bb325SJeff Bonwick zio_write_gang_block(zio_t *pio)
2439fa9e4066Sahrens {
2440e14bb325SJeff Bonwick 	spa_t *spa = pio->io_spa;
24410f7643c7SGeorge Wilson 	metaslab_class_t *mc = spa_normal_class(spa);
2442e14bb325SJeff Bonwick 	blkptr_t *bp = pio->io_bp;
2443f5383399SBill Moore 	zio_t *gio = pio->io_gang_leader;
2444e14bb325SJeff Bonwick 	zio_t *zio;
2445e14bb325SJeff Bonwick 	zio_gang_node_t *gn, **gnpp;
2446fa9e4066Sahrens 	zio_gbh_phys_t *gbh;
2447770499e1SDan Kimmel 	abd_t *gbh_abd;
2448e14bb325SJeff Bonwick 	uint64_t txg = pio->io_txg;
2449e14bb325SJeff Bonwick 	uint64_t resid = pio->io_size;
2450e14bb325SJeff Bonwick 	uint64_t lsize;
2451b24ab676SJeff Bonwick 	int copies = gio->io_prop.zp_copies;
2452b24ab676SJeff Bonwick 	int gbh_copies = MIN(copies + 1, spa_max_replication(spa));
2453e14bb325SJeff Bonwick 	zio_prop_t zp;
2454fa9e4066Sahrens 	int error;
24557341a7deSBrad Lewis 	boolean_t has_data = !(pio->io_flags & ZIO_FLAG_NODATA);
2456fa9e4066Sahrens 
2457eb633035STom Caputi 	/*
2458eb633035STom Caputi 	 * encrypted blocks need DVA[2] free so encrypted gang headers can't
2459eb633035STom Caputi 	 * have a third copy.
2460eb633035STom Caputi 	 */
2461eb633035STom Caputi 	if (gio->io_prop.zp_encrypt && gbh_copies >= SPA_DVAS_PER_BP)
2462eb633035STom Caputi 		gbh_copies = SPA_DVAS_PER_BP - 1;
2463eb633035STom Caputi 
24640f7643c7SGeorge Wilson 	int flags = METASLAB_HINTBP_FAVOR | METASLAB_GANG_HEADER;
24650f7643c7SGeorge Wilson 	if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
24660f7643c7SGeorge Wilson 		ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
24677341a7deSBrad Lewis 		ASSERT(has_data);
24680f7643c7SGeorge Wilson 
24690f7643c7SGeorge Wilson 		flags |= METASLAB_ASYNC_ALLOC;
2470e914ace2STim Schumacher 		VERIFY(zfs_refcount_held(&mc->mc_alloc_slots[pio->io_allocator],
2471f78cdc34SPaul Dagnelie 		    pio));
24720f7643c7SGeorge Wilson 
24730f7643c7SGeorge Wilson 		/*
24740f7643c7SGeorge Wilson 		 * The logical zio has already placed a reservation for
24750f7643c7SGeorge Wilson 		 * 'copies' allocation slots but gang blocks may require
24760f7643c7SGeorge Wilson 		 * additional copies. These additional copies
24770f7643c7SGeorge Wilson 		 * (i.e. gbh_copies - copies) are guaranteed to succeed
24780f7643c7SGeorge Wilson 		 * since metaslab_class_throttle_reserve() always allows
24790f7643c7SGeorge Wilson 		 * additional reservations for gang blocks.
24800f7643c7SGeorge Wilson 		 */
24810f7643c7SGeorge Wilson 		VERIFY(metaslab_class_throttle_reserve(mc, gbh_copies - copies,
2482f78cdc34SPaul Dagnelie 		    pio->io_allocator, pio, flags));
24830f7643c7SGeorge Wilson 	}
24840f7643c7SGeorge Wilson 
24850f7643c7SGeorge Wilson 	error = metaslab_alloc(spa, mc, SPA_GANGBLOCKSIZE,
24868363e80aSGeorge Wilson 	    bp, gbh_copies, txg, pio == gio ? NULL : gio->io_bp, flags,
2487f78cdc34SPaul Dagnelie 	    &pio->io_alloc_list, pio, pio->io_allocator);
2488e05725b1Sbonwick 	if (error) {
24890f7643c7SGeorge Wilson 		if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
24900f7643c7SGeorge Wilson 			ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
24917341a7deSBrad Lewis 			ASSERT(has_data);
24920f7643c7SGeorge Wilson 
24930f7643c7SGeorge Wilson 			/*
24940f7643c7SGeorge Wilson 			 * If we failed to allocate the gang block header then
24950f7643c7SGeorge Wilson 			 * we remove any additional allocation reservations that
24960f7643c7SGeorge Wilson 			 * we placed here. The original reservation will
24970f7643c7SGeorge Wilson 			 * be removed when the logical I/O goes to the ready
24980f7643c7SGeorge Wilson 			 * stage.
24990f7643c7SGeorge Wilson 			 */
25000f7643c7SGeorge Wilson 			metaslab_class_throttle_unreserve(mc,
2501f78cdc34SPaul Dagnelie 			    gbh_copies - copies, pio->io_allocator, pio);
25020f7643c7SGeorge Wilson 		}
2503e14bb325SJeff Bonwick 		pio->io_error = error;
2504e05725b1Sbonwick 		return (ZIO_PIPELINE_CONTINUE);
2505e05725b1Sbonwick 	}
2506fa9e4066Sahrens 
2507f5383399SBill Moore 	if (pio == gio) {
2508f5383399SBill Moore 		gnpp = &gio->io_gang_tree;
2509e14bb325SJeff Bonwick 	} else {
2510e14bb325SJeff Bonwick 		gnpp = pio->io_private;
2511e14bb325SJeff Bonwick 		ASSERT(pio->io_ready == zio_write_gang_member_ready);
2512fa9e4066Sahrens 	}
2513fa9e4066Sahrens 
2514e14bb325SJeff Bonwick 	gn = zio_gang_node_alloc(gnpp);
2515e14bb325SJeff Bonwick 	gbh = gn->gn_gbh;
2516e14bb325SJeff Bonwick 	bzero(gbh, SPA_GANGBLOCKSIZE);
2517770499e1SDan Kimmel 	gbh_abd = abd_get_from_buf(gbh, SPA_GANGBLOCKSIZE);
2518fa9e4066Sahrens 
2519e14bb325SJeff Bonwick 	/*
2520e14bb325SJeff Bonwick 	 * Create the gang header.
2521e14bb325SJeff Bonwick 	 */
2522770499e1SDan Kimmel 	zio = zio_rewrite(pio, spa, txg, bp, gbh_abd, SPA_GANGBLOCKSIZE,
2523770499e1SDan Kimmel 	    zio_write_gang_done, NULL, pio->io_priority,
2524770499e1SDan Kimmel 	    ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
2525fa9e4066Sahrens 
2526e14bb325SJeff Bonwick 	/*
2527e14bb325SJeff Bonwick 	 * Create and nowait the gang children.
2528e14bb325SJeff Bonwick 	 */
2529e14bb325SJeff Bonwick 	for (int g = 0; resid != 0; resid -= lsize, g++) {
2530e14bb325SJeff Bonwick 		lsize = P2ROUNDUP(resid / (SPA_GBH_NBLKPTRS - g),
2531e14bb325SJeff Bonwick 		    SPA_MINBLOCKSIZE);
2532e14bb325SJeff Bonwick 		ASSERT(lsize >= SPA_MINBLOCKSIZE && lsize <= resid);
2533e14bb325SJeff Bonwick 
2534f5383399SBill Moore 		zp.zp_checksum = gio->io_prop.zp_checksum;
2535e14bb325SJeff Bonwick 		zp.zp_compress = ZIO_COMPRESS_OFF;
2536e14bb325SJeff Bonwick 		zp.zp_type = DMU_OT_NONE;
2537e14bb325SJeff Bonwick 		zp.zp_level = 0;
2538b24ab676SJeff Bonwick 		zp.zp_copies = gio->io_prop.zp_copies;
253980901aeaSGeorge Wilson 		zp.zp_dedup = B_FALSE;
254080901aeaSGeorge Wilson 		zp.zp_dedup_verify = B_FALSE;
254180901aeaSGeorge Wilson 		zp.zp_nopwrite = B_FALSE;
2542eb633035STom Caputi 		zp.zp_encrypt = gio->io_prop.zp_encrypt;
2543eb633035STom Caputi 		zp.zp_byteorder = gio->io_prop.zp_byteorder;
2544eb633035STom Caputi 		bzero(zp.zp_salt, ZIO_DATA_SALT_LEN);
2545eb633035STom Caputi 		bzero(zp.zp_iv, ZIO_DATA_IV_LEN);
2546eb633035STom Caputi 		bzero(zp.zp_mac, ZIO_DATA_MAC_LEN);
2547e14bb325SJeff Bonwick 
25480f7643c7SGeorge Wilson 		zio_t *cio = zio_write(zio, spa, txg, &gbh->zg_blkptr[g],
25497341a7deSBrad Lewis 		    has_data ? abd_get_offset(pio->io_abd, pio->io_size -
25507341a7deSBrad Lewis 		    resid) : NULL, lsize, lsize, &zp,
25517341a7deSBrad Lewis 		    zio_write_gang_member_ready, NULL, NULL,
2552770499e1SDan Kimmel 		    zio_write_gang_done, &gn->gn_child[g], pio->io_priority,
25530f7643c7SGeorge Wilson 		    ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
25540f7643c7SGeorge Wilson 
25550f7643c7SGeorge Wilson 		if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
25560f7643c7SGeorge Wilson 			ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
25577341a7deSBrad Lewis 			ASSERT(has_data);
25580f7643c7SGeorge Wilson 
25590f7643c7SGeorge Wilson 			/*
25600f7643c7SGeorge Wilson 			 * Gang children won't throttle but we should
25610f7643c7SGeorge Wilson 			 * account for their work, so reserve an allocation
25620f7643c7SGeorge Wilson 			 * slot for them here.
25630f7643c7SGeorge Wilson 			 */
25640f7643c7SGeorge Wilson 			VERIFY(metaslab_class_throttle_reserve(mc,
2565f78cdc34SPaul Dagnelie 			    zp.zp_copies, cio->io_allocator, cio, flags));
25660f7643c7SGeorge Wilson 		}
25670f7643c7SGeorge Wilson 		zio_nowait(cio);
2568e14bb325SJeff Bonwick 	}
2569e05725b1Sbonwick 
257044cd46caSbillm 	/*
2571e14bb325SJeff Bonwick 	 * Set pio's pipeline to just wait for zio to finish.
257244cd46caSbillm 	 */
2573e14bb325SJeff Bonwick 	pio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
2574e14bb325SJeff Bonwick 
2575e14bb325SJeff Bonwick 	zio_nowait(zio);
2576e14bb325SJeff Bonwick 
2577e14bb325SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
2578fa9e4066Sahrens }
2579fa9e4066Sahrens 
258080901aeaSGeorge Wilson /*
258145818ee1SMatthew Ahrens  * The zio_nop_write stage in the pipeline determines if allocating a
258245818ee1SMatthew Ahrens  * new bp is necessary.  The nopwrite feature can handle writes in
258345818ee1SMatthew Ahrens  * either syncing or open context (i.e. zil writes) and as a result is
258445818ee1SMatthew Ahrens  * mutually exclusive with dedup.
258545818ee1SMatthew Ahrens  *
258645818ee1SMatthew Ahrens  * By leveraging a cryptographically secure checksum, such as SHA256, we
258745818ee1SMatthew Ahrens  * can compare the checksums of the new data and the old to determine if
258845818ee1SMatthew Ahrens  * allocating a new block is required.  Note that our requirements for
258945818ee1SMatthew Ahrens  * cryptographic strength are fairly weak: there can't be any accidental
259045818ee1SMatthew Ahrens  * hash collisions, but we don't need to be secure against intentional
259145818ee1SMatthew Ahrens  * (malicious) collisions.  To trigger a nopwrite, you have to be able
259245818ee1SMatthew Ahrens  * to write the file to begin with, and triggering an incorrect (hash
259345818ee1SMatthew Ahrens  * collision) nopwrite is no worse than simply writing to the file.
259445818ee1SMatthew Ahrens  * That said, there are no known attacks against the checksum algorithms
259545818ee1SMatthew Ahrens  * used for nopwrite, assuming that the salt and the checksums
259645818ee1SMatthew Ahrens  * themselves remain secret.
259780901aeaSGeorge Wilson  */
259880901aeaSGeorge Wilson static int
259980901aeaSGeorge Wilson zio_nop_write(zio_t *zio)
260080901aeaSGeorge Wilson {
260180901aeaSGeorge Wilson 	blkptr_t *bp = zio->io_bp;
260280901aeaSGeorge Wilson 	blkptr_t *bp_orig = &zio->io_bp_orig;
260380901aeaSGeorge Wilson 	zio_prop_t *zp = &zio->io_prop;
260480901aeaSGeorge Wilson 
260580901aeaSGeorge Wilson 	ASSERT(BP_GET_LEVEL(bp) == 0);
260680901aeaSGeorge Wilson 	ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REWRITE));
260780901aeaSGeorge Wilson 	ASSERT(zp->zp_nopwrite);
260880901aeaSGeorge Wilson 	ASSERT(!zp->zp_dedup);
260980901aeaSGeorge Wilson 	ASSERT(zio->io_bp_override == NULL);
261080901aeaSGeorge Wilson 	ASSERT(IO_IS_ALLOCATING(zio));
261180901aeaSGeorge Wilson 
261280901aeaSGeorge Wilson 	/*
261380901aeaSGeorge Wilson 	 * Check to see if the original bp and the new bp have matching
261480901aeaSGeorge Wilson 	 * characteristics (i.e. same checksum, compression algorithms, etc).
261580901aeaSGeorge Wilson 	 * If they don't then just continue with the pipeline which will
261680901aeaSGeorge Wilson 	 * allocate a new bp.
261780901aeaSGeorge Wilson 	 */
261880901aeaSGeorge Wilson 	if (BP_IS_HOLE(bp_orig) ||
261945818ee1SMatthew Ahrens 	    !(zio_checksum_table[BP_GET_CHECKSUM(bp)].ci_flags &
262045818ee1SMatthew Ahrens 	    ZCHECKSUM_FLAG_NOPWRITE) ||
2621eb633035STom Caputi 	    BP_IS_ENCRYPTED(bp) || BP_IS_ENCRYPTED(bp_orig) ||
262280901aeaSGeorge Wilson 	    BP_GET_CHECKSUM(bp) != BP_GET_CHECKSUM(bp_orig) ||
262380901aeaSGeorge Wilson 	    BP_GET_COMPRESS(bp) != BP_GET_COMPRESS(bp_orig) ||
262480901aeaSGeorge Wilson 	    BP_GET_DEDUP(bp) != BP_GET_DEDUP(bp_orig) ||
262580901aeaSGeorge Wilson 	    zp->zp_copies != BP_GET_NDVAS(bp_orig))
262680901aeaSGeorge Wilson 		return (ZIO_PIPELINE_CONTINUE);
262780901aeaSGeorge Wilson 
262880901aeaSGeorge Wilson 	/*
262980901aeaSGeorge Wilson 	 * If the checksums match then reset the pipeline so that we
263080901aeaSGeorge Wilson 	 * avoid allocating a new bp and issuing any I/O.
263180901aeaSGeorge Wilson 	 */
263280901aeaSGeorge Wilson 	if (ZIO_CHECKSUM_EQUAL(bp->blk_cksum, bp_orig->blk_cksum)) {
263345818ee1SMatthew Ahrens 		ASSERT(zio_checksum_table[zp->zp_checksum].ci_flags &
263445818ee1SMatthew Ahrens 		    ZCHECKSUM_FLAG_NOPWRITE);
263580901aeaSGeorge Wilson 		ASSERT3U(BP_GET_PSIZE(bp), ==, BP_GET_PSIZE(bp_orig));
263680901aeaSGeorge Wilson 		ASSERT3U(BP_GET_LSIZE(bp), ==, BP_GET_LSIZE(bp_orig));
263780901aeaSGeorge Wilson 		ASSERT(zp->zp_compress != ZIO_COMPRESS_OFF);
263880901aeaSGeorge Wilson 		ASSERT(bcmp(&bp->blk_prop, &bp_orig->blk_prop,
263980901aeaSGeorge Wilson 		    sizeof (uint64_t)) == 0);
264080901aeaSGeorge Wilson 
264180901aeaSGeorge Wilson 		*bp = *bp_orig;
264280901aeaSGeorge Wilson 		zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
264380901aeaSGeorge Wilson 		zio->io_flags |= ZIO_FLAG_NOPWRITE;
264480901aeaSGeorge Wilson 	}
264580901aeaSGeorge Wilson 
264680901aeaSGeorge Wilson 	return (ZIO_PIPELINE_CONTINUE);
264780901aeaSGeorge Wilson }
264880901aeaSGeorge Wilson 
2649fa9e4066Sahrens /*
2650fa9e4066Sahrens  * ==========================================================================
2651b24ab676SJeff Bonwick  * Dedup
2652fa9e4066Sahrens  * ==========================================================================
2653fa9e4066Sahrens  */
2654b24ab676SJeff Bonwick static void
2655b24ab676SJeff Bonwick zio_ddt_child_read_done(zio_t *zio)
2656b24ab676SJeff Bonwick {
2657b24ab676SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2658b24ab676SJeff Bonwick 	ddt_entry_t *dde = zio->io_private;
2659b24ab676SJeff Bonwick 	ddt_phys_t *ddp;
2660b24ab676SJeff Bonwick 	zio_t *pio = zio_unique_parent(zio);
2661b24ab676SJeff Bonwick 
2662b24ab676SJeff Bonwick 	mutex_enter(&pio->io_lock);
2663b24ab676SJeff Bonwick 	ddp = ddt_phys_select(dde, bp);
2664b24ab676SJeff Bonwick 	if (zio->io_error == 0)
2665b24ab676SJeff Bonwick 		ddt_phys_clear(ddp);	/* this ddp doesn't need repair */
2666770499e1SDan Kimmel 
2667770499e1SDan Kimmel 	if (zio->io_error == 0 && dde->dde_repair_abd == NULL)
2668770499e1SDan Kimmel 		dde->dde_repair_abd = zio->io_abd;
2669b24ab676SJeff Bonwick 	else
2670770499e1SDan Kimmel 		abd_free(zio->io_abd);
2671b24ab676SJeff Bonwick 	mutex_exit(&pio->io_lock);
2672b24ab676SJeff Bonwick }
2673b24ab676SJeff Bonwick 
2674b24ab676SJeff Bonwick static int
2675b24ab676SJeff Bonwick zio_ddt_read_start(zio_t *zio)
2676b24ab676SJeff Bonwick {
2677b24ab676SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2678b24ab676SJeff Bonwick 
2679b24ab676SJeff Bonwick 	ASSERT(BP_GET_DEDUP(bp));
2680b24ab676SJeff Bonwick 	ASSERT(BP_GET_PSIZE(bp) == zio->io_size);
2681b24ab676SJeff Bonwick 	ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
2682b24ab676SJeff Bonwick 
2683b24ab676SJeff Bonwick 	if (zio->io_child_error[ZIO_CHILD_DDT]) {
2684b24ab676SJeff Bonwick 		ddt_t *ddt = ddt_select(zio->io_spa, bp);
2685b24ab676SJeff Bonwick 		ddt_entry_t *dde = ddt_repair_start(ddt, bp);
2686b24ab676SJeff Bonwick 		ddt_phys_t *ddp = dde->dde_phys;
2687b24ab676SJeff Bonwick 		ddt_phys_t *ddp_self = ddt_phys_select(dde, bp);
2688b24ab676SJeff Bonwick 		blkptr_t blk;
2689b24ab676SJeff Bonwick 
2690b24ab676SJeff Bonwick 		ASSERT(zio->io_vsd == NULL);
2691b24ab676SJeff Bonwick 		zio->io_vsd = dde;
2692b24ab676SJeff Bonwick 
2693b24ab676SJeff Bonwick 		if (ddp_self == NULL)
2694b24ab676SJeff Bonwick 			return (ZIO_PIPELINE_CONTINUE);
2695b24ab676SJeff Bonwick 
2696b24ab676SJeff Bonwick 		for (int p = 0; p < DDT_PHYS_TYPES; p++, ddp++) {
2697b24ab676SJeff Bonwick 			if (ddp->ddp_phys_birth == 0 || ddp == ddp_self)
2698b24ab676SJeff Bonwick 				continue;
2699bbfd46c4SJeff Bonwick 			ddt_bp_create(ddt->ddt_checksum, &dde->dde_key, ddp,
2700bbfd46c4SJeff Bonwick 			    &blk);
2701b24ab676SJeff Bonwick 			zio_nowait(zio_read(zio, zio->io_spa, &blk,
2702770499e1SDan Kimmel 			    abd_alloc_for_io(zio->io_size, B_TRUE),
2703770499e1SDan Kimmel 			    zio->io_size, zio_ddt_child_read_done, dde,
2704770499e1SDan Kimmel 			    zio->io_priority, ZIO_DDT_CHILD_FLAGS(zio) |
2705770499e1SDan Kimmel 			    ZIO_FLAG_DONT_PROPAGATE, &zio->io_bookmark));
2706b24ab676SJeff Bonwick 		}
2707b24ab676SJeff Bonwick 		return (ZIO_PIPELINE_CONTINUE);
2708b24ab676SJeff Bonwick 	}
2709b24ab676SJeff Bonwick 
2710b24ab676SJeff Bonwick 	zio_nowait(zio_read(zio, zio->io_spa, bp,
2711770499e1SDan Kimmel 	    zio->io_abd, zio->io_size, NULL, NULL, zio->io_priority,
2712b24ab676SJeff Bonwick 	    ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark));
2713b24ab676SJeff Bonwick 
2714b24ab676SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
2715b24ab676SJeff Bonwick }
2716e14bb325SJeff Bonwick 
2717b24ab676SJeff Bonwick static int
2718b24ab676SJeff Bonwick zio_ddt_read_done(zio_t *zio)
2719b24ab676SJeff Bonwick {
2720b24ab676SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2721b24ab676SJeff Bonwick 
2722d6e1c446SGeorge Wilson 	if (zio_wait_for_children(zio, ZIO_CHILD_DDT_BIT, ZIO_WAIT_DONE)) {
2723b24ab676SJeff Bonwick 		return (ZIO_PIPELINE_STOP);
2724d6e1c446SGeorge Wilson 	}
2725b24ab676SJeff Bonwick 
2726b24ab676SJeff Bonwick 	ASSERT(BP_GET_DEDUP(bp));
2727b24ab676SJeff Bonwick 	ASSERT(BP_GET_PSIZE(bp) == zio->io_size);
2728b24ab676SJeff Bonwick 	ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
2729b24ab676SJeff Bonwick 
2730b24ab676SJeff Bonwick 	if (zio->io_child_error[ZIO_CHILD_DDT]) {
2731b24ab676SJeff Bonwick 		ddt_t *ddt = ddt_select(zio->io_spa, bp);
2732b24ab676SJeff Bonwick 		ddt_entry_t *dde = zio->io_vsd;
2733b24ab676SJeff Bonwick 		if (ddt == NULL) {
2734b16da2e2SGeorge Wilson 			ASSERT(spa_load_state(zio->io_spa) != SPA_LOAD_NONE);
2735b24ab676SJeff Bonwick 			return (ZIO_PIPELINE_CONTINUE);
2736b24ab676SJeff Bonwick 		}
2737b24ab676SJeff Bonwick 		if (dde == NULL) {
2738b24ab676SJeff Bonwick 			zio->io_stage = ZIO_STAGE_DDT_READ_START >> 1;
273935a5a358SJonathan Adams 			zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_FALSE);
2740b24ab676SJeff Bonwick 			return (ZIO_PIPELINE_STOP);
2741b24ab676SJeff Bonwick 		}
2742770499e1SDan Kimmel 		if (dde->dde_repair_abd != NULL) {
2743770499e1SDan Kimmel 			abd_copy(zio->io_abd, dde->dde_repair_abd,
2744770499e1SDan Kimmel 			    zio->io_size);
2745b24ab676SJeff Bonwick 			zio->io_child_error[ZIO_CHILD_DDT] = 0;
2746b24ab676SJeff Bonwick 		}
2747b24ab676SJeff Bonwick 		ddt_repair_done(ddt, dde);
2748b24ab676SJeff Bonwick 		zio->io_vsd = NULL;
2749b24ab676SJeff Bonwick 	}
2750b24ab676SJeff Bonwick 
2751b24ab676SJeff Bonwick 	ASSERT(zio->io_vsd == NULL);
2752b24ab676SJeff Bonwick 
2753b24ab676SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
2754b24ab676SJeff Bonwick }
2755b24ab676SJeff Bonwick 
2756b24ab676SJeff Bonwick static boolean_t
2757b24ab676SJeff Bonwick zio_ddt_collision(zio_t *zio, ddt_t *ddt, ddt_entry_t *dde)
2758b24ab676SJeff Bonwick {
2759b24ab676SJeff Bonwick 	spa_t *spa = zio->io_spa;
2760eb633035STom Caputi 	boolean_t do_raw = !!(zio->io_flags & ZIO_FLAG_RAW);
27615602294fSDan Kimmel 
27625602294fSDan Kimmel 	/* We should never get a raw, override zio */
27635602294fSDan Kimmel 	ASSERT(!(zio->io_bp_override && do_raw));
2764b24ab676SJeff Bonwick 
2765b24ab676SJeff Bonwick 	/*
2766b24ab676SJeff Bonwick 	 * Note: we compare the original data, not the transformed data,
2767b24ab676SJeff Bonwick 	 * because when zio->io_bp is an override bp, we will not have
2768b24ab676SJeff Bonwick 	 * pushed the I/O transforms.  That's an important optimization
2769b24ab676SJeff Bonwick 	 * because otherwise we'd compress/encrypt all dmu_sync() data twice.
2770eb633035STom Caputi 	 * However, we should never get a raw, override zio so in these
2771eb633035STom Caputi 	 * cases we can compare the io_data directly. This is useful because
2772eb633035STom Caputi 	 * it allows us to do dedup verification even if we don't have access
2773eb633035STom Caputi 	 * to the original data (for instance, if the encryption keys aren't
2774eb633035STom Caputi 	 * loaded).
2775b24ab676SJeff Bonwick 	 */
2776eb633035STom Caputi 
2777b24ab676SJeff Bonwick 	for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++) {
2778b24ab676SJeff Bonwick 		zio_t *lio = dde->dde_lead_zio[p];
2779b24ab676SJeff Bonwick 
2780eb633035STom Caputi 		if (lio != NULL && do_raw) {
2781eb633035STom Caputi 			return (lio->io_size != zio->io_size ||
2782eb633035STom Caputi 			    abd_cmp(zio->io_abd, lio->io_abd,
2783eb633035STom Caputi 			    zio->io_size) != 0);
2784eb633035STom Caputi 		} else if (lio != NULL) {
2785b24ab676SJeff Bonwick 			return (lio->io_orig_size != zio->io_orig_size ||
2786770499e1SDan Kimmel 			    abd_cmp(zio->io_orig_abd, lio->io_orig_abd,
2787b24ab676SJeff Bonwick 			    zio->io_orig_size) != 0);
2788b24ab676SJeff Bonwick 		}
2789b24ab676SJeff Bonwick 	}
2790b24ab676SJeff Bonwick 
2791b24ab676SJeff Bonwick 	for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++) {
2792b24ab676SJeff Bonwick 		ddt_phys_t *ddp = &dde->dde_phys[p];
2793b24ab676SJeff Bonwick 
2794eb633035STom Caputi 		if (ddp->ddp_phys_birth != 0 && do_raw) {
2795eb633035STom Caputi 			blkptr_t blk = *zio->io_bp;
2796eb633035STom Caputi 			uint64_t psize;
2797eb633035STom Caputi 			abd_t *tmpabd;
2798eb633035STom Caputi 			int error;
2799eb633035STom Caputi 
2800eb633035STom Caputi 			ddt_bp_fill(ddp, &blk, ddp->ddp_phys_birth);
2801eb633035STom Caputi 			psize = BP_GET_PSIZE(&blk);
2802eb633035STom Caputi 
2803eb633035STom Caputi 			if (psize != zio->io_size)
2804eb633035STom Caputi 				return (B_TRUE);
2805eb633035STom Caputi 
2806eb633035STom Caputi 			ddt_exit(ddt);
2807eb633035STom Caputi 
2808eb633035STom Caputi 			tmpabd = abd_alloc_for_io(psize, B_TRUE);
2809eb633035STom Caputi 
2810eb633035STom Caputi 			error = zio_wait(zio_read(NULL, spa, &blk, tmpabd,
2811eb633035STom Caputi 			    psize, NULL, NULL, ZIO_PRIORITY_SYNC_READ,
2812eb633035STom Caputi 			    ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE |
2813eb633035STom Caputi 			    ZIO_FLAG_RAW, &zio->io_bookmark));
2814eb633035STom Caputi 
2815eb633035STom Caputi 			if (error == 0) {
2816eb633035STom Caputi 				if (abd_cmp(tmpabd, zio->io_abd, psize) != 0)
2817eb633035STom Caputi 					error = SET_ERROR(ENOENT);
2818eb633035STom Caputi 			}
2819eb633035STom Caputi 
2820eb633035STom Caputi 			abd_free(tmpabd);
2821eb633035STom Caputi 			ddt_enter(ddt);
2822eb633035STom Caputi 			return (error != 0);
2823eb633035STom Caputi 		} else if (ddp->ddp_phys_birth != 0) {
2824b24ab676SJeff Bonwick 			arc_buf_t *abuf = NULL;
28257adb730bSGeorge Wilson 			arc_flags_t aflags = ARC_FLAG_WAIT;
28265602294fSDan Kimmel 			int zio_flags = ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE;
2827b24ab676SJeff Bonwick 			blkptr_t blk = *zio->io_bp;
2828b24ab676SJeff Bonwick 			int error;
2829b24ab676SJeff Bonwick 
2830b24ab676SJeff Bonwick 			ddt_bp_fill(ddp, &blk, ddp->ddp_phys_birth);
2831b24ab676SJeff Bonwick 
2832eb633035STom Caputi 			if (BP_GET_LSIZE(&blk) != zio->io_orig_size)
2833eb633035STom Caputi 				return (B_TRUE);
2834eb633035STom Caputi 
2835b24ab676SJeff Bonwick 			ddt_exit(ddt);
2836b24ab676SJeff Bonwick 
28375602294fSDan Kimmel 			/*
28385602294fSDan Kimmel 			 * Intuitively, it would make more sense to compare
2839770499e1SDan Kimmel 			 * io_abd than io_orig_abd in the raw case since you
28405602294fSDan Kimmel 			 * don't want to look at any transformations that have
28415602294fSDan Kimmel 			 * happened to the data. However, for raw I/Os the
2842770499e1SDan Kimmel 			 * data will actually be the same in io_abd and
2843770499e1SDan Kimmel 			 * io_orig_abd, so all we have to do is issue this as
28445602294fSDan Kimmel 			 * a raw ARC read.
28455602294fSDan Kimmel 			 */
28465602294fSDan Kimmel 			if (do_raw) {
28475602294fSDan Kimmel 				zio_flags |= ZIO_FLAG_RAW;
28485602294fSDan Kimmel 				ASSERT3U(zio->io_size, ==, zio->io_orig_size);
2849770499e1SDan Kimmel 				ASSERT0(abd_cmp(zio->io_abd, zio->io_orig_abd,
28505602294fSDan Kimmel 				    zio->io_size));
28515602294fSDan Kimmel 				ASSERT3P(zio->io_transform_stack, ==, NULL);
28525602294fSDan Kimmel 			}
28535602294fSDan Kimmel 
28541b912ec7SGeorge Wilson 			error = arc_read(NULL, spa, &blk,
2855b24ab676SJeff Bonwick 			    arc_getbuf_func, &abuf, ZIO_PRIORITY_SYNC_READ,
28565602294fSDan Kimmel 			    zio_flags, &aflags, &zio->io_bookmark);
2857b24ab676SJeff Bonwick 
2858b24ab676SJeff Bonwick 			if (error == 0) {
2859eb633035STom Caputi 				if (abd_cmp_buf(zio->io_orig_abd, abuf->b_data,
2860b24ab676SJeff Bonwick 				    zio->io_orig_size) != 0)
2861eb633035STom Caputi 					error = SET_ERROR(ENOENT);
2862dcbf3bd6SGeorge Wilson 				arc_buf_destroy(abuf, &abuf);
2863b24ab676SJeff Bonwick 			}
2864b24ab676SJeff Bonwick 
2865b24ab676SJeff Bonwick 			ddt_enter(ddt);
2866b24ab676SJeff Bonwick 			return (error != 0);
2867b24ab676SJeff Bonwick 		}
2868b24ab676SJeff Bonwick 	}
2869b24ab676SJeff Bonwick 
2870b24ab676SJeff Bonwick 	return (B_FALSE);
2871b24ab676SJeff Bonwick }
2872b24ab676SJeff Bonwick 
2873b24ab676SJeff Bonwick static void
2874b24ab676SJeff Bonwick zio_ddt_child_write_ready(zio_t *zio)
2875b24ab676SJeff Bonwick {
2876b24ab676SJeff Bonwick 	int p = zio->io_prop.zp_copies;
2877b24ab676SJeff Bonwick 	ddt_t *ddt = ddt_select(zio->io_spa, zio->io_bp);
2878b24ab676SJeff Bonwick 	ddt_entry_t *dde = zio->io_private;
2879b24ab676SJeff Bonwick 	ddt_phys_t *ddp = &dde->dde_phys[p];
2880b24ab676SJeff Bonwick 	zio_t *pio;
2881b24ab676SJeff Bonwick 
2882b24ab676SJeff Bonwick 	if (zio->io_error)
2883b24ab676SJeff Bonwick 		return;
2884b24ab676SJeff Bonwick 
2885b24ab676SJeff Bonwick 	ddt_enter(ddt);
2886b24ab676SJeff Bonwick 
2887b24ab676SJeff Bonwick 	ASSERT(dde->dde_lead_zio[p] == zio);
2888b24ab676SJeff Bonwick 
2889b24ab676SJeff Bonwick 	ddt_phys_fill(ddp, zio->io_bp);
2890b24ab676SJeff Bonwick 
28910f7643c7SGeorge Wilson 	zio_link_t *zl = NULL;
28920f7643c7SGeorge Wilson 	while ((pio = zio_walk_parents(zio, &zl)) != NULL)
2893b24ab676SJeff Bonwick 		ddt_bp_fill(ddp, pio->io_bp, zio->io_txg);
2894b24ab676SJeff Bonwick 
2895b24ab676SJeff Bonwick 	ddt_exit(ddt);
2896b24ab676SJeff Bonwick }
2897b24ab676SJeff Bonwick 
2898b24ab676SJeff Bonwick static void
2899b24ab676SJeff Bonwick zio_ddt_child_write_done(zio_t *zio)
2900b24ab676SJeff Bonwick {
2901b24ab676SJeff Bonwick 	int p = zio->io_prop.zp_copies;
2902b24ab676SJeff Bonwick 	ddt_t *ddt = ddt_select(zio->io_spa, zio->io_bp);
2903b24ab676SJeff Bonwick 	ddt_entry_t *dde = zio->io_private;
2904b24ab676SJeff Bonwick 	ddt_phys_t *ddp = &dde->dde_phys[p];
2905b24ab676SJeff Bonwick 
2906b24ab676SJeff Bonwick 	ddt_enter(ddt);
2907b24ab676SJeff Bonwick 
2908b24ab676SJeff Bonwick 	ASSERT(ddp->ddp_refcnt == 0);
2909b24ab676SJeff Bonwick 	ASSERT(dde->dde_lead_zio[p] == zio);
2910b24ab676SJeff Bonwick 	dde->dde_lead_zio[p] = NULL;
2911b24ab676SJeff Bonwick 
2912b24ab676SJeff Bonwick 	if (zio->io_error == 0) {
29130f7643c7SGeorge Wilson 		zio_link_t *zl = NULL;
29140f7643c7SGeorge Wilson 		while (zio_walk_parents(zio, &zl) != NULL)
2915b24ab676SJeff Bonwick 			ddt_phys_addref(ddp);
2916b24ab676SJeff Bonwick 	} else {
2917b24ab676SJeff Bonwick 		ddt_phys_clear(ddp);
2918b24ab676SJeff Bonwick 	}
2919b24ab676SJeff Bonwick 
2920b24ab676SJeff Bonwick 	ddt_exit(ddt);
2921b24ab676SJeff Bonwick }
2922b24ab676SJeff Bonwick 
2923b24ab676SJeff Bonwick static void
2924b24ab676SJeff Bonwick zio_ddt_ditto_write_done(zio_t *zio)
2925b24ab676SJeff Bonwick {
2926b24ab676SJeff Bonwick 	int p = DDT_PHYS_DITTO;
2927b24ab676SJeff Bonwick 	zio_prop_t *zp = &zio->io_prop;
2928b24ab676SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2929b24ab676SJeff Bonwick 	ddt_t *ddt = ddt_select(zio->io_spa, bp);
2930b24ab676SJeff Bonwick 	ddt_entry_t *dde = zio->io_private;
2931b24ab676SJeff Bonwick 	ddt_phys_t *ddp = &dde->dde_phys[p];
2932b24ab676SJeff Bonwick 	ddt_key_t *ddk = &dde->dde_key;
2933b24ab676SJeff Bonwick 
2934b24ab676SJeff Bonwick 	ddt_enter(ddt);
2935b24ab676SJeff Bonwick 
2936b24ab676SJeff Bonwick 	ASSERT(ddp->ddp_refcnt == 0);
2937b24ab676SJeff Bonwick 	ASSERT(dde->dde_lead_zio[p] == zio);
2938b24ab676SJeff Bonwick 	dde->dde_lead_zio[p] = NULL;
2939b24ab676SJeff Bonwick 
2940b24ab676SJeff Bonwick 	if (zio->io_error == 0) {
2941b24ab676SJeff Bonwick 		ASSERT(ZIO_CHECKSUM_EQUAL(bp->blk_cksum, ddk->ddk_cksum));
2942b24ab676SJeff Bonwick 		ASSERT(zp->zp_copies < SPA_DVAS_PER_BP);
2943b24ab676SJeff Bonwick 		ASSERT(zp->zp_copies == BP_GET_NDVAS(bp) - BP_IS_GANG(bp));
2944b24ab676SJeff Bonwick 		if (ddp->ddp_phys_birth != 0)
2945b24ab676SJeff Bonwick 			ddt_phys_free(ddt, ddk, ddp, zio->io_txg);
2946b24ab676SJeff Bonwick 		ddt_phys_fill(ddp, bp);
2947b24ab676SJeff Bonwick 	}
2948b24ab676SJeff Bonwick 
2949b24ab676SJeff Bonwick 	ddt_exit(ddt);
2950b24ab676SJeff Bonwick }
2951b24ab676SJeff Bonwick 
2952b24ab676SJeff Bonwick static int
2953b24ab676SJeff Bonwick zio_ddt_write(zio_t *zio)
2954b24ab676SJeff Bonwick {
2955b24ab676SJeff Bonwick 	spa_t *spa = zio->io_spa;
2956b24ab676SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
2957b24ab676SJeff Bonwick 	uint64_t txg = zio->io_txg;
2958b24ab676SJeff Bonwick 	zio_prop_t *zp = &zio->io_prop;
2959b24ab676SJeff Bonwick 	int p = zp->zp_copies;
2960b24ab676SJeff Bonwick 	int ditto_copies;
2961b24ab676SJeff Bonwick 	zio_t *cio = NULL;
2962b24ab676SJeff Bonwick 	zio_t *dio = NULL;
2963b24ab676SJeff Bonwick 	ddt_t *ddt = ddt_select(spa, bp);
2964b24ab676SJeff Bonwick 	ddt_entry_t *dde;
2965b24ab676SJeff Bonwick 	ddt_phys_t *ddp;
2966b24ab676SJeff Bonwick 
2967b24ab676SJeff Bonwick 	ASSERT(BP_GET_DEDUP(bp));
2968b24ab676SJeff Bonwick 	ASSERT(BP_GET_CHECKSUM(bp) == zp->zp_checksum);
2969b24ab676SJeff Bonwick 	ASSERT(BP_IS_HOLE(bp) || zio->io_bp_override);
29705602294fSDan Kimmel 	ASSERT(!(zio->io_bp_override && (zio->io_flags & ZIO_FLAG_RAW)));
2971b24ab676SJeff Bonwick 
2972b24ab676SJeff Bonwick 	ddt_enter(ddt);
2973b24ab676SJeff Bonwick 	dde = ddt_lookup(ddt, bp, B_TRUE);
2974b24ab676SJeff Bonwick 	ddp = &dde->dde_phys[p];
2975b24ab676SJeff Bonwick 
2976b24ab676SJeff Bonwick 	if (zp->zp_dedup_verify && zio_ddt_collision(zio, ddt, dde)) {
2977b24ab676SJeff Bonwick 		/*
2978b24ab676SJeff Bonwick 		 * If we're using a weak checksum, upgrade to a strong checksum
2979b24ab676SJeff Bonwick 		 * and try again.  If we're already using a strong checksum,
2980b24ab676SJeff Bonwick 		 * we can't resolve it, so just convert to an ordinary write.
2981b24ab676SJeff Bonwick 		 * (And automatically e-mail a paper to Nature?)
2982b24ab676SJeff Bonwick 		 */
298345818ee1SMatthew Ahrens 		if (!(zio_checksum_table[zp->zp_checksum].ci_flags &
298445818ee1SMatthew Ahrens 		    ZCHECKSUM_FLAG_DEDUP)) {
2985b24ab676SJeff Bonwick 			zp->zp_checksum = spa_dedup_checksum(spa);
2986b24ab676SJeff Bonwick 			zio_pop_transforms(zio);
2987b24ab676SJeff Bonwick 			zio->io_stage = ZIO_STAGE_OPEN;
2988b24ab676SJeff Bonwick 			BP_ZERO(bp);
2989b24ab676SJeff Bonwick 		} else {
299080901aeaSGeorge Wilson 			zp->zp_dedup = B_FALSE;
29915602294fSDan Kimmel 			BP_SET_DEDUP(bp, B_FALSE);
2992b24ab676SJeff Bonwick 		}
29935602294fSDan Kimmel 		ASSERT(!BP_GET_DEDUP(bp));
2994b24ab676SJeff Bonwick 		zio->io_pipeline = ZIO_WRITE_PIPELINE;
2995b24ab676SJeff Bonwick 		ddt_exit(ddt);
2996b24ab676SJeff Bonwick 		return (ZIO_PIPELINE_CONTINUE);
2997b24ab676SJeff Bonwick 	}
2998b24ab676SJeff Bonwick 
2999b24ab676SJeff Bonwick 	ditto_copies = ddt_ditto_copies_needed(ddt, dde, ddp);
3000b24ab676SJeff Bonwick 	ASSERT(ditto_copies < SPA_DVAS_PER_BP);
3001b24ab676SJeff Bonwick 
3002b24ab676SJeff Bonwick 	if (ditto_copies > ddt_ditto_copies_present(dde) &&
3003b24ab676SJeff Bonwick 	    dde->dde_lead_zio[DDT_PHYS_DITTO] == NULL) {
3004b24ab676SJeff Bonwick 		zio_prop_t czp = *zp;
3005b24ab676SJeff Bonwick 
3006b24ab676SJeff Bonwick 		czp.zp_copies = ditto_copies;
3007b24ab676SJeff Bonwick 
3008b24ab676SJeff Bonwick 		/*
3009b24ab676SJeff Bonwick 		 * If we arrived here with an override bp, we won't have run
3010b24ab676SJeff Bonwick 		 * the transform stack, so we won't have the data we need to
3011b24ab676SJeff Bonwick 		 * generate a child i/o.  So, toss the override bp and restart.
3012b24ab676SJeff Bonwick 		 * This is safe, because using the override bp is just an
3013b24ab676SJeff Bonwick 		 * optimization; and it's rare, so the cost doesn't matter.
3014b24ab676SJeff Bonwick 		 */
3015b24ab676SJeff Bonwick 		if (zio->io_bp_override) {
3016b24ab676SJeff Bonwick 			zio_pop_transforms(zio);
3017b24ab676SJeff Bonwick 			zio->io_stage = ZIO_STAGE_OPEN;
3018b24ab676SJeff Bonwick 			zio->io_pipeline = ZIO_WRITE_PIPELINE;
3019b24ab676SJeff Bonwick 			zio->io_bp_override = NULL;
3020b24ab676SJeff Bonwick 			BP_ZERO(bp);
3021b24ab676SJeff Bonwick 			ddt_exit(ddt);
3022b24ab676SJeff Bonwick 			return (ZIO_PIPELINE_CONTINUE);
3023b24ab676SJeff Bonwick 		}
3024b24ab676SJeff Bonwick 
3025770499e1SDan Kimmel 		dio = zio_write(zio, spa, txg, bp, zio->io_orig_abd,
30265602294fSDan Kimmel 		    zio->io_orig_size, zio->io_orig_size, &czp, NULL, NULL,
30278df0bcf0SPaul Dagnelie 		    NULL, zio_ddt_ditto_write_done, dde, zio->io_priority,
3028b24ab676SJeff Bonwick 		    ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark);
3029b24ab676SJeff Bonwick 
3030770499e1SDan Kimmel 		zio_push_transform(dio, zio->io_abd, zio->io_size, 0, NULL);
3031b24ab676SJeff Bonwick 		dde->dde_lead_zio[DDT_PHYS_DITTO] = dio;
3032b24ab676SJeff Bonwick 	}
3033b24ab676SJeff Bonwick 
3034b24ab676SJeff Bonwick 	if (ddp->ddp_phys_birth != 0 || dde->dde_lead_zio[p] != NULL) {
3035b24ab676SJeff Bonwick 		if (ddp->ddp_phys_birth != 0)
3036b24ab676SJeff Bonwick 			ddt_bp_fill(ddp, bp, txg);
3037b24ab676SJeff Bonwick 		if (dde->dde_lead_zio[p] != NULL)
3038b24ab676SJeff Bonwick 			zio_add_child(zio, dde->dde_lead_zio[p]);
3039b24ab676SJeff Bonwick 		else
3040b24ab676SJeff Bonwick 			ddt_phys_addref(ddp);
3041b24ab676SJeff Bonwick 	} else if (zio->io_bp_override) {
3042b24ab676SJeff Bonwick 		ASSERT(bp->blk_birth == txg);
3043b24ab676SJeff Bonwick 		ASSERT(BP_EQUAL(bp, zio->io_bp_override));
3044b24ab676SJeff Bonwick 		ddt_phys_fill(ddp, bp);
3045b24ab676SJeff Bonwick 		ddt_phys_addref(ddp);
3046b24ab676SJeff Bonwick 	} else {
3047770499e1SDan Kimmel 		cio = zio_write(zio, spa, txg, bp, zio->io_orig_abd,
30485602294fSDan Kimmel 		    zio->io_orig_size, zio->io_orig_size, zp,
30498df0bcf0SPaul Dagnelie 		    zio_ddt_child_write_ready, NULL, NULL,
3050b24ab676SJeff Bonwick 		    zio_ddt_child_write_done, dde, zio->io_priority,
3051b24ab676SJeff Bonwick 		    ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark);
3052b24ab676SJeff Bonwick 
3053770499e1SDan Kimmel 		zio_push_transform(cio, zio->io_abd, zio->io_size, 0, NULL);
3054b24ab676SJeff Bonwick 		dde->dde_lead_zio[p] = cio;
3055b24ab676SJeff Bonwick 	}
3056b24ab676SJeff Bonwick 
3057b24ab676SJeff Bonwick 	ddt_exit(ddt);
3058b24ab676SJeff Bonwick 
3059b24ab676SJeff Bonwick 	if (cio)
3060b24ab676SJeff Bonwick 		zio_nowait(cio);
3061b24ab676SJeff Bonwick 	if (dio)
3062b24ab676SJeff Bonwick 		zio_nowait(dio);
3063b24ab676SJeff Bonwick 
3064b24ab676SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
3065b24ab676SJeff Bonwick }
3066b24ab676SJeff Bonwick 
30673f9d6ad7SLin Ling ddt_entry_t *freedde; /* for debugging */
30683f9d6ad7SLin Ling 
3069b24ab676SJeff Bonwick static int
3070b24ab676SJeff Bonwick zio_ddt_free(zio_t *zio)
3071b24ab676SJeff Bonwick {
3072b24ab676SJeff Bonwick 	spa_t *spa = zio->io_spa;
3073b24ab676SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
3074b24ab676SJeff Bonwick 	ddt_t *ddt = ddt_select(spa, bp);
3075b24ab676SJeff Bonwick 	ddt_entry_t *dde;
3076b24ab676SJeff Bonwick 	ddt_phys_t *ddp;
3077b24ab676SJeff Bonwick 
3078b24ab676SJeff Bonwick 	ASSERT(BP_GET_DEDUP(bp));
3079b24ab676SJeff Bonwick 	ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
3080b24ab676SJeff Bonwick 
3081b24ab676SJeff Bonwick 	ddt_enter(ddt);
30823f9d6ad7SLin Ling 	freedde = dde = ddt_lookup(ddt, bp, B_TRUE);
3083b24ab676SJeff Bonwick 	ddp = ddt_phys_select(dde, bp);
3084b24ab676SJeff Bonwick 	ddt_phys_decref(ddp);
3085b24ab676SJeff Bonwick 	ddt_exit(ddt);
3086b24ab676SJeff Bonwick 
3087b24ab676SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
3088b24ab676SJeff Bonwick }
3089b24ab676SJeff Bonwick 
3090b24ab676SJeff Bonwick /*
3091b24ab676SJeff Bonwick  * ==========================================================================
3092b24ab676SJeff Bonwick  * Allocate and free blocks
3093b24ab676SJeff Bonwick  * ==========================================================================
3094b24ab676SJeff Bonwick  */
30950f7643c7SGeorge Wilson 
30960f7643c7SGeorge Wilson static zio_t *
3097f78cdc34SPaul Dagnelie zio_io_to_allocate(spa_t *spa, int allocator)
30980f7643c7SGeorge Wilson {
30990f7643c7SGeorge Wilson 	zio_t *zio;
31000f7643c7SGeorge Wilson 
3101f78cdc34SPaul Dagnelie 	ASSERT(MUTEX_HELD(&spa->spa_alloc_locks[allocator]));
31020f7643c7SGeorge Wilson 
3103f78cdc34SPaul Dagnelie 	zio = avl_first(&spa->spa_alloc_trees[allocator]);
31040f7643c7SGeorge Wilson 	if (zio == NULL)
31050f7643c7SGeorge Wilson 		return (NULL);
31060f7643c7SGeorge Wilson 
31070f7643c7SGeorge Wilson 	ASSERT(IO_IS_ALLOCATING(zio));
31080f7643c7SGeorge Wilson 
31090f7643c7SGeorge Wilson 	/*
31100f7643c7SGeorge Wilson 	 * Try to place a reservation for this zio. If we're unable to
31110f7643c7SGeorge Wilson 	 * reserve then we throttle.
31120f7643c7SGeorge Wilson 	 */
3113f78cdc34SPaul Dagnelie 	ASSERT3U(zio->io_allocator, ==, allocator);
3114663207adSDon Brady 	if (!metaslab_class_throttle_reserve(zio->io_metaslab_class,
3115f78cdc34SPaul Dagnelie 	    zio->io_prop.zp_copies, zio->io_allocator, zio, 0)) {
31160f7643c7SGeorge Wilson 		return (NULL);
31170f7643c7SGeorge Wilson 	}
31180f7643c7SGeorge Wilson 
3119f78cdc34SPaul Dagnelie 	avl_remove(&spa->spa_alloc_trees[allocator], zio);
31200f7643c7SGeorge Wilson 	ASSERT3U(zio->io_stage, <, ZIO_STAGE_DVA_ALLOCATE);
31210f7643c7SGeorge Wilson 
31220f7643c7SGeorge Wilson 	return (zio);
31230f7643c7SGeorge Wilson }
31240f7643c7SGeorge Wilson 
31250f7643c7SGeorge Wilson static int
31260f7643c7SGeorge Wilson zio_dva_throttle(zio_t *zio)
31270f7643c7SGeorge Wilson {
31280f7643c7SGeorge Wilson 	spa_t *spa = zio->io_spa;
31290f7643c7SGeorge Wilson 	zio_t *nio;
3130663207adSDon Brady 	metaslab_class_t *mc;
3131663207adSDon Brady 
3132663207adSDon Brady 	/* locate an appropriate allocation class */
3133663207adSDon Brady 	mc = spa_preferred_class(spa, zio->io_size, zio->io_prop.zp_type,
3134663207adSDon Brady 	    zio->io_prop.zp_level, zio->io_prop.zp_zpl_smallblk);
31350f7643c7SGeorge Wilson 
31360f7643c7SGeorge Wilson 	if (zio->io_priority == ZIO_PRIORITY_SYNC_WRITE ||
3137663207adSDon Brady 	    !mc->mc_alloc_throttle_enabled ||
31380f7643c7SGeorge Wilson 	    zio->io_child_type == ZIO_CHILD_GANG ||
31390f7643c7SGeorge Wilson 	    zio->io_flags & ZIO_FLAG_NODATA) {
31400f7643c7SGeorge Wilson 		return (ZIO_PIPELINE_CONTINUE);
31410f7643c7SGeorge Wilson 	}
31420f7643c7SGeorge Wilson 
31430f7643c7SGeorge Wilson 	ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
31440f7643c7SGeorge Wilson 
31450f7643c7SGeorge Wilson 	ASSERT3U(zio->io_queued_timestamp, >, 0);
31460f7643c7SGeorge Wilson 	ASSERT(zio->io_stage == ZIO_STAGE_DVA_THROTTLE);
31470f7643c7SGeorge Wilson 
3148f78cdc34SPaul Dagnelie 	zbookmark_phys_t *bm = &zio->io_bookmark;
3149f78cdc34SPaul Dagnelie 	/*
3150f78cdc34SPaul Dagnelie 	 * We want to try to use as many allocators as possible to help improve
3151f78cdc34SPaul Dagnelie 	 * performance, but we also want logically adjacent IOs to be physically
3152f78cdc34SPaul Dagnelie 	 * adjacent to improve sequential read performance. We chunk each object
3153f78cdc34SPaul Dagnelie 	 * into 2^20 block regions, and then hash based on the objset, object,
3154f78cdc34SPaul Dagnelie 	 * level, and region to accomplish both of these goals.
3155f78cdc34SPaul Dagnelie 	 */
3156f78cdc34SPaul Dagnelie 	zio->io_allocator = cityhash4(bm->zb_objset, bm->zb_object,
3157f78cdc34SPaul Dagnelie 	    bm->zb_level, bm->zb_blkid >> 20) % spa->spa_alloc_count;
3158f78cdc34SPaul Dagnelie 	mutex_enter(&spa->spa_alloc_locks[zio->io_allocator]);
31590f7643c7SGeorge Wilson 	ASSERT(zio->io_type == ZIO_TYPE_WRITE);
3160663207adSDon Brady 	zio->io_metaslab_class = mc;
3161f78cdc34SPaul Dagnelie 	avl_add(&spa->spa_alloc_trees[zio->io_allocator], zio);
3162663207adSDon Brady 	nio = zio_io_to_allocate(spa, zio->io_allocator);
3163f78cdc34SPaul Dagnelie 	mutex_exit(&spa->spa_alloc_locks[zio->io_allocator]);
31640f7643c7SGeorge Wilson 
31650f7643c7SGeorge Wilson 	if (nio == zio)
31660f7643c7SGeorge Wilson 		return (ZIO_PIPELINE_CONTINUE);
31670f7643c7SGeorge Wilson 
31680f7643c7SGeorge Wilson 	if (nio != NULL) {
31690f7643c7SGeorge Wilson 		ASSERT(nio->io_stage == ZIO_STAGE_DVA_THROTTLE);
31700f7643c7SGeorge Wilson 		/*
31710f7643c7SGeorge Wilson 		 * We are passing control to a new zio so make sure that
31720f7643c7SGeorge Wilson 		 * it is processed by a different thread. We do this to
31730f7643c7SGeorge Wilson 		 * avoid stack overflows that can occur when parents are
31740f7643c7SGeorge Wilson 		 * throttled and children are making progress. We allow
31750f7643c7SGeorge Wilson 		 * it to go to the head of the taskq since it's already
31760f7643c7SGeorge Wilson 		 * been waiting.
31770f7643c7SGeorge Wilson 		 */
31780f7643c7SGeorge Wilson 		zio_taskq_dispatch(nio, ZIO_TASKQ_ISSUE, B_TRUE);
31790f7643c7SGeorge Wilson 	}
31800f7643c7SGeorge Wilson 	return (ZIO_PIPELINE_STOP);
31810f7643c7SGeorge Wilson }
31820f7643c7SGeorge Wilson 
3183663207adSDon Brady static void
3184f78cdc34SPaul Dagnelie zio_allocate_dispatch(spa_t *spa, int allocator)
31850f7643c7SGeorge Wilson {
31860f7643c7SGeorge Wilson 	zio_t *zio;
31870f7643c7SGeorge Wilson 
3188f78cdc34SPaul Dagnelie 	mutex_enter(&spa->spa_alloc_locks[allocator]);
3189f78cdc34SPaul Dagnelie 	zio = zio_io_to_allocate(spa, allocator);
3190f78cdc34SPaul Dagnelie 	mutex_exit(&spa->spa_alloc_locks[allocator]);
31910f7643c7SGeorge Wilson 	if (zio == NULL)
31920f7643c7SGeorge Wilson 		return;
31930f7643c7SGeorge Wilson 
31940f7643c7SGeorge Wilson 	ASSERT3U(zio->io_stage, ==, ZIO_STAGE_DVA_THROTTLE);
31950f7643c7SGeorge Wilson 	ASSERT0(zio->io_error);
31960f7643c7SGeorge Wilson 	zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_TRUE);
31970f7643c7SGeorge Wilson }
31980f7643c7SGeorge Wilson 
3199e05725b1Sbonwick static int
3200fa9e4066Sahrens zio_dva_allocate(zio_t *zio)
3201fa9e4066Sahrens {
32028654d025Sperrin 	spa_t *spa = zio->io_spa;
3203663207adSDon Brady 	metaslab_class_t *mc;
3204fa9e4066Sahrens 	blkptr_t *bp = zio->io_bp;
3205fa9e4066Sahrens 	int error;
320609c9d376SGeorge Wilson 	int flags = 0;
3207fa9e4066Sahrens 
3208f5383399SBill Moore 	if (zio->io_gang_leader == NULL) {
3209f5383399SBill Moore 		ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
3210f5383399SBill Moore 		zio->io_gang_leader = zio;
3211f5383399SBill Moore 	}
3212f5383399SBill Moore 
3213fa9e4066Sahrens 	ASSERT(BP_IS_HOLE(bp));
3214fb09f5aaSMadhav Suresh 	ASSERT0(BP_GET_NDVAS(bp));
3215b24ab676SJeff Bonwick 	ASSERT3U(zio->io_prop.zp_copies, >, 0);
3216b24ab676SJeff Bonwick 	ASSERT3U(zio->io_prop.zp_copies, <=, spa_max_replication(spa));
3217fa9e4066Sahrens 	ASSERT3U(zio->io_size, ==, BP_GET_PSIZE(bp));
3218fa9e4066Sahrens 
3219663207adSDon Brady 	if (zio->io_flags & ZIO_FLAG_NODATA)
32200f7643c7SGeorge Wilson 		flags |= METASLAB_DONT_THROTTLE;
3221663207adSDon Brady 	if (zio->io_flags & ZIO_FLAG_GANG_CHILD)
32220f7643c7SGeorge Wilson 		flags |= METASLAB_GANG_CHILD;
3223663207adSDon Brady 	if (zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE)
32240f7643c7SGeorge Wilson 		flags |= METASLAB_ASYNC_ALLOC;
3225663207adSDon Brady 
3226663207adSDon Brady 	/*
3227663207adSDon Brady 	 * if not already chosen, locate an appropriate allocation class
3228663207adSDon Brady 	 */
3229663207adSDon Brady 	mc = zio->io_metaslab_class;
3230663207adSDon Brady 	if (mc == NULL) {
3231663207adSDon Brady 		mc = spa_preferred_class(spa, zio->io_size,
3232663207adSDon Brady 		    zio->io_prop.zp_type, zio->io_prop.zp_level,
3233663207adSDon Brady 		    zio->io_prop.zp_zpl_smallblk);
3234663207adSDon Brady 		zio->io_metaslab_class = mc;
32350f7643c7SGeorge Wilson 	}
32360f7643c7SGeorge Wilson 
3237e14bb325SJeff Bonwick 	error = metaslab_alloc(spa, mc, zio->io_size, bp,
32388363e80aSGeorge Wilson 	    zio->io_prop.zp_copies, zio->io_txg, NULL, flags,
3239f78cdc34SPaul Dagnelie 	    &zio->io_alloc_list, zio, zio->io_allocator);
3240fa9e4066Sahrens 
3241663207adSDon Brady 	/*
3242663207adSDon Brady 	 * Fallback to normal class when an alloc class is full
3243663207adSDon Brady 	 */
3244663207adSDon Brady 	if (error == ENOSPC && mc != spa_normal_class(spa)) {
3245663207adSDon Brady 		/*
3246663207adSDon Brady 		 * If throttling, transfer reservation over to normal class.
3247663207adSDon Brady 		 * The io_allocator slot can remain the same even though we
3248663207adSDon Brady 		 * are switching classes.
3249663207adSDon Brady 		 */
3250663207adSDon Brady 		if (mc->mc_alloc_throttle_enabled &&
3251663207adSDon Brady 		    (zio->io_flags & ZIO_FLAG_IO_ALLOCATING)) {
3252663207adSDon Brady 			metaslab_class_throttle_unreserve(mc,
3253663207adSDon Brady 			    zio->io_prop.zp_copies, zio->io_allocator, zio);
3254663207adSDon Brady 			zio->io_flags &= ~ZIO_FLAG_IO_ALLOCATING;
3255663207adSDon Brady 
3256663207adSDon Brady 			mc = spa_normal_class(spa);
3257663207adSDon Brady 			VERIFY(metaslab_class_throttle_reserve(mc,
3258663207adSDon Brady 			    zio->io_prop.zp_copies, zio->io_allocator, zio,
3259663207adSDon Brady 			    flags | METASLAB_MUST_RESERVE));
3260663207adSDon Brady 		} else {
3261663207adSDon Brady 			mc = spa_normal_class(spa);
3262663207adSDon Brady 		}
3263663207adSDon Brady 		zio->io_metaslab_class = mc;
3264663207adSDon Brady 
3265663207adSDon Brady 		error = metaslab_alloc(spa, mc, zio->io_size, bp,
3266663207adSDon Brady 		    zio->io_prop.zp_copies, zio->io_txg, NULL, flags,
3267663207adSDon Brady 		    &zio->io_alloc_list, zio, zio->io_allocator);
3268663207adSDon Brady 	}
3269663207adSDon Brady 
32700f7643c7SGeorge Wilson 	if (error != 0) {
327121f7c81cSMatthew Ahrens 		zfs_dbgmsg("%s: metaslab allocation failure: zio %p, "
327209c9d376SGeorge Wilson 		    "size %llu, error %d", spa_name(spa), zio, zio->io_size,
327309c9d376SGeorge Wilson 		    error);
3274e14bb325SJeff Bonwick 		if (error == ENOSPC && zio->io_size > SPA_MINBLOCKSIZE)
3275e14bb325SJeff Bonwick 			return (zio_write_gang_block(zio));
3276fa9e4066Sahrens 		zio->io_error = error;
3277fa9e4066Sahrens 	}
3278e05725b1Sbonwick 
3279e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
3280fa9e4066Sahrens }
3281fa9e4066Sahrens 
3282e05725b1Sbonwick static int
3283fa9e4066Sahrens zio_dva_free(zio_t *zio)
3284fa9e4066Sahrens {
3285e14bb325SJeff Bonwick 	metaslab_free(zio->io_spa, zio->io_bp, zio->io_txg, B_FALSE);
3286fa9e4066Sahrens 
3287e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
3288fa9e4066Sahrens }
3289fa9e4066Sahrens 
3290e05725b1Sbonwick static int
3291fa9e4066Sahrens zio_dva_claim(zio_t *zio)
3292fa9e4066Sahrens {
3293e14bb325SJeff Bonwick 	int error;
3294e14bb325SJeff Bonwick 
3295e14bb325SJeff Bonwick 	error = metaslab_claim(zio->io_spa, zio->io_bp, zio->io_txg);
3296e14bb325SJeff Bonwick 	if (error)
3297e14bb325SJeff Bonwick 		zio->io_error = error;
3298fa9e4066Sahrens 
3299e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
3300fa9e4066Sahrens }
3301fa9e4066Sahrens 
3302e14bb325SJeff Bonwick /*
3303e14bb325SJeff Bonwick  * Undo an allocation.  This is used by zio_done() when an I/O fails
3304e14bb325SJeff Bonwick  * and we want to give back the block we just allocated.
3305e14bb325SJeff Bonwick  * This handles both normal blocks and gang blocks.
3306e14bb325SJeff Bonwick  */
3307e14bb325SJeff Bonwick static void
3308e14bb325SJeff Bonwick zio_dva_unallocate(zio_t *zio, zio_gang_node_t *gn, blkptr_t *bp)
3309e14bb325SJeff Bonwick {
3310e14bb325SJeff Bonwick 	ASSERT(bp->blk_birth == zio->io_txg || BP_IS_HOLE(bp));
3311b24ab676SJeff Bonwick 	ASSERT(zio->io_bp_override == NULL);
3312e14bb325SJeff Bonwick 
3313e14bb325SJeff Bonwick 	if (!BP_IS_HOLE(bp))
3314b24ab676SJeff Bonwick 		metaslab_free(zio->io_spa, bp, bp->blk_birth, B_TRUE);
3315e14bb325SJeff Bonwick 
3316e14bb325SJeff Bonwick 	if (gn != NULL) {
3317e14bb325SJeff Bonwick 		for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
3318e14bb325SJeff Bonwick 			zio_dva_unallocate(zio, gn->gn_child[g],
3319e14bb325SJeff Bonwick 			    &gn->gn_gbh->zg_blkptr[g]);
3320e14bb325SJeff Bonwick 		}
3321e14bb325SJeff Bonwick 	}
3322e14bb325SJeff Bonwick }
3323e14bb325SJeff Bonwick 
3324e14bb325SJeff Bonwick /*
3325e14bb325SJeff Bonwick  * Try to allocate an intent log block.  Return 0 on success, errno on failure.
3326e14bb325SJeff Bonwick  */
3327e14bb325SJeff Bonwick int
3328eb633035STom Caputi zio_alloc_zil(spa_t *spa, objset_t *os, uint64_t txg, blkptr_t *new_bp,
3329f78cdc34SPaul Dagnelie     blkptr_t *old_bp, uint64_t size, boolean_t *slog)
3330e14bb325SJeff Bonwick {
3331e09fa4daSNeil Perrin 	int error = 1;
33328363e80aSGeorge Wilson 	zio_alloc_list_t io_alloc_list;
3333e14bb325SJeff Bonwick 
3334b24ab676SJeff Bonwick 	ASSERT(txg > spa_syncing_txg(spa));
3335b24ab676SJeff Bonwick 
33368363e80aSGeorge Wilson 	metaslab_trace_init(&io_alloc_list);
3337663207adSDon Brady 
3338663207adSDon Brady 	/*
3339663207adSDon Brady 	 * Block pointer fields are useful to metaslabs for stats and debugging.
3340663207adSDon Brady 	 * Fill in the obvious ones before calling into metaslab_alloc().
3341663207adSDon Brady 	 */
3342663207adSDon Brady 	BP_SET_TYPE(new_bp, DMU_OT_INTENT_LOG);
3343663207adSDon Brady 	BP_SET_PSIZE(new_bp, size);
3344663207adSDon Brady 	BP_SET_LEVEL(new_bp, 0);
3345663207adSDon Brady 
3346f78cdc34SPaul Dagnelie 	/*
3347f78cdc34SPaul Dagnelie 	 * When allocating a zil block, we don't have information about
3348f78cdc34SPaul Dagnelie 	 * the final destination of the block except the objset it's part
3349f78cdc34SPaul Dagnelie 	 * of, so we just hash the objset ID to pick the allocator to get
3350f78cdc34SPaul Dagnelie 	 * some parallelism.
3351f78cdc34SPaul Dagnelie 	 */
3352c5ee4681SAlexander Motin 	error = metaslab_alloc(spa, spa_log_class(spa), size, new_bp, 1,
3353f78cdc34SPaul Dagnelie 	    txg, old_bp, METASLAB_HINTBP_AVOID, &io_alloc_list, NULL,
3354eb633035STom Caputi 	    cityhash4(0, 0, 0,
3355eb633035STom Caputi 	    os->os_dsl_dataset->ds_object) % spa->spa_alloc_count);
3356c5ee4681SAlexander Motin 	if (error == 0) {
3357c5ee4681SAlexander Motin 		*slog = TRUE;
3358c5ee4681SAlexander Motin 	} else {
3359b24ab676SJeff Bonwick 		error = metaslab_alloc(spa, spa_normal_class(spa), size,
33608363e80aSGeorge Wilson 		    new_bp, 1, txg, old_bp, METASLAB_HINTBP_AVOID,
3361eb633035STom Caputi 		    &io_alloc_list, NULL, cityhash4(0, 0, 0,
3362eb633035STom Caputi 		    os->os_dsl_dataset->ds_object) % spa->spa_alloc_count);
3363c5ee4681SAlexander Motin 		if (error == 0)
3364c5ee4681SAlexander Motin 			*slog = FALSE;
3365840345f6SGeorge Wilson 	}
33668363e80aSGeorge Wilson 	metaslab_trace_fini(&io_alloc_list);
3367e14bb325SJeff Bonwick 
3368e14bb325SJeff Bonwick 	if (error == 0) {
3369e14bb325SJeff Bonwick 		BP_SET_LSIZE(new_bp, size);
3370e14bb325SJeff Bonwick 		BP_SET_PSIZE(new_bp, size);
3371e14bb325SJeff Bonwick 		BP_SET_COMPRESS(new_bp, ZIO_COMPRESS_OFF);
33726e1f5caaSNeil Perrin 		BP_SET_CHECKSUM(new_bp,
33736e1f5caaSNeil Perrin 		    spa_version(spa) >= SPA_VERSION_SLIM_ZIL
33746e1f5caaSNeil Perrin 		    ? ZIO_CHECKSUM_ZILOG2 : ZIO_CHECKSUM_ZILOG);
3375e14bb325SJeff Bonwick 		BP_SET_TYPE(new_bp, DMU_OT_INTENT_LOG);
3376e14bb325SJeff Bonwick 		BP_SET_LEVEL(new_bp, 0);
3377b24ab676SJeff Bonwick 		BP_SET_DEDUP(new_bp, 0);
3378e14bb325SJeff Bonwick 		BP_SET_BYTEORDER(new_bp, ZFS_HOST_BYTEORDER);
3379eb633035STom Caputi 
3380eb633035STom Caputi 		/*
3381eb633035STom Caputi 		 * encrypted blocks will require an IV and salt. We generate
3382eb633035STom Caputi 		 * these now since we will not be rewriting the bp at
3383eb633035STom Caputi 		 * rewrite time.
3384eb633035STom Caputi 		 */
3385eb633035STom Caputi 		if (os->os_encrypted) {
3386eb633035STom Caputi 			uint8_t iv[ZIO_DATA_IV_LEN];
3387eb633035STom Caputi 			uint8_t salt[ZIO_DATA_SALT_LEN];
3388eb633035STom Caputi 
3389eb633035STom Caputi 			BP_SET_CRYPT(new_bp, B_TRUE);
3390eb633035STom Caputi 			VERIFY0(spa_crypt_get_salt(spa,
3391eb633035STom Caputi 			    dmu_objset_id(os), salt));
3392eb633035STom Caputi 			VERIFY0(zio_crypt_generate_iv(iv));
3393eb633035STom Caputi 
3394eb633035STom Caputi 			zio_crypt_encode_params_bp(new_bp, salt, iv);
3395eb633035STom Caputi 		}
33961271e4b1SPrakash Surya 	} else {
33971271e4b1SPrakash Surya 		zfs_dbgmsg("%s: zil block allocation failure: "
33981271e4b1SPrakash Surya 		    "size %llu, error %d", spa_name(spa), size, error);
3399e14bb325SJeff Bonwick 	}
3400e14bb325SJeff Bonwick 
3401e14bb325SJeff Bonwick 	return (error);
3402e14bb325SJeff Bonwick }
3403e14bb325SJeff Bonwick 
3404fa9e4066Sahrens /*
3405fa9e4066Sahrens  * ==========================================================================
3406fa9e4066Sahrens  * Read and write to physical devices
3407fa9e4066Sahrens  * ==========================================================================
3408fa9e4066Sahrens  */
3409738f37bcSGeorge Wilson 
3410738f37bcSGeorge Wilson /*
3411738f37bcSGeorge Wilson  * Issue an I/O to the underlying vdev. Typically the issue pipeline
3412738f37bcSGeorge Wilson  * stops after this stage and will resume upon I/O completion.
3413738f37bcSGeorge Wilson  * However, there are instances where the vdev layer may need to
3414738f37bcSGeorge Wilson  * continue the pipeline when an I/O was not issued. Since the I/O
3415738f37bcSGeorge Wilson  * that was sent to the vdev layer might be different than the one
3416738f37bcSGeorge Wilson  * currently active in the pipeline (see vdev_queue_io()), we explicitly
3417738f37bcSGeorge Wilson  * force the underlying vdev layers to call either zio_execute() or
3418738f37bcSGeorge Wilson  * zio_interrupt() to ensure that the pipeline continues with the correct I/O.
3419738f37bcSGeorge Wilson  */
3420e05725b1Sbonwick static int
342144cd46caSbillm zio_vdev_io_start(zio_t *zio)
3422fa9e4066Sahrens {
3423fa9e4066Sahrens 	vdev_t *vd = zio->io_vd;
342444cd46caSbillm 	uint64_t align;
34250a4e9518Sgw 	spa_t *spa = zio->io_spa;
34260a4e9518Sgw 
3427dd50e0ccSTony Hutter 	zio->io_delay = 0;
3428dd50e0ccSTony Hutter 
3429e14bb325SJeff Bonwick 	ASSERT(zio->io_error == 0);
3430e14bb325SJeff Bonwick 	ASSERT(zio->io_child_error[ZIO_CHILD_VDEV] == 0);
3431fa9e4066Sahrens 
3432e14bb325SJeff Bonwick 	if (vd == NULL) {
3433e14bb325SJeff Bonwick 		if (!(zio->io_flags & ZIO_FLAG_CONFIG_WRITER))
3434e14bb325SJeff Bonwick 			spa_config_enter(spa, SCL_ZIO, zio, RW_READER);
3435fa9e4066Sahrens 
3436e14bb325SJeff Bonwick 		/*
3437e14bb325SJeff Bonwick 		 * The mirror_ops handle multiple DVAs in a single BP.
3438e14bb325SJeff Bonwick 		 */
3439738f37bcSGeorge Wilson 		vdev_mirror_ops.vdev_op_io_start(zio);
3440738f37bcSGeorge Wilson 		return (ZIO_PIPELINE_STOP);
3441fa9e4066Sahrens 	}
3442fa9e4066Sahrens 
34430f7643c7SGeorge Wilson 	ASSERT3P(zio->io_logical, !=, zio);
34446f793812SPavel Zakharov 	if (zio->io_type == ZIO_TYPE_WRITE) {
34456f793812SPavel Zakharov 		ASSERT(spa->spa_trust_config);
34466f793812SPavel Zakharov 
3447a3874b8bSToomas Soome 		/*
3448a3874b8bSToomas Soome 		 * Note: the code can handle other kinds of writes,
3449a3874b8bSToomas Soome 		 * but we don't expect them.
3450a3874b8bSToomas Soome 		 */
34516f793812SPavel Zakharov 		if (zio->io_vd->vdev_removing) {
34526f793812SPavel Zakharov 			ASSERT(zio->io_flags &
34536f793812SPavel Zakharov 			    (ZIO_FLAG_PHYSICAL | ZIO_FLAG_SELF_HEAL |
34543a4b1be9SMatthew Ahrens 			    ZIO_FLAG_RESILVER | ZIO_FLAG_INDUCE_DAMAGE));
34556f793812SPavel Zakharov 		}
34565cabbc6bSPrashanth Sreenivasa 	}
34570f7643c7SGeorge Wilson 
3458e14bb325SJeff Bonwick 	align = 1ULL << vd->vdev_top->vdev_ashift;
3459e14bb325SJeff Bonwick 
34602a104a52SAlex Reece 	if (!(zio->io_flags & ZIO_FLAG_PHYSICAL) &&
34612a104a52SAlex Reece 	    P2PHASE(zio->io_size, align) != 0) {
34622a104a52SAlex Reece 		/* Transform logical writes to be a full physical block size. */
3463ecc2d604Sbonwick 		uint64_t asize = P2ROUNDUP(zio->io_size, align);
3464770499e1SDan Kimmel 		abd_t *abuf = abd_alloc_sametype(zio->io_abd, asize);
3465e14bb325SJeff Bonwick 		ASSERT(vd == vd->vdev_top);
3466ecc2d604Sbonwick 		if (zio->io_type == ZIO_TYPE_WRITE) {
3467770499e1SDan Kimmel 			abd_copy(abuf, zio->io_abd, zio->io_size);
3468770499e1SDan Kimmel 			abd_zero_off(abuf, zio->io_size, asize - zio->io_size);
3469ecc2d604Sbonwick 		}
3470e14bb325SJeff Bonwick 		zio_push_transform(zio, abuf, asize, asize, zio_subblock);
3471ecc2d604Sbonwick 	}
3472ecc2d604Sbonwick 
34732a104a52SAlex Reece 	/*
34742a104a52SAlex Reece 	 * If this is not a physical io, make sure that it is properly aligned
34752a104a52SAlex Reece 	 * before proceeding.
34762a104a52SAlex Reece 	 */
34772a104a52SAlex Reece 	if (!(zio->io_flags & ZIO_FLAG_PHYSICAL)) {
34782a104a52SAlex Reece 		ASSERT0(P2PHASE(zio->io_offset, align));
34792a104a52SAlex Reece 		ASSERT0(P2PHASE(zio->io_size, align));
34802a104a52SAlex Reece 	} else {
34812a104a52SAlex Reece 		/*
34822a104a52SAlex Reece 		 * For physical writes, we allow 512b aligned writes and assume
34832a104a52SAlex Reece 		 * the device will perform a read-modify-write as necessary.
34842a104a52SAlex Reece 		 */
34852a104a52SAlex Reece 		ASSERT0(P2PHASE(zio->io_offset, SPA_MINBLOCKSIZE));
34862a104a52SAlex Reece 		ASSERT0(P2PHASE(zio->io_size, SPA_MINBLOCKSIZE));
34872a104a52SAlex Reece 	}
34882a104a52SAlex Reece 
3489f9af39baSGeorge Wilson 	VERIFY(zio->io_type != ZIO_TYPE_WRITE || spa_writeable(spa));
34908ad4d6ddSJeff Bonwick 
34918ad4d6ddSJeff Bonwick 	/*
34928ad4d6ddSJeff Bonwick 	 * If this is a repair I/O, and there's no self-healing involved --
34938ad4d6ddSJeff Bonwick 	 * that is, we're just resilvering what we expect to resilver --
34948ad4d6ddSJeff Bonwick 	 * then don't do the I/O unless zio's txg is actually in vd's DTL.
34953a4b1be9SMatthew Ahrens 	 * This prevents spurious resilvering.
34963a4b1be9SMatthew Ahrens 	 *
34973a4b1be9SMatthew Ahrens 	 * There are a few ways that we can end up creating these spurious
34983a4b1be9SMatthew Ahrens 	 * resilver i/os:
34993a4b1be9SMatthew Ahrens 	 *
35003a4b1be9SMatthew Ahrens 	 * 1. A resilver i/o will be issued if any DVA in the BP has a
35013a4b1be9SMatthew Ahrens 	 * dirty DTL.  The mirror code will issue resilver writes to
35023a4b1be9SMatthew Ahrens 	 * each DVA, including the one(s) that are not on vdevs with dirty
35033a4b1be9SMatthew Ahrens 	 * DTLs.
35043a4b1be9SMatthew Ahrens 	 *
35053a4b1be9SMatthew Ahrens 	 * 2. With nested replication, which happens when we have a
35063a4b1be9SMatthew Ahrens 	 * "replacing" or "spare" vdev that's a child of a mirror or raidz.
35073a4b1be9SMatthew Ahrens 	 * For example, given mirror(replacing(A+B), C), it's likely that
35083a4b1be9SMatthew Ahrens 	 * only A is out of date (it's the new device). In this case, we'll
35093a4b1be9SMatthew Ahrens 	 * read from C, then use the data to resilver A+B -- but we don't
35103a4b1be9SMatthew Ahrens 	 * actually want to resilver B, just A. The top-level mirror has no
35113a4b1be9SMatthew Ahrens 	 * way to know this, so instead we just discard unnecessary repairs
35123a4b1be9SMatthew Ahrens 	 * as we work our way down the vdev tree.
35133a4b1be9SMatthew Ahrens 	 *
35143a4b1be9SMatthew Ahrens 	 * 3. ZTEST also creates mirrors of mirrors, mirrors of raidz, etc.
35153a4b1be9SMatthew Ahrens 	 * The same logic applies to any form of nested replication: ditto
35163a4b1be9SMatthew Ahrens 	 * + mirror, RAID-Z + replacing, etc.
35173a4b1be9SMatthew Ahrens 	 *
35183a4b1be9SMatthew Ahrens 	 * However, indirect vdevs point off to other vdevs which may have
35193a4b1be9SMatthew Ahrens 	 * DTL's, so we never bypass them.  The child i/os on concrete vdevs
35203a4b1be9SMatthew Ahrens 	 * will be properly bypassed instead.
35218ad4d6ddSJeff Bonwick 	 */
35228ad4d6ddSJeff Bonwick 	if ((zio->io_flags & ZIO_FLAG_IO_REPAIR) &&
35238ad4d6ddSJeff Bonwick 	    !(zio->io_flags & ZIO_FLAG_SELF_HEAL) &&
35248ad4d6ddSJeff Bonwick 	    zio->io_txg != 0 &&	/* not a delegated i/o */
35253a4b1be9SMatthew Ahrens 	    vd->vdev_ops != &vdev_indirect_ops &&
35268ad4d6ddSJeff Bonwick 	    !vdev_dtl_contains(vd, DTL_PARTIAL, zio->io_txg, 1)) {
35278ad4d6ddSJeff Bonwick 		ASSERT(zio->io_type == ZIO_TYPE_WRITE);
35288ad4d6ddSJeff Bonwick 		zio_vdev_io_bypass(zio);
35298ad4d6ddSJeff Bonwick 		return (ZIO_PIPELINE_CONTINUE);
35308ad4d6ddSJeff Bonwick 	}
3531fa9e4066Sahrens 
3532084fd14fSBrian Behlendorf 	if (vd->vdev_ops->vdev_op_leaf && (zio->io_type == ZIO_TYPE_READ ||
3533084fd14fSBrian Behlendorf 	    zio->io_type == ZIO_TYPE_WRITE || zio->io_type == ZIO_TYPE_TRIM)) {
3534e14bb325SJeff Bonwick 
353543466aaeSMax Grossman 		if (zio->io_type == ZIO_TYPE_READ && vdev_cache_read(zio))
3536a3f829aeSBill Moore 			return (ZIO_PIPELINE_CONTINUE);
3537e14bb325SJeff Bonwick 
3538e14bb325SJeff Bonwick 		if ((zio = vdev_queue_io(zio)) == NULL)
3539e14bb325SJeff Bonwick 			return (ZIO_PIPELINE_STOP);
3540e14bb325SJeff Bonwick 
3541e14bb325SJeff Bonwick 		if (!vdev_accessible(vd, zio)) {
3542be6fd75aSMatthew Ahrens 			zio->io_error = SET_ERROR(ENXIO);
3543e14bb325SJeff Bonwick 			zio_interrupt(zio);
3544e14bb325SJeff Bonwick 			return (ZIO_PIPELINE_STOP);
3545e14bb325SJeff Bonwick 		}
3546dd50e0ccSTony Hutter 		zio->io_delay = gethrtime();
3547e14bb325SJeff Bonwick 	}
3548e14bb325SJeff Bonwick 
3549738f37bcSGeorge Wilson 	vd->vdev_ops->vdev_op_io_start(zio);
3550738f37bcSGeorge Wilson 	return (ZIO_PIPELINE_STOP);
3551fa9e4066Sahrens }
3552fa9e4066Sahrens 
3553e05725b1Sbonwick static int
3554fa9e4066Sahrens zio_vdev_io_done(zio_t *zio)
3555fa9e4066Sahrens {
3556e14bb325SJeff Bonwick 	vdev_t *vd = zio->io_vd;
3557e14bb325SJeff Bonwick 	vdev_ops_t *ops = vd ? vd->vdev_ops : &vdev_mirror_ops;
3558e14bb325SJeff Bonwick 	boolean_t unexpected_error = B_FALSE;
3559e05725b1Sbonwick 
3560d6e1c446SGeorge Wilson 	if (zio_wait_for_children(zio, ZIO_CHILD_VDEV_BIT, ZIO_WAIT_DONE)) {
3561e14bb325SJeff Bonwick 		return (ZIO_PIPELINE_STOP);
3562d6e1c446SGeorge Wilson 	}
3563fa9e4066Sahrens 
3564084fd14fSBrian Behlendorf 	ASSERT(zio->io_type == ZIO_TYPE_READ ||
3565084fd14fSBrian Behlendorf 	    zio->io_type == ZIO_TYPE_WRITE || zio->io_type == ZIO_TYPE_TRIM);
3566e14bb325SJeff Bonwick 
3567dd50e0ccSTony Hutter 	if (zio->io_delay)
3568dd50e0ccSTony Hutter 		zio->io_delay = gethrtime() - zio->io_delay;
3569dd50e0ccSTony Hutter 
3570e14bb325SJeff Bonwick 	if (vd != NULL && vd->vdev_ops->vdev_op_leaf) {
3571e14bb325SJeff Bonwick 
3572e14bb325SJeff Bonwick 		vdev_queue_io_done(zio);
3573fa9e4066Sahrens 
3574e14bb325SJeff Bonwick 		if (zio->io_type == ZIO_TYPE_WRITE)
3575e14bb325SJeff Bonwick 			vdev_cache_write(zio);
3576e14bb325SJeff Bonwick 
3577e14bb325SJeff Bonwick 		if (zio_injection_enabled && zio->io_error == 0)
35788956713aSEric Schrock 			zio->io_error = zio_handle_device_injection(vd,
35798956713aSEric Schrock 			    zio, EIO);
3580e14bb325SJeff Bonwick 
3581e14bb325SJeff Bonwick 		if (zio_injection_enabled && zio->io_error == 0)
3582e14bb325SJeff Bonwick 			zio->io_error = zio_handle_label_injection(zio, EIO);
3583e14bb325SJeff Bonwick 
3584084fd14fSBrian Behlendorf 		if (zio->io_error && zio->io_type != ZIO_TYPE_TRIM) {
3585e14bb325SJeff Bonwick 			if (!vdev_accessible(vd, zio)) {
3586be6fd75aSMatthew Ahrens 				zio->io_error = SET_ERROR(ENXIO);
3587e14bb325SJeff Bonwick 			} else {
3588e14bb325SJeff Bonwick 				unexpected_error = B_TRUE;
3589e14bb325SJeff Bonwick 			}
3590e14bb325SJeff Bonwick 		}
359151ece835Seschrock 	}
3592fa9e4066Sahrens 
3593e14bb325SJeff Bonwick 	ops->vdev_op_io_done(zio);
3594e14bb325SJeff Bonwick 
3595e14bb325SJeff Bonwick 	if (unexpected_error)
3596a3f829aeSBill Moore 		VERIFY(vdev_probe(vd, zio) == NULL);
3597e14bb325SJeff Bonwick 
3598e14bb325SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
3599fa9e4066Sahrens }
3600fa9e4066Sahrens 
3601a3874b8bSToomas Soome /*
3602a3874b8bSToomas Soome  * This function is used to change the priority of an existing zio that is
3603a3874b8bSToomas Soome  * currently in-flight. This is used by the arc to upgrade priority in the
3604a3874b8bSToomas Soome  * event that a demand read is made for a block that is currently queued
3605a3874b8bSToomas Soome  * as a scrub or async read IO. Otherwise, the high priority read request
3606a3874b8bSToomas Soome  * would end up having to wait for the lower priority IO.
3607a3874b8bSToomas Soome  */
3608a3874b8bSToomas Soome void
3609a3874b8bSToomas Soome zio_change_priority(zio_t *pio, zio_priority_t priority)
3610a3874b8bSToomas Soome {
3611a3874b8bSToomas Soome 	zio_t *cio, *cio_next;
3612a3874b8bSToomas Soome 	zio_link_t *zl = NULL;
3613a3874b8bSToomas Soome 
3614a3874b8bSToomas Soome 	ASSERT3U(priority, <, ZIO_PRIORITY_NUM_QUEUEABLE);
3615a3874b8bSToomas Soome 
3616a3874b8bSToomas Soome 	if (pio->io_vd != NULL && pio->io_vd->vdev_ops->vdev_op_leaf) {
3617a3874b8bSToomas Soome 		vdev_queue_change_io_priority(pio, priority);
3618a3874b8bSToomas Soome 	} else {
3619a3874b8bSToomas Soome 		pio->io_priority = priority;
3620a3874b8bSToomas Soome 	}
3621a3874b8bSToomas Soome 
3622a3874b8bSToomas Soome 	mutex_enter(&pio->io_lock);
3623a3874b8bSToomas Soome 	for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
3624a3874b8bSToomas Soome 		cio_next = zio_walk_children(pio, &zl);
3625a3874b8bSToomas Soome 		zio_change_priority(cio, priority);
3626a3874b8bSToomas Soome 	}
3627a3874b8bSToomas Soome 	mutex_exit(&pio->io_lock);
3628a3874b8bSToomas Soome }
3629a3874b8bSToomas Soome 
363022fe2c88SJonathan Adams /*
363122fe2c88SJonathan Adams  * For non-raidz ZIOs, we can just copy aside the bad data read from the
363222fe2c88SJonathan Adams  * disk, and use that to finish the checksum ereport later.
363322fe2c88SJonathan Adams  */
363422fe2c88SJonathan Adams static void
363522fe2c88SJonathan Adams zio_vsd_default_cksum_finish(zio_cksum_report_t *zcr,
3636eb633035STom Caputi     const abd_t *good_buf)
363722fe2c88SJonathan Adams {
363822fe2c88SJonathan Adams 	/* no processing needed */
363922fe2c88SJonathan Adams 	zfs_ereport_finish_checksum(zcr, good_buf, zcr->zcr_cbdata, B_FALSE);
364022fe2c88SJonathan Adams }
364122fe2c88SJonathan Adams 
364222fe2c88SJonathan Adams /*ARGSUSED*/
364322fe2c88SJonathan Adams void
364422fe2c88SJonathan Adams zio_vsd_default_cksum_report(zio_t *zio, zio_cksum_report_t *zcr, void *ignored)
364522fe2c88SJonathan Adams {
3646eb633035STom Caputi 	void *abd = abd_alloc_sametype(zio->io_abd, zio->io_size);
364722fe2c88SJonathan Adams 
3648eb633035STom Caputi 	abd_copy(abd, zio->io_abd, zio->io_size);
364922fe2c88SJonathan Adams 
365022fe2c88SJonathan Adams 	zcr->zcr_cbinfo = zio->io_size;
3651eb633035STom Caputi 	zcr->zcr_cbdata = abd;
365222fe2c88SJonathan Adams 	zcr->zcr_finish = zio_vsd_default_cksum_finish;
3653eb633035STom Caputi 	zcr->zcr_free = zio_abd_free;
365422fe2c88SJonathan Adams }
365522fe2c88SJonathan Adams 
3656e05725b1Sbonwick static int
3657fa9e4066Sahrens zio_vdev_io_assess(zio_t *zio)
3658fa9e4066Sahrens {
3659fa9e4066Sahrens 	vdev_t *vd = zio->io_vd;
3660e14bb325SJeff Bonwick 
3661d6e1c446SGeorge Wilson 	if (zio_wait_for_children(zio, ZIO_CHILD_VDEV_BIT, ZIO_WAIT_DONE)) {
3662e14bb325SJeff Bonwick 		return (ZIO_PIPELINE_STOP);
3663d6e1c446SGeorge Wilson 	}
3664e14bb325SJeff Bonwick 
3665e14bb325SJeff Bonwick 	if (vd == NULL && !(zio->io_flags & ZIO_FLAG_CONFIG_WRITER))
3666e14bb325SJeff Bonwick 		spa_config_exit(zio->io_spa, SCL_ZIO, zio);
3667e14bb325SJeff Bonwick 
3668e14bb325SJeff Bonwick 	if (zio->io_vsd != NULL) {
366922fe2c88SJonathan Adams 		zio->io_vsd_ops->vsd_free(zio);
3670e14bb325SJeff Bonwick 		zio->io_vsd = NULL;
3671ecc2d604Sbonwick 	}
3672ecc2d604Sbonwick 
3673e14bb325SJeff Bonwick 	if (zio_injection_enabled && zio->io_error == 0)
3674ea8dc4b6Seschrock 		zio->io_error = zio_handle_fault_injection(zio, EIO);
3675ea8dc4b6Seschrock 
3676fa9e4066Sahrens 	/*
3677fa9e4066Sahrens 	 * If the I/O failed, determine whether we should attempt to retry it.
367835a5a358SJonathan Adams 	 *
367935a5a358SJonathan Adams 	 * On retry, we cut in line in the issue queue, since we don't want
368035a5a358SJonathan Adams 	 * compression/checksumming/etc. work to prevent our (cheap) IO reissue.
3681fa9e4066Sahrens 	 */
3682e14bb325SJeff Bonwick 	if (zio->io_error && vd == NULL &&
3683e14bb325SJeff Bonwick 	    !(zio->io_flags & (ZIO_FLAG_DONT_RETRY | ZIO_FLAG_IO_RETRY))) {
3684e14bb325SJeff Bonwick 		ASSERT(!(zio->io_flags & ZIO_FLAG_DONT_QUEUE));	/* not a leaf */
3685e14bb325SJeff Bonwick 		ASSERT(!(zio->io_flags & ZIO_FLAG_IO_BYPASS));	/* not a leaf */
3686fa9e4066Sahrens 		zio->io_error = 0;
3687e14bb325SJeff Bonwick 		zio->io_flags |= ZIO_FLAG_IO_RETRY |
3688e14bb325SJeff Bonwick 		    ZIO_FLAG_DONT_CACHE | ZIO_FLAG_DONT_AGGREGATE;
3689b24ab676SJeff Bonwick 		zio->io_stage = ZIO_STAGE_VDEV_IO_START >> 1;
369035a5a358SJonathan Adams 		zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE,
369135a5a358SJonathan Adams 		    zio_requeue_io_start_cut_in_line);
3692e14bb325SJeff Bonwick 		return (ZIO_PIPELINE_STOP);
3693ea8dc4b6Seschrock 	}
3694fa9e4066Sahrens 
3695e14bb325SJeff Bonwick 	/*
3696e14bb325SJeff Bonwick 	 * If we got an error on a leaf device, convert it to ENXIO
3697e14bb325SJeff Bonwick 	 * if the device is not accessible at all.
3698e14bb325SJeff Bonwick 	 */
3699e14bb325SJeff Bonwick 	if (zio->io_error && vd != NULL && vd->vdev_ops->vdev_op_leaf &&
3700e14bb325SJeff Bonwick 	    !vdev_accessible(vd, zio))
3701be6fd75aSMatthew Ahrens 		zio->io_error = SET_ERROR(ENXIO);
3702e14bb325SJeff Bonwick 
3703e14bb325SJeff Bonwick 	/*
3704e14bb325SJeff Bonwick 	 * If we can't write to an interior vdev (mirror or RAID-Z),
3705e14bb325SJeff Bonwick 	 * set vdev_cant_write so that we stop trying to allocate from it.
3706e14bb325SJeff Bonwick 	 */
3707e14bb325SJeff Bonwick 	if (zio->io_error == ENXIO && zio->io_type == ZIO_TYPE_WRITE &&
37083b2aab18SMatthew Ahrens 	    vd != NULL && !vd->vdev_ops->vdev_op_leaf) {
3709e14bb325SJeff Bonwick 		vd->vdev_cant_write = B_TRUE;
37103b2aab18SMatthew Ahrens 	}
3711e14bb325SJeff Bonwick 
3712295438baSHans Rosenfeld 	/*
3713295438baSHans Rosenfeld 	 * If a cache flush returns ENOTSUP or ENOTTY, we know that no future
3714084fd14fSBrian Behlendorf 	 * attempts will ever succeed. In this case we set a persistent
3715084fd14fSBrian Behlendorf 	 * boolean flag so that we don't bother with it in the future.
3716295438baSHans Rosenfeld 	 */
3717295438baSHans Rosenfeld 	if ((zio->io_error == ENOTSUP || zio->io_error == ENOTTY) &&
3718295438baSHans Rosenfeld 	    zio->io_type == ZIO_TYPE_IOCTL &&
3719295438baSHans Rosenfeld 	    zio->io_cmd == DKIOCFLUSHWRITECACHE && vd != NULL)
3720295438baSHans Rosenfeld 		vd->vdev_nowritecache = B_TRUE;
3721295438baSHans Rosenfeld 
3722e14bb325SJeff Bonwick 	if (zio->io_error)
3723e14bb325SJeff Bonwick 		zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
3724e14bb325SJeff Bonwick 
372569962b56SMatthew Ahrens 	if (vd != NULL && vd->vdev_ops->vdev_op_leaf &&
372669962b56SMatthew Ahrens 	    zio->io_physdone != NULL) {
372769962b56SMatthew Ahrens 		ASSERT(!(zio->io_flags & ZIO_FLAG_DELEGATED));
372869962b56SMatthew Ahrens 		ASSERT(zio->io_child_type == ZIO_CHILD_VDEV);
372969962b56SMatthew Ahrens 		zio->io_physdone(zio->io_logical);
373069962b56SMatthew Ahrens 	}
373169962b56SMatthew Ahrens 
3732e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
3733fa9e4066Sahrens }
3734fa9e4066Sahrens 
3735fa9e4066Sahrens void
3736fa9e4066Sahrens zio_vdev_io_reissue(zio_t *zio)
3737fa9e4066Sahrens {
3738fa9e4066Sahrens 	ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_START);
3739fa9e4066Sahrens 	ASSERT(zio->io_error == 0);
3740fa9e4066Sahrens 
3741b24ab676SJeff Bonwick 	zio->io_stage >>= 1;
3742fa9e4066Sahrens }
3743fa9e4066Sahrens 
3744fa9e4066Sahrens void
3745fa9e4066Sahrens zio_vdev_io_redone(zio_t *zio)
3746fa9e4066Sahrens {
3747fa9e4066Sahrens 	ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_DONE);
3748fa9e4066Sahrens 
3749b24ab676SJeff Bonwick 	zio->io_stage >>= 1;
3750fa9e4066Sahrens }
3751fa9e4066Sahrens 
3752fa9e4066Sahrens void
3753fa9e4066Sahrens zio_vdev_io_bypass(zio_t *zio)
3754fa9e4066Sahrens {
3755fa9e4066Sahrens 	ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_START);
3756fa9e4066Sahrens 	ASSERT(zio->io_error == 0);
3757fa9e4066Sahrens 
3758fa9e4066Sahrens 	zio->io_flags |= ZIO_FLAG_IO_BYPASS;
3759b24ab676SJeff Bonwick 	zio->io_stage = ZIO_STAGE_VDEV_IO_ASSESS >> 1;
3760fa9e4066Sahrens }
3761fa9e4066Sahrens 
3762eb633035STom Caputi /*
3763eb633035STom Caputi  * ==========================================================================
3764eb633035STom Caputi  * Encrypt and store encryption parameters
3765eb633035STom Caputi  * ==========================================================================
3766eb633035STom Caputi  */
3767eb633035STom Caputi 
3768eb633035STom Caputi 
3769eb633035STom Caputi /*
3770eb633035STom Caputi  * This function is used for ZIO_STAGE_ENCRYPT. It is responsible for
3771eb633035STom Caputi  * managing the storage of encryption parameters and passing them to the
3772eb633035STom Caputi  * lower-level encryption functions.
3773eb633035STom Caputi  */
3774eb633035STom Caputi static int
3775eb633035STom Caputi zio_encrypt(zio_t *zio)
3776eb633035STom Caputi {
3777eb633035STom Caputi 	zio_prop_t *zp = &zio->io_prop;
3778eb633035STom Caputi 	spa_t *spa = zio->io_spa;
3779eb633035STom Caputi 	blkptr_t *bp = zio->io_bp;
3780eb633035STom Caputi 	uint64_t psize = BP_GET_PSIZE(bp);
3781eb633035STom Caputi 	uint64_t dsobj = zio->io_bookmark.zb_objset;
3782eb633035STom Caputi 	dmu_object_type_t ot = BP_GET_TYPE(bp);
3783eb633035STom Caputi 	void *enc_buf = NULL;
3784eb633035STom Caputi 	abd_t *eabd = NULL;
3785eb633035STom Caputi 	uint8_t salt[ZIO_DATA_SALT_LEN];
3786eb633035STom Caputi 	uint8_t iv[ZIO_DATA_IV_LEN];
3787eb633035STom Caputi 	uint8_t mac[ZIO_DATA_MAC_LEN];
3788eb633035STom Caputi 	boolean_t no_crypt = B_FALSE;
3789eb633035STom Caputi 
3790eb633035STom Caputi 	/* the root zio already encrypted the data */
3791eb633035STom Caputi 	if (zio->io_child_type == ZIO_CHILD_GANG)
3792eb633035STom Caputi 		return (ZIO_PIPELINE_CONTINUE);
3793eb633035STom Caputi 
3794eb633035STom Caputi 	/* only ZIL blocks are re-encrypted on rewrite */
3795eb633035STom Caputi 	if (!IO_IS_ALLOCATING(zio) && ot != DMU_OT_INTENT_LOG)
3796eb633035STom Caputi 		return (ZIO_PIPELINE_CONTINUE);
3797eb633035STom Caputi 
3798eb633035STom Caputi 	if (!(zp->zp_encrypt || BP_IS_ENCRYPTED(bp))) {
3799eb633035STom Caputi 		BP_SET_CRYPT(bp, B_FALSE);
3800eb633035STom Caputi 		return (ZIO_PIPELINE_CONTINUE);
3801eb633035STom Caputi 	}
3802eb633035STom Caputi 
3803eb633035STom Caputi 	/* if we are doing raw encryption set the provided encryption params */
3804eb633035STom Caputi 	if (zio->io_flags & ZIO_FLAG_RAW_ENCRYPT) {
3805eb633035STom Caputi 		ASSERT0(BP_GET_LEVEL(bp));
3806eb633035STom Caputi 		BP_SET_CRYPT(bp, B_TRUE);
3807eb633035STom Caputi 		BP_SET_BYTEORDER(bp, zp->zp_byteorder);
3808eb633035STom Caputi 		if (ot != DMU_OT_OBJSET)
3809eb633035STom Caputi 			zio_crypt_encode_mac_bp(bp, zp->zp_mac);
3810eb633035STom Caputi 
3811eb633035STom Caputi 		/* dnode blocks must be written out in the provided byteorder */
3812eb633035STom Caputi 		if (zp->zp_byteorder != ZFS_HOST_BYTEORDER &&
3813eb633035STom Caputi 		    ot == DMU_OT_DNODE) {
3814eb633035STom Caputi 			void *bswap_buf = zio_buf_alloc(psize);
3815eb633035STom Caputi 			abd_t *babd = abd_get_from_buf(bswap_buf, psize);
3816eb633035STom Caputi 
3817eb633035STom Caputi 			ASSERT3U(BP_GET_COMPRESS(bp), ==, ZIO_COMPRESS_OFF);
3818eb633035STom Caputi 			abd_copy_to_buf(bswap_buf, zio->io_abd, psize);
3819eb633035STom Caputi 			dmu_ot_byteswap[DMU_OT_BYTESWAP(ot)].ob_func(bswap_buf,
3820eb633035STom Caputi 			    psize);
3821eb633035STom Caputi 
3822eb633035STom Caputi 			abd_take_ownership_of_buf(babd, B_TRUE);
3823eb633035STom Caputi 			zio_push_transform(zio, babd, psize, psize, NULL);
3824eb633035STom Caputi 		}
3825eb633035STom Caputi 
3826eb633035STom Caputi 		if (DMU_OT_IS_ENCRYPTED(ot))
3827eb633035STom Caputi 			zio_crypt_encode_params_bp(bp, zp->zp_salt, zp->zp_iv);
3828eb633035STom Caputi 		return (ZIO_PIPELINE_CONTINUE);
3829eb633035STom Caputi 	}
3830eb633035STom Caputi 
3831eb633035STom Caputi 	/* indirect blocks only maintain a cksum of the lower level MACs */
3832eb633035STom Caputi 	if (BP_GET_LEVEL(bp) > 0) {
3833eb633035STom Caputi 		BP_SET_CRYPT(bp, B_TRUE);
3834eb633035STom Caputi 		VERIFY0(zio_crypt_do_indirect_mac_checksum_abd(B_TRUE,
3835eb633035STom Caputi 		    zio->io_orig_abd, BP_GET_LSIZE(bp), BP_SHOULD_BYTESWAP(bp),
3836eb633035STom Caputi 		    mac));
3837eb633035STom Caputi 		zio_crypt_encode_mac_bp(bp, mac);
3838eb633035STom Caputi 		return (ZIO_PIPELINE_CONTINUE);
3839eb633035STom Caputi 	}
3840eb633035STom Caputi 
3841eb633035STom Caputi 	/*
3842eb633035STom Caputi 	 * Objset blocks are a special case since they have 2 256-bit MACs
3843eb633035STom Caputi 	 * embedded within them.
3844eb633035STom Caputi 	 */
3845eb633035STom Caputi 	if (ot == DMU_OT_OBJSET) {
3846eb633035STom Caputi 		ASSERT0(DMU_OT_IS_ENCRYPTED(ot));
3847eb633035STom Caputi 		ASSERT3U(BP_GET_COMPRESS(bp), ==, ZIO_COMPRESS_OFF);
3848eb633035STom Caputi 		BP_SET_CRYPT(bp, B_TRUE);
3849eb633035STom Caputi 		VERIFY0(spa_do_crypt_objset_mac_abd(B_TRUE, spa, dsobj,
3850eb633035STom Caputi 		    zio->io_abd, psize, BP_SHOULD_BYTESWAP(bp)));
3851eb633035STom Caputi 		return (ZIO_PIPELINE_CONTINUE);
3852eb633035STom Caputi 	}
3853eb633035STom Caputi 
3854eb633035STom Caputi 	/* unencrypted object types are only authenticated with a MAC */
3855eb633035STom Caputi 	if (!DMU_OT_IS_ENCRYPTED(ot)) {
3856eb633035STom Caputi 		BP_SET_CRYPT(bp, B_TRUE);
3857eb633035STom Caputi 		VERIFY0(spa_do_crypt_mac_abd(B_TRUE, spa, dsobj,
3858eb633035STom Caputi 		    zio->io_abd, psize, mac));
3859eb633035STom Caputi 		zio_crypt_encode_mac_bp(bp, mac);
3860eb633035STom Caputi 		return (ZIO_PIPELINE_CONTINUE);
3861eb633035STom Caputi 	}
3862eb633035STom Caputi 
3863eb633035STom Caputi 	/*
3864eb633035STom Caputi 	 * Later passes of sync-to-convergence may decide to rewrite data
3865eb633035STom Caputi 	 * in place to avoid more disk reallocations. This presents a problem
3866eb633035STom Caputi 	 * for encryption because this consitutes rewriting the new data with
3867eb633035STom Caputi 	 * the same encryption key and IV. However, this only applies to blocks
3868eb633035STom Caputi 	 * in the MOS (particularly the spacemaps) and we do not encrypt the
3869eb633035STom Caputi 	 * MOS. We assert that the zio is allocating or an intent log write
3870eb633035STom Caputi 	 * to enforce this.
3871eb633035STom Caputi 	 */
3872eb633035STom Caputi 	ASSERT(IO_IS_ALLOCATING(zio) || ot == DMU_OT_INTENT_LOG);
3873eb633035STom Caputi 	ASSERT(BP_GET_LEVEL(bp) == 0 || ot == DMU_OT_INTENT_LOG);
3874eb633035STom Caputi 	ASSERT(spa_feature_is_active(spa, SPA_FEATURE_ENCRYPTION));
3875eb633035STom Caputi 	ASSERT3U(psize, !=, 0);
3876eb633035STom Caputi 
3877eb633035STom Caputi 	enc_buf = zio_buf_alloc(psize);
3878eb633035STom Caputi 	eabd = abd_get_from_buf(enc_buf, psize);
3879eb633035STom Caputi 	abd_take_ownership_of_buf(eabd, B_TRUE);
3880eb633035STom Caputi 
3881eb633035STom Caputi 	/*
3882eb633035STom Caputi 	 * For an explanation of what encryption parameters are stored
3883eb633035STom Caputi 	 * where, see the block comment in zio_crypt.c.
3884eb633035STom Caputi 	 */
3885eb633035STom Caputi 	if (ot == DMU_OT_INTENT_LOG) {
3886eb633035STom Caputi 		zio_crypt_decode_params_bp(bp, salt, iv);
3887eb633035STom Caputi 	} else {
3888eb633035STom Caputi 		BP_SET_CRYPT(bp, B_TRUE);
3889eb633035STom Caputi 	}
3890eb633035STom Caputi 
3891eb633035STom Caputi 	/* Perform the encryption. This should not fail */
3892eb633035STom Caputi 	VERIFY0(spa_do_crypt_abd(B_TRUE, spa, &zio->io_bookmark,
3893eb633035STom Caputi 	    BP_GET_TYPE(bp), BP_GET_DEDUP(bp), BP_SHOULD_BYTESWAP(bp),
3894eb633035STom Caputi 	    salt, iv, mac, psize, zio->io_abd, eabd, &no_crypt));
3895eb633035STom Caputi 
3896eb633035STom Caputi 	/* encode encryption metadata into the bp */
3897eb633035STom Caputi 	if (ot == DMU_OT_INTENT_LOG) {
3898eb633035STom Caputi 		/*
3899eb633035STom Caputi 		 * ZIL blocks store the MAC in the embedded checksum, so the
3900eb633035STom Caputi 		 * transform must always be applied.
3901eb633035STom Caputi 		 */
3902eb633035STom Caputi 		zio_crypt_encode_mac_zil(enc_buf, mac);
3903eb633035STom Caputi 		zio_push_transform(zio, eabd, psize, psize, NULL);
3904eb633035STom Caputi 	} else {
3905eb633035STom Caputi 		BP_SET_CRYPT(bp, B_TRUE);
3906eb633035STom Caputi 		zio_crypt_encode_params_bp(bp, salt, iv);
3907eb633035STom Caputi 		zio_crypt_encode_mac_bp(bp, mac);
3908eb633035STom Caputi 
3909eb633035STom Caputi 		if (no_crypt) {
3910eb633035STom Caputi 			ASSERT3U(ot, ==, DMU_OT_DNODE);
3911eb633035STom Caputi 			abd_free(eabd);
3912eb633035STom Caputi 		} else {
3913eb633035STom Caputi 			zio_push_transform(zio, eabd, psize, psize, NULL);
3914eb633035STom Caputi 		}
3915eb633035STom Caputi 	}
3916eb633035STom Caputi 
3917eb633035STom Caputi 	return (ZIO_PIPELINE_CONTINUE);
3918eb633035STom Caputi }
3919eb633035STom Caputi 
3920fa9e4066Sahrens /*
3921fa9e4066Sahrens  * ==========================================================================
3922fa9e4066Sahrens  * Generate and verify checksums
3923fa9e4066Sahrens  * ==========================================================================
3924fa9e4066Sahrens  */
3925e05725b1Sbonwick static int
3926fa9e4066Sahrens zio_checksum_generate(zio_t *zio)
3927fa9e4066Sahrens {
3928fa9e4066Sahrens 	blkptr_t *bp = zio->io_bp;
3929e14bb325SJeff Bonwick 	enum zio_checksum checksum;
3930fa9e4066Sahrens 
3931e14bb325SJeff Bonwick 	if (bp == NULL) {
3932e14bb325SJeff Bonwick 		/*
3933e14bb325SJeff Bonwick 		 * This is zio_write_phys().
3934e14bb325SJeff Bonwick 		 * We're either generating a label checksum, or none at all.
3935e14bb325SJeff Bonwick 		 */
3936e14bb325SJeff Bonwick 		checksum = zio->io_prop.zp_checksum;
3937e14bb325SJeff Bonwick 
3938e14bb325SJeff Bonwick 		if (checksum == ZIO_CHECKSUM_OFF)
3939e14bb325SJeff Bonwick 			return (ZIO_PIPELINE_CONTINUE);
3940fa9e4066Sahrens 
3941e14bb325SJeff Bonwick 		ASSERT(checksum == ZIO_CHECKSUM_LABEL);
3942e14bb325SJeff Bonwick 	} else {
3943e14bb325SJeff Bonwick 		if (BP_IS_GANG(bp) && zio->io_child_type == ZIO_CHILD_GANG) {
3944e14bb325SJeff Bonwick 			ASSERT(!IO_IS_ALLOCATING(zio));
3945e14bb325SJeff Bonwick 			checksum = ZIO_CHECKSUM_GANG_HEADER;
3946e14bb325SJeff Bonwick 		} else {
3947e14bb325SJeff Bonwick 			checksum = BP_GET_CHECKSUM(bp);
3948e14bb325SJeff Bonwick 		}
3949e14bb325SJeff Bonwick 	}
3950fa9e4066Sahrens 
3951770499e1SDan Kimmel 	zio_checksum_compute(zio, checksum, zio->io_abd, zio->io_size);
3952fa9e4066Sahrens 
3953e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
3954fa9e4066Sahrens }
3955fa9e4066Sahrens 
3956e05725b1Sbonwick static int
3957e14bb325SJeff Bonwick zio_checksum_verify(zio_t *zio)
3958fa9e4066Sahrens {
395922fe2c88SJonathan Adams 	zio_bad_cksum_t info;
3960e14bb325SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
3961e14bb325SJeff Bonwick 	int error;
3962fa9e4066Sahrens 
3963b24ab676SJeff Bonwick 	ASSERT(zio->io_vd != NULL);
3964b24ab676SJeff Bonwick 
3965e14bb325SJeff Bonwick 	if (bp == NULL) {
3966e14bb325SJeff Bonwick 		/*
3967e14bb325SJeff Bonwick 		 * This is zio_read_phys().
3968e14bb325SJeff Bonwick 		 * We're either verifying a label checksum, or nothing at all.
3969e14bb325SJeff Bonwick 		 */
3970e14bb325SJeff Bonwick 		if (zio->io_prop.zp_checksum == ZIO_CHECKSUM_OFF)
3971e14bb325SJeff Bonwick 			return (ZIO_PIPELINE_CONTINUE);
3972fa9e4066Sahrens 
3973e14bb325SJeff Bonwick 		ASSERT(zio->io_prop.zp_checksum == ZIO_CHECKSUM_LABEL);
3974e14bb325SJeff Bonwick 	}
3975fa9e4066Sahrens 
397622fe2c88SJonathan Adams 	if ((error = zio_checksum_error(zio, &info)) != 0) {
3977e14bb325SJeff Bonwick 		zio->io_error = error;
3978373dc1cfSMatthew Ahrens 		if (error == ECKSUM &&
3979373dc1cfSMatthew Ahrens 		    !(zio->io_flags & ZIO_FLAG_SPECULATIVE)) {
398022fe2c88SJonathan Adams 			zfs_ereport_start_checksum(zio->io_spa,
3981eb633035STom Caputi 			    zio->io_vd, &zio->io_bookmark, zio,
3982eb633035STom Caputi 			    zio->io_offset, zio->io_size, NULL, &info);
3983e14bb325SJeff Bonwick 		}
3984fa9e4066Sahrens 	}
3985fa9e4066Sahrens 
3986e05725b1Sbonwick 	return (ZIO_PIPELINE_CONTINUE);
3987fa9e4066Sahrens }
3988fa9e4066Sahrens 
3989fa9e4066Sahrens /*
3990fa9e4066Sahrens  * Called by RAID-Z to ensure we don't compute the checksum twice.
3991fa9e4066Sahrens  */
3992fa9e4066Sahrens void
3993fa9e4066Sahrens zio_checksum_verified(zio_t *zio)
3994fa9e4066Sahrens {
3995b24ab676SJeff Bonwick 	zio->io_pipeline &= ~ZIO_STAGE_CHECKSUM_VERIFY;
3996fa9e4066Sahrens }
3997fa9e4066Sahrens 
3998fa9e4066Sahrens /*
3999e14bb325SJeff Bonwick  * ==========================================================================
4000e14bb325SJeff Bonwick  * Error rank.  Error are ranked in the order 0, ENXIO, ECKSUM, EIO, other.
40015d7b4d43SMatthew Ahrens  * An error of 0 indicates success.  ENXIO indicates whole-device failure,
4002e14bb325SJeff Bonwick  * which may be transient (e.g. unplugged) or permament.  ECKSUM and EIO
4003e14bb325SJeff Bonwick  * indicate errors that are specific to one I/O, and most likely permanent.
4004e14bb325SJeff Bonwick  * Any other error is presumed to be worse because we weren't expecting it.
4005e14bb325SJeff Bonwick  * ==========================================================================
4006fa9e4066Sahrens  */
4007e14bb325SJeff Bonwick int
4008e14bb325SJeff Bonwick zio_worst_error(int e1, int e2)
4009fa9e4066Sahrens {
4010e14bb325SJeff Bonwick 	static int zio_error_rank[] = { 0, ENXIO, ECKSUM, EIO };
4011e14bb325SJeff Bonwick 	int r1, r2;
4012e14bb325SJeff Bonwick 
4013e14bb325SJeff Bonwick 	for (r1 = 0; r1 < sizeof (zio_error_rank) / sizeof (int); r1++)
4014e14bb325SJeff Bonwick 		if (e1 == zio_error_rank[r1])
4015e14bb325SJeff Bonwick 			break;
4016e14bb325SJeff Bonwick 
4017e14bb325SJeff Bonwick 	for (r2 = 0; r2 < sizeof (zio_error_rank) / sizeof (int); r2++)
4018e14bb325SJeff Bonwick 		if (e2 == zio_error_rank[r2])
4019e14bb325SJeff Bonwick 			break;
402044cd46caSbillm 
4021e14bb325SJeff Bonwick 	return (r1 > r2 ? e1 : e2);
4022fa9e4066Sahrens }
4023fa9e4066Sahrens 
4024fa9e4066Sahrens /*
4025fa9e4066Sahrens  * ==========================================================================
4026e14bb325SJeff Bonwick  * I/O completion
4027fa9e4066Sahrens  * ==========================================================================
4028fa9e4066Sahrens  */
4029e14bb325SJeff Bonwick static int
4030e14bb325SJeff Bonwick zio_ready(zio_t *zio)
4031fa9e4066Sahrens {
4032e14bb325SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
4033a3f829aeSBill Moore 	zio_t *pio, *pio_next;
40340f7643c7SGeorge Wilson 	zio_link_t *zl = NULL;
4035fa9e4066Sahrens 
4036d6e1c446SGeorge Wilson 	if (zio_wait_for_children(zio, ZIO_CHILD_GANG_BIT | ZIO_CHILD_DDT_BIT,
4037d6e1c446SGeorge Wilson 	    ZIO_WAIT_READY)) {
4038f5383399SBill Moore 		return (ZIO_PIPELINE_STOP);
4039d6e1c446SGeorge Wilson 	}
4040fa9e4066Sahrens 
4041f5383399SBill Moore 	if (zio->io_ready) {
4042e14bb325SJeff Bonwick 		ASSERT(IO_IS_ALLOCATING(zio));
404380901aeaSGeorge Wilson 		ASSERT(bp->blk_birth == zio->io_txg || BP_IS_HOLE(bp) ||
404480901aeaSGeorge Wilson 		    (zio->io_flags & ZIO_FLAG_NOPWRITE));
4045e14bb325SJeff Bonwick 		ASSERT(zio->io_children[ZIO_CHILD_GANG][ZIO_WAIT_READY] == 0);
4046fa9e4066Sahrens 
4047e14bb325SJeff Bonwick 		zio->io_ready(zio);
4048e14bb325SJeff Bonwick 	}
4049fa9e4066Sahrens 
4050e14bb325SJeff Bonwick 	if (bp != NULL && bp != &zio->io_bp_copy)
4051e14bb325SJeff Bonwick 		zio->io_bp_copy = *bp;
4052fa9e4066Sahrens 
40530f7643c7SGeorge Wilson 	if (zio->io_error != 0) {
4054e14bb325SJeff Bonwick 		zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
4055fa9e4066Sahrens 
40560f7643c7SGeorge Wilson 		if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
40570f7643c7SGeorge Wilson 			ASSERT(IO_IS_ALLOCATING(zio));
40580f7643c7SGeorge Wilson 			ASSERT(zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
4059663207adSDon Brady 			ASSERT(zio->io_metaslab_class != NULL);
4060663207adSDon Brady 
40610f7643c7SGeorge Wilson 			/*
40620f7643c7SGeorge Wilson 			 * We were unable to allocate anything, unreserve and
40630f7643c7SGeorge Wilson 			 * issue the next I/O to allocate.
40640f7643c7SGeorge Wilson 			 */
40650f7643c7SGeorge Wilson 			metaslab_class_throttle_unreserve(
4066663207adSDon Brady 			    zio->io_metaslab_class, zio->io_prop.zp_copies,
4067663207adSDon Brady 			    zio->io_allocator, zio);
4068f78cdc34SPaul Dagnelie 			zio_allocate_dispatch(zio->io_spa, zio->io_allocator);
40690f7643c7SGeorge Wilson 		}
40700f7643c7SGeorge Wilson 	}
40710f7643c7SGeorge Wilson 
4072a3f829aeSBill Moore 	mutex_enter(&zio->io_lock);
4073a3f829aeSBill Moore 	zio->io_state[ZIO_WAIT_READY] = 1;
40740f7643c7SGeorge Wilson 	pio = zio_walk_parents(zio, &zl);
4075a3f829aeSBill Moore 	mutex_exit(&zio->io_lock);
4076a3f829aeSBill Moore 
4077a3f829aeSBill Moore 	/*
4078a3f829aeSBill Moore 	 * As we notify zio's parents, new parents could be added.
4079a3f829aeSBill Moore 	 * New parents go to the head of zio's io_parent_list, however,
4080a3f829aeSBill Moore 	 * so we will (correctly) not notify them.  The remainder of zio's
4081a3f829aeSBill Moore 	 * io_parent_list, from 'pio_next' onward, cannot change because
4082a3f829aeSBill Moore 	 * all parents must wait for us to be done before they can be done.
4083a3f829aeSBill Moore 	 */
4084a3f829aeSBill Moore 	for (; pio != NULL; pio = pio_next) {
40850f7643c7SGeorge Wilson 		pio_next = zio_walk_parents(zio, &zl);
4086e14bb325SJeff Bonwick 		zio_notify_parent(pio, zio, ZIO_WAIT_READY);
4087a3f829aeSBill Moore 	}
4088fa9e4066Sahrens 
4089b24ab676SJeff Bonwick 	if (zio->io_flags & ZIO_FLAG_NODATA) {
4090b24ab676SJeff Bonwick 		if (BP_IS_GANG(bp)) {
4091b24ab676SJeff Bonwick 			zio->io_flags &= ~ZIO_FLAG_NODATA;
4092b24ab676SJeff Bonwick 		} else {
4093770499e1SDan Kimmel 			ASSERT((uintptr_t)zio->io_abd < SPA_MAXBLOCKSIZE);
4094b24ab676SJeff Bonwick 			zio->io_pipeline &= ~ZIO_VDEV_IO_STAGES;
4095b24ab676SJeff Bonwick 		}
4096b24ab676SJeff Bonwick 	}
4097b24ab676SJeff Bonwick 
4098a33cae98STim Haley 	if (zio_injection_enabled &&
4099a33cae98STim Haley 	    zio->io_spa->spa_syncing_txg == zio->io_txg)
4100a33cae98STim Haley 		zio_handle_ignored_writes(zio);
4101a33cae98STim Haley 
4102e14bb325SJeff Bonwick 	return (ZIO_PIPELINE_CONTINUE);
4103fa9e4066Sahrens }
4104fa9e4066Sahrens 
41050f7643c7SGeorge Wilson /*
41060f7643c7SGeorge Wilson  * Update the allocation throttle accounting.
41070f7643c7SGeorge Wilson  */
41080f7643c7SGeorge Wilson static void
41090f7643c7SGeorge Wilson zio_dva_throttle_done(zio_t *zio)
41100f7643c7SGeorge Wilson {
41110f7643c7SGeorge Wilson 	zio_t *lio = zio->io_logical;
41120f7643c7SGeorge Wilson 	zio_t *pio = zio_unique_parent(zio);
41130f7643c7SGeorge Wilson 	vdev_t *vd = zio->io_vd;
41140f7643c7SGeorge Wilson 	int flags = METASLAB_ASYNC_ALLOC;
41150f7643c7SGeorge Wilson 
41160f7643c7SGeorge Wilson 	ASSERT3P(zio->io_bp, !=, NULL);
41170f7643c7SGeorge Wilson 	ASSERT3U(zio->io_type, ==, ZIO_TYPE_WRITE);
41180f7643c7SGeorge Wilson 	ASSERT3U(zio->io_priority, ==, ZIO_PRIORITY_ASYNC_WRITE);
41190f7643c7SGeorge Wilson 	ASSERT3U(zio->io_child_type, ==, ZIO_CHILD_VDEV);
41200f7643c7SGeorge Wilson 	ASSERT(vd != NULL);
41210f7643c7SGeorge Wilson 	ASSERT3P(vd, ==, vd->vdev_top);
41220f7643c7SGeorge Wilson 	ASSERT(!(zio->io_flags & (ZIO_FLAG_IO_REPAIR | ZIO_FLAG_IO_RETRY)));
41230f7643c7SGeorge Wilson 	ASSERT(zio->io_flags & ZIO_FLAG_IO_ALLOCATING);
41240f7643c7SGeorge Wilson 	ASSERT(!(lio->io_flags & ZIO_FLAG_IO_REWRITE));
41250f7643c7SGeorge Wilson 	ASSERT(!(lio->io_orig_flags & ZIO_FLAG_NODATA));
41260f7643c7SGeorge Wilson 
41270f7643c7SGeorge Wilson 	/*
41280f7643c7SGeorge Wilson 	 * Parents of gang children can have two flavors -- ones that
41290f7643c7SGeorge Wilson 	 * allocated the gang header (will have ZIO_FLAG_IO_REWRITE set)
41300f7643c7SGeorge Wilson 	 * and ones that allocated the constituent blocks. The allocation
41310f7643c7SGeorge Wilson 	 * throttle needs to know the allocating parent zio so we must find
41320f7643c7SGeorge Wilson 	 * it here.
41330f7643c7SGeorge Wilson 	 */
41340f7643c7SGeorge Wilson 	if (pio->io_child_type == ZIO_CHILD_GANG) {
41350f7643c7SGeorge Wilson 		/*
41360f7643c7SGeorge Wilson 		 * If our parent is a rewrite gang child then our grandparent
41370f7643c7SGeorge Wilson 		 * would have been the one that performed the allocation.
41380f7643c7SGeorge Wilson 		 */
41390f7643c7SGeorge Wilson 		if (pio->io_flags & ZIO_FLAG_IO_REWRITE)
41400f7643c7SGeorge Wilson 			pio = zio_unique_parent(pio);
41410f7643c7SGeorge Wilson 		flags |= METASLAB_GANG_CHILD;
41420f7643c7SGeorge Wilson 	}
41430f7643c7SGeorge Wilson 
41440f7643c7SGeorge Wilson 	ASSERT(IO_IS_ALLOCATING(pio));
41450f7643c7SGeorge Wilson 	ASSERT3P(zio, !=, zio->io_logical);
41460f7643c7SGeorge Wilson 	ASSERT(zio->io_logical != NULL);
41470f7643c7SGeorge Wilson 	ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REPAIR));
41480f7643c7SGeorge Wilson 	ASSERT0(zio->io_flags & ZIO_FLAG_NOPWRITE);
4149663207adSDon Brady 	ASSERT(zio->io_metaslab_class != NULL);
41500f7643c7SGeorge Wilson 
41510f7643c7SGeorge Wilson 	mutex_enter(&pio->io_lock);
4152f78cdc34SPaul Dagnelie 	metaslab_group_alloc_decrement(zio->io_spa, vd->vdev_id, pio, flags,
4153f78cdc34SPaul Dagnelie 	    pio->io_allocator, B_TRUE);
41540f7643c7SGeorge Wilson 	mutex_exit(&pio->io_lock);
41550f7643c7SGeorge Wilson 
4156663207adSDon Brady 	metaslab_class_throttle_unreserve(zio->io_metaslab_class, 1,
4157663207adSDon Brady 	    pio->io_allocator, pio);
41580f7643c7SGeorge Wilson 
41590f7643c7SGeorge Wilson 	/*
41600f7643c7SGeorge Wilson 	 * Call into the pipeline to see if there is more work that
41610f7643c7SGeorge Wilson 	 * needs to be done. If there is work to be done it will be
41620f7643c7SGeorge Wilson 	 * dispatched to another taskq thread.
41630f7643c7SGeorge Wilson 	 */
4164f78cdc34SPaul Dagnelie 	zio_allocate_dispatch(zio->io_spa, pio->io_allocator);
41650f7643c7SGeorge Wilson }
41660f7643c7SGeorge Wilson 
4167e14bb325SJeff Bonwick static int
4168e14bb325SJeff Bonwick zio_done(zio_t *zio)
4169d63d470bSgw {
4170e14bb325SJeff Bonwick 	spa_t *spa = zio->io_spa;
4171e14bb325SJeff Bonwick 	zio_t *lio = zio->io_logical;
4172e14bb325SJeff Bonwick 	blkptr_t *bp = zio->io_bp;
4173e14bb325SJeff Bonwick 	vdev_t *vd = zio->io_vd;
4174e14bb325SJeff Bonwick 	uint64_t psize = zio->io_size;
4175a3f829aeSBill Moore 	zio_t *pio, *pio_next;
41760f7643c7SGeorge Wilson 	zio_link_t *zl = NULL;
4177d63d470bSgw 
4178e14bb325SJeff Bonwick 	/*
4179f5383399SBill Moore 	 * If our children haven't all completed,
4180e14bb325SJeff Bonwick 	 * wait for them and then repeat this pipeline stage.
4181e14bb325SJeff Bonwick 	 */
4182d6e1c446SGeorge Wilson 	if (zio_wait_for_children(zio, ZIO_CHILD_ALL_BITS, ZIO_WAIT_DONE)) {
4183e14bb325SJeff Bonwick 		return (ZIO_PIPELINE_STOP);
4184d6e1c446SGeorge Wilson 	}
4185d63d470bSgw 
41860f7643c7SGeorge Wilson 	/*
41870f7643c7SGeorge Wilson 	 * If the allocation throttle is enabled, then update the accounting.
41880f7643c7SGeorge Wilson 	 * We only track child I/Os that are part of an allocating async
41890f7643c7SGeorge Wilson 	 * write. We must do this since the allocation is performed
41900f7643c7SGeorge Wilson 	 * by the logical I/O but the actual write is done by child I/Os.
41910f7643c7SGeorge Wilson 	 */
41920f7643c7SGeorge Wilson 	if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING &&
41930f7643c7SGeorge Wilson 	    zio->io_child_type == ZIO_CHILD_VDEV) {
4194663207adSDon Brady 		ASSERT(zio->io_metaslab_class != NULL);
4195663207adSDon Brady 		ASSERT(zio->io_metaslab_class->mc_alloc_throttle_enabled);
41960f7643c7SGeorge Wilson 		zio_dva_throttle_done(zio);
41970f7643c7SGeorge Wilson 	}
41980f7643c7SGeorge Wilson 
41990f7643c7SGeorge Wilson 	/*
42000f7643c7SGeorge Wilson 	 * If the allocation throttle is enabled, verify that
42010f7643c7SGeorge Wilson 	 * we have decremented the refcounts for every I/O that was throttled.
42020f7643c7SGeorge Wilson 	 */
42030f7643c7SGeorge Wilson 	if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
42040f7643c7SGeorge Wilson 		ASSERT(zio->io_type == ZIO_TYPE_WRITE);
42050f7643c7SGeorge Wilson 		ASSERT(zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
42060f7643c7SGeorge Wilson 		ASSERT(bp != NULL);
4207663207adSDon Brady 
4208f78cdc34SPaul Dagnelie 		metaslab_group_alloc_verify(spa, zio->io_bp, zio,
4209f78cdc34SPaul Dagnelie 		    zio->io_allocator);
4210e914ace2STim Schumacher 		VERIFY(zfs_refcount_not_held(
4211663207adSDon Brady 		    &zio->io_metaslab_class->mc_alloc_slots[zio->io_allocator],
4212663207adSDon Brady 		    zio));
42130f7643c7SGeorge Wilson 	}
42140f7643c7SGeorge Wilson 
4215e14bb325SJeff Bonwick 	for (int c = 0; c < ZIO_CHILD_TYPES; c++)
4216e14bb325SJeff Bonwick 		for (int w = 0; w < ZIO_WAIT_TYPES; w++)
4217e14bb325SJeff Bonwick 			ASSERT(zio->io_children[c][w] == 0);
4218e14bb325SJeff Bonwick 
42195d7b4d43SMatthew Ahrens 	if (bp != NULL && !BP_IS_EMBEDDED(bp)) {
4220e14bb325SJeff Bonwick 		ASSERT(bp->blk_pad[0] == 0);
4221e14bb325SJeff Bonwick 		ASSERT(bp->blk_pad[1] == 0);
4222e14bb325SJeff Bonwick 		ASSERT(bcmp(bp, &zio->io_bp_copy, sizeof (blkptr_t)) == 0 ||
4223a3f829aeSBill Moore 		    (bp == zio_unique_parent(zio)->io_bp));
4224e14bb325SJeff Bonwick 		if (zio->io_type == ZIO_TYPE_WRITE && !BP_IS_HOLE(bp) &&
4225b24ab676SJeff Bonwick 		    zio->io_bp_override == NULL &&
4226e14bb325SJeff Bonwick 		    !(zio->io_flags & ZIO_FLAG_IO_REPAIR)) {
4227b24ab676SJeff Bonwick 			ASSERT3U(zio->io_prop.zp_copies, <=, BP_GET_NDVAS(bp));
4228e14bb325SJeff Bonwick 			ASSERT(BP_COUNT_GANG(bp) == 0 ||
4229e14bb325SJeff Bonwick 			    (BP_COUNT_GANG(bp) == BP_GET_NDVAS(bp)));
4230e14bb325SJeff Bonwick 		}
423180901aeaSGeorge Wilson 		if (zio->io_flags & ZIO_FLAG_NOPWRITE)
423280901aeaSGeorge Wilson 			VERIFY(BP_EQUAL(bp, &zio->io_bp_orig));
4233e14bb325SJeff Bonwick 	}
4234fa9e4066Sahrens 
4235e14bb325SJeff Bonwick 	/*
4236b24ab676SJeff Bonwick 	 * If there were child vdev/gang/ddt errors, they apply to us now.
4237e14bb325SJeff Bonwick 	 */
4238e14bb325SJeff Bonwick 	zio_inherit_child_errors(zio, ZIO_CHILD_VDEV);
4239e14bb325SJeff Bonwick 	zio_inherit_child_errors(zio, ZIO_CHILD_GANG);
4240b24ab676SJeff Bonwick 	zio_inherit_child_errors(zio, ZIO_CHILD_DDT);
4241b24ab676SJeff Bonwick 
4242b24ab676SJeff Bonwick 	/*
4243b24ab676SJeff Bonwick 	 * If the I/O on the transformed data was successful, generate any
4244b24ab676SJeff Bonwick 	 * checksum reports now while we still have the transformed data.
4245b24ab676SJeff Bonwick 	 */
4246b24ab676SJeff Bonwick 	if (zio->io_error == 0) {
4247b24ab676SJeff Bonwick 		while (zio->io_cksum_report != NULL) {
4248b24ab676SJeff Bonwick 			zio_cksum_report_t *zcr = zio->io_cksum_report;
4249b24ab676SJeff Bonwick 			uint64_t align = zcr->zcr_align;
4250b24ab676SJeff Bonwick 			uint64_t asize = P2ROUNDUP(psize, align);
4251770499e1SDan Kimmel 			abd_t *adata = zio->io_abd;
4252b24ab676SJeff Bonwick 
4253b24ab676SJeff Bonwick 			if (asize != psize) {
4254eb633035STom Caputi 				adata = abd_alloc(asize, B_TRUE);
4255770499e1SDan Kimmel 				abd_copy(adata, zio->io_abd, psize);
4256770499e1SDan Kimmel 				abd_zero_off(adata, psize, asize - psize);
4257b24ab676SJeff Bonwick 			}
4258b24ab676SJeff Bonwick 
4259b24ab676SJeff Bonwick 			zio->io_cksum_report = zcr->zcr_next;
4260b24ab676SJeff Bonwick 			zcr->zcr_next = NULL;
4261eb633035STom Caputi 			zcr->zcr_finish(zcr, adata);
4262b24ab676SJeff Bonwick 			zfs_ereport_free_checksum(zcr);
4263b24ab676SJeff Bonwick 
4264b24ab676SJeff Bonwick 			if (asize != psize)
4265770499e1SDan Kimmel 				abd_free(adata);
4266b24ab676SJeff Bonwick 		}
4267b24ab676SJeff Bonwick 	}
4268e14bb325SJeff Bonwick 
4269e14bb325SJeff Bonwick 	zio_pop_transforms(zio);	/* note: may set zio->io_error */
4270e14bb325SJeff Bonwick 
4271e14bb325SJeff Bonwick 	vdev_stat_update(zio, psize);
4272e14bb325SJeff Bonwick 
4273dd50e0ccSTony Hutter 	if (zio->io_delay >= MSEC2NSEC(zio_slow_io_ms)) {
4274dd50e0ccSTony Hutter 		if (zio->io_vd != NULL && !vdev_is_dead(zio->io_vd)) {
4275dd50e0ccSTony Hutter 			/*
4276dd50e0ccSTony Hutter 			 * We want to only increment our slow IO counters if
4277dd50e0ccSTony Hutter 			 * the IO is valid (i.e. not if the drive is removed).
4278dd50e0ccSTony Hutter 			 *
4279dd50e0ccSTony Hutter 			 * zfs_ereport_post() will also do these checks, but
4280dd50e0ccSTony Hutter 			 * it can also have other failures, so we need to
4281dd50e0ccSTony Hutter 			 * increment the slow_io counters independent of it.
4282dd50e0ccSTony Hutter 			 */
4283dd50e0ccSTony Hutter 			if (zfs_ereport_is_valid(FM_EREPORT_ZFS_DELAY,
4284dd50e0ccSTony Hutter 			    zio->io_spa, zio->io_vd, zio)) {
4285dd50e0ccSTony Hutter 				mutex_enter(&zio->io_vd->vdev_stat_lock);
4286dd50e0ccSTony Hutter 				zio->io_vd->vdev_stat.vs_slow_ios++;
4287dd50e0ccSTony Hutter 				mutex_exit(&zio->io_vd->vdev_stat_lock);
4288dd50e0ccSTony Hutter 
42899b088140SToomas Soome 				(void) zfs_ereport_post(FM_EREPORT_ZFS_DELAY,
4290dd50e0ccSTony Hutter 				    zio->io_spa, zio->io_vd, &zio->io_bookmark,
4291dd50e0ccSTony Hutter 				    zio, 0, 0);
4292dd50e0ccSTony Hutter 			}
4293dd50e0ccSTony Hutter 		}
4294dd50e0ccSTony Hutter 	}
4295dd50e0ccSTony Hutter 
4296e14bb325SJeff Bonwick 	if (zio->io_error) {
4297e14bb325SJeff Bonwick 		/*
4298e14bb325SJeff Bonwick 		 * If this I/O is attached to a particular vdev,
4299e14bb325SJeff Bonwick 		 * generate an error message describing the I/O failure
4300e14bb325SJeff Bonwick 		 * at the block level.  We ignore these errors if the
4301e14bb325SJeff Bonwick 		 * device is currently unavailable.
4302e14bb325SJeff Bonwick 		 */
4303e14bb325SJeff Bonwick 		if (zio->io_error != ECKSUM && vd != NULL && !vdev_is_dead(vd))
43049b088140SToomas Soome 			(void) zfs_ereport_post(FM_EREPORT_ZFS_IO, spa, vd,
4305eb633035STom Caputi 			    &zio->io_bookmark, zio, 0, 0);
4306e14bb325SJeff Bonwick 
43078f18d1faSGeorge Wilson 		if ((zio->io_error == EIO || !(zio->io_flags &
43088f18d1faSGeorge Wilson 		    (ZIO_FLAG_SPECULATIVE | ZIO_FLAG_DONT_PROPAGATE))) &&
43098f18d1faSGeorge Wilson 		    zio == lio) {
4310e14bb325SJeff Bonwick 			/*
4311e14bb325SJeff Bonwick 			 * For logical I/O requests, tell the SPA to log the
4312e14bb325SJeff Bonwick 			 * error and generate a logical data ereport.
4313e14bb325SJeff Bonwick 			 */
4314eb633035STom Caputi 			spa_log_error(spa, &zio->io_bookmark);
43159b088140SToomas Soome 			(void) zfs_ereport_post(FM_EREPORT_ZFS_DATA, spa, NULL,
4316eb633035STom Caputi 			    &zio->io_bookmark, zio, 0, 0);
4317e14bb325SJeff Bonwick 		}
4318e14bb325SJeff Bonwick 	}
4319fa9e4066Sahrens 
4320e14bb325SJeff Bonwick 	if (zio->io_error && zio == lio) {
4321e14bb325SJeff Bonwick 		/*
4322e14bb325SJeff Bonwick 		 * Determine whether zio should be reexecuted.  This will
4323e14bb325SJeff Bonwick 		 * propagate all the way to the root via zio_notify_parent().
4324e14bb325SJeff Bonwick 		 */
4325e14bb325SJeff Bonwick 		ASSERT(vd == NULL && bp != NULL);
4326b24ab676SJeff Bonwick 		ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
4327e14bb325SJeff Bonwick 
4328b24ab676SJeff Bonwick 		if (IO_IS_ALLOCATING(zio) &&
4329b24ab676SJeff Bonwick 		    !(zio->io_flags & ZIO_FLAG_CANFAIL)) {
4330e14bb325SJeff Bonwick 			if (zio->io_error != ENOSPC)
4331e14bb325SJeff Bonwick 				zio->io_reexecute |= ZIO_REEXECUTE_NOW;
4332e14bb325SJeff Bonwick 			else
4333e14bb325SJeff Bonwick 				zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
4334b24ab676SJeff Bonwick 		}
4335e14bb325SJeff Bonwick 
4336e14bb325SJeff Bonwick 		if ((zio->io_type == ZIO_TYPE_READ ||
4337e14bb325SJeff Bonwick 		    zio->io_type == ZIO_TYPE_FREE) &&
433844ecc532SGeorge Wilson 		    !(zio->io_flags & ZIO_FLAG_SCAN_THREAD) &&
4339e14bb325SJeff Bonwick 		    zio->io_error == ENXIO &&
4340b16da2e2SGeorge Wilson 		    spa_load_state(spa) == SPA_LOAD_NONE &&
4341e14bb325SJeff Bonwick 		    spa_get_failmode(spa) != ZIO_FAILURE_MODE_CONTINUE)
4342e14bb325SJeff Bonwick 			zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
4343e14bb325SJeff Bonwick 
4344e14bb325SJeff Bonwick 		if (!(zio->io_flags & ZIO_FLAG_CANFAIL) && !zio->io_reexecute)
4345e14bb325SJeff Bonwick 			zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
434622fe2c88SJonathan Adams 
434722fe2c88SJonathan Adams 		/*
434822fe2c88SJonathan Adams 		 * Here is a possibly good place to attempt to do
434922fe2c88SJonathan Adams 		 * either combinatorial reconstruction or error correction
435022fe2c88SJonathan Adams 		 * based on checksums.  It also might be a good place
435122fe2c88SJonathan Adams 		 * to send out preliminary ereports before we suspend
435222fe2c88SJonathan Adams 		 * processing.
435322fe2c88SJonathan Adams 		 */
4354d63d470bSgw 	}
4355d63d470bSgw 
435667bd71c6Sperrin 	/*
4357e14bb325SJeff Bonwick 	 * If there were logical child errors, they apply to us now.
4358e14bb325SJeff Bonwick 	 * We defer this until now to avoid conflating logical child
4359e14bb325SJeff Bonwick 	 * errors with errors that happened to the zio itself when
4360e14bb325SJeff Bonwick 	 * updating vdev stats and reporting FMA events above.
436167bd71c6Sperrin 	 */
4362e14bb325SJeff Bonwick 	zio_inherit_child_errors(zio, ZIO_CHILD_LOGICAL);
43638654d025Sperrin 
4364b24ab676SJeff Bonwick 	if ((zio->io_error || zio->io_reexecute) &&
4365b24ab676SJeff Bonwick 	    IO_IS_ALLOCATING(zio) && zio->io_gang_leader == zio &&
436680901aeaSGeorge Wilson 	    !(zio->io_flags & (ZIO_FLAG_IO_REWRITE | ZIO_FLAG_NOPWRITE)))
4367f5383399SBill Moore 		zio_dva_unallocate(zio, zio->io_gang_tree, bp);
4368f5383399SBill Moore 
4369f5383399SBill Moore 	zio_gang_tree_free(&zio->io_gang_tree);
4370f5383399SBill Moore 
437133a372edSGeorge Wilson 	/*
437233a372edSGeorge Wilson 	 * Godfather I/Os should never suspend.
437333a372edSGeorge Wilson 	 */
437433a372edSGeorge Wilson 	if ((zio->io_flags & ZIO_FLAG_GODFATHER) &&
437533a372edSGeorge Wilson 	    (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND))
437633a372edSGeorge Wilson 		zio->io_reexecute = 0;
437733a372edSGeorge Wilson 
437833a372edSGeorge Wilson 	if (zio->io_reexecute) {
4379e14bb325SJeff Bonwick 		/*
4380e14bb325SJeff Bonwick 		 * This is a logical I/O that wants to reexecute.
4381e14bb325SJeff Bonwick 		 *
4382e14bb325SJeff Bonwick 		 * Reexecute is top-down.  When an i/o fails, if it's not
4383e14bb325SJeff Bonwick 		 * the root, it simply notifies its parent and sticks around.
4384e14bb325SJeff Bonwick 		 * The parent, seeing that it still has children in zio_done(),
4385e14bb325SJeff Bonwick 		 * does the same.  This percolates all the way up to the root.
4386e14bb325SJeff Bonwick 		 * The root i/o will reexecute or suspend the entire tree.
4387e14bb325SJeff Bonwick 		 *
4388e14bb325SJeff Bonwick 		 * This approach ensures that zio_reexecute() honors
4389e14bb325SJeff Bonwick 		 * all the original i/o dependency relationships, e.g.
4390e14bb325SJeff Bonwick 		 * parents not executing until children are ready.
4391e14bb325SJeff Bonwick 		 */
4392e14bb325SJeff Bonwick 		ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
4393fa9e4066Sahrens 
4394f5383399SBill Moore 		zio->io_gang_leader = NULL;
4395e14bb325SJeff Bonwick 
4396a3f829aeSBill Moore 		mutex_enter(&zio->io_lock);
4397a3f829aeSBill Moore 		zio->io_state[ZIO_WAIT_DONE] = 1;
4398a3f829aeSBill Moore 		mutex_exit(&zio->io_lock);
4399a3f829aeSBill Moore 
440054d692b7SGeorge Wilson 		/*
440154d692b7SGeorge Wilson 		 * "The Godfather" I/O monitors its children but is
440254d692b7SGeorge Wilson 		 * not a true parent to them. It will track them through
440354d692b7SGeorge Wilson 		 * the pipeline but severs its ties whenever they get into
440454d692b7SGeorge Wilson 		 * trouble (e.g. suspended). This allows "The Godfather"
440554d692b7SGeorge Wilson 		 * I/O to return status without blocking.
440654d692b7SGeorge Wilson 		 */
44070f7643c7SGeorge Wilson 		zl = NULL;
44080f7643c7SGeorge Wilson 		for (pio = zio_walk_parents(zio, &zl); pio != NULL;
44090f7643c7SGeorge Wilson 		    pio = pio_next) {
44100f7643c7SGeorge Wilson 			zio_link_t *remove_zl = zl;
44110f7643c7SGeorge Wilson 			pio_next = zio_walk_parents(zio, &zl);
441254d692b7SGeorge Wilson 
441354d692b7SGeorge Wilson 			if ((pio->io_flags & ZIO_FLAG_GODFATHER) &&
441454d692b7SGeorge Wilson 			    (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND)) {
44150f7643c7SGeorge Wilson 				zio_remove_child(pio, zio, remove_zl);
441654d692b7SGeorge Wilson 				zio_notify_parent(pio, zio, ZIO_WAIT_DONE);
441754d692b7SGeorge Wilson 			}
441854d692b7SGeorge Wilson 		}
441954d692b7SGeorge Wilson 
4420a3f829aeSBill Moore 		if ((pio = zio_unique_parent(zio)) != NULL) {
4421e14bb325SJeff Bonwick 			/*
4422e14bb325SJeff Bonwick 			 * We're not a root i/o, so there's nothing to do
4423e14bb325SJeff Bonwick 			 * but notify our parent.  Don't propagate errors
4424e14bb325SJeff Bonwick 			 * upward since we haven't permanently failed yet.
4425e14bb325SJeff Bonwick 			 */
442633a372edSGeorge Wilson 			ASSERT(!(zio->io_flags & ZIO_FLAG_GODFATHER));
4427e14bb325SJeff Bonwick 			zio->io_flags |= ZIO_FLAG_DONT_PROPAGATE;
4428e14bb325SJeff Bonwick 			zio_notify_parent(pio, zio, ZIO_WAIT_DONE);
4429e14bb325SJeff Bonwick 		} else if (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND) {
4430e14bb325SJeff Bonwick 			/*
4431e14bb325SJeff Bonwick 			 * We'd fail again if we reexecuted now, so suspend
4432e14bb325SJeff Bonwick 			 * until conditions improve (e.g. device comes online).
4433e14bb325SJeff Bonwick 			 */
4434e0f1c0afSOlaf Faaland 			zio_suspend(zio->io_spa, zio, ZIO_SUSPEND_IOERR);
4435e14bb325SJeff Bonwick 		} else {
4436e14bb325SJeff Bonwick 			/*
4437e14bb325SJeff Bonwick 			 * Reexecution is potentially a huge amount of work.
4438e14bb325SJeff Bonwick 			 * Hand it off to the otherwise-unused claim taskq.
4439e14bb325SJeff Bonwick 			 */
44405aeb9474SGarrett D'Amore 			ASSERT(zio->io_tqent.tqent_next == NULL);
4441ec94d322SAdam Leventhal 			spa_taskq_dispatch_ent(spa, ZIO_TYPE_CLAIM,
4442ec94d322SAdam Leventhal 			    ZIO_TASKQ_ISSUE, (task_func_t *)zio_reexecute, zio,
4443ec94d322SAdam Leventhal 			    0, &zio->io_tqent);
4444e14bb325SJeff Bonwick 		}
4445e14bb325SJeff Bonwick 		return (ZIO_PIPELINE_STOP);
4446fa9e4066Sahrens 	}
4447fa9e4066Sahrens 
4448b24ab676SJeff Bonwick 	ASSERT(zio->io_child_count == 0);
444933a372edSGeorge Wilson 	ASSERT(zio->io_reexecute == 0);
4450e14bb325SJeff Bonwick 	ASSERT(zio->io_error == 0 || (zio->io_flags & ZIO_FLAG_CANFAIL));
4451fa9e4066Sahrens 
4452b24ab676SJeff Bonwick 	/*
4453b24ab676SJeff Bonwick 	 * Report any checksum errors, since the I/O is complete.
4454b24ab676SJeff Bonwick 	 */
445522fe2c88SJonathan Adams 	while (zio->io_cksum_report != NULL) {
4456b24ab676SJeff Bonwick 		zio_cksum_report_t *zcr = zio->io_cksum_report;
4457b24ab676SJeff Bonwick 		zio->io_cksum_report = zcr->zcr_next;
4458b24ab676SJeff Bonwick 		zcr->zcr_next = NULL;
4459b24ab676SJeff Bonwick 		zcr->zcr_finish(zcr, NULL);
4460b24ab676SJeff Bonwick 		zfs_ereport_free_checksum(zcr);
446122fe2c88SJonathan Adams 	}
446222fe2c88SJonathan Adams 
4463a3f829aeSBill Moore 	/*
4464a3f829aeSBill Moore 	 * It is the responsibility of the done callback to ensure that this
4465a3f829aeSBill Moore 	 * particular zio is no longer discoverable for adoption, and as
4466a3f829aeSBill Moore 	 * such, cannot acquire any new parents.
4467a3f829aeSBill Moore 	 */
4468e14bb325SJeff Bonwick 	if (zio->io_done)
4469e14bb325SJeff Bonwick 		zio->io_done(zio);
4470fa9e4066Sahrens 
4471a3f829aeSBill Moore 	mutex_enter(&zio->io_lock);
4472a3f829aeSBill Moore 	zio->io_state[ZIO_WAIT_DONE] = 1;
4473a3f829aeSBill Moore 	mutex_exit(&zio->io_lock);
4474fa9e4066Sahrens 
44750f7643c7SGeorge Wilson 	zl = NULL;
44760f7643c7SGeorge Wilson 	for (pio = zio_walk_parents(zio, &zl); pio != NULL; pio = pio_next) {
44770f7643c7SGeorge Wilson 		zio_link_t *remove_zl = zl;
44780f7643c7SGeorge Wilson 		pio_next = zio_walk_parents(zio, &zl);
44790f7643c7SGeorge Wilson 		zio_remove_child(pio, zio, remove_zl);
4480e14bb325SJeff Bonwick 		zio_notify_parent(pio, zio, ZIO_WAIT_DONE);
4481e14bb325SJeff Bonwick 	}
4482fa9e4066Sahrens 
4483e14bb325SJeff Bonwick 	if (zio->io_waiter != NULL) {
4484e14bb325SJeff Bonwick 		mutex_enter(&zio->io_lock);
4485e14bb325SJeff Bonwick 		zio->io_executor = NULL;
4486e14bb325SJeff Bonwick 		cv_broadcast(&zio->io_cv);
4487e14bb325SJeff Bonwick 		mutex_exit(&zio->io_lock);
4488e14bb325SJeff Bonwick 	} else {
4489e14bb325SJeff Bonwick 		zio_destroy(zio);
4490e14bb325SJeff Bonwick 	}
4491fa9e4066Sahrens 
4492e14bb325SJeff Bonwick 	return (ZIO_PIPELINE_STOP);
4493fa9e4066Sahrens }
449446341222Sperrin 
449546341222Sperrin /*
4496e14bb325SJeff Bonwick  * ==========================================================================
4497e14bb325SJeff Bonwick  * I/O pipeline definition
4498e14bb325SJeff Bonwick  * ==========================================================================
449946341222Sperrin  */
4500b24ab676SJeff Bonwick static zio_pipe_stage_t *zio_pipeline[] = {
4501e14bb325SJeff Bonwick 	NULL,
4502e14bb325SJeff Bonwick 	zio_read_bp_init,
45030f7643c7SGeorge Wilson 	zio_write_bp_init,
4504b24ab676SJeff Bonwick 	zio_free_bp_init,
4505b24ab676SJeff Bonwick 	zio_issue_async,
45060f7643c7SGeorge Wilson 	zio_write_compress,
4507eb633035STom Caputi 	zio_encrypt,
4508e14bb325SJeff Bonwick 	zio_checksum_generate,
450980901aeaSGeorge Wilson 	zio_nop_write,
4510b24ab676SJeff Bonwick 	zio_ddt_read_start,
4511b24ab676SJeff Bonwick 	zio_ddt_read_done,
4512b24ab676SJeff Bonwick 	zio_ddt_write,
4513b24ab676SJeff Bonwick 	zio_ddt_free,
4514e14bb325SJeff Bonwick 	zio_gang_assemble,
4515e14bb325SJeff Bonwick 	zio_gang_issue,
45160f7643c7SGeorge Wilson 	zio_dva_throttle,
4517e14bb325SJeff Bonwick 	zio_dva_allocate,
4518e14bb325SJeff Bonwick 	zio_dva_free,
4519e14bb325SJeff Bonwick 	zio_dva_claim,
4520e14bb325SJeff Bonwick 	zio_ready,
4521e14bb325SJeff Bonwick 	zio_vdev_io_start,
4522e14bb325SJeff Bonwick 	zio_vdev_io_done,
4523e14bb325SJeff Bonwick 	zio_vdev_io_assess,
4524e14bb325SJeff Bonwick 	zio_checksum_verify,
4525e14bb325SJeff Bonwick 	zio_done
4526e14bb325SJeff Bonwick };
4527ad135b5dSChristopher Siden 
4528ad135b5dSChristopher Siden 
4529ad135b5dSChristopher Siden 
4530ad135b5dSChristopher Siden 
4531a2cdcdd2SPaul Dagnelie /*
4532a2cdcdd2SPaul Dagnelie  * Compare two zbookmark_phys_t's to see which we would reach first in a
4533a2cdcdd2SPaul Dagnelie  * pre-order traversal of the object tree.
4534a2cdcdd2SPaul Dagnelie  *
4535a2cdcdd2SPaul Dagnelie  * This is simple in every case aside from the meta-dnode object. For all other
4536a2cdcdd2SPaul Dagnelie  * objects, we traverse them in order (object 1 before object 2, and so on).
4537a2cdcdd2SPaul Dagnelie  * However, all of these objects are traversed while traversing object 0, since
4538a2cdcdd2SPaul Dagnelie  * the data it points to is the list of objects.  Thus, we need to convert to a
4539a2cdcdd2SPaul Dagnelie  * canonical representation so we can compare meta-dnode bookmarks to
4540a2cdcdd2SPaul Dagnelie  * non-meta-dnode bookmarks.
4541a2cdcdd2SPaul Dagnelie  *
4542a2cdcdd2SPaul Dagnelie  * We do this by calculating "equivalents" for each field of the zbookmark.
4543a2cdcdd2SPaul Dagnelie  * zbookmarks outside of the meta-dnode use their own object and level, and
4544a2cdcdd2SPaul Dagnelie  * calculate the level 0 equivalent (the first L0 blkid that is contained in the
4545a2cdcdd2SPaul Dagnelie  * blocks this bookmark refers to) by multiplying their blkid by their span
4546a2cdcdd2SPaul Dagnelie  * (the number of L0 blocks contained within one block at their level).
4547a2cdcdd2SPaul Dagnelie  * zbookmarks inside the meta-dnode calculate their object equivalent
4548a2cdcdd2SPaul Dagnelie  * (which is L0equiv * dnodes per data block), use 0 for their L0equiv, and use
4549a2cdcdd2SPaul Dagnelie  * level + 1<<31 (any value larger than a level could ever be) for their level.
4550a2cdcdd2SPaul Dagnelie  * This causes them to always compare before a bookmark in their object
4551a2cdcdd2SPaul Dagnelie  * equivalent, compare appropriately to bookmarks in other objects, and to
4552a2cdcdd2SPaul Dagnelie  * compare appropriately to other bookmarks in the meta-dnode.
4553a2cdcdd2SPaul Dagnelie  */
4554a2cdcdd2SPaul Dagnelie int
4555a2cdcdd2SPaul Dagnelie zbookmark_compare(uint16_t dbss1, uint8_t ibs1, uint16_t dbss2, uint8_t ibs2,
4556a2cdcdd2SPaul Dagnelie     const zbookmark_phys_t *zb1, const zbookmark_phys_t *zb2)
4557a2cdcdd2SPaul Dagnelie {
4558a2cdcdd2SPaul Dagnelie 	/*
4559a2cdcdd2SPaul Dagnelie 	 * These variables represent the "equivalent" values for the zbookmark,
4560a2cdcdd2SPaul Dagnelie 	 * after converting zbookmarks inside the meta dnode to their
4561a2cdcdd2SPaul Dagnelie 	 * normal-object equivalents.
4562a2cdcdd2SPaul Dagnelie 	 */
4563a2cdcdd2SPaul Dagnelie 	uint64_t zb1obj, zb2obj;
4564a2cdcdd2SPaul Dagnelie 	uint64_t zb1L0, zb2L0;
4565a2cdcdd2SPaul Dagnelie 	uint64_t zb1level, zb2level;
4566ad135b5dSChristopher Siden 
4567a2cdcdd2SPaul Dagnelie 	if (zb1->zb_object == zb2->zb_object &&
4568a2cdcdd2SPaul Dagnelie 	    zb1->zb_level == zb2->zb_level &&
4569a2cdcdd2SPaul Dagnelie 	    zb1->zb_blkid == zb2->zb_blkid)
4570a2cdcdd2SPaul Dagnelie 		return (0);
4571a2cdcdd2SPaul Dagnelie 
4572a2cdcdd2SPaul Dagnelie 	/*
4573a2cdcdd2SPaul Dagnelie 	 * BP_SPANB calculates the span in blocks.
4574a2cdcdd2SPaul Dagnelie 	 */
4575a2cdcdd2SPaul Dagnelie 	zb1L0 = (zb1->zb_blkid) * BP_SPANB(ibs1, zb1->zb_level);
4576a2cdcdd2SPaul Dagnelie 	zb2L0 = (zb2->zb_blkid) * BP_SPANB(ibs2, zb2->zb_level);
4577ad135b5dSChristopher Siden 
4578ad135b5dSChristopher Siden 	if (zb1->zb_object == DMU_META_DNODE_OBJECT) {
4579a2cdcdd2SPaul Dagnelie 		zb1obj = zb1L0 * (dbss1 << (SPA_MINBLOCKSHIFT - DNODE_SHIFT));
4580a2cdcdd2SPaul Dagnelie 		zb1L0 = 0;
4581a2cdcdd2SPaul Dagnelie 		zb1level = zb1->zb_level + COMPARE_META_LEVEL;
4582a2cdcdd2SPaul Dagnelie 	} else {
4583a2cdcdd2SPaul Dagnelie 		zb1obj = zb1->zb_object;
4584a2cdcdd2SPaul Dagnelie 		zb1level = zb1->zb_level;
4585ad135b5dSChristopher Siden 	}
4586ad135b5dSChristopher Siden 
4587a2cdcdd2SPaul Dagnelie 	if (zb2->zb_object == DMU_META_DNODE_OBJECT) {
4588a2cdcdd2SPaul Dagnelie 		zb2obj = zb2L0 * (dbss2 << (SPA_MINBLOCKSHIFT - DNODE_SHIFT));
4589a2cdcdd2SPaul Dagnelie 		zb2L0 = 0;
4590a2cdcdd2SPaul Dagnelie 		zb2level = zb2->zb_level + COMPARE_META_LEVEL;
4591a2cdcdd2SPaul Dagnelie 	} else {
4592a2cdcdd2SPaul Dagnelie 		zb2obj = zb2->zb_object;
4593a2cdcdd2SPaul Dagnelie 		zb2level = zb2->zb_level;
4594a2cdcdd2SPaul Dagnelie 	}
4595a2cdcdd2SPaul Dagnelie 
4596a2cdcdd2SPaul Dagnelie 	/* Now that we have a canonical representation, do the comparison. */
4597a2cdcdd2SPaul Dagnelie 	if (zb1obj != zb2obj)
4598a2cdcdd2SPaul Dagnelie 		return (zb1obj < zb2obj ? -1 : 1);
4599a2cdcdd2SPaul Dagnelie 	else if (zb1L0 != zb2L0)
4600a2cdcdd2SPaul Dagnelie 		return (zb1L0 < zb2L0 ? -1 : 1);
4601a2cdcdd2SPaul Dagnelie 	else if (zb1level != zb2level)
4602a2cdcdd2SPaul Dagnelie 		return (zb1level > zb2level ? -1 : 1);
4603a2cdcdd2SPaul Dagnelie 	/*
4604a2cdcdd2SPaul Dagnelie 	 * This can (theoretically) happen if the bookmarks have the same object
4605a2cdcdd2SPaul Dagnelie 	 * and level, but different blkids, if the block sizes are not the same.
4606a2cdcdd2SPaul Dagnelie 	 * There is presently no way to change the indirect block sizes
4607a2cdcdd2SPaul Dagnelie 	 */
4608a2cdcdd2SPaul Dagnelie 	return (0);
4609a2cdcdd2SPaul Dagnelie }
4610a2cdcdd2SPaul Dagnelie 
4611a2cdcdd2SPaul Dagnelie /*
4612a2cdcdd2SPaul Dagnelie  *  This function checks the following: given that last_block is the place that
4613a2cdcdd2SPaul Dagnelie  *  our traversal stopped last time, does that guarantee that we've visited
4614a2cdcdd2SPaul Dagnelie  *  every node under subtree_root?  Therefore, we can't just use the raw output
4615a2cdcdd2SPaul Dagnelie  *  of zbookmark_compare.  We have to pass in a modified version of
4616a2cdcdd2SPaul Dagnelie  *  subtree_root; by incrementing the block id, and then checking whether
4617a2cdcdd2SPaul Dagnelie  *  last_block is before or equal to that, we can tell whether or not having
4618a2cdcdd2SPaul Dagnelie  *  visited last_block implies that all of subtree_root's children have been
4619a2cdcdd2SPaul Dagnelie  *  visited.
4620a2cdcdd2SPaul Dagnelie  */
4621a2cdcdd2SPaul Dagnelie boolean_t
4622a2cdcdd2SPaul Dagnelie zbookmark_subtree_completed(const dnode_phys_t *dnp,
4623a2cdcdd2SPaul Dagnelie     const zbookmark_phys_t *subtree_root, const zbookmark_phys_t *last_block)
4624a2cdcdd2SPaul Dagnelie {
4625a2cdcdd2SPaul Dagnelie 	zbookmark_phys_t mod_zb = *subtree_root;
4626a2cdcdd2SPaul Dagnelie 	mod_zb.zb_blkid++;
4627a2cdcdd2SPaul Dagnelie 	ASSERT(last_block->zb_level == 0);
4628a2cdcdd2SPaul Dagnelie 
4629a2cdcdd2SPaul Dagnelie 	/* The objset_phys_t isn't before anything. */
4630a2cdcdd2SPaul Dagnelie 	if (dnp == NULL)
4631ad135b5dSChristopher Siden 		return (B_FALSE);
4632a2cdcdd2SPaul Dagnelie 
4633a2cdcdd2SPaul Dagnelie 	/*
4634a2cdcdd2SPaul Dagnelie 	 * We pass in 1ULL << (DNODE_BLOCK_SHIFT - SPA_MINBLOCKSHIFT) for the
4635a2cdcdd2SPaul Dagnelie 	 * data block size in sectors, because that variable is only used if
4636a2cdcdd2SPaul Dagnelie 	 * the bookmark refers to a block in the meta-dnode.  Since we don't
4637a2cdcdd2SPaul Dagnelie 	 * know without examining it what object it refers to, and there's no
4638a2cdcdd2SPaul Dagnelie 	 * harm in passing in this value in other cases, we always pass it in.
4639a2cdcdd2SPaul Dagnelie 	 *
4640a2cdcdd2SPaul Dagnelie 	 * We pass in 0 for the indirect block size shift because zb2 must be
4641a2cdcdd2SPaul Dagnelie 	 * level 0.  The indirect block size is only used to calculate the span
4642a2cdcdd2SPaul Dagnelie 	 * of the bookmark, but since the bookmark must be level 0, the span is
4643a2cdcdd2SPaul Dagnelie 	 * always 1, so the math works out.
4644a2cdcdd2SPaul Dagnelie 	 *
4645a2cdcdd2SPaul Dagnelie 	 * If you make changes to how the zbookmark_compare code works, be sure
4646a2cdcdd2SPaul Dagnelie 	 * to make sure that this code still works afterwards.
4647a2cdcdd2SPaul Dagnelie 	 */
4648a2cdcdd2SPaul Dagnelie 	return (zbookmark_compare(dnp->dn_datablkszsec, dnp->dn_indblkshift,
4649a2cdcdd2SPaul Dagnelie 	    1ULL << (DNODE_BLOCK_SHIFT - SPA_MINBLOCKSHIFT), 0, &mod_zb,
4650a2cdcdd2SPaul Dagnelie 	    last_block) <= 0);
4651ad135b5dSChristopher Siden }
4652