quota.c 41.2 KB
Newer Older
D
David Teigland 已提交
1 2
/*
 * Copyright (C) Sistina Software, Inc.  1997-2003 All rights reserved.
3
 * Copyright (C) 2004-2007 Red Hat, Inc.  All rights reserved.
D
David Teigland 已提交
4 5 6
 *
 * This copyrighted material is made available to anyone wishing to use,
 * modify, copy, or redistribute it subject to the terms and conditions
7
 * of the GNU General Public License version 2.
D
David Teigland 已提交
8 9 10 11 12 13 14 15 16 17
 */

/*
 * Quota change tags are associated with each transaction that allocates or
 * deallocates space.  Those changes are accumulated locally to each node (in a
 * per-node file) and then are periodically synced to the quota file.  This
 * avoids the bottleneck of constantly touching the quota file, but introduces
 * fuzziness in the current usage value of IDs that are being used on different
 * nodes in the cluster simultaneously.  So, it is possible for a user on
 * multiple nodes to overrun their quota, but that overrun is controlable.
18
 * Since quota tags are part of transactions, there is no need for a quota check
D
David Teigland 已提交
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
 * program to be run on node crashes or anything like that.
 *
 * There are couple of knobs that let the administrator manage the quota
 * fuzziness.  "quota_quantum" sets the maximum time a quota change can be
 * sitting on one node before being synced to the quota file.  (The default is
 * 60 seconds.)  Another knob, "quota_scale" controls how quickly the frequency
 * of quota file syncs increases as the user moves closer to their limit.  The
 * more frequent the syncs, the more accurate the quota enforcement, but that
 * means that there is more contention between the nodes for the quota file.
 * The default value is one.  This sets the maximum theoretical quota overrun
 * (with infinite node with infinite bandwidth) to twice the user's limit.  (In
 * practice, the maximum overrun you see should be much less.)  A "quota_scale"
 * number greater than one makes quota syncs more frequent and reduces the
 * maximum overrun.  Numbers less than one (but greater than zero) make quota
 * syncs less frequent.
 *
 * GFS quotas also use per-ID Lock Value Blocks (LVBs) to cache the contents of
 * the quota file, so it is not being constantly read.
 */

39 40
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

D
David Teigland 已提交
41 42
#include <linux/sched.h>
#include <linux/slab.h>
43
#include <linux/mm.h>
D
David Teigland 已提交
44 45 46 47
#include <linux/spinlock.h>
#include <linux/completion.h>
#include <linux/buffer_head.h>
#include <linux/sort.h>
48
#include <linux/fs.h>
49
#include <linux/bio.h>
50
#include <linux/gfs2_ondisk.h>
51 52
#include <linux/kthread.h>
#include <linux/freezer.h>
S
Steven Whitehouse 已提交
53
#include <linux/quota.h>
54
#include <linux/dqblk_xfs.h>
55
#include <linux/lockref.h>
56
#include <linux/list_lru.h>
57 58 59 60
#include <linux/rcupdate.h>
#include <linux/rculist_bl.h>
#include <linux/bit_spinlock.h>
#include <linux/jhash.h>
61
#include <linux/vmalloc.h>
D
David Teigland 已提交
62 63

#include "gfs2.h"
64
#include "incore.h"
D
David Teigland 已提交
65 66 67 68 69 70 71 72 73
#include "bmap.h"
#include "glock.h"
#include "glops.h"
#include "log.h"
#include "meta_io.h"
#include "quota.h"
#include "rgrp.h"
#include "super.h"
#include "trans.h"
74
#include "inode.h"
75
#include "util.h"
D
David Teigland 已提交
76

77 78 79 80 81
#define GFS2_QD_HASH_SHIFT      12
#define GFS2_QD_HASH_SIZE       (1 << GFS2_QD_HASH_SHIFT)
#define GFS2_QD_HASH_MASK       (GFS2_QD_HASH_SIZE - 1)

/* Lock order: qd_lock -> bucket lock -> qd->lockref.lock -> lru lock */
82
/*                     -> sd_bitmap_lock                              */
83
static DEFINE_SPINLOCK(qd_lock);
84
struct list_lru gfs2_qd_lru;
85

86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
static struct hlist_bl_head qd_hash_table[GFS2_QD_HASH_SIZE];

static unsigned int gfs2_qd_hash(const struct gfs2_sbd *sdp,
				 const struct kqid qid)
{
	unsigned int h;

	h = jhash(&sdp, sizeof(struct gfs2_sbd *), 0);
	h = jhash(&qid, sizeof(struct kqid), h);

	return h & GFS2_QD_HASH_MASK;
}

static inline void spin_lock_bucket(unsigned int hash)
{
        hlist_bl_lock(&qd_hash_table[hash]);
}

static inline void spin_unlock_bucket(unsigned int hash)
{
        hlist_bl_unlock(&qd_hash_table[hash]);
}

static void gfs2_qd_dealloc(struct rcu_head *rcu)
{
	struct gfs2_quota_data *qd = container_of(rcu, struct gfs2_quota_data, qd_rcu);
	kmem_cache_free(gfs2_quotad_cachep, qd);
}

115
static void gfs2_qd_dispose(struct list_head *list)
116 117 118 119
{
	struct gfs2_quota_data *qd;
	struct gfs2_sbd *sdp;

120 121
	while (!list_empty(list)) {
		qd = list_entry(list->next, struct gfs2_quota_data, qd_lru);
122 123
		sdp = qd->qd_gl->gl_sbd;

124 125
		list_del(&qd->qd_lru);

126
		/* Free from the filesystem-specific list */
127
		spin_lock(&qd_lock);
128
		list_del(&qd->qd_list);
129
		spin_unlock(&qd_lock);
130

131 132 133 134
		spin_lock_bucket(qd->qd_hash);
		hlist_bl_del_rcu(&qd->qd_hlist);
		spin_unlock_bucket(qd->qd_hash);

135 136 137 138
		gfs2_assert_warn(sdp, !qd->qd_change);
		gfs2_assert_warn(sdp, !qd->qd_slot_count);
		gfs2_assert_warn(sdp, !qd->qd_bh_count);

139
		gfs2_glock_put(qd->qd_gl);
140 141 142
		atomic_dec(&sdp->sd_quota_count);

		/* Delete it from the common reclaim list */
143
		call_rcu(&qd->qd_rcu, gfs2_qd_dealloc);
144
	}
145 146 147
}


148 149
static enum lru_status gfs2_qd_isolate(struct list_head *item,
		struct list_lru_one *lru, spinlock_t *lru_lock, void *arg)
150 151 152 153 154 155 156 157 158
{
	struct list_head *dispose = arg;
	struct gfs2_quota_data *qd = list_entry(item, struct gfs2_quota_data, qd_lru);

	if (!spin_trylock(&qd->qd_lockref.lock))
		return LRU_SKIP;

	if (qd->qd_lockref.count == 0) {
		lockref_mark_dead(&qd->qd_lockref);
159
		list_lru_isolate_move(lru, &qd->qd_lru, dispose);
160 161 162 163 164 165 166 167 168 169 170 171 172 173 174
	}

	spin_unlock(&qd->qd_lockref.lock);
	return LRU_REMOVED;
}

static unsigned long gfs2_qd_shrink_scan(struct shrinker *shrink,
					 struct shrink_control *sc)
{
	LIST_HEAD(dispose);
	unsigned long freed;

	if (!(sc->gfp_mask & __GFP_FS))
		return SHRINK_STOP;

175 176
	freed = list_lru_shrink_walk(&gfs2_qd_lru, sc,
				     gfs2_qd_isolate, &dispose);
177 178 179

	gfs2_qd_dispose(&dispose);

180 181
	return freed;
}
182

183 184
static unsigned long gfs2_qd_shrink_count(struct shrinker *shrink,
					  struct shrink_control *sc)
185
{
186
	return vfs_pressure_ratio(list_lru_shrink_count(&gfs2_qd_lru, sc));
187 188
}

189 190 191 192 193 194 195 196
struct shrinker gfs2_qd_shrinker = {
	.count_objects = gfs2_qd_shrink_count,
	.scan_objects = gfs2_qd_shrink_scan,
	.seeks = DEFAULT_SEEKS,
	.flags = SHRINKER_NUMA_AWARE,
};


E
Eric W. Biederman 已提交
197 198
static u64 qd2index(struct gfs2_quota_data *qd)
{
199 200
	struct kqid qid = qd->qd_id;
	return (2 * (u64)from_kqid(&init_user_ns, qid)) +
B
Bob Peterson 已提交
201
		((qid.type == USRQUOTA) ? 0 : 1);
E
Eric W. Biederman 已提交
202 203
}

204
static u64 qd2offset(struct gfs2_quota_data *qd)
D
David Teigland 已提交
205
{
206
	u64 offset;
D
David Teigland 已提交
207

E
Eric W. Biederman 已提交
208
	offset = qd2index(qd);
D
David Teigland 已提交
209 210 211 212 213
	offset *= sizeof(struct gfs2_quota);

	return offset;
}

214
static struct gfs2_quota_data *qd_alloc(unsigned hash, struct gfs2_sbd *sdp, struct kqid qid)
D
David Teigland 已提交
215 216 217 218
{
	struct gfs2_quota_data *qd;
	int error;

219
	qd = kmem_cache_zalloc(gfs2_quotad_cachep, GFP_NOFS);
D
David Teigland 已提交
220
	if (!qd)
221
		return NULL;
D
David Teigland 已提交
222

223
	qd->qd_sbd = sdp;
224 225
	qd->qd_lockref.count = 1;
	spin_lock_init(&qd->qd_lockref.lock);
226
	qd->qd_id = qid;
D
David Teigland 已提交
227
	qd->qd_slot = -1;
228
	INIT_LIST_HEAD(&qd->qd_lru);
229
	qd->qd_hash = hash;
D
David Teigland 已提交
230

E
Eric W. Biederman 已提交
231
	error = gfs2_glock_get(sdp, qd2index(qd),
D
David Teigland 已提交
232 233 234 235
			      &gfs2_quota_glops, CREATE, &qd->qd_gl);
	if (error)
		goto fail;

236
	return qd;
D
David Teigland 已提交
237

238
fail:
239
	kmem_cache_free(gfs2_quotad_cachep, qd);
240
	return NULL;
D
David Teigland 已提交
241 242
}

243 244 245
static struct gfs2_quota_data *gfs2_qd_search_bucket(unsigned int hash,
						     const struct gfs2_sbd *sdp,
						     struct kqid qid)
D
David Teigland 已提交
246
{
247 248
	struct gfs2_quota_data *qd;
	struct hlist_bl_node *h;
D
David Teigland 已提交
249

250 251 252 253 254 255 256 257
	hlist_bl_for_each_entry_rcu(qd, h, &qd_hash_table[hash], qd_hlist) {
		if (!qid_eq(qd->qd_id, qid))
			continue;
		if (qd->qd_sbd != sdp)
			continue;
		if (lockref_get_not_dead(&qd->qd_lockref)) {
			list_lru_del(&gfs2_qd_lru, &qd->qd_lru);
			return qd;
D
David Teigland 已提交
258
		}
259
	}
D
David Teigland 已提交
260

261 262
	return NULL;
}
D
David Teigland 已提交
263 264


265 266 267 268 269
static int qd_get(struct gfs2_sbd *sdp, struct kqid qid,
		  struct gfs2_quota_data **qdp)
{
	struct gfs2_quota_data *qd, *new_qd;
	unsigned int hash = gfs2_qd_hash(sdp, qid);
D
David Teigland 已提交
270

271 272 273
	rcu_read_lock();
	*qdp = qd = gfs2_qd_search_bucket(hash, sdp, qid);
	rcu_read_unlock();
D
David Teigland 已提交
274

275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296
	if (qd)
		return 0;

	new_qd = qd_alloc(hash, sdp, qid);
	if (!new_qd)
		return -ENOMEM;

	spin_lock(&qd_lock);
	spin_lock_bucket(hash);
	*qdp = qd = gfs2_qd_search_bucket(hash, sdp, qid);
	if (qd == NULL) {
		*qdp = new_qd;
		list_add(&new_qd->qd_list, &sdp->sd_quota_list);
		hlist_bl_add_head_rcu(&new_qd->qd_hlist, &qd_hash_table[hash]);
		atomic_inc(&sdp->sd_quota_count);
	}
	spin_unlock_bucket(hash);
	spin_unlock(&qd_lock);

	if (qd) {
		gfs2_glock_put(new_qd->qd_gl);
		kmem_cache_free(gfs2_quotad_cachep, new_qd);
D
David Teigland 已提交
297
	}
298 299

	return 0;
D
David Teigland 已提交
300 301
}

302

D
David Teigland 已提交
303 304 305
static void qd_hold(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
306 307
	gfs2_assert(sdp, !__lockref_is_dead(&qd->qd_lockref));
	lockref_get(&qd->qd_lockref);
D
David Teigland 已提交
308 309 310 311
}

static void qd_put(struct gfs2_quota_data *qd)
{
312 313
	if (lockref_put_or_lock(&qd->qd_lockref))
		return;
314

315 316 317
	qd->qd_lockref.count = 0;
	list_lru_add(&gfs2_qd_lru, &qd->qd_lru);
	spin_unlock(&qd->qd_lockref.lock);
318

D
David Teigland 已提交
319 320 321 322
}

static int slot_get(struct gfs2_quota_data *qd)
{
323 324 325
	struct gfs2_sbd *sdp = qd->qd_sbd;
	unsigned int bit;
	int error = 0;
D
David Teigland 已提交
326

327
	spin_lock(&sdp->sd_bitmap_lock);
328 329
	if (qd->qd_slot_count != 0)
		goto out;
D
David Teigland 已提交
330

331 332 333 334 335
	error = -ENOSPC;
	bit = find_first_zero_bit(sdp->sd_quota_bitmap, sdp->sd_quota_slots);
	if (bit < sdp->sd_quota_slots) {
		set_bit(bit, sdp->sd_quota_bitmap);
		qd->qd_slot = bit;
A
Abhi Das 已提交
336
		error = 0;
337 338
out:
		qd->qd_slot_count++;
D
David Teigland 已提交
339
	}
340
	spin_unlock(&sdp->sd_bitmap_lock);
D
David Teigland 已提交
341

342
	return error;
D
David Teigland 已提交
343 344 345 346
}

static void slot_hold(struct gfs2_quota_data *qd)
{
347
	struct gfs2_sbd *sdp = qd->qd_sbd;
D
David Teigland 已提交
348

349
	spin_lock(&sdp->sd_bitmap_lock);
D
David Teigland 已提交
350 351
	gfs2_assert(sdp, qd->qd_slot_count);
	qd->qd_slot_count++;
352
	spin_unlock(&sdp->sd_bitmap_lock);
D
David Teigland 已提交
353 354 355 356
}

static void slot_put(struct gfs2_quota_data *qd)
{
357
	struct gfs2_sbd *sdp = qd->qd_sbd;
D
David Teigland 已提交
358

359
	spin_lock(&sdp->sd_bitmap_lock);
D
David Teigland 已提交
360 361
	gfs2_assert(sdp, qd->qd_slot_count);
	if (!--qd->qd_slot_count) {
362
		BUG_ON(!test_and_clear_bit(qd->qd_slot, sdp->sd_quota_bitmap));
D
David Teigland 已提交
363 364
		qd->qd_slot = -1;
	}
365
	spin_unlock(&sdp->sd_bitmap_lock);
D
David Teigland 已提交
366 367 368 369 370
}

static int bh_get(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
371
	struct gfs2_inode *ip = GFS2_I(sdp->sd_qc_inode);
D
David Teigland 已提交
372 373 374
	unsigned int block, offset;
	struct buffer_head *bh;
	int error;
375
	struct buffer_head bh_map = { .b_state = 0, .b_blocknr = 0 };
D
David Teigland 已提交
376

377
	mutex_lock(&sdp->sd_quota_mutex);
D
David Teigland 已提交
378 379

	if (qd->qd_bh_count++) {
380
		mutex_unlock(&sdp->sd_quota_mutex);
D
David Teigland 已提交
381 382 383 384
		return 0;
	}

	block = qd->qd_slot / sdp->sd_qc_per_block;
385
	offset = qd->qd_slot % sdp->sd_qc_per_block;
D
David Teigland 已提交
386

387
	bh_map.b_size = 1 << ip->i_inode.i_blkbits;
388
	error = gfs2_block_map(&ip->i_inode, block, &bh_map, 0);
D
David Teigland 已提交
389 390
	if (error)
		goto fail;
S
Steven Whitehouse 已提交
391
	error = gfs2_meta_read(ip->i_gl, bh_map.b_blocknr, DIO_WAIT, &bh);
D
David Teigland 已提交
392 393 394 395 396 397 398 399 400 401 402
	if (error)
		goto fail;
	error = -EIO;
	if (gfs2_metatype_check(sdp, bh, GFS2_METATYPE_QC))
		goto fail_brelse;

	qd->qd_bh = bh;
	qd->qd_bh_qc = (struct gfs2_quota_change *)
		(bh->b_data + sizeof(struct gfs2_meta_header) +
		 offset * sizeof(struct gfs2_quota_change));

J
Josef Whiter 已提交
403
	mutex_unlock(&sdp->sd_quota_mutex);
D
David Teigland 已提交
404 405 406

	return 0;

407
fail_brelse:
D
David Teigland 已提交
408
	brelse(bh);
409
fail:
D
David Teigland 已提交
410
	qd->qd_bh_count--;
411
	mutex_unlock(&sdp->sd_quota_mutex);
D
David Teigland 已提交
412 413 414 415 416 417 418
	return error;
}

static void bh_put(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;

419
	mutex_lock(&sdp->sd_quota_mutex);
D
David Teigland 已提交
420 421 422 423 424 425
	gfs2_assert(sdp, qd->qd_bh_count);
	if (!--qd->qd_bh_count) {
		brelse(qd->qd_bh);
		qd->qd_bh = NULL;
		qd->qd_bh_qc = NULL;
	}
426
	mutex_unlock(&sdp->sd_quota_mutex);
D
David Teigland 已提交
427 428
}

429 430 431 432 433 434 435 436
static int qd_check_sync(struct gfs2_sbd *sdp, struct gfs2_quota_data *qd,
			 u64 *sync_gen)
{
	if (test_bit(QDF_LOCKED, &qd->qd_flags) ||
	    !test_bit(QDF_CHANGE, &qd->qd_flags) ||
	    (sync_gen && (qd->qd_sync_gen >= *sync_gen)))
		return 0;

437 438
	if (!lockref_get_not_dead(&qd->qd_lockref))
		return 0;
439

440
	list_move_tail(&qd->qd_list, &sdp->sd_quota_list);
441 442
	set_bit(QDF_LOCKED, &qd->qd_flags);
	qd->qd_change_sync = qd->qd_change;
443
	slot_hold(qd);
444 445 446
	return 1;
}

D
David Teigland 已提交
447 448 449 450 451 452 453 454 455 456 457
static int qd_fish(struct gfs2_sbd *sdp, struct gfs2_quota_data **qdp)
{
	struct gfs2_quota_data *qd = NULL;
	int error;
	int found = 0;

	*qdp = NULL;

	if (sdp->sd_vfs->s_flags & MS_RDONLY)
		return 0;

458
	spin_lock(&qd_lock);
D
David Teigland 已提交
459 460

	list_for_each_entry(qd, &sdp->sd_quota_list, qd_list) {
461 462 463
		found = qd_check_sync(sdp, qd, &sdp->sd_quota_sync_gen);
		if (found)
			break;
D
David Teigland 已提交
464 465 466 467 468
	}

	if (!found)
		qd = NULL;

469
	spin_unlock(&qd_lock);
D
David Teigland 已提交
470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488

	if (qd) {
		gfs2_assert_warn(sdp, qd->qd_change_sync);
		error = bh_get(qd);
		if (error) {
			clear_bit(QDF_LOCKED, &qd->qd_flags);
			slot_put(qd);
			qd_put(qd);
			return error;
		}
	}

	*qdp = qd;

	return 0;
}

static void qd_unlock(struct gfs2_quota_data *qd)
{
489 490
	gfs2_assert_warn(qd->qd_gl->gl_sbd,
			 test_bit(QDF_LOCKED, &qd->qd_flags));
D
David Teigland 已提交
491 492 493 494 495 496
	clear_bit(QDF_LOCKED, &qd->qd_flags);
	bh_put(qd);
	slot_put(qd);
	qd_put(qd);
}

497
static int qdsb_get(struct gfs2_sbd *sdp, struct kqid qid,
D
David Teigland 已提交
498 499 500 501
		    struct gfs2_quota_data **qdp)
{
	int error;

502
	error = qd_get(sdp, qid, qdp);
D
David Teigland 已提交
503 504 505 506 507 508 509 510 511 512 513 514 515
	if (error)
		return error;

	error = slot_get(*qdp);
	if (error)
		goto fail;

	error = bh_get(*qdp);
	if (error)
		goto fail_slot;

	return 0;

516
fail_slot:
D
David Teigland 已提交
517
	slot_put(*qdp);
518
fail:
D
David Teigland 已提交
519 520 521 522 523 524 525 526 527 528 529
	qd_put(*qdp);
	return error;
}

static void qdsb_put(struct gfs2_quota_data *qd)
{
	bh_put(qd);
	slot_put(qd);
	qd_put(qd);
}

530
int gfs2_quota_hold(struct gfs2_inode *ip, kuid_t uid, kgid_t gid)
D
David Teigland 已提交
531
{
532
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
533
	struct gfs2_quota_data **qd;
D
David Teigland 已提交
534 535
	int error;

536 537 538 539 540
	if (ip->i_res == NULL) {
		error = gfs2_rs_alloc(ip);
		if (error)
			return error;
	}
541 542 543 544

	qd = ip->i_res->rs_qa_qd;

	if (gfs2_assert_warn(sdp, !ip->i_res->rs_qa_qd_num) ||
D
David Teigland 已提交
545 546 547 548 549 550
	    gfs2_assert_warn(sdp, !test_bit(GIF_QD_LOCKED, &ip->i_flags)))
		return -EIO;

	if (sdp->sd_args.ar_quota == GFS2_QUOTA_OFF)
		return 0;

551
	error = qdsb_get(sdp, make_kqid_uid(ip->i_inode.i_uid), qd);
D
David Teigland 已提交
552 553
	if (error)
		goto out;
554
	ip->i_res->rs_qa_qd_num++;
D
David Teigland 已提交
555 556
	qd++;

557
	error = qdsb_get(sdp, make_kqid_gid(ip->i_inode.i_gid), qd);
D
David Teigland 已提交
558 559
	if (error)
		goto out;
560
	ip->i_res->rs_qa_qd_num++;
D
David Teigland 已提交
561 562
	qd++;

563 564
	if (!uid_eq(uid, NO_UID_QUOTA_CHANGE) &&
	    !uid_eq(uid, ip->i_inode.i_uid)) {
565
		error = qdsb_get(sdp, make_kqid_uid(uid), qd);
D
David Teigland 已提交
566 567
		if (error)
			goto out;
568
		ip->i_res->rs_qa_qd_num++;
D
David Teigland 已提交
569 570 571
		qd++;
	}

572 573
	if (!gid_eq(gid, NO_GID_QUOTA_CHANGE) &&
	    !gid_eq(gid, ip->i_inode.i_gid)) {
574
		error = qdsb_get(sdp, make_kqid_gid(gid), qd);
D
David Teigland 已提交
575 576
		if (error)
			goto out;
577
		ip->i_res->rs_qa_qd_num++;
D
David Teigland 已提交
578 579 580
		qd++;
	}

581
out:
D
David Teigland 已提交
582 583 584 585 586 587 588
	if (error)
		gfs2_quota_unhold(ip);
	return error;
}

void gfs2_quota_unhold(struct gfs2_inode *ip)
{
589
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
D
David Teigland 已提交
590 591
	unsigned int x;

592 593
	if (ip->i_res == NULL)
		return;
D
David Teigland 已提交
594 595
	gfs2_assert_warn(sdp, !test_bit(GIF_QD_LOCKED, &ip->i_flags));

596 597 598
	for (x = 0; x < ip->i_res->rs_qa_qd_num; x++) {
		qdsb_put(ip->i_res->rs_qa_qd[x]);
		ip->i_res->rs_qa_qd[x] = NULL;
D
David Teigland 已提交
599
	}
600
	ip->i_res->rs_qa_qd_num = 0;
D
David Teigland 已提交
601 602 603 604
}

static int sort_qd(const void *a, const void *b)
{
605 606
	const struct gfs2_quota_data *qd_a = *(const struct gfs2_quota_data **)a;
	const struct gfs2_quota_data *qd_b = *(const struct gfs2_quota_data **)b;
D
David Teigland 已提交
607

608
	if (qid_lt(qd_a->qd_id, qd_b->qd_id))
609
		return -1;
610
	if (qid_lt(qd_b->qd_id, qd_a->qd_id))
611 612
		return 1;
	return 0;
D
David Teigland 已提交
613 614
}

615
static void do_qc(struct gfs2_quota_data *qd, s64 change)
D
David Teigland 已提交
616 617
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
618
	struct gfs2_inode *ip = GFS2_I(sdp->sd_qc_inode);
D
David Teigland 已提交
619
	struct gfs2_quota_change *qc = qd->qd_bh_qc;
620
	s64 x;
D
David Teigland 已提交
621

622
	mutex_lock(&sdp->sd_quota_mutex);
623
	gfs2_trans_add_meta(ip->i_gl, qd->qd_bh);
D
David Teigland 已提交
624 625 626 627

	if (!test_bit(QDF_CHANGE, &qd->qd_flags)) {
		qc->qc_change = 0;
		qc->qc_flags = 0;
628
		if (qd->qd_id.type == USRQUOTA)
D
David Teigland 已提交
629
			qc->qc_flags = cpu_to_be32(GFS2_QCF_USER);
630
		qc->qc_id = cpu_to_be32(from_kqid(&init_user_ns, qd->qd_id));
D
David Teigland 已提交
631 632
	}

A
Al Viro 已提交
633
	x = be64_to_cpu(qc->qc_change) + change;
D
David Teigland 已提交
634 635
	qc->qc_change = cpu_to_be64(x);

636
	spin_lock(&qd_lock);
D
David Teigland 已提交
637
	qd->qd_change = x;
638
	spin_unlock(&qd_lock);
D
David Teigland 已提交
639 640 641 642 643 644 645 646 647 648 649 650

	if (!x) {
		gfs2_assert_warn(sdp, test_bit(QDF_CHANGE, &qd->qd_flags));
		clear_bit(QDF_CHANGE, &qd->qd_flags);
		qc->qc_flags = 0;
		qc->qc_id = 0;
		slot_put(qd);
		qd_put(qd);
	} else if (!test_and_set_bit(QDF_CHANGE, &qd->qd_flags)) {
		qd_hold(qd);
		slot_hold(qd);
	}
651

652
	mutex_unlock(&sdp->sd_quota_mutex);
D
David Teigland 已提交
653 654
}

655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 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 751 752 753 754 755 756 757 758 759 760
static int gfs2_write_buf_to_page(struct gfs2_inode *ip, unsigned long index,
				  unsigned off, void *buf, unsigned bytes)
{
	struct inode *inode = &ip->i_inode;
	struct gfs2_sbd *sdp = GFS2_SB(inode);
	struct address_space *mapping = inode->i_mapping;
	struct page *page;
	struct buffer_head *bh;
	void *kaddr;
	u64 blk;
	unsigned bsize = sdp->sd_sb.sb_bsize, bnum = 0, boff = 0;
	unsigned to_write = bytes, pg_off = off;
	int done = 0;

	blk = index << (PAGE_CACHE_SHIFT - sdp->sd_sb.sb_bsize_shift);
	boff = off % bsize;

	page = find_or_create_page(mapping, index, GFP_NOFS);
	if (!page)
		return -ENOMEM;
	if (!page_has_buffers(page))
		create_empty_buffers(page, bsize, 0);

	bh = page_buffers(page);
	while (!done) {
		/* Find the beginning block within the page */
		if (pg_off >= ((bnum * bsize) + bsize)) {
			bh = bh->b_this_page;
			bnum++;
			blk++;
			continue;
		}
		if (!buffer_mapped(bh)) {
			gfs2_block_map(inode, blk, bh, 1);
			if (!buffer_mapped(bh))
				goto unlock_out;
			/* If it's a newly allocated disk block, zero it */
			if (buffer_new(bh))
				zero_user(page, bnum * bsize, bh->b_size);
		}
		if (PageUptodate(page))
			set_buffer_uptodate(bh);
		if (!buffer_uptodate(bh)) {
			ll_rw_block(READ | REQ_META, 1, &bh);
			wait_on_buffer(bh);
			if (!buffer_uptodate(bh))
				goto unlock_out;
		}
		gfs2_trans_add_data(ip->i_gl, bh);

		/* If we need to write to the next block as well */
		if (to_write > (bsize - boff)) {
			pg_off += (bsize - boff);
			to_write -= (bsize - boff);
			boff = pg_off % bsize;
			continue;
		}
		done = 1;
	}

	/* Write to the page, now that we have setup the buffer(s) */
	kaddr = kmap_atomic(page);
	memcpy(kaddr + off, buf, bytes);
	flush_dcache_page(page);
	kunmap_atomic(kaddr);
	unlock_page(page);
	page_cache_release(page);

	return 0;

unlock_out:
	unlock_page(page);
	page_cache_release(page);
	return -EIO;
}

static int gfs2_write_disk_quota(struct gfs2_inode *ip, struct gfs2_quota *qp,
				 loff_t loc)
{
	unsigned long pg_beg;
	unsigned pg_off, nbytes, overflow = 0;
	int pg_oflow = 0, error;
	void *ptr;

	nbytes = sizeof(struct gfs2_quota);

	pg_beg = loc >> PAGE_CACHE_SHIFT;
	pg_off = loc % PAGE_CACHE_SIZE;

	/* If the quota straddles a page boundary, split the write in two */
	if ((pg_off + nbytes) > PAGE_CACHE_SIZE) {
		pg_oflow = 1;
		overflow = (pg_off + nbytes) - PAGE_CACHE_SIZE;
	}

	ptr = qp;
	error = gfs2_write_buf_to_page(ip, pg_beg, pg_off, ptr,
				       nbytes - overflow);
	/* If there's an overflow, write the remaining bytes to the next page */
	if (!error && pg_oflow)
		error = gfs2_write_buf_to_page(ip, pg_beg + 1, 0,
					       ptr + nbytes - overflow,
					       overflow);
	return error;
}

761
/**
762 763 764
 * gfs2_adjust_quota - adjust record of current block usage
 * @ip: The quota inode
 * @loc: Offset of the entry in the quota file
S
Steven Whitehouse 已提交
765
 * @change: The amount of usage change to record
766
 * @qd: The quota data
S
Steven Whitehouse 已提交
767
 * @fdq: The updated limits to record
768 769 770
 *
 * This function was mostly borrowed from gfs2_block_truncate_page which was
 * in turn mostly borrowed from ext3
771 772
 *
 * Returns: 0 or -ve on error
773
 */
774

775
static int gfs2_adjust_quota(struct gfs2_inode *ip, loff_t loc,
S
Steven Whitehouse 已提交
776
			     s64 change, struct gfs2_quota_data *qd,
777
			     struct qc_dqblk *fdq)
778
{
779
	struct inode *inode = &ip->i_inode;
780
	struct gfs2_sbd *sdp = GFS2_SB(inode);
A
Al Viro 已提交
781
	struct gfs2_quota q;
782
	int err;
S
Steven Whitehouse 已提交
783
	u64 size;
784

S
Steven Whitehouse 已提交
785 786 787 788 789
	if (gfs2_is_stuffed(ip)) {
		err = gfs2_unstuff_dinode(ip, NULL);
		if (err)
			return err;
	}
790 791

	memset(&q, 0, sizeof(struct gfs2_quota));
792
	err = gfs2_internal_read(ip, (char *)&q, &loc, sizeof(q));
793 794 795
	if (err < 0)
		return err;

796
	loc -= sizeof(q); /* gfs2_internal_read would've advanced the loc ptr */
797
	err = -EIO;
A
Al Viro 已提交
798
	be64_add_cpu(&q.qu_value, change);
799 800
	if (be64_to_cpu(q.qu_value) < 0)
		q.qu_value = 0; /* Never go negative on quota usage */
A
Al Viro 已提交
801
	qd->qd_qb.qb_value = q.qu_value;
802
	if (fdq) {
803 804
		if (fdq->d_fieldmask & QC_SPC_SOFT) {
			q.qu_warn = cpu_to_be64(fdq->d_spc_softlimit >> sdp->sd_sb.sb_bsize_shift);
A
Al Viro 已提交
805
			qd->qd_qb.qb_warn = q.qu_warn;
806
		}
807 808
		if (fdq->d_fieldmask & QC_SPC_HARD) {
			q.qu_limit = cpu_to_be64(fdq->d_spc_hardlimit >> sdp->sd_sb.sb_bsize_shift);
A
Al Viro 已提交
809
			qd->qd_qb.qb_limit = q.qu_limit;
810
		}
811 812
		if (fdq->d_fieldmask & QC_SPACE) {
			q.qu_value = cpu_to_be64(fdq->d_space >> sdp->sd_sb.sb_bsize_shift);
A
Al Viro 已提交
813
			qd->qd_qb.qb_value = q.qu_value;
814
		}
815 816
	}

817 818 819 820 821 822 823 824
	err = gfs2_write_disk_quota(ip, &q, loc);
	if (!err) {
		size = loc + sizeof(struct gfs2_quota);
		if (size > inode->i_size)
			i_size_write(inode, size);
		inode->i_mtime = inode->i_atime = CURRENT_TIME;
		mark_inode_dirty(inode);
		set_bit(QDF_REFRESH, &qd->qd_flags);
825 826 827 828 829
	}

	return err;
}

D
David Teigland 已提交
830 831 832
static int do_sync(unsigned int num_qd, struct gfs2_quota_data **qda)
{
	struct gfs2_sbd *sdp = (*qda)->qd_gl->gl_sbd;
833
	struct gfs2_inode *ip = GFS2_I(sdp->sd_quota_inode);
834
	struct gfs2_alloc_parms ap = { .aflags = 0, };
D
David Teigland 已提交
835 836 837 838
	unsigned int data_blocks, ind_blocks;
	struct gfs2_holder *ghs, i_gh;
	unsigned int qx, x;
	struct gfs2_quota_data *qd;
839
	unsigned reserved;
840
	loff_t offset;
841
	unsigned int nalloc = 0, blocks;
D
David Teigland 已提交
842 843
	int error;

844 845 846 847
	error = gfs2_rs_alloc(ip);
	if (error)
		return error;

D
David Teigland 已提交
848 849 850
	gfs2_write_calc_reserv(ip, sizeof(struct gfs2_quota),
			      &data_blocks, &ind_blocks);

J
Josef Bacik 已提交
851
	ghs = kcalloc(num_qd, sizeof(struct gfs2_holder), GFP_NOFS);
D
David Teigland 已提交
852 853 854 855
	if (!ghs)
		return -ENOMEM;

	sort(qda, num_qd, sizeof(struct gfs2_quota_data *), sort_qd, NULL);
J
Jan Kara 已提交
856
	mutex_lock(&ip->i_inode.i_mutex);
D
David Teigland 已提交
857
	for (qx = 0; qx < num_qd; qx++) {
858
		error = gfs2_glock_nq_init(qda[qx]->qd_gl, LM_ST_EXCLUSIVE,
D
David Teigland 已提交
859 860 861 862 863 864 865 866 867 868 869
					   GL_NOCACHE, &ghs[qx]);
		if (error)
			goto out;
	}

	error = gfs2_glock_nq_init(ip->i_gl, LM_ST_EXCLUSIVE, 0, &i_gh);
	if (error)
		goto out;

	for (x = 0; x < num_qd; x++) {
		offset = qd2offset(qda[x]);
870 871
		if (gfs2_write_alloc_required(ip, offset,
					      sizeof(struct gfs2_quota)))
D
David Teigland 已提交
872 873 874
			nalloc++;
	}

875 876 877 878 879 880
	/* 
	 * 1 blk for unstuffing inode if stuffed. We add this extra
	 * block to the reservation unconditionally. If the inode
	 * doesn't need unstuffing, the block will be released to the 
	 * rgrp since it won't be allocated during the transaction
	 */
881 882 883 884
	/* +3 in the end for unstuffing block, inode size update block
	 * and another block in case quota straddles page boundary and 
	 * two blocks need to be updated instead of 1 */
	blocks = num_qd * data_blocks + RES_DINODE + num_qd + 3;
D
David Teigland 已提交
885

886
	reserved = 1 + (nalloc * (data_blocks + ind_blocks));
887 888
	ap.target = reserved;
	error = gfs2_inplace_reserve(ip, &ap);
889 890
	if (error)
		goto out_alloc;
D
David Teigland 已提交
891

892
	if (nalloc)
893
		blocks += gfs2_rg_blocks(ip, reserved) + nalloc * ind_blocks + RES_STATFS;
894 895 896 897

	error = gfs2_trans_begin(sdp, blocks, 0);
	if (error)
		goto out_ipres;
D
David Teigland 已提交
898 899 900 901

	for (x = 0; x < num_qd; x++) {
		qd = qda[x];
		offset = qd2offset(qd);
S
Steven Whitehouse 已提交
902
		error = gfs2_adjust_quota(ip, offset, qd->qd_change_sync, qd, NULL);
903
		if (error)
D
David Teigland 已提交
904 905 906
			goto out_end_trans;

		do_qc(qd, -qd->qd_change_sync);
907
		set_bit(QDF_REFRESH, &qd->qd_flags);
D
David Teigland 已提交
908 909 910 911
	}

	error = 0;

912
out_end_trans:
D
David Teigland 已提交
913
	gfs2_trans_end(sdp);
914
out_ipres:
915
	gfs2_inplace_release(ip);
916
out_alloc:
D
David Teigland 已提交
917
	gfs2_glock_dq_uninit(&i_gh);
918
out:
D
David Teigland 已提交
919 920
	while (qx--)
		gfs2_glock_dq_uninit(&ghs[qx]);
S
Steven Whitehouse 已提交
921
	mutex_unlock(&ip->i_inode.i_mutex);
D
David Teigland 已提交
922
	kfree(ghs);
923
	gfs2_log_flush(ip->i_gl->gl_sbd, ip->i_gl, NORMAL_FLUSH);
D
David Teigland 已提交
924 925 926
	return error;
}

S
Steven Whitehouse 已提交
927 928 929 930 931 932 933 934 935 936
static int update_qd(struct gfs2_sbd *sdp, struct gfs2_quota_data *qd)
{
	struct gfs2_inode *ip = GFS2_I(sdp->sd_quota_inode);
	struct gfs2_quota q;
	struct gfs2_quota_lvb *qlvb;
	loff_t pos;
	int error;

	memset(&q, 0, sizeof(struct gfs2_quota));
	pos = qd2offset(qd);
937
	error = gfs2_internal_read(ip, (char *)&q, &pos, sizeof(q));
S
Steven Whitehouse 已提交
938 939 940
	if (error < 0)
		return error;

941
	qlvb = (struct gfs2_quota_lvb *)qd->qd_gl->gl_lksb.sb_lvbptr;
S
Steven Whitehouse 已提交
942 943 944 945 946 947 948 949 950 951
	qlvb->qb_magic = cpu_to_be32(GFS2_MAGIC);
	qlvb->__pad = 0;
	qlvb->qb_limit = q.qu_limit;
	qlvb->qb_warn = q.qu_warn;
	qlvb->qb_value = q.qu_value;
	qd->qd_qb = *qlvb;

	return 0;
}

D
David Teigland 已提交
952 953 954 955
static int do_glock(struct gfs2_quota_data *qd, int force_refresh,
		    struct gfs2_holder *q_gh)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
956
	struct gfs2_inode *ip = GFS2_I(sdp->sd_quota_inode);
D
David Teigland 已提交
957 958 959
	struct gfs2_holder i_gh;
	int error;

960
restart:
D
David Teigland 已提交
961 962 963 964
	error = gfs2_glock_nq_init(qd->qd_gl, LM_ST_SHARED, 0, q_gh);
	if (error)
		return error;

965 966 967
	if (test_and_clear_bit(QDF_REFRESH, &qd->qd_flags))
		force_refresh = FORCE;

968
	qd->qd_qb = *(struct gfs2_quota_lvb *)qd->qd_gl->gl_lksb.sb_lvbptr;
D
David Teigland 已提交
969

970
	if (force_refresh || qd->qd_qb.qb_magic != cpu_to_be32(GFS2_MAGIC)) {
D
David Teigland 已提交
971
		gfs2_glock_dq_uninit(q_gh);
972 973
		error = gfs2_glock_nq_init(qd->qd_gl, LM_ST_EXCLUSIVE,
					   GL_NOCACHE, q_gh);
D
David Teigland 已提交
974 975 976
		if (error)
			return error;

977
		error = gfs2_glock_nq_init(ip->i_gl, LM_ST_SHARED, 0, &i_gh);
D
David Teigland 已提交
978 979 980
		if (error)
			goto fail;

S
Steven Whitehouse 已提交
981 982
		error = update_qd(sdp, qd);
		if (error)
983
			goto fail_gunlock;
D
David Teigland 已提交
984

S
Steven Whitehouse 已提交
985
		gfs2_glock_dq_uninit(&i_gh);
986 987 988
		gfs2_glock_dq_uninit(q_gh);
		force_refresh = 0;
		goto restart;
D
David Teigland 已提交
989 990 991 992
	}

	return 0;

993
fail_gunlock:
D
David Teigland 已提交
994
	gfs2_glock_dq_uninit(&i_gh);
995
fail:
D
David Teigland 已提交
996 997 998 999
	gfs2_glock_dq_uninit(q_gh);
	return error;
}

1000
int gfs2_quota_lock(struct gfs2_inode *ip, kuid_t uid, kgid_t gid)
D
David Teigland 已提交
1001
{
1002
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
1003
	struct gfs2_quota_data *qd;
D
David Teigland 已提交
1004 1005 1006
	unsigned int x;
	int error = 0;

S
Steven Whitehouse 已提交
1007 1008 1009
	error = gfs2_quota_hold(ip, uid, gid);
	if (error)
		return error;
D
David Teigland 已提交
1010 1011 1012 1013 1014

	if (capable(CAP_SYS_RESOURCE) ||
	    sdp->sd_args.ar_quota != GFS2_QUOTA_ON)
		return 0;

1015 1016
	sort(ip->i_res->rs_qa_qd, ip->i_res->rs_qa_qd_num,
	     sizeof(struct gfs2_quota_data *), sort_qd, NULL);
D
David Teigland 已提交
1017

1018 1019
	for (x = 0; x < ip->i_res->rs_qa_qd_num; x++) {
		qd = ip->i_res->rs_qa_qd[x];
1020
		error = do_glock(qd, NO_FORCE, &ip->i_res->rs_qa_qd_ghs[x]);
D
David Teigland 已提交
1021 1022 1023 1024 1025 1026 1027 1028
		if (error)
			break;
	}

	if (!error)
		set_bit(GIF_QD_LOCKED, &ip->i_flags);
	else {
		while (x--)
1029
			gfs2_glock_dq_uninit(&ip->i_res->rs_qa_qd_ghs[x]);
D
David Teigland 已提交
1030 1031 1032 1033 1034 1035 1036 1037 1038 1039
		gfs2_quota_unhold(ip);
	}

	return error;
}

static int need_sync(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
	struct gfs2_tune *gt = &sdp->sd_tune;
1040
	s64 value;
D
David Teigland 已提交
1041 1042 1043 1044 1045 1046
	unsigned int num, den;
	int do_sync = 1;

	if (!qd->qd_qb.qb_limit)
		return 0;

1047
	spin_lock(&qd_lock);
D
David Teigland 已提交
1048
	value = qd->qd_change;
1049
	spin_unlock(&qd_lock);
D
David Teigland 已提交
1050 1051 1052 1053 1054 1055 1056 1057

	spin_lock(&gt->gt_spin);
	num = gt->gt_quota_scale_num;
	den = gt->gt_quota_scale_den;
	spin_unlock(&gt->gt_spin);

	if (value < 0)
		do_sync = 0;
1058 1059
	else if ((s64)be64_to_cpu(qd->qd_qb.qb_value) >=
		 (s64)be64_to_cpu(qd->qd_qb.qb_limit))
D
David Teigland 已提交
1060 1061 1062
		do_sync = 0;
	else {
		value *= gfs2_jindex_size(sdp) * num;
1063
		value = div_s64(value, den);
1064
		value += (s64)be64_to_cpu(qd->qd_qb.qb_value);
1065
		if (value < (s64)be64_to_cpu(qd->qd_qb.qb_limit))
D
David Teigland 已提交
1066 1067 1068 1069 1070 1071 1072 1073
			do_sync = 0;
	}

	return do_sync;
}

void gfs2_quota_unlock(struct gfs2_inode *ip)
{
1074
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
D
David Teigland 已提交
1075 1076 1077
	struct gfs2_quota_data *qda[4];
	unsigned int count = 0;
	unsigned int x;
1078
	int found;
D
David Teigland 已提交
1079 1080 1081 1082

	if (!test_and_clear_bit(GIF_QD_LOCKED, &ip->i_flags))
		goto out;

1083
	for (x = 0; x < ip->i_res->rs_qa_qd_num; x++) {
D
David Teigland 已提交
1084 1085 1086
		struct gfs2_quota_data *qd;
		int sync;

1087
		qd = ip->i_res->rs_qa_qd[x];
D
David Teigland 已提交
1088 1089
		sync = need_sync(qd);

1090
		gfs2_glock_dq_uninit(&ip->i_res->rs_qa_qd_ghs[x]);
1091 1092 1093
		if (!sync)
			continue;

1094
		spin_lock(&qd_lock);
1095
		found = qd_check_sync(sdp, qd, NULL);
1096
		spin_unlock(&qd_lock);
1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107

		if (!found)
			continue;

		gfs2_assert_warn(sdp, qd->qd_change_sync);
		if (bh_get(qd)) {
			clear_bit(QDF_LOCKED, &qd->qd_flags);
			slot_put(qd);
			qd_put(qd);
			continue;
		}
D
David Teigland 已提交
1108

1109
		qda[count++] = qd;
D
David Teigland 已提交
1110 1111 1112 1113 1114 1115 1116 1117
	}

	if (count) {
		do_sync(count, qda);
		for (x = 0; x < count; x++)
			qd_unlock(qda[x]);
	}

1118
out:
D
David Teigland 已提交
1119 1120 1121 1122 1123 1124 1125 1126 1127
	gfs2_quota_unhold(ip);
}

#define MAX_LINE 256

static int print_message(struct gfs2_quota_data *qd, char *type)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;

J
Joe Perches 已提交
1128 1129
	fs_info(sdp, "quota %s for %s %u\n",
		type,
1130 1131
		(qd->qd_id.type == USRQUOTA) ? "user" : "group",
		from_kqid(&init_user_ns, qd->qd_id));
D
David Teigland 已提交
1132 1133 1134 1135

	return 0;
}

1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152
/**
 * gfs2_quota_check - check if allocating new blocks will exceed quota
 * @ip:  The inode for which this check is being performed
 * @uid: The uid to check against
 * @gid: The gid to check against
 * @ap:  The allocation parameters. ap->target contains the requested
 *       blocks. ap->min_target, if set, contains the minimum blks
 *       requested.
 *
 * Returns: 0 on success.
 *                  min_req = ap->min_target ? ap->min_target : ap->target;
 *                  quota must allow atleast min_req blks for success and
 *                  ap->allowed is set to the number of blocks allowed
 *
 *          -EDQUOT otherwise, quota violation. ap->allowed is set to number
 *                  of blocks available.
 */
1153 1154
int gfs2_quota_check(struct gfs2_inode *ip, kuid_t uid, kgid_t gid,
		     struct gfs2_alloc_parms *ap)
D
David Teigland 已提交
1155
{
1156
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
D
David Teigland 已提交
1157
	struct gfs2_quota_data *qd;
1158
	s64 value, warn, limit;
D
David Teigland 已提交
1159 1160 1161
	unsigned int x;
	int error = 0;

1162
	ap->allowed = UINT_MAX; /* Assume we are permitted a whole lot */
D
David Teigland 已提交
1163 1164 1165 1166 1167 1168
	if (!test_bit(GIF_QD_LOCKED, &ip->i_flags))
		return 0;

        if (sdp->sd_args.ar_quota != GFS2_QUOTA_ON)
                return 0;

1169 1170
	for (x = 0; x < ip->i_res->rs_qa_qd_num; x++) {
		qd = ip->i_res->rs_qa_qd[x];
D
David Teigland 已提交
1171

1172 1173
		if (!(qid_eq(qd->qd_id, make_kqid_uid(uid)) ||
		      qid_eq(qd->qd_id, make_kqid_gid(gid))))
D
David Teigland 已提交
1174 1175
			continue;

1176 1177
		warn = (s64)be64_to_cpu(qd->qd_qb.qb_warn);
		limit = (s64)be64_to_cpu(qd->qd_qb.qb_limit);
1178
		value = (s64)be64_to_cpu(qd->qd_qb.qb_value);
1179
		spin_lock(&qd_lock);
1180
		value += qd->qd_change;
1181
		spin_unlock(&qd_lock);
D
David Teigland 已提交
1182

1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197
		if (limit > 0 && (limit - value) < ap->allowed)
			ap->allowed = limit - value;
		/* If we can't meet the target */
		if (limit && limit < (value + (s64)ap->target)) {
			/* If no min_target specified or we don't meet
			 * min_target, return -EDQUOT */
			if (!ap->min_target || ap->min_target > ap->allowed) {
				print_message(qd, "exceeded");
				quota_send_warning(qd->qd_id,
						   sdp->sd_vfs->s_dev,
						   QUOTA_NL_BHARDWARN);
				error = -EDQUOT;
				break;
			}
		} else if (warn && warn < value &&
D
David Teigland 已提交
1198
			   time_after_eq(jiffies, qd->qd_last_warn +
1199 1200
					 gfs2_tune_get(sdp, gt_quota_warn_period)
					 * HZ)) {
1201
			quota_send_warning(qd->qd_id,
S
Steven Whitehouse 已提交
1202
					   sdp->sd_vfs->s_dev, QUOTA_NL_BSOFTWARN);
D
David Teigland 已提交
1203 1204 1205 1206 1207 1208 1209
			error = print_message(qd, "warning");
			qd->qd_last_warn = jiffies;
		}
	}
	return error;
}

1210
void gfs2_quota_change(struct gfs2_inode *ip, s64 change,
1211
		       kuid_t uid, kgid_t gid)
D
David Teigland 已提交
1212 1213 1214 1215
{
	struct gfs2_quota_data *qd;
	unsigned int x;

1216
	if (gfs2_assert_warn(GFS2_SB(&ip->i_inode), change))
D
David Teigland 已提交
1217
		return;
1218
	if (ip->i_diskflags & GFS2_DIF_SYSTEM)
D
David Teigland 已提交
1219 1220
		return;

1221 1222
	for (x = 0; x < ip->i_res->rs_qa_qd_num; x++) {
		qd = ip->i_res->rs_qa_qd[x];
D
David Teigland 已提交
1223

1224 1225
		if (qid_eq(qd->qd_id, make_kqid_uid(uid)) ||
		    qid_eq(qd->qd_id, make_kqid_gid(gid))) {
D
David Teigland 已提交
1226 1227 1228 1229 1230
			do_qc(qd, change);
		}
	}
}

1231
int gfs2_quota_sync(struct super_block *sb, int type)
D
David Teigland 已提交
1232
{
1233
	struct gfs2_sbd *sdp = sb->s_fs_info;
D
David Teigland 已提交
1234
	struct gfs2_quota_data **qda;
1235
	unsigned int max_qd = PAGE_SIZE/sizeof(struct gfs2_holder);
D
David Teigland 已提交
1236 1237 1238 1239 1240 1241 1242 1243
	unsigned int num_qd;
	unsigned int x;
	int error = 0;

	qda = kcalloc(max_qd, sizeof(struct gfs2_quota_data *), GFP_KERNEL);
	if (!qda)
		return -ENOMEM;

1244 1245 1246
	mutex_lock(&sdp->sd_quota_sync_mutex);
	sdp->sd_quota_sync_gen++;

D
David Teigland 已提交
1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
	do {
		num_qd = 0;

		for (;;) {
			error = qd_fish(sdp, qda + num_qd);
			if (error || !qda[num_qd])
				break;
			if (++num_qd == max_qd)
				break;
		}

		if (num_qd) {
			if (!error)
				error = do_sync(num_qd, qda);
			if (!error)
				for (x = 0; x < num_qd; x++)
					qda[x]->qd_sync_gen =
						sdp->sd_quota_sync_gen;

			for (x = 0; x < num_qd; x++)
				qd_unlock(qda[x]);
		}
	} while (!error && num_qd == max_qd);

1271
	mutex_unlock(&sdp->sd_quota_sync_mutex);
D
David Teigland 已提交
1272 1273 1274 1275 1276
	kfree(qda);

	return error;
}

1277
int gfs2_quota_refresh(struct gfs2_sbd *sdp, struct kqid qid)
D
David Teigland 已提交
1278 1279 1280 1281 1282
{
	struct gfs2_quota_data *qd;
	struct gfs2_holder q_gh;
	int error;

1283
	error = qd_get(sdp, qid, &qd);
D
David Teigland 已提交
1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296
	if (error)
		return error;

	error = do_glock(qd, FORCE, &q_gh);
	if (!error)
		gfs2_glock_dq_uninit(&q_gh);

	qd_put(qd);
	return error;
}

int gfs2_quota_init(struct gfs2_sbd *sdp)
{
1297
	struct gfs2_inode *ip = GFS2_I(sdp->sd_qc_inode);
S
Steven Whitehouse 已提交
1298 1299
	u64 size = i_size_read(sdp->sd_qc_inode);
	unsigned int blocks = size >> sdp->sd_sb.sb_bsize_shift;
D
David Teigland 已提交
1300 1301
	unsigned int x, slot = 0;
	unsigned int found = 0;
1302
	unsigned int hash;
1303
	unsigned int bm_size;
1304 1305
	u64 dblock;
	u32 extlen = 0;
D
David Teigland 已提交
1306 1307
	int error;

S
Steven Whitehouse 已提交
1308
	if (gfs2_check_internal_file_size(sdp->sd_qc_inode, 1, 64 << 20))
1309
		return -EIO;
S
Steven Whitehouse 已提交
1310

D
David Teigland 已提交
1311
	sdp->sd_quota_slots = blocks * sdp->sd_qc_per_block;
1312 1313
	bm_size = DIV_ROUND_UP(sdp->sd_quota_slots, 8 * sizeof(unsigned long));
	bm_size *= sizeof(unsigned long);
D
David Teigland 已提交
1314
	error = -ENOMEM;
1315
	sdp->sd_quota_bitmap = kzalloc(bm_size, GFP_NOFS | __GFP_NOWARN);
1316
	if (sdp->sd_quota_bitmap == NULL)
1317 1318
		sdp->sd_quota_bitmap = __vmalloc(bm_size, GFP_NOFS |
						 __GFP_ZERO, PAGE_KERNEL);
D
David Teigland 已提交
1319 1320 1321 1322 1323
	if (!sdp->sd_quota_bitmap)
		return error;

	for (x = 0; x < blocks; x++) {
		struct buffer_head *bh;
1324
		const struct gfs2_quota_change *qc;
D
David Teigland 已提交
1325 1326 1327 1328
		unsigned int y;

		if (!extlen) {
			int new = 0;
1329
			error = gfs2_extent_map(&ip->i_inode, x, &new, &dblock, &extlen);
D
David Teigland 已提交
1330 1331 1332 1333
			if (error)
				goto fail;
		}
		error = -EIO;
S
Steven Whitehouse 已提交
1334 1335 1336
		bh = gfs2_meta_ra(ip->i_gl, dblock, extlen);
		if (!bh)
			goto fail;
D
David Teigland 已提交
1337 1338 1339 1340 1341
		if (gfs2_metatype_check(sdp, bh, GFS2_METATYPE_QC)) {
			brelse(bh);
			goto fail;
		}

1342
		qc = (const struct gfs2_quota_change *)(bh->b_data + sizeof(struct gfs2_meta_header));
S
Steven Whitehouse 已提交
1343
		for (y = 0; y < sdp->sd_qc_per_block && slot < sdp->sd_quota_slots;
D
David Teigland 已提交
1344 1345
		     y++, slot++) {
			struct gfs2_quota_data *qd;
1346 1347 1348 1349 1350 1351 1352 1353
			s64 qc_change = be64_to_cpu(qc->qc_change);
			u32 qc_flags = be32_to_cpu(qc->qc_flags);
			enum quota_type qtype = (qc_flags & GFS2_QCF_USER) ?
						USRQUOTA : GRPQUOTA;
			struct kqid qc_id = make_kqid(&init_user_ns, qtype,
						      be32_to_cpu(qc->qc_id));
			qc++;
			if (!qc_change)
D
David Teigland 已提交
1354 1355
				continue;

1356 1357 1358
			hash = gfs2_qd_hash(sdp, qc_id);
			qd = qd_alloc(hash, sdp, qc_id);
			if (qd == NULL) {
D
David Teigland 已提交
1359 1360 1361 1362 1363
				brelse(bh);
				goto fail;
			}

			set_bit(QDF_CHANGE, &qd->qd_flags);
1364
			qd->qd_change = qc_change;
D
David Teigland 已提交
1365 1366 1367
			qd->qd_slot = slot;
			qd->qd_slot_count = 1;

1368
			spin_lock(&qd_lock);
1369
			BUG_ON(test_and_set_bit(slot, sdp->sd_quota_bitmap));
D
David Teigland 已提交
1370 1371
			list_add(&qd->qd_list, &sdp->sd_quota_list);
			atomic_inc(&sdp->sd_quota_count);
1372
			spin_unlock(&qd_lock);
D
David Teigland 已提交
1373

1374 1375 1376 1377
			spin_lock_bucket(hash);
			hlist_bl_add_head_rcu(&qd->qd_hlist, &qd_hash_table[hash]);
			spin_unlock_bucket(hash);

D
David Teigland 已提交
1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390
			found++;
		}

		brelse(bh);
		dblock++;
		extlen--;
	}

	if (found)
		fs_info(sdp, "found %u quota changes\n", found);

	return 0;

1391
fail:
D
David Teigland 已提交
1392 1393 1394 1395 1396 1397 1398 1399 1400
	gfs2_quota_cleanup(sdp);
	return error;
}

void gfs2_quota_cleanup(struct gfs2_sbd *sdp)
{
	struct list_head *head = &sdp->sd_quota_list;
	struct gfs2_quota_data *qd;

1401
	spin_lock(&qd_lock);
D
David Teigland 已提交
1402 1403 1404 1405
	while (!list_empty(head)) {
		qd = list_entry(head->prev, struct gfs2_quota_data, qd_list);

		list_del(&qd->qd_list);
1406

1407
		/* Also remove if this qd exists in the reclaim list */
1408
		list_lru_del(&gfs2_qd_lru, &qd->qd_lru);
D
David Teigland 已提交
1409
		atomic_dec(&sdp->sd_quota_count);
1410
		spin_unlock(&qd_lock);
D
David Teigland 已提交
1411

1412 1413 1414 1415
		spin_lock_bucket(qd->qd_hash);
		hlist_bl_del_rcu(&qd->qd_hlist);
		spin_unlock_bucket(qd->qd_hash);

1416 1417
		gfs2_assert_warn(sdp, !qd->qd_change);
		gfs2_assert_warn(sdp, !qd->qd_slot_count);
D
David Teigland 已提交
1418 1419
		gfs2_assert_warn(sdp, !qd->qd_bh_count);

1420
		gfs2_glock_put(qd->qd_gl);
1421
		call_rcu(&qd->qd_rcu, gfs2_qd_dealloc);
D
David Teigland 已提交
1422

1423
		spin_lock(&qd_lock);
D
David Teigland 已提交
1424
	}
1425
	spin_unlock(&qd_lock);
D
David Teigland 已提交
1426 1427 1428

	gfs2_assert_warn(sdp, !atomic_read(&sdp->sd_quota_count));

1429 1430
	kvfree(sdp->sd_quota_bitmap);
	sdp->sd_quota_bitmap = NULL;
D
David Teigland 已提交
1431 1432
}

1433 1434 1435 1436 1437 1438 1439 1440 1441
static void quotad_error(struct gfs2_sbd *sdp, const char *msg, int error)
{
	if (error == 0 || error == -EROFS)
		return;
	if (!test_bit(SDF_SHUTDOWN, &sdp->sd_flags))
		fs_err(sdp, "gfs2_quotad: %s error %d\n", msg, error);
}

static void quotad_check_timeo(struct gfs2_sbd *sdp, const char *msg,
1442
			       int (*fxn)(struct super_block *sb, int type),
1443 1444 1445 1446
			       unsigned long t, unsigned long *timeo,
			       unsigned int *new_timeo)
{
	if (t >= *timeo) {
1447
		int error = fxn(sdp->sd_vfs, 0);
1448 1449 1450 1451 1452 1453 1454
		quotad_error(sdp, msg, error);
		*timeo = gfs2_tune_get_i(&sdp->sd_tune, new_timeo) * HZ;
	} else {
		*timeo -= t;
	}
}

1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473
static void quotad_check_trunc_list(struct gfs2_sbd *sdp)
{
	struct gfs2_inode *ip;

	while(1) {
		ip = NULL;
		spin_lock(&sdp->sd_trunc_lock);
		if (!list_empty(&sdp->sd_trunc_list)) {
			ip = list_entry(sdp->sd_trunc_list.next,
					struct gfs2_inode, i_trunc_list);
			list_del_init(&ip->i_trunc_list);
		}
		spin_unlock(&sdp->sd_trunc_lock);
		if (ip == NULL)
			return;
		gfs2_glock_finish_truncate(ip);
	}
}

1474 1475 1476 1477 1478 1479 1480 1481
void gfs2_wake_up_statfs(struct gfs2_sbd *sdp) {
	if (!sdp->sd_statfs_force_sync) {
		sdp->sd_statfs_force_sync = 1;
		wake_up(&sdp->sd_quota_wait);
	}
}


1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495
/**
 * gfs2_quotad - Write cached quota changes into the quota file
 * @sdp: Pointer to GFS2 superblock
 *
 */

int gfs2_quotad(void *data)
{
	struct gfs2_sbd *sdp = data;
	struct gfs2_tune *tune = &sdp->sd_tune;
	unsigned long statfs_timeo = 0;
	unsigned long quotad_timeo = 0;
	unsigned long t = 0;
	DEFINE_WAIT(wait);
1496
	int empty;
1497 1498 1499 1500

	while (!kthread_should_stop()) {

		/* Update the master statfs file */
1501 1502 1503 1504 1505 1506 1507 1508 1509
		if (sdp->sd_statfs_force_sync) {
			int error = gfs2_statfs_sync(sdp->sd_vfs, 0);
			quotad_error(sdp, "statfs", error);
			statfs_timeo = gfs2_tune_get(sdp, gt_statfs_quantum) * HZ;
		}
		else
			quotad_check_timeo(sdp, "statfs", gfs2_statfs_sync, t,
				   	   &statfs_timeo,
					   &tune->gt_statfs_quantum);
1510 1511

		/* Update quota file */
1512
		quotad_check_timeo(sdp, "sync", gfs2_quota_sync, t,
1513 1514
				   &quotad_timeo, &tune->gt_quota_quantum);

1515 1516 1517
		/* Check for & recover partially truncated inodes */
		quotad_check_trunc_list(sdp);

1518 1519
		try_to_freeze();

1520 1521
		t = min(quotad_timeo, statfs_timeo);

1522
		prepare_to_wait(&sdp->sd_quota_wait, &wait, TASK_INTERRUPTIBLE);
1523 1524 1525
		spin_lock(&sdp->sd_trunc_lock);
		empty = list_empty(&sdp->sd_trunc_list);
		spin_unlock(&sdp->sd_trunc_lock);
1526
		if (empty && !sdp->sd_statfs_force_sync)
1527 1528 1529
			t -= schedule_timeout(t);
		else
			t = 0;
1530 1531 1532 1533 1534 1535
		finish_wait(&sdp->sd_quota_wait, &wait);
	}

	return 0;
}

1536 1537 1538 1539 1540 1541 1542
static int gfs2_quota_get_xstate(struct super_block *sb,
				 struct fs_quota_stat *fqs)
{
	struct gfs2_sbd *sdp = sb->s_fs_info;

	memset(fqs, 0, sizeof(struct fs_quota_stat));
	fqs->qs_version = FS_QSTAT_VERSION;
1543 1544 1545

	switch (sdp->sd_args.ar_quota) {
	case GFS2_QUOTA_ON:
1546
		fqs->qs_flags |= (FS_QUOTA_UDQ_ENFD | FS_QUOTA_GDQ_ENFD);
1547 1548
		/*FALLTHRU*/
	case GFS2_QUOTA_ACCOUNT:
1549
		fqs->qs_flags |= (FS_QUOTA_UDQ_ACCT | FS_QUOTA_GDQ_ACCT);
1550 1551 1552 1553 1554
		break;
	case GFS2_QUOTA_OFF:
		break;
	}

1555 1556 1557 1558 1559 1560
	if (sdp->sd_quota_inode) {
		fqs->qs_uquota.qfs_ino = GFS2_I(sdp->sd_quota_inode)->i_no_addr;
		fqs->qs_uquota.qfs_nblks = sdp->sd_quota_inode->i_blocks;
	}
	fqs->qs_uquota.qfs_nextents = 1; /* unsupported */
	fqs->qs_gquota = fqs->qs_uquota; /* its the same inode in both cases */
1561
	fqs->qs_incoredqs = list_lru_count(&gfs2_qd_lru);
1562 1563 1564
	return 0;
}

E
Eric W. Biederman 已提交
1565
static int gfs2_get_dqblk(struct super_block *sb, struct kqid qid,
1566
			  struct qc_dqblk *fdq)
S
Steven Whitehouse 已提交
1567 1568 1569 1570 1571 1572 1573
{
	struct gfs2_sbd *sdp = sb->s_fs_info;
	struct gfs2_quota_lvb *qlvb;
	struct gfs2_quota_data *qd;
	struct gfs2_holder q_gh;
	int error;

1574
	memset(fdq, 0, sizeof(*fdq));
S
Steven Whitehouse 已提交
1575 1576 1577 1578

	if (sdp->sd_args.ar_quota == GFS2_QUOTA_OFF)
		return -ESRCH; /* Crazy XFS error code */

1579 1580
	if ((qid.type != USRQUOTA) &&
	    (qid.type != GRPQUOTA))
S
Steven Whitehouse 已提交
1581 1582
		return -EINVAL;

1583
	error = qd_get(sdp, qid, &qd);
S
Steven Whitehouse 已提交
1584 1585 1586 1587 1588 1589
	if (error)
		return error;
	error = do_glock(qd, FORCE, &q_gh);
	if (error)
		goto out;

1590
	qlvb = (struct gfs2_quota_lvb *)qd->qd_gl->gl_lksb.sb_lvbptr;
1591 1592 1593
	fdq->d_spc_hardlimit = be64_to_cpu(qlvb->qb_limit) << sdp->sd_sb.sb_bsize_shift;
	fdq->d_spc_softlimit = be64_to_cpu(qlvb->qb_warn) << sdp->sd_sb.sb_bsize_shift;
	fdq->d_space = be64_to_cpu(qlvb->qb_value) << sdp->sd_sb.sb_bsize_shift;
S
Steven Whitehouse 已提交
1594 1595 1596 1597 1598 1599 1600

	gfs2_glock_dq_uninit(&q_gh);
out:
	qd_put(qd);
	return error;
}

S
Steven Whitehouse 已提交
1601
/* GFS2 only supports a subset of the XFS fields */
1602
#define GFS2_FIELDMASK (QC_SPC_SOFT|QC_SPC_HARD|QC_SPACE)
S
Steven Whitehouse 已提交
1603

E
Eric W. Biederman 已提交
1604
static int gfs2_set_dqblk(struct super_block *sb, struct kqid qid,
1605
			  struct qc_dqblk *fdq)
S
Steven Whitehouse 已提交
1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
{
	struct gfs2_sbd *sdp = sb->s_fs_info;
	struct gfs2_inode *ip = GFS2_I(sdp->sd_quota_inode);
	struct gfs2_quota_data *qd;
	struct gfs2_holder q_gh, i_gh;
	unsigned int data_blocks, ind_blocks;
	unsigned int blocks = 0;
	int alloc_required;
	loff_t offset;
	int error;

	if (sdp->sd_args.ar_quota == GFS2_QUOTA_OFF)
		return -ESRCH; /* Crazy XFS error code */

1620 1621
	if ((qid.type != USRQUOTA) &&
	    (qid.type != GRPQUOTA))
S
Steven Whitehouse 已提交
1622 1623 1624 1625 1626
		return -EINVAL;

	if (fdq->d_fieldmask & ~GFS2_FIELDMASK)
		return -EINVAL;

1627
	error = qd_get(sdp, qid, &qd);
S
Steven Whitehouse 已提交
1628 1629 1630
	if (error)
		return error;

1631 1632 1633 1634
	error = gfs2_rs_alloc(ip);
	if (error)
		goto out_put;

S
Steven Whitehouse 已提交
1635 1636 1637
	mutex_lock(&ip->i_inode.i_mutex);
	error = gfs2_glock_nq_init(qd->qd_gl, LM_ST_EXCLUSIVE, 0, &q_gh);
	if (error)
1638
		goto out_unlockput;
S
Steven Whitehouse 已提交
1639 1640 1641 1642 1643 1644 1645 1646 1647 1648
	error = gfs2_glock_nq_init(ip->i_gl, LM_ST_EXCLUSIVE, 0, &i_gh);
	if (error)
		goto out_q;

	/* Check for existing entry, if none then alloc new blocks */
	error = update_qd(sdp, qd);
	if (error)
		goto out_i;

	/* If nothing has changed, this is a no-op */
1649 1650 1651
	if ((fdq->d_fieldmask & QC_SPC_SOFT) &&
	    ((fdq->d_spc_softlimit >> sdp->sd_sb.sb_bsize_shift) == be64_to_cpu(qd->qd_qb.qb_warn)))
		fdq->d_fieldmask ^= QC_SPC_SOFT;
1652

1653 1654 1655
	if ((fdq->d_fieldmask & QC_SPC_HARD) &&
	    ((fdq->d_spc_hardlimit >> sdp->sd_sb.sb_bsize_shift) == be64_to_cpu(qd->qd_qb.qb_limit)))
		fdq->d_fieldmask ^= QC_SPC_HARD;
1656

1657 1658 1659
	if ((fdq->d_fieldmask & QC_SPACE) &&
	    ((fdq->d_space >> sdp->sd_sb.sb_bsize_shift) == be64_to_cpu(qd->qd_qb.qb_value)))
		fdq->d_fieldmask ^= QC_SPACE;
1660

S
Steven Whitehouse 已提交
1661 1662 1663 1664
	if (fdq->d_fieldmask == 0)
		goto out_i;

	offset = qd2offset(qd);
1665
	alloc_required = gfs2_write_alloc_required(ip, offset, sizeof(struct gfs2_quota));
A
Abhijith Das 已提交
1666 1667
	if (gfs2_is_stuffed(ip))
		alloc_required = 1;
S
Steven Whitehouse 已提交
1668
	if (alloc_required) {
1669
		struct gfs2_alloc_parms ap = { .aflags = 0, };
S
Steven Whitehouse 已提交
1670 1671
		gfs2_write_calc_reserv(ip, sizeof(struct gfs2_quota),
				       &data_blocks, &ind_blocks);
1672
		blocks = 1 + data_blocks + ind_blocks;
1673 1674
		ap.target = blocks;
		error = gfs2_inplace_reserve(ip, &ap);
S
Steven Whitehouse 已提交
1675
		if (error)
1676
			goto out_i;
1677
		blocks += gfs2_rg_blocks(ip, blocks);
S
Steven Whitehouse 已提交
1678 1679
	}

A
Abhijith Das 已提交
1680 1681 1682
	/* Some quotas span block boundaries and can update two blocks,
	   adding an extra block to the transaction to handle such quotas */
	error = gfs2_trans_begin(sdp, blocks + RES_DINODE + 2, 0);
S
Steven Whitehouse 已提交
1683 1684 1685 1686 1687 1688 1689 1690
	if (error)
		goto out_release;

	/* Apply changes */
	error = gfs2_adjust_quota(ip, offset, 0, qd, fdq);

	gfs2_trans_end(sdp);
out_release:
1691
	if (alloc_required)
S
Steven Whitehouse 已提交
1692 1693 1694 1695 1696
		gfs2_inplace_release(ip);
out_i:
	gfs2_glock_dq_uninit(&i_gh);
out_q:
	gfs2_glock_dq_uninit(&q_gh);
1697
out_unlockput:
S
Steven Whitehouse 已提交
1698
	mutex_unlock(&ip->i_inode.i_mutex);
1699
out_put:
S
Steven Whitehouse 已提交
1700 1701 1702 1703
	qd_put(qd);
	return error;
}

1704 1705
const struct quotactl_ops gfs2_quotactl_ops = {
	.quota_sync     = gfs2_quota_sync,
1706
	.get_xstate     = gfs2_quota_get_xstate,
C
Christoph Hellwig 已提交
1707
	.get_dqblk	= gfs2_get_dqblk,
C
Christoph Hellwig 已提交
1708
	.set_dqblk	= gfs2_set_dqblk,
1709
};
1710 1711 1712 1713 1714 1715 1716 1717

void __init gfs2_quota_hash_init(void)
{
	unsigned i;

	for(i = 0; i < GFS2_QD_HASH_SIZE; i++)
		INIT_HLIST_BL_HEAD(&qd_hash_table[i]);
}