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

/* Quota counters for live quotacheck. */
struct xqcheck_dquot {
        /* block usage count */
        int64_t                 bcount;

        /* inode usage count */
        int64_t                 icount;

        /* realtime block usage count */
        int64_t                 rtbcount;

        /* Record state */
        unsigned int            flags;
};

/*
 * This incore dquot record has been written at least once.  We never want to
 * store an xqcheck_dquot that looks uninitialized.
 */
#define XQCHECK_DQUOT_WRITTEN           (1U << 0)

/* Already checked this dquot. */
#define XQCHECK_DQUOT_COMPARE_SCANNED   (1U << 1)

/* Already repaired this dquot. */
#define XQCHECK_DQUOT_REPAIR_SCANNED    (1U << 2)

/* Live quotacheck control structure. */
struct xqcheck {
        struct xfs_scrub        *sc;

        /* Shadow dquot counter data. */
        struct xfarray          *ucounts;
        struct xfarray          *gcounts;
        struct xfarray          *pcounts;

        /* Lock protecting quotacheck count observations */
        struct mutex            lock;

        struct xchk_iscan       iscan;

        /* Hooks into the quota code. */
        struct xfs_dqtrx_hook   qhook;

        /* Shadow quota delta tracking structure. */
        struct rhashtable       shadow_dquot_acct;
};

/* Return the incore counter array for a given quota type. */
static inline struct xfarray *
xqcheck_counters_for(
        struct xqcheck          *xqc,
        xfs_dqtype_t            dqtype)
{
        switch (dqtype) {
        case XFS_DQTYPE_USER:
                return xqc->ucounts;
        case XFS_DQTYPE_GROUP:
                return xqc->gcounts;
        case XFS_DQTYPE_PROJ:
                return xqc->pcounts;
        }

        ASSERT(0);
        return NULL;
}

#endif /* __XFS_SCRUB_QUOTACHECK_H__ */