/*

 * CDDL HEADER START

 *

 * The contents of this file are subject to the terms of the

 *

 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

 * or http://www.opensolaris.org/os/licensing.

 * See the License for the specific language governing permissions

 * and limitations under the License.

 *

 * When distributing Covered Code, include this CDDL HEADER in each

 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.

 * If applicable, add the following below this CDDL HEADER, with the

 * fields enclosed by brackets "[]" replaced with your own identifying

 * information: Portions Copyright [yyyy] [name of copyright owner]

 *

 * CDDL HEADER END

 */

/*

 * Use is subject to license terms.

 */

#ifndef	_SYS_DMU_TRAVERSE_H

#define	_SYS_DMU_TRAVERSE_H

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#include <sys/zfs_context.h>

#include <sys/spa.h>

#include <sys/zio.h>

#include <sys/dmu.h>

#include <sys/dnode.h>

#include <sys/arc.h>

#ifdef	__cplusplus

extern "C" {

#endif

#define	ADVANCE_POST	0		/* post-order traversal */

#define	ADVANCE_PRE	0x01		/* pre-order traversal */

#define	ADVANCE_PRUNE	0x02		/* prune by prev snapshot birth time */

#define	ADVANCE_DATA	0x04		/* read user data blocks */

#define	ADVANCE_HOLES	0x08		/* visit holes */

#define	ZB_NO_LEVEL	-2

#define	ZB_MAXLEVEL	32		/* Next power of 2 >= DN_MAX_LEVELS */

#define	ZB_MAXBLKID	(1ULL << 62)

#define	ZB_MAXOBJSET	(1ULL << 62)

#define	ZB_MAXOBJECT	(1ULL << 62)

#define	ZB_MOS_CACHE	0

#define	ZB_MDN_CACHE	1

#define	ZB_DN_CACHE	2

#define	ZB_DEPTH	3

typedef struct zseg {

	uint64_t	seg_mintxg;

	uint64_t	seg_maxtxg;

	zbookmark_t	seg_start;

	zbookmark_t	seg_end;

	list_node_t	seg_node;

} zseg_t;

typedef struct traverse_blk_cache {

	zbookmark_t	bc_bookmark;

	blkptr_t	bc_blkptr;

	void		*bc_data;

	dnode_phys_t	*bc_dnode;

	int		bc_errno;

	int		bc_pad1;

	uint64_t	bc_pad2;

} traverse_blk_cache_t;

typedef int (blkptr_cb_t)(traverse_blk_cache_t *bc, spa_t *spa, void *arg);

struct traverse_handle {

	spa_t		*th_spa;

	blkptr_cb_t	*th_func;

	void		*th_arg;

	int		th_advance;

	int		th_zio_flags;

	list_t		th_seglist;

	traverse_blk_cache_t th_cache[ZB_DEPTH][ZB_MAXLEVEL];

	uint64_t	th_hits;

	uint64_t	th_arc_hits;

	uint64_t	th_reads;

	uint64_t	th_callbacks;

	uint64_t	th_syncs;

	uint64_t	th_restarts;

	zbookmark_t	th_noread;

	zbookmark_t	th_lastcb;

};

int traverse_dsl_dataset(struct dsl_dataset *ds, uint64_t txg_start,

    int advance, blkptr_cb_t func, void *arg);

traverse_handle_t *traverse_init(spa_t *spa, blkptr_cb_t *func, void *arg,

    int advance, int zio_flags);

void traverse_fini(traverse_handle_t *th);

void traverse_add_dnode(traverse_handle_t *th,

    uint64_t mintxg, uint64_t maxtxg, uint64_t objset, uint64_t object);

void traverse_add_objset(traverse_handle_t *th,

    uint64_t mintxg, uint64_t maxtxg, uint64_t objset);

void traverse_add_pool(traverse_handle_t *th, uint64_t mintxg, uint64_t maxtxg);

int traverse_more(traverse_handle_t *th);

#ifdef	__cplusplus

}

#endif

#endif /* _SYS_DMU_TRAVERSE_H */