]> git.proxmox.com Git - mirror_zfs.git/blame - module/zfs/dmu_traverse.c
Retire KM_NODEBUG
[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.
1fa8f795 23 * Copyright (c) 2012, 2014 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;
5dbd68a3 58 zbookmark_phys_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;
5dbd68a3 75 zbookmark_phys_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;
5dbd68a3 99 zbookmark_phys_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,
5dbd68a3 153 const zbookmark_phys_t *zb)
9ae529ec
CS
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,
5dbd68a3 178 const blkptr_t *bp, const zbookmark_phys_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
1fa8f795
MA
200static boolean_t
201prefetch_needed(prefetch_data_t *pfd, const blkptr_t *bp)
202{
203 ASSERT(pfd->pd_flags & TRAVERSE_PREFETCH_DATA);
204 if (BP_IS_HOLE(bp) || BP_IS_EMBEDDED(bp) ||
205 BP_GET_TYPE(bp) == DMU_OT_INTENT_LOG)
206 return (B_FALSE);
207 return (B_TRUE);
208}
209
c7f8f831
BB
210static int
211traverse_visitbp(traverse_data_t *td, const dnode_phys_t *dnp,
5dbd68a3 212 const blkptr_t *bp, const zbookmark_phys_t *zb)
6656bf56 213{
fbeddd60 214 int err = 0;
c7f8f831 215 arc_buf_t *buf = NULL;
9ae529ec
CS
216
217 switch (resume_skip_check(td, dnp, zb)) {
218 case RESUME_SKIP_ALL:
219 return (0);
220 case RESUME_SKIP_CHILDREN:
221 goto post;
222 case RESUME_SKIP_NONE:
223 break;
224 default:
225 ASSERT(0);
226 }
6656bf56 227
b0bc7a84
MG
228 if (bp->blk_birth == 0) {
229 if (spa_feature_is_active(td->td_spa, SPA_FEATURE_HOLE_BIRTH)) {
230 /*
231 * Since this block has a birth time of 0 it must be a
232 * hole created before the SPA_FEATURE_HOLE_BIRTH
233 * feature was enabled. If SPA_FEATURE_HOLE_BIRTH
234 * was enabled before the min_txg for this traveral we
235 * know the hole must have been created before the
236 * min_txg for this traveral, so we can skip it. If
237 * SPA_FEATURE_HOLE_BIRTH was enabled after the min_txg
238 * for this traveral we cannot tell if the hole was
239 * created before or after the min_txg for this
240 * traversal, so we cannot skip it.
241 */
242 uint64_t hole_birth_enabled_txg;
243 VERIFY(spa_feature_enabled_txg(td->td_spa,
244 SPA_FEATURE_HOLE_BIRTH, &hole_birth_enabled_txg));
245 if (hole_birth_enabled_txg < td->td_min_txg)
246 return (0);
247 }
248 } else if (bp->blk_birth <= td->td_min_txg) {
249 return (0);
250 }
251
1fa8f795
MA
252 if (td->td_pfd != NULL && !td->td_pfd->pd_exited &&
253 prefetch_needed(td->td_pfd, bp)) {
a1687880
BB
254 mutex_enter(&td->td_pfd->pd_mtx);
255 ASSERT(td->td_pfd->pd_blks_fetched >= 0);
256 while (td->td_pfd->pd_blks_fetched == 0 &&
257 !td->td_pfd->pd_exited)
258 cv_wait(&td->td_pfd->pd_cv, &td->td_pfd->pd_mtx);
259 td->td_pfd->pd_blks_fetched--;
260 cv_broadcast(&td->td_pfd->pd_cv);
261 mutex_exit(&td->td_pfd->pd_mtx);
34dc7c2f
BB
262 }
263
1fa8f795
MA
264 if (BP_IS_HOLE(bp)) {
265 err = td->td_func(td->td_spa, NULL, bp, zb, dnp, td->td_arg);
266 if (err != 0)
267 goto post;
268 return (0);
269 }
270
c7f8f831 271 if (td->td_flags & TRAVERSE_PRE) {
294f6806 272 err = td->td_func(td->td_spa, NULL, bp, zb, dnp,
c7f8f831
BB
273 td->td_arg);
274 if (err == TRAVERSE_VISIT_NO_CHILDREN)
572e2857 275 return (0);
9ae529ec
CS
276 if (err != 0)
277 goto post;
34dc7c2f
BB
278 }
279
c7f8f831
BB
280 if (BP_GET_LEVEL(bp) > 0) {
281 uint32_t flags = ARC_WAIT;
a1687880
BB
282 int32_t i;
283 int32_t epb = BP_GET_LSIZE(bp) >> SPA_BLKPTRSHIFT;
5dbd68a3 284 zbookmark_phys_t *czb;
b128c09f 285
294f6806 286 err = arc_read(NULL, td->td_spa, bp, arc_getbuf_func, &buf,
c7f8f831 287 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
13fe0198 288 if (err != 0)
fbeddd60 289 goto post;
a1687880 290
5dbd68a3 291 czb = kmem_alloc(sizeof (zbookmark_phys_t), KM_PUSHPAGE);
96b89346
MA
292
293 for (i = 0; i < epb; i++) {
a1687880 294 SET_BOOKMARK(czb, zb->zb_objset, zb->zb_object,
96b89346
MA
295 zb->zb_level - 1,
296 zb->zb_blkid * epb + i);
a1687880
BB
297 traverse_prefetch_metadata(td,
298 &((blkptr_t *)buf->b_data)[i], czb);
96b89346 299 }
b128c09f
BB
300
301 /* recursively visitbp() blocks below this */
96b89346 302 for (i = 0; i < epb; i++) {
a1687880 303 SET_BOOKMARK(czb, zb->zb_objset, zb->zb_object,
c7f8f831
BB
304 zb->zb_level - 1,
305 zb->zb_blkid * epb + i);
a1687880
BB
306 err = traverse_visitbp(td, dnp,
307 &((blkptr_t *)buf->b_data)[i], czb);
fbeddd60
MA
308 if (err != 0)
309 break;
b128c09f 310 }
a1687880 311
5dbd68a3 312 kmem_free(czb, sizeof (zbookmark_phys_t));
a1687880 313
c7f8f831
BB
314 } else if (BP_GET_TYPE(bp) == DMU_OT_DNODE) {
315 uint32_t flags = ARC_WAIT;
a1687880
BB
316 int32_t i;
317 int32_t epb = BP_GET_LSIZE(bp) >> DNODE_SHIFT;
b128c09f 318
294f6806 319 err = arc_read(NULL, td->td_spa, bp, arc_getbuf_func, &buf,
c7f8f831 320 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
13fe0198 321 if (err != 0)
fbeddd60 322 goto post;
96b89346
MA
323 dnp = buf->b_data;
324
325 for (i = 0; i < epb; i++) {
294f6806 326 prefetch_dnode_metadata(td, &dnp[i], zb->zb_objset,
96b89346
MA
327 zb->zb_blkid * epb + i);
328 }
b128c09f
BB
329
330 /* recursively visitbp() blocks below this */
96b89346 331 for (i = 0; i < epb; i++) {
294f6806 332 err = traverse_dnode(td, &dnp[i], zb->zb_objset,
c7f8f831 333 zb->zb_blkid * epb + i);
fbeddd60
MA
334 if (err != 0)
335 break;
34dc7c2f 336 }
c7f8f831
BB
337 } else if (BP_GET_TYPE(bp) == DMU_OT_OBJSET) {
338 uint32_t flags = ARC_WAIT;
339 objset_phys_t *osp;
340 dnode_phys_t *dnp;
341
294f6806 342 err = arc_read(NULL, td->td_spa, bp, arc_getbuf_func, &buf,
c7f8f831 343 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, zb);
13fe0198 344 if (err != 0)
fbeddd60 345 goto post;
c7f8f831
BB
346
347 osp = buf->b_data;
348 dnp = &osp->os_meta_dnode;
294f6806 349 prefetch_dnode_metadata(td, dnp, zb->zb_objset,
96b89346
MA
350 DMU_META_DNODE_OBJECT);
351 if (arc_buf_size(buf) >= sizeof (objset_phys_t)) {
96b89346 352 prefetch_dnode_metadata(td, &osp->os_groupused_dnode,
8b464649
MA
353 zb->zb_objset, DMU_GROUPUSED_OBJECT);
354 prefetch_dnode_metadata(td, &osp->os_userused_dnode,
294f6806 355 zb->zb_objset, DMU_USERUSED_OBJECT);
96b89346
MA
356 }
357
294f6806 358 err = traverse_dnode(td, dnp, zb->zb_objset,
c7f8f831 359 DMU_META_DNODE_OBJECT);
c7f8f831 360 if (err == 0 && arc_buf_size(buf) >= sizeof (objset_phys_t)) {
8b464649 361 dnp = &osp->os_groupused_dnode;
294f6806 362 err = traverse_dnode(td, dnp, zb->zb_objset,
8b464649 363 DMU_GROUPUSED_OBJECT);
9babb374 364 }
c7f8f831 365 if (err == 0 && arc_buf_size(buf) >= sizeof (objset_phys_t)) {
8b464649 366 dnp = &osp->os_userused_dnode;
294f6806 367 err = traverse_dnode(td, dnp, zb->zb_objset,
8b464649 368 DMU_USERUSED_OBJECT);
34dc7c2f 369 }
34dc7c2f
BB
370 }
371
c7f8f831
BB
372 if (buf)
373 (void) arc_buf_remove_ref(buf, &buf);
34dc7c2f 374
9ae529ec 375post:
fbeddd60 376 if (err == 0 && (td->td_flags & TRAVERSE_POST))
294f6806 377 err = td->td_func(td->td_spa, NULL, bp, zb, dnp, td->td_arg);
fbeddd60
MA
378
379 if ((td->td_flags & TRAVERSE_HARD) && (err == EIO || err == ECKSUM)) {
380 /*
381 * Ignore this disk error as requested by the HARD flag,
382 * and continue traversal.
383 */
384 err = 0;
9ae529ec
CS
385 }
386
fbeddd60
MA
387 /*
388 * If we are stopping here, set td_resume.
389 */
390 if (td->td_resume != NULL && err != 0 && !td->td_paused) {
391 td->td_resume->zb_objset = zb->zb_objset;
392 td->td_resume->zb_object = zb->zb_object;
393 td->td_resume->zb_level = 0;
394 /*
395 * If we have stopped on an indirect block (e.g. due to
396 * i/o error), we have not visited anything below it.
397 * Set the bookmark to the first level-0 block that we need
398 * to visit. This way, the resuming code does not need to
399 * deal with resuming from indirect blocks.
400 */
401 td->td_resume->zb_blkid = zb->zb_blkid <<
402 (zb->zb_level * (dnp->dn_indblkshift - SPA_BLKPTRSHIFT));
403 td->td_paused = B_TRUE;
428870ff 404 }
34dc7c2f 405
fbeddd60 406 return (err);
34dc7c2f
BB
407}
408
96b89346
MA
409static void
410prefetch_dnode_metadata(traverse_data_t *td, const dnode_phys_t *dnp,
294f6806 411 uint64_t objset, uint64_t object)
96b89346
MA
412{
413 int j;
5dbd68a3 414 zbookmark_phys_t czb;
96b89346
MA
415
416 for (j = 0; j < dnp->dn_nblkptr; j++) {
417 SET_BOOKMARK(&czb, objset, object, dnp->dn_nlevels - 1, j);
294f6806 418 traverse_prefetch_metadata(td, &dnp->dn_blkptr[j], &czb);
96b89346
MA
419 }
420
421 if (dnp->dn_flags & DNODE_FLAG_SPILL_BLKPTR) {
422 SET_BOOKMARK(&czb, objset, object, 0, DMU_SPILL_BLKID);
294f6806 423 traverse_prefetch_metadata(td, &dnp->dn_spill, &czb);
96b89346
MA
424 }
425}
426
9babb374 427static int
572e2857 428traverse_dnode(traverse_data_t *td, const dnode_phys_t *dnp,
294f6806 429 uint64_t objset, uint64_t object)
9babb374 430{
fbeddd60 431 int j, err = 0;
5dbd68a3 432 zbookmark_phys_t czb;
9babb374
BB
433
434 for (j = 0; j < dnp->dn_nblkptr; j++) {
435 SET_BOOKMARK(&czb, objset, object, dnp->dn_nlevels - 1, j);
294f6806 436 err = traverse_visitbp(td, dnp, &dnp->dn_blkptr[j], &czb);
fbeddd60
MA
437 if (err != 0)
438 break;
9babb374 439 }
428870ff
BB
440
441 if (dnp->dn_flags & DNODE_FLAG_SPILL_BLKPTR) {
96b89346 442 SET_BOOKMARK(&czb, objset, object, 0, DMU_SPILL_BLKID);
294f6806 443 err = traverse_visitbp(td, dnp, &dnp->dn_spill, &czb);
428870ff 444 }
fbeddd60 445 return (err);
9babb374
BB
446}
447
b128c09f
BB
448/* ARGSUSED */
449static int
428870ff 450traverse_prefetcher(spa_t *spa, zilog_t *zilog, const blkptr_t *bp,
5dbd68a3 451 const zbookmark_phys_t *zb, const dnode_phys_t *dnp, void *arg)
34dc7c2f 452{
572e2857 453 prefetch_data_t *pfd = arg;
b128c09f 454 uint32_t aflags = ARC_NOWAIT | ARC_PREFETCH;
34dc7c2f 455
b128c09f
BB
456 ASSERT(pfd->pd_blks_fetched >= 0);
457 if (pfd->pd_cancel)
2e528b49 458 return (SET_ERROR(EINTR));
34dc7c2f 459
1fa8f795 460 if (!prefetch_needed(pfd, bp))
34dc7c2f
BB
461 return (0);
462
b128c09f
BB
463 mutex_enter(&pfd->pd_mtx);
464 while (!pfd->pd_cancel && pfd->pd_blks_fetched >= pfd->pd_blks_max)
465 cv_wait(&pfd->pd_cv, &pfd->pd_mtx);
466 pfd->pd_blks_fetched++;
467 cv_broadcast(&pfd->pd_cv);
468 mutex_exit(&pfd->pd_mtx);
34dc7c2f 469
294f6806
GW
470 (void) arc_read(NULL, spa, bp, NULL, NULL, ZIO_PRIORITY_ASYNC_READ,
471 ZIO_FLAG_CANFAIL | ZIO_FLAG_SPECULATIVE, &aflags, zb);
34dc7c2f 472
b128c09f 473 return (0);
34dc7c2f
BB
474}
475
34dc7c2f 476static void
b128c09f 477traverse_prefetch_thread(void *arg)
34dc7c2f 478{
572e2857
BB
479 traverse_data_t *td_main = arg;
480 traverse_data_t td = *td_main;
5dbd68a3 481 zbookmark_phys_t czb;
34dc7c2f 482
b128c09f
BB
483 td.td_func = traverse_prefetcher;
484 td.td_arg = td_main->td_pfd;
485 td.td_pfd = NULL;
34dc7c2f 486
428870ff
BB
487 SET_BOOKMARK(&czb, td.td_objset,
488 ZB_ROOT_OBJECT, ZB_ROOT_LEVEL, ZB_ROOT_BLKID);
294f6806 489 (void) traverse_visitbp(&td, NULL, td.td_rootbp, &czb);
34dc7c2f 490
b128c09f
BB
491 mutex_enter(&td_main->td_pfd->pd_mtx);
492 td_main->td_pfd->pd_exited = B_TRUE;
493 cv_broadcast(&td_main->td_pfd->pd_cv);
494 mutex_exit(&td_main->td_pfd->pd_mtx);
34dc7c2f
BB
495}
496
b128c09f
BB
497/*
498 * NB: dataset must not be changing on-disk (eg, is a snapshot or we are
499 * in syncing context).
500 */
501static int
9ae529ec 502traverse_impl(spa_t *spa, dsl_dataset_t *ds, uint64_t objset, blkptr_t *rootbp,
5dbd68a3 503 uint64_t txg_start, zbookmark_phys_t *resume, int flags,
9ae529ec 504 blkptr_cb_t func, void *arg)
34dc7c2f 505{
47050a88
BB
506 traverse_data_t *td;
507 prefetch_data_t *pd;
5dbd68a3 508 zbookmark_phys_t *czb;
b128c09f 509 int err;
34dc7c2f 510
9ae529ec
CS
511 ASSERT(ds == NULL || objset == ds->ds_object);
512 ASSERT(!(flags & TRAVERSE_PRE) || !(flags & TRAVERSE_POST));
513
96b89346
MA
514 /*
515 * The data prefetching mechanism (the prefetch thread) is incompatible
516 * with resuming from a bookmark.
517 */
518 ASSERT(resume == NULL || !(flags & TRAVERSE_PREFETCH_DATA));
519
a1687880
BB
520 td = kmem_alloc(sizeof (traverse_data_t), KM_PUSHPAGE);
521 pd = kmem_zalloc(sizeof (prefetch_data_t), KM_PUSHPAGE);
5dbd68a3 522 czb = kmem_alloc(sizeof (zbookmark_phys_t), KM_PUSHPAGE);
47050a88
BB
523
524 td->td_spa = spa;
9ae529ec 525 td->td_objset = objset;
47050a88
BB
526 td->td_rootbp = rootbp;
527 td->td_min_txg = txg_start;
9ae529ec 528 td->td_resume = resume;
47050a88
BB
529 td->td_func = func;
530 td->td_arg = arg;
531 td->td_pfd = pd;
532 td->td_flags = flags;
fbeddd60 533 td->td_paused = B_FALSE;
b128c09f 534
47050a88
BB
535 pd->pd_blks_max = zfs_pd_blks_max;
536 pd->pd_flags = flags;
537 mutex_init(&pd->pd_mtx, NULL, MUTEX_DEFAULT, NULL);
538 cv_init(&pd->pd_cv, NULL, CV_DEFAULT, NULL);
b128c09f 539
1421c891
PS
540 SET_BOOKMARK(czb, td->td_objset,
541 ZB_ROOT_OBJECT, ZB_ROOT_LEVEL, ZB_ROOT_BLKID);
542
572e2857 543 /* See comment on ZIL traversal in dsl_scan_visitds. */
13fe0198
MA
544 if (ds != NULL && !dsl_dataset_is_snapshot(ds) && !BP_IS_HOLE(rootbp)) {
545 uint32_t flags = ARC_WAIT;
546 objset_phys_t *osp;
547 arc_buf_t *buf;
572e2857 548
13fe0198
MA
549 err = arc_read(NULL, td->td_spa, rootbp,
550 arc_getbuf_func, &buf,
1421c891 551 ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_CANFAIL, &flags, czb);
13fe0198 552 if (err != 0)
572e2857
BB
553 return (err);
554
13fe0198
MA
555 osp = buf->b_data;
556 traverse_zil(td, &osp->os_zil_header);
557 (void) arc_buf_remove_ref(buf, &buf);
572e2857
BB
558 }
559
96b89346 560 if (!(flags & TRAVERSE_PREFETCH_DATA) ||
b128c09f 561 0 == taskq_dispatch(system_taskq, traverse_prefetch_thread,
47050a88
BB
562 td, TQ_NOQUEUE))
563 pd->pd_exited = B_TRUE;
b128c09f 564
294f6806 565 err = traverse_visitbp(td, NULL, rootbp, czb);
47050a88
BB
566
567 mutex_enter(&pd->pd_mtx);
568 pd->pd_cancel = B_TRUE;
569 cv_broadcast(&pd->pd_cv);
570 while (!pd->pd_exited)
571 cv_wait(&pd->pd_cv, &pd->pd_mtx);
572 mutex_exit(&pd->pd_mtx);
b128c09f 573
47050a88
BB
574 mutex_destroy(&pd->pd_mtx);
575 cv_destroy(&pd->pd_cv);
b128c09f 576
5dbd68a3 577 kmem_free(czb, sizeof (zbookmark_phys_t));
a1687880
BB
578 kmem_free(pd, sizeof (struct prefetch_data));
579 kmem_free(td, sizeof (struct traverse_data));
34dc7c2f 580
b128c09f 581 return (err);
34dc7c2f
BB
582}
583
b128c09f
BB
584/*
585 * NB: dataset must not be changing on-disk (eg, is a snapshot or we are
586 * in syncing context).
587 */
588int
589traverse_dataset(dsl_dataset_t *ds, uint64_t txg_start, int flags,
590 blkptr_cb_t func, void *arg)
34dc7c2f 591{
9ae529ec
CS
592 return (traverse_impl(ds->ds_dir->dd_pool->dp_spa, ds, ds->ds_object,
593 &ds->ds_phys->ds_bp, txg_start, NULL, flags, func, arg));
594}
595
596int
597traverse_dataset_destroyed(spa_t *spa, blkptr_t *blkptr,
5dbd68a3 598 uint64_t txg_start, zbookmark_phys_t *resume, int flags,
9ae529ec
CS
599 blkptr_cb_t func, void *arg)
600{
601 return (traverse_impl(spa, NULL, ZB_DESTROYED_OBJSET,
602 blkptr, txg_start, resume, flags, func, arg));
34dc7c2f
BB
603}
604
b128c09f
BB
605/*
606 * NB: pool must not be changing on-disk (eg, from zdb or sync context).
607 */
608int
428870ff
BB
609traverse_pool(spa_t *spa, uint64_t txg_start, int flags,
610 blkptr_cb_t func, void *arg)
34dc7c2f 611{
fbeddd60 612 int err;
b128c09f
BB
613 uint64_t obj;
614 dsl_pool_t *dp = spa_get_dsl(spa);
615 objset_t *mos = dp->dp_meta_objset;
428870ff 616 boolean_t hard = (flags & TRAVERSE_HARD);
b128c09f
BB
617
618 /* visit the MOS */
9ae529ec
CS
619 err = traverse_impl(spa, NULL, 0, spa_get_rootblkptr(spa),
620 txg_start, NULL, flags, func, arg);
13fe0198 621 if (err != 0)
b128c09f
BB
622 return (err);
623
624 /* visit each dataset */
fbeddd60 625 for (obj = 1; err == 0;
428870ff 626 err = dmu_object_next(mos, &obj, FALSE, txg_start)) {
b128c09f
BB
627 dmu_object_info_t doi;
628
629 err = dmu_object_info(mos, obj, &doi);
13fe0198 630 if (err != 0) {
fbeddd60
MA
631 if (hard)
632 continue;
633 break;
428870ff 634 }
b128c09f 635
fa86b5db 636 if (doi.doi_bonus_type == DMU_OT_DSL_DATASET) {
b128c09f 637 dsl_dataset_t *ds;
428870ff
BB
638 uint64_t txg = txg_start;
639
13fe0198 640 dsl_pool_config_enter(dp, FTAG);
b128c09f 641 err = dsl_dataset_hold_obj(dp, obj, FTAG, &ds);
13fe0198
MA
642 dsl_pool_config_exit(dp, FTAG);
643 if (err != 0) {
fbeddd60
MA
644 if (hard)
645 continue;
646 break;
428870ff
BB
647 }
648 if (ds->ds_phys->ds_prev_snap_txg > txg)
649 txg = ds->ds_phys->ds_prev_snap_txg;
650 err = traverse_dataset(ds, txg, flags, func, arg);
b128c09f 651 dsl_dataset_rele(ds, FTAG);
fbeddd60
MA
652 if (err != 0)
653 break;
b128c09f 654 }
34dc7c2f 655 }
b128c09f
BB
656 if (err == ESRCH)
657 err = 0;
fbeddd60 658 return (err);
34dc7c2f 659}
c28b2279
BB
660
661#if defined(_KERNEL) && defined(HAVE_SPL)
662EXPORT_SYMBOL(traverse_dataset);
663EXPORT_SYMBOL(traverse_pool);
c409e464
BB
664
665module_param(zfs_pd_blks_max, int, 0644);
666MODULE_PARM_DESC(zfs_pd_blks_max, "Max number of blocks to prefetch");
c28b2279 667#endif