lustre/quota/quota_check.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 (c) 2007, 2010, Oracle and/or its affiliates. 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
  37
  38 #ifndef EXPORT_SYMTAB
  39 # define EXPORT_SYMTAB
  40 #endif
  41 #define DEBUG_SUBSYSTEM S_LQUOTA
  42
  43 #ifdef __KERNEL__
  44 # include <linux/version.h>
  45 # include <linux/module.h>
  46 # include <linux/init.h>
  47 # include <linux/fs.h>
  48 # include <linux/jbd.h>
  49 # include <linux/ext3_fs.h>
  50 # include <linux/smp_lock.h>
  51 # include <linux/buffer_head.h>
  52 # include <linux/workqueue.h>
  53 # include <linux/mount.h>
  54 #else /* __KERNEL__ */
  55 # include <liblustre.h>
  56 #endif
  57
  58 #include <obd_class.h>
  59 #include <lustre_mds.h>
  60 #include <lustre_dlm.h>
  61 #include <lustre_cfg.h>
  62 #include <obd_ost.h>
  63 #include <lustre_fsfilt.h>
  64 #include <lustre_quota.h>
  65 #include "quota_internal.h"
  66
  67 #ifdef HAVE_QUOTA_SUPPORT
  68 #ifdef __KERNEL__
  69 static int target_quotacheck_callback(struct obd_export *exp,
  70                                       struct obd_quotactl *oqctl)
  71 {
  72         struct ptlrpc_request *req;
  73         struct obd_quotactl   *body;
  74         int                    rc;
  75         ENTRY;
  76
  77         req = ptlrpc_request_alloc_pack(exp->exp_imp_reverse, &RQF_QC_CALLBACK,
  78                                         LUSTRE_OBD_VERSION, OBD_QC_CALLBACK);
  79         if (req == NULL)
  80                 RETURN(-ENOMEM);
  81
  82         body = req_capsule_client_get(&req->rq_pill, &RMF_OBD_QUOTACTL);
  83         *body = *oqctl;
  84
  85         ptlrpc_request_set_replen(req);
  86
  87         rc = ptlrpc_queue_wait(req);
  88         ptlrpc_req_finished(req);
  89
  90         RETURN(rc);
  91 }
  92
  93 static int target_quotacheck_thread(void *data)
  94 {
  95         struct quotacheck_thread_args *qta = data;
  96         struct obd_export *exp;
  97         struct obd_device *obd;
  98         struct obd_quotactl *oqctl;
  99         struct lvfs_run_ctxt saved;
 100         int rc;
 101
 102         cfs_daemonize_ctxt("quotacheck");
 103
 104         exp = qta->qta_exp;
 105         obd = qta->qta_obd;
 106         oqctl = &qta->qta_oqctl;
 107
 108         push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
 109
 110         rc = fsfilt_quotacheck(obd, qta->qta_sb, oqctl);
 111         if (rc)
 112                 CERROR("%s: fsfilt_quotacheck: %d\n", obd->obd_name, rc);
 113
 114         pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
 115
 116         rc = target_quotacheck_callback(exp, oqctl);
 117         class_export_put(exp);
 118         cfs_up(qta->qta_sem);
 119         OBD_FREE_PTR(qta);
 120         return rc;
 121 }
 122
 123 int target_quota_check(struct obd_device *obd, struct obd_export *exp,
 124                        struct obd_quotactl *oqctl)
 125 {
 126         struct obd_device_target *obt = &obd->u.obt;
 127         struct quotacheck_thread_args *qta;
 128         int rc = 0;
 129         ENTRY;
 130
 131         OBD_ALLOC_PTR(qta);
 132         if (!qta)
 133                 RETURN(ENOMEM);
 134
 135         cfs_down(&obt->obt_quotachecking);
 136
 137         qta->qta_exp = exp;
 138         qta->qta_obd = obd;
 139         qta->qta_oqctl = *oqctl;
 140         qta->qta_oqctl.qc_id = obt->obt_qfmt; /* override qfmt version */
 141         qta->qta_sb = obt->obt_sb;
 142         qta->qta_sem = &obt->obt_quotachecking;
 143
 144         /* quotaoff firstly */
 145         oqctl->qc_cmd = Q_QUOTAOFF;
 146         if (!strcmp(obd->obd_type->typ_name, LUSTRE_MDS_NAME)) {
 147                 rc = do_mds_quota_off(obd, oqctl);
 148                 if (rc && rc != -EALREADY) {
 149                         CERROR("off quota on MDS failed: %d\n", rc);
 150                         GOTO(out, rc);
 151                 }
 152
 153                 /* quota master */
 154                 rc = init_admin_quotafiles(obd, &qta->qta_oqctl);
 155                 if (rc) {
 156                         CERROR("init_admin_quotafiles failed: %d\n", rc);
 157                         GOTO(out, rc);
 158                 }
 159         } else {
 160                 struct lvfs_run_ctxt saved;
 161                 struct lustre_quota_ctxt *qctxt = &obt->obt_qctxt;
 162
 163                 push_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
 164                 rc = fsfilt_quotactl(obd, obt->obt_sb, oqctl);
 165                 pop_ctxt(&saved, &obd->obd_lvfs_ctxt, NULL);
 166                 if (!rc) {
 167                         qctxt->lqc_flags &= ~UGQUOTA2LQC(oqctl->qc_type);
 168                 } else if (!quota_is_off(qctxt, oqctl)) {
 169                         CERROR("off quota on OSS failed: %d\n", rc);
 170                         GOTO(out, rc);
 171                 }
 172         }
 173
 174         /* we get ref for exp because target_quotacheck_callback() will use this
 175          * export later b=18126 */
 176         class_export_get(exp);
 177         rc = cfs_kernel_thread(target_quotacheck_thread, qta,
 178                                CLONE_VM|CLONE_FILES);
 179         if (rc >= 0) {
 180                 /* target_quotacheck_thread will drop the ref on exp and release
 181                  * obt_quotachecking */
 182                 CDEBUG(D_INFO, "%s: target_quotacheck_thread: %d\n",
 183                        obd->obd_name, rc);
 184                 RETURN(0);
 185         } else {
 186                 CERROR("%s: error starting quotacheck_thread: %d\n",
 187                        obd->obd_name, rc);
 188                 class_export_put(exp);
 189                 EXIT;
 190         }
 191
 192 out:
 193         cfs_up(&obt->obt_quotachecking);
 194         OBD_FREE_PTR(qta);
 195         return rc;
 196 }
 197
 198 #endif /* __KERNEL__ */
 199 #endif /* HAVE_QUOTA_SUPPORT */
 200
 201 int client_quota_check(struct obd_device *unused, struct obd_export *exp,
 202                        struct obd_quotactl *oqctl)
 203 {
 204         struct client_obd       *cli = &exp->exp_obd->u.cli;
 205         struct ptlrpc_request   *req;
 206         struct obd_quotactl     *body;
 207         const struct req_format *rf;
 208         int                      ver, opc, rc;
 209         ENTRY;
 210
 211         if (!strcmp(exp->exp_obd->obd_type->typ_name, LUSTRE_MDC_NAME)) {
 212                 rf  = &RQF_MDS_QUOTACHECK;
 213                 ver = LUSTRE_MDS_VERSION;
 214                 opc = MDS_QUOTACHECK;
 215         } else if (!strcmp(exp->exp_obd->obd_type->typ_name, LUSTRE_OSC_NAME)) {
 216                 rf  = &RQF_OST_QUOTACHECK;
 217                 ver = LUSTRE_OST_VERSION;
 218                 opc = OST_QUOTACHECK;
 219         } else {
 220                 RETURN(-EINVAL);
 221         }
 222
 223         req = ptlrpc_request_alloc_pack(class_exp2cliimp(exp), rf, ver, opc);
 224         if (req == NULL)
 225                 RETURN(-ENOMEM);
 226
 227         body = req_capsule_client_get(&req->rq_pill, &RMF_OBD_QUOTACTL);
 228         *body = *oqctl;
 229
 230         ptlrpc_request_set_replen(req);
 231
 232         /* the next poll will find -ENODATA, that means quotacheck is
 233          * going on */
 234         cli->cl_qchk_stat = -ENODATA;
 235         rc = ptlrpc_queue_wait(req);
 236         if (rc)
 237                 cli->cl_qchk_stat = rc;
 238         ptlrpc_req_finished(req);
 239         RETURN(rc);
 240 }
 241
 242 int client_quota_poll_check(struct obd_export *exp, struct if_quotacheck *qchk)
 243 {
 244         struct client_obd *cli = &exp->exp_obd->u.cli;
 245         int rc;
 246         ENTRY;
 247
 248         rc = cli->cl_qchk_stat;
 249
 250         /* the client is not the previous one */
 251         if (rc == CL_NOT_QUOTACHECKED)
 252                 rc = -EINTR;
 253
 254         qchk->obd_uuid = cli->cl_target_uuid;
 255         /* FIXME change strncmp to strcmp and save the strlen op */
 256         if (strncmp(exp->exp_obd->obd_type->typ_name, LUSTRE_OSC_NAME,
 257                     strlen(LUSTRE_OSC_NAME)) == 0)
 258                 memcpy(qchk->obd_type, LUSTRE_OST_NAME,
 259                        strlen(LUSTRE_OST_NAME));
 260         else if (strncmp(exp->exp_obd->obd_type->typ_name, LUSTRE_MDC_NAME,
 261                          strlen(LUSTRE_MDC_NAME)) == 0)
 262                 memcpy(qchk->obd_type, LUSTRE_MDS_NAME,
 263                        strlen(LUSTRE_MDS_NAME));
 264
 265         RETURN(rc);
 266 }
 267
 268 int lmv_quota_check(struct obd_device *unused, struct obd_export *exp,
 269                     struct obd_quotactl *oqctl)
 270 {
 271         struct obd_device *obd = class_exp2obd(exp);
 272         struct lmv_obd *lmv = &obd->u.lmv;
 273         struct lmv_tgt_desc *tgt;
 274         int i, rc = 0;
 275         ENTRY;
 276
 277         for (i = 0, tgt = lmv->tgts; i < lmv->desc.ld_tgt_count; i++, tgt++) {
 278                 int err;
 279
 280                 if (!tgt->ltd_active) {
 281                         CERROR("lmv idx %d inactive\n", i);
 282                         RETURN(-EIO);
 283                 }
 284
 285                 err = obd_quotacheck(tgt->ltd_exp, oqctl);
 286                 if (err && !rc)
 287                         rc = err;
 288         }
 289
 290         RETURN(rc);
 291 }
 292
 293 int lov_quota_check(struct obd_device *unused, struct obd_export *exp,
 294                     struct obd_quotactl *oqctl)
 295 {
 296         struct obd_device *obd = class_exp2obd(exp);
 297         struct lov_obd *lov = &obd->u.lov;
 298         int i, rc = 0;
 299         ENTRY;
 300
 301         obd_getref(obd);
 302
 303         for (i = 0; i < lov->desc.ld_tgt_count; i++) {
 304                 if (!lov->lov_tgts[i])
 305                         continue;
 306
 307                 /* Skip quota check on the administratively disabled OSTs. */
 308                 if (!lov->lov_tgts[i]->ltd_activate) {
 309                         CWARN("lov idx %d was administratively disabled, "
 310                               "skip quotacheck on it.\n", i);
 311                         continue;
 312                 }
 313
 314                 if (!lov->lov_tgts[i]->ltd_active) {
 315                         CERROR("lov idx %d inactive\n", i);
 316                         rc = -EIO;
 317                         goto out;
 318                 }
 319         }
 320
 321         for (i = 0; i < lov->desc.ld_tgt_count; i++) {
 322                 int err;
 323
 324                 if (!lov->lov_tgts[i] || !lov->lov_tgts[i]->ltd_activate)
 325                         continue;
 326
 327                 err = obd_quotacheck(lov->lov_tgts[i]->ltd_exp, oqctl);
 328                 if (err && !rc)
 329                         rc = err;
 330         }
 331
 332 out:
 333         obd_putref(obd);
 334
 335         RETURN(rc);
 336 }