xref: /freebsd/sys/contrib/openzfs/cmd/zdb/zdb.c (revision 2f10ffc003be396f3fc23cd2888023896560252b)
1 // SPDX-License-Identifier: CDDL-1.0
2 /*
3  * This file and its contents are supplied under the terms of the
4  * Common Development and Distribution License ("CDDL"), version 1.0.
5  * You may only use this file in accordance with the terms of version
6  * 1.0 of the CDDL.
7  *
8  * A full copy of the text of the CDDL should have accompanied this
9  * source.  A copy of the CDDL is also available via the Internet at
10  * https://opensource.org/license/CDDL-1.0.
11  */
12 
13 /*
14  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
15  * Copyright (c) 2011, 2019 by Delphix. All rights reserved.
16  * Copyright (c) 2014 Integros [integros.com]
17  * Copyright 2016 Nexenta Systems, Inc.
18  * Copyright (c) 2017, 2018 Lawrence Livermore National Security, LLC.
19  * Copyright (c) 2015, 2017, Intel Corporation.
20  * Copyright (c) 2020 Datto Inc.
21  * Copyright (c) 2020, The FreeBSD Foundation [1]
22  *
23  * [1] Portions of this software were developed by Allan Jude
24  *     under sponsorship from the FreeBSD Foundation.
25  * Copyright (c) 2021 Allan Jude
26  * Copyright (c) 2021 Toomas Soome <tsoome@me.com>
27  * Copyright (c) 2023, 2024, Klara Inc.
28  * Copyright (c) 2023, Rob Norris <robn@despairlabs.com>
29  * Copyright (c) 2026, TrueNAS.
30  * Copyright 2026 Edgecast Cloud LLC.
31  */
32 
33 #include <stdio.h>
34 #include <unistd.h>
35 #include <stdlib.h>
36 #include <ctype.h>
37 #include <getopt.h>
38 #include <openssl/evp.h>
39 #include <sys/zfs_context.h>
40 #include <sys/spa.h>
41 #include <sys/spa_impl.h>
42 #include <sys/dmu.h>
43 #include <sys/zap.h>
44 #include <sys/zap_impl.h>
45 #include <sys/fs/zfs.h>
46 #include <sys/zfs_znode.h>
47 #include <sys/zfs_sa.h>
48 #include <sys/sa.h>
49 #include <sys/sa_impl.h>
50 #include <sys/vdev.h>
51 #include <sys/vdev_impl.h>
52 #include <sys/metaslab_impl.h>
53 #include <sys/dmu_objset.h>
54 #include <sys/dsl_dir.h>
55 #include <sys/dsl_dataset.h>
56 #include <sys/dsl_pool.h>
57 #include <sys/dsl_bookmark.h>
58 #include <sys/dbuf.h>
59 #include <sys/zil.h>
60 #include <sys/zil_impl.h>
61 #include <sys/stat.h>
62 #include <sys/resource.h>
63 #include <sys/dmu_send.h>
64 #include <sys/dmu_traverse.h>
65 #include <sys/zio_checksum.h>
66 #include <sys/zio_compress.h>
67 #include <sys/zfs_fuid.h>
68 #include <sys/arc.h>
69 #include <sys/arc_impl.h>
70 #include <sys/ddt.h>
71 #include <sys/ddt_impl.h>
72 #include <sys/zfeature.h>
73 #include <sys/abd.h>
74 #include <sys/blkptr.h>
75 #include <sys/dsl_crypt.h>
76 #include <sys/dsl_scan.h>
77 #include <sys/btree.h>
78 #include <sys/brt.h>
79 #include <sys/brt_impl.h>
80 #include <zfs_comutil.h>
81 #include <sys/vdev_raidz.h>
82 #include <sys/vdev_raidz_impl.h>
83 #include <sys/zstd/zstd.h>
84 #include <sys/backtrace.h>
85 
86 #include <libzpool.h>
87 #include <libnvpair.h>
88 #include <libzutil.h>
89 #include <libzfs_core.h>
90 
91 #include <libzdb.h>
92 
93 #include "zdb.h"
94 
95 
96 extern int reference_tracking_enable;
97 extern int zfs_recover;
98 extern uint_t zfs_vdev_async_read_max_active;
99 extern boolean_t spa_load_verify_dryrun;
100 extern boolean_t spa_mode_readable_spacemaps;
101 extern uint_t zfs_reconstruct_indirect_combinations_max;
102 extern uint_t zfs_btree_verify_intensity;
103 
104 enum {
105 	ARG_ALLOCATED = 256,
106 	ARG_BLOCK_BIN_MODE,
107 	ARG_BLOCK_CLASSES,
108 };
109 
110 static const char cmdname[] = "zdb";
111 uint8_t dump_opt[512];
112 
113 typedef void object_viewer_t(objset_t *, uint64_t, void *data, size_t size);
114 
115 static uint64_t *zopt_metaslab = NULL;
116 static unsigned zopt_metaslab_args = 0;
117 
118 
119 static zopt_object_range_t *zopt_object_ranges = NULL;
120 static unsigned zopt_object_args = 0;
121 
122 static int flagbits[256];
123 
124 
125 static uint64_t max_inflight_bytes = 256 * 1024 * 1024; /* 256MB */
126 static int leaked_objects = 0;
127 static zfs_range_tree_t *mos_refd_objs;
128 static spa_t *spa;
129 static objset_t *os;
130 static boolean_t kernel_init_done;
131 static boolean_t corruption_found = B_FALSE;
132 
133 static enum {
134 	BIN_AUTO = 0,
135 	BIN_PSIZE,
136 	BIN_LSIZE,
137 	BIN_ASIZE,
138 } block_bin_mode = BIN_AUTO;
139 
140 static enum {
141 	CLASS_NORMAL = 1 << 1,
142 	CLASS_SPECIAL = 1 << 2,
143 	CLASS_DEDUP = 1 << 3,
144 	CLASS_OTHER = 1 << 4,
145 } block_classes = 0;
146 
147 static void snprintf_blkptr_compact(char *, size_t, const blkptr_t *,
148     boolean_t);
149 static void mos_obj_refd(uint64_t);
150 static void mos_obj_refd_multiple(uint64_t);
151 static int dump_bpobj_cb(void *arg, const blkptr_t *bp, boolean_t free,
152     dmu_tx_t *tx);
153 
154 
155 
156 static void zdb_print_blkptr(const blkptr_t *bp, int flags);
157 static void zdb_exit(int reason);
158 
159 typedef struct sublivelist_verify_block_refcnt {
160 	/* block pointer entry in livelist being verified */
161 	blkptr_t svbr_blk;
162 
163 	/*
164 	 * Refcount gets incremented to 1 when we encounter the first
165 	 * FREE entry for the svfbr block pointer and a node for it
166 	 * is created in our ZDB verification/tracking metadata.
167 	 *
168 	 * As we encounter more FREE entries we increment this counter
169 	 * and similarly decrement it whenever we find the respective
170 	 * ALLOC entries for this block.
171 	 *
172 	 * When the refcount gets to 0 it means that all the FREE and
173 	 * ALLOC entries of this block have paired up and we no longer
174 	 * need to track it in our verification logic (e.g. the node
175 	 * containing this struct in our verification data structure
176 	 * should be freed).
177 	 *
178 	 * [refer to sublivelist_verify_blkptr() for the actual code]
179 	 */
180 	uint32_t svbr_refcnt;
181 } sublivelist_verify_block_refcnt_t;
182 
183 static int
sublivelist_block_refcnt_compare(const void * larg,const void * rarg)184 sublivelist_block_refcnt_compare(const void *larg, const void *rarg)
185 {
186 	const sublivelist_verify_block_refcnt_t *l = larg;
187 	const sublivelist_verify_block_refcnt_t *r = rarg;
188 	return (livelist_compare(&l->svbr_blk, &r->svbr_blk));
189 }
190 
191 static int
sublivelist_verify_blkptr(void * arg,const blkptr_t * bp,boolean_t free,dmu_tx_t * tx)192 sublivelist_verify_blkptr(void *arg, const blkptr_t *bp, boolean_t free,
193     dmu_tx_t *tx)
194 {
195 	ASSERT0P(tx);
196 	struct sublivelist_verify *sv = arg;
197 	sublivelist_verify_block_refcnt_t current = {
198 			.svbr_blk = *bp,
199 
200 			/*
201 			 * Start with 1 in case this is the first free entry.
202 			 * This field is not used for our B-Tree comparisons
203 			 * anyway.
204 			 */
205 			.svbr_refcnt = 1,
206 	};
207 
208 	zfs_btree_index_t where;
209 	sublivelist_verify_block_refcnt_t *pair =
210 	    zfs_btree_find(&sv->sv_pair, &current, &where);
211 	if (free) {
212 		if (pair == NULL) {
213 			/* first free entry for this block pointer */
214 			zfs_btree_add(&sv->sv_pair, &current);
215 		} else {
216 			pair->svbr_refcnt++;
217 		}
218 	} else {
219 		if (pair == NULL) {
220 			/* block that is currently marked as allocated */
221 			for (int i = 0; i < SPA_DVAS_PER_BP; i++) {
222 				if (DVA_IS_EMPTY(&bp->blk_dva[i]))
223 					break;
224 				sublivelist_verify_block_t svb = {
225 				    .svb_dva = bp->blk_dva[i],
226 				    .svb_allocated_txg =
227 				    BP_GET_BIRTH(bp)
228 				};
229 
230 				if (zfs_btree_find(&sv->sv_leftover, &svb,
231 				    &where) == NULL) {
232 					zfs_btree_add_idx(&sv->sv_leftover,
233 					    &svb, &where);
234 				}
235 			}
236 		} else {
237 			/* alloc matches a free entry */
238 			pair->svbr_refcnt--;
239 			if (pair->svbr_refcnt == 0) {
240 				/* all allocs and frees have been matched */
241 				zfs_btree_remove_idx(&sv->sv_pair, &where);
242 			}
243 		}
244 	}
245 
246 	return (0);
247 }
248 
249 static int
sublivelist_verify_func(void * args,dsl_deadlist_entry_t * dle)250 sublivelist_verify_func(void *args, dsl_deadlist_entry_t *dle)
251 {
252 	int err;
253 	struct sublivelist_verify *sv = args;
254 
255 	zfs_btree_create(&sv->sv_pair, sublivelist_block_refcnt_compare, NULL,
256 	    sizeof (sublivelist_verify_block_refcnt_t));
257 
258 	err = bpobj_iterate_nofree(&dle->dle_bpobj, sublivelist_verify_blkptr,
259 	    sv, NULL);
260 
261 	sublivelist_verify_block_refcnt_t *e;
262 	zfs_btree_index_t *cookie = NULL;
263 	while ((e = zfs_btree_destroy_nodes(&sv->sv_pair, &cookie)) != NULL) {
264 		char blkbuf[BP_SPRINTF_LEN];
265 		snprintf_blkptr_compact(blkbuf, sizeof (blkbuf),
266 		    &e->svbr_blk, B_TRUE);
267 		(void) printf("\tERROR: %d unmatched FREE(s): %s\n",
268 		    e->svbr_refcnt, blkbuf);
269 		corruption_found = B_TRUE;
270 	}
271 	zfs_btree_destroy(&sv->sv_pair);
272 
273 	return (err);
274 }
275 
276 static int
livelist_block_compare(const void * larg,const void * rarg)277 livelist_block_compare(const void *larg, const void *rarg)
278 {
279 	const sublivelist_verify_block_t *l = larg;
280 	const sublivelist_verify_block_t *r = rarg;
281 
282 	if (DVA_GET_VDEV(&l->svb_dva) < DVA_GET_VDEV(&r->svb_dva))
283 		return (-1);
284 	else if (DVA_GET_VDEV(&l->svb_dva) > DVA_GET_VDEV(&r->svb_dva))
285 		return (+1);
286 
287 	if (DVA_GET_OFFSET(&l->svb_dva) < DVA_GET_OFFSET(&r->svb_dva))
288 		return (-1);
289 	else if (DVA_GET_OFFSET(&l->svb_dva) > DVA_GET_OFFSET(&r->svb_dva))
290 		return (+1);
291 
292 	if (DVA_GET_ASIZE(&l->svb_dva) < DVA_GET_ASIZE(&r->svb_dva))
293 		return (-1);
294 	else if (DVA_GET_ASIZE(&l->svb_dva) > DVA_GET_ASIZE(&r->svb_dva))
295 		return (+1);
296 
297 	return (0);
298 }
299 
300 /*
301  * Check for errors in a livelist while tracking all unfreed ALLOCs in the
302  * sublivelist_verify_t: sv->sv_leftover
303  */
304 static void
livelist_verify(dsl_deadlist_t * dl,void * arg)305 livelist_verify(dsl_deadlist_t *dl, void *arg)
306 {
307 	sublivelist_verify_t *sv = arg;
308 	dsl_deadlist_iterate(dl, sublivelist_verify_func, sv);
309 }
310 
311 /*
312  * Check for errors in the livelist entry and discard the intermediary
313  * data structures
314  */
315 static int
sublivelist_verify_lightweight(void * args,dsl_deadlist_entry_t * dle)316 sublivelist_verify_lightweight(void *args, dsl_deadlist_entry_t *dle)
317 {
318 	(void) args;
319 	sublivelist_verify_t sv;
320 	zfs_btree_create(&sv.sv_leftover, livelist_block_compare, NULL,
321 	    sizeof (sublivelist_verify_block_t));
322 	int err = sublivelist_verify_func(&sv, dle);
323 	zfs_btree_clear(&sv.sv_leftover);
324 	zfs_btree_destroy(&sv.sv_leftover);
325 	return (err);
326 }
327 
328 typedef struct metaslab_verify {
329 	/*
330 	 * Tree containing all the leftover ALLOCs from the livelists
331 	 * that are part of this metaslab.
332 	 */
333 	zfs_btree_t mv_livelist_allocs;
334 
335 	/*
336 	 * Metaslab information.
337 	 */
338 	uint64_t mv_vdid;
339 	uint64_t mv_msid;
340 	uint64_t mv_start;
341 	uint64_t mv_end;
342 
343 	/*
344 	 * What's currently allocated for this metaslab.
345 	 */
346 	zfs_range_tree_t *mv_allocated;
347 } metaslab_verify_t;
348 
349 typedef void ll_iter_t(dsl_deadlist_t *ll, void *arg);
350 
351 typedef int (*zdb_log_sm_cb_t)(spa_t *spa, space_map_entry_t *sme, uint64_t txg,
352     void *arg);
353 
354 typedef struct unflushed_iter_cb_arg {
355 	spa_t *uic_spa;
356 	uint64_t uic_txg;
357 	void *uic_arg;
358 	zdb_log_sm_cb_t uic_cb;
359 } unflushed_iter_cb_arg_t;
360 
361 static int
iterate_through_spacemap_logs_cb(space_map_entry_t * sme,void * arg)362 iterate_through_spacemap_logs_cb(space_map_entry_t *sme, void *arg)
363 {
364 	unflushed_iter_cb_arg_t *uic = arg;
365 	return (uic->uic_cb(uic->uic_spa, sme, uic->uic_txg, uic->uic_arg));
366 }
367 
368 static void
iterate_through_spacemap_logs(spa_t * spa,zdb_log_sm_cb_t cb,void * arg)369 iterate_through_spacemap_logs(spa_t *spa, zdb_log_sm_cb_t cb, void *arg)
370 {
371 	if (!spa_feature_is_active(spa, SPA_FEATURE_LOG_SPACEMAP))
372 		return;
373 
374 	spa_config_enter(spa, SCL_CONFIG, FTAG, RW_READER);
375 	for (spa_log_sm_t *sls = avl_first(&spa->spa_sm_logs_by_txg);
376 	    sls; sls = AVL_NEXT(&spa->spa_sm_logs_by_txg, sls)) {
377 		space_map_t *sm = NULL;
378 		VERIFY0(space_map_open(&sm, spa_meta_objset(spa),
379 		    sls->sls_sm_obj, 0, UINT64_MAX, SPA_MINBLOCKSHIFT));
380 
381 		unflushed_iter_cb_arg_t uic = {
382 			.uic_spa = spa,
383 			.uic_txg = sls->sls_txg,
384 			.uic_arg = arg,
385 			.uic_cb = cb
386 		};
387 		VERIFY0(space_map_iterate(sm, space_map_length(sm),
388 		    iterate_through_spacemap_logs_cb, &uic));
389 		space_map_close(sm);
390 	}
391 	spa_config_exit(spa, SCL_CONFIG, FTAG);
392 }
393 
394 static void
verify_livelist_allocs(metaslab_verify_t * mv,uint64_t txg,uint64_t offset,uint64_t size)395 verify_livelist_allocs(metaslab_verify_t *mv, uint64_t txg,
396     uint64_t offset, uint64_t size)
397 {
398 	sublivelist_verify_block_t svb = {{{0}}};
399 	DVA_SET_VDEV(&svb.svb_dva, mv->mv_vdid);
400 	DVA_SET_OFFSET(&svb.svb_dva, offset);
401 	DVA_SET_ASIZE(&svb.svb_dva, 0);
402 	zfs_btree_index_t where;
403 	uint64_t end_offset = offset + size;
404 
405 	/*
406 	 *  Look for an exact match for spacemap entry in the livelist entries.
407 	 *  Then, look for other livelist entries that fall within the range
408 	 *  of the spacemap entry as it may have been condensed
409 	 */
410 	sublivelist_verify_block_t *found =
411 	    zfs_btree_find(&mv->mv_livelist_allocs, &svb, &where);
412 	if (found == NULL) {
413 		found = zfs_btree_next(&mv->mv_livelist_allocs, &where, &where);
414 	}
415 	for (; found != NULL && DVA_GET_VDEV(&found->svb_dva) == mv->mv_vdid &&
416 	    DVA_GET_OFFSET(&found->svb_dva) < end_offset;
417 	    found = zfs_btree_next(&mv->mv_livelist_allocs, &where, &where)) {
418 		if (found->svb_allocated_txg <= txg) {
419 			(void) printf("ERROR: Livelist ALLOC [%llx:%llx] "
420 			    "from TXG %llx FREED at TXG %llx\n",
421 			    (u_longlong_t)DVA_GET_OFFSET(&found->svb_dva),
422 			    (u_longlong_t)DVA_GET_ASIZE(&found->svb_dva),
423 			    (u_longlong_t)found->svb_allocated_txg,
424 			    (u_longlong_t)txg);
425 			corruption_found = B_TRUE;
426 		}
427 	}
428 }
429 
430 static int
metaslab_spacemap_validation_cb(space_map_entry_t * sme,void * arg)431 metaslab_spacemap_validation_cb(space_map_entry_t *sme, void *arg)
432 {
433 	metaslab_verify_t *mv = arg;
434 	uint64_t offset = sme->sme_offset;
435 	uint64_t size = sme->sme_run;
436 	uint64_t txg = sme->sme_txg;
437 
438 	if (sme->sme_type == SM_ALLOC) {
439 		if (zfs_range_tree_contains(mv->mv_allocated,
440 		    offset, size)) {
441 			(void) printf("ERROR: DOUBLE ALLOC: "
442 			    "%llu [%llx:%llx] "
443 			    "%llu:%llu LOG_SM\n",
444 			    (u_longlong_t)txg, (u_longlong_t)offset,
445 			    (u_longlong_t)size, (u_longlong_t)mv->mv_vdid,
446 			    (u_longlong_t)mv->mv_msid);
447 			corruption_found = B_TRUE;
448 		} else {
449 			zfs_range_tree_add(mv->mv_allocated,
450 			    offset, size);
451 		}
452 	} else {
453 		if (!zfs_range_tree_contains(mv->mv_allocated,
454 		    offset, size)) {
455 			(void) printf("ERROR: DOUBLE FREE: "
456 			    "%llu [%llx:%llx] "
457 			    "%llu:%llu LOG_SM\n",
458 			    (u_longlong_t)txg, (u_longlong_t)offset,
459 			    (u_longlong_t)size, (u_longlong_t)mv->mv_vdid,
460 			    (u_longlong_t)mv->mv_msid);
461 			corruption_found = B_TRUE;
462 		} else {
463 			zfs_range_tree_remove(mv->mv_allocated,
464 			    offset, size);
465 		}
466 	}
467 
468 	if (sme->sme_type != SM_ALLOC) {
469 		/*
470 		 * If something is freed in the spacemap, verify that
471 		 * it is not listed as allocated in the livelist.
472 		 */
473 		verify_livelist_allocs(mv, txg, offset, size);
474 	}
475 	return (0);
476 }
477 
478 static int
spacemap_check_sm_log_cb(spa_t * spa,space_map_entry_t * sme,uint64_t txg,void * arg)479 spacemap_check_sm_log_cb(spa_t *spa, space_map_entry_t *sme,
480     uint64_t txg, void *arg)
481 {
482 	metaslab_verify_t *mv = arg;
483 	uint64_t offset = sme->sme_offset;
484 	uint64_t vdev_id = sme->sme_vdev;
485 
486 	vdev_t *vd = vdev_lookup_top(spa, vdev_id);
487 
488 	/* skip indirect vdevs */
489 	if (!vdev_is_concrete(vd))
490 		return (0);
491 
492 	if (vdev_id != mv->mv_vdid)
493 		return (0);
494 
495 	metaslab_t *ms = vd->vdev_ms[offset >> vd->vdev_ms_shift];
496 	if (ms->ms_id != mv->mv_msid)
497 		return (0);
498 
499 	if (txg < metaslab_unflushed_txg(ms))
500 		return (0);
501 
502 
503 	ASSERT3U(txg, ==, sme->sme_txg);
504 	return (metaslab_spacemap_validation_cb(sme, mv));
505 }
506 
507 static void
spacemap_check_sm_log(spa_t * spa,metaslab_verify_t * mv)508 spacemap_check_sm_log(spa_t *spa, metaslab_verify_t *mv)
509 {
510 	iterate_through_spacemap_logs(spa, spacemap_check_sm_log_cb, mv);
511 }
512 
513 static void
spacemap_check_ms_sm(space_map_t * sm,metaslab_verify_t * mv)514 spacemap_check_ms_sm(space_map_t  *sm, metaslab_verify_t *mv)
515 {
516 	if (sm == NULL)
517 		return;
518 
519 	VERIFY0(space_map_iterate(sm, space_map_length(sm),
520 	    metaslab_spacemap_validation_cb, mv));
521 }
522 
523 static void iterate_deleted_livelists(spa_t *spa, ll_iter_t func, void *arg);
524 
525 /*
526  * Transfer blocks from sv_leftover tree to the mv_livelist_allocs if
527  * they are part of that metaslab (mv_msid).
528  */
529 static void
mv_populate_livelist_allocs(metaslab_verify_t * mv,sublivelist_verify_t * sv)530 mv_populate_livelist_allocs(metaslab_verify_t *mv, sublivelist_verify_t *sv)
531 {
532 	zfs_btree_index_t where;
533 	sublivelist_verify_block_t *svb;
534 	ASSERT3U(zfs_btree_numnodes(&mv->mv_livelist_allocs), ==, 0);
535 	for (svb = zfs_btree_first(&sv->sv_leftover, &where);
536 	    svb != NULL;
537 	    svb = zfs_btree_next(&sv->sv_leftover, &where, &where)) {
538 		if (DVA_GET_VDEV(&svb->svb_dva) != mv->mv_vdid)
539 			continue;
540 
541 		if (DVA_GET_OFFSET(&svb->svb_dva) < mv->mv_start &&
542 		    (DVA_GET_OFFSET(&svb->svb_dva) +
543 		    DVA_GET_ASIZE(&svb->svb_dva)) > mv->mv_start) {
544 			(void) printf("ERROR: Found block that crosses "
545 			    "metaslab boundary: <%llu:%llx:%llx>\n",
546 			    (u_longlong_t)DVA_GET_VDEV(&svb->svb_dva),
547 			    (u_longlong_t)DVA_GET_OFFSET(&svb->svb_dva),
548 			    (u_longlong_t)DVA_GET_ASIZE(&svb->svb_dva));
549 			corruption_found = B_TRUE;
550 			continue;
551 		}
552 
553 		if (DVA_GET_OFFSET(&svb->svb_dva) < mv->mv_start)
554 			continue;
555 
556 		if (DVA_GET_OFFSET(&svb->svb_dva) >= mv->mv_end)
557 			continue;
558 
559 		if ((DVA_GET_OFFSET(&svb->svb_dva) +
560 		    DVA_GET_ASIZE(&svb->svb_dva)) > mv->mv_end) {
561 			(void) printf("ERROR: Found block that crosses "
562 			    "metaslab boundary: <%llu:%llx:%llx>\n",
563 			    (u_longlong_t)DVA_GET_VDEV(&svb->svb_dva),
564 			    (u_longlong_t)DVA_GET_OFFSET(&svb->svb_dva),
565 			    (u_longlong_t)DVA_GET_ASIZE(&svb->svb_dva));
566 			corruption_found = B_TRUE;
567 			continue;
568 		}
569 
570 		zfs_btree_add(&mv->mv_livelist_allocs, svb);
571 	}
572 
573 	for (svb = zfs_btree_first(&mv->mv_livelist_allocs, &where);
574 	    svb != NULL;
575 	    svb = zfs_btree_next(&mv->mv_livelist_allocs, &where, &where)) {
576 		zfs_btree_remove(&sv->sv_leftover, svb);
577 	}
578 }
579 
580 /*
581  * [Livelist Check]
582  * Iterate through all the sublivelists and:
583  * - report leftover frees (**)
584  * - record leftover ALLOCs together with their TXG [see Cross Check]
585  *
586  * (**) Note: Double ALLOCs are valid in datasets that have dedup
587  *      enabled. Similarly double FREEs are allowed as well but
588  *      only if they pair up with a corresponding ALLOC entry once
589  *      we our done with our sublivelist iteration.
590  *
591  * [Spacemap Check]
592  * for each metaslab:
593  * - iterate over spacemap and then the metaslab's entries in the
594  *   spacemap log, then report any double FREEs and ALLOCs (do not
595  *   blow up).
596  *
597  * [Cross Check]
598  * After finishing the Livelist Check phase and while being in the
599  * Spacemap Check phase, we find all the recorded leftover ALLOCs
600  * of the livelist check that are part of the metaslab that we are
601  * currently looking at in the Spacemap Check. We report any entries
602  * that are marked as ALLOCs in the livelists but have been actually
603  * freed (and potentially allocated again) after their TXG stamp in
604  * the spacemaps. Also report any ALLOCs from the livelists that
605  * belong to indirect vdevs (e.g. their vdev completed removal).
606  *
607  * Note that this will miss Log Spacemap entries that cancelled each other
608  * out before being flushed to the metaslab, so we are not guaranteed
609  * to match all erroneous ALLOCs.
610  */
611 static void
livelist_metaslab_validate(spa_t * spa)612 livelist_metaslab_validate(spa_t *spa)
613 {
614 	(void) printf("Verifying deleted livelist entries\n");
615 
616 	sublivelist_verify_t sv;
617 	zfs_btree_create(&sv.sv_leftover, livelist_block_compare, NULL,
618 	    sizeof (sublivelist_verify_block_t));
619 	iterate_deleted_livelists(spa, livelist_verify, &sv);
620 
621 	(void) printf("Verifying metaslab entries\n");
622 	vdev_t *rvd = spa->spa_root_vdev;
623 	for (uint64_t c = 0; c < rvd->vdev_children; c++) {
624 		vdev_t *vd = rvd->vdev_child[c];
625 
626 		if (!vdev_is_concrete(vd))
627 			continue;
628 
629 		for (uint64_t mid = 0; mid < vd->vdev_ms_count; mid++) {
630 			metaslab_t *m = vd->vdev_ms[mid];
631 
632 			(void) fprintf(stderr,
633 			    "\rverifying concrete vdev %llu, "
634 			    "metaslab %llu of %llu ...",
635 			    (longlong_t)vd->vdev_id,
636 			    (longlong_t)mid,
637 			    (longlong_t)vd->vdev_ms_count);
638 
639 			uint64_t shift, start;
640 			zfs_range_seg_type_t type =
641 			    metaslab_calculate_range_tree_type(vd, m,
642 			    &start, &shift);
643 			metaslab_verify_t mv;
644 			mv.mv_allocated = zfs_range_tree_create_flags(
645 			    NULL, type, NULL, start, shift,
646 			    0, "livelist_metaslab_validate:mv_allocated");
647 			mv.mv_vdid = vd->vdev_id;
648 			mv.mv_msid = m->ms_id;
649 			mv.mv_start = m->ms_start;
650 			mv.mv_end = m->ms_start + m->ms_size;
651 			zfs_btree_create(&mv.mv_livelist_allocs,
652 			    livelist_block_compare, NULL,
653 			    sizeof (sublivelist_verify_block_t));
654 
655 			mv_populate_livelist_allocs(&mv, &sv);
656 
657 			spacemap_check_ms_sm(m->ms_sm, &mv);
658 			spacemap_check_sm_log(spa, &mv);
659 
660 			zfs_range_tree_vacate(mv.mv_allocated, NULL, NULL);
661 			zfs_range_tree_destroy(mv.mv_allocated);
662 			zfs_btree_clear(&mv.mv_livelist_allocs);
663 			zfs_btree_destroy(&mv.mv_livelist_allocs);
664 		}
665 	}
666 	(void) fprintf(stderr, "\n");
667 
668 	/*
669 	 * If there are any segments in the leftover tree after we walked
670 	 * through all the metaslabs in the concrete vdevs then this means
671 	 * that we have segments in the livelists that belong to indirect
672 	 * vdevs and are marked as allocated.
673 	 */
674 	if (zfs_btree_numnodes(&sv.sv_leftover) == 0) {
675 		zfs_btree_destroy(&sv.sv_leftover);
676 		return;
677 	}
678 	(void) printf("ERROR: Found livelist blocks marked as allocated "
679 	    "for indirect vdevs:\n");
680 	corruption_found = B_TRUE;
681 
682 	zfs_btree_index_t *where = NULL;
683 	sublivelist_verify_block_t *svb;
684 	while ((svb = zfs_btree_destroy_nodes(&sv.sv_leftover, &where)) !=
685 	    NULL) {
686 		int vdev_id = DVA_GET_VDEV(&svb->svb_dva);
687 		ASSERT3U(vdev_id, <, rvd->vdev_children);
688 		vdev_t *vd = rvd->vdev_child[vdev_id];
689 		ASSERT(!vdev_is_concrete(vd));
690 		(void) printf("<%d:%llx:%llx> TXG %llx\n",
691 		    vdev_id, (u_longlong_t)DVA_GET_OFFSET(&svb->svb_dva),
692 		    (u_longlong_t)DVA_GET_ASIZE(&svb->svb_dva),
693 		    (u_longlong_t)svb->svb_allocated_txg);
694 	}
695 	(void) printf("\n");
696 	zfs_btree_destroy(&sv.sv_leftover);
697 }
698 
699 /*
700  * These libumem hooks provide a reasonable set of defaults for the allocator's
701  * debugging facilities.
702  */
703 const char *
_umem_debug_init(void)704 _umem_debug_init(void)
705 {
706 	return ("default,verbose"); /* $UMEM_DEBUG setting */
707 }
708 
709 const char *
_umem_logging_init(void)710 _umem_logging_init(void)
711 {
712 	return ("fail,contents"); /* $UMEM_LOGGING setting */
713 }
714 
715 static void
usage(void)716 usage(void)
717 {
718 	(void) fprintf(stderr,
719 	    "Usage:\t%s [-AbcdDFGhikLMPsvXy] [-e [-V] [-p <path> ...]] "
720 	    "[-I <inflight I/Os>]\n"
721 	    "\t\t[-o <var>=<value>]... [-t <txg>] [-U <cache>] [-x <dumpdir>]\n"
722 	    "\t\t[-K <key>]\n"
723 	    "\t\t[<poolname>[/<dataset | objset id>] [<object | range> ...]]\n"
724 	    "\t%s [-AdiPv] [-e [-V] [-p <path> ...]] [-U <cache>] [-K <key>]\n"
725 	    "\t\t[<poolname>[/<dataset | objset id>] [<object | range> ...]\n"
726 	    "\t%s -B [-e [-V] [-p <path> ...]] [-I <inflight I/Os>]\n"
727 	    "\t\t[-o <var>=<value>]... [-t <txg>] [-U <cache>] [-x <dumpdir>]\n"
728 	    "\t\t[-K <key>] <poolname>/<objset id> [<backupflags>]\n"
729 	    "\t%s [-v] <bookmark>\n"
730 	    "\t%s -C [-A] [-U <cache>] [<poolname>]\n"
731 	    "\t%s -l [-Aqu] <device>\n"
732 	    "\t%s -m [-AFLPX] [-e [-V] [-p <path> ...]] [-t <txg>] "
733 	    "[-U <cache>]\n\t\t<poolname> [<vdev> [<metaslab> ...]]\n"
734 	    "\t%s -O [-K <key>] <dataset> <path>\n"
735 	    "\t%s -r [-K <key>] <dataset> <path> <destination>\n"
736 	    "\t%s -r [-K <key>] -O <dataset> <object-id> <destination>\n"
737 	    "\t%s -R [-A] [-e [-V] [-p <path> ...]] [-U <cache>]\n"
738 	    "\t\t<poolname> <vdev>:<offset>:<size>[:<flags>]\n"
739 	    "\t%s -f [-H] [-e [-V] [-p <path> ...]] [-U <cache>]\n"
740 	    "\t\t[<poolname>[/<dataset | objset id>] [<object | range> ...]]\n"
741 	    "\t%s -E [-A] word0:word1:...:word15\n"
742 	    "\t%s -S [-AP] [-e [-V] [-p <path> ...]] [-U <cache>] "
743 	    "<poolname>\n\n",
744 	    cmdname, cmdname, cmdname, cmdname, cmdname, cmdname, cmdname,
745 	    cmdname, cmdname, cmdname, cmdname, cmdname, cmdname, cmdname);
746 
747 	(void) fprintf(stderr, "    Dataset name must include at least one "
748 	    "separator character '/' or '@'\n");
749 	(void) fprintf(stderr, "    If dataset name is specified, only that "
750 	    "dataset is dumped\n");
751 	(void) fprintf(stderr,  "    If object numbers or object number "
752 	    "ranges are specified, only those\n"
753 	    "    objects or ranges are dumped.\n\n");
754 	(void) fprintf(stderr,
755 	    "    Object ranges take the form <start>:<end>[:<flags>]\n"
756 	    "        start    Starting object number\n"
757 	    "        end      Ending object number, or -1 for no upper bound\n"
758 	    "        flags    Optional flags to select object types:\n"
759 	    "            A     All objects (this is the default)\n"
760 	    "            d     ZFS directories\n"
761 	    "            f     ZFS files \n"
762 	    "            m     SPA space maps\n"
763 	    "            v     ZVols\n"
764 	    "            z     ZAPs\n"
765 	    "            -     Negate effect of next flag\n\n");
766 	(void) fprintf(stderr, "    Options to control amount of output:\n");
767 	(void) fprintf(stderr, "        -b --block-stats             "
768 	    "block statistics\n");
769 	(void) fprintf(stderr, "           --bin=(lsize|psize|asize) "
770 	    "bin blocks based on this size in all three columns\n");
771 	(void) fprintf(stderr,
772 	    "           --class=(normal|special|dedup|other)[,...]\n"
773 	    "                                     only consider blocks from "
774 	    "these allocation classes\n");
775 	(void) fprintf(stderr, "        -B --backup                  "
776 	    "backup stream\n");
777 	(void) fprintf(stderr, "        -c --checksum                "
778 	    "checksum all metadata (twice for all data) blocks\n");
779 	(void) fprintf(stderr, "        -C --config                  "
780 	    "config (or cachefile if alone)\n");
781 	(void) fprintf(stderr, "        -d --datasets                "
782 	    "dataset(s)\n");
783 	(void) fprintf(stderr, "        -D --dedup-stats             "
784 	    "dedup statistics\n");
785 	(void) fprintf(stderr, "        -f --file-layout             "
786 	    "display a file's layout across raidz disks\n");
787 	(void) fprintf(stderr, "        -E --embedded-block-pointer=INTEGER\n"
788 	    "                                     decode and display block "
789 	    "from an embedded block pointer\n");
790 	(void) fprintf(stderr, "        -h --history                 "
791 	    "pool history\n");
792 	(void) fprintf(stderr, "        -i --intent-logs             "
793 	    "intent logs\n");
794 	(void) fprintf(stderr, "        -l --label                   "
795 	    "read label contents\n");
796 	(void) fprintf(stderr, "        -k --checkpointed-state      "
797 	    "examine the checkpointed state of the pool\n");
798 	(void) fprintf(stderr, "        -L --disable-leak-tracking   "
799 	    "disable leak tracking (do not load spacemaps)\n");
800 	(void) fprintf(stderr, "        -m --metaslabs               "
801 	    "metaslabs\n");
802 	(void) fprintf(stderr, "        -M --metaslab-groups         "
803 	    "metaslab groups\n");
804 	(void) fprintf(stderr, "        -O --object-lookups          "
805 	    "perform object lookups by path\n");
806 	(void) fprintf(stderr, "        -r --copy-object             "
807 	    "copy an object by path to file\n");
808 	(void) fprintf(stderr, "        -R --read-block              "
809 	    "read and display block from a device\n");
810 	(void) fprintf(stderr, "        -s --io-stats                "
811 	    "report stats on zdb's I/O\n");
812 	(void) fprintf(stderr, "        -S --simulate-dedup          "
813 	    "simulate dedup to measure effect\n");
814 	(void) fprintf(stderr, "        -v --verbose                 "
815 	    "verbose (applies to all others)\n");
816 	(void) fprintf(stderr, "        -y --livelist                "
817 	    "perform livelist and metaslab validation on any livelists being "
818 	    "deleted\n\n");
819 
820 	(void) fprintf(stderr, "    Below options are intended for use "
821 	    "with other options:\n");
822 	(void) fprintf(stderr, "        -A --ignore-assertions       "
823 	    "ignore assertions (-A), enable panic recovery (-AA) or both "
824 	    "(-AAA)\n");
825 	(void) fprintf(stderr, "        -e --exported                "
826 	    "pool is exported/destroyed/has altroot/not in a cachefile\n");
827 	(void) fprintf(stderr, "        -F --automatic-rewind        "
828 	    "attempt automatic rewind within safe range of transaction "
829 	    "groups\n");
830 	(void) fprintf(stderr, "        -G --dump-debug-msg          "
831 	    "dump zfs_dbgmsg buffer before exiting\n");
832 	(void) fprintf(stderr, "        -H --scripting-mode          "
833 	    "dump output in a format that is scripting friendly\n");
834 	(void) fprintf(stderr, "        -I --inflight=INTEGER        "
835 	    "specify the maximum number of checksumming I/Os "
836 	    "[default is 200]\n");
837 	(void) fprintf(stderr, "        -K --key=KEY                 "
838 	    "decryption key for encrypted dataset\n");
839 	(void) fprintf(stderr, "        -o --option=\"NAME=VALUE\" "
840 	    "set the named tunable to the given value\n");
841 	(void) fprintf(stderr, "        -p --path==PATH              "
842 	    "use one or more with -e to specify path to vdev dir\n");
843 	(void) fprintf(stderr, "        -P --parseable               "
844 	    "print numbers in parseable form\n");
845 	(void) fprintf(stderr, "        -q --skip-label              "
846 	    "don't print label contents\n");
847 	(void) fprintf(stderr, "        -t --txg=INTEGER             "
848 	    "highest txg to use when searching for uberblocks\n");
849 	(void) fprintf(stderr, "        -T --brt-stats               "
850 	    "BRT statistics\n");
851 	(void) fprintf(stderr, "        -u --uberblock               "
852 	    "uberblock\n");
853 	(void) fprintf(stderr, "        -U --cachefile=PATH          "
854 	    "use alternate cachefile\n");
855 	(void) fprintf(stderr, "        -V --verbatim                "
856 	    "do verbatim import\n");
857 	(void) fprintf(stderr, "        -x --dump-blocks=PATH        "
858 	    "dump all read blocks into specified directory\n");
859 	(void) fprintf(stderr, "        -X --extreme-rewind          "
860 	    "attempt extreme rewind (does not work with dataset)\n");
861 	(void) fprintf(stderr, "        -Y --all-reconstruction      "
862 	    "attempt all reconstruction combinations for split blocks\n");
863 	(void) fprintf(stderr, "        -Z --zstd-headers            "
864 	    "show ZSTD headers \n");
865 	(void) fprintf(stderr, "Specify an option more than once (e.g. -bb) "
866 	    "to make only that option verbose\n");
867 	(void) fprintf(stderr, "Default is to dump everything non-verbosely\n");
868 	zdb_exit(2);
869 }
870 
871 static void
dump_debug_buffer(void)872 dump_debug_buffer(void)
873 {
874 	ssize_t ret __attribute__((unused));
875 
876 	if (!dump_opt['G'])
877 		return;
878 	/*
879 	 * We use write() instead of printf() so that this function
880 	 * is safe to call from a signal handler.
881 	 */
882 	ret = write(STDERR_FILENO, "\n", 1);
883 	zfs_dbgmsg_print(STDERR_FILENO, "zdb");
884 }
885 
sig_handler(int signo)886 static void sig_handler(int signo)
887 {
888 	struct sigaction action;
889 
890 	libspl_backtrace(STDERR_FILENO);
891 	dump_debug_buffer();
892 
893 	/*
894 	 * Restore default action and re-raise signal so SIGSEGV and
895 	 * SIGABRT can trigger a core dump.
896 	 */
897 	action.sa_handler = SIG_DFL;
898 	sigemptyset(&action.sa_mask);
899 	action.sa_flags = 0;
900 	(void) sigaction(signo, &action, NULL);
901 	raise(signo);
902 }
903 
904 /*
905  * Called for usage errors that are discovered after a call to spa_open(),
906  * dmu_bonus_hold(), or pool_match().  abort() is called for other errors.
907  */
908 
909 static void
fatal(const char * fmt,...)910 fatal(const char *fmt, ...)
911 {
912 	va_list ap;
913 
914 	va_start(ap, fmt);
915 	(void) fprintf(stderr, "%s: ", cmdname);
916 	(void) vfprintf(stderr, fmt, ap);
917 	va_end(ap);
918 	(void) fprintf(stderr, "\n");
919 
920 	dump_debug_buffer();
921 
922 	zdb_exit(1);
923 }
924 
925 static void
dump_packed_nvlist(objset_t * os,uint64_t object,void * data,size_t size)926 dump_packed_nvlist(objset_t *os, uint64_t object, void *data, size_t size)
927 {
928 	(void) size;
929 	nvlist_t *nv;
930 	size_t nvsize = *(uint64_t *)data;
931 	char *packed = umem_alloc(nvsize, UMEM_NOFAIL);
932 	int err;
933 
934 	err = dmu_read(os, object, 0, nvsize, packed, DMU_READ_PREFETCH);
935 	if (err != 0) {
936 		(void) printf("got error %u from dmu_read\n", err);
937 		umem_free(packed, nvsize);
938 		return;
939 	}
940 
941 	err = nvlist_unpack(packed, nvsize, &nv, 0);
942 
943 	umem_free(packed, nvsize);
944 
945 	if (err != 0) {
946 		(void) printf("got error %u from nvlist_unpack\n", err);
947 		return;
948 	}
949 
950 	dump_nvlist(nv, 8);
951 
952 	nvlist_free(nv);
953 }
954 
955 static void
dump_history_offsets(objset_t * os,uint64_t object,void * data,size_t size)956 dump_history_offsets(objset_t *os, uint64_t object, void *data, size_t size)
957 {
958 	(void) os, (void) object, (void) size;
959 	spa_history_phys_t *shp = data;
960 
961 	if (shp == NULL)
962 		return;
963 
964 	(void) printf("\t\tpool_create_len = %llu\n",
965 	    (u_longlong_t)shp->sh_pool_create_len);
966 	(void) printf("\t\tphys_max_off = %llu\n",
967 	    (u_longlong_t)shp->sh_phys_max_off);
968 	(void) printf("\t\tbof = %llu\n",
969 	    (u_longlong_t)shp->sh_bof);
970 	(void) printf("\t\teof = %llu\n",
971 	    (u_longlong_t)shp->sh_eof);
972 	(void) printf("\t\trecords_lost = %llu\n",
973 	    (u_longlong_t)shp->sh_records_lost);
974 }
975 
976 static void
zdb_nicenum(uint64_t num,char * buf,size_t buflen)977 zdb_nicenum(uint64_t num, char *buf, size_t buflen)
978 {
979 	if (dump_opt['P'])
980 		(void) snprintf(buf, buflen, "%llu", (longlong_t)num);
981 	else
982 		nicenum(num, buf, buflen);
983 }
984 
985 static void
zdb_nicebytes(uint64_t bytes,char * buf,size_t buflen)986 zdb_nicebytes(uint64_t bytes, char *buf, size_t buflen)
987 {
988 	if (dump_opt['P'])
989 		(void) snprintf(buf, buflen, "%llu", (longlong_t)bytes);
990 	else
991 		zfs_nicebytes(bytes, buf, buflen);
992 }
993 
994 static const char histo_stars[] = "****************************************";
995 static const uint64_t histo_width = sizeof (histo_stars) - 1;
996 
997 static void
dump_histogram(const uint64_t * histo,int size,int offset)998 dump_histogram(const uint64_t *histo, int size, int offset)
999 {
1000 	int i;
1001 	int minidx = size - 1;
1002 	int maxidx = 0;
1003 	uint64_t max = 0;
1004 
1005 	for (i = 0; i < size; i++) {
1006 		if (histo[i] == 0)
1007 			continue;
1008 		if (histo[i] > max)
1009 			max = histo[i];
1010 		if (i > maxidx)
1011 			maxidx = i;
1012 		if (i < minidx)
1013 			minidx = i;
1014 	}
1015 
1016 	if (max < histo_width)
1017 		max = histo_width;
1018 
1019 	for (i = minidx; i <= maxidx; i++) {
1020 		(void) printf("\t\t\t%3u: %6llu %s\n",
1021 		    i + offset, (u_longlong_t)histo[i],
1022 		    &histo_stars[(max - histo[i]) * histo_width / max]);
1023 	}
1024 }
1025 
1026 static void
dump_zap_stats(objset_t * os,uint64_t object)1027 dump_zap_stats(objset_t *os, uint64_t object)
1028 {
1029 	int error;
1030 	zap_stats_t zs;
1031 
1032 	error = zap_get_stats(os, object, &zs);
1033 	if (error)
1034 		return;
1035 
1036 	if (zs.zs_ptrtbl_len == 0) {
1037 		ASSERT(zs.zs_num_blocks == 1);
1038 		(void) printf("\tmicrozap: %llu bytes, %llu entries\n",
1039 		    (u_longlong_t)zs.zs_blocksize,
1040 		    (u_longlong_t)zs.zs_num_entries);
1041 		return;
1042 	}
1043 
1044 	(void) printf("\tFat ZAP stats:\n");
1045 
1046 	(void) printf("\t\tPointer table:\n");
1047 	(void) printf("\t\t\t%llu elements\n",
1048 	    (u_longlong_t)zs.zs_ptrtbl_len);
1049 	(void) printf("\t\t\tzt_blk: %llu\n",
1050 	    (u_longlong_t)zs.zs_ptrtbl_zt_blk);
1051 	(void) printf("\t\t\tzt_numblks: %llu\n",
1052 	    (u_longlong_t)zs.zs_ptrtbl_zt_numblks);
1053 	(void) printf("\t\t\tzt_shift: %llu\n",
1054 	    (u_longlong_t)zs.zs_ptrtbl_zt_shift);
1055 	(void) printf("\t\t\tzt_blks_copied: %llu\n",
1056 	    (u_longlong_t)zs.zs_ptrtbl_blks_copied);
1057 	(void) printf("\t\t\tzt_nextblk: %llu\n",
1058 	    (u_longlong_t)zs.zs_ptrtbl_nextblk);
1059 
1060 	(void) printf("\t\tZAP entries: %llu\n",
1061 	    (u_longlong_t)zs.zs_num_entries);
1062 	(void) printf("\t\tLeaf blocks: %llu\n",
1063 	    (u_longlong_t)zs.zs_num_leafs);
1064 	(void) printf("\t\tTotal blocks: %llu\n",
1065 	    (u_longlong_t)zs.zs_num_blocks);
1066 	(void) printf("\t\tzap_block_type: 0x%llx\n",
1067 	    (u_longlong_t)zs.zs_block_type);
1068 	(void) printf("\t\tzap_magic: 0x%llx\n",
1069 	    (u_longlong_t)zs.zs_magic);
1070 	(void) printf("\t\tzap_salt: 0x%llx\n",
1071 	    (u_longlong_t)zs.zs_salt);
1072 
1073 	(void) printf("\t\tLeafs with 2^n pointers:\n");
1074 	dump_histogram(zs.zs_leafs_with_2n_pointers, ZAP_HISTOGRAM_SIZE, 0);
1075 
1076 	(void) printf("\t\tBlocks with n*5 entries:\n");
1077 	dump_histogram(zs.zs_blocks_with_n5_entries, ZAP_HISTOGRAM_SIZE, 0);
1078 
1079 	(void) printf("\t\tBlocks n/10 full:\n");
1080 	dump_histogram(zs.zs_blocks_n_tenths_full, ZAP_HISTOGRAM_SIZE, 0);
1081 
1082 	(void) printf("\t\tEntries with n chunks:\n");
1083 	dump_histogram(zs.zs_entries_using_n_chunks, ZAP_HISTOGRAM_SIZE, 0);
1084 
1085 	(void) printf("\t\tBuckets with n entries:\n");
1086 	dump_histogram(zs.zs_buckets_with_n_entries, ZAP_HISTOGRAM_SIZE, 0);
1087 }
1088 
1089 static void
dump_none(objset_t * os,uint64_t object,void * data,size_t size)1090 dump_none(objset_t *os, uint64_t object, void *data, size_t size)
1091 {
1092 	(void) os, (void) object, (void) data, (void) size;
1093 }
1094 
1095 static void
dump_unknown(objset_t * os,uint64_t object,void * data,size_t size)1096 dump_unknown(objset_t *os, uint64_t object, void *data, size_t size)
1097 {
1098 	(void) os, (void) object, (void) data, (void) size;
1099 	(void) printf("\tUNKNOWN OBJECT TYPE\n");
1100 }
1101 
1102 static void
dump_uint8(objset_t * os,uint64_t object,void * data,size_t size)1103 dump_uint8(objset_t *os, uint64_t object, void *data, size_t size)
1104 {
1105 	(void) os, (void) object, (void) data, (void) size;
1106 }
1107 
1108 static void
dump_uint64(objset_t * os,uint64_t object,void * data,size_t size)1109 dump_uint64(objset_t *os, uint64_t object, void *data, size_t size)
1110 {
1111 	uint64_t *arr;
1112 	uint64_t oursize;
1113 	if (dump_opt['d'] < 6)
1114 		return;
1115 
1116 	if (data == NULL) {
1117 		dmu_object_info_t doi;
1118 
1119 		VERIFY0(dmu_object_info(os, object, &doi));
1120 		size = doi.doi_max_offset;
1121 		/*
1122 		 * We cap the size at 1 mebibyte here to prevent
1123 		 * allocation failures and nigh-infinite printing if the
1124 		 * object is extremely large.
1125 		 */
1126 		oursize = MIN(size, 1 << 20);
1127 		arr = kmem_alloc(oursize, KM_SLEEP);
1128 
1129 		int err = dmu_read(os, object, 0, oursize, arr, 0);
1130 		if (err != 0) {
1131 			(void) printf("got error %u from dmu_read\n", err);
1132 			kmem_free(arr, oursize);
1133 			return;
1134 		}
1135 	} else {
1136 		/*
1137 		 * Even though the allocation is already done in this code path,
1138 		 * we still cap the size to prevent excessive printing.
1139 		 */
1140 		oursize = MIN(size, 1 << 20);
1141 		arr = data;
1142 	}
1143 
1144 	if (size == 0) {
1145 		if (data == NULL)
1146 			kmem_free(arr, oursize);
1147 		(void) printf("\t\t[]\n");
1148 		return;
1149 	}
1150 
1151 	(void) printf("\t\t[%0llx", (u_longlong_t)arr[0]);
1152 	for (size_t i = 1; i * sizeof (uint64_t) < oursize; i++) {
1153 		if (i % 4 != 0)
1154 			(void) printf(", %0llx", (u_longlong_t)arr[i]);
1155 		else
1156 			(void) printf(",\n\t\t%0llx", (u_longlong_t)arr[i]);
1157 	}
1158 	if (oursize != size)
1159 		(void) printf(", ... ");
1160 	(void) printf("]\n");
1161 
1162 	if (data == NULL)
1163 		kmem_free(arr, oursize);
1164 }
1165 
1166 static void
dump_zap(objset_t * os,uint64_t object,void * data,size_t size)1167 dump_zap(objset_t *os, uint64_t object, void *data, size_t size)
1168 {
1169 	(void) data, (void) size;
1170 	zap_cursor_t zc;
1171 	zap_attribute_t *attrp = zap_attribute_long_alloc();
1172 	void *prop;
1173 	unsigned i;
1174 
1175 	dump_zap_stats(os, object);
1176 	(void) printf("\n");
1177 
1178 	for (zap_cursor_init(&zc, os, object);
1179 	    zap_cursor_retrieve(&zc, attrp) == 0;
1180 	    zap_cursor_advance(&zc)) {
1181 		boolean_t key64 =
1182 		    !!(zap_getflags(zc.zc_zap) & ZAP_FLAG_UINT64_KEY);
1183 
1184 		if (key64)
1185 			(void) printf("\t\t0x%010" PRIu64 "x = ",
1186 			    *(uint64_t *)attrp->za_name);
1187 		else
1188 			(void) printf("\t\t%s = ", attrp->za_name);
1189 
1190 		if (attrp->za_num_integers == 0) {
1191 			(void) printf("\n");
1192 			continue;
1193 		}
1194 		prop = umem_zalloc(attrp->za_num_integers *
1195 		    attrp->za_integer_length, UMEM_NOFAIL);
1196 
1197 		if (key64)
1198 			(void) zap_lookup_uint64(os, object,
1199 			    (const uint64_t *)attrp->za_name, 1,
1200 			    attrp->za_integer_length, attrp->za_num_integers,
1201 			    prop);
1202 		else
1203 			(void) zap_lookup(os, object, attrp->za_name,
1204 			    attrp->za_integer_length, attrp->za_num_integers,
1205 			    prop);
1206 
1207 		if (attrp->za_integer_length == 1 && !key64) {
1208 			if (strcmp(attrp->za_name,
1209 			    DSL_CRYPTO_KEY_MASTER_KEY) == 0 ||
1210 			    strcmp(attrp->za_name,
1211 			    DSL_CRYPTO_KEY_HMAC_KEY) == 0 ||
1212 			    strcmp(attrp->za_name, DSL_CRYPTO_KEY_IV) == 0 ||
1213 			    strcmp(attrp->za_name, DSL_CRYPTO_KEY_MAC) == 0 ||
1214 			    strcmp(attrp->za_name,
1215 			    DMU_POOL_CHECKSUM_SALT) == 0) {
1216 				uint8_t *u8 = prop;
1217 
1218 				for (i = 0; i < attrp->za_num_integers; i++) {
1219 					(void) printf("%02x", u8[i]);
1220 				}
1221 			} else {
1222 				(void) printf("%s", (char *)prop);
1223 			}
1224 		} else {
1225 			for (i = 0; i < attrp->za_num_integers; i++) {
1226 				switch (attrp->za_integer_length) {
1227 				case 1:
1228 					(void) printf("%u ",
1229 					    ((uint8_t *)prop)[i]);
1230 					break;
1231 				case 2:
1232 					(void) printf("%u ",
1233 					    ((uint16_t *)prop)[i]);
1234 					break;
1235 				case 4:
1236 					(void) printf("%u ",
1237 					    ((uint32_t *)prop)[i]);
1238 					break;
1239 				case 8:
1240 					(void) printf("%lld ",
1241 					    (u_longlong_t)((int64_t *)prop)[i]);
1242 					break;
1243 				}
1244 			}
1245 		}
1246 		(void) printf("\n");
1247 		umem_free(prop,
1248 		    attrp->za_num_integers * attrp->za_integer_length);
1249 	}
1250 	zap_cursor_fini(&zc);
1251 	zap_attribute_free(attrp);
1252 }
1253 
1254 static void
dump_bpobj(objset_t * os,uint64_t object,void * data,size_t size)1255 dump_bpobj(objset_t *os, uint64_t object, void *data, size_t size)
1256 {
1257 	bpobj_phys_t *bpop = data;
1258 	uint64_t i;
1259 	char bytes[32], comp[32], uncomp[32];
1260 
1261 	/* make sure the output won't get truncated */
1262 	_Static_assert(sizeof (bytes) >= NN_NUMBUF_SZ, "bytes truncated");
1263 	_Static_assert(sizeof (comp) >= NN_NUMBUF_SZ, "comp truncated");
1264 	_Static_assert(sizeof (uncomp) >= NN_NUMBUF_SZ, "uncomp truncated");
1265 
1266 	if (bpop == NULL)
1267 		return;
1268 
1269 	zdb_nicenum(bpop->bpo_bytes, bytes, sizeof (bytes));
1270 	zdb_nicenum(bpop->bpo_comp, comp, sizeof (comp));
1271 	zdb_nicenum(bpop->bpo_uncomp, uncomp, sizeof (uncomp));
1272 
1273 	(void) printf("\t\tnum_blkptrs = %llu\n",
1274 	    (u_longlong_t)bpop->bpo_num_blkptrs);
1275 	(void) printf("\t\tbytes = %s\n", bytes);
1276 	if (size >= BPOBJ_SIZE_V1) {
1277 		(void) printf("\t\tcomp = %s\n", comp);
1278 		(void) printf("\t\tuncomp = %s\n", uncomp);
1279 	}
1280 	if (size >= BPOBJ_SIZE_V2) {
1281 		(void) printf("\t\tsubobjs = %llu\n",
1282 		    (u_longlong_t)bpop->bpo_subobjs);
1283 		(void) printf("\t\tnum_subobjs = %llu\n",
1284 		    (u_longlong_t)bpop->bpo_num_subobjs);
1285 	}
1286 	if (size >= sizeof (*bpop)) {
1287 		(void) printf("\t\tnum_freed = %llu\n",
1288 		    (u_longlong_t)bpop->bpo_num_freed);
1289 	}
1290 
1291 	if (dump_opt['d'] < 5)
1292 		return;
1293 
1294 	for (i = 0; i < bpop->bpo_num_blkptrs; i++) {
1295 		char blkbuf[BP_SPRINTF_LEN];
1296 		blkptr_t bp;
1297 
1298 		int err = dmu_read(os, object,
1299 		    i * sizeof (bp), sizeof (bp), &bp, 0);
1300 		if (err != 0) {
1301 			(void) printf("got error %u from dmu_read\n", err);
1302 			break;
1303 		}
1304 		snprintf_blkptr_compact(blkbuf, sizeof (blkbuf), &bp,
1305 		    BP_GET_FREE(&bp));
1306 		(void) printf("\t%s\n", blkbuf);
1307 	}
1308 }
1309 
1310 static void
dump_bpobj_subobjs(objset_t * os,uint64_t object,void * data,size_t size)1311 dump_bpobj_subobjs(objset_t *os, uint64_t object, void *data, size_t size)
1312 {
1313 	(void) data, (void) size;
1314 	dmu_object_info_t doi;
1315 	int64_t i;
1316 
1317 	VERIFY0(dmu_object_info(os, object, &doi));
1318 	uint64_t *subobjs = kmem_alloc(doi.doi_max_offset, KM_SLEEP);
1319 
1320 	int err = dmu_read(os, object, 0, doi.doi_max_offset, subobjs, 0);
1321 	if (err != 0) {
1322 		(void) printf("got error %u from dmu_read\n", err);
1323 		kmem_free(subobjs, doi.doi_max_offset);
1324 		return;
1325 	}
1326 
1327 	int64_t last_nonzero = -1;
1328 	for (i = 0; i < doi.doi_max_offset / 8; i++) {
1329 		if (subobjs[i] != 0)
1330 			last_nonzero = i;
1331 	}
1332 
1333 	for (i = 0; i <= last_nonzero; i++) {
1334 		(void) printf("\t%llu\n", (u_longlong_t)subobjs[i]);
1335 	}
1336 	kmem_free(subobjs, doi.doi_max_offset);
1337 }
1338 
1339 static void
dump_ddt_zap(objset_t * os,uint64_t object,void * data,size_t size)1340 dump_ddt_zap(objset_t *os, uint64_t object, void *data, size_t size)
1341 {
1342 	(void) data, (void) size;
1343 	dump_zap_stats(os, object);
1344 	/* contents are printed elsewhere, properly decoded */
1345 }
1346 
1347 static void
dump_sa_attrs(objset_t * os,uint64_t object,void * data,size_t size)1348 dump_sa_attrs(objset_t *os, uint64_t object, void *data, size_t size)
1349 {
1350 	(void) data, (void) size;
1351 	zap_cursor_t zc;
1352 	zap_attribute_t *attrp = zap_attribute_alloc();
1353 
1354 	dump_zap_stats(os, object);
1355 	(void) printf("\n");
1356 
1357 	for (zap_cursor_init(&zc, os, object);
1358 	    zap_cursor_retrieve(&zc, attrp) == 0;
1359 	    zap_cursor_advance(&zc)) {
1360 		(void) printf("\t\t%s = ", attrp->za_name);
1361 		if (attrp->za_num_integers == 0) {
1362 			(void) printf("\n");
1363 			continue;
1364 		}
1365 		(void) printf(" %llx : [%d:%d:%d]\n",
1366 		    (u_longlong_t)attrp->za_first_integer,
1367 		    (int)ATTR_LENGTH(attrp->za_first_integer),
1368 		    (int)ATTR_BSWAP(attrp->za_first_integer),
1369 		    (int)ATTR_NUM(attrp->za_first_integer));
1370 	}
1371 	zap_cursor_fini(&zc);
1372 	zap_attribute_free(attrp);
1373 }
1374 
1375 static void
dump_sa_layouts(objset_t * os,uint64_t object,void * data,size_t size)1376 dump_sa_layouts(objset_t *os, uint64_t object, void *data, size_t size)
1377 {
1378 	(void) data, (void) size;
1379 	zap_cursor_t zc;
1380 	zap_attribute_t *attrp = zap_attribute_alloc();
1381 	uint16_t *layout_attrs;
1382 	unsigned i;
1383 
1384 	dump_zap_stats(os, object);
1385 	(void) printf("\n");
1386 
1387 	for (zap_cursor_init(&zc, os, object);
1388 	    zap_cursor_retrieve(&zc, attrp) == 0;
1389 	    zap_cursor_advance(&zc)) {
1390 		(void) printf("\t\t%s = [", attrp->za_name);
1391 		if (attrp->za_num_integers == 0) {
1392 			(void) printf("\n");
1393 			continue;
1394 		}
1395 
1396 		VERIFY(attrp->za_integer_length == 2);
1397 		layout_attrs = umem_zalloc(attrp->za_num_integers *
1398 		    attrp->za_integer_length, UMEM_NOFAIL);
1399 
1400 		VERIFY(zap_lookup(os, object, attrp->za_name,
1401 		    attrp->za_integer_length,
1402 		    attrp->za_num_integers, layout_attrs) == 0);
1403 
1404 		for (i = 0; i != attrp->za_num_integers; i++)
1405 			(void) printf(" %d ", (int)layout_attrs[i]);
1406 		(void) printf("]\n");
1407 		umem_free(layout_attrs,
1408 		    attrp->za_num_integers * attrp->za_integer_length);
1409 	}
1410 	zap_cursor_fini(&zc);
1411 	zap_attribute_free(attrp);
1412 }
1413 
1414 static void
dump_zpldir(objset_t * os,uint64_t object,void * data,size_t size)1415 dump_zpldir(objset_t *os, uint64_t object, void *data, size_t size)
1416 {
1417 	(void) data, (void) size;
1418 	zap_cursor_t zc;
1419 	zap_attribute_t *attrp = zap_attribute_long_alloc();
1420 	const char *typenames[] = {
1421 		/* 0 */ "not specified",
1422 		/* 1 */ "FIFO",
1423 		/* 2 */ "Character Device",
1424 		/* 3 */ "3 (invalid)",
1425 		/* 4 */ "Directory",
1426 		/* 5 */ "5 (invalid)",
1427 		/* 6 */ "Block Device",
1428 		/* 7 */ "7 (invalid)",
1429 		/* 8 */ "Regular File",
1430 		/* 9 */ "9 (invalid)",
1431 		/* 10 */ "Symbolic Link",
1432 		/* 11 */ "11 (invalid)",
1433 		/* 12 */ "Socket",
1434 		/* 13 */ "Door",
1435 		/* 14 */ "Event Port",
1436 		/* 15 */ "15 (invalid)",
1437 	};
1438 
1439 	dump_zap_stats(os, object);
1440 	(void) printf("\n");
1441 
1442 	for (zap_cursor_init(&zc, os, object);
1443 	    zap_cursor_retrieve(&zc, attrp) == 0;
1444 	    zap_cursor_advance(&zc)) {
1445 		(void) printf("\t\t%s = %lld (type: %s)\n",
1446 		    attrp->za_name, ZFS_DIRENT_OBJ(attrp->za_first_integer),
1447 		    typenames[ZFS_DIRENT_TYPE(attrp->za_first_integer)]);
1448 	}
1449 	zap_cursor_fini(&zc);
1450 	zap_attribute_free(attrp);
1451 }
1452 
1453 static uint64_t
get_dtl_refcount(vdev_t * vd,zfs_range_tree_t * spacemap_objs)1454 get_dtl_refcount(vdev_t *vd, zfs_range_tree_t *spacemap_objs)
1455 {
1456 	uint64_t refcount = 0;
1457 
1458 	if (vd->vdev_ops->vdev_op_leaf) {
1459 		space_map_t *sm = vd->vdev_dtl_sm;
1460 
1461 		if (sm != NULL &&
1462 		    sm->sm_dbuf->db_size == sizeof (space_map_phys_t)) {
1463 			if (spacemap_objs != NULL &&
1464 			    !zfs_range_tree_contains(spacemap_objs,
1465 			    space_map_object(sm), 1)) {
1466 				zfs_range_tree_add(spacemap_objs,
1467 				    space_map_object(sm), 1);
1468 			}
1469 			return (1);
1470 		}
1471 		return (0);
1472 	}
1473 
1474 	for (unsigned c = 0; c < vd->vdev_children; c++)
1475 		refcount += get_dtl_refcount(vd->vdev_child[c], spacemap_objs);
1476 	return (refcount);
1477 }
1478 
1479 static uint64_t
get_metaslab_refcount(vdev_t * vd,zfs_range_tree_t * spacemap_objs)1480 get_metaslab_refcount(vdev_t *vd, zfs_range_tree_t *spacemap_objs)
1481 {
1482 	uint64_t refcount = 0;
1483 
1484 	if (vd->vdev_top == vd) {
1485 		for (uint64_t m = 0; m < vd->vdev_ms_count; m++) {
1486 			space_map_t *sm = vd->vdev_ms[m]->ms_sm;
1487 
1488 			if (sm != NULL &&
1489 			    sm->sm_dbuf->db_size == sizeof (space_map_phys_t)) {
1490 				if (spacemap_objs != NULL &&
1491 				    !zfs_range_tree_contains(spacemap_objs,
1492 				    space_map_object(sm), 1)) {
1493 					zfs_range_tree_add(spacemap_objs,
1494 					    space_map_object(sm), 1);
1495 				}
1496 				refcount++;
1497 			}
1498 		}
1499 	}
1500 	for (unsigned c = 0; c < vd->vdev_children; c++)
1501 		refcount += get_metaslab_refcount(vd->vdev_child[c],
1502 		    spacemap_objs);
1503 
1504 	return (refcount);
1505 }
1506 
1507 static uint64_t
get_obsolete_refcount(vdev_t * vd,zfs_range_tree_t * spacemap_objs)1508 get_obsolete_refcount(vdev_t *vd, zfs_range_tree_t *spacemap_objs)
1509 {
1510 	uint64_t obsolete_sm_object;
1511 	uint64_t refcount = 0;
1512 
1513 	VERIFY0(vdev_obsolete_sm_object(vd, &obsolete_sm_object));
1514 	if (vd->vdev_top == vd && obsolete_sm_object != 0) {
1515 		dmu_object_info_t doi;
1516 		VERIFY0(dmu_object_info(vd->vdev_spa->spa_meta_objset,
1517 		    obsolete_sm_object, &doi));
1518 		if (doi.doi_bonus_size == sizeof (space_map_phys_t)) {
1519 			if (spacemap_objs != NULL &&
1520 			    !zfs_range_tree_contains(spacemap_objs,
1521 			    obsolete_sm_object, 1)) {
1522 				zfs_range_tree_add(spacemap_objs,
1523 				    obsolete_sm_object, 1);
1524 			}
1525 			refcount++;
1526 		}
1527 	} else {
1528 		ASSERT0P(vd->vdev_obsolete_sm);
1529 		ASSERT0(obsolete_sm_object);
1530 	}
1531 	for (unsigned c = 0; c < vd->vdev_children; c++) {
1532 		refcount += get_obsolete_refcount(vd->vdev_child[c],
1533 		    spacemap_objs);
1534 	}
1535 
1536 	return (refcount);
1537 }
1538 
1539 static uint64_t
get_prev_obsolete_spacemap_refcount(spa_t * spa,zfs_range_tree_t * spacemap_objs)1540 get_prev_obsolete_spacemap_refcount(spa_t *spa,
1541     zfs_range_tree_t *spacemap_objs)
1542 {
1543 	uint64_t prev_obj =
1544 	    spa->spa_condensing_indirect_phys.scip_prev_obsolete_sm_object;
1545 	if (prev_obj != 0) {
1546 		dmu_object_info_t doi;
1547 		VERIFY0(dmu_object_info(spa->spa_meta_objset, prev_obj, &doi));
1548 		if (doi.doi_bonus_size == sizeof (space_map_phys_t)) {
1549 			if (spacemap_objs != NULL &&
1550 			    !zfs_range_tree_contains(spacemap_objs,
1551 			    prev_obj, 1)) {
1552 				zfs_range_tree_add(spacemap_objs,
1553 				    prev_obj, 1);
1554 			}
1555 			return (1);
1556 		}
1557 	}
1558 	return (0);
1559 }
1560 
1561 static uint64_t
get_checkpoint_refcount(vdev_t * vd,zfs_range_tree_t * spacemap_objs)1562 get_checkpoint_refcount(vdev_t *vd, zfs_range_tree_t *spacemap_objs)
1563 {
1564 	uint64_t refcount = 0;
1565 
1566 	if (vd->vdev_top == vd && vd->vdev_top_zap != 0) {
1567 		uint64_t checkpoint_sm_obj;
1568 		int error = zap_lookup(spa_meta_objset(vd->vdev_spa),
1569 		    vd->vdev_top_zap, VDEV_TOP_ZAP_POOL_CHECKPOINT_SM,
1570 		    sizeof (checkpoint_sm_obj), 1, &checkpoint_sm_obj);
1571 		if (error == 0) {
1572 			if (spacemap_objs != NULL &&
1573 			    !zfs_range_tree_contains(spacemap_objs,
1574 			    checkpoint_sm_obj, 1)) {
1575 				zfs_range_tree_add(spacemap_objs,
1576 				    checkpoint_sm_obj, 1);
1577 			}
1578 			refcount++;
1579 		}
1580 	}
1581 
1582 	for (uint64_t c = 0; c < vd->vdev_children; c++)
1583 		refcount += get_checkpoint_refcount(vd->vdev_child[c],
1584 		    spacemap_objs);
1585 
1586 	return (refcount);
1587 }
1588 
1589 static uint64_t
get_log_spacemap_refcount(spa_t * spa,zfs_range_tree_t * spacemap_objs)1590 get_log_spacemap_refcount(spa_t *spa, zfs_range_tree_t *spacemap_objs)
1591 {
1592 	uint64_t refcount = 0;
1593 
1594 	for (spa_log_sm_t *sls = avl_first(&spa->spa_sm_logs_by_txg);
1595 	    sls != NULL;
1596 	    sls = AVL_NEXT(&spa->spa_sm_logs_by_txg, sls)) {
1597 		if (spacemap_objs != NULL &&
1598 		    !zfs_range_tree_contains(spacemap_objs, sls->sls_sm_obj,
1599 		    1)) {
1600 			zfs_range_tree_add(spacemap_objs, sls->sls_sm_obj, 1);
1601 		}
1602 		refcount++;
1603 	}
1604 
1605 	return (refcount);
1606 }
1607 
1608 static void
dump_spacemap_refcount_mismatch_details(spa_t * spa,uint64_t expected_refcount,uint64_t actual_refcount,zfs_range_tree_t * spacemap_objs)1609 dump_spacemap_refcount_mismatch_details(spa_t *spa,
1610     uint64_t expected_refcount, uint64_t actual_refcount,
1611     zfs_range_tree_t *spacemap_objs)
1612 {
1613 	objset_t *mos = spa->spa_meta_objset;
1614 	uint64_t total_histogram_sm = 0;
1615 	uint64_t unreferenced_histogram_sm = 0;
1616 	uint64_t object = 0;
1617 	boolean_t printed_unreferenced_header = B_FALSE;
1618 
1619 	(void) printf("\tdelta(expected-actual)=%lld\n",
1620 	    (longlong_t)expected_refcount - (longlong_t)actual_refcount);
1621 
1622 	while (dmu_object_next(mos, &object, B_FALSE, 0) == 0) {
1623 		dmu_object_info_t doi;
1624 		VERIFY0(dmu_object_info(mos, object, &doi));
1625 		if (doi.doi_type != DMU_OT_SPACE_MAP ||
1626 		    doi.doi_bonus_size != sizeof (space_map_phys_t))
1627 			continue;
1628 
1629 		total_histogram_sm++;
1630 		if (zfs_range_tree_contains(spacemap_objs, object, 1))
1631 			continue;
1632 
1633 		unreferenced_histogram_sm++;
1634 		if (!printed_unreferenced_header) {
1635 			(void) printf(
1636 			    "\t  unreferenced histogram space maps:\n");
1637 			printed_unreferenced_header = B_TRUE;
1638 		}
1639 
1640 		dmu_buf_t *db = NULL;
1641 		int error = dmu_bonus_hold(mos, object, FTAG, &db);
1642 		if (error != 0) {
1643 			(void) printf("\t    object %llu "
1644 			    "(bonus hold error: %s)\n",
1645 			    (u_longlong_t)object, strerror(error));
1646 			continue;
1647 		}
1648 
1649 		space_map_phys_t *smp = db->db_data;
1650 		(void) printf("\t    object %llu smp_alloc=0x%llx "
1651 		    "smp_length=0x%llx\n", (u_longlong_t)object,
1652 		    (u_longlong_t)smp->smp_alloc,
1653 		    (u_longlong_t)smp->smp_length);
1654 		dmu_buf_rele(db, FTAG);
1655 	}
1656 
1657 	(void) printf("\t  allocated histogram space maps in MOS=%llu\n",
1658 	    (u_longlong_t)total_histogram_sm);
1659 	if (expected_refcount != total_histogram_sm) {
1660 		(void) printf("\t  WARNING: feature refcount disagrees "
1661 		    "with MOS scan by %lld\n",
1662 		    (longlong_t)expected_refcount -
1663 		    (longlong_t)total_histogram_sm);
1664 	}
1665 	if (unreferenced_histogram_sm == 0) {
1666 		(void) printf(
1667 		    "\t  no unreferenced histogram space maps found\n");
1668 	} else {
1669 		(void) printf("\t  total unreferenced histogram space "
1670 		    "maps=%llu\n",
1671 		    (u_longlong_t)unreferenced_histogram_sm);
1672 	}
1673 }
1674 
1675 static int
verify_spacemap_refcounts(spa_t * spa)1676 verify_spacemap_refcounts(spa_t *spa)
1677 {
1678 	uint64_t expected_refcount = 0;
1679 	uint64_t actual_refcount = 0;
1680 	uint64_t dtl_refcount, metaslab_refcount, obsolete_refcount,
1681 	    prev_obsolete_refcount, checkpoint_refcount, log_spacemap_refcount;
1682 	zfs_range_tree_t *spacemap_objs = zfs_range_tree_create_flags(
1683 	    NULL, ZFS_RANGE_SEG64, NULL, 0, 0, 0,
1684 	    "verify_spacemap_refcounts:spacemap_objs");
1685 
1686 	(void) feature_get_refcount(spa,
1687 	    &spa_feature_table[SPA_FEATURE_SPACEMAP_HISTOGRAM],
1688 	    &expected_refcount);
1689 	dtl_refcount = get_dtl_refcount(spa->spa_root_vdev, spacemap_objs);
1690 	metaslab_refcount = get_metaslab_refcount(spa->spa_root_vdev,
1691 	    spacemap_objs);
1692 	obsolete_refcount = get_obsolete_refcount(spa->spa_root_vdev,
1693 	    spacemap_objs);
1694 	prev_obsolete_refcount = get_prev_obsolete_spacemap_refcount(spa,
1695 	    spacemap_objs);
1696 	checkpoint_refcount = get_checkpoint_refcount(spa->spa_root_vdev,
1697 	    spacemap_objs);
1698 	log_spacemap_refcount = get_log_spacemap_refcount(spa, spacemap_objs);
1699 	actual_refcount = dtl_refcount + metaslab_refcount +
1700 	    obsolete_refcount + prev_obsolete_refcount +
1701 	    checkpoint_refcount + log_spacemap_refcount;
1702 
1703 	if (expected_refcount != actual_refcount) {
1704 		(void) printf("space map refcount mismatch: expected %llu != "
1705 		    "actual %llu\n",
1706 		    (u_longlong_t)expected_refcount,
1707 		    (u_longlong_t)actual_refcount);
1708 		(void) printf("\tDTL: %llu, metaslab: %llu, obsolete: %llu, "
1709 		    "prev obsolete: %llu, checkpoint: %llu, "
1710 		    "log spacemap: %llu\n",
1711 		    (u_longlong_t)dtl_refcount,
1712 		    (u_longlong_t)metaslab_refcount,
1713 		    (u_longlong_t)obsolete_refcount,
1714 		    (u_longlong_t)prev_obsolete_refcount,
1715 		    (u_longlong_t)checkpoint_refcount,
1716 		    (u_longlong_t)log_spacemap_refcount);
1717 		dump_spacemap_refcount_mismatch_details(spa, expected_refcount,
1718 		    actual_refcount, spacemap_objs);
1719 		zfs_range_tree_vacate(spacemap_objs, NULL, NULL);
1720 		zfs_range_tree_destroy(spacemap_objs);
1721 		return (2);
1722 	}
1723 
1724 	zfs_range_tree_vacate(spacemap_objs, NULL, NULL);
1725 	zfs_range_tree_destroy(spacemap_objs);
1726 	return (0);
1727 }
1728 
1729 static void
dump_spacemap(objset_t * os,space_map_t * sm)1730 dump_spacemap(objset_t *os, space_map_t *sm)
1731 {
1732 	const char *ddata[] = { "ALLOC", "FREE", "CONDENSE", "INVALID",
1733 	    "INVALID", "INVALID", "INVALID", "INVALID" };
1734 
1735 	if (sm == NULL)
1736 		return;
1737 
1738 	(void) printf("space map object %llu:\n",
1739 	    (longlong_t)sm->sm_object);
1740 	(void) printf("  smp_length = 0x%llx\n",
1741 	    (longlong_t)sm->sm_phys->smp_length);
1742 	(void) printf("  smp_alloc = 0x%llx\n",
1743 	    (longlong_t)sm->sm_phys->smp_alloc);
1744 
1745 	if (dump_opt['d'] < 6 && dump_opt['m'] < 4)
1746 		return;
1747 
1748 	/*
1749 	 * Print out the freelist entries in both encoded and decoded form.
1750 	 */
1751 	uint8_t mapshift = sm->sm_shift;
1752 	int64_t alloc = 0;
1753 	uint64_t word, entry_id = 0;
1754 	for (uint64_t offset = 0; offset < space_map_length(sm);
1755 	    offset += sizeof (word)) {
1756 
1757 		VERIFY0(dmu_read(os, space_map_object(sm), offset,
1758 		    sizeof (word), &word, DMU_READ_PREFETCH));
1759 
1760 		if (sm_entry_is_debug(word)) {
1761 			uint64_t de_txg = SM_DEBUG_TXG_DECODE(word);
1762 			uint64_t de_sync_pass = SM_DEBUG_SYNCPASS_DECODE(word);
1763 			if (de_txg == 0) {
1764 				(void) printf(
1765 				    "\t    [%6llu] PADDING\n",
1766 				    (u_longlong_t)entry_id);
1767 			} else {
1768 				(void) printf(
1769 				    "\t    [%6llu] %s: txg %llu pass %llu\n",
1770 				    (u_longlong_t)entry_id,
1771 				    ddata[SM_DEBUG_ACTION_DECODE(word)],
1772 				    (u_longlong_t)de_txg,
1773 				    (u_longlong_t)de_sync_pass);
1774 			}
1775 			entry_id++;
1776 			continue;
1777 		}
1778 
1779 		char entry_type;
1780 		uint64_t entry_off, entry_run, entry_vdev;
1781 
1782 		if (sm_entry_is_single_word(word)) {
1783 			entry_type = (SM_TYPE_DECODE(word) == SM_ALLOC) ?
1784 			    'A' : 'F';
1785 			entry_off = (SM_OFFSET_DECODE(word) << mapshift) +
1786 			    sm->sm_start;
1787 			entry_run = SM_RUN_DECODE(word) << mapshift;
1788 
1789 			(void) printf("\t    [%6llu] %c "
1790 			    "range: %012llx-%012llx size: %08llx\n",
1791 			    (u_longlong_t)entry_id, entry_type,
1792 			    (u_longlong_t)entry_off,
1793 			    (u_longlong_t)(entry_off + entry_run - 1),
1794 			    (u_longlong_t)entry_run);
1795 		} else {
1796 			/* it is a two-word entry so we read another word */
1797 			ASSERT(sm_entry_is_double_word(word));
1798 
1799 			uint64_t extra_word;
1800 			offset += sizeof (extra_word);
1801 			ASSERT3U(offset, <, space_map_length(sm));
1802 			VERIFY0(dmu_read(os, space_map_object(sm), offset,
1803 			    sizeof (extra_word), &extra_word,
1804 			    DMU_READ_PREFETCH));
1805 
1806 			entry_run = SM2_RUN_DECODE(word) << mapshift;
1807 			entry_vdev = SM2_VDEV_DECODE(word);
1808 			entry_type = (SM2_TYPE_DECODE(extra_word) == SM_ALLOC) ?
1809 			    'A' : 'F';
1810 			entry_off = (SM2_OFFSET_DECODE(extra_word) <<
1811 			    mapshift) + sm->sm_start;
1812 
1813 			if (zopt_metaslab_args == 0 ||
1814 			    zopt_metaslab[0] == entry_vdev) {
1815 				(void) printf("\t    [%6llu] %c "
1816 				    "range: %012llx-%012llx size: %08llx "
1817 				    "vdev: %llu\n",
1818 				    (u_longlong_t)entry_id, entry_type,
1819 				    (u_longlong_t)entry_off,
1820 				    (u_longlong_t)(entry_off + entry_run - 1),
1821 				    (u_longlong_t)entry_run,
1822 				    (u_longlong_t)entry_vdev);
1823 			}
1824 		}
1825 
1826 		if (entry_type == 'A')
1827 			alloc += entry_run;
1828 		else
1829 			alloc -= entry_run;
1830 		entry_id++;
1831 	}
1832 	if (alloc != space_map_allocated(sm)) {
1833 		(void) printf("space_map_object alloc (%lld) INCONSISTENT "
1834 		    "with space map summary (%lld)\n",
1835 		    (longlong_t)space_map_allocated(sm), (longlong_t)alloc);
1836 	}
1837 }
1838 
1839 static void
dump_metaslab_stats(metaslab_t * msp)1840 dump_metaslab_stats(metaslab_t *msp)
1841 {
1842 	char maxbuf[32];
1843 	zfs_range_tree_t *rt = msp->ms_allocatable;
1844 	zfs_btree_t *t = &msp->ms_allocatable_by_size;
1845 	int free_pct = zfs_range_tree_space(rt) * 100 / msp->ms_size;
1846 
1847 	/* max sure nicenum has enough space */
1848 	_Static_assert(sizeof (maxbuf) >= NN_NUMBUF_SZ, "maxbuf truncated");
1849 
1850 	zdb_nicenum(metaslab_largest_allocatable(msp), maxbuf, sizeof (maxbuf));
1851 
1852 	(void) printf("\t %25s %10lu   %7s  %6s   %4s %4d%%\n",
1853 	    "segments", zfs_btree_numnodes(t), "maxsize", maxbuf,
1854 	    "freepct", free_pct);
1855 	(void) printf("\tIn-memory histogram:\n");
1856 	dump_histogram(rt->rt_histogram, ZFS_RANGE_TREE_HISTOGRAM_SIZE, 0);
1857 }
1858 
1859 static void
dump_allocated(void * arg,uint64_t start,uint64_t size)1860 dump_allocated(void *arg, uint64_t start, uint64_t size)
1861 {
1862 	uint64_t *off = arg;
1863 	if (*off != start)
1864 		(void) printf("ALLOC: %"PRIu64" %"PRIu64"\n", *off,
1865 		    start - *off);
1866 	*off = start + size;
1867 }
1868 
1869 static void
dump_metaslab(metaslab_t * msp)1870 dump_metaslab(metaslab_t *msp)
1871 {
1872 	vdev_t *vd = msp->ms_group->mg_vd;
1873 	spa_t *spa = vd->vdev_spa;
1874 	space_map_t *sm = msp->ms_sm;
1875 	char freebuf[32];
1876 
1877 	zdb_nicenum(msp->ms_size - space_map_allocated(sm), freebuf,
1878 	    sizeof (freebuf));
1879 
1880 	(void) printf(
1881 	    "\tmetaslab %6llu   offset %12llx   spacemap %6llu   free    %5s\n",
1882 	    (u_longlong_t)msp->ms_id, (u_longlong_t)msp->ms_start,
1883 	    (u_longlong_t)space_map_object(sm), freebuf);
1884 
1885 	if (dump_opt[ARG_ALLOCATED] ||
1886 	    (dump_opt['m'] > 2 && !dump_opt['L'])) {
1887 		mutex_enter(&msp->ms_lock);
1888 		VERIFY0(metaslab_load(msp));
1889 	}
1890 
1891 	if (dump_opt['m'] > 2 && !dump_opt['L']) {
1892 		zfs_range_tree_stat_verify(msp->ms_allocatable);
1893 		dump_metaslab_stats(msp);
1894 	}
1895 
1896 	if (dump_opt[ARG_ALLOCATED]) {
1897 		uint64_t off = msp->ms_start;
1898 		zfs_range_tree_walk(msp->ms_allocatable, dump_allocated,
1899 		    &off);
1900 		if (off != msp->ms_start + msp->ms_size)
1901 			(void) printf("ALLOC: %"PRIu64" %"PRIu64"\n", off,
1902 			    msp->ms_size - off);
1903 	}
1904 
1905 	if (dump_opt['m'] > 1 && sm != NULL &&
1906 	    spa_feature_is_active(spa, SPA_FEATURE_SPACEMAP_HISTOGRAM)) {
1907 		/*
1908 		 * The space map histogram represents free space in chunks
1909 		 * of sm_shift (i.e. bucket 0 refers to 2^sm_shift).
1910 		 */
1911 		(void) printf("\tOn-disk histogram:\t\tfragmentation %llu\n",
1912 		    (u_longlong_t)msp->ms_fragmentation);
1913 		dump_histogram(sm->sm_phys->smp_histogram,
1914 		    SPACE_MAP_HISTOGRAM_SIZE, sm->sm_shift);
1915 	}
1916 
1917 	if (dump_opt[ARG_ALLOCATED] ||
1918 	    (dump_opt['m'] > 2 && !dump_opt['L'])) {
1919 		metaslab_unload(msp);
1920 		mutex_exit(&msp->ms_lock);
1921 	}
1922 
1923 	if (vd->vdev_ops == &vdev_draid_ops)
1924 		ASSERT3U(msp->ms_size, <=, 1ULL << vd->vdev_ms_shift);
1925 	else
1926 		ASSERT3U(msp->ms_size, ==, 1ULL << vd->vdev_ms_shift);
1927 
1928 	dump_spacemap(spa->spa_meta_objset, msp->ms_sm);
1929 
1930 	if (spa_feature_is_active(spa, SPA_FEATURE_LOG_SPACEMAP)) {
1931 		(void) printf("\tFlush data:\n\tunflushed txg=%llu\n\n",
1932 		    (u_longlong_t)metaslab_unflushed_txg(msp));
1933 	}
1934 }
1935 
1936 static void
print_vdev_metaslab_header(vdev_t * vd)1937 print_vdev_metaslab_header(vdev_t *vd)
1938 {
1939 	vdev_alloc_bias_t alloc_bias = vd->vdev_alloc_bias;
1940 	const char *bias_str = "";
1941 	if (alloc_bias == VDEV_BIAS_LOG || vd->vdev_islog) {
1942 		bias_str = VDEV_ALLOC_BIAS_LOG;
1943 	} else if (alloc_bias == VDEV_BIAS_SPECIAL) {
1944 		bias_str = VDEV_ALLOC_BIAS_SPECIAL;
1945 	} else if (alloc_bias == VDEV_BIAS_DEDUP) {
1946 		bias_str = VDEV_ALLOC_BIAS_DEDUP;
1947 	}
1948 
1949 	uint64_t ms_flush_data_obj = 0;
1950 	if (vd->vdev_top_zap != 0) {
1951 		int error = zap_lookup(spa_meta_objset(vd->vdev_spa),
1952 		    vd->vdev_top_zap, VDEV_TOP_ZAP_MS_UNFLUSHED_PHYS_TXGS,
1953 		    sizeof (uint64_t), 1, &ms_flush_data_obj);
1954 		if (error != ENOENT) {
1955 			ASSERT0(error);
1956 		}
1957 	}
1958 
1959 	(void) printf("\tvdev %10llu\t%s  metaslab shift %4llu",
1960 	    (u_longlong_t)vd->vdev_id, bias_str,
1961 	    (u_longlong_t)vd->vdev_ms_shift);
1962 
1963 	if (ms_flush_data_obj != 0) {
1964 		(void) printf("   ms_unflushed_phys object %llu",
1965 		    (u_longlong_t)ms_flush_data_obj);
1966 	}
1967 
1968 	(void) printf("\n\t%-10s%5llu   %-19s   %-15s   %-12s\n",
1969 	    "metaslabs", (u_longlong_t)vd->vdev_ms_count,
1970 	    "offset", "spacemap", "free");
1971 	(void) printf("\t%15s   %19s   %15s   %12s\n",
1972 	    "---------------", "-------------------",
1973 	    "---------------", "------------");
1974 }
1975 
1976 static void
dump_metaslab_groups(spa_t * spa,boolean_t show_special)1977 dump_metaslab_groups(spa_t *spa, boolean_t show_special)
1978 {
1979 	vdev_t *rvd = spa->spa_root_vdev;
1980 	metaslab_class_t *mc = spa_normal_class(spa);
1981 	metaslab_class_t *smc = spa_special_class(spa);
1982 	uint64_t fragmentation;
1983 
1984 	metaslab_class_histogram_verify(mc);
1985 
1986 	for (unsigned c = 0; c < rvd->vdev_children; c++) {
1987 		vdev_t *tvd = rvd->vdev_child[c];
1988 		metaslab_group_t *mg = tvd->vdev_mg;
1989 
1990 		if (mg == NULL || (mg->mg_class != mc &&
1991 		    (!show_special || mg->mg_class != smc)))
1992 			continue;
1993 
1994 		metaslab_group_histogram_verify(mg);
1995 		mg->mg_fragmentation = metaslab_group_fragmentation(mg);
1996 
1997 		(void) printf("\tvdev %10llu\t\tmetaslabs%5llu\t\t"
1998 		    "fragmentation",
1999 		    (u_longlong_t)tvd->vdev_id,
2000 		    (u_longlong_t)tvd->vdev_ms_count);
2001 		if (mg->mg_fragmentation == ZFS_FRAG_INVALID) {
2002 			(void) printf("%3s\n", "-");
2003 		} else {
2004 			(void) printf("%3llu%%\n",
2005 			    (u_longlong_t)mg->mg_fragmentation);
2006 		}
2007 		dump_histogram(mg->mg_histogram,
2008 		    ZFS_RANGE_TREE_HISTOGRAM_SIZE, 0);
2009 	}
2010 
2011 	(void) printf("\tpool %s\tfragmentation", spa_name(spa));
2012 	fragmentation = metaslab_class_fragmentation(mc);
2013 	if (fragmentation == ZFS_FRAG_INVALID)
2014 		(void) printf("\t%3s\n", "-");
2015 	else
2016 		(void) printf("\t%3llu%%\n", (u_longlong_t)fragmentation);
2017 	dump_histogram(mc->mc_histogram, ZFS_RANGE_TREE_HISTOGRAM_SIZE, 0);
2018 }
2019 
2020 static void
print_vdev_indirect(vdev_t * vd)2021 print_vdev_indirect(vdev_t *vd)
2022 {
2023 	vdev_indirect_config_t *vic = &vd->vdev_indirect_config;
2024 	vdev_indirect_mapping_t *vim = vd->vdev_indirect_mapping;
2025 	vdev_indirect_births_t *vib = vd->vdev_indirect_births;
2026 
2027 	if (vim == NULL) {
2028 		ASSERT0P(vib);
2029 		return;
2030 	}
2031 
2032 	ASSERT3U(vdev_indirect_mapping_object(vim), ==,
2033 	    vic->vic_mapping_object);
2034 	ASSERT3U(vdev_indirect_births_object(vib), ==,
2035 	    vic->vic_births_object);
2036 
2037 	(void) printf("indirect births obj %llu:\n",
2038 	    (longlong_t)vic->vic_births_object);
2039 	(void) printf("    vib_count = %llu\n",
2040 	    (longlong_t)vdev_indirect_births_count(vib));
2041 	for (uint64_t i = 0; i < vdev_indirect_births_count(vib); i++) {
2042 		vdev_indirect_birth_entry_phys_t *cur_vibe =
2043 		    &vib->vib_entries[i];
2044 		(void) printf("\toffset %llx -> txg %llu\n",
2045 		    (longlong_t)cur_vibe->vibe_offset,
2046 		    (longlong_t)cur_vibe->vibe_phys_birth_txg);
2047 	}
2048 	(void) printf("\n");
2049 
2050 	(void) printf("indirect mapping obj %llu:\n",
2051 	    (longlong_t)vic->vic_mapping_object);
2052 	(void) printf("    vim_max_offset = 0x%llx\n",
2053 	    (longlong_t)vdev_indirect_mapping_max_offset(vim));
2054 	(void) printf("    vim_bytes_mapped = 0x%llx\n",
2055 	    (longlong_t)vdev_indirect_mapping_bytes_mapped(vim));
2056 	(void) printf("    vim_count = %llu\n",
2057 	    (longlong_t)vdev_indirect_mapping_num_entries(vim));
2058 
2059 	if (dump_opt['d'] <= 5 && dump_opt['m'] <= 3)
2060 		return;
2061 
2062 	uint32_t *counts = vdev_indirect_mapping_load_obsolete_counts(vim);
2063 
2064 	for (uint64_t i = 0; i < vdev_indirect_mapping_num_entries(vim); i++) {
2065 		vdev_indirect_mapping_entry_phys_t *vimep =
2066 		    &vim->vim_entries[i];
2067 		(void) printf("\t<%llx:%llx:%llx> -> "
2068 		    "<%llx:%llx:%llx> (%x obsolete)\n",
2069 		    (longlong_t)vd->vdev_id,
2070 		    (longlong_t)DVA_MAPPING_GET_SRC_OFFSET(vimep),
2071 		    (longlong_t)DVA_GET_ASIZE(&vimep->vimep_dst),
2072 		    (longlong_t)DVA_GET_VDEV(&vimep->vimep_dst),
2073 		    (longlong_t)DVA_GET_OFFSET(&vimep->vimep_dst),
2074 		    (longlong_t)DVA_GET_ASIZE(&vimep->vimep_dst),
2075 		    counts[i]);
2076 	}
2077 	(void) printf("\n");
2078 
2079 	uint64_t obsolete_sm_object;
2080 	VERIFY0(vdev_obsolete_sm_object(vd, &obsolete_sm_object));
2081 	if (obsolete_sm_object != 0) {
2082 		objset_t *mos = vd->vdev_spa->spa_meta_objset;
2083 		(void) printf("obsolete space map object %llu:\n",
2084 		    (u_longlong_t)obsolete_sm_object);
2085 		ASSERT(vd->vdev_obsolete_sm != NULL);
2086 		ASSERT3U(space_map_object(vd->vdev_obsolete_sm), ==,
2087 		    obsolete_sm_object);
2088 		dump_spacemap(mos, vd->vdev_obsolete_sm);
2089 		(void) printf("\n");
2090 	}
2091 }
2092 
2093 static void
dump_metaslabs(spa_t * spa)2094 dump_metaslabs(spa_t *spa)
2095 {
2096 	vdev_t *vd, *rvd = spa->spa_root_vdev;
2097 	uint64_t m, c = 0, children = rvd->vdev_children;
2098 
2099 	(void) printf("\nMetaslabs:\n");
2100 
2101 	if (zopt_metaslab_args > 0) {
2102 		c = zopt_metaslab[0];
2103 
2104 		if (c >= children)
2105 			(void) fatal("bad vdev id: %llu", (u_longlong_t)c);
2106 
2107 		if (zopt_metaslab_args > 1) {
2108 			vd = rvd->vdev_child[c];
2109 			print_vdev_metaslab_header(vd);
2110 
2111 			for (m = 1; m < zopt_metaslab_args; m++) {
2112 				if (zopt_metaslab[m] < vd->vdev_ms_count)
2113 					dump_metaslab(
2114 					    vd->vdev_ms[zopt_metaslab[m]]);
2115 				else
2116 					(void) fprintf(stderr, "bad metaslab "
2117 					    "number %llu\n",
2118 					    (u_longlong_t)zopt_metaslab[m]);
2119 			}
2120 			(void) printf("\n");
2121 			return;
2122 		}
2123 		children = c + 1;
2124 	}
2125 	for (; c < children; c++) {
2126 		vd = rvd->vdev_child[c];
2127 		print_vdev_metaslab_header(vd);
2128 
2129 		print_vdev_indirect(vd);
2130 
2131 		for (m = 0; m < vd->vdev_ms_count; m++)
2132 			dump_metaslab(vd->vdev_ms[m]);
2133 		(void) printf("\n");
2134 	}
2135 }
2136 
2137 static void
dump_log_spacemaps(spa_t * spa)2138 dump_log_spacemaps(spa_t *spa)
2139 {
2140 	if (!spa_feature_is_active(spa, SPA_FEATURE_LOG_SPACEMAP))
2141 		return;
2142 
2143 	(void) printf("\nLog Space Maps in Pool:\n");
2144 	for (spa_log_sm_t *sls = avl_first(&spa->spa_sm_logs_by_txg);
2145 	    sls; sls = AVL_NEXT(&spa->spa_sm_logs_by_txg, sls)) {
2146 		space_map_t *sm = NULL;
2147 		VERIFY0(space_map_open(&sm, spa_meta_objset(spa),
2148 		    sls->sls_sm_obj, 0, UINT64_MAX, SPA_MINBLOCKSHIFT));
2149 
2150 		(void) printf("Log Spacemap object %llu txg %llu\n",
2151 		    (u_longlong_t)sls->sls_sm_obj, (u_longlong_t)sls->sls_txg);
2152 		dump_spacemap(spa->spa_meta_objset, sm);
2153 		space_map_close(sm);
2154 	}
2155 	(void) printf("\n");
2156 }
2157 
2158 static void
dump_ddt_entry(const ddt_t * ddt,const ddt_lightweight_entry_t * ddlwe,uint64_t index)2159 dump_ddt_entry(const ddt_t *ddt, const ddt_lightweight_entry_t *ddlwe,
2160     uint64_t index)
2161 {
2162 	const ddt_key_t *ddk = &ddlwe->ddlwe_key;
2163 	char blkbuf[BP_SPRINTF_LEN];
2164 	blkptr_t blk;
2165 	int p;
2166 
2167 	for (p = 0; p < DDT_NPHYS(ddt); p++) {
2168 		const ddt_univ_phys_t *ddp = &ddlwe->ddlwe_phys;
2169 		ddt_phys_variant_t v = DDT_PHYS_VARIANT(ddt, p);
2170 
2171 		if (ddt_phys_birth(ddp, v) == 0)
2172 			continue;
2173 		ddt_bp_create(ddt->ddt_checksum, ddk, ddp, v, &blk);
2174 		snprintf_blkptr(blkbuf, sizeof (blkbuf), &blk);
2175 		(void) printf("index %llx refcnt %llu phys %d %s\n",
2176 		    (u_longlong_t)index, (u_longlong_t)ddt_phys_refcnt(ddp, v),
2177 		    p, blkbuf);
2178 	}
2179 }
2180 
2181 static void
dump_dedup_ratio(const ddt_stat_t * dds)2182 dump_dedup_ratio(const ddt_stat_t *dds)
2183 {
2184 	double rL, rP, rD, D, dedup, compress, copies;
2185 
2186 	if (dds->dds_blocks == 0)
2187 		return;
2188 
2189 	rL = (double)dds->dds_ref_lsize;
2190 	rP = (double)dds->dds_ref_psize;
2191 	rD = (double)dds->dds_ref_dsize;
2192 	D = (double)dds->dds_dsize;
2193 
2194 	dedup = rD / D;
2195 	compress = rL / rP;
2196 	copies = rD / rP;
2197 
2198 	(void) printf("dedup = %.2f, compress = %.2f, copies = %.2f, "
2199 	    "dedup * compress / copies = %.2f\n\n",
2200 	    dedup, compress, copies, dedup * compress / copies);
2201 }
2202 
2203 static void
dump_ddt_log(ddt_t * ddt)2204 dump_ddt_log(ddt_t *ddt)
2205 {
2206 	if (ddt->ddt_version != DDT_VERSION_FDT ||
2207 	    !(ddt->ddt_flags & DDT_FLAG_LOG))
2208 		return;
2209 
2210 	for (int n = 0; n < 2; n++) {
2211 		ddt_log_t *ddl = &ddt->ddt_log[n];
2212 
2213 		char flagstr[64] = {0};
2214 		if (ddl->ddl_flags > 0) {
2215 			flagstr[0] = ' ';
2216 			int c = 1;
2217 			if (ddl->ddl_flags & DDL_FLAG_FLUSHING)
2218 				c += strlcpy(&flagstr[c], " FLUSHING",
2219 				    sizeof (flagstr) - c);
2220 			if (ddl->ddl_flags & DDL_FLAG_CHECKPOINT)
2221 				c += strlcpy(&flagstr[c], " CHECKPOINT",
2222 				    sizeof (flagstr) - c);
2223 			if (ddl->ddl_flags &
2224 			    ~(DDL_FLAG_FLUSHING|DDL_FLAG_CHECKPOINT))
2225 				c += strlcpy(&flagstr[c], " UNKNOWN",
2226 				    sizeof (flagstr) - c);
2227 			flagstr[1] = '[';
2228 			flagstr[c] = ']';
2229 		}
2230 
2231 		uint64_t count = avl_numnodes(&ddl->ddl_tree);
2232 
2233 		printf(DMU_POOL_DDT_LOG ": flags=0x%02x%s; obj=%llu; "
2234 		    "len=%llu; txg=%llu; entries=%llu\n",
2235 		    zio_checksum_table[ddt->ddt_checksum].ci_name, n,
2236 		    ddl->ddl_flags, flagstr,
2237 		    (u_longlong_t)ddl->ddl_object,
2238 		    (u_longlong_t)ddl->ddl_length,
2239 		    (u_longlong_t)ddl->ddl_first_txg, (u_longlong_t)count);
2240 
2241 		if (ddl->ddl_flags & DDL_FLAG_CHECKPOINT) {
2242 			const ddt_key_t *ddk = &ddl->ddl_checkpoint;
2243 			printf("    checkpoint: "
2244 			    "%016llx:%016llx:%016llx:%016llx:%016llx\n",
2245 			    (u_longlong_t)ddk->ddk_cksum.zc_word[0],
2246 			    (u_longlong_t)ddk->ddk_cksum.zc_word[1],
2247 			    (u_longlong_t)ddk->ddk_cksum.zc_word[2],
2248 			    (u_longlong_t)ddk->ddk_cksum.zc_word[3],
2249 			    (u_longlong_t)ddk->ddk_prop);
2250 		}
2251 
2252 		if (count == 0 || dump_opt['D'] < 4)
2253 			continue;
2254 
2255 		ddt_lightweight_entry_t ddlwe;
2256 		uint64_t index = 0;
2257 		for (ddt_log_entry_t *ddle = avl_first(&ddl->ddl_tree);
2258 		    ddle; ddle = AVL_NEXT(&ddl->ddl_tree, ddle)) {
2259 			DDT_LOG_ENTRY_TO_LIGHTWEIGHT(ddt, ddle, &ddlwe);
2260 			dump_ddt_entry(ddt, &ddlwe, index++);
2261 		}
2262 	}
2263 }
2264 
2265 static void
dump_ddt_object(ddt_t * ddt,ddt_type_t type,ddt_class_t class)2266 dump_ddt_object(ddt_t *ddt, ddt_type_t type, ddt_class_t class)
2267 {
2268 	char name[DDT_NAMELEN];
2269 	ddt_lightweight_entry_t ddlwe;
2270 	uint64_t walk = 0;
2271 	dmu_object_info_t doi;
2272 	uint64_t count, dspace, mspace;
2273 	int error;
2274 
2275 	error = ddt_object_info(ddt, type, class, &doi);
2276 
2277 	if (error == ENOENT)
2278 		return;
2279 	ASSERT0(error);
2280 
2281 	error = ddt_object_count(ddt, type, class, &count);
2282 	ASSERT0(error);
2283 	if (count == 0)
2284 		return;
2285 
2286 	dspace = doi.doi_physical_blocks_512 << 9;
2287 	mspace = doi.doi_fill_count * doi.doi_data_block_size;
2288 
2289 	ddt_object_name(ddt, type, class, name);
2290 
2291 	(void) printf("%s: dspace=%llu; mspace=%llu; entries=%llu\n", name,
2292 	    (u_longlong_t)dspace, (u_longlong_t)mspace, (u_longlong_t)count);
2293 
2294 	if (dump_opt['D'] < 3)
2295 		return;
2296 
2297 	(void) printf("%s: object=%llu\n", name,
2298 	    (u_longlong_t)ddt->ddt_object[type][class]);
2299 	zpool_dump_ddt(NULL, &ddt->ddt_histogram[type][class],
2300 	    dump_opt['P'] > 0);
2301 
2302 	if (dump_opt['D'] < 4)
2303 		return;
2304 
2305 	if (dump_opt['D'] < 5 && class == DDT_CLASS_UNIQUE)
2306 		return;
2307 
2308 	(void) printf("%s contents:\n\n", name);
2309 
2310 	while ((error = ddt_object_walk(ddt, type, class, &walk, &ddlwe)) == 0)
2311 		dump_ddt_entry(ddt, &ddlwe, walk);
2312 
2313 	ASSERT3U(error, ==, ENOENT);
2314 
2315 	(void) printf("\n");
2316 }
2317 
2318 static void
dump_ddt(ddt_t * ddt)2319 dump_ddt(ddt_t *ddt)
2320 {
2321 	if (!ddt || ddt->ddt_version == DDT_VERSION_UNCONFIGURED)
2322 		return;
2323 
2324 	char flagstr[64] = {0};
2325 	if (ddt->ddt_flags > 0) {
2326 		flagstr[0] = ' ';
2327 		int c = 1;
2328 		if (ddt->ddt_flags & DDT_FLAG_FLAT)
2329 			c += strlcpy(&flagstr[c], " FLAT",
2330 			    sizeof (flagstr) - c);
2331 		if (ddt->ddt_flags & DDT_FLAG_LOG)
2332 			c += strlcpy(&flagstr[c], " LOG",
2333 			    sizeof (flagstr) - c);
2334 		if (ddt->ddt_flags & ~DDT_FLAG_MASK)
2335 			c += strlcpy(&flagstr[c], " UNKNOWN",
2336 			    sizeof (flagstr) - c);
2337 		flagstr[1] = '[';
2338 		flagstr[c] = ']';
2339 	}
2340 
2341 	printf("DDT-%s: version=%llu [%s]; flags=0x%02llx%s; rootobj=%llu\n",
2342 	    zio_checksum_table[ddt->ddt_checksum].ci_name,
2343 	    (u_longlong_t)ddt->ddt_version,
2344 	    (ddt->ddt_version == 0) ? "LEGACY" :
2345 	    (ddt->ddt_version == 1) ? "FDT" : "UNKNOWN",
2346 	    (u_longlong_t)ddt->ddt_flags, flagstr,
2347 	    (u_longlong_t)ddt->ddt_dir_object);
2348 
2349 	for (ddt_type_t type = 0; type < DDT_TYPES; type++)
2350 		for (ddt_class_t class = 0; class < DDT_CLASSES; class++)
2351 			dump_ddt_object(ddt, type, class);
2352 
2353 	dump_ddt_log(ddt);
2354 }
2355 
2356 static void
dump_all_ddts(spa_t * spa)2357 dump_all_ddts(spa_t *spa)
2358 {
2359 	ddt_histogram_t ddh_total = {{{0}}};
2360 	ddt_stat_t dds_total = {0};
2361 
2362 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++)
2363 		dump_ddt(spa->spa_ddt[c]);
2364 
2365 	ddt_get_dedup_stats(spa, &dds_total);
2366 
2367 	if (dds_total.dds_blocks == 0) {
2368 		(void) printf("All DDTs are empty\n");
2369 		return;
2370 	}
2371 
2372 	(void) printf("\n");
2373 
2374 	if (dump_opt['D'] > 1) {
2375 		(void) printf("DDT histogram (aggregated over all DDTs):\n");
2376 		ddt_get_dedup_histogram(spa, &ddh_total);
2377 		zpool_dump_ddt(&dds_total, &ddh_total, dump_opt['P'] > 0);
2378 	}
2379 
2380 	dump_dedup_ratio(&dds_total);
2381 
2382 	/*
2383 	 * Dump a histogram of unique class entry age
2384 	 */
2385 	if (dump_opt['D'] == 3 && getenv("ZDB_DDT_UNIQUE_AGE_HIST") != NULL) {
2386 		ddt_age_histo_t histogram;
2387 
2388 		(void) printf("DDT walk unique, building age histogram...\n");
2389 		ddt_prune_walk(spa, 0, &histogram);
2390 
2391 		/*
2392 		 * print out histogram for unique entry class birth
2393 		 */
2394 		if (histogram.dah_entries > 0) {
2395 			(void) printf("%5s  %9s  %4s\n",
2396 			    "age", "blocks", "amnt");
2397 			(void) printf("%5s  %9s  %4s\n",
2398 			    "-----", "---------", "----");
2399 			for (int i = 0; i < HIST_BINS; i++) {
2400 				(void) printf("%5d  %9d %4d%%\n", 1 << i,
2401 				    (int)histogram.dah_age_histo[i],
2402 				    (int)((histogram.dah_age_histo[i] * 100) /
2403 				    histogram.dah_entries));
2404 			}
2405 		}
2406 	}
2407 }
2408 
2409 static void
dump_brt(spa_t * spa)2410 dump_brt(spa_t *spa)
2411 {
2412 	if (!spa_feature_is_enabled(spa, SPA_FEATURE_BLOCK_CLONING)) {
2413 		printf("BRT: unsupported on this pool\n");
2414 		return;
2415 	}
2416 
2417 	if (!spa_feature_is_active(spa, SPA_FEATURE_BLOCK_CLONING)) {
2418 		printf("BRT: empty\n");
2419 		return;
2420 	}
2421 
2422 	char count[32], used[32], saved[32];
2423 	zdb_nicebytes(brt_get_used(spa), used, sizeof (used));
2424 	zdb_nicebytes(brt_get_saved(spa), saved, sizeof (saved));
2425 	uint64_t ratio = brt_get_ratio(spa);
2426 	printf("BRT: used %s; saved %s; ratio %llu.%02llux\n", used, saved,
2427 	    (u_longlong_t)(ratio / 100), (u_longlong_t)(ratio % 100));
2428 
2429 	if (dump_opt['T'] < 2)
2430 		return;
2431 
2432 	for (uint64_t vdevid = 0; vdevid < spa->spa_brt_nvdevs; vdevid++) {
2433 		brt_vdev_t *brtvd = spa->spa_brt_vdevs[vdevid];
2434 		if (!brtvd->bv_initiated) {
2435 			printf("BRT: vdev %" PRIu64 ": empty\n", vdevid);
2436 			continue;
2437 		}
2438 
2439 		zdb_nicenum(brtvd->bv_totalcount, count, sizeof (count));
2440 		zdb_nicebytes(brtvd->bv_usedspace, used, sizeof (used));
2441 		zdb_nicebytes(brtvd->bv_savedspace, saved, sizeof (saved));
2442 		printf("BRT: vdev %" PRIu64 ": refcnt %s; used %s; saved %s\n",
2443 		    vdevid, count, used, saved);
2444 	}
2445 
2446 	if (dump_opt['T'] < 3)
2447 		return;
2448 
2449 	/* -TTT shows a per-vdev histograms; -TTTT shows all entries */
2450 	boolean_t do_histo = dump_opt['T'] == 3;
2451 
2452 	char dva[64];
2453 
2454 	if (!do_histo)
2455 		printf("\n%-16s %-10s\n", "DVA", "REFCNT");
2456 
2457 	for (uint64_t vdevid = 0; vdevid < spa->spa_brt_nvdevs; vdevid++) {
2458 		brt_vdev_t *brtvd = spa->spa_brt_vdevs[vdevid];
2459 		if (!brtvd->bv_initiated)
2460 			continue;
2461 
2462 		uint64_t counts[64] = {};
2463 
2464 		zap_cursor_t zc;
2465 		zap_attribute_t *za = zap_attribute_alloc();
2466 		for (zap_cursor_init(&zc, spa->spa_meta_objset,
2467 		    brtvd->bv_mos_entries);
2468 		    zap_cursor_retrieve(&zc, za) == 0;
2469 		    zap_cursor_advance(&zc)) {
2470 			uint64_t refcnt;
2471 			VERIFY0(zap_lookup_uint64(spa->spa_meta_objset,
2472 			    brtvd->bv_mos_entries,
2473 			    (const uint64_t *)za->za_name, 1,
2474 			    za->za_integer_length, za->za_num_integers,
2475 			    &refcnt));
2476 
2477 			if (do_histo)
2478 				counts[highbit64(refcnt)]++;
2479 			else {
2480 				uint64_t offset =
2481 				    *(const uint64_t *)za->za_name;
2482 
2483 				snprintf(dva, sizeof (dva), "%" PRIu64 ":%llx",
2484 				    vdevid, (u_longlong_t)offset);
2485 				printf("%-16s %-10llu\n", dva,
2486 				    (u_longlong_t)refcnt);
2487 			}
2488 		}
2489 		zap_cursor_fini(&zc);
2490 		zap_attribute_free(za);
2491 
2492 		if (do_histo) {
2493 			printf("\nBRT: vdev %" PRIu64
2494 			    ": DVAs with 2^n refcnts:\n", vdevid);
2495 			dump_histogram(counts, 64, 0);
2496 		}
2497 	}
2498 }
2499 
2500 static void
dump_dtl_seg(void * arg,uint64_t start,uint64_t size)2501 dump_dtl_seg(void *arg, uint64_t start, uint64_t size)
2502 {
2503 	char *prefix = arg;
2504 
2505 	(void) printf("%s [%llu,%llu) length %llu\n",
2506 	    prefix,
2507 	    (u_longlong_t)start,
2508 	    (u_longlong_t)(start + size),
2509 	    (u_longlong_t)(size));
2510 }
2511 
2512 static void
dump_dtl(vdev_t * vd,int indent)2513 dump_dtl(vdev_t *vd, int indent)
2514 {
2515 	spa_t *spa = vd->vdev_spa;
2516 	boolean_t required;
2517 	const char *name[DTL_TYPES] = { "missing", "partial", "scrub",
2518 		"outage" };
2519 	char prefix[256];
2520 
2521 	spa_vdev_state_enter(spa, SCL_NONE);
2522 	required = vdev_dtl_required(vd);
2523 	(void) spa_vdev_state_exit(spa, NULL, 0);
2524 
2525 	if (indent == 0)
2526 		(void) printf("\nDirty time logs:\n\n");
2527 
2528 	(void) printf("\t%*s%s [%s]\n", indent, "",
2529 	    vd->vdev_path ? vd->vdev_path :
2530 	    vd->vdev_parent ? vd->vdev_ops->vdev_op_type : spa_name(spa),
2531 	    required ? "DTL-required" : "DTL-expendable");
2532 
2533 	for (int t = 0; t < DTL_TYPES; t++) {
2534 		zfs_range_tree_t *rt = vd->vdev_dtl[t];
2535 		if (zfs_range_tree_space(rt) == 0)
2536 			continue;
2537 		(void) snprintf(prefix, sizeof (prefix), "\t%*s%s",
2538 		    indent + 2, "", name[t]);
2539 		zfs_range_tree_walk(rt, dump_dtl_seg, prefix);
2540 		if (dump_opt['d'] > 5 && vd->vdev_children == 0)
2541 			dump_spacemap(spa->spa_meta_objset,
2542 			    vd->vdev_dtl_sm);
2543 	}
2544 
2545 	for (unsigned c = 0; c < vd->vdev_children; c++)
2546 		dump_dtl(vd->vdev_child[c], indent + 4);
2547 }
2548 
2549 static void
dump_history(spa_t * spa)2550 dump_history(spa_t *spa)
2551 {
2552 	nvlist_t **events = NULL;
2553 	char *buf;
2554 	uint64_t resid, len, off = 0;
2555 	uint_t num = 0;
2556 	int error;
2557 	char tbuf[30];
2558 
2559 	if ((buf = malloc(SPA_OLD_MAXBLOCKSIZE)) == NULL) {
2560 		(void) fprintf(stderr, "%s: unable to allocate I/O buffer\n",
2561 		    __func__);
2562 		return;
2563 	}
2564 
2565 	do {
2566 		len = SPA_OLD_MAXBLOCKSIZE;
2567 
2568 		if ((error = spa_history_get(spa, &off, &len, buf)) != 0) {
2569 			(void) fprintf(stderr, "Unable to read history: "
2570 			    "error %d\n", error);
2571 			free(buf);
2572 			return;
2573 		}
2574 
2575 		if (zpool_history_unpack(buf, len, &resid, &events, &num) != 0)
2576 			break;
2577 
2578 		off -= resid;
2579 	} while (len != 0);
2580 
2581 	(void) printf("\nHistory:\n");
2582 	for (unsigned i = 0; i < num; i++) {
2583 		boolean_t printed = B_FALSE;
2584 
2585 		if (nvlist_exists(events[i], ZPOOL_HIST_TIME)) {
2586 			time_t tsec;
2587 			struct tm t;
2588 
2589 			tsec = fnvlist_lookup_uint64(events[i],
2590 			    ZPOOL_HIST_TIME);
2591 			(void) localtime_r(&tsec, &t);
2592 			(void) strftime(tbuf, sizeof (tbuf), "%F.%T", &t);
2593 		} else {
2594 			tbuf[0] = '\0';
2595 		}
2596 
2597 		if (nvlist_exists(events[i], ZPOOL_HIST_CMD)) {
2598 			(void) printf("%s %s\n", tbuf,
2599 			    fnvlist_lookup_string(events[i], ZPOOL_HIST_CMD));
2600 		} else if (nvlist_exists(events[i], ZPOOL_HIST_INT_EVENT)) {
2601 			uint64_t ievent;
2602 
2603 			ievent = fnvlist_lookup_uint64(events[i],
2604 			    ZPOOL_HIST_INT_EVENT);
2605 			if (ievent >= ZFS_NUM_LEGACY_HISTORY_EVENTS)
2606 				goto next;
2607 
2608 			(void) printf(" %s [internal %s txg:%ju] %s\n",
2609 			    tbuf,
2610 			    zfs_history_event_names[ievent],
2611 			    fnvlist_lookup_uint64(events[i],
2612 			    ZPOOL_HIST_TXG),
2613 			    fnvlist_lookup_string(events[i],
2614 			    ZPOOL_HIST_INT_STR));
2615 		} else if (nvlist_exists(events[i], ZPOOL_HIST_INT_NAME)) {
2616 			(void) printf("%s [txg:%ju] %s", tbuf,
2617 			    fnvlist_lookup_uint64(events[i],
2618 			    ZPOOL_HIST_TXG),
2619 			    fnvlist_lookup_string(events[i],
2620 			    ZPOOL_HIST_INT_NAME));
2621 
2622 			if (nvlist_exists(events[i], ZPOOL_HIST_DSNAME)) {
2623 				(void) printf(" %s (%llu)",
2624 				    fnvlist_lookup_string(events[i],
2625 				    ZPOOL_HIST_DSNAME),
2626 				    (u_longlong_t)fnvlist_lookup_uint64(
2627 				    events[i],
2628 				    ZPOOL_HIST_DSID));
2629 			}
2630 
2631 			(void) printf(" %s\n", fnvlist_lookup_string(events[i],
2632 			    ZPOOL_HIST_INT_STR));
2633 		} else if (nvlist_exists(events[i], ZPOOL_HIST_IOCTL)) {
2634 			(void) printf("%s ioctl %s\n", tbuf,
2635 			    fnvlist_lookup_string(events[i],
2636 			    ZPOOL_HIST_IOCTL));
2637 
2638 			if (nvlist_exists(events[i], ZPOOL_HIST_INPUT_NVL)) {
2639 				(void) printf("    input:\n");
2640 				dump_nvlist(fnvlist_lookup_nvlist(events[i],
2641 				    ZPOOL_HIST_INPUT_NVL), 8);
2642 			}
2643 			if (nvlist_exists(events[i], ZPOOL_HIST_OUTPUT_NVL)) {
2644 				(void) printf("    output:\n");
2645 				dump_nvlist(fnvlist_lookup_nvlist(events[i],
2646 				    ZPOOL_HIST_OUTPUT_NVL), 8);
2647 			}
2648 			if (nvlist_exists(events[i], ZPOOL_HIST_ERRNO)) {
2649 				(void) printf("    errno: %lld\n",
2650 				    (longlong_t)fnvlist_lookup_int64(events[i],
2651 				    ZPOOL_HIST_ERRNO));
2652 			}
2653 		} else {
2654 			goto next;
2655 		}
2656 
2657 		printed = B_TRUE;
2658 next:
2659 		if (dump_opt['h'] > 1) {
2660 			if (!printed)
2661 				(void) printf("unrecognized record:\n");
2662 			dump_nvlist(events[i], 2);
2663 		}
2664 	}
2665 	free(buf);
2666 }
2667 
2668 static void
dump_dnode(objset_t * os,uint64_t object,void * data,size_t size)2669 dump_dnode(objset_t *os, uint64_t object, void *data, size_t size)
2670 {
2671 	(void) os, (void) object, (void) data, (void) size;
2672 }
2673 
2674 static uint64_t
blkid2offset(const dnode_phys_t * dnp,const blkptr_t * bp,const zbookmark_phys_t * zb)2675 blkid2offset(const dnode_phys_t *dnp, const blkptr_t *bp,
2676     const zbookmark_phys_t *zb)
2677 {
2678 	if (dnp == NULL) {
2679 		ASSERT(zb->zb_level < 0);
2680 		if (zb->zb_object == 0)
2681 			return (zb->zb_blkid);
2682 		return (zb->zb_blkid * BP_GET_LSIZE(bp));
2683 	}
2684 
2685 	ASSERT(zb->zb_level >= 0);
2686 
2687 	return ((zb->zb_blkid <<
2688 	    (zb->zb_level * (dnp->dn_indblkshift - SPA_BLKPTRSHIFT))) *
2689 	    dnp->dn_datablkszsec << SPA_MINBLOCKSHIFT);
2690 }
2691 
2692 static void
snprintf_zstd_header(spa_t * spa,char * blkbuf,size_t buflen,const blkptr_t * bp)2693 snprintf_zstd_header(spa_t *spa, char *blkbuf, size_t buflen,
2694     const blkptr_t *bp)
2695 {
2696 	static abd_t *pabd = NULL;
2697 	void *buf;
2698 	zio_t *zio;
2699 	zfs_zstdhdr_t zstd_hdr;
2700 	int error;
2701 
2702 	if (BP_GET_COMPRESS(bp) != ZIO_COMPRESS_ZSTD)
2703 		return;
2704 
2705 	if (BP_IS_HOLE(bp))
2706 		return;
2707 
2708 	if (BP_IS_EMBEDDED(bp)) {
2709 		buf = malloc(SPA_MAXBLOCKSIZE);
2710 		if (buf == NULL) {
2711 			(void) fprintf(stderr, "out of memory\n");
2712 			zdb_exit(1);
2713 		}
2714 		decode_embedded_bp_compressed(bp, buf);
2715 		memcpy(&zstd_hdr, buf, sizeof (zstd_hdr));
2716 		free(buf);
2717 		zstd_hdr.c_len = BE_32(zstd_hdr.c_len);
2718 		zstd_hdr.raw_version_level = BE_32(zstd_hdr.raw_version_level);
2719 		(void) snprintf(blkbuf + strlen(blkbuf),
2720 		    buflen - strlen(blkbuf),
2721 		    " ZSTD:size=%u:version=%u:level=%u:EMBEDDED",
2722 		    zstd_hdr.c_len, zfs_get_hdrversion(&zstd_hdr),
2723 		    zfs_get_hdrlevel(&zstd_hdr));
2724 		return;
2725 	}
2726 
2727 	if (!pabd)
2728 		pabd = abd_alloc_for_io(SPA_MAXBLOCKSIZE, B_FALSE);
2729 	zio = zio_root(spa, NULL, NULL, 0);
2730 
2731 	/* Decrypt but don't decompress so we can read the compression header */
2732 	zio_nowait(zio_read(zio, spa, bp, pabd, BP_GET_PSIZE(bp), NULL, NULL,
2733 	    ZIO_PRIORITY_SYNC_READ, ZIO_FLAG_CANFAIL | ZIO_FLAG_RAW_COMPRESS,
2734 	    NULL));
2735 	error = zio_wait(zio);
2736 	if (error) {
2737 		(void) fprintf(stderr, "read failed: %d\n", error);
2738 		return;
2739 	}
2740 	buf = abd_borrow_buf_copy(pabd, BP_GET_LSIZE(bp));
2741 	memcpy(&zstd_hdr, buf, sizeof (zstd_hdr));
2742 	zstd_hdr.c_len = BE_32(zstd_hdr.c_len);
2743 	zstd_hdr.raw_version_level = BE_32(zstd_hdr.raw_version_level);
2744 
2745 	(void) snprintf(blkbuf + strlen(blkbuf),
2746 	    buflen - strlen(blkbuf),
2747 	    " ZSTD:size=%u:version=%u:level=%u:NORMAL",
2748 	    zstd_hdr.c_len, zfs_get_hdrversion(&zstd_hdr),
2749 	    zfs_get_hdrlevel(&zstd_hdr));
2750 
2751 	abd_return_buf_copy(pabd, buf, BP_GET_LSIZE(bp));
2752 }
2753 
2754 static void
snprintf_blkptr_compact(char * blkbuf,size_t buflen,const blkptr_t * bp,boolean_t bp_freed)2755 snprintf_blkptr_compact(char *blkbuf, size_t buflen, const blkptr_t *bp,
2756     boolean_t bp_freed)
2757 {
2758 	const dva_t *dva = bp->blk_dva;
2759 	int ndvas = dump_opt['d'] > 5 ? BP_GET_NDVAS(bp) : 1;
2760 	int i;
2761 
2762 	if (dump_opt['b'] >= 6) {
2763 		snprintf_blkptr(blkbuf, buflen, bp);
2764 		if (bp_freed) {
2765 			(void) snprintf(blkbuf + strlen(blkbuf),
2766 			    buflen - strlen(blkbuf), " %s", "FREE");
2767 		}
2768 		return;
2769 	}
2770 
2771 	if (BP_IS_EMBEDDED(bp)) {
2772 		(void) sprintf(blkbuf,
2773 		    "EMBEDDED et=%u %llxL/%llxP B=%llu",
2774 		    (int)BPE_GET_ETYPE(bp),
2775 		    (u_longlong_t)BPE_GET_LSIZE(bp),
2776 		    (u_longlong_t)BPE_GET_PSIZE(bp),
2777 		    (u_longlong_t)BP_GET_LOGICAL_BIRTH(bp));
2778 		return;
2779 	}
2780 
2781 	blkbuf[0] = '\0';
2782 
2783 	for (i = 0; i < ndvas; i++) {
2784 		(void) snprintf(blkbuf + strlen(blkbuf),
2785 		    buflen - strlen(blkbuf), "%llu:%llx:%llx%s ",
2786 		    (u_longlong_t)DVA_GET_VDEV(&dva[i]),
2787 		    (u_longlong_t)DVA_GET_OFFSET(&dva[i]),
2788 		    (u_longlong_t)DVA_GET_ASIZE(&dva[i]),
2789 		    (DVA_GET_GANG(&dva[i]) ? "G" : ""));
2790 	}
2791 
2792 	if (BP_IS_HOLE(bp)) {
2793 		(void) snprintf(blkbuf + strlen(blkbuf),
2794 		    buflen - strlen(blkbuf),
2795 		    "%llxL B=%llu",
2796 		    (u_longlong_t)BP_GET_LSIZE(bp),
2797 		    (u_longlong_t)BP_GET_LOGICAL_BIRTH(bp));
2798 	} else {
2799 		(void) snprintf(blkbuf + strlen(blkbuf),
2800 		    buflen - strlen(blkbuf),
2801 		    "%llxL/%llxP F=%llu B=%llu/%llu",
2802 		    (u_longlong_t)BP_GET_LSIZE(bp),
2803 		    (u_longlong_t)BP_GET_PSIZE(bp),
2804 		    (u_longlong_t)BP_GET_FILL(bp),
2805 		    (u_longlong_t)BP_GET_LOGICAL_BIRTH(bp),
2806 		    (u_longlong_t)BP_GET_PHYSICAL_BIRTH(bp));
2807 		if (bp_freed)
2808 			(void) snprintf(blkbuf + strlen(blkbuf),
2809 			    buflen - strlen(blkbuf), " %s", "FREE");
2810 
2811 		(void) snprintf(blkbuf + strlen(blkbuf),
2812 		    buflen - strlen(blkbuf),
2813 		    " cksum=%016llx:%016llx:%016llx:%016llx",
2814 		    (u_longlong_t)bp->blk_cksum.zc_word[0],
2815 		    (u_longlong_t)bp->blk_cksum.zc_word[1],
2816 		    (u_longlong_t)bp->blk_cksum.zc_word[2],
2817 		    (u_longlong_t)bp->blk_cksum.zc_word[3]);
2818 	}
2819 }
2820 
2821 static void
inverse_text(boolean_t on)2822 inverse_text(boolean_t on)
2823 {
2824 	if (isatty(STDOUT_FILENO)) {
2825 		if (on)
2826 			fputs("\x1b[7m", stdout);
2827 		else
2828 			fputs("\x1b[m", stdout);
2829 	}
2830 }
2831 
2832 static void
print_file_layout_line(int line,int first_disk,int last_disk,int ashift,raidz_row_t * rr)2833 print_file_layout_line(int line, int first_disk, int last_disk, int ashift,
2834     raidz_row_t *rr)
2835 {
2836 	if (first_disk != 0) {
2837 		/* Account for empty columns */
2838 		for (int c = 0; c < first_disk; c++) {
2839 			(void) printf("%s%s", (c == 0) ? " │ " : "  ",
2840 			    line == 0 ? "` ` ` ` ` " : " ` ` ` ` `");
2841 		}
2842 	}
2843 
2844 	/*
2845 	 * Check if we need to account for out of order disks in raidz1.
2846 	 * Here flip means the disks in first two column were exchanged.
2847 	 */
2848 	boolean_t flip = rr->rr_firstdatacol == 1 &&
2849 	    (rr->rr_offset & (1ULL << 20));
2850 
2851 	for (int c = 0; c < rr->rr_cols; c++) {
2852 		raidz_col_t *rc = &rr->rr_col[c];
2853 		char colname[8];
2854 		boolean_t pcol = flip ? (c == 1) : (c < rr->rr_firstdatacol);
2855 
2856 		if (rc->rc_devidx > last_disk)
2857 			continue;
2858 
2859 		(void) snprintf(colname, sizeof (colname), "%c%d",
2860 		    pcol ? 'P' : 'D', (flip && c < 2) ? 0 :
2861 		    pcol ? c : c - rr->rr_firstdatacol);
2862 
2863 		(void) printf("%s", (rc->rc_devidx == 0) ? " │ " : "  ");
2864 		if (pcol)
2865 			inverse_text(B_TRUE);
2866 		if (line == 0) {
2867 			(void) printf("%3s  %5d", colname,
2868 			    (int)(rc->rc_size >> ashift));
2869 		} else {
2870 			(void) printf("%10llx",
2871 			    (u_longlong_t)((rc->rc_offset +
2872 			    VDEV_LABEL_START_SIZE) >> ashift));
2873 		}
2874 		if (pcol)
2875 			inverse_text(B_FALSE);
2876 		if (rc->rc_devidx == last_disk) {
2877 			/* empty columns */
2878 			for (int c = last_disk + 1; c < rr->rr_cols; c++) {
2879 				(void) printf("  %s",
2880 				    line == 0 ? "` ` ` ` ` " : " ` ` ` ` `");
2881 			}
2882 			break;
2883 		}
2884 	}
2885 	(void) printf(" │\n");
2886 }
2887 
2888 
2889 static u_longlong_t
print_indirect(spa_t * spa,blkptr_t * bp,const zbookmark_phys_t * zb,const dnode_phys_t * dnp)2890 print_indirect(spa_t *spa, blkptr_t *bp, const zbookmark_phys_t *zb,
2891     const dnode_phys_t *dnp)
2892 {
2893 	char blkbuf[BP_SPRINTF_LEN];
2894 	u_longlong_t offset;
2895 	int l;
2896 
2897 	offset = (u_longlong_t)blkid2offset(dnp, bp, zb);
2898 
2899 	(void) printf("%16llx ", offset);
2900 
2901 	ASSERT(zb->zb_level >= 0);
2902 
2903 	for (l = dnp->dn_nlevels - 1; l >= -1; l--) {
2904 		if (l == zb->zb_level) {
2905 			(void) printf("L%llx", (u_longlong_t)zb->zb_level);
2906 		} else {
2907 			(void) printf(" ");
2908 		}
2909 	}
2910 
2911 	snprintf_blkptr_compact(blkbuf, sizeof (blkbuf), bp, B_FALSE);
2912 	if (dump_opt['Z'] && BP_GET_COMPRESS(bp) == ZIO_COMPRESS_ZSTD)
2913 		snprintf_zstd_header(spa, blkbuf, sizeof (blkbuf), bp);
2914 	(void) printf("%s", blkbuf);
2915 
2916 	if (!BP_IS_EMBEDDED(bp)) {
2917 		if (BP_GET_TYPE(bp) != dnp->dn_type) {
2918 			(void) printf(" (ERROR: Block pointer type "
2919 			    "(%llu) does not match dnode type (%hhu))",
2920 			    BP_GET_TYPE(bp), dnp->dn_type);
2921 			corruption_found = B_TRUE;
2922 		}
2923 		if (BP_GET_LEVEL(bp) != zb->zb_level) {
2924 			(void) printf(" (ERROR: Block pointer level "
2925 			    "(%llu) does not match bookmark level (%lld))",
2926 			    BP_GET_LEVEL(bp), (longlong_t)zb->zb_level);
2927 			corruption_found = B_TRUE;
2928 		}
2929 	}
2930 	(void) printf("\n");
2931 
2932 	return (offset);
2933 }
2934 
2935 static void
print_file_layout_raidz(vdev_t * vd,blkptr_t * bp,uint64_t file_offset,boolean_t last)2936 print_file_layout_raidz(vdev_t *vd, blkptr_t *bp, uint64_t file_offset,
2937     boolean_t last)
2938 {
2939 	/*
2940 	 * RAIDZ file data layout
2941 	 *
2942 	 * Use vdev_raidz_map_alloc() to get the actual layout
2943 	 */
2944 	const dva_t *dva = bp->blk_dva;
2945 	zio_t zio = {0};
2946 	zio.io_size = P2ROUNDUP_TYPED(BP_GET_PSIZE(bp),
2947 	    1ULL << vd->vdev_ashift, uint64_t);
2948 	zio.io_offset = DVA_GET_OFFSET(&dva[0]);
2949 	zio.io_type = ZIO_TYPE_READ;
2950 	zio.io_abd = abd_alloc_for_io(zio.io_size, B_FALSE);
2951 
2952 	vdev_raidz_t *vdrz = vd->vdev_tsd;
2953 	raidz_map_t *rm = vdev_raidz_map_alloc(&zio, vd->vdev_ashift,
2954 	    vd->vdev_children, vdrz->vd_nparity);
2955 	raidz_row_t *rr = rm->rm_row[0];
2956 
2957 	if (!dump_opt['H']) {
2958 		int last_disk = vd->vdev_children - 1;
2959 		/*
2960 		 * Account for out of order disks in raidz1.
2961 		 * For now just reverse them back and adjust for it later.
2962 		 */
2963 		if (rr->rr_firstdatacol == 1 &&
2964 		    (zio.io_offset & (1ULL << 20))) {
2965 			uint64_t devidx = rr->rr_col[0].rc_devidx;
2966 			rr->rr_col[0].rc_devidx = rr->rr_col[1].rc_devidx;
2967 			rr->rr_col[1].rc_devidx = devidx;
2968 		}
2969 		int first_disk = rr->rr_col[0].rc_devidx;
2970 
2971 		(void) printf("%12llx", (u_longlong_t)file_offset);
2972 		print_file_layout_line(0, first_disk, last_disk,
2973 		    vd->vdev_ashift, rr);
2974 		(void) printf("%*c", 12, ' ');
2975 		print_file_layout_line(1, first_disk, last_disk,
2976 		    vd->vdev_ashift, rr);
2977 		/* Check for split row */
2978 		if (first_disk != 0) {
2979 			(void) printf("%*c", 12, ' ');
2980 			print_file_layout_line(0, 0, first_disk - 1,
2981 			    vd->vdev_ashift, rr);
2982 			(void) printf("%*c", 12, ' ');
2983 			print_file_layout_line(1, 0, first_disk - 1,
2984 			    vd->vdev_ashift, rr);
2985 		}
2986 		/* seperate rows with a line */
2987 		(void) printf("%*c", 12, ' ');
2988 		for (int c = 0; c < vd->vdev_children; c++) {
2989 			(void) printf("%s%10s", (c == 0) ?
2990 			    (last ? " └─" : " ├─") : "──",
2991 			    "──────────");
2992 		}
2993 		(void) printf("─%s\n", last ? "┘" : "┤");
2994 	} else {
2995 		static uint64_t next_offset = 0;
2996 
2997 		if (next_offset != file_offset) {
2998 			(void) printf("skip hole\t-\t\t%lld\n",
2999 			    (u_longlong_t)((file_offset - next_offset) / 512));
3000 		}
3001 		next_offset = file_offset + BP_GET_LSIZE(bp);
3002 		uint64_t tmp_offset = file_offset;
3003 
3004 
3005 		for (int c = 0; c < rr->rr_cols; c++) {
3006 			boolean_t pcol = c < rr->rr_firstdatacol;
3007 			raidz_col_t *rc = &rr->rr_col[c];
3008 			char *path = vd->vdev_child[rc->rc_devidx]->vdev_path;
3009 
3010 			if (rc->rc_size == 0)
3011 				continue;
3012 			(void) printf("%s\t\t%llu\t%d",
3013 			    zfs_basename(path),
3014 			    (u_longlong_t)(rc->rc_offset +
3015 			    VDEV_LABEL_START_SIZE)/512,
3016 			    (int)rc->rc_size/512);
3017 			if (dump_opt['v']) {
3018 				char label = pcol ? 'P' : 'D';
3019 				int num;
3020 
3021 				if (c < 2) {
3022 					num = 0;
3023 				} else {
3024 					num = pcol ? c :
3025 					    (c - rr->rr_firstdatacol);
3026 				}
3027 				printf("\t%c%d", label, num);
3028 				if (dump_opt['v'] > 1) {
3029 					unsigned long long off;
3030 					if (pcol)
3031 						off = file_offset;
3032 					else
3033 						off = tmp_offset;
3034 					off = off / 512ULL;
3035 					printf("\t%llu", off);
3036 				}
3037 			}
3038 			if (!pcol)
3039 				tmp_offset += rc->rc_size;
3040 			printf("\n");
3041 		}
3042 	}
3043 }
3044 
3045 static void
print_file_layout(spa_t * spa,blkptr_t * bp,const zbookmark_phys_t * zb,const dnode_phys_t * dnp)3046 print_file_layout(spa_t *spa, blkptr_t *bp, const zbookmark_phys_t *zb,
3047     const dnode_phys_t *dnp)
3048 {
3049 	if (!BP_IS_EMBEDDED(bp)) {
3050 		ASSERT3U(BP_GET_TYPE(bp), ==, dnp->dn_type);
3051 		ASSERT3U(BP_GET_LEVEL(bp), ==, zb->zb_level);
3052 	}
3053 	ASSERT(zb->zb_level >= 0);
3054 
3055 	if (BP_IS_HOLE(bp))
3056 		return;
3057 
3058 	if (BP_IS_EMBEDDED(bp))
3059 		return;
3060 
3061 	const dva_t *dva = bp->blk_dva;
3062 	vdev_t *vd = spa->spa_root_vdev->vdev_child[DVA_GET_VDEV(&dva[0])];
3063 	uint64_t file_offset = blkid2offset(dnp, bp, zb);
3064 
3065 	if (strcmp("raidz", vd->vdev_ops->vdev_op_type) == 0) {
3066 		boolean_t last = (file_offset + BP_GET_LSIZE(bp)) >=
3067 		    (dnp->dn_used & ~0x03ffULL);
3068 		print_file_layout_raidz(vd, bp, file_offset, last);
3069 	}
3070 }
3071 
3072 static int
visit_indirect(spa_t * spa,const dnode_phys_t * dnp,blkptr_t * bp,const zbookmark_phys_t * zb)3073 visit_indirect(spa_t *spa, const dnode_phys_t *dnp,
3074     blkptr_t *bp, const zbookmark_phys_t *zb)
3075 {
3076 	u_longlong_t offset = 0;
3077 	int err = 0;
3078 
3079 	if (BP_GET_BIRTH(bp) == 0)
3080 		return (0);
3081 
3082 	if (dump_opt['f']) {
3083 		if (BP_GET_LEVEL(bp) == 0)
3084 			print_file_layout(spa, bp, zb, dnp);
3085 	} else {
3086 		print_indirect(spa, bp, zb, dnp);
3087 	}
3088 
3089 	if (BP_GET_LEVEL(bp) > 0 && !BP_IS_HOLE(bp)) {
3090 		arc_flags_t flags = ARC_FLAG_WAIT;
3091 		int i;
3092 		blkptr_t *cbp;
3093 		int epb = BP_GET_LSIZE(bp) >> SPA_BLKPTRSHIFT;
3094 		arc_buf_t *buf;
3095 		uint64_t fill = 0;
3096 		ASSERT(!BP_IS_REDACTED(bp));
3097 
3098 		err = arc_read(NULL, spa, bp, arc_getbuf_func, &buf,
3099 		    ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
3100 		if (err)
3101 			return (err);
3102 		ASSERT(buf->b_data);
3103 
3104 		/* recursively visit blocks below this */
3105 		cbp = buf->b_data;
3106 		for (i = 0; i < epb; i++, cbp++) {
3107 			zbookmark_phys_t czb;
3108 
3109 			SET_BOOKMARK(&czb, zb->zb_objset, zb->zb_object,
3110 			    zb->zb_level - 1,
3111 			    zb->zb_blkid * epb + i);
3112 			err = visit_indirect(spa, dnp, cbp, &czb);
3113 			if (err)
3114 				break;
3115 			fill += BP_GET_FILL(cbp);
3116 		}
3117 		if (!err) {
3118 			if (fill != BP_GET_FILL(bp)) {
3119 				(void) printf("%16llx: Block pointer "
3120 				    "fill (%llu) does not match calculated "
3121 				    "value (%llu)\n", offset, BP_GET_FILL(bp),
3122 				    (u_longlong_t)fill);
3123 				corruption_found = B_TRUE;
3124 			}
3125 		}
3126 		arc_buf_destroy(buf, &buf);
3127 	}
3128 
3129 	return (err);
3130 }
3131 
3132 static void
dump_indirect(dnode_t * dn)3133 dump_indirect(dnode_t *dn)
3134 {
3135 	dnode_phys_t *dnp = dn->dn_phys;
3136 	zbookmark_phys_t czb;
3137 
3138 	(void) printf("Indirect blocks:\n");
3139 	SET_BOOKMARK(&czb, dmu_objset_id(dn->dn_objset),
3140 	    dn->dn_object, dnp->dn_nlevels - 1, 0);
3141 	for (int j = 0; j < dnp->dn_nblkptr; j++) {
3142 		czb.zb_blkid = j;
3143 		(void) visit_indirect(dmu_objset_spa(dn->dn_objset), dnp,
3144 		    &dnp->dn_blkptr[j], &czb);
3145 	}
3146 
3147 	(void) printf("\n");
3148 }
3149 
3150 static int
dump_indirect_layout(dnode_t * dn)3151 dump_indirect_layout(dnode_t *dn)
3152 {
3153 	dnode_phys_t *dnp = dn->dn_phys;
3154 	zbookmark_phys_t czb;
3155 
3156 	spa_t *spa = dmu_objset_spa(dn->dn_objset);
3157 	int ashift = spa->spa_root_vdev->vdev_child[0]->vdev_ashift;
3158 	int children = spa->spa_root_vdev->vdev_child[0]->vdev_children;
3159 
3160 	if (strcmp(spa->spa_root_vdev->vdev_child[0]->vdev_ops->vdev_op_type,
3161 	    "raidz") != 0) {
3162 		(void) fprintf(stderr, "file layout only supports raidz\n");
3163 		return (ENOTSUP);
3164 	}
3165 
3166 	/*
3167 	 * Start layout with a header
3168 	 */
3169 	if (dump_opt['H']) {
3170 		(void) printf("DISK\t\t\tLBA\tCOUNT");
3171 		if (dump_opt['v'])
3172 			(void) printf("\tTYPE");
3173 		if (dump_opt['v'] > 1)
3174 			(void) printf("\tOFFSET");
3175 		printf("\n");
3176 	} else {
3177 		char diskhdr[16];
3178 
3179 		(void) printf("%12s: %d\n", "block size", dn->dn_datablksz);
3180 		(void) printf("%12s: %s%d\n", "vdev_type",
3181 		    spa->spa_root_vdev->vdev_child[0]->vdev_ops->vdev_op_type,
3182 		    (int)vdev_get_nparity(spa->spa_root_vdev->vdev_child[0]));
3183 		(void) printf("%12s: %d\n", "sector size", 1 << ashift);
3184 		(void) printf("%12s: %d\n\n", "child disks", children);
3185 
3186 		(void) printf("%*c ", 12, ' ');
3187 
3188 		for (int c = 0; c < children; c++) {
3189 			(void) snprintf(diskhdr, sizeof (diskhdr),
3190 			    "V%d:DISK-%d", 0, c);
3191 			(void) printf(" %10s ", diskhdr);
3192 		}
3193 		(void) printf("\n");
3194 
3195 		(void) printf("%12s", "FILE OFFSET");
3196 		for (int c = 0; c < children; c++)
3197 			(void) printf("%s%10s", (c == 0) ? " ┌─" : "──",
3198 			    "──────────");
3199 		(void) printf("─┐\n");
3200 	}
3201 
3202 	SET_BOOKMARK(&czb, dmu_objset_id(dn->dn_objset),
3203 	    dn->dn_object, dnp->dn_nlevels - 1, 0);
3204 	for (int j = 0; j < dnp->dn_nblkptr; j++) {
3205 		czb.zb_blkid = j;
3206 		(void) visit_indirect(dmu_objset_spa(dn->dn_objset), dnp,
3207 		    &dnp->dn_blkptr[j], &czb);
3208 	}
3209 
3210 	(void) printf("\n");
3211 	return (0);
3212 }
3213 
3214 static void
dump_dsl_dir(objset_t * os,uint64_t object,void * data,size_t size)3215 dump_dsl_dir(objset_t *os, uint64_t object, void *data, size_t size)
3216 {
3217 	(void) os, (void) object;
3218 	dsl_dir_phys_t *dd = data;
3219 	time_t crtime;
3220 	char nice[32];
3221 
3222 	/* make sure nicenum has enough space */
3223 	_Static_assert(sizeof (nice) >= NN_NUMBUF_SZ, "nice truncated");
3224 
3225 	if (dd == NULL)
3226 		return;
3227 
3228 	ASSERT3U(size, >=, sizeof (dsl_dir_phys_t));
3229 
3230 	crtime = dd->dd_creation_time;
3231 	(void) printf("\t\tcreation_time = %s", ctime(&crtime));
3232 	(void) printf("\t\thead_dataset_obj = %llu\n",
3233 	    (u_longlong_t)dd->dd_head_dataset_obj);
3234 	(void) printf("\t\tparent_dir_obj = %llu\n",
3235 	    (u_longlong_t)dd->dd_parent_obj);
3236 	(void) printf("\t\torigin_obj = %llu\n",
3237 	    (u_longlong_t)dd->dd_origin_obj);
3238 	(void) printf("\t\tchild_dir_zapobj = %llu\n",
3239 	    (u_longlong_t)dd->dd_child_dir_zapobj);
3240 	zdb_nicenum(dd->dd_used_bytes, nice, sizeof (nice));
3241 	(void) printf("\t\tused_bytes = %s\n", nice);
3242 	zdb_nicenum(dd->dd_compressed_bytes, nice, sizeof (nice));
3243 	(void) printf("\t\tcompressed_bytes = %s\n", nice);
3244 	zdb_nicenum(dd->dd_uncompressed_bytes, nice, sizeof (nice));
3245 	(void) printf("\t\tuncompressed_bytes = %s\n", nice);
3246 	zdb_nicenum(dd->dd_quota, nice, sizeof (nice));
3247 	(void) printf("\t\tquota = %s\n", nice);
3248 	zdb_nicenum(dd->dd_reserved, nice, sizeof (nice));
3249 	(void) printf("\t\treserved = %s\n", nice);
3250 	(void) printf("\t\tprops_zapobj = %llu\n",
3251 	    (u_longlong_t)dd->dd_props_zapobj);
3252 	(void) printf("\t\tdeleg_zapobj = %llu\n",
3253 	    (u_longlong_t)dd->dd_deleg_zapobj);
3254 	(void) printf("\t\tflags = %llx\n",
3255 	    (u_longlong_t)dd->dd_flags);
3256 
3257 #define	DO(which) \
3258 	zdb_nicenum(dd->dd_used_breakdown[DD_USED_ ## which], nice, \
3259 	    sizeof (nice)); \
3260 	(void) printf("\t\tused_breakdown[" #which "] = %s\n", nice)
3261 	DO(HEAD);
3262 	DO(SNAP);
3263 	DO(CHILD);
3264 	DO(CHILD_RSRV);
3265 	DO(REFRSRV);
3266 #undef DO
3267 	(void) printf("\t\tclones = %llu\n",
3268 	    (u_longlong_t)dd->dd_clones);
3269 }
3270 
3271 static void
dump_dsl_dataset(objset_t * os,uint64_t object,void * data,size_t size)3272 dump_dsl_dataset(objset_t *os, uint64_t object, void *data, size_t size)
3273 {
3274 	(void) os, (void) object;
3275 	dsl_dataset_phys_t *ds = data;
3276 	time_t crtime;
3277 	char used[32], compressed[32], uncompressed[32], unique[32];
3278 	char blkbuf[BP_SPRINTF_LEN];
3279 
3280 	/* make sure nicenum has enough space */
3281 	_Static_assert(sizeof (used) >= NN_NUMBUF_SZ, "used truncated");
3282 	_Static_assert(sizeof (compressed) >= NN_NUMBUF_SZ,
3283 	    "compressed truncated");
3284 	_Static_assert(sizeof (uncompressed) >= NN_NUMBUF_SZ,
3285 	    "uncompressed truncated");
3286 	_Static_assert(sizeof (unique) >= NN_NUMBUF_SZ, "unique truncated");
3287 
3288 	if (ds == NULL)
3289 		return;
3290 
3291 	ASSERT(size == sizeof (*ds));
3292 	crtime = ds->ds_creation_time;
3293 	zdb_nicenum(ds->ds_referenced_bytes, used, sizeof (used));
3294 	zdb_nicenum(ds->ds_compressed_bytes, compressed, sizeof (compressed));
3295 	zdb_nicenum(ds->ds_uncompressed_bytes, uncompressed,
3296 	    sizeof (uncompressed));
3297 	zdb_nicenum(ds->ds_unique_bytes, unique, sizeof (unique));
3298 	snprintf_blkptr(blkbuf, sizeof (blkbuf), &ds->ds_bp);
3299 
3300 	(void) printf("\t\tdir_obj = %llu\n",
3301 	    (u_longlong_t)ds->ds_dir_obj);
3302 	(void) printf("\t\tprev_snap_obj = %llu\n",
3303 	    (u_longlong_t)ds->ds_prev_snap_obj);
3304 	(void) printf("\t\tprev_snap_txg = %llu\n",
3305 	    (u_longlong_t)ds->ds_prev_snap_txg);
3306 	(void) printf("\t\tnext_snap_obj = %llu\n",
3307 	    (u_longlong_t)ds->ds_next_snap_obj);
3308 	(void) printf("\t\tsnapnames_zapobj = %llu\n",
3309 	    (u_longlong_t)ds->ds_snapnames_zapobj);
3310 	(void) printf("\t\tnum_children = %llu\n",
3311 	    (u_longlong_t)ds->ds_num_children);
3312 	(void) printf("\t\tuserrefs_obj = %llu\n",
3313 	    (u_longlong_t)ds->ds_userrefs_obj);
3314 	(void) printf("\t\tcreation_time = %s", ctime(&crtime));
3315 	(void) printf("\t\tcreation_txg = %llu\n",
3316 	    (u_longlong_t)ds->ds_creation_txg);
3317 	(void) printf("\t\tdeadlist_obj = %llu\n",
3318 	    (u_longlong_t)ds->ds_deadlist_obj);
3319 	(void) printf("\t\tused_bytes = %s\n", used);
3320 	(void) printf("\t\tcompressed_bytes = %s\n", compressed);
3321 	(void) printf("\t\tuncompressed_bytes = %s\n", uncompressed);
3322 	(void) printf("\t\tunique = %s\n", unique);
3323 	(void) printf("\t\tfsid_guid = %llu\n",
3324 	    (u_longlong_t)ds->ds_fsid_guid);
3325 	(void) printf("\t\tguid = %llu\n",
3326 	    (u_longlong_t)ds->ds_guid);
3327 	(void) printf("\t\tflags = %llx\n",
3328 	    (u_longlong_t)ds->ds_flags);
3329 	(void) printf("\t\tnext_clones_obj = %llu\n",
3330 	    (u_longlong_t)ds->ds_next_clones_obj);
3331 	(void) printf("\t\tprops_obj = %llu\n",
3332 	    (u_longlong_t)ds->ds_props_obj);
3333 	(void) printf("\t\tbp = %s\n", blkbuf);
3334 }
3335 
3336 static int
dump_bptree_cb(void * arg,const blkptr_t * bp,dmu_tx_t * tx)3337 dump_bptree_cb(void *arg, const blkptr_t *bp, dmu_tx_t *tx)
3338 {
3339 	(void) arg, (void) tx;
3340 	char blkbuf[BP_SPRINTF_LEN];
3341 
3342 	if (BP_GET_BIRTH(bp) != 0) {
3343 		snprintf_blkptr(blkbuf, sizeof (blkbuf), bp);
3344 		(void) printf("\t%s\n", blkbuf);
3345 	}
3346 	return (0);
3347 }
3348 
3349 static void
dump_bptree(objset_t * os,uint64_t obj,const char * name)3350 dump_bptree(objset_t *os, uint64_t obj, const char *name)
3351 {
3352 	char bytes[32];
3353 	bptree_phys_t *bt;
3354 	dmu_buf_t *db;
3355 
3356 	/* make sure nicenum has enough space */
3357 	_Static_assert(sizeof (bytes) >= NN_NUMBUF_SZ, "bytes truncated");
3358 
3359 	if (dump_opt['d'] < 3)
3360 		return;
3361 
3362 	VERIFY3U(0, ==, dmu_bonus_hold(os, obj, FTAG, &db));
3363 	bt = db->db_data;
3364 	zdb_nicenum(bt->bt_bytes, bytes, sizeof (bytes));
3365 	(void) printf("\n    %s: %llu datasets, %s\n",
3366 	    name, (unsigned long long)(bt->bt_end - bt->bt_begin), bytes);
3367 	dmu_buf_rele(db, FTAG);
3368 
3369 	if (dump_opt['d'] < 5)
3370 		return;
3371 
3372 	(void) printf("\n");
3373 
3374 	(void) bptree_iterate(os, obj, B_FALSE, dump_bptree_cb, NULL, NULL);
3375 }
3376 
3377 static int
dump_bpobj_cb(void * arg,const blkptr_t * bp,boolean_t bp_freed,dmu_tx_t * tx)3378 dump_bpobj_cb(void *arg, const blkptr_t *bp, boolean_t bp_freed, dmu_tx_t *tx)
3379 {
3380 	(void) arg, (void) tx;
3381 	char blkbuf[BP_SPRINTF_LEN];
3382 
3383 	ASSERT(BP_GET_BIRTH(bp) != 0);
3384 	snprintf_blkptr_compact(blkbuf, sizeof (blkbuf), bp, bp_freed);
3385 	(void) printf("\t%s\n", blkbuf);
3386 	return (0);
3387 }
3388 
3389 static void
dump_full_bpobj(bpobj_t * bpo,const char * name,int indent)3390 dump_full_bpobj(bpobj_t *bpo, const char *name, int indent)
3391 {
3392 	char bytes[32];
3393 	char comp[32];
3394 	char uncomp[32];
3395 	uint64_t i;
3396 
3397 	/* make sure nicenum has enough space */
3398 	_Static_assert(sizeof (bytes) >= NN_NUMBUF_SZ, "bytes truncated");
3399 	_Static_assert(sizeof (comp) >= NN_NUMBUF_SZ, "comp truncated");
3400 	_Static_assert(sizeof (uncomp) >= NN_NUMBUF_SZ, "uncomp truncated");
3401 
3402 	if (dump_opt['d'] < 3)
3403 		return;
3404 
3405 	zdb_nicenum(bpo->bpo_phys->bpo_bytes, bytes, sizeof (bytes));
3406 	if (bpo->bpo_havesubobj && bpo->bpo_phys->bpo_subobjs != 0) {
3407 		zdb_nicenum(bpo->bpo_phys->bpo_comp, comp, sizeof (comp));
3408 		zdb_nicenum(bpo->bpo_phys->bpo_uncomp, uncomp, sizeof (uncomp));
3409 		if (bpo->bpo_havefreed) {
3410 			(void) printf("    %*s: object %llu, %llu local "
3411 			    "blkptrs, %llu freed, %llu subobjs in object %llu, "
3412 			    "%s (%s/%s comp)\n",
3413 			    indent * 8, name,
3414 			    (u_longlong_t)bpo->bpo_object,
3415 			    (u_longlong_t)bpo->bpo_phys->bpo_num_blkptrs,
3416 			    (u_longlong_t)bpo->bpo_phys->bpo_num_freed,
3417 			    (u_longlong_t)bpo->bpo_phys->bpo_num_subobjs,
3418 			    (u_longlong_t)bpo->bpo_phys->bpo_subobjs,
3419 			    bytes, comp, uncomp);
3420 		} else {
3421 			(void) printf("    %*s: object %llu, %llu local "
3422 			    "blkptrs, %llu subobjs in object %llu, "
3423 			    "%s (%s/%s comp)\n",
3424 			    indent * 8, name,
3425 			    (u_longlong_t)bpo->bpo_object,
3426 			    (u_longlong_t)bpo->bpo_phys->bpo_num_blkptrs,
3427 			    (u_longlong_t)bpo->bpo_phys->bpo_num_subobjs,
3428 			    (u_longlong_t)bpo->bpo_phys->bpo_subobjs,
3429 			    bytes, comp, uncomp);
3430 		}
3431 
3432 		for (i = 0; i < bpo->bpo_phys->bpo_num_subobjs; i++) {
3433 			uint64_t subobj;
3434 			bpobj_t subbpo;
3435 			int error;
3436 			VERIFY0(dmu_read(bpo->bpo_os,
3437 			    bpo->bpo_phys->bpo_subobjs,
3438 			    i * sizeof (subobj), sizeof (subobj), &subobj, 0));
3439 			error = bpobj_open(&subbpo, bpo->bpo_os, subobj);
3440 			if (error != 0) {
3441 				(void) printf("ERROR %u while trying to open "
3442 				    "subobj id %llu\n",
3443 				    error, (u_longlong_t)subobj);
3444 				corruption_found = B_TRUE;
3445 				continue;
3446 			}
3447 			dump_full_bpobj(&subbpo, "subobj", indent + 1);
3448 			bpobj_close(&subbpo);
3449 		}
3450 	} else {
3451 		if (bpo->bpo_havefreed) {
3452 			(void) printf("    %*s: object %llu, %llu blkptrs, "
3453 			    "%llu freed, %s\n",
3454 			    indent * 8, name,
3455 			    (u_longlong_t)bpo->bpo_object,
3456 			    (u_longlong_t)bpo->bpo_phys->bpo_num_blkptrs,
3457 			    (u_longlong_t)bpo->bpo_phys->bpo_num_freed,
3458 			    bytes);
3459 		} else {
3460 			(void) printf("    %*s: object %llu, %llu blkptrs, "
3461 			    "%s\n",
3462 			    indent * 8, name,
3463 			    (u_longlong_t)bpo->bpo_object,
3464 			    (u_longlong_t)bpo->bpo_phys->bpo_num_blkptrs,
3465 			    bytes);
3466 		}
3467 	}
3468 
3469 	if (dump_opt['d'] < 5)
3470 		return;
3471 
3472 	if (indent == 0) {
3473 		(void) bpobj_iterate_nofree(bpo, dump_bpobj_cb, NULL, NULL);
3474 		(void) printf("\n");
3475 	}
3476 }
3477 
3478 static int
dump_bookmark(dsl_pool_t * dp,char * name,boolean_t print_redact,boolean_t print_list)3479 dump_bookmark(dsl_pool_t *dp, char *name, boolean_t print_redact,
3480     boolean_t print_list)
3481 {
3482 	int err = 0;
3483 	zfs_bookmark_phys_t prop;
3484 	objset_t *mos = dp->dp_spa->spa_meta_objset;
3485 	err = dsl_bookmark_lookup(dp, name, NULL, &prop);
3486 
3487 	if (err != 0) {
3488 		return (err);
3489 	}
3490 
3491 	(void) printf("\t#%s: ", strchr(name, '#') + 1);
3492 	(void) printf("{guid: %llx creation_txg: %llu creation_time: "
3493 	    "%llu redaction_obj: %llu}\n", (u_longlong_t)prop.zbm_guid,
3494 	    (u_longlong_t)prop.zbm_creation_txg,
3495 	    (u_longlong_t)prop.zbm_creation_time,
3496 	    (u_longlong_t)prop.zbm_redaction_obj);
3497 
3498 	IMPLY(print_list, print_redact);
3499 	if (!print_redact || prop.zbm_redaction_obj == 0)
3500 		return (0);
3501 
3502 	redaction_list_t *rl;
3503 	VERIFY0(dsl_redaction_list_hold_obj(dp,
3504 	    prop.zbm_redaction_obj, FTAG, &rl));
3505 
3506 	redaction_list_phys_t *rlp = rl->rl_phys;
3507 	(void) printf("\tRedacted:\n\t\tProgress: ");
3508 	if (rlp->rlp_last_object != UINT64_MAX ||
3509 	    rlp->rlp_last_blkid != UINT64_MAX) {
3510 		(void) printf("%llu %llu (incomplete)\n",
3511 		    (u_longlong_t)rlp->rlp_last_object,
3512 		    (u_longlong_t)rlp->rlp_last_blkid);
3513 	} else {
3514 		(void) printf("complete\n");
3515 	}
3516 	(void) printf("\t\tSnapshots: [");
3517 	for (unsigned int i = 0; i < rlp->rlp_num_snaps; i++) {
3518 		if (i > 0)
3519 			(void) printf(", ");
3520 		(void) printf("%0llu",
3521 		    (u_longlong_t)rlp->rlp_snaps[i]);
3522 	}
3523 	(void) printf("]\n\t\tLength: %llu\n",
3524 	    (u_longlong_t)rlp->rlp_num_entries);
3525 
3526 	if (!print_list) {
3527 		dsl_redaction_list_rele(rl, FTAG);
3528 		return (0);
3529 	}
3530 
3531 	if (rlp->rlp_num_entries == 0) {
3532 		dsl_redaction_list_rele(rl, FTAG);
3533 		(void) printf("\t\tRedaction List: []\n\n");
3534 		return (0);
3535 	}
3536 
3537 	redact_block_phys_t *rbp_buf;
3538 	uint64_t size;
3539 	dmu_object_info_t doi;
3540 
3541 	VERIFY0(dmu_object_info(mos, prop.zbm_redaction_obj, &doi));
3542 	size = doi.doi_max_offset;
3543 	rbp_buf = kmem_alloc(size, KM_SLEEP);
3544 
3545 	err = dmu_read(mos, prop.zbm_redaction_obj, 0, size,
3546 	    rbp_buf, 0);
3547 	if (err != 0) {
3548 		dsl_redaction_list_rele(rl, FTAG);
3549 		kmem_free(rbp_buf, size);
3550 		return (err);
3551 	}
3552 
3553 	(void) printf("\t\tRedaction List: [{object: %llx, offset: "
3554 	    "%llx, blksz: %x, count: %llx}",
3555 	    (u_longlong_t)rbp_buf[0].rbp_object,
3556 	    (u_longlong_t)rbp_buf[0].rbp_blkid,
3557 	    (uint_t)(redact_block_get_size(&rbp_buf[0])),
3558 	    (u_longlong_t)redact_block_get_count(&rbp_buf[0]));
3559 
3560 	for (size_t i = 1; i < rlp->rlp_num_entries; i++) {
3561 		(void) printf(",\n\t\t{object: %llx, offset: %llx, "
3562 		    "blksz: %x, count: %llx}",
3563 		    (u_longlong_t)rbp_buf[i].rbp_object,
3564 		    (u_longlong_t)rbp_buf[i].rbp_blkid,
3565 		    (uint_t)(redact_block_get_size(&rbp_buf[i])),
3566 		    (u_longlong_t)redact_block_get_count(&rbp_buf[i]));
3567 	}
3568 	dsl_redaction_list_rele(rl, FTAG);
3569 	kmem_free(rbp_buf, size);
3570 	(void) printf("]\n\n");
3571 	return (0);
3572 }
3573 
3574 static void
dump_bookmarks(objset_t * os,int verbosity)3575 dump_bookmarks(objset_t *os, int verbosity)
3576 {
3577 	zap_cursor_t zc;
3578 	zap_attribute_t *attrp;
3579 	dsl_dataset_t *ds = dmu_objset_ds(os);
3580 	dsl_pool_t *dp = spa_get_dsl(os->os_spa);
3581 	objset_t *mos = os->os_spa->spa_meta_objset;
3582 	if (verbosity < 4)
3583 		return;
3584 	attrp = zap_attribute_alloc();
3585 	dsl_pool_config_enter(dp, FTAG);
3586 
3587 	for (zap_cursor_init(&zc, mos, ds->ds_bookmarks_obj);
3588 	    zap_cursor_retrieve(&zc, attrp) == 0;
3589 	    zap_cursor_advance(&zc)) {
3590 		char osname[ZFS_MAX_DATASET_NAME_LEN];
3591 		char buf[ZFS_MAX_DATASET_NAME_LEN];
3592 		int len;
3593 		dmu_objset_name(os, osname);
3594 		len = snprintf(buf, sizeof (buf), "%s#%s", osname,
3595 		    attrp->za_name);
3596 		VERIFY3S(len, <, ZFS_MAX_DATASET_NAME_LEN);
3597 		(void) dump_bookmark(dp, buf, verbosity >= 5, verbosity >= 6);
3598 	}
3599 	zap_cursor_fini(&zc);
3600 	dsl_pool_config_exit(dp, FTAG);
3601 	zap_attribute_free(attrp);
3602 }
3603 
3604 static void
bpobj_count_refd(bpobj_t * bpo)3605 bpobj_count_refd(bpobj_t *bpo)
3606 {
3607 	mos_obj_refd(bpo->bpo_object);
3608 
3609 	if (bpo->bpo_havesubobj && bpo->bpo_phys->bpo_subobjs != 0) {
3610 		mos_obj_refd(bpo->bpo_phys->bpo_subobjs);
3611 		for (uint64_t i = 0; i < bpo->bpo_phys->bpo_num_subobjs; i++) {
3612 			uint64_t subobj;
3613 			bpobj_t subbpo;
3614 			int error;
3615 			VERIFY0(dmu_read(bpo->bpo_os,
3616 			    bpo->bpo_phys->bpo_subobjs,
3617 			    i * sizeof (subobj), sizeof (subobj), &subobj, 0));
3618 			error = bpobj_open(&subbpo, bpo->bpo_os, subobj);
3619 			if (error != 0) {
3620 				(void) printf("ERROR %u while trying to open "
3621 				    "subobj id %llu\n",
3622 				    error, (u_longlong_t)subobj);
3623 				corruption_found = B_TRUE;
3624 				continue;
3625 			}
3626 			bpobj_count_refd(&subbpo);
3627 			bpobj_close(&subbpo);
3628 		}
3629 	}
3630 }
3631 
3632 static int
dsl_deadlist_entry_count_refd(void * arg,dsl_deadlist_entry_t * dle)3633 dsl_deadlist_entry_count_refd(void *arg, dsl_deadlist_entry_t *dle)
3634 {
3635 	spa_t *spa = arg;
3636 	uint64_t empty_bpobj = spa->spa_dsl_pool->dp_empty_bpobj;
3637 	if (dle->dle_bpobj.bpo_object != empty_bpobj)
3638 		bpobj_count_refd(&dle->dle_bpobj);
3639 	return (0);
3640 }
3641 
3642 static int
dsl_deadlist_entry_dump(void * arg,dsl_deadlist_entry_t * dle)3643 dsl_deadlist_entry_dump(void *arg, dsl_deadlist_entry_t *dle)
3644 {
3645 	ASSERT0P(arg);
3646 	if (dump_opt['d'] >= 5) {
3647 		char buf[128];
3648 		(void) snprintf(buf, sizeof (buf),
3649 		    "mintxg %llu -> obj %llu",
3650 		    (longlong_t)dle->dle_mintxg,
3651 		    (longlong_t)dle->dle_bpobj.bpo_object);
3652 
3653 		dump_full_bpobj(&dle->dle_bpobj, buf, 0);
3654 	} else {
3655 		(void) printf("mintxg %llu -> obj %llu\n",
3656 		    (longlong_t)dle->dle_mintxg,
3657 		    (longlong_t)dle->dle_bpobj.bpo_object);
3658 	}
3659 	return (0);
3660 }
3661 
3662 static void
dump_blkptr_list(dsl_deadlist_t * dl,const char * name)3663 dump_blkptr_list(dsl_deadlist_t *dl, const char *name)
3664 {
3665 	char bytes[32];
3666 	char comp[32];
3667 	char uncomp[32];
3668 	char entries[32];
3669 	spa_t *spa = dmu_objset_spa(dl->dl_os);
3670 	uint64_t empty_bpobj = spa->spa_dsl_pool->dp_empty_bpobj;
3671 
3672 	if (dl->dl_oldfmt) {
3673 		if (dl->dl_bpobj.bpo_object != empty_bpobj)
3674 			bpobj_count_refd(&dl->dl_bpobj);
3675 	} else {
3676 		mos_obj_refd(dl->dl_object);
3677 		dsl_deadlist_iterate(dl, dsl_deadlist_entry_count_refd, spa);
3678 	}
3679 
3680 	/* make sure nicenum has enough space */
3681 	_Static_assert(sizeof (bytes) >= NN_NUMBUF_SZ, "bytes truncated");
3682 	_Static_assert(sizeof (comp) >= NN_NUMBUF_SZ, "comp truncated");
3683 	_Static_assert(sizeof (uncomp) >= NN_NUMBUF_SZ, "uncomp truncated");
3684 	_Static_assert(sizeof (entries) >= NN_NUMBUF_SZ, "entries truncated");
3685 
3686 	if (dump_opt['d'] < 3)
3687 		return;
3688 
3689 	if (dl->dl_oldfmt) {
3690 		dump_full_bpobj(&dl->dl_bpobj, "old-format deadlist", 0);
3691 		return;
3692 	}
3693 
3694 	zdb_nicenum(dl->dl_phys->dl_used, bytes, sizeof (bytes));
3695 	zdb_nicenum(dl->dl_phys->dl_comp, comp, sizeof (comp));
3696 	zdb_nicenum(dl->dl_phys->dl_uncomp, uncomp, sizeof (uncomp));
3697 	zdb_nicenum(avl_numnodes(&dl->dl_tree), entries, sizeof (entries));
3698 	(void) printf("\n    %s: %s (%s/%s comp), %s entries\n",
3699 	    name, bytes, comp, uncomp, entries);
3700 
3701 	if (dump_opt['d'] < 4)
3702 		return;
3703 
3704 	(void) putchar('\n');
3705 
3706 	dsl_deadlist_iterate(dl, dsl_deadlist_entry_dump, NULL);
3707 }
3708 
3709 static int
verify_dd_livelist(objset_t * os)3710 verify_dd_livelist(objset_t *os)
3711 {
3712 	uint64_t ll_used, used, ll_comp, comp, ll_uncomp, uncomp;
3713 	dsl_pool_t *dp = spa_get_dsl(os->os_spa);
3714 	dsl_dir_t  *dd = os->os_dsl_dataset->ds_dir;
3715 
3716 	ASSERT(!dmu_objset_is_snapshot(os));
3717 	if (!dsl_deadlist_is_open(&dd->dd_livelist))
3718 		return (0);
3719 
3720 	/* Iterate through the livelist to check for duplicates */
3721 	dsl_deadlist_iterate(&dd->dd_livelist, sublivelist_verify_lightweight,
3722 	    NULL);
3723 
3724 	dsl_pool_config_enter(dp, FTAG);
3725 	dsl_deadlist_space(&dd->dd_livelist, &ll_used,
3726 	    &ll_comp, &ll_uncomp);
3727 
3728 	dsl_dataset_t *origin_ds;
3729 	ASSERT(dsl_pool_config_held(dp));
3730 	VERIFY0(dsl_dataset_hold_obj(dp,
3731 	    dsl_dir_phys(dd)->dd_origin_obj, FTAG, &origin_ds));
3732 	VERIFY0(dsl_dataset_space_written(origin_ds, os->os_dsl_dataset,
3733 	    &used, &comp, &uncomp));
3734 	dsl_dataset_rele(origin_ds, FTAG);
3735 	dsl_pool_config_exit(dp, FTAG);
3736 	/*
3737 	 *  It's possible that the dataset's uncomp space is larger than the
3738 	 *  livelist's because livelists do not track embedded block pointers
3739 	 */
3740 	if (used != ll_used || comp != ll_comp || uncomp < ll_uncomp) {
3741 		char nice_used[32], nice_comp[32], nice_uncomp[32];
3742 		(void) printf("Discrepancy in space accounting:\n");
3743 		zdb_nicenum(used, nice_used, sizeof (nice_used));
3744 		zdb_nicenum(comp, nice_comp, sizeof (nice_comp));
3745 		zdb_nicenum(uncomp, nice_uncomp, sizeof (nice_uncomp));
3746 		(void) printf("dir: used %s, comp %s, uncomp %s\n",
3747 		    nice_used, nice_comp, nice_uncomp);
3748 		zdb_nicenum(ll_used, nice_used, sizeof (nice_used));
3749 		zdb_nicenum(ll_comp, nice_comp, sizeof (nice_comp));
3750 		zdb_nicenum(ll_uncomp, nice_uncomp, sizeof (nice_uncomp));
3751 		(void) printf("livelist: used %s, comp %s, uncomp %s\n",
3752 		    nice_used, nice_comp, nice_uncomp);
3753 		return (1);
3754 	}
3755 	return (0);
3756 }
3757 
3758 static char *key_material = NULL;
3759 
3760 static boolean_t
zdb_derive_key(dsl_dir_t * dd,uint8_t * key_out)3761 zdb_derive_key(dsl_dir_t *dd, uint8_t *key_out)
3762 {
3763 	uint64_t keyformat, salt, iters;
3764 	int i;
3765 	unsigned char c;
3766 	FILE *f;
3767 
3768 	VERIFY0(zap_lookup(dd->dd_pool->dp_meta_objset, dd->dd_crypto_obj,
3769 	    zfs_prop_to_name(ZFS_PROP_KEYFORMAT), sizeof (uint64_t),
3770 	    1, &keyformat));
3771 
3772 	switch (keyformat) {
3773 	case ZFS_KEYFORMAT_HEX:
3774 		for (i = 0; i < WRAPPING_KEY_LEN * 2; i += 2) {
3775 			if (!isxdigit(key_material[i]) ||
3776 			    !isxdigit(key_material[i+1]))
3777 				return (B_FALSE);
3778 			if (sscanf(&key_material[i], "%02hhx", &c) != 1)
3779 				return (B_FALSE);
3780 			key_out[i / 2] = c;
3781 		}
3782 		break;
3783 
3784 	case ZFS_KEYFORMAT_PASSPHRASE:
3785 		VERIFY0(zap_lookup(dd->dd_pool->dp_meta_objset,
3786 		    dd->dd_crypto_obj, zfs_prop_to_name(ZFS_PROP_PBKDF2_SALT),
3787 		    sizeof (uint64_t), 1, &salt));
3788 		VERIFY0(zap_lookup(dd->dd_pool->dp_meta_objset,
3789 		    dd->dd_crypto_obj, zfs_prop_to_name(ZFS_PROP_PBKDF2_ITERS),
3790 		    sizeof (uint64_t), 1, &iters));
3791 
3792 		if (PKCS5_PBKDF2_HMAC_SHA1(key_material, strlen(key_material),
3793 		    ((uint8_t *)&salt), sizeof (uint64_t), iters,
3794 		    WRAPPING_KEY_LEN, key_out) != 1)
3795 			return (B_FALSE);
3796 
3797 		break;
3798 
3799 	case ZFS_KEYFORMAT_RAW:
3800 		if ((f = fopen(key_material, "r")) == NULL)
3801 			return (B_FALSE);
3802 
3803 		if (fread(key_out, 1, WRAPPING_KEY_LEN, f) !=
3804 		    WRAPPING_KEY_LEN) {
3805 			(void) fclose(f);
3806 			return (B_FALSE);
3807 		}
3808 
3809 		/* Check the key length */
3810 		if (fgetc(f) != EOF) {
3811 			(void) fclose(f);
3812 			return (B_FALSE);
3813 		}
3814 
3815 		(void) fclose(f);
3816 		break;
3817 
3818 	default:
3819 		fatal("no support for key format %u\n",
3820 		    (unsigned int) keyformat);
3821 	}
3822 
3823 	return (B_TRUE);
3824 }
3825 
3826 static char encroot[ZFS_MAX_DATASET_NAME_LEN];
3827 static boolean_t key_loaded = B_FALSE;
3828 
3829 static int
zdb_load_key(objset_t * os)3830 zdb_load_key(objset_t *os)
3831 {
3832 	dsl_pool_t *dp;
3833 	dsl_dir_t *dd, *rdd;
3834 	uint8_t key[WRAPPING_KEY_LEN];
3835 	uint64_t rddobj;
3836 	int err = 0;
3837 
3838 	dp = spa_get_dsl(os->os_spa);
3839 	dd = os->os_dsl_dataset->ds_dir;
3840 
3841 	dsl_pool_config_enter(dp, FTAG);
3842 	VERIFY0(zap_lookup(dd->dd_pool->dp_meta_objset, dd->dd_crypto_obj,
3843 	    DSL_CRYPTO_KEY_ROOT_DDOBJ, sizeof (uint64_t), 1, &rddobj));
3844 	VERIFY0(dsl_dir_hold_obj(dd->dd_pool, rddobj, NULL, FTAG, &rdd));
3845 	dsl_dir_name(rdd, encroot);
3846 	dsl_dir_rele(rdd, FTAG);
3847 
3848 	if (!zdb_derive_key(dd, key))
3849 		err = EINVAL;
3850 	dsl_pool_config_exit(dp, FTAG);
3851 
3852 	if (err != 0) {
3853 		fprintf(stderr, "couldn't derive encryption key\n");
3854 		return (err);
3855 	}
3856 
3857 	ASSERT3U(dsl_dataset_get_keystatus(dd), ==, ZFS_KEYSTATUS_UNAVAILABLE);
3858 
3859 	dsl_crypto_params_t *dcp;
3860 	nvlist_t *crypto_args;
3861 
3862 	crypto_args = fnvlist_alloc();
3863 	fnvlist_add_uint8_array(crypto_args, "wkeydata",
3864 	    (uint8_t *)key, WRAPPING_KEY_LEN);
3865 	VERIFY0(dsl_crypto_params_create_nvlist(DCP_CMD_NONE,
3866 	    NULL, crypto_args, &dcp));
3867 	err = spa_keystore_load_wkey(encroot, dcp, B_FALSE);
3868 
3869 	dsl_crypto_params_free(dcp, (err != 0));
3870 	fnvlist_free(crypto_args);
3871 
3872 	if (err != 0) {
3873 		fprintf(stderr,
3874 		    "couldn't load encryption key for %s: %s\n",
3875 		    encroot, err == ZFS_ERR_CRYPTO_NOTSUP ?
3876 		    "crypto params not supported" : strerror(err));
3877 		return (err);
3878 	}
3879 
3880 	ASSERT3U(dsl_dataset_get_keystatus(dd), ==, ZFS_KEYSTATUS_AVAILABLE);
3881 
3882 	printf("Unlocked encryption root: %s\n", encroot);
3883 	key_loaded = B_TRUE;
3884 
3885 	return (0);
3886 }
3887 
3888 static void
zdb_unload_key(void)3889 zdb_unload_key(void)
3890 {
3891 	if (!key_loaded)
3892 		return;
3893 
3894 	VERIFY0(spa_keystore_unload_wkey(encroot));
3895 	key_loaded = B_FALSE;
3896 }
3897 
3898 static avl_tree_t idx_tree;
3899 static avl_tree_t domain_tree;
3900 static boolean_t fuid_table_loaded;
3901 static objset_t *sa_os = NULL;
3902 static sa_attr_type_t *sa_attr_table = NULL;
3903 
3904 static int
open_objset(const char * path,const void * tag,objset_t ** osp)3905 open_objset(const char *path, const void *tag, objset_t **osp)
3906 {
3907 	int err;
3908 	uint64_t sa_attrs = 0;
3909 	uint64_t version = 0;
3910 
3911 	VERIFY0P(sa_os);
3912 
3913 	/*
3914 	 * We can't own an objset if it's redacted.  Therefore, we do this
3915 	 * dance: hold the objset, then acquire a long hold on its dataset, then
3916 	 * release the pool (which is held as part of holding the objset).
3917 	 */
3918 
3919 	if (dump_opt['K']) {
3920 		/* decryption requested, try to load keys */
3921 		err = dmu_objset_hold(path, tag, osp);
3922 		if (err != 0) {
3923 			(void) fprintf(stderr, "failed to hold dataset "
3924 			    "'%s': %s\n",
3925 			    path, strerror(err));
3926 			return (err);
3927 		}
3928 
3929 		/*
3930 		 * Only try to load the key and unlock the dataset if it is
3931 		 * actually encrypted; otherwise we'll just crash. Just
3932 		 * ignore the -K switch entirely otherwise; it's useful to be
3933 		 * able to provide even if it's not needed.
3934 		 */
3935 		if ((*osp)->os_encrypted) {
3936 			dsl_dataset_long_hold(dmu_objset_ds(*osp), tag);
3937 			dsl_pool_rele(dmu_objset_pool(*osp), tag);
3938 
3939 			err = zdb_load_key(*osp);
3940 
3941 			/* release it all */
3942 			dsl_dataset_long_rele(dmu_objset_ds(*osp), tag);
3943 			dsl_dataset_rele(dmu_objset_ds(*osp), tag);
3944 
3945 			if (err != 0) {
3946 				*osp = NULL;
3947 				return (err);
3948 			}
3949 		} else {
3950 			dmu_objset_rele(*osp, tag);
3951 		}
3952 	}
3953 
3954 	int ds_hold_flags = key_loaded ? DS_HOLD_FLAG_DECRYPT : 0;
3955 
3956 	err = dmu_objset_hold_flags(path, ds_hold_flags, tag, osp);
3957 	if (err != 0) {
3958 		(void) fprintf(stderr, "failed to hold dataset '%s': %s\n",
3959 		    path, strerror(err));
3960 		*osp = NULL;
3961 		return (err);
3962 	}
3963 	dsl_dataset_long_hold(dmu_objset_ds(*osp), tag);
3964 	dsl_pool_rele(dmu_objset_pool(*osp), tag);
3965 
3966 	if (dmu_objset_type(*osp) == DMU_OST_ZFS &&
3967 	    (key_loaded || !(*osp)->os_encrypted)) {
3968 		(void) zap_lookup(*osp, MASTER_NODE_OBJ, ZPL_VERSION_STR,
3969 		    8, 1, &version);
3970 		if (version >= ZPL_VERSION_SA) {
3971 			(void) zap_lookup(*osp, MASTER_NODE_OBJ, ZFS_SA_ATTRS,
3972 			    8, 1, &sa_attrs);
3973 		}
3974 		err = sa_setup(*osp, sa_attrs, zfs_attr_table, ZPL_END,
3975 		    &sa_attr_table);
3976 		if (err != 0) {
3977 			(void) fprintf(stderr, "sa_setup failed: %s\n",
3978 			    strerror(err));
3979 			dsl_dataset_long_rele(dmu_objset_ds(*osp), tag);
3980 			dsl_dataset_rele_flags(dmu_objset_ds(*osp),
3981 			    ds_hold_flags, tag);
3982 			*osp = NULL;
3983 		}
3984 	}
3985 	sa_os = *osp;
3986 
3987 	return (err);
3988 }
3989 
3990 static void
close_objset(objset_t * os,const void * tag)3991 close_objset(objset_t *os, const void *tag)
3992 {
3993 	VERIFY3P(os, ==, sa_os);
3994 	if (os->os_sa != NULL)
3995 		sa_tear_down(os);
3996 	dsl_dataset_long_rele(dmu_objset_ds(os), tag);
3997 	dsl_dataset_rele_flags(dmu_objset_ds(os),
3998 	    key_loaded ? DS_HOLD_FLAG_DECRYPT : 0, tag);
3999 	sa_attr_table = NULL;
4000 	sa_os = NULL;
4001 
4002 	zdb_unload_key();
4003 }
4004 
4005 static void
fuid_table_destroy(void)4006 fuid_table_destroy(void)
4007 {
4008 	if (fuid_table_loaded) {
4009 		zfs_fuid_table_destroy(&idx_tree, &domain_tree);
4010 		fuid_table_loaded = B_FALSE;
4011 	}
4012 }
4013 
4014 /*
4015  * Clean up DDT internal state. ddt_lookup() adds entries to ddt_tree, which on
4016  * a live pool are normally cleaned up during ddt_sync(). We can't do that (and
4017  * wouldn't want to anyway), but if we don't clean up the presence of stuff on
4018  * ddt_tree will trip asserts in ddt_table_free(). So, we clean up ourselves.
4019  *
4020  * Note that this is not a particularly efficient way to do this, but
4021  * ddt_remove() is the only public method that can do the work we need, and it
4022  * requires the right locks and etc to do the job. This is only ever called
4023  * during zdb shutdown so efficiency is not especially important.
4024  */
4025 static void
zdb_ddt_cleanup(spa_t * spa)4026 zdb_ddt_cleanup(spa_t *spa)
4027 {
4028 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
4029 		ddt_t *ddt = spa->spa_ddt[c];
4030 		if (!ddt)
4031 			continue;
4032 
4033 		spa_config_enter(spa, SCL_CONFIG, FTAG, RW_READER);
4034 		ddt_enter(ddt);
4035 		ddt_entry_t *dde = avl_first(&ddt->ddt_tree), *next;
4036 		while (dde) {
4037 			next = AVL_NEXT(&ddt->ddt_tree, dde);
4038 			dde->dde_io = NULL;
4039 			ddt_remove(ddt, dde);
4040 			dde = next;
4041 		}
4042 		ddt_exit(ddt);
4043 		spa_config_exit(spa, SCL_CONFIG, FTAG);
4044 	}
4045 }
4046 
4047 static void
zdb_exit(int reason)4048 zdb_exit(int reason)
4049 {
4050 	if (spa != NULL)
4051 		zdb_ddt_cleanup(spa);
4052 
4053 	if (os != NULL) {
4054 		close_objset(os, FTAG);
4055 	} else if (spa != NULL) {
4056 		spa_close(spa, FTAG);
4057 	}
4058 
4059 	fuid_table_destroy();
4060 
4061 	if (kernel_init_done)
4062 		kernel_fini();
4063 
4064 	exit(reason);
4065 }
4066 
4067 /*
4068  * print uid or gid information.
4069  * For normal POSIX id just the id is printed in decimal format.
4070  * For CIFS files with FUID the fuid is printed in hex followed by
4071  * the domain-rid string.
4072  */
4073 static void
print_idstr(uint64_t id,const char * id_type)4074 print_idstr(uint64_t id, const char *id_type)
4075 {
4076 	if (FUID_INDEX(id)) {
4077 		const char *domain =
4078 		    zfs_fuid_idx_domain(&idx_tree, FUID_INDEX(id));
4079 		(void) printf("\t%s     %llx [%s-%d]\n", id_type,
4080 		    (u_longlong_t)id, domain, (int)FUID_RID(id));
4081 	} else {
4082 		(void) printf("\t%s     %llu\n", id_type, (u_longlong_t)id);
4083 	}
4084 
4085 }
4086 
4087 static void
dump_uidgid(objset_t * os,uint64_t uid,uint64_t gid)4088 dump_uidgid(objset_t *os, uint64_t uid, uint64_t gid)
4089 {
4090 	uint32_t uid_idx, gid_idx;
4091 
4092 	uid_idx = FUID_INDEX(uid);
4093 	gid_idx = FUID_INDEX(gid);
4094 
4095 	/* Load domain table, if not already loaded */
4096 	if (!fuid_table_loaded && (uid_idx || gid_idx)) {
4097 		uint64_t fuid_obj;
4098 
4099 		/* first find the fuid object.  It lives in the master node */
4100 		VERIFY0(zap_lookup(os, MASTER_NODE_OBJ, ZFS_FUID_TABLES,
4101 		    8, 1, &fuid_obj));
4102 		zfs_fuid_avl_tree_create(&idx_tree, &domain_tree);
4103 		(void) zfs_fuid_table_load(os, fuid_obj,
4104 		    &idx_tree, &domain_tree);
4105 		fuid_table_loaded = B_TRUE;
4106 	}
4107 
4108 	print_idstr(uid, "uid");
4109 	print_idstr(gid, "gid");
4110 }
4111 
4112 static void
dump_znode_sa_xattr(sa_handle_t * hdl)4113 dump_znode_sa_xattr(sa_handle_t *hdl)
4114 {
4115 	nvlist_t *sa_xattr;
4116 	nvpair_t *elem = NULL;
4117 	int sa_xattr_size = 0;
4118 	int sa_xattr_entries = 0;
4119 	int error;
4120 	char *sa_xattr_packed;
4121 
4122 	error = sa_size(hdl, sa_attr_table[ZPL_DXATTR], &sa_xattr_size);
4123 	if (error || sa_xattr_size == 0)
4124 		return;
4125 
4126 	sa_xattr_packed = malloc(sa_xattr_size);
4127 	if (sa_xattr_packed == NULL)
4128 		return;
4129 
4130 	error = sa_lookup(hdl, sa_attr_table[ZPL_DXATTR],
4131 	    sa_xattr_packed, sa_xattr_size);
4132 	if (error) {
4133 		free(sa_xattr_packed);
4134 		return;
4135 	}
4136 
4137 	error = nvlist_unpack(sa_xattr_packed, sa_xattr_size, &sa_xattr, 0);
4138 	if (error) {
4139 		free(sa_xattr_packed);
4140 		return;
4141 	}
4142 
4143 	while ((elem = nvlist_next_nvpair(sa_xattr, elem)) != NULL)
4144 		sa_xattr_entries++;
4145 
4146 	(void) printf("\tSA xattrs: %d bytes, %d entries\n\n",
4147 	    sa_xattr_size, sa_xattr_entries);
4148 	while ((elem = nvlist_next_nvpair(sa_xattr, elem)) != NULL) {
4149 		boolean_t can_print = !dump_opt['P'];
4150 		uchar_t *value;
4151 		uint_t cnt, idx;
4152 
4153 		(void) printf("\t\t%s = ", nvpair_name(elem));
4154 		nvpair_value_byte_array(elem, &value, &cnt);
4155 
4156 		for (idx = 0; idx < cnt; ++idx) {
4157 			if (!isprint(value[idx])) {
4158 				can_print = B_FALSE;
4159 				break;
4160 			}
4161 		}
4162 
4163 		for (idx = 0; idx < cnt; ++idx) {
4164 			if (can_print)
4165 				(void) putchar(value[idx]);
4166 			else
4167 				(void) printf("\\%3.3o", value[idx]);
4168 		}
4169 		(void) putchar('\n');
4170 	}
4171 
4172 	nvlist_free(sa_xattr);
4173 	free(sa_xattr_packed);
4174 }
4175 
4176 static void
dump_znode_symlink(sa_handle_t * hdl)4177 dump_znode_symlink(sa_handle_t *hdl)
4178 {
4179 	int sa_symlink_size = 0;
4180 	char linktarget[MAXPATHLEN];
4181 	int error;
4182 
4183 	error = sa_size(hdl, sa_attr_table[ZPL_SYMLINK], &sa_symlink_size);
4184 	if (error || sa_symlink_size == 0) {
4185 		return;
4186 	}
4187 	if (sa_symlink_size >= sizeof (linktarget)) {
4188 		(void) printf("symlink size %d is too large\n",
4189 		    sa_symlink_size);
4190 		return;
4191 	}
4192 	linktarget[sa_symlink_size] = '\0';
4193 	if (sa_lookup(hdl, sa_attr_table[ZPL_SYMLINK],
4194 	    &linktarget, sa_symlink_size) == 0)
4195 		(void) printf("\ttarget	%s\n", linktarget);
4196 }
4197 
4198 static void
dump_znode(objset_t * os,uint64_t object,void * data,size_t size)4199 dump_znode(objset_t *os, uint64_t object, void *data, size_t size)
4200 {
4201 	(void) data, (void) size;
4202 	char path[MAXPATHLEN * 2];	/* allow for xattr and failure prefix */
4203 	sa_handle_t *hdl;
4204 	uint64_t xattr, rdev, gen;
4205 	uint64_t uid, gid, mode, fsize, parent, links;
4206 	uint64_t pflags;
4207 	uint64_t acctm[2], modtm[2], chgtm[2], crtm[2];
4208 	time_t z_crtime, z_atime, z_mtime, z_ctime;
4209 	sa_bulk_attr_t bulk[12];
4210 	int idx = 0;
4211 	int error;
4212 
4213 	VERIFY3P(os, ==, sa_os);
4214 	if (sa_handle_get(os, object, NULL, SA_HDL_PRIVATE, &hdl)) {
4215 		(void) printf("Failed to get handle for SA znode\n");
4216 		return;
4217 	}
4218 
4219 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_UID], NULL, &uid, 8);
4220 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_GID], NULL, &gid, 8);
4221 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_LINKS], NULL,
4222 	    &links, 8);
4223 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_GEN], NULL, &gen, 8);
4224 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_MODE], NULL,
4225 	    &mode, 8);
4226 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_PARENT],
4227 	    NULL, &parent, 8);
4228 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_SIZE], NULL,
4229 	    &fsize, 8);
4230 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_ATIME], NULL,
4231 	    acctm, 16);
4232 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_MTIME], NULL,
4233 	    modtm, 16);
4234 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_CRTIME], NULL,
4235 	    crtm, 16);
4236 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_CTIME], NULL,
4237 	    chgtm, 16);
4238 	SA_ADD_BULK_ATTR(bulk, idx, sa_attr_table[ZPL_FLAGS], NULL,
4239 	    &pflags, 8);
4240 
4241 	if (sa_bulk_lookup(hdl, bulk, idx)) {
4242 		(void) sa_handle_destroy(hdl);
4243 		return;
4244 	}
4245 
4246 	z_crtime = (time_t)crtm[0];
4247 	z_atime = (time_t)acctm[0];
4248 	z_mtime = (time_t)modtm[0];
4249 	z_ctime = (time_t)chgtm[0];
4250 
4251 	if (dump_opt['d'] > 4) {
4252 		error = zfs_obj_to_path(os, object, path, sizeof (path));
4253 		if (error == ESTALE) {
4254 			(void) snprintf(path, sizeof (path), "on delete queue");
4255 		} else if (error != 0) {
4256 			leaked_objects++;
4257 			(void) snprintf(path, sizeof (path),
4258 			    "path not found, possibly leaked");
4259 		}
4260 		(void) printf("\tpath	%s\n", path);
4261 	}
4262 
4263 	if (S_ISLNK(mode))
4264 		dump_znode_symlink(hdl);
4265 	dump_uidgid(os, uid, gid);
4266 	(void) printf("\tatime	%s", ctime(&z_atime));
4267 	(void) printf("\tmtime	%s", ctime(&z_mtime));
4268 	(void) printf("\tctime	%s", ctime(&z_ctime));
4269 	(void) printf("\tcrtime	%s", ctime(&z_crtime));
4270 	(void) printf("\tgen	%llu\n", (u_longlong_t)gen);
4271 	(void) printf("\tmode	%llo\n", (u_longlong_t)mode);
4272 	(void) printf("\tsize	%llu\n", (u_longlong_t)fsize);
4273 	(void) printf("\tparent	%llu\n", (u_longlong_t)parent);
4274 	(void) printf("\tlinks	%llu\n", (u_longlong_t)links);
4275 	(void) printf("\tpflags	%llx\n", (u_longlong_t)pflags);
4276 	if (dmu_objset_projectquota_enabled(os) && (pflags & ZFS_PROJID)) {
4277 		uint64_t projid;
4278 
4279 		if (sa_lookup(hdl, sa_attr_table[ZPL_PROJID], &projid,
4280 		    sizeof (uint64_t)) == 0)
4281 			(void) printf("\tprojid	%llu\n", (u_longlong_t)projid);
4282 	}
4283 	if (sa_lookup(hdl, sa_attr_table[ZPL_XATTR], &xattr,
4284 	    sizeof (uint64_t)) == 0)
4285 		(void) printf("\txattr	%llu\n", (u_longlong_t)xattr);
4286 	if (sa_lookup(hdl, sa_attr_table[ZPL_RDEV], &rdev,
4287 	    sizeof (uint64_t)) == 0)
4288 		(void) printf("\trdev	0x%016llx\n", (u_longlong_t)rdev);
4289 	dump_znode_sa_xattr(hdl);
4290 	sa_handle_destroy(hdl);
4291 }
4292 
4293 static void
dump_acl(objset_t * os,uint64_t object,void * data,size_t size)4294 dump_acl(objset_t *os, uint64_t object, void *data, size_t size)
4295 {
4296 	(void) os, (void) object, (void) data, (void) size;
4297 }
4298 
4299 static void
dump_dmu_objset(objset_t * os,uint64_t object,void * data,size_t size)4300 dump_dmu_objset(objset_t *os, uint64_t object, void *data, size_t size)
4301 {
4302 	(void) os, (void) object, (void) data, (void) size;
4303 }
4304 
4305 static object_viewer_t *object_viewer[DMU_OT_NUMTYPES + 1] = {
4306 	dump_none,		/* unallocated			*/
4307 	dump_zap,		/* object directory		*/
4308 	dump_uint64,		/* object array			*/
4309 	dump_none,		/* packed nvlist		*/
4310 	dump_packed_nvlist,	/* packed nvlist size		*/
4311 	dump_none,		/* bpobj			*/
4312 	dump_bpobj,		/* bpobj header			*/
4313 	dump_none,		/* SPA space map header		*/
4314 	dump_none,		/* SPA space map		*/
4315 	dump_none,		/* ZIL intent log		*/
4316 	dump_dnode,		/* DMU dnode			*/
4317 	dump_dmu_objset,	/* DMU objset			*/
4318 	dump_dsl_dir,		/* DSL directory		*/
4319 	dump_zap,		/* DSL directory child map	*/
4320 	dump_zap,		/* DSL dataset snap map		*/
4321 	dump_zap,		/* DSL props			*/
4322 	dump_dsl_dataset,	/* DSL dataset			*/
4323 	dump_znode,		/* ZFS znode			*/
4324 	dump_acl,		/* ZFS V0 ACL			*/
4325 	dump_uint8,		/* ZFS plain file		*/
4326 	dump_zpldir,		/* ZFS directory		*/
4327 	dump_zap,		/* ZFS master node		*/
4328 	dump_zap,		/* ZFS delete queue		*/
4329 	dump_uint8,		/* zvol object			*/
4330 	dump_zap,		/* zvol prop			*/
4331 	dump_uint8,		/* other uint8[]		*/
4332 	dump_uint64,		/* other uint64[]		*/
4333 	dump_zap,		/* other ZAP			*/
4334 	dump_zap,		/* persistent error log		*/
4335 	dump_uint8,		/* SPA history			*/
4336 	dump_history_offsets,	/* SPA history offsets		*/
4337 	dump_zap,		/* Pool properties		*/
4338 	dump_zap,		/* DSL permissions		*/
4339 	dump_acl,		/* ZFS ACL			*/
4340 	dump_uint8,		/* ZFS SYSACL			*/
4341 	dump_none,		/* FUID nvlist			*/
4342 	dump_packed_nvlist,	/* FUID nvlist size		*/
4343 	dump_zap,		/* DSL dataset next clones	*/
4344 	dump_zap,		/* DSL scrub queue		*/
4345 	dump_zap,		/* ZFS user/group/project used	*/
4346 	dump_zap,		/* ZFS user/group/project quota	*/
4347 	dump_zap,		/* snapshot refcount tags	*/
4348 	dump_ddt_zap,		/* DDT ZAP object		*/
4349 	dump_zap,		/* DDT statistics		*/
4350 	dump_znode,		/* SA object			*/
4351 	dump_zap,		/* SA Master Node		*/
4352 	dump_sa_attrs,		/* SA attribute registration	*/
4353 	dump_sa_layouts,	/* SA attribute layouts		*/
4354 	dump_zap,		/* DSL scrub translations	*/
4355 	dump_none,		/* fake dedup BP		*/
4356 	dump_zap,		/* deadlist			*/
4357 	dump_none,		/* deadlist hdr			*/
4358 	dump_zap,		/* dsl clones			*/
4359 	dump_bpobj_subobjs,	/* bpobj subobjs		*/
4360 	dump_unknown,		/* Unknown type, must be last	*/
4361 };
4362 
4363 static boolean_t
match_object_type(dmu_object_type_t obj_type,uint64_t flags)4364 match_object_type(dmu_object_type_t obj_type, uint64_t flags)
4365 {
4366 	boolean_t match = B_TRUE;
4367 
4368 	switch (obj_type) {
4369 	case DMU_OT_DIRECTORY_CONTENTS:
4370 		if (!(flags & ZOR_FLAG_DIRECTORY))
4371 			match = B_FALSE;
4372 		break;
4373 	case DMU_OT_PLAIN_FILE_CONTENTS:
4374 		if (!(flags & ZOR_FLAG_PLAIN_FILE))
4375 			match = B_FALSE;
4376 		break;
4377 	case DMU_OT_SPACE_MAP:
4378 		if (!(flags & ZOR_FLAG_SPACE_MAP))
4379 			match = B_FALSE;
4380 		break;
4381 	case DMU_OT_ZVOL:
4382 		if (!(flags & ZOR_FLAG_ZVOL))
4383 			match = B_FALSE;
4384 		break;
4385 	default:
4386 		if (strcmp(zdb_ot_name(obj_type), "zap") == 0) {
4387 			if (!(flags & ZOR_FLAG_ZAP))
4388 				match = B_FALSE;
4389 			break;
4390 		}
4391 
4392 		/*
4393 		 * If all bits except some of the supported flags are
4394 		 * set, the user combined the all-types flag (A) with
4395 		 * a negated flag to exclude some types (e.g. A-f to
4396 		 * show all object types except plain files).
4397 		 */
4398 		if ((flags | ZOR_SUPPORTED_FLAGS) != ZOR_FLAG_ALL_TYPES)
4399 			match = B_FALSE;
4400 
4401 		break;
4402 	}
4403 
4404 	return (match);
4405 }
4406 
4407 static void
dump_object(objset_t * os,uint64_t object,int verbosity,boolean_t * print_header,uint64_t * dnode_slots_used,uint64_t flags)4408 dump_object(objset_t *os, uint64_t object, int verbosity,
4409     boolean_t *print_header, uint64_t *dnode_slots_used, uint64_t flags)
4410 {
4411 	dmu_buf_t *db = NULL;
4412 	dmu_object_info_t doi;
4413 	dnode_t *dn;
4414 	boolean_t dnode_held = B_FALSE;
4415 	void *bonus = NULL;
4416 	size_t bsize = 0;
4417 	char iblk[32], dblk[32], lsize[32], asize[32], fill[32], dnsize[32];
4418 	char bonus_size[32];
4419 	char aux[50];
4420 	int error;
4421 
4422 	/* make sure nicenum has enough space */
4423 	_Static_assert(sizeof (iblk) >= NN_NUMBUF_SZ, "iblk truncated");
4424 	_Static_assert(sizeof (dblk) >= NN_NUMBUF_SZ, "dblk truncated");
4425 	_Static_assert(sizeof (lsize) >= NN_NUMBUF_SZ, "lsize truncated");
4426 	_Static_assert(sizeof (asize) >= NN_NUMBUF_SZ, "asize truncated");
4427 	_Static_assert(sizeof (bonus_size) >= NN_NUMBUF_SZ,
4428 	    "bonus_size truncated");
4429 
4430 	if (*print_header) {
4431 		(void) printf("\n%10s  %3s  %5s  %5s  %5s  %6s  %5s  %6s  %s\n",
4432 		    "Object", "lvl", "iblk", "dblk", "dsize", "dnsize",
4433 		    "lsize", "%full", "type");
4434 		*print_header = 0;
4435 	}
4436 
4437 	if (object == 0) {
4438 		dn = DMU_META_DNODE(os);
4439 		dmu_object_info_from_dnode(dn, &doi);
4440 	} else {
4441 		/*
4442 		 * Encrypted datasets will have sensitive bonus buffers
4443 		 * encrypted. Therefore we cannot hold the bonus buffer and
4444 		 * must hold the dnode itself instead.
4445 		 */
4446 		error = dmu_object_info(os, object, &doi);
4447 		if (error)
4448 			fatal("dmu_object_info() failed, errno %u", error);
4449 
4450 		if (!key_loaded && os->os_encrypted &&
4451 		    DMU_OT_IS_ENCRYPTED(doi.doi_bonus_type)) {
4452 			error = dnode_hold(os, object, FTAG, &dn);
4453 			if (error)
4454 				fatal("dnode_hold() failed, errno %u", error);
4455 			dnode_held = B_TRUE;
4456 		} else {
4457 			error = dmu_bonus_hold(os, object, FTAG, &db);
4458 			if (error)
4459 				fatal("dmu_bonus_hold(%llu) failed, errno %u",
4460 				    object, error);
4461 			bonus = db->db_data;
4462 			bsize = db->db_size;
4463 			dn = DB_DNODE((dmu_buf_impl_t *)db);
4464 		}
4465 	}
4466 
4467 	/*
4468 	 * Default to showing all object types if no flags were specified.
4469 	 */
4470 	if (flags != 0 && flags != ZOR_FLAG_ALL_TYPES &&
4471 	    !match_object_type(doi.doi_type, flags))
4472 		goto out;
4473 
4474 	if (dnode_slots_used)
4475 		*dnode_slots_used = doi.doi_dnodesize / DNODE_MIN_SIZE;
4476 
4477 	zdb_nicenum(doi.doi_metadata_block_size, iblk, sizeof (iblk));
4478 	zdb_nicenum(doi.doi_data_block_size, dblk, sizeof (dblk));
4479 	zdb_nicenum(doi.doi_max_offset, lsize, sizeof (lsize));
4480 	zdb_nicenum(doi.doi_physical_blocks_512 << 9, asize, sizeof (asize));
4481 	zdb_nicenum(doi.doi_bonus_size, bonus_size, sizeof (bonus_size));
4482 	zdb_nicenum(doi.doi_dnodesize, dnsize, sizeof (dnsize));
4483 	(void) snprintf(fill, sizeof (fill), "%6.2f", 100.0 *
4484 	    doi.doi_fill_count * doi.doi_data_block_size / (object == 0 ?
4485 	    DNODES_PER_BLOCK : 1) / doi.doi_max_offset);
4486 
4487 	aux[0] = '\0';
4488 
4489 	if (doi.doi_checksum != ZIO_CHECKSUM_INHERIT || verbosity >= 6) {
4490 		(void) snprintf(aux + strlen(aux), sizeof (aux) - strlen(aux),
4491 		    " (K=%s)", ZDB_CHECKSUM_NAME(doi.doi_checksum));
4492 	}
4493 
4494 	if (doi.doi_compress == ZIO_COMPRESS_INHERIT &&
4495 	    ZIO_COMPRESS_HASLEVEL(os->os_compress) && verbosity >= 6) {
4496 		const char *compname = NULL;
4497 		if (zfs_prop_index_to_string(ZFS_PROP_COMPRESSION,
4498 		    ZIO_COMPRESS_RAW(os->os_compress, os->os_complevel),
4499 		    &compname) == 0) {
4500 			(void) snprintf(aux + strlen(aux),
4501 			    sizeof (aux) - strlen(aux), " (Z=inherit=%s)",
4502 			    compname);
4503 		} else {
4504 			(void) snprintf(aux + strlen(aux),
4505 			    sizeof (aux) - strlen(aux),
4506 			    " (Z=inherit=%s-unknown)",
4507 			    ZDB_COMPRESS_NAME(os->os_compress));
4508 		}
4509 	} else if (doi.doi_compress == ZIO_COMPRESS_INHERIT && verbosity >= 6) {
4510 		(void) snprintf(aux + strlen(aux), sizeof (aux) - strlen(aux),
4511 		    " (Z=inherit=%s)", ZDB_COMPRESS_NAME(os->os_compress));
4512 	} else if (doi.doi_compress != ZIO_COMPRESS_INHERIT || verbosity >= 6) {
4513 		(void) snprintf(aux + strlen(aux), sizeof (aux) - strlen(aux),
4514 		    " (Z=%s)", ZDB_COMPRESS_NAME(doi.doi_compress));
4515 	}
4516 
4517 	(void) printf("%10lld  %3u  %5s  %5s  %5s  %6s  %5s  %6s  %s%s\n",
4518 	    (u_longlong_t)object, doi.doi_indirection, iblk, dblk,
4519 	    asize, dnsize, lsize, fill, zdb_ot_name(doi.doi_type), aux);
4520 
4521 	if (doi.doi_bonus_type != DMU_OT_NONE && verbosity > 3) {
4522 		(void) printf("%10s  %3s  %5s  %5s  %5s  %5s  %5s  %6s  %s\n",
4523 		    "", "", "", "", "", "", bonus_size, "bonus",
4524 		    zdb_ot_name(doi.doi_bonus_type));
4525 	}
4526 
4527 	if (verbosity >= 4) {
4528 		(void) printf("\tdnode flags: %s%s%s%s\n",
4529 		    (dn->dn_phys->dn_flags & DNODE_FLAG_USED_BYTES) ?
4530 		    "USED_BYTES " : "",
4531 		    (dn->dn_phys->dn_flags & DNODE_FLAG_USERUSED_ACCOUNTED) ?
4532 		    "USERUSED_ACCOUNTED " : "",
4533 		    (dn->dn_phys->dn_flags & DNODE_FLAG_USEROBJUSED_ACCOUNTED) ?
4534 		    "USEROBJUSED_ACCOUNTED " : "",
4535 		    (dn->dn_phys->dn_flags & DNODE_FLAG_SPILL_BLKPTR) ?
4536 		    "SPILL_BLKPTR" : "");
4537 		(void) printf("\tdnode maxblkid: %llu\n",
4538 		    (longlong_t)dn->dn_phys->dn_maxblkid);
4539 
4540 		if (!dnode_held) {
4541 			object_viewer[ZDB_OT_TYPE(doi.doi_bonus_type)](os,
4542 			    object, bonus, bsize);
4543 		} else {
4544 			(void) printf("\t\t(bonus encrypted)\n");
4545 		}
4546 
4547 		if (key_loaded ||
4548 		    (!os->os_encrypted || !DMU_OT_IS_ENCRYPTED(doi.doi_type))) {
4549 			object_viewer[ZDB_OT_TYPE(doi.doi_type)](os, object,
4550 			    NULL, 0);
4551 		} else {
4552 			(void) printf("\t\t(object encrypted)\n");
4553 		}
4554 
4555 		*print_header = B_TRUE;
4556 	}
4557 
4558 	if (verbosity >= 5) {
4559 		if (dn->dn_phys->dn_flags & DNODE_FLAG_SPILL_BLKPTR) {
4560 			char blkbuf[BP_SPRINTF_LEN];
4561 			snprintf_blkptr_compact(blkbuf, sizeof (blkbuf),
4562 			    DN_SPILL_BLKPTR(dn->dn_phys), B_FALSE);
4563 			(void) printf("\nSpill block: %s\n", blkbuf);
4564 		}
4565 		dump_indirect(dn);
4566 	}
4567 
4568 	if (verbosity >= 5) {
4569 		/*
4570 		 * Report the list of segments that comprise the object.
4571 		 */
4572 		uint64_t start = 0;
4573 		uint64_t end;
4574 		uint64_t blkfill = 1;
4575 		int minlvl = 1;
4576 
4577 		if (dn->dn_type == DMU_OT_DNODE) {
4578 			minlvl = 0;
4579 			blkfill = DNODES_PER_BLOCK;
4580 		}
4581 
4582 		for (;;) {
4583 			char segsize[32];
4584 			/* make sure nicenum has enough space */
4585 			_Static_assert(sizeof (segsize) >= NN_NUMBUF_SZ,
4586 			    "segsize truncated");
4587 			error = dnode_next_offset(dn,
4588 			    0, &start, minlvl, blkfill, 0);
4589 			if (error)
4590 				break;
4591 			end = start;
4592 			error = dnode_next_offset(dn,
4593 			    DNODE_FIND_HOLE, &end, minlvl, blkfill, 0);
4594 			zdb_nicenum(end - start, segsize, sizeof (segsize));
4595 			(void) printf("\t\tsegment [%016llx, %016llx)"
4596 			    " size %5s\n", (u_longlong_t)start,
4597 			    (u_longlong_t)end, segsize);
4598 			if (error)
4599 				break;
4600 			start = end;
4601 		}
4602 	}
4603 out:
4604 	if (db != NULL)
4605 		dmu_buf_rele(db, FTAG);
4606 	if (dnode_held)
4607 		dnode_rele(dn, FTAG);
4608 }
4609 
4610 static void
dump_object_file_layout(objset_t * os,uint64_t object,int verbosity)4611 dump_object_file_layout(objset_t *os, uint64_t object, int verbosity)
4612 {
4613 	(void) verbosity;
4614 	dmu_buf_t *db = NULL;
4615 	dmu_object_info_t doi;
4616 	dnode_t *dn;
4617 	boolean_t dnode_held = B_FALSE;
4618 	int error;
4619 
4620 	char osname[ZFS_MAX_DATASET_NAME_LEN];
4621 	dmu_objset_name(os, osname);
4622 	(void) printf("%12s: '%s'\n", "objset", osname);
4623 	(void) printf("%12s: %d\n", "object", (int)object);
4624 
4625 	if (object == 0) {
4626 		dn = DMU_META_DNODE(os);
4627 		dmu_object_info_from_dnode(dn, &doi);
4628 	} else {
4629 		/*
4630 		 * Encrypted datasets will have sensitive bonus buffers
4631 		 * encrypted. Therefore we cannot hold the bonus buffer and
4632 		 * must hold the dnode itself instead.
4633 		 */
4634 		error = dmu_object_info(os, object, &doi);
4635 		if (error)
4636 			fatal("dmu_object_info() failed, errno %u", error);
4637 
4638 		if (!key_loaded && os->os_encrypted &&
4639 		    DMU_OT_IS_ENCRYPTED(doi.doi_bonus_type)) {
4640 			error = dnode_hold(os, object, FTAG, &dn);
4641 			if (error)
4642 				fatal("dnode_hold() failed, errno %u", error);
4643 			dnode_held = B_TRUE;
4644 		} else {
4645 			error = dmu_bonus_hold(os, object, FTAG, &db);
4646 			if (error)
4647 				fatal("dmu_bonus_hold(%llu) failed, errno %u",
4648 				    object, error);
4649 			dn = DB_DNODE((dmu_buf_impl_t *)db);
4650 		}
4651 	}
4652 
4653 	error = dump_indirect_layout(dn);
4654 
4655 	if (db != NULL)
4656 		dmu_buf_rele(db, FTAG);
4657 	if (dnode_held)
4658 		dnode_rele(dn, FTAG);
4659 
4660 	if (error)
4661 		zdb_exit(1);
4662 }
4663 
4664 static void
count_dir_mos_objects(dsl_dir_t * dd)4665 count_dir_mos_objects(dsl_dir_t *dd)
4666 {
4667 	mos_obj_refd(dd->dd_object);
4668 	mos_obj_refd(dsl_dir_phys(dd)->dd_child_dir_zapobj);
4669 	mos_obj_refd(dsl_dir_phys(dd)->dd_deleg_zapobj);
4670 	mos_obj_refd(dsl_dir_phys(dd)->dd_props_zapobj);
4671 	mos_obj_refd(dsl_dir_phys(dd)->dd_clones);
4672 
4673 	/*
4674 	 * The dd_crypto_obj can be referenced by multiple dsl_dir's.
4675 	 * Ignore the references after the first one.
4676 	 */
4677 	mos_obj_refd_multiple(dd->dd_crypto_obj);
4678 }
4679 
4680 static void
count_ds_mos_objects(dsl_dataset_t * ds)4681 count_ds_mos_objects(dsl_dataset_t *ds)
4682 {
4683 	mos_obj_refd(ds->ds_object);
4684 	mos_obj_refd(dsl_dataset_phys(ds)->ds_next_clones_obj);
4685 	mos_obj_refd(dsl_dataset_phys(ds)->ds_props_obj);
4686 	mos_obj_refd(dsl_dataset_phys(ds)->ds_userrefs_obj);
4687 	mos_obj_refd(dsl_dataset_phys(ds)->ds_snapnames_zapobj);
4688 	mos_obj_refd(ds->ds_bookmarks_obj);
4689 
4690 	if (!dsl_dataset_is_snapshot(ds)) {
4691 		count_dir_mos_objects(ds->ds_dir);
4692 	}
4693 }
4694 
4695 static const char *const objset_types[DMU_OST_NUMTYPES] = {
4696 	"NONE", "META", "ZPL", "ZVOL", "OTHER", "ANY" };
4697 
4698 /*
4699  * Parse a string denoting a range of object IDs of the form
4700  * <start>[:<end>[:flags]], and store the results in zor.
4701  * Return 0 on success. On error, return 1 and update the msg
4702  * pointer to point to a descriptive error message.
4703  */
4704 static int
parse_object_range(char * range,zopt_object_range_t * zor,const char ** msg)4705 parse_object_range(char *range, zopt_object_range_t *zor, const char **msg)
4706 {
4707 	uint64_t flags = 0;
4708 	char *p, *s, *dup, *flagstr, *tmp = NULL;
4709 	size_t len;
4710 	int i;
4711 	int rc = 0;
4712 
4713 	if (strchr(range, ':') == NULL) {
4714 		zor->zor_obj_start = strtoull(range, &p, 0);
4715 		if (*p != '\0') {
4716 			*msg = "Invalid characters in object ID";
4717 			rc = 1;
4718 		}
4719 		zor->zor_obj_start = ZDB_MAP_OBJECT_ID(zor->zor_obj_start);
4720 		zor->zor_obj_end = zor->zor_obj_start;
4721 		return (rc);
4722 	}
4723 
4724 	if (strchr(range, ':') == range) {
4725 		*msg = "Invalid leading colon";
4726 		rc = 1;
4727 		return (rc);
4728 	}
4729 
4730 	len = strlen(range);
4731 	if (range[len - 1] == ':') {
4732 		*msg = "Invalid trailing colon";
4733 		rc = 1;
4734 		return (rc);
4735 	}
4736 
4737 	dup = strdup(range);
4738 	s = strtok_r(dup, ":", &tmp);
4739 	zor->zor_obj_start = strtoull(s, &p, 0);
4740 
4741 	if (*p != '\0') {
4742 		*msg = "Invalid characters in start object ID";
4743 		rc = 1;
4744 		goto out;
4745 	}
4746 
4747 	s = strtok_r(NULL, ":", &tmp);
4748 	zor->zor_obj_end = strtoull(s, &p, 0);
4749 
4750 	if (*p != '\0') {
4751 		*msg = "Invalid characters in end object ID";
4752 		rc = 1;
4753 		goto out;
4754 	}
4755 
4756 	if (zor->zor_obj_start > zor->zor_obj_end) {
4757 		*msg = "Start object ID may not exceed end object ID";
4758 		rc = 1;
4759 		goto out;
4760 	}
4761 
4762 	s = strtok_r(NULL, ":", &tmp);
4763 	if (s == NULL) {
4764 		zor->zor_flags = ZOR_FLAG_ALL_TYPES;
4765 		goto out;
4766 	} else if (strtok_r(NULL, ":", &tmp) != NULL) {
4767 		*msg = "Invalid colon-delimited field after flags";
4768 		rc = 1;
4769 		goto out;
4770 	}
4771 
4772 	flagstr = s;
4773 	for (i = 0; flagstr[i]; i++) {
4774 		int bit;
4775 		boolean_t negation = (flagstr[i] == '-');
4776 
4777 		if (negation) {
4778 			i++;
4779 			if (flagstr[i] == '\0') {
4780 				*msg = "Invalid trailing negation operator";
4781 				rc = 1;
4782 				goto out;
4783 			}
4784 		}
4785 		bit = flagbits[(uchar_t)flagstr[i]];
4786 		if (bit == 0) {
4787 			*msg = "Invalid flag";
4788 			rc = 1;
4789 			goto out;
4790 		}
4791 		if (negation)
4792 			flags &= ~bit;
4793 		else
4794 			flags |= bit;
4795 	}
4796 	zor->zor_flags = flags;
4797 
4798 	zor->zor_obj_start = ZDB_MAP_OBJECT_ID(zor->zor_obj_start);
4799 	zor->zor_obj_end = ZDB_MAP_OBJECT_ID(zor->zor_obj_end);
4800 
4801 out:
4802 	free(dup);
4803 	return (rc);
4804 }
4805 
4806 static void
dump_objset(objset_t * os)4807 dump_objset(objset_t *os)
4808 {
4809 	dmu_objset_stats_t dds = { 0 };
4810 	uint64_t object, object_count;
4811 	uint64_t refdbytes, usedobjs, scratch;
4812 	char numbuf[32];
4813 	char blkbuf[BP_SPRINTF_LEN + 20];
4814 	char osname[ZFS_MAX_DATASET_NAME_LEN];
4815 	const char *type = "UNKNOWN";
4816 	int verbosity = dump_opt['d'];
4817 	boolean_t print_header;
4818 	unsigned i;
4819 	int error;
4820 	uint64_t total_slots_used = 0;
4821 	uint64_t max_slot_used = 0;
4822 	uint64_t dnode_slots;
4823 	uint64_t obj_start;
4824 	uint64_t obj_end;
4825 	uint64_t flags;
4826 
4827 	/* make sure nicenum has enough space */
4828 	_Static_assert(sizeof (numbuf) >= NN_NUMBUF_SZ, "numbuf truncated");
4829 
4830 	dsl_pool_config_enter(dmu_objset_pool(os), FTAG);
4831 	dmu_objset_fast_stat(os, &dds);
4832 	dsl_pool_config_exit(dmu_objset_pool(os), FTAG);
4833 
4834 	print_header = B_TRUE;
4835 
4836 	if (dds.dds_type < DMU_OST_NUMTYPES)
4837 		type = objset_types[dds.dds_type];
4838 
4839 	if (dds.dds_type == DMU_OST_META) {
4840 		dds.dds_creation_txg = TXG_INITIAL;
4841 		usedobjs = BP_GET_FILL(os->os_rootbp);
4842 		refdbytes = dsl_dir_phys(os->os_spa->spa_dsl_pool->dp_mos_dir)->
4843 		    dd_used_bytes;
4844 	} else {
4845 		dmu_objset_space(os, &refdbytes, &scratch, &usedobjs, &scratch);
4846 	}
4847 
4848 	ASSERT3U(usedobjs, ==, BP_GET_FILL(os->os_rootbp));
4849 
4850 	zdb_nicenum(refdbytes, numbuf, sizeof (numbuf));
4851 
4852 	if (verbosity >= 4 || dump_opt['d']) {
4853 		(void) snprintf(blkbuf, sizeof (blkbuf), ", rootbp ");
4854 		(void) snprintf_blkptr(blkbuf + strlen(blkbuf),
4855 		    sizeof (blkbuf) - strlen(blkbuf), os->os_rootbp);
4856 	} else {
4857 		blkbuf[0] = '\0';
4858 	}
4859 
4860 	dmu_objset_name(os, osname);
4861 
4862 	(void) printf("Dataset %s [%s], ID %llu, cr_txg %llu, "
4863 	    "%s, %llu objects%s%s\n",
4864 	    osname, type, (u_longlong_t)dmu_objset_id(os),
4865 	    (u_longlong_t)dds.dds_creation_txg,
4866 	    numbuf, (u_longlong_t)usedobjs, blkbuf,
4867 	    (dds.dds_inconsistent) ? " (inconsistent)" : "");
4868 
4869 	for (i = 0; i < zopt_object_args; i++) {
4870 		obj_start = zopt_object_ranges[i].zor_obj_start;
4871 		obj_end = zopt_object_ranges[i].zor_obj_end;
4872 		flags = zopt_object_ranges[i].zor_flags;
4873 
4874 		object = obj_start;
4875 		if (object == 0 || obj_start == obj_end) {
4876 			dump_object(os, object, verbosity,
4877 			    &print_header, NULL, flags);
4878 		} else {
4879 			object--;
4880 		}
4881 
4882 		while ((dmu_object_next(os, &object, B_FALSE, 0) == 0) &&
4883 		    object <= obj_end) {
4884 			dump_object(os, object, verbosity,
4885 			    &print_header, NULL, flags);
4886 		}
4887 	}
4888 
4889 	if (zopt_object_args > 0) {
4890 		return;
4891 	}
4892 
4893 	if (dump_opt['i'] != 0 || verbosity >= 2)
4894 		dump_intent_log(dmu_objset_zil(os));
4895 
4896 	if (dmu_objset_ds(os) != NULL) {
4897 		dsl_dataset_t *ds = dmu_objset_ds(os);
4898 		dump_blkptr_list(&ds->ds_deadlist, "Deadlist");
4899 		if (dsl_deadlist_is_open(&ds->ds_dir->dd_livelist) &&
4900 		    !dmu_objset_is_snapshot(os)) {
4901 			dump_blkptr_list(&ds->ds_dir->dd_livelist, "Livelist");
4902 			if (verify_dd_livelist(os) != 0)
4903 				fatal("livelist is incorrect");
4904 		}
4905 
4906 		if (dsl_dataset_remap_deadlist_exists(ds)) {
4907 			(void) printf("ds_remap_deadlist:\n");
4908 			dump_blkptr_list(&ds->ds_remap_deadlist, "Deadlist");
4909 		}
4910 		count_ds_mos_objects(ds);
4911 	}
4912 
4913 	if (dmu_objset_ds(os) != NULL)
4914 		dump_bookmarks(os, verbosity);
4915 
4916 	if (verbosity < 2)
4917 		return;
4918 
4919 	if (BP_IS_HOLE(os->os_rootbp))
4920 		return;
4921 
4922 	dump_object(os, 0, verbosity, &print_header, NULL, 0);
4923 	object_count = 0;
4924 	if (DMU_USERUSED_DNODE(os) != NULL &&
4925 	    DMU_USERUSED_DNODE(os)->dn_type != 0) {
4926 		dump_object(os, DMU_USERUSED_OBJECT, verbosity, &print_header,
4927 		    NULL, 0);
4928 		dump_object(os, DMU_GROUPUSED_OBJECT, verbosity, &print_header,
4929 		    NULL, 0);
4930 	}
4931 
4932 	if (DMU_PROJECTUSED_DNODE(os) != NULL &&
4933 	    DMU_PROJECTUSED_DNODE(os)->dn_type != 0)
4934 		dump_object(os, DMU_PROJECTUSED_OBJECT, verbosity,
4935 		    &print_header, NULL, 0);
4936 
4937 	object = 0;
4938 	while ((error = dmu_object_next(os, &object, B_FALSE, 0)) == 0) {
4939 		dump_object(os, object, verbosity, &print_header, &dnode_slots,
4940 		    0);
4941 		object_count++;
4942 		total_slots_used += dnode_slots;
4943 		max_slot_used = object + dnode_slots - 1;
4944 	}
4945 
4946 	(void) printf("\n");
4947 
4948 	(void) printf("    Dnode slots:\n");
4949 	(void) printf("\tTotal used:    %10llu\n",
4950 	    (u_longlong_t)total_slots_used);
4951 	(void) printf("\tMax used:      %10llu\n",
4952 	    (u_longlong_t)max_slot_used);
4953 	(void) printf("\tPercent empty: %10lf\n",
4954 	    (double)(max_slot_used - total_slots_used)*100 /
4955 	    (double)max_slot_used);
4956 	(void) printf("\n");
4957 
4958 	if (error != ESRCH) {
4959 		(void) fprintf(stderr, "dmu_object_next() = %d\n", error);
4960 		abort();
4961 	}
4962 
4963 	ASSERT3U(object_count, ==, usedobjs);
4964 
4965 	if (leaked_objects != 0) {
4966 		(void) printf("%d potentially leaked objects detected\n",
4967 		    leaked_objects);
4968 		leaked_objects = 0;
4969 	}
4970 }
4971 
4972 static void
dump_file_data_layout(objset_t * os)4973 dump_file_data_layout(objset_t *os)
4974 {
4975 	uint64_t object;
4976 	int verbosity = dump_opt['d'];
4977 	unsigned i;
4978 	uint64_t obj_start;
4979 	uint64_t obj_end;
4980 
4981 	for (i = 0; i < zopt_object_args; i++) {
4982 		obj_start = zopt_object_ranges[i].zor_obj_start;
4983 		obj_end = zopt_object_ranges[i].zor_obj_end;
4984 
4985 		object = obj_start;
4986 		if (object == 0 || obj_start == obj_end)
4987 			dump_object_file_layout(os, object, verbosity);
4988 		else
4989 			object--;
4990 
4991 		while ((dmu_object_next(os, &object, B_FALSE, 0) == 0) &&
4992 		    object <= obj_end) {
4993 			dump_object_file_layout(os, object, verbosity);
4994 		}
4995 	}
4996 }
4997 
4998 static void
dump_uberblock(uberblock_t * ub,const char * header,const char * footer)4999 dump_uberblock(uberblock_t *ub, const char *header, const char *footer)
5000 {
5001 	time_t timestamp = ub->ub_timestamp;
5002 
5003 	(void) printf("%s", header ? header : "");
5004 	(void) printf("\tmagic = %016llx\n", (u_longlong_t)ub->ub_magic);
5005 	(void) printf("\tversion = %llu\n", (u_longlong_t)ub->ub_version);
5006 	(void) printf("\ttxg = %llu\n", (u_longlong_t)ub->ub_txg);
5007 	(void) printf("\tguid_sum = %llu\n", (u_longlong_t)ub->ub_guid_sum);
5008 	(void) printf("\ttimestamp = %llu UTC = %s",
5009 	    (u_longlong_t)ub->ub_timestamp, ctime(&timestamp));
5010 
5011 	char blkbuf[BP_SPRINTF_LEN];
5012 	snprintf_blkptr(blkbuf, sizeof (blkbuf), &ub->ub_rootbp);
5013 	(void) printf("\tbp = %s\n", blkbuf);
5014 
5015 	(void) printf("\tmmp_magic = %016llx\n",
5016 	    (u_longlong_t)ub->ub_mmp_magic);
5017 	if (MMP_VALID(ub)) {
5018 		(void) printf("\tmmp_delay = %0llu\n",
5019 		    (u_longlong_t)ub->ub_mmp_delay);
5020 		if (MMP_SEQ_VALID(ub))
5021 			(void) printf("\tmmp_seq = %u\n",
5022 			    (unsigned int) MMP_SEQ(ub));
5023 		if (MMP_FAIL_INT_VALID(ub))
5024 			(void) printf("\tmmp_fail = %u\n",
5025 			    (unsigned int) MMP_FAIL_INT(ub));
5026 		if (MMP_INTERVAL_VALID(ub))
5027 			(void) printf("\tmmp_write = %u\n",
5028 			    (unsigned int) MMP_INTERVAL(ub));
5029 		/* After MMP_* to make summarize_uberblock_mmp cleaner */
5030 		(void) printf("\tmmp_valid = %x\n",
5031 		    (unsigned int) ub->ub_mmp_config & 0xFF);
5032 	}
5033 
5034 	if (dump_opt['u'] >= 4) {
5035 		char blkbuf[BP_SPRINTF_LEN];
5036 		snprintf_blkptr(blkbuf, sizeof (blkbuf), &ub->ub_rootbp);
5037 		(void) printf("\trootbp = %s\n", blkbuf);
5038 	}
5039 	(void) printf("\tcheckpoint_txg = %llu\n",
5040 	    (u_longlong_t)ub->ub_checkpoint_txg);
5041 
5042 	(void) printf("\traidz_reflow state=%u off=%llu\n",
5043 	    (int)RRSS_GET_STATE(ub),
5044 	    (u_longlong_t)RRSS_GET_OFFSET(ub));
5045 
5046 	(void) printf("%s", footer ? footer : "");
5047 }
5048 
5049 static void
dump_config(spa_t * spa)5050 dump_config(spa_t *spa)
5051 {
5052 	dmu_buf_t *db;
5053 	size_t nvsize = 0;
5054 	int error = 0;
5055 
5056 
5057 	error = dmu_bonus_hold(spa->spa_meta_objset,
5058 	    spa->spa_config_object, FTAG, &db);
5059 
5060 	if (error == 0) {
5061 		nvsize = *(uint64_t *)db->db_data;
5062 		dmu_buf_rele(db, FTAG);
5063 
5064 		(void) printf("\nMOS Configuration:\n");
5065 		dump_packed_nvlist(spa->spa_meta_objset,
5066 		    spa->spa_config_object, (void *)&nvsize, 1);
5067 	} else {
5068 		(void) fprintf(stderr, "dmu_bonus_hold(%llu) failed, errno %d",
5069 		    (u_longlong_t)spa->spa_config_object, error);
5070 	}
5071 }
5072 
5073 static void
dump_cachefile(const char * cachefile)5074 dump_cachefile(const char *cachefile)
5075 {
5076 	int fd;
5077 	struct stat64 statbuf;
5078 	char *buf;
5079 	nvlist_t *config;
5080 
5081 	if ((fd = open64(cachefile, O_RDONLY)) < 0) {
5082 		(void) printf("cannot open '%s': %s\n", cachefile,
5083 		    strerror(errno));
5084 		zdb_exit(1);
5085 	}
5086 
5087 	if (fstat64(fd, &statbuf) != 0) {
5088 		(void) printf("failed to stat '%s': %s\n", cachefile,
5089 		    strerror(errno));
5090 		zdb_exit(1);
5091 	}
5092 
5093 	if ((buf = malloc(statbuf.st_size)) == NULL) {
5094 		(void) fprintf(stderr, "failed to allocate %llu bytes\n",
5095 		    (u_longlong_t)statbuf.st_size);
5096 		zdb_exit(1);
5097 	}
5098 
5099 	if (read(fd, buf, statbuf.st_size) != statbuf.st_size) {
5100 		(void) fprintf(stderr, "failed to read %llu bytes\n",
5101 		    (u_longlong_t)statbuf.st_size);
5102 		zdb_exit(1);
5103 	}
5104 
5105 	(void) close(fd);
5106 
5107 	if (nvlist_unpack(buf, statbuf.st_size, &config, 0) != 0) {
5108 		(void) fprintf(stderr, "failed to unpack nvlist\n");
5109 		zdb_exit(1);
5110 	}
5111 
5112 	free(buf);
5113 
5114 	dump_nvlist(config, 0);
5115 
5116 	nvlist_free(config);
5117 }
5118 
5119 /*
5120  * ZFS label nvlist stats
5121  */
5122 typedef struct zdb_nvl_stats {
5123 	int		zns_list_count;
5124 	int		zns_leaf_count;
5125 	size_t		zns_leaf_largest;
5126 	size_t		zns_leaf_total;
5127 	nvlist_t	*zns_string;
5128 	nvlist_t	*zns_uint64;
5129 	nvlist_t	*zns_boolean;
5130 } zdb_nvl_stats_t;
5131 
5132 static void
collect_nvlist_stats(nvlist_t * nvl,zdb_nvl_stats_t * stats)5133 collect_nvlist_stats(nvlist_t *nvl, zdb_nvl_stats_t *stats)
5134 {
5135 	nvlist_t *list, **array;
5136 	nvpair_t *nvp = NULL;
5137 	const char *name;
5138 	uint_t i, items;
5139 
5140 	stats->zns_list_count++;
5141 
5142 	while ((nvp = nvlist_next_nvpair(nvl, nvp)) != NULL) {
5143 		name = nvpair_name(nvp);
5144 
5145 		switch (nvpair_type(nvp)) {
5146 		case DATA_TYPE_STRING:
5147 			fnvlist_add_string(stats->zns_string, name,
5148 			    fnvpair_value_string(nvp));
5149 			break;
5150 		case DATA_TYPE_UINT64:
5151 			fnvlist_add_uint64(stats->zns_uint64, name,
5152 			    fnvpair_value_uint64(nvp));
5153 			break;
5154 		case DATA_TYPE_BOOLEAN:
5155 			fnvlist_add_boolean(stats->zns_boolean, name);
5156 			break;
5157 		case DATA_TYPE_NVLIST:
5158 			if (nvpair_value_nvlist(nvp, &list) == 0)
5159 				collect_nvlist_stats(list, stats);
5160 			break;
5161 		case DATA_TYPE_NVLIST_ARRAY:
5162 			if (nvpair_value_nvlist_array(nvp, &array, &items) != 0)
5163 				break;
5164 
5165 			for (i = 0; i < items; i++) {
5166 				collect_nvlist_stats(array[i], stats);
5167 
5168 				/* collect stats on leaf vdev */
5169 				if (strcmp(name, "children") == 0) {
5170 					size_t size;
5171 
5172 					(void) nvlist_size(array[i], &size,
5173 					    NV_ENCODE_XDR);
5174 					stats->zns_leaf_total += size;
5175 					if (size > stats->zns_leaf_largest)
5176 						stats->zns_leaf_largest = size;
5177 					stats->zns_leaf_count++;
5178 				}
5179 			}
5180 			break;
5181 		default:
5182 			(void) printf("skip type %d!\n", (int)nvpair_type(nvp));
5183 		}
5184 	}
5185 }
5186 
5187 static void
dump_nvlist_stats(nvlist_t * nvl,size_t cap)5188 dump_nvlist_stats(nvlist_t *nvl, size_t cap)
5189 {
5190 	zdb_nvl_stats_t stats = { 0 };
5191 	size_t size, sum = 0, total;
5192 	size_t noise;
5193 
5194 	/* requires nvlist with non-unique names for stat collection */
5195 	VERIFY0(nvlist_alloc(&stats.zns_string, 0, 0));
5196 	VERIFY0(nvlist_alloc(&stats.zns_uint64, 0, 0));
5197 	VERIFY0(nvlist_alloc(&stats.zns_boolean, 0, 0));
5198 	VERIFY0(nvlist_size(stats.zns_boolean, &noise, NV_ENCODE_XDR));
5199 
5200 	(void) printf("\n\nZFS Label NVList Config Stats:\n");
5201 
5202 	VERIFY0(nvlist_size(nvl, &total, NV_ENCODE_XDR));
5203 	(void) printf("  %d bytes used, %d bytes free (using %4.1f%%)\n\n",
5204 	    (int)total, (int)(cap - total), 100.0 * total / cap);
5205 
5206 	collect_nvlist_stats(nvl, &stats);
5207 
5208 	VERIFY0(nvlist_size(stats.zns_uint64, &size, NV_ENCODE_XDR));
5209 	size -= noise;
5210 	sum += size;
5211 	(void) printf("%12s %4d %6d bytes (%5.2f%%)\n", "integers:",
5212 	    (int)fnvlist_num_pairs(stats.zns_uint64),
5213 	    (int)size, 100.0 * size / total);
5214 
5215 	VERIFY0(nvlist_size(stats.zns_string, &size, NV_ENCODE_XDR));
5216 	size -= noise;
5217 	sum += size;
5218 	(void) printf("%12s %4d %6d bytes (%5.2f%%)\n", "strings:",
5219 	    (int)fnvlist_num_pairs(stats.zns_string),
5220 	    (int)size, 100.0 * size / total);
5221 
5222 	VERIFY0(nvlist_size(stats.zns_boolean, &size, NV_ENCODE_XDR));
5223 	size -= noise;
5224 	sum += size;
5225 	(void) printf("%12s %4d %6d bytes (%5.2f%%)\n", "booleans:",
5226 	    (int)fnvlist_num_pairs(stats.zns_boolean),
5227 	    (int)size, 100.0 * size / total);
5228 
5229 	size = total - sum;	/* treat remainder as nvlist overhead */
5230 	(void) printf("%12s %4d %6d bytes (%5.2f%%)\n\n", "nvlists:",
5231 	    stats.zns_list_count, (int)size, 100.0 * size / total);
5232 
5233 	if (stats.zns_leaf_count > 0) {
5234 		size_t average = stats.zns_leaf_total / stats.zns_leaf_count;
5235 
5236 		(void) printf("%12s %4d %6d bytes average\n", "leaf vdevs:",
5237 		    stats.zns_leaf_count, (int)average);
5238 		(void) printf("%24d bytes largest\n",
5239 		    (int)stats.zns_leaf_largest);
5240 
5241 		if (dump_opt['l'] >= 3 && average > 0)
5242 			(void) printf("  space for %d additional leaf vdevs\n",
5243 			    (int)((cap - total) / average));
5244 	}
5245 	(void) printf("\n");
5246 
5247 	nvlist_free(stats.zns_string);
5248 	nvlist_free(stats.zns_uint64);
5249 	nvlist_free(stats.zns_boolean);
5250 }
5251 
5252 typedef struct cksum_record {
5253 	zio_cksum_t cksum;
5254 	boolean_t labels[VDEV_LABELS];
5255 	avl_node_t link;
5256 } cksum_record_t;
5257 
5258 static int
cksum_record_compare(const void * x1,const void * x2)5259 cksum_record_compare(const void *x1, const void *x2)
5260 {
5261 	const cksum_record_t *l = (cksum_record_t *)x1;
5262 	const cksum_record_t *r = (cksum_record_t *)x2;
5263 	int arraysize = ARRAY_SIZE(l->cksum.zc_word);
5264 	int difference = 0;
5265 
5266 	for (int i = 0; i < arraysize; i++) {
5267 		difference = TREE_CMP(l->cksum.zc_word[i], r->cksum.zc_word[i]);
5268 		if (difference)
5269 			break;
5270 	}
5271 
5272 	return (difference);
5273 }
5274 
5275 static cksum_record_t *
cksum_record_alloc(zio_cksum_t * cksum,int l)5276 cksum_record_alloc(zio_cksum_t *cksum, int l)
5277 {
5278 	cksum_record_t *rec;
5279 
5280 	rec = umem_zalloc(sizeof (*rec), UMEM_NOFAIL);
5281 	rec->cksum = *cksum;
5282 	rec->labels[l] = B_TRUE;
5283 
5284 	return (rec);
5285 }
5286 
5287 static cksum_record_t *
cksum_record_lookup(avl_tree_t * tree,zio_cksum_t * cksum)5288 cksum_record_lookup(avl_tree_t *tree, zio_cksum_t *cksum)
5289 {
5290 	cksum_record_t lookup = { .cksum = *cksum };
5291 	avl_index_t where;
5292 
5293 	return (avl_find(tree, &lookup, &where));
5294 }
5295 
5296 static cksum_record_t *
cksum_record_insert(avl_tree_t * tree,zio_cksum_t * cksum,int l)5297 cksum_record_insert(avl_tree_t *tree, zio_cksum_t *cksum, int l)
5298 {
5299 	cksum_record_t *rec;
5300 
5301 	rec = cksum_record_lookup(tree, cksum);
5302 	if (rec) {
5303 		rec->labels[l] = B_TRUE;
5304 	} else {
5305 		rec = cksum_record_alloc(cksum, l);
5306 		avl_add(tree, rec);
5307 	}
5308 
5309 	return (rec);
5310 }
5311 
5312 static int
first_label(cksum_record_t * rec)5313 first_label(cksum_record_t *rec)
5314 {
5315 	for (int i = 0; i < VDEV_LABELS; i++)
5316 		if (rec->labels[i])
5317 			return (i);
5318 
5319 	return (-1);
5320 }
5321 
5322 static void
print_label_numbers(const char * prefix,const cksum_record_t * rec)5323 print_label_numbers(const char *prefix, const cksum_record_t *rec)
5324 {
5325 	fputs(prefix, stdout);
5326 	for (int i = 0; i < VDEV_LABELS; i++)
5327 		if (rec->labels[i] == B_TRUE)
5328 			printf("%d ", i);
5329 	putchar('\n');
5330 }
5331 
5332 #define	MAX_UBERBLOCK_COUNT (VDEV_UBERBLOCK_RING >> UBERBLOCK_SHIFT)
5333 
5334 typedef struct zdb_label {
5335 	vdev_label_t label;
5336 	uint64_t label_offset;
5337 	nvlist_t *config_nv;
5338 	cksum_record_t *config;
5339 	cksum_record_t *uberblocks[MAX_UBERBLOCK_COUNT];
5340 	boolean_t header_printed;
5341 	boolean_t read_failed;
5342 	boolean_t cksum_valid;
5343 } zdb_label_t;
5344 
5345 static void
print_label_header(zdb_label_t * label,int l)5346 print_label_header(zdb_label_t *label, int l)
5347 {
5348 
5349 	if (dump_opt['q'])
5350 		return;
5351 
5352 	if (label->header_printed == B_TRUE)
5353 		return;
5354 
5355 	(void) printf("------------------------------------\n");
5356 	(void) printf("LABEL %d %s\n", l,
5357 	    label->cksum_valid ? "" : "(Bad label cksum)");
5358 	(void) printf("------------------------------------\n");
5359 
5360 	label->header_printed = B_TRUE;
5361 }
5362 
5363 static void
print_l2arc_header(void)5364 print_l2arc_header(void)
5365 {
5366 	(void) printf("------------------------------------\n");
5367 	(void) printf("L2ARC device header\n");
5368 	(void) printf("------------------------------------\n");
5369 }
5370 
5371 static void
print_l2arc_log_blocks(void)5372 print_l2arc_log_blocks(void)
5373 {
5374 	(void) printf("------------------------------------\n");
5375 	(void) printf("L2ARC device log blocks\n");
5376 	(void) printf("------------------------------------\n");
5377 }
5378 
5379 static void
dump_l2arc_log_entries(uint64_t log_entries,l2arc_log_ent_phys_t * le,uint64_t i)5380 dump_l2arc_log_entries(uint64_t log_entries,
5381     l2arc_log_ent_phys_t *le, uint64_t i)
5382 {
5383 	for (int j = 0; j < log_entries; j++) {
5384 		dva_t dva = le[j].le_dva;
5385 		(void) printf("lb[%4llu]\tle[%4d]\tDVA asize: %llu, "
5386 		    "vdev: %llu, offset: %llu\n",
5387 		    (u_longlong_t)i, j + 1,
5388 		    (u_longlong_t)DVA_GET_ASIZE(&dva),
5389 		    (u_longlong_t)DVA_GET_VDEV(&dva),
5390 		    (u_longlong_t)DVA_GET_OFFSET(&dva));
5391 		(void) printf("|\t\t\t\tbirth: %llu\n",
5392 		    (u_longlong_t)le[j].le_birth);
5393 		(void) printf("|\t\t\t\tlsize: %llu\n",
5394 		    (u_longlong_t)L2BLK_GET_LSIZE((&le[j])->le_prop));
5395 		(void) printf("|\t\t\t\tpsize: %llu\n",
5396 		    (u_longlong_t)L2BLK_GET_PSIZE((&le[j])->le_prop));
5397 		(void) printf("|\t\t\t\tcompr: %llu\n",
5398 		    (u_longlong_t)L2BLK_GET_COMPRESS((&le[j])->le_prop));
5399 		(void) printf("|\t\t\t\tcomplevel: %llu\n",
5400 		    (u_longlong_t)(&le[j])->le_complevel);
5401 		(void) printf("|\t\t\t\ttype: %llu\n",
5402 		    (u_longlong_t)L2BLK_GET_TYPE((&le[j])->le_prop));
5403 		(void) printf("|\t\t\t\tprotected: %llu\n",
5404 		    (u_longlong_t)L2BLK_GET_PROTECTED((&le[j])->le_prop));
5405 		(void) printf("|\t\t\t\tprefetch: %llu\n",
5406 		    (u_longlong_t)L2BLK_GET_PREFETCH((&le[j])->le_prop));
5407 		(void) printf("|\t\t\t\taddress: %llu\n",
5408 		    (u_longlong_t)le[j].le_daddr);
5409 		(void) printf("|\t\t\t\tARC state: %llu\n",
5410 		    (u_longlong_t)L2BLK_GET_STATE((&le[j])->le_prop));
5411 		(void) printf("|\n");
5412 	}
5413 	(void) printf("\n");
5414 }
5415 
5416 static void
dump_l2arc_log_blkptr(const l2arc_log_blkptr_t * lbps)5417 dump_l2arc_log_blkptr(const l2arc_log_blkptr_t *lbps)
5418 {
5419 	(void) printf("|\t\tdaddr: %llu\n", (u_longlong_t)lbps->lbp_daddr);
5420 	(void) printf("|\t\tpayload_asize: %llu\n",
5421 	    (u_longlong_t)lbps->lbp_payload_asize);
5422 	(void) printf("|\t\tpayload_start: %llu\n",
5423 	    (u_longlong_t)lbps->lbp_payload_start);
5424 	(void) printf("|\t\tlsize: %llu\n",
5425 	    (u_longlong_t)L2BLK_GET_LSIZE(lbps->lbp_prop));
5426 	(void) printf("|\t\tasize: %llu\n",
5427 	    (u_longlong_t)L2BLK_GET_PSIZE(lbps->lbp_prop));
5428 	(void) printf("|\t\tcompralgo: %llu\n",
5429 	    (u_longlong_t)L2BLK_GET_COMPRESS(lbps->lbp_prop));
5430 	(void) printf("|\t\tcksumalgo: %llu\n",
5431 	    (u_longlong_t)L2BLK_GET_CHECKSUM(lbps->lbp_prop));
5432 	(void) printf("|\n\n");
5433 }
5434 
5435 static void
dump_l2arc_log_blocks(int fd,const l2arc_dev_hdr_phys_t * l2dhdr,l2arc_dev_hdr_phys_t * rebuild)5436 dump_l2arc_log_blocks(int fd, const l2arc_dev_hdr_phys_t *l2dhdr,
5437     l2arc_dev_hdr_phys_t *rebuild)
5438 {
5439 	l2arc_log_blk_phys_t this_lb;
5440 	uint64_t asize;
5441 	l2arc_log_blkptr_t lbps[2];
5442 	zio_cksum_t cksum;
5443 	int failed = 0;
5444 	l2arc_dev_t dev;
5445 
5446 	if (!dump_opt['q'])
5447 		print_l2arc_log_blocks();
5448 	memcpy(lbps, l2dhdr->dh_start_lbps, sizeof (lbps));
5449 
5450 	dev.l2ad_evict = l2dhdr->dh_evict;
5451 	dev.l2ad_start = l2dhdr->dh_start;
5452 	dev.l2ad_end = l2dhdr->dh_end;
5453 
5454 	if (l2dhdr->dh_start_lbps[0].lbp_daddr == 0) {
5455 		/* no log blocks to read */
5456 		if (!dump_opt['q']) {
5457 			(void) printf("No log blocks to read\n");
5458 			(void) printf("\n");
5459 		}
5460 		return;
5461 	} else {
5462 		dev.l2ad_hand = lbps[0].lbp_daddr +
5463 		    L2BLK_GET_PSIZE((&lbps[0])->lbp_prop);
5464 	}
5465 
5466 	dev.l2ad_first = !!(l2dhdr->dh_flags & L2ARC_DEV_HDR_EVICT_FIRST);
5467 
5468 	for (;;) {
5469 		if (!l2arc_log_blkptr_valid(&dev, &lbps[0]))
5470 			break;
5471 
5472 		/* L2BLK_GET_PSIZE returns aligned size for log blocks */
5473 		asize = L2BLK_GET_PSIZE((&lbps[0])->lbp_prop);
5474 		if (pread64(fd, &this_lb, asize, lbps[0].lbp_daddr) != asize) {
5475 			if (!dump_opt['q']) {
5476 				(void) printf("Error while reading next log "
5477 				    "block\n\n");
5478 			}
5479 			break;
5480 		}
5481 
5482 		fletcher_4_native_varsize(&this_lb, asize, &cksum);
5483 		if (!ZIO_CHECKSUM_EQUAL(cksum, lbps[0].lbp_cksum)) {
5484 			failed++;
5485 			if (!dump_opt['q']) {
5486 				(void) printf("Invalid cksum\n");
5487 				dump_l2arc_log_blkptr(&lbps[0]);
5488 			}
5489 			break;
5490 		}
5491 
5492 		switch (L2BLK_GET_COMPRESS((&lbps[0])->lbp_prop)) {
5493 		case ZIO_COMPRESS_OFF:
5494 			break;
5495 		default: {
5496 			abd_t *abd = abd_alloc_linear(asize, B_TRUE);
5497 			abd_copy_from_buf_off(abd, &this_lb, 0, asize);
5498 			abd_t dabd;
5499 			abd_get_from_buf_struct(&dabd, &this_lb,
5500 			    sizeof (this_lb));
5501 			int err = zio_decompress_data(L2BLK_GET_COMPRESS(
5502 			    (&lbps[0])->lbp_prop), abd, &dabd,
5503 			    asize, sizeof (this_lb), NULL);
5504 			abd_free(&dabd);
5505 			abd_free(abd);
5506 			if (err != 0) {
5507 				(void) printf("L2ARC block decompression "
5508 				    "failed\n");
5509 				goto out;
5510 			}
5511 			break;
5512 		}
5513 		}
5514 
5515 		if (this_lb.lb_magic == BSWAP_64(L2ARC_LOG_BLK_MAGIC))
5516 			byteswap_uint64_array(&this_lb, sizeof (this_lb));
5517 		if (this_lb.lb_magic != L2ARC_LOG_BLK_MAGIC) {
5518 			if (!dump_opt['q'])
5519 				(void) printf("Invalid log block magic\n\n");
5520 			break;
5521 		}
5522 
5523 		rebuild->dh_lb_count++;
5524 		rebuild->dh_lb_asize += asize;
5525 		if (dump_opt['l'] > 1 && !dump_opt['q']) {
5526 			(void) printf("lb[%4llu]\tmagic: %llu\n",
5527 			    (u_longlong_t)rebuild->dh_lb_count,
5528 			    (u_longlong_t)this_lb.lb_magic);
5529 			dump_l2arc_log_blkptr(&lbps[0]);
5530 		}
5531 
5532 		if (dump_opt['l'] > 2 && !dump_opt['q'])
5533 			dump_l2arc_log_entries(l2dhdr->dh_log_entries,
5534 			    this_lb.lb_entries,
5535 			    rebuild->dh_lb_count);
5536 
5537 		if (l2arc_range_check_overlap(lbps[1].lbp_payload_start,
5538 		    lbps[0].lbp_payload_start, dev.l2ad_evict) &&
5539 		    !dev.l2ad_first)
5540 			break;
5541 
5542 		lbps[0] = lbps[1];
5543 		lbps[1] = this_lb.lb_prev_lbp;
5544 	}
5545 out:
5546 	if (!dump_opt['q']) {
5547 		(void) printf("log_blk_count:\t %llu with valid cksum\n",
5548 		    (u_longlong_t)rebuild->dh_lb_count);
5549 		(void) printf("\t\t %d with invalid cksum\n", failed);
5550 		(void) printf("log_blk_asize:\t %llu\n\n",
5551 		    (u_longlong_t)rebuild->dh_lb_asize);
5552 	}
5553 }
5554 
5555 static int
dump_l2arc_header(int fd)5556 dump_l2arc_header(int fd)
5557 {
5558 	l2arc_dev_hdr_phys_t l2dhdr = {0}, rebuild = {0};
5559 	int error = B_FALSE;
5560 
5561 	if (pread64(fd, &l2dhdr, sizeof (l2dhdr),
5562 	    VDEV_LABEL_START_SIZE) != sizeof (l2dhdr)) {
5563 		error = B_TRUE;
5564 	} else {
5565 		if (l2dhdr.dh_magic == BSWAP_64(L2ARC_DEV_HDR_MAGIC))
5566 			byteswap_uint64_array(&l2dhdr, sizeof (l2dhdr));
5567 
5568 		if (l2dhdr.dh_magic != L2ARC_DEV_HDR_MAGIC)
5569 			error = B_TRUE;
5570 	}
5571 
5572 	if (error) {
5573 		(void) printf("L2ARC device header not found\n\n");
5574 		/* Do not return an error here for backward compatibility */
5575 		return (0);
5576 	} else if (!dump_opt['q']) {
5577 		print_l2arc_header();
5578 
5579 		(void) printf("    magic: %llu\n",
5580 		    (u_longlong_t)l2dhdr.dh_magic);
5581 		(void) printf("    version: %llu\n",
5582 		    (u_longlong_t)l2dhdr.dh_version);
5583 		(void) printf("    pool_guid: %llu\n",
5584 		    (u_longlong_t)l2dhdr.dh_spa_guid);
5585 		(void) printf("    flags: %llu\n",
5586 		    (u_longlong_t)l2dhdr.dh_flags);
5587 		(void) printf("    start_lbps[0]: %llu\n",
5588 		    (u_longlong_t)
5589 		    l2dhdr.dh_start_lbps[0].lbp_daddr);
5590 		(void) printf("    start_lbps[1]: %llu\n",
5591 		    (u_longlong_t)
5592 		    l2dhdr.dh_start_lbps[1].lbp_daddr);
5593 		(void) printf("    log_blk_ent: %llu\n",
5594 		    (u_longlong_t)l2dhdr.dh_log_entries);
5595 		(void) printf("    start: %llu\n",
5596 		    (u_longlong_t)l2dhdr.dh_start);
5597 		(void) printf("    end: %llu\n",
5598 		    (u_longlong_t)l2dhdr.dh_end);
5599 		(void) printf("    evict: %llu\n",
5600 		    (u_longlong_t)l2dhdr.dh_evict);
5601 		(void) printf("    lb_asize_refcount: %llu\n",
5602 		    (u_longlong_t)l2dhdr.dh_lb_asize);
5603 		(void) printf("    lb_count_refcount: %llu\n",
5604 		    (u_longlong_t)l2dhdr.dh_lb_count);
5605 		(void) printf("    trim_action_time: %llu\n",
5606 		    (u_longlong_t)l2dhdr.dh_trim_action_time);
5607 		(void) printf("    trim_state: %llu\n\n",
5608 		    (u_longlong_t)l2dhdr.dh_trim_state);
5609 	}
5610 
5611 	dump_l2arc_log_blocks(fd, &l2dhdr, &rebuild);
5612 	/*
5613 	 * The total aligned size of log blocks and the number of log blocks
5614 	 * reported in the header of the device may be less than what zdb
5615 	 * reports by dump_l2arc_log_blocks() which emulates l2arc_rebuild().
5616 	 * This happens because dump_l2arc_log_blocks() lacks the memory
5617 	 * pressure valve that l2arc_rebuild() has. Thus, if we are on a system
5618 	 * with low memory, l2arc_rebuild will exit prematurely and dh_lb_asize
5619 	 * and dh_lb_count will be lower to begin with than what exists on the
5620 	 * device. This is normal and zdb should not exit with an error. The
5621 	 * opposite case should never happen though, the values reported in the
5622 	 * header should never be higher than what dump_l2arc_log_blocks() and
5623 	 * l2arc_rebuild() report. If this happens there is a leak in the
5624 	 * accounting of log blocks.
5625 	 */
5626 	if (l2dhdr.dh_lb_asize > rebuild.dh_lb_asize ||
5627 	    l2dhdr.dh_lb_count > rebuild.dh_lb_count)
5628 		return (1);
5629 
5630 	return (0);
5631 }
5632 
5633 static void
dump_config_from_label(zdb_label_t * label,size_t buflen,int l)5634 dump_config_from_label(zdb_label_t *label, size_t buflen, int l)
5635 {
5636 	if (dump_opt['q'])
5637 		return;
5638 
5639 	if ((dump_opt['l'] < 3) && (first_label(label->config) != l))
5640 		return;
5641 
5642 	print_label_header(label, l);
5643 	dump_nvlist(label->config_nv, 4);
5644 	print_label_numbers("    labels = ", label->config);
5645 
5646 	if (dump_opt['l'] >= 2)
5647 		dump_nvlist_stats(label->config_nv, buflen);
5648 }
5649 
5650 #define	ZDB_MAX_UB_HEADER_SIZE 32
5651 
5652 static void
dump_label_uberblocks(zdb_label_t * label,uint64_t ashift,int label_num)5653 dump_label_uberblocks(zdb_label_t *label, uint64_t ashift, int label_num)
5654 {
5655 
5656 	vdev_t vd;
5657 	char header[ZDB_MAX_UB_HEADER_SIZE];
5658 
5659 	vd.vdev_ashift = ashift;
5660 	vd.vdev_top = &vd;
5661 
5662 	for (int i = 0; i < VDEV_UBERBLOCK_COUNT(&vd); i++) {
5663 		uint64_t uoff = VDEV_UBERBLOCK_OFFSET(&vd, i);
5664 		uberblock_t *ub = (void *)((char *)&label->label + uoff);
5665 		cksum_record_t *rec = label->uberblocks[i];
5666 
5667 		if (rec == NULL) {
5668 			if (dump_opt['u'] >= 2) {
5669 				print_label_header(label, label_num);
5670 				(void) printf("    Uberblock[%d] invalid\n", i);
5671 			}
5672 			continue;
5673 		}
5674 
5675 		if ((dump_opt['u'] < 3) && (first_label(rec) != label_num))
5676 			continue;
5677 
5678 		if ((dump_opt['u'] < 4) &&
5679 		    (ub->ub_mmp_magic == MMP_MAGIC) && ub->ub_mmp_delay &&
5680 		    (i >= VDEV_UBERBLOCK_COUNT(&vd) - MMP_BLOCKS_PER_LABEL))
5681 			continue;
5682 
5683 		print_label_header(label, label_num);
5684 		(void) snprintf(header, ZDB_MAX_UB_HEADER_SIZE,
5685 		    "    Uberblock[%d]\n", i);
5686 		dump_uberblock(ub, header, "");
5687 		print_label_numbers("        labels = ", rec);
5688 	}
5689 }
5690 
5691 static char curpath[PATH_MAX];
5692 
5693 /*
5694  * Iterate through the path components, recursively passing
5695  * current one's obj and remaining path until we find the obj
5696  * for the last one.
5697  */
5698 static int
dump_path_impl(objset_t * os,uint64_t obj,char * name,uint64_t * retobj)5699 dump_path_impl(objset_t *os, uint64_t obj, char *name, uint64_t *retobj)
5700 {
5701 	int err;
5702 	boolean_t header = B_TRUE;
5703 	uint64_t child_obj;
5704 	char *s;
5705 	dmu_buf_t *db;
5706 	dmu_object_info_t doi;
5707 
5708 	if ((s = strchr(name, '/')) != NULL)
5709 		*s = '\0';
5710 	err = zap_lookup(os, obj, name, 8, 1, &child_obj);
5711 
5712 	(void) strlcat(curpath, name, sizeof (curpath));
5713 
5714 	if (err != 0) {
5715 		(void) fprintf(stderr, "failed to lookup %s: %s\n",
5716 		    curpath, strerror(err));
5717 		return (err);
5718 	}
5719 
5720 	child_obj = ZFS_DIRENT_OBJ(child_obj);
5721 	err = sa_buf_hold(os, child_obj, FTAG, &db);
5722 	if (err != 0) {
5723 		(void) fprintf(stderr,
5724 		    "failed to get SA dbuf for obj %llu: %s\n",
5725 		    (u_longlong_t)child_obj, strerror(err));
5726 		return (EINVAL);
5727 	}
5728 	dmu_object_info_from_db(db, &doi);
5729 	sa_buf_rele(db, FTAG);
5730 
5731 	if (doi.doi_bonus_type != DMU_OT_SA &&
5732 	    doi.doi_bonus_type != DMU_OT_ZNODE) {
5733 		(void) fprintf(stderr, "invalid bonus type %d for obj %llu\n",
5734 		    doi.doi_bonus_type, (u_longlong_t)child_obj);
5735 		return (EINVAL);
5736 	}
5737 
5738 	if (dump_opt['v'] > 6) {
5739 		(void) printf("obj=%llu %s type=%d bonustype=%d\n",
5740 		    (u_longlong_t)child_obj, curpath, doi.doi_type,
5741 		    doi.doi_bonus_type);
5742 	}
5743 
5744 	(void) strlcat(curpath, "/", sizeof (curpath));
5745 
5746 	switch (doi.doi_type) {
5747 	case DMU_OT_DIRECTORY_CONTENTS:
5748 		if (s != NULL && *(s + 1) != '\0')
5749 			return (dump_path_impl(os, child_obj, s + 1, retobj));
5750 		zfs_fallthrough;
5751 	case DMU_OT_PLAIN_FILE_CONTENTS:
5752 		if (retobj != NULL) {
5753 			*retobj = child_obj;
5754 		} else {
5755 			dump_object(os, child_obj, dump_opt['v'], &header,
5756 			    NULL, 0);
5757 		}
5758 		return (0);
5759 	default:
5760 		(void) fprintf(stderr, "object %llu has non-file/directory "
5761 		    "type %d\n", (u_longlong_t)obj, doi.doi_type);
5762 		break;
5763 	}
5764 
5765 	return (EINVAL);
5766 }
5767 
5768 /*
5769  * Dump the blocks for the object specified by path inside the dataset.
5770  */
5771 static int
dump_path(char * ds,char * path,uint64_t * retobj)5772 dump_path(char *ds, char *path, uint64_t *retobj)
5773 {
5774 	int err;
5775 	objset_t *os;
5776 	uint64_t root_obj;
5777 
5778 	err = open_objset(ds, FTAG, &os);
5779 	if (err != 0)
5780 		return (err);
5781 
5782 	err = zap_lookup(os, MASTER_NODE_OBJ, ZFS_ROOT_OBJ, 8, 1, &root_obj);
5783 	if (err != 0) {
5784 		(void) fprintf(stderr, "can't lookup root znode: %s\n",
5785 		    strerror(err));
5786 		close_objset(os, FTAG);
5787 		return (EINVAL);
5788 	}
5789 
5790 	(void) snprintf(curpath, sizeof (curpath), "dataset=%s path=/", ds);
5791 
5792 	err = dump_path_impl(os, root_obj, path, retobj);
5793 
5794 	close_objset(os, FTAG);
5795 	return (err);
5796 }
5797 
5798 static int
dump_backup_bytes(objset_t * os,void * buf,int len,void * arg)5799 dump_backup_bytes(objset_t *os, void *buf, int len, void *arg)
5800 {
5801 	const char *p = (const char *)buf;
5802 	ssize_t nwritten;
5803 
5804 	(void) os;
5805 	(void) arg;
5806 
5807 	/* Write the data out, handling short writes and signals. */
5808 	while ((nwritten = write(STDOUT_FILENO, p, len)) < len) {
5809 		if (nwritten < 0) {
5810 			if (errno == EINTR)
5811 				continue;
5812 			return (errno);
5813 		}
5814 		p += nwritten;
5815 		len -= nwritten;
5816 	}
5817 
5818 	return (0);
5819 }
5820 
5821 static void
dump_backup(const char * pool,uint64_t objset_id,const char * flagstr)5822 dump_backup(const char *pool, uint64_t objset_id, const char *flagstr)
5823 {
5824 	boolean_t embed = B_FALSE;
5825 	boolean_t large_block = B_FALSE;
5826 	boolean_t compress = B_FALSE;
5827 	boolean_t raw = B_FALSE;
5828 
5829 	const char *c;
5830 	for (c = flagstr; c != NULL && *c != '\0'; c++) {
5831 		switch (*c) {
5832 			case 'e':
5833 				embed = B_TRUE;
5834 				break;
5835 			case 'L':
5836 				large_block = B_TRUE;
5837 				break;
5838 			case 'c':
5839 				compress = B_TRUE;
5840 				break;
5841 			case 'w':
5842 				raw = B_TRUE;
5843 				break;
5844 			default:
5845 				fprintf(stderr, "dump_backup: invalid flag "
5846 				    "'%c'\n", *c);
5847 				return;
5848 		}
5849 	}
5850 
5851 	if (isatty(STDOUT_FILENO)) {
5852 		fprintf(stderr, "dump_backup: stream cannot be written "
5853 		    "to a terminal\n");
5854 		return;
5855 	}
5856 
5857 	offset_t off = 0;
5858 	dmu_send_outparams_t out = {
5859 	    .dso_outfunc = dump_backup_bytes,
5860 	    .dso_dryrun  = B_FALSE,
5861 	};
5862 
5863 	int err = dmu_send_obj(pool, objset_id, /* fromsnap */0, embed,
5864 	    large_block, compress, raw, /* saved */ B_FALSE, STDOUT_FILENO,
5865 	    &off, &out);
5866 	if (err != 0) {
5867 		fprintf(stderr, "dump_backup: dmu_send_obj: %s\n",
5868 		    strerror(err));
5869 		return;
5870 	}
5871 }
5872 
5873 static int
zdb_copy_object(objset_t * os,uint64_t srcobj,char * destfile)5874 zdb_copy_object(objset_t *os, uint64_t srcobj, char *destfile)
5875 {
5876 	int err = 0;
5877 	uint64_t size, readsize, oursize, offset;
5878 	ssize_t writesize;
5879 	sa_handle_t *hdl;
5880 
5881 	(void) printf("Copying object %" PRIu64 " to file %s\n", srcobj,
5882 	    destfile);
5883 
5884 	VERIFY3P(os, ==, sa_os);
5885 	if ((err = sa_handle_get(os, srcobj, NULL, SA_HDL_PRIVATE, &hdl))) {
5886 		(void) printf("Failed to get handle for SA znode\n");
5887 		return (err);
5888 	}
5889 	if ((err = sa_lookup(hdl, sa_attr_table[ZPL_SIZE], &size, 8))) {
5890 		(void) sa_handle_destroy(hdl);
5891 		return (err);
5892 	}
5893 	(void) sa_handle_destroy(hdl);
5894 
5895 	(void) printf("Object %" PRIu64 " is %" PRIu64 " bytes\n", srcobj,
5896 	    size);
5897 	if (size == 0) {
5898 		return (EINVAL);
5899 	}
5900 
5901 	int fd = open(destfile, O_WRONLY | O_CREAT | O_TRUNC, 0644);
5902 	if (fd == -1)
5903 		return (errno);
5904 	/*
5905 	 * We cap the size at 1 mebibyte here to prevent
5906 	 * allocation failures and nigh-infinite printing if the
5907 	 * object is extremely large.
5908 	 */
5909 	oursize = MIN(size, 1 << 20);
5910 	offset = 0;
5911 	char *buf = kmem_alloc(oursize, KM_NOSLEEP);
5912 	if (buf == NULL) {
5913 		(void) close(fd);
5914 		return (ENOMEM);
5915 	}
5916 
5917 	while (offset < size) {
5918 		readsize = MIN(size - offset, 1 << 20);
5919 		err = dmu_read(os, srcobj, offset, readsize, buf, 0);
5920 		if (err != 0) {
5921 			(void) printf("got error %u from dmu_read\n", err);
5922 			kmem_free(buf, oursize);
5923 			(void) close(fd);
5924 			return (err);
5925 		}
5926 		if (dump_opt['v'] > 3) {
5927 			(void) printf("Read offset=%" PRIu64 " size=%" PRIu64
5928 			    " error=%d\n", offset, readsize, err);
5929 		}
5930 
5931 		writesize = write(fd, buf, readsize);
5932 		if (writesize < 0) {
5933 			err = errno;
5934 			break;
5935 		} else if (writesize != readsize) {
5936 			/* Incomplete write */
5937 			(void) fprintf(stderr, "Short write, only wrote %llu of"
5938 			    " %" PRIu64 " bytes, exiting...\n",
5939 			    (u_longlong_t)writesize, readsize);
5940 			break;
5941 		}
5942 
5943 		offset += readsize;
5944 	}
5945 
5946 	(void) close(fd);
5947 
5948 	if (buf != NULL)
5949 		kmem_free(buf, oursize);
5950 
5951 	return (err);
5952 }
5953 
5954 static boolean_t
label_cksum_valid(vdev_label_t * label,uint64_t offset)5955 label_cksum_valid(vdev_label_t *label, uint64_t offset)
5956 {
5957 	zio_checksum_info_t *ci = &zio_checksum_table[ZIO_CHECKSUM_LABEL];
5958 	zio_cksum_t expected_cksum;
5959 	zio_cksum_t actual_cksum;
5960 	zio_cksum_t verifier;
5961 	zio_eck_t *eck;
5962 	int byteswap;
5963 
5964 	void *data = (char *)label + offsetof(vdev_label_t, vl_vdev_phys);
5965 	eck = (zio_eck_t *)((char *)(data) + VDEV_PHYS_SIZE) - 1;
5966 
5967 	offset += offsetof(vdev_label_t, vl_vdev_phys);
5968 	ZIO_SET_CHECKSUM(&verifier, offset, 0, 0, 0);
5969 
5970 	byteswap = (eck->zec_magic == BSWAP_64(ZEC_MAGIC));
5971 	if (byteswap)
5972 		byteswap_uint64_array(&verifier, sizeof (zio_cksum_t));
5973 
5974 	expected_cksum = eck->zec_cksum;
5975 	eck->zec_cksum = verifier;
5976 
5977 	abd_t *abd = abd_get_from_buf(data, VDEV_PHYS_SIZE);
5978 	ci->ci_func[byteswap](abd, VDEV_PHYS_SIZE, NULL, &actual_cksum);
5979 	abd_free(abd);
5980 
5981 	if (byteswap)
5982 		byteswap_uint64_array(&expected_cksum, sizeof (zio_cksum_t));
5983 
5984 	if (ZIO_CHECKSUM_EQUAL(actual_cksum, expected_cksum))
5985 		return (B_TRUE);
5986 
5987 	return (B_FALSE);
5988 }
5989 
5990 static int
dump_label(const char * dev)5991 dump_label(const char *dev)
5992 {
5993 	char path[MAXPATHLEN];
5994 	zdb_label_t labels[VDEV_LABELS] = {{{{0}}}};
5995 	uint64_t psize, ashift, l2cache;
5996 	struct stat64 statbuf;
5997 	boolean_t config_found = B_FALSE;
5998 	boolean_t error = B_FALSE;
5999 	boolean_t read_l2arc_header = B_FALSE;
6000 	avl_tree_t config_tree;
6001 	avl_tree_t uberblock_tree;
6002 	void *node, *cookie;
6003 	int fd;
6004 
6005 	/*
6006 	 * Check if we were given absolute path and use it as is.
6007 	 * Otherwise if the provided vdev name doesn't point to a file,
6008 	 * try prepending expected disk paths and partition numbers.
6009 	 */
6010 	(void) strlcpy(path, dev, sizeof (path));
6011 	if (dev[0] != '/' && stat64(path, &statbuf) != 0) {
6012 		int error;
6013 
6014 		error = zfs_resolve_shortname(dev, path, MAXPATHLEN);
6015 		if (error == 0 && zfs_dev_is_whole_disk(path)) {
6016 			if (zfs_append_partition(path, MAXPATHLEN) == -1)
6017 				error = ENOENT;
6018 		}
6019 
6020 		if (error || (stat64(path, &statbuf) != 0)) {
6021 			(void) printf("failed to find device %s, try "
6022 			    "specifying absolute path instead\n", dev);
6023 			return (1);
6024 		}
6025 	}
6026 
6027 	if ((fd = open64(path, O_RDONLY)) < 0) {
6028 		(void) printf("cannot open '%s': %s\n", path, strerror(errno));
6029 		zdb_exit(1);
6030 	}
6031 
6032 	if (fstat64_blk(fd, &statbuf) != 0) {
6033 		(void) printf("failed to stat '%s': %s\n", path,
6034 		    strerror(errno));
6035 		(void) close(fd);
6036 		zdb_exit(1);
6037 	}
6038 
6039 	if (S_ISBLK(statbuf.st_mode) && zfs_dev_flush(fd) != 0)
6040 		(void) printf("failed to invalidate cache '%s' : %s\n", path,
6041 		    strerror(errno));
6042 
6043 	avl_create(&config_tree, cksum_record_compare,
6044 	    sizeof (cksum_record_t), offsetof(cksum_record_t, link));
6045 	avl_create(&uberblock_tree, cksum_record_compare,
6046 	    sizeof (cksum_record_t), offsetof(cksum_record_t, link));
6047 
6048 	psize = statbuf.st_size;
6049 	psize = P2ALIGN_TYPED(psize, sizeof (vdev_label_t), uint64_t);
6050 	ashift = SPA_MINBLOCKSHIFT;
6051 
6052 	/*
6053 	 * 1. Read the label from disk
6054 	 * 2. Verify label cksum
6055 	 * 3. Unpack the configuration and insert in config tree.
6056 	 * 4. Traverse all uberblocks and insert in uberblock tree.
6057 	 */
6058 	for (int l = 0; l < VDEV_LABELS; l++) {
6059 		zdb_label_t *label = &labels[l];
6060 		char *buf = label->label.vl_vdev_phys.vp_nvlist;
6061 		size_t buflen = sizeof (label->label.vl_vdev_phys.vp_nvlist);
6062 		nvlist_t *config;
6063 		cksum_record_t *rec;
6064 		zio_cksum_t cksum;
6065 		vdev_t vd;
6066 
6067 		label->label_offset = vdev_label_offset(psize, l, 0);
6068 
6069 		if (pread64(fd, &label->label, sizeof (label->label),
6070 		    label->label_offset) != sizeof (label->label)) {
6071 			if (!dump_opt['q'])
6072 				(void) printf("failed to read label %d\n", l);
6073 			label->read_failed = B_TRUE;
6074 			error = B_TRUE;
6075 			continue;
6076 		}
6077 
6078 		label->read_failed = B_FALSE;
6079 		label->cksum_valid = label_cksum_valid(&label->label,
6080 		    label->label_offset);
6081 
6082 		if (nvlist_unpack(buf, buflen, &config, 0) == 0) {
6083 			nvlist_t *vdev_tree = NULL;
6084 			size_t size;
6085 
6086 			if ((nvlist_lookup_nvlist(config,
6087 			    ZPOOL_CONFIG_VDEV_TREE, &vdev_tree) != 0) ||
6088 			    (nvlist_lookup_uint64(vdev_tree,
6089 			    ZPOOL_CONFIG_ASHIFT, &ashift) != 0))
6090 				ashift = SPA_MINBLOCKSHIFT;
6091 
6092 			if (nvlist_size(config, &size, NV_ENCODE_XDR) != 0)
6093 				size = buflen;
6094 
6095 			/* If the device is a cache device read the header. */
6096 			if (!read_l2arc_header) {
6097 				if (nvlist_lookup_uint64(config,
6098 				    ZPOOL_CONFIG_POOL_STATE, &l2cache) == 0 &&
6099 				    l2cache == POOL_STATE_L2CACHE) {
6100 					read_l2arc_header = B_TRUE;
6101 				}
6102 			}
6103 
6104 			fletcher_4_native_varsize(buf, size, &cksum);
6105 			rec = cksum_record_insert(&config_tree, &cksum, l);
6106 
6107 			label->config = rec;
6108 			label->config_nv = config;
6109 			config_found = B_TRUE;
6110 		} else {
6111 			error = B_TRUE;
6112 		}
6113 
6114 		vd.vdev_ashift = ashift;
6115 		vd.vdev_top = &vd;
6116 
6117 		for (int i = 0; i < VDEV_UBERBLOCK_COUNT(&vd); i++) {
6118 			uint64_t uoff = VDEV_UBERBLOCK_OFFSET(&vd, i);
6119 			uberblock_t *ub = (void *)((char *)label + uoff);
6120 
6121 			if (uberblock_verify(ub))
6122 				continue;
6123 
6124 			fletcher_4_native_varsize(ub, sizeof (*ub), &cksum);
6125 			rec = cksum_record_insert(&uberblock_tree, &cksum, l);
6126 
6127 			label->uberblocks[i] = rec;
6128 		}
6129 	}
6130 
6131 	/*
6132 	 * Dump the label and uberblocks.
6133 	 */
6134 	for (int l = 0; l < VDEV_LABELS; l++) {
6135 		zdb_label_t *label = &labels[l];
6136 		size_t buflen = sizeof (label->label.vl_vdev_phys.vp_nvlist);
6137 
6138 		if (label->read_failed == B_TRUE)
6139 			continue;
6140 
6141 		if (label->config_nv) {
6142 			dump_config_from_label(label, buflen, l);
6143 		} else {
6144 			if (!dump_opt['q'])
6145 				(void) printf("failed to unpack label %d\n", l);
6146 		}
6147 
6148 		if (dump_opt['u'])
6149 			dump_label_uberblocks(label, ashift, l);
6150 
6151 		nvlist_free(label->config_nv);
6152 	}
6153 
6154 	/*
6155 	 * Dump the L2ARC header, if existent.
6156 	 */
6157 	if (read_l2arc_header)
6158 		error |= dump_l2arc_header(fd);
6159 
6160 	cookie = NULL;
6161 	while ((node = avl_destroy_nodes(&config_tree, &cookie)) != NULL)
6162 		umem_free(node, sizeof (cksum_record_t));
6163 
6164 	cookie = NULL;
6165 	while ((node = avl_destroy_nodes(&uberblock_tree, &cookie)) != NULL)
6166 		umem_free(node, sizeof (cksum_record_t));
6167 
6168 	avl_destroy(&config_tree);
6169 	avl_destroy(&uberblock_tree);
6170 
6171 	(void) close(fd);
6172 
6173 	return (config_found == B_FALSE ? 2 :
6174 	    (error == B_TRUE ? 1 : 0));
6175 }
6176 
6177 static uint64_t dataset_feature_count[SPA_FEATURES];
6178 static uint64_t global_feature_count[SPA_FEATURES];
6179 static uint64_t remap_deadlist_count = 0;
6180 
6181 static int
dump_one_objset(const char * dsname,void * arg)6182 dump_one_objset(const char *dsname, void *arg)
6183 {
6184 	(void) arg;
6185 	int error;
6186 	objset_t *os;
6187 	spa_feature_t f;
6188 
6189 	error = open_objset(dsname, FTAG, &os);
6190 	if (error != 0)
6191 		return (0);
6192 
6193 	for (f = 0; f < SPA_FEATURES; f++) {
6194 		if (!dsl_dataset_feature_is_active(dmu_objset_ds(os), f))
6195 			continue;
6196 		ASSERT(spa_feature_table[f].fi_flags &
6197 		    ZFEATURE_FLAG_PER_DATASET);
6198 		dataset_feature_count[f]++;
6199 	}
6200 
6201 	if (dsl_dataset_remap_deadlist_exists(dmu_objset_ds(os))) {
6202 		remap_deadlist_count++;
6203 	}
6204 
6205 	for (dsl_bookmark_node_t *dbn =
6206 	    avl_first(&dmu_objset_ds(os)->ds_bookmarks); dbn != NULL;
6207 	    dbn = AVL_NEXT(&dmu_objset_ds(os)->ds_bookmarks, dbn)) {
6208 		mos_obj_refd(dbn->dbn_phys.zbm_redaction_obj);
6209 		if (dbn->dbn_phys.zbm_redaction_obj != 0) {
6210 			global_feature_count[
6211 			    SPA_FEATURE_REDACTION_BOOKMARKS]++;
6212 			objset_t *mos = os->os_spa->spa_meta_objset;
6213 			dnode_t *rl;
6214 			VERIFY0(dnode_hold(mos,
6215 			    dbn->dbn_phys.zbm_redaction_obj, FTAG, &rl));
6216 			if (rl->dn_have_spill) {
6217 				global_feature_count[
6218 				    SPA_FEATURE_REDACTION_LIST_SPILL]++;
6219 			}
6220 		}
6221 		if (dbn->dbn_phys.zbm_flags & ZBM_FLAG_HAS_FBN)
6222 			global_feature_count[SPA_FEATURE_BOOKMARK_WRITTEN]++;
6223 	}
6224 
6225 	if (dsl_deadlist_is_open(&dmu_objset_ds(os)->ds_dir->dd_livelist) &&
6226 	    !dmu_objset_is_snapshot(os)) {
6227 		global_feature_count[SPA_FEATURE_LIVELIST]++;
6228 	}
6229 	dump_objset(os);
6230 	close_objset(os, FTAG);
6231 	fuid_table_destroy();
6232 	return (0);
6233 }
6234 
6235 /*
6236  * Block statistics.
6237  */
6238 #define	PSIZE_HISTO_SIZE (SPA_OLD_MAXBLOCKSIZE / SPA_MINBLOCKSIZE + 2)
6239 typedef struct zdb_blkstats {
6240 	uint64_t zb_asize;
6241 	uint64_t zb_lsize;
6242 	uint64_t zb_psize;
6243 	uint64_t zb_count;
6244 	uint64_t zb_gangs;
6245 	uint64_t zb_ditto_samevdev;
6246 	uint64_t zb_ditto_same_ms;
6247 	uint64_t zb_psize_histogram[PSIZE_HISTO_SIZE];
6248 } zdb_blkstats_t;
6249 
6250 /*
6251  * Extended object types to report deferred frees and dedup auto-ditto blocks.
6252  */
6253 #define	ZDB_OT_DEFERRED	(DMU_OT_NUMTYPES + 0)
6254 #define	ZDB_OT_DITTO	(DMU_OT_NUMTYPES + 1)
6255 #define	ZDB_OT_OTHER	(DMU_OT_NUMTYPES + 2)
6256 #define	ZDB_OT_TOTAL	(DMU_OT_NUMTYPES + 3)
6257 
6258 static const char *zdb_ot_extname[] = {
6259 	"deferred free",
6260 	"dedup ditto",
6261 	"other",
6262 	"Total",
6263 };
6264 
6265 #define	ZB_TOTAL	DN_MAX_LEVELS
6266 #define	SPA_MAX_FOR_16M	(SPA_MAXBLOCKSHIFT+1)
6267 
6268 typedef struct zdb_brt_entry {
6269 	dva_t		zbre_dva;
6270 	uint64_t	zbre_refcount;
6271 	avl_node_t	zbre_node;
6272 } zdb_brt_entry_t;
6273 
6274 typedef struct zdb_cb {
6275 	zdb_blkstats_t	zcb_type[ZB_TOTAL + 1][ZDB_OT_TOTAL + 1];
6276 	uint64_t	zcb_removing_size;
6277 	uint64_t	zcb_checkpoint_size;
6278 	uint64_t	zcb_dedup_asize;
6279 	uint64_t	zcb_dedup_blocks;
6280 	uint64_t	zcb_clone_asize;
6281 	uint64_t	zcb_clone_blocks;
6282 	uint64_t	zcb_psize_count[SPA_MAX_FOR_16M];
6283 	uint64_t	zcb_lsize_count[SPA_MAX_FOR_16M];
6284 	uint64_t	zcb_asize_count[SPA_MAX_FOR_16M];
6285 	uint64_t	zcb_psize_len[SPA_MAX_FOR_16M];
6286 	uint64_t	zcb_lsize_len[SPA_MAX_FOR_16M];
6287 	uint64_t	zcb_asize_len[SPA_MAX_FOR_16M];
6288 	uint64_t	zcb_psize_total;
6289 	uint64_t	zcb_lsize_total;
6290 	uint64_t	zcb_asize_total;
6291 	uint64_t	zcb_embedded_blocks[NUM_BP_EMBEDDED_TYPES];
6292 	uint64_t	zcb_embedded_histogram[NUM_BP_EMBEDDED_TYPES]
6293 	    [BPE_PAYLOAD_SIZE + 1];
6294 	uint64_t	zcb_start;
6295 	hrtime_t	zcb_lastprint;
6296 	uint64_t	zcb_totalasize;
6297 	uint64_t	zcb_errors[256];
6298 	int		zcb_readfails;
6299 	int		zcb_haderrors;
6300 	spa_t		*zcb_spa;
6301 	uint32_t	**zcb_vd_obsolete_counts;
6302 	avl_tree_t	zcb_brt;
6303 	boolean_t	zcb_brt_is_active;
6304 } zdb_cb_t;
6305 
6306 /* test if two DVA offsets from same vdev are within the same metaslab */
6307 static boolean_t
same_metaslab(spa_t * spa,uint64_t vdev,uint64_t off1,uint64_t off2)6308 same_metaslab(spa_t *spa, uint64_t vdev, uint64_t off1, uint64_t off2)
6309 {
6310 	vdev_t *vd = vdev_lookup_top(spa, vdev);
6311 	uint64_t ms_shift = vd->vdev_ms_shift;
6312 
6313 	return ((off1 >> ms_shift) == (off2 >> ms_shift));
6314 }
6315 
6316 /*
6317  * Used to simplify reporting of the histogram data.
6318  */
6319 typedef struct one_histo {
6320 	const char *name;
6321 	uint64_t *count;
6322 	uint64_t *len;
6323 	uint64_t cumulative;
6324 } one_histo_t;
6325 
6326 /*
6327  * The number of separate histograms processed for psize, lsize and asize.
6328  */
6329 #define	NUM_HISTO 3
6330 
6331 /*
6332  * This routine will create a fixed column size output of three different
6333  * histograms showing by blocksize of 512 - 2^ SPA_MAX_FOR_16M
6334  * the count, length and cumulative length of the psize, lsize and
6335  * asize blocks.
6336  *
6337  * All three types of blocks are listed on a single line
6338  *
6339  * By default the table is printed in nicenumber format (e.g. 123K) but
6340  * if the '-P' parameter is specified then the full raw number (parseable)
6341  * is printed out.
6342  */
6343 static void
dump_size_histograms(zdb_cb_t * zcb)6344 dump_size_histograms(zdb_cb_t *zcb)
6345 {
6346 	/*
6347 	 * A temporary buffer that allows us to convert a number into
6348 	 * a string using zdb_nicenumber to allow either raw or human
6349 	 * readable numbers to be output.
6350 	 */
6351 	char numbuf[32];
6352 
6353 	/*
6354 	 * Define titles which are used in the headers of the tables
6355 	 * printed by this routine.
6356 	 */
6357 	const char blocksize_title1[] = "block";
6358 	const char blocksize_title2[] = "size";
6359 	const char count_title[] = "Count";
6360 	const char length_title[] = "Size";
6361 	const char cumulative_title[] = "Cum.";
6362 
6363 	/*
6364 	 * Setup the histogram arrays (psize, lsize, and asize).
6365 	 */
6366 	one_histo_t parm_histo[NUM_HISTO];
6367 
6368 	parm_histo[0].name = "psize";
6369 	parm_histo[0].count = zcb->zcb_psize_count;
6370 	parm_histo[0].len = zcb->zcb_psize_len;
6371 	parm_histo[0].cumulative = 0;
6372 
6373 	parm_histo[1].name = "lsize";
6374 	parm_histo[1].count = zcb->zcb_lsize_count;
6375 	parm_histo[1].len = zcb->zcb_lsize_len;
6376 	parm_histo[1].cumulative = 0;
6377 
6378 	parm_histo[2].name = "asize";
6379 	parm_histo[2].count = zcb->zcb_asize_count;
6380 	parm_histo[2].len = zcb->zcb_asize_len;
6381 	parm_histo[2].cumulative = 0;
6382 
6383 
6384 	(void) printf("\nBlock Size Histogram\n");
6385 	switch (block_bin_mode) {
6386 	case BIN_PSIZE:
6387 		printf("(note: all categories are binned by %s)\n", "psize");
6388 		break;
6389 	case BIN_LSIZE:
6390 		printf("(note: all categories are binned by %s)\n", "lsize");
6391 		break;
6392 	case BIN_ASIZE:
6393 		printf("(note: all categories are binned by %s)\n", "asize");
6394 		break;
6395 	default:
6396 		printf("(note: all categories are binned separately)\n");
6397 		break;
6398 	}
6399 	if (block_classes != 0) {
6400 		char buf[256] = "";
6401 		if (block_classes & CLASS_NORMAL)
6402 			strlcat(buf, "\"normal\", ", sizeof (buf));
6403 		if (block_classes & CLASS_SPECIAL)
6404 			strlcat(buf, "\"special\", ", sizeof (buf));
6405 		if (block_classes & CLASS_DEDUP)
6406 			strlcat(buf, "\"dedup\", ", sizeof (buf));
6407 		if (block_classes & CLASS_OTHER)
6408 			strlcat(buf, "\"other\", ", sizeof (buf));
6409 		buf[strlen(buf)-2] = '\0';
6410 		printf("(note: only blocks in these classes are counted: %s)\n",
6411 		    buf);
6412 	}
6413 	/*
6414 	 * Print the first line titles
6415 	 */
6416 	if (dump_opt['P'])
6417 		(void) printf("\n%s\t", blocksize_title1);
6418 	else
6419 		(void) printf("\n%7s   ", blocksize_title1);
6420 
6421 	for (int j = 0; j < NUM_HISTO; j++) {
6422 		if (dump_opt['P']) {
6423 			if (j < NUM_HISTO - 1) {
6424 				(void) printf("%s\t\t\t", parm_histo[j].name);
6425 			} else {
6426 				/* Don't print trailing spaces */
6427 				(void) printf("  %s", parm_histo[j].name);
6428 			}
6429 		} else {
6430 			if (j < NUM_HISTO - 1) {
6431 				/* Left aligned strings in the output */
6432 				(void) printf("%-7s              ",
6433 				    parm_histo[j].name);
6434 			} else {
6435 				/* Don't print trailing spaces */
6436 				(void) printf("%s", parm_histo[j].name);
6437 			}
6438 		}
6439 	}
6440 	(void) printf("\n");
6441 
6442 	/*
6443 	 * Print the second line titles
6444 	 */
6445 	if (dump_opt['P']) {
6446 		(void) printf("%s\t", blocksize_title2);
6447 	} else {
6448 		(void) printf("%7s ", blocksize_title2);
6449 	}
6450 
6451 	for (int i = 0; i < NUM_HISTO; i++) {
6452 		if (dump_opt['P']) {
6453 			(void) printf("%s\t%s\t%s\t",
6454 			    count_title, length_title, cumulative_title);
6455 		} else {
6456 			(void) printf("%7s%7s%7s",
6457 			    count_title, length_title, cumulative_title);
6458 		}
6459 	}
6460 	(void) printf("\n");
6461 
6462 	/*
6463 	 * Print the rows
6464 	 */
6465 	for (int i = SPA_MINBLOCKSHIFT; i < SPA_MAX_FOR_16M; i++) {
6466 
6467 		/*
6468 		 * Print the first column showing the blocksize
6469 		 */
6470 		zdb_nicenum((1ULL << i), numbuf, sizeof (numbuf));
6471 
6472 		if (dump_opt['P']) {
6473 			printf("%s", numbuf);
6474 		} else {
6475 			printf("%7s:", numbuf);
6476 		}
6477 
6478 		/*
6479 		 * Print the remaining set of 3 columns per size:
6480 		 * for psize, lsize and asize
6481 		 */
6482 		for (int j = 0; j < NUM_HISTO; j++) {
6483 			parm_histo[j].cumulative += parm_histo[j].len[i];
6484 
6485 			zdb_nicenum(parm_histo[j].count[i],
6486 			    numbuf, sizeof (numbuf));
6487 			if (dump_opt['P'])
6488 				(void) printf("\t%s", numbuf);
6489 			else
6490 				(void) printf("%7s", numbuf);
6491 
6492 			zdb_nicenum(parm_histo[j].len[i],
6493 			    numbuf, sizeof (numbuf));
6494 			if (dump_opt['P'])
6495 				(void) printf("\t%s", numbuf);
6496 			else
6497 				(void) printf("%7s", numbuf);
6498 
6499 			zdb_nicenum(parm_histo[j].cumulative,
6500 			    numbuf, sizeof (numbuf));
6501 			if (dump_opt['P'])
6502 				(void) printf("\t%s", numbuf);
6503 			else
6504 				(void) printf("%7s", numbuf);
6505 		}
6506 		(void) printf("\n");
6507 	}
6508 }
6509 
6510 static void
zdb_count_block(zdb_cb_t * zcb,zilog_t * zilog,const blkptr_t * bp,dmu_object_type_t type)6511 zdb_count_block(zdb_cb_t *zcb, zilog_t *zilog, const blkptr_t *bp,
6512     dmu_object_type_t type)
6513 {
6514 	int i;
6515 	boolean_t claimed = B_FALSE;
6516 	boolean_t ddt_block = B_FALSE;
6517 	boolean_t brt_block = B_FALSE;
6518 
6519 	ASSERT(type < ZDB_OT_TOTAL);
6520 
6521 	if (zilog && zil_bp_tree_add(zilog, bp) != 0)
6522 		return;
6523 
6524 	spa_config_enter(zcb->zcb_spa, SCL_CONFIG, FTAG, RW_READER);
6525 
6526 	blkptr_t tempbp;
6527 	if (BP_GET_DEDUP(bp)) {
6528 		/*
6529 		 * Dedup'd blocks are special. We need to count them, so we can
6530 		 * later uncount them when reporting leaked space, and we must
6531 		 * only claim them once.
6532 		 *
6533 		 * We use the existing dedup system to track what we've seen.
6534 		 * The first time we see a block, we do a ddt_lookup() to see
6535 		 * if it exists in the DDT. If we're doing leak tracking, we
6536 		 * claim the block at this time.
6537 		 *
6538 		 * Each time we see a block, we reduce the refcount in the
6539 		 * entry by one, and add to the size and count of dedup'd
6540 		 * blocks to report at the end.
6541 		 */
6542 
6543 		ddt_t *ddt = ddt_select(zcb->zcb_spa, bp);
6544 
6545 		ddt_enter(ddt);
6546 
6547 		/*
6548 		 * Find the block. This will create the entry in memory, but
6549 		 * we'll know if that happened by its refcount.
6550 		 */
6551 		ddt_entry_t *dde = ddt_lookup(ddt, bp, B_TRUE);
6552 
6553 		/*
6554 		 * ddt_lookup() can return NULL when unique entries are pruned
6555 		 * from the DDT.
6556 		 */
6557 		if (dde == NULL) {
6558 			ddt_exit(ddt);
6559 			goto ddt_done;
6560 		}
6561 
6562 		/* Get the phys for this variant */
6563 		ddt_phys_variant_t v = ddt_phys_select(ddt, dde, bp);
6564 
6565 		/*
6566 		 * DDT_PHYS_NONE means the block has the dedup bit set but
6567 		 * its DVA doesn't match any phys in the entry.  This can
6568 		 * happen when a DVA was evicted from the DDT and re-added
6569 		 * on a hash collision.  The block may still have a BRT entry.
6570 		 */
6571 		if (v == DDT_PHYS_NONE) {
6572 			ddt_exit(ddt);
6573 			goto ddt_done;
6574 		}
6575 
6576 		ddt_block = B_TRUE;
6577 
6578 		/*
6579 		 * This entry may have multiple sets of DVAs. We must claim
6580 		 * each set the first time we see them in a real block on disk,
6581 		 * or count them on subsequent occurences. We don't have a
6582 		 * convenient way to track the first time we see each variant,
6583 		 * so we repurpose dde_io as a set of "seen" flag bits. We can
6584 		 * do this safely in zdb because it never writes, so it will
6585 		 * never have a writing zio for this block in that pointer.
6586 		 */
6587 		boolean_t seen = !!(((uintptr_t)dde->dde_io) & (1 << v));
6588 		if (!seen)
6589 			dde->dde_io =
6590 			    (void *)(((uintptr_t)dde->dde_io) | (1 << v));
6591 
6592 		/*
6593 		 * Consume a reference.  If this variant's refcount is already
6594 		 * zero, the DDT tracking is exhausted — more filesystem
6595 		 * references exist than the DDT accounts for.
6596 		 */
6597 		boolean_t ddt_refcnt_exhausted =
6598 		    (ddt_phys_refcnt(dde->dde_phys, v) == 0);
6599 		if (!ddt_refcnt_exhausted)
6600 			ddt_phys_decref(dde->dde_phys, v);
6601 
6602 		/*
6603 		 * If this entry has a single flat phys, it may have been
6604 		 * extended with additional DVAs at some time in its life.
6605 		 * This block might be from before it was fully extended, and
6606 		 * so have fewer DVAs.
6607 		 *
6608 		 * If this is the first time we've seen this block, and we
6609 		 * claimed it as-is, then we would miss the claim on some
6610 		 * number of DVAs, which would then be seen as leaked.
6611 		 *
6612 		 * In all cases, if we've had fewer DVAs, then the asize would
6613 		 * be too small, and would lead to the pool apparently using
6614 		 * more space than allocated.
6615 		 *
6616 		 * To handle this, we copy the canonical set of DVAs from the
6617 		 * entry back to the block pointer before we claim it.
6618 		 */
6619 		if (v == DDT_PHYS_FLAT) {
6620 			ASSERT3U(BP_GET_PHYSICAL_BIRTH(bp), ==,
6621 			    ddt_phys_birth(dde->dde_phys, v));
6622 			tempbp = *bp;
6623 			ddt_bp_fill(dde->dde_phys, v, &tempbp,
6624 			    BP_GET_PHYSICAL_BIRTH(bp));
6625 			bp = &tempbp;
6626 		}
6627 
6628 		if (seen && !ddt_refcnt_exhausted) {
6629 			/*
6630 			 * The second or later time we see this block,
6631 			 * it's a duplicate and we count it.
6632 			 */
6633 			zcb->zcb_dedup_asize += BP_GET_ASIZE(bp);
6634 			zcb->zcb_dedup_blocks++;
6635 			claimed = B_TRUE;
6636 		}
6637 
6638 		ddt_exit(ddt);
6639 	}
6640 
6641 ddt_done:
6642 	if (!claimed && zcb->zcb_brt_is_active &&
6643 	    brt_maybe_exists(zcb->zcb_spa, bp)) {
6644 		/*
6645 		 * Cloned blocks are special. We need to count them, so we can
6646 		 * later uncount them when reporting leaked space, and we must
6647 		 * only claim them once.
6648 		 *
6649 		 * To do this, we keep our own in-memory BRT. For each block
6650 		 * we haven't seen before, we look it up in the real BRT. If
6651 		 * we see the block again, we count it as a clone.
6652 		 */
6653 		zdb_brt_entry_t zbre_search, *zbre;
6654 		avl_index_t where;
6655 
6656 		zbre_search.zbre_dva = bp->blk_dva[0];
6657 		zbre = avl_find(&zcb->zcb_brt, &zbre_search, &where);
6658 		if (zbre == NULL) {
6659 			uint64_t refcnt =
6660 			    brt_entry_get_refcount(zcb->zcb_spa, bp);
6661 			if (refcnt > 0) {
6662 				brt_block = B_TRUE;
6663 				zbre = umem_zalloc(sizeof (zdb_brt_entry_t),
6664 				    UMEM_NOFAIL);
6665 				zbre->zbre_dva = bp->blk_dva[0];
6666 				zbre->zbre_refcount = refcnt;
6667 				avl_insert(&zcb->zcb_brt, zbre, where);
6668 			}
6669 		} else {
6670 			brt_block = B_TRUE;
6671 			if (zbre->zbre_refcount > 0) {
6672 				zcb->zcb_clone_asize += BP_GET_ASIZE(bp);
6673 				zcb->zcb_clone_blocks++;
6674 				zbre->zbre_refcount--;
6675 				claimed = B_TRUE;
6676 			}
6677 		}
6678 	}
6679 
6680 	for (i = 0; i < 4; i++) {
6681 		int l = (i < 2) ? BP_GET_LEVEL(bp) : ZB_TOTAL;
6682 		int t = (i & 1) ? type : ZDB_OT_TOTAL;
6683 		int equal;
6684 		zdb_blkstats_t *zb = &zcb->zcb_type[l][t];
6685 
6686 		zb->zb_asize += BP_GET_ASIZE(bp);
6687 		zb->zb_lsize += BP_GET_LSIZE(bp);
6688 		zb->zb_psize += BP_GET_PSIZE(bp);
6689 		zb->zb_count++;
6690 
6691 		/*
6692 		 * The histogram is only big enough to record blocks up to
6693 		 * SPA_OLD_MAXBLOCKSIZE; larger blocks go into the last,
6694 		 * "other", bucket.
6695 		 */
6696 		unsigned idx = BP_GET_PSIZE(bp) >> SPA_MINBLOCKSHIFT;
6697 		idx = MIN(idx, SPA_OLD_MAXBLOCKSIZE / SPA_MINBLOCKSIZE + 1);
6698 		zb->zb_psize_histogram[idx]++;
6699 
6700 		zb->zb_gangs += BP_COUNT_GANG(bp);
6701 
6702 		switch (BP_GET_NDVAS(bp)) {
6703 		case 2:
6704 			if (DVA_GET_VDEV(&bp->blk_dva[0]) ==
6705 			    DVA_GET_VDEV(&bp->blk_dva[1])) {
6706 				zb->zb_ditto_samevdev++;
6707 
6708 				if (same_metaslab(zcb->zcb_spa,
6709 				    DVA_GET_VDEV(&bp->blk_dva[0]),
6710 				    DVA_GET_OFFSET(&bp->blk_dva[0]),
6711 				    DVA_GET_OFFSET(&bp->blk_dva[1])))
6712 					zb->zb_ditto_same_ms++;
6713 			}
6714 			break;
6715 		case 3:
6716 			equal = (DVA_GET_VDEV(&bp->blk_dva[0]) ==
6717 			    DVA_GET_VDEV(&bp->blk_dva[1])) +
6718 			    (DVA_GET_VDEV(&bp->blk_dva[0]) ==
6719 			    DVA_GET_VDEV(&bp->blk_dva[2])) +
6720 			    (DVA_GET_VDEV(&bp->blk_dva[1]) ==
6721 			    DVA_GET_VDEV(&bp->blk_dva[2]));
6722 			if (equal != 0) {
6723 				zb->zb_ditto_samevdev++;
6724 
6725 				if (DVA_GET_VDEV(&bp->blk_dva[0]) ==
6726 				    DVA_GET_VDEV(&bp->blk_dva[1]) &&
6727 				    same_metaslab(zcb->zcb_spa,
6728 				    DVA_GET_VDEV(&bp->blk_dva[0]),
6729 				    DVA_GET_OFFSET(&bp->blk_dva[0]),
6730 				    DVA_GET_OFFSET(&bp->blk_dva[1])))
6731 					zb->zb_ditto_same_ms++;
6732 				else if (DVA_GET_VDEV(&bp->blk_dva[0]) ==
6733 				    DVA_GET_VDEV(&bp->blk_dva[2]) &&
6734 				    same_metaslab(zcb->zcb_spa,
6735 				    DVA_GET_VDEV(&bp->blk_dva[0]),
6736 				    DVA_GET_OFFSET(&bp->blk_dva[0]),
6737 				    DVA_GET_OFFSET(&bp->blk_dva[2])))
6738 					zb->zb_ditto_same_ms++;
6739 				else if (DVA_GET_VDEV(&bp->blk_dva[1]) ==
6740 				    DVA_GET_VDEV(&bp->blk_dva[2]) &&
6741 				    same_metaslab(zcb->zcb_spa,
6742 				    DVA_GET_VDEV(&bp->blk_dva[1]),
6743 				    DVA_GET_OFFSET(&bp->blk_dva[1]),
6744 				    DVA_GET_OFFSET(&bp->blk_dva[2])))
6745 					zb->zb_ditto_same_ms++;
6746 			}
6747 			break;
6748 		}
6749 	}
6750 
6751 	spa_config_exit(zcb->zcb_spa, SCL_CONFIG, FTAG);
6752 
6753 	if (BP_IS_EMBEDDED(bp)) {
6754 		zcb->zcb_embedded_blocks[BPE_GET_ETYPE(bp)]++;
6755 		zcb->zcb_embedded_histogram[BPE_GET_ETYPE(bp)]
6756 		    [BPE_GET_PSIZE(bp)]++;
6757 		return;
6758 	}
6759 
6760 	if (block_classes != 0) {
6761 		spa_config_enter(zcb->zcb_spa, SCL_CONFIG, FTAG, RW_READER);
6762 
6763 		uint64_t vdev = DVA_GET_VDEV(&bp->blk_dva[0]);
6764 		uint64_t offset = DVA_GET_OFFSET(&bp->blk_dva[0]);
6765 		vdev_t *vd = vdev_lookup_top(zcb->zcb_spa, vdev);
6766 		ASSERT(vd != NULL);
6767 		metaslab_t *ms = vd->vdev_ms[offset >> vd->vdev_ms_shift];
6768 		ASSERT(ms != NULL);
6769 		metaslab_group_t *mg = ms->ms_group;
6770 		ASSERT(mg != NULL);
6771 		metaslab_class_t *mc = mg->mg_class;
6772 		ASSERT(mc != NULL);
6773 
6774 		spa_config_exit(zcb->zcb_spa, SCL_CONFIG, FTAG);
6775 
6776 		int class;
6777 		if (mc == spa_normal_class(zcb->zcb_spa)) {
6778 			class = CLASS_NORMAL;
6779 		} else if (mc == spa_special_class(zcb->zcb_spa)) {
6780 			class = CLASS_SPECIAL;
6781 		} else if (mc == spa_dedup_class(zcb->zcb_spa)) {
6782 			class = CLASS_DEDUP;
6783 		} else {
6784 			class = CLASS_OTHER;
6785 		}
6786 
6787 		if (!(block_classes & class)) {
6788 			goto hist_skipped;
6789 		}
6790 	}
6791 
6792 	/*
6793 	 * The binning histogram bins by powers of two up to
6794 	 * SPA_MAXBLOCKSIZE rather than creating bins for
6795 	 * every possible blocksize found in the pool.
6796 	 */
6797 	int bin;
6798 
6799 	/*
6800 	 * Binning strategy: each bin includes blocks up to and including
6801 	 * the given size (excluding blocks that fit into the previous bin).
6802 	 * This way, the "4K" bin includes blocks within the (2K; 4K] range.
6803 	 */
6804 #define	BIN(size) (highbit64((size) - 1))
6805 
6806 	switch (block_bin_mode) {
6807 	case BIN_PSIZE: bin = BIN(BP_GET_PSIZE(bp)); break;
6808 	case BIN_LSIZE: bin = BIN(BP_GET_LSIZE(bp)); break;
6809 	case BIN_ASIZE: bin = BIN(BP_GET_ASIZE(bp)); break;
6810 	case BIN_AUTO: break;
6811 	default: PANIC("bad block_bin_mode"); abort();
6812 	}
6813 
6814 	if (block_bin_mode == BIN_AUTO)
6815 		bin = BIN(BP_GET_PSIZE(bp));
6816 
6817 	zcb->zcb_psize_count[bin]++;
6818 	zcb->zcb_psize_len[bin] += BP_GET_PSIZE(bp);
6819 	zcb->zcb_psize_total += BP_GET_PSIZE(bp);
6820 
6821 	if (block_bin_mode == BIN_AUTO)
6822 		bin = BIN(BP_GET_LSIZE(bp));
6823 
6824 	zcb->zcb_lsize_count[bin]++;
6825 	zcb->zcb_lsize_len[bin] += BP_GET_LSIZE(bp);
6826 	zcb->zcb_lsize_total += BP_GET_LSIZE(bp);
6827 
6828 	if (block_bin_mode == BIN_AUTO)
6829 		bin = BIN(BP_GET_ASIZE(bp));
6830 
6831 	zcb->zcb_asize_count[bin]++;
6832 	zcb->zcb_asize_len[bin] += BP_GET_ASIZE(bp);
6833 	zcb->zcb_asize_total += BP_GET_ASIZE(bp);
6834 
6835 #undef BIN
6836 
6837 hist_skipped:
6838 	if (claimed || dump_opt['L'])
6839 		return;
6840 
6841 	int claim_err = zio_wait(zio_claim(NULL, zcb->zcb_spa,
6842 	    spa_min_claim_txg(zcb->zcb_spa), bp, NULL, NULL,
6843 	    ZIO_FLAG_CANFAIL));
6844 	if (claim_err != 0) {
6845 		char blkbuf[BP_SPRINTF_LEN];
6846 		snprintf_blkptr(blkbuf, sizeof (blkbuf), bp);
6847 		(void) printf("block claim error %d%s%s: %s\n",
6848 		    claim_err, brt_block ? " (BRT)" : "",
6849 		    ddt_block ? " (DDT)" : "", blkbuf);
6850 		zcb->zcb_haderrors = 1;
6851 		zcb->zcb_errors[claim_err]++;
6852 	}
6853 }
6854 
6855 static void
zdb_blkptr_done(zio_t * zio)6856 zdb_blkptr_done(zio_t *zio)
6857 {
6858 	spa_t *spa = zio->io_spa;
6859 	blkptr_t *bp = zio->io_bp;
6860 	int ioerr = zio->io_error;
6861 	zdb_cb_t *zcb = zio->io_private;
6862 	zbookmark_phys_t *zb = &zio->io_bookmark;
6863 
6864 	mutex_enter(&spa->spa_scrub_lock);
6865 	spa->spa_load_verify_bytes -= BP_GET_PSIZE(bp);
6866 	cv_broadcast(&spa->spa_scrub_io_cv);
6867 
6868 	if (ioerr && !(zio->io_flags & ZIO_FLAG_SPECULATIVE)) {
6869 		char blkbuf[BP_SPRINTF_LEN];
6870 
6871 		zcb->zcb_haderrors = 1;
6872 		zcb->zcb_errors[ioerr]++;
6873 
6874 		if (dump_opt['b'] >= 2)
6875 			snprintf_blkptr(blkbuf, sizeof (blkbuf), bp);
6876 		else
6877 			blkbuf[0] = '\0';
6878 
6879 		(void) printf("zdb_blkptr_cb: "
6880 		    "Got error %d reading "
6881 		    "<%llu, %llu, %lld, %llx> %s -- skipping\n",
6882 		    ioerr,
6883 		    (u_longlong_t)zb->zb_objset,
6884 		    (u_longlong_t)zb->zb_object,
6885 		    (u_longlong_t)zb->zb_level,
6886 		    (u_longlong_t)zb->zb_blkid,
6887 		    blkbuf);
6888 	}
6889 	mutex_exit(&spa->spa_scrub_lock);
6890 
6891 	abd_free(zio->io_abd);
6892 }
6893 
6894 static int
zdb_blkptr_cb(spa_t * spa,zilog_t * zilog,const blkptr_t * bp,const zbookmark_phys_t * zb,const dnode_phys_t * dnp,void * arg)6895 zdb_blkptr_cb(spa_t *spa, zilog_t *zilog, const blkptr_t *bp,
6896     const zbookmark_phys_t *zb, const dnode_phys_t *dnp, void *arg)
6897 {
6898 	zdb_cb_t *zcb = arg;
6899 	dmu_object_type_t type;
6900 	boolean_t is_metadata;
6901 
6902 	if (zb->zb_level == ZB_DNODE_LEVEL)
6903 		return (0);
6904 
6905 	if (dump_opt['b'] >= 5 && BP_GET_BIRTH(bp) > 0) {
6906 		char blkbuf[BP_SPRINTF_LEN];
6907 		snprintf_blkptr(blkbuf, sizeof (blkbuf), bp);
6908 		(void) printf("objset %llu object %llu "
6909 		    "level %lld offset 0x%llx %s\n",
6910 		    (u_longlong_t)zb->zb_objset,
6911 		    (u_longlong_t)zb->zb_object,
6912 		    (longlong_t)zb->zb_level,
6913 		    (u_longlong_t)blkid2offset(dnp, bp, zb),
6914 		    blkbuf);
6915 	}
6916 
6917 	if (BP_IS_HOLE(bp) || BP_IS_REDACTED(bp))
6918 		return (0);
6919 
6920 	type = BP_GET_TYPE(bp);
6921 
6922 	zdb_count_block(zcb, zilog, bp,
6923 	    (type & DMU_OT_NEWTYPE) ? ZDB_OT_OTHER : type);
6924 
6925 	is_metadata = (BP_GET_LEVEL(bp) != 0 || DMU_OT_IS_METADATA(type));
6926 
6927 	if (!BP_IS_EMBEDDED(bp) &&
6928 	    (dump_opt['c'] > 1 || (dump_opt['c'] && is_metadata))) {
6929 		size_t size = BP_GET_PSIZE(bp);
6930 		abd_t *abd = abd_alloc(size, B_FALSE);
6931 		int flags = ZIO_FLAG_CANFAIL | ZIO_FLAG_SCRUB | ZIO_FLAG_RAW;
6932 
6933 		/* If it's an intent log block, failure is expected. */
6934 		if (zb->zb_level == ZB_ZIL_LEVEL)
6935 			flags |= ZIO_FLAG_SPECULATIVE;
6936 
6937 		mutex_enter(&spa->spa_scrub_lock);
6938 		while (spa->spa_load_verify_bytes > max_inflight_bytes)
6939 			cv_wait(&spa->spa_scrub_io_cv, &spa->spa_scrub_lock);
6940 		spa->spa_load_verify_bytes += size;
6941 		mutex_exit(&spa->spa_scrub_lock);
6942 
6943 		zio_nowait(zio_read(NULL, spa, bp, abd, size,
6944 		    zdb_blkptr_done, zcb, ZIO_PRIORITY_ASYNC_READ, flags, zb));
6945 	}
6946 
6947 	zcb->zcb_readfails = 0;
6948 
6949 	/* only call gethrtime() every 100 blocks */
6950 	static int iters;
6951 	if (++iters > 100)
6952 		iters = 0;
6953 	else
6954 		return (0);
6955 
6956 	if (dump_opt['b'] < 5 && gethrtime() > zcb->zcb_lastprint + NANOSEC) {
6957 		uint64_t now = gethrtime();
6958 		char buf[10];
6959 		uint64_t bytes = zcb->zcb_type[ZB_TOTAL][ZDB_OT_TOTAL].zb_asize;
6960 		uint64_t kb_per_sec =
6961 		    1 + bytes / (1 + ((now - zcb->zcb_start) / 1000 / 1000));
6962 		uint64_t sec_remaining =
6963 		    (zcb->zcb_totalasize - bytes) / 1024 / kb_per_sec;
6964 
6965 		/* make sure nicenum has enough space */
6966 		_Static_assert(sizeof (buf) >= NN_NUMBUF_SZ, "buf truncated");
6967 
6968 		zfs_nicebytes(bytes, buf, sizeof (buf));
6969 		(void) fprintf(stderr,
6970 		    "\r%5s completed (%4"PRIu64"MB/s) "
6971 		    "estimated time remaining: "
6972 		    "%"PRIu64"hr %02"PRIu64"min %02"PRIu64"sec        ",
6973 		    buf, kb_per_sec / 1024,
6974 		    sec_remaining / 60 / 60,
6975 		    sec_remaining / 60 % 60,
6976 		    sec_remaining % 60);
6977 
6978 		zcb->zcb_lastprint = now;
6979 	}
6980 
6981 	return (0);
6982 }
6983 
6984 static void
zdb_leak(void * arg,uint64_t start,uint64_t size)6985 zdb_leak(void *arg, uint64_t start, uint64_t size)
6986 {
6987 	vdev_t *vd = arg;
6988 
6989 	(void) printf("leaked space: vdev %llu, offset 0x%llx, size %llu\n",
6990 	    (u_longlong_t)vd->vdev_id, (u_longlong_t)start, (u_longlong_t)size);
6991 }
6992 
6993 static metaslab_ops_t zdb_metaslab_ops = {
6994 	NULL	/* alloc */
6995 };
6996 
6997 static int
load_unflushed_svr_segs_cb(spa_t * spa,space_map_entry_t * sme,uint64_t txg,void * arg)6998 load_unflushed_svr_segs_cb(spa_t *spa, space_map_entry_t *sme,
6999     uint64_t txg, void *arg)
7000 {
7001 	spa_vdev_removal_t *svr = arg;
7002 
7003 	uint64_t offset = sme->sme_offset;
7004 	uint64_t size = sme->sme_run;
7005 
7006 	/* skip vdevs we don't care about */
7007 	if (sme->sme_vdev != svr->svr_vdev_id)
7008 		return (0);
7009 
7010 	vdev_t *vd = vdev_lookup_top(spa, sme->sme_vdev);
7011 	metaslab_t *ms = vd->vdev_ms[offset >> vd->vdev_ms_shift];
7012 	ASSERT(sme->sme_type == SM_ALLOC || sme->sme_type == SM_FREE);
7013 
7014 	if (txg < metaslab_unflushed_txg(ms))
7015 		return (0);
7016 
7017 	if (sme->sme_type == SM_ALLOC)
7018 		zfs_range_tree_add(svr->svr_allocd_segs, offset, size);
7019 	else
7020 		zfs_range_tree_remove(svr->svr_allocd_segs, offset, size);
7021 
7022 	return (0);
7023 }
7024 
7025 static void
claim_segment_impl_cb(uint64_t inner_offset,vdev_t * vd,uint64_t offset,uint64_t size,void * arg)7026 claim_segment_impl_cb(uint64_t inner_offset, vdev_t *vd, uint64_t offset,
7027     uint64_t size, void *arg)
7028 {
7029 	(void) inner_offset, (void) arg;
7030 
7031 	/*
7032 	 * This callback was called through a remap from
7033 	 * a device being removed. Therefore, the vdev that
7034 	 * this callback is applied to is a concrete
7035 	 * vdev.
7036 	 */
7037 	ASSERT(vdev_is_concrete(vd));
7038 
7039 	VERIFY0(metaslab_claim_impl(vd, offset, size,
7040 	    spa_min_claim_txg(vd->vdev_spa)));
7041 }
7042 
7043 static void
claim_segment_cb(void * arg,uint64_t offset,uint64_t size)7044 claim_segment_cb(void *arg, uint64_t offset, uint64_t size)
7045 {
7046 	vdev_t *vd = arg;
7047 
7048 	vdev_indirect_ops.vdev_op_remap(vd, offset, size,
7049 	    claim_segment_impl_cb, NULL);
7050 }
7051 
7052 /*
7053  * After accounting for all allocated blocks that are directly referenced,
7054  * we might have missed a reference to a block from a partially complete
7055  * (and thus unused) indirect mapping object. We perform a secondary pass
7056  * through the metaslabs we have already mapped and claim the destination
7057  * blocks.
7058  */
7059 static void
zdb_claim_removing(spa_t * spa,zdb_cb_t * zcb)7060 zdb_claim_removing(spa_t *spa, zdb_cb_t *zcb)
7061 {
7062 	if (dump_opt['L'])
7063 		return;
7064 
7065 	if (spa->spa_vdev_removal == NULL)
7066 		return;
7067 
7068 	spa_config_enter(spa, SCL_CONFIG, FTAG, RW_READER);
7069 
7070 	spa_vdev_removal_t *svr = spa->spa_vdev_removal;
7071 	vdev_t *vd = vdev_lookup_top(spa, svr->svr_vdev_id);
7072 	vdev_indirect_mapping_t *vim = vd->vdev_indirect_mapping;
7073 
7074 	ASSERT0(zfs_range_tree_space(svr->svr_allocd_segs));
7075 
7076 	zfs_range_tree_t *allocs = zfs_range_tree_create_flags(
7077 	    NULL, ZFS_RANGE_SEG64, NULL, 0, 0,
7078 	    0, "zdb_claim_removing:allocs");
7079 	for (uint64_t msi = 0; msi < vd->vdev_ms_count; msi++) {
7080 		metaslab_t *msp = vd->vdev_ms[msi];
7081 
7082 		ASSERT0(zfs_range_tree_space(allocs));
7083 		if (msp->ms_sm != NULL)
7084 			VERIFY0(space_map_load(msp->ms_sm, allocs, SM_ALLOC));
7085 		zfs_range_tree_vacate(allocs, zfs_range_tree_add,
7086 		    svr->svr_allocd_segs);
7087 	}
7088 	zfs_range_tree_destroy(allocs);
7089 
7090 	iterate_through_spacemap_logs(spa, load_unflushed_svr_segs_cb, svr);
7091 
7092 	/*
7093 	 * Clear everything past what has been synced,
7094 	 * because we have not allocated mappings for
7095 	 * it yet.
7096 	 */
7097 	zfs_range_tree_clear(svr->svr_allocd_segs,
7098 	    vdev_indirect_mapping_max_offset(vim),
7099 	    vd->vdev_asize - vdev_indirect_mapping_max_offset(vim));
7100 
7101 	zcb->zcb_removing_size += zfs_range_tree_space(svr->svr_allocd_segs);
7102 	zfs_range_tree_vacate(svr->svr_allocd_segs, claim_segment_cb, vd);
7103 
7104 	spa_config_exit(spa, SCL_CONFIG, FTAG);
7105 }
7106 
7107 static int
increment_indirect_mapping_cb(void * arg,const blkptr_t * bp,boolean_t bp_freed,dmu_tx_t * tx)7108 increment_indirect_mapping_cb(void *arg, const blkptr_t *bp, boolean_t bp_freed,
7109     dmu_tx_t *tx)
7110 {
7111 	(void) tx;
7112 	zdb_cb_t *zcb = arg;
7113 	spa_t *spa = zcb->zcb_spa;
7114 	vdev_t *vd;
7115 	const dva_t *dva = &bp->blk_dva[0];
7116 
7117 	ASSERT(!bp_freed);
7118 	ASSERT(!dump_opt['L']);
7119 	ASSERT3U(BP_GET_NDVAS(bp), ==, 1);
7120 
7121 	spa_config_enter(spa, SCL_VDEV, FTAG, RW_READER);
7122 	vd = vdev_lookup_top(zcb->zcb_spa, DVA_GET_VDEV(dva));
7123 	ASSERT3P(vd, !=, NULL);
7124 	spa_config_exit(spa, SCL_VDEV, FTAG);
7125 
7126 	ASSERT(vd->vdev_indirect_config.vic_mapping_object != 0);
7127 	ASSERT3P(zcb->zcb_vd_obsolete_counts[vd->vdev_id], !=, NULL);
7128 
7129 	vdev_indirect_mapping_increment_obsolete_count(
7130 	    vd->vdev_indirect_mapping,
7131 	    DVA_GET_OFFSET(dva), DVA_GET_ASIZE(dva),
7132 	    zcb->zcb_vd_obsolete_counts[vd->vdev_id]);
7133 
7134 	return (0);
7135 }
7136 
7137 static uint32_t *
zdb_load_obsolete_counts(vdev_t * vd)7138 zdb_load_obsolete_counts(vdev_t *vd)
7139 {
7140 	vdev_indirect_mapping_t *vim = vd->vdev_indirect_mapping;
7141 	spa_t *spa = vd->vdev_spa;
7142 	spa_condensing_indirect_phys_t *scip =
7143 	    &spa->spa_condensing_indirect_phys;
7144 	uint64_t obsolete_sm_object;
7145 	uint32_t *counts;
7146 
7147 	VERIFY0(vdev_obsolete_sm_object(vd, &obsolete_sm_object));
7148 	EQUIV(obsolete_sm_object != 0, vd->vdev_obsolete_sm != NULL);
7149 	counts = vdev_indirect_mapping_load_obsolete_counts(vim);
7150 	if (vd->vdev_obsolete_sm != NULL) {
7151 		vdev_indirect_mapping_load_obsolete_spacemap(vim, counts,
7152 		    vd->vdev_obsolete_sm);
7153 	}
7154 	if (scip->scip_vdev == vd->vdev_id &&
7155 	    scip->scip_prev_obsolete_sm_object != 0) {
7156 		space_map_t *prev_obsolete_sm = NULL;
7157 		VERIFY0(space_map_open(&prev_obsolete_sm, spa->spa_meta_objset,
7158 		    scip->scip_prev_obsolete_sm_object, 0, vd->vdev_asize, 0));
7159 		vdev_indirect_mapping_load_obsolete_spacemap(vim, counts,
7160 		    prev_obsolete_sm);
7161 		space_map_close(prev_obsolete_sm);
7162 	}
7163 	return (counts);
7164 }
7165 
7166 typedef struct checkpoint_sm_exclude_entry_arg {
7167 	vdev_t *cseea_vd;
7168 	uint64_t cseea_checkpoint_size;
7169 } checkpoint_sm_exclude_entry_arg_t;
7170 
7171 static int
checkpoint_sm_exclude_entry_cb(space_map_entry_t * sme,void * arg)7172 checkpoint_sm_exclude_entry_cb(space_map_entry_t *sme, void *arg)
7173 {
7174 	checkpoint_sm_exclude_entry_arg_t *cseea = arg;
7175 	vdev_t *vd = cseea->cseea_vd;
7176 	metaslab_t *ms = vd->vdev_ms[sme->sme_offset >> vd->vdev_ms_shift];
7177 	uint64_t end = sme->sme_offset + sme->sme_run;
7178 
7179 	ASSERT(sme->sme_type == SM_FREE);
7180 
7181 	/*
7182 	 * Since the vdev_checkpoint_sm exists in the vdev level
7183 	 * and the ms_sm space maps exist in the metaslab level,
7184 	 * an entry in the checkpoint space map could theoretically
7185 	 * cross the boundaries of the metaslab that it belongs.
7186 	 *
7187 	 * In reality, because of the way that we populate and
7188 	 * manipulate the checkpoint's space maps currently,
7189 	 * there shouldn't be any entries that cross metaslabs.
7190 	 * Hence the assertion below.
7191 	 *
7192 	 * That said, there is no fundamental requirement that
7193 	 * the checkpoint's space map entries should not cross
7194 	 * metaslab boundaries. So if needed we could add code
7195 	 * that handles metaslab-crossing segments in the future.
7196 	 */
7197 	VERIFY3U(sme->sme_offset, >=, ms->ms_start);
7198 	VERIFY3U(end, <=, ms->ms_start + ms->ms_size);
7199 
7200 	/*
7201 	 * By removing the entry from the allocated segments we
7202 	 * also verify that the entry is there to begin with.
7203 	 */
7204 	mutex_enter(&ms->ms_lock);
7205 	zfs_range_tree_remove(ms->ms_allocatable, sme->sme_offset,
7206 	    sme->sme_run);
7207 	mutex_exit(&ms->ms_lock);
7208 
7209 	cseea->cseea_checkpoint_size += sme->sme_run;
7210 	return (0);
7211 }
7212 
7213 static void
zdb_leak_init_vdev_exclude_checkpoint(vdev_t * vd,zdb_cb_t * zcb)7214 zdb_leak_init_vdev_exclude_checkpoint(vdev_t *vd, zdb_cb_t *zcb)
7215 {
7216 	spa_t *spa = vd->vdev_spa;
7217 	space_map_t *checkpoint_sm = NULL;
7218 	uint64_t checkpoint_sm_obj;
7219 
7220 	/*
7221 	 * If there is no vdev_top_zap, we are in a pool whose
7222 	 * version predates the pool checkpoint feature.
7223 	 */
7224 	if (vd->vdev_top_zap == 0)
7225 		return;
7226 
7227 	/*
7228 	 * If there is no reference of the vdev_checkpoint_sm in
7229 	 * the vdev_top_zap, then one of the following scenarios
7230 	 * is true:
7231 	 *
7232 	 * 1] There is no checkpoint
7233 	 * 2] There is a checkpoint, but no checkpointed blocks
7234 	 *    have been freed yet
7235 	 * 3] The current vdev is indirect
7236 	 *
7237 	 * In these cases we return immediately.
7238 	 */
7239 	if (zap_contains(spa_meta_objset(spa), vd->vdev_top_zap,
7240 	    VDEV_TOP_ZAP_POOL_CHECKPOINT_SM) != 0)
7241 		return;
7242 
7243 	VERIFY0(zap_lookup(spa_meta_objset(spa), vd->vdev_top_zap,
7244 	    VDEV_TOP_ZAP_POOL_CHECKPOINT_SM, sizeof (uint64_t), 1,
7245 	    &checkpoint_sm_obj));
7246 
7247 	checkpoint_sm_exclude_entry_arg_t cseea;
7248 	cseea.cseea_vd = vd;
7249 	cseea.cseea_checkpoint_size = 0;
7250 
7251 	VERIFY0(space_map_open(&checkpoint_sm, spa_meta_objset(spa),
7252 	    checkpoint_sm_obj, 0, vd->vdev_asize, vd->vdev_ashift));
7253 
7254 	VERIFY0(space_map_iterate(checkpoint_sm,
7255 	    space_map_length(checkpoint_sm),
7256 	    checkpoint_sm_exclude_entry_cb, &cseea));
7257 	space_map_close(checkpoint_sm);
7258 
7259 	zcb->zcb_checkpoint_size += cseea.cseea_checkpoint_size;
7260 }
7261 
7262 static void
zdb_leak_init_exclude_checkpoint(spa_t * spa,zdb_cb_t * zcb)7263 zdb_leak_init_exclude_checkpoint(spa_t *spa, zdb_cb_t *zcb)
7264 {
7265 	ASSERT(!dump_opt['L']);
7266 
7267 	vdev_t *rvd = spa->spa_root_vdev;
7268 	for (uint64_t c = 0; c < rvd->vdev_children; c++) {
7269 		ASSERT3U(c, ==, rvd->vdev_child[c]->vdev_id);
7270 		zdb_leak_init_vdev_exclude_checkpoint(rvd->vdev_child[c], zcb);
7271 	}
7272 }
7273 
7274 static int
count_unflushed_space_cb(spa_t * spa,space_map_entry_t * sme,uint64_t txg,void * arg)7275 count_unflushed_space_cb(spa_t *spa, space_map_entry_t *sme,
7276     uint64_t txg, void *arg)
7277 {
7278 	int64_t *ualloc_space = arg;
7279 
7280 	uint64_t offset = sme->sme_offset;
7281 	uint64_t vdev_id = sme->sme_vdev;
7282 
7283 	vdev_t *vd = vdev_lookup_top(spa, vdev_id);
7284 	if (!vdev_is_concrete(vd))
7285 		return (0);
7286 
7287 	metaslab_t *ms = vd->vdev_ms[offset >> vd->vdev_ms_shift];
7288 	ASSERT(sme->sme_type == SM_ALLOC || sme->sme_type == SM_FREE);
7289 
7290 	if (txg < metaslab_unflushed_txg(ms))
7291 		return (0);
7292 
7293 	if (sme->sme_type == SM_ALLOC)
7294 		*ualloc_space += sme->sme_run;
7295 	else
7296 		*ualloc_space -= sme->sme_run;
7297 
7298 	return (0);
7299 }
7300 
7301 static int64_t
get_unflushed_alloc_space(spa_t * spa)7302 get_unflushed_alloc_space(spa_t *spa)
7303 {
7304 	if (dump_opt['L'])
7305 		return (0);
7306 
7307 	int64_t ualloc_space = 0;
7308 	iterate_through_spacemap_logs(spa, count_unflushed_space_cb,
7309 	    &ualloc_space);
7310 	return (ualloc_space);
7311 }
7312 
7313 static int
load_unflushed_cb(spa_t * spa,space_map_entry_t * sme,uint64_t txg,void * arg)7314 load_unflushed_cb(spa_t *spa, space_map_entry_t *sme, uint64_t txg, void *arg)
7315 {
7316 	maptype_t *uic_maptype = arg;
7317 
7318 	uint64_t offset = sme->sme_offset;
7319 	uint64_t size = sme->sme_run;
7320 	uint64_t vdev_id = sme->sme_vdev;
7321 
7322 	vdev_t *vd = vdev_lookup_top(spa, vdev_id);
7323 
7324 	/* skip indirect vdevs */
7325 	if (!vdev_is_concrete(vd))
7326 		return (0);
7327 
7328 	metaslab_t *ms = vd->vdev_ms[offset >> vd->vdev_ms_shift];
7329 
7330 	ASSERT(sme->sme_type == SM_ALLOC || sme->sme_type == SM_FREE);
7331 	ASSERT(*uic_maptype == SM_ALLOC || *uic_maptype == SM_FREE);
7332 
7333 	if (txg < metaslab_unflushed_txg(ms))
7334 		return (0);
7335 
7336 	if (*uic_maptype == sme->sme_type)
7337 		zfs_range_tree_add(ms->ms_allocatable, offset, size);
7338 	else
7339 		zfs_range_tree_remove(ms->ms_allocatable, offset, size);
7340 
7341 	return (0);
7342 }
7343 
7344 static void
load_unflushed_to_ms_allocatables(spa_t * spa,maptype_t maptype)7345 load_unflushed_to_ms_allocatables(spa_t *spa, maptype_t maptype)
7346 {
7347 	iterate_through_spacemap_logs(spa, load_unflushed_cb, &maptype);
7348 }
7349 
7350 static void
load_concrete_ms_allocatable_trees(spa_t * spa,maptype_t maptype)7351 load_concrete_ms_allocatable_trees(spa_t *spa, maptype_t maptype)
7352 {
7353 	vdev_t *rvd = spa->spa_root_vdev;
7354 	for (uint64_t i = 0; i < rvd->vdev_children; i++) {
7355 		vdev_t *vd = rvd->vdev_child[i];
7356 
7357 		ASSERT3U(i, ==, vd->vdev_id);
7358 
7359 		if (vd->vdev_ops == &vdev_indirect_ops)
7360 			continue;
7361 
7362 		for (uint64_t m = 0; m < vd->vdev_ms_count; m++) {
7363 			metaslab_t *msp = vd->vdev_ms[m];
7364 
7365 			(void) fprintf(stderr,
7366 			    "\rloading concrete vdev %llu, "
7367 			    "metaslab %llu of %llu ...",
7368 			    (longlong_t)vd->vdev_id,
7369 			    (longlong_t)msp->ms_id,
7370 			    (longlong_t)vd->vdev_ms_count);
7371 
7372 			mutex_enter(&msp->ms_lock);
7373 			zfs_range_tree_vacate(msp->ms_allocatable, NULL, NULL);
7374 
7375 			/*
7376 			 * We don't want to spend the CPU manipulating the
7377 			 * size-ordered tree, so clear the range_tree ops.
7378 			 */
7379 			msp->ms_allocatable->rt_ops = NULL;
7380 
7381 			if (msp->ms_sm != NULL) {
7382 				VERIFY0(space_map_load(msp->ms_sm,
7383 				    msp->ms_allocatable, maptype));
7384 			}
7385 			if (!msp->ms_loaded)
7386 				msp->ms_loaded = B_TRUE;
7387 			mutex_exit(&msp->ms_lock);
7388 		}
7389 	}
7390 
7391 	load_unflushed_to_ms_allocatables(spa, maptype);
7392 }
7393 
7394 /*
7395  * vm_idxp is an in-out parameter which (for indirect vdevs) is the
7396  * index in vim_entries that has the first entry in this metaslab.
7397  * On return, it will be set to the first entry after this metaslab.
7398  */
7399 static void
load_indirect_ms_allocatable_tree(vdev_t * vd,metaslab_t * msp,uint64_t * vim_idxp)7400 load_indirect_ms_allocatable_tree(vdev_t *vd, metaslab_t *msp,
7401     uint64_t *vim_idxp)
7402 {
7403 	vdev_indirect_mapping_t *vim = vd->vdev_indirect_mapping;
7404 
7405 	mutex_enter(&msp->ms_lock);
7406 	zfs_range_tree_vacate(msp->ms_allocatable, NULL, NULL);
7407 
7408 	/*
7409 	 * We don't want to spend the CPU manipulating the
7410 	 * size-ordered tree, so clear the range_tree ops.
7411 	 */
7412 	msp->ms_allocatable->rt_ops = NULL;
7413 
7414 	for (; *vim_idxp < vdev_indirect_mapping_num_entries(vim);
7415 	    (*vim_idxp)++) {
7416 		vdev_indirect_mapping_entry_phys_t *vimep =
7417 		    &vim->vim_entries[*vim_idxp];
7418 		uint64_t ent_offset = DVA_MAPPING_GET_SRC_OFFSET(vimep);
7419 		uint64_t ent_len = DVA_GET_ASIZE(&vimep->vimep_dst);
7420 		ASSERT3U(ent_offset, >=, msp->ms_start);
7421 		if (ent_offset >= msp->ms_start + msp->ms_size)
7422 			break;
7423 
7424 		/*
7425 		 * Mappings do not cross metaslab boundaries,
7426 		 * because we create them by walking the metaslabs.
7427 		 */
7428 		ASSERT3U(ent_offset + ent_len, <=,
7429 		    msp->ms_start + msp->ms_size);
7430 		zfs_range_tree_add(msp->ms_allocatable, ent_offset, ent_len);
7431 	}
7432 
7433 	if (!msp->ms_loaded)
7434 		msp->ms_loaded = B_TRUE;
7435 	mutex_exit(&msp->ms_lock);
7436 }
7437 
7438 static void
zdb_leak_init_prepare_indirect_vdevs(spa_t * spa,zdb_cb_t * zcb)7439 zdb_leak_init_prepare_indirect_vdevs(spa_t *spa, zdb_cb_t *zcb)
7440 {
7441 	ASSERT(!dump_opt['L']);
7442 
7443 	vdev_t *rvd = spa->spa_root_vdev;
7444 	for (uint64_t c = 0; c < rvd->vdev_children; c++) {
7445 		vdev_t *vd = rvd->vdev_child[c];
7446 
7447 		ASSERT3U(c, ==, vd->vdev_id);
7448 
7449 		if (vd->vdev_ops != &vdev_indirect_ops)
7450 			continue;
7451 
7452 		/*
7453 		 * Note: we don't check for mapping leaks on
7454 		 * removing vdevs because their ms_allocatable's
7455 		 * are used to look for leaks in allocated space.
7456 		 */
7457 		zcb->zcb_vd_obsolete_counts[c] = zdb_load_obsolete_counts(vd);
7458 
7459 		/*
7460 		 * Normally, indirect vdevs don't have any
7461 		 * metaslabs.  We want to set them up for
7462 		 * zio_claim().
7463 		 */
7464 		vdev_metaslab_group_create(vd);
7465 		VERIFY0(vdev_metaslab_init(vd, 0));
7466 
7467 		vdev_indirect_mapping_t *vim __maybe_unused =
7468 		    vd->vdev_indirect_mapping;
7469 		uint64_t vim_idx = 0;
7470 		for (uint64_t m = 0; m < vd->vdev_ms_count; m++) {
7471 
7472 			(void) fprintf(stderr,
7473 			    "\rloading indirect vdev %llu, "
7474 			    "metaslab %llu of %llu ...",
7475 			    (longlong_t)vd->vdev_id,
7476 			    (longlong_t)vd->vdev_ms[m]->ms_id,
7477 			    (longlong_t)vd->vdev_ms_count);
7478 
7479 			load_indirect_ms_allocatable_tree(vd, vd->vdev_ms[m],
7480 			    &vim_idx);
7481 		}
7482 		ASSERT3U(vim_idx, ==, vdev_indirect_mapping_num_entries(vim));
7483 	}
7484 }
7485 
7486 static void
zdb_leak_init(spa_t * spa,zdb_cb_t * zcb)7487 zdb_leak_init(spa_t *spa, zdb_cb_t *zcb)
7488 {
7489 	zcb->zcb_spa = spa;
7490 
7491 	if (dump_opt['L'])
7492 		return;
7493 
7494 	dsl_pool_t *dp = spa->spa_dsl_pool;
7495 	vdev_t *rvd = spa->spa_root_vdev;
7496 
7497 	/*
7498 	 * We are going to be changing the meaning of the metaslab's
7499 	 * ms_allocatable.  Ensure that the allocator doesn't try to
7500 	 * use the tree.
7501 	 */
7502 	spa->spa_normal_class->mc_ops = &zdb_metaslab_ops;
7503 	spa->spa_log_class->mc_ops = &zdb_metaslab_ops;
7504 	spa->spa_embedded_log_class->mc_ops = &zdb_metaslab_ops;
7505 	spa->spa_special_embedded_log_class->mc_ops = &zdb_metaslab_ops;
7506 
7507 	zcb->zcb_vd_obsolete_counts =
7508 	    umem_zalloc(rvd->vdev_children * sizeof (uint32_t *),
7509 	    UMEM_NOFAIL);
7510 
7511 	/*
7512 	 * For leak detection, we overload the ms_allocatable trees
7513 	 * to contain allocated segments instead of free segments.
7514 	 * As a result, we can't use the normal metaslab_load/unload
7515 	 * interfaces.
7516 	 */
7517 	zdb_leak_init_prepare_indirect_vdevs(spa, zcb);
7518 	load_concrete_ms_allocatable_trees(spa, SM_ALLOC);
7519 
7520 	/*
7521 	 * On load_concrete_ms_allocatable_trees() we loaded all the
7522 	 * allocated entries from the ms_sm to the ms_allocatable for
7523 	 * each metaslab. If the pool has a checkpoint or is in the
7524 	 * middle of discarding a checkpoint, some of these blocks
7525 	 * may have been freed but their ms_sm may not have been
7526 	 * updated because they are referenced by the checkpoint. In
7527 	 * order to avoid false-positives during leak-detection, we
7528 	 * go through the vdev's checkpoint space map and exclude all
7529 	 * its entries from their relevant ms_allocatable.
7530 	 *
7531 	 * We also aggregate the space held by the checkpoint and add
7532 	 * it to zcb_checkpoint_size.
7533 	 *
7534 	 * Note that at this point we are also verifying that all the
7535 	 * entries on the checkpoint_sm are marked as allocated in
7536 	 * the ms_sm of their relevant metaslab.
7537 	 * [see comment in checkpoint_sm_exclude_entry_cb()]
7538 	 */
7539 	zdb_leak_init_exclude_checkpoint(spa, zcb);
7540 	ASSERT3U(zcb->zcb_checkpoint_size, ==, spa_get_checkpoint_space(spa));
7541 
7542 	/* for cleaner progress output */
7543 	(void) fprintf(stderr, "\n");
7544 
7545 	if (bpobj_is_open(&dp->dp_obsolete_bpobj)) {
7546 		ASSERT(spa_feature_is_enabled(spa,
7547 		    SPA_FEATURE_DEVICE_REMOVAL));
7548 		(void) bpobj_iterate_nofree(&dp->dp_obsolete_bpobj,
7549 		    increment_indirect_mapping_cb, zcb, NULL);
7550 	}
7551 }
7552 
7553 static boolean_t
zdb_check_for_obsolete_leaks(vdev_t * vd,zdb_cb_t * zcb)7554 zdb_check_for_obsolete_leaks(vdev_t *vd, zdb_cb_t *zcb)
7555 {
7556 	boolean_t leaks = B_FALSE;
7557 	vdev_indirect_mapping_t *vim = vd->vdev_indirect_mapping;
7558 	uint64_t total_leaked = 0;
7559 	boolean_t are_precise = B_FALSE;
7560 
7561 	ASSERT(vim != NULL);
7562 
7563 	for (uint64_t i = 0; i < vdev_indirect_mapping_num_entries(vim); i++) {
7564 		vdev_indirect_mapping_entry_phys_t *vimep =
7565 		    &vim->vim_entries[i];
7566 		uint64_t obsolete_bytes = 0;
7567 		uint64_t offset = DVA_MAPPING_GET_SRC_OFFSET(vimep);
7568 		metaslab_t *msp = vd->vdev_ms[offset >> vd->vdev_ms_shift];
7569 
7570 		/*
7571 		 * This is not very efficient but it's easy to
7572 		 * verify correctness.
7573 		 */
7574 		for (uint64_t inner_offset = 0;
7575 		    inner_offset < DVA_GET_ASIZE(&vimep->vimep_dst);
7576 		    inner_offset += 1ULL << vd->vdev_ashift) {
7577 			if (zfs_range_tree_contains(msp->ms_allocatable,
7578 			    offset + inner_offset, 1ULL << vd->vdev_ashift)) {
7579 				obsolete_bytes += 1ULL << vd->vdev_ashift;
7580 			}
7581 		}
7582 
7583 		int64_t bytes_leaked = obsolete_bytes -
7584 		    zcb->zcb_vd_obsolete_counts[vd->vdev_id][i];
7585 		ASSERT3U(DVA_GET_ASIZE(&vimep->vimep_dst), >=,
7586 		    zcb->zcb_vd_obsolete_counts[vd->vdev_id][i]);
7587 
7588 		VERIFY0(vdev_obsolete_counts_are_precise(vd, &are_precise));
7589 		if (bytes_leaked != 0 && (are_precise || dump_opt['d'] >= 5)) {
7590 			(void) printf("obsolete indirect mapping count "
7591 			    "mismatch on %llu:%llx:%llx : %llx bytes leaked\n",
7592 			    (u_longlong_t)vd->vdev_id,
7593 			    (u_longlong_t)DVA_MAPPING_GET_SRC_OFFSET(vimep),
7594 			    (u_longlong_t)DVA_GET_ASIZE(&vimep->vimep_dst),
7595 			    (u_longlong_t)bytes_leaked);
7596 		}
7597 		total_leaked += ABS(bytes_leaked);
7598 	}
7599 
7600 	VERIFY0(vdev_obsolete_counts_are_precise(vd, &are_precise));
7601 	if (!are_precise && total_leaked > 0) {
7602 		int pct_leaked = total_leaked * 100 /
7603 		    vdev_indirect_mapping_bytes_mapped(vim);
7604 		(void) printf("cannot verify obsolete indirect mapping "
7605 		    "counts of vdev %llu because precise feature was not "
7606 		    "enabled when it was removed: %d%% (%llx bytes) of mapping"
7607 		    "unreferenced\n",
7608 		    (u_longlong_t)vd->vdev_id, pct_leaked,
7609 		    (u_longlong_t)total_leaked);
7610 	} else if (total_leaked > 0) {
7611 		(void) printf("obsolete indirect mapping count mismatch "
7612 		    "for vdev %llu -- %llx total bytes mismatched\n",
7613 		    (u_longlong_t)vd->vdev_id,
7614 		    (u_longlong_t)total_leaked);
7615 		leaks |= B_TRUE;
7616 	}
7617 
7618 	vdev_indirect_mapping_free_obsolete_counts(vim,
7619 	    zcb->zcb_vd_obsolete_counts[vd->vdev_id]);
7620 	zcb->zcb_vd_obsolete_counts[vd->vdev_id] = NULL;
7621 
7622 	return (leaks);
7623 }
7624 
7625 static boolean_t
zdb_leak_fini(spa_t * spa,zdb_cb_t * zcb)7626 zdb_leak_fini(spa_t *spa, zdb_cb_t *zcb)
7627 {
7628 	boolean_t leaks = B_FALSE;
7629 
7630 	/*
7631 	 * Report leaked BRT entries whose refcount was not fully consumed by
7632 	 * the traversal.
7633 	 */
7634 	if (zcb->zcb_brt_is_active) {
7635 		void *cookie = NULL;
7636 		zdb_brt_entry_t *zbre;
7637 		while ((zbre = avl_destroy_nodes(
7638 		    &zcb->zcb_brt, &cookie)) != NULL) {
7639 			if (!dump_opt['L'] && zbre->zbre_refcount != 0) {
7640 				(void) printf("BRT leak: vdev %llu, "
7641 				    "offset 0x%llx, refcount %llu\n",
7642 				    (u_longlong_t)DVA_GET_VDEV(
7643 				    &zbre->zbre_dva),
7644 				    (u_longlong_t)DVA_GET_OFFSET(
7645 				    &zbre->zbre_dva),
7646 				    (u_longlong_t)zbre->zbre_refcount);
7647 				leaks = B_TRUE;
7648 			}
7649 			umem_free(zbre, sizeof (zdb_brt_entry_t));
7650 		}
7651 		avl_destroy(&zcb->zcb_brt);
7652 	}
7653 
7654 	if (dump_opt['L'])
7655 		return (leaks);
7656 
7657 	/*
7658 	 * Report leaked DDT entries whose refcount was not fully consumed by
7659 	 * the traversal.  Entries in the DDT ZAP that were never looked up
7660 	 * are not detected here.
7661 	 */
7662 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
7663 		ddt_t *ddt = spa->spa_ddt[c];
7664 		if (ddt == NULL)
7665 			continue;
7666 		ddt_enter(ddt);
7667 		for (ddt_entry_t *dde = avl_first(&ddt->ddt_tree); dde != NULL;
7668 		    dde = AVL_NEXT(&ddt->ddt_tree, dde)) {
7669 			for (int p = 0; p < DDT_NPHYS(ddt); p++) {
7670 				ddt_phys_variant_t v = DDT_PHYS_VARIANT(ddt, p);
7671 				uint64_t refcnt = ddt_phys_refcnt(dde->dde_phys,
7672 				    v);
7673 				if (refcnt == 0)
7674 					continue;
7675 				blkptr_t blk;
7676 				char blkbuf[BP_SPRINTF_LEN];
7677 				ddt_bp_create(ddt->ddt_checksum, &dde->dde_key,
7678 				    dde->dde_phys, v, &blk);
7679 				snprintf_blkptr(blkbuf, sizeof (blkbuf), &blk);
7680 				(void) printf("DDT leak: refcount %llu %s\n",
7681 				    (u_longlong_t)refcnt, blkbuf);
7682 				leaks = B_TRUE;
7683 			}
7684 		}
7685 		ddt_exit(ddt);
7686 	}
7687 
7688 	vdev_t *rvd = spa->spa_root_vdev;
7689 	for (unsigned c = 0; c < rvd->vdev_children; c++) {
7690 		vdev_t *vd = rvd->vdev_child[c];
7691 
7692 		if (zcb->zcb_vd_obsolete_counts[c] != NULL) {
7693 			leaks |= zdb_check_for_obsolete_leaks(vd, zcb);
7694 		}
7695 
7696 		for (uint64_t m = 0; m < vd->vdev_ms_count; m++) {
7697 			metaslab_t *msp = vd->vdev_ms[m];
7698 			ASSERT3P(msp->ms_group, ==, (msp->ms_group->mg_class ==
7699 			    spa_embedded_log_class(spa) ||
7700 			    msp->ms_group->mg_class ==
7701 			    spa_special_embedded_log_class(spa)) ?
7702 			    vd->vdev_log_mg : vd->vdev_mg);
7703 
7704 			/*
7705 			 * ms_allocatable has been overloaded
7706 			 * to contain allocated segments. Now that
7707 			 * we finished traversing all blocks, any
7708 			 * block that remains in the ms_allocatable
7709 			 * represents an allocated block that we
7710 			 * did not claim during the traversal.
7711 			 * Claimed blocks would have been removed
7712 			 * from the ms_allocatable.  For indirect
7713 			 * vdevs, space remaining in the tree
7714 			 * represents parts of the mapping that are
7715 			 * not referenced, which is not a bug.
7716 			 */
7717 			if (vd->vdev_ops == &vdev_indirect_ops) {
7718 				zfs_range_tree_vacate(msp->ms_allocatable,
7719 				    NULL, NULL);
7720 			} else {
7721 				zfs_range_tree_vacate(msp->ms_allocatable,
7722 				    zdb_leak, vd);
7723 			}
7724 			if (msp->ms_loaded) {
7725 				msp->ms_loaded = B_FALSE;
7726 			}
7727 		}
7728 	}
7729 
7730 	umem_free(zcb->zcb_vd_obsolete_counts,
7731 	    rvd->vdev_children * sizeof (uint32_t *));
7732 	zcb->zcb_vd_obsolete_counts = NULL;
7733 
7734 	return (leaks);
7735 }
7736 
7737 static int
count_block_cb(void * arg,const blkptr_t * bp,dmu_tx_t * tx)7738 count_block_cb(void *arg, const blkptr_t *bp, dmu_tx_t *tx)
7739 {
7740 	(void) tx;
7741 	zdb_cb_t *zcb = arg;
7742 
7743 	if (dump_opt['b'] >= 5) {
7744 		char blkbuf[BP_SPRINTF_LEN];
7745 		snprintf_blkptr(blkbuf, sizeof (blkbuf), bp);
7746 		(void) printf("[%s] %s\n",
7747 		    "deferred free", blkbuf);
7748 	}
7749 	zdb_count_block(zcb, NULL, bp, ZDB_OT_DEFERRED);
7750 	return (0);
7751 }
7752 
7753 /*
7754  * Iterate over livelists which have been destroyed by the user but
7755  * are still present in the MOS, waiting to be freed
7756  */
7757 static void
iterate_deleted_livelists(spa_t * spa,ll_iter_t func,void * arg)7758 iterate_deleted_livelists(spa_t *spa, ll_iter_t func, void *arg)
7759 {
7760 	objset_t *mos = spa->spa_meta_objset;
7761 	uint64_t zap_obj;
7762 	int err = zap_lookup(mos, DMU_POOL_DIRECTORY_OBJECT,
7763 	    DMU_POOL_DELETED_CLONES, sizeof (uint64_t), 1, &zap_obj);
7764 	if (err == ENOENT)
7765 		return;
7766 	ASSERT0(err);
7767 
7768 	zap_cursor_t zc;
7769 	zap_attribute_t *attrp = zap_attribute_alloc();
7770 	dsl_deadlist_t ll;
7771 	/* NULL out os prior to dsl_deadlist_open in case it's garbage */
7772 	ll.dl_os = NULL;
7773 	for (zap_cursor_init(&zc, mos, zap_obj);
7774 	    zap_cursor_retrieve(&zc, attrp) == 0;
7775 	    (void) zap_cursor_advance(&zc)) {
7776 		VERIFY0(dsl_deadlist_open(&ll, mos, attrp->za_first_integer));
7777 		func(&ll, arg);
7778 		dsl_deadlist_close(&ll);
7779 	}
7780 	zap_cursor_fini(&zc);
7781 	zap_attribute_free(attrp);
7782 }
7783 
7784 static int
bpobj_count_block_cb(void * arg,const blkptr_t * bp,boolean_t bp_freed,dmu_tx_t * tx)7785 bpobj_count_block_cb(void *arg, const blkptr_t *bp, boolean_t bp_freed,
7786     dmu_tx_t *tx)
7787 {
7788 	ASSERT(!bp_freed);
7789 	return (count_block_cb(arg, bp, tx));
7790 }
7791 
7792 static int
livelist_entry_count_blocks_cb(void * args,dsl_deadlist_entry_t * dle)7793 livelist_entry_count_blocks_cb(void *args, dsl_deadlist_entry_t *dle)
7794 {
7795 	zdb_cb_t *zbc = args;
7796 	bplist_t blks;
7797 	bplist_create(&blks);
7798 	/* determine which blocks have been alloc'd but not freed */
7799 	VERIFY0(dsl_process_sub_livelist(&dle->dle_bpobj, &blks, NULL, NULL));
7800 	/* count those blocks */
7801 	(void) bplist_iterate(&blks, count_block_cb, zbc, NULL);
7802 	bplist_destroy(&blks);
7803 	return (0);
7804 }
7805 
7806 static void
livelist_count_blocks(dsl_deadlist_t * ll,void * arg)7807 livelist_count_blocks(dsl_deadlist_t *ll, void *arg)
7808 {
7809 	dsl_deadlist_iterate(ll, livelist_entry_count_blocks_cb, arg);
7810 }
7811 
7812 /*
7813  * Count the blocks in the livelists that have been destroyed by the user
7814  * but haven't yet been freed.
7815  */
7816 static void
deleted_livelists_count_blocks(spa_t * spa,zdb_cb_t * zbc)7817 deleted_livelists_count_blocks(spa_t *spa, zdb_cb_t *zbc)
7818 {
7819 	iterate_deleted_livelists(spa, livelist_count_blocks, zbc);
7820 }
7821 
7822 static void
dump_livelist_cb(dsl_deadlist_t * ll,void * arg)7823 dump_livelist_cb(dsl_deadlist_t *ll, void *arg)
7824 {
7825 	ASSERT0P(arg);
7826 	global_feature_count[SPA_FEATURE_LIVELIST]++;
7827 	dump_blkptr_list(ll, "Deleted Livelist");
7828 	dsl_deadlist_iterate(ll, sublivelist_verify_lightweight, NULL);
7829 }
7830 
7831 /*
7832  * Print out, register object references to, and increment feature counts for
7833  * livelists that have been destroyed by the user but haven't yet been freed.
7834  */
7835 static void
deleted_livelists_dump_mos(spa_t * spa)7836 deleted_livelists_dump_mos(spa_t *spa)
7837 {
7838 	uint64_t zap_obj;
7839 	objset_t *mos = spa->spa_meta_objset;
7840 	int err = zap_lookup(mos, DMU_POOL_DIRECTORY_OBJECT,
7841 	    DMU_POOL_DELETED_CLONES, sizeof (uint64_t), 1, &zap_obj);
7842 	if (err == ENOENT)
7843 		return;
7844 	mos_obj_refd(zap_obj);
7845 	iterate_deleted_livelists(spa, dump_livelist_cb, NULL);
7846 }
7847 
7848 static int
zdb_brt_entry_compare(const void * zcn1,const void * zcn2)7849 zdb_brt_entry_compare(const void *zcn1, const void *zcn2)
7850 {
7851 	const dva_t *dva1 = &((const zdb_brt_entry_t *)zcn1)->zbre_dva;
7852 	const dva_t *dva2 = &((const zdb_brt_entry_t *)zcn2)->zbre_dva;
7853 	int cmp;
7854 
7855 	cmp = TREE_CMP(DVA_GET_VDEV(dva1), DVA_GET_VDEV(dva2));
7856 	if (cmp == 0)
7857 		cmp = TREE_CMP(DVA_GET_OFFSET(dva1), DVA_GET_OFFSET(dva2));
7858 
7859 	return (cmp);
7860 }
7861 
7862 static int
dump_block_stats(spa_t * spa)7863 dump_block_stats(spa_t *spa)
7864 {
7865 	zdb_cb_t *zcb;
7866 	zdb_blkstats_t *zb, *tzb;
7867 	uint64_t norm_alloc, norm_space, total_alloc, total_found;
7868 	int flags = TRAVERSE_PRE | TRAVERSE_PREFETCH_METADATA |
7869 	    TRAVERSE_NO_DECRYPT | TRAVERSE_HARD;
7870 	boolean_t leaks = B_FALSE;
7871 	int e, c, err;
7872 	bp_embedded_type_t i;
7873 
7874 	ddt_prefetch_all(spa);
7875 
7876 	zcb = umem_zalloc(sizeof (zdb_cb_t), UMEM_NOFAIL);
7877 
7878 	if (spa_feature_is_active(spa, SPA_FEATURE_BLOCK_CLONING)) {
7879 		avl_create(&zcb->zcb_brt, zdb_brt_entry_compare,
7880 		    sizeof (zdb_brt_entry_t),
7881 		    offsetof(zdb_brt_entry_t, zbre_node));
7882 		zcb->zcb_brt_is_active = B_TRUE;
7883 	}
7884 
7885 	(void) printf("\nTraversing all blocks %s%s%s%s%s...\n\n",
7886 	    (dump_opt['c'] || !dump_opt['L']) ? "to verify " : "",
7887 	    (dump_opt['c'] == 1) ? "metadata " : "",
7888 	    dump_opt['c'] ? "checksums " : "",
7889 	    (dump_opt['c'] && !dump_opt['L']) ? "and verify " : "",
7890 	    !dump_opt['L'] ? "nothing leaked " : "");
7891 
7892 	/*
7893 	 * When leak detection is enabled we load all space maps as SM_ALLOC
7894 	 * maps, then traverse the pool claiming each block we discover. If
7895 	 * the pool is perfectly consistent, the segment trees will be empty
7896 	 * when we're done. Anything left over is a leak; any block we can't
7897 	 * claim (because it's not part of any space map) is a double
7898 	 * allocation, reference to a freed block, or an unclaimed log block.
7899 	 *
7900 	 * When leak detection is disabled (-L option) we still traverse the
7901 	 * pool claiming each block we discover, but we skip opening any space
7902 	 * maps.
7903 	 */
7904 	zdb_leak_init(spa, zcb);
7905 
7906 	/*
7907 	 * If there's a deferred-free bplist, process that first.
7908 	 */
7909 	(void) bpobj_iterate_nofree(&spa->spa_deferred_bpobj,
7910 	    bpobj_count_block_cb, zcb, NULL);
7911 
7912 	if (spa_version(spa) >= SPA_VERSION_DEADLISTS) {
7913 		(void) bpobj_iterate_nofree(&spa->spa_dsl_pool->dp_free_bpobj,
7914 		    bpobj_count_block_cb, zcb, NULL);
7915 	}
7916 
7917 	zdb_claim_removing(spa, zcb);
7918 
7919 	if (spa_feature_is_active(spa, SPA_FEATURE_ASYNC_DESTROY)) {
7920 		VERIFY3U(0, ==, bptree_iterate(spa->spa_meta_objset,
7921 		    spa->spa_dsl_pool->dp_bptree_obj, B_FALSE, count_block_cb,
7922 		    zcb, NULL));
7923 	}
7924 
7925 	deleted_livelists_count_blocks(spa, zcb);
7926 
7927 	if (dump_opt['c'] > 1)
7928 		flags |= TRAVERSE_PREFETCH_DATA;
7929 
7930 	zcb->zcb_totalasize = metaslab_class_get_alloc(spa_normal_class(spa));
7931 	zcb->zcb_totalasize += metaslab_class_get_alloc(spa_special_class(spa));
7932 	zcb->zcb_totalasize += metaslab_class_get_alloc(spa_dedup_class(spa));
7933 	zcb->zcb_totalasize +=
7934 	    metaslab_class_get_alloc(spa_embedded_log_class(spa));
7935 	zcb->zcb_totalasize +=
7936 	    metaslab_class_get_alloc(spa_special_embedded_log_class(spa));
7937 	zcb->zcb_start = zcb->zcb_lastprint = gethrtime();
7938 	err = traverse_pool(spa, 0, flags, zdb_blkptr_cb, zcb);
7939 
7940 	/*
7941 	 * If we've traversed the data blocks then we need to wait for those
7942 	 * I/Os to complete. We leverage "The Godfather" zio to wait on
7943 	 * all async I/Os to complete.
7944 	 */
7945 	if (dump_opt['c']) {
7946 		for (c = 0; c < max_ncpus; c++) {
7947 			(void) zio_wait(spa->spa_async_zio_root[c]);
7948 			spa->spa_async_zio_root[c] = zio_root(spa, NULL, NULL,
7949 			    ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE |
7950 			    ZIO_FLAG_GODFATHER);
7951 		}
7952 	}
7953 	ASSERT0(spa->spa_load_verify_bytes);
7954 
7955 	/*
7956 	 * Done after zio_wait() since zcb_haderrors is modified in
7957 	 * zdb_blkptr_done()
7958 	 */
7959 	zcb->zcb_haderrors |= err;
7960 
7961 	if (zcb->zcb_haderrors) {
7962 		(void) printf("\nError counts:\n\n");
7963 		(void) printf("\t%5s  %s\n", "errno", "count");
7964 		for (e = 0; e < 256; e++) {
7965 			if (zcb->zcb_errors[e] != 0) {
7966 				(void) printf("\t%5d  %llu\n",
7967 				    e, (u_longlong_t)zcb->zcb_errors[e]);
7968 			}
7969 		}
7970 	}
7971 
7972 	/*
7973 	 * Report any leaked segments.
7974 	 */
7975 	leaks |= zdb_leak_fini(spa, zcb);
7976 
7977 	tzb = &zcb->zcb_type[ZB_TOTAL][ZDB_OT_TOTAL];
7978 
7979 	norm_alloc = metaslab_class_get_alloc(spa_normal_class(spa));
7980 	norm_space = metaslab_class_get_space(spa_normal_class(spa));
7981 
7982 	total_alloc = norm_alloc +
7983 	    metaslab_class_get_alloc(spa_log_class(spa)) +
7984 	    metaslab_class_get_alloc(spa_embedded_log_class(spa)) +
7985 	    metaslab_class_get_alloc(spa_special_embedded_log_class(spa)) +
7986 	    metaslab_class_get_alloc(spa_special_class(spa)) +
7987 	    metaslab_class_get_alloc(spa_dedup_class(spa)) +
7988 	    get_unflushed_alloc_space(spa);
7989 	total_found =
7990 	    tzb->zb_asize - zcb->zcb_dedup_asize - zcb->zcb_clone_asize +
7991 	    zcb->zcb_removing_size + zcb->zcb_checkpoint_size;
7992 
7993 	if (total_found == total_alloc && !dump_opt['L']) {
7994 		(void) printf("\n\tNo leaks (block sum matches space"
7995 		    " maps exactly)\n");
7996 	} else if (!dump_opt['L']) {
7997 		(void) printf("block traversal size %llu != alloc %llu "
7998 		    "(%s %lld)\n",
7999 		    (u_longlong_t)total_found,
8000 		    (u_longlong_t)total_alloc,
8001 		    (dump_opt['L']) ? "unreachable" : "leaked",
8002 		    (longlong_t)(total_alloc - total_found));
8003 	}
8004 
8005 	if (tzb->zb_count == 0) {
8006 		umem_free(zcb, sizeof (zdb_cb_t));
8007 		return (2);
8008 	}
8009 
8010 	(void) printf("\n");
8011 	(void) printf("\t%-16s %14llu\n", "bp count:",
8012 	    (u_longlong_t)tzb->zb_count);
8013 	(void) printf("\t%-16s %14llu\n", "ganged count:",
8014 	    (longlong_t)tzb->zb_gangs);
8015 	(void) printf("\t%-16s %14llu      avg: %6llu\n", "bp logical:",
8016 	    (u_longlong_t)tzb->zb_lsize,
8017 	    (u_longlong_t)(tzb->zb_lsize / tzb->zb_count));
8018 	(void) printf("\t%-16s %14llu      avg: %6llu     compression: %6.2f\n",
8019 	    "bp physical:", (u_longlong_t)tzb->zb_psize,
8020 	    (u_longlong_t)(tzb->zb_psize / tzb->zb_count),
8021 	    (double)tzb->zb_lsize / tzb->zb_psize);
8022 	(void) printf("\t%-16s %14llu      avg: %6llu     compression: %6.2f\n",
8023 	    "bp allocated:", (u_longlong_t)tzb->zb_asize,
8024 	    (u_longlong_t)(tzb->zb_asize / tzb->zb_count),
8025 	    (double)tzb->zb_lsize / tzb->zb_asize);
8026 	(void) printf("\t%-16s %14llu    ref>1: %6llu   deduplication: %6.2f\n",
8027 	    "bp deduped:", (u_longlong_t)zcb->zcb_dedup_asize,
8028 	    (u_longlong_t)zcb->zcb_dedup_blocks,
8029 	    (double)zcb->zcb_dedup_asize / tzb->zb_asize + 1.0);
8030 	(void) printf("\t%-16s %14llu    count: %6llu\n",
8031 	    "bp cloned:", (u_longlong_t)zcb->zcb_clone_asize,
8032 	    (u_longlong_t)zcb->zcb_clone_blocks);
8033 	(void) printf("\t%-16s %14llu     used: %5.2f%%\n", "Normal class:",
8034 	    (u_longlong_t)norm_alloc, 100.0 * norm_alloc / norm_space);
8035 
8036 	if (spa_special_class(spa)->mc_allocator[0].mca_rotor != NULL) {
8037 		uint64_t alloc = metaslab_class_get_alloc(
8038 		    spa_special_class(spa));
8039 		uint64_t space = metaslab_class_get_space(
8040 		    spa_special_class(spa));
8041 
8042 		(void) printf("\t%-16s %14llu     used: %5.2f%%\n",
8043 		    "Special class", (u_longlong_t)alloc,
8044 		    100.0 * alloc / space);
8045 	}
8046 
8047 	if (spa_dedup_class(spa)->mc_allocator[0].mca_rotor != NULL) {
8048 		uint64_t alloc = metaslab_class_get_alloc(
8049 		    spa_dedup_class(spa));
8050 		uint64_t space = metaslab_class_get_space(
8051 		    spa_dedup_class(spa));
8052 
8053 		(void) printf("\t%-16s %14llu     used: %5.2f%%\n",
8054 		    "Dedup class", (u_longlong_t)alloc,
8055 		    100.0 * alloc / space);
8056 	}
8057 
8058 	if (spa_embedded_log_class(spa)->mc_allocator[0].mca_rotor != NULL) {
8059 		uint64_t alloc = metaslab_class_get_alloc(
8060 		    spa_embedded_log_class(spa));
8061 		uint64_t space = metaslab_class_get_space(
8062 		    spa_embedded_log_class(spa));
8063 
8064 		(void) printf("\t%-16s %14llu     used: %5.2f%%\n",
8065 		    "Embedded log class", (u_longlong_t)alloc,
8066 		    100.0 * alloc / space);
8067 	}
8068 
8069 	if (spa_special_embedded_log_class(spa)->mc_allocator[0].mca_rotor
8070 	    != NULL) {
8071 		uint64_t alloc = metaslab_class_get_alloc(
8072 		    spa_special_embedded_log_class(spa));
8073 		uint64_t space = metaslab_class_get_space(
8074 		    spa_special_embedded_log_class(spa));
8075 
8076 		(void) printf("\t%-16s %14llu     used: %5.2f%%\n",
8077 		    "Special embedded log", (u_longlong_t)alloc,
8078 		    100.0 * alloc / space);
8079 	}
8080 
8081 	for (i = 0; i < NUM_BP_EMBEDDED_TYPES; i++) {
8082 		if (zcb->zcb_embedded_blocks[i] == 0)
8083 			continue;
8084 		(void) printf("\n");
8085 		(void) printf("\tadditional, non-pointer bps of type %u: "
8086 		    "%10llu\n",
8087 		    i, (u_longlong_t)zcb->zcb_embedded_blocks[i]);
8088 
8089 		if (dump_opt['b'] >= 3) {
8090 			(void) printf("\t number of (compressed) bytes:  "
8091 			    "number of bps\n");
8092 			dump_histogram(zcb->zcb_embedded_histogram[i],
8093 			    sizeof (zcb->zcb_embedded_histogram[i]) /
8094 			    sizeof (zcb->zcb_embedded_histogram[i][0]), 0);
8095 		}
8096 	}
8097 
8098 	if (tzb->zb_ditto_samevdev != 0) {
8099 		(void) printf("\tDittoed blocks on same vdev: %llu\n",
8100 		    (longlong_t)tzb->zb_ditto_samevdev);
8101 	}
8102 	if (tzb->zb_ditto_same_ms != 0) {
8103 		(void) printf("\tDittoed blocks in same metaslab: %llu\n",
8104 		    (longlong_t)tzb->zb_ditto_same_ms);
8105 	}
8106 
8107 	for (uint64_t v = 0; v < spa->spa_root_vdev->vdev_children; v++) {
8108 		vdev_t *vd = spa->spa_root_vdev->vdev_child[v];
8109 		vdev_indirect_mapping_t *vim = vd->vdev_indirect_mapping;
8110 
8111 		if (vim == NULL) {
8112 			continue;
8113 		}
8114 
8115 		char mem[32];
8116 		zdb_nicenum(vdev_indirect_mapping_num_entries(vim),
8117 		    mem, vdev_indirect_mapping_size(vim));
8118 
8119 		(void) printf("\tindirect vdev id %llu has %llu segments "
8120 		    "(%s in memory)\n",
8121 		    (longlong_t)vd->vdev_id,
8122 		    (longlong_t)vdev_indirect_mapping_num_entries(vim), mem);
8123 	}
8124 
8125 	if (dump_opt['b'] >= 2) {
8126 		int l, t, level;
8127 		char csize[32], lsize[32], psize[32], asize[32];
8128 		char avg[32], gang[32];
8129 		(void) printf("\nBlocks\tLSIZE\tPSIZE\tASIZE"
8130 		    "\t  avg\t comp\t%%Total\tType\n");
8131 
8132 		zfs_blkstat_t *mdstats = umem_zalloc(sizeof (zfs_blkstat_t),
8133 		    UMEM_NOFAIL);
8134 
8135 		for (t = 0; t <= ZDB_OT_TOTAL; t++) {
8136 			const char *typename;
8137 
8138 			/* make sure nicenum has enough space */
8139 			_Static_assert(sizeof (csize) >= NN_NUMBUF_SZ,
8140 			    "csize truncated");
8141 			_Static_assert(sizeof (lsize) >= NN_NUMBUF_SZ,
8142 			    "lsize truncated");
8143 			_Static_assert(sizeof (psize) >= NN_NUMBUF_SZ,
8144 			    "psize truncated");
8145 			_Static_assert(sizeof (asize) >= NN_NUMBUF_SZ,
8146 			    "asize truncated");
8147 			_Static_assert(sizeof (avg) >= NN_NUMBUF_SZ,
8148 			    "avg truncated");
8149 			_Static_assert(sizeof (gang) >= NN_NUMBUF_SZ,
8150 			    "gang truncated");
8151 
8152 			if (t < DMU_OT_NUMTYPES)
8153 				typename = dmu_ot[t].ot_name;
8154 			else
8155 				typename = zdb_ot_extname[t - DMU_OT_NUMTYPES];
8156 
8157 			if (zcb->zcb_type[ZB_TOTAL][t].zb_asize == 0) {
8158 				(void) printf("%6s\t%5s\t%5s\t%5s"
8159 				    "\t%5s\t%5s\t%6s\t%s\n",
8160 				    "-",
8161 				    "-",
8162 				    "-",
8163 				    "-",
8164 				    "-",
8165 				    "-",
8166 				    "-",
8167 				    typename);
8168 				continue;
8169 			}
8170 
8171 			for (l = ZB_TOTAL - 1; l >= -1; l--) {
8172 				level = (l == -1 ? ZB_TOTAL : l);
8173 				zb = &zcb->zcb_type[level][t];
8174 
8175 				if (zb->zb_asize == 0)
8176 					continue;
8177 
8178 				if (level != ZB_TOTAL && t < DMU_OT_NUMTYPES &&
8179 				    (level > 0 || DMU_OT_IS_METADATA(t))) {
8180 					mdstats->zb_count += zb->zb_count;
8181 					mdstats->zb_lsize += zb->zb_lsize;
8182 					mdstats->zb_psize += zb->zb_psize;
8183 					mdstats->zb_asize += zb->zb_asize;
8184 					mdstats->zb_gangs += zb->zb_gangs;
8185 				}
8186 
8187 				if (dump_opt['b'] < 3 && level != ZB_TOTAL)
8188 					continue;
8189 
8190 				if (level == 0 && zb->zb_asize ==
8191 				    zcb->zcb_type[ZB_TOTAL][t].zb_asize)
8192 					continue;
8193 
8194 				zdb_nicenum(zb->zb_count, csize,
8195 				    sizeof (csize));
8196 				zdb_nicenum(zb->zb_lsize, lsize,
8197 				    sizeof (lsize));
8198 				zdb_nicenum(zb->zb_psize, psize,
8199 				    sizeof (psize));
8200 				zdb_nicenum(zb->zb_asize, asize,
8201 				    sizeof (asize));
8202 				zdb_nicenum(zb->zb_asize / zb->zb_count, avg,
8203 				    sizeof (avg));
8204 				zdb_nicenum(zb->zb_gangs, gang, sizeof (gang));
8205 
8206 				(void) printf("%6s\t%5s\t%5s\t%5s\t%5s"
8207 				    "\t%5.2f\t%6.2f\t",
8208 				    csize, lsize, psize, asize, avg,
8209 				    (double)zb->zb_lsize / zb->zb_psize,
8210 				    100.0 * zb->zb_asize / tzb->zb_asize);
8211 
8212 				if (level == ZB_TOTAL)
8213 					(void) printf("%s\n", typename);
8214 				else
8215 					(void) printf("    L%d %s\n",
8216 					    level, typename);
8217 
8218 				if (dump_opt['b'] >= 3 && zb->zb_gangs > 0) {
8219 					(void) printf("\t number of ganged "
8220 					    "blocks: %s\n", gang);
8221 				}
8222 
8223 				if (dump_opt['b'] >= 4) {
8224 					(void) printf("psize "
8225 					    "(in 512-byte sectors): "
8226 					    "number of blocks\n");
8227 					dump_histogram(zb->zb_psize_histogram,
8228 					    PSIZE_HISTO_SIZE, 0);
8229 				}
8230 			}
8231 		}
8232 		zdb_nicenum(mdstats->zb_count, csize,
8233 		    sizeof (csize));
8234 		zdb_nicenum(mdstats->zb_lsize, lsize,
8235 		    sizeof (lsize));
8236 		zdb_nicenum(mdstats->zb_psize, psize,
8237 		    sizeof (psize));
8238 		zdb_nicenum(mdstats->zb_asize, asize,
8239 		    sizeof (asize));
8240 		zdb_nicenum(mdstats->zb_asize / mdstats->zb_count, avg,
8241 		    sizeof (avg));
8242 		zdb_nicenum(mdstats->zb_gangs, gang, sizeof (gang));
8243 
8244 		(void) printf("%6s\t%5s\t%5s\t%5s\t%5s"
8245 		    "\t%5.2f\t%6.2f\t",
8246 		    csize, lsize, psize, asize, avg,
8247 		    (double)mdstats->zb_lsize / mdstats->zb_psize,
8248 		    100.0 * mdstats->zb_asize / tzb->zb_asize);
8249 		(void) printf("%s\n", "Metadata Total");
8250 
8251 		/* Output a table summarizing block sizes in the pool */
8252 		if (dump_opt['b'] >= 2) {
8253 			dump_size_histograms(zcb);
8254 		}
8255 
8256 		umem_free(mdstats, sizeof (zfs_blkstat_t));
8257 	}
8258 
8259 	(void) printf("\n");
8260 
8261 	if (leaks) {
8262 		umem_free(zcb, sizeof (zdb_cb_t));
8263 		return (2);
8264 	}
8265 
8266 	if (zcb->zcb_haderrors) {
8267 		umem_free(zcb, sizeof (zdb_cb_t));
8268 		return (3);
8269 	}
8270 
8271 	umem_free(zcb, sizeof (zdb_cb_t));
8272 	return (0);
8273 }
8274 
8275 typedef struct zdb_ddt_entry {
8276 	/* key must be first for ddt_key_compare */
8277 	ddt_key_t	zdde_key;
8278 	uint64_t	zdde_ref_blocks;
8279 	uint64_t	zdde_ref_lsize;
8280 	uint64_t	zdde_ref_psize;
8281 	uint64_t	zdde_ref_dsize;
8282 	avl_node_t	zdde_node;
8283 } zdb_ddt_entry_t;
8284 
8285 static int
zdb_ddt_add_cb(spa_t * spa,zilog_t * zilog,const blkptr_t * bp,const zbookmark_phys_t * zb,const dnode_phys_t * dnp,void * arg)8286 zdb_ddt_add_cb(spa_t *spa, zilog_t *zilog, const blkptr_t *bp,
8287     const zbookmark_phys_t *zb, const dnode_phys_t *dnp, void *arg)
8288 {
8289 	(void) zilog, (void) dnp;
8290 	avl_tree_t *t = arg;
8291 	avl_index_t where;
8292 	zdb_ddt_entry_t *zdde, zdde_search;
8293 
8294 	if (zb->zb_level == ZB_DNODE_LEVEL || BP_IS_HOLE(bp) ||
8295 	    BP_IS_EMBEDDED(bp))
8296 		return (0);
8297 
8298 	if (dump_opt['S'] > 1 && zb->zb_level == ZB_ROOT_LEVEL) {
8299 		(void) printf("traversing objset %llu, %llu objects, "
8300 		    "%lu blocks so far\n",
8301 		    (u_longlong_t)zb->zb_objset,
8302 		    (u_longlong_t)BP_GET_FILL(bp),
8303 		    avl_numnodes(t));
8304 	}
8305 
8306 	if (BP_IS_HOLE(bp) || BP_GET_CHECKSUM(bp) == ZIO_CHECKSUM_OFF ||
8307 	    BP_GET_LEVEL(bp) > 0 || DMU_OT_IS_METADATA(BP_GET_TYPE(bp)))
8308 		return (0);
8309 
8310 	ddt_key_fill(&zdde_search.zdde_key, bp);
8311 
8312 	zdde = avl_find(t, &zdde_search, &where);
8313 
8314 	if (zdde == NULL) {
8315 		zdde = umem_zalloc(sizeof (*zdde), UMEM_NOFAIL);
8316 		zdde->zdde_key = zdde_search.zdde_key;
8317 		avl_insert(t, zdde, where);
8318 	}
8319 
8320 	zdde->zdde_ref_blocks += 1;
8321 	zdde->zdde_ref_lsize += BP_GET_LSIZE(bp);
8322 	zdde->zdde_ref_psize += BP_GET_PSIZE(bp);
8323 	zdde->zdde_ref_dsize += bp_get_dsize_sync(spa, bp);
8324 
8325 	return (0);
8326 }
8327 
8328 static void
dump_simulated_ddt(spa_t * spa)8329 dump_simulated_ddt(spa_t *spa)
8330 {
8331 	avl_tree_t t;
8332 	void *cookie = NULL;
8333 	zdb_ddt_entry_t *zdde;
8334 	ddt_histogram_t ddh_total = {{{0}}};
8335 	ddt_stat_t dds_total = {0};
8336 
8337 	avl_create(&t, ddt_key_compare,
8338 	    sizeof (zdb_ddt_entry_t), offsetof(zdb_ddt_entry_t, zdde_node));
8339 
8340 	spa_config_enter(spa, SCL_CONFIG, FTAG, RW_READER);
8341 
8342 	(void) traverse_pool(spa, 0, TRAVERSE_PRE | TRAVERSE_PREFETCH_METADATA |
8343 	    TRAVERSE_NO_DECRYPT, zdb_ddt_add_cb, &t);
8344 
8345 	spa_config_exit(spa, SCL_CONFIG, FTAG);
8346 
8347 	while ((zdde = avl_destroy_nodes(&t, &cookie)) != NULL) {
8348 		uint64_t refcnt = zdde->zdde_ref_blocks;
8349 		ASSERT(refcnt != 0);
8350 
8351 		ddt_stat_t *dds = &ddh_total.ddh_stat[highbit64(refcnt) - 1];
8352 
8353 		dds->dds_blocks += zdde->zdde_ref_blocks / refcnt;
8354 		dds->dds_lsize += zdde->zdde_ref_lsize / refcnt;
8355 		dds->dds_psize += zdde->zdde_ref_psize / refcnt;
8356 		dds->dds_dsize += zdde->zdde_ref_dsize / refcnt;
8357 
8358 		dds->dds_ref_blocks += zdde->zdde_ref_blocks;
8359 		dds->dds_ref_lsize += zdde->zdde_ref_lsize;
8360 		dds->dds_ref_psize += zdde->zdde_ref_psize;
8361 		dds->dds_ref_dsize += zdde->zdde_ref_dsize;
8362 
8363 		umem_free(zdde, sizeof (*zdde));
8364 	}
8365 
8366 	avl_destroy(&t);
8367 
8368 	ddt_histogram_total(&dds_total, &ddh_total);
8369 
8370 	(void) printf("Simulated DDT histogram:\n");
8371 
8372 	zpool_dump_ddt(&dds_total, &ddh_total, dump_opt['P'] > 0);
8373 
8374 	dump_dedup_ratio(&dds_total);
8375 }
8376 
8377 static int
verify_device_removal_feature_counts(spa_t * spa)8378 verify_device_removal_feature_counts(spa_t *spa)
8379 {
8380 	uint64_t dr_feature_refcount = 0;
8381 	uint64_t oc_feature_refcount = 0;
8382 	uint64_t indirect_vdev_count = 0;
8383 	uint64_t precise_vdev_count = 0;
8384 	uint64_t obsolete_counts_object_count = 0;
8385 	uint64_t obsolete_sm_count = 0;
8386 	uint64_t obsolete_counts_count = 0;
8387 	uint64_t scip_count = 0;
8388 	uint64_t obsolete_bpobj_count = 0;
8389 	int ret = 0;
8390 
8391 	spa_condensing_indirect_phys_t *scip =
8392 	    &spa->spa_condensing_indirect_phys;
8393 	if (scip->scip_next_mapping_object != 0) {
8394 		vdev_t *vd = spa->spa_root_vdev->vdev_child[scip->scip_vdev];
8395 		ASSERT(scip->scip_prev_obsolete_sm_object != 0);
8396 		ASSERT3P(vd->vdev_ops, ==, &vdev_indirect_ops);
8397 
8398 		(void) printf("Condensing indirect vdev %llu: new mapping "
8399 		    "object %llu, prev obsolete sm %llu\n",
8400 		    (u_longlong_t)scip->scip_vdev,
8401 		    (u_longlong_t)scip->scip_next_mapping_object,
8402 		    (u_longlong_t)scip->scip_prev_obsolete_sm_object);
8403 		if (scip->scip_prev_obsolete_sm_object != 0) {
8404 			space_map_t *prev_obsolete_sm = NULL;
8405 			VERIFY0(space_map_open(&prev_obsolete_sm,
8406 			    spa->spa_meta_objset,
8407 			    scip->scip_prev_obsolete_sm_object,
8408 			    0, vd->vdev_asize, 0));
8409 			dump_spacemap(spa->spa_meta_objset, prev_obsolete_sm);
8410 			(void) printf("\n");
8411 			space_map_close(prev_obsolete_sm);
8412 		}
8413 
8414 		scip_count += 2;
8415 	}
8416 
8417 	for (uint64_t i = 0; i < spa->spa_root_vdev->vdev_children; i++) {
8418 		vdev_t *vd = spa->spa_root_vdev->vdev_child[i];
8419 		vdev_indirect_config_t *vic = &vd->vdev_indirect_config;
8420 
8421 		if (vic->vic_mapping_object != 0) {
8422 			ASSERT(vd->vdev_ops == &vdev_indirect_ops ||
8423 			    vd->vdev_removing);
8424 			indirect_vdev_count++;
8425 
8426 			if (vd->vdev_indirect_mapping->vim_havecounts) {
8427 				obsolete_counts_count++;
8428 			}
8429 		}
8430 
8431 		boolean_t are_precise;
8432 		VERIFY0(vdev_obsolete_counts_are_precise(vd, &are_precise));
8433 		if (are_precise) {
8434 			ASSERT(vic->vic_mapping_object != 0);
8435 			precise_vdev_count++;
8436 		}
8437 
8438 		uint64_t obsolete_sm_object;
8439 		VERIFY0(vdev_obsolete_sm_object(vd, &obsolete_sm_object));
8440 		if (obsolete_sm_object != 0) {
8441 			ASSERT(vic->vic_mapping_object != 0);
8442 			obsolete_sm_count++;
8443 		}
8444 	}
8445 
8446 	(void) feature_get_refcount(spa,
8447 	    &spa_feature_table[SPA_FEATURE_DEVICE_REMOVAL],
8448 	    &dr_feature_refcount);
8449 	(void) feature_get_refcount(spa,
8450 	    &spa_feature_table[SPA_FEATURE_OBSOLETE_COUNTS],
8451 	    &oc_feature_refcount);
8452 
8453 	if (dr_feature_refcount != indirect_vdev_count) {
8454 		ret = 1;
8455 		(void) printf("Number of indirect vdevs (%llu) " \
8456 		    "does not match feature count (%llu)\n",
8457 		    (u_longlong_t)indirect_vdev_count,
8458 		    (u_longlong_t)dr_feature_refcount);
8459 	} else {
8460 		(void) printf("Verified device_removal feature refcount " \
8461 		    "of %llu is correct\n",
8462 		    (u_longlong_t)dr_feature_refcount);
8463 	}
8464 
8465 	if (zap_contains(spa_meta_objset(spa), DMU_POOL_DIRECTORY_OBJECT,
8466 	    DMU_POOL_OBSOLETE_BPOBJ) == 0) {
8467 		obsolete_bpobj_count++;
8468 	}
8469 
8470 
8471 	obsolete_counts_object_count = precise_vdev_count;
8472 	obsolete_counts_object_count += obsolete_sm_count;
8473 	obsolete_counts_object_count += obsolete_counts_count;
8474 	obsolete_counts_object_count += scip_count;
8475 	obsolete_counts_object_count += obsolete_bpobj_count;
8476 	obsolete_counts_object_count += remap_deadlist_count;
8477 
8478 	if (oc_feature_refcount != obsolete_counts_object_count) {
8479 		ret = 1;
8480 		(void) printf("Number of obsolete counts objects (%llu) " \
8481 		    "does not match feature count (%llu)\n",
8482 		    (u_longlong_t)obsolete_counts_object_count,
8483 		    (u_longlong_t)oc_feature_refcount);
8484 		(void) printf("pv:%llu os:%llu oc:%llu sc:%llu "
8485 		    "ob:%llu rd:%llu\n",
8486 		    (u_longlong_t)precise_vdev_count,
8487 		    (u_longlong_t)obsolete_sm_count,
8488 		    (u_longlong_t)obsolete_counts_count,
8489 		    (u_longlong_t)scip_count,
8490 		    (u_longlong_t)obsolete_bpobj_count,
8491 		    (u_longlong_t)remap_deadlist_count);
8492 	} else {
8493 		(void) printf("Verified indirect_refcount feature refcount " \
8494 		    "of %llu is correct\n",
8495 		    (u_longlong_t)oc_feature_refcount);
8496 	}
8497 	return (ret);
8498 }
8499 
8500 static void
zdb_set_skip_mmp(char * target)8501 zdb_set_skip_mmp(char *target)
8502 {
8503 	spa_t *spa;
8504 
8505 	/*
8506 	 * Disable the activity check to allow examination of
8507 	 * active pools.
8508 	 */
8509 	spa_namespace_enter(FTAG);
8510 	if ((spa = spa_lookup(target)) != NULL) {
8511 		spa->spa_import_flags |= ZFS_IMPORT_SKIP_MMP;
8512 	}
8513 	spa_namespace_exit(FTAG);
8514 }
8515 
8516 #define	BOGUS_SUFFIX "_CHECKPOINTED_UNIVERSE"
8517 /*
8518  * Import the checkpointed state of the pool specified by the target
8519  * parameter as readonly. The function also accepts a pool config
8520  * as an optional parameter, else it attempts to infer the config by
8521  * the name of the target pool.
8522  *
8523  * Note that the checkpointed state's pool name will be the name of
8524  * the original pool with the above suffix appended to it. In addition,
8525  * if the target is not a pool name (e.g. a path to a dataset) then
8526  * the new_path parameter is populated with the updated path to
8527  * reflect the fact that we are looking into the checkpointed state.
8528  *
8529  * The function returns a newly-allocated copy of the name of the
8530  * pool containing the checkpointed state. When this copy is no
8531  * longer needed it should be freed with free(3C). Same thing
8532  * applies to the new_path parameter if allocated.
8533  */
8534 static char *
import_checkpointed_state(char * target,nvlist_t * cfg,boolean_t target_is_spa,char ** new_path)8535 import_checkpointed_state(char *target, nvlist_t *cfg, boolean_t target_is_spa,
8536     char **new_path)
8537 {
8538 	int error = 0;
8539 	char *poolname, *bogus_name = NULL;
8540 	boolean_t freecfg = B_FALSE;
8541 
8542 	/* If the target is not a pool, the extract the pool name */
8543 	char *path_start = strchr(target, '/');
8544 	if (target_is_spa || path_start == NULL) {
8545 		poolname = target;
8546 	} else {
8547 		size_t poolname_len = path_start - target;
8548 		poolname = strndup(target, poolname_len);
8549 	}
8550 
8551 	if (cfg == NULL) {
8552 		zdb_set_skip_mmp(poolname);
8553 		error = spa_get_stats(poolname, &cfg, NULL, 0);
8554 		if (error != 0) {
8555 			fatal("Tried to read config of pool \"%s\" but "
8556 			    "spa_get_stats() failed with error %d\n",
8557 			    poolname, error);
8558 		}
8559 		freecfg = B_TRUE;
8560 	}
8561 
8562 	if (asprintf(&bogus_name, "%s%s", poolname, BOGUS_SUFFIX) == -1) {
8563 		if (target != poolname)
8564 			free(poolname);
8565 		return (NULL);
8566 	}
8567 	fnvlist_add_string(cfg, ZPOOL_CONFIG_POOL_NAME, bogus_name);
8568 
8569 	error = spa_import(bogus_name, cfg, NULL,
8570 	    ZFS_IMPORT_MISSING_LOG | ZFS_IMPORT_CHECKPOINT |
8571 	    ZFS_IMPORT_SKIP_MMP);
8572 	if (freecfg)
8573 		nvlist_free(cfg);
8574 	if (error != 0) {
8575 		fatal("Tried to import pool \"%s\" but spa_import() failed "
8576 		    "with error %d\n", bogus_name, error);
8577 	}
8578 
8579 	if (new_path != NULL && !target_is_spa) {
8580 		if (asprintf(new_path, "%s%s", bogus_name,
8581 		    path_start != NULL ? path_start : "") == -1) {
8582 			free(bogus_name);
8583 			if (!target_is_spa && path_start != NULL)
8584 				free(poolname);
8585 			return (NULL);
8586 		}
8587 	}
8588 
8589 	if (target != poolname)
8590 		free(poolname);
8591 
8592 	return (bogus_name);
8593 }
8594 
8595 typedef struct verify_checkpoint_sm_entry_cb_arg {
8596 	vdev_t *vcsec_vd;
8597 
8598 	/* the following fields are only used for printing progress */
8599 	uint64_t vcsec_entryid;
8600 	uint64_t vcsec_num_entries;
8601 } verify_checkpoint_sm_entry_cb_arg_t;
8602 
8603 #define	ENTRIES_PER_PROGRESS_UPDATE 10000
8604 
8605 static int
verify_checkpoint_sm_entry_cb(space_map_entry_t * sme,void * arg)8606 verify_checkpoint_sm_entry_cb(space_map_entry_t *sme, void *arg)
8607 {
8608 	verify_checkpoint_sm_entry_cb_arg_t *vcsec = arg;
8609 	vdev_t *vd = vcsec->vcsec_vd;
8610 	metaslab_t *ms = vd->vdev_ms[sme->sme_offset >> vd->vdev_ms_shift];
8611 	uint64_t end = sme->sme_offset + sme->sme_run;
8612 
8613 	ASSERT(sme->sme_type == SM_FREE);
8614 
8615 	if ((vcsec->vcsec_entryid % ENTRIES_PER_PROGRESS_UPDATE) == 0) {
8616 		(void) fprintf(stderr,
8617 		    "\rverifying vdev %llu, space map entry %llu of %llu ...",
8618 		    (longlong_t)vd->vdev_id,
8619 		    (longlong_t)vcsec->vcsec_entryid,
8620 		    (longlong_t)vcsec->vcsec_num_entries);
8621 	}
8622 	vcsec->vcsec_entryid++;
8623 
8624 	/*
8625 	 * See comment in checkpoint_sm_exclude_entry_cb()
8626 	 */
8627 	VERIFY3U(sme->sme_offset, >=, ms->ms_start);
8628 	VERIFY3U(end, <=, ms->ms_start + ms->ms_size);
8629 
8630 	/*
8631 	 * The entries in the vdev_checkpoint_sm should be marked as
8632 	 * allocated in the checkpointed state of the pool, therefore
8633 	 * their respective ms_allocateable trees should not contain them.
8634 	 */
8635 	mutex_enter(&ms->ms_lock);
8636 	zfs_range_tree_verify_not_present(ms->ms_allocatable,
8637 	    sme->sme_offset, sme->sme_run);
8638 	mutex_exit(&ms->ms_lock);
8639 
8640 	return (0);
8641 }
8642 
8643 /*
8644  * Verify that all segments in the vdev_checkpoint_sm are allocated
8645  * according to the checkpoint's ms_sm (i.e. are not in the checkpoint's
8646  * ms_allocatable).
8647  *
8648  * Do so by comparing the checkpoint space maps (vdev_checkpoint_sm) of
8649  * each vdev in the current state of the pool to the metaslab space maps
8650  * (ms_sm) of the checkpointed state of the pool.
8651  *
8652  * Note that the function changes the state of the ms_allocatable
8653  * trees of the current spa_t. The entries of these ms_allocatable
8654  * trees are cleared out and then repopulated from with the free
8655  * entries of their respective ms_sm space maps.
8656  */
8657 static void
verify_checkpoint_vdev_spacemaps(spa_t * checkpoint,spa_t * current)8658 verify_checkpoint_vdev_spacemaps(spa_t *checkpoint, spa_t *current)
8659 {
8660 	vdev_t *ckpoint_rvd = checkpoint->spa_root_vdev;
8661 	vdev_t *current_rvd = current->spa_root_vdev;
8662 
8663 	load_concrete_ms_allocatable_trees(checkpoint, SM_FREE);
8664 
8665 	for (uint64_t c = 0; c < ckpoint_rvd->vdev_children; c++) {
8666 		vdev_t *ckpoint_vd = ckpoint_rvd->vdev_child[c];
8667 		vdev_t *current_vd = current_rvd->vdev_child[c];
8668 
8669 		space_map_t *checkpoint_sm = NULL;
8670 		uint64_t checkpoint_sm_obj;
8671 
8672 		if (ckpoint_vd->vdev_ops == &vdev_indirect_ops) {
8673 			/*
8674 			 * Since we don't allow device removal in a pool
8675 			 * that has a checkpoint, we expect that all removed
8676 			 * vdevs were removed from the pool before the
8677 			 * checkpoint.
8678 			 */
8679 			ASSERT3P(current_vd->vdev_ops, ==, &vdev_indirect_ops);
8680 			continue;
8681 		}
8682 
8683 		/*
8684 		 * If the checkpoint space map doesn't exist, then nothing
8685 		 * here is checkpointed so there's nothing to verify.
8686 		 */
8687 		if (current_vd->vdev_top_zap == 0 ||
8688 		    zap_contains(spa_meta_objset(current),
8689 		    current_vd->vdev_top_zap,
8690 		    VDEV_TOP_ZAP_POOL_CHECKPOINT_SM) != 0)
8691 			continue;
8692 
8693 		VERIFY0(zap_lookup(spa_meta_objset(current),
8694 		    current_vd->vdev_top_zap, VDEV_TOP_ZAP_POOL_CHECKPOINT_SM,
8695 		    sizeof (uint64_t), 1, &checkpoint_sm_obj));
8696 
8697 		VERIFY0(space_map_open(&checkpoint_sm, spa_meta_objset(current),
8698 		    checkpoint_sm_obj, 0, current_vd->vdev_asize,
8699 		    current_vd->vdev_ashift));
8700 
8701 		verify_checkpoint_sm_entry_cb_arg_t vcsec;
8702 		vcsec.vcsec_vd = ckpoint_vd;
8703 		vcsec.vcsec_entryid = 0;
8704 		vcsec.vcsec_num_entries =
8705 		    space_map_length(checkpoint_sm) / sizeof (uint64_t);
8706 		VERIFY0(space_map_iterate(checkpoint_sm,
8707 		    space_map_length(checkpoint_sm),
8708 		    verify_checkpoint_sm_entry_cb, &vcsec));
8709 		if (dump_opt['m'] > 3)
8710 			dump_spacemap(current->spa_meta_objset, checkpoint_sm);
8711 		space_map_close(checkpoint_sm);
8712 	}
8713 
8714 	/*
8715 	 * If we've added vdevs since we took the checkpoint, ensure
8716 	 * that their checkpoint space maps are empty.
8717 	 */
8718 	if (ckpoint_rvd->vdev_children < current_rvd->vdev_children) {
8719 		for (uint64_t c = ckpoint_rvd->vdev_children;
8720 		    c < current_rvd->vdev_children; c++) {
8721 			vdev_t *current_vd = current_rvd->vdev_child[c];
8722 			VERIFY0P(current_vd->vdev_checkpoint_sm);
8723 		}
8724 	}
8725 
8726 	/* for cleaner progress output */
8727 	(void) fprintf(stderr, "\n");
8728 }
8729 
8730 /*
8731  * Verifies that all space that's allocated in the checkpoint is
8732  * still allocated in the current version, by checking that everything
8733  * in checkpoint's ms_allocatable (which is actually allocated, not
8734  * allocatable/free) is not present in current's ms_allocatable.
8735  *
8736  * Note that the function changes the state of the ms_allocatable
8737  * trees of both spas when called. The entries of all ms_allocatable
8738  * trees are cleared out and then repopulated from their respective
8739  * ms_sm space maps. In the checkpointed state we load the allocated
8740  * entries, and in the current state we load the free entries.
8741  */
8742 static void
verify_checkpoint_ms_spacemaps(spa_t * checkpoint,spa_t * current)8743 verify_checkpoint_ms_spacemaps(spa_t *checkpoint, spa_t *current)
8744 {
8745 	vdev_t *ckpoint_rvd = checkpoint->spa_root_vdev;
8746 	vdev_t *current_rvd = current->spa_root_vdev;
8747 
8748 	load_concrete_ms_allocatable_trees(checkpoint, SM_ALLOC);
8749 	load_concrete_ms_allocatable_trees(current, SM_FREE);
8750 
8751 	for (uint64_t i = 0; i < ckpoint_rvd->vdev_children; i++) {
8752 		vdev_t *ckpoint_vd = ckpoint_rvd->vdev_child[i];
8753 		vdev_t *current_vd = current_rvd->vdev_child[i];
8754 
8755 		if (ckpoint_vd->vdev_ops == &vdev_indirect_ops) {
8756 			/*
8757 			 * See comment in verify_checkpoint_vdev_spacemaps()
8758 			 */
8759 			ASSERT3P(current_vd->vdev_ops, ==, &vdev_indirect_ops);
8760 			continue;
8761 		}
8762 
8763 		for (uint64_t m = 0; m < ckpoint_vd->vdev_ms_count; m++) {
8764 			metaslab_t *ckpoint_msp = ckpoint_vd->vdev_ms[m];
8765 			metaslab_t *current_msp = current_vd->vdev_ms[m];
8766 
8767 			(void) fprintf(stderr,
8768 			    "\rverifying vdev %llu of %llu, "
8769 			    "metaslab %llu of %llu ...",
8770 			    (longlong_t)current_vd->vdev_id,
8771 			    (longlong_t)current_rvd->vdev_children,
8772 			    (longlong_t)current_vd->vdev_ms[m]->ms_id,
8773 			    (longlong_t)current_vd->vdev_ms_count);
8774 
8775 			/*
8776 			 * We walk through the ms_allocatable trees that
8777 			 * are loaded with the allocated blocks from the
8778 			 * ms_sm spacemaps of the checkpoint. For each
8779 			 * one of these ranges we ensure that none of them
8780 			 * exists in the ms_allocatable trees of the
8781 			 * current state which are loaded with the ranges
8782 			 * that are currently free.
8783 			 *
8784 			 * This way we ensure that none of the blocks that
8785 			 * are part of the checkpoint were freed by mistake.
8786 			 */
8787 			zfs_range_tree_walk(ckpoint_msp->ms_allocatable,
8788 			    (zfs_range_tree_func_t *)
8789 			    zfs_range_tree_verify_not_present,
8790 			    current_msp->ms_allocatable);
8791 		}
8792 	}
8793 
8794 	/* for cleaner progress output */
8795 	(void) fprintf(stderr, "\n");
8796 }
8797 
8798 static void
verify_checkpoint_blocks(spa_t * spa)8799 verify_checkpoint_blocks(spa_t *spa)
8800 {
8801 	ASSERT(!dump_opt['L']);
8802 
8803 	spa_t *checkpoint_spa;
8804 	char *checkpoint_pool;
8805 	int error = 0;
8806 
8807 	/*
8808 	 * We import the checkpointed state of the pool (under a different
8809 	 * name) so we can do verification on it against the current state
8810 	 * of the pool.
8811 	 */
8812 	checkpoint_pool = import_checkpointed_state(spa->spa_name, NULL, B_TRUE,
8813 	    NULL);
8814 	ASSERT(strcmp(spa->spa_name, checkpoint_pool) != 0);
8815 
8816 	error = spa_open(checkpoint_pool, &checkpoint_spa, FTAG);
8817 	if (error != 0) {
8818 		fatal("Tried to open pool \"%s\" but spa_open() failed with "
8819 		    "error %d\n", checkpoint_pool, error);
8820 	}
8821 
8822 	/*
8823 	 * Ensure that ranges in the checkpoint space maps of each vdev
8824 	 * are allocated according to the checkpointed state's metaslab
8825 	 * space maps.
8826 	 */
8827 	verify_checkpoint_vdev_spacemaps(checkpoint_spa, spa);
8828 
8829 	/*
8830 	 * Ensure that allocated ranges in the checkpoint's metaslab
8831 	 * space maps remain allocated in the metaslab space maps of
8832 	 * the current state.
8833 	 */
8834 	verify_checkpoint_ms_spacemaps(checkpoint_spa, spa);
8835 
8836 	/*
8837 	 * Once we are done, we get rid of the checkpointed state.
8838 	 */
8839 	spa_close(checkpoint_spa, FTAG);
8840 	free(checkpoint_pool);
8841 }
8842 
8843 static void
dump_leftover_checkpoint_blocks(spa_t * spa)8844 dump_leftover_checkpoint_blocks(spa_t *spa)
8845 {
8846 	vdev_t *rvd = spa->spa_root_vdev;
8847 
8848 	for (uint64_t i = 0; i < rvd->vdev_children; i++) {
8849 		vdev_t *vd = rvd->vdev_child[i];
8850 
8851 		space_map_t *checkpoint_sm = NULL;
8852 		uint64_t checkpoint_sm_obj;
8853 
8854 		if (vd->vdev_top_zap == 0)
8855 			continue;
8856 
8857 		if (zap_contains(spa_meta_objset(spa), vd->vdev_top_zap,
8858 		    VDEV_TOP_ZAP_POOL_CHECKPOINT_SM) != 0)
8859 			continue;
8860 
8861 		VERIFY0(zap_lookup(spa_meta_objset(spa), vd->vdev_top_zap,
8862 		    VDEV_TOP_ZAP_POOL_CHECKPOINT_SM,
8863 		    sizeof (uint64_t), 1, &checkpoint_sm_obj));
8864 
8865 		VERIFY0(space_map_open(&checkpoint_sm, spa_meta_objset(spa),
8866 		    checkpoint_sm_obj, 0, vd->vdev_asize, vd->vdev_ashift));
8867 		dump_spacemap(spa->spa_meta_objset, checkpoint_sm);
8868 		space_map_close(checkpoint_sm);
8869 	}
8870 }
8871 
8872 static int
verify_checkpoint(spa_t * spa)8873 verify_checkpoint(spa_t *spa)
8874 {
8875 	uberblock_t checkpoint;
8876 	int error;
8877 
8878 	if (!spa_feature_is_active(spa, SPA_FEATURE_POOL_CHECKPOINT))
8879 		return (0);
8880 
8881 	error = zap_lookup(spa->spa_meta_objset, DMU_POOL_DIRECTORY_OBJECT,
8882 	    DMU_POOL_ZPOOL_CHECKPOINT, sizeof (uint64_t),
8883 	    sizeof (uberblock_t) / sizeof (uint64_t), &checkpoint);
8884 
8885 	if (error == ENOENT && !dump_opt['L']) {
8886 		/*
8887 		 * If the feature is active but the uberblock is missing
8888 		 * then we must be in the middle of discarding the
8889 		 * checkpoint.
8890 		 */
8891 		(void) printf("\nPartially discarded checkpoint "
8892 		    "state found:\n");
8893 		if (dump_opt['m'] > 3)
8894 			dump_leftover_checkpoint_blocks(spa);
8895 		return (0);
8896 	} else if (error != 0) {
8897 		(void) printf("lookup error %d when looking for "
8898 		    "checkpointed uberblock in MOS\n", error);
8899 		return (error);
8900 	}
8901 	dump_uberblock(&checkpoint, "\nCheckpointed uberblock found:\n", "\n");
8902 
8903 	if (checkpoint.ub_checkpoint_txg == 0) {
8904 		(void) printf("\nub_checkpoint_txg not set in checkpointed "
8905 		    "uberblock\n");
8906 		error = 3;
8907 	}
8908 
8909 	if (error == 0 && !dump_opt['L'])
8910 		verify_checkpoint_blocks(spa);
8911 
8912 	return (error);
8913 }
8914 
8915 static void
mos_leaks_cb(void * arg,uint64_t start,uint64_t size)8916 mos_leaks_cb(void *arg, uint64_t start, uint64_t size)
8917 {
8918 	(void) arg;
8919 	for (uint64_t i = start; i < size; i++) {
8920 		(void) printf("MOS object %llu referenced but not allocated\n",
8921 		    (u_longlong_t)i);
8922 	}
8923 }
8924 
8925 static void
mos_obj_refd(uint64_t obj)8926 mos_obj_refd(uint64_t obj)
8927 {
8928 	if (obj != 0 && mos_refd_objs != NULL)
8929 		zfs_range_tree_add(mos_refd_objs, obj, 1);
8930 }
8931 
8932 /*
8933  * Call on a MOS object that may already have been referenced.
8934  */
8935 static void
mos_obj_refd_multiple(uint64_t obj)8936 mos_obj_refd_multiple(uint64_t obj)
8937 {
8938 	if (obj != 0 && mos_refd_objs != NULL &&
8939 	    !zfs_range_tree_contains(mos_refd_objs, obj, 1))
8940 		zfs_range_tree_add(mos_refd_objs, obj, 1);
8941 }
8942 
8943 static void
dump_mos_leaked_object_details(objset_t * mos,uint64_t object,const dmu_object_info_t * doi)8944 dump_mos_leaked_object_details(objset_t *mos, uint64_t object,
8945     const dmu_object_info_t *doi)
8946 {
8947 	if (dump_opt['d'] < 2)
8948 		return;
8949 
8950 	if (doi->doi_type == DMU_OT_DSL_CLONES) {
8951 		uint64_t entries = 0;
8952 		int error = zap_count(mos, object, &entries);
8953 		if (error == 0) {
8954 			(void) printf("\tleak detail: clone entries=%llu\n",
8955 			    (u_longlong_t)entries);
8956 		} else {
8957 			(void) printf("\tleak detail: clone entry count "
8958 			    "failed: %s\n", strerror(error));
8959 		}
8960 		return;
8961 	}
8962 
8963 	if (doi->doi_type == DMU_OT_SPACE_MAP &&
8964 	    doi->doi_bonus_size == sizeof (space_map_phys_t)) {
8965 		dmu_buf_t *db = NULL;
8966 		int error = dmu_bonus_hold(mos, object, FTAG, &db);
8967 		if (error != 0) {
8968 			(void) printf("\tleak detail: space map bonus hold "
8969 			    "failed: %s\n", strerror(error));
8970 			return;
8971 		}
8972 
8973 		space_map_phys_t *smp = db->db_data;
8974 		(void) printf("\tleak detail: smp_alloc=0x%llx "
8975 		    "smp_length=0x%llx\n", (u_longlong_t)smp->smp_alloc,
8976 		    (u_longlong_t)smp->smp_length);
8977 		dmu_buf_rele(db, FTAG);
8978 	}
8979 }
8980 
8981 static void
mos_leak_vdev_top_zap(vdev_t * vd)8982 mos_leak_vdev_top_zap(vdev_t *vd)
8983 {
8984 	uint64_t ms_flush_data_obj;
8985 	int error = zap_lookup(spa_meta_objset(vd->vdev_spa),
8986 	    vd->vdev_top_zap, VDEV_TOP_ZAP_MS_UNFLUSHED_PHYS_TXGS,
8987 	    sizeof (ms_flush_data_obj), 1, &ms_flush_data_obj);
8988 	if (error == ENOENT)
8989 		return;
8990 	ASSERT0(error);
8991 
8992 	mos_obj_refd(ms_flush_data_obj);
8993 }
8994 
8995 static void
mos_leak_vdev(vdev_t * vd)8996 mos_leak_vdev(vdev_t *vd)
8997 {
8998 	mos_obj_refd(vd->vdev_dtl_object);
8999 	mos_obj_refd(vd->vdev_ms_array);
9000 	mos_obj_refd(vd->vdev_indirect_config.vic_births_object);
9001 	mos_obj_refd(vd->vdev_indirect_config.vic_mapping_object);
9002 	mos_obj_refd(vd->vdev_leaf_zap);
9003 	if (vd->vdev_checkpoint_sm != NULL)
9004 		mos_obj_refd(vd->vdev_checkpoint_sm->sm_object);
9005 	if (vd->vdev_indirect_mapping != NULL) {
9006 		mos_obj_refd(vd->vdev_indirect_mapping->
9007 		    vim_phys->vimp_counts_object);
9008 	}
9009 	if (vd->vdev_obsolete_sm != NULL)
9010 		mos_obj_refd(vd->vdev_obsolete_sm->sm_object);
9011 
9012 	for (uint64_t m = 0; m < vd->vdev_ms_count; m++) {
9013 		metaslab_t *ms = vd->vdev_ms[m];
9014 		mos_obj_refd(space_map_object(ms->ms_sm));
9015 	}
9016 
9017 	if (vd->vdev_root_zap != 0)
9018 		mos_obj_refd(vd->vdev_root_zap);
9019 
9020 	if (vd->vdev_top_zap != 0) {
9021 		mos_obj_refd(vd->vdev_top_zap);
9022 		mos_leak_vdev_top_zap(vd);
9023 	}
9024 
9025 	for (uint64_t c = 0; c < vd->vdev_children; c++) {
9026 		mos_leak_vdev(vd->vdev_child[c]);
9027 	}
9028 }
9029 
9030 static void
mos_leak_log_spacemaps(spa_t * spa)9031 mos_leak_log_spacemaps(spa_t *spa)
9032 {
9033 	uint64_t spacemap_zap;
9034 	int error = zap_lookup(spa_meta_objset(spa),
9035 	    DMU_POOL_DIRECTORY_OBJECT, DMU_POOL_LOG_SPACEMAP_ZAP,
9036 	    sizeof (spacemap_zap), 1, &spacemap_zap);
9037 	if (error == ENOENT)
9038 		return;
9039 	ASSERT0(error);
9040 
9041 	mos_obj_refd(spacemap_zap);
9042 	for (spa_log_sm_t *sls = avl_first(&spa->spa_sm_logs_by_txg);
9043 	    sls; sls = AVL_NEXT(&spa->spa_sm_logs_by_txg, sls))
9044 		mos_obj_refd(sls->sls_sm_obj);
9045 }
9046 
9047 static void
errorlog_count_refd(objset_t * mos,uint64_t errlog)9048 errorlog_count_refd(objset_t *mos, uint64_t errlog)
9049 {
9050 	zap_cursor_t zc;
9051 	zap_attribute_t *za = zap_attribute_alloc();
9052 	for (zap_cursor_init(&zc, mos, errlog);
9053 	    zap_cursor_retrieve(&zc, za) == 0;
9054 	    zap_cursor_advance(&zc)) {
9055 		mos_obj_refd(za->za_first_integer);
9056 	}
9057 	zap_cursor_fini(&zc);
9058 	zap_attribute_free(za);
9059 }
9060 
9061 static int
dump_mos_leaks(spa_t * spa)9062 dump_mos_leaks(spa_t *spa)
9063 {
9064 	int rv = 0;
9065 	objset_t *mos = spa->spa_meta_objset;
9066 	dsl_pool_t *dp = spa->spa_dsl_pool;
9067 
9068 	/* Visit and mark all referenced objects in the MOS */
9069 
9070 	mos_obj_refd(DMU_POOL_DIRECTORY_OBJECT);
9071 	mos_obj_refd(spa->spa_pool_props_object);
9072 	mos_obj_refd(spa->spa_config_object);
9073 	mos_obj_refd(spa->spa_ddt_stat_object);
9074 	mos_obj_refd(spa->spa_feat_desc_obj);
9075 	mos_obj_refd(spa->spa_feat_enabled_txg_obj);
9076 	mos_obj_refd(spa->spa_feat_for_read_obj);
9077 	mos_obj_refd(spa->spa_feat_for_write_obj);
9078 	mos_obj_refd(spa->spa_history);
9079 	mos_obj_refd(spa->spa_errlog_last);
9080 	mos_obj_refd(spa->spa_errlog_scrub);
9081 
9082 	if (spa_feature_is_enabled(spa, SPA_FEATURE_HEAD_ERRLOG)) {
9083 		errorlog_count_refd(mos, spa->spa_errlog_last);
9084 		errorlog_count_refd(mos, spa->spa_errlog_scrub);
9085 	}
9086 
9087 	mos_obj_refd(spa->spa_all_vdev_zaps);
9088 	mos_obj_refd(spa->spa_dsl_pool->dp_bptree_obj);
9089 	mos_obj_refd(spa->spa_dsl_pool->dp_tmp_userrefs_obj);
9090 	mos_obj_refd(spa->spa_dsl_pool->dp_scan->scn_phys.scn_queue_obj);
9091 	bpobj_count_refd(&spa->spa_deferred_bpobj);
9092 	mos_obj_refd(dp->dp_empty_bpobj);
9093 	bpobj_count_refd(&dp->dp_obsolete_bpobj);
9094 	bpobj_count_refd(&dp->dp_free_bpobj);
9095 	mos_obj_refd(spa->spa_l2cache.sav_object);
9096 	mos_obj_refd(spa->spa_spares.sav_object);
9097 
9098 	if (spa->spa_syncing_log_sm != NULL)
9099 		mos_obj_refd(spa->spa_syncing_log_sm->sm_object);
9100 	mos_leak_log_spacemaps(spa);
9101 
9102 	mos_obj_refd(spa->spa_condensing_indirect_phys.
9103 	    scip_next_mapping_object);
9104 	mos_obj_refd(spa->spa_condensing_indirect_phys.
9105 	    scip_prev_obsolete_sm_object);
9106 	if (spa->spa_condensing_indirect_phys.scip_next_mapping_object != 0) {
9107 		vdev_indirect_mapping_t *vim =
9108 		    vdev_indirect_mapping_open(mos,
9109 		    spa->spa_condensing_indirect_phys.scip_next_mapping_object);
9110 		mos_obj_refd(vim->vim_phys->vimp_counts_object);
9111 		vdev_indirect_mapping_close(vim);
9112 	}
9113 	deleted_livelists_dump_mos(spa);
9114 
9115 	if (dp->dp_origin_snap != NULL) {
9116 		dsl_dataset_t *ds;
9117 
9118 		dsl_pool_config_enter(dp, FTAG);
9119 		VERIFY0(dsl_dataset_hold_obj(dp,
9120 		    dsl_dataset_phys(dp->dp_origin_snap)->ds_next_snap_obj,
9121 		    FTAG, &ds));
9122 		count_ds_mos_objects(ds);
9123 		dump_blkptr_list(&ds->ds_deadlist, "Deadlist");
9124 		dsl_dataset_rele(ds, FTAG);
9125 		dsl_pool_config_exit(dp, FTAG);
9126 
9127 		count_ds_mos_objects(dp->dp_origin_snap);
9128 		dump_blkptr_list(&dp->dp_origin_snap->ds_deadlist, "Deadlist");
9129 	}
9130 	count_dir_mos_objects(dp->dp_mos_dir);
9131 	if (dp->dp_free_dir != NULL)
9132 		count_dir_mos_objects(dp->dp_free_dir);
9133 	if (dp->dp_leak_dir != NULL)
9134 		count_dir_mos_objects(dp->dp_leak_dir);
9135 
9136 	mos_leak_vdev(spa->spa_root_vdev);
9137 
9138 	for (uint64_t c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
9139 		ddt_t *ddt = spa->spa_ddt[c];
9140 		if (!ddt || ddt->ddt_version == DDT_VERSION_UNCONFIGURED)
9141 			continue;
9142 
9143 		/* DDT store objects */
9144 		for (ddt_type_t type = 0; type < DDT_TYPES; type++) {
9145 			for (ddt_class_t class = 0; class < DDT_CLASSES;
9146 			    class++) {
9147 				mos_obj_refd(ddt->ddt_object[type][class]);
9148 			}
9149 		}
9150 
9151 		/* FDT container */
9152 		if (ddt->ddt_version == DDT_VERSION_FDT)
9153 			mos_obj_refd(ddt->ddt_dir_object);
9154 
9155 		/* FDT log objects */
9156 		if (ddt->ddt_flags & DDT_FLAG_LOG) {
9157 			mos_obj_refd(ddt->ddt_log[0].ddl_object);
9158 			mos_obj_refd(ddt->ddt_log[1].ddl_object);
9159 		}
9160 	}
9161 
9162 	for (uint64_t vdevid = 0; vdevid < spa->spa_brt_nvdevs; vdevid++) {
9163 		brt_vdev_t *brtvd = spa->spa_brt_vdevs[vdevid];
9164 		if (brtvd->bv_initiated) {
9165 			mos_obj_refd(brtvd->bv_mos_brtvdev);
9166 			mos_obj_refd(brtvd->bv_mos_entries);
9167 		}
9168 	}
9169 
9170 	/*
9171 	 * Visit all allocated objects and make sure they are referenced.
9172 	 */
9173 	uint64_t object = 0;
9174 	while (dmu_object_next(mos, &object, B_FALSE, 0) == 0) {
9175 		if (zfs_range_tree_contains(mos_refd_objs, object, 1)) {
9176 			zfs_range_tree_remove(mos_refd_objs, object, 1);
9177 		} else {
9178 			dmu_object_info_t doi;
9179 			const char *name;
9180 			VERIFY0(dmu_object_info(mos, object, &doi));
9181 			if (doi.doi_type & DMU_OT_NEWTYPE) {
9182 				dmu_object_byteswap_t bswap =
9183 				    DMU_OT_BYTESWAP(doi.doi_type);
9184 				name = dmu_ot_byteswap[bswap].ob_name;
9185 			} else {
9186 				name = dmu_ot[doi.doi_type].ot_name;
9187 			}
9188 
9189 			(void) printf("MOS object %llu (%s) leaked\n",
9190 			    (u_longlong_t)object, name);
9191 			dump_mos_leaked_object_details(mos, object,
9192 			    &doi);
9193 			rv = 2;
9194 		}
9195 	}
9196 	(void) zfs_range_tree_walk(mos_refd_objs, mos_leaks_cb, NULL);
9197 	if (!zfs_range_tree_is_empty(mos_refd_objs))
9198 		rv = 2;
9199 	zfs_range_tree_vacate(mos_refd_objs, NULL, NULL);
9200 	zfs_range_tree_destroy(mos_refd_objs);
9201 	return (rv);
9202 }
9203 
9204 typedef struct log_sm_obsolete_stats_arg {
9205 	uint64_t lsos_current_txg;
9206 
9207 	uint64_t lsos_total_entries;
9208 	uint64_t lsos_valid_entries;
9209 
9210 	uint64_t lsos_sm_entries;
9211 	uint64_t lsos_valid_sm_entries;
9212 } log_sm_obsolete_stats_arg_t;
9213 
9214 static int
log_spacemap_obsolete_stats_cb(spa_t * spa,space_map_entry_t * sme,uint64_t txg,void * arg)9215 log_spacemap_obsolete_stats_cb(spa_t *spa, space_map_entry_t *sme,
9216     uint64_t txg, void *arg)
9217 {
9218 	log_sm_obsolete_stats_arg_t *lsos = arg;
9219 
9220 	uint64_t offset = sme->sme_offset;
9221 	uint64_t vdev_id = sme->sme_vdev;
9222 
9223 	if (lsos->lsos_current_txg == 0) {
9224 		/* this is the first log */
9225 		lsos->lsos_current_txg = txg;
9226 	} else if (lsos->lsos_current_txg < txg) {
9227 		/* we just changed log - print stats and reset */
9228 		(void) printf("%-8llu valid entries out of %-8llu - txg %llu\n",
9229 		    (u_longlong_t)lsos->lsos_valid_sm_entries,
9230 		    (u_longlong_t)lsos->lsos_sm_entries,
9231 		    (u_longlong_t)lsos->lsos_current_txg);
9232 		lsos->lsos_valid_sm_entries = 0;
9233 		lsos->lsos_sm_entries = 0;
9234 		lsos->lsos_current_txg = txg;
9235 	}
9236 	ASSERT3U(lsos->lsos_current_txg, ==, txg);
9237 
9238 	lsos->lsos_sm_entries++;
9239 	lsos->lsos_total_entries++;
9240 
9241 	vdev_t *vd = vdev_lookup_top(spa, vdev_id);
9242 	if (!vdev_is_concrete(vd))
9243 		return (0);
9244 
9245 	metaslab_t *ms = vd->vdev_ms[offset >> vd->vdev_ms_shift];
9246 	ASSERT(sme->sme_type == SM_ALLOC || sme->sme_type == SM_FREE);
9247 
9248 	if (txg < metaslab_unflushed_txg(ms))
9249 		return (0);
9250 	lsos->lsos_valid_sm_entries++;
9251 	lsos->lsos_valid_entries++;
9252 	return (0);
9253 }
9254 
9255 static void
dump_log_spacemap_obsolete_stats(spa_t * spa)9256 dump_log_spacemap_obsolete_stats(spa_t *spa)
9257 {
9258 	if (!spa_feature_is_active(spa, SPA_FEATURE_LOG_SPACEMAP))
9259 		return;
9260 
9261 	log_sm_obsolete_stats_arg_t lsos = {0};
9262 
9263 	(void) printf("Log Space Map Obsolete Entry Statistics:\n");
9264 
9265 	iterate_through_spacemap_logs(spa,
9266 	    log_spacemap_obsolete_stats_cb, &lsos);
9267 
9268 	/* print stats for latest log */
9269 	(void) printf("%-8llu valid entries out of %-8llu - txg %llu\n",
9270 	    (u_longlong_t)lsos.lsos_valid_sm_entries,
9271 	    (u_longlong_t)lsos.lsos_sm_entries,
9272 	    (u_longlong_t)lsos.lsos_current_txg);
9273 
9274 	(void) printf("%-8llu valid entries out of %-8llu - total\n\n",
9275 	    (u_longlong_t)lsos.lsos_valid_entries,
9276 	    (u_longlong_t)lsos.lsos_total_entries);
9277 }
9278 
9279 static void
dump_zpool(spa_t * spa)9280 dump_zpool(spa_t *spa)
9281 {
9282 	dsl_pool_t *dp = spa_get_dsl(spa);
9283 	int rc = 0;
9284 
9285 	if (dump_opt['y']) {
9286 		livelist_metaslab_validate(spa);
9287 	}
9288 
9289 	if (dump_opt['S']) {
9290 		dump_simulated_ddt(spa);
9291 		return;
9292 	}
9293 
9294 	if (!dump_opt['e'] && dump_opt['C'] > 1) {
9295 		(void) printf("\nCached configuration:\n");
9296 		dump_nvlist(spa->spa_config, 8);
9297 	}
9298 
9299 	if (dump_opt['C'])
9300 		dump_config(spa);
9301 
9302 	if (dump_opt['u'])
9303 		dump_uberblock(&spa->spa_uberblock, "\nUberblock:\n", "\n");
9304 
9305 	if (dump_opt['D'])
9306 		dump_all_ddts(spa);
9307 
9308 	if (dump_opt['T'])
9309 		dump_brt(spa);
9310 
9311 	if (dump_opt['d'] > 2 || dump_opt['m'])
9312 		dump_metaslabs(spa);
9313 	if (dump_opt['M'])
9314 		dump_metaslab_groups(spa, dump_opt['M'] > 1);
9315 	if (dump_opt['d'] > 2 || dump_opt['m']) {
9316 		dump_log_spacemaps(spa);
9317 		dump_log_spacemap_obsolete_stats(spa);
9318 	}
9319 
9320 	if (dump_opt['d'] || dump_opt['i']) {
9321 		spa_feature_t f;
9322 		mos_refd_objs = zfs_range_tree_create_flags(
9323 		    NULL, ZFS_RANGE_SEG64, NULL, 0, 0,
9324 		    0, "dump_zpool:mos_refd_objs");
9325 		dump_objset(dp->dp_meta_objset);
9326 
9327 		if (dump_opt['d'] >= 3) {
9328 			dsl_pool_t *dp = spa->spa_dsl_pool;
9329 			dump_full_bpobj(&spa->spa_deferred_bpobj,
9330 			    "Deferred frees", 0);
9331 			if (spa_version(spa) >= SPA_VERSION_DEADLISTS) {
9332 				dump_full_bpobj(&dp->dp_free_bpobj,
9333 				    "Pool snapshot frees", 0);
9334 			}
9335 			if (bpobj_is_open(&dp->dp_obsolete_bpobj)) {
9336 				ASSERT(spa_feature_is_enabled(spa,
9337 				    SPA_FEATURE_DEVICE_REMOVAL));
9338 				dump_full_bpobj(&dp->dp_obsolete_bpobj,
9339 				    "Pool obsolete blocks", 0);
9340 			}
9341 
9342 			if (spa_feature_is_active(spa,
9343 			    SPA_FEATURE_ASYNC_DESTROY)) {
9344 				dump_bptree(spa->spa_meta_objset,
9345 				    dp->dp_bptree_obj,
9346 				    "Pool dataset frees");
9347 			}
9348 			dump_dtl(spa->spa_root_vdev, 0);
9349 		}
9350 
9351 		for (spa_feature_t f = 0; f < SPA_FEATURES; f++)
9352 			global_feature_count[f] = UINT64_MAX;
9353 		global_feature_count[SPA_FEATURE_REDACTION_BOOKMARKS] = 0;
9354 		global_feature_count[SPA_FEATURE_REDACTION_LIST_SPILL] = 0;
9355 		global_feature_count[SPA_FEATURE_BOOKMARK_WRITTEN] = 0;
9356 		global_feature_count[SPA_FEATURE_LIVELIST] = 0;
9357 
9358 		(void) dmu_objset_find(spa_name(spa), dump_one_objset,
9359 		    NULL, DS_FIND_SNAPSHOTS | DS_FIND_CHILDREN);
9360 
9361 		if (rc == 0 && !dump_opt['L'])
9362 			rc = dump_mos_leaks(spa);
9363 
9364 		for (f = 0; f < SPA_FEATURES; f++) {
9365 			uint64_t refcount;
9366 
9367 			uint64_t *arr;
9368 			if (!(spa_feature_table[f].fi_flags &
9369 			    ZFEATURE_FLAG_PER_DATASET)) {
9370 				if (global_feature_count[f] == UINT64_MAX)
9371 					continue;
9372 				if (!spa_feature_is_enabled(spa, f)) {
9373 					ASSERT0(global_feature_count[f]);
9374 					continue;
9375 				}
9376 				arr = global_feature_count;
9377 			} else {
9378 				if (!spa_feature_is_enabled(spa, f)) {
9379 					ASSERT0(dataset_feature_count[f]);
9380 					continue;
9381 				}
9382 				arr = dataset_feature_count;
9383 			}
9384 			if (feature_get_refcount(spa, &spa_feature_table[f],
9385 			    &refcount) == ENOTSUP)
9386 				continue;
9387 			if (arr[f] != refcount) {
9388 				(void) printf("%s feature refcount mismatch: "
9389 				    "%lld consumers != %lld refcount\n",
9390 				    spa_feature_table[f].fi_uname,
9391 				    (longlong_t)arr[f], (longlong_t)refcount);
9392 				rc = 2;
9393 			} else {
9394 				(void) printf("Verified %s feature refcount "
9395 				    "of %llu is correct\n",
9396 				    spa_feature_table[f].fi_uname,
9397 				    (longlong_t)refcount);
9398 			}
9399 		}
9400 
9401 		if (rc == 0)
9402 			rc = verify_device_removal_feature_counts(spa);
9403 	}
9404 
9405 	if (rc == 0 && (dump_opt['b'] || dump_opt['c']))
9406 		rc = dump_block_stats(spa);
9407 
9408 	if (rc == 0)
9409 		rc = verify_spacemap_refcounts(spa);
9410 
9411 	if (dump_opt['s'])
9412 		show_pool_stats(spa);
9413 
9414 	if (dump_opt['h'])
9415 		dump_history(spa);
9416 
9417 	if (rc == 0)
9418 		rc = verify_checkpoint(spa);
9419 
9420 	if (rc != 0) {
9421 		dump_debug_buffer();
9422 		zdb_exit(rc);
9423 	}
9424 }
9425 
9426 #define	ZDB_FLAG_CHECKSUM	0x0001
9427 #define	ZDB_FLAG_DECOMPRESS	0x0002
9428 #define	ZDB_FLAG_BSWAP		0x0004
9429 #define	ZDB_FLAG_GBH		0x0008
9430 #define	ZDB_FLAG_INDIRECT	0x0010
9431 #define	ZDB_FLAG_RAW		0x0020
9432 #define	ZDB_FLAG_PRINT_BLKPTR	0x0040
9433 #define	ZDB_FLAG_VERBOSE	0x0080
9434 
9435 static int flagbits[256];
9436 static char flagbitstr[16];
9437 
9438 static void
zdb_print_blkptr(const blkptr_t * bp,int flags)9439 zdb_print_blkptr(const blkptr_t *bp, int flags)
9440 {
9441 	char blkbuf[BP_SPRINTF_LEN];
9442 
9443 	if (flags & ZDB_FLAG_BSWAP)
9444 		byteswap_uint64_array((void *)bp, sizeof (blkptr_t));
9445 
9446 	snprintf_blkptr(blkbuf, sizeof (blkbuf), bp);
9447 	(void) printf("%s\n", blkbuf);
9448 }
9449 
9450 static void
zdb_dump_indirect(blkptr_t * bp,int nbps,int flags)9451 zdb_dump_indirect(blkptr_t *bp, int nbps, int flags)
9452 {
9453 	int i;
9454 
9455 	for (i = 0; i < nbps; i++)
9456 		zdb_print_blkptr(&bp[i], flags);
9457 }
9458 
9459 static void
zdb_dump_gbh(void * buf,uint64_t size,int flags)9460 zdb_dump_gbh(void *buf, uint64_t size, int flags)
9461 {
9462 	zdb_dump_indirect((blkptr_t *)buf, gbh_nblkptrs(size), flags);
9463 }
9464 
9465 static void
zdb_dump_block_raw(void * buf,uint64_t size,int flags)9466 zdb_dump_block_raw(void *buf, uint64_t size, int flags)
9467 {
9468 	if (flags & ZDB_FLAG_BSWAP)
9469 		byteswap_uint64_array(buf, size);
9470 	VERIFY(write(fileno(stdout), buf, size) == size);
9471 }
9472 
9473 static void
zdb_dump_block(char * label,void * buf,uint64_t size,int flags)9474 zdb_dump_block(char *label, void *buf, uint64_t size, int flags)
9475 {
9476 	uint64_t *d = (uint64_t *)buf;
9477 	unsigned nwords = size / sizeof (uint64_t);
9478 	int do_bswap = !!(flags & ZDB_FLAG_BSWAP);
9479 	unsigned i, j;
9480 	const char *hdr;
9481 	char *c;
9482 
9483 
9484 	if (do_bswap)
9485 		hdr = " 7 6 5 4 3 2 1 0   f e d c b a 9 8";
9486 	else
9487 		hdr = " 0 1 2 3 4 5 6 7   8 9 a b c d e f";
9488 
9489 	(void) printf("\n%s\n%6s   %s  0123456789abcdef\n", label, "", hdr);
9490 
9491 #ifdef _ZFS_LITTLE_ENDIAN
9492 	/* correct the endianness */
9493 	do_bswap = !do_bswap;
9494 #endif
9495 	for (i = 0; i < nwords; i += 2) {
9496 		(void) printf("%06llx:  %016llx  %016llx  ",
9497 		    (u_longlong_t)(i * sizeof (uint64_t)),
9498 		    (u_longlong_t)(do_bswap ? BSWAP_64(d[i]) : d[i]),
9499 		    (u_longlong_t)(do_bswap ? BSWAP_64(d[i + 1]) : d[i + 1]));
9500 
9501 		c = (char *)&d[i];
9502 		for (j = 0; j < 2 * sizeof (uint64_t); j++)
9503 			(void) printf("%c", isprint(c[j]) ? c[j] : '.');
9504 		(void) printf("\n");
9505 	}
9506 }
9507 
9508 /*
9509  * There are two acceptable formats:
9510  *	leaf_name	  - For example: c1t0d0 or /tmp/ztest.0a
9511  *	child[.child]*    - For example: 0.1.1
9512  *
9513  * The second form can be used to specify arbitrary vdevs anywhere
9514  * in the hierarchy.  For example, in a pool with a mirror of
9515  * RAID-Zs, you can specify either RAID-Z vdev with 0.0 or 0.1 .
9516  */
9517 static vdev_t *
zdb_vdev_lookup(vdev_t * vdev,const char * path)9518 zdb_vdev_lookup(vdev_t *vdev, const char *path)
9519 {
9520 	char *s, *p, *q;
9521 	unsigned i;
9522 
9523 	if (vdev == NULL)
9524 		return (NULL);
9525 
9526 	/* First, assume the x.x.x.x format */
9527 	i = strtoul(path, &s, 10);
9528 	if (s == path || (s && *s != '.' && *s != '\0'))
9529 		goto name;
9530 	if (i >= vdev->vdev_children)
9531 		return (NULL);
9532 
9533 	vdev = vdev->vdev_child[i];
9534 	if (s && *s == '\0')
9535 		return (vdev);
9536 	return (zdb_vdev_lookup(vdev, s+1));
9537 
9538 name:
9539 	for (i = 0; i < vdev->vdev_children; i++) {
9540 		vdev_t *vc = vdev->vdev_child[i];
9541 
9542 		if (vc->vdev_path == NULL) {
9543 			vc = zdb_vdev_lookup(vc, path);
9544 			if (vc == NULL)
9545 				continue;
9546 			else
9547 				return (vc);
9548 		}
9549 
9550 		p = strrchr(vc->vdev_path, '/');
9551 		p = p ? p + 1 : vc->vdev_path;
9552 		q = &vc->vdev_path[strlen(vc->vdev_path) - 2];
9553 
9554 		if (strcmp(vc->vdev_path, path) == 0)
9555 			return (vc);
9556 		if (strcmp(p, path) == 0)
9557 			return (vc);
9558 		if (strcmp(q, "s0") == 0 && strncmp(p, path, q - p) == 0)
9559 			return (vc);
9560 	}
9561 
9562 	return (NULL);
9563 }
9564 
9565 static int
name_from_objset_id(spa_t * spa,uint64_t objset_id,char * outstr)9566 name_from_objset_id(spa_t *spa, uint64_t objset_id, char *outstr)
9567 {
9568 	dsl_dataset_t *ds;
9569 
9570 	dsl_pool_config_enter(spa->spa_dsl_pool, FTAG);
9571 	int error = dsl_dataset_hold_obj(spa->spa_dsl_pool, objset_id,
9572 	    NULL, &ds);
9573 	if (error != 0) {
9574 		(void) fprintf(stderr, "failed to hold objset %llu: %s\n",
9575 		    (u_longlong_t)objset_id, strerror(error));
9576 		dsl_pool_config_exit(spa->spa_dsl_pool, FTAG);
9577 		return (error);
9578 	}
9579 	dsl_dataset_name(ds, outstr);
9580 	dsl_dataset_rele(ds, NULL);
9581 	dsl_pool_config_exit(spa->spa_dsl_pool, FTAG);
9582 	return (0);
9583 }
9584 
9585 static boolean_t
zdb_parse_block_sizes(char * sizes,uint64_t * lsize,uint64_t * psize)9586 zdb_parse_block_sizes(char *sizes, uint64_t *lsize, uint64_t *psize)
9587 {
9588 	char *s0, *s1, *tmp = NULL;
9589 
9590 	if (sizes == NULL)
9591 		return (B_FALSE);
9592 
9593 	s0 = strtok_r(sizes, "/", &tmp);
9594 	if (s0 == NULL)
9595 		return (B_FALSE);
9596 	s1 = strtok_r(NULL, "/", &tmp);
9597 	*lsize = strtoull(s0, NULL, 16);
9598 	*psize = s1 ? strtoull(s1, NULL, 16) : *lsize;
9599 	return (*lsize >= *psize && *psize > 0);
9600 }
9601 
9602 #define	ZIO_COMPRESS_MASK(alg)	(1ULL << (ZIO_COMPRESS_##alg))
9603 
9604 static boolean_t
try_decompress_block(abd_t * pabd,uint64_t lsize,uint64_t psize,int flags,int cfunc,void * lbuf,void * lbuf2)9605 try_decompress_block(abd_t *pabd, uint64_t lsize, uint64_t psize,
9606     int flags, int cfunc, void *lbuf, void *lbuf2)
9607 {
9608 	if (flags & ZDB_FLAG_VERBOSE) {
9609 		(void) fprintf(stderr,
9610 		    "Trying %05llx -> %05llx (%s)\n",
9611 		    (u_longlong_t)psize,
9612 		    (u_longlong_t)lsize,
9613 		    zio_compress_table[cfunc].ci_name);
9614 	}
9615 
9616 	/*
9617 	 * We set lbuf to all zeros and lbuf2 to all
9618 	 * ones, then decompress to both buffers and
9619 	 * compare their contents. This way we can
9620 	 * know if decompression filled exactly to
9621 	 * lsize or if it left some bytes unwritten.
9622 	 */
9623 
9624 	memset(lbuf, 0x00, lsize);
9625 	memset(lbuf2, 0xff, lsize);
9626 
9627 	abd_t labd, labd2;
9628 	abd_get_from_buf_struct(&labd, lbuf, lsize);
9629 	abd_get_from_buf_struct(&labd2, lbuf2, lsize);
9630 
9631 	boolean_t ret = B_FALSE;
9632 	if (zio_decompress_data(cfunc, pabd,
9633 	    &labd, psize, lsize, NULL) == 0 &&
9634 	    zio_decompress_data(cfunc, pabd,
9635 	    &labd2, psize, lsize, NULL) == 0 &&
9636 	    memcmp(lbuf, lbuf2, lsize) == 0)
9637 		ret = B_TRUE;
9638 
9639 	abd_free(&labd2);
9640 	abd_free(&labd);
9641 
9642 	return (ret);
9643 }
9644 
9645 static uint64_t
zdb_decompress_block(abd_t * pabd,void * buf,void * lbuf,uint64_t lsize,uint64_t psize,int flags)9646 zdb_decompress_block(abd_t *pabd, void *buf, void *lbuf, uint64_t lsize,
9647     uint64_t psize, int flags)
9648 {
9649 	(void) buf;
9650 	uint64_t orig_lsize = lsize;
9651 	boolean_t tryzle = ((getenv("ZDB_NO_ZLE") == NULL));
9652 	/*
9653 	 * We don't know how the data was compressed, so just try
9654 	 * every decompress function at every inflated blocksize.
9655 	 */
9656 	void *lbuf2 = umem_alloc(SPA_MAXBLOCKSIZE, UMEM_NOFAIL);
9657 	int cfuncs[ZIO_COMPRESS_FUNCTIONS] = { 0 };
9658 	int *cfuncp = cfuncs;
9659 	uint64_t maxlsize = SPA_MAXBLOCKSIZE;
9660 	uint64_t mask = ZIO_COMPRESS_MASK(ON) | ZIO_COMPRESS_MASK(OFF) |
9661 	    ZIO_COMPRESS_MASK(INHERIT) | ZIO_COMPRESS_MASK(EMPTY) |
9662 	    ZIO_COMPRESS_MASK(ZLE);
9663 	*cfuncp++ = ZIO_COMPRESS_LZ4;
9664 	*cfuncp++ = ZIO_COMPRESS_LZJB;
9665 	mask |= ZIO_COMPRESS_MASK(LZ4) | ZIO_COMPRESS_MASK(LZJB);
9666 	/*
9667 	 * Every gzip level has the same decompressor, no need to
9668 	 * run it 9 times per bruteforce attempt.
9669 	 */
9670 	mask |= ZIO_COMPRESS_MASK(GZIP_2) | ZIO_COMPRESS_MASK(GZIP_3);
9671 	mask |= ZIO_COMPRESS_MASK(GZIP_4) | ZIO_COMPRESS_MASK(GZIP_5);
9672 	mask |= ZIO_COMPRESS_MASK(GZIP_6) | ZIO_COMPRESS_MASK(GZIP_7);
9673 	mask |= ZIO_COMPRESS_MASK(GZIP_8) | ZIO_COMPRESS_MASK(GZIP_9);
9674 	for (int c = 0; c < ZIO_COMPRESS_FUNCTIONS; c++)
9675 		if (((1ULL << c) & mask) == 0)
9676 			*cfuncp++ = c;
9677 
9678 	/*
9679 	 * On the one hand, with SPA_MAXBLOCKSIZE at 16MB, this
9680 	 * could take a while and we should let the user know
9681 	 * we are not stuck.  On the other hand, printing progress
9682 	 * info gets old after a while.  User can specify 'v' flag
9683 	 * to see the progression.
9684 	 */
9685 	if (lsize == psize)
9686 		lsize += SPA_MINBLOCKSIZE;
9687 	else
9688 		maxlsize = lsize;
9689 
9690 	for (; lsize <= maxlsize; lsize += SPA_MINBLOCKSIZE) {
9691 		for (cfuncp = cfuncs; *cfuncp; cfuncp++) {
9692 			if (try_decompress_block(pabd, lsize, psize, flags,
9693 			    *cfuncp, lbuf, lbuf2)) {
9694 				tryzle = B_FALSE;
9695 				break;
9696 			}
9697 		}
9698 		if (*cfuncp != 0)
9699 			break;
9700 	}
9701 	if (tryzle) {
9702 		for (lsize = orig_lsize; lsize <= maxlsize;
9703 		    lsize += SPA_MINBLOCKSIZE) {
9704 			if (try_decompress_block(pabd, lsize, psize, flags,
9705 			    ZIO_COMPRESS_ZLE, lbuf, lbuf2)) {
9706 				*cfuncp = ZIO_COMPRESS_ZLE;
9707 				break;
9708 			}
9709 		}
9710 	}
9711 	umem_free(lbuf2, SPA_MAXBLOCKSIZE);
9712 
9713 	if (*cfuncp == ZIO_COMPRESS_ZLE) {
9714 		printf("\nZLE decompression was selected. If you "
9715 		    "suspect the results are wrong,\ntry avoiding ZLE "
9716 		    "by setting and exporting ZDB_NO_ZLE=\"true\"\n");
9717 	}
9718 
9719 	return (lsize > maxlsize ? -1 : lsize);
9720 }
9721 
9722 /*
9723  * Read a block from a pool and print it out.  The syntax of the
9724  * block descriptor is:
9725  *
9726  *	pool:vdev_specifier:offset:[lsize/]psize[:flags]
9727  *
9728  *	pool           - The name of the pool you wish to read from
9729  *	vdev_specifier - Which vdev (see comment for zdb_vdev_lookup)
9730  *	offset         - offset, in hex, in bytes
9731  *	size           - Amount of data to read, in hex, in bytes
9732  *	flags          - A string of characters specifying options
9733  *		 b: Decode a blkptr at given offset within block
9734  *		 c: Calculate and display checksums
9735  *		 d: Decompress data before dumping
9736  *		 e: Byteswap data before dumping
9737  *		 g: Display data as a gang block header
9738  *		 i: Display as an indirect block
9739  *		 r: Dump raw data to stdout
9740  *		 v: Verbose
9741  *
9742  */
9743 static void
zdb_read_block(char * thing,spa_t * spa)9744 zdb_read_block(char *thing, spa_t *spa)
9745 {
9746 	blkptr_t blk, *bp = &blk;
9747 	dva_t *dva = bp->blk_dva;
9748 	int flags = 0;
9749 	uint64_t offset = 0, psize = 0, lsize = 0, blkptr_offset = 0;
9750 	zio_t *zio;
9751 	vdev_t *vd;
9752 	abd_t *pabd;
9753 	void *lbuf, *buf;
9754 	char *s, *p, *dup, *flagstr, *sizes, *tmp = NULL;
9755 	const char *vdev, *errmsg = NULL;
9756 	int i, len, error;
9757 	boolean_t borrowed = B_FALSE, found = B_FALSE;
9758 
9759 	dup = strdup(thing);
9760 	s = strtok_r(dup, ":", &tmp);
9761 	vdev = s ?: "";
9762 	s = strtok_r(NULL, ":", &tmp);
9763 	offset = strtoull(s ? s : "", NULL, 16);
9764 	sizes = strtok_r(NULL, ":", &tmp);
9765 	s = strtok_r(NULL, ":", &tmp);
9766 	flagstr = strdup(s ?: "");
9767 
9768 	if (!zdb_parse_block_sizes(sizes, &lsize, &psize))
9769 		errmsg = "invalid size(s)";
9770 	if (!IS_P2ALIGNED(psize, DEV_BSIZE) || !IS_P2ALIGNED(lsize, DEV_BSIZE))
9771 		errmsg = "size must be a multiple of sector size";
9772 	if (!IS_P2ALIGNED(offset, DEV_BSIZE))
9773 		errmsg = "offset must be a multiple of sector size";
9774 	if (errmsg) {
9775 		(void) printf("Invalid block specifier: %s  - %s\n",
9776 		    thing, errmsg);
9777 		goto done;
9778 	}
9779 
9780 	tmp = NULL;
9781 	for (s = strtok_r(flagstr, ":", &tmp);
9782 	    s != NULL;
9783 	    s = strtok_r(NULL, ":", &tmp)) {
9784 		len = strlen(flagstr);
9785 		for (i = 0; i < len; i++) {
9786 			int bit = flagbits[(uchar_t)flagstr[i]];
9787 
9788 			if (bit == 0) {
9789 				(void) printf("***Ignoring flag: %c\n",
9790 				    (uchar_t)flagstr[i]);
9791 				continue;
9792 			}
9793 			found = B_TRUE;
9794 			flags |= bit;
9795 
9796 			p = &flagstr[i + 1];
9797 			if (*p != ':' && *p != '\0') {
9798 				int j = 0, nextbit = flagbits[(uchar_t)*p];
9799 				char *end, offstr[8] = { 0 };
9800 				if ((bit == ZDB_FLAG_PRINT_BLKPTR) &&
9801 				    (nextbit == 0)) {
9802 					/* look ahead to isolate the offset */
9803 					while (nextbit == 0 &&
9804 					    strchr(flagbitstr, *p) == NULL) {
9805 						offstr[j] = *p;
9806 						j++;
9807 						if (i + j > strlen(flagstr))
9808 							break;
9809 						p++;
9810 						nextbit = flagbits[(uchar_t)*p];
9811 					}
9812 					blkptr_offset = strtoull(offstr, &end,
9813 					    16);
9814 					i += j;
9815 				} else if (nextbit == 0) {
9816 					(void) printf("***Ignoring flag arg:"
9817 					    " '%c'\n", (uchar_t)*p);
9818 				}
9819 			}
9820 		}
9821 	}
9822 	if (blkptr_offset % sizeof (blkptr_t)) {
9823 		printf("Block pointer offset 0x%llx "
9824 		    "must be divisible by 0x%x\n",
9825 		    (longlong_t)blkptr_offset, (int)sizeof (blkptr_t));
9826 		goto done;
9827 	}
9828 	if (found == B_FALSE && strlen(flagstr) > 0) {
9829 		printf("Invalid flag arg: '%s'\n", flagstr);
9830 		goto done;
9831 	}
9832 
9833 	vd = zdb_vdev_lookup(spa->spa_root_vdev, vdev);
9834 	if (vd == NULL) {
9835 		(void) printf("***Invalid vdev: %s\n", vdev);
9836 		goto done;
9837 	} else {
9838 		if (vd->vdev_path)
9839 			(void) fprintf(stderr, "Found vdev: %s\n",
9840 			    vd->vdev_path);
9841 		else
9842 			(void) fprintf(stderr, "Found vdev type: %s\n",
9843 			    vd->vdev_ops->vdev_op_type);
9844 	}
9845 
9846 	pabd = abd_alloc_for_io(SPA_MAXBLOCKSIZE, B_FALSE);
9847 	lbuf = umem_alloc(SPA_MAXBLOCKSIZE, UMEM_NOFAIL);
9848 
9849 	BP_ZERO(bp);
9850 
9851 	DVA_SET_VDEV(&dva[0], vd->vdev_id);
9852 	DVA_SET_OFFSET(&dva[0], offset);
9853 	DVA_SET_GANG(&dva[0], 0);
9854 	DVA_SET_ASIZE(&dva[0], vdev_psize_to_asize(vd, psize));
9855 
9856 	BP_SET_BIRTH(bp, TXG_INITIAL, TXG_INITIAL);
9857 
9858 	BP_SET_LSIZE(bp, lsize);
9859 	BP_SET_PSIZE(bp, psize);
9860 	BP_SET_COMPRESS(bp, ZIO_COMPRESS_OFF);
9861 	BP_SET_CHECKSUM(bp, ZIO_CHECKSUM_OFF);
9862 	BP_SET_TYPE(bp, DMU_OT_NONE);
9863 	BP_SET_LEVEL(bp, 0);
9864 	BP_SET_DEDUP(bp, 0);
9865 	BP_SET_BYTEORDER(bp, ZFS_HOST_BYTEORDER);
9866 
9867 	spa_config_enter(spa, SCL_STATE, FTAG, RW_READER);
9868 	zio = zio_root(spa, NULL, NULL, ZIO_FLAG_CANFAIL);
9869 
9870 	if (vd == vd->vdev_top) {
9871 		/*
9872 		 * Treat this as a normal block read.
9873 		 */
9874 		zio_nowait(zio_read(zio, spa, bp, pabd, psize, NULL, NULL,
9875 		    ZIO_PRIORITY_SYNC_READ,
9876 		    ZIO_FLAG_CANFAIL | ZIO_FLAG_RAW, NULL));
9877 	} else {
9878 		/*
9879 		 * Treat this as a vdev child I/O.
9880 		 */
9881 		zio_nowait(zio_vdev_child_io(zio, bp, vd, offset, pabd,
9882 		    psize, ZIO_TYPE_READ, ZIO_PRIORITY_SYNC_READ,
9883 		    ZIO_FLAG_DONT_PROPAGATE | ZIO_FLAG_DONT_RETRY |
9884 		    ZIO_FLAG_CANFAIL | ZIO_FLAG_RAW | ZIO_FLAG_OPTIONAL,
9885 		    NULL, NULL));
9886 	}
9887 
9888 	error = zio_wait(zio);
9889 	spa_config_exit(spa, SCL_STATE, FTAG);
9890 
9891 	if (error) {
9892 		(void) printf("Read of %s failed, error: %d\n", thing, error);
9893 		goto out;
9894 	}
9895 
9896 	uint64_t orig_lsize = lsize;
9897 	buf = lbuf;
9898 	if (flags & ZDB_FLAG_DECOMPRESS) {
9899 		lsize = zdb_decompress_block(pabd, buf, lbuf,
9900 		    lsize, psize, flags);
9901 		if (lsize == -1) {
9902 			(void) printf("Decompress of %s failed\n", thing);
9903 			goto out;
9904 		}
9905 	} else {
9906 		buf = abd_borrow_buf_copy(pabd, lsize);
9907 		borrowed = B_TRUE;
9908 	}
9909 	/*
9910 	 * Try to detect invalid block pointer.  If invalid, try
9911 	 * decompressing.
9912 	 */
9913 	if ((flags & ZDB_FLAG_PRINT_BLKPTR || flags & ZDB_FLAG_INDIRECT) &&
9914 	    !(flags & ZDB_FLAG_DECOMPRESS)) {
9915 		const blkptr_t *b = (const blkptr_t *)(void *)
9916 		    ((uintptr_t)buf + (uintptr_t)blkptr_offset);
9917 		if (zfs_blkptr_verify(spa, b,
9918 		    BLK_CONFIG_NEEDED, BLK_VERIFY_ONLY)) {
9919 			abd_return_buf_copy(pabd, buf, lsize);
9920 			borrowed = B_FALSE;
9921 			buf = lbuf;
9922 			lsize = zdb_decompress_block(pabd, buf,
9923 			    lbuf, lsize, psize, flags);
9924 			b = (const blkptr_t *)(void *)
9925 			    ((uintptr_t)buf + (uintptr_t)blkptr_offset);
9926 			if (lsize == -1 || zfs_blkptr_verify(spa, b,
9927 			    BLK_CONFIG_NEEDED, BLK_VERIFY_LOG)) {
9928 				printf("invalid block pointer at this DVA\n");
9929 				goto out;
9930 			}
9931 		}
9932 	}
9933 
9934 	if (flags & ZDB_FLAG_PRINT_BLKPTR)
9935 		zdb_print_blkptr((blkptr_t *)(void *)
9936 		    ((uintptr_t)buf + (uintptr_t)blkptr_offset), flags);
9937 	else if (flags & ZDB_FLAG_RAW)
9938 		zdb_dump_block_raw(buf, lsize, flags);
9939 	else if (flags & ZDB_FLAG_INDIRECT)
9940 		zdb_dump_indirect((blkptr_t *)buf,
9941 		    orig_lsize / sizeof (blkptr_t), flags);
9942 	else if (flags & ZDB_FLAG_GBH)
9943 		zdb_dump_gbh(buf, lsize, flags);
9944 	else
9945 		zdb_dump_block(thing, buf, lsize, flags);
9946 
9947 	/*
9948 	 * If :c was specified, iterate through the checksum table to
9949 	 * calculate and display each checksum for our specified
9950 	 * DVA and length.
9951 	 */
9952 	if ((flags & ZDB_FLAG_CHECKSUM) && !(flags & ZDB_FLAG_RAW) &&
9953 	    !(flags & ZDB_FLAG_GBH)) {
9954 		zio_t *czio;
9955 		(void) printf("\n");
9956 		for (enum zio_checksum ck = ZIO_CHECKSUM_LABEL;
9957 		    ck < ZIO_CHECKSUM_FUNCTIONS; ck++) {
9958 
9959 			if ((zio_checksum_table[ck].ci_flags &
9960 			    ZCHECKSUM_FLAG_EMBEDDED) ||
9961 			    ck == ZIO_CHECKSUM_NOPARITY) {
9962 				continue;
9963 			}
9964 			BP_SET_CHECKSUM(bp, ck);
9965 			spa_config_enter(spa, SCL_STATE, FTAG, RW_READER);
9966 			czio = zio_root(spa, NULL, NULL, ZIO_FLAG_CANFAIL);
9967 			if (vd == vd->vdev_top) {
9968 				zio_nowait(zio_read(czio, spa, bp, pabd, psize,
9969 				    NULL, NULL,
9970 				    ZIO_PRIORITY_SYNC_READ,
9971 				    ZIO_FLAG_CANFAIL | ZIO_FLAG_RAW |
9972 				    ZIO_FLAG_DONT_RETRY, NULL));
9973 			} else {
9974 				zio_nowait(zio_vdev_child_io(czio, bp, vd,
9975 				    offset, pabd, psize, ZIO_TYPE_READ,
9976 				    ZIO_PRIORITY_SYNC_READ,
9977 				    ZIO_FLAG_DONT_PROPAGATE |
9978 				    ZIO_FLAG_DONT_RETRY |
9979 				    ZIO_FLAG_CANFAIL | ZIO_FLAG_RAW |
9980 				    ZIO_FLAG_SPECULATIVE |
9981 				    ZIO_FLAG_OPTIONAL, NULL, NULL));
9982 			}
9983 			error = zio_wait(czio);
9984 			if (error == 0 || error == ECKSUM) {
9985 				zio_t *ck_zio = zio_null(NULL, spa, NULL,
9986 				    NULL, NULL, 0);
9987 				ck_zio->io_offset =
9988 				    DVA_GET_OFFSET(&bp->blk_dva[0]);
9989 				ck_zio->io_bp = bp;
9990 				zio_checksum_compute(ck_zio, ck, pabd, psize);
9991 				printf(
9992 				    "%12s\t"
9993 				    "cksum=%016llx:%016llx:%016llx:%016llx\n",
9994 				    zio_checksum_table[ck].ci_name,
9995 				    (u_longlong_t)bp->blk_cksum.zc_word[0],
9996 				    (u_longlong_t)bp->blk_cksum.zc_word[1],
9997 				    (u_longlong_t)bp->blk_cksum.zc_word[2],
9998 				    (u_longlong_t)bp->blk_cksum.zc_word[3]);
9999 				zio_wait(ck_zio);
10000 			} else {
10001 				printf("error %d reading block\n", error);
10002 			}
10003 			spa_config_exit(spa, SCL_STATE, FTAG);
10004 		}
10005 	}
10006 
10007 	if (borrowed)
10008 		abd_return_buf_copy(pabd, buf, lsize);
10009 
10010 out:
10011 	abd_free(pabd);
10012 	umem_free(lbuf, SPA_MAXBLOCKSIZE);
10013 done:
10014 	free(flagstr);
10015 	free(dup);
10016 }
10017 
10018 static void
zdb_embedded_block(char * thing)10019 zdb_embedded_block(char *thing)
10020 {
10021 	blkptr_t bp = {{{{0}}}};
10022 	unsigned long long *words = (void *)&bp;
10023 	char *buf;
10024 	int err;
10025 
10026 	err = sscanf(thing, "%llx:%llx:%llx:%llx:%llx:%llx:%llx:%llx:"
10027 	    "%llx:%llx:%llx:%llx:%llx:%llx:%llx:%llx",
10028 	    words + 0, words + 1, words + 2, words + 3,
10029 	    words + 4, words + 5, words + 6, words + 7,
10030 	    words + 8, words + 9, words + 10, words + 11,
10031 	    words + 12, words + 13, words + 14, words + 15);
10032 	if (err != 16) {
10033 		(void) fprintf(stderr, "invalid input format\n");
10034 		zdb_exit(1);
10035 	}
10036 	ASSERT3U(BPE_GET_LSIZE(&bp), <=, SPA_MAXBLOCKSIZE);
10037 	buf = malloc(SPA_MAXBLOCKSIZE);
10038 	if (buf == NULL) {
10039 		(void) fprintf(stderr, "out of memory\n");
10040 		zdb_exit(1);
10041 	}
10042 	err = decode_embedded_bp(&bp, buf, BPE_GET_LSIZE(&bp));
10043 	if (err != 0) {
10044 		(void) fprintf(stderr, "decode failed: %u\n", err);
10045 		zdb_exit(1);
10046 	}
10047 	zdb_dump_block_raw(buf, BPE_GET_LSIZE(&bp), 0);
10048 	free(buf);
10049 }
10050 
10051 /* check for valid hex or decimal numeric string */
10052 static boolean_t
zdb_numeric(char * str)10053 zdb_numeric(char *str)
10054 {
10055 	int i = 0, len;
10056 
10057 	len = strlen(str);
10058 	if (len == 0)
10059 		return (B_FALSE);
10060 	if (strncmp(str, "0x", 2) == 0 || strncmp(str, "0X", 2) == 0)
10061 		i = 2;
10062 	for (; i < len; i++) {
10063 		if (!isxdigit(str[i]))
10064 			return (B_FALSE);
10065 	}
10066 	return (B_TRUE);
10067 }
10068 
10069 static int
dummy_get_file_info(dmu_object_type_t bonustype,const void * data,zfs_file_info_t * zoi)10070 dummy_get_file_info(dmu_object_type_t bonustype, const void *data,
10071     zfs_file_info_t *zoi)
10072 {
10073 	(void) data, (void) zoi;
10074 
10075 	if (bonustype != DMU_OT_ZNODE && bonustype != DMU_OT_SA)
10076 		return (ENOENT);
10077 
10078 	(void) fprintf(stderr, "dummy_get_file_info: not implemented");
10079 	abort();
10080 }
10081 
10082 int
main(int argc,char ** argv)10083 main(int argc, char **argv)
10084 {
10085 	int c;
10086 	int dump_all = 1;
10087 	int verbose = 0;
10088 	int error = 0;
10089 	char **searchdirs = NULL;
10090 	int nsearch = 0;
10091 	char *target, *target_pool, dsname[ZFS_MAX_DATASET_NAME_LEN];
10092 	nvlist_t *policy = NULL;
10093 	uint64_t max_txg = UINT64_MAX;
10094 	int64_t objset_id = -1;
10095 	uint64_t object;
10096 	int flags = ZFS_IMPORT_MISSING_LOG;
10097 	int rewind = ZPOOL_NEVER_REWIND;
10098 	char *spa_config_path_env, *objset_str;
10099 	boolean_t target_is_spa = B_TRUE, dataset_lookup = B_FALSE;
10100 	nvlist_t *cfg = NULL;
10101 	struct sigaction action;
10102 	boolean_t force_import = B_FALSE;
10103 	boolean_t config_path_console = B_FALSE;
10104 	char pbuf[MAXPATHLEN];
10105 
10106 	dprintf_setup(&argc, argv);
10107 
10108 	/*
10109 	 * Set up signal handlers, so if we crash due to bad on-disk data we
10110 	 * can get more info. Unlike ztest, we don't bail out if we can't set
10111 	 * up signal handlers, because zdb is very useful without them.
10112 	 */
10113 	action.sa_handler = sig_handler;
10114 	sigemptyset(&action.sa_mask);
10115 	action.sa_flags = 0;
10116 	if (sigaction(SIGSEGV, &action, NULL) < 0) {
10117 		(void) fprintf(stderr, "zdb: cannot catch SIGSEGV: %s\n",
10118 		    strerror(errno));
10119 	}
10120 	if (sigaction(SIGABRT, &action, NULL) < 0) {
10121 		(void) fprintf(stderr, "zdb: cannot catch SIGABRT: %s\n",
10122 		    strerror(errno));
10123 	}
10124 
10125 	/*
10126 	 * If there is an environment variable SPA_CONFIG_PATH it overrides
10127 	 * default spa_config_path setting. If -U flag is specified it will
10128 	 * override this environment variable settings once again.
10129 	 */
10130 	spa_config_path_env = getenv("SPA_CONFIG_PATH");
10131 	if (spa_config_path_env != NULL)
10132 		spa_config_path = spa_config_path_env;
10133 
10134 	/*
10135 	 * For performance reasons, we set this tunable down. We do so before
10136 	 * the arg parsing section so that the user can override this value if
10137 	 * they choose.
10138 	 */
10139 	zfs_btree_verify_intensity = 3;
10140 
10141 	struct option long_options[] = {
10142 		{"ignore-assertions",	no_argument,		NULL, 'A'},
10143 		{"block-stats",		no_argument,		NULL, 'b'},
10144 		{"backup",		no_argument,		NULL, 'B'},
10145 		{"checksum",		no_argument,		NULL, 'c'},
10146 		{"config",		no_argument,		NULL, 'C'},
10147 		{"datasets",		no_argument,		NULL, 'd'},
10148 		{"dedup-stats",		no_argument,		NULL, 'D'},
10149 		{"exported",		no_argument,		NULL, 'e'},
10150 		{"embedded-block-pointer",	no_argument,	NULL, 'E'},
10151 		{"file-layout",		no_argument,		NULL, 'f'},
10152 		{"automatic-rewind",	no_argument,		NULL, 'F'},
10153 		{"dump-debug-msg",	no_argument,		NULL, 'G'},
10154 		{"history",		no_argument,		NULL, 'h'},
10155 		{"scripting-mode",	no_argument,		NULL, 'H'},
10156 		{"intent-logs",		no_argument,		NULL, 'i'},
10157 		{"inflight",		required_argument,	NULL, 'I'},
10158 		{"checkpointed-state",	no_argument,		NULL, 'k'},
10159 		{"key",			required_argument,	NULL, 'K'},
10160 		{"label",		no_argument,		NULL, 'l'},
10161 		{"disable-leak-tracking",	no_argument,	NULL, 'L'},
10162 		{"metaslabs",		no_argument,		NULL, 'm'},
10163 		{"metaslab-groups",	no_argument,		NULL, 'M'},
10164 		{"numeric",		no_argument,		NULL, 'N'},
10165 		{"option",		required_argument,	NULL, 'o'},
10166 		{"object-lookups",	no_argument,		NULL, 'O'},
10167 		{"path",		required_argument,	NULL, 'p'},
10168 		{"parseable",		no_argument,		NULL, 'P'},
10169 		{"skip-label",		no_argument,		NULL, 'q'},
10170 		{"copy-object",		no_argument,		NULL, 'r'},
10171 		{"read-block",		no_argument,		NULL, 'R'},
10172 		{"io-stats",		no_argument,		NULL, 's'},
10173 		{"simulate-dedup",	no_argument,		NULL, 'S'},
10174 		{"txg",			required_argument,	NULL, 't'},
10175 		{"brt-stats",		no_argument,		NULL, 'T'},
10176 		{"uberblock",		no_argument,		NULL, 'u'},
10177 		{"cachefile",		required_argument,	NULL, 'U'},
10178 		{"verbose",		no_argument,		NULL, 'v'},
10179 		{"verbatim",		no_argument,		NULL, 'V'},
10180 		{"dump-blocks",		required_argument,	NULL, 'x'},
10181 		{"extreme-rewind",	no_argument,		NULL, 'X'},
10182 		{"all-reconstruction",	no_argument,		NULL, 'Y'},
10183 		{"livelist",		no_argument,		NULL, 'y'},
10184 		{"zstd-headers",	no_argument,		NULL, 'Z'},
10185 		{"allocated-map",	no_argument,		NULL,
10186 		    ARG_ALLOCATED},
10187 		{"bin",			required_argument,	NULL,
10188 		    ARG_BLOCK_BIN_MODE},
10189 		{"class",		required_argument,	NULL,
10190 		    ARG_BLOCK_CLASSES},
10191 		{0, 0, 0, 0}
10192 	};
10193 
10194 	while ((c = getopt_long(argc, argv,
10195 	    "AbBcCdDeEfFGhHiI:kK:lLmMNo:Op:PqrRsSt:TuU:vVx:XYyZ",
10196 	    long_options, NULL)) != -1) {
10197 		switch (c) {
10198 		case 'b':
10199 		case 'B':
10200 		case 'c':
10201 		case 'C':
10202 		case 'd':
10203 		case 'D':
10204 		case 'E':
10205 		case 'f':
10206 		case 'G':
10207 		case 'h':
10208 		case 'i':
10209 		case 'l':
10210 		case 'm':
10211 		case 'M':
10212 		case 'N':
10213 		case 'O':
10214 		case 'r':
10215 		case 'R':
10216 		case 's':
10217 		case 'S':
10218 		case 'T':
10219 		case 'u':
10220 		case 'y':
10221 		case 'Z':
10222 		case ARG_ALLOCATED:
10223 			dump_opt[c]++;
10224 			dump_all = 0;
10225 			break;
10226 		case 'A':
10227 		case 'e':
10228 		case 'F':
10229 		case 'H':
10230 		case 'k':
10231 		case 'L':
10232 		case 'P':
10233 		case 'q':
10234 		case 'X':
10235 			dump_opt[c]++;
10236 			break;
10237 		case 'Y':
10238 			zfs_reconstruct_indirect_combinations_max = INT_MAX;
10239 			zfs_deadman_enabled = 0;
10240 			break;
10241 		/* NB: Sort single match options below. */
10242 		case 'I':
10243 			max_inflight_bytes = strtoull(optarg, NULL, 0);
10244 			if (max_inflight_bytes == 0) {
10245 				(void) fprintf(stderr, "maximum number "
10246 				    "of inflight bytes must be greater "
10247 				    "than 0\n");
10248 				usage();
10249 			}
10250 			break;
10251 		case 'K':
10252 			dump_opt[c]++;
10253 			key_material = strdup(optarg);
10254 			/* redact key material in process table */
10255 			while (*optarg != '\0') { *optarg++ = '*'; }
10256 			break;
10257 		case 'o':
10258 			dump_opt[c]++;
10259 			dump_all = 0;
10260 			error = handle_tunable_option(optarg, B_FALSE);
10261 			if (error != 0)
10262 				zdb_exit(1);
10263 			break;
10264 		case 'p':
10265 			if (searchdirs == NULL) {
10266 				searchdirs = umem_alloc(sizeof (char *),
10267 				    UMEM_NOFAIL);
10268 			} else {
10269 				char **tmp = umem_alloc((nsearch + 1) *
10270 				    sizeof (char *), UMEM_NOFAIL);
10271 				memcpy(tmp, searchdirs, nsearch *
10272 				    sizeof (char *));
10273 				umem_free(searchdirs,
10274 				    nsearch * sizeof (char *));
10275 				searchdirs = tmp;
10276 			}
10277 			searchdirs[nsearch++] = optarg;
10278 			break;
10279 		case 't':
10280 			max_txg = strtoull(optarg, NULL, 0);
10281 			if (max_txg < TXG_INITIAL) {
10282 				(void) fprintf(stderr, "incorrect txg "
10283 				    "specified: %s\n", optarg);
10284 				usage();
10285 			}
10286 			break;
10287 		case 'U':
10288 			config_path_console = B_TRUE;
10289 			spa_config_path = optarg;
10290 			if (spa_config_path[0] != '/') {
10291 				(void) fprintf(stderr,
10292 				    "cachefile must be an absolute path "
10293 				    "(i.e. start with a slash)\n");
10294 				usage();
10295 			}
10296 			break;
10297 		case 'v':
10298 			verbose++;
10299 			break;
10300 		case 'V':
10301 			flags = ZFS_IMPORT_VERBATIM;
10302 			break;
10303 		case 'x':
10304 			vn_dumpdir = optarg;
10305 			break;
10306 		case ARG_BLOCK_BIN_MODE:
10307 			if (strcmp(optarg, "lsize") == 0) {
10308 				block_bin_mode = BIN_LSIZE;
10309 			} else if (strcmp(optarg, "psize") == 0) {
10310 				block_bin_mode = BIN_PSIZE;
10311 			} else if (strcmp(optarg, "asize") == 0) {
10312 				block_bin_mode = BIN_ASIZE;
10313 			} else {
10314 				(void) fprintf(stderr,
10315 				    "--bin=\"%s\" must be one of \"lsize\", "
10316 				    "\"psize\" or \"asize\"\n", optarg);
10317 				usage();
10318 			}
10319 			break;
10320 
10321 		case ARG_BLOCK_CLASSES: {
10322 			char *buf = strdup(optarg), *tok = buf, *next,
10323 			    *save = NULL;
10324 
10325 			while ((next = strtok_r(tok, ",", &save)) != NULL) {
10326 				tok = NULL;
10327 
10328 				if (strcmp(next, "normal") == 0) {
10329 					block_classes |= CLASS_NORMAL;
10330 				} else if (strcmp(next, "special") == 0) {
10331 					block_classes |= CLASS_SPECIAL;
10332 				} else if (strcmp(next, "dedup") == 0) {
10333 					block_classes |= CLASS_DEDUP;
10334 				} else if (strcmp(next, "other") == 0) {
10335 					block_classes |= CLASS_OTHER;
10336 				} else {
10337 					(void) fprintf(stderr,
10338 					    "--class=\"%s\" must be a "
10339 					    "comma-separated list of either "
10340 					    "\"normal\", \"special\", "
10341 					    "\"asize\" or \"other\"; "
10342 					    "got \"%s\"\n",
10343 					    optarg, next);
10344 					usage();
10345 				}
10346 			}
10347 
10348 			if (block_classes == 0) {
10349 				(void) fprintf(stderr,
10350 				    "--class= must be a comma-separated "
10351 				    "list of either \"normal\", \"special\", "
10352 				    "\"asize\" or \"other\"; got empty\n");
10353 				usage();
10354 			}
10355 
10356 			free(buf);
10357 			break;
10358 		}
10359 		default:
10360 			usage();
10361 			break;
10362 		}
10363 	}
10364 
10365 	if (!dump_opt['e'] && searchdirs != NULL) {
10366 		(void) fprintf(stderr, "-p option requires use of -e\n");
10367 		usage();
10368 	}
10369 	if (dump_opt['H'] && !dump_opt['f']) {
10370 		(void) fprintf(stderr, "-H option requires use of -f\n");
10371 		usage();
10372 	}
10373 #if defined(_LP64)
10374 	/*
10375 	 * ZDB does not typically re-read blocks; therefore limit the ARC
10376 	 * to 256 MB, which can be used entirely for metadata.
10377 	 */
10378 	zfs_arc_min = 2ULL << SPA_MAXBLOCKSHIFT;
10379 	zfs_arc_max = 256 * 1024 * 1024;
10380 #endif
10381 
10382 	/*
10383 	 * "zdb -c" uses checksum-verifying scrub i/os which are async reads.
10384 	 * "zdb -b" uses traversal prefetch which uses async reads.
10385 	 * For good performance, let several of them be active at once.
10386 	 */
10387 	zfs_vdev_async_read_max_active = 10;
10388 
10389 	/*
10390 	 * Disable reference tracking for better performance.
10391 	 */
10392 	reference_tracking_enable = B_FALSE;
10393 
10394 	/*
10395 	 * Do not fail spa_load when spa_load_verify fails. This is needed
10396 	 * to load non-idle pools.
10397 	 */
10398 	spa_load_verify_dryrun = B_TRUE;
10399 
10400 	/*
10401 	 * ZDB should have ability to read spacemaps.
10402 	 */
10403 	spa_mode_readable_spacemaps = B_TRUE;
10404 
10405 	libspl_set_assert_ok((dump_opt['A'] == 1) || (dump_opt['A'] > 2));
10406 	zfs_recover = (dump_opt['A'] > 1);
10407 
10408 	if (dump_all)
10409 		verbose = MAX(verbose, 1);
10410 
10411 	for (c = 0; c < 256; c++) {
10412 		if (dump_all && strchr("ABeEfFkKlLNOPrRSXy", c) == NULL)
10413 			dump_opt[c] = 1;
10414 		if (dump_opt[c])
10415 			dump_opt[c] += verbose;
10416 	}
10417 
10418 	argc -= optind;
10419 	argv += optind;
10420 	if (argc < 2 && dump_opt['R'])
10421 		usage();
10422 
10423 	target = argv[0];
10424 
10425 	/*
10426 	 * Automate cachefile
10427 	 */
10428 	if (!spa_config_path_env && !config_path_console && target &&
10429 	    !dump_opt['l'] && libzfs_core_init() == 0) {
10430 		char *pname = strdup(target);
10431 		const char *value;
10432 		nvlist_t *pnvl = NULL;
10433 		nvlist_t *vnvl = NULL;
10434 
10435 		if (strpbrk(pname, "/@") != NULL)
10436 			*strpbrk(pname, "/@") = '\0';
10437 
10438 		if (pname && lzc_get_props(pname, &pnvl) == 0) {
10439 			if (nvlist_lookup_nvlist(pnvl, "cachefile",
10440 			    &vnvl) == 0) {
10441 				value = fnvlist_lookup_string(vnvl,
10442 				    ZPROP_VALUE);
10443 			} else {
10444 				value = "-";
10445 			}
10446 			strlcpy(pbuf, value, sizeof (pbuf));
10447 			if (pbuf[0] != '\0') {
10448 				if (pbuf[0] == '/') {
10449 					if (access(pbuf, F_OK) == 0)
10450 						spa_config_path = pbuf;
10451 					else
10452 						force_import = B_TRUE;
10453 				} else if ((strcmp(pbuf, "-") == 0 &&
10454 				    access(ZPOOL_CACHE, F_OK) != 0) ||
10455 				    strcmp(pbuf, "none") == 0) {
10456 					force_import = B_TRUE;
10457 				}
10458 			}
10459 			nvlist_free(vnvl);
10460 		}
10461 
10462 		free(pname);
10463 		nvlist_free(pnvl);
10464 		libzfs_core_fini();
10465 	}
10466 
10467 	dmu_objset_register_type(DMU_OST_ZFS, dummy_get_file_info);
10468 	kernel_init(SPA_MODE_READ);
10469 	kernel_init_done = B_TRUE;
10470 
10471 	if (dump_opt['E']) {
10472 		if (argc != 1)
10473 			usage();
10474 		zdb_embedded_block(argv[0]);
10475 		error = 0;
10476 		goto fini;
10477 	}
10478 
10479 	if (argc < 1) {
10480 		if (!dump_opt['e'] && dump_opt['C']) {
10481 			dump_cachefile(spa_config_path);
10482 			error = 0;
10483 			goto fini;
10484 		}
10485 		if (dump_opt['o'])
10486 			/*
10487 			 * Avoid blasting tunable options off the top of the
10488 			 * screen.
10489 			 */
10490 			zdb_exit(1);
10491 		usage();
10492 	}
10493 
10494 	if (dump_opt['l']) {
10495 		error = dump_label(argv[0]);
10496 		goto fini;
10497 	}
10498 
10499 	if (dump_opt['X'] || dump_opt['F'])
10500 		rewind = ZPOOL_DO_REWIND |
10501 		    (dump_opt['X'] ? ZPOOL_EXTREME_REWIND : 0);
10502 
10503 	/* -N implies -d */
10504 	if (dump_opt['N'] && dump_opt['d'] == 0)
10505 		dump_opt['d'] = dump_opt['N'];
10506 
10507 	if (nvlist_alloc(&policy, NV_UNIQUE_NAME_TYPE, 0) != 0 ||
10508 	    nvlist_add_uint64(policy, ZPOOL_LOAD_REQUEST_TXG, max_txg) != 0 ||
10509 	    nvlist_add_uint32(policy, ZPOOL_LOAD_REWIND_POLICY, rewind) != 0)
10510 		fatal("internal error: %s", strerror(ENOMEM));
10511 
10512 	error = 0;
10513 
10514 	if (strpbrk(target, "/@") != NULL) {
10515 		size_t targetlen;
10516 
10517 		target_pool = strdup(target);
10518 		*strpbrk(target_pool, "/@") = '\0';
10519 
10520 		target_is_spa = B_FALSE;
10521 		targetlen = strlen(target);
10522 		if (targetlen && target[targetlen - 1] == '/')
10523 			target[targetlen - 1] = '\0';
10524 
10525 		/*
10526 		 * See if an objset ID was supplied (-d <pool>/<objset ID>).
10527 		 * To disambiguate tank/100, consider the 100 as objsetID
10528 		 * if -N was given, otherwise 100 is an objsetID iff
10529 		 * tank/100 as a named dataset fails on lookup.
10530 		 */
10531 		objset_str = strchr(target, '/');
10532 		if (objset_str && strlen(objset_str) > 1 &&
10533 		    zdb_numeric(objset_str + 1)) {
10534 			char *endptr;
10535 			errno = 0;
10536 			objset_str++;
10537 			objset_id = strtoull(objset_str, &endptr, 0);
10538 			/* dataset 0 is the same as opening the pool */
10539 			if (errno == 0 && endptr != objset_str &&
10540 			    objset_id != 0) {
10541 				if (dump_opt['N'])
10542 					dataset_lookup = B_TRUE;
10543 			}
10544 			/* normal dataset name not an objset ID */
10545 			if (endptr == objset_str) {
10546 				objset_id = -1;
10547 			}
10548 		} else if (objset_str && !zdb_numeric(objset_str + 1) &&
10549 		    dump_opt['N']) {
10550 			printf("Supply a numeric objset ID with -N\n");
10551 			error = 2;
10552 			goto fini;
10553 		}
10554 	} else {
10555 		target_pool = target;
10556 	}
10557 
10558 	if (dump_opt['e'] || force_import) {
10559 		importargs_t args = { 0 };
10560 
10561 		/*
10562 		 * If path is not provided, search in /dev
10563 		 */
10564 		if (searchdirs == NULL) {
10565 			searchdirs = umem_alloc(sizeof (char *), UMEM_NOFAIL);
10566 			searchdirs[nsearch++] = (char *)ZFS_DEVDIR;
10567 		}
10568 
10569 		args.paths = nsearch;
10570 		args.path = searchdirs;
10571 		args.can_be_active = B_TRUE;
10572 
10573 		libpc_handle_t lpch = {
10574 			.lpc_lib_handle = NULL,
10575 			.lpc_ops = &libzpool_config_ops,
10576 			.lpc_printerr = B_TRUE
10577 		};
10578 		error = zpool_find_config(&lpch, target_pool, &cfg, &args);
10579 
10580 		if (error == 0) {
10581 
10582 			if (nvlist_add_nvlist(cfg,
10583 			    ZPOOL_LOAD_POLICY, policy) != 0) {
10584 				fatal("can't open '%s': %s",
10585 				    target, strerror(ENOMEM));
10586 			}
10587 
10588 			if (dump_opt['C'] > 1) {
10589 				(void) printf("\nConfiguration for import:\n");
10590 				dump_nvlist(cfg, 8);
10591 			}
10592 
10593 			/*
10594 			 * Disable the activity check to allow examination of
10595 			 * active pools.
10596 			 */
10597 			error = spa_import(target_pool, cfg, NULL,
10598 			    flags | ZFS_IMPORT_SKIP_MMP);
10599 		}
10600 	}
10601 
10602 	if (searchdirs != NULL) {
10603 		umem_free(searchdirs, nsearch * sizeof (char *));
10604 		searchdirs = NULL;
10605 	}
10606 
10607 	/*
10608 	 * We need to make sure to process -O option or call
10609 	 * dump_path after the -e option has been processed,
10610 	 * which imports the pool to the namespace if it's
10611 	 * not in the cachefile.
10612 	 */
10613 	if (dump_opt['O'] && !dump_opt['r']) {
10614 		if (argc != 2)
10615 			usage();
10616 		dump_opt['v'] = verbose + 3;
10617 		error = dump_path(argv[0], argv[1], NULL);
10618 		goto fini;
10619 	}
10620 
10621 	if (dump_opt['r']) {
10622 		target_is_spa = B_FALSE;
10623 		if (argc != 3)
10624 			usage();
10625 		dump_opt['v'] = verbose;
10626 		if (dump_opt['O']) {
10627 			object = strtoull(argv[1], NULL, 0);
10628 		} else {
10629 			error = dump_path(argv[0], argv[1], &object);
10630 		}
10631 		if (error != 0)
10632 			fatal("internal error: %s", strerror(error));
10633 	}
10634 
10635 	/*
10636 	 * import_checkpointed_state makes the assumption that the
10637 	 * target pool that we pass it is already part of the spa
10638 	 * namespace. Because of that we need to make sure to call
10639 	 * it always after the -e option has been processed, which
10640 	 * imports the pool to the namespace if it's not in the
10641 	 * cachefile.
10642 	 */
10643 	char *checkpoint_pool = NULL;
10644 	char *checkpoint_target = NULL;
10645 	if (dump_opt['k']) {
10646 		checkpoint_pool = import_checkpointed_state(target, cfg,
10647 		    target_is_spa, &checkpoint_target);
10648 
10649 		if (checkpoint_target != NULL)
10650 			target = checkpoint_target;
10651 	}
10652 
10653 	if (cfg != NULL) {
10654 		nvlist_free(cfg);
10655 		cfg = NULL;
10656 	}
10657 
10658 	if (target_pool != target)
10659 		free(target_pool);
10660 
10661 	if (error == 0) {
10662 		if (dump_opt['k'] && (target_is_spa || dump_opt['R'])) {
10663 			ASSERT(checkpoint_pool != NULL);
10664 			ASSERT0P(checkpoint_target);
10665 
10666 			error = spa_open(checkpoint_pool, &spa, FTAG);
10667 			if (error != 0) {
10668 				fatal("Tried to open pool \"%s\" but "
10669 				    "spa_open() failed with error %d\n",
10670 				    checkpoint_pool, error);
10671 			}
10672 
10673 		} else if (target_is_spa || dump_opt['R'] || dump_opt['B'] ||
10674 		    objset_id == 0) {
10675 			zdb_set_skip_mmp(target);
10676 			error = spa_open_rewind(target, &spa, FTAG, policy,
10677 			    NULL);
10678 			if (error) {
10679 				/*
10680 				 * If we're missing the log device then
10681 				 * try opening the pool after clearing the
10682 				 * log state.  Keep the global spa NULL
10683 				 * meanwhile: the failed open left it that
10684 				 * way, we hold no reference on what the
10685 				 * lookup returns, and zdb_exit() would
10686 				 * spa_close() it on the way out.
10687 				 */
10688 				spa_t *found;
10689 
10690 				spa_namespace_enter(FTAG);
10691 				if ((found = spa_lookup(target)) != NULL &&
10692 				    found->spa_log_state == SPA_LOG_MISSING) {
10693 					found->spa_log_state = SPA_LOG_CLEAR;
10694 					error = 0;
10695 				}
10696 				spa_namespace_exit(FTAG);
10697 
10698 				if (!error) {
10699 					error = spa_open_rewind(target, &spa,
10700 					    FTAG, policy, NULL);
10701 				}
10702 			}
10703 		} else if (strpbrk(target, "#") != NULL) {
10704 			dsl_pool_t *dp;
10705 			error = dsl_pool_hold(target, FTAG, &dp);
10706 			if (error != 0) {
10707 				fatal("can't dump '%s': %s", target,
10708 				    strerror(error));
10709 			}
10710 			error = dump_bookmark(dp, target, B_TRUE, verbose > 1);
10711 			dsl_pool_rele(dp, FTAG);
10712 			if (error != 0) {
10713 				fatal("can't dump '%s': %s", target,
10714 				    strerror(error));
10715 			}
10716 			goto fini;
10717 		} else {
10718 			target_pool = strdup(target);
10719 			if (strpbrk(target, "/@") != NULL)
10720 				*strpbrk(target_pool, "/@") = '\0';
10721 
10722 			zdb_set_skip_mmp(target);
10723 			/*
10724 			 * If -N was supplied, the user has indicated that
10725 			 * zdb -d <pool>/<objsetID> is in effect.  Otherwise
10726 			 * we first assume that the dataset string is the
10727 			 * dataset name.  If dmu_objset_hold fails with the
10728 			 * dataset string, and we have an objset_id, retry the
10729 			 * lookup with the objsetID.
10730 			 */
10731 			boolean_t retry = B_TRUE;
10732 retry_lookup:
10733 			if (dataset_lookup == B_TRUE) {
10734 				/*
10735 				 * Use the supplied id to get the name
10736 				 * for open_objset.
10737 				 */
10738 				error = spa_open(target_pool, &spa, FTAG);
10739 				if (error == 0) {
10740 					error = name_from_objset_id(spa,
10741 					    objset_id, dsname);
10742 					spa_close(spa, FTAG);
10743 					if (error == 0)
10744 						target = dsname;
10745 				}
10746 			}
10747 			if (error == 0) {
10748 				if (objset_id > 0 && retry) {
10749 					int err = dmu_objset_hold(target, FTAG,
10750 					    &os);
10751 					if (err) {
10752 						dataset_lookup = B_TRUE;
10753 						retry = B_FALSE;
10754 						goto retry_lookup;
10755 					} else {
10756 						dmu_objset_rele(os, FTAG);
10757 					}
10758 				}
10759 				error = open_objset(target, FTAG, &os);
10760 			}
10761 			if (error == 0)
10762 				spa = dmu_objset_spa(os);
10763 			free(target_pool);
10764 		}
10765 	}
10766 	nvlist_free(policy);
10767 
10768 	if (error)
10769 		fatal("can't open '%s': %s", target, strerror(error));
10770 
10771 	/*
10772 	 * Set the pool failure mode to panic in order to prevent the pool
10773 	 * from suspending.  A suspended I/O will have no way to resume and
10774 	 * can prevent the zdb(8) command from terminating as expected.
10775 	 */
10776 	if (spa != NULL)
10777 		spa->spa_failmode = ZIO_FAILURE_MODE_PANIC;
10778 
10779 	argv++;
10780 	argc--;
10781 	if (dump_opt['r']) {
10782 		error = zdb_copy_object(os, object, argv[1]);
10783 	} else if (!dump_opt['R']) {
10784 		flagbits['d'] = ZOR_FLAG_DIRECTORY;
10785 		flagbits['f'] = ZOR_FLAG_PLAIN_FILE;
10786 		flagbits['m'] = ZOR_FLAG_SPACE_MAP;
10787 		flagbits['v'] = ZOR_FLAG_ZVOL;
10788 		flagbits['z'] = ZOR_FLAG_ZAP;
10789 		flagbits['A'] = ZOR_FLAG_ALL_TYPES;
10790 
10791 		if (argc > 0 && (dump_opt['d'] || dump_opt['f'])) {
10792 			zopt_object_args = argc;
10793 			zopt_object_ranges = calloc(zopt_object_args,
10794 			    sizeof (zopt_object_range_t));
10795 			for (unsigned i = 0; i < zopt_object_args; i++) {
10796 				int err;
10797 				const char *msg = NULL;
10798 
10799 				err = parse_object_range(argv[i],
10800 				    &zopt_object_ranges[i], &msg);
10801 				if (err != 0)
10802 					fatal("Bad object or range: '%s': %s\n",
10803 					    argv[i], msg ?: "");
10804 			}
10805 		} else if (argc > 0 && dump_opt['m']) {
10806 			zopt_metaslab_args = argc;
10807 			zopt_metaslab = calloc(zopt_metaslab_args,
10808 			    sizeof (uint64_t));
10809 			for (unsigned i = 0; i < zopt_metaslab_args; i++) {
10810 				errno = 0;
10811 				zopt_metaslab[i] = strtoull(argv[i], NULL, 0);
10812 				if (zopt_metaslab[i] == 0 && errno != 0)
10813 					fatal("bad number %s: %s", argv[i],
10814 					    strerror(errno));
10815 			}
10816 		}
10817 
10818 		if (dump_opt['f'] && os != NULL) {
10819 			dump_opt['v'] = verbose;
10820 			dump_file_data_layout(os);
10821 		} else if (dump_opt['B']) {
10822 			dump_backup(target, objset_id,
10823 			    argc > 0 ? argv[0] : NULL);
10824 		} else if (os != NULL) {
10825 			dump_objset(os);
10826 		} else if (zopt_object_args > 0 && !dump_opt['m']) {
10827 			dump_objset(spa->spa_meta_objset);
10828 		} else {
10829 			dump_zpool(spa);
10830 		}
10831 	} else {
10832 		flagbits['b'] = ZDB_FLAG_PRINT_BLKPTR;
10833 		flagbits['c'] = ZDB_FLAG_CHECKSUM;
10834 		flagbits['d'] = ZDB_FLAG_DECOMPRESS;
10835 		flagbits['e'] = ZDB_FLAG_BSWAP;
10836 		flagbits['g'] = ZDB_FLAG_GBH;
10837 		flagbits['i'] = ZDB_FLAG_INDIRECT;
10838 		flagbits['r'] = ZDB_FLAG_RAW;
10839 		flagbits['v'] = ZDB_FLAG_VERBOSE;
10840 
10841 		for (int i = 0; i < argc; i++)
10842 			zdb_read_block(argv[i], spa);
10843 	}
10844 
10845 	if (dump_opt['k']) {
10846 		free(checkpoint_pool);
10847 		if (!target_is_spa)
10848 			free(checkpoint_target);
10849 	}
10850 
10851 fini:
10852 	if (spa != NULL)
10853 		zdb_ddt_cleanup(spa);
10854 
10855 	if (os != NULL) {
10856 		close_objset(os, FTAG);
10857 	} else if (spa != NULL) {
10858 		spa_close(spa, FTAG);
10859 	}
10860 
10861 	fuid_table_destroy();
10862 
10863 	dump_debug_buffer();
10864 
10865 	if (kernel_init_done)
10866 		kernel_fini();
10867 
10868 	if (corruption_found && error == 0)
10869 		error = 3;
10870 
10871 	return (error);
10872 }
10873