]> git.proxmox.com Git - mirror_zfs.git/blame - module/zfs/zio.c
ZFS mounted NFSv3 shares fail lock reclaims
[mirror_zfs.git] / module / zfs / zio.c
CommitLineData
34dc7c2f
BB
1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21/*
428870ff 22 * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
492f64e9 23 * Copyright (c) 2011, 2018 by Delphix. All rights reserved.
a38718a6 24 * Copyright (c) 2011 Nexenta Systems, Inc. All rights reserved.
cc99f275 25 * Copyright (c) 2017, Intel Corporation.
34dc7c2f
BB
26 */
27
f1512ee6 28#include <sys/sysmacros.h>
34dc7c2f
BB
29#include <sys/zfs_context.h>
30#include <sys/fm/fs/zfs.h>
31#include <sys/spa.h>
32#include <sys/txg.h>
33#include <sys/spa_impl.h>
34#include <sys/vdev_impl.h>
35#include <sys/zio_impl.h>
36#include <sys/zio_compress.h>
37#include <sys/zio_checksum.h>
428870ff
BB
38#include <sys/dmu_objset.h>
39#include <sys/arc.h>
40#include <sys/ddt.h>
9b67f605 41#include <sys/blkptr.h>
b0bc7a84 42#include <sys/zfeature.h>
d4a72f23 43#include <sys/dsl_scan.h>
3dfb57a3 44#include <sys/metaslab_impl.h>
193a37cb 45#include <sys/time.h>
26ef0cc7 46#include <sys/trace_zio.h>
a6255b7f 47#include <sys/abd.h>
b5256303 48#include <sys/dsl_crypt.h>
492f64e9 49#include <sys/cityhash.h>
34dc7c2f 50
34dc7c2f
BB
51/*
52 * ==========================================================================
53 * I/O type descriptions
54 * ==========================================================================
55 */
e8b96c60 56const char *zio_type_name[ZIO_TYPES] = {
3dfb57a3
DB
57 /*
58 * Note: Linux kernel thread name length is limited
59 * so these names will differ from upstream open zfs.
60 */
451041db 61 "z_null", "z_rd", "z_wr", "z_fr", "z_cl", "z_ioctl"
428870ff 62};
34dc7c2f 63
27f2b90d 64int zio_dva_throttle_enabled = B_TRUE;
638dd5f4 65int zio_deadman_log_all = B_FALSE;
3dfb57a3 66
34dc7c2f
BB
67/*
68 * ==========================================================================
69 * I/O kmem caches
70 * ==========================================================================
71 */
72kmem_cache_t *zio_cache;
d164b209 73kmem_cache_t *zio_link_cache;
34dc7c2f
BB
74kmem_cache_t *zio_buf_cache[SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT];
75kmem_cache_t *zio_data_buf_cache[SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT];
a6255b7f
DQ
76#if defined(ZFS_DEBUG) && !defined(_KERNEL)
77uint64_t zio_buf_cache_allocs[SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT];
78uint64_t zio_buf_cache_frees[SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT];
79#endif
80
ad796b8a
TH
81/* Mark IOs as "slow" if they take longer than 30 seconds */
82int zio_slow_io_ms = (30 * MILLISEC);
34dc7c2f 83
fcff0f35
PD
84#define BP_SPANB(indblkshift, level) \
85 (((uint64_t)1) << ((level) * ((indblkshift) - SPA_BLKPTRSHIFT)))
86#define COMPARE_META_LEVEL 0x80000000ul
55d85d5a
GW
87/*
88 * The following actions directly effect the spa's sync-to-convergence logic.
89 * The values below define the sync pass when we start performing the action.
90 * Care should be taken when changing these values as they directly impact
91 * spa_sync() performance. Tuning these values may introduce subtle performance
92 * pathologies and should only be done in the context of performance analysis.
93 * These tunables will eventually be removed and replaced with #defines once
94 * enough analysis has been done to determine optimal values.
95 *
96 * The 'zfs_sync_pass_deferred_free' pass must be greater than 1 to ensure that
97 * regular blocks are not deferred.
98 */
99int zfs_sync_pass_deferred_free = 2; /* defer frees starting in this pass */
100int zfs_sync_pass_dont_compress = 5; /* don't compress starting in this pass */
101int zfs_sync_pass_rewrite = 2; /* rewrite new bps starting in this pass */
102
34dc7c2f 103/*
b128c09f
BB
104 * An allocating zio is one that either currently has the DVA allocate
105 * stage set or will have it later in its lifetime.
34dc7c2f 106 */
428870ff
BB
107#define IO_IS_ALLOCATING(zio) ((zio)->io_orig_pipeline & ZIO_STAGE_DVA_ALLOCATE)
108
c409e464 109int zio_requeue_io_start_cut_in_line = 1;
428870ff
BB
110
111#ifdef ZFS_DEBUG
112int zio_buf_debug_limit = 16384;
113#else
114int zio_buf_debug_limit = 0;
115#endif
34dc7c2f 116
da6b4005
NB
117static inline void __zio_execute(zio_t *zio);
118
3dfb57a3
DB
119static void zio_taskq_dispatch(zio_t *, zio_taskq_type_t, boolean_t);
120
34dc7c2f
BB
121void
122zio_init(void)
123{
124 size_t c;
125 vmem_t *data_alloc_arena = NULL;
126
3941503c
BB
127 zio_cache = kmem_cache_create("zio_cache",
128 sizeof (zio_t), 0, NULL, NULL, NULL, NULL, NULL, 0);
d164b209 129 zio_link_cache = kmem_cache_create("zio_link_cache",
6795a698 130 sizeof (zio_link_t), 0, NULL, NULL, NULL, NULL, NULL, 0);
34dc7c2f
BB
131
132 /*
133 * For small buffers, we want a cache for each multiple of
f1512ee6
MA
134 * SPA_MINBLOCKSIZE. For larger buffers, we want a cache
135 * for each quarter-power of 2.
34dc7c2f
BB
136 */
137 for (c = 0; c < SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT; c++) {
138 size_t size = (c + 1) << SPA_MINBLOCKSHIFT;
139 size_t p2 = size;
140 size_t align = 0;
6442f3cf 141 size_t cflags = (size > zio_buf_debug_limit) ? KMC_NODEBUG : 0;
34dc7c2f 142
34328f3c 143#if defined(_ILP32) && defined(_KERNEL)
f1512ee6
MA
144 /*
145 * Cache size limited to 1M on 32-bit platforms until ARC
146 * buffers no longer require virtual address space.
147 */
148 if (size > zfs_max_recordsize)
149 break;
150#endif
151
152 while (!ISP2(p2))
34dc7c2f
BB
153 p2 &= p2 - 1;
154
498877ba
MA
155#ifndef _KERNEL
156 /*
157 * If we are using watchpoints, put each buffer on its own page,
158 * to eliminate the performance overhead of trapping to the
159 * kernel when modifying a non-watched buffer that shares the
160 * page with a watched buffer.
161 */
162 if (arc_watch && !IS_P2ALIGNED(size, PAGESIZE))
163 continue;
fcf64f45
BB
164 /*
165 * Here's the problem - on 4K native devices in userland on
166 * Linux using O_DIRECT, buffers must be 4K aligned or I/O
167 * will fail with EINVAL, causing zdb (and others) to coredump.
168 * Since userland probably doesn't need optimized buffer caches,
169 * we just force 4K alignment on everything.
170 */
171 align = 8 * SPA_MINBLOCKSIZE;
172#else
24fa2034 173 if (size < PAGESIZE) {
34dc7c2f 174 align = SPA_MINBLOCKSIZE;
498877ba 175 } else if (IS_P2ALIGNED(size, p2 >> 2)) {
24fa2034 176 align = PAGESIZE;
34dc7c2f 177 }
fcf64f45 178#endif
34dc7c2f
BB
179
180 if (align != 0) {
181 char name[36];
182 (void) sprintf(name, "zio_buf_%lu", (ulong_t)size);
183 zio_buf_cache[c] = kmem_cache_create(name, size,
6442f3cf 184 align, NULL, NULL, NULL, NULL, NULL, cflags);
34dc7c2f
BB
185
186 (void) sprintf(name, "zio_data_buf_%lu", (ulong_t)size);
187 zio_data_buf_cache[c] = kmem_cache_create(name, size,
ae6ba3db 188 align, NULL, NULL, NULL, NULL,
6442f3cf 189 data_alloc_arena, cflags);
34dc7c2f
BB
190 }
191 }
192
193 while (--c != 0) {
194 ASSERT(zio_buf_cache[c] != NULL);
195 if (zio_buf_cache[c - 1] == NULL)
196 zio_buf_cache[c - 1] = zio_buf_cache[c];
197
198 ASSERT(zio_data_buf_cache[c] != NULL);
199 if (zio_data_buf_cache[c - 1] == NULL)
200 zio_data_buf_cache[c - 1] = zio_data_buf_cache[c];
201 }
202
34dc7c2f 203 zio_inject_init();
9759c60f
ED
204
205 lz4_init();
34dc7c2f
BB
206}
207
208void
209zio_fini(void)
210{
211 size_t c;
212 kmem_cache_t *last_cache = NULL;
213 kmem_cache_t *last_data_cache = NULL;
214
215 for (c = 0; c < SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT; c++) {
f1512ee6
MA
216#ifdef _ILP32
217 /*
218 * Cache size limited to 1M on 32-bit platforms until ARC
219 * buffers no longer require virtual address space.
220 */
221 if (((c + 1) << SPA_MINBLOCKSHIFT) > zfs_max_recordsize)
222 break;
a6255b7f
DQ
223#endif
224#if defined(ZFS_DEBUG) && !defined(_KERNEL)
225 if (zio_buf_cache_allocs[c] != zio_buf_cache_frees[c])
226 (void) printf("zio_fini: [%d] %llu != %llu\n",
227 (int)((c + 1) << SPA_MINBLOCKSHIFT),
228 (long long unsigned)zio_buf_cache_allocs[c],
229 (long long unsigned)zio_buf_cache_frees[c]);
f1512ee6 230#endif
34dc7c2f
BB
231 if (zio_buf_cache[c] != last_cache) {
232 last_cache = zio_buf_cache[c];
233 kmem_cache_destroy(zio_buf_cache[c]);
234 }
235 zio_buf_cache[c] = NULL;
236
237 if (zio_data_buf_cache[c] != last_data_cache) {
238 last_data_cache = zio_data_buf_cache[c];
239 kmem_cache_destroy(zio_data_buf_cache[c]);
240 }
241 zio_data_buf_cache[c] = NULL;
242 }
243
d164b209 244 kmem_cache_destroy(zio_link_cache);
34dc7c2f
BB
245 kmem_cache_destroy(zio_cache);
246
247 zio_inject_fini();
9759c60f
ED
248
249 lz4_fini();
34dc7c2f
BB
250}
251
252/*
253 * ==========================================================================
254 * Allocate and free I/O buffers
255 * ==========================================================================
256 */
257
258/*
259 * Use zio_buf_alloc to allocate ZFS metadata. This data will appear in a
260 * crashdump if the kernel panics, so use it judiciously. Obviously, it's
261 * useful to inspect ZFS metadata, but if possible, we should avoid keeping
262 * excess / transient data in-core during a crashdump.
263 */
264void *
265zio_buf_alloc(size_t size)
266{
267 size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
268
63e3a861 269 VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
a6255b7f
DQ
270#if defined(ZFS_DEBUG) && !defined(_KERNEL)
271 atomic_add_64(&zio_buf_cache_allocs[c], 1);
272#endif
34dc7c2f 273
efcd79a8 274 return (kmem_cache_alloc(zio_buf_cache[c], KM_PUSHPAGE));
34dc7c2f
BB
275}
276
277/*
278 * Use zio_data_buf_alloc to allocate data. The data will not appear in a
279 * crashdump if the kernel panics. This exists so that we will limit the amount
280 * of ZFS data that shows up in a kernel crashdump. (Thus reducing the amount
281 * of kernel heap dumped to disk when the kernel panics)
282 */
283void *
284zio_data_buf_alloc(size_t size)
285{
286 size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
287
63e3a861 288 VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
34dc7c2f 289
efcd79a8 290 return (kmem_cache_alloc(zio_data_buf_cache[c], KM_PUSHPAGE));
34dc7c2f
BB
291}
292
293void
294zio_buf_free(void *buf, size_t size)
295{
296 size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
297
63e3a861 298 VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
a6255b7f
DQ
299#if defined(ZFS_DEBUG) && !defined(_KERNEL)
300 atomic_add_64(&zio_buf_cache_frees[c], 1);
301#endif
34dc7c2f
BB
302
303 kmem_cache_free(zio_buf_cache[c], buf);
304}
305
306void
307zio_data_buf_free(void *buf, size_t size)
308{
309 size_t c = (size - 1) >> SPA_MINBLOCKSHIFT;
310
63e3a861 311 VERIFY3U(c, <, SPA_MAXBLOCKSIZE >> SPA_MINBLOCKSHIFT);
34dc7c2f
BB
312
313 kmem_cache_free(zio_data_buf_cache[c], buf);
314}
315
84c07ada
GN
316static void
317zio_abd_free(void *abd, size_t size)
318{
319 abd_free((abd_t *)abd);
320}
321
34dc7c2f
BB
322/*
323 * ==========================================================================
324 * Push and pop I/O transform buffers
325 * ==========================================================================
326 */
d3c2ae1c 327void
a6255b7f 328zio_push_transform(zio_t *zio, abd_t *data, uint64_t size, uint64_t bufsize,
e9aa730c 329 zio_transform_func_t *transform)
34dc7c2f 330{
79c76d5b 331 zio_transform_t *zt = kmem_alloc(sizeof (zio_transform_t), KM_SLEEP);
34dc7c2f 332
a6255b7f
DQ
333 /*
334 * Ensure that anyone expecting this zio to contain a linear ABD isn't
335 * going to get a nasty surprise when they try to access the data.
336 */
337 IMPLY(abd_is_linear(zio->io_abd), abd_is_linear(data));
338
339 zt->zt_orig_abd = zio->io_abd;
b128c09f 340 zt->zt_orig_size = zio->io_size;
34dc7c2f 341 zt->zt_bufsize = bufsize;
b128c09f 342 zt->zt_transform = transform;
34dc7c2f
BB
343
344 zt->zt_next = zio->io_transform_stack;
345 zio->io_transform_stack = zt;
346
a6255b7f 347 zio->io_abd = data;
34dc7c2f
BB
348 zio->io_size = size;
349}
350
d3c2ae1c 351void
b128c09f 352zio_pop_transforms(zio_t *zio)
34dc7c2f 353{
b128c09f
BB
354 zio_transform_t *zt;
355
356 while ((zt = zio->io_transform_stack) != NULL) {
357 if (zt->zt_transform != NULL)
358 zt->zt_transform(zio,
a6255b7f 359 zt->zt_orig_abd, zt->zt_orig_size);
34dc7c2f 360
428870ff 361 if (zt->zt_bufsize != 0)
a6255b7f 362 abd_free(zio->io_abd);
34dc7c2f 363
a6255b7f 364 zio->io_abd = zt->zt_orig_abd;
b128c09f
BB
365 zio->io_size = zt->zt_orig_size;
366 zio->io_transform_stack = zt->zt_next;
34dc7c2f 367
b128c09f 368 kmem_free(zt, sizeof (zio_transform_t));
34dc7c2f
BB
369 }
370}
371
b128c09f
BB
372/*
373 * ==========================================================================
b5256303 374 * I/O transform callbacks for subblocks, decompression, and decryption
b128c09f
BB
375 * ==========================================================================
376 */
377static void
a6255b7f 378zio_subblock(zio_t *zio, abd_t *data, uint64_t size)
b128c09f
BB
379{
380 ASSERT(zio->io_size > size);
381
382 if (zio->io_type == ZIO_TYPE_READ)
a6255b7f 383 abd_copy(data, zio->io_abd, size);
b128c09f
BB
384}
385
386static void
a6255b7f 387zio_decompress(zio_t *zio, abd_t *data, uint64_t size)
b128c09f 388{
a6255b7f
DQ
389 if (zio->io_error == 0) {
390 void *tmp = abd_borrow_buf(data, size);
391 int ret = zio_decompress_data(BP_GET_COMPRESS(zio->io_bp),
392 zio->io_abd, tmp, zio->io_size, size);
393 abd_return_buf_copy(data, tmp, size);
394
c3bd3fb4
TC
395 if (zio_injection_enabled && ret == 0)
396 ret = zio_handle_fault_injection(zio, EINVAL);
397
a6255b7f
DQ
398 if (ret != 0)
399 zio->io_error = SET_ERROR(EIO);
400 }
b128c09f
BB
401}
402
b5256303
TC
403static void
404zio_decrypt(zio_t *zio, abd_t *data, uint64_t size)
405{
406 int ret;
407 void *tmp;
408 blkptr_t *bp = zio->io_bp;
ae76f45c
TC
409 spa_t *spa = zio->io_spa;
410 uint64_t dsobj = zio->io_bookmark.zb_objset;
b5256303
TC
411 uint64_t lsize = BP_GET_LSIZE(bp);
412 dmu_object_type_t ot = BP_GET_TYPE(bp);
413 uint8_t salt[ZIO_DATA_SALT_LEN];
414 uint8_t iv[ZIO_DATA_IV_LEN];
415 uint8_t mac[ZIO_DATA_MAC_LEN];
416 boolean_t no_crypt = B_FALSE;
417
418 ASSERT(BP_USES_CRYPT(bp));
419 ASSERT3U(size, !=, 0);
420
421 if (zio->io_error != 0)
422 return;
423
424 /*
425 * Verify the cksum of MACs stored in an indirect bp. It will always
426 * be possible to verify this since it does not require an encryption
427 * key.
428 */
429 if (BP_HAS_INDIRECT_MAC_CKSUM(bp)) {
430 zio_crypt_decode_mac_bp(bp, mac);
431
432 if (BP_GET_COMPRESS(bp) != ZIO_COMPRESS_OFF) {
433 /*
434 * We haven't decompressed the data yet, but
435 * zio_crypt_do_indirect_mac_checksum() requires
436 * decompressed data to be able to parse out the MACs
437 * from the indirect block. We decompress it now and
438 * throw away the result after we are finished.
439 */
440 tmp = zio_buf_alloc(lsize);
441 ret = zio_decompress_data(BP_GET_COMPRESS(bp),
442 zio->io_abd, tmp, zio->io_size, lsize);
443 if (ret != 0) {
444 ret = SET_ERROR(EIO);
445 goto error;
446 }
447 ret = zio_crypt_do_indirect_mac_checksum(B_FALSE,
448 tmp, lsize, BP_SHOULD_BYTESWAP(bp), mac);
449 zio_buf_free(tmp, lsize);
450 } else {
451 ret = zio_crypt_do_indirect_mac_checksum_abd(B_FALSE,
452 zio->io_abd, size, BP_SHOULD_BYTESWAP(bp), mac);
453 }
454 abd_copy(data, zio->io_abd, size);
455
be9a5c35
TC
456 if (zio_injection_enabled && ot != DMU_OT_DNODE && ret == 0) {
457 ret = zio_handle_decrypt_injection(spa,
458 &zio->io_bookmark, ot, ECKSUM);
459 }
b5256303
TC
460 if (ret != 0)
461 goto error;
462
463 return;
464 }
465
466 /*
467 * If this is an authenticated block, just check the MAC. It would be
468 * nice to separate this out into its own flag, but for the moment
469 * enum zio_flag is out of bits.
470 */
471 if (BP_IS_AUTHENTICATED(bp)) {
472 if (ot == DMU_OT_OBJSET) {
ae76f45c
TC
473 ret = spa_do_crypt_objset_mac_abd(B_FALSE, spa,
474 dsobj, zio->io_abd, size, BP_SHOULD_BYTESWAP(bp));
b5256303
TC
475 } else {
476 zio_crypt_decode_mac_bp(bp, mac);
ae76f45c
TC
477 ret = spa_do_crypt_mac_abd(B_FALSE, spa, dsobj,
478 zio->io_abd, size, mac);
be9a5c35
TC
479 if (zio_injection_enabled && ret == 0) {
480 ret = zio_handle_decrypt_injection(spa,
481 &zio->io_bookmark, ot, ECKSUM);
482 }
b5256303
TC
483 }
484 abd_copy(data, zio->io_abd, size);
485
486 if (ret != 0)
487 goto error;
488
489 return;
490 }
491
492 zio_crypt_decode_params_bp(bp, salt, iv);
493
494 if (ot == DMU_OT_INTENT_LOG) {
495 tmp = abd_borrow_buf_copy(zio->io_abd, sizeof (zil_chain_t));
496 zio_crypt_decode_mac_zil(tmp, mac);
497 abd_return_buf(zio->io_abd, tmp, sizeof (zil_chain_t));
498 } else {
499 zio_crypt_decode_mac_bp(bp, mac);
500 }
501
be9a5c35
TC
502 ret = spa_do_crypt_abd(B_FALSE, spa, &zio->io_bookmark, BP_GET_TYPE(bp),
503 BP_GET_DEDUP(bp), BP_SHOULD_BYTESWAP(bp), salt, iv, mac, size, data,
504 zio->io_abd, &no_crypt);
b5256303
TC
505 if (no_crypt)
506 abd_copy(data, zio->io_abd, size);
507
508 if (ret != 0)
509 goto error;
510
511 return;
512
513error:
514 /* assert that the key was found unless this was speculative */
be9a5c35 515 ASSERT(ret != EACCES || (zio->io_flags & ZIO_FLAG_SPECULATIVE));
b5256303
TC
516
517 /*
518 * If there was a decryption / authentication error return EIO as
519 * the io_error. If this was not a speculative zio, create an ereport.
520 */
521 if (ret == ECKSUM) {
a2c2ed1b 522 zio->io_error = SET_ERROR(EIO);
b5256303 523 if ((zio->io_flags & ZIO_FLAG_SPECULATIVE) == 0) {
be9a5c35 524 spa_log_error(spa, &zio->io_bookmark);
b5256303 525 zfs_ereport_post(FM_EREPORT_ZFS_AUTHENTICATION,
ae76f45c 526 spa, NULL, &zio->io_bookmark, zio, 0, 0);
b5256303
TC
527 }
528 } else {
529 zio->io_error = ret;
530 }
531}
532
b128c09f
BB
533/*
534 * ==========================================================================
535 * I/O parent/child relationships and pipeline interlocks
536 * ==========================================================================
537 */
d164b209 538zio_t *
3dfb57a3 539zio_walk_parents(zio_t *cio, zio_link_t **zl)
d164b209 540{
d164b209 541 list_t *pl = &cio->io_parent_list;
b128c09f 542
3dfb57a3
DB
543 *zl = (*zl == NULL) ? list_head(pl) : list_next(pl, *zl);
544 if (*zl == NULL)
d164b209
BB
545 return (NULL);
546
3dfb57a3
DB
547 ASSERT((*zl)->zl_child == cio);
548 return ((*zl)->zl_parent);
d164b209
BB
549}
550
551zio_t *
3dfb57a3 552zio_walk_children(zio_t *pio, zio_link_t **zl)
d164b209 553{
d164b209
BB
554 list_t *cl = &pio->io_child_list;
555
a8b2e306
TC
556 ASSERT(MUTEX_HELD(&pio->io_lock));
557
3dfb57a3
DB
558 *zl = (*zl == NULL) ? list_head(cl) : list_next(cl, *zl);
559 if (*zl == NULL)
d164b209
BB
560 return (NULL);
561
3dfb57a3
DB
562 ASSERT((*zl)->zl_parent == pio);
563 return ((*zl)->zl_child);
d164b209
BB
564}
565
566zio_t *
567zio_unique_parent(zio_t *cio)
568{
3dfb57a3
DB
569 zio_link_t *zl = NULL;
570 zio_t *pio = zio_walk_parents(cio, &zl);
d164b209 571
3dfb57a3 572 VERIFY3P(zio_walk_parents(cio, &zl), ==, NULL);
d164b209
BB
573 return (pio);
574}
575
576void
577zio_add_child(zio_t *pio, zio_t *cio)
b128c09f 578{
79c76d5b 579 zio_link_t *zl = kmem_cache_alloc(zio_link_cache, KM_SLEEP);
d164b209
BB
580
581 /*
582 * Logical I/Os can have logical, gang, or vdev children.
583 * Gang I/Os can have gang or vdev children.
584 * Vdev I/Os can only have vdev children.
585 * The following ASSERT captures all of these constraints.
586 */
1ce23dca 587 ASSERT3S(cio->io_child_type, <=, pio->io_child_type);
d164b209
BB
588
589 zl->zl_parent = pio;
590 zl->zl_child = cio;
591
b128c09f 592 mutex_enter(&pio->io_lock);
a8b2e306 593 mutex_enter(&cio->io_lock);
d164b209
BB
594
595 ASSERT(pio->io_state[ZIO_WAIT_DONE] == 0);
596
1c27024e 597 for (int w = 0; w < ZIO_WAIT_TYPES; w++)
d164b209
BB
598 pio->io_children[cio->io_child_type][w] += !cio->io_state[w];
599
600 list_insert_head(&pio->io_child_list, zl);
601 list_insert_head(&cio->io_parent_list, zl);
602
428870ff
BB
603 pio->io_child_count++;
604 cio->io_parent_count++;
605
d164b209 606 mutex_exit(&cio->io_lock);
a8b2e306 607 mutex_exit(&pio->io_lock);
b128c09f
BB
608}
609
34dc7c2f 610static void
d164b209 611zio_remove_child(zio_t *pio, zio_t *cio, zio_link_t *zl)
b128c09f 612{
d164b209
BB
613 ASSERT(zl->zl_parent == pio);
614 ASSERT(zl->zl_child == cio);
b128c09f
BB
615
616 mutex_enter(&pio->io_lock);
a8b2e306 617 mutex_enter(&cio->io_lock);
d164b209
BB
618
619 list_remove(&pio->io_child_list, zl);
620 list_remove(&cio->io_parent_list, zl);
621
428870ff
BB
622 pio->io_child_count--;
623 cio->io_parent_count--;
624
d164b209 625 mutex_exit(&cio->io_lock);
a8b2e306 626 mutex_exit(&pio->io_lock);
d164b209 627 kmem_cache_free(zio_link_cache, zl);
b128c09f
BB
628}
629
630static boolean_t
ddc751d5 631zio_wait_for_children(zio_t *zio, uint8_t childbits, enum zio_wait_type wait)
34dc7c2f 632{
b128c09f
BB
633 boolean_t waiting = B_FALSE;
634
635 mutex_enter(&zio->io_lock);
636 ASSERT(zio->io_stall == NULL);
ddc751d5
GW
637 for (int c = 0; c < ZIO_CHILD_TYPES; c++) {
638 if (!(ZIO_CHILD_BIT_IS_SET(childbits, c)))
639 continue;
640
641 uint64_t *countp = &zio->io_children[c][wait];
642 if (*countp != 0) {
643 zio->io_stage >>= 1;
644 ASSERT3U(zio->io_stage, !=, ZIO_STAGE_OPEN);
645 zio->io_stall = countp;
646 waiting = B_TRUE;
647 break;
648 }
b128c09f
BB
649 }
650 mutex_exit(&zio->io_lock);
b128c09f
BB
651 return (waiting);
652}
34dc7c2f 653
bf701a83
BB
654__attribute__((always_inline))
655static inline void
62840030
MA
656zio_notify_parent(zio_t *pio, zio_t *zio, enum zio_wait_type wait,
657 zio_t **next_to_executep)
b128c09f
BB
658{
659 uint64_t *countp = &pio->io_children[zio->io_child_type][wait];
660 int *errorp = &pio->io_child_error[zio->io_child_type];
34dc7c2f 661
b128c09f
BB
662 mutex_enter(&pio->io_lock);
663 if (zio->io_error && !(zio->io_flags & ZIO_FLAG_DONT_PROPAGATE))
664 *errorp = zio_worst_error(*errorp, zio->io_error);
665 pio->io_reexecute |= zio->io_reexecute;
666 ASSERT3U(*countp, >, 0);
e8b96c60
MA
667
668 (*countp)--;
669
670 if (*countp == 0 && pio->io_stall == countp) {
3dfb57a3
DB
671 zio_taskq_type_t type =
672 pio->io_stage < ZIO_STAGE_VDEV_IO_START ? ZIO_TASKQ_ISSUE :
673 ZIO_TASKQ_INTERRUPT;
b128c09f
BB
674 pio->io_stall = NULL;
675 mutex_exit(&pio->io_lock);
62840030 676
3dfb57a3 677 /*
62840030
MA
678 * If we can tell the caller to execute this parent next, do
679 * so. Otherwise dispatch the parent zio as its own task.
680 *
681 * Having the caller execute the parent when possible reduces
682 * locking on the zio taskq's, reduces context switch
683 * overhead, and has no recursion penalty. Note that one
684 * read from disk typically causes at least 3 zio's: a
685 * zio_null(), the logical zio_read(), and then a physical
686 * zio. When the physical ZIO completes, we are able to call
687 * zio_done() on all 3 of these zio's from one invocation of
688 * zio_execute() by returning the parent back to
689 * zio_execute(). Since the parent isn't executed until this
690 * thread returns back to zio_execute(), the caller should do
691 * so promptly.
692 *
693 * In other cases, dispatching the parent prevents
694 * overflowing the stack when we have deeply nested
695 * parent-child relationships, as we do with the "mega zio"
696 * of writes for spa_sync(), and the chain of ZIL blocks.
3dfb57a3 697 */
62840030
MA
698 if (next_to_executep != NULL && *next_to_executep == NULL) {
699 *next_to_executep = pio;
700 } else {
701 zio_taskq_dispatch(pio, type, B_FALSE);
702 }
b128c09f
BB
703 } else {
704 mutex_exit(&pio->io_lock);
34dc7c2f
BB
705 }
706}
707
b128c09f
BB
708static void
709zio_inherit_child_errors(zio_t *zio, enum zio_child c)
710{
711 if (zio->io_child_error[c] != 0 && zio->io_error == 0)
712 zio->io_error = zio->io_child_error[c];
713}
714
3dfb57a3 715int
64fc7762 716zio_bookmark_compare(const void *x1, const void *x2)
3dfb57a3
DB
717{
718 const zio_t *z1 = x1;
719 const zio_t *z2 = x2;
3dfb57a3 720
64fc7762
MA
721 if (z1->io_bookmark.zb_objset < z2->io_bookmark.zb_objset)
722 return (-1);
723 if (z1->io_bookmark.zb_objset > z2->io_bookmark.zb_objset)
724 return (1);
3dfb57a3 725
64fc7762
MA
726 if (z1->io_bookmark.zb_object < z2->io_bookmark.zb_object)
727 return (-1);
728 if (z1->io_bookmark.zb_object > z2->io_bookmark.zb_object)
729 return (1);
3dfb57a3 730
64fc7762
MA
731 if (z1->io_bookmark.zb_level < z2->io_bookmark.zb_level)
732 return (-1);
733 if (z1->io_bookmark.zb_level > z2->io_bookmark.zb_level)
734 return (1);
735
736 if (z1->io_bookmark.zb_blkid < z2->io_bookmark.zb_blkid)
737 return (-1);
738 if (z1->io_bookmark.zb_blkid > z2->io_bookmark.zb_blkid)
739 return (1);
740
741 if (z1 < z2)
742 return (-1);
743 if (z1 > z2)
744 return (1);
745
746 return (0);
3dfb57a3
DB
747}
748
34dc7c2f
BB
749/*
750 * ==========================================================================
b128c09f 751 * Create the various types of I/O (read, write, free, etc)
34dc7c2f
BB
752 * ==========================================================================
753 */
754static zio_t *
428870ff 755zio_create(zio_t *pio, spa_t *spa, uint64_t txg, const blkptr_t *bp,
a6255b7f 756 abd_t *data, uint64_t lsize, uint64_t psize, zio_done_func_t *done,
2aa34383
DK
757 void *private, zio_type_t type, zio_priority_t priority,
758 enum zio_flag flags, vdev_t *vd, uint64_t offset,
759 const zbookmark_phys_t *zb, enum zio_stage stage,
760 enum zio_stage pipeline)
34dc7c2f
BB
761{
762 zio_t *zio;
763
2aa34383
DK
764 ASSERT3U(psize, <=, SPA_MAXBLOCKSIZE);
765 ASSERT(P2PHASE(psize, SPA_MINBLOCKSIZE) == 0);
b128c09f
BB
766 ASSERT(P2PHASE(offset, SPA_MINBLOCKSIZE) == 0);
767
768 ASSERT(!vd || spa_config_held(spa, SCL_STATE_ALL, RW_READER));
769 ASSERT(!bp || !(flags & ZIO_FLAG_CONFIG_WRITER));
770 ASSERT(vd || stage == ZIO_STAGE_OPEN);
34dc7c2f 771
b5256303 772 IMPLY(lsize != psize, (flags & ZIO_FLAG_RAW_COMPRESS) != 0);
2aa34383 773
79c76d5b 774 zio = kmem_cache_alloc(zio_cache, KM_SLEEP);
3941503c
BB
775 bzero(zio, sizeof (zio_t));
776
448d7aaa 777 mutex_init(&zio->io_lock, NULL, MUTEX_NOLOCKDEP, NULL);
3941503c
BB
778 cv_init(&zio->io_cv, NULL, CV_DEFAULT, NULL);
779
780 list_create(&zio->io_parent_list, sizeof (zio_link_t),
781 offsetof(zio_link_t, zl_parent_node));
782 list_create(&zio->io_child_list, sizeof (zio_link_t),
783 offsetof(zio_link_t, zl_child_node));
4e21fd06 784 metaslab_trace_init(&zio->io_alloc_list);
d164b209 785
b128c09f
BB
786 if (vd != NULL)
787 zio->io_child_type = ZIO_CHILD_VDEV;
788 else if (flags & ZIO_FLAG_GANG_CHILD)
789 zio->io_child_type = ZIO_CHILD_GANG;
428870ff
BB
790 else if (flags & ZIO_FLAG_DDT_CHILD)
791 zio->io_child_type = ZIO_CHILD_DDT;
b128c09f
BB
792 else
793 zio->io_child_type = ZIO_CHILD_LOGICAL;
794
34dc7c2f 795 if (bp != NULL) {
428870ff 796 zio->io_bp = (blkptr_t *)bp;
34dc7c2f
BB
797 zio->io_bp_copy = *bp;
798 zio->io_bp_orig = *bp;
428870ff
BB
799 if (type != ZIO_TYPE_WRITE ||
800 zio->io_child_type == ZIO_CHILD_DDT)
b128c09f 801 zio->io_bp = &zio->io_bp_copy; /* so caller can free */
9babb374 802 if (zio->io_child_type == ZIO_CHILD_LOGICAL)
b128c09f 803 zio->io_logical = zio;
9babb374
BB
804 if (zio->io_child_type > ZIO_CHILD_GANG && BP_IS_GANG(bp))
805 pipeline |= ZIO_GANG_STAGES;
34dc7c2f 806 }
b128c09f
BB
807
808 zio->io_spa = spa;
809 zio->io_txg = txg;
34dc7c2f
BB
810 zio->io_done = done;
811 zio->io_private = private;
812 zio->io_type = type;
813 zio->io_priority = priority;
b128c09f
BB
814 zio->io_vd = vd;
815 zio->io_offset = offset;
a6255b7f 816 zio->io_orig_abd = zio->io_abd = data;
2aa34383
DK
817 zio->io_orig_size = zio->io_size = psize;
818 zio->io_lsize = lsize;
b128c09f
BB
819 zio->io_orig_flags = zio->io_flags = flags;
820 zio->io_orig_stage = zio->io_stage = stage;
821 zio->io_orig_pipeline = zio->io_pipeline = pipeline;
3dfb57a3 822 zio->io_pipeline_trace = ZIO_STAGE_OPEN;
34dc7c2f 823
d164b209
BB
824 zio->io_state[ZIO_WAIT_READY] = (stage >= ZIO_STAGE_READY);
825 zio->io_state[ZIO_WAIT_DONE] = (stage >= ZIO_STAGE_DONE);
826
b128c09f
BB
827 if (zb != NULL)
828 zio->io_bookmark = *zb;
829
830 if (pio != NULL) {
cc99f275
DB
831 if (zio->io_metaslab_class == NULL)
832 zio->io_metaslab_class = pio->io_metaslab_class;
b128c09f 833 if (zio->io_logical == NULL)
34dc7c2f 834 zio->io_logical = pio->io_logical;
9babb374
BB
835 if (zio->io_child_type == ZIO_CHILD_GANG)
836 zio->io_gang_leader = pio->io_gang_leader;
b128c09f 837 zio_add_child(pio, zio);
34dc7c2f
BB
838 }
839
a38718a6
GA
840 taskq_init_ent(&zio->io_tqent);
841
34dc7c2f
BB
842 return (zio);
843}
844
845static void
b128c09f 846zio_destroy(zio_t *zio)
34dc7c2f 847{
4e21fd06 848 metaslab_trace_fini(&zio->io_alloc_list);
3941503c
BB
849 list_destroy(&zio->io_parent_list);
850 list_destroy(&zio->io_child_list);
851 mutex_destroy(&zio->io_lock);
852 cv_destroy(&zio->io_cv);
b128c09f 853 kmem_cache_free(zio_cache, zio);
34dc7c2f
BB
854}
855
856zio_t *
d164b209 857zio_null(zio_t *pio, spa_t *spa, vdev_t *vd, zio_done_func_t *done,
428870ff 858 void *private, enum zio_flag flags)
34dc7c2f
BB
859{
860 zio_t *zio;
861
2aa34383 862 zio = zio_create(pio, spa, 0, NULL, NULL, 0, 0, done, private,
d164b209 863 ZIO_TYPE_NULL, ZIO_PRIORITY_NOW, flags, vd, 0, NULL,
b128c09f 864 ZIO_STAGE_OPEN, ZIO_INTERLOCK_PIPELINE);
34dc7c2f
BB
865
866 return (zio);
867}
868
869zio_t *
428870ff 870zio_root(spa_t *spa, zio_done_func_t *done, void *private, enum zio_flag flags)
34dc7c2f 871{
d164b209 872 return (zio_null(NULL, spa, NULL, done, private, flags));
34dc7c2f
BB
873}
874
63e3a861
MA
875void
876zfs_blkptr_verify(spa_t *spa, const blkptr_t *bp)
877{
63e3a861
MA
878 if (!DMU_OT_IS_VALID(BP_GET_TYPE(bp))) {
879 zfs_panic_recover("blkptr at %p has invalid TYPE %llu",
880 bp, (longlong_t)BP_GET_TYPE(bp));
881 }
882 if (BP_GET_CHECKSUM(bp) >= ZIO_CHECKSUM_FUNCTIONS ||
883 BP_GET_CHECKSUM(bp) <= ZIO_CHECKSUM_ON) {
884 zfs_panic_recover("blkptr at %p has invalid CHECKSUM %llu",
885 bp, (longlong_t)BP_GET_CHECKSUM(bp));
886 }
887 if (BP_GET_COMPRESS(bp) >= ZIO_COMPRESS_FUNCTIONS ||
888 BP_GET_COMPRESS(bp) <= ZIO_COMPRESS_ON) {
889 zfs_panic_recover("blkptr at %p has invalid COMPRESS %llu",
890 bp, (longlong_t)BP_GET_COMPRESS(bp));
891 }
892 if (BP_GET_LSIZE(bp) > SPA_MAXBLOCKSIZE) {
893 zfs_panic_recover("blkptr at %p has invalid LSIZE %llu",
894 bp, (longlong_t)BP_GET_LSIZE(bp));
895 }
896 if (BP_GET_PSIZE(bp) > SPA_MAXBLOCKSIZE) {
897 zfs_panic_recover("blkptr at %p has invalid PSIZE %llu",
898 bp, (longlong_t)BP_GET_PSIZE(bp));
899 }
900
901 if (BP_IS_EMBEDDED(bp)) {
902 if (BPE_GET_ETYPE(bp) > NUM_BP_EMBEDDED_TYPES) {
903 zfs_panic_recover("blkptr at %p has invalid ETYPE %llu",
904 bp, (longlong_t)BPE_GET_ETYPE(bp));
905 }
906 }
907
6cb8e530
PZ
908 /*
909 * Do not verify individual DVAs if the config is not trusted. This
910 * will be done once the zio is executed in vdev_mirror_map_alloc.
911 */
912 if (!spa->spa_trust_config)
913 return;
914
63e3a861
MA
915 /*
916 * Pool-specific checks.
917 *
918 * Note: it would be nice to verify that the blk_birth and
919 * BP_PHYSICAL_BIRTH() are not too large. However, spa_freeze()
920 * allows the birth time of log blocks (and dmu_sync()-ed blocks
921 * that are in the log) to be arbitrarily large.
922 */
1c27024e 923 for (int i = 0; i < BP_GET_NDVAS(bp); i++) {
63e3a861 924 uint64_t vdevid = DVA_GET_VDEV(&bp->blk_dva[i]);
1c27024e 925
63e3a861
MA
926 if (vdevid >= spa->spa_root_vdev->vdev_children) {
927 zfs_panic_recover("blkptr at %p DVA %u has invalid "
928 "VDEV %llu",
929 bp, i, (longlong_t)vdevid);
ee3a23b8 930 continue;
63e3a861 931 }
1c27024e 932 vdev_t *vd = spa->spa_root_vdev->vdev_child[vdevid];
63e3a861
MA
933 if (vd == NULL) {
934 zfs_panic_recover("blkptr at %p DVA %u has invalid "
935 "VDEV %llu",
936 bp, i, (longlong_t)vdevid);
ee3a23b8 937 continue;
63e3a861
MA
938 }
939 if (vd->vdev_ops == &vdev_hole_ops) {
940 zfs_panic_recover("blkptr at %p DVA %u has hole "
941 "VDEV %llu",
942 bp, i, (longlong_t)vdevid);
ee3a23b8 943 continue;
63e3a861
MA
944 }
945 if (vd->vdev_ops == &vdev_missing_ops) {
946 /*
947 * "missing" vdevs are valid during import, but we
948 * don't have their detailed info (e.g. asize), so
949 * we can't perform any more checks on them.
950 */
951 continue;
952 }
1c27024e
DB
953 uint64_t offset = DVA_GET_OFFSET(&bp->blk_dva[i]);
954 uint64_t asize = DVA_GET_ASIZE(&bp->blk_dva[i]);
63e3a861
MA
955 if (BP_IS_GANG(bp))
956 asize = vdev_psize_to_asize(vd, SPA_GANGBLOCKSIZE);
957 if (offset + asize > vd->vdev_asize) {
958 zfs_panic_recover("blkptr at %p DVA %u has invalid "
959 "OFFSET %llu",
960 bp, i, (longlong_t)offset);
961 }
962 }
963}
964
6cb8e530
PZ
965boolean_t
966zfs_dva_valid(spa_t *spa, const dva_t *dva, const blkptr_t *bp)
967{
968 uint64_t vdevid = DVA_GET_VDEV(dva);
969
970 if (vdevid >= spa->spa_root_vdev->vdev_children)
971 return (B_FALSE);
972
973 vdev_t *vd = spa->spa_root_vdev->vdev_child[vdevid];
974 if (vd == NULL)
975 return (B_FALSE);
976
977 if (vd->vdev_ops == &vdev_hole_ops)
978 return (B_FALSE);
979
980 if (vd->vdev_ops == &vdev_missing_ops) {
981 return (B_FALSE);
982 }
983
984 uint64_t offset = DVA_GET_OFFSET(dva);
985 uint64_t asize = DVA_GET_ASIZE(dva);
986
987 if (BP_IS_GANG(bp))
988 asize = vdev_psize_to_asize(vd, SPA_GANGBLOCKSIZE);
989 if (offset + asize > vd->vdev_asize)
990 return (B_FALSE);
991
992 return (B_TRUE);
993}
994
34dc7c2f 995zio_t *
b128c09f 996zio_read(zio_t *pio, spa_t *spa, const blkptr_t *bp,
a6255b7f 997 abd_t *data, uint64_t size, zio_done_func_t *done, void *private,
5dbd68a3 998 zio_priority_t priority, enum zio_flag flags, const zbookmark_phys_t *zb)
34dc7c2f
BB
999{
1000 zio_t *zio;
1001
63e3a861
MA
1002 zfs_blkptr_verify(spa, bp);
1003
428870ff 1004 zio = zio_create(pio, spa, BP_PHYSICAL_BIRTH(bp), bp,
2aa34383 1005 data, size, size, done, private,
b128c09f 1006 ZIO_TYPE_READ, priority, flags, NULL, 0, zb,
428870ff
BB
1007 ZIO_STAGE_OPEN, (flags & ZIO_FLAG_DDT_CHILD) ?
1008 ZIO_DDT_CHILD_READ_PIPELINE : ZIO_READ_PIPELINE);
34dc7c2f 1009
b128c09f
BB
1010 return (zio);
1011}
34dc7c2f 1012
34dc7c2f 1013zio_t *
b128c09f 1014zio_write(zio_t *pio, spa_t *spa, uint64_t txg, blkptr_t *bp,
a6255b7f 1015 abd_t *data, uint64_t lsize, uint64_t psize, const zio_prop_t *zp,
bc77ba73
PD
1016 zio_done_func_t *ready, zio_done_func_t *children_ready,
1017 zio_done_func_t *physdone, zio_done_func_t *done,
1018 void *private, zio_priority_t priority, enum zio_flag flags,
1019 const zbookmark_phys_t *zb)
34dc7c2f
BB
1020{
1021 zio_t *zio;
1022
b128c09f
BB
1023 ASSERT(zp->zp_checksum >= ZIO_CHECKSUM_OFF &&
1024 zp->zp_checksum < ZIO_CHECKSUM_FUNCTIONS &&
1025 zp->zp_compress >= ZIO_COMPRESS_OFF &&
1026 zp->zp_compress < ZIO_COMPRESS_FUNCTIONS &&
9ae529ec 1027 DMU_OT_IS_VALID(zp->zp_type) &&
b128c09f 1028 zp->zp_level < 32 &&
428870ff 1029 zp->zp_copies > 0 &&
03c6040b 1030 zp->zp_copies <= spa_max_replication(spa));
34dc7c2f 1031
2aa34383 1032 zio = zio_create(pio, spa, txg, bp, data, lsize, psize, done, private,
b128c09f 1033 ZIO_TYPE_WRITE, priority, flags, NULL, 0, zb,
428870ff
BB
1034 ZIO_STAGE_OPEN, (flags & ZIO_FLAG_DDT_CHILD) ?
1035 ZIO_DDT_CHILD_WRITE_PIPELINE : ZIO_WRITE_PIPELINE);
34dc7c2f
BB
1036
1037 zio->io_ready = ready;
bc77ba73 1038 zio->io_children_ready = children_ready;
e8b96c60 1039 zio->io_physdone = physdone;
b128c09f 1040 zio->io_prop = *zp;
34dc7c2f 1041
9b67f605
MA
1042 /*
1043 * Data can be NULL if we are going to call zio_write_override() to
1044 * provide the already-allocated BP. But we may need the data to
1045 * verify a dedup hit (if requested). In this case, don't try to
b5256303
TC
1046 * dedup (just take the already-allocated BP verbatim). Encrypted
1047 * dedup blocks need data as well so we also disable dedup in this
1048 * case.
9b67f605 1049 */
b5256303
TC
1050 if (data == NULL &&
1051 (zio->io_prop.zp_dedup_verify || zio->io_prop.zp_encrypt)) {
9b67f605
MA
1052 zio->io_prop.zp_dedup = zio->io_prop.zp_dedup_verify = B_FALSE;
1053 }
1054
34dc7c2f
BB
1055 return (zio);
1056}
1057
1058zio_t *
a6255b7f 1059zio_rewrite(zio_t *pio, spa_t *spa, uint64_t txg, blkptr_t *bp, abd_t *data,
e8b96c60 1060 uint64_t size, zio_done_func_t *done, void *private,
5dbd68a3 1061 zio_priority_t priority, enum zio_flag flags, zbookmark_phys_t *zb)
34dc7c2f
BB
1062{
1063 zio_t *zio;
1064
2aa34383 1065 zio = zio_create(pio, spa, txg, bp, data, size, size, done, private,
3dfb57a3 1066 ZIO_TYPE_WRITE, priority, flags | ZIO_FLAG_IO_REWRITE, NULL, 0, zb,
b128c09f 1067 ZIO_STAGE_OPEN, ZIO_REWRITE_PIPELINE);
34dc7c2f
BB
1068
1069 return (zio);
1070}
1071
428870ff 1072void
03c6040b 1073zio_write_override(zio_t *zio, blkptr_t *bp, int copies, boolean_t nopwrite)
428870ff
BB
1074{
1075 ASSERT(zio->io_type == ZIO_TYPE_WRITE);
1076 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
1077 ASSERT(zio->io_stage == ZIO_STAGE_OPEN);
1078 ASSERT(zio->io_txg == spa_syncing_txg(zio->io_spa));
1079
03c6040b
GW
1080 /*
1081 * We must reset the io_prop to match the values that existed
1082 * when the bp was first written by dmu_sync() keeping in mind
1083 * that nopwrite and dedup are mutually exclusive.
1084 */
1085 zio->io_prop.zp_dedup = nopwrite ? B_FALSE : zio->io_prop.zp_dedup;
1086 zio->io_prop.zp_nopwrite = nopwrite;
428870ff
BB
1087 zio->io_prop.zp_copies = copies;
1088 zio->io_bp_override = bp;
1089}
1090
1091void
1092zio_free(spa_t *spa, uint64_t txg, const blkptr_t *bp)
1093{
9b67f605 1094
a1d477c2
MA
1095 zfs_blkptr_verify(spa, bp);
1096
9b67f605
MA
1097 /*
1098 * The check for EMBEDDED is a performance optimization. We
1099 * process the free here (by ignoring it) rather than
1100 * putting it on the list and then processing it in zio_free_sync().
1101 */
1102 if (BP_IS_EMBEDDED(bp))
1103 return;
13fe0198 1104 metaslab_check_free(spa, bp);
2883cad5
MA
1105
1106 /*
1107 * Frees that are for the currently-syncing txg, are not going to be
1108 * deferred, and which will not need to do a read (i.e. not GANG or
1109 * DEDUP), can be processed immediately. Otherwise, put them on the
1110 * in-memory list for later processing.
1111 */
1112 if (BP_IS_GANG(bp) || BP_GET_DEDUP(bp) ||
1113 txg != spa->spa_syncing_txg ||
1114 spa_sync_pass(spa) >= zfs_sync_pass_deferred_free) {
1115 bplist_append(&spa->spa_free_bplist[txg & TXG_MASK], bp);
1116 } else {
1117 VERIFY0(zio_wait(zio_free_sync(NULL, spa, txg, bp, 0)));
1118 }
428870ff
BB
1119}
1120
34dc7c2f 1121zio_t *
428870ff
BB
1122zio_free_sync(zio_t *pio, spa_t *spa, uint64_t txg, const blkptr_t *bp,
1123 enum zio_flag flags)
34dc7c2f
BB
1124{
1125 zio_t *zio;
2883cad5 1126 enum zio_stage stage = ZIO_FREE_PIPELINE;
34dc7c2f 1127
428870ff
BB
1128 ASSERT(!BP_IS_HOLE(bp));
1129 ASSERT(spa_syncing_txg(spa) == txg);
55d85d5a 1130 ASSERT(spa_sync_pass(spa) < zfs_sync_pass_deferred_free);
34dc7c2f 1131
9b67f605
MA
1132 if (BP_IS_EMBEDDED(bp))
1133 return (zio_null(pio, spa, NULL, NULL, NULL, 0));
1134
13fe0198 1135 metaslab_check_free(spa, bp);
8c841793 1136 arc_freed(spa, bp);
d4a72f23 1137 dsl_scan_freed(spa, bp);
13fe0198 1138
2883cad5
MA
1139 /*
1140 * GANG and DEDUP blocks can induce a read (for the gang block header,
1141 * or the DDT), so issue them asynchronously so that this thread is
1142 * not tied up.
1143 */
1144 if (BP_IS_GANG(bp) || BP_GET_DEDUP(bp))
1145 stage |= ZIO_STAGE_ISSUE_ASYNC;
1146
b128c09f 1147 zio = zio_create(pio, spa, txg, bp, NULL, BP_GET_PSIZE(bp),
2aa34383
DK
1148 BP_GET_PSIZE(bp), NULL, NULL, ZIO_TYPE_FREE, ZIO_PRIORITY_NOW,
1149 flags, NULL, 0, NULL, ZIO_STAGE_OPEN, stage);
2883cad5 1150
34dc7c2f
BB
1151 return (zio);
1152}
1153
1154zio_t *
428870ff
BB
1155zio_claim(zio_t *pio, spa_t *spa, uint64_t txg, const blkptr_t *bp,
1156 zio_done_func_t *done, void *private, enum zio_flag flags)
34dc7c2f
BB
1157{
1158 zio_t *zio;
1159
a1d477c2 1160 zfs_blkptr_verify(spa, bp);
9b67f605
MA
1161
1162 if (BP_IS_EMBEDDED(bp))
1163 return (zio_null(pio, spa, NULL, NULL, NULL, 0));
1164
34dc7c2f
BB
1165 /*
1166 * A claim is an allocation of a specific block. Claims are needed
1167 * to support immediate writes in the intent log. The issue is that
1168 * immediate writes contain committed data, but in a txg that was
1169 * *not* committed. Upon opening the pool after an unclean shutdown,
1170 * the intent log claims all blocks that contain immediate write data
1171 * so that the SPA knows they're in use.
1172 *
1173 * All claims *must* be resolved in the first txg -- before the SPA
1174 * starts allocating blocks -- so that nothing is allocated twice.
428870ff 1175 * If txg == 0 we just verify that the block is claimable.
34dc7c2f 1176 */
d2734cce
SD
1177 ASSERT3U(spa->spa_uberblock.ub_rootbp.blk_birth, <,
1178 spa_min_claim_txg(spa));
1179 ASSERT(txg == spa_min_claim_txg(spa) || txg == 0);
428870ff 1180 ASSERT(!BP_GET_DEDUP(bp) || !spa_writeable(spa)); /* zdb(1M) */
34dc7c2f 1181
b128c09f 1182 zio = zio_create(pio, spa, txg, bp, NULL, BP_GET_PSIZE(bp),
2aa34383
DK
1183 BP_GET_PSIZE(bp), done, private, ZIO_TYPE_CLAIM, ZIO_PRIORITY_NOW,
1184 flags, NULL, 0, NULL, ZIO_STAGE_OPEN, ZIO_CLAIM_PIPELINE);
3dfb57a3 1185 ASSERT0(zio->io_queued_timestamp);
34dc7c2f
BB
1186
1187 return (zio);
1188}
1189
1190zio_t *
1191zio_ioctl(zio_t *pio, spa_t *spa, vdev_t *vd, int cmd,
e8b96c60 1192 zio_done_func_t *done, void *private, enum zio_flag flags)
34dc7c2f
BB
1193{
1194 zio_t *zio;
1195 int c;
1196
1197 if (vd->vdev_children == 0) {
2aa34383 1198 zio = zio_create(pio, spa, 0, NULL, NULL, 0, 0, done, private,
e8b96c60 1199 ZIO_TYPE_IOCTL, ZIO_PRIORITY_NOW, flags, vd, 0, NULL,
34dc7c2f
BB
1200 ZIO_STAGE_OPEN, ZIO_IOCTL_PIPELINE);
1201
34dc7c2f
BB
1202 zio->io_cmd = cmd;
1203 } else {
d164b209 1204 zio = zio_null(pio, spa, NULL, NULL, NULL, flags);
34dc7c2f
BB
1205
1206 for (c = 0; c < vd->vdev_children; c++)
1207 zio_nowait(zio_ioctl(zio, spa, vd->vdev_child[c], cmd,
e8b96c60 1208 done, private, flags));
34dc7c2f
BB
1209 }
1210
1211 return (zio);
1212}
1213
34dc7c2f
BB
1214zio_t *
1215zio_read_phys(zio_t *pio, vdev_t *vd, uint64_t offset, uint64_t size,
a6255b7f 1216 abd_t *data, int checksum, zio_done_func_t *done, void *private,
e8b96c60 1217 zio_priority_t priority, enum zio_flag flags, boolean_t labels)
34dc7c2f
BB
1218{
1219 zio_t *zio;
34dc7c2f 1220
b128c09f
BB
1221 ASSERT(vd->vdev_children == 0);
1222 ASSERT(!labels || offset + size <= VDEV_LABEL_START_SIZE ||
1223 offset >= vd->vdev_psize - VDEV_LABEL_END_SIZE);
1224 ASSERT3U(offset + size, <=, vd->vdev_psize);
34dc7c2f 1225
2aa34383
DK
1226 zio = zio_create(pio, vd->vdev_spa, 0, NULL, data, size, size, done,
1227 private, ZIO_TYPE_READ, priority, flags | ZIO_FLAG_PHYSICAL, vd,
1228 offset, NULL, ZIO_STAGE_OPEN, ZIO_READ_PHYS_PIPELINE);
34dc7c2f 1229
b128c09f 1230 zio->io_prop.zp_checksum = checksum;
34dc7c2f
BB
1231
1232 return (zio);
1233}
1234
1235zio_t *
1236zio_write_phys(zio_t *pio, vdev_t *vd, uint64_t offset, uint64_t size,
a6255b7f 1237 abd_t *data, int checksum, zio_done_func_t *done, void *private,
e8b96c60 1238 zio_priority_t priority, enum zio_flag flags, boolean_t labels)
34dc7c2f 1239{
34dc7c2f 1240 zio_t *zio;
34dc7c2f 1241
b128c09f
BB
1242 ASSERT(vd->vdev_children == 0);
1243 ASSERT(!labels || offset + size <= VDEV_LABEL_START_SIZE ||
1244 offset >= vd->vdev_psize - VDEV_LABEL_END_SIZE);
1245 ASSERT3U(offset + size, <=, vd->vdev_psize);
34dc7c2f 1246
2aa34383
DK
1247 zio = zio_create(pio, vd->vdev_spa, 0, NULL, data, size, size, done,
1248 private, ZIO_TYPE_WRITE, priority, flags | ZIO_FLAG_PHYSICAL, vd,
1249 offset, NULL, ZIO_STAGE_OPEN, ZIO_WRITE_PHYS_PIPELINE);
34dc7c2f 1250
b128c09f 1251 zio->io_prop.zp_checksum = checksum;
34dc7c2f 1252
3c67d83a 1253 if (zio_checksum_table[checksum].ci_flags & ZCHECKSUM_FLAG_EMBEDDED) {
34dc7c2f 1254 /*
428870ff 1255 * zec checksums are necessarily destructive -- they modify
b128c09f 1256 * the end of the write buffer to hold the verifier/checksum.
34dc7c2f 1257 * Therefore, we must make a local copy in case the data is
b128c09f 1258 * being written to multiple places in parallel.
34dc7c2f 1259 */
a6255b7f
DQ
1260 abd_t *wbuf = abd_alloc_sametype(data, size);
1261 abd_copy(wbuf, data, size);
1262
b128c09f 1263 zio_push_transform(zio, wbuf, size, size, NULL);
34dc7c2f
BB
1264 }
1265
1266 return (zio);
1267}
1268
1269/*
b128c09f 1270 * Create a child I/O to do some work for us.
34dc7c2f
BB
1271 */
1272zio_t *
b128c09f 1273zio_vdev_child_io(zio_t *pio, blkptr_t *bp, vdev_t *vd, uint64_t offset,
4ea3f864
GM
1274 abd_t *data, uint64_t size, int type, zio_priority_t priority,
1275 enum zio_flag flags, zio_done_func_t *done, void *private)
34dc7c2f 1276{
428870ff 1277 enum zio_stage pipeline = ZIO_VDEV_CHILD_PIPELINE;
b128c09f
BB
1278 zio_t *zio;
1279
a1d477c2
MA
1280 /*
1281 * vdev child I/Os do not propagate their error to the parent.
1282 * Therefore, for correct operation the caller *must* check for
1283 * and handle the error in the child i/o's done callback.
1284 * The only exceptions are i/os that we don't care about
1285 * (OPTIONAL or REPAIR).
1286 */
1287 ASSERT((flags & ZIO_FLAG_OPTIONAL) || (flags & ZIO_FLAG_IO_REPAIR) ||
1288 done != NULL);
1289
34dc7c2f
BB
1290 if (type == ZIO_TYPE_READ && bp != NULL) {
1291 /*
1292 * If we have the bp, then the child should perform the
1293 * checksum and the parent need not. This pushes error
1294 * detection as close to the leaves as possible and
1295 * eliminates redundant checksums in the interior nodes.
1296 */
428870ff
BB
1297 pipeline |= ZIO_STAGE_CHECKSUM_VERIFY;
1298 pio->io_pipeline &= ~ZIO_STAGE_CHECKSUM_VERIFY;
34dc7c2f
BB
1299 }
1300
a1d477c2
MA
1301 if (vd->vdev_ops->vdev_op_leaf) {
1302 ASSERT0(vd->vdev_children);
b128c09f 1303 offset += VDEV_LABEL_START_SIZE;
a1d477c2 1304 }
b128c09f 1305
a1d477c2 1306 flags |= ZIO_VDEV_CHILD_FLAGS(pio);
428870ff
BB
1307
1308 /*
1309 * If we've decided to do a repair, the write is not speculative --
1310 * even if the original read was.
1311 */
1312 if (flags & ZIO_FLAG_IO_REPAIR)
1313 flags &= ~ZIO_FLAG_SPECULATIVE;
1314
3dfb57a3
DB
1315 /*
1316 * If we're creating a child I/O that is not associated with a
1317 * top-level vdev, then the child zio is not an allocating I/O.
1318 * If this is a retried I/O then we ignore it since we will
1319 * have already processed the original allocating I/O.
1320 */
1321 if (flags & ZIO_FLAG_IO_ALLOCATING &&
1322 (vd != vd->vdev_top || (flags & ZIO_FLAG_IO_RETRY))) {
cc99f275
DB
1323 ASSERT(pio->io_metaslab_class != NULL);
1324 ASSERT(pio->io_metaslab_class->mc_alloc_throttle_enabled);
3dfb57a3
DB
1325 ASSERT(type == ZIO_TYPE_WRITE);
1326 ASSERT(priority == ZIO_PRIORITY_ASYNC_WRITE);
1327 ASSERT(!(flags & ZIO_FLAG_IO_REPAIR));
1328 ASSERT(!(pio->io_flags & ZIO_FLAG_IO_REWRITE) ||
1329 pio->io_child_type == ZIO_CHILD_GANG);
1330
1331 flags &= ~ZIO_FLAG_IO_ALLOCATING;
1332 }
1333
1334
2aa34383 1335 zio = zio_create(pio, pio->io_spa, pio->io_txg, bp, data, size, size,
428870ff
BB
1336 done, private, type, priority, flags, vd, offset, &pio->io_bookmark,
1337 ZIO_STAGE_VDEV_IO_START >> 1, pipeline);
3dfb57a3 1338 ASSERT3U(zio->io_child_type, ==, ZIO_CHILD_VDEV);
34dc7c2f 1339
e8b96c60
MA
1340 zio->io_physdone = pio->io_physdone;
1341 if (vd->vdev_ops->vdev_op_leaf && zio->io_logical != NULL)
1342 zio->io_logical->io_phys_children++;
1343
b128c09f 1344 return (zio);
34dc7c2f
BB
1345}
1346
b128c09f 1347zio_t *
a6255b7f 1348zio_vdev_delegated_io(vdev_t *vd, uint64_t offset, abd_t *data, uint64_t size,
9e052db4 1349 zio_type_t type, zio_priority_t priority, enum zio_flag flags,
e9aa730c 1350 zio_done_func_t *done, void *private)
34dc7c2f 1351{
b128c09f 1352 zio_t *zio;
34dc7c2f 1353
b128c09f 1354 ASSERT(vd->vdev_ops->vdev_op_leaf);
34dc7c2f 1355
b128c09f 1356 zio = zio_create(NULL, vd->vdev_spa, 0, NULL,
2aa34383 1357 data, size, size, done, private, type, priority,
e8b96c60 1358 flags | ZIO_FLAG_CANFAIL | ZIO_FLAG_DONT_RETRY | ZIO_FLAG_DELEGATED,
b128c09f 1359 vd, offset, NULL,
428870ff 1360 ZIO_STAGE_VDEV_IO_START >> 1, ZIO_VDEV_CHILD_PIPELINE);
34dc7c2f 1361
b128c09f 1362 return (zio);
34dc7c2f
BB
1363}
1364
1365void
b128c09f 1366zio_flush(zio_t *zio, vdev_t *vd)
34dc7c2f 1367{
b128c09f 1368 zio_nowait(zio_ioctl(zio, zio->io_spa, vd, DKIOCFLUSHWRITECACHE,
e8b96c60 1369 NULL, NULL,
b128c09f 1370 ZIO_FLAG_CANFAIL | ZIO_FLAG_DONT_PROPAGATE | ZIO_FLAG_DONT_RETRY));
34dc7c2f
BB
1371}
1372
428870ff
BB
1373void
1374zio_shrink(zio_t *zio, uint64_t size)
1375{
1ce23dca
PS
1376 ASSERT3P(zio->io_executor, ==, NULL);
1377 ASSERT3U(zio->io_orig_size, ==, zio->io_size);
1378 ASSERT3U(size, <=, zio->io_size);
428870ff
BB
1379
1380 /*
1381 * We don't shrink for raidz because of problems with the
1382 * reconstruction when reading back less than the block size.
1383 * Note, BP_IS_RAIDZ() assumes no compression.
1384 */
1385 ASSERT(BP_GET_COMPRESS(zio->io_bp) == ZIO_COMPRESS_OFF);
2aa34383
DK
1386 if (!BP_IS_RAIDZ(zio->io_bp)) {
1387 /* we are not doing a raw write */
1388 ASSERT3U(zio->io_size, ==, zio->io_lsize);
1389 zio->io_orig_size = zio->io_size = zio->io_lsize = size;
1390 }
428870ff
BB
1391}
1392
34dc7c2f
BB
1393/*
1394 * ==========================================================================
b128c09f 1395 * Prepare to read and write logical blocks
34dc7c2f
BB
1396 * ==========================================================================
1397 */
b128c09f 1398
62840030 1399static zio_t *
b128c09f 1400zio_read_bp_init(zio_t *zio)
34dc7c2f 1401{
b128c09f 1402 blkptr_t *bp = zio->io_bp;
b5256303
TC
1403 uint64_t psize =
1404 BP_IS_EMBEDDED(bp) ? BPE_GET_PSIZE(bp) : BP_GET_PSIZE(bp);
34dc7c2f 1405
a1d477c2
MA
1406 ASSERT3P(zio->io_bp, ==, &zio->io_bp_copy);
1407
fb5f0bc8 1408 if (BP_GET_COMPRESS(bp) != ZIO_COMPRESS_OFF &&
9babb374 1409 zio->io_child_type == ZIO_CHILD_LOGICAL &&
b5256303 1410 !(zio->io_flags & ZIO_FLAG_RAW_COMPRESS)) {
a6255b7f
DQ
1411 zio_push_transform(zio, abd_alloc_sametype(zio->io_abd, psize),
1412 psize, psize, zio_decompress);
34dc7c2f 1413 }
34dc7c2f 1414
b5256303
TC
1415 if (((BP_IS_PROTECTED(bp) && !(zio->io_flags & ZIO_FLAG_RAW_ENCRYPT)) ||
1416 BP_HAS_INDIRECT_MAC_CKSUM(bp)) &&
1417 zio->io_child_type == ZIO_CHILD_LOGICAL) {
1418 zio_push_transform(zio, abd_alloc_sametype(zio->io_abd, psize),
1419 psize, psize, zio_decrypt);
1420 }
1421
9b67f605 1422 if (BP_IS_EMBEDDED(bp) && BPE_GET_ETYPE(bp) == BP_EMBEDDED_TYPE_DATA) {
a6255b7f
DQ
1423 int psize = BPE_GET_PSIZE(bp);
1424 void *data = abd_borrow_buf(zio->io_abd, psize);
1425
9b67f605 1426 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
a6255b7f
DQ
1427 decode_embedded_bp_compressed(bp, data);
1428 abd_return_buf_copy(zio->io_abd, data, psize);
9b67f605
MA
1429 } else {
1430 ASSERT(!BP_IS_EMBEDDED(bp));
a1d477c2 1431 ASSERT3P(zio->io_bp, ==, &zio->io_bp_copy);
9b67f605
MA
1432 }
1433
9ae529ec 1434 if (!DMU_OT_IS_METADATA(BP_GET_TYPE(bp)) && BP_GET_LEVEL(bp) == 0)
b128c09f
BB
1435 zio->io_flags |= ZIO_FLAG_DONT_CACHE;
1436
428870ff
BB
1437 if (BP_GET_TYPE(bp) == DMU_OT_DDT_ZAP)
1438 zio->io_flags |= ZIO_FLAG_DONT_CACHE;
1439
1440 if (BP_GET_DEDUP(bp) && zio->io_child_type == ZIO_CHILD_LOGICAL)
1441 zio->io_pipeline = ZIO_DDT_READ_PIPELINE;
1442
62840030 1443 return (zio);
34dc7c2f
BB
1444}
1445
62840030 1446static zio_t *
b128c09f 1447zio_write_bp_init(zio_t *zio)
34dc7c2f 1448{
b128c09f 1449 if (!IO_IS_ALLOCATING(zio))
62840030 1450 return (zio);
34dc7c2f 1451
428870ff
BB
1452 ASSERT(zio->io_child_type != ZIO_CHILD_DDT);
1453
1454 if (zio->io_bp_override) {
3dfb57a3
DB
1455 blkptr_t *bp = zio->io_bp;
1456 zio_prop_t *zp = &zio->io_prop;
1457
428870ff
BB
1458 ASSERT(bp->blk_birth != zio->io_txg);
1459 ASSERT(BP_GET_DEDUP(zio->io_bp_override) == 0);
1460
1461 *bp = *zio->io_bp_override;
1462 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
1463
9b67f605 1464 if (BP_IS_EMBEDDED(bp))
62840030 1465 return (zio);
9b67f605 1466
03c6040b
GW
1467 /*
1468 * If we've been overridden and nopwrite is set then
1469 * set the flag accordingly to indicate that a nopwrite
1470 * has already occurred.
1471 */
1472 if (!BP_IS_HOLE(bp) && zp->zp_nopwrite) {
1473 ASSERT(!zp->zp_dedup);
3dfb57a3 1474 ASSERT3U(BP_GET_CHECKSUM(bp), ==, zp->zp_checksum);
03c6040b 1475 zio->io_flags |= ZIO_FLAG_NOPWRITE;
62840030 1476 return (zio);
03c6040b
GW
1477 }
1478
1479 ASSERT(!zp->zp_nopwrite);
1480
428870ff 1481 if (BP_IS_HOLE(bp) || !zp->zp_dedup)
62840030 1482 return (zio);
428870ff 1483
3c67d83a
TH
1484 ASSERT((zio_checksum_table[zp->zp_checksum].ci_flags &
1485 ZCHECKSUM_FLAG_DEDUP) || zp->zp_dedup_verify);
428870ff 1486
b5256303
TC
1487 if (BP_GET_CHECKSUM(bp) == zp->zp_checksum &&
1488 !zp->zp_encrypt) {
428870ff
BB
1489 BP_SET_DEDUP(bp, 1);
1490 zio->io_pipeline |= ZIO_STAGE_DDT_WRITE;
62840030 1491 return (zio);
428870ff 1492 }
3dfb57a3
DB
1493
1494 /*
1495 * We were unable to handle this as an override bp, treat
1496 * it as a regular write I/O.
1497 */
5511754b 1498 zio->io_bp_override = NULL;
3dfb57a3
DB
1499 *bp = zio->io_bp_orig;
1500 zio->io_pipeline = zio->io_orig_pipeline;
1501 }
1502
62840030 1503 return (zio);
3dfb57a3
DB
1504}
1505
62840030 1506static zio_t *
3dfb57a3
DB
1507zio_write_compress(zio_t *zio)
1508{
1509 spa_t *spa = zio->io_spa;
1510 zio_prop_t *zp = &zio->io_prop;
1511 enum zio_compress compress = zp->zp_compress;
1512 blkptr_t *bp = zio->io_bp;
1513 uint64_t lsize = zio->io_lsize;
1514 uint64_t psize = zio->io_size;
1515 int pass = 1;
1516
3dfb57a3
DB
1517 /*
1518 * If our children haven't all reached the ready stage,
1519 * wait for them and then repeat this pipeline stage.
1520 */
ddc751d5
GW
1521 if (zio_wait_for_children(zio, ZIO_CHILD_LOGICAL_BIT |
1522 ZIO_CHILD_GANG_BIT, ZIO_WAIT_READY)) {
62840030 1523 return (NULL);
ddc751d5 1524 }
3dfb57a3
DB
1525
1526 if (!IO_IS_ALLOCATING(zio))
62840030 1527 return (zio);
3dfb57a3
DB
1528
1529 if (zio->io_children_ready != NULL) {
1530 /*
1531 * Now that all our children are ready, run the callback
1532 * associated with this zio in case it wants to modify the
1533 * data to be written.
1534 */
1535 ASSERT3U(zp->zp_level, >, 0);
1536 zio->io_children_ready(zio);
428870ff 1537 }
34dc7c2f 1538
3dfb57a3
DB
1539 ASSERT(zio->io_child_type != ZIO_CHILD_DDT);
1540 ASSERT(zio->io_bp_override == NULL);
1541
b0bc7a84 1542 if (!BP_IS_HOLE(bp) && bp->blk_birth == zio->io_txg) {
b128c09f
BB
1543 /*
1544 * We're rewriting an existing block, which means we're
1545 * working on behalf of spa_sync(). For spa_sync() to
1546 * converge, it must eventually be the case that we don't
1547 * have to allocate new blocks. But compression changes
1548 * the blocksize, which forces a reallocate, and makes
1549 * convergence take longer. Therefore, after the first
1550 * few passes, stop compressing to ensure convergence.
1551 */
428870ff
BB
1552 pass = spa_sync_pass(spa);
1553
1554 ASSERT(zio->io_txg == spa_syncing_txg(spa));
1555 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
1556 ASSERT(!BP_GET_DEDUP(bp));
34dc7c2f 1557
55d85d5a 1558 if (pass >= zfs_sync_pass_dont_compress)
b128c09f 1559 compress = ZIO_COMPRESS_OFF;
34dc7c2f 1560
b128c09f 1561 /* Make sure someone doesn't change their mind on overwrites */
9b67f605 1562 ASSERT(BP_IS_EMBEDDED(bp) || MIN(zp->zp_copies + BP_IS_GANG(bp),
428870ff 1563 spa_max_replication(spa)) == BP_GET_NDVAS(bp));
b128c09f 1564 }
34dc7c2f 1565
2aa34383 1566 /* If it's a compressed write that is not raw, compress the buffer. */
b5256303
TC
1567 if (compress != ZIO_COMPRESS_OFF &&
1568 !(zio->io_flags & ZIO_FLAG_RAW_COMPRESS)) {
428870ff 1569 void *cbuf = zio_buf_alloc(lsize);
a6255b7f 1570 psize = zio_compress_data(compress, zio->io_abd, cbuf, lsize);
428870ff 1571 if (psize == 0 || psize == lsize) {
b128c09f 1572 compress = ZIO_COMPRESS_OFF;
428870ff 1573 zio_buf_free(cbuf, lsize);
b5256303
TC
1574 } else if (!zp->zp_dedup && !zp->zp_encrypt &&
1575 psize <= BPE_PAYLOAD_SIZE &&
9b67f605
MA
1576 zp->zp_level == 0 && !DMU_OT_HAS_FILL(zp->zp_type) &&
1577 spa_feature_is_enabled(spa, SPA_FEATURE_EMBEDDED_DATA)) {
1578 encode_embedded_bp_compressed(bp,
1579 cbuf, compress, lsize, psize);
1580 BPE_SET_ETYPE(bp, BP_EMBEDDED_TYPE_DATA);
1581 BP_SET_TYPE(bp, zio->io_prop.zp_type);
1582 BP_SET_LEVEL(bp, zio->io_prop.zp_level);
1583 zio_buf_free(cbuf, lsize);
1584 bp->blk_birth = zio->io_txg;
1585 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
1586 ASSERT(spa_feature_is_active(spa,
1587 SPA_FEATURE_EMBEDDED_DATA));
62840030 1588 return (zio);
428870ff 1589 } else {
9b67f605 1590 /*
c3520e7f
MA
1591 * Round up compressed size up to the ashift
1592 * of the smallest-ashift device, and zero the tail.
1593 * This ensures that the compressed size of the BP
1594 * (and thus compressratio property) are correct,
1595 * in that we charge for the padding used to fill out
1596 * the last sector.
9b67f605 1597 */
c3520e7f 1598 ASSERT3U(spa->spa_min_ashift, >=, SPA_MINBLOCKSHIFT);
1c27024e 1599 size_t rounded = (size_t)P2ROUNDUP(psize,
c3520e7f
MA
1600 1ULL << spa->spa_min_ashift);
1601 if (rounded >= lsize) {
9b67f605
MA
1602 compress = ZIO_COMPRESS_OFF;
1603 zio_buf_free(cbuf, lsize);
c3520e7f 1604 psize = lsize;
9b67f605 1605 } else {
a6255b7f
DQ
1606 abd_t *cdata = abd_get_from_buf(cbuf, lsize);
1607 abd_take_ownership_of_buf(cdata, B_TRUE);
1608 abd_zero_off(cdata, psize, rounded - psize);
c3520e7f 1609 psize = rounded;
a6255b7f 1610 zio_push_transform(zio, cdata,
9b67f605
MA
1611 psize, lsize, NULL);
1612 }
b128c09f 1613 }
3dfb57a3
DB
1614
1615 /*
1616 * We were unable to handle this as an override bp, treat
1617 * it as a regular write I/O.
1618 */
1619 zio->io_bp_override = NULL;
1620 *bp = zio->io_bp_orig;
1621 zio->io_pipeline = zio->io_orig_pipeline;
1622
b1d21733
TC
1623 } else if ((zio->io_flags & ZIO_FLAG_RAW_ENCRYPT) != 0 &&
1624 zp->zp_type == DMU_OT_DNODE) {
1625 /*
1626 * The DMU actually relies on the zio layer's compression
1627 * to free metadnode blocks that have had all contained
1628 * dnodes freed. As a result, even when doing a raw
1629 * receive, we must check whether the block can be compressed
1630 * to a hole.
1631 */
1632 psize = zio_compress_data(ZIO_COMPRESS_EMPTY,
1633 zio->io_abd, NULL, lsize);
1634 if (psize == 0)
1635 compress = ZIO_COMPRESS_OFF;
2aa34383
DK
1636 } else {
1637 ASSERT3U(psize, !=, 0);
b128c09f 1638 }
34dc7c2f 1639
b128c09f
BB
1640 /*
1641 * The final pass of spa_sync() must be all rewrites, but the first
1642 * few passes offer a trade-off: allocating blocks defers convergence,
1643 * but newly allocated blocks are sequential, so they can be written
1644 * to disk faster. Therefore, we allow the first few passes of
1645 * spa_sync() to allocate new blocks, but force rewrites after that.
1646 * There should only be a handful of blocks after pass 1 in any case.
1647 */
b0bc7a84
MG
1648 if (!BP_IS_HOLE(bp) && bp->blk_birth == zio->io_txg &&
1649 BP_GET_PSIZE(bp) == psize &&
55d85d5a 1650 pass >= zfs_sync_pass_rewrite) {
cc99f275 1651 VERIFY3U(psize, !=, 0);
1c27024e 1652 enum zio_stage gang_stages = zio->io_pipeline & ZIO_GANG_STAGES;
cc99f275 1653
b128c09f
BB
1654 zio->io_pipeline = ZIO_REWRITE_PIPELINE | gang_stages;
1655 zio->io_flags |= ZIO_FLAG_IO_REWRITE;
1656 } else {
1657 BP_ZERO(bp);
1658 zio->io_pipeline = ZIO_WRITE_PIPELINE;
1659 }
34dc7c2f 1660
428870ff 1661 if (psize == 0) {
b0bc7a84
MG
1662 if (zio->io_bp_orig.blk_birth != 0 &&
1663 spa_feature_is_active(spa, SPA_FEATURE_HOLE_BIRTH)) {
1664 BP_SET_LSIZE(bp, lsize);
1665 BP_SET_TYPE(bp, zp->zp_type);
1666 BP_SET_LEVEL(bp, zp->zp_level);
1667 BP_SET_BIRTH(bp, zio->io_txg, 0);
1668 }
b128c09f
BB
1669 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
1670 } else {
1671 ASSERT(zp->zp_checksum != ZIO_CHECKSUM_GANG_HEADER);
1672 BP_SET_LSIZE(bp, lsize);
b0bc7a84
MG
1673 BP_SET_TYPE(bp, zp->zp_type);
1674 BP_SET_LEVEL(bp, zp->zp_level);
428870ff 1675 BP_SET_PSIZE(bp, psize);
b128c09f
BB
1676 BP_SET_COMPRESS(bp, compress);
1677 BP_SET_CHECKSUM(bp, zp->zp_checksum);
428870ff 1678 BP_SET_DEDUP(bp, zp->zp_dedup);
b128c09f 1679 BP_SET_BYTEORDER(bp, ZFS_HOST_BYTEORDER);
428870ff
BB
1680 if (zp->zp_dedup) {
1681 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
1682 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REWRITE));
b5256303
TC
1683 ASSERT(!zp->zp_encrypt ||
1684 DMU_OT_IS_ENCRYPTED(zp->zp_type));
428870ff
BB
1685 zio->io_pipeline = ZIO_DDT_WRITE_PIPELINE;
1686 }
03c6040b
GW
1687 if (zp->zp_nopwrite) {
1688 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
1689 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REWRITE));
1690 zio->io_pipeline |= ZIO_STAGE_NOP_WRITE;
1691 }
428870ff 1692 }
62840030 1693 return (zio);
428870ff
BB
1694}
1695
62840030 1696static zio_t *
428870ff
BB
1697zio_free_bp_init(zio_t *zio)
1698{
1699 blkptr_t *bp = zio->io_bp;
1700
1701 if (zio->io_child_type == ZIO_CHILD_LOGICAL) {
1702 if (BP_GET_DEDUP(bp))
1703 zio->io_pipeline = ZIO_DDT_FREE_PIPELINE;
b128c09f 1704 }
34dc7c2f 1705
a1d477c2
MA
1706 ASSERT3P(zio->io_bp, ==, &zio->io_bp_copy);
1707
62840030 1708 return (zio);
34dc7c2f
BB
1709}
1710
b128c09f
BB
1711/*
1712 * ==========================================================================
1713 * Execute the I/O pipeline
1714 * ==========================================================================
1715 */
1716
1717static void
7ef5e54e 1718zio_taskq_dispatch(zio_t *zio, zio_taskq_type_t q, boolean_t cutinline)
34dc7c2f 1719{
428870ff 1720 spa_t *spa = zio->io_spa;
b128c09f 1721 zio_type_t t = zio->io_type;
a38718a6 1722 int flags = (cutinline ? TQ_FRONT : 0);
34dc7c2f
BB
1723
1724 /*
9babb374
BB
1725 * If we're a config writer or a probe, the normal issue and
1726 * interrupt threads may all be blocked waiting for the config lock.
1727 * In this case, select the otherwise-unused taskq for ZIO_TYPE_NULL.
34dc7c2f 1728 */
9babb374 1729 if (zio->io_flags & (ZIO_FLAG_CONFIG_WRITER | ZIO_FLAG_PROBE))
b128c09f 1730 t = ZIO_TYPE_NULL;
34dc7c2f
BB
1731
1732 /*
b128c09f 1733 * A similar issue exists for the L2ARC write thread until L2ARC 2.0.
34dc7c2f 1734 */
b128c09f
BB
1735 if (t == ZIO_TYPE_WRITE && zio->io_vd && zio->io_vd->vdev_aux)
1736 t = ZIO_TYPE_NULL;
34dc7c2f 1737
428870ff 1738 /*
7ef5e54e
AL
1739 * If this is a high priority I/O, then use the high priority taskq if
1740 * available.
428870ff 1741 */
18b14b17
GW
1742 if ((zio->io_priority == ZIO_PRIORITY_NOW ||
1743 zio->io_priority == ZIO_PRIORITY_SYNC_WRITE) &&
7ef5e54e 1744 spa->spa_zio_taskq[t][q + 1].stqs_count != 0)
428870ff
BB
1745 q++;
1746
1747 ASSERT3U(q, <, ZIO_TASKQ_TYPES);
5cc556b4 1748
a38718a6
GA
1749 /*
1750 * NB: We are assuming that the zio can only be dispatched
1751 * to a single taskq at a time. It would be a grievous error
1752 * to dispatch the zio to another taskq at the same time.
1753 */
1754 ASSERT(taskq_empty_ent(&zio->io_tqent));
7ef5e54e
AL
1755 spa_taskq_dispatch_ent(spa, t, q, (task_func_t *)zio_execute, zio,
1756 flags, &zio->io_tqent);
b128c09f 1757}
34dc7c2f 1758
b128c09f 1759static boolean_t
7ef5e54e 1760zio_taskq_member(zio_t *zio, zio_taskq_type_t q)
b128c09f
BB
1761{
1762 kthread_t *executor = zio->io_executor;
1763 spa_t *spa = zio->io_spa;
34dc7c2f 1764
1c27024e 1765 for (zio_type_t t = 0; t < ZIO_TYPES; t++) {
7ef5e54e
AL
1766 spa_taskqs_t *tqs = &spa->spa_zio_taskq[t][q];
1767 uint_t i;
1768 for (i = 0; i < tqs->stqs_count; i++) {
1769 if (taskq_member(tqs->stqs_taskq[i], executor))
1770 return (B_TRUE);
1771 }
1772 }
34dc7c2f 1773
b128c09f
BB
1774 return (B_FALSE);
1775}
34dc7c2f 1776
62840030 1777static zio_t *
b128c09f
BB
1778zio_issue_async(zio_t *zio)
1779{
428870ff 1780 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_FALSE);
b128c09f 1781
62840030 1782 return (NULL);
34dc7c2f
BB
1783}
1784
b128c09f
BB
1785void
1786zio_interrupt(zio_t *zio)
34dc7c2f 1787{
428870ff 1788 zio_taskq_dispatch(zio, ZIO_TASKQ_INTERRUPT, B_FALSE);
b128c09f 1789}
34dc7c2f 1790
26ef0cc7
TH
1791void
1792zio_delay_interrupt(zio_t *zio)
1793{
1794 /*
1795 * The timeout_generic() function isn't defined in userspace, so
1796 * rather than trying to implement the function, the zio delay
1797 * functionality has been disabled for userspace builds.
1798 */
1799
1800#ifdef _KERNEL
1801 /*
1802 * If io_target_timestamp is zero, then no delay has been registered
1803 * for this IO, thus jump to the end of this function and "skip" the
1804 * delay; issuing it directly to the zio layer.
1805 */
1806 if (zio->io_target_timestamp != 0) {
1807 hrtime_t now = gethrtime();
1808
1809 if (now >= zio->io_target_timestamp) {
1810 /*
1811 * This IO has already taken longer than the target
1812 * delay to complete, so we don't want to delay it
1813 * any longer; we "miss" the delay and issue it
1814 * directly to the zio layer. This is likely due to
1815 * the target latency being set to a value less than
1816 * the underlying hardware can satisfy (e.g. delay
1817 * set to 1ms, but the disks take 10ms to complete an
1818 * IO request).
1819 */
1820
1821 DTRACE_PROBE2(zio__delay__miss, zio_t *, zio,
1822 hrtime_t, now);
1823
1824 zio_interrupt(zio);
1825 } else {
1826 taskqid_t tid;
1827 hrtime_t diff = zio->io_target_timestamp - now;
1828 clock_t expire_at_tick = ddi_get_lbolt() +
1829 NSEC_TO_TICK(diff);
1830
1831 DTRACE_PROBE3(zio__delay__hit, zio_t *, zio,
1832 hrtime_t, now, hrtime_t, diff);
1833
1834 if (NSEC_TO_TICK(diff) == 0) {
1835 /* Our delay is less than a jiffy - just spin */
1836 zfs_sleep_until(zio->io_target_timestamp);
1837 } else {
1838 /*
1839 * Use taskq_dispatch_delay() in the place of
1840 * OpenZFS's timeout_generic().
1841 */
1842 tid = taskq_dispatch_delay(system_taskq,
02730c33 1843 (task_func_t *)zio_interrupt,
26ef0cc7 1844 zio, TQ_NOSLEEP, expire_at_tick);
48d3eb40 1845 if (tid == TASKQID_INVALID) {
26ef0cc7
TH
1846 /*
1847 * Couldn't allocate a task. Just
1848 * finish the zio without a delay.
1849 */
1850 zio_interrupt(zio);
1851 }
1852 }
1853 }
1854 return;
1855 }
1856#endif
1857 DTRACE_PROBE1(zio__delay__skip, zio_t *, zio);
1858 zio_interrupt(zio);
1859}
1860
8fb1ede1 1861static void
638dd5f4 1862zio_deadman_impl(zio_t *pio, int ziodepth)
8fb1ede1
BB
1863{
1864 zio_t *cio, *cio_next;
1865 zio_link_t *zl = NULL;
1866 vdev_t *vd = pio->io_vd;
1867
638dd5f4
TC
1868 if (zio_deadman_log_all || (vd != NULL && vd->vdev_ops->vdev_op_leaf)) {
1869 vdev_queue_t *vq = vd ? &vd->vdev_queue : NULL;
8fb1ede1
BB
1870 zbookmark_phys_t *zb = &pio->io_bookmark;
1871 uint64_t delta = gethrtime() - pio->io_timestamp;
1872 uint64_t failmode = spa_get_deadman_failmode(pio->io_spa);
1873
638dd5f4 1874 zfs_dbgmsg("slow zio[%d]: zio=%p timestamp=%llu "
8fb1ede1
BB
1875 "delta=%llu queued=%llu io=%llu "
1876 "path=%s last=%llu "
1877 "type=%d priority=%d flags=0x%x "
1878 "stage=0x%x pipeline=0x%x pipeline-trace=0x%x "
1879 "objset=%llu object=%llu level=%llu blkid=%llu "
1880 "offset=%llu size=%llu error=%d",
638dd5f4 1881 ziodepth, pio, pio->io_timestamp,
8fb1ede1 1882 delta, pio->io_delta, pio->io_delay,
638dd5f4 1883 vd ? vd->vdev_path : "NULL", vq ? vq->vq_io_complete_ts : 0,
8fb1ede1 1884 pio->io_type, pio->io_priority, pio->io_flags,
638dd5f4 1885 pio->io_stage, pio->io_pipeline, pio->io_pipeline_trace,
8fb1ede1
BB
1886 zb->zb_objset, zb->zb_object, zb->zb_level, zb->zb_blkid,
1887 pio->io_offset, pio->io_size, pio->io_error);
1888 zfs_ereport_post(FM_EREPORT_ZFS_DEADMAN,
1889 pio->io_spa, vd, zb, pio, 0, 0);
1890
1891 if (failmode == ZIO_FAILURE_MODE_CONTINUE &&
1892 taskq_empty_ent(&pio->io_tqent)) {
1893 zio_interrupt(pio);
1894 }
1895 }
1896
1897 mutex_enter(&pio->io_lock);
1898 for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
1899 cio_next = zio_walk_children(pio, &zl);
638dd5f4 1900 zio_deadman_impl(cio, ziodepth + 1);
8fb1ede1
BB
1901 }
1902 mutex_exit(&pio->io_lock);
1903}
1904
1905/*
1906 * Log the critical information describing this zio and all of its children
1907 * using the zfs_dbgmsg() interface then post deadman event for the ZED.
1908 */
1909void
1910zio_deadman(zio_t *pio, char *tag)
1911{
1912 spa_t *spa = pio->io_spa;
1913 char *name = spa_name(spa);
1914
1915 if (!zfs_deadman_enabled || spa_suspended(spa))
1916 return;
1917
638dd5f4 1918 zio_deadman_impl(pio, 0);
8fb1ede1
BB
1919
1920 switch (spa_get_deadman_failmode(spa)) {
1921 case ZIO_FAILURE_MODE_WAIT:
1922 zfs_dbgmsg("%s waiting for hung I/O to pool '%s'", tag, name);
1923 break;
1924
1925 case ZIO_FAILURE_MODE_CONTINUE:
1926 zfs_dbgmsg("%s restarting hung I/O for pool '%s'", tag, name);
1927 break;
1928
1929 case ZIO_FAILURE_MODE_PANIC:
1930 fm_panic("%s determined I/O to pool '%s' is hung.", tag, name);
1931 break;
1932 }
1933}
1934
b128c09f
BB
1935/*
1936 * Execute the I/O pipeline until one of the following occurs:
1937 * (1) the I/O completes; (2) the pipeline stalls waiting for
1938 * dependent child I/Os; (3) the I/O issues, so we're waiting
1939 * for an I/O completion interrupt; (4) the I/O is delegated by
1940 * vdev-level caching or aggregation; (5) the I/O is deferred
1941 * due to vdev-level queueing; (6) the I/O is handed off to
1942 * another thread. In all cases, the pipeline stops whenever
8e07b99b 1943 * there's no CPU work; it never burns a thread in cv_wait_io().
b128c09f
BB
1944 *
1945 * There's no locking on io_stage because there's no legitimate way
1946 * for multiple threads to be attempting to process the same I/O.
1947 */
428870ff 1948static zio_pipe_stage_t *zio_pipeline[];
34dc7c2f 1949
da6b4005
NB
1950/*
1951 * zio_execute() is a wrapper around the static function
1952 * __zio_execute() so that we can force __zio_execute() to be
1953 * inlined. This reduces stack overhead which is important
1954 * because __zio_execute() is called recursively in several zio
1955 * code paths. zio_execute() itself cannot be inlined because
1956 * it is externally visible.
1957 */
b128c09f
BB
1958void
1959zio_execute(zio_t *zio)
da6b4005 1960{
92119cc2
BB
1961 fstrans_cookie_t cookie;
1962
1963 cookie = spl_fstrans_mark();
da6b4005 1964 __zio_execute(zio);
92119cc2 1965 spl_fstrans_unmark(cookie);
da6b4005
NB
1966}
1967
b58986ee
BB
1968/*
1969 * Used to determine if in the current context the stack is sized large
1970 * enough to allow zio_execute() to be called recursively. A minimum
1971 * stack size of 16K is required to avoid needing to re-dispatch the zio.
1972 */
1973boolean_t
1974zio_execute_stack_check(zio_t *zio)
1975{
1976#if !defined(HAVE_LARGE_STACKS)
1977 dsl_pool_t *dp = spa_get_dsl(zio->io_spa);
1978
1979 /* Executing in txg_sync_thread() context. */
1980 if (dp && curthread == dp->dp_tx.tx_sync_thread)
1981 return (B_TRUE);
1982
1983 /* Pool initialization outside of zio_taskq context. */
1984 if (dp && spa_is_initializing(dp->dp_spa) &&
1985 !zio_taskq_member(zio, ZIO_TASKQ_ISSUE) &&
1986 !zio_taskq_member(zio, ZIO_TASKQ_ISSUE_HIGH))
1987 return (B_TRUE);
1988#endif /* HAVE_LARGE_STACKS */
1989
1990 return (B_FALSE);
1991}
1992
da6b4005
NB
1993__attribute__((always_inline))
1994static inline void
1995__zio_execute(zio_t *zio)
b128c09f 1996{
3dfb57a3
DB
1997 ASSERT3U(zio->io_queued_timestamp, >, 0);
1998
b128c09f 1999 while (zio->io_stage < ZIO_STAGE_DONE) {
428870ff
BB
2000 enum zio_stage pipeline = zio->io_pipeline;
2001 enum zio_stage stage = zio->io_stage;
62840030
MA
2002
2003 zio->io_executor = curthread;
34dc7c2f 2004
b128c09f 2005 ASSERT(!MUTEX_HELD(&zio->io_lock));
428870ff
BB
2006 ASSERT(ISP2(stage));
2007 ASSERT(zio->io_stall == NULL);
34dc7c2f 2008
428870ff
BB
2009 do {
2010 stage <<= 1;
2011 } while ((stage & pipeline) == 0);
b128c09f
BB
2012
2013 ASSERT(stage <= ZIO_STAGE_DONE);
34dc7c2f
BB
2014
2015 /*
b128c09f
BB
2016 * If we are in interrupt context and this pipeline stage
2017 * will grab a config lock that is held across I/O,
428870ff
BB
2018 * or may wait for an I/O that needs an interrupt thread
2019 * to complete, issue async to avoid deadlock.
2020 *
2021 * For VDEV_IO_START, we cut in line so that the io will
2022 * be sent to disk promptly.
34dc7c2f 2023 */
91579709
BB
2024 if ((stage & ZIO_BLOCKING_STAGES) && zio->io_vd == NULL &&
2025 zio_taskq_member(zio, ZIO_TASKQ_INTERRUPT)) {
b58986ee
BB
2026 boolean_t cut = (stage == ZIO_STAGE_VDEV_IO_START) ?
2027 zio_requeue_io_start_cut_in_line : B_FALSE;
91579709
BB
2028 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, cut);
2029 return;
2030 }
2031
2032 /*
b58986ee
BB
2033 * If the current context doesn't have large enough stacks
2034 * the zio must be issued asynchronously to prevent overflow.
91579709 2035 */
b58986ee
BB
2036 if (zio_execute_stack_check(zio)) {
2037 boolean_t cut = (stage == ZIO_STAGE_VDEV_IO_START) ?
2038 zio_requeue_io_start_cut_in_line : B_FALSE;
428870ff 2039 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, cut);
b128c09f 2040 return;
34dc7c2f
BB
2041 }
2042
b128c09f 2043 zio->io_stage = stage;
3dfb57a3 2044 zio->io_pipeline_trace |= zio->io_stage;
34dc7c2f 2045
62840030
MA
2046 /*
2047 * The zio pipeline stage returns the next zio to execute
2048 * (typically the same as this one), or NULL if we should
2049 * stop.
2050 */
2051 zio = zio_pipeline[highbit64(stage) - 1](zio);
34dc7c2f 2052
62840030
MA
2053 if (zio == NULL)
2054 return;
b128c09f 2055 }
34dc7c2f
BB
2056}
2057
da6b4005 2058
b128c09f
BB
2059/*
2060 * ==========================================================================
2061 * Initiate I/O, either sync or async
2062 * ==========================================================================
2063 */
2064int
2065zio_wait(zio_t *zio)
34dc7c2f 2066{
8fb1ede1 2067 long timeout = MSEC_TO_TICK(zfs_deadman_ziotime_ms);
b128c09f 2068 int error;
34dc7c2f 2069
1ce23dca
PS
2070 ASSERT3S(zio->io_stage, ==, ZIO_STAGE_OPEN);
2071 ASSERT3P(zio->io_executor, ==, NULL);
34dc7c2f 2072
b128c09f 2073 zio->io_waiter = curthread;
3dfb57a3
DB
2074 ASSERT0(zio->io_queued_timestamp);
2075 zio->io_queued_timestamp = gethrtime();
34dc7c2f 2076
da6b4005 2077 __zio_execute(zio);
34dc7c2f 2078
b128c09f 2079 mutex_enter(&zio->io_lock);
8fb1ede1
BB
2080 while (zio->io_executor != NULL) {
2081 error = cv_timedwait_io(&zio->io_cv, &zio->io_lock,
2082 ddi_get_lbolt() + timeout);
2083
2084 if (zfs_deadman_enabled && error == -1 &&
2085 gethrtime() - zio->io_queued_timestamp >
2086 spa_deadman_ziotime(zio->io_spa)) {
2087 mutex_exit(&zio->io_lock);
2088 timeout = MSEC_TO_TICK(zfs_deadman_checktime_ms);
2089 zio_deadman(zio, FTAG);
2090 mutex_enter(&zio->io_lock);
2091 }
2092 }
b128c09f 2093 mutex_exit(&zio->io_lock);
34dc7c2f 2094
b128c09f
BB
2095 error = zio->io_error;
2096 zio_destroy(zio);
34dc7c2f 2097
b128c09f
BB
2098 return (error);
2099}
34dc7c2f 2100
b128c09f
BB
2101void
2102zio_nowait(zio_t *zio)
2103{
1ce23dca 2104 ASSERT3P(zio->io_executor, ==, NULL);
34dc7c2f 2105
d164b209
BB
2106 if (zio->io_child_type == ZIO_CHILD_LOGICAL &&
2107 zio_unique_parent(zio) == NULL) {
8878261f
BB
2108 zio_t *pio;
2109
34dc7c2f 2110 /*
b128c09f 2111 * This is a logical async I/O with no parent to wait for it.
9babb374
BB
2112 * We add it to the spa_async_root_zio "Godfather" I/O which
2113 * will ensure they complete prior to unloading the pool.
34dc7c2f 2114 */
b128c09f 2115 spa_t *spa = zio->io_spa;
8878261f
BB
2116 kpreempt_disable();
2117 pio = spa->spa_async_zio_root[CPU_SEQID];
2118 kpreempt_enable();
9babb374 2119
8878261f 2120 zio_add_child(pio, zio);
b128c09f 2121 }
34dc7c2f 2122
3dfb57a3
DB
2123 ASSERT0(zio->io_queued_timestamp);
2124 zio->io_queued_timestamp = gethrtime();
da6b4005 2125 __zio_execute(zio);
b128c09f 2126}
34dc7c2f 2127
b128c09f
BB
2128/*
2129 * ==========================================================================
1ce23dca 2130 * Reexecute, cancel, or suspend/resume failed I/O
b128c09f
BB
2131 * ==========================================================================
2132 */
34dc7c2f 2133
b128c09f
BB
2134static void
2135zio_reexecute(zio_t *pio)
2136{
d164b209
BB
2137 zio_t *cio, *cio_next;
2138
2139 ASSERT(pio->io_child_type == ZIO_CHILD_LOGICAL);
2140 ASSERT(pio->io_orig_stage == ZIO_STAGE_OPEN);
9babb374
BB
2141 ASSERT(pio->io_gang_leader == NULL);
2142 ASSERT(pio->io_gang_tree == NULL);
34dc7c2f 2143
b128c09f
BB
2144 pio->io_flags = pio->io_orig_flags;
2145 pio->io_stage = pio->io_orig_stage;
2146 pio->io_pipeline = pio->io_orig_pipeline;
2147 pio->io_reexecute = 0;
03c6040b 2148 pio->io_flags |= ZIO_FLAG_REEXECUTED;
3dfb57a3 2149 pio->io_pipeline_trace = 0;
b128c09f 2150 pio->io_error = 0;
1c27024e 2151 for (int w = 0; w < ZIO_WAIT_TYPES; w++)
d164b209 2152 pio->io_state[w] = 0;
1c27024e 2153 for (int c = 0; c < ZIO_CHILD_TYPES; c++)
b128c09f 2154 pio->io_child_error[c] = 0;
34dc7c2f 2155
428870ff
BB
2156 if (IO_IS_ALLOCATING(pio))
2157 BP_ZERO(pio->io_bp);
34dc7c2f 2158
b128c09f
BB
2159 /*
2160 * As we reexecute pio's children, new children could be created.
d164b209 2161 * New children go to the head of pio's io_child_list, however,
b128c09f 2162 * so we will (correctly) not reexecute them. The key is that
d164b209
BB
2163 * the remainder of pio's io_child_list, from 'cio_next' onward,
2164 * cannot be affected by any side effects of reexecuting 'cio'.
b128c09f 2165 */
1c27024e 2166 zio_link_t *zl = NULL;
a8b2e306 2167 mutex_enter(&pio->io_lock);
3dfb57a3
DB
2168 for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
2169 cio_next = zio_walk_children(pio, &zl);
1c27024e 2170 for (int w = 0; w < ZIO_WAIT_TYPES; w++)
d164b209 2171 pio->io_children[cio->io_child_type][w]++;
b128c09f 2172 mutex_exit(&pio->io_lock);
d164b209 2173 zio_reexecute(cio);
a8b2e306 2174 mutex_enter(&pio->io_lock);
34dc7c2f 2175 }
a8b2e306 2176 mutex_exit(&pio->io_lock);
34dc7c2f 2177
b128c09f
BB
2178 /*
2179 * Now that all children have been reexecuted, execute the parent.
9babb374 2180 * We don't reexecute "The Godfather" I/O here as it's the
9e2c3bb4 2181 * responsibility of the caller to wait on it.
b128c09f 2182 */
3dfb57a3
DB
2183 if (!(pio->io_flags & ZIO_FLAG_GODFATHER)) {
2184 pio->io_queued_timestamp = gethrtime();
da6b4005 2185 __zio_execute(pio);
3dfb57a3 2186 }
34dc7c2f
BB
2187}
2188
b128c09f 2189void
cec3a0a1 2190zio_suspend(spa_t *spa, zio_t *zio, zio_suspend_reason_t reason)
34dc7c2f 2191{
b128c09f
BB
2192 if (spa_get_failmode(spa) == ZIO_FAILURE_MODE_PANIC)
2193 fm_panic("Pool '%s' has encountered an uncorrectable I/O "
2194 "failure and the failure mode property for this pool "
2195 "is set to panic.", spa_name(spa));
34dc7c2f 2196
bf89c199
BB
2197 cmn_err(CE_WARN, "Pool '%s' has encountered an uncorrectable I/O "
2198 "failure and has been suspended.\n", spa_name(spa));
2199
b5256303
TC
2200 zfs_ereport_post(FM_EREPORT_ZFS_IO_FAILURE, spa, NULL,
2201 NULL, NULL, 0, 0);
34dc7c2f 2202
b128c09f 2203 mutex_enter(&spa->spa_suspend_lock);
34dc7c2f 2204
b128c09f 2205 if (spa->spa_suspend_zio_root == NULL)
9babb374
BB
2206 spa->spa_suspend_zio_root = zio_root(spa, NULL, NULL,
2207 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE |
2208 ZIO_FLAG_GODFATHER);
34dc7c2f 2209
cec3a0a1 2210 spa->spa_suspended = reason;
34dc7c2f 2211
b128c09f 2212 if (zio != NULL) {
9babb374 2213 ASSERT(!(zio->io_flags & ZIO_FLAG_GODFATHER));
b128c09f
BB
2214 ASSERT(zio != spa->spa_suspend_zio_root);
2215 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
d164b209 2216 ASSERT(zio_unique_parent(zio) == NULL);
b128c09f
BB
2217 ASSERT(zio->io_stage == ZIO_STAGE_DONE);
2218 zio_add_child(spa->spa_suspend_zio_root, zio);
2219 }
34dc7c2f 2220
b128c09f
BB
2221 mutex_exit(&spa->spa_suspend_lock);
2222}
34dc7c2f 2223
9babb374 2224int
b128c09f
BB
2225zio_resume(spa_t *spa)
2226{
9babb374 2227 zio_t *pio;
34dc7c2f
BB
2228
2229 /*
b128c09f 2230 * Reexecute all previously suspended i/o.
34dc7c2f 2231 */
b128c09f 2232 mutex_enter(&spa->spa_suspend_lock);
cec3a0a1 2233 spa->spa_suspended = ZIO_SUSPEND_NONE;
b128c09f
BB
2234 cv_broadcast(&spa->spa_suspend_cv);
2235 pio = spa->spa_suspend_zio_root;
2236 spa->spa_suspend_zio_root = NULL;
2237 mutex_exit(&spa->spa_suspend_lock);
2238
2239 if (pio == NULL)
9babb374 2240 return (0);
34dc7c2f 2241
9babb374
BB
2242 zio_reexecute(pio);
2243 return (zio_wait(pio));
b128c09f
BB
2244}
2245
2246void
2247zio_resume_wait(spa_t *spa)
2248{
2249 mutex_enter(&spa->spa_suspend_lock);
2250 while (spa_suspended(spa))
2251 cv_wait(&spa->spa_suspend_cv, &spa->spa_suspend_lock);
2252 mutex_exit(&spa->spa_suspend_lock);
34dc7c2f
BB
2253}
2254
2255/*
2256 * ==========================================================================
b128c09f
BB
2257 * Gang blocks.
2258 *
2259 * A gang block is a collection of small blocks that looks to the DMU
2260 * like one large block. When zio_dva_allocate() cannot find a block
2261 * of the requested size, due to either severe fragmentation or the pool
2262 * being nearly full, it calls zio_write_gang_block() to construct the
2263 * block from smaller fragments.
2264 *
2265 * A gang block consists of a gang header (zio_gbh_phys_t) and up to
2266 * three (SPA_GBH_NBLKPTRS) gang members. The gang header is just like
2267 * an indirect block: it's an array of block pointers. It consumes
2268 * only one sector and hence is allocatable regardless of fragmentation.
2269 * The gang header's bps point to its gang members, which hold the data.
2270 *
2271 * Gang blocks are self-checksumming, using the bp's <vdev, offset, txg>
2272 * as the verifier to ensure uniqueness of the SHA256 checksum.
2273 * Critically, the gang block bp's blk_cksum is the checksum of the data,
2274 * not the gang header. This ensures that data block signatures (needed for
2275 * deduplication) are independent of how the block is physically stored.
2276 *
2277 * Gang blocks can be nested: a gang member may itself be a gang block.
2278 * Thus every gang block is a tree in which root and all interior nodes are
2279 * gang headers, and the leaves are normal blocks that contain user data.
2280 * The root of the gang tree is called the gang leader.
2281 *
2282 * To perform any operation (read, rewrite, free, claim) on a gang block,
2283 * zio_gang_assemble() first assembles the gang tree (minus data leaves)
2284 * in the io_gang_tree field of the original logical i/o by recursively
2285 * reading the gang leader and all gang headers below it. This yields
2286 * an in-core tree containing the contents of every gang header and the
2287 * bps for every constituent of the gang block.
2288 *
2289 * With the gang tree now assembled, zio_gang_issue() just walks the gang tree
2290 * and invokes a callback on each bp. To free a gang block, zio_gang_issue()
2291 * calls zio_free_gang() -- a trivial wrapper around zio_free() -- for each bp.
2292 * zio_claim_gang() provides a similarly trivial wrapper for zio_claim().
2293 * zio_read_gang() is a wrapper around zio_read() that omits reading gang
2294 * headers, since we already have those in io_gang_tree. zio_rewrite_gang()
2295 * performs a zio_rewrite() of the data or, for gang headers, a zio_rewrite()
2296 * of the gang header plus zio_checksum_compute() of the data to update the
2297 * gang header's blk_cksum as described above.
2298 *
2299 * The two-phase assemble/issue model solves the problem of partial failure --
2300 * what if you'd freed part of a gang block but then couldn't read the
2301 * gang header for another part? Assembling the entire gang tree first
2302 * ensures that all the necessary gang header I/O has succeeded before
2303 * starting the actual work of free, claim, or write. Once the gang tree
2304 * is assembled, free and claim are in-memory operations that cannot fail.
2305 *
2306 * In the event that a gang write fails, zio_dva_unallocate() walks the
2307 * gang tree to immediately free (i.e. insert back into the space map)
2308 * everything we've allocated. This ensures that we don't get ENOSPC
2309 * errors during repeated suspend/resume cycles due to a flaky device.
2310 *
2311 * Gang rewrites only happen during sync-to-convergence. If we can't assemble
2312 * the gang tree, we won't modify the block, so we can safely defer the free
2313 * (knowing that the block is still intact). If we *can* assemble the gang
2314 * tree, then even if some of the rewrites fail, zio_dva_unallocate() will free
2315 * each constituent bp and we can allocate a new block on the next sync pass.
2316 *
2317 * In all cases, the gang tree allows complete recovery from partial failure.
34dc7c2f
BB
2318 * ==========================================================================
2319 */
b128c09f 2320
a6255b7f
DQ
2321static void
2322zio_gang_issue_func_done(zio_t *zio)
2323{
2324 abd_put(zio->io_abd);
2325}
2326
b128c09f 2327static zio_t *
a6255b7f
DQ
2328zio_read_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2329 uint64_t offset)
34dc7c2f 2330{
b128c09f
BB
2331 if (gn != NULL)
2332 return (pio);
34dc7c2f 2333
a6255b7f
DQ
2334 return (zio_read(pio, pio->io_spa, bp, abd_get_offset(data, offset),
2335 BP_GET_PSIZE(bp), zio_gang_issue_func_done,
2336 NULL, pio->io_priority, ZIO_GANG_CHILD_FLAGS(pio),
b128c09f
BB
2337 &pio->io_bookmark));
2338}
2339
a6255b7f
DQ
2340static zio_t *
2341zio_rewrite_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2342 uint64_t offset)
b128c09f
BB
2343{
2344 zio_t *zio;
2345
2346 if (gn != NULL) {
a6255b7f
DQ
2347 abd_t *gbh_abd =
2348 abd_get_from_buf(gn->gn_gbh, SPA_GANGBLOCKSIZE);
b128c09f 2349 zio = zio_rewrite(pio, pio->io_spa, pio->io_txg, bp,
a6255b7f
DQ
2350 gbh_abd, SPA_GANGBLOCKSIZE, zio_gang_issue_func_done, NULL,
2351 pio->io_priority, ZIO_GANG_CHILD_FLAGS(pio),
2352 &pio->io_bookmark);
34dc7c2f 2353 /*
b128c09f
BB
2354 * As we rewrite each gang header, the pipeline will compute
2355 * a new gang block header checksum for it; but no one will
2356 * compute a new data checksum, so we do that here. The one
2357 * exception is the gang leader: the pipeline already computed
2358 * its data checksum because that stage precedes gang assembly.
2359 * (Presently, nothing actually uses interior data checksums;
2360 * this is just good hygiene.)
34dc7c2f 2361 */
9babb374 2362 if (gn != pio->io_gang_leader->io_gang_tree) {
a6255b7f
DQ
2363 abd_t *buf = abd_get_offset(data, offset);
2364
b128c09f 2365 zio_checksum_compute(zio, BP_GET_CHECKSUM(bp),
a6255b7f
DQ
2366 buf, BP_GET_PSIZE(bp));
2367
2368 abd_put(buf);
b128c09f 2369 }
428870ff
BB
2370 /*
2371 * If we are here to damage data for testing purposes,
2372 * leave the GBH alone so that we can detect the damage.
2373 */
2374 if (pio->io_gang_leader->io_flags & ZIO_FLAG_INDUCE_DAMAGE)
2375 zio->io_pipeline &= ~ZIO_VDEV_IO_STAGES;
34dc7c2f 2376 } else {
b128c09f 2377 zio = zio_rewrite(pio, pio->io_spa, pio->io_txg, bp,
a6255b7f
DQ
2378 abd_get_offset(data, offset), BP_GET_PSIZE(bp),
2379 zio_gang_issue_func_done, NULL, pio->io_priority,
b128c09f 2380 ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
34dc7c2f
BB
2381 }
2382
b128c09f
BB
2383 return (zio);
2384}
34dc7c2f 2385
b128c09f 2386/* ARGSUSED */
a6255b7f
DQ
2387static zio_t *
2388zio_free_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2389 uint64_t offset)
b128c09f 2390{
428870ff
BB
2391 return (zio_free_sync(pio, pio->io_spa, pio->io_txg, bp,
2392 ZIO_GANG_CHILD_FLAGS(pio)));
34dc7c2f
BB
2393}
2394
b128c09f 2395/* ARGSUSED */
a6255b7f
DQ
2396static zio_t *
2397zio_claim_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2398 uint64_t offset)
34dc7c2f 2399{
b128c09f
BB
2400 return (zio_claim(pio, pio->io_spa, pio->io_txg, bp,
2401 NULL, NULL, ZIO_GANG_CHILD_FLAGS(pio)));
2402}
2403
2404static zio_gang_issue_func_t *zio_gang_issue_func[ZIO_TYPES] = {
2405 NULL,
2406 zio_read_gang,
2407 zio_rewrite_gang,
2408 zio_free_gang,
2409 zio_claim_gang,
2410 NULL
2411};
34dc7c2f 2412
b128c09f 2413static void zio_gang_tree_assemble_done(zio_t *zio);
34dc7c2f 2414
b128c09f
BB
2415static zio_gang_node_t *
2416zio_gang_node_alloc(zio_gang_node_t **gnpp)
2417{
2418 zio_gang_node_t *gn;
34dc7c2f 2419
b128c09f 2420 ASSERT(*gnpp == NULL);
34dc7c2f 2421
79c76d5b 2422 gn = kmem_zalloc(sizeof (*gn), KM_SLEEP);
b128c09f
BB
2423 gn->gn_gbh = zio_buf_alloc(SPA_GANGBLOCKSIZE);
2424 *gnpp = gn;
34dc7c2f 2425
b128c09f 2426 return (gn);
34dc7c2f
BB
2427}
2428
34dc7c2f 2429static void
b128c09f 2430zio_gang_node_free(zio_gang_node_t **gnpp)
34dc7c2f 2431{
b128c09f 2432 zio_gang_node_t *gn = *gnpp;
34dc7c2f 2433
1c27024e 2434 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++)
b128c09f
BB
2435 ASSERT(gn->gn_child[g] == NULL);
2436
2437 zio_buf_free(gn->gn_gbh, SPA_GANGBLOCKSIZE);
2438 kmem_free(gn, sizeof (*gn));
2439 *gnpp = NULL;
34dc7c2f
BB
2440}
2441
b128c09f
BB
2442static void
2443zio_gang_tree_free(zio_gang_node_t **gnpp)
34dc7c2f 2444{
b128c09f 2445 zio_gang_node_t *gn = *gnpp;
34dc7c2f 2446
b128c09f
BB
2447 if (gn == NULL)
2448 return;
34dc7c2f 2449
1c27024e 2450 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++)
b128c09f 2451 zio_gang_tree_free(&gn->gn_child[g]);
34dc7c2f 2452
b128c09f 2453 zio_gang_node_free(gnpp);
34dc7c2f
BB
2454}
2455
b128c09f 2456static void
9babb374 2457zio_gang_tree_assemble(zio_t *gio, blkptr_t *bp, zio_gang_node_t **gnpp)
34dc7c2f 2458{
b128c09f 2459 zio_gang_node_t *gn = zio_gang_node_alloc(gnpp);
a6255b7f 2460 abd_t *gbh_abd = abd_get_from_buf(gn->gn_gbh, SPA_GANGBLOCKSIZE);
b128c09f 2461
9babb374 2462 ASSERT(gio->io_gang_leader == gio);
b128c09f 2463 ASSERT(BP_IS_GANG(bp));
34dc7c2f 2464
a6255b7f
DQ
2465 zio_nowait(zio_read(gio, gio->io_spa, bp, gbh_abd, SPA_GANGBLOCKSIZE,
2466 zio_gang_tree_assemble_done, gn, gio->io_priority,
2467 ZIO_GANG_CHILD_FLAGS(gio), &gio->io_bookmark));
b128c09f 2468}
34dc7c2f 2469
b128c09f
BB
2470static void
2471zio_gang_tree_assemble_done(zio_t *zio)
2472{
9babb374 2473 zio_t *gio = zio->io_gang_leader;
b128c09f
BB
2474 zio_gang_node_t *gn = zio->io_private;
2475 blkptr_t *bp = zio->io_bp;
34dc7c2f 2476
9babb374 2477 ASSERT(gio == zio_unique_parent(zio));
428870ff 2478 ASSERT(zio->io_child_count == 0);
34dc7c2f 2479
b128c09f
BB
2480 if (zio->io_error)
2481 return;
34dc7c2f 2482
a6255b7f 2483 /* this ABD was created from a linear buf in zio_gang_tree_assemble */
b128c09f 2484 if (BP_SHOULD_BYTESWAP(bp))
a6255b7f 2485 byteswap_uint64_array(abd_to_buf(zio->io_abd), zio->io_size);
34dc7c2f 2486
a6255b7f 2487 ASSERT3P(abd_to_buf(zio->io_abd), ==, gn->gn_gbh);
b128c09f 2488 ASSERT(zio->io_size == SPA_GANGBLOCKSIZE);
428870ff 2489 ASSERT(gn->gn_gbh->zg_tail.zec_magic == ZEC_MAGIC);
34dc7c2f 2490
a6255b7f
DQ
2491 abd_put(zio->io_abd);
2492
1c27024e 2493 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
b128c09f
BB
2494 blkptr_t *gbp = &gn->gn_gbh->zg_blkptr[g];
2495 if (!BP_IS_GANG(gbp))
2496 continue;
9babb374 2497 zio_gang_tree_assemble(gio, gbp, &gn->gn_child[g]);
b128c09f 2498 }
34dc7c2f
BB
2499}
2500
b128c09f 2501static void
a6255b7f
DQ
2502zio_gang_tree_issue(zio_t *pio, zio_gang_node_t *gn, blkptr_t *bp, abd_t *data,
2503 uint64_t offset)
34dc7c2f 2504{
9babb374 2505 zio_t *gio = pio->io_gang_leader;
b128c09f 2506 zio_t *zio;
34dc7c2f 2507
b128c09f 2508 ASSERT(BP_IS_GANG(bp) == !!gn);
9babb374
BB
2509 ASSERT(BP_GET_CHECKSUM(bp) == BP_GET_CHECKSUM(gio->io_bp));
2510 ASSERT(BP_GET_LSIZE(bp) == BP_GET_PSIZE(bp) || gn == gio->io_gang_tree);
34dc7c2f 2511
b128c09f
BB
2512 /*
2513 * If you're a gang header, your data is in gn->gn_gbh.
2514 * If you're a gang member, your data is in 'data' and gn == NULL.
2515 */
a6255b7f 2516 zio = zio_gang_issue_func[gio->io_type](pio, bp, gn, data, offset);
34dc7c2f 2517
b128c09f 2518 if (gn != NULL) {
428870ff 2519 ASSERT(gn->gn_gbh->zg_tail.zec_magic == ZEC_MAGIC);
34dc7c2f 2520
1c27024e 2521 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
b128c09f
BB
2522 blkptr_t *gbp = &gn->gn_gbh->zg_blkptr[g];
2523 if (BP_IS_HOLE(gbp))
2524 continue;
a6255b7f
DQ
2525 zio_gang_tree_issue(zio, gn->gn_child[g], gbp, data,
2526 offset);
2527 offset += BP_GET_PSIZE(gbp);
b128c09f 2528 }
34dc7c2f
BB
2529 }
2530
9babb374 2531 if (gn == gio->io_gang_tree)
a6255b7f 2532 ASSERT3U(gio->io_size, ==, offset);
34dc7c2f 2533
b128c09f
BB
2534 if (zio != pio)
2535 zio_nowait(zio);
34dc7c2f
BB
2536}
2537
62840030 2538static zio_t *
b128c09f 2539zio_gang_assemble(zio_t *zio)
34dc7c2f 2540{
b128c09f 2541 blkptr_t *bp = zio->io_bp;
34dc7c2f 2542
9babb374
BB
2543 ASSERT(BP_IS_GANG(bp) && zio->io_gang_leader == NULL);
2544 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
2545
2546 zio->io_gang_leader = zio;
34dc7c2f 2547
b128c09f 2548 zio_gang_tree_assemble(zio, bp, &zio->io_gang_tree);
34dc7c2f 2549
62840030 2550 return (zio);
34dc7c2f
BB
2551}
2552
62840030 2553static zio_t *
b128c09f 2554zio_gang_issue(zio_t *zio)
34dc7c2f 2555{
b128c09f 2556 blkptr_t *bp = zio->io_bp;
34dc7c2f 2557
ddc751d5 2558 if (zio_wait_for_children(zio, ZIO_CHILD_GANG_BIT, ZIO_WAIT_DONE)) {
62840030 2559 return (NULL);
ddc751d5 2560 }
34dc7c2f 2561
9babb374
BB
2562 ASSERT(BP_IS_GANG(bp) && zio->io_gang_leader == zio);
2563 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
34dc7c2f 2564
b128c09f 2565 if (zio->io_child_error[ZIO_CHILD_GANG] == 0)
a6255b7f
DQ
2566 zio_gang_tree_issue(zio, zio->io_gang_tree, bp, zio->io_abd,
2567 0);
b128c09f 2568 else
9babb374 2569 zio_gang_tree_free(&zio->io_gang_tree);
34dc7c2f 2570
b128c09f 2571 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
34dc7c2f 2572
62840030 2573 return (zio);
34dc7c2f
BB
2574}
2575
2576static void
b128c09f 2577zio_write_gang_member_ready(zio_t *zio)
34dc7c2f 2578{
d164b209 2579 zio_t *pio = zio_unique_parent(zio);
34dc7c2f
BB
2580 dva_t *cdva = zio->io_bp->blk_dva;
2581 dva_t *pdva = pio->io_bp->blk_dva;
2582 uint64_t asize;
d1d7e268 2583 ASSERTV(zio_t *gio = zio->io_gang_leader);
34dc7c2f 2584
b128c09f
BB
2585 if (BP_IS_HOLE(zio->io_bp))
2586 return;
2587
2588 ASSERT(BP_IS_HOLE(&zio->io_bp_orig));
2589
2590 ASSERT(zio->io_child_type == ZIO_CHILD_GANG);
428870ff
BB
2591 ASSERT3U(zio->io_prop.zp_copies, ==, gio->io_prop.zp_copies);
2592 ASSERT3U(zio->io_prop.zp_copies, <=, BP_GET_NDVAS(zio->io_bp));
2593 ASSERT3U(pio->io_prop.zp_copies, <=, BP_GET_NDVAS(pio->io_bp));
34dc7c2f 2594 ASSERT3U(BP_GET_NDVAS(zio->io_bp), <=, BP_GET_NDVAS(pio->io_bp));
34dc7c2f
BB
2595
2596 mutex_enter(&pio->io_lock);
1c27024e 2597 for (int d = 0; d < BP_GET_NDVAS(zio->io_bp); d++) {
34dc7c2f
BB
2598 ASSERT(DVA_GET_GANG(&pdva[d]));
2599 asize = DVA_GET_ASIZE(&pdva[d]);
2600 asize += DVA_GET_ASIZE(&cdva[d]);
2601 DVA_SET_ASIZE(&pdva[d], asize);
2602 }
2603 mutex_exit(&pio->io_lock);
2604}
2605
a6255b7f
DQ
2606static void
2607zio_write_gang_done(zio_t *zio)
2608{
c955398b
BL
2609 /*
2610 * The io_abd field will be NULL for a zio with no data. The io_flags
2611 * will initially have the ZIO_FLAG_NODATA bit flag set, but we can't
2612 * check for it here as it is cleared in zio_ready.
2613 */
2614 if (zio->io_abd != NULL)
2615 abd_put(zio->io_abd);
a6255b7f
DQ
2616}
2617
62840030 2618static zio_t *
b128c09f 2619zio_write_gang_block(zio_t *pio)
34dc7c2f 2620{
b128c09f 2621 spa_t *spa = pio->io_spa;
3dfb57a3 2622 metaslab_class_t *mc = spa_normal_class(spa);
b128c09f 2623 blkptr_t *bp = pio->io_bp;
9babb374 2624 zio_t *gio = pio->io_gang_leader;
b128c09f
BB
2625 zio_t *zio;
2626 zio_gang_node_t *gn, **gnpp;
34dc7c2f 2627 zio_gbh_phys_t *gbh;
a6255b7f 2628 abd_t *gbh_abd;
b128c09f
BB
2629 uint64_t txg = pio->io_txg;
2630 uint64_t resid = pio->io_size;
2631 uint64_t lsize;
428870ff 2632 int copies = gio->io_prop.zp_copies;
b5256303 2633 int gbh_copies;
b128c09f 2634 zio_prop_t zp;
1c27024e 2635 int error;
c955398b 2636 boolean_t has_data = !(pio->io_flags & ZIO_FLAG_NODATA);
b5256303
TC
2637
2638 /*
2639 * encrypted blocks need DVA[2] free so encrypted gang headers can't
2640 * have a third copy.
2641 */
2642 gbh_copies = MIN(copies + 1, spa_max_replication(spa));
2643 if (gio->io_prop.zp_encrypt && gbh_copies >= SPA_DVAS_PER_BP)
2644 gbh_copies = SPA_DVAS_PER_BP - 1;
2645
1c27024e 2646 int flags = METASLAB_HINTBP_FAVOR | METASLAB_GANG_HEADER;
3dfb57a3
DB
2647 if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
2648 ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
c955398b 2649 ASSERT(has_data);
3dfb57a3
DB
2650
2651 flags |= METASLAB_ASYNC_ALLOC;
424fd7c3 2652 VERIFY(zfs_refcount_held(&mc->mc_alloc_slots[pio->io_allocator],
492f64e9 2653 pio));
3dfb57a3
DB
2654
2655 /*
2656 * The logical zio has already placed a reservation for
2657 * 'copies' allocation slots but gang blocks may require
2658 * additional copies. These additional copies
2659 * (i.e. gbh_copies - copies) are guaranteed to succeed
2660 * since metaslab_class_throttle_reserve() always allows
2661 * additional reservations for gang blocks.
2662 */
2663 VERIFY(metaslab_class_throttle_reserve(mc, gbh_copies - copies,
492f64e9 2664 pio->io_allocator, pio, flags));
3dfb57a3
DB
2665 }
2666
2667 error = metaslab_alloc(spa, mc, SPA_GANGBLOCKSIZE,
4e21fd06 2668 bp, gbh_copies, txg, pio == gio ? NULL : gio->io_bp, flags,
492f64e9 2669 &pio->io_alloc_list, pio, pio->io_allocator);
34dc7c2f 2670 if (error) {
3dfb57a3
DB
2671 if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
2672 ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
c955398b 2673 ASSERT(has_data);
3dfb57a3
DB
2674
2675 /*
2676 * If we failed to allocate the gang block header then
2677 * we remove any additional allocation reservations that
2678 * we placed here. The original reservation will
2679 * be removed when the logical I/O goes to the ready
2680 * stage.
2681 */
2682 metaslab_class_throttle_unreserve(mc,
492f64e9 2683 gbh_copies - copies, pio->io_allocator, pio);
3dfb57a3
DB
2684 }
2685
b128c09f 2686 pio->io_error = error;
62840030 2687 return (pio);
34dc7c2f
BB
2688 }
2689
9babb374
BB
2690 if (pio == gio) {
2691 gnpp = &gio->io_gang_tree;
b128c09f
BB
2692 } else {
2693 gnpp = pio->io_private;
2694 ASSERT(pio->io_ready == zio_write_gang_member_ready);
34dc7c2f
BB
2695 }
2696
b128c09f
BB
2697 gn = zio_gang_node_alloc(gnpp);
2698 gbh = gn->gn_gbh;
2699 bzero(gbh, SPA_GANGBLOCKSIZE);
a6255b7f 2700 gbh_abd = abd_get_from_buf(gbh, SPA_GANGBLOCKSIZE);
34dc7c2f 2701
b128c09f
BB
2702 /*
2703 * Create the gang header.
2704 */
a6255b7f
DQ
2705 zio = zio_rewrite(pio, spa, txg, bp, gbh_abd, SPA_GANGBLOCKSIZE,
2706 zio_write_gang_done, NULL, pio->io_priority,
2707 ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
34dc7c2f 2708
b128c09f
BB
2709 /*
2710 * Create and nowait the gang children.
2711 */
1c27024e 2712 for (int g = 0; resid != 0; resid -= lsize, g++) {
b128c09f
BB
2713 lsize = P2ROUNDUP(resid / (SPA_GBH_NBLKPTRS - g),
2714 SPA_MINBLOCKSIZE);
2715 ASSERT(lsize >= SPA_MINBLOCKSIZE && lsize <= resid);
2716
9babb374 2717 zp.zp_checksum = gio->io_prop.zp_checksum;
b128c09f
BB
2718 zp.zp_compress = ZIO_COMPRESS_OFF;
2719 zp.zp_type = DMU_OT_NONE;
2720 zp.zp_level = 0;
428870ff 2721 zp.zp_copies = gio->io_prop.zp_copies;
03c6040b
GW
2722 zp.zp_dedup = B_FALSE;
2723 zp.zp_dedup_verify = B_FALSE;
2724 zp.zp_nopwrite = B_FALSE;
4807c0ba
TC
2725 zp.zp_encrypt = gio->io_prop.zp_encrypt;
2726 zp.zp_byteorder = gio->io_prop.zp_byteorder;
b5256303
TC
2727 bzero(zp.zp_salt, ZIO_DATA_SALT_LEN);
2728 bzero(zp.zp_iv, ZIO_DATA_IV_LEN);
2729 bzero(zp.zp_mac, ZIO_DATA_MAC_LEN);
b128c09f 2730
1c27024e 2731 zio_t *cio = zio_write(zio, spa, txg, &gbh->zg_blkptr[g],
c955398b
BL
2732 has_data ? abd_get_offset(pio->io_abd, pio->io_size -
2733 resid) : NULL, lsize, lsize, &zp,
2734 zio_write_gang_member_ready, NULL, NULL,
a6255b7f 2735 zio_write_gang_done, &gn->gn_child[g], pio->io_priority,
3dfb57a3
DB
2736 ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
2737
2738 if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
2739 ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
c955398b 2740 ASSERT(has_data);
3dfb57a3
DB
2741
2742 /*
2743 * Gang children won't throttle but we should
2744 * account for their work, so reserve an allocation
2745 * slot for them here.
2746 */
2747 VERIFY(metaslab_class_throttle_reserve(mc,
492f64e9 2748 zp.zp_copies, cio->io_allocator, cio, flags));
3dfb57a3
DB
2749 }
2750 zio_nowait(cio);
b128c09f 2751 }
34dc7c2f
BB
2752
2753 /*
b128c09f 2754 * Set pio's pipeline to just wait for zio to finish.
34dc7c2f 2755 */
b128c09f
BB
2756 pio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
2757
920dd524
ED
2758 /*
2759 * We didn't allocate this bp, so make sure it doesn't get unmarked.
2760 */
2761 pio->io_flags &= ~ZIO_FLAG_FASTWRITE;
2762
b128c09f
BB
2763 zio_nowait(zio);
2764
62840030 2765 return (pio);
34dc7c2f
BB
2766}
2767
03c6040b 2768/*
3c67d83a
TH
2769 * The zio_nop_write stage in the pipeline determines if allocating a
2770 * new bp is necessary. The nopwrite feature can handle writes in
2771 * either syncing or open context (i.e. zil writes) and as a result is
2772 * mutually exclusive with dedup.
2773 *
2774 * By leveraging a cryptographically secure checksum, such as SHA256, we
2775 * can compare the checksums of the new data and the old to determine if
2776 * allocating a new block is required. Note that our requirements for
2777 * cryptographic strength are fairly weak: there can't be any accidental
2778 * hash collisions, but we don't need to be secure against intentional
2779 * (malicious) collisions. To trigger a nopwrite, you have to be able
2780 * to write the file to begin with, and triggering an incorrect (hash
2781 * collision) nopwrite is no worse than simply writing to the file.
2782 * That said, there are no known attacks against the checksum algorithms
2783 * used for nopwrite, assuming that the salt and the checksums
2784 * themselves remain secret.
03c6040b 2785 */
62840030 2786static zio_t *
03c6040b
GW
2787zio_nop_write(zio_t *zio)
2788{
2789 blkptr_t *bp = zio->io_bp;
2790 blkptr_t *bp_orig = &zio->io_bp_orig;
2791 zio_prop_t *zp = &zio->io_prop;
2792
2793 ASSERT(BP_GET_LEVEL(bp) == 0);
2794 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REWRITE));
2795 ASSERT(zp->zp_nopwrite);
2796 ASSERT(!zp->zp_dedup);
2797 ASSERT(zio->io_bp_override == NULL);
2798 ASSERT(IO_IS_ALLOCATING(zio));
2799
2800 /*
2801 * Check to see if the original bp and the new bp have matching
2802 * characteristics (i.e. same checksum, compression algorithms, etc).
2803 * If they don't then just continue with the pipeline which will
2804 * allocate a new bp.
2805 */
2806 if (BP_IS_HOLE(bp_orig) ||
3c67d83a
TH
2807 !(zio_checksum_table[BP_GET_CHECKSUM(bp)].ci_flags &
2808 ZCHECKSUM_FLAG_NOPWRITE) ||
b5256303 2809 BP_IS_ENCRYPTED(bp) || BP_IS_ENCRYPTED(bp_orig) ||
03c6040b
GW
2810 BP_GET_CHECKSUM(bp) != BP_GET_CHECKSUM(bp_orig) ||
2811 BP_GET_COMPRESS(bp) != BP_GET_COMPRESS(bp_orig) ||
2812 BP_GET_DEDUP(bp) != BP_GET_DEDUP(bp_orig) ||
2813 zp->zp_copies != BP_GET_NDVAS(bp_orig))
62840030 2814 return (zio);
03c6040b
GW
2815
2816 /*
2817 * If the checksums match then reset the pipeline so that we
2818 * avoid allocating a new bp and issuing any I/O.
2819 */
2820 if (ZIO_CHECKSUM_EQUAL(bp->blk_cksum, bp_orig->blk_cksum)) {
3c67d83a
TH
2821 ASSERT(zio_checksum_table[zp->zp_checksum].ci_flags &
2822 ZCHECKSUM_FLAG_NOPWRITE);
03c6040b
GW
2823 ASSERT3U(BP_GET_PSIZE(bp), ==, BP_GET_PSIZE(bp_orig));
2824 ASSERT3U(BP_GET_LSIZE(bp), ==, BP_GET_LSIZE(bp_orig));
2825 ASSERT(zp->zp_compress != ZIO_COMPRESS_OFF);
2826 ASSERT(bcmp(&bp->blk_prop, &bp_orig->blk_prop,
2827 sizeof (uint64_t)) == 0);
2828
2829 *bp = *bp_orig;
2830 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
2831 zio->io_flags |= ZIO_FLAG_NOPWRITE;
2832 }
2833
62840030 2834 return (zio);
03c6040b
GW
2835}
2836
34dc7c2f
BB
2837/*
2838 * ==========================================================================
428870ff 2839 * Dedup
34dc7c2f
BB
2840 * ==========================================================================
2841 */
428870ff
BB
2842static void
2843zio_ddt_child_read_done(zio_t *zio)
2844{
2845 blkptr_t *bp = zio->io_bp;
2846 ddt_entry_t *dde = zio->io_private;
2847 ddt_phys_t *ddp;
2848 zio_t *pio = zio_unique_parent(zio);
2849
2850 mutex_enter(&pio->io_lock);
2851 ddp = ddt_phys_select(dde, bp);
2852 if (zio->io_error == 0)
2853 ddt_phys_clear(ddp); /* this ddp doesn't need repair */
a6255b7f
DQ
2854
2855 if (zio->io_error == 0 && dde->dde_repair_abd == NULL)
2856 dde->dde_repair_abd = zio->io_abd;
428870ff 2857 else
a6255b7f 2858 abd_free(zio->io_abd);
428870ff
BB
2859 mutex_exit(&pio->io_lock);
2860}
2861
62840030 2862static zio_t *
428870ff
BB
2863zio_ddt_read_start(zio_t *zio)
2864{
2865 blkptr_t *bp = zio->io_bp;
2866
2867 ASSERT(BP_GET_DEDUP(bp));
2868 ASSERT(BP_GET_PSIZE(bp) == zio->io_size);
2869 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
2870
2871 if (zio->io_child_error[ZIO_CHILD_DDT]) {
2872 ddt_t *ddt = ddt_select(zio->io_spa, bp);
2873 ddt_entry_t *dde = ddt_repair_start(ddt, bp);
2874 ddt_phys_t *ddp = dde->dde_phys;
2875 ddt_phys_t *ddp_self = ddt_phys_select(dde, bp);
2876 blkptr_t blk;
2877
2878 ASSERT(zio->io_vsd == NULL);
2879 zio->io_vsd = dde;
2880
2881 if (ddp_self == NULL)
62840030 2882 return (zio);
428870ff 2883
1c27024e 2884 for (int p = 0; p < DDT_PHYS_TYPES; p++, ddp++) {
428870ff
BB
2885 if (ddp->ddp_phys_birth == 0 || ddp == ddp_self)
2886 continue;
2887 ddt_bp_create(ddt->ddt_checksum, &dde->dde_key, ddp,
2888 &blk);
2889 zio_nowait(zio_read(zio, zio->io_spa, &blk,
a6255b7f
DQ
2890 abd_alloc_for_io(zio->io_size, B_TRUE),
2891 zio->io_size, zio_ddt_child_read_done, dde,
2892 zio->io_priority, ZIO_DDT_CHILD_FLAGS(zio) |
2893 ZIO_FLAG_DONT_PROPAGATE, &zio->io_bookmark));
428870ff 2894 }
62840030 2895 return (zio);
428870ff
BB
2896 }
2897
2898 zio_nowait(zio_read(zio, zio->io_spa, bp,
a6255b7f 2899 zio->io_abd, zio->io_size, NULL, NULL, zio->io_priority,
428870ff
BB
2900 ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark));
2901
62840030 2902 return (zio);
428870ff
BB
2903}
2904
62840030 2905static zio_t *
428870ff
BB
2906zio_ddt_read_done(zio_t *zio)
2907{
2908 blkptr_t *bp = zio->io_bp;
2909
ddc751d5 2910 if (zio_wait_for_children(zio, ZIO_CHILD_DDT_BIT, ZIO_WAIT_DONE)) {
62840030 2911 return (NULL);
ddc751d5 2912 }
428870ff
BB
2913
2914 ASSERT(BP_GET_DEDUP(bp));
2915 ASSERT(BP_GET_PSIZE(bp) == zio->io_size);
2916 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
2917
2918 if (zio->io_child_error[ZIO_CHILD_DDT]) {
2919 ddt_t *ddt = ddt_select(zio->io_spa, bp);
2920 ddt_entry_t *dde = zio->io_vsd;
2921 if (ddt == NULL) {
2922 ASSERT(spa_load_state(zio->io_spa) != SPA_LOAD_NONE);
62840030 2923 return (zio);
428870ff
BB
2924 }
2925 if (dde == NULL) {
2926 zio->io_stage = ZIO_STAGE_DDT_READ_START >> 1;
2927 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_FALSE);
62840030 2928 return (NULL);
428870ff 2929 }
a6255b7f
DQ
2930 if (dde->dde_repair_abd != NULL) {
2931 abd_copy(zio->io_abd, dde->dde_repair_abd,
2932 zio->io_size);
428870ff
BB
2933 zio->io_child_error[ZIO_CHILD_DDT] = 0;
2934 }
2935 ddt_repair_done(ddt, dde);
2936 zio->io_vsd = NULL;
2937 }
2938
2939 ASSERT(zio->io_vsd == NULL);
2940
62840030 2941 return (zio);
428870ff
BB
2942}
2943
2944static boolean_t
2945zio_ddt_collision(zio_t *zio, ddt_t *ddt, ddt_entry_t *dde)
2946{
2947 spa_t *spa = zio->io_spa;
c17bcf83 2948 boolean_t do_raw = !!(zio->io_flags & ZIO_FLAG_RAW);
428870ff 2949
c17bcf83 2950 ASSERT(!(zio->io_bp_override && do_raw));
2aa34383 2951
428870ff
BB
2952 /*
2953 * Note: we compare the original data, not the transformed data,
2954 * because when zio->io_bp is an override bp, we will not have
2955 * pushed the I/O transforms. That's an important optimization
2956 * because otherwise we'd compress/encrypt all dmu_sync() data twice.
c17bcf83 2957 * However, we should never get a raw, override zio so in these
b5256303 2958 * cases we can compare the io_abd directly. This is useful because
c17bcf83
TC
2959 * it allows us to do dedup verification even if we don't have access
2960 * to the original data (for instance, if the encryption keys aren't
2961 * loaded).
428870ff 2962 */
c17bcf83 2963
1c27024e 2964 for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++) {
428870ff
BB
2965 zio_t *lio = dde->dde_lead_zio[p];
2966
c17bcf83
TC
2967 if (lio != NULL && do_raw) {
2968 return (lio->io_size != zio->io_size ||
a6255b7f 2969 abd_cmp(zio->io_abd, lio->io_abd) != 0);
c17bcf83 2970 } else if (lio != NULL) {
428870ff 2971 return (lio->io_orig_size != zio->io_orig_size ||
a6255b7f 2972 abd_cmp(zio->io_orig_abd, lio->io_orig_abd) != 0);
428870ff
BB
2973 }
2974 }
2975
1c27024e 2976 for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++) {
428870ff
BB
2977 ddt_phys_t *ddp = &dde->dde_phys[p];
2978
c17bcf83
TC
2979 if (ddp->ddp_phys_birth != 0 && do_raw) {
2980 blkptr_t blk = *zio->io_bp;
2981 uint64_t psize;
a6255b7f 2982 abd_t *tmpabd;
c17bcf83
TC
2983 int error;
2984
2985 ddt_bp_fill(ddp, &blk, ddp->ddp_phys_birth);
2986 psize = BP_GET_PSIZE(&blk);
2987
2988 if (psize != zio->io_size)
2989 return (B_TRUE);
2990
2991 ddt_exit(ddt);
2992
a6255b7f 2993 tmpabd = abd_alloc_for_io(psize, B_TRUE);
c17bcf83 2994
a6255b7f 2995 error = zio_wait(zio_read(NULL, spa, &blk, tmpabd,
c17bcf83
TC
2996 psize, NULL, NULL, ZIO_PRIORITY_SYNC_READ,
2997 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE |
2998 ZIO_FLAG_RAW, &zio->io_bookmark));
2999
3000 if (error == 0) {
a6255b7f 3001 if (abd_cmp(tmpabd, zio->io_abd) != 0)
c17bcf83
TC
3002 error = SET_ERROR(ENOENT);
3003 }
3004
a6255b7f 3005 abd_free(tmpabd);
c17bcf83
TC
3006 ddt_enter(ddt);
3007 return (error != 0);
3008 } else if (ddp->ddp_phys_birth != 0) {
428870ff 3009 arc_buf_t *abuf = NULL;
2a432414 3010 arc_flags_t aflags = ARC_FLAG_WAIT;
428870ff
BB
3011 blkptr_t blk = *zio->io_bp;
3012 int error;
3013
3014 ddt_bp_fill(ddp, &blk, ddp->ddp_phys_birth);
3015
c17bcf83
TC
3016 if (BP_GET_LSIZE(&blk) != zio->io_orig_size)
3017 return (B_TRUE);
3018
428870ff
BB
3019 ddt_exit(ddt);
3020
294f6806 3021 error = arc_read(NULL, spa, &blk,
428870ff
BB
3022 arc_getbuf_func, &abuf, ZIO_PRIORITY_SYNC_READ,
3023 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE,
3024 &aflags, &zio->io_bookmark);
3025
3026 if (error == 0) {
a6255b7f 3027 if (abd_cmp_buf(zio->io_orig_abd, abuf->b_data,
428870ff 3028 zio->io_orig_size) != 0)
c17bcf83 3029 error = SET_ERROR(ENOENT);
d3c2ae1c 3030 arc_buf_destroy(abuf, &abuf);
428870ff
BB
3031 }
3032
3033 ddt_enter(ddt);
3034 return (error != 0);
3035 }
3036 }
3037
3038 return (B_FALSE);
3039}
3040
3041static void
3042zio_ddt_child_write_ready(zio_t *zio)
3043{
3044 int p = zio->io_prop.zp_copies;
3045 ddt_t *ddt = ddt_select(zio->io_spa, zio->io_bp);
3046 ddt_entry_t *dde = zio->io_private;
3047 ddt_phys_t *ddp = &dde->dde_phys[p];
3048 zio_t *pio;
3049
3050 if (zio->io_error)
3051 return;
3052
3053 ddt_enter(ddt);
3054
3055 ASSERT(dde->dde_lead_zio[p] == zio);
3056
3057 ddt_phys_fill(ddp, zio->io_bp);
3058
1c27024e 3059 zio_link_t *zl = NULL;
3dfb57a3 3060 while ((pio = zio_walk_parents(zio, &zl)) != NULL)
428870ff
BB
3061 ddt_bp_fill(ddp, pio->io_bp, zio->io_txg);
3062
3063 ddt_exit(ddt);
3064}
3065
3066static void
3067zio_ddt_child_write_done(zio_t *zio)
3068{
3069 int p = zio->io_prop.zp_copies;
3070 ddt_t *ddt = ddt_select(zio->io_spa, zio->io_bp);
3071 ddt_entry_t *dde = zio->io_private;
3072 ddt_phys_t *ddp = &dde->dde_phys[p];
3073
3074 ddt_enter(ddt);
3075
3076 ASSERT(ddp->ddp_refcnt == 0);
3077 ASSERT(dde->dde_lead_zio[p] == zio);
3078 dde->dde_lead_zio[p] = NULL;
3079
3080 if (zio->io_error == 0) {
3dfb57a3
DB
3081 zio_link_t *zl = NULL;
3082 while (zio_walk_parents(zio, &zl) != NULL)
428870ff
BB
3083 ddt_phys_addref(ddp);
3084 } else {
3085 ddt_phys_clear(ddp);
3086 }
3087
3088 ddt_exit(ddt);
3089}
3090
3091static void
3092zio_ddt_ditto_write_done(zio_t *zio)
3093{
3094 int p = DDT_PHYS_DITTO;
1c27024e 3095 ASSERTV(zio_prop_t *zp = &zio->io_prop);
428870ff
BB
3096 blkptr_t *bp = zio->io_bp;
3097 ddt_t *ddt = ddt_select(zio->io_spa, bp);
3098 ddt_entry_t *dde = zio->io_private;
3099 ddt_phys_t *ddp = &dde->dde_phys[p];
3100 ddt_key_t *ddk = &dde->dde_key;
3101
3102 ddt_enter(ddt);
3103
3104 ASSERT(ddp->ddp_refcnt == 0);
3105 ASSERT(dde->dde_lead_zio[p] == zio);
3106 dde->dde_lead_zio[p] = NULL;
3107
3108 if (zio->io_error == 0) {
3109 ASSERT(ZIO_CHECKSUM_EQUAL(bp->blk_cksum, ddk->ddk_cksum));
3110 ASSERT(zp->zp_copies < SPA_DVAS_PER_BP);
3111 ASSERT(zp->zp_copies == BP_GET_NDVAS(bp) - BP_IS_GANG(bp));
3112 if (ddp->ddp_phys_birth != 0)
3113 ddt_phys_free(ddt, ddk, ddp, zio->io_txg);
3114 ddt_phys_fill(ddp, bp);
3115 }
3116
3117 ddt_exit(ddt);
3118}
3119
62840030 3120static zio_t *
428870ff
BB
3121zio_ddt_write(zio_t *zio)
3122{
3123 spa_t *spa = zio->io_spa;
3124 blkptr_t *bp = zio->io_bp;
3125 uint64_t txg = zio->io_txg;
3126 zio_prop_t *zp = &zio->io_prop;
3127 int p = zp->zp_copies;
3128 int ditto_copies;
3129 zio_t *cio = NULL;
3130 zio_t *dio = NULL;
3131 ddt_t *ddt = ddt_select(spa, bp);
3132 ddt_entry_t *dde;
3133 ddt_phys_t *ddp;
3134
3135 ASSERT(BP_GET_DEDUP(bp));
3136 ASSERT(BP_GET_CHECKSUM(bp) == zp->zp_checksum);
3137 ASSERT(BP_IS_HOLE(bp) || zio->io_bp_override);
c17bcf83 3138 ASSERT(!(zio->io_bp_override && (zio->io_flags & ZIO_FLAG_RAW)));
428870ff
BB
3139
3140 ddt_enter(ddt);
3141 dde = ddt_lookup(ddt, bp, B_TRUE);
3142 ddp = &dde->dde_phys[p];
3143
3144 if (zp->zp_dedup_verify && zio_ddt_collision(zio, ddt, dde)) {
3145 /*
3146 * If we're using a weak checksum, upgrade to a strong checksum
3147 * and try again. If we're already using a strong checksum,
3148 * we can't resolve it, so just convert to an ordinary write.
3149 * (And automatically e-mail a paper to Nature?)
3150 */
3c67d83a
TH
3151 if (!(zio_checksum_table[zp->zp_checksum].ci_flags &
3152 ZCHECKSUM_FLAG_DEDUP)) {
428870ff
BB
3153 zp->zp_checksum = spa_dedup_checksum(spa);
3154 zio_pop_transforms(zio);
3155 zio->io_stage = ZIO_STAGE_OPEN;
3156 BP_ZERO(bp);
3157 } else {
03c6040b 3158 zp->zp_dedup = B_FALSE;
428870ff
BB
3159 }
3160 zio->io_pipeline = ZIO_WRITE_PIPELINE;
3161 ddt_exit(ddt);
62840030 3162 return (zio);
428870ff
BB
3163 }
3164
3165 ditto_copies = ddt_ditto_copies_needed(ddt, dde, ddp);
3166 ASSERT(ditto_copies < SPA_DVAS_PER_BP);
3167
3168 if (ditto_copies > ddt_ditto_copies_present(dde) &&
3169 dde->dde_lead_zio[DDT_PHYS_DITTO] == NULL) {
3170 zio_prop_t czp = *zp;
3171
3172 czp.zp_copies = ditto_copies;
3173
3174 /*
3175 * If we arrived here with an override bp, we won't have run
3176 * the transform stack, so we won't have the data we need to
3177 * generate a child i/o. So, toss the override bp and restart.
3178 * This is safe, because using the override bp is just an
3179 * optimization; and it's rare, so the cost doesn't matter.
3180 */
3181 if (zio->io_bp_override) {
3182 zio_pop_transforms(zio);
3183 zio->io_stage = ZIO_STAGE_OPEN;
3184 zio->io_pipeline = ZIO_WRITE_PIPELINE;
3185 zio->io_bp_override = NULL;
3186 BP_ZERO(bp);
3187 ddt_exit(ddt);
62840030 3188 return (zio);
428870ff
BB
3189 }
3190
a6255b7f 3191 dio = zio_write(zio, spa, txg, bp, zio->io_orig_abd,
2aa34383 3192 zio->io_orig_size, zio->io_orig_size, &czp, NULL, NULL,
bc77ba73 3193 NULL, zio_ddt_ditto_write_done, dde, zio->io_priority,
428870ff
BB
3194 ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark);
3195
a6255b7f 3196 zio_push_transform(dio, zio->io_abd, zio->io_size, 0, NULL);
428870ff
BB
3197 dde->dde_lead_zio[DDT_PHYS_DITTO] = dio;
3198 }
3199
3200 if (ddp->ddp_phys_birth != 0 || dde->dde_lead_zio[p] != NULL) {
3201 if (ddp->ddp_phys_birth != 0)
3202 ddt_bp_fill(ddp, bp, txg);
3203 if (dde->dde_lead_zio[p] != NULL)
3204 zio_add_child(zio, dde->dde_lead_zio[p]);
3205 else
3206 ddt_phys_addref(ddp);
3207 } else if (zio->io_bp_override) {
3208 ASSERT(bp->blk_birth == txg);
3209 ASSERT(BP_EQUAL(bp, zio->io_bp_override));
3210 ddt_phys_fill(ddp, bp);
3211 ddt_phys_addref(ddp);
3212 } else {
a6255b7f 3213 cio = zio_write(zio, spa, txg, bp, zio->io_orig_abd,
2aa34383 3214 zio->io_orig_size, zio->io_orig_size, zp,
bc77ba73 3215 zio_ddt_child_write_ready, NULL, NULL,
428870ff
BB
3216 zio_ddt_child_write_done, dde, zio->io_priority,
3217 ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark);
3218
a6255b7f 3219 zio_push_transform(cio, zio->io_abd, zio->io_size, 0, NULL);
428870ff
BB
3220 dde->dde_lead_zio[p] = cio;
3221 }
3222
3223 ddt_exit(ddt);
3224
3225 if (cio)
3226 zio_nowait(cio);
3227 if (dio)
3228 zio_nowait(dio);
3229
62840030 3230 return (zio);
428870ff
BB
3231}
3232
3233ddt_entry_t *freedde; /* for debugging */
b128c09f 3234
62840030 3235static zio_t *
428870ff
BB
3236zio_ddt_free(zio_t *zio)
3237{
3238 spa_t *spa = zio->io_spa;
3239 blkptr_t *bp = zio->io_bp;
3240 ddt_t *ddt = ddt_select(spa, bp);
3241 ddt_entry_t *dde;
3242 ddt_phys_t *ddp;
3243
3244 ASSERT(BP_GET_DEDUP(bp));
3245 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
3246
3247 ddt_enter(ddt);
3248 freedde = dde = ddt_lookup(ddt, bp, B_TRUE);
5dc6af0e
BB
3249 if (dde) {
3250 ddp = ddt_phys_select(dde, bp);
3251 if (ddp)
3252 ddt_phys_decref(ddp);
3253 }
428870ff
BB
3254 ddt_exit(ddt);
3255
62840030 3256 return (zio);
428870ff
BB
3257}
3258
3259/*
3260 * ==========================================================================
3261 * Allocate and free blocks
3262 * ==========================================================================
3263 */
3dfb57a3
DB
3264
3265static zio_t *
492f64e9 3266zio_io_to_allocate(spa_t *spa, int allocator)
3dfb57a3
DB
3267{
3268 zio_t *zio;
3269
492f64e9 3270 ASSERT(MUTEX_HELD(&spa->spa_alloc_locks[allocator]));
3dfb57a3 3271
492f64e9 3272 zio = avl_first(&spa->spa_alloc_trees[allocator]);
3dfb57a3
DB
3273 if (zio == NULL)
3274 return (NULL);
3275
3276 ASSERT(IO_IS_ALLOCATING(zio));
3277
3278 /*
3279 * Try to place a reservation for this zio. If we're unable to
3280 * reserve then we throttle.
3281 */
492f64e9 3282 ASSERT3U(zio->io_allocator, ==, allocator);
cc99f275 3283 if (!metaslab_class_throttle_reserve(zio->io_metaslab_class,
492f64e9 3284 zio->io_prop.zp_copies, zio->io_allocator, zio, 0)) {
3dfb57a3
DB
3285 return (NULL);
3286 }
3287
492f64e9 3288 avl_remove(&spa->spa_alloc_trees[allocator], zio);
3dfb57a3
DB
3289 ASSERT3U(zio->io_stage, <, ZIO_STAGE_DVA_ALLOCATE);
3290
3291 return (zio);
3292}
3293
62840030 3294static zio_t *
3dfb57a3
DB
3295zio_dva_throttle(zio_t *zio)
3296{
3297 spa_t *spa = zio->io_spa;
3298 zio_t *nio;
cc99f275
DB
3299 metaslab_class_t *mc;
3300
3301 /* locate an appropriate allocation class */
3302 mc = spa_preferred_class(spa, zio->io_size, zio->io_prop.zp_type,
3303 zio->io_prop.zp_level, zio->io_prop.zp_zpl_smallblk);
3dfb57a3
DB
3304
3305 if (zio->io_priority == ZIO_PRIORITY_SYNC_WRITE ||
cc99f275 3306 !mc->mc_alloc_throttle_enabled ||
3dfb57a3
DB
3307 zio->io_child_type == ZIO_CHILD_GANG ||
3308 zio->io_flags & ZIO_FLAG_NODATA) {
62840030 3309 return (zio);
3dfb57a3
DB
3310 }
3311
3312 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
3313
3314 ASSERT3U(zio->io_queued_timestamp, >, 0);
3315 ASSERT(zio->io_stage == ZIO_STAGE_DVA_THROTTLE);
3316
492f64e9
PD
3317 zbookmark_phys_t *bm = &zio->io_bookmark;
3318 /*
3319 * We want to try to use as many allocators as possible to help improve
3320 * performance, but we also want logically adjacent IOs to be physically
3321 * adjacent to improve sequential read performance. We chunk each object
3322 * into 2^20 block regions, and then hash based on the objset, object,
3323 * level, and region to accomplish both of these goals.
3324 */
3325 zio->io_allocator = cityhash4(bm->zb_objset, bm->zb_object,
3326 bm->zb_level, bm->zb_blkid >> 20) % spa->spa_alloc_count;
3327 mutex_enter(&spa->spa_alloc_locks[zio->io_allocator]);
3dfb57a3 3328 ASSERT(zio->io_type == ZIO_TYPE_WRITE);
cc99f275 3329 zio->io_metaslab_class = mc;
492f64e9 3330 avl_add(&spa->spa_alloc_trees[zio->io_allocator], zio);
cc99f275 3331 nio = zio_io_to_allocate(spa, zio->io_allocator);
492f64e9 3332 mutex_exit(&spa->spa_alloc_locks[zio->io_allocator]);
62840030 3333 return (nio);
3dfb57a3
DB
3334}
3335
cc99f275 3336static void
492f64e9 3337zio_allocate_dispatch(spa_t *spa, int allocator)
3dfb57a3
DB
3338{
3339 zio_t *zio;
3340
492f64e9
PD
3341 mutex_enter(&spa->spa_alloc_locks[allocator]);
3342 zio = zio_io_to_allocate(spa, allocator);
3343 mutex_exit(&spa->spa_alloc_locks[allocator]);
3dfb57a3
DB
3344 if (zio == NULL)
3345 return;
3346
3347 ASSERT3U(zio->io_stage, ==, ZIO_STAGE_DVA_THROTTLE);
3348 ASSERT0(zio->io_error);
3349 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_TRUE);
3350}
3351
62840030 3352static zio_t *
34dc7c2f
BB
3353zio_dva_allocate(zio_t *zio)
3354{
3355 spa_t *spa = zio->io_spa;
cc99f275 3356 metaslab_class_t *mc;
34dc7c2f
BB
3357 blkptr_t *bp = zio->io_bp;
3358 int error;
6d974228 3359 int flags = 0;
34dc7c2f 3360
9babb374
BB
3361 if (zio->io_gang_leader == NULL) {
3362 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
3363 zio->io_gang_leader = zio;
3364 }
3365
34dc7c2f 3366 ASSERT(BP_IS_HOLE(bp));
c99c9001 3367 ASSERT0(BP_GET_NDVAS(bp));
428870ff
BB
3368 ASSERT3U(zio->io_prop.zp_copies, >, 0);
3369 ASSERT3U(zio->io_prop.zp_copies, <=, spa_max_replication(spa));
34dc7c2f
BB
3370 ASSERT3U(zio->io_size, ==, BP_GET_PSIZE(bp));
3371
920dd524 3372 flags |= (zio->io_flags & ZIO_FLAG_FASTWRITE) ? METASLAB_FASTWRITE : 0;
3dfb57a3
DB
3373 if (zio->io_flags & ZIO_FLAG_NODATA)
3374 flags |= METASLAB_DONT_THROTTLE;
3375 if (zio->io_flags & ZIO_FLAG_GANG_CHILD)
3376 flags |= METASLAB_GANG_CHILD;
3377 if (zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE)
3378 flags |= METASLAB_ASYNC_ALLOC;
3379
cc99f275
DB
3380 /*
3381 * if not already chosen, locate an appropriate allocation class
3382 */
3383 mc = zio->io_metaslab_class;
3384 if (mc == NULL) {
3385 mc = spa_preferred_class(spa, zio->io_size,
3386 zio->io_prop.zp_type, zio->io_prop.zp_level,
3387 zio->io_prop.zp_zpl_smallblk);
3388 zio->io_metaslab_class = mc;
3389 }
3390
b128c09f 3391 error = metaslab_alloc(spa, mc, zio->io_size, bp,
4e21fd06 3392 zio->io_prop.zp_copies, zio->io_txg, NULL, flags,
492f64e9 3393 &zio->io_alloc_list, zio, zio->io_allocator);
34dc7c2f 3394
cc99f275
DB
3395 /*
3396 * Fallback to normal class when an alloc class is full
3397 */
3398 if (error == ENOSPC && mc != spa_normal_class(spa)) {
3399 /*
3400 * If throttling, transfer reservation over to normal class.
3401 * The io_allocator slot can remain the same even though we
3402 * are switching classes.
3403 */
3404 if (mc->mc_alloc_throttle_enabled &&
3405 (zio->io_flags & ZIO_FLAG_IO_ALLOCATING)) {
3406 metaslab_class_throttle_unreserve(mc,
3407 zio->io_prop.zp_copies, zio->io_allocator, zio);
3408 zio->io_flags &= ~ZIO_FLAG_IO_ALLOCATING;
3409
3410 mc = spa_normal_class(spa);
3411 VERIFY(metaslab_class_throttle_reserve(mc,
3412 zio->io_prop.zp_copies, zio->io_allocator, zio,
3413 flags | METASLAB_MUST_RESERVE));
3414 } else {
3415 mc = spa_normal_class(spa);
3416 }
3417 zio->io_metaslab_class = mc;
3418
3419 error = metaslab_alloc(spa, mc, zio->io_size, bp,
3420 zio->io_prop.zp_copies, zio->io_txg, NULL, flags,
3421 &zio->io_alloc_list, zio, zio->io_allocator);
3422 }
3423
3dfb57a3 3424 if (error != 0) {
964c2d69 3425 zfs_dbgmsg("%s: metaslab allocation failure: zio %p, "
6d974228
GW
3426 "size %llu, error %d", spa_name(spa), zio, zio->io_size,
3427 error);
b128c09f
BB
3428 if (error == ENOSPC && zio->io_size > SPA_MINBLOCKSIZE)
3429 return (zio_write_gang_block(zio));
34dc7c2f
BB
3430 zio->io_error = error;
3431 }
3432
62840030 3433 return (zio);
34dc7c2f
BB
3434}
3435
62840030 3436static zio_t *
34dc7c2f
BB
3437zio_dva_free(zio_t *zio)
3438{
b128c09f 3439 metaslab_free(zio->io_spa, zio->io_bp, zio->io_txg, B_FALSE);
34dc7c2f 3440
62840030 3441 return (zio);
34dc7c2f
BB
3442}
3443
62840030 3444static zio_t *
34dc7c2f
BB
3445zio_dva_claim(zio_t *zio)
3446{
b128c09f
BB
3447 int error;
3448
3449 error = metaslab_claim(zio->io_spa, zio->io_bp, zio->io_txg);
3450 if (error)
3451 zio->io_error = error;
34dc7c2f 3452
62840030 3453 return (zio);
34dc7c2f
BB
3454}
3455
b128c09f
BB
3456/*
3457 * Undo an allocation. This is used by zio_done() when an I/O fails
3458 * and we want to give back the block we just allocated.
3459 * This handles both normal blocks and gang blocks.
3460 */
3461static void
3462zio_dva_unallocate(zio_t *zio, zio_gang_node_t *gn, blkptr_t *bp)
3463{
b128c09f 3464 ASSERT(bp->blk_birth == zio->io_txg || BP_IS_HOLE(bp));
428870ff 3465 ASSERT(zio->io_bp_override == NULL);
b128c09f
BB
3466
3467 if (!BP_IS_HOLE(bp))
428870ff 3468 metaslab_free(zio->io_spa, bp, bp->blk_birth, B_TRUE);
b128c09f
BB
3469
3470 if (gn != NULL) {
1c27024e 3471 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
b128c09f
BB
3472 zio_dva_unallocate(zio, gn->gn_child[g],
3473 &gn->gn_gbh->zg_blkptr[g]);
3474 }
3475 }
3476}
3477
3478/*
3479 * Try to allocate an intent log block. Return 0 on success, errno on failure.
3480 */
3481int
b5256303
TC
3482zio_alloc_zil(spa_t *spa, objset_t *os, uint64_t txg, blkptr_t *new_bp,
3483 uint64_t size, boolean_t *slog)
b128c09f 3484{
428870ff 3485 int error = 1;
4e21fd06 3486 zio_alloc_list_t io_alloc_list;
b128c09f 3487
428870ff
BB
3488 ASSERT(txg > spa_syncing_txg(spa));
3489
4e21fd06 3490 metaslab_trace_init(&io_alloc_list);
cc99f275
DB
3491
3492 /*
3493 * Block pointer fields are useful to metaslabs for stats and debugging.
3494 * Fill in the obvious ones before calling into metaslab_alloc().
3495 */
3496 BP_SET_TYPE(new_bp, DMU_OT_INTENT_LOG);
3497 BP_SET_PSIZE(new_bp, size);
3498 BP_SET_LEVEL(new_bp, 0);
3499
492f64e9
PD
3500 /*
3501 * When allocating a zil block, we don't have information about
3502 * the final destination of the block except the objset it's part
3503 * of, so we just hash the objset ID to pick the allocator to get
3504 * some parallelism.
3505 */
1b7c1e5c 3506 error = metaslab_alloc(spa, spa_log_class(spa), size, new_bp, 1,
492f64e9
PD
3507 txg, NULL, METASLAB_FASTWRITE, &io_alloc_list, NULL,
3508 cityhash4(0, 0, 0, os->os_dsl_dataset->ds_object) %
3509 spa->spa_alloc_count);
1b7c1e5c
GDN
3510 if (error == 0) {
3511 *slog = TRUE;
3512 } else {
428870ff 3513 error = metaslab_alloc(spa, spa_normal_class(spa), size,
4e21fd06 3514 new_bp, 1, txg, NULL, METASLAB_FASTWRITE,
492f64e9
PD
3515 &io_alloc_list, NULL, cityhash4(0, 0, 0,
3516 os->os_dsl_dataset->ds_object) % spa->spa_alloc_count);
1b7c1e5c
GDN
3517 if (error == 0)
3518 *slog = FALSE;
ebf8e3a2 3519 }
4e21fd06 3520 metaslab_trace_fini(&io_alloc_list);
b128c09f
BB
3521
3522 if (error == 0) {
3523 BP_SET_LSIZE(new_bp, size);
3524 BP_SET_PSIZE(new_bp, size);
3525 BP_SET_COMPRESS(new_bp, ZIO_COMPRESS_OFF);
428870ff
BB
3526 BP_SET_CHECKSUM(new_bp,
3527 spa_version(spa) >= SPA_VERSION_SLIM_ZIL
3528 ? ZIO_CHECKSUM_ZILOG2 : ZIO_CHECKSUM_ZILOG);
b128c09f
BB
3529 BP_SET_TYPE(new_bp, DMU_OT_INTENT_LOG);
3530 BP_SET_LEVEL(new_bp, 0);
428870ff 3531 BP_SET_DEDUP(new_bp, 0);
b128c09f 3532 BP_SET_BYTEORDER(new_bp, ZFS_HOST_BYTEORDER);
b5256303
TC
3533
3534 /*
3535 * encrypted blocks will require an IV and salt. We generate
3536 * these now since we will not be rewriting the bp at
3537 * rewrite time.
3538 */
3539 if (os->os_encrypted) {
3540 uint8_t iv[ZIO_DATA_IV_LEN];
3541 uint8_t salt[ZIO_DATA_SALT_LEN];
3542
3543 BP_SET_CRYPT(new_bp, B_TRUE);
3544 VERIFY0(spa_crypt_get_salt(spa,
3545 dmu_objset_id(os), salt));
3546 VERIFY0(zio_crypt_generate_iv(iv));
3547
3548 zio_crypt_encode_params_bp(new_bp, salt, iv);
3549 }
1ce23dca
PS
3550 } else {
3551 zfs_dbgmsg("%s: zil block allocation failure: "
3552 "size %llu, error %d", spa_name(spa), size, error);
b128c09f
BB
3553 }
3554
3555 return (error);
3556}
3557
34dc7c2f
BB
3558/*
3559 * ==========================================================================
3560 * Read and write to physical devices
3561 * ==========================================================================
3562 */
98b25418
GW
3563
3564
3565/*
3566 * Issue an I/O to the underlying vdev. Typically the issue pipeline
3567 * stops after this stage and will resume upon I/O completion.
3568 * However, there are instances where the vdev layer may need to
3569 * continue the pipeline when an I/O was not issued. Since the I/O
3570 * that was sent to the vdev layer might be different than the one
3571 * currently active in the pipeline (see vdev_queue_io()), we explicitly
3572 * force the underlying vdev layers to call either zio_execute() or
3573 * zio_interrupt() to ensure that the pipeline continues with the correct I/O.
3574 */
62840030 3575static zio_t *
34dc7c2f
BB
3576zio_vdev_io_start(zio_t *zio)
3577{
3578 vdev_t *vd = zio->io_vd;
34dc7c2f
BB
3579 uint64_t align;
3580 spa_t *spa = zio->io_spa;
3581
193a37cb
TH
3582 zio->io_delay = 0;
3583
b128c09f
BB
3584 ASSERT(zio->io_error == 0);
3585 ASSERT(zio->io_child_error[ZIO_CHILD_VDEV] == 0);
34dc7c2f 3586
b128c09f
BB
3587 if (vd == NULL) {
3588 if (!(zio->io_flags & ZIO_FLAG_CONFIG_WRITER))
3589 spa_config_enter(spa, SCL_ZIO, zio, RW_READER);
34dc7c2f 3590
b128c09f
BB
3591 /*
3592 * The mirror_ops handle multiple DVAs in a single BP.
3593 */
98b25418 3594 vdev_mirror_ops.vdev_op_io_start(zio);
62840030 3595 return (NULL);
34dc7c2f
BB
3596 }
3597
3dfb57a3 3598 ASSERT3P(zio->io_logical, !=, zio);
6cb8e530
PZ
3599 if (zio->io_type == ZIO_TYPE_WRITE) {
3600 ASSERT(spa->spa_trust_config);
3601
a1d477c2
MA
3602 /*
3603 * Note: the code can handle other kinds of writes,
3604 * but we don't expect them.
3605 */
6cb8e530
PZ
3606 if (zio->io_vd->vdev_removing) {
3607 ASSERT(zio->io_flags &
3608 (ZIO_FLAG_PHYSICAL | ZIO_FLAG_SELF_HEAL |
3609 ZIO_FLAG_RESILVER | ZIO_FLAG_INDUCE_DAMAGE));
3610 }
a1d477c2 3611 }
3dfb57a3 3612
b128c09f
BB
3613 align = 1ULL << vd->vdev_top->vdev_ashift;
3614
b02fe35d
AR
3615 if (!(zio->io_flags & ZIO_FLAG_PHYSICAL) &&
3616 P2PHASE(zio->io_size, align) != 0) {
3617 /* Transform logical writes to be a full physical block size. */
34dc7c2f 3618 uint64_t asize = P2ROUNDUP(zio->io_size, align);
a6255b7f 3619 abd_t *abuf = abd_alloc_sametype(zio->io_abd, asize);
178e73b3 3620 ASSERT(vd == vd->vdev_top);
34dc7c2f 3621 if (zio->io_type == ZIO_TYPE_WRITE) {
a6255b7f
DQ
3622 abd_copy(abuf, zio->io_abd, zio->io_size);
3623 abd_zero_off(abuf, zio->io_size, asize - zio->io_size);
34dc7c2f 3624 }
b128c09f 3625 zio_push_transform(zio, abuf, asize, asize, zio_subblock);
34dc7c2f
BB
3626 }
3627
b02fe35d
AR
3628 /*
3629 * If this is not a physical io, make sure that it is properly aligned
3630 * before proceeding.
3631 */
3632 if (!(zio->io_flags & ZIO_FLAG_PHYSICAL)) {
3633 ASSERT0(P2PHASE(zio->io_offset, align));
3634 ASSERT0(P2PHASE(zio->io_size, align));
3635 } else {
3636 /*
3637 * For physical writes, we allow 512b aligned writes and assume
3638 * the device will perform a read-modify-write as necessary.
3639 */
3640 ASSERT0(P2PHASE(zio->io_offset, SPA_MINBLOCKSIZE));
3641 ASSERT0(P2PHASE(zio->io_size, SPA_MINBLOCKSIZE));
3642 }
3643
572e2857 3644 VERIFY(zio->io_type != ZIO_TYPE_WRITE || spa_writeable(spa));
fb5f0bc8
BB
3645
3646 /*
3647 * If this is a repair I/O, and there's no self-healing involved --
3648 * that is, we're just resilvering what we expect to resilver --
3649 * then don't do the I/O unless zio's txg is actually in vd's DTL.
9e052db4
MA
3650 * This prevents spurious resilvering.
3651 *
3652 * There are a few ways that we can end up creating these spurious
3653 * resilver i/os:
3654 *
3655 * 1. A resilver i/o will be issued if any DVA in the BP has a
3656 * dirty DTL. The mirror code will issue resilver writes to
3657 * each DVA, including the one(s) that are not on vdevs with dirty
3658 * DTLs.
3659 *
3660 * 2. With nested replication, which happens when we have a
3661 * "replacing" or "spare" vdev that's a child of a mirror or raidz.
3662 * For example, given mirror(replacing(A+B), C), it's likely that
3663 * only A is out of date (it's the new device). In this case, we'll
3664 * read from C, then use the data to resilver A+B -- but we don't
3665 * actually want to resilver B, just A. The top-level mirror has no
3666 * way to know this, so instead we just discard unnecessary repairs
3667 * as we work our way down the vdev tree.
3668 *
3669 * 3. ZTEST also creates mirrors of mirrors, mirrors of raidz, etc.
3670 * The same logic applies to any form of nested replication: ditto
3671 * + mirror, RAID-Z + replacing, etc.
3672 *
3673 * However, indirect vdevs point off to other vdevs which may have
3674 * DTL's, so we never bypass them. The child i/os on concrete vdevs
3675 * will be properly bypassed instead.
fb5f0bc8
BB
3676 */
3677 if ((zio->io_flags & ZIO_FLAG_IO_REPAIR) &&
3678 !(zio->io_flags & ZIO_FLAG_SELF_HEAL) &&
3679 zio->io_txg != 0 && /* not a delegated i/o */
9e052db4 3680 vd->vdev_ops != &vdev_indirect_ops &&
fb5f0bc8
BB
3681 !vdev_dtl_contains(vd, DTL_PARTIAL, zio->io_txg, 1)) {
3682 ASSERT(zio->io_type == ZIO_TYPE_WRITE);
fb5f0bc8 3683 zio_vdev_io_bypass(zio);
62840030 3684 return (zio);
fb5f0bc8 3685 }
34dc7c2f 3686
b128c09f
BB
3687 if (vd->vdev_ops->vdev_op_leaf &&
3688 (zio->io_type == ZIO_TYPE_READ || zio->io_type == ZIO_TYPE_WRITE)) {
3689
b0bc7a84 3690 if (zio->io_type == ZIO_TYPE_READ && vdev_cache_read(zio))
62840030 3691 return (zio);
b128c09f
BB
3692
3693 if ((zio = vdev_queue_io(zio)) == NULL)
62840030 3694 return (NULL);
b128c09f
BB
3695
3696 if (!vdev_accessible(vd, zio)) {
2e528b49 3697 zio->io_error = SET_ERROR(ENXIO);
b128c09f 3698 zio_interrupt(zio);
62840030 3699 return (NULL);
b128c09f 3700 }
67103816 3701 zio->io_delay = gethrtime();
b128c09f
BB
3702 }
3703
98b25418 3704 vd->vdev_ops->vdev_op_io_start(zio);
62840030 3705 return (NULL);
34dc7c2f
BB
3706}
3707
62840030 3708static zio_t *
34dc7c2f
BB
3709zio_vdev_io_done(zio_t *zio)
3710{
b128c09f
BB
3711 vdev_t *vd = zio->io_vd;
3712 vdev_ops_t *ops = vd ? vd->vdev_ops : &vdev_mirror_ops;
3713 boolean_t unexpected_error = B_FALSE;
34dc7c2f 3714
ddc751d5 3715 if (zio_wait_for_children(zio, ZIO_CHILD_VDEV_BIT, ZIO_WAIT_DONE)) {
62840030 3716 return (NULL);
ddc751d5 3717 }
34dc7c2f 3718
b128c09f
BB
3719 ASSERT(zio->io_type == ZIO_TYPE_READ || zio->io_type == ZIO_TYPE_WRITE);
3720
193a37cb
TH
3721 if (zio->io_delay)
3722 zio->io_delay = gethrtime() - zio->io_delay;
3723
b128c09f
BB
3724 if (vd != NULL && vd->vdev_ops->vdev_op_leaf) {
3725
3726 vdev_queue_io_done(zio);
3727
3728 if (zio->io_type == ZIO_TYPE_WRITE)
3729 vdev_cache_write(zio);
3730
3731 if (zio_injection_enabled && zio->io_error == 0)
d977122d
DB
3732 zio->io_error = zio_handle_device_injections(vd, zio,
3733 EIO, EILSEQ);
b128c09f
BB
3734
3735 if (zio_injection_enabled && zio->io_error == 0)
3736 zio->io_error = zio_handle_label_injection(zio, EIO);
3737
3738 if (zio->io_error) {
3739 if (!vdev_accessible(vd, zio)) {
2e528b49 3740 zio->io_error = SET_ERROR(ENXIO);
b128c09f
BB
3741 } else {
3742 unexpected_error = B_TRUE;
3743 }
3744 }
3745 }
3746
3747 ops->vdev_op_io_done(zio);
34dc7c2f 3748
f43615d0 3749 if (unexpected_error)
d164b209 3750 VERIFY(vdev_probe(vd, zio) == NULL);
34dc7c2f 3751
62840030 3752 return (zio);
34dc7c2f
BB
3753}
3754
a8b2e306
TC
3755/*
3756 * This function is used to change the priority of an existing zio that is
3757 * currently in-flight. This is used by the arc to upgrade priority in the
3758 * event that a demand read is made for a block that is currently queued
3759 * as a scrub or async read IO. Otherwise, the high priority read request
3760 * would end up having to wait for the lower priority IO.
3761 */
3762void
3763zio_change_priority(zio_t *pio, zio_priority_t priority)
3764{
3765 zio_t *cio, *cio_next;
3766 zio_link_t *zl = NULL;
3767
3768 ASSERT3U(priority, <, ZIO_PRIORITY_NUM_QUEUEABLE);
3769
3770 if (pio->io_vd != NULL && pio->io_vd->vdev_ops->vdev_op_leaf) {
3771 vdev_queue_change_io_priority(pio, priority);
3772 } else {
3773 pio->io_priority = priority;
3774 }
3775
3776 mutex_enter(&pio->io_lock);
3777 for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
3778 cio_next = zio_walk_children(pio, &zl);
3779 zio_change_priority(cio, priority);
3780 }
3781 mutex_exit(&pio->io_lock);
3782}
3783
428870ff
BB
3784/*
3785 * For non-raidz ZIOs, we can just copy aside the bad data read from the
3786 * disk, and use that to finish the checksum ereport later.
3787 */
3788static void
3789zio_vsd_default_cksum_finish(zio_cksum_report_t *zcr,
84c07ada 3790 const abd_t *good_buf)
428870ff
BB
3791{
3792 /* no processing needed */
3793 zfs_ereport_finish_checksum(zcr, good_buf, zcr->zcr_cbdata, B_FALSE);
3794}
3795
3796/*ARGSUSED*/
3797void
3798zio_vsd_default_cksum_report(zio_t *zio, zio_cksum_report_t *zcr, void *ignored)
3799{
84c07ada 3800 void *abd = abd_alloc_sametype(zio->io_abd, zio->io_size);
428870ff 3801
84c07ada 3802 abd_copy(abd, zio->io_abd, zio->io_size);
428870ff
BB
3803
3804 zcr->zcr_cbinfo = zio->io_size;
84c07ada 3805 zcr->zcr_cbdata = abd;
428870ff 3806 zcr->zcr_finish = zio_vsd_default_cksum_finish;
84c07ada 3807 zcr->zcr_free = zio_abd_free;
428870ff
BB
3808}
3809
62840030 3810static zio_t *
34dc7c2f
BB
3811zio_vdev_io_assess(zio_t *zio)
3812{
3813 vdev_t *vd = zio->io_vd;
b128c09f 3814
ddc751d5 3815 if (zio_wait_for_children(zio, ZIO_CHILD_VDEV_BIT, ZIO_WAIT_DONE)) {
62840030 3816 return (NULL);
ddc751d5 3817 }
b128c09f
BB
3818
3819 if (vd == NULL && !(zio->io_flags & ZIO_FLAG_CONFIG_WRITER))
3820 spa_config_exit(zio->io_spa, SCL_ZIO, zio);
3821
3822 if (zio->io_vsd != NULL) {
428870ff 3823 zio->io_vsd_ops->vsd_free(zio);
b128c09f 3824 zio->io_vsd = NULL;
34dc7c2f
BB
3825 }
3826
b128c09f 3827 if (zio_injection_enabled && zio->io_error == 0)
34dc7c2f
BB
3828 zio->io_error = zio_handle_fault_injection(zio, EIO);
3829
3830 /*
3831 * If the I/O failed, determine whether we should attempt to retry it.
428870ff
BB
3832 *
3833 * On retry, we cut in line in the issue queue, since we don't want
3834 * compression/checksumming/etc. work to prevent our (cheap) IO reissue.
34dc7c2f 3835 */
b128c09f
BB
3836 if (zio->io_error && vd == NULL &&
3837 !(zio->io_flags & (ZIO_FLAG_DONT_RETRY | ZIO_FLAG_IO_RETRY))) {
3838 ASSERT(!(zio->io_flags & ZIO_FLAG_DONT_QUEUE)); /* not a leaf */
3839 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_BYPASS)); /* not a leaf */
34dc7c2f 3840 zio->io_error = 0;
b128c09f
BB
3841 zio->io_flags |= ZIO_FLAG_IO_RETRY |
3842 ZIO_FLAG_DONT_CACHE | ZIO_FLAG_DONT_AGGREGATE;
428870ff
BB
3843 zio->io_stage = ZIO_STAGE_VDEV_IO_START >> 1;
3844 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE,
3845 zio_requeue_io_start_cut_in_line);
62840030 3846 return (NULL);
34dc7c2f
BB
3847 }
3848
b128c09f
BB
3849 /*
3850 * If we got an error on a leaf device, convert it to ENXIO
3851 * if the device is not accessible at all.
3852 */
3853 if (zio->io_error && vd != NULL && vd->vdev_ops->vdev_op_leaf &&
3854 !vdev_accessible(vd, zio))
2e528b49 3855 zio->io_error = SET_ERROR(ENXIO);
b128c09f
BB
3856
3857 /*
3858 * If we can't write to an interior vdev (mirror or RAID-Z),
3859 * set vdev_cant_write so that we stop trying to allocate from it.
3860 */
3861 if (zio->io_error == ENXIO && zio->io_type == ZIO_TYPE_WRITE &&
13fe0198 3862 vd != NULL && !vd->vdev_ops->vdev_op_leaf) {
b128c09f 3863 vd->vdev_cant_write = B_TRUE;
13fe0198 3864 }
b128c09f 3865
298ec40b
GM
3866 /*
3867 * If a cache flush returns ENOTSUP or ENOTTY, we know that no future
3868 * attempts will ever succeed. In this case we set a persistent bit so
3869 * that we don't bother with it in the future.
3870 */
3871 if ((zio->io_error == ENOTSUP || zio->io_error == ENOTTY) &&
3872 zio->io_type == ZIO_TYPE_IOCTL &&
3873 zio->io_cmd == DKIOCFLUSHWRITECACHE && vd != NULL)
3874 vd->vdev_nowritecache = B_TRUE;
3875
b128c09f
BB
3876 if (zio->io_error)
3877 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
3878
e8b96c60
MA
3879 if (vd != NULL && vd->vdev_ops->vdev_op_leaf &&
3880 zio->io_physdone != NULL) {
3881 ASSERT(!(zio->io_flags & ZIO_FLAG_DELEGATED));
3882 ASSERT(zio->io_child_type == ZIO_CHILD_VDEV);
3883 zio->io_physdone(zio->io_logical);
3884 }
3885
62840030 3886 return (zio);
34dc7c2f
BB
3887}
3888
3889void
3890zio_vdev_io_reissue(zio_t *zio)
3891{
3892 ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_START);
3893 ASSERT(zio->io_error == 0);
3894
428870ff 3895 zio->io_stage >>= 1;
34dc7c2f
BB
3896}
3897
3898void
3899zio_vdev_io_redone(zio_t *zio)
3900{
3901 ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_DONE);
3902
428870ff 3903 zio->io_stage >>= 1;
34dc7c2f
BB
3904}
3905
3906void
3907zio_vdev_io_bypass(zio_t *zio)
3908{
3909 ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_START);
3910 ASSERT(zio->io_error == 0);
3911
3912 zio->io_flags |= ZIO_FLAG_IO_BYPASS;
428870ff 3913 zio->io_stage = ZIO_STAGE_VDEV_IO_ASSESS >> 1;
34dc7c2f
BB
3914}
3915
b5256303
TC
3916/*
3917 * ==========================================================================
3918 * Encrypt and store encryption parameters
3919 * ==========================================================================
3920 */
3921
3922
3923/*
3924 * This function is used for ZIO_STAGE_ENCRYPT. It is responsible for
3925 * managing the storage of encryption parameters and passing them to the
3926 * lower-level encryption functions.
3927 */
62840030 3928static zio_t *
b5256303
TC
3929zio_encrypt(zio_t *zio)
3930{
3931 zio_prop_t *zp = &zio->io_prop;
3932 spa_t *spa = zio->io_spa;
3933 blkptr_t *bp = zio->io_bp;
3934 uint64_t psize = BP_GET_PSIZE(bp);
ae76f45c 3935 uint64_t dsobj = zio->io_bookmark.zb_objset;
b5256303
TC
3936 dmu_object_type_t ot = BP_GET_TYPE(bp);
3937 void *enc_buf = NULL;
3938 abd_t *eabd = NULL;
3939 uint8_t salt[ZIO_DATA_SALT_LEN];
3940 uint8_t iv[ZIO_DATA_IV_LEN];
3941 uint8_t mac[ZIO_DATA_MAC_LEN];
3942 boolean_t no_crypt = B_FALSE;
3943
3944 /* the root zio already encrypted the data */
3945 if (zio->io_child_type == ZIO_CHILD_GANG)
62840030 3946 return (zio);
b5256303
TC
3947
3948 /* only ZIL blocks are re-encrypted on rewrite */
3949 if (!IO_IS_ALLOCATING(zio) && ot != DMU_OT_INTENT_LOG)
62840030 3950 return (zio);
b5256303
TC
3951
3952 if (!(zp->zp_encrypt || BP_IS_ENCRYPTED(bp))) {
3953 BP_SET_CRYPT(bp, B_FALSE);
62840030 3954 return (zio);
b5256303
TC
3955 }
3956
3957 /* if we are doing raw encryption set the provided encryption params */
3958 if (zio->io_flags & ZIO_FLAG_RAW_ENCRYPT) {
ae76f45c 3959 ASSERT0(BP_GET_LEVEL(bp));
b5256303
TC
3960 BP_SET_CRYPT(bp, B_TRUE);
3961 BP_SET_BYTEORDER(bp, zp->zp_byteorder);
3962 if (ot != DMU_OT_OBJSET)
3963 zio_crypt_encode_mac_bp(bp, zp->zp_mac);
ae76f45c
TC
3964
3965 /* dnode blocks must be written out in the provided byteorder */
3966 if (zp->zp_byteorder != ZFS_HOST_BYTEORDER &&
3967 ot == DMU_OT_DNODE) {
3968 void *bswap_buf = zio_buf_alloc(psize);
3969 abd_t *babd = abd_get_from_buf(bswap_buf, psize);
3970
3971 ASSERT3U(BP_GET_COMPRESS(bp), ==, ZIO_COMPRESS_OFF);
3972 abd_copy_to_buf(bswap_buf, zio->io_abd, psize);
3973 dmu_ot_byteswap[DMU_OT_BYTESWAP(ot)].ob_func(bswap_buf,
3974 psize);
3975
3976 abd_take_ownership_of_buf(babd, B_TRUE);
3977 zio_push_transform(zio, babd, psize, psize, NULL);
3978 }
3979
b5256303
TC
3980 if (DMU_OT_IS_ENCRYPTED(ot))
3981 zio_crypt_encode_params_bp(bp, zp->zp_salt, zp->zp_iv);
62840030 3982 return (zio);
b5256303
TC
3983 }
3984
3985 /* indirect blocks only maintain a cksum of the lower level MACs */
3986 if (BP_GET_LEVEL(bp) > 0) {
3987 BP_SET_CRYPT(bp, B_TRUE);
3988 VERIFY0(zio_crypt_do_indirect_mac_checksum_abd(B_TRUE,
3989 zio->io_orig_abd, BP_GET_LSIZE(bp), BP_SHOULD_BYTESWAP(bp),
3990 mac));
3991 zio_crypt_encode_mac_bp(bp, mac);
62840030 3992 return (zio);
b5256303
TC
3993 }
3994
3995 /*
3996 * Objset blocks are a special case since they have 2 256-bit MACs
3997 * embedded within them.
3998 */
3999 if (ot == DMU_OT_OBJSET) {
4000 ASSERT0(DMU_OT_IS_ENCRYPTED(ot));
4001 ASSERT3U(BP_GET_COMPRESS(bp), ==, ZIO_COMPRESS_OFF);
4002 BP_SET_CRYPT(bp, B_TRUE);
ae76f45c
TC
4003 VERIFY0(spa_do_crypt_objset_mac_abd(B_TRUE, spa, dsobj,
4004 zio->io_abd, psize, BP_SHOULD_BYTESWAP(bp)));
62840030 4005 return (zio);
b5256303
TC
4006 }
4007
4008 /* unencrypted object types are only authenticated with a MAC */
4009 if (!DMU_OT_IS_ENCRYPTED(ot)) {
4010 BP_SET_CRYPT(bp, B_TRUE);
ae76f45c
TC
4011 VERIFY0(spa_do_crypt_mac_abd(B_TRUE, spa, dsobj,
4012 zio->io_abd, psize, mac));
b5256303 4013 zio_crypt_encode_mac_bp(bp, mac);
62840030 4014 return (zio);
b5256303
TC
4015 }
4016
4017 /*
4018 * Later passes of sync-to-convergence may decide to rewrite data
4019 * in place to avoid more disk reallocations. This presents a problem
d611989f 4020 * for encryption because this constitutes rewriting the new data with
b5256303
TC
4021 * the same encryption key and IV. However, this only applies to blocks
4022 * in the MOS (particularly the spacemaps) and we do not encrypt the
4023 * MOS. We assert that the zio is allocating or an intent log write
4024 * to enforce this.
4025 */
4026 ASSERT(IO_IS_ALLOCATING(zio) || ot == DMU_OT_INTENT_LOG);
4027 ASSERT(BP_GET_LEVEL(bp) == 0 || ot == DMU_OT_INTENT_LOG);
4028 ASSERT(spa_feature_is_active(spa, SPA_FEATURE_ENCRYPTION));
4029 ASSERT3U(psize, !=, 0);
4030
4031 enc_buf = zio_buf_alloc(psize);
4032 eabd = abd_get_from_buf(enc_buf, psize);
4033 abd_take_ownership_of_buf(eabd, B_TRUE);
4034
4035 /*
4036 * For an explanation of what encryption parameters are stored
4037 * where, see the block comment in zio_crypt.c.
4038 */
4039 if (ot == DMU_OT_INTENT_LOG) {
4040 zio_crypt_decode_params_bp(bp, salt, iv);
4041 } else {
4042 BP_SET_CRYPT(bp, B_TRUE);
4043 }
4044
4045 /* Perform the encryption. This should not fail */
be9a5c35
TC
4046 VERIFY0(spa_do_crypt_abd(B_TRUE, spa, &zio->io_bookmark,
4047 BP_GET_TYPE(bp), BP_GET_DEDUP(bp), BP_SHOULD_BYTESWAP(bp),
4048 salt, iv, mac, psize, zio->io_abd, eabd, &no_crypt));
b5256303
TC
4049
4050 /* encode encryption metadata into the bp */
4051 if (ot == DMU_OT_INTENT_LOG) {
4052 /*
4053 * ZIL blocks store the MAC in the embedded checksum, so the
4054 * transform must always be applied.
4055 */
4056 zio_crypt_encode_mac_zil(enc_buf, mac);
4057 zio_push_transform(zio, eabd, psize, psize, NULL);
4058 } else {
4059 BP_SET_CRYPT(bp, B_TRUE);
4060 zio_crypt_encode_params_bp(bp, salt, iv);
4061 zio_crypt_encode_mac_bp(bp, mac);
4062
4063 if (no_crypt) {
4064 ASSERT3U(ot, ==, DMU_OT_DNODE);
4065 abd_free(eabd);
4066 } else {
4067 zio_push_transform(zio, eabd, psize, psize, NULL);
4068 }
4069 }
4070
62840030 4071 return (zio);
b5256303
TC
4072}
4073
34dc7c2f
BB
4074/*
4075 * ==========================================================================
4076 * Generate and verify checksums
4077 * ==========================================================================
4078 */
62840030 4079static zio_t *
34dc7c2f
BB
4080zio_checksum_generate(zio_t *zio)
4081{
34dc7c2f 4082 blkptr_t *bp = zio->io_bp;
b128c09f 4083 enum zio_checksum checksum;
34dc7c2f 4084
b128c09f
BB
4085 if (bp == NULL) {
4086 /*
4087 * This is zio_write_phys().
4088 * We're either generating a label checksum, or none at all.
4089 */
4090 checksum = zio->io_prop.zp_checksum;
34dc7c2f 4091
b128c09f 4092 if (checksum == ZIO_CHECKSUM_OFF)
62840030 4093 return (zio);
b128c09f
BB
4094
4095 ASSERT(checksum == ZIO_CHECKSUM_LABEL);
4096 } else {
4097 if (BP_IS_GANG(bp) && zio->io_child_type == ZIO_CHILD_GANG) {
4098 ASSERT(!IO_IS_ALLOCATING(zio));
4099 checksum = ZIO_CHECKSUM_GANG_HEADER;
4100 } else {
4101 checksum = BP_GET_CHECKSUM(bp);
4102 }
4103 }
34dc7c2f 4104
a6255b7f 4105 zio_checksum_compute(zio, checksum, zio->io_abd, zio->io_size);
34dc7c2f 4106
62840030 4107 return (zio);
34dc7c2f
BB
4108}
4109
62840030 4110static zio_t *
b128c09f 4111zio_checksum_verify(zio_t *zio)
34dc7c2f 4112{
428870ff 4113 zio_bad_cksum_t info;
b128c09f
BB
4114 blkptr_t *bp = zio->io_bp;
4115 int error;
34dc7c2f 4116
428870ff
BB
4117 ASSERT(zio->io_vd != NULL);
4118
b128c09f
BB
4119 if (bp == NULL) {
4120 /*
4121 * This is zio_read_phys().
4122 * We're either verifying a label checksum, or nothing at all.
4123 */
4124 if (zio->io_prop.zp_checksum == ZIO_CHECKSUM_OFF)
62840030 4125 return (zio);
34dc7c2f 4126
b128c09f
BB
4127 ASSERT(zio->io_prop.zp_checksum == ZIO_CHECKSUM_LABEL);
4128 }
34dc7c2f 4129
428870ff 4130 if ((error = zio_checksum_error(zio, &info)) != 0) {
b128c09f 4131 zio->io_error = error;
7a3066ff
MA
4132 if (error == ECKSUM &&
4133 !(zio->io_flags & ZIO_FLAG_SPECULATIVE)) {
428870ff 4134 zfs_ereport_start_checksum(zio->io_spa,
b5256303
TC
4135 zio->io_vd, &zio->io_bookmark, zio,
4136 zio->io_offset, zio->io_size, NULL, &info);
b128c09f 4137 }
34dc7c2f
BB
4138 }
4139
62840030 4140 return (zio);
34dc7c2f
BB
4141}
4142
4143/*
4144 * Called by RAID-Z to ensure we don't compute the checksum twice.
4145 */
4146void
4147zio_checksum_verified(zio_t *zio)
4148{
428870ff 4149 zio->io_pipeline &= ~ZIO_STAGE_CHECKSUM_VERIFY;
34dc7c2f
BB
4150}
4151
4152/*
b128c09f
BB
4153 * ==========================================================================
4154 * Error rank. Error are ranked in the order 0, ENXIO, ECKSUM, EIO, other.
9b67f605 4155 * An error of 0 indicates success. ENXIO indicates whole-device failure,
d611989f 4156 * which may be transient (e.g. unplugged) or permanent. ECKSUM and EIO
b128c09f
BB
4157 * indicate errors that are specific to one I/O, and most likely permanent.
4158 * Any other error is presumed to be worse because we weren't expecting it.
4159 * ==========================================================================
34dc7c2f 4160 */
b128c09f
BB
4161int
4162zio_worst_error(int e1, int e2)
34dc7c2f 4163{
b128c09f
BB
4164 static int zio_error_rank[] = { 0, ENXIO, ECKSUM, EIO };
4165 int r1, r2;
4166
4167 for (r1 = 0; r1 < sizeof (zio_error_rank) / sizeof (int); r1++)
4168 if (e1 == zio_error_rank[r1])
4169 break;
34dc7c2f 4170
b128c09f
BB
4171 for (r2 = 0; r2 < sizeof (zio_error_rank) / sizeof (int); r2++)
4172 if (e2 == zio_error_rank[r2])
4173 break;
4174
4175 return (r1 > r2 ? e1 : e2);
34dc7c2f
BB
4176}
4177
4178/*
4179 * ==========================================================================
b128c09f 4180 * I/O completion
34dc7c2f
BB
4181 * ==========================================================================
4182 */
62840030 4183static zio_t *
b128c09f 4184zio_ready(zio_t *zio)
34dc7c2f 4185{
b128c09f 4186 blkptr_t *bp = zio->io_bp;
d164b209 4187 zio_t *pio, *pio_next;
3dfb57a3 4188 zio_link_t *zl = NULL;
34dc7c2f 4189
ddc751d5
GW
4190 if (zio_wait_for_children(zio, ZIO_CHILD_GANG_BIT | ZIO_CHILD_DDT_BIT,
4191 ZIO_WAIT_READY)) {
62840030 4192 return (NULL);
ddc751d5 4193 }
34dc7c2f 4194
9babb374 4195 if (zio->io_ready) {
b128c09f 4196 ASSERT(IO_IS_ALLOCATING(zio));
03c6040b
GW
4197 ASSERT(bp->blk_birth == zio->io_txg || BP_IS_HOLE(bp) ||
4198 (zio->io_flags & ZIO_FLAG_NOPWRITE));
b128c09f 4199 ASSERT(zio->io_children[ZIO_CHILD_GANG][ZIO_WAIT_READY] == 0);
34dc7c2f 4200
b128c09f
BB
4201 zio->io_ready(zio);
4202 }
34dc7c2f 4203
b128c09f
BB
4204 if (bp != NULL && bp != &zio->io_bp_copy)
4205 zio->io_bp_copy = *bp;
34dc7c2f 4206
3dfb57a3 4207 if (zio->io_error != 0) {
b128c09f 4208 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
34dc7c2f 4209
3dfb57a3
DB
4210 if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
4211 ASSERT(IO_IS_ALLOCATING(zio));
4212 ASSERT(zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
cc99f275
DB
4213 ASSERT(zio->io_metaslab_class != NULL);
4214
3dfb57a3
DB
4215 /*
4216 * We were unable to allocate anything, unreserve and
4217 * issue the next I/O to allocate.
4218 */
4219 metaslab_class_throttle_unreserve(
cc99f275
DB
4220 zio->io_metaslab_class, zio->io_prop.zp_copies,
4221 zio->io_allocator, zio);
492f64e9 4222 zio_allocate_dispatch(zio->io_spa, zio->io_allocator);
3dfb57a3
DB
4223 }
4224 }
4225
d164b209
BB
4226 mutex_enter(&zio->io_lock);
4227 zio->io_state[ZIO_WAIT_READY] = 1;
3dfb57a3 4228 pio = zio_walk_parents(zio, &zl);
d164b209
BB
4229 mutex_exit(&zio->io_lock);
4230
4231 /*
4232 * As we notify zio's parents, new parents could be added.
4233 * New parents go to the head of zio's io_parent_list, however,
4234 * so we will (correctly) not notify them. The remainder of zio's
4235 * io_parent_list, from 'pio_next' onward, cannot change because
4236 * all parents must wait for us to be done before they can be done.
4237 */
4238 for (; pio != NULL; pio = pio_next) {
3dfb57a3 4239 pio_next = zio_walk_parents(zio, &zl);
62840030 4240 zio_notify_parent(pio, zio, ZIO_WAIT_READY, NULL);
d164b209 4241 }
34dc7c2f 4242
428870ff
BB
4243 if (zio->io_flags & ZIO_FLAG_NODATA) {
4244 if (BP_IS_GANG(bp)) {
4245 zio->io_flags &= ~ZIO_FLAG_NODATA;
4246 } else {
a6255b7f 4247 ASSERT((uintptr_t)zio->io_abd < SPA_MAXBLOCKSIZE);
428870ff
BB
4248 zio->io_pipeline &= ~ZIO_VDEV_IO_STAGES;
4249 }
4250 }
4251
4252 if (zio_injection_enabled &&
4253 zio->io_spa->spa_syncing_txg == zio->io_txg)
4254 zio_handle_ignored_writes(zio);
4255
62840030 4256 return (zio);
34dc7c2f
BB
4257}
4258
3dfb57a3
DB
4259/*
4260 * Update the allocation throttle accounting.
4261 */
4262static void
4263zio_dva_throttle_done(zio_t *zio)
4264{
1c27024e 4265 ASSERTV(zio_t *lio = zio->io_logical);
3dfb57a3
DB
4266 zio_t *pio = zio_unique_parent(zio);
4267 vdev_t *vd = zio->io_vd;
4268 int flags = METASLAB_ASYNC_ALLOC;
4269
4270 ASSERT3P(zio->io_bp, !=, NULL);
4271 ASSERT3U(zio->io_type, ==, ZIO_TYPE_WRITE);
4272 ASSERT3U(zio->io_priority, ==, ZIO_PRIORITY_ASYNC_WRITE);
4273 ASSERT3U(zio->io_child_type, ==, ZIO_CHILD_VDEV);
4274 ASSERT(vd != NULL);
4275 ASSERT3P(vd, ==, vd->vdev_top);
21df134f
SB
4276 ASSERT(zio_injection_enabled || !(zio->io_flags & ZIO_FLAG_IO_RETRY));
4277 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REPAIR));
3dfb57a3
DB
4278 ASSERT(zio->io_flags & ZIO_FLAG_IO_ALLOCATING);
4279 ASSERT(!(lio->io_flags & ZIO_FLAG_IO_REWRITE));
4280 ASSERT(!(lio->io_orig_flags & ZIO_FLAG_NODATA));
4281
4282 /*
4283 * Parents of gang children can have two flavors -- ones that
4284 * allocated the gang header (will have ZIO_FLAG_IO_REWRITE set)
4285 * and ones that allocated the constituent blocks. The allocation
4286 * throttle needs to know the allocating parent zio so we must find
4287 * it here.
4288 */
4289 if (pio->io_child_type == ZIO_CHILD_GANG) {
4290 /*
4291 * If our parent is a rewrite gang child then our grandparent
4292 * would have been the one that performed the allocation.
4293 */
4294 if (pio->io_flags & ZIO_FLAG_IO_REWRITE)
4295 pio = zio_unique_parent(pio);
4296 flags |= METASLAB_GANG_CHILD;
4297 }
4298
4299 ASSERT(IO_IS_ALLOCATING(pio));
4300 ASSERT3P(zio, !=, zio->io_logical);
4301 ASSERT(zio->io_logical != NULL);
4302 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REPAIR));
4303 ASSERT0(zio->io_flags & ZIO_FLAG_NOPWRITE);
cc99f275 4304 ASSERT(zio->io_metaslab_class != NULL);
3dfb57a3
DB
4305
4306 mutex_enter(&pio->io_lock);
492f64e9
PD
4307 metaslab_group_alloc_decrement(zio->io_spa, vd->vdev_id, pio, flags,
4308 pio->io_allocator, B_TRUE);
3dfb57a3
DB
4309 mutex_exit(&pio->io_lock);
4310
cc99f275
DB
4311 metaslab_class_throttle_unreserve(zio->io_metaslab_class, 1,
4312 pio->io_allocator, pio);
3dfb57a3
DB
4313
4314 /*
4315 * Call into the pipeline to see if there is more work that
4316 * needs to be done. If there is work to be done it will be
4317 * dispatched to another taskq thread.
4318 */
492f64e9 4319 zio_allocate_dispatch(zio->io_spa, pio->io_allocator);
3dfb57a3
DB
4320}
4321
62840030 4322static zio_t *
b128c09f 4323zio_done(zio_t *zio)
34dc7c2f 4324{
3dfb57a3
DB
4325 /*
4326 * Always attempt to keep stack usage minimal here since
d611989f 4327 * we can be called recursively up to 19 levels deep.
3dfb57a3 4328 */
84c07ada 4329 const uint64_t psize = zio->io_size;
d164b209 4330 zio_t *pio, *pio_next;
3dfb57a3 4331 zio_link_t *zl = NULL;
34dc7c2f 4332
b128c09f 4333 /*
9babb374 4334 * If our children haven't all completed,
b128c09f
BB
4335 * wait for them and then repeat this pipeline stage.
4336 */
ddc751d5 4337 if (zio_wait_for_children(zio, ZIO_CHILD_ALL_BITS, ZIO_WAIT_DONE)) {
62840030 4338 return (NULL);
ddc751d5 4339 }
34dc7c2f 4340
3dfb57a3
DB
4341 /*
4342 * If the allocation throttle is enabled, then update the accounting.
4343 * We only track child I/Os that are part of an allocating async
4344 * write. We must do this since the allocation is performed
4345 * by the logical I/O but the actual write is done by child I/Os.
4346 */
4347 if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING &&
4348 zio->io_child_type == ZIO_CHILD_VDEV) {
cc99f275
DB
4349 ASSERT(zio->io_metaslab_class != NULL);
4350 ASSERT(zio->io_metaslab_class->mc_alloc_throttle_enabled);
3dfb57a3
DB
4351 zio_dva_throttle_done(zio);
4352 }
4353
4354 /*
4355 * If the allocation throttle is enabled, verify that
4356 * we have decremented the refcounts for every I/O that was throttled.
4357 */
4358 if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
4359 ASSERT(zio->io_type == ZIO_TYPE_WRITE);
4360 ASSERT(zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
4361 ASSERT(zio->io_bp != NULL);
cc99f275 4362
492f64e9
PD
4363 metaslab_group_alloc_verify(zio->io_spa, zio->io_bp, zio,
4364 zio->io_allocator);
424fd7c3 4365 VERIFY(zfs_refcount_not_held(
cc99f275 4366 &zio->io_metaslab_class->mc_alloc_slots[zio->io_allocator],
492f64e9 4367 zio));
3dfb57a3
DB
4368 }
4369
4370
1c27024e
DB
4371 for (int c = 0; c < ZIO_CHILD_TYPES; c++)
4372 for (int w = 0; w < ZIO_WAIT_TYPES; w++)
b128c09f
BB
4373 ASSERT(zio->io_children[c][w] == 0);
4374
9b67f605 4375 if (zio->io_bp != NULL && !BP_IS_EMBEDDED(zio->io_bp)) {
c776b317
BB
4376 ASSERT(zio->io_bp->blk_pad[0] == 0);
4377 ASSERT(zio->io_bp->blk_pad[1] == 0);
d1d7e268
MK
4378 ASSERT(bcmp(zio->io_bp, &zio->io_bp_copy,
4379 sizeof (blkptr_t)) == 0 ||
c776b317
BB
4380 (zio->io_bp == zio_unique_parent(zio)->io_bp));
4381 if (zio->io_type == ZIO_TYPE_WRITE && !BP_IS_HOLE(zio->io_bp) &&
428870ff 4382 zio->io_bp_override == NULL &&
b128c09f 4383 !(zio->io_flags & ZIO_FLAG_IO_REPAIR)) {
d1d7e268
MK
4384 ASSERT3U(zio->io_prop.zp_copies, <=,
4385 BP_GET_NDVAS(zio->io_bp));
c776b317 4386 ASSERT(BP_COUNT_GANG(zio->io_bp) == 0 ||
d1d7e268
MK
4387 (BP_COUNT_GANG(zio->io_bp) ==
4388 BP_GET_NDVAS(zio->io_bp)));
b128c09f 4389 }
03c6040b
GW
4390 if (zio->io_flags & ZIO_FLAG_NOPWRITE)
4391 VERIFY(BP_EQUAL(zio->io_bp, &zio->io_bp_orig));
b128c09f
BB
4392 }
4393
4394 /*
428870ff 4395 * If there were child vdev/gang/ddt errors, they apply to us now.
b128c09f
BB
4396 */
4397 zio_inherit_child_errors(zio, ZIO_CHILD_VDEV);
4398 zio_inherit_child_errors(zio, ZIO_CHILD_GANG);
428870ff
BB
4399 zio_inherit_child_errors(zio, ZIO_CHILD_DDT);
4400
4401 /*
4402 * If the I/O on the transformed data was successful, generate any
4403 * checksum reports now while we still have the transformed data.
4404 */
4405 if (zio->io_error == 0) {
4406 while (zio->io_cksum_report != NULL) {
4407 zio_cksum_report_t *zcr = zio->io_cksum_report;
4408 uint64_t align = zcr->zcr_align;
a6255b7f 4409 uint64_t asize = P2ROUNDUP(psize, align);
a6255b7f
DQ
4410 abd_t *adata = zio->io_abd;
4411
4412 if (asize != psize) {
84c07ada 4413 adata = abd_alloc(asize, B_TRUE);
a6255b7f
DQ
4414 abd_copy(adata, zio->io_abd, psize);
4415 abd_zero_off(adata, psize, asize - psize);
428870ff
BB
4416 }
4417
4418 zio->io_cksum_report = zcr->zcr_next;
4419 zcr->zcr_next = NULL;
84c07ada 4420 zcr->zcr_finish(zcr, adata);
428870ff
BB
4421 zfs_ereport_free_checksum(zcr);
4422
a6255b7f
DQ
4423 if (asize != psize)
4424 abd_free(adata);
428870ff
BB
4425 }
4426 }
b128c09f
BB
4427
4428 zio_pop_transforms(zio); /* note: may set zio->io_error */
4429
a6255b7f 4430 vdev_stat_update(zio, psize);
b128c09f 4431
a69052be 4432 /*
cc92e9d0 4433 * If this I/O is attached to a particular vdev is slow, exceeding
72f53c56
MJ
4434 * 30 seconds to complete, post an error described the I/O delay.
4435 * We ignore these errors if the device is currently unavailable.
a69052be 4436 */
ad796b8a
TH
4437 if (zio->io_delay >= MSEC2NSEC(zio_slow_io_ms)) {
4438 if (zio->io_vd != NULL && !vdev_is_dead(zio->io_vd)) {
4439 /*
4440 * We want to only increment our slow IO counters if
4441 * the IO is valid (i.e. not if the drive is removed).
4442 *
4443 * zfs_ereport_post() will also do these checks, but
4444 * it can also ratelimit and have other failures, so we
4445 * need to increment the slow_io counters independent
4446 * of it.
4447 */
4448 if (zfs_ereport_is_valid(FM_EREPORT_ZFS_DELAY,
4449 zio->io_spa, zio->io_vd, zio)) {
4450 mutex_enter(&zio->io_vd->vdev_stat_lock);
4451 zio->io_vd->vdev_stat.vs_slow_ios++;
4452 mutex_exit(&zio->io_vd->vdev_stat_lock);
4453
4454 zfs_ereport_post(FM_EREPORT_ZFS_DELAY,
4455 zio->io_spa, zio->io_vd, &zio->io_bookmark,
4456 zio, 0, 0);
4457 }
4458 }
72f53c56 4459 }
a69052be 4460
b128c09f
BB
4461 if (zio->io_error) {
4462 /*
4463 * If this I/O is attached to a particular vdev,
4464 * generate an error message describing the I/O failure
4465 * at the block level. We ignore these errors if the
4466 * device is currently unavailable.
4467 */
c776b317 4468 if (zio->io_error != ECKSUM && zio->io_vd != NULL &&
02730c33 4469 !vdev_is_dead(zio->io_vd))
c776b317 4470 zfs_ereport_post(FM_EREPORT_ZFS_IO, zio->io_spa,
b5256303 4471 zio->io_vd, &zio->io_bookmark, zio, 0, 0);
34dc7c2f 4472
428870ff
BB
4473 if ((zio->io_error == EIO || !(zio->io_flags &
4474 (ZIO_FLAG_SPECULATIVE | ZIO_FLAG_DONT_PROPAGATE))) &&
c776b317 4475 zio == zio->io_logical) {
b128c09f
BB
4476 /*
4477 * For logical I/O requests, tell the SPA to log the
4478 * error and generate a logical data ereport.
4479 */
b5256303 4480 spa_log_error(zio->io_spa, &zio->io_bookmark);
d1d7e268 4481 zfs_ereport_post(FM_EREPORT_ZFS_DATA, zio->io_spa,
b5256303 4482 NULL, &zio->io_bookmark, zio, 0, 0);
b128c09f
BB
4483 }
4484 }
34dc7c2f 4485
c776b317 4486 if (zio->io_error && zio == zio->io_logical) {
b128c09f
BB
4487 /*
4488 * Determine whether zio should be reexecuted. This will
4489 * propagate all the way to the root via zio_notify_parent().
4490 */
c776b317 4491 ASSERT(zio->io_vd == NULL && zio->io_bp != NULL);
428870ff 4492 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
b128c09f 4493
428870ff
BB
4494 if (IO_IS_ALLOCATING(zio) &&
4495 !(zio->io_flags & ZIO_FLAG_CANFAIL)) {
b128c09f
BB
4496 if (zio->io_error != ENOSPC)
4497 zio->io_reexecute |= ZIO_REEXECUTE_NOW;
4498 else
4499 zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
428870ff 4500 }
b128c09f
BB
4501
4502 if ((zio->io_type == ZIO_TYPE_READ ||
4503 zio->io_type == ZIO_TYPE_FREE) &&
572e2857 4504 !(zio->io_flags & ZIO_FLAG_SCAN_THREAD) &&
b128c09f 4505 zio->io_error == ENXIO &&
c776b317
BB
4506 spa_load_state(zio->io_spa) == SPA_LOAD_NONE &&
4507 spa_get_failmode(zio->io_spa) != ZIO_FAILURE_MODE_CONTINUE)
b128c09f
BB
4508 zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
4509
4510 if (!(zio->io_flags & ZIO_FLAG_CANFAIL) && !zio->io_reexecute)
4511 zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
428870ff
BB
4512
4513 /*
4514 * Here is a possibly good place to attempt to do
4515 * either combinatorial reconstruction or error correction
4516 * based on checksums. It also might be a good place
4517 * to send out preliminary ereports before we suspend
4518 * processing.
4519 */
34dc7c2f
BB
4520 }
4521
4522 /*
b128c09f
BB
4523 * If there were logical child errors, they apply to us now.
4524 * We defer this until now to avoid conflating logical child
4525 * errors with errors that happened to the zio itself when
4526 * updating vdev stats and reporting FMA events above.
34dc7c2f 4527 */
b128c09f 4528 zio_inherit_child_errors(zio, ZIO_CHILD_LOGICAL);
34dc7c2f 4529
428870ff
BB
4530 if ((zio->io_error || zio->io_reexecute) &&
4531 IO_IS_ALLOCATING(zio) && zio->io_gang_leader == zio &&
03c6040b 4532 !(zio->io_flags & (ZIO_FLAG_IO_REWRITE | ZIO_FLAG_NOPWRITE)))
c776b317 4533 zio_dva_unallocate(zio, zio->io_gang_tree, zio->io_bp);
9babb374
BB
4534
4535 zio_gang_tree_free(&zio->io_gang_tree);
4536
4537 /*
4538 * Godfather I/Os should never suspend.
4539 */
4540 if ((zio->io_flags & ZIO_FLAG_GODFATHER) &&
4541 (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND))
a32494d2 4542 zio->io_reexecute &= ~ZIO_REEXECUTE_SUSPEND;
9babb374 4543
b128c09f
BB
4544 if (zio->io_reexecute) {
4545 /*
4546 * This is a logical I/O that wants to reexecute.
4547 *
4548 * Reexecute is top-down. When an i/o fails, if it's not
4549 * the root, it simply notifies its parent and sticks around.
4550 * The parent, seeing that it still has children in zio_done(),
4551 * does the same. This percolates all the way up to the root.
4552 * The root i/o will reexecute or suspend the entire tree.
4553 *
4554 * This approach ensures that zio_reexecute() honors
4555 * all the original i/o dependency relationships, e.g.
4556 * parents not executing until children are ready.
4557 */
4558 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
34dc7c2f 4559
9babb374 4560 zio->io_gang_leader = NULL;
b128c09f 4561
d164b209
BB
4562 mutex_enter(&zio->io_lock);
4563 zio->io_state[ZIO_WAIT_DONE] = 1;
4564 mutex_exit(&zio->io_lock);
4565
9babb374
BB
4566 /*
4567 * "The Godfather" I/O monitors its children but is
4568 * not a true parent to them. It will track them through
4569 * the pipeline but severs its ties whenever they get into
4570 * trouble (e.g. suspended). This allows "The Godfather"
4571 * I/O to return status without blocking.
4572 */
3dfb57a3
DB
4573 zl = NULL;
4574 for (pio = zio_walk_parents(zio, &zl); pio != NULL;
4575 pio = pio_next) {
4576 zio_link_t *remove_zl = zl;
4577 pio_next = zio_walk_parents(zio, &zl);
9babb374
BB
4578
4579 if ((pio->io_flags & ZIO_FLAG_GODFATHER) &&
4580 (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND)) {
3dfb57a3 4581 zio_remove_child(pio, zio, remove_zl);
62840030
MA
4582 /*
4583 * This is a rare code path, so we don't
4584 * bother with "next_to_execute".
4585 */
4586 zio_notify_parent(pio, zio, ZIO_WAIT_DONE,
4587 NULL);
9babb374
BB
4588 }
4589 }
4590
d164b209 4591 if ((pio = zio_unique_parent(zio)) != NULL) {
b128c09f
BB
4592 /*
4593 * We're not a root i/o, so there's nothing to do
4594 * but notify our parent. Don't propagate errors
4595 * upward since we haven't permanently failed yet.
4596 */
9babb374 4597 ASSERT(!(zio->io_flags & ZIO_FLAG_GODFATHER));
b128c09f 4598 zio->io_flags |= ZIO_FLAG_DONT_PROPAGATE;
62840030
MA
4599 /*
4600 * This is a rare code path, so we don't bother with
4601 * "next_to_execute".
4602 */
4603 zio_notify_parent(pio, zio, ZIO_WAIT_DONE, NULL);
b128c09f
BB
4604 } else if (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND) {
4605 /*
4606 * We'd fail again if we reexecuted now, so suspend
4607 * until conditions improve (e.g. device comes online).
4608 */
cec3a0a1 4609 zio_suspend(zio->io_spa, zio, ZIO_SUSPEND_IOERR);
b128c09f
BB
4610 } else {
4611 /*
4612 * Reexecution is potentially a huge amount of work.
4613 * Hand it off to the otherwise-unused claim taskq.
4614 */
a38718a6 4615 ASSERT(taskq_empty_ent(&zio->io_tqent));
7ef5e54e
AL
4616 spa_taskq_dispatch_ent(zio->io_spa,
4617 ZIO_TYPE_CLAIM, ZIO_TASKQ_ISSUE,
a38718a6
GA
4618 (task_func_t *)zio_reexecute, zio, 0,
4619 &zio->io_tqent);
b128c09f 4620 }
62840030 4621 return (NULL);
34dc7c2f
BB
4622 }
4623
428870ff 4624 ASSERT(zio->io_child_count == 0);
b128c09f
BB
4625 ASSERT(zio->io_reexecute == 0);
4626 ASSERT(zio->io_error == 0 || (zio->io_flags & ZIO_FLAG_CANFAIL));
34dc7c2f 4627
428870ff
BB
4628 /*
4629 * Report any checksum errors, since the I/O is complete.
4630 */
4631 while (zio->io_cksum_report != NULL) {
4632 zio_cksum_report_t *zcr = zio->io_cksum_report;
4633 zio->io_cksum_report = zcr->zcr_next;
4634 zcr->zcr_next = NULL;
4635 zcr->zcr_finish(zcr, NULL);
4636 zfs_ereport_free_checksum(zcr);
4637 }
4638
920dd524 4639 if (zio->io_flags & ZIO_FLAG_FASTWRITE && zio->io_bp &&
9b67f605
MA
4640 !BP_IS_HOLE(zio->io_bp) && !BP_IS_EMBEDDED(zio->io_bp) &&
4641 !(zio->io_flags & ZIO_FLAG_NOPWRITE)) {
920dd524
ED
4642 metaslab_fastwrite_unmark(zio->io_spa, zio->io_bp);
4643 }
4644
d164b209
BB
4645 /*
4646 * It is the responsibility of the done callback to ensure that this
4647 * particular zio is no longer discoverable for adoption, and as
4648 * such, cannot acquire any new parents.
4649 */
b128c09f
BB
4650 if (zio->io_done)
4651 zio->io_done(zio);
34dc7c2f 4652
d164b209
BB
4653 mutex_enter(&zio->io_lock);
4654 zio->io_state[ZIO_WAIT_DONE] = 1;
4655 mutex_exit(&zio->io_lock);
34dc7c2f 4656
62840030
MA
4657 /*
4658 * We are done executing this zio. We may want to execute a parent
4659 * next. See the comment in zio_notify_parent().
4660 */
4661 zio_t *next_to_execute = NULL;
3dfb57a3
DB
4662 zl = NULL;
4663 for (pio = zio_walk_parents(zio, &zl); pio != NULL; pio = pio_next) {
4664 zio_link_t *remove_zl = zl;
4665 pio_next = zio_walk_parents(zio, &zl);
4666 zio_remove_child(pio, zio, remove_zl);
62840030 4667 zio_notify_parent(pio, zio, ZIO_WAIT_DONE, &next_to_execute);
b128c09f 4668 }
34dc7c2f 4669
b128c09f
BB
4670 if (zio->io_waiter != NULL) {
4671 mutex_enter(&zio->io_lock);
4672 zio->io_executor = NULL;
4673 cv_broadcast(&zio->io_cv);
4674 mutex_exit(&zio->io_lock);
4675 } else {
4676 zio_destroy(zio);
4677 }
34dc7c2f 4678
62840030 4679 return (next_to_execute);
34dc7c2f
BB
4680}
4681
4682/*
b128c09f
BB
4683 * ==========================================================================
4684 * I/O pipeline definition
4685 * ==========================================================================
34dc7c2f 4686 */
428870ff 4687static zio_pipe_stage_t *zio_pipeline[] = {
b128c09f 4688 NULL,
b128c09f 4689 zio_read_bp_init,
3dfb57a3 4690 zio_write_bp_init,
428870ff
BB
4691 zio_free_bp_init,
4692 zio_issue_async,
3dfb57a3 4693 zio_write_compress,
b5256303 4694 zio_encrypt,
b128c09f 4695 zio_checksum_generate,
03c6040b 4696 zio_nop_write,
428870ff
BB
4697 zio_ddt_read_start,
4698 zio_ddt_read_done,
4699 zio_ddt_write,
4700 zio_ddt_free,
b128c09f
BB
4701 zio_gang_assemble,
4702 zio_gang_issue,
3dfb57a3 4703 zio_dva_throttle,
b128c09f
BB
4704 zio_dva_allocate,
4705 zio_dva_free,
4706 zio_dva_claim,
4707 zio_ready,
4708 zio_vdev_io_start,
4709 zio_vdev_io_done,
4710 zio_vdev_io_assess,
4711 zio_checksum_verify,
4712 zio_done
4713};
c28b2279 4714
9ae529ec 4715
9ae529ec 4716
9ae529ec 4717
fcff0f35
PD
4718/*
4719 * Compare two zbookmark_phys_t's to see which we would reach first in a
4720 * pre-order traversal of the object tree.
4721 *
4722 * This is simple in every case aside from the meta-dnode object. For all other
4723 * objects, we traverse them in order (object 1 before object 2, and so on).
4724 * However, all of these objects are traversed while traversing object 0, since
4725 * the data it points to is the list of objects. Thus, we need to convert to a
4726 * canonical representation so we can compare meta-dnode bookmarks to
4727 * non-meta-dnode bookmarks.
4728 *
4729 * We do this by calculating "equivalents" for each field of the zbookmark.
4730 * zbookmarks outside of the meta-dnode use their own object and level, and
4731 * calculate the level 0 equivalent (the first L0 blkid that is contained in the
4732 * blocks this bookmark refers to) by multiplying their blkid by their span
4733 * (the number of L0 blocks contained within one block at their level).
4734 * zbookmarks inside the meta-dnode calculate their object equivalent
4735 * (which is L0equiv * dnodes per data block), use 0 for their L0equiv, and use
4736 * level + 1<<31 (any value larger than a level could ever be) for their level.
4737 * This causes them to always compare before a bookmark in their object
4738 * equivalent, compare appropriately to bookmarks in other objects, and to
4739 * compare appropriately to other bookmarks in the meta-dnode.
4740 */
4741int
4742zbookmark_compare(uint16_t dbss1, uint8_t ibs1, uint16_t dbss2, uint8_t ibs2,
4743 const zbookmark_phys_t *zb1, const zbookmark_phys_t *zb2)
4744{
4745 /*
4746 * These variables represent the "equivalent" values for the zbookmark,
4747 * after converting zbookmarks inside the meta dnode to their
4748 * normal-object equivalents.
4749 */
4750 uint64_t zb1obj, zb2obj;
4751 uint64_t zb1L0, zb2L0;
4752 uint64_t zb1level, zb2level;
4753
4754 if (zb1->zb_object == zb2->zb_object &&
4755 zb1->zb_level == zb2->zb_level &&
4756 zb1->zb_blkid == zb2->zb_blkid)
4757 return (0);
9ae529ec 4758
fcff0f35
PD
4759 /*
4760 * BP_SPANB calculates the span in blocks.
4761 */
4762 zb1L0 = (zb1->zb_blkid) * BP_SPANB(ibs1, zb1->zb_level);
4763 zb2L0 = (zb2->zb_blkid) * BP_SPANB(ibs2, zb2->zb_level);
9ae529ec
CS
4764
4765 if (zb1->zb_object == DMU_META_DNODE_OBJECT) {
fcff0f35
PD
4766 zb1obj = zb1L0 * (dbss1 << (SPA_MINBLOCKSHIFT - DNODE_SHIFT));
4767 zb1L0 = 0;
4768 zb1level = zb1->zb_level + COMPARE_META_LEVEL;
4769 } else {
4770 zb1obj = zb1->zb_object;
4771 zb1level = zb1->zb_level;
9ae529ec
CS
4772 }
4773
fcff0f35
PD
4774 if (zb2->zb_object == DMU_META_DNODE_OBJECT) {
4775 zb2obj = zb2L0 * (dbss2 << (SPA_MINBLOCKSHIFT - DNODE_SHIFT));
4776 zb2L0 = 0;
4777 zb2level = zb2->zb_level + COMPARE_META_LEVEL;
4778 } else {
4779 zb2obj = zb2->zb_object;
4780 zb2level = zb2->zb_level;
4781 }
4782
4783 /* Now that we have a canonical representation, do the comparison. */
4784 if (zb1obj != zb2obj)
4785 return (zb1obj < zb2obj ? -1 : 1);
4786 else if (zb1L0 != zb2L0)
4787 return (zb1L0 < zb2L0 ? -1 : 1);
4788 else if (zb1level != zb2level)
4789 return (zb1level > zb2level ? -1 : 1);
4790 /*
4791 * This can (theoretically) happen if the bookmarks have the same object
4792 * and level, but different blkids, if the block sizes are not the same.
4793 * There is presently no way to change the indirect block sizes
4794 */
4795 return (0);
4796}
4797
4798/*
4799 * This function checks the following: given that last_block is the place that
4800 * our traversal stopped last time, does that guarantee that we've visited
4801 * every node under subtree_root? Therefore, we can't just use the raw output
4802 * of zbookmark_compare. We have to pass in a modified version of
4803 * subtree_root; by incrementing the block id, and then checking whether
4804 * last_block is before or equal to that, we can tell whether or not having
4805 * visited last_block implies that all of subtree_root's children have been
4806 * visited.
4807 */
4808boolean_t
4809zbookmark_subtree_completed(const dnode_phys_t *dnp,
4810 const zbookmark_phys_t *subtree_root, const zbookmark_phys_t *last_block)
4811{
4812 zbookmark_phys_t mod_zb = *subtree_root;
4813 mod_zb.zb_blkid++;
4814 ASSERT(last_block->zb_level == 0);
4815
4816 /* The objset_phys_t isn't before anything. */
4817 if (dnp == NULL)
9ae529ec 4818 return (B_FALSE);
fcff0f35
PD
4819
4820 /*
4821 * We pass in 1ULL << (DNODE_BLOCK_SHIFT - SPA_MINBLOCKSHIFT) for the
4822 * data block size in sectors, because that variable is only used if
4823 * the bookmark refers to a block in the meta-dnode. Since we don't
4824 * know without examining it what object it refers to, and there's no
4825 * harm in passing in this value in other cases, we always pass it in.
4826 *
4827 * We pass in 0 for the indirect block size shift because zb2 must be
4828 * level 0. The indirect block size is only used to calculate the span
4829 * of the bookmark, but since the bookmark must be level 0, the span is
4830 * always 1, so the math works out.
4831 *
4832 * If you make changes to how the zbookmark_compare code works, be sure
4833 * to make sure that this code still works afterwards.
4834 */
4835 return (zbookmark_compare(dnp->dn_datablkszsec, dnp->dn_indblkshift,
4836 1ULL << (DNODE_BLOCK_SHIFT - SPA_MINBLOCKSHIFT), 0, &mod_zb,
4837 last_block) <= 0);
9ae529ec
CS
4838}
4839
93ce2b4c 4840#if defined(_KERNEL)
c28b2279 4841EXPORT_SYMBOL(zio_type_name);
81971b13
BB
4842EXPORT_SYMBOL(zio_buf_alloc);
4843EXPORT_SYMBOL(zio_data_buf_alloc);
4844EXPORT_SYMBOL(zio_buf_free);
4845EXPORT_SYMBOL(zio_data_buf_free);
c28b2279 4846
ad796b8a
TH
4847module_param(zio_slow_io_ms, int, 0644);
4848MODULE_PARM_DESC(zio_slow_io_ms,
4849 "Max I/O completion time (milliseconds) before marking it as slow");
c409e464
BB
4850
4851module_param(zio_requeue_io_start_cut_in_line, int, 0644);
4852MODULE_PARM_DESC(zio_requeue_io_start_cut_in_line, "Prioritize requeued I/O");
29dee3ee
CP
4853
4854module_param(zfs_sync_pass_deferred_free, int, 0644);
4855MODULE_PARM_DESC(zfs_sync_pass_deferred_free,
d1d7e268 4856 "Defer frees starting in this pass");
29dee3ee
CP
4857
4858module_param(zfs_sync_pass_dont_compress, int, 0644);
4859MODULE_PARM_DESC(zfs_sync_pass_dont_compress,
d1d7e268 4860 "Don't compress starting in this pass");
29dee3ee
CP
4861
4862module_param(zfs_sync_pass_rewrite, int, 0644);
4863MODULE_PARM_DESC(zfs_sync_pass_rewrite,
d1d7e268 4864 "Rewrite new bps starting in this pass");
3dfb57a3
DB
4865
4866module_param(zio_dva_throttle_enabled, int, 0644);
4867MODULE_PARM_DESC(zio_dva_throttle_enabled,
4868 "Throttle block allocations in the ZIO pipeline");
638dd5f4
TC
4869
4870module_param(zio_deadman_log_all, int, 0644);
4871MODULE_PARM_DESC(zio_deadman_log_all,
4872 "Log all slow ZIOs, not just those with vdevs");
c28b2279 4873#endif