Whamcloud - gitweb
LU-3536 osp: move update packing into out_lib.c
[fs/lustre-release.git] / lustre / osd-ldiskfs / osd_quota.c
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 only,
8  * as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License version 2 for more details (a copy is included
14  * in the LICENSE file that accompanied this code).
15  *
16  * You should have received a copy of the GNU General Public License
17  * version 2 along with this program; if not, write to the
18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19  * Boston, MA 021110-1307, USA
20  *
21  * GPL HEADER END
22  */
23 /*
24  * Copyright (c) 2012, 2013, Intel Corporation.
25  * Use is subject to license terms.
26  *
27  * Author: Johann Lombardi <johann@whamcloud.com>
28  * Author: Niu    Yawei    <niu@whamcloud.com>
29  */
30
31 #include <lustre_quota.h>
32 #include "osd_internal.h"
33
34 /**
35  * Helpers function to find out the quota type (USRQUOTA/GRPQUOTA) of a
36  * given object
37  */
38 static inline int fid2type(const struct lu_fid *fid)
39 {
40         LASSERT(fid_is_acct(fid));
41         if (fid_oid(fid) == ACCT_GROUP_OID)
42                 return GRPQUOTA;
43         return USRQUOTA;
44 }
45
46 static inline int obj2type(struct dt_object *obj)
47 {
48         return fid2type(lu_object_fid(&obj->do_lu));
49 }
50
51 /**
52  * Space Accounting Management
53  */
54
55 /**
56  * Look up an accounting object based on its fid.
57  *
58  * \param info - is the osd thread info passed by the caller
59  * \param osd  - is the osd device
60  * \param fid  - is the fid of the accounting object we want to look up
61  * \param id   - is the osd_inode_id struct to fill with the inode number of
62  *               the quota file if the lookup is successful
63  */
64 int osd_acct_obj_lookup(struct osd_thread_info *info, struct osd_device *osd,
65                         const struct lu_fid *fid, struct osd_inode_id *id)
66 {
67         struct super_block *sb = osd_sb(osd);
68         unsigned long qf_inums[2] = {
69                 le32_to_cpu(LDISKFS_SB(sb)->s_es->s_usr_quota_inum),
70                 le32_to_cpu(LDISKFS_SB(sb)->s_es->s_grp_quota_inum)
71         };
72
73         ENTRY;
74         LASSERT(fid_is_acct(fid));
75
76         if (!LDISKFS_HAS_RO_COMPAT_FEATURE(sb,
77                                            LDISKFS_FEATURE_RO_COMPAT_QUOTA))
78                 RETURN(-ENOENT);
79
80         id->oii_gen = OSD_OII_NOGEN;
81         id->oii_ino = qf_inums[fid2type(fid)];
82         if (!ldiskfs_valid_inum(sb, id->oii_ino))
83                 RETURN(-ENOENT);
84         RETURN(0);
85 }
86
87 /**
88  * Return space usage (#blocks & #inodes) consumed by a given uid or gid.
89  *
90  * \param env   - is the environment passed by the caller
91  * \param dtobj - is the accounting object
92  * \param dtrec - is the record to fill with space usage information
93  * \param dtkey - is the id the of the user or group for which we would
94  *                like to access disk usage.
95  * \param capa - is the capability, not used.
96  *
97  * \retval +ve - success : exact match
98  * \retval -ve - failure
99  */
100 static int osd_acct_index_lookup(const struct lu_env *env,
101                                  struct dt_object *dtobj,
102                                  struct dt_rec *dtrec,
103                                  const struct dt_key *dtkey,
104                                  struct lustre_capa *capa)
105 {
106         struct osd_thread_info  *info = osd_oti_get(env);
107 #ifdef HAVE_DQUOT_FS_DISK_QUOTA
108         struct fs_disk_quota    *dqblk = &info->oti_fdq;
109 #else
110         struct if_dqblk         *dqblk = &info->oti_dqblk;
111 #endif
112         struct super_block      *sb = osd_sb(osd_obj2dev(osd_dt_obj(dtobj)));
113         struct lquota_acct_rec  *rec = (struct lquota_acct_rec *)dtrec;
114         __u64                    id = *((__u64 *)dtkey);
115         int                      rc;
116 #ifdef HAVE_DQUOT_KQID
117         struct kqid             qid;
118 #endif
119
120         ENTRY;
121
122         memset((void *)dqblk, 0, sizeof(struct obd_dqblk));
123 #ifdef HAVE_DQUOT_KQID
124         qid = make_kqid(&init_user_ns, obj2type(dtobj), id);
125         rc = sb->s_qcop->get_dqblk(sb, qid, dqblk);
126 #else
127         rc = sb->s_qcop->get_dqblk(sb, obj2type(dtobj), (qid_t) id, dqblk);
128 #endif
129         if (rc)
130                 RETURN(rc);
131 #ifdef HAVE_DQUOT_FS_DISK_QUOTA
132         rec->bspace = dqblk->d_bcount;
133         rec->ispace = dqblk->d_icount;
134 #else
135         rec->bspace = dqblk->dqb_curspace;
136         rec->ispace = dqblk->dqb_curinodes;
137 #endif
138         RETURN(+1);
139 }
140
141 #define QUOTA_IT_READ_ERROR(it, rc)                                    \
142         CERROR("%s: Error while trying to read quota information, "    \
143                "failed with %d\n",                                     \
144                osd_dev(it->oiq_obj->oo_dt.do_lu.lo_dev)->od_svname, rc); \
145
146 /**
147  * Initialize osd Iterator for given osd index object.
148  *
149  * \param  dt    - osd index object
150  * \param  attr  - not used
151  * \param  capa  - BYPASS_CAPA
152  */
153 static struct dt_it *osd_it_acct_init(const struct lu_env *env,
154                                       struct dt_object *dt,
155                                       __u32 attr, struct lustre_capa *capa)
156 {
157         struct osd_thread_info  *info = osd_oti_get(env);
158         struct osd_it_quota     *it;
159         struct lu_object        *lo = &dt->do_lu;
160         struct osd_object       *obj = osd_dt_obj(dt);
161
162         ENTRY;
163
164         LASSERT(lu_object_exists(lo));
165
166         if (info == NULL)
167                 RETURN(ERR_PTR(-ENOMEM));
168
169         it = &info->oti_it_quota;
170         memset(it, 0, sizeof(*it));
171         lu_object_get(lo);
172         it->oiq_obj = obj;
173         INIT_LIST_HEAD(&it->oiq_list);
174
175         /* LUSTRE_DQTREEOFF is the initial offset where the tree can be found */
176         it->oiq_blk[0] = LUSTRE_DQTREEOFF;
177
178         /* NB: we don't need to store the tree depth since it is always
179          * equal to LUSTRE_DQTREEDEPTH - 1 (root has depth = 0) for a leaf
180          * block. */
181         RETURN((struct dt_it *)it);
182 }
183
184 /**
185  * Free given iterator.
186  *
187  * \param  di   - osd iterator
188  */
189 static void osd_it_acct_fini(const struct lu_env *env, struct dt_it *di)
190 {
191         struct osd_it_quota *it = (struct osd_it_quota *)di;
192         struct osd_quota_leaf *leaf, *tmp;
193         ENTRY;
194
195         lu_object_put(env, &it->oiq_obj->oo_dt.do_lu);
196
197         list_for_each_entry_safe(leaf, tmp, &it->oiq_list, oql_link) {
198                 list_del_init(&leaf->oql_link);
199                 OBD_FREE_PTR(leaf);
200         }
201         EXIT;
202 }
203
204 /**
205  * Move Iterator to record specified by \a key, if the \a key isn't found,
206  * move to the first valid record.
207  *
208  * \param  di   - osd iterator
209  * \param  key  - uid or gid
210  *
211  * \retval +ve  - di points to the first valid record
212  * \retval  +1  - di points to exact matched key
213  * \retval -ve  - failure
214  */
215 static int osd_it_acct_get(const struct lu_env *env, struct dt_it *di,
216                            const struct dt_key *key)
217 {
218         struct osd_it_quota     *it = (struct osd_it_quota *)di;
219         const struct lu_fid     *fid =
220                                 lu_object_fid(&it->oiq_obj->oo_dt.do_lu);
221         int                      type = fid2type(fid);
222         qid_t                    dqid = *(qid_t *)key;
223         loff_t                   offset;
224         int                      rc;
225
226         ENTRY;
227
228         offset = find_tree_dqentry(env, it->oiq_obj, type, dqid,
229                                    LUSTRE_DQTREEOFF, 0, it);
230         if (offset > 0) { /* Found */
231                 RETURN(+1);
232         } else if (offset < 0) { /* Error */
233                 QUOTA_IT_READ_ERROR(it, (int)offset);
234                 RETURN((int)offset);
235         }
236
237         /* The @key is not found, move to the first valid entry */
238         rc = walk_tree_dqentry(env, it->oiq_obj, type, it->oiq_blk[0], 0,
239                                0, it);
240         if (rc == 0)
241                 rc = 1;
242         else if (rc > 0)
243                 rc = -ENOENT;
244
245         RETURN(rc);
246 }
247
248 /**
249  * Release Iterator
250  *
251  * \param  di   - osd iterator
252  */
253 static void osd_it_acct_put(const struct lu_env *env, struct dt_it *di)
254 {
255         return;
256 }
257
258 static int osd_it_add_processed(struct osd_it_quota *it, int depth)
259 {
260         struct osd_quota_leaf *leaf;
261
262         OBD_ALLOC_PTR(leaf);
263         if (leaf == NULL)
264                 RETURN(-ENOMEM);
265         INIT_LIST_HEAD(&leaf->oql_link);
266         leaf->oql_blk = it->oiq_blk[depth];
267         list_add_tail(&leaf->oql_link, &it->oiq_list);
268         RETURN(0);
269 }
270
271 /**
272  * Move on to the next valid entry.
273  *
274  * \param  di   - osd iterator
275  *
276  * \retval +ve  - iterator reached the end
277  * \retval   0  - iterator has not reached the end yet
278  * \retval -ve  - unexpected failure
279  */
280 static int osd_it_acct_next(const struct lu_env *env, struct dt_it *di)
281 {
282         struct osd_it_quota     *it = (struct osd_it_quota *)di;
283         const struct lu_fid     *fid =
284                                 lu_object_fid(&it->oiq_obj->oo_dt.do_lu);
285         int                      type = fid2type(fid);
286         int                      depth, rc;
287         uint                     index;
288
289         ENTRY;
290
291         /* Let's first check if there are any remaining valid entry in the
292          * current leaf block. Start with the next entry after the current one.
293          */
294         depth = LUSTRE_DQTREEDEPTH;
295         index = it->oiq_index[depth];
296         if (++index < LUSTRE_DQSTRINBLK) {
297                 /* Search for the next valid entry from current index */
298                 rc = walk_block_dqentry(env, it->oiq_obj, type,
299                                         it->oiq_blk[depth], index, it);
300                 if (rc < 0) {
301                         QUOTA_IT_READ_ERROR(it, rc);
302                         RETURN(rc);
303                 } else if (rc == 0) {
304                         /* Found on entry, @it is already updated to the
305                          * new position in walk_block_dqentry(). */
306                         RETURN(0);
307                 } else {
308                         rc = osd_it_add_processed(it, depth);
309                         if (rc)
310                                 RETURN(rc);
311                 }
312         } else {
313                 rc = osd_it_add_processed(it, depth);
314                 if (rc)
315                         RETURN(rc);
316         }
317         rc = 1;
318
319         /* We have consumed all the entries of the current leaf block, move on
320          * to the next one. */
321         depth--;
322
323         /* We keep searching as long as walk_tree_dqentry() returns +1
324          * (= no valid entry found). */
325         for (; depth >= 0 && rc > 0; depth--) {
326                 index = it->oiq_index[depth];
327                 if (++index > 0xff)
328                         continue;
329                 rc = walk_tree_dqentry(env, it->oiq_obj, type,
330                                        it->oiq_blk[depth], depth, index, it);
331         }
332
333         if (rc < 0)
334                 QUOTA_IT_READ_ERROR(it, rc);
335         RETURN(rc);
336 }
337
338 /**
339  * Return pointer to the key under iterator.
340  *
341  * \param  di   - osd iterator
342  */
343 static struct dt_key *osd_it_acct_key(const struct lu_env *env,
344                                       const struct dt_it *di)
345 {
346         struct osd_it_quota *it = (struct osd_it_quota *)di;
347
348         ENTRY;
349         RETURN((struct dt_key *)&it->oiq_id);
350 }
351
352 /**
353  * Return size of key under iterator (in bytes)
354  *
355  * \param  di   - osd iterator
356  */
357 static int osd_it_acct_key_size(const struct lu_env *env,
358                                 const struct dt_it *di)
359 {
360         struct osd_it_quota *it = (struct osd_it_quota *)di;
361
362         ENTRY;
363         RETURN((int)sizeof(it->oiq_id));
364 }
365
366 /**
367  * Return pointer to the record under iterator.
368  *
369  * \param  di    - osd iterator
370  * \param  attr  - not used
371  */
372 static int osd_it_acct_rec(const struct lu_env *env,
373                            const struct dt_it *di,
374                            struct dt_rec *dtrec, __u32 attr)
375 {
376         struct osd_it_quota     *it = (struct osd_it_quota *)di;
377         const struct dt_key     *key = osd_it_acct_key(env, di);
378         int                      rc;
379
380         ENTRY;
381
382         rc = osd_acct_index_lookup(env, &it->oiq_obj->oo_dt, dtrec, key,
383                                    BYPASS_CAPA);
384         RETURN(rc > 0 ? 0 : rc);
385 }
386
387 /**
388  * Returns cookie for current Iterator position.
389  *
390  * \param  di    - osd iterator
391  */
392 static __u64 osd_it_acct_store(const struct lu_env *env,
393                                const struct dt_it *di)
394 {
395         struct osd_it_quota *it = (struct osd_it_quota *)di;
396
397         ENTRY;
398         RETURN(it->oiq_id);
399 }
400
401 /**
402  * Restore iterator from cookie. if the \a hash isn't found,
403  * restore the first valid record.
404  *
405  * \param  di    - osd iterator
406  * \param  hash  - iterator location cookie
407  *
408  * \retval +ve   - di points to the first valid record
409  * \retval  +1   - di points to exact matched hash
410  * \retval -ve   - failure
411  */
412 static int osd_it_acct_load(const struct lu_env *env,
413                             const struct dt_it *di, __u64 hash)
414 {
415         ENTRY;
416         RETURN(osd_it_acct_get(env, (struct dt_it *)di,
417                                (const struct dt_key *)&hash));
418 }
419
420 /**
421  * Index and Iterator operations for accounting objects
422  */
423 const struct dt_index_operations osd_acct_index_ops = {
424         .dio_lookup     = osd_acct_index_lookup,
425         .dio_it         = {
426                 .init           = osd_it_acct_init,
427                 .fini           = osd_it_acct_fini,
428                 .get            = osd_it_acct_get,
429                 .put            = osd_it_acct_put,
430                 .next           = osd_it_acct_next,
431                 .key            = osd_it_acct_key,
432                 .key_size       = osd_it_acct_key_size,
433                 .rec            = osd_it_acct_rec,
434                 .store          = osd_it_acct_store,
435                 .load           = osd_it_acct_load
436         }
437 };
438
439 static inline void osd_quota_swab(char *ptr, size_t size)
440 {
441         int offset;
442
443         LASSERT((size & (sizeof(__u64) - 1)) == 0);
444
445         for (offset = 0; offset < size; offset += sizeof(__u64))
446              __swab64s((__u64 *)(ptr + offset));
447 }
448
449 const struct dt_rec *osd_quota_pack(struct osd_object *obj,
450                                     const struct dt_rec *rec,
451                                     union lquota_rec *quota_rec)
452 {
453 #ifdef __BIG_ENDIAN
454         struct iam_descr        *descr;
455
456         LASSERT(obj->oo_dir != NULL);
457         descr = obj->oo_dir->od_container.ic_descr;
458
459         memcpy(quota_rec, rec, descr->id_rec_size);
460
461         osd_quota_swab((char *)quota_rec, descr->id_rec_size);
462         return (const struct dt_rec *)quota_rec;
463 #else
464         return rec;
465 #endif
466 }
467
468 void osd_quota_unpack(struct osd_object *obj, const struct dt_rec *rec)
469 {
470 #ifdef __BIG_ENDIAN
471         struct iam_descr *descr;
472
473         LASSERT(obj->oo_dir != NULL);
474         descr = obj->oo_dir->od_container.ic_descr;
475
476         osd_quota_swab((char *)rec, descr->id_rec_size);
477 #else
478         return;
479 #endif
480 }
481
482 static inline int osd_qid_type(struct osd_thandle *oh, int i)
483 {
484         return (oh->ot_id_type & (1 << i)) ? GRPQUOTA : USRQUOTA;
485 }
486
487 static inline void osd_qid_set_type(struct osd_thandle *oh, int i, int type)
488 {
489         oh->ot_id_type |= ((type == GRPQUOTA) ? (1 << i) : 0);
490 }
491
492 /**
493  * Reserve journal credits for quota files update first, then call
494  * ->op_begin() to perform quota enforcement.
495  *
496  * \param  env    - the environment passed by the caller
497  * \param  oh     - osd transaction handle
498  * \param  qi     - quota id & space required for this operation
499  * \param  allocated - dquot entry in quota accounting file has been allocated
500  * \param  flags  - if the operation is write, return no user quota, no
501  *                  group quota, or sync commit flags to the caller
502  *
503  * \retval 0      - success
504  * \retval -ve    - failure
505  */
506 int osd_declare_qid(const struct lu_env *env, struct osd_thandle *oh,
507                     struct lquota_id_info *qi, bool allocated, int *flags)
508 {
509         struct osd_thread_info  *info = osd_oti_get(env);
510         struct osd_device       *dev = info->oti_dev;
511         struct qsd_instance     *qsd = dev->od_quota_slave;
512         int                      i, rc;
513         bool                     found = false;
514         ENTRY;
515
516         LASSERT(oh != NULL);
517         LASSERTF(oh->ot_id_cnt <= OSD_MAX_UGID_CNT, "count=%d\n",
518                  oh->ot_id_cnt);
519
520         for (i = 0; i < oh->ot_id_cnt; i++) {
521                 if (oh->ot_id_array[i] == qi->lqi_id.qid_uid &&
522                     osd_qid_type(oh, i) == qi->lqi_type) {
523                         found = true;
524                         break;
525                 }
526         }
527
528         if (!found) {
529                 /* we need to account for credits for this new ID */
530                 if (i >= OSD_MAX_UGID_CNT) {
531                         CERROR("Too many(%d) trans qids!\n", i + 1);
532                         RETURN(-EOVERFLOW);
533                 }
534
535                 osd_trans_declare_op(env, oh, OSD_OT_QUOTA,
536                                      (allocated || qi->lqi_id.qid_uid == 0) ?
537                                      1: LDISKFS_QUOTA_INIT_BLOCKS(osd_sb(dev)));
538
539                 oh->ot_id_array[i] = qi->lqi_id.qid_uid;
540                 osd_qid_set_type(oh, i, qi->lqi_type);
541                 oh->ot_id_cnt++;
542         }
543
544         if (unlikely(qsd == NULL))
545                 /* quota slave instance hasn't been allocated yet */
546                 RETURN(0);
547
548         /* check quota */
549         rc = qsd_op_begin(env, qsd, oh->ot_quota_trans, qi, flags);
550         RETURN(rc);
551 }
552
553 /**
554  * Wrapper for osd_declare_qid()
555  *
556  * \param  env    - the environment passed by the caller
557  * \param  uid    - user id of the inode
558  * \param  gid    - group id of the inode
559  * \param  space  - how many blocks/inodes will be consumed/released
560  * \param  oh     - osd transaction handle
561  * \param  is_blk - block quota or inode quota?
562  * \param  allocated - dquot entry in quota accounting file has been allocated
563  * \param  flags  - if the operation is write, return no user quota, no
564  *                  group quota, or sync commit flags to the caller
565  * \param force   - set to 1 when changes are performed by root user and thus
566  *                  can't failed with EDQUOT
567  *
568  * \retval 0      - success
569  * \retval -ve    - failure
570  */
571 int osd_declare_inode_qid(const struct lu_env *env, qid_t uid, qid_t gid,
572                           long long space, struct osd_thandle *oh,
573                           bool is_blk, bool allocated, int *flags, bool force)
574 {
575         struct osd_thread_info  *info = osd_oti_get(env);
576         struct lquota_id_info   *qi = &info->oti_qi;
577         int                      rcu, rcg; /* user & group rc */
578         ENTRY;
579
580         /* let's start with user quota */
581         qi->lqi_id.qid_uid = uid;
582         qi->lqi_type       = USRQUOTA;
583         qi->lqi_space      = space;
584         qi->lqi_is_blk     = is_blk;
585         rcu = osd_declare_qid(env, oh, qi, allocated, flags);
586
587         if (force && (rcu == -EDQUOT || rcu == -EINPROGRESS))
588                 /* ignore EDQUOT & EINPROGRESS when changes are done by root */
589                 rcu = 0;
590
591         /* For non-fatal error, we want to continue to get the noquota flags
592          * for group id. This is only for commit write, which has @flags passed
593          * in. See osd_declare_write_commit().
594          * When force is set to true, we also want to proceed with the gid */
595         if (rcu && (rcu != -EDQUOT || flags == NULL))
596                 RETURN(rcu);
597
598         /* and now group quota */
599         qi->lqi_id.qid_gid = gid;
600         qi->lqi_type       = GRPQUOTA;
601         rcg = osd_declare_qid(env, oh, qi, allocated, flags);
602
603         if (force && (rcg == -EDQUOT || rcg == -EINPROGRESS))
604                 /* as before, ignore EDQUOT & EINPROGRESS for root */
605                 rcg = 0;
606
607         RETURN(rcu ? rcu : rcg);
608 }
609
610 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 7, 53, 0)
611 /* Following code is used to migrate old admin quota files (in Linux quota
612  * file v2 format) into the new quota global indexes (in IAM format). */
613
614 /* copied from osd_it_acct_get(), only changed the 'type' to -1 */
615 static int osd_it_admin_get(const struct lu_env *env, struct dt_it *di,
616                             const struct dt_key *key)
617 {
618         struct osd_it_quota     *it = (struct osd_it_quota *)di;
619         int                      type = -1;
620         qid_t                    dqid = *(qid_t *)key;
621         loff_t                   offset;
622         int                      rc;
623         ENTRY;
624
625         offset = find_tree_dqentry(env, it->oiq_obj, type, dqid,
626                                    LUSTRE_DQTREEOFF, 0, it);
627         if (offset > 0) { /* Found */
628                 RETURN(+1);
629         } else if (offset < 0) { /* Error */
630                 QUOTA_IT_READ_ERROR(it, (int)offset);
631                 RETURN((int)offset);
632         }
633
634         /* The @key is not found, move to the first valid entry */
635         rc = walk_tree_dqentry(env, it->oiq_obj, type, it->oiq_blk[0], 0,
636                                0, it);
637         if (rc > 0)
638                 /* no valid entry found */
639                 rc = -ENOENT;
640         RETURN(rc);
641 }
642
643 static int osd_it_admin_load(const struct lu_env *env,
644                              const struct dt_it *di, __u64 hash)
645 {
646         int rc;
647         ENTRY;
648
649         rc = osd_it_admin_get(env, (struct dt_it *)di,
650                               (const struct dt_key *)&hash);
651         RETURN(rc);
652 }
653
654 static int osd_it_admin_rec(const struct lu_env *env,
655                             const struct dt_it *di,
656                             struct dt_rec *dtrec, __u32 attr)
657 {
658         struct osd_it_quota     *it = (struct osd_it_quota *)di;
659         struct lu_buf            buf;
660         loff_t                   pos;
661         int                      rc;
662         struct lustre_disk_dqblk_v2 *dqblk =
663                 (struct lustre_disk_dqblk_v2 *)dtrec;
664         ENTRY;
665
666         buf.lb_buf = dqblk;
667         buf.lb_len = sizeof(*dqblk);
668
669         pos = it->oiq_offset;
670         rc = dt_record_read(env, &it->oiq_obj->oo_dt, &buf, &pos);
671         RETURN(rc);
672 }
673
674 /* copied from osd_it_acct_next(), only changed the 'type' to -1 */
675 static int osd_it_admin_next(const struct lu_env *env, struct dt_it *di)
676 {
677         struct osd_it_quota     *it = (struct osd_it_quota *)di;
678         int                      type = -1;
679         int                      depth, rc;
680         uint                     index;
681         ENTRY;
682
683         /* Let's first check if there are any remaining valid entry in the
684          * current leaf block. Start with the next entry after the current one.
685          */
686         depth = LUSTRE_DQTREEDEPTH;
687         index = it->oiq_index[depth];
688         if (++index < LUSTRE_DQSTRINBLK) {
689                 /* Search for the next valid entry from current index */
690                 rc = walk_block_dqentry(env, it->oiq_obj, type,
691                                         it->oiq_blk[depth], index, it);
692                 if (rc < 0) {
693                         QUOTA_IT_READ_ERROR(it, rc);
694                         RETURN(rc);
695                 } else if (rc == 0) {
696                         /* Found on entry, @it is already updated to the
697                          * new position in walk_block_dqentry(). */
698                         RETURN(0);
699                 } else {
700                         rc = osd_it_add_processed(it, depth);
701                         if (rc)
702                                 RETURN(rc);
703                 }
704         } else {
705                 rc = osd_it_add_processed(it, depth);
706                 if (rc)
707                         RETURN(rc);
708         }
709         rc = 1;
710
711         /* We have consumed all the entries of the current leaf block, move on
712          * to the next one. */
713         depth--;
714
715         /* We keep searching as long as walk_tree_dqentry() returns +1
716          * (= no valid entry found). */
717         for (; depth >= 0 && rc > 0; depth--) {
718                 index = it->oiq_index[depth];
719                 if (++index > 0xff)
720                         continue;
721                 rc = walk_tree_dqentry(env, it->oiq_obj, type,
722                                        it->oiq_blk[depth], depth, index, it);
723         }
724
725         if (rc < 0)
726                 QUOTA_IT_READ_ERROR(it, rc);
727         RETURN(rc);
728 }
729
730 const struct dt_index_operations osd_admin_index_ops = {
731         .dio_lookup     = osd_acct_index_lookup,
732         .dio_it         = {
733                 .init     = osd_it_acct_init,
734                 .fini     = osd_it_acct_fini,
735                 .get      = osd_it_admin_get,
736                 .put      = osd_it_acct_put,
737                 .next     = osd_it_admin_next,
738                 .key      = osd_it_acct_key,
739                 .key_size = osd_it_acct_key_size,
740                 .rec      = osd_it_admin_rec,
741                 .store    = osd_it_acct_store,
742                 .load     = osd_it_admin_load
743         }
744 };
745
746 static int convert_quota_file(const struct lu_env *env,
747                               struct dt_object *old, struct dt_object *new,
748                               bool isblk)
749 {
750         const struct dt_it_ops  *iops = &old->do_index_ops->dio_it;
751         struct osd_object       *obj;
752         struct lu_buf            buf;
753         struct dt_it            *it;
754         struct dt_key           *key;
755         __u32                    grace;
756         struct lquota_glb_rec   *glb_rec = NULL;
757         loff_t                   pos;
758         int                      rc;
759         struct lustre_disk_dqblk_v2     *dqblk = NULL;
760         struct lustre_disk_dqinfo       *dqinfo = NULL;
761         ENTRY;
762
763         obj = osd_dt_obj(old);
764         LASSERT(obj->oo_inode);
765
766         if (i_size_read(obj->oo_inode) == 0)
767                 RETURN(0);
768
769         /* allocate buffers */
770         OBD_ALLOC_PTR(dqinfo);
771         if (dqinfo == NULL)
772                 RETURN(-ENOMEM);
773
774         OBD_ALLOC_PTR(glb_rec);
775         if (glb_rec == NULL)
776                 GOTO(out, rc = -ENOMEM);
777
778         OBD_ALLOC_PTR(dqblk);
779         if (dqblk == NULL)
780                 GOTO(out, rc = -ENOMEM);
781
782         /* convert the old igrace/bgrace */
783         buf.lb_buf = dqinfo;
784         buf.lb_len = sizeof(*dqinfo);
785         pos = LUSTRE_DQINFOOFF;
786
787         rc = dt_record_read(env, old, &buf, &pos);
788         if (rc)
789                 GOTO(out, rc);
790
791         /* keep it in little endian */
792         grace = isblk ? dqinfo->dqi_bgrace : dqinfo->dqi_igrace;
793         if (grace != 0) {
794                 glb_rec->qbr_time = grace;
795                 rc = lquota_disk_write_glb(env, new, 0, glb_rec);
796                 if (rc)
797                         GOTO(out, rc);
798                 glb_rec->qbr_time = 0;
799         }
800
801         /* iterate the old admin file, insert each record into the
802          * new index file. */
803         it = iops->init(env, old, 0, BYPASS_CAPA);
804         if (IS_ERR(it))
805                 GOTO(out, rc = PTR_ERR(it));
806
807         rc = iops->load(env, it, 0);
808         if (rc == -ENOENT)
809                 GOTO(out_it, rc = 0);
810         else if (rc < 0)
811                 GOTO(out_it, rc);
812
813         do {
814                 key = iops->key(env, it);
815                 if (IS_ERR(key))
816                         GOTO(out_it, rc = PTR_ERR(key));
817
818                 /* skip the root user/group */
819                 if (*((__u64 *)key) == 0)
820                         goto next;
821
822                 rc = iops->rec(env, it, (struct dt_rec *)dqblk, 0);
823                 if (rc)
824                         GOTO(out_it, rc);
825
826                 /* keep the value in little endian */
827                 glb_rec->qbr_hardlimit = isblk ? dqblk->dqb_bhardlimit :
828                                                  dqblk->dqb_ihardlimit;
829                 glb_rec->qbr_softlimit = isblk ? dqblk->dqb_bsoftlimit :
830                                                  dqblk->dqb_isoftlimit;
831
832                 rc = lquota_disk_write_glb(env, new, *((__u64 *)key), glb_rec);
833                 if (rc)
834                         GOTO(out_it, rc);
835 next:
836                 rc = iops->next(env, it);
837         } while (rc == 0);
838
839         /* reach the end */
840         if (rc > 0)
841                 rc = 0;
842
843 out_it:
844         iops->put(env, it);
845         iops->fini(env, it);
846 out:
847         if (dqblk != NULL)
848                 OBD_FREE_PTR(dqblk);
849         if (glb_rec != NULL)
850                 OBD_FREE_PTR(glb_rec);
851         if (dqinfo != NULL)
852                 OBD_FREE_PTR(dqinfo);
853         return rc;
854 }
855
856 /* Nobdy else can access the global index now, it's safe to truncate and
857  * reinitialize it */
858 static int truncate_quota_index(const struct lu_env *env, struct dt_object *dt,
859                                 const struct dt_index_features *feat)
860 {
861         struct osd_device       *osd = osd_obj2dev(osd_dt_obj(dt));
862         struct thandle          *th;
863         struct lu_attr          *attr;
864         struct osd_thandle      *oth;
865         struct inode            *inode;
866         int                      rc;
867         struct iam_container    *bag = &(osd_dt_obj(dt))->oo_dir->od_container;
868         struct lu_buf           *lb = &osd_oti_get(env)->oti_buf;
869         ENTRY;
870
871         LASSERT(bag->ic_root_bh != NULL);
872         iam_container_fini(bag);
873
874         LASSERT(fid_seq(lu_object_fid(&dt->do_lu)) == FID_SEQ_QUOTA_GLB);
875
876         OBD_ALLOC_PTR(attr);
877         if (attr == NULL)
878                 RETURN(-ENOMEM);
879
880         attr->la_size = 0;
881         attr->la_valid = LA_SIZE;
882
883         th = dt_trans_create(env, &osd->od_dt_dev);
884         if (IS_ERR(th)) {
885                 OBD_FREE_PTR(attr);
886                 RETURN(PTR_ERR(th));
887         }
888
889         rc = dt_declare_punch(env, dt, 0, OBD_OBJECT_EOF, th);
890         if (rc)
891                 GOTO(out, rc);
892
893         rc = dt_declare_attr_set(env, dt, attr, th);
894         if (rc)
895                 GOTO(out, rc);
896
897         inode = osd_dt_obj(dt)->oo_inode;
898         LASSERT(inode);
899
900         /* iam_lfix_create() writes two blocks at the beginning */
901         lb->lb_len = osd_sb(osd)->s_blocksize * 2;
902         rc = dt_declare_record_write(env, dt, lb, 0, th);
903         if (rc)
904                 GOTO(out, rc);
905
906         rc = dt_trans_start_local(env, &osd->od_dt_dev, th);
907         if (rc)
908                 GOTO(out, rc);
909
910         dt_write_lock(env, dt, 0);
911         rc = dt_punch(env, dt, 0, OBD_OBJECT_EOF, th, BYPASS_CAPA);
912         if (rc)
913                 GOTO(out_lock, rc);
914
915         rc = dt_attr_set(env, dt, attr, th, BYPASS_CAPA);
916         if (rc)
917                 GOTO(out_lock, rc);
918
919         oth = container_of(th, struct osd_thandle, ot_super);
920
921         if (feat->dif_flags & DT_IND_VARKEY)
922                 rc = iam_lvar_create(osd_dt_obj(dt)->oo_inode,
923                                      feat->dif_keysize_max,
924                                      feat->dif_ptrsize,
925                                      feat->dif_recsize_max, oth->ot_handle);
926         else
927                 rc = iam_lfix_create(osd_dt_obj(dt)->oo_inode,
928                                      feat->dif_keysize_max,
929                                      feat->dif_ptrsize,
930                                      feat->dif_recsize_max, oth->ot_handle);
931 out_lock:
932         dt_write_unlock(env, dt);
933 out:
934         dt_trans_stop(env, &osd->od_dt_dev, th);
935         OBD_FREE_PTR(attr);
936
937         if (rc == 0) {
938                 rc  = iam_container_setup(bag);
939                 if (rc != 0)
940                         iam_container_fini(bag);
941         }
942         RETURN(rc);
943 }
944
945 static int set_quota_index_version(const struct lu_env *env,
946                                    struct dt_object *dt,
947                                    dt_obj_version_t version)
948 {
949         struct osd_device       *osd = osd_obj2dev(osd_dt_obj(dt));
950         struct thandle          *th;
951         int                      rc;
952         ENTRY;
953
954         th = dt_trans_create(env, &osd->od_dt_dev);
955         if (IS_ERR(th))
956                 RETURN(PTR_ERR(th));
957
958         rc = dt_declare_version_set(env, dt, th);
959         if (rc)
960                 GOTO(out, rc);
961
962         rc = dt_trans_start_local(env, &osd->od_dt_dev, th);
963         if (rc)
964                 GOTO(out, rc);
965
966         th->th_sync = 1;
967         dt_version_set(env, dt, version, th);
968 out:
969         dt_trans_stop(env, &osd->od_dt_dev, th);
970         RETURN(rc);
971 }
972
973 int osd_quota_migration(const struct lu_env *env, struct dt_object *dt,
974                         const struct dt_index_features *feat)
975 {
976         struct osd_thread_info  *oti = osd_oti_get(env);
977         struct osd_device       *osd = osd_obj2dev(osd_dt_obj(dt));
978         struct dt_object        *root, *parent = NULL, *admin = NULL;
979         dt_obj_version_t         version;
980         char                    *fname;
981         bool                     isblk = false, converted = false;
982         int                      rc;
983         ENTRY;
984
985         /* not newly created global index */
986         version = dt_version_get(env, dt);
987         if (version != 0)
988                 RETURN(0);
989
990         /* locate root */
991         rc = dt_root_get(env, &osd->od_dt_dev, &oti->oti_fid);
992         if (rc) {
993                 CERROR("%s: Can't get root FID, rc:%d\n", osd->od_svname, rc);
994                 RETURN(rc);
995         }
996
997         root = dt_locate(env, &osd->od_dt_dev, &oti->oti_fid);
998         if (IS_ERR(root)) {
999                 CERROR("%s: Failed to locate root "DFID", rc:%ld\n",
1000                        osd->od_svname, PFID(&oti->oti_fid), PTR_ERR(root));
1001                 RETURN(PTR_ERR(root));
1002         }
1003
1004         /* locate /OBJECTS */
1005         rc = dt_lookup_dir(env, root, OBJECTS, &oti->oti_fid);
1006         if (rc == -ENOENT) {
1007                 GOTO(out, rc = 0);
1008         } else if (rc) {
1009                 CERROR("%s: Failed to lookup %s, rc:%d\n",
1010                        osd->od_svname, OBJECTS, rc);
1011                 GOTO(out, rc);
1012         }
1013
1014         parent = dt_locate(env, &osd->od_dt_dev, &oti->oti_fid);
1015         if (IS_ERR(parent)) {
1016                 CERROR("%s: Failed to locate %s "DFID", rc:%ld\n",
1017                        osd->od_svname, OBJECTS, PFID(&oti->oti_fid),
1018                        PTR_ERR(parent));
1019                 GOTO(out, rc = PTR_ERR(parent));
1020         }
1021
1022         /* locate quota admin file */
1023         if (feat == &dt_quota_iusr_features) {
1024                 fname = ADMIN_USR;
1025                 isblk = false;
1026         } else if (feat == &dt_quota_busr_features) {
1027                 fname = ADMIN_USR;
1028                 isblk = true;
1029         } else if (feat == &dt_quota_igrp_features) {
1030                 fname = ADMIN_GRP;
1031                 isblk = false;
1032         } else {
1033                 fname = ADMIN_GRP;
1034                 isblk = true;
1035         }
1036
1037         rc = dt_lookup_dir(env, parent, fname, &oti->oti_fid);
1038         if (rc == -ENOENT) {
1039                 GOTO(out, rc = 0);
1040         } else if (rc) {
1041                 CERROR("%s: Failed to lookup %s, rc:%d\n",
1042                        osd->od_svname, fname, rc);
1043                 GOTO(out, rc);
1044         }
1045
1046         admin = dt_locate(env, &osd->od_dt_dev, &oti->oti_fid);
1047         if (IS_ERR(admin)) {
1048                 CERROR("%s: Failed to locate %s "DFID", rc:%d\n",
1049                        osd->od_svname, fname, PFID(&oti->oti_fid), rc);
1050                 GOTO(out, rc = PTR_ERR(admin));
1051         }
1052
1053         if (!dt_object_exists(admin)) {
1054                 CERROR("%s: Old admin file %s doesn't exist, but is still "
1055                        " referenced in parent directory.\n",
1056                        osd->od_svname, fname);
1057                 GOTO(out, rc = -ENOENT);
1058         }
1059
1060         /* truncate the new quota index file in case of any leftovers
1061          * from last failed migration */
1062         rc = truncate_quota_index(env, dt, feat);
1063         if (rc) {
1064                 CERROR("%s: Failed to truncate the quota index "DFID", rc:%d\n",
1065                        osd->od_svname, PFID(lu_object_fid(&dt->do_lu)), rc);
1066                 GOTO(out, rc);
1067         }
1068
1069         /* set up indexing operations for the admin file */
1070         admin->do_index_ops = &osd_admin_index_ops;
1071
1072         LCONSOLE_INFO("%s: Migrate %s quota from old admin quota file(%s) to "
1073                       "new IAM quota index("DFID").\n", osd->od_svname,
1074                       isblk ? "block" : "inode", fname,
1075                       PFID(lu_object_fid(&dt->do_lu)));
1076
1077         /* iterate the admin quota file, and insert each record into
1078          * the new index file */
1079         rc = convert_quota_file(env, admin, dt, isblk);
1080         if (rc)
1081                 CERROR("%s: Migrate old admin quota file(%s) failed, rc:%d\n",
1082                        osd->od_svname, fname, rc);
1083         converted = true;
1084 out:
1085         /* if no migration happen, we need to set the default grace time. */
1086         if (!converted && rc == 0) {
1087                 struct lquota_glb_rec *rec = &oti->oti_quota_rec.lqr_glb_rec;
1088
1089                 rec->qbr_hardlimit = 0;
1090                 rec->qbr_softlimit = 0;
1091                 rec->qbr_granted = 0;
1092                 rec->qbr_time = isblk ? MAX_DQ_TIME : MAX_IQ_TIME;
1093
1094                 rc = lquota_disk_write_glb(env, dt, 0, rec);
1095                 if (rc)
1096                         CERROR("%s: Failed to set default grace time for "
1097                                "index("DFID"), rc:%d\n", osd->od_svname,
1098                                PFID(lu_object_fid(&dt->do_lu)), rc);
1099         }
1100
1101         /* bump index version to 1 (or 2 if migration happened), so the
1102          * migration will be skipped next time. */
1103         if (rc == 0) {
1104                 rc = set_quota_index_version(env , dt, converted ? 2 : 1);
1105                 if (rc)
1106                         CERROR("%s: Failed to set quota index("DFID") "
1107                                "version, rc:%d\n", osd->od_svname,
1108                                PFID(lu_object_fid(&dt->do_lu)), rc);
1109         }
1110
1111         if (admin && !IS_ERR(admin))
1112                 lu_object_put(env, &admin->do_lu);
1113         if (parent && !IS_ERR(parent))
1114                 lu_object_put(env, &parent->do_lu);
1115         lu_object_put(env, &root->do_lu);
1116
1117         RETURN(rc);
1118 }
1119 #endif /* LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 7, 53, 0) */