#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_btree_staging.h"
#include "xfs_bit.h"
#include "xfs_log_format.h"
#include "xfs_trans.h"
#include "xfs_sb.h"
#include "xfs_alloc.h"
#include "xfs_ialloc.h"
#include "xfs_rmap.h"
#include "xfs_rmap_btree.h"
#include "xfs_rtrmap_btree.h"
#include "xfs_refcount.h"
#include "xfs_rtrefcount_btree.h"
#include "xfs_error.h"
#include "xfs_health.h"
#include "xfs_inode.h"
#include "xfs_quota.h"
#include "xfs_rtalloc.h"
#include "xfs_ag.h"
#include "xfs_rtgroup.h"
#include "xfs_rtbitmap.h"
#include "scrub/xfs_scrub.h"
#include "scrub/scrub.h"
#include "scrub/common.h"
#include "scrub/btree.h"
#include "scrub/trace.h"
#include "scrub/repair.h"
#include "scrub/bitmap.h"
#include "scrub/fsb_bitmap.h"
#include "scrub/xfile.h"
#include "scrub/xfarray.h"
#include "scrub/newbt.h"
#include "scrub/reap.h"
#include "scrub/rcbag.h"
struct xrep_rtrefc {
struct xfarray *refcount_records;
struct xrep_newbt new_btree;
struct xfsb_bitmap old_rtrefcountbt_blocks;
struct xfs_scrub *sc;
xfarray_idx_t array_cur;
xfs_filblks_t btblocks;
};
int
xrep_setup_rtrefcountbt(
struct xfs_scrub *sc)
{
return xrep_setup_xfbtree(sc, "realtime rmap record bag");
}
STATIC int
xrep_rtrefc_check_ext(
struct xfs_scrub *sc,
const struct xfs_refcount_irec *rec)
{
xfs_rgblock_t last;
if (xfs_rtrefcount_check_irec(sc->sr.rtg, rec) != NULL)
return -EFSCORRUPTED;
if (xfs_rgbno_to_rtxoff(sc->mp, rec->rc_startblock) != 0)
return -EFSCORRUPTED;
last = rec->rc_startblock + rec->rc_blockcount - 1;
if (xfs_rgbno_to_rtxoff(sc->mp, last) != sc->mp->m_sb.sb_rextsize - 1)
return -EFSCORRUPTED;
return xrep_require_rtext_inuse(sc, rec->rc_startblock,
rec->rc_blockcount);
}
STATIC int
xrep_rtrefc_stash(
struct xrep_rtrefc *rr,
enum xfs_refc_domain domain,
xfs_rgblock_t bno,
xfs_extlen_t len,
uint64_t refcount)
{
struct xfs_refcount_irec irec = {
.rc_startblock = bno,
.rc_blockcount = len,
.rc_refcount = refcount,
.rc_domain = domain,
};
int error = 0;
if (xchk_should_terminate(rr->sc, &error))
return error;
irec.rc_refcount = min_t(uint64_t, XFS_REFC_REFCOUNT_MAX, refcount);
error = xrep_rtrefc_check_ext(rr->sc, &irec);
if (error)
return error;
trace_xrep_refc_found(rtg_group(rr->sc->sr.rtg), &irec);
return xfarray_append(rr->refcount_records, &irec);
}
STATIC int
xrep_rtrefc_stash_cow(
struct xrep_rtrefc *rr,
xfs_rgblock_t bno,
xfs_extlen_t len)
{
return xrep_rtrefc_stash(rr, XFS_REFC_DOMAIN_COW, bno, len, 1);
}
static inline bool
xrep_rtrefc_rmap_shareable(
const struct xfs_rmap_irec *rmap)
{
if (XFS_RMAP_NON_INODE_OWNER(rmap->rm_owner))
return false;
if (rmap->rm_flags & XFS_RMAP_UNWRITTEN)
return false;
return true;
}
STATIC int
xrep_rtrefc_walk_rmaps(
struct xrep_rtrefc *rr,
struct xfs_rmap_irec *rmap,
bool *have_rec)
{
struct xfs_btree_cur *cur = rr->sc->sr.rmap_cur;
struct xfs_mount *mp = cur->bc_mp;
int have_gt;
int error = 0;
*have_rec = false;
do {
if (xchk_should_terminate(rr->sc, &error))
return error;
error = xfs_btree_increment(cur, 0, &have_gt);
if (error)
return error;
if (!have_gt)
return 0;
error = xfs_rmap_get_rec(cur, rmap, &have_gt);
if (error)
return error;
if (XFS_IS_CORRUPT(mp, !have_gt)) {
xfs_btree_mark_sick(cur);
return -EFSCORRUPTED;
}
if (rmap->rm_owner == XFS_RMAP_OWN_COW) {
error = xrep_rtrefc_stash_cow(rr, rmap->rm_startblock,
rmap->rm_blockcount);
if (error)
return error;
} else if (xfs_is_sb_inum(mp, rmap->rm_owner) ||
(rmap->rm_flags & (XFS_RMAP_ATTR_FORK |
XFS_RMAP_BMBT_BLOCK))) {
xfs_btree_mark_sick(cur);
return -EFSCORRUPTED;
}
} while (!xrep_rtrefc_rmap_shareable(rmap));
*have_rec = true;
return 0;
}
static inline uint32_t
xrep_rtrefc_encode_startblock(
const struct xfs_refcount_irec *irec)
{
uint32_t start;
start = irec->rc_startblock & ~XFS_REFC_COWFLAG;
if (irec->rc_domain == XFS_REFC_DOMAIN_COW)
start |= XFS_REFC_COWFLAG;
return start;
}
static int
xrep_rtrefc_extent_cmp(
const void *a,
const void *b)
{
const struct xfs_refcount_irec *ap = a;
const struct xfs_refcount_irec *bp = b;
uint32_t sa, sb;
sa = xrep_rtrefc_encode_startblock(ap);
sb = xrep_rtrefc_encode_startblock(bp);
if (sa > sb)
return 1;
if (sa < sb)
return -1;
return 0;
}
STATIC int
xrep_rtrefc_sort_records(
struct xrep_rtrefc *rr)
{
struct xfs_refcount_irec irec;
xfarray_idx_t cur;
enum xfs_refc_domain dom = XFS_REFC_DOMAIN_SHARED;
xfs_rgblock_t next_rgbno = 0;
int error;
error = xfarray_sort(rr->refcount_records, xrep_rtrefc_extent_cmp,
XFARRAY_SORT_KILLABLE);
if (error)
return error;
foreach_xfarray_idx(rr->refcount_records, cur) {
if (xchk_should_terminate(rr->sc, &error))
return error;
error = xfarray_load(rr->refcount_records, cur, &irec);
if (error)
return error;
if (dom == XFS_REFC_DOMAIN_SHARED &&
irec.rc_domain == XFS_REFC_DOMAIN_COW) {
dom = irec.rc_domain;
next_rgbno = 0;
}
if (dom != irec.rc_domain)
return -EFSCORRUPTED;
if (irec.rc_startblock < next_rgbno)
return -EFSCORRUPTED;
next_rgbno = irec.rc_startblock + irec.rc_blockcount;
}
return error;
}
STATIC int
xrep_rtrefc_walk_rmap(
struct xfs_btree_cur *cur,
const struct xfs_rmap_irec *rec,
void *priv)
{
struct xrep_rtrefc *rr = priv;
int error = 0;
if (xchk_should_terminate(rr->sc, &error))
return error;
if (rec->rm_owner != rr->sc->ip->i_ino)
return 0;
error = xrep_check_ino_btree_mapping(rr->sc, rec);
if (error)
return error;
return xfsb_bitmap_set(&rr->old_rtrefcountbt_blocks,
xfs_gbno_to_fsb(cur->bc_group, rec->rm_startblock),
rec->rm_blockcount);
}
static int
xrep_rtrefc_push_rmaps_at(
struct xrep_rtrefc *rr,
struct rcbag *rcstack,
xfs_rgblock_t bno,
struct xfs_rmap_irec *rmap,
bool *have)
{
struct xfs_scrub *sc = rr->sc;
int have_gt;
int error;
while (*have && rmap->rm_startblock == bno) {
error = rcbag_add(rcstack, rr->sc->tp, rmap);
if (error)
return error;
error = xrep_rtrefc_walk_rmaps(rr, rmap, have);
if (error)
return error;
}
error = xfs_btree_decrement(sc->sr.rmap_cur, 0, &have_gt);
if (error)
return error;
if (XFS_IS_CORRUPT(sc->mp, !have_gt)) {
xfs_btree_mark_sick(sc->sr.rmap_cur);
return -EFSCORRUPTED;
}
return 0;
}
STATIC int
xrep_rtrefc_scan_ag(
struct xrep_rtrefc *rr,
struct xfs_perag *pag)
{
struct xfs_scrub *sc = rr->sc;
int error;
error = xrep_ag_init(sc, pag, &sc->sa);
if (error)
return error;
error = xfs_rmap_query_all(sc->sa.rmap_cur, xrep_rtrefc_walk_rmap, rr);
xchk_ag_free(sc, &sc->sa);
return error;
}
STATIC int
xrep_rtrefc_find_refcounts(
struct xrep_rtrefc *rr)
{
struct xfs_scrub *sc = rr->sc;
struct rcbag *rcstack;
struct xfs_perag *pag = NULL;
uint64_t old_stack_height;
xfs_rgblock_t sbno;
xfs_rgblock_t cbno;
xfs_rgblock_t nbno;
bool have;
int error;
while ((pag = xfs_perag_next(sc->mp, pag))) {
error = xrep_rtrefc_scan_ag(rr, pag);
if (error) {
xfs_perag_rele(pag);
return error;
}
}
xrep_rtgroup_btcur_init(sc, &sc->sr);
error = rcbag_init(sc->mp, sc->xmbtp, &rcstack);
if (error)
goto out_cur;
error = xfs_btree_goto_left_edge(sc->sr.rmap_cur);
if (error)
goto out_bag;
while (xfs_btree_has_more_records(sc->sr.rmap_cur)) {
struct xfs_rmap_irec rmap;
error = xrep_rtrefc_walk_rmaps(rr, &rmap, &have);
if (error)
goto out_bag;
if (!have)
break;
sbno = cbno = rmap.rm_startblock;
error = xrep_rtrefc_push_rmaps_at(rr, rcstack, sbno, &rmap,
&have);
if (error)
goto out_bag;
error = rcbag_next_edge(rcstack, sc->tp, &rmap, have, &nbno);
if (error)
goto out_bag;
ASSERT(nbno > sbno);
old_stack_height = rcbag_count(rcstack);
while (rcbag_count(rcstack) > 0) {
error = rcbag_remove_ending_at(rcstack, sc->tp, nbno);
if (error)
goto out_bag;
error = xrep_rtrefc_walk_rmaps(rr, &rmap, &have);
if (error)
goto out_bag;
if (have) {
error = xrep_rtrefc_push_rmaps_at(rr, rcstack,
nbno, &rmap, &have);
if (error)
goto out_bag;
}
ASSERT(nbno > cbno);
if (rcbag_count(rcstack) != old_stack_height) {
if (old_stack_height > 1) {
error = xrep_rtrefc_stash(rr,
XFS_REFC_DOMAIN_SHARED,
cbno, nbno - cbno,
old_stack_height);
if (error)
goto out_bag;
}
cbno = nbno;
}
if (rcbag_count(rcstack) == 0)
break;
old_stack_height = rcbag_count(rcstack);
sbno = nbno;
error = rcbag_next_edge(rcstack, sc->tp, &rmap, have,
&nbno);
if (error)
goto out_bag;
ASSERT(nbno > sbno);
}
}
ASSERT(rcbag_count(rcstack) == 0);
out_bag:
rcbag_free(&rcstack);
out_cur:
xchk_rtgroup_btcur_free(&sc->sr);
return error;
}
STATIC int
xrep_rtrefc_get_records(
struct xfs_btree_cur *cur,
unsigned int idx,
struct xfs_btree_block *block,
unsigned int nr_wanted,
void *priv)
{
struct xrep_rtrefc *rr = priv;
union xfs_btree_rec *block_rec;
unsigned int loaded;
int error;
for (loaded = 0; loaded < nr_wanted; loaded++, idx++) {
error = xfarray_load(rr->refcount_records, rr->array_cur++,
&cur->bc_rec.rc);
if (error)
return error;
block_rec = xfs_btree_rec_addr(cur, idx, block);
cur->bc_ops->init_rec_from_cur(cur, block_rec);
}
return loaded;
}
STATIC int
xrep_rtrefc_claim_block(
struct xfs_btree_cur *cur,
union xfs_btree_ptr *ptr,
void *priv)
{
struct xrep_rtrefc *rr = priv;
return xrep_newbt_claim_block(cur, &rr->new_btree, ptr);
}
STATIC size_t
xrep_rtrefc_iroot_size(
struct xfs_btree_cur *cur,
unsigned int level,
unsigned int nr_this_level,
void *priv)
{
return xfs_rtrefcount_broot_space_calc(cur->bc_mp, level,
nr_this_level);
}
STATIC int
xrep_rtrefc_build_new_tree(
struct xrep_rtrefc *rr)
{
struct xfs_scrub *sc = rr->sc;
struct xfs_rtgroup *rtg = sc->sr.rtg;
struct xfs_btree_cur *refc_cur;
int error;
error = xrep_rtrefc_sort_records(rr);
if (error)
return error;
error = xrep_newbt_init_metadir_inode(&rr->new_btree, sc);
if (error)
return error;
rr->new_btree.bload.get_records = xrep_rtrefc_get_records;
rr->new_btree.bload.claim_block = xrep_rtrefc_claim_block;
rr->new_btree.bload.iroot_size = xrep_rtrefc_iroot_size;
refc_cur = xfs_rtrefcountbt_init_cursor(NULL, rtg);
xfs_btree_stage_ifakeroot(refc_cur, &rr->new_btree.ifake);
error = xfs_btree_bload_compute_geometry(refc_cur, &rr->new_btree.bload,
xfarray_length(rr->refcount_records));
if (error)
goto err_cur;
if (xchk_should_terminate(sc, &error))
goto err_cur;
error = xfs_trans_reserve_more_inode(sc->tp, rtg_refcount(rtg),
rr->new_btree.bload.nr_blocks, 0, true);
if (error)
goto err_cur;
error = xrep_newbt_alloc_blocks(&rr->new_btree,
rr->new_btree.bload.nr_blocks);
if (error)
goto err_cur;
rr->new_btree.ifake.if_fork->if_format = XFS_DINODE_FMT_META_BTREE;
rr->array_cur = XFARRAY_CURSOR_INIT;
error = xfs_btree_bload(refc_cur, &rr->new_btree.bload, rr);
if (error)
goto err_cur;
xfs_rtrefcountbt_commit_staged_btree(refc_cur, sc->tp);
xrep_inode_set_nblocks(rr->sc, rr->new_btree.ifake.if_blocks);
xfs_btree_del_cursor(refc_cur, 0);
error = xrep_newbt_commit(&rr->new_btree);
if (error)
return error;
return xrep_roll_trans(sc);
err_cur:
xfs_btree_del_cursor(refc_cur, error);
xrep_newbt_cancel(&rr->new_btree);
return error;
}
int
xrep_rtrefcountbt(
struct xfs_scrub *sc)
{
struct xrep_rtrefc *rr;
struct xfs_mount *mp = sc->mp;
int error;
if (!xfs_has_rtrmapbt(mp))
return -EOPNOTSUPP;
error = xrep_metadata_inode_forks(sc);
if (error)
return error;
rr = kzalloc_obj(struct xrep_rtrefc, XCHK_GFP_FLAGS);
if (!rr)
return -ENOMEM;
rr->sc = sc;
error = xfarray_create("realtime reference count records",
mp->m_sb.sb_rextents, sizeof(struct xfs_refcount_irec),
&rr->refcount_records);
if (error)
goto out_rr;
xfsb_bitmap_init(&rr->old_rtrefcountbt_blocks);
error = xrep_rtrefc_find_refcounts(rr);
if (error)
goto out_bitmap;
xfs_trans_ijoin(sc->tp, sc->ip, 0);
error = xrep_rtrefc_build_new_tree(rr);
if (error)
goto out_bitmap;
error = xrep_reap_metadir_fsblocks(rr->sc,
&rr->old_rtrefcountbt_blocks);
if (error)
goto out_bitmap;
out_bitmap:
xfsb_bitmap_destroy(&rr->old_rtrefcountbt_blocks);
xfarray_destroy(rr->refcount_records);
out_rr:
kfree(rr);
return error;
}