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

/*
 * We use an xfile to construct new bitmap blocks for the portion of the
 * rtbitmap file that we're replacing.  Whereas the ondisk bitmap must be
 * accessed through the buffer cache, the xfile bitmap supports direct
 * word-level accesses.  Therefore, we create a small abstraction for linear
 * access.
 */
typedef unsigned long long xrep_wordoff_t;
typedef unsigned int xrep_wordcnt_t;

/* Mask to round an rtx down to the nearest bitmap word. */
#define XREP_RTBMP_WORDMASK     ((1ULL << XFS_NBWORDLOG) - 1)


struct xchk_rtbitmap {
        struct xfs_scrub        *sc;

        uint64_t                rextents;
        uint64_t                rbmblocks;
        unsigned int            rextslog;
        unsigned int            resblks;

        /* The next free rt group block number that we expect to see. */
        xfs_rgblock_t           next_free_rgbno;

#ifdef CONFIG_XFS_ONLINE_REPAIR
        /* stuff for staging a new bitmap */
        struct xfs_rtalloc_args args;
        struct xrep_tempexch    tempexch;
#endif

        /* The next rtgroup block we expect to see during our rtrmapbt walk. */
        xfs_rgblock_t           next_rgbno;

        /* rtgroup lock flags */
        unsigned int            rtglock_flags;

        /* rtword position of xfile as we write buffers to disk. */
        xrep_wordoff_t          prep_wordoff;

        /* In-Memory rtbitmap for repair. */
        union xfs_rtword_raw    words[];
};

#ifdef CONFIG_XFS_ONLINE_REPAIR
int xrep_setup_rtbitmap(struct xfs_scrub *sc, struct xchk_rtbitmap *rtb);

/*
 * How big should the words[] buffer be?
 *
 * For repairs, we want a full fsblock worth of space so that we can memcpy a
 * buffer full of 1s into the xfile bitmap.  The xfile bitmap doesn't have
 * rtbitmap block headers, so we don't use blockwsize.  Scrub doesn't use the
 * words buffer at all.
 */
static inline unsigned int
xchk_rtbitmap_wordcnt(
        struct xfs_scrub        *sc)
{
        if (xchk_could_repair(sc))
                return sc->mp->m_sb.sb_blocksize >> XFS_WORDLOG;
        return 0;
}
#else
# define xrep_setup_rtbitmap(sc, rtb)   (0)
# define xchk_rtbitmap_wordcnt(sc)      (0)
#endif /* CONFIG_XFS_ONLINE_REPAIR */

#endif /* __XFS_SCRUB_RTBITMAP_H__ */