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