Whamcloud - gitweb
b=21174 fix quotacheck with non-consecutive OST indices
[fs/lustre-release.git] / 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                 if (!lov->lov_tgts[i]->ltd_active) {
308                         CERROR("lov idx %d inactive\n", i);
309                         rc = -EIO;
310                         goto out;
311                 }
312         }
313
314         for (i = 0; i < lov->desc.ld_tgt_count; i++) {
315                 int err;
316
317                 if (!lov->lov_tgts[i])
318                         continue;
319
320                 err = obd_quotacheck(lov->lov_tgts[i]->ltd_exp, oqctl);
321                 if (err && !rc)
322                         rc = err;
323         }
324
325 out:
326         obd_putref(obd);
327
328         RETURN(rc);
329 }