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