/* 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__ */