xref: /openbmc/linux/fs/xfs/xfs_trans_dquot.c (revision 36acd5e2)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (c) 2000-2002 Silicon Graphics, Inc.
4  * All Rights Reserved.
5  */
6 #include "xfs.h"
7 #include "xfs_fs.h"
8 #include "xfs_shared.h"
9 #include "xfs_format.h"
10 #include "xfs_log_format.h"
11 #include "xfs_trans_resv.h"
12 #include "xfs_mount.h"
13 #include "xfs_inode.h"
14 #include "xfs_trans.h"
15 #include "xfs_trans_priv.h"
16 #include "xfs_quota.h"
17 #include "xfs_qm.h"
18 #include "xfs_trace.h"
19 
20 STATIC void	xfs_trans_alloc_dqinfo(xfs_trans_t *);
21 
22 /*
23  * Add the locked dquot to the transaction.
24  * The dquot must be locked, and it cannot be associated with any
25  * transaction.
26  */
27 void
28 xfs_trans_dqjoin(
29 	struct xfs_trans	*tp,
30 	struct xfs_dquot	*dqp)
31 {
32 	ASSERT(XFS_DQ_IS_LOCKED(dqp));
33 	ASSERT(dqp->q_logitem.qli_dquot == dqp);
34 
35 	/*
36 	 * Get a log_item_desc to point at the new item.
37 	 */
38 	xfs_trans_add_item(tp, &dqp->q_logitem.qli_item);
39 }
40 
41 /*
42  * This is called to mark the dquot as needing
43  * to be logged when the transaction is committed.  The dquot must
44  * already be associated with the given transaction.
45  * Note that it marks the entire transaction as dirty. In the ordinary
46  * case, this gets called via xfs_trans_commit, after the transaction
47  * is already dirty. However, there's nothing stop this from getting
48  * called directly, as done by xfs_qm_scall_setqlim. Hence, the TRANS_DIRTY
49  * flag.
50  */
51 void
52 xfs_trans_log_dquot(
53 	struct xfs_trans	*tp,
54 	struct xfs_dquot	*dqp)
55 {
56 	ASSERT(XFS_DQ_IS_LOCKED(dqp));
57 
58 	/* Upgrade the dquot to bigtime format if possible. */
59 	if (dqp->q_id != 0 &&
60 	    xfs_sb_version_hasbigtime(&tp->t_mountp->m_sb) &&
61 	    !(dqp->q_type & XFS_DQTYPE_BIGTIME))
62 		dqp->q_type |= XFS_DQTYPE_BIGTIME;
63 
64 	tp->t_flags |= XFS_TRANS_DIRTY;
65 	set_bit(XFS_LI_DIRTY, &dqp->q_logitem.qli_item.li_flags);
66 }
67 
68 /*
69  * Carry forward whatever is left of the quota blk reservation to
70  * the spanky new transaction
71  */
72 void
73 xfs_trans_dup_dqinfo(
74 	struct xfs_trans	*otp,
75 	struct xfs_trans	*ntp)
76 {
77 	struct xfs_dqtrx	*oq, *nq;
78 	int			i, j;
79 	struct xfs_dqtrx	*oqa, *nqa;
80 	uint64_t		blk_res_used;
81 
82 	if (!otp->t_dqinfo)
83 		return;
84 
85 	xfs_trans_alloc_dqinfo(ntp);
86 
87 	for (j = 0; j < XFS_QM_TRANS_DQTYPES; j++) {
88 		oqa = otp->t_dqinfo->dqs[j];
89 		nqa = ntp->t_dqinfo->dqs[j];
90 		for (i = 0; i < XFS_QM_TRANS_MAXDQS; i++) {
91 			blk_res_used = 0;
92 
93 			if (oqa[i].qt_dquot == NULL)
94 				break;
95 			oq = &oqa[i];
96 			nq = &nqa[i];
97 
98 			if (oq->qt_blk_res && oq->qt_bcount_delta > 0)
99 				blk_res_used = oq->qt_bcount_delta;
100 
101 			nq->qt_dquot = oq->qt_dquot;
102 			nq->qt_bcount_delta = nq->qt_icount_delta = 0;
103 			nq->qt_rtbcount_delta = 0;
104 
105 			/*
106 			 * Transfer whatever is left of the reservations.
107 			 */
108 			nq->qt_blk_res = oq->qt_blk_res - blk_res_used;
109 			oq->qt_blk_res = blk_res_used;
110 
111 			nq->qt_rtblk_res = oq->qt_rtblk_res -
112 				oq->qt_rtblk_res_used;
113 			oq->qt_rtblk_res = oq->qt_rtblk_res_used;
114 
115 			nq->qt_ino_res = oq->qt_ino_res - oq->qt_ino_res_used;
116 			oq->qt_ino_res = oq->qt_ino_res_used;
117 
118 		}
119 	}
120 }
121 
122 /*
123  * Wrap around mod_dquot to account for both user and group quotas.
124  */
125 void
126 xfs_trans_mod_dquot_byino(
127 	xfs_trans_t	*tp,
128 	xfs_inode_t	*ip,
129 	uint		field,
130 	int64_t		delta)
131 {
132 	xfs_mount_t	*mp = tp->t_mountp;
133 
134 	if (!XFS_IS_QUOTA_RUNNING(mp) ||
135 	    !XFS_IS_QUOTA_ON(mp) ||
136 	    xfs_is_quota_inode(&mp->m_sb, ip->i_ino))
137 		return;
138 
139 	if (XFS_IS_UQUOTA_ON(mp) && ip->i_udquot)
140 		(void) xfs_trans_mod_dquot(tp, ip->i_udquot, field, delta);
141 	if (XFS_IS_GQUOTA_ON(mp) && ip->i_gdquot)
142 		(void) xfs_trans_mod_dquot(tp, ip->i_gdquot, field, delta);
143 	if (XFS_IS_PQUOTA_ON(mp) && ip->i_pdquot)
144 		(void) xfs_trans_mod_dquot(tp, ip->i_pdquot, field, delta);
145 }
146 
147 STATIC struct xfs_dqtrx *
148 xfs_trans_get_dqtrx(
149 	struct xfs_trans	*tp,
150 	struct xfs_dquot	*dqp)
151 {
152 	int			i;
153 	struct xfs_dqtrx	*qa;
154 
155 	switch (xfs_dquot_type(dqp)) {
156 	case XFS_DQTYPE_USER:
157 		qa = tp->t_dqinfo->dqs[XFS_QM_TRANS_USR];
158 		break;
159 	case XFS_DQTYPE_GROUP:
160 		qa = tp->t_dqinfo->dqs[XFS_QM_TRANS_GRP];
161 		break;
162 	case XFS_DQTYPE_PROJ:
163 		qa = tp->t_dqinfo->dqs[XFS_QM_TRANS_PRJ];
164 		break;
165 	default:
166 		return NULL;
167 	}
168 
169 	for (i = 0; i < XFS_QM_TRANS_MAXDQS; i++) {
170 		if (qa[i].qt_dquot == NULL ||
171 		    qa[i].qt_dquot == dqp)
172 			return &qa[i];
173 	}
174 
175 	return NULL;
176 }
177 
178 /*
179  * Make the changes in the transaction structure.
180  * The moral equivalent to xfs_trans_mod_sb().
181  * We don't touch any fields in the dquot, so we don't care
182  * if it's locked or not (most of the time it won't be).
183  */
184 void
185 xfs_trans_mod_dquot(
186 	struct xfs_trans	*tp,
187 	struct xfs_dquot	*dqp,
188 	uint			field,
189 	int64_t			delta)
190 {
191 	struct xfs_dqtrx	*qtrx;
192 
193 	ASSERT(tp);
194 	ASSERT(XFS_IS_QUOTA_RUNNING(tp->t_mountp));
195 	qtrx = NULL;
196 
197 	if (!delta)
198 		return;
199 
200 	if (tp->t_dqinfo == NULL)
201 		xfs_trans_alloc_dqinfo(tp);
202 	/*
203 	 * Find either the first free slot or the slot that belongs
204 	 * to this dquot.
205 	 */
206 	qtrx = xfs_trans_get_dqtrx(tp, dqp);
207 	ASSERT(qtrx);
208 	if (qtrx->qt_dquot == NULL)
209 		qtrx->qt_dquot = dqp;
210 
211 	trace_xfs_trans_mod_dquot_before(qtrx);
212 	trace_xfs_trans_mod_dquot(tp, dqp, field, delta);
213 
214 	switch (field) {
215 	/* regular disk blk reservation */
216 	case XFS_TRANS_DQ_RES_BLKS:
217 		qtrx->qt_blk_res += delta;
218 		break;
219 
220 	/* inode reservation */
221 	case XFS_TRANS_DQ_RES_INOS:
222 		qtrx->qt_ino_res += delta;
223 		break;
224 
225 	/* disk blocks used. */
226 	case XFS_TRANS_DQ_BCOUNT:
227 		qtrx->qt_bcount_delta += delta;
228 		break;
229 
230 	case XFS_TRANS_DQ_DELBCOUNT:
231 		qtrx->qt_delbcnt_delta += delta;
232 		break;
233 
234 	/* Inode Count */
235 	case XFS_TRANS_DQ_ICOUNT:
236 		if (qtrx->qt_ino_res && delta > 0) {
237 			qtrx->qt_ino_res_used += delta;
238 			ASSERT(qtrx->qt_ino_res >= qtrx->qt_ino_res_used);
239 		}
240 		qtrx->qt_icount_delta += delta;
241 		break;
242 
243 	/* rtblk reservation */
244 	case XFS_TRANS_DQ_RES_RTBLKS:
245 		qtrx->qt_rtblk_res += delta;
246 		break;
247 
248 	/* rtblk count */
249 	case XFS_TRANS_DQ_RTBCOUNT:
250 		if (qtrx->qt_rtblk_res && delta > 0) {
251 			qtrx->qt_rtblk_res_used += delta;
252 			ASSERT(qtrx->qt_rtblk_res >= qtrx->qt_rtblk_res_used);
253 		}
254 		qtrx->qt_rtbcount_delta += delta;
255 		break;
256 
257 	case XFS_TRANS_DQ_DELRTBCOUNT:
258 		qtrx->qt_delrtb_delta += delta;
259 		break;
260 
261 	default:
262 		ASSERT(0);
263 	}
264 
265 	trace_xfs_trans_mod_dquot_after(qtrx);
266 }
267 
268 
269 /*
270  * Given an array of dqtrx structures, lock all the dquots associated and join
271  * them to the transaction, provided they have been modified.  We know that the
272  * highest number of dquots of one type - usr, grp and prj - involved in a
273  * transaction is 3 so we don't need to make this very generic.
274  */
275 STATIC void
276 xfs_trans_dqlockedjoin(
277 	struct xfs_trans	*tp,
278 	struct xfs_dqtrx	*q)
279 {
280 	ASSERT(q[0].qt_dquot != NULL);
281 	if (q[1].qt_dquot == NULL) {
282 		xfs_dqlock(q[0].qt_dquot);
283 		xfs_trans_dqjoin(tp, q[0].qt_dquot);
284 	} else {
285 		ASSERT(XFS_QM_TRANS_MAXDQS == 2);
286 		xfs_dqlock2(q[0].qt_dquot, q[1].qt_dquot);
287 		xfs_trans_dqjoin(tp, q[0].qt_dquot);
288 		xfs_trans_dqjoin(tp, q[1].qt_dquot);
289 	}
290 }
291 
292 /* Apply dqtrx changes to the quota reservation counters. */
293 static inline void
294 xfs_apply_quota_reservation_deltas(
295 	struct xfs_dquot_res	*res,
296 	uint64_t		reserved,
297 	int64_t			res_used,
298 	int64_t			count_delta)
299 {
300 	if (reserved != 0) {
301 		/*
302 		 * Subtle math here: If reserved > res_used (the normal case),
303 		 * we're simply subtracting the unused transaction quota
304 		 * reservation from the dquot reservation.
305 		 *
306 		 * If, however, res_used > reserved, then we have allocated
307 		 * more quota blocks than were reserved for the transaction.
308 		 * We must add that excess to the dquot reservation since it
309 		 * tracks (usage + resv) and by definition we didn't reserve
310 		 * that excess.
311 		 */
312 		res->reserved -= abs(reserved - res_used);
313 	} else if (count_delta != 0) {
314 		/*
315 		 * These blks were never reserved, either inside a transaction
316 		 * or outside one (in a delayed allocation). Also, this isn't
317 		 * always a negative number since we sometimes deliberately
318 		 * skip quota reservations.
319 		 */
320 		res->reserved += count_delta;
321 	}
322 }
323 
324 /*
325  * Called by xfs_trans_commit() and similar in spirit to
326  * xfs_trans_apply_sb_deltas().
327  * Go thru all the dquots belonging to this transaction and modify the
328  * INCORE dquot to reflect the actual usages.
329  * Unreserve just the reservations done by this transaction.
330  * dquot is still left locked at exit.
331  */
332 void
333 xfs_trans_apply_dquot_deltas(
334 	struct xfs_trans	*tp)
335 {
336 	int			i, j;
337 	struct xfs_dquot	*dqp;
338 	struct xfs_dqtrx	*qtrx, *qa;
339 	int64_t			totalbdelta;
340 	int64_t			totalrtbdelta;
341 
342 	if (!tp->t_dqinfo)
343 		return;
344 
345 	ASSERT(tp->t_dqinfo);
346 	for (j = 0; j < XFS_QM_TRANS_DQTYPES; j++) {
347 		qa = tp->t_dqinfo->dqs[j];
348 		if (qa[0].qt_dquot == NULL)
349 			continue;
350 
351 		/*
352 		 * Lock all of the dquots and join them to the transaction.
353 		 */
354 		xfs_trans_dqlockedjoin(tp, qa);
355 
356 		for (i = 0; i < XFS_QM_TRANS_MAXDQS; i++) {
357 			uint64_t	blk_res_used;
358 
359 			qtrx = &qa[i];
360 			/*
361 			 * The array of dquots is filled
362 			 * sequentially, not sparsely.
363 			 */
364 			if ((dqp = qtrx->qt_dquot) == NULL)
365 				break;
366 
367 			ASSERT(XFS_DQ_IS_LOCKED(dqp));
368 
369 			/*
370 			 * adjust the actual number of blocks used
371 			 */
372 
373 			/*
374 			 * The issue here is - sometimes we don't make a blkquota
375 			 * reservation intentionally to be fair to users
376 			 * (when the amount is small). On the other hand,
377 			 * delayed allocs do make reservations, but that's
378 			 * outside of a transaction, so we have no
379 			 * idea how much was really reserved.
380 			 * So, here we've accumulated delayed allocation blks and
381 			 * non-delay blks. The assumption is that the
382 			 * delayed ones are always reserved (outside of a
383 			 * transaction), and the others may or may not have
384 			 * quota reservations.
385 			 */
386 			totalbdelta = qtrx->qt_bcount_delta +
387 				qtrx->qt_delbcnt_delta;
388 			totalrtbdelta = qtrx->qt_rtbcount_delta +
389 				qtrx->qt_delrtb_delta;
390 
391 			if (totalbdelta != 0 || totalrtbdelta != 0 ||
392 			    qtrx->qt_icount_delta != 0) {
393 				trace_xfs_trans_apply_dquot_deltas_before(dqp);
394 				trace_xfs_trans_apply_dquot_deltas(qtrx);
395 			}
396 
397 #ifdef DEBUG
398 			if (totalbdelta < 0)
399 				ASSERT(dqp->q_blk.count >= -totalbdelta);
400 
401 			if (totalrtbdelta < 0)
402 				ASSERT(dqp->q_rtb.count >= -totalrtbdelta);
403 
404 			if (qtrx->qt_icount_delta < 0)
405 				ASSERT(dqp->q_ino.count >= -qtrx->qt_icount_delta);
406 #endif
407 			if (totalbdelta)
408 				dqp->q_blk.count += totalbdelta;
409 
410 			if (qtrx->qt_icount_delta)
411 				dqp->q_ino.count += qtrx->qt_icount_delta;
412 
413 			if (totalrtbdelta)
414 				dqp->q_rtb.count += totalrtbdelta;
415 
416 			if (totalbdelta != 0 || totalrtbdelta != 0 ||
417 			    qtrx->qt_icount_delta != 0)
418 				trace_xfs_trans_apply_dquot_deltas_after(dqp);
419 
420 			/*
421 			 * Get any default limits in use.
422 			 * Start/reset the timer(s) if needed.
423 			 */
424 			if (dqp->q_id) {
425 				xfs_qm_adjust_dqlimits(dqp);
426 				xfs_qm_adjust_dqtimers(dqp);
427 			}
428 
429 			dqp->q_flags |= XFS_DQFLAG_DIRTY;
430 			/*
431 			 * add this to the list of items to get logged
432 			 */
433 			xfs_trans_log_dquot(tp, dqp);
434 			/*
435 			 * Take off what's left of the original reservation.
436 			 * In case of delayed allocations, there's no
437 			 * reservation that a transaction structure knows of.
438 			 */
439 			blk_res_used = max_t(int64_t, 0, qtrx->qt_bcount_delta);
440 			xfs_apply_quota_reservation_deltas(&dqp->q_blk,
441 					qtrx->qt_blk_res, blk_res_used,
442 					qtrx->qt_bcount_delta);
443 
444 			/*
445 			 * Adjust the RT reservation.
446 			 */
447 			xfs_apply_quota_reservation_deltas(&dqp->q_rtb,
448 					qtrx->qt_rtblk_res,
449 					qtrx->qt_rtblk_res_used,
450 					qtrx->qt_rtbcount_delta);
451 
452 			/*
453 			 * Adjust the inode reservation.
454 			 */
455 			ASSERT(qtrx->qt_ino_res >= qtrx->qt_ino_res_used);
456 			xfs_apply_quota_reservation_deltas(&dqp->q_ino,
457 					qtrx->qt_ino_res,
458 					qtrx->qt_ino_res_used,
459 					qtrx->qt_icount_delta);
460 
461 			ASSERT(dqp->q_blk.reserved >= dqp->q_blk.count);
462 			ASSERT(dqp->q_ino.reserved >= dqp->q_ino.count);
463 			ASSERT(dqp->q_rtb.reserved >= dqp->q_rtb.count);
464 		}
465 	}
466 }
467 
468 /*
469  * Release the reservations, and adjust the dquots accordingly.
470  * This is called only when the transaction is being aborted. If by
471  * any chance we have done dquot modifications incore (ie. deltas) already,
472  * we simply throw those away, since that's the expected behavior
473  * when a transaction is curtailed without a commit.
474  */
475 void
476 xfs_trans_unreserve_and_mod_dquots(
477 	struct xfs_trans	*tp)
478 {
479 	int			i, j;
480 	struct xfs_dquot	*dqp;
481 	struct xfs_dqtrx	*qtrx, *qa;
482 	bool			locked;
483 
484 	if (!tp->t_dqinfo)
485 		return;
486 
487 	for (j = 0; j < XFS_QM_TRANS_DQTYPES; j++) {
488 		qa = tp->t_dqinfo->dqs[j];
489 
490 		for (i = 0; i < XFS_QM_TRANS_MAXDQS; i++) {
491 			qtrx = &qa[i];
492 			/*
493 			 * We assume that the array of dquots is filled
494 			 * sequentially, not sparsely.
495 			 */
496 			if ((dqp = qtrx->qt_dquot) == NULL)
497 				break;
498 			/*
499 			 * Unreserve the original reservation. We don't care
500 			 * about the number of blocks used field, or deltas.
501 			 * Also we don't bother to zero the fields.
502 			 */
503 			locked = false;
504 			if (qtrx->qt_blk_res) {
505 				xfs_dqlock(dqp);
506 				locked = true;
507 				dqp->q_blk.reserved -=
508 					(xfs_qcnt_t)qtrx->qt_blk_res;
509 			}
510 			if (qtrx->qt_ino_res) {
511 				if (!locked) {
512 					xfs_dqlock(dqp);
513 					locked = true;
514 				}
515 				dqp->q_ino.reserved -=
516 					(xfs_qcnt_t)qtrx->qt_ino_res;
517 			}
518 
519 			if (qtrx->qt_rtblk_res) {
520 				if (!locked) {
521 					xfs_dqlock(dqp);
522 					locked = true;
523 				}
524 				dqp->q_rtb.reserved -=
525 					(xfs_qcnt_t)qtrx->qt_rtblk_res;
526 			}
527 			if (locked)
528 				xfs_dqunlock(dqp);
529 
530 		}
531 	}
532 }
533 
534 STATIC void
535 xfs_quota_warn(
536 	struct xfs_mount	*mp,
537 	struct xfs_dquot	*dqp,
538 	int			type)
539 {
540 	enum quota_type		qtype;
541 
542 	switch (xfs_dquot_type(dqp)) {
543 	case XFS_DQTYPE_PROJ:
544 		qtype = PRJQUOTA;
545 		break;
546 	case XFS_DQTYPE_USER:
547 		qtype = USRQUOTA;
548 		break;
549 	case XFS_DQTYPE_GROUP:
550 		qtype = GRPQUOTA;
551 		break;
552 	default:
553 		return;
554 	}
555 
556 	quota_send_warning(make_kqid(&init_user_ns, qtype, dqp->q_id),
557 			   mp->m_super->s_dev, type);
558 }
559 
560 /*
561  * Decide if we can make an additional reservation against a quota resource.
562  * Returns an inode QUOTA_NL_ warning code and whether or not it's fatal.
563  *
564  * Note that we assume that the numeric difference between the inode and block
565  * warning codes will always be 3 since it's userspace ABI now, and will never
566  * decrease the quota reservation, so the *BELOW messages are irrelevant.
567  */
568 static inline int
569 xfs_dqresv_check(
570 	struct xfs_dquot_res	*res,
571 	struct xfs_quota_limits	*qlim,
572 	int64_t			delta,
573 	bool			*fatal)
574 {
575 	xfs_qcnt_t		hardlimit = res->hardlimit;
576 	xfs_qcnt_t		softlimit = res->softlimit;
577 	xfs_qcnt_t		total_count = res->reserved + delta;
578 
579 	BUILD_BUG_ON(QUOTA_NL_BHARDWARN     != QUOTA_NL_IHARDWARN + 3);
580 	BUILD_BUG_ON(QUOTA_NL_BSOFTLONGWARN != QUOTA_NL_ISOFTLONGWARN + 3);
581 	BUILD_BUG_ON(QUOTA_NL_BSOFTWARN     != QUOTA_NL_ISOFTWARN + 3);
582 
583 	*fatal = false;
584 	if (delta <= 0)
585 		return QUOTA_NL_NOWARN;
586 
587 	if (!hardlimit)
588 		hardlimit = qlim->hard;
589 	if (!softlimit)
590 		softlimit = qlim->soft;
591 
592 	if (hardlimit && total_count > hardlimit) {
593 		*fatal = true;
594 		return QUOTA_NL_IHARDWARN;
595 	}
596 
597 	if (softlimit && total_count > softlimit) {
598 		time64_t	now = ktime_get_real_seconds();
599 
600 		if ((res->timer != 0 && now > res->timer) ||
601 		    (res->warnings != 0 && res->warnings >= qlim->warn)) {
602 			*fatal = true;
603 			return QUOTA_NL_ISOFTLONGWARN;
604 		}
605 
606 		res->warnings++;
607 		return QUOTA_NL_ISOFTWARN;
608 	}
609 
610 	return QUOTA_NL_NOWARN;
611 }
612 
613 /*
614  * This reserves disk blocks and inodes against a dquot.
615  * Flags indicate if the dquot is to be locked here and also
616  * if the blk reservation is for RT or regular blocks.
617  * Sending in XFS_QMOPT_FORCE_RES flag skips the quota check.
618  */
619 STATIC int
620 xfs_trans_dqresv(
621 	struct xfs_trans	*tp,
622 	struct xfs_mount	*mp,
623 	struct xfs_dquot	*dqp,
624 	int64_t			nblks,
625 	long			ninos,
626 	uint			flags)
627 {
628 	struct xfs_quotainfo	*q = mp->m_quotainfo;
629 	struct xfs_def_quota	*defq;
630 	struct xfs_dquot_res	*blkres;
631 	struct xfs_quota_limits	*qlim;
632 
633 	xfs_dqlock(dqp);
634 
635 	defq = xfs_get_defquota(q, xfs_dquot_type(dqp));
636 
637 	if (flags & XFS_TRANS_DQ_RES_BLKS) {
638 		blkres = &dqp->q_blk;
639 		qlim = &defq->blk;
640 	} else {
641 		blkres = &dqp->q_rtb;
642 		qlim = &defq->rtb;
643 	}
644 
645 	if ((flags & XFS_QMOPT_FORCE_RES) == 0 && dqp->q_id &&
646 	    xfs_dquot_is_enforced(dqp)) {
647 		int		quota_nl;
648 		bool		fatal;
649 
650 		/*
651 		 * dquot is locked already. See if we'd go over the hardlimit
652 		 * or exceed the timelimit if we'd reserve resources.
653 		 */
654 		quota_nl = xfs_dqresv_check(blkres, qlim, nblks, &fatal);
655 		if (quota_nl != QUOTA_NL_NOWARN) {
656 			/*
657 			 * Quota block warning codes are 3 more than the inode
658 			 * codes, which we check above.
659 			 */
660 			xfs_quota_warn(mp, dqp, quota_nl + 3);
661 			if (fatal)
662 				goto error_return;
663 		}
664 
665 		quota_nl = xfs_dqresv_check(&dqp->q_ino, &defq->ino, ninos,
666 				&fatal);
667 		if (quota_nl != QUOTA_NL_NOWARN) {
668 			xfs_quota_warn(mp, dqp, quota_nl);
669 			if (fatal)
670 				goto error_return;
671 		}
672 	}
673 
674 	/*
675 	 * Change the reservation, but not the actual usage.
676 	 * Note that q_blk.reserved = q_blk.count + resv
677 	 */
678 	blkres->reserved += (xfs_qcnt_t)nblks;
679 	dqp->q_ino.reserved += (xfs_qcnt_t)ninos;
680 
681 	/*
682 	 * note the reservation amt in the trans struct too,
683 	 * so that the transaction knows how much was reserved by
684 	 * it against this particular dquot.
685 	 * We don't do this when we are reserving for a delayed allocation,
686 	 * because we don't have the luxury of a transaction envelope then.
687 	 */
688 	if (tp) {
689 		ASSERT(flags & XFS_QMOPT_RESBLK_MASK);
690 		xfs_trans_mod_dquot(tp, dqp, flags & XFS_QMOPT_RESBLK_MASK,
691 				    nblks);
692 		xfs_trans_mod_dquot(tp, dqp, XFS_TRANS_DQ_RES_INOS, ninos);
693 	}
694 	ASSERT(dqp->q_blk.reserved >= dqp->q_blk.count);
695 	ASSERT(dqp->q_rtb.reserved >= dqp->q_rtb.count);
696 	ASSERT(dqp->q_ino.reserved >= dqp->q_ino.count);
697 
698 	xfs_dqunlock(dqp);
699 	return 0;
700 
701 error_return:
702 	xfs_dqunlock(dqp);
703 	if (xfs_dquot_type(dqp) == XFS_DQTYPE_PROJ)
704 		return -ENOSPC;
705 	return -EDQUOT;
706 }
707 
708 
709 /*
710  * Given dquot(s), make disk block and/or inode reservations against them.
711  * The fact that this does the reservation against user, group and
712  * project quotas is important, because this follows a all-or-nothing
713  * approach.
714  *
715  * flags = XFS_QMOPT_FORCE_RES evades limit enforcement. Used by chown.
716  *	   XFS_QMOPT_ENOSPC returns ENOSPC not EDQUOT.  Used by pquota.
717  *	   XFS_TRANS_DQ_RES_BLKS reserves regular disk blocks
718  *	   XFS_TRANS_DQ_RES_RTBLKS reserves realtime disk blocks
719  * dquots are unlocked on return, if they were not locked by caller.
720  */
721 int
722 xfs_trans_reserve_quota_bydquots(
723 	struct xfs_trans	*tp,
724 	struct xfs_mount	*mp,
725 	struct xfs_dquot	*udqp,
726 	struct xfs_dquot	*gdqp,
727 	struct xfs_dquot	*pdqp,
728 	int64_t			nblks,
729 	long			ninos,
730 	uint			flags)
731 {
732 	int		error;
733 
734 	if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
735 		return 0;
736 
737 	ASSERT(flags & XFS_QMOPT_RESBLK_MASK);
738 
739 	if (udqp) {
740 		error = xfs_trans_dqresv(tp, mp, udqp, nblks, ninos, flags);
741 		if (error)
742 			return error;
743 	}
744 
745 	if (gdqp) {
746 		error = xfs_trans_dqresv(tp, mp, gdqp, nblks, ninos, flags);
747 		if (error)
748 			goto unwind_usr;
749 	}
750 
751 	if (pdqp) {
752 		error = xfs_trans_dqresv(tp, mp, pdqp, nblks, ninos, flags);
753 		if (error)
754 			goto unwind_grp;
755 	}
756 
757 	/*
758 	 * Didn't change anything critical, so, no need to log
759 	 */
760 	return 0;
761 
762 unwind_grp:
763 	flags |= XFS_QMOPT_FORCE_RES;
764 	if (gdqp)
765 		xfs_trans_dqresv(tp, mp, gdqp, -nblks, -ninos, flags);
766 unwind_usr:
767 	flags |= XFS_QMOPT_FORCE_RES;
768 	if (udqp)
769 		xfs_trans_dqresv(tp, mp, udqp, -nblks, -ninos, flags);
770 	return error;
771 }
772 
773 
774 /*
775  * Lock the dquot and change the reservation if we can.
776  * This doesn't change the actual usage, just the reservation.
777  * The inode sent in is locked.
778  */
779 int
780 xfs_trans_reserve_quota_nblks(
781 	struct xfs_trans	*tp,
782 	struct xfs_inode	*ip,
783 	int64_t			nblks,
784 	long			ninos,
785 	uint			flags)
786 {
787 	struct xfs_mount	*mp = ip->i_mount;
788 
789 	if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
790 		return 0;
791 
792 	ASSERT(!xfs_is_quota_inode(&mp->m_sb, ip->i_ino));
793 
794 	ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
795 	ASSERT((flags & ~(XFS_QMOPT_FORCE_RES)) == XFS_TRANS_DQ_RES_RTBLKS ||
796 	       (flags & ~(XFS_QMOPT_FORCE_RES)) == XFS_TRANS_DQ_RES_BLKS);
797 
798 	/*
799 	 * Reserve nblks against these dquots, with trans as the mediator.
800 	 */
801 	return xfs_trans_reserve_quota_bydquots(tp, mp,
802 						ip->i_udquot, ip->i_gdquot,
803 						ip->i_pdquot,
804 						nblks, ninos, flags);
805 }
806 
807 /*
808  * This routine is called to allocate a quotaoff log item.
809  */
810 struct xfs_qoff_logitem *
811 xfs_trans_get_qoff_item(
812 	struct xfs_trans	*tp,
813 	struct xfs_qoff_logitem	*startqoff,
814 	uint			flags)
815 {
816 	struct xfs_qoff_logitem	*q;
817 
818 	ASSERT(tp != NULL);
819 
820 	q = xfs_qm_qoff_logitem_init(tp->t_mountp, startqoff, flags);
821 	ASSERT(q != NULL);
822 
823 	/*
824 	 * Get a log_item_desc to point at the new item.
825 	 */
826 	xfs_trans_add_item(tp, &q->qql_item);
827 	return q;
828 }
829 
830 
831 /*
832  * This is called to mark the quotaoff logitem as needing
833  * to be logged when the transaction is committed.  The logitem must
834  * already be associated with the given transaction.
835  */
836 void
837 xfs_trans_log_quotaoff_item(
838 	struct xfs_trans	*tp,
839 	struct xfs_qoff_logitem	*qlp)
840 {
841 	tp->t_flags |= XFS_TRANS_DIRTY;
842 	set_bit(XFS_LI_DIRTY, &qlp->qql_item.li_flags);
843 }
844 
845 STATIC void
846 xfs_trans_alloc_dqinfo(
847 	xfs_trans_t	*tp)
848 {
849 	tp->t_dqinfo = kmem_cache_zalloc(xfs_qm_dqtrxzone,
850 					 GFP_KERNEL | __GFP_NOFAIL);
851 }
852 
853 void
854 xfs_trans_free_dqinfo(
855 	xfs_trans_t	*tp)
856 {
857 	if (!tp->t_dqinfo)
858 		return;
859 	kmem_cache_free(xfs_qm_dqtrxzone, tp->t_dqinfo);
860 	tp->t_dqinfo = NULL;
861 }
862