// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2022-2023 Oracle.  All Rights Reserved.
 * Author: Darrick J. Wong <djwong@kernel.org>
 */
#ifndef __XFS_SCRUB_NEWBT_H__
#define __XFS_SCRUB_NEWBT_H__

struct xfs_alloc_arg;

struct xrep_newbt_resv {
        /* Link to list of extents that we've reserved. */
        struct list_head        list;

        struct xfs_perag        *pag;

        /* Auto-freeing this reservation if we don't commit. */
        struct xfs_alloc_autoreap autoreap;

        /* AG block of the extent we reserved. */
        xfs_agblock_t           agbno;

        /* Length of the reservation. */
        xfs_extlen_t            len;

        /* How much of this reservation has been used. */
        xfs_extlen_t            used;
};

struct xrep_newbt {
        struct xfs_scrub        *sc;

        /* Custom allocation function, or NULL for xfs_alloc_vextent */
        int                     (*alloc_vextent)(struct xfs_scrub *sc,
                                                 struct xfs_alloc_arg *args,
                                                 xfs_fsblock_t alloc_hint);

        /* List of extents that we've reserved. */
        struct list_head        resv_list;

        /* Fake root for new btree. */
        union {
                struct xbtree_afakeroot afake;
                struct xbtree_ifakeroot ifake;
        };

        /* rmap owner of these blocks */
        struct xfs_owner_info   oinfo;

        /* btree geometry for the bulk loader */
        struct xfs_btree_bload  bload;

        /* Allocation hint */
        xfs_fsblock_t           alloc_hint;

        /* per-ag reservation type */
        enum xfs_ag_resv_type   resv;
};

void xrep_newbt_init_bare(struct xrep_newbt *xnr, struct xfs_scrub *sc);
void xrep_newbt_init_ag(struct xrep_newbt *xnr, struct xfs_scrub *sc,
                const struct xfs_owner_info *oinfo, xfs_fsblock_t alloc_hint,
                enum xfs_ag_resv_type resv);
int xrep_newbt_init_inode(struct xrep_newbt *xnr, struct xfs_scrub *sc,
                int whichfork, const struct xfs_owner_info *oinfo);
int xrep_newbt_init_metadir_inode(struct xrep_newbt *xnr, struct xfs_scrub *sc);
int xrep_newbt_alloc_blocks(struct xrep_newbt *xnr, uint64_t nr_blocks);
int xrep_newbt_add_extent(struct xrep_newbt *xnr, struct xfs_perag *pag,
                xfs_agblock_t agbno, xfs_extlen_t len);
void xrep_newbt_cancel(struct xrep_newbt *xnr);
int xrep_newbt_commit(struct xrep_newbt *xnr);
int xrep_newbt_claim_block(struct xfs_btree_cur *cur, struct xrep_newbt *xnr,
                union xfs_btree_ptr *ptr);
unsigned int xrep_newbt_unused_blocks(struct xrep_newbt *xnr);

#endif /* __XFS_SCRUB_NEWBT_H__ */