Mention branches and keyring.
[releases.git] / libxfs / xfs_ag.c
1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * Copyright (c) 2000-2005 Silicon Graphics, Inc.
4  * Copyright (c) 2018 Red Hat, Inc.
5  * All rights reserved.
6  */
7
8 #include "xfs.h"
9 #include "xfs_fs.h"
10 #include "xfs_shared.h"
11 #include "xfs_format.h"
12 #include "xfs_trans_resv.h"
13 #include "xfs_bit.h"
14 #include "xfs_sb.h"
15 #include "xfs_mount.h"
16 #include "xfs_btree.h"
17 #include "xfs_alloc_btree.h"
18 #include "xfs_rmap_btree.h"
19 #include "xfs_alloc.h"
20 #include "xfs_ialloc.h"
21 #include "xfs_rmap.h"
22 #include "xfs_ag.h"
23 #include "xfs_ag_resv.h"
24 #include "xfs_health.h"
25 #include "xfs_error.h"
26 #include "xfs_bmap.h"
27 #include "xfs_defer.h"
28 #include "xfs_log_format.h"
29 #include "xfs_trans.h"
30 #include "xfs_trace.h"
31 #include "xfs_inode.h"
32 #include "xfs_icache.h"
33
34
35 /*
36  * Passive reference counting access wrappers to the perag structures.  If the
37  * per-ag structure is to be freed, the freeing code is responsible for cleaning
38  * up objects with passive references before freeing the structure. This is
39  * things like cached buffers.
40  */
41 struct xfs_perag *
42 xfs_perag_get(
43         struct xfs_mount        *mp,
44         xfs_agnumber_t          agno)
45 {
46         struct xfs_perag        *pag;
47         int                     ref = 0;
48
49         rcu_read_lock();
50         pag = radix_tree_lookup(&mp->m_perag_tree, agno);
51         if (pag) {
52                 ASSERT(atomic_read(&pag->pag_ref) >= 0);
53                 ref = atomic_inc_return(&pag->pag_ref);
54         }
55         rcu_read_unlock();
56         trace_xfs_perag_get(mp, agno, ref, _RET_IP_);
57         return pag;
58 }
59
60 /*
61  * search from @first to find the next perag with the given tag set.
62  */
63 struct xfs_perag *
64 xfs_perag_get_tag(
65         struct xfs_mount        *mp,
66         xfs_agnumber_t          first,
67         unsigned int            tag)
68 {
69         struct xfs_perag        *pag;
70         int                     found;
71         int                     ref;
72
73         rcu_read_lock();
74         found = radix_tree_gang_lookup_tag(&mp->m_perag_tree,
75                                         (void **)&pag, first, 1, tag);
76         if (found <= 0) {
77                 rcu_read_unlock();
78                 return NULL;
79         }
80         ref = atomic_inc_return(&pag->pag_ref);
81         rcu_read_unlock();
82         trace_xfs_perag_get_tag(mp, pag->pag_agno, ref, _RET_IP_);
83         return pag;
84 }
85
86 void
87 xfs_perag_put(
88         struct xfs_perag        *pag)
89 {
90         int     ref;
91
92         ASSERT(atomic_read(&pag->pag_ref) > 0);
93         ref = atomic_dec_return(&pag->pag_ref);
94         trace_xfs_perag_put(pag->pag_mount, pag->pag_agno, ref, _RET_IP_);
95 }
96
97 /*
98  * xfs_initialize_perag_data
99  *
100  * Read in each per-ag structure so we can count up the number of
101  * allocated inodes, free inodes and used filesystem blocks as this
102  * information is no longer persistent in the superblock. Once we have
103  * this information, write it into the in-core superblock structure.
104  */
105 int
106 xfs_initialize_perag_data(
107         struct xfs_mount        *mp,
108         xfs_agnumber_t          agcount)
109 {
110         xfs_agnumber_t          index;
111         struct xfs_perag        *pag;
112         struct xfs_sb           *sbp = &mp->m_sb;
113         uint64_t                ifree = 0;
114         uint64_t                ialloc = 0;
115         uint64_t                bfree = 0;
116         uint64_t                bfreelst = 0;
117         uint64_t                btree = 0;
118         uint64_t                fdblocks;
119         int                     error = 0;
120
121         for (index = 0; index < agcount; index++) {
122                 /*
123                  * Read the AGF and AGI buffers to populate the per-ag
124                  * structures for us.
125                  */
126                 pag = xfs_perag_get(mp, index);
127                 error = xfs_alloc_read_agf(pag, NULL, 0, NULL);
128                 if (!error)
129                         error = xfs_ialloc_read_agi(pag, NULL, NULL);
130                 if (error) {
131                         xfs_perag_put(pag);
132                         return error;
133                 }
134
135                 ifree += pag->pagi_freecount;
136                 ialloc += pag->pagi_count;
137                 bfree += pag->pagf_freeblks;
138                 bfreelst += pag->pagf_flcount;
139                 btree += pag->pagf_btreeblks;
140                 xfs_perag_put(pag);
141         }
142         fdblocks = bfree + bfreelst + btree;
143
144         /*
145          * If the new summary counts are obviously incorrect, fail the
146          * mount operation because that implies the AGFs are also corrupt.
147          * Clear FS_COUNTERS so that we don't unmount with a dirty log, which
148          * will prevent xfs_repair from fixing anything.
149          */
150         if (fdblocks > sbp->sb_dblocks || ifree > ialloc) {
151                 xfs_alert(mp, "AGF corruption. Please run xfs_repair.");
152                 error = -EFSCORRUPTED;
153                 goto out;
154         }
155
156         /* Overwrite incore superblock counters with just-read data */
157         spin_lock(&mp->m_sb_lock);
158         sbp->sb_ifree = ifree;
159         sbp->sb_icount = ialloc;
160         sbp->sb_fdblocks = fdblocks;
161         spin_unlock(&mp->m_sb_lock);
162
163         xfs_reinit_percpu_counters(mp);
164 out:
165         xfs_fs_mark_healthy(mp, XFS_SICK_FS_COUNTERS);
166         return error;
167 }
168
169 STATIC void
170 __xfs_free_perag(
171         struct rcu_head *head)
172 {
173         struct xfs_perag *pag = container_of(head, struct xfs_perag, rcu_head);
174
175         ASSERT(!delayed_work_pending(&pag->pag_blockgc_work));
176         kmem_free(pag);
177 }
178
179 /*
180  * Free up the per-ag resources associated with the mount structure.
181  */
182 void
183 xfs_free_perag(
184         struct xfs_mount        *mp)
185 {
186         struct xfs_perag        *pag;
187         xfs_agnumber_t          agno;
188
189         for (agno = 0; agno < mp->m_sb.sb_agcount; agno++) {
190                 spin_lock(&mp->m_perag_lock);
191                 pag = radix_tree_delete(&mp->m_perag_tree, agno);
192                 spin_unlock(&mp->m_perag_lock);
193                 ASSERT(pag);
194                 XFS_IS_CORRUPT(pag->pag_mount, atomic_read(&pag->pag_ref) != 0);
195
196                 cancel_delayed_work_sync(&pag->pag_blockgc_work);
197                 xfs_buf_hash_destroy(pag);
198
199                 call_rcu(&pag->rcu_head, __xfs_free_perag);
200         }
201 }
202
203 /* Find the size of the AG, in blocks. */
204 static xfs_agblock_t
205 __xfs_ag_block_count(
206         struct xfs_mount        *mp,
207         xfs_agnumber_t          agno,
208         xfs_agnumber_t          agcount,
209         xfs_rfsblock_t          dblocks)
210 {
211         ASSERT(agno < agcount);
212
213         if (agno < agcount - 1)
214                 return mp->m_sb.sb_agblocks;
215         return dblocks - (agno * mp->m_sb.sb_agblocks);
216 }
217
218 xfs_agblock_t
219 xfs_ag_block_count(
220         struct xfs_mount        *mp,
221         xfs_agnumber_t          agno)
222 {
223         return __xfs_ag_block_count(mp, agno, mp->m_sb.sb_agcount,
224                         mp->m_sb.sb_dblocks);
225 }
226
227 /* Calculate the first and last possible inode number in an AG. */
228 static void
229 __xfs_agino_range(
230         struct xfs_mount        *mp,
231         xfs_agblock_t           eoag,
232         xfs_agino_t             *first,
233         xfs_agino_t             *last)
234 {
235         xfs_agblock_t           bno;
236
237         /*
238          * Calculate the first inode, which will be in the first
239          * cluster-aligned block after the AGFL.
240          */
241         bno = round_up(XFS_AGFL_BLOCK(mp) + 1, M_IGEO(mp)->cluster_align);
242         *first = XFS_AGB_TO_AGINO(mp, bno);
243
244         /*
245          * Calculate the last inode, which will be at the end of the
246          * last (aligned) cluster that can be allocated in the AG.
247          */
248         bno = round_down(eoag, M_IGEO(mp)->cluster_align);
249         *last = XFS_AGB_TO_AGINO(mp, bno) - 1;
250 }
251
252 void
253 xfs_agino_range(
254         struct xfs_mount        *mp,
255         xfs_agnumber_t          agno,
256         xfs_agino_t             *first,
257         xfs_agino_t             *last)
258 {
259         return __xfs_agino_range(mp, xfs_ag_block_count(mp, agno), first, last);
260 }
261
262 int
263 xfs_initialize_perag(
264         struct xfs_mount        *mp,
265         xfs_agnumber_t          agcount,
266         xfs_rfsblock_t          dblocks,
267         xfs_agnumber_t          *maxagi)
268 {
269         struct xfs_perag        *pag;
270         xfs_agnumber_t          index;
271         xfs_agnumber_t          first_initialised = NULLAGNUMBER;
272         int                     error;
273
274         /*
275          * Walk the current per-ag tree so we don't try to initialise AGs
276          * that already exist (growfs case). Allocate and insert all the
277          * AGs we don't find ready for initialisation.
278          */
279         for (index = 0; index < agcount; index++) {
280                 pag = xfs_perag_get(mp, index);
281                 if (pag) {
282                         xfs_perag_put(pag);
283                         continue;
284                 }
285
286                 pag = kmem_zalloc(sizeof(*pag), KM_MAYFAIL);
287                 if (!pag) {
288                         error = -ENOMEM;
289                         goto out_unwind_new_pags;
290                 }
291                 pag->pag_agno = index;
292                 pag->pag_mount = mp;
293
294                 error = radix_tree_preload(GFP_NOFS);
295                 if (error)
296                         goto out_free_pag;
297
298                 spin_lock(&mp->m_perag_lock);
299                 if (radix_tree_insert(&mp->m_perag_tree, index, pag)) {
300                         WARN_ON_ONCE(1);
301                         spin_unlock(&mp->m_perag_lock);
302                         radix_tree_preload_end();
303                         error = -EEXIST;
304                         goto out_free_pag;
305                 }
306                 spin_unlock(&mp->m_perag_lock);
307                 radix_tree_preload_end();
308
309 #ifdef __KERNEL__
310                 /* Place kernel structure only init below this point. */
311                 spin_lock_init(&pag->pag_ici_lock);
312                 spin_lock_init(&pag->pagb_lock);
313                 spin_lock_init(&pag->pag_state_lock);
314                 INIT_DELAYED_WORK(&pag->pag_blockgc_work, xfs_blockgc_worker);
315                 INIT_RADIX_TREE(&pag->pag_ici_root, GFP_ATOMIC);
316                 init_waitqueue_head(&pag->pagb_wait);
317                 pag->pagb_count = 0;
318                 pag->pagb_tree = RB_ROOT;
319 #endif /* __KERNEL__ */
320
321                 error = xfs_buf_hash_init(pag);
322                 if (error)
323                         goto out_remove_pag;
324
325                 /* first new pag is fully initialized */
326                 if (first_initialised == NULLAGNUMBER)
327                         first_initialised = index;
328
329                 /*
330                  * Pre-calculated geometry
331                  */
332                 pag->block_count = __xfs_ag_block_count(mp, index, agcount,
333                                 dblocks);
334                 pag->min_block = XFS_AGFL_BLOCK(mp);
335                 __xfs_agino_range(mp, pag->block_count, &pag->agino_min,
336                                 &pag->agino_max);
337         }
338
339         index = xfs_set_inode_alloc(mp, agcount);
340
341         if (maxagi)
342                 *maxagi = index;
343
344         mp->m_ag_prealloc_blocks = xfs_prealloc_blocks(mp);
345         return 0;
346
347 out_remove_pag:
348         radix_tree_delete(&mp->m_perag_tree, index);
349 out_free_pag:
350         kmem_free(pag);
351 out_unwind_new_pags:
352         /* unwind any prior newly initialized pags */
353         for (index = first_initialised; index < agcount; index++) {
354                 pag = radix_tree_delete(&mp->m_perag_tree, index);
355                 if (!pag)
356                         break;
357                 xfs_buf_hash_destroy(pag);
358                 kmem_free(pag);
359         }
360         return error;
361 }
362
363 static int
364 xfs_get_aghdr_buf(
365         struct xfs_mount        *mp,
366         xfs_daddr_t             blkno,
367         size_t                  numblks,
368         struct xfs_buf          **bpp,
369         const struct xfs_buf_ops *ops)
370 {
371         struct xfs_buf          *bp;
372         int                     error;
373
374         error = xfs_buf_get_uncached(mp->m_ddev_targp, numblks, 0, &bp);
375         if (error)
376                 return error;
377
378         bp->b_maps[0].bm_bn = blkno;
379         bp->b_ops = ops;
380
381         *bpp = bp;
382         return 0;
383 }
384
385 /*
386  * Generic btree root block init function
387  */
388 static void
389 xfs_btroot_init(
390         struct xfs_mount        *mp,
391         struct xfs_buf          *bp,
392         struct aghdr_init_data  *id)
393 {
394         xfs_btree_init_block(mp, bp, id->type, 0, 0, id->agno);
395 }
396
397 /* Finish initializing a free space btree. */
398 static void
399 xfs_freesp_init_recs(
400         struct xfs_mount        *mp,
401         struct xfs_buf          *bp,
402         struct aghdr_init_data  *id)
403 {
404         struct xfs_alloc_rec    *arec;
405         struct xfs_btree_block  *block = XFS_BUF_TO_BLOCK(bp);
406
407         arec = XFS_ALLOC_REC_ADDR(mp, XFS_BUF_TO_BLOCK(bp), 1);
408         arec->ar_startblock = cpu_to_be32(mp->m_ag_prealloc_blocks);
409
410         if (xfs_ag_contains_log(mp, id->agno)) {
411                 struct xfs_alloc_rec    *nrec;
412                 xfs_agblock_t           start = XFS_FSB_TO_AGBNO(mp,
413                                                         mp->m_sb.sb_logstart);
414
415                 ASSERT(start >= mp->m_ag_prealloc_blocks);
416                 if (start != mp->m_ag_prealloc_blocks) {
417                         /*
418                          * Modify first record to pad stripe align of log
419                          */
420                         arec->ar_blockcount = cpu_to_be32(start -
421                                                 mp->m_ag_prealloc_blocks);
422                         nrec = arec + 1;
423
424                         /*
425                          * Insert second record at start of internal log
426                          * which then gets trimmed.
427                          */
428                         nrec->ar_startblock = cpu_to_be32(
429                                         be32_to_cpu(arec->ar_startblock) +
430                                         be32_to_cpu(arec->ar_blockcount));
431                         arec = nrec;
432                         be16_add_cpu(&block->bb_numrecs, 1);
433                 }
434                 /*
435                  * Change record start to after the internal log
436                  */
437                 be32_add_cpu(&arec->ar_startblock, mp->m_sb.sb_logblocks);
438         }
439
440         /*
441          * Calculate the record block count and check for the case where
442          * the log might have consumed all available space in the AG. If
443          * so, reset the record count to 0 to avoid exposure of an invalid
444          * record start block.
445          */
446         arec->ar_blockcount = cpu_to_be32(id->agsize -
447                                           be32_to_cpu(arec->ar_startblock));
448         if (!arec->ar_blockcount)
449                 block->bb_numrecs = 0;
450 }
451
452 /*
453  * Alloc btree root block init functions
454  */
455 static void
456 xfs_bnoroot_init(
457         struct xfs_mount        *mp,
458         struct xfs_buf          *bp,
459         struct aghdr_init_data  *id)
460 {
461         xfs_btree_init_block(mp, bp, XFS_BTNUM_BNO, 0, 1, id->agno);
462         xfs_freesp_init_recs(mp, bp, id);
463 }
464
465 static void
466 xfs_cntroot_init(
467         struct xfs_mount        *mp,
468         struct xfs_buf          *bp,
469         struct aghdr_init_data  *id)
470 {
471         xfs_btree_init_block(mp, bp, XFS_BTNUM_CNT, 0, 1, id->agno);
472         xfs_freesp_init_recs(mp, bp, id);
473 }
474
475 /*
476  * Reverse map root block init
477  */
478 static void
479 xfs_rmaproot_init(
480         struct xfs_mount        *mp,
481         struct xfs_buf          *bp,
482         struct aghdr_init_data  *id)
483 {
484         struct xfs_btree_block  *block = XFS_BUF_TO_BLOCK(bp);
485         struct xfs_rmap_rec     *rrec;
486
487         xfs_btree_init_block(mp, bp, XFS_BTNUM_RMAP, 0, 4, id->agno);
488
489         /*
490          * mark the AG header regions as static metadata The BNO
491          * btree block is the first block after the headers, so
492          * it's location defines the size of region the static
493          * metadata consumes.
494          *
495          * Note: unlike mkfs, we never have to account for log
496          * space when growing the data regions
497          */
498         rrec = XFS_RMAP_REC_ADDR(block, 1);
499         rrec->rm_startblock = 0;
500         rrec->rm_blockcount = cpu_to_be32(XFS_BNO_BLOCK(mp));
501         rrec->rm_owner = cpu_to_be64(XFS_RMAP_OWN_FS);
502         rrec->rm_offset = 0;
503
504         /* account freespace btree root blocks */
505         rrec = XFS_RMAP_REC_ADDR(block, 2);
506         rrec->rm_startblock = cpu_to_be32(XFS_BNO_BLOCK(mp));
507         rrec->rm_blockcount = cpu_to_be32(2);
508         rrec->rm_owner = cpu_to_be64(XFS_RMAP_OWN_AG);
509         rrec->rm_offset = 0;
510
511         /* account inode btree root blocks */
512         rrec = XFS_RMAP_REC_ADDR(block, 3);
513         rrec->rm_startblock = cpu_to_be32(XFS_IBT_BLOCK(mp));
514         rrec->rm_blockcount = cpu_to_be32(XFS_RMAP_BLOCK(mp) -
515                                           XFS_IBT_BLOCK(mp));
516         rrec->rm_owner = cpu_to_be64(XFS_RMAP_OWN_INOBT);
517         rrec->rm_offset = 0;
518
519         /* account for rmap btree root */
520         rrec = XFS_RMAP_REC_ADDR(block, 4);
521         rrec->rm_startblock = cpu_to_be32(XFS_RMAP_BLOCK(mp));
522         rrec->rm_blockcount = cpu_to_be32(1);
523         rrec->rm_owner = cpu_to_be64(XFS_RMAP_OWN_AG);
524         rrec->rm_offset = 0;
525
526         /* account for refc btree root */
527         if (xfs_has_reflink(mp)) {
528                 rrec = XFS_RMAP_REC_ADDR(block, 5);
529                 rrec->rm_startblock = cpu_to_be32(xfs_refc_block(mp));
530                 rrec->rm_blockcount = cpu_to_be32(1);
531                 rrec->rm_owner = cpu_to_be64(XFS_RMAP_OWN_REFC);
532                 rrec->rm_offset = 0;
533                 be16_add_cpu(&block->bb_numrecs, 1);
534         }
535
536         /* account for the log space */
537         if (xfs_ag_contains_log(mp, id->agno)) {
538                 rrec = XFS_RMAP_REC_ADDR(block,
539                                 be16_to_cpu(block->bb_numrecs) + 1);
540                 rrec->rm_startblock = cpu_to_be32(
541                                 XFS_FSB_TO_AGBNO(mp, mp->m_sb.sb_logstart));
542                 rrec->rm_blockcount = cpu_to_be32(mp->m_sb.sb_logblocks);
543                 rrec->rm_owner = cpu_to_be64(XFS_RMAP_OWN_LOG);
544                 rrec->rm_offset = 0;
545                 be16_add_cpu(&block->bb_numrecs, 1);
546         }
547 }
548
549 /*
550  * Initialise new secondary superblocks with the pre-grow geometry, but mark
551  * them as "in progress" so we know they haven't yet been activated. This will
552  * get cleared when the update with the new geometry information is done after
553  * changes to the primary are committed. This isn't strictly necessary, but we
554  * get it for free with the delayed buffer write lists and it means we can tell
555  * if a grow operation didn't complete properly after the fact.
556  */
557 static void
558 xfs_sbblock_init(
559         struct xfs_mount        *mp,
560         struct xfs_buf          *bp,
561         struct aghdr_init_data  *id)
562 {
563         struct xfs_dsb          *dsb = bp->b_addr;
564
565         xfs_sb_to_disk(dsb, &mp->m_sb);
566         dsb->sb_inprogress = 1;
567 }
568
569 static void
570 xfs_agfblock_init(
571         struct xfs_mount        *mp,
572         struct xfs_buf          *bp,
573         struct aghdr_init_data  *id)
574 {
575         struct xfs_agf          *agf = bp->b_addr;
576         xfs_extlen_t            tmpsize;
577
578         agf->agf_magicnum = cpu_to_be32(XFS_AGF_MAGIC);
579         agf->agf_versionnum = cpu_to_be32(XFS_AGF_VERSION);
580         agf->agf_seqno = cpu_to_be32(id->agno);
581         agf->agf_length = cpu_to_be32(id->agsize);
582         agf->agf_roots[XFS_BTNUM_BNOi] = cpu_to_be32(XFS_BNO_BLOCK(mp));
583         agf->agf_roots[XFS_BTNUM_CNTi] = cpu_to_be32(XFS_CNT_BLOCK(mp));
584         agf->agf_levels[XFS_BTNUM_BNOi] = cpu_to_be32(1);
585         agf->agf_levels[XFS_BTNUM_CNTi] = cpu_to_be32(1);
586         if (xfs_has_rmapbt(mp)) {
587                 agf->agf_roots[XFS_BTNUM_RMAPi] =
588                                         cpu_to_be32(XFS_RMAP_BLOCK(mp));
589                 agf->agf_levels[XFS_BTNUM_RMAPi] = cpu_to_be32(1);
590                 agf->agf_rmap_blocks = cpu_to_be32(1);
591         }
592
593         agf->agf_flfirst = cpu_to_be32(1);
594         agf->agf_fllast = 0;
595         agf->agf_flcount = 0;
596         tmpsize = id->agsize - mp->m_ag_prealloc_blocks;
597         agf->agf_freeblks = cpu_to_be32(tmpsize);
598         agf->agf_longest = cpu_to_be32(tmpsize);
599         if (xfs_has_crc(mp))
600                 uuid_copy(&agf->agf_uuid, &mp->m_sb.sb_meta_uuid);
601         if (xfs_has_reflink(mp)) {
602                 agf->agf_refcount_root = cpu_to_be32(
603                                 xfs_refc_block(mp));
604                 agf->agf_refcount_level = cpu_to_be32(1);
605                 agf->agf_refcount_blocks = cpu_to_be32(1);
606         }
607
608         if (xfs_ag_contains_log(mp, id->agno)) {
609                 int64_t logblocks = mp->m_sb.sb_logblocks;
610
611                 be32_add_cpu(&agf->agf_freeblks, -logblocks);
612                 agf->agf_longest = cpu_to_be32(id->agsize -
613                         XFS_FSB_TO_AGBNO(mp, mp->m_sb.sb_logstart) - logblocks);
614         }
615 }
616
617 static void
618 xfs_agflblock_init(
619         struct xfs_mount        *mp,
620         struct xfs_buf          *bp,
621         struct aghdr_init_data  *id)
622 {
623         struct xfs_agfl         *agfl = XFS_BUF_TO_AGFL(bp);
624         __be32                  *agfl_bno;
625         int                     bucket;
626
627         if (xfs_has_crc(mp)) {
628                 agfl->agfl_magicnum = cpu_to_be32(XFS_AGFL_MAGIC);
629                 agfl->agfl_seqno = cpu_to_be32(id->agno);
630                 uuid_copy(&agfl->agfl_uuid, &mp->m_sb.sb_meta_uuid);
631         }
632
633         agfl_bno = xfs_buf_to_agfl_bno(bp);
634         for (bucket = 0; bucket < xfs_agfl_size(mp); bucket++)
635                 agfl_bno[bucket] = cpu_to_be32(NULLAGBLOCK);
636 }
637
638 static void
639 xfs_agiblock_init(
640         struct xfs_mount        *mp,
641         struct xfs_buf          *bp,
642         struct aghdr_init_data  *id)
643 {
644         struct xfs_agi          *agi = bp->b_addr;
645         int                     bucket;
646
647         agi->agi_magicnum = cpu_to_be32(XFS_AGI_MAGIC);
648         agi->agi_versionnum = cpu_to_be32(XFS_AGI_VERSION);
649         agi->agi_seqno = cpu_to_be32(id->agno);
650         agi->agi_length = cpu_to_be32(id->agsize);
651         agi->agi_count = 0;
652         agi->agi_root = cpu_to_be32(XFS_IBT_BLOCK(mp));
653         agi->agi_level = cpu_to_be32(1);
654         agi->agi_freecount = 0;
655         agi->agi_newino = cpu_to_be32(NULLAGINO);
656         agi->agi_dirino = cpu_to_be32(NULLAGINO);
657         if (xfs_has_crc(mp))
658                 uuid_copy(&agi->agi_uuid, &mp->m_sb.sb_meta_uuid);
659         if (xfs_has_finobt(mp)) {
660                 agi->agi_free_root = cpu_to_be32(XFS_FIBT_BLOCK(mp));
661                 agi->agi_free_level = cpu_to_be32(1);
662         }
663         for (bucket = 0; bucket < XFS_AGI_UNLINKED_BUCKETS; bucket++)
664                 agi->agi_unlinked[bucket] = cpu_to_be32(NULLAGINO);
665         if (xfs_has_inobtcounts(mp)) {
666                 agi->agi_iblocks = cpu_to_be32(1);
667                 if (xfs_has_finobt(mp))
668                         agi->agi_fblocks = cpu_to_be32(1);
669         }
670 }
671
672 typedef void (*aghdr_init_work_f)(struct xfs_mount *mp, struct xfs_buf *bp,
673                                   struct aghdr_init_data *id);
674 static int
675 xfs_ag_init_hdr(
676         struct xfs_mount        *mp,
677         struct aghdr_init_data  *id,
678         aghdr_init_work_f       work,
679         const struct xfs_buf_ops *ops)
680 {
681         struct xfs_buf          *bp;
682         int                     error;
683
684         error = xfs_get_aghdr_buf(mp, id->daddr, id->numblks, &bp, ops);
685         if (error)
686                 return error;
687
688         (*work)(mp, bp, id);
689
690         xfs_buf_delwri_queue(bp, &id->buffer_list);
691         xfs_buf_relse(bp);
692         return 0;
693 }
694
695 struct xfs_aghdr_grow_data {
696         xfs_daddr_t             daddr;
697         size_t                  numblks;
698         const struct xfs_buf_ops *ops;
699         aghdr_init_work_f       work;
700         xfs_btnum_t             type;
701         bool                    need_init;
702 };
703
704 /*
705  * Prepare new AG headers to be written to disk. We use uncached buffers here,
706  * as it is assumed these new AG headers are currently beyond the currently
707  * valid filesystem address space. Using cached buffers would trip over EOFS
708  * corruption detection alogrithms in the buffer cache lookup routines.
709  *
710  * This is a non-transactional function, but the prepared buffers are added to a
711  * delayed write buffer list supplied by the caller so they can submit them to
712  * disk and wait on them as required.
713  */
714 int
715 xfs_ag_init_headers(
716         struct xfs_mount        *mp,
717         struct aghdr_init_data  *id)
718
719 {
720         struct xfs_aghdr_grow_data aghdr_data[] = {
721         { /* SB */
722                 .daddr = XFS_AG_DADDR(mp, id->agno, XFS_SB_DADDR),
723                 .numblks = XFS_FSS_TO_BB(mp, 1),
724                 .ops = &xfs_sb_buf_ops,
725                 .work = &xfs_sbblock_init,
726                 .need_init = true
727         },
728         { /* AGF */
729                 .daddr = XFS_AG_DADDR(mp, id->agno, XFS_AGF_DADDR(mp)),
730                 .numblks = XFS_FSS_TO_BB(mp, 1),
731                 .ops = &xfs_agf_buf_ops,
732                 .work = &xfs_agfblock_init,
733                 .need_init = true
734         },
735         { /* AGFL */
736                 .daddr = XFS_AG_DADDR(mp, id->agno, XFS_AGFL_DADDR(mp)),
737                 .numblks = XFS_FSS_TO_BB(mp, 1),
738                 .ops = &xfs_agfl_buf_ops,
739                 .work = &xfs_agflblock_init,
740                 .need_init = true
741         },
742         { /* AGI */
743                 .daddr = XFS_AG_DADDR(mp, id->agno, XFS_AGI_DADDR(mp)),
744                 .numblks = XFS_FSS_TO_BB(mp, 1),
745                 .ops = &xfs_agi_buf_ops,
746                 .work = &xfs_agiblock_init,
747                 .need_init = true
748         },
749         { /* BNO root block */
750                 .daddr = XFS_AGB_TO_DADDR(mp, id->agno, XFS_BNO_BLOCK(mp)),
751                 .numblks = BTOBB(mp->m_sb.sb_blocksize),
752                 .ops = &xfs_bnobt_buf_ops,
753                 .work = &xfs_bnoroot_init,
754                 .need_init = true
755         },
756         { /* CNT root block */
757                 .daddr = XFS_AGB_TO_DADDR(mp, id->agno, XFS_CNT_BLOCK(mp)),
758                 .numblks = BTOBB(mp->m_sb.sb_blocksize),
759                 .ops = &xfs_cntbt_buf_ops,
760                 .work = &xfs_cntroot_init,
761                 .need_init = true
762         },
763         { /* INO root block */
764                 .daddr = XFS_AGB_TO_DADDR(mp, id->agno, XFS_IBT_BLOCK(mp)),
765                 .numblks = BTOBB(mp->m_sb.sb_blocksize),
766                 .ops = &xfs_inobt_buf_ops,
767                 .work = &xfs_btroot_init,
768                 .type = XFS_BTNUM_INO,
769                 .need_init = true
770         },
771         { /* FINO root block */
772                 .daddr = XFS_AGB_TO_DADDR(mp, id->agno, XFS_FIBT_BLOCK(mp)),
773                 .numblks = BTOBB(mp->m_sb.sb_blocksize),
774                 .ops = &xfs_finobt_buf_ops,
775                 .work = &xfs_btroot_init,
776                 .type = XFS_BTNUM_FINO,
777                 .need_init =  xfs_has_finobt(mp)
778         },
779         { /* RMAP root block */
780                 .daddr = XFS_AGB_TO_DADDR(mp, id->agno, XFS_RMAP_BLOCK(mp)),
781                 .numblks = BTOBB(mp->m_sb.sb_blocksize),
782                 .ops = &xfs_rmapbt_buf_ops,
783                 .work = &xfs_rmaproot_init,
784                 .need_init = xfs_has_rmapbt(mp)
785         },
786         { /* REFC root block */
787                 .daddr = XFS_AGB_TO_DADDR(mp, id->agno, xfs_refc_block(mp)),
788                 .numblks = BTOBB(mp->m_sb.sb_blocksize),
789                 .ops = &xfs_refcountbt_buf_ops,
790                 .work = &xfs_btroot_init,
791                 .type = XFS_BTNUM_REFC,
792                 .need_init = xfs_has_reflink(mp)
793         },
794         { /* NULL terminating block */
795                 .daddr = XFS_BUF_DADDR_NULL,
796         }
797         };
798         struct  xfs_aghdr_grow_data *dp;
799         int                     error = 0;
800
801         /* Account for AG free space in new AG */
802         id->nfree += id->agsize - mp->m_ag_prealloc_blocks;
803         for (dp = &aghdr_data[0]; dp->daddr != XFS_BUF_DADDR_NULL; dp++) {
804                 if (!dp->need_init)
805                         continue;
806
807                 id->daddr = dp->daddr;
808                 id->numblks = dp->numblks;
809                 id->type = dp->type;
810                 error = xfs_ag_init_hdr(mp, id, dp->work, dp->ops);
811                 if (error)
812                         break;
813         }
814         return error;
815 }
816
817 int
818 xfs_ag_shrink_space(
819         struct xfs_perag        *pag,
820         struct xfs_trans        **tpp,
821         xfs_extlen_t            delta)
822 {
823         struct xfs_mount        *mp = pag->pag_mount;
824         struct xfs_alloc_arg    args = {
825                 .tp     = *tpp,
826                 .mp     = mp,
827                 .type   = XFS_ALLOCTYPE_THIS_BNO,
828                 .minlen = delta,
829                 .maxlen = delta,
830                 .oinfo  = XFS_RMAP_OINFO_SKIP_UPDATE,
831                 .resv   = XFS_AG_RESV_NONE,
832                 .prod   = 1
833         };
834         struct xfs_buf          *agibp, *agfbp;
835         struct xfs_agi          *agi;
836         struct xfs_agf          *agf;
837         xfs_agblock_t           aglen;
838         int                     error, err2;
839
840         ASSERT(pag->pag_agno == mp->m_sb.sb_agcount - 1);
841         error = xfs_ialloc_read_agi(pag, *tpp, &agibp);
842         if (error)
843                 return error;
844
845         agi = agibp->b_addr;
846
847         error = xfs_alloc_read_agf(pag, *tpp, 0, &agfbp);
848         if (error)
849                 return error;
850
851         agf = agfbp->b_addr;
852         aglen = be32_to_cpu(agi->agi_length);
853         /* some extra paranoid checks before we shrink the ag */
854         if (XFS_IS_CORRUPT(mp, agf->agf_length != agi->agi_length))
855                 return -EFSCORRUPTED;
856         if (delta >= aglen)
857                 return -EINVAL;
858
859         args.fsbno = XFS_AGB_TO_FSB(mp, pag->pag_agno, aglen - delta);
860
861         /*
862          * Make sure that the last inode cluster cannot overlap with the new
863          * end of the AG, even if it's sparse.
864          */
865         error = xfs_ialloc_check_shrink(*tpp, pag->pag_agno, agibp,
866                         aglen - delta);
867         if (error)
868                 return error;
869
870         /*
871          * Disable perag reservations so it doesn't cause the allocation request
872          * to fail. We'll reestablish reservation before we return.
873          */
874         error = xfs_ag_resv_free(pag);
875         if (error)
876                 return error;
877
878         /* internal log shouldn't also show up in the free space btrees */
879         error = xfs_alloc_vextent(&args);
880         if (!error && args.agbno == NULLAGBLOCK)
881                 error = -ENOSPC;
882
883         if (error) {
884                 /*
885                  * if extent allocation fails, need to roll the transaction to
886                  * ensure that the AGFL fixup has been committed anyway.
887                  */
888                 xfs_trans_bhold(*tpp, agfbp);
889                 err2 = xfs_trans_roll(tpp);
890                 if (err2)
891                         return err2;
892                 xfs_trans_bjoin(*tpp, agfbp);
893                 goto resv_init_out;
894         }
895
896         /*
897          * if successfully deleted from freespace btrees, need to confirm
898          * per-AG reservation works as expected.
899          */
900         be32_add_cpu(&agi->agi_length, -delta);
901         be32_add_cpu(&agf->agf_length, -delta);
902
903         err2 = xfs_ag_resv_init(pag, *tpp);
904         if (err2) {
905                 be32_add_cpu(&agi->agi_length, delta);
906                 be32_add_cpu(&agf->agf_length, delta);
907                 if (err2 != -ENOSPC)
908                         goto resv_err;
909
910                 __xfs_free_extent_later(*tpp, args.fsbno, delta, NULL, true);
911
912                 /*
913                  * Roll the transaction before trying to re-init the per-ag
914                  * reservation. The new transaction is clean so it will cancel
915                  * without any side effects.
916                  */
917                 error = xfs_defer_finish(tpp);
918                 if (error)
919                         return error;
920
921                 error = -ENOSPC;
922                 goto resv_init_out;
923         }
924         xfs_ialloc_log_agi(*tpp, agibp, XFS_AGI_LENGTH);
925         xfs_alloc_log_agf(*tpp, agfbp, XFS_AGF_LENGTH);
926         return 0;
927
928 resv_init_out:
929         err2 = xfs_ag_resv_init(pag, *tpp);
930         if (!err2)
931                 return error;
932 resv_err:
933         xfs_warn(mp, "Error %d reserving per-AG metadata reserve pool.", err2);
934         xfs_force_shutdown(mp, SHUTDOWN_CORRUPT_INCORE);
935         return err2;
936 }
937
938 /*
939  * Extent the AG indicated by the @id by the length passed in
940  */
941 int
942 xfs_ag_extend_space(
943         struct xfs_perag        *pag,
944         struct xfs_trans        *tp,
945         xfs_extlen_t            len)
946 {
947         struct xfs_buf          *bp;
948         struct xfs_agi          *agi;
949         struct xfs_agf          *agf;
950         int                     error;
951
952         ASSERT(pag->pag_agno == pag->pag_mount->m_sb.sb_agcount - 1);
953
954         error = xfs_ialloc_read_agi(pag, tp, &bp);
955         if (error)
956                 return error;
957
958         agi = bp->b_addr;
959         be32_add_cpu(&agi->agi_length, len);
960         xfs_ialloc_log_agi(tp, bp, XFS_AGI_LENGTH);
961
962         /*
963          * Change agf length.
964          */
965         error = xfs_alloc_read_agf(pag, tp, 0, &bp);
966         if (error)
967                 return error;
968
969         agf = bp->b_addr;
970         be32_add_cpu(&agf->agf_length, len);
971         ASSERT(agf->agf_length == agi->agi_length);
972         xfs_alloc_log_agf(tp, bp, XFS_AGF_LENGTH);
973
974         /*
975          * Free the new space.
976          *
977          * XFS_RMAP_OINFO_SKIP_UPDATE is used here to tell the rmap btree that
978          * this doesn't actually exist in the rmap btree.
979          */
980         error = xfs_rmap_free(tp, bp, pag, be32_to_cpu(agf->agf_length) - len,
981                                 len, &XFS_RMAP_OINFO_SKIP_UPDATE);
982         if (error)
983                 return error;
984
985         error = xfs_free_extent(tp, XFS_AGB_TO_FSB(pag->pag_mount, pag->pag_agno,
986                                         be32_to_cpu(agf->agf_length) - len),
987                                 len, &XFS_RMAP_OINFO_SKIP_UPDATE,
988                                 XFS_AG_RESV_NONE);
989         if (error)
990                 return error;
991
992         /* Update perag geometry */
993         pag->block_count = be32_to_cpu(agf->agf_length);
994         __xfs_agino_range(pag->pag_mount, pag->block_count, &pag->agino_min,
995                                 &pag->agino_max);
996         return 0;
997 }
998
999 /* Retrieve AG geometry. */
1000 int
1001 xfs_ag_get_geometry(
1002         struct xfs_perag        *pag,
1003         struct xfs_ag_geometry  *ageo)
1004 {
1005         struct xfs_buf          *agi_bp;
1006         struct xfs_buf          *agf_bp;
1007         struct xfs_agi          *agi;
1008         struct xfs_agf          *agf;
1009         unsigned int            freeblks;
1010         int                     error;
1011
1012         /* Lock the AG headers. */
1013         error = xfs_ialloc_read_agi(pag, NULL, &agi_bp);
1014         if (error)
1015                 return error;
1016         error = xfs_alloc_read_agf(pag, NULL, 0, &agf_bp);
1017         if (error)
1018                 goto out_agi;
1019
1020         /* Fill out form. */
1021         memset(ageo, 0, sizeof(*ageo));
1022         ageo->ag_number = pag->pag_agno;
1023
1024         agi = agi_bp->b_addr;
1025         ageo->ag_icount = be32_to_cpu(agi->agi_count);
1026         ageo->ag_ifree = be32_to_cpu(agi->agi_freecount);
1027
1028         agf = agf_bp->b_addr;
1029         ageo->ag_length = be32_to_cpu(agf->agf_length);
1030         freeblks = pag->pagf_freeblks +
1031                    pag->pagf_flcount +
1032                    pag->pagf_btreeblks -
1033                    xfs_ag_resv_needed(pag, XFS_AG_RESV_NONE);
1034         ageo->ag_freeblks = freeblks;
1035         xfs_ag_geom_health(pag, ageo);
1036
1037         /* Release resources. */
1038         xfs_buf_relse(agf_bp);
1039 out_agi:
1040         xfs_buf_relse(agi_bp);
1041         return error;
1042 }