lustre/quota/quota_context.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * GPL HEADER START
   5  *
   6  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 only,
  10  * as published by the Free Software Foundation.
  11  *
  12  * This program is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * General Public License version 2 for more details (a copy is included
  16  * in the LICENSE file that accompanied this code).
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * version 2 along with this program; If not, see
  20  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  21  *
  22  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  23  * CA 95054 USA or visit www.sun.com if you need additional information or
  24  * have any questions.
  25  *
  26  * GPL HEADER END
  27  */
  28 /*
  29  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
  30  * Use is subject to license terms.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * lustre/quota/quota_context.c
  37  *
  38  * Lustre Quota Context
  39  *
  40  * Author: Niu YaWei <niu@clusterfs.com>
  41  */
  42
  43 #ifndef EXPORT_SYMTAB
  44 # define EXPORT_SYMTAB
  45 #endif
  46
  47 #define DEBUG_SUBSYSTEM S_LQUOTA
  48
  49 #include <linux/version.h>
  50 #include <linux/fs.h>
  51 #include <asm/unistd.h>
  52 #include <linux/slab.h>
  53 #include <linux/quotaops.h>
  54 #include <linux/module.h>
  55 #include <linux/init.h>
  56
  57 #include <obd_class.h>
  58 #include <lustre_quota.h>
  59 #include <lustre_fsfilt.h>
  60 #include <lprocfs_status.h>
  61 #include "quota_internal.h"
  62
  63 #ifdef HAVE_QUOTA_SUPPORT
  64
  65 static cfs_hash_ops_t lqs_hash_ops;
  66
  67 unsigned long default_bunit_sz = 128 * 1024 * 1024; /* 128M bytes */
  68 unsigned long default_btune_ratio = 50;             /* 50 percentage */
  69 unsigned long default_iunit_sz = 5120;              /* 5120 inodes */
  70 unsigned long default_itune_ratio = 50;             /* 50 percentage */
  71
  72 cfs_mem_cache_t *qunit_cachep = NULL;
  73 struct list_head qunit_hash[NR_DQHASH];
  74 spinlock_t qunit_hash_lock = SPIN_LOCK_UNLOCKED;
  75
  76 /* please sync qunit_state with qunit_state_names */
  77 enum qunit_state {
  78         /**
  79          * a qunit is created
  80          */
  81         QUNIT_CREATED      = 0,
  82         /**
  83          * a qunit is added into qunit hash, that means
  84          * a quota req will be sent or is flying
  85          */
  86         QUNIT_IN_HASH      = 1,
  87         /**
  88          * a qunit is removed from qunit hash, that
  89          * means a quota req is handled and comes back
  90          */
  91         QUNIT_RM_FROM_HASH = 2,
  92         /**
  93          * qunit can wake up all threads waiting for it
  94          */
  95         QUNIT_FINISHED     = 3,
  96 };
  97
  98 static const char *qunit_state_names[] = {
  99         [QUNIT_CREATED]      = "CREATED",
 100         [QUNIT_IN_HASH]      = "IN_HASH",
 101         [QUNIT_RM_FROM_HASH] = "RM_FROM_HASH",
 102         [QUNIT_FINISHED]     = "FINISHED",
 103 };
 104
 105 struct lustre_qunit {
 106         struct list_head lq_hash;          /** Hash list in memory */
 107         atomic_t lq_refcnt;                /** Use count */
 108         struct lustre_quota_ctxt *lq_ctxt; /** Quota context this applies to */
 109         struct qunit_data lq_data;         /** See qunit_data */
 110         unsigned int lq_opc;               /** QUOTA_DQACQ, QUOTA_DQREL */
 111         cfs_waitq_t lq_waitq;              /** Threads waiting for this qunit */
 112         spinlock_t lq_lock;                /** Protect the whole structure */
 113         enum qunit_state lq_state;         /** Present the status of qunit */
 114         int lq_rc;                         /** The rc of lq_data */
 115         pid_t lq_owner;
 116 };
 117
 118 #define QUNIT_SET_STATE(qunit, state)                                   \
 119 do {                                                                    \
 120         spin_lock(&qunit->lq_lock);                                     \
 121         QDATA_DEBUG((&qunit->lq_data), "qunit(%p) lq_state(%s->%s), "   \
 122                     "lq_rc(%d), lq_owner(%d)\n",                        \
 123                     qunit, qunit_state_names[qunit->lq_state],          \
 124                     qunit_state_names[state], qunit->lq_rc,             \
 125                     qunit->lq_owner);                                   \
 126         qunit->lq_state = state;                                        \
 127         spin_unlock(&qunit->lq_lock);                                   \
 128 } while(0)
 129
 130 #define QUNIT_SET_STATE_AND_RC(qunit, state, rc)                        \
 131 do {                                                                    \
 132         spin_lock(&qunit->lq_lock);                                     \
 133         qunit->lq_rc = rc;                                              \
 134         QDATA_DEBUG((&qunit->lq_data), "qunit(%p) lq_state(%s->%s), "   \
 135                     "lq_rc(%d), lq_owner(%d)\n",                        \
 136                     qunit, qunit_state_names[qunit->lq_state],          \
 137                     qunit_state_names[state], qunit->lq_rc,             \
 138                     qunit->lq_owner);                                   \
 139         qunit->lq_state = state;                                        \
 140         spin_unlock(&qunit->lq_lock);                                   \
 141 } while(0)
 142
 143 int should_translate_quota (struct obd_import *imp)
 144 {
 145         ENTRY;
 146
 147         LASSERT(imp);
 148         if (imp->imp_connect_data.ocd_connect_flags & OBD_CONNECT_QUOTA64)
 149                 RETURN(0);
 150         else
 151                 RETURN(1);
 152 }
 153
 154 void qunit_cache_cleanup(void)
 155 {
 156         int i;
 157         ENTRY;
 158
 159         spin_lock(&qunit_hash_lock);
 160         for (i = 0; i < NR_DQHASH; i++)
 161                 LASSERT(list_empty(qunit_hash + i));
 162         spin_unlock(&qunit_hash_lock);
 163
 164         if (qunit_cachep) {
 165                 int rc;
 166                 rc = cfs_mem_cache_destroy(qunit_cachep);
 167                 LASSERTF(rc == 0, "couldn't destroy qunit_cache slab\n");
 168                 qunit_cachep = NULL;
 169         }
 170         EXIT;
 171 }
 172
 173 int qunit_cache_init(void)
 174 {
 175         int i;
 176         ENTRY;
 177
 178         LASSERT(qunit_cachep == NULL);
 179         qunit_cachep = cfs_mem_cache_create("ll_qunit_cache",
 180                                             sizeof(struct lustre_qunit),
 181                                             0, 0);
 182         if (!qunit_cachep)
 183                 RETURN(-ENOMEM);
 184
 185         spin_lock(&qunit_hash_lock);
 186         for (i = 0; i < NR_DQHASH; i++)
 187                 CFS_INIT_LIST_HEAD(qunit_hash + i);
 188         spin_unlock(&qunit_hash_lock);
 189         RETURN(0);
 190 }
 191
 192 static inline int
 193 qunit_hashfn(struct lustre_quota_ctxt *qctxt, struct qunit_data *qdata)
 194              __attribute__((__const__));
 195
 196 static inline int
 197 qunit_hashfn(struct lustre_quota_ctxt *qctxt, struct qunit_data *qdata)
 198 {
 199         unsigned int id = qdata->qd_id;
 200         unsigned int type = QDATA_IS_GRP(qdata);
 201
 202         unsigned long tmp = ((unsigned long)qctxt >> L1_CACHE_SHIFT) ^ id;
 203         tmp = (tmp * (MAXQUOTAS - type)) % NR_DQHASH;
 204         return tmp;
 205 }
 206
 207 /* caller must hold qunit_hash_lock */
 208 static inline struct lustre_qunit *find_qunit(unsigned int hashent,
 209                                               struct lustre_quota_ctxt *qctxt,
 210                                               struct qunit_data *qdata)
 211 {
 212         struct lustre_qunit *qunit = NULL;
 213         struct qunit_data *tmp;
 214
 215         LASSERT_SPIN_LOCKED(&qunit_hash_lock);
 216         list_for_each_entry(qunit, qunit_hash + hashent, lq_hash) {
 217                 tmp = &qunit->lq_data;
 218                 if (qunit->lq_ctxt == qctxt &&
 219                     qdata->qd_id == tmp->qd_id &&
 220                     (qdata->qd_flags & LQUOTA_QUNIT_FLAGS) ==
 221                     (tmp->qd_flags & LQUOTA_QUNIT_FLAGS))
 222                         return qunit;
 223         }
 224         return NULL;
 225 }
 226
 227 /* check_cur_qunit - check the current usage of qunit.
 228  * @qctxt: quota context
 229  * @qdata: the type of quota unit to be checked
 230  *
 231  * return: 1 - need acquire qunit;
 232  *         2 - need release qunit;
 233  *         0 - need do nothing.
 234  *       < 0 - error.
 235  */
 236 static int
 237 check_cur_qunit(struct obd_device *obd,
 238                 struct lustre_quota_ctxt *qctxt, struct qunit_data *qdata)
 239 {
 240         struct super_block *sb = qctxt->lqc_sb;
 241         unsigned long qunit_sz, tune_sz;
 242         __u64 usage, limit, limit_org, pending_write = 0;
 243         long long record = 0;
 244         struct obd_quotactl *qctl;
 245         struct lustre_qunit_size *lqs = NULL;
 246         int ret = 0;
 247         ENTRY;
 248
 249         if (!ll_sb_any_quota_active(sb))
 250                 RETURN(0);
 251
 252         spin_lock(&qctxt->lqc_lock);
 253         if (!qctxt->lqc_valid){
 254                 spin_unlock(&qctxt->lqc_lock);
 255                 RETURN(0);
 256         }
 257         spin_unlock(&qctxt->lqc_lock);
 258
 259         OBD_ALLOC_PTR(qctl);
 260         if (qctl == NULL)
 261                 RETURN(-ENOMEM);
 262
 263         /* get fs quota usage & limit */
 264         qctl->qc_cmd = Q_GETQUOTA;
 265         qctl->qc_id = qdata->qd_id;
 266         qctl->qc_type = QDATA_IS_GRP(qdata);
 267         ret = fsfilt_quotactl(obd, sb, qctl);
 268         if (ret) {
 269                 if (ret == -ESRCH)      /* no limit */
 270                         ret = 0;
 271                 else
 272                         CERROR("can't get fs quota usage! (rc:%d)\n", ret);
 273                 GOTO(out, ret);
 274         }
 275
 276         if (QDATA_IS_BLK(qdata)) {
 277                 usage = qctl->qc_dqblk.dqb_curspace;
 278                 limit = qctl->qc_dqblk.dqb_bhardlimit << QUOTABLOCK_BITS;
 279         } else {
 280                 usage = qctl->qc_dqblk.dqb_curinodes;
 281                 limit = qctl->qc_dqblk.dqb_ihardlimit;
 282         }
 283
 284         /* ignore the no quota limit case; and it can avoid creating
 285          * unnecessary lqs for uid/gid */
 286         if (!limit)
 287                 GOTO(out, ret = 0);
 288
 289         lqs = quota_search_lqs(LQS_KEY(QDATA_IS_GRP(qdata), qdata->qd_id),
 290                                qctxt, 0);
 291         if (IS_ERR(lqs) || lqs == NULL) {
 292                 CDEBUG(D_ERROR, "fail to find a lqs(%s id: %u)!\n",
 293                        QDATA_IS_GRP(qdata) ? "group" : "user", qdata->qd_id);
 294                 GOTO (out, ret = 0);
 295         }
 296         spin_lock(&lqs->lqs_lock);
 297
 298         if (QDATA_IS_BLK(qdata)) {
 299                 qunit_sz = lqs->lqs_bunit_sz;
 300                 tune_sz  = lqs->lqs_btune_sz;
 301                 pending_write = lqs->lqs_bwrite_pending;
 302                 record   = lqs->lqs_blk_rec;
 303                 LASSERT(!(qunit_sz % QUOTABLOCK_SIZE));
 304         } else {
 305                 /* we didn't need change inode qunit size now */
 306                 qunit_sz = lqs->lqs_iunit_sz;
 307                 tune_sz  = lqs->lqs_itune_sz;
 308                 pending_write = lqs->lqs_iwrite_pending;
 309                 record   = lqs->lqs_ino_rec;
 310         }
 311
 312         /* we don't count the MIN_QLIMIT */
 313         if ((limit == MIN_QLIMIT && !QDATA_IS_BLK(qdata)) ||
 314             (toqb(limit) == MIN_QLIMIT && QDATA_IS_BLK(qdata)))
 315                 limit = 0;
 316
 317         usage += pending_write;
 318         limit_org = limit;
 319         /* when a releasing quota req is sent, before it returned
 320            limit is assigned a small value. limit will overflow */
 321         if (limit + record < 0)
 322                 usage -= record;
 323         else
 324                 limit += record;
 325
 326         LASSERT(qdata->qd_count == 0);
 327         if (limit <= usage + tune_sz) {
 328                 while (qdata->qd_count + limit <=
 329                        usage + tune_sz)
 330                         qdata->qd_count += qunit_sz;
 331                 ret = 1;
 332         } else if (limit > usage + qunit_sz + tune_sz &&
 333                    limit_org > qdata->qd_count + qunit_sz) {
 334                 while (limit - qdata->qd_count > usage + qunit_sz + tune_sz &&
 335                        limit_org > qdata->qd_count + qunit_sz)
 336                         qdata->qd_count += qunit_sz;
 337                 ret = 2;
 338                 /* if there are other pending writes for this uid/gid, releasing
 339                  * quota is put off until the last pending write b=16645 */
 340                 if (ret == 2 && pending_write) {
 341                         CDEBUG(D_QUOTA, "delay quota release\n");
 342                         ret = 0;
 343                 }
 344         }
 345         CDEBUG(D_QUOTA, "type: %c, limit: "LPU64", usage: "LPU64
 346                ", pending_write: "LPU64", record: "LPD64
 347                ", qunit_sz: %lu, tune_sz: %lu, ret: %d.\n",
 348                QDATA_IS_BLK(qdata) ? 'b' : 'i', limit, usage, pending_write,
 349                record, qunit_sz, tune_sz, ret);
 350         LASSERT(ret == 0 || qdata->qd_count);
 351
 352         spin_unlock(&lqs->lqs_lock);
 353         lqs_putref(lqs);
 354
 355         EXIT;
 356  out:
 357         OBD_FREE_PTR(qctl);
 358         return ret;
 359 }
 360
 361 /**
 362  * Compute the remaining quota for certain gid or uid b=11693
 363  */
 364 int compute_remquota(struct obd_device *obd, struct lustre_quota_ctxt *qctxt,
 365                      struct qunit_data *qdata, int isblk)
 366 {
 367         struct super_block *sb = qctxt->lqc_sb;
 368         __u64 usage, limit;
 369         struct obd_quotactl *qctl;
 370         int ret = QUOTA_RET_OK;
 371         ENTRY;
 372
 373         if (!ll_sb_any_quota_active(sb))
 374                 RETURN(QUOTA_RET_NOQUOTA);
 375
 376         /* ignore root user */
 377         if (qdata->qd_id == 0 && QDATA_IS_GRP(qdata) == USRQUOTA)
 378                 RETURN(QUOTA_RET_NOLIMIT);
 379
 380         OBD_ALLOC_PTR(qctl);
 381         if (qctl == NULL)
 382                 RETURN(-ENOMEM);
 383
 384         /* get fs quota usage & limit */
 385         qctl->qc_cmd = Q_GETQUOTA;
 386         qctl->qc_id = qdata->qd_id;
 387         qctl->qc_type = QDATA_IS_GRP(qdata);
 388         ret = fsfilt_quotactl(obd, sb, qctl);
 389         if (ret) {
 390                 if (ret == -ESRCH)      /* no limit */
 391                         ret = QUOTA_RET_NOLIMIT;
 392                 else
 393                         CDEBUG(D_QUOTA, "can't get fs quota usage! (rc:%d)",
 394                                ret);
 395                 GOTO(out, ret);
 396         }
 397
 398         usage = isblk ? qctl->qc_dqblk.dqb_curspace :
 399                 qctl->qc_dqblk.dqb_curinodes;
 400         limit = isblk ? qctl->qc_dqblk.dqb_bhardlimit << QUOTABLOCK_BITS :
 401                 qctl->qc_dqblk.dqb_ihardlimit;
 402         if (!limit){            /* no limit */
 403                 ret = QUOTA_RET_NOLIMIT;
 404                 GOTO(out, ret);
 405         }
 406
 407         if (limit >= usage)
 408                 qdata->qd_count = limit - usage;
 409         else
 410                 qdata->qd_count = 0;
 411         EXIT;
 412 out:
 413         OBD_FREE_PTR(qctl);
 414         return ret;
 415 }
 416
 417 static struct lustre_qunit *alloc_qunit(struct lustre_quota_ctxt *qctxt,
 418                                         struct qunit_data *qdata, int opc)
 419 {
 420         struct lustre_qunit *qunit = NULL;
 421         ENTRY;
 422
 423         OBD_SLAB_ALLOC_PTR_GFP(qunit, qunit_cachep, CFS_ALLOC_IO);
 424         if (qunit == NULL)
 425                 RETURN(NULL);
 426
 427         CFS_INIT_LIST_HEAD(&qunit->lq_hash);
 428         init_waitqueue_head(&qunit->lq_waitq);
 429         atomic_set(&qunit->lq_refcnt, 1);
 430         qunit->lq_ctxt = qctxt;
 431         memcpy(&qunit->lq_data, qdata, sizeof(*qdata));
 432         qunit->lq_opc = opc;
 433         qunit->lq_lock = SPIN_LOCK_UNLOCKED;
 434         QUNIT_SET_STATE_AND_RC(qunit, QUNIT_CREATED, 0);
 435         qunit->lq_owner = cfs_curproc_pid();
 436         RETURN(qunit);
 437 }
 438
 439 static inline void free_qunit(struct lustre_qunit *qunit)
 440 {
 441         OBD_SLAB_FREE(qunit, qunit_cachep, sizeof(*qunit));
 442 }
 443
 444 static inline void qunit_get(struct lustre_qunit *qunit)
 445 {
 446         atomic_inc(&qunit->lq_refcnt);
 447 }
 448
 449 static void qunit_put(struct lustre_qunit *qunit)
 450 {
 451         LASSERT(atomic_read(&qunit->lq_refcnt));
 452         if (atomic_dec_and_test(&qunit->lq_refcnt))
 453                 free_qunit(qunit);
 454 }
 455
 456 /* caller must hold qunit_hash_lock and release ref of qunit after using it */
 457 static struct lustre_qunit *dqacq_in_flight(struct lustre_quota_ctxt *qctxt,
 458                                             struct qunit_data *qdata)
 459 {
 460         unsigned int hashent = qunit_hashfn(qctxt, qdata);
 461         struct lustre_qunit *qunit;
 462         ENTRY;
 463
 464         LASSERT_SPIN_LOCKED(&qunit_hash_lock);
 465         qunit = find_qunit(hashent, qctxt, qdata);
 466         if (qunit)
 467                 qunit_get(qunit);
 468         RETURN(qunit);
 469 }
 470
 471 static void
 472 insert_qunit_nolock(struct lustre_quota_ctxt *qctxt, struct lustre_qunit *qunit)
 473 {
 474         struct list_head *head;
 475
 476         LASSERT(list_empty(&qunit->lq_hash));
 477         qunit_get(qunit);
 478         head = qunit_hash + qunit_hashfn(qctxt, &qunit->lq_data);
 479         list_add(&qunit->lq_hash, head);
 480         QUNIT_SET_STATE(qunit, QUNIT_IN_HASH);
 481 }
 482
 483 static void compute_lqs_after_removing_qunit(struct lustre_qunit *qunit)
 484 {
 485         struct lustre_qunit_size *lqs;
 486
 487         lqs = quota_search_lqs(LQS_KEY(QDATA_IS_GRP(&qunit->lq_data),
 488                                        qunit->lq_data.qd_id),
 489                                qunit->lq_ctxt, 0);
 490         if (lqs && !IS_ERR(lqs)) {
 491                 spin_lock(&lqs->lqs_lock);
 492                 if (qunit->lq_opc == QUOTA_DQACQ)
 493                         quota_compute_lqs(&qunit->lq_data, lqs, 0, 1);
 494                 if (qunit->lq_opc == QUOTA_DQREL)
 495                         quota_compute_lqs(&qunit->lq_data, lqs, 0, 0);
 496                 spin_unlock(&lqs->lqs_lock);
 497                 /* this is for quota_search_lqs */
 498                 lqs_putref(lqs);
 499                 /* this is for schedule_dqacq */
 500                 lqs_putref(lqs);
 501         }
 502 }
 503
 504 static void remove_qunit_nolock(struct lustre_qunit *qunit)
 505 {
 506         LASSERT(!list_empty(&qunit->lq_hash));
 507         LASSERT_SPIN_LOCKED(&qunit_hash_lock);
 508
 509         list_del_init(&qunit->lq_hash);
 510         QUNIT_SET_STATE(qunit, QUNIT_RM_FROM_HASH);
 511         qunit_put(qunit);
 512 }
 513
 514 void* quota_barrier(struct lustre_quota_ctxt *qctxt,
 515                     struct obd_quotactl *oqctl, int isblk)
 516 {
 517         struct lustre_qunit *qunit, *find_qunit;
 518         int cycle = 1;
 519
 520         OBD_SLAB_ALLOC(qunit, qunit_cachep, CFS_ALLOC_IO, sizeof(*qunit));
 521         if (qunit == NULL) {
 522                 CERROR("locating qunit failed.(id=%u isblk=%d %s)\n",
 523                        oqctl->qc_id, isblk, oqctl->qc_type ? "grp" : "usr");
 524                 qctxt_wait_pending_dqacq(qctxt, oqctl->qc_id,
 525                                          oqctl->qc_type, isblk);
 526                 return NULL;
 527         }
 528
 529         INIT_LIST_HEAD(&qunit->lq_hash);
 530         qunit->lq_lock = SPIN_LOCK_UNLOCKED;
 531         init_waitqueue_head(&qunit->lq_waitq);
 532         atomic_set(&qunit->lq_refcnt, 1);
 533         qunit->lq_ctxt = qctxt;
 534         qunit->lq_data.qd_id = oqctl->qc_id;
 535         qunit->lq_data.qd_flags =  oqctl->qc_type;
 536         if (isblk)
 537                 QDATA_SET_BLK(&qunit->lq_data);
 538         QUNIT_SET_STATE_AND_RC(qunit, QUNIT_CREATED, 0);
 539         /* it means it is only an invalid qunit for barrier */
 540         qunit->lq_opc = QUOTA_LAST_OPC;
 541
 542         while (1) {
 543                 spin_lock(&qunit_hash_lock);
 544                 find_qunit = dqacq_in_flight(qctxt, &qunit->lq_data);
 545                 if (find_qunit) {
 546                         spin_unlock(&qunit_hash_lock);
 547                         qunit_put(find_qunit);
 548                         qctxt_wait_pending_dqacq(qctxt, oqctl->qc_id,
 549                                                  oqctl->qc_type, isblk);
 550                         CDEBUG(D_QUOTA, "cycle=%d\n", cycle++);
 551                         continue;
 552                 }
 553                 break;
 554         }
 555         insert_qunit_nolock(qctxt, qunit);
 556         spin_unlock(&qunit_hash_lock);
 557         return qunit;
 558 }
 559
 560 void quota_unbarrier(void *handle)
 561 {
 562         struct lustre_qunit *qunit = (struct lustre_qunit *)handle;
 563
 564         if (qunit == NULL) {
 565                 CERROR("handle is NULL\n");
 566                 return;
 567         }
 568
 569         LASSERT(qunit->lq_opc == QUOTA_LAST_OPC);
 570         spin_lock(&qunit_hash_lock);
 571         remove_qunit_nolock(qunit);
 572         spin_unlock(&qunit_hash_lock);
 573         QUNIT_SET_STATE_AND_RC(qunit, QUNIT_FINISHED, QUOTA_REQ_RETURNED);
 574         wake_up(&qunit->lq_waitq);
 575         qunit_put(qunit);
 576 }
 577
 578 #define INC_QLIMIT(limit, count) (limit == MIN_QLIMIT) ? \
 579                                  (limit = count) : (limit += count)
 580
 581
 582 static inline int is_master(struct lustre_quota_ctxt *qctxt)
 583 {
 584         return qctxt->lqc_handler ? 1 : 0;
 585 }
 586
 587 static int
 588 schedule_dqacq(struct obd_device *obd, struct lustre_quota_ctxt *qctxt,
 589                struct qunit_data *qdata, int opc, int wait,
 590                struct obd_trans_info *oti);
 591
 592 static inline void qdata_to_oqaq(struct qunit_data *qdata,
 593                                  struct quota_adjust_qunit *oqaq)
 594 {
 595         LASSERT(qdata);
 596         LASSERT(oqaq);
 597
 598         oqaq->qaq_flags = qdata->qd_flags;
 599         oqaq->qaq_id    = qdata->qd_id;
 600         if (QDATA_IS_ADJBLK(qdata))
 601                 oqaq->qaq_bunit_sz = qdata->qd_qunit;
 602         if (QDATA_IS_ADJINO(qdata))
 603                 oqaq->qaq_iunit_sz = qdata->qd_qunit;
 604 }
 605
 606 static int
 607 dqacq_completion(struct obd_device *obd, struct lustre_quota_ctxt *qctxt,
 608                  struct qunit_data *qdata, int rc, int opc)
 609 {
 610         struct lustre_qunit *qunit = NULL;
 611         struct super_block *sb = qctxt->lqc_sb;
 612         int err = 0;
 613         struct quota_adjust_qunit *oqaq = NULL;
 614         int rc1 = 0;
 615         ENTRY;
 616
 617         LASSERT(qdata);
 618         QDATA_DEBUG(qdata, "obd(%s): complete %s quota req\n",
 619                     obd->obd_name, (opc == QUOTA_DQACQ) ? "acq" : "rel");
 620
 621         /* do it only when a releasing quota req more than 5MB b=18491 */
 622         if (opc == QUOTA_DQREL && qdata->qd_count >= 5242880)
 623                 OBD_FAIL_TIMEOUT(OBD_FAIL_QUOTA_DELAY_REL, 5);
 624
 625         /* update local operational quota file */
 626         if (rc == 0) {
 627                 __u64 count = QUSG(qdata->qd_count, QDATA_IS_BLK(qdata));
 628                 struct obd_quotactl *qctl;
 629                 __u64 *hardlimit;
 630
 631                 OBD_ALLOC_PTR(qctl);
 632                 if (qctl == NULL)
 633                         GOTO(out, err = -ENOMEM);
 634
 635                 /* acq/rel qunit for specified uid/gid is serialized,
 636                  * so there is no race between get fs quota limit and
 637                  * set fs quota limit */
 638                 qctl->qc_cmd = Q_GETQUOTA;
 639                 qctl->qc_id = qdata->qd_id;
 640                 qctl->qc_type = QDATA_IS_GRP(qdata);
 641                 err = fsfilt_quotactl(obd, sb, qctl);
 642                 if (err) {
 643                         CERROR("error get quota fs limit! (rc:%d)\n", err);
 644                         GOTO(out_mem, err);
 645                 }
 646
 647                 if (QDATA_IS_BLK(qdata)) {
 648                         qctl->qc_dqblk.dqb_valid = QIF_BLIMITS;
 649                         hardlimit = &qctl->qc_dqblk.dqb_bhardlimit;
 650                 } else {
 651                         qctl->qc_dqblk.dqb_valid = QIF_ILIMITS;
 652                         hardlimit = &qctl->qc_dqblk.dqb_ihardlimit;
 653                 }
 654
 655                 CDEBUG(D_QUOTA, "hardlimt: "LPU64"\n", *hardlimit);
 656
 657                 if (*hardlimit == 0)
 658                         goto out_mem;
 659
 660                 switch (opc) {
 661                 case QUOTA_DQACQ:
 662                         INC_QLIMIT(*hardlimit, count);
 663                         break;
 664                 case QUOTA_DQREL:
 665                         LASSERTF(count < *hardlimit,
 666                                  "id(%u) flag(%u) type(%c) isblk(%c) "
 667                                  "count("LPU64") qd_qunit("LPU64") "
 668                                  "hardlimit("LPU64").\n",
 669                                  qdata->qd_id, qdata->qd_flags,
 670                                  QDATA_IS_GRP(qdata) ? 'g' : 'u',
 671                                  QDATA_IS_BLK(qdata) ? 'b': 'i',
 672                                  qdata->qd_count, qdata->qd_qunit, *hardlimit);
 673                         *hardlimit -= count;
 674                         break;
 675                 default:
 676                         LBUG();
 677                 }
 678
 679                 /* clear quota limit */
 680                 if (count == 0)
 681                         *hardlimit = 0;
 682
 683                 qctl->qc_cmd = Q_SETQUOTA;
 684                 err = fsfilt_quotactl(obd, sb, qctl);
 685                 if (err)
 686                         CERROR("error set quota fs limit! (rc:%d)\n", err);
 687
 688                 QDATA_DEBUG(qdata, "%s completion\n",
 689                             opc == QUOTA_DQACQ ? "DQACQ" : "DQREL");
 690 out_mem:
 691                 OBD_FREE_PTR(qctl);
 692         } else if (rc == -EDQUOT) {
 693                 QDATA_DEBUG(qdata, "acquire qunit got EDQUOT.\n");
 694         } else if (rc == -EBUSY) {
 695                 QDATA_DEBUG(qdata, "it's is recovering, got EBUSY.\n");
 696         } else {
 697                 CERROR("acquire qunit got error! (rc:%d)\n", rc);
 698         }
 699 out:
 700         /* remove the qunit from hash */
 701         spin_lock(&qunit_hash_lock);
 702
 703         qunit = dqacq_in_flight(qctxt, qdata);
 704         /* this qunit has been removed by qctxt_cleanup() */
 705         if (!qunit) {
 706                 spin_unlock(&qunit_hash_lock);
 707                 QDATA_DEBUG(qdata, "%s is discarded because qunit isn't found\n",
 708                             opc == QUOTA_DQACQ ? "DQACQ" : "DQREL");
 709                 RETURN(err);
 710         }
 711
 712         LASSERT(opc == qunit->lq_opc);
 713         /* remove this qunit from lq_hash so that new processes cannot be added
 714          * to qunit->lq_waiters */
 715         remove_qunit_nolock(qunit);
 716         spin_unlock(&qunit_hash_lock);
 717
 718         compute_lqs_after_removing_qunit(qunit);
 719
 720         if (rc == 0)
 721                 rc = QUOTA_REQ_RETURNED;
 722         QUNIT_SET_STATE_AND_RC(qunit, QUNIT_FINISHED, rc);
 723         /* wake up all waiters */
 724         wake_up_all(&qunit->lq_waitq);
 725
 726         /* this is for dqacq_in_flight() */
 727         qunit_put(qunit);
 728         /* this is for alloc_qunit() */
 729         qunit_put(qunit);
 730         if (rc < 0 && rc != -EDQUOT)
 731                  RETURN(err);
 732
 733         /* don't reschedule in such cases:
 734          *   - acq/rel failure and qunit isn't changed,
 735          *     but not for quota recovery.
 736          *   - local dqacq/dqrel.
 737          *   - local disk io failure.
 738          */
 739          OBD_ALLOC_PTR(oqaq);
 740          if (!oqaq)
 741                  RETURN(-ENOMEM);
 742          qdata_to_oqaq(qdata, oqaq);
 743          /* adjust the qunit size in slaves */
 744          rc1 = quota_adjust_slave_lqs(oqaq, qctxt);
 745          OBD_FREE_PTR(oqaq);
 746          if (rc1 < 0) {
 747                  CERROR("adjust slave's qunit size failed!(rc:%d)\n", rc1);
 748                  RETURN(rc1);
 749          }
 750          if (err || (rc < 0 && rc != -EBUSY && rc1 == 0) || is_master(qctxt))
 751                  RETURN(err);
 752
 753          if (opc == QUOTA_DQREL && qdata->qd_count >= 5242880 &&
 754              OBD_FAIL_CHECK(OBD_FAIL_QUOTA_DELAY_REL))
 755                  RETURN(err);
 756
 757         /* reschedule another dqacq/dqrel if needed */
 758         qdata->qd_count = 0;
 759         qdata->qd_flags &= LQUOTA_QUNIT_FLAGS;
 760         rc1 = check_cur_qunit(obd, qctxt, qdata);
 761         if (rc1 > 0) {
 762                 int opc;
 763                 opc = rc1 == 1 ? QUOTA_DQACQ : QUOTA_DQREL;
 764                 rc1 = schedule_dqacq(obd, qctxt, qdata, opc, 0, NULL);
 765                 QDATA_DEBUG(qdata, "reschedudle opc(%d) rc(%d)\n", opc, rc1);
 766         }
 767         RETURN(err);
 768 }
 769
 770 struct dqacq_async_args {
 771         struct lustre_quota_ctxt *aa_ctxt;
 772         struct lustre_qunit *aa_qunit;
 773 };
 774
 775 static int dqacq_interpret(const struct lu_env *env,
 776                            struct ptlrpc_request *req, void *data, int rc)
 777 {
 778         struct dqacq_async_args *aa = (struct dqacq_async_args *)data;
 779         struct lustre_quota_ctxt *qctxt = aa->aa_ctxt;
 780         struct obd_device_target *obt = qctxt->lqc_obt;
 781         struct lustre_qunit *qunit = aa->aa_qunit;
 782         struct obd_device *obd = req->rq_import->imp_obd;
 783         struct qunit_data *qdata = NULL;
 784         ENTRY;
 785
 786         LASSERT(req);
 787         LASSERT(req->rq_import);
 788
 789         down_read(&obt->obt_rwsem);
 790         /* if a quota req timeouts or is dropped, we should update quota
 791          * statistics which will be handled in dqacq_completion. And in
 792          * this situation we should get qdata from request instead of
 793          * reply */
 794         qdata = quota_get_qdata(req, (rc != 0) ? QUOTA_REQUEST : QUOTA_REPLY,
 795                                 QUOTA_IMPORT);
 796         if (IS_ERR(qdata)) {
 797                 rc = PTR_ERR(qdata);
 798                 DEBUG_REQ(D_ERROR, req,
 799                           "error unpacking qunit_data(rc: %ld)\n",
 800                           PTR_ERR(qdata));
 801                 RETURN(PTR_ERR(qdata));
 802         }
 803
 804         QDATA_DEBUG(qdata, "qdata: interpret rc(%d).\n", rc);
 805         QDATA_DEBUG((&qunit->lq_data), "lq_data: \n");
 806
 807         if (qdata->qd_id != qunit->lq_data.qd_id ||
 808             OBD_FAIL_CHECK(OBD_FAIL_QUOTA_RET_QDATA)) {
 809                 CDEBUG(D_ERROR, "the returned qd_id isn't expected!"
 810                        "(qdata: %u, lq_data: %u)\n", qdata->qd_id,
 811                        qunit->lq_data.qd_id);
 812                 qdata->qd_id = qunit->lq_data.qd_id;
 813                 rc = -EPROTO;
 814         }
 815         if (QDATA_IS_GRP(qdata) != QDATA_IS_GRP(&qunit->lq_data)) {
 816                 CDEBUG(D_ERROR, "the returned grp/usr isn't expected!"
 817                        "(qdata: %u, lq_data: %u)\n", qdata->qd_flags,
 818                        qunit->lq_data.qd_flags);
 819                 if (QDATA_IS_GRP(&qunit->lq_data))
 820                         QDATA_SET_GRP(qdata);
 821                 else
 822                         QDATA_CLR_GRP(qdata);
 823                 rc = -EPROTO;
 824         }
 825         if (qdata->qd_count > qunit->lq_data.qd_count) {
 826                 CDEBUG(D_ERROR, "the returned qd_count isn't expected!"
 827                        "(qdata: "LPU64", lq_data: "LPU64")\n", qdata->qd_count,
 828                        qunit->lq_data.qd_count);
 829                 rc = -EPROTO;
 830         }
 831
 832         if (unlikely(rc == -ESRCH))
 833                 CERROR("quota for %s has been enabled by master, but disabled "
 834                        "by slave.\n", QDATA_IS_GRP(qdata) ? "group" : "user");
 835
 836         rc = dqacq_completion(obd, qctxt, qdata, rc,
 837                               lustre_msg_get_opc(req->rq_reqmsg));
 838
 839         up_read(&obt->obt_rwsem);
 840         RETURN(rc);
 841 }
 842
 843 /**
 844  * check if quota master is online
 845  */
 846 int check_qm(struct lustre_quota_ctxt *qctxt)
 847 {
 848         int rc;
 849         ENTRY;
 850
 851         spin_lock(&qctxt->lqc_lock);
 852         /* quit waiting when mds is back or qctxt is cleaned up */
 853         rc = qctxt->lqc_import || !qctxt->lqc_valid;
 854         spin_unlock(&qctxt->lqc_lock);
 855
 856         RETURN(rc);
 857 }
 858
 859 /* wake up all waiting threads when lqc_import is NULL */
 860 void dqacq_interrupt(struct lustre_quota_ctxt *qctxt)
 861 {
 862         struct lustre_qunit *qunit, *tmp;
 863         int i;
 864         ENTRY;
 865
 866         spin_lock(&qunit_hash_lock);
 867         for (i = 0; i < NR_DQHASH; i++) {
 868                 list_for_each_entry_safe(qunit, tmp, &qunit_hash[i], lq_hash) {
 869                         if (qunit->lq_ctxt != qctxt)
 870                                 continue;
 871
 872                         /* Wake up all waiters. Do not change lq_state.
 873                          * The waiters will check lq_rc which is kept as 0
 874                          * if no others change it, then the waiters will return
 875                          * -EAGAIN to caller who can perform related quota
 876                          * acq/rel if necessary. */
 877                         wake_up_all(&qunit->lq_waitq);
 878                 }
 879         }
 880         spin_unlock(&qunit_hash_lock);
 881         EXIT;
 882 }
 883
 884 static int got_qunit(struct lustre_qunit *qunit, int is_master)
 885 {
 886         struct lustre_quota_ctxt *qctxt = qunit->lq_ctxt;
 887         int rc = 0;
 888         ENTRY;
 889
 890         spin_lock(&qunit->lq_lock);
 891         switch (qunit->lq_state) {
 892         case QUNIT_IN_HASH:
 893         case QUNIT_RM_FROM_HASH:
 894                 break;
 895         case QUNIT_FINISHED:
 896                 rc = 1;
 897                 break;
 898         default:
 899                 CERROR("invalid qunit state %d\n", qunit->lq_state);
 900         }
 901         spin_unlock(&qunit->lq_lock);
 902
 903         if (!rc) {
 904                 spin_lock(&qctxt->lqc_lock);
 905                 rc = !qctxt->lqc_valid;
 906                 if (!is_master)
 907                         rc |= !qctxt->lqc_import;
 908                 spin_unlock(&qctxt->lqc_lock);
 909         }
 910
 911         RETURN(rc);
 912 }
 913
 914 static int
 915 schedule_dqacq(struct obd_device *obd, struct lustre_quota_ctxt *qctxt,
 916                struct qunit_data *qdata, int opc, int wait,
 917                struct obd_trans_info *oti)
 918 {
 919         struct lustre_qunit *qunit, *empty;
 920         struct l_wait_info lwi = { 0 };
 921         struct ptlrpc_request *req;
 922         struct dqacq_async_args *aa;
 923         struct obd_import *imp = NULL;
 924         struct lustre_qunit_size *lqs = NULL;
 925         struct timeval work_start;
 926         struct timeval work_end;
 927         long timediff;
 928         int rc = 0;
 929         ENTRY;
 930
 931         LASSERT(opc == QUOTA_DQACQ || opc == QUOTA_DQREL);
 932         do_gettimeofday(&work_start);
 933         if ((empty = alloc_qunit(qctxt, qdata, opc)) == NULL)
 934                 RETURN(-ENOMEM);
 935
 936         spin_lock(&qunit_hash_lock);
 937         qunit = dqacq_in_flight(qctxt, qdata);
 938         if (qunit) {
 939                 spin_unlock(&qunit_hash_lock);
 940                 qunit_put(empty);
 941
 942                 goto wait_completion;
 943         }
 944         qunit = empty;
 945         qunit_get(qunit);
 946         insert_qunit_nolock(qctxt, qunit);
 947         spin_unlock(&qunit_hash_lock);
 948
 949         lqs = quota_search_lqs(LQS_KEY(QDATA_IS_GRP(qdata), qdata->qd_id),
 950                                qctxt, 0);
 951         if (lqs && !IS_ERR(lqs)) {
 952                 spin_lock(&lqs->lqs_lock);
 953                 quota_compute_lqs(qdata, lqs, 1, (opc == QUOTA_DQACQ) ? 1 : 0);
 954                 /* when this qdata returned from mds, it will call lqs_putref */
 955                 lqs_getref(lqs);
 956                 spin_unlock(&lqs->lqs_lock);
 957                 /* this is for quota_search_lqs */
 958                 lqs_putref(lqs);
 959         } else {
 960                 CDEBUG(D_ERROR, "Can't find the lustre qunit size!\n");
 961         }
 962
 963         QDATA_DEBUG(qdata, "obd(%s): send %s quota req\n",
 964                     obd->obd_name, (opc == QUOTA_DQACQ) ? "acq" : "rel");
 965         /* master is going to dqacq/dqrel from itself */
 966         if (is_master(qctxt)) {
 967                 int rc2;
 968                 QDATA_DEBUG(qdata, "local %s.\n",
 969                             opc == QUOTA_DQACQ ? "DQACQ" : "DQREL");
 970                 QDATA_SET_CHANGE_QS(qdata);
 971                 rc = qctxt->lqc_handler(obd, qdata, opc);
 972                 rc2 = dqacq_completion(obd, qctxt, qdata, rc, opc);
 973                 /* this is for qunit_get() */
 974                 qunit_put(qunit);
 975
 976                 do_gettimeofday(&work_end);
 977                 timediff = cfs_timeval_sub(&work_end, &work_start, NULL);
 978                 if (opc == QUOTA_DQACQ)
 979                         lprocfs_counter_add(qctxt->lqc_stats,
 980                                             wait ? LQUOTA_SYNC_ACQ : LQUOTA_ASYNC_ACQ,
 981                                             timediff);
 982                 else
 983                         lprocfs_counter_add(qctxt->lqc_stats,
 984                                             wait ? LQUOTA_SYNC_REL : LQUOTA_ASYNC_REL,
 985                                             timediff);
 986                 RETURN(rc ? rc : rc2);
 987         }
 988
 989         spin_lock(&qctxt->lqc_lock);
 990         if (!qctxt->lqc_import) {
 991                 spin_unlock(&qctxt->lqc_lock);
 992                 QDATA_DEBUG(qdata, "lqc_import is invalid.\n");
 993
 994                 spin_lock(&qunit_hash_lock);
 995                 remove_qunit_nolock(qunit);
 996                 spin_unlock(&qunit_hash_lock);
 997
 998                 compute_lqs_after_removing_qunit(qunit);
 999
1000                 QUNIT_SET_STATE_AND_RC(qunit, QUNIT_FINISHED, -EAGAIN);
1001                 wake_up_all(&qunit->lq_waitq);
1002
1003                 /* this is for qunit_get() */
1004                 qunit_put(qunit);
1005                 /* this for alloc_qunit() */
1006                 qunit_put(qunit);
1007                 spin_lock(&qctxt->lqc_lock);
1008                 if (wait && !qctxt->lqc_import) {
1009                         spin_unlock(&qctxt->lqc_lock);
1010
1011                         LASSERT(oti && oti->oti_thread &&
1012                                 oti->oti_thread->t_watchdog);
1013
1014                         lc_watchdog_disable(oti->oti_thread->t_watchdog);
1015                         CDEBUG(D_QUOTA, "sleep for quota master\n");
1016                         l_wait_event(qctxt->lqc_wait_for_qmaster,
1017                                      check_qm(qctxt), &lwi);
1018                         CDEBUG(D_QUOTA, "wake up when quota master is back\n");
1019                         lc_watchdog_touch(oti->oti_thread->t_watchdog,
1020                                  GET_TIMEOUT(oti->oti_thread->t_svc));
1021                 } else {
1022                         spin_unlock(&qctxt->lqc_lock);
1023                 }
1024
1025                 RETURN(-EAGAIN);
1026         }
1027         imp = class_import_get(qctxt->lqc_import);
1028         spin_unlock(&qctxt->lqc_lock);
1029
1030         /* build dqacq/dqrel request */
1031         LASSERT(imp);
1032
1033         req = ptlrpc_request_alloc_pack(imp, &RQF_MDS_QUOTA_DQACQ,
1034                                         LUSTRE_MDS_VERSION, opc);
1035         class_import_put(imp);
1036         if (req == NULL) {
1037                 CDEBUG(D_ERROR, "Can't alloc request\n");
1038                 dqacq_completion(obd, qctxt, qdata, -ENOMEM, opc);
1039                 /* this is for qunit_get() */
1040                 qunit_put(qunit);
1041                 RETURN(-ENOMEM);
1042         }
1043
1044         ptlrpc_request_set_replen(req);
1045         req->rq_no_resend = req->rq_no_delay = 1;
1046         rc = quota_copy_qdata(req, qdata, QUOTA_REQUEST, QUOTA_IMPORT);
1047         if (rc < 0) {
1048                 CDEBUG(D_ERROR, "Can't pack qunit_data(rc: %d)\n", rc);
1049                 ptlrpc_req_finished(req);
1050                 dqacq_completion(obd, qctxt, qdata, -EPROTO, opc);
1051                 /* this is for qunit_get() */
1052                 qunit_put(qunit);
1053                 RETURN(rc);
1054         }
1055
1056         CLASSERT(sizeof(*aa) <= sizeof(req->rq_async_args));
1057         aa = ptlrpc_req_async_args(req);
1058         aa->aa_ctxt = qctxt;
1059         aa->aa_qunit = qunit;
1060
1061         req->rq_interpret_reply = dqacq_interpret;
1062         ptlrpcd_add_req(req, PSCOPE_OTHER);
1063
1064         QDATA_DEBUG(qdata, "%s scheduled.\n",
1065                     opc == QUOTA_DQACQ ? "DQACQ" : "DQREL");
1066 wait_completion:
1067         if (wait && qunit) {
1068                 struct qunit_data *p = &qunit->lq_data;
1069
1070                 QDATA_DEBUG(p, "qunit(%p) is waiting for dqacq.\n", qunit);
1071                 l_wait_event(qunit->lq_waitq, got_qunit(qunit, is_master(qctxt)),
1072                              &lwi);
1073                 /* rc = -EAGAIN, it means the quota master isn't ready yet
1074                  * rc = QUOTA_REQ_RETURNED, it means a quota req is finished;
1075                  * rc = -EDQUOT, it means out of quota
1076                  * rc = -EBUSY, it means recovery is happening
1077                  * other rc < 0, it means real errors, functions who call
1078                  * schedule_dqacq should take care of this */
1079                 spin_lock(&qunit->lq_lock);
1080                 rc = qunit->lq_rc;
1081                 spin_unlock(&qunit->lq_lock);
1082                 CDEBUG(D_QUOTA, "qunit(%p) finishes waiting: id(%u) flag(%u) "
1083                        "rc(%d) owner(%d)\n", qunit, qunit->lq_data.qd_id,
1084                        qunit->lq_data.qd_flags, rc, qunit->lq_owner);
1085         }
1086
1087         qunit_put(qunit);
1088         do_gettimeofday(&work_end);
1089         timediff = cfs_timeval_sub(&work_end, &work_start, NULL);
1090         if (opc == QUOTA_DQACQ)
1091                 lprocfs_counter_add(qctxt->lqc_stats,
1092                                     wait ? LQUOTA_SYNC_ACQ : LQUOTA_ASYNC_ACQ,
1093                                     timediff);
1094         else
1095                 lprocfs_counter_add(qctxt->lqc_stats,
1096                                     wait ? LQUOTA_SYNC_REL : LQUOTA_ASYNC_REL,
1097                                     timediff);
1098
1099         RETURN(rc);
1100 }
1101
1102 int
1103 qctxt_adjust_qunit(struct obd_device *obd, struct lustre_quota_ctxt *qctxt,
1104                    const unsigned int id[], __u32 isblk, int wait,
1105                    struct obd_trans_info *oti)
1106 {
1107         int rc = 0, i = USRQUOTA;
1108         struct qunit_data qdata[MAXQUOTAS];
1109         ENTRY;
1110
1111         if (quota_is_set(obd, id, isblk ? QB_SET : QI_SET) == 0)
1112                 RETURN(0);
1113
1114         for (i = 0; i < MAXQUOTAS; i++) {
1115                 qdata[i].qd_id = id[i];
1116                 qdata[i].qd_flags = i;
1117                 if (isblk)
1118                         QDATA_SET_BLK(&qdata[i]);
1119                 qdata[i].qd_count = 0;
1120
1121                 rc = check_cur_qunit(obd, qctxt, &qdata[i]);
1122                 if (rc > 0) {
1123                         int opc;
1124                         /* need acquire or release */
1125                         opc = rc == 1 ? QUOTA_DQACQ : QUOTA_DQREL;
1126                         rc = schedule_dqacq(obd, qctxt, &qdata[i], opc,
1127                                             wait,oti);
1128                         if (rc < 0)
1129                                 RETURN(rc);
1130                 } else if (wait == 1) {
1131                         /* when wait equates 1, that means mds_quota_acquire
1132                          * or filter_quota_acquire is calling it. */
1133                         rc = qctxt_wait_pending_dqacq(qctxt, id[i], i, isblk);
1134                         if (rc < 0)
1135                                 RETURN(rc);
1136                 }
1137         }
1138
1139         RETURN(rc);
1140 }
1141
1142 int
1143 qctxt_wait_pending_dqacq(struct lustre_quota_ctxt *qctxt, unsigned int id,
1144                          unsigned short type, int isblk)
1145 {
1146         struct lustre_qunit *qunit = NULL;
1147         struct qunit_data qdata;
1148         struct timeval work_start;
1149         struct timeval work_end;
1150         long timediff;
1151         struct l_wait_info lwi = { 0 };
1152         int rc = 0;
1153         ENTRY;
1154
1155         do_gettimeofday(&work_start);
1156         qdata.qd_id = id;
1157         qdata.qd_flags = type;
1158         if (isblk)
1159                 QDATA_SET_BLK(&qdata);
1160         qdata.qd_count = 0;
1161
1162         spin_lock(&qunit_hash_lock);
1163         qunit = dqacq_in_flight(qctxt, &qdata);
1164         spin_unlock(&qunit_hash_lock);
1165
1166         if (qunit) {
1167                 struct qunit_data *p = &qunit->lq_data;
1168
1169                 QDATA_DEBUG(p, "qunit(%p) is waiting for dqacq.\n", qunit);
1170                 l_wait_event(qunit->lq_waitq, got_qunit(qunit, is_master(qctxt)),
1171                              &lwi);
1172                 CDEBUG(D_QUOTA, "qunit(%p) finishes waiting: rc(%d) "
1173                        "owner(%d)\n", qunit, qunit->lq_rc, qunit->lq_owner);
1174                 /* keep same as schedule_dqacq() b=17030 */
1175                 spin_lock(&qunit->lq_lock);
1176                 rc = qunit->lq_rc;
1177                 spin_unlock(&qunit->lq_lock);
1178                 /* this is for dqacq_in_flight() */
1179                 qunit_put(qunit);
1180                 do_gettimeofday(&work_end);
1181                 timediff = cfs_timeval_sub(&work_end, &work_start, NULL);
1182                 lprocfs_counter_add(qctxt->lqc_stats,
1183                                     isblk ? LQUOTA_WAIT_PENDING_BLK_QUOTA :
1184                                             LQUOTA_WAIT_PENDING_INO_QUOTA,
1185                                     timediff);
1186         } else {
1187                 do_gettimeofday(&work_end);
1188                 timediff = cfs_timeval_sub(&work_end, &work_start, NULL);
1189                 lprocfs_counter_add(qctxt->lqc_stats,
1190                                     isblk ? LQUOTA_NOWAIT_PENDING_BLK_QUOTA :
1191                                             LQUOTA_NOWAIT_PENDING_INO_QUOTA,
1192                                     timediff);
1193         }
1194
1195         RETURN(rc);
1196 }
1197
1198 int
1199 qctxt_init(struct obd_device *obd, dqacq_handler_t handler)
1200 {
1201         struct lustre_quota_ctxt *qctxt = &obd->u.obt.obt_qctxt;
1202         struct obd_device_target *obt = &obd->u.obt;
1203         struct super_block *sb = obt->obt_sb;
1204         int rc = 0;
1205         ENTRY;
1206
1207         LASSERT(qctxt);
1208
1209         rc = ptlrpcd_addref();
1210         if (rc)
1211                 RETURN(rc);
1212
1213         cfs_waitq_init(&qctxt->lqc_wait_for_qmaster);
1214         cfs_waitq_init(&qctxt->lqc_lqs_waitq);
1215         atomic_set(&qctxt->lqc_lqs, 0);
1216         spin_lock_init(&qctxt->lqc_lock);
1217         spin_lock(&qctxt->lqc_lock);
1218         qctxt->lqc_handler = handler;
1219         qctxt->lqc_sb = sb;
1220         qctxt->lqc_obt = obt;
1221         qctxt->lqc_import = NULL;
1222         qctxt->lqc_recovery = 0;
1223         qctxt->lqc_switch_qs = 1; /* Change qunit size in default setting */
1224         qctxt->lqc_valid = 1;
1225         qctxt->lqc_cqs_boundary_factor = 4;
1226         qctxt->lqc_cqs_least_bunit = PTLRPC_MAX_BRW_SIZE;
1227         qctxt->lqc_cqs_least_iunit = 2;
1228         qctxt->lqc_cqs_qs_factor = 2;
1229         qctxt->lqc_flags = 0;
1230         QUOTA_MASTER_UNREADY(qctxt);
1231         qctxt->lqc_bunit_sz = default_bunit_sz;
1232         qctxt->lqc_btune_sz = default_bunit_sz / 100 * default_btune_ratio;
1233         qctxt->lqc_iunit_sz = default_iunit_sz;
1234         qctxt->lqc_itune_sz = default_iunit_sz * default_itune_ratio / 100;
1235         qctxt->lqc_switch_seconds = 300; /* enlarging will wait 5 minutes
1236                                           * after the last shrinking */
1237         qctxt->lqc_sync_blk = 0;
1238         spin_unlock(&qctxt->lqc_lock);
1239
1240         qctxt->lqc_lqs_hash = cfs_hash_create("LQS_HASH",
1241                                               HASH_LQS_CUR_BITS,
1242                                               HASH_LQS_MAX_BITS,
1243                                               &lqs_hash_ops, CFS_HASH_REHASH);
1244         if (!qctxt->lqc_lqs_hash) {
1245                 CERROR("initialize hash lqs for %s error!\n", obd->obd_name);
1246                 RETURN(-ENOMEM);
1247         }
1248
1249 #ifdef LPROCFS
1250         rc = lquota_proc_setup(obd, is_master(qctxt));
1251         if (rc)
1252                 CERROR("initialize proc for %s error!\n", obd->obd_name);
1253 #endif
1254
1255         RETURN(rc);
1256 }
1257
1258 static int check_lqs(struct lustre_quota_ctxt *qctxt)
1259 {
1260         int rc;
1261         ENTRY;
1262
1263         rc = !atomic_read(&qctxt->lqc_lqs);
1264
1265         RETURN(rc);
1266 }
1267
1268
1269 void hash_put_lqs(void *obj, void *data)
1270 {
1271         lqs_putref((struct lustre_qunit_size *)obj);
1272 }
1273
1274 void qctxt_cleanup(struct lustre_quota_ctxt *qctxt, int force)
1275 {
1276         struct lustre_qunit *qunit, *tmp;
1277         struct list_head tmp_list;
1278         struct l_wait_info lwi = { 0 };
1279         struct obd_device_target *obt = qctxt->lqc_obt;
1280         int i;
1281         ENTRY;
1282
1283         CFS_INIT_LIST_HEAD(&tmp_list);
1284
1285         spin_lock(&qctxt->lqc_lock);
1286         qctxt->lqc_valid = 0;
1287         spin_unlock(&qctxt->lqc_lock);
1288
1289         spin_lock(&qunit_hash_lock);
1290         for (i = 0; i < NR_DQHASH; i++) {
1291                 list_for_each_entry_safe(qunit, tmp, &qunit_hash[i], lq_hash) {
1292                         if (qunit->lq_ctxt != qctxt)
1293                                 continue;
1294                         remove_qunit_nolock(qunit);
1295                         list_add(&qunit->lq_hash, &tmp_list);
1296                 }
1297         }
1298         spin_unlock(&qunit_hash_lock);
1299
1300         list_for_each_entry_safe(qunit, tmp, &tmp_list, lq_hash) {
1301                 list_del_init(&qunit->lq_hash);
1302                 compute_lqs_after_removing_qunit(qunit);
1303
1304                 /* wake up all waiters */
1305                 QUNIT_SET_STATE_AND_RC(qunit, QUNIT_FINISHED, 0);
1306                 wake_up_all(&qunit->lq_waitq);
1307                 qunit_put(qunit);
1308         }
1309
1310         /* after qctxt_cleanup, qctxt might be freed, then check_qm() is
1311          * unpredicted. So we must wait until lqc_wait_for_qmaster is empty */
1312         while (cfs_waitq_active(&qctxt->lqc_wait_for_qmaster)) {
1313                 cfs_waitq_signal(&qctxt->lqc_wait_for_qmaster);
1314                 cfs_schedule_timeout(CFS_TASK_INTERRUPTIBLE,
1315                                      cfs_time_seconds(1));
1316         }
1317
1318         cfs_hash_for_each_safe(qctxt->lqc_lqs_hash, hash_put_lqs, NULL);
1319         l_wait_event(qctxt->lqc_lqs_waitq, check_lqs(qctxt), &lwi);
1320         down_write(&obt->obt_rwsem);
1321         cfs_hash_destroy(qctxt->lqc_lqs_hash);
1322         qctxt->lqc_lqs_hash = NULL;
1323         up_write(&obt->obt_rwsem);
1324
1325         ptlrpcd_decref();
1326
1327 #ifdef LPROCFS
1328         if (lquota_proc_cleanup(qctxt))
1329                 CERROR("cleanup proc error!\n");
1330 #endif
1331
1332         EXIT;
1333 }
1334
1335 struct qslave_recov_thread_data {
1336         struct obd_device *obd;
1337         struct lustre_quota_ctxt *qctxt;
1338         struct completion comp;
1339 };
1340
1341 /* FIXME only recovery block quota by now */
1342 static int qslave_recovery_main(void *arg)
1343 {
1344         struct qslave_recov_thread_data *data = arg;
1345         struct obd_device *obd = data->obd;
1346         struct lustre_quota_ctxt *qctxt = data->qctxt;
1347         unsigned int type;
1348         int rc = 0;
1349         ENTRY;
1350
1351         cfs_daemonize_ctxt("qslave_recovd");
1352
1353         /* for obdfilter */
1354         class_incref(obd, "qslave_recovd_filter", obd);
1355
1356         complete(&data->comp);
1357
1358         spin_lock(&qctxt->lqc_lock);
1359         if (qctxt->lqc_recovery) {
1360                 spin_unlock(&qctxt->lqc_lock);
1361                 class_decref(obd, "qslave_recovd_filter", obd);
1362                 RETURN(0);
1363         } else {
1364                 qctxt->lqc_recovery = 1;
1365                 spin_unlock(&qctxt->lqc_lock);
1366         }
1367
1368         for (type = USRQUOTA; type < MAXQUOTAS; type++) {
1369                 struct qunit_data qdata;
1370                 struct quota_info *dqopt = sb_dqopt(qctxt->lqc_sb);
1371                 struct list_head id_list;
1372                 struct dquot_id *dqid, *tmp;
1373                 int ret;
1374
1375                 LOCK_DQONOFF_MUTEX(dqopt);
1376                 if (!ll_sb_has_quota_active(qctxt->lqc_sb, type)) {
1377                         UNLOCK_DQONOFF_MUTEX(dqopt);
1378                         break;
1379                 }
1380
1381                 LASSERT(dqopt->files[type] != NULL);
1382                 CFS_INIT_LIST_HEAD(&id_list);
1383 #ifndef KERNEL_SUPPORTS_QUOTA_READ
1384                 rc = fsfilt_qids(obd, dqopt->files[type], NULL, type, &id_list);
1385 #else
1386                 rc = fsfilt_qids(obd, NULL, dqopt->files[type], type, &id_list);
1387 #endif
1388                 UNLOCK_DQONOFF_MUTEX(dqopt);
1389                 if (rc)
1390                         CERROR("Get ids from quota file failed. (rc:%d)\n", rc);
1391
1392                 list_for_each_entry_safe(dqid, tmp, &id_list, di_link) {
1393                         list_del_init(&dqid->di_link);
1394                         /* skip slave recovery on itself */
1395                         if (is_master(qctxt))
1396                                 goto free;
1397                         if (rc && rc != -EBUSY)
1398                                 goto free;
1399
1400                         qdata.qd_id = dqid->di_id;
1401                         qdata.qd_flags = type;
1402                         QDATA_SET_BLK(&qdata);
1403                         qdata.qd_count = 0;
1404
1405                         ret = check_cur_qunit(obd, qctxt, &qdata);
1406                         if (ret > 0) {
1407                                 int opc;
1408                                 opc = ret == 1 ? QUOTA_DQACQ : QUOTA_DQREL;
1409                                 rc = schedule_dqacq(obd, qctxt, &qdata, opc,
1410                                                     0, NULL);
1411                                 if (rc == -EDQUOT)
1412                                         rc = 0;
1413                         } else {
1414                                 rc = 0;
1415                         }
1416
1417                         if (rc)
1418                                 CDEBUG(rc == -EBUSY ? D_QUOTA : D_ERROR,
1419                                        "qslave recovery failed! (id:%d type:%d "
1420                                        " rc:%d)\n", dqid->di_id, type, rc);
1421 free:
1422                         OBD_FREE_PTR(dqid);
1423                 }
1424         }
1425
1426         spin_lock(&qctxt->lqc_lock);
1427         qctxt->lqc_recovery = 0;
1428         spin_unlock(&qctxt->lqc_lock);
1429         class_decref(obd, "qslave_recovd_filter", obd);
1430         RETURN(rc);
1431 }
1432
1433 void
1434 qslave_start_recovery(struct obd_device *obd, struct lustre_quota_ctxt *qctxt)
1435 {
1436         struct qslave_recov_thread_data data;
1437         int rc;
1438         ENTRY;
1439
1440         if (!ll_sb_any_quota_active(qctxt->lqc_sb))
1441                 goto exit;
1442
1443         data.obd = obd;
1444         data.qctxt = qctxt;
1445         init_completion(&data.comp);
1446
1447         rc = kernel_thread(qslave_recovery_main, &data, CLONE_VM|CLONE_FILES);
1448         if (rc < 0) {
1449                 CERROR("Cannot start quota recovery thread: rc %d\n", rc);
1450                 goto exit;
1451         }
1452         wait_for_completion(&data.comp);
1453 exit:
1454         EXIT;
1455 }
1456
1457 int quota_is_on(struct lustre_quota_ctxt *qctxt, struct obd_quotactl *oqctl)
1458 {
1459         unsigned int type;
1460
1461         for (type = USRQUOTA; type < MAXQUOTAS; type++) {
1462                 if (!Q_TYPESET(oqctl, type))
1463                         continue;
1464                 if (!(qctxt->lqc_flags & UGQUOTA2LQC(oqctl->qc_type)))
1465                         return 0;
1466         }
1467         return 1;
1468 }
1469
1470 int quota_is_off(struct lustre_quota_ctxt *qctxt, struct obd_quotactl *oqctl)
1471 {
1472         unsigned int type;
1473
1474         for (type = USRQUOTA; type < MAXQUOTAS; type++) {
1475                 if (!Q_TYPESET(oqctl, type))
1476                         continue;
1477                 if (qctxt->lqc_flags & UGQUOTA2LQC(oqctl->qc_type))
1478                         return 0;
1479         }
1480         return 1;
1481 }
1482
1483 /**
1484  * When quotaon, build a lqs for every uid/gid who has been set limitation
1485  * for quota. After quota_search_lqs, it will hold one ref for the lqs.
1486  * It will be released when qctxt_cleanup() is executed b=18574
1487  *
1488  * Should be called with obt->obt_quotachecking held. b=20152
1489  */
1490 void build_lqs(struct obd_device *obd)
1491 {
1492         struct obd_device_target *obt = &obd->u.obt;
1493         struct lustre_quota_ctxt *qctxt = &obt->obt_qctxt;
1494         struct list_head id_list;
1495         int i, rc;
1496
1497         LASSERT_SEM_LOCKED(&obt->obt_quotachecking);
1498         INIT_LIST_HEAD(&id_list);
1499         for (i = 0; i < MAXQUOTAS; i++) {
1500                 struct dquot_id *dqid, *tmp;
1501
1502                 if (sb_dqopt(qctxt->lqc_sb)->files[i] == NULL)
1503                         continue;
1504
1505 #ifndef KERNEL_SUPPORTS_QUOTA_READ
1506                 rc = fsfilt_qids(obd, sb_dqopt(qctxt->lqc_sb)->files[i], NULL,
1507                                  i, &id_list);
1508 #else
1509                 rc = fsfilt_qids(obd, NULL, sb_dqopt(qctxt->lqc_sb)->files[i],
1510                                  i, &id_list);
1511 #endif
1512                 if (rc) {
1513                         CDEBUG(D_ERROR, "fail to get %s qids!\n",
1514                                i ? "group" : "user");
1515                         continue;
1516                 }
1517
1518                 list_for_each_entry_safe(dqid, tmp, &id_list,
1519                                          di_link) {
1520                         struct lustre_qunit_size *lqs;
1521
1522                         list_del_init(&dqid->di_link);
1523                         lqs = quota_search_lqs(LQS_KEY(i, dqid->di_id),
1524                                                qctxt, 1);
1525                         if (lqs && !IS_ERR(lqs)) {
1526                                 lqs->lqs_flags |= dqid->di_flag;
1527                                 lqs_putref(lqs);
1528                         } else {
1529                                 CDEBUG(D_ERROR, "fail to create a lqs"
1530                                        "(%s id: %u)!\n", i ? "group" : "user",
1531                                        dqid->di_id);
1532                         }
1533
1534                         OBD_FREE_PTR(dqid);
1535                 }
1536         }
1537 }
1538
1539 /**
1540  * lqs<->qctxt hash operations
1541  */
1542
1543 /**
1544  * string hashing using djb2 hash algorithm
1545  */
1546 static unsigned
1547 lqs_hash(cfs_hash_t *hs, void *key, unsigned mask)
1548 {
1549         struct quota_adjust_qunit *lqs_key;
1550         unsigned hash;
1551         ENTRY;
1552
1553         LASSERT(key);
1554         lqs_key = (struct quota_adjust_qunit *)key;
1555         hash = (QAQ_IS_GRP(lqs_key) ? 5381 : 5387) * lqs_key->qaq_id;
1556
1557         RETURN(hash & mask);
1558 }
1559
1560 static int
1561 lqs_compare(void *key, struct hlist_node *hnode)
1562 {
1563         struct lustre_qunit_size *q;
1564         int rc;
1565         ENTRY;
1566
1567         LASSERT(key);
1568         q = hlist_entry(hnode, struct lustre_qunit_size, lqs_hash);
1569
1570         spin_lock(&q->lqs_lock);
1571         rc = (q->lqs_key == *((unsigned long long *)key));
1572         spin_unlock(&q->lqs_lock);
1573
1574         RETURN(rc);
1575 }
1576
1577 static void *
1578 lqs_get(struct hlist_node *hnode)
1579 {
1580         struct lustre_qunit_size *q =
1581                 hlist_entry(hnode, struct lustre_qunit_size, lqs_hash);
1582         ENTRY;
1583
1584         __lqs_getref(q);
1585
1586         RETURN(q);
1587 }
1588
1589 static void *
1590 lqs_put(struct hlist_node *hnode)
1591 {
1592         struct lustre_qunit_size *q =
1593                 hlist_entry(hnode, struct lustre_qunit_size, lqs_hash);
1594         ENTRY;
1595
1596         __lqs_putref(q);
1597
1598         RETURN(q);
1599 }
1600
1601 static void
1602 lqs_exit(struct hlist_node *hnode)
1603 {
1604         struct lustre_qunit_size *q =
1605                 hlist_entry(hnode, struct lustre_qunit_size, lqs_hash);
1606         ENTRY;
1607
1608         /*
1609          * Nothing should be left. User of lqs put it and
1610          * lqs also was deleted from table by this time
1611          * so we should have 0 refs.
1612          */
1613         LASSERTF(atomic_read(&q->lqs_refcount) == 0,
1614                  "Busy lqs %p with %d refs\n", q,
1615                  atomic_read(&q->lqs_refcount));
1616         OBD_FREE_PTR(q);
1617         EXIT;
1618 }
1619
1620 static cfs_hash_ops_t lqs_hash_ops = {
1621         .hs_hash    = lqs_hash,
1622         .hs_compare = lqs_compare,
1623         .hs_get     = lqs_get,
1624         .hs_put     = lqs_put,
1625         .hs_exit    = lqs_exit
1626 };
1627 #endif /* HAVE_QUOTA_SUPPORT */