10b61f8a4SDave Chinner // SPDX-License-Identifier: GPL-2.0+ 21946b91cSDarrick J. Wong /* 31946b91cSDarrick J. Wong * Copyright (C) 2016 Oracle. All Rights Reserved. 41946b91cSDarrick J. Wong * Author: Darrick J. Wong <darrick.wong@oracle.com> 51946b91cSDarrick J. Wong */ 61946b91cSDarrick J. Wong #include "xfs.h" 71946b91cSDarrick J. Wong #include "xfs_fs.h" 81946b91cSDarrick J. Wong #include "xfs_shared.h" 91946b91cSDarrick J. Wong #include "xfs_format.h" 101946b91cSDarrick J. Wong #include "xfs_log_format.h" 111946b91cSDarrick J. Wong #include "xfs_trans_resv.h" 121946b91cSDarrick J. Wong #include "xfs_mount.h" 131946b91cSDarrick J. Wong #include "xfs_btree.h" 1456e98164SDarrick J. Wong #include "xfs_btree_staging.h" 151946b91cSDarrick J. Wong #include "xfs_refcount_btree.h" 161946b91cSDarrick J. Wong #include "xfs_alloc.h" 171946b91cSDarrick J. Wong #include "xfs_error.h" 181946b91cSDarrick J. Wong #include "xfs_trace.h" 191946b91cSDarrick J. Wong #include "xfs_trans.h" 201946b91cSDarrick J. Wong #include "xfs_bit.h" 21bdf28630SDarrick J. Wong #include "xfs_rmap.h" 229bbafc71SDave Chinner #include "xfs_ag.h" 231946b91cSDarrick J. Wong 241946b91cSDarrick J. Wong static struct xfs_btree_cur * 251946b91cSDarrick J. Wong xfs_refcountbt_dup_cursor( 261946b91cSDarrick J. Wong struct xfs_btree_cur *cur) 271946b91cSDarrick J. Wong { 281946b91cSDarrick J. Wong return xfs_refcountbt_init_cursor(cur->bc_mp, cur->bc_tp, 29a81a0621SDave Chinner cur->bc_ag.agbp, cur->bc_ag.pag); 301946b91cSDarrick J. Wong } 311946b91cSDarrick J. Wong 32bdf28630SDarrick J. Wong STATIC void 33bdf28630SDarrick J. Wong xfs_refcountbt_set_root( 34bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 35b5a6e5feSDarrick J. Wong const union xfs_btree_ptr *ptr, 36bdf28630SDarrick J. Wong int inc) 37bdf28630SDarrick J. Wong { 38576af732SDave Chinner struct xfs_buf *agbp = cur->bc_ag.agbp; 399798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 4092a00544SGao Xiang struct xfs_perag *pag = agbp->b_pag; 41bdf28630SDarrick J. Wong 42bdf28630SDarrick J. Wong ASSERT(ptr->s != 0); 43bdf28630SDarrick J. Wong 44bdf28630SDarrick J. Wong agf->agf_refcount_root = ptr->s; 45bdf28630SDarrick J. Wong be32_add_cpu(&agf->agf_refcount_level, inc); 46bdf28630SDarrick J. Wong pag->pagf_refcount_level += inc; 47bdf28630SDarrick J. Wong 48bdf28630SDarrick J. Wong xfs_alloc_log_agf(cur->bc_tp, agbp, 49bdf28630SDarrick J. Wong XFS_AGF_REFCOUNT_ROOT | XFS_AGF_REFCOUNT_LEVEL); 50bdf28630SDarrick J. Wong } 51bdf28630SDarrick J. Wong 52bdf28630SDarrick J. Wong STATIC int 53bdf28630SDarrick J. Wong xfs_refcountbt_alloc_block( 54bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 55deb06b9aSDarrick J. Wong const union xfs_btree_ptr *start, 56bdf28630SDarrick J. Wong union xfs_btree_ptr *new, 57bdf28630SDarrick J. Wong int *stat) 58bdf28630SDarrick J. Wong { 59576af732SDave Chinner struct xfs_buf *agbp = cur->bc_ag.agbp; 609798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 61bdf28630SDarrick J. Wong struct xfs_alloc_arg args; /* block allocation args */ 62bdf28630SDarrick J. Wong int error; /* error return value */ 63bdf28630SDarrick J. Wong 64bdf28630SDarrick J. Wong memset(&args, 0, sizeof(args)); 65bdf28630SDarrick J. Wong args.tp = cur->bc_tp; 66bdf28630SDarrick J. Wong args.mp = cur->bc_mp; 67bdf28630SDarrick J. Wong args.type = XFS_ALLOCTYPE_NEAR_BNO; 6850f02fe3SDave Chinner args.fsbno = XFS_AGB_TO_FSB(cur->bc_mp, cur->bc_ag.pag->pag_agno, 69bdf28630SDarrick J. Wong xfs_refc_block(args.mp)); 707280fedaSDarrick J. Wong args.oinfo = XFS_RMAP_OINFO_REFC; 71bdf28630SDarrick J. Wong args.minlen = args.maxlen = args.prod = 1; 7284d69619SDarrick J. Wong args.resv = XFS_AG_RESV_METADATA; 73bdf28630SDarrick J. Wong 74bdf28630SDarrick J. Wong error = xfs_alloc_vextent(&args); 75bdf28630SDarrick J. Wong if (error) 76bdf28630SDarrick J. Wong goto out_error; 7750f02fe3SDave Chinner trace_xfs_refcountbt_alloc_block(cur->bc_mp, cur->bc_ag.pag->pag_agno, 78bdf28630SDarrick J. Wong args.agbno, 1); 79bdf28630SDarrick J. Wong if (args.fsbno == NULLFSBLOCK) { 80bdf28630SDarrick J. Wong *stat = 0; 81bdf28630SDarrick J. Wong return 0; 82bdf28630SDarrick J. Wong } 8350f02fe3SDave Chinner ASSERT(args.agno == cur->bc_ag.pag->pag_agno); 84bdf28630SDarrick J. Wong ASSERT(args.len == 1); 85bdf28630SDarrick J. Wong 86bdf28630SDarrick J. Wong new->s = cpu_to_be32(args.agbno); 87bdf28630SDarrick J. Wong be32_add_cpu(&agf->agf_refcount_blocks, 1); 88bdf28630SDarrick J. Wong xfs_alloc_log_agf(cur->bc_tp, agbp, XFS_AGF_REFCOUNT_BLOCKS); 89bdf28630SDarrick J. Wong 90bdf28630SDarrick J. Wong *stat = 1; 91bdf28630SDarrick J. Wong return 0; 92bdf28630SDarrick J. Wong 93bdf28630SDarrick J. Wong out_error: 94bdf28630SDarrick J. Wong return error; 95bdf28630SDarrick J. Wong } 96bdf28630SDarrick J. Wong 97bdf28630SDarrick J. Wong STATIC int 98bdf28630SDarrick J. Wong xfs_refcountbt_free_block( 99bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 100bdf28630SDarrick J. Wong struct xfs_buf *bp) 101bdf28630SDarrick J. Wong { 102bdf28630SDarrick J. Wong struct xfs_mount *mp = cur->bc_mp; 103576af732SDave Chinner struct xfs_buf *agbp = cur->bc_ag.agbp; 1049798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 105*04fcad80SDave Chinner xfs_fsblock_t fsbno = XFS_DADDR_TO_FSB(mp, xfs_buf_daddr(bp)); 10684d69619SDarrick J. Wong int error; 107bdf28630SDarrick J. Wong 10850f02fe3SDave Chinner trace_xfs_refcountbt_free_block(cur->bc_mp, cur->bc_ag.pag->pag_agno, 109bdf28630SDarrick J. Wong XFS_FSB_TO_AGBNO(cur->bc_mp, fsbno), 1); 110bdf28630SDarrick J. Wong be32_add_cpu(&agf->agf_refcount_blocks, -1); 111bdf28630SDarrick J. Wong xfs_alloc_log_agf(cur->bc_tp, agbp, XFS_AGF_REFCOUNT_BLOCKS); 1127280fedaSDarrick J. Wong error = xfs_free_extent(cur->bc_tp, fsbno, 1, &XFS_RMAP_OINFO_REFC, 11384d69619SDarrick J. Wong XFS_AG_RESV_METADATA); 11484d69619SDarrick J. Wong if (error) 11584d69619SDarrick J. Wong return error; 116bdf28630SDarrick J. Wong 11784d69619SDarrick J. Wong return error; 118bdf28630SDarrick J. Wong } 119bdf28630SDarrick J. Wong 120bdf28630SDarrick J. Wong STATIC int 121bdf28630SDarrick J. Wong xfs_refcountbt_get_minrecs( 122bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 123bdf28630SDarrick J. Wong int level) 124bdf28630SDarrick J. Wong { 125bdf28630SDarrick J. Wong return cur->bc_mp->m_refc_mnr[level != 0]; 126bdf28630SDarrick J. Wong } 127bdf28630SDarrick J. Wong 128bdf28630SDarrick J. Wong STATIC int 129bdf28630SDarrick J. Wong xfs_refcountbt_get_maxrecs( 130bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 131bdf28630SDarrick J. Wong int level) 132bdf28630SDarrick J. Wong { 133bdf28630SDarrick J. Wong return cur->bc_mp->m_refc_mxr[level != 0]; 134bdf28630SDarrick J. Wong } 135bdf28630SDarrick J. Wong 136bdf28630SDarrick J. Wong STATIC void 137bdf28630SDarrick J. Wong xfs_refcountbt_init_key_from_rec( 138bdf28630SDarrick J. Wong union xfs_btree_key *key, 13923825cd1SDarrick J. Wong const union xfs_btree_rec *rec) 140bdf28630SDarrick J. Wong { 141bdf28630SDarrick J. Wong key->refc.rc_startblock = rec->refc.rc_startblock; 142bdf28630SDarrick J. Wong } 143bdf28630SDarrick J. Wong 144bdf28630SDarrick J. Wong STATIC void 145bdf28630SDarrick J. Wong xfs_refcountbt_init_high_key_from_rec( 146bdf28630SDarrick J. Wong union xfs_btree_key *key, 14723825cd1SDarrick J. Wong const union xfs_btree_rec *rec) 148bdf28630SDarrick J. Wong { 149bdf28630SDarrick J. Wong __u32 x; 150bdf28630SDarrick J. Wong 151bdf28630SDarrick J. Wong x = be32_to_cpu(rec->refc.rc_startblock); 152bdf28630SDarrick J. Wong x += be32_to_cpu(rec->refc.rc_blockcount) - 1; 153bdf28630SDarrick J. Wong key->refc.rc_startblock = cpu_to_be32(x); 154bdf28630SDarrick J. Wong } 155bdf28630SDarrick J. Wong 156bdf28630SDarrick J. Wong STATIC void 157bdf28630SDarrick J. Wong xfs_refcountbt_init_rec_from_cur( 158bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 159bdf28630SDarrick J. Wong union xfs_btree_rec *rec) 160bdf28630SDarrick J. Wong { 161bdf28630SDarrick J. Wong rec->refc.rc_startblock = cpu_to_be32(cur->bc_rec.rc.rc_startblock); 162bdf28630SDarrick J. Wong rec->refc.rc_blockcount = cpu_to_be32(cur->bc_rec.rc.rc_blockcount); 163bdf28630SDarrick J. Wong rec->refc.rc_refcount = cpu_to_be32(cur->bc_rec.rc.rc_refcount); 164bdf28630SDarrick J. Wong } 165bdf28630SDarrick J. Wong 166bdf28630SDarrick J. Wong STATIC void 167bdf28630SDarrick J. Wong xfs_refcountbt_init_ptr_from_cur( 168bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 169bdf28630SDarrick J. Wong union xfs_btree_ptr *ptr) 170bdf28630SDarrick J. Wong { 171576af732SDave Chinner struct xfs_agf *agf = cur->bc_ag.agbp->b_addr; 172bdf28630SDarrick J. Wong 17350f02fe3SDave Chinner ASSERT(cur->bc_ag.pag->pag_agno == be32_to_cpu(agf->agf_seqno)); 174bdf28630SDarrick J. Wong 175bdf28630SDarrick J. Wong ptr->s = agf->agf_refcount_root; 176bdf28630SDarrick J. Wong } 177bdf28630SDarrick J. Wong 178c8ce540dSDarrick J. Wong STATIC int64_t 179bdf28630SDarrick J. Wong xfs_refcountbt_key_diff( 180bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 181d29d5577SDarrick J. Wong const union xfs_btree_key *key) 182bdf28630SDarrick J. Wong { 183bdf28630SDarrick J. Wong struct xfs_refcount_irec *rec = &cur->bc_rec.rc; 184d29d5577SDarrick J. Wong const struct xfs_refcount_key *kp = &key->refc; 185bdf28630SDarrick J. Wong 186c8ce540dSDarrick J. Wong return (int64_t)be32_to_cpu(kp->rc_startblock) - rec->rc_startblock; 187bdf28630SDarrick J. Wong } 188bdf28630SDarrick J. Wong 189c8ce540dSDarrick J. Wong STATIC int64_t 190bdf28630SDarrick J. Wong xfs_refcountbt_diff_two_keys( 191bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 192d29d5577SDarrick J. Wong const union xfs_btree_key *k1, 193d29d5577SDarrick J. Wong const union xfs_btree_key *k2) 194bdf28630SDarrick J. Wong { 195c8ce540dSDarrick J. Wong return (int64_t)be32_to_cpu(k1->refc.rc_startblock) - 196bdf28630SDarrick J. Wong be32_to_cpu(k2->refc.rc_startblock); 197bdf28630SDarrick J. Wong } 198bdf28630SDarrick J. Wong 199a6a781a5SDarrick J. Wong STATIC xfs_failaddr_t 2001946b91cSDarrick J. Wong xfs_refcountbt_verify( 2011946b91cSDarrick J. Wong struct xfs_buf *bp) 2021946b91cSDarrick J. Wong { 203dbd329f1SChristoph Hellwig struct xfs_mount *mp = bp->b_mount; 2041946b91cSDarrick J. Wong struct xfs_btree_block *block = XFS_BUF_TO_BLOCK(bp); 2051946b91cSDarrick J. Wong struct xfs_perag *pag = bp->b_pag; 206a6a781a5SDarrick J. Wong xfs_failaddr_t fa; 2071946b91cSDarrick J. Wong unsigned int level; 2081946b91cSDarrick J. Wong 20939708c20SBrian Foster if (!xfs_verify_magic(bp, block->bb_magic)) 210a6a781a5SDarrick J. Wong return __this_address; 2111946b91cSDarrick J. Wong 21238c26bfdSDave Chinner if (!xfs_has_reflink(mp)) 213a6a781a5SDarrick J. Wong return __this_address; 214a6a781a5SDarrick J. Wong fa = xfs_btree_sblock_v5hdr_verify(bp); 215a6a781a5SDarrick J. Wong if (fa) 216a6a781a5SDarrick J. Wong return fa; 2171946b91cSDarrick J. Wong 2181946b91cSDarrick J. Wong level = be16_to_cpu(block->bb_level); 2191946b91cSDarrick J. Wong if (pag && pag->pagf_init) { 2201946b91cSDarrick J. Wong if (level >= pag->pagf_refcount_level) 221a6a781a5SDarrick J. Wong return __this_address; 2221946b91cSDarrick J. Wong } else if (level >= mp->m_refc_maxlevels) 223a6a781a5SDarrick J. Wong return __this_address; 2241946b91cSDarrick J. Wong 2251946b91cSDarrick J. Wong return xfs_btree_sblock_verify(bp, mp->m_refc_mxr[level != 0]); 2261946b91cSDarrick J. Wong } 2271946b91cSDarrick J. Wong 2281946b91cSDarrick J. Wong STATIC void 2291946b91cSDarrick J. Wong xfs_refcountbt_read_verify( 2301946b91cSDarrick J. Wong struct xfs_buf *bp) 2311946b91cSDarrick J. Wong { 232bc1a09b8SDarrick J. Wong xfs_failaddr_t fa; 233bc1a09b8SDarrick J. Wong 2341946b91cSDarrick J. Wong if (!xfs_btree_sblock_verify_crc(bp)) 235bc1a09b8SDarrick J. Wong xfs_verifier_error(bp, -EFSBADCRC, __this_address); 236bc1a09b8SDarrick J. Wong else { 237bc1a09b8SDarrick J. Wong fa = xfs_refcountbt_verify(bp); 238bc1a09b8SDarrick J. Wong if (fa) 239bc1a09b8SDarrick J. Wong xfs_verifier_error(bp, -EFSCORRUPTED, fa); 240bc1a09b8SDarrick J. Wong } 2411946b91cSDarrick J. Wong 24231ca03c9SDarrick J. Wong if (bp->b_error) 2431946b91cSDarrick J. Wong trace_xfs_btree_corrupt(bp, _RET_IP_); 2441946b91cSDarrick J. Wong } 2451946b91cSDarrick J. Wong 2461946b91cSDarrick J. Wong STATIC void 2471946b91cSDarrick J. Wong xfs_refcountbt_write_verify( 2481946b91cSDarrick J. Wong struct xfs_buf *bp) 2491946b91cSDarrick J. Wong { 250bc1a09b8SDarrick J. Wong xfs_failaddr_t fa; 251bc1a09b8SDarrick J. Wong 252bc1a09b8SDarrick J. Wong fa = xfs_refcountbt_verify(bp); 253bc1a09b8SDarrick J. Wong if (fa) { 2541946b91cSDarrick J. Wong trace_xfs_btree_corrupt(bp, _RET_IP_); 255bc1a09b8SDarrick J. Wong xfs_verifier_error(bp, -EFSCORRUPTED, fa); 2561946b91cSDarrick J. Wong return; 2571946b91cSDarrick J. Wong } 2581946b91cSDarrick J. Wong xfs_btree_sblock_calc_crc(bp); 2591946b91cSDarrick J. Wong 2601946b91cSDarrick J. Wong } 2611946b91cSDarrick J. Wong 2621946b91cSDarrick J. Wong const struct xfs_buf_ops xfs_refcountbt_buf_ops = { 2631946b91cSDarrick J. Wong .name = "xfs_refcountbt", 26439708c20SBrian Foster .magic = { 0, cpu_to_be32(XFS_REFC_CRC_MAGIC) }, 2651946b91cSDarrick J. Wong .verify_read = xfs_refcountbt_read_verify, 2661946b91cSDarrick J. Wong .verify_write = xfs_refcountbt_write_verify, 267b5572597SDarrick J. Wong .verify_struct = xfs_refcountbt_verify, 2681946b91cSDarrick J. Wong }; 2691946b91cSDarrick J. Wong 270bdf28630SDarrick J. Wong STATIC int 271bdf28630SDarrick J. Wong xfs_refcountbt_keys_inorder( 272bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 2738e38dc88SDarrick J. Wong const union xfs_btree_key *k1, 2748e38dc88SDarrick J. Wong const union xfs_btree_key *k2) 275bdf28630SDarrick J. Wong { 276bdf28630SDarrick J. Wong return be32_to_cpu(k1->refc.rc_startblock) < 277bdf28630SDarrick J. Wong be32_to_cpu(k2->refc.rc_startblock); 278bdf28630SDarrick J. Wong } 279bdf28630SDarrick J. Wong 280bdf28630SDarrick J. Wong STATIC int 281bdf28630SDarrick J. Wong xfs_refcountbt_recs_inorder( 282bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 2838e38dc88SDarrick J. Wong const union xfs_btree_rec *r1, 2848e38dc88SDarrick J. Wong const union xfs_btree_rec *r2) 285bdf28630SDarrick J. Wong { 286bdf28630SDarrick J. Wong return be32_to_cpu(r1->refc.rc_startblock) + 287bdf28630SDarrick J. Wong be32_to_cpu(r1->refc.rc_blockcount) <= 288bdf28630SDarrick J. Wong be32_to_cpu(r2->refc.rc_startblock); 289bdf28630SDarrick J. Wong } 290bdf28630SDarrick J. Wong 2911946b91cSDarrick J. Wong static const struct xfs_btree_ops xfs_refcountbt_ops = { 2921946b91cSDarrick J. Wong .rec_len = sizeof(struct xfs_refcount_rec), 2931946b91cSDarrick J. Wong .key_len = sizeof(struct xfs_refcount_key), 2941946b91cSDarrick J. Wong 2951946b91cSDarrick J. Wong .dup_cursor = xfs_refcountbt_dup_cursor, 296bdf28630SDarrick J. Wong .set_root = xfs_refcountbt_set_root, 297bdf28630SDarrick J. Wong .alloc_block = xfs_refcountbt_alloc_block, 298bdf28630SDarrick J. Wong .free_block = xfs_refcountbt_free_block, 299bdf28630SDarrick J. Wong .get_minrecs = xfs_refcountbt_get_minrecs, 300bdf28630SDarrick J. Wong .get_maxrecs = xfs_refcountbt_get_maxrecs, 301bdf28630SDarrick J. Wong .init_key_from_rec = xfs_refcountbt_init_key_from_rec, 302bdf28630SDarrick J. Wong .init_high_key_from_rec = xfs_refcountbt_init_high_key_from_rec, 303bdf28630SDarrick J. Wong .init_rec_from_cur = xfs_refcountbt_init_rec_from_cur, 304bdf28630SDarrick J. Wong .init_ptr_from_cur = xfs_refcountbt_init_ptr_from_cur, 305bdf28630SDarrick J. Wong .key_diff = xfs_refcountbt_key_diff, 3061946b91cSDarrick J. Wong .buf_ops = &xfs_refcountbt_buf_ops, 307bdf28630SDarrick J. Wong .diff_two_keys = xfs_refcountbt_diff_two_keys, 308bdf28630SDarrick J. Wong .keys_inorder = xfs_refcountbt_keys_inorder, 309bdf28630SDarrick J. Wong .recs_inorder = xfs_refcountbt_recs_inorder, 3101946b91cSDarrick J. Wong }; 3111946b91cSDarrick J. Wong 3121946b91cSDarrick J. Wong /* 31356e98164SDarrick J. Wong * Initialize a new refcount btree cursor. 3141946b91cSDarrick J. Wong */ 31556e98164SDarrick J. Wong static struct xfs_btree_cur * 31656e98164SDarrick J. Wong xfs_refcountbt_init_common( 31756e98164SDarrick J. Wong struct xfs_mount *mp, 31856e98164SDarrick J. Wong struct xfs_trans *tp, 319be9fb17dSDave Chinner struct xfs_perag *pag) 32056e98164SDarrick J. Wong { 32156e98164SDarrick J. Wong struct xfs_btree_cur *cur; 32256e98164SDarrick J. Wong 323a81a0621SDave Chinner ASSERT(pag->pag_agno < mp->m_sb.sb_agcount); 32456e98164SDarrick J. Wong 32532a2b11fSCarlos Maiolino cur = kmem_cache_zalloc(xfs_btree_cur_zone, GFP_NOFS | __GFP_NOFAIL); 32656e98164SDarrick J. Wong cur->bc_tp = tp; 32756e98164SDarrick J. Wong cur->bc_mp = mp; 32856e98164SDarrick J. Wong cur->bc_btnum = XFS_BTNUM_REFC; 32956e98164SDarrick J. Wong cur->bc_blocklog = mp->m_sb.sb_blocklog; 33056e98164SDarrick J. Wong cur->bc_statoff = XFS_STATS_CALC_INDEX(xs_refcbt_2); 33156e98164SDarrick J. Wong 33256e98164SDarrick J. Wong cur->bc_flags |= XFS_BTREE_CRC_BLOCKS; 333a81a0621SDave Chinner 334be9fb17dSDave Chinner /* take a reference for the cursor */ 335be9fb17dSDave Chinner atomic_inc(&pag->pag_ref); 336be9fb17dSDave Chinner cur->bc_ag.pag = pag; 33756e98164SDarrick J. Wong 33856e98164SDarrick J. Wong cur->bc_ag.refc.nr_ops = 0; 33956e98164SDarrick J. Wong cur->bc_ag.refc.shape_changes = 0; 34056e98164SDarrick J. Wong cur->bc_ops = &xfs_refcountbt_ops; 34156e98164SDarrick J. Wong return cur; 34256e98164SDarrick J. Wong } 34356e98164SDarrick J. Wong 34456e98164SDarrick J. Wong /* Create a btree cursor. */ 3451946b91cSDarrick J. Wong struct xfs_btree_cur * 3461946b91cSDarrick J. Wong xfs_refcountbt_init_cursor( 3471946b91cSDarrick J. Wong struct xfs_mount *mp, 3481946b91cSDarrick J. Wong struct xfs_trans *tp, 3491946b91cSDarrick J. Wong struct xfs_buf *agbp, 350be9fb17dSDave Chinner struct xfs_perag *pag) 3511946b91cSDarrick J. Wong { 3529798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 3531946b91cSDarrick J. Wong struct xfs_btree_cur *cur; 3541946b91cSDarrick J. Wong 355a81a0621SDave Chinner cur = xfs_refcountbt_init_common(mp, tp, pag); 3561946b91cSDarrick J. Wong cur->bc_nlevels = be32_to_cpu(agf->agf_refcount_level); 357576af732SDave Chinner cur->bc_ag.agbp = agbp; 3581946b91cSDarrick J. Wong return cur; 3591946b91cSDarrick J. Wong } 3601946b91cSDarrick J. Wong 36156e98164SDarrick J. Wong /* Create a btree cursor with a fake root for staging. */ 36256e98164SDarrick J. Wong struct xfs_btree_cur * 36356e98164SDarrick J. Wong xfs_refcountbt_stage_cursor( 36456e98164SDarrick J. Wong struct xfs_mount *mp, 36556e98164SDarrick J. Wong struct xbtree_afakeroot *afake, 366a81a0621SDave Chinner struct xfs_perag *pag) 36756e98164SDarrick J. Wong { 36856e98164SDarrick J. Wong struct xfs_btree_cur *cur; 36956e98164SDarrick J. Wong 370a81a0621SDave Chinner cur = xfs_refcountbt_init_common(mp, NULL, pag); 37156e98164SDarrick J. Wong xfs_btree_stage_afakeroot(cur, afake); 37256e98164SDarrick J. Wong return cur; 37356e98164SDarrick J. Wong } 37456e98164SDarrick J. Wong 37556e98164SDarrick J. Wong /* 37656e98164SDarrick J. Wong * Swap in the new btree root. Once we pass this point the newly rebuilt btree 37756e98164SDarrick J. Wong * is in place and we have to kill off all the old btree blocks. 37856e98164SDarrick J. Wong */ 37956e98164SDarrick J. Wong void 38056e98164SDarrick J. Wong xfs_refcountbt_commit_staged_btree( 38156e98164SDarrick J. Wong struct xfs_btree_cur *cur, 38256e98164SDarrick J. Wong struct xfs_trans *tp, 38356e98164SDarrick J. Wong struct xfs_buf *agbp) 38456e98164SDarrick J. Wong { 38556e98164SDarrick J. Wong struct xfs_agf *agf = agbp->b_addr; 38656e98164SDarrick J. Wong struct xbtree_afakeroot *afake = cur->bc_ag.afake; 38756e98164SDarrick J. Wong 38856e98164SDarrick J. Wong ASSERT(cur->bc_flags & XFS_BTREE_STAGING); 38956e98164SDarrick J. Wong 39056e98164SDarrick J. Wong agf->agf_refcount_root = cpu_to_be32(afake->af_root); 39156e98164SDarrick J. Wong agf->agf_refcount_level = cpu_to_be32(afake->af_levels); 39256e98164SDarrick J. Wong agf->agf_refcount_blocks = cpu_to_be32(afake->af_blocks); 39356e98164SDarrick J. Wong xfs_alloc_log_agf(tp, agbp, XFS_AGF_REFCOUNT_BLOCKS | 39456e98164SDarrick J. Wong XFS_AGF_REFCOUNT_ROOT | 39556e98164SDarrick J. Wong XFS_AGF_REFCOUNT_LEVEL); 39656e98164SDarrick J. Wong xfs_btree_commit_afakeroot(cur, tp, agbp, &xfs_refcountbt_ops); 39756e98164SDarrick J. Wong } 39856e98164SDarrick J. Wong 3991946b91cSDarrick J. Wong /* 4001946b91cSDarrick J. Wong * Calculate the number of records in a refcount btree block. 4011946b91cSDarrick J. Wong */ 4021946b91cSDarrick J. Wong int 4031946b91cSDarrick J. Wong xfs_refcountbt_maxrecs( 4041946b91cSDarrick J. Wong int blocklen, 4051946b91cSDarrick J. Wong bool leaf) 4061946b91cSDarrick J. Wong { 4071946b91cSDarrick J. Wong blocklen -= XFS_REFCOUNT_BLOCK_LEN; 4081946b91cSDarrick J. Wong 4091946b91cSDarrick J. Wong if (leaf) 4101946b91cSDarrick J. Wong return blocklen / sizeof(struct xfs_refcount_rec); 4111946b91cSDarrick J. Wong return blocklen / (sizeof(struct xfs_refcount_key) + 4121946b91cSDarrick J. Wong sizeof(xfs_refcount_ptr_t)); 4131946b91cSDarrick J. Wong } 4141946b91cSDarrick J. Wong 4151946b91cSDarrick J. Wong /* Compute the maximum height of a refcount btree. */ 4161946b91cSDarrick J. Wong void 4171946b91cSDarrick J. Wong xfs_refcountbt_compute_maxlevels( 4181946b91cSDarrick J. Wong struct xfs_mount *mp) 4191946b91cSDarrick J. Wong { 420a1f69417SEric Sandeen mp->m_refc_maxlevels = xfs_btree_compute_maxlevels( 4211946b91cSDarrick J. Wong mp->m_refc_mnr, mp->m_sb.sb_agblocks); 4221946b91cSDarrick J. Wong } 42384d69619SDarrick J. Wong 42484d69619SDarrick J. Wong /* Calculate the refcount btree size for some records. */ 42584d69619SDarrick J. Wong xfs_extlen_t 42684d69619SDarrick J. Wong xfs_refcountbt_calc_size( 42784d69619SDarrick J. Wong struct xfs_mount *mp, 42884d69619SDarrick J. Wong unsigned long long len) 42984d69619SDarrick J. Wong { 430a1f69417SEric Sandeen return xfs_btree_calc_size(mp->m_refc_mnr, len); 43184d69619SDarrick J. Wong } 43284d69619SDarrick J. Wong 43384d69619SDarrick J. Wong /* 43484d69619SDarrick J. Wong * Calculate the maximum refcount btree size. 43584d69619SDarrick J. Wong */ 43684d69619SDarrick J. Wong xfs_extlen_t 43784d69619SDarrick J. Wong xfs_refcountbt_max_size( 43820e73b00SDarrick J. Wong struct xfs_mount *mp, 43920e73b00SDarrick J. Wong xfs_agblock_t agblocks) 44084d69619SDarrick J. Wong { 44184d69619SDarrick J. Wong /* Bail out if we're uninitialized, which can happen in mkfs. */ 44284d69619SDarrick J. Wong if (mp->m_refc_mxr[0] == 0) 44384d69619SDarrick J. Wong return 0; 44484d69619SDarrick J. Wong 44520e73b00SDarrick J. Wong return xfs_refcountbt_calc_size(mp, agblocks); 44684d69619SDarrick J. Wong } 44784d69619SDarrick J. Wong 44884d69619SDarrick J. Wong /* 44984d69619SDarrick J. Wong * Figure out how many blocks to reserve and how many are used by this btree. 45084d69619SDarrick J. Wong */ 45184d69619SDarrick J. Wong int 45284d69619SDarrick J. Wong xfs_refcountbt_calc_reserves( 45384d69619SDarrick J. Wong struct xfs_mount *mp, 454ebcbef3aSDarrick J. Wong struct xfs_trans *tp, 45530933120SDave Chinner struct xfs_perag *pag, 45684d69619SDarrick J. Wong xfs_extlen_t *ask, 45784d69619SDarrick J. Wong xfs_extlen_t *used) 45884d69619SDarrick J. Wong { 45984d69619SDarrick J. Wong struct xfs_buf *agbp; 46084d69619SDarrick J. Wong struct xfs_agf *agf; 46120e73b00SDarrick J. Wong xfs_agblock_t agblocks; 46284d69619SDarrick J. Wong xfs_extlen_t tree_len; 46384d69619SDarrick J. Wong int error; 46484d69619SDarrick J. Wong 46538c26bfdSDave Chinner if (!xfs_has_reflink(mp)) 46684d69619SDarrick J. Wong return 0; 46784d69619SDarrick J. Wong 46830933120SDave Chinner error = xfs_alloc_read_agf(mp, tp, pag->pag_agno, 0, &agbp); 46984d69619SDarrick J. Wong if (error) 47084d69619SDarrick J. Wong return error; 47184d69619SDarrick J. Wong 4729798f615SChristoph Hellwig agf = agbp->b_addr; 47320e73b00SDarrick J. Wong agblocks = be32_to_cpu(agf->agf_length); 47484d69619SDarrick J. Wong tree_len = be32_to_cpu(agf->agf_refcount_blocks); 475ebcbef3aSDarrick J. Wong xfs_trans_brelse(tp, agbp); 47684d69619SDarrick J. Wong 4775cd213b0SDarrick J. Wong /* 4785cd213b0SDarrick J. Wong * The log is permanently allocated, so the space it occupies will 4795cd213b0SDarrick J. Wong * never be available for the kinds of things that would require btree 4805cd213b0SDarrick J. Wong * expansion. We therefore can pretend the space isn't there. 4815cd213b0SDarrick J. Wong */ 4825cd213b0SDarrick J. Wong if (mp->m_sb.sb_logstart && 48330933120SDave Chinner XFS_FSB_TO_AGNO(mp, mp->m_sb.sb_logstart) == pag->pag_agno) 4845cd213b0SDarrick J. Wong agblocks -= mp->m_sb.sb_logblocks; 4855cd213b0SDarrick J. Wong 48620e73b00SDarrick J. Wong *ask += xfs_refcountbt_max_size(mp, agblocks); 48784d69619SDarrick J. Wong *used += tree_len; 48884d69619SDarrick J. Wong 48984d69619SDarrick J. Wong return error; 49084d69619SDarrick J. Wong } 491