cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

xfs_iwalk.c (21088B)


      1// SPDX-License-Identifier: GPL-2.0-or-later
      2/*
      3 * Copyright (C) 2019 Oracle.  All Rights Reserved.
      4 * Author: Darrick J. Wong <darrick.wong@oracle.com>
      5 */
      6#include "xfs.h"
      7#include "xfs_fs.h"
      8#include "xfs_shared.h"
      9#include "xfs_format.h"
     10#include "xfs_log_format.h"
     11#include "xfs_trans_resv.h"
     12#include "xfs_mount.h"
     13#include "xfs_inode.h"
     14#include "xfs_btree.h"
     15#include "xfs_ialloc.h"
     16#include "xfs_ialloc_btree.h"
     17#include "xfs_iwalk.h"
     18#include "xfs_error.h"
     19#include "xfs_trace.h"
     20#include "xfs_icache.h"
     21#include "xfs_health.h"
     22#include "xfs_trans.h"
     23#include "xfs_pwork.h"
     24#include "xfs_ag.h"
     25
     26/*
     27 * Walking Inodes in the Filesystem
     28 * ================================
     29 *
     30 * This iterator function walks a subset of filesystem inodes in increasing
     31 * order from @startino until there are no more inodes.  For each allocated
     32 * inode it finds, it calls a walk function with the relevant inode number and
     33 * a pointer to caller-provided data.  The walk function can return the usual
     34 * negative error code to stop the iteration; 0 to continue the iteration; or
     35 * -ECANCELED to stop the iteration.  This return value is returned to the
     36 * caller.
     37 *
     38 * Internally, we allow the walk function to do anything, which means that we
     39 * cannot maintain the inobt cursor or our lock on the AGI buffer.  We
     40 * therefore cache the inobt records in kernel memory and only call the walk
     41 * function when our memory buffer is full.  @nr_recs is the number of records
     42 * that we've cached, and @sz_recs is the size of our cache.
     43 *
     44 * It is the responsibility of the walk function to ensure it accesses
     45 * allocated inodes, as the inobt records may be stale by the time they are
     46 * acted upon.
     47 */
     48
     49struct xfs_iwalk_ag {
     50	/* parallel work control data; will be null if single threaded */
     51	struct xfs_pwork		pwork;
     52
     53	struct xfs_mount		*mp;
     54	struct xfs_trans		*tp;
     55	struct xfs_perag		*pag;
     56
     57	/* Where do we start the traversal? */
     58	xfs_ino_t			startino;
     59
     60	/* What was the last inode number we saw when iterating the inobt? */
     61	xfs_ino_t			lastino;
     62
     63	/* Array of inobt records we cache. */
     64	struct xfs_inobt_rec_incore	*recs;
     65
     66	/* Number of entries allocated for the @recs array. */
     67	unsigned int			sz_recs;
     68
     69	/* Number of entries in the @recs array that are in use. */
     70	unsigned int			nr_recs;
     71
     72	/* Inode walk function and data pointer. */
     73	xfs_iwalk_fn			iwalk_fn;
     74	xfs_inobt_walk_fn		inobt_walk_fn;
     75	void				*data;
     76
     77	/*
     78	 * Make it look like the inodes up to startino are free so that
     79	 * bulkstat can start its inode iteration at the correct place without
     80	 * needing to special case everywhere.
     81	 */
     82	unsigned int			trim_start:1;
     83
     84	/* Skip empty inobt records? */
     85	unsigned int			skip_empty:1;
     86
     87	/* Drop the (hopefully empty) transaction when calling iwalk_fn. */
     88	unsigned int			drop_trans:1;
     89};
     90
     91/*
     92 * Loop over all clusters in a chunk for a given incore inode allocation btree
     93 * record.  Do a readahead if there are any allocated inodes in that cluster.
     94 */
     95STATIC void
     96xfs_iwalk_ichunk_ra(
     97	struct xfs_mount		*mp,
     98	struct xfs_perag		*pag,
     99	struct xfs_inobt_rec_incore	*irec)
    100{
    101	struct xfs_ino_geometry		*igeo = M_IGEO(mp);
    102	xfs_agblock_t			agbno;
    103	struct blk_plug			plug;
    104	int				i;	/* inode chunk index */
    105
    106	agbno = XFS_AGINO_TO_AGBNO(mp, irec->ir_startino);
    107
    108	blk_start_plug(&plug);
    109	for (i = 0; i < XFS_INODES_PER_CHUNK; i += igeo->inodes_per_cluster) {
    110		xfs_inofree_t	imask;
    111
    112		imask = xfs_inobt_maskn(i, igeo->inodes_per_cluster);
    113		if (imask & ~irec->ir_free) {
    114			xfs_btree_reada_bufs(mp, pag->pag_agno, agbno,
    115					igeo->blocks_per_cluster,
    116					&xfs_inode_buf_ops);
    117		}
    118		agbno += igeo->blocks_per_cluster;
    119	}
    120	blk_finish_plug(&plug);
    121}
    122
    123/*
    124 * Set the bits in @irec's free mask that correspond to the inodes before
    125 * @agino so that we skip them.  This is how we restart an inode walk that was
    126 * interrupted in the middle of an inode record.
    127 */
    128STATIC void
    129xfs_iwalk_adjust_start(
    130	xfs_agino_t			agino,	/* starting inode of chunk */
    131	struct xfs_inobt_rec_incore	*irec)	/* btree record */
    132{
    133	int				idx;	/* index into inode chunk */
    134	int				i;
    135
    136	idx = agino - irec->ir_startino;
    137
    138	/*
    139	 * We got a right chunk with some left inodes allocated at it.  Grab
    140	 * the chunk record.  Mark all the uninteresting inodes free because
    141	 * they're before our start point.
    142	 */
    143	for (i = 0; i < idx; i++) {
    144		if (XFS_INOBT_MASK(i) & ~irec->ir_free)
    145			irec->ir_freecount++;
    146	}
    147
    148	irec->ir_free |= xfs_inobt_maskn(0, idx);
    149}
    150
    151/* Allocate memory for a walk. */
    152STATIC int
    153xfs_iwalk_alloc(
    154	struct xfs_iwalk_ag	*iwag)
    155{
    156	size_t			size;
    157
    158	ASSERT(iwag->recs == NULL);
    159	iwag->nr_recs = 0;
    160
    161	/* Allocate a prefetch buffer for inobt records. */
    162	size = iwag->sz_recs * sizeof(struct xfs_inobt_rec_incore);
    163	iwag->recs = kmem_alloc(size, KM_MAYFAIL);
    164	if (iwag->recs == NULL)
    165		return -ENOMEM;
    166
    167	return 0;
    168}
    169
    170/* Free memory we allocated for a walk. */
    171STATIC void
    172xfs_iwalk_free(
    173	struct xfs_iwalk_ag	*iwag)
    174{
    175	kmem_free(iwag->recs);
    176	iwag->recs = NULL;
    177}
    178
    179/* For each inuse inode in each cached inobt record, call our function. */
    180STATIC int
    181xfs_iwalk_ag_recs(
    182	struct xfs_iwalk_ag	*iwag)
    183{
    184	struct xfs_mount	*mp = iwag->mp;
    185	struct xfs_trans	*tp = iwag->tp;
    186	struct xfs_perag	*pag = iwag->pag;
    187	xfs_ino_t		ino;
    188	unsigned int		i, j;
    189	int			error;
    190
    191	for (i = 0; i < iwag->nr_recs; i++) {
    192		struct xfs_inobt_rec_incore	*irec = &iwag->recs[i];
    193
    194		trace_xfs_iwalk_ag_rec(mp, pag->pag_agno, irec);
    195
    196		if (xfs_pwork_want_abort(&iwag->pwork))
    197			return 0;
    198
    199		if (iwag->inobt_walk_fn) {
    200			error = iwag->inobt_walk_fn(mp, tp, pag->pag_agno, irec,
    201					iwag->data);
    202			if (error)
    203				return error;
    204		}
    205
    206		if (!iwag->iwalk_fn)
    207			continue;
    208
    209		for (j = 0; j < XFS_INODES_PER_CHUNK; j++) {
    210			if (xfs_pwork_want_abort(&iwag->pwork))
    211				return 0;
    212
    213			/* Skip if this inode is free */
    214			if (XFS_INOBT_MASK(j) & irec->ir_free)
    215				continue;
    216
    217			/* Otherwise call our function. */
    218			ino = XFS_AGINO_TO_INO(mp, pag->pag_agno,
    219						irec->ir_startino + j);
    220			error = iwag->iwalk_fn(mp, tp, ino, iwag->data);
    221			if (error)
    222				return error;
    223		}
    224	}
    225
    226	return 0;
    227}
    228
    229/* Delete cursor and let go of AGI. */
    230static inline void
    231xfs_iwalk_del_inobt(
    232	struct xfs_trans	*tp,
    233	struct xfs_btree_cur	**curpp,
    234	struct xfs_buf		**agi_bpp,
    235	int			error)
    236{
    237	if (*curpp) {
    238		xfs_btree_del_cursor(*curpp, error);
    239		*curpp = NULL;
    240	}
    241	if (*agi_bpp) {
    242		xfs_trans_brelse(tp, *agi_bpp);
    243		*agi_bpp = NULL;
    244	}
    245}
    246
    247/*
    248 * Set ourselves up for walking inobt records starting from a given point in
    249 * the filesystem.
    250 *
    251 * If caller passed in a nonzero start inode number, load the record from the
    252 * inobt and make the record look like all the inodes before agino are free so
    253 * that we skip them, and then move the cursor to the next inobt record.  This
    254 * is how we support starting an iwalk in the middle of an inode chunk.
    255 *
    256 * If the caller passed in a start number of zero, move the cursor to the first
    257 * inobt record.
    258 *
    259 * The caller is responsible for cleaning up the cursor and buffer pointer
    260 * regardless of the error status.
    261 */
    262STATIC int
    263xfs_iwalk_ag_start(
    264	struct xfs_iwalk_ag	*iwag,
    265	xfs_agino_t		agino,
    266	struct xfs_btree_cur	**curpp,
    267	struct xfs_buf		**agi_bpp,
    268	int			*has_more)
    269{
    270	struct xfs_mount	*mp = iwag->mp;
    271	struct xfs_trans	*tp = iwag->tp;
    272	struct xfs_perag	*pag = iwag->pag;
    273	struct xfs_inobt_rec_incore *irec;
    274	int			error;
    275
    276	/* Set up a fresh cursor and empty the inobt cache. */
    277	iwag->nr_recs = 0;
    278	error = xfs_inobt_cur(mp, tp, pag, XFS_BTNUM_INO, curpp, agi_bpp);
    279	if (error)
    280		return error;
    281
    282	/* Starting at the beginning of the AG?  That's easy! */
    283	if (agino == 0)
    284		return xfs_inobt_lookup(*curpp, 0, XFS_LOOKUP_GE, has_more);
    285
    286	/*
    287	 * Otherwise, we have to grab the inobt record where we left off, stuff
    288	 * the record into our cache, and then see if there are more records.
    289	 * We require a lookup cache of at least two elements so that the
    290	 * caller doesn't have to deal with tearing down the cursor to walk the
    291	 * records.
    292	 */
    293	error = xfs_inobt_lookup(*curpp, agino, XFS_LOOKUP_LE, has_more);
    294	if (error)
    295		return error;
    296
    297	/*
    298	 * If the LE lookup at @agino yields no records, jump ahead to the
    299	 * inobt cursor increment to see if there are more records to process.
    300	 */
    301	if (!*has_more)
    302		goto out_advance;
    303
    304	/* Get the record, should always work */
    305	irec = &iwag->recs[iwag->nr_recs];
    306	error = xfs_inobt_get_rec(*curpp, irec, has_more);
    307	if (error)
    308		return error;
    309	if (XFS_IS_CORRUPT(mp, *has_more != 1))
    310		return -EFSCORRUPTED;
    311
    312	iwag->lastino = XFS_AGINO_TO_INO(mp, pag->pag_agno,
    313				irec->ir_startino + XFS_INODES_PER_CHUNK - 1);
    314
    315	/*
    316	 * If the LE lookup yielded an inobt record before the cursor position,
    317	 * skip it and see if there's another one after it.
    318	 */
    319	if (irec->ir_startino + XFS_INODES_PER_CHUNK <= agino)
    320		goto out_advance;
    321
    322	/*
    323	 * If agino fell in the middle of the inode record, make it look like
    324	 * the inodes up to agino are free so that we don't return them again.
    325	 */
    326	if (iwag->trim_start)
    327		xfs_iwalk_adjust_start(agino, irec);
    328
    329	/*
    330	 * The prefetch calculation is supposed to give us a large enough inobt
    331	 * record cache that grab_ichunk can stage a partial first record and
    332	 * the loop body can cache a record without having to check for cache
    333	 * space until after it reads an inobt record.
    334	 */
    335	iwag->nr_recs++;
    336	ASSERT(iwag->nr_recs < iwag->sz_recs);
    337
    338out_advance:
    339	return xfs_btree_increment(*curpp, 0, has_more);
    340}
    341
    342/*
    343 * The inobt record cache is full, so preserve the inobt cursor state and
    344 * run callbacks on the cached inobt records.  When we're done, restore the
    345 * cursor state to wherever the cursor would have been had the cache not been
    346 * full (and therefore we could've just incremented the cursor) if *@has_more
    347 * is true.  On exit, *@has_more will indicate whether or not the caller should
    348 * try for more inode records.
    349 */
    350STATIC int
    351xfs_iwalk_run_callbacks(
    352	struct xfs_iwalk_ag		*iwag,
    353	struct xfs_btree_cur		**curpp,
    354	struct xfs_buf			**agi_bpp,
    355	int				*has_more)
    356{
    357	struct xfs_mount		*mp = iwag->mp;
    358	struct xfs_inobt_rec_incore	*irec;
    359	xfs_agino_t			next_agino;
    360	int				error;
    361
    362	next_agino = XFS_INO_TO_AGINO(mp, iwag->lastino) + 1;
    363
    364	ASSERT(iwag->nr_recs > 0);
    365
    366	/* Delete cursor but remember the last record we cached... */
    367	xfs_iwalk_del_inobt(iwag->tp, curpp, agi_bpp, 0);
    368	irec = &iwag->recs[iwag->nr_recs - 1];
    369	ASSERT(next_agino >= irec->ir_startino + XFS_INODES_PER_CHUNK);
    370
    371	if (iwag->drop_trans) {
    372		xfs_trans_cancel(iwag->tp);
    373		iwag->tp = NULL;
    374	}
    375
    376	error = xfs_iwalk_ag_recs(iwag);
    377	if (error)
    378		return error;
    379
    380	/* ...empty the cache... */
    381	iwag->nr_recs = 0;
    382
    383	if (!has_more)
    384		return 0;
    385
    386	if (iwag->drop_trans) {
    387		error = xfs_trans_alloc_empty(mp, &iwag->tp);
    388		if (error)
    389			return error;
    390	}
    391
    392	/* ...and recreate the cursor just past where we left off. */
    393	error = xfs_inobt_cur(mp, iwag->tp, iwag->pag, XFS_BTNUM_INO, curpp,
    394			agi_bpp);
    395	if (error)
    396		return error;
    397
    398	return xfs_inobt_lookup(*curpp, next_agino, XFS_LOOKUP_GE, has_more);
    399}
    400
    401/* Walk all inodes in a single AG, from @iwag->startino to the end of the AG. */
    402STATIC int
    403xfs_iwalk_ag(
    404	struct xfs_iwalk_ag		*iwag)
    405{
    406	struct xfs_mount		*mp = iwag->mp;
    407	struct xfs_perag		*pag = iwag->pag;
    408	struct xfs_buf			*agi_bp = NULL;
    409	struct xfs_btree_cur		*cur = NULL;
    410	xfs_agino_t			agino;
    411	int				has_more;
    412	int				error = 0;
    413
    414	/* Set up our cursor at the right place in the inode btree. */
    415	ASSERT(pag->pag_agno == XFS_INO_TO_AGNO(mp, iwag->startino));
    416	agino = XFS_INO_TO_AGINO(mp, iwag->startino);
    417	error = xfs_iwalk_ag_start(iwag, agino, &cur, &agi_bp, &has_more);
    418
    419	while (!error && has_more) {
    420		struct xfs_inobt_rec_incore	*irec;
    421		xfs_ino_t			rec_fsino;
    422
    423		cond_resched();
    424		if (xfs_pwork_want_abort(&iwag->pwork))
    425			goto out;
    426
    427		/* Fetch the inobt record. */
    428		irec = &iwag->recs[iwag->nr_recs];
    429		error = xfs_inobt_get_rec(cur, irec, &has_more);
    430		if (error || !has_more)
    431			break;
    432
    433		/* Make sure that we always move forward. */
    434		rec_fsino = XFS_AGINO_TO_INO(mp, pag->pag_agno, irec->ir_startino);
    435		if (iwag->lastino != NULLFSINO &&
    436		    XFS_IS_CORRUPT(mp, iwag->lastino >= rec_fsino)) {
    437			error = -EFSCORRUPTED;
    438			goto out;
    439		}
    440		iwag->lastino = rec_fsino + XFS_INODES_PER_CHUNK - 1;
    441
    442		/* No allocated inodes in this chunk; skip it. */
    443		if (iwag->skip_empty && irec->ir_freecount == irec->ir_count) {
    444			error = xfs_btree_increment(cur, 0, &has_more);
    445			if (error)
    446				break;
    447			continue;
    448		}
    449
    450		/*
    451		 * Start readahead for this inode chunk in anticipation of
    452		 * walking the inodes.
    453		 */
    454		if (iwag->iwalk_fn)
    455			xfs_iwalk_ichunk_ra(mp, pag, irec);
    456
    457		/*
    458		 * If there's space in the buffer for more records, increment
    459		 * the btree cursor and grab more.
    460		 */
    461		if (++iwag->nr_recs < iwag->sz_recs) {
    462			error = xfs_btree_increment(cur, 0, &has_more);
    463			if (error || !has_more)
    464				break;
    465			continue;
    466		}
    467
    468		/*
    469		 * Otherwise, we need to save cursor state and run the callback
    470		 * function on the cached records.  The run_callbacks function
    471		 * is supposed to return a cursor pointing to the record where
    472		 * we would be if we had been able to increment like above.
    473		 */
    474		ASSERT(has_more);
    475		error = xfs_iwalk_run_callbacks(iwag, &cur, &agi_bp, &has_more);
    476	}
    477
    478	if (iwag->nr_recs == 0 || error)
    479		goto out;
    480
    481	/* Walk the unprocessed records in the cache. */
    482	error = xfs_iwalk_run_callbacks(iwag, &cur, &agi_bp, &has_more);
    483
    484out:
    485	xfs_iwalk_del_inobt(iwag->tp, &cur, &agi_bp, error);
    486	return error;
    487}
    488
    489/*
    490 * We experimentally determined that the reduction in ioctl call overhead
    491 * diminishes when userspace asks for more than 2048 inodes, so we'll cap
    492 * prefetch at this point.
    493 */
    494#define IWALK_MAX_INODE_PREFETCH	(2048U)
    495
    496/*
    497 * Given the number of inodes to prefetch, set the number of inobt records that
    498 * we cache in memory, which controls the number of inodes we try to read
    499 * ahead.  Set the maximum if @inodes == 0.
    500 */
    501static inline unsigned int
    502xfs_iwalk_prefetch(
    503	unsigned int		inodes)
    504{
    505	unsigned int		inobt_records;
    506
    507	/*
    508	 * If the caller didn't tell us the number of inodes they wanted,
    509	 * assume the maximum prefetch possible for best performance.
    510	 * Otherwise, cap prefetch at that maximum so that we don't start an
    511	 * absurd amount of prefetch.
    512	 */
    513	if (inodes == 0)
    514		inodes = IWALK_MAX_INODE_PREFETCH;
    515	inodes = min(inodes, IWALK_MAX_INODE_PREFETCH);
    516
    517	/* Round the inode count up to a full chunk. */
    518	inodes = round_up(inodes, XFS_INODES_PER_CHUNK);
    519
    520	/*
    521	 * In order to convert the number of inodes to prefetch into an
    522	 * estimate of the number of inobt records to cache, we require a
    523	 * conversion factor that reflects our expectations of the average
    524	 * loading factor of an inode chunk.  Based on data gathered, most
    525	 * (but not all) filesystems manage to keep the inode chunks totally
    526	 * full, so we'll underestimate slightly so that our readahead will
    527	 * still deliver the performance we want on aging filesystems:
    528	 *
    529	 * inobt = inodes / (INODES_PER_CHUNK * (4 / 5));
    530	 *
    531	 * The funny math is to avoid integer division.
    532	 */
    533	inobt_records = (inodes * 5) / (4 * XFS_INODES_PER_CHUNK);
    534
    535	/*
    536	 * Allocate enough space to prefetch at least two inobt records so that
    537	 * we can cache both the record where the iwalk started and the next
    538	 * record.  This simplifies the AG inode walk loop setup code.
    539	 */
    540	return max(inobt_records, 2U);
    541}
    542
    543/*
    544 * Walk all inodes in the filesystem starting from @startino.  The @iwalk_fn
    545 * will be called for each allocated inode, being passed the inode's number and
    546 * @data.  @max_prefetch controls how many inobt records' worth of inodes we
    547 * try to readahead.
    548 */
    549int
    550xfs_iwalk(
    551	struct xfs_mount	*mp,
    552	struct xfs_trans	*tp,
    553	xfs_ino_t		startino,
    554	unsigned int		flags,
    555	xfs_iwalk_fn		iwalk_fn,
    556	unsigned int		inode_records,
    557	void			*data)
    558{
    559	struct xfs_iwalk_ag	iwag = {
    560		.mp		= mp,
    561		.tp		= tp,
    562		.iwalk_fn	= iwalk_fn,
    563		.data		= data,
    564		.startino	= startino,
    565		.sz_recs	= xfs_iwalk_prefetch(inode_records),
    566		.trim_start	= 1,
    567		.skip_empty	= 1,
    568		.pwork		= XFS_PWORK_SINGLE_THREADED,
    569		.lastino	= NULLFSINO,
    570	};
    571	struct xfs_perag	*pag;
    572	xfs_agnumber_t		agno = XFS_INO_TO_AGNO(mp, startino);
    573	int			error;
    574
    575	ASSERT(agno < mp->m_sb.sb_agcount);
    576	ASSERT(!(flags & ~XFS_IWALK_FLAGS_ALL));
    577
    578	error = xfs_iwalk_alloc(&iwag);
    579	if (error)
    580		return error;
    581
    582	for_each_perag_from(mp, agno, pag) {
    583		iwag.pag = pag;
    584		error = xfs_iwalk_ag(&iwag);
    585		if (error)
    586			break;
    587		iwag.startino = XFS_AGINO_TO_INO(mp, agno + 1, 0);
    588		if (flags & XFS_INOBT_WALK_SAME_AG)
    589			break;
    590		iwag.pag = NULL;
    591	}
    592
    593	if (iwag.pag)
    594		xfs_perag_put(pag);
    595	xfs_iwalk_free(&iwag);
    596	return error;
    597}
    598
    599/* Run per-thread iwalk work. */
    600static int
    601xfs_iwalk_ag_work(
    602	struct xfs_mount	*mp,
    603	struct xfs_pwork	*pwork)
    604{
    605	struct xfs_iwalk_ag	*iwag;
    606	int			error = 0;
    607
    608	iwag = container_of(pwork, struct xfs_iwalk_ag, pwork);
    609	if (xfs_pwork_want_abort(pwork))
    610		goto out;
    611
    612	error = xfs_iwalk_alloc(iwag);
    613	if (error)
    614		goto out;
    615	/*
    616	 * Grab an empty transaction so that we can use its recursive buffer
    617	 * locking abilities to detect cycles in the inobt without deadlocking.
    618	 */
    619	error = xfs_trans_alloc_empty(mp, &iwag->tp);
    620	if (error)
    621		goto out;
    622	iwag->drop_trans = 1;
    623
    624	error = xfs_iwalk_ag(iwag);
    625	if (iwag->tp)
    626		xfs_trans_cancel(iwag->tp);
    627	xfs_iwalk_free(iwag);
    628out:
    629	xfs_perag_put(iwag->pag);
    630	kmem_free(iwag);
    631	return error;
    632}
    633
    634/*
    635 * Walk all the inodes in the filesystem using multiple threads to process each
    636 * AG.
    637 */
    638int
    639xfs_iwalk_threaded(
    640	struct xfs_mount	*mp,
    641	xfs_ino_t		startino,
    642	unsigned int		flags,
    643	xfs_iwalk_fn		iwalk_fn,
    644	unsigned int		inode_records,
    645	bool			polled,
    646	void			*data)
    647{
    648	struct xfs_pwork_ctl	pctl;
    649	struct xfs_perag	*pag;
    650	xfs_agnumber_t		agno = XFS_INO_TO_AGNO(mp, startino);
    651	int			error;
    652
    653	ASSERT(agno < mp->m_sb.sb_agcount);
    654	ASSERT(!(flags & ~XFS_IWALK_FLAGS_ALL));
    655
    656	error = xfs_pwork_init(mp, &pctl, xfs_iwalk_ag_work, "xfs_iwalk");
    657	if (error)
    658		return error;
    659
    660	for_each_perag_from(mp, agno, pag) {
    661		struct xfs_iwalk_ag	*iwag;
    662
    663		if (xfs_pwork_ctl_want_abort(&pctl))
    664			break;
    665
    666		iwag = kmem_zalloc(sizeof(struct xfs_iwalk_ag), 0);
    667		iwag->mp = mp;
    668
    669		/*
    670		 * perag is being handed off to async work, so take another
    671		 * reference for the async work to release.
    672		 */
    673		atomic_inc(&pag->pag_ref);
    674		iwag->pag = pag;
    675		iwag->iwalk_fn = iwalk_fn;
    676		iwag->data = data;
    677		iwag->startino = startino;
    678		iwag->sz_recs = xfs_iwalk_prefetch(inode_records);
    679		iwag->lastino = NULLFSINO;
    680		xfs_pwork_queue(&pctl, &iwag->pwork);
    681		startino = XFS_AGINO_TO_INO(mp, pag->pag_agno + 1, 0);
    682		if (flags & XFS_INOBT_WALK_SAME_AG)
    683			break;
    684	}
    685	if (pag)
    686		xfs_perag_put(pag);
    687	if (polled)
    688		xfs_pwork_poll(&pctl);
    689	return xfs_pwork_destroy(&pctl);
    690}
    691
    692/*
    693 * Allow callers to cache up to a page's worth of inobt records.  This reflects
    694 * the existing inumbers prefetching behavior.  Since the inobt walk does not
    695 * itself do anything with the inobt records, we can set a fairly high limit
    696 * here.
    697 */
    698#define MAX_INOBT_WALK_PREFETCH	\
    699	(PAGE_SIZE / sizeof(struct xfs_inobt_rec_incore))
    700
    701/*
    702 * Given the number of records that the user wanted, set the number of inobt
    703 * records that we buffer in memory.  Set the maximum if @inobt_records == 0.
    704 */
    705static inline unsigned int
    706xfs_inobt_walk_prefetch(
    707	unsigned int		inobt_records)
    708{
    709	/*
    710	 * If the caller didn't tell us the number of inobt records they
    711	 * wanted, assume the maximum prefetch possible for best performance.
    712	 */
    713	if (inobt_records == 0)
    714		inobt_records = MAX_INOBT_WALK_PREFETCH;
    715
    716	/*
    717	 * Allocate enough space to prefetch at least two inobt records so that
    718	 * we can cache both the record where the iwalk started and the next
    719	 * record.  This simplifies the AG inode walk loop setup code.
    720	 */
    721	inobt_records = max(inobt_records, 2U);
    722
    723	/*
    724	 * Cap prefetch at that maximum so that we don't use an absurd amount
    725	 * of memory.
    726	 */
    727	return min_t(unsigned int, inobt_records, MAX_INOBT_WALK_PREFETCH);
    728}
    729
    730/*
    731 * Walk all inode btree records in the filesystem starting from @startino.  The
    732 * @inobt_walk_fn will be called for each btree record, being passed the incore
    733 * record and @data.  @max_prefetch controls how many inobt records we try to
    734 * cache ahead of time.
    735 */
    736int
    737xfs_inobt_walk(
    738	struct xfs_mount	*mp,
    739	struct xfs_trans	*tp,
    740	xfs_ino_t		startino,
    741	unsigned int		flags,
    742	xfs_inobt_walk_fn	inobt_walk_fn,
    743	unsigned int		inobt_records,
    744	void			*data)
    745{
    746	struct xfs_iwalk_ag	iwag = {
    747		.mp		= mp,
    748		.tp		= tp,
    749		.inobt_walk_fn	= inobt_walk_fn,
    750		.data		= data,
    751		.startino	= startino,
    752		.sz_recs	= xfs_inobt_walk_prefetch(inobt_records),
    753		.pwork		= XFS_PWORK_SINGLE_THREADED,
    754		.lastino	= NULLFSINO,
    755	};
    756	struct xfs_perag	*pag;
    757	xfs_agnumber_t		agno = XFS_INO_TO_AGNO(mp, startino);
    758	int			error;
    759
    760	ASSERT(agno < mp->m_sb.sb_agcount);
    761	ASSERT(!(flags & ~XFS_INOBT_WALK_FLAGS_ALL));
    762
    763	error = xfs_iwalk_alloc(&iwag);
    764	if (error)
    765		return error;
    766
    767	for_each_perag_from(mp, agno, pag) {
    768		iwag.pag = pag;
    769		error = xfs_iwalk_ag(&iwag);
    770		if (error)
    771			break;
    772		iwag.startino = XFS_AGINO_TO_INO(mp, pag->pag_agno + 1, 0);
    773		if (flags & XFS_INOBT_WALK_SAME_AG)
    774			break;
    775		iwag.pag = NULL;
    776	}
    777
    778	if (iwag.pag)
    779		xfs_perag_put(pag);
    780	xfs_iwalk_free(&iwag);
    781	return error;
    782}