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" 169a50ee4fSDarrick J. Wong #include "xfs_refcount.h" 171946b91cSDarrick J. Wong #include "xfs_alloc.h" 181946b91cSDarrick J. Wong #include "xfs_error.h" 191946b91cSDarrick J. Wong #include "xfs_trace.h" 201946b91cSDarrick J. Wong #include "xfs_trans.h" 211946b91cSDarrick J. Wong #include "xfs_bit.h" 22bdf28630SDarrick J. Wong #include "xfs_rmap.h" 239bbafc71SDave Chinner #include "xfs_ag.h" 241946b91cSDarrick J. Wong 25e7720afaSDarrick J. Wong static struct kmem_cache *xfs_refcountbt_cur_cache; 269fa47bdcSDarrick J. Wong 271946b91cSDarrick J. Wong static struct xfs_btree_cur * 281946b91cSDarrick J. Wong xfs_refcountbt_dup_cursor( 291946b91cSDarrick J. Wong struct xfs_btree_cur *cur) 301946b91cSDarrick J. Wong { 311946b91cSDarrick J. Wong return xfs_refcountbt_init_cursor(cur->bc_mp, cur->bc_tp, 32a81a0621SDave Chinner cur->bc_ag.agbp, cur->bc_ag.pag); 331946b91cSDarrick J. Wong } 341946b91cSDarrick J. Wong 35bdf28630SDarrick J. Wong STATIC void 36bdf28630SDarrick J. Wong xfs_refcountbt_set_root( 37bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 38b5a6e5feSDarrick J. Wong const union xfs_btree_ptr *ptr, 39bdf28630SDarrick J. Wong int inc) 40bdf28630SDarrick J. Wong { 41576af732SDave Chinner struct xfs_buf *agbp = cur->bc_ag.agbp; 429798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 4392a00544SGao Xiang struct xfs_perag *pag = agbp->b_pag; 44bdf28630SDarrick J. Wong 45bdf28630SDarrick J. Wong ASSERT(ptr->s != 0); 46bdf28630SDarrick J. Wong 47bdf28630SDarrick J. Wong agf->agf_refcount_root = ptr->s; 48bdf28630SDarrick J. Wong be32_add_cpu(&agf->agf_refcount_level, inc); 49bdf28630SDarrick J. Wong pag->pagf_refcount_level += inc; 50bdf28630SDarrick J. Wong 51bdf28630SDarrick J. Wong xfs_alloc_log_agf(cur->bc_tp, agbp, 52bdf28630SDarrick J. Wong XFS_AGF_REFCOUNT_ROOT | XFS_AGF_REFCOUNT_LEVEL); 53bdf28630SDarrick J. Wong } 54bdf28630SDarrick J. Wong 55bdf28630SDarrick J. Wong STATIC int 56bdf28630SDarrick J. Wong xfs_refcountbt_alloc_block( 57bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 58deb06b9aSDarrick J. Wong const union xfs_btree_ptr *start, 59bdf28630SDarrick J. Wong union xfs_btree_ptr *new, 60bdf28630SDarrick J. Wong int *stat) 61bdf28630SDarrick J. Wong { 62576af732SDave Chinner struct xfs_buf *agbp = cur->bc_ag.agbp; 639798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 64bdf28630SDarrick J. Wong struct xfs_alloc_arg args; /* block allocation args */ 65bdf28630SDarrick J. Wong int error; /* error return value */ 66bdf28630SDarrick J. Wong 67bdf28630SDarrick J. Wong memset(&args, 0, sizeof(args)); 68bdf28630SDarrick J. Wong args.tp = cur->bc_tp; 69bdf28630SDarrick J. Wong args.mp = cur->bc_mp; 7074c36a86SDave Chinner args.pag = cur->bc_ag.pag; 717280fedaSDarrick J. Wong args.oinfo = XFS_RMAP_OINFO_REFC; 72bdf28630SDarrick J. Wong args.minlen = args.maxlen = args.prod = 1; 7384d69619SDarrick J. Wong args.resv = XFS_AG_RESV_METADATA; 74bdf28630SDarrick J. Wong 75*db4710fdSDave Chinner error = xfs_alloc_vextent_near_bno(&args, 76*db4710fdSDave Chinner XFS_AGB_TO_FSB(args.mp, args.pag->pag_agno, 77*db4710fdSDave Chinner xfs_refc_block(args.mp))); 78bdf28630SDarrick J. Wong if (error) 79bdf28630SDarrick J. Wong goto out_error; 8050f02fe3SDave Chinner trace_xfs_refcountbt_alloc_block(cur->bc_mp, cur->bc_ag.pag->pag_agno, 81bdf28630SDarrick J. Wong args.agbno, 1); 82bdf28630SDarrick J. Wong if (args.fsbno == NULLFSBLOCK) { 83bdf28630SDarrick J. Wong *stat = 0; 84bdf28630SDarrick J. Wong return 0; 85bdf28630SDarrick J. Wong } 8650f02fe3SDave Chinner ASSERT(args.agno == cur->bc_ag.pag->pag_agno); 87bdf28630SDarrick J. Wong ASSERT(args.len == 1); 88bdf28630SDarrick J. Wong 89bdf28630SDarrick J. Wong new->s = cpu_to_be32(args.agbno); 90bdf28630SDarrick J. Wong be32_add_cpu(&agf->agf_refcount_blocks, 1); 91bdf28630SDarrick J. Wong xfs_alloc_log_agf(cur->bc_tp, agbp, XFS_AGF_REFCOUNT_BLOCKS); 92bdf28630SDarrick J. Wong 93bdf28630SDarrick J. Wong *stat = 1; 94bdf28630SDarrick J. Wong return 0; 95bdf28630SDarrick J. Wong 96bdf28630SDarrick J. Wong out_error: 97bdf28630SDarrick J. Wong return error; 98bdf28630SDarrick J. Wong } 99bdf28630SDarrick J. Wong 100bdf28630SDarrick J. Wong STATIC int 101bdf28630SDarrick J. Wong xfs_refcountbt_free_block( 102bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 103bdf28630SDarrick J. Wong struct xfs_buf *bp) 104bdf28630SDarrick J. Wong { 105bdf28630SDarrick J. Wong struct xfs_mount *mp = cur->bc_mp; 106576af732SDave Chinner struct xfs_buf *agbp = cur->bc_ag.agbp; 1079798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 10804fcad80SDave Chinner xfs_fsblock_t fsbno = XFS_DADDR_TO_FSB(mp, xfs_buf_daddr(bp)); 10984d69619SDarrick J. Wong int error; 110bdf28630SDarrick J. Wong 11150f02fe3SDave Chinner trace_xfs_refcountbt_free_block(cur->bc_mp, cur->bc_ag.pag->pag_agno, 112bdf28630SDarrick J. Wong XFS_FSB_TO_AGBNO(cur->bc_mp, fsbno), 1); 113bdf28630SDarrick J. Wong be32_add_cpu(&agf->agf_refcount_blocks, -1); 114bdf28630SDarrick J. Wong xfs_alloc_log_agf(cur->bc_tp, agbp, XFS_AGF_REFCOUNT_BLOCKS); 1157280fedaSDarrick J. Wong error = xfs_free_extent(cur->bc_tp, fsbno, 1, &XFS_RMAP_OINFO_REFC, 11684d69619SDarrick J. Wong XFS_AG_RESV_METADATA); 11784d69619SDarrick J. Wong if (error) 11884d69619SDarrick J. Wong return error; 119bdf28630SDarrick J. Wong 12084d69619SDarrick J. Wong return error; 121bdf28630SDarrick J. Wong } 122bdf28630SDarrick J. Wong 123bdf28630SDarrick J. Wong STATIC int 124bdf28630SDarrick J. Wong xfs_refcountbt_get_minrecs( 125bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 126bdf28630SDarrick J. Wong int level) 127bdf28630SDarrick J. Wong { 128bdf28630SDarrick J. Wong return cur->bc_mp->m_refc_mnr[level != 0]; 129bdf28630SDarrick J. Wong } 130bdf28630SDarrick J. Wong 131bdf28630SDarrick J. Wong STATIC int 132bdf28630SDarrick J. Wong xfs_refcountbt_get_maxrecs( 133bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 134bdf28630SDarrick J. Wong int level) 135bdf28630SDarrick J. Wong { 136bdf28630SDarrick J. Wong return cur->bc_mp->m_refc_mxr[level != 0]; 137bdf28630SDarrick J. Wong } 138bdf28630SDarrick J. Wong 139bdf28630SDarrick J. Wong STATIC void 140bdf28630SDarrick J. Wong xfs_refcountbt_init_key_from_rec( 141bdf28630SDarrick J. Wong union xfs_btree_key *key, 14223825cd1SDarrick J. Wong const union xfs_btree_rec *rec) 143bdf28630SDarrick J. Wong { 144bdf28630SDarrick J. Wong key->refc.rc_startblock = rec->refc.rc_startblock; 145bdf28630SDarrick J. Wong } 146bdf28630SDarrick J. Wong 147bdf28630SDarrick J. Wong STATIC void 148bdf28630SDarrick J. Wong xfs_refcountbt_init_high_key_from_rec( 149bdf28630SDarrick J. Wong union xfs_btree_key *key, 15023825cd1SDarrick J. Wong const union xfs_btree_rec *rec) 151bdf28630SDarrick J. Wong { 152bdf28630SDarrick J. Wong __u32 x; 153bdf28630SDarrick J. Wong 154bdf28630SDarrick J. Wong x = be32_to_cpu(rec->refc.rc_startblock); 155bdf28630SDarrick J. Wong x += be32_to_cpu(rec->refc.rc_blockcount) - 1; 156bdf28630SDarrick J. Wong key->refc.rc_startblock = cpu_to_be32(x); 157bdf28630SDarrick J. Wong } 158bdf28630SDarrick J. Wong 159bdf28630SDarrick J. Wong STATIC void 160bdf28630SDarrick J. Wong xfs_refcountbt_init_rec_from_cur( 161bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 162bdf28630SDarrick J. Wong union xfs_btree_rec *rec) 163bdf28630SDarrick J. Wong { 1649a50ee4fSDarrick J. Wong const struct xfs_refcount_irec *irec = &cur->bc_rec.rc; 1659a50ee4fSDarrick J. Wong uint32_t start; 1669a50ee4fSDarrick J. Wong 1679a50ee4fSDarrick J. Wong start = xfs_refcount_encode_startblock(irec->rc_startblock, 1689a50ee4fSDarrick J. Wong irec->rc_domain); 1699a50ee4fSDarrick J. Wong rec->refc.rc_startblock = cpu_to_be32(start); 170bdf28630SDarrick J. Wong rec->refc.rc_blockcount = cpu_to_be32(cur->bc_rec.rc.rc_blockcount); 171bdf28630SDarrick J. Wong rec->refc.rc_refcount = cpu_to_be32(cur->bc_rec.rc.rc_refcount); 172bdf28630SDarrick J. Wong } 173bdf28630SDarrick J. Wong 174bdf28630SDarrick J. Wong STATIC void 175bdf28630SDarrick J. Wong xfs_refcountbt_init_ptr_from_cur( 176bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 177bdf28630SDarrick J. Wong union xfs_btree_ptr *ptr) 178bdf28630SDarrick J. Wong { 179576af732SDave Chinner struct xfs_agf *agf = cur->bc_ag.agbp->b_addr; 180bdf28630SDarrick J. Wong 18150f02fe3SDave Chinner ASSERT(cur->bc_ag.pag->pag_agno == be32_to_cpu(agf->agf_seqno)); 182bdf28630SDarrick J. Wong 183bdf28630SDarrick J. Wong ptr->s = agf->agf_refcount_root; 184bdf28630SDarrick J. Wong } 185bdf28630SDarrick J. Wong 186c8ce540dSDarrick J. Wong STATIC int64_t 187bdf28630SDarrick J. Wong xfs_refcountbt_key_diff( 188bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 189d29d5577SDarrick J. Wong const union xfs_btree_key *key) 190bdf28630SDarrick J. Wong { 191d29d5577SDarrick J. Wong const struct xfs_refcount_key *kp = &key->refc; 1929a50ee4fSDarrick J. Wong const struct xfs_refcount_irec *irec = &cur->bc_rec.rc; 1939a50ee4fSDarrick J. Wong uint32_t start; 194bdf28630SDarrick J. Wong 1959a50ee4fSDarrick J. Wong start = xfs_refcount_encode_startblock(irec->rc_startblock, 1969a50ee4fSDarrick J. Wong irec->rc_domain); 1979a50ee4fSDarrick J. Wong return (int64_t)be32_to_cpu(kp->rc_startblock) - start; 198bdf28630SDarrick J. Wong } 199bdf28630SDarrick J. Wong 200c8ce540dSDarrick J. Wong STATIC int64_t 201bdf28630SDarrick J. Wong xfs_refcountbt_diff_two_keys( 202bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 203d29d5577SDarrick J. Wong const union xfs_btree_key *k1, 204d29d5577SDarrick J. Wong const union xfs_btree_key *k2) 205bdf28630SDarrick J. Wong { 206c8ce540dSDarrick J. Wong return (int64_t)be32_to_cpu(k1->refc.rc_startblock) - 207bdf28630SDarrick J. Wong be32_to_cpu(k2->refc.rc_startblock); 208bdf28630SDarrick J. Wong } 209bdf28630SDarrick J. Wong 210a6a781a5SDarrick J. Wong STATIC xfs_failaddr_t 2111946b91cSDarrick J. Wong xfs_refcountbt_verify( 2121946b91cSDarrick J. Wong struct xfs_buf *bp) 2131946b91cSDarrick J. Wong { 214dbd329f1SChristoph Hellwig struct xfs_mount *mp = bp->b_mount; 2151946b91cSDarrick J. Wong struct xfs_btree_block *block = XFS_BUF_TO_BLOCK(bp); 2161946b91cSDarrick J. Wong struct xfs_perag *pag = bp->b_pag; 217a6a781a5SDarrick J. Wong xfs_failaddr_t fa; 2181946b91cSDarrick J. Wong unsigned int level; 2191946b91cSDarrick J. Wong 22039708c20SBrian Foster if (!xfs_verify_magic(bp, block->bb_magic)) 221a6a781a5SDarrick J. Wong return __this_address; 2221946b91cSDarrick J. Wong 22338c26bfdSDave Chinner if (!xfs_has_reflink(mp)) 224a6a781a5SDarrick J. Wong return __this_address; 225a6a781a5SDarrick J. Wong fa = xfs_btree_sblock_v5hdr_verify(bp); 226a6a781a5SDarrick J. Wong if (fa) 227a6a781a5SDarrick J. Wong return fa; 2281946b91cSDarrick J. Wong 2291946b91cSDarrick J. Wong level = be16_to_cpu(block->bb_level); 2307ac2ff8bSDave Chinner if (pag && xfs_perag_initialised_agf(pag)) { 2311946b91cSDarrick J. Wong if (level >= pag->pagf_refcount_level) 232a6a781a5SDarrick J. Wong return __this_address; 2331946b91cSDarrick J. Wong } else if (level >= mp->m_refc_maxlevels) 234a6a781a5SDarrick J. Wong return __this_address; 2351946b91cSDarrick J. Wong 2361946b91cSDarrick J. Wong return xfs_btree_sblock_verify(bp, mp->m_refc_mxr[level != 0]); 2371946b91cSDarrick J. Wong } 2381946b91cSDarrick J. Wong 2391946b91cSDarrick J. Wong STATIC void 2401946b91cSDarrick J. Wong xfs_refcountbt_read_verify( 2411946b91cSDarrick J. Wong struct xfs_buf *bp) 2421946b91cSDarrick J. Wong { 243bc1a09b8SDarrick J. Wong xfs_failaddr_t fa; 244bc1a09b8SDarrick J. Wong 2451946b91cSDarrick J. Wong if (!xfs_btree_sblock_verify_crc(bp)) 246bc1a09b8SDarrick J. Wong xfs_verifier_error(bp, -EFSBADCRC, __this_address); 247bc1a09b8SDarrick J. Wong else { 248bc1a09b8SDarrick J. Wong fa = xfs_refcountbt_verify(bp); 249bc1a09b8SDarrick J. Wong if (fa) 250bc1a09b8SDarrick J. Wong xfs_verifier_error(bp, -EFSCORRUPTED, fa); 251bc1a09b8SDarrick J. Wong } 2521946b91cSDarrick J. Wong 25331ca03c9SDarrick J. Wong if (bp->b_error) 2541946b91cSDarrick J. Wong trace_xfs_btree_corrupt(bp, _RET_IP_); 2551946b91cSDarrick J. Wong } 2561946b91cSDarrick J. Wong 2571946b91cSDarrick J. Wong STATIC void 2581946b91cSDarrick J. Wong xfs_refcountbt_write_verify( 2591946b91cSDarrick J. Wong struct xfs_buf *bp) 2601946b91cSDarrick J. Wong { 261bc1a09b8SDarrick J. Wong xfs_failaddr_t fa; 262bc1a09b8SDarrick J. Wong 263bc1a09b8SDarrick J. Wong fa = xfs_refcountbt_verify(bp); 264bc1a09b8SDarrick J. Wong if (fa) { 2651946b91cSDarrick J. Wong trace_xfs_btree_corrupt(bp, _RET_IP_); 266bc1a09b8SDarrick J. Wong xfs_verifier_error(bp, -EFSCORRUPTED, fa); 2671946b91cSDarrick J. Wong return; 2681946b91cSDarrick J. Wong } 2691946b91cSDarrick J. Wong xfs_btree_sblock_calc_crc(bp); 2701946b91cSDarrick J. Wong 2711946b91cSDarrick J. Wong } 2721946b91cSDarrick J. Wong 2731946b91cSDarrick J. Wong const struct xfs_buf_ops xfs_refcountbt_buf_ops = { 2741946b91cSDarrick J. Wong .name = "xfs_refcountbt", 27539708c20SBrian Foster .magic = { 0, cpu_to_be32(XFS_REFC_CRC_MAGIC) }, 2761946b91cSDarrick J. Wong .verify_read = xfs_refcountbt_read_verify, 2771946b91cSDarrick J. Wong .verify_write = xfs_refcountbt_write_verify, 278b5572597SDarrick J. Wong .verify_struct = xfs_refcountbt_verify, 2791946b91cSDarrick J. Wong }; 2801946b91cSDarrick J. Wong 281bdf28630SDarrick J. Wong STATIC int 282bdf28630SDarrick J. Wong xfs_refcountbt_keys_inorder( 283bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 2848e38dc88SDarrick J. Wong const union xfs_btree_key *k1, 2858e38dc88SDarrick J. Wong const union xfs_btree_key *k2) 286bdf28630SDarrick J. Wong { 287bdf28630SDarrick J. Wong return be32_to_cpu(k1->refc.rc_startblock) < 288bdf28630SDarrick J. Wong be32_to_cpu(k2->refc.rc_startblock); 289bdf28630SDarrick J. Wong } 290bdf28630SDarrick J. Wong 291bdf28630SDarrick J. Wong STATIC int 292bdf28630SDarrick J. Wong xfs_refcountbt_recs_inorder( 293bdf28630SDarrick J. Wong struct xfs_btree_cur *cur, 2948e38dc88SDarrick J. Wong const union xfs_btree_rec *r1, 2958e38dc88SDarrick J. Wong const union xfs_btree_rec *r2) 296bdf28630SDarrick J. Wong { 297bdf28630SDarrick J. Wong return be32_to_cpu(r1->refc.rc_startblock) + 298bdf28630SDarrick J. Wong be32_to_cpu(r1->refc.rc_blockcount) <= 299bdf28630SDarrick J. Wong be32_to_cpu(r2->refc.rc_startblock); 300bdf28630SDarrick J. Wong } 301bdf28630SDarrick J. Wong 3021946b91cSDarrick J. Wong static const struct xfs_btree_ops xfs_refcountbt_ops = { 3031946b91cSDarrick J. Wong .rec_len = sizeof(struct xfs_refcount_rec), 3041946b91cSDarrick J. Wong .key_len = sizeof(struct xfs_refcount_key), 3051946b91cSDarrick J. Wong 3061946b91cSDarrick J. Wong .dup_cursor = xfs_refcountbt_dup_cursor, 307bdf28630SDarrick J. Wong .set_root = xfs_refcountbt_set_root, 308bdf28630SDarrick J. Wong .alloc_block = xfs_refcountbt_alloc_block, 309bdf28630SDarrick J. Wong .free_block = xfs_refcountbt_free_block, 310bdf28630SDarrick J. Wong .get_minrecs = xfs_refcountbt_get_minrecs, 311bdf28630SDarrick J. Wong .get_maxrecs = xfs_refcountbt_get_maxrecs, 312bdf28630SDarrick J. Wong .init_key_from_rec = xfs_refcountbt_init_key_from_rec, 313bdf28630SDarrick J. Wong .init_high_key_from_rec = xfs_refcountbt_init_high_key_from_rec, 314bdf28630SDarrick J. Wong .init_rec_from_cur = xfs_refcountbt_init_rec_from_cur, 315bdf28630SDarrick J. Wong .init_ptr_from_cur = xfs_refcountbt_init_ptr_from_cur, 316bdf28630SDarrick J. Wong .key_diff = xfs_refcountbt_key_diff, 3171946b91cSDarrick J. Wong .buf_ops = &xfs_refcountbt_buf_ops, 318bdf28630SDarrick J. Wong .diff_two_keys = xfs_refcountbt_diff_two_keys, 319bdf28630SDarrick J. Wong .keys_inorder = xfs_refcountbt_keys_inorder, 320bdf28630SDarrick J. Wong .recs_inorder = xfs_refcountbt_recs_inorder, 3211946b91cSDarrick J. Wong }; 3221946b91cSDarrick J. Wong 3231946b91cSDarrick J. Wong /* 32456e98164SDarrick J. Wong * Initialize a new refcount btree cursor. 3251946b91cSDarrick J. Wong */ 32656e98164SDarrick J. Wong static struct xfs_btree_cur * 32756e98164SDarrick J. Wong xfs_refcountbt_init_common( 32856e98164SDarrick J. Wong struct xfs_mount *mp, 32956e98164SDarrick J. Wong struct xfs_trans *tp, 330be9fb17dSDave Chinner struct xfs_perag *pag) 33156e98164SDarrick J. Wong { 33256e98164SDarrick J. Wong struct xfs_btree_cur *cur; 33356e98164SDarrick J. Wong 334a81a0621SDave Chinner ASSERT(pag->pag_agno < mp->m_sb.sb_agcount); 33556e98164SDarrick J. Wong 336c940a0c5SDarrick J. Wong cur = xfs_btree_alloc_cursor(mp, tp, XFS_BTNUM_REFC, 3379fa47bdcSDarrick J. Wong mp->m_refc_maxlevels, xfs_refcountbt_cur_cache); 33856e98164SDarrick J. Wong cur->bc_statoff = XFS_STATS_CALC_INDEX(xs_refcbt_2); 33956e98164SDarrick J. Wong 34056e98164SDarrick J. Wong cur->bc_flags |= XFS_BTREE_CRC_BLOCKS; 341a81a0621SDave Chinner 342be9fb17dSDave Chinner /* take a reference for the cursor */ 343be9fb17dSDave Chinner atomic_inc(&pag->pag_ref); 344be9fb17dSDave Chinner cur->bc_ag.pag = pag; 34556e98164SDarrick J. Wong 34656e98164SDarrick J. Wong cur->bc_ag.refc.nr_ops = 0; 34756e98164SDarrick J. Wong cur->bc_ag.refc.shape_changes = 0; 34856e98164SDarrick J. Wong cur->bc_ops = &xfs_refcountbt_ops; 34956e98164SDarrick J. Wong return cur; 35056e98164SDarrick J. Wong } 35156e98164SDarrick J. Wong 35256e98164SDarrick J. Wong /* Create a btree cursor. */ 3531946b91cSDarrick J. Wong struct xfs_btree_cur * 3541946b91cSDarrick J. Wong xfs_refcountbt_init_cursor( 3551946b91cSDarrick J. Wong struct xfs_mount *mp, 3561946b91cSDarrick J. Wong struct xfs_trans *tp, 3571946b91cSDarrick J. Wong struct xfs_buf *agbp, 358be9fb17dSDave Chinner struct xfs_perag *pag) 3591946b91cSDarrick J. Wong { 3609798f615SChristoph Hellwig struct xfs_agf *agf = agbp->b_addr; 3611946b91cSDarrick J. Wong struct xfs_btree_cur *cur; 3621946b91cSDarrick J. Wong 363a81a0621SDave Chinner cur = xfs_refcountbt_init_common(mp, tp, pag); 3641946b91cSDarrick J. Wong cur->bc_nlevels = be32_to_cpu(agf->agf_refcount_level); 365576af732SDave Chinner cur->bc_ag.agbp = agbp; 3661946b91cSDarrick J. Wong return cur; 3671946b91cSDarrick J. Wong } 3681946b91cSDarrick J. Wong 36956e98164SDarrick J. Wong /* Create a btree cursor with a fake root for staging. */ 37056e98164SDarrick J. Wong struct xfs_btree_cur * 37156e98164SDarrick J. Wong xfs_refcountbt_stage_cursor( 37256e98164SDarrick J. Wong struct xfs_mount *mp, 37356e98164SDarrick J. Wong struct xbtree_afakeroot *afake, 374a81a0621SDave Chinner struct xfs_perag *pag) 37556e98164SDarrick J. Wong { 37656e98164SDarrick J. Wong struct xfs_btree_cur *cur; 37756e98164SDarrick J. Wong 378a81a0621SDave Chinner cur = xfs_refcountbt_init_common(mp, NULL, pag); 37956e98164SDarrick J. Wong xfs_btree_stage_afakeroot(cur, afake); 38056e98164SDarrick J. Wong return cur; 38156e98164SDarrick J. Wong } 38256e98164SDarrick J. Wong 38356e98164SDarrick J. Wong /* 38456e98164SDarrick J. Wong * Swap in the new btree root. Once we pass this point the newly rebuilt btree 38556e98164SDarrick J. Wong * is in place and we have to kill off all the old btree blocks. 38656e98164SDarrick J. Wong */ 38756e98164SDarrick J. Wong void 38856e98164SDarrick J. Wong xfs_refcountbt_commit_staged_btree( 38956e98164SDarrick J. Wong struct xfs_btree_cur *cur, 39056e98164SDarrick J. Wong struct xfs_trans *tp, 39156e98164SDarrick J. Wong struct xfs_buf *agbp) 39256e98164SDarrick J. Wong { 39356e98164SDarrick J. Wong struct xfs_agf *agf = agbp->b_addr; 39456e98164SDarrick J. Wong struct xbtree_afakeroot *afake = cur->bc_ag.afake; 39556e98164SDarrick J. Wong 39656e98164SDarrick J. Wong ASSERT(cur->bc_flags & XFS_BTREE_STAGING); 39756e98164SDarrick J. Wong 39856e98164SDarrick J. Wong agf->agf_refcount_root = cpu_to_be32(afake->af_root); 39956e98164SDarrick J. Wong agf->agf_refcount_level = cpu_to_be32(afake->af_levels); 40056e98164SDarrick J. Wong agf->agf_refcount_blocks = cpu_to_be32(afake->af_blocks); 40156e98164SDarrick J. Wong xfs_alloc_log_agf(tp, agbp, XFS_AGF_REFCOUNT_BLOCKS | 40256e98164SDarrick J. Wong XFS_AGF_REFCOUNT_ROOT | 40356e98164SDarrick J. Wong XFS_AGF_REFCOUNT_LEVEL); 40456e98164SDarrick J. Wong xfs_btree_commit_afakeroot(cur, tp, agbp, &xfs_refcountbt_ops); 40556e98164SDarrick J. Wong } 40656e98164SDarrick J. Wong 4070ed5f735SDarrick J. Wong /* Calculate number of records in a refcount btree block. */ 4080ed5f735SDarrick J. Wong static inline unsigned int 4090ed5f735SDarrick J. Wong xfs_refcountbt_block_maxrecs( 4100ed5f735SDarrick J. Wong unsigned int blocklen, 4110ed5f735SDarrick J. Wong bool leaf) 4120ed5f735SDarrick J. Wong { 4130ed5f735SDarrick J. Wong if (leaf) 4140ed5f735SDarrick J. Wong return blocklen / sizeof(struct xfs_refcount_rec); 4150ed5f735SDarrick J. Wong return blocklen / (sizeof(struct xfs_refcount_key) + 4160ed5f735SDarrick J. Wong sizeof(xfs_refcount_ptr_t)); 4170ed5f735SDarrick J. Wong } 4180ed5f735SDarrick J. Wong 4191946b91cSDarrick J. Wong /* 4201946b91cSDarrick J. Wong * Calculate the number of records in a refcount btree block. 4211946b91cSDarrick J. Wong */ 4221946b91cSDarrick J. Wong int 4231946b91cSDarrick J. Wong xfs_refcountbt_maxrecs( 4241946b91cSDarrick J. Wong int blocklen, 4251946b91cSDarrick J. Wong bool leaf) 4261946b91cSDarrick J. Wong { 4271946b91cSDarrick J. Wong blocklen -= XFS_REFCOUNT_BLOCK_LEN; 4280ed5f735SDarrick J. Wong return xfs_refcountbt_block_maxrecs(blocklen, leaf); 4290ed5f735SDarrick J. Wong } 4301946b91cSDarrick J. Wong 4310ed5f735SDarrick J. Wong /* Compute the max possible height of the maximally sized refcount btree. */ 4320ed5f735SDarrick J. Wong unsigned int 4330ed5f735SDarrick J. Wong xfs_refcountbt_maxlevels_ondisk(void) 4340ed5f735SDarrick J. Wong { 4350ed5f735SDarrick J. Wong unsigned int minrecs[2]; 4360ed5f735SDarrick J. Wong unsigned int blocklen; 4370ed5f735SDarrick J. Wong 4380ed5f735SDarrick J. Wong blocklen = XFS_MIN_CRC_BLOCKSIZE - XFS_BTREE_SBLOCK_CRC_LEN; 4390ed5f735SDarrick J. Wong 4400ed5f735SDarrick J. Wong minrecs[0] = xfs_refcountbt_block_maxrecs(blocklen, true) / 2; 4410ed5f735SDarrick J. Wong minrecs[1] = xfs_refcountbt_block_maxrecs(blocklen, false) / 2; 4420ed5f735SDarrick J. Wong 4430ed5f735SDarrick J. Wong return xfs_btree_compute_maxlevels(minrecs, XFS_MAX_CRC_AG_BLOCKS); 4441946b91cSDarrick J. Wong } 4451946b91cSDarrick J. Wong 4461946b91cSDarrick J. Wong /* Compute the maximum height of a refcount btree. */ 4471946b91cSDarrick J. Wong void 4481946b91cSDarrick J. Wong xfs_refcountbt_compute_maxlevels( 4491946b91cSDarrick J. Wong struct xfs_mount *mp) 4501946b91cSDarrick J. Wong { 4510ed5f735SDarrick J. Wong if (!xfs_has_reflink(mp)) { 4520ed5f735SDarrick J. Wong mp->m_refc_maxlevels = 0; 4530ed5f735SDarrick J. Wong return; 4540ed5f735SDarrick J. Wong } 4550ed5f735SDarrick J. Wong 456a1f69417SEric Sandeen mp->m_refc_maxlevels = xfs_btree_compute_maxlevels( 4571946b91cSDarrick J. Wong mp->m_refc_mnr, mp->m_sb.sb_agblocks); 4580ed5f735SDarrick J. Wong ASSERT(mp->m_refc_maxlevels <= xfs_refcountbt_maxlevels_ondisk()); 4591946b91cSDarrick J. Wong } 46084d69619SDarrick J. Wong 46184d69619SDarrick J. Wong /* Calculate the refcount btree size for some records. */ 46284d69619SDarrick J. Wong xfs_extlen_t 46384d69619SDarrick J. Wong xfs_refcountbt_calc_size( 46484d69619SDarrick J. Wong struct xfs_mount *mp, 46584d69619SDarrick J. Wong unsigned long long len) 46684d69619SDarrick J. Wong { 467a1f69417SEric Sandeen return xfs_btree_calc_size(mp->m_refc_mnr, len); 46884d69619SDarrick J. Wong } 46984d69619SDarrick J. Wong 47084d69619SDarrick J. Wong /* 47184d69619SDarrick J. Wong * Calculate the maximum refcount btree size. 47284d69619SDarrick J. Wong */ 47384d69619SDarrick J. Wong xfs_extlen_t 47484d69619SDarrick J. Wong xfs_refcountbt_max_size( 47520e73b00SDarrick J. Wong struct xfs_mount *mp, 47620e73b00SDarrick J. Wong xfs_agblock_t agblocks) 47784d69619SDarrick J. Wong { 47884d69619SDarrick J. Wong /* Bail out if we're uninitialized, which can happen in mkfs. */ 47984d69619SDarrick J. Wong if (mp->m_refc_mxr[0] == 0) 48084d69619SDarrick J. Wong return 0; 48184d69619SDarrick J. Wong 48220e73b00SDarrick J. Wong return xfs_refcountbt_calc_size(mp, agblocks); 48384d69619SDarrick J. Wong } 48484d69619SDarrick J. Wong 48584d69619SDarrick J. Wong /* 48684d69619SDarrick J. Wong * Figure out how many blocks to reserve and how many are used by this btree. 48784d69619SDarrick J. Wong */ 48884d69619SDarrick J. Wong int 48984d69619SDarrick J. Wong xfs_refcountbt_calc_reserves( 49084d69619SDarrick J. Wong struct xfs_mount *mp, 491ebcbef3aSDarrick J. Wong struct xfs_trans *tp, 49230933120SDave Chinner struct xfs_perag *pag, 49384d69619SDarrick J. Wong xfs_extlen_t *ask, 49484d69619SDarrick J. Wong xfs_extlen_t *used) 49584d69619SDarrick J. Wong { 49684d69619SDarrick J. Wong struct xfs_buf *agbp; 49784d69619SDarrick J. Wong struct xfs_agf *agf; 49820e73b00SDarrick J. Wong xfs_agblock_t agblocks; 49984d69619SDarrick J. Wong xfs_extlen_t tree_len; 50084d69619SDarrick J. Wong int error; 50184d69619SDarrick J. Wong 50238c26bfdSDave Chinner if (!xfs_has_reflink(mp)) 50384d69619SDarrick J. Wong return 0; 50484d69619SDarrick J. Wong 50508d3e84fSDave Chinner error = xfs_alloc_read_agf(pag, tp, 0, &agbp); 50684d69619SDarrick J. Wong if (error) 50784d69619SDarrick J. Wong return error; 50884d69619SDarrick J. Wong 5099798f615SChristoph Hellwig agf = agbp->b_addr; 51020e73b00SDarrick J. Wong agblocks = be32_to_cpu(agf->agf_length); 51184d69619SDarrick J. Wong tree_len = be32_to_cpu(agf->agf_refcount_blocks); 512ebcbef3aSDarrick J. Wong xfs_trans_brelse(tp, agbp); 51384d69619SDarrick J. Wong 5145cd213b0SDarrick J. Wong /* 5155cd213b0SDarrick J. Wong * The log is permanently allocated, so the space it occupies will 5165cd213b0SDarrick J. Wong * never be available for the kinds of things that would require btree 5175cd213b0SDarrick J. Wong * expansion. We therefore can pretend the space isn't there. 5185cd213b0SDarrick J. Wong */ 51936029deeSDave Chinner if (xfs_ag_contains_log(mp, pag->pag_agno)) 5205cd213b0SDarrick J. Wong agblocks -= mp->m_sb.sb_logblocks; 5215cd213b0SDarrick J. Wong 52220e73b00SDarrick J. Wong *ask += xfs_refcountbt_max_size(mp, agblocks); 52384d69619SDarrick J. Wong *used += tree_len; 52484d69619SDarrick J. Wong 52584d69619SDarrick J. Wong return error; 52684d69619SDarrick J. Wong } 5279fa47bdcSDarrick J. Wong 5289fa47bdcSDarrick J. Wong int __init 5299fa47bdcSDarrick J. Wong xfs_refcountbt_init_cur_cache(void) 5309fa47bdcSDarrick J. Wong { 5319fa47bdcSDarrick J. Wong xfs_refcountbt_cur_cache = kmem_cache_create("xfs_refcbt_cur", 5329fa47bdcSDarrick J. Wong xfs_btree_cur_sizeof(xfs_refcountbt_maxlevels_ondisk()), 5339fa47bdcSDarrick J. Wong 0, 0, NULL); 5349fa47bdcSDarrick J. Wong 5359fa47bdcSDarrick J. Wong if (!xfs_refcountbt_cur_cache) 5369fa47bdcSDarrick J. Wong return -ENOMEM; 5379fa47bdcSDarrick J. Wong return 0; 5389fa47bdcSDarrick J. Wong } 5399fa47bdcSDarrick J. Wong 5409fa47bdcSDarrick J. Wong void 5419fa47bdcSDarrick J. Wong xfs_refcountbt_destroy_cur_cache(void) 5429fa47bdcSDarrick J. Wong { 5439fa47bdcSDarrick J. Wong kmem_cache_destroy(xfs_refcountbt_cur_cache); 5449fa47bdcSDarrick J. Wong xfs_refcountbt_cur_cache = NULL; 5459fa47bdcSDarrick J. Wong } 546