]> git.proxmox.com Git - mirror_zfs.git/blame - module/zfs/dmu_traverse.c
Illumos 4897 - Space accounting mismatch in L2ARC/zpool
[mirror_zfs.git] / module / zfs / dmu_traverse.c
CommitLineData
34dc7c2f
BB
1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21/*
428870ff 22 * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
2e528b49 23 * Copyright (c) 2013 by Delphix. All rights reserved.
34dc7c2f
BB
24 */
25
34dc7c2f
BB
26#include <sys/zfs_context.h>
27#include <sys/dmu_objset.h>
28#include <sys/dmu_traverse.h>
29#include <sys/dsl_dataset.h>
30#include <sys/dsl_dir.h>
31#include <sys/dsl_pool.h>
32#include <sys/dnode.h>
33#include <sys/spa.h>
34#include <sys/zio.h>
35#include <sys/dmu_impl.h>
428870ff
BB
36#include <sys/sa.h>
37#include <sys/sa_impl.h>
b128c09f 38#include <sys/callb.h>
b0bc7a84 39#include <sys/zfeature.h>
b128c09f 40
572e2857
BB
41int zfs_pd_blks_max = 100;
42
43typedef struct prefetch_data {
b128c09f
BB
44 kmutex_t pd_mtx;
45 kcondvar_t pd_cv;
46 int pd_blks_max;
47 int pd_blks_fetched;
48 int pd_flags;
49 boolean_t pd_cancel;
50 boolean_t pd_exited;
572e2857 51} prefetch_data_t;
b128c09f 52
572e2857 53typedef struct traverse_data {
b128c09f
BB
54 spa_t *td_spa;
55 uint64_t td_objset;
56 blkptr_t *td_rootbp;
57 uint64_t td_min_txg;
9ae529ec 58 zbookmark_t *td_resume;
b128c09f 59 int td_flags;
572e2857 60 prefetch_data_t *td_pfd;
fbeddd60 61 boolean_t td_paused;
b128c09f
BB
62 blkptr_cb_t *td_func;
63 void *td_arg;
572e2857 64} traverse_data_t;
34dc7c2f 65
572e2857 66static int traverse_dnode(traverse_data_t *td, const dnode_phys_t *dnp,
294f6806 67 uint64_t objset, uint64_t object);
96b89346 68static void prefetch_dnode_metadata(traverse_data_t *td, const dnode_phys_t *,
294f6806 69 uint64_t objset, uint64_t object);
9babb374 70
428870ff 71static int
34dc7c2f
BB
72traverse_zil_block(zilog_t *zilog, blkptr_t *bp, void *arg, uint64_t claim_txg)
73{
572e2857 74 traverse_data_t *td = arg;
b128c09f 75 zbookmark_t zb;
34dc7c2f 76
b0bc7a84 77 if (BP_IS_HOLE(bp))
428870ff 78 return (0);
34dc7c2f 79
b128c09f 80 if (claim_txg == 0 && bp->blk_birth >= spa_first_txg(td->td_spa))
428870ff
BB
81 return (0);
82
83 SET_BOOKMARK(&zb, td->td_objset, ZB_ZIL_OBJECT, ZB_ZIL_LEVEL,
84 bp->blk_cksum.zc_word[ZIL_ZC_SEQ]);
85
294f6806 86 (void) td->td_func(td->td_spa, zilog, bp, &zb, NULL, td->td_arg);
b128c09f 87
428870ff 88 return (0);
34dc7c2f
BB
89}
90
428870ff 91static int
34dc7c2f
BB
92traverse_zil_record(zilog_t *zilog, lr_t *lrc, void *arg, uint64_t claim_txg)
93{
572e2857 94 traverse_data_t *td = arg;
34dc7c2f
BB
95
96 if (lrc->lrc_txtype == TX_WRITE) {
97 lr_write_t *lr = (lr_write_t *)lrc;
98 blkptr_t *bp = &lr->lr_blkptr;
b128c09f 99 zbookmark_t zb;
34dc7c2f 100
b0bc7a84 101 if (BP_IS_HOLE(bp))
428870ff 102 return (0);
34dc7c2f 103
b128c09f 104 if (claim_txg == 0 || bp->blk_birth < claim_txg)
428870ff 105 return (0);
b128c09f 106
572e2857
BB
107 SET_BOOKMARK(&zb, td->td_objset, lr->lr_foid,
108 ZB_ZIL_LEVEL, lr->lr_offset / BP_GET_LSIZE(bp));
428870ff 109
294f6806 110 (void) td->td_func(td->td_spa, zilog, bp, &zb, NULL,
428870ff 111 td->td_arg);
34dc7c2f 112 }
428870ff 113 return (0);
34dc7c2f
BB
114}
115
116static void
572e2857 117traverse_zil(traverse_data_t *td, zil_header_t *zh)
34dc7c2f 118{
34dc7c2f
BB
119 uint64_t claim_txg = zh->zh_claim_txg;
120 zilog_t *zilog;
121
34dc7c2f
BB
122 /*
123 * We only want to visit blocks that have been claimed but not yet
428870ff 124 * replayed; plus, in read-only mode, blocks that are already stable.
34dc7c2f 125 */
fb5f0bc8 126 if (claim_txg == 0 && spa_writeable(td->td_spa))
34dc7c2f
BB
127 return;
128
b128c09f 129 zilog = zil_alloc(spa_get_dsl(td->td_spa)->dp_meta_objset, zh);
34dc7c2f 130
b128c09f 131 (void) zil_parse(zilog, traverse_zil_block, traverse_zil_record, td,
34dc7c2f
BB
132 claim_txg);
133
134 zil_free(zilog);
135}
136
9ae529ec
CS
137typedef enum resume_skip {
138 RESUME_SKIP_ALL,
139 RESUME_SKIP_NONE,
140 RESUME_SKIP_CHILDREN
141} resume_skip_t;
142
143/*
144 * Returns RESUME_SKIP_ALL if td indicates that we are resuming a traversal and
145 * the block indicated by zb does not need to be visited at all. Returns
146 * RESUME_SKIP_CHILDREN if we are resuming a post traversal and we reach the
147 * resume point. This indicates that this block should be visited but not its
148 * children (since they must have been visited in a previous traversal).
149 * Otherwise returns RESUME_SKIP_NONE.
150 */
151static resume_skip_t
152resume_skip_check(traverse_data_t *td, const dnode_phys_t *dnp,
153 const zbookmark_t *zb)
154{
155 if (td->td_resume != NULL && !ZB_IS_ZERO(td->td_resume)) {
156 /*
157 * If we already visited this bp & everything below,
158 * don't bother doing it again.
159 */
160 if (zbookmark_is_before(dnp, zb, td->td_resume))
161 return (RESUME_SKIP_ALL);
162
163 /*
164 * If we found the block we're trying to resume from, zero
165 * the bookmark out to indicate that we have resumed.
166 */
9ae529ec
CS
167 if (bcmp(zb, td->td_resume, sizeof (*zb)) == 0) {
168 bzero(td->td_resume, sizeof (*zb));
169 if (td->td_flags & TRAVERSE_POST)
170 return (RESUME_SKIP_CHILDREN);
171 }
172 }
173 return (RESUME_SKIP_NONE);
174}
175
96b89346
MA
176static void
177traverse_prefetch_metadata(traverse_data_t *td,
294f6806 178 const blkptr_t *bp, const zbookmark_t *zb)
96b89346
MA
179{
180 uint32_t flags = ARC_NOWAIT | ARC_PREFETCH;
181
182 if (!(td->td_flags & TRAVERSE_PREFETCH_METADATA))
183 return;
184 /*
185 * If we are in the process of resuming, don't prefetch, because
186 * some children will not be needed (and in fact may have already
187 * been freed).
188 */
189 if (td->td_resume != NULL && !ZB_IS_ZERO(td->td_resume))
190 return;
191 if (BP_IS_HOLE(bp) || bp->blk_birth <= td->td_min_txg)
192 return;
193 if (BP_GET_LEVEL(bp) == 0 && BP_GET_TYPE(bp) != DMU_OT_DNODE)
194 return;
195
294f6806
GW
196 (void) arc_read(NULL, td->td_spa, bp, NULL, NULL,
197 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
96b89346
MA
198}
199
c7f8f831
BB
200static int
201traverse_visitbp(traverse_data_t *td, const dnode_phys_t *dnp,
294f6806 202 const blkptr_t *bp, const zbookmark_t *zb)
6656bf56 203{
fbeddd60 204 int err = 0;
c7f8f831 205 arc_buf_t *buf = NULL;
9ae529ec
CS
206
207 switch (resume_skip_check(td, dnp, zb)) {
208 case RESUME_SKIP_ALL:
209 return (0);
210 case RESUME_SKIP_CHILDREN:
211 goto post;
212 case RESUME_SKIP_NONE:
213 break;
214 default:
215 ASSERT(0);
216 }
6656bf56 217
b0bc7a84
MG
218 if (bp->blk_birth == 0) {
219 if (spa_feature_is_active(td->td_spa, SPA_FEATURE_HOLE_BIRTH)) {
220 /*
221 * Since this block has a birth time of 0 it must be a
222 * hole created before the SPA_FEATURE_HOLE_BIRTH
223 * feature was enabled. If SPA_FEATURE_HOLE_BIRTH
224 * was enabled before the min_txg for this traveral we
225 * know the hole must have been created before the
226 * min_txg for this traveral, so we can skip it. If
227 * SPA_FEATURE_HOLE_BIRTH was enabled after the min_txg
228 * for this traveral we cannot tell if the hole was
229 * created before or after the min_txg for this
230 * traversal, so we cannot skip it.
231 */
232 uint64_t hole_birth_enabled_txg;
233 VERIFY(spa_feature_enabled_txg(td->td_spa,
234 SPA_FEATURE_HOLE_BIRTH, &hole_birth_enabled_txg));
235 if (hole_birth_enabled_txg < td->td_min_txg)
236 return (0);
237 }
238 } else if (bp->blk_birth <= td->td_min_txg) {
239 return (0);
240 }
241
9ae529ec 242 if (BP_IS_HOLE(bp)) {
b0bc7a84 243 err = td->td_func(td->td_spa, NULL, bp, zb, dnp, td->td_arg);
fbeddd60
MA
244 if (err != 0)
245 goto post;
246 return (0);
34dc7c2f
BB
247 }
248
a1687880
BB
249 if (td->td_pfd && !td->td_pfd->pd_exited &&
250 ((td->td_pfd->pd_flags & TRAVERSE_PREFETCH_DATA) ||
c7f8f831 251 BP_GET_TYPE(bp) == DMU_OT_DNODE || BP_GET_LEVEL(bp) > 0)) {
a1687880
BB
252 mutex_enter(&td->td_pfd->pd_mtx);
253 ASSERT(td->td_pfd->pd_blks_fetched >= 0);
254 while (td->td_pfd->pd_blks_fetched == 0 &&
255 !td->td_pfd->pd_exited)
256 cv_wait(&td->td_pfd->pd_cv, &td->td_pfd->pd_mtx);
257 td->td_pfd->pd_blks_fetched--;
258 cv_broadcast(&td->td_pfd->pd_cv);
259 mutex_exit(&td->td_pfd->pd_mtx);
34dc7c2f
BB
260 }
261
c7f8f831 262 if (td->td_flags & TRAVERSE_PRE) {
294f6806 263 err = td->td_func(td->td_spa, NULL, bp, zb, dnp,
c7f8f831
BB
264 td->td_arg);
265 if (err == TRAVERSE_VISIT_NO_CHILDREN)
572e2857 266 return (0);
9ae529ec
CS
267 if (err != 0)
268 goto post;
34dc7c2f
BB
269 }
270
c7f8f831
BB
271 if (BP_GET_LEVEL(bp) > 0) {
272 uint32_t flags = ARC_WAIT;
a1687880
BB
273 int32_t i;
274 int32_t epb = BP_GET_LSIZE(bp) >> SPA_BLKPTRSHIFT;
275 zbookmark_t *czb;
b128c09f 276
294f6806 277 err = arc_read(NULL, td->td_spa, bp, arc_getbuf_func, &buf,
c7f8f831 278 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
13fe0198 279 if (err != 0)
fbeddd60 280 goto post;
a1687880
BB
281
282 czb = kmem_alloc(sizeof (zbookmark_t), KM_PUSHPAGE);
96b89346
MA
283
284 for (i = 0; i < epb; i++) {
a1687880 285 SET_BOOKMARK(czb, zb->zb_objset, zb->zb_object,
96b89346
MA
286 zb->zb_level - 1,
287 zb->zb_blkid * epb + i);
a1687880
BB
288 traverse_prefetch_metadata(td,
289 &((blkptr_t *)buf->b_data)[i], czb);
96b89346 290 }
b128c09f
BB
291
292 /* recursively visitbp() blocks below this */
96b89346 293 for (i = 0; i < epb; i++) {
a1687880 294 SET_BOOKMARK(czb, zb->zb_objset, zb->zb_object,
c7f8f831
BB
295 zb->zb_level - 1,
296 zb->zb_blkid * epb + i);
a1687880
BB
297 err = traverse_visitbp(td, dnp,
298 &((blkptr_t *)buf->b_data)[i], czb);
fbeddd60
MA
299 if (err != 0)
300 break;
b128c09f 301 }
a1687880
BB
302
303 kmem_free(czb, sizeof (zbookmark_t));
304
c7f8f831
BB
305 } else if (BP_GET_TYPE(bp) == DMU_OT_DNODE) {
306 uint32_t flags = ARC_WAIT;
a1687880
BB
307 int32_t i;
308 int32_t epb = BP_GET_LSIZE(bp) >> DNODE_SHIFT;
b128c09f 309
294f6806 310 err = arc_read(NULL, td->td_spa, bp, arc_getbuf_func, &buf,
c7f8f831 311 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
13fe0198 312 if (err != 0)
fbeddd60 313 goto post;
96b89346
MA
314 dnp = buf->b_data;
315
316 for (i = 0; i < epb; i++) {
294f6806 317 prefetch_dnode_metadata(td, &dnp[i], zb->zb_objset,
96b89346
MA
318 zb->zb_blkid * epb + i);
319 }
b128c09f
BB
320
321 /* recursively visitbp() blocks below this */
96b89346 322 for (i = 0; i < epb; i++) {
294f6806 323 err = traverse_dnode(td, &dnp[i], zb->zb_objset,
c7f8f831 324 zb->zb_blkid * epb + i);
fbeddd60
MA
325 if (err != 0)
326 break;
34dc7c2f 327 }
c7f8f831
BB
328 } else if (BP_GET_TYPE(bp) == DMU_OT_OBJSET) {
329 uint32_t flags = ARC_WAIT;
330 objset_phys_t *osp;
331 dnode_phys_t *dnp;
332
294f6806 333 err = arc_read(NULL, td->td_spa, bp, arc_getbuf_func, &buf,
c7f8f831 334 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
13fe0198 335 if (err != 0)
fbeddd60 336 goto post;
c7f8f831
BB
337
338 osp = buf->b_data;
339 dnp = &osp->os_meta_dnode;
294f6806 340 prefetch_dnode_metadata(td, dnp, zb->zb_objset,
96b89346
MA
341 DMU_META_DNODE_OBJECT);
342 if (arc_buf_size(buf) >= sizeof (objset_phys_t)) {
96b89346 343 prefetch_dnode_metadata(td, &osp->os_groupused_dnode,
8b464649
MA
344 zb->zb_objset, DMU_GROUPUSED_OBJECT);
345 prefetch_dnode_metadata(td, &osp->os_userused_dnode,
294f6806 346 zb->zb_objset, DMU_USERUSED_OBJECT);
96b89346
MA
347 }
348
294f6806 349 err = traverse_dnode(td, dnp, zb->zb_objset,
c7f8f831 350 DMU_META_DNODE_OBJECT);
c7f8f831 351 if (err == 0 && arc_buf_size(buf) >= sizeof (objset_phys_t)) {
8b464649 352 dnp = &osp->os_groupused_dnode;
294f6806 353 err = traverse_dnode(td, dnp, zb->zb_objset,
8b464649 354 DMU_GROUPUSED_OBJECT);
9babb374 355 }
c7f8f831 356 if (err == 0 && arc_buf_size(buf) >= sizeof (objset_phys_t)) {
8b464649 357 dnp = &osp->os_userused_dnode;
294f6806 358 err = traverse_dnode(td, dnp, zb->zb_objset,
8b464649 359 DMU_USERUSED_OBJECT);
34dc7c2f 360 }
34dc7c2f
BB
361 }
362
c7f8f831
BB
363 if (buf)
364 (void) arc_buf_remove_ref(buf, &buf);
34dc7c2f 365
9ae529ec 366post:
fbeddd60 367 if (err == 0 && (td->td_flags & TRAVERSE_POST))
294f6806 368 err = td->td_func(td->td_spa, NULL, bp, zb, dnp, td->td_arg);
fbeddd60
MA
369
370 if ((td->td_flags & TRAVERSE_HARD) && (err == EIO || err == ECKSUM)) {
371 /*
372 * Ignore this disk error as requested by the HARD flag,
373 * and continue traversal.
374 */
375 err = 0;
9ae529ec
CS
376 }
377
fbeddd60
MA
378 /*
379 * If we are stopping here, set td_resume.
380 */
381 if (td->td_resume != NULL && err != 0 && !td->td_paused) {
382 td->td_resume->zb_objset = zb->zb_objset;
383 td->td_resume->zb_object = zb->zb_object;
384 td->td_resume->zb_level = 0;
385 /*
386 * If we have stopped on an indirect block (e.g. due to
387 * i/o error), we have not visited anything below it.
388 * Set the bookmark to the first level-0 block that we need
389 * to visit. This way, the resuming code does not need to
390 * deal with resuming from indirect blocks.
391 */
392 td->td_resume->zb_blkid = zb->zb_blkid <<
393 (zb->zb_level * (dnp->dn_indblkshift - SPA_BLKPTRSHIFT));
394 td->td_paused = B_TRUE;
428870ff 395 }
34dc7c2f 396
fbeddd60 397 return (err);
34dc7c2f
BB
398}
399
96b89346
MA
400static void
401prefetch_dnode_metadata(traverse_data_t *td, const dnode_phys_t *dnp,
294f6806 402 uint64_t objset, uint64_t object)
96b89346
MA
403{
404 int j;
405 zbookmark_t czb;
406
407 for (j = 0; j < dnp->dn_nblkptr; j++) {
408 SET_BOOKMARK(&czb, objset, object, dnp->dn_nlevels - 1, j);
294f6806 409 traverse_prefetch_metadata(td, &dnp->dn_blkptr[j], &czb);
96b89346
MA
410 }
411
412 if (dnp->dn_flags & DNODE_FLAG_SPILL_BLKPTR) {
413 SET_BOOKMARK(&czb, objset, object, 0, DMU_SPILL_BLKID);
294f6806 414 traverse_prefetch_metadata(td, &dnp->dn_spill, &czb);
96b89346
MA
415 }
416}
417
9babb374 418static int
572e2857 419traverse_dnode(traverse_data_t *td, const dnode_phys_t *dnp,
294f6806 420 uint64_t objset, uint64_t object)
9babb374 421{
fbeddd60 422 int j, err = 0;
9babb374
BB
423 zbookmark_t czb;
424
425 for (j = 0; j < dnp->dn_nblkptr; j++) {
426 SET_BOOKMARK(&czb, objset, object, dnp->dn_nlevels - 1, j);
294f6806 427 err = traverse_visitbp(td, dnp, &dnp->dn_blkptr[j], &czb);
fbeddd60
MA
428 if (err != 0)
429 break;
9babb374 430 }
428870ff
BB
431
432 if (dnp->dn_flags & DNODE_FLAG_SPILL_BLKPTR) {
96b89346 433 SET_BOOKMARK(&czb, objset, object, 0, DMU_SPILL_BLKID);
294f6806 434 err = traverse_visitbp(td, dnp, &dnp->dn_spill, &czb);
428870ff 435 }
fbeddd60 436 return (err);
9babb374
BB
437}
438
b128c09f
BB
439/* ARGSUSED */
440static int
428870ff 441traverse_prefetcher(spa_t *spa, zilog_t *zilog, const blkptr_t *bp,
294f6806 442 const zbookmark_t *zb, const dnode_phys_t *dnp, void *arg)
34dc7c2f 443{
572e2857 444 prefetch_data_t *pfd = arg;
b128c09f 445 uint32_t aflags = ARC_NOWAIT | ARC_PREFETCH;
34dc7c2f 446
b128c09f
BB
447 ASSERT(pfd->pd_blks_fetched >= 0);
448 if (pfd->pd_cancel)
2e528b49 449 return (SET_ERROR(EINTR));
34dc7c2f 450
9b67f605 451 if (BP_IS_HOLE(bp) || BP_IS_EMBEDDED(bp) ||
b0bc7a84 452 !((pfd->pd_flags & TRAVERSE_PREFETCH_DATA) ||
428870ff
BB
453 BP_GET_TYPE(bp) == DMU_OT_DNODE || BP_GET_LEVEL(bp) > 0) ||
454 BP_GET_TYPE(bp) == DMU_OT_INTENT_LOG)
34dc7c2f
BB
455 return (0);
456
b128c09f
BB
457 mutex_enter(&pfd->pd_mtx);
458 while (!pfd->pd_cancel && pfd->pd_blks_fetched >= pfd->pd_blks_max)
459 cv_wait(&pfd->pd_cv, &pfd->pd_mtx);
460 pfd->pd_blks_fetched++;
461 cv_broadcast(&pfd->pd_cv);
462 mutex_exit(&pfd->pd_mtx);
34dc7c2f 463
294f6806
GW
464 (void) arc_read(NULL, spa, bp, NULL, NULL, ZIO_PRIORITY_ASYNC_READ,
465 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE, &aflags, zb);
34dc7c2f 466
b128c09f 467 return (0);
34dc7c2f
BB
468}
469
34dc7c2f 470static void
b128c09f 471traverse_prefetch_thread(void *arg)
34dc7c2f 472{
572e2857
BB
473 traverse_data_t *td_main = arg;
474 traverse_data_t td = *td_main;
b128c09f 475 zbookmark_t czb;
34dc7c2f 476
b128c09f
BB
477 td.td_func = traverse_prefetcher;
478 td.td_arg = td_main->td_pfd;
479 td.td_pfd = NULL;
34dc7c2f 480
428870ff
BB
481 SET_BOOKMARK(&czb, td.td_objset,
482 ZB_ROOT_OBJECT, ZB_ROOT_LEVEL, ZB_ROOT_BLKID);
294f6806 483 (void) traverse_visitbp(&td, NULL, td.td_rootbp, &czb);
34dc7c2f 484
b128c09f
BB
485 mutex_enter(&td_main->td_pfd->pd_mtx);
486 td_main->td_pfd->pd_exited = B_TRUE;
487 cv_broadcast(&td_main->td_pfd->pd_cv);
488 mutex_exit(&td_main->td_pfd->pd_mtx);
34dc7c2f
BB
489}
490
b128c09f
BB
491/*
492 * NB: dataset must not be changing on-disk (eg, is a snapshot or we are
493 * in syncing context).
494 */
495static int
9ae529ec
CS
496traverse_impl(spa_t *spa, dsl_dataset_t *ds, uint64_t objset, blkptr_t *rootbp,
497 uint64_t txg_start, zbookmark_t *resume, int flags,
498 blkptr_cb_t func, void *arg)
34dc7c2f 499{
47050a88
BB
500 traverse_data_t *td;
501 prefetch_data_t *pd;
502 zbookmark_t *czb;
b128c09f 503 int err;
34dc7c2f 504
9ae529ec
CS
505 ASSERT(ds == NULL || objset == ds->ds_object);
506 ASSERT(!(flags & TRAVERSE_PRE) || !(flags & TRAVERSE_POST));
507
96b89346
MA
508 /*
509 * The data prefetching mechanism (the prefetch thread) is incompatible
510 * with resuming from a bookmark.
511 */
512 ASSERT(resume == NULL || !(flags & TRAVERSE_PREFETCH_DATA));
513
a1687880
BB
514 td = kmem_alloc(sizeof (traverse_data_t), KM_PUSHPAGE);
515 pd = kmem_zalloc(sizeof (prefetch_data_t), KM_PUSHPAGE);
516 czb = kmem_alloc(sizeof (zbookmark_t), KM_PUSHPAGE);
47050a88
BB
517
518 td->td_spa = spa;
9ae529ec 519 td->td_objset = objset;
47050a88
BB
520 td->td_rootbp = rootbp;
521 td->td_min_txg = txg_start;
9ae529ec 522 td->td_resume = resume;
47050a88
BB
523 td->td_func = func;
524 td->td_arg = arg;
525 td->td_pfd = pd;
526 td->td_flags = flags;
fbeddd60 527 td->td_paused = B_FALSE;
b128c09f 528
47050a88
BB
529 pd->pd_blks_max = zfs_pd_blks_max;
530 pd->pd_flags = flags;
531 mutex_init(&pd->pd_mtx, NULL, MUTEX_DEFAULT, NULL);
532 cv_init(&pd->pd_cv, NULL, CV_DEFAULT, NULL);
b128c09f 533
1421c891
PS
534 SET_BOOKMARK(czb, td->td_objset,
535 ZB_ROOT_OBJECT, ZB_ROOT_LEVEL, ZB_ROOT_BLKID);
536
572e2857 537 /* See comment on ZIL traversal in dsl_scan_visitds. */
13fe0198
MA
538 if (ds != NULL && !dsl_dataset_is_snapshot(ds) && !BP_IS_HOLE(rootbp)) {
539 uint32_t flags = ARC_WAIT;
540 objset_phys_t *osp;
541 arc_buf_t *buf;
572e2857 542
13fe0198
MA
543 err = arc_read(NULL, td->td_spa, rootbp,
544 arc_getbuf_func, &buf,
1421c891 545 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, czb);
13fe0198 546 if (err != 0)
572e2857
BB
547 return (err);
548
13fe0198
MA
549 osp = buf->b_data;
550 traverse_zil(td, &osp->os_zil_header);
551 (void) arc_buf_remove_ref(buf, &buf);
572e2857
BB
552 }
553
96b89346 554 if (!(flags & TRAVERSE_PREFETCH_DATA) ||
b128c09f 555 0 == taskq_dispatch(system_taskq, traverse_prefetch_thread,
47050a88
BB
556 td, TQ_NOQUEUE))
557 pd->pd_exited = B_TRUE;
b128c09f 558
294f6806 559 err = traverse_visitbp(td, NULL, rootbp, czb);
47050a88
BB
560
561 mutex_enter(&pd->pd_mtx);
562 pd->pd_cancel = B_TRUE;
563 cv_broadcast(&pd->pd_cv);
564 while (!pd->pd_exited)
565 cv_wait(&pd->pd_cv, &pd->pd_mtx);
566 mutex_exit(&pd->pd_mtx);
b128c09f 567
47050a88
BB
568 mutex_destroy(&pd->pd_mtx);
569 cv_destroy(&pd->pd_cv);
b128c09f 570
a1687880
BB
571 kmem_free(czb, sizeof (zbookmark_t));
572 kmem_free(pd, sizeof (struct prefetch_data));
573 kmem_free(td, sizeof (struct traverse_data));
34dc7c2f 574
b128c09f 575 return (err);
34dc7c2f
BB
576}
577
b128c09f
BB
578/*
579 * NB: dataset must not be changing on-disk (eg, is a snapshot or we are
580 * in syncing context).
581 */
582int
583traverse_dataset(dsl_dataset_t *ds, uint64_t txg_start, int flags,
584 blkptr_cb_t func, void *arg)
34dc7c2f 585{
9ae529ec
CS
586 return (traverse_impl(ds->ds_dir->dd_pool->dp_spa, ds, ds->ds_object,
587 &ds->ds_phys->ds_bp, txg_start, NULL, flags, func, arg));
588}
589
590int
591traverse_dataset_destroyed(spa_t *spa, blkptr_t *blkptr,
592 uint64_t txg_start, zbookmark_t *resume, int flags,
593 blkptr_cb_t func, void *arg)
594{
595 return (traverse_impl(spa, NULL, ZB_DESTROYED_OBJSET,
596 blkptr, txg_start, resume, flags, func, arg));
34dc7c2f
BB
597}
598
b128c09f
BB
599/*
600 * NB: pool must not be changing on-disk (eg, from zdb or sync context).
601 */
602int
428870ff
BB
603traverse_pool(spa_t *spa, uint64_t txg_start, int flags,
604 blkptr_cb_t func, void *arg)
34dc7c2f 605{
fbeddd60 606 int err;
b128c09f
BB
607 uint64_t obj;
608 dsl_pool_t *dp = spa_get_dsl(spa);
609 objset_t *mos = dp->dp_meta_objset;
428870ff 610 boolean_t hard = (flags & TRAVERSE_HARD);
b128c09f
BB
611
612 /* visit the MOS */
9ae529ec
CS
613 err = traverse_impl(spa, NULL, 0, spa_get_rootblkptr(spa),
614 txg_start, NULL, flags, func, arg);
13fe0198 615 if (err != 0)
b128c09f
BB
616 return (err);
617
618 /* visit each dataset */
fbeddd60 619 for (obj = 1; err == 0;
428870ff 620 err = dmu_object_next(mos, &obj, FALSE, txg_start)) {
b128c09f
BB
621 dmu_object_info_t doi;
622
623 err = dmu_object_info(mos, obj, &doi);
13fe0198 624 if (err != 0) {
fbeddd60
MA
625 if (hard)
626 continue;
627 break;
428870ff 628 }
b128c09f 629
fa86b5db 630 if (doi.doi_bonus_type == DMU_OT_DSL_DATASET) {
b128c09f 631 dsl_dataset_t *ds;
428870ff
BB
632 uint64_t txg = txg_start;
633
13fe0198 634 dsl_pool_config_enter(dp, FTAG);
b128c09f 635 err = dsl_dataset_hold_obj(dp, obj, FTAG, &ds);
13fe0198
MA
636 dsl_pool_config_exit(dp, FTAG);
637 if (err != 0) {
fbeddd60
MA
638 if (hard)
639 continue;
640 break;
428870ff
BB
641 }
642 if (ds->ds_phys->ds_prev_snap_txg > txg)
643 txg = ds->ds_phys->ds_prev_snap_txg;
644 err = traverse_dataset(ds, txg, flags, func, arg);
b128c09f 645 dsl_dataset_rele(ds, FTAG);
fbeddd60
MA
646 if (err != 0)
647 break;
b128c09f 648 }
34dc7c2f 649 }
b128c09f
BB
650 if (err == ESRCH)
651 err = 0;
fbeddd60 652 return (err);
34dc7c2f 653}
c28b2279
BB
654
655#if defined(_KERNEL) && defined(HAVE_SPL)
656EXPORT_SYMBOL(traverse_dataset);
657EXPORT_SYMBOL(traverse_pool);
c409e464
BB
658
659module_param(zfs_pd_blks_max, int, 0644);
660MODULE_PARM_DESC(zfs_pd_blks_max, "Max number of blocks to prefetch");
c28b2279 661#endif