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