Whamcloud - gitweb
LU-5436 quota: use OBD_SLAB_FREE_PTR() to free lqe
[fs/lustre-release.git] / lustre / quota / lquota_lib.c
index 0f5d984..7c0fb54 100644 (file)
  * GPL HEADER END
  */
 /*
- * Copyright (c) 2011, 2012, Whamcloud, Inc.
+ * Copyright (c) 2012, 2013, Intel Corporation.
  * Use is subject to license terms.
  *
- * Author: Johann Lombardi <johann@whamcloud.com>
- * Author: Niu    Yawei    <niu@whamcloud.com>
+ * Author: Johann Lombardi <johann.lombardi@intel.com>
+ * Author: Niu    Yawei    <yawei.niu@intel.com>
  */
 
-#ifndef EXPORT_SYMTAB
-# define EXPORT_SYMTAB
-#endif
-
 #define DEBUG_SUBSYSTEM S_LQUOTA
 
+#include <linux/version.h>
+#include <linux/module.h>
+#include <linux/init.h>
+
 #include "lquota_internal.h"
 
-static struct dt_object_format dt_acct_format = {
-       .dof_type               = DFT_INDEX,
-       .u.dof_idx.di_feat      = &dt_acct_features
+struct kmem_cache *lqe_kmem;
+
+struct lu_kmem_descr lquota_caches[] = {
+       {
+               .ckd_cache = &lqe_kmem,
+               .ckd_name  = "lqe_kmem",
+               .ckd_size  = sizeof(struct lquota_entry)
+       },
+       {
+               .ckd_cache = NULL
+       }
 };
 
+/* register lquota key */
+LU_KEY_INIT_FINI(lquota, struct lquota_thread_info);
+LU_CONTEXT_KEY_DEFINE(lquota, LCT_MD_THREAD | LCT_DT_THREAD | LCT_LOCAL);
+LU_KEY_INIT_GENERIC(lquota);
+
 /**
  * Look-up accounting object to collect space usage information for user
  * or group.
  *
- * \param env - is the environment passed by the caller
- * \param dev - is the dt_device storing the accounting object
- * \param oid - is the object id of the accounting object to initialize, must be
- *              either ACCT_USER_OID or ACCT_GROUP_OID.
+ * \param env  - is the environment passed by the caller
+ * \param dev  - is the dt_device storing the accounting object
+ * \param type - is the quota type, either USRQUOTA or GRPQUOTA
  */
 struct dt_object *acct_obj_lookup(const struct lu_env *env,
-                                 struct dt_device *dev, __u32 oid)
+                                 struct dt_device *dev, int type)
 {
-       struct dt_object        *obj = NULL;
-       struct lu_fid            fid;
-       struct lu_attr           attr;
-       int                      rc;
+       struct lquota_thread_info       *qti = lquota_info(env);
+       struct dt_object                *obj = NULL;
        ENTRY;
 
-       memset(&attr, 0, sizeof(attr));
-       attr.la_valid = LA_MODE;
-       attr.la_mode = S_IFREG | S_IRUGO | S_IWUSR;
-       lu_local_obj_fid(&fid, oid);
+       lu_local_obj_fid(&qti->qti_fid,
+                        type == USRQUOTA ? ACCT_USER_OID : ACCT_GROUP_OID);
 
-       /* lookup/create the accounting object */
-       obj = dt_find_or_create(env, dev, &fid, &dt_acct_format, &attr);
+       /* lookup the accounting object */
+       obj = dt_locate(env, dev, &qti->qti_fid);
        if (IS_ERR(obj))
                RETURN(obj);
 
+       if (!dt_object_exists(obj)) {
+               lu_object_put(env, &obj->do_lu);
+               RETURN(ERR_PTR(-ENOENT));
+       }
+
        if (obj->do_index_ops == NULL) {
+               int rc;
+
                /* set up indexing operations */
                rc = obj->do_ops->do_index_try(env, obj, &dt_acct_features);
                if (rc) {
+                       CERROR("%s: failed to set up indexing operations for %s"
+                              " acct object rc:%d\n",
+                              dev->dd_lu_dev.ld_obd->obd_name,
+                              QTYPE_NAME(type), rc);
+                       lu_object_put(env, &obj->do_lu);
+                       RETURN(ERR_PTR(rc));
+               }
+       }
+       RETURN(obj);
+}
+
+/**
+ * Initialize slave index object to collect local quota limit for user or group.
+ *
+ * \param env - is the environment passed by the caller
+ * \param dev - is the dt_device storing the slave index object
+ * \param type - is the quota type, either USRQUOTA or GRPQUOTA
+ */
+static struct dt_object *quota_obj_lookup(const struct lu_env *env,
+                                         struct dt_device *dev, int type)
+{
+       struct lquota_thread_info       *qti = lquota_info(env);
+       struct dt_object                *obj = NULL;
+       ENTRY;
+
+       qti->qti_fid.f_seq = FID_SEQ_QUOTA;
+       qti->qti_fid.f_oid = type == USRQUOTA ? LQUOTA_USR_OID : LQUOTA_GRP_OID;
+       qti->qti_fid.f_ver = 0;
+
+       /* lookup the quota object */
+       obj = dt_locate(env, dev, &qti->qti_fid);
+       if (IS_ERR(obj))
+               RETURN(obj);
+
+       if (!dt_object_exists(obj)) {
+               lu_object_put(env, &obj->do_lu);
+               RETURN(ERR_PTR(-ENOENT));
+       }
+
+       if (obj->do_index_ops == NULL) {
+               int rc;
+
+               /* set up indexing operations */
+               rc = obj->do_ops->do_index_try(env, obj,
+                                              &dt_quota_slv_features);
+               if (rc) {
+                       CERROR("%s: failed to set up indexing operations for %s"
+                              " slave index object rc:%d\n",
+                              dev->dd_lu_dev.ld_obd->obd_name,
+                              QTYPE_NAME(type), rc);
                        lu_object_put(env, &obj->do_lu);
                        RETURN(ERR_PTR(rc));
                }
@@ -93,51 +158,228 @@ struct dt_object *acct_obj_lookup(const struct lu_env *env,
 int lquotactl_slv(const struct lu_env *env, struct dt_device *dev,
                  struct obd_quotactl *oqctl)
 {
-       struct acct_rec          rec;
-       __u64                    key;
-       struct dt_object        *obj;
-       int                      rc = 0;
+       struct lquota_thread_info       *qti = lquota_info(env);
+       __u64                            key;
+       struct dt_object                *obj;
+       struct obd_dqblk                *dqblk = &oqctl->qc_dqblk;
+       int                              rc;
        ENTRY;
 
        if (oqctl->qc_cmd != Q_GETOQUOTA) {
                /* as in many other places, dev->dd_lu_dev.ld_obd->obd_name
-                * point to a valid obd_name, to be fixed in LU-1574 */
+                * point to an invalid obd_name, to be fixed in LU-1574 */
                CERROR("%s: Unsupported quotactl command: %x\n",
                       dev->dd_lu_dev.ld_obd->obd_name, oqctl->qc_cmd);
                RETURN(-EOPNOTSUPP);
        }
 
-       if (oqctl->qc_type == USRQUOTA)
-               obj = acct_obj_lookup(env, dev, ACCT_USER_OID);
-       else if (oqctl->qc_type == GRPQUOTA)
-               obj = acct_obj_lookup(env, dev, ACCT_GROUP_OID);
-       else
+       if (oqctl->qc_type != USRQUOTA && oqctl->qc_type != GRPQUOTA)
                /* no support for directory quota yet */
                RETURN(-EOPNOTSUPP);
 
+       /* qc_id is a 32-bit field while a key has 64 bits */
+       key = oqctl->qc_id;
+
+       /* Step 1: collect accounting information */
+
+       obj = acct_obj_lookup(env, dev, oqctl->qc_type);
        if (IS_ERR(obj))
                RETURN(-EOPNOTSUPP);
        if (obj->do_index_ops == NULL)
                GOTO(out, rc = -EINVAL);
 
-       /* qc_id is a 32-bit field while a key has 64 bits */
-       key = oqctl->qc_id;
-
        /* lookup record storing space accounting information for this ID */
-       rc = dt_lookup(env, obj, (struct dt_rec *)&rec, (struct dt_key *)&key,
-                      BYPASS_CAPA);
+       rc = dt_lookup(env, obj, (struct dt_rec *)&qti->qti_acct_rec,
+                      (struct dt_key *)&key, BYPASS_CAPA);
        if (rc < 0)
                GOTO(out, rc);
 
        memset(&oqctl->qc_dqblk, 0, sizeof(struct obd_dqblk));
-       oqctl->qc_dqblk.dqb_curspace  = rec.bspace;
-       oqctl->qc_dqblk.dqb_curinodes = rec.ispace;
-       oqctl->qc_dqblk.dqb_valid     = QIF_USAGE;
-       /* TODO: must set {hard,soft}limit and grace time */
+       dqblk->dqb_curspace     = qti->qti_acct_rec.bspace;
+       dqblk->dqb_curinodes    = qti->qti_acct_rec.ispace;
+       dqblk->dqb_valid        = QIF_USAGE;
+
+       lu_object_put(env, &obj->do_lu);
+
+       /* Step 2: collect enforcement information */
+
+       obj = quota_obj_lookup(env, dev, oqctl->qc_type);
+       if (IS_ERR(obj))
+               RETURN(0);
+       if (obj->do_index_ops == NULL)
+               GOTO(out, rc = 0);
+
+       memset(&qti->qti_slv_rec, 0, sizeof(qti->qti_slv_rec));
+       /* lookup record storing enforcement information for this ID */
+       rc = dt_lookup(env, obj, (struct dt_rec *)&qti->qti_slv_rec,
+                      (struct dt_key *)&key, BYPASS_CAPA);
+       if (rc < 0 && rc != -ENOENT)
+               GOTO(out, rc = 0);
+
+       if (lu_device_is_md(dev->dd_lu_dev.ld_site->ls_top_dev)) {
+               dqblk->dqb_ihardlimit = qti->qti_slv_rec.qsr_granted;
+               dqblk->dqb_bhardlimit = 0;
+       } else {
+               dqblk->dqb_ihardlimit = 0;
+               dqblk->dqb_bhardlimit = qti->qti_slv_rec.qsr_granted;
+       }
+       dqblk->dqb_valid |= QIF_LIMITS;
 
-       EXIT;
+       GOTO(out, rc = 0);
 out:
        lu_object_put(env, &obj->do_lu);
         return rc;
 }
 EXPORT_SYMBOL(lquotactl_slv);
+
+/**
+ * Helper routine returning the FID associated with the global index storing
+ * quota settings for the storage pool \pool_id, resource type \pool_type and
+ * the quota type \quota_type.
+ */
+void lquota_generate_fid(struct lu_fid *fid, int pool_id, int pool_type,
+                         int quota_type)
+{
+       __u8     qtype;
+
+       qtype = (quota_type == USRQUOTA) ? LQUOTA_TYPE_USR : LQUOTA_TYPE_GRP;
+
+       fid->f_seq = FID_SEQ_QUOTA_GLB;
+       fid->f_oid = (qtype << 24) | (pool_type << 16) | (__u16)pool_id;
+       fid->f_ver = 0;
+}
+
+/**
+ * Helper routine used to extract pool ID, pool type and quota type from a
+ * given FID.
+ */
+int lquota_extract_fid(const struct lu_fid *fid, int *pool_id, int *pool_type,
+                      int *quota_type)
+{
+       unsigned int     tmp;
+       ENTRY;
+
+       if (fid->f_seq != FID_SEQ_QUOTA_GLB)
+               RETURN(-EINVAL);
+
+       if (pool_id != NULL) {
+               tmp = fid->f_oid & 0xffffU;
+               if (tmp != 0)
+                       /* we only support pool ID 0 for the time being */
+                       RETURN(-ENOTSUPP);
+               *pool_id = tmp;
+       }
+
+       if (pool_type != NULL) {
+               tmp = (fid->f_oid >> 16) & 0xffU;
+               if (tmp >= LQUOTA_LAST_RES)
+                       RETURN(-ENOTSUPP);
+
+               *pool_type = tmp;
+       }
+
+       if (quota_type != NULL) {
+               tmp = fid->f_oid >> 24;
+               if (tmp >= LQUOTA_TYPE_MAX)
+                       RETURN(-ENOTSUPP);
+
+               *quota_type = (tmp == LQUOTA_TYPE_USR) ? USRQUOTA : GRPQUOTA;
+       }
+
+       RETURN(0);
+}
+
+#if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2,7,50,0)
+/* Index features supported by the global index objects.
+ * We actually use one dt_index_features structure for each quota combination
+ * of quota type x [inode, block] to allow the ldiskfs OSD to recognize those
+ * objects and to handle the conversion from the old administrative quota file
+ * format */
+struct dt_index_features dt_quota_iusr_features;
+EXPORT_SYMBOL(dt_quota_iusr_features);
+struct dt_index_features dt_quota_busr_features;
+EXPORT_SYMBOL(dt_quota_busr_features);
+struct dt_index_features dt_quota_igrp_features;
+EXPORT_SYMBOL(dt_quota_igrp_features);
+struct dt_index_features dt_quota_bgrp_features;
+EXPORT_SYMBOL(dt_quota_bgrp_features);
+
+/**
+ * Helper routine returning the right index feature structure to be used
+ * depending on the FID of the global index.
+ */
+const struct dt_index_features *glb_idx_feature(struct lu_fid *fid)
+{
+       int     res_type, quota_type, rc;
+
+       rc = lquota_extract_fid(fid, NULL, &res_type, &quota_type);
+       if (rc)
+               return ERR_PTR(rc);
+
+       if (quota_type == USRQUOTA) {
+               if (res_type == LQUOTA_RES_MD)
+                       return &dt_quota_iusr_features;
+               else
+                       return &dt_quota_busr_features;
+       } else {
+               if (res_type == LQUOTA_RES_MD)
+                       return &dt_quota_igrp_features;
+               else
+                       return &dt_quota_bgrp_features;
+       }
+}
+#else
+#warning "remove old quota compatibility code"
+#endif
+
+static int __init init_lquota(void)
+{
+       int     rc;
+       ENTRY;
+
+       lquota_key_init_generic(&lquota_thread_key, NULL);
+       lu_context_key_register(&lquota_thread_key);
+
+#if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2,7,50,0)
+       dt_quota_iusr_features = dt_quota_busr_features = dt_quota_glb_features;
+       dt_quota_igrp_features = dt_quota_bgrp_features = dt_quota_glb_features;
+#else
+#warning "remove old quota compatibility code"
+#endif
+
+       rc = lu_kmem_init(lquota_caches);
+       if (rc)
+               GOTO(out_key, rc);
+
+       rc = qmt_glb_init();
+       if (rc)
+               GOTO(out_caches, rc);
+
+       rc = qsd_glb_init();
+       if (rc)
+               GOTO(out_qmt, rc);
+
+       RETURN(0);
+
+out_qmt:
+       qmt_glb_fini();
+out_caches:
+       lu_kmem_fini(lquota_caches);
+out_key:
+       lu_context_key_degister(&lquota_thread_key);
+       return rc;
+}
+
+static void exit_lquota(void)
+{
+       qsd_glb_fini();
+       qmt_glb_fini();
+       lu_kmem_fini(lquota_caches);
+       lu_context_key_degister(&lquota_thread_key);
+}
+
+MODULE_AUTHOR("Intel Corporation <http://www.intel.com/>");
+MODULE_DESCRIPTION("Lustre Quota");
+MODULE_LICENSE("GPL");
+
+cfs_module(lquota, "2.4.0", init_lquota, exit_lquota);