]> git.proxmox.com Git - mirror_zfs.git/blame - module/zfs/zio.c
Delay injection can cause indefinitely hung zios
[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);
f545b6ae 1837 zio_interrupt(zio);
26ef0cc7
TH
1838 } else {
1839 /*
1840 * Use taskq_dispatch_delay() in the place of
1841 * OpenZFS's timeout_generic().
1842 */
1843 tid = taskq_dispatch_delay(system_taskq,
02730c33 1844 (task_func_t *)zio_interrupt,
26ef0cc7 1845 zio, TQ_NOSLEEP, expire_at_tick);
48d3eb40 1846 if (tid == TASKQID_INVALID) {
26ef0cc7
TH
1847 /*
1848 * Couldn't allocate a task. Just
1849 * finish the zio without a delay.
1850 */
1851 zio_interrupt(zio);
1852 }
1853 }
1854 }
1855 return;
1856 }
1857#endif
1858 DTRACE_PROBE1(zio__delay__skip, zio_t *, zio);
1859 zio_interrupt(zio);
1860}
1861
8fb1ede1 1862static void
638dd5f4 1863zio_deadman_impl(zio_t *pio, int ziodepth)
8fb1ede1
BB
1864{
1865 zio_t *cio, *cio_next;
1866 zio_link_t *zl = NULL;
1867 vdev_t *vd = pio->io_vd;
1868
638dd5f4
TC
1869 if (zio_deadman_log_all || (vd != NULL && vd->vdev_ops->vdev_op_leaf)) {
1870 vdev_queue_t *vq = vd ? &vd->vdev_queue : NULL;
8fb1ede1
BB
1871 zbookmark_phys_t *zb = &pio->io_bookmark;
1872 uint64_t delta = gethrtime() - pio->io_timestamp;
1873 uint64_t failmode = spa_get_deadman_failmode(pio->io_spa);
1874
638dd5f4 1875 zfs_dbgmsg("slow zio[%d]: zio=%p timestamp=%llu "
8fb1ede1
BB
1876 "delta=%llu queued=%llu io=%llu "
1877 "path=%s last=%llu "
1878 "type=%d priority=%d flags=0x%x "
1879 "stage=0x%x pipeline=0x%x pipeline-trace=0x%x "
1880 "objset=%llu object=%llu level=%llu blkid=%llu "
1881 "offset=%llu size=%llu error=%d",
638dd5f4 1882 ziodepth, pio, pio->io_timestamp,
8fb1ede1 1883 delta, pio->io_delta, pio->io_delay,
638dd5f4 1884 vd ? vd->vdev_path : "NULL", vq ? vq->vq_io_complete_ts : 0,
8fb1ede1 1885 pio->io_type, pio->io_priority, pio->io_flags,
638dd5f4 1886 pio->io_stage, pio->io_pipeline, pio->io_pipeline_trace,
8fb1ede1
BB
1887 zb->zb_objset, zb->zb_object, zb->zb_level, zb->zb_blkid,
1888 pio->io_offset, pio->io_size, pio->io_error);
1889 zfs_ereport_post(FM_EREPORT_ZFS_DEADMAN,
1890 pio->io_spa, vd, zb, pio, 0, 0);
1891
1892 if (failmode == ZIO_FAILURE_MODE_CONTINUE &&
1893 taskq_empty_ent(&pio->io_tqent)) {
1894 zio_interrupt(pio);
1895 }
1896 }
1897
1898 mutex_enter(&pio->io_lock);
1899 for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
1900 cio_next = zio_walk_children(pio, &zl);
638dd5f4 1901 zio_deadman_impl(cio, ziodepth + 1);
8fb1ede1
BB
1902 }
1903 mutex_exit(&pio->io_lock);
1904}
1905
1906/*
1907 * Log the critical information describing this zio and all of its children
1908 * using the zfs_dbgmsg() interface then post deadman event for the ZED.
1909 */
1910void
1911zio_deadman(zio_t *pio, char *tag)
1912{
1913 spa_t *spa = pio->io_spa;
1914 char *name = spa_name(spa);
1915
1916 if (!zfs_deadman_enabled || spa_suspended(spa))
1917 return;
1918
638dd5f4 1919 zio_deadman_impl(pio, 0);
8fb1ede1
BB
1920
1921 switch (spa_get_deadman_failmode(spa)) {
1922 case ZIO_FAILURE_MODE_WAIT:
1923 zfs_dbgmsg("%s waiting for hung I/O to pool '%s'", tag, name);
1924 break;
1925
1926 case ZIO_FAILURE_MODE_CONTINUE:
1927 zfs_dbgmsg("%s restarting hung I/O for pool '%s'", tag, name);
1928 break;
1929
1930 case ZIO_FAILURE_MODE_PANIC:
1931 fm_panic("%s determined I/O to pool '%s' is hung.", tag, name);
1932 break;
1933 }
1934}
1935
b128c09f
BB
1936/*
1937 * Execute the I/O pipeline until one of the following occurs:
1938 * (1) the I/O completes; (2) the pipeline stalls waiting for
1939 * dependent child I/Os; (3) the I/O issues, so we're waiting
1940 * for an I/O completion interrupt; (4) the I/O is delegated by
1941 * vdev-level caching or aggregation; (5) the I/O is deferred
1942 * due to vdev-level queueing; (6) the I/O is handed off to
1943 * another thread. In all cases, the pipeline stops whenever
8e07b99b 1944 * there's no CPU work; it never burns a thread in cv_wait_io().
b128c09f
BB
1945 *
1946 * There's no locking on io_stage because there's no legitimate way
1947 * for multiple threads to be attempting to process the same I/O.
1948 */
428870ff 1949static zio_pipe_stage_t *zio_pipeline[];
34dc7c2f 1950
da6b4005
NB
1951/*
1952 * zio_execute() is a wrapper around the static function
1953 * __zio_execute() so that we can force __zio_execute() to be
1954 * inlined. This reduces stack overhead which is important
1955 * because __zio_execute() is called recursively in several zio
1956 * code paths. zio_execute() itself cannot be inlined because
1957 * it is externally visible.
1958 */
b128c09f
BB
1959void
1960zio_execute(zio_t *zio)
da6b4005 1961{
92119cc2
BB
1962 fstrans_cookie_t cookie;
1963
1964 cookie = spl_fstrans_mark();
da6b4005 1965 __zio_execute(zio);
92119cc2 1966 spl_fstrans_unmark(cookie);
da6b4005
NB
1967}
1968
b58986ee
BB
1969/*
1970 * Used to determine if in the current context the stack is sized large
1971 * enough to allow zio_execute() to be called recursively. A minimum
1972 * stack size of 16K is required to avoid needing to re-dispatch the zio.
1973 */
1974boolean_t
1975zio_execute_stack_check(zio_t *zio)
1976{
1977#if !defined(HAVE_LARGE_STACKS)
1978 dsl_pool_t *dp = spa_get_dsl(zio->io_spa);
1979
1980 /* Executing in txg_sync_thread() context. */
1981 if (dp && curthread == dp->dp_tx.tx_sync_thread)
1982 return (B_TRUE);
1983
1984 /* Pool initialization outside of zio_taskq context. */
1985 if (dp && spa_is_initializing(dp->dp_spa) &&
1986 !zio_taskq_member(zio, ZIO_TASKQ_ISSUE) &&
1987 !zio_taskq_member(zio, ZIO_TASKQ_ISSUE_HIGH))
1988 return (B_TRUE);
1989#endif /* HAVE_LARGE_STACKS */
1990
1991 return (B_FALSE);
1992}
1993
da6b4005
NB
1994__attribute__((always_inline))
1995static inline void
1996__zio_execute(zio_t *zio)
b128c09f 1997{
3dfb57a3
DB
1998 ASSERT3U(zio->io_queued_timestamp, >, 0);
1999
b128c09f 2000 while (zio->io_stage < ZIO_STAGE_DONE) {
428870ff
BB
2001 enum zio_stage pipeline = zio->io_pipeline;
2002 enum zio_stage stage = zio->io_stage;
62840030
MA
2003
2004 zio->io_executor = curthread;
34dc7c2f 2005
b128c09f 2006 ASSERT(!MUTEX_HELD(&zio->io_lock));
428870ff
BB
2007 ASSERT(ISP2(stage));
2008 ASSERT(zio->io_stall == NULL);
34dc7c2f 2009
428870ff
BB
2010 do {
2011 stage <<= 1;
2012 } while ((stage & pipeline) == 0);
b128c09f
BB
2013
2014 ASSERT(stage <= ZIO_STAGE_DONE);
34dc7c2f
BB
2015
2016 /*
b128c09f
BB
2017 * If we are in interrupt context and this pipeline stage
2018 * will grab a config lock that is held across I/O,
428870ff
BB
2019 * or may wait for an I/O that needs an interrupt thread
2020 * to complete, issue async to avoid deadlock.
2021 *
2022 * For VDEV_IO_START, we cut in line so that the io will
2023 * be sent to disk promptly.
34dc7c2f 2024 */
91579709
BB
2025 if ((stage & ZIO_BLOCKING_STAGES) && zio->io_vd == NULL &&
2026 zio_taskq_member(zio, ZIO_TASKQ_INTERRUPT)) {
b58986ee
BB
2027 boolean_t cut = (stage == ZIO_STAGE_VDEV_IO_START) ?
2028 zio_requeue_io_start_cut_in_line : B_FALSE;
91579709
BB
2029 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, cut);
2030 return;
2031 }
2032
2033 /*
b58986ee
BB
2034 * If the current context doesn't have large enough stacks
2035 * the zio must be issued asynchronously to prevent overflow.
91579709 2036 */
b58986ee
BB
2037 if (zio_execute_stack_check(zio)) {
2038 boolean_t cut = (stage == ZIO_STAGE_VDEV_IO_START) ?
2039 zio_requeue_io_start_cut_in_line : B_FALSE;
428870ff 2040 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, cut);
b128c09f 2041 return;
34dc7c2f
BB
2042 }
2043
b128c09f 2044 zio->io_stage = stage;
3dfb57a3 2045 zio->io_pipeline_trace |= zio->io_stage;
34dc7c2f 2046
62840030
MA
2047 /*
2048 * The zio pipeline stage returns the next zio to execute
2049 * (typically the same as this one), or NULL if we should
2050 * stop.
2051 */
2052 zio = zio_pipeline[highbit64(stage) - 1](zio);
34dc7c2f 2053
62840030
MA
2054 if (zio == NULL)
2055 return;
b128c09f 2056 }
34dc7c2f
BB
2057}
2058
da6b4005 2059
b128c09f
BB
2060/*
2061 * ==========================================================================
2062 * Initiate I/O, either sync or async
2063 * ==========================================================================
2064 */
2065int
2066zio_wait(zio_t *zio)
34dc7c2f 2067{
8fb1ede1 2068 long timeout = MSEC_TO_TICK(zfs_deadman_ziotime_ms);
b128c09f 2069 int error;
34dc7c2f 2070
1ce23dca
PS
2071 ASSERT3S(zio->io_stage, ==, ZIO_STAGE_OPEN);
2072 ASSERT3P(zio->io_executor, ==, NULL);
34dc7c2f 2073
b128c09f 2074 zio->io_waiter = curthread;
3dfb57a3
DB
2075 ASSERT0(zio->io_queued_timestamp);
2076 zio->io_queued_timestamp = gethrtime();
34dc7c2f 2077
da6b4005 2078 __zio_execute(zio);
34dc7c2f 2079
b128c09f 2080 mutex_enter(&zio->io_lock);
8fb1ede1
BB
2081 while (zio->io_executor != NULL) {
2082 error = cv_timedwait_io(&zio->io_cv, &zio->io_lock,
2083 ddi_get_lbolt() + timeout);
2084
2085 if (zfs_deadman_enabled && error == -1 &&
2086 gethrtime() - zio->io_queued_timestamp >
2087 spa_deadman_ziotime(zio->io_spa)) {
2088 mutex_exit(&zio->io_lock);
2089 timeout = MSEC_TO_TICK(zfs_deadman_checktime_ms);
2090 zio_deadman(zio, FTAG);
2091 mutex_enter(&zio->io_lock);
2092 }
2093 }
b128c09f 2094 mutex_exit(&zio->io_lock);
34dc7c2f 2095
b128c09f
BB
2096 error = zio->io_error;
2097 zio_destroy(zio);
34dc7c2f 2098
b128c09f
BB
2099 return (error);
2100}
34dc7c2f 2101
b128c09f
BB
2102void
2103zio_nowait(zio_t *zio)
2104{
1ce23dca 2105 ASSERT3P(zio->io_executor, ==, NULL);
34dc7c2f 2106
d164b209
BB
2107 if (zio->io_child_type == ZIO_CHILD_LOGICAL &&
2108 zio_unique_parent(zio) == NULL) {
8878261f
BB
2109 zio_t *pio;
2110
34dc7c2f 2111 /*
b128c09f 2112 * This is a logical async I/O with no parent to wait for it.
9babb374
BB
2113 * We add it to the spa_async_root_zio "Godfather" I/O which
2114 * will ensure they complete prior to unloading the pool.
34dc7c2f 2115 */
b128c09f 2116 spa_t *spa = zio->io_spa;
8878261f
BB
2117 kpreempt_disable();
2118 pio = spa->spa_async_zio_root[CPU_SEQID];
2119 kpreempt_enable();
9babb374 2120
8878261f 2121 zio_add_child(pio, zio);
b128c09f 2122 }
34dc7c2f 2123
3dfb57a3
DB
2124 ASSERT0(zio->io_queued_timestamp);
2125 zio->io_queued_timestamp = gethrtime();
da6b4005 2126 __zio_execute(zio);
b128c09f 2127}
34dc7c2f 2128
b128c09f
BB
2129/*
2130 * ==========================================================================
1ce23dca 2131 * Reexecute, cancel, or suspend/resume failed I/O
b128c09f
BB
2132 * ==========================================================================
2133 */
34dc7c2f 2134
b128c09f
BB
2135static void
2136zio_reexecute(zio_t *pio)
2137{
d164b209
BB
2138 zio_t *cio, *cio_next;
2139
2140 ASSERT(pio->io_child_type == ZIO_CHILD_LOGICAL);
2141 ASSERT(pio->io_orig_stage == ZIO_STAGE_OPEN);
9babb374
BB
2142 ASSERT(pio->io_gang_leader == NULL);
2143 ASSERT(pio->io_gang_tree == NULL);
34dc7c2f 2144
b128c09f
BB
2145 pio->io_flags = pio->io_orig_flags;
2146 pio->io_stage = pio->io_orig_stage;
2147 pio->io_pipeline = pio->io_orig_pipeline;
2148 pio->io_reexecute = 0;
03c6040b 2149 pio->io_flags |= ZIO_FLAG_REEXECUTED;
3dfb57a3 2150 pio->io_pipeline_trace = 0;
b128c09f 2151 pio->io_error = 0;
1c27024e 2152 for (int w = 0; w < ZIO_WAIT_TYPES; w++)
d164b209 2153 pio->io_state[w] = 0;
1c27024e 2154 for (int c = 0; c < ZIO_CHILD_TYPES; c++)
b128c09f 2155 pio->io_child_error[c] = 0;
34dc7c2f 2156
428870ff
BB
2157 if (IO_IS_ALLOCATING(pio))
2158 BP_ZERO(pio->io_bp);
34dc7c2f 2159
b128c09f
BB
2160 /*
2161 * As we reexecute pio's children, new children could be created.
d164b209 2162 * New children go to the head of pio's io_child_list, however,
b128c09f 2163 * so we will (correctly) not reexecute them. The key is that
d164b209
BB
2164 * the remainder of pio's io_child_list, from 'cio_next' onward,
2165 * cannot be affected by any side effects of reexecuting 'cio'.
b128c09f 2166 */
1c27024e 2167 zio_link_t *zl = NULL;
a8b2e306 2168 mutex_enter(&pio->io_lock);
3dfb57a3
DB
2169 for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
2170 cio_next = zio_walk_children(pio, &zl);
1c27024e 2171 for (int w = 0; w < ZIO_WAIT_TYPES; w++)
d164b209 2172 pio->io_children[cio->io_child_type][w]++;
b128c09f 2173 mutex_exit(&pio->io_lock);
d164b209 2174 zio_reexecute(cio);
a8b2e306 2175 mutex_enter(&pio->io_lock);
34dc7c2f 2176 }
a8b2e306 2177 mutex_exit(&pio->io_lock);
34dc7c2f 2178
b128c09f
BB
2179 /*
2180 * Now that all children have been reexecuted, execute the parent.
9babb374 2181 * We don't reexecute "The Godfather" I/O here as it's the
9e2c3bb4 2182 * responsibility of the caller to wait on it.
b128c09f 2183 */
3dfb57a3
DB
2184 if (!(pio->io_flags & ZIO_FLAG_GODFATHER)) {
2185 pio->io_queued_timestamp = gethrtime();
da6b4005 2186 __zio_execute(pio);
3dfb57a3 2187 }
34dc7c2f
BB
2188}
2189
b128c09f 2190void
cec3a0a1 2191zio_suspend(spa_t *spa, zio_t *zio, zio_suspend_reason_t reason)
34dc7c2f 2192{
b128c09f
BB
2193 if (spa_get_failmode(spa) == ZIO_FAILURE_MODE_PANIC)
2194 fm_panic("Pool '%s' has encountered an uncorrectable I/O "
2195 "failure and the failure mode property for this pool "
2196 "is set to panic.", spa_name(spa));
34dc7c2f 2197
bf89c199
BB
2198 cmn_err(CE_WARN, "Pool '%s' has encountered an uncorrectable I/O "
2199 "failure and has been suspended.\n", spa_name(spa));
2200
b5256303
TC
2201 zfs_ereport_post(FM_EREPORT_ZFS_IO_FAILURE, spa, NULL,
2202 NULL, NULL, 0, 0);
34dc7c2f 2203
b128c09f 2204 mutex_enter(&spa->spa_suspend_lock);
34dc7c2f 2205
b128c09f 2206 if (spa->spa_suspend_zio_root == NULL)
9babb374
BB
2207 spa->spa_suspend_zio_root = zio_root(spa, NULL, NULL,
2208 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE |
2209 ZIO_FLAG_GODFATHER);
34dc7c2f 2210
cec3a0a1 2211 spa->spa_suspended = reason;
34dc7c2f 2212
b128c09f 2213 if (zio != NULL) {
9babb374 2214 ASSERT(!(zio->io_flags & ZIO_FLAG_GODFATHER));
b128c09f
BB
2215 ASSERT(zio != spa->spa_suspend_zio_root);
2216 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
d164b209 2217 ASSERT(zio_unique_parent(zio) == NULL);
b128c09f
BB
2218 ASSERT(zio->io_stage == ZIO_STAGE_DONE);
2219 zio_add_child(spa->spa_suspend_zio_root, zio);
2220 }
34dc7c2f 2221
b128c09f
BB
2222 mutex_exit(&spa->spa_suspend_lock);
2223}
34dc7c2f 2224
9babb374 2225int
b128c09f
BB
2226zio_resume(spa_t *spa)
2227{
9babb374 2228 zio_t *pio;
34dc7c2f
BB
2229
2230 /*
b128c09f 2231 * Reexecute all previously suspended i/o.
34dc7c2f 2232 */
b128c09f 2233 mutex_enter(&spa->spa_suspend_lock);
cec3a0a1 2234 spa->spa_suspended = ZIO_SUSPEND_NONE;
b128c09f
BB
2235 cv_broadcast(&spa->spa_suspend_cv);
2236 pio = spa->spa_suspend_zio_root;
2237 spa->spa_suspend_zio_root = NULL;
2238 mutex_exit(&spa->spa_suspend_lock);
2239
2240 if (pio == NULL)
9babb374 2241 return (0);
34dc7c2f 2242
9babb374
BB
2243 zio_reexecute(pio);
2244 return (zio_wait(pio));
b128c09f
BB
2245}
2246
2247void
2248zio_resume_wait(spa_t *spa)
2249{
2250 mutex_enter(&spa->spa_suspend_lock);
2251 while (spa_suspended(spa))
2252 cv_wait(&spa->spa_suspend_cv, &spa->spa_suspend_lock);
2253 mutex_exit(&spa->spa_suspend_lock);
34dc7c2f
BB
2254}
2255
2256/*
2257 * ==========================================================================
b128c09f
BB
2258 * Gang blocks.
2259 *
2260 * A gang block is a collection of small blocks that looks to the DMU
2261 * like one large block. When zio_dva_allocate() cannot find a block
2262 * of the requested size, due to either severe fragmentation or the pool
2263 * being nearly full, it calls zio_write_gang_block() to construct the
2264 * block from smaller fragments.
2265 *
2266 * A gang block consists of a gang header (zio_gbh_phys_t) and up to
2267 * three (SPA_GBH_NBLKPTRS) gang members. The gang header is just like
2268 * an indirect block: it's an array of block pointers. It consumes
2269 * only one sector and hence is allocatable regardless of fragmentation.
2270 * The gang header's bps point to its gang members, which hold the data.
2271 *
2272 * Gang blocks are self-checksumming, using the bp's <vdev, offset, txg>
2273 * as the verifier to ensure uniqueness of the SHA256 checksum.
2274 * Critically, the gang block bp's blk_cksum is the checksum of the data,
2275 * not the gang header. This ensures that data block signatures (needed for
2276 * deduplication) are independent of how the block is physically stored.
2277 *
2278 * Gang blocks can be nested: a gang member may itself be a gang block.
2279 * Thus every gang block is a tree in which root and all interior nodes are
2280 * gang headers, and the leaves are normal blocks that contain user data.
2281 * The root of the gang tree is called the gang leader.
2282 *
2283 * To perform any operation (read, rewrite, free, claim) on a gang block,
2284 * zio_gang_assemble() first assembles the gang tree (minus data leaves)
2285 * in the io_gang_tree field of the original logical i/o by recursively
2286 * reading the gang leader and all gang headers below it. This yields
2287 * an in-core tree containing the contents of every gang header and the
2288 * bps for every constituent of the gang block.
2289 *
2290 * With the gang tree now assembled, zio_gang_issue() just walks the gang tree
2291 * and invokes a callback on each bp. To free a gang block, zio_gang_issue()
2292 * calls zio_free_gang() -- a trivial wrapper around zio_free() -- for each bp.
2293 * zio_claim_gang() provides a similarly trivial wrapper for zio_claim().
2294 * zio_read_gang() is a wrapper around zio_read() that omits reading gang
2295 * headers, since we already have those in io_gang_tree. zio_rewrite_gang()
2296 * performs a zio_rewrite() of the data or, for gang headers, a zio_rewrite()
2297 * of the gang header plus zio_checksum_compute() of the data to update the
2298 * gang header's blk_cksum as described above.
2299 *
2300 * The two-phase assemble/issue model solves the problem of partial failure --
2301 * what if you'd freed part of a gang block but then couldn't read the
2302 * gang header for another part? Assembling the entire gang tree first
2303 * ensures that all the necessary gang header I/O has succeeded before
2304 * starting the actual work of free, claim, or write. Once the gang tree
2305 * is assembled, free and claim are in-memory operations that cannot fail.
2306 *
2307 * In the event that a gang write fails, zio_dva_unallocate() walks the
2308 * gang tree to immediately free (i.e. insert back into the space map)
2309 * everything we've allocated. This ensures that we don't get ENOSPC
2310 * errors during repeated suspend/resume cycles due to a flaky device.
2311 *
2312 * Gang rewrites only happen during sync-to-convergence. If we can't assemble
2313 * the gang tree, we won't modify the block, so we can safely defer the free
2314 * (knowing that the block is still intact). If we *can* assemble the gang
2315 * tree, then even if some of the rewrites fail, zio_dva_unallocate() will free
2316 * each constituent bp and we can allocate a new block on the next sync pass.
2317 *
2318 * In all cases, the gang tree allows complete recovery from partial failure.
34dc7c2f
BB
2319 * ==========================================================================
2320 */
b128c09f 2321
a6255b7f
DQ
2322static void
2323zio_gang_issue_func_done(zio_t *zio)
2324{
2325 abd_put(zio->io_abd);
2326}
2327
b128c09f 2328static zio_t *
a6255b7f
DQ
2329zio_read_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2330 uint64_t offset)
34dc7c2f 2331{
b128c09f
BB
2332 if (gn != NULL)
2333 return (pio);
34dc7c2f 2334
a6255b7f
DQ
2335 return (zio_read(pio, pio->io_spa, bp, abd_get_offset(data, offset),
2336 BP_GET_PSIZE(bp), zio_gang_issue_func_done,
2337 NULL, pio->io_priority, ZIO_GANG_CHILD_FLAGS(pio),
b128c09f
BB
2338 &pio->io_bookmark));
2339}
2340
a6255b7f
DQ
2341static zio_t *
2342zio_rewrite_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2343 uint64_t offset)
b128c09f
BB
2344{
2345 zio_t *zio;
2346
2347 if (gn != NULL) {
a6255b7f
DQ
2348 abd_t *gbh_abd =
2349 abd_get_from_buf(gn->gn_gbh, SPA_GANGBLOCKSIZE);
b128c09f 2350 zio = zio_rewrite(pio, pio->io_spa, pio->io_txg, bp,
a6255b7f
DQ
2351 gbh_abd, SPA_GANGBLOCKSIZE, zio_gang_issue_func_done, NULL,
2352 pio->io_priority, ZIO_GANG_CHILD_FLAGS(pio),
2353 &pio->io_bookmark);
34dc7c2f 2354 /*
b128c09f
BB
2355 * As we rewrite each gang header, the pipeline will compute
2356 * a new gang block header checksum for it; but no one will
2357 * compute a new data checksum, so we do that here. The one
2358 * exception is the gang leader: the pipeline already computed
2359 * its data checksum because that stage precedes gang assembly.
2360 * (Presently, nothing actually uses interior data checksums;
2361 * this is just good hygiene.)
34dc7c2f 2362 */
9babb374 2363 if (gn != pio->io_gang_leader->io_gang_tree) {
a6255b7f
DQ
2364 abd_t *buf = abd_get_offset(data, offset);
2365
b128c09f 2366 zio_checksum_compute(zio, BP_GET_CHECKSUM(bp),
a6255b7f
DQ
2367 buf, BP_GET_PSIZE(bp));
2368
2369 abd_put(buf);
b128c09f 2370 }
428870ff
BB
2371 /*
2372 * If we are here to damage data for testing purposes,
2373 * leave the GBH alone so that we can detect the damage.
2374 */
2375 if (pio->io_gang_leader->io_flags & ZIO_FLAG_INDUCE_DAMAGE)
2376 zio->io_pipeline &= ~ZIO_VDEV_IO_STAGES;
34dc7c2f 2377 } else {
b128c09f 2378 zio = zio_rewrite(pio, pio->io_spa, pio->io_txg, bp,
a6255b7f
DQ
2379 abd_get_offset(data, offset), BP_GET_PSIZE(bp),
2380 zio_gang_issue_func_done, NULL, pio->io_priority,
b128c09f 2381 ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
34dc7c2f
BB
2382 }
2383
b128c09f
BB
2384 return (zio);
2385}
34dc7c2f 2386
b128c09f 2387/* ARGSUSED */
a6255b7f
DQ
2388static zio_t *
2389zio_free_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2390 uint64_t offset)
b128c09f 2391{
428870ff
BB
2392 return (zio_free_sync(pio, pio->io_spa, pio->io_txg, bp,
2393 ZIO_GANG_CHILD_FLAGS(pio)));
34dc7c2f
BB
2394}
2395
b128c09f 2396/* ARGSUSED */
a6255b7f
DQ
2397static zio_t *
2398zio_claim_gang(zio_t *pio, blkptr_t *bp, zio_gang_node_t *gn, abd_t *data,
2399 uint64_t offset)
34dc7c2f 2400{
b128c09f
BB
2401 return (zio_claim(pio, pio->io_spa, pio->io_txg, bp,
2402 NULL, NULL, ZIO_GANG_CHILD_FLAGS(pio)));
2403}
2404
2405static zio_gang_issue_func_t *zio_gang_issue_func[ZIO_TYPES] = {
2406 NULL,
2407 zio_read_gang,
2408 zio_rewrite_gang,
2409 zio_free_gang,
2410 zio_claim_gang,
2411 NULL
2412};
34dc7c2f 2413
b128c09f 2414static void zio_gang_tree_assemble_done(zio_t *zio);
34dc7c2f 2415
b128c09f
BB
2416static zio_gang_node_t *
2417zio_gang_node_alloc(zio_gang_node_t **gnpp)
2418{
2419 zio_gang_node_t *gn;
34dc7c2f 2420
b128c09f 2421 ASSERT(*gnpp == NULL);
34dc7c2f 2422
79c76d5b 2423 gn = kmem_zalloc(sizeof (*gn), KM_SLEEP);
b128c09f
BB
2424 gn->gn_gbh = zio_buf_alloc(SPA_GANGBLOCKSIZE);
2425 *gnpp = gn;
34dc7c2f 2426
b128c09f 2427 return (gn);
34dc7c2f
BB
2428}
2429
34dc7c2f 2430static void
b128c09f 2431zio_gang_node_free(zio_gang_node_t **gnpp)
34dc7c2f 2432{
b128c09f 2433 zio_gang_node_t *gn = *gnpp;
34dc7c2f 2434
1c27024e 2435 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++)
b128c09f
BB
2436 ASSERT(gn->gn_child[g] == NULL);
2437
2438 zio_buf_free(gn->gn_gbh, SPA_GANGBLOCKSIZE);
2439 kmem_free(gn, sizeof (*gn));
2440 *gnpp = NULL;
34dc7c2f
BB
2441}
2442
b128c09f
BB
2443static void
2444zio_gang_tree_free(zio_gang_node_t **gnpp)
34dc7c2f 2445{
b128c09f 2446 zio_gang_node_t *gn = *gnpp;
34dc7c2f 2447
b128c09f
BB
2448 if (gn == NULL)
2449 return;
34dc7c2f 2450
1c27024e 2451 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++)
b128c09f 2452 zio_gang_tree_free(&gn->gn_child[g]);
34dc7c2f 2453
b128c09f 2454 zio_gang_node_free(gnpp);
34dc7c2f
BB
2455}
2456
b128c09f 2457static void
9babb374 2458zio_gang_tree_assemble(zio_t *gio, blkptr_t *bp, zio_gang_node_t **gnpp)
34dc7c2f 2459{
b128c09f 2460 zio_gang_node_t *gn = zio_gang_node_alloc(gnpp);
a6255b7f 2461 abd_t *gbh_abd = abd_get_from_buf(gn->gn_gbh, SPA_GANGBLOCKSIZE);
b128c09f 2462
9babb374 2463 ASSERT(gio->io_gang_leader == gio);
b128c09f 2464 ASSERT(BP_IS_GANG(bp));
34dc7c2f 2465
a6255b7f
DQ
2466 zio_nowait(zio_read(gio, gio->io_spa, bp, gbh_abd, SPA_GANGBLOCKSIZE,
2467 zio_gang_tree_assemble_done, gn, gio->io_priority,
2468 ZIO_GANG_CHILD_FLAGS(gio), &gio->io_bookmark));
b128c09f 2469}
34dc7c2f 2470
b128c09f
BB
2471static void
2472zio_gang_tree_assemble_done(zio_t *zio)
2473{
9babb374 2474 zio_t *gio = zio->io_gang_leader;
b128c09f
BB
2475 zio_gang_node_t *gn = zio->io_private;
2476 blkptr_t *bp = zio->io_bp;
34dc7c2f 2477
9babb374 2478 ASSERT(gio == zio_unique_parent(zio));
428870ff 2479 ASSERT(zio->io_child_count == 0);
34dc7c2f 2480
b128c09f
BB
2481 if (zio->io_error)
2482 return;
34dc7c2f 2483
a6255b7f 2484 /* this ABD was created from a linear buf in zio_gang_tree_assemble */
b128c09f 2485 if (BP_SHOULD_BYTESWAP(bp))
a6255b7f 2486 byteswap_uint64_array(abd_to_buf(zio->io_abd), zio->io_size);
34dc7c2f 2487
a6255b7f 2488 ASSERT3P(abd_to_buf(zio->io_abd), ==, gn->gn_gbh);
b128c09f 2489 ASSERT(zio->io_size == SPA_GANGBLOCKSIZE);
428870ff 2490 ASSERT(gn->gn_gbh->zg_tail.zec_magic == ZEC_MAGIC);
34dc7c2f 2491
a6255b7f
DQ
2492 abd_put(zio->io_abd);
2493
1c27024e 2494 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
b128c09f
BB
2495 blkptr_t *gbp = &gn->gn_gbh->zg_blkptr[g];
2496 if (!BP_IS_GANG(gbp))
2497 continue;
9babb374 2498 zio_gang_tree_assemble(gio, gbp, &gn->gn_child[g]);
b128c09f 2499 }
34dc7c2f
BB
2500}
2501
b128c09f 2502static void
a6255b7f
DQ
2503zio_gang_tree_issue(zio_t *pio, zio_gang_node_t *gn, blkptr_t *bp, abd_t *data,
2504 uint64_t offset)
34dc7c2f 2505{
9babb374 2506 zio_t *gio = pio->io_gang_leader;
b128c09f 2507 zio_t *zio;
34dc7c2f 2508
b128c09f 2509 ASSERT(BP_IS_GANG(bp) == !!gn);
9babb374
BB
2510 ASSERT(BP_GET_CHECKSUM(bp) == BP_GET_CHECKSUM(gio->io_bp));
2511 ASSERT(BP_GET_LSIZE(bp) == BP_GET_PSIZE(bp) || gn == gio->io_gang_tree);
34dc7c2f 2512
b128c09f
BB
2513 /*
2514 * If you're a gang header, your data is in gn->gn_gbh.
2515 * If you're a gang member, your data is in 'data' and gn == NULL.
2516 */
a6255b7f 2517 zio = zio_gang_issue_func[gio->io_type](pio, bp, gn, data, offset);
34dc7c2f 2518
b128c09f 2519 if (gn != NULL) {
428870ff 2520 ASSERT(gn->gn_gbh->zg_tail.zec_magic == ZEC_MAGIC);
34dc7c2f 2521
1c27024e 2522 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
b128c09f
BB
2523 blkptr_t *gbp = &gn->gn_gbh->zg_blkptr[g];
2524 if (BP_IS_HOLE(gbp))
2525 continue;
a6255b7f
DQ
2526 zio_gang_tree_issue(zio, gn->gn_child[g], gbp, data,
2527 offset);
2528 offset += BP_GET_PSIZE(gbp);
b128c09f 2529 }
34dc7c2f
BB
2530 }
2531
9babb374 2532 if (gn == gio->io_gang_tree)
a6255b7f 2533 ASSERT3U(gio->io_size, ==, offset);
34dc7c2f 2534
b128c09f
BB
2535 if (zio != pio)
2536 zio_nowait(zio);
34dc7c2f
BB
2537}
2538
62840030 2539static zio_t *
b128c09f 2540zio_gang_assemble(zio_t *zio)
34dc7c2f 2541{
b128c09f 2542 blkptr_t *bp = zio->io_bp;
34dc7c2f 2543
9babb374
BB
2544 ASSERT(BP_IS_GANG(bp) && zio->io_gang_leader == NULL);
2545 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
2546
2547 zio->io_gang_leader = zio;
34dc7c2f 2548
b128c09f 2549 zio_gang_tree_assemble(zio, bp, &zio->io_gang_tree);
34dc7c2f 2550
62840030 2551 return (zio);
34dc7c2f
BB
2552}
2553
62840030 2554static zio_t *
b128c09f 2555zio_gang_issue(zio_t *zio)
34dc7c2f 2556{
b128c09f 2557 blkptr_t *bp = zio->io_bp;
34dc7c2f 2558
ddc751d5 2559 if (zio_wait_for_children(zio, ZIO_CHILD_GANG_BIT, ZIO_WAIT_DONE)) {
62840030 2560 return (NULL);
ddc751d5 2561 }
34dc7c2f 2562
9babb374
BB
2563 ASSERT(BP_IS_GANG(bp) && zio->io_gang_leader == zio);
2564 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
34dc7c2f 2565
b128c09f 2566 if (zio->io_child_error[ZIO_CHILD_GANG] == 0)
a6255b7f
DQ
2567 zio_gang_tree_issue(zio, zio->io_gang_tree, bp, zio->io_abd,
2568 0);
b128c09f 2569 else
9babb374 2570 zio_gang_tree_free(&zio->io_gang_tree);
34dc7c2f 2571
b128c09f 2572 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
34dc7c2f 2573
62840030 2574 return (zio);
34dc7c2f
BB
2575}
2576
2577static void
b128c09f 2578zio_write_gang_member_ready(zio_t *zio)
34dc7c2f 2579{
d164b209 2580 zio_t *pio = zio_unique_parent(zio);
34dc7c2f
BB
2581 dva_t *cdva = zio->io_bp->blk_dva;
2582 dva_t *pdva = pio->io_bp->blk_dva;
2583 uint64_t asize;
d1d7e268 2584 ASSERTV(zio_t *gio = zio->io_gang_leader);
34dc7c2f 2585
b128c09f
BB
2586 if (BP_IS_HOLE(zio->io_bp))
2587 return;
2588
2589 ASSERT(BP_IS_HOLE(&zio->io_bp_orig));
2590
2591 ASSERT(zio->io_child_type == ZIO_CHILD_GANG);
428870ff
BB
2592 ASSERT3U(zio->io_prop.zp_copies, ==, gio->io_prop.zp_copies);
2593 ASSERT3U(zio->io_prop.zp_copies, <=, BP_GET_NDVAS(zio->io_bp));
2594 ASSERT3U(pio->io_prop.zp_copies, <=, BP_GET_NDVAS(pio->io_bp));
34dc7c2f 2595 ASSERT3U(BP_GET_NDVAS(zio->io_bp), <=, BP_GET_NDVAS(pio->io_bp));
34dc7c2f
BB
2596
2597 mutex_enter(&pio->io_lock);
1c27024e 2598 for (int d = 0; d < BP_GET_NDVAS(zio->io_bp); d++) {
34dc7c2f
BB
2599 ASSERT(DVA_GET_GANG(&pdva[d]));
2600 asize = DVA_GET_ASIZE(&pdva[d]);
2601 asize += DVA_GET_ASIZE(&cdva[d]);
2602 DVA_SET_ASIZE(&pdva[d], asize);
2603 }
2604 mutex_exit(&pio->io_lock);
2605}
2606
a6255b7f
DQ
2607static void
2608zio_write_gang_done(zio_t *zio)
2609{
c955398b
BL
2610 /*
2611 * The io_abd field will be NULL for a zio with no data. The io_flags
2612 * will initially have the ZIO_FLAG_NODATA bit flag set, but we can't
2613 * check for it here as it is cleared in zio_ready.
2614 */
2615 if (zio->io_abd != NULL)
2616 abd_put(zio->io_abd);
a6255b7f
DQ
2617}
2618
62840030 2619static zio_t *
b128c09f 2620zio_write_gang_block(zio_t *pio)
34dc7c2f 2621{
b128c09f 2622 spa_t *spa = pio->io_spa;
3dfb57a3 2623 metaslab_class_t *mc = spa_normal_class(spa);
b128c09f 2624 blkptr_t *bp = pio->io_bp;
9babb374 2625 zio_t *gio = pio->io_gang_leader;
b128c09f
BB
2626 zio_t *zio;
2627 zio_gang_node_t *gn, **gnpp;
34dc7c2f 2628 zio_gbh_phys_t *gbh;
a6255b7f 2629 abd_t *gbh_abd;
b128c09f
BB
2630 uint64_t txg = pio->io_txg;
2631 uint64_t resid = pio->io_size;
2632 uint64_t lsize;
428870ff 2633 int copies = gio->io_prop.zp_copies;
b5256303 2634 int gbh_copies;
b128c09f 2635 zio_prop_t zp;
1c27024e 2636 int error;
c955398b 2637 boolean_t has_data = !(pio->io_flags & ZIO_FLAG_NODATA);
b5256303
TC
2638
2639 /*
2640 * encrypted blocks need DVA[2] free so encrypted gang headers can't
2641 * have a third copy.
2642 */
2643 gbh_copies = MIN(copies + 1, spa_max_replication(spa));
2644 if (gio->io_prop.zp_encrypt && gbh_copies >= SPA_DVAS_PER_BP)
2645 gbh_copies = SPA_DVAS_PER_BP - 1;
2646
1c27024e 2647 int flags = METASLAB_HINTBP_FAVOR | METASLAB_GANG_HEADER;
3dfb57a3
DB
2648 if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
2649 ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
c955398b 2650 ASSERT(has_data);
3dfb57a3
DB
2651
2652 flags |= METASLAB_ASYNC_ALLOC;
424fd7c3 2653 VERIFY(zfs_refcount_held(&mc->mc_alloc_slots[pio->io_allocator],
492f64e9 2654 pio));
3dfb57a3
DB
2655
2656 /*
2657 * The logical zio has already placed a reservation for
2658 * 'copies' allocation slots but gang blocks may require
2659 * additional copies. These additional copies
2660 * (i.e. gbh_copies - copies) are guaranteed to succeed
2661 * since metaslab_class_throttle_reserve() always allows
2662 * additional reservations for gang blocks.
2663 */
2664 VERIFY(metaslab_class_throttle_reserve(mc, gbh_copies - copies,
492f64e9 2665 pio->io_allocator, pio, flags));
3dfb57a3
DB
2666 }
2667
2668 error = metaslab_alloc(spa, mc, SPA_GANGBLOCKSIZE,
4e21fd06 2669 bp, gbh_copies, txg, pio == gio ? NULL : gio->io_bp, flags,
492f64e9 2670 &pio->io_alloc_list, pio, pio->io_allocator);
34dc7c2f 2671 if (error) {
3dfb57a3
DB
2672 if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
2673 ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
c955398b 2674 ASSERT(has_data);
3dfb57a3
DB
2675
2676 /*
2677 * If we failed to allocate the gang block header then
2678 * we remove any additional allocation reservations that
2679 * we placed here. The original reservation will
2680 * be removed when the logical I/O goes to the ready
2681 * stage.
2682 */
2683 metaslab_class_throttle_unreserve(mc,
492f64e9 2684 gbh_copies - copies, pio->io_allocator, pio);
3dfb57a3
DB
2685 }
2686
b128c09f 2687 pio->io_error = error;
62840030 2688 return (pio);
34dc7c2f
BB
2689 }
2690
9babb374
BB
2691 if (pio == gio) {
2692 gnpp = &gio->io_gang_tree;
b128c09f
BB
2693 } else {
2694 gnpp = pio->io_private;
2695 ASSERT(pio->io_ready == zio_write_gang_member_ready);
34dc7c2f
BB
2696 }
2697
b128c09f
BB
2698 gn = zio_gang_node_alloc(gnpp);
2699 gbh = gn->gn_gbh;
2700 bzero(gbh, SPA_GANGBLOCKSIZE);
a6255b7f 2701 gbh_abd = abd_get_from_buf(gbh, SPA_GANGBLOCKSIZE);
34dc7c2f 2702
b128c09f
BB
2703 /*
2704 * Create the gang header.
2705 */
a6255b7f
DQ
2706 zio = zio_rewrite(pio, spa, txg, bp, gbh_abd, SPA_GANGBLOCKSIZE,
2707 zio_write_gang_done, NULL, pio->io_priority,
2708 ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
34dc7c2f 2709
b128c09f
BB
2710 /*
2711 * Create and nowait the gang children.
2712 */
1c27024e 2713 for (int g = 0; resid != 0; resid -= lsize, g++) {
b128c09f
BB
2714 lsize = P2ROUNDUP(resid / (SPA_GBH_NBLKPTRS - g),
2715 SPA_MINBLOCKSIZE);
2716 ASSERT(lsize >= SPA_MINBLOCKSIZE && lsize <= resid);
2717
9babb374 2718 zp.zp_checksum = gio->io_prop.zp_checksum;
b128c09f
BB
2719 zp.zp_compress = ZIO_COMPRESS_OFF;
2720 zp.zp_type = DMU_OT_NONE;
2721 zp.zp_level = 0;
428870ff 2722 zp.zp_copies = gio->io_prop.zp_copies;
03c6040b
GW
2723 zp.zp_dedup = B_FALSE;
2724 zp.zp_dedup_verify = B_FALSE;
2725 zp.zp_nopwrite = B_FALSE;
4807c0ba
TC
2726 zp.zp_encrypt = gio->io_prop.zp_encrypt;
2727 zp.zp_byteorder = gio->io_prop.zp_byteorder;
b5256303
TC
2728 bzero(zp.zp_salt, ZIO_DATA_SALT_LEN);
2729 bzero(zp.zp_iv, ZIO_DATA_IV_LEN);
2730 bzero(zp.zp_mac, ZIO_DATA_MAC_LEN);
b128c09f 2731
1c27024e 2732 zio_t *cio = zio_write(zio, spa, txg, &gbh->zg_blkptr[g],
c955398b
BL
2733 has_data ? abd_get_offset(pio->io_abd, pio->io_size -
2734 resid) : NULL, lsize, lsize, &zp,
2735 zio_write_gang_member_ready, NULL, NULL,
a6255b7f 2736 zio_write_gang_done, &gn->gn_child[g], pio->io_priority,
3dfb57a3
DB
2737 ZIO_GANG_CHILD_FLAGS(pio), &pio->io_bookmark);
2738
2739 if (pio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
2740 ASSERT(pio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
c955398b 2741 ASSERT(has_data);
3dfb57a3
DB
2742
2743 /*
2744 * Gang children won't throttle but we should
2745 * account for their work, so reserve an allocation
2746 * slot for them here.
2747 */
2748 VERIFY(metaslab_class_throttle_reserve(mc,
492f64e9 2749 zp.zp_copies, cio->io_allocator, cio, flags));
3dfb57a3
DB
2750 }
2751 zio_nowait(cio);
b128c09f 2752 }
34dc7c2f
BB
2753
2754 /*
b128c09f 2755 * Set pio's pipeline to just wait for zio to finish.
34dc7c2f 2756 */
b128c09f
BB
2757 pio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
2758
920dd524
ED
2759 /*
2760 * We didn't allocate this bp, so make sure it doesn't get unmarked.
2761 */
2762 pio->io_flags &= ~ZIO_FLAG_FASTWRITE;
2763
b128c09f
BB
2764 zio_nowait(zio);
2765
62840030 2766 return (pio);
34dc7c2f
BB
2767}
2768
03c6040b 2769/*
3c67d83a
TH
2770 * The zio_nop_write stage in the pipeline determines if allocating a
2771 * new bp is necessary. The nopwrite feature can handle writes in
2772 * either syncing or open context (i.e. zil writes) and as a result is
2773 * mutually exclusive with dedup.
2774 *
2775 * By leveraging a cryptographically secure checksum, such as SHA256, we
2776 * can compare the checksums of the new data and the old to determine if
2777 * allocating a new block is required. Note that our requirements for
2778 * cryptographic strength are fairly weak: there can't be any accidental
2779 * hash collisions, but we don't need to be secure against intentional
2780 * (malicious) collisions. To trigger a nopwrite, you have to be able
2781 * to write the file to begin with, and triggering an incorrect (hash
2782 * collision) nopwrite is no worse than simply writing to the file.
2783 * That said, there are no known attacks against the checksum algorithms
2784 * used for nopwrite, assuming that the salt and the checksums
2785 * themselves remain secret.
03c6040b 2786 */
62840030 2787static zio_t *
03c6040b
GW
2788zio_nop_write(zio_t *zio)
2789{
2790 blkptr_t *bp = zio->io_bp;
2791 blkptr_t *bp_orig = &zio->io_bp_orig;
2792 zio_prop_t *zp = &zio->io_prop;
2793
2794 ASSERT(BP_GET_LEVEL(bp) == 0);
2795 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REWRITE));
2796 ASSERT(zp->zp_nopwrite);
2797 ASSERT(!zp->zp_dedup);
2798 ASSERT(zio->io_bp_override == NULL);
2799 ASSERT(IO_IS_ALLOCATING(zio));
2800
2801 /*
2802 * Check to see if the original bp and the new bp have matching
2803 * characteristics (i.e. same checksum, compression algorithms, etc).
2804 * If they don't then just continue with the pipeline which will
2805 * allocate a new bp.
2806 */
2807 if (BP_IS_HOLE(bp_orig) ||
3c67d83a
TH
2808 !(zio_checksum_table[BP_GET_CHECKSUM(bp)].ci_flags &
2809 ZCHECKSUM_FLAG_NOPWRITE) ||
b5256303 2810 BP_IS_ENCRYPTED(bp) || BP_IS_ENCRYPTED(bp_orig) ||
03c6040b
GW
2811 BP_GET_CHECKSUM(bp) != BP_GET_CHECKSUM(bp_orig) ||
2812 BP_GET_COMPRESS(bp) != BP_GET_COMPRESS(bp_orig) ||
2813 BP_GET_DEDUP(bp) != BP_GET_DEDUP(bp_orig) ||
2814 zp->zp_copies != BP_GET_NDVAS(bp_orig))
62840030 2815 return (zio);
03c6040b
GW
2816
2817 /*
2818 * If the checksums match then reset the pipeline so that we
2819 * avoid allocating a new bp and issuing any I/O.
2820 */
2821 if (ZIO_CHECKSUM_EQUAL(bp->blk_cksum, bp_orig->blk_cksum)) {
3c67d83a
TH
2822 ASSERT(zio_checksum_table[zp->zp_checksum].ci_flags &
2823 ZCHECKSUM_FLAG_NOPWRITE);
03c6040b
GW
2824 ASSERT3U(BP_GET_PSIZE(bp), ==, BP_GET_PSIZE(bp_orig));
2825 ASSERT3U(BP_GET_LSIZE(bp), ==, BP_GET_LSIZE(bp_orig));
2826 ASSERT(zp->zp_compress != ZIO_COMPRESS_OFF);
2827 ASSERT(bcmp(&bp->blk_prop, &bp_orig->blk_prop,
2828 sizeof (uint64_t)) == 0);
2829
2830 *bp = *bp_orig;
2831 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
2832 zio->io_flags |= ZIO_FLAG_NOPWRITE;
2833 }
2834
62840030 2835 return (zio);
03c6040b
GW
2836}
2837
34dc7c2f
BB
2838/*
2839 * ==========================================================================
428870ff 2840 * Dedup
34dc7c2f
BB
2841 * ==========================================================================
2842 */
428870ff
BB
2843static void
2844zio_ddt_child_read_done(zio_t *zio)
2845{
2846 blkptr_t *bp = zio->io_bp;
2847 ddt_entry_t *dde = zio->io_private;
2848 ddt_phys_t *ddp;
2849 zio_t *pio = zio_unique_parent(zio);
2850
2851 mutex_enter(&pio->io_lock);
2852 ddp = ddt_phys_select(dde, bp);
2853 if (zio->io_error == 0)
2854 ddt_phys_clear(ddp); /* this ddp doesn't need repair */
a6255b7f
DQ
2855
2856 if (zio->io_error == 0 && dde->dde_repair_abd == NULL)
2857 dde->dde_repair_abd = zio->io_abd;
428870ff 2858 else
a6255b7f 2859 abd_free(zio->io_abd);
428870ff
BB
2860 mutex_exit(&pio->io_lock);
2861}
2862
62840030 2863static zio_t *
428870ff
BB
2864zio_ddt_read_start(zio_t *zio)
2865{
2866 blkptr_t *bp = zio->io_bp;
2867
2868 ASSERT(BP_GET_DEDUP(bp));
2869 ASSERT(BP_GET_PSIZE(bp) == zio->io_size);
2870 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
2871
2872 if (zio->io_child_error[ZIO_CHILD_DDT]) {
2873 ddt_t *ddt = ddt_select(zio->io_spa, bp);
2874 ddt_entry_t *dde = ddt_repair_start(ddt, bp);
2875 ddt_phys_t *ddp = dde->dde_phys;
2876 ddt_phys_t *ddp_self = ddt_phys_select(dde, bp);
2877 blkptr_t blk;
2878
2879 ASSERT(zio->io_vsd == NULL);
2880 zio->io_vsd = dde;
2881
2882 if (ddp_self == NULL)
62840030 2883 return (zio);
428870ff 2884
1c27024e 2885 for (int p = 0; p < DDT_PHYS_TYPES; p++, ddp++) {
428870ff
BB
2886 if (ddp->ddp_phys_birth == 0 || ddp == ddp_self)
2887 continue;
2888 ddt_bp_create(ddt->ddt_checksum, &dde->dde_key, ddp,
2889 &blk);
2890 zio_nowait(zio_read(zio, zio->io_spa, &blk,
a6255b7f
DQ
2891 abd_alloc_for_io(zio->io_size, B_TRUE),
2892 zio->io_size, zio_ddt_child_read_done, dde,
2893 zio->io_priority, ZIO_DDT_CHILD_FLAGS(zio) |
2894 ZIO_FLAG_DONT_PROPAGATE, &zio->io_bookmark));
428870ff 2895 }
62840030 2896 return (zio);
428870ff
BB
2897 }
2898
2899 zio_nowait(zio_read(zio, zio->io_spa, bp,
a6255b7f 2900 zio->io_abd, zio->io_size, NULL, NULL, zio->io_priority,
428870ff
BB
2901 ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark));
2902
62840030 2903 return (zio);
428870ff
BB
2904}
2905
62840030 2906static zio_t *
428870ff
BB
2907zio_ddt_read_done(zio_t *zio)
2908{
2909 blkptr_t *bp = zio->io_bp;
2910
ddc751d5 2911 if (zio_wait_for_children(zio, ZIO_CHILD_DDT_BIT, ZIO_WAIT_DONE)) {
62840030 2912 return (NULL);
ddc751d5 2913 }
428870ff
BB
2914
2915 ASSERT(BP_GET_DEDUP(bp));
2916 ASSERT(BP_GET_PSIZE(bp) == zio->io_size);
2917 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
2918
2919 if (zio->io_child_error[ZIO_CHILD_DDT]) {
2920 ddt_t *ddt = ddt_select(zio->io_spa, bp);
2921 ddt_entry_t *dde = zio->io_vsd;
2922 if (ddt == NULL) {
2923 ASSERT(spa_load_state(zio->io_spa) != SPA_LOAD_NONE);
62840030 2924 return (zio);
428870ff
BB
2925 }
2926 if (dde == NULL) {
2927 zio->io_stage = ZIO_STAGE_DDT_READ_START >> 1;
2928 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_FALSE);
62840030 2929 return (NULL);
428870ff 2930 }
a6255b7f
DQ
2931 if (dde->dde_repair_abd != NULL) {
2932 abd_copy(zio->io_abd, dde->dde_repair_abd,
2933 zio->io_size);
428870ff
BB
2934 zio->io_child_error[ZIO_CHILD_DDT] = 0;
2935 }
2936 ddt_repair_done(ddt, dde);
2937 zio->io_vsd = NULL;
2938 }
2939
2940 ASSERT(zio->io_vsd == NULL);
2941
62840030 2942 return (zio);
428870ff
BB
2943}
2944
2945static boolean_t
2946zio_ddt_collision(zio_t *zio, ddt_t *ddt, ddt_entry_t *dde)
2947{
2948 spa_t *spa = zio->io_spa;
c17bcf83 2949 boolean_t do_raw = !!(zio->io_flags & ZIO_FLAG_RAW);
428870ff 2950
c17bcf83 2951 ASSERT(!(zio->io_bp_override && do_raw));
2aa34383 2952
428870ff
BB
2953 /*
2954 * Note: we compare the original data, not the transformed data,
2955 * because when zio->io_bp is an override bp, we will not have
2956 * pushed the I/O transforms. That's an important optimization
2957 * because otherwise we'd compress/encrypt all dmu_sync() data twice.
c17bcf83 2958 * However, we should never get a raw, override zio so in these
b5256303 2959 * cases we can compare the io_abd directly. This is useful because
c17bcf83
TC
2960 * it allows us to do dedup verification even if we don't have access
2961 * to the original data (for instance, if the encryption keys aren't
2962 * loaded).
428870ff 2963 */
c17bcf83 2964
1c27024e 2965 for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++) {
428870ff
BB
2966 zio_t *lio = dde->dde_lead_zio[p];
2967
c17bcf83
TC
2968 if (lio != NULL && do_raw) {
2969 return (lio->io_size != zio->io_size ||
a6255b7f 2970 abd_cmp(zio->io_abd, lio->io_abd) != 0);
c17bcf83 2971 } else if (lio != NULL) {
428870ff 2972 return (lio->io_orig_size != zio->io_orig_size ||
a6255b7f 2973 abd_cmp(zio->io_orig_abd, lio->io_orig_abd) != 0);
428870ff
BB
2974 }
2975 }
2976
1c27024e 2977 for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++) {
428870ff
BB
2978 ddt_phys_t *ddp = &dde->dde_phys[p];
2979
c17bcf83
TC
2980 if (ddp->ddp_phys_birth != 0 && do_raw) {
2981 blkptr_t blk = *zio->io_bp;
2982 uint64_t psize;
a6255b7f 2983 abd_t *tmpabd;
c17bcf83
TC
2984 int error;
2985
2986 ddt_bp_fill(ddp, &blk, ddp->ddp_phys_birth);
2987 psize = BP_GET_PSIZE(&blk);
2988
2989 if (psize != zio->io_size)
2990 return (B_TRUE);
2991
2992 ddt_exit(ddt);
2993
a6255b7f 2994 tmpabd = abd_alloc_for_io(psize, B_TRUE);
c17bcf83 2995
a6255b7f 2996 error = zio_wait(zio_read(NULL, spa, &blk, tmpabd,
c17bcf83
TC
2997 psize, NULL, NULL, ZIO_PRIORITY_SYNC_READ,
2998 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE |
2999 ZIO_FLAG_RAW, &zio->io_bookmark));
3000
3001 if (error == 0) {
a6255b7f 3002 if (abd_cmp(tmpabd, zio->io_abd) != 0)
c17bcf83
TC
3003 error = SET_ERROR(ENOENT);
3004 }
3005
a6255b7f 3006 abd_free(tmpabd);
c17bcf83
TC
3007 ddt_enter(ddt);
3008 return (error != 0);
3009 } else if (ddp->ddp_phys_birth != 0) {
428870ff 3010 arc_buf_t *abuf = NULL;
2a432414 3011 arc_flags_t aflags = ARC_FLAG_WAIT;
428870ff
BB
3012 blkptr_t blk = *zio->io_bp;
3013 int error;
3014
3015 ddt_bp_fill(ddp, &blk, ddp->ddp_phys_birth);
3016
c17bcf83
TC
3017 if (BP_GET_LSIZE(&blk) != zio->io_orig_size)
3018 return (B_TRUE);
3019
428870ff
BB
3020 ddt_exit(ddt);
3021
294f6806 3022 error = arc_read(NULL, spa, &blk,
428870ff
BB
3023 arc_getbuf_func, &abuf, ZIO_PRIORITY_SYNC_READ,
3024 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE,
3025 &aflags, &zio->io_bookmark);
3026
3027 if (error == 0) {
a6255b7f 3028 if (abd_cmp_buf(zio->io_orig_abd, abuf->b_data,
428870ff 3029 zio->io_orig_size) != 0)
c17bcf83 3030 error = SET_ERROR(ENOENT);
d3c2ae1c 3031 arc_buf_destroy(abuf, &abuf);
428870ff
BB
3032 }
3033
3034 ddt_enter(ddt);
3035 return (error != 0);
3036 }
3037 }
3038
3039 return (B_FALSE);
3040}
3041
3042static void
3043zio_ddt_child_write_ready(zio_t *zio)
3044{
3045 int p = zio->io_prop.zp_copies;
3046 ddt_t *ddt = ddt_select(zio->io_spa, zio->io_bp);
3047 ddt_entry_t *dde = zio->io_private;
3048 ddt_phys_t *ddp = &dde->dde_phys[p];
3049 zio_t *pio;
3050
3051 if (zio->io_error)
3052 return;
3053
3054 ddt_enter(ddt);
3055
3056 ASSERT(dde->dde_lead_zio[p] == zio);
3057
3058 ddt_phys_fill(ddp, zio->io_bp);
3059
1c27024e 3060 zio_link_t *zl = NULL;
3dfb57a3 3061 while ((pio = zio_walk_parents(zio, &zl)) != NULL)
428870ff
BB
3062 ddt_bp_fill(ddp, pio->io_bp, zio->io_txg);
3063
3064 ddt_exit(ddt);
3065}
3066
3067static void
3068zio_ddt_child_write_done(zio_t *zio)
3069{
3070 int p = zio->io_prop.zp_copies;
3071 ddt_t *ddt = ddt_select(zio->io_spa, zio->io_bp);
3072 ddt_entry_t *dde = zio->io_private;
3073 ddt_phys_t *ddp = &dde->dde_phys[p];
3074
3075 ddt_enter(ddt);
3076
3077 ASSERT(ddp->ddp_refcnt == 0);
3078 ASSERT(dde->dde_lead_zio[p] == zio);
3079 dde->dde_lead_zio[p] = NULL;
3080
3081 if (zio->io_error == 0) {
3dfb57a3
DB
3082 zio_link_t *zl = NULL;
3083 while (zio_walk_parents(zio, &zl) != NULL)
428870ff
BB
3084 ddt_phys_addref(ddp);
3085 } else {
3086 ddt_phys_clear(ddp);
3087 }
3088
3089 ddt_exit(ddt);
3090}
3091
3092static void
3093zio_ddt_ditto_write_done(zio_t *zio)
3094{
3095 int p = DDT_PHYS_DITTO;
1c27024e 3096 ASSERTV(zio_prop_t *zp = &zio->io_prop);
428870ff
BB
3097 blkptr_t *bp = zio->io_bp;
3098 ddt_t *ddt = ddt_select(zio->io_spa, bp);
3099 ddt_entry_t *dde = zio->io_private;
3100 ddt_phys_t *ddp = &dde->dde_phys[p];
3101 ddt_key_t *ddk = &dde->dde_key;
3102
3103 ddt_enter(ddt);
3104
3105 ASSERT(ddp->ddp_refcnt == 0);
3106 ASSERT(dde->dde_lead_zio[p] == zio);
3107 dde->dde_lead_zio[p] = NULL;
3108
3109 if (zio->io_error == 0) {
3110 ASSERT(ZIO_CHECKSUM_EQUAL(bp->blk_cksum, ddk->ddk_cksum));
3111 ASSERT(zp->zp_copies < SPA_DVAS_PER_BP);
3112 ASSERT(zp->zp_copies == BP_GET_NDVAS(bp) - BP_IS_GANG(bp));
3113 if (ddp->ddp_phys_birth != 0)
3114 ddt_phys_free(ddt, ddk, ddp, zio->io_txg);
3115 ddt_phys_fill(ddp, bp);
3116 }
3117
3118 ddt_exit(ddt);
3119}
3120
62840030 3121static zio_t *
428870ff
BB
3122zio_ddt_write(zio_t *zio)
3123{
3124 spa_t *spa = zio->io_spa;
3125 blkptr_t *bp = zio->io_bp;
3126 uint64_t txg = zio->io_txg;
3127 zio_prop_t *zp = &zio->io_prop;
3128 int p = zp->zp_copies;
3129 int ditto_copies;
3130 zio_t *cio = NULL;
3131 zio_t *dio = NULL;
3132 ddt_t *ddt = ddt_select(spa, bp);
3133 ddt_entry_t *dde;
3134 ddt_phys_t *ddp;
3135
3136 ASSERT(BP_GET_DEDUP(bp));
3137 ASSERT(BP_GET_CHECKSUM(bp) == zp->zp_checksum);
3138 ASSERT(BP_IS_HOLE(bp) || zio->io_bp_override);
c17bcf83 3139 ASSERT(!(zio->io_bp_override && (zio->io_flags & ZIO_FLAG_RAW)));
428870ff
BB
3140
3141 ddt_enter(ddt);
3142 dde = ddt_lookup(ddt, bp, B_TRUE);
3143 ddp = &dde->dde_phys[p];
3144
3145 if (zp->zp_dedup_verify && zio_ddt_collision(zio, ddt, dde)) {
3146 /*
3147 * If we're using a weak checksum, upgrade to a strong checksum
3148 * and try again. If we're already using a strong checksum,
3149 * we can't resolve it, so just convert to an ordinary write.
3150 * (And automatically e-mail a paper to Nature?)
3151 */
3c67d83a
TH
3152 if (!(zio_checksum_table[zp->zp_checksum].ci_flags &
3153 ZCHECKSUM_FLAG_DEDUP)) {
428870ff
BB
3154 zp->zp_checksum = spa_dedup_checksum(spa);
3155 zio_pop_transforms(zio);
3156 zio->io_stage = ZIO_STAGE_OPEN;
3157 BP_ZERO(bp);
3158 } else {
03c6040b 3159 zp->zp_dedup = B_FALSE;
428870ff
BB
3160 }
3161 zio->io_pipeline = ZIO_WRITE_PIPELINE;
3162 ddt_exit(ddt);
62840030 3163 return (zio);
428870ff
BB
3164 }
3165
3166 ditto_copies = ddt_ditto_copies_needed(ddt, dde, ddp);
3167 ASSERT(ditto_copies < SPA_DVAS_PER_BP);
3168
3169 if (ditto_copies > ddt_ditto_copies_present(dde) &&
3170 dde->dde_lead_zio[DDT_PHYS_DITTO] == NULL) {
3171 zio_prop_t czp = *zp;
3172
3173 czp.zp_copies = ditto_copies;
3174
3175 /*
3176 * If we arrived here with an override bp, we won't have run
3177 * the transform stack, so we won't have the data we need to
3178 * generate a child i/o. So, toss the override bp and restart.
3179 * This is safe, because using the override bp is just an
3180 * optimization; and it's rare, so the cost doesn't matter.
3181 */
3182 if (zio->io_bp_override) {
3183 zio_pop_transforms(zio);
3184 zio->io_stage = ZIO_STAGE_OPEN;
3185 zio->io_pipeline = ZIO_WRITE_PIPELINE;
3186 zio->io_bp_override = NULL;
3187 BP_ZERO(bp);
3188 ddt_exit(ddt);
62840030 3189 return (zio);
428870ff
BB
3190 }
3191
a6255b7f 3192 dio = zio_write(zio, spa, txg, bp, zio->io_orig_abd,
2aa34383 3193 zio->io_orig_size, zio->io_orig_size, &czp, NULL, NULL,
bc77ba73 3194 NULL, zio_ddt_ditto_write_done, dde, zio->io_priority,
428870ff
BB
3195 ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark);
3196
a6255b7f 3197 zio_push_transform(dio, zio->io_abd, zio->io_size, 0, NULL);
428870ff
BB
3198 dde->dde_lead_zio[DDT_PHYS_DITTO] = dio;
3199 }
3200
3201 if (ddp->ddp_phys_birth != 0 || dde->dde_lead_zio[p] != NULL) {
3202 if (ddp->ddp_phys_birth != 0)
3203 ddt_bp_fill(ddp, bp, txg);
3204 if (dde->dde_lead_zio[p] != NULL)
3205 zio_add_child(zio, dde->dde_lead_zio[p]);
3206 else
3207 ddt_phys_addref(ddp);
3208 } else if (zio->io_bp_override) {
3209 ASSERT(bp->blk_birth == txg);
3210 ASSERT(BP_EQUAL(bp, zio->io_bp_override));
3211 ddt_phys_fill(ddp, bp);
3212 ddt_phys_addref(ddp);
3213 } else {
a6255b7f 3214 cio = zio_write(zio, spa, txg, bp, zio->io_orig_abd,
2aa34383 3215 zio->io_orig_size, zio->io_orig_size, zp,
bc77ba73 3216 zio_ddt_child_write_ready, NULL, NULL,
428870ff
BB
3217 zio_ddt_child_write_done, dde, zio->io_priority,
3218 ZIO_DDT_CHILD_FLAGS(zio), &zio->io_bookmark);
3219
a6255b7f 3220 zio_push_transform(cio, zio->io_abd, zio->io_size, 0, NULL);
428870ff
BB
3221 dde->dde_lead_zio[p] = cio;
3222 }
3223
3224 ddt_exit(ddt);
3225
3226 if (cio)
3227 zio_nowait(cio);
3228 if (dio)
3229 zio_nowait(dio);
3230
62840030 3231 return (zio);
428870ff
BB
3232}
3233
3234ddt_entry_t *freedde; /* for debugging */
b128c09f 3235
62840030 3236static zio_t *
428870ff
BB
3237zio_ddt_free(zio_t *zio)
3238{
3239 spa_t *spa = zio->io_spa;
3240 blkptr_t *bp = zio->io_bp;
3241 ddt_t *ddt = ddt_select(spa, bp);
3242 ddt_entry_t *dde;
3243 ddt_phys_t *ddp;
3244
3245 ASSERT(BP_GET_DEDUP(bp));
3246 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
3247
3248 ddt_enter(ddt);
3249 freedde = dde = ddt_lookup(ddt, bp, B_TRUE);
5dc6af0e
BB
3250 if (dde) {
3251 ddp = ddt_phys_select(dde, bp);
3252 if (ddp)
3253 ddt_phys_decref(ddp);
3254 }
428870ff
BB
3255 ddt_exit(ddt);
3256
62840030 3257 return (zio);
428870ff
BB
3258}
3259
3260/*
3261 * ==========================================================================
3262 * Allocate and free blocks
3263 * ==========================================================================
3264 */
3dfb57a3
DB
3265
3266static zio_t *
492f64e9 3267zio_io_to_allocate(spa_t *spa, int allocator)
3dfb57a3
DB
3268{
3269 zio_t *zio;
3270
492f64e9 3271 ASSERT(MUTEX_HELD(&spa->spa_alloc_locks[allocator]));
3dfb57a3 3272
492f64e9 3273 zio = avl_first(&spa->spa_alloc_trees[allocator]);
3dfb57a3
DB
3274 if (zio == NULL)
3275 return (NULL);
3276
3277 ASSERT(IO_IS_ALLOCATING(zio));
3278
3279 /*
3280 * Try to place a reservation for this zio. If we're unable to
3281 * reserve then we throttle.
3282 */
492f64e9 3283 ASSERT3U(zio->io_allocator, ==, allocator);
cc99f275 3284 if (!metaslab_class_throttle_reserve(zio->io_metaslab_class,
492f64e9 3285 zio->io_prop.zp_copies, zio->io_allocator, zio, 0)) {
3dfb57a3
DB
3286 return (NULL);
3287 }
3288
492f64e9 3289 avl_remove(&spa->spa_alloc_trees[allocator], zio);
3dfb57a3
DB
3290 ASSERT3U(zio->io_stage, <, ZIO_STAGE_DVA_ALLOCATE);
3291
3292 return (zio);
3293}
3294
62840030 3295static zio_t *
3dfb57a3
DB
3296zio_dva_throttle(zio_t *zio)
3297{
3298 spa_t *spa = zio->io_spa;
3299 zio_t *nio;
cc99f275
DB
3300 metaslab_class_t *mc;
3301
3302 /* locate an appropriate allocation class */
3303 mc = spa_preferred_class(spa, zio->io_size, zio->io_prop.zp_type,
3304 zio->io_prop.zp_level, zio->io_prop.zp_zpl_smallblk);
3dfb57a3
DB
3305
3306 if (zio->io_priority == ZIO_PRIORITY_SYNC_WRITE ||
cc99f275 3307 !mc->mc_alloc_throttle_enabled ||
3dfb57a3
DB
3308 zio->io_child_type == ZIO_CHILD_GANG ||
3309 zio->io_flags & ZIO_FLAG_NODATA) {
62840030 3310 return (zio);
3dfb57a3
DB
3311 }
3312
3313 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
3314
3315 ASSERT3U(zio->io_queued_timestamp, >, 0);
3316 ASSERT(zio->io_stage == ZIO_STAGE_DVA_THROTTLE);
3317
492f64e9
PD
3318 zbookmark_phys_t *bm = &zio->io_bookmark;
3319 /*
3320 * We want to try to use as many allocators as possible to help improve
3321 * performance, but we also want logically adjacent IOs to be physically
3322 * adjacent to improve sequential read performance. We chunk each object
3323 * into 2^20 block regions, and then hash based on the objset, object,
3324 * level, and region to accomplish both of these goals.
3325 */
3326 zio->io_allocator = cityhash4(bm->zb_objset, bm->zb_object,
3327 bm->zb_level, bm->zb_blkid >> 20) % spa->spa_alloc_count;
3328 mutex_enter(&spa->spa_alloc_locks[zio->io_allocator]);
3dfb57a3 3329 ASSERT(zio->io_type == ZIO_TYPE_WRITE);
cc99f275 3330 zio->io_metaslab_class = mc;
492f64e9 3331 avl_add(&spa->spa_alloc_trees[zio->io_allocator], zio);
cc99f275 3332 nio = zio_io_to_allocate(spa, zio->io_allocator);
492f64e9 3333 mutex_exit(&spa->spa_alloc_locks[zio->io_allocator]);
62840030 3334 return (nio);
3dfb57a3
DB
3335}
3336
cc99f275 3337static void
492f64e9 3338zio_allocate_dispatch(spa_t *spa, int allocator)
3dfb57a3
DB
3339{
3340 zio_t *zio;
3341
492f64e9
PD
3342 mutex_enter(&spa->spa_alloc_locks[allocator]);
3343 zio = zio_io_to_allocate(spa, allocator);
3344 mutex_exit(&spa->spa_alloc_locks[allocator]);
3dfb57a3
DB
3345 if (zio == NULL)
3346 return;
3347
3348 ASSERT3U(zio->io_stage, ==, ZIO_STAGE_DVA_THROTTLE);
3349 ASSERT0(zio->io_error);
3350 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE, B_TRUE);
3351}
3352
62840030 3353static zio_t *
34dc7c2f
BB
3354zio_dva_allocate(zio_t *zio)
3355{
3356 spa_t *spa = zio->io_spa;
cc99f275 3357 metaslab_class_t *mc;
34dc7c2f
BB
3358 blkptr_t *bp = zio->io_bp;
3359 int error;
6d974228 3360 int flags = 0;
34dc7c2f 3361
9babb374
BB
3362 if (zio->io_gang_leader == NULL) {
3363 ASSERT(zio->io_child_type > ZIO_CHILD_GANG);
3364 zio->io_gang_leader = zio;
3365 }
3366
34dc7c2f 3367 ASSERT(BP_IS_HOLE(bp));
c99c9001 3368 ASSERT0(BP_GET_NDVAS(bp));
428870ff
BB
3369 ASSERT3U(zio->io_prop.zp_copies, >, 0);
3370 ASSERT3U(zio->io_prop.zp_copies, <=, spa_max_replication(spa));
34dc7c2f
BB
3371 ASSERT3U(zio->io_size, ==, BP_GET_PSIZE(bp));
3372
920dd524 3373 flags |= (zio->io_flags & ZIO_FLAG_FASTWRITE) ? METASLAB_FASTWRITE : 0;
3dfb57a3
DB
3374 if (zio->io_flags & ZIO_FLAG_NODATA)
3375 flags |= METASLAB_DONT_THROTTLE;
3376 if (zio->io_flags & ZIO_FLAG_GANG_CHILD)
3377 flags |= METASLAB_GANG_CHILD;
3378 if (zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE)
3379 flags |= METASLAB_ASYNC_ALLOC;
3380
cc99f275
DB
3381 /*
3382 * if not already chosen, locate an appropriate allocation class
3383 */
3384 mc = zio->io_metaslab_class;
3385 if (mc == NULL) {
3386 mc = spa_preferred_class(spa, zio->io_size,
3387 zio->io_prop.zp_type, zio->io_prop.zp_level,
3388 zio->io_prop.zp_zpl_smallblk);
3389 zio->io_metaslab_class = mc;
3390 }
3391
b128c09f 3392 error = metaslab_alloc(spa, mc, zio->io_size, bp,
4e21fd06 3393 zio->io_prop.zp_copies, zio->io_txg, NULL, flags,
492f64e9 3394 &zio->io_alloc_list, zio, zio->io_allocator);
34dc7c2f 3395
cc99f275
DB
3396 /*
3397 * Fallback to normal class when an alloc class is full
3398 */
3399 if (error == ENOSPC && mc != spa_normal_class(spa)) {
3400 /*
3401 * If throttling, transfer reservation over to normal class.
3402 * The io_allocator slot can remain the same even though we
3403 * are switching classes.
3404 */
3405 if (mc->mc_alloc_throttle_enabled &&
3406 (zio->io_flags & ZIO_FLAG_IO_ALLOCATING)) {
3407 metaslab_class_throttle_unreserve(mc,
3408 zio->io_prop.zp_copies, zio->io_allocator, zio);
3409 zio->io_flags &= ~ZIO_FLAG_IO_ALLOCATING;
3410
3411 mc = spa_normal_class(spa);
3412 VERIFY(metaslab_class_throttle_reserve(mc,
3413 zio->io_prop.zp_copies, zio->io_allocator, zio,
3414 flags | METASLAB_MUST_RESERVE));
3415 } else {
3416 mc = spa_normal_class(spa);
3417 }
3418 zio->io_metaslab_class = mc;
3419
3420 error = metaslab_alloc(spa, mc, zio->io_size, bp,
3421 zio->io_prop.zp_copies, zio->io_txg, NULL, flags,
3422 &zio->io_alloc_list, zio, zio->io_allocator);
3423 }
3424
3dfb57a3 3425 if (error != 0) {
964c2d69 3426 zfs_dbgmsg("%s: metaslab allocation failure: zio %p, "
6d974228
GW
3427 "size %llu, error %d", spa_name(spa), zio, zio->io_size,
3428 error);
b128c09f
BB
3429 if (error == ENOSPC && zio->io_size > SPA_MINBLOCKSIZE)
3430 return (zio_write_gang_block(zio));
34dc7c2f
BB
3431 zio->io_error = error;
3432 }
3433
62840030 3434 return (zio);
34dc7c2f
BB
3435}
3436
62840030 3437static zio_t *
34dc7c2f
BB
3438zio_dva_free(zio_t *zio)
3439{
b128c09f 3440 metaslab_free(zio->io_spa, zio->io_bp, zio->io_txg, B_FALSE);
34dc7c2f 3441
62840030 3442 return (zio);
34dc7c2f
BB
3443}
3444
62840030 3445static zio_t *
34dc7c2f
BB
3446zio_dva_claim(zio_t *zio)
3447{
b128c09f
BB
3448 int error;
3449
3450 error = metaslab_claim(zio->io_spa, zio->io_bp, zio->io_txg);
3451 if (error)
3452 zio->io_error = error;
34dc7c2f 3453
62840030 3454 return (zio);
34dc7c2f
BB
3455}
3456
b128c09f
BB
3457/*
3458 * Undo an allocation. This is used by zio_done() when an I/O fails
3459 * and we want to give back the block we just allocated.
3460 * This handles both normal blocks and gang blocks.
3461 */
3462static void
3463zio_dva_unallocate(zio_t *zio, zio_gang_node_t *gn, blkptr_t *bp)
3464{
b128c09f 3465 ASSERT(bp->blk_birth == zio->io_txg || BP_IS_HOLE(bp));
428870ff 3466 ASSERT(zio->io_bp_override == NULL);
b128c09f
BB
3467
3468 if (!BP_IS_HOLE(bp))
428870ff 3469 metaslab_free(zio->io_spa, bp, bp->blk_birth, B_TRUE);
b128c09f
BB
3470
3471 if (gn != NULL) {
1c27024e 3472 for (int g = 0; g < SPA_GBH_NBLKPTRS; g++) {
b128c09f
BB
3473 zio_dva_unallocate(zio, gn->gn_child[g],
3474 &gn->gn_gbh->zg_blkptr[g]);
3475 }
3476 }
3477}
3478
3479/*
3480 * Try to allocate an intent log block. Return 0 on success, errno on failure.
3481 */
3482int
b5256303
TC
3483zio_alloc_zil(spa_t *spa, objset_t *os, uint64_t txg, blkptr_t *new_bp,
3484 uint64_t size, boolean_t *slog)
b128c09f 3485{
428870ff 3486 int error = 1;
4e21fd06 3487 zio_alloc_list_t io_alloc_list;
b128c09f 3488
428870ff
BB
3489 ASSERT(txg > spa_syncing_txg(spa));
3490
4e21fd06 3491 metaslab_trace_init(&io_alloc_list);
cc99f275
DB
3492
3493 /*
3494 * Block pointer fields are useful to metaslabs for stats and debugging.
3495 * Fill in the obvious ones before calling into metaslab_alloc().
3496 */
3497 BP_SET_TYPE(new_bp, DMU_OT_INTENT_LOG);
3498 BP_SET_PSIZE(new_bp, size);
3499 BP_SET_LEVEL(new_bp, 0);
3500
492f64e9
PD
3501 /*
3502 * When allocating a zil block, we don't have information about
3503 * the final destination of the block except the objset it's part
3504 * of, so we just hash the objset ID to pick the allocator to get
3505 * some parallelism.
3506 */
1b7c1e5c 3507 error = metaslab_alloc(spa, spa_log_class(spa), size, new_bp, 1,
492f64e9
PD
3508 txg, NULL, METASLAB_FASTWRITE, &io_alloc_list, NULL,
3509 cityhash4(0, 0, 0, os->os_dsl_dataset->ds_object) %
3510 spa->spa_alloc_count);
1b7c1e5c
GDN
3511 if (error == 0) {
3512 *slog = TRUE;
3513 } else {
428870ff 3514 error = metaslab_alloc(spa, spa_normal_class(spa), size,
4e21fd06 3515 new_bp, 1, txg, NULL, METASLAB_FASTWRITE,
492f64e9
PD
3516 &io_alloc_list, NULL, cityhash4(0, 0, 0,
3517 os->os_dsl_dataset->ds_object) % spa->spa_alloc_count);
1b7c1e5c
GDN
3518 if (error == 0)
3519 *slog = FALSE;
ebf8e3a2 3520 }
4e21fd06 3521 metaslab_trace_fini(&io_alloc_list);
b128c09f
BB
3522
3523 if (error == 0) {
3524 BP_SET_LSIZE(new_bp, size);
3525 BP_SET_PSIZE(new_bp, size);
3526 BP_SET_COMPRESS(new_bp, ZIO_COMPRESS_OFF);
428870ff
BB
3527 BP_SET_CHECKSUM(new_bp,
3528 spa_version(spa) >= SPA_VERSION_SLIM_ZIL
3529 ? ZIO_CHECKSUM_ZILOG2 : ZIO_CHECKSUM_ZILOG);
b128c09f
BB
3530 BP_SET_TYPE(new_bp, DMU_OT_INTENT_LOG);
3531 BP_SET_LEVEL(new_bp, 0);
428870ff 3532 BP_SET_DEDUP(new_bp, 0);
b128c09f 3533 BP_SET_BYTEORDER(new_bp, ZFS_HOST_BYTEORDER);
b5256303
TC
3534
3535 /*
3536 * encrypted blocks will require an IV and salt. We generate
3537 * these now since we will not be rewriting the bp at
3538 * rewrite time.
3539 */
3540 if (os->os_encrypted) {
3541 uint8_t iv[ZIO_DATA_IV_LEN];
3542 uint8_t salt[ZIO_DATA_SALT_LEN];
3543
3544 BP_SET_CRYPT(new_bp, B_TRUE);
3545 VERIFY0(spa_crypt_get_salt(spa,
3546 dmu_objset_id(os), salt));
3547 VERIFY0(zio_crypt_generate_iv(iv));
3548
3549 zio_crypt_encode_params_bp(new_bp, salt, iv);
3550 }
1ce23dca
PS
3551 } else {
3552 zfs_dbgmsg("%s: zil block allocation failure: "
3553 "size %llu, error %d", spa_name(spa), size, error);
b128c09f
BB
3554 }
3555
3556 return (error);
3557}
3558
34dc7c2f
BB
3559/*
3560 * ==========================================================================
3561 * Read and write to physical devices
3562 * ==========================================================================
3563 */
98b25418
GW
3564
3565
3566/*
3567 * Issue an I/O to the underlying vdev. Typically the issue pipeline
3568 * stops after this stage and will resume upon I/O completion.
3569 * However, there are instances where the vdev layer may need to
3570 * continue the pipeline when an I/O was not issued. Since the I/O
3571 * that was sent to the vdev layer might be different than the one
3572 * currently active in the pipeline (see vdev_queue_io()), we explicitly
3573 * force the underlying vdev layers to call either zio_execute() or
3574 * zio_interrupt() to ensure that the pipeline continues with the correct I/O.
3575 */
62840030 3576static zio_t *
34dc7c2f
BB
3577zio_vdev_io_start(zio_t *zio)
3578{
3579 vdev_t *vd = zio->io_vd;
34dc7c2f
BB
3580 uint64_t align;
3581 spa_t *spa = zio->io_spa;
3582
193a37cb
TH
3583 zio->io_delay = 0;
3584
b128c09f
BB
3585 ASSERT(zio->io_error == 0);
3586 ASSERT(zio->io_child_error[ZIO_CHILD_VDEV] == 0);
34dc7c2f 3587
b128c09f
BB
3588 if (vd == NULL) {
3589 if (!(zio->io_flags & ZIO_FLAG_CONFIG_WRITER))
3590 spa_config_enter(spa, SCL_ZIO, zio, RW_READER);
34dc7c2f 3591
b128c09f
BB
3592 /*
3593 * The mirror_ops handle multiple DVAs in a single BP.
3594 */
98b25418 3595 vdev_mirror_ops.vdev_op_io_start(zio);
62840030 3596 return (NULL);
34dc7c2f
BB
3597 }
3598
3dfb57a3 3599 ASSERT3P(zio->io_logical, !=, zio);
6cb8e530
PZ
3600 if (zio->io_type == ZIO_TYPE_WRITE) {
3601 ASSERT(spa->spa_trust_config);
3602
a1d477c2
MA
3603 /*
3604 * Note: the code can handle other kinds of writes,
3605 * but we don't expect them.
3606 */
6cb8e530
PZ
3607 if (zio->io_vd->vdev_removing) {
3608 ASSERT(zio->io_flags &
3609 (ZIO_FLAG_PHYSICAL | ZIO_FLAG_SELF_HEAL |
3610 ZIO_FLAG_RESILVER | ZIO_FLAG_INDUCE_DAMAGE));
3611 }
a1d477c2 3612 }
3dfb57a3 3613
b128c09f
BB
3614 align = 1ULL << vd->vdev_top->vdev_ashift;
3615
b02fe35d
AR
3616 if (!(zio->io_flags & ZIO_FLAG_PHYSICAL) &&
3617 P2PHASE(zio->io_size, align) != 0) {
3618 /* Transform logical writes to be a full physical block size. */
34dc7c2f 3619 uint64_t asize = P2ROUNDUP(zio->io_size, align);
a6255b7f 3620 abd_t *abuf = abd_alloc_sametype(zio->io_abd, asize);
178e73b3 3621 ASSERT(vd == vd->vdev_top);
34dc7c2f 3622 if (zio->io_type == ZIO_TYPE_WRITE) {
a6255b7f
DQ
3623 abd_copy(abuf, zio->io_abd, zio->io_size);
3624 abd_zero_off(abuf, zio->io_size, asize - zio->io_size);
34dc7c2f 3625 }
b128c09f 3626 zio_push_transform(zio, abuf, asize, asize, zio_subblock);
34dc7c2f
BB
3627 }
3628
b02fe35d
AR
3629 /*
3630 * If this is not a physical io, make sure that it is properly aligned
3631 * before proceeding.
3632 */
3633 if (!(zio->io_flags & ZIO_FLAG_PHYSICAL)) {
3634 ASSERT0(P2PHASE(zio->io_offset, align));
3635 ASSERT0(P2PHASE(zio->io_size, align));
3636 } else {
3637 /*
3638 * For physical writes, we allow 512b aligned writes and assume
3639 * the device will perform a read-modify-write as necessary.
3640 */
3641 ASSERT0(P2PHASE(zio->io_offset, SPA_MINBLOCKSIZE));
3642 ASSERT0(P2PHASE(zio->io_size, SPA_MINBLOCKSIZE));
3643 }
3644
572e2857 3645 VERIFY(zio->io_type != ZIO_TYPE_WRITE || spa_writeable(spa));
fb5f0bc8
BB
3646
3647 /*
3648 * If this is a repair I/O, and there's no self-healing involved --
3649 * that is, we're just resilvering what we expect to resilver --
3650 * then don't do the I/O unless zio's txg is actually in vd's DTL.
9e052db4
MA
3651 * This prevents spurious resilvering.
3652 *
3653 * There are a few ways that we can end up creating these spurious
3654 * resilver i/os:
3655 *
3656 * 1. A resilver i/o will be issued if any DVA in the BP has a
3657 * dirty DTL. The mirror code will issue resilver writes to
3658 * each DVA, including the one(s) that are not on vdevs with dirty
3659 * DTLs.
3660 *
3661 * 2. With nested replication, which happens when we have a
3662 * "replacing" or "spare" vdev that's a child of a mirror or raidz.
3663 * For example, given mirror(replacing(A+B), C), it's likely that
3664 * only A is out of date (it's the new device). In this case, we'll
3665 * read from C, then use the data to resilver A+B -- but we don't
3666 * actually want to resilver B, just A. The top-level mirror has no
3667 * way to know this, so instead we just discard unnecessary repairs
3668 * as we work our way down the vdev tree.
3669 *
3670 * 3. ZTEST also creates mirrors of mirrors, mirrors of raidz, etc.
3671 * The same logic applies to any form of nested replication: ditto
3672 * + mirror, RAID-Z + replacing, etc.
3673 *
3674 * However, indirect vdevs point off to other vdevs which may have
3675 * DTL's, so we never bypass them. The child i/os on concrete vdevs
3676 * will be properly bypassed instead.
fb5f0bc8
BB
3677 */
3678 if ((zio->io_flags & ZIO_FLAG_IO_REPAIR) &&
3679 !(zio->io_flags & ZIO_FLAG_SELF_HEAL) &&
3680 zio->io_txg != 0 && /* not a delegated i/o */
9e052db4 3681 vd->vdev_ops != &vdev_indirect_ops &&
fb5f0bc8
BB
3682 !vdev_dtl_contains(vd, DTL_PARTIAL, zio->io_txg, 1)) {
3683 ASSERT(zio->io_type == ZIO_TYPE_WRITE);
fb5f0bc8 3684 zio_vdev_io_bypass(zio);
62840030 3685 return (zio);
fb5f0bc8 3686 }
34dc7c2f 3687
b128c09f
BB
3688 if (vd->vdev_ops->vdev_op_leaf &&
3689 (zio->io_type == ZIO_TYPE_READ || zio->io_type == ZIO_TYPE_WRITE)) {
3690
b0bc7a84 3691 if (zio->io_type == ZIO_TYPE_READ && vdev_cache_read(zio))
62840030 3692 return (zio);
b128c09f
BB
3693
3694 if ((zio = vdev_queue_io(zio)) == NULL)
62840030 3695 return (NULL);
b128c09f
BB
3696
3697 if (!vdev_accessible(vd, zio)) {
2e528b49 3698 zio->io_error = SET_ERROR(ENXIO);
b128c09f 3699 zio_interrupt(zio);
62840030 3700 return (NULL);
b128c09f 3701 }
67103816 3702 zio->io_delay = gethrtime();
b128c09f
BB
3703 }
3704
98b25418 3705 vd->vdev_ops->vdev_op_io_start(zio);
62840030 3706 return (NULL);
34dc7c2f
BB
3707}
3708
62840030 3709static zio_t *
34dc7c2f
BB
3710zio_vdev_io_done(zio_t *zio)
3711{
b128c09f
BB
3712 vdev_t *vd = zio->io_vd;
3713 vdev_ops_t *ops = vd ? vd->vdev_ops : &vdev_mirror_ops;
3714 boolean_t unexpected_error = B_FALSE;
34dc7c2f 3715
ddc751d5 3716 if (zio_wait_for_children(zio, ZIO_CHILD_VDEV_BIT, ZIO_WAIT_DONE)) {
62840030 3717 return (NULL);
ddc751d5 3718 }
34dc7c2f 3719
b128c09f
BB
3720 ASSERT(zio->io_type == ZIO_TYPE_READ || zio->io_type == ZIO_TYPE_WRITE);
3721
193a37cb
TH
3722 if (zio->io_delay)
3723 zio->io_delay = gethrtime() - zio->io_delay;
3724
b128c09f
BB
3725 if (vd != NULL && vd->vdev_ops->vdev_op_leaf) {
3726
3727 vdev_queue_io_done(zio);
3728
3729 if (zio->io_type == ZIO_TYPE_WRITE)
3730 vdev_cache_write(zio);
3731
3732 if (zio_injection_enabled && zio->io_error == 0)
d977122d
DB
3733 zio->io_error = zio_handle_device_injections(vd, zio,
3734 EIO, EILSEQ);
b128c09f
BB
3735
3736 if (zio_injection_enabled && zio->io_error == 0)
3737 zio->io_error = zio_handle_label_injection(zio, EIO);
3738
3739 if (zio->io_error) {
3740 if (!vdev_accessible(vd, zio)) {
2e528b49 3741 zio->io_error = SET_ERROR(ENXIO);
b128c09f
BB
3742 } else {
3743 unexpected_error = B_TRUE;
3744 }
3745 }
3746 }
3747
3748 ops->vdev_op_io_done(zio);
34dc7c2f 3749
f43615d0 3750 if (unexpected_error)
d164b209 3751 VERIFY(vdev_probe(vd, zio) == NULL);
34dc7c2f 3752
62840030 3753 return (zio);
34dc7c2f
BB
3754}
3755
a8b2e306
TC
3756/*
3757 * This function is used to change the priority of an existing zio that is
3758 * currently in-flight. This is used by the arc to upgrade priority in the
3759 * event that a demand read is made for a block that is currently queued
3760 * as a scrub or async read IO. Otherwise, the high priority read request
3761 * would end up having to wait for the lower priority IO.
3762 */
3763void
3764zio_change_priority(zio_t *pio, zio_priority_t priority)
3765{
3766 zio_t *cio, *cio_next;
3767 zio_link_t *zl = NULL;
3768
3769 ASSERT3U(priority, <, ZIO_PRIORITY_NUM_QUEUEABLE);
3770
3771 if (pio->io_vd != NULL && pio->io_vd->vdev_ops->vdev_op_leaf) {
3772 vdev_queue_change_io_priority(pio, priority);
3773 } else {
3774 pio->io_priority = priority;
3775 }
3776
3777 mutex_enter(&pio->io_lock);
3778 for (cio = zio_walk_children(pio, &zl); cio != NULL; cio = cio_next) {
3779 cio_next = zio_walk_children(pio, &zl);
3780 zio_change_priority(cio, priority);
3781 }
3782 mutex_exit(&pio->io_lock);
3783}
3784
428870ff
BB
3785/*
3786 * For non-raidz ZIOs, we can just copy aside the bad data read from the
3787 * disk, and use that to finish the checksum ereport later.
3788 */
3789static void
3790zio_vsd_default_cksum_finish(zio_cksum_report_t *zcr,
84c07ada 3791 const abd_t *good_buf)
428870ff
BB
3792{
3793 /* no processing needed */
3794 zfs_ereport_finish_checksum(zcr, good_buf, zcr->zcr_cbdata, B_FALSE);
3795}
3796
3797/*ARGSUSED*/
3798void
3799zio_vsd_default_cksum_report(zio_t *zio, zio_cksum_report_t *zcr, void *ignored)
3800{
84c07ada 3801 void *abd = abd_alloc_sametype(zio->io_abd, zio->io_size);
428870ff 3802
84c07ada 3803 abd_copy(abd, zio->io_abd, zio->io_size);
428870ff
BB
3804
3805 zcr->zcr_cbinfo = zio->io_size;
84c07ada 3806 zcr->zcr_cbdata = abd;
428870ff 3807 zcr->zcr_finish = zio_vsd_default_cksum_finish;
84c07ada 3808 zcr->zcr_free = zio_abd_free;
428870ff
BB
3809}
3810
62840030 3811static zio_t *
34dc7c2f
BB
3812zio_vdev_io_assess(zio_t *zio)
3813{
3814 vdev_t *vd = zio->io_vd;
b128c09f 3815
ddc751d5 3816 if (zio_wait_for_children(zio, ZIO_CHILD_VDEV_BIT, ZIO_WAIT_DONE)) {
62840030 3817 return (NULL);
ddc751d5 3818 }
b128c09f
BB
3819
3820 if (vd == NULL && !(zio->io_flags & ZIO_FLAG_CONFIG_WRITER))
3821 spa_config_exit(zio->io_spa, SCL_ZIO, zio);
3822
3823 if (zio->io_vsd != NULL) {
428870ff 3824 zio->io_vsd_ops->vsd_free(zio);
b128c09f 3825 zio->io_vsd = NULL;
34dc7c2f
BB
3826 }
3827
b128c09f 3828 if (zio_injection_enabled && zio->io_error == 0)
34dc7c2f
BB
3829 zio->io_error = zio_handle_fault_injection(zio, EIO);
3830
3831 /*
3832 * If the I/O failed, determine whether we should attempt to retry it.
428870ff
BB
3833 *
3834 * On retry, we cut in line in the issue queue, since we don't want
3835 * compression/checksumming/etc. work to prevent our (cheap) IO reissue.
34dc7c2f 3836 */
b128c09f
BB
3837 if (zio->io_error && vd == NULL &&
3838 !(zio->io_flags & (ZIO_FLAG_DONT_RETRY | ZIO_FLAG_IO_RETRY))) {
3839 ASSERT(!(zio->io_flags & ZIO_FLAG_DONT_QUEUE)); /* not a leaf */
3840 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_BYPASS)); /* not a leaf */
34dc7c2f 3841 zio->io_error = 0;
b128c09f
BB
3842 zio->io_flags |= ZIO_FLAG_IO_RETRY |
3843 ZIO_FLAG_DONT_CACHE | ZIO_FLAG_DONT_AGGREGATE;
428870ff
BB
3844 zio->io_stage = ZIO_STAGE_VDEV_IO_START >> 1;
3845 zio_taskq_dispatch(zio, ZIO_TASKQ_ISSUE,
3846 zio_requeue_io_start_cut_in_line);
62840030 3847 return (NULL);
34dc7c2f
BB
3848 }
3849
b128c09f
BB
3850 /*
3851 * If we got an error on a leaf device, convert it to ENXIO
3852 * if the device is not accessible at all.
3853 */
3854 if (zio->io_error && vd != NULL && vd->vdev_ops->vdev_op_leaf &&
3855 !vdev_accessible(vd, zio))
2e528b49 3856 zio->io_error = SET_ERROR(ENXIO);
b128c09f
BB
3857
3858 /*
3859 * If we can't write to an interior vdev (mirror or RAID-Z),
3860 * set vdev_cant_write so that we stop trying to allocate from it.
3861 */
3862 if (zio->io_error == ENXIO && zio->io_type == ZIO_TYPE_WRITE &&
13fe0198 3863 vd != NULL && !vd->vdev_ops->vdev_op_leaf) {
b128c09f 3864 vd->vdev_cant_write = B_TRUE;
13fe0198 3865 }
b128c09f 3866
298ec40b
GM
3867 /*
3868 * If a cache flush returns ENOTSUP or ENOTTY, we know that no future
3869 * attempts will ever succeed. In this case we set a persistent bit so
3870 * that we don't bother with it in the future.
3871 */
3872 if ((zio->io_error == ENOTSUP || zio->io_error == ENOTTY) &&
3873 zio->io_type == ZIO_TYPE_IOCTL &&
3874 zio->io_cmd == DKIOCFLUSHWRITECACHE && vd != NULL)
3875 vd->vdev_nowritecache = B_TRUE;
3876
b128c09f
BB
3877 if (zio->io_error)
3878 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
3879
e8b96c60
MA
3880 if (vd != NULL && vd->vdev_ops->vdev_op_leaf &&
3881 zio->io_physdone != NULL) {
3882 ASSERT(!(zio->io_flags & ZIO_FLAG_DELEGATED));
3883 ASSERT(zio->io_child_type == ZIO_CHILD_VDEV);
3884 zio->io_physdone(zio->io_logical);
3885 }
3886
62840030 3887 return (zio);
34dc7c2f
BB
3888}
3889
3890void
3891zio_vdev_io_reissue(zio_t *zio)
3892{
3893 ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_START);
3894 ASSERT(zio->io_error == 0);
3895
428870ff 3896 zio->io_stage >>= 1;
34dc7c2f
BB
3897}
3898
3899void
3900zio_vdev_io_redone(zio_t *zio)
3901{
3902 ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_DONE);
3903
428870ff 3904 zio->io_stage >>= 1;
34dc7c2f
BB
3905}
3906
3907void
3908zio_vdev_io_bypass(zio_t *zio)
3909{
3910 ASSERT(zio->io_stage == ZIO_STAGE_VDEV_IO_START);
3911 ASSERT(zio->io_error == 0);
3912
3913 zio->io_flags |= ZIO_FLAG_IO_BYPASS;
428870ff 3914 zio->io_stage = ZIO_STAGE_VDEV_IO_ASSESS >> 1;
34dc7c2f
BB
3915}
3916
b5256303
TC
3917/*
3918 * ==========================================================================
3919 * Encrypt and store encryption parameters
3920 * ==========================================================================
3921 */
3922
3923
3924/*
3925 * This function is used for ZIO_STAGE_ENCRYPT. It is responsible for
3926 * managing the storage of encryption parameters and passing them to the
3927 * lower-level encryption functions.
3928 */
62840030 3929static zio_t *
b5256303
TC
3930zio_encrypt(zio_t *zio)
3931{
3932 zio_prop_t *zp = &zio->io_prop;
3933 spa_t *spa = zio->io_spa;
3934 blkptr_t *bp = zio->io_bp;
3935 uint64_t psize = BP_GET_PSIZE(bp);
ae76f45c 3936 uint64_t dsobj = zio->io_bookmark.zb_objset;
b5256303
TC
3937 dmu_object_type_t ot = BP_GET_TYPE(bp);
3938 void *enc_buf = NULL;
3939 abd_t *eabd = NULL;
3940 uint8_t salt[ZIO_DATA_SALT_LEN];
3941 uint8_t iv[ZIO_DATA_IV_LEN];
3942 uint8_t mac[ZIO_DATA_MAC_LEN];
3943 boolean_t no_crypt = B_FALSE;
3944
3945 /* the root zio already encrypted the data */
3946 if (zio->io_child_type == ZIO_CHILD_GANG)
62840030 3947 return (zio);
b5256303
TC
3948
3949 /* only ZIL blocks are re-encrypted on rewrite */
3950 if (!IO_IS_ALLOCATING(zio) && ot != DMU_OT_INTENT_LOG)
62840030 3951 return (zio);
b5256303
TC
3952
3953 if (!(zp->zp_encrypt || BP_IS_ENCRYPTED(bp))) {
3954 BP_SET_CRYPT(bp, B_FALSE);
62840030 3955 return (zio);
b5256303
TC
3956 }
3957
3958 /* if we are doing raw encryption set the provided encryption params */
3959 if (zio->io_flags & ZIO_FLAG_RAW_ENCRYPT) {
ae76f45c 3960 ASSERT0(BP_GET_LEVEL(bp));
b5256303
TC
3961 BP_SET_CRYPT(bp, B_TRUE);
3962 BP_SET_BYTEORDER(bp, zp->zp_byteorder);
3963 if (ot != DMU_OT_OBJSET)
3964 zio_crypt_encode_mac_bp(bp, zp->zp_mac);
ae76f45c
TC
3965
3966 /* dnode blocks must be written out in the provided byteorder */
3967 if (zp->zp_byteorder != ZFS_HOST_BYTEORDER &&
3968 ot == DMU_OT_DNODE) {
3969 void *bswap_buf = zio_buf_alloc(psize);
3970 abd_t *babd = abd_get_from_buf(bswap_buf, psize);
3971
3972 ASSERT3U(BP_GET_COMPRESS(bp), ==, ZIO_COMPRESS_OFF);
3973 abd_copy_to_buf(bswap_buf, zio->io_abd, psize);
3974 dmu_ot_byteswap[DMU_OT_BYTESWAP(ot)].ob_func(bswap_buf,
3975 psize);
3976
3977 abd_take_ownership_of_buf(babd, B_TRUE);
3978 zio_push_transform(zio, babd, psize, psize, NULL);
3979 }
3980
b5256303
TC
3981 if (DMU_OT_IS_ENCRYPTED(ot))
3982 zio_crypt_encode_params_bp(bp, zp->zp_salt, zp->zp_iv);
62840030 3983 return (zio);
b5256303
TC
3984 }
3985
3986 /* indirect blocks only maintain a cksum of the lower level MACs */
3987 if (BP_GET_LEVEL(bp) > 0) {
3988 BP_SET_CRYPT(bp, B_TRUE);
3989 VERIFY0(zio_crypt_do_indirect_mac_checksum_abd(B_TRUE,
3990 zio->io_orig_abd, BP_GET_LSIZE(bp), BP_SHOULD_BYTESWAP(bp),
3991 mac));
3992 zio_crypt_encode_mac_bp(bp, mac);
62840030 3993 return (zio);
b5256303
TC
3994 }
3995
3996 /*
3997 * Objset blocks are a special case since they have 2 256-bit MACs
3998 * embedded within them.
3999 */
4000 if (ot == DMU_OT_OBJSET) {
4001 ASSERT0(DMU_OT_IS_ENCRYPTED(ot));
4002 ASSERT3U(BP_GET_COMPRESS(bp), ==, ZIO_COMPRESS_OFF);
4003 BP_SET_CRYPT(bp, B_TRUE);
ae76f45c
TC
4004 VERIFY0(spa_do_crypt_objset_mac_abd(B_TRUE, spa, dsobj,
4005 zio->io_abd, psize, BP_SHOULD_BYTESWAP(bp)));
62840030 4006 return (zio);
b5256303
TC
4007 }
4008
4009 /* unencrypted object types are only authenticated with a MAC */
4010 if (!DMU_OT_IS_ENCRYPTED(ot)) {
4011 BP_SET_CRYPT(bp, B_TRUE);
ae76f45c
TC
4012 VERIFY0(spa_do_crypt_mac_abd(B_TRUE, spa, dsobj,
4013 zio->io_abd, psize, mac));
b5256303 4014 zio_crypt_encode_mac_bp(bp, mac);
62840030 4015 return (zio);
b5256303
TC
4016 }
4017
4018 /*
4019 * Later passes of sync-to-convergence may decide to rewrite data
4020 * in place to avoid more disk reallocations. This presents a problem
d611989f 4021 * for encryption because this constitutes rewriting the new data with
b5256303
TC
4022 * the same encryption key and IV. However, this only applies to blocks
4023 * in the MOS (particularly the spacemaps) and we do not encrypt the
4024 * MOS. We assert that the zio is allocating or an intent log write
4025 * to enforce this.
4026 */
4027 ASSERT(IO_IS_ALLOCATING(zio) || ot == DMU_OT_INTENT_LOG);
4028 ASSERT(BP_GET_LEVEL(bp) == 0 || ot == DMU_OT_INTENT_LOG);
4029 ASSERT(spa_feature_is_active(spa, SPA_FEATURE_ENCRYPTION));
4030 ASSERT3U(psize, !=, 0);
4031
4032 enc_buf = zio_buf_alloc(psize);
4033 eabd = abd_get_from_buf(enc_buf, psize);
4034 abd_take_ownership_of_buf(eabd, B_TRUE);
4035
4036 /*
4037 * For an explanation of what encryption parameters are stored
4038 * where, see the block comment in zio_crypt.c.
4039 */
4040 if (ot == DMU_OT_INTENT_LOG) {
4041 zio_crypt_decode_params_bp(bp, salt, iv);
4042 } else {
4043 BP_SET_CRYPT(bp, B_TRUE);
4044 }
4045
4046 /* Perform the encryption. This should not fail */
be9a5c35
TC
4047 VERIFY0(spa_do_crypt_abd(B_TRUE, spa, &zio->io_bookmark,
4048 BP_GET_TYPE(bp), BP_GET_DEDUP(bp), BP_SHOULD_BYTESWAP(bp),
4049 salt, iv, mac, psize, zio->io_abd, eabd, &no_crypt));
b5256303
TC
4050
4051 /* encode encryption metadata into the bp */
4052 if (ot == DMU_OT_INTENT_LOG) {
4053 /*
4054 * ZIL blocks store the MAC in the embedded checksum, so the
4055 * transform must always be applied.
4056 */
4057 zio_crypt_encode_mac_zil(enc_buf, mac);
4058 zio_push_transform(zio, eabd, psize, psize, NULL);
4059 } else {
4060 BP_SET_CRYPT(bp, B_TRUE);
4061 zio_crypt_encode_params_bp(bp, salt, iv);
4062 zio_crypt_encode_mac_bp(bp, mac);
4063
4064 if (no_crypt) {
4065 ASSERT3U(ot, ==, DMU_OT_DNODE);
4066 abd_free(eabd);
4067 } else {
4068 zio_push_transform(zio, eabd, psize, psize, NULL);
4069 }
4070 }
4071
62840030 4072 return (zio);
b5256303
TC
4073}
4074
34dc7c2f
BB
4075/*
4076 * ==========================================================================
4077 * Generate and verify checksums
4078 * ==========================================================================
4079 */
62840030 4080static zio_t *
34dc7c2f
BB
4081zio_checksum_generate(zio_t *zio)
4082{
34dc7c2f 4083 blkptr_t *bp = zio->io_bp;
b128c09f 4084 enum zio_checksum checksum;
34dc7c2f 4085
b128c09f
BB
4086 if (bp == NULL) {
4087 /*
4088 * This is zio_write_phys().
4089 * We're either generating a label checksum, or none at all.
4090 */
4091 checksum = zio->io_prop.zp_checksum;
34dc7c2f 4092
b128c09f 4093 if (checksum == ZIO_CHECKSUM_OFF)
62840030 4094 return (zio);
b128c09f
BB
4095
4096 ASSERT(checksum == ZIO_CHECKSUM_LABEL);
4097 } else {
4098 if (BP_IS_GANG(bp) && zio->io_child_type == ZIO_CHILD_GANG) {
4099 ASSERT(!IO_IS_ALLOCATING(zio));
4100 checksum = ZIO_CHECKSUM_GANG_HEADER;
4101 } else {
4102 checksum = BP_GET_CHECKSUM(bp);
4103 }
4104 }
34dc7c2f 4105
a6255b7f 4106 zio_checksum_compute(zio, checksum, zio->io_abd, zio->io_size);
34dc7c2f 4107
62840030 4108 return (zio);
34dc7c2f
BB
4109}
4110
62840030 4111static zio_t *
b128c09f 4112zio_checksum_verify(zio_t *zio)
34dc7c2f 4113{
428870ff 4114 zio_bad_cksum_t info;
b128c09f
BB
4115 blkptr_t *bp = zio->io_bp;
4116 int error;
34dc7c2f 4117
428870ff
BB
4118 ASSERT(zio->io_vd != NULL);
4119
b128c09f
BB
4120 if (bp == NULL) {
4121 /*
4122 * This is zio_read_phys().
4123 * We're either verifying a label checksum, or nothing at all.
4124 */
4125 if (zio->io_prop.zp_checksum == ZIO_CHECKSUM_OFF)
62840030 4126 return (zio);
34dc7c2f 4127
b128c09f
BB
4128 ASSERT(zio->io_prop.zp_checksum == ZIO_CHECKSUM_LABEL);
4129 }
34dc7c2f 4130
428870ff 4131 if ((error = zio_checksum_error(zio, &info)) != 0) {
b128c09f 4132 zio->io_error = error;
7a3066ff
MA
4133 if (error == ECKSUM &&
4134 !(zio->io_flags & ZIO_FLAG_SPECULATIVE)) {
428870ff 4135 zfs_ereport_start_checksum(zio->io_spa,
b5256303
TC
4136 zio->io_vd, &zio->io_bookmark, zio,
4137 zio->io_offset, zio->io_size, NULL, &info);
b128c09f 4138 }
34dc7c2f
BB
4139 }
4140
62840030 4141 return (zio);
34dc7c2f
BB
4142}
4143
4144/*
4145 * Called by RAID-Z to ensure we don't compute the checksum twice.
4146 */
4147void
4148zio_checksum_verified(zio_t *zio)
4149{
428870ff 4150 zio->io_pipeline &= ~ZIO_STAGE_CHECKSUM_VERIFY;
34dc7c2f
BB
4151}
4152
4153/*
b128c09f
BB
4154 * ==========================================================================
4155 * Error rank. Error are ranked in the order 0, ENXIO, ECKSUM, EIO, other.
9b67f605 4156 * An error of 0 indicates success. ENXIO indicates whole-device failure,
d611989f 4157 * which may be transient (e.g. unplugged) or permanent. ECKSUM and EIO
b128c09f
BB
4158 * indicate errors that are specific to one I/O, and most likely permanent.
4159 * Any other error is presumed to be worse because we weren't expecting it.
4160 * ==========================================================================
34dc7c2f 4161 */
b128c09f
BB
4162int
4163zio_worst_error(int e1, int e2)
34dc7c2f 4164{
b128c09f
BB
4165 static int zio_error_rank[] = { 0, ENXIO, ECKSUM, EIO };
4166 int r1, r2;
4167
4168 for (r1 = 0; r1 < sizeof (zio_error_rank) / sizeof (int); r1++)
4169 if (e1 == zio_error_rank[r1])
4170 break;
34dc7c2f 4171
b128c09f
BB
4172 for (r2 = 0; r2 < sizeof (zio_error_rank) / sizeof (int); r2++)
4173 if (e2 == zio_error_rank[r2])
4174 break;
4175
4176 return (r1 > r2 ? e1 : e2);
34dc7c2f
BB
4177}
4178
4179/*
4180 * ==========================================================================
b128c09f 4181 * I/O completion
34dc7c2f
BB
4182 * ==========================================================================
4183 */
62840030 4184static zio_t *
b128c09f 4185zio_ready(zio_t *zio)
34dc7c2f 4186{
b128c09f 4187 blkptr_t *bp = zio->io_bp;
d164b209 4188 zio_t *pio, *pio_next;
3dfb57a3 4189 zio_link_t *zl = NULL;
34dc7c2f 4190
ddc751d5
GW
4191 if (zio_wait_for_children(zio, ZIO_CHILD_GANG_BIT | ZIO_CHILD_DDT_BIT,
4192 ZIO_WAIT_READY)) {
62840030 4193 return (NULL);
ddc751d5 4194 }
34dc7c2f 4195
9babb374 4196 if (zio->io_ready) {
b128c09f 4197 ASSERT(IO_IS_ALLOCATING(zio));
03c6040b
GW
4198 ASSERT(bp->blk_birth == zio->io_txg || BP_IS_HOLE(bp) ||
4199 (zio->io_flags & ZIO_FLAG_NOPWRITE));
b128c09f 4200 ASSERT(zio->io_children[ZIO_CHILD_GANG][ZIO_WAIT_READY] == 0);
34dc7c2f 4201
b128c09f
BB
4202 zio->io_ready(zio);
4203 }
34dc7c2f 4204
b128c09f
BB
4205 if (bp != NULL && bp != &zio->io_bp_copy)
4206 zio->io_bp_copy = *bp;
34dc7c2f 4207
3dfb57a3 4208 if (zio->io_error != 0) {
b128c09f 4209 zio->io_pipeline = ZIO_INTERLOCK_PIPELINE;
34dc7c2f 4210
3dfb57a3
DB
4211 if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
4212 ASSERT(IO_IS_ALLOCATING(zio));
4213 ASSERT(zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
cc99f275
DB
4214 ASSERT(zio->io_metaslab_class != NULL);
4215
3dfb57a3
DB
4216 /*
4217 * We were unable to allocate anything, unreserve and
4218 * issue the next I/O to allocate.
4219 */
4220 metaslab_class_throttle_unreserve(
cc99f275
DB
4221 zio->io_metaslab_class, zio->io_prop.zp_copies,
4222 zio->io_allocator, zio);
492f64e9 4223 zio_allocate_dispatch(zio->io_spa, zio->io_allocator);
3dfb57a3
DB
4224 }
4225 }
4226
d164b209
BB
4227 mutex_enter(&zio->io_lock);
4228 zio->io_state[ZIO_WAIT_READY] = 1;
3dfb57a3 4229 pio = zio_walk_parents(zio, &zl);
d164b209
BB
4230 mutex_exit(&zio->io_lock);
4231
4232 /*
4233 * As we notify zio's parents, new parents could be added.
4234 * New parents go to the head of zio's io_parent_list, however,
4235 * so we will (correctly) not notify them. The remainder of zio's
4236 * io_parent_list, from 'pio_next' onward, cannot change because
4237 * all parents must wait for us to be done before they can be done.
4238 */
4239 for (; pio != NULL; pio = pio_next) {
3dfb57a3 4240 pio_next = zio_walk_parents(zio, &zl);
62840030 4241 zio_notify_parent(pio, zio, ZIO_WAIT_READY, NULL);
d164b209 4242 }
34dc7c2f 4243
428870ff
BB
4244 if (zio->io_flags & ZIO_FLAG_NODATA) {
4245 if (BP_IS_GANG(bp)) {
4246 zio->io_flags &= ~ZIO_FLAG_NODATA;
4247 } else {
a6255b7f 4248 ASSERT((uintptr_t)zio->io_abd < SPA_MAXBLOCKSIZE);
428870ff
BB
4249 zio->io_pipeline &= ~ZIO_VDEV_IO_STAGES;
4250 }
4251 }
4252
4253 if (zio_injection_enabled &&
4254 zio->io_spa->spa_syncing_txg == zio->io_txg)
4255 zio_handle_ignored_writes(zio);
4256
62840030 4257 return (zio);
34dc7c2f
BB
4258}
4259
3dfb57a3
DB
4260/*
4261 * Update the allocation throttle accounting.
4262 */
4263static void
4264zio_dva_throttle_done(zio_t *zio)
4265{
1c27024e 4266 ASSERTV(zio_t *lio = zio->io_logical);
3dfb57a3
DB
4267 zio_t *pio = zio_unique_parent(zio);
4268 vdev_t *vd = zio->io_vd;
4269 int flags = METASLAB_ASYNC_ALLOC;
4270
4271 ASSERT3P(zio->io_bp, !=, NULL);
4272 ASSERT3U(zio->io_type, ==, ZIO_TYPE_WRITE);
4273 ASSERT3U(zio->io_priority, ==, ZIO_PRIORITY_ASYNC_WRITE);
4274 ASSERT3U(zio->io_child_type, ==, ZIO_CHILD_VDEV);
4275 ASSERT(vd != NULL);
4276 ASSERT3P(vd, ==, vd->vdev_top);
21df134f
SB
4277 ASSERT(zio_injection_enabled || !(zio->io_flags & ZIO_FLAG_IO_RETRY));
4278 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REPAIR));
3dfb57a3
DB
4279 ASSERT(zio->io_flags & ZIO_FLAG_IO_ALLOCATING);
4280 ASSERT(!(lio->io_flags & ZIO_FLAG_IO_REWRITE));
4281 ASSERT(!(lio->io_orig_flags & ZIO_FLAG_NODATA));
4282
4283 /*
4284 * Parents of gang children can have two flavors -- ones that
4285 * allocated the gang header (will have ZIO_FLAG_IO_REWRITE set)
4286 * and ones that allocated the constituent blocks. The allocation
4287 * throttle needs to know the allocating parent zio so we must find
4288 * it here.
4289 */
4290 if (pio->io_child_type == ZIO_CHILD_GANG) {
4291 /*
4292 * If our parent is a rewrite gang child then our grandparent
4293 * would have been the one that performed the allocation.
4294 */
4295 if (pio->io_flags & ZIO_FLAG_IO_REWRITE)
4296 pio = zio_unique_parent(pio);
4297 flags |= METASLAB_GANG_CHILD;
4298 }
4299
4300 ASSERT(IO_IS_ALLOCATING(pio));
4301 ASSERT3P(zio, !=, zio->io_logical);
4302 ASSERT(zio->io_logical != NULL);
4303 ASSERT(!(zio->io_flags & ZIO_FLAG_IO_REPAIR));
4304 ASSERT0(zio->io_flags & ZIO_FLAG_NOPWRITE);
cc99f275 4305 ASSERT(zio->io_metaslab_class != NULL);
3dfb57a3
DB
4306
4307 mutex_enter(&pio->io_lock);
492f64e9
PD
4308 metaslab_group_alloc_decrement(zio->io_spa, vd->vdev_id, pio, flags,
4309 pio->io_allocator, B_TRUE);
3dfb57a3
DB
4310 mutex_exit(&pio->io_lock);
4311
cc99f275
DB
4312 metaslab_class_throttle_unreserve(zio->io_metaslab_class, 1,
4313 pio->io_allocator, pio);
3dfb57a3
DB
4314
4315 /*
4316 * Call into the pipeline to see if there is more work that
4317 * needs to be done. If there is work to be done it will be
4318 * dispatched to another taskq thread.
4319 */
492f64e9 4320 zio_allocate_dispatch(zio->io_spa, pio->io_allocator);
3dfb57a3
DB
4321}
4322
62840030 4323static zio_t *
b128c09f 4324zio_done(zio_t *zio)
34dc7c2f 4325{
3dfb57a3
DB
4326 /*
4327 * Always attempt to keep stack usage minimal here since
d611989f 4328 * we can be called recursively up to 19 levels deep.
3dfb57a3 4329 */
84c07ada 4330 const uint64_t psize = zio->io_size;
d164b209 4331 zio_t *pio, *pio_next;
3dfb57a3 4332 zio_link_t *zl = NULL;
34dc7c2f 4333
b128c09f 4334 /*
9babb374 4335 * If our children haven't all completed,
b128c09f
BB
4336 * wait for them and then repeat this pipeline stage.
4337 */
ddc751d5 4338 if (zio_wait_for_children(zio, ZIO_CHILD_ALL_BITS, ZIO_WAIT_DONE)) {
62840030 4339 return (NULL);
ddc751d5 4340 }
34dc7c2f 4341
3dfb57a3
DB
4342 /*
4343 * If the allocation throttle is enabled, then update the accounting.
4344 * We only track child I/Os that are part of an allocating async
4345 * write. We must do this since the allocation is performed
4346 * by the logical I/O but the actual write is done by child I/Os.
4347 */
4348 if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING &&
4349 zio->io_child_type == ZIO_CHILD_VDEV) {
cc99f275
DB
4350 ASSERT(zio->io_metaslab_class != NULL);
4351 ASSERT(zio->io_metaslab_class->mc_alloc_throttle_enabled);
3dfb57a3
DB
4352 zio_dva_throttle_done(zio);
4353 }
4354
4355 /*
4356 * If the allocation throttle is enabled, verify that
4357 * we have decremented the refcounts for every I/O that was throttled.
4358 */
4359 if (zio->io_flags & ZIO_FLAG_IO_ALLOCATING) {
4360 ASSERT(zio->io_type == ZIO_TYPE_WRITE);
4361 ASSERT(zio->io_priority == ZIO_PRIORITY_ASYNC_WRITE);
4362 ASSERT(zio->io_bp != NULL);
cc99f275 4363
492f64e9
PD
4364 metaslab_group_alloc_verify(zio->io_spa, zio->io_bp, zio,
4365 zio->io_allocator);
424fd7c3 4366 VERIFY(zfs_refcount_not_held(
cc99f275 4367 &zio->io_metaslab_class->mc_alloc_slots[zio->io_allocator],
492f64e9 4368 zio));
3dfb57a3
DB
4369 }
4370
4371
1c27024e
DB
4372 for (int c = 0; c < ZIO_CHILD_TYPES; c++)
4373 for (int w = 0; w < ZIO_WAIT_TYPES; w++)
b128c09f
BB
4374 ASSERT(zio->io_children[c][w] == 0);
4375
9b67f605 4376 if (zio->io_bp != NULL && !BP_IS_EMBEDDED(zio->io_bp)) {
c776b317
BB
4377 ASSERT(zio->io_bp->blk_pad[0] == 0);
4378 ASSERT(zio->io_bp->blk_pad[1] == 0);
d1d7e268
MK
4379 ASSERT(bcmp(zio->io_bp, &zio->io_bp_copy,
4380 sizeof (blkptr_t)) == 0 ||
c776b317
BB
4381 (zio->io_bp == zio_unique_parent(zio)->io_bp));
4382 if (zio->io_type == ZIO_TYPE_WRITE && !BP_IS_HOLE(zio->io_bp) &&
428870ff 4383 zio->io_bp_override == NULL &&
b128c09f 4384 !(zio->io_flags & ZIO_FLAG_IO_REPAIR)) {
d1d7e268
MK
4385 ASSERT3U(zio->io_prop.zp_copies, <=,
4386 BP_GET_NDVAS(zio->io_bp));
c776b317 4387 ASSERT(BP_COUNT_GANG(zio->io_bp) == 0 ||
d1d7e268
MK
4388 (BP_COUNT_GANG(zio->io_bp) ==
4389 BP_GET_NDVAS(zio->io_bp)));
b128c09f 4390 }
03c6040b
GW
4391 if (zio->io_flags & ZIO_FLAG_NOPWRITE)
4392 VERIFY(BP_EQUAL(zio->io_bp, &zio->io_bp_orig));
b128c09f
BB
4393 }
4394
4395 /*
428870ff 4396 * If there were child vdev/gang/ddt errors, they apply to us now.
b128c09f
BB
4397 */
4398 zio_inherit_child_errors(zio, ZIO_CHILD_VDEV);
4399 zio_inherit_child_errors(zio, ZIO_CHILD_GANG);
428870ff
BB
4400 zio_inherit_child_errors(zio, ZIO_CHILD_DDT);
4401
4402 /*
4403 * If the I/O on the transformed data was successful, generate any
4404 * checksum reports now while we still have the transformed data.
4405 */
4406 if (zio->io_error == 0) {
4407 while (zio->io_cksum_report != NULL) {
4408 zio_cksum_report_t *zcr = zio->io_cksum_report;
4409 uint64_t align = zcr->zcr_align;
a6255b7f 4410 uint64_t asize = P2ROUNDUP(psize, align);
a6255b7f
DQ
4411 abd_t *adata = zio->io_abd;
4412
4413 if (asize != psize) {
84c07ada 4414 adata = abd_alloc(asize, B_TRUE);
a6255b7f
DQ
4415 abd_copy(adata, zio->io_abd, psize);
4416 abd_zero_off(adata, psize, asize - psize);
428870ff
BB
4417 }
4418
4419 zio->io_cksum_report = zcr->zcr_next;
4420 zcr->zcr_next = NULL;
84c07ada 4421 zcr->zcr_finish(zcr, adata);
428870ff
BB
4422 zfs_ereport_free_checksum(zcr);
4423
a6255b7f
DQ
4424 if (asize != psize)
4425 abd_free(adata);
428870ff
BB
4426 }
4427 }
b128c09f
BB
4428
4429 zio_pop_transforms(zio); /* note: may set zio->io_error */
4430
a6255b7f 4431 vdev_stat_update(zio, psize);
b128c09f 4432
a69052be 4433 /*
cc92e9d0 4434 * If this I/O is attached to a particular vdev is slow, exceeding
72f53c56
MJ
4435 * 30 seconds to complete, post an error described the I/O delay.
4436 * We ignore these errors if the device is currently unavailable.
a69052be 4437 */
ad796b8a
TH
4438 if (zio->io_delay >= MSEC2NSEC(zio_slow_io_ms)) {
4439 if (zio->io_vd != NULL && !vdev_is_dead(zio->io_vd)) {
4440 /*
4441 * We want to only increment our slow IO counters if
4442 * the IO is valid (i.e. not if the drive is removed).
4443 *
4444 * zfs_ereport_post() will also do these checks, but
4445 * it can also ratelimit and have other failures, so we
4446 * need to increment the slow_io counters independent
4447 * of it.
4448 */
4449 if (zfs_ereport_is_valid(FM_EREPORT_ZFS_DELAY,
4450 zio->io_spa, zio->io_vd, zio)) {
4451 mutex_enter(&zio->io_vd->vdev_stat_lock);
4452 zio->io_vd->vdev_stat.vs_slow_ios++;
4453 mutex_exit(&zio->io_vd->vdev_stat_lock);
4454
4455 zfs_ereport_post(FM_EREPORT_ZFS_DELAY,
4456 zio->io_spa, zio->io_vd, &zio->io_bookmark,
4457 zio, 0, 0);
4458 }
4459 }
72f53c56 4460 }
a69052be 4461
b128c09f
BB
4462 if (zio->io_error) {
4463 /*
4464 * If this I/O is attached to a particular vdev,
4465 * generate an error message describing the I/O failure
4466 * at the block level. We ignore these errors if the
4467 * device is currently unavailable.
4468 */
c776b317 4469 if (zio->io_error != ECKSUM && zio->io_vd != NULL &&
02730c33 4470 !vdev_is_dead(zio->io_vd))
c776b317 4471 zfs_ereport_post(FM_EREPORT_ZFS_IO, zio->io_spa,
b5256303 4472 zio->io_vd, &zio->io_bookmark, zio, 0, 0);
34dc7c2f 4473
428870ff
BB
4474 if ((zio->io_error == EIO || !(zio->io_flags &
4475 (ZIO_FLAG_SPECULATIVE | ZIO_FLAG_DONT_PROPAGATE))) &&
c776b317 4476 zio == zio->io_logical) {
b128c09f
BB
4477 /*
4478 * For logical I/O requests, tell the SPA to log the
4479 * error and generate a logical data ereport.
4480 */
b5256303 4481 spa_log_error(zio->io_spa, &zio->io_bookmark);
d1d7e268 4482 zfs_ereport_post(FM_EREPORT_ZFS_DATA, zio->io_spa,
b5256303 4483 NULL, &zio->io_bookmark, zio, 0, 0);
b128c09f
BB
4484 }
4485 }
34dc7c2f 4486
c776b317 4487 if (zio->io_error && zio == zio->io_logical) {
b128c09f
BB
4488 /*
4489 * Determine whether zio should be reexecuted. This will
4490 * propagate all the way to the root via zio_notify_parent().
4491 */
c776b317 4492 ASSERT(zio->io_vd == NULL && zio->io_bp != NULL);
428870ff 4493 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
b128c09f 4494
428870ff
BB
4495 if (IO_IS_ALLOCATING(zio) &&
4496 !(zio->io_flags & ZIO_FLAG_CANFAIL)) {
b128c09f
BB
4497 if (zio->io_error != ENOSPC)
4498 zio->io_reexecute |= ZIO_REEXECUTE_NOW;
4499 else
4500 zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
428870ff 4501 }
b128c09f
BB
4502
4503 if ((zio->io_type == ZIO_TYPE_READ ||
4504 zio->io_type == ZIO_TYPE_FREE) &&
572e2857 4505 !(zio->io_flags & ZIO_FLAG_SCAN_THREAD) &&
b128c09f 4506 zio->io_error == ENXIO &&
c776b317
BB
4507 spa_load_state(zio->io_spa) == SPA_LOAD_NONE &&
4508 spa_get_failmode(zio->io_spa) != ZIO_FAILURE_MODE_CONTINUE)
b128c09f
BB
4509 zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
4510
4511 if (!(zio->io_flags & ZIO_FLAG_CANFAIL) && !zio->io_reexecute)
4512 zio->io_reexecute |= ZIO_REEXECUTE_SUSPEND;
428870ff
BB
4513
4514 /*
4515 * Here is a possibly good place to attempt to do
4516 * either combinatorial reconstruction or error correction
4517 * based on checksums. It also might be a good place
4518 * to send out preliminary ereports before we suspend
4519 * processing.
4520 */
34dc7c2f
BB
4521 }
4522
4523 /*
b128c09f
BB
4524 * If there were logical child errors, they apply to us now.
4525 * We defer this until now to avoid conflating logical child
4526 * errors with errors that happened to the zio itself when
4527 * updating vdev stats and reporting FMA events above.
34dc7c2f 4528 */
b128c09f 4529 zio_inherit_child_errors(zio, ZIO_CHILD_LOGICAL);
34dc7c2f 4530
428870ff
BB
4531 if ((zio->io_error || zio->io_reexecute) &&
4532 IO_IS_ALLOCATING(zio) && zio->io_gang_leader == zio &&
03c6040b 4533 !(zio->io_flags & (ZIO_FLAG_IO_REWRITE | ZIO_FLAG_NOPWRITE)))
c776b317 4534 zio_dva_unallocate(zio, zio->io_gang_tree, zio->io_bp);
9babb374
BB
4535
4536 zio_gang_tree_free(&zio->io_gang_tree);
4537
4538 /*
4539 * Godfather I/Os should never suspend.
4540 */
4541 if ((zio->io_flags & ZIO_FLAG_GODFATHER) &&
4542 (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND))
a32494d2 4543 zio->io_reexecute &= ~ZIO_REEXECUTE_SUSPEND;
9babb374 4544
b128c09f
BB
4545 if (zio->io_reexecute) {
4546 /*
4547 * This is a logical I/O that wants to reexecute.
4548 *
4549 * Reexecute is top-down. When an i/o fails, if it's not
4550 * the root, it simply notifies its parent and sticks around.
4551 * The parent, seeing that it still has children in zio_done(),
4552 * does the same. This percolates all the way up to the root.
4553 * The root i/o will reexecute or suspend the entire tree.
4554 *
4555 * This approach ensures that zio_reexecute() honors
4556 * all the original i/o dependency relationships, e.g.
4557 * parents not executing until children are ready.
4558 */
4559 ASSERT(zio->io_child_type == ZIO_CHILD_LOGICAL);
34dc7c2f 4560
9babb374 4561 zio->io_gang_leader = NULL;
b128c09f 4562
d164b209
BB
4563 mutex_enter(&zio->io_lock);
4564 zio->io_state[ZIO_WAIT_DONE] = 1;
4565 mutex_exit(&zio->io_lock);
4566
9babb374
BB
4567 /*
4568 * "The Godfather" I/O monitors its children but is
4569 * not a true parent to them. It will track them through
4570 * the pipeline but severs its ties whenever they get into
4571 * trouble (e.g. suspended). This allows "The Godfather"
4572 * I/O to return status without blocking.
4573 */
3dfb57a3
DB
4574 zl = NULL;
4575 for (pio = zio_walk_parents(zio, &zl); pio != NULL;
4576 pio = pio_next) {
4577 zio_link_t *remove_zl = zl;
4578 pio_next = zio_walk_parents(zio, &zl);
9babb374
BB
4579
4580 if ((pio->io_flags & ZIO_FLAG_GODFATHER) &&
4581 (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND)) {
3dfb57a3 4582 zio_remove_child(pio, zio, remove_zl);
62840030
MA
4583 /*
4584 * This is a rare code path, so we don't
4585 * bother with "next_to_execute".
4586 */
4587 zio_notify_parent(pio, zio, ZIO_WAIT_DONE,
4588 NULL);
9babb374
BB
4589 }
4590 }
4591
d164b209 4592 if ((pio = zio_unique_parent(zio)) != NULL) {
b128c09f
BB
4593 /*
4594 * We're not a root i/o, so there's nothing to do
4595 * but notify our parent. Don't propagate errors
4596 * upward since we haven't permanently failed yet.
4597 */
9babb374 4598 ASSERT(!(zio->io_flags & ZIO_FLAG_GODFATHER));
b128c09f 4599 zio->io_flags |= ZIO_FLAG_DONT_PROPAGATE;
62840030
MA
4600 /*
4601 * This is a rare code path, so we don't bother with
4602 * "next_to_execute".
4603 */
4604 zio_notify_parent(pio, zio, ZIO_WAIT_DONE, NULL);
b128c09f
BB
4605 } else if (zio->io_reexecute & ZIO_REEXECUTE_SUSPEND) {
4606 /*
4607 * We'd fail again if we reexecuted now, so suspend
4608 * until conditions improve (e.g. device comes online).
4609 */
cec3a0a1 4610 zio_suspend(zio->io_spa, zio, ZIO_SUSPEND_IOERR);
b128c09f
BB
4611 } else {
4612 /*
4613 * Reexecution is potentially a huge amount of work.
4614 * Hand it off to the otherwise-unused claim taskq.
4615 */
a38718a6 4616 ASSERT(taskq_empty_ent(&zio->io_tqent));
7ef5e54e
AL
4617 spa_taskq_dispatch_ent(zio->io_spa,
4618 ZIO_TYPE_CLAIM, ZIO_TASKQ_ISSUE,
a38718a6
GA
4619 (task_func_t *)zio_reexecute, zio, 0,
4620 &zio->io_tqent);
b128c09f 4621 }
62840030 4622 return (NULL);
34dc7c2f
BB
4623 }
4624
428870ff 4625 ASSERT(zio->io_child_count == 0);
b128c09f
BB
4626 ASSERT(zio->io_reexecute == 0);
4627 ASSERT(zio->io_error == 0 || (zio->io_flags & ZIO_FLAG_CANFAIL));
34dc7c2f 4628
428870ff
BB
4629 /*
4630 * Report any checksum errors, since the I/O is complete.
4631 */
4632 while (zio->io_cksum_report != NULL) {
4633 zio_cksum_report_t *zcr = zio->io_cksum_report;
4634 zio->io_cksum_report = zcr->zcr_next;
4635 zcr->zcr_next = NULL;
4636 zcr->zcr_finish(zcr, NULL);
4637 zfs_ereport_free_checksum(zcr);
4638 }
4639
920dd524 4640 if (zio->io_flags & ZIO_FLAG_FASTWRITE && zio->io_bp &&
9b67f605
MA
4641 !BP_IS_HOLE(zio->io_bp) && !BP_IS_EMBEDDED(zio->io_bp) &&
4642 !(zio->io_flags & ZIO_FLAG_NOPWRITE)) {
920dd524
ED
4643 metaslab_fastwrite_unmark(zio->io_spa, zio->io_bp);
4644 }
4645
d164b209
BB
4646 /*
4647 * It is the responsibility of the done callback to ensure that this
4648 * particular zio is no longer discoverable for adoption, and as
4649 * such, cannot acquire any new parents.
4650 */
b128c09f
BB
4651 if (zio->io_done)
4652 zio->io_done(zio);
34dc7c2f 4653
d164b209
BB
4654 mutex_enter(&zio->io_lock);
4655 zio->io_state[ZIO_WAIT_DONE] = 1;
4656 mutex_exit(&zio->io_lock);
34dc7c2f 4657
62840030
MA
4658 /*
4659 * We are done executing this zio. We may want to execute a parent
4660 * next. See the comment in zio_notify_parent().
4661 */
4662 zio_t *next_to_execute = NULL;
3dfb57a3
DB
4663 zl = NULL;
4664 for (pio = zio_walk_parents(zio, &zl); pio != NULL; pio = pio_next) {
4665 zio_link_t *remove_zl = zl;
4666 pio_next = zio_walk_parents(zio, &zl);
4667 zio_remove_child(pio, zio, remove_zl);
62840030 4668 zio_notify_parent(pio, zio, ZIO_WAIT_DONE, &next_to_execute);
b128c09f 4669 }
34dc7c2f 4670
b128c09f
BB
4671 if (zio->io_waiter != NULL) {
4672 mutex_enter(&zio->io_lock);
4673 zio->io_executor = NULL;
4674 cv_broadcast(&zio->io_cv);
4675 mutex_exit(&zio->io_lock);
4676 } else {
4677 zio_destroy(zio);
4678 }
34dc7c2f 4679
62840030 4680 return (next_to_execute);
34dc7c2f
BB
4681}
4682
4683/*
b128c09f
BB
4684 * ==========================================================================
4685 * I/O pipeline definition
4686 * ==========================================================================
34dc7c2f 4687 */
428870ff 4688static zio_pipe_stage_t *zio_pipeline[] = {
b128c09f 4689 NULL,
b128c09f 4690 zio_read_bp_init,
3dfb57a3 4691 zio_write_bp_init,
428870ff
BB
4692 zio_free_bp_init,
4693 zio_issue_async,
3dfb57a3 4694 zio_write_compress,
b5256303 4695 zio_encrypt,
b128c09f 4696 zio_checksum_generate,
03c6040b 4697 zio_nop_write,
428870ff
BB
4698 zio_ddt_read_start,
4699 zio_ddt_read_done,
4700 zio_ddt_write,
4701 zio_ddt_free,
b128c09f
BB
4702 zio_gang_assemble,
4703 zio_gang_issue,
3dfb57a3 4704 zio_dva_throttle,
b128c09f
BB
4705 zio_dva_allocate,
4706 zio_dva_free,
4707 zio_dva_claim,
4708 zio_ready,
4709 zio_vdev_io_start,
4710 zio_vdev_io_done,
4711 zio_vdev_io_assess,
4712 zio_checksum_verify,
4713 zio_done
4714};
c28b2279 4715
9ae529ec 4716
9ae529ec 4717
9ae529ec 4718
fcff0f35
PD
4719/*
4720 * Compare two zbookmark_phys_t's to see which we would reach first in a
4721 * pre-order traversal of the object tree.
4722 *
4723 * This is simple in every case aside from the meta-dnode object. For all other
4724 * objects, we traverse them in order (object 1 before object 2, and so on).
4725 * However, all of these objects are traversed while traversing object 0, since
4726 * the data it points to is the list of objects. Thus, we need to convert to a
4727 * canonical representation so we can compare meta-dnode bookmarks to
4728 * non-meta-dnode bookmarks.
4729 *
4730 * We do this by calculating "equivalents" for each field of the zbookmark.
4731 * zbookmarks outside of the meta-dnode use their own object and level, and
4732 * calculate the level 0 equivalent (the first L0 blkid that is contained in the
4733 * blocks this bookmark refers to) by multiplying their blkid by their span
4734 * (the number of L0 blocks contained within one block at their level).
4735 * zbookmarks inside the meta-dnode calculate their object equivalent
4736 * (which is L0equiv * dnodes per data block), use 0 for their L0equiv, and use
4737 * level + 1<<31 (any value larger than a level could ever be) for their level.
4738 * This causes them to always compare before a bookmark in their object
4739 * equivalent, compare appropriately to bookmarks in other objects, and to
4740 * compare appropriately to other bookmarks in the meta-dnode.
4741 */
4742int
4743zbookmark_compare(uint16_t dbss1, uint8_t ibs1, uint16_t dbss2, uint8_t ibs2,
4744 const zbookmark_phys_t *zb1, const zbookmark_phys_t *zb2)
4745{
4746 /*
4747 * These variables represent the "equivalent" values for the zbookmark,
4748 * after converting zbookmarks inside the meta dnode to their
4749 * normal-object equivalents.
4750 */
4751 uint64_t zb1obj, zb2obj;
4752 uint64_t zb1L0, zb2L0;
4753 uint64_t zb1level, zb2level;
4754
4755 if (zb1->zb_object == zb2->zb_object &&
4756 zb1->zb_level == zb2->zb_level &&
4757 zb1->zb_blkid == zb2->zb_blkid)
4758 return (0);
9ae529ec 4759
fcff0f35
PD
4760 /*
4761 * BP_SPANB calculates the span in blocks.
4762 */
4763 zb1L0 = (zb1->zb_blkid) * BP_SPANB(ibs1, zb1->zb_level);
4764 zb2L0 = (zb2->zb_blkid) * BP_SPANB(ibs2, zb2->zb_level);
9ae529ec
CS
4765
4766 if (zb1->zb_object == DMU_META_DNODE_OBJECT) {
fcff0f35
PD
4767 zb1obj = zb1L0 * (dbss1 << (SPA_MINBLOCKSHIFT - DNODE_SHIFT));
4768 zb1L0 = 0;
4769 zb1level = zb1->zb_level + COMPARE_META_LEVEL;
4770 } else {
4771 zb1obj = zb1->zb_object;
4772 zb1level = zb1->zb_level;
9ae529ec
CS
4773 }
4774
fcff0f35
PD
4775 if (zb2->zb_object == DMU_META_DNODE_OBJECT) {
4776 zb2obj = zb2L0 * (dbss2 << (SPA_MINBLOCKSHIFT - DNODE_SHIFT));
4777 zb2L0 = 0;
4778 zb2level = zb2->zb_level + COMPARE_META_LEVEL;
4779 } else {
4780 zb2obj = zb2->zb_object;
4781 zb2level = zb2->zb_level;
4782 }
4783
4784 /* Now that we have a canonical representation, do the comparison. */
4785 if (zb1obj != zb2obj)
4786 return (zb1obj < zb2obj ? -1 : 1);
4787 else if (zb1L0 != zb2L0)
4788 return (zb1L0 < zb2L0 ? -1 : 1);
4789 else if (zb1level != zb2level)
4790 return (zb1level > zb2level ? -1 : 1);
4791 /*
4792 * This can (theoretically) happen if the bookmarks have the same object
4793 * and level, but different blkids, if the block sizes are not the same.
4794 * There is presently no way to change the indirect block sizes
4795 */
4796 return (0);
4797}
4798
4799/*
4800 * This function checks the following: given that last_block is the place that
4801 * our traversal stopped last time, does that guarantee that we've visited
4802 * every node under subtree_root? Therefore, we can't just use the raw output
4803 * of zbookmark_compare. We have to pass in a modified version of
4804 * subtree_root; by incrementing the block id, and then checking whether
4805 * last_block is before or equal to that, we can tell whether or not having
4806 * visited last_block implies that all of subtree_root's children have been
4807 * visited.
4808 */
4809boolean_t
4810zbookmark_subtree_completed(const dnode_phys_t *dnp,
4811 const zbookmark_phys_t *subtree_root, const zbookmark_phys_t *last_block)
4812{
4813 zbookmark_phys_t mod_zb = *subtree_root;
4814 mod_zb.zb_blkid++;
4815 ASSERT(last_block->zb_level == 0);
4816
4817 /* The objset_phys_t isn't before anything. */
4818 if (dnp == NULL)
9ae529ec 4819 return (B_FALSE);
fcff0f35
PD
4820
4821 /*
4822 * We pass in 1ULL << (DNODE_BLOCK_SHIFT - SPA_MINBLOCKSHIFT) for the
4823 * data block size in sectors, because that variable is only used if
4824 * the bookmark refers to a block in the meta-dnode. Since we don't
4825 * know without examining it what object it refers to, and there's no
4826 * harm in passing in this value in other cases, we always pass it in.
4827 *
4828 * We pass in 0 for the indirect block size shift because zb2 must be
4829 * level 0. The indirect block size is only used to calculate the span
4830 * of the bookmark, but since the bookmark must be level 0, the span is
4831 * always 1, so the math works out.
4832 *
4833 * If you make changes to how the zbookmark_compare code works, be sure
4834 * to make sure that this code still works afterwards.
4835 */
4836 return (zbookmark_compare(dnp->dn_datablkszsec, dnp->dn_indblkshift,
4837 1ULL << (DNODE_BLOCK_SHIFT - SPA_MINBLOCKSHIFT), 0, &mod_zb,
4838 last_block) <= 0);
9ae529ec
CS
4839}
4840
93ce2b4c 4841#if defined(_KERNEL)
c28b2279 4842EXPORT_SYMBOL(zio_type_name);
81971b13
BB
4843EXPORT_SYMBOL(zio_buf_alloc);
4844EXPORT_SYMBOL(zio_data_buf_alloc);
4845EXPORT_SYMBOL(zio_buf_free);
4846EXPORT_SYMBOL(zio_data_buf_free);
c28b2279 4847
ad796b8a
TH
4848module_param(zio_slow_io_ms, int, 0644);
4849MODULE_PARM_DESC(zio_slow_io_ms,
4850 "Max I/O completion time (milliseconds) before marking it as slow");
c409e464
BB
4851
4852module_param(zio_requeue_io_start_cut_in_line, int, 0644);
4853MODULE_PARM_DESC(zio_requeue_io_start_cut_in_line, "Prioritize requeued I/O");
29dee3ee
CP
4854
4855module_param(zfs_sync_pass_deferred_free, int, 0644);
4856MODULE_PARM_DESC(zfs_sync_pass_deferred_free,
d1d7e268 4857 "Defer frees starting in this pass");
29dee3ee
CP
4858
4859module_param(zfs_sync_pass_dont_compress, int, 0644);
4860MODULE_PARM_DESC(zfs_sync_pass_dont_compress,
d1d7e268 4861 "Don't compress starting in this pass");
29dee3ee
CP
4862
4863module_param(zfs_sync_pass_rewrite, int, 0644);
4864MODULE_PARM_DESC(zfs_sync_pass_rewrite,
d1d7e268 4865 "Rewrite new bps starting in this pass");
3dfb57a3
DB
4866
4867module_param(zio_dva_throttle_enabled, int, 0644);
4868MODULE_PARM_DESC(zio_dva_throttle_enabled,
4869 "Throttle block allocations in the ZIO pipeline");
638dd5f4
TC
4870
4871module_param(zio_deadman_log_all, int, 0644);
4872MODULE_PARM_DESC(zio_deadman_log_all,
4873 "Log all slow ZIOs, not just those with vdevs");
c28b2279 4874#endif