fs/xfs/xfs_qm.c

   1 /*
   2  * Copyright (c) 2000-2005 Silicon Graphics, Inc.
   3  * All Rights Reserved.
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License as
   7  * published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it would be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write the Free Software Foundation,
  16  * Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17  */
  18 #include "xfs.h"
  19 #include "xfs_fs.h"
  20 #include "xfs_shared.h"
  21 #include "xfs_format.h"
  22 #include "xfs_log_format.h"
  23 #include "xfs_trans_resv.h"
  24 #include "xfs_bit.h"
  25 #include "xfs_sb.h"
  26 #include "xfs_mount.h"
  27 #include "xfs_inode.h"
  28 #include "xfs_ialloc.h"
  29 #include "xfs_itable.h"
  30 #include "xfs_quota.h"
  31 #include "xfs_error.h"
  32 #include "xfs_bmap.h"
  33 #include "xfs_bmap_btree.h"
  34 #include "xfs_bmap_util.h"
  35 #include "xfs_trans.h"
  36 #include "xfs_trans_space.h"
  37 #include "xfs_qm.h"
  38 #include "xfs_trace.h"
  39 #include "xfs_icache.h"
  40 #include "xfs_cksum.h"
  41
  42 /*
  43  * The global quota manager. There is only one of these for the entire
  44  * system, _not_ one per file system. XQM keeps track of the overall
  45  * quota functionality, including maintaining the freelist and hash
  46  * tables of dquots.
  47  */
  48 STATIC int      xfs_qm_init_quotainos(xfs_mount_t *);
  49 STATIC int      xfs_qm_init_quotainfo(xfs_mount_t *);
  50
  51 STATIC void     xfs_qm_destroy_quotainos(xfs_quotainfo_t *qi);
  52 STATIC void     xfs_qm_dqfree_one(struct xfs_dquot *dqp);
  53 /*
  54  * We use the batch lookup interface to iterate over the dquots as it
  55  * currently is the only interface into the radix tree code that allows
  56  * fuzzy lookups instead of exact matches.  Holding the lock over multiple
  57  * operations is fine as all callers are used either during mount/umount
  58  * or quotaoff.
  59  */
  60 #define XFS_DQ_LOOKUP_BATCH     32
  61
  62 STATIC int
  63 xfs_qm_dquot_walk(
  64         struct xfs_mount        *mp,
  65         int                     type,
  66         int                     (*execute)(struct xfs_dquot *dqp, void *data),
  67         void                    *data)
  68 {
  69         struct xfs_quotainfo    *qi = mp->m_quotainfo;
  70         struct radix_tree_root  *tree = xfs_dquot_tree(qi, type);
  71         uint32_t                next_index;
  72         int                     last_error = 0;
  73         int                     skipped;
  74         int                     nr_found;
  75
  76 restart:
  77         skipped = 0;
  78         next_index = 0;
  79         nr_found = 0;
  80
  81         while (1) {
  82                 struct xfs_dquot *batch[XFS_DQ_LOOKUP_BATCH];
  83                 int             error = 0;
  84                 int             i;
  85
  86                 mutex_lock(&qi->qi_tree_lock);
  87                 nr_found = radix_tree_gang_lookup(tree, (void **)batch,
  88                                         next_index, XFS_DQ_LOOKUP_BATCH);
  89                 if (!nr_found) {
  90                         mutex_unlock(&qi->qi_tree_lock);
  91                         break;
  92                 }
  93
  94                 for (i = 0; i < nr_found; i++) {
  95                         struct xfs_dquot *dqp = batch[i];
  96
  97                         next_index = be32_to_cpu(dqp->q_core.d_id) + 1;
  98
  99                         error = execute(batch[i], data);
 100                         if (error == -EAGAIN) {
 101                                 skipped++;
 102                                 continue;
 103                         }
 104                         if (error && last_error != -EFSCORRUPTED)
 105                                 last_error = error;
 106                 }
 107
 108                 mutex_unlock(&qi->qi_tree_lock);
 109
 110                 /* bail out if the filesystem is corrupted.  */
 111                 if (last_error == -EFSCORRUPTED) {
 112                         skipped = 0;
 113                         break;
 114                 }
 115                 /* we're done if id overflows back to zero */
 116                 if (!next_index)
 117                         break;
 118         }
 119
 120         if (skipped) {
 121                 delay(1);
 122                 goto restart;
 123         }
 124
 125         return last_error;
 126 }
 127
 128
 129 /*
 130  * Purge a dquot from all tracking data structures and free it.
 131  */
 132 STATIC int
 133 xfs_qm_dqpurge(
 134         struct xfs_dquot        *dqp,
 135         void                    *data)
 136 {
 137         struct xfs_mount        *mp = dqp->q_mount;
 138         struct xfs_quotainfo    *qi = mp->m_quotainfo;
 139
 140         xfs_dqlock(dqp);
 141         if ((dqp->dq_flags & XFS_DQ_FREEING) || dqp->q_nrefs != 0) {
 142                 xfs_dqunlock(dqp);
 143                 return -EAGAIN;
 144         }
 145
 146         dqp->dq_flags |= XFS_DQ_FREEING;
 147
 148         xfs_dqflock(dqp);
 149
 150         /*
 151          * If we are turning this type of quotas off, we don't care
 152          * about the dirty metadata sitting in this dquot. OTOH, if
 153          * we're unmounting, we do care, so we flush it and wait.
 154          */
 155         if (XFS_DQ_IS_DIRTY(dqp)) {
 156                 struct xfs_buf  *bp = NULL;
 157                 int             error;
 158
 159                 /*
 160                  * We don't care about getting disk errors here. We need
 161                  * to purge this dquot anyway, so we go ahead regardless.
 162                  */
 163                 error = xfs_qm_dqflush(dqp, &bp);
 164                 if (error) {
 165                         xfs_warn(mp, "%s: dquot "PTR_FMT" flush failed",
 166                                 __func__, dqp);
 167                 } else {
 168                         error = xfs_bwrite(bp);
 169                         xfs_buf_relse(bp);
 170                 }
 171                 xfs_dqflock(dqp);
 172         }
 173
 174         ASSERT(atomic_read(&dqp->q_pincount) == 0);
 175         ASSERT(XFS_FORCED_SHUTDOWN(mp) ||
 176                !(dqp->q_logitem.qli_item.li_flags & XFS_LI_IN_AIL));
 177
 178         xfs_dqfunlock(dqp);
 179         xfs_dqunlock(dqp);
 180
 181         radix_tree_delete(xfs_dquot_tree(qi, dqp->q_core.d_flags),
 182                           be32_to_cpu(dqp->q_core.d_id));
 183         qi->qi_dquots--;
 184
 185         /*
 186          * We move dquots to the freelist as soon as their reference count
 187          * hits zero, so it really should be on the freelist here.
 188          */
 189         ASSERT(!list_empty(&dqp->q_lru));
 190         list_lru_del(&qi->qi_lru, &dqp->q_lru);
 191         XFS_STATS_DEC(mp, xs_qm_dquot_unused);
 192
 193         xfs_qm_dqdestroy(dqp);
 194         return 0;
 195 }
 196
 197 /*
 198  * Purge the dquot cache.
 199  */
 200 void
 201 xfs_qm_dqpurge_all(
 202         struct xfs_mount        *mp,
 203         uint                    flags)
 204 {
 205         if (flags & XFS_QMOPT_UQUOTA)
 206                 xfs_qm_dquot_walk(mp, XFS_DQ_USER, xfs_qm_dqpurge, NULL);
 207         if (flags & XFS_QMOPT_GQUOTA)
 208                 xfs_qm_dquot_walk(mp, XFS_DQ_GROUP, xfs_qm_dqpurge, NULL);
 209         if (flags & XFS_QMOPT_PQUOTA)
 210                 xfs_qm_dquot_walk(mp, XFS_DQ_PROJ, xfs_qm_dqpurge, NULL);
 211 }
 212
 213 /*
 214  * Just destroy the quotainfo structure.
 215  */
 216 void
 217 xfs_qm_unmount(
 218         struct xfs_mount        *mp)
 219 {
 220         if (mp->m_quotainfo) {
 221                 xfs_qm_dqpurge_all(mp, XFS_QMOPT_QUOTALL);
 222                 xfs_qm_destroy_quotainfo(mp);
 223         }
 224 }
 225
 226 /*
 227  * Called from the vfsops layer.
 228  */
 229 void
 230 xfs_qm_unmount_quotas(
 231         xfs_mount_t     *mp)
 232 {
 233         /*
 234          * Release the dquots that root inode, et al might be holding,
 235          * before we flush quotas and blow away the quotainfo structure.
 236          */
 237         ASSERT(mp->m_rootip);
 238         xfs_qm_dqdetach(mp->m_rootip);
 239         if (mp->m_rbmip)
 240                 xfs_qm_dqdetach(mp->m_rbmip);
 241         if (mp->m_rsumip)
 242                 xfs_qm_dqdetach(mp->m_rsumip);
 243
 244         /*
 245          * Release the quota inodes.
 246          */
 247         if (mp->m_quotainfo) {
 248                 if (mp->m_quotainfo->qi_uquotaip) {
 249                         IRELE(mp->m_quotainfo->qi_uquotaip);
 250                         mp->m_quotainfo->qi_uquotaip = NULL;
 251                 }
 252                 if (mp->m_quotainfo->qi_gquotaip) {
 253                         IRELE(mp->m_quotainfo->qi_gquotaip);
 254                         mp->m_quotainfo->qi_gquotaip = NULL;
 255                 }
 256                 if (mp->m_quotainfo->qi_pquotaip) {
 257                         IRELE(mp->m_quotainfo->qi_pquotaip);
 258                         mp->m_quotainfo->qi_pquotaip = NULL;
 259                 }
 260         }
 261 }
 262
 263 STATIC int
 264 xfs_qm_dqattach_one(
 265         xfs_inode_t     *ip,
 266         xfs_dqid_t      id,
 267         uint            type,
 268         uint            doalloc,
 269         xfs_dquot_t     **IO_idqpp)
 270 {
 271         xfs_dquot_t     *dqp;
 272         int             error;
 273
 274         ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
 275         error = 0;
 276
 277         /*
 278          * See if we already have it in the inode itself. IO_idqpp is &i_udquot
 279          * or &i_gdquot. This made the code look weird, but made the logic a lot
 280          * simpler.
 281          */
 282         dqp = *IO_idqpp;
 283         if (dqp) {
 284                 trace_xfs_dqattach_found(dqp);
 285                 return 0;
 286         }
 287
 288         /*
 289          * Find the dquot from somewhere. This bumps the reference count of
 290          * dquot and returns it locked.  This can return ENOENT if dquot didn't
 291          * exist on disk and we didn't ask it to allocate; ESRCH if quotas got
 292          * turned off suddenly.
 293          */
 294         error = xfs_qm_dqget(ip->i_mount, ip, id, type, doalloc, &dqp);
 295         if (error)
 296                 return error;
 297
 298         trace_xfs_dqattach_get(dqp);
 299
 300         /*
 301          * dqget may have dropped and re-acquired the ilock, but it guarantees
 302          * that the dquot returned is the one that should go in the inode.
 303          */
 304         *IO_idqpp = dqp;
 305         xfs_dqunlock(dqp);
 306         return 0;
 307 }
 308
 309 static bool
 310 xfs_qm_need_dqattach(
 311         struct xfs_inode        *ip)
 312 {
 313         struct xfs_mount        *mp = ip->i_mount;
 314
 315         if (!XFS_IS_QUOTA_RUNNING(mp))
 316                 return false;
 317         if (!XFS_IS_QUOTA_ON(mp))
 318                 return false;
 319         if (!XFS_NOT_DQATTACHED(mp, ip))
 320                 return false;
 321         if (xfs_is_quota_inode(&mp->m_sb, ip->i_ino))
 322                 return false;
 323         return true;
 324 }
 325
 326 /*
 327  * Given a locked inode, attach dquot(s) to it, taking U/G/P-QUOTAON
 328  * into account.
 329  * If XFS_QMOPT_DQALLOC, the dquot(s) will be allocated if needed.
 330  * Inode may get unlocked and relocked in here, and the caller must deal with
 331  * the consequences.
 332  */
 333 int
 334 xfs_qm_dqattach_locked(
 335         xfs_inode_t     *ip,
 336         uint            flags)
 337 {
 338         xfs_mount_t     *mp = ip->i_mount;
 339         int             error = 0;
 340
 341         if (!xfs_qm_need_dqattach(ip))
 342                 return 0;
 343
 344         ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
 345
 346         if (XFS_IS_UQUOTA_ON(mp) && !ip->i_udquot) {
 347                 error = xfs_qm_dqattach_one(ip, ip->i_d.di_uid, XFS_DQ_USER,
 348                                                 flags & XFS_QMOPT_DQALLOC,
 349                                                 &ip->i_udquot);
 350                 if (error)
 351                         goto done;
 352                 ASSERT(ip->i_udquot);
 353         }
 354
 355         if (XFS_IS_GQUOTA_ON(mp) && !ip->i_gdquot) {
 356                 error = xfs_qm_dqattach_one(ip, ip->i_d.di_gid, XFS_DQ_GROUP,
 357                                                 flags & XFS_QMOPT_DQALLOC,
 358                                                 &ip->i_gdquot);
 359                 if (error)
 360                         goto done;
 361                 ASSERT(ip->i_gdquot);
 362         }
 363
 364         if (XFS_IS_PQUOTA_ON(mp) && !ip->i_pdquot) {
 365                 error = xfs_qm_dqattach_one(ip, xfs_get_projid(ip), XFS_DQ_PROJ,
 366                                                 flags & XFS_QMOPT_DQALLOC,
 367                                                 &ip->i_pdquot);
 368                 if (error)
 369                         goto done;
 370                 ASSERT(ip->i_pdquot);
 371         }
 372
 373 done:
 374         /*
 375          * Don't worry about the dquots that we may have attached before any
 376          * error - they'll get detached later if it has not already been done.
 377          */
 378         ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
 379         return error;
 380 }
 381
 382 int
 383 xfs_qm_dqattach(
 384         struct xfs_inode        *ip,
 385         uint                    flags)
 386 {
 387         int                     error;
 388
 389         if (!xfs_qm_need_dqattach(ip))
 390                 return 0;
 391
 392         xfs_ilock(ip, XFS_ILOCK_EXCL);
 393         error = xfs_qm_dqattach_locked(ip, flags);
 394         xfs_iunlock(ip, XFS_ILOCK_EXCL);
 395
 396         return error;
 397 }
 398
 399 /*
 400  * Release dquots (and their references) if any.
 401  * The inode should be locked EXCL except when this's called by
 402  * xfs_ireclaim.
 403  */
 404 void
 405 xfs_qm_dqdetach(
 406         xfs_inode_t     *ip)
 407 {
 408         if (!(ip->i_udquot || ip->i_gdquot || ip->i_pdquot))
 409                 return;
 410
 411         trace_xfs_dquot_dqdetach(ip);
 412
 413         ASSERT(!xfs_is_quota_inode(&ip->i_mount->m_sb, ip->i_ino));
 414         if (ip->i_udquot) {
 415                 xfs_qm_dqrele(ip->i_udquot);
 416                 ip->i_udquot = NULL;
 417         }
 418         if (ip->i_gdquot) {
 419                 xfs_qm_dqrele(ip->i_gdquot);
 420                 ip->i_gdquot = NULL;
 421         }
 422         if (ip->i_pdquot) {
 423                 xfs_qm_dqrele(ip->i_pdquot);
 424                 ip->i_pdquot = NULL;
 425         }
 426 }
 427
 428 struct xfs_qm_isolate {
 429         struct list_head        buffers;
 430         struct list_head        dispose;
 431 };
 432
 433 static enum lru_status
 434 xfs_qm_dquot_isolate(
 435         struct list_head        *item,
 436         struct list_lru_one     *lru,
 437         spinlock_t              *lru_lock,
 438         void                    *arg)
 439                 __releases(lru_lock) __acquires(lru_lock)
 440 {
 441         struct xfs_dquot        *dqp = container_of(item,
 442                                                 struct xfs_dquot, q_lru);
 443         struct xfs_qm_isolate   *isol = arg;
 444
 445         if (!xfs_dqlock_nowait(dqp))
 446                 goto out_miss_busy;
 447
 448         /*
 449          * This dquot has acquired a reference in the meantime remove it from
 450          * the freelist and try again.
 451          */
 452         if (dqp->q_nrefs) {
 453                 xfs_dqunlock(dqp);
 454                 XFS_STATS_INC(dqp->q_mount, xs_qm_dqwants);
 455
 456                 trace_xfs_dqreclaim_want(dqp);
 457                 list_lru_isolate(lru, &dqp->q_lru);
 458                 XFS_STATS_DEC(dqp->q_mount, xs_qm_dquot_unused);
 459                 return LRU_REMOVED;
 460         }
 461
 462         /*
 463          * If the dquot is dirty, flush it. If it's already being flushed, just
 464          * skip it so there is time for the IO to complete before we try to
 465          * reclaim it again on the next LRU pass.
 466          */
 467         if (!xfs_dqflock_nowait(dqp)) {
 468                 xfs_dqunlock(dqp);
 469                 goto out_miss_busy;
 470         }
 471
 472         if (XFS_DQ_IS_DIRTY(dqp)) {
 473                 struct xfs_buf  *bp = NULL;
 474                 int             error;
 475
 476                 trace_xfs_dqreclaim_dirty(dqp);
 477
 478                 /* we have to drop the LRU lock to flush the dquot */
 479                 spin_unlock(lru_lock);
 480
 481                 error = xfs_qm_dqflush(dqp, &bp);
 482                 if (error) {
 483                         xfs_warn(dqp->q_mount, "%s: dquot "PTR_FMT" flush failed",
 484                                  __func__, dqp);
 485                         goto out_unlock_dirty;
 486                 }
 487
 488                 xfs_buf_delwri_queue(bp, &isol->buffers);
 489                 xfs_buf_relse(bp);
 490                 goto out_unlock_dirty;
 491         }
 492         xfs_dqfunlock(dqp);
 493
 494         /*
 495          * Prevent lookups now that we are past the point of no return.
 496          */
 497         dqp->dq_flags |= XFS_DQ_FREEING;
 498         xfs_dqunlock(dqp);
 499
 500         ASSERT(dqp->q_nrefs == 0);
 501         list_lru_isolate_move(lru, &dqp->q_lru, &isol->dispose);
 502         XFS_STATS_DEC(dqp->q_mount, xs_qm_dquot_unused);
 503         trace_xfs_dqreclaim_done(dqp);
 504         XFS_STATS_INC(dqp->q_mount, xs_qm_dqreclaims);
 505         return LRU_REMOVED;
 506
 507 out_miss_busy:
 508         trace_xfs_dqreclaim_busy(dqp);
 509         XFS_STATS_INC(dqp->q_mount, xs_qm_dqreclaim_misses);
 510         return LRU_SKIP;
 511
 512 out_unlock_dirty:
 513         trace_xfs_dqreclaim_busy(dqp);
 514         XFS_STATS_INC(dqp->q_mount, xs_qm_dqreclaim_misses);
 515         xfs_dqunlock(dqp);
 516         spin_lock(lru_lock);
 517         return LRU_RETRY;
 518 }
 519
 520 static unsigned long
 521 xfs_qm_shrink_scan(
 522         struct shrinker         *shrink,
 523         struct shrink_control   *sc)
 524 {
 525         struct xfs_quotainfo    *qi = container_of(shrink,
 526                                         struct xfs_quotainfo, qi_shrinker);
 527         struct xfs_qm_isolate   isol;
 528         unsigned long           freed;
 529         int                     error;
 530
 531         if ((sc->gfp_mask & (__GFP_FS|__GFP_DIRECT_RECLAIM)) != (__GFP_FS|__GFP_DIRECT_RECLAIM))
 532                 return 0;
 533
 534         INIT_LIST_HEAD(&isol.buffers);
 535         INIT_LIST_HEAD(&isol.dispose);
 536
 537         freed = list_lru_shrink_walk(&qi->qi_lru, sc,
 538                                      xfs_qm_dquot_isolate, &isol);
 539
 540         error = xfs_buf_delwri_submit(&isol.buffers);
 541         if (error)
 542                 xfs_warn(NULL, "%s: dquot reclaim failed", __func__);
 543
 544         while (!list_empty(&isol.dispose)) {
 545                 struct xfs_dquot        *dqp;
 546
 547                 dqp = list_first_entry(&isol.dispose, struct xfs_dquot, q_lru);
 548                 list_del_init(&dqp->q_lru);
 549                 xfs_qm_dqfree_one(dqp);
 550         }
 551
 552         return freed;
 553 }
 554
 555 static unsigned long
 556 xfs_qm_shrink_count(
 557         struct shrinker         *shrink,
 558         struct shrink_control   *sc)
 559 {
 560         struct xfs_quotainfo    *qi = container_of(shrink,
 561                                         struct xfs_quotainfo, qi_shrinker);
 562
 563         return list_lru_shrink_count(&qi->qi_lru, sc);
 564 }
 565
 566 STATIC void
 567 xfs_qm_set_defquota(
 568         xfs_mount_t     *mp,
 569         uint            type,
 570         xfs_quotainfo_t *qinf)
 571 {
 572         xfs_dquot_t             *dqp;
 573         struct xfs_def_quota    *defq;
 574         int                     error;
 575
 576         error = xfs_qm_dqread(mp, 0, type, 0, &dqp);
 577
 578         if (!error) {
 579                 xfs_disk_dquot_t        *ddqp = &dqp->q_core;
 580
 581                 defq = xfs_get_defquota(dqp, qinf);
 582
 583                 /*
 584                  * Timers and warnings have been already set, let's just set the
 585                  * default limits for this quota type
 586                  */
 587                 defq->bhardlimit = be64_to_cpu(ddqp->d_blk_hardlimit);
 588                 defq->bsoftlimit = be64_to_cpu(ddqp->d_blk_softlimit);
 589                 defq->ihardlimit = be64_to_cpu(ddqp->d_ino_hardlimit);
 590                 defq->isoftlimit = be64_to_cpu(ddqp->d_ino_softlimit);
 591                 defq->rtbhardlimit = be64_to_cpu(ddqp->d_rtb_hardlimit);
 592                 defq->rtbsoftlimit = be64_to_cpu(ddqp->d_rtb_softlimit);
 593                 xfs_qm_dqdestroy(dqp);
 594         }
 595 }
 596
 597 /*
 598  * This initializes all the quota information that's kept in the
 599  * mount structure
 600  */
 601 STATIC int
 602 xfs_qm_init_quotainfo(
 603         xfs_mount_t     *mp)
 604 {
 605         xfs_quotainfo_t *qinf;
 606         int             error;
 607         xfs_dquot_t     *dqp;
 608
 609         ASSERT(XFS_IS_QUOTA_RUNNING(mp));
 610
 611         qinf = mp->m_quotainfo = kmem_zalloc(sizeof(xfs_quotainfo_t), KM_SLEEP);
 612
 613         error = list_lru_init(&qinf->qi_lru);
 614         if (error)
 615                 goto out_free_qinf;
 616
 617         /*
 618          * See if quotainodes are setup, and if not, allocate them,
 619          * and change the superblock accordingly.
 620          */
 621         error = xfs_qm_init_quotainos(mp);
 622         if (error)
 623                 goto out_free_lru;
 624
 625         INIT_RADIX_TREE(&qinf->qi_uquota_tree, GFP_NOFS);
 626         INIT_RADIX_TREE(&qinf->qi_gquota_tree, GFP_NOFS);
 627         INIT_RADIX_TREE(&qinf->qi_pquota_tree, GFP_NOFS);
 628         mutex_init(&qinf->qi_tree_lock);
 629
 630         /* mutex used to serialize quotaoffs */
 631         mutex_init(&qinf->qi_quotaofflock);
 632
 633         /* Precalc some constants */
 634         qinf->qi_dqchunklen = XFS_FSB_TO_BB(mp, XFS_DQUOT_CLUSTER_SIZE_FSB);
 635         qinf->qi_dqperchunk = xfs_calc_dquots_per_chunk(qinf->qi_dqchunklen);
 636
 637         mp->m_qflags |= (mp->m_sb.sb_qflags & XFS_ALL_QUOTA_CHKD);
 638
 639         /*
 640          * We try to get the limits from the superuser's limits fields.
 641          * This is quite hacky, but it is standard quota practice.
 642          *
 643          * Since we may not have done a quotacheck by this point, just read
 644          * the dquot without attaching it to any hashtables or lists.
 645          *
 646          * Timers and warnings are globally set by the first timer found in
 647          * user/group/proj quota types, otherwise a default value is used.
 648          * This should be split into different fields per quota type.
 649          */
 650         error = xfs_qm_dqread(mp, 0,
 651                         XFS_IS_UQUOTA_RUNNING(mp) ? XFS_DQ_USER :
 652                          (XFS_IS_GQUOTA_RUNNING(mp) ? XFS_DQ_GROUP :
 653                           XFS_DQ_PROJ),
 654                         0, &dqp);
 655
 656         if (!error) {
 657                 xfs_disk_dquot_t        *ddqp = &dqp->q_core;
 658
 659                 /*
 660                  * The warnings and timers set the grace period given to
 661                  * a user or group before he or she can not perform any
 662                  * more writing. If it is zero, a default is used.
 663                  */
 664                 qinf->qi_btimelimit = ddqp->d_btimer ?
 665                         be32_to_cpu(ddqp->d_btimer) : XFS_QM_BTIMELIMIT;
 666                 qinf->qi_itimelimit = ddqp->d_itimer ?
 667                         be32_to_cpu(ddqp->d_itimer) : XFS_QM_ITIMELIMIT;
 668                 qinf->qi_rtbtimelimit = ddqp->d_rtbtimer ?
 669                         be32_to_cpu(ddqp->d_rtbtimer) : XFS_QM_RTBTIMELIMIT;
 670                 qinf->qi_bwarnlimit = ddqp->d_bwarns ?
 671                         be16_to_cpu(ddqp->d_bwarns) : XFS_QM_BWARNLIMIT;
 672                 qinf->qi_iwarnlimit = ddqp->d_iwarns ?
 673                         be16_to_cpu(ddqp->d_iwarns) : XFS_QM_IWARNLIMIT;
 674                 qinf->qi_rtbwarnlimit = ddqp->d_rtbwarns ?
 675                         be16_to_cpu(ddqp->d_rtbwarns) : XFS_QM_RTBWARNLIMIT;
 676                 xfs_qm_dqdestroy(dqp);
 677         } else {
 678                 qinf->qi_btimelimit = XFS_QM_BTIMELIMIT;
 679                 qinf->qi_itimelimit = XFS_QM_ITIMELIMIT;
 680                 qinf->qi_rtbtimelimit = XFS_QM_RTBTIMELIMIT;
 681                 qinf->qi_bwarnlimit = XFS_QM_BWARNLIMIT;
 682                 qinf->qi_iwarnlimit = XFS_QM_IWARNLIMIT;
 683                 qinf->qi_rtbwarnlimit = XFS_QM_RTBWARNLIMIT;
 684         }
 685
 686         if (XFS_IS_UQUOTA_RUNNING(mp))
 687                 xfs_qm_set_defquota(mp, XFS_DQ_USER, qinf);
 688         if (XFS_IS_GQUOTA_RUNNING(mp))
 689                 xfs_qm_set_defquota(mp, XFS_DQ_GROUP, qinf);
 690         if (XFS_IS_PQUOTA_RUNNING(mp))
 691                 xfs_qm_set_defquota(mp, XFS_DQ_PROJ, qinf);
 692
 693         qinf->qi_shrinker.count_objects = xfs_qm_shrink_count;
 694         qinf->qi_shrinker.scan_objects = xfs_qm_shrink_scan;
 695         qinf->qi_shrinker.seeks = DEFAULT_SEEKS;
 696         qinf->qi_shrinker.flags = SHRINKER_NUMA_AWARE;
 697
 698         error = register_shrinker(&qinf->qi_shrinker);
 699         if (error)
 700                 goto out_free_inos;
 701
 702         return 0;
 703
 704 out_free_inos:
 705         mutex_destroy(&qinf->qi_quotaofflock);
 706         mutex_destroy(&qinf->qi_tree_lock);
 707         xfs_qm_destroy_quotainos(qinf);
 708 out_free_lru:
 709         list_lru_destroy(&qinf->qi_lru);
 710 out_free_qinf:
 711         kmem_free(qinf);
 712         mp->m_quotainfo = NULL;
 713         return error;
 714 }
 715
 716 /*
 717  * Gets called when unmounting a filesystem or when all quotas get
 718  * turned off.
 719  * This purges the quota inodes, destroys locks and frees itself.
 720  */
 721 void
 722 xfs_qm_destroy_quotainfo(
 723         xfs_mount_t     *mp)
 724 {
 725         xfs_quotainfo_t *qi;
 726
 727         qi = mp->m_quotainfo;
 728         ASSERT(qi != NULL);
 729
 730         unregister_shrinker(&qi->qi_shrinker);
 731         list_lru_destroy(&qi->qi_lru);
 732         xfs_qm_destroy_quotainos(qi);
 733         mutex_destroy(&qi->qi_tree_lock);
 734         mutex_destroy(&qi->qi_quotaofflock);
 735         kmem_free(qi);
 736         mp->m_quotainfo = NULL;
 737 }
 738
 739 /*
 740  * Create an inode and return with a reference already taken, but unlocked
 741  * This is how we create quota inodes
 742  */
 743 STATIC int
 744 xfs_qm_qino_alloc(
 745         xfs_mount_t     *mp,
 746         xfs_inode_t     **ip,
 747         uint            flags)
 748 {
 749         xfs_trans_t     *tp;
 750         int             error;
 751         int             committed;
 752         bool            need_alloc = true;
 753
 754         *ip = NULL;
 755         /*
 756          * With superblock that doesn't have separate pquotino, we
 757          * share an inode between gquota and pquota. If the on-disk
 758          * superblock has GQUOTA and the filesystem is now mounted
 759          * with PQUOTA, just use sb_gquotino for sb_pquotino and
 760          * vice-versa.
 761          */
 762         if (!xfs_sb_version_has_pquotino(&mp->m_sb) &&
 763                         (flags & (XFS_QMOPT_PQUOTA|XFS_QMOPT_GQUOTA))) {
 764                 xfs_ino_t ino = NULLFSINO;
 765
 766                 if ((flags & XFS_QMOPT_PQUOTA) &&
 767                              (mp->m_sb.sb_gquotino != NULLFSINO)) {
 768                         ino = mp->m_sb.sb_gquotino;
 769                         ASSERT(mp->m_sb.sb_pquotino == NULLFSINO);
 770                 } else if ((flags & XFS_QMOPT_GQUOTA) &&
 771                              (mp->m_sb.sb_pquotino != NULLFSINO)) {
 772                         ino = mp->m_sb.sb_pquotino;
 773                         ASSERT(mp->m_sb.sb_gquotino == NULLFSINO);
 774                 }
 775                 if (ino != NULLFSINO) {
 776                         error = xfs_iget(mp, NULL, ino, 0, 0, ip);
 777                         if (error)
 778                                 return error;
 779                         mp->m_sb.sb_gquotino = NULLFSINO;
 780                         mp->m_sb.sb_pquotino = NULLFSINO;
 781                         need_alloc = false;
 782                 }
 783         }
 784
 785         error = xfs_trans_alloc(mp, &M_RES(mp)->tr_create,
 786                         XFS_QM_QINOCREATE_SPACE_RES(mp), 0, 0, &tp);
 787         if (error)
 788                 return error;
 789
 790         if (need_alloc) {
 791                 error = xfs_dir_ialloc(&tp, NULL, S_IFREG, 1, 0, 0, ip,
 792                                 &committed);
 793                 if (error) {
 794                         xfs_trans_cancel(tp);
 795                         return error;
 796                 }
 797         }
 798
 799         /*
 800          * Make the changes in the superblock, and log those too.
 801          * sbfields arg may contain fields other than *QUOTINO;
 802          * VERSIONNUM for example.
 803          */
 804         spin_lock(&mp->m_sb_lock);
 805         if (flags & XFS_QMOPT_SBVERSION) {
 806                 ASSERT(!xfs_sb_version_hasquota(&mp->m_sb));
 807
 808                 xfs_sb_version_addquota(&mp->m_sb);
 809                 mp->m_sb.sb_uquotino = NULLFSINO;
 810                 mp->m_sb.sb_gquotino = NULLFSINO;
 811                 mp->m_sb.sb_pquotino = NULLFSINO;
 812
 813                 /* qflags will get updated fully _after_ quotacheck */
 814                 mp->m_sb.sb_qflags = mp->m_qflags & XFS_ALL_QUOTA_ACCT;
 815         }
 816         if (flags & XFS_QMOPT_UQUOTA)
 817                 mp->m_sb.sb_uquotino = (*ip)->i_ino;
 818         else if (flags & XFS_QMOPT_GQUOTA)
 819                 mp->m_sb.sb_gquotino = (*ip)->i_ino;
 820         else
 821                 mp->m_sb.sb_pquotino = (*ip)->i_ino;
 822         spin_unlock(&mp->m_sb_lock);
 823         xfs_log_sb(tp);
 824
 825         error = xfs_trans_commit(tp);
 826         if (error) {
 827                 ASSERT(XFS_FORCED_SHUTDOWN(mp));
 828                 xfs_alert(mp, "%s failed (error %d)!", __func__, error);
 829         }
 830         if (need_alloc)
 831                 xfs_finish_inode_setup(*ip);
 832         return error;
 833 }
 834
 835
 836 STATIC void
 837 xfs_qm_reset_dqcounts(
 838         xfs_mount_t     *mp,
 839         xfs_buf_t       *bp,
 840         xfs_dqid_t      id,
 841         uint            type)
 842 {
 843         struct xfs_dqblk        *dqb;
 844         int                     j;
 845         xfs_failaddr_t          fa;
 846
 847         trace_xfs_reset_dqcounts(bp, _RET_IP_);
 848
 849         /*
 850          * Reset all counters and timers. They'll be
 851          * started afresh by xfs_qm_quotacheck.
 852          */
 853 #ifdef DEBUG
 854         j = (int)XFS_FSB_TO_B(mp, XFS_DQUOT_CLUSTER_SIZE_FSB) /
 855                 sizeof(xfs_dqblk_t);
 856         ASSERT(mp->m_quotainfo->qi_dqperchunk == j);
 857 #endif
 858         dqb = bp->b_addr;
 859         for (j = 0; j < mp->m_quotainfo->qi_dqperchunk; j++) {
 860                 struct xfs_disk_dquot   *ddq;
 861
 862                 ddq = (struct xfs_disk_dquot *)&dqb[j];
 863
 864                 /*
 865                  * Do a sanity check, and if needed, repair the dqblk. Don't
 866                  * output any warnings because it's perfectly possible to
 867                  * find uninitialised dquot blks. See comment in
 868                  * xfs_dquot_verify.
 869                  */
 870                 fa = xfs_dquot_verify(mp, ddq, id + j, type, 0);
 871                 if (fa)
 872                         xfs_dquot_repair(mp, ddq, id + j, type);
 873
 874                 /*
 875                  * Reset type in case we are reusing group quota file for
 876                  * project quotas or vice versa
 877                  */
 878                 ddq->d_flags = type;
 879                 ddq->d_bcount = 0;
 880                 ddq->d_icount = 0;
 881                 ddq->d_rtbcount = 0;
 882                 ddq->d_btimer = 0;
 883                 ddq->d_itimer = 0;
 884                 ddq->d_rtbtimer = 0;
 885                 ddq->d_bwarns = 0;
 886                 ddq->d_iwarns = 0;
 887                 ddq->d_rtbwarns = 0;
 888
 889                 if (xfs_sb_version_hascrc(&mp->m_sb)) {
 890                         xfs_update_cksum((char *)&dqb[j],
 891                                          sizeof(struct xfs_dqblk),
 892                                          XFS_DQUOT_CRC_OFF);
 893                 }
 894         }
 895 }
 896
 897 STATIC int
 898 xfs_qm_dqiter_bufs(
 899         struct xfs_mount        *mp,
 900         xfs_dqid_t              firstid,
 901         xfs_fsblock_t           bno,
 902         xfs_filblks_t           blkcnt,
 903         uint                    flags,
 904         struct list_head        *buffer_list)
 905 {
 906         struct xfs_buf          *bp;
 907         int                     error;
 908         int                     type;
 909
 910         ASSERT(blkcnt > 0);
 911         type = flags & XFS_QMOPT_UQUOTA ? XFS_DQ_USER :
 912                 (flags & XFS_QMOPT_PQUOTA ? XFS_DQ_PROJ : XFS_DQ_GROUP);
 913         error = 0;
 914
 915         /*
 916          * Blkcnt arg can be a very big number, and might even be
 917          * larger than the log itself. So, we have to break it up into
 918          * manageable-sized transactions.
 919          * Note that we don't start a permanent transaction here; we might
 920          * not be able to get a log reservation for the whole thing up front,
 921          * and we don't really care to either, because we just discard
 922          * everything if we were to crash in the middle of this loop.
 923          */
 924         while (blkcnt--) {
 925                 error = xfs_trans_read_buf(mp, NULL, mp->m_ddev_targp,
 926                               XFS_FSB_TO_DADDR(mp, bno),
 927                               mp->m_quotainfo->qi_dqchunklen, 0, &bp,
 928                               &xfs_dquot_buf_ops);
 929
 930                 /*
 931                  * CRC and validation errors will return a EFSCORRUPTED here. If
 932                  * this occurs, re-read without CRC validation so that we can
 933                  * repair the damage via xfs_qm_reset_dqcounts(). This process
 934                  * will leave a trace in the log indicating corruption has
 935                  * been detected.
 936                  */
 937                 if (error == -EFSCORRUPTED) {
 938                         error = xfs_trans_read_buf(mp, NULL, mp->m_ddev_targp,
 939                                       XFS_FSB_TO_DADDR(mp, bno),
 940                                       mp->m_quotainfo->qi_dqchunklen, 0, &bp,
 941                                       NULL);
 942                 }
 943
 944                 if (error)
 945                         break;
 946
 947                 /*
 948                  * A corrupt buffer might not have a verifier attached, so
 949                  * make sure we have the correct one attached before writeback
 950                  * occurs.
 951                  */
 952                 bp->b_ops = &xfs_dquot_buf_ops;
 953                 xfs_qm_reset_dqcounts(mp, bp, firstid, type);
 954                 xfs_buf_delwri_queue(bp, buffer_list);
 955                 xfs_buf_relse(bp);
 956
 957                 /* goto the next block. */
 958                 bno++;
 959                 firstid += mp->m_quotainfo->qi_dqperchunk;
 960         }
 961
 962         return error;
 963 }
 964
 965 /*
 966  * Iterate over all allocated USR/GRP/PRJ dquots in the system, calling a
 967  * caller supplied function for every chunk of dquots that we find.
 968  */
 969 STATIC int
 970 xfs_qm_dqiterate(
 971         struct xfs_mount        *mp,
 972         struct xfs_inode        *qip,
 973         uint                    flags,
 974         struct list_head        *buffer_list)
 975 {
 976         struct xfs_bmbt_irec    *map;
 977         int                     i, nmaps;       /* number of map entries */
 978         int                     error;          /* return value */
 979         xfs_fileoff_t           lblkno;
 980         xfs_filblks_t           maxlblkcnt;
 981         xfs_dqid_t              firstid;
 982         xfs_fsblock_t           rablkno;
 983         xfs_filblks_t           rablkcnt;
 984
 985         error = 0;
 986         /*
 987          * This looks racy, but we can't keep an inode lock across a
 988          * trans_reserve. But, this gets called during quotacheck, and that
 989          * happens only at mount time which is single threaded.
 990          */
 991         if (qip->i_d.di_nblocks == 0)
 992                 return 0;
 993
 994         map = kmem_alloc(XFS_DQITER_MAP_SIZE * sizeof(*map), KM_SLEEP);
 995
 996         lblkno = 0;
 997         maxlblkcnt = XFS_B_TO_FSB(mp, mp->m_super->s_maxbytes);
 998         do {
 999                 uint            lock_mode;
1000
1001                 nmaps = XFS_DQITER_MAP_SIZE;
1002                 /*
1003                  * We aren't changing the inode itself. Just changing
1004                  * some of its data. No new blocks are added here, and
1005                  * the inode is never added to the transaction.
1006                  */
1007                 lock_mode = xfs_ilock_data_map_shared(qip);
1008                 error = xfs_bmapi_read(qip, lblkno, maxlblkcnt - lblkno,
1009                                        map, &nmaps, 0);
1010                 xfs_iunlock(qip, lock_mode);
1011                 if (error)
1012                         break;
1013
1014                 ASSERT(nmaps <= XFS_DQITER_MAP_SIZE);
1015                 for (i = 0; i < nmaps; i++) {
1016                         ASSERT(map[i].br_startblock != DELAYSTARTBLOCK);
1017                         ASSERT(map[i].br_blockcount);
1018
1019
1020                         lblkno += map[i].br_blockcount;
1021
1022                         if (map[i].br_startblock == HOLESTARTBLOCK)
1023                                 continue;
1024
1025                         firstid = (xfs_dqid_t) map[i].br_startoff *
1026                                 mp->m_quotainfo->qi_dqperchunk;
1027                         /*
1028                          * Do a read-ahead on the next extent.
1029                          */
1030                         if ((i+1 < nmaps) &&
1031                             (map[i+1].br_startblock != HOLESTARTBLOCK)) {
1032                                 rablkcnt =  map[i+1].br_blockcount;
1033                                 rablkno = map[i+1].br_startblock;
1034                                 while (rablkcnt--) {
1035                                         xfs_buf_readahead(mp->m_ddev_targp,
1036                                                XFS_FSB_TO_DADDR(mp, rablkno),
1037                                                mp->m_quotainfo->qi_dqchunklen,
1038                                                &xfs_dquot_buf_ops);
1039                                         rablkno++;
1040                                 }
1041                         }
1042                         /*
1043                          * Iterate thru all the blks in the extent and
1044                          * reset the counters of all the dquots inside them.
1045                          */
1046                         error = xfs_qm_dqiter_bufs(mp, firstid,
1047                                                    map[i].br_startblock,
1048                                                    map[i].br_blockcount,
1049                                                    flags, buffer_list);
1050                         if (error)
1051                                 goto out;
1052                 }
1053         } while (nmaps > 0);
1054
1055 out:
1056         kmem_free(map);
1057         return error;
1058 }
1059
1060 /*
1061  * Called by dqusage_adjust in doing a quotacheck.
1062  *
1063  * Given the inode, and a dquot id this updates both the incore dqout as well
1064  * as the buffer copy. This is so that once the quotacheck is done, we can
1065  * just log all the buffers, as opposed to logging numerous updates to
1066  * individual dquots.
1067  */
1068 STATIC int
1069 xfs_qm_quotacheck_dqadjust(
1070         struct xfs_inode        *ip,
1071         xfs_dqid_t              id,
1072         uint                    type,
1073         xfs_qcnt_t              nblks,
1074         xfs_qcnt_t              rtblks)
1075 {
1076         struct xfs_mount        *mp = ip->i_mount;
1077         struct xfs_dquot        *dqp;
1078         int                     error;
1079
1080         error = xfs_qm_dqget(mp, ip, id, type, XFS_QMOPT_DQALLOC, &dqp);
1081         if (error) {
1082                 /*
1083                  * Shouldn't be able to turn off quotas here.
1084                  */
1085                 ASSERT(error != -ESRCH);
1086                 ASSERT(error != -ENOENT);
1087                 return error;
1088         }
1089
1090         trace_xfs_dqadjust(dqp);
1091
1092         /*
1093          * Adjust the inode count and the block count to reflect this inode's
1094          * resource usage.
1095          */
1096         be64_add_cpu(&dqp->q_core.d_icount, 1);
1097         dqp->q_res_icount++;
1098         if (nblks) {
1099                 be64_add_cpu(&dqp->q_core.d_bcount, nblks);
1100                 dqp->q_res_bcount += nblks;
1101         }
1102         if (rtblks) {
1103                 be64_add_cpu(&dqp->q_core.d_rtbcount, rtblks);
1104                 dqp->q_res_rtbcount += rtblks;
1105         }
1106
1107         /*
1108          * Set default limits, adjust timers (since we changed usages)
1109          *
1110          * There are no timers for the default values set in the root dquot.
1111          */
1112         if (dqp->q_core.d_id) {
1113                 xfs_qm_adjust_dqlimits(mp, dqp);
1114                 xfs_qm_adjust_dqtimers(mp, &dqp->q_core);
1115         }
1116
1117         dqp->dq_flags |= XFS_DQ_DIRTY;
1118         xfs_qm_dqput(dqp);
1119         return 0;
1120 }
1121
1122 /*
1123  * callback routine supplied to bulkstat(). Given an inumber, find its
1124  * dquots and update them to account for resources taken by that inode.
1125  */
1126 /* ARGSUSED */
1127 STATIC int
1128 xfs_qm_dqusage_adjust(
1129         xfs_mount_t     *mp,            /* mount point for filesystem */
1130         xfs_ino_t       ino,            /* inode number to get data for */
1131         void            __user *buffer, /* not used */
1132         int             ubsize,         /* not used */
1133         int             *ubused,        /* not used */
1134         int             *res)           /* result code value */
1135 {
1136         xfs_inode_t     *ip;
1137         xfs_qcnt_t      nblks;
1138         xfs_filblks_t   rtblks = 0;     /* total rt blks */
1139         int             error;
1140
1141         ASSERT(XFS_IS_QUOTA_RUNNING(mp));
1142
1143         /*
1144          * rootino must have its resources accounted for, not so with the quota
1145          * inodes.
1146          */
1147         if (xfs_is_quota_inode(&mp->m_sb, ino)) {
1148                 *res = BULKSTAT_RV_NOTHING;
1149                 return -EINVAL;
1150         }
1151
1152         /*
1153          * We don't _need_ to take the ilock EXCL. However, the xfs_qm_dqget
1154          * interface expects the inode to be exclusively locked because that's
1155          * the case in all other instances. It's OK that we do this because
1156          * quotacheck is done only at mount time.
1157          */
1158         error = xfs_iget(mp, NULL, ino, XFS_IGET_DONTCACHE, XFS_ILOCK_EXCL,
1159                          &ip);
1160         if (error) {
1161                 *res = BULKSTAT_RV_NOTHING;
1162                 return error;
1163         }
1164
1165         ASSERT(ip->i_delayed_blks == 0);
1166
1167         if (XFS_IS_REALTIME_INODE(ip)) {
1168                 struct xfs_ifork        *ifp = XFS_IFORK_PTR(ip, XFS_DATA_FORK);
1169
1170                 if (!(ifp->if_flags & XFS_IFEXTENTS)) {
1171                         error = xfs_iread_extents(NULL, ip, XFS_DATA_FORK);
1172                         if (error)
1173                                 goto error0;
1174                 }
1175
1176                 xfs_bmap_count_leaves(ifp, &rtblks);
1177         }
1178
1179         nblks = (xfs_qcnt_t)ip->i_d.di_nblocks - rtblks;
1180
1181         /*
1182          * Add the (disk blocks and inode) resources occupied by this
1183          * inode to its dquots. We do this adjustment in the incore dquot,
1184          * and also copy the changes to its buffer.
1185          * We don't care about putting these changes in a transaction
1186          * envelope because if we crash in the middle of a 'quotacheck'
1187          * we have to start from the beginning anyway.
1188          * Once we're done, we'll log all the dquot bufs.
1189          *
1190          * The *QUOTA_ON checks below may look pretty racy, but quotachecks
1191          * and quotaoffs don't race. (Quotachecks happen at mount time only).
1192          */
1193         if (XFS_IS_UQUOTA_ON(mp)) {
1194                 error = xfs_qm_quotacheck_dqadjust(ip, ip->i_d.di_uid,
1195                                                    XFS_DQ_USER, nblks, rtblks);
1196                 if (error)
1197                         goto error0;
1198         }
1199
1200         if (XFS_IS_GQUOTA_ON(mp)) {
1201                 error = xfs_qm_quotacheck_dqadjust(ip, ip->i_d.di_gid,
1202                                                    XFS_DQ_GROUP, nblks, rtblks);
1203                 if (error)
1204                         goto error0;
1205         }
1206
1207         if (XFS_IS_PQUOTA_ON(mp)) {
1208                 error = xfs_qm_quotacheck_dqadjust(ip, xfs_get_projid(ip),
1209                                                    XFS_DQ_PROJ, nblks, rtblks);
1210                 if (error)
1211                         goto error0;
1212         }
1213
1214         xfs_iunlock(ip, XFS_ILOCK_EXCL);
1215         IRELE(ip);
1216         *res = BULKSTAT_RV_DIDONE;
1217         return 0;
1218
1219 error0:
1220         xfs_iunlock(ip, XFS_ILOCK_EXCL);
1221         IRELE(ip);
1222         *res = BULKSTAT_RV_GIVEUP;
1223         return error;
1224 }
1225
1226 STATIC int
1227 xfs_qm_flush_one(
1228         struct xfs_dquot        *dqp,
1229         void                    *data)
1230 {
1231         struct xfs_mount        *mp = dqp->q_mount;
1232         struct list_head        *buffer_list = data;
1233         struct xfs_buf          *bp = NULL;
1234         int                     error = 0;
1235
1236         xfs_dqlock(dqp);
1237         if (dqp->dq_flags & XFS_DQ_FREEING)
1238                 goto out_unlock;
1239         if (!XFS_DQ_IS_DIRTY(dqp))
1240                 goto out_unlock;
1241
1242         /*
1243          * The only way the dquot is already flush locked by the time quotacheck
1244          * gets here is if reclaim flushed it before the dqadjust walk dirtied
1245          * it for the final time. Quotacheck collects all dquot bufs in the
1246          * local delwri queue before dquots are dirtied, so reclaim can't have
1247          * possibly queued it for I/O. The only way out is to push the buffer to
1248          * cycle the flush lock.
1249          */
1250         if (!xfs_dqflock_nowait(dqp)) {
1251                 /* buf is pinned in-core by delwri list */
1252                 DEFINE_SINGLE_BUF_MAP(map, dqp->q_blkno,
1253                                       mp->m_quotainfo->qi_dqchunklen);
1254                 bp = _xfs_buf_find(mp->m_ddev_targp, &map, 1, 0, NULL);
1255                 if (!bp) {
1256                         error = -EINVAL;
1257                         goto out_unlock;
1258                 }
1259                 xfs_buf_unlock(bp);
1260
1261                 xfs_buf_delwri_pushbuf(bp, buffer_list);
1262                 xfs_buf_rele(bp);
1263
1264                 error = -EAGAIN;
1265                 goto out_unlock;
1266         }
1267
1268         error = xfs_qm_dqflush(dqp, &bp);
1269         if (error)
1270                 goto out_unlock;
1271
1272         xfs_buf_delwri_queue(bp, buffer_list);
1273         xfs_buf_relse(bp);
1274 out_unlock:
1275         xfs_dqunlock(dqp);
1276         return error;
1277 }
1278
1279 /*
1280  * Walk thru all the filesystem inodes and construct a consistent view
1281  * of the disk quota world. If the quotacheck fails, disable quotas.
1282  */
1283 STATIC int
1284 xfs_qm_quotacheck(
1285         xfs_mount_t     *mp)
1286 {
1287         int                     done, count, error, error2;
1288         xfs_ino_t               lastino;
1289         size_t                  structsz;
1290         uint                    flags;
1291         LIST_HEAD               (buffer_list);
1292         struct xfs_inode        *uip = mp->m_quotainfo->qi_uquotaip;
1293         struct xfs_inode        *gip = mp->m_quotainfo->qi_gquotaip;
1294         struct xfs_inode        *pip = mp->m_quotainfo->qi_pquotaip;
1295
1296         count = INT_MAX;
1297         structsz = 1;
1298         lastino = 0;
1299         flags = 0;
1300
1301         ASSERT(uip || gip || pip);
1302         ASSERT(XFS_IS_QUOTA_RUNNING(mp));
1303
1304         xfs_notice(mp, "Quotacheck needed: Please wait.");
1305
1306         /*
1307          * First we go thru all the dquots on disk, USR and GRP/PRJ, and reset
1308          * their counters to zero. We need a clean slate.
1309          * We don't log our changes till later.
1310          */
1311         if (uip) {
1312                 error = xfs_qm_dqiterate(mp, uip, XFS_QMOPT_UQUOTA,
1313                                          &buffer_list);
1314                 if (error)
1315                         goto error_return;
1316                 flags |= XFS_UQUOTA_CHKD;
1317         }
1318
1319         if (gip) {
1320                 error = xfs_qm_dqiterate(mp, gip, XFS_QMOPT_GQUOTA,
1321                                          &buffer_list);
1322                 if (error)
1323                         goto error_return;
1324                 flags |= XFS_GQUOTA_CHKD;
1325         }
1326
1327         if (pip) {
1328                 error = xfs_qm_dqiterate(mp, pip, XFS_QMOPT_PQUOTA,
1329                                          &buffer_list);
1330                 if (error)
1331                         goto error_return;
1332                 flags |= XFS_PQUOTA_CHKD;
1333         }
1334
1335         do {
1336                 /*
1337                  * Iterate thru all the inodes in the file system,
1338                  * adjusting the corresponding dquot counters in core.
1339                  */
1340                 error = xfs_bulkstat(mp, &lastino, &count,
1341                                      xfs_qm_dqusage_adjust,
1342                                      structsz, NULL, &done);
1343                 if (error)
1344                         break;
1345
1346         } while (!done);
1347
1348         /*
1349          * We've made all the changes that we need to make incore.  Flush them
1350          * down to disk buffers if everything was updated successfully.
1351          */
1352         if (XFS_IS_UQUOTA_ON(mp)) {
1353                 error = xfs_qm_dquot_walk(mp, XFS_DQ_USER, xfs_qm_flush_one,
1354                                           &buffer_list);
1355         }
1356         if (XFS_IS_GQUOTA_ON(mp)) {
1357                 error2 = xfs_qm_dquot_walk(mp, XFS_DQ_GROUP, xfs_qm_flush_one,
1358                                            &buffer_list);
1359                 if (!error)
1360                         error = error2;
1361         }
1362         if (XFS_IS_PQUOTA_ON(mp)) {
1363                 error2 = xfs_qm_dquot_walk(mp, XFS_DQ_PROJ, xfs_qm_flush_one,
1364                                            &buffer_list);
1365                 if (!error)
1366                         error = error2;
1367         }
1368
1369         error2 = xfs_buf_delwri_submit(&buffer_list);
1370         if (!error)
1371                 error = error2;
1372
1373         /*
1374          * We can get this error if we couldn't do a dquot allocation inside
1375          * xfs_qm_dqusage_adjust (via bulkstat). We don't care about the
1376          * dirty dquots that might be cached, we just want to get rid of them
1377          * and turn quotaoff. The dquots won't be attached to any of the inodes
1378          * at this point (because we intentionally didn't in dqget_noattach).
1379          */
1380         if (error) {
1381                 xfs_qm_dqpurge_all(mp, XFS_QMOPT_QUOTALL);
1382                 goto error_return;
1383         }
1384
1385         /*
1386          * If one type of quotas is off, then it will lose its
1387          * quotachecked status, since we won't be doing accounting for
1388          * that type anymore.
1389          */
1390         mp->m_qflags &= ~XFS_ALL_QUOTA_CHKD;
1391         mp->m_qflags |= flags;
1392
1393  error_return:
1394         xfs_buf_delwri_cancel(&buffer_list);
1395
1396         if (error) {
1397                 xfs_warn(mp,
1398         "Quotacheck: Unsuccessful (Error %d): Disabling quotas.",
1399                         error);
1400                 /*
1401                  * We must turn off quotas.
1402                  */
1403                 ASSERT(mp->m_quotainfo != NULL);
1404                 xfs_qm_destroy_quotainfo(mp);
1405                 if (xfs_mount_reset_sbqflags(mp)) {
1406                         xfs_warn(mp,
1407                                 "Quotacheck: Failed to reset quota flags.");
1408                 }
1409         } else
1410                 xfs_notice(mp, "Quotacheck: Done.");
1411         return error;
1412 }
1413
1414 /*
1415  * This is called from xfs_mountfs to start quotas and initialize all
1416  * necessary data structures like quotainfo.  This is also responsible for
1417  * running a quotacheck as necessary.  We are guaranteed that the superblock
1418  * is consistently read in at this point.
1419  *
1420  * If we fail here, the mount will continue with quota turned off. We don't
1421  * need to inidicate success or failure at all.
1422  */
1423 void
1424 xfs_qm_mount_quotas(
1425         struct xfs_mount        *mp)
1426 {
1427         int                     error = 0;
1428         uint                    sbf;
1429
1430         /*
1431          * If quotas on realtime volumes is not supported, we disable
1432          * quotas immediately.
1433          */
1434         if (mp->m_sb.sb_rextents) {
1435                 xfs_notice(mp, "Cannot turn on quotas for realtime filesystem");
1436                 mp->m_qflags = 0;
1437                 goto write_changes;
1438         }
1439
1440         ASSERT(XFS_IS_QUOTA_RUNNING(mp));
1441
1442         /*
1443          * Allocate the quotainfo structure inside the mount struct, and
1444          * create quotainode(s), and change/rev superblock if necessary.
1445          */
1446         error = xfs_qm_init_quotainfo(mp);
1447         if (error) {
1448                 /*
1449                  * We must turn off quotas.
1450                  */
1451                 ASSERT(mp->m_quotainfo == NULL);
1452                 mp->m_qflags = 0;
1453                 goto write_changes;
1454         }
1455         /*
1456          * If any of the quotas are not consistent, do a quotacheck.
1457          */
1458         if (XFS_QM_NEED_QUOTACHECK(mp)) {
1459                 error = xfs_qm_quotacheck(mp);
1460                 if (error) {
1461                         /* Quotacheck failed and disabled quotas. */
1462                         return;
1463                 }
1464         }
1465         /*
1466          * If one type of quotas is off, then it will lose its
1467          * quotachecked status, since we won't be doing accounting for
1468          * that type anymore.
1469          */
1470         if (!XFS_IS_UQUOTA_ON(mp))
1471                 mp->m_qflags &= ~XFS_UQUOTA_CHKD;
1472         if (!XFS_IS_GQUOTA_ON(mp))
1473                 mp->m_qflags &= ~XFS_GQUOTA_CHKD;
1474         if (!XFS_IS_PQUOTA_ON(mp))
1475                 mp->m_qflags &= ~XFS_PQUOTA_CHKD;
1476
1477  write_changes:
1478         /*
1479          * We actually don't have to acquire the m_sb_lock at all.
1480          * This can only be called from mount, and that's single threaded. XXX
1481          */
1482         spin_lock(&mp->m_sb_lock);
1483         sbf = mp->m_sb.sb_qflags;
1484         mp->m_sb.sb_qflags = mp->m_qflags & XFS_MOUNT_QUOTA_ALL;
1485         spin_unlock(&mp->m_sb_lock);
1486
1487         if (sbf != (mp->m_qflags & XFS_MOUNT_QUOTA_ALL)) {
1488                 if (xfs_sync_sb(mp, false)) {
1489                         /*
1490                          * We could only have been turning quotas off.
1491                          * We aren't in very good shape actually because
1492                          * the incore structures are convinced that quotas are
1493                          * off, but the on disk superblock doesn't know that !
1494                          */
1495                         ASSERT(!(XFS_IS_QUOTA_RUNNING(mp)));
1496                         xfs_alert(mp, "%s: Superblock update failed!",
1497                                 __func__);
1498                 }
1499         }
1500
1501         if (error) {
1502                 xfs_warn(mp, "Failed to initialize disk quotas.");
1503                 return;
1504         }
1505 }
1506
1507 /*
1508  * This is called after the superblock has been read in and we're ready to
1509  * iget the quota inodes.
1510  */
1511 STATIC int
1512 xfs_qm_init_quotainos(
1513         xfs_mount_t     *mp)
1514 {
1515         struct xfs_inode        *uip = NULL;
1516         struct xfs_inode        *gip = NULL;
1517         struct xfs_inode        *pip = NULL;
1518         int                     error;
1519         uint                    flags = 0;
1520
1521         ASSERT(mp->m_quotainfo);
1522
1523         /*
1524          * Get the uquota and gquota inodes
1525          */
1526         if (xfs_sb_version_hasquota(&mp->m_sb)) {
1527                 if (XFS_IS_UQUOTA_ON(mp) &&
1528                     mp->m_sb.sb_uquotino != NULLFSINO) {
1529                         ASSERT(mp->m_sb.sb_uquotino > 0);
1530                         error = xfs_iget(mp, NULL, mp->m_sb.sb_uquotino,
1531                                              0, 0, &uip);
1532                         if (error)
1533                                 return error;
1534                 }
1535                 if (XFS_IS_GQUOTA_ON(mp) &&
1536                     mp->m_sb.sb_gquotino != NULLFSINO) {
1537                         ASSERT(mp->m_sb.sb_gquotino > 0);
1538                         error = xfs_iget(mp, NULL, mp->m_sb.sb_gquotino,
1539                                              0, 0, &gip);
1540                         if (error)
1541                                 goto error_rele;
1542                 }
1543                 if (XFS_IS_PQUOTA_ON(mp) &&
1544                     mp->m_sb.sb_pquotino != NULLFSINO) {
1545                         ASSERT(mp->m_sb.sb_pquotino > 0);
1546                         error = xfs_iget(mp, NULL, mp->m_sb.sb_pquotino,
1547                                              0, 0, &pip);
1548                         if (error)
1549                                 goto error_rele;
1550                 }
1551         } else {
1552                 flags |= XFS_QMOPT_SBVERSION;
1553         }
1554
1555         /*
1556          * Create the three inodes, if they don't exist already. The changes
1557          * made above will get added to a transaction and logged in one of
1558          * the qino_alloc calls below.  If the device is readonly,
1559          * temporarily switch to read-write to do this.
1560          */
1561         if (XFS_IS_UQUOTA_ON(mp) && uip == NULL) {
1562                 error = xfs_qm_qino_alloc(mp, &uip,
1563                                               flags | XFS_QMOPT_UQUOTA);
1564                 if (error)
1565                         goto error_rele;
1566
1567                 flags &= ~XFS_QMOPT_SBVERSION;
1568         }
1569         if (XFS_IS_GQUOTA_ON(mp) && gip == NULL) {
1570                 error = xfs_qm_qino_alloc(mp, &gip,
1571                                           flags | XFS_QMOPT_GQUOTA);
1572                 if (error)
1573                         goto error_rele;
1574
1575                 flags &= ~XFS_QMOPT_SBVERSION;
1576         }
1577         if (XFS_IS_PQUOTA_ON(mp) && pip == NULL) {
1578                 error = xfs_qm_qino_alloc(mp, &pip,
1579                                           flags | XFS_QMOPT_PQUOTA);
1580                 if (error)
1581                         goto error_rele;
1582         }
1583
1584         mp->m_quotainfo->qi_uquotaip = uip;
1585         mp->m_quotainfo->qi_gquotaip = gip;
1586         mp->m_quotainfo->qi_pquotaip = pip;
1587
1588         return 0;
1589
1590 error_rele:
1591         if (uip)
1592                 IRELE(uip);
1593         if (gip)
1594                 IRELE(gip);
1595         if (pip)
1596                 IRELE(pip);
1597         return error;
1598 }
1599
1600 STATIC void
1601 xfs_qm_destroy_quotainos(
1602         xfs_quotainfo_t *qi)
1603 {
1604         if (qi->qi_uquotaip) {
1605                 IRELE(qi->qi_uquotaip);
1606                 qi->qi_uquotaip = NULL; /* paranoia */
1607         }
1608         if (qi->qi_gquotaip) {
1609                 IRELE(qi->qi_gquotaip);
1610                 qi->qi_gquotaip = NULL;
1611         }
1612         if (qi->qi_pquotaip) {
1613                 IRELE(qi->qi_pquotaip);
1614                 qi->qi_pquotaip = NULL;
1615         }
1616 }
1617
1618 STATIC void
1619 xfs_qm_dqfree_one(
1620         struct xfs_dquot        *dqp)
1621 {
1622         struct xfs_mount        *mp = dqp->q_mount;
1623         struct xfs_quotainfo    *qi = mp->m_quotainfo;
1624
1625         mutex_lock(&qi->qi_tree_lock);
1626         radix_tree_delete(xfs_dquot_tree(qi, dqp->q_core.d_flags),
1627                           be32_to_cpu(dqp->q_core.d_id));
1628
1629         qi->qi_dquots--;
1630         mutex_unlock(&qi->qi_tree_lock);
1631
1632         xfs_qm_dqdestroy(dqp);
1633 }
1634
1635 /* --------------- utility functions for vnodeops ---------------- */
1636
1637
1638 /*
1639  * Given an inode, a uid, gid and prid make sure that we have
1640  * allocated relevant dquot(s) on disk, and that we won't exceed inode
1641  * quotas by creating this file.
1642  * This also attaches dquot(s) to the given inode after locking it,
1643  * and returns the dquots corresponding to the uid and/or gid.
1644  *
1645  * in   : inode (unlocked)
1646  * out  : udquot, gdquot with references taken and unlocked
1647  */
1648 int
1649 xfs_qm_vop_dqalloc(
1650         struct xfs_inode        *ip,
1651         xfs_dqid_t              uid,
1652         xfs_dqid_t              gid,
1653         prid_t                  prid,
1654         uint                    flags,
1655         struct xfs_dquot        **O_udqpp,
1656         struct xfs_dquot        **O_gdqpp,
1657         struct xfs_dquot        **O_pdqpp)
1658 {
1659         struct xfs_mount        *mp = ip->i_mount;
1660         struct xfs_dquot        *uq = NULL;
1661         struct xfs_dquot        *gq = NULL;
1662         struct xfs_dquot        *pq = NULL;
1663         int                     error;
1664         uint                    lockflags;
1665
1666         if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
1667                 return 0;
1668
1669         lockflags = XFS_ILOCK_EXCL;
1670         xfs_ilock(ip, lockflags);
1671
1672         if ((flags & XFS_QMOPT_INHERIT) && XFS_INHERIT_GID(ip))
1673                 gid = ip->i_d.di_gid;
1674
1675         /*
1676          * Attach the dquot(s) to this inode, doing a dquot allocation
1677          * if necessary. The dquot(s) will not be locked.
1678          */
1679         if (XFS_NOT_DQATTACHED(mp, ip)) {
1680                 error = xfs_qm_dqattach_locked(ip, XFS_QMOPT_DQALLOC);
1681                 if (error) {
1682                         xfs_iunlock(ip, lockflags);
1683                         return error;
1684                 }
1685         }
1686
1687         if ((flags & XFS_QMOPT_UQUOTA) && XFS_IS_UQUOTA_ON(mp)) {
1688                 if (ip->i_d.di_uid != uid) {
1689                         /*
1690                          * What we need is the dquot that has this uid, and
1691                          * if we send the inode to dqget, the uid of the inode
1692                          * takes priority over what's sent in the uid argument.
1693                          * We must unlock inode here before calling dqget if
1694                          * we're not sending the inode, because otherwise
1695                          * we'll deadlock by doing trans_reserve while
1696                          * holding ilock.
1697                          */
1698                         xfs_iunlock(ip, lockflags);
1699                         error = xfs_qm_dqget(mp, NULL, uid,
1700                                                  XFS_DQ_USER,
1701                                                  XFS_QMOPT_DQALLOC,
1702                                                  &uq);
1703                         if (error) {
1704                                 ASSERT(error != -ENOENT);
1705                                 return error;
1706                         }
1707                         /*
1708                          * Get the ilock in the right order.
1709                          */
1710                         xfs_dqunlock(uq);
1711                         lockflags = XFS_ILOCK_SHARED;
1712                         xfs_ilock(ip, lockflags);
1713                 } else {
1714                         /*
1715                          * Take an extra reference, because we'll return
1716                          * this to caller
1717                          */
1718                         ASSERT(ip->i_udquot);
1719                         uq = xfs_qm_dqhold(ip->i_udquot);
1720                 }
1721         }
1722         if ((flags & XFS_QMOPT_GQUOTA) && XFS_IS_GQUOTA_ON(mp)) {
1723                 if (ip->i_d.di_gid != gid) {
1724                         xfs_iunlock(ip, lockflags);
1725                         error = xfs_qm_dqget(mp, NULL, gid,
1726                                                  XFS_DQ_GROUP,
1727                                                  XFS_QMOPT_DQALLOC,
1728                                                  &gq);
1729                         if (error) {
1730                                 ASSERT(error != -ENOENT);
1731                                 goto error_rele;
1732                         }
1733                         xfs_dqunlock(gq);
1734                         lockflags = XFS_ILOCK_SHARED;
1735                         xfs_ilock(ip, lockflags);
1736                 } else {
1737                         ASSERT(ip->i_gdquot);
1738                         gq = xfs_qm_dqhold(ip->i_gdquot);
1739                 }
1740         }
1741         if ((flags & XFS_QMOPT_PQUOTA) && XFS_IS_PQUOTA_ON(mp)) {
1742                 if (xfs_get_projid(ip) != prid) {
1743                         xfs_iunlock(ip, lockflags);
1744                         error = xfs_qm_dqget(mp, NULL, (xfs_dqid_t)prid,
1745                                                  XFS_DQ_PROJ,
1746                                                  XFS_QMOPT_DQALLOC,
1747                                                  &pq);
1748                         if (error) {
1749                                 ASSERT(error != -ENOENT);
1750                                 goto error_rele;
1751                         }
1752                         xfs_dqunlock(pq);
1753                         lockflags = XFS_ILOCK_SHARED;
1754                         xfs_ilock(ip, lockflags);
1755                 } else {
1756                         ASSERT(ip->i_pdquot);
1757                         pq = xfs_qm_dqhold(ip->i_pdquot);
1758                 }
1759         }
1760         if (uq)
1761                 trace_xfs_dquot_dqalloc(ip);
1762
1763         xfs_iunlock(ip, lockflags);
1764         if (O_udqpp)
1765                 *O_udqpp = uq;
1766         else
1767                 xfs_qm_dqrele(uq);
1768         if (O_gdqpp)
1769                 *O_gdqpp = gq;
1770         else
1771                 xfs_qm_dqrele(gq);
1772         if (O_pdqpp)
1773                 *O_pdqpp = pq;
1774         else
1775                 xfs_qm_dqrele(pq);
1776         return 0;
1777
1778 error_rele:
1779         xfs_qm_dqrele(gq);
1780         xfs_qm_dqrele(uq);
1781         return error;
1782 }
1783
1784 /*
1785  * Actually transfer ownership, and do dquot modifications.
1786  * These were already reserved.
1787  */
1788 xfs_dquot_t *
1789 xfs_qm_vop_chown(
1790         xfs_trans_t     *tp,
1791         xfs_inode_t     *ip,
1792         xfs_dquot_t     **IO_olddq,
1793         xfs_dquot_t     *newdq)
1794 {
1795         xfs_dquot_t     *prevdq;
1796         uint            bfield = XFS_IS_REALTIME_INODE(ip) ?
1797                                  XFS_TRANS_DQ_RTBCOUNT : XFS_TRANS_DQ_BCOUNT;
1798
1799
1800         ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
1801         ASSERT(XFS_IS_QUOTA_RUNNING(ip->i_mount));
1802
1803         /* old dquot */
1804         prevdq = *IO_olddq;
1805         ASSERT(prevdq);
1806         ASSERT(prevdq != newdq);
1807
1808         xfs_trans_mod_dquot(tp, prevdq, bfield, -(ip->i_d.di_nblocks));
1809         xfs_trans_mod_dquot(tp, prevdq, XFS_TRANS_DQ_ICOUNT, -1);
1810
1811         /* the sparkling new dquot */
1812         xfs_trans_mod_dquot(tp, newdq, bfield, ip->i_d.di_nblocks);
1813         xfs_trans_mod_dquot(tp, newdq, XFS_TRANS_DQ_ICOUNT, 1);
1814
1815         /*
1816          * Take an extra reference, because the inode is going to keep
1817          * this dquot pointer even after the trans_commit.
1818          */
1819         *IO_olddq = xfs_qm_dqhold(newdq);
1820
1821         return prevdq;
1822 }
1823
1824 /*
1825  * Quota reservations for setattr(AT_UID|AT_GID|AT_PROJID).
1826  */
1827 int
1828 xfs_qm_vop_chown_reserve(
1829         struct xfs_trans        *tp,
1830         struct xfs_inode        *ip,
1831         struct xfs_dquot        *udqp,
1832         struct xfs_dquot        *gdqp,
1833         struct xfs_dquot        *pdqp,
1834         uint                    flags)
1835 {
1836         struct xfs_mount        *mp = ip->i_mount;
1837         uint                    delblks, blkflags, prjflags = 0;
1838         struct xfs_dquot        *udq_unres = NULL;
1839         struct xfs_dquot        *gdq_unres = NULL;
1840         struct xfs_dquot        *pdq_unres = NULL;
1841         struct xfs_dquot        *udq_delblks = NULL;
1842         struct xfs_dquot        *gdq_delblks = NULL;
1843         struct xfs_dquot        *pdq_delblks = NULL;
1844         int                     error;
1845
1846
1847         ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL|XFS_ILOCK_SHARED));
1848         ASSERT(XFS_IS_QUOTA_RUNNING(mp));
1849
1850         delblks = ip->i_delayed_blks;
1851         blkflags = XFS_IS_REALTIME_INODE(ip) ?
1852                         XFS_QMOPT_RES_RTBLKS : XFS_QMOPT_RES_REGBLKS;
1853
1854         if (XFS_IS_UQUOTA_ON(mp) && udqp &&
1855             ip->i_d.di_uid != be32_to_cpu(udqp->q_core.d_id)) {
1856                 udq_delblks = udqp;
1857                 /*
1858                  * If there are delayed allocation blocks, then we have to
1859                  * unreserve those from the old dquot, and add them to the
1860                  * new dquot.
1861                  */
1862                 if (delblks) {
1863                         ASSERT(ip->i_udquot);
1864                         udq_unres = ip->i_udquot;
1865                 }
1866         }
1867         if (XFS_IS_GQUOTA_ON(ip->i_mount) && gdqp &&
1868             ip->i_d.di_gid != be32_to_cpu(gdqp->q_core.d_id)) {
1869                 gdq_delblks = gdqp;
1870                 if (delblks) {
1871                         ASSERT(ip->i_gdquot);
1872                         gdq_unres = ip->i_gdquot;
1873                 }
1874         }
1875
1876         if (XFS_IS_PQUOTA_ON(ip->i_mount) && pdqp &&
1877             xfs_get_projid(ip) != be32_to_cpu(pdqp->q_core.d_id)) {
1878                 prjflags = XFS_QMOPT_ENOSPC;
1879                 pdq_delblks = pdqp;
1880                 if (delblks) {
1881                         ASSERT(ip->i_pdquot);
1882                         pdq_unres = ip->i_pdquot;
1883                 }
1884         }
1885
1886         error = xfs_trans_reserve_quota_bydquots(tp, ip->i_mount,
1887                                 udq_delblks, gdq_delblks, pdq_delblks,
1888                                 ip->i_d.di_nblocks, 1,
1889                                 flags | blkflags | prjflags);
1890         if (error)
1891                 return error;
1892
1893         /*
1894          * Do the delayed blks reservations/unreservations now. Since, these
1895          * are done without the help of a transaction, if a reservation fails
1896          * its previous reservations won't be automatically undone by trans
1897          * code. So, we have to do it manually here.
1898          */
1899         if (delblks) {
1900                 /*
1901                  * Do the reservations first. Unreservation can't fail.
1902                  */
1903                 ASSERT(udq_delblks || gdq_delblks || pdq_delblks);
1904                 ASSERT(udq_unres || gdq_unres || pdq_unres);
1905                 error = xfs_trans_reserve_quota_bydquots(NULL, ip->i_mount,
1906                             udq_delblks, gdq_delblks, pdq_delblks,
1907                             (xfs_qcnt_t)delblks, 0,
1908                             flags | blkflags | prjflags);
1909                 if (error)
1910                         return error;
1911                 xfs_trans_reserve_quota_bydquots(NULL, ip->i_mount,
1912                                 udq_unres, gdq_unres, pdq_unres,
1913                                 -((xfs_qcnt_t)delblks), 0, blkflags);
1914         }
1915
1916         return 0;
1917 }
1918
1919 int
1920 xfs_qm_vop_rename_dqattach(
1921         struct xfs_inode        **i_tab)
1922 {
1923         struct xfs_mount        *mp = i_tab[0]->i_mount;
1924         int                     i;
1925
1926         if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
1927                 return 0;
1928
1929         for (i = 0; (i < 4 && i_tab[i]); i++) {
1930                 struct xfs_inode        *ip = i_tab[i];
1931                 int                     error;
1932
1933                 /*
1934                  * Watch out for duplicate entries in the table.
1935                  */
1936                 if (i == 0 || ip != i_tab[i-1]) {
1937                         if (XFS_NOT_DQATTACHED(mp, ip)) {
1938                                 error = xfs_qm_dqattach(ip, 0);
1939                                 if (error)
1940                                         return error;
1941                         }
1942                 }
1943         }
1944         return 0;
1945 }
1946
1947 void
1948 xfs_qm_vop_create_dqattach(
1949         struct xfs_trans        *tp,
1950         struct xfs_inode        *ip,
1951         struct xfs_dquot        *udqp,
1952         struct xfs_dquot        *gdqp,
1953         struct xfs_dquot        *pdqp)
1954 {
1955         struct xfs_mount        *mp = tp->t_mountp;
1956
1957         if (!XFS_IS_QUOTA_RUNNING(mp) || !XFS_IS_QUOTA_ON(mp))
1958                 return;
1959
1960         ASSERT(xfs_isilocked(ip, XFS_ILOCK_EXCL));
1961         ASSERT(XFS_IS_QUOTA_RUNNING(mp));
1962
1963         if (udqp && XFS_IS_UQUOTA_ON(mp)) {
1964                 ASSERT(ip->i_udquot == NULL);
1965                 ASSERT(ip->i_d.di_uid == be32_to_cpu(udqp->q_core.d_id));
1966
1967                 ip->i_udquot = xfs_qm_dqhold(udqp);
1968                 xfs_trans_mod_dquot(tp, udqp, XFS_TRANS_DQ_ICOUNT, 1);
1969         }
1970         if (gdqp && XFS_IS_GQUOTA_ON(mp)) {
1971                 ASSERT(ip->i_gdquot == NULL);
1972                 ASSERT(ip->i_d.di_gid == be32_to_cpu(gdqp->q_core.d_id));
1973                 ip->i_gdquot = xfs_qm_dqhold(gdqp);
1974                 xfs_trans_mod_dquot(tp, gdqp, XFS_TRANS_DQ_ICOUNT, 1);
1975         }
1976         if (pdqp && XFS_IS_PQUOTA_ON(mp)) {
1977                 ASSERT(ip->i_pdquot == NULL);
1978                 ASSERT(xfs_get_projid(ip) == be32_to_cpu(pdqp->q_core.d_id));
1979
1980                 ip->i_pdquot = xfs_qm_dqhold(pdqp);
1981                 xfs_trans_mod_dquot(tp, pdqp, XFS_TRANS_DQ_ICOUNT, 1);
1982         }
1983 }
1984