// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (c) 2018-2024 Oracle.  All Rights Reserved.
 * Author: Darrick J. Wong <djwong@kernel.org>
 */
#include "xfs_platform.h"
#include "xfs_fs.h"
#include "xfs_shared.h"
#include "xfs_format.h"
#include "xfs_trans_resv.h"
#include "xfs_mount.h"
#include "xfs_defer.h"
#include "xfs_btree.h"
#include "xfs_bit.h"
#include "xfs_log_format.h"
#include "xfs_trans.h"
#include "xfs_sb.h"
#include "xfs_inode.h"
#include "xfs_alloc.h"
#include "xfs_ialloc.h"
#include "xfs_rmap.h"
#include "xfs_health.h"
#include "scrub/xfs_scrub.h"
#include "scrub/scrub.h"
#include "scrub/common.h"
#include "scrub/trace.h"
#include "scrub/repair.h"
#include "scrub/fscounters.h"

/*
 * FS Summary Counters
 * ===================
 *
 * We correct errors in the filesystem summary counters by setting them to the
 * values computed during the obligatory scrub phase.  However, we must be
 * careful not to allow any other thread to change the counters while we're
 * computing and setting new values.  To achieve this, we freeze the
 * filesystem for the whole operation if the REPAIR flag is set.  The checking
 * function is stricter when we've frozen the fs.
 */

/*
 * Reset the superblock counters.  Caller is responsible for freezing the
 * filesystem during the calculation and reset phases.
 */
int
xrep_fscounters(
        struct xfs_scrub        *sc)
{
        struct xfs_mount        *mp = sc->mp;
        struct xchk_fscounters  *fsc = sc->buf;

        /*
         * Reinitialize the in-core counters from what we computed.  We froze
         * the filesystem, so there shouldn't be anyone else trying to modify
         * these counters.
         */
        if (!fsc->frozen) {
                ASSERT(fsc->frozen);
                return -EFSCORRUPTED;
        }

        trace_xrep_reset_counters(mp, fsc);

        percpu_counter_set(&mp->m_icount, fsc->icount);
        percpu_counter_set(&mp->m_ifree, fsc->ifree);
        xfs_set_freecounter(mp, XC_FREE_BLOCKS, fsc->fdblocks);

        /*
         * Online repair is only supported on v5 file systems, which require
         * lazy sb counters and thus no update of sb_fdblocks here.  But
         * sb_frextents only uses a lazy counter with rtgroups, and thus needs
         * to be updated directly here otherwise.  And for that we need to keep
         * track of the delalloc reservations separately, as they are are
         * subtracted from m_frextents, but not included in sb_frextents.
         */
        if (!xfs_has_zoned(mp)) {
                xfs_set_freecounter(mp, XC_FREE_RTEXTENTS,
                                fsc->frextents - fsc->frextents_delayed);
                if (!xfs_has_rtgroups(mp))
                        mp->m_sb.sb_frextents = fsc->frextents;
        }

        return 0;
}