quota.c 38.1 KB
Newer Older
David Teigland's avatar
David Teigland committed
1 2
/*
 * Copyright (C) Sistina Software, Inc.  1997-2003 All rights reserved.
3
 * Copyright (C) 2004-2007 Red Hat, Inc.  All rights reserved.
David Teigland's avatar
David Teigland committed
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.
David Teigland's avatar
David Teigland committed
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
David Teigland's avatar
David Teigland committed
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
 * 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.
 */

#include <linux/sched.h>
#include <linux/slab.h>
41
#include <linux/mm.h>
David Teigland's avatar
David Teigland committed
42 43 44 45
#include <linux/spinlock.h>
#include <linux/completion.h>
#include <linux/buffer_head.h>
#include <linux/sort.h>
46
#include <linux/fs.h>
47
#include <linux/bio.h>
48
#include <linux/gfs2_ondisk.h>
49 50
#include <linux/kthread.h>
#include <linux/freezer.h>
51
#include <linux/quota.h>
52
#include <linux/dqblk_xfs.h>
David Teigland's avatar
David Teigland committed
53 54

#include "gfs2.h"
55
#include "incore.h"
David Teigland's avatar
David Teigland committed
56 57 58 59 60 61 62 63 64
#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"
65
#include "inode.h"
66
#include "util.h"
David Teigland's avatar
David Teigland committed
67 68 69 70

#define QUOTA_USER 1
#define QUOTA_GROUP 0

71 72 73 74 75 76
struct gfs2_quota_change_host {
	u64 qc_change;
	u32 qc_flags; /* GFS2_QCF_... */
	u32 qc_id;
};

77 78
static LIST_HEAD(qd_lru_list);
static atomic_t qd_lru_count = ATOMIC_INIT(0);
Xu Gang's avatar
Xu Gang committed
79
static DEFINE_SPINLOCK(qd_lru_lock);
80

81
int gfs2_shrink_qd_memory(struct shrinker *shrink, struct shrink_control *sc)
82 83 84
{
	struct gfs2_quota_data *qd;
	struct gfs2_sbd *sdp;
85
	int nr_to_scan = sc->nr_to_scan;
86

87
	if (nr_to_scan == 0)
88 89
		goto out;

90
	if (!(sc->gfp_mask & __GFP_FS))
91 92 93
		return -1;

	spin_lock(&qd_lru_lock);
94
	while (nr_to_scan && !list_empty(&qd_lru_list)) {
95 96 97 98 99 100 101 102 103 104 105
		qd = list_entry(qd_lru_list.next,
				struct gfs2_quota_data, qd_reclaim);
		sdp = qd->qd_gl->gl_sbd;

		/* Free from the filesystem-specific list */
		list_del(&qd->qd_list);

		gfs2_assert_warn(sdp, !qd->qd_change);
		gfs2_assert_warn(sdp, !qd->qd_slot_count);
		gfs2_assert_warn(sdp, !qd->qd_bh_count);

106
		gfs2_glock_put(qd->qd_gl);
107 108 109 110 111 112 113 114
		atomic_dec(&sdp->sd_quota_count);

		/* Delete it from the common reclaim list */
		list_del_init(&qd->qd_reclaim);
		atomic_dec(&qd_lru_count);
		spin_unlock(&qd_lru_lock);
		kmem_cache_free(gfs2_quotad_cachep, qd);
		spin_lock(&qd_lru_lock);
115
		nr_to_scan--;
116 117 118 119 120 121 122
	}
	spin_unlock(&qd_lru_lock);

out:
	return (atomic_read(&qd_lru_count) * sysctl_vfs_cache_pressure) / 100;
}

123
static u64 qd2offset(struct gfs2_quota_data *qd)
David Teigland's avatar
David Teigland committed
124
{
125
	u64 offset;
David Teigland's avatar
David Teigland committed
126

127
	offset = 2 * (u64)qd->qd_id + !test_bit(QDF_USER, &qd->qd_flags);
David Teigland's avatar
David Teigland committed
128 129 130 131 132
	offset *= sizeof(struct gfs2_quota);

	return offset;
}

133
static int qd_alloc(struct gfs2_sbd *sdp, int user, u32 id,
David Teigland's avatar
David Teigland committed
134 135 136 137 138
		    struct gfs2_quota_data **qdp)
{
	struct gfs2_quota_data *qd;
	int error;

139
	qd = kmem_cache_zalloc(gfs2_quotad_cachep, GFP_NOFS);
David Teigland's avatar
David Teigland committed
140 141 142
	if (!qd)
		return -ENOMEM;

143
	atomic_set(&qd->qd_count, 1);
David Teigland's avatar
David Teigland committed
144 145 146 147
	qd->qd_id = id;
	if (user)
		set_bit(QDF_USER, &qd->qd_flags);
	qd->qd_slot = -1;
148
	INIT_LIST_HEAD(&qd->qd_reclaim);
David Teigland's avatar
David Teigland committed
149

150
	error = gfs2_glock_get(sdp, 2 * (u64)id + !user,
David Teigland's avatar
David Teigland committed
151 152 153 154 155 156 157 158
			      &gfs2_quota_glops, CREATE, &qd->qd_gl);
	if (error)
		goto fail;

	*qdp = qd;

	return 0;

159
fail:
160
	kmem_cache_free(gfs2_quotad_cachep, qd);
David Teigland's avatar
David Teigland committed
161 162 163
	return error;
}

164
static int qd_get(struct gfs2_sbd *sdp, int user, u32 id,
David Teigland's avatar
David Teigland committed
165 166 167 168 169 170 171 172 173
		  struct gfs2_quota_data **qdp)
{
	struct gfs2_quota_data *qd = NULL, *new_qd = NULL;
	int error, found;

	*qdp = NULL;

	for (;;) {
		found = 0;
174
		spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
175 176 177
		list_for_each_entry(qd, &sdp->sd_quota_list, qd_list) {
			if (qd->qd_id == id &&
			    !test_bit(QDF_USER, &qd->qd_flags) == !user) {
178 179 180 181 182 183 184
				if (!atomic_read(&qd->qd_count) &&
				    !list_empty(&qd->qd_reclaim)) {
					/* Remove it from reclaim list */
					list_del_init(&qd->qd_reclaim);
					atomic_dec(&qd_lru_count);
				}
				atomic_inc(&qd->qd_count);
David Teigland's avatar
David Teigland committed
185 186 187 188 189 190 191 192 193 194 195 196 197 198 199
				found = 1;
				break;
			}
		}

		if (!found)
			qd = NULL;

		if (!qd && new_qd) {
			qd = new_qd;
			list_add(&qd->qd_list, &sdp->sd_quota_list);
			atomic_inc(&sdp->sd_quota_count);
			new_qd = NULL;
		}

200
		spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
201

202
		if (qd) {
David Teigland's avatar
David Teigland committed
203
			if (new_qd) {
204
				gfs2_glock_put(new_qd->qd_gl);
205
				kmem_cache_free(gfs2_quotad_cachep, new_qd);
David Teigland's avatar
David Teigland committed
206 207 208 209 210 211 212 213 214 215 216 217 218 219
			}
			*qdp = qd;
			return 0;
		}

		error = qd_alloc(sdp, user, id, &new_qd);
		if (error)
			return error;
	}
}

static void qd_hold(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
220 221
	gfs2_assert(sdp, atomic_read(&qd->qd_count));
	atomic_inc(&qd->qd_count);
David Teigland's avatar
David Teigland committed
222 223 224 225
}

static void qd_put(struct gfs2_quota_data *qd)
{
226 227 228 229 230 231
	if (atomic_dec_and_lock(&qd->qd_count, &qd_lru_lock)) {
		/* Add to the reclaim list */
		list_add_tail(&qd->qd_reclaim, &qd_lru_list);
		atomic_inc(&qd_lru_count);
		spin_unlock(&qd_lru_lock);
	}
David Teigland's avatar
David Teigland committed
232 233 234 235 236 237 238 239
}

static int slot_get(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
	unsigned int c, o = 0, b;
	unsigned char byte = 0;

240
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
241 242

	if (qd->qd_slot_count++) {
243
		spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
244 245 246 247 248 249 250 251 252 253 254 255
		return 0;
	}

	for (c = 0; c < sdp->sd_quota_chunks; c++)
		for (o = 0; o < PAGE_SIZE; o++) {
			byte = sdp->sd_quota_bitmap[c][o];
			if (byte != 0xFF)
				goto found;
		}

	goto fail;

256
found:
David Teigland's avatar
David Teigland committed
257 258 259 260 261 262 263 264 265 266
	for (b = 0; b < 8; b++)
		if (!(byte & (1 << b)))
			break;
	qd->qd_slot = c * (8 * PAGE_SIZE) + o * 8 + b;

	if (qd->qd_slot >= sdp->sd_quota_slots)
		goto fail;

	sdp->sd_quota_bitmap[c][o] |= 1 << b;

267
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
268 269 270

	return 0;

271
fail:
David Teigland's avatar
David Teigland committed
272
	qd->qd_slot_count--;
273
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
274 275 276 277 278 279 280
	return -ENOSPC;
}

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

281
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
282 283
	gfs2_assert(sdp, qd->qd_slot_count);
	qd->qd_slot_count++;
284
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
285 286 287 288 289 290
}

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

291
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
292 293 294 295 296
	gfs2_assert(sdp, qd->qd_slot_count);
	if (!--qd->qd_slot_count) {
		gfs2_icbit_munge(sdp, sdp->sd_quota_bitmap, qd->qd_slot, 0);
		qd->qd_slot = -1;
	}
297
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
298 299 300 301 302
}

static int bh_get(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
303
	struct gfs2_inode *ip = GFS2_I(sdp->sd_qc_inode);
David Teigland's avatar
David Teigland committed
304 305 306
	unsigned int block, offset;
	struct buffer_head *bh;
	int error;
307
	struct buffer_head bh_map = { .b_state = 0, .b_blocknr = 0 };
David Teigland's avatar
David Teigland committed
308

309
	mutex_lock(&sdp->sd_quota_mutex);
David Teigland's avatar
David Teigland committed
310 311

	if (qd->qd_bh_count++) {
312
		mutex_unlock(&sdp->sd_quota_mutex);
David Teigland's avatar
David Teigland committed
313 314 315 316
		return 0;
	}

	block = qd->qd_slot / sdp->sd_qc_per_block;
317
	offset = qd->qd_slot % sdp->sd_qc_per_block;
David Teigland's avatar
David Teigland committed
318

319
	bh_map.b_size = 1 << ip->i_inode.i_blkbits;
320
	error = gfs2_block_map(&ip->i_inode, block, &bh_map, 0);
David Teigland's avatar
David Teigland committed
321 322
	if (error)
		goto fail;
323
	error = gfs2_meta_read(ip->i_gl, bh_map.b_blocknr, DIO_WAIT, &bh);
David Teigland's avatar
David Teigland committed
324 325 326 327 328 329 330 331 332 333 334
	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));

Josef Whiter's avatar
Josef Whiter committed
335
	mutex_unlock(&sdp->sd_quota_mutex);
David Teigland's avatar
David Teigland committed
336 337 338

	return 0;

339
fail_brelse:
David Teigland's avatar
David Teigland committed
340
	brelse(bh);
341
fail:
David Teigland's avatar
David Teigland committed
342
	qd->qd_bh_count--;
343
	mutex_unlock(&sdp->sd_quota_mutex);
David Teigland's avatar
David Teigland committed
344 345 346 347 348 349 350
	return error;
}

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

351
	mutex_lock(&sdp->sd_quota_mutex);
David Teigland's avatar
David Teigland committed
352 353 354 355 356 357
	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;
	}
358
	mutex_unlock(&sdp->sd_quota_mutex);
David Teigland's avatar
David Teigland committed
359 360 361 362 363 364 365 366 367 368 369 370 371
}

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;

372
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
373 374 375 376 377 378 379 380 381 382

	list_for_each_entry(qd, &sdp->sd_quota_list, qd_list) {
		if (test_bit(QDF_LOCKED, &qd->qd_flags) ||
		    !test_bit(QDF_CHANGE, &qd->qd_flags) ||
		    qd->qd_sync_gen >= sdp->sd_quota_sync_gen)
			continue;

		list_move_tail(&qd->qd_list, &sdp->sd_quota_list);

		set_bit(QDF_LOCKED, &qd->qd_flags);
383 384
		gfs2_assert_warn(sdp, atomic_read(&qd->qd_count));
		atomic_inc(&qd->qd_count);
David Teigland's avatar
David Teigland committed
385 386 387 388 389 390 391 392 393 394 395
		qd->qd_change_sync = qd->qd_change;
		gfs2_assert_warn(sdp, qd->qd_slot_count);
		qd->qd_slot_count++;
		found = 1;

		break;
	}

	if (!found)
		qd = NULL;

396
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420

	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 int qd_trylock(struct gfs2_quota_data *qd)
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;

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

421
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
422 423 424

	if (test_bit(QDF_LOCKED, &qd->qd_flags) ||
	    !test_bit(QDF_CHANGE, &qd->qd_flags)) {
425
		spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
426 427 428 429 430 431
		return 0;
	}

	list_move_tail(&qd->qd_list, &sdp->sd_quota_list);

	set_bit(QDF_LOCKED, &qd->qd_flags);
432 433
	gfs2_assert_warn(sdp, atomic_read(&qd->qd_count));
	atomic_inc(&qd->qd_count);
David Teigland's avatar
David Teigland committed
434 435 436 437
	qd->qd_change_sync = qd->qd_change;
	gfs2_assert_warn(sdp, qd->qd_slot_count);
	qd->qd_slot_count++;

438
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
439 440 441 442 443 444 445 446 447 448 449 450 451 452

	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);
		return 0;
	}

	return 1;
}

static void qd_unlock(struct gfs2_quota_data *qd)
{
453 454
	gfs2_assert_warn(qd->qd_gl->gl_sbd,
			 test_bit(QDF_LOCKED, &qd->qd_flags));
David Teigland's avatar
David Teigland committed
455 456 457 458 459 460
	clear_bit(QDF_LOCKED, &qd->qd_flags);
	bh_put(qd);
	slot_put(qd);
	qd_put(qd);
}

461
static int qdsb_get(struct gfs2_sbd *sdp, int user, u32 id,
David Teigland's avatar
David Teigland committed
462 463 464 465
		    struct gfs2_quota_data **qdp)
{
	int error;

466
	error = qd_get(sdp, user, id, qdp);
David Teigland's avatar
David Teigland committed
467 468 469 470 471 472 473 474 475 476 477 478 479
	if (error)
		return error;

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

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

	return 0;

480
fail_slot:
David Teigland's avatar
David Teigland committed
481
	slot_put(*qdp);
482
fail:
David Teigland's avatar
David Teigland committed
483 484 485 486 487 488 489 490 491 492 493
	qd_put(*qdp);
	return error;
}

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

494
int gfs2_quota_hold(struct gfs2_inode *ip, u32 uid, u32 gid)
David Teigland's avatar
David Teigland committed
495
{
496
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
497 498
	struct gfs2_qadata *qa = ip->i_qadata;
	struct gfs2_quota_data **qd = qa->qa_qd;
David Teigland's avatar
David Teigland committed
499 500
	int error;

501
	if (gfs2_assert_warn(sdp, !qa->qa_qd_num) ||
David Teigland's avatar
David Teigland committed
502 503 504 505 506 507
	    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;

508
	error = qdsb_get(sdp, QUOTA_USER, ip->i_inode.i_uid, qd);
David Teigland's avatar
David Teigland committed
509 510
	if (error)
		goto out;
511
	qa->qa_qd_num++;
David Teigland's avatar
David Teigland committed
512 513
	qd++;

514
	error = qdsb_get(sdp, QUOTA_GROUP, ip->i_inode.i_gid, qd);
David Teigland's avatar
David Teigland committed
515 516
	if (error)
		goto out;
517
	qa->qa_qd_num++;
David Teigland's avatar
David Teigland committed
518 519
	qd++;

520
	if (uid != NO_QUOTA_CHANGE && uid != ip->i_inode.i_uid) {
521
		error = qdsb_get(sdp, QUOTA_USER, uid, qd);
David Teigland's avatar
David Teigland committed
522 523
		if (error)
			goto out;
524
		qa->qa_qd_num++;
David Teigland's avatar
David Teigland committed
525 526 527
		qd++;
	}

528
	if (gid != NO_QUOTA_CHANGE && gid != ip->i_inode.i_gid) {
529
		error = qdsb_get(sdp, QUOTA_GROUP, gid, qd);
David Teigland's avatar
David Teigland committed
530 531
		if (error)
			goto out;
532
		qa->qa_qd_num++;
David Teigland's avatar
David Teigland committed
533 534 535
		qd++;
	}

536
out:
David Teigland's avatar
David Teigland committed
537 538 539 540 541 542 543
	if (error)
		gfs2_quota_unhold(ip);
	return error;
}

void gfs2_quota_unhold(struct gfs2_inode *ip)
{
544
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
545
	struct gfs2_qadata *qa = ip->i_qadata;
David Teigland's avatar
David Teigland committed
546 547 548 549
	unsigned int x;

	gfs2_assert_warn(sdp, !test_bit(GIF_QD_LOCKED, &ip->i_flags));

550 551 552
	for (x = 0; x < qa->qa_qd_num; x++) {
		qdsb_put(qa->qa_qd[x]);
		qa->qa_qd[x] = NULL;
David Teigland's avatar
David Teigland committed
553
	}
554
	qa->qa_qd_num = 0;
David Teigland's avatar
David Teigland committed
555 556 557 558
}

static int sort_qd(const void *a, const void *b)
{
559 560
	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;
David Teigland's avatar
David Teigland committed
561 562 563 564

	if (!test_bit(QDF_USER, &qd_a->qd_flags) !=
	    !test_bit(QDF_USER, &qd_b->qd_flags)) {
		if (test_bit(QDF_USER, &qd_a->qd_flags))
565
			return -1;
David Teigland's avatar
David Teigland committed
566
		else
567
			return 1;
David Teigland's avatar
David Teigland committed
568
	}
569 570 571 572
	if (qd_a->qd_id < qd_b->qd_id)
		return -1;
	if (qd_a->qd_id > qd_b->qd_id)
		return 1;
David Teigland's avatar
David Teigland committed
573

574
	return 0;
David Teigland's avatar
David Teigland committed
575 576
}

577
static void do_qc(struct gfs2_quota_data *qd, s64 change)
David Teigland's avatar
David Teigland committed
578 579
{
	struct gfs2_sbd *sdp = qd->qd_gl->gl_sbd;
580
	struct gfs2_inode *ip = GFS2_I(sdp->sd_qc_inode);
David Teigland's avatar
David Teigland committed
581
	struct gfs2_quota_change *qc = qd->qd_bh_qc;
582
	s64 x;
David Teigland's avatar
David Teigland committed
583

584
	mutex_lock(&sdp->sd_quota_mutex);
585
	gfs2_trans_add_bh(ip->i_gl, qd->qd_bh, 1);
David Teigland's avatar
David Teigland committed
586 587 588 589 590 591 592 593 594

	if (!test_bit(QDF_CHANGE, &qd->qd_flags)) {
		qc->qc_change = 0;
		qc->qc_flags = 0;
		if (test_bit(QDF_USER, &qd->qd_flags))
			qc->qc_flags = cpu_to_be32(GFS2_QCF_USER);
		qc->qc_id = cpu_to_be32(qd->qd_id);
	}

595
	x = be64_to_cpu(qc->qc_change) + change;
David Teigland's avatar
David Teigland committed
596 597
	qc->qc_change = cpu_to_be64(x);

598
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
599
	qd->qd_change = x;
600
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
601 602 603 604 605 606 607 608 609 610 611 612

	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);
	}
613

614
	mutex_unlock(&sdp->sd_quota_mutex);
David Teigland's avatar
David Teigland committed
615 616
}

617
/**
618 619 620
 * gfs2_adjust_quota - adjust record of current block usage
 * @ip: The quota inode
 * @loc: Offset of the entry in the quota file
621
 * @change: The amount of usage change to record
622
 * @qd: The quota data
623
 * @fdq: The updated limits to record
624 625 626
 *
 * This function was mostly borrowed from gfs2_block_truncate_page which was
 * in turn mostly borrowed from ext3
627 628
 *
 * Returns: 0 or -ve on error
629
 */
630

631
static int gfs2_adjust_quota(struct gfs2_inode *ip, loff_t loc,
632 633
			     s64 change, struct gfs2_quota_data *qd,
			     struct fs_disk_quota *fdq)
634
{
635
	struct inode *inode = &ip->i_inode;
636
	struct gfs2_sbd *sdp = GFS2_SB(inode);
637 638
	struct address_space *mapping = inode->i_mapping;
	unsigned long index = loc >> PAGE_CACHE_SHIFT;
639
	unsigned offset = loc & (PAGE_CACHE_SIZE - 1);
640
	unsigned blocksize, iblock, pos;
641
	struct buffer_head *bh;
642
	struct page *page;
643 644 645
	void *kaddr, *ptr;
	struct gfs2_quota q, *qp;
	int err, nbytes;
646
	u64 size;
647

Steven Whitehouse's avatar
Steven Whitehouse committed
648 649 650 651 652
	if (gfs2_is_stuffed(ip)) {
		err = gfs2_unstuff_dinode(ip, NULL);
		if (err)
			return err;
	}
653 654

	memset(&q, 0, sizeof(struct gfs2_quota));
655
	err = gfs2_internal_read(ip, (char *)&q, &loc, sizeof(q));
656 657 658 659 660 661 662 663 664 665 666
	if (err < 0)
		return err;

	err = -EIO;
	qp = &q;
	qp->qu_value = be64_to_cpu(qp->qu_value);
	qp->qu_value += change;
	qp->qu_value = cpu_to_be64(qp->qu_value);
	qd->qd_qb.qb_value = qp->qu_value;
	if (fdq) {
		if (fdq->d_fieldmask & FS_DQ_BSOFT) {
667
			qp->qu_warn = cpu_to_be64(fdq->d_blk_softlimit >> sdp->sd_fsb2bb_shift);
668 669 670
			qd->qd_qb.qb_warn = qp->qu_warn;
		}
		if (fdq->d_fieldmask & FS_DQ_BHARD) {
671
			qp->qu_limit = cpu_to_be64(fdq->d_blk_hardlimit >> sdp->sd_fsb2bb_shift);
672 673
			qd->qd_qb.qb_limit = qp->qu_limit;
		}
674 675 676 677
		if (fdq->d_fieldmask & FS_DQ_BCOUNT) {
			qp->qu_value = cpu_to_be64(fdq->d_bcount >> sdp->sd_fsb2bb_shift);
			qd->qd_qb.qb_value = qp->qu_value;
		}
678 679 680 681 682 683
	}

	/* Write the quota into the quota file on disk */
	ptr = qp;
	nbytes = sizeof(struct gfs2_quota);
get_a_page:
684
	page = find_or_create_page(mapping, index, GFP_NOFS);
685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702
	if (!page)
		return -ENOMEM;

	blocksize = inode->i_sb->s_blocksize;
	iblock = index << (PAGE_CACHE_SHIFT - inode->i_sb->s_blocksize_bits);

	if (!page_has_buffers(page))
		create_empty_buffers(page, blocksize, 0);

	bh = page_buffers(page);
	pos = blocksize;
	while (offset >= pos) {
		bh = bh->b_this_page;
		iblock++;
		pos += blocksize;
	}

	if (!buffer_mapped(bh)) {
703
		gfs2_block_map(inode, iblock, bh, 1);
704
		if (!buffer_mapped(bh))
705 706
			goto unlock_out;
		/* If it's a newly allocated disk block for quota, zero it */
707 708
		if (buffer_new(bh))
			zero_user(page, pos - blocksize, bh->b_size);
709 710 711 712 713 714
	}

	if (PageUptodate(page))
		set_buffer_uptodate(bh);

	if (!buffer_uptodate(bh)) {
Steven Whitehouse's avatar
Steven Whitehouse committed
715
		ll_rw_block(READ | REQ_META, 1, &bh);
716 717
		wait_on_buffer(bh);
		if (!buffer_uptodate(bh))
718
			goto unlock_out;
719 720 721 722
	}

	gfs2_trans_add_bh(ip->i_gl, bh, 0);

723
	kaddr = kmap_atomic(page);
724 725 726
	if (offset + sizeof(struct gfs2_quota) > PAGE_CACHE_SIZE)
		nbytes = PAGE_CACHE_SIZE - offset;
	memcpy(kaddr + offset, ptr, nbytes);
727
	flush_dcache_page(page);
728
	kunmap_atomic(kaddr);
729 730 731 732 733
	unlock_page(page);
	page_cache_release(page);

	/* If quota straddles page boundary, we need to update the rest of the
	 * quota at the beginning of the next page */
734
	if ((offset + sizeof(struct gfs2_quota)) > PAGE_CACHE_SIZE) {
735 736 737 738 739 740
		ptr = ptr + nbytes;
		nbytes = sizeof(struct gfs2_quota) - nbytes;
		offset = 0;
		index++;
		goto get_a_page;
	}
741 742

	size = loc + sizeof(struct gfs2_quota);
Steven Whitehouse's avatar
Steven Whitehouse committed
743
	if (size > inode->i_size)
744 745 746
		i_size_write(inode, size);
	inode->i_mtime = inode->i_atime = CURRENT_TIME;
	mark_inode_dirty(inode);
747
	return 0;
748

749
unlock_out:
750 751 752 753 754
	unlock_page(page);
	page_cache_release(page);
	return err;
}

David Teigland's avatar
David Teigland committed
755 756 757
static int do_sync(unsigned int num_qd, struct gfs2_quota_data **qda)
{
	struct gfs2_sbd *sdp = (*qda)->qd_gl->gl_sbd;
758
	struct gfs2_inode *ip = GFS2_I(sdp->sd_quota_inode);
David Teigland's avatar
David Teigland committed
759 760 761 762
	unsigned int data_blocks, ind_blocks;
	struct gfs2_holder *ghs, i_gh;
	unsigned int qx, x;
	struct gfs2_quota_data *qd;
763
	loff_t offset;
764
	unsigned int nalloc = 0, blocks;
David Teigland's avatar
David Teigland committed
765 766
	int error;

767 768 769 770
	error = gfs2_rs_alloc(ip);
	if (error)
		return error;

David Teigland's avatar
David Teigland committed
771 772 773
	gfs2_write_calc_reserv(ip, sizeof(struct gfs2_quota),
			      &data_blocks, &ind_blocks);

Josef Bacik's avatar
Josef Bacik committed
774
	ghs = kcalloc(num_qd, sizeof(struct gfs2_holder), GFP_NOFS);
David Teigland's avatar
David Teigland committed
775 776 777 778
	if (!ghs)
		return -ENOMEM;

	sort(qda, num_qd, sizeof(struct gfs2_quota_data *), sort_qd, NULL);
779
	mutex_lock_nested(&ip->i_inode.i_mutex, I_MUTEX_QUOTA);
David Teigland's avatar
David Teigland committed
780
	for (qx = 0; qx < num_qd; qx++) {
781
		error = gfs2_glock_nq_init(qda[qx]->qd_gl, LM_ST_EXCLUSIVE,
David Teigland's avatar
David Teigland committed
782 783 784 785 786 787 788 789 790 791 792
					   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]);
793 794
		if (gfs2_write_alloc_required(ip, offset,
					      sizeof(struct gfs2_quota)))
David Teigland's avatar
David Teigland committed
795 796 797
			nalloc++;
	}

798 799 800 801 802 803
	/* 
	 * 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
	 */
804 805 806 807
	/* +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;
David Teigland's avatar
David Teigland committed
808

809 810
	error = gfs2_inplace_reserve(ip, 1 +
				     (nalloc * (data_blocks + ind_blocks)));
811 812
	if (error)
		goto out_alloc;
David Teigland's avatar
David Teigland committed
813

814
	if (nalloc)
815
		blocks += gfs2_rg_blocks(ip) + nalloc * ind_blocks + RES_STATFS;
816 817 818 819

	error = gfs2_trans_begin(sdp, blocks, 0);
	if (error)
		goto out_ipres;
David Teigland's avatar
David Teigland committed
820 821 822 823

	for (x = 0; x < num_qd; x++) {
		qd = qda[x];
		offset = qd2offset(qd);
824
		error = gfs2_adjust_quota(ip, offset, qd->qd_change_sync, qd, NULL);
825
		if (error)
David Teigland's avatar
David Teigland committed
826 827 828
			goto out_end_trans;

		do_qc(qd, -qd->qd_change_sync);
829
		set_bit(QDF_REFRESH, &qd->qd_flags);
David Teigland's avatar
David Teigland committed
830 831 832 833
	}

	error = 0;

834
out_end_trans:
David Teigland's avatar
David Teigland committed
835
	gfs2_trans_end(sdp);
836
out_ipres:
837
	gfs2_inplace_release(ip);
838
out_alloc:
David Teigland's avatar
David Teigland committed
839
	gfs2_glock_dq_uninit(&i_gh);
840
out:
David Teigland's avatar
David Teigland committed
841 842
	while (qx--)
		gfs2_glock_dq_uninit(&ghs[qx]);
843
	mutex_unlock(&ip->i_inode.i_mutex);
David Teigland's avatar
David Teigland committed
844
	kfree(ghs);
845
	gfs2_log_flush(ip->i_gl->gl_sbd, ip->i_gl);
David Teigland's avatar
David Teigland committed
846 847 848
	return error;
}

849 850 851 852 853 854 855 856 857 858
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);
859
	error = gfs2_internal_read(ip, (char *)&q, &pos, sizeof(q));
860 861 862 863 864 865 866 867 868 869 870 871 872 873
	if (error < 0)
		return error;

	qlvb = (struct gfs2_quota_lvb *)qd->qd_gl->gl_lvb;
	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;
}

David Teigland's avatar
David Teigland committed
874 875 876 877
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;
878
	struct gfs2_inode *ip = GFS2_I(sdp->sd_quota_inode);
David Teigland's avatar
David Teigland committed
879 880 881
	struct gfs2_holder i_gh;
	int error;

882
restart:
David Teigland's avatar
David Teigland committed
883 884 885 886
	error = gfs2_glock_nq_init(qd->qd_gl, LM_ST_SHARED, 0, q_gh);
	if (error)
		return error;

887
	qd->qd_qb = *(struct gfs2_quota_lvb *)qd->qd_gl->gl_lvb;
David Teigland's avatar
David Teigland committed
888

889
	if (force_refresh || qd->qd_qb.qb_magic != cpu_to_be32(GFS2_MAGIC)) {
David Teigland's avatar
David Teigland committed
890
		gfs2_glock_dq_uninit(q_gh);
891 892
		error = gfs2_glock_nq_init(qd->qd_gl, LM_ST_EXCLUSIVE,
					   GL_NOCACHE, q_gh);
David Teigland's avatar
David Teigland committed
893 894 895
		if (error)
			return error;

896
		error = gfs2_glock_nq_init(ip->i_gl, LM_ST_SHARED, 0, &i_gh);
David Teigland's avatar
David Teigland committed
897 898 899
		if (error)
			goto fail;

900 901
		error = update_qd(sdp, qd);
		if (error)
902
			goto fail_gunlock;
David Teigland's avatar
David Teigland committed
903

904
		gfs2_glock_dq_uninit(&i_gh);
905 906 907
		gfs2_glock_dq_uninit(q_gh);
		force_refresh = 0;
		goto restart;
David Teigland's avatar
David Teigland committed
908 909 910 911
	}

	return 0;

912
fail_gunlock:
David Teigland's avatar
David Teigland committed
913
	gfs2_glock_dq_uninit(&i_gh);
914
fail:
David Teigland's avatar
David Teigland committed
915 916 917 918
	gfs2_glock_dq_uninit(q_gh);
	return error;
}

919
int gfs2_quota_lock(struct gfs2_inode *ip, u32 uid, u32 gid)
David Teigland's avatar
David Teigland committed
920
{
921
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
922
	struct gfs2_qadata *qa = ip->i_qadata;
923
	struct gfs2_quota_data *qd;
David Teigland's avatar
David Teigland committed
924 925 926
	unsigned int x;
	int error = 0;

Steven Whitehouse's avatar
Steven Whitehouse committed
927 928 929
	error = gfs2_quota_hold(ip, uid, gid);
	if (error)
		return error;
David Teigland's avatar
David Teigland committed
930 931 932 933 934

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

935
	sort(qa->qa_qd, qa->qa_qd_num, sizeof(struct gfs2_quota_data *),
David Teigland's avatar
David Teigland committed
936 937
	     sort_qd, NULL);

938
	for (x = 0; x < qa->qa_qd_num; x++) {
939
		int force = NO_FORCE;
940
		qd = qa->qa_qd[x];
941 942
		if (test_and_clear_bit(QDF_REFRESH, &qd->qd_flags))
			force = FORCE;
943
		error = do_glock(qd, force, &qa->qa_qd_ghs[x]);
David Teigland's avatar
David Teigland committed
944 945 946 947 948 949 950 951
		if (error)
			break;
	}

	if (!error)
		set_bit(GIF_QD_LOCKED, &ip->i_flags);
	else {
		while (x--)
952
			gfs2_glock_dq_uninit(&qa->qa_qd_ghs[x]);
David Teigland's avatar
David Teigland committed
953 954 955 956 957 958 959 960 961 962
		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;
963
	s64 value;
David Teigland's avatar
David Teigland committed
964 965 966 967 968 969
	unsigned int num, den;
	int do_sync = 1;

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

970
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
971
	value = qd->qd_change;
972
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
973 974 975 976 977 978 979 980

	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;
981 982
	else if ((s64)be64_to_cpu(qd->qd_qb.qb_value) >=
		 (s64)be64_to_cpu(qd->qd_qb.qb_limit))
David Teigland's avatar
David Teigland committed
983 984 985
		do_sync = 0;
	else {
		value *= gfs2_jindex_size(sdp) * num;
986
		value = div_s64(value, den);
987
		value += (s64)be64_to_cpu(qd->qd_qb.qb_value);
988
		if (value < (s64)be64_to_cpu(qd->qd_qb.qb_limit))
David Teigland's avatar
David Teigland committed
989 990 991 992 993 994 995 996
			do_sync = 0;
	}

	return do_sync;
}

void gfs2_quota_unlock(struct gfs2_inode *ip)
{
997
	struct gfs2_qadata *qa = ip->i_qadata;
David Teigland's avatar
David Teigland committed
998 999 1000 1001 1002 1003 1004
	struct gfs2_quota_data *qda[4];
	unsigned int count = 0;
	unsigned int x;

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

1005
	for (x = 0; x < qa->qa_qd_num; x++) {
David Teigland's avatar
David Teigland committed
1006 1007 1008
		struct gfs2_quota_data *qd;
		int sync;

1009
		qd = qa->qa_qd[x];
David Teigland's avatar
David Teigland committed
1010 1011
		sync = need_sync(qd);

1012
		gfs2_glock_dq_uninit(&qa->qa_qd_ghs[x]);
David Teigland's avatar
David Teigland committed
1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023

		if (sync && qd_trylock(qd))
			qda[count++] = qd;
	}

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

1024
out:
David Teigland's avatar
David Teigland committed
1025 1026 1027 1028 1029 1030 1031 1032 1033
	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;

1034
	printk(KERN_INFO "GFS2: fsid=%s: quota %s for %s %u\n",
1035 1036 1037
	       sdp->sd_fsname, type,
	       (test_bit(QDF_USER, &qd->qd_flags)) ? "user" : "group",
	       qd->qd_id);
David Teigland's avatar
David Teigland committed
1038 1039 1040 1041

	return 0;
}

1042
int gfs2_quota_check(struct gfs2_inode *ip, u32 uid, u32 gid)
David Teigland's avatar
David Teigland committed
1043
{
1044
	struct gfs2_sbd *sdp = GFS2_SB(&ip->i_inode);
1045
	struct gfs2_qadata *qa = ip->i_qadata;
David Teigland's avatar
David Teigland committed
1046
	struct gfs2_quota_data *qd;
1047
	s64 value;
David Teigland's avatar
David Teigland committed
1048 1049 1050 1051 1052 1053 1054 1055 1056
	unsigned int x;
	int error = 0;

	if (!test_bit(GIF_QD_LOCKED, &ip->i_flags))
		return 0;

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

1057 1058
	for (x = 0; x < qa->qa_qd_num; x++) {
		qd = qa->qa_qd[x];
David Teigland's avatar
David Teigland committed
1059 1060 1061 1062 1063

		if (!((qd->qd_id == uid && test_bit(QDF_USER, &qd->qd_flags)) ||
		      (qd->qd_id == gid && !test_bit(QDF_USER, &qd->qd_flags))))
			continue;

1064
		value = (s64)be64_to_cpu(qd->qd_qb.qb_value);
1065
		spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1066
		value += qd->qd_change;
1067
		spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1068

1069
		if (be64_to_cpu(qd->qd_qb.qb_limit) && (s64)be64_to_cpu(qd->qd_qb.qb_limit) < value) {
David Teigland's avatar
David Teigland committed
1070
			print_message(qd, "exceeded");
1071 1072 1073 1074
			quota_send_warning(test_bit(QDF_USER, &qd->qd_flags) ?
					   USRQUOTA : GRPQUOTA, qd->qd_id,
					   sdp->sd_vfs->s_dev, QUOTA_NL_BHARDWARN);

David Teigland's avatar
David Teigland committed
1075 1076
			error = -EDQUOT;
			break;
1077
		} else if (be64_to_cpu(qd->qd_qb.qb_warn) &&
1078
			   (s64)be64_to_cpu(qd->qd_qb.qb_warn) < value &&
David Teigland's avatar
David Teigland committed
1079
			   time_after_eq(jiffies, qd->qd_last_warn +
1080 1081
					 gfs2_tune_get(sdp,
						gt_quota_warn_period) * HZ)) {
1082 1083 1084
			quota_send_warning(test_bit(QDF_USER, &qd->qd_flags) ?
					   USRQUOTA : GRPQUOTA, qd->qd_id,
					   sdp->sd_vfs->s_dev, QUOTA_NL_BSOFTWARN);
David Teigland's avatar
David Teigland committed
1085 1086 1087 1088 1089 1090 1091 1092
			error = print_message(qd, "warning");
			qd->qd_last_warn = jiffies;
		}
	}

	return error;
}

1093 1094
void gfs2_quota_change(struct gfs2_inode *ip, s64 change,
		       u32 uid, u32 gid)
David Teigland's avatar
David Teigland committed
1095
{
1096
	struct gfs2_qadata *qa = ip->i_qadata;
David Teigland's avatar
David Teigland committed
1097 1098 1099
	struct gfs2_quota_data *qd;
	unsigned int x;

1100
	if (gfs2_assert_warn(GFS2_SB(&ip->i_inode), change))
David Teigland's avatar
David Teigland committed
1101
		return;
1102
	if (ip->i_diskflags & GFS2_DIF_SYSTEM)
David Teigland's avatar
David Teigland committed
1103 1104
		return;

1105 1106
	for (x = 0; x < qa->qa_qd_num; x++) {
		qd = qa->qa_qd[x];
David Teigland's avatar
David Teigland committed
1107 1108 1109 1110 1111 1112 1113 1114

		if ((qd->qd_id == uid && test_bit(QDF_USER, &qd->qd_flags)) ||
		    (qd->qd_id == gid && !test_bit(QDF_USER, &qd->qd_flags))) {
			do_qc(qd, change);
		}
	}
}

1115
int gfs2_quota_sync(struct super_block *sb, int type, int wait)
David Teigland's avatar
David Teigland committed
1116
{
1117
	struct gfs2_sbd *sdp = sb->s_fs_info;
David Teigland's avatar
David Teigland committed
1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158
	struct gfs2_quota_data **qda;
	unsigned int max_qd = gfs2_tune_get(sdp, gt_quota_simul_sync);
	unsigned int num_qd;
	unsigned int x;
	int error = 0;

	sdp->sd_quota_sync_gen++;

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

	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);

	kfree(qda);

	return error;
}

1159 1160 1161 1162 1163
static int gfs2_quota_sync_timeo(struct super_block *sb, int type)
{
	return gfs2_quota_sync(sb, type, 0);
}

1164
int gfs2_quota_refresh(struct gfs2_sbd *sdp, int user, u32 id)
David Teigland's avatar
David Teigland committed
1165 1166 1167 1168 1169
{
	struct gfs2_quota_data *qd;
	struct gfs2_holder q_gh;
	int error;

1170
	error = qd_get(sdp, user, id, &qd);
David Teigland's avatar
David Teigland committed
1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
	if (error)
		return error;

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

	qd_put(qd);
	return error;
}

1182 1183 1184 1185 1186 1187 1188 1189 1190
static void gfs2_quota_change_in(struct gfs2_quota_change_host *qc, const void *buf)
{
	const struct gfs2_quota_change *str = buf;

	qc->qc_change = be64_to_cpu(str->qc_change);
	qc->qc_flags = be32_to_cpu(str->qc_flags);
	qc->qc_id = be32_to_cpu(str->qc_id);
}

David Teigland's avatar
David Teigland committed
1191 1192
int gfs2_quota_init(struct gfs2_sbd *sdp)
{
1193
	struct gfs2_inode *ip = GFS2_I(sdp->sd_qc_inode);
Steven Whitehouse's avatar
Steven Whitehouse committed
1194 1195
	u64 size = i_size_read(sdp->sd_qc_inode);
	unsigned int blocks = size >> sdp->sd_sb.sb_bsize_shift;
David Teigland's avatar
David Teigland committed
1196 1197
	unsigned int x, slot = 0;
	unsigned int found = 0;
1198 1199
	u64 dblock;
	u32 extlen = 0;
David Teigland's avatar
David Teigland committed
1200 1201
	int error;

Steven Whitehouse's avatar
Steven Whitehouse committed
1202
	if (gfs2_check_internal_file_size(sdp->sd_qc_inode, 1, 64 << 20))
1203
		return -EIO;
Steven Whitehouse's avatar
Steven Whitehouse committed
1204

David Teigland's avatar
David Teigland committed
1205
	sdp->sd_quota_slots = blocks * sdp->sd_qc_per_block;
1206
	sdp->sd_quota_chunks = DIV_ROUND_UP(sdp->sd_quota_slots, 8 * PAGE_SIZE);
David Teigland's avatar
David Teigland committed
1207 1208 1209 1210

	error = -ENOMEM;

	sdp->sd_quota_bitmap = kcalloc(sdp->sd_quota_chunks,
Josef Bacik's avatar
Josef Bacik committed
1211
				       sizeof(unsigned char *), GFP_NOFS);
David Teigland's avatar
David Teigland committed
1212 1213 1214 1215
	if (!sdp->sd_quota_bitmap)
		return error;

	for (x = 0; x < sdp->sd_quota_chunks; x++) {
Josef Bacik's avatar
Josef Bacik committed
1216
		sdp->sd_quota_bitmap[x] = kzalloc(PAGE_SIZE, GFP_NOFS);
David Teigland's avatar
David Teigland committed
1217 1218 1219 1220 1221 1222 1223 1224 1225 1226
		if (!sdp->sd_quota_bitmap[x])
			goto fail;
	}

	for (x = 0; x < blocks; x++) {
		struct buffer_head *bh;
		unsigned int y;

		if (!extlen) {
			int new = 0;
1227
			error = gfs2_extent_map(&ip->i_inode, x, &new, &dblock, &extlen);
David Teigland's avatar
David Teigland committed
1228 1229 1230 1231
			if (error)
				goto fail;
		}
		error = -EIO;
1232 1233 1234
		bh = gfs2_meta_ra(ip->i_gl, dblock, extlen);
		if (!bh)
			goto fail;
David Teigland's avatar
David Teigland committed
1235 1236 1237 1238 1239
		if (gfs2_metatype_check(sdp, bh, GFS2_METATYPE_QC)) {
			brelse(bh);
			goto fail;
		}

1240
		for (y = 0; y < sdp->sd_qc_per_block && slot < sdp->sd_quota_slots;
David Teigland's avatar
David Teigland committed
1241
		     y++, slot++) {
1242
			struct gfs2_quota_change_host qc;
David Teigland's avatar
David Teigland committed
1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262
			struct gfs2_quota_data *qd;

			gfs2_quota_change_in(&qc, bh->b_data +
					  sizeof(struct gfs2_meta_header) +
					  y * sizeof(struct gfs2_quota_change));
			if (!qc.qc_change)
				continue;

			error = qd_alloc(sdp, (qc.qc_flags & GFS2_QCF_USER),
					 qc.qc_id, &qd);
			if (error) {
				brelse(bh);
				goto fail;
			}

			set_bit(QDF_CHANGE, &qd->qd_flags);
			qd->qd_change = qc.qc_change;
			qd->qd_slot = slot;
			qd->qd_slot_count = 1;

1263
			spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1264 1265 1266
			gfs2_icbit_munge(sdp, sdp->sd_quota_bitmap, slot, 1);
			list_add(&qd->qd_list, &sdp->sd_quota_list);
			atomic_inc(&sdp->sd_quota_count);
1267
			spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281

			found++;
		}

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

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

	return 0;

1282
fail:
David Teigland's avatar
David Teigland committed
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292
	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;
	unsigned int x;

1293
	spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1294 1295 1296
	while (!list_empty(head)) {
		qd = list_entry(head->prev, struct gfs2_quota_data, qd_list);

1297 1298 1299 1300 1301
		if (atomic_read(&qd->qd_count) > 1 ||
		    (atomic_read(&qd->qd_count) &&
		     !test_bit(QDF_CHANGE, &qd->qd_flags))) {
			list_move(&qd->qd_list, head);
			spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1302
			schedule();
1303
			spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1304 1305 1306 1307
			continue;
		}

		list_del(&qd->qd_list);
1308 1309 1310 1311 1312
		/* Also remove if this qd exists in the reclaim list */
		if (!list_empty(&qd->qd_reclaim)) {
			list_del_init(&qd->qd_reclaim);
			atomic_dec(&qd_lru_count);
		}
David Teigland's avatar
David Teigland committed
1313
		atomic_dec(&sdp->sd_quota_count);
1314
		spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1315

1316
		if (!atomic_read(&qd->qd_count)) {
David Teigland's avatar
David Teigland committed
1317 1318 1319 1320 1321 1322
			gfs2_assert_warn(sdp, !qd->qd_change);
			gfs2_assert_warn(sdp, !qd->qd_slot_count);
		} else
			gfs2_assert_warn(sdp, qd->qd_slot_count == 1);
		gfs2_assert_warn(sdp, !qd->qd_bh_count);

1323
		gfs2_glock_put(qd->qd_gl);
1324
		kmem_cache_free(gfs2_quotad_cachep, qd);
David Teigland's avatar
David Teigland committed
1325

1326
		spin_lock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1327
	}
1328
	spin_unlock(&qd_lru_lock);
David Teigland's avatar
David Teigland committed
1329 1330 1331 1332 1333 1334 1335 1336 1337 1338

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

	if (sdp->sd_quota_bitmap) {
		for (x = 0; x < sdp->sd_quota_chunks; x++)
			kfree(sdp->sd_quota_bitmap[x]);
		kfree(sdp->sd_quota_bitmap);
	}
}

1339 1340 1341 1342 1343 1344 1345 1346 1347
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,
1348
			       int (*fxn)(struct super_block *sb, int type),
1349 1350 1351 1352
			       unsigned long t, unsigned long *timeo,
			       unsigned int *new_timeo)
{
	if (t >= *timeo) {
1353
		int error = fxn(sdp->sd_vfs, 0);
1354 1355 1356 1357 1358 1359 1360
		quotad_error(sdp, msg, error);
		*timeo = gfs2_tune_get_i(&sdp->sd_tune, new_timeo) * HZ;
	} else {
		*timeo -= t;
	}
}

1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379
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);
	}
}

1380 1381 1382 1383 1384 1385 1386 1387
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);
	}
}


1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401
/**
 * 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);
1402
	int empty;
1403 1404 1405 1406

	while (!kthread_should_stop()) {

		/* Update the master statfs file */
1407 1408 1409 1410 1411 1412 1413 1414 1415
		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);
1416 1417

		/* Update quota file */
1418
		quotad_check_timeo(sdp, "sync", gfs2_quota_sync_timeo, t,
1419 1420
				   &quotad_timeo, &tune->gt_quota_quantum);

1421 1422 1423
		/* Check for & recover partially truncated inodes */
		quotad_check_trunc_list(sdp);

1424 1425
		try_to_freeze();

1426 1427
		t = min(quotad_timeo, statfs_timeo);

1428
		prepare_to_wait(&sdp->sd_quota_wait, &wait, TASK_INTERRUPTIBLE);
1429 1430 1431
		spin_lock(&sdp->sd_trunc_lock);
		empty = list_empty(&sdp->sd_trunc_list);
		spin_unlock(&sdp->sd_trunc_lock);
1432
		if (empty && !sdp->sd_statfs_force_sync)
1433 1434 1435
			t -= schedule_timeout(t);
		else
			t = 0;
1436 1437 1438 1439 1440 1441
		finish_wait(&sdp->sd_quota_wait, &wait);
	}

	return 0;
}

1442 1443 1444 1445 1446 1447 1448
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;
1449 1450 1451

	switch (sdp->sd_args.ar_quota) {
	case GFS2_QUOTA_ON:
1452
		fqs->qs_flags |= (FS_QUOTA_UDQ_ENFD | FS_QUOTA_GDQ_ENFD);
1453 1454
		/*FALLTHRU*/
	case GFS2_QUOTA_ACCOUNT:
1455
		fqs->qs_flags |= (FS_QUOTA_UDQ_ACCT | FS_QUOTA_GDQ_ACCT);
1456 1457 1458 1459 1460
		break;
	case GFS2_QUOTA_OFF:
		break;
	}

1461 1462 1463 1464 1465 1466 1467 1468 1469 1470
	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 */
	fqs->qs_incoredqs = atomic_read(&qd_lru_count);
	return 0;
}

Christoph Hellwig's avatar
Christoph Hellwig committed
1471 1472
static int gfs2_get_dqblk(struct super_block *sb, int type, qid_t id,
			  struct fs_disk_quota *fdq)
1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500
{
	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;

	memset(fdq, 0, sizeof(struct fs_disk_quota));

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

	if (type == USRQUOTA)
		type = QUOTA_USER;
	else if (type == GRPQUOTA)
		type = QUOTA_GROUP;
	else
		return -EINVAL;

	error = qd_get(sdp, type, id, &qd);
	if (error)
		return error;
	error = do_glock(qd, FORCE, &q_gh);
	if (error)
		goto out;

	qlvb = (struct gfs2_quota_lvb *)qd->qd_gl->gl_lvb;
	fdq->d_version = FS_DQUOT_VERSION;
1501
	fdq->d_flags = (type == QUOTA_USER) ? FS_USER_QUOTA : FS_GROUP_QUOTA;
1502
	fdq->d_id = id;
1503 1504 1505
	fdq->d_blk_hardlimit = be64_to_cpu(qlvb->qb_limit) << sdp->sd_fsb2bb_shift;
	fdq->d_blk_softlimit = be64_to_cpu(qlvb->qb_warn) << sdp->sd_fsb2bb_shift;
	fdq->d_bcount = be64_to_cpu(qlvb->qb_value) << sdp->sd_fsb2bb_shift;
1506 1507 1508 1509 1510 1511 1512

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

1513
/* GFS2 only supports a subset of the XFS fields */
1514
#define GFS2_FIELDMASK (FS_DQ_BSOFT|FS_DQ_BHARD|FS_DQ_BCOUNT)
1515

Christoph Hellwig's avatar
Christoph Hellwig committed
1516 1517
static int gfs2_set_dqblk(struct super_block *sb, int type, qid_t id,
			  struct fs_disk_quota *fdq)
1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534
{
	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 */

	switch(type) {
	case USRQUOTA:
		type = QUOTA_USER;
1535
		if (fdq->d_flags != FS_USER_QUOTA)
1536 1537 1538 1539
			return -EINVAL;
		break;
	case GRPQUOTA:
		type = QUOTA_GROUP;
1540
		if (fdq->d_flags != FS_GROUP_QUOTA)
1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555
			return -EINVAL;
		break;
	default:
		return -EINVAL;
	}

	if (fdq->d_fieldmask & ~GFS2_FIELDMASK)
		return -EINVAL;
	if (fdq->d_id != id)
		return -EINVAL;

	error = qd_get(sdp, type, id, &qd);
	if (error)
		return error;

1556 1557 1558 1559
	error = gfs2_rs_alloc(ip);
	if (error)
		goto out_put;

1560 1561 1562
	mutex_lock(&ip->i_inode.i_mutex);
	error = gfs2_glock_nq_init(qd->qd_gl, LM_ST_EXCLUSIVE, 0, &q_gh);
	if (error)
1563
		goto out_unlockput;
1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574
	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 */
	if ((fdq->d_fieldmask & FS_DQ_BSOFT) &&
1575
	    ((fdq->d_blk_softlimit >> sdp->sd_fsb2bb_shift) == be64_to_cpu(qd->qd_qb.qb_warn)))
1576
		fdq->d_fieldmask ^= FS_DQ_BSOFT;
1577

1578
	if ((fdq->d_fieldmask & FS_DQ_BHARD) &&
1579
	    ((fdq->d_blk_hardlimit >> sdp->sd_fsb2bb_shift) == be64_to_cpu(qd->qd_qb.qb_limit)))
1580
		fdq->d_fieldmask ^= FS_DQ_BHARD;
1581 1582 1583 1584 1585

	if ((fdq->d_fieldmask & FS_DQ_BCOUNT) &&
	    ((fdq->d_bcount >> sdp->sd_fsb2bb_shift) == be64_to_cpu(qd->qd_qb.qb_value)))
		fdq->d_fieldmask ^= FS_DQ_BCOUNT;

1586 1587 1588 1589
	if (fdq->d_fieldmask == 0)
		goto out_i;

	offset = qd2offset(qd);
1590
	alloc_required = gfs2_write_alloc_required(ip, offset, sizeof(struct gfs2_quota));
1591 1592
	if (gfs2_is_stuffed(ip))
		alloc_required = 1;
1593 1594 1595
	if (alloc_required) {
		gfs2_write_calc_reserv(ip, sizeof(struct gfs2_quota),
				       &data_blocks, &ind_blocks);
1596 1597
		blocks = 1 + data_blocks + ind_blocks;
		error = gfs2_inplace_reserve(ip, blocks);
1598
		if (error)
1599
			goto out_i;
1600
		blocks += gfs2_rg_blocks(ip);
1601 1602
	}

1603 1604 1605
	/* 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);
1606 1607 1608 1609 1610 1611 1612 1613
	if (error)
		goto out_release;

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

	gfs2_trans_end(sdp);
out_release:
1614
	if (alloc_required)
1615 1616 1617 1618 1619
		gfs2_inplace_release(ip);
out_i:
	gfs2_glock_dq_uninit(&i_gh);
out_q:
	gfs2_glock_dq_uninit(&q_gh);
1620
out_unlockput:
1621
	mutex_unlock(&ip->i_inode.i_mutex);
1622
out_put:
1623 1624 1625 1626
	qd_put(qd);
	return error;
}

1627 1628
const struct quotactl_ops gfs2_quotactl_ops = {
	.quota_sync     = gfs2_quota_sync,
1629
	.get_xstate     = gfs2_quota_get_xstate,
Christoph Hellwig's avatar
Christoph Hellwig committed
1630
	.get_dqblk	= gfs2_get_dqblk,
Christoph Hellwig's avatar
Christoph Hellwig committed
1631
	.set_dqblk	= gfs2_set_dqblk,
1632
};