xfs_qm.c 46.5 KB
Newer Older
L
Linus Torvalds 已提交
1
/*
2 3
 * Copyright (c) 2000-2005 Silicon Graphics, Inc.
 * All Rights Reserved.
L
Linus Torvalds 已提交
4
 *
5 6
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
L
Linus Torvalds 已提交
7 8
 * published by the Free Software Foundation.
 *
9 10 11 12
 * This program is distributed in the hope that it would be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
L
Linus Torvalds 已提交
13
 *
14 15 16
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write the Free Software Foundation,
 * Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
L
Linus Torvalds 已提交
17 18 19
 */
#include "xfs.h"
#include "xfs_fs.h"
20
#include "xfs_shared.h"
21
#include "xfs_format.h"
22 23
#include "xfs_log_format.h"
#include "xfs_trans_resv.h"
24
#include "xfs_bit.h"
L
Linus Torvalds 已提交
25 26 27
#include "xfs_sb.h"
#include "xfs_mount.h"
#include "xfs_inode.h"
28 29
#include "xfs_ialloc.h"
#include "xfs_itable.h"
30
#include "xfs_quota.h"
L
Linus Torvalds 已提交
31
#include "xfs_error.h"
32
#include "xfs_bmap.h"
33
#include "xfs_bmap_btree.h"
34
#include "xfs_trans.h"
L
Linus Torvalds 已提交
35 36
#include "xfs_trans_space.h"
#include "xfs_qm.h"
C
Christoph Hellwig 已提交
37
#include "xfs_trace.h"
D
Dave Chinner 已提交
38
#include "xfs_icache.h"
D
Dave Chinner 已提交
39
#include "xfs_cksum.h"
L
Linus Torvalds 已提交
40 41 42 43 44 45 46 47

/*
 * The global quota manager. There is only one of these for the entire
 * system, _not_ one per file system. XQM keeps track of the overall
 * quota functionality, including maintaining the freelist and hash
 * tables of dquots.
 */
STATIC int	xfs_qm_init_quotainos(xfs_mount_t *);
48
STATIC int	xfs_qm_init_quotainfo(xfs_mount_t *);
L
Linus Torvalds 已提交
49

50 51

STATIC void	xfs_qm_dqfree_one(struct xfs_dquot *dqp);
52 53 54 55 56 57 58 59 60 61 62 63 64
/*
 * We use the batch lookup interface to iterate over the dquots as it
 * currently is the only interface into the radix tree code that allows
 * fuzzy lookups instead of exact matches.  Holding the lock over multiple
 * operations is fine as all callers are used either during mount/umount
 * or quotaoff.
 */
#define XFS_DQ_LOOKUP_BATCH	32

STATIC int
xfs_qm_dquot_walk(
	struct xfs_mount	*mp,
	int			type,
65 66
	int			(*execute)(struct xfs_dquot *dqp, void *data),
	void			*data)
67 68
{
	struct xfs_quotainfo	*qi = mp->m_quotainfo;
69
	struct radix_tree_root	*tree = xfs_dquot_tree(qi, type);
70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
	uint32_t		next_index;
	int			last_error = 0;
	int			skipped;
	int			nr_found;

restart:
	skipped = 0;
	next_index = 0;
	nr_found = 0;

	while (1) {
		struct xfs_dquot *batch[XFS_DQ_LOOKUP_BATCH];
		int		error = 0;
		int		i;

		mutex_lock(&qi->qi_tree_lock);
		nr_found = radix_tree_gang_lookup(tree, (void **)batch,
					next_index, XFS_DQ_LOOKUP_BATCH);
		if (!nr_found) {
			mutex_unlock(&qi->qi_tree_lock);
			break;
		}

		for (i = 0; i < nr_found; i++) {
			struct xfs_dquot *dqp = batch[i];

			next_index = be32_to_cpu(dqp->q_core.d_id) + 1;

98
			error = execute(batch[i], data);
D
Dave Chinner 已提交
99
			if (error == -EAGAIN) {
100 101 102
				skipped++;
				continue;
			}
D
Dave Chinner 已提交
103
			if (error && last_error != -EFSCORRUPTED)
104 105 106 107 108 109
				last_error = error;
		}

		mutex_unlock(&qi->qi_tree_lock);

		/* bail out if the filesystem is corrupted.  */
D
Dave Chinner 已提交
110
		if (last_error == -EFSCORRUPTED) {
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
			skipped = 0;
			break;
		}
	}

	if (skipped) {
		delay(1);
		goto restart;
	}

	return last_error;
}


/*
 * Purge a dquot from all tracking data structures and free it.
 */
STATIC int
xfs_qm_dqpurge(
130 131
	struct xfs_dquot	*dqp,
	void			*data)
132 133 134 135 136 137 138
{
	struct xfs_mount	*mp = dqp->q_mount;
	struct xfs_quotainfo	*qi = mp->m_quotainfo;

	xfs_dqlock(dqp);
	if ((dqp->dq_flags & XFS_DQ_FREEING) || dqp->q_nrefs != 0) {
		xfs_dqunlock(dqp);
D
Dave Chinner 已提交
139
		return -EAGAIN;
140 141 142 143
	}

	dqp->dq_flags |= XFS_DQ_FREEING;

144
	xfs_dqflock(dqp);
145 146 147 148 149 150 151

	/*
	 * If we are turning this type of quotas off, we don't care
	 * about the dirty metadata sitting in this dquot. OTOH, if
	 * we're unmounting, we do care, so we flush it and wait.
	 */
	if (XFS_DQ_IS_DIRTY(dqp)) {
152 153
		struct xfs_buf	*bp = NULL;
		int		error;
154 155 156 157 158

		/*
		 * We don't care about getting disk errors here. We need
		 * to purge this dquot anyway, so we go ahead regardless.
		 */
159 160
		error = xfs_qm_dqflush(dqp, &bp);
		if (error) {
161 162
			xfs_warn(mp, "%s: dquot %p flush failed",
				__func__, dqp);
163 164 165 166
		} else {
			error = xfs_bwrite(bp);
			xfs_buf_relse(bp);
		}
167 168 169 170 171 172 173 174 175 176
		xfs_dqflock(dqp);
	}

	ASSERT(atomic_read(&dqp->q_pincount) == 0);
	ASSERT(XFS_FORCED_SHUTDOWN(mp) ||
	       !(dqp->q_logitem.qli_item.li_flags & XFS_LI_IN_AIL));

	xfs_dqfunlock(dqp);
	xfs_dqunlock(dqp);

177
	radix_tree_delete(xfs_dquot_tree(qi, dqp->q_core.d_flags),
178 179 180 181 182 183 184 185
			  be32_to_cpu(dqp->q_core.d_id));
	qi->qi_dquots--;

	/*
	 * We move dquots to the freelist as soon as their reference count
	 * hits zero, so it really should be on the freelist here.
	 */
	ASSERT(!list_empty(&dqp->q_lru));
186
	list_lru_del(&qi->qi_lru, &dqp->q_lru);
187 188 189
	XFS_STATS_DEC(xs_qm_dquot_unused);

	xfs_qm_dqdestroy(dqp);
190 191 192
	return 0;
}

193 194 195 196 197 198 199 200
/*
 * Purge the dquot cache.
 */
void
xfs_qm_dqpurge_all(
	struct xfs_mount	*mp,
	uint			flags)
{
D
Dave Chinner 已提交
201 202
	if (flags & XFS_QMOPT_UQUOTA)
		xfs_qm_dquot_walk(mp, XFS_DQ_USER, xfs_qm_dqpurge, NULL);
203
	if (flags & XFS_QMOPT_GQUOTA)
204
		xfs_qm_dquot_walk(mp, XFS_DQ_GROUP, xfs_qm_dqpurge, NULL);
205
	if (flags & XFS_QMOPT_PQUOTA)
206
		xfs_qm_dquot_walk(mp, XFS_DQ_PROJ, xfs_qm_dqpurge, NULL);
207 208
}

L
Linus Torvalds 已提交
209 210 211 212
/*
 * Just destroy the quotainfo structure.
 */
void
C
Christoph Hellwig 已提交
213 214
xfs_qm_unmount(
	struct xfs_mount	*mp)
L
Linus Torvalds 已提交
215
{
C
Christoph Hellwig 已提交
216
	if (mp->m_quotainfo) {
217
		xfs_qm_dqpurge_all(mp, XFS_QMOPT_QUOTALL);
L
Linus Torvalds 已提交
218
		xfs_qm_destroy_quotainfo(mp);
C
Christoph Hellwig 已提交
219
	}
L
Linus Torvalds 已提交
220 221 222 223 224
}

/*
 * Called from the vfsops layer.
 */
225
void
L
Linus Torvalds 已提交
226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
xfs_qm_unmount_quotas(
	xfs_mount_t	*mp)
{
	/*
	 * Release the dquots that root inode, et al might be holding,
	 * before we flush quotas and blow away the quotainfo structure.
	 */
	ASSERT(mp->m_rootip);
	xfs_qm_dqdetach(mp->m_rootip);
	if (mp->m_rbmip)
		xfs_qm_dqdetach(mp->m_rbmip);
	if (mp->m_rsumip)
		xfs_qm_dqdetach(mp->m_rsumip);

	/*
241
	 * Release the quota inodes.
L
Linus Torvalds 已提交
242 243
	 */
	if (mp->m_quotainfo) {
244 245 246
		if (mp->m_quotainfo->qi_uquotaip) {
			IRELE(mp->m_quotainfo->qi_uquotaip);
			mp->m_quotainfo->qi_uquotaip = NULL;
L
Linus Torvalds 已提交
247
		}
248 249 250
		if (mp->m_quotainfo->qi_gquotaip) {
			IRELE(mp->m_quotainfo->qi_gquotaip);
			mp->m_quotainfo->qi_gquotaip = NULL;
L
Linus Torvalds 已提交
251
		}
252 253 254 255
		if (mp->m_quotainfo->qi_pquotaip) {
			IRELE(mp->m_quotainfo->qi_pquotaip);
			mp->m_quotainfo->qi_pquotaip = NULL;
		}
L
Linus Torvalds 已提交
256 257 258 259 260 261 262 263 264 265 266 267 268 269
	}
}

STATIC int
xfs_qm_dqattach_one(
	xfs_inode_t	*ip,
	xfs_dqid_t	id,
	uint		type,
	uint		doalloc,
	xfs_dquot_t	**IO_idqpp)
{
	xfs_dquot_t	*dqp;
	int		error;

C
Christoph Hellwig 已提交
270
	ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
L
Linus Torvalds 已提交
271
	error = 0;
272

L
Linus Torvalds 已提交
273
	/*
D
Dave Chinner 已提交
274 275 276
	 * See if we already have it in the inode itself. IO_idqpp is &i_udquot
	 * or &i_gdquot. This made the code look weird, but made the logic a lot
	 * simpler.
L
Linus Torvalds 已提交
277
	 */
278 279
	dqp = *IO_idqpp;
	if (dqp) {
C
Christoph Hellwig 已提交
280
		trace_xfs_dqattach_found(dqp);
281
		return 0;
L
Linus Torvalds 已提交
282 283 284
	}

	/*
D
Dave Chinner 已提交
285 286 287 288
	 * Find the dquot from somewhere. This bumps the reference count of
	 * dquot and returns it locked.  This can return ENOENT if dquot didn't
	 * exist on disk and we didn't ask it to allocate; ESRCH if quotas got
	 * turned off suddenly.
L
Linus Torvalds 已提交
289
	 */
290 291
	error = xfs_qm_dqget(ip->i_mount, ip, id, type,
			     doalloc | XFS_QMOPT_DOWARN, &dqp);
292 293
	if (error)
		return error;
L
Linus Torvalds 已提交
294

C
Christoph Hellwig 已提交
295
	trace_xfs_dqattach_get(dqp);
296

L
Linus Torvalds 已提交
297 298 299 300 301
	/*
	 * dqget may have dropped and re-acquired the ilock, but it guarantees
	 * that the dquot returned is the one that should go in the inode.
	 */
	*IO_idqpp = dqp;
302 303
	xfs_dqunlock(dqp);
	return 0;
L
Linus Torvalds 已提交
304 305
}

306 307 308 309 310 311 312 313 314 315 316 317
static bool
xfs_qm_need_dqattach(
	struct xfs_inode	*ip)
{
	struct xfs_mount	*mp = ip->i_mount;

	if (!XFS_IS_QUOTA_RUNNING(mp))
		return false;
	if (!XFS_IS_QUOTA_ON(mp))
		return false;
	if (!XFS_NOT_DQATTACHED(mp, ip))
		return false;
318
	if (xfs_is_quota_inode(&mp->m_sb, ip->i_ino))
319 320 321
		return false;
	return true;
}
L
Linus Torvalds 已提交
322 323

/*
324 325
 * Given a locked inode, attach dquot(s) to it, taking U/G/P-QUOTAON
 * into account.
L
Linus Torvalds 已提交
326 327 328 329 330
 * If XFS_QMOPT_DQALLOC, the dquot(s) will be allocated if needed.
 * Inode may get unlocked and relocked in here, and the caller must deal with
 * the consequences.
 */
int
C
Christoph Hellwig 已提交
331
xfs_qm_dqattach_locked(
L
Linus Torvalds 已提交
332 333 334 335 336 337
	xfs_inode_t	*ip,
	uint		flags)
{
	xfs_mount_t	*mp = ip->i_mount;
	int		error = 0;

338
	if (!xfs_qm_need_dqattach(ip))
339
		return 0;
L
Linus Torvalds 已提交
340

C
Christoph Hellwig 已提交
341
	ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
L
Linus Torvalds 已提交
342

D
Dave Chinner 已提交
343
	if (XFS_IS_UQUOTA_ON(mp) && !ip->i_udquot) {
L
Linus Torvalds 已提交
344 345
		error = xfs_qm_dqattach_one(ip, ip->i_d.di_uid, XFS_DQ_USER,
						flags & XFS_QMOPT_DQALLOC,
D
Dave Chinner 已提交
346
						&ip->i_udquot);
L
Linus Torvalds 已提交
347 348
		if (error)
			goto done;
D
Dave Chinner 已提交
349
		ASSERT(ip->i_udquot);
L
Linus Torvalds 已提交
350
	}
C
Christoph Hellwig 已提交
351

D
Dave Chinner 已提交
352
	if (XFS_IS_GQUOTA_ON(mp) && !ip->i_gdquot) {
353
		error = xfs_qm_dqattach_one(ip, ip->i_d.di_gid, XFS_DQ_GROUP,
L
Linus Torvalds 已提交
354
						flags & XFS_QMOPT_DQALLOC,
D
Dave Chinner 已提交
355
						&ip->i_gdquot);
L
Linus Torvalds 已提交
356 357
		if (error)
			goto done;
D
Dave Chinner 已提交
358
		ASSERT(ip->i_gdquot);
L
Linus Torvalds 已提交
359 360
	}

D
Dave Chinner 已提交
361
	if (XFS_IS_PQUOTA_ON(mp) && !ip->i_pdquot) {
362 363
		error = xfs_qm_dqattach_one(ip, xfs_get_projid(ip), XFS_DQ_PROJ,
						flags & XFS_QMOPT_DQALLOC,
D
Dave Chinner 已提交
364
						&ip->i_pdquot);
365 366
		if (error)
			goto done;
D
Dave Chinner 已提交
367
		ASSERT(ip->i_pdquot);
368 369
	}

D
Dave Chinner 已提交
370
done:
L
Linus Torvalds 已提交
371
	/*
D
Dave Chinner 已提交
372 373
	 * Don't worry about the dquots that we may have attached before any
	 * error - they'll get detached later if it has not already been done.
L
Linus Torvalds 已提交
374
	 */
C
Christoph Hellwig 已提交
375 376 377
	ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
	return error;
}
L
Linus Torvalds 已提交
378

C
Christoph Hellwig 已提交
379 380 381 382 383 384 385
int
xfs_qm_dqattach(
	struct xfs_inode	*ip,
	uint			flags)
{
	int			error;

386 387 388
	if (!xfs_qm_need_dqattach(ip))
		return 0;

C
Christoph Hellwig 已提交
389 390 391
	xfs_ilock(ip, XFS_ILOCK_EXCL);
	error = xfs_qm_dqattach_locked(ip, flags);
	xfs_iunlock(ip, XFS_ILOCK_EXCL);
L
Linus Torvalds 已提交
392

393
	return error;
L
Linus Torvalds 已提交
394 395 396 397 398 399 400 401 402 403 404
}

/*
 * Release dquots (and their references) if any.
 * The inode should be locked EXCL except when this's called by
 * xfs_ireclaim.
 */
void
xfs_qm_dqdetach(
	xfs_inode_t	*ip)
{
405
	if (!(ip->i_udquot || ip->i_gdquot || ip->i_pdquot))
L
Linus Torvalds 已提交
406 407
		return;

C
Christoph Hellwig 已提交
408 409
	trace_xfs_dquot_dqdetach(ip);

410
	ASSERT(!xfs_is_quota_inode(&ip->i_mount->m_sb, ip->i_ino));
L
Linus Torvalds 已提交
411 412 413 414 415 416 417 418
	if (ip->i_udquot) {
		xfs_qm_dqrele(ip->i_udquot);
		ip->i_udquot = NULL;
	}
	if (ip->i_gdquot) {
		xfs_qm_dqrele(ip->i_gdquot);
		ip->i_gdquot = NULL;
	}
419 420 421 422
	if (ip->i_pdquot) {
		xfs_qm_dqrele(ip->i_pdquot);
		ip->i_pdquot = NULL;
	}
L
Linus Torvalds 已提交
423 424
}

425 426 427 428 429 430 431 432
struct xfs_qm_isolate {
	struct list_head	buffers;
	struct list_head	dispose;
};

static enum lru_status
xfs_qm_dquot_isolate(
	struct list_head	*item,
433
	struct list_lru_one	*lru,
434 435
	spinlock_t		*lru_lock,
	void			*arg)
436
		__releases(lru_lock) __acquires(lru_lock)
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453
{
	struct xfs_dquot	*dqp = container_of(item,
						struct xfs_dquot, q_lru);
	struct xfs_qm_isolate	*isol = arg;

	if (!xfs_dqlock_nowait(dqp))
		goto out_miss_busy;

	/*
	 * This dquot has acquired a reference in the meantime remove it from
	 * the freelist and try again.
	 */
	if (dqp->q_nrefs) {
		xfs_dqunlock(dqp);
		XFS_STATS_INC(xs_qm_dqwants);

		trace_xfs_dqreclaim_want(dqp);
454
		list_lru_isolate(lru, &dqp->q_lru);
455
		XFS_STATS_DEC(xs_qm_dquot_unused);
D
Dave Chinner 已提交
456
		return LRU_REMOVED;
457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497
	}

	/*
	 * If the dquot is dirty, flush it. If it's already being flushed, just
	 * skip it so there is time for the IO to complete before we try to
	 * reclaim it again on the next LRU pass.
	 */
	if (!xfs_dqflock_nowait(dqp)) {
		xfs_dqunlock(dqp);
		goto out_miss_busy;
	}

	if (XFS_DQ_IS_DIRTY(dqp)) {
		struct xfs_buf	*bp = NULL;
		int		error;

		trace_xfs_dqreclaim_dirty(dqp);

		/* we have to drop the LRU lock to flush the dquot */
		spin_unlock(lru_lock);

		error = xfs_qm_dqflush(dqp, &bp);
		if (error) {
			xfs_warn(dqp->q_mount, "%s: dquot %p flush failed",
				 __func__, dqp);
			goto out_unlock_dirty;
		}

		xfs_buf_delwri_queue(bp, &isol->buffers);
		xfs_buf_relse(bp);
		goto out_unlock_dirty;
	}
	xfs_dqfunlock(dqp);

	/*
	 * Prevent lookups now that we are past the point of no return.
	 */
	dqp->dq_flags |= XFS_DQ_FREEING;
	xfs_dqunlock(dqp);

	ASSERT(dqp->q_nrefs == 0);
498
	list_lru_isolate_move(lru, &dqp->q_lru, &isol->dispose);
499 500 501
	XFS_STATS_DEC(xs_qm_dquot_unused);
	trace_xfs_dqreclaim_done(dqp);
	XFS_STATS_INC(xs_qm_dqreclaims);
D
Dave Chinner 已提交
502
	return LRU_REMOVED;
503 504 505 506

out_miss_busy:
	trace_xfs_dqreclaim_busy(dqp);
	XFS_STATS_INC(xs_qm_dqreclaim_misses);
D
Dave Chinner 已提交
507
	return LRU_SKIP;
508 509 510 511

out_unlock_dirty:
	trace_xfs_dqreclaim_busy(dqp);
	XFS_STATS_INC(xs_qm_dqreclaim_misses);
D
Dave Chinner 已提交
512 513 514
	xfs_dqunlock(dqp);
	spin_lock(lru_lock);
	return LRU_RETRY;
515 516
}

517
static unsigned long
518 519 520 521 522 523 524
xfs_qm_shrink_scan(
	struct shrinker		*shrink,
	struct shrink_control	*sc)
{
	struct xfs_quotainfo	*qi = container_of(shrink,
					struct xfs_quotainfo, qi_shrinker);
	struct xfs_qm_isolate	isol;
525
	unsigned long		freed;
526 527 528 529 530 531 532 533
	int			error;

	if ((sc->gfp_mask & (__GFP_FS|__GFP_WAIT)) != (__GFP_FS|__GFP_WAIT))
		return 0;

	INIT_LIST_HEAD(&isol.buffers);
	INIT_LIST_HEAD(&isol.dispose);

534 535
	freed = list_lru_shrink_walk(&qi->qi_lru, sc,
				     xfs_qm_dquot_isolate, &isol);
536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551

	error = xfs_buf_delwri_submit(&isol.buffers);
	if (error)
		xfs_warn(NULL, "%s: dquot reclaim failed", __func__);

	while (!list_empty(&isol.dispose)) {
		struct xfs_dquot	*dqp;

		dqp = list_first_entry(&isol.dispose, struct xfs_dquot, q_lru);
		list_del_init(&dqp->q_lru);
		xfs_qm_dqfree_one(dqp);
	}

	return freed;
}

552
static unsigned long
553 554 555 556 557 558 559
xfs_qm_shrink_count(
	struct shrinker		*shrink,
	struct shrink_control	*sc)
{
	struct xfs_quotainfo	*qi = container_of(shrink,
					struct xfs_quotainfo, qi_shrinker);

560
	return list_lru_shrink_count(&qi->qi_lru, sc);
561 562
}

L
Linus Torvalds 已提交
563 564 565 566
/*
 * This initializes all the quota information that's kept in the
 * mount structure
 */
567
STATIC int
L
Linus Torvalds 已提交
568 569 570 571 572 573 574 575 576 577 578
xfs_qm_init_quotainfo(
	xfs_mount_t	*mp)
{
	xfs_quotainfo_t *qinf;
	int		error;
	xfs_dquot_t	*dqp;

	ASSERT(XFS_IS_QUOTA_RUNNING(mp));

	qinf = mp->m_quotainfo = kmem_zalloc(sizeof(xfs_quotainfo_t), KM_SLEEP);

D
Dave Chinner 已提交
579
	error = list_lru_init(&qinf->qi_lru);
580 581
	if (error)
		goto out_free_qinf;
582

L
Linus Torvalds 已提交
583 584 585 586
	/*
	 * See if quotainodes are setup, and if not, allocate them,
	 * and change the superblock accordingly.
	 */
587 588 589
	error = xfs_qm_init_quotainos(mp);
	if (error)
		goto out_free_lru;
L
Linus Torvalds 已提交
590

591 592
	INIT_RADIX_TREE(&qinf->qi_uquota_tree, GFP_NOFS);
	INIT_RADIX_TREE(&qinf->qi_gquota_tree, GFP_NOFS);
593
	INIT_RADIX_TREE(&qinf->qi_pquota_tree, GFP_NOFS);
594 595
	mutex_init(&qinf->qi_tree_lock);

L
Linus Torvalds 已提交
596
	/* mutex used to serialize quotaoffs */
597
	mutex_init(&qinf->qi_quotaofflock);
L
Linus Torvalds 已提交
598 599 600

	/* Precalc some constants */
	qinf->qi_dqchunklen = XFS_FSB_TO_BB(mp, XFS_DQUOT_CLUSTER_SIZE_FSB);
601
	qinf->qi_dqperchunk = xfs_calc_dquots_per_chunk(qinf->qi_dqchunklen);
L
Linus Torvalds 已提交
602 603 604 605 606 607

	mp->m_qflags |= (mp->m_sb.sb_qflags & XFS_ALL_QUOTA_CHKD);

	/*
	 * We try to get the limits from the superuser's limits fields.
	 * This is quite hacky, but it is standard quota practice.
C
Christoph Hellwig 已提交
608
	 *
L
Linus Torvalds 已提交
609 610 611 612
	 * We look at the USR dquot with id == 0 first, but if user quotas
	 * are not enabled we goto the GRP dquot with id == 0.
	 * We don't really care to keep separate default limits for user
	 * and group quotas, at least not at this point.
C
Christoph Hellwig 已提交
613 614 615
	 *
	 * Since we may not have done a quotacheck by this point, just read
	 * the dquot without attaching it to any hashtables or lists.
L
Linus Torvalds 已提交
616
	 */
C
Christoph Hellwig 已提交
617 618 619 620 621 622
	error = xfs_qm_dqread(mp, 0,
			XFS_IS_UQUOTA_RUNNING(mp) ? XFS_DQ_USER :
			 (XFS_IS_GQUOTA_RUNNING(mp) ? XFS_DQ_GROUP :
			  XFS_DQ_PROJ),
			XFS_QMOPT_DOWARN, &dqp);
	if (!error) {
L
Linus Torvalds 已提交
623 624 625 626 627 628 629
		xfs_disk_dquot_t	*ddqp = &dqp->q_core;

		/*
		 * The warnings and timers set the grace period given to
		 * a user or group before he or she can not perform any
		 * more writing. If it is zero, a default is used.
		 */
630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647
		qinf->qi_btimelimit = ddqp->d_btimer ?
			be32_to_cpu(ddqp->d_btimer) : XFS_QM_BTIMELIMIT;
		qinf->qi_itimelimit = ddqp->d_itimer ?
			be32_to_cpu(ddqp->d_itimer) : XFS_QM_ITIMELIMIT;
		qinf->qi_rtbtimelimit = ddqp->d_rtbtimer ?
			be32_to_cpu(ddqp->d_rtbtimer) : XFS_QM_RTBTIMELIMIT;
		qinf->qi_bwarnlimit = ddqp->d_bwarns ?
			be16_to_cpu(ddqp->d_bwarns) : XFS_QM_BWARNLIMIT;
		qinf->qi_iwarnlimit = ddqp->d_iwarns ?
			be16_to_cpu(ddqp->d_iwarns) : XFS_QM_IWARNLIMIT;
		qinf->qi_rtbwarnlimit = ddqp->d_rtbwarns ?
			be16_to_cpu(ddqp->d_rtbwarns) : XFS_QM_RTBWARNLIMIT;
		qinf->qi_bhardlimit = be64_to_cpu(ddqp->d_blk_hardlimit);
		qinf->qi_bsoftlimit = be64_to_cpu(ddqp->d_blk_softlimit);
		qinf->qi_ihardlimit = be64_to_cpu(ddqp->d_ino_hardlimit);
		qinf->qi_isoftlimit = be64_to_cpu(ddqp->d_ino_softlimit);
		qinf->qi_rtbhardlimit = be64_to_cpu(ddqp->d_rtb_hardlimit);
		qinf->qi_rtbsoftlimit = be64_to_cpu(ddqp->d_rtb_softlimit);
648

L
Linus Torvalds 已提交
649 650 651 652 653 654 655
		xfs_qm_dqdestroy(dqp);
	} else {
		qinf->qi_btimelimit = XFS_QM_BTIMELIMIT;
		qinf->qi_itimelimit = XFS_QM_ITIMELIMIT;
		qinf->qi_rtbtimelimit = XFS_QM_RTBTIMELIMIT;
		qinf->qi_bwarnlimit = XFS_QM_BWARNLIMIT;
		qinf->qi_iwarnlimit = XFS_QM_IWARNLIMIT;
656
		qinf->qi_rtbwarnlimit = XFS_QM_RTBWARNLIMIT;
L
Linus Torvalds 已提交
657 658
	}

659 660
	qinf->qi_shrinker.count_objects = xfs_qm_shrink_count;
	qinf->qi_shrinker.scan_objects = xfs_qm_shrink_scan;
661
	qinf->qi_shrinker.seeks = DEFAULT_SEEKS;
662
	qinf->qi_shrinker.flags = SHRINKER_NUMA_AWARE;
663
	register_shrinker(&qinf->qi_shrinker);
664
	return 0;
665 666 667 668 669 670 671

out_free_lru:
	list_lru_destroy(&qinf->qi_lru);
out_free_qinf:
	kmem_free(qinf);
	mp->m_quotainfo = NULL;
	return error;
L
Linus Torvalds 已提交
672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688
}


/*
 * Gets called when unmounting a filesystem or when all quotas get
 * turned off.
 * This purges the quota inodes, destroys locks and frees itself.
 */
void
xfs_qm_destroy_quotainfo(
	xfs_mount_t	*mp)
{
	xfs_quotainfo_t *qi;

	qi = mp->m_quotainfo;
	ASSERT(qi != NULL);

689
	unregister_shrinker(&qi->qi_shrinker);
G
Glauber Costa 已提交
690
	list_lru_destroy(&qi->qi_lru);
691

L
Linus Torvalds 已提交
692
	if (qi->qi_uquotaip) {
693
		IRELE(qi->qi_uquotaip);
L
Linus Torvalds 已提交
694 695 696
		qi->qi_uquotaip = NULL; /* paranoia */
	}
	if (qi->qi_gquotaip) {
697
		IRELE(qi->qi_gquotaip);
L
Linus Torvalds 已提交
698 699
		qi->qi_gquotaip = NULL;
	}
700 701 702 703
	if (qi->qi_pquotaip) {
		IRELE(qi->qi_pquotaip);
		qi->qi_pquotaip = NULL;
	}
L
Linus Torvalds 已提交
704
	mutex_destroy(&qi->qi_quotaofflock);
705
	kmem_free(qi);
L
Linus Torvalds 已提交
706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721
	mp->m_quotainfo = NULL;
}

/*
 * Create an inode and return with a reference already taken, but unlocked
 * This is how we create quota inodes
 */
STATIC int
xfs_qm_qino_alloc(
	xfs_mount_t	*mp,
	xfs_inode_t	**ip,
	uint		flags)
{
	xfs_trans_t	*tp;
	int		error;
	int		committed;
722
	bool		need_alloc = true;
L
Linus Torvalds 已提交
723

724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750
	*ip = NULL;
	/*
	 * With superblock that doesn't have separate pquotino, we
	 * share an inode between gquota and pquota. If the on-disk
	 * superblock has GQUOTA and the filesystem is now mounted
	 * with PQUOTA, just use sb_gquotino for sb_pquotino and
	 * vice-versa.
	 */
	if (!xfs_sb_version_has_pquotino(&mp->m_sb) &&
			(flags & (XFS_QMOPT_PQUOTA|XFS_QMOPT_GQUOTA))) {
		xfs_ino_t ino = NULLFSINO;

		if ((flags & XFS_QMOPT_PQUOTA) &&
			     (mp->m_sb.sb_gquotino != NULLFSINO)) {
			ino = mp->m_sb.sb_gquotino;
			ASSERT(mp->m_sb.sb_pquotino == NULLFSINO);
		} else if ((flags & XFS_QMOPT_GQUOTA) &&
			     (mp->m_sb.sb_pquotino != NULLFSINO)) {
			ino = mp->m_sb.sb_pquotino;
			ASSERT(mp->m_sb.sb_gquotino == NULLFSINO);
		}
		if (ino != NULLFSINO) {
			error = xfs_iget(mp, NULL, ino, 0, 0, ip);
			if (error)
				return error;
			mp->m_sb.sb_gquotino = NULLFSINO;
			mp->m_sb.sb_pquotino = NULLFSINO;
751
			need_alloc = false;
752 753 754
		}
	}

755
	tp = xfs_trans_alloc(mp, XFS_TRANS_QM_QINOCREATE);
756 757 758
	error = xfs_trans_reserve(tp, &M_RES(mp)->tr_create,
				  XFS_QM_QINOCREATE_SPACE_RES(mp), 0);
	if (error) {
759
		xfs_trans_cancel(tp);
760
		return error;
L
Linus Torvalds 已提交
761 762
	}

763
	if (need_alloc) {
764 765 766
		error = xfs_dir_ialloc(&tp, NULL, S_IFREG, 1, 0, 0, 1, ip,
								&committed);
		if (error) {
767
			xfs_trans_cancel(tp);
768 769
			return error;
		}
L
Linus Torvalds 已提交
770 771 772 773 774 775 776
	}

	/*
	 * Make the changes in the superblock, and log those too.
	 * sbfields arg may contain fields other than *QUOTINO;
	 * VERSIONNUM for example.
	 */
E
Eric Sandeen 已提交
777
	spin_lock(&mp->m_sb_lock);
L
Linus Torvalds 已提交
778
	if (flags & XFS_QMOPT_SBVERSION) {
779
		ASSERT(!xfs_sb_version_hasquota(&mp->m_sb));
L
Linus Torvalds 已提交
780

781
		xfs_sb_version_addquota(&mp->m_sb);
L
Linus Torvalds 已提交
782 783
		mp->m_sb.sb_uquotino = NULLFSINO;
		mp->m_sb.sb_gquotino = NULLFSINO;
784
		mp->m_sb.sb_pquotino = NULLFSINO;
L
Linus Torvalds 已提交
785

786 787
		/* qflags will get updated fully _after_ quotacheck */
		mp->m_sb.sb_qflags = mp->m_qflags & XFS_ALL_QUOTA_ACCT;
L
Linus Torvalds 已提交
788 789 790
	}
	if (flags & XFS_QMOPT_UQUOTA)
		mp->m_sb.sb_uquotino = (*ip)->i_ino;
791
	else if (flags & XFS_QMOPT_GQUOTA)
L
Linus Torvalds 已提交
792
		mp->m_sb.sb_gquotino = (*ip)->i_ino;
793 794
	else
		mp->m_sb.sb_pquotino = (*ip)->i_ino;
E
Eric Sandeen 已提交
795
	spin_unlock(&mp->m_sb_lock);
796
	xfs_log_sb(tp);
L
Linus Torvalds 已提交
797

798 799 800
	error = xfs_trans_commit(tp, XFS_TRANS_RELEASE_LOG_RES);
	if (error) {
		ASSERT(XFS_FORCED_SHUTDOWN(mp));
801
		xfs_alert(mp, "%s failed (error %d)!", __func__, error);
L
Linus Torvalds 已提交
802
	}
803 804 805
	if (need_alloc)
		xfs_finish_inode_setup(*ip);
	return error;
L
Linus Torvalds 已提交
806 807 808
}


809
STATIC void
L
Linus Torvalds 已提交
810 811 812 813 814 815
xfs_qm_reset_dqcounts(
	xfs_mount_t	*mp,
	xfs_buf_t	*bp,
	xfs_dqid_t	id,
	uint		type)
{
D
Dave Chinner 已提交
816
	struct xfs_dqblk	*dqb;
L
Linus Torvalds 已提交
817 818
	int			j;

C
Christoph Hellwig 已提交
819 820
	trace_xfs_reset_dqcounts(bp, _RET_IP_);

L
Linus Torvalds 已提交
821 822 823 824 825 826 827
	/*
	 * Reset all counters and timers. They'll be
	 * started afresh by xfs_qm_quotacheck.
	 */
#ifdef DEBUG
	j = XFS_FSB_TO_B(mp, XFS_DQUOT_CLUSTER_SIZE_FSB);
	do_div(j, sizeof(xfs_dqblk_t));
828
	ASSERT(mp->m_quotainfo->qi_dqperchunk == j);
L
Linus Torvalds 已提交
829
#endif
D
Dave Chinner 已提交
830
	dqb = bp->b_addr;
831
	for (j = 0; j < mp->m_quotainfo->qi_dqperchunk; j++) {
D
Dave Chinner 已提交
832 833 834 835
		struct xfs_disk_dquot	*ddq;

		ddq = (struct xfs_disk_dquot *)&dqb[j];

L
Linus Torvalds 已提交
836 837 838
		/*
		 * Do a sanity check, and if needed, repair the dqblk. Don't
		 * output any warnings because it's perfectly possible to
839
		 * find uninitialised dquot blks. See comment in xfs_dqcheck.
L
Linus Torvalds 已提交
840
		 */
841 842
		xfs_dqcheck(mp, ddq, id+j, type, XFS_QMOPT_DQREPAIR,
			    "xfs_quotacheck");
843 844 845 846 847
		/*
		 * Reset type in case we are reusing group quota file for
		 * project quotas or vice versa
		 */
		ddq->d_flags = type;
848 849 850 851 852 853 854 855 856
		ddq->d_bcount = 0;
		ddq->d_icount = 0;
		ddq->d_rtbcount = 0;
		ddq->d_btimer = 0;
		ddq->d_itimer = 0;
		ddq->d_rtbtimer = 0;
		ddq->d_bwarns = 0;
		ddq->d_iwarns = 0;
		ddq->d_rtbwarns = 0;
D
Dave Chinner 已提交
857 858 859 860 861 862

		if (xfs_sb_version_hascrc(&mp->m_sb)) {
			xfs_update_cksum((char *)&dqb[j],
					 sizeof(struct xfs_dqblk),
					 XFS_DQUOT_CRC_OFF);
		}
L
Linus Torvalds 已提交
863 864 865 866 867
	}
}

STATIC int
xfs_qm_dqiter_bufs(
868 869 870 871 872 873
	struct xfs_mount	*mp,
	xfs_dqid_t		firstid,
	xfs_fsblock_t		bno,
	xfs_filblks_t		blkcnt,
	uint			flags,
	struct list_head	*buffer_list)
L
Linus Torvalds 已提交
874
{
875 876 877
	struct xfs_buf		*bp;
	int			error;
	int			type;
L
Linus Torvalds 已提交
878 879

	ASSERT(blkcnt > 0);
880 881
	type = flags & XFS_QMOPT_UQUOTA ? XFS_DQ_USER :
		(flags & XFS_QMOPT_PQUOTA ? XFS_DQ_PROJ : XFS_DQ_GROUP);
L
Linus Torvalds 已提交
882 883 884 885 886 887 888 889 890 891 892 893 894 895
	error = 0;

	/*
	 * Blkcnt arg can be a very big number, and might even be
	 * larger than the log itself. So, we have to break it up into
	 * manageable-sized transactions.
	 * Note that we don't start a permanent transaction here; we might
	 * not be able to get a log reservation for the whole thing up front,
	 * and we don't really care to either, because we just discard
	 * everything if we were to crash in the middle of this loop.
	 */
	while (blkcnt--) {
		error = xfs_trans_read_buf(mp, NULL, mp->m_ddev_targp,
			      XFS_FSB_TO_DADDR(mp, bno),
896
			      mp->m_quotainfo->qi_dqchunklen, 0, &bp,
897
			      &xfs_dquot_buf_ops);
L
Linus Torvalds 已提交
898

899
		/*
D
Dave Chinner 已提交
900 901 902 903 904
		 * CRC and validation errors will return a EFSCORRUPTED here. If
		 * this occurs, re-read without CRC validation so that we can
		 * repair the damage via xfs_qm_reset_dqcounts(). This process
		 * will leave a trace in the log indicating corruption has
		 * been detected.
905
		 */
D
Dave Chinner 已提交
906
		if (error == -EFSCORRUPTED) {
D
Dave Chinner 已提交
907 908 909 910 911 912 913 914 915
			error = xfs_trans_read_buf(mp, NULL, mp->m_ddev_targp,
				      XFS_FSB_TO_DADDR(mp, bno),
				      mp->m_quotainfo->qi_dqchunklen, 0, &bp,
				      NULL);
		}

		if (error)
			break;

916 917 918 919 920 921
		/*
		 * A corrupt buffer might not have a verifier attached, so
		 * make sure we have the correct one attached before writeback
		 * occurs.
		 */
		bp->b_ops = &xfs_dquot_buf_ops;
922
		xfs_qm_reset_dqcounts(mp, bp, firstid, type);
923
		xfs_buf_delwri_queue(bp, buffer_list);
924
		xfs_buf_relse(bp);
D
Dave Chinner 已提交
925 926

		/* goto the next block. */
L
Linus Torvalds 已提交
927
		bno++;
928
		firstid += mp->m_quotainfo->qi_dqperchunk;
L
Linus Torvalds 已提交
929
	}
930

931
	return error;
L
Linus Torvalds 已提交
932 933 934
}

/*
935
 * Iterate over all allocated USR/GRP/PRJ dquots in the system, calling a
L
Linus Torvalds 已提交
936 937 938 939
 * caller supplied function for every chunk of dquots that we find.
 */
STATIC int
xfs_qm_dqiterate(
940 941 942 943
	struct xfs_mount	*mp,
	struct xfs_inode	*qip,
	uint			flags,
	struct list_head	*buffer_list)
L
Linus Torvalds 已提交
944
{
945
	struct xfs_bmbt_irec	*map;
L
Linus Torvalds 已提交
946 947 948 949 950 951 952 953 954 955
	int			i, nmaps;	/* number of map entries */
	int			error;		/* return value */
	xfs_fileoff_t		lblkno;
	xfs_filblks_t		maxlblkcnt;
	xfs_dqid_t		firstid;
	xfs_fsblock_t		rablkno;
	xfs_filblks_t		rablkcnt;

	error = 0;
	/*
956
	 * This looks racy, but we can't keep an inode lock across a
L
Linus Torvalds 已提交
957 958 959 960
	 * trans_reserve. But, this gets called during quotacheck, and that
	 * happens only at mount time which is single threaded.
	 */
	if (qip->i_d.di_nblocks == 0)
961
		return 0;
L
Linus Torvalds 已提交
962 963 964 965

	map = kmem_alloc(XFS_DQITER_MAP_SIZE * sizeof(*map), KM_SLEEP);

	lblkno = 0;
966
	maxlblkcnt = XFS_B_TO_FSB(mp, mp->m_super->s_maxbytes);
L
Linus Torvalds 已提交
967
	do {
968 969
		uint		lock_mode;

L
Linus Torvalds 已提交
970 971 972 973 974 975
		nmaps = XFS_DQITER_MAP_SIZE;
		/*
		 * We aren't changing the inode itself. Just changing
		 * some of its data. No new blocks are added here, and
		 * the inode is never added to the transaction.
		 */
976
		lock_mode = xfs_ilock_data_map_shared(qip);
D
Dave Chinner 已提交
977 978
		error = xfs_bmapi_read(qip, lblkno, maxlblkcnt - lblkno,
				       map, &nmaps, 0);
979
		xfs_iunlock(qip, lock_mode);
L
Linus Torvalds 已提交
980 981 982 983 984 985 986 987 988 989 990 991 992 993 994
		if (error)
			break;

		ASSERT(nmaps <= XFS_DQITER_MAP_SIZE);
		for (i = 0; i < nmaps; i++) {
			ASSERT(map[i].br_startblock != DELAYSTARTBLOCK);
			ASSERT(map[i].br_blockcount);


			lblkno += map[i].br_blockcount;

			if (map[i].br_startblock == HOLESTARTBLOCK)
				continue;

			firstid = (xfs_dqid_t) map[i].br_startoff *
995
				mp->m_quotainfo->qi_dqperchunk;
L
Linus Torvalds 已提交
996 997 998 999 1000 1001 1002 1003
			/*
			 * Do a read-ahead on the next extent.
			 */
			if ((i+1 < nmaps) &&
			    (map[i+1].br_startblock != HOLESTARTBLOCK)) {
				rablkcnt =  map[i+1].br_blockcount;
				rablkno = map[i+1].br_startblock;
				while (rablkcnt--) {
C
Christoph Hellwig 已提交
1004
					xfs_buf_readahead(mp->m_ddev_targp,
L
Linus Torvalds 已提交
1005
					       XFS_FSB_TO_DADDR(mp, rablkno),
1006
					       mp->m_quotainfo->qi_dqchunklen,
1007
					       &xfs_dquot_buf_ops);
L
Linus Torvalds 已提交
1008 1009 1010 1011 1012 1013 1014
					rablkno++;
				}
			}
			/*
			 * Iterate thru all the blks in the extent and
			 * reset the counters of all the dquots inside them.
			 */
1015 1016 1017 1018 1019 1020
			error = xfs_qm_dqiter_bufs(mp, firstid,
						   map[i].br_startblock,
						   map[i].br_blockcount,
						   flags, buffer_list);
			if (error)
				goto out;
L
Linus Torvalds 已提交
1021 1022 1023
		}
	} while (nmaps > 0);

1024
out:
1025
	kmem_free(map);
1026
	return error;
L
Linus Torvalds 已提交
1027 1028 1029 1030
}

/*
 * Called by dqusage_adjust in doing a quotacheck.
1031 1032 1033 1034 1035
 *
 * Given the inode, and a dquot id this updates both the incore dqout as well
 * as the buffer copy. This is so that once the quotacheck is done, we can
 * just log all the buffers, as opposed to logging numerous updates to
 * individual dquots.
L
Linus Torvalds 已提交
1036
 */
1037
STATIC int
L
Linus Torvalds 已提交
1038
xfs_qm_quotacheck_dqadjust(
1039 1040 1041
	struct xfs_inode	*ip,
	xfs_dqid_t		id,
	uint			type,
L
Linus Torvalds 已提交
1042 1043 1044
	xfs_qcnt_t		nblks,
	xfs_qcnt_t		rtblks)
{
1045 1046 1047 1048 1049 1050 1051 1052 1053 1054
	struct xfs_mount	*mp = ip->i_mount;
	struct xfs_dquot	*dqp;
	int			error;

	error = xfs_qm_dqget(mp, ip, id, type,
			     XFS_QMOPT_DQALLOC | XFS_QMOPT_DOWARN, &dqp);
	if (error) {
		/*
		 * Shouldn't be able to turn off quotas here.
		 */
D
Dave Chinner 已提交
1055 1056
		ASSERT(error != -ESRCH);
		ASSERT(error != -ENOENT);
1057 1058
		return error;
	}
C
Christoph Hellwig 已提交
1059 1060 1061

	trace_xfs_dqadjust(dqp);

L
Linus Torvalds 已提交
1062 1063 1064 1065
	/*
	 * Adjust the inode count and the block count to reflect this inode's
	 * resource usage.
	 */
1066
	be64_add_cpu(&dqp->q_core.d_icount, 1);
L
Linus Torvalds 已提交
1067 1068
	dqp->q_res_icount++;
	if (nblks) {
1069
		be64_add_cpu(&dqp->q_core.d_bcount, nblks);
L
Linus Torvalds 已提交
1070 1071 1072
		dqp->q_res_bcount += nblks;
	}
	if (rtblks) {
1073
		be64_add_cpu(&dqp->q_core.d_rtbcount, rtblks);
L
Linus Torvalds 已提交
1074 1075 1076 1077 1078
		dqp->q_res_rtbcount += rtblks;
	}

	/*
	 * Set default limits, adjust timers (since we changed usages)
1079 1080
	 *
	 * There are no timers for the default values set in the root dquot.
L
Linus Torvalds 已提交
1081
	 */
1082
	if (dqp->q_core.d_id) {
1083
		xfs_qm_adjust_dqlimits(mp, dqp);
1084
		xfs_qm_adjust_dqtimers(mp, &dqp->q_core);
L
Linus Torvalds 已提交
1085 1086 1087
	}

	dqp->dq_flags |= XFS_DQ_DIRTY;
1088 1089
	xfs_qm_dqput(dqp);
	return 0;
L
Linus Torvalds 已提交
1090 1091 1092 1093 1094 1095 1096 1097
}

STATIC int
xfs_qm_get_rtblks(
	xfs_inode_t	*ip,
	xfs_qcnt_t	*O_rtblks)
{
	xfs_filblks_t	rtblks;			/* total rt blks */
1098
	xfs_extnum_t	idx;			/* extent record index */
L
Linus Torvalds 已提交
1099 1100 1101 1102 1103 1104 1105 1106
	xfs_ifork_t	*ifp;			/* inode fork pointer */
	xfs_extnum_t	nextents;		/* number of extent entries */
	int		error;

	ASSERT(XFS_IS_REALTIME_INODE(ip));
	ifp = XFS_IFORK_PTR(ip, XFS_DATA_FORK);
	if (!(ifp->if_flags & XFS_IFEXTENTS)) {
		if ((error = xfs_iread_extents(NULL, ip, XFS_DATA_FORK)))
1107
			return error;
L
Linus Torvalds 已提交
1108 1109
	}
	rtblks = 0;
1110
	nextents = ifp->if_bytes / (uint)sizeof(xfs_bmbt_rec_t);
1111 1112
	for (idx = 0; idx < nextents; idx++)
		rtblks += xfs_bmbt_get_blockcount(xfs_iext_get_ext(ifp, idx));
L
Linus Torvalds 已提交
1113
	*O_rtblks = (xfs_qcnt_t)rtblks;
1114
	return 0;
L
Linus Torvalds 已提交
1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131
}

/*
 * callback routine supplied to bulkstat(). Given an inumber, find its
 * dquots and update them to account for resources taken by that inode.
 */
/* ARGSUSED */
STATIC int
xfs_qm_dqusage_adjust(
	xfs_mount_t	*mp,		/* mount point for filesystem */
	xfs_ino_t	ino,		/* inode number to get data for */
	void		__user *buffer,	/* not used */
	int		ubsize,		/* not used */
	int		*ubused,	/* not used */
	int		*res)		/* result code value */
{
	xfs_inode_t	*ip;
1132
	xfs_qcnt_t	nblks, rtblks = 0;
L
Linus Torvalds 已提交
1133 1134 1135 1136 1137 1138 1139 1140
	int		error;

	ASSERT(XFS_IS_QUOTA_RUNNING(mp));

	/*
	 * rootino must have its resources accounted for, not so with the quota
	 * inodes.
	 */
1141
	if (xfs_is_quota_inode(&mp->m_sb, ino)) {
L
Linus Torvalds 已提交
1142
		*res = BULKSTAT_RV_NOTHING;
D
Dave Chinner 已提交
1143
		return -EINVAL;
L
Linus Torvalds 已提交
1144 1145 1146 1147 1148 1149 1150 1151
	}

	/*
	 * We don't _need_ to take the ilock EXCL. However, the xfs_qm_dqget
	 * interface expects the inode to be exclusively locked because that's
	 * the case in all other instances. It's OK that we do this because
	 * quotacheck is done only at mount time.
	 */
1152 1153
	error = xfs_iget(mp, NULL, ino, 0, XFS_ILOCK_EXCL, &ip);
	if (error) {
L
Linus Torvalds 已提交
1154
		*res = BULKSTAT_RV_NOTHING;
1155
		return error;
L
Linus Torvalds 已提交
1156 1157
	}

1158
	ASSERT(ip->i_delayed_blks == 0);
L
Linus Torvalds 已提交
1159

1160
	if (XFS_IS_REALTIME_INODE(ip)) {
L
Linus Torvalds 已提交
1161 1162 1163
		/*
		 * Walk thru the extent list and count the realtime blocks.
		 */
1164 1165 1166
		error = xfs_qm_get_rtblks(ip, &rtblks);
		if (error)
			goto error0;
L
Linus Torvalds 已提交
1167 1168
	}

1169
	nblks = (xfs_qcnt_t)ip->i_d.di_nblocks - rtblks;
L
Linus Torvalds 已提交
1170 1171 1172 1173 1174 1175 1176 1177 1178 1179

	/*
	 * Add the (disk blocks and inode) resources occupied by this
	 * inode to its dquots. We do this adjustment in the incore dquot,
	 * and also copy the changes to its buffer.
	 * We don't care about putting these changes in a transaction
	 * envelope because if we crash in the middle of a 'quotacheck'
	 * we have to start from the beginning anyway.
	 * Once we're done, we'll log all the dquot bufs.
	 *
1180
	 * The *QUOTA_ON checks below may look pretty racy, but quotachecks
L
Linus Torvalds 已提交
1181 1182 1183
	 * and quotaoffs don't race. (Quotachecks happen at mount time only).
	 */
	if (XFS_IS_UQUOTA_ON(mp)) {
1184 1185 1186 1187
		error = xfs_qm_quotacheck_dqadjust(ip, ip->i_d.di_uid,
						   XFS_DQ_USER, nblks, rtblks);
		if (error)
			goto error0;
L
Linus Torvalds 已提交
1188
	}
1189 1190 1191 1192 1193 1194

	if (XFS_IS_GQUOTA_ON(mp)) {
		error = xfs_qm_quotacheck_dqadjust(ip, ip->i_d.di_gid,
						   XFS_DQ_GROUP, nblks, rtblks);
		if (error)
			goto error0;
L
Linus Torvalds 已提交
1195 1196
	}

1197
	if (XFS_IS_PQUOTA_ON(mp)) {
1198
		error = xfs_qm_quotacheck_dqadjust(ip, xfs_get_projid(ip),
1199 1200 1201 1202 1203 1204 1205
						   XFS_DQ_PROJ, nblks, rtblks);
		if (error)
			goto error0;
	}

	xfs_iunlock(ip, XFS_ILOCK_EXCL);
	IRELE(ip);
L
Linus Torvalds 已提交
1206
	*res = BULKSTAT_RV_DIDONE;
1207
	return 0;
1208 1209 1210 1211 1212 1213

error0:
	xfs_iunlock(ip, XFS_ILOCK_EXCL);
	IRELE(ip);
	*res = BULKSTAT_RV_GIVEUP;
	return error;
L
Linus Torvalds 已提交
1214 1215
}

1216 1217
STATIC int
xfs_qm_flush_one(
1218 1219
	struct xfs_dquot	*dqp,
	void			*data)
1220
{
1221
	struct list_head	*buffer_list = data;
1222
	struct xfs_buf		*bp = NULL;
1223 1224 1225 1226 1227 1228 1229 1230
	int			error = 0;

	xfs_dqlock(dqp);
	if (dqp->dq_flags & XFS_DQ_FREEING)
		goto out_unlock;
	if (!XFS_DQ_IS_DIRTY(dqp))
		goto out_unlock;

1231
	xfs_dqflock(dqp);
1232 1233 1234
	error = xfs_qm_dqflush(dqp, &bp);
	if (error)
		goto out_unlock;
1235

1236
	xfs_buf_delwri_queue(bp, buffer_list);
1237
	xfs_buf_relse(bp);
1238 1239 1240 1241 1242
out_unlock:
	xfs_dqunlock(dqp);
	return error;
}

L
Linus Torvalds 已提交
1243 1244 1245 1246
/*
 * Walk thru all the filesystem inodes and construct a consistent view
 * of the disk quota world. If the quotacheck fails, disable quotas.
 */
J
Jie Liu 已提交
1247
STATIC int
L
Linus Torvalds 已提交
1248 1249 1250
xfs_qm_quotacheck(
	xfs_mount_t	*mp)
{
1251 1252 1253 1254 1255 1256 1257
	int			done, count, error, error2;
	xfs_ino_t		lastino;
	size_t			structsz;
	uint			flags;
	LIST_HEAD		(buffer_list);
	struct xfs_inode	*uip = mp->m_quotainfo->qi_uquotaip;
	struct xfs_inode	*gip = mp->m_quotainfo->qi_gquotaip;
1258
	struct xfs_inode	*pip = mp->m_quotainfo->qi_pquotaip;
L
Linus Torvalds 已提交
1259 1260 1261 1262 1263 1264

	count = INT_MAX;
	structsz = 1;
	lastino = 0;
	flags = 0;

1265
	ASSERT(uip || gip || pip);
L
Linus Torvalds 已提交
1266 1267
	ASSERT(XFS_IS_QUOTA_RUNNING(mp));

1268
	xfs_notice(mp, "Quotacheck needed: Please wait.");
L
Linus Torvalds 已提交
1269 1270

	/*
1271
	 * First we go thru all the dquots on disk, USR and GRP/PRJ, and reset
L
Linus Torvalds 已提交
1272 1273 1274
	 * their counters to zero. We need a clean slate.
	 * We don't log our changes till later.
	 */
1275
	if (uip) {
1276 1277
		error = xfs_qm_dqiterate(mp, uip, XFS_QMOPT_UQUOTA,
					 &buffer_list);
1278
		if (error)
L
Linus Torvalds 已提交
1279 1280 1281 1282
			goto error_return;
		flags |= XFS_UQUOTA_CHKD;
	}

1283
	if (gip) {
1284
		error = xfs_qm_dqiterate(mp, gip, XFS_QMOPT_GQUOTA,
1285
					 &buffer_list);
1286
		if (error)
L
Linus Torvalds 已提交
1287
			goto error_return;
1288 1289 1290 1291 1292 1293 1294 1295 1296
		flags |= XFS_GQUOTA_CHKD;
	}

	if (pip) {
		error = xfs_qm_dqiterate(mp, pip, XFS_QMOPT_PQUOTA,
					 &buffer_list);
		if (error)
			goto error_return;
		flags |= XFS_PQUOTA_CHKD;
L
Linus Torvalds 已提交
1297 1298 1299 1300 1301 1302 1303
	}

	do {
		/*
		 * Iterate thru all the inodes in the file system,
		 * adjusting the corresponding dquot counters in core.
		 */
1304 1305 1306 1307
		error = xfs_bulkstat(mp, &lastino, &count,
				     xfs_qm_dqusage_adjust,
				     structsz, NULL, &done);
		if (error)
L
Linus Torvalds 已提交
1308 1309
			break;

1310
	} while (!done);
L
Linus Torvalds 已提交
1311

1312
	/*
1313 1314
	 * We've made all the changes that we need to make incore.  Flush them
	 * down to disk buffers if everything was updated successfully.
1315
	 */
1316 1317 1318 1319
	if (XFS_IS_UQUOTA_ON(mp)) {
		error = xfs_qm_dquot_walk(mp, XFS_DQ_USER, xfs_qm_flush_one,
					  &buffer_list);
	}
1320
	if (XFS_IS_GQUOTA_ON(mp)) {
1321 1322
		error2 = xfs_qm_dquot_walk(mp, XFS_DQ_GROUP, xfs_qm_flush_one,
					   &buffer_list);
1323 1324 1325 1326
		if (!error)
			error = error2;
	}
	if (XFS_IS_PQUOTA_ON(mp)) {
1327 1328
		error2 = xfs_qm_dquot_walk(mp, XFS_DQ_PROJ, xfs_qm_flush_one,
					   &buffer_list);
1329 1330 1331
		if (!error)
			error = error2;
	}
1332

1333 1334 1335 1336
	error2 = xfs_buf_delwri_submit(&buffer_list);
	if (!error)
		error = error2;

L
Linus Torvalds 已提交
1337 1338 1339 1340 1341 1342 1343 1344
	/*
	 * We can get this error if we couldn't do a dquot allocation inside
	 * xfs_qm_dqusage_adjust (via bulkstat). We don't care about the
	 * dirty dquots that might be cached, we just want to get rid of them
	 * and turn quotaoff. The dquots won't be attached to any of the inodes
	 * at this point (because we intentionally didn't in dqget_noattach).
	 */
	if (error) {
1345
		xfs_qm_dqpurge_all(mp, XFS_QMOPT_QUOTALL);
L
Linus Torvalds 已提交
1346 1347 1348 1349 1350 1351 1352 1353
		goto error_return;
	}

	/*
	 * If one type of quotas is off, then it will lose its
	 * quotachecked status, since we won't be doing accounting for
	 * that type anymore.
	 */
1354
	mp->m_qflags &= ~XFS_ALL_QUOTA_CHKD;
L
Linus Torvalds 已提交
1355 1356 1357
	mp->m_qflags |= flags;

 error_return:
1358 1359 1360 1361 1362 1363 1364
	while (!list_empty(&buffer_list)) {
		struct xfs_buf *bp =
			list_first_entry(&buffer_list, struct xfs_buf, b_list);
		list_del_init(&bp->b_list);
		xfs_buf_relse(bp);
	}

L
Linus Torvalds 已提交
1365
	if (error) {
1366 1367 1368
		xfs_warn(mp,
	"Quotacheck: Unsuccessful (Error %d): Disabling quotas.",
			error);
L
Linus Torvalds 已提交
1369 1370 1371 1372 1373
		/*
		 * We must turn off quotas.
		 */
		ASSERT(mp->m_quotainfo != NULL);
		xfs_qm_destroy_quotainfo(mp);
1374
		if (xfs_mount_reset_sbqflags(mp)) {
1375 1376
			xfs_warn(mp,
				"Quotacheck: Failed to reset quota flags.");
1377
		}
1378 1379
	} else
		xfs_notice(mp, "Quotacheck: Done.");
E
Eric Sandeen 已提交
1380
	return error;
L
Linus Torvalds 已提交
1381 1382
}

J
Jie Liu 已提交
1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456
/*
 * This is called from xfs_mountfs to start quotas and initialize all
 * necessary data structures like quotainfo.  This is also responsible for
 * running a quotacheck as necessary.  We are guaranteed that the superblock
 * is consistently read in at this point.
 *
 * If we fail here, the mount will continue with quota turned off. We don't
 * need to inidicate success or failure at all.
 */
void
xfs_qm_mount_quotas(
	struct xfs_mount	*mp)
{
	int			error = 0;
	uint			sbf;

	/*
	 * If quotas on realtime volumes is not supported, we disable
	 * quotas immediately.
	 */
	if (mp->m_sb.sb_rextents) {
		xfs_notice(mp, "Cannot turn on quotas for realtime filesystem");
		mp->m_qflags = 0;
		goto write_changes;
	}

	ASSERT(XFS_IS_QUOTA_RUNNING(mp));

	/*
	 * Allocate the quotainfo structure inside the mount struct, and
	 * create quotainode(s), and change/rev superblock if necessary.
	 */
	error = xfs_qm_init_quotainfo(mp);
	if (error) {
		/*
		 * We must turn off quotas.
		 */
		ASSERT(mp->m_quotainfo == NULL);
		mp->m_qflags = 0;
		goto write_changes;
	}
	/*
	 * If any of the quotas are not consistent, do a quotacheck.
	 */
	if (XFS_QM_NEED_QUOTACHECK(mp)) {
		error = xfs_qm_quotacheck(mp);
		if (error) {
			/* Quotacheck failed and disabled quotas. */
			return;
		}
	}
	/*
	 * If one type of quotas is off, then it will lose its
	 * quotachecked status, since we won't be doing accounting for
	 * that type anymore.
	 */
	if (!XFS_IS_UQUOTA_ON(mp))
		mp->m_qflags &= ~XFS_UQUOTA_CHKD;
	if (!XFS_IS_GQUOTA_ON(mp))
		mp->m_qflags &= ~XFS_GQUOTA_CHKD;
	if (!XFS_IS_PQUOTA_ON(mp))
		mp->m_qflags &= ~XFS_PQUOTA_CHKD;

 write_changes:
	/*
	 * We actually don't have to acquire the m_sb_lock at all.
	 * This can only be called from mount, and that's single threaded. XXX
	 */
	spin_lock(&mp->m_sb_lock);
	sbf = mp->m_sb.sb_qflags;
	mp->m_sb.sb_qflags = mp->m_qflags & XFS_MOUNT_QUOTA_ALL;
	spin_unlock(&mp->m_sb_lock);

	if (sbf != (mp->m_qflags & XFS_MOUNT_QUOTA_ALL)) {
1457
		if (xfs_sync_sb(mp, false)) {
J
Jie Liu 已提交
1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475
			/*
			 * We could only have been turning quotas off.
			 * We aren't in very good shape actually because
			 * the incore structures are convinced that quotas are
			 * off, but the on disk superblock doesn't know that !
			 */
			ASSERT(!(XFS_IS_QUOTA_RUNNING(mp)));
			xfs_alert(mp, "%s: Superblock update failed!",
				__func__);
		}
	}

	if (error) {
		xfs_warn(mp, "Failed to initialize disk quotas.");
		return;
	}
}

L
Linus Torvalds 已提交
1476 1477 1478 1479 1480 1481 1482 1483
/*
 * This is called after the superblock has been read in and we're ready to
 * iget the quota inodes.
 */
STATIC int
xfs_qm_init_quotainos(
	xfs_mount_t	*mp)
{
1484 1485
	struct xfs_inode	*uip = NULL;
	struct xfs_inode	*gip = NULL;
1486
	struct xfs_inode	*pip = NULL;
1487 1488
	int			error;
	uint			flags = 0;
L
Linus Torvalds 已提交
1489 1490 1491 1492 1493 1494

	ASSERT(mp->m_quotainfo);

	/*
	 * Get the uquota and gquota inodes
	 */
1495
	if (xfs_sb_version_hasquota(&mp->m_sb)) {
L
Linus Torvalds 已提交
1496 1497 1498
		if (XFS_IS_UQUOTA_ON(mp) &&
		    mp->m_sb.sb_uquotino != NULLFSINO) {
			ASSERT(mp->m_sb.sb_uquotino > 0);
1499 1500 1501
			error = xfs_iget(mp, NULL, mp->m_sb.sb_uquotino,
					     0, 0, &uip);
			if (error)
E
Eric Sandeen 已提交
1502
				return error;
L
Linus Torvalds 已提交
1503
		}
1504
		if (XFS_IS_GQUOTA_ON(mp) &&
L
Linus Torvalds 已提交
1505 1506
		    mp->m_sb.sb_gquotino != NULLFSINO) {
			ASSERT(mp->m_sb.sb_gquotino > 0);
1507 1508 1509 1510
			error = xfs_iget(mp, NULL, mp->m_sb.sb_gquotino,
					     0, 0, &gip);
			if (error)
				goto error_rele;
L
Linus Torvalds 已提交
1511
		}
1512
		if (XFS_IS_PQUOTA_ON(mp) &&
1513 1514 1515
		    mp->m_sb.sb_pquotino != NULLFSINO) {
			ASSERT(mp->m_sb.sb_pquotino > 0);
			error = xfs_iget(mp, NULL, mp->m_sb.sb_pquotino,
1516 1517 1518 1519
					     0, 0, &pip);
			if (error)
				goto error_rele;
		}
L
Linus Torvalds 已提交
1520 1521 1522 1523 1524
	} else {
		flags |= XFS_QMOPT_SBVERSION;
	}

	/*
1525
	 * Create the three inodes, if they don't exist already. The changes
L
Linus Torvalds 已提交
1526 1527 1528 1529 1530
	 * made above will get added to a transaction and logged in one of
	 * the qino_alloc calls below.  If the device is readonly,
	 * temporarily switch to read-write to do this.
	 */
	if (XFS_IS_UQUOTA_ON(mp) && uip == NULL) {
1531 1532 1533 1534
		error = xfs_qm_qino_alloc(mp, &uip,
					      flags | XFS_QMOPT_UQUOTA);
		if (error)
			goto error_rele;
L
Linus Torvalds 已提交
1535 1536 1537

		flags &= ~XFS_QMOPT_SBVERSION;
	}
1538
	if (XFS_IS_GQUOTA_ON(mp) && gip == NULL) {
1539
		error = xfs_qm_qino_alloc(mp, &gip,
1540 1541 1542 1543 1544 1545 1546 1547 1548
					  flags | XFS_QMOPT_GQUOTA);
		if (error)
			goto error_rele;

		flags &= ~XFS_QMOPT_SBVERSION;
	}
	if (XFS_IS_PQUOTA_ON(mp) && pip == NULL) {
		error = xfs_qm_qino_alloc(mp, &pip,
					  flags | XFS_QMOPT_PQUOTA);
1549 1550
		if (error)
			goto error_rele;
L
Linus Torvalds 已提交
1551 1552
	}

1553 1554
	mp->m_quotainfo->qi_uquotaip = uip;
	mp->m_quotainfo->qi_gquotaip = gip;
1555
	mp->m_quotainfo->qi_pquotaip = pip;
L
Linus Torvalds 已提交
1556

1557
	return 0;
1558 1559 1560 1561 1562 1563

error_rele:
	if (uip)
		IRELE(uip);
	if (gip)
		IRELE(gip);
1564 1565
	if (pip)
		IRELE(pip);
E
Eric Sandeen 已提交
1566
	return error;
L
Linus Torvalds 已提交
1567 1568
}

1569 1570 1571 1572 1573 1574
STATIC void
xfs_qm_dqfree_one(
	struct xfs_dquot	*dqp)
{
	struct xfs_mount	*mp = dqp->q_mount;
	struct xfs_quotainfo	*qi = mp->m_quotainfo;
L
Linus Torvalds 已提交
1575

1576
	mutex_lock(&qi->qi_tree_lock);
1577
	radix_tree_delete(xfs_dquot_tree(qi, dqp->q_core.d_flags),
1578
			  be32_to_cpu(dqp->q_core.d_id));
1579

1580
	qi->qi_dquots--;
1581
	mutex_unlock(&qi->qi_tree_lock);
1582 1583 1584 1585

	xfs_qm_dqdestroy(dqp);
}

L
Linus Torvalds 已提交
1586 1587 1588 1589
/* --------------- utility functions for vnodeops ---------------- */


/*
C
Christoph Hellwig 已提交
1590
 * Given an inode, a uid, gid and prid make sure that we have
L
Linus Torvalds 已提交
1591 1592 1593 1594 1595 1596 1597 1598 1599 1600
 * allocated relevant dquot(s) on disk, and that we won't exceed inode
 * quotas by creating this file.
 * This also attaches dquot(s) to the given inode after locking it,
 * and returns the dquots corresponding to the uid and/or gid.
 *
 * in	: inode (unlocked)
 * out	: udquot, gdquot with references taken and unlocked
 */
int
xfs_qm_vop_dqalloc(
C
Christoph Hellwig 已提交
1601
	struct xfs_inode	*ip,
1602 1603
	xfs_dqid_t		uid,
	xfs_dqid_t		gid,
C
Christoph Hellwig 已提交
1604 1605 1606
	prid_t			prid,
	uint			flags,
	struct xfs_dquot	**O_udqpp,
1607 1608
	struct xfs_dquot	**O_gdqpp,
	struct xfs_dquot	**O_pdqpp)
L
Linus Torvalds 已提交
1609
{
C
Christoph Hellwig 已提交
1610
	struct xfs_mount	*mp = ip->i_mount;
1611 1612
	struct xfs_dquot	*uq = NULL;
	struct xfs_dquot	*gq = NULL;
1613
	struct xfs_dquot	*pq = NULL;
C
Christoph Hellwig 已提交
1614 1615
	int			error;
	uint			lockflags;
L
Linus Torvalds 已提交
1616

C
Christoph Hellwig 已提交
1617
	if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
L
Linus Torvalds 已提交
1618 1619 1620 1621 1622
		return 0;

	lockflags = XFS_ILOCK_EXCL;
	xfs_ilock(ip, lockflags);

1623
	if ((flags & XFS_QMOPT_INHERIT) && XFS_INHERIT_GID(ip))
L
Linus Torvalds 已提交
1624 1625 1626 1627 1628 1629 1630
		gid = ip->i_d.di_gid;

	/*
	 * Attach the dquot(s) to this inode, doing a dquot allocation
	 * if necessary. The dquot(s) will not be locked.
	 */
	if (XFS_NOT_DQATTACHED(mp, ip)) {
C
Christoph Hellwig 已提交
1631 1632
		error = xfs_qm_dqattach_locked(ip, XFS_QMOPT_DQALLOC);
		if (error) {
L
Linus Torvalds 已提交
1633
			xfs_iunlock(ip, lockflags);
1634
			return error;
L
Linus Torvalds 已提交
1635 1636 1637
		}
	}

1638
	if ((flags & XFS_QMOPT_UQUOTA) && XFS_IS_UQUOTA_ON(mp)) {
L
Linus Torvalds 已提交
1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649
		if (ip->i_d.di_uid != uid) {
			/*
			 * What we need is the dquot that has this uid, and
			 * if we send the inode to dqget, the uid of the inode
			 * takes priority over what's sent in the uid argument.
			 * We must unlock inode here before calling dqget if
			 * we're not sending the inode, because otherwise
			 * we'll deadlock by doing trans_reserve while
			 * holding ilock.
			 */
			xfs_iunlock(ip, lockflags);
1650
			error = xfs_qm_dqget(mp, NULL, uid,
L
Linus Torvalds 已提交
1651 1652 1653
						 XFS_DQ_USER,
						 XFS_QMOPT_DQALLOC |
						 XFS_QMOPT_DOWARN,
1654 1655
						 &uq);
			if (error) {
D
Dave Chinner 已提交
1656
				ASSERT(error != -ENOENT);
1657
				return error;
L
Linus Torvalds 已提交
1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670
			}
			/*
			 * Get the ilock in the right order.
			 */
			xfs_dqunlock(uq);
			lockflags = XFS_ILOCK_SHARED;
			xfs_ilock(ip, lockflags);
		} else {
			/*
			 * Take an extra reference, because we'll return
			 * this to caller
			 */
			ASSERT(ip->i_udquot);
C
Christoph Hellwig 已提交
1671
			uq = xfs_qm_dqhold(ip->i_udquot);
L
Linus Torvalds 已提交
1672 1673
		}
	}
1674
	if ((flags & XFS_QMOPT_GQUOTA) && XFS_IS_GQUOTA_ON(mp)) {
L
Linus Torvalds 已提交
1675 1676
		if (ip->i_d.di_gid != gid) {
			xfs_iunlock(ip, lockflags);
1677
			error = xfs_qm_dqget(mp, NULL, gid,
L
Linus Torvalds 已提交
1678 1679 1680
						 XFS_DQ_GROUP,
						 XFS_QMOPT_DQALLOC |
						 XFS_QMOPT_DOWARN,
1681 1682
						 &gq);
			if (error) {
D
Dave Chinner 已提交
1683
				ASSERT(error != -ENOENT);
1684
				goto error_rele;
L
Linus Torvalds 已提交
1685 1686 1687 1688 1689 1690
			}
			xfs_dqunlock(gq);
			lockflags = XFS_ILOCK_SHARED;
			xfs_ilock(ip, lockflags);
		} else {
			ASSERT(ip->i_gdquot);
C
Christoph Hellwig 已提交
1691
			gq = xfs_qm_dqhold(ip->i_gdquot);
L
Linus Torvalds 已提交
1692
		}
1693 1694
	}
	if ((flags & XFS_QMOPT_PQUOTA) && XFS_IS_PQUOTA_ON(mp)) {
1695
		if (xfs_get_projid(ip) != prid) {
1696
			xfs_iunlock(ip, lockflags);
1697
			error = xfs_qm_dqget(mp, NULL, (xfs_dqid_t)prid,
1698 1699 1700
						 XFS_DQ_PROJ,
						 XFS_QMOPT_DQALLOC |
						 XFS_QMOPT_DOWARN,
1701
						 &pq);
1702
			if (error) {
D
Dave Chinner 已提交
1703
				ASSERT(error != -ENOENT);
1704
				goto error_rele;
1705
			}
1706
			xfs_dqunlock(pq);
1707 1708 1709
			lockflags = XFS_ILOCK_SHARED;
			xfs_ilock(ip, lockflags);
		} else {
1710 1711
			ASSERT(ip->i_pdquot);
			pq = xfs_qm_dqhold(ip->i_pdquot);
1712
		}
L
Linus Torvalds 已提交
1713 1714
	}
	if (uq)
C
Christoph Hellwig 已提交
1715
		trace_xfs_dquot_dqalloc(ip);
L
Linus Torvalds 已提交
1716 1717 1718 1719

	xfs_iunlock(ip, lockflags);
	if (O_udqpp)
		*O_udqpp = uq;
1720
	else
L
Linus Torvalds 已提交
1721 1722 1723
		xfs_qm_dqrele(uq);
	if (O_gdqpp)
		*O_gdqpp = gq;
1724
	else
L
Linus Torvalds 已提交
1725
		xfs_qm_dqrele(gq);
1726 1727
	if (O_pdqpp)
		*O_pdqpp = pq;
1728
	else
1729
		xfs_qm_dqrele(pq);
1730
	return 0;
1731 1732

error_rele:
1733 1734
	xfs_qm_dqrele(gq);
	xfs_qm_dqrele(uq);
1735
	return error;
L
Linus Torvalds 已提交
1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749
}

/*
 * Actually transfer ownership, and do dquot modifications.
 * These were already reserved.
 */
xfs_dquot_t *
xfs_qm_vop_chown(
	xfs_trans_t	*tp,
	xfs_inode_t	*ip,
	xfs_dquot_t	**IO_olddq,
	xfs_dquot_t	*newdq)
{
	xfs_dquot_t	*prevdq;
1750 1751 1752
	uint		bfield = XFS_IS_REALTIME_INODE(ip) ?
				 XFS_TRANS_DQ_RTBCOUNT : XFS_TRANS_DQ_BCOUNT;

C
Christoph Hellwig 已提交
1753

C
Christoph Hellwig 已提交
1754
	ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
L
Linus Torvalds 已提交
1755 1756 1757 1758 1759 1760 1761
	ASSERT(XFS_IS_QUOTA_RUNNING(ip->i_mount));

	/* old dquot */
	prevdq = *IO_olddq;
	ASSERT(prevdq);
	ASSERT(prevdq != newdq);

1762 1763
	xfs_trans_mod_dquot(tp, prevdq, bfield, -(ip->i_d.di_nblocks));
	xfs_trans_mod_dquot(tp, prevdq, XFS_TRANS_DQ_ICOUNT, -1);
L
Linus Torvalds 已提交
1764 1765

	/* the sparkling new dquot */
1766 1767
	xfs_trans_mod_dquot(tp, newdq, bfield, ip->i_d.di_nblocks);
	xfs_trans_mod_dquot(tp, newdq, XFS_TRANS_DQ_ICOUNT, 1);
L
Linus Torvalds 已提交
1768 1769

	/*
C
Christoph Hellwig 已提交
1770 1771
	 * Take an extra reference, because the inode is going to keep
	 * this dquot pointer even after the trans_commit.
L
Linus Torvalds 已提交
1772
	 */
C
Christoph Hellwig 已提交
1773
	*IO_olddq = xfs_qm_dqhold(newdq);
L
Linus Torvalds 已提交
1774

1775
	return prevdq;
L
Linus Torvalds 已提交
1776 1777 1778
}

/*
1779
 * Quota reservations for setattr(AT_UID|AT_GID|AT_PROJID).
L
Linus Torvalds 已提交
1780 1781 1782
 */
int
xfs_qm_vop_chown_reserve(
1783 1784 1785 1786
	struct xfs_trans	*tp,
	struct xfs_inode	*ip,
	struct xfs_dquot	*udqp,
	struct xfs_dquot	*gdqp,
1787
	struct xfs_dquot	*pdqp,
1788
	uint			flags)
L
Linus Torvalds 已提交
1789
{
1790 1791 1792 1793
	struct xfs_mount	*mp = ip->i_mount;
	uint			delblks, blkflags, prjflags = 0;
	struct xfs_dquot	*udq_unres = NULL;
	struct xfs_dquot	*gdq_unres = NULL;
1794
	struct xfs_dquot	*pdq_unres = NULL;
1795 1796
	struct xfs_dquot	*udq_delblks = NULL;
	struct xfs_dquot	*gdq_delblks = NULL;
1797
	struct xfs_dquot	*pdq_delblks = NULL;
1798
	int			error;
C
Christoph Hellwig 已提交
1799

L
Linus Torvalds 已提交
1800

C
Christoph Hellwig 已提交
1801
	ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL|XFS_ILOCK_SHARED));
L
Linus Torvalds 已提交
1802 1803 1804
	ASSERT(XFS_IS_QUOTA_RUNNING(mp));

	delblks = ip->i_delayed_blks;
1805 1806
	blkflags = XFS_IS_REALTIME_INODE(ip) ?
			XFS_QMOPT_RES_RTBLKS : XFS_QMOPT_RES_REGBLKS;
L
Linus Torvalds 已提交
1807 1808

	if (XFS_IS_UQUOTA_ON(mp) && udqp &&
1809
	    ip->i_d.di_uid != be32_to_cpu(udqp->q_core.d_id)) {
1810
		udq_delblks = udqp;
L
Linus Torvalds 已提交
1811 1812 1813 1814 1815 1816 1817
		/*
		 * If there are delayed allocation blocks, then we have to
		 * unreserve those from the old dquot, and add them to the
		 * new dquot.
		 */
		if (delblks) {
			ASSERT(ip->i_udquot);
1818
			udq_unres = ip->i_udquot;
L
Linus Torvalds 已提交
1819 1820
		}
	}
1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836
	if (XFS_IS_GQUOTA_ON(ip->i_mount) && gdqp &&
	    ip->i_d.di_gid != be32_to_cpu(gdqp->q_core.d_id)) {
		gdq_delblks = gdqp;
		if (delblks) {
			ASSERT(ip->i_gdquot);
			gdq_unres = ip->i_gdquot;
		}
	}

	if (XFS_IS_PQUOTA_ON(ip->i_mount) && pdqp &&
	    xfs_get_projid(ip) != be32_to_cpu(pdqp->q_core.d_id)) {
		prjflags = XFS_QMOPT_ENOSPC;
		pdq_delblks = pdqp;
		if (delblks) {
			ASSERT(ip->i_pdquot);
			pdq_unres = ip->i_pdquot;
L
Linus Torvalds 已提交
1837 1838 1839
		}
	}

1840
	error = xfs_trans_reserve_quota_bydquots(tp, ip->i_mount,
1841 1842
				udq_delblks, gdq_delblks, pdq_delblks,
				ip->i_d.di_nblocks, 1,
1843 1844 1845
				flags | blkflags | prjflags);
	if (error)
		return error;
L
Linus Torvalds 已提交
1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856

	/*
	 * Do the delayed blks reservations/unreservations now. Since, these
	 * are done without the help of a transaction, if a reservation fails
	 * its previous reservations won't be automatically undone by trans
	 * code. So, we have to do it manually here.
	 */
	if (delblks) {
		/*
		 * Do the reservations first. Unreservation can't fail.
		 */
1857 1858
		ASSERT(udq_delblks || gdq_delblks || pdq_delblks);
		ASSERT(udq_unres || gdq_unres || pdq_unres);
1859
		error = xfs_trans_reserve_quota_bydquots(NULL, ip->i_mount,
1860 1861
			    udq_delblks, gdq_delblks, pdq_delblks,
			    (xfs_qcnt_t)delblks, 0,
1862 1863 1864
			    flags | blkflags | prjflags);
		if (error)
			return error;
L
Linus Torvalds 已提交
1865
		xfs_trans_reserve_quota_bydquots(NULL, ip->i_mount,
1866 1867
				udq_unres, gdq_unres, pdq_unres,
				-((xfs_qcnt_t)delblks), 0, blkflags);
L
Linus Torvalds 已提交
1868 1869
	}

E
Eric Sandeen 已提交
1870
	return 0;
L
Linus Torvalds 已提交
1871 1872 1873 1874
}

int
xfs_qm_vop_rename_dqattach(
C
Christoph Hellwig 已提交
1875
	struct xfs_inode	**i_tab)
L
Linus Torvalds 已提交
1876
{
C
Christoph Hellwig 已提交
1877 1878
	struct xfs_mount	*mp = i_tab[0]->i_mount;
	int			i;
L
Linus Torvalds 已提交
1879

C
Christoph Hellwig 已提交
1880
	if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
1881
		return 0;
L
Linus Torvalds 已提交
1882

C
Christoph Hellwig 已提交
1883 1884 1885 1886
	for (i = 0; (i < 4 && i_tab[i]); i++) {
		struct xfs_inode	*ip = i_tab[i];
		int			error;

L
Linus Torvalds 已提交
1887 1888 1889
		/*
		 * Watch out for duplicate entries in the table.
		 */
C
Christoph Hellwig 已提交
1890 1891
		if (i == 0 || ip != i_tab[i-1]) {
			if (XFS_NOT_DQATTACHED(mp, ip)) {
L
Linus Torvalds 已提交
1892 1893
				error = xfs_qm_dqattach(ip, 0);
				if (error)
1894
					return error;
L
Linus Torvalds 已提交
1895 1896 1897
			}
		}
	}
1898
	return 0;
L
Linus Torvalds 已提交
1899 1900 1901
}

void
C
Christoph Hellwig 已提交
1902 1903 1904 1905
xfs_qm_vop_create_dqattach(
	struct xfs_trans	*tp,
	struct xfs_inode	*ip,
	struct xfs_dquot	*udqp,
1906 1907
	struct xfs_dquot	*gdqp,
	struct xfs_dquot	*pdqp)
L
Linus Torvalds 已提交
1908
{
C
Christoph Hellwig 已提交
1909 1910 1911
	struct xfs_mount	*mp = tp->t_mountp;

	if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
L
Linus Torvalds 已提交
1912 1913
		return;

C
Christoph Hellwig 已提交
1914
	ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
C
Christoph Hellwig 已提交
1915
	ASSERT(XFS_IS_QUOTA_RUNNING(mp));
L
Linus Torvalds 已提交
1916

1917
	if (udqp && XFS_IS_UQUOTA_ON(mp)) {
L
Linus Torvalds 已提交
1918
		ASSERT(ip->i_udquot == NULL);
1919
		ASSERT(ip->i_d.di_uid == be32_to_cpu(udqp->q_core.d_id));
C
Christoph Hellwig 已提交
1920 1921

		ip->i_udquot = xfs_qm_dqhold(udqp);
L
Linus Torvalds 已提交
1922 1923
		xfs_trans_mod_dquot(tp, udqp, XFS_TRANS_DQ_ICOUNT, 1);
	}
1924
	if (gdqp && XFS_IS_GQUOTA_ON(mp)) {
L
Linus Torvalds 已提交
1925
		ASSERT(ip->i_gdquot == NULL);
1926
		ASSERT(ip->i_d.di_gid == be32_to_cpu(gdqp->q_core.d_id));
C
Christoph Hellwig 已提交
1927
		ip->i_gdquot = xfs_qm_dqhold(gdqp);
L
Linus Torvalds 已提交
1928 1929
		xfs_trans_mod_dquot(tp, gdqp, XFS_TRANS_DQ_ICOUNT, 1);
	}
1930
	if (pdqp && XFS_IS_PQUOTA_ON(mp)) {
1931 1932 1933 1934 1935 1936
		ASSERT(ip->i_pdquot == NULL);
		ASSERT(xfs_get_projid(ip) == be32_to_cpu(pdqp->q_core.d_id));

		ip->i_pdquot = xfs_qm_dqhold(pdqp);
		xfs_trans_mod_dquot(tp, pdqp, XFS_TRANS_DQ_ICOUNT, 1);
	}
L
Linus Torvalds 已提交
1937 1938
}