Whamcloud - gitweb
b=1028 drop SOM dependency on quota
[fs/lustre-release.git] / lustre / quota / quota_master.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_master.c
37  *
38  * Lustre Quota Master request handler
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 #include <linux/quota.h>
57
58 #include <obd_class.h>
59 #include <lustre_quota.h>
60 #include <lustre_fsfilt.h>
61 #include <lustre_mds.h>
62
63 #include "quota_internal.h"
64
65 #ifdef HAVE_QUOTA_SUPPORT
66
67 /* lock ordering: mds->mds_qonoff_sem > dquot->dq_sem */
68 static cfs_list_t lustre_dquot_hash[NR_DQHASH];
69 static cfs_spinlock_t dquot_hash_lock = CFS_SPIN_LOCK_UNLOCKED;
70
71 cfs_mem_cache_t *lustre_dquot_cachep;
72
73 int lustre_dquot_init(void)
74 {
75         int i;
76         ENTRY;
77
78         LASSERT(lustre_dquot_cachep == NULL);
79         lustre_dquot_cachep = cfs_mem_cache_create("lustre_dquot_cache",
80                                                    sizeof(struct lustre_dquot),
81                                                    0, 0);
82         if (!lustre_dquot_cachep)
83                 return (-ENOMEM);
84
85         for (i = 0; i < NR_DQHASH; i++) {
86                 CFS_INIT_LIST_HEAD(lustre_dquot_hash + i);
87         }
88         RETURN(0);
89 }
90
91 void lustre_dquot_exit(void)
92 {
93         int i;
94         ENTRY;
95         /* FIXME cleanup work ?? */
96
97         for (i = 0; i < NR_DQHASH; i++) {
98                 LASSERT(cfs_list_empty(lustre_dquot_hash + i));
99         }
100         if (lustre_dquot_cachep) {
101                 int rc;
102                 rc = cfs_mem_cache_destroy(lustre_dquot_cachep);
103                 LASSERTF(rc == 0,"couldn't destroy lustre_dquot_cachep slab\n");
104                 lustre_dquot_cachep = NULL;
105         }
106         EXIT;
107 }
108
109 static inline int
110 dquot_hashfn(struct lustre_quota_info *info, unsigned int id, int type)
111              __attribute__((__const__));
112
113 static inline int
114 dquot_hashfn(struct lustre_quota_info *info, unsigned int id, int type)
115 {
116         unsigned long tmp = ((unsigned long)info >> L1_CACHE_SHIFT) ^ id;
117         tmp = (tmp * (MAXQUOTAS - type)) % NR_DQHASH;
118         return tmp;
119 }
120
121 /* caller must hold dquot_hash_lock */
122 static struct lustre_dquot *find_dquot(int hashent,
123                                        struct lustre_quota_info *lqi, qid_t id,
124                                        int type)
125 {
126         struct lustre_dquot *dquot;
127         ENTRY;
128
129         LASSERT_SPIN_LOCKED(&dquot_hash_lock);
130         cfs_list_for_each_entry(dquot, &lustre_dquot_hash[hashent], dq_hash) {
131                 if (dquot->dq_info == lqi &&
132                     dquot->dq_id == id && dquot->dq_type == type)
133                         RETURN(dquot);
134         }
135         RETURN(NULL);
136 }
137
138 static struct lustre_dquot *alloc_dquot(struct lustre_quota_info *lqi,
139                                         qid_t id, int type)
140 {
141         struct lustre_dquot *dquot = NULL;
142         ENTRY;
143
144         OBD_SLAB_ALLOC_PTR_GFP(dquot, lustre_dquot_cachep, CFS_ALLOC_IO);
145         if (dquot == NULL)
146                 RETURN(NULL);
147
148         CFS_INIT_LIST_HEAD(&dquot->dq_hash);
149         cfs_init_mutex_locked(&dquot->dq_sem);
150         dquot->dq_refcnt = 1;
151         dquot->dq_info = lqi;
152         dquot->dq_id = id;
153         dquot->dq_type = type;
154         dquot->dq_status = DQ_STATUS_AVAIL;
155
156         RETURN(dquot);
157 }
158
159 static void free_dquot(struct lustre_dquot *dquot)
160 {
161         OBD_SLAB_FREE(dquot, lustre_dquot_cachep, sizeof(*dquot));
162 }
163
164 static void insert_dquot_nolock(struct lustre_dquot *dquot)
165 {
166         cfs_list_t *head = lustre_dquot_hash +
167             dquot_hashfn(dquot->dq_info, dquot->dq_id, dquot->dq_type);
168         LASSERT(cfs_list_empty(&dquot->dq_hash));
169         cfs_list_add(&dquot->dq_hash, head);
170 }
171
172 static void remove_dquot_nolock(struct lustre_dquot *dquot)
173 {
174         LASSERT(!cfs_list_empty(&dquot->dq_hash));
175         cfs_list_del_init(&dquot->dq_hash);
176 }
177
178 static void lustre_dqput(struct lustre_dquot *dquot)
179 {
180         ENTRY;
181         cfs_spin_lock(&dquot_hash_lock);
182         LASSERT(dquot->dq_refcnt);
183         dquot->dq_refcnt--;
184         if (!dquot->dq_refcnt) {
185                 remove_dquot_nolock(dquot);
186                 free_dquot(dquot);
187         }
188         cfs_spin_unlock(&dquot_hash_lock);
189         EXIT;
190 }
191
192 static struct lustre_dquot *lustre_dqget(struct obd_device *obd,
193                                          struct lustre_quota_info *lqi,
194                                          qid_t id, int type)
195 {
196         unsigned int hashent = dquot_hashfn(lqi, id, type);
197         struct lustre_dquot *dquot, *empty;
198         ENTRY;
199
200         if ((empty = alloc_dquot(lqi, id, type)) == NULL)
201                 RETURN(ERR_PTR(-ENOMEM));
202
203         cfs_spin_lock(&dquot_hash_lock);
204         if ((dquot = find_dquot(hashent, lqi, id, type)) != NULL) {
205                 dquot->dq_refcnt++;
206                 cfs_spin_unlock(&dquot_hash_lock);
207                 free_dquot(empty);
208         } else {
209                 int rc;
210
211                 dquot = empty;
212                 insert_dquot_nolock(dquot);
213                 cfs_spin_unlock(&dquot_hash_lock);
214
215                 rc = fsfilt_dquot(obd, dquot, QFILE_RD_DQUOT);
216                 cfs_up(&dquot->dq_sem);
217                 if (rc) {
218                         CERROR("can't read dquot from admin quotafile! "
219                                "(rc:%d)\n", rc);
220                         lustre_dqput(dquot);
221                         RETURN(ERR_PTR(rc));
222                 }
223
224         }
225
226         LASSERT(dquot);
227         RETURN(dquot);
228 }
229
230 static void init_oqaq(struct quota_adjust_qunit *oqaq,
231                       struct lustre_quota_ctxt *qctxt,
232                       qid_t id, int type)
233 {
234         struct lustre_qunit_size *lqs = NULL;
235
236         oqaq->qaq_id = id;
237         oqaq->qaq_flags = type;
238         lqs = quota_search_lqs(LQS_KEY(type, id), qctxt, 0);
239         if (lqs && !IS_ERR(lqs)) {
240                 cfs_spin_lock(&lqs->lqs_lock);
241                 oqaq->qaq_bunit_sz = lqs->lqs_bunit_sz;
242                 oqaq->qaq_iunit_sz = lqs->lqs_iunit_sz;
243                 oqaq->qaq_flags    = lqs->lqs_flags;
244                 cfs_spin_unlock(&lqs->lqs_lock);
245                 lqs_putref(lqs);
246         } else {
247                 CDEBUG(D_QUOTA, "Can't find the lustre qunit size!\n");
248                 oqaq->qaq_bunit_sz = qctxt->lqc_bunit_sz;
249                 oqaq->qaq_iunit_sz = qctxt->lqc_iunit_sz;
250         }
251 }
252
253 int dqacq_adjust_qunit_sz(struct obd_device *obd, qid_t id, int type,
254                           __u32 is_blk)
255 {
256         struct mds_obd *mds = &obd->u.mds;
257         struct lustre_quota_ctxt *qctxt = &mds->mds_obt.obt_qctxt;
258         __u32 ost_num = mds->mds_lov_objid_count, mdt_num = 1;
259         struct quota_adjust_qunit *oqaq = NULL;
260         unsigned int qid[MAXQUOTAS] = { 0, 0 };
261         struct lustre_quota_info *info = &mds->mds_quota_info;
262         struct lustre_dquot *dquot = NULL;
263         int adjust_res = 0;
264         int rc = 0;
265         ENTRY;
266
267         LASSERT(mds);
268         dquot = lustre_dqget(obd, info, id, type);
269         if (IS_ERR(dquot))
270                 RETURN(PTR_ERR(dquot));
271
272         OBD_ALLOC_PTR(oqaq);
273         if (!oqaq)
274                 GOTO(out, rc = -ENOMEM);
275
276         cfs_down(&dquot->dq_sem);
277         init_oqaq(oqaq, qctxt, id, type);
278
279         rc = dquot_create_oqaq(qctxt, dquot, ost_num, mdt_num,
280                                is_blk ? LQUOTA_FLAGS_ADJBLK :
281                                LQUOTA_FLAGS_ADJINO, oqaq);
282
283         if (rc < 0) {
284                 CERROR("create oqaq failed! (rc:%d)\n", rc);
285                 GOTO(out_sem, rc);
286         }
287         QAQ_DEBUG(oqaq, "show oqaq.\n")
288
289         if (!QAQ_IS_ADJBLK(oqaq) && !QAQ_IS_ADJINO(oqaq))
290                 GOTO(out_sem, rc);
291
292         /* adjust the mds slave qunit size */
293         adjust_res = quota_adjust_slave_lqs(oqaq, qctxt);
294         if (adjust_res <= 0) {
295                 if (adjust_res < 0) {
296                         rc = adjust_res;
297                         CERROR("adjust mds slave's qunit size failed! "
298                                "(rc:%d)\n", rc);
299                 } else {
300                         CDEBUG(D_QUOTA, "qunit doesn't need to be adjusted.\n");
301                 }
302                 GOTO(out_sem, rc);
303         }
304
305         if (type)
306                 qid[GRPQUOTA] = dquot->dq_id;
307         else
308                 qid[USRQUOTA] = dquot->dq_id;
309
310         cfs_up(&dquot->dq_sem);
311
312         rc = qctxt_adjust_qunit(obd, qctxt, qid, is_blk, 0, NULL);
313         if (rc == -EDQUOT || rc == -EBUSY) {
314                 CDEBUG(D_QUOTA, "rc: %d.\n", rc);
315                 rc = 0;
316         }
317         if (rc) {
318                 CERROR("%s: mds fail to adjust file quota! (rc:%d)\n",
319                        obd->obd_name, rc);
320                 GOTO(out, rc);
321         }
322
323         /* only when block qunit is reduced, boardcast to osts */
324         if ((adjust_res & LQS_BLK_DECREASE) && QAQ_IS_ADJBLK(oqaq))
325                 rc = obd_quota_adjust_qunit(mds->mds_osc_exp, oqaq, qctxt);
326
327 out:
328         lustre_dqput(dquot);
329         if (oqaq)
330                 OBD_FREE_PTR(oqaq);
331
332         RETURN(rc);
333 out_sem:
334         cfs_up(&dquot->dq_sem);
335         goto out;
336 }
337
338 int dqacq_handler(struct obd_device *obd, struct qunit_data *qdata, int opc)
339 {
340         struct mds_obd *mds = &obd->u.mds;
341         struct lustre_quota_ctxt *qctxt = &mds->mds_obt.obt_qctxt;
342         struct lustre_quota_info *info = &mds->mds_quota_info;
343         struct lustre_dquot *dquot = NULL;
344         __u64 *usage = NULL;
345         __u64 hlimit = 0, slimit = 0;
346         time_t *time = NULL;
347         unsigned int grace = 0;
348         struct lustre_qunit_size *lqs = NULL;
349         int rc = 0;
350         ENTRY;
351
352         if (OBD_FAIL_CHECK(OBD_FAIL_OBD_DQACQ))
353                 RETURN(-EIO);
354
355         dquot = lustre_dqget(obd, info, qdata->qd_id, QDATA_IS_GRP(qdata));
356         if (IS_ERR(dquot))
357                 RETURN(PTR_ERR(dquot));
358
359         DQUOT_DEBUG(dquot, "get dquot in dqacq_handler\n");
360         QINFO_DEBUG(dquot->dq_info, "get dquot in dqadq_handler\n");
361
362         cfs_down(&mds->mds_qonoff_sem);
363         cfs_down(&dquot->dq_sem);
364
365         if (dquot->dq_status & DQ_STATUS_RECOVERY) {
366                 DQUOT_DEBUG(dquot, "this dquot is under recovering.\n");
367                 GOTO(out, rc = -EBUSY);
368         }
369
370         if (QDATA_IS_BLK(qdata)) {
371                 grace = info->qi_info[QDATA_IS_GRP(qdata)].dqi_bgrace;
372                 usage = &dquot->dq_dqb.dqb_curspace;
373                 hlimit = dquot->dq_dqb.dqb_bhardlimit;
374                 slimit = dquot->dq_dqb.dqb_bsoftlimit;
375                 time = &dquot->dq_dqb.dqb_btime;
376         } else {
377                 grace = info->qi_info[QDATA_IS_GRP(qdata)].dqi_igrace;
378                 usage = (__u64 *) & dquot->dq_dqb.dqb_curinodes;
379                 hlimit = dquot->dq_dqb.dqb_ihardlimit;
380                 slimit = dquot->dq_dqb.dqb_isoftlimit;
381                 time = &dquot->dq_dqb.dqb_itime;
382         }
383
384         /* if the quota limit in admin quotafile is zero, we just inform
385          * slave to clear quota limit with zero qd_count */
386         if (hlimit == 0 && slimit == 0) {
387                 qdata->qd_count = 0;
388                 GOTO(out, rc);
389         }
390
391         switch (opc) {
392         case QUOTA_DQACQ:
393                 if (hlimit &&
394                     QUSG(*usage + qdata->qd_count, QDATA_IS_BLK(qdata)) > hlimit)
395                 {
396                         if (QDATA_IS_CHANGE_QS(qdata) &&
397                             QUSG(*usage, QDATA_IS_BLK(qdata)) < hlimit)
398                                 qdata->qd_count = (hlimit -
399                                         QUSG(*usage, QDATA_IS_BLK(qdata)))
400                                         * (QDATA_IS_BLK(qdata) ?
401                                            QUOTABLOCK_SIZE : 1);
402                         else
403                                 GOTO(out, rc = -EDQUOT);
404                 }
405
406                 if (slimit &&
407                     QUSG(*usage + qdata->qd_count, QDATA_IS_BLK(qdata)) > slimit) {
408                         if (*time && cfs_time_current_sec() >= *time)
409                                 GOTO(out, rc = -EDQUOT);
410                         else if (!*time)
411                                 *time = cfs_time_current_sec() + grace;
412                 }
413
414                 *usage += qdata->qd_count;
415                 break;
416         case QUOTA_DQREL:
417                 /* The usage in administrative file might be incorrect before
418                  * recovery done */
419                 if (*usage - qdata->qd_count < 0)
420                         *usage = 0;
421                 else
422                         *usage -= qdata->qd_count;
423
424                 /* (usage <= soft limit) but not (usage < soft limit) */
425                 if (!slimit || QUSG(*usage, QDATA_IS_BLK(qdata)) <= slimit)
426                         *time = 0;
427                 break;
428         default:
429                 LBUG();
430         }
431
432         rc = fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
433         EXIT;
434 out:
435         cfs_up(&dquot->dq_sem);
436         cfs_up(&mds->mds_qonoff_sem);
437         lustre_dqput(dquot);
438         if (rc != -EDQUOT)
439                 dqacq_adjust_qunit_sz(obd, qdata->qd_id, QDATA_IS_GRP(qdata),
440                                       QDATA_IS_BLK(qdata));
441
442         lqs = quota_search_lqs(LQS_KEY(QDATA_IS_GRP(qdata), qdata->qd_id),
443                                qctxt, 0);
444         if (lqs == NULL || IS_ERR(lqs)) {
445                 CDEBUG(D_INFO, "Can't find the lustre qunit size!\n");
446                 qdata->qd_qunit  = QDATA_IS_BLK(qdata) ? qctxt->lqc_bunit_sz :
447                                                          qctxt->lqc_iunit_sz;
448         } else {
449                 cfs_spin_lock(&lqs->lqs_lock);
450                 qdata->qd_qunit  = QDATA_IS_BLK(qdata) ? lqs->lqs_bunit_sz :
451                                                          lqs->lqs_iunit_sz;
452                 cfs_spin_unlock(&lqs->lqs_lock);
453         }
454
455         if (QDATA_IS_BLK(qdata))
456                 QDATA_SET_ADJBLK(qdata);
457         else
458                 QDATA_SET_ADJINO(qdata);
459
460         QDATA_DEBUG(qdata, "alloc/release qunit in dqacq_handler\n");
461         if (lqs)
462                 lqs_putref(lqs);
463
464         return rc;
465 }
466
467 int mds_quota_adjust(struct obd_device *obd, const unsigned int qcids[],
468                      const unsigned int qpids[], int rc, int opc)
469 {
470         struct lustre_quota_ctxt *qctxt = &obd->u.obt.obt_qctxt;
471         int rc2 = 0;
472         ENTRY;
473
474         if (rc && rc != -EDQUOT && rc != ENOLCK)
475                 RETURN(0);
476
477         switch (opc) {
478         case FSFILT_OP_SETATTR:
479                 /* release file quota on original owner */
480                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids, 0, 0, NULL);
481                 /* release block quota on original owner */
482                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids, 1, 0, NULL);
483                 /* acquire file quota on current owner */
484                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 0, 0, NULL);
485                 /* acquire block quota on current owner */
486                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 1, 0, NULL);
487                 break;
488         case FSFILT_OP_UNLINK_PARTIAL_CHILD:
489                 /* release file quota on child */
490                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 0, 0, NULL);
491                 /* rlease block quota on child */
492                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 1, 0, NULL);
493                 break;
494         case FSFILT_OP_CREATE_PARTIAL_CHILD:
495                 /* acquire file quota on child */
496                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 0, 0, NULL);
497                 /* acquire block quota on child */
498                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 1, 0, NULL);
499                 break;
500         case FSFILT_OP_LINK:
501                 /* acquire block quota on parent */
502                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids, 1, 0, NULL);
503                 break;
504         case FSFILT_OP_UNLINK:
505                 /* release block quota on parent */
506                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids, 1, 0, NULL);
507                 /* release file quota on child */
508                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 0, 0, NULL);
509                 if (qpids[0] != qcids[0] || qpids[1] != qcids[1])
510                         /* release block quota on child */
511                         rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 1, 0,
512                                                   NULL);
513                 break;
514         case FSFILT_OP_UNLINK_PARTIAL_PARENT:
515                 /* release block quota on parent */
516                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids, 1, 0, NULL);
517                 break;
518         case FSFILT_OP_CREATE:
519                 /* acquire block quota on parent */
520                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids, 1, 0, NULL);
521                 /* acquire file quota on child */
522                 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 0, 0, NULL);
523                 if (qpids[0] != qcids[0] || qpids[1] != qcids[1])
524                         /* acquire block quota on child */
525                         rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids, 1, 0,
526                                                   NULL);
527                 break;
528         default:
529                 LBUG();
530                 break;
531         }
532
533         if (rc2)
534                 CDEBUG(D_QUOTA,
535                        "mds adjust qunit %ssuccessfully! (opc:%d rc:%d)\n",
536                        rc2 == QUOTA_REQ_RETURNED ? "" : "un", opc, rc2);
537         RETURN(0);
538 }
539
540 int filter_quota_adjust(struct obd_device *obd, const unsigned int qcids[],
541                         const unsigned int qpids[], int rc, int opc)
542 {
543         struct lustre_quota_ctxt *qctxt = &obd->u.obt.obt_qctxt;
544         int rc2 = 0;
545         ENTRY;
546
547         if (rc && rc != -EDQUOT)
548                 RETURN(0);
549
550         switch (opc) {
551         case FSFILT_OP_SETATTR:
552                 /* acquire/release block quota on original & current owner */
553                 rc = qctxt_adjust_qunit(obd, qctxt, qcids, 1, 0, NULL);
554                 rc2 = qctxt_adjust_qunit(obd, qctxt, qpids, 1, 0, NULL);
555                 break;
556         case FSFILT_OP_UNLINK:
557                 /* release block quota on this owner */
558         case FSFILT_OP_CREATE: /* XXX for write operation on obdfilter */
559                 /* acquire block quota on this owner */
560                 rc = qctxt_adjust_qunit(obd, qctxt, qcids, 1, 0, NULL);
561                 break;
562         default:
563                 LBUG();
564                 break;
565         }
566
567         if (rc || rc2) {
568                 if (!rc)
569                         rc = rc2;
570                 CDEBUG(D_QUOTA,
571                        "filter adjust qunit %ssuccessfully! (opc:%d rc%d)\n",
572                        rc == QUOTA_REQ_RETURNED ? "" : "un", opc, rc);
573         }
574
575         RETURN(0);
576 }
577
578 static const char prefix[] = "OBJECTS/";
579
580 int mds_quota_invalidate(struct obd_device *obd, struct obd_quotactl *oqctl)
581 {
582         struct mds_obd *mds = &obd->u.mds;
583         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
584         struct obd_device_target *obt = &obd->u.obt;
585         int rc = 0, i, rc1 = 0;
586         char *quotafile[] = LUSTRE_ADMIN_QUOTAFILES_V2;
587         char name[64];
588         struct lvfs_run_ctxt saved;
589         ENTRY;
590
591         LASSERT(qinfo->qi_version == LUSTRE_QUOTA_V2);
592
593         if (oqctl->qc_type != USRQUOTA &&
594             oqctl->qc_type != GRPQUOTA &&
595             oqctl->qc_type != UGQUOTA)
596                 RETURN(-EINVAL);
597
598         cfs_down(&obt->obt_quotachecking);
599         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
600         cfs_down(&mds->mds_qonoff_sem);
601
602         for (i = 0; i < MAXQUOTAS; i++) {
603                 struct file *fp;
604
605                 if (!Q_TYPESET(oqctl, i))
606                         continue;
607
608                 /* quota file has been opened ? */
609                 if (qinfo->qi_files[i]) {
610                         CWARN("quota[%d] is on yet\n", i);
611                         rc1 = -EBUSY;
612                         continue;
613                 }
614
615                 LASSERT(strlen(quotafile[i]) + sizeof(prefix) <= sizeof(name));
616                 sprintf(name, "%s%s", prefix, quotafile[i]);
617
618                 fp = filp_open(name, O_CREAT | O_TRUNC | O_RDWR, 0644);
619                 if (IS_ERR(fp)) {
620                         rc = PTR_ERR(fp);
621                         CERROR("%s: error invalidating admin quotafile %s (rc:%d)\n",
622                                obd->obd_name, name, rc);
623                 }
624                 else
625                         filp_close(fp, 0);
626         }
627
628         cfs_up(&mds->mds_qonoff_sem);
629         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
630         cfs_up(&obt->obt_quotachecking);
631         RETURN(rc ? : rc1);
632 }
633
634 int mds_quota_finvalidate(struct obd_device *obd, struct obd_quotactl *oqctl)
635 {
636         struct mds_obd *mds = &obd->u.mds;
637         struct obd_device_target *obt = &obd->u.obt;
638         int rc;
639         struct lvfs_run_ctxt saved;
640         ENTRY;
641
642         if (oqctl->qc_type != USRQUOTA &&
643             oqctl->qc_type != GRPQUOTA &&
644             oqctl->qc_type != UGQUOTA)
645                 RETURN(-EINVAL);
646
647         cfs_down(&obt->obt_quotachecking);
648         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
649         cfs_down(&mds->mds_qonoff_sem);
650
651         oqctl->qc_cmd = Q_FINVALIDATE;
652         rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
653         if (!rc)
654                 rc = obd_quotactl(mds->mds_osc_exp, oqctl);
655
656         cfs_up(&mds->mds_qonoff_sem);
657         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
658         cfs_up(&obt->obt_quotachecking);
659         RETURN(rc);
660 }
661
662 int init_admin_quotafiles(struct obd_device *obd, struct obd_quotactl *oqctl)
663 {
664         struct mds_obd *mds = &obd->u.mds;
665         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
666         const char *quotafile[] = LUSTRE_ADMIN_QUOTAFILES_V2;
667         struct lvfs_run_ctxt saved;
668         char name[64];
669         int i, rc = 0;
670         ENTRY;
671
672         LASSERT(qinfo->qi_version == LUSTRE_QUOTA_V2);
673
674         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
675
676         cfs_down(&mds->mds_qonoff_sem);
677
678         for (i = 0; i < MAXQUOTAS && !rc; i++) {
679                 struct file *fp;
680
681                 if (!Q_TYPESET(oqctl, i))
682                         continue;
683
684                 /* quota file has been opened ? */
685                 if (qinfo->qi_files[i]) {
686                         CWARN("init %s admin quotafile while quota on.\n",
687                               i == USRQUOTA ? "user" : "group");
688                         continue;
689                 }
690
691                 LASSERT(strlen(quotafile[i]) + sizeof(prefix) <= sizeof(name));
692                 sprintf(name, "%s%s", prefix, quotafile[i]);
693
694                 /* check if quota file exists and is correct */
695                 fp = filp_open(name, O_RDONLY, 0);
696                 if (!IS_ERR(fp)) {
697                         /* irregular file is not the right place for quota */
698                         if (!S_ISREG(fp->f_dentry->d_inode->i_mode)) {
699                                 CERROR("admin quota file %s is not "
700                                        "regular!", name);
701                                 filp_close(fp, 0);
702                                 rc = -EINVAL;
703                                 break;
704                         }
705                         qinfo->qi_files[i] = fp;
706                         rc = fsfilt_quotainfo(obd, qinfo, i, QFILE_CHK);
707                         qinfo->qi_files[i] = 0;
708                         filp_close(fp, 0);
709                 }
710                 else
711                         rc = PTR_ERR(fp);
712
713                 if (!rc)
714                         continue;
715
716                 /* -EINVAL may be returned by quotainfo for bad quota file */
717                 if (rc != -ENOENT && rc != -EINVAL) {
718                         CERROR("%s: error opening old quota file %s (%d)\n",
719                                obd->obd_name, name, rc);
720                         break;
721                 }
722
723                 CDEBUG(D_INFO, "%s new quota file %s\n", name,
724                        rc == -ENOENT ? "creating" : "overwriting");
725
726                 /* create quota file overwriting old if needed */
727                 fp = filp_open(name, O_CREAT | O_TRUNC | O_RDWR, 0644);
728                 if (IS_ERR(fp)) {
729                         rc = PTR_ERR(fp);
730                         CERROR("%s: error creating admin quotafile %s (rc:%d)\n",
731                                obd->obd_name, name, rc);
732                         break;
733                 }
734
735                 qinfo->qi_files[i] = fp;
736
737                 rc = fsfilt_quotainfo(obd, qinfo, i, QFILE_INIT_INFO);
738                 if (rc)
739                         CERROR("error init %s admin quotafile! (rc:%d)\n",
740                                i == USRQUOTA ? "user" : "group", rc);
741
742                 filp_close(fp, 0);
743                 qinfo->qi_files[i] = NULL;
744         }
745         cfs_up(&mds->mds_qonoff_sem);
746
747         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
748         RETURN(rc);
749 }
750
751 static int close_quota_files(struct obd_quotactl *oqctl,
752                              struct lustre_quota_info *qinfo)
753 {
754         int i, rc = 0;
755         ENTRY;
756
757         for (i = 0; i < MAXQUOTAS; i++) {
758                 if (!Q_TYPESET(oqctl, i))
759                         continue;
760                 if (qinfo->qi_files[i] == NULL) {
761                         CWARN("quota[%d] is off already\n", i);
762                         rc = -EALREADY;
763                         continue;
764                 }
765                 filp_close(qinfo->qi_files[i], 0);
766                 qinfo->qi_files[i] = NULL;
767         }
768         RETURN(rc);
769 }
770
771 int mds_admin_quota_on(struct obd_device *obd, struct obd_quotactl *oqctl)
772 {
773         struct mds_obd *mds = &obd->u.mds;
774         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
775         const char *quotafile[] = LUSTRE_ADMIN_QUOTAFILES_V2;
776         char name[64];
777         int i, rc = 0, rc1 = 0;
778         ENTRY;
779
780         LASSERT(qinfo->qi_version == LUSTRE_QUOTA_V2);
781
782         /* open admin quota files and read quotafile info */
783         for (i = 0; i < MAXQUOTAS; i++) {
784                 struct file *fp;
785
786                 if (!Q_TYPESET(oqctl, i))
787                         continue;
788
789                 LASSERT(strlen(quotafile[i])
790                         + sizeof(prefix) <= sizeof(name));
791                 sprintf(name, "%s%s", prefix, quotafile[i]);
792
793                 if (qinfo->qi_files[i] != NULL) {
794                         CWARN("quota[%d] is on already\n", i);
795                         rc1 = -EALREADY;
796                         continue;
797                 }
798
799                 fp = filp_open(name, O_RDWR, 0);
800                 if (IS_ERR(fp) || !S_ISREG(fp->f_dentry->d_inode->i_mode)) {
801                         rc = IS_ERR(fp) ? PTR_ERR(fp) : -EINVAL;
802                         CERROR("error open/create %s! (rc:%d)\n", name, rc);
803                         break;
804                 }
805                 qinfo->qi_files[i] = fp;
806
807                 rc = fsfilt_quotainfo(obd, qinfo, i, QFILE_CHK);
808                 if (rc) {
809                         CERROR("invalid quota file %s! (rc:%d)\n", name, rc);
810                         break;
811                 }
812
813                 rc = fsfilt_quotainfo(obd, qinfo, i, QFILE_RD_INFO);
814                 if (rc) {
815                         CERROR("error read quotainfo of %s! (rc:%d)\n", name,
816                                rc);
817                         break;
818                 }
819         }
820
821         if (rc && rc1 != -EALREADY)
822                 close_quota_files(oqctl, qinfo);
823
824         RETURN(rc ? : rc1);
825 }
826
827 int mds_admin_quota_off(struct obd_device *obd,
828                         struct obd_quotactl *oqctl)
829 {
830         struct mds_obd *mds = &obd->u.mds;
831         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
832         int rc;
833         ENTRY;
834
835         /* close admin quota files */
836         rc = close_quota_files(oqctl, qinfo);
837         RETURN(rc);
838 }
839
840 int mds_quota_on(struct obd_device *obd, struct obd_quotactl *oqctl)
841 {
842         struct mds_obd *mds = &obd->u.mds;
843         struct obd_device_target *obt = &obd->u.obt;
844         struct lustre_quota_ctxt *qctxt = &obt->obt_qctxt;
845         struct lvfs_run_ctxt saved;
846         int rc = 0, rc1 = 0, rc2 = 0;
847         ENTRY;
848
849         if (oqctl->qc_type != USRQUOTA &&
850             oqctl->qc_type != GRPQUOTA &&
851             oqctl->qc_type != UGQUOTA)
852                 RETURN(-EINVAL);
853
854         cfs_down(&obt->obt_quotachecking);
855         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
856         cfs_down(&mds->mds_qonoff_sem);
857         rc2 = mds_admin_quota_on(obd, oqctl);
858         if (rc2 && rc2 != -EALREADY) {
859                 CWARN("mds quota[%d] is failed to be on for %d\n", oqctl->qc_type, rc2);
860                 GOTO(out, rc2);
861         }
862
863         rc1 = fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
864         if (!rc1) {
865                 qctxt->lqc_flags |= UGQUOTA2LQC(oqctl->qc_type);
866                 /* when quotaon, create lqs for every quota uid/gid b=18574 */
867                 build_lqs(obd);
868         } else if (rc1 == -EBUSY && quota_is_on(qctxt, oqctl)) {
869                 CWARN("mds local quota[%d] is on already\n", oqctl->qc_type);
870                 rc1 = -EALREADY;
871         } else {
872                 if (rc2 != -EALREADY) {
873                         CWARN("mds local quota[%d] is failed to be on for %d\n",
874                               oqctl->qc_type, rc1);
875                         oqctl->qc_cmd = Q_QUOTAOFF;
876                         mds_admin_quota_off(obd, oqctl);
877                         oqctl->qc_cmd = Q_QUOTAON;
878                 }
879                 GOTO(out, rc1);
880         }
881
882         rc = obd_quotactl(mds->mds_osc_exp, oqctl);
883         if (rc && rc != -EALREADY) {
884                 CWARN("mds remote quota[%d] is failed to be on for %d\n",
885                       oqctl->qc_type, rc);
886                 oqctl->qc_cmd = Q_QUOTAOFF;
887                 if (rc2 != -EALREADY)
888                         mds_admin_quota_off(obd, oqctl);
889                 if (rc1 != -EALREADY) {
890                         fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
891                         qctxt->lqc_flags &= ~UGQUOTA2LQC(oqctl->qc_type);
892                 }
893                 oqctl->qc_cmd = Q_QUOTAON;
894         }
895
896         EXIT;
897
898 out:
899         cfs_up(&mds->mds_qonoff_sem);
900         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
901         cfs_up(&obt->obt_quotachecking);
902         return rc ? : (rc1 ? : rc2);
903 }
904
905 /* with obt->obt_quotachecking held */
906 int do_mds_quota_off(struct obd_device *obd, struct obd_quotactl *oqctl)
907 {
908         struct mds_obd *mds = &obd->u.mds;
909         struct obd_device_target *obt = &obd->u.obt;
910         struct lustre_quota_ctxt *qctxt = &obt->obt_qctxt;
911         struct lvfs_run_ctxt saved;
912         int rc = 0, rc1 = 0, rc2 = 0;
913         ENTRY;
914
915         LASSERT_SEM_LOCKED(&obt->obt_quotachecking);
916
917         if (oqctl->qc_type != USRQUOTA &&
918             oqctl->qc_type != GRPQUOTA &&
919             oqctl->qc_type != UGQUOTA)
920                 RETURN(-EINVAL);
921
922         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
923         cfs_down(&mds->mds_qonoff_sem);
924         /* close admin quota files */
925         rc2 = mds_admin_quota_off(obd, oqctl);
926         if (rc2 && rc2 != -EALREADY) {
927                 CWARN("mds quota[%d] is failed to be off for %d\n", oqctl->qc_type, rc2);
928                 GOTO(out, rc2);
929         }
930
931         rc1 = fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
932         if (!rc1) {
933                 obt->obt_qctxt.lqc_flags &= ~UGQUOTA2LQC(oqctl->qc_type);
934         } else if (quota_is_off(qctxt, oqctl)) {
935                 CWARN("mds local quota[%d] is off already\n", oqctl->qc_type);
936                 rc1 = -EALREADY;
937         } else {
938                 if (rc2 != -EALREADY) {
939                         CWARN("mds local quota[%d] is failed to be off for %d\n",
940                               oqctl->qc_type, rc1);
941                         oqctl->qc_cmd = Q_QUOTAON;
942                         mds_admin_quota_on(obd, oqctl);
943                         oqctl->qc_cmd = Q_QUOTAOFF;
944                 }
945                 GOTO(out, rc1);
946         }
947
948         rc = obd_quotactl(mds->mds_osc_exp, oqctl);
949         if (rc && rc != -EALREADY) {
950                 CWARN("mds remote quota[%d] is failed to be off for %d\n",
951                       oqctl->qc_type, rc);
952                 oqctl->qc_cmd = Q_QUOTAON;
953                 if (rc2 != -EALREADY)
954                         mds_admin_quota_on(obd, oqctl);
955                 if (rc1 != -EALREADY) {
956                         fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
957                         qctxt->lqc_flags |= UGQUOTA2LQC(oqctl->qc_type);
958                 }
959                 oqctl->qc_cmd = Q_QUOTAOFF;
960         }
961         EXIT;
962
963 out:
964         cfs_up(&mds->mds_qonoff_sem);
965         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
966         return rc ? : (rc1 ? : rc2);
967 }
968
969 int mds_quota_off(struct obd_device *obd, struct obd_quotactl *oqctl)
970 {
971         struct obd_device_target *obt = &obd->u.obt;
972         int rc;
973         ENTRY;
974
975         cfs_down(&obt->obt_quotachecking);
976         rc = do_mds_quota_off(obd, oqctl);
977         cfs_up(&obt->obt_quotachecking);
978         RETURN(rc);
979 }
980
981 int mds_set_dqinfo(struct obd_device *obd, struct obd_quotactl *oqctl)
982 {
983         struct mds_obd *mds = &obd->u.mds;
984         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
985         struct obd_dqinfo *dqinfo = &oqctl->qc_dqinfo;
986         int rc;
987         ENTRY;
988
989         if (oqctl->qc_type != USRQUOTA &&
990             oqctl->qc_type != GRPQUOTA)
991                 RETURN(-EINVAL);
992
993         cfs_down(&mds->mds_qonoff_sem);
994         if (qinfo->qi_files[oqctl->qc_type] == NULL) {
995                 CWARN("quota[%u] is off\n", oqctl->qc_type);
996                 GOTO(out, rc = -ESRCH);
997         }
998
999         qinfo->qi_info[oqctl->qc_type].dqi_bgrace = dqinfo->dqi_bgrace;
1000         qinfo->qi_info[oqctl->qc_type].dqi_igrace = dqinfo->dqi_igrace;
1001         qinfo->qi_info[oqctl->qc_type].dqi_flags = dqinfo->dqi_flags;
1002
1003         rc = fsfilt_quotainfo(obd, qinfo, oqctl->qc_type, QFILE_WR_INFO);
1004         EXIT;
1005
1006 out:
1007         cfs_up(&mds->mds_qonoff_sem);
1008         return rc;
1009 }
1010
1011 int mds_get_dqinfo(struct obd_device *obd, struct obd_quotactl *oqctl)
1012 {
1013         struct mds_obd *mds = &obd->u.mds;
1014         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
1015         struct obd_dqinfo *dqinfo = &oqctl->qc_dqinfo;
1016         int rc = 0;
1017         ENTRY;
1018
1019         if (oqctl->qc_type != USRQUOTA &&
1020             oqctl->qc_type != GRPQUOTA)
1021                 RETURN(-EINVAL);
1022
1023         cfs_down(&mds->mds_qonoff_sem);
1024         if (qinfo->qi_files[oqctl->qc_type] == NULL) {
1025                 CWARN("quota[%u] is off\n", oqctl->qc_type);
1026                 GOTO(out, rc = -ESRCH);
1027         }
1028
1029         dqinfo->dqi_bgrace = qinfo->qi_info[oqctl->qc_type].dqi_bgrace;
1030         dqinfo->dqi_igrace = qinfo->qi_info[oqctl->qc_type].dqi_igrace;
1031         dqinfo->dqi_flags = qinfo->qi_info[oqctl->qc_type].dqi_flags;
1032         EXIT;
1033
1034 out:
1035         cfs_up(&mds->mds_qonoff_sem);
1036         return rc;
1037 }
1038
1039 int dquot_create_oqaq(struct lustre_quota_ctxt *qctxt,
1040                       struct lustre_dquot *dquot, __u32 ost_num, __u32 mdt_num,
1041                       int type, struct quota_adjust_qunit *oqaq)
1042 {
1043         __u64 bunit_curr_o, iunit_curr_o;
1044         unsigned long shrink_qunit_limit = qctxt->lqc_cqs_boundary_factor;
1045         unsigned long cqs_factor = qctxt->lqc_cqs_qs_factor;
1046         __u64 blimit = dquot->dq_dqb.dqb_bhardlimit ?
1047                 dquot->dq_dqb.dqb_bhardlimit : dquot->dq_dqb.dqb_bsoftlimit;
1048         __u64 ilimit = dquot->dq_dqb.dqb_ihardlimit ?
1049                 dquot->dq_dqb.dqb_ihardlimit : dquot->dq_dqb.dqb_isoftlimit;
1050         int rc = 0;
1051         ENTRY;
1052
1053         if (!dquot || !oqaq)
1054                 RETURN(-EINVAL);
1055         LASSERT_SEM_LOCKED(&dquot->dq_sem);
1056         LASSERT(oqaq->qaq_iunit_sz);
1057         LASSERT(oqaq->qaq_bunit_sz);
1058
1059         /* don't change qunit size */
1060         if (!qctxt->lqc_switch_qs)
1061                 RETURN(rc);
1062
1063         bunit_curr_o = oqaq->qaq_bunit_sz;
1064         iunit_curr_o = oqaq->qaq_iunit_sz;
1065
1066         if (dquot->dq_type == GRPQUOTA)
1067                 QAQ_SET_GRP(oqaq);
1068
1069         if ((type & LQUOTA_FLAGS_ADJBLK) && blimit) {
1070                 __u64 b_limitation =
1071                         oqaq->qaq_bunit_sz * (ost_num + 1) * shrink_qunit_limit;
1072                 /* enlarge block qunit size */
1073                 while (blimit >
1074                        QUSG(dquot->dq_dqb.dqb_curspace + 2 * b_limitation, 1)) {
1075                         oqaq->qaq_bunit_sz =
1076                                 QUSG(oqaq->qaq_bunit_sz * cqs_factor, 1)
1077                                 << QUOTABLOCK_BITS;
1078                         b_limitation = oqaq->qaq_bunit_sz * (ost_num + 1) *
1079                                 shrink_qunit_limit;
1080                 }
1081
1082                 if (oqaq->qaq_bunit_sz > qctxt->lqc_bunit_sz)
1083                         oqaq->qaq_bunit_sz = qctxt->lqc_bunit_sz;
1084
1085                 /* shrink block qunit size */
1086                 while (blimit <
1087                        QUSG(dquot->dq_dqb.dqb_curspace + b_limitation, 1)) {
1088                         do_div(oqaq->qaq_bunit_sz , cqs_factor);
1089                         oqaq->qaq_bunit_sz = QUSG(oqaq->qaq_bunit_sz, 1) <<
1090                                 QUOTABLOCK_BITS;
1091                         b_limitation = oqaq->qaq_bunit_sz * (ost_num + 1) *
1092                                 shrink_qunit_limit;
1093                         if (oqaq->qaq_bunit_sz <  qctxt->lqc_cqs_least_bunit)
1094                                 break;
1095                 }
1096
1097                 if (oqaq->qaq_bunit_sz < qctxt->lqc_cqs_least_bunit)
1098                         oqaq->qaq_bunit_sz = qctxt->lqc_cqs_least_bunit;
1099
1100                 if (bunit_curr_o != oqaq->qaq_bunit_sz)
1101                         QAQ_SET_ADJBLK(oqaq);
1102
1103         }
1104
1105         if ((type & LQUOTA_FLAGS_ADJINO) && ilimit) {
1106                 __u64 i_limitation =
1107                         oqaq->qaq_iunit_sz * mdt_num * shrink_qunit_limit;
1108                 /* enlarge file qunit size */
1109                 while (ilimit > dquot->dq_dqb.dqb_curinodes
1110                        + 2 * i_limitation) {
1111                         oqaq->qaq_iunit_sz = oqaq->qaq_iunit_sz * cqs_factor;
1112                         i_limitation = oqaq->qaq_iunit_sz * mdt_num *
1113                                 shrink_qunit_limit;
1114                 }
1115
1116                 if (oqaq->qaq_iunit_sz > qctxt->lqc_iunit_sz)
1117                         oqaq->qaq_iunit_sz = qctxt->lqc_iunit_sz;
1118
1119                 /* shrink file qunit size */
1120                 while (ilimit < dquot->dq_dqb.dqb_curinodes
1121                        + i_limitation) {
1122                         do_div(oqaq->qaq_iunit_sz, cqs_factor);
1123                         i_limitation = oqaq->qaq_iunit_sz * mdt_num *
1124                                        shrink_qunit_limit;
1125                         if (oqaq->qaq_iunit_sz < qctxt->lqc_cqs_least_iunit)
1126                                 break;
1127                 }
1128
1129                 if (oqaq->qaq_iunit_sz < qctxt->lqc_cqs_least_iunit)
1130                         oqaq->qaq_iunit_sz = qctxt->lqc_cqs_least_iunit;
1131
1132                 if (iunit_curr_o != oqaq->qaq_iunit_sz)
1133                         QAQ_SET_ADJINO(oqaq);
1134
1135         }
1136
1137         QAQ_DEBUG(oqaq, "the oqaq computed\n");
1138
1139         RETURN(rc);
1140 }
1141
1142 static int mds_init_slave_ilimits(struct obd_device *obd,
1143                                   struct obd_quotactl *oqctl, int set)
1144 {
1145         /* XXX: for file limits only adjust local now */
1146         struct obd_device_target *obt = &obd->u.obt;
1147         struct lustre_quota_ctxt *qctxt = &obt->obt_qctxt;
1148         unsigned int id[MAXQUOTAS] = { 0, 0 };
1149         struct obd_quotactl *ioqc = NULL;
1150         struct lustre_qunit_size *lqs;
1151         int flag;
1152         int rc;
1153         ENTRY;
1154
1155         /* if we are going to set zero limit, needn't init slaves */
1156         if (!oqctl->qc_dqblk.dqb_ihardlimit && !oqctl->qc_dqblk.dqb_isoftlimit &&
1157             !set)
1158                 RETURN(0);
1159
1160         OBD_ALLOC_PTR(ioqc);
1161         if (!ioqc)
1162                 RETURN(-ENOMEM);
1163
1164         flag = oqctl->qc_dqblk.dqb_ihardlimit ||
1165                oqctl->qc_dqblk.dqb_isoftlimit || !set;
1166         ioqc->qc_cmd = flag ? Q_INITQUOTA : Q_SETQUOTA;
1167         ioqc->qc_id = oqctl->qc_id;
1168         ioqc->qc_type = oqctl->qc_type;
1169         ioqc->qc_dqblk.dqb_valid = QIF_ILIMITS;
1170         ioqc->qc_dqblk.dqb_ihardlimit = flag ? MIN_QLIMIT : 0;
1171
1172         /* build lqs for mds */
1173         lqs = quota_search_lqs(LQS_KEY(oqctl->qc_type, oqctl->qc_id),
1174                                qctxt, flag ? 1 : 0);
1175         if (lqs && !IS_ERR(lqs)) {
1176                 if (flag)
1177                         lqs->lqs_flags |= QI_SET;
1178                 else
1179                         lqs->lqs_flags &= ~QI_SET;
1180                 lqs_putref(lqs);
1181         } else {
1182                 CERROR("fail to %s lqs for inode(%s id: %u)!\n",
1183                        flag ? "create" : "search",
1184                        oqctl->qc_type ? "group" : "user",
1185                        oqctl->qc_id);
1186                 GOTO(out, rc = PTR_ERR(lqs));
1187         }
1188
1189         /* set local limit to MIN_QLIMIT */
1190         rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, ioqc);
1191         if (rc)
1192                 GOTO(out, rc);
1193
1194         /* trigger local qunit pre-acquire */
1195         if (oqctl->qc_type == USRQUOTA)
1196                 id[USRQUOTA] = oqctl->qc_id;
1197         else
1198                 id[GRPQUOTA] = oqctl->qc_id;
1199
1200         rc = qctxt_adjust_qunit(obd, &obd->u.obt.obt_qctxt, id, 0, 0, NULL);
1201         if (rc == -EDQUOT || rc == -EBUSY) {
1202                 CDEBUG(D_QUOTA, "rc: %d.\n", rc);
1203                 rc = 0;
1204         }
1205         if (rc) {
1206                 CDEBUG(D_QUOTA,"error mds adjust local file quota! (rc:%d)\n",
1207                        rc);
1208                 GOTO(out, rc);
1209         }
1210         /* FIXME initialize all slaves in CMD */
1211         EXIT;
1212 out:
1213         if (ioqc)
1214                 OBD_FREE_PTR(ioqc);
1215         return rc;
1216 }
1217
1218 static int mds_init_slave_blimits(struct obd_device *obd,
1219                                   struct obd_quotactl *oqctl, int set)
1220 {
1221         struct obd_device_target *obt = &obd->u.obt;
1222         struct lustre_quota_ctxt *qctxt = &obt->obt_qctxt;
1223         struct mds_obd *mds = &obd->u.mds;
1224         struct obd_quotactl *ioqc;
1225         struct lustre_qunit_size *lqs;
1226         unsigned int id[MAXQUOTAS] = { 0, 0 };
1227         int rc;
1228         int flag;
1229         ENTRY;
1230
1231         /* if we are going to set zero limit, needn't init slaves */
1232         if (!oqctl->qc_dqblk.dqb_bhardlimit && !oqctl->qc_dqblk.dqb_bsoftlimit &&
1233             !set)
1234                 RETURN(0);
1235
1236         OBD_ALLOC_PTR(ioqc);
1237         if (!ioqc)
1238                 RETURN(-ENOMEM);
1239
1240         flag = oqctl->qc_dqblk.dqb_bhardlimit ||
1241                oqctl->qc_dqblk.dqb_bsoftlimit || !set;
1242         ioqc->qc_cmd = flag ? Q_INITQUOTA : Q_SETQUOTA;
1243         ioqc->qc_id = oqctl->qc_id;
1244         ioqc->qc_type = oqctl->qc_type;
1245         ioqc->qc_dqblk.dqb_valid = QIF_BLIMITS;
1246         ioqc->qc_dqblk.dqb_bhardlimit = flag ? MIN_QLIMIT : 0;
1247
1248         /* build lqs for mds */
1249         lqs = quota_search_lqs(LQS_KEY(oqctl->qc_type, oqctl->qc_id),
1250                                qctxt, flag ? 1 : 0);
1251         if (lqs && !IS_ERR(lqs)) {
1252                 if (flag)
1253                         lqs->lqs_flags |= QB_SET;
1254                 else
1255                         lqs->lqs_flags &= ~QB_SET;
1256                 lqs_putref(lqs);
1257         } else {
1258                 CERROR("fail to %s lqs for block(%s id: %u)!\n",
1259                        flag ? "create" : "search",
1260                        oqctl->qc_type ? "group" : "user",
1261                        oqctl->qc_id);
1262                 GOTO(out, rc = PTR_ERR(lqs));
1263         }
1264
1265         rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, ioqc);
1266         if (rc)
1267                 GOTO(out, rc);
1268
1269         /* trigger local qunit pre-acquire */
1270         if (oqctl->qc_type == USRQUOTA)
1271                 id[USRQUOTA] = oqctl->qc_id;
1272         else
1273                 id[GRPQUOTA] = oqctl->qc_id;
1274
1275         /* initialize all slave's limit */
1276         rc = obd_quotactl(mds->mds_osc_exp, ioqc);
1277
1278         rc = qctxt_adjust_qunit(obd, &obd->u.obt.obt_qctxt, id, 1, 0, NULL);
1279         if (rc == -EDQUOT || rc == -EBUSY) {
1280                 CDEBUG(D_QUOTA, "rc: %d.\n", rc);
1281                 rc = 0;
1282         }
1283         if (rc) {
1284                 CERROR("error mds adjust local block quota! (rc:%d)\n", rc);
1285                 GOTO(out, rc);
1286         }
1287
1288         EXIT;
1289 out:
1290         OBD_FREE_PTR(ioqc);
1291         return rc;
1292 }
1293
1294 static void adjust_lqs(struct obd_device *obd, struct quota_adjust_qunit *qaq)
1295 {
1296         struct lustre_quota_ctxt *qctxt = &obd->u.obt.obt_qctxt;
1297         int rc = 0;
1298
1299         QAQ_SET_CREATE_LQS(qaq);
1300         /* adjust local lqs */
1301         rc = quota_adjust_slave_lqs(qaq, qctxt);
1302         if (rc < 0)
1303                 CERROR("adjust master's qunit size failed!(rc=%d)\n", rc);
1304
1305         /* adjust remote lqs */
1306         if (QAQ_IS_ADJBLK(qaq)) {
1307                 rc = obd_quota_adjust_qunit(obd->u.mds.mds_osc_exp, qaq, qctxt);
1308                 if (rc < 0)
1309                         CERROR("adjust slaves' qunit size failed!(rc=%d)\n", rc);
1310
1311         }
1312 }
1313
1314 int mds_set_dqblk(struct obd_device *obd, struct obd_quotactl *oqctl)
1315 {
1316         struct mds_obd *mds = &obd->u.mds;
1317         struct lustre_quota_ctxt *qctxt = &mds->mds_obt.obt_qctxt;
1318         struct obd_device *lov_obd = class_exp2obd(mds->mds_osc_exp);
1319         struct lov_obd *lov = &lov_obd->u.lov;
1320         struct quota_adjust_qunit *oqaq = NULL;
1321         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
1322         __u64 ihardlimit, isoftlimit, bhardlimit, bsoftlimit;
1323         time_t btime, itime;
1324         struct lustre_dquot *dquot;
1325         struct obd_dqblk *dqblk = &oqctl->qc_dqblk;
1326         /* orig_set means if quota was set before; now_set means we are
1327          * setting/cancelling quota */
1328         int orig_set, now_set;
1329         int rc, rc2 = 0, flag = 0;
1330         ENTRY;
1331
1332         if (oqctl->qc_type != USRQUOTA &&
1333             oqctl->qc_type != GRPQUOTA)
1334                 RETURN(-EINVAL);
1335
1336         OBD_ALLOC_PTR(oqaq);
1337         if (!oqaq)
1338                 RETURN(-ENOMEM);
1339         cfs_down(&mds->mds_qonoff_sem);
1340         init_oqaq(oqaq, qctxt, oqctl->qc_id, oqctl->qc_type);
1341
1342         if (qinfo->qi_files[oqctl->qc_type] == NULL) {
1343                 CWARN("quota[%u] is off\n", oqctl->qc_type);
1344                 GOTO(out_sem, rc = -ESRCH);
1345         }
1346
1347         dquot = lustre_dqget(obd, qinfo, oqctl->qc_id, oqctl->qc_type);
1348         if (IS_ERR(dquot))
1349                 GOTO(out_sem, rc = PTR_ERR(dquot));
1350         DQUOT_DEBUG(dquot, "get dquot in mds_set_blk\n");
1351         QINFO_DEBUG(dquot->dq_info, "get dquot in mds_set_blk\n");
1352
1353         cfs_down(&dquot->dq_sem);
1354
1355         if (dquot->dq_status) {
1356                 cfs_up(&dquot->dq_sem);
1357                 lustre_dqput(dquot);
1358                 GOTO(out_sem, rc = -EBUSY);
1359         }
1360         dquot->dq_status |= DQ_STATUS_SET;
1361
1362         ihardlimit = dquot->dq_dqb.dqb_ihardlimit;
1363         isoftlimit = dquot->dq_dqb.dqb_isoftlimit;
1364         bhardlimit = dquot->dq_dqb.dqb_bhardlimit;
1365         bsoftlimit = dquot->dq_dqb.dqb_bsoftlimit;
1366         btime = dquot->dq_dqb.dqb_btime;
1367         itime = dquot->dq_dqb.dqb_itime;
1368
1369         if (dqblk->dqb_valid & QIF_BTIME)
1370                 dquot->dq_dqb.dqb_btime = dqblk->dqb_btime;
1371         if (dqblk->dqb_valid & QIF_ITIME)
1372                 dquot->dq_dqb.dqb_itime = dqblk->dqb_itime;
1373
1374         if (dqblk->dqb_valid & QIF_BLIMITS) {
1375                 dquot->dq_dqb.dqb_bhardlimit = dqblk->dqb_bhardlimit;
1376                 dquot->dq_dqb.dqb_bsoftlimit = dqblk->dqb_bsoftlimit;
1377                 /* clear usage (limit pool) */
1378                 if (!dquot->dq_dqb.dqb_bhardlimit &&
1379                     !dquot->dq_dqb.dqb_bsoftlimit)
1380                         dquot->dq_dqb.dqb_curspace = 0;
1381
1382                 /* clear grace time */
1383                 if (!dqblk->dqb_bsoftlimit ||
1384                     toqb(dquot->dq_dqb.dqb_curspace) <= dqblk->dqb_bsoftlimit)
1385                         dquot->dq_dqb.dqb_btime = 0;
1386                 /* set grace only if user hasn't provided his own */
1387                 else if (!(dqblk->dqb_valid & QIF_BTIME))
1388                         dquot->dq_dqb.dqb_btime = cfs_time_current_sec() +
1389                                 qinfo->qi_info[dquot->dq_type].dqi_bgrace;
1390
1391                 flag |= LQUOTA_FLAGS_ADJBLK;
1392         }
1393
1394         if (dqblk->dqb_valid & QIF_ILIMITS) {
1395                 dquot->dq_dqb.dqb_ihardlimit = dqblk->dqb_ihardlimit;
1396                 dquot->dq_dqb.dqb_isoftlimit = dqblk->dqb_isoftlimit;
1397                 /* clear usage (limit pool) */
1398                 if (!dquot->dq_dqb.dqb_ihardlimit &&
1399                     !dquot->dq_dqb.dqb_isoftlimit)
1400                         dquot->dq_dqb.dqb_curinodes = 0;
1401
1402                 if (!dqblk->dqb_isoftlimit ||
1403                     dquot->dq_dqb.dqb_curinodes <= dqblk->dqb_isoftlimit)
1404                         dquot->dq_dqb.dqb_itime = 0;
1405                 else if (!(dqblk->dqb_valid & QIF_ITIME))
1406                         dquot->dq_dqb.dqb_itime = cfs_time_current_sec() +
1407                                 qinfo->qi_info[dquot->dq_type].dqi_igrace;
1408
1409                 flag |= LQUOTA_FLAGS_ADJINO;
1410         }
1411         QAQ_DEBUG(oqaq, "before dquot_create_oqaq\n");
1412         rc = dquot_create_oqaq(qctxt, dquot, lov->desc.ld_tgt_count, 1,
1413                                flag, oqaq);
1414         QAQ_DEBUG(oqaq, "after dquot_create_oqaq\n");
1415         if (rc < 0)
1416                 CDEBUG(D_QUOTA, "adjust qunit size failed! (rc:%d)\n", rc);
1417
1418
1419         rc = fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
1420
1421         cfs_up(&dquot->dq_sem);
1422
1423         if (rc) {
1424                 CERROR("set limit failed! (rc:%d)\n", rc);
1425                 goto out;
1426         }
1427
1428         cfs_up(&mds->mds_qonoff_sem);
1429
1430         adjust_lqs(obd, oqaq);
1431
1432         orig_set = ihardlimit || isoftlimit;
1433         now_set  = dqblk->dqb_ihardlimit || dqblk->dqb_isoftlimit;
1434         if (dqblk->dqb_valid & QIF_ILIMITS && orig_set != now_set) {
1435                 cfs_down(&dquot->dq_sem);
1436                 dquot->dq_dqb.dqb_curinodes = 0;
1437                 cfs_up(&dquot->dq_sem);
1438                 rc = mds_init_slave_ilimits(obd, oqctl, orig_set);
1439                 if (rc) {
1440                         CERROR("init slave ilimits failed! (rc:%d)\n", rc);
1441                         goto revoke_out;
1442                 }
1443         }
1444
1445         orig_set = bhardlimit || bsoftlimit;
1446         now_set  = dqblk->dqb_bhardlimit || dqblk->dqb_bsoftlimit;
1447         if (dqblk->dqb_valid & QIF_BLIMITS && orig_set != now_set) {
1448                 cfs_down(&dquot->dq_sem);
1449                 dquot->dq_dqb.dqb_curspace = 0;
1450                 cfs_up(&dquot->dq_sem);
1451                 rc = mds_init_slave_blimits(obd, oqctl, orig_set);
1452                 if (rc) {
1453                         CERROR("init slave blimits failed! (rc:%d)\n", rc);
1454                         goto revoke_out;
1455                 }
1456         }
1457
1458 revoke_out:
1459         cfs_down(&mds->mds_qonoff_sem);
1460         cfs_down(&dquot->dq_sem);
1461         if (rc) {
1462                 /* cancel previous setting */
1463                 dquot->dq_dqb.dqb_ihardlimit = ihardlimit;
1464                 dquot->dq_dqb.dqb_isoftlimit = isoftlimit;
1465                 dquot->dq_dqb.dqb_bhardlimit = bhardlimit;
1466                 dquot->dq_dqb.dqb_bsoftlimit = bsoftlimit;
1467                 dquot->dq_dqb.dqb_btime = btime;
1468                 dquot->dq_dqb.dqb_itime = itime;
1469         }
1470         rc2 = fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
1471         cfs_up(&dquot->dq_sem);
1472
1473 out:
1474         cfs_down(&dquot->dq_sem);
1475         dquot->dq_status &= ~DQ_STATUS_SET;
1476         cfs_up(&dquot->dq_sem);
1477         lustre_dqput(dquot);
1478         EXIT;
1479 out_sem:
1480         cfs_up(&mds->mds_qonoff_sem);
1481
1482         if (oqaq)
1483                 OBD_FREE_PTR(oqaq);
1484
1485         return rc ? rc : rc2;
1486 }
1487
1488 static int mds_get_space(struct obd_device *obd, struct obd_quotactl *oqctl)
1489 {
1490         struct obd_quotactl *soqc;
1491         struct lvfs_run_ctxt saved;
1492         int rc, rc1;
1493         ENTRY;
1494
1495         OBD_ALLOC_PTR(soqc);
1496         if (!soqc)
1497                 RETURN(-ENOMEM);
1498
1499         soqc->qc_cmd = Q_GETOQUOTA;
1500         soqc->qc_id = oqctl->qc_id;
1501         soqc->qc_type = oqctl->qc_type;
1502
1503         /* get block usage from OSS */
1504         soqc->qc_dqblk.dqb_curspace = 0;
1505         rc = obd_quotactl(obd->u.mds.mds_osc_exp, soqc);
1506         if (!rc || rc == -EREMOTEIO) {
1507                 oqctl->qc_dqblk.dqb_curspace = soqc->qc_dqblk.dqb_curspace;
1508                 oqctl->qc_dqblk.dqb_valid |= QIF_SPACE;
1509         }
1510
1511         /* get block/inode usage from MDS */
1512         soqc->qc_dqblk.dqb_curspace = 0;
1513         soqc->qc_dqblk.dqb_curinodes = 0;
1514         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
1515         rc1 = fsfilt_quotactl(obd, obd->u.obt.obt_sb, soqc);
1516         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
1517         if (!rc1) {
1518                 oqctl->qc_dqblk.dqb_curspace += soqc->qc_dqblk.dqb_curspace;
1519                 oqctl->qc_dqblk.dqb_curinodes = soqc->qc_dqblk.dqb_curinodes;
1520                 oqctl->qc_dqblk.dqb_valid |= QIF_INODES;
1521         }
1522
1523         OBD_FREE_PTR(soqc);
1524
1525         RETURN(rc ? : rc1);
1526 }
1527
1528 int mds_get_dqblk(struct obd_device *obd, struct obd_quotactl *oqctl)
1529 {
1530         struct mds_obd *mds = &obd->u.mds;
1531         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
1532         struct lustre_dquot *dquot;
1533         struct obd_dqblk *dqblk = &oqctl->qc_dqblk;
1534         int rc;
1535         ENTRY;
1536
1537         if (oqctl->qc_type != USRQUOTA &&
1538             oqctl->qc_type != GRPQUOTA)
1539                 RETURN(-EINVAL);
1540
1541         cfs_down(&mds->mds_qonoff_sem);
1542         dqblk->dqb_valid = 0;
1543         if (qinfo->qi_files[oqctl->qc_type] == NULL) {
1544                 CWARN("quota[%u] is off\n", oqctl->qc_type);
1545                 GOTO(out, rc = -ESRCH);
1546         }
1547
1548         dquot = lustre_dqget(obd, qinfo, oqctl->qc_id, oqctl->qc_type);
1549         if (IS_ERR(dquot))
1550                 GOTO(out, rc = PTR_ERR(dquot));
1551
1552         cfs_down(&dquot->dq_sem);
1553         dqblk->dqb_ihardlimit = dquot->dq_dqb.dqb_ihardlimit;
1554         dqblk->dqb_isoftlimit = dquot->dq_dqb.dqb_isoftlimit;
1555         dqblk->dqb_bhardlimit = dquot->dq_dqb.dqb_bhardlimit;
1556         dqblk->dqb_bsoftlimit = dquot->dq_dqb.dqb_bsoftlimit;
1557         dqblk->dqb_btime = dquot->dq_dqb.dqb_btime;
1558         dqblk->dqb_itime = dquot->dq_dqb.dqb_itime;
1559         dqblk->dqb_valid |= QIF_LIMITS | QIF_TIMES;
1560         cfs_up(&dquot->dq_sem);
1561
1562         lustre_dqput(dquot);
1563         cfs_up(&mds->mds_qonoff_sem);
1564
1565         /* the usages in admin quota file is inaccurate */
1566         dqblk->dqb_curinodes = 0;
1567         dqblk->dqb_curspace = 0;
1568         rc = mds_get_space(obd, oqctl);
1569         EXIT;
1570         return rc;
1571 out:
1572         cfs_up(&mds->mds_qonoff_sem);
1573         return rc;
1574 }
1575
1576 int mds_get_obd_quota(struct obd_device *obd, struct obd_quotactl *oqctl)
1577 {
1578         struct lvfs_run_ctxt saved;
1579         int rc;
1580         ENTRY;
1581
1582         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
1583         rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
1584         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
1585
1586         RETURN(rc);
1587 }
1588
1589
1590 /* FIXME we only recovery block limit by now, need recovery inode
1591  * limits also after CMD involved in */
1592 static int 
1593 dquot_recovery(struct obd_device *obd, unsigned int id, unsigned short type)
1594 {
1595         struct mds_obd *mds = &obd->u.mds;
1596         struct lustre_quota_info *qinfo= &mds->mds_quota_info;
1597         struct lustre_dquot *dquot;
1598         struct obd_quotactl *qctl;
1599         __u64 total_limits = 0;
1600         int rc;
1601         ENTRY;
1602
1603         OBD_ALLOC_PTR(qctl);
1604         if (qctl == NULL)
1605                 RETURN(-ENOMEM);
1606
1607         dquot = lustre_dqget(obd, qinfo, id, type);
1608         if (IS_ERR(dquot)) {
1609                 CERROR("Get dquot failed. (rc:%ld)\n", PTR_ERR(dquot));
1610                 OBD_FREE_PTR(qctl);
1611                 RETURN(PTR_ERR(dquot));
1612         }
1613
1614         cfs_down(&dquot->dq_sem);
1615
1616         /* don't recovery the dquot without limits or under setting */
1617         if (!(dquot->dq_dqb.dqb_bhardlimit || dquot->dq_dqb.dqb_bsoftlimit) ||
1618             dquot->dq_status)
1619                 GOTO(skip, rc = 0);
1620         dquot->dq_status |= DQ_STATUS_RECOVERY;
1621
1622         cfs_up(&dquot->dq_sem);
1623
1624         /* get real bhardlimit from all slaves. */
1625         qctl->qc_cmd = Q_GETOQUOTA;
1626         qctl->qc_type = type;
1627         qctl->qc_id = id;
1628         qctl->qc_stat = QUOTA_RECOVERING;
1629         rc = obd_quotactl(mds->mds_osc_exp, qctl);
1630         if (rc)
1631                 GOTO(out, rc);
1632         total_limits = qctl->qc_dqblk.dqb_bhardlimit;
1633
1634         /* get real bhardlimit from master */
1635         rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, qctl);
1636         if (rc)
1637                 GOTO(out, rc);
1638         total_limits += qctl->qc_dqblk.dqb_bhardlimit;
1639
1640         /* amend the usage of the administrative quotafile */
1641         cfs_down(&mds->mds_qonoff_sem);
1642         cfs_down(&dquot->dq_sem);
1643
1644         dquot->dq_dqb.dqb_curspace = total_limits << QUOTABLOCK_BITS;
1645
1646         rc = fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
1647         if (rc)
1648                 CERROR("write dquot failed! (rc:%d)\n", rc);
1649
1650         cfs_up(&dquot->dq_sem);
1651         cfs_up(&mds->mds_qonoff_sem);
1652         EXIT;
1653 out:
1654         cfs_down(&dquot->dq_sem);
1655         dquot->dq_status &= ~DQ_STATUS_RECOVERY;
1656 skip:
1657         cfs_up(&dquot->dq_sem);
1658
1659         lustre_dqput(dquot);
1660         OBD_FREE_PTR(qctl);
1661         return rc;
1662 }
1663
1664 struct qmaster_recov_thread_data {
1665         struct obd_device *obd;
1666         cfs_completion_t comp;
1667 };
1668
1669 static int qmaster_recovery_main(void *arg)
1670 {
1671         struct qmaster_recov_thread_data *data = arg;
1672         struct obd_device *obd = data->obd;
1673         struct mds_obd *mds = &obd->u.mds;
1674         struct lustre_quota_info *qinfo = &mds->mds_quota_info;
1675         int rc = 0;
1676         unsigned short type;
1677         ENTRY;
1678
1679         cfs_daemonize_ctxt("qmaster_recovd");
1680
1681         /* for mds */
1682         class_incref(obd, "qmaster_recovd_mds", obd);
1683         /* for lov */
1684         class_incref(mds->mds_osc_obd, "qmaster_recovd_lov", mds->mds_osc_obd);
1685
1686         cfs_complete(&data->comp);
1687
1688         for (type = USRQUOTA; type < MAXQUOTAS; type++) {
1689                 cfs_list_t id_list;
1690                 struct dquot_id *dqid, *tmp;
1691
1692                 cfs_down(&mds->mds_qonoff_sem);
1693                 if (qinfo->qi_files[type] == NULL) {
1694                         cfs_up(&mds->mds_qonoff_sem);
1695                         continue;
1696                 }
1697                 CFS_INIT_LIST_HEAD(&id_list);
1698                 rc = fsfilt_qids(obd, qinfo->qi_files[type], NULL, type,
1699                                  &id_list);
1700                 cfs_up(&mds->mds_qonoff_sem);
1701
1702                 if (rc)
1703                         CERROR("error get ids from admin quotafile.(%d)\n", rc);
1704
1705                 cfs_list_for_each_entry_safe(dqid, tmp, &id_list, di_link) {
1706                         cfs_list_del_init(&dqid->di_link);
1707                         if (rc)
1708                                 goto free;
1709
1710                         rc = dquot_recovery(obd, dqid->di_id, type);
1711                         if (rc)
1712                                 CERROR("%s: qmaster recovery failed for %sid %d"
1713                                        " rc:%d)\n", obd->obd_name,
1714                                        type ? "g" : "u", dqid->di_id, rc);
1715 free:
1716                         OBD_FREE_PTR(dqid);
1717                 }
1718         }
1719         class_decref(mds->mds_osc_obd, "qmaster_recovd_lov", mds->mds_osc_obd);
1720         class_decref(obd, "qmaster_recovd_mds", obd);
1721         RETURN(rc);
1722 }
1723
1724 int mds_quota_recovery(struct obd_device *obd)
1725 {
1726         struct mds_obd *mds = &obd->u.mds;
1727         struct qmaster_recov_thread_data data;
1728         int rc = 0;
1729         ENTRY;
1730
1731         if (unlikely(!mds->mds_quota || obd->obd_stopping))
1732                 RETURN(rc);
1733
1734         cfs_mutex_down(&obd->obd_dev_sem);
1735         if (mds->mds_lov_desc.ld_active_tgt_count != mds->mds_lov_objid_count) {
1736                 CWARN("Only %u/%u OSTs are active, abort quota recovery\n",
1737                       mds->mds_lov_desc.ld_active_tgt_count,
1738                       mds->mds_lov_objid_count);
1739                 cfs_mutex_up(&obd->obd_dev_sem);
1740                 RETURN(rc);
1741         }
1742         cfs_mutex_up(&obd->obd_dev_sem);
1743
1744         data.obd = obd;
1745         cfs_init_completion(&data.comp);
1746
1747         rc = cfs_kernel_thread(qmaster_recovery_main, &data,
1748                                CLONE_VM|CLONE_FILES);
1749         if (rc < 0)
1750                 CERROR("%s: cannot start quota recovery thread: rc %d\n",
1751                        obd->obd_name, rc);
1752
1753         cfs_wait_for_completion(&data.comp);
1754         RETURN(rc);
1755 }
1756
1757 #endif /* HAVE_QUOTA_SUPPORT */