xref: /illumos-gate/usr/src/uts/common/fs/zfs/ddt.c (revision 3f9d6ad73e45c6823b409f93b0c8d4f62861d2d5)
1b24ab676SJeff Bonwick /*
2b24ab676SJeff Bonwick  * CDDL HEADER START
3b24ab676SJeff Bonwick  *
4b24ab676SJeff Bonwick  * The contents of this file are subject to the terms of the
5b24ab676SJeff Bonwick  * Common Development and Distribution License (the "License").
6b24ab676SJeff Bonwick  * You may not use this file except in compliance with the License.
7b24ab676SJeff Bonwick  *
8b24ab676SJeff Bonwick  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9b24ab676SJeff Bonwick  * or http://www.opensolaris.org/os/licensing.
10b24ab676SJeff Bonwick  * See the License for the specific language governing permissions
11b24ab676SJeff Bonwick  * and limitations under the License.
12b24ab676SJeff Bonwick  *
13b24ab676SJeff Bonwick  * When distributing Covered Code, include this CDDL HEADER in each
14b24ab676SJeff Bonwick  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15b24ab676SJeff Bonwick  * If applicable, add the following below this CDDL HEADER, with the
16b24ab676SJeff Bonwick  * fields enclosed by brackets "[]" replaced with your own identifying
17b24ab676SJeff Bonwick  * information: Portions Copyright [yyyy] [name of copyright owner]
18b24ab676SJeff Bonwick  *
19b24ab676SJeff Bonwick  * CDDL HEADER END
20b24ab676SJeff Bonwick  */
21b24ab676SJeff Bonwick 
22b24ab676SJeff Bonwick /*
23*3f9d6ad7SLin Ling  * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
24b24ab676SJeff Bonwick  */
25b24ab676SJeff Bonwick 
26b24ab676SJeff Bonwick #include <sys/zfs_context.h>
27b24ab676SJeff Bonwick #include <sys/spa.h>
28b24ab676SJeff Bonwick #include <sys/spa_impl.h>
29b24ab676SJeff Bonwick #include <sys/zio.h>
30b24ab676SJeff Bonwick #include <sys/ddt.h>
31b24ab676SJeff Bonwick #include <sys/zap.h>
32b24ab676SJeff Bonwick #include <sys/dmu_tx.h>
33b24ab676SJeff Bonwick #include <sys/arc.h>
34bbfd46c4SJeff Bonwick #include <sys/dsl_pool.h>
35b24ab676SJeff Bonwick #include <sys/zio_checksum.h>
36b24ab676SJeff Bonwick #include <sys/zio_compress.h>
37*3f9d6ad7SLin Ling #include <sys/dsl_scan.h>
38b24ab676SJeff Bonwick 
39b24ab676SJeff Bonwick static const ddt_ops_t *ddt_ops[DDT_TYPES] = {
40b24ab676SJeff Bonwick 	&ddt_zap_ops,
41b24ab676SJeff Bonwick };
42b24ab676SJeff Bonwick 
43b24ab676SJeff Bonwick static const char *ddt_class_name[DDT_CLASSES] = {
44b24ab676SJeff Bonwick 	"ditto",
45b24ab676SJeff Bonwick 	"duplicate",
46b24ab676SJeff Bonwick 	"unique",
47b24ab676SJeff Bonwick };
48b24ab676SJeff Bonwick 
49b24ab676SJeff Bonwick static void
50b24ab676SJeff Bonwick ddt_object_create(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
51b24ab676SJeff Bonwick     dmu_tx_t *tx)
52b24ab676SJeff Bonwick {
53b24ab676SJeff Bonwick 	spa_t *spa = ddt->ddt_spa;
54b24ab676SJeff Bonwick 	objset_t *os = ddt->ddt_os;
55b24ab676SJeff Bonwick 	uint64_t *objectp = &ddt->ddt_object[type][class];
56b24ab676SJeff Bonwick 	boolean_t prehash = zio_checksum_table[ddt->ddt_checksum].ci_dedup;
57b24ab676SJeff Bonwick 	char name[DDT_NAMELEN];
58b24ab676SJeff Bonwick 
59b24ab676SJeff Bonwick 	ddt_object_name(ddt, type, class, name);
60b24ab676SJeff Bonwick 
61b24ab676SJeff Bonwick 	ASSERT(*objectp == 0);
62b24ab676SJeff Bonwick 	VERIFY(ddt_ops[type]->ddt_op_create(os, objectp, tx, prehash) == 0);
63b24ab676SJeff Bonwick 	ASSERT(*objectp != 0);
64b24ab676SJeff Bonwick 
65b24ab676SJeff Bonwick 	VERIFY(zap_add(os, DMU_POOL_DIRECTORY_OBJECT, name,
66b24ab676SJeff Bonwick 	    sizeof (uint64_t), 1, objectp, tx) == 0);
67b24ab676SJeff Bonwick 
68b24ab676SJeff Bonwick 	VERIFY(zap_add(os, spa->spa_ddt_stat_object, name,
69b24ab676SJeff Bonwick 	    sizeof (uint64_t), sizeof (ddt_histogram_t) / sizeof (uint64_t),
70b24ab676SJeff Bonwick 	    &ddt->ddt_histogram[type][class], tx) == 0);
71b24ab676SJeff Bonwick }
72b24ab676SJeff Bonwick 
73b24ab676SJeff Bonwick static void
74b24ab676SJeff Bonwick ddt_object_destroy(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
75b24ab676SJeff Bonwick     dmu_tx_t *tx)
76b24ab676SJeff Bonwick {
77b24ab676SJeff Bonwick 	spa_t *spa = ddt->ddt_spa;
78b24ab676SJeff Bonwick 	objset_t *os = ddt->ddt_os;
79b24ab676SJeff Bonwick 	uint64_t *objectp = &ddt->ddt_object[type][class];
80b24ab676SJeff Bonwick 	char name[DDT_NAMELEN];
81b24ab676SJeff Bonwick 
82b24ab676SJeff Bonwick 	ddt_object_name(ddt, type, class, name);
83b24ab676SJeff Bonwick 
84b24ab676SJeff Bonwick 	ASSERT(*objectp != 0);
85b24ab676SJeff Bonwick 	ASSERT(ddt_object_count(ddt, type, class) == 0);
86b24ab676SJeff Bonwick 	ASSERT(ddt_histogram_empty(&ddt->ddt_histogram[type][class]));
87b24ab676SJeff Bonwick 	VERIFY(zap_remove(os, DMU_POOL_DIRECTORY_OBJECT, name, tx) == 0);
88b24ab676SJeff Bonwick 	VERIFY(zap_remove(os, spa->spa_ddt_stat_object, name, tx) == 0);
89b24ab676SJeff Bonwick 	VERIFY(ddt_ops[type]->ddt_op_destroy(os, *objectp, tx) == 0);
9030f608a4SGeorge Wilson 	bzero(&ddt->ddt_object_stats[type][class], sizeof (ddt_object_t));
91b24ab676SJeff Bonwick 
92b24ab676SJeff Bonwick 	*objectp = 0;
93b24ab676SJeff Bonwick }
94b24ab676SJeff Bonwick 
95b24ab676SJeff Bonwick static int
96b24ab676SJeff Bonwick ddt_object_load(ddt_t *ddt, enum ddt_type type, enum ddt_class class)
97b24ab676SJeff Bonwick {
9830f608a4SGeorge Wilson 	ddt_object_t *ddo = &ddt->ddt_object_stats[type][class];
9930f608a4SGeorge Wilson 	dmu_object_info_t doi;
100b24ab676SJeff Bonwick 	char name[DDT_NAMELEN];
101b24ab676SJeff Bonwick 	int error;
102b24ab676SJeff Bonwick 
103b24ab676SJeff Bonwick 	ddt_object_name(ddt, type, class, name);
104b24ab676SJeff Bonwick 
105b24ab676SJeff Bonwick 	error = zap_lookup(ddt->ddt_os, DMU_POOL_DIRECTORY_OBJECT, name,
106b24ab676SJeff Bonwick 	    sizeof (uint64_t), 1, &ddt->ddt_object[type][class]);
107b24ab676SJeff Bonwick 
108b24ab676SJeff Bonwick 	if (error)
109b24ab676SJeff Bonwick 		return (error);
110b24ab676SJeff Bonwick 
111b24ab676SJeff Bonwick 	error = zap_lookup(ddt->ddt_os, ddt->ddt_spa->spa_ddt_stat_object, name,
112b24ab676SJeff Bonwick 	    sizeof (uint64_t), sizeof (ddt_histogram_t) / sizeof (uint64_t),
113b24ab676SJeff Bonwick 	    &ddt->ddt_histogram[type][class]);
114b24ab676SJeff Bonwick 
11530f608a4SGeorge Wilson 	/*
11630f608a4SGeorge Wilson 	 * Seed the cached statistics.
11730f608a4SGeorge Wilson 	 */
11830f608a4SGeorge Wilson 	VERIFY(ddt_object_info(ddt, type, class, &doi) == 0);
11930f608a4SGeorge Wilson 
12030f608a4SGeorge Wilson 	ddo->ddo_count = ddt_object_count(ddt, type, class);
12130f608a4SGeorge Wilson 	ddo->ddo_dspace = doi.doi_physical_blocks_512 << 9;
12230f608a4SGeorge Wilson 	ddo->ddo_mspace = doi.doi_fill_count * doi.doi_data_block_size;
12330f608a4SGeorge Wilson 
124b24ab676SJeff Bonwick 	ASSERT(error == 0);
125b24ab676SJeff Bonwick 	return (error);
126b24ab676SJeff Bonwick }
127b24ab676SJeff Bonwick 
128b24ab676SJeff Bonwick static void
129b24ab676SJeff Bonwick ddt_object_sync(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
130b24ab676SJeff Bonwick     dmu_tx_t *tx)
131b24ab676SJeff Bonwick {
13230f608a4SGeorge Wilson 	ddt_object_t *ddo = &ddt->ddt_object_stats[type][class];
13330f608a4SGeorge Wilson 	dmu_object_info_t doi;
134b24ab676SJeff Bonwick 	char name[DDT_NAMELEN];
135b24ab676SJeff Bonwick 
136b24ab676SJeff Bonwick 	ddt_object_name(ddt, type, class, name);
137b24ab676SJeff Bonwick 
138b24ab676SJeff Bonwick 	VERIFY(zap_update(ddt->ddt_os, ddt->ddt_spa->spa_ddt_stat_object, name,
139b24ab676SJeff Bonwick 	    sizeof (uint64_t), sizeof (ddt_histogram_t) / sizeof (uint64_t),
140b24ab676SJeff Bonwick 	    &ddt->ddt_histogram[type][class], tx) == 0);
14130f608a4SGeorge Wilson 
14230f608a4SGeorge Wilson 	/*
14330f608a4SGeorge Wilson 	 * Cache DDT statistics; this is the only time they'll change.
14430f608a4SGeorge Wilson 	 */
14530f608a4SGeorge Wilson 	VERIFY(ddt_object_info(ddt, type, class, &doi) == 0);
14630f608a4SGeorge Wilson 
14730f608a4SGeorge Wilson 	ddo->ddo_count = ddt_object_count(ddt, type, class);
14830f608a4SGeorge Wilson 	ddo->ddo_dspace = doi.doi_physical_blocks_512 << 9;
14930f608a4SGeorge Wilson 	ddo->ddo_mspace = doi.doi_fill_count * doi.doi_data_block_size;
150b24ab676SJeff Bonwick }
151b24ab676SJeff Bonwick 
152b24ab676SJeff Bonwick static int
153b24ab676SJeff Bonwick ddt_object_lookup(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
154b24ab676SJeff Bonwick     ddt_entry_t *dde)
155b24ab676SJeff Bonwick {
156b24ab676SJeff Bonwick 	if (!ddt_object_exists(ddt, type, class))
157b24ab676SJeff Bonwick 		return (ENOENT);
158b24ab676SJeff Bonwick 
159b24ab676SJeff Bonwick 	return (ddt_ops[type]->ddt_op_lookup(ddt->ddt_os,
160b24ab676SJeff Bonwick 	    ddt->ddt_object[type][class], dde));
161b24ab676SJeff Bonwick }
162b24ab676SJeff Bonwick 
163*3f9d6ad7SLin Ling int
164b24ab676SJeff Bonwick ddt_object_update(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
165b24ab676SJeff Bonwick     ddt_entry_t *dde, dmu_tx_t *tx)
166b24ab676SJeff Bonwick {
167b24ab676SJeff Bonwick 	ASSERT(ddt_object_exists(ddt, type, class));
168b24ab676SJeff Bonwick 
169b24ab676SJeff Bonwick 	return (ddt_ops[type]->ddt_op_update(ddt->ddt_os,
170b24ab676SJeff Bonwick 	    ddt->ddt_object[type][class], dde, tx));
171b24ab676SJeff Bonwick }
172b24ab676SJeff Bonwick 
173b24ab676SJeff Bonwick static int
174b24ab676SJeff Bonwick ddt_object_remove(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
175b24ab676SJeff Bonwick     ddt_entry_t *dde, dmu_tx_t *tx)
176b24ab676SJeff Bonwick {
177b24ab676SJeff Bonwick 	ASSERT(ddt_object_exists(ddt, type, class));
178b24ab676SJeff Bonwick 
179b24ab676SJeff Bonwick 	return (ddt_ops[type]->ddt_op_remove(ddt->ddt_os,
180b24ab676SJeff Bonwick 	    ddt->ddt_object[type][class], dde, tx));
181b24ab676SJeff Bonwick }
182b24ab676SJeff Bonwick 
183b24ab676SJeff Bonwick int
184b24ab676SJeff Bonwick ddt_object_walk(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
185bbfd46c4SJeff Bonwick     uint64_t *walk, ddt_entry_t *dde)
186b24ab676SJeff Bonwick {
187b24ab676SJeff Bonwick 	ASSERT(ddt_object_exists(ddt, type, class));
188b24ab676SJeff Bonwick 
189b24ab676SJeff Bonwick 	return (ddt_ops[type]->ddt_op_walk(ddt->ddt_os,
190b24ab676SJeff Bonwick 	    ddt->ddt_object[type][class], dde, walk));
191b24ab676SJeff Bonwick }
192b24ab676SJeff Bonwick 
193b24ab676SJeff Bonwick uint64_t
194b24ab676SJeff Bonwick ddt_object_count(ddt_t *ddt, enum ddt_type type, enum ddt_class class)
195b24ab676SJeff Bonwick {
196b24ab676SJeff Bonwick 	ASSERT(ddt_object_exists(ddt, type, class));
197b24ab676SJeff Bonwick 
198b24ab676SJeff Bonwick 	return (ddt_ops[type]->ddt_op_count(ddt->ddt_os,
199b24ab676SJeff Bonwick 	    ddt->ddt_object[type][class]));
200b24ab676SJeff Bonwick }
201b24ab676SJeff Bonwick 
202b24ab676SJeff Bonwick int
203b24ab676SJeff Bonwick ddt_object_info(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
204b24ab676SJeff Bonwick     dmu_object_info_t *doi)
205b24ab676SJeff Bonwick {
206b24ab676SJeff Bonwick 	if (!ddt_object_exists(ddt, type, class))
207b24ab676SJeff Bonwick 		return (ENOENT);
208b24ab676SJeff Bonwick 
209b24ab676SJeff Bonwick 	return (dmu_object_info(ddt->ddt_os, ddt->ddt_object[type][class],
210b24ab676SJeff Bonwick 	    doi));
211b24ab676SJeff Bonwick }
212b24ab676SJeff Bonwick 
213b24ab676SJeff Bonwick boolean_t
214b24ab676SJeff Bonwick ddt_object_exists(ddt_t *ddt, enum ddt_type type, enum ddt_class class)
215b24ab676SJeff Bonwick {
216b24ab676SJeff Bonwick 	return (!!ddt->ddt_object[type][class]);
217b24ab676SJeff Bonwick }
218b24ab676SJeff Bonwick 
219b24ab676SJeff Bonwick void
220b24ab676SJeff Bonwick ddt_object_name(ddt_t *ddt, enum ddt_type type, enum ddt_class class,
221b24ab676SJeff Bonwick     char *name)
222b24ab676SJeff Bonwick {
223b24ab676SJeff Bonwick 	(void) sprintf(name, DMU_POOL_DDT,
224b24ab676SJeff Bonwick 	    zio_checksum_table[ddt->ddt_checksum].ci_name,
225b24ab676SJeff Bonwick 	    ddt_ops[type]->ddt_op_name, ddt_class_name[class]);
226b24ab676SJeff Bonwick }
227b24ab676SJeff Bonwick 
228b24ab676SJeff Bonwick void
229b24ab676SJeff Bonwick ddt_bp_fill(const ddt_phys_t *ddp, blkptr_t *bp, uint64_t txg)
230b24ab676SJeff Bonwick {
231b24ab676SJeff Bonwick 	ASSERT(txg != 0);
232b24ab676SJeff Bonwick 
233b24ab676SJeff Bonwick 	for (int d = 0; d < SPA_DVAS_PER_BP; d++)
234b24ab676SJeff Bonwick 		bp->blk_dva[d] = ddp->ddp_dva[d];
235b24ab676SJeff Bonwick 	BP_SET_BIRTH(bp, txg, ddp->ddp_phys_birth);
236b24ab676SJeff Bonwick }
237b24ab676SJeff Bonwick 
238b24ab676SJeff Bonwick void
239bbfd46c4SJeff Bonwick ddt_bp_create(enum zio_checksum checksum,
240bbfd46c4SJeff Bonwick     const ddt_key_t *ddk, const ddt_phys_t *ddp, blkptr_t *bp)
241b24ab676SJeff Bonwick {
242b24ab676SJeff Bonwick 	BP_ZERO(bp);
243b24ab676SJeff Bonwick 
244b24ab676SJeff Bonwick 	if (ddp != NULL)
245b24ab676SJeff Bonwick 		ddt_bp_fill(ddp, bp, ddp->ddp_phys_birth);
246b24ab676SJeff Bonwick 
247b24ab676SJeff Bonwick 	bp->blk_cksum = ddk->ddk_cksum;
248*3f9d6ad7SLin Ling 	bp->blk_fill = 1;
249b24ab676SJeff Bonwick 
250b24ab676SJeff Bonwick 	BP_SET_LSIZE(bp, DDK_GET_LSIZE(ddk));
251b24ab676SJeff Bonwick 	BP_SET_PSIZE(bp, DDK_GET_PSIZE(ddk));
252b24ab676SJeff Bonwick 	BP_SET_COMPRESS(bp, DDK_GET_COMPRESS(ddk));
253bbfd46c4SJeff Bonwick 	BP_SET_CHECKSUM(bp, checksum);
254*3f9d6ad7SLin Ling 	BP_SET_TYPE(bp, DMU_OT_DEDUP);
255b24ab676SJeff Bonwick 	BP_SET_LEVEL(bp, 0);
256b24ab676SJeff Bonwick 	BP_SET_DEDUP(bp, 0);
257b24ab676SJeff Bonwick 	BP_SET_BYTEORDER(bp, ZFS_HOST_BYTEORDER);
258b24ab676SJeff Bonwick }
259b24ab676SJeff Bonwick 
260b24ab676SJeff Bonwick void
261b24ab676SJeff Bonwick ddt_key_fill(ddt_key_t *ddk, const blkptr_t *bp)
262b24ab676SJeff Bonwick {
263b24ab676SJeff Bonwick 	ddk->ddk_cksum = bp->blk_cksum;
264b24ab676SJeff Bonwick 	ddk->ddk_prop = 0;
265b24ab676SJeff Bonwick 
266b24ab676SJeff Bonwick 	DDK_SET_LSIZE(ddk, BP_GET_LSIZE(bp));
267b24ab676SJeff Bonwick 	DDK_SET_PSIZE(ddk, BP_GET_PSIZE(bp));
268b24ab676SJeff Bonwick 	DDK_SET_COMPRESS(ddk, BP_GET_COMPRESS(bp));
269b24ab676SJeff Bonwick }
270b24ab676SJeff Bonwick 
271b24ab676SJeff Bonwick void
272b24ab676SJeff Bonwick ddt_phys_fill(ddt_phys_t *ddp, const blkptr_t *bp)
273b24ab676SJeff Bonwick {
274b24ab676SJeff Bonwick 	ASSERT(ddp->ddp_phys_birth == 0);
275b24ab676SJeff Bonwick 
276b24ab676SJeff Bonwick 	for (int d = 0; d < SPA_DVAS_PER_BP; d++)
277b24ab676SJeff Bonwick 		ddp->ddp_dva[d] = bp->blk_dva[d];
278b24ab676SJeff Bonwick 	ddp->ddp_phys_birth = BP_PHYSICAL_BIRTH(bp);
279b24ab676SJeff Bonwick }
280b24ab676SJeff Bonwick 
281b24ab676SJeff Bonwick void
282b24ab676SJeff Bonwick ddt_phys_clear(ddt_phys_t *ddp)
283b24ab676SJeff Bonwick {
284b24ab676SJeff Bonwick 	bzero(ddp, sizeof (*ddp));
285b24ab676SJeff Bonwick }
286b24ab676SJeff Bonwick 
287b24ab676SJeff Bonwick void
288b24ab676SJeff Bonwick ddt_phys_addref(ddt_phys_t *ddp)
289b24ab676SJeff Bonwick {
290b24ab676SJeff Bonwick 	ddp->ddp_refcnt++;
291b24ab676SJeff Bonwick }
292b24ab676SJeff Bonwick 
293b24ab676SJeff Bonwick void
294b24ab676SJeff Bonwick ddt_phys_decref(ddt_phys_t *ddp)
295b24ab676SJeff Bonwick {
296b24ab676SJeff Bonwick 	ASSERT((int64_t)ddp->ddp_refcnt > 0);
297b24ab676SJeff Bonwick 	ddp->ddp_refcnt--;
298b24ab676SJeff Bonwick }
299b24ab676SJeff Bonwick 
300b24ab676SJeff Bonwick void
301b24ab676SJeff Bonwick ddt_phys_free(ddt_t *ddt, ddt_key_t *ddk, ddt_phys_t *ddp, uint64_t txg)
302b24ab676SJeff Bonwick {
303b24ab676SJeff Bonwick 	blkptr_t blk;
304b24ab676SJeff Bonwick 
305bbfd46c4SJeff Bonwick 	ddt_bp_create(ddt->ddt_checksum, ddk, ddp, &blk);
306b24ab676SJeff Bonwick 	ddt_phys_clear(ddp);
307b24ab676SJeff Bonwick 	zio_free(ddt->ddt_spa, txg, &blk);
308b24ab676SJeff Bonwick }
309b24ab676SJeff Bonwick 
310b24ab676SJeff Bonwick ddt_phys_t *
311b24ab676SJeff Bonwick ddt_phys_select(const ddt_entry_t *dde, const blkptr_t *bp)
312b24ab676SJeff Bonwick {
313b24ab676SJeff Bonwick 	ddt_phys_t *ddp = (ddt_phys_t *)dde->dde_phys;
314b24ab676SJeff Bonwick 
315b24ab676SJeff Bonwick 	for (int p = 0; p < DDT_PHYS_TYPES; p++, ddp++) {
316b24ab676SJeff Bonwick 		if (DVA_EQUAL(BP_IDENTITY(bp), &ddp->ddp_dva[0]) &&
317b24ab676SJeff Bonwick 		    BP_PHYSICAL_BIRTH(bp) == ddp->ddp_phys_birth)
318b24ab676SJeff Bonwick 			return (ddp);
319b24ab676SJeff Bonwick 	}
320b24ab676SJeff Bonwick 	return (NULL);
321b24ab676SJeff Bonwick }
322b24ab676SJeff Bonwick 
323b24ab676SJeff Bonwick uint64_t
324b24ab676SJeff Bonwick ddt_phys_total_refcnt(const ddt_entry_t *dde)
325b24ab676SJeff Bonwick {
326b24ab676SJeff Bonwick 	uint64_t refcnt = 0;
327b24ab676SJeff Bonwick 
328b24ab676SJeff Bonwick 	for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++)
329b24ab676SJeff Bonwick 		refcnt += dde->dde_phys[p].ddp_refcnt;
330b24ab676SJeff Bonwick 
331b24ab676SJeff Bonwick 	return (refcnt);
332b24ab676SJeff Bonwick }
333b24ab676SJeff Bonwick 
334b24ab676SJeff Bonwick static void
335b24ab676SJeff Bonwick ddt_stat_generate(ddt_t *ddt, ddt_entry_t *dde, ddt_stat_t *dds)
336b24ab676SJeff Bonwick {
337b24ab676SJeff Bonwick 	spa_t *spa = ddt->ddt_spa;
338b24ab676SJeff Bonwick 	ddt_phys_t *ddp = dde->dde_phys;
339b24ab676SJeff Bonwick 	ddt_key_t *ddk = &dde->dde_key;
340b24ab676SJeff Bonwick 	uint64_t lsize = DDK_GET_LSIZE(ddk);
341b24ab676SJeff Bonwick 	uint64_t psize = DDK_GET_PSIZE(ddk);
342b24ab676SJeff Bonwick 
343b24ab676SJeff Bonwick 	bzero(dds, sizeof (*dds));
344b24ab676SJeff Bonwick 
345b24ab676SJeff Bonwick 	for (int p = 0; p < DDT_PHYS_TYPES; p++, ddp++) {
346b24ab676SJeff Bonwick 		uint64_t dsize = 0;
347b24ab676SJeff Bonwick 		uint64_t refcnt = ddp->ddp_refcnt;
348b24ab676SJeff Bonwick 
349b24ab676SJeff Bonwick 		if (ddp->ddp_phys_birth == 0)
350b24ab676SJeff Bonwick 			continue;
351b24ab676SJeff Bonwick 
352b24ab676SJeff Bonwick 		for (int d = 0; d < SPA_DVAS_PER_BP; d++)
353b24ab676SJeff Bonwick 			dsize += dva_get_dsize_sync(spa, &ddp->ddp_dva[d]);
354b24ab676SJeff Bonwick 
355b24ab676SJeff Bonwick 		dds->dds_blocks += 1;
356b24ab676SJeff Bonwick 		dds->dds_lsize += lsize;
357b24ab676SJeff Bonwick 		dds->dds_psize += psize;
358b24ab676SJeff Bonwick 		dds->dds_dsize += dsize;
359b24ab676SJeff Bonwick 
360b24ab676SJeff Bonwick 		dds->dds_ref_blocks += refcnt;
361b24ab676SJeff Bonwick 		dds->dds_ref_lsize += lsize * refcnt;
362b24ab676SJeff Bonwick 		dds->dds_ref_psize += psize * refcnt;
363b24ab676SJeff Bonwick 		dds->dds_ref_dsize += dsize * refcnt;
364b24ab676SJeff Bonwick 	}
365b24ab676SJeff Bonwick }
366b24ab676SJeff Bonwick 
367b24ab676SJeff Bonwick void
368b24ab676SJeff Bonwick ddt_stat_add(ddt_stat_t *dst, const ddt_stat_t *src, uint64_t neg)
369b24ab676SJeff Bonwick {
370b24ab676SJeff Bonwick 	const uint64_t *s = (const uint64_t *)src;
371b24ab676SJeff Bonwick 	uint64_t *d = (uint64_t *)dst;
372b24ab676SJeff Bonwick 	uint64_t *d_end = (uint64_t *)(dst + 1);
373b24ab676SJeff Bonwick 
374b24ab676SJeff Bonwick 	ASSERT(neg == 0 || neg == -1ULL);	/* add or subtract */
375b24ab676SJeff Bonwick 
376b24ab676SJeff Bonwick 	while (d < d_end)
377b24ab676SJeff Bonwick 		*d++ += (*s++ ^ neg) - neg;
378b24ab676SJeff Bonwick }
379b24ab676SJeff Bonwick 
380b24ab676SJeff Bonwick static void
381b24ab676SJeff Bonwick ddt_stat_update(ddt_t *ddt, ddt_entry_t *dde, uint64_t neg)
382b24ab676SJeff Bonwick {
383b24ab676SJeff Bonwick 	ddt_stat_t dds;
384b24ab676SJeff Bonwick 	ddt_histogram_t *ddh;
385b24ab676SJeff Bonwick 	int bucket;
386b24ab676SJeff Bonwick 
387b24ab676SJeff Bonwick 	ddt_stat_generate(ddt, dde, &dds);
388b24ab676SJeff Bonwick 
389b24ab676SJeff Bonwick 	bucket = highbit(dds.dds_ref_blocks) - 1;
390b24ab676SJeff Bonwick 	ASSERT(bucket >= 0);
391b24ab676SJeff Bonwick 
392b24ab676SJeff Bonwick 	ddh = &ddt->ddt_histogram[dde->dde_type][dde->dde_class];
393b24ab676SJeff Bonwick 
394b24ab676SJeff Bonwick 	ddt_stat_add(&ddh->ddh_stat[bucket], &dds, neg);
395b24ab676SJeff Bonwick }
396b24ab676SJeff Bonwick 
397b24ab676SJeff Bonwick void
398b24ab676SJeff Bonwick ddt_histogram_add(ddt_histogram_t *dst, const ddt_histogram_t *src)
399b24ab676SJeff Bonwick {
400b24ab676SJeff Bonwick 	for (int h = 0; h < 64; h++)
401b24ab676SJeff Bonwick 		ddt_stat_add(&dst->ddh_stat[h], &src->ddh_stat[h], 0);
402b24ab676SJeff Bonwick }
403b24ab676SJeff Bonwick 
404b24ab676SJeff Bonwick void
405b24ab676SJeff Bonwick ddt_histogram_stat(ddt_stat_t *dds, const ddt_histogram_t *ddh)
406b24ab676SJeff Bonwick {
407b24ab676SJeff Bonwick 	bzero(dds, sizeof (*dds));
408b24ab676SJeff Bonwick 
409b24ab676SJeff Bonwick 	for (int h = 0; h < 64; h++)
410b24ab676SJeff Bonwick 		ddt_stat_add(dds, &ddh->ddh_stat[h], 0);
411b24ab676SJeff Bonwick }
412b24ab676SJeff Bonwick 
413b24ab676SJeff Bonwick boolean_t
414b24ab676SJeff Bonwick ddt_histogram_empty(const ddt_histogram_t *ddh)
415b24ab676SJeff Bonwick {
416b24ab676SJeff Bonwick 	const uint64_t *s = (const uint64_t *)ddh;
417b24ab676SJeff Bonwick 	const uint64_t *s_end = (const uint64_t *)(ddh + 1);
418b24ab676SJeff Bonwick 
419b24ab676SJeff Bonwick 	while (s < s_end)
420b24ab676SJeff Bonwick 		if (*s++ != 0)
421b24ab676SJeff Bonwick 			return (B_FALSE);
422b24ab676SJeff Bonwick 
423b24ab676SJeff Bonwick 	return (B_TRUE);
424b24ab676SJeff Bonwick }
425b24ab676SJeff Bonwick 
4269eb19f4dSGeorge Wilson void
42730f608a4SGeorge Wilson ddt_get_dedup_object_stats(spa_t *spa, ddt_object_t *ddo_total)
428b24ab676SJeff Bonwick {
42930f608a4SGeorge Wilson 	/* Sum the statistics we cached in ddt_object_sync(). */
430b24ab676SJeff Bonwick 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
431b24ab676SJeff Bonwick 		ddt_t *ddt = spa->spa_ddt[c];
432b24ab676SJeff Bonwick 		for (enum ddt_type type = 0; type < DDT_TYPES; type++) {
433b24ab676SJeff Bonwick 			for (enum ddt_class class = 0; class < DDT_CLASSES;
434b24ab676SJeff Bonwick 			    class++) {
43530f608a4SGeorge Wilson 				ddt_object_t *ddo =
43630f608a4SGeorge Wilson 				    &ddt->ddt_object_stats[type][class];
43730f608a4SGeorge Wilson 				ddo_total->ddo_count += ddo->ddo_count;
43830f608a4SGeorge Wilson 				ddo_total->ddo_dspace += ddo->ddo_dspace;
43930f608a4SGeorge Wilson 				ddo_total->ddo_mspace += ddo->ddo_mspace;
4409eb19f4dSGeorge Wilson 			}
4419eb19f4dSGeorge Wilson 		}
4429eb19f4dSGeorge Wilson 	}
44330f608a4SGeorge Wilson 
44430f608a4SGeorge Wilson 	/* ... and compute the averages. */
44530f608a4SGeorge Wilson 	if (ddo_total->ddo_count != 0) {
44630f608a4SGeorge Wilson 		ddo_total->ddo_dspace /= ddo_total->ddo_count;
44730f608a4SGeorge Wilson 		ddo_total->ddo_mspace /= ddo_total->ddo_count;
44830f608a4SGeorge Wilson 	} else {
44930f608a4SGeorge Wilson 		ASSERT(ddo_total->ddo_dspace == 0);
45030f608a4SGeorge Wilson 		ASSERT(ddo_total->ddo_mspace == 0);
45130f608a4SGeorge Wilson 	}
4529eb19f4dSGeorge Wilson }
4539eb19f4dSGeorge Wilson 
4549eb19f4dSGeorge Wilson void
4559eb19f4dSGeorge Wilson ddt_get_dedup_histogram(spa_t *spa, ddt_histogram_t *ddh)
4569eb19f4dSGeorge Wilson {
4579eb19f4dSGeorge Wilson 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
4589eb19f4dSGeorge Wilson 		ddt_t *ddt = spa->spa_ddt[c];
4599eb19f4dSGeorge Wilson 		for (enum ddt_type type = 0; type < DDT_TYPES; type++) {
4609eb19f4dSGeorge Wilson 			for (enum ddt_class class = 0; class < DDT_CLASSES;
4619eb19f4dSGeorge Wilson 			    class++) {
4629eb19f4dSGeorge Wilson 				ddt_histogram_add(ddh,
46330f608a4SGeorge Wilson 				    &ddt->ddt_histogram_cache[type][class]);
464b24ab676SJeff Bonwick 			}
465b24ab676SJeff Bonwick 		}
466b24ab676SJeff Bonwick 	}
4679eb19f4dSGeorge Wilson }
4689eb19f4dSGeorge Wilson 
4699eb19f4dSGeorge Wilson void
4709eb19f4dSGeorge Wilson ddt_get_dedup_stats(spa_t *spa, ddt_stat_t *dds_total)
4719eb19f4dSGeorge Wilson {
4729eb19f4dSGeorge Wilson 	ddt_histogram_t *ddh_total;
473b24ab676SJeff Bonwick 
4749eb19f4dSGeorge Wilson 	ddh_total = kmem_zalloc(sizeof (ddt_histogram_t), KM_SLEEP);
4759eb19f4dSGeorge Wilson 	ddt_get_dedup_histogram(spa, ddh_total);
4769eb19f4dSGeorge Wilson 	ddt_histogram_stat(dds_total, ddh_total);
4779eb19f4dSGeorge Wilson 	kmem_free(ddh_total, sizeof (ddt_histogram_t));
478485bbbf5SGeorge Wilson }
479485bbbf5SGeorge Wilson 
480485bbbf5SGeorge Wilson uint64_t
481485bbbf5SGeorge Wilson ddt_get_dedup_dspace(spa_t *spa)
482485bbbf5SGeorge Wilson {
483485bbbf5SGeorge Wilson 	ddt_stat_t dds_total = { 0 };
484485bbbf5SGeorge Wilson 
485485bbbf5SGeorge Wilson 	ddt_get_dedup_stats(spa, &dds_total);
486485bbbf5SGeorge Wilson 	return (dds_total.dds_ref_dsize - dds_total.dds_dsize);
487485bbbf5SGeorge Wilson }
488485bbbf5SGeorge Wilson 
489485bbbf5SGeorge Wilson uint64_t
490485bbbf5SGeorge Wilson ddt_get_pool_dedup_ratio(spa_t *spa)
491485bbbf5SGeorge Wilson {
492485bbbf5SGeorge Wilson 	ddt_stat_t dds_total = { 0 };
493b24ab676SJeff Bonwick 
494485bbbf5SGeorge Wilson 	ddt_get_dedup_stats(spa, &dds_total);
495b24ab676SJeff Bonwick 	if (dds_total.dds_dsize == 0)
496b24ab676SJeff Bonwick 		return (100);
497b24ab676SJeff Bonwick 
498b24ab676SJeff Bonwick 	return (dds_total.dds_ref_dsize * 100 / dds_total.dds_dsize);
499b24ab676SJeff Bonwick }
500b24ab676SJeff Bonwick 
501b24ab676SJeff Bonwick int
502b24ab676SJeff Bonwick ddt_ditto_copies_needed(ddt_t *ddt, ddt_entry_t *dde, ddt_phys_t *ddp_willref)
503b24ab676SJeff Bonwick {
504b24ab676SJeff Bonwick 	spa_t *spa = ddt->ddt_spa;
505b24ab676SJeff Bonwick 	uint64_t total_refcnt = 0;
506b24ab676SJeff Bonwick 	uint64_t ditto = spa->spa_dedup_ditto;
507b24ab676SJeff Bonwick 	int total_copies = 0;
508b24ab676SJeff Bonwick 	int desired_copies = 0;
509b24ab676SJeff Bonwick 
510b24ab676SJeff Bonwick 	for (int p = DDT_PHYS_SINGLE; p <= DDT_PHYS_TRIPLE; p++) {
511b24ab676SJeff Bonwick 		ddt_phys_t *ddp = &dde->dde_phys[p];
512b24ab676SJeff Bonwick 		zio_t *zio = dde->dde_lead_zio[p];
513b24ab676SJeff Bonwick 		uint64_t refcnt = ddp->ddp_refcnt;	/* committed refs */
514b24ab676SJeff Bonwick 		if (zio != NULL)
515b24ab676SJeff Bonwick 			refcnt += zio->io_parent_count;	/* pending refs */
516b24ab676SJeff Bonwick 		if (ddp == ddp_willref)
517b24ab676SJeff Bonwick 			refcnt++;			/* caller's ref */
518b24ab676SJeff Bonwick 		if (refcnt != 0) {
519b24ab676SJeff Bonwick 			total_refcnt += refcnt;
520b24ab676SJeff Bonwick 			total_copies += p;
521b24ab676SJeff Bonwick 		}
522b24ab676SJeff Bonwick 	}
523b24ab676SJeff Bonwick 
524b24ab676SJeff Bonwick 	if (ditto == 0 || ditto > UINT32_MAX)
525b24ab676SJeff Bonwick 		ditto = UINT32_MAX;
526b24ab676SJeff Bonwick 
527b24ab676SJeff Bonwick 	if (total_refcnt >= 1)
528b24ab676SJeff Bonwick 		desired_copies++;
529b24ab676SJeff Bonwick 	if (total_refcnt >= ditto)
530b24ab676SJeff Bonwick 		desired_copies++;
531b24ab676SJeff Bonwick 	if (total_refcnt >= ditto * ditto)
532b24ab676SJeff Bonwick 		desired_copies++;
533b24ab676SJeff Bonwick 
534b24ab676SJeff Bonwick 	return (MAX(desired_copies, total_copies) - total_copies);
535b24ab676SJeff Bonwick }
536b24ab676SJeff Bonwick 
537b24ab676SJeff Bonwick int
538b24ab676SJeff Bonwick ddt_ditto_copies_present(ddt_entry_t *dde)
539b24ab676SJeff Bonwick {
540b24ab676SJeff Bonwick 	ddt_phys_t *ddp = &dde->dde_phys[DDT_PHYS_DITTO];
541b24ab676SJeff Bonwick 	dva_t *dva = ddp->ddp_dva;
542b24ab676SJeff Bonwick 	int copies = 0 - DVA_GET_GANG(dva);
543b24ab676SJeff Bonwick 
544b24ab676SJeff Bonwick 	for (int d = 0; d < SPA_DVAS_PER_BP; d++, dva++)
545b24ab676SJeff Bonwick 		if (DVA_IS_VALID(dva))
546b24ab676SJeff Bonwick 			copies++;
547b24ab676SJeff Bonwick 
548b24ab676SJeff Bonwick 	ASSERT(copies >= 0 && copies < SPA_DVAS_PER_BP);
549b24ab676SJeff Bonwick 
550b24ab676SJeff Bonwick 	return (copies);
551b24ab676SJeff Bonwick }
552b24ab676SJeff Bonwick 
553b24ab676SJeff Bonwick size_t
554b24ab676SJeff Bonwick ddt_compress(void *src, uchar_t *dst, size_t s_len, size_t d_len)
555b24ab676SJeff Bonwick {
556b24ab676SJeff Bonwick 	uchar_t *version = dst++;
557b24ab676SJeff Bonwick 	int cpfunc = ZIO_COMPRESS_ZLE;
558b24ab676SJeff Bonwick 	zio_compress_info_t *ci = &zio_compress_table[cpfunc];
559b24ab676SJeff Bonwick 	size_t c_len;
560b24ab676SJeff Bonwick 
561b24ab676SJeff Bonwick 	ASSERT(d_len >= s_len + 1);	/* no compression plus version byte */
562b24ab676SJeff Bonwick 
563b24ab676SJeff Bonwick 	c_len = ci->ci_compress(src, dst, s_len, d_len - 1, ci->ci_level);
564b24ab676SJeff Bonwick 
565b24ab676SJeff Bonwick 	if (c_len == s_len) {
566b24ab676SJeff Bonwick 		cpfunc = ZIO_COMPRESS_OFF;
567b24ab676SJeff Bonwick 		bcopy(src, dst, s_len);
568b24ab676SJeff Bonwick 	}
569b24ab676SJeff Bonwick 
570b24ab676SJeff Bonwick 	*version = (ZFS_HOST_BYTEORDER & DDT_COMPRESS_BYTEORDER_MASK) | cpfunc;
571b24ab676SJeff Bonwick 
572b24ab676SJeff Bonwick 	return (c_len + 1);
573b24ab676SJeff Bonwick }
574b24ab676SJeff Bonwick 
575b24ab676SJeff Bonwick void
576b24ab676SJeff Bonwick ddt_decompress(uchar_t *src, void *dst, size_t s_len, size_t d_len)
577b24ab676SJeff Bonwick {
578b24ab676SJeff Bonwick 	uchar_t version = *src++;
579b24ab676SJeff Bonwick 	int cpfunc = version & DDT_COMPRESS_FUNCTION_MASK;
580b24ab676SJeff Bonwick 	zio_compress_info_t *ci = &zio_compress_table[cpfunc];
581b24ab676SJeff Bonwick 
582b24ab676SJeff Bonwick 	if (ci->ci_decompress != NULL)
583b24ab676SJeff Bonwick 		(void) ci->ci_decompress(src, dst, s_len, d_len, ci->ci_level);
584b24ab676SJeff Bonwick 	else
585b24ab676SJeff Bonwick 		bcopy(src, dst, d_len);
586b24ab676SJeff Bonwick 
587b24ab676SJeff Bonwick 	if ((version ^ ZFS_HOST_BYTEORDER) & DDT_COMPRESS_BYTEORDER_MASK)
588b24ab676SJeff Bonwick 		byteswap_uint64_array(dst, d_len);
589b24ab676SJeff Bonwick }
590b24ab676SJeff Bonwick 
591b24ab676SJeff Bonwick ddt_t *
592b24ab676SJeff Bonwick ddt_select_by_checksum(spa_t *spa, enum zio_checksum c)
593b24ab676SJeff Bonwick {
594b24ab676SJeff Bonwick 	return (spa->spa_ddt[c]);
595b24ab676SJeff Bonwick }
596b24ab676SJeff Bonwick 
597b24ab676SJeff Bonwick ddt_t *
598b24ab676SJeff Bonwick ddt_select(spa_t *spa, const blkptr_t *bp)
599b24ab676SJeff Bonwick {
600b24ab676SJeff Bonwick 	return (spa->spa_ddt[BP_GET_CHECKSUM(bp)]);
601b24ab676SJeff Bonwick }
602b24ab676SJeff Bonwick 
603b24ab676SJeff Bonwick void
604b24ab676SJeff Bonwick ddt_enter(ddt_t *ddt)
605b24ab676SJeff Bonwick {
606b24ab676SJeff Bonwick 	mutex_enter(&ddt->ddt_lock);
607b24ab676SJeff Bonwick }
608b24ab676SJeff Bonwick 
609b24ab676SJeff Bonwick void
610b24ab676SJeff Bonwick ddt_exit(ddt_t *ddt)
611b24ab676SJeff Bonwick {
612b24ab676SJeff Bonwick 	mutex_exit(&ddt->ddt_lock);
613b24ab676SJeff Bonwick }
614b24ab676SJeff Bonwick 
615b24ab676SJeff Bonwick static ddt_entry_t *
616b24ab676SJeff Bonwick ddt_alloc(const ddt_key_t *ddk)
617b24ab676SJeff Bonwick {
618b24ab676SJeff Bonwick 	ddt_entry_t *dde;
619b24ab676SJeff Bonwick 
620b24ab676SJeff Bonwick 	dde = kmem_zalloc(sizeof (ddt_entry_t), KM_SLEEP);
621b24ab676SJeff Bonwick 	cv_init(&dde->dde_cv, NULL, CV_DEFAULT, NULL);
622b24ab676SJeff Bonwick 
623b24ab676SJeff Bonwick 	dde->dde_key = *ddk;
624b24ab676SJeff Bonwick 
625b24ab676SJeff Bonwick 	return (dde);
626b24ab676SJeff Bonwick }
627b24ab676SJeff Bonwick 
628b24ab676SJeff Bonwick static void
629b24ab676SJeff Bonwick ddt_free(ddt_entry_t *dde)
630b24ab676SJeff Bonwick {
631b24ab676SJeff Bonwick 	ASSERT(!dde->dde_loading);
632b24ab676SJeff Bonwick 
633b24ab676SJeff Bonwick 	for (int p = 0; p < DDT_PHYS_TYPES; p++)
634b24ab676SJeff Bonwick 		ASSERT(dde->dde_lead_zio[p] == NULL);
635b24ab676SJeff Bonwick 
636b24ab676SJeff Bonwick 	if (dde->dde_repair_data != NULL)
637b24ab676SJeff Bonwick 		zio_buf_free(dde->dde_repair_data,
638b24ab676SJeff Bonwick 		    DDK_GET_PSIZE(&dde->dde_key));
639b24ab676SJeff Bonwick 
640b24ab676SJeff Bonwick 	cv_destroy(&dde->dde_cv);
641b24ab676SJeff Bonwick 	kmem_free(dde, sizeof (*dde));
642b24ab676SJeff Bonwick }
643b24ab676SJeff Bonwick 
644b24ab676SJeff Bonwick void
645b24ab676SJeff Bonwick ddt_remove(ddt_t *ddt, ddt_entry_t *dde)
646b24ab676SJeff Bonwick {
647b24ab676SJeff Bonwick 	ASSERT(MUTEX_HELD(&ddt->ddt_lock));
648b24ab676SJeff Bonwick 
649b24ab676SJeff Bonwick 	avl_remove(&ddt->ddt_tree, dde);
650b24ab676SJeff Bonwick 	ddt_free(dde);
651b24ab676SJeff Bonwick }
652b24ab676SJeff Bonwick 
653b24ab676SJeff Bonwick ddt_entry_t *
654b24ab676SJeff Bonwick ddt_lookup(ddt_t *ddt, const blkptr_t *bp, boolean_t add)
655b24ab676SJeff Bonwick {
656b24ab676SJeff Bonwick 	ddt_entry_t *dde, dde_search;
657b24ab676SJeff Bonwick 	enum ddt_type type;
658b24ab676SJeff Bonwick 	enum ddt_class class;
659b24ab676SJeff Bonwick 	avl_index_t where;
660b24ab676SJeff Bonwick 	int error;
661b24ab676SJeff Bonwick 
662b24ab676SJeff Bonwick 	ASSERT(MUTEX_HELD(&ddt->ddt_lock));
663b24ab676SJeff Bonwick 
664b24ab676SJeff Bonwick 	ddt_key_fill(&dde_search.dde_key, bp);
665b24ab676SJeff Bonwick 
666b24ab676SJeff Bonwick 	dde = avl_find(&ddt->ddt_tree, &dde_search, &where);
667b24ab676SJeff Bonwick 	if (dde == NULL) {
668b24ab676SJeff Bonwick 		if (!add)
669b24ab676SJeff Bonwick 			return (NULL);
670b24ab676SJeff Bonwick 		dde = ddt_alloc(&dde_search.dde_key);
671b24ab676SJeff Bonwick 		avl_insert(&ddt->ddt_tree, dde, where);
672b24ab676SJeff Bonwick 	}
673b24ab676SJeff Bonwick 
674b24ab676SJeff Bonwick 	while (dde->dde_loading)
675b24ab676SJeff Bonwick 		cv_wait(&dde->dde_cv, &ddt->ddt_lock);
676b24ab676SJeff Bonwick 
677b24ab676SJeff Bonwick 	if (dde->dde_loaded)
678b24ab676SJeff Bonwick 		return (dde);
679b24ab676SJeff Bonwick 
680b24ab676SJeff Bonwick 	dde->dde_loading = B_TRUE;
681b24ab676SJeff Bonwick 
682b24ab676SJeff Bonwick 	ddt_exit(ddt);
683b24ab676SJeff Bonwick 
684b24ab676SJeff Bonwick 	error = ENOENT;
685b24ab676SJeff Bonwick 
686b24ab676SJeff Bonwick 	for (type = 0; type < DDT_TYPES; type++) {
687b24ab676SJeff Bonwick 		for (class = 0; class < DDT_CLASSES; class++) {
688b24ab676SJeff Bonwick 			error = ddt_object_lookup(ddt, type, class, dde);
689b24ab676SJeff Bonwick 			if (error != ENOENT)
690b24ab676SJeff Bonwick 				break;
691b24ab676SJeff Bonwick 		}
692b24ab676SJeff Bonwick 		if (error != ENOENT)
693b24ab676SJeff Bonwick 			break;
694b24ab676SJeff Bonwick 	}
695b24ab676SJeff Bonwick 
696b24ab676SJeff Bonwick 	ASSERT(error == 0 || error == ENOENT);
697b24ab676SJeff Bonwick 
698b24ab676SJeff Bonwick 	ddt_enter(ddt);
699b24ab676SJeff Bonwick 
700b24ab676SJeff Bonwick 	ASSERT(dde->dde_loaded == B_FALSE);
701b24ab676SJeff Bonwick 	ASSERT(dde->dde_loading == B_TRUE);
702b24ab676SJeff Bonwick 
703b24ab676SJeff Bonwick 	dde->dde_type = type;	/* will be DDT_TYPES if no entry found */
704b24ab676SJeff Bonwick 	dde->dde_class = class;	/* will be DDT_CLASSES if no entry found */
705b24ab676SJeff Bonwick 	dde->dde_loaded = B_TRUE;
706b24ab676SJeff Bonwick 	dde->dde_loading = B_FALSE;
707b24ab676SJeff Bonwick 
708b24ab676SJeff Bonwick 	if (error == 0)
709b24ab676SJeff Bonwick 		ddt_stat_update(ddt, dde, -1ULL);
710b24ab676SJeff Bonwick 
711b24ab676SJeff Bonwick 	cv_broadcast(&dde->dde_cv);
712b24ab676SJeff Bonwick 
713b24ab676SJeff Bonwick 	return (dde);
714b24ab676SJeff Bonwick }
715b24ab676SJeff Bonwick 
716b24ab676SJeff Bonwick int
717b24ab676SJeff Bonwick ddt_entry_compare(const void *x1, const void *x2)
718b24ab676SJeff Bonwick {
719b24ab676SJeff Bonwick 	const ddt_entry_t *dde1 = x1;
720b24ab676SJeff Bonwick 	const ddt_entry_t *dde2 = x2;
721b24ab676SJeff Bonwick 	const uint64_t *u1 = (const uint64_t *)&dde1->dde_key;
722b24ab676SJeff Bonwick 	const uint64_t *u2 = (const uint64_t *)&dde2->dde_key;
723b24ab676SJeff Bonwick 
724b24ab676SJeff Bonwick 	for (int i = 0; i < DDT_KEY_WORDS; i++) {
725b24ab676SJeff Bonwick 		if (u1[i] < u2[i])
726b24ab676SJeff Bonwick 			return (-1);
727b24ab676SJeff Bonwick 		if (u1[i] > u2[i])
728b24ab676SJeff Bonwick 			return (1);
729b24ab676SJeff Bonwick 	}
730b24ab676SJeff Bonwick 
731b24ab676SJeff Bonwick 	return (0);
732b24ab676SJeff Bonwick }
733b24ab676SJeff Bonwick 
734b24ab676SJeff Bonwick static ddt_t *
735b24ab676SJeff Bonwick ddt_table_alloc(spa_t *spa, enum zio_checksum c)
736b24ab676SJeff Bonwick {
737b24ab676SJeff Bonwick 	ddt_t *ddt;
738b24ab676SJeff Bonwick 
739b24ab676SJeff Bonwick 	ddt = kmem_zalloc(sizeof (*ddt), KM_SLEEP);
740b24ab676SJeff Bonwick 
741b24ab676SJeff Bonwick 	mutex_init(&ddt->ddt_lock, NULL, MUTEX_DEFAULT, NULL);
742b24ab676SJeff Bonwick 	avl_create(&ddt->ddt_tree, ddt_entry_compare,
743b24ab676SJeff Bonwick 	    sizeof (ddt_entry_t), offsetof(ddt_entry_t, dde_node));
744b24ab676SJeff Bonwick 	avl_create(&ddt->ddt_repair_tree, ddt_entry_compare,
745b24ab676SJeff Bonwick 	    sizeof (ddt_entry_t), offsetof(ddt_entry_t, dde_node));
746b24ab676SJeff Bonwick 	ddt->ddt_checksum = c;
747b24ab676SJeff Bonwick 	ddt->ddt_spa = spa;
748b24ab676SJeff Bonwick 	ddt->ddt_os = spa->spa_meta_objset;
749b24ab676SJeff Bonwick 
750b24ab676SJeff Bonwick 	return (ddt);
751b24ab676SJeff Bonwick }
752b24ab676SJeff Bonwick 
753b24ab676SJeff Bonwick static void
754b24ab676SJeff Bonwick ddt_table_free(ddt_t *ddt)
755b24ab676SJeff Bonwick {
756b24ab676SJeff Bonwick 	ASSERT(avl_numnodes(&ddt->ddt_tree) == 0);
757b24ab676SJeff Bonwick 	ASSERT(avl_numnodes(&ddt->ddt_repair_tree) == 0);
758b24ab676SJeff Bonwick 	avl_destroy(&ddt->ddt_tree);
759b24ab676SJeff Bonwick 	avl_destroy(&ddt->ddt_repair_tree);
760b24ab676SJeff Bonwick 	mutex_destroy(&ddt->ddt_lock);
761b24ab676SJeff Bonwick 	kmem_free(ddt, sizeof (*ddt));
762b24ab676SJeff Bonwick }
763b24ab676SJeff Bonwick 
764b24ab676SJeff Bonwick void
765b24ab676SJeff Bonwick ddt_create(spa_t *spa)
766b24ab676SJeff Bonwick {
767b24ab676SJeff Bonwick 	spa->spa_dedup_checksum = ZIO_DEDUPCHECKSUM;
768b24ab676SJeff Bonwick 
769b24ab676SJeff Bonwick 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++)
770b24ab676SJeff Bonwick 		spa->spa_ddt[c] = ddt_table_alloc(spa, c);
771b24ab676SJeff Bonwick }
772b24ab676SJeff Bonwick 
773b24ab676SJeff Bonwick int
774b24ab676SJeff Bonwick ddt_load(spa_t *spa)
775b24ab676SJeff Bonwick {
776b24ab676SJeff Bonwick 	int error;
777b24ab676SJeff Bonwick 
778b24ab676SJeff Bonwick 	ddt_create(spa);
779b24ab676SJeff Bonwick 
780b24ab676SJeff Bonwick 	error = zap_lookup(spa->spa_meta_objset, DMU_POOL_DIRECTORY_OBJECT,
781b24ab676SJeff Bonwick 	    DMU_POOL_DDT_STATS, sizeof (uint64_t), 1,
782b24ab676SJeff Bonwick 	    &spa->spa_ddt_stat_object);
783b24ab676SJeff Bonwick 
784b24ab676SJeff Bonwick 	if (error)
785b24ab676SJeff Bonwick 		return (error == ENOENT ? 0 : error);
786b24ab676SJeff Bonwick 
787b24ab676SJeff Bonwick 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
78830f608a4SGeorge Wilson 		ddt_t *ddt = spa->spa_ddt[c];
789b24ab676SJeff Bonwick 		for (enum ddt_type type = 0; type < DDT_TYPES; type++) {
790b24ab676SJeff Bonwick 			for (enum ddt_class class = 0; class < DDT_CLASSES;
791b24ab676SJeff Bonwick 			    class++) {
792b24ab676SJeff Bonwick 				error = ddt_object_load(ddt, type, class);
793b24ab676SJeff Bonwick 				if (error != 0 && error != ENOENT)
794b24ab676SJeff Bonwick 					return (error);
795b24ab676SJeff Bonwick 			}
796b24ab676SJeff Bonwick 		}
79730f608a4SGeorge Wilson 
79830f608a4SGeorge Wilson 		/*
79930f608a4SGeorge Wilson 		 * Seed the cached histograms.
80030f608a4SGeorge Wilson 		 */
80130f608a4SGeorge Wilson 		bcopy(ddt->ddt_histogram, &ddt->ddt_histogram_cache,
80230f608a4SGeorge Wilson 		    sizeof (ddt->ddt_histogram));
803b24ab676SJeff Bonwick 	}
804b24ab676SJeff Bonwick 
805b24ab676SJeff Bonwick 	return (0);
806b24ab676SJeff Bonwick }
807b24ab676SJeff Bonwick 
808b24ab676SJeff Bonwick void
809b24ab676SJeff Bonwick ddt_unload(spa_t *spa)
810b24ab676SJeff Bonwick {
811b24ab676SJeff Bonwick 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
812b24ab676SJeff Bonwick 		if (spa->spa_ddt[c]) {
813b24ab676SJeff Bonwick 			ddt_table_free(spa->spa_ddt[c]);
814b24ab676SJeff Bonwick 			spa->spa_ddt[c] = NULL;
815b24ab676SJeff Bonwick 		}
816b24ab676SJeff Bonwick 	}
817b24ab676SJeff Bonwick }
818b24ab676SJeff Bonwick 
819bbfd46c4SJeff Bonwick boolean_t
820bbfd46c4SJeff Bonwick ddt_class_contains(spa_t *spa, enum ddt_class max_class, const blkptr_t *bp)
821bbfd46c4SJeff Bonwick {
822bbfd46c4SJeff Bonwick 	ddt_t *ddt;
823bbfd46c4SJeff Bonwick 	ddt_entry_t dde;
824bbfd46c4SJeff Bonwick 
825bbfd46c4SJeff Bonwick 	if (!BP_GET_DEDUP(bp))
826bbfd46c4SJeff Bonwick 		return (B_FALSE);
827bbfd46c4SJeff Bonwick 
828bbfd46c4SJeff Bonwick 	if (max_class == DDT_CLASS_UNIQUE)
829bbfd46c4SJeff Bonwick 		return (B_TRUE);
830bbfd46c4SJeff Bonwick 
831bbfd46c4SJeff Bonwick 	ddt = spa->spa_ddt[BP_GET_CHECKSUM(bp)];
832bbfd46c4SJeff Bonwick 
833bbfd46c4SJeff Bonwick 	ddt_key_fill(&dde.dde_key, bp);
834bbfd46c4SJeff Bonwick 
835bbfd46c4SJeff Bonwick 	for (enum ddt_type type = 0; type < DDT_TYPES; type++)
836bbfd46c4SJeff Bonwick 		for (enum ddt_class class = 0; class <= max_class; class++)
837bbfd46c4SJeff Bonwick 			if (ddt_object_lookup(ddt, type, class, &dde) == 0)
838bbfd46c4SJeff Bonwick 				return (B_TRUE);
839bbfd46c4SJeff Bonwick 
840bbfd46c4SJeff Bonwick 	return (B_FALSE);
841bbfd46c4SJeff Bonwick }
842bbfd46c4SJeff Bonwick 
843b24ab676SJeff Bonwick ddt_entry_t *
844b24ab676SJeff Bonwick ddt_repair_start(ddt_t *ddt, const blkptr_t *bp)
845b24ab676SJeff Bonwick {
846b24ab676SJeff Bonwick 	ddt_key_t ddk;
847b24ab676SJeff Bonwick 	ddt_entry_t *dde;
848b24ab676SJeff Bonwick 
849b24ab676SJeff Bonwick 	ddt_key_fill(&ddk, bp);
850b24ab676SJeff Bonwick 
851b24ab676SJeff Bonwick 	dde = ddt_alloc(&ddk);
852b24ab676SJeff Bonwick 
853b24ab676SJeff Bonwick 	for (enum ddt_type type = 0; type < DDT_TYPES; type++) {
854b24ab676SJeff Bonwick 		for (enum ddt_class class = 0; class < DDT_CLASSES; class++) {
855b24ab676SJeff Bonwick 			/*
856b24ab676SJeff Bonwick 			 * We can only do repair if there are multiple copies
857b24ab676SJeff Bonwick 			 * of the block.  For anything in the UNIQUE class,
858b24ab676SJeff Bonwick 			 * there's definitely only one copy, so don't even try.
859b24ab676SJeff Bonwick 			 */
860b24ab676SJeff Bonwick 			if (class != DDT_CLASS_UNIQUE &&
861b24ab676SJeff Bonwick 			    ddt_object_lookup(ddt, type, class, dde) == 0)
862b24ab676SJeff Bonwick 				return (dde);
863b24ab676SJeff Bonwick 		}
864b24ab676SJeff Bonwick 	}
865b24ab676SJeff Bonwick 
866b24ab676SJeff Bonwick 	bzero(dde->dde_phys, sizeof (dde->dde_phys));
867b24ab676SJeff Bonwick 
868b24ab676SJeff Bonwick 	return (dde);
869b24ab676SJeff Bonwick }
870b24ab676SJeff Bonwick 
871b24ab676SJeff Bonwick void
872b24ab676SJeff Bonwick ddt_repair_done(ddt_t *ddt, ddt_entry_t *dde)
873b24ab676SJeff Bonwick {
874b24ab676SJeff Bonwick 	avl_index_t where;
875b24ab676SJeff Bonwick 
876b24ab676SJeff Bonwick 	ddt_enter(ddt);
877b24ab676SJeff Bonwick 
878b24ab676SJeff Bonwick 	if (dde->dde_repair_data != NULL && spa_writeable(ddt->ddt_spa) &&
879b24ab676SJeff Bonwick 	    avl_find(&ddt->ddt_repair_tree, dde, &where) == NULL)
880b24ab676SJeff Bonwick 		avl_insert(&ddt->ddt_repair_tree, dde, where);
881b24ab676SJeff Bonwick 	else
882b24ab676SJeff Bonwick 		ddt_free(dde);
883b24ab676SJeff Bonwick 
884b24ab676SJeff Bonwick 	ddt_exit(ddt);
885b24ab676SJeff Bonwick }
886b24ab676SJeff Bonwick 
887b24ab676SJeff Bonwick static void
888b24ab676SJeff Bonwick ddt_repair_entry_done(zio_t *zio)
889b24ab676SJeff Bonwick {
890b24ab676SJeff Bonwick 	ddt_entry_t *rdde = zio->io_private;
891b24ab676SJeff Bonwick 
892b24ab676SJeff Bonwick 	ddt_free(rdde);
893b24ab676SJeff Bonwick }
894b24ab676SJeff Bonwick 
895b24ab676SJeff Bonwick static void
896b24ab676SJeff Bonwick ddt_repair_entry(ddt_t *ddt, ddt_entry_t *dde, ddt_entry_t *rdde, zio_t *rio)
897b24ab676SJeff Bonwick {
898b24ab676SJeff Bonwick 	ddt_phys_t *ddp = dde->dde_phys;
899b24ab676SJeff Bonwick 	ddt_phys_t *rddp = rdde->dde_phys;
900b24ab676SJeff Bonwick 	ddt_key_t *ddk = &dde->dde_key;
901b24ab676SJeff Bonwick 	ddt_key_t *rddk = &rdde->dde_key;
902b24ab676SJeff Bonwick 	zio_t *zio;
903b24ab676SJeff Bonwick 	blkptr_t blk;
904b24ab676SJeff Bonwick 
905b24ab676SJeff Bonwick 	zio = zio_null(rio, rio->io_spa, NULL,
906b24ab676SJeff Bonwick 	    ddt_repair_entry_done, rdde, rio->io_flags);
907b24ab676SJeff Bonwick 
908b24ab676SJeff Bonwick 	for (int p = 0; p < DDT_PHYS_TYPES; p++, ddp++, rddp++) {
909b24ab676SJeff Bonwick 		if (ddp->ddp_phys_birth == 0 ||
910b24ab676SJeff Bonwick 		    ddp->ddp_phys_birth != rddp->ddp_phys_birth ||
911b24ab676SJeff Bonwick 		    bcmp(ddp->ddp_dva, rddp->ddp_dva, sizeof (ddp->ddp_dva)))
912b24ab676SJeff Bonwick 			continue;
913bbfd46c4SJeff Bonwick 		ddt_bp_create(ddt->ddt_checksum, ddk, ddp, &blk);
914b24ab676SJeff Bonwick 		zio_nowait(zio_rewrite(zio, zio->io_spa, 0, &blk,
915b24ab676SJeff Bonwick 		    rdde->dde_repair_data, DDK_GET_PSIZE(rddk), NULL, NULL,
916b24ab676SJeff Bonwick 		    ZIO_PRIORITY_SYNC_WRITE, ZIO_DDT_CHILD_FLAGS(zio), NULL));
917b24ab676SJeff Bonwick 	}
918b24ab676SJeff Bonwick 
919b24ab676SJeff Bonwick 	zio_nowait(zio);
920b24ab676SJeff Bonwick }
921b24ab676SJeff Bonwick 
922b24ab676SJeff Bonwick static void
923b24ab676SJeff Bonwick ddt_repair_table(ddt_t *ddt, zio_t *rio)
924b24ab676SJeff Bonwick {
925b24ab676SJeff Bonwick 	spa_t *spa = ddt->ddt_spa;
926b24ab676SJeff Bonwick 	ddt_entry_t *dde, *rdde_next, *rdde;
927b24ab676SJeff Bonwick 	avl_tree_t *t = &ddt->ddt_repair_tree;
928b24ab676SJeff Bonwick 	blkptr_t blk;
929b24ab676SJeff Bonwick 
930b24ab676SJeff Bonwick 	if (spa_sync_pass(spa) > 1)
931b24ab676SJeff Bonwick 		return;
932b24ab676SJeff Bonwick 
933b24ab676SJeff Bonwick 	ddt_enter(ddt);
934b24ab676SJeff Bonwick 	for (rdde = avl_first(t); rdde != NULL; rdde = rdde_next) {
935b24ab676SJeff Bonwick 		rdde_next = AVL_NEXT(t, rdde);
936b24ab676SJeff Bonwick 		avl_remove(&ddt->ddt_repair_tree, rdde);
937b24ab676SJeff Bonwick 		ddt_exit(ddt);
938bbfd46c4SJeff Bonwick 		ddt_bp_create(ddt->ddt_checksum, &rdde->dde_key, NULL, &blk);
939b24ab676SJeff Bonwick 		dde = ddt_repair_start(ddt, &blk);
940b24ab676SJeff Bonwick 		ddt_repair_entry(ddt, dde, rdde, rio);
941b24ab676SJeff Bonwick 		ddt_repair_done(ddt, dde);
942b24ab676SJeff Bonwick 		ddt_enter(ddt);
943b24ab676SJeff Bonwick 	}
944b24ab676SJeff Bonwick 	ddt_exit(ddt);
945b24ab676SJeff Bonwick }
946b24ab676SJeff Bonwick 
947b24ab676SJeff Bonwick static void
948b24ab676SJeff Bonwick ddt_sync_entry(ddt_t *ddt, ddt_entry_t *dde, dmu_tx_t *tx, uint64_t txg)
949b24ab676SJeff Bonwick {
950bbfd46c4SJeff Bonwick 	dsl_pool_t *dp = ddt->ddt_spa->spa_dsl_pool;
951b24ab676SJeff Bonwick 	ddt_phys_t *ddp = dde->dde_phys;
952b24ab676SJeff Bonwick 	ddt_key_t *ddk = &dde->dde_key;
953b24ab676SJeff Bonwick 	enum ddt_type otype = dde->dde_type;
954b24ab676SJeff Bonwick 	enum ddt_type ntype = DDT_TYPE_CURRENT;
955b24ab676SJeff Bonwick 	enum ddt_class oclass = dde->dde_class;
956b24ab676SJeff Bonwick 	enum ddt_class nclass;
957b24ab676SJeff Bonwick 	uint64_t total_refcnt = 0;
958b24ab676SJeff Bonwick 
959b24ab676SJeff Bonwick 	ASSERT(dde->dde_loaded);
960b24ab676SJeff Bonwick 	ASSERT(!dde->dde_loading);
961b24ab676SJeff Bonwick 
962b24ab676SJeff Bonwick 	for (int p = 0; p < DDT_PHYS_TYPES; p++, ddp++) {
963b24ab676SJeff Bonwick 		ASSERT(dde->dde_lead_zio[p] == NULL);
964b24ab676SJeff Bonwick 		ASSERT((int64_t)ddp->ddp_refcnt >= 0);
965b24ab676SJeff Bonwick 		if (ddp->ddp_phys_birth == 0) {
966b24ab676SJeff Bonwick 			ASSERT(ddp->ddp_refcnt == 0);
967b24ab676SJeff Bonwick 			continue;
968b24ab676SJeff Bonwick 		}
969b24ab676SJeff Bonwick 		if (p == DDT_PHYS_DITTO) {
970b24ab676SJeff Bonwick 			if (ddt_ditto_copies_needed(ddt, dde, NULL) == 0)
971b24ab676SJeff Bonwick 				ddt_phys_free(ddt, ddk, ddp, txg);
972b24ab676SJeff Bonwick 			continue;
973b24ab676SJeff Bonwick 		}
974b24ab676SJeff Bonwick 		if (ddp->ddp_refcnt == 0)
975b24ab676SJeff Bonwick 			ddt_phys_free(ddt, ddk, ddp, txg);
976b24ab676SJeff Bonwick 		total_refcnt += ddp->ddp_refcnt;
977b24ab676SJeff Bonwick 	}
978b24ab676SJeff Bonwick 
979b24ab676SJeff Bonwick 	if (dde->dde_phys[DDT_PHYS_DITTO].ddp_phys_birth != 0)
980b24ab676SJeff Bonwick 		nclass = DDT_CLASS_DITTO;
981b24ab676SJeff Bonwick 	else if (total_refcnt > 1)
982b24ab676SJeff Bonwick 		nclass = DDT_CLASS_DUPLICATE;
983b24ab676SJeff Bonwick 	else
984b24ab676SJeff Bonwick 		nclass = DDT_CLASS_UNIQUE;
985b24ab676SJeff Bonwick 
986b24ab676SJeff Bonwick 	if (otype != DDT_TYPES &&
987b24ab676SJeff Bonwick 	    (otype != ntype || oclass != nclass || total_refcnt == 0)) {
988b24ab676SJeff Bonwick 		VERIFY(ddt_object_remove(ddt, otype, oclass, dde, tx) == 0);
989b24ab676SJeff Bonwick 		ASSERT(ddt_object_lookup(ddt, otype, oclass, dde) == ENOENT);
990b24ab676SJeff Bonwick 	}
991b24ab676SJeff Bonwick 
992b24ab676SJeff Bonwick 	if (total_refcnt != 0) {
993b24ab676SJeff Bonwick 		dde->dde_type = ntype;
994b24ab676SJeff Bonwick 		dde->dde_class = nclass;
995b24ab676SJeff Bonwick 		ddt_stat_update(ddt, dde, 0);
996b24ab676SJeff Bonwick 		if (!ddt_object_exists(ddt, ntype, nclass))
997b24ab676SJeff Bonwick 			ddt_object_create(ddt, ntype, nclass, tx);
998b24ab676SJeff Bonwick 		VERIFY(ddt_object_update(ddt, ntype, nclass, dde, tx) == 0);
999bbfd46c4SJeff Bonwick 
1000*3f9d6ad7SLin Ling 		/*
1001*3f9d6ad7SLin Ling 		 * If the class changes, the order that we scan this bp
1002*3f9d6ad7SLin Ling 		 * changes.  If it decreases, we could miss it, so
1003*3f9d6ad7SLin Ling 		 * scan it right now.  (This covers both class changing
1004*3f9d6ad7SLin Ling 		 * while we are doing ddt_walk(), and when we are
1005*3f9d6ad7SLin Ling 		 * traversing.)
1006*3f9d6ad7SLin Ling 		 */
1007*3f9d6ad7SLin Ling 		if (nclass < oclass) {
1008*3f9d6ad7SLin Ling 			dsl_scan_ddt_entry(dp->dp_scan,
1009*3f9d6ad7SLin Ling 			    ddt->ddt_checksum, dde, tx);
1010*3f9d6ad7SLin Ling 		}
1011b24ab676SJeff Bonwick 	}
1012b24ab676SJeff Bonwick }
1013b24ab676SJeff Bonwick 
1014b24ab676SJeff Bonwick static void
1015b24ab676SJeff Bonwick ddt_sync_table(ddt_t *ddt, dmu_tx_t *tx, uint64_t txg)
1016b24ab676SJeff Bonwick {
1017b24ab676SJeff Bonwick 	spa_t *spa = ddt->ddt_spa;
1018b24ab676SJeff Bonwick 	ddt_entry_t *dde;
1019b24ab676SJeff Bonwick 	void *cookie = NULL;
1020b24ab676SJeff Bonwick 
1021b24ab676SJeff Bonwick 	if (avl_numnodes(&ddt->ddt_tree) == 0)
1022b24ab676SJeff Bonwick 		return;
1023b24ab676SJeff Bonwick 
1024b24ab676SJeff Bonwick 	ASSERT(spa->spa_uberblock.ub_version >= SPA_VERSION_DEDUP);
1025b24ab676SJeff Bonwick 
1026b24ab676SJeff Bonwick 	if (spa->spa_ddt_stat_object == 0) {
1027b24ab676SJeff Bonwick 		spa->spa_ddt_stat_object = zap_create(ddt->ddt_os,
1028b24ab676SJeff Bonwick 		    DMU_OT_DDT_STATS, DMU_OT_NONE, 0, tx);
1029b24ab676SJeff Bonwick 		VERIFY(zap_add(ddt->ddt_os, DMU_POOL_DIRECTORY_OBJECT,
1030b24ab676SJeff Bonwick 		    DMU_POOL_DDT_STATS, sizeof (uint64_t), 1,
1031b24ab676SJeff Bonwick 		    &spa->spa_ddt_stat_object, tx) == 0);
1032b24ab676SJeff Bonwick 	}
1033b24ab676SJeff Bonwick 
1034b24ab676SJeff Bonwick 	while ((dde = avl_destroy_nodes(&ddt->ddt_tree, &cookie)) != NULL) {
1035b24ab676SJeff Bonwick 		ddt_sync_entry(ddt, dde, tx, txg);
1036b24ab676SJeff Bonwick 		ddt_free(dde);
1037b24ab676SJeff Bonwick 	}
1038b24ab676SJeff Bonwick 
1039b24ab676SJeff Bonwick 	for (enum ddt_type type = 0; type < DDT_TYPES; type++) {
1040b24ab676SJeff Bonwick 		for (enum ddt_class class = 0; class < DDT_CLASSES; class++) {
1041b24ab676SJeff Bonwick 			if (!ddt_object_exists(ddt, type, class))
1042b24ab676SJeff Bonwick 				continue;
1043b24ab676SJeff Bonwick 			ddt_object_sync(ddt, type, class, tx);
1044b24ab676SJeff Bonwick 			if (ddt_object_count(ddt, type, class) == 0)
1045b24ab676SJeff Bonwick 				ddt_object_destroy(ddt, type, class, tx);
1046b24ab676SJeff Bonwick 		}
1047b24ab676SJeff Bonwick 	}
104830f608a4SGeorge Wilson 
104930f608a4SGeorge Wilson 	bcopy(ddt->ddt_histogram, &ddt->ddt_histogram_cache,
105030f608a4SGeorge Wilson 	    sizeof (ddt->ddt_histogram));
1051b24ab676SJeff Bonwick }
1052b24ab676SJeff Bonwick 
1053b24ab676SJeff Bonwick void
1054b24ab676SJeff Bonwick ddt_sync(spa_t *spa, uint64_t txg)
1055b24ab676SJeff Bonwick {
1056b24ab676SJeff Bonwick 	dmu_tx_t *tx;
1057b24ab676SJeff Bonwick 	zio_t *rio = zio_root(spa, NULL, NULL,
1058b24ab676SJeff Bonwick 	    ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE);
1059b24ab676SJeff Bonwick 
1060b24ab676SJeff Bonwick 	ASSERT(spa_syncing_txg(spa) == txg);
1061b24ab676SJeff Bonwick 
1062b24ab676SJeff Bonwick 	tx = dmu_tx_create_assigned(spa->spa_dsl_pool, txg);
1063b24ab676SJeff Bonwick 
1064b24ab676SJeff Bonwick 	for (enum zio_checksum c = 0; c < ZIO_CHECKSUM_FUNCTIONS; c++) {
1065b24ab676SJeff Bonwick 		ddt_t *ddt = spa->spa_ddt[c];
1066b24ab676SJeff Bonwick 		if (ddt == NULL)
1067b24ab676SJeff Bonwick 			continue;
1068b24ab676SJeff Bonwick 		ddt_sync_table(ddt, tx, txg);
1069b24ab676SJeff Bonwick 		ddt_repair_table(ddt, rio);
1070b24ab676SJeff Bonwick 	}
1071b24ab676SJeff Bonwick 
1072b24ab676SJeff Bonwick 	(void) zio_wait(rio);
1073b24ab676SJeff Bonwick 
1074b24ab676SJeff Bonwick 	dmu_tx_commit(tx);
1075b24ab676SJeff Bonwick }
1076bbfd46c4SJeff Bonwick 
1077bbfd46c4SJeff Bonwick int
1078bbfd46c4SJeff Bonwick ddt_walk(spa_t *spa, ddt_bookmark_t *ddb, ddt_entry_t *dde)
1079bbfd46c4SJeff Bonwick {
1080bbfd46c4SJeff Bonwick 	do {
1081bbfd46c4SJeff Bonwick 		do {
1082bbfd46c4SJeff Bonwick 			do {
1083bbfd46c4SJeff Bonwick 				ddt_t *ddt = spa->spa_ddt[ddb->ddb_checksum];
1084bbfd46c4SJeff Bonwick 				int error = ENOENT;
1085bbfd46c4SJeff Bonwick 				if (ddt_object_exists(ddt, ddb->ddb_type,
1086bbfd46c4SJeff Bonwick 				    ddb->ddb_class)) {
1087bbfd46c4SJeff Bonwick 					error = ddt_object_walk(ddt,
1088bbfd46c4SJeff Bonwick 					    ddb->ddb_type, ddb->ddb_class,
1089bbfd46c4SJeff Bonwick 					    &ddb->ddb_cursor, dde);
1090bbfd46c4SJeff Bonwick 				}
1091*3f9d6ad7SLin Ling 				dde->dde_type = ddb->ddb_type;
1092*3f9d6ad7SLin Ling 				dde->dde_class = ddb->ddb_class;
1093bbfd46c4SJeff Bonwick 				if (error == 0)
1094bbfd46c4SJeff Bonwick 					return (0);
1095bbfd46c4SJeff Bonwick 				if (error != ENOENT)
1096bbfd46c4SJeff Bonwick 					return (error);
1097bbfd46c4SJeff Bonwick 				ddb->ddb_cursor = 0;
1098bbfd46c4SJeff Bonwick 			} while (++ddb->ddb_checksum < ZIO_CHECKSUM_FUNCTIONS);
1099bbfd46c4SJeff Bonwick 			ddb->ddb_checksum = 0;
1100bbfd46c4SJeff Bonwick 		} while (++ddb->ddb_type < DDT_TYPES);
1101bbfd46c4SJeff Bonwick 		ddb->ddb_type = 0;
1102bbfd46c4SJeff Bonwick 	} while (++ddb->ddb_class < DDT_CLASSES);
1103bbfd46c4SJeff Bonwick 
1104bbfd46c4SJeff Bonwick 	return (ENOENT);
1105bbfd46c4SJeff Bonwick }
1106