1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2 * vim:expandtab:shiftwidth=8:tabstop=8:
4 * lustre/quota/quota_master.c
5 * Lustre Quota Master request handler
7 * Copyright (c) 2001-2005 Cluster File Systems, Inc.
8 * Author: Niu YaWei <niu@clusterfs.com>
10 * This file is part of Lustre, http://www.lustre.org.
12 * No redistribution or use is permitted outside of Cluster File Systems, Inc.
16 # define EXPORT_SYMTAB
19 #define DEBUG_SUBSYSTEM S_MDS
21 #include <linux/version.h>
23 #include <asm/unistd.h>
24 #include <linux/slab.h>
25 #include <linux/quotaops.h>
26 #include <linux/module.h>
27 #include <linux/init.h>
28 #include <linux/quota.h>
30 #include <obd_class.h>
31 #include <lustre_quota.h>
32 #include <lustre_fsfilt.h>
33 #include <lustre_mds.h>
35 #include "quota_internal.h"
38 * mds->mds_qonoff_sem > dquot->dq_sem */
39 static struct list_head lustre_dquot_hash[NR_DQHASH];
40 static spinlock_t dquot_hash_lock = SPIN_LOCK_UNLOCKED;
42 cfs_mem_cache_t *lustre_dquot_cachep;
44 int lustre_dquot_init(void)
49 LASSERT(lustre_dquot_cachep == NULL);
50 lustre_dquot_cachep = cfs_mem_cache_create("lustre_dquot_cache",
51 sizeof(struct lustre_dquot),
53 if (!lustre_dquot_cachep)
56 for (i = 0; i < NR_DQHASH; i++) {
57 INIT_LIST_HEAD(lustre_dquot_hash + i);
62 void lustre_dquot_exit(void)
66 /* FIXME cleanup work ?? */
68 for (i = 0; i < NR_DQHASH; i++) {
69 LASSERT(list_empty(lustre_dquot_hash + i));
71 if (lustre_dquot_cachep) {
73 rc = cfs_mem_cache_destroy(lustre_dquot_cachep);
74 LASSERTF(rc == 0,"couldn't destroy lustre_dquot_cachep slab\n");
75 lustre_dquot_cachep = NULL;
81 dquot_hashfn(struct lustre_quota_info *info, unsigned int id, int type)
82 __attribute__((__const__));
85 dquot_hashfn(struct lustre_quota_info *info, unsigned int id, int type)
87 unsigned long tmp = ((unsigned long)info >> L1_CACHE_SHIFT) ^ id;
88 tmp = (tmp * (MAXQUOTAS - type)) % NR_DQHASH;
92 /* caller must hold dquot_hash_lock */
93 static struct lustre_dquot *find_dquot(int hashent,
94 struct lustre_quota_info *lqi, qid_t id,
97 struct lustre_dquot *dquot;
100 LASSERT_SPIN_LOCKED(&dquot_hash_lock);
101 list_for_each_entry(dquot, &lustre_dquot_hash[hashent], dq_hash) {
102 if (dquot->dq_info == lqi &&
103 dquot->dq_id == id && dquot->dq_type == type)
109 static struct lustre_dquot *alloc_dquot(struct lustre_quota_info *lqi,
112 struct lustre_dquot *dquot = NULL;
115 OBD_SLAB_ALLOC(dquot, lustre_dquot_cachep, CFS_ALLOC_IO, sizeof(*dquot));
119 INIT_LIST_HEAD(&dquot->dq_hash);
120 init_mutex_locked(&dquot->dq_sem);
121 dquot->dq_refcnt = 1;
122 dquot->dq_info = lqi;
124 dquot->dq_type = type;
125 dquot->dq_status = DQ_STATUS_AVAIL;
130 static void free_dquot(struct lustre_dquot *dquot)
132 OBD_SLAB_FREE(dquot, lustre_dquot_cachep, sizeof(*dquot));
135 static void insert_dquot_nolock(struct lustre_dquot *dquot)
137 struct list_head *head = lustre_dquot_hash +
138 dquot_hashfn(dquot->dq_info, dquot->dq_id, dquot->dq_type);
139 LASSERT(list_empty(&dquot->dq_hash));
140 list_add(&dquot->dq_hash, head);
143 static void remove_dquot_nolock(struct lustre_dquot *dquot)
145 LASSERT(!list_empty(&dquot->dq_hash));
146 list_del_init(&dquot->dq_hash);
149 static void lustre_dqput(struct lustre_dquot *dquot)
152 spin_lock(&dquot_hash_lock);
153 LASSERT(dquot->dq_refcnt);
155 if (!dquot->dq_refcnt) {
156 remove_dquot_nolock(dquot);
159 spin_unlock(&dquot_hash_lock);
163 static struct lustre_dquot *lustre_dqget(struct obd_device *obd,
164 struct lustre_quota_info *lqi,
167 unsigned int hashent = dquot_hashfn(lqi, id, type);
168 struct lustre_dquot *dquot, *empty;
171 if ((empty = alloc_dquot(lqi, id, type)) == NULL)
172 RETURN(ERR_PTR(-ENOMEM));
174 spin_lock(&dquot_hash_lock);
175 if ((dquot = find_dquot(hashent, lqi, id, type)) != NULL) {
177 spin_unlock(&dquot_hash_lock);
183 insert_dquot_nolock(dquot);
184 spin_unlock(&dquot_hash_lock);
186 rc = fsfilt_dquot(obd, dquot, QFILE_RD_DQUOT);
189 CERROR("can't read dquot from admin quotafile! "
201 int dqacq_handler(struct obd_device *obd, struct qunit_data *qdata, int opc)
203 struct mds_obd *mds = &obd->u.mds;
204 struct lustre_quota_info *info = &mds->mds_quota_info;
205 struct lustre_dquot *dquot = NULL;
207 __u32 hlimit = 0, slimit = 0;
208 __u32 qdata_type = qdata->qd_flags & QUOTA_IS_GRP;
209 __u32 is_blk = (qdata->qd_flags & QUOTA_IS_BLOCK) >> 1;
211 unsigned int grace = 0;
215 OBD_FAIL_RETURN(OBD_FAIL_OBD_DQACQ, -EIO);
217 dquot = lustre_dqget(obd, info, qdata->qd_id, qdata_type);
219 RETURN(PTR_ERR(dquot));
221 DQUOT_DEBUG(dquot, "get dquot in dqacq_handler\n");
222 QINFO_DEBUG(dquot->dq_info, "get dquot in dqadq_handler\n");
224 down(&mds->mds_qonoff_sem);
225 down(&dquot->dq_sem);
227 if (dquot->dq_status & DQ_STATUS_RECOVERY) {
228 DQUOT_DEBUG(dquot, "this dquot is under recovering.\n");
229 GOTO(out, rc = -EBUSY);
233 grace = info->qi_info[qdata_type].dqi_bgrace;
234 usage = &dquot->dq_dqb.dqb_curspace;
235 hlimit = dquot->dq_dqb.dqb_bhardlimit;
236 slimit = dquot->dq_dqb.dqb_bsoftlimit;
237 time = &dquot->dq_dqb.dqb_btime;
239 grace = info->qi_info[qdata_type].dqi_igrace;
240 usage = (__u64 *) & dquot->dq_dqb.dqb_curinodes;
241 hlimit = dquot->dq_dqb.dqb_ihardlimit;
242 slimit = dquot->dq_dqb.dqb_isoftlimit;
243 time = &dquot->dq_dqb.dqb_itime;
246 /* if the quota limit in admin quotafile is zero, we just inform
247 * slave to clear quota limit with zero qd_count */
248 if (hlimit == 0 && slimit == 0) {
256 QUSG(*usage + qdata->qd_count, is_blk) > hlimit)
257 GOTO(out, rc = -EDQUOT);
260 QUSG(*usage + qdata->qd_count, is_blk) > slimit) {
261 if (*time && cfs_time_current_sec() >= *time)
262 GOTO(out, rc = -EDQUOT);
264 *time = cfs_time_current_sec() + grace;
267 *usage += qdata->qd_count;
270 /* The usage in administrative file might be incorrect before
272 if (*usage - qdata->qd_count < 0)
275 *usage -= qdata->qd_count;
277 /* (usage <= soft limit) but not (usage < soft limit) */
278 if (!slimit || QUSG(*usage, is_blk) <= slimit)
285 rc = fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
289 up(&mds->mds_qonoff_sem);
294 int mds_quota_adjust(struct obd_device *obd, unsigned int qcids[],
295 unsigned int qpids[], int rc, int opc)
297 struct lustre_quota_ctxt *qctxt = &obd->u.obt.obt_qctxt;
301 if (rc && rc != -EDQUOT)
305 case FSFILT_OP_RENAME:
306 /* acquire/release block quota on owner of original parent */
307 rc2 = qctxt_adjust_qunit(obd, qctxt, qpids[2], qpids[3], 1, 0);
309 case FSFILT_OP_SETATTR:
310 /* acquire/release file quota on original owner */
311 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids[0], qpids[1], 0, 0);
313 case FSFILT_OP_CREATE:
314 case FSFILT_OP_UNLINK:
315 /* acquire/release file/block quota on owner of child (or current owner) */
316 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids[0], qcids[1], 0, 0);
317 rc2 |= qctxt_adjust_qunit(obd, qctxt, qcids[0], qcids[1], 1, 0);
318 /* acquire/release block quota on owner of parent (or original owner) */
319 rc2 |= qctxt_adjust_qunit(obd, qctxt, qpids[0], qpids[1], 1, 0);
327 CERROR("mds adjust qunit failed! (opc:%d rc:%d)\n", opc, rc2);
331 int filter_quota_adjust(struct obd_device *obd, unsigned int qcids[],
332 unsigned int qpids[], int rc, int opc)
334 struct lustre_quota_ctxt *qctxt = &obd->u.obt.obt_qctxt;
338 if (rc && rc != -EDQUOT)
342 case FSFILT_OP_SETATTR:
343 /* acquire/release block quota on original & current owner */
344 rc = qctxt_adjust_qunit(obd, qctxt, qcids[0], qcids[1], 1, 0);
345 rc2 = qctxt_adjust_qunit(obd, qctxt, qpids[0], qpids[1], 1, 0);
347 case FSFILT_OP_UNLINK:
348 /* release block quota on this owner */
349 case FSFILT_OP_CREATE: /* XXX for write operation on obdfilter */
350 /* acquire block quota on this owner */
351 rc = qctxt_adjust_qunit(obd, qctxt, qcids[0], qcids[1], 1, 0);
359 CERROR("filter adjust qunit failed! (opc:%d rc%d)\n",
364 #define LUSTRE_ADMIN_QUOTAFILES {\
365 "admin_quotafile.usr", /* user admin quotafile */\
366 "admin_quotafile.grp" /* group admin quotafile */\
368 static const char prefix[] = "OBJECTS/";
370 int init_admin_quotafiles(struct obd_device *obd, struct obd_quotactl *oqctl)
372 struct mds_obd *mds = &obd->u.mds;
373 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
374 const char *quotafiles[] = LUSTRE_ADMIN_QUOTAFILES;
375 struct lvfs_run_ctxt saved;
378 struct dentry *dparent = mds->mds_objects_dir;
379 struct inode *iparent = dparent->d_inode;
383 push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
385 down(&mds->mds_qonoff_sem);
386 for (i = 0; i < MAXQUOTAS; i++) {
390 if (!Q_TYPESET(oqctl, i))
393 /* quota file has been opened ? */
394 if (qinfo->qi_files[i]) {
395 CWARN("init %s admin quotafile while quota on.\n",
396 i == USRQUOTA ? "user" : "group");
400 /* lookup quota file */
402 LOCK_INODE_MUTEX(iparent);
403 de = lookup_one_len(quotafiles[i], dparent,
404 strlen(quotafiles[i]));
405 UNLOCK_INODE_MUTEX(iparent);
406 if (IS_ERR(de) || de->d_inode == NULL ||
407 !S_ISREG(de->d_inode->i_mode))
408 rc = IS_ERR(de) ? PTR_ERR(de) : -ENOENT;
412 if (rc && rc != -ENOENT) {
413 CERROR("error lookup quotafile %s! (rc:%d)\n",
420 LASSERT(strlen(quotafiles[i]) + sizeof(prefix) <= sizeof(name));
421 sprintf(name, "%s%s", prefix, quotafiles[i]);
423 LASSERT(rc == -ENOENT);
424 /* create quota file */
425 fp = filp_open(name, O_CREAT | O_EXCL, 0644);
426 if (IS_ERR(fp) || !S_ISREG(fp->f_dentry->d_inode->i_mode)) {
428 CERROR("error creating admin quotafile %s (rc:%d)\n",
433 qinfo->qi_files[i] = fp;
434 rc = fsfilt_quotainfo(obd, qinfo, i, QFILE_INIT_INFO);
436 qinfo->qi_files[i] = NULL;
439 CERROR("error init %s admin quotafile! (rc:%d)\n",
440 i == USRQUOTA ? "user" : "group", rc);
444 up(&mds->mds_qonoff_sem);
446 pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
450 static int close_quota_files(struct obd_quotactl *oqctl,
451 struct lustre_quota_info *qinfo)
456 for (i = 0; i < MAXQUOTAS; i++) {
457 if (!Q_TYPESET(oqctl, i))
459 if (qinfo->qi_files[i] == NULL) {
463 filp_close(qinfo->qi_files[i], 0);
464 qinfo->qi_files[i] = NULL;
469 int mds_admin_quota_on(struct obd_device *obd, struct obd_quotactl *oqctl)
471 struct mds_obd *mds = &obd->u.mds;
472 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
473 const char *quotafiles[] = LUSTRE_ADMIN_QUOTAFILES;
476 struct inode *iparent = mds->mds_objects_dir->d_inode;
481 /* open admin quota files and read quotafile info */
482 for (i = 0; i < MAXQUOTAS; i++) {
485 if (!Q_TYPESET(oqctl, i))
488 LASSERT(strlen(quotafiles[i]) + sizeof(prefix) <= sizeof(name));
489 sprintf(name, "%s%s", prefix, quotafiles[i]);
491 if (qinfo->qi_files[i] != NULL) {
496 fp = filp_open(name, O_RDWR | O_EXCL, 0644);
497 if (IS_ERR(fp) || !S_ISREG(fp->f_dentry->d_inode->i_mode)) {
499 CDEBUG(rc == -ENOENT ? D_QUOTA : D_ERROR,
500 "open %s failed! (rc:%d)\n", name, rc);
503 qinfo->qi_files[i] = fp;
505 rc = fsfilt_quotainfo(obd, qinfo, i, QFILE_RD_INFO);
507 CERROR("error read quotainfo of %s! (rc:%d)\n",
513 if (rc && rc != -EBUSY)
514 close_quota_files(oqctl, qinfo);
519 static int mds_admin_quota_off(struct obd_device *obd,
520 struct obd_quotactl *oqctl)
522 struct mds_obd *mds = &obd->u.mds;
523 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
527 /* close admin quota files */
528 rc = close_quota_files(oqctl, qinfo);
532 int mds_quota_on(struct obd_device *obd, struct obd_quotactl *oqctl)
534 struct mds_obd *mds = &obd->u.mds;
535 struct obd_device_target *obt = &obd->u.obt;
536 struct lvfs_run_ctxt saved;
540 if (!atomic_dec_and_test(&obt->obt_quotachecking)) {
541 CDEBUG(D_INFO, "other people are doing quotacheck\n");
542 atomic_inc(&obt->obt_quotachecking);
546 down(&mds->mds_qonoff_sem);
547 push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
548 rc = mds_admin_quota_on(obd, oqctl);
552 rc = obd_quotactl(mds->mds_osc_exp, oqctl);
556 rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
558 obt->obt_qctxt.lqc_status = 1;
560 pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
561 up(&mds->mds_qonoff_sem);
562 atomic_inc(&obt->obt_quotachecking);
566 int mds_quota_off(struct obd_device *obd, struct obd_quotactl *oqctl)
568 struct mds_obd *mds = &obd->u.mds;
569 struct obd_device_target *obt = &obd->u.obt;
570 struct lvfs_run_ctxt saved;
574 if (!atomic_dec_and_test(&obt->obt_quotachecking)) {
575 CDEBUG(D_INFO, "other people are doing quotacheck\n");
576 atomic_inc(&obt->obt_quotachecking);
580 down(&mds->mds_qonoff_sem);
581 /* close admin quota files */
582 push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
583 mds_admin_quota_off(obd, oqctl);
585 rc = obd_quotactl(mds->mds_osc_exp, oqctl);
586 rc2 = fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
588 obt->obt_qctxt.lqc_status = 0;
590 pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
591 up(&mds->mds_qonoff_sem);
592 atomic_inc(&obt->obt_quotachecking);
597 int mds_set_dqinfo(struct obd_device *obd, struct obd_quotactl *oqctl)
599 struct mds_obd *mds = &obd->u.mds;
600 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
601 struct obd_dqinfo *dqinfo = &oqctl->qc_dqinfo;
605 down(&mds->mds_qonoff_sem);
606 if (qinfo->qi_files[oqctl->qc_type] == NULL) {
611 qinfo->qi_info[oqctl->qc_type].dqi_bgrace = dqinfo->dqi_bgrace;
612 qinfo->qi_info[oqctl->qc_type].dqi_igrace = dqinfo->dqi_igrace;
613 qinfo->qi_info[oqctl->qc_type].dqi_flags = dqinfo->dqi_flags;
615 rc = fsfilt_quotainfo(obd, qinfo, oqctl->qc_type, QFILE_WR_INFO);
618 up(&mds->mds_qonoff_sem);
622 int mds_get_dqinfo(struct obd_device *obd, struct obd_quotactl *oqctl)
624 struct mds_obd *mds = &obd->u.mds;
625 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
626 struct obd_dqinfo *dqinfo = &oqctl->qc_dqinfo;
630 down(&mds->mds_qonoff_sem);
631 if (qinfo->qi_files[oqctl->qc_type] == NULL) {
636 dqinfo->dqi_bgrace = qinfo->qi_info[oqctl->qc_type].dqi_bgrace;
637 dqinfo->dqi_igrace = qinfo->qi_info[oqctl->qc_type].dqi_igrace;
638 dqinfo->dqi_flags = qinfo->qi_info[oqctl->qc_type].dqi_flags;
641 up(&mds->mds_qonoff_sem);
645 static int mds_init_slave_ilimits(struct obd_device *obd,
646 struct obd_quotactl *oqctl, int set)
648 /* XXX: for file limits only adjust local now */
649 unsigned int uid = 0, gid = 0;
650 struct obd_quotactl *ioqc = NULL;
655 /* if we are going to set zero limit, needn't init slaves */
656 if (!oqctl->qc_dqblk.dqb_ihardlimit && !oqctl->qc_dqblk.dqb_isoftlimit &&
664 flag = oqctl->qc_dqblk.dqb_ihardlimit ||
665 oqctl->qc_dqblk.dqb_isoftlimit || set;
666 ioqc->qc_cmd = flag ? Q_INITQUOTA : Q_SETQUOTA;
667 ioqc->qc_id = oqctl->qc_id;
668 ioqc->qc_type = oqctl->qc_type;
669 ioqc->qc_dqblk.dqb_valid = QIF_ILIMITS;
670 ioqc->qc_dqblk.dqb_ihardlimit = flag ? MIN_QLIMIT : 0;
672 /* set local limit to MIN_QLIMIT */
673 rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, ioqc);
677 /* trigger local qunit pre-acquire */
678 if (oqctl->qc_type == USRQUOTA)
683 rc = qctxt_adjust_qunit(obd, &obd->u.obt.obt_qctxt, uid, gid, 0, 0);
685 CERROR("error mds adjust local file quota! (rc:%d)\n", rc);
688 /* FIXME initialize all slaves in CMD */
696 static int mds_init_slave_blimits(struct obd_device *obd,
697 struct obd_quotactl *oqctl, int set)
699 struct mds_obd *mds = &obd->u.mds;
700 struct obd_quotactl *ioqc;
701 unsigned int uid = 0, gid = 0;
706 /* if we are going to set zero limit, needn't init slaves */
707 if (!oqctl->qc_dqblk.dqb_bhardlimit && !oqctl->qc_dqblk.dqb_bsoftlimit &&
715 flag = oqctl->qc_dqblk.dqb_bhardlimit ||
716 oqctl->qc_dqblk.dqb_bsoftlimit || set;
717 ioqc->qc_cmd = flag ? Q_INITQUOTA : Q_SETQUOTA;
718 ioqc->qc_id = oqctl->qc_id;
719 ioqc->qc_type = oqctl->qc_type;
720 ioqc->qc_dqblk.dqb_valid = QIF_BLIMITS;
721 ioqc->qc_dqblk.dqb_bhardlimit = flag ? MIN_QLIMIT : 0;
723 rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, ioqc);
727 /* trigger local qunit pre-acquire */
728 if (oqctl->qc_type == USRQUOTA)
733 rc = qctxt_adjust_qunit(obd, &obd->u.obt.obt_qctxt, uid, gid, 1, 0);
735 CERROR("error mds adjust local block quota! (rc:%d)\n", rc);
739 /* initialize all slave's limit */
740 rc = obd_quotactl(mds->mds_osc_exp, ioqc);
747 int mds_set_dqblk(struct obd_device *obd, struct obd_quotactl *oqctl)
749 struct mds_obd *mds = &obd->u.mds;
750 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
751 __u32 ihardlimit, isoftlimit, bhardlimit, bsoftlimit;
753 struct lustre_dquot *dquot;
754 struct obd_dqblk *dqblk = &oqctl->qc_dqblk;
758 down(&mds->mds_qonoff_sem);
759 if (qinfo->qi_files[oqctl->qc_type] == NULL)
760 GOTO(out_sem, rc = -ESRCH);
762 dquot = lustre_dqget(obd, qinfo, oqctl->qc_id, oqctl->qc_type);
764 GOTO(out_sem, rc = PTR_ERR(dquot));
765 DQUOT_DEBUG(dquot, "get dquot in mds_set_blk\n");
766 QINFO_DEBUG(dquot->dq_info, "get dquot in mds_set_blk\n");
768 down(&dquot->dq_sem);
770 if (dquot->dq_status) {
773 GOTO(out_sem, rc = -EBUSY);
775 dquot->dq_status |= DQ_STATUS_SET;
777 ihardlimit = dquot->dq_dqb.dqb_ihardlimit;
778 isoftlimit = dquot->dq_dqb.dqb_isoftlimit;
779 bhardlimit = dquot->dq_dqb.dqb_bhardlimit;
780 bsoftlimit = dquot->dq_dqb.dqb_bsoftlimit;
781 btime = dquot->dq_dqb.dqb_btime;
782 itime = dquot->dq_dqb.dqb_itime;
784 if (dqblk->dqb_valid & QIF_BTIME)
785 dquot->dq_dqb.dqb_btime = dqblk->dqb_btime;
786 if (dqblk->dqb_valid & QIF_ITIME)
787 dquot->dq_dqb.dqb_itime = dqblk->dqb_itime;
789 if (dqblk->dqb_valid & QIF_BLIMITS) {
790 dquot->dq_dqb.dqb_bhardlimit = dqblk->dqb_bhardlimit;
791 dquot->dq_dqb.dqb_bsoftlimit = dqblk->dqb_bsoftlimit;
792 /* clear usage (limit pool) */
793 if (!dquot->dq_dqb.dqb_bhardlimit &&
794 !dquot->dq_dqb.dqb_bsoftlimit)
795 dquot->dq_dqb.dqb_curspace = 0;
797 /* clear grace time */
798 if (!dqblk->dqb_bsoftlimit ||
799 toqb(dquot->dq_dqb.dqb_curspace) <= dqblk->dqb_bsoftlimit)
800 dquot->dq_dqb.dqb_btime = 0;
801 /* set grace only if user hasn't provided his own */
802 else if (!(dqblk->dqb_valid & QIF_BTIME))
803 dquot->dq_dqb.dqb_btime = cfs_time_current_sec() +
804 qinfo->qi_info[dquot->dq_type].dqi_bgrace;
807 if (dqblk->dqb_valid & QIF_ILIMITS) {
808 dquot->dq_dqb.dqb_ihardlimit = dqblk->dqb_ihardlimit;
809 dquot->dq_dqb.dqb_isoftlimit = dqblk->dqb_isoftlimit;
810 /* clear usage (limit pool) */
811 if (!dquot->dq_dqb.dqb_ihardlimit &&
812 !dquot->dq_dqb.dqb_isoftlimit)
813 dquot->dq_dqb.dqb_curinodes = 0;
815 if (!dqblk->dqb_isoftlimit ||
816 dquot->dq_dqb.dqb_curinodes <= dqblk->dqb_isoftlimit)
817 dquot->dq_dqb.dqb_itime = 0;
818 else if (!(dqblk->dqb_valid & QIF_ITIME))
819 dquot->dq_dqb.dqb_itime = cfs_time_current_sec() +
820 qinfo->qi_info[dquot->dq_type].dqi_igrace;
823 rc = fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
828 CERROR("set limit failed! (rc:%d)\n", rc);
832 up(&mds->mds_qonoff_sem);
833 if (dqblk->dqb_valid & QIF_ILIMITS) {
834 set = !(ihardlimit || isoftlimit);
835 rc = mds_init_slave_ilimits(obd, oqctl, set);
837 CERROR("init slave ilimits failed! (rc:%d)\n", rc);
842 if (dqblk->dqb_valid & QIF_BLIMITS) {
843 set = !(bhardlimit || bsoftlimit);
844 rc = mds_init_slave_blimits(obd, oqctl, set);
846 CERROR("init slave blimits failed! (rc:%d)\n", rc);
850 down(&mds->mds_qonoff_sem);
854 /* cancel previous setting */
855 down(&dquot->dq_sem);
856 dquot->dq_dqb.dqb_ihardlimit = ihardlimit;
857 dquot->dq_dqb.dqb_isoftlimit = isoftlimit;
858 dquot->dq_dqb.dqb_bhardlimit = bhardlimit;
859 dquot->dq_dqb.dqb_bsoftlimit = bsoftlimit;
860 dquot->dq_dqb.dqb_btime = btime;
861 dquot->dq_dqb.dqb_itime = itime;
862 fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
866 down(&dquot->dq_sem);
867 dquot->dq_status &= ~DQ_STATUS_SET;
872 up(&mds->mds_qonoff_sem);
876 static int mds_get_space(struct obd_device *obd, struct obd_quotactl *oqctl)
878 struct obd_quotactl *soqc;
879 struct lvfs_run_ctxt saved;
887 soqc->qc_cmd = Q_GETOQUOTA;
888 soqc->qc_id = oqctl->qc_id;
889 soqc->qc_type = oqctl->qc_type;
891 rc = obd_quotactl(obd->u.mds.mds_osc_exp, soqc);
895 oqctl->qc_dqblk.dqb_curspace = soqc->qc_dqblk.dqb_curspace;
897 push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
898 soqc->qc_dqblk.dqb_curspace = 0;
899 rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, soqc);
900 pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
905 oqctl->qc_dqblk.dqb_curinodes += soqc->qc_dqblk.dqb_curinodes;
906 oqctl->qc_dqblk.dqb_curspace += soqc->qc_dqblk.dqb_curspace;
913 int mds_get_dqblk(struct obd_device *obd, struct obd_quotactl *oqctl)
915 struct mds_obd *mds = &obd->u.mds;
916 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
917 struct lustre_dquot *dquot;
918 struct obd_dqblk *dqblk = &oqctl->qc_dqblk;
922 down(&mds->mds_qonoff_sem);
923 if (qinfo->qi_files[oqctl->qc_type] == NULL)
924 GOTO(out, rc = -ESRCH);
926 dquot = lustre_dqget(obd, qinfo, oqctl->qc_id, oqctl->qc_type);
928 GOTO(out, rc = PTR_ERR(dquot));
930 down(&dquot->dq_sem);
931 dqblk->dqb_ihardlimit = dquot->dq_dqb.dqb_ihardlimit;
932 dqblk->dqb_isoftlimit = dquot->dq_dqb.dqb_isoftlimit;
933 dqblk->dqb_bhardlimit = dquot->dq_dqb.dqb_bhardlimit;
934 dqblk->dqb_bsoftlimit = dquot->dq_dqb.dqb_bsoftlimit;
935 dqblk->dqb_btime = dquot->dq_dqb.dqb_btime;
936 dqblk->dqb_itime = dquot->dq_dqb.dqb_itime;
941 /* the usages in admin quota file is inaccurate */
942 dqblk->dqb_curinodes = 0;
943 dqblk->dqb_curspace = 0;
944 rc = mds_get_space(obd, oqctl);
947 up(&mds->mds_qonoff_sem);
951 int mds_get_obd_quota(struct obd_device *obd, struct obd_quotactl *oqctl)
953 struct lvfs_run_ctxt saved;
957 push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
958 rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, oqctl);
959 pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
965 /* FIXME we only recovery block limit by now, need recovery inode
966 * limits also after CMD involved in */
968 dquot_recovery(struct obd_device *obd, unsigned int id, unsigned short type)
970 struct mds_obd *mds = &obd->u.mds;
971 struct lustre_quota_info *qinfo= &obd->u.mds.mds_quota_info;
972 struct lustre_dquot *dquot;
973 struct obd_quotactl *qctl;
974 __u64 total_limits = 0;
982 dquot = lustre_dqget(obd, qinfo, id, type);
984 CERROR("Get dquot failed. (rc:%ld)\n", PTR_ERR(dquot));
986 RETURN(PTR_ERR(dquot));
989 down(&dquot->dq_sem);
991 /* don't recovery the dquot without limits or under setting */
992 if (!(dquot->dq_dqb.dqb_bhardlimit || dquot->dq_dqb.dqb_bsoftlimit) ||
995 dquot->dq_status |= DQ_STATUS_RECOVERY;
999 /* get real bhardlimit from all slaves. */
1000 qctl->qc_cmd = Q_GETOQUOTA;
1001 qctl->qc_type = type;
1003 qctl->qc_stat = QUOTA_RECOVERING;
1004 rc = obd_quotactl(obd->u.mds.mds_osc_exp, qctl);
1007 total_limits = qctl->qc_dqblk.dqb_bhardlimit;
1009 /* get real bhardlimit from master */
1010 rc = fsfilt_quotactl(obd, obd->u.obt.obt_sb, qctl);
1013 total_limits += qctl->qc_dqblk.dqb_bhardlimit;
1015 /* amend the usage of the administrative quotafile */
1016 down(&mds->mds_qonoff_sem);
1017 down(&dquot->dq_sem);
1019 dquot->dq_dqb.dqb_curspace = total_limits << QUOTABLOCK_BITS;
1021 rc = fsfilt_dquot(obd, dquot, QFILE_WR_DQUOT);
1023 CERROR("write dquot failed! (rc:%d)\n", rc);
1026 up(&mds->mds_qonoff_sem);
1029 down(&dquot->dq_sem);
1030 dquot->dq_status &= ~DQ_STATUS_RECOVERY;
1034 lustre_dqput(dquot);
1039 struct qmaster_recov_thread_data {
1040 struct obd_device *obd;
1041 struct completion comp;
1044 static int qmaster_recovery_main(void *arg)
1046 struct qmaster_recov_thread_data *data = arg;
1047 struct obd_device *obd = data->obd;
1049 unsigned short type;
1052 ptlrpc_daemonize("qmaster_recovd");
1054 complete(&data->comp);
1056 for (type = USRQUOTA; type < MAXQUOTAS; type++) {
1057 struct mds_obd *mds = &obd->u.mds;
1058 struct lustre_quota_info *qinfo = &mds->mds_quota_info;
1059 struct list_head id_list;
1060 struct dquot_id *dqid, *tmp;
1062 down(&mds->mds_qonoff_sem);
1063 if (qinfo->qi_files[type] == NULL) {
1064 up(&mds->mds_qonoff_sem);
1067 INIT_LIST_HEAD(&id_list);
1068 rc = fsfilt_qids(obd, qinfo->qi_files[type], NULL, type,
1070 up(&mds->mds_qonoff_sem);
1073 CERROR("error get ids from admin quotafile.(%d)\n", rc);
1075 list_for_each_entry_safe(dqid, tmp, &id_list, di_link) {
1076 list_del_init(&dqid->di_link);
1080 rc = dquot_recovery(obd, dqid->di_id, type);
1082 CERROR("qmaster recovery failed! (id:%d type:%d"
1083 " rc:%d)\n", dqid->di_id, type, rc);
1091 int mds_quota_recovery(struct obd_device *obd)
1093 struct lov_obd *lov = &obd->u.mds.mds_osc_obd->u.lov;
1094 struct qmaster_recov_thread_data data;
1098 mutex_down(&lov->lov_lock);
1099 if (lov->desc.ld_tgt_count != lov->desc.ld_active_tgt_count) {
1100 CWARN("Not all osts are active, abort quota recovery\n");
1101 mutex_up(&lov->lov_lock);
1104 mutex_up(&lov->lov_lock);
1107 init_completion(&data.comp);
1109 rc = kernel_thread(qmaster_recovery_main, &data, CLONE_VM|CLONE_FILES);
1111 CERROR("Cannot start quota recovery thread: rc %d\n", rc);
1113 wait_for_completion(&data.comp);